/* Feel free to use this example code in any way you see fit (Public Domain) */ #include #ifndef _WIN32 #include #include #else #include #endif #include #include #include #include #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 = "\n\ Upload a file, please!
\n\ There are %u clients uploading at the moment.
\n\ \n\ "; const char *busypage = "This server is busy, please try again later."; const char *completepage = "The upload has been completed."; const char *errorpage = "This doesn't seem to be right."; const char *servererrorpage = "An internal server error has occured."; const char *fileexistspage = "This file already exists."; 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; }