1 // 2 // detail/win_iocp_handle_write_op.hpp 3 // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ 4 // 5 // Copyright (c) 2003-2021 Christopher M. Kohlhoff (chris at kohlhoff dot com) 6 // Copyright (c) 2008 Rep Invariant Systems, Inc. ([email protected]) 7 // 8 // Distributed under the Boost Software License, Version 1.0. (See accompanying 9 // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) 10 // 11 12 #ifndef BOOST_ASIO_DETAIL_WIN_IOCP_HANDLE_WRITE_OP_HPP 13 #define BOOST_ASIO_DETAIL_WIN_IOCP_HANDLE_WRITE_OP_HPP 14 15 #if defined(_MSC_VER) && (_MSC_VER >= 1200) 16 # pragma once 17 #endif // defined(_MSC_VER) && (_MSC_VER >= 1200) 18 19 #include <boost/asio/detail/config.hpp> 20 21 #if defined(BOOST_ASIO_HAS_IOCP) 22 23 #include <boost/asio/detail/bind_handler.hpp> 24 #include <boost/asio/detail/buffer_sequence_adapter.hpp> 25 #include <boost/asio/detail/fenced_block.hpp> 26 #include <boost/asio/detail/handler_alloc_helpers.hpp> 27 #include <boost/asio/detail/handler_invoke_helpers.hpp> 28 #include <boost/asio/detail/handler_work.hpp> 29 #include <boost/asio/detail/memory.hpp> 30 #include <boost/asio/detail/operation.hpp> 31 #include <boost/asio/error.hpp> 32 33 #include <boost/asio/detail/push_options.hpp> 34 35 namespace boost { 36 namespace asio { 37 namespace detail { 38 39 template <typename ConstBufferSequence, typename Handler, typename IoExecutor> 40 class win_iocp_handle_write_op : public operation 41 { 42 public: 43 BOOST_ASIO_DEFINE_HANDLER_PTR(win_iocp_handle_write_op); 44 win_iocp_handle_write_op(const ConstBufferSequence & buffers,Handler & handler,const IoExecutor & io_ex)45 win_iocp_handle_write_op(const ConstBufferSequence& buffers, 46 Handler& handler, const IoExecutor& io_ex) 47 : operation(&win_iocp_handle_write_op::do_complete), 48 buffers_(buffers), 49 handler_(BOOST_ASIO_MOVE_CAST(Handler)(handler)), 50 work_(handler_, io_ex) 51 { 52 } 53 do_complete(void * owner,operation * base,const boost::system::error_code & ec,std::size_t bytes_transferred)54 static void do_complete(void* owner, operation* base, 55 const boost::system::error_code& ec, std::size_t bytes_transferred) 56 { 57 // Take ownership of the operation object. 58 win_iocp_handle_write_op* o(static_cast<win_iocp_handle_write_op*>(base)); 59 ptr p = { boost::asio::detail::addressof(o->handler_), o, o }; 60 61 BOOST_ASIO_HANDLER_COMPLETION((*o)); 62 63 // Take ownership of the operation's outstanding work. 64 handler_work<Handler, IoExecutor> w( 65 BOOST_ASIO_MOVE_CAST2(handler_work<Handler, IoExecutor>)( 66 o->work_)); 67 68 #if defined(BOOST_ASIO_ENABLE_BUFFER_DEBUGGING) 69 if (owner) 70 { 71 // Check whether buffers are still valid. 72 buffer_sequence_adapter<boost::asio::const_buffer, 73 ConstBufferSequence>::validate(o->buffers_); 74 } 75 #endif // defined(BOOST_ASIO_ENABLE_BUFFER_DEBUGGING) 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::binder2<Handler, boost::system::error_code, std::size_t> 84 handler(o->handler_, ec, bytes_transferred); 85 p.h = boost::asio::detail::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_, handler.arg2_)); 93 w.complete(handler, handler.handler_); 94 BOOST_ASIO_HANDLER_INVOCATION_END; 95 } 96 } 97 98 private: 99 ConstBufferSequence buffers_; 100 Handler handler_; 101 handler_work<Handler, IoExecutor> work_; 102 }; 103 104 } // namespace detail 105 } // namespace asio 106 } // namespace boost 107 108 #include <boost/asio/detail/pop_options.hpp> 109 110 #endif // defined(BOOST_ASIO_HAS_IOCP) 111 112 #endif // BOOST_ASIO_DETAIL_WIN_IOCP_HANDLE_WRITE_OP_HPP 113