1 //===- ProvenanceAnalysisEvaluator.cpp - ObjC ARC Optimization ------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "ProvenanceAnalysis.h" 10 #include "llvm/ADT/SetVector.h" 11 #include "llvm/Analysis/AliasAnalysis.h" 12 #include "llvm/Analysis/Passes.h" 13 #include "llvm/IR/Function.h" 14 #include "llvm/IR/InstIterator.h" 15 #include "llvm/IR/Module.h" 16 #include "llvm/Pass.h" 17 #include "llvm/Support/raw_ostream.h" 18 19 using namespace llvm; 20 using namespace llvm::objcarc; 21 22 namespace { 23 class PAEval : public FunctionPass { 24 25 public: 26 static char ID; 27 PAEval(); 28 void getAnalysisUsage(AnalysisUsage &AU) const override; 29 bool runOnFunction(Function &F) override; 30 }; 31 } 32 33 char PAEval::ID = 0; 34 PAEval::PAEval() : FunctionPass(ID) {} 35 36 void PAEval::getAnalysisUsage(AnalysisUsage &AU) const { 37 AU.addRequired<AAResultsWrapperPass>(); 38 } 39 40 static StringRef getName(Value *V) { 41 StringRef Name = V->getName(); 42 if (Name.startswith("\1")) 43 return Name.substr(1); 44 return Name; 45 } 46 47 static void insertIfNamed(SetVector<Value *> &Values, Value *V) { 48 if (!V->hasName()) 49 return; 50 Values.insert(V); 51 } 52 53 bool 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<AAResultsWrapperPass>().getAAResults()); 68 const DataLayout &DL = F.getParent()->getDataLayout(); 69 70 for (Value *V1 : Values) { 71 StringRef NameV1 = getName(V1); 72 for (Value *V2 : Values) { 73 StringRef NameV2 = getName(V2); 74 if (NameV1 >= NameV2) 75 continue; 76 errs() << NameV1 << " and " << NameV2; 77 if (PA.related(V1, V2, DL)) 78 errs() << " are related.\n"; 79 else 80 errs() << " are not related.\n"; 81 } 82 } 83 84 return false; 85 } 86 87 FunctionPass *llvm::createPAEvalPass() { return new PAEval(); } 88 89 INITIALIZE_PASS_BEGIN(PAEval, "pa-eval", 90 "Evaluate ProvenanceAnalysis on all pairs", false, true) 91 INITIALIZE_PASS_DEPENDENCY(AAResultsWrapperPass) 92 INITIALIZE_PASS_END(PAEval, "pa-eval", 93 "Evaluate ProvenanceAnalysis on all pairs", false, true) 94