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