1193323Sed//===- AliasDebugger.cpp - Simple Alias Analysis Use Checker --------------===//
2193323Sed//
3193323Sed//                     The LLVM Compiler Infrastructure
4193323Sed//
5193323Sed// This file is distributed under the University of Illinois Open Source
6193323Sed// License. See LICENSE.TXT for details.
7193323Sed//
8193323Sed//===----------------------------------------------------------------------===//
9193323Sed//
10193323Sed// This simple pass checks alias analysis users to ensure that if they
11193323Sed// create a new value, they do not query AA without informing it of the value.
12193323Sed// It acts as a shim over any other AA pass you want.
13193323Sed//
14193323Sed// Yes keeping track of every value in the program is expensive, but this is
15193323Sed// a debugging pass.
16193323Sed//
17193323Sed//===----------------------------------------------------------------------===//
18193323Sed
19193323Sed#include "llvm/Analysis/Passes.h"
20249423Sdim#include "llvm/Analysis/AliasAnalysis.h"
21249423Sdim#include "llvm/IR/Constants.h"
22249423Sdim#include "llvm/IR/DerivedTypes.h"
23249423Sdim#include "llvm/IR/Instructions.h"
24249423Sdim#include "llvm/IR/Module.h"
25193323Sed#include "llvm/Pass.h"
26193323Sed#include <set>
27193323Sedusing namespace llvm;
28193323Sed
29193323Sednamespace {
30193323Sed
31198892Srdivacky  class AliasDebugger : public ModulePass, public AliasAnalysis {
32193323Sed
33193323Sed    //What we do is simple.  Keep track of every value the AA could
34193323Sed    //know about, and verify that queries are one of those.
35193323Sed    //A query to a value that didn't exist when the AA was created
36193323Sed    //means someone forgot to update the AA when creating new values
37193323Sed
38193323Sed    std::set<const Value*> Vals;
39193323Sed
40193323Sed  public:
41193323Sed    static char ID; // Class identification, replacement for typeinfo
42218893Sdim    AliasDebugger() : ModulePass(ID) {
43218893Sdim      initializeAliasDebuggerPass(*PassRegistry::getPassRegistry());
44218893Sdim    }
45193323Sed
46193323Sed    bool runOnModule(Module &M) {
47193323Sed      InitializeAliasAnalysis(this);                 // set up super class
48193323Sed
49193323Sed      for(Module::global_iterator I = M.global_begin(),
50210299Sed            E = M.global_end(); I != E; ++I) {
51193323Sed        Vals.insert(&*I);
52210299Sed        for (User::const_op_iterator OI = I->op_begin(),
53210299Sed             OE = I->op_end(); OI != OE; ++OI)
54210299Sed          Vals.insert(*OI);
55210299Sed      }
56193323Sed
57193323Sed      for(Module::iterator I = M.begin(),
58193323Sed            E = M.end(); I != E; ++I){
59193323Sed        Vals.insert(&*I);
60193323Sed        if(!I->isDeclaration()) {
61193323Sed          for (Function::arg_iterator AI = I->arg_begin(), AE = I->arg_end();
62193323Sed               AI != AE; ++AI)
63193323Sed            Vals.insert(&*AI);
64193323Sed          for (Function::const_iterator FI = I->begin(), FE = I->end();
65193323Sed               FI != FE; ++FI)
66193323Sed            for (BasicBlock::const_iterator BI = FI->begin(), BE = FI->end();
67210299Sed                 BI != BE; ++BI) {
68193323Sed              Vals.insert(&*BI);
69210299Sed              for (User::const_op_iterator OI = BI->op_begin(),
70210299Sed                   OE = BI->op_end(); OI != OE; ++OI)
71210299Sed                Vals.insert(*OI);
72210299Sed            }
73193323Sed        }
74193323Sed
75193323Sed      }
76193323Sed      return false;
77193323Sed    }
78193323Sed
79193323Sed    virtual void getAnalysisUsage(AnalysisUsage &AU) const {
80193323Sed      AliasAnalysis::getAnalysisUsage(AU);
81193323Sed      AU.setPreservesAll();                         // Does not transform code
82193323Sed    }
83193323Sed
84202878Srdivacky    /// getAdjustedAnalysisPointer - This method is used when a pass implements
85202878Srdivacky    /// an analysis interface through multiple inheritance.  If needed, it
86202878Srdivacky    /// should override this to adjust the this pointer as needed for the
87202878Srdivacky    /// specified pass info.
88212904Sdim    virtual void *getAdjustedAnalysisPointer(AnalysisID PI) {
89212904Sdim      if (PI == &AliasAnalysis::ID)
90202878Srdivacky        return (AliasAnalysis*)this;
91202878Srdivacky      return this;
92202878Srdivacky    }
93202878Srdivacky
94193323Sed    //------------------------------------------------
95193323Sed    // Implement the AliasAnalysis API
96193323Sed    //
97218893Sdim    AliasResult alias(const Location &LocA, const Location &LocB) {
98218893Sdim      assert(Vals.find(LocA.Ptr) != Vals.end() &&
99218893Sdim             "Never seen value in AA before");
100218893Sdim      assert(Vals.find(LocB.Ptr) != Vals.end() &&
101218893Sdim             "Never seen value in AA before");
102218893Sdim      return AliasAnalysis::alias(LocA, LocB);
103193323Sed    }
104193323Sed
105212904Sdim    ModRefResult getModRefInfo(ImmutableCallSite CS,
106218893Sdim                               const Location &Loc) {
107218893Sdim      assert(Vals.find(Loc.Ptr) != Vals.end() && "Never seen value in AA before");
108218893Sdim      return AliasAnalysis::getModRefInfo(CS, Loc);
109193323Sed    }
110193323Sed
111212904Sdim    ModRefResult getModRefInfo(ImmutableCallSite CS1,
112212904Sdim                               ImmutableCallSite CS2) {
113193323Sed      return AliasAnalysis::getModRefInfo(CS1,CS2);
114193323Sed    }
115193323Sed
116218893Sdim    bool pointsToConstantMemory(const Location &Loc, bool OrLocal) {
117218893Sdim      assert(Vals.find(Loc.Ptr) != Vals.end() && "Never seen value in AA before");
118218893Sdim      return AliasAnalysis::pointsToConstantMemory(Loc, OrLocal);
119193323Sed    }
120193323Sed
121193323Sed    virtual void deleteValue(Value *V) {
122193323Sed      assert(Vals.find(V) != Vals.end() && "Never seen value in AA before");
123193323Sed      AliasAnalysis::deleteValue(V);
124193323Sed    }
125193323Sed    virtual void copyValue(Value *From, Value *To) {
126193323Sed      Vals.insert(To);
127193323Sed      AliasAnalysis::copyValue(From, To);
128193323Sed    }
129193323Sed
130193323Sed  };
131193323Sed}
132193323Sed
133193323Sedchar AliasDebugger::ID = 0;
134212904SdimINITIALIZE_AG_PASS(AliasDebugger, AliasAnalysis, "debug-aa",
135218893Sdim                   "AA use debugger", false, true, false)
136193323Sed
137193323SedPass *llvm::createAliasDebugger() { return new AliasDebugger(); }
138193323Sed
139