1  
//
1  
//
2  
// Copyright (c) 2025 Vinnie Falco (vinnie.falco@gmail.com)
2  
// Copyright (c) 2025 Vinnie Falco (vinnie.falco@gmail.com)
3  
// Copyright (c) 2026 Steve Gerbino
3  
// Copyright (c) 2026 Steve Gerbino
4  
//
4  
//
5  
// Distributed under the Boost Software License, Version 1.0. (See accompanying
5  
// Distributed under the Boost Software License, Version 1.0. (See accompanying
6  
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
6  
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
7  
//
7  
//
8  
// Official repository: https://github.com/cppalliance/corosio
8  
// Official repository: https://github.com/cppalliance/corosio
9  
//
9  
//
10  

10  

11  
#ifndef BOOST_COROSIO_TCP_SOCKET_HPP
11  
#ifndef BOOST_COROSIO_TCP_SOCKET_HPP
12  
#define BOOST_COROSIO_TCP_SOCKET_HPP
12  
#define BOOST_COROSIO_TCP_SOCKET_HPP
13  

13  

14  
#include <boost/corosio/detail/config.hpp>
14  
#include <boost/corosio/detail/config.hpp>
15  
#include <boost/corosio/detail/platform.hpp>
15  
#include <boost/corosio/detail/platform.hpp>
16  
#include <boost/corosio/detail/except.hpp>
16  
#include <boost/corosio/detail/except.hpp>
17  
#include <boost/corosio/detail/native_handle.hpp>
17  
#include <boost/corosio/detail/native_handle.hpp>
18  
#include <boost/corosio/detail/op_base.hpp>
18  
#include <boost/corosio/detail/op_base.hpp>
19  
#include <boost/corosio/io/io_stream.hpp>
19  
#include <boost/corosio/io/io_stream.hpp>
20  
#include <boost/capy/io_result.hpp>
20  
#include <boost/capy/io_result.hpp>
21  
#include <boost/corosio/detail/buffer_param.hpp>
21  
#include <boost/corosio/detail/buffer_param.hpp>
22  
#include <boost/corosio/endpoint.hpp>
22  
#include <boost/corosio/endpoint.hpp>
23  
#include <boost/corosio/shutdown_type.hpp>
23  
#include <boost/corosio/shutdown_type.hpp>
24  
#include <boost/corosio/tcp.hpp>
24  
#include <boost/corosio/tcp.hpp>
25  
#include <boost/capy/ex/executor_ref.hpp>
25  
#include <boost/capy/ex/executor_ref.hpp>
26  
#include <boost/capy/ex/execution_context.hpp>
26  
#include <boost/capy/ex/execution_context.hpp>
27  
#include <boost/capy/ex/io_env.hpp>
27  
#include <boost/capy/ex/io_env.hpp>
28  
#include <boost/capy/concept/executor.hpp>
28  
#include <boost/capy/concept/executor.hpp>
29  

29  

30  
#include <system_error>
30  
#include <system_error>
31  

31  

32  
#include <concepts>
32  
#include <concepts>
33  
#include <coroutine>
33  
#include <coroutine>
34  
#include <cstddef>
34  
#include <cstddef>
35  
#include <stop_token>
35  
#include <stop_token>
36  
#include <type_traits>
36  
#include <type_traits>
37  

37  

