92 lines
2.7 KiB
C++
92 lines
2.7 KiB
C++
#include "App.h"
|
|
|
|
#include "helpers.h"
|
|
|
|
/* 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);
|
|
|
|
us_listen_socket_t *listenSocket;
|
|
|
|
/* ws->getUserData returns one of these */
|
|
struct PerSocketData {
|
|
int nothing;
|
|
};
|
|
|
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
|
|
|
|
/* First byte determines what compressor to use */
|
|
if (size < 1) {
|
|
return 0;
|
|
}
|
|
|
|
uWS::CompressOptions compressors[] = {
|
|
uWS::DISABLED,
|
|
uWS::SHARED_COMPRESSOR,
|
|
uWS::DEDICATED_COMPRESSOR_3KB,
|
|
uWS::DEDICATED_COMPRESSOR_4KB,
|
|
uWS::DEDICATED_COMPRESSOR_8KB,
|
|
uWS::DEDICATED_COMPRESSOR_16KB,
|
|
uWS::DEDICATED_COMPRESSOR_32KB,
|
|
uWS::DEDICATED_COMPRESSOR_64KB,
|
|
uWS::DEDICATED_COMPRESSOR_128KB,
|
|
uWS::DEDICATED_COMPRESSOR_256KB
|
|
};
|
|
|
|
uWS::CompressOptions compressor = compressors[data[0] % 10];
|
|
data++;
|
|
size--;
|
|
|
|
/* Very simple WebSocket echo server */
|
|
auto app = uWS::TemplatedApp<false>(uWS::App().ws<PerSocketData>("/*", {
|
|
/* Settings */
|
|
.compression = compressor,
|
|
/* We want this to be low so that we can hit it, yet bigger than 256 */
|
|
.maxPayloadLength = 300,
|
|
.idleTimeout = 10,
|
|
/* Handlers */
|
|
.open = [](auto *ws) {
|
|
//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) {
|
|
if (message.length() > 300) {
|
|
/* Inform the sanitizer of the fault */
|
|
fprintf(stderr, "Too long message passed\n");
|
|
free((void *) -1);
|
|
}
|
|
|
|
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 */
|
|
},
|
|
.ping = [](auto *ws) {
|
|
|
|
},
|
|
.pong = [](auto *ws) {
|
|
|
|
},
|
|
.close = [](auto *ws, int code, std::string_view message) {
|
|
|
|
}
|
|
}).listen(9001, [](us_listen_socket_t *listenSocket) {
|
|
::listenSocket = listenSocket;
|
|
})).run();
|
|
|
|
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;
|
|
}
|