include/boost/corosio/native/detail/select/select_tcp_service.hpp

73.9% Lines (65/88) 87.5% List of functions (14/16) 50.0% Branches (11/22)

[ ] prev/next file n p prev/next uncovered line

select_tcp_service.hpp

f(x) Functions (16)

Function Calls Lines Branches Blocks

Line	Branch	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_NATIVE_DETAIL_SELECT_SELECT_TCP_SERVICE_HPP
11			#define BOOST_COROSIO_NATIVE_DETAIL_SELECT_SELECT_TCP_SERVICE_HPP
12
13			#include <boost/corosio/detail/platform.hpp>
14
15			#if BOOST_COROSIO_HAS_SELECT
16
17			#include <boost/corosio/detail/config.hpp>
18			#include <boost/corosio/detail/tcp_service.hpp>
19
20			#include <boost/corosio/native/detail/select/select_tcp_socket.hpp>
21			#include <boost/corosio/native/detail/select/select_scheduler.hpp>
22			#include <boost/corosio/native/detail/reactor/reactor_socket_service.hpp>
23
24			#include <boost/corosio/native/detail/reactor/reactor_op_complete.hpp>
25
26			#include <coroutine>
27			#include <mutex>
28
29			#include <errno.h>
30			#include <fcntl.h>
31			#include <netinet/in.h>
32			#include <netinet/tcp.h>
33			#include <sys/select.h>
34			#include <sys/socket.h>
35			#include <unistd.h>
36
37			/*
38			Each I/O op tries the syscall speculatively; only registers with
39			the reactor on EAGAIN. Fd is registered once at open time and
40			stays registered until close. The reactor only marks ready_events_;
41			actual I/O happens in invoke_deferred_io(). cancel() captures
42			shared_from_this() into op.impl_ptr to keep the impl alive.
43			*/
44
45			namespace boost::corosio::detail {
46
47			/** select TCP service implementation.
48
49			Inherits from tcp_service to enable runtime polymorphism.
50			Uses key_type = tcp_service for service lookup.
51			*/
52			class BOOST_COROSIO_DECL select_tcp_service final
53			: public reactor_socket_service<
54			select_tcp_service,
55			tcp_service,
56			select_scheduler,
57			select_tcp_socket>
58			{
59			public:
60		484x	explicit select_tcp_service(capy::execution_context& ctx)
61		242x	: reactor_socket_service(ctx)
62		484x	{
63		484x	}
64
65			std::error_code open_socket(
66			tcp_socket::implementation& impl,
67			int family,
68			int type,
69			int protocol) override;
70
71			std::error_code
72			bind_socket(tcp_socket::implementation& impl, endpoint ep) override;
73			};
74
75			inline void
76		✗	select_connect_op::cancel() noexcept
77			{
78		✗	if (socket_impl_)
79		✗	socket_impl_->cancel_single_op(*this);
80			else
81		✗	request_cancel();
82		✗	}
83
84			inline void
85		30x	select_read_op::cancel() noexcept
86			{
87	1/2 ✓ Branch 0 taken 30 times. ✗ Branch 1 not taken.	30x	if (socket_impl_)
88		30x	socket_impl_->cancel_single_op(*this);
89			else
90		✗	request_cancel();
91		30x	}
92
93			inline void
94		✗	select_write_op::cancel() noexcept
95			{
96		✗	if (socket_impl_)
97		✗	socket_impl_->cancel_single_op(*this);
98			else
99		✗	request_cancel();
100		✗	}
101
102			inline void
103		203956x	select_op::operator()()
104			{
105		203956x	complete_io_op(*this);
106		203956x	}
107
108			inline void
109		1739x	select_connect_op::operator()()
110			{
111		1739x	complete_connect_op(*this);
112		1739x	}
113
114		10494x	inline select_tcp_socket::select_tcp_socket(select_tcp_service& svc) noexcept
115		5247x	: reactor_stream_socket(svc)
116		10494x	{
117		10494x	}
118
119		10494x	inline select_tcp_socket::~select_tcp_socket() = default;
120
121			inline std::coroutine_handle<>
122		1739x	select_tcp_socket::connect(
123			std::coroutine_handle<> h,
124			capy::executor_ref ex,
125			endpoint ep,
126			std::stop_token token,
127			std::error_code* ec)
128			{
129		1739x	auto result = do_connect(h, ex, ep, token, ec);
130			// Rebuild fd_sets so select() watches for writability
131	1/2 ✗ Branch 0 not taken. ✓ Branch 1 taken 1739 times.	1739x	if (result == std::noop_coroutine())
132		1739x	svc_.scheduler().notify_reactor();
133		1739x	return result;
134			}
135
136			inline std::coroutine_handle<>
137		444210x	select_tcp_socket::read_some(
138			std::coroutine_handle<> h,
139			capy::executor_ref ex,
140			buffer_param param,
141			std::stop_token token,
142			std::error_code* ec,
143			std::size_t* bytes_out)
144			{
145		444210x	return do_read_some(h, ex, param, token, ec, bytes_out);
146			}
147
148			inline std::coroutine_handle<>
149		445570x	select_tcp_socket::write_some(
150			std::coroutine_handle<> h,
151			capy::executor_ref ex,
152			buffer_param param,
153			std::stop_token token,
154			std::error_code* ec,
155			std::size_t* bytes_out)
156			{
157		445570x	auto result = do_write_some(h, ex, param, token, ec, bytes_out);
158			// Rebuild fd_sets so select() watches for writability
159	2/2 ✓ Branch 0 taken 365413 times. ✓ Branch 1 taken 80157 times.	445570x	if (result == std::noop_coroutine())
160		80157x	svc_.scheduler().notify_reactor();
161		445570x	return result;
162			}
163
164			inline void
165		42x	select_tcp_socket::cancel() noexcept
166			{
167		42x	do_cancel();
168		42x	}
169
170			inline void
171		15753x	select_tcp_socket::close_socket() noexcept
172			{
173		15753x	do_close_socket();
174		15753x	}
175
176			inline std::error_code
177		1761x	select_tcp_service::open_socket(
178			tcp_socket::implementation& impl, int family, int type, int protocol)
179			{
180		1761x	auto* select_impl = static_cast<select_tcp_socket*>(&impl);
181		1761x	select_impl->close_socket();
182
183		1761x	int fd = ::socket(family, type, protocol);
184	1/2 ✓ Branch 0 taken 1761 times. ✗ Branch 1 not taken.	1761x	if (fd < 0)
185		✗	return make_err(errno);
186
187	2/2 ✓ Branch 0 taken 1755 times. ✓ Branch 1 taken 6 times.	1761x	if (family == AF_INET6)
188			{
189		6x	int one = 1;
190		6x	::setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY, &one, sizeof(one));
191		6x	}
192
193		1761x	int flags = ::fcntl(fd, F_GETFL, 0);
194	1/2 ✓ Branch 0 taken 1761 times. ✗ Branch 1 not taken.	1761x	if (flags == -1)
195			{
196		✗	int errn = errno;
197		✗	::close(fd);
198		✗	return make_err(errn);
199			}
200	1/2 ✓ Branch 0 taken 1761 times. ✗ Branch 1 not taken.	1761x	if (::fcntl(fd, F_SETFL, flags \| O_NONBLOCK) == -1)
201			{
202		✗	int errn = errno;
203		✗	::close(fd);
204		✗	return make_err(errn);
205			}
206	1/2 ✓ Branch 0 taken 1761 times. ✗ Branch 1 not taken.	1761x	if (::fcntl(fd, F_SETFD, FD_CLOEXEC) == -1)
207			{
208		✗	int errn = errno;
209		✗	::close(fd);
210		✗	return make_err(errn);
211			}
212
213	1/2 ✗ Branch 0 not taken. ✓ Branch 1 taken 1761 times.	1761x	if (fd >= FD_SETSIZE)
214			{
215		✗	::close(fd);
216		✗	return make_err(EMFILE);
217			}
218
219			#ifdef SO_NOSIGPIPE
220			{
221		1761x	int one = 1;
222		1761x	::setsockopt(fd, SOL_SOCKET, SO_NOSIGPIPE, &one, sizeof(one));
223			}
224			#endif
225
226		1761x	select_impl->fd_ = fd;
227
228		1761x	select_impl->desc_state_.fd = fd;
229			{
230		1761x	std::lock_guard lock(select_impl->desc_state_.mutex);
231		1761x	select_impl->desc_state_.read_op = nullptr;
232		1761x	select_impl->desc_state_.write_op = nullptr;
233		1761x	select_impl->desc_state_.connect_op = nullptr;
234		1761x	}
235		1761x	scheduler().register_descriptor(fd, &select_impl->desc_state_);
236
237		1761x	return {};
238		1761x	}
239
240			inline std::error_code
241		6x	select_tcp_service::bind_socket(
242			tcp_socket::implementation& impl, endpoint ep)
243			{
244		6x	return static_cast<select_tcp_socket*>(&impl)->do_bind(ep);
245			}
246
247			} // namespace boost::corosio::detail
248
249			#endif // BOOST_COROSIO_HAS_SELECT
250
251			#endif // BOOST_COROSIO_NATIVE_DETAIL_SELECT_SELECT_TCP_SERVICE_HPP
252