12946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
22946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
32946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
410ab2aceSGeorge Karpenkov
5*a1e7e401SKazuaki Ishizaki // Simple test for a custom mutator.
610ab2aceSGeorge Karpenkov #include <assert.h>
710ab2aceSGeorge Karpenkov #include <cstddef>
810ab2aceSGeorge Karpenkov #include <cstdint>
910ab2aceSGeorge Karpenkov #include <cstdlib>
1010ab2aceSGeorge Karpenkov #include <iostream>
1110ab2aceSGeorge Karpenkov #include <ostream>
1210ab2aceSGeorge Karpenkov
1310ab2aceSGeorge Karpenkov #include "FuzzerInterface.h"
1410ab2aceSGeorge Karpenkov
1510ab2aceSGeorge Karpenkov static volatile int Sink;
1610ab2aceSGeorge Karpenkov
LLVMFuzzerTestOneInput(const uint8_t * Data,size_t Size)1710ab2aceSGeorge Karpenkov extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
1810ab2aceSGeorge Karpenkov assert(Data);
1910ab2aceSGeorge Karpenkov if (Size > 0 && Data[0] == 'H') {
2010ab2aceSGeorge Karpenkov Sink = 1;
2110ab2aceSGeorge Karpenkov if (Size > 1 && Data[1] == 'i') {
2210ab2aceSGeorge Karpenkov Sink = 2;
2310ab2aceSGeorge Karpenkov if (Size > 2 && Data[2] == '!') {
2410ab2aceSGeorge Karpenkov std::cout << "BINGO; Found the target, exiting\n" << std::flush;
2510ab2aceSGeorge Karpenkov exit(1);
2610ab2aceSGeorge Karpenkov }
2710ab2aceSGeorge Karpenkov }
2810ab2aceSGeorge Karpenkov }
2910ab2aceSGeorge Karpenkov return 0;
3010ab2aceSGeorge Karpenkov }
3110ab2aceSGeorge Karpenkov
LLVMFuzzerCustomMutator(uint8_t * Data,size_t Size,size_t MaxSize,unsigned int Seed)3210ab2aceSGeorge Karpenkov extern "C" size_t LLVMFuzzerCustomMutator(uint8_t *Data, size_t Size,
3310ab2aceSGeorge Karpenkov size_t MaxSize, unsigned int Seed) {
3410ab2aceSGeorge Karpenkov static bool Printed;
3510ab2aceSGeorge Karpenkov if (!Printed) {
3610ab2aceSGeorge Karpenkov std::cerr << "In LLVMFuzzerCustomMutator\n";
3710ab2aceSGeorge Karpenkov Printed = true;
3810ab2aceSGeorge Karpenkov }
3910ab2aceSGeorge Karpenkov return LLVMFuzzerMutate(Data, Size, MaxSize);
4010ab2aceSGeorge Karpenkov }
41