registered_buffer.hpp 7.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344
  1. //
  2. // registered_buffer.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 ASIO_REGISTERED_BUFFER_HPP
  11. #define ASIO_REGISTERED_BUFFER_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/push_options.hpp"
  18. namespace asio {
  19. namespace detail {
  20. class buffer_registration_base;
  21. } // namespace detail
  22. class const_registered_buffer;
  23. /// Type used to identify a registered buffer.
  24. class registered_buffer_id
  25. {
  26. public:
  27. /// The native buffer identifier type.
  28. typedef int native_handle_type;
  29. /// Default constructor creates an invalid registered buffer identifier.
  30. registered_buffer_id() noexcept
  31. : scope_(0),
  32. index_(-1)
  33. {
  34. }
  35. /// Get the native buffer identifier type.
  36. native_handle_type native_handle() const noexcept
  37. {
  38. return index_;
  39. }
  40. /// Compare two IDs for equality.
  41. friend bool operator==(const registered_buffer_id& lhs,
  42. const registered_buffer_id& rhs) noexcept
  43. {
  44. return lhs.scope_ == rhs.scope_ && lhs.index_ == rhs.index_;
  45. }
  46. /// Compare two IDs for equality.
  47. friend bool operator!=(const registered_buffer_id& lhs,
  48. const registered_buffer_id& rhs) noexcept
  49. {
  50. return lhs.scope_ != rhs.scope_ || lhs.index_ != rhs.index_;
  51. }
  52. private:
  53. friend class detail::buffer_registration_base;
  54. // Hidden constructor used by buffer registration.
  55. registered_buffer_id(const void* scope, int index) noexcept
  56. : scope_(scope),
  57. index_(index)
  58. {
  59. }
  60. const void* scope_;
  61. int index_;
  62. };
  63. /// Holds a registered buffer over modifiable data.
  64. /**
  65. * Satisfies the @c MutableBufferSequence type requirements.
  66. */
  67. class mutable_registered_buffer
  68. {
  69. public:
  70. /// Default constructor creates an invalid registered buffer.
  71. mutable_registered_buffer() noexcept
  72. : buffer_(),
  73. id_()
  74. {
  75. }
  76. /// Get the underlying mutable buffer.
  77. const mutable_buffer& buffer() const noexcept
  78. {
  79. return buffer_;
  80. }
  81. /// Get a pointer to the beginning of the memory range.
  82. /**
  83. * @returns <tt>buffer().data()</tt>.
  84. */
  85. void* data() const noexcept
  86. {
  87. return buffer_.data();
  88. }
  89. /// Get the size of the memory range.
  90. /**
  91. * @returns <tt>buffer().size()</tt>.
  92. */
  93. std::size_t size() const noexcept
  94. {
  95. return buffer_.size();
  96. }
  97. /// Get the registered buffer identifier.
  98. const registered_buffer_id& id() const noexcept
  99. {
  100. return id_;
  101. }
  102. /// Move the start of the buffer by the specified number of bytes.
  103. mutable_registered_buffer& operator+=(std::size_t n) noexcept
  104. {
  105. buffer_ += n;
  106. return *this;
  107. }
  108. private:
  109. friend class detail::buffer_registration_base;
  110. // Hidden constructor used by buffer registration and operators.
  111. mutable_registered_buffer(const mutable_buffer& b,
  112. const registered_buffer_id& i) noexcept
  113. : buffer_(b),
  114. id_(i)
  115. {
  116. }
  117. #if !defined(GENERATING_DOCUMENTATION)
  118. friend mutable_registered_buffer buffer(
  119. const mutable_registered_buffer& b, std::size_t n) noexcept;
  120. #endif // !defined(GENERATING_DOCUMENTATION)
  121. mutable_buffer buffer_;
  122. registered_buffer_id id_;
  123. };
  124. /// Holds a registered buffer over non-modifiable data.
  125. /**
  126. * Satisfies the @c ConstBufferSequence type requirements.
  127. */
  128. class const_registered_buffer
  129. {
  130. public:
  131. /// Default constructor creates an invalid registered buffer.
  132. const_registered_buffer() noexcept
  133. : buffer_(),
  134. id_()
  135. {
  136. }
  137. /// Construct a non-modifiable buffer from a modifiable one.
  138. const_registered_buffer(
  139. const mutable_registered_buffer& b) noexcept
  140. : buffer_(b.buffer()),
  141. id_(b.id())
  142. {
  143. }
  144. /// Get the underlying constant buffer.
  145. const const_buffer& buffer() const noexcept
  146. {
  147. return buffer_;
  148. }
  149. /// Get a pointer to the beginning of the memory range.
  150. /**
  151. * @returns <tt>buffer().data()</tt>.
  152. */
  153. const void* data() const noexcept
  154. {
  155. return buffer_.data();
  156. }
  157. /// Get the size of the memory range.
  158. /**
  159. * @returns <tt>buffer().size()</tt>.
  160. */
  161. std::size_t size() const noexcept
  162. {
  163. return buffer_.size();
  164. }
  165. /// Get the registered buffer identifier.
  166. const registered_buffer_id& id() const noexcept
  167. {
  168. return id_;
  169. }
  170. /// Move the start of the buffer by the specified number of bytes.
  171. const_registered_buffer& operator+=(std::size_t n) noexcept
  172. {
  173. buffer_ += n;
  174. return *this;
  175. }
  176. private:
  177. // Hidden constructor used by operators.
  178. const_registered_buffer(const const_buffer& b,
  179. const registered_buffer_id& i) noexcept
  180. : buffer_(b),
  181. id_(i)
  182. {
  183. }
  184. #if !defined(GENERATING_DOCUMENTATION)
  185. friend const_registered_buffer buffer(
  186. const const_registered_buffer& b, std::size_t n) noexcept;
  187. #endif // !defined(GENERATING_DOCUMENTATION)
  188. const_buffer buffer_;
  189. registered_buffer_id id_;
  190. };
  191. /** @addtogroup buffer_sequence_begin */
  192. /// Get an iterator to the first element in a buffer sequence.
  193. inline const mutable_buffer* buffer_sequence_begin(
  194. const mutable_registered_buffer& b) noexcept
  195. {
  196. return &b.buffer();
  197. }
  198. /// Get an iterator to the first element in a buffer sequence.
  199. inline const const_buffer* buffer_sequence_begin(
  200. const const_registered_buffer& b) noexcept
  201. {
  202. return &b.buffer();
  203. }
  204. /** @} */
  205. /** @addtogroup buffer_sequence_end */
  206. /// Get an iterator to one past the end element in a buffer sequence.
  207. inline const mutable_buffer* buffer_sequence_end(
  208. const mutable_registered_buffer& b) noexcept
  209. {
  210. return &b.buffer() + 1;
  211. }
  212. /// Get an iterator to one past the end element in a buffer sequence.
  213. inline const const_buffer* buffer_sequence_end(
  214. const const_registered_buffer& b) noexcept
  215. {
  216. return &b.buffer() + 1;
  217. }
  218. /** @} */
  219. /** @addtogroup buffer */
  220. /// Obtain a buffer representing the entire registered buffer.
  221. inline mutable_registered_buffer buffer(
  222. const mutable_registered_buffer& b) noexcept
  223. {
  224. return b;
  225. }
  226. /// Obtain a buffer representing the entire registered buffer.
  227. inline const_registered_buffer buffer(
  228. const const_registered_buffer& b) noexcept
  229. {
  230. return b;
  231. }
  232. /// Obtain a buffer representing part of a registered buffer.
  233. inline mutable_registered_buffer buffer(
  234. const mutable_registered_buffer& b, std::size_t n) noexcept
  235. {
  236. return mutable_registered_buffer(buffer(b.buffer_, n), b.id_);
  237. }
  238. /// Obtain a buffer representing part of a registered buffer.
  239. inline const_registered_buffer buffer(
  240. const const_registered_buffer& b, std::size_t n) noexcept
  241. {
  242. return const_registered_buffer(buffer(b.buffer_, n), b.id_);
  243. }
  244. /** @} */
  245. /// Create a new modifiable registered buffer that is offset from the start of
  246. /// another.
  247. /**
  248. * @relates mutable_registered_buffer
  249. */
  250. inline mutable_registered_buffer operator+(
  251. const mutable_registered_buffer& b, std::size_t n) noexcept
  252. {
  253. mutable_registered_buffer tmp(b);
  254. tmp += n;
  255. return tmp;
  256. }
  257. /// Create a new modifiable buffer that is offset from the start of another.
  258. /**
  259. * @relates mutable_registered_buffer
  260. */
  261. inline mutable_registered_buffer operator+(std::size_t n,
  262. const mutable_registered_buffer& b) noexcept
  263. {
  264. return b + n;
  265. }
  266. /// Create a new non-modifiable registered buffer that is offset from the start
  267. /// of another.
  268. /**
  269. * @relates const_registered_buffer
  270. */
  271. inline const_registered_buffer operator+(const const_registered_buffer& b,
  272. std::size_t n) noexcept
  273. {
  274. const_registered_buffer tmp(b);
  275. tmp += n;
  276. return tmp;
  277. }
  278. /// Create a new non-modifiable buffer that is offset from the start of another.
  279. /**
  280. * @relates const_registered_buffer
  281. */
  282. inline const_registered_buffer operator+(std::size_t n,
  283. const const_registered_buffer& b) noexcept
  284. {
  285. return b + n;
  286. }
  287. } // namespace asio
  288. #include "asio/detail/pop_options.hpp"
  289. #endif // ASIO_REGISTERED_BUFFER_HPP