handle_form.inl 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896
  1. /* Copyright (c) 2016-2017 the Civetweb developers
  2. *
  3. * Permission is hereby granted, free of charge, to any person obtaining a copy
  4. * of this software and associated documentation files (the "Software"), to deal
  5. * in the Software without restriction, including without limitation the rights
  6. * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  7. * copies of the Software, and to permit persons to whom the Software is
  8. * furnished to do so, subject to the following conditions:
  9. *
  10. * The above copyright notice and this permission notice shall be included in
  11. * all copies or substantial portions of the Software.
  12. *
  13. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  14. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  15. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  16. * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  17. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  18. * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  19. * THE SOFTWARE.
  20. */
  21. static int
  22. url_encoded_field_found(const struct mg_connection *conn,
  23. const char *key,
  24. size_t key_len,
  25. const char *filename,
  26. size_t filename_len,
  27. char *path,
  28. size_t path_len,
  29. struct mg_form_data_handler *fdh)
  30. {
  31. char key_dec[1024];
  32. char filename_dec[1024];
  33. int key_dec_len;
  34. int filename_dec_len;
  35. int ret;
  36. key_dec_len =
  37. mg_url_decode(key, (int)key_len, key_dec, (int)sizeof(key_dec), 1);
  38. if (((size_t)key_dec_len >= (size_t)sizeof(key_dec)) || (key_dec_len < 0)) {
  39. return FORM_FIELD_STORAGE_SKIP;
  40. }
  41. if (filename) {
  42. filename_dec_len = mg_url_decode(filename,
  43. (int)filename_len,
  44. filename_dec,
  45. (int)sizeof(filename_dec),
  46. 1);
  47. if (((size_t)filename_dec_len >= (size_t)sizeof(filename_dec))
  48. || (filename_dec_len < 0)) {
  49. /* Log error message and skip this field. */
  50. mg_cry(conn, "%s: Cannot decode filename", __func__);
  51. return FORM_FIELD_STORAGE_SKIP;
  52. }
  53. } else {
  54. filename_dec[0] = 0;
  55. }
  56. ret =
  57. fdh->field_found(key_dec, filename_dec, path, path_len, fdh->user_data);
  58. if ((ret & 0xF) == FORM_FIELD_STORAGE_GET) {
  59. if (fdh->field_get == NULL) {
  60. mg_cry(conn, "%s: Function \"Get\" not available", __func__);
  61. return FORM_FIELD_STORAGE_SKIP;
  62. }
  63. }
  64. if ((ret & 0xF) == FORM_FIELD_STORAGE_STORE) {
  65. if (fdh->field_store == NULL) {
  66. mg_cry(conn, "%s: Function \"Store\" not available", __func__);
  67. return FORM_FIELD_STORAGE_SKIP;
  68. }
  69. }
  70. return ret;
  71. }
  72. static int
  73. url_encoded_field_get(const struct mg_connection *conn,
  74. const char *key,
  75. size_t key_len,
  76. const char *value,
  77. size_t value_len,
  78. struct mg_form_data_handler *fdh)
  79. {
  80. char key_dec[1024];
  81. char *value_dec = (char *)mg_malloc_ctx(value_len + 1, conn->ctx);
  82. int value_dec_len, ret;
  83. if (!value_dec) {
  84. /* Log error message and stop parsing the form data. */
  85. mg_cry(conn,
  86. "%s: Not enough memory (required: %lu)",
  87. __func__,
  88. (unsigned long)(value_len + 1));
  89. return FORM_FIELD_STORAGE_ABORT;
  90. }
  91. mg_url_decode(key, (int)key_len, key_dec, (int)sizeof(key_dec), 1);
  92. value_dec_len =
  93. mg_url_decode(value, (int)value_len, value_dec, (int)value_len + 1, 1);
  94. ret = fdh->field_get(key_dec,
  95. value_dec,
  96. (size_t)value_dec_len,
  97. fdh->user_data);
  98. mg_free(value_dec);
  99. return ret;
  100. }
  101. static int
  102. unencoded_field_get(const struct mg_connection *conn,
  103. const char *key,
  104. size_t key_len,
  105. const char *value,
  106. size_t value_len,
  107. struct mg_form_data_handler *fdh)
  108. {
  109. char key_dec[1024];
  110. (void)conn;
  111. mg_url_decode(key, (int)key_len, key_dec, (int)sizeof(key_dec), 1);
  112. return fdh->field_get(key_dec, value, value_len, fdh->user_data);
  113. }
  114. static int
  115. field_stored(const struct mg_connection *conn,
  116. const char *path,
  117. long long file_size,
  118. struct mg_form_data_handler *fdh)
  119. {
  120. /* Equivalent to "upload" callback of "mg_upload". */
  121. (void)conn; /* we do not need mg_cry here, so conn is currently unused */
  122. return fdh->field_store(path, file_size, fdh->user_data);
  123. }
  124. static const char *
  125. search_boundary(const char *buf,
  126. size_t buf_len,
  127. const char *boundary,
  128. size_t boundary_len)
  129. {
  130. /* We must do a binary search here, not a string search, since the buffer
  131. * may contain '\x00' bytes, if binary data is transferred. */
  132. int clen = (int)buf_len - (int)boundary_len - 4;
  133. int i;
  134. for (i = 0; i <= clen; i++) {
  135. if (!memcmp(buf + i, "\r\n--", 4)) {
  136. if (!memcmp(buf + i + 4, boundary, boundary_len)) {
  137. return buf + i;
  138. }
  139. }
  140. }
  141. return NULL;
  142. }
  143. int
  144. mg_handle_form_request(struct mg_connection *conn,
  145. struct mg_form_data_handler *fdh)
  146. {
  147. const char *content_type;
  148. char path[512];
  149. char buf[1024];
  150. int field_storage;
  151. int buf_fill = 0;
  152. int r;
  153. int field_count = 0;
  154. struct mg_file fstore = STRUCT_FILE_INITIALIZER;
  155. int64_t file_size = 0; /* init here, to a avoid a false positive
  156. "uninitialized variable used" warning */
  157. int has_body_data =
  158. (conn->request_info.content_length > 0) || (conn->is_chunked);
  159. /* There are three ways to encode data from a HTML form:
  160. * 1) method: GET (default)
  161. * The form data is in the HTTP query string.
  162. * 2) method: POST, enctype: "application/x-www-form-urlencoded"
  163. * The form data is in the request body.
  164. * The body is url encoded (the default encoding for POST).
  165. * 3) method: POST, enctype: "multipart/form-data".
  166. * The form data is in the request body of a multipart message.
  167. * This is the typical way to handle file upload from a form.
  168. */
  169. if (!has_body_data) {
  170. const char *data;
  171. if (strcmp(conn->request_info.request_method, "GET")) {
  172. /* No body data, but not a GET request.
  173. * This is not a valid form request. */
  174. return -1;
  175. }
  176. /* GET request: form data is in the query string. */
  177. /* The entire data has already been loaded, so there is no nead to
  178. * call mg_read. We just need to split the query string into key-value
  179. * pairs. */
  180. data = conn->request_info.query_string;
  181. if (!data) {
  182. /* No query string. */
  183. return -1;
  184. }
  185. /* Split data in a=1&b=xy&c=3&c=4 ... */
  186. while (*data) {
  187. const char *val = strchr(data, '=');
  188. const char *next;
  189. ptrdiff_t keylen, vallen;
  190. if (!val) {
  191. break;
  192. }
  193. keylen = val - data;
  194. /* In every "field_found" callback we ask what to do with the
  195. * data ("field_storage"). This could be:
  196. * FORM_FIELD_STORAGE_SKIP (0) ... ignore the value of this field
  197. * FORM_FIELD_STORAGE_GET (1) ... read the data and call the get
  198. * callback function
  199. * FORM_FIELD_STORAGE_STORE (2) ... store the data in a file
  200. * FORM_FIELD_STORAGE_READ (3) ... let the user read the data
  201. * (for parsing long data on the fly)
  202. * (currently not implemented)
  203. * FORM_FIELD_STORAGE_ABORT (flag) ... stop parsing
  204. */
  205. memset(path, 0, sizeof(path));
  206. field_count++;
  207. field_storage = url_encoded_field_found(conn,
  208. data,
  209. (size_t)keylen,
  210. NULL,
  211. 0,
  212. path,
  213. sizeof(path) - 1,
  214. fdh);
  215. val++;
  216. next = strchr(val, '&');
  217. if (next) {
  218. vallen = next - val;
  219. next++;
  220. } else {
  221. vallen = (ptrdiff_t)strlen(val);
  222. next = val + vallen;
  223. }
  224. if (field_storage == FORM_FIELD_STORAGE_GET) {
  225. /* Call callback */
  226. url_encoded_field_get(
  227. conn, data, (size_t)keylen, val, (size_t)vallen, fdh);
  228. }
  229. if (field_storage == FORM_FIELD_STORAGE_STORE) {
  230. /* Store the content to a file */
  231. if (mg_fopen(conn, path, MG_FOPEN_MODE_WRITE, &fstore) == 0) {
  232. fstore.access.fp = NULL;
  233. }
  234. file_size = 0;
  235. if (fstore.access.fp != NULL) {
  236. size_t n = (size_t)
  237. fwrite(val, 1, (size_t)vallen, fstore.access.fp);
  238. if ((n != (size_t)vallen) || (ferror(fstore.access.fp))) {
  239. mg_cry(conn,
  240. "%s: Cannot write file %s",
  241. __func__,
  242. path);
  243. (void)mg_fclose(&fstore.access);
  244. remove_bad_file(conn, path);
  245. }
  246. file_size += (int64_t)n;
  247. if (fstore.access.fp) {
  248. r = mg_fclose(&fstore.access);
  249. if (r == 0) {
  250. /* stored successfully */
  251. field_stored(conn, path, file_size, fdh);
  252. } else {
  253. mg_cry(conn,
  254. "%s: Error saving file %s",
  255. __func__,
  256. path);
  257. remove_bad_file(conn, path);
  258. }
  259. fstore.access.fp = NULL;
  260. }
  261. } else {
  262. mg_cry(conn, "%s: Cannot create file %s", __func__, path);
  263. }
  264. }
  265. /* if (field_storage == FORM_FIELD_STORAGE_READ) { */
  266. /* The idea of "field_storage=read" is to let the API user read
  267. * data chunk by chunk and to some data processing on the fly.
  268. * This should avoid the need to store data in the server:
  269. * It should neither be stored in memory, like
  270. * "field_storage=get" does, nor in a file like
  271. * "field_storage=store".
  272. * However, for a "GET" request this does not make any much
  273. * sense, since the data is already stored in memory, as it is
  274. * part of the query string.
  275. */
  276. /* } */
  277. if ((field_storage & FORM_FIELD_STORAGE_ABORT)
  278. == FORM_FIELD_STORAGE_ABORT) {
  279. /* Stop parsing the request */
  280. break;
  281. }
  282. /* Proceed to next entry */
  283. data = next;
  284. }
  285. return field_count;
  286. }
  287. content_type = mg_get_header(conn, "Content-Type");
  288. if (!content_type
  289. || !mg_strcasecmp(content_type, "APPLICATION/X-WWW-FORM-URLENCODED")
  290. || !mg_strcasecmp(content_type, "APPLICATION/WWW-FORM-URLENCODED")) {
  291. /* The form data is in the request body data, encoded in key/value
  292. * pairs. */
  293. int all_data_read = 0;
  294. /* Read body data and split it in keys and values.
  295. * The encoding is like in the "GET" case above: a=1&b&c=3&c=4.
  296. * Here we use "POST", and read the data from the request body.
  297. * The data read on the fly, so it is not required to buffer the
  298. * entire request in memory before processing it. */
  299. for (;;) {
  300. const char *val;
  301. const char *next;
  302. ptrdiff_t keylen, vallen;
  303. ptrdiff_t used;
  304. int end_of_key_value_pair_found = 0;
  305. int get_block;
  306. if ((size_t)buf_fill < (sizeof(buf) - 1)) {
  307. size_t to_read = sizeof(buf) - 1 - (size_t)buf_fill;
  308. r = mg_read(conn, buf + (size_t)buf_fill, to_read);
  309. if (r < 0) {
  310. /* read error */
  311. return -1;
  312. }
  313. if (r != (int)to_read) {
  314. /* TODO: Create a function to get "all_data_read" from
  315. * the conn object. All data is read if the Content-Length
  316. * has been reached, or if chunked encoding is used and
  317. * the end marker has been read, or if the connection has
  318. * been closed. */
  319. all_data_read = 1;
  320. }
  321. buf_fill += r;
  322. buf[buf_fill] = 0;
  323. if (buf_fill < 1) {
  324. break;
  325. }
  326. }
  327. val = strchr(buf, '=');
  328. if (!val) {
  329. break;
  330. }
  331. keylen = val - buf;
  332. val++;
  333. /* Call callback */
  334. memset(path, 0, sizeof(path));
  335. field_count++;
  336. field_storage = url_encoded_field_found(conn,
  337. buf,
  338. (size_t)keylen,
  339. NULL,
  340. 0,
  341. path,
  342. sizeof(path) - 1,
  343. fdh);
  344. if ((field_storage & FORM_FIELD_STORAGE_ABORT)
  345. == FORM_FIELD_STORAGE_ABORT) {
  346. /* Stop parsing the request */
  347. break;
  348. }
  349. if (field_storage == FORM_FIELD_STORAGE_STORE) {
  350. if (mg_fopen(conn, path, MG_FOPEN_MODE_WRITE, &fstore) == 0) {
  351. fstore.access.fp = NULL;
  352. }
  353. file_size = 0;
  354. if (!fstore.access.fp) {
  355. mg_cry(conn, "%s: Cannot create file %s", __func__, path);
  356. }
  357. }
  358. get_block = 0;
  359. /* Loop to read values larger than sizeof(buf)-keylen-2 */
  360. do {
  361. next = strchr(val, '&');
  362. if (next) {
  363. vallen = next - val;
  364. next++;
  365. end_of_key_value_pair_found = 1;
  366. } else {
  367. vallen = (ptrdiff_t)strlen(val);
  368. next = val + vallen;
  369. }
  370. if (field_storage == FORM_FIELD_STORAGE_GET) {
  371. #if 0
  372. if (!end_of_key_value_pair_found && !all_data_read) {
  373. /* This callback will deliver partial contents */
  374. }
  375. #else
  376. (void)all_data_read; /* avoid warning */
  377. #endif
  378. /* Call callback */
  379. url_encoded_field_get(conn,
  380. ((get_block > 0) ? NULL : buf),
  381. ((get_block > 0) ? 0
  382. : (size_t)keylen),
  383. val,
  384. (size_t)vallen,
  385. fdh);
  386. get_block++;
  387. }
  388. if (fstore.access.fp) {
  389. size_t n = (size_t)
  390. fwrite(val, 1, (size_t)vallen, fstore.access.fp);
  391. if ((n != (size_t)vallen) || (ferror(fstore.access.fp))) {
  392. mg_cry(conn,
  393. "%s: Cannot write file %s",
  394. __func__,
  395. path);
  396. mg_fclose(&fstore.access);
  397. remove_bad_file(conn, path);
  398. }
  399. file_size += (int64_t)n;
  400. }
  401. if (!end_of_key_value_pair_found) {
  402. used = next - buf;
  403. memmove(buf,
  404. buf + (size_t)used,
  405. sizeof(buf) - (size_t)used);
  406. buf_fill -= (int)used;
  407. if ((size_t)buf_fill < (sizeof(buf) - 1)) {
  408. size_t to_read = sizeof(buf) - 1 - (size_t)buf_fill;
  409. r = mg_read(conn, buf + (size_t)buf_fill, to_read);
  410. if (r < 0) {
  411. /* read error */
  412. return -1;
  413. }
  414. if (r != (int)to_read) {
  415. /* TODO: Create a function to get "all_data_read"
  416. * from the conn object. All data is read if the
  417. * Content-Length has been reached, or if chunked
  418. * encoding is used and the end marker has been
  419. * read, or if the connection has been closed. */
  420. all_data_read = 1;
  421. }
  422. buf_fill += r;
  423. buf[buf_fill] = 0;
  424. if (buf_fill < 1) {
  425. break;
  426. }
  427. val = buf;
  428. }
  429. }
  430. } while (!end_of_key_value_pair_found);
  431. if (fstore.access.fp) {
  432. r = mg_fclose(&fstore.access);
  433. if (r == 0) {
  434. /* stored successfully */
  435. field_stored(conn, path, file_size, fdh);
  436. } else {
  437. mg_cry(conn, "%s: Error saving file %s", __func__, path);
  438. remove_bad_file(conn, path);
  439. }
  440. fstore.access.fp = NULL;
  441. }
  442. /* Proceed to next entry */
  443. used = next - buf;
  444. memmove(buf, buf + (size_t)used, sizeof(buf) - (size_t)used);
  445. buf_fill -= (int)used;
  446. }
  447. return field_count;
  448. }
  449. if (!mg_strncasecmp(content_type, "MULTIPART/FORM-DATA;", 20)) {
  450. /* The form data is in the request body data, encoded as multipart
  451. * content (see https://www.ietf.org/rfc/rfc1867.txt,
  452. * https://www.ietf.org/rfc/rfc2388.txt). */
  453. const char *boundary;
  454. size_t bl;
  455. ptrdiff_t used;
  456. struct mg_request_info part_header;
  457. char *hbuf;
  458. const char *content_disp, *hend, *fbeg, *fend, *nbeg, *nend;
  459. const char *next;
  460. memset(&part_header, 0, sizeof(part_header));
  461. /* Skip all spaces between MULTIPART/FORM-DATA; and BOUNDARY= */
  462. bl = 20;
  463. while (content_type[bl] == ' ') {
  464. bl++;
  465. }
  466. /* There has to be a BOUNDARY definition in the Content-Type header */
  467. if (mg_strncasecmp(content_type + bl, "BOUNDARY=", 9)) {
  468. /* Malformed request */
  469. return -1;
  470. }
  471. boundary = content_type + bl + 9;
  472. bl = strlen(boundary);
  473. if (boundary[0] == '"') {
  474. /* RFC 2046 permits the boundary string to be quoted. */
  475. hbuf = strchr(boundary + 1, '"');
  476. if (!hbuf) {
  477. /* Malformed request */
  478. return -1;
  479. }
  480. if (*hbuf) {
  481. *hbuf = 0;
  482. boundary++;
  483. bl = strlen(boundary);
  484. } else {
  485. /* Malformed request */
  486. return -1;
  487. }
  488. }
  489. if (bl + 800 > sizeof(buf)) {
  490. /* Sanity check: The algorithm can not work if bl >= sizeof(buf),
  491. * and it will not work effectively, if the buf is only a few byte
  492. * larger than bl, or if buf can not hold the multipart header
  493. * plus the boundary.
  494. * Check some reasonable number here, that should be fulfilled by
  495. * any reasonable request from every browser. If it is not
  496. * fulfilled, it might be a hand-made request, intended to
  497. * interfere with the algorithm. */
  498. return -1;
  499. }
  500. if (bl < 4) {
  501. /* Sanity check: A boundary string of less than 4 bytes makes
  502. * no sense either. */
  503. return -1;
  504. }
  505. for (;;) {
  506. size_t towrite, n;
  507. int get_block;
  508. r = mg_read(conn,
  509. buf + (size_t)buf_fill,
  510. sizeof(buf) - 1 - (size_t)buf_fill);
  511. if (r < 0) {
  512. /* read error */
  513. return -1;
  514. }
  515. buf_fill += r;
  516. buf[buf_fill] = 0;
  517. if (buf_fill < 1) {
  518. /* No data */
  519. return -1;
  520. }
  521. if (buf[0] != '-' || buf[1] != '-') {
  522. /* Malformed request */
  523. return -1;
  524. }
  525. if (strncmp(buf + 2, boundary, bl)) {
  526. /* Malformed request */
  527. return -1;
  528. }
  529. if (buf[bl + 2] != '\r' || buf[bl + 3] != '\n') {
  530. /* Every part must end with \r\n, if there is another part.
  531. * The end of the request has an extra -- */
  532. if (((size_t)buf_fill != (size_t)(bl + 6))
  533. || (strncmp(buf + bl + 2, "--\r\n", 4))) {
  534. /* Malformed request */
  535. return -1;
  536. }
  537. /* End of the request */
  538. break;
  539. }
  540. /* Next, we need to get the part header: Read until \r\n\r\n */
  541. hbuf = buf + bl + 4;
  542. hend = strstr(hbuf, "\r\n\r\n");
  543. if (!hend) {
  544. /* Malformed request */
  545. return -1;
  546. }
  547. part_header.num_headers =
  548. parse_http_headers(&hbuf, part_header.http_headers);
  549. if ((hend + 2) != hbuf) {
  550. /* Malformed request */
  551. return -1;
  552. }
  553. /* Skip \r\n\r\n */
  554. hend += 4;
  555. /* According to the RFC, every part has to have a header field like:
  556. * Content-Disposition: form-data; name="..." */
  557. content_disp = get_header(part_header.http_headers,
  558. part_header.num_headers,
  559. "Content-Disposition");
  560. if (!content_disp) {
  561. /* Malformed request */
  562. return -1;
  563. }
  564. /* Get the mandatory name="..." part of the Content-Disposition
  565. * header. */
  566. nbeg = strstr(content_disp, "name=\"");
  567. while ((nbeg != NULL) && (strcspn(nbeg - 1, ":,; \t") != 0)) {
  568. /* It could be somethingname= instead of name= */
  569. nbeg = strstr(nbeg + 1, "name=\"");
  570. }
  571. /* This line is not required, but otherwise some compilers
  572. * generate spurious warnings. */
  573. nend = nbeg;
  574. /* And others complain, the result is unused. */
  575. (void)nend;
  576. /* If name=" is found, search for the closing " */
  577. if (nbeg) {
  578. nbeg += 6;
  579. nend = strchr(nbeg, '\"');
  580. if (!nend) {
  581. /* Malformed request */
  582. return -1;
  583. }
  584. } else {
  585. /* name= without quotes is also allowed */
  586. nbeg = strstr(content_disp, "name=");
  587. while ((nbeg != NULL) && (strcspn(nbeg - 1, ":,; \t") != 0)) {
  588. /* It could be somethingname= instead of name= */
  589. nbeg = strstr(nbeg + 1, "name=");
  590. }
  591. if (!nbeg) {
  592. /* Malformed request */
  593. return -1;
  594. }
  595. nbeg += 5;
  596. /* RFC 2616 Sec. 2.2 defines a list of allowed
  597. * separators, but many of them make no sense
  598. * here, e.g. various brackets or slashes.
  599. * If they are used, probably someone is
  600. * trying to attack with curious hand made
  601. * requests. Only ; , space and tab seem to be
  602. * reasonable here. Ignore everything else. */
  603. nend = nbeg + strcspn(nbeg, ",; \t");
  604. }
  605. /* Get the optional filename="..." part of the Content-Disposition
  606. * header. */
  607. fbeg = strstr(content_disp, "filename=\"");
  608. while ((fbeg != NULL) && (strcspn(fbeg - 1, ":,; \t") != 0)) {
  609. /* It could be somethingfilename= instead of filename= */
  610. fbeg = strstr(fbeg + 1, "filename=\"");
  611. }
  612. /* This line is not required, but otherwise some compilers
  613. * generate spurious warnings. */
  614. fend = fbeg;
  615. /* If filename=" is found, search for the closing " */
  616. if (fbeg) {
  617. fbeg += 10;
  618. fend = strchr(fbeg, '\"');
  619. if (!fend) {
  620. /* Malformed request (the filename field is optional, but if
  621. * it exists, it needs to be terminated correctly). */
  622. return -1;
  623. }
  624. /* TODO: check Content-Type */
  625. /* Content-Type: application/octet-stream */
  626. }
  627. if (!fbeg) {
  628. /* Try the same without quotes */
  629. fbeg = strstr(content_disp, "filename=");
  630. while ((fbeg != NULL) && (strcspn(fbeg - 1, ":,; \t") != 0)) {
  631. /* It could be somethingfilename= instead of filename= */
  632. fbeg = strstr(fbeg + 1, "filename=");
  633. }
  634. if (fbeg) {
  635. fbeg += 9;
  636. fend = fbeg + strcspn(fbeg, ",; \t");
  637. }
  638. }
  639. if (!fbeg) {
  640. fend = NULL;
  641. }
  642. /* In theory, it could be possible that someone crafts
  643. * a request like name=filename=xyz. Check if name and
  644. * filename do not overlap. */
  645. if (!(((ptrdiff_t)fbeg > (ptrdiff_t)nend)
  646. || ((ptrdiff_t)nbeg > (ptrdiff_t)fend))) {
  647. return -1;
  648. }
  649. /* Call callback for new field */
  650. memset(path, 0, sizeof(path));
  651. field_count++;
  652. field_storage = url_encoded_field_found(conn,
  653. nbeg,
  654. (size_t)(nend - nbeg),
  655. fbeg,
  656. (size_t)(fend - fbeg),
  657. path,
  658. sizeof(path) - 1,
  659. fdh);
  660. /* If the boundary is already in the buffer, get the address,
  661. * otherwise next will be NULL. */
  662. next = search_boundary(hbuf,
  663. (size_t)((buf - hbuf) + buf_fill),
  664. boundary,
  665. bl);
  666. if (field_storage == FORM_FIELD_STORAGE_STORE) {
  667. /* Store the content to a file */
  668. if (mg_fopen(conn, path, MG_FOPEN_MODE_WRITE, &fstore) == 0) {
  669. fstore.access.fp = NULL;
  670. }
  671. file_size = 0;
  672. if (!fstore.access.fp) {
  673. mg_cry(conn, "%s: Cannot create file %s", __func__, path);
  674. }
  675. }
  676. get_block = 0;
  677. while (!next) {
  678. /* Set "towrite" to the number of bytes available
  679. * in the buffer */
  680. towrite = (size_t)(buf - hend + buf_fill);
  681. /* Subtract the boundary length, to deal with
  682. * cases the boundary is only partially stored
  683. * in the buffer. */
  684. towrite -= bl + 4;
  685. if (field_storage == FORM_FIELD_STORAGE_GET) {
  686. unencoded_field_get(conn,
  687. ((get_block > 0) ? NULL : nbeg),
  688. ((get_block > 0)
  689. ? 0
  690. : (size_t)(nend - nbeg)),
  691. hend,
  692. towrite,
  693. fdh);
  694. get_block++;
  695. }
  696. if (field_storage == FORM_FIELD_STORAGE_STORE) {
  697. if (fstore.access.fp) {
  698. /* Store the content of the buffer. */
  699. n = (size_t)fwrite(hend, 1, towrite, fstore.access.fp);
  700. if ((n != towrite) || (ferror(fstore.access.fp))) {
  701. mg_cry(conn,
  702. "%s: Cannot write file %s",
  703. __func__,
  704. path);
  705. mg_fclose(&fstore.access);
  706. remove_bad_file(conn, path);
  707. }
  708. file_size += (int64_t)n;
  709. }
  710. }
  711. memmove(buf, hend + towrite, bl + 4);
  712. buf_fill = (int)(bl + 4);
  713. hend = buf;
  714. /* Read new data */
  715. r = mg_read(conn,
  716. buf + (size_t)buf_fill,
  717. sizeof(buf) - 1 - (size_t)buf_fill);
  718. if (r < 0) {
  719. /* read error */
  720. return -1;
  721. }
  722. buf_fill += r;
  723. buf[buf_fill] = 0;
  724. if (buf_fill < 1) {
  725. /* No data */
  726. return -1;
  727. }
  728. /* Find boundary */
  729. next = search_boundary(buf, (size_t)buf_fill, boundary, bl);
  730. }
  731. towrite = (size_t)(next - hend);
  732. if (field_storage == FORM_FIELD_STORAGE_GET) {
  733. /* Call callback */
  734. unencoded_field_get(conn,
  735. ((get_block > 0) ? NULL : nbeg),
  736. ((get_block > 0) ? 0
  737. : (size_t)(nend - nbeg)),
  738. hend,
  739. towrite,
  740. fdh);
  741. }
  742. if (field_storage == FORM_FIELD_STORAGE_STORE) {
  743. if (fstore.access.fp) {
  744. n = (size_t)fwrite(hend, 1, towrite, fstore.access.fp);
  745. if ((n != towrite) || (ferror(fstore.access.fp))) {
  746. mg_cry(conn,
  747. "%s: Cannot write file %s",
  748. __func__,
  749. path);
  750. mg_fclose(&fstore.access);
  751. remove_bad_file(conn, path);
  752. } else {
  753. file_size += (int64_t)n;
  754. r = mg_fclose(&fstore.access);
  755. if (r == 0) {
  756. /* stored successfully */
  757. field_stored(conn, path, file_size, fdh);
  758. } else {
  759. mg_cry(conn,
  760. "%s: Error saving file %s",
  761. __func__,
  762. path);
  763. remove_bad_file(conn, path);
  764. }
  765. }
  766. fstore.access.fp = NULL;
  767. }
  768. }
  769. if ((field_storage & FORM_FIELD_STORAGE_ABORT)
  770. == FORM_FIELD_STORAGE_ABORT) {
  771. /* Stop parsing the request */
  772. break;
  773. }
  774. /* Remove from the buffer */
  775. used = next - buf + 2;
  776. memmove(buf, buf + (size_t)used, sizeof(buf) - (size_t)used);
  777. buf_fill -= (int)used;
  778. }
  779. /* All parts handled */
  780. return field_count;
  781. }
  782. /* Unknown Content-Type */
  783. return -1;
  784. }
  785. /* End of handle_form.inl */