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
+3 -4
View File
@@ -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;