forked from boostorg/beast
Vinnie Falco
c29451a8dd
fix #373 * concepts.hpp is renamed to type_traits.hpp * Body reader and writer concepts are renamed
334 lines
8.6 KiB
C++
334 lines
8.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_IMPL_READ_IPP_HPP
|
|
#define BEAST_HTTP_IMPL_READ_IPP_HPP
|
|
|
|
#include <beast/http/type_traits.hpp>
|
|
#include <beast/http/error.hpp>
|
|
#include <beast/http/message_parser.hpp>
|
|
#include <beast/http/read.hpp>
|
|
#include <beast/core/bind_handler.hpp>
|
|
#include <beast/core/handler_ptr.hpp>
|
|
#include <beast/core/type_traits.hpp>
|
|
#include <beast/core/detail/read_size_helper.hpp>
|
|
#include <boost/asio/handler_alloc_hook.hpp>
|
|
#include <boost/asio/handler_continuation_hook.hpp>
|
|
#include <boost/asio/handler_invoke_hook.hpp>
|
|
#include <boost/assert.hpp>
|
|
#include <boost/optional.hpp>
|
|
#include <boost/throw_exception.hpp>
|
|
|
|
namespace beast {
|
|
namespace http {
|
|
|
|
namespace detail {
|
|
|
|
template<
|
|
class SyncReadStream,
|
|
class DynamicBuffer,
|
|
bool isRequest, bool isDirect, class Derived>
|
|
inline
|
|
std::size_t
|
|
read_some_buffer(
|
|
SyncReadStream& stream,
|
|
DynamicBuffer& buffer,
|
|
basic_parser<isRequest, isDirect, Derived>& parser,
|
|
error_code& ec)
|
|
{
|
|
std::size_t bytes_used;
|
|
if(buffer.size() == 0)
|
|
goto do_read;
|
|
for(;;)
|
|
{
|
|
bytes_used = parser.write(
|
|
buffer.data(), ec);
|
|
if(ec)
|
|
return 0;
|
|
if(bytes_used > 0)
|
|
goto do_finish;
|
|
do_read:
|
|
boost::optional<typename
|
|
DynamicBuffer::mutable_buffers_type> mb;
|
|
using beast::detail::read_size_helper;
|
|
auto const size =
|
|
read_size_helper(buffer, 65536);
|
|
BOOST_ASSERT(size > 0);
|
|
try
|
|
{
|
|
mb.emplace(buffer.prepare(size));
|
|
}
|
|
catch(std::length_error const&)
|
|
{
|
|
ec = error::buffer_overflow;
|
|
return 0;
|
|
}
|
|
auto const bytes_transferred =
|
|
stream.read_some(*mb, ec);
|
|
if(ec == boost::asio::error::eof)
|
|
{
|
|
BOOST_ASSERT(bytes_transferred == 0);
|
|
bytes_used = 0;
|
|
if(parser.got_some())
|
|
{
|
|
// caller sees EOF on next read
|
|
ec = {};
|
|
parser.write_eof(ec);
|
|
if(ec)
|
|
return 0;
|
|
BOOST_ASSERT(parser.is_complete());
|
|
break;
|
|
}
|
|
ec = error::end_of_stream;
|
|
break;
|
|
}
|
|
else if(ec)
|
|
{
|
|
return 0;
|
|
}
|
|
BOOST_ASSERT(bytes_transferred > 0);
|
|
buffer.commit(bytes_transferred);
|
|
}
|
|
do_finish:
|
|
return bytes_used;
|
|
}
|
|
|
|
template<
|
|
class SyncReadStream,
|
|
class DynamicBuffer,
|
|
bool isRequest, class Derived>
|
|
inline
|
|
std::size_t
|
|
read_some_body(
|
|
SyncReadStream& stream,
|
|
DynamicBuffer& buffer,
|
|
basic_parser<isRequest, true, Derived>& parser,
|
|
error_code& ec)
|
|
{
|
|
if(buffer.size() > 0)
|
|
return parser.copy_body(buffer);
|
|
boost::optional<typename
|
|
Derived::mutable_buffers_type> mb;
|
|
try
|
|
{
|
|
parser.prepare_body(mb, 65536);
|
|
}
|
|
catch(std::length_error const&)
|
|
{
|
|
ec = error::buffer_overflow;
|
|
return 0;
|
|
}
|
|
auto const bytes_transferred =
|
|
stream.read_some(*mb, ec);
|
|
if(ec == boost::asio::error::eof)
|
|
{
|
|
BOOST_ASSERT(bytes_transferred == 0);
|
|
// caller sees EOF on next read
|
|
ec = {};
|
|
parser.write_eof(ec);
|
|
if(ec)
|
|
return 0;
|
|
BOOST_ASSERT(parser.is_complete());
|
|
}
|
|
else if(! ec)
|
|
{
|
|
parser.commit_body(bytes_transferred);
|
|
return 0;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
template<
|
|
class SyncReadStream,
|
|
class DynamicBuffer,
|
|
bool isRequest, class Derived>
|
|
inline
|
|
std::size_t
|
|
read_some(
|
|
SyncReadStream& stream,
|
|
DynamicBuffer& buffer,
|
|
basic_parser<isRequest, true, Derived>& parser,
|
|
error_code& ec)
|
|
{
|
|
switch(parser.state())
|
|
{
|
|
case parse_state::header:
|
|
case parse_state::chunk_header:
|
|
return detail::read_some_buffer(
|
|
stream, buffer, parser, ec);
|
|
|
|
default:
|
|
return detail::read_some_body(
|
|
stream, buffer, parser, ec);
|
|
}
|
|
}
|
|
|
|
template<
|
|
class SyncReadStream,
|
|
class DynamicBuffer,
|
|
bool isRequest, class Derived>
|
|
inline
|
|
std::size_t
|
|
read_some(
|
|
SyncReadStream& stream,
|
|
DynamicBuffer& buffer,
|
|
basic_parser<isRequest, false, Derived>& parser,
|
|
error_code& ec)
|
|
{
|
|
return detail::read_some_buffer(
|
|
stream, buffer, parser, ec);
|
|
}
|
|
|
|
} // detail
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
template<
|
|
class SyncReadStream,
|
|
class DynamicBuffer,
|
|
bool isRequest, bool isDirect, class Derived>
|
|
std::size_t
|
|
read_some(
|
|
SyncReadStream& stream,
|
|
DynamicBuffer& buffer,
|
|
basic_parser<isRequest, isDirect, Derived>& parser)
|
|
{
|
|
static_assert(is_sync_read_stream<SyncReadStream>::value,
|
|
"SyncReadStream requirements not met");
|
|
static_assert(is_dynamic_buffer<DynamicBuffer>::value,
|
|
"DynamicBuffer requirements not met");
|
|
BOOST_ASSERT(! parser.is_complete());
|
|
error_code ec;
|
|
auto const bytes_used =
|
|
read_some(stream, buffer, parser, ec);
|
|
if(ec)
|
|
BOOST_THROW_EXCEPTION(system_error{ec});
|
|
return bytes_used;
|
|
}
|
|
|
|
template<
|
|
class SyncReadStream,
|
|
class DynamicBuffer,
|
|
bool isRequest, bool isDirect, class Derived>
|
|
std::size_t
|
|
read_some(
|
|
SyncReadStream& stream,
|
|
DynamicBuffer& buffer,
|
|
basic_parser<isRequest, isDirect, Derived>& parser,
|
|
error_code& ec)
|
|
{
|
|
static_assert(is_sync_read_stream<SyncReadStream>::value,
|
|
"SyncReadStream requirements not met");
|
|
static_assert(is_dynamic_buffer<DynamicBuffer>::value,
|
|
"DynamicBuffer requirements not met");
|
|
BOOST_ASSERT(! parser.is_complete());
|
|
return detail::read_some(stream, buffer, parser, ec);
|
|
}
|
|
|
|
template<
|
|
class SyncReadStream,
|
|
class DynamicBuffer,
|
|
bool isRequest, bool isDirect, class Derived>
|
|
void
|
|
read(
|
|
SyncReadStream& stream,
|
|
DynamicBuffer& buffer,
|
|
basic_parser<isRequest, isDirect, Derived>& parser)
|
|
{
|
|
static_assert(is_sync_read_stream<SyncReadStream>::value,
|
|
"SyncReadStream requirements not met");
|
|
static_assert(is_dynamic_buffer<DynamicBuffer>::value,
|
|
"DynamicBuffer requirements not met");
|
|
BOOST_ASSERT(! parser.is_complete());
|
|
error_code ec;
|
|
read(stream, buffer, parser, ec);
|
|
if(ec)
|
|
BOOST_THROW_EXCEPTION(system_error{ec});
|
|
}
|
|
|
|
template<
|
|
class SyncReadStream,
|
|
class DynamicBuffer,
|
|
bool isRequest, bool isDirect, class Derived>
|
|
void
|
|
read(
|
|
SyncReadStream& stream,
|
|
DynamicBuffer& buffer,
|
|
basic_parser<isRequest, isDirect, Derived>& parser,
|
|
error_code& ec)
|
|
{
|
|
static_assert(is_sync_read_stream<SyncReadStream>::value,
|
|
"SyncReadStream requirements not met");
|
|
static_assert(is_dynamic_buffer<DynamicBuffer>::value,
|
|
"DynamicBuffer requirements not met");
|
|
BOOST_ASSERT(! parser.is_complete());
|
|
do
|
|
{
|
|
auto const bytes_used =
|
|
read_some(stream, buffer, parser, ec);
|
|
if(ec)
|
|
return;
|
|
buffer.consume(bytes_used);
|
|
}
|
|
while(! parser.is_complete());
|
|
}
|
|
|
|
template<
|
|
class SyncReadStream,
|
|
class DynamicBuffer,
|
|
bool isRequest, class Body, class Fields>
|
|
void
|
|
read(
|
|
SyncReadStream& stream,
|
|
DynamicBuffer& buffer,
|
|
message<isRequest, Body, Fields>& msg)
|
|
{
|
|
static_assert(is_sync_read_stream<SyncReadStream>::value,
|
|
"SyncReadStream requirements not met");
|
|
static_assert(is_dynamic_buffer<DynamicBuffer>::value,
|
|
"DynamicBuffer requirements not met");
|
|
static_assert(is_body<Body>::value,
|
|
"Body requirements not met");
|
|
static_assert(is_body_writer<Body>::value,
|
|
"BodyWriter requirements not met");
|
|
error_code ec;
|
|
beast::http::read(stream, buffer, msg, ec);
|
|
if(ec)
|
|
BOOST_THROW_EXCEPTION(system_error{ec});
|
|
}
|
|
|
|
template<
|
|
class SyncReadStream,
|
|
class DynamicBuffer,
|
|
bool isRequest, class Body, class Fields>
|
|
void
|
|
read(
|
|
SyncReadStream& stream,
|
|
DynamicBuffer& buffer,
|
|
message<isRequest, Body, Fields>& msg,
|
|
error_code& ec)
|
|
{
|
|
static_assert(is_sync_read_stream<SyncReadStream>::value,
|
|
"SyncReadStream requirements not met");
|
|
static_assert(is_dynamic_buffer<DynamicBuffer>::value,
|
|
"DynamicBuffer requirements not met");
|
|
static_assert(is_body<Body>::value,
|
|
"Body requirements not met");
|
|
static_assert(is_body_writer<Body>::value,
|
|
"BodyWriter requirements not met");
|
|
message_parser<isRequest, Body, Fields> p;
|
|
beast::http::read(stream, buffer, p, ec);
|
|
if(ec)
|
|
return;
|
|
msg = p.release();
|
|
}
|
|
|
|
} // http
|
|
} // beast
|
|
|
|
#endif
|