1 //===- ObjCARC.h - ObjC ARC Optimization --------------*- C++ -*-----------===// 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 /// \file 9 /// This file defines common definitions/declarations used by the ObjC ARC 10 /// Optimizer. ARC stands for Automatic Reference Counting and is a system for 11 /// managing reference counts for objects in Objective C. 12 /// 13 /// WARNING: This file knows about certain library functions. It recognizes them 14 /// by name, and hardwires knowledge of their semantics. 15 /// 16 /// WARNING: This file knows about how certain Objective-C library functions are 17 /// used. Naive LLVM IR transformations which would otherwise be 18 /// behavior-preserving may break these assumptions. 19 /// 20 //===----------------------------------------------------------------------===// 21 22 #ifndef LLVM_LIB_TRANSFORMS_OBJCARC_OBJCARC_H 23 #define LLVM_LIB_TRANSFORMS_OBJCARC_OBJCARC_H 24 25 #include "llvm/Analysis/ObjCARCAnalysisUtils.h" 26 #include "llvm/Analysis/ObjCARCInstKind.h" 27 #include "llvm/Transforms/Utils/Local.h" 28 #include "llvm/Transforms/ObjCARC.h" 29 30 namespace llvm { 31 namespace objcarc { 32 33 /// Erase the given instruction. 34 /// 35 /// Many ObjC calls return their argument verbatim, 36 /// so if it's such a call and the return value has users, replace them with the 37 /// argument value. 38 /// 39 static inline void EraseInstruction(Instruction *CI) { 40 Value *OldArg = cast<CallInst>(CI)->getArgOperand(0); 41 42 bool Unused = CI->use_empty(); 43 44 if (!Unused) { 45 // Replace the return value with the argument. 46 assert((IsForwarding(GetBasicARCInstKind(CI)) || 47 (IsNoopOnNull(GetBasicARCInstKind(CI)) && 48 IsNullOrUndef(OldArg->stripPointerCasts()))) && 49 "Can't delete non-forwarding instruction with users!"); 50 CI->replaceAllUsesWith(OldArg); 51 } 52 53 CI->eraseFromParent(); 54 55 if (Unused) 56 RecursivelyDeleteTriviallyDeadInstructions(OldArg); 57 } 58 59 /// If Inst is a ReturnRV and its operand is a call or invoke, return the 60 /// operand. Otherwise return null. 61 static inline const Instruction *getreturnRVOperand(const Instruction &Inst, 62 ARCInstKind Class) { 63 if (Class != ARCInstKind::RetainRV) 64 return nullptr; 65 66 const auto *Opnd = Inst.getOperand(0)->stripPointerCasts(); 67 if (const auto *C = dyn_cast<CallInst>(Opnd)) 68 return C; 69 return dyn_cast<InvokeInst>(Opnd); 70 } 71 72 /// Return the list of PHI nodes that are equivalent to PN. 73 template<class PHINodeTy, class VectorTy> 74 void getEquivalentPHIs(PHINodeTy &PN, VectorTy &PHIList) { 75 auto *BB = PN.getParent(); 76 for (auto &P : BB->phis()) { 77 if (&P == &PN) // Do not add PN to the list. 78 continue; 79 unsigned I = 0, E = PN.getNumIncomingValues(); 80 for (; I < E; ++I) { 81 auto *BB = PN.getIncomingBlock(I); 82 auto *PNOpnd = PN.getIncomingValue(I)->stripPointerCasts(); 83 auto *POpnd = P.getIncomingValueForBlock(BB)->stripPointerCasts(); 84 if (PNOpnd != POpnd) 85 break; 86 } 87 if (I == E) 88 PHIList.push_back(&P); 89 } 90 } 91 92 } // end namespace objcarc 93 } // end namespace llvm 94 95 #endif 96