forked from boostorg/beast
146 lines
3.6 KiB
C++
146 lines
3.6 KiB
C++
//
|
|
// Copyright (c) 2013-2017 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_STRING_BODY_HPP
|
|
#define BEAST_HTTP_STRING_BODY_HPP
|
|
|
|
#include <beast/config.hpp>
|
|
#include <beast/http/error.hpp>
|
|
#include <beast/http/message.hpp>
|
|
#include <beast/core/detail/type_traits.hpp>
|
|
#include <boost/asio/buffer.hpp>
|
|
#include <boost/optional.hpp>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <utility>
|
|
|
|
namespace beast {
|
|
namespace http {
|
|
|
|
/** An HTTP message body represented by a `std::string`.
|
|
|
|
Meets the requirements of @b Body.
|
|
*/
|
|
struct string_body
|
|
{
|
|
/// The type of the body member when used in a message.
|
|
using value_type = std::string;
|
|
|
|
/// Returns the content length of the body in a message.
|
|
static
|
|
std::uint64_t
|
|
size(value_type const& v)
|
|
{
|
|
return v.size();
|
|
}
|
|
|
|
#if BEAST_DOXYGEN
|
|
/// The algorithm to obtain buffers representing the body
|
|
using reader = implementation_defined;
|
|
#else
|
|
class reader
|
|
{
|
|
value_type const& body_;
|
|
|
|
public:
|
|
using const_buffers_type =
|
|
boost::asio::const_buffers_1;
|
|
|
|
template<bool isRequest, class Fields>
|
|
explicit
|
|
reader(message<isRequest, string_body,
|
|
Fields> const& msg, error_code& ec)
|
|
: body_(msg.body)
|
|
{
|
|
ec.assign(0, ec.category());
|
|
}
|
|
|
|
boost::optional<std::pair<const_buffers_type, bool>>
|
|
get(error_code& ec)
|
|
{
|
|
ec.assign(0, ec.category());
|
|
return {{const_buffers_type{
|
|
body_.data(), body_.size()}, false}};
|
|
}
|
|
};
|
|
#endif
|
|
|
|
#if BEAST_DOXYGEN
|
|
/// The algorithm used store buffers in this body
|
|
using writer = implementation_defined;
|
|
#else
|
|
class writer
|
|
{
|
|
value_type& body_;
|
|
|
|
public:
|
|
template<bool isRequest, class Fields>
|
|
explicit
|
|
writer(message<isRequest, string_body, Fields>& m,
|
|
boost::optional<std::uint64_t> content_length,
|
|
error_code& ec)
|
|
: body_(m.body)
|
|
{
|
|
if(content_length)
|
|
{
|
|
if(*content_length > (std::numeric_limits<
|
|
std::size_t>::max)())
|
|
{
|
|
ec = error::buffer_overflow;
|
|
return;
|
|
}
|
|
try
|
|
{
|
|
body_.reserve(static_cast<
|
|
std::size_t>(*content_length));
|
|
}
|
|
catch(std::exception const&)
|
|
{
|
|
ec = error::buffer_overflow;
|
|
return;
|
|
}
|
|
}
|
|
ec.assign(0, ec.category());
|
|
}
|
|
|
|
template<class ConstBufferSequence>
|
|
std::size_t
|
|
put(ConstBufferSequence const& buffers,
|
|
error_code& ec)
|
|
{
|
|
using boost::asio::buffer_size;
|
|
using boost::asio::buffer_copy;
|
|
auto const n = buffer_size(buffers);
|
|
auto const len = body_.size();
|
|
try
|
|
{
|
|
body_.resize(len + n);
|
|
}
|
|
catch(std::exception const&)
|
|
{
|
|
ec = error::buffer_overflow;
|
|
return 0;
|
|
}
|
|
ec.assign(0, ec.category());
|
|
return buffer_copy(boost::asio::buffer(
|
|
&body_[0] + len, n), buffers);
|
|
}
|
|
|
|
void
|
|
finish(error_code& ec)
|
|
{
|
|
ec.assign(0, ec.category());
|
|
}
|
|
};
|
|
#endif
|
|
};
|
|
|
|
} // http
|
|
} // beast
|
|
|
|
#endif
|