buffered_read_stream.hpp 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360
  1. //
  2. // impl/buffered_read_stream.hpp
  3. // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  4. //
  5. // Copyright (c) 2003-2013 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/detail/handler_alloc_helpers.hpp>
  16. #include <boost/asio/detail/handler_cont_helpers.hpp>
  17. #include <boost/asio/detail/handler_invoke_helpers.hpp>
  18. #include <boost/asio/detail/handler_type_requirements.hpp>
  19. #include <boost/asio/detail/push_options.hpp>
  20. namespace boost {
  21. namespace asio {
  22. template <typename Stream>
  23. std::size_t buffered_read_stream<Stream>::fill()
  24. {
  25. detail::buffer_resize_guard<detail::buffered_stream_storage>
  26. resize_guard(storage_);
  27. std::size_t previous_size = storage_.size();
  28. storage_.resize(storage_.capacity());
  29. storage_.resize(previous_size + next_layer_.read_some(buffer(
  30. storage_.data() + previous_size,
  31. storage_.size() - previous_size)));
  32. resize_guard.commit();
  33. return storage_.size() - previous_size;
  34. }
  35. template <typename Stream>
  36. std::size_t buffered_read_stream<Stream>::fill(boost::system::error_code& ec)
  37. {
  38. detail::buffer_resize_guard<detail::buffered_stream_storage>
  39. resize_guard(storage_);
  40. std::size_t previous_size = storage_.size();
  41. storage_.resize(storage_.capacity());
  42. storage_.resize(previous_size + next_layer_.read_some(buffer(
  43. storage_.data() + previous_size,
  44. storage_.size() - previous_size),
  45. ec));
  46. resize_guard.commit();
  47. return storage_.size() - previous_size;
  48. }
  49. namespace detail
  50. {
  51. template <typename ReadHandler>
  52. class buffered_fill_handler
  53. {
  54. public:
  55. buffered_fill_handler(detail::buffered_stream_storage& storage,
  56. std::size_t previous_size, ReadHandler& handler)
  57. : storage_(storage),
  58. previous_size_(previous_size),
  59. handler_(BOOST_ASIO_MOVE_CAST(ReadHandler)(handler))
  60. {
  61. }
  62. #if defined(BOOST_ASIO_HAS_MOVE)
  63. buffered_fill_handler(const buffered_fill_handler& other)
  64. : storage_(other.storage_),
  65. previous_size_(other.previous_size_),
  66. handler_(other.handler_)
  67. {
  68. }
  69. buffered_fill_handler(buffered_fill_handler&& other)
  70. : storage_(other.storage_),
  71. previous_size_(other.previous_size_),
  72. handler_(BOOST_ASIO_MOVE_CAST(ReadHandler)(other.handler_))
  73. {
  74. }
  75. #endif // defined(BOOST_ASIO_HAS_MOVE)
  76. void operator()(const boost::system::error_code& ec,
  77. const std::size_t bytes_transferred)
  78. {
  79. storage_.resize(previous_size_ + bytes_transferred);
  80. handler_(ec, bytes_transferred);
  81. }
  82. //private:
  83. detail::buffered_stream_storage& storage_;
  84. std::size_t previous_size_;
  85. ReadHandler handler_;
  86. };
  87. template <typename ReadHandler>
  88. inline void* asio_handler_allocate(std::size_t size,
  89. buffered_fill_handler<ReadHandler>* this_handler)
  90. {
  91. return boost_asio_handler_alloc_helpers::allocate(
  92. size, this_handler->handler_);
  93. }
  94. template <typename ReadHandler>
  95. inline void asio_handler_deallocate(void* pointer, std::size_t size,
  96. buffered_fill_handler<ReadHandler>* this_handler)
  97. {
  98. boost_asio_handler_alloc_helpers::deallocate(
  99. pointer, size, this_handler->handler_);
  100. }
  101. template <typename ReadHandler>
  102. inline bool asio_handler_is_continuation(
  103. buffered_fill_handler<ReadHandler>* this_handler)
  104. {
  105. return boost_asio_handler_cont_helpers::is_continuation(
  106. this_handler->handler_);
  107. }
  108. template <typename Function, typename ReadHandler>
  109. inline void asio_handler_invoke(Function& function,
  110. buffered_fill_handler<ReadHandler>* this_handler)
  111. {
  112. boost_asio_handler_invoke_helpers::invoke(
  113. function, this_handler->handler_);
  114. }
  115. template <typename Function, typename ReadHandler>
  116. inline void asio_handler_invoke(const Function& function,
  117. buffered_fill_handler<ReadHandler>* this_handler)
  118. {
  119. boost_asio_handler_invoke_helpers::invoke(
  120. function, this_handler->handler_);
  121. }
  122. } // namespace detail
  123. template <typename Stream>
  124. template <typename ReadHandler>
  125. BOOST_ASIO_INITFN_RESULT_TYPE(ReadHandler,
  126. void (boost::system::error_code, std::size_t))
  127. buffered_read_stream<Stream>::async_fill(
  128. BOOST_ASIO_MOVE_ARG(ReadHandler) handler)
  129. {
  130. // If you get an error on the following line it means that your handler does
  131. // not meet the documented type requirements for a ReadHandler.
  132. BOOST_ASIO_READ_HANDLER_CHECK(ReadHandler, handler) type_check;
  133. detail::async_result_init<
  134. ReadHandler, void (boost::system::error_code, std::size_t)> init(
  135. BOOST_ASIO_MOVE_CAST(ReadHandler)(handler));
  136. std::size_t previous_size = storage_.size();
  137. storage_.resize(storage_.capacity());
  138. next_layer_.async_read_some(
  139. buffer(
  140. storage_.data() + previous_size,
  141. storage_.size() - previous_size),
  142. detail::buffered_fill_handler<BOOST_ASIO_HANDLER_TYPE(
  143. ReadHandler, void (boost::system::error_code, std::size_t))>(
  144. storage_, previous_size, init.handler));
  145. return init.result.get();
  146. }
  147. template <typename Stream>
  148. template <typename MutableBufferSequence>
  149. std::size_t buffered_read_stream<Stream>::read_some(
  150. const MutableBufferSequence& buffers)
  151. {
  152. if (boost::asio::buffer_size(buffers) == 0)
  153. return 0;
  154. if (storage_.empty())
  155. this->fill();
  156. return this->copy(buffers);
  157. }
  158. template <typename Stream>
  159. template <typename MutableBufferSequence>
  160. std::size_t buffered_read_stream<Stream>::read_some(
  161. const MutableBufferSequence& buffers, boost::system::error_code& ec)
  162. {
  163. ec = boost::system::error_code();
  164. if (boost::asio::buffer_size(buffers) == 0)
  165. return 0;
  166. if (storage_.empty() && !this->fill(ec))
  167. return 0;
  168. return this->copy(buffers);
  169. }
  170. namespace detail
  171. {
  172. template <typename MutableBufferSequence, typename ReadHandler>
  173. class buffered_read_some_handler
  174. {
  175. public:
  176. buffered_read_some_handler(detail::buffered_stream_storage& storage,
  177. const MutableBufferSequence& buffers, ReadHandler& handler)
  178. : storage_(storage),
  179. buffers_(buffers),
  180. handler_(handler)
  181. {
  182. }
  183. #if defined(BOOST_ASIO_HAS_MOVE)
  184. buffered_read_some_handler(const buffered_read_some_handler& other)
  185. : storage_(other.storage_),
  186. buffers_(other.buffers_),
  187. handler_(other.handler_)
  188. {
  189. }
  190. buffered_read_some_handler(buffered_read_some_handler&& other)
  191. : storage_(other.storage_),
  192. buffers_(other.buffers_),
  193. handler_(BOOST_ASIO_MOVE_CAST(ReadHandler)(other.handler_))
  194. {
  195. }
  196. #endif // defined(BOOST_ASIO_HAS_MOVE)
  197. void operator()(const boost::system::error_code& ec, std::size_t)
  198. {
  199. if (ec || storage_.empty())
  200. {
  201. const std::size_t length = 0;
  202. handler_(ec, length);
  203. }
  204. else
  205. {
  206. const std::size_t bytes_copied = boost::asio::buffer_copy(
  207. buffers_, storage_.data(), storage_.size());
  208. storage_.consume(bytes_copied);
  209. handler_(ec, bytes_copied);
  210. }
  211. }
  212. //private:
  213. detail::buffered_stream_storage& storage_;
  214. MutableBufferSequence buffers_;
  215. ReadHandler handler_;
  216. };
  217. template <typename MutableBufferSequence, typename ReadHandler>
  218. inline void* asio_handler_allocate(std::size_t size,
  219. buffered_read_some_handler<
  220. MutableBufferSequence, ReadHandler>* this_handler)
  221. {
  222. return boost_asio_handler_alloc_helpers::allocate(
  223. size, this_handler->handler_);
  224. }
  225. template <typename MutableBufferSequence, typename ReadHandler>
  226. inline void asio_handler_deallocate(void* pointer, std::size_t size,
  227. buffered_read_some_handler<
  228. MutableBufferSequence, ReadHandler>* this_handler)
  229. {
  230. boost_asio_handler_alloc_helpers::deallocate(
  231. pointer, size, this_handler->handler_);
  232. }
  233. template <typename MutableBufferSequence, typename ReadHandler>
  234. inline bool asio_handler_is_continuation(
  235. buffered_read_some_handler<
  236. MutableBufferSequence, ReadHandler>* this_handler)
  237. {
  238. return boost_asio_handler_cont_helpers::is_continuation(
  239. this_handler->handler_);
  240. }
  241. template <typename Function, typename MutableBufferSequence,
  242. typename ReadHandler>
  243. inline void asio_handler_invoke(Function& function,
  244. buffered_read_some_handler<
  245. MutableBufferSequence, ReadHandler>* this_handler)
  246. {
  247. boost_asio_handler_invoke_helpers::invoke(
  248. function, this_handler->handler_);
  249. }
  250. template <typename Function, typename MutableBufferSequence,
  251. typename ReadHandler>
  252. inline void asio_handler_invoke(const Function& function,
  253. buffered_read_some_handler<
  254. MutableBufferSequence, ReadHandler>* this_handler)
  255. {
  256. boost_asio_handler_invoke_helpers::invoke(
  257. function, this_handler->handler_);
  258. }
  259. } // namespace detail
  260. template <typename Stream>
  261. template <typename MutableBufferSequence, typename ReadHandler>
  262. BOOST_ASIO_INITFN_RESULT_TYPE(ReadHandler,
  263. void (boost::system::error_code, std::size_t))
  264. buffered_read_stream<Stream>::async_read_some(
  265. const MutableBufferSequence& buffers,
  266. BOOST_ASIO_MOVE_ARG(ReadHandler) handler)
  267. {
  268. // If you get an error on the following line it means that your handler does
  269. // not meet the documented type requirements for a ReadHandler.
  270. BOOST_ASIO_READ_HANDLER_CHECK(ReadHandler, handler) type_check;
  271. detail::async_result_init<
  272. ReadHandler, void (boost::system::error_code, std::size_t)> init(
  273. BOOST_ASIO_MOVE_CAST(ReadHandler)(handler));
  274. if (boost::asio::buffer_size(buffers) == 0 || !storage_.empty())
  275. {
  276. next_layer_.async_read_some(boost::asio::mutable_buffers_1(0, 0),
  277. detail::buffered_read_some_handler<
  278. MutableBufferSequence, BOOST_ASIO_HANDLER_TYPE(
  279. ReadHandler, void (boost::system::error_code, std::size_t))>(
  280. storage_, buffers, init.handler));
  281. }
  282. else
  283. {
  284. this->async_fill(detail::buffered_read_some_handler<
  285. MutableBufferSequence, BOOST_ASIO_HANDLER_TYPE(
  286. ReadHandler, void (boost::system::error_code, std::size_t))>(
  287. storage_, buffers, init.handler));
  288. }
  289. return init.result.get();
  290. }
  291. template <typename Stream>
  292. template <typename MutableBufferSequence>
  293. std::size_t buffered_read_stream<Stream>::peek(
  294. const MutableBufferSequence& buffers)
  295. {
  296. if (storage_.empty())
  297. this->fill();
  298. return this->peek_copy(buffers);
  299. }
  300. template <typename Stream>
  301. template <typename MutableBufferSequence>
  302. std::size_t buffered_read_stream<Stream>::peek(
  303. const MutableBufferSequence& buffers, boost::system::error_code& ec)
  304. {
  305. ec = boost::system::error_code();
  306. if (storage_.empty() && !this->fill(ec))
  307. return 0;
  308. return this->peek_copy(buffers);
  309. }
  310. } // namespace asio
  311. } // namespace boost
  312. #include <boost/asio/detail/pop_options.hpp>
  313. #endif // BOOST_ASIO_IMPL_BUFFERED_READ_STREAM_HPP