diff options
Diffstat (limited to 'lib/Fuzzer/test/InitializeTest.cpp')
-rw-r--r-- | lib/Fuzzer/test/InitializeTest.cpp | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/lib/Fuzzer/test/InitializeTest.cpp b/lib/Fuzzer/test/InitializeTest.cpp new file mode 100644 index 0000000000000..d40ff2f793653 --- /dev/null +++ b/lib/Fuzzer/test/InitializeTest.cpp @@ -0,0 +1,26 @@ +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. + +// Make sure LLVMFuzzerInitialize is called. +#include <assert.h> +#include <stddef.h> +#include <stdint.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +static char *argv0; + +extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv) { + assert(argc > 0); + argv0 = **argv; + return 0; +} + +extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) { + if (strncmp(reinterpret_cast<const char*>(Data), argv0, Size)) { + fprintf(stderr, "BINGO\n"); + exit(1); + } + return 0; +} |