Blame src/examples/fileserver_example_external_select.c

Packit 875988
/*
Packit 875988
     This file is part of libmicrohttpd
Packit 875988
     Copyright (C) 2007, 2008 Christian Grothoff (and other contributing authors)
Packit 875988
Packit 875988
     This library is free software; you can redistribute it and/or
Packit 875988
     modify it under the terms of the GNU Lesser General Public
Packit 875988
     License as published by the Free Software Foundation; either
Packit 875988
     version 2.1 of the License, or (at your option) any later version.
Packit 875988
Packit 875988
     This library is distributed in the hope that it will be useful,
Packit 875988
     but WITHOUT ANY WARRANTY; without even the implied warranty of
Packit 875988
     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
Packit 875988
     Lesser General Public License for more details.
Packit 875988
Packit 875988
     You should have received a copy of the GNU Lesser General Public
Packit 875988
     License along with this library; if not, write to the Free Software
Packit 875988
     Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
Packit 875988
*/
Packit 875988
/**
Packit 875988
 * @file fileserver_example_external_select.c
Packit 875988
 * @brief minimal example for how to use libmicrohttpd to server files
Packit 875988
 * @author Christian Grothoff
Packit 875988
 */
Packit 875988
Packit 875988
#include "platform.h"
Packit 875988
#include <microhttpd.h>
Packit 875988
#include <sys/stat.h>
Packit 875988
#include <unistd.h>
Packit 875988
Packit 875988
#define PAGE "<html><head><title>File not found</title></head><body>File not found</body></html>"
Packit 875988
Packit 875988
static ssize_t
Packit 875988
file_reader (void *cls, uint64_t pos, char *buf, size_t max)
Packit 875988
{
Packit 875988
  FILE *file = cls;
Packit 875988
Packit 875988
  (void) fseek (file, pos, SEEK_SET);
Packit 875988
  return fread (buf, 1, max, file);
Packit 875988
}
Packit 875988
Packit 875988
Packit 875988
static void
Packit 875988
free_callback (void *cls)
Packit 875988
{
Packit 875988
  FILE *file = cls;
Packit 875988
  fclose (file);
Packit 875988
}
Packit 875988
Packit 875988
Packit 875988
static int
Packit 875988
ahc_echo (void *cls,
Packit 875988
          struct MHD_Connection *connection,
Packit 875988
          const char *url,
Packit 875988
          const char *method,
Packit 875988
          const char *version,
Packit 875988
          const char *upload_data,
Packit 875988
	  size_t *upload_data_size, void **ptr)
