mirror of
https://github.com/XRPLF/rippled.git
synced 2025-11-20 19:15:54 +00:00
c060d08767
New parser set_option function for controlling independent size limits on headers and body. By default request and response parsers are set up with reasonable limits to prevent resource exhaustion attacks. * Parser adheres strictly to rfc7230 * Increased test coverage * Headers and body maximum size limit options
113 lines
2.1 KiB
C++
113 lines
2.1 KiB
C++
//
|
|
// Copyright (c) 2013-2016 Vinnie Falco (vinnie dot falco at gmail dot com)
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
|
|
#ifndef BEAST_HTTP_TEST_FAIL_PARSER_HPP
|
|
#define BEAST_HTTP_TEST_FAIL_PARSER_HPP
|
|
|
|
#include <beast/http/basic_parser_v1.hpp>
|
|
#include <beast/test/fail_counter.hpp>
|
|
|
|
namespace beast {
|
|
namespace http {
|
|
|
|
template<bool isRequest>
|
|
class fail_parser
|
|
: public basic_parser_v1<isRequest, fail_parser<isRequest>>
|
|
{
|
|
test::fail_counter& fc_;
|
|
std::uint64_t content_length_ = no_content_length;
|
|
int body_rv_ = 0;
|
|
|
|
public:
|
|
std::string body;
|
|
|
|
template<class... Args>
|
|
explicit
|
|
fail_parser(test::fail_counter& fc, Args&&... args)
|
|
: fc_(fc)
|
|
{
|
|
}
|
|
|
|
void
|
|
on_body_rv(int rv)
|
|
{
|
|
body_rv_ = rv;
|
|
}
|
|
|
|
// valid on successful parse
|
|
std::uint64_t
|
|
content_length() const
|
|
{
|
|
return content_length_;
|
|
}
|
|
|
|
void on_start(error_code& ec)
|
|
{
|
|
fc_.fail(ec);
|
|
}
|
|
|
|
void on_method(boost::string_ref const&, error_code& ec)
|
|
{
|
|
fc_.fail(ec);
|
|
}
|
|
|
|
void on_uri(boost::string_ref const&, error_code& ec)
|
|
{
|
|
fc_.fail(ec);
|
|
}
|
|
|
|
void on_reason(boost::string_ref const&, error_code& ec)
|
|
{
|
|
fc_.fail(ec);
|
|
}
|
|
|
|
void on_request(error_code& ec)
|
|
{
|
|
fc_.fail(ec);
|
|
}
|
|
|
|
void on_response(error_code& ec)
|
|
{
|
|
fc_.fail(ec);
|
|
}
|
|
|
|
void on_field(boost::string_ref const&, error_code& ec)
|
|
{
|
|
fc_.fail(ec);
|
|
}
|
|
|
|
void on_value(boost::string_ref const&, error_code& ec)
|
|
{
|
|
fc_.fail(ec);
|
|
}
|
|
|
|
int on_headers(std::uint64_t content_length, error_code& ec)
|
|
{
|
|
if(fc_.fail(ec))
|
|
return 0;
|
|
content_length_ = content_length;
|
|
return body_rv_;
|
|
}
|
|
|
|
void on_body(boost::string_ref const& s, error_code& ec)
|
|
{
|
|
if(fc_.fail(ec))
|
|
return;
|
|
body.append(s.data(), s.size());
|
|
}
|
|
|
|
void on_complete(error_code& ec)
|
|
{
|
|
fc_.fail(ec);
|
|
}
|
|
};
|
|
|
|
} // http
|
|
} // beast
|
|
|
|
#endif
|