// // impl/connect.hpp // ~~~~~~~~~~~~~~~~ // // Copyright (c) 2003-2012 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 BOOST_ASIO_IMPL_CONNECT_HPP #define BOOST_ASIO_IMPL_CONNECT_HPP #if defined(_MSC_VER) && (_MSC_VER >= 1200) # pragma once #endif // defined(_MSC_VER) && (_MSC_VER >= 1200) #include <boost/asio/detail/bind_handler.hpp> #include <boost/asio/detail/consuming_buffers.hpp> #include <boost/asio/detail/handler_alloc_helpers.hpp> #include <boost/asio/detail/handler_invoke_helpers.hpp> #include <boost/asio/detail/handler_type_requirements.hpp> #include <boost/asio/detail/throw_error.hpp> #include <boost/asio/error.hpp> #include <boost/asio/detail/push_options.hpp> namespace boost { namespace asio { namespace detail { struct default_connect_condition { template <typename Iterator> Iterator operator()(const boost::system::error_code&, Iterator next) { return next; } }; } template <typename Protocol, typename SocketService, typename Iterator> Iterator connect(basic_socket<Protocol, SocketService>& s, Iterator begin) { boost::system::error_code ec; Iterator result = connect(s, begin, ec); boost::asio::detail::throw_error(ec, "connect"); return result; } template <typename Protocol, typename SocketService, typename Iterator> inline Iterator connect(basic_socket<Protocol, SocketService>& s, Iterator begin, boost::system::error_code& ec) { return connect(s, begin, Iterator(), detail::default_connect_condition(), ec); } template <typename Protocol, typename SocketService, typename Iterator> Iterator connect(basic_socket<Protocol, SocketService>& s, Iterator begin, Iterator end) { boost::system::error_code ec; Iterator result = connect(s, begin, end, ec); boost::asio::detail::throw_error(ec, "connect"); return result; } template <typename Protocol, typename SocketService, typename Iterator> inline Iterator connect(basic_socket<Protocol, SocketService>& s, Iterator begin, Iterator end, boost::system::error_code& ec) { return connect(s, begin, end, detail::default_connect_condition(), ec); } template <typename Protocol, typename SocketService, typename Iterator, typename ConnectCondition> Iterator connect(basic_socket<Protocol, SocketService>& s, Iterator begin, ConnectCondition connect_condition) { boost::system::error_code ec; Iterator result = connect(s, begin, connect_condition, ec); boost::asio::detail::throw_error(ec, "connect"); return result; } template <typename Protocol, typename SocketService, typename Iterator, typename ConnectCondition> inline Iterator connect(basic_socket<Protocol, SocketService>& s, Iterator begin, ConnectCondition connect_condition, boost::system::error_code& ec) { return connect(s, begin, Iterator(), connect_condition, ec); } template <typename Protocol, typename SocketService, typename Iterator, typename ConnectCondition> Iterator connect(basic_socket<Protocol, SocketService>& s, Iterator begin, Iterator end, ConnectCondition connect_condition) { boost::system::error_code ec; Iterator result = connect(s, begin, end, connect_condition, ec); boost::asio::detail::throw_error(ec, "connect"); return result; } template <typename Protocol, typename SocketService, typename Iterator, typename ConnectCondition> Iterator connect(basic_socket<Protocol, SocketService>& s, Iterator begin, Iterator end, ConnectCondition connect_condition, boost::system::error_code& ec) { ec = boost::system::error_code(); for (Iterator iter = begin; iter != end; ++iter) { iter = connect_condition(ec, iter); if (iter != end) { s.close(ec); s.connect(*iter, ec); if (!ec) return iter; } } if (!ec) ec = boost::asio::error::not_found; return end; } namespace detail { // Enable the empty base class optimisation for the connect condition. template <typename ConnectCondition> class base_from_connect_condition { protected: explicit base_from_connect_condition( const ConnectCondition& connect_condition) : connect_condition_(connect_condition) { } template <typename Iterator> void check_condition(const boost::system::error_code& ec, Iterator& iter, Iterator& end) { if (iter != end) iter = connect_condition_(ec, static_cast<const Iterator&>(iter)); } private: ConnectCondition connect_condition_; }; // The default_connect_condition implementation is essentially a no-op. This // template specialisation lets us eliminate all costs associated with it. template <> class base_from_connect_condition<default_connect_condition> { protected: explicit base_from_connect_condition(const default_connect_condition&) { } template <typename Iterator> void check_condition(const boost::system::error_code&, Iterator&, Iterator&) { } }; template <typename Protocol, typename SocketService, typename Iterator, typename ConnectCondition, typename ComposedConnectHandler> class connect_op : base_from_connect_condition<ConnectCondition> { public: connect_op(basic_socket<Protocol, SocketService>& sock, const Iterator& begin, const Iterator& end, const ConnectCondition& connect_condition, ComposedConnectHandler& handler) : base_from_connect_condition<ConnectCondition>(connect_condition), socket_(sock), iter_(begin), end_(end), handler_(BOOST_ASIO_MOVE_CAST(ComposedConnectHandler)(handler)) { } #if defined(BOOST_ASIO_HAS_MOVE) connect_op(const connect_op& other) : base_from_connect_condition<ConnectCondition>(other), socket_(other.socket_), iter_(other.iter_), end_(other.end_), handler_(other.handler_) { } connect_op(connect_op&& other) : base_from_connect_condition<ConnectCondition>(other), socket_(other.socket_), iter_(other.iter_), end_(other.end_), handler_(BOOST_ASIO_MOVE_CAST(ComposedConnectHandler)(other.handler_)) { } #endif // defined(BOOST_ASIO_HAS_MOVE) void operator()(boost::system::error_code ec, int start = 0) { switch (start) { case 1: for (;;) { this->check_condition(ec, iter_, end_); if (iter_ != end_) { socket_.close(ec); socket_.async_connect(*iter_, BOOST_ASIO_MOVE_CAST(connect_op)(*this)); return; } if (start) { ec = boost::asio::error::not_found; socket_.get_io_service().post(detail::bind_handler(*this, ec)); return; } default: if (iter_ == end_) break; if (!socket_.is_open()) { ec = boost::asio::error::operation_aborted; break; } if (!ec) break; ++iter_; } handler_(static_cast<const boost::system::error_code&>(ec), static_cast<const Iterator&>(iter_)); } } //private: basic_socket<Protocol, SocketService>& socket_; Iterator iter_; Iterator end_; ComposedConnectHandler handler_; }; template <typename Protocol, typename SocketService, typename Iterator, typename ConnectCondition, typename ComposedConnectHandler> inline void* asio_handler_allocate(std::size_t size, connect_op<Protocol, SocketService, Iterator, ConnectCondition, ComposedConnectHandler>* this_handler) { return boost_asio_handler_alloc_helpers::allocate( size, this_handler->handler_); } template <typename Protocol, typename SocketService, typename Iterator, typename ConnectCondition, typename ComposedConnectHandler> inline void asio_handler_deallocate(void* pointer, std::size_t size, connect_op<Protocol, SocketService, Iterator, ConnectCondition, ComposedConnectHandler>* this_handler) { boost_asio_handler_alloc_helpers::deallocate( pointer, size, this_handler->handler_); } template <typename Function, typename Protocol, typename SocketService, typename Iterator, typename ConnectCondition, typename ComposedConnectHandler> inline void asio_handler_invoke(Function& function, connect_op<Protocol, SocketService, Iterator, ConnectCondition, ComposedConnectHandler>* this_handler) { boost_asio_handler_invoke_helpers::invoke( function, this_handler->handler_); } template <typename Function, typename Protocol, typename SocketService, typename Iterator, typename ConnectCondition, typename ComposedConnectHandler> inline void asio_handler_invoke(const Function& function, connect_op<Protocol, SocketService, Iterator, ConnectCondition, ComposedConnectHandler>* this_handler) { boost_asio_handler_invoke_helpers::invoke( function, this_handler->handler_); } template <typename Protocol, typename SocketService, typename Iterator, typename ConnectCondition, typename ComposedConnectHandler> inline connect_op<Protocol, SocketService, Iterator, ConnectCondition, ComposedConnectHandler> make_connect_op(basic_socket<Protocol, SocketService>& sock, const Iterator& begin, const Iterator& end, const ConnectCondition& connect_condition, ComposedConnectHandler handler) { return connect_op<Protocol, SocketService, Iterator, ConnectCondition, ComposedConnectHandler>( sock, begin, end, connect_condition, handler); } } // namespace detail template <typename Protocol, typename SocketService, typename Iterator, typename ComposedConnectHandler> inline void async_connect(basic_socket<Protocol, SocketService>& s, Iterator begin, BOOST_ASIO_MOVE_ARG(ComposedConnectHandler) handler) { // If you get an error on the following line it means that your handler does // not meet the documented type requirements for a ComposedConnectHandler. BOOST_ASIO_COMPOSED_CONNECT_HANDLER_CHECK( ComposedConnectHandler, handler, Iterator) type_check; detail::make_connect_op(s, begin, Iterator(), detail::default_connect_condition(), BOOST_ASIO_MOVE_CAST(ComposedConnectHandler)(handler))( boost::system::error_code(), 1); } template <typename Protocol, typename SocketService, typename Iterator, typename ComposedConnectHandler> inline void async_connect(basic_socket<Protocol, SocketService>& s, Iterator begin, Iterator end, BOOST_ASIO_MOVE_ARG(ComposedConnectHandler) handler) { // If you get an error on the following line it means that your handler does // not meet the documented type requirements for a ComposedConnectHandler. BOOST_ASIO_COMPOSED_CONNECT_HANDLER_CHECK( ComposedConnectHandler, handler, Iterator) type_check; detail::make_connect_op(s, begin, end, detail::default_connect_condition(), BOOST_ASIO_MOVE_CAST(ComposedConnectHandler)(handler))( boost::system::error_code(), 1); } template <typename Protocol, typename SocketService, typename Iterator, typename ConnectCondition, typename ComposedConnectHandler> inline void async_connect(basic_socket<Protocol, SocketService>& s, Iterator begin, ConnectCondition connect_condition, BOOST_ASIO_MOVE_ARG(ComposedConnectHandler) handler) { // If you get an error on the following line it means that your handler does // not meet the documented type requirements for a ComposedConnectHandler. BOOST_ASIO_COMPOSED_CONNECT_HANDLER_CHECK( ComposedConnectHandler, handler, Iterator) type_check; detail::make_connect_op(s, begin, Iterator(), connect_condition, BOOST_ASIO_MOVE_CAST(ComposedConnectHandler)(handler))( boost::system::error_code(), 1); } template <typename Protocol, typename SocketService, typename Iterator, typename ConnectCondition, typename ComposedConnectHandler> void async_connect(basic_socket<Protocol, SocketService>& s, Iterator begin, Iterator end, ConnectCondition connect_condition, BOOST_ASIO_MOVE_ARG(ComposedConnectHandler) handler) { // If you get an error on the following line it means that your handler does // not meet the documented type requirements for a ComposedConnectHandler. BOOST_ASIO_COMPOSED_CONNECT_HANDLER_CHECK( ComposedConnectHandler, handler, Iterator) type_check; detail::make_connect_op(s, begin, end, connect_condition, BOOST_ASIO_MOVE_CAST(ComposedConnectHandler)(handler))( boost::system::error_code(), 1); } } // namespace asio } // namespace boost #include <boost/asio/detail/pop_options.hpp> #endif // BOOST_ASIO_IMPL_CONNECT_HPP