TLA Line data Source code
1 : //
2 : // Copyright (c) 2021 Vinnie Falco (vinnie.falco@gmail.com)
3 : // Copyright (c) 2024 Christian Mazakas
4 : //
5 : // Distributed under the Boost Software License, Version 1.0. (See accompanying
6 : // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
7 : //
8 : // Official repository: https://github.com/cppalliance/http
9 : //
10 :
11 : #include <boost/http/message_base.hpp>
12 : #include <boost/http/rfc/list_rule.hpp>
13 : #include <boost/http/rfc/token_rule.hpp>
14 : #include <boost/http/detail/except.hpp>
15 : #include "detail/number_string.hpp"
16 : #include <boost/url/grammar/parse.hpp>
17 : #include <boost/url/grammar/ci_string.hpp>
18 :
19 : namespace boost {
20 : namespace http {
21 :
22 : void
23 HIT 37 : message_base::
24 : set_payload_size(
25 : std::uint64_t n)
26 : {
27 : //if(! is_head_response())
28 : if(true)
29 : {
30 : // comes first for exception safety
31 37 : set_content_length(n);
32 :
33 37 : set_chunked(false);
34 : }
35 : else
36 : {
37 : // VFALCO ?
38 : }
39 37 : }
40 :
41 : void
42 37 : message_base::
43 : set_content_length(
44 : std::uint64_t n)
45 : {
46 37 : set(field::content_length,
47 37 : detail::number_string(n));
48 37 : }
49 :
50 : void
51 70 : message_base::
52 : set_chunked(bool value)
53 : {
54 70 : if(value)
55 : {
56 : // set chunked
57 33 : if(! h_.md.transfer_encoding.is_chunked )
58 : {
59 33 : append(
60 : field::transfer_encoding,
61 : "chunked");
62 33 : return;
63 : }
64 : }
65 : else
66 : {
67 : // clear chunked
68 : // VFALCO ?
69 : }
70 : }
71 :
72 : void
73 13 : message_base::
74 : set_keep_alive(bool value)
75 : {
76 13 : if(h_.md.connection.ec)
77 : {
78 : // throw? return false?
79 6 : return;
80 : }
81 :
82 13 : if(h_.md.connection.count == 0)
83 : {
84 : // no Connection field
85 6 : switch(h_.version)
86 : {
87 4 : default:
88 : case http::version::http_1_1:
89 4 : if(! value)
90 3 : set(field::connection, "close");
91 4 : break;
92 :
93 2 : case http::version::http_1_0:
94 2 : if(value)
95 1 : set(field::connection, "keep-alive");
96 2 : break;
97 : }
98 6 : return;
99 : }
100 :
101 : // VFALCO TODO iterate in reverse order,
102 : // and cache the last iterator to use
103 : // for appending
104 :
105 : // one or more Connection fields
106 7 : auto it = begin();
107 : auto const erase_token =
108 6 : [&](core::string_view token)
109 : {
110 14 : while(it != end())
111 : {
112 8 : if(it->id != field::connection)
113 : {
114 MIS 0 : ++it;
115 HIT 4 : continue;
116 : }
117 : auto rv = grammar::parse(
118 8 : it->value,
119 16 : list_rule(token_rule, 1));
120 8 : BOOST_ASSERT(! rv.has_error());
121 8 : BOOST_ASSERT(! rv->empty());
122 8 : auto itv = rv->begin();
123 8 : if(urls::grammar::ci_is_equal(
124 8 : *itv, token))
125 : {
126 4 : if(rv->size() == 1)
127 : {
128 : // only one token
129 3 : it = erase(it);
130 : }
131 : else
132 : {
133 : // first token matches
134 1 : ++itv;
135 1 : set(it,
136 1 : it->value.substr(
137 2 : (*itv).data() -
138 1 : it->value.data()));
139 1 : ++it;
140 : }
141 4 : continue;
142 : }
143 : // search remaining tokens
144 4 : std::string s = *itv++;
145 7 : while(itv != rv->end())
146 : {
147 3 : if(! urls::grammar::ci_is_equal(
148 3 : *itv, token))
149 2 : s += ", " + std::string(*itv);
150 3 : ++itv;
151 : }
152 4 : set(it, s);
153 4 : ++it;
154 8 : }
155 6 : };
156 7 : if(value)
157 : {
158 6 : if(h_.md.connection.close)
159 5 : erase_token("close");
160 : }
161 : else
162 : {
163 1 : if(h_.md.connection.keep_alive)
164 1 : erase_token("keep-alive");
165 : }
166 7 : switch(h_.version)
167 : {
168 5 : default:
169 : case http::version::http_1_1:
170 5 : if(! value)
171 : {
172 : // add one "close" token if needed
173 MIS 0 : if(! h_.md.connection.close)
174 0 : append(field::connection, "close");
175 : }
176 HIT 5 : break;
177 :
178 2 : case http::version::http_1_0:
179 2 : if(value)
180 : {
181 : // add one "keep-alive" token if needed
182 1 : if(! h_.md.connection.keep_alive)
183 MIS 0 : append(field::connection, "keep-alive");
184 : }
185 HIT 2 : break;
186 : }
187 : }
188 :
189 : } // http
190 : } // boost
|
