forked from boostorg/beast
108 lines
2.9 KiB
C++
108 lines
2.9 KiB
C++
//
|
|
// Copyright (c) 2016-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)
|
|
//
|
|
// Official repository: https://github.com/boostorg/beast
|
|
//
|
|
|
|
#ifndef BOOST_BEAST_TEST_BUFFER_TEST_HPP
|
|
#define BOOST_BEAST_TEST_BUFFER_TEST_HPP
|
|
|
|
#include <boost/beast/core/buffers_to_string.hpp>
|
|
#include <boost/beast/core/string.hpp>
|
|
#include <boost/beast/core/read_size.hpp>
|
|
#include <boost/beast/core/type_traits.hpp>
|
|
#include <boost/beast/core/detail/type_traits.hpp>
|
|
#include <boost/asio/buffer.hpp>
|
|
#include <algorithm>
|
|
#include <string>
|
|
#include <type_traits>
|
|
|
|
namespace boost {
|
|
namespace beast {
|
|
namespace test {
|
|
|
|
template<class DynamicBuffer>
|
|
void
|
|
write_buffer(DynamicBuffer& b, string_view s)
|
|
{
|
|
b.commit(boost::asio::buffer_copy(
|
|
b.prepare(s.size()), boost::asio::buffer(
|
|
s.data(), s.size())));
|
|
}
|
|
|
|
template<class ConstBufferSequence>
|
|
typename std::enable_if<
|
|
boost::asio::is_const_buffer_sequence<ConstBufferSequence>::value,
|
|
std::size_t>::type
|
|
buffer_count(ConstBufferSequence const& buffers)
|
|
{
|
|
return std::distance(buffers.begin(), buffers.end());
|
|
}
|
|
|
|
template<class ConstBufferSequence>
|
|
typename std::enable_if<
|
|
boost::asio::is_const_buffer_sequence<ConstBufferSequence>::value,
|
|
std::size_t>::type
|
|
size_pre(ConstBufferSequence const& buffers)
|
|
{
|
|
std::size_t n = 0;
|
|
for(auto it = buffers.begin(); it != buffers.end(); ++it)
|
|
{
|
|
typename ConstBufferSequence::const_iterator it0(std::move(it));
|
|
typename ConstBufferSequence::const_iterator it1(it0);
|
|
typename ConstBufferSequence::const_iterator it2;
|
|
it2 = it1;
|
|
n += boost::asio::buffer_size(*it2);
|
|
it = std::move(it2);
|
|
}
|
|
return n;
|
|
}
|
|
|
|
template<class ConstBufferSequence>
|
|
typename std::enable_if<
|
|
boost::asio::is_const_buffer_sequence<ConstBufferSequence>::value,
|
|
std::size_t>::type
|
|
size_post(ConstBufferSequence const& buffers)
|
|
{
|
|
std::size_t n = 0;
|
|
for(auto it = buffers.begin(); it != buffers.end(); it++)
|
|
n += boost::asio::buffer_size(*it);
|
|
return n;
|
|
}
|
|
|
|
template<class ConstBufferSequence>
|
|
typename std::enable_if<
|
|
boost::asio::is_const_buffer_sequence<ConstBufferSequence>::value,
|
|
std::size_t>::type
|
|
size_rev_pre(ConstBufferSequence const& buffers)
|
|
{
|
|
std::size_t n = 0;
|
|
for(auto it = buffers.end(); it != buffers.begin();)
|
|
n += boost::asio::buffer_size(*--it);
|
|
return n;
|
|
}
|
|
|
|
template<class ConstBufferSequence>
|
|
typename std::enable_if<
|
|
boost::asio::is_const_buffer_sequence<ConstBufferSequence>::value,
|
|
std::size_t>::type
|
|
size_rev_post(ConstBufferSequence const& buffers)
|
|
{
|
|
std::size_t n = 0;
|
|
for(auto it = buffers.end(); it != buffers.begin();)
|
|
{
|
|
it--;
|
|
n += boost::asio::buffer_size(*it);
|
|
}
|
|
return n;
|
|
}
|
|
|
|
} // test
|
|
} // beast
|
|
} // boost
|
|
|
|
#endif
|