1 //===- FuzzerDefs.h - Internal header for the Fuzzer ------------*- C++ -* ===//
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 // Basic definitions.
9 //===----------------------------------------------------------------------===//
10
11 #ifndef LLVM_FUZZER_DEFS_H
12 #define LLVM_FUZZER_DEFS_H
13
14 #include <cassert>
15 #include <cstddef>
16 #include <cstdint>
17 #include <cstring>
18 #include <memory>
19 #include <set>
20 #include <string>
21 #include <vector>
22
23
24 namespace fuzzer {
25
Min(T a,T b)26 template <class T> T Min(T a, T b) { return a < b ? a : b; }
Max(T a,T b)27 template <class T> T Max(T a, T b) { return a > b ? a : b; }
28
29 class Random;
30 class Dictionary;
31 class DictionaryEntry;
32 class MutationDispatcher;
33 struct FuzzingOptions;
34 class InputCorpus;
35 struct InputInfo;
36 struct ExternalFunctions;
37
38 // Global interface to functions that may or may not be available.
39 extern ExternalFunctions *EF;
40
41 // We are using a custom allocator to give a different symbol name to STL
42 // containers in order to avoid ODR violations.
43 template<typename T>
44 class fuzzer_allocator: public std::allocator<T> {
45 public:
46 fuzzer_allocator() = default;
47
48 template<class U>
fuzzer_allocator(const fuzzer_allocator<U> &)49 fuzzer_allocator(const fuzzer_allocator<U>&) {}
50
51 template<class Other>
52 struct rebind { typedef fuzzer_allocator<Other> other; };
53 };
54
55 template<typename T>
56 using Vector = std::vector<T, fuzzer_allocator<T>>;
57
58 template<typename T>
59 using Set = std::set<T, std::less<T>, fuzzer_allocator<T>>;
60
61 typedef Vector<uint8_t> Unit;
62 typedef Vector<Unit> UnitVector;
63 typedef int (*UserCallback)(const uint8_t *Data, size_t Size);
64
65 int FuzzerDriver(int *argc, char ***argv, UserCallback Callback);
66
67 uint8_t *ExtraCountersBegin();
68 uint8_t *ExtraCountersEnd();
69 void ClearExtraCounters();
70
71 extern bool RunningUserCallback;
72
73 } // namespace fuzzer
74
75 #endif // LLVM_FUZZER_DEFS_H
76