mirror of
https://github.com/Xahau/xahaud.git
synced 2025-12-06 17:27:52 +00:00
8921da91b8
Core: * Test buffer_cat iterator move members HTTP: * Fixed yield / resume in writer * Fixed message serialization with chunked encoding * Test yield / resume in writer * Test all conditional branches during message serialization * Test chunked encoding * Increase coverage on parse_error * Add parse_error::general WebSocket: * Add error::general * Increase coverage in error
127 lines
3.3 KiB
C++
127 lines
3.3 KiB
C++
//
|
|
// Copyright (c) 2013-2016 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_TEST_STRING_STREAM_HPP
|
|
#define BEAST_TEST_STRING_STREAM_HPP
|
|
|
|
#include <beast/core/bind_handler.hpp>
|
|
#include <beast/core/error.hpp>
|
|
#include <boost/asio/buffer.hpp>
|
|
#include <boost/asio/io_service.hpp>
|
|
#include <string>
|
|
|
|
namespace beast {
|
|
namespace test {
|
|
|
|
/** A SyncStream and AsyncStream that reads from a string.
|
|
|
|
This class behaves like a socket, except that written data is simply
|
|
discarded, and when data is read it comes from a string provided
|
|
at construction.
|
|
*/
|
|
class string_stream
|
|
{
|
|
std::string s_;
|
|
boost::asio::io_service& ios_;
|
|
|
|
public:
|
|
string_stream(boost::asio::io_service& ios,
|
|
std::string s)
|
|
: s_(std::move(s))
|
|
, ios_(ios)
|
|
{
|
|
}
|
|
|
|
boost::asio::io_service&
|
|
get_io_service()
|
|
{
|
|
return ios_;
|
|
}
|
|
|
|
template<class MutableBufferSequence>
|
|
std::size_t
|
|
read_some(MutableBufferSequence const& buffers)
|
|
{
|
|
error_code ec;
|
|
auto const n = read_some(buffers, ec);
|
|
if(ec)
|
|
throw system_error{ec};
|
|
return n;
|
|
}
|
|
|
|
template<class MutableBufferSequence>
|
|
std::size_t
|
|
read_some(MutableBufferSequence const& buffers,
|
|
error_code& ec)
|
|
{
|
|
auto const n = boost::asio::buffer_copy(
|
|
buffers, boost::asio::buffer(s_));
|
|
if(n > 0)
|
|
s_.erase(0, n);
|
|
else
|
|
ec = boost::asio::error::eof;
|
|
return n;
|
|
}
|
|
|
|
template<class MutableBufferSequence, class ReadHandler>
|
|
typename async_completion<ReadHandler,
|
|
void(error_code, std::size_t)>::result_type
|
|
async_read_some(MutableBufferSequence const& buffers,
|
|
ReadHandler&& handler)
|
|
{
|
|
auto const n = boost::asio::buffer_copy(
|
|
buffers, boost::asio::buffer(s_));
|
|
error_code ec;
|
|
if(n > 0)
|
|
s_.erase(0, n);
|
|
else
|
|
ec = boost::asio::error::eof;
|
|
async_completion<ReadHandler,
|
|
void(error_code, std::size_t)> completion(handler);
|
|
ios_.post(bind_handler(
|
|
completion.handler, ec, n));
|
|
return completion.result.get();
|
|
}
|
|
|
|
template<class ConstBufferSequence>
|
|
std::size_t
|
|
write_some(ConstBufferSequence const& buffers)
|
|
{
|
|
error_code ec;
|
|
auto const n = write_some(buffers, ec);
|
|
if(ec)
|
|
throw system_error{ec};
|
|
return n;
|
|
}
|
|
|
|
template<class ConstBufferSequence>
|
|
std::size_t
|
|
write_some(ConstBufferSequence const& buffers,
|
|
error_code&)
|
|
{
|
|
return boost::asio::buffer_size(buffers);
|
|
}
|
|
|
|
template<class ConstBuffeSequence, class WriteHandler>
|
|
typename async_completion<WriteHandler,
|
|
void(error_code, std::size_t)>::result_type
|
|
async_write_some(ConstBuffeSequence const& buffers,
|
|
WriteHandler&& handler)
|
|
{
|
|
async_completion<WriteHandler,
|
|
void(error_code, std::size_t)> completion(handler);
|
|
ios_.post(bind_handler(completion.handler,
|
|
error_code{}, boost::asio::buffer_size(buffers)));
|
|
return completion.result.get();
|
|
}
|
|
};
|
|
|
|
} // test
|
|
} // beast
|
|
|
|
#endif
|