38  
namespace boost::corosio {
38  
namespace boost::corosio {
39  

39  

40  
/** An asynchronous TCP socket for coroutine I/O.
40  
/** An asynchronous TCP socket for coroutine I/O.
41  

41  

42  
    This class provides asynchronous TCP socket operations that return
42  
    This class provides asynchronous TCP socket operations that return
43  
    awaitable types. Each operation participates in the affine awaitable
43  
    awaitable types. Each operation participates in the affine awaitable
44  
    protocol, ensuring coroutines resume on the correct executor.
44  
    protocol, ensuring coroutines resume on the correct executor.
45  

45  

46  
    The socket must be opened before performing I/O operations. Operations
46  
    The socket must be opened before performing I/O operations. Operations
47  
    support cancellation through `std::stop_token` via the affine protocol,
47  
    support cancellation through `std::stop_token` via the affine protocol,
48  
    or explicitly through the `cancel()` member function.
48  
    or explicitly through the `cancel()` member function.
49  

49  

50  
    @par Thread Safety
50  
    @par Thread Safety
51  
    Distinct objects: Safe.@n
51  
    Distinct objects: Safe.@n
52  
    Shared objects: Unsafe. A socket must not have concurrent operations
52  
    Shared objects: Unsafe. A socket must not have concurrent operations
53  
    of the same type (e.g., two simultaneous reads). One read and one
53  
    of the same type (e.g., two simultaneous reads). One read and one
54  
    write may be in flight simultaneously.
54  
    write may be in flight simultaneously.
55  

55  

56  
    @par Semantics
56  
    @par Semantics
57  
    Wraps the platform TCP/IP stack. Operations dispatch to
57  
    Wraps the platform TCP/IP stack. Operations dispatch to
58  
    OS socket APIs via the io_context reactor (epoll, IOCP,
58  
    OS socket APIs via the io_context reactor (epoll, IOCP,
59  
    kqueue). Satisfies @ref capy::Stream.
59  
    kqueue). Satisfies @ref capy::Stream.
60  

60  

61  
    @par Example
61  
    @par Example
62  
    @code
62  
    @code
63  
    io_context ioc;
63  
    io_context ioc;
64  
    tcp_socket s(ioc);
64  
    tcp_socket s(ioc);
65  
    s.open();
65  
    s.open();
66  

66  

67  
    // Using structured bindings
67  
    // Using structured bindings
68  
    auto [ec] = co_await s.connect(
68  
    auto [ec] = co_await s.connect(
69  
        endpoint(ipv4_address::loopback(), 8080));
69  
        endpoint(ipv4_address::loopback(), 8080));
70  
    if (ec)
70  
    if (ec)
71  
        co_return;
71  
        co_return;
72  

72  

73  
    char buf[1024];
73  
    char buf[1024];
74  
    auto [read_ec, n] = co_await s.read_some(
74  
    auto [read_ec, n] = co_await s.read_some(
75  
        capy::mutable_buffer(buf, sizeof(buf)));
75  
        capy::mutable_buffer(buf, sizeof(buf)));
76  
    @endcode
76  
    @endcode
77  
*/
77  
*/
78  
class BOOST_COROSIO_DECL tcp_socket : public io_stream
78  
class BOOST_COROSIO_DECL tcp_socket : public io_stream
79  
{
79  
{
80 -
    /// The endpoint type used by this socket.
 
81 -
    using endpoint_type = corosio::endpoint;
 
82 -

 
83  
public:
80  
public:
84  
    using shutdown_type = corosio::shutdown_type;
81  
    using shutdown_type = corosio::shutdown_type;
85  
    using enum corosio::shutdown_type;
82  
    using enum corosio::shutdown_type;
86  

83  

87  
    /** Define backend hooks for TCP socket operations.
84  
    /** Define backend hooks for TCP socket operations.
88  

85  

89  
        Platform backends (epoll, IOCP, kqueue, select) derive from
86  
        Platform backends (epoll, IOCP, kqueue, select) derive from
90  
        this to implement socket I/O, connection, and option management.
87  
        this to implement socket I/O, connection, and option management.
91  
    */
88  
    */
92  
    struct implementation : io_stream::implementation
89  
    struct implementation : io_stream::implementation
93  
    {
90  
    {
94  
        /** Initiate an asynchronous connect to the given endpoint.
91  
        /** Initiate an asynchronous connect to the given endpoint.
95  

92  

96  
            @param h Coroutine handle to resume on completion.
93  
            @param h Coroutine handle to resume on completion.
97  
            @param ex Executor for dispatching the completion.
94  
            @param ex Executor for dispatching the completion.
98  
            @param ep The remote endpoint to connect to.
95  
            @param ep The remote endpoint to connect to.
99  
            @param token Stop token for cancellation.
96  
            @param token Stop token for cancellation.
100  
            @param ec Output error code.
97  
            @param ec Output error code.
101  

98  

102  
            @return Coroutine handle to resume immediately.
99  
            @return Coroutine handle to resume immediately.
103  
        */
100  
        */
104  
        virtual std::coroutine_handle<> connect(
101  
        virtual std::coroutine_handle<> connect(
105  
            std::coroutine_handle<> h,
102  
            std::coroutine_handle<> h,
106  
            capy::executor_ref ex,
103  
            capy::executor_ref ex,
107  
            endpoint ep,
104  
            endpoint ep,
108  
            std::stop_token token,
105  
            std::stop_token token,
109  
            std::error_code* ec) = 0;
106  
            std::error_code* ec) = 0;
110  

107  

111  
        /** Shut down the socket for the given direction(s).
108  
        /** Shut down the socket for the given direction(s).
112  

109  

113  
            @param what The shutdown direction.
110  
            @param what The shutdown direction.
114  

111  

115  
            @return Error code on failure, empty on success.
112  
            @return Error code on failure, empty on success.
116  
        */
113  
        */
117  
        virtual std::error_code shutdown(shutdown_type what) noexcept = 0;
114  
        virtual std::error_code shutdown(shutdown_type what) noexcept = 0;
118  

115  

119  
        /// Return the platform socket descriptor.
116  
        /// Return the platform socket descriptor.
120  
        virtual native_handle_type native_handle() const noexcept = 0;
117  
        virtual native_handle_type native_handle() const noexcept = 0;
121  

118  

122  
        /** Request cancellation of pending asynchronous operations.
119  
        /** Request cancellation of pending asynchronous operations.
123  

120  

124  
            All outstanding operations complete with operation_canceled error.
121  
            All outstanding operations complete with operation_canceled error.
125  
            Check `ec == cond::canceled` for portable comparison.
122  
            Check `ec == cond::canceled` for portable comparison.
126  
        */
123  
        */
127  
        virtual void cancel() noexcept = 0;
124  
        virtual void cancel() noexcept = 0;
128  

125  

129  
        /** Set a socket option.
126  
        /** Set a socket option.
130  

127  

131  
            @param level The protocol level (e.g. `SOL_SOCKET`).
128  
            @param level The protocol level (e.g. `SOL_SOCKET`).
132  
            @param optname The option name (e.g. `SO_KEEPALIVE`).
129  
            @param optname The option name (e.g. `SO_KEEPALIVE`).
133  
            @param data Pointer to the option value.
130  
            @param data Pointer to the option value.
134  
            @param size Size of the option value in bytes.
131  
            @param size Size of the option value in bytes.
135  
            @return Error code on failure, empty on success.
132  
            @return Error code on failure, empty on success.
136  
        */
133  
        */
137  
        virtual std::error_code set_option(
134  
        virtual std::error_code set_option(
138  
            int level,
135  
            int level,
139  
            int optname,
136  
            int optname,
140  
            void const* data,
137  
            void const* data,
141  
            std::size_t size) noexcept = 0;
138  
            std::size_t size) noexcept = 0;
142  

139  

143  
        /** Get a socket option.
140  
        /** Get a socket option.
144  

141  

145  
            @param level The protocol level (e.g. `SOL_SOCKET`).
142  
            @param level The protocol level (e.g. `SOL_SOCKET`).
146  
            @param optname The option name (e.g. `SO_KEEPALIVE`).
143  
            @param optname The option name (e.g. `SO_KEEPALIVE`).
147  
            @param data Pointer to receive the option value.
144  
            @param data Pointer to receive the option value.
148  
            @param size On entry, the size of the buffer. On exit,
145  
            @param size On entry, the size of the buffer. On exit,
149  
                the size of the option value.
146  
                the size of the option value.
150  
            @return Error code on failure, empty on success.
147  
            @return Error code on failure, empty on success.
151  
        */
148  
        */
152  
        virtual std::error_code
149  
        virtual std::error_code
153  
        get_option(int level, int optname, void* data, std::size_t* size)
150  
        get_option(int level, int optname, void* data, std::size_t* size)
154  
            const noexcept = 0;
151  
            const noexcept = 0;
155  

152  

156  
        /// Return the cached local endpoint.
153  
        /// Return the cached local endpoint.
157  
        virtual endpoint local_endpoint() const noexcept = 0;
154  
        virtual endpoint local_endpoint() const noexcept = 0;
158  

155  

159  
        /// Return the cached remote endpoint.
156  
        /// Return the cached remote endpoint.
160  
        virtual endpoint remote_endpoint() const noexcept = 0;
157  
        virtual endpoint remote_endpoint() const noexcept = 0;
161  
    };
158  
    };
162  

159  

163  
    /// Represent the awaitable returned by @ref connect.
160  
    /// Represent the awaitable returned by @ref connect.
164  
    struct connect_awaitable
161  
    struct connect_awaitable
165  
        : detail::void_op_base<connect_awaitable>
162  
        : detail::void_op_base<connect_awaitable>
166  
    {
163  
    {
167  
        tcp_socket& s_;
164  
        tcp_socket& s_;
168  
        endpoint endpoint_;
165  
        endpoint endpoint_;
169  

166  

170  
        connect_awaitable(tcp_socket& s, endpoint ep) noexcept
167  
        connect_awaitable(tcp_socket& s, endpoint ep) noexcept
171  
            : s_(s), endpoint_(ep) {}
168  
            : s_(s), endpoint_(ep) {}
172  

169  

173  
        std::coroutine_handle<> dispatch(
170  
        std::coroutine_handle<> dispatch(
174  
            std::coroutine_handle<> h, capy::executor_ref ex) const
171  
            std::coroutine_handle<> h, capy::executor_ref ex) const
175  
        {
172  
        {
176  
            return s_.get().connect(h, ex, endpoint_, token_, &ec_);
173  
            return s_.get().connect(h, ex, endpoint_, token_, &ec_);
177  
        }
174  
        }
178  
    };
175  
    };
179  

176  

180  
public:
177  
public:
181  
    /** Destructor.
178  
    /** Destructor.
182  

179  

183  
        Closes the socket if open, cancelling any pending operations.
180  
        Closes the socket if open, cancelling any pending operations.
184  
    */
181  
    */
185  
    ~tcp_socket() override;
182  
    ~tcp_socket() override;
186  

183  

187  
    /** Construct a socket from an execution context.
184  
    /** Construct a socket from an execution context.
188  

185  

189  
        @param ctx The execution context that will own this socket.
186  
        @param ctx The execution context that will own this socket.
190  
    */
187  
    */
191  
    explicit tcp_socket(capy::execution_context& ctx);
188  
    explicit tcp_socket(capy::execution_context& ctx);
192  

189  

193  
    /** Construct a socket from an executor.
190  
    /** Construct a socket from an executor.
194  

191  

195  
        The socket is associated with the executor's context.
192  
        The socket is associated with the executor's context.
196  

193  

197  
        @param ex The executor whose context will own the socket.
194  
        @param ex The executor whose context will own the socket.
198  
    */
195  
    */
199  
    template<class Ex>
196  
    template<class Ex>
200  
        requires(!std::same_as<std::remove_cvref_t<Ex>, tcp_socket>) &&
197  
        requires(!std::same_as<std::remove_cvref_t<Ex>, tcp_socket>) &&
201  
        capy::Executor<Ex>
198  
        capy::Executor<Ex>
202  
    explicit tcp_socket(Ex const& ex) : tcp_socket(ex.context())
199  
    explicit tcp_socket(Ex const& ex) : tcp_socket(ex.context())
203  
    {
200  
    {
204  
    }
201  
    }
205  

202  

206  
    /** Move constructor.
203  
    /** Move constructor.
207  

204  

208  
        Transfers ownership of the socket resources.
205  
        Transfers ownership of the socket resources.
209  

206  

210  
        @param other The socket to move from.
207  
        @param other The socket to move from.
211  

208  

212  
        @pre No awaitables returned by @p other's methods exist.
209  
        @pre No awaitables returned by @p other's methods exist.
213  
        @pre @p other is not referenced as a peer in any outstanding
210  
        @pre @p other is not referenced as a peer in any outstanding
214  
            accept awaitable.
211  
            accept awaitable.
215  
        @pre The execution context associated with @p other must
212  
        @pre The execution context associated with @p other must
216  
            outlive this socket.
213  
            outlive this socket.
217  
    */
214  
    */
218  
    tcp_socket(tcp_socket&& other) noexcept : io_object(std::move(other)) {}
215  
    tcp_socket(tcp_socket&& other) noexcept : io_object(std::move(other)) {}
219  

216  

220  
    /** Move assignment operator.
217  
    /** Move assignment operator.
221  

218  

222  
        Closes any existing socket and transfers ownership.
219  
        Closes any existing socket and transfers ownership.
223  

220  

224  
        @param other The socket to move from.
221  
        @param other The socket to move from.
225  

222  

226  
        @pre No awaitables returned by either `*this` or @p other's
223  
        @pre No awaitables returned by either `*this` or @p other's
227  
            methods exist.
224  
            methods exist.
228  
        @pre Neither `*this` nor @p other is referenced as a peer in
225  
        @pre Neither `*this` nor @p other is referenced as a peer in
229  
            any outstanding accept awaitable.
226  
            any outstanding accept awaitable.
230  
        @pre The execution context associated with @p other must
227  
        @pre The execution context associated with @p other must
231  
            outlive this socket.
228  
            outlive this socket.
232  

229  

233  
        @return Reference to this socket.
230  
        @return Reference to this socket.
234  
    */
231  
    */
235  
    tcp_socket& operator=(tcp_socket&& other) noexcept
232  
    tcp_socket& operator=(tcp_socket&& other) noexcept
236  
    {
233  
    {
237  
        if (this != &other)
234  
        if (this != &other)
238  
        {
235  
        {
239  
            close();
236  
            close();
240  
            h_ = std::move(other.h_);
237  
            h_ = std::move(other.h_);
241  
        }
238  
        }
242  
        return *this;
239  
        return *this;
243  
    }
240  
    }
244  

241  

245  
    tcp_socket(tcp_socket const&)            = delete;
242  
    tcp_socket(tcp_socket const&)            = delete;
246  
    tcp_socket& operator=(tcp_socket const&) = delete;
243  
    tcp_socket& operator=(tcp_socket const&) = delete;
247  

244  

248  
    /** Open the socket.
245  
    /** Open the socket.
249  

246  

250  
        Creates a TCP socket and associates it with the platform
247  
        Creates a TCP socket and associates it with the platform
251  
        reactor (IOCP on Windows). Calling @ref connect on a closed
248  
        reactor (IOCP on Windows). Calling @ref connect on a closed
252  
        socket opens it automatically with the endpoint's address family,
249  
        socket opens it automatically with the endpoint's address family,
253  
        so explicit `open()` is only needed when socket options must be
250  
        so explicit `open()` is only needed when socket options must be
254  
        set before connecting.
251  
        set before connecting.
255  

252  

256  
        @param proto The protocol (IPv4 or IPv6). Defaults to
253  
        @param proto The protocol (IPv4 or IPv6). Defaults to
257  
            `tcp::v4()`.
254  
            `tcp::v4()`.
258  

255  

259  
        @throws std::system_error on failure.
256  
        @throws std::system_error on failure.
260  
    */
257  
    */
261  
    void open(tcp proto = tcp::v4());
258  
    void open(tcp proto = tcp::v4());
262  

259  

263  
    /** Bind the socket to a local endpoint.
260  
    /** Bind the socket to a local endpoint.
264  

261  

265  
        Associates the socket with a local address and port before
262  
        Associates the socket with a local address and port before
266  
        connecting. Useful for multi-homed hosts or source-port
263  
        connecting. Useful for multi-homed hosts or source-port
267  
        pinning.
264  
        pinning.
268  

265  

269  
        @param ep The local endpoint to bind to.
266  
        @param ep The local endpoint to bind to.
270  

267  

271  
        @return An error code indicating success or the reason for
268  
        @return An error code indicating success or the reason for
272  
            failure.
269  
            failure.
273  

270  

274  
        @par Error Conditions
271  
        @par Error Conditions
275  
        @li `errc::address_in_use`: The endpoint is already in use.
272  
        @li `errc::address_in_use`: The endpoint is already in use.
276  
        @li `errc::address_not_available`: The address is not
273  
        @li `errc::address_not_available`: The address is not
277  
            available on any local interface.
274  
            available on any local interface.
278  
        @li `errc::permission_denied`: Insufficient privileges to
275  
        @li `errc::permission_denied`: Insufficient privileges to
279  
            bind to the endpoint (e.g., privileged port).
276  
            bind to the endpoint (e.g., privileged port).
280  

277  

281  
        @throws std::logic_error if the socket is not open.
278  
        @throws std::logic_error if the socket is not open.
282  
    */
279  
    */
283  
    [[nodiscard]] std::error_code bind(endpoint ep);
280  
    [[nodiscard]] std::error_code bind(endpoint ep);
284  

281  

285  
    /** Close the socket.
282  
    /** Close the socket.
286  

283  

287  
        Releases socket resources. Any pending operations complete
284  
        Releases socket resources. Any pending operations complete
288  
        with `errc::operation_canceled`.
285  
        with `errc::operation_canceled`.
289  
    */
286  
    */
290  
    void close();
287  
    void close();
291  

288  

292  
    /** Check if the socket is open.
289  
    /** Check if the socket is open.
293  

290  

294  
        @return `true` if the socket is open and ready for operations.
291  
        @return `true` if the socket is open and ready for operations.
295  
    */
292  
    */
296  
    bool is_open() const noexcept
293  
    bool is_open() const noexcept
297  
    {
294  
    {
298  
#if BOOST_COROSIO_HAS_IOCP && !defined(BOOST_COROSIO_MRDOCS)
295  
#if BOOST_COROSIO_HAS_IOCP && !defined(BOOST_COROSIO_MRDOCS)
299  
        return h_ && get().native_handle() != ~native_handle_type(0);
296  
        return h_ && get().native_handle() != ~native_handle_type(0);
300  
#else
297  
#else
301  
        return h_ && get().native_handle() >= 0;
298  
        return h_ && get().native_handle() >= 0;
302  
#endif
299  
#endif
303  
    }
300  
    }
304  

301  

305  
    /** Initiate an asynchronous connect operation.
302  
    /** Initiate an asynchronous connect operation.
306  

303  

307  
        If the socket is not already open, it is opened automatically
304  
        If the socket is not already open, it is opened automatically
308  
        using the address family of @p ep (IPv4 or IPv6). If the socket
305  
        using the address family of @p ep (IPv4 or IPv6). If the socket
309  
        is already open, the existing file descriptor is used as-is.
306  
        is already open, the existing file descriptor is used as-is.
310  

307  

311  
        The operation supports cancellation via `std::stop_token` through
308  
        The operation supports cancellation via `std::stop_token` through
312  
        the affine awaitable protocol. If the associated stop token is
309  
        the affine awaitable protocol. If the associated stop token is
313  
        triggered, the operation completes immediately with
310  
        triggered, the operation completes immediately with
314  
        `errc::operation_canceled`.
311  
        `errc::operation_canceled`.
315  

312  

316  
        @param ep The remote endpoint to connect to.
313  
        @param ep The remote endpoint to connect to.
317  

314  

318  
        @return An awaitable that completes with `io_result<>`.
315  
        @return An awaitable that completes with `io_result<>`.
319  
            Returns success (default error_code) on successful connection,
316  
            Returns success (default error_code) on successful connection,
320  
            or an error code on failure including:
317  
            or an error code on failure including:
321  
            - connection_refused: No server listening at endpoint
318  
            - connection_refused: No server listening at endpoint
322  
            - timed_out: Connection attempt timed out
319  
            - timed_out: Connection attempt timed out
323  
            - network_unreachable: No route to host
320  
            - network_unreachable: No route to host
324  
            - operation_canceled: Cancelled via stop_token or cancel().
321  
            - operation_canceled: Cancelled via stop_token or cancel().
325  
                Check `ec == cond::canceled` for portable comparison.
322  
                Check `ec == cond::canceled` for portable comparison.
326  

323  

327  
        @throws std::system_error if the socket needs to be opened
324  
        @throws std::system_error if the socket needs to be opened
328  
            and the open fails.
325  
            and the open fails.
329  

326  

330  
        @par Preconditions
327  
        @par Preconditions
331  
        This socket must outlive the returned awaitable.
328  
        This socket must outlive the returned awaitable.
332  

329  

333  
        @par Example
330  
        @par Example
334  
        @code
331  
        @code
335  
        // Socket opened automatically with correct address family:
332  
        // Socket opened automatically with correct address family:
336  
        auto [ec] = co_await s.connect(endpoint);
333  
        auto [ec] = co_await s.connect(endpoint);
337  
        if (ec) { ... }
334  
        if (ec) { ... }
338  
        @endcode
335  
        @endcode
339  
    */
336  
    */
340  
    auto connect(endpoint ep)
337  
    auto connect(endpoint ep)
341  
    {
338  
    {
342  
        if (!is_open())
339  
        if (!is_open())
343  
            open(ep.is_v6() ? tcp::v6() : tcp::v4());
340  
            open(ep.is_v6() ? tcp::v6() : tcp::v4());
344  
        return connect_awaitable(*this, ep);
341  
        return connect_awaitable(*this, ep);
345  
    }
342  
    }
346  

343  

347  
    /** Cancel any pending asynchronous operations.
344  
    /** Cancel any pending asynchronous operations.
348  

345  

349  
        All outstanding operations complete with `errc::operation_canceled`.
346  
        All outstanding operations complete with `errc::operation_canceled`.
350  
        Check `ec == cond::canceled` for portable comparison.
347  
        Check `ec == cond::canceled` for portable comparison.
351  
    */
348  
    */
352  
    void cancel();
349  
    void cancel();
353  

350  

354  
    /** Get the native socket handle.
351  
    /** Get the native socket handle.
355  

352  

356  
        Returns the underlying platform-specific socket descriptor.
353  
        Returns the underlying platform-specific socket descriptor.
357  
        On POSIX systems this is an `int` file descriptor.
354  
        On POSIX systems this is an `int` file descriptor.
358  
        On Windows this is a `SOCKET` handle.
355  
        On Windows this is a `SOCKET` handle.
359  

356  

360  
        @return The native socket handle, or -1/INVALID_SOCKET if not open.
357  
        @return The native socket handle, or -1/INVALID_SOCKET if not open.
361  

358  

362  
        @par Preconditions
359  
        @par Preconditions
363  
        None. May be called on closed sockets.
360  
        None. May be called on closed sockets.
364  
    */
361  
    */
365  
    native_handle_type native_handle() const noexcept;
362  
    native_handle_type native_handle() const noexcept;
366  

363  

367  
    /** Disable sends or receives on the socket.
364  
    /** Disable sends or receives on the socket.
368  

365  

369  
        TCP connections are full-duplex: each direction (send and receive)
366  
        TCP connections are full-duplex: each direction (send and receive)
370  
        operates independently. This function allows you to close one or
367  
        operates independently. This function allows you to close one or
371  
        both directions without destroying the socket.
368  
        both directions without destroying the socket.
372  

369  

373  
        @li @ref shutdown_send sends a TCP FIN packet to the peer,
370  
        @li @ref shutdown_send sends a TCP FIN packet to the peer,
374  
            signaling that you have no more data to send. You can still
371  
            signaling that you have no more data to send. You can still
375  
            receive data until the peer also closes their send direction.
372  
            receive data until the peer also closes their send direction.
376  
            This is the most common use case, typically called before
373  
            This is the most common use case, typically called before
377  
            close() to ensure graceful connection termination.
374  
            close() to ensure graceful connection termination.
378  

375  

379  
        @li @ref shutdown_receive disables reading on the socket. This
376  
        @li @ref shutdown_receive disables reading on the socket. This
380  
            does NOT send anything to the peer - they are not informed
377  
            does NOT send anything to the peer - they are not informed
381  
            and may continue sending data. Subsequent reads will fail
378  
            and may continue sending data. Subsequent reads will fail
382  
            or return end-of-file. Incoming data may be discarded or
379  
            or return end-of-file. Incoming data may be discarded or
383  
            buffered depending on the operating system.
380  
            buffered depending on the operating system.
384  

381  

385  
        @li @ref shutdown_both combines both effects: sends a FIN and
382  
        @li @ref shutdown_both combines both effects: sends a FIN and
386  
            disables reading.
383  
            disables reading.
387  

384  

388  
        When the peer shuts down their send direction (sends a FIN),
385  
        When the peer shuts down their send direction (sends a FIN),
389  
        subsequent read operations will complete with `capy::cond::eof`.
386  
        subsequent read operations will complete with `capy::cond::eof`.
390  
        Use the portable condition test rather than comparing error
387  
        Use the portable condition test rather than comparing error
391  
        codes directly:
388  
        codes directly:
392  

389  

393  
        @code
390  
        @code
394  
        auto [ec, n] = co_await sock.read_some(buffer);
391  
        auto [ec, n] = co_await sock.read_some(buffer);
395  
        if (ec == capy::cond::eof)
392  
        if (ec == capy::cond::eof)
396  
        {
393  
        {
397  
            // Peer closed their send direction
394  
            // Peer closed their send direction
398  
        }
395  
        }
399  
        @endcode
396  
        @endcode
400  

397  

401  
        Any error from the underlying system call is silently discarded
398  
        Any error from the underlying system call is silently discarded
402  
        because it is unlikely to be helpful.
399  
        because it is unlikely to be helpful.
403  

400  

404  
        @param what Determines what operations will no longer be allowed.
401  
        @param what Determines what operations will no longer be allowed.
405  
    */
402  
    */
406  
    void shutdown(shutdown_type what);
403  
    void shutdown(shutdown_type what);
407  

404  

408  
    /** Set a socket option.
405  
    /** Set a socket option.
409  

406  

410  
        Applies a type-safe socket option to the underlying socket.
407  
        Applies a type-safe socket option to the underlying socket.
411  
        The option type encodes the protocol level and option name.
408  
        The option type encodes the protocol level and option name.
412  

409  

413  
        @par Example
410  
        @par Example
414  
        @code
411  
        @code
415  
        sock.set_option( socket_option::no_delay( true ) );
412  
        sock.set_option( socket_option::no_delay( true ) );
416  
        sock.set_option( socket_option::receive_buffer_size( 65536 ) );
413  
        sock.set_option( socket_option::receive_buffer_size( 65536 ) );
417  
        @endcode
414  
        @endcode
418  

415  

419  
        @param opt The option to set.
416  
        @param opt The option to set.
420  

417  

421  
        @throws std::logic_error if the socket is not open.
418  
        @throws std::logic_error if the socket is not open.
422  
        @throws std::system_error on failure.
419  
        @throws std::system_error on failure.
423  
    */
420  
    */
424  
    template<class Option>
421  
    template<class Option>
425  
    void set_option(Option const& opt)
422  
    void set_option(Option const& opt)
426  
    {
423  
    {
427  
        if (!is_open())
424  
        if (!is_open())
428  
            detail::throw_logic_error("set_option: socket not open");
425  
            detail::throw_logic_error("set_option: socket not open");
429  
        std::error_code ec = get().set_option(
426  
        std::error_code ec = get().set_option(
430  
            Option::level(), Option::name(), opt.data(), opt.size());
427  
            Option::level(), Option::name(), opt.data(), opt.size());
431  
        if (ec)
428  
        if (ec)
432  
            detail::throw_system_error(ec, "tcp_socket::set_option");
429  
            detail::throw_system_error(ec, "tcp_socket::set_option");
433  
    }
430  
    }
434  

431  

435  
    /** Get a socket option.
432  
    /** Get a socket option.
436  

433  

437  
        Retrieves the current value of a type-safe socket option.
434  
        Retrieves the current value of a type-safe socket option.
438  

435  

439  
        @par Example
436  
        @par Example
440  
        @code
437  
        @code
441  
        auto nd = sock.get_option<socket_option::no_delay>();
438  
        auto nd = sock.get_option<socket_option::no_delay>();
442  
        if ( nd.value() )
439  
        if ( nd.value() )
443  
            // Nagle's algorithm is disabled
440  
            // Nagle's algorithm is disabled
444  
        @endcode
441  
        @endcode
445  

442  

446  
        @return The current option value.
443  
        @return The current option value.
447  

444  

448  
        @throws std::logic_error if the socket is not open.
445  
        @throws std::logic_error if the socket is not open.
449  
        @throws std::system_error on failure.
446  
        @throws std::system_error on failure.
450  
    */
447  
    */
451  
    template<class Option>
448  
    template<class Option>
452  
    Option get_option() const
449  
    Option get_option() const
453  
    {
450  
    {
454  
        if (!is_open())
451  
        if (!is_open())
455  
            detail::throw_logic_error("get_option: socket not open");
452  
            detail::throw_logic_error("get_option: socket not open");
456  
        Option opt{};
453  
        Option opt{};
457  
        std::size_t sz = opt.size();
454  
        std::size_t sz = opt.size();
458  
        std::error_code ec =
455  
        std::error_code ec =
459  
            get().get_option(Option::level(), Option::name(), opt.data(), &sz);
456  
            get().get_option(Option::level(), Option::name(), opt.data(), &sz);
460  
        if (ec)
457  
        if (ec)
461  
            detail::throw_system_error(ec, "tcp_socket::get_option");
458  
            detail::throw_system_error(ec, "tcp_socket::get_option");
462  
        opt.resize(sz);
459  
        opt.resize(sz);
463  
        return opt;
460  
        return opt;
464  
    }
461  
    }
465  

462  

466  
    /** Get the local endpoint of the socket.
463  
    /** Get the local endpoint of the socket.
467  

464  

468  
        Returns the local address and port to which the socket is bound.
465  
        Returns the local address and port to which the socket is bound.
469  
        For a connected socket, this is the local side of the connection.
466  
        For a connected socket, this is the local side of the connection.
470  
        The endpoint is cached when the connection is established.
467  
        The endpoint is cached when the connection is established.
471  

468  

472  
        @return The local endpoint, or a default endpoint (0.0.0.0:0) if
469  
        @return The local endpoint, or a default endpoint (0.0.0.0:0) if
473  
            the socket is not connected.
470  
            the socket is not connected.
474  

471  

475  
        @par Thread Safety
472  
        @par Thread Safety
476  
        The cached endpoint value is set during connect/accept completion
473  
        The cached endpoint value is set during connect/accept completion
477  
        and cleared during close(). This function may be called concurrently
474  
        and cleared during close(). This function may be called concurrently
478  
        with I/O operations, but must not be called concurrently with
475  
        with I/O operations, but must not be called concurrently with
479  
        connect(), accept(), or close().
476  
        connect(), accept(), or close().
480  
    */
477  
    */
481  
    endpoint local_endpoint() const noexcept;
478  
    endpoint local_endpoint() const noexcept;
482  

479  

483  
    /** Get the remote endpoint of the socket.
480  
    /** Get the remote endpoint of the socket.
484  

481  

485  
        Returns the remote address and port to which the socket is connected.
482  
        Returns the remote address and port to which the socket is connected.
486  
        The endpoint is cached when the connection is established.
483  
        The endpoint is cached when the connection is established.
487  

484  

488  
        @return The remote endpoint, or a default endpoint (0.0.0.0:0) if
485  
        @return The remote endpoint, or a default endpoint (0.0.0.0:0) if
489  
            the socket is not connected.
486  
            the socket is not connected.
490  

487  

491  
        @par Thread Safety
488  
        @par Thread Safety
492  
        The cached endpoint value is set during connect/accept completion
489  
        The cached endpoint value is set during connect/accept completion
493  
        and cleared during close(). This function may be called concurrently
490  
        and cleared during close(). This function may be called concurrently
494  
        with I/O operations, but must not be called concurrently with
491  
        with I/O operations, but must not be called concurrently with
495  
        connect(), accept(), or close().
492  
        connect(), accept(), or close().
496  
    */
493  
    */
497  
    endpoint remote_endpoint() const noexcept;
494  
    endpoint remote_endpoint() const noexcept;
498  

495  

499  
protected:
496  
protected:
500  
    tcp_socket() noexcept = default;
497  
    tcp_socket() noexcept = default;
501  

498  

502  
    explicit tcp_socket(handle h) noexcept : io_object(std::move(h)) {}
499  
    explicit tcp_socket(handle h) noexcept : io_object(std::move(h)) {}
503  

500  

504  
private:
501  
private:
505  
    friend class tcp_acceptor;
502  
    friend class tcp_acceptor;
506  

503  

507  
    /// Open the socket for the given protocol triple.
504  
    /// Open the socket for the given protocol triple.
508  
    void open_for_family(int family, int type, int protocol);
505  
    void open_for_family(int family, int type, int protocol);
509  

506  

510  
    inline implementation& get() const noexcept
507  
    inline implementation& get() const noexcept
511  
    {
508  
    {
512  
        return *static_cast<implementation*>(h_.get());
509  
        return *static_cast<implementation*>(h_.get());
513  
    }
510  
    }
514  
};
511  
};
515  

512  

516  
} // namespace boost::corosio
513  
} // namespace boost::corosio
517  

514  

518  
#endif
515  
#endif