123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563 |
- //
- // Copyright (c) 2016-2019 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)
- //
- // Official repository: https://github.com/boostorg/beast
- //
- #ifndef BOOST_BEAST_HTTP_IMPL_READ_HPP
- #define BOOST_BEAST_HTTP_IMPL_READ_HPP
- #include <boost/beast/http/type_traits.hpp>
- #include <boost/beast/http/error.hpp>
- #include <boost/beast/http/parser.hpp>
- #include <boost/beast/http/read.hpp>
- #include <boost/beast/core/async_base.hpp>
- #include <boost/beast/core/stream_traits.hpp>
- #include <boost/beast/core/detail/read.hpp>
- #include <boost/asio/error.hpp>
- namespace boost {
- namespace beast {
- namespace http {
- namespace detail {
- // The default maximum number of bytes to transfer in a single operation.
- std::size_t constexpr default_max_transfer_size = 65536;
- template<
- class DynamicBuffer,
- bool isRequest,
- class Condition>
- std::size_t
- parse_until(
- DynamicBuffer& buffer,
- basic_parser<isRequest>& parser,
- error_code& ec,
- Condition cond)
- {
- if(ec == net::error::eof)
- {
- if(parser.got_some())
- {
- // Caller sees EOF on next read
- ec = {};
- parser.put_eof(ec);
- BOOST_ASSERT(ec || parser.is_done());
- }
- else
- {
- ec = error::end_of_stream;
- }
- return 0;
- }
- if(ec)
- {
- // Upgrade the error if we have a partial message.
- // This causes SSL short reads (and every other error)
- // to be converted into something else, allowing the
- // caller to distinguish an SSL short read which
- // represents a safe connection closure, versus
- // a closure with data loss.
- if( ec != net::error::operation_aborted &&
- parser.got_some() && ! parser.is_done())
- {
- ec = error::partial_message;
- }
- return 0;
- }
- if(parser.is_done())
- return 0;
- if(buffer.size() > 0)
- {
- auto const bytes_used =
- parser.put(buffer.data(), ec);
- // total = total + bytes_used; // VFALCO Can't do this in a condition
- buffer.consume(bytes_used);
- if(ec == http::error::need_more)
- {
- if(buffer.size() >= buffer.max_size())
- {
- ec = http::error::buffer_overflow;
- return 0;
- }
- ec = {};
- }
- else if(ec || cond())
- {
- return 0;
- }
- }
- return default_max_transfer_size;
- }
- // predicate is true on any forward parser progress
- template<bool isRequest>
- struct read_some_condition
- {
- basic_parser<isRequest>& parser;
- template<class DynamicBuffer>
- std::size_t
- operator()(error_code& ec, std::size_t,
- DynamicBuffer& buffer)
- {
- return detail::parse_until(
- buffer, parser, ec,
- []
- {
- return true;
- });
- }
- };
- // predicate is true when parser header is complete
- template<bool isRequest>
- struct read_header_condition
- {
- basic_parser<isRequest>& parser;
- template<class DynamicBuffer>
- std::size_t
- operator()(error_code& ec, std::size_t,
- DynamicBuffer& buffer)
- {
- return detail::parse_until(
- buffer, parser, ec,
- [this]
- {
- return parser.is_header_done();
- });
- }
- };
- // predicate is true when parser message is complete
- template<bool isRequest>
- struct read_all_condition
- {
- basic_parser<isRequest>& parser;
- template<class DynamicBuffer>
- std::size_t
- operator()(error_code& ec, std::size_t,
- DynamicBuffer& buffer)
- {
- return detail::parse_until(
- buffer, parser, ec,
- [this]
- {
- return parser.is_done();
- });
- }
- };
- //------------------------------------------------------------------------------
- template<
- class Stream, class DynamicBuffer,
- bool isRequest, class Body, class Allocator,
- class Handler>
- class read_msg_op
- : public beast::stable_async_base<
- Handler, beast::executor_type<Stream>>
- , public asio::coroutine
- {
- using parser_type =
- parser<isRequest, Body, Allocator>;
- using message_type =
- typename parser_type::value_type;
- struct data
- {
- Stream& s;
- message_type& m;
- parser_type p;
- data(
- Stream& s_,
- message_type& m_)
- : s(s_)
- , m(m_)
- , p(std::move(m))
- {
- }
- };
- data& d_;
- public:
- template<class Handler_>
- read_msg_op(
- Handler_&& h,
- Stream& s,
- DynamicBuffer& b,
- message_type& m)
- : stable_async_base<
- Handler, beast::executor_type<Stream>>(
- std::forward<Handler_>(h), s.get_executor())
- , d_(beast::allocate_stable<data>(
- *this, s, m))
- {
- http::async_read(d_.s, b, d_.p, std::move(*this));
- }
- void
- operator()(
- error_code ec,
- std::size_t bytes_transferred)
- {
- if(! ec)
- d_.m = d_.p.release();
- this->complete_now(ec, bytes_transferred);
- }
- };
- struct run_read_msg_op
- {
- template<
- class ReadHandler,
- class AsyncReadStream,
- class DynamicBuffer,
- bool isRequest, class Body, class Allocator>
- void
- operator()(
- ReadHandler&& h,
- AsyncReadStream* s,
- DynamicBuffer* b,
- message<isRequest, Body,
- basic_fields<Allocator>>* m)
- {
- // If you get an error on the following line it means
- // that your handler does not meet the documented type
- // requirements for the handler.
- static_assert(
- beast::detail::is_invocable<ReadHandler,
- void(error_code, std::size_t)>::value,
- "ReadHandler type requirements not met");
- read_msg_op<
- AsyncReadStream,
- DynamicBuffer,
- isRequest, Body, Allocator,
- typename std::decay<ReadHandler>::type>(
- std::forward<ReadHandler>(h), *s, *b, *m);
- }
- };
- } // detail
- //------------------------------------------------------------------------------
- template<
- class SyncReadStream,
- class DynamicBuffer,
- bool isRequest>
- std::size_t
- read_some(
- SyncReadStream& stream,
- DynamicBuffer& buffer,
- basic_parser<isRequest>& parser)
- {
- static_assert(
- is_sync_read_stream<SyncReadStream>::value,
- "SyncReadStream type requirements not met");
- static_assert(
- net::is_dynamic_buffer<DynamicBuffer>::value,
- "DynamicBuffer type requirements not met");
- error_code ec;
- auto const bytes_transferred =
- http::read_some(stream, buffer, parser, ec);
- if(ec)
- BOOST_THROW_EXCEPTION(system_error{ec});
- return bytes_transferred;
- }
- template<
- class SyncReadStream,
- class DynamicBuffer,
- bool isRequest>
- std::size_t
- read_some(
- SyncReadStream& stream,
- DynamicBuffer& buffer,
- basic_parser<isRequest>& parser,
- error_code& ec)
- {
- static_assert(
- is_sync_read_stream<SyncReadStream>::value,
- "SyncReadStream type requirements not met");
- static_assert(
- net::is_dynamic_buffer<DynamicBuffer>::value,
- "DynamicBuffer type requirements not met");
- return beast::detail::read(stream, buffer,
- detail::read_some_condition<
- isRequest>{parser}, ec);
- }
- template<
- class AsyncReadStream,
- class DynamicBuffer,
- bool isRequest,
- class ReadHandler>
- BOOST_BEAST_ASYNC_RESULT2(ReadHandler)
- async_read_some(
- AsyncReadStream& stream,
- DynamicBuffer& buffer,
- basic_parser<isRequest>& parser,
- ReadHandler&& handler)
- {
- return beast::detail::async_read(
- stream,
- buffer,
- detail::read_some_condition<
- isRequest>{parser},
- std::forward<ReadHandler>(handler));
- }
- //------------------------------------------------------------------------------
- template<
- class SyncReadStream,
- class DynamicBuffer,
- bool isRequest>
- std::size_t
- read_header(
- SyncReadStream& stream,
- DynamicBuffer& buffer,
- basic_parser<isRequest>& parser)
- {
- static_assert(
- is_sync_read_stream<SyncReadStream>::value,
- "SyncReadStream type requirements not met");
- static_assert(
- net::is_dynamic_buffer<DynamicBuffer>::value,
- "DynamicBuffer type requirements not met");
- error_code ec;
- auto const bytes_transferred =
- http::read_header(stream, buffer, parser, ec);
- if(ec)
- BOOST_THROW_EXCEPTION(system_error{ec});
- return bytes_transferred;
- }
- template<
- class SyncReadStream,
- class DynamicBuffer,
- bool isRequest>
- std::size_t
- read_header(
- SyncReadStream& stream,
- DynamicBuffer& buffer,
- basic_parser<isRequest>& parser,
- error_code& ec)
- {
- static_assert(
- is_sync_read_stream<SyncReadStream>::value,
- "SyncReadStream type requirements not met");
- static_assert(
- net::is_dynamic_buffer<DynamicBuffer>::value,
- "DynamicBuffer type requirements not met");
- parser.eager(false);
- return beast::detail::read(stream, buffer,
- detail::read_header_condition<
- isRequest>{parser}, ec);
- }
- template<
- class AsyncReadStream,
- class DynamicBuffer,
- bool isRequest,
- class ReadHandler>
- BOOST_BEAST_ASYNC_RESULT2(ReadHandler)
- async_read_header(
- AsyncReadStream& stream,
- DynamicBuffer& buffer,
- basic_parser<isRequest>& parser,
- ReadHandler&& handler)
- {
- parser.eager(false);
- return beast::detail::async_read(
- stream,
- buffer,
- detail::read_header_condition<
- isRequest>{parser},
- std::forward<ReadHandler>(handler));
- }
- //------------------------------------------------------------------------------
- template<
- class SyncReadStream,
- class DynamicBuffer,
- bool isRequest>
- std::size_t
- read(
- SyncReadStream& stream,
- DynamicBuffer& buffer,
- basic_parser<isRequest>& parser)
- {
- static_assert(
- is_sync_read_stream<SyncReadStream>::value,
- "SyncReadStream type requirements not met");
- static_assert(
- net::is_dynamic_buffer<DynamicBuffer>::value,
- "DynamicBuffer type requirements not met");
- error_code ec;
- auto const bytes_transferred =
- http::read(stream, buffer, parser, ec);
- if(ec)
- BOOST_THROW_EXCEPTION(system_error{ec});
- return bytes_transferred;
- }
- template<
- class SyncReadStream,
- class DynamicBuffer,
- bool isRequest>
- std::size_t
- read(
- SyncReadStream& stream,
- DynamicBuffer& buffer,
- basic_parser<isRequest>& parser,
- error_code& ec)
- {
- static_assert(
- is_sync_read_stream<SyncReadStream>::value,
- "SyncReadStream type requirements not met");
- static_assert(
- net::is_dynamic_buffer<DynamicBuffer>::value,
- "DynamicBuffer type requirements not met");
- parser.eager(true);
- return beast::detail::read(stream, buffer,
- detail::read_all_condition<
- isRequest>{parser}, ec);
- }
- template<
- class AsyncReadStream,
- class DynamicBuffer,
- bool isRequest,
- class ReadHandler>
- BOOST_BEAST_ASYNC_RESULT2(ReadHandler)
- async_read(
- AsyncReadStream& stream,
- DynamicBuffer& buffer,
- basic_parser<isRequest>& parser,
- ReadHandler&& handler)
- {
- static_assert(
- is_async_read_stream<AsyncReadStream>::value,
- "AsyncReadStream type requirements not met");
- static_assert(
- net::is_dynamic_buffer<DynamicBuffer>::value,
- "DynamicBuffer type requirements not met");
- parser.eager(true);
- return beast::detail::async_read(
- stream,
- buffer,
- detail::read_all_condition<
- isRequest>{parser},
- std::forward<ReadHandler>(handler));
- }
- //------------------------------------------------------------------------------
- template<
- class SyncReadStream,
- class DynamicBuffer,
- bool isRequest, class Body, class Allocator>
- std::size_t
- read(
- SyncReadStream& stream,
- DynamicBuffer& buffer,
- message<isRequest, Body, basic_fields<Allocator>>& msg)
- {
- static_assert(
- is_sync_read_stream<SyncReadStream>::value,
- "SyncReadStream type requirements not met");
- static_assert(
- net::is_dynamic_buffer<DynamicBuffer>::value,
- "DynamicBuffer type requirements not met");
- static_assert(is_body<Body>::value,
- "Body type requirements not met");
- static_assert(is_body_reader<Body>::value,
- "BodyReader type requirements not met");
- error_code ec;
- auto const bytes_transferred =
- http::read(stream, buffer, msg, ec);
- if(ec)
- BOOST_THROW_EXCEPTION(system_error{ec});
- return bytes_transferred;
- }
- template<
- class SyncReadStream,
- class DynamicBuffer,
- bool isRequest, class Body, class Allocator>
- std::size_t
- read(
- SyncReadStream& stream,
- DynamicBuffer& buffer,
- message<isRequest, Body, basic_fields<Allocator>>& msg,
- error_code& ec)
- {
- static_assert(
- is_sync_read_stream<SyncReadStream>::value,
- "SyncReadStream type requirements not met");
- static_assert(
- net::is_dynamic_buffer<DynamicBuffer>::value,
- "DynamicBuffer type requirements not met");
- static_assert(is_body<Body>::value,
- "Body type requirements not met");
- static_assert(is_body_reader<Body>::value,
- "BodyReader type requirements not met");
- parser<isRequest, Body, Allocator> p(std::move(msg));
- p.eager(true);
- auto const bytes_transferred =
- http::read(stream, buffer, p, ec);
- if(ec)
- return bytes_transferred;
- msg = p.release();
- return bytes_transferred;
- }
- template<
- class AsyncReadStream,
- class DynamicBuffer,
- bool isRequest, class Body, class Allocator,
- class ReadHandler>
- BOOST_BEAST_ASYNC_RESULT2(ReadHandler)
- async_read(
- AsyncReadStream& stream,
- DynamicBuffer& buffer,
- message<isRequest, Body, basic_fields<Allocator>>& msg,
- ReadHandler&& handler)
- {
- static_assert(
- is_async_read_stream<AsyncReadStream>::value,
- "AsyncReadStream type requirements not met");
- static_assert(
- net::is_dynamic_buffer<DynamicBuffer>::value,
- "DynamicBuffer type requirements not met");
- static_assert(is_body<Body>::value,
- "Body type requirements not met");
- static_assert(is_body_reader<Body>::value,
- "BodyReader type requirements not met");
- return net::async_initiate<
- ReadHandler,
- void(error_code, std::size_t)>(
- detail::run_read_msg_op{},
- handler, &stream, &buffer, &msg);
- }
- } // http
- } // beast
- } // boost
- #endif
|