httplib.h 62 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904
  1. //
  2. // httplib.h
  3. //
  4. // Copyright (c) 2022 Yuji Hirose. All rights reserved.
  5. // MIT License
  6. //
  7. #ifndef CPPHTTPLIB_HTTPLIB_H
  8. #define CPPHTTPLIB_HTTPLIB_H
  9. #define CPPHTTPLIB_VERSION "0.11.1"
  10. /*
  11. * Configuration
  12. */
  13. #ifndef CPPHTTPLIB_KEEPALIVE_TIMEOUT_SECOND
  14. #define CPPHTTPLIB_KEEPALIVE_TIMEOUT_SECOND 5
  15. #endif
  16. #ifndef CPPHTTPLIB_KEEPALIVE_MAX_COUNT
  17. #define CPPHTTPLIB_KEEPALIVE_MAX_COUNT 5
  18. #endif
  19. #ifndef CPPHTTPLIB_CONNECTION_TIMEOUT_SECOND
  20. #define CPPHTTPLIB_CONNECTION_TIMEOUT_SECOND 300
  21. #endif
  22. #ifndef CPPHTTPLIB_CONNECTION_TIMEOUT_USECOND
  23. #define CPPHTTPLIB_CONNECTION_TIMEOUT_USECOND 0
  24. #endif
  25. #ifndef CPPHTTPLIB_READ_TIMEOUT_SECOND
  26. #define CPPHTTPLIB_READ_TIMEOUT_SECOND 5
  27. #endif
  28. #ifndef CPPHTTPLIB_READ_TIMEOUT_USECOND
  29. #define CPPHTTPLIB_READ_TIMEOUT_USECOND 0
  30. #endif
  31. #ifndef CPPHTTPLIB_WRITE_TIMEOUT_SECOND
  32. #define CPPHTTPLIB_WRITE_TIMEOUT_SECOND 5
  33. #endif
  34. #ifndef CPPHTTPLIB_WRITE_TIMEOUT_USECOND
  35. #define CPPHTTPLIB_WRITE_TIMEOUT_USECOND 0
  36. #endif
  37. #ifndef CPPHTTPLIB_IDLE_INTERVAL_SECOND
  38. #define CPPHTTPLIB_IDLE_INTERVAL_SECOND 0
  39. #endif
  40. #ifndef CPPHTTPLIB_IDLE_INTERVAL_USECOND
  41. #ifdef _WIN32
  42. #define CPPHTTPLIB_IDLE_INTERVAL_USECOND 10000
  43. #else
  44. #define CPPHTTPLIB_IDLE_INTERVAL_USECOND 0
  45. #endif
  46. #endif
  47. #ifndef CPPHTTPLIB_REQUEST_URI_MAX_LENGTH
  48. #define CPPHTTPLIB_REQUEST_URI_MAX_LENGTH 8192
  49. #endif
  50. #ifndef CPPHTTPLIB_HEADER_MAX_LENGTH
  51. #define CPPHTTPLIB_HEADER_MAX_LENGTH 8192
  52. #endif
  53. #ifndef CPPHTTPLIB_REDIRECT_MAX_COUNT
  54. #define CPPHTTPLIB_REDIRECT_MAX_COUNT 20
  55. #endif
  56. #ifndef CPPHTTPLIB_MULTIPART_FORM_DATA_FILE_MAX_COUNT
  57. #define CPPHTTPLIB_MULTIPART_FORM_DATA_FILE_MAX_COUNT 1024
  58. #endif
  59. #ifndef CPPHTTPLIB_PAYLOAD_MAX_LENGTH
  60. #define CPPHTTPLIB_PAYLOAD_MAX_LENGTH ((std::numeric_limits<size_t>::max)())
  61. #endif
  62. #ifndef CPPHTTPLIB_FORM_URL_ENCODED_PAYLOAD_MAX_LENGTH
  63. #define CPPHTTPLIB_FORM_URL_ENCODED_PAYLOAD_MAX_LENGTH 8192
  64. #endif
  65. #ifndef CPPHTTPLIB_TCP_NODELAY
  66. #define CPPHTTPLIB_TCP_NODELAY false
  67. #endif
  68. #ifndef CPPHTTPLIB_RECV_BUFSIZ
  69. #define CPPHTTPLIB_RECV_BUFSIZ size_t(4096u)
  70. #endif
  71. #ifndef CPPHTTPLIB_COMPRESSION_BUFSIZ
  72. #define CPPHTTPLIB_COMPRESSION_BUFSIZ size_t(16384u)
  73. #endif
  74. #ifndef CPPHTTPLIB_THREAD_POOL_COUNT
  75. #define CPPHTTPLIB_THREAD_POOL_COUNT \
  76. ((std::max)(8u, std::thread::hardware_concurrency() > 0 \
  77. ? std::thread::hardware_concurrency() - 1 \
  78. : 0))
  79. #endif
  80. #ifndef CPPHTTPLIB_RECV_FLAGS
  81. #define CPPHTTPLIB_RECV_FLAGS 0
  82. #endif
  83. #ifndef CPPHTTPLIB_SEND_FLAGS
  84. #define CPPHTTPLIB_SEND_FLAGS 0
  85. #endif
  86. #ifndef CPPHTTPLIB_LISTEN_BACKLOG
  87. #define CPPHTTPLIB_LISTEN_BACKLOG 5
  88. #endif
  89. /*
  90. * Headers
  91. */
  92. #ifdef _WIN32
  93. #ifndef _CRT_SECURE_NO_WARNINGS
  94. #define _CRT_SECURE_NO_WARNINGS
  95. #endif //_CRT_SECURE_NO_WARNINGS
  96. #ifndef _CRT_NONSTDC_NO_DEPRECATE
  97. #define _CRT_NONSTDC_NO_DEPRECATE
  98. #endif //_CRT_NONSTDC_NO_DEPRECATE
  99. #if defined(_MSC_VER)
  100. #if _MSC_VER < 1900
  101. #error Sorry, Visual Studio versions prior to 2015 are not supported
  102. #endif
  103. #pragma comment(lib, "ws2_32.lib")
  104. #ifdef _WIN64
  105. using ssize_t = __int64;
  106. #else
  107. using ssize_t = long;
  108. #endif
  109. #endif // _MSC_VER
  110. #ifndef S_ISREG
  111. #define S_ISREG(m) (((m)&S_IFREG) == S_IFREG)
  112. #endif // S_ISREG
  113. #ifndef S_ISDIR
  114. #define S_ISDIR(m) (((m)&S_IFDIR) == S_IFDIR)
  115. #endif // S_ISDIR
  116. #ifndef NOMINMAX
  117. #define NOMINMAX
  118. #endif // NOMINMAX
  119. #include <io.h>
  120. #include <winsock2.h>
  121. #include <ws2tcpip.h>
  122. #ifndef WSA_FLAG_NO_HANDLE_INHERIT
  123. #define WSA_FLAG_NO_HANDLE_INHERIT 0x80
  124. #endif
  125. #ifndef strcasecmp
  126. #define strcasecmp _stricmp
  127. #endif // strcasecmp
  128. using socket_t = SOCKET;
  129. #ifdef CPPHTTPLIB_USE_POLL
  130. #define poll(fds, nfds, timeout) WSAPoll(fds, nfds, timeout)
  131. #endif
  132. #else // not _WIN32
  133. #include <arpa/inet.h>
  134. #include <ifaddrs.h>
  135. #include <net/if.h>
  136. #include <netdb.h>
  137. #include <netinet/in.h>
  138. #ifdef __linux__
  139. #include <resolv.h>
  140. #endif
  141. #include <netinet/tcp.h>
  142. #ifdef CPPHTTPLIB_USE_POLL
  143. #include <poll.h>
  144. #endif
  145. #include <csignal>
  146. #include <pthread.h>
  147. #include <sys/select.h>
  148. #include <sys/socket.h>
  149. #include <sys/un.h>
  150. #include <unistd.h>
  151. using socket_t = int;
  152. #ifndef INVALID_SOCKET
  153. #define INVALID_SOCKET (-1)
  154. #endif
  155. #endif //_WIN32
  156. #include <algorithm>
  157. #include <array>
  158. #include <atomic>
  159. #include <cassert>
  160. #include <cctype>
  161. #include <climits>
  162. #include <condition_variable>
  163. #include <cstring>
  164. #include <errno.h>
  165. #include <fcntl.h>
  166. #include <fstream>
  167. #include <functional>
  168. #include <iomanip>
  169. #include <iostream>
  170. #include <list>
  171. #include <map>
  172. #include <memory>
  173. #include <mutex>
  174. #include <random>
  175. #include <regex>
  176. #include <set>
  177. #include <sstream>
  178. #include <string>
  179. #include <sys/stat.h>
  180. #include <thread>
  181. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  182. #ifdef _WIN32
  183. #include <wincrypt.h>
  184. // these are defined in wincrypt.h and it breaks compilation if BoringSSL is
  185. // used
  186. #undef X509_NAME
  187. #undef X509_CERT_PAIR
  188. #undef X509_EXTENSIONS
  189. #undef PKCS7_SIGNER_INFO
  190. #ifdef _MSC_VER
  191. #pragma comment(lib, "crypt32.lib")
  192. #pragma comment(lib, "cryptui.lib")
  193. #endif
  194. #endif //_WIN32
  195. #include <openssl/err.h>
  196. #include <openssl/evp.h>
  197. #include <openssl/ssl.h>
  198. #include <openssl/x509v3.h>
  199. #if defined(_WIN32) && defined(OPENSSL_USE_APPLINK)
  200. #include <openssl/applink.c>
  201. #endif
  202. #include <iostream>
  203. #include <sstream>
  204. #if OPENSSL_VERSION_NUMBER < 0x1010100fL
  205. #error Sorry, OpenSSL versions prior to 1.1.1 are not supported
  206. #endif
  207. #if OPENSSL_VERSION_NUMBER < 0x10100000L
  208. #include <openssl/crypto.h>
  209. inline const unsigned char *ASN1_STRING_get0_data(const ASN1_STRING *asn1) {
  210. return M_ASN1_STRING_data(asn1);
  211. }
  212. #endif
  213. #endif
  214. #ifdef CPPHTTPLIB_ZLIB_SUPPORT
  215. #include <zlib.h>
  216. #endif
  217. #ifdef CPPHTTPLIB_BROTLI_SUPPORT
  218. #include <brotli/decode.h>
  219. #include <brotli/encode.h>
  220. #endif
  221. /*
  222. * Declaration
  223. */
  224. namespace httplib {
  225. namespace detail {
  226. /*
  227. * Backport std::make_unique from C++14.
  228. *
  229. * NOTE: This code came up with the following stackoverflow post:
  230. * https://stackoverflow.com/questions/10149840/c-arrays-and-make-unique
  231. *
  232. */
  233. template <class T, class... Args>
  234. typename std::enable_if<!std::is_array<T>::value, std::unique_ptr<T>>::type
  235. make_unique(Args &&...args) {
  236. return std::unique_ptr<T>(new T(std::forward<Args>(args)...));
  237. }
  238. template <class T>
  239. typename std::enable_if<std::is_array<T>::value, std::unique_ptr<T>>::type
  240. make_unique(std::size_t n) {
  241. typedef typename std::remove_extent<T>::type RT;
  242. return std::unique_ptr<T>(new RT[n]);
  243. }
  244. struct ci {
  245. bool operator()(const std::string &s1, const std::string &s2) const {
  246. return std::lexicographical_compare(s1.begin(), s1.end(), s2.begin(),
  247. s2.end(),
  248. [](unsigned char c1, unsigned char c2) {
  249. return ::tolower(c1) < ::tolower(c2);
  250. });
  251. }
  252. };
  253. } // namespace detail
  254. using Headers = std::multimap<std::string, std::string, detail::ci>;
  255. using Params = std::multimap<std::string, std::string>;
  256. using Match = std::smatch;
  257. using Progress = std::function<bool(uint64_t current, uint64_t total)>;
  258. struct Response;
  259. using ResponseHandler = std::function<bool(const Response &response)>;
  260. struct MultipartFormData {
  261. std::string name;
  262. std::string content;
  263. std::string filename;
  264. std::string content_type;
  265. };
  266. using MultipartFormDataItems = std::vector<MultipartFormData>;
  267. using MultipartFormDataMap = std::multimap<std::string, MultipartFormData>;
  268. class DataSink {
  269. public:
  270. DataSink() : os(&sb_), sb_(*this) {}
  271. DataSink(const DataSink &) = delete;
  272. DataSink &operator=(const DataSink &) = delete;
  273. DataSink(DataSink &&) = delete;
  274. DataSink &operator=(DataSink &&) = delete;
  275. std::function<bool(const char *data, size_t data_len)> write;
  276. std::function<void()> done;
  277. std::function<bool()> is_writable;
  278. std::ostream os;
  279. private:
  280. class data_sink_streambuf : public std::streambuf {
  281. public:
  282. explicit data_sink_streambuf(DataSink &sink) : sink_(sink) {}
  283. protected:
  284. std::streamsize xsputn(const char *s, std::streamsize n) {
  285. sink_.write(s, static_cast<size_t>(n));
  286. return n;
  287. }
  288. private:
  289. DataSink &sink_;
  290. };
  291. data_sink_streambuf sb_;
  292. };
  293. using ContentProvider =
  294. std::function<bool(size_t offset, size_t length, DataSink &sink)>;
  295. using ContentProviderWithoutLength =
  296. std::function<bool(size_t offset, DataSink &sink)>;
  297. using ContentProviderResourceReleaser = std::function<void(bool success)>;
  298. using ContentReceiverWithProgress =
  299. std::function<bool(const char *data, size_t data_length, uint64_t offset,
  300. uint64_t total_length)>;
  301. using ContentReceiver =
  302. std::function<bool(const char *data, size_t data_length)>;
  303. using MultipartContentHeader =
  304. std::function<bool(const MultipartFormData &file)>;
  305. class ContentReader {
  306. public:
  307. using Reader = std::function<bool(ContentReceiver receiver)>;
  308. using MultipartReader = std::function<bool(MultipartContentHeader header,
  309. ContentReceiver receiver)>;
  310. ContentReader(Reader reader, MultipartReader multipart_reader)
  311. : reader_(std::move(reader)),
  312. multipart_reader_(std::move(multipart_reader)) {}
  313. bool operator()(MultipartContentHeader header,
  314. ContentReceiver receiver) const {
  315. return multipart_reader_(std::move(header), std::move(receiver));
  316. }
  317. bool operator()(ContentReceiver receiver) const {
  318. return reader_(std::move(receiver));
  319. }
  320. Reader reader_;
  321. MultipartReader multipart_reader_;
  322. };
  323. using Range = std::pair<ssize_t, ssize_t>;
  324. using Ranges = std::vector<Range>;
  325. struct Request {
  326. std::string method;
  327. std::string path;
  328. Headers headers;
  329. std::string body;
  330. std::string remote_addr;
  331. int remote_port = -1;
  332. // for server
  333. std::string version;
  334. std::string target;
  335. Params params;
  336. MultipartFormDataMap files;
  337. Ranges ranges;
  338. Match matches;
  339. // for client
  340. ResponseHandler response_handler;
  341. ContentReceiverWithProgress content_receiver;
  342. Progress progress;
  343. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  344. const SSL *ssl = nullptr;
  345. #endif
  346. bool has_header(const std::string &key) const;
  347. std::string get_header_value(const std::string &key, size_t id = 0) const;
  348. template <typename T>
  349. T get_header_value(const std::string &key, size_t id = 0) const;
  350. size_t get_header_value_count(const std::string &key) const;
  351. void set_header(const std::string &key, const std::string &val);
  352. bool has_param(const std::string &key) const;
  353. std::string get_param_value(const std::string &key, size_t id = 0) const;
  354. size_t get_param_value_count(const std::string &key) const;
  355. bool is_multipart_form_data() const;
  356. bool has_file(const std::string &key) const;
  357. MultipartFormData get_file_value(const std::string &key) const;
  358. // private members...
  359. size_t redirect_count_ = CPPHTTPLIB_REDIRECT_MAX_COUNT;
  360. size_t content_length_ = 0;
  361. ContentProvider content_provider_;
  362. bool is_chunked_content_provider_ = false;
  363. size_t authorization_count_ = 0;
  364. };
  365. struct Response {
  366. std::string version;
  367. int status = -1;
  368. std::string reason;
  369. Headers headers;
  370. std::string body;
  371. std::string location; // Redirect location
  372. bool has_header(const std::string &key) const;
  373. std::string get_header_value(const std::string &key, size_t id = 0) const;
  374. template <typename T>
  375. T get_header_value(const std::string &key, size_t id = 0) const;
  376. size_t get_header_value_count(const std::string &key) const;
  377. void set_header(const std::string &key, const std::string &val);
  378. void set_redirect(const std::string &url, int status = 302);
  379. void set_content(const char *s, size_t n, const std::string &content_type);
  380. void set_content(const std::string &s, const std::string &content_type);
  381. void set_content_provider(
  382. size_t length, const std::string &content_type, ContentProvider provider,
  383. ContentProviderResourceReleaser resource_releaser = nullptr);
  384. void set_content_provider(
  385. const std::string &content_type, ContentProviderWithoutLength provider,
  386. ContentProviderResourceReleaser resource_releaser = nullptr);
  387. void set_chunked_content_provider(
  388. const std::string &content_type, ContentProviderWithoutLength provider,
  389. ContentProviderResourceReleaser resource_releaser = nullptr);
  390. Response() = default;
  391. Response(const Response &) = default;
  392. Response &operator=(const Response &) = default;
  393. Response(Response &&) = default;
  394. Response &operator=(Response &&) = default;
  395. ~Response() {
  396. if (content_provider_resource_releaser_) {
  397. content_provider_resource_releaser_(content_provider_success_);
  398. }
  399. }
  400. // private members...
  401. size_t content_length_ = 0;
  402. ContentProvider content_provider_;
  403. ContentProviderResourceReleaser content_provider_resource_releaser_;
  404. bool is_chunked_content_provider_ = false;
  405. bool content_provider_success_ = false;
  406. };
  407. class Stream {
  408. public:
  409. virtual ~Stream() = default;
  410. virtual bool is_readable() const = 0;
  411. virtual bool is_writable() const = 0;
  412. virtual ssize_t read(char *ptr, size_t size) = 0;
  413. virtual ssize_t write(const char *ptr, size_t size) = 0;
  414. virtual void get_remote_ip_and_port(std::string &ip, int &port) const = 0;
  415. virtual socket_t socket() const = 0;
  416. template <typename... Args>
  417. ssize_t write_format(const char *fmt, const Args &...args);
  418. ssize_t write(const char *ptr);
  419. ssize_t write(const std::string &s);
  420. };
  421. class TaskQueue {
  422. public:
  423. TaskQueue() = default;
  424. virtual ~TaskQueue() = default;
  425. virtual void enqueue(std::function<void()> fn) = 0;
  426. virtual void shutdown() = 0;
  427. virtual void on_idle() {}
  428. };
  429. class ThreadPool : public TaskQueue {
  430. public:
  431. explicit ThreadPool(size_t n) : shutdown_(false) {
  432. while (n) {
  433. threads_.emplace_back(worker(*this));
  434. n--;
  435. }
  436. }
  437. ThreadPool(const ThreadPool &) = delete;
  438. ~ThreadPool() override = default;
  439. void enqueue(std::function<void()> fn) override {
  440. std::unique_lock<std::mutex> lock(mutex_);
  441. jobs_.push_back(std::move(fn));
  442. cond_.notify_one();
  443. }
  444. void shutdown() override {
  445. // Stop all worker threads...
  446. {
  447. std::unique_lock<std::mutex> lock(mutex_);
  448. shutdown_ = true;
  449. }
  450. cond_.notify_all();
  451. // Join...
  452. for (auto &t : threads_) {
  453. t.join();
  454. }
  455. }
  456. private:
  457. struct worker {
  458. explicit worker(ThreadPool &pool) : pool_(pool) {}
  459. void operator()() {
  460. for (;;) {
  461. std::function<void()> fn;
  462. {
  463. std::unique_lock<std::mutex> lock(pool_.mutex_);
  464. pool_.cond_.wait(
  465. lock, [&] { return !pool_.jobs_.empty() || pool_.shutdown_; });
  466. if (pool_.shutdown_ && pool_.jobs_.empty()) { break; }
  467. fn = pool_.jobs_.front();
  468. pool_.jobs_.pop_front();
  469. }
  470. assert(true == static_cast<bool>(fn));
  471. fn();
  472. }
  473. }
  474. ThreadPool &pool_;
  475. };
  476. friend struct worker;
  477. std::vector<std::thread> threads_;
  478. std::list<std::function<void()>> jobs_;
  479. bool shutdown_;
  480. std::condition_variable cond_;
  481. std::mutex mutex_;
  482. };
  483. using Logger = std::function<void(const Request &, const Response &)>;
  484. using SocketOptions = std::function<void(socket_t sock)>;
  485. void default_socket_options(socket_t sock);
  486. class Server {
  487. public:
  488. using Handler = std::function<void(const Request &, Response &)>;
  489. using ExceptionHandler =
  490. std::function<void(const Request &, Response &, std::exception_ptr ep)>;
  491. enum class HandlerResponse {
  492. Handled,
  493. Unhandled,
  494. };
  495. using HandlerWithResponse =
  496. std::function<HandlerResponse(const Request &, Response &)>;
  497. using HandlerWithContentReader = std::function<void(
  498. const Request &, Response &, const ContentReader &content_reader)>;
  499. using Expect100ContinueHandler =
  500. std::function<int(const Request &, Response &)>;
  501. Server();
  502. virtual ~Server();
  503. virtual bool is_valid() const;
  504. Server &Get(const std::string &pattern, Handler handler);
  505. Server &Post(const std::string &pattern, Handler handler);
  506. Server &Post(const std::string &pattern, HandlerWithContentReader handler);
  507. Server &Put(const std::string &pattern, Handler handler);
  508. Server &Put(const std::string &pattern, HandlerWithContentReader handler);
  509. Server &Patch(const std::string &pattern, Handler handler);
  510. Server &Patch(const std::string &pattern, HandlerWithContentReader handler);
  511. Server &Delete(const std::string &pattern, Handler handler);
  512. Server &Delete(const std::string &pattern, HandlerWithContentReader handler);
  513. Server &Options(const std::string &pattern, Handler handler);
  514. bool set_base_dir(const std::string &dir,
  515. const std::string &mount_point = std::string());
  516. bool set_mount_point(const std::string &mount_point, const std::string &dir,
  517. Headers headers = Headers());
  518. bool remove_mount_point(const std::string &mount_point);
  519. Server &set_file_extension_and_mimetype_mapping(const std::string &ext,
  520. const std::string &mime);
  521. Server &set_file_request_handler(Handler handler);
  522. Server &set_error_handler(HandlerWithResponse handler);
  523. Server &set_error_handler(Handler handler);
  524. Server &set_exception_handler(ExceptionHandler handler);
  525. Server &set_pre_routing_handler(HandlerWithResponse handler);
  526. Server &set_post_routing_handler(Handler handler);
  527. Server &set_expect_100_continue_handler(Expect100ContinueHandler handler);
  528. Server &set_logger(Logger logger);
  529. Server &set_address_family(int family);
  530. Server &set_tcp_nodelay(bool on);
  531. Server &set_socket_options(SocketOptions socket_options);
  532. Server &set_default_headers(Headers headers);
  533. Server &set_keep_alive_max_count(size_t count);
  534. Server &set_keep_alive_timeout(time_t sec);
  535. Server &set_read_timeout(time_t sec, time_t usec = 0);
  536. template <class Rep, class Period>
  537. Server &set_read_timeout(const std::chrono::duration<Rep, Period> &duration);
  538. Server &set_write_timeout(time_t sec, time_t usec = 0);
  539. template <class Rep, class Period>
  540. Server &set_write_timeout(const std::chrono::duration<Rep, Period> &duration);
  541. Server &set_idle_interval(time_t sec, time_t usec = 0);
  542. template <class Rep, class Period>
  543. Server &set_idle_interval(const std::chrono::duration<Rep, Period> &duration);
  544. Server &set_payload_max_length(size_t length);
  545. bool bind_to_port(const std::string &host, int port, int socket_flags = 0);
  546. int bind_to_any_port(const std::string &host, int socket_flags = 0);
  547. bool listen_after_bind();
  548. bool listen(const std::string &host, int port, int socket_flags = 0);
  549. bool is_running() const;
  550. void stop();
  551. std::function<TaskQueue *(void)> new_task_queue;
  552. protected:
  553. bool process_request(Stream &strm, bool close_connection,
  554. bool &connection_closed,
  555. const std::function<void(Request &)> &setup_request);
  556. std::atomic<socket_t> svr_sock_;
  557. size_t keep_alive_max_count_ = CPPHTTPLIB_KEEPALIVE_MAX_COUNT;
  558. time_t keep_alive_timeout_sec_ = CPPHTTPLIB_KEEPALIVE_TIMEOUT_SECOND;
  559. time_t read_timeout_sec_ = CPPHTTPLIB_READ_TIMEOUT_SECOND;
  560. time_t read_timeout_usec_ = CPPHTTPLIB_READ_TIMEOUT_USECOND;
  561. time_t write_timeout_sec_ = CPPHTTPLIB_WRITE_TIMEOUT_SECOND;
  562. time_t write_timeout_usec_ = CPPHTTPLIB_WRITE_TIMEOUT_USECOND;
  563. time_t idle_interval_sec_ = CPPHTTPLIB_IDLE_INTERVAL_SECOND;
  564. time_t idle_interval_usec_ = CPPHTTPLIB_IDLE_INTERVAL_USECOND;
  565. size_t payload_max_length_ = CPPHTTPLIB_PAYLOAD_MAX_LENGTH;
  566. private:
  567. using Handlers = std::vector<std::pair<std::regex, Handler>>;
  568. using HandlersForContentReader =
  569. std::vector<std::pair<std::regex, HandlerWithContentReader>>;
  570. socket_t create_server_socket(const std::string &host, int port,
  571. int socket_flags,
  572. SocketOptions socket_options) const;
  573. int bind_internal(const std::string &host, int port, int socket_flags);
  574. bool listen_internal();
  575. bool routing(Request &req, Response &res, Stream &strm);
  576. bool handle_file_request(const Request &req, Response &res,
  577. bool head = false);
  578. bool dispatch_request(Request &req, Response &res, const Handlers &handlers);
  579. bool
  580. dispatch_request_for_content_reader(Request &req, Response &res,
  581. ContentReader content_reader,
  582. const HandlersForContentReader &handlers);
  583. bool parse_request_line(const char *s, Request &req);
  584. void apply_ranges(const Request &req, Response &res,
  585. std::string &content_type, std::string &boundary);
  586. bool write_response(Stream &strm, bool close_connection, const Request &req,
  587. Response &res);
  588. bool write_response_with_content(Stream &strm, bool close_connection,
  589. const Request &req, Response &res);
  590. bool write_response_core(Stream &strm, bool close_connection,
  591. const Request &req, Response &res,
  592. bool need_apply_ranges);
  593. bool write_content_with_provider(Stream &strm, const Request &req,
  594. Response &res, const std::string &boundary,
  595. const std::string &content_type);
  596. bool read_content(Stream &strm, Request &req, Response &res);
  597. bool
  598. read_content_with_content_receiver(Stream &strm, Request &req, Response &res,
  599. ContentReceiver receiver,
  600. MultipartContentHeader multipart_header,
  601. ContentReceiver multipart_receiver);
  602. bool read_content_core(Stream &strm, Request &req, Response &res,
  603. ContentReceiver receiver,
  604. MultipartContentHeader mulitpart_header,
  605. ContentReceiver multipart_receiver);
  606. virtual bool process_and_close_socket(socket_t sock);
  607. struct MountPointEntry {
  608. std::string mount_point;
  609. std::string base_dir;
  610. Headers headers;
  611. };
  612. std::vector<MountPointEntry> base_dirs_;
  613. std::atomic<bool> is_running_;
  614. std::map<std::string, std::string> file_extension_and_mimetype_map_;
  615. Handler file_request_handler_;
  616. Handlers get_handlers_;
  617. Handlers post_handlers_;
  618. HandlersForContentReader post_handlers_for_content_reader_;
  619. Handlers put_handlers_;
  620. HandlersForContentReader put_handlers_for_content_reader_;
  621. Handlers patch_handlers_;
  622. HandlersForContentReader patch_handlers_for_content_reader_;
  623. Handlers delete_handlers_;
  624. HandlersForContentReader delete_handlers_for_content_reader_;
  625. Handlers options_handlers_;
  626. HandlerWithResponse error_handler_;
  627. ExceptionHandler exception_handler_;
  628. HandlerWithResponse pre_routing_handler_;
  629. Handler post_routing_handler_;
  630. Logger logger_;
  631. Expect100ContinueHandler expect_100_continue_handler_;
  632. int address_family_ = AF_UNSPEC;
  633. bool tcp_nodelay_ = CPPHTTPLIB_TCP_NODELAY;
  634. SocketOptions socket_options_ = default_socket_options;
  635. Headers default_headers_;
  636. };
  637. enum class Error {
  638. Success = 0,
  639. Unknown,
  640. Connection,
  641. BindIPAddress,
  642. Read,
  643. Write,
  644. ExceedRedirectCount,
  645. Canceled,
  646. SSLConnection,
  647. SSLLoadingCerts,
  648. SSLServerVerification,
  649. UnsupportedMultipartBoundaryChars,
  650. Compression,
  651. ConnectionTimeout,
  652. };
  653. std::string to_string(const Error error);
  654. std::ostream &operator<<(std::ostream &os, const Error &obj);
  655. class Result {
  656. public:
  657. Result(std::unique_ptr<Response> &&res, Error err,
  658. Headers &&request_headers = Headers{})
  659. : res_(std::move(res)), err_(err),
  660. request_headers_(std::move(request_headers)) {}
  661. // Response
  662. operator bool() const { return res_ != nullptr; }
  663. bool operator==(std::nullptr_t) const { return res_ == nullptr; }
  664. bool operator!=(std::nullptr_t) const { return res_ != nullptr; }
  665. const Response &value() const { return *res_; }
  666. Response &value() { return *res_; }
  667. const Response &operator*() const { return *res_; }
  668. Response &operator*() { return *res_; }
  669. const Response *operator->() const { return res_.get(); }
  670. Response *operator->() { return res_.get(); }
  671. // Error
  672. Error error() const { return err_; }
  673. // Request Headers
  674. bool has_request_header(const std::string &key) const;
  675. std::string get_request_header_value(const std::string &key,
  676. size_t id = 0) const;
  677. template <typename T>
  678. T get_request_header_value(const std::string &key, size_t id = 0) const;
  679. size_t get_request_header_value_count(const std::string &key) const;
  680. private:
  681. std::unique_ptr<Response> res_;
  682. Error err_;
  683. Headers request_headers_;
  684. };
  685. class ClientImpl {
  686. public:
  687. explicit ClientImpl(const std::string &host);
  688. explicit ClientImpl(const std::string &host, int port);
  689. explicit ClientImpl(const std::string &host, int port,
  690. const std::string &client_cert_path,
  691. const std::string &client_key_path);
  692. virtual ~ClientImpl();
  693. virtual bool is_valid() const;
  694. Result Get(const std::string &path);
  695. Result Get(const std::string &path, const Headers &headers);
  696. Result Get(const std::string &path, Progress progress);
  697. Result Get(const std::string &path, const Headers &headers,
  698. Progress progress);
  699. Result Get(const std::string &path, ContentReceiver content_receiver);
  700. Result Get(const std::string &path, const Headers &headers,
  701. ContentReceiver content_receiver);
  702. Result Get(const std::string &path, ContentReceiver content_receiver,
  703. Progress progress);
  704. Result Get(const std::string &path, const Headers &headers,
  705. ContentReceiver content_receiver, Progress progress);
  706. Result Get(const std::string &path, ResponseHandler response_handler,
  707. ContentReceiver content_receiver);
  708. Result Get(const std::string &path, const Headers &headers,
  709. ResponseHandler response_handler,
  710. ContentReceiver content_receiver);
  711. Result Get(const std::string &path, ResponseHandler response_handler,
  712. ContentReceiver content_receiver, Progress progress);
  713. Result Get(const std::string &path, const Headers &headers,
  714. ResponseHandler response_handler, ContentReceiver content_receiver,
  715. Progress progress);
  716. Result Get(const std::string &path, const Params &params,
  717. const Headers &headers, Progress progress = nullptr);
  718. Result Get(const std::string &path, const Params &params,
  719. const Headers &headers, ContentReceiver content_receiver,
  720. Progress progress = nullptr);
  721. Result Get(const std::string &path, const Params &params,
  722. const Headers &headers, ResponseHandler response_handler,
  723. ContentReceiver content_receiver, Progress progress = nullptr);
  724. Result Head(const std::string &path);
  725. Result Head(const std::string &path, const Headers &headers);
  726. Result Post(const std::string &path);
  727. Result Post(const std::string &path, const char *body, size_t content_length,
  728. const std::string &content_type);
  729. Result Post(const std::string &path, const Headers &headers, const char *body,
  730. size_t content_length, const std::string &content_type);
  731. Result Post(const std::string &path, const std::string &body,
  732. const std::string &content_type);
  733. Result Post(const std::string &path, const Headers &headers,
  734. const std::string &body, const std::string &content_type);
  735. Result Post(const std::string &path, size_t content_length,
  736. ContentProvider content_provider,
  737. const std::string &content_type);
  738. Result Post(const std::string &path,
  739. ContentProviderWithoutLength content_provider,
  740. const std::string &content_type);
  741. Result Post(const std::string &path, const Headers &headers,
  742. size_t content_length, ContentProvider content_provider,
  743. const std::string &content_type);
  744. Result Post(const std::string &path, const Headers &headers,
  745. ContentProviderWithoutLength content_provider,
  746. const std::string &content_type);
  747. Result Post(const std::string &path, const Params &params);
  748. Result Post(const std::string &path, const Headers &headers,
  749. const Params &params);
  750. Result Post(const std::string &path, const MultipartFormDataItems &items);
  751. Result Post(const std::string &path, const Headers &headers,
  752. const MultipartFormDataItems &items);
  753. Result Post(const std::string &path, const Headers &headers,
  754. const MultipartFormDataItems &items, const std::string &boundary);
  755. Result Put(const std::string &path);
  756. Result Put(const std::string &path, const char *body, size_t content_length,
  757. const std::string &content_type);
  758. Result Put(const std::string &path, const Headers &headers, const char *body,
  759. size_t content_length, const std::string &content_type);
  760. Result Put(const std::string &path, const std::string &body,
  761. const std::string &content_type);
  762. Result Put(const std::string &path, const Headers &headers,
  763. const std::string &body, const std::string &content_type);
  764. Result Put(const std::string &path, size_t content_length,
  765. ContentProvider content_provider, const std::string &content_type);
  766. Result Put(const std::string &path,
  767. ContentProviderWithoutLength content_provider,
  768. const std::string &content_type);
  769. Result Put(const std::string &path, const Headers &headers,
  770. size_t content_length, ContentProvider content_provider,
  771. const std::string &content_type);
  772. Result Put(const std::string &path, const Headers &headers,
  773. ContentProviderWithoutLength content_provider,
  774. const std::string &content_type);
  775. Result Put(const std::string &path, const Params &params);
  776. Result Put(const std::string &path, const Headers &headers,
  777. const Params &params);
  778. Result Put(const std::string &path, const MultipartFormDataItems &items);
  779. Result Put(const std::string &path, const Headers &headers,
  780. const MultipartFormDataItems &items);
  781. Result Put(const std::string &path, const Headers &headers,
  782. const MultipartFormDataItems &items, const std::string &boundary);
  783. Result Patch(const std::string &path);
  784. Result Patch(const std::string &path, const char *body, size_t content_length,
  785. const std::string &content_type);
  786. Result Patch(const std::string &path, const Headers &headers,
  787. const char *body, size_t content_length,
  788. const std::string &content_type);
  789. Result Patch(const std::string &path, const std::string &body,
  790. const std::string &content_type);
  791. Result Patch(const std::string &path, const Headers &headers,
  792. const std::string &body, const std::string &content_type);
  793. Result Patch(const std::string &path, size_t content_length,
  794. ContentProvider content_provider,
  795. const std::string &content_type);
  796. Result Patch(const std::string &path,
  797. ContentProviderWithoutLength content_provider,
  798. const std::string &content_type);
  799. Result Patch(const std::string &path, const Headers &headers,
  800. size_t content_length, ContentProvider content_provider,
  801. const std::string &content_type);
  802. Result Patch(const std::string &path, const Headers &headers,
  803. ContentProviderWithoutLength content_provider,
  804. const std::string &content_type);
  805. Result Delete(const std::string &path);
  806. Result Delete(const std::string &path, const Headers &headers);
  807. Result Delete(const std::string &path, const char *body,
  808. size_t content_length, const std::string &content_type);
  809. Result Delete(const std::string &path, const Headers &headers,
  810. const char *body, size_t content_length,
  811. const std::string &content_type);
  812. Result Delete(const std::string &path, const std::string &body,
  813. const std::string &content_type);
  814. Result Delete(const std::string &path, const Headers &headers,
  815. const std::string &body, const std::string &content_type);
  816. Result Options(const std::string &path);
  817. Result Options(const std::string &path, const Headers &headers);
  818. bool send(Request &req, Response &res, Error &error);
  819. Result send(const Request &req);
  820. size_t is_socket_open() const;
  821. socket_t socket() const;
  822. void stop();
  823. void set_hostname_addr_map(std::map<std::string, std::string> addr_map);
  824. void set_default_headers(Headers headers);
  825. void set_address_family(int family);
  826. void set_tcp_nodelay(bool on);
  827. void set_socket_options(SocketOptions socket_options);
  828. void set_connection_timeout(time_t sec, time_t usec = 0);
  829. template <class Rep, class Period>
  830. void
  831. set_connection_timeout(const std::chrono::duration<Rep, Period> &duration);
  832. void set_read_timeout(time_t sec, time_t usec = 0);
  833. template <class Rep, class Period>
  834. void set_read_timeout(const std::chrono::duration<Rep, Period> &duration);
  835. void set_write_timeout(time_t sec, time_t usec = 0);
  836. template <class Rep, class Period>
  837. void set_write_timeout(const std::chrono::duration<Rep, Period> &duration);
  838. void set_basic_auth(const std::string &username, const std::string &password);
  839. void set_bearer_token_auth(const std::string &token);
  840. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  841. void set_digest_auth(const std::string &username,
  842. const std::string &password);
  843. #endif
  844. void set_keep_alive(bool on);
  845. void set_follow_location(bool on);
  846. void set_url_encode(bool on);
  847. void set_compress(bool on);
  848. void set_decompress(bool on);
  849. void set_interface(const std::string &intf);
  850. void set_proxy(const std::string &host, int port);
  851. void set_proxy_basic_auth(const std::string &username,
  852. const std::string &password);
  853. void set_proxy_bearer_token_auth(const std::string &token);
  854. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  855. void set_proxy_digest_auth(const std::string &username,
  856. const std::string &password);
  857. #endif
  858. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  859. void set_ca_cert_path(const std::string &ca_cert_file_path,
  860. const std::string &ca_cert_dir_path = std::string());
  861. void set_ca_cert_store(X509_STORE *ca_cert_store);
  862. #endif
  863. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  864. void enable_server_certificate_verification(bool enabled);
  865. #endif
  866. void set_logger(Logger logger);
  867. protected:
  868. struct Socket {
  869. socket_t sock = INVALID_SOCKET;
  870. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  871. SSL *ssl = nullptr;
  872. #endif
  873. bool is_open() const { return sock != INVALID_SOCKET; }
  874. };
  875. Result send_(Request &&req);
  876. virtual bool create_and_connect_socket(Socket &socket, Error &error);
  877. // All of:
  878. // shutdown_ssl
  879. // shutdown_socket
  880. // close_socket
  881. // should ONLY be called when socket_mutex_ is locked.
  882. // Also, shutdown_ssl and close_socket should also NOT be called concurrently
  883. // with a DIFFERENT thread sending requests using that socket.
  884. virtual void shutdown_ssl(Socket &socket, bool shutdown_gracefully);
  885. void shutdown_socket(Socket &socket);
  886. void close_socket(Socket &socket);
  887. bool process_request(Stream &strm, Request &req, Response &res,
  888. bool close_connection, Error &error);
  889. bool write_content_with_provider(Stream &strm, const Request &req,
  890. Error &error);
  891. void copy_settings(const ClientImpl &rhs);
  892. // Socket endoint information
  893. const std::string host_;
  894. const int port_;
  895. const std::string host_and_port_;
  896. // Current open socket
  897. Socket socket_;
  898. mutable std::mutex socket_mutex_;
  899. std::recursive_mutex request_mutex_;
  900. // These are all protected under socket_mutex
  901. size_t socket_requests_in_flight_ = 0;
  902. std::thread::id socket_requests_are_from_thread_ = std::thread::id();
  903. bool socket_should_be_closed_when_request_is_done_ = false;
  904. // Hostname-IP map
  905. std::map<std::string, std::string> addr_map_;
  906. // Default headers
  907. Headers default_headers_;
  908. // Settings
  909. std::string client_cert_path_;
  910. std::string client_key_path_;
  911. time_t connection_timeout_sec_ = CPPHTTPLIB_CONNECTION_TIMEOUT_SECOND;
  912. time_t connection_timeout_usec_ = CPPHTTPLIB_CONNECTION_TIMEOUT_USECOND;
  913. time_t read_timeout_sec_ = CPPHTTPLIB_READ_TIMEOUT_SECOND;
  914. time_t read_timeout_usec_ = CPPHTTPLIB_READ_TIMEOUT_USECOND;
  915. time_t write_timeout_sec_ = CPPHTTPLIB_WRITE_TIMEOUT_SECOND;
  916. time_t write_timeout_usec_ = CPPHTTPLIB_WRITE_TIMEOUT_USECOND;
  917. std::string basic_auth_username_;
  918. std::string basic_auth_password_;
  919. std::string bearer_token_auth_token_;
  920. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  921. std::string digest_auth_username_;
  922. std::string digest_auth_password_;
  923. #endif
  924. bool keep_alive_ = false;
  925. bool follow_location_ = false;
  926. bool url_encode_ = true;
  927. int address_family_ = AF_UNSPEC;
  928. bool tcp_nodelay_ = CPPHTTPLIB_TCP_NODELAY;
  929. SocketOptions socket_options_ = nullptr;
  930. bool compress_ = false;
  931. bool decompress_ = true;
  932. std::string interface_;
  933. std::string proxy_host_;
  934. int proxy_port_ = -1;
  935. std::string proxy_basic_auth_username_;
  936. std::string proxy_basic_auth_password_;
  937. std::string proxy_bearer_token_auth_token_;
  938. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  939. std::string proxy_digest_auth_username_;
  940. std::string proxy_digest_auth_password_;
  941. #endif
  942. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  943. std::string ca_cert_file_path_;
  944. std::string ca_cert_dir_path_;
  945. X509_STORE *ca_cert_store_ = nullptr;
  946. #endif
  947. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  948. bool server_certificate_verification_ = true;
  949. #endif
  950. Logger logger_;
  951. private:
  952. socket_t create_client_socket(Error &error) const;
  953. bool read_response_line(Stream &strm, const Request &req, Response &res);
  954. bool write_request(Stream &strm, Request &req, bool close_connection,
  955. Error &error);
  956. bool redirect(Request &req, Response &res, Error &error);
  957. bool handle_request(Stream &strm, Request &req, Response &res,
  958. bool close_connection, Error &error);
  959. std::unique_ptr<Response> send_with_content_provider(
  960. Request &req, const char *body, size_t content_length,
  961. ContentProvider content_provider,
  962. ContentProviderWithoutLength content_provider_without_length,
  963. const std::string &content_type, Error &error);
  964. Result send_with_content_provider(
  965. const std::string &method, const std::string &path,
  966. const Headers &headers, const char *body, size_t content_length,
  967. ContentProvider content_provider,
  968. ContentProviderWithoutLength content_provider_without_length,
  969. const std::string &content_type);
  970. std::string adjust_host_string(const std::string &host) const;
  971. virtual bool process_socket(const Socket &socket,
  972. std::function<bool(Stream &strm)> callback);
  973. virtual bool is_ssl() const;
  974. };
  975. class Client {
  976. public:
  977. // Universal interface
  978. explicit Client(const std::string &scheme_host_port);
  979. explicit Client(const std::string &scheme_host_port,
  980. const std::string &client_cert_path,
  981. const std::string &client_key_path);
  982. // HTTP only interface
  983. explicit Client(const std::string &host, int port);
  984. explicit Client(const std::string &host, int port,
  985. const std::string &client_cert_path,
  986. const std::string &client_key_path);
  987. Client(Client &&) = default;
  988. ~Client();
  989. bool is_valid() const;
  990. Result Get(const std::string &path);
  991. Result Get(const std::string &path, const Headers &headers);
  992. Result Get(const std::string &path, Progress progress);
  993. Result Get(const std::string &path, const Headers &headers,
  994. Progress progress);
  995. Result Get(const std::string &path, ContentReceiver content_receiver);
  996. Result Get(const std::string &path, const Headers &headers,
  997. ContentReceiver content_receiver);
  998. Result Get(const std::string &path, ContentReceiver content_receiver,
  999. Progress progress);
  1000. Result Get(const std::string &path, const Headers &headers,
  1001. ContentReceiver content_receiver, Progress progress);
  1002. Result Get(const std::string &path, ResponseHandler response_handler,
  1003. ContentReceiver content_receiver);
  1004. Result Get(const std::string &path, const Headers &headers,
  1005. ResponseHandler response_handler,
  1006. ContentReceiver content_receiver);
  1007. Result Get(const std::string &path, const Headers &headers,
  1008. ResponseHandler response_handler, ContentReceiver content_receiver,
  1009. Progress progress);
  1010. Result Get(const std::string &path, ResponseHandler response_handler,
  1011. ContentReceiver content_receiver, Progress progress);
  1012. Result Get(const std::string &path, const Params &params,
  1013. const Headers &headers, Progress progress = nullptr);
  1014. Result Get(const std::string &path, const Params &params,
  1015. const Headers &headers, ContentReceiver content_receiver,
  1016. Progress progress = nullptr);
  1017. Result Get(const std::string &path, const Params &params,
  1018. const Headers &headers, ResponseHandler response_handler,
  1019. ContentReceiver content_receiver, Progress progress = nullptr);
  1020. Result Head(const std::string &path);
  1021. Result Head(const std::string &path, const Headers &headers);
  1022. Result Post(const std::string &path);
  1023. Result Post(const std::string &path, const char *body, size_t content_length,
  1024. const std::string &content_type);
  1025. Result Post(const std::string &path, const Headers &headers, const char *body,
  1026. size_t content_length, const std::string &content_type);
  1027. Result Post(const std::string &path, const std::string &body,
  1028. const std::string &content_type);
  1029. Result Post(const std::string &path, const Headers &headers,
  1030. const std::string &body, const std::string &content_type);
  1031. Result Post(const std::string &path, size_t content_length,
  1032. ContentProvider content_provider,
  1033. const std::string &content_type);
  1034. Result Post(const std::string &path,
  1035. ContentProviderWithoutLength content_provider,
  1036. const std::string &content_type);
  1037. Result Post(const std::string &path, const Headers &headers,
  1038. size_t content_length, ContentProvider content_provider,
  1039. const std::string &content_type);
  1040. Result Post(const std::string &path, const Headers &headers,
  1041. ContentProviderWithoutLength content_provider,
  1042. const std::string &content_type);
  1043. Result Post(const std::string &path, const Params &params);
  1044. Result Post(const std::string &path, const Headers &headers,
  1045. const Params &params);
  1046. Result Post(const std::string &path, const MultipartFormDataItems &items);
  1047. Result Post(const std::string &path, const Headers &headers,
  1048. const MultipartFormDataItems &items);
  1049. Result Post(const std::string &path, const Headers &headers,
  1050. const MultipartFormDataItems &items, const std::string &boundary);
  1051. Result Put(const std::string &path);
  1052. Result Put(const std::string &path, const char *body, size_t content_length,
  1053. const std::string &content_type);
  1054. Result Put(const std::string &path, const Headers &headers, const char *body,
  1055. size_t content_length, const std::string &content_type);
  1056. Result Put(const std::string &path, const std::string &body,
  1057. const std::string &content_type);
  1058. Result Put(const std::string &path, const Headers &headers,
  1059. const std::string &body, const std::string &content_type);
  1060. Result Put(const std::string &path, size_t content_length,
  1061. ContentProvider content_provider, const std::string &content_type);
  1062. Result Put(const std::string &path,
  1063. ContentProviderWithoutLength content_provider,
  1064. const std::string &content_type);
  1065. Result Put(const std::string &path, const Headers &headers,
  1066. size_t content_length, ContentProvider content_provider,
  1067. const std::string &content_type);
  1068. Result Put(const std::string &path, const Headers &headers,
  1069. ContentProviderWithoutLength content_provider,
  1070. const std::string &content_type);
  1071. Result Put(const std::string &path, const Params &params);
  1072. Result Put(const std::string &path, const Headers &headers,
  1073. const Params &params);
  1074. Result Put(const std::string &path, const MultipartFormDataItems &items);
  1075. Result Put(const std::string &path, const Headers &headers,
  1076. const MultipartFormDataItems &items);
  1077. Result Put(const std::string &path, const Headers &headers,
  1078. const MultipartFormDataItems &items, const std::string &boundary);
  1079. Result Patch(const std::string &path);
  1080. Result Patch(const std::string &path, const char *body, size_t content_length,
  1081. const std::string &content_type);
  1082. Result Patch(const std::string &path, const Headers &headers,
  1083. const char *body, size_t content_length,
  1084. const std::string &content_type);
  1085. Result Patch(const std::string &path, const std::string &body,
  1086. const std::string &content_type);
  1087. Result Patch(const std::string &path, const Headers &headers,
  1088. const std::string &body, const std::string &content_type);
  1089. Result Patch(const std::string &path, size_t content_length,
  1090. ContentProvider content_provider,
  1091. const std::string &content_type);
  1092. Result Patch(const std::string &path,
  1093. ContentProviderWithoutLength content_provider,
  1094. const std::string &content_type);
  1095. Result Patch(const std::string &path, const Headers &headers,
  1096. size_t content_length, ContentProvider content_provider,
  1097. const std::string &content_type);
  1098. Result Patch(const std::string &path, const Headers &headers,
  1099. ContentProviderWithoutLength content_provider,
  1100. const std::string &content_type);
  1101. Result Delete(const std::string &path);
  1102. Result Delete(const std::string &path, const Headers &headers);
  1103. Result Delete(const std::string &path, const char *body,
  1104. size_t content_length, const std::string &content_type);
  1105. Result Delete(const std::string &path, const Headers &headers,
  1106. const char *body, size_t content_length,
  1107. const std::string &content_type);
  1108. Result Delete(const std::string &path, const std::string &body,
  1109. const std::string &content_type);
  1110. Result Delete(const std::string &path, const Headers &headers,
  1111. const std::string &body, const std::string &content_type);
  1112. Result Options(const std::string &path);
  1113. Result Options(const std::string &path, const Headers &headers);
  1114. bool send(Request &req, Response &res, Error &error);
  1115. Result send(const Request &req);
  1116. size_t is_socket_open() const;
  1117. socket_t socket() const;
  1118. void stop();
  1119. void set_hostname_addr_map(std::map<std::string, std::string> addr_map);
  1120. void set_default_headers(Headers headers);
  1121. void set_address_family(int family);
  1122. void set_tcp_nodelay(bool on);
  1123. void set_socket_options(SocketOptions socket_options);
  1124. void set_connection_timeout(time_t sec, time_t usec = 0);
  1125. template <class Rep, class Period>
  1126. void
  1127. set_connection_timeout(const std::chrono::duration<Rep, Period> &duration);
  1128. void set_read_timeout(time_t sec, time_t usec = 0);
  1129. template <class Rep, class Period>
  1130. void set_read_timeout(const std::chrono::duration<Rep, Period> &duration);
  1131. void set_write_timeout(time_t sec, time_t usec = 0);
  1132. template <class Rep, class Period>
  1133. void set_write_timeout(const std::chrono::duration<Rep, Period> &duration);
  1134. void set_basic_auth(const std::string &username, const std::string &password);
  1135. void set_bearer_token_auth(const std::string &token);
  1136. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  1137. void set_digest_auth(const std::string &username,
  1138. const std::string &password);
  1139. #endif
  1140. void set_keep_alive(bool on);
  1141. void set_follow_location(bool on);
  1142. void set_url_encode(bool on);
  1143. void set_compress(bool on);
  1144. void set_decompress(bool on);
  1145. void set_interface(const std::string &intf);
  1146. void set_proxy(const std::string &host, int port);
  1147. void set_proxy_basic_auth(const std::string &username,
  1148. const std::string &password);
  1149. void set_proxy_bearer_token_auth(const std::string &token);
  1150. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  1151. void set_proxy_digest_auth(const std::string &username,
  1152. const std::string &password);
  1153. #endif
  1154. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  1155. void enable_server_certificate_verification(bool enabled);
  1156. #endif
  1157. void set_logger(Logger logger);
  1158. // SSL
  1159. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  1160. void set_ca_cert_path(const std::string &ca_cert_file_path,
  1161. const std::string &ca_cert_dir_path = std::string());
  1162. void set_ca_cert_store(X509_STORE *ca_cert_store);
  1163. long get_openssl_verify_result() const;
  1164. SSL_CTX *ssl_context() const;
  1165. #endif
  1166. private:
  1167. std::unique_ptr<ClientImpl> cli_;
  1168. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  1169. bool is_ssl_ = false;
  1170. #endif
  1171. };
  1172. #ifdef CPPHTTPLIB_OPENSSL_SUPPORT
  1173. class SSLServer : public Server {
  1174. public:
  1175. SSLServer(const char *cert_path, const char *private_key_path,
  1176. const char *client_ca_cert_file_path = nullptr,
  1177. const char *client_ca_cert_dir_path = nullptr,
  1178. const char *private_key_password = nullptr);
  1179. SSLServer(X509 *cert, EVP_PKEY *private_key,
  1180. X509_STORE *client_ca_cert_store = nullptr);
  1181. SSLServer(
  1182. const std::function<bool(SSL_CTX &ssl_ctx)> &setup_ssl_ctx_callback);
  1183. ~SSLServer() override;
  1184. bool is_valid() const override;
  1185. SSL_CTX *ssl_context() const;
  1186. private:
  1187. bool process_and_close_socket(socket_t sock) override;
  1188. SSL_CTX *ctx_;
  1189. std::mutex ctx_mutex_;
  1190. };
  1191. class SSLClient : public ClientImpl {
  1192. public:
  1193. explicit SSLClient(const std::string &host);
  1194. explicit SSLClient(const std::string &host, int port);
  1195. explicit SSLClient(const std::string &host, int port,
  1196. const std::string &client_cert_path,
  1197. const std::string &client_key_path);
  1198. explicit SSLClient(const std::string &host, int port, X509 *client_cert,
  1199. EVP_PKEY *client_key);
  1200. ~SSLClient() override;
  1201. bool is_valid() const override;
  1202. void set_ca_cert_store(X509_STORE *ca_cert_store);
  1203. long get_openssl_verify_result() const;
  1204. SSL_CTX *ssl_context() const;
  1205. private:
  1206. bool create_and_connect_socket(Socket &socket, Error &error) override;
  1207. void shutdown_ssl(Socket &socket, bool shutdown_gracefully) override;
  1208. void shutdown_ssl_impl(Socket &socket, bool shutdown_socket);
  1209. bool process_socket(const Socket &socket,
  1210. std::function<bool(Stream &strm)> callback) override;
  1211. bool is_ssl() const override;
  1212. bool connect_with_proxy(Socket &sock, Response &res, bool &success,
  1213. Error &error);
  1214. bool initialize_ssl(Socket &socket, Error &error);
  1215. bool load_certs();
  1216. bool verify_host(X509 *server_cert) const;
  1217. bool verify_host_with_subject_alt_name(X509 *server_cert) const;
  1218. bool verify_host_with_common_name(X509 *server_cert) const;
  1219. bool check_host_name(const char *pattern, size_t pattern_len) const;
  1220. SSL_CTX *ctx_;
  1221. std::mutex ctx_mutex_;
  1222. std::once_flag initialize_cert_;
  1223. std::vector<std::string> host_components_;
  1224. long verify_result_ = 0;
  1225. friend class ClientImpl;
  1226. };
  1227. #endif
  1228. /*
  1229. * Implementation of template methods.
  1230. */
  1231. namespace detail {
  1232. template <typename T, typename U>
  1233. inline void duration_to_sec_and_usec(const T &duration, U callback) {
  1234. auto sec = std::chrono::duration_cast<std::chrono::seconds>(duration).count();
  1235. auto usec = std::chrono::duration_cast<std::chrono::microseconds>(
  1236. duration - std::chrono::seconds(sec))
  1237. .count();
  1238. callback(sec, usec);
  1239. }
  1240. template <typename T>
  1241. inline T get_header_value(const Headers & /*headers*/,
  1242. const std::string & /*key*/, size_t /*id*/ = 0,
  1243. uint64_t /*def*/ = 0) {}
  1244. template <>
  1245. inline uint64_t get_header_value<uint64_t>(const Headers &headers,
  1246. const std::string &key, size_t id,
  1247. uint64_t def) {
  1248. auto rng = headers.equal_range(key);
  1249. auto it = rng.first;
  1250. std::advance(it, static_cast<ssize_t>(id));
  1251. if (it != rng.second) {
  1252. return std::strtoull(it->second.data(), nullptr, 10);
  1253. }
  1254. return def;
  1255. }
  1256. } // namespace detail
  1257. template <typename T>
  1258. inline T Request::get_header_value(const std::string &key, size_t id) const {
  1259. return detail::get_header_value<T>(headers, key, id, 0);
  1260. }
  1261. template <typename T>
  1262. inline T Response::get_header_value(const std::string &key, size_t id) const {
  1263. return detail::get_header_value<T>(headers, key, id, 0);
  1264. }
  1265. template <typename... Args>
  1266. inline ssize_t Stream::write_format(const char *fmt, const Args &...args) {
  1267. const auto bufsiz = 2048;
  1268. std::array<char, bufsiz> buf{};
  1269. auto sn = snprintf(buf.data(), buf.size() - 1, fmt, args...);
  1270. if (sn <= 0) { return sn; }
  1271. auto n = static_cast<size_t>(sn);
  1272. if (n >= buf.size() - 1) {
  1273. std::vector<char> glowable_buf(buf.size());
  1274. while (n >= glowable_buf.size() - 1) {
  1275. glowable_buf.resize(glowable_buf.size() * 2);
  1276. n = static_cast<size_t>(
  1277. snprintf(&glowable_buf[0], glowable_buf.size() - 1, fmt, args...));
  1278. }
  1279. return write(&glowable_buf[0], n);
  1280. } else {
  1281. return write(buf.data(), n);
  1282. }
  1283. }
  1284. inline void default_socket_options(socket_t sock) {
  1285. int yes = 1;
  1286. #ifdef _WIN32
  1287. setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, reinterpret_cast<char *>(&yes),
  1288. sizeof(yes));
  1289. setsockopt(sock, SOL_SOCKET, SO_EXCLUSIVEADDRUSE,
  1290. reinterpret_cast<char *>(&yes), sizeof(yes));
  1291. #else
  1292. #ifdef SO_REUSEPORT
  1293. setsockopt(sock, SOL_SOCKET, SO_REUSEPORT, reinterpret_cast<void *>(&yes),
  1294. sizeof(yes));
  1295. #else
  1296. setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, reinterpret_cast<void *>(&yes),
  1297. sizeof(yes));
  1298. #endif
  1299. #endif
  1300. }
  1301. template <class Rep, class Period>
  1302. inline Server &
  1303. Server::set_read_timeout(const std::chrono::duration<Rep, Period> &duration) {
  1304. detail::duration_to_sec_and_usec(
  1305. duration, [&](time_t sec, time_t usec) { set_read_timeout(sec, usec); });
  1306. return *this;
  1307. }
  1308. template <class Rep, class Period>
  1309. inline Server &
  1310. Server::set_write_timeout(const std::chrono::duration<Rep, Period> &duration) {
  1311. detail::duration_to_sec_and_usec(
  1312. duration, [&](time_t sec, time_t usec) { set_write_timeout(sec, usec); });
  1313. return *this;
  1314. }
  1315. template <class Rep, class Period>
  1316. inline Server &
  1317. Server::set_idle_interval(const std::chrono::duration<Rep, Period> &duration) {
  1318. detail::duration_to_sec_and_usec(
  1319. duration, [&](time_t sec, time_t usec) { set_idle_interval(sec, usec); });
  1320. return *this;
  1321. }
  1322. inline std::string to_string(const Error error) {
  1323. switch (error) {
  1324. case Error::Success: return "Success";
  1325. case Error::Connection: return "Connection";
  1326. case Error::BindIPAddress: return "BindIPAddress";
  1327. case Error::Read: return "Read";
  1328. case Error::Write: return "Write";
  1329. case Error::ExceedRedirectCount: return "ExceedRedirectCount";
  1330. case Error::Canceled: return "Canceled";
  1331. case Error::SSLConnection: return "SSLConnection";
  1332. case Error::SSLLoadingCerts: return "SSLLoadingCerts";
  1333. case Error::SSLServerVerification: return "SSLServerVerification";
  1334. case Error::UnsupportedMultipartBoundaryChars:
  1335. return "UnsupportedMultipartBoundaryChars";
  1336. case Error::Compression: return "Compression";
  1337. case Error::ConnectionTimeout: return "ConnectionTimeout";
  1338. case Error::Unknown: return "Unknown";
  1339. default: break;
  1340. }
  1341. return "Invalid";
  1342. }
  1343. inline std::ostream &operator<<(std::ostream &os, const Error &obj) {
  1344. os << to_string(obj);
  1345. os << " (" << static_cast<std::underlying_type<Error>::type>(obj) << ')';
  1346. return os;
  1347. }
  1348. template <typename T>
  1349. inline T Result::get_request_header_value(const std::string &key,
  1350. size_t id) const {
  1351. return detail::get_header_value<T>(request_headers_, key, id, 0);
  1352. }
  1353. template <class Rep, class Period>
  1354. inline void ClientImpl::set_connection_timeout(
  1355. const std::chrono::duration<Rep, Period> &duration) {
  1356. detail::duration_to_sec_and_usec(duration, [&](time_t sec, time_t usec) {
  1357. set_connection_timeout(sec, usec);
  1358. });
  1359. }
  1360. template <class Rep, class Period>
  1361. inline void ClientImpl::set_read_timeout(
  1362. const std::chrono::duration<Rep, Period> &duration) {
  1363. detail::duration_to_sec_and_usec(
  1364. duration, [&](time_t sec, time_t usec) { set_read_timeout(sec, usec); });
  1365. }
  1366. template <class Rep, class Period>
  1367. inline void ClientImpl::set_write_timeout(
  1368. const std::chrono::duration<Rep, Period> &duration) {
  1369. detail::duration_to_sec_and_usec(
  1370. duration, [&](time_t sec, time_t usec) { set_write_timeout(sec, usec); });
  1371. }
  1372. template <class Rep, class Period>
  1373. inline void Client::set_connection_timeout(
  1374. const std::chrono::duration<Rep, Period> &duration) {
  1375. cli_->set_connection_timeout(duration);
  1376. }
  1377. template <class Rep, class Period>
  1378. inline void
  1379. Client::set_read_timeout(const std::chrono::duration<Rep, Period> &duration) {
  1380. cli_->set_read_timeout(duration);
  1381. }
  1382. template <class Rep, class Period>
  1383. inline void
  1384. Client::set_write_timeout(const std::chrono::duration<Rep, Period> &duration) {
  1385. cli_->set_write_timeout(duration);
  1386. }
  1387. /*
  1388. * Forward declarations and types that will be part of the .h file if split into
  1389. * .h + .cc.
  1390. */
  1391. std::string hosted_at(const std::string &hostname);
  1392. void hosted_at(const std::string &hostname, std::vector<std::string> &addrs);
  1393. std::string append_query_params(const std::string &path, const Params &params);
  1394. std::pair<std::string, std::string> make_range_header(Ranges ranges);
  1395. std::pair<std::string, std::string>
  1396. make_basic_authentication_header(const std::string &username,
  1397. const std::string &password,
  1398. bool is_proxy = false);
  1399. namespace detail {
  1400. std::string encode_query_param(const std::string &value);
  1401. std::string decode_url(const std::string &s, bool convert_plus_to_space);
  1402. void read_file(const std::string &path, std::string &out);
  1403. std::string trim_copy(const std::string &s);
  1404. void split(const char *b, const char *e, char d,
  1405. std::function<void(const char *, const char *)> fn);
  1406. bool process_client_socket(socket_t sock, time_t read_timeout_sec,
  1407. time_t read_timeout_usec, time_t write_timeout_sec,
  1408. time_t write_timeout_usec,
  1409. std::function<bool(Stream &)> callback);
  1410. socket_t create_client_socket(
  1411. const std::string &host, const std::string &ip, int port,
  1412. int address_family, bool tcp_nodelay, SocketOptions socket_options,
  1413. time_t connection_timeout_sec, time_t connection_timeout_usec,
  1414. time_t read_timeout_sec, time_t read_timeout_usec, time_t write_timeout_sec,
  1415. time_t write_timeout_usec, const std::string &intf, Error &error);
  1416. const char *get_header_value(const Headers &headers, const std::string &key,
  1417. size_t id = 0, const char *def = nullptr);
  1418. std::string params_to_query_str(const Params &params);
  1419. void parse_query_text(const std::string &s, Params &params);
  1420. bool parse_range_header(const std::string &s, Ranges &ranges);
  1421. int close_socket(socket_t sock);
  1422. ssize_t send_socket(socket_t sock, const void *ptr, size_t size, int flags);
  1423. ssize_t read_socket(socket_t sock, void *ptr, size_t size, int flags);
  1424. enum class EncodingType { None = 0, Gzip, Brotli };
  1425. EncodingType encoding_type(const Request &req, const Response &res);
  1426. class BufferStream : public Stream {
  1427. public:
  1428. BufferStream() = default;
  1429. ~BufferStream() override = default;
  1430. bool is_readable() const override;
  1431. bool is_writable() const override;
  1432. ssize_t read(char *ptr, size_t size) override;
  1433. ssize_t write(const char *ptr, size_t size) override;
  1434. void get_remote_ip_and_port(std::string &ip, int &port) const override;
  1435. socket_t socket() const override;
  1436. const std::string &get_buffer() const;
  1437. private:
  1438. std::string buffer;
  1439. size_t position = 0;
  1440. };
  1441. class compressor {
  1442. public:
  1443. virtual ~compressor() = default;
  1444. typedef std::function<bool(const char *data, size_t data_len)> Callback;
  1445. virtual bool compress(const char *data, size_t data_length, bool last,
  1446. Callback callback) = 0;
  1447. };
  1448. class decompressor {
  1449. public:
  1450. virtual ~decompressor() = default;
  1451. virtual bool is_valid() const = 0;
  1452. typedef std::function<bool(const char *data, size_t data_len)> Callback;
  1453. virtual bool decompress(const char *data, size_t data_length,
  1454. Callback callback) = 0;
  1455. };
  1456. class nocompressor : public compressor {
  1457. public:
  1458. virtual ~nocompressor() = default;
  1459. bool compress(const char *data, size_t data_length, bool /*last*/,
  1460. Callback callback) override;
  1461. };
  1462. #ifdef CPPHTTPLIB_ZLIB_SUPPORT
  1463. class gzip_compressor : public compressor {
  1464. public:
  1465. gzip_compressor();
  1466. ~gzip_compressor();
  1467. bool compress(const char *data, size_t data_length, bool last,
  1468. Callback callback) override;
  1469. private:
  1470. bool is_valid_ = false;
  1471. z_stream strm_;
  1472. };
  1473. class gzip_decompressor : public decompressor {
  1474. public:
  1475. gzip_decompressor();
  1476. ~gzip_decompressor();
  1477. bool is_valid() const override;
  1478. bool decompress(const char *data, size_t data_length,
  1479. Callback callback) override;
  1480. private:
  1481. bool is_valid_ = false;
  1482. z_stream strm_;
  1483. };
  1484. #endif
  1485. #ifdef CPPHTTPLIB_BROTLI_SUPPORT
  1486. class brotli_compressor : public compressor {
  1487. public:
  1488. brotli_compressor();
  1489. ~brotli_compressor();
  1490. bool compress(const char *data, size_t data_length, bool last,
  1491. Callback callback) override;
  1492. private:
  1493. BrotliEncoderState *state_ = nullptr;
  1494. };
  1495. class brotli_decompressor : public decompressor {
  1496. public:
  1497. brotli_decompressor();
  1498. ~brotli_decompressor();
  1499. bool is_valid() const override;
  1500. bool decompress(const char *data, size_t data_length,
  1501. Callback callback) override;
  1502. private:
  1503. BrotliDecoderResult decoder_r;
  1504. BrotliDecoderState *decoder_s = nullptr;
  1505. };
  1506. #endif
  1507. // NOTE: until the read size reaches `fixed_buffer_size`, use `fixed_buffer`
  1508. // to store data. The call can set memory on stack for performance.
  1509. class stream_line_reader {
  1510. public:
  1511. stream_line_reader(Stream &strm, char *fixed_buffer,
  1512. size_t fixed_buffer_size);
  1513. const char *ptr() const;
  1514. size_t size() const;
  1515. bool end_with_crlf() const;
  1516. bool getline();
  1517. private:
  1518. void append(char c);
  1519. Stream &strm_;
  1520. char *fixed_buffer_;
  1521. const size_t fixed_buffer_size_;
  1522. size_t fixed_buffer_used_size_ = 0;
  1523. std::string glowable_buffer_;
  1524. };
  1525. } // namespace detail
  1526. } // namespace httplib
  1527. #endif // CPPHTTPLIB_HTTPLIB_H