asio/detail/winrt_resolve_op.hpp

0001 //
0002 // detail/winrt_resolve_op.hpp
0003 // ~~~~~~~~~~~~~~~~~~~~~~~~~~~
0004 //
0005 // Copyright (c) 2003-2023 Christopher M. Kohlhoff (chris at kohlhoff dot com)
0006 //
0007 // Distributed under the Boost Software License, Version 1.0. (See accompanying
0008 // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
0009 //
0010
0011 #ifndef BOOST_ASIO_DETAIL_WINRT_RESOLVE_OP_HPP
0012 #define BOOST_ASIO_DETAIL_WINRT_RESOLVE_OP_HPP
0013
0014 #if defined(_MSC_VER) && (_MSC_VER >= 1200)
0015 # pragma once
0016 #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
0017
0018 #include <boost/asio/detail/config.hpp>
0019
0020 #if defined(BOOST_ASIO_WINDOWS_RUNTIME)
0021
0022 #include <boost/asio/detail/bind_handler.hpp>
0023 #include <boost/asio/detail/fenced_block.hpp>
0024 #include <boost/asio/detail/handler_alloc_helpers.hpp>
0025 #include <boost/asio/detail/handler_work.hpp>
0026 #include <boost/asio/detail/memory.hpp>
0027 #include <boost/asio/detail/winrt_async_op.hpp>
0028 #include <boost/asio/ip/basic_resolver_results.hpp>
0029 #include <boost/asio/error.hpp>
0030
0031 #include <boost/asio/detail/push_options.hpp>
0032
0033 namespace boost {
0034 namespace asio {
0035 namespace detail {
0036
0037 template <typename Protocol, typename Handler, typename IoExecutor>
0038 class winrt_resolve_op :
0039   public winrt_async_op<
0040     Windows::Foundation::Collections::IVectorView<
0041       Windows::Networking::EndpointPair^>^>
0042 {
0043 public:
0044   BOOST_ASIO_DEFINE_HANDLER_PTR(winrt_resolve_op);
0045
0046   typedef typename Protocol::endpoint endpoint_type;
0047   typedef boost::asio::ip::basic_resolver_query<Protocol> query_type;
0048   typedef boost::asio::ip::basic_resolver_results<Protocol> results_type;
0049
0050   winrt_resolve_op(const query_type& query,
0051       Handler& handler, const IoExecutor& io_ex)
0052     : winrt_async_op<
0053         Windows::Foundation::Collections::IVectorView<
0054           Windows::Networking::EndpointPair^>^>(
0055             &winrt_resolve_op::do_complete),
0056       query_(query),
0057       handler_(static_cast<Handler&&>(handler)),
0058       work_(handler_, io_ex)
0059   {
0060   }
0061
0062   static void do_complete(void* owner, operation* base,
0063       const boost::system::error_code&, std::size_t)
0064   {
0065     // Take ownership of the operation object.
0066     BOOST_ASIO_ASSUME(base != 0);
0067     winrt_resolve_op* o(static_cast<winrt_resolve_op*>(base));
0068     ptr p = { boost::asio::detail::addressof(o->handler_), o, o };
0069
0070     BOOST_ASIO_HANDLER_COMPLETION((*o));
0071
0072     // Take ownership of the operation's outstanding work.
0073     handler_work<Handler, IoExecutor> w(
0074         static_cast<handler_work<Handler, IoExecutor>&&>(
0075           o->work_));
0076
0077     results_type results = results_type();
0078     if (!o->ec_)
0079     {
0080       try
0081       {
0082         results = results_type::create(o->result_, o->query_.hints(),
0083             o->query_.host_name(), o->query_.service_name());
0084       }
0085       catch (Platform::Exception^ e)
0086       {
0087         o->ec_ = boost::system::error_code(e->HResult,
0088             boost::system::system_category());
0089       }
0090     }
0091
0092     // Make a copy of the handler so that the memory can be deallocated before
0093     // the upcall is made. Even if we're not about to make an upcall, a
0094     // sub-object of the handler may be the true owner of the memory associated
0095     // with the handler. Consequently, a local copy of the handler is required
0096     // to ensure that any owning sub-object remains valid until after we have
0097     // deallocated the memory here.
0098     detail::binder2<Handler, boost::system::error_code, results_type>
0099       handler(o->handler_, o->ec_, results);
0100     p.h = boost::asio::detail::addressof(handler.handler_);
0101     p.reset();
0102
0103     // Make the upcall if required.
0104     if (owner)
0105     {
0106       fenced_block b(fenced_block::half);
0107       BOOST_ASIO_HANDLER_INVOCATION_BEGIN((handler.arg1_, "..."));
0108       w.complete(handler, handler.handler_);
0109       BOOST_ASIO_HANDLER_INVOCATION_END;
0110     }
0111   }
0112
0113 private:
0114   query_type query_;
0115   Handler handler_;
0116   handler_work<Handler, IoExecutor> executor_;
0117 };
0118
0119 } // namespace detail
0120 } // namespace asio
0121 } // namespace boost
0122
0123 #include <boost/asio/detail/pop_options.hpp>
0124
0125 #endif // defined(BOOST_ASIO_WINDOWS_RUNTIME)
0126
0127 #endif // BOOST_ASIO_DETAIL_WINRT_RESOLVE_OP_HPP