xref: /llvm-project/compiler-rt/test/fuzzer/InitializeTest.cpp (revision 2946cd701067404b99c39fb29dc9c74bd7193eb3)
1 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
2 // See https://llvm.org/LICENSE.txt for license information.
3 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
4 
5 // Make sure LLVMFuzzerInitialize is called.
6 #include <assert.h>
7 #include <stddef.h>
8 #include <stdint.h>
9 #include <stdio.h>
10 #include <stdlib.h>
11 #include <string.h>
12 
13 static char *argv0 = NULL;
14 
LLVMFuzzerInitialize(int * argc,char *** argv)15 extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv) {
16   assert(*argc > 0);
17   argv0 = **argv;
18   fprintf(stderr, "LLVMFuzzerInitialize: %s\n", argv0);
19   return 0;
20 }
21 
LLVMFuzzerTestOneInput(const uint8_t * Data,size_t Size)22 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
23   assert(argv0);
24   if (argv0 && Size >= 4 && !memcmp(Data, "fuzz", 4)) {
25     fprintf(stderr, "BINGO %s\n", argv0);
26     exit(1);
27   }
28   return 0;
29 }
30