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