TLA  Line data    Source code

       1                 : //
       2                 : // Copyright (c) 2021 Vinnie Falco (vinnie.falco@gmail.com)
       3                 : // Copyright (c) 2024 Christian Mazakas
       4                 : // Copyright (c) 2025 Mohammad Nejati
       5                 : //
       6                 : // Distributed under the Boost Software License, Version 1.0. (See accompanying
       7                 : // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
       8                 : //
       9                 : // Official repository: https://github.com/cppalliance/http
      10                 : //
      11                 : 
      12                 : #include <boost/http/request_base.hpp>
      13                 : 
      14                 : #include <cstring>
      15                 : 
      16                 : namespace boost {
      17                 : namespace http {
      18                 : 
      19                 : void
      20 HIT          10 : request_base::
      21                 : set_expect_100_continue(bool b)
      22                 : {
      23              10 :     if(h_.md.expect.count == 0)
      24                 :     {
      25               3 :         BOOST_ASSERT(
      26                 :             ! h_.md.expect.ec);
      27               3 :         BOOST_ASSERT(
      28                 :             ! h_.md.expect.is_100_continue);
      29               3 :         if( b )
      30                 :         {
      31               2 :             append(
      32                 :                 field::expect,
      33                 :                 "100-continue");
      34               2 :             return;
      35                 :         }
      36               1 :         return;
      37                 :     }
      38                 : 
      39               7 :     if(h_.md.expect.count == 1)
      40                 :     {
      41               3 :         if(b)
      42                 :         {
      43               2 :             if(! h_.md.expect.ec)
      44                 :             {
      45               1 :                 BOOST_ASSERT(
      46                 :                     h_.md.expect.is_100_continue);
      47               1 :                 return;
      48                 :             }
      49               1 :             BOOST_ASSERT(
      50                 :                 ! h_.md.expect.is_100_continue);
      51               1 :             auto it = find(field::expect);
      52               1 :             BOOST_ASSERT(it != end());
      53               1 :             set(it, "100-continue");
      54               1 :             return;
      55                 :         }
      56                 : 
      57               1 :         auto it = find(field::expect);
      58               1 :         BOOST_ASSERT(it != end());
      59               1 :         erase(it);
      60               1 :         return;
      61                 :     }
      62                 : 
      63               4 :     BOOST_ASSERT(h_.md.expect.ec);
      64                 : 
      65               4 :     auto nc = (b ? 1 : 0);
      66               4 :     auto ne = h_.md.expect.count - nc;
      67               4 :     if( b )
      68               3 :         set(find(field::expect), "100-continue");
      69                 : 
      70               4 :     raw_erase_n(field::expect, ne);
      71               4 :     h_.md.expect.count = nc;
      72               4 :     h_.md.expect.ec = {};
      73               4 :     h_.md.expect.is_100_continue = b;
      74                 : }
      75                 : 
      76                 : //------------------------------------------------
      77                 : 
      78                 : void
      79              44 : request_base::
      80                 : set_start_line_impl(
      81                 :     http::method m,
      82                 :     core::string_view ms,
      83                 :     core::string_view t,
      84                 :     http::version v)
      85                 : {
      86                 :     // TODO: check validity
      87              44 :     auto const vs = to_string(v);
      88                 :     auto const new_prefix =
      89              44 :         ms.size() + 1 + // method SP
      90              44 :         t.size() + 1 +  // request-target SP
      91              44 :         vs.size() + 2;  // HTTP-version CRLF
      92                 : 
      93                 :     // Introduce a new scope so that prefix_op's
      94                 :     // destructor runs before h_.on_start_line().
      95                 :     {
      96                 :         auto op = prefix_op_t(
      97              44 :             *this, new_prefix, &ms, &t);
      98                 : 
      99              42 :         h_.version = v;
     100              42 :         h_.req.method = m;
     101              42 :         h_.req.method_len = static_cast<
     102              42 :             offset_type>(ms.size());
     103              42 :         h_.req.target_len = static_cast<
     104              42 :             offset_type>(t.size());
     105                 : 
     106              42 :         char* m_dest = h_.buf;
     107              42 :         char* t_dest = h_.buf + ms.size() + 1;
     108              42 :         char* v_dest = t_dest + t.size() + 1;
     109                 : 
     110              42 :         std::memmove(t_dest, t.data(), t.size());
     111              42 :         t_dest[t.size()] = ' ';
     112                 : 
     113                 :         // memmove after target because could overlap
     114              42 :         std::memmove(m_dest, ms.data(), ms.size());
     115              42 :         m_dest[ms.size()] = ' ';
     116                 : 
     117              42 :         std::memcpy(v_dest, vs.data(), vs.size());
     118              42 :         v_dest[vs.size() + 0] = '\r';
     119              42 :         v_dest[vs.size() + 1] = '\n';
     120              42 :     }
     121                 : 
     122              42 :     h_.on_start_line();
     123              42 : }
     124                 : 
     125                 : } // http
     126                 : } // boost