eden-emu/cpp-httplib
15
0
Fork 0
mirror of https://github.com/yhirose/cpp-httplib.git synced 2025-05-21 04:08:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Support Content-Encoding: gzip on server side

Browse source 
If the client specifies Content-Encoding: gzip for POST requests,
decompress the body before attempting to parse it.
This commit is contained in:
Scott Graham 2018-04-16 11:01:17 -07:00
parent 0e239a0014
commit 5579d4d101
2 changed files with 117 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

48
httplib.h
View file

@ -701,6 +701,7 @@ inline const char* status_message(int status)
case 200: return "OK";
case 400: return "Bad Request";
case 404: return "Not Found";
case 406: return "Not Acceptable";
default:
case 500: return "Internal Server Error";
}
@ -1187,6 +1188,43 @@ inline void compress(const Request& req, Response& res)
deflateEnd(&strm);
}
inline void decompress_request_body(Request& req)
{
if (req.get_header_value("Content-Encoding") != "gzip")
return;
z_stream strm;
strm.zalloc = Z_NULL;
strm.zfree = Z_NULL;
strm.opaque = Z_NULL;
// 15 is the value of wbits, which should be at the maximum possible value to ensure
// that any gzip stream can be decoded. The offset of 16 specifies that the stream
// to decompress will be formatted with a gzip wrapper.
auto ret = inflateInit2(&strm, 16 + 15);
if (ret != Z_OK) {
return;
}
strm.avail_in = req.body.size();
strm.next_in = (Bytef *)req.body.data();
std::string decompressed;
const auto bufsiz = 16384;
char buff[bufsiz];
do {
strm.avail_out = bufsiz;
strm.next_out = (Bytef *)buff;
inflate(&strm, Z_NO_FLUSH);
decompressed.append(buff, bufsiz - strm.avail_out);
} while (strm.avail_out == 0);
req.body.swap(decompressed);
inflateEnd(&strm);
}
#endif
#ifdef _WIN32
@ -1629,6 +1667,16 @@ inline bool Server::process_request(Stream& strm, bool last_connection)
const auto& content_type = req.get_header_value("Content-Type");
#ifdef CPPHTTPLIB_ZLIB_SUPPORT
detail::decompress_request_body(req);
#else
if (req.get_header_value("Content-Encoding") == "gzip") {
res.status = 406;
write_response(strm, last_connection, req, res);
return ret;
}
#endif
if (!content_type.find("application/x-www-form-urlencoded")) {
detail::parse_query_text(req.body, req.params);
} else if(!content_type.find("multipart/form-data")) {