summaryrefslogtreecommitdiff
path: root/test/fuzzer/AFLDriverTest.cpp
blob: 84b5f9f6ba26c7925b73d9d1e306c14b8ea8714e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
// 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];
}