. /include/boost/corosio /test /socket_pair.hpp
71.79% Lines (28/39) 100.00% Functions (3/3)
[ ] prev/next file n p prev/next uncovered line
TLA Baseline Branch
Line Hits Code Line Hits Code
1   // 1   //
2   // Copyright (c) 2025 Vinnie Falco (vinnie.falco@gmail.com) 2   // Copyright (c) 2025 Vinnie Falco (vinnie.falco@gmail.com)
3   // Copyright (c) 2026 Steve Gerbino 3   // Copyright (c) 2026 Steve Gerbino
4   // 4   //
5   // Distributed under the Boost Software License, Version 1.0. (See accompanying 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) 6   // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
7   // 7   //
8   // Official repository: https://github.com/cppalliance/corosio 8   // Official repository: https://github.com/cppalliance/corosio
9   // 9   //
10   10  
11   #ifndef BOOST_COROSIO_TEST_SOCKET_PAIR_HPP 11   #ifndef BOOST_COROSIO_TEST_SOCKET_PAIR_HPP
12   #define BOOST_COROSIO_TEST_SOCKET_PAIR_HPP 12   #define BOOST_COROSIO_TEST_SOCKET_PAIR_HPP
13   13  
14   #include <boost/corosio/io_context.hpp> 14   #include <boost/corosio/io_context.hpp>
15   #include <boost/corosio/tcp_acceptor.hpp> 15   #include <boost/corosio/tcp_acceptor.hpp>
16   #include <boost/corosio/tcp_socket.hpp> 16   #include <boost/corosio/tcp_socket.hpp>
17   #include <boost/corosio/socket_option.hpp> 17   #include <boost/corosio/socket_option.hpp>
18   #include <boost/capy/ex/run_async.hpp> 18   #include <boost/capy/ex/run_async.hpp>
19   #include <boost/capy/task.hpp> 19   #include <boost/capy/task.hpp>
20   20  
21   #include <cstdio> 21   #include <cstdio>
22   #include <stdexcept> 22   #include <stdexcept>
23   #include <system_error> 23   #include <system_error>
24   #include <utility> 24   #include <utility>
25   25  
26   namespace boost::corosio::test { 26   namespace boost::corosio::test {
27   27  
28   /** Create a connected pair of sockets. 28   /** Create a connected pair of sockets.
29   29  
30   Creates two sockets connected via loopback TCP sockets. 30   Creates two sockets connected via loopback TCP sockets.
31   Data written to one socket can be read from the other. 31   Data written to one socket can be read from the other.
32   32  
33   @tparam Socket The socket type (default `tcp_socket`). 33   @tparam Socket The socket type (default `tcp_socket`).
34   @tparam Acceptor The acceptor type (default `tcp_acceptor`). 34   @tparam Acceptor The acceptor type (default `tcp_acceptor`).
35   35  
36   @param ctx The I/O context for the sockets. 36   @param ctx The I/O context for the sockets.
37   37  
38   @return A pair of connected sockets. 38   @return A pair of connected sockets.
39   */ 39   */
40   template< 40   template<
41   class Socket = tcp_socket, 41   class Socket = tcp_socket,
42   class Acceptor = tcp_acceptor, 42   class Acceptor = tcp_acceptor,
43   bool Linger = true> 43   bool Linger = true>
44   std::pair<Socket, Socket> 44   std::pair<Socket, Socket>
HITCBC 45   254 make_socket_pair(io_context& ctx) 45   254 make_socket_pair(io_context& ctx)
46   { 46   {
HITCBC 47   254 auto ex = ctx.get_executor(); 47   254 auto ex = ctx.get_executor();
48   48  
HITCBC 49   254 std::error_code accept_ec; 49   254 std::error_code accept_ec;
HITCBC 50   254 std::error_code connect_ec; 50   254 std::error_code connect_ec;
HITCBC 51   254 bool accept_done = false; 51   254 bool accept_done = false;
HITCBC 52   254 bool connect_done = false; 52   254 bool connect_done = false;
53   53  
HITCBC 54   254 Acceptor acc(ctx); 54   254 Acceptor acc(ctx);
HITCBC 55   254 if (auto open_ec = acc.open()) 55   254 if (auto open_ec = acc.open())
MISUBC 56 - ✗ throw std::runtime_error("socket_pair open failed: " + open_ec.message()); 56 + ✗ throw std::runtime_error(
  57 + "socket_pair open failed: " + open_ec.message());
HITCBC 57   254 acc.set_option(socket_option::reuse_address(true)); 58   254 acc.set_option(socket_option::reuse_address(true));
HITCBC 58   254 if (auto ec = acc.bind(endpoint(ipv4_address::loopback(), 0))) 59   254 if (auto ec = acc.bind(endpoint(ipv4_address::loopback(), 0)))
MISUBC 59   ✗ throw std::runtime_error("socket_pair bind failed: " + ec.message()); 60   ✗ throw std::runtime_error("socket_pair bind failed: " + ec.message());
HITCBC 60   254 if (auto ec = acc.listen()) 61   254 if (auto ec = acc.listen())
MISUBC 61   ✗ throw std::runtime_error("socket_pair listen failed: " + ec.message()); 62   ✗ throw std::runtime_error("socket_pair listen failed: " + ec.message());
HITCBC 62   254 auto port = acc.local_endpoint().port(); 63   254 auto port = acc.local_endpoint().port();
63   64  
HITCBC 64   254 Socket s1(ctx); 65   254 Socket s1(ctx);
HITCBC 65   254 Socket s2(ctx); 66   254 Socket s2(ctx);
HITCBC 66   254 if (auto open_ec = s2.open()) 67   254 if (auto open_ec = s2.open())
MISUBC 67 - ✗ throw std::runtime_error("socket_pair open failed: " + open_ec.message()); 68 + ✗ throw std::runtime_error(
  69 + "socket_pair open failed: " + open_ec.message());
68   70  
HITCBC 69   254 capy::run_async(ex)( 71   254 capy::run_async(ex)(
HITCBC 70   508 [](Acceptor& a, Socket& s, std::error_code& ec_out, 72   508 [](Acceptor& a, Socket& s, std::error_code& ec_out,
71   bool& done_out) -> capy::task<> { 73   bool& done_out) -> capy::task<> {
72   auto [ec] = co_await a.accept(s); 74   auto [ec] = co_await a.accept(s);
73   ec_out = ec; 75   ec_out = ec;
74   done_out = true; 76   done_out = true;
75   }(acc, s1, accept_ec, accept_done)); 77   }(acc, s1, accept_ec, accept_done));
76   78  
HITCBC 77   254 capy::run_async(ex)( 79   254 capy::run_async(ex)(
HITCBC 78   508 [](Socket& s, endpoint ep, std::error_code& ec_out, 80   508 [](Socket& s, endpoint ep, std::error_code& ec_out,
79   bool& done_out) -> capy::task<> { 81   bool& done_out) -> capy::task<> {
80   auto [ec] = co_await s.connect(ep); 82   auto [ec] = co_await s.connect(ep);
81   ec_out = ec; 83   ec_out = ec;
82   done_out = true; 84   done_out = true;
83   }(s2, endpoint(ipv4_address::loopback(), port), connect_ec, 85   }(s2, endpoint(ipv4_address::loopback(), port), connect_ec,
84   connect_done)); 86   connect_done));
85   87  
HITCBC 86   254 ctx.run(); 88   254 ctx.run();
HITCBC 87   254 ctx.restart(); 89   254 ctx.restart();
88   90  
HITCBC 89   254 if (!accept_done || accept_ec) 91   254 if (!accept_done || accept_ec)
90   { 92   {
MISUBC 91   ✗ std::fprintf( 93   ✗ std::fprintf(
92   stderr, "socket_pair: accept failed (done=%d, ec=%s)\n", 94   stderr, "socket_pair: accept failed (done=%d, ec=%s)\n",
93   accept_done, accept_ec.message().c_str()); 95   accept_done, accept_ec.message().c_str());
MISUBC 94   ✗ acc.close(); 96   ✗ acc.close();
MISUBC 95   ✗ throw std::runtime_error("socket_pair accept failed"); 97   ✗ throw std::runtime_error("socket_pair accept failed");
96   } 98   }
97   99  
HITCBC 98   254 if (!connect_done || connect_ec) 100   254 if (!connect_done || connect_ec)
99   { 101   {
MISUBC 100   ✗ std::fprintf( 102   ✗ std::fprintf(
101   stderr, "socket_pair: connect failed (done=%d, ec=%s)\n", 103   stderr, "socket_pair: connect failed (done=%d, ec=%s)\n",
102   connect_done, connect_ec.message().c_str()); 104   connect_done, connect_ec.message().c_str());
MISUBC 103   ✗ acc.close(); 105   ✗ acc.close();
MISUBC 104   ✗ s1.close(); 106   ✗ s1.close();
MISUBC 105   ✗ throw std::runtime_error("socket_pair connect failed"); 107   ✗ throw std::runtime_error("socket_pair connect failed");
106   } 108   }
107   109  
HITCBC 108   254 acc.close(); 110   254 acc.close();
109   111  
110   if constexpr (Linger) 112   if constexpr (Linger)
111   { 113   {
HITCBC 112   93 s1.set_option(socket_option::linger(true, 0)); 114   93 s1.set_option(socket_option::linger(true, 0));
HITCBC 113   93 s2.set_option(socket_option::linger(true, 0)); 115   93 s2.set_option(socket_option::linger(true, 0));
114   } 116   }
115   117  
HITCBC 116   508 return {std::move(s1), std::move(s2)}; 118   508 return {std::move(s1), std::move(s2)};
HITCBC 117   254 } 119   254 }
118   120  
119   } // namespace boost::corosio::test 121   } // namespace boost::corosio::test
120   122  
121   #endif 123   #endif