ProvenanceAnalysisEvaluator.cpp revision 37ed9c199ca639565f6ce88105f9e39e898d82d0 
1//===- ProvenanceAnalysisEvaluator.cpp - ObjC ARC Optimization ------------===//
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 "ProvenanceAnalysis.h"
11#include "llvm/Pass.h"
12#include "llvm/ADT/SetVector.h"
13#include "llvm/Analysis/AliasAnalysis.h"
14#include "llvm/Analysis/Passes.h"
15#include "llvm/IR/InstIterator.h"
16#include "llvm/IR/Function.h"
17#include "llvm/Support/raw_ostream.h"
18
19using namespace llvm;
20using namespace llvm::objcarc;
21
22namespace {
23class PAEval : public FunctionPass {
24
25public:
26  static char ID;
27  PAEval();
28  void getAnalysisUsage(AnalysisUsage &AU) const override;
29  bool runOnFunction(Function &F) override;
30};
31}
32
33char PAEval::ID = 0;
34PAEval::PAEval() : FunctionPass(ID) {}
35
36void PAEval::getAnalysisUsage(AnalysisUsage &AU) const {
37  AU.addRequired<AliasAnalysis>();
38}
39
40static StringRef getName(Value *V) {
41  StringRef Name = V->getName();
42  if (Name.startswith("\1"))
43    return Name.substr(1);
44  return Name;
45}
46
47static void insertIfNamed(SetVector<Value *> &Values, Value *V) {
48  if (!V->hasName())
49    return;
50  Values.insert(V);
51}
52
53bool PAEval::runOnFunction(Function &F) {
54  SetVector<Value *> Values;
55
56  for (auto &Arg : F.args())
57    insertIfNamed(Values, &Arg);
58
59  for (auto I = inst_begin(F), E = inst_end(F); I != E; ++I) {
60    insertIfNamed(Values, &*I);
61
62    for (auto &Op : I->operands())
63    insertIfNamed(Values, Op);
64  }
65
66  ProvenanceAnalysis PA;
67  PA.setAA(&getAnalysis<AliasAnalysis>());
68
69  for (Value *V1 : Values) {
70    StringRef NameV1 = getName(V1);
71    for (Value *V2 : Values) {
72      StringRef NameV2 = getName(V2);
73      if (NameV1 >= NameV2)
74        continue;
75      errs() << NameV1 << " and " << NameV2;
76      if (PA.related(V1, V2))
77        errs() << " are related.\n";
78      else
79        errs() << " are not related.\n";
80    }
81  }
82
83  return false;
84}
85
86FunctionPass *llvm::createPAEvalPass() { return new PAEval(); }
87
88INITIALIZE_PASS_BEGIN(PAEval, "pa-eval",
89                      "Evaluate ProvenanceAnalysis on all pairs", false, true)
90INITIALIZE_AG_DEPENDENCY(AliasAnalysis)
91INITIALIZE_PASS_END(PAEval, "pa-eval",
92                    "Evaluate ProvenanceAnalysis on all pairs", false, true)
93