From 09aef610e0dbc1de7297fda3a873a6d84498d4f7 Mon Sep 17 00:00:00 2001
From: Alex Hultman <alexhultman@gmail.com>
Date: Sun, 21 Feb 2021 17:11:38 +0100
Subject: [PATCH] Fix EpollEchoServer fuzzing

---
 fuzzing/EpollEchoServer.cpp | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/fuzzing/EpollEchoServer.cpp b/fuzzing/EpollEchoServer.cpp
index 1388baa..ba97741 100644
--- a/fuzzing/EpollEchoServer.cpp
+++ b/fuzzing/EpollEchoServer.cpp
@@ -45,8 +45,8 @@ void test() {
             .idleTimeout = 12,
             /* Handlers */
             .open = [](auto *ws) {
-                    /* Subscribe to anything */
-                    ws->subscribe(/*req->getHeader(*/"topic"/*)*/);
+                /* Subscribe to anything */
+                ws->subscribe(/*req->getHeader(*/"topic"/*)*/);
             },
             .message = [](auto *ws, std::string_view message, uWS::OpCode opCode) {
                 if (message.length() && message[0] == 'C') {
@@ -58,7 +58,7 @@ void test() {
                     ws->publish(message, message, opCode, true);
 
                     if (message.length() && message[0] == 'U') {
-                    ws->unsubscribe(message);
+                        ws->unsubscribe(message);
                     }
                 }
             },
@@ -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;