-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathwss_examples.cpp
141 lines (114 loc) · 5.97 KB
/
wss_examples.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
#include "client_wss.hpp"
#include "server_wss.hpp"
using namespace std;
using WssServer = SimpleWeb::SocketServer<SimpleWeb::WSS>;
using WssClient = SimpleWeb::SocketClient<SimpleWeb::WSS>;
int main() {
// WebSocket Secure (WSS)-server at port 8080 using 1 thread
WssServer server("server.crt", "server.key");
server.config.port = 8080;
// Example 1: echo WebSocket Secure endpoint
// Added debug messages for example use of the callbacks
// Test with the following JavaScript:
// var wss=new WebSocket("wss://localhost:8080/echo");
// wss.onmessage=function(evt){console.log(evt.data);};
// wss.send("test");
auto &echo = server.endpoint["^/echo/?$"];
echo.on_message = [](shared_ptr<WssServer::Connection> connection, shared_ptr<WssServer::InMessage> in_message) {
auto out_message = in_message->string();
cout << "Server: Message received: \"" << out_message << "\" from " << connection.get() << endl;
cout << "Server: Sending message \"" << out_message << "\" to " << connection.get() << endl;
// connection->send is an asynchronous function
connection->send(out_message, [](const SimpleWeb::error_code &ec) {
if(ec) {
cout << "Server: Error sending message. " <<
// See http://www.boost.org/doc/libs/1_55_0/doc/html/boost_asio/reference.html, Error Codes for error code meanings
"Error: " << ec << ", error message: " << ec.message() << endl;
}
});
// Alternatively use streams:
// auto out_message = make_shared<WssServer::OutMessage>();
// *out_message << in_message->string();
// connection->send(out_message);
};
echo.on_open = [](shared_ptr<WssServer::Connection> connection) {
cout << "Server: Opened connection " << connection.get() << endl;
};
// See RFC 6455 7.4.1. for status codes
echo.on_close = [](shared_ptr<WssServer::Connection> connection, int status, const string & /*reason*/) {
cout << "Server: Closed connection " << connection.get() << " with status code " << status << endl;
};
// See http://www.boost.org/doc/libs/1_55_0/doc/html/boost_asio/reference.html, Error Codes for error code meanings
echo.on_error = [](shared_ptr<WssServer::Connection> connection, const SimpleWeb::error_code &ec) {
cout << "Server: Error in connection " << connection.get() << ". "
<< "Error: " << ec << ", error message: " << ec.message() << endl;
};
// Example 2: Echo thrice
// Demonstrating queuing of messages by sending a received message three times back to the client.
// Concurrent send operations are automatically queued by the library.
// Test with the following JavaScript:
// var wss=new WebSocket("wss://localhost:8080/echo_thrice");
// wss.onmessage=function(evt){console.log(evt.data);};
// wss.send("test");
auto &echo_thrice = server.endpoint["^/echo_thrice/?$"];
echo_thrice.on_message = [](shared_ptr<WssServer::Connection> connection, shared_ptr<WssServer::InMessage> in_message) {
auto out_message = make_shared<string>(in_message->string());
connection->send(*out_message, [connection, out_message](const SimpleWeb::error_code &ec) {
if(!ec)
connection->send(*out_message); // Sent after the first send operation is finished
});
connection->send(*out_message); // Most likely queued. Sent after the first send operation is finished.
};
// Example 3: Echo to all WebSocket Secure endpoints
// Sending received messages to all connected clients
// Test with the following JavaScript on more than one browser windows:
// var wss=new WebSocket("wss://localhost:8080/echo_all");
// wss.onmessage=function(evt){console.log(evt.data);};
// wss.send("test");
auto &echo_all = server.endpoint["^/echo_all/?$"];
echo_all.on_message = [&server](shared_ptr<WssServer::Connection> /*connection*/, shared_ptr<WssServer::InMessage> in_message) {
auto out_message = in_message->string();
// echo_all.get_connections() can also be used to solely receive connections on this endpoint
for(auto &a_connection : server.get_connections())
a_connection->send(out_message);
};
thread server_thread([&server]() {
// Start WSS-server
server.start();
});
// Wait for server to start so that the client can connect
this_thread::sleep_for(chrono::seconds(1));
// Example 4: Client communication with server
// Second Client() parameter set to false: no certificate verification
// Possible output:
// Server: Opened connection 0x7fcf21600380
// Client: Opened connection
// Client: Sending message: "Hello"
// Server: Message received: "Hello" from 0x7fcf21600380
// Server: Sending message "Hello" to 0x7fcf21600380
// Client: Message received: "Hello"
// Client: Sending close connection
// Server: Closed connection 0x7fcf21600380 with status code 1000
// Client: Closed connection with status code 1000
WssClient client("localhost:8080/echo", false);
client.on_message = [](shared_ptr<WssClient::Connection> connection, shared_ptr<WssClient::InMessage> in_message) {
cout << "Client: Message received: \"" << in_message->string() << "\"" << endl;
cout << "Client: Sending close connection" << endl;
connection->send_close(1000);
};
client.on_open = [](shared_ptr<WssClient::Connection> connection) {
cout << "Client: Opened connection" << endl;
string out_message("Hello");
cout << "Client: Sending message: \"" << out_message << "\"" << endl;
connection->send(out_message);
};
client.on_close = [](shared_ptr<WssClient::Connection> /*connection*/, int status, const string & /*reason*/) {
cout << "Client: Closed connection with status code " << status << endl;
};
// See http://www.boost.org/doc/libs/1_55_0/doc/html/boost_asio/reference.html, Error Codes for error code meanings
client.on_error = [](shared_ptr<WssClient::Connection> /*connection*/, const SimpleWeb::error_code &ec) {
cout << "Client: Error: " << ec << ", error message: " << ec.message() << endl;
};
client.start();
server_thread.join();
}