1 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
2 // See https://llvm.org/LICENSE.txt for license information.
3 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
4
5 // Simple test for a cutom mutator.
6 #include <assert.h>
7 #include <cstddef>
8 #include <cstdint>
9 #include <cstdlib>
10 #include <iostream>
11 #include <ostream>
12
13 #include "FuzzerInterface.h"
14
15 static volatile int Sink;
16
LLVMFuzzerTestOneInput(const uint8_t * Data,size_t Size)17 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
18 assert(Data);
19 if (Size > 0 && Data[0] == 'H') {
20 Sink = 1;
21 if (Size > 1 && Data[1] == 'i') {
22 Sink = 2;
23 if (Size > 2 && Data[2] == '!') {
24 std::cout << "BINGO; Found the target, exiting\n" << std::flush;
25 exit(1);
26 }
27 }
28 }
29 return 0;
30 }
31
LLVMFuzzerCustomMutator(uint8_t * Data,size_t Size,size_t MaxSize,unsigned int Seed)32 extern "C" size_t LLVMFuzzerCustomMutator(uint8_t *Data, size_t Size,
33 size_t MaxSize, unsigned int Seed) {
34 static bool Printed;
35 if (!Printed) {
36 std::cerr << "In LLVMFuzzerCustomMutator\n";
37 Printed = true;
38 }
39 return LLVMFuzzerMutate(Data, Size, MaxSize);
40 }
41