Line data    Source code

       1             : //
       2             : // Copyright (c) 2019 Vinnie Falco (vinnie.falco@gmail.com)
       3             : //
       4             : // Distributed under the Boost Software License, Version 1.0. (See accompanying
       5             : // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
       6             : //
       7             : // Official repository: https://github.com/cppalliance/http_proto
       8             : //
       9             : 
      10             : #include <boost/http_proto/buffered_base.hpp>
      11             : #include <boost/http_proto/detail/except.hpp>
      12             : 
      13             : namespace boost {
      14             : namespace http_proto {
      15             : 
      16             : buffered_base::
      17             : ~buffered_base() = default;
      18             : 
      19             : void
      20           8 : buffered_base::
      21             : on_init(allocator&)
      22             : {
      23           8 : }
      24             : 
      25             : void
      26           8 : buffered_base::
      27             : init(
      28             :     allocator& a,
      29             :     std::size_t max_size)
      30             : {
      31             :     // max_size exceeds limit
      32           8 :     if(max_size > a.max_size())
      33           1 :         detail::throw_invalid_argument();
      34             : 
      35             :     struct restorer
      36             :     {
      37             :         allocator& a;
      38             :         std::size_t n;
      39             : 
      40           7 :         ~restorer()
      41           7 :         {
      42           7 :             a.restore(n);
      43           7 :         }
      44             :     };
      45             : 
      46             :     auto const n =
      47           7 :         a.max_size() - max_size;
      48           7 :     a.remove(n);
      49           7 :     restorer r{a, n};
      50           7 :     init(a);
      51           5 : }
      52             : 
      53             : //------------------------------------------------
      54             : 
      55             : void*
      56          10 : buffered_base::
      57             : allocator::
      58             : allocate(
      59             :     std::size_t n)
      60             : {
      61             :     // n exceeds available space
      62          10 :     if(n > size_)
      63           3 :         detail::throw_invalid_argument();
      64             : 
      65           7 :     size_used_ += n;
      66           7 :     if(down_)
      67             :     {
      68           3 :         auto p = base_ + size_ - n;
      69           3 :         size_ -= n;
      70           3 :         return p;
      71             :     }
      72           4 :     auto p = base_;
      73           4 :     base_ += n;
      74           4 :     size_ -= n;
      75           4 :     return p;
      76             : }
      77             : 
      78             : } // http_proto
      79             : } // boost