mirror of
https://git.gnunet.org/libmicrohttpd.git
synced 2026-09-25 04:09:31 +03:00
296 lines
7.4 KiB
C
296 lines
7.4 KiB
C
/* Feel free to use this example code in any way
|
|
you see fit (Public Domain) */
|
|
|
|
#include <sys/types.h>
|
|
#ifndef _WIN32
|
|
#include <sys/select.h>
|
|
#include <sys/socket.h>
|
|
#else
|
|
#include <winsock2.h>
|
|
#endif
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <microhttpd.h>
|
|
|
|
#ifdef _MSC_VER
|
|
#ifndef strcasecmp
|
|
#define strcasecmp(a,b) _stricmp((a),(b))
|
|
#endif /* !strcasecmp */
|
|
#endif /* _MSC_VER */
|
|
|
|
#if defined(_MSC_VER) && _MSC_VER+0 <= 1800
|
|
/* Substitution is OK while return value is not used */
|
|
#define snprintf _snprintf
|
|
#endif
|
|
|
|
#define PORT 8888
|
|
#define POSTBUFFERSIZE 512
|
|
#define MAXCLIENTS 2
|
|
|
|
enum ConnectionType
|
|
{
|
|
GET = 0,
|
|
POST = 1
|
|
};
|
|
|
|
static unsigned int nr_of_uploading_clients = 0;
|
|
|
|
struct connection_info_struct
|
|
{
|
|
enum ConnectionType connectiontype;
|
|
struct MHD_PostProcessor *postprocessor;
|
|
FILE *fp;
|
|
const char *answerstring;
|
|
int answercode;
|
|
};
|
|
|
|
const char *askpage = "<html><body>\n\
|
|
Upload a file, please!<br>\n\
|
|
There are %u clients uploading at the moment.<br>\n\
|
|
<form action=\"/filepost\" method=\"post\" enctype=\"multipart/form-data\">\n\
|
|
<input name=\"file\" type=\"file\">\n\
|
|
<input type=\"submit\" value=\" Send \"></form>\n\
|
|
</body></html>";
|
|
|
|
const char *busypage =
|
|
"<html><body>This server is busy, please try again later.</body></html>";
|
|
|
|
const char *completepage =
|
|
"<html><body>The upload has been completed.</body></html>";
|
|
|
|
const char *errorpage =
|
|
"<html><body>This doesn't seem to be right.</body></html>";
|
|
const char *servererrorpage =
|
|
"<html><body>An internal server error has occured.</body></html>";
|
|
const char *fileexistspage =
|
|
"<html><body>This file already exists.</body></html>";
|
|
const char* const postprocerror =
|
|
"<html><head><title>Error</title></head><body>Error processing POST data</body></html>";
|
|
|
|
|
|
static int
|
|
send_page (struct MHD_Connection *connection,
|
|
const char *page,
|
|
int status_code)
|
|
{
|
|
int ret;
|
|
struct MHD_Response *response;
|
|
|
|
response =
|
|
MHD_create_response_from_buffer (strlen (page),
|
|
(void *) page,
|
|
MHD_RESPMEM_MUST_COPY);
|
|
if (!response)
|
|
return MHD_NO;
|
|
MHD_add_response_header (response,
|
|
MHD_HTTP_HEADER_CONTENT_TYPE,
|
|
"text/html");
|
|
ret = MHD_queue_response (connection,
|
|
status_code,
|
|
response);
|
|
MHD_destroy_response (response);
|
|
|
|
return ret;
|
|
}
|
|
|
|
|
|
static int
|
|
iterate_post (void *coninfo_cls,
|
|
enum MHD_ValueKind kind,
|
|
const char *key,
|
|
const char *filename,
|
|
const char *content_type,
|
|
const char *transfer_encoding,
|
|
const char *data,
|
|
uint64_t off,
|
|
size_t size)
|
|
{
|
|
struct connection_info_struct *con_info = coninfo_cls;
|
|
FILE *fp;
|
|
|
|
con_info->answerstring = servererrorpage;
|
|
con_info->answercode = MHD_HTTP_INTERNAL_SERVER_ERROR;
|
|
|
|
if (0 != strcmp (key, "file"))
|
|
return MHD_NO;
|
|
|
|
if (! con_info->fp)
|
|
{
|
|
if (NULL != (fp = fopen (filename, "rb")))
|
|
{
|
|
fclose (fp);
|
|
con_info->answerstring = fileexistspage;
|
|
con_info->answercode = MHD_HTTP_FORBIDDEN;
|
|
return MHD_NO;
|
|
}
|
|
|
|
con_info->fp = fopen (filename, "ab");
|
|
if (!con_info->fp)
|
|
return MHD_NO;
|
|
}
|
|
|
|
if (size > 0)
|
|
{
|
|
if (! fwrite (data, sizeof (char), size, con_info->fp))
|
|
return MHD_NO;
|
|
}
|
|
|
|
con_info->answerstring = completepage;
|
|
con_info->answercode = MHD_HTTP_OK;
|
|
|
|
return MHD_YES;
|
|
}
|
|
|
|
|
|
static void
|
|
request_completed (void *cls,
|
|
struct MHD_Connection *connection,
|
|
void **con_cls,
|
|
enum MHD_RequestTerminationCode toe)
|
|
{
|
|
struct connection_info_struct *con_info = *con_cls;
|
|
|
|
if (NULL == con_info)
|
|
return;
|
|
|
|
if (con_info->connectiontype == POST)
|
|
{
|
|
if (NULL != con_info->postprocessor)
|
|
{
|
|
MHD_destroy_post_processor (con_info->postprocessor);
|
|
nr_of_uploading_clients--;
|
|
}
|
|
|
|
if (con_info->fp)
|
|
fclose (con_info->fp);
|
|
}
|
|
|
|
free (con_info);
|
|
*con_cls = NULL;
|
|
}
|
|
|
|
|
|
static int
|
|
answer_to_connection (void *cls,
|
|
struct MHD_Connection *connection,
|
|
const char *url,
|
|
const char *method,
|
|
const char *version,
|
|
const char *upload_data,
|
|
size_t *upload_data_size,
|
|
void **con_cls)
|
|
{
|
|
if (NULL == *con_cls)
|
|
{
|
|
struct connection_info_struct *con_info;
|
|
|
|
if (nr_of_uploading_clients >= MAXCLIENTS)
|
|
return send_page (connection,
|
|
busypage,
|
|
MHD_HTTP_SERVICE_UNAVAILABLE);
|
|
|
|
con_info = malloc (sizeof (struct connection_info_struct));
|
|
if (NULL == con_info)
|
|
return MHD_NO;
|
|
|
|
con_info->fp = NULL;
|
|
|
|
if (0 == strcasecmp (method, MHD_HTTP_METHOD_POST))
|
|
{
|
|
con_info->postprocessor =
|
|
MHD_create_post_processor (connection,
|
|
POSTBUFFERSIZE,
|
|
&iterate_post,
|
|
(void *) con_info);
|
|
|
|
if (NULL == con_info->postprocessor)
|
|
{
|
|
free (con_info);
|
|
return MHD_NO;
|
|
}
|
|
|
|
nr_of_uploading_clients++;
|
|
|
|
con_info->connectiontype = POST;
|
|
con_info->answercode = MHD_HTTP_OK;
|
|
con_info->answerstring = completepage;
|
|
}
|
|
else
|
|
con_info->connectiontype = GET;
|
|
|
|
*con_cls = (void *) con_info;
|
|
|
|
return MHD_YES;
|
|
}
|
|
|
|
if (0 == strcasecmp (method, MHD_HTTP_METHOD_GET))
|
|
{
|
|
char buffer[1024];
|
|
|
|
snprintf (buffer,
|
|
sizeof (buffer),
|
|
askpage,
|
|
nr_of_uploading_clients);
|
|
return send_page (connection,
|
|
buffer,
|
|
MHD_HTTP_OK);
|
|
}
|
|
|
|
if (0 == strcasecmp (method, MHD_HTTP_METHOD_POST))
|
|
{
|
|
struct connection_info_struct *con_info = *con_cls;
|
|
|
|
if (0 != *upload_data_size)
|
|
{
|
|
if (MHD_post_process (con_info->postprocessor,
|
|
upload_data,
|
|
*upload_data_size) != MHD_YES)
|
|
{
|
|
return send_page (connection,
|
|
postprocerror,
|
|
MHD_HTTP_BAD_REQUEST);
|
|
}
|
|
*upload_data_size = 0;
|
|
|
|
return MHD_YES;
|
|
}
|
|
else
|
|
{
|
|
if (NULL != con_info->fp)
|
|
{
|
|
fclose (con_info->fp);
|
|
con_info->fp = NULL;
|
|
}
|
|
/* Now it is safe to open and inspect the file before
|
|
calling send_page with a response */
|
|
return send_page (connection,
|
|
con_info->answerstring,
|
|
con_info->answercode);
|
|
}
|
|
|
|
}
|
|
|
|
return send_page (connection,
|
|
errorpage,
|
|
MHD_HTTP_BAD_REQUEST);
|
|
}
|
|
|
|
|
|
int
|
|
main ()
|
|
{
|
|
struct MHD_Daemon *daemon;
|
|
|
|
daemon = MHD_start_daemon (MHD_USE_SELECT_INTERNALLY,
|
|
PORT, NULL, NULL,
|
|
&answer_to_connection, NULL,
|
|
MHD_OPTION_NOTIFY_COMPLETED, &request_completed, NULL,
|
|
MHD_OPTION_END);
|
|
if (NULL == daemon)
|
|
return 1;
|
|
(void) getchar ();
|
|
MHD_stop_daemon (daemon);
|
|
return 0;
|
|
}
|