test.cc 51 KB

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