buffer_sequence_adapter.hpp 9.5 KB

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