diff options
author | DongHun Kwak <dh0128.kwak@samsung.com> | 2019-12-05 15:11:01 +0900 |
---|---|---|
committer | DongHun Kwak <dh0128.kwak@samsung.com> | 2019-12-05 15:11:01 +0900 |
commit | 3fdc3e5ee96dca5b11d1694975a65200787eab86 (patch) | |
tree | 5c1733853892b8397d67706fa453a9bd978d2102 /boost/beast/core/impl/read_size.ipp | |
parent | 88e602c57797660ebe0f9e15dbd64c1ff16dead3 (diff) | |
download | boost-3fdc3e5ee96dca5b11d1694975a65200787eab86.tar.gz boost-3fdc3e5ee96dca5b11d1694975a65200787eab86.tar.bz2 boost-3fdc3e5ee96dca5b11d1694975a65200787eab86.zip |
Imported Upstream version 1.66.0upstream/1.66.0
Diffstat (limited to 'boost/beast/core/impl/read_size.ipp')
-rw-r--r-- | boost/beast/core/impl/read_size.ipp | 80 |
1 files changed, 80 insertions, 0 deletions
diff --git a/boost/beast/core/impl/read_size.ipp b/boost/beast/core/impl/read_size.ipp new file mode 100644 index 0000000000..ddcaf397a9 --- /dev/null +++ b/boost/beast/core/impl/read_size.ipp @@ -0,0 +1,80 @@ +// +// 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_IMPL_READ_SIZE_IPP +#define BOOST_BEAST_IMPL_READ_SIZE_IPP + +namespace boost { +namespace beast { + +namespace detail { + +template<class T, class = void> +struct has_read_size_helper : std::false_type {}; + +template<class T> +struct has_read_size_helper<T, decltype( + read_size_helper(std::declval<T&>(), 512), + (void)0)> : std::true_type +{ +}; + +template<class DynamicBuffer> +std::size_t +read_size(DynamicBuffer& buffer, + std::size_t max_size, std::true_type) +{ + return read_size_helper(buffer, max_size); +} + +template<class DynamicBuffer> +std::size_t +read_size(DynamicBuffer& buffer, + std::size_t max_size, std::false_type) +{ + static_assert( + boost::asio::is_dynamic_buffer<DynamicBuffer>::value, + "DynamicBuffer requirements not met"); + BOOST_ASSERT(max_size >= 1); + auto const size = buffer.size(); + auto const limit = buffer.max_size() - size; + BOOST_ASSERT(size <= buffer.max_size()); + return (std::min<std::size_t>)( + (std::max<std::size_t>)(512, buffer.capacity() - size), + (std::min<std::size_t>)(max_size, limit)); +} + +} // detail + +template<class DynamicBuffer> +inline +std::size_t +read_size( + DynamicBuffer& buffer, std::size_t max_size) +{ + return detail::read_size(buffer, max_size, + detail::has_read_size_helper<DynamicBuffer>{}); +} + +template<class DynamicBuffer> +std::size_t +read_size_or_throw( + DynamicBuffer& buffer, std::size_t max_size) +{ + auto const n = read_size(buffer, max_size); + if(n == 0) + BOOST_THROW_EXCEPTION(std::length_error{ + "buffer overflow"}); + return n; +} + +} // beast +} // boost + +#endif |