OSSIA
Open Scenario System for Interactive Application
Loading...
Searching...
No Matches
tcp_socket.hpp
1#pragma once
2#include <ossia/network/context.hpp>
3#include <ossia/network/sockets/configuration.hpp>
4
5#include <boost/asio/io_context.hpp>
6#include <boost/asio/ip/tcp.hpp>
7#include <boost/asio/ip/udp.hpp>
8#include <boost/asio/local/datagram_protocol.hpp>
9#include <boost/asio/placeholders.hpp>
10#include <boost/asio/strand.hpp>
11#include <boost/asio/write.hpp>
12
13#include <nano_signal_slot.hpp>
14
15namespace ossia::net
16{
17class tcp_listener
18{
19public:
20 using proto = boost::asio::ip::tcp;
21 using socket = typename proto::socket;
22
23 tcp_listener() = delete;
24 tcp_listener(const tcp_listener&) = delete;
25 tcp_listener& operator=(const tcp_listener&) = delete;
26 tcp_listener(tcp_listener&&) = default;
27 tcp_listener& operator=(tcp_listener&&) = default;
28 explicit tcp_listener(proto::socket sock)
29 : m_socket{std::move(sock)}
30 {
31 }
32
33 void close()
34 {
35 m_socket.shutdown(boost::asio::ip::tcp::socket::shutdown_both);
36 m_socket.close();
37 }
38
39 void write(const boost::asio::const_buffer& buf)
40 {
41 boost::system::error_code ec;
42 boost::asio::write(m_socket, buf, ec);
43 }
44
45 void on_close() { }
46
47 void on_fail() { }
48
49 proto::socket m_socket;
50};
51
52class tcp_server
53{
54public:
55 using proto = boost::asio::ip::tcp;
56 using socket = typename proto::socket;
57 using listener = tcp_listener;
58
59 tcp_server(const inbound_socket_configuration& conf, boost::asio::io_context& ctx)
60 : m_context{ctx}
61 , m_acceptor{
62 boost::asio::make_strand(ctx),
63 proto::endpoint{boost::asio::ip::make_address(conf.bind), conf.port}}
64 {
65 }
66
67 tcp_server(
68 const inbound_socket_configuration& conf, ossia::net::network_context_ptr ctx)
69 : tcp_server{conf, ctx->context}
70 {
71 }
72
73 boost::asio::io_context& m_context;
74 proto::acceptor m_acceptor;
75};
76
77class tcp_client
78{
79public:
80 using proto = boost::asio::ip::tcp;
81 using socket = typename proto::socket;
82
83 tcp_client(const outbound_socket_configuration& conf, boost::asio::io_context& ctx)
84 : m_context{ctx}
85 , m_endpoint{boost::asio::ip::make_address(conf.host), conf.port}
86 , m_socket{boost::asio::make_strand(ctx)}
87 {
88 }
89
90 void connect()
91 {
92 boost::system::error_code ec;
93 m_socket.set_option(boost::asio::ip::tcp::no_delay{true}, ec);
94 m_socket.set_option(boost::asio::socket_base::reuse_address{true}, ec);
95
96 m_socket.async_connect(
97 m_endpoint, [this](const boost::system::error_code& ec, auto&&...) {
98 if(m_socket.is_open() && !ec)
99 {
100 m_connected = true;
101 on_open();
102 }
103 else
104 {
105 m_connected = false;
106 puts(ec.message().c_str());
107 on_fail();
108 }
109 });
110 }
111
112 bool connected() const { return m_connected; }
113
114 void close()
115 {
116 boost::asio::post(m_context, [this] {
117 m_socket.shutdown(boost::asio::ip::tcp::socket::shutdown_both);
118 m_socket.close();
119 on_close();
120 });
121 }
122
123 void write(const char* data, std::size_t sz)
124 {
125 boost::system::error_code ec;
126 boost::asio::write(m_socket, boost::asio::const_buffer(data, sz), ec);
127 }
128
129 Nano::Signal<void()> on_open;
130 Nano::Signal<void()> on_close;
131 Nano::Signal<void()> on_fail;
132
133 boost::asio::io_context& m_context;
134 proto::endpoint m_endpoint;
135 proto::socket m_socket;
136 bool m_connected{false};
137};
138}