242 lines
6.1 KiB
C
242 lines
6.1 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>
|
|
|
|
#define PORT 8888
|
|
#define POSTBUFFERSIZE 512
|
|
#define MAXCLIENTS 2
|
|
|
|
#define GET 0
|
|
#define POST 1
|
|
|
|
static unsigned int nr_of_uploading_clients = 0;
|
|
|
|
struct connection_info_struct
|
|
{
|
|
int 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>";
|
|
|
|
|
|
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, size, sizeof (char), 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 == strcmp (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 == strcmp (method, "GET"))
|
|
{
|
|
char buffer[1024];
|
|
|
|
snprintf (buffer, sizeof (buffer), askpage, nr_of_uploading_clients);
|
|
return send_page (connection, buffer, MHD_HTTP_OK);
|
|
}
|
|
|
|
if (0 == strcmp (method, "POST"))
|
|
{
|
|
struct connection_info_struct *con_info = *con_cls;
|
|
|
|
if (0 != *upload_data_size)
|
|
{
|
|
MHD_post_process (con_info->postprocessor, upload_data,
|
|
*upload_data_size);
|
|
*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;
|
|
}
|