1//===--- MixedTBAATest.cpp - Mixed TBAA unit tests ------------------------===// 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#include "llvm/Analysis/Passes.h" 11#include "llvm/IR/Constants.h" 12#include "llvm/IR/Instructions.h" 13#include "llvm/IR/LLVMContext.h" 14#include "llvm/IR/MDBuilder.h" 15#include "llvm/IR/Module.h" 16#include "llvm/PassManager.h" 17#include "llvm/Support/CommandLine.h" 18#include "gtest/gtest.h" 19 20namespace llvm { 21namespace { 22 23class MixedTBAATest : public testing::Test { 24protected: 25 MixedTBAATest() : M("MixedTBAATest", C), MD(C) {} 26 27 LLVMContext C; 28 Module M; 29 MDBuilder MD; 30 PassManager PM; 31}; 32 33TEST_F(MixedTBAATest, MixedTBAA) { 34 // Setup function. 35 FunctionType *FTy = FunctionType::get(Type::getVoidTy(C), 36 std::vector<Type *>(), false); 37 auto *F = cast<Function>(M.getOrInsertFunction("f", FTy)); 38 auto *BB = BasicBlock::Create(C, "entry", F); 39 auto IntType = Type::getInt32Ty(C); 40 auto PtrType = Type::getInt32PtrTy(C); 41 auto *Value = ConstantInt::get(IntType, 42); 42 auto *Addr = ConstantPointerNull::get(PtrType); 43 44 auto *Store1 = new StoreInst(Value, Addr, BB); 45 auto *Store2 = new StoreInst(Value, Addr, BB); 46 ReturnInst::Create(C, nullptr, BB); 47 48 // New TBAA metadata 49 { 50 auto RootMD = MD.createTBAARoot("Simple C/C++ TBAA"); 51 auto MD1 = MD.createTBAAScalarTypeNode("omnipotent char", RootMD); 52 auto MD2 = MD.createTBAAScalarTypeNode("int", MD1); 53 auto MD3 = MD.createTBAAStructTagNode(MD2, MD2, 0); 54 Store2->setMetadata(LLVMContext::MD_tbaa, MD3); 55 } 56 57 // Old TBAA metadata 58 { 59 auto RootMD = MD.createTBAARoot("Simple C/C++ TBAA"); 60 auto MD1 = MD.createTBAANode("omnipotent char", RootMD); 61 auto MD2 = MD.createTBAANode("int", MD1); 62 Store1->setMetadata(LLVMContext::MD_tbaa, MD2); 63 } 64 65 // Run the TBAA eval pass on a mixture of path-aware and non-path-aware TBAA. 66 // The order of the metadata (path-aware vs non-path-aware) is important, 67 // because the AA eval pass only runs one test per store-pair. 68 const char* args[] = { "MixedTBAATest", "-evaluate-tbaa" }; 69 cl::ParseCommandLineOptions(sizeof(args) / sizeof(const char*), args); 70 PM.add(createTypeBasedAliasAnalysisPass()); 71 PM.add(createAAEvalPass()); 72 PM.run(M); 73} 74 75} // end anonymous namspace 76} // end llvm namespace 77 78