1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
/*========================== begin_copyright_notice ============================
Copyright (C) 2024 Intel Corporation
SPDX-License-Identifier: MIT
============================= end_copyright_notice ===========================*/
#pragma once
#include "AdaptorCommon/ImplicitArgs.hpp"
#include "Compiler/Optimizer/OpenCLPasses/KernelArgs/KernelArgs.hpp"
#include "Compiler/MetaDataUtilsWrapper.h"
#include "common/LLVMWarningsPush.hpp"
#include "llvmWrapper/IR/Module.h"
#include <llvm/Pass.h>
#include <llvm/IR/InstVisitor.h>
#include <llvm/IR/IRBuilder.h>
#include "common/LLVMWarningsPop.hpp"
namespace IGC {
class BufferBoundsChecking : public llvm::ModulePass, public llvm::InstVisitor<BufferBoundsChecking> {
public:
static char ID;
BufferBoundsChecking();
~BufferBoundsChecking() = default;
virtual void getAnalysisUsage(llvm::AnalysisUsage &analysisUsage) const override {
analysisUsage.addRequired<MetaDataUtilsWrapper>();
analysisUsage.addRequired<CodeGenContextWrapper>();
}
virtual llvm::StringRef getPassName() const override { return "BufferBoundsChecking"; }
virtual bool runOnModule(llvm::Module &M) override;
void visitLoadInst(llvm::LoadInst &load);
void visitStoreInst(llvm::StoreInst &store);
private:
struct AccessInfo {
llvm::StringRef filename;
int line;
int column;
llvm::StringRef bufferName;
llvm::Value *bufferAddress;
llvm::Value *bufferOffsetInBytes;
uint32_t implicitArgBufferSizeIndex;
llvm::Value *elementSizeInBytes;
};
bool modified = false;
IGCMD::MetaDataUtils *metadataUtils = nullptr;
ModuleMetaData *moduleMetadata = nullptr;
ImplicitArgs *implicitArgs = nullptr;
KernelArgs *kernelArgs = nullptr;
llvm::SmallVector<llvm::Instruction *, 4> loadsAndStoresToCheck;
llvm::DenseMap<llvm::StringRef, llvm::GlobalVariable *> stringsCache;
llvm::DICompileUnit *compileUnit = nullptr;
llvm::Function *bufferSizePlaceholderFunction = nullptr;
llvm::Value *localId0;
llvm::Value *localId1;
llvm::Value *localId2;
llvm::Value *globalId0;
llvm::Value *globalId1;
llvm::Value *globalId2;
std::tuple<llvm::Value *, llvm::Value *, llvm::Value *> getLocalIds(llvm::Function &function);
std::tuple<llvm::Value *, llvm::Value *, llvm::Value *> getGlobalIds(llvm::Function &function);
void handleLoadStore(llvm::Instruction *instruction);
bool argumentQualifiesForChecking(const llvm::Argument *argument);
llvm::Value *createBoundsCheckingCondition(const AccessInfo &accessInfo, llvm::Instruction *insertBefore);
llvm::Value *createLoadStoreReplacement(llvm::Instruction *instruction, llvm::Instruction *insertBefore);
void createAssertCall(const AccessInfo &accessInfo, llvm::Instruction *insertBefore);
llvm::SmallVector<llvm::Value *, 4> createAssertArgs(const AccessInfo &accessInfo, llvm::Instruction *insertBefore);
llvm::GlobalVariable *getOrCreateGlobalConstantString(llvm::Module *M, llvm::StringRef format);
void createBoundsCheckingCode(llvm::Instruction *instruction, const AccessInfo &accessInfo);
const IGC::KernelArg *getKernelArg(llvm::Value *value);
const IGC::KernelArg *getKernelArgFromPtr(const llvm::PointerType &pointerType, llvm::Value *value);
llvm::Argument *getBufferSizeArg(llvm::Function *function, uint32_t n);
AccessInfo getAccessInfo(llvm::Instruction *instruction, llvm::Value *value);
llvm::Value *createBufferSizePlaceholder(uint32_t implicitArgBufferSizeIndex, llvm::Instruction *insertBefore);
};
} // namespace IGC
|
