lib/libhttp
1
0
Fork 0
mirror of https://github.com/lammertb/libhttp.git synced 2025-08-07 16:02:55 +03:00
Code Activity

Add example for parsing uploaded files on the fly

Browse Source
This commit is contained in:
bel
2016-08-20 01:43:14 +02:00
parent 67eb68c5b7
commit c8b8e9f74d
1 changed files with 172 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

182
examples/embedded_c/embedded_c.c
View File

@@ -40,7 +40,9 @@ int exitNow = 0;
int int
ExampleHandler(struct mg_connection *conn, void *cbdata) ExampleHandler(struct mg_connection *conn, void *cbdata)
{ {
mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n"); mg_printf(conn,
"HTTP/1.1 200 OK\r\nContent-Type: text/html\r\nConnection: "
"close\r\n\r\n");
mg_printf(conn, "<html><body>"); mg_printf(conn, "<html><body>");
mg_printf(conn, "<h2>This is an example text from a C handler</h2>"); mg_printf(conn, "<h2>This is an example text from a C handler</h2>");
mg_printf( mg_printf(
@@ -74,6 +76,10 @@ ExampleHandler(struct mg_connection *conn, void *cbdata)
mg_printf(conn, mg_printf(conn,
"<p>To see a page from the CookieHandler handler <a " "<p>To see a page from the CookieHandler handler <a "
"href=\"cookie\">click cookie</a></p>"); "href=\"cookie\">click cookie</a></p>");
mg_printf(conn,
"<p>To see an example for parsing files on the fly <a "
"href=\"on_the_fly_form\">click form</a> (form for "
"uploading files)</p>");
#ifdef USE_WEBSOCKET #ifdef USE_WEBSOCKET
mg_printf(conn, mg_printf(conn,
@@ -89,7 +95,9 @@ ExampleHandler(struct mg_connection *conn, void *cbdata)
int int
ExitHandler(struct mg_connection *conn, void *cbdata) ExitHandler(struct mg_connection *conn, void *cbdata)
{ {
mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/plain\r\n\r\n"); mg_printf(conn,
"HTTP/1.1 200 OK\r\nContent-Type: "
"text/plain\r\nConnection: close\r\n\r\n");
mg_printf(conn, "Server will shut down.\n"); mg_printf(conn, "Server will shut down.\n");
mg_printf(conn, "Bye!\n"); mg_printf(conn, "Bye!\n");
exitNow = 1; exitNow = 1;
@@ -100,7 +108,9 @@ ExitHandler(struct mg_connection *conn, void *cbdata)
int int
AHandler(struct mg_connection *conn, void *cbdata) AHandler(struct mg_connection *conn, void *cbdata)
{ {
mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n"); mg_printf(conn,
"HTTP/1.1 200 OK\r\nContent-Type: text/html\r\nConnection: "
"close\r\n\r\n");
mg_printf(conn, "<html><body>"); mg_printf(conn, "<html><body>");
mg_printf(conn, "<h2>This is the A handler!!!</h2>"); mg_printf(conn, "<h2>This is the A handler!!!</h2>");
mg_printf(conn, "</body></html>\n"); mg_printf(conn, "</body></html>\n");
@@ -111,7 +121,9 @@ AHandler(struct mg_connection *conn, void *cbdata)
int int
ABHandler(struct mg_connection *conn, void *cbdata) ABHandler(struct mg_connection *conn, void *cbdata)
{ {
mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n"); mg_printf(conn,
"HTTP/1.1 200 OK\r\nContent-Type: text/html\r\nConnection: "
"close\r\n\r\n");
mg_printf(conn, "<html><body>"); mg_printf(conn, "<html><body>");
mg_printf(conn, "<h2>This is the AB handler!!!</h2>"); mg_printf(conn, "<h2>This is the AB handler!!!</h2>");
mg_printf(conn, "</body></html>\n"); mg_printf(conn, "</body></html>\n");
@@ -125,7 +137,9 @@ BXHandler(struct mg_connection *conn, void *cbdata)
/* Handler may access the request info using mg_get_request_info */ /* Handler may access the request info using mg_get_request_info */
const struct mg_request_info *req_info = mg_get_request_info(conn); const struct mg_request_info *req_info = mg_get_request_info(conn);
mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n"); mg_printf(conn,
"HTTP/1.1 200 OK\r\nContent-Type: text/html\r\nConnection: "
"close\r\n\r\n");
mg_printf(conn, "<html><body>"); mg_printf(conn, "<html><body>");
mg_printf(conn, "<h2>This is the BX handler %p!!!</h2>", cbdata); mg_printf(conn, "<h2>This is the BX handler %p!!!</h2>", cbdata);
mg_printf(conn, "<p>The actual uri is %s</p>", req_info->uri); mg_printf(conn, "<p>The actual uri is %s</p>", req_info->uri);
@@ -140,7 +154,9 @@ FooHandler(struct mg_connection *conn, void *cbdata)
/* Handler may access the request info using mg_get_request_info */ /* Handler may access the request info using mg_get_request_info */
const struct mg_request_info *req_info = mg_get_request_info(conn); const struct mg_request_info *req_info = mg_get_request_info(conn);
mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n"); mg_printf(conn,
"HTTP/1.1 200 OK\r\nContent-Type: text/html\r\nConnection: "
"close\r\n\r\n");
mg_printf(conn, "<html><body>"); mg_printf(conn, "<html><body>");
mg_printf(conn, "<h2>This is the Foo handler!!!</h2>"); mg_printf(conn, "<h2>This is the Foo handler!!!</h2>");
mg_printf(conn, mg_printf(conn,
@@ -159,7 +175,9 @@ CloseHandler(struct mg_connection *conn, void *cbdata)
/* Handler may access the request info using mg_get_request_info */ /* Handler may access the request info using mg_get_request_info */
const struct mg_request_info *req_info = mg_get_request_info(conn); const struct mg_request_info *req_info = mg_get_request_info(conn);
mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n"); mg_printf(conn,
"HTTP/1.1 200 OK\r\nContent-Type: text/html\r\nConnection: "
"close\r\n\r\n");
mg_printf(conn, "<html><body>"); mg_printf(conn, "<html><body>");
mg_printf(conn, mg_printf(conn,
"<h2>This handler will close the connection in a second</h2>"); "<h2>This handler will close the connection in a second</h2>");
@@ -256,7 +274,9 @@ FormHandler(struct mg_connection *conn, void *cbdata)
* mg_handle_form_request. */ * mg_handle_form_request. */
(void)req_info; (void)req_info;
mg_printf(conn, "HTTP/1.0 200 OK\r\nContent-Type: text/plain\r\n\r\n"); mg_printf(conn,
"HTTP/1.1 200 OK\r\nContent-Type: "
"text/plain\r\nConnection: close\r\n\r\n");
fdh.user_data = (void *)conn; fdh.user_data = (void *)conn;
/* Call the form handler */ /* Call the form handler */
@@ -268,6 +288,136 @@ FormHandler(struct mg_connection *conn, void *cbdata)
} }
int
FileUploadForm(struct mg_connection *conn, void *cbdata)
{
mg_printf(conn,
"HTTP/1.1 200 OK\r\nContent-Type: text/html\r\nConnection: "
"close\r\n\r\n");
mg_printf(conn, "<!DOCTYPE html>\n");
mg_printf(conn, "<html>\n<head>\n");
mg_printf(conn, "<meta charset=\"UTF-8\">\n");
mg_printf(conn, "<title>File upload</title>\n");
mg_printf(conn, "</head>\n<body>\n");
mg_printf(conn,
"<form action=\"%s\" method=\"POST\" "
"enctype=\"multipart/form-data\">\n",
(const char *)cbdata);
mg_printf(conn, "<input type=\"file\" name=\"filesin\" multiple>\n");
mg_printf(conn, "<input type=\"submit\" value=\"Submit\">\n");
mg_printf(conn, "</form>\n</body>\n</html>\n");
return 1;
}
#define MD5_STATIC static
#include "../src/md5.inl"
struct tfile_checksum {
char name[128];
unsigned long long length;
md5_state_t chksum;
};
#define MAX_FILES (10)
struct tfiles_checksums {
int index;
struct tfile_checksum file[MAX_FILES];
};
int
field_disp_read_on_the_fly(const char *key,
const char *filename,
char *path,
size_t pathlen,
void *user_data)
{
struct tfiles_checksums *context = (struct tfiles_checksums *)user_data;
(void)key;
(void)path;
(void)pathlen;
if (context->index < MAX_FILES) {
context->index++;
strncpy(context->file[context->index - 1].name, filename, 128);
context->file[context->index - 1].name[127] = 0;
context->file[context->index - 1].length = 0;
md5_init(&(context->file[context->index - 1].chksum));
return FORM_FIELD_STORAGE_GET;
}
return FORM_FIELD_STORAGE_ABORT;
}
int
field_get_checksum(const char *key,
const char *value,
size_t valuelen,
void *user_data)
{
struct tfiles_checksums *context = (struct tfiles_checksums *)user_data;
char path[1026];
FILE *f;
(void)key;
context->file[context->index - 1].length += valuelen;
md5_append(&(context->file[context->index - 1].chksum),
(const md5_byte_t *)value,
valuelen);
return 0;
}
int
CheckSumHandler(struct mg_connection *conn, void *cbdata)
{
/* Handler may access the request info using mg_get_request_info */
const struct mg_request_info *req_info = mg_get_request_info(conn);
int i, j, ret;
struct tfiles_checksums chksums;
md5_byte_t digest[16];
struct mg_form_data_handler fdh = {field_disp_read_on_the_fly,
field_get_checksum,
0,
(void *)&chksums};
/* It would be possible to check the request info here before calling
* mg_handle_form_request. */
(void)req_info;
memset(&chksums, 0, sizeof(chksums));
mg_printf(conn,
"HTTP/1.1 200 OK\r\n"
"Content-Type: text/plain\r\n"
"Connection: close\r\n\r\n");
/* Call the form handler */
mg_printf(conn, "File checksums:");
ret = mg_handle_form_request(conn, &fdh);
for (i = 0; i < chksums.index; i++) {
md5_finish(&(chksums.file[i].chksum), digest);
/* Visual Studio 2010+ support llu */
mg_printf(conn,
"\r\n%s %llu ",
chksums.file[i].name,
chksums.file[i].length);
for (j = 0; j < 16; j++) {
mg_printf(conn, "%02x", (unsigned int)digest[j]);
}
}
mg_printf(conn, "\r\n%i files\r\n", ret);
return 1;
}
int int
CookieHandler(struct mg_connection *conn, void *cbdata) CookieHandler(struct mg_connection *conn, void *cbdata)
{ {
@@ -280,7 +430,7 @@ CookieHandler(struct mg_connection *conn, void *cbdata)
(void)mg_get_cookie(cookie, "first", first_str, sizeof(first_str)); (void)mg_get_cookie(cookie, "first", first_str, sizeof(first_str));
(void)mg_get_cookie(cookie, "count", count_str, sizeof(count_str)); (void)mg_get_cookie(cookie, "count", count_str, sizeof(count_str));
mg_printf(conn, "HTTP/1.1 200 OK\r\n"); mg_printf(conn, "HTTP/1.1 200 OK\r\nConnection: close\r\n");
if (first_str[0] == 0) { if (first_str[0] == 0) {
time_t t = time(0); time_t t = time(0);
struct tm *ptm = localtime(&t); struct tm *ptm = localtime(&t);
@@ -316,7 +466,9 @@ CookieHandler(struct mg_connection *conn, void *cbdata)
int int
WebSocketStartHandler(struct mg_connection *conn, void *cbdata) WebSocketStartHandler(struct mg_connection *conn, void *cbdata)
{ {
mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n"); mg_printf(conn,
"HTTP/1.1 200 OK\r\nContent-Type: text/html\r\nConnection: "
"close\r\n\r\n");
mg_printf(conn, "<!DOCTYPE html>\n"); mg_printf(conn, "<!DOCTYPE html>\n");
mg_printf(conn, "<html>\n<head>\n"); mg_printf(conn, "<html>\n<head>\n");
@@ -577,6 +729,16 @@ main(int argc, char *argv[])
FormHandler, FormHandler,
(void *)0); (void *)0);
/* Add a file upload handler for parsing files on the fly */
mg_set_request_handler(ctx,
"/on_the_fly_form",
FileUploadForm,
(void *)"/on_the_fly_form.md5.callback");
mg_set_request_handler(ctx,
"/on_the_fly_form.md5.callback",
CheckSumHandler,
(void *)0);
/* Add handler for /cookie example */ /* Add handler for /cookie example */
mg_set_request_handler(ctx, "/cookie", CookieHandler, 0); mg_set_request_handler(ctx, "/cookie", CookieHandler, 0);