You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and dots ('.'), can be up to 35 characters long. Letters must be lowercase.
391 lines
13 KiB
391 lines
13 KiB
// |
|
// impl/connect.hpp |
|
// ~~~~~~~~~~~~~~~~ |
|
// |
|
// Copyright (c) 2003-2011 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
|
|
|