123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232 |
- //
- // 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_BUFFER_BODY_HPP
- #define BOOST_BEAST_HTTP_BUFFER_BODY_HPP
- #include <boost/beast/core/detail/config.hpp>
- #include <boost/beast/core/buffer_traits.hpp>
- #include <boost/beast/http/error.hpp>
- #include <boost/beast/http/message.hpp>
- #include <boost/beast/http/type_traits.hpp>
- #include <boost/optional.hpp>
- #include <cstdint>
- #include <type_traits>
- #include <utility>
- namespace boost {
- namespace beast {
- namespace http {
- /** A <em>Body</em> using a caller provided buffer
- Messages using this body type may be serialized and parsed.
- To use this class, the caller must initialize the members
- of @ref buffer_body::value_type to appropriate values before
- each call to read or write during a stream operation.
- */
- struct buffer_body
- {
- /// The type of the body member when used in a message.
- struct value_type
- {
- /** A pointer to a contiguous area of memory of @ref size octets, else `nullptr`.
- @par When Serializing
- If this is `nullptr` and `more` is `true`, the error
- @ref error::need_buffer will be returned from @ref serializer::get
- Otherwise, the serializer will use the memory pointed to
- by `data` having `size` octets of valid storage as the
- next buffer representing the body.
- @par When Parsing
- If this is `nullptr`, the error @ref error::need_buffer
- will be returned from @ref parser::put. Otherwise, the
- parser will store body octets into the memory pointed to
- by `data` having `size` octets of valid storage. After
- octets are stored, the `data` and `size` members are
- adjusted: `data` is incremented to point to the next
- octet after the data written, while `size` is decremented
- to reflect the remaining space at the memory location
- pointed to by `data`.
- */
- void* data = nullptr;
- /** The number of octets in the buffer pointed to by @ref data.
- @par When Serializing
- If `data` is `nullptr` during serialization, this value
- is ignored. Otherwise, it represents the number of valid
- body octets pointed to by `data`.
- @par When Parsing
- The value of this field will be decremented during parsing
- to indicate the number of remaining free octets in the
- buffer pointed to by `data`. When it reaches zero, the
- parser will return @ref error::need_buffer, indicating to
- the caller that the values of `data` and `size` should be
- updated to point to a new memory buffer.
- */
- std::size_t size = 0;
- /** `true` if this is not the last buffer.
- @par When Serializing
-
- If this is `true` and `data` is `nullptr`, the error
- @ref error::need_buffer will be returned from @ref serializer::get
- @par When Parsing
- This field is not used during parsing.
- */
- bool more = true;
- };
- /** The algorithm for parsing the body
- Meets the requirements of <em>BodyReader</em>.
- */
- #if BOOST_BEAST_DOXYGEN
- using reader = __implementation_defined__;
- #else
- class reader
- {
- value_type& body_;
- public:
- template<bool isRequest, class Fields>
- explicit
- reader(header<isRequest, Fields>&, value_type& b)
- : body_(b)
- {
- }
- void
- init(boost::optional<std::uint64_t> const&, error_code& ec)
- {
- ec = {};
- }
- template<class ConstBufferSequence>
- std::size_t
- put(ConstBufferSequence const& buffers,
- error_code& ec)
- {
- if(! body_.data)
- {
- BOOST_BEAST_ASSIGN_EC(ec, error::need_buffer);
- return 0;
- }
- auto const bytes_transferred =
- net::buffer_copy(net::buffer(
- body_.data, body_.size), buffers);
- body_.data = static_cast<char*>(
- body_.data) + bytes_transferred;
- body_.size -= bytes_transferred;
- if(bytes_transferred == buffer_bytes(buffers))
- ec = {};
- else
- {
- BOOST_BEAST_ASSIGN_EC(ec, error::need_buffer);
- }
- return bytes_transferred;
- }
- void
- finish(error_code& ec)
- {
- ec = {};
- }
- };
- #endif
- /** The algorithm for serializing the body
- Meets the requirements of <em>BodyWriter</em>.
- */
- #if BOOST_BEAST_DOXYGEN
- using writer = __implementation_defined__;
- #else
- class writer
- {
- bool toggle_ = false;
- value_type const& body_;
- public:
- using const_buffers_type =
- net::const_buffer;
- template<bool isRequest, class Fields>
- explicit
- writer(header<isRequest, Fields> const&, value_type const& b)
- : body_(b)
- {
- }
- void
- init(error_code& ec)
- {
- ec = {};
- }
- boost::optional<
- std::pair<const_buffers_type, bool>>
- get(error_code& ec)
- {
- if(toggle_)
- {
- if(body_.more)
- {
- toggle_ = false;
- BOOST_BEAST_ASSIGN_EC(ec, error::need_buffer);
- }
- else
- {
- ec = {};
- }
- return boost::none;
- }
- if(body_.data)
- {
- ec = {};
- toggle_ = true;
- return {{const_buffers_type{
- body_.data, body_.size}, body_.more}};
- }
- if(body_.more)
- {
- BOOST_BEAST_ASSIGN_EC(ec, error::need_buffer);
- }
- else
- ec = {};
- return boost::none;
- }
- };
- #endif
- };
- #if ! BOOST_BEAST_DOXYGEN
- // operator<< is not supported for buffer_body
- template<bool isRequest, class Fields>
- std::ostream&
- operator<<(std::ostream& os, message<isRequest,
- buffer_body, Fields> const& msg) = delete;
- #endif
- } // http
- } // beast
- } // boost
- #endif
|