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
|
//===--- MixedTBAATest.cpp - Mixed TBAA unit tests ------------------------===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#include "llvm/Analysis/Passes.h"
#include "llvm/IR/Constants.h"
#include "llvm/IR/Instructions.h"
#include "llvm/IR/LLVMContext.h"
#include "llvm/IR/MDBuilder.h"
#include "llvm/IR/Module.h"
#include "llvm/PassManager.h"
#include "llvm/Support/CommandLine.h"
#include "gtest/gtest.h"
namespace llvm {
namespace {
class MixedTBAATest : public testing::Test {
protected:
MixedTBAATest() : M("MixedTBAATest", C), MD(C) {}
LLVMContext C;
Module M;
MDBuilder MD;
PassManager PM;
};
TEST_F(MixedTBAATest, MixedTBAA) {
// Setup function.
FunctionType *FTy = FunctionType::get(Type::getVoidTy(C),
std::vector<Type *>(), false);
auto *F = cast<Function>(M.getOrInsertFunction("f", FTy));
auto *BB = BasicBlock::Create(C, "entry", F);
auto IntType = Type::getInt32Ty(C);
auto PtrType = Type::getInt32PtrTy(C);
auto *Value = ConstantInt::get(IntType, 42);
auto *Addr = ConstantPointerNull::get(PtrType);
auto *Store1 = new StoreInst(Value, Addr, BB);
auto *Store2 = new StoreInst(Value, Addr, BB);
ReturnInst::Create(C, nullptr, BB);
// New TBAA metadata
{
auto RootMD = MD.createTBAARoot("Simple C/C++ TBAA");
auto MD1 = MD.createTBAAScalarTypeNode("omnipotent char", RootMD);
auto MD2 = MD.createTBAAScalarTypeNode("int", MD1);
auto MD3 = MD.createTBAAStructTagNode(MD2, MD2, 0);
Store2->setMetadata(LLVMContext::MD_tbaa, MD3);
}
// Old TBAA metadata
{
auto RootMD = MD.createTBAARoot("Simple C/C++ TBAA");
auto MD1 = MD.createTBAANode("omnipotent char", RootMD);
auto MD2 = MD.createTBAANode("int", MD1);
Store1->setMetadata(LLVMContext::MD_tbaa, MD2);
}
// Run the TBAA eval pass on a mixture of path-aware and non-path-aware TBAA.
// The order of the metadata (path-aware vs non-path-aware) is important,
// because the AA eval pass only runs one test per store-pair.
const char* args[] = { "MixedTBAATest", "-evaluate-tbaa" };
cl::ParseCommandLineOptions(sizeof(args) / sizeof(const char*), args);
PM.add(createTypeBasedAliasAnalysisPass());
PM.add(createAAEvalPass());
PM.run(M);
}
} // end anonymous namspace
} // end llvm namespace
|