123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242 |
- //
- // 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_IMPL_BUFFERED_READ_STREAM_HPP
- #define BOOST_BEAST_IMPL_BUFFERED_READ_STREAM_HPP
- #include <boost/beast/core/async_base.hpp>
- #include <boost/beast/core/bind_handler.hpp>
- #include <boost/beast/core/error.hpp>
- #include <boost/beast/core/read_size.hpp>
- #include <boost/beast/core/stream_traits.hpp>
- #include <boost/beast/core/detail/is_invocable.hpp>
- #include <boost/asio/post.hpp>
- #include <boost/throw_exception.hpp>
- namespace boost {
- namespace beast {
- template<class Stream, class DynamicBuffer>
- struct buffered_read_stream<Stream, DynamicBuffer>::ops
- {
- template<class MutableBufferSequence, class Handler>
- class read_op
- : public async_base<Handler,
- beast::executor_type<buffered_read_stream>>
- {
- buffered_read_stream& s_;
- MutableBufferSequence b_;
- int step_ = 0;
- public:
- read_op(read_op&&) = default;
- read_op(read_op const&) = delete;
- template<class Handler_>
- read_op(
- Handler_&& h,
- buffered_read_stream& s,
- MutableBufferSequence const& b)
- : async_base<
- Handler, beast::executor_type<buffered_read_stream>>(
- std::forward<Handler_>(h), s.get_executor())
- , s_(s)
- , b_(b)
- {
- (*this)({}, 0);
- }
- void
- operator()(
- error_code ec,
- std::size_t bytes_transferred)
- {
- // VFALCO TODO Rewrite this using reenter/yield
- switch(step_)
- {
- case 0:
- if(s_.buffer_.size() == 0)
- {
- if(s_.capacity_ == 0)
- {
- // read (unbuffered)
- step_ = 1;
- return s_.next_layer_.async_read_some(
- b_, std::move(*this));
- }
- // read
- step_ = 2;
- return s_.next_layer_.async_read_some(
- s_.buffer_.prepare(read_size(
- s_.buffer_, s_.capacity_)),
- std::move(*this));
- }
- step_ = 3;
- return net::post(
- s_.get_executor(),
- beast::bind_front_handler(
- std::move(*this), ec, 0));
- case 1:
- // upcall
- break;
- case 2:
- s_.buffer_.commit(bytes_transferred);
- BOOST_FALLTHROUGH;
- case 3:
- bytes_transferred =
- net::buffer_copy(b_, s_.buffer_.data());
- s_.buffer_.consume(bytes_transferred);
- break;
- }
- this->complete_now(ec, bytes_transferred);
- }
- };
- struct run_read_op
- {
- template<class ReadHandler, class Buffers>
- void
- operator()(
- ReadHandler&& h,
- buffered_read_stream* s,
- Buffers const& b)
- {
- // 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_op<
- Buffers,
- typename std::decay<ReadHandler>::type>(
- std::forward<ReadHandler>(h), *s, b);
- }
- };
- };
- //------------------------------------------------------------------------------
- template<class Stream, class DynamicBuffer>
- template<class... Args>
- buffered_read_stream<Stream, DynamicBuffer>::
- buffered_read_stream(Args&&... args)
- : next_layer_(std::forward<Args>(args)...)
- {
- }
- template<class Stream, class DynamicBuffer>
- template<class ConstBufferSequence, class WriteHandler>
- BOOST_BEAST_ASYNC_RESULT2(WriteHandler)
- buffered_read_stream<Stream, DynamicBuffer>::
- async_write_some(
- ConstBufferSequence const& buffers,
- WriteHandler&& handler)
- {
- static_assert(is_async_write_stream<next_layer_type>::value,
- "AsyncWriteStream type requirements not met");
- static_assert(net::is_const_buffer_sequence<
- ConstBufferSequence>::value,
- "ConstBufferSequence type requirements not met");
- static_assert(detail::is_invocable<WriteHandler,
- void(error_code, std::size_t)>::value,
- "WriteHandler type requirements not met");
- return next_layer_.async_write_some(buffers,
- std::forward<WriteHandler>(handler));
- }
- template<class Stream, class DynamicBuffer>
- template<class MutableBufferSequence>
- std::size_t
- buffered_read_stream<Stream, DynamicBuffer>::
- read_some(
- MutableBufferSequence const& buffers)
- {
- static_assert(is_sync_read_stream<next_layer_type>::value,
- "SyncReadStream type requirements not met");
- static_assert(net::is_mutable_buffer_sequence<
- MutableBufferSequence>::value,
- "MutableBufferSequence type requirements not met");
- error_code ec;
- auto n = read_some(buffers, ec);
- if(ec)
- BOOST_THROW_EXCEPTION(system_error{ec});
- return n;
- }
- template<class Stream, class DynamicBuffer>
- template<class MutableBufferSequence>
- std::size_t
- buffered_read_stream<Stream, DynamicBuffer>::
- read_some(MutableBufferSequence const& buffers,
- error_code& ec)
- {
- static_assert(is_sync_read_stream<next_layer_type>::value,
- "SyncReadStream type requirements not met");
- static_assert(net::is_mutable_buffer_sequence<
- MutableBufferSequence>::value,
- "MutableBufferSequence type requirements not met");
- if(buffer_.size() == 0)
- {
- if(capacity_ == 0)
- return next_layer_.read_some(buffers, ec);
- buffer_.commit(next_layer_.read_some(
- buffer_.prepare(read_size(buffer_,
- capacity_)), ec));
- if(ec)
- return 0;
- }
- else
- {
- ec = {};
- }
- auto bytes_transferred =
- net::buffer_copy(buffers, buffer_.data());
- buffer_.consume(bytes_transferred);
- return bytes_transferred;
- }
- template<class Stream, class DynamicBuffer>
- template<class MutableBufferSequence, class ReadHandler>
- BOOST_BEAST_ASYNC_RESULT2(ReadHandler)
- buffered_read_stream<Stream, DynamicBuffer>::
- async_read_some(
- MutableBufferSequence const& buffers,
- ReadHandler&& handler)
- {
- static_assert(is_async_read_stream<next_layer_type>::value,
- "AsyncReadStream type requirements not met");
- static_assert(net::is_mutable_buffer_sequence<
- MutableBufferSequence>::value,
- "MutableBufferSequence type requirements not met");
- if(buffer_.size() == 0 && capacity_ == 0)
- return next_layer_.async_read_some(buffers,
- std::forward<ReadHandler>(handler));
- return net::async_initiate<
- ReadHandler,
- void(error_code, std::size_t)>(
- typename ops::run_read_op{},
- handler,
- this,
- buffers);
- }
- } // beast
- } // boost
- #endif
|