lib/cpp-httplib
1
0
Fork 0
mirror of synced 2025-06-10 20:21:41 +03:00
Code Activity

BUGFIX: crash when "content-length" > max_int

Browse Source
This commit is contained in:
Manny
2019-04-12 12:16:32 +02:00
committed by GitHub
parent 5d082f1da4
commit 07ed076499
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
httplib.h
View File

@ -788,6 +788,15 @@ inline int get_header_value_int(const Headers &headers, const char *key,
return def; return def;
} }
inline unsigned long long get_header_value_uint64(const Headers &headers, const char *key,
int def = 0) {
auto it = headers.find(key);
if (it != headers.end()) {
return std::strtoull(it->second.data(), nullptr, 10);
}
return def;
}
inline bool read_headers(Stream &strm, Headers &headers) { inline bool read_headers(Stream &strm, Headers &headers) {
static std::regex re(R"((.+?):\s*(.+?)\s*\r\n)"); static std::regex re(R"((.+?):\s*(.+?)\s*\r\n)");
@ -881,7 +890,7 @@ inline bool read_content_chunked(Stream &strm, std::string &out) {
template <typename T> template <typename T>
bool read_content(Stream &strm, T &x, Progress progress = Progress()) { bool read_content(Stream &strm, T &x, Progress progress = Progress()) {
if (has_header(x.headers, "Content-Length")) { if (has_header(x.headers, "Content-Length")) {
auto len = get_header_value_int(x.headers, "Content-Length", 0); auto len = get_header_value_uint64(x.headers, "Content-Length", 0);
if (len == 0) { if (len == 0) {
const auto &encoding = const auto &encoding =
get_header_value(x.headers, "Transfer-Encoding", 0, ""); get_header_value(x.headers, "Transfer-Encoding", 0, "");