| 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 custom 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 | |
| 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(status: 1); |
| 26 | } |
| 27 | } |
| 28 | } |
| 29 | return 0; |
| 30 | } |
| 31 | |
| 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 |