add boost on mac
This commit is contained in:
40
macx64/include/boost/beast/_experimental/test/error.hpp
Normal file
40
macx64/include/boost/beast/_experimental/test/error.hpp
Normal file
@@ -0,0 +1,40 @@
|
||||
//
|
||||
// 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_TEST_ERROR_HPP
|
||||
#define BOOST_BEAST_TEST_ERROR_HPP
|
||||
|
||||
#include <boost/beast/core/detail/config.hpp>
|
||||
#include <boost/beast/core/error.hpp>
|
||||
|
||||
namespace boost {
|
||||
namespace beast {
|
||||
namespace test {
|
||||
|
||||
/// Error codes returned from unit testing algorithms
|
||||
enum class error
|
||||
{
|
||||
/** The test stream generated a simulated testing error
|
||||
|
||||
This error is returned by a @ref fail_count object
|
||||
when it generates a simulated error.
|
||||
*/
|
||||
test_failure = 1
|
||||
};
|
||||
|
||||
} // test
|
||||
} // beast
|
||||
} // boost
|
||||
|
||||
#include <boost/beast/_experimental/test/impl/error.hpp>
|
||||
#ifdef BOOST_BEAST_HEADER_ONLY
|
||||
#include <boost/beast/_experimental/test/impl/error.ipp>
|
||||
#endif
|
||||
|
||||
#endif
|
||||
70
macx64/include/boost/beast/_experimental/test/fail_count.hpp
Normal file
70
macx64/include/boost/beast/_experimental/test/fail_count.hpp
Normal file
@@ -0,0 +1,70 @@
|
||||
//
|
||||
// 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_TEST_FAIL_COUNT_HPP
|
||||
#define BOOST_BEAST_TEST_FAIL_COUNT_HPP
|
||||
|
||||
#include <boost/beast/core/detail/config.hpp>
|
||||
#include <boost/beast/_experimental/test/error.hpp>
|
||||
#include <cstdlib>
|
||||
|
||||
namespace boost {
|
||||
namespace beast {
|
||||
namespace test {
|
||||
|
||||
/** A countdown to simulated failure
|
||||
|
||||
On the Nth operation, the class will fail with the specified
|
||||
error code, or the default error code of @ref error::test_failure.
|
||||
|
||||
Instances of this class may be used to build objects which
|
||||
are specifically designed to aid in writing unit tests, for
|
||||
interfaces which can throw exceptions or return `error_code`
|
||||
values representing failure.
|
||||
*/
|
||||
class fail_count
|
||||
{
|
||||
std::size_t n_;
|
||||
std::size_t i_ = 0;
|
||||
error_code ec_;
|
||||
|
||||
public:
|
||||
fail_count(fail_count&&) = default;
|
||||
|
||||
/** Construct a counter
|
||||
|
||||
@param n The 0-based index of the operation to fail on or after
|
||||
@param ev An optional error code to use when generating a simulated failure
|
||||
*/
|
||||
BOOST_BEAST_DECL
|
||||
explicit
|
||||
fail_count(
|
||||
std::size_t n,
|
||||
error_code ev = error::test_failure);
|
||||
|
||||
/// Throw an exception on the Nth failure
|
||||
BOOST_BEAST_DECL
|
||||
void
|
||||
fail();
|
||||
|
||||
/// Set an error code on the Nth failure
|
||||
BOOST_BEAST_DECL
|
||||
bool
|
||||
fail(error_code& ec);
|
||||
};
|
||||
|
||||
} // test
|
||||
} // beast
|
||||
} // boost
|
||||
|
||||
#ifdef BOOST_BEAST_HEADER_ONLY
|
||||
#include <boost/beast/_experimental/test/impl/fail_count.ipp>
|
||||
#endif
|
||||
|
||||
#endif
|
||||
188
macx64/include/boost/beast/_experimental/test/handler.hpp
Normal file
188
macx64/include/boost/beast/_experimental/test/handler.hpp
Normal file
@@ -0,0 +1,188 @@
|
||||
//
|
||||
// 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_TEST_HANDLER_HPP
|
||||
#define BOOST_BEAST_TEST_HANDLER_HPP
|
||||
|
||||
#include <boost/beast/_experimental/unit_test/suite.hpp>
|
||||
#include <boost/beast/core/error.hpp>
|
||||
#include <boost/asio/io_context.hpp>
|
||||
#include <boost/core/exchange.hpp>
|
||||
#include <boost/optional.hpp>
|
||||
|
||||
namespace boost {
|
||||
namespace beast {
|
||||
namespace test {
|
||||
|
||||
/** A CompletionHandler used for testing.
|
||||
|
||||
This completion handler is used by tests to ensure correctness
|
||||
of behavior. It is designed as a single type to reduce template
|
||||
instantiations, with configurable settings through constructor
|
||||
arguments. Typically this type will be used in type lists and
|
||||
not instantiated directly; instances of this class are returned
|
||||
by the helper functions listed below.
|
||||
|
||||
@see success_handler, @ref fail_handler, @ref any_handler
|
||||
*/
|
||||
class handler
|
||||
{
|
||||
boost::optional<error_code> ec_;
|
||||
bool pass_ = false;
|
||||
|
||||
public:
|
||||
handler() = default;
|
||||
|
||||
explicit
|
||||
handler(error_code ec)
|
||||
: ec_(ec)
|
||||
{
|
||||
}
|
||||
|
||||
explicit
|
||||
handler(boost::none_t)
|
||||
{
|
||||
}
|
||||
|
||||
handler(handler&& other)
|
||||
: ec_(other.ec_)
|
||||
, pass_(boost::exchange(other.pass_, true))
|
||||
{
|
||||
}
|
||||
|
||||
~handler()
|
||||
{
|
||||
BEAST_EXPECT(pass_);
|
||||
}
|
||||
|
||||
template<class... Args>
|
||||
void
|
||||
operator()(error_code ec, Args&&...)
|
||||
{
|
||||
BEAST_EXPECT(! pass_); // can't call twice
|
||||
BEAST_EXPECTS(! ec_ || ec == *ec_,
|
||||
ec.message());
|
||||
pass_ = true;
|
||||
}
|
||||
|
||||
void
|
||||
operator()()
|
||||
{
|
||||
BEAST_EXPECT(! pass_); // can't call twice
|
||||
BEAST_EXPECT(! ec_);
|
||||
pass_ = true;
|
||||
}
|
||||
|
||||
template<class Arg0, class... Args,
|
||||
class = typename std::enable_if<
|
||||
! std::is_convertible<Arg0, error_code>::value>::type>
|
||||
void
|
||||
operator()(Arg0&&, Args&&...)
|
||||
{
|
||||
BEAST_EXPECT(! pass_); // can't call twice
|
||||
BEAST_EXPECT(! ec_);
|
||||
pass_ = true;
|
||||
}
|
||||
};
|
||||
|
||||
/** Return a test CompletionHandler which requires success.
|
||||
|
||||
The returned handler can be invoked with any signature whose
|
||||
first parameter is an `error_code`. The handler fails the test
|
||||
if:
|
||||
|
||||
@li The handler is destroyed without being invoked, or
|
||||
|
||||
@li The handler is invoked with a non-successful error code.
|
||||
*/
|
||||
inline
|
||||
handler
|
||||
success_handler() noexcept
|
||||
{
|
||||
return handler(error_code{});
|
||||
}
|
||||
|
||||
/** Return a test CompletionHandler which requires invocation.
|
||||
|
||||
The returned handler can be invoked with any signature.
|
||||
The handler fails the test if:
|
||||
|
||||
@li The handler is destroyed without being invoked.
|
||||
*/
|
||||
inline
|
||||
handler
|
||||
any_handler() noexcept
|
||||
{
|
||||
return handler(boost::none);
|
||||
}
|
||||
|
||||
/** Return a test CompletionHandler which requires a specific error code.
|
||||
|
||||
This handler can be invoked with any signature whose first
|
||||
parameter is an `error_code`. The handler fails the test if:
|
||||
|
||||
@li The handler is destroyed without being invoked.
|
||||
|
||||
@li The handler is invoked with an error code different from
|
||||
what is specified.
|
||||
|
||||
@param ec The error code to specify.
|
||||
*/
|
||||
inline
|
||||
handler
|
||||
fail_handler(error_code ec) noexcept
|
||||
{
|
||||
return handler(ec);
|
||||
}
|
||||
|
||||
/** Run an I/O context.
|
||||
|
||||
This function runs and dispatches handlers on the specified
|
||||
I/O context, until one of the following conditions is true:
|
||||
|
||||
@li The I/O context runs out of work.
|
||||
|
||||
@param ioc The I/O context to run
|
||||
*/
|
||||
inline
|
||||
void
|
||||
run(net::io_context& ioc)
|
||||
{
|
||||
ioc.run();
|
||||
ioc.restart();
|
||||
}
|
||||
|
||||
/** Run an I/O context for a certain amount of time.
|
||||
|
||||
This function runs and dispatches handlers on the specified
|
||||
I/O context, until one of the following conditions is true:
|
||||
|
||||
@li The I/O context runs out of work.
|
||||
|
||||
@li No completions occur and the specified amount of time has elapsed.
|
||||
|
||||
@param ioc The I/O context to run
|
||||
|
||||
@param elapsed The maximum amount of time to run for.
|
||||
*/
|
||||
template<class Rep, class Period>
|
||||
void
|
||||
run_for(
|
||||
net::io_context& ioc,
|
||||
std::chrono::duration<Rep, Period> elapsed)
|
||||
{
|
||||
ioc.run_for(elapsed);
|
||||
ioc.restart();
|
||||
}
|
||||
|
||||
} // test
|
||||
} // beast
|
||||
} // boost
|
||||
|
||||
#endif
|
||||
40
macx64/include/boost/beast/_experimental/test/impl/error.hpp
Normal file
40
macx64/include/boost/beast/_experimental/test/impl/error.hpp
Normal file
@@ -0,0 +1,40 @@
|
||||
//
|
||||
// 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_TEST_IMPL_ERROR_HPP
|
||||
#define BOOST_BEAST_TEST_IMPL_ERROR_HPP
|
||||
|
||||
#include <boost/beast/core/error.hpp>
|
||||
#include <boost/beast/core/string.hpp>
|
||||
#include <type_traits>
|
||||
|
||||
namespace boost {
|
||||
namespace system {
|
||||
template<>
|
||||
struct is_error_code_enum<
|
||||
boost::beast::test::error>
|
||||
: std::true_type
|
||||
{
|
||||
};
|
||||
} // system
|
||||
} // boost
|
||||
|
||||
namespace boost {
|
||||
namespace beast {
|
||||
namespace test {
|
||||
|
||||
BOOST_BEAST_DECL
|
||||
error_code
|
||||
make_error_code(error e) noexcept;
|
||||
|
||||
} // test
|
||||
} // beast
|
||||
} // boost
|
||||
|
||||
#endif
|
||||
65
macx64/include/boost/beast/_experimental/test/impl/error.ipp
Normal file
65
macx64/include/boost/beast/_experimental/test/impl/error.ipp
Normal file
@@ -0,0 +1,65 @@
|
||||
//
|
||||
// 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_TEST_IMPL_ERROR_IPP
|
||||
#define BOOST_BEAST_TEST_IMPL_ERROR_IPP
|
||||
|
||||
#include <boost/beast/_experimental/test/error.hpp>
|
||||
|
||||
namespace boost {
|
||||
namespace beast {
|
||||
namespace test {
|
||||
|
||||
namespace detail {
|
||||
|
||||
class error_codes : public error_category
|
||||
{
|
||||
public:
|
||||
BOOST_BEAST_DECL
|
||||
const char*
|
||||
name() const noexcept override
|
||||
{
|
||||
return "boost.beast.test";
|
||||
}
|
||||
|
||||
BOOST_BEAST_DECL
|
||||
std::string
|
||||
message(int ev) const override
|
||||
{
|
||||
switch(static_cast<error>(ev))
|
||||
{
|
||||
default:
|
||||
case error::test_failure: return
|
||||
"An automatic unit test failure occurred";
|
||||
}
|
||||
}
|
||||
|
||||
BOOST_BEAST_DECL
|
||||
error_condition
|
||||
default_error_condition(int ev) const noexcept override
|
||||
{
|
||||
return error_condition{ev, *this};
|
||||
}
|
||||
};
|
||||
|
||||
} // detail
|
||||
|
||||
error_code
|
||||
make_error_code(error e) noexcept
|
||||
{
|
||||
static detail::error_codes const cat{};
|
||||
return error_code{static_cast<
|
||||
std::underlying_type<error>::type>(e), cat};
|
||||
}
|
||||
|
||||
} // test
|
||||
} // beast
|
||||
} // boost
|
||||
|
||||
#endif
|
||||
@@ -0,0 +1,58 @@
|
||||
//
|
||||
// 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_TEST_IMPL_FAIL_COUNT_IPP
|
||||
#define BOOST_BEAST_TEST_IMPL_FAIL_COUNT_IPP
|
||||
|
||||
#include <boost/beast/_experimental/test/fail_count.hpp>
|
||||
#include <boost/throw_exception.hpp>
|
||||
|
||||
namespace boost {
|
||||
namespace beast {
|
||||
namespace test {
|
||||
|
||||
fail_count::
|
||||
fail_count(
|
||||
std::size_t n,
|
||||
error_code ev)
|
||||
: n_(n)
|
||||
, ec_(ev)
|
||||
{
|
||||
}
|
||||
|
||||
void
|
||||
fail_count::
|
||||
fail()
|
||||
{
|
||||
if(i_ < n_)
|
||||
++i_;
|
||||
if(i_ == n_)
|
||||
BOOST_THROW_EXCEPTION(system_error{ec_});
|
||||
}
|
||||
|
||||
bool
|
||||
fail_count::
|
||||
fail(error_code& ec)
|
||||
{
|
||||
if(i_ < n_)
|
||||
++i_;
|
||||
if(i_ == n_)
|
||||
{
|
||||
ec = ec_;
|
||||
return true;
|
||||
}
|
||||
ec = {};
|
||||
return false;
|
||||
}
|
||||
|
||||
} // test
|
||||
} // beast
|
||||
} // boost
|
||||
|
||||
#endif
|
||||
453
macx64/include/boost/beast/_experimental/test/impl/stream.hpp
Normal file
453
macx64/include/boost/beast/_experimental/test/impl/stream.hpp
Normal file
@@ -0,0 +1,453 @@
|
||||
//
|
||||
// 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_TEST_IMPL_STREAM_HPP
|
||||
#define BOOST_BEAST_TEST_IMPL_STREAM_HPP
|
||||
|
||||
#include <boost/beast/core/bind_handler.hpp>
|
||||
#include <boost/beast/core/buffer_traits.hpp>
|
||||
#include <boost/beast/core/buffers_prefix.hpp>
|
||||
#include <boost/beast/core/detail/service_base.hpp>
|
||||
#include <boost/beast/core/detail/type_traits.hpp>
|
||||
#include <mutex>
|
||||
#include <stdexcept>
|
||||
#include <vector>
|
||||
|
||||
namespace boost {
|
||||
namespace beast {
|
||||
namespace test {
|
||||
|
||||
//------------------------------------------------------------------------------
|
||||
|
||||
struct stream::service_impl
|
||||
{
|
||||
std::mutex m_;
|
||||
std::vector<state*> v_;
|
||||
|
||||
BOOST_BEAST_DECL
|
||||
void
|
||||
remove(state& impl);
|
||||
};
|
||||
|
||||
class stream::service
|
||||
: public beast::detail::service_base<service>
|
||||
{
|
||||
boost::shared_ptr<service_impl> sp_;
|
||||
|
||||
BOOST_BEAST_DECL
|
||||
void
|
||||
shutdown() override;
|
||||
|
||||
public:
|
||||
BOOST_BEAST_DECL
|
||||
explicit
|
||||
service(net::execution_context& ctx);
|
||||
|
||||
BOOST_BEAST_DECL
|
||||
static
|
||||
auto
|
||||
make_impl(
|
||||
net::io_context& ctx,
|
||||
test::fail_count* fc) ->
|
||||
boost::shared_ptr<state>;
|
||||
};
|
||||
|
||||
//------------------------------------------------------------------------------
|
||||
|
||||
template<class Handler, class Buffers>
|
||||
class stream::read_op : public stream::read_op_base
|
||||
{
|
||||
using ex1_type =
|
||||
net::io_context::executor_type;
|
||||
using ex2_type
|
||||
= net::associated_executor_t<Handler, ex1_type>;
|
||||
|
||||
struct lambda
|
||||
{
|
||||
Handler h_;
|
||||
boost::weak_ptr<state> wp_;
|
||||
Buffers b_;
|
||||
net::executor_work_guard<ex2_type> wg2_;
|
||||
|
||||
lambda(lambda&&) = default;
|
||||
lambda(lambda const&) = default;
|
||||
|
||||
template<class Handler_>
|
||||
lambda(
|
||||
Handler_&& h,
|
||||
boost::shared_ptr<state> const& s,
|
||||
Buffers const& b)
|
||||
: h_(std::forward<Handler_>(h))
|
||||
, wp_(s)
|
||||
, b_(b)
|
||||
, wg2_(net::get_associated_executor(
|
||||
h_, s->ioc.get_executor()))
|
||||
{
|
||||
}
|
||||
|
||||
void
|
||||
operator()(error_code ec)
|
||||
{
|
||||
std::size_t bytes_transferred = 0;
|
||||
auto sp = wp_.lock();
|
||||
if(! sp)
|
||||
ec = net::error::operation_aborted;
|
||||
if(! ec)
|
||||
{
|
||||
std::lock_guard<std::mutex> lock(sp->m);
|
||||
BOOST_ASSERT(! sp->op);
|
||||
if(sp->b.size() > 0)
|
||||
{
|
||||
bytes_transferred =
|
||||
net::buffer_copy(
|
||||
b_, sp->b.data(), sp->read_max);
|
||||
sp->b.consume(bytes_transferred);
|
||||
}
|
||||
else if (buffer_bytes(b_) > 0)
|
||||
{
|
||||
ec = net::error::eof;
|
||||
}
|
||||
}
|
||||
|
||||
auto alloc = net::get_associated_allocator(h_);
|
||||
wg2_.get_executor().dispatch(
|
||||
beast::bind_front_handler(std::move(h_),
|
||||
ec, bytes_transferred), alloc);
|
||||
wg2_.reset();
|
||||
}
|
||||
};
|
||||
|
||||
lambda fn_;
|
||||
net::executor_work_guard<ex1_type> wg1_;
|
||||
|
||||
public:
|
||||
template<class Handler_>
|
||||
read_op(
|
||||
Handler_&& h,
|
||||
boost::shared_ptr<state> const& s,
|
||||
Buffers const& b)
|
||||
: fn_(std::forward<Handler_>(h), s, b)
|
||||
, wg1_(s->ioc.get_executor())
|
||||
{
|
||||
}
|
||||
|
||||
void
|
||||
operator()(error_code ec) override
|
||||
{
|
||||
|
||||
auto alloc = net::get_associated_allocator(fn_.h_);
|
||||
wg1_.get_executor().post(
|
||||
beast::bind_front_handler(std::move(fn_), ec), alloc);
|
||||
wg1_.reset();
|
||||
}
|
||||
};
|
||||
|
||||
struct stream::run_read_op
|
||||
{
|
||||
template<
|
||||
class ReadHandler,
|
||||
class MutableBufferSequence>
|
||||
void
|
||||
operator()(
|
||||
ReadHandler&& h,
|
||||
boost::shared_ptr<state> const& in,
|
||||
MutableBufferSequence const& buffers)
|
||||
{
|
||||
// If you get an error on the following line it means
|
||||
// that your handler does not meet the documented type
|
||||
// requirements for the handler.
|
||||
|
||||
static_assert(
|
||||
beast::detail::is_invocable<ReadHandler,
|
||||
void(error_code, std::size_t)>::value,
|
||||
"ReadHandler type requirements not met");
|
||||
|
||||
initiate_read(
|
||||
in,
|
||||
std::unique_ptr<read_op_base>{
|
||||
new read_op<
|
||||
typename std::decay<ReadHandler>::type,
|
||||
MutableBufferSequence>(
|
||||
std::move(h),
|
||||
in,
|
||||
buffers)},
|
||||
buffer_bytes(buffers));
|
||||
}
|
||||
};
|
||||
|
||||
struct stream::run_write_op
|
||||
{
|
||||
template<
|
||||
class WriteHandler,
|
||||
class ConstBufferSequence>
|
||||
void
|
||||
operator()(
|
||||
WriteHandler&& h,
|
||||
boost::shared_ptr<state> in_,
|
||||
boost::weak_ptr<state> out_,
|
||||
ConstBufferSequence const& buffers)
|
||||
{
|
||||
// If you get an error on the following line it means
|
||||
// that your handler does not meet the documented type
|
||||
// requirements for the handler.
|
||||
|
||||
static_assert(
|
||||
beast::detail::is_invocable<WriteHandler,
|
||||
void(error_code, std::size_t)>::value,
|
||||
"WriteHandler type requirements not met");
|
||||
|
||||
++in_->nwrite;
|
||||
auto const upcall = [&](error_code ec, std::size_t n)
|
||||
{
|
||||
net::post(
|
||||
in_->ioc.get_executor(),
|
||||
beast::bind_front_handler(std::move(h), ec, n));
|
||||
};
|
||||
|
||||
// test failure
|
||||
error_code ec;
|
||||
std::size_t n = 0;
|
||||
if(in_->fc && in_->fc->fail(ec))
|
||||
return upcall(ec, n);
|
||||
|
||||
// A request to write 0 bytes to a stream is a no-op.
|
||||
if(buffer_bytes(buffers) == 0)
|
||||
return upcall(ec, n);
|
||||
|
||||
// connection closed
|
||||
auto out = out_.lock();
|
||||
if(! out)
|
||||
return upcall(net::error::connection_reset, n);
|
||||
|
||||
// copy buffers
|
||||
n = std::min<std::size_t>(
|
||||
buffer_bytes(buffers), in_->write_max);
|
||||
{
|
||||
std::lock_guard<std::mutex> lock(out->m);
|
||||
n = net::buffer_copy(out->b.prepare(n), buffers);
|
||||
out->b.commit(n);
|
||||
out->notify_read();
|
||||
}
|
||||
BOOST_ASSERT(! ec);
|
||||
upcall(ec, n);
|
||||
}
|
||||
};
|
||||
|
||||
//------------------------------------------------------------------------------
|
||||
|
||||
template<class MutableBufferSequence>
|
||||
std::size_t
|
||||
stream::
|
||||
read_some(MutableBufferSequence const& buffers)
|
||||
{
|
||||
static_assert(net::is_mutable_buffer_sequence<
|
||||
MutableBufferSequence>::value,
|
||||
"MutableBufferSequence type requirements not met");
|
||||
error_code ec;
|
||||
auto const n = read_some(buffers, ec);
|
||||
if(ec)
|
||||
BOOST_THROW_EXCEPTION(system_error{ec});
|
||||
return n;
|
||||
}
|
||||
|
||||
template<class MutableBufferSequence>
|
||||
std::size_t
|
||||
stream::
|
||||
read_some(MutableBufferSequence const& buffers,
|
||||
error_code& ec)
|
||||
{
|
||||
static_assert(net::is_mutable_buffer_sequence<
|
||||
MutableBufferSequence>::value,
|
||||
"MutableBufferSequence type requirements not met");
|
||||
|
||||
++in_->nread;
|
||||
|
||||
// test failure
|
||||
if(in_->fc && in_->fc->fail(ec))
|
||||
return 0;
|
||||
|
||||
// A request to read 0 bytes from a stream is a no-op.
|
||||
if(buffer_bytes(buffers) == 0)
|
||||
{
|
||||
ec = {};
|
||||
return 0;
|
||||
}
|
||||
|
||||
std::unique_lock<std::mutex> lock{in_->m};
|
||||
BOOST_ASSERT(! in_->op);
|
||||
in_->cv.wait(lock,
|
||||
[&]()
|
||||
{
|
||||
return
|
||||
in_->b.size() > 0 ||
|
||||
in_->code != status::ok;
|
||||
});
|
||||
|
||||
// deliver bytes before eof
|
||||
if(in_->b.size() > 0)
|
||||
{
|
||||
auto const n = net::buffer_copy(
|
||||
buffers, in_->b.data(), in_->read_max);
|
||||
in_->b.consume(n);
|
||||
return n;
|
||||
}
|
||||
|
||||
// deliver error
|
||||
BOOST_ASSERT(in_->code != status::ok);
|
||||
ec = net::error::eof;
|
||||
return 0;
|
||||
}
|
||||
|
||||
template<class MutableBufferSequence, class ReadHandler>
|
||||
BOOST_BEAST_ASYNC_RESULT2(ReadHandler)
|
||||
stream::
|
||||
async_read_some(
|
||||
MutableBufferSequence const& buffers,
|
||||
ReadHandler&& handler)
|
||||
{
|
||||
static_assert(net::is_mutable_buffer_sequence<
|
||||
MutableBufferSequence>::value,
|
||||
"MutableBufferSequence type requirements not met");
|
||||
|
||||
return net::async_initiate<
|
||||
ReadHandler,
|
||||
void(error_code, std::size_t)>(
|
||||
run_read_op{},
|
||||
handler,
|
||||
in_,
|
||||
buffers);
|
||||
}
|
||||
|
||||
template<class ConstBufferSequence>
|
||||
std::size_t
|
||||
stream::
|
||||
write_some(ConstBufferSequence const& buffers)
|
||||
{
|
||||
static_assert(net::is_const_buffer_sequence<
|
||||
ConstBufferSequence>::value,
|
||||
"ConstBufferSequence type requirements not met");
|
||||
error_code ec;
|
||||
auto const bytes_transferred =
|
||||
write_some(buffers, ec);
|
||||
if(ec)
|
||||
BOOST_THROW_EXCEPTION(system_error{ec});
|
||||
return bytes_transferred;
|
||||
}
|
||||
|
||||
template<class ConstBufferSequence>
|
||||
std::size_t
|
||||
stream::
|
||||
write_some(
|
||||
ConstBufferSequence const& buffers, error_code& ec)
|
||||
{
|
||||
static_assert(net::is_const_buffer_sequence<
|
||||
ConstBufferSequence>::value,
|
||||
"ConstBufferSequence type requirements not met");
|
||||
|
||||
++in_->nwrite;
|
||||
|
||||
// test failure
|
||||
if(in_->fc && in_->fc->fail(ec))
|
||||
return 0;
|
||||
|
||||
// A request to write 0 bytes to a stream is a no-op.
|
||||
if(buffer_bytes(buffers) == 0)
|
||||
{
|
||||
ec = {};
|
||||
return 0;
|
||||
}
|
||||
|
||||
// connection closed
|
||||
auto out = out_.lock();
|
||||
if(! out)
|
||||
{
|
||||
ec = net::error::connection_reset;
|
||||
return 0;
|
||||
}
|
||||
|
||||
// copy buffers
|
||||
auto n = std::min<std::size_t>(
|
||||
buffer_bytes(buffers), in_->write_max);
|
||||
{
|
||||
std::lock_guard<std::mutex> lock(out->m);
|
||||
n = net::buffer_copy(out->b.prepare(n), buffers);
|
||||
out->b.commit(n);
|
||||
out->notify_read();
|
||||
}
|
||||
return n;
|
||||
}
|
||||
|
||||
template<class ConstBufferSequence, class WriteHandler>
|
||||
BOOST_BEAST_ASYNC_RESULT2(WriteHandler)
|
||||
stream::
|
||||
async_write_some(
|
||||
ConstBufferSequence const& buffers,
|
||||
WriteHandler&& handler)
|
||||
{
|
||||
static_assert(net::is_const_buffer_sequence<
|
||||
ConstBufferSequence>::value,
|
||||
"ConstBufferSequence type requirements not met");
|
||||
|
||||
return net::async_initiate<
|
||||
WriteHandler,
|
||||
void(error_code, std::size_t)>(
|
||||
run_write_op{},
|
||||
handler,
|
||||
in_,
|
||||
out_,
|
||||
buffers);
|
||||
}
|
||||
|
||||
//------------------------------------------------------------------------------
|
||||
|
||||
template<class TeardownHandler>
|
||||
void
|
||||
async_teardown(
|
||||
role_type,
|
||||
stream& s,
|
||||
TeardownHandler&& handler)
|
||||
{
|
||||
error_code ec;
|
||||
if( s.in_->fc &&
|
||||
s.in_->fc->fail(ec))
|
||||
return net::post(
|
||||
s.get_executor(),
|
||||
beast::bind_front_handler(
|
||||
std::move(handler), ec));
|
||||
s.close();
|
||||
if( s.in_->fc &&
|
||||
s.in_->fc->fail(ec))
|
||||
ec = net::error::eof;
|
||||
else
|
||||
ec = {};
|
||||
|
||||
net::post(
|
||||
s.get_executor(),
|
||||
beast::bind_front_handler(
|
||||
std::move(handler), ec));
|
||||
}
|
||||
|
||||
//------------------------------------------------------------------------------
|
||||
|
||||
template<class Arg1, class... ArgN>
|
||||
stream
|
||||
connect(stream& to, Arg1&& arg1, ArgN&&... argn)
|
||||
{
|
||||
stream from{
|
||||
std::forward<Arg1>(arg1),
|
||||
std::forward<ArgN>(argn)...};
|
||||
from.connect(to);
|
||||
return from;
|
||||
}
|
||||
|
||||
} // test
|
||||
} // beast
|
||||
} // boost
|
||||
|
||||
#endif
|
||||
375
macx64/include/boost/beast/_experimental/test/impl/stream.ipp
Normal file
375
macx64/include/boost/beast/_experimental/test/impl/stream.ipp
Normal file
@@ -0,0 +1,375 @@
|
||||
//
|
||||
// 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_TEST_IMPL_STREAM_IPP
|
||||
#define BOOST_BEAST_TEST_IMPL_STREAM_IPP
|
||||
|
||||
#include <boost/beast/_experimental/test/stream.hpp>
|
||||
#include <boost/beast/core/bind_handler.hpp>
|
||||
#include <boost/beast/core/buffer_traits.hpp>
|
||||
#include <boost/beast/core/buffers_prefix.hpp>
|
||||
#include <boost/make_shared.hpp>
|
||||
#include <stdexcept>
|
||||
#include <vector>
|
||||
|
||||
namespace boost {
|
||||
namespace beast {
|
||||
namespace test {
|
||||
|
||||
//------------------------------------------------------------------------------
|
||||
|
||||
stream::
|
||||
service::
|
||||
service(net::execution_context& ctx)
|
||||
: beast::detail::service_base<service>(ctx)
|
||||
, sp_(boost::make_shared<service_impl>())
|
||||
{
|
||||
}
|
||||
|
||||
void
|
||||
stream::
|
||||
service::
|
||||
shutdown()
|
||||
{
|
||||
std::vector<std::unique_ptr<read_op_base>> v;
|
||||
std::lock_guard<std::mutex> g1(sp_->m_);
|
||||
v.reserve(sp_->v_.size());
|
||||
for(auto p : sp_->v_)
|
||||
{
|
||||
std::lock_guard<std::mutex> g2(p->m);
|
||||
v.emplace_back(std::move(p->op));
|
||||
p->code = status::eof;
|
||||
}
|
||||
}
|
||||
|
||||
auto
|
||||
stream::
|
||||
service::
|
||||
make_impl(
|
||||
net::io_context& ctx,
|
||||
test::fail_count* fc) ->
|
||||
boost::shared_ptr<state>
|
||||
{
|
||||
auto& svc = net::use_service<service>(ctx);
|
||||
auto sp = boost::make_shared<state>(ctx, svc.sp_, fc);
|
||||
std::lock_guard<std::mutex> g(svc.sp_->m_);
|
||||
svc.sp_->v_.push_back(sp.get());
|
||||
return sp;
|
||||
}
|
||||
|
||||
void
|
||||
stream::
|
||||
service_impl::
|
||||
remove(state& impl)
|
||||
{
|
||||
std::lock_guard<std::mutex> g(m_);
|
||||
*std::find(
|
||||
v_.begin(), v_.end(),
|
||||
&impl) = std::move(v_.back());
|
||||
v_.pop_back();
|
||||
}
|
||||
|
||||
//------------------------------------------------------------------------------
|
||||
|
||||
void stream::initiate_read(
|
||||
boost::shared_ptr<state> const& in_,
|
||||
std::unique_ptr<stream::read_op_base>&& op,
|
||||
std::size_t buf_size)
|
||||
{
|
||||
std::unique_lock<std::mutex> lock(in_->m);
|
||||
|
||||
++in_->nread;
|
||||
if(in_->op != nullptr)
|
||||
BOOST_THROW_EXCEPTION(
|
||||
std::logic_error{"in_->op != nullptr"});
|
||||
|
||||
// test failure
|
||||
error_code ec;
|
||||
if(in_->fc && in_->fc->fail(ec))
|
||||
{
|
||||
lock.unlock();
|
||||
(*op)(ec);
|
||||
return;
|
||||
}
|
||||
|
||||
// A request to read 0 bytes from a stream is a no-op.
|
||||
if(buf_size == 0 || buffer_bytes(in_->b.data()) > 0)
|
||||
{
|
||||
lock.unlock();
|
||||
(*op)(ec);
|
||||
return;
|
||||
}
|
||||
|
||||
// deliver error
|
||||
if(in_->code != status::ok)
|
||||
{
|
||||
lock.unlock();
|
||||
(*op)(net::error::eof);
|
||||
return;
|
||||
}
|
||||
|
||||
// complete when bytes available or closed
|
||||
in_->op = std::move(op);
|
||||
}
|
||||
|
||||
stream::
|
||||
state::
|
||||
state(
|
||||
net::io_context& ioc_,
|
||||
boost::weak_ptr<service_impl> wp_,
|
||||
fail_count* fc_)
|
||||
: ioc(ioc_)
|
||||
, wp(std::move(wp_))
|
||||
, fc(fc_)
|
||||
{
|
||||
}
|
||||
|
||||
stream::
|
||||
state::
|
||||
~state()
|
||||
{
|
||||
// cancel outstanding read
|
||||
if(op != nullptr)
|
||||
(*op)(net::error::operation_aborted);
|
||||
}
|
||||
|
||||
void
|
||||
stream::
|
||||
state::
|
||||
remove() noexcept
|
||||
{
|
||||
auto sp = wp.lock();
|
||||
|
||||
// If this goes off, it means the lifetime of a test::stream object
|
||||
// extended beyond the lifetime of the associated execution context.
|
||||
BOOST_ASSERT(sp);
|
||||
|
||||
sp->remove(*this);
|
||||
}
|
||||
|
||||
void
|
||||
stream::
|
||||
state::
|
||||
notify_read()
|
||||
{
|
||||
if(op)
|
||||
{
|
||||
auto op_ = std::move(op);
|
||||
op_->operator()(error_code{});
|
||||
}
|
||||
else
|
||||
{
|
||||
cv.notify_all();
|
||||
}
|
||||
}
|
||||
|
||||
void
|
||||
stream::
|
||||
state::
|
||||
cancel_read()
|
||||
{
|
||||
std::unique_ptr<read_op_base> p;
|
||||
{
|
||||
std::lock_guard<std::mutex> lock(m);
|
||||
code = status::eof;
|
||||
p = std::move(op);
|
||||
}
|
||||
if(p != nullptr)
|
||||
(*p)(net::error::operation_aborted);
|
||||
}
|
||||
|
||||
//------------------------------------------------------------------------------
|
||||
|
||||
stream::
|
||||
~stream()
|
||||
{
|
||||
close();
|
||||
in_->remove();
|
||||
}
|
||||
|
||||
stream::
|
||||
stream(stream&& other)
|
||||
{
|
||||
auto in = service::make_impl(
|
||||
other.in_->ioc, other.in_->fc);
|
||||
in_ = std::move(other.in_);
|
||||
out_ = std::move(other.out_);
|
||||
other.in_ = in;
|
||||
}
|
||||
|
||||
stream&
|
||||
stream::
|
||||
operator=(stream&& other)
|
||||
{
|
||||
close();
|
||||
auto in = service::make_impl(
|
||||
other.in_->ioc, other.in_->fc);
|
||||
in_->remove();
|
||||
in_ = std::move(other.in_);
|
||||
out_ = std::move(other.out_);
|
||||
other.in_ = in;
|
||||
return *this;
|
||||
}
|
||||
|
||||
//------------------------------------------------------------------------------
|
||||
|
||||
stream::
|
||||
stream(net::io_context& ioc)
|
||||
: in_(service::make_impl(ioc, nullptr))
|
||||
{
|
||||
}
|
||||
|
||||
stream::
|
||||
stream(
|
||||
net::io_context& ioc,
|
||||
fail_count& fc)
|
||||
: in_(service::make_impl(ioc, &fc))
|
||||
{
|
||||
}
|
||||
|
||||
stream::
|
||||
stream(
|
||||
net::io_context& ioc,
|
||||
string_view s)
|
||||
: in_(service::make_impl(ioc, nullptr))
|
||||
{
|
||||
in_->b.commit(net::buffer_copy(
|
||||
in_->b.prepare(s.size()),
|
||||
net::buffer(s.data(), s.size())));
|
||||
}
|
||||
|
||||
stream::
|
||||
stream(
|
||||
net::io_context& ioc,
|
||||
fail_count& fc,
|
||||
string_view s)
|
||||
: in_(service::make_impl(ioc, &fc))
|
||||
{
|
||||
in_->b.commit(net::buffer_copy(
|
||||
in_->b.prepare(s.size()),
|
||||
net::buffer(s.data(), s.size())));
|
||||
}
|
||||
|
||||
void
|
||||
stream::
|
||||
connect(stream& remote)
|
||||
{
|
||||
BOOST_ASSERT(! out_.lock());
|
||||
BOOST_ASSERT(! remote.out_.lock());
|
||||
out_ = remote.in_;
|
||||
remote.out_ = in_;
|
||||
in_->code = status::ok;
|
||||
remote.in_->code = status::ok;
|
||||
}
|
||||
|
||||
string_view
|
||||
stream::
|
||||
str() const
|
||||
{
|
||||
auto const bs = in_->b.data();
|
||||
if(buffer_bytes(bs) == 0)
|
||||
return {};
|
||||
auto const b = beast::buffers_front(bs);
|
||||
return {static_cast<char const*>(b.data()), b.size()};
|
||||
}
|
||||
|
||||
void
|
||||
stream::
|
||||
append(string_view s)
|
||||
{
|
||||
std::lock_guard<std::mutex> lock{in_->m};
|
||||
in_->b.commit(net::buffer_copy(
|
||||
in_->b.prepare(s.size()),
|
||||
net::buffer(s.data(), s.size())));
|
||||
}
|
||||
|
||||
void
|
||||
stream::
|
||||
clear()
|
||||
{
|
||||
std::lock_guard<std::mutex> lock{in_->m};
|
||||
in_->b.consume(in_->b.size());
|
||||
}
|
||||
|
||||
void
|
||||
stream::
|
||||
close()
|
||||
{
|
||||
in_->cancel_read();
|
||||
|
||||
// disconnect
|
||||
{
|
||||
auto out = out_.lock();
|
||||
out_.reset();
|
||||
|
||||
// notify peer
|
||||
if(out)
|
||||
{
|
||||
std::lock_guard<std::mutex> lock(out->m);
|
||||
if(out->code == status::ok)
|
||||
{
|
||||
out->code = status::eof;
|
||||
out->notify_read();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void
|
||||
stream::
|
||||
close_remote()
|
||||
{
|
||||
std::lock_guard<std::mutex> lock{in_->m};
|
||||
if(in_->code == status::ok)
|
||||
{
|
||||
in_->code = status::eof;
|
||||
in_->notify_read();
|
||||
}
|
||||
}
|
||||
|
||||
void
|
||||
teardown(
|
||||
role_type,
|
||||
stream& s,
|
||||
boost::system::error_code& ec)
|
||||
{
|
||||
if( s.in_->fc &&
|
||||
s.in_->fc->fail(ec))
|
||||
return;
|
||||
|
||||
s.close();
|
||||
|
||||
if( s.in_->fc &&
|
||||
s.in_->fc->fail(ec))
|
||||
ec = net::error::eof;
|
||||
else
|
||||
ec = {};
|
||||
}
|
||||
|
||||
//------------------------------------------------------------------------------
|
||||
|
||||
stream
|
||||
connect(stream& to)
|
||||
{
|
||||
stream from{to.get_executor().context()};
|
||||
from.connect(to);
|
||||
return from;
|
||||
}
|
||||
|
||||
void
|
||||
connect(stream& s1, stream& s2)
|
||||
{
|
||||
s1.connect(s2);
|
||||
}
|
||||
|
||||
} // test
|
||||
} // beast
|
||||
} // boost
|
||||
|
||||
#endif
|
||||
607
macx64/include/boost/beast/_experimental/test/stream.hpp
Normal file
607
macx64/include/boost/beast/_experimental/test/stream.hpp
Normal file
@@ -0,0 +1,607 @@
|
||||
//
|
||||
// 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_TEST_STREAM_HPP
|
||||
#define BOOST_BEAST_TEST_STREAM_HPP
|
||||
|
||||
#include <boost/beast/core/detail/config.hpp>
|
||||
#include <boost/beast/core/bind_handler.hpp>
|
||||
#include <boost/beast/core/flat_buffer.hpp>
|
||||
#include <boost/beast/core/role.hpp>
|
||||
#include <boost/beast/core/string.hpp>
|
||||
#include <boost/beast/_experimental/test/fail_count.hpp>
|
||||
#include <boost/asio/async_result.hpp>
|
||||
#include <boost/asio/buffer.hpp>
|
||||
#include <boost/asio/error.hpp>
|
||||
#include <boost/asio/executor_work_guard.hpp>
|
||||
#include <boost/asio/io_context.hpp>
|
||||
#include <boost/asio/post.hpp>
|
||||
#include <boost/assert.hpp>
|
||||
#include <boost/shared_ptr.hpp>
|
||||
#include <boost/weak_ptr.hpp>
|
||||
#include <boost/throw_exception.hpp>
|
||||
#include <condition_variable>
|
||||
#include <limits>
|
||||
#include <memory>
|
||||
#include <mutex>
|
||||
#include <utility>
|
||||
|
||||
#if ! BOOST_BEAST_DOXYGEN
|
||||
namespace boost {
|
||||
namespace asio {
|
||||
namespace ssl {
|
||||
template<typename> class stream;
|
||||
} // ssl
|
||||
} // asio
|
||||
} // boost
|
||||
#endif
|
||||
|
||||
namespace boost {
|
||||
namespace beast {
|
||||
namespace test {
|
||||
|
||||
/** A two-way socket useful for unit testing
|
||||
|
||||
An instance of this class simulates a traditional socket,
|
||||
while also providing features useful for unit testing.
|
||||
Each endpoint maintains an independent buffer called
|
||||
the input area. Writes from one endpoint append data
|
||||
to the peer's pending input area. When an endpoint performs
|
||||
a read and data is present in the input area, the data is
|
||||
delivered to the blocking or asynchronous operation. Otherwise
|
||||
the operation is blocked or deferred until data is made
|
||||
available, or until the endpoints become disconnected.
|
||||
|
||||
These streams may be used anywhere an algorithm accepts a
|
||||
reference to a synchronous or asynchronous read or write
|
||||
stream. It is possible to use a test stream in a call to
|
||||
`net::read_until`, or in a call to
|
||||
@ref boost::beast::http::async_write for example.
|
||||
|
||||
As with Boost.Asio I/O objects, a @ref stream constructs
|
||||
with a reference to the `net::io_context` to use for
|
||||
handling asynchronous I/O. For asynchronous operations, the
|
||||
stream follows the same rules as a traditional asio socket
|
||||
with respect to how completion handlers for asynchronous
|
||||
operations are performed.
|
||||
|
||||
To facilitate testing, these streams support some additional
|
||||
features:
|
||||
|
||||
@li The input area, represented by a @ref beast::basic_flat_buffer,
|
||||
may be directly accessed by the caller to inspect the contents
|
||||
before or after the remote endpoint writes data. This allows
|
||||
a unit test to verify that the received data matches.
|
||||
|
||||
@li Data may be manually appended to the input area. This data
|
||||
will delivered in the next call to
|
||||
@ref stream::read_some or @ref stream::async_read_some.
|
||||
This allows predefined test vectors to be set up for testing
|
||||
read algorithms.
|
||||
|
||||
@li The stream may be constructed with a fail count. The
|
||||
stream will eventually fail with a predefined error after a
|
||||
certain number of operations, where the number of operations
|
||||
is controlled by the test. When a test loops over a range of
|
||||
operation counts, it is possible to exercise every possible
|
||||
point of failure in the algorithm being tested. When used
|
||||
correctly the technique allows the tests to reach a high
|
||||
percentage of code coverage.
|
||||
|
||||
@par Thread Safety
|
||||
@e Distinct @e objects: Safe.@n
|
||||
@e Shared @e objects: Unsafe.
|
||||
The application must also ensure that all asynchronous
|
||||
operations are performed within the same implicit or explicit strand.
|
||||
|
||||
@par Concepts
|
||||
@li <em>SyncReadStream</em>
|
||||
@li <em>SyncWriteStream</em>
|
||||
@li <em>AsyncReadStream</em>
|
||||
@li <em>AsyncWriteStream</em>
|
||||
*/
|
||||
class stream
|
||||
{
|
||||
struct state;
|
||||
|
||||
boost::shared_ptr<state> in_;
|
||||
boost::weak_ptr<state> out_;
|
||||
|
||||
enum class status
|
||||
{
|
||||
ok,
|
||||
eof,
|
||||
};
|
||||
|
||||
class service;
|
||||
struct service_impl;
|
||||
|
||||
struct read_op_base
|
||||
{
|
||||
virtual ~read_op_base() = default;
|
||||
virtual void operator()(error_code ec) = 0;
|
||||
};
|
||||
|
||||
struct state
|
||||
{
|
||||
friend class stream;
|
||||
|
||||
net::io_context& ioc;
|
||||
boost::weak_ptr<service_impl> wp;
|
||||
std::mutex m;
|
||||
flat_buffer b;
|
||||
std::condition_variable cv;
|
||||
std::unique_ptr<read_op_base> op;
|
||||
status code = status::ok;
|
||||
fail_count* fc = nullptr;
|
||||
std::size_t nread = 0;
|
||||
std::size_t nwrite = 0;
|
||||
std::size_t read_max =
|
||||
(std::numeric_limits<std::size_t>::max)();
|
||||
std::size_t write_max =
|
||||
(std::numeric_limits<std::size_t>::max)();
|
||||
|
||||
BOOST_BEAST_DECL
|
||||
state(
|
||||
net::io_context& ioc_,
|
||||
boost::weak_ptr<service_impl> wp_,
|
||||
fail_count* fc_);
|
||||
|
||||
|
||||
BOOST_BEAST_DECL
|
||||
~state();
|
||||
|
||||
BOOST_BEAST_DECL
|
||||
void
|
||||
remove() noexcept;
|
||||
|
||||
BOOST_BEAST_DECL
|
||||
void
|
||||
notify_read();
|
||||
|
||||
BOOST_BEAST_DECL
|
||||
void
|
||||
cancel_read();
|
||||
};
|
||||
|
||||
template<class Handler, class Buffers>
|
||||
class read_op;
|
||||
|
||||
struct run_read_op;
|
||||
struct run_write_op;
|
||||
|
||||
BOOST_BEAST_DECL
|
||||
static
|
||||
void
|
||||
initiate_read(
|
||||
boost::shared_ptr<state> const& in,
|
||||
std::unique_ptr<read_op_base>&& op,
|
||||
std::size_t buf_size);
|
||||
|
||||
#if ! BOOST_BEAST_DOXYGEN
|
||||
// boost::asio::ssl::stream needs these
|
||||
// DEPRECATED
|
||||
template<class>
|
||||
friend class boost::asio::ssl::stream;
|
||||
// DEPRECATED
|
||||
using lowest_layer_type = stream;
|
||||
// DEPRECATED
|
||||
lowest_layer_type&
|
||||
lowest_layer() noexcept
|
||||
{
|
||||
return *this;
|
||||
}
|
||||
// DEPRECATED
|
||||
lowest_layer_type const&
|
||||
lowest_layer() const noexcept
|
||||
{
|
||||
return *this;
|
||||
}
|
||||
#endif
|
||||
|
||||
public:
|
||||
using buffer_type = flat_buffer;
|
||||
|
||||
/** Destructor
|
||||
|
||||
If an asynchronous read operation is pending, it will
|
||||
simply be discarded with no notification to the completion
|
||||
handler.
|
||||
|
||||
If a connection is established while the stream is destroyed,
|
||||
the peer will see the error `net::error::connection_reset`
|
||||
when performing any reads or writes.
|
||||
*/
|
||||
BOOST_BEAST_DECL
|
||||
~stream();
|
||||
|
||||
/** Move Constructor
|
||||
|
||||
Moving the stream while asynchronous operations are pending
|
||||
results in undefined behavior.
|
||||
*/
|
||||
BOOST_BEAST_DECL
|
||||
stream(stream&& other);
|
||||
|
||||
/** Move Assignment
|
||||
|
||||
Moving the stream while asynchronous operations are pending
|
||||
results in undefined behavior.
|
||||
*/
|
||||
BOOST_BEAST_DECL
|
||||
stream&
|
||||
operator=(stream&& other);
|
||||
|
||||
/** Construct a stream
|
||||
|
||||
The stream will be created in a disconnected state.
|
||||
|
||||
@param ioc The `io_context` object that the stream will use to
|
||||
dispatch handlers for any asynchronous operations.
|
||||
*/
|
||||
BOOST_BEAST_DECL
|
||||
explicit
|
||||
stream(net::io_context& ioc);
|
||||
|
||||
/** Construct a stream
|
||||
|
||||
The stream will be created in a disconnected state.
|
||||
|
||||
@param ioc The `io_context` object that the stream will use to
|
||||
dispatch handlers for any asynchronous operations.
|
||||
|
||||
@param fc The @ref fail_count to associate with the stream.
|
||||
Each I/O operation performed on the stream will increment the
|
||||
fail count. When the fail count reaches its internal limit,
|
||||
a simulated failure error will be raised.
|
||||
*/
|
||||
BOOST_BEAST_DECL
|
||||
stream(
|
||||
net::io_context& ioc,
|
||||
fail_count& fc);
|
||||
|
||||
/** Construct a stream
|
||||
|
||||
The stream will be created in a disconnected state.
|
||||
|
||||
@param ioc The `io_context` object that the stream will use to
|
||||
dispatch handlers for any asynchronous operations.
|
||||
|
||||
@param s A string which will be appended to the input area, not
|
||||
including the null terminator.
|
||||
*/
|
||||
BOOST_BEAST_DECL
|
||||
stream(
|
||||
net::io_context& ioc,
|
||||
string_view s);
|
||||
|
||||
/** Construct a stream
|
||||
|
||||
The stream will be created in a disconnected state.
|
||||
|
||||
@param ioc The `io_context` object that the stream will use to
|
||||
dispatch handlers for any asynchronous operations.
|
||||
|
||||
@param fc The @ref fail_count to associate with the stream.
|
||||
Each I/O operation performed on the stream will increment the
|
||||
fail count. When the fail count reaches its internal limit,
|
||||
a simulated failure error will be raised.
|
||||
|
||||
@param s A string which will be appended to the input area, not
|
||||
including the null terminator.
|
||||
*/
|
||||
BOOST_BEAST_DECL
|
||||
stream(
|
||||
net::io_context& ioc,
|
||||
fail_count& fc,
|
||||
string_view s);
|
||||
|
||||
/// Establish a connection
|
||||
BOOST_BEAST_DECL
|
||||
void
|
||||
connect(stream& remote);
|
||||
|
||||
/// The type of the executor associated with the object.
|
||||
using executor_type =
|
||||
net::io_context::executor_type;
|
||||
|
||||
/// Return the executor associated with the object.
|
||||
executor_type
|
||||
get_executor() noexcept
|
||||
{
|
||||
return in_->ioc.get_executor();
|
||||
};
|
||||
|
||||
/// Set the maximum number of bytes returned by read_some
|
||||
void
|
||||
read_size(std::size_t n) noexcept
|
||||
{
|
||||
in_->read_max = n;
|
||||
}
|
||||
|
||||
/// Set the maximum number of bytes returned by write_some
|
||||
void
|
||||
write_size(std::size_t n) noexcept
|
||||
{
|
||||
in_->write_max = n;
|
||||
}
|
||||
|
||||
/// Direct input buffer access
|
||||
buffer_type&
|
||||
buffer() noexcept
|
||||
{
|
||||
return in_->b;
|
||||
}
|
||||
|
||||
/// Returns a string view representing the pending input data
|
||||
BOOST_BEAST_DECL
|
||||
string_view
|
||||
str() const;
|
||||
|
||||
/// Appends a string to the pending input data
|
||||
BOOST_BEAST_DECL
|
||||
void
|
||||
append(string_view s);
|
||||
|
||||
/// Clear the pending input area
|
||||
BOOST_BEAST_DECL
|
||||
void
|
||||
clear();
|
||||
|
||||
/// Return the number of reads
|
||||
std::size_t
|
||||
nread() const noexcept
|
||||
{
|
||||
return in_->nread;
|
||||
}
|
||||
|
||||
/// Return the number of writes
|
||||
std::size_t
|
||||
nwrite() const noexcept
|
||||
{
|
||||
return in_->nwrite;
|
||||
}
|
||||
|
||||
/** Close the stream.
|
||||
|
||||
The other end of the connection will see
|
||||
`error::eof` after reading all the remaining data.
|
||||
*/
|
||||
BOOST_BEAST_DECL
|
||||
void
|
||||
close();
|
||||
|
||||
/** Close the other end of the stream.
|
||||
|
||||
This end of the connection will see
|
||||
`error::eof` after reading all the remaining data.
|
||||
*/
|
||||
BOOST_BEAST_DECL
|
||||
void
|
||||
close_remote();
|
||||
|
||||
/** Read some data from the stream.
|
||||
|
||||
This function is used to read data from the stream. The function call will
|
||||
block until one or more bytes of data has been read successfully, or until
|
||||
an error occurs.
|
||||
|
||||
@param buffers The buffers into which the data will be read.
|
||||
|
||||
@returns The number of bytes read.
|
||||
|
||||
@throws boost::system::system_error Thrown on failure.
|
||||
|
||||
@note The `read_some` operation may not read all of the requested number of
|
||||
bytes. Consider using the function `net::read` if you need to ensure
|
||||
that the requested amount of data is read before the blocking operation
|
||||
completes.
|
||||
*/
|
||||
template<class MutableBufferSequence>
|
||||
std::size_t
|
||||
read_some(MutableBufferSequence const& buffers);
|
||||
|
||||
/** Read some data from the stream.
|
||||
|
||||
This function is used to read data from the stream. The function call will
|
||||
block until one or more bytes of data has been read successfully, or until
|
||||
an error occurs.
|
||||
|
||||
@param buffers The buffers into which the data will be read.
|
||||
|
||||
@param ec Set to indicate what error occurred, if any.
|
||||
|
||||
@returns The number of bytes read.
|
||||
|
||||
@note The `read_some` operation may not read all of the requested number of
|
||||
bytes. Consider using the function `net::read` if you need to ensure
|
||||
that the requested amount of data is read before the blocking operation
|
||||
completes.
|
||||
*/
|
||||
template<class MutableBufferSequence>
|
||||
std::size_t
|
||||
read_some(MutableBufferSequence const& buffers,
|
||||
error_code& ec);
|
||||
|
||||
/** Start an asynchronous read.
|
||||
|
||||
This function is used to asynchronously read one or more bytes of data from
|
||||
the stream. The function call always returns immediately.
|
||||
|
||||
@param buffers The buffers into which the data will be read. Although the
|
||||
buffers object may be copied as necessary, ownership of the underlying
|
||||
buffers is retained by the caller, which must guarantee that they remain
|
||||
valid until the handler is called.
|
||||
|
||||
@param handler The completion handler to invoke when the operation
|
||||
completes. The implementation takes ownership of the handler by
|
||||
performing a decay-copy. The equivalent function signature of
|
||||
the handler must be:
|
||||
@code
|
||||
void handler(
|
||||
error_code const& ec, // Result of operation.
|
||||
std::size_t bytes_transferred // Number of bytes read.
|
||||
);
|
||||
@endcode
|
||||
Regardless of whether the asynchronous operation completes
|
||||
immediately or not, the handler will not be invoked from within
|
||||
this function. Invocation of the handler will be performed in a
|
||||
manner equivalent to using `net::post`.
|
||||
|
||||
@note The `async_read_some` operation may not read all of the requested number of
|
||||
bytes. Consider using the function `net::async_read` if you need
|
||||
to ensure that the requested amount of data is read before the asynchronous
|
||||
operation completes.
|
||||
*/
|
||||
template<class MutableBufferSequence, class ReadHandler>
|
||||
BOOST_BEAST_ASYNC_RESULT2(ReadHandler)
|
||||
async_read_some(MutableBufferSequence const& buffers,
|
||||
ReadHandler&& handler);
|
||||
|
||||
/** Write some data to the stream.
|
||||
|
||||
This function is used to write data on the stream. The function call will
|
||||
block until one or more bytes of data has been written successfully, or
|
||||
until an error occurs.
|
||||
|
||||
@param buffers The data to be written.
|
||||
|
||||
@returns The number of bytes written.
|
||||
|
||||
@throws boost::system::system_error Thrown on failure.
|
||||
|
||||
@note The `write_some` operation may not transmit all of the data to the
|
||||
peer. Consider using the function `net::write` if you need to
|
||||
ensure that all data is written before the blocking operation completes.
|
||||
*/
|
||||
template<class ConstBufferSequence>
|
||||
std::size_t
|
||||
write_some(ConstBufferSequence const& buffers);
|
||||
|
||||
/** Write some data to the stream.
|
||||
|
||||
This function is used to write data on the stream. The function call will
|
||||
block until one or more bytes of data has been written successfully, or
|
||||
until an error occurs.
|
||||
|
||||
@param buffers The data to be written.
|
||||
|
||||
@param ec Set to indicate what error occurred, if any.
|
||||
|
||||
@returns The number of bytes written.
|
||||
|
||||
@note The `write_some` operation may not transmit all of the data to the
|
||||
peer. Consider using the function `net::write` if you need to
|
||||
ensure that all data is written before the blocking operation completes.
|
||||
*/
|
||||
template<class ConstBufferSequence>
|
||||
std::size_t
|
||||
write_some(
|
||||
ConstBufferSequence const& buffers, error_code& ec);
|
||||
|
||||
/** Start an asynchronous write.
|
||||
|
||||
This function is used to asynchronously write one or more bytes of data to
|
||||
the stream. The function call always returns immediately.
|
||||
|
||||
@param buffers The data to be written to the stream. Although the buffers
|
||||
object may be copied as necessary, ownership of the underlying buffers is
|
||||
retained by the caller, which must guarantee that they remain valid until
|
||||
the handler is called.
|
||||
|
||||
@param handler The completion handler to invoke when the operation
|
||||
completes. The implementation takes ownership of the handler by
|
||||
performing a decay-copy. The equivalent function signature of
|
||||
the handler must be:
|
||||
@code
|
||||
void handler(
|
||||
error_code const& ec, // Result of operation.
|
||||
std::size_t bytes_transferred // Number of bytes written.
|
||||
);
|
||||
@endcode
|
||||
Regardless of whether the asynchronous operation completes
|
||||
immediately or not, the handler will not be invoked from within
|
||||
this function. Invocation of the handler will be performed in a
|
||||
manner equivalent to using `net::post`.
|
||||
|
||||
@note The `async_write_some` operation may not transmit all of the data to
|
||||
the peer. Consider using the function `net::async_write` if you need
|
||||
to ensure that all data is written before the asynchronous operation completes.
|
||||
*/
|
||||
template<class ConstBufferSequence, class WriteHandler>
|
||||
BOOST_BEAST_ASYNC_RESULT2(WriteHandler)
|
||||
async_write_some(ConstBufferSequence const& buffers,
|
||||
WriteHandler&& handler);
|
||||
|
||||
#if ! BOOST_BEAST_DOXYGEN
|
||||
friend
|
||||
BOOST_BEAST_DECL
|
||||
void
|
||||
teardown(
|
||||
role_type,
|
||||
stream& s,
|
||||
boost::system::error_code& ec);
|
||||
|
||||
template<class TeardownHandler>
|
||||
friend
|
||||
BOOST_BEAST_DECL
|
||||
void
|
||||
async_teardown(
|
||||
role_type role,
|
||||
stream& s,
|
||||
TeardownHandler&& handler);
|
||||
#endif
|
||||
};
|
||||
|
||||
#if ! BOOST_BEAST_DOXYGEN
|
||||
inline
|
||||
void
|
||||
beast_close_socket(stream& s)
|
||||
{
|
||||
s.close();
|
||||
}
|
||||
#endif
|
||||
|
||||
#if BOOST_BEAST_DOXYGEN
|
||||
/** Return a new stream connected to the given stream
|
||||
|
||||
@param to The stream to connect to.
|
||||
|
||||
@param args Optional arguments forwarded to the new stream's constructor.
|
||||
|
||||
@return The new, connected stream.
|
||||
*/
|
||||
template<class... Args>
|
||||
stream
|
||||
connect(stream& to, Args&&... args);
|
||||
|
||||
#else
|
||||
BOOST_BEAST_DECL
|
||||
stream
|
||||
connect(stream& to);
|
||||
|
||||
BOOST_BEAST_DECL
|
||||
void
|
||||
connect(stream& s1, stream& s2);
|
||||
|
||||
template<class Arg1, class... ArgN>
|
||||
stream
|
||||
connect(stream& to, Arg1&& arg1, ArgN&&... argn);
|
||||
#endif
|
||||
|
||||
} // test
|
||||
} // beast
|
||||
} // boost
|
||||
|
||||
#include <boost/beast/_experimental/test/impl/stream.hpp>
|
||||
#ifdef BOOST_BEAST_HEADER_ONLY
|
||||
#include <boost/beast/_experimental/test/impl/stream.ipp>
|
||||
#endif
|
||||
|
||||
#endif
|
||||
77
macx64/include/boost/beast/_experimental/test/tcp.hpp
Normal file
77
macx64/include/boost/beast/_experimental/test/tcp.hpp
Normal file
@@ -0,0 +1,77 @@
|
||||
//
|
||||
// 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_TEST_TCP_HPP
|
||||
#define BOOST_BEAST_TEST_TCP_HPP
|
||||
|
||||
#include <boost/beast/core/detail/config.hpp>
|
||||
#include <boost/beast/core/detail/get_io_context.hpp>
|
||||
#include <boost/beast/_experimental/unit_test/suite.hpp>
|
||||
#include <boost/beast/_experimental/test/handler.hpp>
|
||||
#include <boost/asio/ip/tcp.hpp>
|
||||
#include <chrono>
|
||||
|
||||
namespace boost {
|
||||
namespace beast {
|
||||
namespace test {
|
||||
|
||||
/** Connect two TCP sockets together.
|
||||
*/
|
||||
template<class Executor>
|
||||
bool
|
||||
connect(
|
||||
net::basic_stream_socket<net::ip::tcp, Executor>& s1,
|
||||
net::basic_stream_socket<net::ip::tcp, Executor>& s2)
|
||||
|
||||
{
|
||||
auto ioc1 = beast::detail::get_io_context(s1);
|
||||
auto ioc2 = beast::detail::get_io_context(s2);
|
||||
if(! BEAST_EXPECT(ioc1 != nullptr))
|
||||
return false;
|
||||
if(! BEAST_EXPECT(ioc2 != nullptr))
|
||||
return false;
|
||||
if(! BEAST_EXPECT(ioc1 == ioc2))
|
||||
return false;
|
||||
auto& ioc = *ioc1;
|
||||
try
|
||||
{
|
||||
net::basic_socket_acceptor<
|
||||
net::ip::tcp, Executor> a(s1.get_executor());
|
||||
auto ep = net::ip::tcp::endpoint(
|
||||
net::ip::make_address_v4("127.0.0.1"), 0);
|
||||
a.open(ep.protocol());
|
||||
a.set_option(
|
||||
net::socket_base::reuse_address(true));
|
||||
a.bind(ep);
|
||||
a.listen(0);
|
||||
ep = a.local_endpoint();
|
||||
a.async_accept(s2, test::success_handler());
|
||||
s1.async_connect(ep, test::success_handler());
|
||||
run(ioc);
|
||||
if(! BEAST_EXPECT(
|
||||
s1.remote_endpoint() == s2.local_endpoint()))
|
||||
return false;
|
||||
if(! BEAST_EXPECT(
|
||||
s2.remote_endpoint() == s1.local_endpoint()))
|
||||
return false;
|
||||
}
|
||||
catch(std::exception const& e)
|
||||
{
|
||||
beast::unit_test::suite::this_suite()->fail(
|
||||
e.what(), __FILE__, __LINE__);
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
} // test
|
||||
} // beast
|
||||
} // boost
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user