eden-emu/cpp-httplib
14
0
Fork 0
mirror of https://github.com/yhirose/cpp-httplib.git synced 2025-05-15 09:18:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix #425

Browse source
This commit is contained in:
yhirose 2020-04-12 15:33:08 -04:00
parent ed8efea98b
commit 85327e19ae
2 changed files with 63 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

33
httplib.h
View file

@ -2524,6 +2524,17 @@ inline bool expect_content(const Request &req) {
return false;
}
inline bool has_crlf(const char* s) {
auto p = s;
while (*p) {
if (*p == '\r' || *p == '\n') {
return true;
}
p++;
}
return false;
}
#ifdef CPPHTTPLIB_OPENSSL_SUPPORT
template <typename CTX, typename Init, typename Update, typename Final>
inline std::string message_digest(const std::string &s, Init init,
@ -2710,11 +2721,15 @@ inline size_t Request::get_header_value_count(const char *key) const {
}
inline void Request::set_header(const char *key, const char *val) {
headers.emplace(key, val);
if (!detail::has_crlf(key) && !detail::has_crlf(val)) {
headers.emplace(key, val);
}
}
inline void Request::set_header(const char *key, const std::string &val) {
headers.emplace(key, val);
if (!detail::has_crlf(key) && !detail::has_crlf(val.c_str())) {
headers.emplace(key, val);
}
}
inline bool Request::has_param(const char *key) const {
@ -2764,16 +2779,22 @@ inline size_t Response::get_header_value_count(const char *key) const {
}
inline void Response::set_header(const char *key, const char *val) {
headers.emplace(key, val);
if (!detail::has_crlf(key) && !detail::has_crlf(val)) {
headers.emplace(key, val);
}
}
inline void Response::set_header(const char *key, const std::string &val) {
headers.emplace(key, val);
if (!detail::has_crlf(key) && !detail::has_crlf(val.c_str())) {
headers.emplace(key, val);
}
}
inline void Response::set_redirect(const char *url) {
set_header("Location", url);
status = 302;
if (!detail::has_crlf(url)) {
set_header("Location", url);
status = 302;
}
}
inline void Response::set_content(const char *s, size_t n,