include/boost/corosio/detail/op_base.hpp

100.0% Lines (47 / 47) 55.2% Branches (74 / 134) 99.7% Functions (639 / 641)

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

op_base.hpp

f(x) Functions (641)

Function Calls Lines Branches Blocks

Line	Branch	Hits	Source Code
1			//
2			// Copyright (c) 2026 Michael Vandeberg
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_OP_BASE_HPP
11			#define BOOST_COROSIO_DETAIL_OP_BASE_HPP
12
13			#include <boost/capy/error.hpp>
14			#include <boost/capy/continuation.hpp>
15			#include <boost/capy/results.hpp>
16			#include <boost/capy/ex/executor_ref.hpp>
17			#include <boost/capy/ex/await_context.hpp>
18
19			#include <coroutine>
20			#include <cstddef>
21			#include <stop_token>
22			#include <system_error>
23
24			namespace boost::corosio::detail {
25
26			/* CRTP base for awaitables that return results<std::size_t>.
27
28			Derived classes must provide:
29
30			std::coroutine_handle<> dispatch(
31			capy::continuation& cont,
32			capy::executor_ref ex) const;
33
34			which forwards to the backend implementation method, passing
35			token_, &ec_, and &bytes_ as the cancellation/output parameters.
36			*/
37			template<class Derived>
38			class bytes_op_base
39			{
40			friend Derived;
41		2776694x	bytes_op_base() = default;
42
43			public:
44			std::stop_token token_;
45			mutable std::error_code ec_;
46		1388347x	mutable std::size_t bytes_ = 0;
47			// Lives in the awaiting frame until resumption, so a completion
48			// posted through an executor never shares the object's state.
49			mutable capy::continuation cont_;
50
51		1388349x	bool await_ready() const noexcept
52			{
53			// A pre-set ec_ means the initiator failed before dispatch
54			// (e.g. a closed object); complete immediately with that error.
55		1388349x	return static_cast<bool>(ec_);
56			}
57
58		1387884x	[[nodiscard]] capy::results<std::size_t> await_resume() const noexcept
59			{
60		1387884x	return {ec_, bytes_};
61			}
62
63		1388312x	auto await_suspend(std::coroutine_handle<> h, capy::await_context const* env)
64			-> std::coroutine_handle<>
65			{
66		1388312x	token_ = env->stop_token;
67			// A pre-stopped token short-circuits before dispatch so no I/O
68			// is performed. A stop landing after dispatch is decoded by the
69			// backend, and a completed transfer is reported verbatim: the
70			// stream contracts forbid discarding the byte count.
71	34/68 ✓ Branch 0 taken 47 times. ✓ Branch 1 taken 1267781 times. ✓ Branch 2 taken 11 times. ✓ Branch 3 taken 119602 times. ✓ Branch 4 taken 10 times. ✓ Branch 5 taken 516 times. ✓ Branch 6 taken 5 times. ✓ Branch 7 taken 124 times. ✓ Branch 8 taken 5 times. ✓ Branch 9 taken 71 times. ✓ Branch 10 taken 4 times. ✓ Branch 11 taken 45 times. ✓ Branch 12 taken 4 times. ✓ Branch 13 taken 30 times. ✓ Branch 14 taken 4 times. ✓ Branch 15 taken 21 times. ✓ Branch 16 taken 1 time. ✓ Branch 17 taken 1 time. ✓ Branch 18 taken 1 time. ✓ Branch 19 taken 1 time. ✗ Branch 20 not taken. ✗ Branch 21 not taken. ✗ Branch 22 not taken. ✗ Branch 23 not taken. ✗ Branch 24 not taken. ✗ Branch 25 not taken. ✗ Branch 26 not taken. ✗ Branch 27 not taken. ✗ Branch 28 not taken. ✗ Branch 29 not taken. ✗ Branch 30 not taken. ✗ Branch 31 not taken. ✗ Branch 32 not taken. ✗ Branch 33 not taken. ✓ Branch 34 taken 2 times. ✓ Branch 35 taken 7 times. ✗ Branch 36 not taken. ✗ Branch 37 not taken. ✓ Branch 38 taken 1 time. ✓ Branch 39 taken 2 times. ✓ Branch 40 taken 1 time. ✓ Branch 41 taken 1 time. ✓ Branch 42 taken 1 time. ✓ Branch 43 taken 2 times. ✓ Branch 44 taken 1 time. ✓ Branch 45 taken 4 times. ✓ Branch 46 taken 1 time. ✓ Branch 47 taken 2 times. ✓ Branch 48 taken 1 time. ✓ Branch 49 taken 2 times. ✗ Branch 50 not taken. ✗ Branch 51 not taken. ✗ Branch 52 not taken. ✗ Branch 53 not taken. ✗ Branch 54 not taken. ✗ Branch 55 not taken. ✗ Branch 56 not taken. ✗ Branch 57 not taken. ✗ Branch 58 not taken. ✗ Branch 59 not taken. ✗ Branch 60 not taken. ✗ Branch 61 not taken. ✗ Branch 62 not taken. ✗ Branch 63 not taken. ✗ Branch 64 not taken. ✗ Branch 65 not taken. ✗ Branch 66 not taken. ✗ Branch 67 not taken.	1388312x	if (token_.stop_requested())
72			{
73		100x	ec_ = capy::error::canceled;
74		100x	return h;
75			}
76		1388212x	cont_.h = h;
77		1388212x	cont_.owner = env->owner;
78		1388212x	return static_cast<Derived const*>(this)->dispatch(
79		1388212x	cont_, env->executor);
80		1388312x	}
81			};
82
83			/* CRTP base for awaitables that return results<Value> for a
84			moved-out result object (e.g. the resolver's result lists).
85
86			Derived classes must provide:
87
88			std::coroutine_handle<> dispatch(
89			capy::continuation& cont,
90			capy::executor_ref ex) const;
91
92			which forwards to the backend implementation method, passing
93			token_, &ec_, and &value_ as the cancellation/output parameters.
94			*/
95			template<class Derived, class Value>
96			class value_op_base
97			{
98			friend Derived;
99		412174x	value_op_base() = default;
100
101			public:
102			std::stop_token token_;
103			mutable std::error_code ec_;
104		206087x	mutable Value value_{};
105			// See bytes_op_base::cont_.
106			mutable capy::continuation cont_;
107
108		206084x	bool await_ready() const noexcept
109			{
110			// A pre-set ec_ means the initiator failed before dispatch;
111			// complete immediately with that error.
112		206084x	return static_cast<bool>(ec_);
113			}
114
115		202857x	[[nodiscard]] capy::results<Value> await_resume() const noexcept
116			{
117		202857x	return {ec_, std::move(value_)};
118			}
119
120		206079x	auto await_suspend(std::coroutine_handle<> h, capy::await_context const* env)
121			-> std::coroutine_handle<>
122			{
123		206079x	token_ = env->stop_token;
124			// A pre-stopped token short-circuits before dispatch; a stop
125			// landing after dispatch leaves the completed result intact.
126	6/6 ✓ Branch 0 taken 2 times. ✓ Branch 1 taken 39 times. ✓ Branch 2 taken 63 times. ✓ Branch 3 taken 205956 times. ✓ Branch 4 taken 1 time. ✓ Branch 5 taken 18 times.	206079x	if (token_.stop_requested())
127			{
128		66x	ec_ = capy::error::canceled;
129		66x	return h;
130			}
131		206013x	cont_.h = h;
132		206013x	cont_.owner = env->owner;
133		206013x	return static_cast<Derived const*>(this)->dispatch(
134		206013x	cont_, env->executor);
135		206079x	}
136			};
137
138			/* CRTP base for awaitables that return results<>.
139
140			Derived classes must provide:
141
142			std::coroutine_handle<> dispatch(
143			capy::continuation& cont,
144			capy::executor_ref ex) const;
145
146			which forwards to the backend implementation method, passing
147			token_ and &ec_ as the cancellation/output parameters.
148			*/
149			template<class Derived>
150			class void_op_base
151			{
152			friend Derived;
153		31349x	void_op_base() = default;
154
155			public:
156			std::stop_token token_;
157			mutable std::error_code ec_;
158			// See bytes_op_base::cont_.
159			mutable capy::continuation cont_;
160
161		31352x	bool await_ready() const noexcept
162			{
163			// A pre-set ec_ means the initiator failed before dispatch
164			// (e.g. auto-open); complete immediately with that error.
165		31352x	return static_cast<bool>(ec_);
166			}
167
168		15936x	[[nodiscard]] capy::results<> await_resume() const noexcept
169			{
170		15936x	return {ec_};
171			}
172
173		31324x	auto await_suspend(std::coroutine_handle<> h, capy::await_context const* env)
174			-> std::coroutine_handle<>
175			{
176		31324x	token_ = env->stop_token;
177			// A pre-stopped token short-circuits before dispatch; a stop
178			// landing after dispatch leaves the completed result intact.
179	34/60 ✓ Branch 0 taken 12 times. ✓ Branch 1 taken 15345 times. ✓ Branch 2 taken 9 times. ✓ Branch 3 taken 14615 times. ✓ Branch 4 taken 1011 times. ✓ Branch 5 taken 175 times. ✓ Branch 6 taken 7 times. ✓ Branch 7 taken 63 times. ✓ Branch 8 taken 3 times. ✓ Branch 9 taken 34 times. ✓ Branch 10 taken 2 times. ✓ Branch 11 taken 6 times. ✓ Branch 12 taken 2 times. ✓ Branch 13 taken 3 times. ✓ Branch 14 taken 1 time. ✓ Branch 15 taken 3 times. ✓ Branch 16 taken 1 time. ✓ Branch 17 taken 5 times. ✓ Branch 18 taken 1 time. ✓ Branch 19 taken 2 times. ✓ Branch 20 taken 1 time. ✓ Branch 21 taken 2 times. ✓ Branch 22 taken 1 time. ✓ Branch 23 taken 2 times. ✗ Branch 24 not taken. ✗ Branch 25 not taken. ✗ Branch 26 not taken. ✗ Branch 27 not taken. ✗ Branch 28 not taken. ✗ Branch 29 not taken. ✗ Branch 30 not taken. ✗ Branch 31 not taken. ✗ Branch 32 not taken. ✗ Branch 33 not taken. ✓ Branch 34 taken 1 time. ✓ Branch 35 taken 4 times. ✓ Branch 36 taken 1 time. ✓ Branch 37 taken 2 times. ✓ Branch 38 taken 1 time. ✓ Branch 39 taken 2 times. ✓ Branch 40 taken 1 time. ✓ Branch 41 taken 4 times. ✓ Branch 42 taken 1 time. ✓ Branch 43 taken 1 time. ✗ Branch 44 not taken. ✗ Branch 45 not taken. ✗ Branch 46 not taken. ✗ Branch 47 not taken. ✗ Branch 48 not taken. ✗ Branch 49 not taken. ✗ Branch 50 not taken. ✗ Branch 51 not taken. ✗ Branch 52 not taken. ✗ Branch 53 not taken. ✗ Branch 54 not taken. ✗ Branch 55 not taken. ✗ Branch 56 not taken. ✗ Branch 57 not taken. ✗ Branch 58 not taken. ✗ Branch 59 not taken.	31324x	if (token_.stop_requested())
180			{
181		1056x	ec_ = capy::error::canceled;
182		1056x	return h;
183			}
184		30268x	cont_.h = h;
185		30268x	cont_.owner = env->owner;
186		30268x	return static_cast<Derived const*>(this)->dispatch(
187		30268x	cont_, env->executor);
188		31324x	}
189			};
190
191			} // namespace boost::corosio::detail
192
193			#endif // BOOST_COROSIO_DETAIL_OP_BASE_HPP
194