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
13 static volatile int Sink;
14
LLVMFuzzerTestOneInput(const uint8_t * Data,size_t Size)15 extern "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
LLVMFuzzerCustomMutator(uint8_t * Data,size_t Size,size_t MaxSize,unsigned int Seed)30 extern "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