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   
        buffer_param,
50   
        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   
        buffer_param,
58   
        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,
71   
        int level,
72   
        int optname,
72   
        int optname,
73   
        void const* data,
73   
        void const* data,
74   
        std::size_t size) noexcept override;
74   
        std::size_t size) noexcept override;
75   
    std::error_code
75   
    std::error_code
76   
    get_option(int level, int optname, void* data, std::size_t* size)
76   
    get_option(int level, int optname, void* data, std::size_t* size)
77   
        const noexcept override;
77   
        const noexcept override;
78  
78  
79   
    endpoint local_endpoint() const noexcept override
79   
    endpoint local_endpoint() const noexcept override
80   
    {
80   
    {
81   
        return local_endpoint_;
81   
        return local_endpoint_;
82   
    }
82   
    }
83   
    endpoint remote_endpoint() const noexcept override
83   
    endpoint remote_endpoint() const noexcept override
84   
    {
84   
    {
85   
        return remote_endpoint_;
85   
        return remote_endpoint_;
86   
    }
86   
    }
87   
    bool is_open() const noexcept
87   
    bool is_open() const noexcept
88   
    {
88   
    {
89   
        return fd_ >= 0;
89   
        return fd_ >= 0;
90   
    }
90   
    }
91   
    void cancel() noexcept override;
91   
    void cancel() noexcept override;
92   
    void cancel_single_op(select_op& op) noexcept;
92   
    void cancel_single_op(select_op& op) noexcept;
93   
    void close_socket() noexcept;
93   
    void close_socket() noexcept;
94   
    void set_socket(int fd) noexcept
94   
    void set_socket(int fd) noexcept
95   
    {
95   
    {
96   
        fd_ = fd;
96   
        fd_ = fd;
97   
    }
97   
    }
98   
    void set_endpoints(endpoint local, endpoint remote) noexcept
98   
    void set_endpoints(endpoint local, endpoint remote) noexcept
99   
    {
99   
    {
100   
        local_endpoint_  = local;
100   
        local_endpoint_  = local;
101   
        remote_endpoint_ = remote;
101   
        remote_endpoint_ = remote;
102   
    }
102   
    }
103  
103  
104   
    select_connect_op conn_;
104   
    select_connect_op conn_;
105   
    select_read_op rd_;
105   
    select_read_op rd_;
106   
    select_write_op wr_;
106   
    select_write_op wr_;
107  
107  
108   
private:
108   
private:
109   
    select_socket_service& svc_;
109   
    select_socket_service& svc_;
110   
    int fd_ = -1;
110   
    int fd_ = -1;
111   
    endpoint local_endpoint_;
111   
    endpoint local_endpoint_;
112   
    endpoint remote_endpoint_;
112   
    endpoint remote_endpoint_;
113   
};
113   
};
114  
114  
115   
} // namespace boost::corosio::detail
115   
} // namespace boost::corosio::detail
116  
116  
117   
#endif // BOOST_COROSIO_HAS_SELECT
117   
#endif // BOOST_COROSIO_HAS_SELECT
118  
118  
119   
#endif // BOOST_COROSIO_NATIVE_DETAIL_SELECT_SELECT_SOCKET_HPP
119   
#endif // BOOST_COROSIO_NATIVE_DETAIL_SELECT_SELECT_SOCKET_HPP