asio.hpp 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408
  1. /*
  2. * Copyright (c) 2017-2023 zhllxt
  3. *
  4. * author : zhllxt
  5. * email : 37792738@qq.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 __ASIO2_ASIO_HPP__
  11. #define __ASIO2_ASIO_HPP__
  12. #if defined(_MSC_VER) && (_MSC_VER >= 1200)
  13. #pragma once
  14. #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
  15. #include <asio2/config.hpp>
  16. #ifdef ASIO2_HEADER_ONLY
  17. # ifndef ASIO_STANDALONE
  18. # define ASIO_STANDALONE
  19. # endif
  20. #endif
  21. #ifdef ASIO_STANDALONE
  22. # ifndef ASIO_HEADER_ONLY
  23. # define ASIO_HEADER_ONLY
  24. # endif
  25. #endif
  26. #include <asio2/base/detail/push_options.hpp>
  27. #ifdef ASIO_STANDALONE
  28. #if __has_include(<asio.hpp>)
  29. #include <asio.hpp>
  30. #elif __has_include(<asio/asio.hpp>)
  31. #include <asio/asio.hpp>
  32. #else
  33. #include "asio/associated_allocator.hpp"
  34. #include "asio/associated_executor.hpp"
  35. #include "asio/async_result.hpp"
  36. #include "asio/awaitable.hpp"
  37. #include "asio/basic_datagram_socket.hpp"
  38. #include "asio/basic_deadline_timer.hpp"
  39. #include "asio/basic_io_object.hpp"
  40. #include "asio/basic_raw_socket.hpp"
  41. #include "asio/basic_seq_packet_socket.hpp"
  42. #include "asio/basic_serial_port.hpp"
  43. #include "asio/basic_signal_set.hpp"
  44. #include "asio/basic_socket.hpp"
  45. #include "asio/basic_socket_acceptor.hpp"
  46. #include "asio/basic_socket_iostream.hpp"
  47. #include "asio/basic_socket_streambuf.hpp"
  48. #include "asio/basic_stream_socket.hpp"
  49. #include "asio/basic_streambuf.hpp"
  50. #include "asio/basic_waitable_timer.hpp"
  51. #include "asio/bind_executor.hpp"
  52. #include "asio/buffer.hpp"
  53. #include "asio/buffered_read_stream_fwd.hpp"
  54. #include "asio/buffered_read_stream.hpp"
  55. #include "asio/buffered_stream_fwd.hpp"
  56. #include "asio/buffered_stream.hpp"
  57. #include "asio/buffered_write_stream_fwd.hpp"
  58. #include "asio/buffered_write_stream.hpp"
  59. #include "asio/buffers_iterator.hpp"
  60. #include "asio/co_spawn.hpp"
  61. #include "asio/completion_condition.hpp"
  62. #include "asio/compose.hpp"
  63. #include "asio/connect.hpp"
  64. #include "asio/coroutine.hpp"
  65. #include "asio/deadline_timer.hpp"
  66. #include "asio/defer.hpp"
  67. #include "asio/detached.hpp"
  68. #include "asio/dispatch.hpp"
  69. #include "asio/error.hpp"
  70. #include "asio/error_code.hpp"
  71. #include "asio/execution.hpp"
  72. #include "asio/execution/allocator.hpp"
  73. #include "asio/execution/any_executor.hpp"
  74. #include "asio/execution/blocking.hpp"
  75. #include "asio/execution/blocking_adaptation.hpp"
  76. #include "asio/execution/bulk_execute.hpp"
  77. #include "asio/execution/bulk_guarantee.hpp"
  78. #include "asio/execution/connect.hpp"
  79. #include "asio/execution/context.hpp"
  80. #include "asio/execution/context_as.hpp"
  81. #include "asio/execution/execute.hpp"
  82. #include "asio/execution/executor.hpp"
  83. #include "asio/execution/invocable_archetype.hpp"
  84. #include "asio/execution/mapping.hpp"
  85. #include "asio/execution/occupancy.hpp"
  86. #include "asio/execution/operation_state.hpp"
  87. #include "asio/execution/outstanding_work.hpp"
  88. #include "asio/execution/prefer_only.hpp"
  89. #include "asio/execution/receiver.hpp"
  90. #include "asio/execution/receiver_invocation_error.hpp"
  91. #include "asio/execution/relationship.hpp"
  92. #include "asio/execution/schedule.hpp"
  93. #include "asio/execution/scheduler.hpp"
  94. #include "asio/execution/sender.hpp"
  95. #include "asio/execution/set_done.hpp"
  96. #include "asio/execution/set_error.hpp"
  97. #include "asio/execution/set_value.hpp"
  98. #include "asio/execution/start.hpp"
  99. #include "asio/execution_context.hpp"
  100. #include "asio/executor.hpp"
  101. #include "asio/executor_work_guard.hpp"
  102. #include "asio/generic/basic_endpoint.hpp"
  103. #include "asio/generic/datagram_protocol.hpp"
  104. #include "asio/generic/raw_protocol.hpp"
  105. #include "asio/generic/seq_packet_protocol.hpp"
  106. #include "asio/generic/stream_protocol.hpp"
  107. #include "asio/handler_alloc_hook.hpp"
  108. #include "asio/handler_continuation_hook.hpp"
  109. #include "asio/handler_invoke_hook.hpp"
  110. #include "asio/high_resolution_timer.hpp"
  111. #include "asio/io_context.hpp"
  112. #include "asio/io_context_strand.hpp"
  113. #include "asio/io_service.hpp"
  114. #include "asio/io_service_strand.hpp"
  115. #include "asio/ip/address.hpp"
  116. #include "asio/ip/address_v4.hpp"
  117. #include "asio/ip/address_v4_iterator.hpp"
  118. #include "asio/ip/address_v4_range.hpp"
  119. #include "asio/ip/address_v6.hpp"
  120. #include "asio/ip/address_v6_iterator.hpp"
  121. #include "asio/ip/address_v6_range.hpp"
  122. #include "asio/ip/network_v4.hpp"
  123. #include "asio/ip/network_v6.hpp"
  124. #include "asio/ip/bad_address_cast.hpp"
  125. #include "asio/ip/basic_endpoint.hpp"
  126. #include "asio/ip/basic_resolver.hpp"
  127. #include "asio/ip/basic_resolver_entry.hpp"
  128. #include "asio/ip/basic_resolver_iterator.hpp"
  129. #include "asio/ip/basic_resolver_query.hpp"
  130. #include "asio/ip/host_name.hpp"
  131. #include "asio/ip/icmp.hpp"
  132. #include "asio/ip/multicast.hpp"
  133. #include "asio/ip/resolver_base.hpp"
  134. #include "asio/ip/resolver_query_base.hpp"
  135. #include "asio/ip/tcp.hpp"
  136. #include "asio/ip/udp.hpp"
  137. #include "asio/ip/unicast.hpp"
  138. #include "asio/ip/v6_only.hpp"
  139. #include "asio/is_applicable_property.hpp"
  140. #include "asio/is_executor.hpp"
  141. #include "asio/is_read_buffered.hpp"
  142. #include "asio/is_write_buffered.hpp"
  143. #include "asio/local/basic_endpoint.hpp"
  144. #include "asio/local/connect_pair.hpp"
  145. #include "asio/local/datagram_protocol.hpp"
  146. #include "asio/local/stream_protocol.hpp"
  147. #include "asio/multiple_exceptions.hpp"
  148. #include "asio/packaged_task.hpp"
  149. #include "asio/placeholders.hpp"
  150. #include "asio/posix/basic_descriptor.hpp"
  151. #include "asio/posix/basic_stream_descriptor.hpp"
  152. #include "asio/posix/descriptor.hpp"
  153. #include "asio/posix/descriptor_base.hpp"
  154. #include "asio/posix/stream_descriptor.hpp"
  155. #include "asio/post.hpp"
  156. #include "asio/prefer.hpp"
  157. #include "asio/query.hpp"
  158. #include "asio/read.hpp"
  159. #include "asio/read_at.hpp"
  160. #include "asio/read_until.hpp"
  161. #include "asio/redirect_error.hpp"
  162. #include "asio/require.hpp"
  163. #include "asio/require_concept.hpp"
  164. #include "asio/serial_port.hpp"
  165. #include "asio/serial_port_base.hpp"
  166. #include "asio/signal_set.hpp"
  167. #include "asio/socket_base.hpp"
  168. #include "asio/static_thread_pool.hpp"
  169. #include "asio/steady_timer.hpp"
  170. #include "asio/strand.hpp"
  171. #include "asio/streambuf.hpp"
  172. #include "asio/system_context.hpp"
  173. #include "asio/system_error.hpp"
  174. #include "asio/system_executor.hpp"
  175. #include "asio/system_timer.hpp"
  176. #include "asio/this_coro.hpp"
  177. #include "asio/thread.hpp"
  178. #include "asio/thread_pool.hpp"
  179. #include "asio/time_traits.hpp"
  180. #include "asio/use_awaitable.hpp"
  181. #include "asio/use_future.hpp"
  182. #include "asio/uses_executor.hpp"
  183. #include "asio/version.hpp"
  184. #include "asio/wait_traits.hpp"
  185. #include "asio/windows/basic_object_handle.hpp"
  186. #include "asio/windows/basic_overlapped_handle.hpp"
  187. #include "asio/windows/basic_random_access_handle.hpp"
  188. #include "asio/windows/basic_stream_handle.hpp"
  189. #include "asio/windows/object_handle.hpp"
  190. #include "asio/windows/overlapped_handle.hpp"
  191. #include "asio/windows/overlapped_ptr.hpp"
  192. #include "asio/windows/random_access_handle.hpp"
  193. #include "asio/windows/stream_handle.hpp"
  194. #include "asio/write.hpp"
  195. #include "asio/write_at.hpp"
  196. #endif
  197. #if defined(ASIO2_ENABLE_SSL) || defined(ASIO2_USE_SSL)
  198. #include <asio/ssl.hpp>
  199. #endif
  200. #else
  201. #include <boost/asio.hpp>
  202. #if defined(ASIO2_ENABLE_SSL) || defined(ASIO2_USE_SSL)
  203. #include <boost/asio/ssl.hpp>
  204. #endif
  205. #if !defined(ASIO_VERSION) && defined(BOOST_ASIO_VERSION)
  206. #define ASIO_VERSION BOOST_ASIO_VERSION
  207. #endif
  208. #if !defined(ASIO_CONST_BUFFER) && defined(BOOST_ASIO_CONST_BUFFER)
  209. #define ASIO_CONST_BUFFER BOOST_ASIO_CONST_BUFFER
  210. #endif
  211. #if !defined(ASIO_MUTABLE_BUFFER) && defined(BOOST_ASIO_MUTABLE_BUFFER)
  212. #define ASIO_MUTABLE_BUFFER BOOST_ASIO_MUTABLE_BUFFER
  213. #endif
  214. #if !defined(ASIO_NOEXCEPT) && defined(BOOST_ASIO_NOEXCEPT)
  215. #define ASIO_NOEXCEPT BOOST_ASIO_NOEXCEPT
  216. #endif
  217. #if !defined(ASIO_NO_EXCEPTIONS) && defined(BOOST_ASIO_NO_EXCEPTIONS)
  218. #define ASIO_NO_EXCEPTIONS BOOST_ASIO_NO_EXCEPTIONS
  219. #endif
  220. #if !defined(ASIO_CORO_REENTER) && defined(BOOST_ASIO_CORO_REENTER)
  221. #define ASIO_CORO_REENTER BOOST_ASIO_CORO_REENTER
  222. #endif
  223. #if !defined(ASIO_CORO_YIELD) && defined(BOOST_ASIO_CORO_YIELD)
  224. #define ASIO_CORO_YIELD BOOST_ASIO_CORO_YIELD
  225. #endif
  226. #if !defined(ASIO_CORO_FORK) && defined(BOOST_ASIO_CORO_FORK)
  227. #define ASIO_CORO_FORK BOOST_ASIO_CORO_FORK
  228. #endif
  229. #if !defined(ASIO_INITFN_AUTO_RESULT_TYPE) && defined(BOOST_ASIO_INITFN_AUTO_RESULT_TYPE)
  230. #define ASIO_INITFN_AUTO_RESULT_TYPE BOOST_ASIO_INITFN_AUTO_RESULT_TYPE
  231. #endif
  232. #if !defined(ASIO_COMPLETION_TOKEN_FOR) && defined(BOOST_ASIO_COMPLETION_TOKEN_FOR)
  233. #define ASIO_COMPLETION_TOKEN_FOR BOOST_ASIO_COMPLETION_TOKEN_FOR
  234. #endif
  235. #if !defined(ASIO_DEFAULT_COMPLETION_TOKEN) && defined(BOOST_ASIO_DEFAULT_COMPLETION_TOKEN)
  236. #define ASIO_DEFAULT_COMPLETION_TOKEN BOOST_ASIO_DEFAULT_COMPLETION_TOKEN
  237. #endif
  238. #if !defined(ASIO_DEFAULT_COMPLETION_TOKEN_TYPE) && defined(BOOST_ASIO_DEFAULT_COMPLETION_TOKEN_TYPE)
  239. #define ASIO_DEFAULT_COMPLETION_TOKEN_TYPE BOOST_ASIO_DEFAULT_COMPLETION_TOKEN_TYPE
  240. #endif
  241. #if !defined(ASIO_INITFN_AUTO_RESULT_TYPE_PREFIX) && defined(BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_PREFIX)
  242. #define ASIO_INITFN_AUTO_RESULT_TYPE_PREFIX BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_PREFIX
  243. #endif
  244. #if !defined(ASIO_INITFN_AUTO_RESULT_TYPE_SUFFIX) && defined(BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_SUFFIX)
  245. #define ASIO_INITFN_AUTO_RESULT_TYPE_SUFFIX BOOST_ASIO_INITFN_AUTO_RESULT_TYPE_SUFFIX
  246. #endif
  247. #if !defined(ASIO_NO_TS_EXECUTORS) && defined(BOOST_ASIO_NO_TS_EXECUTORS)
  248. #define ASIO_NO_TS_EXECUTORS BOOST_ASIO_NO_TS_EXECUTORS
  249. #endif
  250. #if !defined(ASIO_HANDLER_LOCATION) && defined(BOOST_ASIO_HANDLER_LOCATION)
  251. #define ASIO_HANDLER_LOCATION BOOST_ASIO_HANDLER_LOCATION
  252. #endif
  253. #if !defined(ASIO_ENABLE_BUFFER_DEBUGGING) && defined(BOOST_ASIO_ENABLE_BUFFER_DEBUGGING)
  254. #define ASIO_ENABLE_BUFFER_DEBUGGING BOOST_ASIO_ENABLE_BUFFER_DEBUGGING
  255. #endif
  256. #if !defined(ASIO_MOVE_ARG) && defined(BOOST_ASIO_MOVE_ARG)
  257. #define ASIO_MOVE_ARG BOOST_ASIO_MOVE_ARG
  258. #endif
  259. #if !defined(ASIO_MOVE_CAST) && defined(BOOST_ASIO_MOVE_CAST)
  260. #define ASIO_MOVE_CAST BOOST_ASIO_MOVE_CAST
  261. #endif
  262. #if !defined(ASIO_HANDLER_TYPE) && defined(BOOST_ASIO_HANDLER_TYPE)
  263. #define ASIO_HANDLER_TYPE BOOST_ASIO_HANDLER_TYPE
  264. #endif
  265. #if !defined(ASIO_HAS_STD_HASH) && defined(BOOST_ASIO_HAS_STD_HASH)
  266. #define ASIO_HAS_STD_HASH BOOST_ASIO_HAS_STD_HASH
  267. #endif
  268. #if !defined(ASIO_SYNC_OP_VOID) && defined(BOOST_ASIO_SYNC_OP_VOID)
  269. #define ASIO_SYNC_OP_VOID BOOST_ASIO_SYNC_OP_VOID
  270. #endif
  271. #if !defined(ASIO_SYNC_OP_VOID_RETURN) && defined(BOOST_ASIO_SYNC_OP_VOID_RETURN)
  272. #define ASIO_SYNC_OP_VOID_RETURN BOOST_ASIO_SYNC_OP_VOID_RETURN
  273. #endif
  274. #endif // ASIO_STANDALONE
  275. #ifdef ASIO_STANDALONE
  276. namespace asio
  277. {
  278. using error_condition = std::error_condition;
  279. }
  280. #else
  281. namespace boost::asio
  282. {
  283. using error_code = ::boost::system::error_code;
  284. using system_error = ::boost::system::system_error;
  285. using error_condition = ::boost::system::error_condition;
  286. using error_category = ::boost::system::error_category;
  287. }
  288. namespace asio = ::boost::asio;
  289. namespace bho = ::boost; // bho means boost header only
  290. // [ adding definitions to namespace alias ]
  291. // This is currently not allowed and probably won't be in C++1Z either,
  292. // but note that a recent proposal is allowing
  293. // https://stackoverflow.com/questions/31629101/adding-definitions-to-namespace-alias?r=SearchResults
  294. //namespace asio
  295. //{
  296. // using error_code = ::boost::system::error_code;
  297. // using system_error = ::boost::system::system_error;
  298. //}
  299. #endif // ASIO_STANDALONE
  300. namespace asio2
  301. {
  302. using error_code = ::asio::error_code;
  303. using system_error = ::asio::system_error;
  304. using error_condition = ::asio::error_condition;
  305. using error_category = ::asio::error_category;
  306. }
  307. #ifdef ASIO_STANDALONE
  308. namespace asio
  309. #else
  310. namespace boost::asio
  311. #endif
  312. {
  313. /*
  314. * used for rdc mode, call("abc") or async_call("abc")
  315. * Without the following overload, the compilation will fail.
  316. */
  317. template<class CharT, class Traits = std::char_traits<CharT>>
  318. inline typename std::enable_if_t<
  319. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, char > ||
  320. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, wchar_t > ||
  321. #if defined(__cpp_lib_char8_t)
  322. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, char8_t > ||
  323. #endif
  324. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, char16_t> ||
  325. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, char32_t>,
  326. ASIO_CONST_BUFFER> buffer(CharT* & data) ASIO_NOEXCEPT
  327. {
  328. return asio::buffer(std::basic_string_view<CharT>(data));
  329. }
  330. template<class CharT, class Traits = std::char_traits<CharT>>
  331. inline typename std::enable_if_t<
  332. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, char > ||
  333. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, wchar_t > ||
  334. #if defined(__cpp_lib_char8_t)
  335. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, char8_t > ||
  336. #endif
  337. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, char16_t> ||
  338. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, char32_t>,
  339. ASIO_CONST_BUFFER> buffer(const CharT* & data) ASIO_NOEXCEPT
  340. {
  341. return asio::buffer(std::basic_string_view<CharT>(data));
  342. }
  343. template<class CharT, class Traits = std::char_traits<CharT>>
  344. inline typename std::enable_if_t<
  345. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, char > ||
  346. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, wchar_t > ||
  347. #if defined(__cpp_lib_char8_t)
  348. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, char8_t > ||
  349. #endif
  350. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, char16_t> ||
  351. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, char32_t>,
  352. ASIO_CONST_BUFFER> buffer(CharT* const& data) ASIO_NOEXCEPT
  353. {
  354. return asio::buffer(std::basic_string_view<CharT>(data));
  355. }
  356. template<class CharT, class Traits = std::char_traits<CharT>>
  357. inline typename std::enable_if_t<
  358. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, char > ||
  359. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, wchar_t > ||
  360. #if defined(__cpp_lib_char8_t)
  361. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, char8_t > ||
  362. #endif
  363. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, char16_t> ||
  364. std::is_same_v<std::remove_cv_t<std::remove_reference_t<CharT>>, char32_t>,
  365. ASIO_CONST_BUFFER> buffer(const CharT* const& data) ASIO_NOEXCEPT
  366. {
  367. return asio::buffer(std::basic_string_view<CharT>(data));
  368. }
  369. }
  370. #if (defined(ASIO_NO_EXCEPTIONS) || defined(BOOST_ASIO_NO_EXCEPTIONS)) && !defined(ASIO2_NO_EXCEPTIONS_IMPL)
  371. #include <asio2/external/assert.hpp>
  372. #include <iostream>
  373. #ifdef ASIO_STANDALONE
  374. namespace asio::detail
  375. #else
  376. namespace boost::asio::detail
  377. #endif
  378. {
  379. template <typename Exception>
  380. void throw_exception(const Exception& e ASIO_SOURCE_LOCATION_PARAM)
  381. {
  382. std::cerr << "exception occured: " << e.what() << std::endl;
  383. ASIO2_ASSERT(false);
  384. std::terminate();
  385. }
  386. }
  387. #endif
  388. #include <asio2/base/detail/pop_options.hpp>
  389. #endif // !__ASIO2_ASIO_HPP__