buffer_sequence_adapter.hpp 10.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385
  1. //
  2. // detail/buffer_sequence_adapter.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_DETAIL_BUFFER_SEQUENCE_ADAPTER_HPP
  11. #define BOOST_ASIO_DETAIL_BUFFER_SEQUENCE_ADAPTER_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/config.hpp>
  16. #include <boost/asio/buffer.hpp>
  17. #include <boost/asio/detail/array_fwd.hpp>
  18. #include <boost/asio/detail/socket_types.hpp>
  19. #include <boost/asio/detail/push_options.hpp>
  20. namespace boost {
  21. namespace asio {
  22. namespace detail {
  23. class buffer_sequence_adapter_base
  24. {
  25. protected:
  26. #if defined(BOOST_ASIO_WINDOWS_RUNTIME)
  27. // The maximum number of buffers to support in a single operation.
  28. enum { max_buffers = 1 };
  29. typedef Windows::Storage::Streams::IBuffer^ native_buffer_type;
  30. BOOST_ASIO_DECL static void init_native_buffer(
  31. native_buffer_type& buf,
  32. const boost::asio::mutable_buffer& buffer);
  33. BOOST_ASIO_DECL static void init_native_buffer(
  34. native_buffer_type& buf,
  35. const boost::asio::const_buffer& buffer);
  36. #elif defined(BOOST_ASIO_WINDOWS) || defined(__CYGWIN__)
  37. // The maximum number of buffers to support in a single operation.
  38. enum { max_buffers = 64 < max_iov_len ? 64 : max_iov_len };
  39. typedef WSABUF native_buffer_type;
  40. static void init_native_buffer(WSABUF& buf,
  41. const boost::asio::mutable_buffer& buffer)
  42. {
  43. buf.buf = boost::asio::buffer_cast<char*>(buffer);
  44. buf.len = static_cast<ULONG>(boost::asio::buffer_size(buffer));
  45. }
  46. static void init_native_buffer(WSABUF& buf,
  47. const boost::asio::const_buffer& buffer)
  48. {
  49. buf.buf = const_cast<char*>(boost::asio::buffer_cast<const char*>(buffer));
  50. buf.len = static_cast<ULONG>(boost::asio::buffer_size(buffer));
  51. }
  52. #else // defined(BOOST_ASIO_WINDOWS) || defined(__CYGWIN__)
  53. // The maximum number of buffers to support in a single operation.
  54. enum { max_buffers = 64 < max_iov_len ? 64 : max_iov_len };
  55. typedef iovec native_buffer_type;
  56. static void init_iov_base(void*& base, void* addr)
  57. {
  58. base = addr;
  59. }
  60. template <typename T>
  61. static void init_iov_base(T& base, void* addr)
  62. {
  63. base = static_cast<T>(addr);
  64. }
  65. static void init_native_buffer(iovec& iov,
  66. const boost::asio::mutable_buffer& buffer)
  67. {
  68. init_iov_base(iov.iov_base, boost::asio::buffer_cast<void*>(buffer));
  69. iov.iov_len = boost::asio::buffer_size(buffer);
  70. }
  71. static void init_native_buffer(iovec& iov,
  72. const boost::asio::const_buffer& buffer)
  73. {
  74. init_iov_base(iov.iov_base, const_cast<void*>(
  75. boost::asio::buffer_cast<const void*>(buffer)));
  76. iov.iov_len = boost::asio::buffer_size(buffer);
  77. }
  78. #endif // defined(BOOST_ASIO_WINDOWS) || defined(__CYGWIN__)
  79. };
  80. // Helper class to translate buffers into the native buffer representation.
  81. template <typename Buffer, typename Buffers>
  82. class buffer_sequence_adapter
  83. : buffer_sequence_adapter_base
  84. {
  85. public:
  86. explicit buffer_sequence_adapter(const Buffers& buffer_sequence)
  87. : count_(0), total_buffer_size_(0)
  88. {
  89. typename Buffers::const_iterator iter = buffer_sequence.begin();
  90. typename Buffers::const_iterator end = buffer_sequence.end();
  91. for (; iter != end && count_ < max_buffers; ++iter, ++count_)
  92. {
  93. Buffer buffer(*iter);
  94. init_native_buffer(buffers_[count_], buffer);
  95. total_buffer_size_ += boost::asio::buffer_size(buffer);
  96. }
  97. }
  98. native_buffer_type* buffers()
  99. {
  100. return buffers_;
  101. }
  102. std::size_t count() const
  103. {
  104. return count_;
  105. }
  106. bool all_empty() const
  107. {
  108. return total_buffer_size_ == 0;
  109. }
  110. static bool all_empty(const Buffers& buffer_sequence)
  111. {
  112. typename Buffers::const_iterator iter = buffer_sequence.begin();
  113. typename Buffers::const_iterator end = buffer_sequence.end();
  114. std::size_t i = 0;
  115. for (; iter != end && i < max_buffers; ++iter, ++i)
  116. if (boost::asio::buffer_size(Buffer(*iter)) > 0)
  117. return false;
  118. return true;
  119. }
  120. static void validate(const Buffers& buffer_sequence)
  121. {
  122. typename Buffers::const_iterator iter = buffer_sequence.begin();
  123. typename Buffers::const_iterator end = buffer_sequence.end();
  124. for (; iter != end; ++iter)
  125. {
  126. Buffer buffer(*iter);
  127. boost::asio::buffer_cast<const void*>(buffer);
  128. }
  129. }
  130. static Buffer first(const Buffers& buffer_sequence)
  131. {
  132. typename Buffers::const_iterator iter = buffer_sequence.begin();
  133. typename Buffers::const_iterator end = buffer_sequence.end();
  134. for (; iter != end; ++iter)
  135. {
  136. Buffer buffer(*iter);
  137. if (boost::asio::buffer_size(buffer) != 0)
  138. return buffer;
  139. }
  140. return Buffer();
  141. }
  142. private:
  143. native_buffer_type buffers_[max_buffers];
  144. std::size_t count_;
  145. std::size_t total_buffer_size_;
  146. };
  147. template <typename Buffer>
  148. class buffer_sequence_adapter<Buffer, boost::asio::mutable_buffers_1>
  149. : buffer_sequence_adapter_base
  150. {
  151. public:
  152. explicit buffer_sequence_adapter(
  153. const boost::asio::mutable_buffers_1& buffer_sequence)
  154. {
  155. init_native_buffer(buffer_, Buffer(buffer_sequence));
  156. total_buffer_size_ = boost::asio::buffer_size(buffer_sequence);
  157. }
  158. native_buffer_type* buffers()
  159. {
  160. return &buffer_;
  161. }
  162. std::size_t count() const
  163. {
  164. return 1;
  165. }
  166. bool all_empty() const
  167. {
  168. return total_buffer_size_ == 0;
  169. }
  170. static bool all_empty(const boost::asio::mutable_buffers_1& buffer_sequence)
  171. {
  172. return boost::asio::buffer_size(buffer_sequence) == 0;
  173. }
  174. static void validate(const boost::asio::mutable_buffers_1& buffer_sequence)
  175. {
  176. boost::asio::buffer_cast<const void*>(buffer_sequence);
  177. }
  178. static Buffer first(const boost::asio::mutable_buffers_1& buffer_sequence)
  179. {
  180. return Buffer(buffer_sequence);
  181. }
  182. private:
  183. native_buffer_type buffer_;
  184. std::size_t total_buffer_size_;
  185. };
  186. template <typename Buffer>
  187. class buffer_sequence_adapter<Buffer, boost::asio::const_buffers_1>
  188. : buffer_sequence_adapter_base
  189. {
  190. public:
  191. explicit buffer_sequence_adapter(
  192. const boost::asio::const_buffers_1& buffer_sequence)
  193. {
  194. init_native_buffer(buffer_, Buffer(buffer_sequence));
  195. total_buffer_size_ = boost::asio::buffer_size(buffer_sequence);
  196. }
  197. native_buffer_type* buffers()
  198. {
  199. return &buffer_;
  200. }
  201. std::size_t count() const
  202. {
  203. return 1;
  204. }
  205. bool all_empty() const
  206. {
  207. return total_buffer_size_ == 0;
  208. }
  209. static bool all_empty(const boost::asio::const_buffers_1& buffer_sequence)
  210. {
  211. return boost::asio::buffer_size(buffer_sequence) == 0;
  212. }
  213. static void validate(const boost::asio::const_buffers_1& buffer_sequence)
  214. {
  215. boost::asio::buffer_cast<const void*>(buffer_sequence);
  216. }
  217. static Buffer first(const boost::asio::const_buffers_1& buffer_sequence)
  218. {
  219. return Buffer(buffer_sequence);
  220. }
  221. private:
  222. native_buffer_type buffer_;
  223. std::size_t total_buffer_size_;
  224. };
  225. template <typename Buffer, typename Elem>
  226. class buffer_sequence_adapter<Buffer, boost::array<Elem, 2> >
  227. : buffer_sequence_adapter_base
  228. {
  229. public:
  230. explicit buffer_sequence_adapter(
  231. const boost::array<Elem, 2>& buffer_sequence)
  232. {
  233. init_native_buffer(buffers_[0], Buffer(buffer_sequence[0]));
  234. init_native_buffer(buffers_[1], Buffer(buffer_sequence[1]));
  235. total_buffer_size_ = boost::asio::buffer_size(buffer_sequence[0])
  236. + boost::asio::buffer_size(buffer_sequence[1]);
  237. }
  238. native_buffer_type* buffers()
  239. {
  240. return buffers_;
  241. }
  242. std::size_t count() const
  243. {
  244. return 2;
  245. }
  246. bool all_empty() const
  247. {
  248. return total_buffer_size_ == 0;
  249. }
  250. static bool all_empty(const boost::array<Elem, 2>& buffer_sequence)
  251. {
  252. return boost::asio::buffer_size(buffer_sequence[0]) == 0
  253. && boost::asio::buffer_size(buffer_sequence[1]) == 0;
  254. }
  255. static void validate(const boost::array<Elem, 2>& buffer_sequence)
  256. {
  257. boost::asio::buffer_cast<const void*>(buffer_sequence[0]);
  258. boost::asio::buffer_cast<const void*>(buffer_sequence[1]);
  259. }
  260. static Buffer first(const boost::array<Elem, 2>& buffer_sequence)
  261. {
  262. return Buffer(boost::asio::buffer_size(buffer_sequence[0]) != 0
  263. ? buffer_sequence[0] : buffer_sequence[1]);
  264. }
  265. private:
  266. native_buffer_type buffers_[2];
  267. std::size_t total_buffer_size_;
  268. };
  269. #if defined(BOOST_ASIO_HAS_STD_ARRAY)
  270. template <typename Buffer, typename Elem>
  271. class buffer_sequence_adapter<Buffer, std::array<Elem, 2> >
  272. : buffer_sequence_adapter_base
  273. {
  274. public:
  275. explicit buffer_sequence_adapter(
  276. const std::array<Elem, 2>& buffer_sequence)
  277. {
  278. init_native_buffer(buffers_[0], Buffer(buffer_sequence[0]));
  279. init_native_buffer(buffers_[1], Buffer(buffer_sequence[1]));
  280. total_buffer_size_ = boost::asio::buffer_size(buffer_sequence[0])
  281. + boost::asio::buffer_size(buffer_sequence[1]);
  282. }
  283. native_buffer_type* buffers()
  284. {
  285. return buffers_;
  286. }
  287. std::size_t count() const
  288. {
  289. return 2;
  290. }
  291. bool all_empty() const
  292. {
  293. return total_buffer_size_ == 0;
  294. }
  295. static bool all_empty(const std::array<Elem, 2>& buffer_sequence)
  296. {
  297. return boost::asio::buffer_size(buffer_sequence[0]) == 0
  298. && boost::asio::buffer_size(buffer_sequence[1]) == 0;
  299. }
  300. static void validate(const std::array<Elem, 2>& buffer_sequence)
  301. {
  302. boost::asio::buffer_cast<const void*>(buffer_sequence[0]);
  303. boost::asio::buffer_cast<const void*>(buffer_sequence[1]);
  304. }
  305. static Buffer first(const std::array<Elem, 2>& buffer_sequence)
  306. {
  307. return Buffer(boost::asio::buffer_size(buffer_sequence[0]) != 0
  308. ? buffer_sequence[0] : buffer_sequence[1]);
  309. }
  310. private:
  311. native_buffer_type buffers_[2];
  312. std::size_t total_buffer_size_;
  313. };
  314. #endif // defined(BOOST_ASIO_HAS_STD_ARRAY)
  315. } // namespace detail
  316. } // namespace asio
  317. } // namespace boost
  318. #include <boost/asio/detail/pop_options.hpp>
  319. #if defined(BOOST_ASIO_HEADER_ONLY)
  320. # include <boost/asio/detail/impl/buffer_sequence_adapter.ipp>
  321. #endif // defined(BOOST_ASIO_HEADER_ONLY)
  322. #endif // BOOST_ASIO_DETAIL_BUFFER_SEQUENCE_ADAPTER_HPP