Packit 875988
{
Packit 875988
  static int aptr;
Packit 875988
  struct MHD_Response *response;
Packit 875988
  int ret;
Packit 875988
  FILE *file;
Packit 875988
  int fd;
Packit 875988
  struct stat buf;
Packit 875988
  (void)cls;               /* Unused. Silent compiler warning. */
Packit 875988
  (void)version;           /* Unused. Silent compiler warning. */
Packit 875988
  (void)upload_data;       /* Unused. Silent compiler warning. */
Packit 875988
  (void)upload_data_size;  /* Unused. Silent compiler warning. */
Packit 875988
Packit 875988
  if (0 != strcmp (method, MHD_HTTP_METHOD_GET))
Packit 875988
    return MHD_NO;              /* unexpected method */
Packit 875988
  if (&aptr != *ptr)
Packit 875988
    {
Packit 875988
      /* do never respond on first call */
Packit 875988
      *ptr = &apt;;
Packit 875988
      return MHD_YES;
Packit 875988
    }
Packit 875988
  *ptr = NULL;                  /* reset when done */
Packit 875988
Packit 875988
  file = fopen (&url[1], "rb");
Packit 875988
  if (NULL != file)
Packit 875988
    {
Packit 875988
      fd = fileno (file);
Packit 875988
      if (-1 == fd)
Packit 875988
        {
Packit 875988
          (void) fclose (file);
Packit 875988
          return MHD_NO; /* internal error */
Packit 875988
        }
Packit 875988
      if ( (0 != fstat (fd, &buf)) ||
Packit 875988
           (! S_ISREG (buf.st_mode)) )
Packit 875988
        {
Packit 875988
          /* not a regular file, refuse to serve */
Packit 875988
          fclose (file);
Packit 875988
          file = NULL;
Packit 875988
        }
Packit 875988
    }
Packit 875988
Packit 875988
  if (NULL == file)
Packit 875988
    {
Packit 875988
      response = MHD_create_response_from_buffer (strlen (PAGE),
Packit 875988
						  (void *) PAGE,
Packit 875988
						  MHD_RESPMEM_PERSISTENT);
Packit 875988
      ret = MHD_queue_response (connection, MHD_HTTP_NOT_FOUND, response);
Packit 875988
      MHD_destroy_response (response);
Packit 875988
    }
Packit 875988
  else
Packit 875988
    {
Packit 875988
      response = MHD_create_response_from_callback (buf.st_size, 32 * 1024,     /* 32k page size */
Packit 875988
                                                    &file_reader,
Packit 875988
                                                    file,
Packit 875988
                                                    &free_callback);
Packit 875988
      if (NULL == response)
Packit 875988
	{
Packit 875988
	  fclose (file);
Packit 875988
	  return MHD_NO;
Packit 875988
	}
Packit 875988
      ret = MHD_queue_response (connection, MHD_HTTP_OK, response);
Packit 875988
      MHD_destroy_response (response);
Packit 875988
    }
Packit 875988
  return ret;
Packit 875988
}
Packit 875988
Packit 875988
Packit 875988
int
Packit 875988
main (int argc, char *const *argv)
Packit 875988
{
Packit 875988
  struct MHD_Daemon *d;
Packit 875988
  time_t end;
Packit 875988
  time_t t;
Packit 875988
  struct timeval tv;
Packit 875988
  fd_set rs;
Packit 875988
  fd_set ws;
Packit 875988
  fd_set es;
Packit 875988
  MHD_socket max;
Packit 875988
  MHD_UNSIGNED_LONG_LONG mhd_timeout;
Packit 875988
Packit 875988
  if (argc != 3)
Packit 875988
    {
Packit 875988
      printf ("%s PORT SECONDS-TO-RUN\n", argv[0]);
Packit 875988
      return 1;
Packit 875988
    }
Packit 875988
  d = MHD_start_daemon (MHD_USE_ERROR_LOG,
Packit 875988
                        atoi (argv[1]),
Packit 875988
                        NULL, NULL, &ahc_echo, PAGE, MHD_OPTION_END);
Packit 875988
  if (d == NULL)
Packit 875988
    return 1;
Packit 875988
  end = time (NULL) + atoi (argv[2]);
Packit 875988
  while ((t = time (NULL)) < end)
Packit 875988
    {
Packit 875988
      tv.tv_sec = end - t;
Packit 875988
      tv.tv_usec = 0;
Packit 875988
      max = 0;
Packit 875988
      FD_ZERO (&rs);
Packit 875988
      FD_ZERO (&ws);
Packit 875988
      FD_ZERO (&es);
Packit 875988
      if (MHD_YES != MHD_get_fdset (d, &rs, &ws, &es, &max))
Packit 875988
	break; /* fatal internal error */
Packit 875988
      if (MHD_get_timeout (d, &mhd_timeout) == MHD_YES)
Packit 875988
        {
Packit 875988
          if (((MHD_UNSIGNED_LONG_LONG)tv.tv_sec) < mhd_timeout / 1000LL)
Packit 875988
            {
Packit 875988
              tv.tv_sec = mhd_timeout / 1000LL;
Packit 875988
              tv.tv_usec = (mhd_timeout - (tv.tv_sec * 1000LL)) * 1000LL;
Packit 875988
            }
Packit 875988
        }
Packit 875988
      if (-1 == select (max + 1, &rs, &ws, &es, &tv))
Packit 875988
        {
Packit 875988
          if (EINTR != errno)
Packit 875988
            abort ();
Packit 875988
        }
Packit 875988
      MHD_run (d);
Packit 875988
    }
Packit 875988
  MHD_stop_daemon (d);
Packit 875988
  return 0;
Packit 875988
}