1   
//
1   
//
2   
// Copyright (c) 2026 Steve Gerbino
2   
// Copyright (c) 2026 Steve Gerbino
3   
//
3   
//
4   
// Distributed under the Boost Software License, Version 1.0. (See accompanying
4   
// Distributed under the Boost Software License, Version 1.0. (See accompanying
5   
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
5   
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
6   
//
6   
//
7   
// Official repository: https://github.com/cppalliance/corosio
7   
// Official repository: https://github.com/cppalliance/corosio
8   
//
8   
//
9  
9  
10   
#ifndef BOOST_COROSIO_NATIVE_DETAIL_SELECT_SELECT_SOCKET_HPP
10   
#ifndef BOOST_COROSIO_NATIVE_DETAIL_SELECT_SELECT_SOCKET_HPP
11   
#define BOOST_COROSIO_NATIVE_DETAIL_SELECT_SELECT_SOCKET_HPP
11   
#define BOOST_COROSIO_NATIVE_DETAIL_SELECT_SELECT_SOCKET_HPP
12  
12  
13   
#include <boost/corosio/detail/platform.hpp>
13   
#include <boost/corosio/detail/platform.hpp>
14  
14  
15   
#if BOOST_COROSIO_HAS_SELECT
15   
#if BOOST_COROSIO_HAS_SELECT
16  
16  
17   
#include <boost/corosio/tcp_socket.hpp>
17   
#include <boost/corosio/tcp_socket.hpp>
18   
#include <boost/capy/ex/executor_ref.hpp>
18   
#include <boost/capy/ex/executor_ref.hpp>
19   
#include <boost/corosio/detail/intrusive.hpp>
19   
#include <boost/corosio/detail/intrusive.hpp>
20  
20  
21   
#include <boost/corosio/native/detail/select/select_op.hpp>
21   
#include <boost/corosio/native/detail/select/select_op.hpp>
22  
22  
23   
#include <memory>
23   
#include <memory>
24  
24  
25   
namespace boost::corosio::detail {
25   
namespace boost::corosio::detail {
26  
26  
27   
class select_socket_service;
27   
class select_socket_service;
28  
28  
29   
/// Socket implementation for select backend.
29   
/// Socket implementation for select backend.
30   
class select_socket final
30   
class select_socket final
31   
    : public tcp_socket::implementation
31   
    : public tcp_socket::implementation
32   
    , public std::enable_shared_from_this<select_socket>
32   
    , public std::enable_shared_from_this<select_socket>
33   
    , public intrusive_list<select_socket>::node
33   
    , public intrusive_list<select_socket>::node
34   
{
34   
{
35   
    friend class select_socket_service;
35   
    friend class select_socket_service;
36  
36  
37   
public:
37   
public:
38   
    explicit select_socket(select_socket_service& svc) noexcept;
38   
    explicit select_socket(select_socket_service& svc) noexcept;
39  
39  
40   
    std::coroutine_handle<> connect(
40   
    std::coroutine_handle<> connect(
41   
        std::coroutine_handle<>,
41   
        std::coroutine_handle<>,
42   
        capy::executor_ref,
42   
        capy::executor_ref,
43   
        endpoint,
43   
        endpoint,
44   
        std::stop_token,
44   
        std::stop_token,
45   
        std::error_code*) override;
45   
        std::error_code*) override;
46  
46  
47   
    std::coroutine_handle<> read_some(
47   
    std::coroutine_handle<> read_some(
48   
        std::coroutine_handle<>,
48   
        std::coroutine_handle<>,
49   
        capy::executor_ref,
49   
        capy::executor_ref,
50   
        io_buffer_param,
50   
        io_buffer_param,
51   
        std::stop_token,
51   
        std::stop_token,
52   
        std::error_code*,
52   
        std::error_code*,
53   
        std::size_t*) override;
53   
        std::size_t*) override;
54  
54  
55   
    std::coroutine_handle<> write_some(
55   
    std::coroutine_handle<> write_some(
56   
        std::coroutine_handle<>,
56   
        std::coroutine_handle<>,
57   
        capy::executor_ref,
57   
        capy::executor_ref,
58   
        io_buffer_param,
58   
        io_buffer_param,
59   
        std::stop_token,
59   
        std::stop_token,
60   
        std::error_code*,
60   
        std::error_code*,
61   
        std::size_t*) override;
61   
        std::size_t*) override;
62  
62  
63   
    std::error_code shutdown(tcp_socket::shutdown_type what) noexcept override;
63   
    std::error_code shutdown(tcp_socket::shutdown_type what) noexcept override;
64  
64  
65   
    native_handle_type native_handle() const noexcept override
65   
    native_handle_type native_handle() const noexcept override
66   
    {
66   
    {
67   
        return fd_;
67   
        return fd_;
68   
    }
68   
    }
69  
69  
70   
    std::error_code set_option(
70   
    std::error_code set_option(
71   
        int level, int optname,
71   
        int level, int optname,
72   
        void const* data, std::size_t size) noexcept override;
72   
        void const* data, std::size_t size) noexcept override;
73   
    std::error_code get_option(
73   
    std::error_code get_option(
74   
        int level, int optname,
74   
        int level, int optname,
75   
        void* data, std::size_t* size) const noexcept override;
75   
        void* data, std::size_t* size) const noexcept override;
76  
76  
77   
    endpoint local_endpoint() const noexcept override
77   
    endpoint local_endpoint() const noexcept override
78   
    {
78   
    {
79   
        return local_endpoint_;
79   
        return local_endpoint_;
80   
    }
80   
    }
81   
    endpoint remote_endpoint() const noexcept override
81   
    endpoint remote_endpoint() const noexcept override
82   
    {
82   
    {
83   
        return remote_endpoint_;
83   
        return remote_endpoint_;
84   
    }
84   
    }
85   
    bool is_open() const noexcept
85   
    bool is_open() const noexcept
86   
    {
86   
    {
87   
        return fd_ >= 0;
87   
        return fd_ >= 0;
88   
    }
88   
    }
89   
    void cancel() noexcept override;
89   
    void cancel() noexcept override;
90   
    void cancel_single_op(select_op& op) noexcept;
90   
    void cancel_single_op(select_op& op) noexcept;
91   
    void close_socket() noexcept;
91   
    void close_socket() noexcept;
92   
    void set_socket(int fd) noexcept
92   
    void set_socket(int fd) noexcept
93   
    {
93   
    {
94   
        fd_ = fd;
94   
        fd_ = fd;
95   
    }
95   
    }
96   
    void set_endpoints(endpoint local, endpoint remote) noexcept
96   
    void set_endpoints(endpoint local, endpoint remote) noexcept
97   
    {
97   
    {
98   
        local_endpoint_  = local;
98   
        local_endpoint_  = local;
99   
        remote_endpoint_ = remote;
99   
        remote_endpoint_ = remote;
100   
    }
100   
    }
101  
101  
102   
    select_connect_op conn_;
102   
    select_connect_op conn_;
103   
    select_read_op rd_;
103   
    select_read_op rd_;
104   
    select_write_op wr_;
104   
    select_write_op wr_;
105  
105  
106   
private:
106   
private:
107   
    select_socket_service& svc_;
107   
    select_socket_service& svc_;
108   
    int fd_ = -1;
108   
    int fd_ = -1;
109   
    endpoint local_endpoint_;
109   
    endpoint local_endpoint_;
110   
    endpoint remote_endpoint_;
110   
    endpoint remote_endpoint_;
111   
};
111   
};
112  
112  
113   
} // namespace boost::corosio::detail
113   
} // namespace boost::corosio::detail
114  
114  
115   
#endif // BOOST_COROSIO_HAS_SELECT
115   
#endif // BOOST_COROSIO_HAS_SELECT
116  
116  
117   
#endif // BOOST_COROSIO_NATIVE_DETAIL_SELECT_SELECT_SOCKET_HPP
117   
#endif // BOOST_COROSIO_NATIVE_DETAIL_SELECT_SELECT_SOCKET_HPP