Alex Hultman
98 lines
2.5 KiB
C++
98 lines
2.5 KiB
C++
/* This is a fuzz test of the http parser */
|
|
|
|
#define WIN32_EXPORT
|
|
|
|
#include "helpers.h"
|
|
|
|
/* We test the websocket parser */
|
|
#include "../src/HttpParser.h"
|
|
|
|
/* And the router */
|
|
#include "../src/HttpRouter.h"
|
|
|
|
struct StaticData {
|
|
|
|
struct RouterData {
|
|
|
|
};
|
|
|
|
uWS::HttpRouter<RouterData> router;
|
|
|
|
StaticData() {
|
|
|
|
router.add("get", "/:hello/:hi", [](RouterData &user, std::pair<int, std::string_view *> params) mutable {
|
|
|
|
/* This route did handle it */
|
|
return true;
|
|
});
|
|
|
|
router.add("post", "/:hello/:hi/*", [](RouterData &user, std::pair<int, std::string_view *> params) mutable {
|
|
|
|
/* This route did handle it */
|
|
return true;
|
|
});
|
|
|
|
router.add("get", "/*", [](RouterData &user, std::pair<int, std::string_view *> params) mutable {
|
|
|
|
/* This route did not handle it */
|
|
return false;
|
|
});
|
|
|
|
router.add("get", "/hi", [](RouterData &user, std::pair<int, std::string_view *> params) mutable {
|
|
|
|
/* This route did handle it */
|
|
return true;
|
|
});
|
|
}
|
|
} staticData;
|
|
|
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
|
|
/* Create parser */
|
|
uWS::HttpParser httpParser;
|
|
/* User data */
|
|
void *user = (void *) 13;
|
|
|
|
/* Iterate the padded fuzz as chunks */
|
|
makeChunked(makePadded(data, size), size, [&httpParser, user](const uint8_t *data, size_t size) {
|
|
/* We need at least 1 byte post padding */
|
|
if (size) {
|
|
size--;
|
|
}
|
|
|
|
/* Parse it */
|
|
httpParser.consumePostPadded((char *) data, size, user, [](void *s, uWS::HttpRequest *httpRequest) -> void * {
|
|
|
|
readBytes(httpRequest->getHeader(httpRequest->getUrl()));
|
|
readBytes(httpRequest->getMethod());
|
|
readBytes(httpRequest->getQuery());
|
|
|
|
/* Route the method and URL in two passes */
|
|
StaticData::RouterData routerData = {};
|
|
if (!staticData.router.route(httpRequest->getMethod(), httpRequest->getUrl(), routerData)) {
|
|
/* It was not handled */
|
|
return nullptr;
|
|
}
|
|
|
|
for (auto p : *httpRequest) {
|
|
|
|
}
|
|
|
|
/* Return ok */
|
|
return s;
|
|
|
|
}, [](void *user, std::string_view data, bool fin) -> void * {
|
|
|
|
/* Return ok */
|
|
return user;
|
|
|
|
}, [](void *user) {
|
|
|
|
/* Return break */
|
|
return nullptr;
|
|
});
|
|
});
|
|
|
|
return 0;
|
|
}
|
|
|