ceph/src/boost/libs/asio/include/boost/asio/detail/win_iocp_overlapped_ptr.hpp

   1 //
   2 // detail/win_iocp_overlapped_ptr.hpp
   3 // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
   4 //
   5 // Copyright (c) 2003-2016 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_WIN_IOCP_OVERLAPPED_PTR_HPP
  12 #define BOOST_ASIO_DETAIL_WIN_IOCP_OVERLAPPED_PTR_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
  20 #if defined(BOOST_ASIO_HAS_IOCP)
  21
  22 #include <boost/asio/io_service.hpp>
  23 #include <boost/asio/detail/addressof.hpp>
  24 #include <boost/asio/detail/handler_alloc_helpers.hpp>
  25 #include <boost/asio/detail/noncopyable.hpp>
  26 #include <boost/asio/detail/win_iocp_overlapped_op.hpp>
  27 #include <boost/asio/detail/win_iocp_io_service.hpp>
  28
  29 #include <boost/asio/detail/push_options.hpp>
  30
  31 namespace boost {
  32 namespace asio {
  33 namespace detail {
  34
  35 // Wraps a handler to create an OVERLAPPED object for use with overlapped I/O.
  36 class win_iocp_overlapped_ptr
  37   : private noncopyable
  38 {
  39 public:
  40   // Construct an empty win_iocp_overlapped_ptr.
  41   win_iocp_overlapped_ptr()
  42     : ptr_(0),
  43       iocp_service_(0)
  44   {
  45   }
  46
  47   // Construct an win_iocp_overlapped_ptr to contain the specified handler.
  48   template <typename Handler>
  49   explicit win_iocp_overlapped_ptr(
  50       boost::asio::io_service& io_service, BOOST_ASIO_MOVE_ARG(Handler) handler)
  51     : ptr_(0),
  52       iocp_service_(0)
  53   {
  54     this->reset(io_service, BOOST_ASIO_MOVE_CAST(Handler)(handler));
  55   }
  56
  57   // Destructor automatically frees the OVERLAPPED object unless released.
  58   ~win_iocp_overlapped_ptr()
  59   {
  60     reset();
  61   }
  62
  63   // Reset to empty.
  64   void reset()
  65   {
  66     if (ptr_)
  67     {
  68       ptr_->destroy();
  69       ptr_ = 0;
  70       iocp_service_->work_finished();
  71       iocp_service_ = 0;
  72     }
  73   }
  74
  75   // Reset to contain the specified handler, freeing any current OVERLAPPED
  76   // object.
  77   template <typename Handler>
  78   void reset(boost::asio::io_service& io_service, Handler handler)
  79   {
  80     typedef win_iocp_overlapped_op<Handler> op;
  81     typename op::ptr p = { boost::asio::detail::addressof(handler),
  82       boost_asio_handler_alloc_helpers::allocate(
  83         sizeof(op), handler), 0 };
  84     p.p = new (p.v) op(handler);
  85
  86     BOOST_ASIO_HANDLER_CREATION((p.p, "io_service",
  87           &io_service.impl_, "overlapped"));
  88
  89     io_service.impl_.work_started();
  90     reset();
  91     ptr_ = p.p;
  92     p.v = p.p = 0;
  93     iocp_service_ = &io_service.impl_;
  94   }
  95
  96   // Get the contained OVERLAPPED object.
  97   OVERLAPPED* get()
  98   {
  99     return ptr_;
 100   }
 101
 102   // Get the contained OVERLAPPED object.
 103   const OVERLAPPED* get() const
 104   {
 105     return ptr_;
 106   }
 107
 108   // Release ownership of the OVERLAPPED object.
 109   OVERLAPPED* release()
 110   {
 111     if (ptr_)
 112       iocp_service_->on_pending(ptr_);
 113
 114     OVERLAPPED* tmp = ptr_;
 115     ptr_ = 0;
 116     iocp_service_ = 0;
 117     return tmp;
 118   }
 119
 120   // Post completion notification for overlapped operation. Releases ownership.
 121   void complete(const boost::system::error_code& ec,
 122       std::size_t bytes_transferred)
 123   {
 124     if (ptr_)
 125     {
 126       iocp_service_->on_completion(ptr_, ec,
 127           static_cast<DWORD>(bytes_transferred));
 128       ptr_ = 0;
 129       iocp_service_ = 0;
 130     }
 131   }
 132
 133 private:
 134   win_iocp_operation* ptr_;
 135   win_iocp_io_service* iocp_service_;
 136 };
 137
 138 } // namespace detail
 139 } // namespace asio
 140 } // namespace boost
 141
 142 #include <boost/asio/detail/pop_options.hpp>
 143
 144 #endif // defined(BOOST_ASIO_HAS_IOCP)
 145
 146 #endif // BOOST_ASIO_DETAIL_WIN_IOCP_OVERLAPPED_PTR_HPP