1 //===- FuzzerDFSan.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 // DFSan interface.
10 //===----------------------------------------------------------------------===//
11 
12 #ifndef LLVM_FUZZER_DFSAN_H
13 #define LLVM_FUZZER_DFSAN_H
14 
15 #define LLVM_FUZZER_SUPPORTS_DFSAN 0
16 #if defined(__has_include)
17 # if __has_include(<sanitizer/dfsan_interface.h>)
18 #  if defined (__linux__)
19 #   undef LLVM_FUZZER_SUPPORTS_DFSAN
20 #   define LLVM_FUZZER_SUPPORTS_DFSAN 1
21 #   include <sanitizer/dfsan_interface.h>
22 #  endif  // __linux__
23 # endif
24 #endif  // defined(__has_include)
25 
26 #if LLVM_FUZZER_SUPPORTS_DFSAN
27 
28 extern "C" {
29 __attribute__((weak))
30 dfsan_label dfsan_create_label(const char *desc, void *userdata);
31 __attribute__((weak))
32 void dfsan_set_label(dfsan_label label, void *addr, size_t size);
33 __attribute__((weak))
34 void dfsan_add_label(dfsan_label label, void *addr, size_t size);
35 __attribute__((weak))
36 const struct dfsan_label_info *dfsan_get_label_info(dfsan_label label);
37 __attribute__((weak))
38 dfsan_label dfsan_read_label(const void *addr, size_t size);
39 }  // extern "C"
40 
41 namespace fuzzer {
42 static bool ReallyHaveDFSan() {
43   return &dfsan_create_label != nullptr;
44 }
45 }  // namespace fuzzer
46 #else
47 // When compiling with a compiler which does not support dfsan,
48 // this code is still expected to build (but not necessary work).
49 typedef unsigned short dfsan_label;
50 struct dfsan_label_info {
51   dfsan_label l1, l2;
52   const char *desc;
53   void *userdata;
54 };
55 namespace fuzzer {
56 static bool ReallyHaveDFSan() { return false; }
57 }  // namespace fuzzer
58 
59 #endif
60 
61 #endif // LLVM_FUZZER_DFSAN_H
62