1 | // |
2 | // impl/buffered_write_stream.hpp |
3 | // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ |
4 | // |
5 | // Copyright (c) 2003-2015 Christopher M. Kohlhoff (chris at kohlhoff dot com) |
6 | // |
7 | // Distributed under the Boost Software License, Version 1.0. (See accompanying |
8 | // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) |
9 | // |
10 | |
11 | #ifndef BOOST_ASIO_IMPL_BUFFERED_WRITE_STREAM_HPP |
12 | #define BOOST_ASIO_IMPL_BUFFERED_WRITE_STREAM_HPP |
13 | |
14 | #if defined(_MSC_VER) && (_MSC_VER >= 1200) |
15 | # pragma once |
16 | #endif // defined(_MSC_VER) && (_MSC_VER >= 1200) |
17 | |
18 | #include <boost/asio/detail/handler_alloc_helpers.hpp> |
19 | #include <boost/asio/detail/handler_cont_helpers.hpp> |
20 | #include <boost/asio/detail/handler_invoke_helpers.hpp> |
21 | #include <boost/asio/detail/handler_type_requirements.hpp> |
22 | |
23 | #include <boost/asio/detail/push_options.hpp> |
24 | |
25 | namespace boost { |
26 | namespace asio { |
27 | |
28 | template <typename Stream> |
29 | std::size_t buffered_write_stream<Stream>::flush() |
30 | { |
31 | std::size_t bytes_written = write(next_layer_, |
32 | buffer(b: storage_.data(), max_size_in_bytes: storage_.size())); |
33 | storage_.consume(count: bytes_written); |
34 | return bytes_written; |
35 | } |
36 | |
37 | template <typename Stream> |
38 | std::size_t buffered_write_stream<Stream>::flush(boost::system::error_code& ec) |
39 | { |
40 | std::size_t bytes_written = write(next_layer_, |
41 | buffer(b: storage_.data(), max_size_in_bytes: storage_.size()), |
42 | transfer_all(), ec); |
43 | storage_.consume(count: bytes_written); |
44 | return bytes_written; |
45 | } |
46 | |
47 | namespace detail |
48 | { |
49 | template <typename WriteHandler> |
50 | class buffered_flush_handler |
51 | { |
52 | public: |
53 | buffered_flush_handler(detail::buffered_stream_storage& storage, |
54 | WriteHandler& handler) |
55 | : storage_(storage), |
56 | handler_(handler) |
57 | { |
58 | } |
59 | |
60 | #if defined(BOOST_ASIO_HAS_MOVE) |
61 | buffered_flush_handler(const buffered_flush_handler& other) |
62 | : storage_(other.storage_), |
63 | handler_(other.handler_) |
64 | { |
65 | } |
66 | |
67 | buffered_flush_handler(buffered_flush_handler&& other) |
68 | : storage_(other.storage_), |
69 | handler_(BOOST_ASIO_MOVE_CAST(WriteHandler)(other.handler_)) |
70 | { |
71 | } |
72 | #endif // defined(BOOST_ASIO_HAS_MOVE) |
73 | |
74 | void operator()(const boost::system::error_code& ec, |
75 | const std::size_t bytes_written) |
76 | { |
77 | storage_.consume(count: bytes_written); |
78 | handler_(ec, bytes_written); |
79 | } |
80 | |
81 | //private: |
82 | detail::buffered_stream_storage& storage_; |
83 | WriteHandler handler_; |
84 | }; |
85 | |
86 | template <typename WriteHandler> |
87 | inline void* asio_handler_allocate(std::size_t size, |
88 | buffered_flush_handler<WriteHandler>* this_handler) |
89 | { |
90 | return boost_asio_handler_alloc_helpers::allocate( |
91 | size, this_handler->handler_); |
92 | } |
93 | |
94 | template <typename WriteHandler> |
95 | inline void asio_handler_deallocate(void* pointer, std::size_t size, |
96 | buffered_flush_handler<WriteHandler>* this_handler) |
97 | { |
98 | boost_asio_handler_alloc_helpers::deallocate( |
99 | pointer, size, this_handler->handler_); |
100 | } |
101 | |
102 | template <typename WriteHandler> |
103 | inline bool asio_handler_is_continuation( |
104 | buffered_flush_handler<WriteHandler>* this_handler) |
105 | { |
106 | return boost_asio_handler_cont_helpers::is_continuation( |
107 | this_handler->handler_); |
108 | } |
109 | |
110 | template <typename Function, typename WriteHandler> |
111 | inline void asio_handler_invoke(Function& function, |
112 | buffered_flush_handler<WriteHandler>* this_handler) |
113 | { |
114 | boost_asio_handler_invoke_helpers::invoke( |
115 | function, this_handler->handler_); |
116 | } |
117 | |
118 | template <typename Function, typename WriteHandler> |
119 | inline void asio_handler_invoke(const Function& function, |
120 | buffered_flush_handler<WriteHandler>* this_handler) |
121 | { |
122 | boost_asio_handler_invoke_helpers::invoke( |
123 | function, this_handler->handler_); |
124 | } |
125 | } |
126 | |
127 | template <typename Stream> |
128 | template <typename WriteHandler> |
129 | BOOST_ASIO_INITFN_RESULT_TYPE(WriteHandler, |
130 | void (boost::system::error_code, std::size_t)) |
131 | buffered_write_stream<Stream>::async_flush( |
132 | BOOST_ASIO_MOVE_ARG(WriteHandler) handler) |
133 | { |
134 | // If you get an error on the following line it means that your handler does |
135 | // not meet the documented type requirements for a WriteHandler. |
136 | BOOST_ASIO_WRITE_HANDLER_CHECK(WriteHandler, handler) type_check; |
137 | |
138 | detail::async_result_init< |
139 | WriteHandler, void (boost::system::error_code, std::size_t)> init( |
140 | BOOST_ASIO_MOVE_CAST(WriteHandler)(handler)); |
141 | |
142 | async_write(next_layer_, buffer(b: storage_.data(), max_size_in_bytes: storage_.size()), |
143 | detail::buffered_flush_handler<BOOST_ASIO_HANDLER_TYPE( |
144 | WriteHandler, void (boost::system::error_code, std::size_t))>( |
145 | storage_, init.handler)); |
146 | |
147 | return init.result.get(); |
148 | } |
149 | |
150 | template <typename Stream> |
151 | template <typename ConstBufferSequence> |
152 | std::size_t buffered_write_stream<Stream>::write_some( |
153 | const ConstBufferSequence& buffers) |
154 | { |
155 | if (boost::asio::buffer_size(buffers) == 0) |
156 | return 0; |
157 | |
158 | if (storage_.size() == storage_.capacity()) |
159 | this->flush(); |
160 | |
161 | return this->copy(buffers); |
162 | } |
163 | |
164 | template <typename Stream> |
165 | template <typename ConstBufferSequence> |
166 | std::size_t buffered_write_stream<Stream>::write_some( |
167 | const ConstBufferSequence& buffers, boost::system::error_code& ec) |
168 | { |
169 | ec = boost::system::error_code(); |
170 | |
171 | if (boost::asio::buffer_size(buffers) == 0) |
172 | return 0; |
173 | |
174 | if (storage_.size() == storage_.capacity() && !flush(ec)) |
175 | return 0; |
176 | |
177 | return this->copy(buffers); |
178 | } |
179 | |
180 | namespace detail |
181 | { |
182 | template <typename ConstBufferSequence, typename WriteHandler> |
183 | class buffered_write_some_handler |
184 | { |
185 | public: |
186 | buffered_write_some_handler(detail::buffered_stream_storage& storage, |
187 | const ConstBufferSequence& buffers, WriteHandler& handler) |
188 | : storage_(storage), |
189 | buffers_(buffers), |
190 | handler_(handler) |
191 | { |
192 | } |
193 | |
194 | #if defined(BOOST_ASIO_HAS_MOVE) |
195 | buffered_write_some_handler(const buffered_write_some_handler& other) |
196 | : storage_(other.storage_), |
197 | buffers_(other.buffers_), |
198 | handler_(other.handler_) |
199 | { |
200 | } |
201 | |
202 | buffered_write_some_handler(buffered_write_some_handler&& other) |
203 | : storage_(other.storage_), |
204 | buffers_(other.buffers_), |
205 | handler_(BOOST_ASIO_MOVE_CAST(WriteHandler)(other.handler_)) |
206 | { |
207 | } |
208 | #endif // defined(BOOST_ASIO_HAS_MOVE) |
209 | |
210 | void operator()(const boost::system::error_code& ec, std::size_t) |
211 | { |
212 | if (ec) |
213 | { |
214 | const std::size_t length = 0; |
215 | handler_(ec, length); |
216 | } |
217 | else |
218 | { |
219 | std::size_t orig_size = storage_.size(); |
220 | std::size_t space_avail = storage_.capacity() - orig_size; |
221 | std::size_t bytes_avail = boost::asio::buffer_size(buffers_); |
222 | std::size_t length = bytes_avail < space_avail |
223 | ? bytes_avail : space_avail; |
224 | storage_.resize(length: orig_size + length); |
225 | const std::size_t bytes_copied = boost::asio::buffer_copy( |
226 | storage_.data() + orig_size, buffers_, length); |
227 | handler_(ec, bytes_copied); |
228 | } |
229 | } |
230 | |
231 | //private: |
232 | detail::buffered_stream_storage& storage_; |
233 | ConstBufferSequence buffers_; |
234 | WriteHandler handler_; |
235 | }; |
236 | |
237 | template <typename ConstBufferSequence, typename WriteHandler> |
238 | inline void* asio_handler_allocate(std::size_t size, |
239 | buffered_write_some_handler< |
240 | ConstBufferSequence, WriteHandler>* this_handler) |
241 | { |
242 | return boost_asio_handler_alloc_helpers::allocate( |
243 | size, this_handler->handler_); |
244 | } |
245 | |
246 | template <typename ConstBufferSequence, typename WriteHandler> |
247 | inline void asio_handler_deallocate(void* pointer, std::size_t size, |
248 | buffered_write_some_handler< |
249 | ConstBufferSequence, WriteHandler>* this_handler) |
250 | { |
251 | boost_asio_handler_alloc_helpers::deallocate( |
252 | pointer, size, this_handler->handler_); |
253 | } |
254 | |
255 | template <typename ConstBufferSequence, typename WriteHandler> |
256 | inline bool asio_handler_is_continuation( |
257 | buffered_write_some_handler< |
258 | ConstBufferSequence, WriteHandler>* this_handler) |
259 | { |
260 | return boost_asio_handler_cont_helpers::is_continuation( |
261 | this_handler->handler_); |
262 | } |
263 | |
264 | template <typename Function, typename ConstBufferSequence, |
265 | typename WriteHandler> |
266 | inline void asio_handler_invoke(Function& function, |
267 | buffered_write_some_handler< |
268 | ConstBufferSequence, WriteHandler>* this_handler) |
269 | { |
270 | boost_asio_handler_invoke_helpers::invoke( |
271 | function, this_handler->handler_); |
272 | } |
273 | |
274 | template <typename Function, typename ConstBufferSequence, |
275 | typename WriteHandler> |
276 | inline void asio_handler_invoke(const Function& function, |
277 | buffered_write_some_handler< |
278 | ConstBufferSequence, WriteHandler>* this_handler) |
279 | { |
280 | boost_asio_handler_invoke_helpers::invoke( |
281 | function, this_handler->handler_); |
282 | } |
283 | } // namespace detail |
284 | |
285 | template <typename Stream> |
286 | template <typename ConstBufferSequence, typename WriteHandler> |
287 | BOOST_ASIO_INITFN_RESULT_TYPE(WriteHandler, |
288 | void (boost::system::error_code, std::size_t)) |
289 | buffered_write_stream<Stream>::async_write_some( |
290 | const ConstBufferSequence& buffers, |
291 | BOOST_ASIO_MOVE_ARG(WriteHandler) handler) |
292 | { |
293 | // If you get an error on the following line it means that your handler does |
294 | // not meet the documented type requirements for a WriteHandler. |
295 | BOOST_ASIO_WRITE_HANDLER_CHECK(WriteHandler, handler) type_check; |
296 | |
297 | detail::async_result_init< |
298 | WriteHandler, void (boost::system::error_code, std::size_t)> init( |
299 | BOOST_ASIO_MOVE_CAST(WriteHandler)(handler)); |
300 | |
301 | if (boost::asio::buffer_size(buffers) == 0 |
302 | || storage_.size() < storage_.capacity()) |
303 | { |
304 | next_layer_.async_write_some(boost::asio::const_buffers_1(0, 0), |
305 | detail::buffered_write_some_handler< |
306 | ConstBufferSequence, BOOST_ASIO_HANDLER_TYPE( |
307 | WriteHandler, void (boost::system::error_code, std::size_t))>( |
308 | storage_, buffers, init.handler)); |
309 | } |
310 | else |
311 | { |
312 | this->async_flush(detail::buffered_write_some_handler< |
313 | ConstBufferSequence, BOOST_ASIO_HANDLER_TYPE( |
314 | WriteHandler, void (boost::system::error_code, std::size_t))>( |
315 | storage_, buffers, init.handler)); |
316 | } |
317 | |
318 | return init.result.get(); |
319 | } |
320 | |
321 | template <typename Stream> |
322 | template <typename ConstBufferSequence> |
323 | std::size_t buffered_write_stream<Stream>::copy( |
324 | const ConstBufferSequence& buffers) |
325 | { |
326 | std::size_t orig_size = storage_.size(); |
327 | std::size_t space_avail = storage_.capacity() - orig_size; |
328 | std::size_t bytes_avail = boost::asio::buffer_size(buffers); |
329 | std::size_t length = bytes_avail < space_avail ? bytes_avail : space_avail; |
330 | storage_.resize(length: orig_size + length); |
331 | return boost::asio::buffer_copy( |
332 | storage_.data() + orig_size, buffers, length); |
333 | } |
334 | |
335 | } // namespace asio |
336 | } // namespace boost |
337 | |
338 | #include <boost/asio/detail/pop_options.hpp> |
339 | |
340 | #endif // BOOST_ASIO_IMPL_BUFFERED_WRITE_STREAM_HPP |
341 | |