test.cc 54 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693
  1. #include <future>
  2. #include <gtest/gtest.h>
  3. #include <httplib.h>
  4. #define SERVER_CERT_FILE "./cert.pem"
  5. #define SERVER_PRIVATE_KEY_FILE "./key.pem"
  6. #define CA_CERT_FILE "./ca-bundle.crt"
  7. #define CLIENT_CA_CERT_FILE "./rootCA.cert.pem"
  8. #define CLIENT_CA_CERT_DIR "."
  9. #define CLIENT_CERT_FILE "./client.cert.pem"
  10. #define CLIENT_PRIVATE_KEY_FILE "./client.key.pem"
  11. #ifdef _WIN32
  12. #include <process.h>
  13. #define msleep(n) ::Sleep(n)
  14. #else
  15. #define msleep(n) ::usleep(n * 1000)
  16. #endif
  17. using namespace std;
  18. using namespace httplib;
  19. const char *HOST = "localhost";
  20. const int PORT = 1234;
  21. const string LONG_QUERY_VALUE = string(25000, '@');
  22. const string LONG_QUERY_URL = "/long-query-value?key=" + LONG_QUERY_VALUE;
  23. const std::string JSON_DATA = "{\"hello\":\"world\"}";
  24. #ifdef _WIN32
  25. TEST(StartupTest, WSAStartup) {
  26. WSADATA wsaData;
  27. int ret = WSAStartup(0x0002, &wsaData);
  28. ASSERT_EQ(0, ret);
  29. }
  30. #endif
  31. TEST(SplitTest, ParseQueryString) {
  32. string s = "key1=val1&key2=val2&key3=val3";
  33. Params dic;
  34. detail::split(s.c_str(), s.c_str() + s.size(), '&',
  35. [&](const char *b, const char *e) {
  36. string key, val;
  37. detail::split(b, e, '=', [&](const char *b, const char *e) {
  38. if (key.empty()) {
  39. key.assign(b, e);
  40. } else {
  41. val.assign(b, e);
  42. }
  43. });
  44. dic.emplace(key, val);
  45. });
  46. EXPECT_EQ("val1", dic.find("key1")->second);
  47. EXPECT_EQ("val2", dic.find("key2")->second);
  48. EXPECT_EQ("val3", dic.find("key3")->second);
  49. }
  50. TEST(ParseQueryTest, ParseQueryString) {
  51. string s = "key1=val1&key2=val2&key3=val3";
  52. Params dic;
  53. detail::parse_query_text(s, dic);
  54. EXPECT_EQ("val1", dic.find("key1")->second);
  55. EXPECT_EQ("val2", dic.find("key2")->second);
  56. EXPECT_EQ("val3", dic.find("key3")->second);
  57. }
  58. TEST(GetHeaderValueTest, DefaultValue) {
  59. Headers headers = {{"Dummy", "Dummy"}};
  60. auto val = detail::get_header_value(headers, "Content-Type", 0, "text/plain");
  61. EXPECT_STREQ("text/plain", val);
  62. }
  63. TEST(GetHeaderValueTest, DefaultValueInt) {
  64. Headers headers = {{"Dummy", "Dummy"}};
  65. auto val = detail::get_header_value_uint64(headers, "Content-Length", 100);
  66. EXPECT_EQ(100ull, val);
  67. }
  68. TEST(GetHeaderValueTest, RegularValue) {
  69. Headers headers = {{"Content-Type", "text/html"}, {"Dummy", "Dummy"}};
  70. auto val = detail::get_header_value(headers, "Content-Type", 0, "text/plain");
  71. EXPECT_STREQ("text/html", val);
  72. }
  73. TEST(GetHeaderValueTest, RegularValueInt) {
  74. Headers headers = {{"Content-Length", "100"}, {"Dummy", "Dummy"}};
  75. auto val = detail::get_header_value_uint64(headers, "Content-Length", 0);
  76. EXPECT_EQ(100ull, val);
  77. }
  78. TEST(GetHeaderValueTest, Range) {
  79. {
  80. Headers headers = {make_range_header({{1, -1}})};
  81. auto val = detail::get_header_value(headers, "Range", 0, 0);
  82. EXPECT_STREQ("bytes=1-", val);
  83. }
  84. {
  85. Headers headers = {make_range_header({{-1, 1}})};
  86. auto val = detail::get_header_value(headers, "Range", 0, 0);
  87. EXPECT_STREQ("bytes=-1", val);
  88. }
  89. {
  90. Headers headers = {make_range_header({{1, 10}})};
  91. auto val = detail::get_header_value(headers, "Range", 0, 0);
  92. EXPECT_STREQ("bytes=1-10", val);
  93. }
  94. {
  95. Headers headers = {make_range_header({{1, 10}, {100, -1}})};
  96. auto val = detail::get_header_value(headers, "Range", 0, 0);
  97. EXPECT_STREQ("bytes=1-10, 100-", val);
  98. }
  99. {
  100. Headers headers = {make_range_header({{1, 10}, {100, 200}})};
  101. auto val = detail::get_header_value(headers, "Range", 0, 0);
  102. EXPECT_STREQ("bytes=1-10, 100-200", val);
  103. }
  104. {
  105. Headers headers = {make_range_header({{0, 0}, {-1, 1}})};
  106. auto val = detail::get_header_value(headers, "Range", 0, 0);
  107. EXPECT_STREQ("bytes=0-0, -1", val);
  108. }
  109. }
  110. TEST(ParseHeaderValueTest, Range) {
  111. {
  112. Ranges ranges;
  113. auto ret = detail::parse_range_header("bytes=1-", ranges);
  114. EXPECT_TRUE(ret);
  115. EXPECT_EQ(1u, ranges.size());
  116. EXPECT_EQ(1u, ranges[0].first);
  117. EXPECT_EQ(-1, ranges[0].second);
  118. }
  119. {
  120. Ranges ranges;
  121. auto ret = detail::parse_range_header("bytes=-1", ranges);
  122. EXPECT_TRUE(ret);
  123. EXPECT_EQ(1u, ranges.size());
  124. EXPECT_EQ(-1, ranges[0].first);
  125. EXPECT_EQ(1u, ranges[0].second);
  126. }
  127. {
  128. Ranges ranges;
  129. auto ret = detail::parse_range_header("bytes=1-10", ranges);
  130. EXPECT_TRUE(ret);
  131. EXPECT_EQ(1u, ranges.size());
  132. EXPECT_EQ(1u, ranges[0].first);
  133. EXPECT_EQ(10u, ranges[0].second);
  134. }
  135. {
  136. Ranges ranges;
  137. auto ret = detail::parse_range_header("bytes=10-1", ranges);
  138. EXPECT_FALSE(ret);
  139. }
  140. {
  141. Ranges ranges;
  142. auto ret = detail::parse_range_header("bytes=1-10, 100-", ranges);
  143. EXPECT_TRUE(ret);
  144. EXPECT_EQ(2u, ranges.size());
  145. EXPECT_EQ(1u, ranges[0].first);
  146. EXPECT_EQ(10u, ranges[0].second);
  147. EXPECT_EQ(100u, ranges[1].first);
  148. EXPECT_EQ(-1, ranges[1].second);
  149. }
  150. {
  151. Ranges ranges;
  152. auto ret =
  153. detail::parse_range_header("bytes=1-10, 100-200, 300-400", ranges);
  154. EXPECT_TRUE(ret);
  155. EXPECT_EQ(3u, ranges.size());
  156. EXPECT_EQ(1u, ranges[0].first);
  157. EXPECT_EQ(10u, ranges[0].second);
  158. EXPECT_EQ(100u, ranges[1].first);
  159. EXPECT_EQ(200u, ranges[1].second);
  160. EXPECT_EQ(300u, ranges[2].first);
  161. EXPECT_EQ(400u, ranges[2].second);
  162. }
  163. }
  164. TEST(ChunkedEncodingTest, FromHTTPWatch) {
  165. auto host = "www.httpwatch.com";
  166. auto sec = 2;
  167. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  168. auto port = 443;
  169. httplib::SSLClient cli(host, port, sec);
  170. #else
  171. auto port = 80;
  172. httplib::Client cli(host, port, sec);
  173. #endif
  174. auto res =
  175. cli.Get("/httpgallery/chunked/chunkedimage.aspx?0.4153841143030137");
  176. ASSERT_TRUE(res != nullptr);
  177. std::string out;
  178. httplib::detail::read_file("./image.jpg", out);
  179. EXPECT_EQ(200, res->status);
  180. EXPECT_EQ(out, res->body);
  181. }
  182. TEST(ChunkedEncodingTest, WithContentReceiver) {
  183. auto host = "www.httpwatch.com";
  184. auto sec = 2;
  185. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  186. auto port = 443;
  187. httplib::SSLClient cli(host, port, sec);
  188. #else
  189. auto port = 80;
  190. httplib::Client cli(host, port, sec);
  191. #endif
  192. std::string body;
  193. auto res =
  194. cli.Get("/httpgallery/chunked/chunkedimage.aspx?0.4153841143030137",
  195. [&](const char *data, size_t len) { body.append(data, len); });
  196. ASSERT_TRUE(res != nullptr);
  197. std::string out;
  198. httplib::detail::read_file("./image.jpg", out);
  199. EXPECT_EQ(200, res->status);
  200. EXPECT_EQ(out, body);
  201. }
  202. TEST(RangeTest, FromHTTPBin) {
  203. auto host = "httpbin.org";
  204. auto sec = 5;
  205. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  206. auto port = 443;
  207. httplib::SSLClient cli(host, port, sec);
  208. #else
  209. auto port = 80;
  210. httplib::Client cli(host, port, sec);
  211. #endif
  212. {
  213. httplib::Headers headers;
  214. auto res = cli.Get("/range/32", headers);
  215. ASSERT_TRUE(res != nullptr);
  216. EXPECT_EQ(res->body, "abcdefghijklmnopqrstuvwxyzabcdef");
  217. EXPECT_EQ(200, res->status);
  218. }
  219. {
  220. httplib::Headers headers = {httplib::make_range_header({{1, -1}})};
  221. auto res = cli.Get("/range/32", headers);
  222. ASSERT_TRUE(res != nullptr);
  223. EXPECT_EQ(res->body, "bcdefghijklmnopqrstuvwxyzabcdef");
  224. EXPECT_EQ(206, res->status);
  225. }
  226. {
  227. httplib::Headers headers = {httplib::make_range_header({{1, 10}})};
  228. auto res = cli.Get("/range/32", headers);
  229. ASSERT_TRUE(res != nullptr);
  230. EXPECT_EQ(res->body, "bcdefghijk");
  231. EXPECT_EQ(206, res->status);
  232. }
  233. {
  234. httplib::Headers headers = {httplib::make_range_header({{0, 31}})};
  235. auto res = cli.Get("/range/32", headers);
  236. ASSERT_TRUE(res != nullptr);
  237. EXPECT_EQ(res->body, "abcdefghijklmnopqrstuvwxyzabcdef");
  238. EXPECT_EQ(200, res->status);
  239. }
  240. {
  241. httplib::Headers headers = {httplib::make_range_header({{0, -1}})};
  242. auto res = cli.Get("/range/32", headers);
  243. ASSERT_TRUE(res != nullptr);
  244. EXPECT_EQ(res->body, "abcdefghijklmnopqrstuvwxyzabcdef");
  245. EXPECT_EQ(200, res->status);
  246. }
  247. {
  248. httplib::Headers headers = {httplib::make_range_header({{0, 32}})};
  249. auto res = cli.Get("/range/32", headers);
  250. ASSERT_TRUE(res != nullptr);
  251. EXPECT_EQ(416, res->status);
  252. }
  253. }
  254. TEST(ConnectionErrorTest, InvalidHost) {
  255. auto host = "abcde.com";
  256. auto sec = 2;
  257. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  258. auto port = 443;
  259. httplib::SSLClient cli(host, port, sec);
  260. #else
  261. auto port = 80;
  262. httplib::Client cli(host, port, sec);
  263. #endif
  264. auto res = cli.Get("/");
  265. ASSERT_TRUE(res == nullptr);
  266. }
  267. TEST(ConnectionErrorTest, InvalidPort) {
  268. auto host = "localhost";
  269. auto sec = 2;
  270. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  271. auto port = 44380;
  272. httplib::SSLClient cli(host, port, sec);
  273. #else
  274. auto port = 8080;
  275. httplib::Client cli(host, port, sec);
  276. #endif
  277. auto res = cli.Get("/");
  278. ASSERT_TRUE(res == nullptr);
  279. }
  280. TEST(ConnectionErrorTest, Timeout) {
  281. auto host = "google.com";
  282. auto sec = 2;
  283. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  284. auto port = 44380;
  285. httplib::SSLClient cli(host, port, sec);
  286. #else
  287. auto port = 8080;
  288. httplib::Client cli(host, port, sec);
  289. #endif
  290. auto res = cli.Get("/");
  291. ASSERT_TRUE(res == nullptr);
  292. }
  293. TEST(CancelTest, NoCancel) {
  294. auto host = "httpbin.org";
  295. auto sec = 5;
  296. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  297. auto port = 443;
  298. httplib::SSLClient cli(host, port, sec);
  299. #else
  300. auto port = 80;
  301. httplib::Client cli(host, port, sec);
  302. #endif
  303. auto res = cli.Get("/range/32", [](uint64_t, uint64_t) { return true; });
  304. ASSERT_TRUE(res != nullptr);
  305. EXPECT_EQ(res->body, "abcdefghijklmnopqrstuvwxyzabcdef");
  306. EXPECT_EQ(200, res->status);
  307. }
  308. TEST(CancelTest, WithCancelSmallPayload) {
  309. auto host = "httpbin.org";
  310. auto sec = 5;
  311. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  312. auto port = 443;
  313. httplib::SSLClient cli(host, port, sec);
  314. #else
  315. auto port = 80;
  316. httplib::Client cli(host, port, sec);
  317. #endif
  318. auto res = cli.Get("/range/32", [](uint64_t, uint64_t) { return false; });
  319. ASSERT_TRUE(res == nullptr);
  320. }
  321. TEST(CancelTest, WithCancelLargePayload) {
  322. auto host = "httpbin.org";
  323. auto sec = 5;
  324. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  325. auto port = 443;
  326. httplib::SSLClient cli(host, port, sec);
  327. #else
  328. auto port = 80;
  329. httplib::Client cli(host, port, sec);
  330. #endif
  331. uint32_t count = 0;
  332. httplib::Headers headers;
  333. auto res = cli.Get("/range/65536", headers,
  334. [&count](uint64_t, uint64_t) { return (count++ == 0); });
  335. ASSERT_TRUE(res == nullptr);
  336. }
  337. TEST(BaseAuthTest, FromHTTPWatch) {
  338. auto host = "httpbin.org";
  339. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  340. auto port = 443;
  341. httplib::SSLClient cli(host, port);
  342. #else
  343. auto port = 80;
  344. httplib::Client cli(host, port);
  345. #endif
  346. {
  347. auto res = cli.Get("/basic-auth/hello/world");
  348. ASSERT_TRUE(res != nullptr);
  349. EXPECT_EQ(401, res->status);
  350. }
  351. {
  352. auto res =
  353. cli.Get("/basic-auth/hello/world",
  354. {httplib::make_basic_authentication_header("hello", "world")});
  355. ASSERT_TRUE(res != nullptr);
  356. EXPECT_EQ(res->body,
  357. "{\n \"authenticated\": true, \n \"user\": \"hello\"\n}\n");
  358. EXPECT_EQ(200, res->status);
  359. }
  360. }
  361. TEST(Server, BindAndListenSeparately) {
  362. Server svr;
  363. int port = svr.bind_to_any_port("localhost");
  364. ASSERT_TRUE(port > 0);
  365. svr.stop();
  366. }
  367. class ServerTest : public ::testing::Test {
  368. protected:
  369. ServerTest()
  370. : cli_(HOST, PORT)
  371. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  372. ,
  373. svr_(SERVER_CERT_FILE, SERVER_PRIVATE_KEY_FILE)
  374. #endif
  375. {
  376. }
  377. virtual void SetUp() {
  378. svr_.set_base_dir("./www");
  379. svr_.Get("/hi",
  380. [&](const Request & /*req*/, Response &res) {
  381. res.set_content("Hello World!", "text/plain");
  382. })
  383. .Get("/slow",
  384. [&](const Request & /*req*/, Response &res) {
  385. msleep(2000);
  386. res.set_content("slow", "text/plain");
  387. })
  388. .Get("/remote_addr",
  389. [&](const Request &req, Response &res) {
  390. auto remote_addr = req.headers.find("REMOTE_ADDR")->second;
  391. res.set_content(remote_addr.c_str(), "text/plain");
  392. })
  393. .Get("/endwith%",
  394. [&](const Request & /*req*/, Response &res) {
  395. res.set_content("Hello World!", "text/plain");
  396. })
  397. .Get("/", [&](const Request & /*req*/,
  398. Response &res) { res.set_redirect("/hi"); })
  399. .Post("/person",
  400. [&](const Request &req, Response &res) {
  401. if (req.has_param("name") && req.has_param("note")) {
  402. persons_[req.get_param_value("name")] =
  403. req.get_param_value("note");
  404. } else {
  405. res.status = 400;
  406. }
  407. })
  408. .Get("/person/(.*)",
  409. [&](const Request &req, Response &res) {
  410. string name = req.matches[1];
  411. if (persons_.find(name) != persons_.end()) {
  412. auto note = persons_[name];
  413. res.set_content(note, "text/plain");
  414. } else {
  415. res.status = 404;
  416. }
  417. })
  418. .Post("/x-www-form-urlencoded-json",
  419. [&](const Request &req, Response &res) {
  420. auto json = req.get_param_value("json");
  421. ASSERT_EQ(JSON_DATA, json);
  422. res.set_content(json, "appliation/json");
  423. res.status = 200;
  424. })
  425. .Get("/streamed-chunked",
  426. [&](const Request & /*req*/, Response &res) {
  427. res.set_chunked_content_provider(
  428. [](uint64_t /*offset*/, Out out, Done done) {
  429. out("123", 3);
  430. out("456", 3);
  431. out("789", 3);
  432. done();
  433. });
  434. })
  435. .Get("/streamed",
  436. [&](const Request & /*req*/, Response &res) {
  437. res.set_content_provider(
  438. 6, [](uint64_t offset, uint64_t /*length*/, Out out) {
  439. if (offset < 3) {
  440. out("a", 1);
  441. } else {
  442. out("b", 1);
  443. }
  444. });
  445. })
  446. .Get("/streamed-with-range",
  447. [&](const Request & /*req*/, Response &res) {
  448. auto data = std::make_shared<std::string>("abcdefg");
  449. res.set_content_provider(
  450. data->size(),
  451. [data](uint64_t offset, uint64_t length, Out out) {
  452. const uint64_t DATA_CHUNK_SIZE = 4;
  453. const auto &d = *data;
  454. out(&d[offset], std::min(length, DATA_CHUNK_SIZE));
  455. });
  456. })
  457. .Get("/with-range",
  458. [&](const Request & /*req*/, Response &res) {
  459. res.set_content("abcdefg", "text/plain");
  460. })
  461. .Post("/chunked",
  462. [&](const Request &req, Response & /*res*/) {
  463. EXPECT_EQ(req.body, "dechunked post body");
  464. })
  465. .Post("/large-chunked",
  466. [&](const Request &req, Response & /*res*/) {
  467. std::string expected(6 * 30 * 1024u, 'a');
  468. EXPECT_EQ(req.body, expected);
  469. })
  470. .Post("/multipart",
  471. [&](const Request &req, Response & /*res*/) {
  472. EXPECT_EQ(5u, req.files.size());
  473. ASSERT_TRUE(!req.has_file("???"));
  474. {
  475. const auto &file = req.get_file_value("text1");
  476. EXPECT_EQ("", file.filename);
  477. EXPECT_EQ("text default",
  478. req.body.substr(file.offset, file.length));
  479. }
  480. {
  481. const auto &file = req.get_file_value("text2");
  482. EXPECT_EQ("", file.filename);
  483. EXPECT_EQ("aωb", req.body.substr(file.offset, file.length));
  484. }
  485. {
  486. const auto &file = req.get_file_value("file1");
  487. EXPECT_EQ("hello.txt", file.filename);
  488. EXPECT_EQ("text/plain", file.content_type);
  489. EXPECT_EQ("h\ne\n\nl\nl\no\n",
  490. req.body.substr(file.offset, file.length));
  491. }
  492. {
  493. const auto &file = req.get_file_value("file3");
  494. EXPECT_EQ("", file.filename);
  495. EXPECT_EQ("application/octet-stream", file.content_type);
  496. EXPECT_EQ(0u, file.length);
  497. }
  498. })
  499. .Post("/empty",
  500. [&](const Request &req, Response &res) {
  501. EXPECT_EQ(req.body, "");
  502. res.set_content("empty", "text/plain");
  503. })
  504. .Put("/put",
  505. [&](const Request &req, Response &res) {
  506. EXPECT_EQ(req.body, "PUT");
  507. res.set_content(req.body, "text/plain");
  508. })
  509. .Patch("/patch",
  510. [&](const Request &req, Response &res) {
  511. EXPECT_EQ(req.body, "PATCH");
  512. res.set_content(req.body, "text/plain");
  513. })
  514. .Delete("/delete",
  515. [&](const Request & /*req*/, Response &res) {
  516. res.set_content("DELETE", "text/plain");
  517. })
  518. .Options(R"(\*)",
  519. [&](const Request & /*req*/, Response &res) {
  520. res.set_header("Allow", "GET, POST, HEAD, OPTIONS");
  521. })
  522. .Get("/request-target",
  523. [&](const Request &req, Response & /*res*/) {
  524. EXPECT_EQ("/request-target?aaa=bbb&ccc=ddd", req.target);
  525. EXPECT_EQ("bbb", req.get_param_value("aaa"));
  526. EXPECT_EQ("ddd", req.get_param_value("ccc"));
  527. })
  528. .Get("/long-query-value",
  529. [&](const Request &req, Response & /*res*/) {
  530. EXPECT_EQ(LONG_QUERY_URL, req.target);
  531. EXPECT_EQ(LONG_QUERY_VALUE, req.get_param_value("key"));
  532. })
  533. .Get("/array-param",
  534. [&](const Request &req, Response & /*res*/) {
  535. EXPECT_EQ(3u, req.get_param_value_count("array"));
  536. EXPECT_EQ("value1", req.get_param_value("array", 0));
  537. EXPECT_EQ("value2", req.get_param_value("array", 1));
  538. EXPECT_EQ("value3", req.get_param_value("array", 2));
  539. })
  540. .Post("/validate-no-multiple-headers",
  541. [&](const Request &req, Response & /*res*/) {
  542. EXPECT_EQ(1u, req.get_header_value_count("Content-Length"));
  543. EXPECT_EQ("5", req.get_header_value("Content-Length"));
  544. })
  545. #ifdef CPPHTTPLIB_ZLIB_SUPPORT
  546. .Get("/gzip",
  547. [&](const Request & /*req*/, Response &res) {
  548. res.set_content(
  549. "12345678901234567890123456789012345678901234567890123456789"
  550. "01234567890123456789012345678901234567890",
  551. "text/plain");
  552. })
  553. .Get("/nogzip",
  554. [&](const Request & /*req*/, Response &res) {
  555. res.set_content(
  556. "12345678901234567890123456789012345678901234567890123456789"
  557. "01234567890123456789012345678901234567890",
  558. "application/octet-stream");
  559. })
  560. .Post("/gzipmultipart",
  561. [&](const Request &req, Response & /*res*/) {
  562. EXPECT_EQ(2u, req.files.size());
  563. ASSERT_TRUE(!req.has_file("???"));
  564. {
  565. const auto &file = req.get_file_value("key1");
  566. EXPECT_EQ("", file.filename);
  567. EXPECT_EQ("test", req.body.substr(file.offset, file.length));
  568. }
  569. {
  570. const auto &file = req.get_file_value("key2");
  571. EXPECT_EQ("", file.filename);
  572. EXPECT_EQ("--abcdefg123",
  573. req.body.substr(file.offset, file.length));
  574. }
  575. })
  576. #endif
  577. ;
  578. persons_["john"] = "programmer";
  579. t_ = thread([&]() { ASSERT_TRUE(svr_.listen(HOST, PORT)); });
  580. while (!svr_.is_running()) {
  581. msleep(1);
  582. }
  583. }
  584. virtual void TearDown() {
  585. svr_.stop();
  586. for (auto &t : request_threads_) {
  587. t.join();
  588. }
  589. t_.join();
  590. }
  591. map<string, string> persons_;
  592. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  593. SSLClient cli_;
  594. SSLServer svr_;
  595. #else
  596. Client cli_;
  597. Server svr_;
  598. #endif
  599. thread t_;
  600. std::vector<thread> request_threads_;
  601. };
  602. TEST_F(ServerTest, GetMethod200) {
  603. auto res = cli_.Get("/hi");
  604. ASSERT_TRUE(res != nullptr);
  605. EXPECT_EQ("HTTP/1.1", res->version);
  606. EXPECT_EQ(200, res->status);
  607. EXPECT_EQ("text/plain", res->get_header_value("Content-Type"));
  608. EXPECT_EQ(1, res->get_header_value_count("Content-Type"));
  609. EXPECT_EQ("close", res->get_header_value("Connection"));
  610. EXPECT_EQ("Hello World!", res->body);
  611. }
  612. TEST_F(ServerTest, GetMethod302) {
  613. auto res = cli_.Get("/");
  614. ASSERT_TRUE(res != nullptr);
  615. EXPECT_EQ(302, res->status);
  616. EXPECT_EQ("/hi", res->get_header_value("Location"));
  617. }
  618. TEST_F(ServerTest, GetMethod404) {
  619. auto res = cli_.Get("/invalid");
  620. ASSERT_TRUE(res != nullptr);
  621. EXPECT_EQ(404, res->status);
  622. }
  623. TEST_F(ServerTest, HeadMethod200) {
  624. auto res = cli_.Head("/hi");
  625. ASSERT_TRUE(res != nullptr);
  626. EXPECT_EQ(200, res->status);
  627. EXPECT_EQ("text/plain", res->get_header_value("Content-Type"));
  628. EXPECT_EQ("", res->body);
  629. }
  630. TEST_F(ServerTest, HeadMethod404) {
  631. auto res = cli_.Head("/invalid");
  632. ASSERT_TRUE(res != nullptr);
  633. EXPECT_EQ(404, res->status);
  634. EXPECT_EQ("", res->body);
  635. }
  636. TEST_F(ServerTest, GetMethodPersonJohn) {
  637. auto res = cli_.Get("/person/john");
  638. ASSERT_TRUE(res != nullptr);
  639. EXPECT_EQ(200, res->status);
  640. EXPECT_EQ("text/plain", res->get_header_value("Content-Type"));
  641. EXPECT_EQ("programmer", res->body);
  642. }
  643. TEST_F(ServerTest, PostMethod1) {
  644. auto res = cli_.Get("/person/john1");
  645. ASSERT_TRUE(res != nullptr);
  646. ASSERT_EQ(404, res->status);
  647. res = cli_.Post("/person", "name=john1&note=coder",
  648. "application/x-www-form-urlencoded");
  649. ASSERT_TRUE(res != nullptr);
  650. ASSERT_EQ(200, res->status);
  651. res = cli_.Get("/person/john1");
  652. ASSERT_TRUE(res != nullptr);
  653. ASSERT_EQ(200, res->status);
  654. ASSERT_EQ("text/plain", res->get_header_value("Content-Type"));
  655. ASSERT_EQ("coder", res->body);
  656. }
  657. TEST_F(ServerTest, PostMethod2) {
  658. auto res = cli_.Get("/person/john2");
  659. ASSERT_TRUE(res != nullptr);
  660. ASSERT_EQ(404, res->status);
  661. Params params;
  662. params.emplace("name", "john2");
  663. params.emplace("note", "coder");
  664. res = cli_.Post("/person", params);
  665. ASSERT_TRUE(res != nullptr);
  666. ASSERT_EQ(200, res->status);
  667. res = cli_.Get("/person/john2");
  668. ASSERT_TRUE(res != nullptr);
  669. ASSERT_EQ(200, res->status);
  670. ASSERT_EQ("text/plain", res->get_header_value("Content-Type"));
  671. ASSERT_EQ("coder", res->body);
  672. }
  673. TEST_F(ServerTest, PostWwwFormUrlEncodedJson) {
  674. Params params;
  675. params.emplace("json", JSON_DATA);
  676. auto res = cli_.Post("/x-www-form-urlencoded-json", params);
  677. ASSERT_TRUE(res != nullptr);
  678. ASSERT_EQ(200, res->status);
  679. ASSERT_EQ(JSON_DATA, res->body);
  680. }
  681. TEST_F(ServerTest, PostEmptyContent) {
  682. auto res = cli_.Post("/empty", "", "text/plain");
  683. ASSERT_TRUE(res != nullptr);
  684. ASSERT_EQ(200, res->status);
  685. ASSERT_EQ("empty", res->body);
  686. }
  687. TEST_F(ServerTest, GetMethodDir) {
  688. auto res = cli_.Get("/dir/");
  689. ASSERT_TRUE(res != nullptr);
  690. EXPECT_EQ(200, res->status);
  691. EXPECT_EQ("text/html", res->get_header_value("Content-Type"));
  692. auto body = R"(<html>
  693. <head>
  694. </head>
  695. <body>
  696. <a href="/dir/test.html">Test</a>
  697. <a href="/hi">hi</a>
  698. </body>
  699. </html>
  700. )";
  701. EXPECT_EQ(body, res->body);
  702. }
  703. TEST_F(ServerTest, GetMethodDirTest) {
  704. auto res = cli_.Get("/dir/test.html");
  705. ASSERT_TRUE(res != nullptr);
  706. EXPECT_EQ(200, res->status);
  707. EXPECT_EQ("text/html", res->get_header_value("Content-Type"));
  708. EXPECT_EQ("test.html", res->body);
  709. }
  710. TEST_F(ServerTest, GetMethodDirTestWithDoubleDots) {
  711. auto res = cli_.Get("/dir/../dir/test.html");
  712. ASSERT_TRUE(res != nullptr);
  713. EXPECT_EQ(200, res->status);
  714. EXPECT_EQ("text/html", res->get_header_value("Content-Type"));
  715. EXPECT_EQ("test.html", res->body);
  716. }
  717. TEST_F(ServerTest, GetMethodInvalidPath) {
  718. auto res = cli_.Get("/dir/../test.html");
  719. ASSERT_TRUE(res != nullptr);
  720. EXPECT_EQ(404, res->status);
  721. }
  722. TEST_F(ServerTest, GetMethodOutOfBaseDir) {
  723. auto res = cli_.Get("/../www/dir/test.html");
  724. ASSERT_TRUE(res != nullptr);
  725. EXPECT_EQ(404, res->status);
  726. }
  727. TEST_F(ServerTest, GetMethodOutOfBaseDir2) {
  728. auto res = cli_.Get("/dir/../../www/dir/test.html");
  729. ASSERT_TRUE(res != nullptr);
  730. EXPECT_EQ(404, res->status);
  731. }
  732. TEST_F(ServerTest, InvalidBaseDir) {
  733. EXPECT_EQ(false, svr_.set_base_dir("invalid_dir"));
  734. EXPECT_EQ(true, svr_.set_base_dir("."));
  735. }
  736. TEST_F(ServerTest, EmptyRequest) {
  737. auto res = cli_.Get("");
  738. ASSERT_TRUE(res == nullptr);
  739. }
  740. TEST_F(ServerTest, LongRequest) {
  741. auto res =
  742. cli_.Get("/TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  743. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  744. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  745. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  746. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  747. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  748. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  749. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  750. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  751. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  752. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  753. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  754. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  755. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  756. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  757. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  758. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  759. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  760. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  761. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  762. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  763. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  764. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  765. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  766. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  767. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  768. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  769. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  770. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  771. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  772. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  773. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  774. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  775. "TooLongRequest/TooLongRequest/TooLongRequest/__ok__");
  776. ASSERT_TRUE(res != nullptr);
  777. EXPECT_EQ(404, res->status);
  778. }
  779. TEST_F(ServerTest, TooLongRequest) {
  780. auto res =
  781. cli_.Get("/TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  782. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  783. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  784. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  785. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  786. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  787. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  788. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  789. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  790. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  791. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  792. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  793. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  794. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  795. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  796. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  797. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  798. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  799. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  800. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  801. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  802. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  803. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  804. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  805. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  806. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  807. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  808. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  809. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  810. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  811. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  812. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  813. "TooLongRequest/TooLongRequest/TooLongRequest/TooLongRequest/"
  814. "TooLongRequest/TooLongRequest/TooLongRequest/__ng___");
  815. ASSERT_TRUE(res != nullptr);
  816. EXPECT_EQ(404, res->status);
  817. }
  818. TEST_F(ServerTest, LongHeader) {
  819. Request req;
  820. req.method = "GET";
  821. req.path = "/hi";
  822. std::string host_and_port;
  823. host_and_port += HOST;
  824. host_and_port += ":";
  825. host_and_port += std::to_string(PORT);
  826. req.headers.emplace("Host", host_and_port.c_str());
  827. req.headers.emplace("Accept", "*/*");
  828. req.headers.emplace("User-Agent", "cpp-httplib/0.1");
  829. req.headers.emplace(
  830. "Header-Name",
  831. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  832. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  833. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  834. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  835. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  836. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  837. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  838. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  839. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  840. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  841. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  842. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  843. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  844. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  845. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  846. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  847. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  848. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  849. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  850. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  851. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  852. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  853. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  854. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  855. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  856. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  857. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  858. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  859. "@@@@@@@@@@@@@@@@");
  860. auto res = std::make_shared<Response>();
  861. auto ret = cli_.send(req, *res);
  862. ASSERT_TRUE(ret);
  863. EXPECT_EQ(200, res->status);
  864. }
  865. TEST_F(ServerTest, LongQueryValue) {
  866. auto res = cli_.Get(LONG_QUERY_URL.c_str());
  867. ASSERT_TRUE(res != nullptr);
  868. EXPECT_EQ(414, res->status);
  869. }
  870. TEST_F(ServerTest, TooLongHeader) {
  871. Request req;
  872. req.method = "GET";
  873. req.path = "/hi";
  874. std::string host_and_port;
  875. host_and_port += HOST;
  876. host_and_port += ":";
  877. host_and_port += std::to_string(PORT);
  878. req.headers.emplace("Host", host_and_port.c_str());
  879. req.headers.emplace("Accept", "*/*");
  880. req.headers.emplace("User-Agent", "cpp-httplib/0.1");
  881. req.headers.emplace(
  882. "Header-Name",
  883. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  884. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  885. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  886. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  887. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  888. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  889. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  890. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  891. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  892. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  893. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  894. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  895. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  896. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  897. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  898. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  899. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  900. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  901. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  902. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  903. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  904. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  905. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  906. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  907. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  908. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  909. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  910. "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  911. "@@@@@@@@@@@@@@@@@");
  912. auto res = std::make_shared<Response>();
  913. auto ret = cli_.send(req, *res);
  914. ASSERT_TRUE(ret);
  915. EXPECT_EQ(200, res->status);
  916. }
  917. TEST_F(ServerTest, PercentEncoding) {
  918. auto res = cli_.Get("/e%6edwith%");
  919. ASSERT_TRUE(res != nullptr);
  920. EXPECT_EQ(200, res->status);
  921. }
  922. TEST_F(ServerTest, PercentEncodingUnicode) {
  923. auto res = cli_.Get("/e%u006edwith%");
  924. ASSERT_TRUE(res != nullptr);
  925. EXPECT_EQ(200, res->status);
  926. }
  927. TEST_F(ServerTest, InvalidPercentEncoding) {
  928. auto res = cli_.Get("/%endwith%");
  929. ASSERT_TRUE(res != nullptr);
  930. EXPECT_EQ(404, res->status);
  931. }
  932. TEST_F(ServerTest, InvalidPercentEncodingUnicode) {
  933. auto res = cli_.Get("/%uendwith%");
  934. ASSERT_TRUE(res != nullptr);
  935. EXPECT_EQ(404, res->status);
  936. }
  937. TEST_F(ServerTest, EndWithPercentCharacterInQuery) {
  938. auto res = cli_.Get("/hello?aaa=bbb%");
  939. ASSERT_TRUE(res != nullptr);
  940. EXPECT_EQ(404, res->status);
  941. }
  942. TEST_F(ServerTest, MultipartFormData) {
  943. MultipartFormDataItems items = {
  944. {"text1", "text default", "", ""},
  945. {"text2", "aωb", "", ""},
  946. {"file1", "h\ne\n\nl\nl\no\n", "hello.txt", "text/plain"},
  947. {"file2", "{\n \"world\", true\n}\n", "world.json", "application/json"},
  948. {"file3", "", "", "application/octet-stream"},
  949. };
  950. auto res = cli_.Post("/multipart", items);
  951. ASSERT_TRUE(res != nullptr);
  952. EXPECT_EQ(200, res->status);
  953. }
  954. TEST_F(ServerTest, CaseInsensitiveHeaderName) {
  955. auto res = cli_.Get("/hi");
  956. ASSERT_TRUE(res != nullptr);
  957. EXPECT_EQ(200, res->status);
  958. EXPECT_EQ("text/plain", res->get_header_value("content-type"));
  959. EXPECT_EQ("Hello World!", res->body);
  960. }
  961. TEST_F(ServerTest, CaseInsensitiveTransferEncoding) {
  962. Request req;
  963. req.method = "POST";
  964. req.path = "/chunked";
  965. std::string host_and_port;
  966. host_and_port += HOST;
  967. host_and_port += ":";
  968. host_and_port += std::to_string(PORT);
  969. req.headers.emplace("Host", host_and_port.c_str());
  970. req.headers.emplace("Accept", "*/*");
  971. req.headers.emplace("User-Agent", "cpp-httplib/0.1");
  972. req.headers.emplace("Content-Type", "text/plain");
  973. req.headers.emplace("Content-Length", "0");
  974. req.headers.emplace(
  975. "Transfer-Encoding",
  976. "Chunked"); // Note, "Chunked" rather than typical "chunked".
  977. // Client does not chunk, so make a chunked body manually.
  978. req.body = "4\r\ndech\r\nf\r\nunked post body\r\n0\r\n\r\n";
  979. auto res = std::make_shared<Response>();
  980. auto ret = cli_.send(req, *res);
  981. ASSERT_TRUE(ret);
  982. EXPECT_EQ(200, res->status);
  983. }
  984. TEST_F(ServerTest, GetStreamed2) {
  985. auto res = cli_.Get("/streamed", {{make_range_header({{2, 3}})}});
  986. ASSERT_TRUE(res != nullptr);
  987. EXPECT_EQ(206, res->status);
  988. EXPECT_EQ("2", res->get_header_value("Content-Length"));
  989. EXPECT_EQ(std::string("ab"), res->body);
  990. }
  991. TEST_F(ServerTest, GetStreamed) {
  992. auto res = cli_.Get("/streamed");
  993. ASSERT_TRUE(res != nullptr);
  994. EXPECT_EQ(200, res->status);
  995. EXPECT_EQ("6", res->get_header_value("Content-Length"));
  996. EXPECT_EQ(std::string("aaabbb"), res->body);
  997. }
  998. TEST_F(ServerTest, GetStreamedWithRange1) {
  999. auto res = cli_.Get("/streamed-with-range", {{make_range_header({{3, 5}})}});
  1000. ASSERT_TRUE(res != nullptr);
  1001. EXPECT_EQ(206, res->status);
  1002. EXPECT_EQ("3", res->get_header_value("Content-Length"));
  1003. EXPECT_EQ(std::string("def"), res->body);
  1004. }
  1005. TEST_F(ServerTest, GetStreamedWithRange2) {
  1006. auto res = cli_.Get("/streamed-with-range", {{make_range_header({{1, -1}})}});
  1007. ASSERT_TRUE(res != nullptr);
  1008. EXPECT_EQ(206, res->status);
  1009. EXPECT_EQ("6", res->get_header_value("Content-Length"));
  1010. EXPECT_EQ(std::string("bcdefg"), res->body);
  1011. }
  1012. TEST_F(ServerTest, GetStreamedWithRangeMultipart) {
  1013. auto res =
  1014. cli_.Get("/streamed-with-range", {{make_range_header({{1, 2}, {4, 5}})}});
  1015. ASSERT_TRUE(res != nullptr);
  1016. EXPECT_EQ(206, res->status);
  1017. EXPECT_EQ("269", res->get_header_value("Content-Length"));
  1018. EXPECT_EQ(269, res->body.size());
  1019. }
  1020. TEST_F(ServerTest, GetWithRange1) {
  1021. auto res = cli_.Get("/with-range", {{make_range_header({{3, 5}})}});
  1022. ASSERT_TRUE(res != nullptr);
  1023. EXPECT_EQ(206, res->status);
  1024. EXPECT_EQ("3", res->get_header_value("Content-Length"));
  1025. EXPECT_EQ(std::string("def"), res->body);
  1026. }
  1027. TEST_F(ServerTest, GetWithRange2) {
  1028. auto res = cli_.Get("/with-range", {{make_range_header({{1, -1}})}});
  1029. ASSERT_TRUE(res != nullptr);
  1030. EXPECT_EQ(206, res->status);
  1031. EXPECT_EQ("6", res->get_header_value("Content-Length"));
  1032. EXPECT_EQ(std::string("bcdefg"), res->body);
  1033. }
  1034. TEST_F(ServerTest, GetWithRange3) {
  1035. auto res = cli_.Get("/with-range", {{make_range_header({{0, 0}})}});
  1036. ASSERT_TRUE(res != nullptr);
  1037. EXPECT_EQ(206, res->status);
  1038. EXPECT_EQ("1", res->get_header_value("Content-Length"));
  1039. EXPECT_EQ(std::string("a"), res->body);
  1040. }
  1041. TEST_F(ServerTest, GetWithRange4) {
  1042. auto res = cli_.Get("/with-range", {{make_range_header({{-1, 2}})}});
  1043. ASSERT_TRUE(res != nullptr);
  1044. EXPECT_EQ(206, res->status);
  1045. EXPECT_EQ("2", res->get_header_value("Content-Length"));
  1046. EXPECT_EQ(std::string("fg"), res->body);
  1047. }
  1048. TEST_F(ServerTest, GetWithRangeMultipart) {
  1049. auto res = cli_.Get("/with-range", {{make_range_header({{1, 2}, {4, 5}})}});
  1050. ASSERT_TRUE(res != nullptr);
  1051. EXPECT_EQ(206, res->status);
  1052. EXPECT_EQ("269", res->get_header_value("Content-Length"));
  1053. EXPECT_EQ(269, res->body.size());
  1054. }
  1055. TEST_F(ServerTest, GetStreamedChunked) {
  1056. auto res = cli_.Get("/streamed-chunked");
  1057. ASSERT_TRUE(res != nullptr);
  1058. EXPECT_EQ(200, res->status);
  1059. EXPECT_EQ(std::string("123456789"), res->body);
  1060. }
  1061. TEST_F(ServerTest, LargeChunkedPost) {
  1062. Request req;
  1063. req.method = "POST";
  1064. req.path = "/large-chunked";
  1065. std::string host_and_port;
  1066. host_and_port += HOST;
  1067. host_and_port += ":";
  1068. host_and_port += std::to_string(PORT);
  1069. req.headers.emplace("Host", host_and_port.c_str());
  1070. req.headers.emplace("Accept", "*/*");
  1071. req.headers.emplace("User-Agent", "cpp-httplib/0.1");
  1072. req.headers.emplace("Content-Type", "text/plain");
  1073. req.headers.emplace("Content-Length", "0");
  1074. req.headers.emplace("Transfer-Encoding", "chunked");
  1075. std::string long_string(30 * 1024u, 'a');
  1076. std::string chunk = "7800\r\n" + long_string + "\r\n";
  1077. // Attempt to make a large enough post to exceed OS buffers, to test that
  1078. // the server handles short reads if the full chunk data isn't available.
  1079. req.body = chunk + chunk + chunk + chunk + chunk + chunk + "0\r\n\r\n";
  1080. auto res = std::make_shared<Response>();
  1081. auto ret = cli_.send(req, *res);
  1082. ASSERT_TRUE(ret);
  1083. EXPECT_EQ(200, res->status);
  1084. }
  1085. TEST_F(ServerTest, GetMethodRemoteAddr) {
  1086. auto res = cli_.Get("/remote_addr");
  1087. ASSERT_TRUE(res != nullptr);
  1088. EXPECT_EQ(200, res->status);
  1089. EXPECT_EQ("text/plain", res->get_header_value("Content-Type"));
  1090. EXPECT_TRUE(res->body == "::1" || res->body == "127.0.0.1");
  1091. }
  1092. TEST_F(ServerTest, SlowRequest) {
  1093. request_threads_.push_back(
  1094. std::thread([=]() { auto res = cli_.Get("/slow"); }));
  1095. request_threads_.push_back(
  1096. std::thread([=]() { auto res = cli_.Get("/slow"); }));
  1097. request_threads_.push_back(
  1098. std::thread([=]() { auto res = cli_.Get("/slow"); }));
  1099. msleep(100);
  1100. }
  1101. TEST_F(ServerTest, Put) {
  1102. auto res = cli_.Put("/put", "PUT", "text/plain");
  1103. ASSERT_TRUE(res != nullptr);
  1104. EXPECT_EQ(200, res->status);
  1105. EXPECT_EQ("PUT", res->body);
  1106. }
  1107. TEST_F(ServerTest, Patch) {
  1108. auto res = cli_.Patch("/patch", "PATCH", "text/plain");
  1109. ASSERT_TRUE(res != nullptr);
  1110. EXPECT_EQ(200, res->status);
  1111. EXPECT_EQ("PATCH", res->body);
  1112. }
  1113. TEST_F(ServerTest, Delete) {
  1114. auto res = cli_.Delete("/delete");
  1115. ASSERT_TRUE(res != nullptr);
  1116. EXPECT_EQ(200, res->status);
  1117. EXPECT_EQ("DELETE", res->body);
  1118. }
  1119. TEST_F(ServerTest, Options) {
  1120. auto res = cli_.Options("*");
  1121. ASSERT_TRUE(res != nullptr);
  1122. EXPECT_EQ(200, res->status);
  1123. EXPECT_EQ("GET, POST, HEAD, OPTIONS", res->get_header_value("Allow"));
  1124. EXPECT_TRUE(res->body.empty());
  1125. }
  1126. TEST_F(ServerTest, URL) {
  1127. auto res = cli_.Get("/request-target?aaa=bbb&ccc=ddd");
  1128. ASSERT_TRUE(res != nullptr);
  1129. EXPECT_EQ(200, res->status);
  1130. }
  1131. TEST_F(ServerTest, ArrayParam) {
  1132. auto res = cli_.Get("/array-param?array=value1&array=value2&array=value3");
  1133. ASSERT_TRUE(res != nullptr);
  1134. EXPECT_EQ(200, res->status);
  1135. }
  1136. TEST_F(ServerTest, NoMultipleHeaders) {
  1137. Headers headers = {{"Content-Length", "5"}};
  1138. auto res = cli_.Post("/validate-no-multiple-headers", headers, "hello",
  1139. "text/plain");
  1140. ASSERT_TRUE(res != nullptr);
  1141. EXPECT_EQ(200, res->status);
  1142. }
  1143. #ifdef CPPHTTPLIB_ZLIB_SUPPORT
  1144. TEST_F(ServerTest, Gzip) {
  1145. Headers headers;
  1146. headers.emplace("Accept-Encoding", "gzip, deflate");
  1147. auto res = cli_.Get("/gzip", headers);
  1148. ASSERT_TRUE(res != nullptr);
  1149. EXPECT_EQ("gzip", res->get_header_value("Content-Encoding"));
  1150. EXPECT_EQ("text/plain", res->get_header_value("Content-Type"));
  1151. EXPECT_EQ("33", res->get_header_value("Content-Length"));
  1152. EXPECT_EQ("123456789012345678901234567890123456789012345678901234567890123456"
  1153. "7890123456789012345678901234567890",
  1154. res->body);
  1155. EXPECT_EQ(200, res->status);
  1156. }
  1157. TEST_F(ServerTest, GzipWithContentReceiver) {
  1158. Headers headers;
  1159. headers.emplace("Accept-Encoding", "gzip, deflate");
  1160. std::string body;
  1161. auto res = cli_.Get("/gzip", headers, [&](const char *data, size_t len) {
  1162. body.append(data, len);
  1163. });
  1164. ASSERT_TRUE(res != nullptr);
  1165. EXPECT_EQ("gzip", res->get_header_value("Content-Encoding"));
  1166. EXPECT_EQ("text/plain", res->get_header_value("Content-Type"));
  1167. EXPECT_EQ("33", res->get_header_value("Content-Length"));
  1168. EXPECT_EQ("123456789012345678901234567890123456789012345678901234567890123456"
  1169. "7890123456789012345678901234567890",
  1170. body);
  1171. EXPECT_EQ(200, res->status);
  1172. }
  1173. TEST_F(ServerTest, NoGzip) {
  1174. Headers headers;
  1175. headers.emplace("Accept-Encoding", "gzip, deflate");
  1176. auto res = cli_.Get("/nogzip", headers);
  1177. ASSERT_TRUE(res != nullptr);
  1178. EXPECT_EQ(false, res->has_header("Content-Encoding"));
  1179. EXPECT_EQ("application/octet-stream", res->get_header_value("Content-Type"));
  1180. EXPECT_EQ("100", res->get_header_value("Content-Length"));
  1181. EXPECT_EQ("123456789012345678901234567890123456789012345678901234567890123456"
  1182. "7890123456789012345678901234567890",
  1183. res->body);
  1184. EXPECT_EQ(200, res->status);
  1185. }
  1186. TEST_F(ServerTest, NoGzipWithContentReceiver) {
  1187. Headers headers;
  1188. headers.emplace("Accept-Encoding", "gzip, deflate");
  1189. std::string body;
  1190. auto res = cli_.Get("/nogzip", headers, [&](const char *data, size_t len) {
  1191. body.append(data, len);
  1192. });
  1193. ASSERT_TRUE(res != nullptr);
  1194. EXPECT_EQ(false, res->has_header("Content-Encoding"));
  1195. EXPECT_EQ("application/octet-stream", res->get_header_value("Content-Type"));
  1196. EXPECT_EQ("100", res->get_header_value("Content-Length"));
  1197. EXPECT_EQ("123456789012345678901234567890123456789012345678901234567890123456"
  1198. "7890123456789012345678901234567890",
  1199. body);
  1200. EXPECT_EQ(200, res->status);
  1201. }
  1202. TEST_F(ServerTest, MultipartFormDataGzip) {
  1203. Request req;
  1204. req.method = "POST";
  1205. req.path = "/gzipmultipart";
  1206. std::string host_and_port;
  1207. host_and_port += HOST;
  1208. host_and_port += ":";
  1209. host_and_port += std::to_string(PORT);
  1210. req.headers.emplace("Host", host_and_port.c_str());
  1211. req.headers.emplace("Accept", "*/*");
  1212. req.headers.emplace("User-Agent", "cpp-httplib/0.1");
  1213. req.headers.emplace(
  1214. "Content-Type",
  1215. "multipart/form-data; boundary=------------------------fcba8368a9f48c0f");
  1216. req.headers.emplace("Content-Encoding", "gzip");
  1217. // compressed_body generated by creating input.txt to this file:
  1218. /*
  1219. --------------------------fcba8368a9f48c0f
  1220. Content-Disposition: form-data; name="key1"
  1221. test
  1222. --------------------------fcba8368a9f48c0f
  1223. Content-Disposition: form-data; name="key2"
  1224. --abcdefg123
  1225. --------------------------fcba8368a9f48c0f--
  1226. */
  1227. // then running unix2dos input.txt; gzip -9 -c input.txt | xxd -i.
  1228. uint8_t compressed_body[] = {
  1229. 0x1f, 0x8b, 0x08, 0x08, 0x48, 0xf1, 0xd4, 0x5a, 0x02, 0x03, 0x69, 0x6e,
  1230. 0x70, 0x75, 0x74, 0x2e, 0x74, 0x78, 0x74, 0x00, 0xd3, 0xd5, 0xc5, 0x05,
  1231. 0xd2, 0x92, 0x93, 0x12, 0x2d, 0x8c, 0xcd, 0x2c, 0x12, 0x2d, 0xd3, 0x4c,
  1232. 0x2c, 0x92, 0x0d, 0xd2, 0x78, 0xb9, 0x9c, 0xf3, 0xf3, 0x4a, 0x52, 0xf3,
  1233. 0x4a, 0x74, 0x5d, 0x32, 0x8b, 0x0b, 0xf2, 0x8b, 0x33, 0x4b, 0x32, 0xf3,
  1234. 0xf3, 0xac, 0x14, 0xd2, 0xf2, 0x8b, 0x72, 0x75, 0x53, 0x12, 0x4b, 0x12,
  1235. 0xad, 0x15, 0xf2, 0x12, 0x73, 0x53, 0x6d, 0x95, 0xb2, 0x53, 0x2b, 0x0d,
  1236. 0x95, 0x78, 0xb9, 0x78, 0xb9, 0x4a, 0x52, 0x8b, 0x4b, 0x78, 0xb9, 0x74,
  1237. 0x69, 0x61, 0x81, 0x11, 0xd8, 0x02, 0x5d, 0xdd, 0xc4, 0xa4, 0xe4, 0x94,
  1238. 0xd4, 0xb4, 0x74, 0x43, 0x23, 0x63, 0x52, 0x2c, 0xd2, 0xd5, 0xe5, 0xe5,
  1239. 0x02, 0x00, 0xff, 0x0e, 0x72, 0xdf, 0xf8, 0x00, 0x00, 0x00};
  1240. req.body = std::string((char *)compressed_body,
  1241. sizeof(compressed_body) / sizeof(compressed_body[0]));
  1242. auto res = std::make_shared<Response>();
  1243. auto ret = cli_.send(req, *res);
  1244. ASSERT_TRUE(ret);
  1245. EXPECT_EQ(200, res->status);
  1246. }
  1247. #endif
  1248. class ServerTestWithAI_PASSIVE : public ::testing::Test {
  1249. protected:
  1250. ServerTestWithAI_PASSIVE()
  1251. : cli_(HOST, PORT)
  1252. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  1253. ,
  1254. svr_(SERVER_CERT_FILE, SERVER_PRIVATE_KEY_FILE)
  1255. #endif
  1256. {
  1257. }
  1258. virtual void SetUp() {
  1259. svr_.Get("/hi", [&](const Request & /*req*/, Response &res) {
  1260. res.set_content("Hello World!", "text/plain");
  1261. });
  1262. t_ = thread([&]() { ASSERT_TRUE(svr_.listen(nullptr, PORT, AI_PASSIVE)); });
  1263. while (!svr_.is_running()) {
  1264. msleep(1);
  1265. }
  1266. }
  1267. virtual void TearDown() {
  1268. svr_.stop();
  1269. t_.join();
  1270. }
  1271. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  1272. SSLClient cli_;
  1273. SSLServer svr_;
  1274. #else
  1275. Client cli_;
  1276. Server svr_;
  1277. #endif
  1278. thread t_;
  1279. };
  1280. TEST_F(ServerTestWithAI_PASSIVE, GetMethod200) {
  1281. auto res = cli_.Get("/hi");
  1282. ASSERT_TRUE(res != nullptr);
  1283. EXPECT_EQ(200, res->status);
  1284. EXPECT_EQ("text/plain", res->get_header_value("Content-Type"));
  1285. EXPECT_EQ("Hello World!", res->body);
  1286. }
  1287. class ServerUpDownTest : public ::testing::Test {
  1288. protected:
  1289. ServerUpDownTest() : cli_(HOST, PORT) {}
  1290. virtual void SetUp() {
  1291. t_ = thread([&]() {
  1292. svr_.bind_to_any_port(HOST);
  1293. msleep(500);
  1294. ASSERT_TRUE(svr_.listen_after_bind());
  1295. });
  1296. while (!svr_.is_running()) {
  1297. msleep(1);
  1298. }
  1299. }
  1300. virtual void TearDown() {
  1301. svr_.stop();
  1302. t_.join();
  1303. }
  1304. Client cli_;
  1305. Server svr_;
  1306. thread t_;
  1307. };
  1308. TEST_F(ServerUpDownTest, QuickStartStop) {
  1309. // Should not crash, especially when run with
  1310. // --gtest_filter=ServerUpDownTest.QuickStartStop --gtest_repeat=1000
  1311. }
  1312. class PayloadMaxLengthTest : public ::testing::Test {
  1313. protected:
  1314. PayloadMaxLengthTest()
  1315. : cli_(HOST, PORT)
  1316. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  1317. ,
  1318. svr_(SERVER_CERT_FILE, SERVER_PRIVATE_KEY_FILE)
  1319. #endif
  1320. {
  1321. }
  1322. virtual void SetUp() {
  1323. svr_.set_payload_max_length(8);
  1324. svr_.Post("/test", [&](const Request & /*req*/, Response &res) {
  1325. res.set_content("test", "text/plain");
  1326. });
  1327. t_ = thread([&]() { ASSERT_TRUE(svr_.listen(HOST, PORT)); });
  1328. while (!svr_.is_running()) {
  1329. msleep(1);
  1330. }
  1331. }
  1332. virtual void TearDown() {
  1333. svr_.stop();
  1334. t_.join();
  1335. }
  1336. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  1337. SSLClient cli_;
  1338. SSLServer svr_;
  1339. #else
  1340. Client cli_;
  1341. Server svr_;
  1342. #endif
  1343. thread t_;
  1344. };
  1345. TEST_F(PayloadMaxLengthTest, ExceedLimit) {
  1346. auto res = cli_.Post("/test", "123456789", "text/plain");
  1347. ASSERT_TRUE(res != nullptr);
  1348. EXPECT_EQ(413, res->status);
  1349. res = cli_.Post("/test", "12345678", "text/plain");
  1350. ASSERT_TRUE(res != nullptr);
  1351. EXPECT_EQ(200, res->status);
  1352. }
  1353. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  1354. TEST(SSLClientTest, ServerNameIndication) {
  1355. SSLClient cli("httpbin.org", 443);
  1356. auto res = cli.Get("/get");
  1357. ASSERT_TRUE(res != nullptr);
  1358. ASSERT_EQ(200, res->status);
  1359. }
  1360. TEST(SSLClientTest, ServerCertificateVerification) {
  1361. SSLClient cli("google.com");
  1362. auto res = cli.Get("/");
  1363. ASSERT_TRUE(res != nullptr);
  1364. ASSERT_EQ(301, res->status);
  1365. cli.enable_server_certificate_verification(true);
  1366. res = cli.Get("/");
  1367. ASSERT_TRUE(res == nullptr);
  1368. cli.set_ca_cert_path(CA_CERT_FILE);
  1369. res = cli.Get("/");
  1370. ASSERT_TRUE(res != nullptr);
  1371. ASSERT_EQ(301, res->status);
  1372. }
  1373. TEST(SSLClientTest, WildcardHostNameMatch) {
  1374. SSLClient cli("www.youtube.com");
  1375. cli.set_ca_cert_path(CA_CERT_FILE);
  1376. cli.enable_server_certificate_verification(true);
  1377. auto res = cli.Get("/");
  1378. ASSERT_TRUE(res != nullptr);
  1379. ASSERT_EQ(200, res->status);
  1380. }
  1381. TEST(SSLClientServerTest, ClientCertPresent) {
  1382. SSLServer svr(SERVER_CERT_FILE, SERVER_PRIVATE_KEY_FILE, CLIENT_CA_CERT_FILE,
  1383. CLIENT_CA_CERT_DIR);
  1384. ASSERT_TRUE(svr.is_valid());
  1385. svr.Get("/test", [&](const Request &req, Response &res) {
  1386. res.set_content("test", "text/plain");
  1387. svr.stop();
  1388. ASSERT_TRUE(true);
  1389. auto peer_cert = SSL_get_peer_certificate(req.ssl);
  1390. ASSERT_TRUE(peer_cert != nullptr);
  1391. auto subject_name = X509_get_subject_name(peer_cert);
  1392. ASSERT_TRUE(subject_name != nullptr);
  1393. std::string common_name;
  1394. {
  1395. char name[BUFSIZ];
  1396. auto name_len = X509_NAME_get_text_by_NID(subject_name, NID_commonName,
  1397. name, sizeof(name));
  1398. common_name.assign(name, name_len);
  1399. }
  1400. EXPECT_EQ("Common Name", common_name);
  1401. X509_free(peer_cert);
  1402. });
  1403. thread t = thread([&]() { ASSERT_TRUE(svr.listen(HOST, PORT)); });
  1404. msleep(1);
  1405. httplib::SSLClient cli(HOST, PORT, 30, CLIENT_CERT_FILE,
  1406. CLIENT_PRIVATE_KEY_FILE);
  1407. auto res = cli.Get("/test");
  1408. ASSERT_TRUE(res != nullptr);
  1409. ASSERT_EQ(200, res->status);
  1410. t.join();
  1411. }
  1412. TEST(SSLClientServerTest, ClientCertMissing) {
  1413. SSLServer svr(SERVER_CERT_FILE, SERVER_PRIVATE_KEY_FILE, CLIENT_CA_CERT_FILE,
  1414. CLIENT_CA_CERT_DIR);
  1415. ASSERT_TRUE(svr.is_valid());
  1416. svr.Get("/test", [&](const Request &, Response &) { ASSERT_TRUE(false); });
  1417. thread t = thread([&]() { ASSERT_TRUE(svr.listen(HOST, PORT)); });
  1418. msleep(1);
  1419. httplib::SSLClient cli(HOST, PORT, 30);
  1420. auto res = cli.Get("/test");
  1421. ASSERT_TRUE(res == nullptr);
  1422. svr.stop();
  1423. t.join();
  1424. }
  1425. TEST(SSLClientServerTest, TrustDirOptional) {
  1426. SSLServer svr(SERVER_CERT_FILE, SERVER_PRIVATE_KEY_FILE, CLIENT_CA_CERT_FILE);
  1427. ASSERT_TRUE(svr.is_valid());
  1428. svr.Get("/test", [&](const Request &, Response &res) {
  1429. res.set_content("test", "text/plain");
  1430. svr.stop();
  1431. });
  1432. thread t = thread([&]() { ASSERT_TRUE(svr.listen(HOST, PORT)); });
  1433. msleep(1);
  1434. httplib::SSLClient cli(HOST, PORT, 30, CLIENT_CERT_FILE,
  1435. CLIENT_PRIVATE_KEY_FILE);
  1436. auto res = cli.Get("/test");
  1437. ASSERT_TRUE(res != nullptr);
  1438. ASSERT_EQ(200, res->status);
  1439. t.join();
  1440. }
  1441. /* Cannot test this case as there is no external access to SSL object to check
  1442. SSL_get_peer_certificate() == NULL TEST(SSLClientServerTest,
  1443. ClientCAPathRequired) { SSLServer svr(SERVER_CERT_FILE, SERVER_PRIVATE_KEY_FILE,
  1444. nullptr, CLIENT_CA_CERT_DIR);
  1445. }
  1446. */
  1447. #endif
  1448. #ifdef _WIN32
  1449. TEST(CleanupTest, WSACleanup) {
  1450. int ret = WSACleanup();
  1451. ASSERT_EQ(0, ret);
  1452. }
  1453. #endif