From e4c71425c96072ee57a96c7b6206fffa5301e4dc Mon Sep 17 00:00:00 2001 From: Sergey Lyubka Date: Tue, 24 Jun 2014 17:04:26 +0100 Subject: [PATCH] Added big_upload.c --- examples/big_upload.c | 88 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 88 insertions(+) create mode 100644 examples/big_upload.c diff --git a/examples/big_upload.c b/examples/big_upload.c new file mode 100644 index 00000000..b9e28573 --- /dev/null +++ b/examples/big_upload.c @@ -0,0 +1,88 @@ +#include +#include +#include +#include "mongoose.h" + +struct file_data { + FILE *fp; + const char *data; + int data_len; + int written; +}; + +static int handle_request(struct mg_connection *conn) { + const char *data; + int data_len; + char var_name[100], file_name[100], path[100]; + + if (strcmp(conn->uri, "/upload") == 0) { + if (mg_parse_multipart(conn->content, conn->content_len, + var_name, sizeof(var_name), + file_name, sizeof(file_name), + &data, &data_len) > 0) { + struct file_data *p = (struct file_data *) malloc(sizeof(*p)); + snprintf(path, sizeof(path), "UPLOAD_%s", file_name); + p->fp = fopen(path, "wb"); + p->data = data; + p->data_len = data_len; + p->written = 0; + conn->connection_param = p; + mg_send_header(conn, "Content-Type", "text/html"); + } + return MG_MORE; + } else { + mg_printf_data(conn, "%s", + "Upload example." + "
" + "
" + "" + "
"); + return MG_TRUE; + } +} + +static int handle_poll(struct mg_connection *conn) { + struct file_data *p = (struct file_data *) conn->connection_param; + if (p != NULL) { + // Write no more then 100 bytes in one go + int len = p->data_len - p->written; + int n = fwrite(p->data + p->written, 1, len > 100 ? 100 : len, p->fp); + if (n > 0) { + p->written += n; + mg_send_data(conn, " ", 1); // Send something back to wake up select() + } + + // If everything is written, close the connection + if (p->written >= p->data_len) { + mg_printf_data(conn, "Written %d bytes.", p->written); + fclose(p->fp); + free(p); + conn->connection_param = NULL; + return MG_TRUE; + } + } + return MG_FALSE; +} + +static int ev_handler(struct mg_connection *conn, enum mg_event ev) { + switch (ev) { + case MG_AUTH: return MG_TRUE; + case MG_REQUEST: return handle_request(conn); + case MG_POLL: return handle_poll(conn); + default: return MG_FALSE; + } +} + +int main(void) { + struct mg_server *server = mg_create_server(NULL, ev_handler); + mg_set_option(server, "listening_port", "8080"); + printf("Starting on port %s\n", mg_get_option(server, "listening_port")); + + for (;;) { + mg_poll_server(server, 1000); + } + + mg_destroy_server(&server); + return 0; +}