Try deterministically fuzzed MockedEchoServer

This commit is contained in:
Alex Hultman
2020-08-07 16:30:42 +02:00
parent 4ad07a4f93
commit 0d5c0bd2ef
+10 -16
View File
@@ -5,18 +5,17 @@
/* This function pushes data to the uSockets mock */ /* This function pushes data to the uSockets mock */
extern "C" void us_loop_read_mocked_data(struct us_loop *loop, char *data, unsigned int size); extern "C" void us_loop_read_mocked_data(struct us_loop *loop, char *data, unsigned int size);
uWS::TemplatedApp<false> *app;
us_listen_socket_t *listenSocket; us_listen_socket_t *listenSocket;
extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv) { /* ws->getUserData returns one of these */
struct PerSocketData {
int nothing;
};
/* ws->getUserData returns one of these */ extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
struct PerSocketData {
int nothing;
};
/* Very simple WebSocket echo server */ /* Very simple WebSocket echo server */
app = new uWS::TemplatedApp<false>(uWS::App().ws<PerSocketData>("/*", { auto app = uWS::TemplatedApp<false>(uWS::App().ws<PerSocketData>("/*", {
/* Settings */ /* Settings */
.compression = uWS::SHARED_COMPRESSOR, .compression = uWS::SHARED_COMPRESSOR,
/* We want this to be low so that we can hit it, yet bigger than 256 */ /* We want this to be low so that we can hit it, yet bigger than 256 */
@@ -58,18 +57,13 @@ extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv) {
} }
}).listen(9001, [](us_listen_socket_t *listenSocket) { }).listen(9001, [](us_listen_socket_t *listenSocket) {
if (listenSocket) {
std::cout << "Listening on port " << 9001 << std::endl;
::listenSocket = listenSocket; ::listenSocket = listenSocket;
} })).run();
}));
return 0;
}
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
us_loop_read_mocked_data((struct us_loop *) uWS::Loop::get(), (char *) makePadded(data, size), size); us_loop_read_mocked_data((struct us_loop *) uWS::Loop::get(), (char *) makePadded(data, size), size);
us_listen_socket_close(0, listenSocket);
uWS::Loop::get()->free();
return 0; return 0;
} }