diff options
Diffstat (limited to '3rdparty/asio/include/asio/detail/resolve_query_op.hpp')
-rw-r--r-- | 3rdparty/asio/include/asio/detail/resolve_query_op.hpp | 46 |
1 files changed, 32 insertions, 14 deletions
diff --git a/3rdparty/asio/include/asio/detail/resolve_query_op.hpp b/3rdparty/asio/include/asio/detail/resolve_query_op.hpp index 0ce450b5ea5..5a2ffa71a7b 100644 --- a/3rdparty/asio/include/asio/detail/resolve_query_op.hpp +++ b/3rdparty/asio/include/asio/detail/resolve_query_op.hpp @@ -2,7 +2,7 @@ // detail/resolve_query_op.hpp // ~~~~~~~~~~~~~~~~~~~~~~~~~~~ // -// Copyright (c) 2003-2016 Christopher M. Kohlhoff (chris at kohlhoff dot com) +// Copyright (c) 2003-2021 Christopher M. Kohlhoff (chris at kohlhoff dot com) // // Distributed under the Boost Software License, Version 1.0. (See accompanying // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) @@ -16,24 +16,30 @@ #endif // defined(_MSC_VER) && (_MSC_VER >= 1200) #include "asio/detail/config.hpp" -#include "asio/error.hpp" -#include "asio/io_context.hpp" -#include "asio/ip/basic_resolver_query.hpp" -#include "asio/ip/basic_resolver_results.hpp" #include "asio/detail/bind_handler.hpp" #include "asio/detail/fenced_block.hpp" #include "asio/detail/handler_alloc_helpers.hpp" #include "asio/detail/handler_invoke_helpers.hpp" +#include "asio/detail/handler_work.hpp" #include "asio/detail/memory.hpp" #include "asio/detail/resolve_op.hpp" #include "asio/detail/socket_ops.hpp" +#include "asio/error.hpp" +#include "asio/ip/basic_resolver_query.hpp" +#include "asio/ip/basic_resolver_results.hpp" + +#if defined(ASIO_HAS_IOCP) +# include "asio/detail/win_iocp_io_context.hpp" +#else // defined(ASIO_HAS_IOCP) +# include "asio/detail/scheduler.hpp" +#endif // defined(ASIO_HAS_IOCP) #include "asio/detail/push_options.hpp" namespace asio { namespace detail { -template <typename Protocol, typename Handler> +template <typename Protocol, typename Handler, typename IoExecutor> class resolve_query_op : public resolve_op { public: @@ -42,16 +48,23 @@ public: typedef asio::ip::basic_resolver_query<Protocol> query_type; typedef asio::ip::basic_resolver_results<Protocol> results_type; +#if defined(ASIO_HAS_IOCP) + typedef class win_iocp_io_context scheduler_impl; +#else + typedef class scheduler scheduler_impl; +#endif + resolve_query_op(socket_ops::weak_cancel_token_type cancel_token, - const query_type& query, io_context_impl& ioc, Handler& handler) + const query_type& qry, scheduler_impl& sched, + Handler& handler, const IoExecutor& io_ex) : resolve_op(&resolve_query_op::do_complete), cancel_token_(cancel_token), - query_(query), - io_context_impl_(ioc), + query_(qry), + scheduler_(sched), handler_(ASIO_MOVE_CAST(Handler)(handler)), + work_(handler_, io_ex), addrinfo_(0) { - handler_work<Handler>::start(handler_); } ~resolve_query_op() @@ -67,9 +80,8 @@ public: // Take ownership of the operation object. resolve_query_op* o(static_cast<resolve_query_op*>(base)); ptr p = { asio::detail::addressof(o->handler_), o, o }; - handler_work<Handler> w(o->handler_); - if (owner && owner != &o->io_context_impl_) + if (owner && owner != &o->scheduler_) { // The operation is being run on the worker io_context. Time to perform // the resolver operation. @@ -80,7 +92,7 @@ public: o->query_.hints(), &o->addrinfo_, o->ec_); // Pass operation back to main io_context for completion. - o->io_context_impl_.post_deferred_completion(o); + o->scheduler_.post_deferred_completion(o); p.v = p.p = 0; } else @@ -90,6 +102,11 @@ public: ASIO_HANDLER_COMPLETION((*o)); + // Take ownership of the operation's outstanding work. + handler_work<Handler, IoExecutor> w( + ASIO_MOVE_CAST2(handler_work<Handler, IoExecutor>)( + o->work_)); + // Make a copy of the handler so that the memory can be deallocated // before the upcall is made. Even if we're not about to make an upcall, // a sub-object of the handler may be the true owner of the memory @@ -119,8 +136,9 @@ public: private: socket_ops::weak_cancel_token_type cancel_token_; query_type query_; - io_context_impl& io_context_impl_; + scheduler_impl& scheduler_; Handler handler_; + handler_work<Handler, IoExecutor> work_; asio::detail::addrinfo_type* addrinfo_; }; |