PSG
/
civetweb
mirror of https://github.com/civetweb/civetweb.git


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
							// Copyright (c) 2004-2012 Sergey Lyubka
// This file is a part of mongoose project, http://github.com/valenok/mongoose

#include <stdio.h>
#include <string.h>
#include <fcntl.h>
#include <stdlib.h>
#ifdef _WIN32
#include <windows.h>
#include <io.h>
#define strtoll strtol
typedef __int64 int64_t;
#define O_CLOEXEC 0
#define O_EXLOCK 0
#else 
#include <inttypes.h>
#include <unistd.h>
#endif // !_WIN32

#include "mongoose.h"

// Make sure that form has enctype="multipart/form-data" attribute
static const char *html_form =
  "<html><body>Upload example."
  "<form method=\"POST\" action=\"/handle_post_request\" "
  "  enctype=\"multipart/form-data\">"
  "<input type=\"file\" name=\"file\" /> <br/>"
  "<input type=\"submit\" value=\"Upload\" />"
  "</form></body></html>";
 
static const char *HTTP_500 = "HTTP/1.0 500 Server Error\r\n\r\n";

static void handle_file_upload(struct mg_connection *conn) {
  const char *cl_header;
  char post_data[16 * 1024], path[999], file_name[1024], mime_type[100],
       buf[BUFSIZ], *eop, *s, *p;
  FILE *fp;
  int64_t cl, written;
  int fd, n, post_data_len;

  // Figure out total content length. Return if it is not present or invalid.
  cl_header = mg_get_header(conn, "Content-Length");
  if (cl_header == NULL || (cl = strtoll(cl_header, NULL, 10)) <= 0) {
    mg_printf(conn, "%s%s", HTTP_500, "Invalid Conent-Length");
    return;
  }

  // Read the initial chunk into memory. This should be multipart POST data.
  // Parse headers, where we should find file name and content-type.
  post_data_len = mg_read(conn, post_data, sizeof(post_data));
  file_name[0] = mime_type[0] = '\0';
  for (s = p = post_data; p < &post_data[post_data_len]; p++) {
    if (p[0] == '\r' && p[1] == '\n') {
      if (s == p) {
        p += 2;
        break;  // End of headers
      }
      p[0] = p[1] = '\0';
      sscanf(s, "Content-Type: %99s", mime_type);
      // TODO(lsm): don't expect filename to be the 3rd field,
      // parse the header properly instead.
      sscanf(s, "Content-Disposition: %*s %*s filename=\"%1023[^\"]",
             file_name);
      s = p + 2;
    }
  }

  // Finished parsing headers. Now "p" points to the first byte of data.
  // Calculate file size
  cl -= p - post_data;      // Subtract headers size
  cl -= strlen(post_data);  // Subtract the boundary marker at the end
  cl -= 6;                  // Subtract "\r\n" before and after boundary

  // Construct destination file name. Write to /tmp, do not allow
  // paths that contain slashes.
  if ((s = strrchr(file_name, '/')) == NULL) {
    s = file_name;
  }
  snprintf(path, sizeof(path), "/tmp/%s", s);

  if (file_name[0] == '\0') {
    mg_printf(conn, "%s%s", HTTP_500, "Can't get file name");
  } else if (cl <= 0) {
    mg_printf(conn, "%s%s", HTTP_500, "Empty file");
  } else if ((fd = open(path, O_CREAT | O_TRUNC |
                        O_WRONLY | O_EXLOCK | O_CLOEXEC)) < 0) {
    // We're opening the file with exclusive lock held. This guarantee us that
    // there is no other thread can save into the same file simultaneously.
    mg_printf(conn, "%s%s", HTTP_500, "Cannot open file");
  } else if ((fp = fdopen(fd, "w")) == NULL) {
    mg_printf(conn, "%s%s", HTTP_500, "Cannot reopen file stream");
    close(fd);
  } else {
    // Success. Write data into the file.
    eop = post_data + post_data_len;
    n = p + cl > eop ? (int) (eop - p) : (int) cl;
    (void) fwrite(p, 1, n, fp);
    written = n;
    while (written < cl &&
           (n = mg_read(conn, buf, cl - written > (int64_t) sizeof(buf) ?
                        sizeof(buf) : cl - written)) > 0) {
      (void) fwrite(buf, 1, n, fp);
      written += n;
    }
    (void) fclose(fp);
    mg_printf(conn, "HTTP/1.0 200 OK\r\n\r\n"
              "Saved to [%s], written %llu bytes", path, cl);
  }
}

static void *callback(enum mg_event event, struct mg_connection *conn) {
  const struct mg_request_info *ri = mg_get_request_info(conn);

  if (event == MG_NEW_REQUEST) {
    if (!strcmp(ri->uri, "/handle_post_request")) {
      handle_file_upload(conn);
    } else {
      // Show HTML form.
      mg_printf(conn, "HTTP/1.0 200 OK\r\n"
                "Content-Length: %d\r\n"
                "Content-Type: text/html\r\n\r\n%s",
                (int) strlen(html_form), html_form);
    }
    // Mark as processed
    return "";
  } else {
    return NULL;
  }
}

int main(void) {
  struct mg_context *ctx;
  const char *options[] = {"listening_ports", "8080", NULL};

  ctx = mg_start(&callback, NULL, options);
  getchar();  // Wait until user hits "enter"
  mg_stop(ctx);

  return 0;
}