compiler-rt/lib/fuzzer/FuzzerDefs.h

   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
  26 template <class T> T Min(T a, T b) { return a < b ? a : 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>
  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