forked from OSchip/llvm-project
78 lines
2.5 KiB
C++
78 lines
2.5 KiB
C++
//===--- 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-aa-metadata" };
|
|
cl::ParseCommandLineOptions(sizeof(args) / sizeof(const char*), args);
|
|
PM.add(createTypeBasedAliasAnalysisPass());
|
|
PM.add(createAAEvalPass());
|
|
PM.run(M);
|
|
}
|
|
|
|
} // end anonymous namspace
|
|
} // end llvm namespace
|
|
|