src/corosio/src/tcp_socket.cpp

76.9% Lines (40/52) 83.3% List of functions (10/12)
tcp_socket.cpp
f(x) Functions (12)
Line TLA Hits Source Code
1 //
2 // Copyright (c) 2025 Vinnie Falco (vinnie.falco@gmail.com)
3 // Copyright (c) 2026 Steve Gerbino
4 //
5 // Distributed under the Boost Software License, Version 1.0. (See accompanying
6 // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
7 //
8 // Official repository: https://github.com/cppalliance/corosio
9 //
10
11 #include <boost/corosio/tcp_socket.hpp>
12 #include <boost/corosio/detail/except.hpp>
13 #include <boost/corosio/detail/platform.hpp>
14
15 #if BOOST_COROSIO_HAS_IOCP
16 #include <boost/corosio/native/detail/iocp/win_tcp_acceptor_service.hpp>
17 #else
18 #include <boost/corosio/detail/tcp_service.hpp>
19 #endif
20
21 namespace boost::corosio {
22
23 16235x tcp_socket::~tcp_socket()
24 {
25 16235x close();
26 16235x }
27
28 16049x tcp_socket::tcp_socket(capy::execution_context& ctx)
29 #if BOOST_COROSIO_HAS_IOCP
30 : io_object(create_handle<detail::win_tcp_service>(ctx))
31 #else
32 16049x : io_object(create_handle<detail::tcp_service>(ctx))
33 #endif
34 {
35 16049x }
36
37 void
38 8023x tcp_socket::open(tcp proto)
39 {
40 8023x if (is_open())
41 return;
42 8023x open_for_family(proto.family(), proto.type(), proto.protocol());
43 }
44
45 void
46 8023x tcp_socket::open_for_family(int family, int type, int protocol)
47 {
48 #if BOOST_COROSIO_HAS_IOCP
49 auto& svc = static_cast<detail::win_tcp_service&>(h_.service());
50 auto& wrapper = static_cast<tcp_socket::implementation&>(*h_.get());
51 std::error_code ec = svc.open_socket(
52 *static_cast<detail::win_tcp_socket&>(wrapper).get_internal(), family, type,
53 protocol);
54 #else
55 8023x auto& svc = static_cast<detail::tcp_service&>(h_.service());
56 8023x std::error_code ec = svc.open_socket(
57 8023x static_cast<tcp_socket::implementation&>(*h_.get()), family, type,
58 protocol);
59 #endif
60 8023x if (ec)
61 detail::throw_system_error(ec, "tcp_socket::open");
62 8023x }
63
64 std::error_code
65 14x tcp_socket::bind(endpoint ep)
66 {
67 14x if (!is_open())
68 2x detail::throw_logic_error("bind: socket not open");
69 #if BOOST_COROSIO_HAS_IOCP
70 auto& svc = static_cast<detail::win_tcp_service&>(h_.service());
71 auto& wrapper = static_cast<tcp_socket::implementation&>(*h_.get());
72 return svc.bind_socket(
73 *static_cast<detail::win_tcp_socket&>(wrapper).get_internal(), ep);
74 #else
75 12x auto& svc = static_cast<detail::tcp_service&>(h_.service());
76 12x return svc.bind_socket(
77 24x static_cast<tcp_socket::implementation&>(*h_.get()), ep);
78 #endif
79 }
80
81 void
82 32261x tcp_socket::close()
83 {
84 32261x if (!is_open())
85 16259x return;
86 16002x h_.service().close(h_);
87 }
88
89 void
90 184x tcp_socket::cancel()
91 {
92 184x if (!is_open())
93 return;
94 184x get().cancel();
95 }
96
97 void
98 12x tcp_socket::shutdown(shutdown_type what)
99 {
100 12x if (is_open())
101 {
102 // Best-effort: errors like ENOTCONN are expected and unhelpful
103 6x [[maybe_unused]] auto ec = get().shutdown(what);
104 }
105 12x }
106
107 void
108 tcp_socket::shutdown(shutdown_type what, std::error_code& ec) noexcept
109 {
110 ec = {};
111 if (is_open())
112 ec = get().shutdown(what);
113 }
114
115 native_handle_type
116 tcp_socket::native_handle() const noexcept
117 {
118 if (!is_open())
119 {
120 #if BOOST_COROSIO_HAS_IOCP
121 return static_cast<native_handle_type>(~0ull); // INVALID_SOCKET
122 #else
123 return -1;
124 #endif
125 }
126 return get().native_handle();
127 }
128
129 endpoint
130 58x tcp_socket::local_endpoint() const noexcept
131 {
132 58x if (!is_open())
133 10x return endpoint{};
134 48x return get().local_endpoint();
135 }
136
137 endpoint
138 54x tcp_socket::remote_endpoint() const noexcept
139 {
140 54x if (!is_open())
141 10x return endpoint{};
142 44x return get().remote_endpoint();
143 }
144
145 } // namespace boost::corosio
146