xref: /freebsd/contrib/llvm-project/compiler-rt/lib/fuzzer/FuzzerCrossOver.cpp (revision 7fdf597e96a02165cfe22ff357b857d5fa15ed8a)
1 //===- FuzzerCrossOver.cpp - Cross over two test inputs -------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 // Cross over test inputs.
9 //===----------------------------------------------------------------------===//
10 
11 #include "FuzzerDefs.h"
12 #include "FuzzerMutate.h"
13 #include "FuzzerRandom.h"
14 #include <cstring>
15 
16 namespace fuzzer {
17 
18 // Cross Data1 and Data2, store the result (up to MaxOutSize bytes) in Out.
19 size_t MutationDispatcher::CrossOver(const uint8_t *Data1, size_t Size1,
20                                      const uint8_t *Data2, size_t Size2,
21                                      uint8_t *Out, size_t MaxOutSize) {
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
52