ARCHIVE/xahaud
3
0
Fork 0
mirror of https://github.com/Xahau/xahaud.git synced 2025-11-19 18:15:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make the HTTP response size const

Browse Source
This commit is contained in:
Joseph Busch
2019-10-29 10:43:49 -05:00
committed by Manoj doshi
parent e26dd7bdfe
commit 7be7343e05
1 changed files with 22 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
src/ripple/net/impl/HTTPClient.cpp
View File

@@ -52,13 +52,13 @@ class HTTPClientImp
public:
HTTPClientImp (boost::asio::io_service& io_service,
const unsigned short port,
std::size_t responseSize,
std::size_t maxResponseSize,
beast::Journal& j)
: mSocket (io_service, httpClientSSLContext->context ())
, mResolver (io_service)
, mHeader (maxClientHeaderBytes)
, mPort (port)
, mResponseSize (responseSize)
, maxResponseSize_ (maxResponseSize)
, mDeadline (io_service)
, j_ (j)
{
@@ -386,16 +386,29 @@ public:
if (boost::regex_match (strHeader, smMatch, reBody)) // we got some body
mBody = smMatch[1];
if (boost::regex_match (strHeader, smMatch, reSize))
mResponseSize =
beast::lexicalCastThrow <int> (std::string(smMatch[1]));
std::size_t const responseSize = [&] {
if (boost::regex_match(strHeader, smMatch, reSize))
return beast::lexicalCast <std::size_t>(
std::string(smMatch[1]), maxResponseSize_);
return maxResponseSize_;
}();
if (mResponseSize == 0)
if (responseSize > maxResponseSize_)
{
JLOG (j_.trace()) << "Response field too large";
invokeComplete (
boost::system::error_code {
boost::system::errc::value_too_large,
boost::system::system_category ()});
return;
}
if (responseSize == 0)
{
// no body wanted or available
invokeComplete (ecResult, mStatus);
}
else if (mBody.size () >= mResponseSize)
else if (mBody.size () >= responseSize)
{
// we got the whole thing
invokeComplete (ecResult, mStatus, mBody);
@@ -403,7 +416,7 @@ public:
else
{
mSocket.async_read (
mResponse.prepare (mResponseSize - mBody.size ()),
mResponse.prepare (responseSize - mBody.size ()),
boost::asio::transfer_all (),
std::bind (&HTTPClientImp::handleData,
shared_from_this (),
@@ -495,7 +508,7 @@ private:
boost::asio::streambuf mResponse;
std::string mBody;
const unsigned short mPort;
int mResponseSize;
std::size_t const maxResponseSize_;
int mStatus;
std::function<void (boost::asio::streambuf& sb, std::string const& strHost)>
mBuild;