1//
2// detail/reactive_socket_service_base.ipp
3// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
4//
5// Copyright (c) 2003-2021 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_IMPL_REACTIVE_SOCKET_SERVICE_BASE_IPP
12#define BOOST_ASIO_DETAIL_IMPL_REACTIVE_SOCKET_SERVICE_BASE_IPP
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  && !defined(BOOST_ASIO_WINDOWS_RUNTIME)
22
23#include <boost/asio/detail/reactive_socket_service_base.hpp>
24
25#include <boost/asio/detail/push_options.hpp>
26
27namespace boost {
28namespace asio {
29namespace detail {
30
31reactive_socket_service_base::reactive_socket_service_base(
32    execution_context& context)
33  : reactor_(use_service<reactor>(context))
34{
35  reactor_.init_task();
36}
37
38void reactive_socket_service_base::base_shutdown()
39{
40}
41
42void reactive_socket_service_base::construct(
43    reactive_socket_service_base::base_implementation_type& impl)
44{
45  impl.socket_ = invalid_socket;
46  impl.state_ = 0;
47}
48
49void reactive_socket_service_base::base_move_construct(
50    reactive_socket_service_base::base_implementation_type& impl,
51    reactive_socket_service_base::base_implementation_type& other_impl)
52  BOOST_ASIO_NOEXCEPT
53{
54  impl.socket_ = other_impl.socket_;
55  other_impl.socket_ = invalid_socket;
56
57  impl.state_ = other_impl.state_;
58  other_impl.state_ = 0;
59
60  reactor_.move_descriptor(impl.socket_,
61      impl.reactor_data_, other_impl.reactor_data_);
62}
63
64void reactive_socket_service_base::base_move_assign(
65    reactive_socket_service_base::base_implementation_type& impl,
66    reactive_socket_service_base& other_service,
67    reactive_socket_service_base::base_implementation_type& other_impl)
68{
69  destroy(impl);
70
71  impl.socket_ = other_impl.socket_;
72  other_impl.socket_ = invalid_socket;
73
74  impl.state_ = other_impl.state_;
75  other_impl.state_ = 0;
76
77  other_service.reactor_.move_descriptor(impl.socket_,
78      impl.reactor_data_, other_impl.reactor_data_);
79}
80
81void reactive_socket_service_base::destroy(
82    reactive_socket_service_base::base_implementation_type& impl)
83{
84  if (impl.socket_ != invalid_socket)
85  {
86    BOOST_ASIO_HANDLER_OPERATION((reactor_.context(),
87          "socket", &impl, impl.socket_, "close"));
88
89    reactor_.deregister_descriptor(impl.socket_, impl.reactor_data_,
90        (impl.state_ & socket_ops::possible_dup) == 0);
91
92    boost::system::error_code ignored_ec;
93    socket_ops::close(impl.socket_, impl.state_, true, ignored_ec);
94
95    reactor_.cleanup_descriptor_data(impl.reactor_data_);
96  }
97}
98
99boost::system::error_code reactive_socket_service_base::close(
100    reactive_socket_service_base::base_implementation_type& impl,
101    boost::system::error_code& ec)
102{
103  if (is_open(impl))
104  {
105    BOOST_ASIO_HANDLER_OPERATION((reactor_.context(),
106          "socket", &impl, impl.socket_, "close"));
107
108    reactor_.deregister_descriptor(impl.socket_, impl.reactor_data_,
109        (impl.state_ & socket_ops::possible_dup) == 0);
110
111    socket_ops::close(impl.socket_, impl.state_, false, ec);
112
113    reactor_.cleanup_descriptor_data(impl.reactor_data_);
114  }
115  else
116  {
117    ec = boost::system::error_code();
118  }
119
120  // The descriptor is closed by the OS even if close() returns an error.
121  //
122  // (Actually, POSIX says the state of the descriptor is unspecified. On
123  // Linux the descriptor is apparently closed anyway; e.g. see
124  //   http://lkml.org/lkml/2005/9/10/129
125  // We'll just have to assume that other OSes follow the same behaviour. The
126  // known exception is when Windows's closesocket() function fails with
127  // WSAEWOULDBLOCK, but this case is handled inside socket_ops::close().
128  construct(impl);
129
130  return ec;
131}
132
133socket_type reactive_socket_service_base::release(
134    reactive_socket_service_base::base_implementation_type& impl,
135    boost::system::error_code& ec)
136{
137  if (!is_open(impl))
138  {
139    ec = boost::asio::error::bad_descriptor;
140    return invalid_socket;
141  }
142
143  BOOST_ASIO_HANDLER_OPERATION((reactor_.context(),
144        "socket", &impl, impl.socket_, "release"));
145
146  reactor_.deregister_descriptor(impl.socket_, impl.reactor_data_, false);
147  reactor_.cleanup_descriptor_data(impl.reactor_data_);
148  socket_type sock = impl.socket_;
149  construct(impl);
150  ec = boost::system::error_code();
151  return sock;
152}
153
154boost::system::error_code reactive_socket_service_base::cancel(
155    reactive_socket_service_base::base_implementation_type& impl,
156    boost::system::error_code& ec)
157{
158  if (!is_open(impl))
159  {
160    ec = boost::asio::error::bad_descriptor;
161    return ec;
162  }
163
164  BOOST_ASIO_HANDLER_OPERATION((reactor_.context(),
165        "socket", &impl, impl.socket_, "cancel"));
166
167  reactor_.cancel_ops(impl.socket_, impl.reactor_data_);
168  ec = boost::system::error_code();
169  return ec;
170}
171
172boost::system::error_code reactive_socket_service_base::do_open(
173    reactive_socket_service_base::base_implementation_type& impl,
174    int af, int type, int protocol, boost::system::error_code& ec)
175{
176  if (is_open(impl))
177  {
178    ec = boost::asio::error::already_open;
179    return ec;
180  }
181
182  socket_holder sock(socket_ops::socket(af, type, protocol, ec));
183  if (sock.get() == invalid_socket)
184    return ec;
185
186  if (int err = reactor_.register_descriptor(sock.get(), impl.reactor_data_))
187  {
188    ec = boost::system::error_code(err,
189        boost::asio::error::get_system_category());
190    return ec;
191  }
192
193  impl.socket_ = sock.release();
194  switch (type)
195  {
196  case SOCK_STREAM: impl.state_ = socket_ops::stream_oriented; break;
197  case SOCK_DGRAM: impl.state_ = socket_ops::datagram_oriented; break;
198  default: impl.state_ = 0; break;
199  }
200  ec = boost::system::error_code();
201  return ec;
202}
203
204boost::system::error_code reactive_socket_service_base::do_assign(
205    reactive_socket_service_base::base_implementation_type& impl, int type,
206    const reactive_socket_service_base::native_handle_type& native_socket,
207    boost::system::error_code& ec)
208{
209  if (is_open(impl))
210  {
211    ec = boost::asio::error::already_open;
212    return ec;
213  }
214
215  if (int err = reactor_.register_descriptor(
216        native_socket, impl.reactor_data_))
217  {
218    ec = boost::system::error_code(err,
219        boost::asio::error::get_system_category());
220    return ec;
221  }
222
223  impl.socket_ = native_socket;
224  switch (type)
225  {
226  case SOCK_STREAM: impl.state_ = socket_ops::stream_oriented; break;
227  case SOCK_DGRAM: impl.state_ = socket_ops::datagram_oriented; break;
228  default: impl.state_ = 0; break;
229  }
230  impl.state_ |= socket_ops::possible_dup;
231  ec = boost::system::error_code();
232  return ec;
233}
234
235void reactive_socket_service_base::start_op(
236    reactive_socket_service_base::base_implementation_type& impl,
237    int op_type, reactor_op* op, bool is_continuation,
238    bool is_non_blocking, bool noop)
239{
240  if (!noop)
241  {
242    if ((impl.state_ & socket_ops::non_blocking)
243        || socket_ops::set_internal_non_blocking(
244          impl.socket_, impl.state_, true, op->ec_))
245    {
246      reactor_.start_op(op_type, impl.socket_,
247          impl.reactor_data_, op, is_continuation, is_non_blocking);
248      return;
249    }
250  }
251
252  reactor_.post_immediate_completion(op, is_continuation);
253}
254
255void reactive_socket_service_base::start_accept_op(
256    reactive_socket_service_base::base_implementation_type& impl,
257    reactor_op* op, bool is_continuation, bool peer_is_open)
258{
259  if (!peer_is_open)
260    start_op(impl, reactor::read_op, op, is_continuation, true, false);
261  else
262  {
263    op->ec_ = boost::asio::error::already_open;
264    reactor_.post_immediate_completion(op, is_continuation);
265  }
266}
267
268void reactive_socket_service_base::start_connect_op(
269    reactive_socket_service_base::base_implementation_type& impl,
270    reactor_op* op, bool is_continuation,
271    const socket_addr_type* addr, size_t addrlen)
272{
273  if ((impl.state_ & socket_ops::non_blocking)
274      || socket_ops::set_internal_non_blocking(
275        impl.socket_, impl.state_, true, op->ec_))
276  {
277    if (socket_ops::connect(impl.socket_, addr, addrlen, op->ec_) != 0)
278    {
279      if (op->ec_ == boost::asio::error::in_progress
280          || op->ec_ == boost::asio::error::would_block)
281      {
282        op->ec_ = boost::system::error_code();
283        reactor_.start_op(reactor::connect_op, impl.socket_,
284            impl.reactor_data_, op, is_continuation, false);
285        return;
286      }
287    }
288  }
289
290  reactor_.post_immediate_completion(op, is_continuation);
291}
292
293} // namespace detail
294} // namespace asio
295} // namespace boost
296
297#include <boost/asio/detail/pop_options.hpp>
298
299#endif // !defined(BOOST_ASIO_HAS_IOCP)
300       //   && !defined(BOOST_ASIO_WINDOWS_RUNTIME)
301
302#endif // BOOST_ASIO_DETAIL_IMPL_REACTIVE_SOCKET_SERVICE_BASE_IPP
303