read.hpp 48 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278
  1. //
  2. // impl/read.hpp
  3. // ~~~~~~~~~~~~~
  4. //
  5. // Copyright (c) 2003-2023 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_READ_HPP
  11. #define BOOST_ASIO_IMPL_READ_HPP
  12. #if defined(_MSC_VER) && (_MSC_VER >= 1200)
  13. # pragma once
  14. #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
  15. #include <algorithm>
  16. #include <boost/asio/associator.hpp>
  17. #include <boost/asio/buffer.hpp>
  18. #include <boost/asio/detail/array_fwd.hpp>
  19. #include <boost/asio/detail/base_from_cancellation_state.hpp>
  20. #include <boost/asio/detail/base_from_completion_cond.hpp>
  21. #include <boost/asio/detail/bind_handler.hpp>
  22. #include <boost/asio/detail/consuming_buffers.hpp>
  23. #include <boost/asio/detail/dependent_type.hpp>
  24. #include <boost/asio/detail/handler_alloc_helpers.hpp>
  25. #include <boost/asio/detail/handler_cont_helpers.hpp>
  26. #include <boost/asio/detail/handler_invoke_helpers.hpp>
  27. #include <boost/asio/detail/handler_tracking.hpp>
  28. #include <boost/asio/detail/handler_type_requirements.hpp>
  29. #include <boost/asio/detail/non_const_lvalue.hpp>
  30. #include <boost/asio/detail/throw_error.hpp>
  31. #include <boost/asio/error.hpp>
  32. #include <boost/asio/detail/push_options.hpp>
  33. namespace boost {
  34. namespace asio {
  35. namespace detail
  36. {
  37. template <typename SyncReadStream, typename MutableBufferSequence,
  38. typename MutableBufferIterator, typename CompletionCondition>
  39. std::size_t read_buffer_seq(SyncReadStream& s,
  40. const MutableBufferSequence& buffers, const MutableBufferIterator&,
  41. CompletionCondition completion_condition, boost::system::error_code& ec)
  42. {
  43. ec = boost::system::error_code();
  44. boost::asio::detail::consuming_buffers<mutable_buffer,
  45. MutableBufferSequence, MutableBufferIterator> tmp(buffers);
  46. while (!tmp.empty())
  47. {
  48. if (std::size_t max_size = detail::adapt_completion_condition_result(
  49. completion_condition(ec, tmp.total_consumed())))
  50. tmp.consume(s.read_some(tmp.prepare(max_size), ec));
  51. else
  52. break;
  53. }
  54. return tmp.total_consumed();
  55. }
  56. } // namespace detail
  57. template <typename SyncReadStream, typename MutableBufferSequence,
  58. typename CompletionCondition>
  59. std::size_t read(SyncReadStream& s, const MutableBufferSequence& buffers,
  60. CompletionCondition completion_condition, boost::system::error_code& ec,
  61. typename constraint<
  62. is_mutable_buffer_sequence<MutableBufferSequence>::value
  63. >::type)
  64. {
  65. return detail::read_buffer_seq(s, buffers,
  66. boost::asio::buffer_sequence_begin(buffers),
  67. BOOST_ASIO_MOVE_CAST(CompletionCondition)(completion_condition), ec);
  68. }
  69. template <typename SyncReadStream, typename MutableBufferSequence>
  70. inline std::size_t read(SyncReadStream& s, const MutableBufferSequence& buffers,
  71. typename constraint<
  72. is_mutable_buffer_sequence<MutableBufferSequence>::value
  73. >::type)
  74. {
  75. boost::system::error_code ec;
  76. std::size_t bytes_transferred = read(s, buffers, transfer_all(), ec);
  77. boost::asio::detail::throw_error(ec, "read");
  78. return bytes_transferred;
  79. }
  80. template <typename SyncReadStream, typename MutableBufferSequence>
  81. inline std::size_t read(SyncReadStream& s, const MutableBufferSequence& buffers,
  82. boost::system::error_code& ec,
  83. typename constraint<
  84. is_mutable_buffer_sequence<MutableBufferSequence>::value
  85. >::type)
  86. {
  87. return read(s, buffers, transfer_all(), ec);
  88. }
  89. template <typename SyncReadStream, typename MutableBufferSequence,
  90. typename CompletionCondition>
  91. inline std::size_t read(SyncReadStream& s, const MutableBufferSequence& buffers,
  92. CompletionCondition completion_condition,
  93. typename constraint<
  94. is_mutable_buffer_sequence<MutableBufferSequence>::value
  95. >::type)
  96. {
  97. boost::system::error_code ec;
  98. std::size_t bytes_transferred = read(s, buffers,
  99. BOOST_ASIO_MOVE_CAST(CompletionCondition)(completion_condition), ec);
  100. boost::asio::detail::throw_error(ec, "read");
  101. return bytes_transferred;
  102. }
  103. #if !defined(BOOST_ASIO_NO_DYNAMIC_BUFFER_V1)
  104. template <typename SyncReadStream, typename DynamicBuffer_v1,
  105. typename CompletionCondition>
  106. std::size_t read(SyncReadStream& s,
  107. BOOST_ASIO_MOVE_ARG(DynamicBuffer_v1) buffers,
  108. CompletionCondition completion_condition, boost::system::error_code& ec,
  109. typename constraint<
  110. is_dynamic_buffer_v1<typename decay<DynamicBuffer_v1>::type>::value
  111. >::type,
  112. typename constraint<
  113. !is_dynamic_buffer_v2<typename decay<DynamicBuffer_v1>::type>::value
  114. >::type)
  115. {
  116. typename decay<DynamicBuffer_v1>::type b(
  117. BOOST_ASIO_MOVE_CAST(DynamicBuffer_v1)(buffers));
  118. ec = boost::system::error_code();
  119. std::size_t total_transferred = 0;
  120. std::size_t max_size = detail::adapt_completion_condition_result(
  121. completion_condition(ec, total_transferred));
  122. std::size_t bytes_available = std::min<std::size_t>(
  123. std::max<std::size_t>(512, b.capacity() - b.size()),
  124. std::min<std::size_t>(max_size, b.max_size() - b.size()));
  125. while (bytes_available > 0)
  126. {
  127. std::size_t bytes_transferred = s.read_some(b.prepare(bytes_available), ec);
  128. b.commit(bytes_transferred);
  129. total_transferred += bytes_transferred;
  130. max_size = detail::adapt_completion_condition_result(
  131. completion_condition(ec, total_transferred));
  132. bytes_available = std::min<std::size_t>(
  133. std::max<std::size_t>(512, b.capacity() - b.size()),
  134. std::min<std::size_t>(max_size, b.max_size() - b.size()));
  135. }
  136. return total_transferred;
  137. }
  138. template <typename SyncReadStream, typename DynamicBuffer_v1>
  139. inline std::size_t read(SyncReadStream& s,
  140. BOOST_ASIO_MOVE_ARG(DynamicBuffer_v1) buffers,
  141. typename constraint<
  142. is_dynamic_buffer_v1<typename decay<DynamicBuffer_v1>::type>::value
  143. >::type,
  144. typename constraint<
  145. !is_dynamic_buffer_v2<typename decay<DynamicBuffer_v1>::type>::value
  146. >::type)
  147. {
  148. boost::system::error_code ec;
  149. std::size_t bytes_transferred = read(s,
  150. BOOST_ASIO_MOVE_CAST(DynamicBuffer_v1)(buffers), transfer_all(), ec);
  151. boost::asio::detail::throw_error(ec, "read");
  152. return bytes_transferred;
  153. }
  154. template <typename SyncReadStream, typename DynamicBuffer_v1>
  155. inline std::size_t read(SyncReadStream& s,
  156. BOOST_ASIO_MOVE_ARG(DynamicBuffer_v1) buffers,
  157. boost::system::error_code& ec,
  158. typename constraint<
  159. is_dynamic_buffer_v1<typename decay<DynamicBuffer_v1>::type>::value
  160. >::type,
  161. typename constraint<
  162. !is_dynamic_buffer_v2<typename decay<DynamicBuffer_v1>::type>::value
  163. >::type)
  164. {
  165. return read(s, BOOST_ASIO_MOVE_CAST(DynamicBuffer_v1)(buffers),
  166. transfer_all(), ec);
  167. }
  168. template <typename SyncReadStream, typename DynamicBuffer_v1,
  169. typename CompletionCondition>
  170. inline std::size_t read(SyncReadStream& s,
  171. BOOST_ASIO_MOVE_ARG(DynamicBuffer_v1) buffers,
  172. CompletionCondition completion_condition,
  173. typename constraint<
  174. is_dynamic_buffer_v1<typename decay<DynamicBuffer_v1>::type>::value
  175. >::type,
  176. typename constraint<
  177. !is_dynamic_buffer_v2<typename decay<DynamicBuffer_v1>::type>::value
  178. >::type)
  179. {
  180. boost::system::error_code ec;
  181. std::size_t bytes_transferred = read(s,
  182. BOOST_ASIO_MOVE_CAST(DynamicBuffer_v1)(buffers),
  183. BOOST_ASIO_MOVE_CAST(CompletionCondition)(completion_condition), ec);
  184. boost::asio::detail::throw_error(ec, "read");
  185. return bytes_transferred;
  186. }
  187. #if !defined(BOOST_ASIO_NO_EXTENSIONS)
  188. #if !defined(BOOST_ASIO_NO_IOSTREAM)
  189. template <typename SyncReadStream, typename Allocator,
  190. typename CompletionCondition>
  191. inline std::size_t read(SyncReadStream& s,
  192. boost::asio::basic_streambuf<Allocator>& b,
  193. CompletionCondition completion_condition, boost::system::error_code& ec)
  194. {
  195. return read(s, basic_streambuf_ref<Allocator>(b),
  196. BOOST_ASIO_MOVE_CAST(CompletionCondition)(completion_condition), ec);
  197. }
  198. template <typename SyncReadStream, typename Allocator>
  199. inline std::size_t read(SyncReadStream& s,
  200. boost::asio::basic_streambuf<Allocator>& b)
  201. {
  202. return read(s, basic_streambuf_ref<Allocator>(b));
  203. }
  204. template <typename SyncReadStream, typename Allocator>
  205. inline std::size_t read(SyncReadStream& s,
  206. boost::asio::basic_streambuf<Allocator>& b,
  207. boost::system::error_code& ec)
  208. {
  209. return read(s, basic_streambuf_ref<Allocator>(b), ec);
  210. }
  211. template <typename SyncReadStream, typename Allocator,
  212. typename CompletionCondition>
  213. inline std::size_t read(SyncReadStream& s,
  214. boost::asio::basic_streambuf<Allocator>& b,
  215. CompletionCondition completion_condition)
  216. {
  217. return read(s, basic_streambuf_ref<Allocator>(b),
  218. BOOST_ASIO_MOVE_CAST(CompletionCondition)(completion_condition));
  219. }
  220. #endif // !defined(BOOST_ASIO_NO_IOSTREAM)
  221. #endif // !defined(BOOST_ASIO_NO_EXTENSIONS)
  222. #endif // !defined(BOOST_ASIO_NO_DYNAMIC_BUFFER_V1)
  223. template <typename SyncReadStream, typename DynamicBuffer_v2,
  224. typename CompletionCondition>
  225. std::size_t read(SyncReadStream& s, DynamicBuffer_v2 buffers,
  226. CompletionCondition completion_condition, boost::system::error_code& ec,
  227. typename constraint<
  228. is_dynamic_buffer_v2<DynamicBuffer_v2>::value
  229. >::type)
  230. {
  231. DynamicBuffer_v2& b = buffers;
  232. ec = boost::system::error_code();
  233. std::size_t total_transferred = 0;
  234. std::size_t max_size = detail::adapt_completion_condition_result(
  235. completion_condition(ec, total_transferred));
  236. std::size_t bytes_available = std::min<std::size_t>(
  237. std::max<std::size_t>(512, b.capacity() - b.size()),
  238. std::min<std::size_t>(max_size, b.max_size() - b.size()));
  239. while (bytes_available > 0)
  240. {
  241. std::size_t pos = b.size();
  242. b.grow(bytes_available);
  243. std::size_t bytes_transferred = s.read_some(
  244. b.data(pos, bytes_available), ec);
  245. b.shrink(bytes_available - bytes_transferred);
  246. total_transferred += bytes_transferred;
  247. max_size = detail::adapt_completion_condition_result(
  248. completion_condition(ec, total_transferred));
  249. bytes_available = std::min<std::size_t>(
  250. std::max<std::size_t>(512, b.capacity() - b.size()),
  251. std::min<std::size_t>(max_size, b.max_size() - b.size()));
  252. }
  253. return total_transferred;
  254. }
  255. template <typename SyncReadStream, typename DynamicBuffer_v2>
  256. inline std::size_t read(SyncReadStream& s, DynamicBuffer_v2 buffers,
  257. typename constraint<
  258. is_dynamic_buffer_v2<DynamicBuffer_v2>::value
  259. >::type)
  260. {
  261. boost::system::error_code ec;
  262. std::size_t bytes_transferred = read(s,
  263. BOOST_ASIO_MOVE_CAST(DynamicBuffer_v2)(buffers), transfer_all(), ec);
  264. boost::asio::detail::throw_error(ec, "read");
  265. return bytes_transferred;
  266. }
  267. template <typename SyncReadStream, typename DynamicBuffer_v2>
  268. inline std::size_t read(SyncReadStream& s, DynamicBuffer_v2 buffers,
  269. boost::system::error_code& ec,
  270. typename constraint<
  271. is_dynamic_buffer_v2<DynamicBuffer_v2>::value
  272. >::type)
  273. {
  274. return read(s, BOOST_ASIO_MOVE_CAST(DynamicBuffer_v2)(buffers),
  275. transfer_all(), ec);
  276. }
  277. template <typename SyncReadStream, typename DynamicBuffer_v2,
  278. typename CompletionCondition>
  279. inline std::size_t read(SyncReadStream& s, DynamicBuffer_v2 buffers,
  280. CompletionCondition completion_condition,
  281. typename constraint<
  282. is_dynamic_buffer_v2<DynamicBuffer_v2>::value
  283. >::type)
  284. {
  285. boost::system::error_code ec;
  286. std::size_t bytes_transferred = read(s,
  287. BOOST_ASIO_MOVE_CAST(DynamicBuffer_v2)(buffers),
  288. BOOST_ASIO_MOVE_CAST(CompletionCondition)(completion_condition), ec);
  289. boost::asio::detail::throw_error(ec, "read");
  290. return bytes_transferred;
  291. }
  292. namespace detail
  293. {
  294. template <typename AsyncReadStream, typename MutableBufferSequence,
  295. typename MutableBufferIterator, typename CompletionCondition,
  296. typename ReadHandler>
  297. class read_op
  298. : public base_from_cancellation_state<ReadHandler>,
  299. base_from_completion_cond<CompletionCondition>
  300. {
  301. public:
  302. read_op(AsyncReadStream& stream, const MutableBufferSequence& buffers,
  303. CompletionCondition& completion_condition, ReadHandler& handler)
  304. : base_from_cancellation_state<ReadHandler>(
  305. handler, enable_partial_cancellation()),
  306. base_from_completion_cond<CompletionCondition>(completion_condition),
  307. stream_(stream),
  308. buffers_(buffers),
  309. start_(0),
  310. handler_(BOOST_ASIO_MOVE_CAST(ReadHandler)(handler))
  311. {
  312. }
  313. #if defined(BOOST_ASIO_HAS_MOVE)
  314. read_op(const read_op& other)
  315. : base_from_cancellation_state<ReadHandler>(other),
  316. base_from_completion_cond<CompletionCondition>(other),
  317. stream_(other.stream_),
  318. buffers_(other.buffers_),
  319. start_(other.start_),
  320. handler_(other.handler_)
  321. {
  322. }
  323. read_op(read_op&& other)
  324. : base_from_cancellation_state<ReadHandler>(
  325. BOOST_ASIO_MOVE_CAST(base_from_cancellation_state<
  326. ReadHandler>)(other)),
  327. base_from_completion_cond<CompletionCondition>(
  328. BOOST_ASIO_MOVE_CAST(base_from_completion_cond<
  329. CompletionCondition>)(other)),
  330. stream_(other.stream_),
  331. buffers_(BOOST_ASIO_MOVE_CAST(buffers_type)(other.buffers_)),
  332. start_(other.start_),
  333. handler_(BOOST_ASIO_MOVE_CAST(ReadHandler)(other.handler_))
  334. {
  335. }
  336. #endif // defined(BOOST_ASIO_HAS_MOVE)
  337. void operator()(boost::system::error_code ec,
  338. std::size_t bytes_transferred, int start = 0)
  339. {
  340. std::size_t max_size;
  341. switch (start_ = start)
  342. {
  343. case 1:
  344. max_size = this->check_for_completion(ec, buffers_.total_consumed());
  345. for (;;)
  346. {
  347. {
  348. BOOST_ASIO_HANDLER_LOCATION((__FILE__, __LINE__, "async_read"));
  349. stream_.async_read_some(buffers_.prepare(max_size),
  350. BOOST_ASIO_MOVE_CAST(read_op)(*this));
  351. }
  352. return; default:
  353. buffers_.consume(bytes_transferred);
  354. if ((!ec && bytes_transferred == 0) || buffers_.empty())
  355. break;
  356. max_size = this->check_for_completion(ec, buffers_.total_consumed());
  357. if (max_size == 0)
  358. break;
  359. if (this->cancelled() != cancellation_type::none)
  360. {
  361. ec = error::operation_aborted;
  362. break;
  363. }
  364. }
  365. BOOST_ASIO_MOVE_OR_LVALUE(ReadHandler)(handler_)(
  366. static_cast<const boost::system::error_code&>(ec),
  367. static_cast<const std::size_t&>(buffers_.total_consumed()));
  368. }
  369. }
  370. //private:
  371. typedef boost::asio::detail::consuming_buffers<mutable_buffer,
  372. MutableBufferSequence, MutableBufferIterator> buffers_type;
  373. AsyncReadStream& stream_;
  374. buffers_type buffers_;
  375. int start_;
  376. ReadHandler handler_;
  377. };
  378. template <typename AsyncReadStream, typename MutableBufferSequence,
  379. typename MutableBufferIterator, typename CompletionCondition,
  380. typename ReadHandler>
  381. inline asio_handler_allocate_is_deprecated
  382. asio_handler_allocate(std::size_t size,
  383. read_op<AsyncReadStream, MutableBufferSequence, MutableBufferIterator,
  384. CompletionCondition, ReadHandler>* this_handler)
  385. {
  386. #if defined(BOOST_ASIO_NO_DEPRECATED)
  387. boost_asio_handler_alloc_helpers::allocate(size, this_handler->handler_);
  388. return asio_handler_allocate_is_no_longer_used();
  389. #else // defined(BOOST_ASIO_NO_DEPRECATED)
  390. return boost_asio_handler_alloc_helpers::allocate(
  391. size, this_handler->handler_);
  392. #endif // defined(BOOST_ASIO_NO_DEPRECATED)
  393. }
  394. template <typename AsyncReadStream, typename MutableBufferSequence,
  395. typename MutableBufferIterator, typename CompletionCondition,
  396. typename ReadHandler>
  397. inline asio_handler_deallocate_is_deprecated
  398. asio_handler_deallocate(void* pointer, std::size_t size,
  399. read_op<AsyncReadStream, MutableBufferSequence, MutableBufferIterator,
  400. CompletionCondition, ReadHandler>* this_handler)
  401. {
  402. boost_asio_handler_alloc_helpers::deallocate(
  403. pointer, size, this_handler->handler_);
  404. #if defined(BOOST_ASIO_NO_DEPRECATED)
  405. return asio_handler_deallocate_is_no_longer_used();
  406. #endif // defined(BOOST_ASIO_NO_DEPRECATED)
  407. }
  408. template <typename AsyncReadStream, typename MutableBufferSequence,
  409. typename MutableBufferIterator, typename CompletionCondition,
  410. typename ReadHandler>
  411. inline bool asio_handler_is_continuation(
  412. read_op<AsyncReadStream, MutableBufferSequence, MutableBufferIterator,
  413. CompletionCondition, ReadHandler>* this_handler)
  414. {
  415. return this_handler->start_ == 0 ? true
  416. : boost_asio_handler_cont_helpers::is_continuation(
  417. this_handler->handler_);
  418. }
  419. template <typename Function, typename AsyncReadStream,
  420. typename MutableBufferSequence, typename MutableBufferIterator,
  421. typename CompletionCondition, typename ReadHandler>
  422. inline asio_handler_invoke_is_deprecated
  423. asio_handler_invoke(Function& function,
  424. read_op<AsyncReadStream, MutableBufferSequence, MutableBufferIterator,
  425. CompletionCondition, ReadHandler>* this_handler)
  426. {
  427. boost_asio_handler_invoke_helpers::invoke(
  428. function, this_handler->handler_);
  429. #if defined(BOOST_ASIO_NO_DEPRECATED)
  430. return asio_handler_invoke_is_no_longer_used();
  431. #endif // defined(BOOST_ASIO_NO_DEPRECATED)
  432. }
  433. template <typename Function, typename AsyncReadStream,
  434. typename MutableBufferSequence, typename MutableBufferIterator,
  435. typename CompletionCondition, typename ReadHandler>
  436. inline asio_handler_invoke_is_deprecated
  437. asio_handler_invoke(const Function& function,
  438. read_op<AsyncReadStream, MutableBufferSequence, MutableBufferIterator,
  439. CompletionCondition, ReadHandler>* this_handler)
  440. {
  441. boost_asio_handler_invoke_helpers::invoke(
  442. function, this_handler->handler_);
  443. #if defined(BOOST_ASIO_NO_DEPRECATED)
  444. return asio_handler_invoke_is_no_longer_used();
  445. #endif // defined(BOOST_ASIO_NO_DEPRECATED)
  446. }
  447. template <typename AsyncReadStream, typename MutableBufferSequence,
  448. typename MutableBufferIterator, typename CompletionCondition,
  449. typename ReadHandler>
  450. inline void start_read_op(AsyncReadStream& stream,
  451. const MutableBufferSequence& buffers, const MutableBufferIterator&,
  452. CompletionCondition& completion_condition, ReadHandler& handler)
  453. {
  454. read_op<AsyncReadStream, MutableBufferSequence,
  455. MutableBufferIterator, CompletionCondition, ReadHandler>(
  456. stream, buffers, completion_condition, handler)(
  457. boost::system::error_code(), 0, 1);
  458. }
  459. template <typename AsyncReadStream>
  460. class initiate_async_read
  461. {
  462. public:
  463. typedef typename AsyncReadStream::executor_type executor_type;
  464. explicit initiate_async_read(AsyncReadStream& stream)
  465. : stream_(stream)
  466. {
  467. }
  468. executor_type get_executor() const BOOST_ASIO_NOEXCEPT
  469. {
  470. return stream_.get_executor();
  471. }
  472. template <typename ReadHandler, typename MutableBufferSequence,
  473. typename CompletionCondition>
  474. void operator()(BOOST_ASIO_MOVE_ARG(ReadHandler) handler,
  475. const MutableBufferSequence& buffers,
  476. BOOST_ASIO_MOVE_ARG(CompletionCondition) completion_cond) const
  477. {
  478. // If you get an error on the following line it means that your handler
  479. // does not meet the documented type requirements for a ReadHandler.
  480. BOOST_ASIO_READ_HANDLER_CHECK(ReadHandler, handler) type_check;
  481. non_const_lvalue<ReadHandler> handler2(handler);
  482. non_const_lvalue<CompletionCondition> completion_cond2(completion_cond);
  483. start_read_op(stream_, buffers,
  484. boost::asio::buffer_sequence_begin(buffers),
  485. completion_cond2.value, handler2.value);
  486. }
  487. private:
  488. AsyncReadStream& stream_;
  489. };
  490. } // namespace detail
  491. #if !defined(GENERATING_DOCUMENTATION)
  492. template <template <typename, typename> class Associator,
  493. typename AsyncReadStream, typename MutableBufferSequence,
  494. typename MutableBufferIterator, typename CompletionCondition,
  495. typename ReadHandler, typename DefaultCandidate>
  496. struct associator<Associator,
  497. detail::read_op<AsyncReadStream, MutableBufferSequence,
  498. MutableBufferIterator, CompletionCondition, ReadHandler>,
  499. DefaultCandidate>
  500. : Associator<ReadHandler, DefaultCandidate>
  501. {
  502. static typename Associator<ReadHandler, DefaultCandidate>::type
  503. get(const detail::read_op<AsyncReadStream, MutableBufferSequence,
  504. MutableBufferIterator, CompletionCondition, ReadHandler>& h)
  505. BOOST_ASIO_NOEXCEPT
  506. {
  507. return Associator<ReadHandler, DefaultCandidate>::get(h.handler_);
  508. }
  509. static BOOST_ASIO_AUTO_RETURN_TYPE_PREFIX2(
  510. typename Associator<ReadHandler, DefaultCandidate>::type)
  511. get(const detail::read_op<AsyncReadStream, MutableBufferSequence,
  512. MutableBufferIterator, CompletionCondition, ReadHandler>& h,
  513. const DefaultCandidate& c) BOOST_ASIO_NOEXCEPT
  514. BOOST_ASIO_AUTO_RETURN_TYPE_SUFFIX((
  515. Associator<ReadHandler, DefaultCandidate>::get(h.handler_, c)))
  516. {
  517. return Associator<ReadHandler, DefaultCandidate>::get(h.handler_, c);
  518. }
  519. };
  520. #endif // !defined(GENERATING_DOCUMENTATION)
  521. template <typename AsyncReadStream,
  522. typename MutableBufferSequence, typename CompletionCondition,
  523. BOOST_ASIO_COMPLETION_TOKEN_FOR(void (boost::system::error_code,
  524. std::size_t)) ReadToken>
  525. inline BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_PREFIX(ReadToken,
  526. void (boost::system::error_code, std::size_t))
  527. async_read(AsyncReadStream& s, const MutableBufferSequence& buffers,
  528. CompletionCondition completion_condition,
  529. BOOST_ASIO_MOVE_ARG(ReadToken) token,
  530. typename constraint<
  531. is_mutable_buffer_sequence<MutableBufferSequence>::value
  532. >::type)
  533. BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_SUFFIX((
  534. async_initiate<ReadToken,
  535. void (boost::system::error_code, std::size_t)>(
  536. declval<detail::initiate_async_read<AsyncReadStream> >(),
  537. token, buffers,
  538. BOOST_ASIO_MOVE_CAST(CompletionCondition)(completion_condition))))
  539. {
  540. return async_initiate<ReadToken,
  541. void (boost::system::error_code, std::size_t)>(
  542. detail::initiate_async_read<AsyncReadStream>(s),
  543. token, buffers,
  544. BOOST_ASIO_MOVE_CAST(CompletionCondition)(completion_condition));
  545. }
  546. template <typename AsyncReadStream, typename MutableBufferSequence,
  547. BOOST_ASIO_COMPLETION_TOKEN_FOR(void (boost::system::error_code,
  548. std::size_t)) ReadToken>
  549. inline BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_PREFIX(ReadToken,
  550. void (boost::system::error_code, std::size_t))
  551. async_read(AsyncReadStream& s, const MutableBufferSequence& buffers,
  552. BOOST_ASIO_MOVE_ARG(ReadToken) token,
  553. typename constraint<
  554. is_mutable_buffer_sequence<MutableBufferSequence>::value
  555. >::type)
  556. BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_SUFFIX((
  557. async_initiate<ReadToken,
  558. void (boost::system::error_code, std::size_t)>(
  559. declval<detail::initiate_async_read<AsyncReadStream> >(),
  560. token, buffers, transfer_all())))
  561. {
  562. return async_initiate<ReadToken,
  563. void (boost::system::error_code, std::size_t)>(
  564. detail::initiate_async_read<AsyncReadStream>(s),
  565. token, buffers, transfer_all());
  566. }
  567. #if !defined(BOOST_ASIO_NO_DYNAMIC_BUFFER_V1)
  568. namespace detail
  569. {
  570. template <typename AsyncReadStream, typename DynamicBuffer_v1,
  571. typename CompletionCondition, typename ReadHandler>
  572. class read_dynbuf_v1_op
  573. : public base_from_cancellation_state<ReadHandler>,
  574. base_from_completion_cond<CompletionCondition>
  575. {
  576. public:
  577. template <typename BufferSequence>
  578. read_dynbuf_v1_op(AsyncReadStream& stream,
  579. BOOST_ASIO_MOVE_ARG(BufferSequence) buffers,
  580. CompletionCondition& completion_condition, ReadHandler& handler)
  581. : base_from_cancellation_state<ReadHandler>(
  582. handler, enable_partial_cancellation()),
  583. base_from_completion_cond<CompletionCondition>(completion_condition),
  584. stream_(stream),
  585. buffers_(BOOST_ASIO_MOVE_CAST(BufferSequence)(buffers)),
  586. start_(0),
  587. total_transferred_(0),
  588. handler_(BOOST_ASIO_MOVE_CAST(ReadHandler)(handler))
  589. {
  590. }
  591. #if defined(BOOST_ASIO_HAS_MOVE)
  592. read_dynbuf_v1_op(const read_dynbuf_v1_op& other)
  593. : base_from_cancellation_state<ReadHandler>(other),
  594. base_from_completion_cond<CompletionCondition>(other),
  595. stream_(other.stream_),
  596. buffers_(other.buffers_),
  597. start_(other.start_),
  598. total_transferred_(other.total_transferred_),
  599. handler_(other.handler_)
  600. {
  601. }
  602. read_dynbuf_v1_op(read_dynbuf_v1_op&& other)
  603. : base_from_cancellation_state<ReadHandler>(
  604. BOOST_ASIO_MOVE_CAST(base_from_cancellation_state<
  605. ReadHandler>)(other)),
  606. base_from_completion_cond<CompletionCondition>(
  607. BOOST_ASIO_MOVE_CAST(base_from_completion_cond<
  608. CompletionCondition>)(other)),
  609. stream_(other.stream_),
  610. buffers_(BOOST_ASIO_MOVE_CAST(DynamicBuffer_v1)(other.buffers_)),
  611. start_(other.start_),
  612. total_transferred_(other.total_transferred_),
  613. handler_(BOOST_ASIO_MOVE_CAST(ReadHandler)(other.handler_))
  614. {
  615. }
  616. #endif // defined(BOOST_ASIO_HAS_MOVE)
  617. void operator()(boost::system::error_code ec,
  618. std::size_t bytes_transferred, int start = 0)
  619. {
  620. std::size_t max_size, bytes_available;
  621. switch (start_ = start)
  622. {
  623. case 1:
  624. max_size = this->check_for_completion(ec, total_transferred_);
  625. bytes_available = std::min<std::size_t>(
  626. std::max<std::size_t>(512,
  627. buffers_.capacity() - buffers_.size()),
  628. std::min<std::size_t>(max_size,
  629. buffers_.max_size() - buffers_.size()));
  630. for (;;)
  631. {
  632. {
  633. BOOST_ASIO_HANDLER_LOCATION((__FILE__, __LINE__, "async_read"));
  634. stream_.async_read_some(buffers_.prepare(bytes_available),
  635. BOOST_ASIO_MOVE_CAST(read_dynbuf_v1_op)(*this));
  636. }
  637. return; default:
  638. total_transferred_ += bytes_transferred;
  639. buffers_.commit(bytes_transferred);
  640. max_size = this->check_for_completion(ec, total_transferred_);
  641. bytes_available = std::min<std::size_t>(
  642. std::max<std::size_t>(512,
  643. buffers_.capacity() - buffers_.size()),
  644. std::min<std::size_t>(max_size,
  645. buffers_.max_size() - buffers_.size()));
  646. if ((!ec && bytes_transferred == 0) || bytes_available == 0)
  647. break;
  648. if (this->cancelled() != cancellation_type::none)
  649. {
  650. ec = error::operation_aborted;
  651. break;
  652. }
  653. }
  654. BOOST_ASIO_MOVE_OR_LVALUE(ReadHandler)(handler_)(
  655. static_cast<const boost::system::error_code&>(ec),
  656. static_cast<const std::size_t&>(total_transferred_));
  657. }
  658. }
  659. //private:
  660. AsyncReadStream& stream_;
  661. DynamicBuffer_v1 buffers_;
  662. int start_;
  663. std::size_t total_transferred_;
  664. ReadHandler handler_;
  665. };
  666. template <typename AsyncReadStream, typename DynamicBuffer_v1,
  667. typename CompletionCondition, typename ReadHandler>
  668. inline asio_handler_allocate_is_deprecated
  669. asio_handler_allocate(std::size_t size,
  670. read_dynbuf_v1_op<AsyncReadStream, DynamicBuffer_v1,
  671. CompletionCondition, ReadHandler>* this_handler)
  672. {
  673. #if defined(BOOST_ASIO_NO_DEPRECATED)
  674. boost_asio_handler_alloc_helpers::allocate(size, this_handler->handler_);
  675. return asio_handler_allocate_is_no_longer_used();
  676. #else // defined(BOOST_ASIO_NO_DEPRECATED)
  677. return boost_asio_handler_alloc_helpers::allocate(
  678. size, this_handler->handler_);
  679. #endif // defined(BOOST_ASIO_NO_DEPRECATED)
  680. }
  681. template <typename AsyncReadStream, typename DynamicBuffer_v1,
  682. typename CompletionCondition, typename ReadHandler>
  683. inline asio_handler_deallocate_is_deprecated
  684. asio_handler_deallocate(void* pointer, std::size_t size,
  685. read_dynbuf_v1_op<AsyncReadStream, DynamicBuffer_v1,
  686. CompletionCondition, ReadHandler>* this_handler)
  687. {
  688. boost_asio_handler_alloc_helpers::deallocate(
  689. pointer, size, this_handler->handler_);
  690. #if defined(BOOST_ASIO_NO_DEPRECATED)
  691. return asio_handler_deallocate_is_no_longer_used();
  692. #endif // defined(BOOST_ASIO_NO_DEPRECATED)
  693. }
  694. template <typename AsyncReadStream, typename DynamicBuffer_v1,
  695. typename CompletionCondition, typename ReadHandler>
  696. inline bool asio_handler_is_continuation(
  697. read_dynbuf_v1_op<AsyncReadStream, DynamicBuffer_v1,
  698. CompletionCondition, ReadHandler>* this_handler)
  699. {
  700. return this_handler->start_ == 0 ? true
  701. : boost_asio_handler_cont_helpers::is_continuation(
  702. this_handler->handler_);
  703. }
  704. template <typename Function, typename AsyncReadStream,
  705. typename DynamicBuffer_v1, typename CompletionCondition,
  706. typename ReadHandler>
  707. inline asio_handler_invoke_is_deprecated
  708. asio_handler_invoke(Function& function,
  709. read_dynbuf_v1_op<AsyncReadStream, DynamicBuffer_v1,
  710. CompletionCondition, ReadHandler>* this_handler)
  711. {
  712. boost_asio_handler_invoke_helpers::invoke(
  713. function, this_handler->handler_);
  714. #if defined(BOOST_ASIO_NO_DEPRECATED)
  715. return asio_handler_invoke_is_no_longer_used();
  716. #endif // defined(BOOST_ASIO_NO_DEPRECATED)
  717. }
  718. template <typename Function, typename AsyncReadStream,
  719. typename DynamicBuffer_v1, typename CompletionCondition,
  720. typename ReadHandler>
  721. inline asio_handler_invoke_is_deprecated
  722. asio_handler_invoke(const Function& function,
  723. read_dynbuf_v1_op<AsyncReadStream, DynamicBuffer_v1,
  724. CompletionCondition, ReadHandler>* this_handler)
  725. {
  726. boost_asio_handler_invoke_helpers::invoke(
  727. function, this_handler->handler_);
  728. #if defined(BOOST_ASIO_NO_DEPRECATED)
  729. return asio_handler_invoke_is_no_longer_used();
  730. #endif // defined(BOOST_ASIO_NO_DEPRECATED)
  731. }
  732. template <typename AsyncReadStream>
  733. class initiate_async_read_dynbuf_v1
  734. {
  735. public:
  736. typedef typename AsyncReadStream::executor_type executor_type;
  737. explicit initiate_async_read_dynbuf_v1(AsyncReadStream& stream)
  738. : stream_(stream)
  739. {
  740. }
  741. executor_type get_executor() const BOOST_ASIO_NOEXCEPT
  742. {
  743. return stream_.get_executor();
  744. }
  745. template <typename ReadHandler, typename DynamicBuffer_v1,
  746. typename CompletionCondition>
  747. void operator()(BOOST_ASIO_MOVE_ARG(ReadHandler) handler,
  748. BOOST_ASIO_MOVE_ARG(DynamicBuffer_v1) buffers,
  749. BOOST_ASIO_MOVE_ARG(CompletionCondition) completion_cond) const
  750. {
  751. // If you get an error on the following line it means that your handler
  752. // does not meet the documented type requirements for a ReadHandler.
  753. BOOST_ASIO_READ_HANDLER_CHECK(ReadHandler, handler) type_check;
  754. non_const_lvalue<ReadHandler> handler2(handler);
  755. non_const_lvalue<CompletionCondition> completion_cond2(completion_cond);
  756. read_dynbuf_v1_op<AsyncReadStream, typename decay<DynamicBuffer_v1>::type,
  757. CompletionCondition, typename decay<ReadHandler>::type>(
  758. stream_, BOOST_ASIO_MOVE_CAST(DynamicBuffer_v1)(buffers),
  759. completion_cond2.value, handler2.value)(
  760. boost::system::error_code(), 0, 1);
  761. }
  762. private:
  763. AsyncReadStream& stream_;
  764. };
  765. } // namespace detail
  766. #if !defined(GENERATING_DOCUMENTATION)
  767. template <template <typename, typename> class Associator,
  768. typename AsyncReadStream, typename DynamicBuffer_v1,
  769. typename CompletionCondition, typename ReadHandler,
  770. typename DefaultCandidate>
  771. struct associator<Associator,
  772. detail::read_dynbuf_v1_op<AsyncReadStream,
  773. DynamicBuffer_v1, CompletionCondition, ReadHandler>,
  774. DefaultCandidate>
  775. : Associator<ReadHandler, DefaultCandidate>
  776. {
  777. static typename Associator<ReadHandler, DefaultCandidate>::type
  778. get(const detail::read_dynbuf_v1_op<AsyncReadStream, DynamicBuffer_v1,
  779. CompletionCondition, ReadHandler>& h) BOOST_ASIO_NOEXCEPT
  780. {
  781. return Associator<ReadHandler, DefaultCandidate>::get(h.handler_);
  782. }
  783. static BOOST_ASIO_AUTO_RETURN_TYPE_PREFIX2(
  784. typename Associator<ReadHandler, DefaultCandidate>::type)
  785. get(const detail::read_dynbuf_v1_op<AsyncReadStream,
  786. DynamicBuffer_v1, CompletionCondition, ReadHandler>& h,
  787. const DefaultCandidate& c) BOOST_ASIO_NOEXCEPT
  788. BOOST_ASIO_AUTO_RETURN_TYPE_SUFFIX((
  789. Associator<ReadHandler, DefaultCandidate>::get(h.handler_, c)))
  790. {
  791. return Associator<ReadHandler, DefaultCandidate>::get(h.handler_, c);
  792. }
  793. };
  794. #endif // !defined(GENERATING_DOCUMENTATION)
  795. template <typename AsyncReadStream, typename DynamicBuffer_v1,
  796. BOOST_ASIO_COMPLETION_TOKEN_FOR(void (boost::system::error_code,
  797. std::size_t)) ReadToken>
  798. inline BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_PREFIX(ReadToken,
  799. void (boost::system::error_code, std::size_t))
  800. async_read(AsyncReadStream& s,
  801. BOOST_ASIO_MOVE_ARG(DynamicBuffer_v1) buffers,
  802. BOOST_ASIO_MOVE_ARG(ReadToken) token,
  803. typename constraint<
  804. is_dynamic_buffer_v1<typename decay<DynamicBuffer_v1>::type>::value
  805. >::type,
  806. typename constraint<
  807. !is_dynamic_buffer_v2<typename decay<DynamicBuffer_v1>::type>::value
  808. >::type)
  809. BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_SUFFIX((
  810. async_initiate<ReadToken,
  811. void (boost::system::error_code, std::size_t)>(
  812. declval<detail::initiate_async_read_dynbuf_v1<AsyncReadStream> >(),
  813. token, BOOST_ASIO_MOVE_CAST(DynamicBuffer_v1)(buffers),
  814. transfer_all())))
  815. {
  816. return async_initiate<ReadToken,
  817. void (boost::system::error_code, std::size_t)>(
  818. detail::initiate_async_read_dynbuf_v1<AsyncReadStream>(s),
  819. token, BOOST_ASIO_MOVE_CAST(DynamicBuffer_v1)(buffers),
  820. transfer_all());
  821. }
  822. template <typename AsyncReadStream,
  823. typename DynamicBuffer_v1, typename CompletionCondition,
  824. BOOST_ASIO_COMPLETION_TOKEN_FOR(void (boost::system::error_code,
  825. std::size_t)) ReadToken>
  826. inline BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_PREFIX(ReadToken,
  827. void (boost::system::error_code, std::size_t))
  828. async_read(AsyncReadStream& s,
  829. BOOST_ASIO_MOVE_ARG(DynamicBuffer_v1) buffers,
  830. CompletionCondition completion_condition,
  831. BOOST_ASIO_MOVE_ARG(ReadToken) token,
  832. typename constraint<
  833. is_dynamic_buffer_v1<typename decay<DynamicBuffer_v1>::type>::value
  834. >::type,
  835. typename constraint<
  836. !is_dynamic_buffer_v2<typename decay<DynamicBuffer_v1>::type>::value
  837. >::type)
  838. BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_SUFFIX((
  839. async_initiate<ReadToken,
  840. void (boost::system::error_code, std::size_t)>(
  841. declval<detail::initiate_async_read_dynbuf_v1<AsyncReadStream> >(),
  842. token, BOOST_ASIO_MOVE_CAST(DynamicBuffer_v1)(buffers),
  843. BOOST_ASIO_MOVE_CAST(CompletionCondition)(completion_condition))))
  844. {
  845. return async_initiate<ReadToken,
  846. void (boost::system::error_code, std::size_t)>(
  847. detail::initiate_async_read_dynbuf_v1<AsyncReadStream>(s),
  848. token, BOOST_ASIO_MOVE_CAST(DynamicBuffer_v1)(buffers),
  849. BOOST_ASIO_MOVE_CAST(CompletionCondition)(completion_condition));
  850. }
  851. #if !defined(BOOST_ASIO_NO_EXTENSIONS)
  852. #if !defined(BOOST_ASIO_NO_IOSTREAM)
  853. template <typename AsyncReadStream, typename Allocator,
  854. BOOST_ASIO_COMPLETION_TOKEN_FOR(void (boost::system::error_code,
  855. std::size_t)) ReadToken>
  856. inline BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_PREFIX(ReadToken,
  857. void (boost::system::error_code, std::size_t))
  858. async_read(AsyncReadStream& s, basic_streambuf<Allocator>& b,
  859. BOOST_ASIO_MOVE_ARG(ReadToken) token)
  860. BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_SUFFIX((
  861. async_read(s, basic_streambuf_ref<Allocator>(b),
  862. BOOST_ASIO_MOVE_CAST(ReadToken)(token))))
  863. {
  864. return async_read(s, basic_streambuf_ref<Allocator>(b),
  865. BOOST_ASIO_MOVE_CAST(ReadToken)(token));
  866. }
  867. template <typename AsyncReadStream,
  868. typename Allocator, typename CompletionCondition,
  869. BOOST_ASIO_COMPLETION_TOKEN_FOR(void (boost::system::error_code,
  870. std::size_t)) ReadToken>
  871. inline BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_PREFIX(ReadToken,
  872. void (boost::system::error_code, std::size_t))
  873. async_read(AsyncReadStream& s, basic_streambuf<Allocator>& b,
  874. CompletionCondition completion_condition,
  875. BOOST_ASIO_MOVE_ARG(ReadToken) token)
  876. BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_SUFFIX((
  877. async_read(s, basic_streambuf_ref<Allocator>(b),
  878. BOOST_ASIO_MOVE_CAST(CompletionCondition)(completion_condition),
  879. BOOST_ASIO_MOVE_CAST(ReadToken)(token))))
  880. {
  881. return async_read(s, basic_streambuf_ref<Allocator>(b),
  882. BOOST_ASIO_MOVE_CAST(CompletionCondition)(completion_condition),
  883. BOOST_ASIO_MOVE_CAST(ReadToken)(token));
  884. }
  885. #endif // !defined(BOOST_ASIO_NO_IOSTREAM)
  886. #endif // !defined(BOOST_ASIO_NO_EXTENSIONS)
  887. #endif // !defined(BOOST_ASIO_NO_DYNAMIC_BUFFER_V1)
  888. namespace detail
  889. {
  890. template <typename AsyncReadStream, typename DynamicBuffer_v2,
  891. typename CompletionCondition, typename ReadHandler>
  892. class read_dynbuf_v2_op
  893. : public base_from_cancellation_state<ReadHandler>,
  894. base_from_completion_cond<CompletionCondition>
  895. {
  896. public:
  897. template <typename BufferSequence>
  898. read_dynbuf_v2_op(AsyncReadStream& stream,
  899. BOOST_ASIO_MOVE_ARG(BufferSequence) buffers,
  900. CompletionCondition& completion_condition, ReadHandler& handler)
  901. : base_from_cancellation_state<ReadHandler>(
  902. handler, enable_partial_cancellation()),
  903. base_from_completion_cond<CompletionCondition>(completion_condition),
  904. stream_(stream),
  905. buffers_(BOOST_ASIO_MOVE_CAST(BufferSequence)(buffers)),
  906. start_(0),
  907. total_transferred_(0),
  908. bytes_available_(0),
  909. handler_(BOOST_ASIO_MOVE_CAST(ReadHandler)(handler))
  910. {
  911. }
  912. #if defined(BOOST_ASIO_HAS_MOVE)
  913. read_dynbuf_v2_op(const read_dynbuf_v2_op& other)
  914. : base_from_cancellation_state<ReadHandler>(other),
  915. base_from_completion_cond<CompletionCondition>(other),
  916. stream_(other.stream_),
  917. buffers_(other.buffers_),
  918. start_(other.start_),
  919. total_transferred_(other.total_transferred_),
  920. bytes_available_(other.bytes_available_),
  921. handler_(other.handler_)
  922. {
  923. }
  924. read_dynbuf_v2_op(read_dynbuf_v2_op&& other)
  925. : base_from_cancellation_state<ReadHandler>(
  926. BOOST_ASIO_MOVE_CAST(base_from_cancellation_state<
  927. ReadHandler>)(other)),
  928. base_from_completion_cond<CompletionCondition>(
  929. BOOST_ASIO_MOVE_CAST(base_from_completion_cond<
  930. CompletionCondition>)(other)),
  931. stream_(other.stream_),
  932. buffers_(BOOST_ASIO_MOVE_CAST(DynamicBuffer_v2)(other.buffers_)),
  933. start_(other.start_),
  934. total_transferred_(other.total_transferred_),
  935. bytes_available_(other.bytes_available_),
  936. handler_(BOOST_ASIO_MOVE_CAST(ReadHandler)(other.handler_))
  937. {
  938. }
  939. #endif // defined(BOOST_ASIO_HAS_MOVE)
  940. void operator()(boost::system::error_code ec,
  941. std::size_t bytes_transferred, int start = 0)
  942. {
  943. std::size_t max_size, pos;
  944. switch (start_ = start)
  945. {
  946. case 1:
  947. max_size = this->check_for_completion(ec, total_transferred_);
  948. bytes_available_ = std::min<std::size_t>(
  949. std::max<std::size_t>(512,
  950. buffers_.capacity() - buffers_.size()),
  951. std::min<std::size_t>(max_size,
  952. buffers_.max_size() - buffers_.size()));
  953. for (;;)
  954. {
  955. pos = buffers_.size();
  956. buffers_.grow(bytes_available_);
  957. {
  958. BOOST_ASIO_HANDLER_LOCATION((__FILE__, __LINE__, "async_read"));
  959. stream_.async_read_some(buffers_.data(pos, bytes_available_),
  960. BOOST_ASIO_MOVE_CAST(read_dynbuf_v2_op)(*this));
  961. }
  962. return; default:
  963. total_transferred_ += bytes_transferred;
  964. buffers_.shrink(bytes_available_ - bytes_transferred);
  965. max_size = this->check_for_completion(ec, total_transferred_);
  966. bytes_available_ = std::min<std::size_t>(
  967. std::max<std::size_t>(512,
  968. buffers_.capacity() - buffers_.size()),
  969. std::min<std::size_t>(max_size,
  970. buffers_.max_size() - buffers_.size()));
  971. if ((!ec && bytes_transferred == 0) || bytes_available_ == 0)
  972. break;
  973. if (this->cancelled() != cancellation_type::none)
  974. {
  975. ec = error::operation_aborted;
  976. break;
  977. }
  978. }
  979. BOOST_ASIO_MOVE_OR_LVALUE(ReadHandler)(handler_)(
  980. static_cast<const boost::system::error_code&>(ec),
  981. static_cast<const std::size_t&>(total_transferred_));
  982. }
  983. }
  984. //private:
  985. AsyncReadStream& stream_;
  986. DynamicBuffer_v2 buffers_;
  987. int start_;
  988. std::size_t total_transferred_;
  989. std::size_t bytes_available_;
  990. ReadHandler handler_;
  991. };
  992. template <typename AsyncReadStream, typename DynamicBuffer_v2,
  993. typename CompletionCondition, typename ReadHandler>
  994. inline asio_handler_allocate_is_deprecated
  995. asio_handler_allocate(std::size_t size,
  996. read_dynbuf_v2_op<AsyncReadStream, DynamicBuffer_v2,
  997. CompletionCondition, ReadHandler>* this_handler)
  998. {
  999. #if defined(BOOST_ASIO_NO_DEPRECATED)
  1000. boost_asio_handler_alloc_helpers::allocate(size, this_handler->handler_);
  1001. return asio_handler_allocate_is_no_longer_used();
  1002. #else // defined(BOOST_ASIO_NO_DEPRECATED)
  1003. return boost_asio_handler_alloc_helpers::allocate(
  1004. size, this_handler->handler_);
  1005. #endif // defined(BOOST_ASIO_NO_DEPRECATED)
  1006. }
  1007. template <typename AsyncReadStream, typename DynamicBuffer_v2,
  1008. typename CompletionCondition, typename ReadHandler>
  1009. inline asio_handler_deallocate_is_deprecated
  1010. asio_handler_deallocate(void* pointer, std::size_t size,
  1011. read_dynbuf_v2_op<AsyncReadStream, DynamicBuffer_v2,
  1012. CompletionCondition, ReadHandler>* this_handler)
  1013. {
  1014. boost_asio_handler_alloc_helpers::deallocate(
  1015. pointer, size, this_handler->handler_);
  1016. #if defined(BOOST_ASIO_NO_DEPRECATED)
  1017. return asio_handler_deallocate_is_no_longer_used();
  1018. #endif // defined(BOOST_ASIO_NO_DEPRECATED)
  1019. }
  1020. template <typename AsyncReadStream, typename DynamicBuffer_v2,
  1021. typename CompletionCondition, typename ReadHandler>
  1022. inline bool asio_handler_is_continuation(
  1023. read_dynbuf_v2_op<AsyncReadStream, DynamicBuffer_v2,
  1024. CompletionCondition, ReadHandler>* this_handler)
  1025. {
  1026. return this_handler->start_ == 0 ? true
  1027. : boost_asio_handler_cont_helpers::is_continuation(
  1028. this_handler->handler_);
  1029. }
  1030. template <typename Function, typename AsyncReadStream,
  1031. typename DynamicBuffer_v2, typename CompletionCondition,
  1032. typename ReadHandler>
  1033. inline asio_handler_invoke_is_deprecated
  1034. asio_handler_invoke(Function& function,
  1035. read_dynbuf_v2_op<AsyncReadStream, DynamicBuffer_v2,
  1036. CompletionCondition, ReadHandler>* this_handler)
  1037. {
  1038. boost_asio_handler_invoke_helpers::invoke(
  1039. function, this_handler->handler_);
  1040. #if defined(BOOST_ASIO_NO_DEPRECATED)
  1041. return asio_handler_invoke_is_no_longer_used();
  1042. #endif // defined(BOOST_ASIO_NO_DEPRECATED)
  1043. }
  1044. template <typename Function, typename AsyncReadStream,
  1045. typename DynamicBuffer_v2, typename CompletionCondition,
  1046. typename ReadHandler>
  1047. inline asio_handler_invoke_is_deprecated
  1048. asio_handler_invoke(const Function& function,
  1049. read_dynbuf_v2_op<AsyncReadStream, DynamicBuffer_v2,
  1050. CompletionCondition, ReadHandler>* this_handler)
  1051. {
  1052. boost_asio_handler_invoke_helpers::invoke(
  1053. function, this_handler->handler_);
  1054. #if defined(BOOST_ASIO_NO_DEPRECATED)
  1055. return asio_handler_invoke_is_no_longer_used();
  1056. #endif // defined(BOOST_ASIO_NO_DEPRECATED)
  1057. }
  1058. template <typename AsyncReadStream>
  1059. class initiate_async_read_dynbuf_v2
  1060. {
  1061. public:
  1062. typedef typename AsyncReadStream::executor_type executor_type;
  1063. explicit initiate_async_read_dynbuf_v2(AsyncReadStream& stream)
  1064. : stream_(stream)
  1065. {
  1066. }
  1067. executor_type get_executor() const BOOST_ASIO_NOEXCEPT
  1068. {
  1069. return stream_.get_executor();
  1070. }
  1071. template <typename ReadHandler, typename DynamicBuffer_v2,
  1072. typename CompletionCondition>
  1073. void operator()(BOOST_ASIO_MOVE_ARG(ReadHandler) handler,
  1074. BOOST_ASIO_MOVE_ARG(DynamicBuffer_v2) buffers,
  1075. BOOST_ASIO_MOVE_ARG(CompletionCondition) completion_cond) const
  1076. {
  1077. // If you get an error on the following line it means that your handler
  1078. // does not meet the documented type requirements for a ReadHandler.
  1079. BOOST_ASIO_READ_HANDLER_CHECK(ReadHandler, handler) type_check;
  1080. non_const_lvalue<ReadHandler> handler2(handler);
  1081. non_const_lvalue<CompletionCondition> completion_cond2(completion_cond);
  1082. read_dynbuf_v2_op<AsyncReadStream, typename decay<DynamicBuffer_v2>::type,
  1083. CompletionCondition, typename decay<ReadHandler>::type>(
  1084. stream_, BOOST_ASIO_MOVE_CAST(DynamicBuffer_v2)(buffers),
  1085. completion_cond2.value, handler2.value)(
  1086. boost::system::error_code(), 0, 1);
  1087. }
  1088. private:
  1089. AsyncReadStream& stream_;
  1090. };
  1091. } // namespace detail
  1092. #if !defined(GENERATING_DOCUMENTATION)
  1093. template <template <typename, typename> class Associator,
  1094. typename AsyncReadStream, typename DynamicBuffer_v2,
  1095. typename CompletionCondition, typename ReadHandler,
  1096. typename DefaultCandidate>
  1097. struct associator<Associator,
  1098. detail::read_dynbuf_v2_op<AsyncReadStream,
  1099. DynamicBuffer_v2, CompletionCondition, ReadHandler>,
  1100. DefaultCandidate>
  1101. : Associator<ReadHandler, DefaultCandidate>
  1102. {
  1103. static typename Associator<ReadHandler, DefaultCandidate>::type
  1104. get(const detail::read_dynbuf_v2_op<AsyncReadStream, DynamicBuffer_v2,
  1105. CompletionCondition, ReadHandler>& h) BOOST_ASIO_NOEXCEPT
  1106. {
  1107. return Associator<ReadHandler, DefaultCandidate>::get(h.handler_);
  1108. }
  1109. static BOOST_ASIO_AUTO_RETURN_TYPE_PREFIX2(
  1110. typename Associator<ReadHandler, DefaultCandidate>::type)
  1111. get(const detail::read_dynbuf_v2_op<AsyncReadStream,
  1112. DynamicBuffer_v2, CompletionCondition, ReadHandler>& h,
  1113. const DefaultCandidate& c) BOOST_ASIO_NOEXCEPT
  1114. BOOST_ASIO_AUTO_RETURN_TYPE_SUFFIX((
  1115. Associator<ReadHandler, DefaultCandidate>::get(h.handler_, c)))
  1116. {
  1117. return Associator<ReadHandler, DefaultCandidate>::get(h.handler_, c);
  1118. }
  1119. };
  1120. #endif // !defined(GENERATING_DOCUMENTATION)
  1121. template <typename AsyncReadStream, typename DynamicBuffer_v2,
  1122. BOOST_ASIO_COMPLETION_TOKEN_FOR(void (boost::system::error_code,
  1123. std::size_t)) ReadToken>
  1124. inline BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_PREFIX(ReadToken,
  1125. void (boost::system::error_code, std::size_t))
  1126. async_read(AsyncReadStream& s, DynamicBuffer_v2 buffers,
  1127. BOOST_ASIO_MOVE_ARG(ReadToken) token,
  1128. typename constraint<
  1129. is_dynamic_buffer_v2<DynamicBuffer_v2>::value
  1130. >::type)
  1131. BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_SUFFIX((
  1132. async_initiate<ReadToken,
  1133. void (boost::system::error_code, std::size_t)>(
  1134. declval<detail::initiate_async_read_dynbuf_v2<AsyncReadStream> >(),
  1135. token, BOOST_ASIO_MOVE_CAST(DynamicBuffer_v2)(buffers),
  1136. transfer_all())))
  1137. {
  1138. return async_initiate<ReadToken,
  1139. void (boost::system::error_code, std::size_t)>(
  1140. detail::initiate_async_read_dynbuf_v2<AsyncReadStream>(s),
  1141. token, BOOST_ASIO_MOVE_CAST(DynamicBuffer_v2)(buffers),
  1142. transfer_all());
  1143. }
  1144. template <typename AsyncReadStream,
  1145. typename DynamicBuffer_v2, typename CompletionCondition,
  1146. BOOST_ASIO_COMPLETION_TOKEN_FOR(void (boost::system::error_code,
  1147. std::size_t)) ReadToken>
  1148. inline BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_PREFIX(ReadToken,
  1149. void (boost::system::error_code, std::size_t))
  1150. async_read(AsyncReadStream& s, DynamicBuffer_v2 buffers,
  1151. CompletionCondition completion_condition,
  1152. BOOST_ASIO_MOVE_ARG(ReadToken) token,
  1153. typename constraint<
  1154. is_dynamic_buffer_v2<DynamicBuffer_v2>::value
  1155. >::type)
  1156. BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_SUFFIX((
  1157. async_initiate<ReadToken,
  1158. void (boost::system::error_code, std::size_t)>(
  1159. declval<detail::initiate_async_read_dynbuf_v2<AsyncReadStream> >(),
  1160. token, BOOST_ASIO_MOVE_CAST(DynamicBuffer_v2)(buffers),
  1161. BOOST_ASIO_MOVE_CAST(CompletionCondition)(completion_condition))))
  1162. {
  1163. return async_initiate<ReadToken,
  1164. void (boost::system::error_code, std::size_t)>(
  1165. detail::initiate_async_read_dynbuf_v2<AsyncReadStream>(s),
  1166. token, BOOST_ASIO_MOVE_CAST(DynamicBuffer_v2)(buffers),
  1167. BOOST_ASIO_MOVE_CAST(CompletionCondition)(completion_condition));
  1168. }
  1169. } // namespace asio
  1170. } // namespace boost
  1171. #include <boost/asio/detail/pop_options.hpp>
  1172. #endif // BOOST_ASIO_IMPL_READ_HPP