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/detail/workspace.hpp>
      11             : #include <boost/http_proto/detail/except.hpp>
      12             : #include <boost/assert.hpp>
      13             : 
      14             : namespace boost {
      15             : namespace http_proto {
      16             : namespace detail {
      17             : 
      18             : workspace::
      19             : any::
      20             : ~any() = default;
      21             : 
      22         803 : workspace::
      23         803 : ~workspace()
      24             : {
      25         803 :     if(begin_)
      26             :     {
      27         803 :         clear();
      28         803 :         delete[] begin_;
      29             :     }
      30         803 : }
      31             : 
      32          12 : workspace::
      33             : workspace(
      34          12 :     std::size_t n)
      35          12 :     : begin_(new unsigned char[n])
      36          12 :     , front_(begin_)
      37          12 :     , head_(begin_ + n)
      38          12 :     , back_(head_)
      39          12 :     , end_(head_)
      40             : {
      41          12 : }
      42             : 
      43           0 : workspace::
      44             : workspace(
      45           0 :     workspace&& other) noexcept
      46           0 :     : begin_(other.begin_)
      47           0 :     , front_(other.front_)
      48           0 :     , head_(other.end_)
      49           0 :     , back_(other.back_)
      50           0 :     , end_(other.end_)
      51             : {
      52           0 :     other.begin_ = nullptr;
      53           0 :     other.front_ = nullptr;
      54           0 :     other.head_ = nullptr;
      55           0 :     other.back_ = nullptr;
      56           0 :     other.end_ = nullptr;
      57           0 : }
      58             : 
      59             : void
      60         791 : workspace::
      61             : allocate(
      62             :     std::size_t n)
      63             : {
      64             :     // Cannot be empty
      65         791 :     if(n == 0)
      66           0 :         detail::throw_invalid_argument();
      67             : 
      68             :     // Already allocated
      69         791 :     if(begin_ != nullptr)
      70           0 :         detail::throw_logic_error();
      71             : 
      72         791 :     begin_ = new unsigned char[n];
      73         791 :     front_ = begin_;
      74         791 :     head_ = begin_ + n;
      75         791 :     back_ = head_;
      76         791 :     end_ = head_;
      77         791 : }
      78             : 
      79             : void
      80        3561 : workspace::
      81             : clear() noexcept
      82             : {
      83        3561 :     if(! begin_)
      84           0 :         return;
      85             : 
      86        3561 :     auto const end =
      87             :         reinterpret_cast<
      88             :             any const*>(back_);
      89        3561 :     auto p =
      90             :         reinterpret_cast<
      91             :             any const*>(head_);
      92        3897 :     while(p != end)
      93             :     {
      94         336 :         auto next = p->next;
      95         336 :         p->~any();
      96         336 :         p = next;
      97             :     }
      98        3561 :     front_ = begin_;
      99        3561 :     head_ = end_;
     100        3561 :     back_ = end_;
     101             : }
     102             : 
     103             : unsigned char*
     104        1213 : workspace::
     105             : reserve_front(
     106             :     std::size_t n)
     107             : {
     108             :     // 
     109             :     // Requested size exceeds available space.
     110             :     // Note you can never reserve the last byte.
     111        1213 :     if(n >= size())
     112           0 :         detail::throw_length_error();
     113             : 
     114        1213 :     auto const p = front_;
     115        1213 :     front_ += n ;
     116        1213 :     return p;
     117             : }
     118             : 
     119             : unsigned char*
     120        1207 : workspace::
     121             : reserve_back(
     122             :     std::size_t n)
     123             : {
     124             :     // can't reserve after acquire
     125        1207 :     if(head_ != end_)
     126           0 :         detail::throw_logic_error();
     127             : 
     128             :     // can't reserve twice
     129        1207 :     if(back_ != end_)
     130           0 :         detail::throw_logic_error();
     131             : 
     132             :     // over capacity
     133        1207 :     std::size_t const lim =
     134        1207 :         head_ - front_;
     135        1207 :     if(n >= lim)
     136           0 :         detail::throw_length_error();
     137             : 
     138        1207 :     head_ -= n;
     139        1207 :     back_ = head_;
     140        1207 :     return back_;
     141             : }
     142             : 
     143             : // https://fitzgeraldnick.com/2019/11/01/always-bump-downwards.html
     144             : unsigned char*
     145         336 : workspace::
     146             : bump_down(
     147             :     std::size_t size,
     148             :     std::size_t align)
     149             : {
     150         336 :     BOOST_ASSERT(align > 0);
     151         336 :     BOOST_ASSERT(
     152             :         (align & (align - 1)) == 0);
     153         336 :     BOOST_ASSERT(front_);
     154             : 
     155         336 :     auto ip0 = reinterpret_cast<
     156         336 :         std::uintptr_t>(front_);
     157         336 :     auto ip = reinterpret_cast<
     158         336 :         std::uintptr_t>(head_);
     159             : 
     160             :     // If you get an exception here, it
     161             :     // means that a buffer was too small
     162             :     // for your workload. Increase the
     163             :     // buffer size.
     164         336 :     if(size > ip - ip0)
     165           0 :         detail::throw_bad_alloc();
     166             : 
     167         336 :     ip -= size;
     168         336 :     ip &= ~(align - 1);
     169             : 
     170             :     // If you get an exception here, it
     171             :     // means that a buffer was too small
     172             :     // for your workload. Increase the
     173             :     // buffer size.
     174         336 :     if(ip < ip0)
     175           0 :         detail::throw_bad_alloc();
     176             : 
     177             :     return reinterpret_cast<
     178         336 :         unsigned char*>(ip);
     179             : }
     180             : 
     181             : } // detail
     182             : } // http_proto
     183             : } // boost