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 <cstddef>
7 #include <cstdint>
8 #include <cstdlib>
9 #include <iostream>
10 #include <ostream>
11 
12 #include "FuzzerInterface.h"
13 
14 static volatile int Sink;
15 
16 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
17   assert(Data);
18   if (Size > 0 && Data[0] == 'H') {
19     Sink = 1;
20     if (Size > 1 && Data[1] == 'i') {
21       Sink = 2;
22       if (Size > 2 && Data[2] == '!') {
23         std::cout << "BINGO; Found the target, exiting\n" << std::flush;
24         exit(1);
25       }
26     }
27   }
28   return 0;
29 }
30 
31 extern "C" size_t LLVMFuzzerCustomMutator(uint8_t *Data, size_t Size,
32                                           size_t MaxSize, unsigned int Seed) {
33   static bool Printed;
34   if (!Printed) {
35     std::cerr << "In LLVMFuzzerCustomMutator\n";
36     Printed = true;
37   }
38   return LLVMFuzzerMutate(Data, Size, MaxSize);
39 }
40