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

Make sure to call write while fuzzing

Browse Source
This commit is contained in:
Alex Hultman
2020-01-12 20:19:44 +01:00
parent dde628956b
commit f0574d5fbb
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fuzzing/MockedHelloWorld.cpp
+2 -1
View File
@@ -11,7 +11,7 @@ us_listen_socket_t *listenSocket;
extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv) { extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv) {
app = new uWS::TemplatedApp<false>(uWS::App().get("/*", [](auto *res, auto *req) { app = new uWS::TemplatedApp<false>(uWS::App().get("/*", [](auto *res, auto *req) {
if (req->getHeader("use_write").length()) { if (req->getHeader("write").length()) {
res->writeStatus("200 OK")->writeHeader("write", "true")->write("Hello"); res->writeStatus("200 OK")->writeHeader("write", "true")->write("Hello");
res->write(" world!"); res->write(" world!");
res->end(); res->end();
@@ -29,6 +29,7 @@ extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv) {
}); });
res->onData([res](std::string_view chunk, bool isEnd) { res->onData([res](std::string_view chunk, bool isEnd) {
if (isEnd) { if (isEnd) {
res->write("something ahead");
res->end(chunk); res->end(chunk);
} }
}); });