llvm/compiler-rt/test/fuzzer/AFLDriverTest.cpp

// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>

// Dummy functions used to avoid dependency on AFL.
extern "C" void __afl_manual_init() {}

extern "C" int __afl_persistent_loop(unsigned int N) {
  static int Count = N;
  fprintf(stderr, "__afl_persistent_loop called, Count = %d\n", Count);
  return Count--;
}

// This declaration exists to prevent the Darwin linker
// from complaining about this being a missing weak symbol.
extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv) {
  return 0;
}

extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
  puts("STDOUT MESSAGE");
  fflush(stdout);
  fprintf(stderr, "STDERR MESSAGE\n"
                  "LLVMFuzzerTestOneInput called; Size = %zd\n",
          Size);
  if (Size < 4)
    return 0;

  return Data[Size];
}