boost/asio/detail/reactive_socket_connect_op.hpp

   1 //
   2 // detail/reactive_socket_connect_op.hpp
   3 // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
   4 //
   5 // Copyright (c) 2003-2012 Christopher M. Kohlhoff (chris at kohlhoff dot com)
   6 //
   7 // Distributed under the Boost Software License, Version 1.0. (See accompanying
   8 // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
   9 //
  10
  11 #ifndef BOOST_ASIO_DETAIL_REACTIVE_SOCKET_CONNECT_OP_HPP
  12 #define BOOST_ASIO_DETAIL_REACTIVE_SOCKET_CONNECT_OP_HPP
  13
  14 #if defined(_MSC_VER) && (_MSC_VER >= 1200)
  15 # pragma once
  16 #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
  17
  18 #include <boost/asio/detail/config.hpp>
  19 #include <boost/utility/addressof.hpp>
  20 #include <boost/asio/detail/bind_handler.hpp>
  21 #include <boost/asio/detail/buffer_sequence_adapter.hpp>
  22 #include <boost/asio/detail/fenced_block.hpp>
  23 #include <boost/asio/detail/reactor_op.hpp>
  24 #include <boost/asio/detail/socket_ops.hpp>
  25
  26 #include <boost/asio/detail/push_options.hpp>
  27
  28 namespace boost {
  29 namespace asio {
  30 namespace detail {
  31
  32 class reactive_socket_connect_op_base : public reactor_op
  33 {
  34 public:
  35   reactive_socket_connect_op_base(socket_type socket, func_type complete_func)
  36     : reactor_op(&reactive_socket_connect_op_base::do_perform, complete_func),
  37       socket_(socket)
  38   {
  39   }
  40
  41   static bool do_perform(reactor_op* base)
  42   {
  43     reactive_socket_connect_op_base* o(
  44         static_cast<reactive_socket_connect_op_base*>(base));
  45
  46     return socket_ops::non_blocking_connect(o->socket_, o->ec_);
  47   }
  48
  49 private:
  50   socket_type socket_;
  51 };
  52
  53 template <typename Handler>
  54 class reactive_socket_connect_op : public reactive_socket_connect_op_base
  55 {
  56 public:
  57   BOOST_ASIO_DEFINE_HANDLER_PTR(reactive_socket_connect_op);
  58
  59   reactive_socket_connect_op(socket_type socket, Handler& handler)
  60     : reactive_socket_connect_op_base(socket,
  61         &reactive_socket_connect_op::do_complete),
  62       handler_(BOOST_ASIO_MOVE_CAST(Handler)(handler))
  63   {
  64   }
  65
  66   static void do_complete(io_service_impl* owner, operation* base,
  67       const boost::system::error_code& /*ec*/,
  68       std::size_t /*bytes_transferred*/)
  69   {
  70     // Take ownership of the handler object.
  71     reactive_socket_connect_op* o
  72       (static_cast<reactive_socket_connect_op*>(base));
  73     ptr p = { boost::addressof(o->handler_), o, o };
  74
  75     BOOST_ASIO_HANDLER_COMPLETION((o));
  76
  77     // Make a copy of the handler so that the memory can be deallocated before
  78     // the upcall is made. Even if we're not about to make an upcall, a
  79     // sub-object of the handler may be the true owner of the memory associated
  80     // with the handler. Consequently, a local copy of the handler is required
  81     // to ensure that any owning sub-object remains valid until after we have
  82     // deallocated the memory here.
  83     detail::binder1<Handler, boost::system::error_code>
  84       handler(o->handler_, o->ec_);
  85     p.h = boost::addressof(handler.handler_);
  86     p.reset();
  87
  88     // Make the upcall if required.
  89     if (owner)
  90     {
  91       fenced_block b(fenced_block::half);
  92       BOOST_ASIO_HANDLER_INVOCATION_BEGIN((handler.arg1_));
  93       boost_asio_handler_invoke_helpers::invoke(handler, handler);
  94       BOOST_ASIO_HANDLER_INVOCATION_END;
  95     }
  96   }
  97
  98 private:
  99   Handler handler_;
 100 };
 101
 102 } // namespace detail
 103 } // namespace asio
 104 } // namespace boost
 105
 106 #include <boost/asio/detail/pop_options.hpp>
 107
 108 #endif // BOOST_ASIO_DETAIL_REACTIVE_SOCKET_CONNECT_OP_HPP