private.c 49 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534
  1. /* Copyright (c) 2015-2020 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. /**********************************************************************************/
  22. /*
  23. * We include the civetweb.c source file so that we have access to the internal
  24. * private static functions
  25. */
  26. #ifdef _MSC_VER
  27. /* Since the C file is included, declare all API functions as static,
  28. * to avoid linker errors in the test (seems to be required for MS
  29. * toolchain only).
  30. */
  31. #define CIVETWEB_API static
  32. #endif
  33. #include "../src/civetweb.c"
  34. /**********************************************************************************/
  35. /* Standard includes for the test code below */
  36. #include <stdlib.h>
  37. #include <time.h>
  38. #include "private.h"
  39. /* This unit test file uses the excellent Check unit testing library.
  40. * The API documentation is available here:
  41. * http://check.sourceforge.net/doc/check_html/index.html
  42. */
  43. static char tmp_parse_buffer[1024];
  44. static int
  45. test_parse_http_response(char *buf, int len, struct mg_response_info *ri)
  46. {
  47. ck_assert_int_lt(len, (int)sizeof(tmp_parse_buffer));
  48. memcpy(tmp_parse_buffer, buf, (size_t)len);
  49. return parse_http_response(tmp_parse_buffer, len, ri);
  50. }
  51. static int
  52. test_parse_http_request(char *buf, int len, struct mg_request_info *ri)
  53. {
  54. ck_assert_int_lt(len, (int)sizeof(tmp_parse_buffer));
  55. memcpy(tmp_parse_buffer, buf, (size_t)len);
  56. return parse_http_request(tmp_parse_buffer, len, ri);
  57. }
  58. START_TEST(test_parse_http_message)
  59. {
  60. /* Adapted from unit_test.c */
  61. /* Copyright (c) 2013-2015 the Civetweb developers */
  62. /* Copyright (c) 2004-2013 Sergey Lyubka */
  63. struct mg_request_info ri;
  64. struct mg_response_info respi;
  65. char empty[] = "";
  66. char space[] = " \x00";
  67. char req1[] = "GET / HTTP/1.1\r\n\r\n";
  68. char req2[] = "BLAH / HTTP/1.1\r\n\r\n";
  69. char req3[] = "GET / HTTP/1.1\nKey: Val\n\n";
  70. char req4[] =
  71. "GET / HTTP/1.1\r\nA: foo bar\r\nB: bar\r\nskip\r\nbaz:\r\n\r\n";
  72. char req5[] = "GET / HTTP/1.0\n\n";
  73. char req6[] = "G";
  74. char req7[] = " blah ";
  75. char req8[] = "HTTP/1.0 404 Not Found\n\n";
  76. char req9[] = "HTTP/1.1 200 OK\r\nConnection: close\r\n\r\n";
  77. char req10[] = "GET / HTTP/1.1\r\nA: foo bar\r\nB: bar\r\n\r\n";
  78. char req11[] = "GET /\r\nError: X\r\n\r\n";
  79. char req12[] =
  80. "POST /a/b/c.d?e=f&g HTTP/1.1\r\nKey1: val1\r\nKey2: val2\r\n\r\nBODY";
  81. int lenreq1 = (int)strlen(req1);
  82. int lenreq2 = (int)strlen(req2);
  83. int lenreq3 = (int)strlen(req3);
  84. int lenreq4 = (int)strlen(req4);
  85. int lenreq5 = (int)strlen(req5);
  86. int lenreq6 = (int)strlen(req6);
  87. int lenreq7 = (int)strlen(req7);
  88. int lenreq8 = (int)strlen(req8);
  89. int lenreq9 = (int)strlen(req9);
  90. int lenreq10 = (int)strlen(req10);
  91. int lenreq11 = (int)strlen(req11);
  92. int lenreq12 = (int)strlen(req12);
  93. int lenhdr12 = lenreq12 - 4; /* length without body */
  94. mark_point();
  95. /* An empty string is neither a complete request nor a complete
  96. * response, so it must return 0 */
  97. ck_assert_int_eq(0, get_http_header_len(empty, 0));
  98. ck_assert_int_eq(0, test_parse_http_request(empty, 0, &ri));
  99. ck_assert_int_eq(0, test_parse_http_response(empty, 0, &respi));
  100. /* Same is true for a leading space */
  101. ck_assert_int_eq(0, get_http_header_len(space, 1));
  102. ck_assert_int_eq(0, test_parse_http_request(space, 1, &ri));
  103. ck_assert_int_eq(0, test_parse_http_response(space, 1, &respi));
  104. /* But a control character (like 0) makes it invalid */
  105. ck_assert_int_eq(-1, get_http_header_len(space, 2));
  106. ck_assert_int_eq(-1, test_parse_http_request(space, 2, &ri));
  107. ck_assert_int_eq(-1, test_parse_http_response(space, 2, &respi));
  108. /* req1 minus 1 byte at the end is incomplete */
  109. ck_assert_int_eq(0, get_http_header_len(req1, lenreq1 - 1));
  110. /* req1 minus 1 byte at the start is complete but invalid */
  111. ck_assert_int_eq(lenreq1 - 1, get_http_header_len(req1 + 1, lenreq1 - 1));
  112. ck_assert_int_eq(-1, test_parse_http_request(req1 + 1, lenreq1 - 1, &ri));
  113. /* req1 is a valid request */
  114. ck_assert_int_eq(lenreq1, get_http_header_len(req1, lenreq1));
  115. ck_assert_int_eq(-1, test_parse_http_response(req1, lenreq1, &respi));
  116. ck_assert_int_eq(lenreq1, test_parse_http_request(req1, lenreq1, &ri));
  117. ck_assert_str_eq("1.1", ri.http_version);
  118. ck_assert_int_eq(0, ri.num_headers);
  119. /* req2 is a complete, but invalid request */
  120. ck_assert_int_eq(lenreq2, get_http_header_len(req2, lenreq2));
  121. ck_assert_int_eq(-1, test_parse_http_request(req2, lenreq2, &ri));
  122. /* req3 is a complete and valid request */
  123. ck_assert_int_eq(lenreq3, get_http_header_len(req3, lenreq3));
  124. ck_assert_int_eq(lenreq3, test_parse_http_request(req3, lenreq3, &ri));
  125. ck_assert_int_eq(-1, test_parse_http_response(req3, lenreq3, &respi));
  126. /* Multiline header are obsolete, so return an error
  127. * (https://tools.ietf.org/html/rfc7230#section-3.2.4). */
  128. ck_assert_int_eq(-1, test_parse_http_request(req4, lenreq4, &ri));
  129. /* req5 is a complete and valid request (also somewhat malformed,
  130. * since it uses \n\n instead of \r\n\r\n) */
  131. ck_assert_int_eq(lenreq5, get_http_header_len(req5, lenreq5));
  132. ck_assert_int_eq(-1, test_parse_http_response(req5, lenreq5, &respi));
  133. ck_assert_int_eq(lenreq5, test_parse_http_request(req5, lenreq5, &ri));
  134. ck_assert_str_eq("GET", ri.request_method);
  135. ck_assert_str_eq("1.0", ri.http_version);
  136. /* req6 is incomplete */
  137. ck_assert_int_eq(0, get_http_header_len(req6, lenreq6));
  138. ck_assert_int_eq(0, test_parse_http_request(req6, lenreq6, &ri));
  139. /* req7 is invalid, but not yet complete */
  140. ck_assert_int_eq(0, get_http_header_len(req7, lenreq7));
  141. ck_assert_int_eq(0, test_parse_http_request(req7, lenreq7, &ri));
  142. /* req8 is a valid response */
  143. ck_assert_int_eq(lenreq8, get_http_header_len(req8, lenreq8));
  144. ck_assert_int_eq(-1, test_parse_http_request(req8, lenreq8, &ri));
  145. ck_assert_int_eq(lenreq8, test_parse_http_response(req8, lenreq8, &respi));
  146. /* req9 is a valid response */
  147. ck_assert_int_eq(lenreq9, get_http_header_len(req9, lenreq9));
  148. ck_assert_int_eq(-1, test_parse_http_request(req9, lenreq9, &ri));
  149. ck_assert_int_eq(lenreq9, test_parse_http_response(req9, lenreq9, &respi));
  150. ck_assert_int_eq(1, respi.num_headers);
  151. /* req10 is a valid request */
  152. ck_assert_int_eq(lenreq10, get_http_header_len(req10, lenreq10));
  153. ck_assert_int_eq(lenreq10, test_parse_http_request(req10, lenreq10, &ri));
  154. ck_assert_str_eq("1.1", ri.http_version);
  155. ck_assert_int_eq(2, ri.num_headers);
  156. ck_assert_str_eq("A", ri.http_headers[0].name);
  157. ck_assert_str_eq("foo bar", ri.http_headers[0].value);
  158. ck_assert_str_eq("B", ri.http_headers[1].name);
  159. ck_assert_str_eq("bar", ri.http_headers[1].value);
  160. /* req11 is a complete but valid request */
  161. ck_assert_int_eq(-1, test_parse_http_request(req11, lenreq11, &ri));
  162. /* req12 is a valid request with body data */
  163. ck_assert_int_gt(lenreq12, lenhdr12);
  164. ck_assert_int_eq(lenhdr12, get_http_header_len(req12, lenreq12));
  165. ck_assert_int_eq(lenhdr12, test_parse_http_request(req12, lenreq12, &ri));
  166. }
  167. END_TEST
  168. START_TEST(test_should_keep_alive)
  169. {
  170. /* Adapted from unit_test.c */
  171. /* Copyright (c) 2013-2015 the Civetweb developers */
  172. /* Copyright (c) 2004-2013 Sergey Lyubka */
  173. struct mg_connection conn;
  174. struct mg_context ctx;
  175. char req1[] = "GET / HTTP/1.1\r\n\r\n";
  176. char req2[] = "GET / HTTP/1.0\r\n\r\n";
  177. char req3[] = "GET / HTTP/1.1\r\nConnection: close\r\n\r\n";
  178. char req4[] = "GET / HTTP/1.1\r\nConnection: keep-alive\r\n\r\n";
  179. char yes[] = "yes";
  180. char no[] = "no";
  181. int lenreq1 = (int)strlen(req1);
  182. int lenreq2 = (int)strlen(req2);
  183. int lenreq3 = (int)strlen(req3);
  184. int lenreq4 = (int)strlen(req4);
  185. mark_point();
  186. memset(&ctx, 0, sizeof(ctx));
  187. memset(&conn, 0, sizeof(conn));
  188. conn.phys_ctx = &ctx;
  189. conn.dom_ctx = &(ctx.dd);
  190. ck_assert_int_eq(test_parse_http_request(req1, lenreq1, &conn.request_info),
  191. lenreq1);
  192. conn.connection_type = 1; /* Valid request */
  193. ck_assert_int_eq(conn.request_info.num_headers, 0);
  194. ctx.dd.config[ENABLE_KEEP_ALIVE] = no;
  195. ck_assert_int_eq(should_keep_alive(&conn), 0);
  196. ctx.dd.config[ENABLE_KEEP_ALIVE] = yes;
  197. ck_assert_int_eq(should_keep_alive(&conn), 1);
  198. conn.must_close = 1;
  199. ck_assert_int_eq(should_keep_alive(&conn), 0);
  200. conn.must_close = 0;
  201. test_parse_http_request(req2, lenreq2, &conn.request_info);
  202. conn.connection_type = 1; /* Valid request */
  203. ck_assert_int_eq(conn.request_info.num_headers, 0);
  204. ck_assert_int_eq(should_keep_alive(&conn), 0);
  205. test_parse_http_request(req3, lenreq3, &conn.request_info);
  206. conn.connection_type = 1; /* Valid request */
  207. ck_assert_int_eq(conn.request_info.num_headers, 1);
  208. ck_assert_int_eq(should_keep_alive(&conn), 0);
  209. test_parse_http_request(req4, lenreq4, &conn.request_info);
  210. conn.connection_type = 1; /* Valid request */
  211. ck_assert_int_eq(conn.request_info.num_headers, 1);
  212. ck_assert_int_eq(should_keep_alive(&conn), 1);
  213. conn.status_code = 200;
  214. conn.must_close = 1;
  215. ck_assert_int_eq(should_keep_alive(&conn), 0);
  216. conn.status_code = 200;
  217. conn.must_close = 0;
  218. ck_assert_int_eq(should_keep_alive(&conn), 1);
  219. conn.status_code = 200;
  220. conn.must_close = 0;
  221. conn.connection_type = 0; /* invalid */
  222. ck_assert_int_eq(should_keep_alive(&conn), 0);
  223. }
  224. END_TEST
  225. START_TEST(test_match_prefix)
  226. {
  227. /* Adapted from unit_test.c */
  228. /* Copyright (c) 2013-2015 the Civetweb developers */
  229. /* Copyright (c) 2004-2013 Sergey Lyubka */
  230. ck_assert_int_eq(4, match_prefix("/api", 4, "/api"));
  231. ck_assert_int_eq(3, match_prefix("/a/", 3, "/a/b/c"));
  232. ck_assert_int_eq(-1, match_prefix("/a/", 3, "/ab/c"));
  233. ck_assert_int_eq(4, match_prefix("/*/", 3, "/ab/c"));
  234. ck_assert_int_eq(6, match_prefix("**", 2, "/a/b/c"));
  235. ck_assert_int_eq(2, match_prefix("/*", 2, "/a/b/c"));
  236. ck_assert_int_eq(2, match_prefix("*/*", 3, "/a/b/c"));
  237. ck_assert_int_eq(5, match_prefix("**/", 3, "/a/b/c"));
  238. ck_assert_int_eq(5, match_prefix("**.foo|**.bar", 13, "a.bar"));
  239. ck_assert_int_eq(2, match_prefix("a|b|cd", 6, "cdef"));
  240. ck_assert_int_eq(2, match_prefix("a|b|c?", 6, "cdef"));
  241. ck_assert_int_eq(1, match_prefix("a|?|cd", 6, "cdef"));
  242. ck_assert_int_eq(-1, match_prefix("/a/**.cgi", 9, "/foo/bar/x.cgi"));
  243. ck_assert_int_eq(12, match_prefix("/a/**.cgi", 9, "/a/bar/x.cgi"));
  244. ck_assert_int_eq(5, match_prefix("**/", 3, "/a/b/c"));
  245. ck_assert_int_eq(-1, match_prefix("**/$", 4, "/a/b/c"));
  246. ck_assert_int_eq(5, match_prefix("**/$", 4, "/a/b/"));
  247. ck_assert_int_eq(0, match_prefix("$", 1, ""));
  248. ck_assert_int_eq(-1, match_prefix("$", 1, "x"));
  249. ck_assert_int_eq(1, match_prefix("*$", 2, "x"));
  250. ck_assert_int_eq(1, match_prefix("/$", 2, "/"));
  251. ck_assert_int_eq(-1, match_prefix("**/$", 4, "/a/b/c"));
  252. ck_assert_int_eq(5, match_prefix("**/$", 4, "/a/b/"));
  253. ck_assert_int_eq(0, match_prefix("*", 1, "/hello/"));
  254. ck_assert_int_eq(-1, match_prefix("**.a$|**.b$", 11, "/a/b.b/"));
  255. ck_assert_int_eq(6, match_prefix("**.a$|**.b$", 11, "/a/b.b"));
  256. ck_assert_int_eq(6, match_prefix("**.a$|**.b$", 11, "/a/B.A"));
  257. ck_assert_int_eq(5, match_prefix("**o$", 4, "HELLO"));
  258. }
  259. END_TEST
  260. START_TEST(test_remove_dot_segments)
  261. {
  262. int i;
  263. struct {
  264. const char *input;
  265. const char *expected_output;
  266. } tests[] = {
  267. {"/path/to/file.ext", "/path/to/file.ext"},
  268. {"/file.ext", "/file.ext"},
  269. {"/path/../file.ext", "/file.ext"},
  270. {"/../to/file.ext", "/to/file.ext"},
  271. {"/../../file.ext", "/file.ext"},
  272. {"/./../file.ext", "/file.ext"},
  273. {"/.././file.ext", "/file.ext"},
  274. {"/././file.ext", "/file.ext"},
  275. {"/././file.ext", "/file.ext"},
  276. {"/path/.to/..file.ext", "/path/.to/..file.ext"},
  277. {"/file", "/file"},
  278. {"/path/", "/path/"},
  279. {"file.ext", "file.ext"},
  280. {"./file.ext", "file.ext"},
  281. {"../file.ext", "file.ext"},
  282. {".file.ext", ".file.ext"},
  283. {"..file.ext", "..file.ext"},
  284. {"file", "file"},
  285. {"/x/../", "/"},
  286. {"/x/../../", "/"},
  287. {"/x/.././", "/"},
  288. {"/./x/.././", "/"},
  289. /* Windows specific */
  290. {"\\file.ext", "/file.ext"},
  291. {"\\..\\file.ext", "/file.ext"},
  292. {"/file.", "/file"},
  293. {"/path\\to.\\.\\file.", "/path/to/file"},
  294. /* Multiple dots and slashes */
  295. {"\\//\\\\x", "/x"},
  296. {"//", "/"},
  297. {"/./", "/"},
  298. {"/../", "/"},
  299. {"/.../", "/"},
  300. {"/..../", "/"},
  301. {"/...../", "/"},
  302. {"/...../", "/"},
  303. {"/...//", "/"},
  304. {"/..././", "/"},
  305. {"/.../../", "/"},
  306. {"/.../.../", "/"},
  307. /* Test cases from issues */
  308. {"/foo/bar/baz/../qux.txt", "/foo/bar/qux.txt"},
  309. {"/../alice/bob/../carol/david/frank/../../grace.ext/hugo/../irene.jpg",
  310. "/alice/carol/grace.ext/irene.jpg"},
  311. {"/a/b/..", "/a/"},
  312. {"/a/b/c/../d/../..", "/a/"},
  313. {NULL, NULL}};
  314. mark_point();
  315. for (i = 0; (tests[i].input != NULL); i++) {
  316. char inout[256];
  317. strcpy(inout, tests[i].input);
  318. remove_dot_segments(inout);
  319. ck_assert_str_eq(inout, tests[i].expected_output);
  320. }
  321. }
  322. END_TEST
  323. START_TEST(test_is_valid_uri)
  324. {
  325. /* is_valid_uri is superseeded by get_uri_type */
  326. ck_assert_int_eq(2, get_uri_type("/api"));
  327. ck_assert_int_eq(2, get_uri_type("/api/"));
  328. ck_assert_int_eq(2,
  329. get_uri_type("/some/long/path%20with%20space/file.xyz"));
  330. ck_assert_int_eq(0, get_uri_type("api"));
  331. ck_assert_int_eq(1, get_uri_type("*"));
  332. ck_assert_int_eq(0, get_uri_type("*xy"));
  333. ck_assert_int_eq(3, get_uri_type("http://somewhere/"));
  334. ck_assert_int_eq(3, get_uri_type("https://somewhere/some/file.html"));
  335. ck_assert_int_eq(4, get_uri_type("http://somewhere:8080/"));
  336. ck_assert_int_eq(4, get_uri_type("https://somewhere:8080/some/file.html"));
  337. }
  338. END_TEST
  339. START_TEST(test_next_option)
  340. {
  341. /* Adapted from unit_test.c */
  342. /* Copyright (c) 2013-2015 the Civetweb developers */
  343. /* Copyright (c) 2004-2013 Sergey Lyubka */
  344. const char *p, *list = "x/8,/y**=1;2k,z";
  345. struct vec a, b;
  346. int i;
  347. mark_point();
  348. ck_assert(next_option(NULL, &a, &b) == NULL);
  349. for (i = 0, p = list; (p = next_option(p, &a, &b)) != NULL; i++) {
  350. ck_assert(i != 0 || (a.ptr == list && a.len == 3 && b.len == 0));
  351. ck_assert(i != 1
  352. || (a.ptr == list + 4 && a.len == 4 && b.ptr == list + 9
  353. && b.len == 4));
  354. ck_assert(i != 2 || (a.ptr == list + 14 && a.len == 1 && b.len == 0));
  355. }
  356. }
  357. END_TEST
  358. START_TEST(test_skip_quoted)
  359. {
  360. /* Adapted from unit_test.c */
  361. /* Copyright (c) 2013-2015 the Civetweb developers */
  362. /* Copyright (c) 2004-2013 Sergey Lyubka */
  363. char x[] = "a=1, b=2, c='hi \' there', d='here\\, there'", *s = x, *p;
  364. mark_point();
  365. p = skip_quoted(&s, ", ", ", ", 0);
  366. ck_assert(p != NULL && !strcmp(p, "a=1"));
  367. p = skip_quoted(&s, ", ", ", ", 0);
  368. ck_assert(p != NULL && !strcmp(p, "b=2"));
  369. p = skip_quoted(&s, ",", " ", 0);
  370. ck_assert(p != NULL && !strcmp(p, "c='hi \' there'"));
  371. p = skip_quoted(&s, ",", " ", '\\');
  372. ck_assert(p != NULL && !strcmp(p, "d='here, there'"));
  373. ck_assert(*s == 0);
  374. }
  375. END_TEST
  376. static int
  377. alloc_printf(char **buf, size_t size, const char *fmt, ...)
  378. {
  379. /* Test helper function - adapted from unit_test.c */
  380. /* Copyright (c) 2013-2015 the Civetweb developers */
  381. /* Copyright (c) 2004-2013 Sergey Lyubka */
  382. va_list ap;
  383. int ret = 0;
  384. mark_point();
  385. va_start(ap, fmt);
  386. ret = alloc_vprintf(buf, *buf, size, fmt, ap);
  387. va_end(ap);
  388. return ret;
  389. }
  390. static int
  391. alloc_printf2(char **buf, const char *fmt, ...)
  392. {
  393. /* Test alternative implementation */
  394. va_list ap;
  395. int ret = 0;
  396. mark_point();
  397. va_start(ap, fmt);
  398. ret = alloc_vprintf2(buf, fmt, ap);
  399. va_end(ap);
  400. return ret;
  401. }
  402. START_TEST(test_alloc_vprintf)
  403. {
  404. /* Adapted from unit_test.c */
  405. /* Copyright (c) 2013-2015 the Civetweb developers */
  406. /* Copyright (c) 2004-2013 Sergey Lyubka */
  407. char buf[MG_BUF_LEN], *p = buf;
  408. mark_point();
  409. ck_assert(alloc_printf(&p, sizeof(buf), "%s", "hi") == 2);
  410. ck_assert(p == buf);
  411. ck_assert(alloc_printf(&p, sizeof(buf), "%s", "") == 0);
  412. ck_assert(p == buf);
  413. ck_assert(alloc_printf(&p, sizeof(buf), "") == 0);
  414. ck_assert(p == buf);
  415. /* Pass small buffer, make sure alloc_printf allocates */
  416. ck_assert(alloc_printf(&p, 1, "%s", "hello") == 5);
  417. ck_assert(p != buf);
  418. mg_free(p);
  419. p = buf;
  420. /* Test alternative implementation */
  421. ck_assert(alloc_printf2(&p, "%s", "hello") == 5);
  422. ck_assert(p != buf);
  423. mg_free(p);
  424. p = buf;
  425. }
  426. END_TEST
  427. START_TEST(test_mg_vsnprintf)
  428. {
  429. char buf[16];
  430. int is_trunc;
  431. #if defined(_WIN32)
  432. /* If the string is truncated, mg_snprintf calls mg_cry.
  433. * If DEBUG is defined, mg_cry calls DEBUG_TRACE.
  434. * In DEBUG_TRACE_FUNC, flockfile(stdout) is called.
  435. * For Windows, flockfile/funlockfile calls
  436. * pthread_mutex_lock/_unlock(&global_log_file_lock).
  437. * So, we need to initialize global_log_file_lock:
  438. */
  439. pthread_mutex_init(&global_log_file_lock, &pthread_mutex_attr);
  440. #endif
  441. memset(buf, 0, sizeof(buf));
  442. mark_point();
  443. is_trunc = 777;
  444. mg_snprintf(NULL, &is_trunc, buf, 10, "%8i", 123);
  445. ck_assert_str_eq(buf, " 123");
  446. ck_assert_int_eq(is_trunc, 0);
  447. is_trunc = 777;
  448. mg_snprintf(NULL, &is_trunc, buf, 10, "%9i", 123);
  449. ck_assert_str_eq(buf, " 123");
  450. ck_assert_int_eq(is_trunc, 0);
  451. is_trunc = 777;
  452. mg_snprintf(NULL, &is_trunc, buf, 9, "%9i", 123);
  453. ck_assert_str_eq(buf, " 12");
  454. ck_assert_int_eq(is_trunc, 1);
  455. is_trunc = 777;
  456. mg_snprintf(NULL, &is_trunc, buf, 8, "%9i", 123);
  457. ck_assert_str_eq(buf, " 1");
  458. ck_assert_int_eq(is_trunc, 1);
  459. is_trunc = 777;
  460. mg_snprintf(NULL, &is_trunc, buf, 7, "%9i", 123);
  461. ck_assert_str_eq(buf, " ");
  462. ck_assert_int_eq(is_trunc, 1);
  463. is_trunc = 777;
  464. strcpy(buf, "1234567890");
  465. mg_snprintf(NULL, &is_trunc, buf, 0, "%i", 543);
  466. ck_assert_str_eq(buf, "1234567890");
  467. ck_assert_int_eq(is_trunc, 1);
  468. }
  469. END_TEST
  470. START_TEST(test_mg_strcasestr)
  471. {
  472. /* Adapted from unit_test.c */
  473. /* Copyright (c) 2013-2015 the Civetweb developers */
  474. /* Copyright (c) 2004-2013 Sergey Lyubka */
  475. static const char *big1 = "abcdef";
  476. mark_point();
  477. ck_assert(mg_strcasestr("Y", "X") == NULL);
  478. ck_assert(mg_strcasestr("Y", "y") != NULL);
  479. ck_assert(mg_strcasestr(big1, "X") == NULL);
  480. ck_assert(mg_strcasestr(big1, "CD") == big1 + 2);
  481. ck_assert(mg_strcasestr("aa", "AAB") == NULL);
  482. }
  483. END_TEST
  484. START_TEST(test_parse_port_string)
  485. {
  486. /* Adapted from unit_test.c */
  487. /* Copyright (c) 2013-2020 the Civetweb developers */
  488. /* Copyright (c) 2004-2013 Sergey Lyubka */
  489. struct t_test_parse_port_string {
  490. const char *port_string;
  491. int valid;
  492. int ip_family;
  493. uint16_t port_num;
  494. };
  495. static struct t_test_parse_port_string testdata[] =
  496. { {"0", 1, 4, 0},
  497. {"1", 1, 4, 1},
  498. {"65535", 1, 4, 65535},
  499. {"65536", 0, 0, 0},
  500. {"1s", 1, 4, 1},
  501. {"1r", 1, 4, 1},
  502. {"1k", 0, 0, 0},
  503. {"1.2.3", 0, 0, 0},
  504. {"1.2.3.", 0, 0, 0},
  505. {"1.2.3.4", 0, 0, 0},
  506. {"1.2.3.4:", 0, 0, 0},
  507. {"1.2.3.4:0", 1, 4, 0},
  508. {"1.2.3.4:1", 1, 4, 1},
  509. {"1.2.3.4:65535", 1, 4, 65535},
  510. {"1.2.3.4:65536", 0, 0, 0},
  511. {"1.2.3.4:1s", 1, 4, 1},
  512. {"1.2.3.4:1r", 1, 4, 1},
  513. {"1.2.3.4:1k", 0, 0, 0},
  514. #if defined(USE_IPV6)
  515. /* IPv6 config */
  516. {"[::1]:123", 1, 6, 123},
  517. {"[::]:80", 1, 6, 80},
  518. {"[3ffe:2a00:100:7031::1]:900", 1, 6, 900},
  519. /* IPv4 + IPv6 config */
  520. {"+80", 1, 4 + 6, 80},
  521. #else
  522. /* IPv6 config: invalid if IPv6 is not activated */
  523. {"[::1]:123", 0, 0, 123},
  524. {"[::]:80", 0, 0, 80},
  525. {"[3ffe:2a00:100:7031::1]:900", 0, 0, 900},
  526. /* IPv4 + IPv6 config: only IPv4 if IPv6 is not activated */
  527. {"+80", 1, 4, 80},
  528. #endif
  529. {NULL, 0, 0, 0} };
  530. struct socket so;
  531. struct vec vec;
  532. int ip_family;
  533. int i, ret;
  534. mark_point();
  535. for (i = 0; testdata[i].port_string != NULL; i++) {
  536. vec.ptr = testdata[i].port_string;
  537. vec.len = strlen(vec.ptr);
  538. ip_family = 123;
  539. ret = parse_port_string(&vec, &so, &ip_family);
  540. if ((ret != testdata[i].valid)
  541. || (ip_family != testdata[i].ip_family)) {
  542. ck_abort_msg("Port string [%s]: "
  543. "expected valid=%i, family=%i; "
  544. "got valid=%i, family=%i",
  545. testdata[i].port_string,
  546. testdata[i].valid,
  547. testdata[i].ip_family,
  548. ret,
  549. ip_family);
  550. }
  551. if (ip_family == 4) {
  552. ck_assert_int_eq((int)so.lsa.sin.sin_family, (int)AF_INET);
  553. }
  554. if (ip_family == 6) {
  555. ck_assert_int_eq((int)so.lsa.sin.sin_family, (int)AF_INET6);
  556. }
  557. if (ret) {
  558. /* Test valid strings only */
  559. ck_assert_int_eq(htons(so.lsa.sin.sin_port), testdata[i].port_num);
  560. }
  561. }
  562. /* special case: localhost can be ipv4 or ipv6 */
  563. vec.ptr = "localhost:123";
  564. vec.len = strlen(vec.ptr);
  565. ret = parse_port_string(&vec, &so, &ip_family);
  566. if (ret != 1) {
  567. ck_abort_msg("IP of localhost seems to be unknown on this system (%i)",
  568. (int)ret);
  569. }
  570. if ((ip_family != 4) && (ip_family != 6)) {
  571. ck_abort_msg("IP family for localhost must be 4 or 6 but is %i",
  572. (int)ip_family);
  573. }
  574. ck_assert_int_eq((int)htons(so.lsa.sin.sin_port), (int)123);
  575. }
  576. END_TEST
  577. START_TEST(test_parse_http_headers)
  578. {
  579. char buf[2048];
  580. char *ptr;
  581. int ret;
  582. struct mg_header hdr[MG_MAX_HEADERS];
  583. memset(hdr, 0, sizeof(hdr));
  584. memset(buf, 0, sizeof(buf));
  585. ptr = &buf[0];
  586. ret = parse_http_headers(&ptr, hdr);
  587. ck_assert_int_eq(ret, 0);
  588. ck_assert_ptr_eq(ptr, &buf[0]);
  589. ck_assert_ptr_eq(hdr[0].name, NULL);
  590. ck_assert_ptr_eq(hdr[0].value, NULL);
  591. ck_assert_ptr_eq(hdr[1].name, NULL);
  592. ck_assert_ptr_eq(hdr[1].value, NULL);
  593. ck_assert_ptr_eq(hdr[2].name, NULL);
  594. ck_assert_ptr_eq(hdr[2].value, NULL);
  595. memset(hdr, 0, sizeof(hdr));
  596. memset(buf, 0, sizeof(buf));
  597. strcpy(buf, "\r\n");
  598. ptr = &buf[0];
  599. ret = parse_http_headers(&ptr, hdr);
  600. ck_assert_int_eq(ret, 0);
  601. ck_assert_ptr_eq(ptr, &buf[0]);
  602. ck_assert_ptr_eq(hdr[0].name, NULL);
  603. ck_assert_ptr_eq(hdr[0].value, NULL);
  604. ck_assert_ptr_eq(hdr[1].name, NULL);
  605. ck_assert_ptr_eq(hdr[1].value, NULL);
  606. ck_assert_ptr_eq(hdr[2].name, NULL);
  607. ck_assert_ptr_eq(hdr[2].value, NULL);
  608. memset(hdr, 0, sizeof(hdr));
  609. memset(buf, 0, sizeof(buf));
  610. strcpy(buf, "a\r\n");
  611. ptr = &buf[0];
  612. ret = parse_http_headers(&ptr, hdr);
  613. ck_assert_int_eq(ret, -1);
  614. ck_assert_ptr_eq(ptr, &buf[0]);
  615. ck_assert_ptr_eq(hdr[0].name, NULL);
  616. ck_assert_ptr_eq(hdr[0].value, NULL);
  617. ck_assert_ptr_eq(hdr[1].name, NULL);
  618. ck_assert_ptr_eq(hdr[1].value, NULL);
  619. ck_assert_ptr_eq(hdr[2].name, NULL);
  620. ck_assert_ptr_eq(hdr[2].value, NULL);
  621. memset(hdr, 0, sizeof(hdr));
  622. memset(buf, 0, sizeof(buf));
  623. strcpy(buf, "a:b");
  624. ptr = &buf[0];
  625. ret = parse_http_headers(&ptr, hdr);
  626. ck_assert_int_eq(ret, 1);
  627. ck_assert_ptr_eq(ptr, &buf[3]);
  628. ck_assert_str_eq(hdr[0].name, "a");
  629. ck_assert_str_eq(hdr[0].value, "b");
  630. ck_assert_ptr_eq(hdr[0].name, &buf[0]);
  631. ck_assert_ptr_eq(hdr[0].value, &buf[2]);
  632. ck_assert_ptr_eq(hdr[1].name, NULL);
  633. ck_assert_ptr_eq(hdr[1].value, NULL);
  634. ck_assert_ptr_eq(hdr[2].name, NULL);
  635. ck_assert_ptr_eq(hdr[2].value, NULL);
  636. memset(hdr, 0, sizeof(hdr));
  637. memset(buf, 0, sizeof(buf));
  638. strcpy(buf, "a:b\r\n");
  639. ptr = &buf[0];
  640. ret = parse_http_headers(&ptr, hdr);
  641. ck_assert_int_eq(ret, 1);
  642. ck_assert_ptr_eq(ptr, &buf[5]);
  643. ck_assert_str_eq(hdr[0].name, "a");
  644. ck_assert_str_eq(hdr[0].value, "b");
  645. ck_assert_ptr_eq(hdr[0].name, &buf[0]);
  646. ck_assert_ptr_eq(hdr[0].value, &buf[2]);
  647. ck_assert_ptr_eq(hdr[1].name, NULL);
  648. ck_assert_ptr_eq(hdr[1].value, NULL);
  649. ck_assert_ptr_eq(hdr[2].name, NULL);
  650. ck_assert_ptr_eq(hdr[2].value, NULL);
  651. memset(hdr, 0, sizeof(hdr));
  652. memset(buf, 0, sizeof(buf));
  653. strcpy(buf, "a:b\r\n\r\n");
  654. ptr = &buf[0];
  655. ret = parse_http_headers(&ptr, hdr);
  656. ck_assert_int_eq(ret, 1);
  657. ck_assert_ptr_eq(ptr, &buf[5]);
  658. ck_assert_str_eq(hdr[0].name, "a");
  659. ck_assert_str_eq(hdr[0].value, "b");
  660. ck_assert_ptr_eq(hdr[0].name, &buf[0]);
  661. ck_assert_ptr_eq(hdr[0].value, &buf[2]);
  662. ck_assert_ptr_eq(hdr[1].name, NULL);
  663. ck_assert_ptr_eq(hdr[1].value, NULL);
  664. ck_assert_ptr_eq(hdr[2].name, NULL);
  665. ck_assert_ptr_eq(hdr[2].value, NULL);
  666. memset(hdr, 0, sizeof(hdr));
  667. memset(buf, 0, sizeof(buf));
  668. strcpy(buf, "a: b\r\n");
  669. ptr = &buf[0];
  670. ret = parse_http_headers(&ptr, hdr);
  671. ck_assert_int_eq(ret, 1);
  672. ck_assert_ptr_eq(ptr, &buf[6]);
  673. ck_assert_str_eq(hdr[0].name, "a");
  674. ck_assert_str_eq(hdr[0].value, "b");
  675. ck_assert_ptr_eq(hdr[0].name, &buf[0]);
  676. ck_assert_ptr_eq(hdr[0].value, &buf[3]);
  677. ck_assert_ptr_eq(hdr[1].name, NULL);
  678. ck_assert_ptr_eq(hdr[1].value, NULL);
  679. ck_assert_ptr_eq(hdr[2].name, NULL);
  680. ck_assert_ptr_eq(hdr[2].value, NULL);
  681. memset(hdr, 0, sizeof(hdr));
  682. memset(buf, 0, sizeof(buf));
  683. strcpy(buf, "a :b\r\n");
  684. ptr = &buf[0];
  685. ret = parse_http_headers(&ptr, hdr);
  686. ck_assert_int_eq(ret, 1);
  687. ck_assert_ptr_eq(ptr, &buf[6]);
  688. ck_assert_str_eq(hdr[0].name, "a");
  689. ck_assert_str_eq(hdr[0].value, "b");
  690. ck_assert_ptr_eq(hdr[0].name, &buf[0]);
  691. ck_assert_ptr_eq(hdr[0].value, &buf[3]);
  692. ck_assert_ptr_eq(hdr[1].name, NULL);
  693. ck_assert_ptr_eq(hdr[1].value, NULL);
  694. ck_assert_ptr_eq(hdr[2].name, NULL);
  695. ck_assert_ptr_eq(hdr[2].value, NULL);
  696. memset(hdr, 0, sizeof(hdr));
  697. memset(buf, 0, sizeof(buf));
  698. strcpy(buf, "a : b\r\n");
  699. ptr = &buf[0];
  700. ret = parse_http_headers(&ptr, hdr);
  701. ck_assert_int_eq(ret, 1);
  702. ck_assert_ptr_eq(ptr, &buf[7]);
  703. ck_assert_str_eq(hdr[0].name, "a");
  704. ck_assert_str_eq(hdr[0].value, "b");
  705. ck_assert_ptr_eq(hdr[0].name, &buf[0]);
  706. ck_assert_ptr_eq(hdr[0].value, &buf[4]);
  707. ck_assert_ptr_eq(hdr[1].name, NULL);
  708. ck_assert_ptr_eq(hdr[1].value, NULL);
  709. ck_assert_ptr_eq(hdr[2].name, NULL);
  710. ck_assert_ptr_eq(hdr[2].value, NULL);
  711. memset(hdr, 0, sizeof(hdr));
  712. memset(buf, 0, sizeof(buf));
  713. strcpy(buf, "a: b\r\nc: d\r\n");
  714. ptr = &buf[0];
  715. ret = parse_http_headers(&ptr, hdr);
  716. ck_assert_int_eq(ret, 2);
  717. ck_assert_ptr_eq(ptr, &buf[12]);
  718. ck_assert_str_eq(hdr[0].name, "a");
  719. ck_assert_str_eq(hdr[0].value, "b");
  720. ck_assert_str_eq(hdr[1].name, "c");
  721. ck_assert_str_eq(hdr[1].value, "d");
  722. ck_assert_ptr_eq(hdr[0].name, &buf[0]);
  723. ck_assert_ptr_eq(hdr[0].value, &buf[3]);
  724. ck_assert_ptr_eq(hdr[1].name, &buf[6]);
  725. ck_assert_ptr_eq(hdr[1].value, &buf[9]);
  726. ck_assert_ptr_eq(hdr[2].name, NULL);
  727. ck_assert_ptr_eq(hdr[2].value, NULL);
  728. memset(hdr, 0, sizeof(hdr));
  729. memset(buf, 0, sizeof(buf));
  730. strcpy(buf, "a: b\r\nc: d\r\n\r\n");
  731. ptr = &buf[0];
  732. ret = parse_http_headers(&ptr, hdr);
  733. ck_assert_int_eq(ret, 2);
  734. ck_assert_ptr_eq(ptr, &buf[12]);
  735. ck_assert_str_eq(hdr[0].name, "a");
  736. ck_assert_str_eq(hdr[0].value, "b");
  737. ck_assert_str_eq(hdr[1].name, "c");
  738. ck_assert_str_eq(hdr[1].value, "d");
  739. ck_assert_ptr_eq(hdr[0].name, &buf[0]);
  740. ck_assert_ptr_eq(hdr[0].value, &buf[3]);
  741. ck_assert_ptr_eq(hdr[1].name, &buf[6]);
  742. ck_assert_ptr_eq(hdr[1].value, &buf[9]);
  743. ck_assert_ptr_eq(hdr[2].name, NULL);
  744. ck_assert_ptr_eq(hdr[2].value, NULL);
  745. memset(hdr, 0, sizeof(hdr));
  746. memset(buf, 0, sizeof(buf));
  747. strcpy(buf, "a: b\r\nc: d\r\n\r\ne: f\r\n");
  748. ptr = &buf[0];
  749. ret = parse_http_headers(&ptr, hdr);
  750. ck_assert_int_eq(ret, 2);
  751. ck_assert_ptr_eq(ptr, &buf[12]);
  752. ck_assert_str_eq(hdr[0].name, "a");
  753. ck_assert_str_eq(hdr[0].value, "b");
  754. ck_assert_str_eq(hdr[1].name, "c");
  755. ck_assert_str_eq(hdr[1].value, "d");
  756. ck_assert_ptr_eq(hdr[0].name, &buf[0]);
  757. ck_assert_ptr_eq(hdr[0].value, &buf[3]);
  758. ck_assert_ptr_eq(hdr[1].name, &buf[6]);
  759. ck_assert_ptr_eq(hdr[1].value, &buf[9]);
  760. ck_assert_ptr_eq(hdr[2].name, NULL);
  761. ck_assert_ptr_eq(hdr[2].value, NULL);
  762. memset(hdr, 0, sizeof(hdr));
  763. memset(buf, 0, sizeof(buf));
  764. strcpy(buf, "a: b\r\nc: d");
  765. ptr = &buf[0];
  766. ret = parse_http_headers(&ptr, hdr);
  767. ck_assert_int_eq(ret, 2);
  768. ck_assert_ptr_eq(ptr, &buf[10]);
  769. ck_assert_str_eq(hdr[0].name, "a");
  770. ck_assert_str_eq(hdr[0].value, "b");
  771. ck_assert_str_eq(hdr[1].name, "c");
  772. ck_assert_str_eq(hdr[1].value, "d");
  773. ck_assert_ptr_eq(hdr[0].name, &buf[0]);
  774. ck_assert_ptr_eq(hdr[0].value, &buf[3]);
  775. ck_assert_ptr_eq(hdr[1].name, &buf[6]);
  776. ck_assert_ptr_eq(hdr[1].value, &buf[9]);
  777. ck_assert_ptr_eq(hdr[2].name, NULL);
  778. ck_assert_ptr_eq(hdr[2].value, NULL);
  779. memset(hdr, 0, sizeof(hdr));
  780. memset(buf, 0, sizeof(buf));
  781. strcpy(buf, "a: b\r\nc: d\r\ne");
  782. ptr = &buf[0];
  783. ret = parse_http_headers(&ptr, hdr);
  784. ck_assert_int_eq(ret, -1);
  785. /* The following could be undefined, since ret == -1 */
  786. ck_assert_ptr_eq(ptr, &buf[12]);
  787. ck_assert_str_eq(hdr[0].name, "a");
  788. ck_assert_str_eq(hdr[0].value, "b");
  789. ck_assert_str_eq(hdr[1].name, "c");
  790. ck_assert_str_eq(hdr[1].value, "d");
  791. ck_assert_ptr_eq(hdr[0].name, &buf[0]);
  792. ck_assert_ptr_eq(hdr[0].value, &buf[3]);
  793. ck_assert_ptr_eq(hdr[1].name, &buf[6]);
  794. ck_assert_ptr_eq(hdr[1].value, &buf[9]);
  795. ck_assert_ptr_eq(hdr[2].name, NULL);
  796. ck_assert_ptr_eq(hdr[2].value, NULL);
  797. memset(hdr, 0, sizeof(hdr));
  798. memset(buf, 0, sizeof(buf));
  799. strcpy(buf, "a: b\r\nc: d\r\nefg:");
  800. ptr = &buf[0];
  801. ret = parse_http_headers(&ptr, hdr);
  802. ck_assert_int_eq(ret, 3);
  803. ck_assert_ptr_eq(ptr, &buf[16]);
  804. ck_assert_str_eq(hdr[0].name, "a");
  805. ck_assert_str_eq(hdr[0].value, "b");
  806. ck_assert_str_eq(hdr[1].name, "c");
  807. ck_assert_str_eq(hdr[1].value, "d");
  808. ck_assert_str_eq(hdr[2].name, "efg");
  809. ck_assert_str_eq(hdr[2].value, "");
  810. ck_assert_ptr_eq(hdr[0].name, &buf[0]);
  811. ck_assert_ptr_eq(hdr[0].value, &buf[3]);
  812. ck_assert_ptr_eq(hdr[1].name, &buf[6]);
  813. ck_assert_ptr_eq(hdr[1].value, &buf[9]);
  814. ck_assert_ptr_eq(hdr[2].name, &buf[12]);
  815. ck_assert_ptr_eq(hdr[2].value, &buf[16]);
  816. }
  817. END_TEST
  818. START_TEST(test_encode_decode)
  819. {
  820. char buf[128];
  821. const char *alpha = "abcdefghijklmnopqrstuvwxyz";
  822. const char *nonalpha = " !\"#$%&'()*+,-./0123456789:;<=>?@";
  823. const char *nonalpha_url_enc1 =
  824. "%20%21%22%23$%25%26%27()%2a%2b,-.%2f0123456789%3a;%3c%3d%3e%3f%40";
  825. const char *nonalpha_url_enc2 =
  826. "%20!%22%23%24%25%26'()*%2B%2C-.%2F0123456789%3A%3B%3C%3D%3E%3F%40";
  827. int ret;
  828. size_t len;
  829. #if defined(USE_WEBSOCKET) || defined(USE_LUA)
  830. const char *alpha_b64_enc = "YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXo=";
  831. const char *nonalpha_b64_enc =
  832. "ICEiIyQlJicoKSorLC0uLzAxMjM0NTY3ODk6Ozw9Pj9A";
  833. mark_point();
  834. memset(buf, 77, sizeof(buf));
  835. base64_encode((unsigned char *)"a", 1, buf);
  836. ck_assert_str_eq(buf, "YQ==");
  837. memset(buf, 77, sizeof(buf));
  838. base64_encode((unsigned char *)"ab", 1, buf);
  839. ck_assert_str_eq(buf, "YQ==");
  840. memset(buf, 77, sizeof(buf));
  841. base64_encode((unsigned char *)"ab", 2, buf);
  842. ck_assert_str_eq(buf, "YWI=");
  843. memset(buf, 77, sizeof(buf));
  844. base64_encode((unsigned char *)alpha, 3, buf);
  845. ck_assert_str_eq(buf, "YWJj");
  846. memset(buf, 77, sizeof(buf));
  847. base64_encode((unsigned char *)alpha, 4, buf);
  848. ck_assert_str_eq(buf, "YWJjZA==");
  849. memset(buf, 77, sizeof(buf));
  850. base64_encode((unsigned char *)alpha, 5, buf);
  851. ck_assert_str_eq(buf, "YWJjZGU=");
  852. memset(buf, 77, sizeof(buf));
  853. base64_encode((unsigned char *)alpha, 6, buf);
  854. ck_assert_str_eq(buf, "YWJjZGVm");
  855. memset(buf, 77, sizeof(buf));
  856. base64_encode((unsigned char *)alpha, (int)strlen(alpha), buf);
  857. ck_assert_str_eq(buf, alpha_b64_enc);
  858. memset(buf, 77, sizeof(buf));
  859. base64_encode((unsigned char *)nonalpha, (int)strlen(nonalpha), buf);
  860. ck_assert_str_eq(buf, nonalpha_b64_enc);
  861. #endif
  862. #if defined(USE_LUA)
  863. memset(buf, 77, sizeof(buf));
  864. len = 9999;
  865. ret = base64_decode((unsigned char *)alpha_b64_enc,
  866. (int)strlen(alpha_b64_enc),
  867. buf,
  868. &len);
  869. ck_assert_int_eq(ret, -1);
  870. ck_assert_uint_eq((unsigned int)len, (unsigned int)strlen(alpha));
  871. ck_assert_str_eq(buf, alpha);
  872. memset(buf, 77, sizeof(buf));
  873. len = 9999;
  874. ret = base64_decode((unsigned char *)"AAA*AAA", 7, buf, &len);
  875. ck_assert_int_eq(ret, 3);
  876. #endif
  877. memset(buf, 77, sizeof(buf));
  878. ret = mg_url_encode(alpha, buf, sizeof(buf));
  879. ck_assert_int_eq(ret, (int)strlen(buf));
  880. ck_assert_int_eq(ret, (int)strlen(alpha));
  881. ck_assert_str_eq(buf, alpha);
  882. memset(buf, 77, sizeof(buf));
  883. ret = mg_url_encode(nonalpha, buf, sizeof(buf));
  884. ck_assert_int_eq(ret, (int)strlen(buf));
  885. ck_assert_int_eq(ret, (int)strlen(nonalpha_url_enc1));
  886. ck_assert_str_eq(buf, nonalpha_url_enc1);
  887. memset(buf, 77, sizeof(buf));
  888. ret = mg_url_decode(alpha, (int)strlen(alpha), buf, sizeof(buf), 0);
  889. ck_assert_int_eq(ret, (int)strlen(buf));
  890. ck_assert_int_eq(ret, (int)strlen(alpha));
  891. ck_assert_str_eq(buf, alpha);
  892. memset(buf, 77, sizeof(buf));
  893. ret = mg_url_decode(
  894. nonalpha_url_enc1, (int)strlen(nonalpha_url_enc1), buf, sizeof(buf), 0);
  895. ck_assert_int_eq(ret, (int)strlen(buf));
  896. ck_assert_int_eq(ret, (int)strlen(nonalpha));
  897. ck_assert_str_eq(buf, nonalpha);
  898. memset(buf, 77, sizeof(buf));
  899. ret = mg_url_decode(
  900. nonalpha_url_enc2, (int)strlen(nonalpha_url_enc2), buf, sizeof(buf), 0);
  901. ck_assert_int_eq(ret, (int)strlen(buf));
  902. ck_assert_int_eq(ret, (int)strlen(nonalpha));
  903. ck_assert_str_eq(buf, nonalpha);
  904. /* Test url_decode_in_place */
  905. strcpy(buf, "AbcDef");
  906. url_decode_in_place(buf);
  907. ck_assert_str_eq(buf, "AbcDef");
  908. strcpy(buf, "A%20B+C");
  909. url_decode_in_place(buf);
  910. ck_assert_str_eq(buf, "A B C");
  911. /* len could be unused, if base64_decode is not tested because USE_LUA is
  912. * not defined */
  913. (void)len;
  914. }
  915. END_TEST
  916. START_TEST(test_mask_data)
  917. {
  918. #if defined(USE_WEBSOCKET)
  919. char in[1024];
  920. char out[1024];
  921. int i;
  922. #endif
  923. uint32_t mask = 0x61626364;
  924. /* TODO: adapt test for big endian */
  925. ck_assert((*(unsigned char *)&mask) == 0x64u);
  926. #if defined(USE_WEBSOCKET)
  927. memset(in, 0, sizeof(in));
  928. memset(out, 99, sizeof(out));
  929. mask_data(in, sizeof(out), 0, out);
  930. ck_assert(!memcmp(out, in, sizeof(out)));
  931. for (i = 0; i < 1024; i++) {
  932. in[i] = (char)((unsigned char)i);
  933. }
  934. mask_data(in, 107, 0, out);
  935. ck_assert(!memcmp(out, in, 107));
  936. mask_data(in, 256, 0x01010101, out);
  937. for (i = 0; i < 256; i++) {
  938. ck_assert_int_eq((int)((unsigned char)out[i]),
  939. (int)(((unsigned char)in[i]) ^ (char)1u));
  940. }
  941. for (i = 256; i < (int)sizeof(out); i++) {
  942. ck_assert_int_eq((int)((unsigned char)out[i]), (int)0);
  943. }
  944. /* TODO: check this for big endian */
  945. mask_data(in, 5, 0x01020304, out);
  946. ck_assert_uint_eq((unsigned char)out[0], 0u ^ 4u);
  947. ck_assert_uint_eq((unsigned char)out[1], 1u ^ 3u);
  948. ck_assert_uint_eq((unsigned char)out[2], 2u ^ 2u);
  949. ck_assert_uint_eq((unsigned char)out[3], 3u ^ 1u);
  950. ck_assert_uint_eq((unsigned char)out[4], 4u ^ 4u);
  951. #endif
  952. }
  953. END_TEST
  954. START_TEST(test_parse_date_string)
  955. {
  956. #if !defined(NO_CACHING)
  957. time_t now = time(0);
  958. struct tm *tm = gmtime(&now);
  959. char date[64] = {0};
  960. unsigned long i;
  961. ck_assert_uint_eq((unsigned long)parse_date_string("1/Jan/1970 00:01:02"),
  962. 62ul);
  963. ck_assert_uint_eq((unsigned long)parse_date_string("1 Jan 1970 00:02:03"),
  964. 123ul);
  965. ck_assert_uint_eq((unsigned long)parse_date_string("1-Jan-1970 00:03:04"),
  966. 184ul);
  967. ck_assert_uint_eq((unsigned long)parse_date_string(
  968. "Xyz, 1 Jan 1970 00:04:05"),
  969. 245ul);
  970. gmt_time_string(date, sizeof(date), &now);
  971. ck_assert_uint_eq((uintmax_t)parse_date_string(date), (uintmax_t)now);
  972. sprintf(date,
  973. "%02u %s %04u %02u:%02u:%02u",
  974. (unsigned int)tm->tm_mday,
  975. month_names[tm->tm_mon],
  976. (unsigned int)(tm->tm_year + 1900),
  977. (unsigned int)tm->tm_hour,
  978. (unsigned int)tm->tm_min,
  979. (unsigned int)tm->tm_sec);
  980. ck_assert_uint_eq((uintmax_t)parse_date_string(date), (uintmax_t)now);
  981. gmt_time_string(date, 1, NULL);
  982. ck_assert_str_eq(date, "");
  983. gmt_time_string(date, 6, NULL);
  984. ck_assert_str_eq(date,
  985. "Thu, "); /* part of "Thu, 01 Jan 1970 00:00:00 GMT" */
  986. gmt_time_string(date, sizeof(date), NULL);
  987. ck_assert_str_eq(date, "Thu, 01 Jan 1970 00:00:00 GMT");
  988. for (i = 2ul; i < 0x8000000ul; i += i / 2) {
  989. now = (time_t)i;
  990. gmt_time_string(date, sizeof(date), &now);
  991. ck_assert_uint_eq((uintmax_t)parse_date_string(date), (uintmax_t)now);
  992. tm = gmtime(&now);
  993. sprintf(date,
  994. "%02u-%s-%04u %02u:%02u:%02u",
  995. (unsigned int)tm->tm_mday,
  996. month_names[tm->tm_mon],
  997. (unsigned int)(tm->tm_year + 1900),
  998. (unsigned int)tm->tm_hour,
  999. (unsigned int)tm->tm_min,
  1000. (unsigned int)tm->tm_sec);
  1001. ck_assert_uint_eq((uintmax_t)parse_date_string(date), (uintmax_t)now);
  1002. }
  1003. #endif
  1004. }
  1005. END_TEST
  1006. START_TEST(test_sha1)
  1007. {
  1008. #ifdef SHA1_DIGEST_SIZE
  1009. SHA_CTX sha_ctx;
  1010. uint8_t digest[SHA1_DIGEST_SIZE] = {0};
  1011. char str[48] = {0};
  1012. int i;
  1013. const char *test_str;
  1014. ck_assert_uint_eq(sizeof(digest), 20);
  1015. ck_assert_uint_gt(sizeof(str), sizeof(digest) * 2 + 1);
  1016. /* empty string */
  1017. SHA1_Init(&sha_ctx);
  1018. SHA1_Final(digest, &sha_ctx);
  1019. bin2str(str, digest, sizeof(digest));
  1020. ck_assert_uint_eq(strlen(str), 40);
  1021. ck_assert_str_eq(str, "da39a3ee5e6b4b0d3255bfef95601890afd80709");
  1022. /* empty string */
  1023. SHA1_Init(&sha_ctx);
  1024. SHA1_Update(&sha_ctx, (uint8_t *)"abc", 0);
  1025. SHA1_Final(digest, &sha_ctx);
  1026. bin2str(str, digest, sizeof(digest));
  1027. ck_assert_uint_eq(strlen(str), 40);
  1028. ck_assert_str_eq(str, "da39a3ee5e6b4b0d3255bfef95601890afd80709");
  1029. /* "abc" */
  1030. SHA1_Init(&sha_ctx);
  1031. SHA1_Update(&sha_ctx, (uint8_t *)"abc", 3);
  1032. SHA1_Final(digest, &sha_ctx);
  1033. bin2str(str, digest, sizeof(digest));
  1034. ck_assert_uint_eq(strlen(str), 40);
  1035. ck_assert_str_eq(str, "a9993e364706816aba3e25717850c26c9cd0d89d");
  1036. /* "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq" */
  1037. test_str = "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq";
  1038. SHA1_Init(&sha_ctx);
  1039. SHA1_Update(&sha_ctx, (uint8_t *)test_str, (uint32_t)strlen(test_str));
  1040. SHA1_Final(digest, &sha_ctx);
  1041. bin2str(str, digest, sizeof(digest));
  1042. ck_assert_uint_eq(strlen(str), 40);
  1043. ck_assert_str_eq(str, "84983e441c3bd26ebaae4aa1f95129e5e54670f1");
  1044. /* a million "a" */
  1045. SHA1_Init(&sha_ctx);
  1046. for (i = 0; i < 1000000; i++) {
  1047. SHA1_Update(&sha_ctx, (uint8_t *)"a", 1);
  1048. }
  1049. SHA1_Final(digest, &sha_ctx);
  1050. bin2str(str, digest, sizeof(digest));
  1051. ck_assert_uint_eq(strlen(str), 40);
  1052. ck_assert_str_eq(str, "34aa973cd4c4daa4f61eeb2bdbad27316534016f");
  1053. /* a million "a" in blocks of 10 */
  1054. SHA1_Init(&sha_ctx);
  1055. for (i = 0; i < 100000; i++) {
  1056. SHA1_Update(&sha_ctx, (uint8_t *)"aaaaaaaaaa", 10);
  1057. }
  1058. SHA1_Final(digest, &sha_ctx);
  1059. bin2str(str, digest, sizeof(digest));
  1060. ck_assert_uint_eq(strlen(str), 40);
  1061. ck_assert_str_eq(str, "34aa973cd4c4daa4f61eeb2bdbad27316534016f");
  1062. #else
  1063. /* Can not test, if SHA1 is not included */
  1064. ck_assert(1);
  1065. #endif
  1066. }
  1067. END_TEST
  1068. START_TEST(test_config_options)
  1069. {
  1070. /* Check size of config_options vs. number of options in enum. */
  1071. ck_assert_ptr_eq(NULL, config_options[NUM_OPTIONS].name);
  1072. ck_assert_int_eq((int)MG_CONFIG_TYPE_UNKNOWN,
  1073. config_options[NUM_OPTIONS].type);
  1074. ck_assert_uint_eq(sizeof(config_options) / sizeof(config_options[0]),
  1075. (size_t)(NUM_OPTIONS + 1));
  1076. /* Check option enums vs. option names. */
  1077. /* Check if the order in
  1078. * static struct mg_option config_options[]
  1079. * is the same as in the option enum
  1080. * This test allows to reorder config_options and the enum,
  1081. * and check if the order is still consistent. */
  1082. ck_assert_str_eq("cgi_pattern", config_options[CGI_EXTENSIONS].name);
  1083. ck_assert_str_eq("cgi_environment", config_options[CGI_ENVIRONMENT].name);
  1084. ck_assert_str_eq("put_delete_auth_file",
  1085. config_options[PUT_DELETE_PASSWORDS_FILE].name);
  1086. ck_assert_str_eq("cgi_interpreter", config_options[CGI_INTERPRETER].name);
  1087. ck_assert_str_eq("protect_uri", config_options[PROTECT_URI].name);
  1088. ck_assert_str_eq("authentication_domain",
  1089. config_options[AUTHENTICATION_DOMAIN].name);
  1090. ck_assert_str_eq("enable_auth_domain_check",
  1091. config_options[ENABLE_AUTH_DOMAIN_CHECK].name);
  1092. ck_assert_str_eq("ssi_pattern", config_options[SSI_EXTENSIONS].name);
  1093. ck_assert_str_eq("throttle", config_options[THROTTLE].name);
  1094. ck_assert_str_eq("access_log_file", config_options[ACCESS_LOG_FILE].name);
  1095. ck_assert_str_eq("enable_directory_listing",
  1096. config_options[ENABLE_DIRECTORY_LISTING].name);
  1097. ck_assert_str_eq("error_log_file", config_options[ERROR_LOG_FILE].name);
  1098. ck_assert_str_eq("global_auth_file",
  1099. config_options[GLOBAL_PASSWORDS_FILE].name);
  1100. ck_assert_str_eq("index_files", config_options[INDEX_FILES].name);
  1101. ck_assert_str_eq("enable_keep_alive",
  1102. config_options[ENABLE_KEEP_ALIVE].name);
  1103. ck_assert_str_eq("access_control_list",
  1104. config_options[ACCESS_CONTROL_LIST].name);
  1105. ck_assert_str_eq("extra_mime_types", config_options[EXTRA_MIME_TYPES].name);
  1106. ck_assert_str_eq("listening_ports", config_options[LISTENING_PORTS].name);
  1107. ck_assert_str_eq("document_root", config_options[DOCUMENT_ROOT].name);
  1108. ck_assert_str_eq("ssl_certificate", config_options[SSL_CERTIFICATE].name);
  1109. ck_assert_str_eq("ssl_certificate_chain",
  1110. config_options[SSL_CERTIFICATE_CHAIN].name);
  1111. ck_assert_str_eq("num_threads", config_options[NUM_THREADS].name);
  1112. ck_assert_str_eq("run_as_user", config_options[RUN_AS_USER].name);
  1113. ck_assert_str_eq("url_rewrite_patterns",
  1114. config_options[URL_REWRITE_PATTERN].name);
  1115. ck_assert_str_eq("hide_files_patterns", config_options[HIDE_FILES].name);
  1116. ck_assert_str_eq("request_timeout_ms",
  1117. config_options[REQUEST_TIMEOUT].name);
  1118. ck_assert_str_eq("keep_alive_timeout_ms",
  1119. config_options[KEEP_ALIVE_TIMEOUT].name);
  1120. ck_assert_str_eq("linger_timeout_ms", config_options[LINGER_TIMEOUT].name);
  1121. ck_assert_str_eq("listen_backlog",
  1122. config_options[LISTEN_BACKLOG_SIZE].name);
  1123. ck_assert_str_eq("ssl_verify_peer",
  1124. config_options[SSL_DO_VERIFY_PEER].name);
  1125. ck_assert_str_eq("ssl_ca_path", config_options[SSL_CA_PATH].name);
  1126. ck_assert_str_eq("ssl_ca_file", config_options[SSL_CA_FILE].name);
  1127. ck_assert_str_eq("ssl_verify_depth", config_options[SSL_VERIFY_DEPTH].name);
  1128. ck_assert_str_eq("ssl_default_verify_paths",
  1129. config_options[SSL_DEFAULT_VERIFY_PATHS].name);
  1130. ck_assert_str_eq("ssl_cipher_list", config_options[SSL_CIPHER_LIST].name);
  1131. ck_assert_str_eq("ssl_protocol_version",
  1132. config_options[SSL_PROTOCOL_VERSION].name);
  1133. ck_assert_str_eq("ssl_short_trust", config_options[SSL_SHORT_TRUST].name);
  1134. #if defined(USE_WEBSOCKET)
  1135. ck_assert_str_eq("websocket_timeout_ms",
  1136. config_options[WEBSOCKET_TIMEOUT].name);
  1137. ck_assert_str_eq("enable_websocket_ping_pong",
  1138. config_options[ENABLE_WEBSOCKET_PING_PONG].name);
  1139. #endif
  1140. ck_assert_str_eq("decode_url", config_options[DECODE_URL].name);
  1141. ck_assert_str_eq("decode_query_string",
  1142. config_options[DECODE_QUERY_STRING].name);
  1143. #if defined(USE_LUA)
  1144. ck_assert_str_eq("lua_preload_file", config_options[LUA_PRELOAD_FILE].name);
  1145. ck_assert_str_eq("lua_script_pattern",
  1146. config_options[LUA_SCRIPT_EXTENSIONS].name);
  1147. ck_assert_str_eq("lua_server_page_pattern",
  1148. config_options[LUA_SERVER_PAGE_EXTENSIONS].name);
  1149. #endif
  1150. #if defined(USE_DUKTAPE)
  1151. ck_assert_str_eq("duktape_script_pattern",
  1152. config_options[DUKTAPE_SCRIPT_EXTENSIONS].name);
  1153. #endif
  1154. #if defined(USE_WEBSOCKET)
  1155. ck_assert_str_eq("websocket_root", config_options[WEBSOCKET_ROOT].name);
  1156. #endif
  1157. #if defined(USE_LUA) && defined(USE_WEBSOCKET)
  1158. ck_assert_str_eq("lua_websocket_pattern",
  1159. config_options[LUA_WEBSOCKET_EXTENSIONS].name);
  1160. #endif
  1161. ck_assert_str_eq("access_control_allow_origin",
  1162. config_options[ACCESS_CONTROL_ALLOW_ORIGIN].name);
  1163. ck_assert_str_eq("access_control_allow_methods",
  1164. config_options[ACCESS_CONTROL_ALLOW_METHODS].name);
  1165. ck_assert_str_eq("access_control_allow_headers",
  1166. config_options[ACCESS_CONTROL_ALLOW_HEADERS].name);
  1167. ck_assert_str_eq("error_pages", config_options[ERROR_PAGES].name);
  1168. ck_assert_str_eq("tcp_nodelay", config_options[CONFIG_TCP_NODELAY].name);
  1169. #if !defined(NO_CACHING)
  1170. ck_assert_str_eq("static_file_max_age",
  1171. config_options[STATIC_FILE_MAX_AGE].name);
  1172. #endif
  1173. #if !defined(NO_SSL)
  1174. ck_assert_str_eq("strict_transport_security_max_age",
  1175. config_options[STRICT_HTTPS_MAX_AGE].name);
  1176. #endif
  1177. #if defined(__linux__)
  1178. ck_assert_str_eq("allow_sendfile_call",
  1179. config_options[ALLOW_SENDFILE_CALL].name);
  1180. #endif
  1181. #if defined(_WIN32)
  1182. ck_assert_str_eq("case_sensitive",
  1183. config_options[CASE_SENSITIVE_FILES].name);
  1184. #endif
  1185. #if defined(USE_LUA)
  1186. ck_assert_str_eq("lua_background_script",
  1187. config_options[LUA_BACKGROUND_SCRIPT].name);
  1188. ck_assert_str_eq("lua_background_script_params",
  1189. config_options[LUA_BACKGROUND_SCRIPT_PARAMS].name);
  1190. #endif
  1191. ck_assert_str_eq("additional_header",
  1192. config_options[ADDITIONAL_HEADER].name);
  1193. ck_assert_str_eq("max_request_size", config_options[MAX_REQUEST_SIZE].name);
  1194. ck_assert_str_eq("allow_index_script_resource",
  1195. config_options[ALLOW_INDEX_SCRIPT_SUB_RES].name);
  1196. }
  1197. END_TEST
  1198. #if !defined(REPLACE_CHECK_FOR_LOCAL_DEBUGGING)
  1199. Suite *
  1200. make_private_suite(void)
  1201. {
  1202. Suite *const suite = suite_create("Private");
  1203. TCase *const tcase_http_message = tcase_create("HTTP Message");
  1204. TCase *const tcase_http_keep_alive = tcase_create("HTTP Keep Alive");
  1205. TCase *const tcase_url_parsing_1 = tcase_create("URL Parsing 1");
  1206. TCase *const tcase_url_parsing_2 = tcase_create("URL Parsing 2");
  1207. TCase *const tcase_url_parsing_3 = tcase_create("URL Parsing 3");
  1208. TCase *const tcase_internal_parse_1 = tcase_create("Internal Parsing 1");
  1209. TCase *const tcase_internal_parse_2 = tcase_create("Internal Parsing 2");
  1210. TCase *const tcase_internal_parse_3 = tcase_create("Internal Parsing 3");
  1211. TCase *const tcase_internal_parse_4 = tcase_create("Internal Parsing 4");
  1212. TCase *const tcase_internal_parse_5 = tcase_create("Internal Parsing 5");
  1213. TCase *const tcase_internal_parse_6 = tcase_create("Internal Parsing 6");
  1214. TCase *const tcase_internal_parse_7 = tcase_create("Internal Parsing 7");
  1215. TCase *const tcase_encode_decode = tcase_create("Encode Decode");
  1216. TCase *const tcase_mask_data = tcase_create("Mask Data");
  1217. TCase *const tcase_parse_date_string = tcase_create("Date Parsing");
  1218. TCase *const tcase_sha1 = tcase_create("SHA1");
  1219. TCase *const tcase_config_options = tcase_create("Config Options");
  1220. tcase_add_test(tcase_http_message, test_parse_http_message);
  1221. tcase_set_timeout(tcase_http_message, civetweb_min_test_timeout);
  1222. suite_add_tcase(suite, tcase_http_message);
  1223. tcase_add_test(tcase_http_keep_alive, test_should_keep_alive);
  1224. tcase_set_timeout(tcase_http_keep_alive, civetweb_min_test_timeout);
  1225. suite_add_tcase(suite, tcase_http_keep_alive);
  1226. tcase_add_test(tcase_url_parsing_1, test_match_prefix);
  1227. tcase_set_timeout(tcase_url_parsing_1, civetweb_min_test_timeout);
  1228. suite_add_tcase(suite, tcase_url_parsing_1);
  1229. tcase_add_test(tcase_url_parsing_2, test_remove_dot_segments);
  1230. tcase_set_timeout(tcase_url_parsing_2, civetweb_min_test_timeout);
  1231. suite_add_tcase(suite, tcase_url_parsing_2);
  1232. tcase_add_test(tcase_url_parsing_3, test_is_valid_uri);
  1233. tcase_set_timeout(tcase_url_parsing_3, civetweb_min_test_timeout);
  1234. suite_add_tcase(suite, tcase_url_parsing_3);
  1235. tcase_add_test(tcase_internal_parse_1, test_next_option);
  1236. tcase_set_timeout(tcase_internal_parse_1, civetweb_min_test_timeout);
  1237. suite_add_tcase(suite, tcase_internal_parse_1);
  1238. tcase_add_test(tcase_internal_parse_2, test_skip_quoted);
  1239. tcase_set_timeout(tcase_internal_parse_2, civetweb_min_test_timeout);
  1240. suite_add_tcase(suite, tcase_internal_parse_2);
  1241. tcase_add_test(tcase_internal_parse_3, test_mg_strcasestr);
  1242. tcase_set_timeout(tcase_internal_parse_3, civetweb_min_test_timeout);
  1243. suite_add_tcase(suite, tcase_internal_parse_3);
  1244. tcase_add_test(tcase_internal_parse_4, test_alloc_vprintf);
  1245. tcase_set_timeout(tcase_internal_parse_4, civetweb_min_test_timeout);
  1246. suite_add_tcase(suite, tcase_internal_parse_4);
  1247. tcase_add_test(tcase_internal_parse_5, test_mg_vsnprintf);
  1248. tcase_set_timeout(tcase_internal_parse_5, civetweb_min_test_timeout);
  1249. suite_add_tcase(suite, tcase_internal_parse_5);
  1250. tcase_add_test(tcase_internal_parse_6, test_parse_port_string);
  1251. tcase_set_timeout(tcase_internal_parse_6, civetweb_min_test_timeout);
  1252. suite_add_tcase(suite, tcase_internal_parse_6);
  1253. tcase_add_test(tcase_internal_parse_7, test_parse_http_headers);
  1254. tcase_set_timeout(tcase_internal_parse_7, civetweb_min_test_timeout);
  1255. suite_add_tcase(suite, tcase_internal_parse_7);
  1256. tcase_add_test(tcase_encode_decode, test_encode_decode);
  1257. tcase_set_timeout(tcase_encode_decode, civetweb_min_test_timeout);
  1258. suite_add_tcase(suite, tcase_encode_decode);
  1259. tcase_add_test(tcase_mask_data, test_mask_data);
  1260. tcase_set_timeout(tcase_mask_data, civetweb_min_test_timeout);
  1261. suite_add_tcase(suite, tcase_mask_data);
  1262. tcase_add_test(tcase_parse_date_string, test_parse_date_string);
  1263. tcase_set_timeout(tcase_parse_date_string, civetweb_min_test_timeout);
  1264. suite_add_tcase(suite, tcase_parse_date_string);
  1265. tcase_add_test(tcase_sha1, test_sha1);
  1266. tcase_set_timeout(tcase_sha1, civetweb_min_test_timeout);
  1267. suite_add_tcase(suite, tcase_sha1);
  1268. tcase_add_test(tcase_config_options, test_config_options);
  1269. tcase_set_timeout(tcase_config_options, civetweb_min_test_timeout);
  1270. suite_add_tcase(suite, tcase_config_options);
  1271. return suite;
  1272. }
  1273. #endif
  1274. #ifdef REPLACE_CHECK_FOR_LOCAL_DEBUGGING
  1275. /* Used to debug test cases without using the check framework */
  1276. void
  1277. MAIN_PRIVATE(void)
  1278. {
  1279. #if defined(_WIN32)
  1280. /* test_parse_port_string requires WSAStartup for IPv6 */
  1281. WSADATA data;
  1282. WSAStartup(MAKEWORD(2, 2), &data);
  1283. #endif
  1284. test_alloc_vprintf(0);
  1285. test_mg_vsnprintf(0);
  1286. test_remove_dot_segments(0);
  1287. test_parse_date_string(0);
  1288. test_parse_port_string(0);
  1289. test_parse_http_message(0);
  1290. test_parse_http_headers(0);
  1291. test_sha1(0);
  1292. #if defined(_WIN32)
  1293. WSACleanup();
  1294. #endif
  1295. }
  1296. #endif