compiler-rt/test/fuzzer/InitializeTest.cpp

   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
  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
  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 }