123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186 |
- //
- // 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_STRING_BODY_HPP
- #define BOOST_BEAST_HTTP_STRING_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/core/buffers_range.hpp>
- #include <boost/beast/core/detail/clamp.hpp>
- #include <boost/asio/buffer.hpp>
- #include <boost/optional.hpp>
- #include <cstdint>
- #include <limits>
- #include <memory>
- #include <stdexcept>
- #include <string>
- #include <utility>
- namespace boost {
- namespace beast {
- namespace http {
- /** A <em>Body</em> using `std::basic_string`
- This body uses `std::basic_string` as a memory-based container
- for holding message payloads. Messages using this body type
- may be serialized and parsed.
- */
- template<
- class CharT,
- class Traits = std::char_traits<CharT>,
- class Allocator = std::allocator<CharT>>
- struct basic_string_body
- {
- private:
- static_assert(
- std::is_integral<CharT>::value &&
- sizeof(CharT) == 1,
- "CharT requirements not met");
- public:
- /** The type of container used for the body
- This determines the type of @ref message::body
- when this body type is used with a message container.
- */
- using value_type =
- std::basic_string<CharT, Traits, Allocator>;
- /** Returns the payload size of the body
- When this body is used with @ref message::prepare_payload,
- the Content-Length will be set to the payload size, and
- any chunked Transfer-Encoding will be removed.
- */
- static
- std::uint64_t
- size(value_type const& body)
- {
- return body.size();
- }
- /** 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& length, error_code& ec)
- {
- if(length)
- {
- if(*length > body_.max_size())
- {
- ec = error::buffer_overflow;
- return;
- }
- body_.reserve(beast::detail::clamp(*length));
- }
- ec = {};
- }
- template<class ConstBufferSequence>
- std::size_t
- put(ConstBufferSequence const& buffers,
- error_code& ec)
- {
- auto const extra = buffer_bytes(buffers);
- auto const size = body_.size();
- if (extra > body_.max_size() - size)
- {
- ec = error::buffer_overflow;
- return 0;
- }
- body_.resize(size + extra);
- ec = {};
- CharT* dest = &body_[size];
- for(auto b : beast::buffers_range_ref(buffers))
- {
- Traits::copy(dest, static_cast<
- CharT const*>(b.data()), b.size());
- dest += b.size();
- }
- return extra;
- }
- 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
- {
- 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)
- {
- ec = {};
- return {{const_buffers_type{
- body_.data(), body_.size()}, false}};
- }
- };
- #endif
- };
- /// A <em>Body</em> using `std::string`
- using string_body = basic_string_body<char>;
- } // http
- } // beast
- } // boost
- #endif
|