CustomCrossOverTest.cpp
1.77 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// Simple test for a cutom crossover.
#include <assert.h>
#include <cstddef>
#include <cstdint>
#include <cstdlib>
#include <iostream>
#include <ostream>
#include <random>
#include <string.h>
#include <functional>
static const char *Separator = "-########-";
static const char *Target = "A-########-B";
static volatile int sink;
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
assert(Data);
std::string Str(reinterpret_cast<const char *>(Data), Size);
static const size_t TargetHash = std::hash<std::string>{}(std::string(Target));
size_t StrHash = std::hash<std::string>{}(Str);
// Ensure we have 'A' and 'B' in the corpus.
if (Size == 1 && *Data == 'A')
sink++;
if (Size == 1 && *Data == 'B')
sink--;
if (TargetHash == StrHash) {
std::cout << "BINGO; Found the target, exiting\n" << std::flush;
exit(1);
}
return 0;
}
extern "C" size_t LLVMFuzzerCustomCrossOver(const uint8_t *Data1, size_t Size1,
const uint8_t *Data2, size_t Size2,
uint8_t *Out, size_t MaxOutSize,
unsigned int Seed) {
static size_t Printed;
static size_t SeparatorLen = strlen(Separator);
if (Printed++ < 32)
std::cerr << "In LLVMFuzzerCustomCrossover " << Size1 << " " << Size2 << "\n";
size_t Size = Size1 + Size2 + SeparatorLen;
if (Size > MaxOutSize)
return 0;
memcpy(Out, Data1, Size1);
memcpy(Out + Size1, Separator, SeparatorLen);
memcpy(Out + Size1 + SeparatorLen, Data2, Size2);
return Size;
}