buffered_read_stream.hpp 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491
  1. //
  2. // impl/buffered_read_stream.hpp
  3. // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  4. //
  5. // Copyright (c) 2003-2019 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. #ifndef BOOST_ASIO_IMPL_BUFFERED_READ_STREAM_HPP
  11. #define BOOST_ASIO_IMPL_BUFFERED_READ_STREAM_HPP
  12. #if defined(_MSC_VER) && (_MSC_VER >= 1200)
  13. # pragma once
  14. #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
  15. #include <boost/asio/associated_allocator.hpp>
  16. #include <boost/asio/associated_executor.hpp>
  17. #include <boost/asio/detail/handler_alloc_helpers.hpp>
  18. #include <boost/asio/detail/handler_cont_helpers.hpp>
  19. #include <boost/asio/detail/handler_invoke_helpers.hpp>
  20. #include <boost/asio/detail/handler_type_requirements.hpp>
  21. #include <boost/asio/detail/non_const_lvalue.hpp>
  22. #include <boost/asio/detail/type_traits.hpp>
  23. #include <boost/asio/detail/push_options.hpp>
  24. namespace boost {
  25. namespace asio {
  26. template <typename Stream>
  27. std::size_t buffered_read_stream<Stream>::fill()
  28. {
  29. detail::buffer_resize_guard<detail::buffered_stream_storage>
  30. resize_guard(storage_);
  31. std::size_t previous_size = storage_.size();
  32. storage_.resize(storage_.capacity());
  33. storage_.resize(previous_size + next_layer_.read_some(buffer(
  34. storage_.data() + previous_size,
  35. storage_.size() - previous_size)));
  36. resize_guard.commit();
  37. return storage_.size() - previous_size;
  38. }
  39. template <typename Stream>
  40. std::size_t buffered_read_stream<Stream>::fill(boost::system::error_code& ec)
  41. {
  42. detail::buffer_resize_guard<detail::buffered_stream_storage>
  43. resize_guard(storage_);
  44. std::size_t previous_size = storage_.size();
  45. storage_.resize(storage_.capacity());
  46. storage_.resize(previous_size + next_layer_.read_some(buffer(
  47. storage_.data() + previous_size,
  48. storage_.size() - previous_size),
  49. ec));
  50. resize_guard.commit();
  51. return storage_.size() - previous_size;
  52. }
  53. namespace detail
  54. {
  55. template <typename ReadHandler>
  56. class buffered_fill_handler
  57. {
  58. public:
  59. buffered_fill_handler(detail::buffered_stream_storage& storage,
  60. std::size_t previous_size, ReadHandler& handler)
  61. : storage_(storage),
  62. previous_size_(previous_size),
  63. handler_(BOOST_ASIO_MOVE_CAST(ReadHandler)(handler))
  64. {
  65. }
  66. #if defined(BOOST_ASIO_HAS_MOVE)
  67. buffered_fill_handler(const buffered_fill_handler& other)
  68. : storage_(other.storage_),
  69. previous_size_(other.previous_size_),
  70. handler_(other.handler_)
  71. {
  72. }
  73. buffered_fill_handler(buffered_fill_handler&& other)
  74. : storage_(other.storage_),
  75. previous_size_(other.previous_size_),
  76. handler_(BOOST_ASIO_MOVE_CAST(ReadHandler)(other.handler_))
  77. {
  78. }
  79. #endif // defined(BOOST_ASIO_HAS_MOVE)
  80. void operator()(const boost::system::error_code& ec,
  81. const std::size_t bytes_transferred)
  82. {
  83. storage_.resize(previous_size_ + bytes_transferred);
  84. handler_(ec, bytes_transferred);
  85. }
  86. //private:
  87. detail::buffered_stream_storage& storage_;
  88. std::size_t previous_size_;
  89. ReadHandler handler_;
  90. };
  91. template <typename ReadHandler>
  92. inline void* asio_handler_allocate(std::size_t size,
  93. buffered_fill_handler<ReadHandler>* this_handler)
  94. {
  95. return boost_asio_handler_alloc_helpers::allocate(
  96. size, this_handler->handler_);
  97. }
  98. template <typename ReadHandler>
  99. inline void asio_handler_deallocate(void* pointer, std::size_t size,
  100. buffered_fill_handler<ReadHandler>* this_handler)
  101. {
  102. boost_asio_handler_alloc_helpers::deallocate(
  103. pointer, size, this_handler->handler_);
  104. }
  105. template <typename ReadHandler>
  106. inline bool asio_handler_is_continuation(
  107. buffered_fill_handler<ReadHandler>* this_handler)
  108. {
  109. return boost_asio_handler_cont_helpers::is_continuation(
  110. this_handler->handler_);
  111. }
  112. template <typename Function, typename ReadHandler>
  113. inline void asio_handler_invoke(Function& function,
  114. buffered_fill_handler<ReadHandler>* this_handler)
  115. {
  116. boost_asio_handler_invoke_helpers::invoke(
  117. function, this_handler->handler_);
  118. }
  119. template <typename Function, typename ReadHandler>
  120. inline void asio_handler_invoke(const Function& function,
  121. buffered_fill_handler<ReadHandler>* this_handler)
  122. {
  123. boost_asio_handler_invoke_helpers::invoke(
  124. function, this_handler->handler_);
  125. }
  126. template <typename Stream>
  127. class initiate_async_buffered_fill
  128. {
  129. public:
  130. typedef typename remove_reference<
  131. Stream>::type::lowest_layer_type::executor_type executor_type;
  132. explicit initiate_async_buffered_fill(Stream& next_layer)
  133. : next_layer_(next_layer)
  134. {
  135. }
  136. executor_type get_executor() const BOOST_ASIO_NOEXCEPT
  137. {
  138. return next_layer_.lowest_layer().get_executor();
  139. }
  140. template <typename ReadHandler>
  141. void operator()(BOOST_ASIO_MOVE_ARG(ReadHandler) handler,
  142. buffered_stream_storage* storage) const
  143. {
  144. // If you get an error on the following line it means that your handler
  145. // does not meet the documented type requirements for a ReadHandler.
  146. BOOST_ASIO_READ_HANDLER_CHECK(ReadHandler, handler) type_check;
  147. non_const_lvalue<ReadHandler> handler2(handler);
  148. std::size_t previous_size = storage->size();
  149. storage->resize(storage->capacity());
  150. next_layer_.async_read_some(
  151. buffer(
  152. storage->data() + previous_size,
  153. storage->size() - previous_size),
  154. buffered_fill_handler<typename decay<ReadHandler>::type>(
  155. *storage, previous_size, handler2.value));
  156. }
  157. private:
  158. Stream& next_layer_;
  159. };
  160. } // namespace detail
  161. #if !defined(GENERATING_DOCUMENTATION)
  162. template <typename ReadHandler, typename Allocator>
  163. struct associated_allocator<
  164. detail::buffered_fill_handler<ReadHandler>, Allocator>
  165. {
  166. typedef typename associated_allocator<ReadHandler, Allocator>::type type;
  167. static type get(const detail::buffered_fill_handler<ReadHandler>& h,
  168. const Allocator& a = Allocator()) BOOST_ASIO_NOEXCEPT
  169. {
  170. return associated_allocator<ReadHandler, Allocator>::get(h.handler_, a);
  171. }
  172. };
  173. template <typename ReadHandler, typename Executor>
  174. struct associated_executor<
  175. detail::buffered_fill_handler<ReadHandler>, Executor>
  176. {
  177. typedef typename associated_executor<ReadHandler, Executor>::type type;
  178. static type get(const detail::buffered_fill_handler<ReadHandler>& h,
  179. const Executor& ex = Executor()) BOOST_ASIO_NOEXCEPT
  180. {
  181. return associated_executor<ReadHandler, Executor>::get(h.handler_, ex);
  182. }
  183. };
  184. #endif // !defined(GENERATING_DOCUMENTATION)
  185. template <typename Stream>
  186. template <
  187. BOOST_ASIO_COMPLETION_TOKEN_FOR(void (boost::system::error_code,
  188. std::size_t)) ReadHandler>
  189. BOOST_ASIO_INITFN_AUTO_RESULT_TYPE(ReadHandler,
  190. void (boost::system::error_code, std::size_t))
  191. buffered_read_stream<Stream>::async_fill(
  192. BOOST_ASIO_MOVE_ARG(ReadHandler) handler)
  193. {
  194. return async_initiate<ReadHandler,
  195. void (boost::system::error_code, std::size_t)>(
  196. detail::initiate_async_buffered_fill<Stream>(next_layer_),
  197. handler, &storage_);
  198. }
  199. template <typename Stream>
  200. template <typename MutableBufferSequence>
  201. std::size_t buffered_read_stream<Stream>::read_some(
  202. const MutableBufferSequence& buffers)
  203. {
  204. using boost::asio::buffer_size;
  205. if (buffer_size(buffers) == 0)
  206. return 0;
  207. if (storage_.empty())
  208. this->fill();
  209. return this->copy(buffers);
  210. }
  211. template <typename Stream>
  212. template <typename MutableBufferSequence>
  213. std::size_t buffered_read_stream<Stream>::read_some(
  214. const MutableBufferSequence& buffers, boost::system::error_code& ec)
  215. {
  216. ec = boost::system::error_code();
  217. using boost::asio::buffer_size;
  218. if (buffer_size(buffers) == 0)
  219. return 0;
  220. if (storage_.empty() && !this->fill(ec))
  221. return 0;
  222. return this->copy(buffers);
  223. }
  224. namespace detail
  225. {
  226. template <typename MutableBufferSequence, typename ReadHandler>
  227. class buffered_read_some_handler
  228. {
  229. public:
  230. buffered_read_some_handler(detail::buffered_stream_storage& storage,
  231. const MutableBufferSequence& buffers, ReadHandler& handler)
  232. : storage_(storage),
  233. buffers_(buffers),
  234. handler_(BOOST_ASIO_MOVE_CAST(ReadHandler)(handler))
  235. {
  236. }
  237. #if defined(BOOST_ASIO_HAS_MOVE)
  238. buffered_read_some_handler(const buffered_read_some_handler& other)
  239. : storage_(other.storage_),
  240. buffers_(other.buffers_),
  241. handler_(other.handler_)
  242. {
  243. }
  244. buffered_read_some_handler(buffered_read_some_handler&& other)
  245. : storage_(other.storage_),
  246. buffers_(other.buffers_),
  247. handler_(BOOST_ASIO_MOVE_CAST(ReadHandler)(other.handler_))
  248. {
  249. }
  250. #endif // defined(BOOST_ASIO_HAS_MOVE)
  251. void operator()(const boost::system::error_code& ec, std::size_t)
  252. {
  253. if (ec || storage_.empty())
  254. {
  255. const std::size_t length = 0;
  256. handler_(ec, length);
  257. }
  258. else
  259. {
  260. const std::size_t bytes_copied = boost::asio::buffer_copy(
  261. buffers_, storage_.data(), storage_.size());
  262. storage_.consume(bytes_copied);
  263. handler_(ec, bytes_copied);
  264. }
  265. }
  266. //private:
  267. detail::buffered_stream_storage& storage_;
  268. MutableBufferSequence buffers_;
  269. ReadHandler handler_;
  270. };
  271. template <typename MutableBufferSequence, typename ReadHandler>
  272. inline void* asio_handler_allocate(std::size_t size,
  273. buffered_read_some_handler<
  274. MutableBufferSequence, ReadHandler>* this_handler)
  275. {
  276. return boost_asio_handler_alloc_helpers::allocate(
  277. size, this_handler->handler_);
  278. }
  279. template <typename MutableBufferSequence, typename ReadHandler>
  280. inline void asio_handler_deallocate(void* pointer, std::size_t size,
  281. buffered_read_some_handler<
  282. MutableBufferSequence, ReadHandler>* this_handler)
  283. {
  284. boost_asio_handler_alloc_helpers::deallocate(
  285. pointer, size, this_handler->handler_);
  286. }
  287. template <typename MutableBufferSequence, typename ReadHandler>
  288. inline bool asio_handler_is_continuation(
  289. buffered_read_some_handler<
  290. MutableBufferSequence, ReadHandler>* this_handler)
  291. {
  292. return boost_asio_handler_cont_helpers::is_continuation(
  293. this_handler->handler_);
  294. }
  295. template <typename Function, typename MutableBufferSequence,
  296. typename ReadHandler>
  297. inline void asio_handler_invoke(Function& function,
  298. buffered_read_some_handler<
  299. MutableBufferSequence, ReadHandler>* this_handler)
  300. {
  301. boost_asio_handler_invoke_helpers::invoke(
  302. function, this_handler->handler_);
  303. }
  304. template <typename Function, typename MutableBufferSequence,
  305. typename ReadHandler>
  306. inline void asio_handler_invoke(const Function& function,
  307. buffered_read_some_handler<
  308. MutableBufferSequence, ReadHandler>* this_handler)
  309. {
  310. boost_asio_handler_invoke_helpers::invoke(
  311. function, this_handler->handler_);
  312. }
  313. template <typename Stream>
  314. class initiate_async_buffered_read_some
  315. {
  316. public:
  317. typedef typename remove_reference<
  318. Stream>::type::lowest_layer_type::executor_type executor_type;
  319. explicit initiate_async_buffered_read_some(Stream& next_layer)
  320. : next_layer_(next_layer)
  321. {
  322. }
  323. executor_type get_executor() const BOOST_ASIO_NOEXCEPT
  324. {
  325. return next_layer_.lowest_layer().get_executor();
  326. }
  327. template <typename ReadHandler, typename MutableBufferSequence>
  328. void operator()(BOOST_ASIO_MOVE_ARG(ReadHandler) handler,
  329. buffered_stream_storage* storage,
  330. const MutableBufferSequence& buffers) const
  331. {
  332. // If you get an error on the following line it means that your handler
  333. // does not meet the documented type requirements for a ReadHandler.
  334. BOOST_ASIO_READ_HANDLER_CHECK(ReadHandler, handler) type_check;
  335. using boost::asio::buffer_size;
  336. non_const_lvalue<ReadHandler> handler2(handler);
  337. if (buffer_size(buffers) == 0 || !storage->empty())
  338. {
  339. next_layer_.async_read_some(BOOST_ASIO_MUTABLE_BUFFER(0, 0),
  340. buffered_read_some_handler<MutableBufferSequence,
  341. typename decay<ReadHandler>::type>(
  342. *storage, buffers, handler2.value));
  343. }
  344. else
  345. {
  346. initiate_async_buffered_fill<Stream>(this->next_layer_)(
  347. buffered_read_some_handler<MutableBufferSequence,
  348. typename decay<ReadHandler>::type>(
  349. *storage, buffers, handler2.value),
  350. storage);
  351. }
  352. }
  353. private:
  354. Stream& next_layer_;
  355. };
  356. } // namespace detail
  357. #if !defined(GENERATING_DOCUMENTATION)
  358. template <typename MutableBufferSequence,
  359. typename ReadHandler, typename Allocator>
  360. struct associated_allocator<
  361. detail::buffered_read_some_handler<MutableBufferSequence, ReadHandler>,
  362. Allocator>
  363. {
  364. typedef typename associated_allocator<ReadHandler, Allocator>::type type;
  365. static type get(
  366. const detail::buffered_read_some_handler<
  367. MutableBufferSequence, ReadHandler>& h,
  368. const Allocator& a = Allocator()) BOOST_ASIO_NOEXCEPT
  369. {
  370. return associated_allocator<ReadHandler, Allocator>::get(h.handler_, a);
  371. }
  372. };
  373. template <typename MutableBufferSequence,
  374. typename ReadHandler, typename Executor>
  375. struct associated_executor<
  376. detail::buffered_read_some_handler<MutableBufferSequence, ReadHandler>,
  377. Executor>
  378. {
  379. typedef typename associated_executor<ReadHandler, Executor>::type type;
  380. static type get(
  381. const detail::buffered_read_some_handler<
  382. MutableBufferSequence, ReadHandler>& h,
  383. const Executor& ex = Executor()) BOOST_ASIO_NOEXCEPT
  384. {
  385. return associated_executor<ReadHandler, Executor>::get(h.handler_, ex);
  386. }
  387. };
  388. #endif // !defined(GENERATING_DOCUMENTATION)
  389. template <typename Stream>
  390. template <typename MutableBufferSequence,
  391. BOOST_ASIO_COMPLETION_TOKEN_FOR(void (boost::system::error_code,
  392. std::size_t)) ReadHandler>
  393. BOOST_ASIO_INITFN_AUTO_RESULT_TYPE(ReadHandler,
  394. void (boost::system::error_code, std::size_t))
  395. buffered_read_stream<Stream>::async_read_some(
  396. const MutableBufferSequence& buffers,
  397. BOOST_ASIO_MOVE_ARG(ReadHandler) handler)
  398. {
  399. return async_initiate<ReadHandler,
  400. void (boost::system::error_code, std::size_t)>(
  401. detail::initiate_async_buffered_read_some<Stream>(next_layer_),
  402. handler, &storage_, buffers);
  403. }
  404. template <typename Stream>
  405. template <typename MutableBufferSequence>
  406. std::size_t buffered_read_stream<Stream>::peek(
  407. const MutableBufferSequence& buffers)
  408. {
  409. if (storage_.empty())
  410. this->fill();
  411. return this->peek_copy(buffers);
  412. }
  413. template <typename Stream>
  414. template <typename MutableBufferSequence>
  415. std::size_t buffered_read_stream<Stream>::peek(
  416. const MutableBufferSequence& buffers, boost::system::error_code& ec)
  417. {
  418. ec = boost::system::error_code();
  419. if (storage_.empty() && !this->fill(ec))
  420. return 0;
  421. return this->peek_copy(buffers);
  422. }
  423. } // namespace asio
  424. } // namespace boost
  425. #include <boost/asio/detail/pop_options.hpp>
  426. #endif // BOOST_ASIO_IMPL_BUFFERED_READ_STREAM_HPP