blob: 79ecc4308fe7f605de744e5464a171a4bd201a62 [file] [log] [blame]
Nick Lewycky50f02cb2011-12-02 22:16:29 +00001//===-- Analysis.cpp - CodeGen LLVM IR Analysis Utilities -----------------===//
Dan Gohman450aa642010-04-21 01:22:34 +00002//
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//
Eric Christopherdb5028b2014-06-10 20:07:29 +000010// This file defines several CodeGen-specific LLVM IR analysis utilities.
Dan Gohman450aa642010-04-21 01:22:34 +000011//
12//===----------------------------------------------------------------------===//
13
Eric Christopher09fc2762014-06-10 20:39:35 +000014#include "llvm/CodeGen/Analysis.h"
Eric Christopherdda00092014-06-25 22:36:37 +000015#include "llvm/Analysis/ValueTracking.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000016#include "llvm/CodeGen/MachineFunction.h"
David Majnemer16193552015-10-04 02:22:52 +000017#include "llvm/CodeGen/MachineModuleInfo.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000018#include "llvm/IR/DataLayout.h"
19#include "llvm/IR/DerivedTypes.h"
20#include "llvm/IR/Function.h"
21#include "llvm/IR/Instructions.h"
22#include "llvm/IR/IntrinsicInst.h"
23#include "llvm/IR/LLVMContext.h"
24#include "llvm/IR/Module.h"
Dan Gohman450aa642010-04-21 01:22:34 +000025#include "llvm/Support/ErrorHandling.h"
26#include "llvm/Support/MathExtras.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000027#include "llvm/Target/TargetLowering.h"
David Majnemer16193552015-10-04 02:22:52 +000028#include "llvm/Target/TargetInstrInfo.h"
Eric Christopherd9134482014-08-04 21:25:23 +000029#include "llvm/Target/TargetSubtargetInfo.h"
Rafael Espindolaf21434c2014-07-30 19:42:16 +000030#include "llvm/Transforms/Utils/GlobalStatus.h"
Eric Christopherd9134482014-08-04 21:25:23 +000031
Dan Gohman450aa642010-04-21 01:22:34 +000032using namespace llvm;
33
Mehdi Amini8923cc52015-01-14 05:33:01 +000034/// Compute the linearized index of a member in a nested aggregate/struct/array
35/// by recursing and accumulating CurIndex as long as there are indices in the
36/// index list.
Chris Lattner229907c2011-07-18 04:54:35 +000037unsigned llvm::ComputeLinearIndex(Type *Ty,
Dan Gohman450aa642010-04-21 01:22:34 +000038 const unsigned *Indices,
39 const unsigned *IndicesEnd,
40 unsigned CurIndex) {
41 // Base case: We're done.
42 if (Indices && Indices == IndicesEnd)
43 return CurIndex;
44
45 // Given a struct type, recursively traverse the elements.
Chris Lattner229907c2011-07-18 04:54:35 +000046 if (StructType *STy = dyn_cast<StructType>(Ty)) {
Dan Gohman450aa642010-04-21 01:22:34 +000047 for (StructType::element_iterator EB = STy->element_begin(),
48 EI = EB,
49 EE = STy->element_end();
50 EI != EE; ++EI) {
51 if (Indices && *Indices == unsigned(EI - EB))
Dan Gohmanaadc5592010-10-06 16:18:29 +000052 return ComputeLinearIndex(*EI, Indices+1, IndicesEnd, CurIndex);
Craig Topperc0196b12014-04-14 00:51:57 +000053 CurIndex = ComputeLinearIndex(*EI, nullptr, nullptr, CurIndex);
Dan Gohman450aa642010-04-21 01:22:34 +000054 }
Mehdi Amini7b068f62015-01-14 05:38:48 +000055 assert(!Indices && "Unexpected out of bound");
Dan Gohman450aa642010-04-21 01:22:34 +000056 return CurIndex;
57 }
58 // Given an array type, recursively traverse the elements.
Chris Lattner229907c2011-07-18 04:54:35 +000059 else if (ArrayType *ATy = dyn_cast<ArrayType>(Ty)) {
60 Type *EltTy = ATy->getElementType();
Mehdi Amini8923cc52015-01-14 05:33:01 +000061 unsigned NumElts = ATy->getNumElements();
62 // Compute the Linear offset when jumping one element of the array
63 unsigned EltLinearOffset = ComputeLinearIndex(EltTy, nullptr, nullptr, 0);
Mehdi Amini7b068f62015-01-14 05:38:48 +000064 if (Indices) {
65 assert(*Indices < NumElts && "Unexpected out of bound");
Mehdi Amini8923cc52015-01-14 05:33:01 +000066 // If the indice is inside the array, compute the index to the requested
67 // elt and recurse inside the element with the end of the indices list
68 CurIndex += EltLinearOffset* *Indices;
69 return ComputeLinearIndex(EltTy, Indices+1, IndicesEnd, CurIndex);
Dan Gohman450aa642010-04-21 01:22:34 +000070 }
Mehdi Amini8923cc52015-01-14 05:33:01 +000071 CurIndex += EltLinearOffset*NumElts;
Dan Gohman450aa642010-04-21 01:22:34 +000072 return CurIndex;
73 }
74 // We haven't found the type we're looking for, so keep searching.
75 return CurIndex + 1;
76}
77
78/// ComputeValueVTs - Given an LLVM IR type, compute a sequence of
79/// EVTs that represent all the individual underlying
80/// non-aggregate types that comprise it.
81///
82/// If Offsets is non-null, it points to a vector to be filled in
83/// with the in-memory offsets of each of the individual values.
84///
Mehdi Amini56228da2015-07-09 01:57:34 +000085void llvm::ComputeValueVTs(const TargetLowering &TLI, const DataLayout &DL,
86 Type *Ty, SmallVectorImpl<EVT> &ValueVTs,
Dan Gohman450aa642010-04-21 01:22:34 +000087 SmallVectorImpl<uint64_t> *Offsets,
88 uint64_t StartingOffset) {
89 // Given a struct type, recursively traverse the elements.
Chris Lattner229907c2011-07-18 04:54:35 +000090 if (StructType *STy = dyn_cast<StructType>(Ty)) {
Mehdi Amini56228da2015-07-09 01:57:34 +000091 const StructLayout *SL = DL.getStructLayout(STy);
Dan Gohman450aa642010-04-21 01:22:34 +000092 for (StructType::element_iterator EB = STy->element_begin(),
93 EI = EB,
94 EE = STy->element_end();
95 EI != EE; ++EI)
Mehdi Amini56228da2015-07-09 01:57:34 +000096 ComputeValueVTs(TLI, DL, *EI, ValueVTs, Offsets,
Dan Gohman450aa642010-04-21 01:22:34 +000097 StartingOffset + SL->getElementOffset(EI - EB));
98 return;
99 }
100 // Given an array type, recursively traverse the elements.
Chris Lattner229907c2011-07-18 04:54:35 +0000101 if (ArrayType *ATy = dyn_cast<ArrayType>(Ty)) {
102 Type *EltTy = ATy->getElementType();
Mehdi Amini56228da2015-07-09 01:57:34 +0000103 uint64_t EltSize = DL.getTypeAllocSize(EltTy);
Dan Gohman450aa642010-04-21 01:22:34 +0000104 for (unsigned i = 0, e = ATy->getNumElements(); i != e; ++i)
Mehdi Amini56228da2015-07-09 01:57:34 +0000105 ComputeValueVTs(TLI, DL, EltTy, ValueVTs, Offsets,
Dan Gohman450aa642010-04-21 01:22:34 +0000106 StartingOffset + i * EltSize);
107 return;
108 }
109 // Interpret void as zero return values.
110 if (Ty->isVoidTy())
111 return;
112 // Base case: we can get an EVT for this LLVM IR type.
Mehdi Amini44ede332015-07-09 02:09:04 +0000113 ValueVTs.push_back(TLI.getValueType(DL, Ty));
Dan Gohman450aa642010-04-21 01:22:34 +0000114 if (Offsets)
115 Offsets->push_back(StartingOffset);
116}
117
118/// ExtractTypeInfo - Returns the type info, possibly bitcast, encoded in V.
Reid Kleckner283bc2e2014-11-14 00:35:50 +0000119GlobalValue *llvm::ExtractTypeInfo(Value *V) {
Dan Gohman450aa642010-04-21 01:22:34 +0000120 V = V->stripPointerCasts();
Reid Kleckner283bc2e2014-11-14 00:35:50 +0000121 GlobalValue *GV = dyn_cast<GlobalValue>(V);
122 GlobalVariable *Var = dyn_cast<GlobalVariable>(V);
Dan Gohman450aa642010-04-21 01:22:34 +0000123
Reid Kleckner283bc2e2014-11-14 00:35:50 +0000124 if (Var && Var->getName() == "llvm.eh.catch.all.value") {
125 assert(Var->hasInitializer() &&
Dan Gohman450aa642010-04-21 01:22:34 +0000126 "The EH catch-all value must have an initializer");
Reid Kleckner283bc2e2014-11-14 00:35:50 +0000127 Value *Init = Var->getInitializer();
128 GV = dyn_cast<GlobalValue>(Init);
Dan Gohman450aa642010-04-21 01:22:34 +0000129 if (!GV) V = cast<ConstantPointerNull>(Init);
130 }
131
132 assert((GV || isa<ConstantPointerNull>(V)) &&
133 "TypeInfo must be a global variable or NULL");
134 return GV;
135}
136
137/// hasInlineAsmMemConstraint - Return true if the inline asm instruction being
138/// processed uses a memory 'm' constraint.
139bool
John Thompsone8360b72010-10-29 17:29:13 +0000140llvm::hasInlineAsmMemConstraint(InlineAsm::ConstraintInfoVector &CInfos,
Dan Gohman450aa642010-04-21 01:22:34 +0000141 const TargetLowering &TLI) {
142 for (unsigned i = 0, e = CInfos.size(); i != e; ++i) {
143 InlineAsm::ConstraintInfo &CI = CInfos[i];
144 for (unsigned j = 0, ee = CI.Codes.size(); j != ee; ++j) {
145 TargetLowering::ConstraintType CType = TLI.getConstraintType(CI.Codes[j]);
146 if (CType == TargetLowering::C_Memory)
147 return true;
148 }
149
150 // Indirect operand accesses access memory.
151 if (CI.isIndirect)
152 return true;
153 }
154
155 return false;
156}
157
158/// getFCmpCondCode - Return the ISD condition code corresponding to
159/// the given LLVM IR floating-point condition code. This includes
160/// consideration of global floating-point math flags.
161///
162ISD::CondCode llvm::getFCmpCondCode(FCmpInst::Predicate Pred) {
Dan Gohman450aa642010-04-21 01:22:34 +0000163 switch (Pred) {
Nick Lewycky50f02cb2011-12-02 22:16:29 +0000164 case FCmpInst::FCMP_FALSE: return ISD::SETFALSE;
165 case FCmpInst::FCMP_OEQ: return ISD::SETOEQ;
166 case FCmpInst::FCMP_OGT: return ISD::SETOGT;
167 case FCmpInst::FCMP_OGE: return ISD::SETOGE;
168 case FCmpInst::FCMP_OLT: return ISD::SETOLT;
169 case FCmpInst::FCMP_OLE: return ISD::SETOLE;
170 case FCmpInst::FCMP_ONE: return ISD::SETONE;
171 case FCmpInst::FCMP_ORD: return ISD::SETO;
172 case FCmpInst::FCMP_UNO: return ISD::SETUO;
173 case FCmpInst::FCMP_UEQ: return ISD::SETUEQ;
174 case FCmpInst::FCMP_UGT: return ISD::SETUGT;
175 case FCmpInst::FCMP_UGE: return ISD::SETUGE;
176 case FCmpInst::FCMP_ULT: return ISD::SETULT;
177 case FCmpInst::FCMP_ULE: return ISD::SETULE;
178 case FCmpInst::FCMP_UNE: return ISD::SETUNE;
179 case FCmpInst::FCMP_TRUE: return ISD::SETTRUE;
David Blaikie46a9f012012-01-20 21:51:11 +0000180 default: llvm_unreachable("Invalid FCmp predicate opcode!");
Dan Gohman450aa642010-04-21 01:22:34 +0000181 }
Nick Lewycky50f02cb2011-12-02 22:16:29 +0000182}
183
184ISD::CondCode llvm::getFCmpCodeWithoutNaN(ISD::CondCode CC) {
185 switch (CC) {
186 case ISD::SETOEQ: case ISD::SETUEQ: return ISD::SETEQ;
187 case ISD::SETONE: case ISD::SETUNE: return ISD::SETNE;
188 case ISD::SETOLT: case ISD::SETULT: return ISD::SETLT;
189 case ISD::SETOLE: case ISD::SETULE: return ISD::SETLE;
190 case ISD::SETOGT: case ISD::SETUGT: return ISD::SETGT;
191 case ISD::SETOGE: case ISD::SETUGE: return ISD::SETGE;
David Blaikie46a9f012012-01-20 21:51:11 +0000192 default: return CC;
Nick Lewycky50f02cb2011-12-02 22:16:29 +0000193 }
Dan Gohman450aa642010-04-21 01:22:34 +0000194}
195
196/// getICmpCondCode - Return the ISD condition code corresponding to
197/// the given LLVM IR integer condition code.
198///
199ISD::CondCode llvm::getICmpCondCode(ICmpInst::Predicate Pred) {
200 switch (Pred) {
201 case ICmpInst::ICMP_EQ: return ISD::SETEQ;
202 case ICmpInst::ICMP_NE: return ISD::SETNE;
203 case ICmpInst::ICMP_SLE: return ISD::SETLE;
204 case ICmpInst::ICMP_ULE: return ISD::SETULE;
205 case ICmpInst::ICMP_SGE: return ISD::SETGE;
206 case ICmpInst::ICMP_UGE: return ISD::SETUGE;
207 case ICmpInst::ICMP_SLT: return ISD::SETLT;
208 case ICmpInst::ICMP_ULT: return ISD::SETULT;
209 case ICmpInst::ICMP_SGT: return ISD::SETGT;
210 case ICmpInst::ICMP_UGT: return ISD::SETUGT;
211 default:
212 llvm_unreachable("Invalid ICmp predicate opcode!");
Dan Gohman450aa642010-04-21 01:22:34 +0000213 }
214}
215
Stephen Linffc44542013-04-20 04:27:51 +0000216static bool isNoopBitcast(Type *T1, Type *T2,
Michael Gottesmanc0659fa2013-07-22 21:05:47 +0000217 const TargetLoweringBase& TLI) {
Stephen Linffc44542013-04-20 04:27:51 +0000218 return T1 == T2 || (T1->isPointerTy() && T2->isPointerTy()) ||
219 (isa<VectorType>(T1) && isa<VectorType>(T2) &&
220 TLI.isTypeLegal(EVT::getEVT(T1)) && TLI.isTypeLegal(EVT::getEVT(T2)));
Chris Lattner4f3615d2012-06-01 05:01:15 +0000221}
222
Tim Northovera4415852013-08-06 09:12:35 +0000223/// Look through operations that will be free to find the earliest source of
224/// this value.
225///
226/// @param ValLoc If V has aggegate type, we will be interested in a particular
227/// scalar component. This records its address; the reverse of this list gives a
228/// sequence of indices appropriate for an extractvalue to locate the important
229/// value. This value is updated during the function and on exit will indicate
230/// similar information for the Value returned.
231///
232/// @param DataBits If this function looks through truncate instructions, this
233/// will record the smallest size attained.
234static const Value *getNoopInput(const Value *V,
235 SmallVectorImpl<unsigned> &ValLoc,
236 unsigned &DataBits,
Mehdi Amini44ede332015-07-09 02:09:04 +0000237 const TargetLoweringBase &TLI,
238 const DataLayout &DL) {
Stephen Linffc44542013-04-20 04:27:51 +0000239 while (true) {
Stephen Linffc44542013-04-20 04:27:51 +0000240 // Try to look through V1; if V1 is not an instruction, it can't be looked
241 // through.
Tim Northovera4415852013-08-06 09:12:35 +0000242 const Instruction *I = dyn_cast<Instruction>(V);
243 if (!I || I->getNumOperands() == 0) return V;
Craig Topperc0196b12014-04-14 00:51:57 +0000244 const Value *NoopInput = nullptr;
Tim Northovera4415852013-08-06 09:12:35 +0000245
246 Value *Op = I->getOperand(0);
247 if (isa<BitCastInst>(I)) {
248 // Look through truly no-op bitcasts.
249 if (isNoopBitcast(Op->getType(), I->getType(), TLI))
250 NoopInput = Op;
251 } else if (isa<GetElementPtrInst>(I)) {
252 // Look through getelementptr
253 if (cast<GetElementPtrInst>(I)->hasAllZeroIndices())
254 NoopInput = Op;
255 } else if (isa<IntToPtrInst>(I)) {
256 // Look through inttoptr.
257 // Make sure this isn't a truncating or extending cast. We could
258 // support this eventually, but don't bother for now.
259 if (!isa<VectorType>(I->getType()) &&
Mehdi Amini44ede332015-07-09 02:09:04 +0000260 DL.getPointerSizeInBits() ==
261 cast<IntegerType>(Op->getType())->getBitWidth())
Tim Northovera4415852013-08-06 09:12:35 +0000262 NoopInput = Op;
263 } else if (isa<PtrToIntInst>(I)) {
264 // Look through ptrtoint.
265 // Make sure this isn't a truncating or extending cast. We could
266 // support this eventually, but don't bother for now.
267 if (!isa<VectorType>(I->getType()) &&
Mehdi Amini44ede332015-07-09 02:09:04 +0000268 DL.getPointerSizeInBits() ==
269 cast<IntegerType>(I->getType())->getBitWidth())
Tim Northovera4415852013-08-06 09:12:35 +0000270 NoopInput = Op;
271 } else if (isa<TruncInst>(I) &&
272 TLI.allowTruncateForTailCall(Op->getType(), I->getType())) {
273 DataBits = std::min(DataBits, I->getType()->getPrimitiveSizeInBits());
274 NoopInput = Op;
Ahmed Bougacha8a413192017-01-03 21:42:43 +0000275 } else if (auto CS = ImmutableCallSite(I)) {
276 const Value *ReturnedOp = CS.getReturnedArgOperand();
Ahmed Bougacha6aff7442017-01-03 20:33:22 +0000277 if (ReturnedOp && isNoopBitcast(ReturnedOp->getType(), I->getType(), TLI))
278 NoopInput = ReturnedOp;
Tim Northovera4415852013-08-06 09:12:35 +0000279 } else if (const InsertValueInst *IVI = dyn_cast<InsertValueInst>(V)) {
280 // Value may come from either the aggregate or the scalar
281 ArrayRef<unsigned> InsertLoc = IVI->getIndices();
Tim Northovere4310fe2015-05-06 20:07:38 +0000282 if (ValLoc.size() >= InsertLoc.size() &&
283 std::equal(InsertLoc.begin(), InsertLoc.end(), ValLoc.rbegin())) {
Tim Northovera4415852013-08-06 09:12:35 +0000284 // The type being inserted is a nested sub-type of the aggregate; we
285 // have to remove those initial indices to get the location we're
286 // interested in for the operand.
287 ValLoc.resize(ValLoc.size() - InsertLoc.size());
288 NoopInput = IVI->getInsertedValueOperand();
289 } else {
290 // The struct we're inserting into has the value we're interested in, no
291 // change of address.
292 NoopInput = Op;
293 }
294 } else if (const ExtractValueInst *EVI = dyn_cast<ExtractValueInst>(V)) {
295 // The part we're interested in will inevitably be some sub-section of the
296 // previous aggregate. Combine the two paths to obtain the true address of
297 // our element.
298 ArrayRef<unsigned> ExtractLoc = EVI->getIndices();
Benjamin Kramer4f6ac162015-02-28 10:11:12 +0000299 ValLoc.append(ExtractLoc.rbegin(), ExtractLoc.rend());
Tim Northovera4415852013-08-06 09:12:35 +0000300 NoopInput = Op;
Stephen Linffc44542013-04-20 04:27:51 +0000301 }
Tim Northovera4415852013-08-06 09:12:35 +0000302 // Terminate if we couldn't find anything to look through.
303 if (!NoopInput)
304 return V;
Stephen Linffc44542013-04-20 04:27:51 +0000305
Tim Northovera4415852013-08-06 09:12:35 +0000306 V = NoopInput;
Stephen Linffc44542013-04-20 04:27:51 +0000307 }
Stephen Linffc44542013-04-20 04:27:51 +0000308}
Chris Lattner4f3615d2012-06-01 05:01:15 +0000309
Tim Northovera4415852013-08-06 09:12:35 +0000310/// Return true if this scalar return value only has bits discarded on its path
311/// from the "tail call" to the "ret". This includes the obvious noop
312/// instructions handled by getNoopInput above as well as free truncations (or
313/// extensions prior to the call).
314static bool slotOnlyDiscardsData(const Value *RetVal, const Value *CallVal,
315 SmallVectorImpl<unsigned> &RetIndices,
316 SmallVectorImpl<unsigned> &CallIndices,
Tim Northover707d68f2013-08-12 09:45:46 +0000317 bool AllowDifferingSizes,
Mehdi Amini44ede332015-07-09 02:09:04 +0000318 const TargetLoweringBase &TLI,
319 const DataLayout &DL) {
Tim Northovera4415852013-08-06 09:12:35 +0000320
321 // Trace the sub-value needed by the return value as far back up the graph as
322 // possible, in the hope that it will intersect with the value produced by the
323 // call. In the simple case with no "returned" attribute, the hope is actually
324 // that we end up back at the tail call instruction itself.
325 unsigned BitsRequired = UINT_MAX;
Mehdi Amini44ede332015-07-09 02:09:04 +0000326 RetVal = getNoopInput(RetVal, RetIndices, BitsRequired, TLI, DL);
Tim Northovera4415852013-08-06 09:12:35 +0000327
328 // If this slot in the value returned is undef, it doesn't matter what the
329 // call puts there, it'll be fine.
330 if (isa<UndefValue>(RetVal))
331 return true;
332
333 // Now do a similar search up through the graph to find where the value
334 // actually returned by the "tail call" comes from. In the simple case without
335 // a "returned" attribute, the search will be blocked immediately and the loop
336 // a Noop.
337 unsigned BitsProvided = UINT_MAX;
Mehdi Amini44ede332015-07-09 02:09:04 +0000338 CallVal = getNoopInput(CallVal, CallIndices, BitsProvided, TLI, DL);
Tim Northovera4415852013-08-06 09:12:35 +0000339
340 // There's no hope if we can't actually trace them to (the same part of!) the
341 // same value.
342 if (CallVal != RetVal || CallIndices != RetIndices)
343 return false;
344
345 // However, intervening truncates may have made the call non-tail. Make sure
346 // all the bits that are needed by the "ret" have been provided by the "tail
347 // call". FIXME: with sufficiently cunning bit-tracking, we could look through
348 // extensions too.
Tim Northover707d68f2013-08-12 09:45:46 +0000349 if (BitsProvided < BitsRequired ||
350 (!AllowDifferingSizes && BitsProvided != BitsRequired))
Tim Northovera4415852013-08-06 09:12:35 +0000351 return false;
352
353 return true;
354}
355
356/// For an aggregate type, determine whether a given index is within bounds or
357/// not.
358static bool indexReallyValid(CompositeType *T, unsigned Idx) {
359 if (ArrayType *AT = dyn_cast<ArrayType>(T))
360 return Idx < AT->getNumElements();
361
362 return Idx < cast<StructType>(T)->getNumElements();
363}
364
365/// Move the given iterators to the next leaf type in depth first traversal.
366///
367/// Performs a depth-first traversal of the type as specified by its arguments,
368/// stopping at the next leaf node (which may be a legitimate scalar type or an
369/// empty struct or array).
370///
371/// @param SubTypes List of the partial components making up the type from
372/// outermost to innermost non-empty aggregate. The element currently
373/// represented is SubTypes.back()->getTypeAtIndex(Path.back() - 1).
374///
375/// @param Path Set of extractvalue indices leading from the outermost type
376/// (SubTypes[0]) to the leaf node currently represented.
377///
378/// @returns true if a new type was found, false otherwise. Calling this
379/// function again on a finished iterator will repeatedly return
380/// false. SubTypes.back()->getTypeAtIndex(Path.back()) is either an empty
381/// aggregate or a non-aggregate
Benjamin Kramerdf034492013-08-09 14:44:41 +0000382static bool advanceToNextLeafType(SmallVectorImpl<CompositeType *> &SubTypes,
383 SmallVectorImpl<unsigned> &Path) {
Tim Northovera4415852013-08-06 09:12:35 +0000384 // First march back up the tree until we can successfully increment one of the
385 // coordinates in Path.
386 while (!Path.empty() && !indexReallyValid(SubTypes.back(), Path.back() + 1)) {
387 Path.pop_back();
388 SubTypes.pop_back();
389 }
390
391 // If we reached the top, then the iterator is done.
392 if (Path.empty())
393 return false;
394
395 // We know there's *some* valid leaf now, so march back down the tree picking
396 // out the left-most element at each node.
397 ++Path.back();
398 Type *DeeperType = SubTypes.back()->getTypeAtIndex(Path.back());
399 while (DeeperType->isAggregateType()) {
400 CompositeType *CT = cast<CompositeType>(DeeperType);
401 if (!indexReallyValid(CT, 0))
402 return true;
403
404 SubTypes.push_back(CT);
405 Path.push_back(0);
406
407 DeeperType = CT->getTypeAtIndex(0U);
408 }
409
410 return true;
411}
412
413/// Find the first non-empty, scalar-like type in Next and setup the iterator
414/// components.
415///
416/// Assuming Next is an aggregate of some kind, this function will traverse the
417/// tree from left to right (i.e. depth-first) looking for the first
418/// non-aggregate type which will play a role in function return.
419///
420/// For example, if Next was {[0 x i64], {{}, i32, {}}, i32} then we would setup
421/// Path as [1, 1] and SubTypes as [Next, {{}, i32, {}}] to represent the first
422/// i32 in that type.
423static bool firstRealType(Type *Next,
424 SmallVectorImpl<CompositeType *> &SubTypes,
425 SmallVectorImpl<unsigned> &Path) {
426 // First initialise the iterator components to the first "leaf" node
427 // (i.e. node with no valid sub-type at any index, so {} does count as a leaf
428 // despite nominally being an aggregate).
429 while (Next->isAggregateType() &&
430 indexReallyValid(cast<CompositeType>(Next), 0)) {
431 SubTypes.push_back(cast<CompositeType>(Next));
432 Path.push_back(0);
433 Next = cast<CompositeType>(Next)->getTypeAtIndex(0U);
434 }
435
436 // If there's no Path now, Next was originally scalar already (or empty
437 // leaf). We're done.
438 if (Path.empty())
439 return true;
440
441 // Otherwise, use normal iteration to keep looking through the tree until we
442 // find a non-aggregate type.
443 while (SubTypes.back()->getTypeAtIndex(Path.back())->isAggregateType()) {
444 if (!advanceToNextLeafType(SubTypes, Path))
445 return false;
446 }
447
448 return true;
449}
450
451/// Set the iterator data-structures to the next non-empty, non-aggregate
452/// subtype.
Benjamin Kramerdf034492013-08-09 14:44:41 +0000453static bool nextRealType(SmallVectorImpl<CompositeType *> &SubTypes,
454 SmallVectorImpl<unsigned> &Path) {
Tim Northovera4415852013-08-06 09:12:35 +0000455 do {
456 if (!advanceToNextLeafType(SubTypes, Path))
457 return false;
458
459 assert(!Path.empty() && "found a leaf but didn't set the path?");
460 } while (SubTypes.back()->getTypeAtIndex(Path.back())->isAggregateType());
461
462 return true;
463}
464
465
Dan Gohman450aa642010-04-21 01:22:34 +0000466/// Test if the given instruction is in a position to be optimized
467/// with a tail-call. This roughly means that it's in a block with
468/// a return and there's nothing that needs to be scheduled
469/// between it and the return.
470///
471/// This function only tests target-independent requirements.
Juergen Ributzka480872b2014-07-16 00:01:22 +0000472bool llvm::isInTailCallPosition(ImmutableCallSite CS, const TargetMachine &TM) {
Dan Gohman450aa642010-04-21 01:22:34 +0000473 const Instruction *I = CS.getInstruction();
474 const BasicBlock *ExitBB = I->getParent();
475 const TerminatorInst *Term = ExitBB->getTerminator();
476 const ReturnInst *Ret = dyn_cast<ReturnInst>(Term);
Dan Gohman450aa642010-04-21 01:22:34 +0000477
478 // The block must end in a return statement or unreachable.
479 //
480 // FIXME: Decline tailcall if it's not guaranteed and if the block ends in
481 // an unreachable, for now. The way tailcall optimization is currently
482 // implemented means it will add an epilogue followed by a jump. That is
483 // not profitable. Also, if the callee is a special function (e.g.
484 // longjmp on x86), it can end up causing miscompilation that has not
485 // been fully understood.
486 if (!Ret &&
Juergen Ributzka4ce98632014-07-11 20:50:47 +0000487 (!TM.Options.GuaranteedTailCallOpt || !isa<UnreachableInst>(Term)))
Chris Lattner4f3615d2012-06-01 05:01:15 +0000488 return false;
Dan Gohman450aa642010-04-21 01:22:34 +0000489
490 // If I will have a chain, make sure no other instruction that will have a
491 // chain interposes between I and the return.
David Majnemer0a92f862015-08-28 21:13:39 +0000492 if (I->mayHaveSideEffects() || I->mayReadFromMemory() ||
493 !isSafeToSpeculativelyExecute(I))
494 for (BasicBlock::const_iterator BBI = std::prev(ExitBB->end(), 2);; --BBI) {
495 if (&*BBI == I)
496 break;
497 // Debug info intrinsics do not get in the way of tail call optimization.
498 if (isa<DbgInfoIntrinsic>(BBI))
499 continue;
500 if (BBI->mayHaveSideEffects() || BBI->mayReadFromMemory() ||
Duncan P. N. Exon Smith980f8f22015-10-09 18:23:49 +0000501 !isSafeToSpeculativelyExecute(&*BBI))
David Majnemer0a92f862015-08-28 21:13:39 +0000502 return false;
503 }
Dan Gohman450aa642010-04-21 01:22:34 +0000504
Eric Christopherf734a8b2015-02-20 18:44:17 +0000505 const Function *F = ExitBB->getParent();
Eric Christopherd9134482014-08-04 21:25:23 +0000506 return returnTypeIsEligibleForTailCall(
Eric Christopherf734a8b2015-02-20 18:44:17 +0000507 F, I, Ret, *TM.getSubtargetImpl(*F)->getTargetLowering());
Michael Gottesmance0e4c22013-08-20 08:36:50 +0000508}
509
Michael Kupersteinf79af6f2016-09-08 00:48:37 +0000510bool llvm::attributesPermitTailCall(const Function *F, const Instruction *I,
511 const ReturnInst *Ret,
512 const TargetLoweringBase &TLI,
513 bool *AllowDifferingSizes) {
514 // ADS may be null, so don't write to it directly.
515 bool DummyADS;
516 bool &ADS = AllowDifferingSizes ? *AllowDifferingSizes : DummyADS;
517 ADS = true;
518
519 AttrBuilder CallerAttrs(F->getAttributes(),
520 AttributeSet::ReturnIndex);
521 AttrBuilder CalleeAttrs(cast<CallInst>(I)->getAttributes(),
522 AttributeSet::ReturnIndex);
523
524 // Noalias is completely benign as far as calling convention goes, it
525 // shouldn't affect whether the call is a tail call.
Bjorn Pettersson807f7322016-10-27 14:48:09 +0000526 CallerAttrs.removeAttribute(Attribute::NoAlias);
527 CalleeAttrs.removeAttribute(Attribute::NoAlias);
Michael Kupersteinf79af6f2016-09-08 00:48:37 +0000528
529 if (CallerAttrs.contains(Attribute::ZExt)) {
530 if (!CalleeAttrs.contains(Attribute::ZExt))
531 return false;
532
533 ADS = false;
534 CallerAttrs.removeAttribute(Attribute::ZExt);
535 CalleeAttrs.removeAttribute(Attribute::ZExt);
536 } else if (CallerAttrs.contains(Attribute::SExt)) {
537 if (!CalleeAttrs.contains(Attribute::SExt))
538 return false;
539
540 ADS = false;
541 CallerAttrs.removeAttribute(Attribute::SExt);
542 CalleeAttrs.removeAttribute(Attribute::SExt);
543 }
544
545 // If they're still different, there's some facet we don't understand
546 // (currently only "inreg", but in future who knows). It may be OK but the
547 // only safe option is to reject the tail call.
548 return CallerAttrs == CalleeAttrs;
549}
550
Michael Gottesmance0e4c22013-08-20 08:36:50 +0000551bool llvm::returnTypeIsEligibleForTailCall(const Function *F,
552 const Instruction *I,
553 const ReturnInst *Ret,
554 const TargetLoweringBase &TLI) {
Dan Gohman450aa642010-04-21 01:22:34 +0000555 // If the block ends with a void return or unreachable, it doesn't matter
556 // what the call's return type is.
557 if (!Ret || Ret->getNumOperands() == 0) return true;
558
559 // If the return value is undef, it doesn't matter what the call's
560 // return type is.
561 if (isa<UndefValue>(Ret->getOperand(0))) return true;
562
Tim Northover707d68f2013-08-12 09:45:46 +0000563 // Make sure the attributes attached to each return are compatible.
Michael Kupersteinf79af6f2016-09-08 00:48:37 +0000564 bool AllowDifferingSizes;
565 if (!attributesPermitTailCall(F, I, Ret, TLI, &AllowDifferingSizes))
Dan Gohman450aa642010-04-21 01:22:34 +0000566 return false;
567
Tim Northovera4415852013-08-06 09:12:35 +0000568 const Value *RetVal = Ret->getOperand(0), *CallVal = I;
569 SmallVector<unsigned, 4> RetPath, CallPath;
570 SmallVector<CompositeType *, 4> RetSubTypes, CallSubTypes;
571
572 bool RetEmpty = !firstRealType(RetVal->getType(), RetSubTypes, RetPath);
573 bool CallEmpty = !firstRealType(CallVal->getType(), CallSubTypes, CallPath);
574
575 // Nothing's actually returned, it doesn't matter what the callee put there
576 // it's a valid tail call.
577 if (RetEmpty)
578 return true;
579
580 // Iterate pairwise through each of the value types making up the tail call
581 // and the corresponding return. For each one we want to know whether it's
582 // essentially going directly from the tail call to the ret, via operations
583 // that end up not generating any code.
584 //
585 // We allow a certain amount of covariance here. For example it's permitted
586 // for the tail call to define more bits than the ret actually cares about
587 // (e.g. via a truncate).
588 do {
589 if (CallEmpty) {
590 // We've exhausted the values produced by the tail call instruction, the
591 // rest are essentially undef. The type doesn't really matter, but we need
592 // *something*.
593 Type *SlotType = RetSubTypes.back()->getTypeAtIndex(RetPath.back());
594 CallVal = UndefValue::get(SlotType);
595 }
596
597 // The manipulations performed when we're looking through an insertvalue or
598 // an extractvalue would happen at the front of the RetPath list, so since
599 // we have to copy it anyway it's more efficient to create a reversed copy.
Benjamin Kramer4f6ac162015-02-28 10:11:12 +0000600 SmallVector<unsigned, 4> TmpRetPath(RetPath.rbegin(), RetPath.rend());
601 SmallVector<unsigned, 4> TmpCallPath(CallPath.rbegin(), CallPath.rend());
Tim Northovera4415852013-08-06 09:12:35 +0000602
603 // Finally, we can check whether the value produced by the tail call at this
604 // index is compatible with the value we return.
Tim Northover707d68f2013-08-12 09:45:46 +0000605 if (!slotOnlyDiscardsData(RetVal, CallVal, TmpRetPath, TmpCallPath,
Mehdi Amini44ede332015-07-09 02:09:04 +0000606 AllowDifferingSizes, TLI,
607 F->getParent()->getDataLayout()))
Tim Northovera4415852013-08-06 09:12:35 +0000608 return false;
609
610 CallEmpty = !nextRealType(CallSubTypes, CallPath);
611 } while(nextRealType(RetSubTypes, RetPath));
612
613 return true;
Dan Gohman450aa642010-04-21 01:22:34 +0000614}
Rafael Espindolaf21434c2014-07-30 19:42:16 +0000615
616bool llvm::canBeOmittedFromSymbolTable(const GlobalValue *GV) {
617 if (!GV->hasLinkOnceODRLinkage())
618 return false;
619
Peter Collingbourne96efdd62016-06-14 21:01:22 +0000620 // We assume that anyone who sets global unnamed_addr on a non-constant knows
621 // what they're doing.
622 if (GV->hasGlobalUnnamedAddr())
Rafael Espindolaf21434c2014-07-30 19:42:16 +0000623 return true;
624
625 // If it is a non constant variable, it needs to be uniqued across shared
626 // objects.
627 if (const GlobalVariable *Var = dyn_cast<GlobalVariable>(GV)) {
628 if (!Var->isConstant())
629 return false;
630 }
631
Peter Collingbourne96efdd62016-06-14 21:01:22 +0000632 return GV->hasAtLeastLocalUnnamedAddr();
Rafael Espindolaf21434c2014-07-30 19:42:16 +0000633}
David Majnemer16193552015-10-04 02:22:52 +0000634
635static void collectFuncletMembers(
636 DenseMap<const MachineBasicBlock *, int> &FuncletMembership, int Funclet,
637 const MachineBasicBlock *MBB) {
David Majnemer734d7c32016-01-22 18:49:50 +0000638 SmallVector<const MachineBasicBlock *, 16> Worklist = {MBB};
639 while (!Worklist.empty()) {
640 const MachineBasicBlock *Visiting = Worklist.pop_back_val();
641 // Don't follow blocks which start new funclets.
642 if (Visiting->isEHPad() && Visiting != MBB)
643 continue;
David Blaikie7b54b522015-10-26 18:41:13 +0000644
David Majnemer734d7c32016-01-22 18:49:50 +0000645 // Add this MBB to our funclet.
646 auto P = FuncletMembership.insert(std::make_pair(Visiting, Funclet));
647
648 // Don't revisit blocks.
649 if (!P.second) {
650 assert(P.first->second == Funclet && "MBB is part of two funclets!");
651 continue;
652 }
653
654 // Returns are boundaries where funclet transfer can occur, don't follow
655 // successors.
656 if (Visiting->isReturnBlock())
657 continue;
658
659 for (const MachineBasicBlock *Succ : Visiting->successors())
660 Worklist.push_back(Succ);
David Majnemer16193552015-10-04 02:22:52 +0000661 }
David Majnemer16193552015-10-04 02:22:52 +0000662}
663
664DenseMap<const MachineBasicBlock *, int>
665llvm::getFuncletMembership(const MachineFunction &MF) {
666 DenseMap<const MachineBasicBlock *, int> FuncletMembership;
667
668 // We don't have anything to do if there aren't any EH pads.
Matthias Braund0ee66c2016-12-01 19:32:15 +0000669 if (!MF.hasEHFunclets())
David Majnemer16193552015-10-04 02:22:52 +0000670 return FuncletMembership;
671
David Majnemere4f9b092015-10-05 20:09:16 +0000672 int EntryBBNumber = MF.front().getNumber();
David Majnemer16193552015-10-04 02:22:52 +0000673 bool IsSEH = isAsynchronousEHPersonality(
674 classifyEHPersonality(MF.getFunction()->getPersonalityFn()));
675
676 const TargetInstrInfo *TII = MF.getSubtarget().getInstrInfo();
677 SmallVector<const MachineBasicBlock *, 16> FuncletBlocks;
David Majnemere4f9b092015-10-05 20:09:16 +0000678 SmallVector<const MachineBasicBlock *, 16> UnreachableBlocks;
679 SmallVector<const MachineBasicBlock *, 16> SEHCatchPads;
David Majnemer16193552015-10-04 02:22:52 +0000680 SmallVector<std::pair<const MachineBasicBlock *, int>, 16> CatchRetSuccessors;
681 for (const MachineBasicBlock &MBB : MF) {
David Majnemere4f9b092015-10-05 20:09:16 +0000682 if (MBB.isEHFuncletEntry()) {
David Majnemer16193552015-10-04 02:22:52 +0000683 FuncletBlocks.push_back(&MBB);
David Majnemere4f9b092015-10-05 20:09:16 +0000684 } else if (IsSEH && MBB.isEHPad()) {
685 SEHCatchPads.push_back(&MBB);
686 } else if (MBB.pred_empty()) {
687 UnreachableBlocks.push_back(&MBB);
688 }
David Majnemer16193552015-10-04 02:22:52 +0000689
690 MachineBasicBlock::const_iterator MBBI = MBB.getFirstTerminator();
Duncan P. N. Exon Smith2e7af972016-08-11 15:29:02 +0000691
David Majnemer16193552015-10-04 02:22:52 +0000692 // CatchPads are not funclets for SEH so do not consider CatchRet to
693 // transfer control to another funclet.
Reid Kleckner26f9e9e2016-08-11 16:00:43 +0000694 if (MBBI == MBB.end() || MBBI->getOpcode() != TII->getCatchReturnOpcode())
David Majnemer16193552015-10-04 02:22:52 +0000695 continue;
696
David Majnemere4f9b092015-10-05 20:09:16 +0000697 // FIXME: SEH CatchPads are not necessarily in the parent function:
698 // they could be inside a finally block.
David Majnemer16193552015-10-04 02:22:52 +0000699 const MachineBasicBlock *Successor = MBBI->getOperand(0).getMBB();
700 const MachineBasicBlock *SuccessorColor = MBBI->getOperand(1).getMBB();
David Majnemere4f9b092015-10-05 20:09:16 +0000701 CatchRetSuccessors.push_back(
702 {Successor, IsSEH ? EntryBBNumber : SuccessorColor->getNumber()});
David Majnemer16193552015-10-04 02:22:52 +0000703 }
704
705 // We don't have anything to do if there aren't any EH pads.
706 if (FuncletBlocks.empty())
707 return FuncletMembership;
708
709 // Identify all the basic blocks reachable from the function entry.
Duncan P. N. Exon Smith980f8f22015-10-09 18:23:49 +0000710 collectFuncletMembers(FuncletMembership, EntryBBNumber, &MF.front());
David Majnemere4f9b092015-10-05 20:09:16 +0000711 // All blocks not part of a funclet are in the parent function.
712 for (const MachineBasicBlock *MBB : UnreachableBlocks)
713 collectFuncletMembers(FuncletMembership, EntryBBNumber, MBB);
David Majnemer16193552015-10-04 02:22:52 +0000714 // Next, identify all the blocks inside the funclets.
715 for (const MachineBasicBlock *MBB : FuncletBlocks)
716 collectFuncletMembers(FuncletMembership, MBB->getNumber(), MBB);
David Majnemere4f9b092015-10-05 20:09:16 +0000717 // SEH CatchPads aren't really funclets, handle them separately.
718 for (const MachineBasicBlock *MBB : SEHCatchPads)
719 collectFuncletMembers(FuncletMembership, EntryBBNumber, MBB);
David Majnemer16193552015-10-04 02:22:52 +0000720 // Finally, identify all the targets of a catchret.
721 for (std::pair<const MachineBasicBlock *, int> CatchRetPair :
722 CatchRetSuccessors)
723 collectFuncletMembers(FuncletMembership, CatchRetPair.second,
724 CatchRetPair.first);
David Majnemer16193552015-10-04 02:22:52 +0000725 return FuncletMembership;
726}