Home | History | Annotate | Download | only in Fuzzer 
      1 //===- FuzzerDefs.h - Internal header for the Fuzzer ------------*- C++ -* ===//
      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 // Basic definitions.
     10 //===----------------------------------------------------------------------===//
     11 
     12 #ifndef LLVM_FUZZER_DEFS_H
     13 #define LLVM_FUZZER_DEFS_H
     14 
     15 #include <cassert>
     16 #include <cstddef>
     17 #include <cstdint>
     18 #include <cstring>
     19 #include <string>
     20 #include <vector>
     21 
     22 // Platform detection.
     23 #ifdef __linux__
     24 #define LIBFUZZER_APPLE 0
     25 #define LIBFUZZER_LINUX 1
     26 #define LIBFUZZER_WINDOWS 0
     27 #elif __APPLE__
     28 #define LIBFUZZER_APPLE 1
     29 #define LIBFUZZER_LINUX 0
     30 #define LIBFUZZER_WINDOWS 0
     31 #elif _WIN32
     32 #define LIBFUZZER_APPLE 0
     33 #define LIBFUZZER_LINUX 0
     34 #define LIBFUZZER_WINDOWS 1
     35 #else
     36 #error "Support for your platform has not been implemented"
     37 #endif
     38 
     39 #ifndef __has_attribute
     40 #  define __has_attribute(x) 0
     41 #endif
     42 
     43 #define LIBFUZZER_POSIX LIBFUZZER_APPLE || LIBFUZZER_LINUX
     44 
     45 #ifdef __x86_64
     46 #  if __has_attribute(target)
     47 #    define ATTRIBUTE_TARGET_POPCNT __attribute__((target("popcnt")))
     48 #  else
     49 #    define ATTRIBUTE_TARGET_POPCNT
     50 #  endif
     51 #else
     52 #  define ATTRIBUTE_TARGET_POPCNT
     53 #endif
     54 
     55 
     56 #ifdef __clang__  // avoid gcc warning.
     57 #  if __has_attribute(no_sanitize)
     58 #    define ATTRIBUTE_NO_SANITIZE_MEMORY __attribute__((no_sanitize("memory")))
     59 #  else
     60 #    define ATTRIBUTE_NO_SANITIZE_MEMORY
     61 #  endif
     62 #  define ALWAYS_INLINE __attribute__((always_inline))
     63 #else
     64 #  define ATTRIBUTE_NO_SANITIZE_MEMORY
     65 #  define ALWAYS_INLINE
     66 #endif // __clang__
     67 
     68 #define ATTRIBUTE_NO_SANITIZE_ADDRESS __attribute__((no_sanitize_address))
     69 
     70 #if defined(__has_feature)
     71 #  if __has_feature(address_sanitizer)
     72 #    define ATTRIBUTE_NO_SANITIZE_ALL ATTRIBUTE_NO_SANITIZE_ADDRESS
     73 #  elif __has_feature(memory_sanitizer)
     74 #    define ATTRIBUTE_NO_SANITIZE_ALL ATTRIBUTE_NO_SANITIZE_MEMORY
     75 #  else
     76 #    define ATTRIBUTE_NO_SANITIZE_ALL
     77 #  endif
     78 #else
     79 #  define ATTRIBUTE_NO_SANITIZE_ALL
     80 #endif
     81 
     82 #if LIBFUZZER_WINDOWS
     83 #define ATTRIBUTE_INTERFACE __declspec(dllexport)
     84 #else
     85 #define ATTRIBUTE_INTERFACE __attribute__((visibility("default")))
     86 #endif
     87 
     88 namespace fuzzer {
     89 
     90 template <class T> T Min(T a, T b) { return a < b ? a : b; }
     91 template <class T> T Max(T a, T b) { return a > b ? a : b; }
     92 
     93 class Random;
     94 class Dictionary;
     95 class DictionaryEntry;
     96 class MutationDispatcher;
     97 struct FuzzingOptions;
     98 class InputCorpus;
     99 struct InputInfo;
    100 struct ExternalFunctions;
    101 
    102 // Global interface to functions that may or may not be available.
    103 extern ExternalFunctions *EF;
    104 
    105 typedef std::vector<uint8_t> Unit;
    106 typedef std::vector<Unit> UnitVector;
    107 typedef int (*UserCallback)(const uint8_t *Data, size_t Size);
    108 
    109 int FuzzerDriver(int *argc, char ***argv, UserCallback Callback);
    110 
    111 struct ScopedDoingMyOwnMemOrStr {
    112   ScopedDoingMyOwnMemOrStr() { DoingMyOwnMemOrStr++; }
    113   ~ScopedDoingMyOwnMemOrStr() { DoingMyOwnMemOrStr--; }
    114   static int DoingMyOwnMemOrStr;
    115 };
    116 
    117 inline uint8_t  Bswap(uint8_t x)  { return x; }
    118 inline uint16_t Bswap(uint16_t x) { return __builtin_bswap16(x); }
    119 inline uint32_t Bswap(uint32_t x) { return __builtin_bswap32(x); }
    120 inline uint64_t Bswap(uint64_t x) { return __builtin_bswap64(x); }
    121 
    122 uint8_t *ExtraCountersBegin();
    123 uint8_t *ExtraCountersEnd();
    124 void ClearExtraCounters();
    125 
    126 }  // namespace fuzzer
    127 
    128 #endif  // LLVM_FUZZER_DEFS_H
    129