1 //===- llvm/CodeGen/DwarfCompileUnit.cpp - Dwarf Compile Units ------------===//
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 // This file contains support for constructing a dwarf compile unit.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "DwarfCompileUnit.h"
15 #include "AddressPool.h"
16 #include "DwarfDebug.h"
17 #include "DwarfExpression.h"
18 #include "DwarfUnit.h"
19 #include "llvm/ADT/None.h"
20 #include "llvm/ADT/STLExtras.h"
21 #include "llvm/ADT/SmallVector.h"
22 #include "llvm/ADT/StringRef.h"
23 #include "llvm/BinaryFormat/Dwarf.h"
24 #include "llvm/CodeGen/AsmPrinter.h"
25 #include "llvm/CodeGen/DIE.h"
26 #include "llvm/CodeGen/LexicalScopes.h"
27 #include "llvm/CodeGen/MachineFunction.h"
28 #include "llvm/CodeGen/MachineInstr.h"
29 #include "llvm/CodeGen/MachineOperand.h"
30 #include "llvm/CodeGen/TargetFrameLowering.h"
31 #include "llvm/CodeGen/TargetLoweringObjectFile.h"
32 #include "llvm/CodeGen/TargetRegisterInfo.h"
33 #include "llvm/CodeGen/TargetSubtargetInfo.h"
34 #include "llvm/IR/DataLayout.h"
35 #include "llvm/IR/DebugInfo.h"
36 #include "llvm/IR/DebugInfoMetadata.h"
37 #include "llvm/IR/GlobalVariable.h"
38 #include "llvm/MC/MCSection.h"
39 #include "llvm/MC/MCStreamer.h"
40 #include "llvm/MC/MCSymbol.h"
41 #include "llvm/MC/MachineLocation.h"
42 #include "llvm/Support/Casting.h"
43 #include "llvm/Target/TargetMachine.h"
44 #include "llvm/Target/TargetOptions.h"
45 #include <algorithm>
46 #include <cassert>
47 #include <cstdint>
48 #include <iterator>
49 #include <memory>
50 #include <string>
51 #include <utility>
52 
53 using namespace llvm;
54 
55 DwarfCompileUnit::DwarfCompileUnit(unsigned UID, const DICompileUnit *Node,
56                                    AsmPrinter *A, DwarfDebug *DW,
57                                    DwarfFile *DWU)
58     : DwarfUnit(dwarf::DW_TAG_compile_unit, Node, A, DW, DWU), UniqueID(UID) {
59   insertDIE(Node, &getUnitDie());
60   MacroLabelBegin = Asm->createTempSymbol("cu_macro_begin");
61 }
62 
63 /// addLabelAddress - Add a dwarf label attribute data and value using
64 /// DW_FORM_addr or DW_FORM_GNU_addr_index.
65 void DwarfCompileUnit::addLabelAddress(DIE &Die, dwarf::Attribute Attribute,
66                                        const MCSymbol *Label) {
67   // Don't use the address pool in non-fission or in the skeleton unit itself.
68   // FIXME: Once GDB supports this, it's probably worthwhile using the address
69   // pool from the skeleton - maybe even in non-fission (possibly fewer
70   // relocations by sharing them in the pool, but we have other ideas about how
71   // to reduce the number of relocations as well/instead).
72   if (!DD->useSplitDwarf() || !Skeleton)
73     return addLocalLabelAddress(Die, Attribute, Label);
74 
75   if (Label)
76     DD->addArangeLabel(SymbolCU(this, Label));
77 
78   unsigned idx = DD->getAddressPool().getIndex(Label);
79   Die.addValue(DIEValueAllocator, Attribute, dwarf::DW_FORM_GNU_addr_index,
80                DIEInteger(idx));
81 }
82 
83 void DwarfCompileUnit::addLocalLabelAddress(DIE &Die,
84                                             dwarf::Attribute Attribute,
85                                             const MCSymbol *Label) {
86   if (Label)
87     DD->addArangeLabel(SymbolCU(this, Label));
88 
89   if (Label)
90     Die.addValue(DIEValueAllocator, Attribute, dwarf::DW_FORM_addr,
91                  DIELabel(Label));
92   else
93     Die.addValue(DIEValueAllocator, Attribute, dwarf::DW_FORM_addr,
94                  DIEInteger(0));
95 }
96 
97 unsigned DwarfCompileUnit::getOrCreateSourceID(const DIFile *File) {
98   // If we print assembly, we can't separate .file entries according to
99   // compile units. Thus all files will belong to the default compile unit.
100 
101   // FIXME: add a better feature test than hasRawTextSupport. Even better,
102   // extend .file to support this.
103   unsigned CUID = Asm->OutStreamer->hasRawTextSupport() ? 0 : getUniqueID();
104   if (!File)
105     return Asm->OutStreamer->EmitDwarfFileDirective(0, "", "", nullptr, CUID);
106   return Asm->OutStreamer->EmitDwarfFileDirective(
107       0, File->getDirectory(), File->getFilename(), getMD5AsBytes(File), CUID);
108 }
109 
110 DIE *DwarfCompileUnit::getOrCreateGlobalVariableDIE(
111     const DIGlobalVariable *GV, ArrayRef<GlobalExpr> GlobalExprs) {
112   // Check for pre-existence.
113   if (DIE *Die = getDIE(GV))
114     return Die;
115 
116   assert(GV);
117 
118   auto *GVContext = GV->getScope();
119   auto *GTy = DD->resolve(GV->getType());
120 
121   // Construct the context before querying for the existence of the DIE in
122   // case such construction creates the DIE.
123   DIE *ContextDIE = getOrCreateContextDIE(GVContext);
124 
125   // Add to map.
126   DIE *VariableDIE = &createAndAddDIE(GV->getTag(), *ContextDIE, GV);
127   DIScope *DeclContext;
128   if (auto *SDMDecl = GV->getStaticDataMemberDeclaration()) {
129     DeclContext = resolve(SDMDecl->getScope());
130     assert(SDMDecl->isStaticMember() && "Expected static member decl");
131     assert(GV->isDefinition());
132     // We need the declaration DIE that is in the static member's class.
133     DIE *VariableSpecDIE = getOrCreateStaticMemberDIE(SDMDecl);
134     addDIEEntry(*VariableDIE, dwarf::DW_AT_specification, *VariableSpecDIE);
135     // If the global variable's type is different from the one in the class
136     // member type, assume that it's more specific and also emit it.
137     if (GTy != DD->resolve(SDMDecl->getBaseType()))
138       addType(*VariableDIE, GTy);
139   } else {
140     DeclContext = GV->getScope();
141     // Add name and type.
142     addString(*VariableDIE, dwarf::DW_AT_name, GV->getDisplayName());
143     addType(*VariableDIE, GTy);
144 
145     // Add scoping info.
146     if (!GV->isLocalToUnit())
147       addFlag(*VariableDIE, dwarf::DW_AT_external);
148 
149     // Add line number info.
150     addSourceLine(*VariableDIE, GV);
151   }
152 
153   if (!GV->isDefinition())
154     addFlag(*VariableDIE, dwarf::DW_AT_declaration);
155   else
156     addGlobalName(GV->getName(), *VariableDIE, DeclContext);
157 
158   if (uint32_t AlignInBytes = GV->getAlignInBytes())
159     addUInt(*VariableDIE, dwarf::DW_AT_alignment, dwarf::DW_FORM_udata,
160             AlignInBytes);
161 
162   // Add location.
163   bool addToAccelTable = false;
164   DIELoc *Loc = nullptr;
165   std::unique_ptr<DIEDwarfExpression> DwarfExpr;
166   for (const auto &GE : GlobalExprs) {
167     const GlobalVariable *Global = GE.Var;
168     const DIExpression *Expr = GE.Expr;
169 
170     // For compatibility with DWARF 3 and earlier,
171     // DW_AT_location(DW_OP_constu, X, DW_OP_stack_value) becomes
172     // DW_AT_const_value(X).
173     if (GlobalExprs.size() == 1 && Expr && Expr->isConstant()) {
174       addToAccelTable = true;
175       addConstantValue(*VariableDIE, /*Unsigned=*/true, Expr->getElement(1));
176       break;
177     }
178 
179     // We cannot describe the location of dllimport'd variables: the
180     // computation of their address requires loads from the IAT.
181     if (Global && Global->hasDLLImportStorageClass())
182       continue;
183 
184     // Nothing to describe without address or constant.
185     if (!Global && (!Expr || !Expr->isConstant()))
186       continue;
187 
188     if (!Loc) {
189       addToAccelTable = true;
190       Loc = new (DIEValueAllocator) DIELoc;
191       DwarfExpr = llvm::make_unique<DIEDwarfExpression>(*Asm, *this, *Loc);
192     }
193 
194     if (Global) {
195       const MCSymbol *Sym = Asm->getSymbol(Global);
196       if (Global->isThreadLocal()) {
197         if (Asm->TM.Options.EmulatedTLS) {
198           // TODO: add debug info for emulated thread local mode.
199         } else {
200           // FIXME: Make this work with -gsplit-dwarf.
201           unsigned PointerSize = Asm->getDataLayout().getPointerSize();
202           assert((PointerSize == 4 || PointerSize == 8) &&
203                  "Add support for other sizes if necessary");
204           // Based on GCC's support for TLS:
205           if (!DD->useSplitDwarf()) {
206             // 1) Start with a constNu of the appropriate pointer size
207             addUInt(*Loc, dwarf::DW_FORM_data1,
208                     PointerSize == 4 ? dwarf::DW_OP_const4u
209                                      : dwarf::DW_OP_const8u);
210             // 2) containing the (relocated) offset of the TLS variable
211             //    within the module's TLS block.
212             addExpr(*Loc, dwarf::DW_FORM_udata,
213                     Asm->getObjFileLowering().getDebugThreadLocalSymbol(Sym));
214           } else {
215             addUInt(*Loc, dwarf::DW_FORM_data1, dwarf::DW_OP_GNU_const_index);
216             addUInt(*Loc, dwarf::DW_FORM_udata,
217                     DD->getAddressPool().getIndex(Sym, /* TLS */ true));
218           }
219           // 3) followed by an OP to make the debugger do a TLS lookup.
220           addUInt(*Loc, dwarf::DW_FORM_data1,
221                   DD->useGNUTLSOpcode() ? dwarf::DW_OP_GNU_push_tls_address
222                                         : dwarf::DW_OP_form_tls_address);
223         }
224       } else {
225         DD->addArangeLabel(SymbolCU(this, Sym));
226         addOpAddress(*Loc, Sym);
227       }
228     }
229     if (Expr) {
230       DwarfExpr->addFragmentOffset(Expr);
231       DwarfExpr->addExpression(Expr);
232     }
233   }
234   if (Loc)
235     addBlock(*VariableDIE, dwarf::DW_AT_location, DwarfExpr->finalize());
236 
237   if (DD->useAllLinkageNames())
238     addLinkageName(*VariableDIE, GV->getLinkageName());
239 
240   if (addToAccelTable) {
241     DD->addAccelName(GV->getName(), *VariableDIE);
242 
243     // If the linkage name is different than the name, go ahead and output
244     // that as well into the name table.
245     if (GV->getLinkageName() != "" && GV->getName() != GV->getLinkageName())
246       DD->addAccelName(GV->getLinkageName(), *VariableDIE);
247   }
248 
249   return VariableDIE;
250 }
251 
252 void DwarfCompileUnit::addRange(RangeSpan Range) {
253   bool SameAsPrevCU = this == DD->getPrevCU();
254   DD->setPrevCU(this);
255   // If we have no current ranges just add the range and return, otherwise,
256   // check the current section and CU against the previous section and CU we
257   // emitted into and the subprogram was contained within. If these are the
258   // same then extend our current range, otherwise add this as a new range.
259   if (CURanges.empty() || !SameAsPrevCU ||
260       (&CURanges.back().getEnd()->getSection() !=
261        &Range.getEnd()->getSection())) {
262     CURanges.push_back(Range);
263     return;
264   }
265 
266   CURanges.back().setEnd(Range.getEnd());
267 }
268 
269 void DwarfCompileUnit::initStmtList() {
270   // Define start line table label for each Compile Unit.
271   MCSymbol *LineTableStartSym =
272       Asm->OutStreamer->getDwarfLineTableSymbol(getUniqueID());
273 
274   // DW_AT_stmt_list is a offset of line number information for this
275   // compile unit in debug_line section. For split dwarf this is
276   // left in the skeleton CU and so not included.
277   // The line table entries are not always emitted in assembly, so it
278   // is not okay to use line_table_start here.
279   const TargetLoweringObjectFile &TLOF = Asm->getObjFileLowering();
280   StmtListValue =
281       addSectionLabel(getUnitDie(), dwarf::DW_AT_stmt_list, LineTableStartSym,
282                       TLOF.getDwarfLineSection()->getBeginSymbol());
283 }
284 
285 void DwarfCompileUnit::applyStmtList(DIE &D) {
286   D.addValue(DIEValueAllocator, *StmtListValue);
287 }
288 
289 void DwarfCompileUnit::attachLowHighPC(DIE &D, const MCSymbol *Begin,
290                                        const MCSymbol *End) {
291   assert(Begin && "Begin label should not be null!");
292   assert(End && "End label should not be null!");
293   assert(Begin->isDefined() && "Invalid starting label");
294   assert(End->isDefined() && "Invalid end label");
295 
296   addLabelAddress(D, dwarf::DW_AT_low_pc, Begin);
297   if (DD->getDwarfVersion() < 4)
298     addLabelAddress(D, dwarf::DW_AT_high_pc, End);
299   else
300     addLabelDelta(D, dwarf::DW_AT_high_pc, End, Begin);
301 }
302 
303 // Find DIE for the given subprogram and attach appropriate DW_AT_low_pc
304 // and DW_AT_high_pc attributes. If there are global variables in this
305 // scope then create and insert DIEs for these variables.
306 DIE &DwarfCompileUnit::updateSubprogramScopeDIE(const DISubprogram *SP) {
307   DIE *SPDie = getOrCreateSubprogramDIE(SP, includeMinimalInlineScopes());
308 
309   attachLowHighPC(*SPDie, Asm->getFunctionBegin(), Asm->getFunctionEnd());
310   if (DD->useAppleExtensionAttributes() &&
311       !DD->getCurrentFunction()->getTarget().Options.DisableFramePointerElim(
312           *DD->getCurrentFunction()))
313     addFlag(*SPDie, dwarf::DW_AT_APPLE_omit_frame_ptr);
314 
315   // Only include DW_AT_frame_base in full debug info
316   if (!includeMinimalInlineScopes()) {
317     const TargetRegisterInfo *RI = Asm->MF->getSubtarget().getRegisterInfo();
318     MachineLocation Location(RI->getFrameRegister(*Asm->MF));
319     if (RI->isPhysicalRegister(Location.getReg()))
320       addAddress(*SPDie, dwarf::DW_AT_frame_base, Location);
321   }
322 
323   // Add name to the name table, we do this here because we're guaranteed
324   // to have concrete versions of our DW_TAG_subprogram nodes.
325   DD->addSubprogramNames(SP, *SPDie);
326 
327   return *SPDie;
328 }
329 
330 // Construct a DIE for this scope.
331 void DwarfCompileUnit::constructScopeDIE(
332     LexicalScope *Scope, SmallVectorImpl<DIE *> &FinalChildren) {
333   if (!Scope || !Scope->getScopeNode())
334     return;
335 
336   auto *DS = Scope->getScopeNode();
337 
338   assert((Scope->getInlinedAt() || !isa<DISubprogram>(DS)) &&
339          "Only handle inlined subprograms here, use "
340          "constructSubprogramScopeDIE for non-inlined "
341          "subprograms");
342 
343   SmallVector<DIE *, 8> Children;
344 
345   // We try to create the scope DIE first, then the children DIEs. This will
346   // avoid creating un-used children then removing them later when we find out
347   // the scope DIE is null.
348   DIE *ScopeDIE;
349   if (Scope->getParent() && isa<DISubprogram>(DS)) {
350     ScopeDIE = constructInlinedScopeDIE(Scope);
351     if (!ScopeDIE)
352       return;
353     // We create children when the scope DIE is not null.
354     createScopeChildrenDIE(Scope, Children);
355   } else {
356     // Early exit when we know the scope DIE is going to be null.
357     if (DD->isLexicalScopeDIENull(Scope))
358       return;
359 
360     bool HasNonScopeChildren = false;
361 
362     // We create children here when we know the scope DIE is not going to be
363     // null and the children will be added to the scope DIE.
364     createScopeChildrenDIE(Scope, Children, &HasNonScopeChildren);
365 
366     // If there are only other scopes as children, put them directly in the
367     // parent instead, as this scope would serve no purpose.
368     if (!HasNonScopeChildren) {
369       FinalChildren.insert(FinalChildren.end(),
370                            std::make_move_iterator(Children.begin()),
371                            std::make_move_iterator(Children.end()));
372       return;
373     }
374     ScopeDIE = constructLexicalScopeDIE(Scope);
375     assert(ScopeDIE && "Scope DIE should not be null.");
376   }
377 
378   // Add children
379   for (auto &I : Children)
380     ScopeDIE->addChild(std::move(I));
381 
382   FinalChildren.push_back(std::move(ScopeDIE));
383 }
384 
385 void DwarfCompileUnit::addScopeRangeList(DIE &ScopeDIE,
386                                          SmallVector<RangeSpan, 2> Range) {
387   const TargetLoweringObjectFile &TLOF = Asm->getObjFileLowering();
388 
389   // Emit offset in .debug_range as a relocatable label. emitDIE will handle
390   // emitting it appropriately.
391   const MCSymbol *RangeSectionSym =
392       TLOF.getDwarfRangesSection()->getBeginSymbol();
393 
394   RangeSpanList List(Asm->createTempSymbol("debug_ranges"), std::move(Range));
395 
396   // Under fission, ranges are specified by constant offsets relative to the
397   // CU's DW_AT_GNU_ranges_base.
398   if (isDwoUnit())
399     addSectionDelta(ScopeDIE, dwarf::DW_AT_ranges, List.getSym(),
400                     RangeSectionSym);
401   else
402     addSectionLabel(ScopeDIE, dwarf::DW_AT_ranges, List.getSym(),
403                     RangeSectionSym);
404 
405   // Add the range list to the set of ranges to be emitted.
406   (Skeleton ? Skeleton : this)->CURangeLists.push_back(std::move(List));
407 }
408 
409 void DwarfCompileUnit::attachRangesOrLowHighPC(
410     DIE &Die, SmallVector<RangeSpan, 2> Ranges) {
411   if (Ranges.size() == 1) {
412     const auto &single = Ranges.front();
413     attachLowHighPC(Die, single.getStart(), single.getEnd());
414   } else
415     addScopeRangeList(Die, std::move(Ranges));
416 }
417 
418 void DwarfCompileUnit::attachRangesOrLowHighPC(
419     DIE &Die, const SmallVectorImpl<InsnRange> &Ranges) {
420   SmallVector<RangeSpan, 2> List;
421   List.reserve(Ranges.size());
422   for (const InsnRange &R : Ranges)
423     List.push_back(RangeSpan(DD->getLabelBeforeInsn(R.first),
424                              DD->getLabelAfterInsn(R.second)));
425   attachRangesOrLowHighPC(Die, std::move(List));
426 }
427 
428 // This scope represents inlined body of a function. Construct DIE to
429 // represent this concrete inlined copy of the function.
430 DIE *DwarfCompileUnit::constructInlinedScopeDIE(LexicalScope *Scope) {
431   assert(Scope->getScopeNode());
432   auto *DS = Scope->getScopeNode();
433   auto *InlinedSP = getDISubprogram(DS);
434   // Find the subprogram's DwarfCompileUnit in the SPMap in case the subprogram
435   // was inlined from another compile unit.
436   DIE *OriginDIE = getAbstractSPDies()[InlinedSP];
437   assert(OriginDIE && "Unable to find original DIE for an inlined subprogram.");
438 
439   auto ScopeDIE = DIE::get(DIEValueAllocator, dwarf::DW_TAG_inlined_subroutine);
440   addDIEEntry(*ScopeDIE, dwarf::DW_AT_abstract_origin, *OriginDIE);
441 
442   attachRangesOrLowHighPC(*ScopeDIE, Scope->getRanges());
443 
444   // Add the call site information to the DIE.
445   const DILocation *IA = Scope->getInlinedAt();
446   addUInt(*ScopeDIE, dwarf::DW_AT_call_file, None,
447           getOrCreateSourceID(IA->getFile()));
448   addUInt(*ScopeDIE, dwarf::DW_AT_call_line, None, IA->getLine());
449   if (IA->getDiscriminator() && DD->getDwarfVersion() >= 4)
450     addUInt(*ScopeDIE, dwarf::DW_AT_GNU_discriminator, None,
451             IA->getDiscriminator());
452 
453   // Add name to the name table, we do this here because we're guaranteed
454   // to have concrete versions of our DW_TAG_inlined_subprogram nodes.
455   DD->addSubprogramNames(InlinedSP, *ScopeDIE);
456 
457   return ScopeDIE;
458 }
459 
460 // Construct new DW_TAG_lexical_block for this scope and attach
461 // DW_AT_low_pc/DW_AT_high_pc labels.
462 DIE *DwarfCompileUnit::constructLexicalScopeDIE(LexicalScope *Scope) {
463   if (DD->isLexicalScopeDIENull(Scope))
464     return nullptr;
465 
466   auto ScopeDIE = DIE::get(DIEValueAllocator, dwarf::DW_TAG_lexical_block);
467   if (Scope->isAbstractScope())
468     return ScopeDIE;
469 
470   attachRangesOrLowHighPC(*ScopeDIE, Scope->getRanges());
471 
472   return ScopeDIE;
473 }
474 
475 /// constructVariableDIE - Construct a DIE for the given DbgVariable.
476 DIE *DwarfCompileUnit::constructVariableDIE(DbgVariable &DV, bool Abstract) {
477   auto D = constructVariableDIEImpl(DV, Abstract);
478   DV.setDIE(*D);
479   return D;
480 }
481 
482 DIE *DwarfCompileUnit::constructVariableDIEImpl(const DbgVariable &DV,
483                                                 bool Abstract) {
484   // Define variable debug information entry.
485   auto VariableDie = DIE::get(DIEValueAllocator, DV.getTag());
486 
487   if (Abstract) {
488     applyVariableAttributes(DV, *VariableDie);
489     return VariableDie;
490   }
491 
492   // Add variable address.
493 
494   unsigned Offset = DV.getDebugLocListIndex();
495   if (Offset != ~0U) {
496     addLocationList(*VariableDie, dwarf::DW_AT_location, Offset);
497     return VariableDie;
498   }
499 
500   // Check if variable is described by a DBG_VALUE instruction.
501   if (const MachineInstr *DVInsn = DV.getMInsn()) {
502     assert(DVInsn->getNumOperands() == 4);
503     if (DVInsn->getOperand(0).isReg()) {
504       auto RegOp = DVInsn->getOperand(0);
505       auto Op1 = DVInsn->getOperand(1);
506       // If the second operand is an immediate, this is an indirect value.
507       assert((!Op1.isImm() || (Op1.getImm() == 0)) && "unexpected offset");
508       MachineLocation Location(RegOp.getReg(), Op1.isImm());
509       addVariableAddress(DV, *VariableDie, Location);
510     } else if (DVInsn->getOperand(0).isImm()) {
511       // This variable is described by a single constant.
512       // Check whether it has a DIExpression.
513       auto *Expr = DV.getSingleExpression();
514       if (Expr && Expr->getNumElements()) {
515         DIELoc *Loc = new (DIEValueAllocator) DIELoc;
516         DIEDwarfExpression DwarfExpr(*Asm, *this, *Loc);
517         // If there is an expression, emit raw unsigned bytes.
518         DwarfExpr.addFragmentOffset(Expr);
519         DwarfExpr.addUnsignedConstant(DVInsn->getOperand(0).getImm());
520         DwarfExpr.addExpression(Expr);
521         addBlock(*VariableDie, dwarf::DW_AT_location, DwarfExpr.finalize());
522       } else
523         addConstantValue(*VariableDie, DVInsn->getOperand(0), DV.getType());
524     } else if (DVInsn->getOperand(0).isFPImm())
525       addConstantFPValue(*VariableDie, DVInsn->getOperand(0));
526     else if (DVInsn->getOperand(0).isCImm())
527       addConstantValue(*VariableDie, DVInsn->getOperand(0).getCImm(),
528                        DV.getType());
529 
530     return VariableDie;
531   }
532 
533   // .. else use frame index.
534   if (!DV.hasFrameIndexExprs())
535     return VariableDie;
536 
537   DIELoc *Loc = new (DIEValueAllocator) DIELoc;
538   DIEDwarfExpression DwarfExpr(*Asm, *this, *Loc);
539   for (auto &Fragment : DV.getFrameIndexExprs()) {
540     unsigned FrameReg = 0;
541     const DIExpression *Expr = Fragment.Expr;
542     const TargetFrameLowering *TFI = Asm->MF->getSubtarget().getFrameLowering();
543     int Offset = TFI->getFrameIndexReference(*Asm->MF, Fragment.FI, FrameReg);
544     DwarfExpr.addFragmentOffset(Expr);
545     SmallVector<uint64_t, 8> Ops;
546     Ops.push_back(dwarf::DW_OP_plus_uconst);
547     Ops.push_back(Offset);
548     Ops.append(Expr->elements_begin(), Expr->elements_end());
549     DIExpressionCursor Cursor(Ops);
550     DwarfExpr.setMemoryLocationKind();
551     DwarfExpr.addMachineRegExpression(
552         *Asm->MF->getSubtarget().getRegisterInfo(), Cursor, FrameReg);
553     DwarfExpr.addExpression(std::move(Cursor));
554   }
555   addBlock(*VariableDie, dwarf::DW_AT_location, DwarfExpr.finalize());
556 
557   return VariableDie;
558 }
559 
560 DIE *DwarfCompileUnit::constructVariableDIE(DbgVariable &DV,
561                                             const LexicalScope &Scope,
562                                             DIE *&ObjectPointer) {
563   auto Var = constructVariableDIE(DV, Scope.isAbstractScope());
564   if (DV.isObjectPointer())
565     ObjectPointer = Var;
566   return Var;
567 }
568 
569 DIE *DwarfCompileUnit::createScopeChildrenDIE(LexicalScope *Scope,
570                                               SmallVectorImpl<DIE *> &Children,
571                                               bool *HasNonScopeChildren) {
572   assert(Children.empty());
573   DIE *ObjectPointer = nullptr;
574 
575   for (DbgVariable *DV : DU->getScopeVariables().lookup(Scope))
576     Children.push_back(constructVariableDIE(*DV, *Scope, ObjectPointer));
577 
578   // Skip imported directives in gmlt-like data.
579   if (!includeMinimalInlineScopes()) {
580     // There is no need to emit empty lexical block DIE.
581     for (const auto *IE : ImportedEntities[Scope->getScopeNode()])
582       Children.push_back(
583           constructImportedEntityDIE(cast<DIImportedEntity>(IE)));
584   }
585 
586   if (HasNonScopeChildren)
587     *HasNonScopeChildren = !Children.empty();
588 
589   for (LexicalScope *LS : Scope->getChildren())
590     constructScopeDIE(LS, Children);
591 
592   return ObjectPointer;
593 }
594 
595 void DwarfCompileUnit::constructSubprogramScopeDIE(const DISubprogram *Sub, LexicalScope *Scope) {
596   DIE &ScopeDIE = updateSubprogramScopeDIE(Sub);
597 
598   if (Scope) {
599     assert(!Scope->getInlinedAt());
600     assert(!Scope->isAbstractScope());
601     // Collect lexical scope children first.
602     // ObjectPointer might be a local (non-argument) local variable if it's a
603     // block's synthetic this pointer.
604     if (DIE *ObjectPointer = createAndAddScopeChildren(Scope, ScopeDIE))
605       addDIEEntry(ScopeDIE, dwarf::DW_AT_object_pointer, *ObjectPointer);
606   }
607 
608   // If this is a variadic function, add an unspecified parameter.
609   DITypeRefArray FnArgs = Sub->getType()->getTypeArray();
610 
611   // If we have a single element of null, it is a function that returns void.
612   // If we have more than one elements and the last one is null, it is a
613   // variadic function.
614   if (FnArgs.size() > 1 && !FnArgs[FnArgs.size() - 1] &&
615       !includeMinimalInlineScopes())
616     ScopeDIE.addChild(
617         DIE::get(DIEValueAllocator, dwarf::DW_TAG_unspecified_parameters));
618 }
619 
620 DIE *DwarfCompileUnit::createAndAddScopeChildren(LexicalScope *Scope,
621                                                  DIE &ScopeDIE) {
622   // We create children when the scope DIE is not null.
623   SmallVector<DIE *, 8> Children;
624   DIE *ObjectPointer = createScopeChildrenDIE(Scope, Children);
625 
626   // Add children
627   for (auto &I : Children)
628     ScopeDIE.addChild(std::move(I));
629 
630   return ObjectPointer;
631 }
632 
633 void DwarfCompileUnit::constructAbstractSubprogramScopeDIE(
634     LexicalScope *Scope) {
635   DIE *&AbsDef = getAbstractSPDies()[Scope->getScopeNode()];
636   if (AbsDef)
637     return;
638 
639   auto *SP = cast<DISubprogram>(Scope->getScopeNode());
640 
641   DIE *ContextDIE;
642   DwarfCompileUnit *ContextCU = this;
643 
644   if (includeMinimalInlineScopes())
645     ContextDIE = &getUnitDie();
646   // Some of this is duplicated from DwarfUnit::getOrCreateSubprogramDIE, with
647   // the important distinction that the debug node is not associated with the
648   // DIE (since the debug node will be associated with the concrete DIE, if
649   // any). It could be refactored to some common utility function.
650   else if (auto *SPDecl = SP->getDeclaration()) {
651     ContextDIE = &getUnitDie();
652     getOrCreateSubprogramDIE(SPDecl);
653   } else {
654     ContextDIE = getOrCreateContextDIE(resolve(SP->getScope()));
655     // The scope may be shared with a subprogram that has already been
656     // constructed in another CU, in which case we need to construct this
657     // subprogram in the same CU.
658     ContextCU = DD->lookupCU(ContextDIE->getUnitDie());
659   }
660 
661   // Passing null as the associated node because the abstract definition
662   // shouldn't be found by lookup.
663   AbsDef = &ContextCU->createAndAddDIE(dwarf::DW_TAG_subprogram, *ContextDIE, nullptr);
664   ContextCU->applySubprogramAttributesToDefinition(SP, *AbsDef);
665 
666   if (!ContextCU->includeMinimalInlineScopes())
667     ContextCU->addUInt(*AbsDef, dwarf::DW_AT_inline, None, dwarf::DW_INL_inlined);
668   if (DIE *ObjectPointer = ContextCU->createAndAddScopeChildren(Scope, *AbsDef))
669     ContextCU->addDIEEntry(*AbsDef, dwarf::DW_AT_object_pointer, *ObjectPointer);
670 }
671 
672 DIE *DwarfCompileUnit::constructImportedEntityDIE(
673     const DIImportedEntity *Module) {
674   DIE *IMDie = DIE::get(DIEValueAllocator, (dwarf::Tag)Module->getTag());
675   insertDIE(Module, IMDie);
676   DIE *EntityDie;
677   auto *Entity = resolve(Module->getEntity());
678   if (auto *NS = dyn_cast<DINamespace>(Entity))
679     EntityDie = getOrCreateNameSpace(NS);
680   else if (auto *M = dyn_cast<DIModule>(Entity))
681     EntityDie = getOrCreateModule(M);
682   else if (auto *SP = dyn_cast<DISubprogram>(Entity))
683     EntityDie = getOrCreateSubprogramDIE(SP);
684   else if (auto *T = dyn_cast<DIType>(Entity))
685     EntityDie = getOrCreateTypeDIE(T);
686   else if (auto *GV = dyn_cast<DIGlobalVariable>(Entity))
687     EntityDie = getOrCreateGlobalVariableDIE(GV, {});
688   else
689     EntityDie = getDIE(Entity);
690   assert(EntityDie);
691   addSourceLine(*IMDie, Module->getLine(), Module->getFile());
692   addDIEEntry(*IMDie, dwarf::DW_AT_import, *EntityDie);
693   StringRef Name = Module->getName();
694   if (!Name.empty())
695     addString(*IMDie, dwarf::DW_AT_name, Name);
696 
697   return IMDie;
698 }
699 
700 void DwarfCompileUnit::finishSubprogramDefinition(const DISubprogram *SP) {
701   DIE *D = getDIE(SP);
702   if (DIE *AbsSPDIE = getAbstractSPDies().lookup(SP)) {
703     if (D)
704       // If this subprogram has an abstract definition, reference that
705       addDIEEntry(*D, dwarf::DW_AT_abstract_origin, *AbsSPDIE);
706   } else {
707     assert(D || includeMinimalInlineScopes());
708     if (D)
709       // And attach the attributes
710       applySubprogramAttributesToDefinition(SP, *D);
711   }
712 }
713 
714 void DwarfCompileUnit::finishVariableDefinition(const DbgVariable &Var) {
715   DbgVariable *AbsVar = getExistingAbstractVariable(
716       InlinedVariable(Var.getVariable(), Var.getInlinedAt()));
717   auto *VariableDie = Var.getDIE();
718   if (AbsVar && AbsVar->getDIE()) {
719     addDIEEntry(*VariableDie, dwarf::DW_AT_abstract_origin,
720                       *AbsVar->getDIE());
721   } else
722     applyVariableAttributes(Var, *VariableDie);
723 }
724 
725 DbgVariable *DwarfCompileUnit::getExistingAbstractVariable(InlinedVariable IV) {
726   const DILocalVariable *Cleansed;
727   return getExistingAbstractVariable(IV, Cleansed);
728 }
729 
730 // Find abstract variable, if any, associated with Var.
731 DbgVariable *DwarfCompileUnit::getExistingAbstractVariable(
732     InlinedVariable IV, const DILocalVariable *&Cleansed) {
733   // More then one inlined variable corresponds to one abstract variable.
734   Cleansed = IV.first;
735   auto &AbstractVariables = getAbstractVariables();
736   auto I = AbstractVariables.find(Cleansed);
737   if (I != AbstractVariables.end())
738     return I->second.get();
739   return nullptr;
740 }
741 
742 void DwarfCompileUnit::createAbstractVariable(const DILocalVariable *Var,
743                                         LexicalScope *Scope) {
744   assert(Scope && Scope->isAbstractScope());
745   auto AbsDbgVariable = llvm::make_unique<DbgVariable>(Var, /* IA */ nullptr);
746   DU->addScopeVariable(Scope, AbsDbgVariable.get());
747   getAbstractVariables()[Var] = std::move(AbsDbgVariable);
748 }
749 
750 void DwarfCompileUnit::emitHeader(bool UseOffsets) {
751   // Don't bother labeling the .dwo unit, as its offset isn't used.
752   if (!Skeleton) {
753     LabelBegin = Asm->createTempSymbol("cu_begin");
754     Asm->OutStreamer->EmitLabel(LabelBegin);
755   }
756 
757   dwarf::UnitType UT = Skeleton ? dwarf::DW_UT_split_compile
758                                 : DD->useSplitDwarf() ? dwarf::DW_UT_skeleton
759                                                       : dwarf::DW_UT_compile;
760   DwarfUnit::emitCommonHeader(UseOffsets, UT);
761 }
762 
763 bool DwarfCompileUnit::hasDwarfPubSections() const {
764   // Opting in to GNU Pubnames/types overrides the default to ensure these are
765   // generated for things like Gold's gdb_index generation.
766   if (CUNode->getGnuPubnames())
767     return true;
768 
769   return DD->tuneForGDB() && !includeMinimalInlineScopes();
770 }
771 
772 /// addGlobalName - Add a new global name to the compile unit.
773 void DwarfCompileUnit::addGlobalName(StringRef Name, const DIE &Die,
774                                      const DIScope *Context) {
775   if (!hasDwarfPubSections())
776     return;
777   std::string FullName = getParentContextString(Context) + Name.str();
778   GlobalNames[FullName] = &Die;
779 }
780 
781 void DwarfCompileUnit::addGlobalNameForTypeUnit(StringRef Name,
782                                                 const DIScope *Context) {
783   if (!hasDwarfPubSections())
784     return;
785   std::string FullName = getParentContextString(Context) + Name.str();
786   // Insert, allowing the entry to remain as-is if it's already present
787   // This way the CU-level type DIE is preferred over the "can't describe this
788   // type as a unit offset because it's not really in the CU at all, it's only
789   // in a type unit"
790   GlobalNames.insert(std::make_pair(std::move(FullName), &getUnitDie()));
791 }
792 
793 /// Add a new global type to the unit.
794 void DwarfCompileUnit::addGlobalType(const DIType *Ty, const DIE &Die,
795                                      const DIScope *Context) {
796   if (!hasDwarfPubSections())
797     return;
798   std::string FullName = getParentContextString(Context) + Ty->getName().str();
799   GlobalTypes[FullName] = &Die;
800 }
801 
802 void DwarfCompileUnit::addGlobalTypeUnitType(const DIType *Ty,
803                                              const DIScope *Context) {
804   if (!hasDwarfPubSections())
805     return;
806   std::string FullName = getParentContextString(Context) + Ty->getName().str();
807   // Insert, allowing the entry to remain as-is if it's already present
808   // This way the CU-level type DIE is preferred over the "can't describe this
809   // type as a unit offset because it's not really in the CU at all, it's only
810   // in a type unit"
811   GlobalTypes.insert(std::make_pair(std::move(FullName), &getUnitDie()));
812 }
813 
814 /// addVariableAddress - Add DW_AT_location attribute for a
815 /// DbgVariable based on provided MachineLocation.
816 void DwarfCompileUnit::addVariableAddress(const DbgVariable &DV, DIE &Die,
817                                           MachineLocation Location) {
818   // addBlockByrefAddress is obsolete and will be removed soon.
819   // The clang frontend always generates block byref variables with a
820   // complex expression that encodes exactly what addBlockByrefAddress
821   // would do.
822   assert((!DV.isBlockByrefVariable() || DV.hasComplexAddress()) &&
823          "block byref variable without a complex expression");
824   if (DV.hasComplexAddress())
825     addComplexAddress(DV, Die, dwarf::DW_AT_location, Location);
826   else if (DV.isBlockByrefVariable())
827     addBlockByrefAddress(DV, Die, dwarf::DW_AT_location, Location);
828   else
829     addAddress(Die, dwarf::DW_AT_location, Location);
830 }
831 
832 /// Add an address attribute to a die based on the location provided.
833 void DwarfCompileUnit::addAddress(DIE &Die, dwarf::Attribute Attribute,
834                                   const MachineLocation &Location) {
835   DIELoc *Loc = new (DIEValueAllocator) DIELoc;
836   DIEDwarfExpression DwarfExpr(*Asm, *this, *Loc);
837   if (Location.isIndirect())
838     DwarfExpr.setMemoryLocationKind();
839 
840   DIExpressionCursor Cursor({});
841   const TargetRegisterInfo &TRI = *Asm->MF->getSubtarget().getRegisterInfo();
842   if (!DwarfExpr.addMachineRegExpression(TRI, Cursor, Location.getReg()))
843     return;
844   DwarfExpr.addExpression(std::move(Cursor));
845 
846   // Now attach the location information to the DIE.
847   addBlock(Die, Attribute, DwarfExpr.finalize());
848 }
849 
850 /// Start with the address based on the location provided, and generate the
851 /// DWARF information necessary to find the actual variable given the extra
852 /// address information encoded in the DbgVariable, starting from the starting
853 /// location.  Add the DWARF information to the die.
854 void DwarfCompileUnit::addComplexAddress(const DbgVariable &DV, DIE &Die,
855                                          dwarf::Attribute Attribute,
856                                          const MachineLocation &Location) {
857   DIELoc *Loc = new (DIEValueAllocator) DIELoc;
858   DIEDwarfExpression DwarfExpr(*Asm, *this, *Loc);
859   const DIExpression *DIExpr = DV.getSingleExpression();
860   DwarfExpr.addFragmentOffset(DIExpr);
861   if (Location.isIndirect())
862     DwarfExpr.setMemoryLocationKind();
863 
864   DIExpressionCursor Cursor(DIExpr);
865   const TargetRegisterInfo &TRI = *Asm->MF->getSubtarget().getRegisterInfo();
866   if (!DwarfExpr.addMachineRegExpression(TRI, Cursor, Location.getReg()))
867     return;
868   DwarfExpr.addExpression(std::move(Cursor));
869 
870   // Now attach the location information to the DIE.
871   addBlock(Die, Attribute, DwarfExpr.finalize());
872 }
873 
874 /// Add a Dwarf loclistptr attribute data and value.
875 void DwarfCompileUnit::addLocationList(DIE &Die, dwarf::Attribute Attribute,
876                                        unsigned Index) {
877   dwarf::Form Form = DD->getDwarfVersion() >= 4 ? dwarf::DW_FORM_sec_offset
878                                                 : dwarf::DW_FORM_data4;
879   Die.addValue(DIEValueAllocator, Attribute, Form, DIELocList(Index));
880 }
881 
882 void DwarfCompileUnit::applyVariableAttributes(const DbgVariable &Var,
883                                                DIE &VariableDie) {
884   StringRef Name = Var.getName();
885   if (!Name.empty())
886     addString(VariableDie, dwarf::DW_AT_name, Name);
887   const auto *DIVar = Var.getVariable();
888   if (DIVar)
889     if (uint32_t AlignInBytes = DIVar->getAlignInBytes())
890       addUInt(VariableDie, dwarf::DW_AT_alignment, dwarf::DW_FORM_udata,
891               AlignInBytes);
892 
893   addSourceLine(VariableDie, DIVar);
894   addType(VariableDie, Var.getType());
895   if (Var.isArtificial())
896     addFlag(VariableDie, dwarf::DW_AT_artificial);
897 }
898 
899 /// Add a Dwarf expression attribute data and value.
900 void DwarfCompileUnit::addExpr(DIELoc &Die, dwarf::Form Form,
901                                const MCExpr *Expr) {
902   Die.addValue(DIEValueAllocator, (dwarf::Attribute)0, Form, DIEExpr(Expr));
903 }
904 
905 void DwarfCompileUnit::applySubprogramAttributesToDefinition(
906     const DISubprogram *SP, DIE &SPDie) {
907   auto *SPDecl = SP->getDeclaration();
908   auto *Context = resolve(SPDecl ? SPDecl->getScope() : SP->getScope());
909   applySubprogramAttributes(SP, SPDie, includeMinimalInlineScopes());
910   addGlobalName(SP->getName(), SPDie, Context);
911 }
912 
913 bool DwarfCompileUnit::isDwoUnit() const {
914   return DD->useSplitDwarf() && Skeleton;
915 }
916 
917 bool DwarfCompileUnit::includeMinimalInlineScopes() const {
918   return getCUNode()->getEmissionKind() == DICompileUnit::LineTablesOnly ||
919          (DD->useSplitDwarf() && !Skeleton);
920 }
921