1 // 2 // detail/reactive_socket_recvmsg_op.hpp 3 // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ 4 // 5 // Copyright (c) 2003-2015 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 ASIO_DETAIL_REACTIVE_SOCKET_RECVMSG_OP_HPP 12 #define ASIO_DETAIL_REACTIVE_SOCKET_RECVMSG_OP_HPP 13 14 15 #include "asio/detail/config.hpp" 16 #include "asio/detail/addressof.hpp" 17 #include "asio/detail/bind_handler.hpp" 18 #include "asio/detail/buffer_sequence_adapter.hpp" 19 #include "asio/detail/fenced_block.hpp" 20 #include "asio/detail/reactor_op.hpp" 21 #include "asio/detail/socket_ops.hpp" 22 #include "asio/socket_base.hpp" 23 24 #include "asio/detail/push_options.hpp" 25 26 namespace asio { 27 namespace detail { 28 29 template <typename MutableBufferSequence> 30 class reactive_socket_recvmsg_op_base : public reactor_op 31 { 32 public: reactive_socket_recvmsg_op_base(socket_type socket,const MutableBufferSequence & buffers,socket_base::message_flags in_flags,socket_base::message_flags & out_flags,func_type complete_func)33 reactive_socket_recvmsg_op_base(socket_type socket, 34 const MutableBufferSequence& buffers, socket_base::message_flags in_flags, 35 socket_base::message_flags& out_flags, func_type complete_func) 36 : reactor_op(&reactive_socket_recvmsg_op_base::do_perform, complete_func), 37 socket_(socket), 38 buffers_(buffers), 39 in_flags_(in_flags), 40 out_flags_(out_flags) 41 { 42 } 43 do_perform(reactor_op * base)44 static bool do_perform(reactor_op* base) 45 { 46 reactive_socket_recvmsg_op_base* o( 47 static_cast<reactive_socket_recvmsg_op_base*>(base)); 48 49 buffer_sequence_adapter<asio::mutable_buffer, 50 MutableBufferSequence> bufs(o->buffers_); 51 52 return socket_ops::non_blocking_recvmsg(o->socket_, 53 bufs.buffers(), bufs.count(), 54 o->in_flags_, o->out_flags_, 55 o->ec_, o->bytes_transferred_); 56 } 57 58 private: 59 socket_type socket_; 60 MutableBufferSequence buffers_; 61 socket_base::message_flags in_flags_; 62 socket_base::message_flags& out_flags_; 63 }; 64 65 template <typename MutableBufferSequence, typename Handler> 66 class reactive_socket_recvmsg_op : 67 public reactive_socket_recvmsg_op_base<MutableBufferSequence> 68 { 69 public: 70 ASIO_DEFINE_HANDLER_PTR(reactive_socket_recvmsg_op); 71 reactive_socket_recvmsg_op(socket_type socket,const MutableBufferSequence & buffers,socket_base::message_flags in_flags,socket_base::message_flags & out_flags,Handler & handler)72 reactive_socket_recvmsg_op(socket_type socket, 73 const MutableBufferSequence& buffers, socket_base::message_flags in_flags, 74 socket_base::message_flags& out_flags, Handler& handler) 75 : reactive_socket_recvmsg_op_base<MutableBufferSequence>(socket, buffers, 76 in_flags, out_flags, &reactive_socket_recvmsg_op::do_complete), 77 handler_(ASIO_MOVE_CAST(Handler)(handler)) 78 { 79 } 80 do_complete(io_service_impl * owner,operation * base,const asio::error_code &,std::size_t)81 static void do_complete(io_service_impl* owner, operation* base, 82 const asio::error_code& /*ec*/, 83 std::size_t /*bytes_transferred*/) 84 { 85 // Take ownership of the handler object. 86 reactive_socket_recvmsg_op* o( 87 static_cast<reactive_socket_recvmsg_op*>(base)); 88 ptr p = { asio::detail::addressof(o->handler_), o, o }; 89 90 ASIO_HANDLER_COMPLETION((o)); 91 92 // Make a copy of the handler so that the memory can be deallocated before 93 // the upcall is made. Even if we're not about to make an upcall, a 94 // sub-object of the handler may be the true owner of the memory associated 95 // with the handler. Consequently, a local copy of the handler is required 96 // to ensure that any owning sub-object remains valid until after we have 97 // deallocated the memory here. 98 detail::binder2<Handler, asio::error_code, std::size_t> 99 handler(o->handler_, o->ec_, o->bytes_transferred_); 100 p.h = asio::detail::addressof(handler.handler_); 101 p.reset(); 102 103 // Make the upcall if required. 104 if (owner) 105 { 106 fenced_block b(fenced_block::half); 107 ASIO_HANDLER_INVOCATION_BEGIN((handler.arg1_, handler.arg2_)); 108 asio_handler_invoke_helpers::invoke(handler, handler.handler_); 109 ASIO_HANDLER_INVOCATION_END; 110 } 111 } 112 113 private: 114 Handler handler_; 115 }; 116 117 } // namespace detail 118 } // namespace asio 119 120 #include "asio/detail/pop_options.hpp" 121 122 #endif // ASIO_DETAIL_REACTIVE_SOCKET_RECVMSG_OP_HPP 123