148 lines
		
	
	
		
			4.5 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			148 lines
		
	
	
		
			4.5 KiB
		
	
	
	
		
			C++
		
	
	
	
| //
 | |
| // detail/resolver_service.hpp
 | |
| // ~~~~~~~~~~~~~~~~~~~~~~~~~~~
 | |
| //
 | |
| // Copyright (c) 2003-2023 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)
 | |
| //
 | |
| 
 | |
| #ifndef ASIO_DETAIL_RESOLVER_SERVICE_HPP
 | |
| #define ASIO_DETAIL_RESOLVER_SERVICE_HPP
 | |
| 
 | |
| #if defined(_MSC_VER) && (_MSC_VER >= 1200)
 | |
| # pragma once
 | |
| #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
 | |
| 
 | |
| #include "asio/detail/config.hpp"
 | |
| 
 | |
| #if !defined(ASIO_WINDOWS_RUNTIME)
 | |
| 
 | |
| #include "asio/ip/basic_resolver_query.hpp"
 | |
| #include "asio/ip/basic_resolver_results.hpp"
 | |
| #include "asio/detail/concurrency_hint.hpp"
 | |
| #include "asio/detail/memory.hpp"
 | |
| #include "asio/detail/resolve_endpoint_op.hpp"
 | |
| #include "asio/detail/resolve_query_op.hpp"
 | |
| #include "asio/detail/resolver_service_base.hpp"
 | |
| 
 | |
| #include "asio/detail/push_options.hpp"
 | |
| 
 | |
| namespace asio {
 | |
| namespace detail {
 | |
| 
 | |
| template <typename Protocol>
 | |
| class resolver_service :
 | |
|   public execution_context_service_base<resolver_service<Protocol> >,
 | |
|   public resolver_service_base
 | |
| {
 | |
| public:
 | |
|   // The implementation type of the resolver. A cancellation token is used to
 | |
|   // indicate to the background thread that the operation has been cancelled.
 | |
|   typedef socket_ops::shared_cancel_token_type implementation_type;
 | |
| 
 | |
|   // The endpoint type.
 | |
|   typedef typename Protocol::endpoint endpoint_type;
 | |
| 
 | |
|   // The query type.
 | |
|   typedef asio::ip::basic_resolver_query<Protocol> query_type;
 | |
| 
 | |
|   // The results type.
 | |
|   typedef asio::ip::basic_resolver_results<Protocol> results_type;
 | |
| 
 | |
|   // Constructor.
 | |
|   resolver_service(execution_context& context)
 | |
|     : execution_context_service_base<resolver_service<Protocol> >(context),
 | |
|       resolver_service_base(context)
 | |
|   {
 | |
|   }
 | |
| 
 | |
|   // Destroy all user-defined handler objects owned by the service.
 | |
|   void shutdown()
 | |
|   {
 | |
|     this->base_shutdown();
 | |
|   }
 | |
| 
 | |
|   // Perform any fork-related housekeeping.
 | |
|   void notify_fork(execution_context::fork_event fork_ev)
 | |
|   {
 | |
|     this->base_notify_fork(fork_ev);
 | |
|   }
 | |
| 
 | |
|   // Resolve a query to a list of entries.
 | |
|   results_type resolve(implementation_type&, const query_type& qry,
 | |
|       asio::error_code& ec)
 | |
|   {
 | |
|     asio::detail::addrinfo_type* address_info = 0;
 | |
| 
 | |
|     socket_ops::getaddrinfo(qry.host_name().c_str(),
 | |
|         qry.service_name().c_str(), qry.hints(), &address_info, ec);
 | |
|     auto_addrinfo auto_address_info(address_info);
 | |
| 
 | |
|     ASIO_ERROR_LOCATION(ec);
 | |
|     return ec ? results_type() : results_type::create(
 | |
|         address_info, qry.host_name(), qry.service_name());
 | |
|   }
 | |
| 
 | |
|   // Asynchronously resolve a query to a list of entries.
 | |
|   template <typename Handler, typename IoExecutor>
 | |
|   void async_resolve(implementation_type& impl, const query_type& qry,
 | |
|       Handler& handler, const IoExecutor& io_ex)
 | |
|   {
 | |
|     // Allocate and construct an operation to wrap the handler.
 | |
|     typedef resolve_query_op<Protocol, Handler, IoExecutor> op;
 | |
|     typename op::ptr p = { asio::detail::addressof(handler),
 | |
|       op::ptr::allocate(handler), 0 };
 | |
|     p.p = new (p.v) op(impl, qry, scheduler_, handler, io_ex);
 | |
| 
 | |
|     ASIO_HANDLER_CREATION((scheduler_.context(),
 | |
|           *p.p, "resolver", &impl, 0, "async_resolve"));
 | |
| 
 | |
|     start_resolve_op(p.p);
 | |
|     p.v = p.p = 0;
 | |
|   }
 | |
| 
 | |
|   // Resolve an endpoint to a list of entries.
 | |
|   results_type resolve(implementation_type&,
 | |
|       const endpoint_type& endpoint, asio::error_code& ec)
 | |
|   {
 | |
|     char host_name[NI_MAXHOST];
 | |
|     char service_name[NI_MAXSERV];
 | |
|     socket_ops::sync_getnameinfo(endpoint.data(), endpoint.size(),
 | |
|         host_name, NI_MAXHOST, service_name, NI_MAXSERV,
 | |
|         endpoint.protocol().type(), ec);
 | |
| 
 | |
|     ASIO_ERROR_LOCATION(ec);
 | |
|     return ec ? results_type() : results_type::create(
 | |
|         endpoint, host_name, service_name);
 | |
|   }
 | |
| 
 | |
|   // Asynchronously resolve an endpoint to a list of entries.
 | |
|   template <typename Handler, typename IoExecutor>
 | |
|   void async_resolve(implementation_type& impl, const endpoint_type& endpoint,
 | |
|       Handler& handler, const IoExecutor& io_ex)
 | |
|   {
 | |
|     // Allocate and construct an operation to wrap the handler.
 | |
|     typedef resolve_endpoint_op<Protocol, Handler, IoExecutor> op;
 | |
|     typename op::ptr p = { asio::detail::addressof(handler),
 | |
|       op::ptr::allocate(handler), 0 };
 | |
|     p.p = new (p.v) op(impl, endpoint, scheduler_, handler, io_ex);
 | |
| 
 | |
|     ASIO_HANDLER_CREATION((scheduler_.context(),
 | |
|           *p.p, "resolver", &impl, 0, "async_resolve"));
 | |
| 
 | |
|     start_resolve_op(p.p);
 | |
|     p.v = p.p = 0;
 | |
|   }
 | |
| };
 | |
| 
 | |
| } // namespace detail
 | |
| } // namespace asio
 | |
| 
 | |
| #include "asio/detail/pop_options.hpp"
 | |
| 
 | |
| #endif // !defined(ASIO_WINDOWS_RUNTIME)
 | |
| 
 | |
| #endif // ASIO_DETAIL_RESOLVER_SERVICE_HPP
 |