Trigger even more code while fuzzing

2019-06-21 13:27:13 +02:00
parent c0d8b383cc
commit 441a1d33ea
1 changed files with 12 additions and 2 deletions
@@ -23,10 +23,20 @@ extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv) {
        .idleTimeout = 10,
        /* Handlers */
        .open = [](auto *ws, auto *req) {
-
+            if (req->getHeader("close_me").length()) {
+                ws->close();
+            } else if (req->getHeader("end_me").length()) {
+                ws->end(1006);
+            }
        },
        .message = [](auto *ws, std::string_view message, uWS::OpCode opCode) {
-            ws->send(message, opCode);
+            if (message.length() && message[0] == 'C') {
+                ws->close();
+            } else if (message.length() && message[0] == 'E') {
+                ws->end(1006);
+            } else {
+                ws->send(message, opCode, true);
+            }
        },
        .drain = [](auto *ws) {
            /* Check getBufferedAmount here */