Fix EpollEchoServer fuzzing

This commit is contained in:
Alex Hultman
2021-02-21 17:11:38 +01:00
parent 44b57e64dc
commit 09aef610e0
+3 -4
View File
@@ -83,8 +83,7 @@ void test() {
/* Handlers */
.open = [](auto *ws) {
PerSocketData *psd = (PerSocketData *) ws->getUserData();
psd->valid.reset(new bool{true});
ws->getUserData()->valid.reset(new bool{true});
//if (req->getHeader("close_me").length()) {
// ws->close();
@@ -115,7 +114,7 @@ void test() {
PerSocketData *psd = (PerSocketData *) ws->getUserData();
uWS::Loop::get()->defer([ws, valid = psd->valid]() {
if (valid.get()) {
if (*valid.get()) {
/* We haven't been closed */
ws->send("Hello!", uWS::TEXT, false);
ws->end(1000);
@@ -126,7 +125,7 @@ void test() {
},
.close = [](auto *ws, int code, std::string_view message) {
(*ws->getUserData()->valid.get()) = false;
}
}).listen(9001, [](us_listen_socket_t *listenSocket) {
listen_socket = listenSocket;