Index/src / corosio / src / detail / epoll / acceptors.hpp
Prev Index Next

src/corosio/src/detail/epoll/acceptors.hpp

100.0% Lines (8/8) 100.0% Functions (6/6) -% Branches (0/0)
src/corosio/src/detail/epoll/acceptors.hpp
Line Hits Source Code
1 //
2 // Copyright (c) 2026 Steve Gerbino
3 //
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)
6 //
7 // Official repository: https://github.com/cppalliance/corosio
8 //
9
10 #ifndef BOOST_COROSIO_DETAIL_EPOLL_ACCEPTORS_HPP
11 #define BOOST_COROSIO_DETAIL_EPOLL_ACCEPTORS_HPP
12
13 #include <boost/corosio/detail/platform.hpp>
14
15 #if BOOST_COROSIO_HAS_EPOLL
16
17 #include <boost/corosio/detail/config.hpp>
18 #include <boost/corosio/tcp_acceptor.hpp>
19 #include <boost/capy/ex/executor_ref.hpp>
20 #include <boost/capy/ex/execution_context.hpp>
21 #include "src/detail/intrusive.hpp"
22 #include "src/detail/socket_service.hpp"
23
24 #include "src/detail/epoll/op.hpp"
25 #include "src/detail/epoll/scheduler.hpp"
26
27 #include <memory>
28 #include <mutex>
29 #include <unordered_map>
30
31 namespace boost::corosio::detail {
32
33 class epoll_acceptor_service;
34 class epoll_acceptor_impl;
35 class epoll_socket_service;
36
37 /// Acceptor implementation for epoll backend.
38 class epoll_acceptor_impl
39 : public tcp_acceptor::implementation
40 , public std::enable_shared_from_this<epoll_acceptor_impl>
41 , public intrusive_list<epoll_acceptor_impl>::node
42 {
43 friend class epoll_acceptor_service;
44
45 public:
46 explicit epoll_acceptor_impl(epoll_acceptor_service& svc) noexcept;
47
48 std::coroutine_handle<> accept(
49 std::coroutine_handle<>,
50 capy::executor_ref,
51 std::stop_token,
52 std::error_code*,
53 io_object::implementation**) override;
54
55 int native_handle() const noexcept { return fd_; }
56 4746 endpoint local_endpoint() const noexcept override { return local_endpoint_; }
57 4949 bool is_open() const noexcept override { return fd_ >= 0; }
58 void cancel() noexcept override;
59 void cancel_single_op(epoll_op& op) noexcept;
60 void close_socket() noexcept;
61 62 void set_local_endpoint(endpoint ep) noexcept { local_endpoint_ = ep; }
62
63 9407 epoll_acceptor_service& service() noexcept { return svc_; }
64
65 epoll_accept_op acc_;
66 descriptor_state desc_state_;
67
68 private:
69 epoll_acceptor_service& svc_;
70 int fd_ = -1;
71 endpoint local_endpoint_;
72 };
73
74 /** State for epoll acceptor service. */
75 class epoll_acceptor_state
76 {
77 public:
78 203 explicit epoll_acceptor_state(epoll_scheduler& sched) noexcept
79 203 : sched_(sched)
80 {
81 203 }
82
83 epoll_scheduler& sched_;
84 std::mutex mutex_;
85 intrusive_list<epoll_acceptor_impl> acceptor_list_;
86 std::unordered_map<epoll_acceptor_impl*, std::shared_ptr<epoll_acceptor_impl>> acceptor_ptrs_;
87 };
88
89 /** epoll acceptor service implementation.
90
91 Inherits from acceptor_service to enable runtime polymorphism.
92 Uses key_type = acceptor_service for service lookup.
93 */
94 class epoll_acceptor_service : public acceptor_service
95 {
96 public:
97 explicit epoll_acceptor_service(capy::execution_context& ctx);
98 ~epoll_acceptor_service();
99
100 epoll_acceptor_service(epoll_acceptor_service const&) = delete;
101 epoll_acceptor_service& operator=(epoll_acceptor_service const&) = delete;
102
103 void shutdown() override;
104
105 io_object::implementation* construct() override;
106 void destroy(io_object::implementation*) override;
107 void close(io_object::handle&) override;
108 std::error_code open_acceptor(
109 tcp_acceptor::implementation& impl,
110 endpoint ep,
111 int backlog) override;
112
113 4834 epoll_scheduler& scheduler() const noexcept { return state_->sched_; }
114 void post(epoll_op* op);
115 void work_started() noexcept;
116 void work_finished() noexcept;
117
118 /** Get the socket service for creating peer sockets during accept. */
119 epoll_socket_service* socket_service() const noexcept;
120
121 private:
122 capy::execution_context& ctx_;
123 std::unique_ptr<epoll_acceptor_state> state_;
124 };
125
126 } // namespace boost::corosio::detail
127
128 #endif // BOOST_COROSIO_HAS_EPOLL
129
130 #endif // BOOST_COROSIO_DETAIL_EPOLL_ACCEPTORS_HPP
131