lib/Fuzzer/FuzzerCrossOver.cpp

   1 //===- FuzzerCrossOver.cpp - Cross over two test inputs -------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 // Cross over test inputs.
  10 //===----------------------------------------------------------------------===//
  11
  12 #include <cstring>
  13
  14 #include "FuzzerInternal.h"
  15
  16 namespace fuzzer {
  17
  18 // Cross Data1 and Data2, store the result (up to MaxOutSize bytes) in Out.
  19 size_t CrossOver(const uint8_t *Data1, size_t Size1,
  20                  const uint8_t *Data2, size_t Size2,
  21                  uint8_t *Out, size_t MaxOutSize, FuzzerRandomBase &Rand) {
  22   assert(Size1 || Size2);
  23   MaxOutSize = Rand(MaxOutSize) + 1;
  24   size_t OutPos = 0;
  25   size_t Pos1 = 0;
  26   size_t Pos2 = 0;
  27   size_t *InPos = &Pos1;
  28   size_t InSize = Size1;
  29   const uint8_t *Data = Data1;
  30   bool CurrentlyUsingFirstData = true;
  31   while (OutPos < MaxOutSize && (Pos1 < Size1 || Pos2 < Size2)) {
  32     // Merge a part of Data into Out.
  33     size_t OutSizeLeft = MaxOutSize - OutPos;
  34     if (*InPos < InSize) {
  35       size_t InSizeLeft = InSize - *InPos;
  36       size_t MaxExtraSize = std::min(OutSizeLeft, InSizeLeft);
  37       size_t ExtraSize = Rand(MaxExtraSize) + 1;
  38       memcpy(Out + OutPos, Data + *InPos, ExtraSize);
  39       OutPos += ExtraSize;
  40       (*InPos) += ExtraSize;
  41     }
  42     // Use the other input data on the next iteration.
  43     InPos  = CurrentlyUsingFirstData ? &Pos2 : &Pos1;
  44     InSize = CurrentlyUsingFirstData ? Size2 : Size1;
  45     Data   = CurrentlyUsingFirstData ? Data2 : Data1;
  46     CurrentlyUsingFirstData = !CurrentlyUsingFirstData;
  47   }
  48   return OutPos;
  49 }
  50
  51 }  // namespace fuzzer