talksik/uWebSockets
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix EpollEchoServer fuzzing

Browse Source
This commit is contained in:
Alex Hultman
2021-02-21 17:11:38 +01:00
parent 44b57e64dc
commit 09aef610e0
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fuzzing/EpollEchoServer.cpp
+6 -7
View File
@@ -45,8 +45,8 @@ void test() {
.idleTimeout = 12, .idleTimeout = 12,
/* Handlers */ /* Handlers */
.open = [](auto *ws) { .open = [](auto *ws) {
/* Subscribe to anything */ /* Subscribe to anything */
ws->subscribe(/*req->getHeader(*/"topic"/*)*/); ws->subscribe(/*req->getHeader(*/"topic"/*)*/);
}, },
.message = [](auto *ws, std::string_view message, uWS::OpCode opCode) { .message = [](auto *ws, std::string_view message, uWS::OpCode opCode) {
if (message.length() && message[0] == 'C') { if (message.length() && message[0] == 'C') {
@@ -58,7 +58,7 @@ void test() {
ws->publish(message, message, opCode, true); ws->publish(message, message, opCode, true);
if (message.length() && message[0] == 'U') { if (message.length() && message[0] == 'U') {
ws->unsubscribe(message); ws->unsubscribe(message);
} }
} }
}, },
@@ -83,8 +83,7 @@ void test() {
/* Handlers */ /* Handlers */
.open = [](auto *ws) { .open = [](auto *ws) {
PerSocketData *psd = (PerSocketData *) ws->getUserData(); ws->getUserData()->valid.reset(new bool{true});
psd->valid.reset(new bool{true});
//if (req->getHeader("close_me").length()) { //if (req->getHeader("close_me").length()) {
// ws->close(); // ws->close();
@@ -115,7 +114,7 @@ void test() {
PerSocketData *psd = (PerSocketData *) ws->getUserData(); PerSocketData *psd = (PerSocketData *) ws->getUserData();
uWS::Loop::get()->defer([ws, valid = psd->valid]() { uWS::Loop::get()->defer([ws, valid = psd->valid]() {
if (valid.get()) { if (*valid.get()) {
/* We haven't been closed */ /* We haven't been closed */
ws->send("Hello!", uWS::TEXT, false); ws->send("Hello!", uWS::TEXT, false);
ws->end(1000); ws->end(1000);
@@ -126,7 +125,7 @@ void test() {
}, },
.close = [](auto *ws, int code, std::string_view message) { .close = [](auto *ws, int code, std::string_view message) {
(*ws->getUserData()->valid.get()) = false;
} }
}).listen(9001, [](us_listen_socket_t *listenSocket) { }).listen(9001, [](us_listen_socket_t *listenSocket) {
listen_socket = listenSocket; listen_socket = listenSocket;