|
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 |
}
|