Home | History | Annotate | Download | only in test
      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 
     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 
     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