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

35 lines
1011 B
C++

// 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];
}