lib/cpp-httplib
1
0
Fork 0
mirror of synced 2025-04-21 22:25:55 +03:00
Code

Fix #136

Browse Source
This commit is contained in:
yhirose 2019-03-28 19:55:43 -04:00
parent acc5e5a450
commit 8483e5931f
2 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
httplib.h
View File

@ -2119,8 +2119,10 @@ inline void Client::write_request(Stream& strm, Request& req)
req.set_header("Content-Type", "text/plain"); req.set_header("Content-Type", "text/plain");
} }
auto length = std::to_string(req.body.size()); if (!req.has_header("Content-Length")) {
req.set_header("Content-Length", length.c_str()); auto length = std::to_string(req.body.size());
req.set_header("Content-Length", length.c_str());
}
} }
detail::write_headers(bstrm, req); detail::write_headers(bstrm, req);

13
test/test.cc
View File

@ -470,6 +470,10 @@ protected:
EXPECT_EQ("value2", req.get_param_value("array", 1)); EXPECT_EQ("value2", req.get_param_value("array", 1));
EXPECT_EQ("value3", req.get_param_value("array", 2)); EXPECT_EQ("value3", req.get_param_value("array", 2));
}) })
.Post("/validate-no-multiple-headers", [&](const Request& req, Response& res) {
EXPECT_EQ(1u, req.get_header_value_count("Content-Length"));
EXPECT_EQ("5", req.get_header_value("Content-Length"));
})
#ifdef CPPHTTPLIB_ZLIB_SUPPORT #ifdef CPPHTTPLIB_ZLIB_SUPPORT
.Get("/gzip", [&](const Request& /*req*/, Response& res) { .Get("/gzip", [&](const Request& /*req*/, Response& res) {
res.set_content("1234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890", "text/plain"); res.set_content("1234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890", "text/plain");
@ -989,6 +993,15 @@ TEST_F(ServerTest, ArrayParam)
EXPECT_EQ(200, res->status); EXPECT_EQ(200, res->status);
} }
TEST_F(ServerTest, NoMultipleHeaders)
{
Headers headers;
headers.emplace("Content-Length", "5");
auto res = cli_.Post("/validate-no-multiple-headers", headers, "hello", "text/plain");
ASSERT_TRUE(res != nullptr);
EXPECT_EQ(200, res->status);
}
#ifdef CPPHTTPLIB_ZLIB_SUPPORT #ifdef CPPHTTPLIB_ZLIB_SUPPORT
TEST_F(ServerTest, Gzip) TEST_F(ServerTest, Gzip)
{ {