1 //===- SanitizerStats.h - Sanitizer statistics gathering -------*- 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 // 10 // Declares functions and data structures for sanitizer statistics gathering. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #ifndef LLVM_TRANSFORMS_UTILS_SANITIZERSTATS_H 15 #define LLVM_TRANSFORMS_UTILS_SANITIZERSTATS_H 16 17 #include "llvm/IR/IRBuilder.h" 18 19 namespace llvm { 20 21 // Number of bits in data that are used for the sanitizer kind. Needs to match 22 // __sanitizer::kKindBits in compiler-rt/lib/stats/stats.h 23 enum { kSanitizerStatKindBits = 3 }; 24 25 enum SanitizerStatKind { 26 SanStat_CFI_VCall, 27 SanStat_CFI_NVCall, 28 SanStat_CFI_DerivedCast, 29 SanStat_CFI_UnrelatedCast, 30 SanStat_CFI_ICall, 31 }; 32 33 struct SanitizerStatReport { 34 SanitizerStatReport(Module *M); 35 36 /// Generates code into B that increments a location-specific counter tagged 37 /// with the given sanitizer kind SK. 38 void create(IRBuilder<> &B, SanitizerStatKind SK); 39 40 /// Finalize module stats array and add global constructor to register it. 41 void finish(); 42 43 private: 44 Module *M; 45 GlobalVariable *ModuleStatsGV; 46 ArrayType *StatTy; 47 StructType *EmptyModuleStatsTy; 48 49 std::vector<Constant *> Inits; 50 ArrayType *makeModuleStatsArrayTy(); 51 StructType *makeModuleStatsTy(); 52 }; 53 54 } 55 56 #endif 57
