LLVM 20.0.0git
SanitizerStats.cpp
Go to the documentation of this file.
1//===- SanitizerStats.cpp - Sanitizer statistics gathering ----------------===//
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//
9// Implements code generation for sanitizer statistics gathering.
10//
11//===----------------------------------------------------------------------===//
12
14#include "llvm/IR/Constants.h"
17#include "llvm/IR/IRBuilder.h"
18#include "llvm/IR/Module.h"
20
21using namespace llvm;
22
24 StatTy = ArrayType::get(PointerType::getUnqual(M->getContext()), 2);
25 EmptyModuleStatsTy = makeModuleStatsTy();
26
27 ModuleStatsGV = new GlobalVariable(*M, EmptyModuleStatsTy, false,
29}
30
31ArrayType *SanitizerStatReport::makeModuleStatsArrayTy() {
32 return ArrayType::get(StatTy, Inits.size());
33}
34
35StructType *SanitizerStatReport::makeModuleStatsTy() {
36 return StructType::get(M->getContext(),
37 {PointerType::getUnqual(M->getContext()),
38 Type::getInt32Ty(M->getContext()),
39 makeModuleStatsArrayTy()});
40}
41
43 Function *F = B.GetInsertBlock()->getParent();
44 Module *M = F->getParent();
45 PointerType *PtrTy = B.getPtrTy();
46 IntegerType *IntPtrTy = B.getIntPtrTy(M->getDataLayout());
47 ArrayType *StatTy = ArrayType::get(PtrTy, 2);
48
49 Inits.push_back(ConstantArray::get(
50 StatTy,
53 ConstantInt::get(IntPtrTy, uint64_t(SK) << (IntPtrTy->getBitWidth() -
55 PtrTy)}));
56
57 FunctionType *StatReportTy = FunctionType::get(B.getVoidTy(), PtrTy, false);
58 FunctionCallee StatReport =
59 M->getOrInsertFunction("__sanitizer_stat_report", StatReportTy);
60
61 auto InitAddr = ConstantExpr::getGetElementPtr(
62 EmptyModuleStatsTy, ModuleStatsGV,
64 ConstantInt::get(IntPtrTy, 0), ConstantInt::get(B.getInt32Ty(), 2),
65 ConstantInt::get(IntPtrTy, Inits.size() - 1),
66 });
67 B.CreateCall(StatReport, InitAddr);
68}
69
71 if (Inits.empty()) {
72 ModuleStatsGV->eraseFromParent();
73 return;
74 }
75
78 Type *VoidTy = Type::getVoidTy(M->getContext());
79
80 // Create a new ModuleStatsGV to replace the old one. We can't just set the
81 // old one's initializer because its type is different.
82 auto NewModuleStatsGV = new GlobalVariable(
83 *M, makeModuleStatsTy(), false, GlobalValue::InternalLinkage,
85 {Constant::getNullValue(Int8PtrTy),
86 ConstantInt::get(Int32Ty, Inits.size()),
87 ConstantArray::get(makeModuleStatsArrayTy(), Inits)}));
88 ModuleStatsGV->replaceAllUsesWith(NewModuleStatsGV);
89 ModuleStatsGV->eraseFromParent();
90
91 // Create a global constructor to register NewModuleStatsGV.
92 auto F = Function::Create(FunctionType::get(VoidTy, false),
94 auto BB = BasicBlock::Create(M->getContext(), "", F);
95 IRBuilder<> B(BB);
96
97 FunctionType *StatInitTy = FunctionType::get(VoidTy, Int8PtrTy, false);
98 FunctionCallee StatInit =
99 M->getOrInsertFunction("__sanitizer_stat_init", StatInitTy);
100
101 B.CreateCall(StatInit, NewModuleStatsGV);
102 B.CreateRetVoid();
103
104 appendToGlobalCtors(*M, F, 0);
105}
static GCRegistry::Add< OcamlGC > B("ocaml", "ocaml 3.10-compatible GC")
This file contains the declarations for the subclasses of Constant, which represent the different fla...
#define F(x, y, z)
Definition: MD5.cpp:55
Module.h This file contains the declarations for the Module class.
ArrayRef - Represent a constant reference to an array (0 or more elements consecutively in memory),...
Definition: ArrayRef.h:41
Class to represent array types.
Definition: DerivedTypes.h:371
static ArrayType * get(Type *ElementType, uint64_t NumElements)
This static method is the primary way to construct an ArrayType.
Definition: Type.cpp:635
static BasicBlock * Create(LLVMContext &Context, const Twine &Name="", Function *Parent=nullptr, BasicBlock *InsertBefore=nullptr)
Creates a new BasicBlock.
Definition: BasicBlock.h:212
static Constant * get(ArrayType *T, ArrayRef< Constant * > V)
Definition: Constants.cpp:1292
static Constant * getIntToPtr(Constant *C, Type *Ty, bool OnlyIfReduced=false)
Definition: Constants.cpp:2281
static Constant * getGetElementPtr(Type *Ty, Constant *C, ArrayRef< Constant * > IdxList, GEPNoWrapFlags NW=GEPNoWrapFlags::none(), std::optional< ConstantRange > InRange=std::nullopt, Type *OnlyIfReducedTy=nullptr)
Getelementptr form.
Definition: Constants.h:1253
static Constant * getAnon(ArrayRef< Constant * > V, bool Packed=false)
Return an anonymous struct that has the specified elements.
Definition: Constants.h:477
static Constant * getNullValue(Type *Ty)
Constructor to create a '0' constant of arbitrary type.
Definition: Constants.cpp:370
A handy container for a FunctionType+Callee-pointer pair, which can be passed around as a single enti...
Definition: DerivedTypes.h:168
Class to represent function types.
Definition: DerivedTypes.h:103
static FunctionType * get(Type *Result, ArrayRef< Type * > Params, bool isVarArg)
This static method is the primary way of constructing a FunctionType.
static Function * Create(FunctionType *Ty, LinkageTypes Linkage, unsigned AddrSpace, const Twine &N="", Module *M=nullptr)
Definition: Function.h:172
@ InternalLinkage
Rename collisions when linking (static functions).
Definition: GlobalValue.h:59
void eraseFromParent()
eraseFromParent - This method unlinks 'this' from the containing module and deletes it.
Definition: Globals.cpp:481
This provides a uniform API for creating instructions and inserting them into a basic block: either a...
Definition: IRBuilder.h:2674
Class to represent integer types.
Definition: DerivedTypes.h:40
unsigned getBitWidth() const
Get the number of bits in this IntegerType.
Definition: DerivedTypes.h:72
A Module instance is used to store all the information related to an LLVM module.
Definition: Module.h:65
LLVMContext & getContext() const
Get the global data context.
Definition: Module.h:299
FunctionCallee getOrInsertFunction(StringRef Name, FunctionType *T, AttributeList AttributeList)
Look up the specified function in the module symbol table.
Definition: Module.cpp:169
Class to represent pointers.
Definition: DerivedTypes.h:646
static PointerType * getUnqual(Type *ElementType)
This constructs a pointer to an object of the specified type in the default address space (address sp...
Definition: DerivedTypes.h:662
Class to represent struct types.
Definition: DerivedTypes.h:216
static StructType * get(LLVMContext &Context, ArrayRef< Type * > Elements, bool isPacked=false)
This static method is the primary way to create a literal StructType.
Definition: Type.cpp:361
The instances of the Type class are immutable: once they are created, they are never changed.
Definition: Type.h:45
static Type * getVoidTy(LLVMContext &C)
static IntegerType * getInt32Ty(LLVMContext &C)
void replaceAllUsesWith(Value *V)
Change all uses of this to point to a new Value.
Definition: Value.cpp:534
This is an optimization pass for GlobalISel generic memory operations.
Definition: AddressRanges.h:18
SanitizerStatKind
void appendToGlobalCtors(Module &M, Function *F, int Priority, Constant *Data=nullptr)
Append F to the list of global ctors of module M with the given Priority.
Definition: ModuleUtils.cpp:74
@ kSanitizerStatKindBits
void finish()
Finalize module stats array and add global constructor to register it.
void create(IRBuilder<> &B, SanitizerStatKind SK)
Generates code into B that increments a location-specific counter tagged with the given sanitizer kin...