Fixed up very wrong fuzzer to go more for the parser.

This commit is contained in:
David Korczynski 2020-04-19 20:36:38 +01:00
parent 82e052f2ec
commit 676a0afe4c
1 changed files with 34 additions and 34 deletions

View File

@ -2,45 +2,45 @@
#include <string.h>
#include <janet.h>
char *cmd_start = "(def v (unmarshal (slurp ((dyn :";
char *cmd_end = ") 1)) load-image-dict));
int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
char *new_str = (char *)malloc(size + strlen(cmd_start) + strlen(cmd_end) + 1);
if (new_str == NULL) {
return 0;
}
// Copy start of command into the string we will execute
memcpy(new_str, cmd_start, strlen(cmd_start));
// Copy fuzz-data into the exec string
char *datap = new_str + strlen(cmd_start);
memcpy(datap, data, size);
// Copy the end of the command into the exec string
end_data = datap + size;
memcpy(end_data, cmd_end, strlen(cmd_end));
char *null_terminator = end_data + strlen(cmd_end) + 1;
*null_terminator = '\0';
// Remove any parentheses from the fuzz data
for (int i = 0; i < size; i++)
{
if (datap[i] == '(')
datap[i] = (char)(datap[i] + 1);
if (datap[i] == ')')
datap[i] = (char)(datap[i] + 1);
}
// janet logic
/* init Janet */
janet_init();
JanetTable *env = janet_core_env(NULL);
janet_dostring(env, new_str, "main", NULL);
/* fuzz the parser */
JanetParser parser;
janet_parser_init(&parser);
for (int i=0, done = 0; i < size; i++)
{
switch (janet_parser_status(&parser)) {
case JANET_PARSE_DEAD:
case JANET_PARSE_ERROR:
done = 1;
break;
case JANET_PARSE_PENDING:
if (i == size) {
janet_parser_eof(&parser);
} else {
janet_parser_consume(&parser, data[i]);
}
break;
case JANET_PARSE_ROOT:
if (i >= size) {
janet_parser_eof(&parser);
} else {
janet_parser_consume(&parser, data[i]);
}
break;
}
if (done == 1)
break;
}
janet_parser_deinit(&parser);
/* cleanup Janet */
janet_deinit();
free(new_str);
return 0;
}