1// This file is distributed under the University of Illinois Open Source
2// License. See LICENSE.TXT for details.
3
4// Simple test for a cutom mutator.
5#include <assert.h>
6#include <cstdint>
7#include <cstdlib>
8#include <cstddef>
9#include <iostream>
10
11#include "FuzzerInterface.h"
12
13static volatile int Sink;
14
15extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
16  assert(Data);
17  if (Size > 0 && Data[0] == 'H') {
18    Sink = 1;
19    if (Size > 1 && Data[1] == 'i') {
20      Sink = 2;
21      if (Size > 2 && Data[2] == '!') {
22        std::cout << "BINGO; Found the target, exiting\n";
23        exit(1);
24      }
25    }
26  }
27  return 0;
28}
29
30extern "C" size_t LLVMFuzzerCustomMutator(uint8_t *Data, size_t Size,
31                                          size_t MaxSize, unsigned int Seed) {
32  static bool Printed;
33  if (!Printed) {
34    std::cerr << "In LLVMFuzzerCustomMutator\n";
35    Printed = true;
36  }
37  return LLVMFuzzerMutate(Data, Size, MaxSize);
38}
39