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