libfuzzer driver nits

This commit is contained in:
vanhauser-thc
2023-04-13 15:32:06 +02:00
parent f756734ad2
commit 2adf5aac0f
2 changed files with 19 additions and 0 deletions

View File

@ -54,4 +54,5 @@
"__sanitizer_cov_trace_pc_guard";
"__sanitizer_cov_trace_pc_guard_init";
"__sanitizer_cov_trace_switch";
"LLVMFuzzerTestOneInput";
};

View File

@ -33,6 +33,11 @@ $ afl-fuzz -i in -o out -- ./test_fuzzer
*/
#ifdef __cplusplus
extern "C" {
#endif
#include <assert.h>
#include <errno.h>
#include <stdarg.h>
@ -260,6 +265,13 @@ static int ExecuteFilesOnyByOne(int argc, char **argv,
__attribute__((weak)) int main(int argc, char **argv) {
if (!LLVMFuzzerTestOneInput) {
fprintf(stderr, "Error: function LLVMFuzzerTestOneInput() not found!\n");
abort();
}
if (argc < 2 || strncmp(argv[1], "-h", 2) == 0)
printf(
"============================== INFO ================================\n"
@ -409,3 +421,9 @@ __attribute__((weak)) int LLVMFuzzerRunDriver(
}
#ifdef __cplusplus
}
#endif