. /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   188 make_socket_pair(io_context& ctx) 45   254 make_socket_pair(io_context& ctx)
46   { 46   {
HITCBC 47   188 auto ex = ctx.get_executor(); 47   254 auto ex = ctx.get_executor();
48   48  
HITCBC 49   188 std::error_code accept_ec; 49   254 std::error_code accept_ec;
HITCBC 50   188 std::error_code connect_ec; 50   254 std::error_code connect_ec;
HITCBC 51   188 bool accept_done = false; 51   254 bool accept_done = false;
HITCBC 52   188 bool connect_done = false; 52   254 bool connect_done = false;
53   53  
HITCBC 54   188 Acceptor acc(ctx); 54   254 Acceptor acc(ctx);
HITCBC 55   188 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("socket_pair open failed: " + open_ec.message());
HITCBC 57   188 acc.set_option(socket_option::reuse_address(true)); 57   254 acc.set_option(socket_option::reuse_address(true));
HITCBC 58   188 if (auto ec = acc.bind(endpoint(ipv4_address::loopback(), 0))) 58   254 if (auto ec = acc.bind(endpoint(ipv4_address::loopback(), 0)))
MISUBC 59   ✗ throw std::runtime_error("socket_pair bind failed: " + ec.message()); 59   ✗ throw std::runtime_error("socket_pair bind failed: " + ec.message());
HITCBC 60   188 if (auto ec = acc.listen()) 60   254 if (auto ec = acc.listen())
MISUBC 61   ✗ throw std::runtime_error("socket_pair listen failed: " + ec.message()); 61   ✗ throw std::runtime_error("socket_pair listen failed: " + ec.message());
HITCBC 62   188 auto port = acc.local_endpoint().port(); 62   254 auto port = acc.local_endpoint().port();
63   63  
HITCBC 64   188 Socket s1(ctx); 64   254 Socket s1(ctx);
HITCBC 65   188 Socket s2(ctx); 65   254 Socket s2(ctx);
HITCBC 66   188 if (auto open_ec = s2.open()) 66   254 if (auto open_ec = s2.open())
MISUBC 67   ✗ throw std::runtime_error("socket_pair open failed: " + open_ec.message()); 67   ✗ throw std::runtime_error("socket_pair open failed: " + open_ec.message());
68   68  
HITCBC 69   188 capy::run_async(ex)( 69   254 capy::run_async(ex)(
HITCBC 70   376 [](Acceptor& a, Socket& s, std::error_code& ec_out, 70   508 [](Acceptor& a, Socket& s, std::error_code& ec_out,
71   bool& done_out) -> capy::task<> { 71   bool& done_out) -> capy::task<> {
72   auto [ec] = co_await a.accept(s); 72   auto [ec] = co_await a.accept(s);
73   ec_out = ec; 73   ec_out = ec;
74   done_out = true; 74   done_out = true;
75   }(acc, s1, accept_ec, accept_done)); 75   }(acc, s1, accept_ec, accept_done));
76   76  
HITCBC 77   188 capy::run_async(ex)( 77   254 capy::run_async(ex)(
HITCBC 78   376 [](Socket& s, endpoint ep, std::error_code& ec_out, 78   508 [](Socket& s, endpoint ep, std::error_code& ec_out,
79   bool& done_out) -> capy::task<> { 79   bool& done_out) -> capy::task<> {
80   auto [ec] = co_await s.connect(ep); 80   auto [ec] = co_await s.connect(ep);
81   ec_out = ec; 81   ec_out = ec;
82   done_out = true; 82   done_out = true;
83   }(s2, endpoint(ipv4_address::loopback(), port), connect_ec, 83   }(s2, endpoint(ipv4_address::loopback(), port), connect_ec,
84   connect_done)); 84   connect_done));
85   85  
HITCBC 86   188 ctx.run(); 86   254 ctx.run();
HITCBC 87   188 ctx.restart(); 87   254 ctx.restart();
88   88  
HITCBC 89   188 if (!accept_done || accept_ec) 89   254 if (!accept_done || accept_ec)
90   { 90   {
MISUBC 91   ✗ std::fprintf( 91   ✗ std::fprintf(
92   stderr, "socket_pair: accept failed (done=%d, ec=%s)\n", 92   stderr, "socket_pair: accept failed (done=%d, ec=%s)\n",
93   accept_done, accept_ec.message().c_str()); 93   accept_done, accept_ec.message().c_str());
MISUBC 94   ✗ acc.close(); 94   ✗ acc.close();
MISUBC 95   ✗ throw std::runtime_error("socket_pair accept failed"); 95   ✗ throw std::runtime_error("socket_pair accept failed");
96   } 96   }
97   97  
HITCBC 98   188 if (!connect_done || connect_ec) 98   254 if (!connect_done || connect_ec)
99   { 99   {
MISUBC 100   ✗ std::fprintf( 100   ✗ std::fprintf(
101   stderr, "socket_pair: connect failed (done=%d, ec=%s)\n", 101   stderr, "socket_pair: connect failed (done=%d, ec=%s)\n",
102   connect_done, connect_ec.message().c_str()); 102   connect_done, connect_ec.message().c_str());
MISUBC 103   ✗ acc.close(); 103   ✗ acc.close();
MISUBC 104   ✗ s1.close(); 104   ✗ s1.close();
MISUBC 105   ✗ throw std::runtime_error("socket_pair connect failed"); 105   ✗ throw std::runtime_error("socket_pair connect failed");
106   } 106   }
107   107  
HITCBC 108   188 acc.close(); 108   254 acc.close();
109   109  
110   if constexpr (Linger) 110   if constexpr (Linger)
111   { 111   {
HITCBC 112   89 s1.set_option(socket_option::linger(true, 0)); 112   93 s1.set_option(socket_option::linger(true, 0));
HITCBC 113   89 s2.set_option(socket_option::linger(true, 0)); 113   93 s2.set_option(socket_option::linger(true, 0));
114   } 114   }
115   115  
HITCBC 116   376 return {std::move(s1), std::move(s2)}; 116   508 return {std::move(s1), std::move(s2)};
HITCBC 117   188 } 117   254 }
118   118  
119   } // namespace boost::corosio::test 119   } // namespace boost::corosio::test
120   120  
121   #endif 121   #endif