257190446ed6e04e4af56326eb9f4274d866b596
[oota-llvm.git] / tools / dsymutil / DwarfLinker.cpp
1 //===- tools/dsymutil/DwarfLinker.cpp - Dwarf debug info linker -----------===//
2 //
3 //                             The LLVM Linker
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 #include "DebugMap.h"
10 #include "BinaryHolder.h"
11 #include "DebugMap.h"
12 #include "dsymutil.h"
13 #include "MachOUtils.h"
14 #include "NonRelocatableStringpool.h"
15 #include "llvm/ADT/IntervalMap.h"
16 #include "llvm/ADT/StringMap.h"
17 #include "llvm/ADT/STLExtras.h"
18 #include "llvm/CodeGen/AsmPrinter.h"
19 #include "llvm/CodeGen/DIE.h"
20 #include "llvm/Config/config.h"
21 #include "llvm/DebugInfo/DWARF/DWARFContext.h"
22 #include "llvm/DebugInfo/DWARF/DWARFDebugInfoEntry.h"
23 #include "llvm/DebugInfo/DWARF/DWARFFormValue.h"
24 #include "llvm/MC/MCAsmBackend.h"
25 #include "llvm/MC/MCAsmInfo.h"
26 #include "llvm/MC/MCContext.h"
27 #include "llvm/MC/MCCodeEmitter.h"
28 #include "llvm/MC/MCDwarf.h"
29 #include "llvm/MC/MCInstrInfo.h"
30 #include "llvm/MC/MCObjectFileInfo.h"
31 #include "llvm/MC/MCRegisterInfo.h"
32 #include "llvm/MC/MCStreamer.h"
33 #include "llvm/MC/MCSubtargetInfo.h"
34 #include "llvm/Object/MachO.h"
35 #include "llvm/Support/Dwarf.h"
36 #include "llvm/Support/LEB128.h"
37 #include "llvm/Support/TargetRegistry.h"
38 #include "llvm/Target/TargetMachine.h"
39 #include "llvm/Target/TargetOptions.h"
40 #include <string>
41 #include <tuple>
42
43 namespace llvm {
44 namespace dsymutil {
45
46 namespace {
47
48 template <typename KeyT, typename ValT>
49 using HalfOpenIntervalMap =
50     IntervalMap<KeyT, ValT, IntervalMapImpl::NodeSizer<KeyT, ValT>::LeafSize,
51                 IntervalMapHalfOpenInfo<KeyT>>;
52
53 typedef HalfOpenIntervalMap<uint64_t, int64_t> FunctionIntervals;
54
55 // FIXME: Delete this structure.
56 struct PatchLocation {
57   DIE::value_iterator I;
58
59   PatchLocation() = default;
60   PatchLocation(DIE::value_iterator I) : I(I) {}
61
62   void set(uint64_t New) const {
63     assert(I);
64     const auto &Old = *I;
65     assert(Old.getType() == DIEValue::isInteger);
66     *I = DIEValue(Old.getAttribute(), Old.getForm(), DIEInteger(New));
67   }
68
69   uint64_t get() const {
70     assert(I);
71     return I->getDIEInteger().getValue();
72   }
73 };
74
75 class CompileUnit;
76 struct DeclMapInfo;
77
78 /// A DeclContext is a named program scope that is used for ODR
79 /// uniquing of types.
80 /// The set of DeclContext for the ODR-subject parts of a Dwarf link
81 /// is expanded (and uniqued) with each new object file processed. We
82 /// need to determine the context of each DIE in an linked object file
83 /// to see if the corresponding type has already been emitted.
84 ///
85 /// The contexts are conceptually organised as a tree (eg. a function
86 /// scope is contained in a namespace scope that contains other
87 /// scopes), but storing/accessing them in an actual tree is too
88 /// inefficient: we need to be able to very quickly query a context
89 /// for a given child context by name. Storing a StringMap in each
90 /// DeclContext would be too space inefficient.
91 /// The solution here is to give each DeclContext a link to its parent
92 /// (this allows to walk up the tree), but to query the existance of a
93 /// specific DeclContext using a separate DenseMap keyed on the hash
94 /// of the fully qualified name of the context.
95 class DeclContext {
96   unsigned QualifiedNameHash;
97   uint32_t Line;
98   uint32_t ByteSize;
99   uint16_t Tag;
100   StringRef Name;
101   StringRef File;
102   const DeclContext &Parent;
103   const DWARFDebugInfoEntryMinimal *LastSeenDIE;
104   uint32_t LastSeenCompileUnitID;
105   uint32_t CanonicalDIEOffset;
106
107   friend DeclMapInfo;
108
109 public:
110   typedef DenseSet<DeclContext *, DeclMapInfo> Map;
111
112   DeclContext()
113       : QualifiedNameHash(0), Line(0), ByteSize(0),
114         Tag(dwarf::DW_TAG_compile_unit), Name(), File(), Parent(*this),
115         LastSeenDIE(nullptr), LastSeenCompileUnitID(0), CanonicalDIEOffset(0) {}
116
117   DeclContext(unsigned Hash, uint32_t Line, uint32_t ByteSize, uint16_t Tag,
118               StringRef Name, StringRef File, const DeclContext &Parent,
119               const DWARFDebugInfoEntryMinimal *LastSeenDIE = nullptr,
120               unsigned CUId = 0)
121       : QualifiedNameHash(Hash), Line(Line), ByteSize(ByteSize), Tag(Tag),
122         Name(Name), File(File), Parent(Parent), LastSeenDIE(LastSeenDIE),
123         LastSeenCompileUnitID(CUId), CanonicalDIEOffset(0) {}
124
125   uint32_t getQualifiedNameHash() const { return QualifiedNameHash; }
126
127   bool setLastSeenDIE(CompileUnit &U, const DWARFDebugInfoEntryMinimal *Die);
128
129   uint32_t getCanonicalDIEOffset() const { return CanonicalDIEOffset; }
130   void setCanonicalDIEOffset(uint32_t Offset) { CanonicalDIEOffset = Offset; }
131
132   uint16_t getTag() const { return Tag; }
133   StringRef getName() const { return Name; }
134 };
135
136 /// Info type for the DenseMap storing the DeclContext pointers.
137 struct DeclMapInfo : private DenseMapInfo<DeclContext *> {
138   using DenseMapInfo<DeclContext *>::getEmptyKey;
139   using DenseMapInfo<DeclContext *>::getTombstoneKey;
140
141   static unsigned getHashValue(const DeclContext *Ctxt) {
142     return Ctxt->QualifiedNameHash;
143   }
144
145   static bool isEqual(const DeclContext *LHS, const DeclContext *RHS) {
146     if (RHS == getEmptyKey() || RHS == getTombstoneKey())
147       return RHS == LHS;
148     return LHS->QualifiedNameHash == RHS->QualifiedNameHash &&
149            LHS->Line == RHS->Line && LHS->ByteSize == RHS->ByteSize &&
150            LHS->Name.data() == RHS->Name.data() &&
151            LHS->File.data() == RHS->File.data() &&
152            LHS->Parent.QualifiedNameHash == RHS->Parent.QualifiedNameHash;
153   }
154 };
155
156 /// This class gives a tree-like API to the DenseMap that stores the
157 /// DeclContext objects. It also holds the BumpPtrAllocator where
158 /// these objects will be allocated.
159 class DeclContextTree {
160   BumpPtrAllocator Allocator;
161   DeclContext Root;
162   DeclContext::Map Contexts;
163
164 public:
165   /// Get the child of \a Context described by \a DIE in \a Unit. The
166   /// required strings will be interned in \a StringPool.
167   /// \returns The child DeclContext along with one bit that is set if
168   /// this context is invalid.
169   /// FIXME: the invalid bit along the return value is to emulate some
170   /// dsymutil-classic functionality. See the fucntion definition for
171   /// a more thorough discussion of its use.
172   PointerIntPair<DeclContext *, 1>
173   getChildDeclContext(DeclContext &Context,
174                       const DWARFDebugInfoEntryMinimal *DIE, CompileUnit &Unit,
175                       NonRelocatableStringpool &StringPool);
176
177   DeclContext &getRoot() { return Root; }
178 };
179
180 /// \brief Stores all information relating to a compile unit, be it in
181 /// its original instance in the object file to its brand new cloned
182 /// and linked DIE tree.
183 class CompileUnit {
184 public:
185   /// \brief Information gathered about a DIE in the object file.
186   struct DIEInfo {
187     int64_t AddrAdjust; ///< Address offset to apply to the described entity.
188     DeclContext *Ctxt;  ///< ODR Declaration context.
189     DIE *Clone;         ///< Cloned version of that DIE.
190     uint32_t ParentIdx; ///< The index of this DIE's parent.
191     bool Keep;          ///< Is the DIE part of the linked output?
192     bool InDebugMap;    ///< Was this DIE's entity found in the map?
193   };
194
195   CompileUnit(DWARFUnit &OrigUnit, unsigned ID, bool CanUseODR)
196       : OrigUnit(OrigUnit), ID(ID), LowPc(UINT64_MAX), HighPc(0), RangeAlloc(),
197         Ranges(RangeAlloc) {
198     Info.resize(OrigUnit.getNumDIEs());
199
200     const auto *CUDie = OrigUnit.getUnitDIE(false);
201     unsigned Lang = CUDie->getAttributeValueAsUnsignedConstant(
202         &OrigUnit, dwarf::DW_AT_language, 0);
203     HasODR = CanUseODR && (Lang == dwarf::DW_LANG_C_plus_plus ||
204                            Lang == dwarf::DW_LANG_C_plus_plus_03 ||
205                            Lang == dwarf::DW_LANG_C_plus_plus_11 ||
206                            Lang == dwarf::DW_LANG_C_plus_plus_14 ||
207                            Lang == dwarf::DW_LANG_ObjC_plus_plus);
208   }
209
210   CompileUnit(CompileUnit &&RHS)
211       : OrigUnit(RHS.OrigUnit), Info(std::move(RHS.Info)),
212         CUDie(std::move(RHS.CUDie)), StartOffset(RHS.StartOffset),
213         NextUnitOffset(RHS.NextUnitOffset), RangeAlloc(), Ranges(RangeAlloc) {
214     // The CompileUnit container has been 'reserve()'d with the right
215     // size. We cannot move the IntervalMap anyway.
216     llvm_unreachable("CompileUnits should not be moved.");
217   }
218
219   DWARFUnit &getOrigUnit() const { return OrigUnit; }
220
221   unsigned getUniqueID() const { return ID; }
222
223   DIE *getOutputUnitDIE() const { return CUDie; }
224   void setOutputUnitDIE(DIE *Die) { CUDie = Die; }
225
226   bool hasODR() const { return HasODR; }
227
228   DIEInfo &getInfo(unsigned Idx) { return Info[Idx]; }
229   const DIEInfo &getInfo(unsigned Idx) const { return Info[Idx]; }
230
231   uint64_t getStartOffset() const { return StartOffset; }
232   uint64_t getNextUnitOffset() const { return NextUnitOffset; }
233   void setStartOffset(uint64_t DebugInfoSize) { StartOffset = DebugInfoSize; }
234
235   uint64_t getLowPc() const { return LowPc; }
236   uint64_t getHighPc() const { return HighPc; }
237
238   Optional<PatchLocation> getUnitRangesAttribute() const {
239     return UnitRangeAttribute;
240   }
241   const FunctionIntervals &getFunctionRanges() const { return Ranges; }
242   const std::vector<PatchLocation> &getRangesAttributes() const {
243     return RangeAttributes;
244   }
245
246   const std::vector<std::pair<PatchLocation, int64_t>> &
247   getLocationAttributes() const {
248     return LocationAttributes;
249   }
250
251   /// \brief Compute the end offset for this unit. Must be
252   /// called after the CU's DIEs have been cloned.
253   /// \returns the next unit offset (which is also the current
254   /// debug_info section size).
255   uint64_t computeNextUnitOffset();
256
257   /// \brief Keep track of a forward reference to DIE \p Die in \p
258   /// RefUnit by \p Attr. The attribute should be fixed up later to
259   /// point to the absolute offset of \p Die in the debug_info section
260   /// or to the canonical offset of \p Ctxt if it is non-null.
261   void noteForwardReference(DIE *Die, const CompileUnit *RefUnit,
262                             DeclContext *Ctxt, PatchLocation Attr);
263
264   /// \brief Apply all fixups recored by noteForwardReference().
265   void fixupForwardReferences();
266
267   /// \brief Add a function range [\p LowPC, \p HighPC) that is
268   /// relocatad by applying offset \p PCOffset.
269   void addFunctionRange(uint64_t LowPC, uint64_t HighPC, int64_t PCOffset);
270
271   /// \brief Keep track of a DW_AT_range attribute that we will need to
272   /// patch up later.
273   void noteRangeAttribute(const DIE &Die, PatchLocation Attr);
274
275   /// \brief Keep track of a location attribute pointing to a location
276   /// list in the debug_loc section.
277   void noteLocationAttribute(PatchLocation Attr, int64_t PcOffset);
278
279   /// \brief Add a name accelerator entry for \p Die with \p Name
280   /// which is stored in the string table at \p Offset.
281   void addNameAccelerator(const DIE *Die, const char *Name, uint32_t Offset,
282                           bool SkipPubnamesSection = false);
283
284   /// \brief Add a type accelerator entry for \p Die with \p Name
285   /// which is stored in the string table at \p Offset.
286   void addTypeAccelerator(const DIE *Die, const char *Name, uint32_t Offset);
287
288   struct AccelInfo {
289     StringRef Name;      ///< Name of the entry.
290     const DIE *Die;      ///< DIE this entry describes.
291     uint32_t NameOffset; ///< Offset of Name in the string pool.
292     bool SkipPubSection; ///< Emit this entry only in the apple_* sections.
293
294     AccelInfo(StringRef Name, const DIE *Die, uint32_t NameOffset,
295               bool SkipPubSection = false)
296         : Name(Name), Die(Die), NameOffset(NameOffset),
297           SkipPubSection(SkipPubSection) {}
298   };
299
300   const std::vector<AccelInfo> &getPubnames() const { return Pubnames; }
301   const std::vector<AccelInfo> &getPubtypes() const { return Pubtypes; }
302
303   /// Get the full path for file \a FileNum in the line table
304   const char *getResolvedPath(unsigned FileNum) {
305     if (FileNum >= ResolvedPaths.size())
306       return nullptr;
307     return ResolvedPaths[FileNum].size() ? ResolvedPaths[FileNum].c_str()
308                                          : nullptr;
309   }
310
311   /// Set the fully resolved path for the line-table's file \a FileNum
312   /// to \a Path.
313   void setResolvedPath(unsigned FileNum, const std::string &Path) {
314     if (ResolvedPaths.size() <= FileNum)
315       ResolvedPaths.resize(FileNum + 1);
316     ResolvedPaths[FileNum] = Path;
317   }
318
319 private:
320   DWARFUnit &OrigUnit;
321   unsigned ID;
322   std::vector<DIEInfo> Info; ///< DIE info indexed by DIE index.
323   DIE *CUDie;                ///< Root of the linked DIE tree.
324
325   uint64_t StartOffset;
326   uint64_t NextUnitOffset;
327
328   uint64_t LowPc;
329   uint64_t HighPc;
330
331   /// \brief A list of attributes to fixup with the absolute offset of
332   /// a DIE in the debug_info section.
333   ///
334   /// The offsets for the attributes in this array couldn't be set while
335   /// cloning because for cross-cu forward refences the target DIE's
336   /// offset isn't known you emit the reference attribute.
337   std::vector<std::tuple<DIE *, const CompileUnit *, DeclContext *,
338                          PatchLocation>> ForwardDIEReferences;
339
340   FunctionIntervals::Allocator RangeAlloc;
341   /// \brief The ranges in that interval map are the PC ranges for
342   /// functions in this unit, associated with the PC offset to apply
343   /// to the addresses to get the linked address.
344   FunctionIntervals Ranges;
345
346   /// \brief DW_AT_ranges attributes to patch after we have gathered
347   /// all the unit's function addresses.
348   /// @{
349   std::vector<PatchLocation> RangeAttributes;
350   Optional<PatchLocation> UnitRangeAttribute;
351   /// @}
352
353   /// \brief Location attributes that need to be transfered from th
354   /// original debug_loc section to the liked one. They are stored
355   /// along with the PC offset that is to be applied to their
356   /// function's address.
357   std::vector<std::pair<PatchLocation, int64_t>> LocationAttributes;
358
359   /// \brief Accelerator entries for the unit, both for the pub*
360   /// sections and the apple* ones.
361   /// @{
362   std::vector<AccelInfo> Pubnames;
363   std::vector<AccelInfo> Pubtypes;
364   /// @}
365
366   /// Cached resolved paths from the line table.
367   std::vector<std::string> ResolvedPaths;
368
369   /// Is this unit subject to the ODR rule?
370   bool HasODR;
371 };
372
373 uint64_t CompileUnit::computeNextUnitOffset() {
374   NextUnitOffset = StartOffset + 11 /* Header size */;
375   // The root DIE might be null, meaning that the Unit had nothing to
376   // contribute to the linked output. In that case, we will emit the
377   // unit header without any actual DIE.
378   if (CUDie)
379     NextUnitOffset += CUDie->getSize();
380   return NextUnitOffset;
381 }
382
383 /// \brief Keep track of a forward cross-cu reference from this unit
384 /// to \p Die that lives in \p RefUnit.
385 void CompileUnit::noteForwardReference(DIE *Die, const CompileUnit *RefUnit,
386                                        DeclContext *Ctxt, PatchLocation Attr) {
387   ForwardDIEReferences.emplace_back(Die, RefUnit, Ctxt, Attr);
388 }
389
390 /// \brief Apply all fixups recorded by noteForwardReference().
391 void CompileUnit::fixupForwardReferences() {
392   for (const auto &Ref : ForwardDIEReferences) {
393     DIE *RefDie;
394     const CompileUnit *RefUnit;
395     PatchLocation Attr;
396     DeclContext *Ctxt;
397     std::tie(RefDie, RefUnit, Ctxt, Attr) = Ref;
398     if (Ctxt && Ctxt->getCanonicalDIEOffset())
399       Attr.set(Ctxt->getCanonicalDIEOffset());
400     else
401       Attr.set(RefDie->getOffset() + RefUnit->getStartOffset());
402   }
403 }
404
405 void CompileUnit::addFunctionRange(uint64_t FuncLowPc, uint64_t FuncHighPc,
406                                    int64_t PcOffset) {
407   Ranges.insert(FuncLowPc, FuncHighPc, PcOffset);
408   this->LowPc = std::min(LowPc, FuncLowPc + PcOffset);
409   this->HighPc = std::max(HighPc, FuncHighPc + PcOffset);
410 }
411
412 void CompileUnit::noteRangeAttribute(const DIE &Die, PatchLocation Attr) {
413   if (Die.getTag() != dwarf::DW_TAG_compile_unit)
414     RangeAttributes.push_back(Attr);
415   else
416     UnitRangeAttribute = Attr;
417 }
418
419 void CompileUnit::noteLocationAttribute(PatchLocation Attr, int64_t PcOffset) {
420   LocationAttributes.emplace_back(Attr, PcOffset);
421 }
422
423 /// \brief Add a name accelerator entry for \p Die with \p Name
424 /// which is stored in the string table at \p Offset.
425 void CompileUnit::addNameAccelerator(const DIE *Die, const char *Name,
426                                      uint32_t Offset, bool SkipPubSection) {
427   Pubnames.emplace_back(Name, Die, Offset, SkipPubSection);
428 }
429
430 /// \brief Add a type accelerator entry for \p Die with \p Name
431 /// which is stored in the string table at \p Offset.
432 void CompileUnit::addTypeAccelerator(const DIE *Die, const char *Name,
433                                      uint32_t Offset) {
434   Pubtypes.emplace_back(Name, Die, Offset, false);
435 }
436
437 /// \brief The Dwarf streaming logic
438 ///
439 /// All interactions with the MC layer that is used to build the debug
440 /// information binary representation are handled in this class.
441 class DwarfStreamer {
442   /// \defgroup MCObjects MC layer objects constructed by the streamer
443   /// @{
444   std::unique_ptr<MCRegisterInfo> MRI;
445   std::unique_ptr<MCAsmInfo> MAI;
446   std::unique_ptr<MCObjectFileInfo> MOFI;
447   std::unique_ptr<MCContext> MC;
448   MCAsmBackend *MAB; // Owned by MCStreamer
449   std::unique_ptr<MCInstrInfo> MII;
450   std::unique_ptr<MCSubtargetInfo> MSTI;
451   MCCodeEmitter *MCE; // Owned by MCStreamer
452   MCStreamer *MS;     // Owned by AsmPrinter
453   std::unique_ptr<TargetMachine> TM;
454   std::unique_ptr<AsmPrinter> Asm;
455   /// @}
456
457   /// \brief the file we stream the linked Dwarf to.
458   std::unique_ptr<raw_fd_ostream> OutFile;
459
460   uint32_t RangesSectionSize;
461   uint32_t LocSectionSize;
462   uint32_t LineSectionSize;
463   uint32_t FrameSectionSize;
464
465   /// \brief Emit the pubnames or pubtypes section contribution for \p
466   /// Unit into \p Sec. The data is provided in \p Names.
467   void emitPubSectionForUnit(MCSection *Sec, StringRef Name,
468                              const CompileUnit &Unit,
469                              const std::vector<CompileUnit::AccelInfo> &Names);
470
471 public:
472   /// \brief Actually create the streamer and the ouptut file.
473   ///
474   /// This could be done directly in the constructor, but it feels
475   /// more natural to handle errors through return value.
476   bool init(Triple TheTriple, StringRef OutputFilename);
477
478   /// \brief Dump the file to the disk.
479   bool finish(const DebugMap &);
480
481   AsmPrinter &getAsmPrinter() const { return *Asm; }
482
483   /// \brief Set the current output section to debug_info and change
484   /// the MC Dwarf version to \p DwarfVersion.
485   void switchToDebugInfoSection(unsigned DwarfVersion);
486
487   /// \brief Emit the compilation unit header for \p Unit in the
488   /// debug_info section.
489   ///
490   /// As a side effect, this also switches the current Dwarf version
491   /// of the MC layer to the one of U.getOrigUnit().
492   void emitCompileUnitHeader(CompileUnit &Unit);
493
494   /// \brief Recursively emit the DIE tree rooted at \p Die.
495   void emitDIE(DIE &Die);
496
497   /// \brief Emit the abbreviation table \p Abbrevs to the
498   /// debug_abbrev section.
499   void emitAbbrevs(const std::vector<DIEAbbrev *> &Abbrevs);
500
501   /// \brief Emit the string table described by \p Pool.
502   void emitStrings(const NonRelocatableStringpool &Pool);
503
504   /// \brief Emit debug_ranges for \p FuncRange by translating the
505   /// original \p Entries.
506   void emitRangesEntries(
507       int64_t UnitPcOffset, uint64_t OrigLowPc,
508       FunctionIntervals::const_iterator FuncRange,
509       const std::vector<DWARFDebugRangeList::RangeListEntry> &Entries,
510       unsigned AddressSize);
511
512   /// \brief Emit debug_aranges entries for \p Unit and if \p
513   /// DoRangesSection is true, also emit the debug_ranges entries for
514   /// the DW_TAG_compile_unit's DW_AT_ranges attribute.
515   void emitUnitRangesEntries(CompileUnit &Unit, bool DoRangesSection);
516
517   uint32_t getRangesSectionSize() const { return RangesSectionSize; }
518
519   /// \brief Emit the debug_loc contribution for \p Unit by copying
520   /// the entries from \p Dwarf and offseting them. Update the
521   /// location attributes to point to the new entries.
522   void emitLocationsForUnit(const CompileUnit &Unit, DWARFContext &Dwarf);
523
524   /// \brief Emit the line table described in \p Rows into the
525   /// debug_line section.
526   void emitLineTableForUnit(MCDwarfLineTableParams Params,
527                             StringRef PrologueBytes, unsigned MinInstLength,
528                             std::vector<DWARFDebugLine::Row> &Rows,
529                             unsigned AdddressSize);
530
531   uint32_t getLineSectionSize() const { return LineSectionSize; }
532
533   /// \brief Emit the .debug_pubnames contribution for \p Unit.
534   void emitPubNamesForUnit(const CompileUnit &Unit);
535
536   /// \brief Emit the .debug_pubtypes contribution for \p Unit.
537   void emitPubTypesForUnit(const CompileUnit &Unit);
538
539   /// \brief Emit a CIE.
540   void emitCIE(StringRef CIEBytes);
541
542   /// \brief Emit an FDE with data \p Bytes.
543   void emitFDE(uint32_t CIEOffset, uint32_t AddreSize, uint32_t Address,
544                StringRef Bytes);
545
546   uint32_t getFrameSectionSize() const { return FrameSectionSize; }
547 };
548
549 bool DwarfStreamer::init(Triple TheTriple, StringRef OutputFilename) {
550   std::string ErrorStr;
551   std::string TripleName;
552   StringRef Context = "dwarf streamer init";
553
554   // Get the target.
555   const Target *TheTarget =
556       TargetRegistry::lookupTarget(TripleName, TheTriple, ErrorStr);
557   if (!TheTarget)
558     return error(ErrorStr, Context);
559   TripleName = TheTriple.getTriple();
560
561   // Create all the MC Objects.
562   MRI.reset(TheTarget->createMCRegInfo(TripleName));
563   if (!MRI)
564     return error(Twine("no register info for target ") + TripleName, Context);
565
566   MAI.reset(TheTarget->createMCAsmInfo(*MRI, TripleName));
567   if (!MAI)
568     return error("no asm info for target " + TripleName, Context);
569
570   MOFI.reset(new MCObjectFileInfo);
571   MC.reset(new MCContext(MAI.get(), MRI.get(), MOFI.get()));
572   MOFI->InitMCObjectFileInfo(TheTriple, Reloc::Default, CodeModel::Default,
573                              *MC);
574
575   MAB = TheTarget->createMCAsmBackend(*MRI, TripleName, "");
576   if (!MAB)
577     return error("no asm backend for target " + TripleName, Context);
578
579   MII.reset(TheTarget->createMCInstrInfo());
580   if (!MII)
581     return error("no instr info info for target " + TripleName, Context);
582
583   MSTI.reset(TheTarget->createMCSubtargetInfo(TripleName, "", ""));
584   if (!MSTI)
585     return error("no subtarget info for target " + TripleName, Context);
586
587   MCE = TheTarget->createMCCodeEmitter(*MII, *MRI, *MC);
588   if (!MCE)
589     return error("no code emitter for target " + TripleName, Context);
590
591   // Create the output file.
592   std::error_code EC;
593   OutFile =
594       llvm::make_unique<raw_fd_ostream>(OutputFilename, EC, sys::fs::F_None);
595   if (EC)
596     return error(Twine(OutputFilename) + ": " + EC.message(), Context);
597
598   MS = TheTarget->createMCObjectStreamer(TargetTuple(TheTriple), *MC, *MAB,
599                                          *OutFile, MCE, *MSTI, false,
600                                          /*DWARFMustBeAtTheEnd*/ false);
601   if (!MS)
602     return error("no object streamer for target " + TripleName, Context);
603
604   // Finally create the AsmPrinter we'll use to emit the DIEs.
605   TM.reset(TheTarget->createTargetMachine(TripleName, "", "", TargetOptions()));
606   if (!TM)
607     return error("no target machine for target " + TripleName, Context);
608
609   Asm.reset(TheTarget->createAsmPrinter(*TM, std::unique_ptr<MCStreamer>(MS)));
610   if (!Asm)
611     return error("no asm printer for target " + TripleName, Context);
612
613   RangesSectionSize = 0;
614   LocSectionSize = 0;
615   LineSectionSize = 0;
616   FrameSectionSize = 0;
617
618   return true;
619 }
620
621 bool DwarfStreamer::finish(const DebugMap &DM) {
622   if (DM.getTriple().isOSDarwin() && !DM.getBinaryPath().empty())
623     return MachOUtils::generateDsymCompanion(DM, *MS, *OutFile);
624
625   MS->Finish();
626   return true;
627 }
628
629 /// \brief Set the current output section to debug_info and change
630 /// the MC Dwarf version to \p DwarfVersion.
631 void DwarfStreamer::switchToDebugInfoSection(unsigned DwarfVersion) {
632   MS->SwitchSection(MOFI->getDwarfInfoSection());
633   MC->setDwarfVersion(DwarfVersion);
634 }
635
636 /// \brief Emit the compilation unit header for \p Unit in the
637 /// debug_info section.
638 ///
639 /// A Dwarf scetion header is encoded as:
640 ///  uint32_t   Unit length (omiting this field)
641 ///  uint16_t   Version
642 ///  uint32_t   Abbreviation table offset
643 ///  uint8_t    Address size
644 ///
645 /// Leading to a total of 11 bytes.
646 void DwarfStreamer::emitCompileUnitHeader(CompileUnit &Unit) {
647   unsigned Version = Unit.getOrigUnit().getVersion();
648   switchToDebugInfoSection(Version);
649
650   // Emit size of content not including length itself. The size has
651   // already been computed in CompileUnit::computeOffsets(). Substract
652   // 4 to that size to account for the length field.
653   Asm->EmitInt32(Unit.getNextUnitOffset() - Unit.getStartOffset() - 4);
654   Asm->EmitInt16(Version);
655   // We share one abbreviations table across all units so it's always at the
656   // start of the section.
657   Asm->EmitInt32(0);
658   Asm->EmitInt8(Unit.getOrigUnit().getAddressByteSize());
659 }
660
661 /// \brief Emit the \p Abbrevs array as the shared abbreviation table
662 /// for the linked Dwarf file.
663 void DwarfStreamer::emitAbbrevs(const std::vector<DIEAbbrev *> &Abbrevs) {
664   MS->SwitchSection(MOFI->getDwarfAbbrevSection());
665   Asm->emitDwarfAbbrevs(Abbrevs);
666 }
667
668 /// \brief Recursively emit the DIE tree rooted at \p Die.
669 void DwarfStreamer::emitDIE(DIE &Die) {
670   MS->SwitchSection(MOFI->getDwarfInfoSection());
671   Asm->emitDwarfDIE(Die);
672 }
673
674 /// \brief Emit the debug_str section stored in \p Pool.
675 void DwarfStreamer::emitStrings(const NonRelocatableStringpool &Pool) {
676   Asm->OutStreamer->SwitchSection(MOFI->getDwarfStrSection());
677   for (auto *Entry = Pool.getFirstEntry(); Entry;
678        Entry = Pool.getNextEntry(Entry))
679     Asm->OutStreamer->EmitBytes(
680         StringRef(Entry->getKey().data(), Entry->getKey().size() + 1));
681 }
682
683 /// \brief Emit the debug_range section contents for \p FuncRange by
684 /// translating the original \p Entries. The debug_range section
685 /// format is totally trivial, consisting just of pairs of address
686 /// sized addresses describing the ranges.
687 void DwarfStreamer::emitRangesEntries(
688     int64_t UnitPcOffset, uint64_t OrigLowPc,
689     FunctionIntervals::const_iterator FuncRange,
690     const std::vector<DWARFDebugRangeList::RangeListEntry> &Entries,
691     unsigned AddressSize) {
692   MS->SwitchSection(MC->getObjectFileInfo()->getDwarfRangesSection());
693
694   // Offset each range by the right amount.
695   int64_t PcOffset = Entries.empty() ? 0 : FuncRange.value() + UnitPcOffset;
696   for (const auto &Range : Entries) {
697     if (Range.isBaseAddressSelectionEntry(AddressSize)) {
698       warn("unsupported base address selection operation",
699            "emitting debug_ranges");
700       break;
701     }
702     // Do not emit empty ranges.
703     if (Range.StartAddress == Range.EndAddress)
704       continue;
705
706     // All range entries should lie in the function range.
707     if (!(Range.StartAddress + OrigLowPc >= FuncRange.start() &&
708           Range.EndAddress + OrigLowPc <= FuncRange.stop()))
709       warn("inconsistent range data.", "emitting debug_ranges");
710     MS->EmitIntValue(Range.StartAddress + PcOffset, AddressSize);
711     MS->EmitIntValue(Range.EndAddress + PcOffset, AddressSize);
712     RangesSectionSize += 2 * AddressSize;
713   }
714
715   // Add the terminator entry.
716   MS->EmitIntValue(0, AddressSize);
717   MS->EmitIntValue(0, AddressSize);
718   RangesSectionSize += 2 * AddressSize;
719 }
720
721 /// \brief Emit the debug_aranges contribution of a unit and
722 /// if \p DoDebugRanges is true the debug_range contents for a
723 /// compile_unit level DW_AT_ranges attribute (Which are basically the
724 /// same thing with a different base address).
725 /// Just aggregate all the ranges gathered inside that unit.
726 void DwarfStreamer::emitUnitRangesEntries(CompileUnit &Unit,
727                                           bool DoDebugRanges) {
728   unsigned AddressSize = Unit.getOrigUnit().getAddressByteSize();
729   // Gather the ranges in a vector, so that we can simplify them. The
730   // IntervalMap will have coalesced the non-linked ranges, but here
731   // we want to coalesce the linked addresses.
732   std::vector<std::pair<uint64_t, uint64_t>> Ranges;
733   const auto &FunctionRanges = Unit.getFunctionRanges();
734   for (auto Range = FunctionRanges.begin(), End = FunctionRanges.end();
735        Range != End; ++Range)
736     Ranges.push_back(std::make_pair(Range.start() + Range.value(),
737                                     Range.stop() + Range.value()));
738
739   // The object addresses where sorted, but again, the linked
740   // addresses might end up in a different order.
741   std::sort(Ranges.begin(), Ranges.end());
742
743   if (!Ranges.empty()) {
744     MS->SwitchSection(MC->getObjectFileInfo()->getDwarfARangesSection());
745
746     MCSymbol *BeginLabel = Asm->createTempSymbol("Barange");
747     MCSymbol *EndLabel = Asm->createTempSymbol("Earange");
748
749     unsigned HeaderSize =
750         sizeof(int32_t) + // Size of contents (w/o this field
751         sizeof(int16_t) + // DWARF ARange version number
752         sizeof(int32_t) + // Offset of CU in the .debug_info section
753         sizeof(int8_t) +  // Pointer Size (in bytes)
754         sizeof(int8_t);   // Segment Size (in bytes)
755
756     unsigned TupleSize = AddressSize * 2;
757     unsigned Padding = OffsetToAlignment(HeaderSize, TupleSize);
758
759     Asm->EmitLabelDifference(EndLabel, BeginLabel, 4); // Arange length
760     Asm->OutStreamer->EmitLabel(BeginLabel);
761     Asm->EmitInt16(dwarf::DW_ARANGES_VERSION); // Version number
762     Asm->EmitInt32(Unit.getStartOffset());     // Corresponding unit's offset
763     Asm->EmitInt8(AddressSize);                // Address size
764     Asm->EmitInt8(0);                          // Segment size
765
766     Asm->OutStreamer->EmitFill(Padding, 0x0);
767
768     for (auto Range = Ranges.begin(), End = Ranges.end(); Range != End;
769          ++Range) {
770       uint64_t RangeStart = Range->first;
771       MS->EmitIntValue(RangeStart, AddressSize);
772       while ((Range + 1) != End && Range->second == (Range + 1)->first)
773         ++Range;
774       MS->EmitIntValue(Range->second - RangeStart, AddressSize);
775     }
776
777     // Emit terminator
778     Asm->OutStreamer->EmitIntValue(0, AddressSize);
779     Asm->OutStreamer->EmitIntValue(0, AddressSize);
780     Asm->OutStreamer->EmitLabel(EndLabel);
781   }
782
783   if (!DoDebugRanges)
784     return;
785
786   MS->SwitchSection(MC->getObjectFileInfo()->getDwarfRangesSection());
787   // Offset each range by the right amount.
788   int64_t PcOffset = -Unit.getLowPc();
789   // Emit coalesced ranges.
790   for (auto Range = Ranges.begin(), End = Ranges.end(); Range != End; ++Range) {
791     MS->EmitIntValue(Range->first + PcOffset, AddressSize);
792     while (Range + 1 != End && Range->second == (Range + 1)->first)
793       ++Range;
794     MS->EmitIntValue(Range->second + PcOffset, AddressSize);
795     RangesSectionSize += 2 * AddressSize;
796   }
797
798   // Add the terminator entry.
799   MS->EmitIntValue(0, AddressSize);
800   MS->EmitIntValue(0, AddressSize);
801   RangesSectionSize += 2 * AddressSize;
802 }
803
804 /// \brief Emit location lists for \p Unit and update attribtues to
805 /// point to the new entries.
806 void DwarfStreamer::emitLocationsForUnit(const CompileUnit &Unit,
807                                          DWARFContext &Dwarf) {
808   const auto &Attributes = Unit.getLocationAttributes();
809
810   if (Attributes.empty())
811     return;
812
813   MS->SwitchSection(MC->getObjectFileInfo()->getDwarfLocSection());
814
815   unsigned AddressSize = Unit.getOrigUnit().getAddressByteSize();
816   const DWARFSection &InputSec = Dwarf.getLocSection();
817   DataExtractor Data(InputSec.Data, Dwarf.isLittleEndian(), AddressSize);
818   DWARFUnit &OrigUnit = Unit.getOrigUnit();
819   const auto *OrigUnitDie = OrigUnit.getUnitDIE(false);
820   int64_t UnitPcOffset = 0;
821   uint64_t OrigLowPc = OrigUnitDie->getAttributeValueAsAddress(
822       &OrigUnit, dwarf::DW_AT_low_pc, -1ULL);
823   if (OrigLowPc != -1ULL)
824     UnitPcOffset = int64_t(OrigLowPc) - Unit.getLowPc();
825
826   for (const auto &Attr : Attributes) {
827     uint32_t Offset = Attr.first.get();
828     Attr.first.set(LocSectionSize);
829     // This is the quantity to add to the old location address to get
830     // the correct address for the new one.
831     int64_t LocPcOffset = Attr.second + UnitPcOffset;
832     while (Data.isValidOffset(Offset)) {
833       uint64_t Low = Data.getUnsigned(&Offset, AddressSize);
834       uint64_t High = Data.getUnsigned(&Offset, AddressSize);
835       LocSectionSize += 2 * AddressSize;
836       if (Low == 0 && High == 0) {
837         Asm->OutStreamer->EmitIntValue(0, AddressSize);
838         Asm->OutStreamer->EmitIntValue(0, AddressSize);
839         break;
840       }
841       Asm->OutStreamer->EmitIntValue(Low + LocPcOffset, AddressSize);
842       Asm->OutStreamer->EmitIntValue(High + LocPcOffset, AddressSize);
843       uint64_t Length = Data.getU16(&Offset);
844       Asm->OutStreamer->EmitIntValue(Length, 2);
845       // Just copy the bytes over.
846       Asm->OutStreamer->EmitBytes(
847           StringRef(InputSec.Data.substr(Offset, Length)));
848       Offset += Length;
849       LocSectionSize += Length + 2;
850     }
851   }
852 }
853
854 void DwarfStreamer::emitLineTableForUnit(MCDwarfLineTableParams Params,
855                                          StringRef PrologueBytes,
856                                          unsigned MinInstLength,
857                                          std::vector<DWARFDebugLine::Row> &Rows,
858                                          unsigned PointerSize) {
859   // Switch to the section where the table will be emitted into.
860   MS->SwitchSection(MC->getObjectFileInfo()->getDwarfLineSection());
861   MCSymbol *LineStartSym = MC->createTempSymbol();
862   MCSymbol *LineEndSym = MC->createTempSymbol();
863
864   // The first 4 bytes is the total length of the information for this
865   // compilation unit (not including these 4 bytes for the length).
866   Asm->EmitLabelDifference(LineEndSym, LineStartSym, 4);
867   Asm->OutStreamer->EmitLabel(LineStartSym);
868   // Copy Prologue.
869   MS->EmitBytes(PrologueBytes);
870   LineSectionSize += PrologueBytes.size() + 4;
871
872   SmallString<128> EncodingBuffer;
873   raw_svector_ostream EncodingOS(EncodingBuffer);
874
875   if (Rows.empty()) {
876     // We only have the dummy entry, dsymutil emits an entry with a 0
877     // address in that case.
878     MCDwarfLineAddr::Encode(*MC, Params, INT64_MAX, 0, EncodingOS);
879     MS->EmitBytes(EncodingOS.str());
880     LineSectionSize += EncodingBuffer.size();
881     MS->EmitLabel(LineEndSym);
882     return;
883   }
884
885   // Line table state machine fields
886   unsigned FileNum = 1;
887   unsigned LastLine = 1;
888   unsigned Column = 0;
889   unsigned IsStatement = 1;
890   unsigned Isa = 0;
891   uint64_t Address = -1ULL;
892
893   unsigned RowsSinceLastSequence = 0;
894
895   for (unsigned Idx = 0; Idx < Rows.size(); ++Idx) {
896     auto &Row = Rows[Idx];
897
898     int64_t AddressDelta;
899     if (Address == -1ULL) {
900       MS->EmitIntValue(dwarf::DW_LNS_extended_op, 1);
901       MS->EmitULEB128IntValue(PointerSize + 1);
902       MS->EmitIntValue(dwarf::DW_LNE_set_address, 1);
903       MS->EmitIntValue(Row.Address, PointerSize);
904       LineSectionSize += 2 + PointerSize + getULEB128Size(PointerSize + 1);
905       AddressDelta = 0;
906     } else {
907       AddressDelta = (Row.Address - Address) / MinInstLength;
908     }
909
910     // FIXME: code copied and transfromed from
911     // MCDwarf.cpp::EmitDwarfLineTable. We should find a way to share
912     // this code, but the current compatibility requirement with
913     // classic dsymutil makes it hard. Revisit that once this
914     // requirement is dropped.
915
916     if (FileNum != Row.File) {
917       FileNum = Row.File;
918       MS->EmitIntValue(dwarf::DW_LNS_set_file, 1);
919       MS->EmitULEB128IntValue(FileNum);
920       LineSectionSize += 1 + getULEB128Size(FileNum);
921     }
922     if (Column != Row.Column) {
923       Column = Row.Column;
924       MS->EmitIntValue(dwarf::DW_LNS_set_column, 1);
925       MS->EmitULEB128IntValue(Column);
926       LineSectionSize += 1 + getULEB128Size(Column);
927     }
928
929     // FIXME: We should handle the discriminator here, but dsymutil
930     // doesn' consider it, thus ignore it for now.
931
932     if (Isa != Row.Isa) {
933       Isa = Row.Isa;
934       MS->EmitIntValue(dwarf::DW_LNS_set_isa, 1);
935       MS->EmitULEB128IntValue(Isa);
936       LineSectionSize += 1 + getULEB128Size(Isa);
937     }
938     if (IsStatement != Row.IsStmt) {
939       IsStatement = Row.IsStmt;
940       MS->EmitIntValue(dwarf::DW_LNS_negate_stmt, 1);
941       LineSectionSize += 1;
942     }
943     if (Row.BasicBlock) {
944       MS->EmitIntValue(dwarf::DW_LNS_set_basic_block, 1);
945       LineSectionSize += 1;
946     }
947
948     if (Row.PrologueEnd) {
949       MS->EmitIntValue(dwarf::DW_LNS_set_prologue_end, 1);
950       LineSectionSize += 1;
951     }
952
953     if (Row.EpilogueBegin) {
954       MS->EmitIntValue(dwarf::DW_LNS_set_epilogue_begin, 1);
955       LineSectionSize += 1;
956     }
957
958     int64_t LineDelta = int64_t(Row.Line) - LastLine;
959     if (!Row.EndSequence) {
960       MCDwarfLineAddr::Encode(*MC, Params, LineDelta, AddressDelta, EncodingOS);
961       MS->EmitBytes(EncodingOS.str());
962       LineSectionSize += EncodingBuffer.size();
963       EncodingBuffer.resize(0);
964       Address = Row.Address;
965       LastLine = Row.Line;
966       RowsSinceLastSequence++;
967     } else {
968       if (LineDelta) {
969         MS->EmitIntValue(dwarf::DW_LNS_advance_line, 1);
970         MS->EmitSLEB128IntValue(LineDelta);
971         LineSectionSize += 1 + getSLEB128Size(LineDelta);
972       }
973       if (AddressDelta) {
974         MS->EmitIntValue(dwarf::DW_LNS_advance_pc, 1);
975         MS->EmitULEB128IntValue(AddressDelta);
976         LineSectionSize += 1 + getULEB128Size(AddressDelta);
977       }
978       MCDwarfLineAddr::Encode(*MC, Params, INT64_MAX, 0, EncodingOS);
979       MS->EmitBytes(EncodingOS.str());
980       LineSectionSize += EncodingBuffer.size();
981       EncodingBuffer.resize(0);
982       Address = -1ULL;
983       LastLine = FileNum = IsStatement = 1;
984       RowsSinceLastSequence = Column = Isa = 0;
985     }
986   }
987
988   if (RowsSinceLastSequence) {
989     MCDwarfLineAddr::Encode(*MC, Params, INT64_MAX, 0, EncodingOS);
990     MS->EmitBytes(EncodingOS.str());
991     LineSectionSize += EncodingBuffer.size();
992     EncodingBuffer.resize(0);
993   }
994
995   MS->EmitLabel(LineEndSym);
996 }
997
998 /// \brief Emit the pubnames or pubtypes section contribution for \p
999 /// Unit into \p Sec. The data is provided in \p Names.
1000 void DwarfStreamer::emitPubSectionForUnit(
1001     MCSection *Sec, StringRef SecName, const CompileUnit &Unit,
1002     const std::vector<CompileUnit::AccelInfo> &Names) {
1003   if (Names.empty())
1004     return;
1005
1006   // Start the dwarf pubnames section.
1007   Asm->OutStreamer->SwitchSection(Sec);
1008   MCSymbol *BeginLabel = Asm->createTempSymbol("pub" + SecName + "_begin");
1009   MCSymbol *EndLabel = Asm->createTempSymbol("pub" + SecName + "_end");
1010
1011   bool HeaderEmitted = false;
1012   // Emit the pubnames for this compilation unit.
1013   for (const auto &Name : Names) {
1014     if (Name.SkipPubSection)
1015       continue;
1016
1017     if (!HeaderEmitted) {
1018       // Emit the header.
1019       Asm->EmitLabelDifference(EndLabel, BeginLabel, 4); // Length
1020       Asm->OutStreamer->EmitLabel(BeginLabel);
1021       Asm->EmitInt16(dwarf::DW_PUBNAMES_VERSION); // Version
1022       Asm->EmitInt32(Unit.getStartOffset());      // Unit offset
1023       Asm->EmitInt32(Unit.getNextUnitOffset() - Unit.getStartOffset()); // Size
1024       HeaderEmitted = true;
1025     }
1026     Asm->EmitInt32(Name.Die->getOffset());
1027     Asm->OutStreamer->EmitBytes(
1028         StringRef(Name.Name.data(), Name.Name.size() + 1));
1029   }
1030
1031   if (!HeaderEmitted)
1032     return;
1033   Asm->EmitInt32(0); // End marker.
1034   Asm->OutStreamer->EmitLabel(EndLabel);
1035 }
1036
1037 /// \brief Emit .debug_pubnames for \p Unit.
1038 void DwarfStreamer::emitPubNamesForUnit(const CompileUnit &Unit) {
1039   emitPubSectionForUnit(MC->getObjectFileInfo()->getDwarfPubNamesSection(),
1040                         "names", Unit, Unit.getPubnames());
1041 }
1042
1043 /// \brief Emit .debug_pubtypes for \p Unit.
1044 void DwarfStreamer::emitPubTypesForUnit(const CompileUnit &Unit) {
1045   emitPubSectionForUnit(MC->getObjectFileInfo()->getDwarfPubTypesSection(),
1046                         "types", Unit, Unit.getPubtypes());
1047 }
1048
1049 /// \brief Emit a CIE into the debug_frame section.
1050 void DwarfStreamer::emitCIE(StringRef CIEBytes) {
1051   MS->SwitchSection(MC->getObjectFileInfo()->getDwarfFrameSection());
1052
1053   MS->EmitBytes(CIEBytes);
1054   FrameSectionSize += CIEBytes.size();
1055 }
1056
1057 /// \brief Emit a FDE into the debug_frame section. \p FDEBytes
1058 /// contains the FDE data without the length, CIE offset and address
1059 /// which will be replaced with the paramter values.
1060 void DwarfStreamer::emitFDE(uint32_t CIEOffset, uint32_t AddrSize,
1061                             uint32_t Address, StringRef FDEBytes) {
1062   MS->SwitchSection(MC->getObjectFileInfo()->getDwarfFrameSection());
1063
1064   MS->EmitIntValue(FDEBytes.size() + 4 + AddrSize, 4);
1065   MS->EmitIntValue(CIEOffset, 4);
1066   MS->EmitIntValue(Address, AddrSize);
1067   MS->EmitBytes(FDEBytes);
1068   FrameSectionSize += FDEBytes.size() + 8 + AddrSize;
1069 }
1070
1071 /// \brief The core of the Dwarf linking logic.
1072 ///
1073 /// The link of the dwarf information from the object files will be
1074 /// driven by the selection of 'root DIEs', which are DIEs that
1075 /// describe variables or functions that are present in the linked
1076 /// binary (and thus have entries in the debug map). All the debug
1077 /// information that will be linked (the DIEs, but also the line
1078 /// tables, ranges, ...) is derived from that set of root DIEs.
1079 ///
1080 /// The root DIEs are identified because they contain relocations that
1081 /// correspond to a debug map entry at specific places (the low_pc for
1082 /// a function, the location for a variable). These relocations are
1083 /// called ValidRelocs in the DwarfLinker and are gathered as a very
1084 /// first step when we start processing a DebugMapObject.
1085 class DwarfLinker {
1086 public:
1087   DwarfLinker(StringRef OutputFilename, const LinkOptions &Options)
1088       : OutputFilename(OutputFilename), Options(Options),
1089         BinHolder(Options.Verbose), LastCIEOffset(0) {}
1090
1091   ~DwarfLinker() {
1092     for (auto *Abbrev : Abbreviations)
1093       delete Abbrev;
1094   }
1095
1096   /// \brief Link the contents of the DebugMap.
1097   bool link(const DebugMap &);
1098
1099 private:
1100   /// \brief Called at the start of a debug object link.
1101   void startDebugObject(DWARFContext &, DebugMapObject &);
1102
1103   /// \brief Called at the end of a debug object link.
1104   void endDebugObject();
1105
1106   /// Keeps track of relocations.
1107   class RelocationManager {
1108     struct ValidReloc {
1109       uint32_t Offset;
1110       uint32_t Size;
1111       uint64_t Addend;
1112       const DebugMapObject::DebugMapEntry *Mapping;
1113
1114       ValidReloc(uint32_t Offset, uint32_t Size, uint64_t Addend,
1115                  const DebugMapObject::DebugMapEntry *Mapping)
1116           : Offset(Offset), Size(Size), Addend(Addend), Mapping(Mapping) {}
1117
1118       bool operator<(const ValidReloc &RHS) const {
1119         return Offset < RHS.Offset;
1120       }
1121     };
1122
1123     DwarfLinker &Linker;
1124
1125     /// \brief The valid relocations for the current DebugMapObject.
1126     /// This vector is sorted by relocation offset.
1127     std::vector<ValidReloc> ValidRelocs;
1128
1129     /// \brief Index into ValidRelocs of the next relocation to
1130     /// consider. As we walk the DIEs in acsending file offset and as
1131     /// ValidRelocs is sorted by file offset, keeping this index
1132     /// uptodate is all we have to do to have a cheap lookup during the
1133     /// root DIE selection and during DIE cloning.
1134     unsigned NextValidReloc;
1135
1136   public:
1137     RelocationManager(DwarfLinker &Linker)
1138         : Linker(Linker), NextValidReloc(0) {}
1139
1140     bool hasValidRelocs() const { return !ValidRelocs.empty(); }
1141     /// \brief Reset the NextValidReloc counter.
1142     void resetValidRelocs() { NextValidReloc = 0; }
1143
1144     /// \defgroup FindValidRelocations Translate debug map into a list
1145     /// of relevant relocations
1146     ///
1147     /// @{
1148     bool findValidRelocsInDebugInfo(const object::ObjectFile &Obj,
1149                                     const DebugMapObject &DMO);
1150
1151     bool findValidRelocs(const object::SectionRef &Section,
1152                          const object::ObjectFile &Obj,
1153                          const DebugMapObject &DMO);
1154
1155     void findValidRelocsMachO(const object::SectionRef &Section,
1156                               const object::MachOObjectFile &Obj,
1157                               const DebugMapObject &DMO);
1158     /// @}
1159
1160     bool hasValidRelocation(uint32_t StartOffset, uint32_t EndOffset,
1161                             CompileUnit::DIEInfo &Info);
1162
1163     bool applyValidRelocs(MutableArrayRef<char> Data, uint32_t BaseOffset,
1164                           bool isLittleEndian);
1165   };
1166
1167   /// \defgroup FindRootDIEs Find DIEs corresponding to debug map entries.
1168   ///
1169   /// @{
1170   /// \brief Recursively walk the \p DIE tree and look for DIEs to
1171   /// keep. Store that information in \p CU's DIEInfo.
1172   void lookForDIEsToKeep(RelocationManager &RelocMgr,
1173                          const DWARFDebugInfoEntryMinimal &DIE,
1174                          const DebugMapObject &DMO, CompileUnit &CU,
1175                          unsigned Flags);
1176
1177   /// \brief Flags passed to DwarfLinker::lookForDIEsToKeep
1178   enum TravesalFlags {
1179     TF_Keep = 1 << 0,            ///< Mark the traversed DIEs as kept.
1180     TF_InFunctionScope = 1 << 1, ///< Current scope is a fucntion scope.
1181     TF_DependencyWalk = 1 << 2,  ///< Walking the dependencies of a kept DIE.
1182     TF_ParentWalk = 1 << 3,      ///< Walking up the parents of a kept DIE.
1183     TF_ODR = 1 << 4,             ///< Use the ODR whhile keeping dependants.
1184     TF_SkipPC = 1 << 5,          ///< Skip all location attributes.
1185   };
1186
1187   /// \brief Mark the passed DIE as well as all the ones it depends on
1188   /// as kept.
1189   void keepDIEAndDenpendencies(RelocationManager &RelocMgr,
1190                                const DWARFDebugInfoEntryMinimal &DIE,
1191                                CompileUnit::DIEInfo &MyInfo,
1192                                const DebugMapObject &DMO, CompileUnit &CU,
1193                                bool UseODR);
1194
1195   unsigned shouldKeepDIE(RelocationManager &RelocMgr,
1196                          const DWARFDebugInfoEntryMinimal &DIE,
1197                          CompileUnit &Unit, CompileUnit::DIEInfo &MyInfo,
1198                          unsigned Flags);
1199
1200   unsigned shouldKeepVariableDIE(RelocationManager &RelocMgr,
1201                                  const DWARFDebugInfoEntryMinimal &DIE,
1202                                  CompileUnit &Unit,
1203                                  CompileUnit::DIEInfo &MyInfo, unsigned Flags);
1204
1205   unsigned shouldKeepSubprogramDIE(RelocationManager &RelocMgr,
1206                                    const DWARFDebugInfoEntryMinimal &DIE,
1207                                    CompileUnit &Unit,
1208                                    CompileUnit::DIEInfo &MyInfo,
1209                                    unsigned Flags);
1210
1211   bool hasValidRelocation(uint32_t StartOffset, uint32_t EndOffset,
1212                           CompileUnit::DIEInfo &Info);
1213   /// @}
1214
1215   /// \defgroup Linking Methods used to link the debug information
1216   ///
1217   /// @{
1218
1219   class DIECloner {
1220     DwarfLinker &Linker;
1221     RelocationManager &RelocMgr;
1222     /// Allocator used for all the DIEValue objects.
1223     BumpPtrAllocator &DIEAlloc;
1224     MutableArrayRef<CompileUnit> CompileUnits;
1225     LinkOptions Options;
1226
1227   public:
1228     DIECloner(DwarfLinker &Linker, RelocationManager &RelocMgr,
1229               BumpPtrAllocator &DIEAlloc,
1230               MutableArrayRef<CompileUnit> CompileUnits, LinkOptions &Options)
1231         : Linker(Linker), RelocMgr(RelocMgr), DIEAlloc(DIEAlloc),
1232           CompileUnits(CompileUnits), Options(Options) {}
1233
1234     /// Recursively clone \p InputDIE into an tree of DIE objects
1235     /// where useless (as decided by lookForDIEsToKeep()) bits have been
1236     /// stripped out and addresses have been rewritten according to the
1237     /// debug map.
1238     ///
1239     /// \param OutOffset is the offset the cloned DIE in the output
1240     /// compile unit.
1241     /// \param PCOffset (while cloning a function scope) is the offset
1242     /// applied to the entry point of the function to get the linked address.
1243     ///
1244     /// \returns the root of the cloned tree or null if nothing was selected.
1245     DIE *cloneDIE(const DWARFDebugInfoEntryMinimal &InputDIE, CompileUnit &U,
1246                   int64_t PCOffset, uint32_t OutOffset, unsigned Flags);
1247
1248     /// Construct the output DIE tree by cloning the DIEs we
1249     /// chose to keep above. If there are no valid relocs, then there's
1250     /// nothing to clone/emit.
1251     void cloneAllCompileUnits(DWARFContextInMemory &DwarfContext);
1252
1253   private:
1254     typedef DWARFAbbreviationDeclaration::AttributeSpec AttributeSpec;
1255
1256     /// Information gathered and exchanged between the various
1257     /// clone*Attributes helpers about the attributes of a particular DIE.
1258     struct AttributesInfo {
1259       const char *Name, *MangledName;         ///< Names.
1260       uint32_t NameOffset, MangledNameOffset; ///< Offsets in the string pool.
1261
1262       uint64_t OrigLowPc;  ///< Value of AT_low_pc in the input DIE
1263       uint64_t OrigHighPc; ///< Value of AT_high_pc in the input DIE
1264       int64_t PCOffset; ///< Offset to apply to PC addresses inside a function.
1265
1266       bool HasLowPc;      ///< Does the DIE have a low_pc attribute?
1267       bool IsDeclaration; ///< Is this DIE only a declaration?
1268
1269       AttributesInfo()
1270           : Name(nullptr), MangledName(nullptr), NameOffset(0),
1271             MangledNameOffset(0), OrigLowPc(UINT64_MAX), OrigHighPc(0),
1272             PCOffset(0), HasLowPc(false), IsDeclaration(false) {}
1273     };
1274
1275     /// Helper for cloneDIE.
1276     unsigned cloneAttribute(DIE &Die,
1277                             const DWARFDebugInfoEntryMinimal &InputDIE,
1278                             CompileUnit &U, const DWARFFormValue &Val,
1279                             const AttributeSpec AttrSpec, unsigned AttrSize,
1280                             AttributesInfo &AttrInfo);
1281
1282     /// Clone a string attribute described by \p AttrSpec and add
1283     /// it to \p Die.
1284     /// \returns the size of the new attribute.
1285     unsigned cloneStringAttribute(DIE &Die, AttributeSpec AttrSpec,
1286                                   const DWARFFormValue &Val,
1287                                   const DWARFUnit &U);
1288
1289     /// Clone an attribute referencing another DIE and add
1290     /// it to \p Die.
1291     /// \returns the size of the new attribute.
1292     unsigned
1293     cloneDieReferenceAttribute(DIE &Die,
1294                                const DWARFDebugInfoEntryMinimal &InputDIE,
1295                                AttributeSpec AttrSpec, unsigned AttrSize,
1296                                const DWARFFormValue &Val, CompileUnit &Unit);
1297
1298     /// Clone an attribute referencing another DIE and add
1299     /// it to \p Die.
1300     /// \returns the size of the new attribute.
1301     unsigned cloneBlockAttribute(DIE &Die, AttributeSpec AttrSpec,
1302                                  const DWARFFormValue &Val, unsigned AttrSize);
1303
1304     /// Clone an attribute referencing another DIE and add
1305     /// it to \p Die.
1306     /// \returns the size of the new attribute.
1307     unsigned cloneAddressAttribute(DIE &Die, AttributeSpec AttrSpec,
1308                                    const DWARFFormValue &Val,
1309                                    const CompileUnit &Unit,
1310                                    AttributesInfo &Info);
1311
1312     /// Clone a scalar attribute  and add it to \p Die.
1313     /// \returns the size of the new attribute.
1314     unsigned cloneScalarAttribute(DIE &Die,
1315                                   const DWARFDebugInfoEntryMinimal &InputDIE,
1316                                   CompileUnit &U, AttributeSpec AttrSpec,
1317                                   const DWARFFormValue &Val, unsigned AttrSize,
1318                                   AttributesInfo &Info);
1319
1320     /// Get the potential name and mangled name for the entity
1321     /// described by \p Die and store them in \Info if they are not
1322     /// already there.
1323     /// \returns is a name was found.
1324     bool getDIENames(const DWARFDebugInfoEntryMinimal &Die, DWARFUnit &U,
1325                      AttributesInfo &Info);
1326
1327     /// Create a copy of abbreviation Abbrev.
1328     void copyAbbrev(const DWARFAbbreviationDeclaration &Abbrev, bool hasODR);
1329   };
1330
1331   /// \brief Assign an abbreviation number to \p Abbrev
1332   void AssignAbbrev(DIEAbbrev &Abbrev);
1333
1334   /// \brief FoldingSet that uniques the abbreviations.
1335   FoldingSet<DIEAbbrev> AbbreviationsSet;
1336   /// \brief Storage for the unique Abbreviations.
1337   /// This is passed to AsmPrinter::emitDwarfAbbrevs(), thus it cannot
1338   /// be changed to a vecot of unique_ptrs.
1339   std::vector<DIEAbbrev *> Abbreviations;
1340
1341   /// \brief Compute and emit debug_ranges section for \p Unit, and
1342   /// patch the attributes referencing it.
1343   void patchRangesForUnit(const CompileUnit &Unit, DWARFContext &Dwarf) const;
1344
1345   /// \brief Generate and emit the DW_AT_ranges attribute for a
1346   /// compile_unit if it had one.
1347   void generateUnitRanges(CompileUnit &Unit) const;
1348
1349   /// \brief Extract the line tables fromt he original dwarf, extract
1350   /// the relevant parts according to the linked function ranges and
1351   /// emit the result in the debug_line section.
1352   void patchLineTableForUnit(CompileUnit &Unit, DWARFContext &OrigDwarf);
1353
1354   /// \brief Emit the accelerator entries for \p Unit.
1355   void emitAcceleratorEntriesForUnit(CompileUnit &Unit);
1356
1357   /// \brief Patch the frame info for an object file and emit it.
1358   void patchFrameInfoForObject(const DebugMapObject &, DWARFContext &,
1359                                unsigned AddressSize);
1360
1361   /// \brief DIELoc objects that need to be destructed (but not freed!).
1362   std::vector<DIELoc *> DIELocs;
1363   /// \brief DIEBlock objects that need to be destructed (but not freed!).
1364   std::vector<DIEBlock *> DIEBlocks;
1365   /// \brief Allocator used for all the DIEValue objects.
1366   BumpPtrAllocator DIEAlloc;
1367   /// @}
1368
1369   /// ODR Contexts for that link.
1370   DeclContextTree ODRContexts;
1371
1372   /// \defgroup Helpers Various helper methods.
1373   ///
1374   /// @{
1375   const DWARFDebugInfoEntryMinimal *
1376   resolveDIEReference(const DWARFFormValue &RefValue, const DWARFUnit &Unit,
1377                       const DWARFDebugInfoEntryMinimal &DIE,
1378                       CompileUnit *&ReferencedCU);
1379
1380   CompileUnit *getUnitForOffset(unsigned Offset);
1381
1382   void reportWarning(const Twine &Warning, const DWARFUnit *Unit = nullptr,
1383                      const DWARFDebugInfoEntryMinimal *DIE = nullptr) const;
1384
1385   bool createStreamer(Triple TheTriple, StringRef OutputFilename);
1386
1387   /// \brief Attempt to load a debug object from disk.
1388   ErrorOr<const object::ObjectFile &> loadObject(BinaryHolder &BinaryHolder,
1389                                                  DebugMapObject &Obj,
1390                                                  const DebugMap &Map);
1391   /// @}
1392
1393 private:
1394   std::string OutputFilename;
1395   LinkOptions Options;
1396   BinaryHolder BinHolder;
1397   std::unique_ptr<DwarfStreamer> Streamer;
1398   uint64_t OutputDebugInfoSize;
1399
1400   /// The units of the current debug map object.
1401   std::vector<CompileUnit> Units;
1402
1403   /// The debug map object curently under consideration.
1404   DebugMapObject *CurrentDebugObject;
1405
1406   /// \brief The Dwarf string pool
1407   NonRelocatableStringpool StringPool;
1408
1409   /// \brief This map is keyed by the entry PC of functions in that
1410   /// debug object and the associated value is a pair storing the
1411   /// corresponding end PC and the offset to apply to get the linked
1412   /// address.
1413   ///
1414   /// See startDebugObject() for a more complete description of its use.
1415   std::map<uint64_t, std::pair<uint64_t, int64_t>> Ranges;
1416
1417   /// \brief The CIEs that have been emitted in the output
1418   /// section. The actual CIE data serves a the key to this StringMap,
1419   /// this takes care of comparing the semantics of CIEs defined in
1420   /// different object files.
1421   StringMap<uint32_t> EmittedCIEs;
1422
1423   /// Offset of the last CIE that has been emitted in the output
1424   /// debug_frame section.
1425   uint32_t LastCIEOffset;
1426 };
1427
1428 /// \brief Similar to DWARFUnitSection::getUnitForOffset(), but
1429 /// returning our CompileUnit object instead.
1430 CompileUnit *DwarfLinker::getUnitForOffset(unsigned Offset) {
1431   auto CU =
1432       std::upper_bound(Units.begin(), Units.end(), Offset,
1433                        [](uint32_t LHS, const CompileUnit &RHS) {
1434                          return LHS < RHS.getOrigUnit().getNextUnitOffset();
1435                        });
1436   return CU != Units.end() ? &*CU : nullptr;
1437 }
1438
1439 /// \brief Resolve the DIE attribute reference that has been
1440 /// extracted in \p RefValue. The resulting DIE migh be in another
1441 /// CompileUnit which is stored into \p ReferencedCU.
1442 /// \returns null if resolving fails for any reason.
1443 const DWARFDebugInfoEntryMinimal *DwarfLinker::resolveDIEReference(
1444     const DWARFFormValue &RefValue, const DWARFUnit &Unit,
1445     const DWARFDebugInfoEntryMinimal &DIE, CompileUnit *&RefCU) {
1446   assert(RefValue.isFormClass(DWARFFormValue::FC_Reference));
1447   uint64_t RefOffset = *RefValue.getAsReference(&Unit);
1448
1449   if ((RefCU = getUnitForOffset(RefOffset)))
1450     if (const auto *RefDie = RefCU->getOrigUnit().getDIEForOffset(RefOffset))
1451       return RefDie;
1452
1453   reportWarning("could not find referenced DIE", &Unit, &DIE);
1454   return nullptr;
1455 }
1456
1457 /// \returns whether the passed \a Attr type might contain a DIE
1458 /// reference suitable for ODR uniquing.
1459 static bool isODRAttribute(uint16_t Attr) {
1460   switch (Attr) {
1461   default:
1462     return false;
1463   case dwarf::DW_AT_type:
1464   case dwarf::DW_AT_containing_type:
1465   case dwarf::DW_AT_specification:
1466   case dwarf::DW_AT_abstract_origin:
1467   case dwarf::DW_AT_import:
1468     return true;
1469   }
1470   llvm_unreachable("Improper attribute.");
1471 }
1472
1473 /// Set the last DIE/CU a context was seen in and, possibly invalidate
1474 /// the context if it is ambiguous.
1475 ///
1476 /// In the current implementation, we don't handle overloaded
1477 /// functions well, because the argument types are not taken into
1478 /// account when computing the DeclContext tree.
1479 ///
1480 /// Some of this is mitigated byt using mangled names that do contain
1481 /// the arguments types, but sometimes (eg. with function templates)
1482 /// we don't have that. In that case, just do not unique anything that
1483 /// refers to the contexts we are not able to distinguish.
1484 ///
1485 /// If a context that is not a namespace appears twice in the same CU,
1486 /// we know it is ambiguous. Make it invalid.
1487 bool DeclContext::setLastSeenDIE(CompileUnit &U,
1488                                  const DWARFDebugInfoEntryMinimal *Die) {
1489   if (LastSeenCompileUnitID == U.getUniqueID()) {
1490     DWARFUnit &OrigUnit = U.getOrigUnit();
1491     uint32_t FirstIdx = OrigUnit.getDIEIndex(LastSeenDIE);
1492     U.getInfo(FirstIdx).Ctxt = nullptr;
1493     return false;
1494   }
1495
1496   LastSeenCompileUnitID = U.getUniqueID();
1497   LastSeenDIE = Die;
1498   return true;
1499 }
1500
1501 /// Get the child context of \a Context corresponding to \a DIE.
1502 ///
1503 /// \returns the child context or null if we shouldn't track children
1504 /// contexts. It also returns an additional bit meaning 'invalid'. An
1505 /// invalid context means it shouldn't be considered for uniquing, but
1506 /// its not returning null, because some children of that context
1507 /// might be uniquing candidates.
1508 /// FIXME: this is for dsymutil-classic compatibility, I don't think
1509 /// it buys us much.
1510 PointerIntPair<DeclContext *, 1> DeclContextTree::getChildDeclContext(
1511     DeclContext &Context, const DWARFDebugInfoEntryMinimal *DIE, CompileUnit &U,
1512     NonRelocatableStringpool &StringPool) {
1513   unsigned Tag = DIE->getTag();
1514
1515   // FIXME: dsymutil-classic compat: We should bail out here if we
1516   // have a specification or an abstract_origin. We will get the
1517   // parent context wrong here.
1518
1519   switch (Tag) {
1520   default:
1521     // By default stop gathering child contexts.
1522     return PointerIntPair<DeclContext *, 1>(nullptr);
1523   case dwarf::DW_TAG_compile_unit:
1524     // FIXME: Add support for DW_TAG_module.
1525     return PointerIntPair<DeclContext *, 1>(&Context);
1526   case dwarf::DW_TAG_subprogram:
1527     // Do not unique anything inside CU local functions.
1528     if ((Context.getTag() == dwarf::DW_TAG_namespace ||
1529          Context.getTag() == dwarf::DW_TAG_compile_unit) &&
1530         !DIE->getAttributeValueAsUnsignedConstant(&U.getOrigUnit(),
1531                                                   dwarf::DW_AT_external, 0))
1532       return PointerIntPair<DeclContext *, 1>(nullptr);
1533   // Fallthrough
1534   case dwarf::DW_TAG_member:
1535   case dwarf::DW_TAG_namespace:
1536   case dwarf::DW_TAG_structure_type:
1537   case dwarf::DW_TAG_class_type:
1538   case dwarf::DW_TAG_union_type:
1539   case dwarf::DW_TAG_enumeration_type:
1540   case dwarf::DW_TAG_typedef:
1541     // Artificial things might be ambiguous, because they might be
1542     // created on demand. For example implicitely defined constructors
1543     // are ambiguous because of the way we identify contexts, and they
1544     // won't be generated everytime everywhere.
1545     if (DIE->getAttributeValueAsUnsignedConstant(&U.getOrigUnit(),
1546                                                  dwarf::DW_AT_artificial, 0))
1547       return PointerIntPair<DeclContext *, 1>(nullptr);
1548     break;
1549   }
1550
1551   const char *Name = DIE->getName(&U.getOrigUnit(), DINameKind::LinkageName);
1552   const char *ShortName = DIE->getName(&U.getOrigUnit(), DINameKind::ShortName);
1553   StringRef NameRef;
1554   StringRef ShortNameRef;
1555   StringRef FileRef;
1556
1557   if (Name)
1558     NameRef = StringPool.internString(Name);
1559   else if (Tag == dwarf::DW_TAG_namespace)
1560     // FIXME: For dsymutil-classic compatibility. I think uniquing
1561     // within anonymous namespaces is wrong. There is no ODR guarantee
1562     // there.
1563     NameRef = StringPool.internString("(anonymous namespace)");
1564
1565   if (ShortName && ShortName != Name)
1566     ShortNameRef = StringPool.internString(ShortName);
1567   else
1568     ShortNameRef = NameRef;
1569
1570   if (Tag != dwarf::DW_TAG_class_type && Tag != dwarf::DW_TAG_structure_type &&
1571       Tag != dwarf::DW_TAG_union_type &&
1572       Tag != dwarf::DW_TAG_enumeration_type && NameRef.empty())
1573     return PointerIntPair<DeclContext *, 1>(nullptr);
1574
1575   std::string File;
1576   unsigned Line = 0;
1577   unsigned ByteSize = 0;
1578
1579   // Gather some discriminating data about the DeclContext we will be
1580   // creating: File, line number and byte size. This shouldn't be
1581   // necessary, because the ODR is just about names, but given that we
1582   // do some approximations with overloaded functions and anonymous
1583   // namespaces, use these additional data points to make the process safer.
1584   ByteSize = DIE->getAttributeValueAsUnsignedConstant(
1585       &U.getOrigUnit(), dwarf::DW_AT_byte_size, UINT64_MAX);
1586   if (Tag != dwarf::DW_TAG_namespace || !Name) {
1587     if (unsigned FileNum = DIE->getAttributeValueAsUnsignedConstant(
1588             &U.getOrigUnit(), dwarf::DW_AT_decl_file, 0)) {
1589       if (const auto *LT = U.getOrigUnit().getContext().getLineTableForUnit(
1590               &U.getOrigUnit())) {
1591         // FIXME: dsymutil-classic compatibility. I'd rather not
1592         // unique anything in anonymous namespaces, but if we do, then
1593         // verify that the file and line correspond.
1594         if (!Name && Tag == dwarf::DW_TAG_namespace)
1595           FileNum = 1;
1596
1597         // FIXME: Passing U.getOrigUnit().getCompilationDir()
1598         // instead of "" would allow more uniquing, but for now, do
1599         // it this way to match dsymutil-classic.
1600         if (LT->getFileNameByIndex(
1601                 FileNum, "",
1602                 DILineInfoSpecifier::FileLineInfoKind::AbsoluteFilePath,
1603                 File)) {
1604           Line = DIE->getAttributeValueAsUnsignedConstant(
1605               &U.getOrigUnit(), dwarf::DW_AT_decl_line, 0);
1606 #ifdef HAVE_REALPATH
1607           // Cache the resolved paths, because calling realpath is expansive.
1608           if (const char *ResolvedPath = U.getResolvedPath(FileNum)) {
1609             File = ResolvedPath;
1610           } else {
1611             char RealPath[PATH_MAX + 1];
1612             RealPath[PATH_MAX] = 0;
1613             if (::realpath(File.c_str(), RealPath))
1614               File = RealPath;
1615             U.setResolvedPath(FileNum, File);
1616           }
1617 #endif
1618           FileRef = StringPool.internString(File);
1619         }
1620       }
1621     }
1622   }
1623
1624   if (!Line && NameRef.empty())
1625     return PointerIntPair<DeclContext *, 1>(nullptr);
1626
1627   // FIXME: dsymutil-classic compat won't unique the same type
1628   // presented once as a struct and once as a class. Use the Tag in
1629   // the fully qualified name hash to get the same effect.
1630   // We hash NameRef, which is the mangled name, in order to get most
1631   // overloaded functions resolvec correctly.
1632   unsigned Hash = hash_combine(Context.getQualifiedNameHash(), Tag, NameRef);
1633
1634   // FIXME: dsymutil-classic compatibility: when we don't have a name,
1635   // use the filename.
1636   if (Tag == dwarf::DW_TAG_namespace && NameRef == "(anonymous namespace)")
1637     Hash = hash_combine(Hash, FileRef);
1638
1639   // Now look if this context already exists.
1640   DeclContext Key(Hash, Line, ByteSize, Tag, NameRef, FileRef, Context);
1641   auto ContextIter = Contexts.find(&Key);
1642
1643   if (ContextIter == Contexts.end()) {
1644     // The context wasn't found.
1645     bool Inserted;
1646     DeclContext *NewContext =
1647         new (Allocator) DeclContext(Hash, Line, ByteSize, Tag, NameRef, FileRef,
1648                                     Context, DIE, U.getUniqueID());
1649     std::tie(ContextIter, Inserted) = Contexts.insert(NewContext);
1650     assert(Inserted && "Failed to insert DeclContext");
1651     (void)Inserted;
1652   } else if (Tag != dwarf::DW_TAG_namespace &&
1653              !(*ContextIter)->setLastSeenDIE(U, DIE)) {
1654     // The context was found, but it is ambiguous with another context
1655     // in the same file. Mark it invalid.
1656     return PointerIntPair<DeclContext *, 1>(*ContextIter, /* Invalid= */ 1);
1657   }
1658
1659   assert(ContextIter != Contexts.end());
1660   // FIXME: dsymutil-classic compatibility. Union types aren't
1661   // uniques, but their children might be.
1662   if ((Tag == dwarf::DW_TAG_subprogram &&
1663        Context.getTag() != dwarf::DW_TAG_structure_type &&
1664        Context.getTag() != dwarf::DW_TAG_class_type) ||
1665       (Tag == dwarf::DW_TAG_union_type))
1666     return PointerIntPair<DeclContext *, 1>(*ContextIter, /* Invalid= */ 1);
1667
1668   return PointerIntPair<DeclContext *, 1>(*ContextIter);
1669 }
1670
1671 bool DwarfLinker::DIECloner::getDIENames(const DWARFDebugInfoEntryMinimal &Die,
1672                                          DWARFUnit &U, AttributesInfo &Info) {
1673   // FIXME: a bit wasteful as the first getName might return the
1674   // short name.
1675   if (!Info.MangledName &&
1676       (Info.MangledName = Die.getName(&U, DINameKind::LinkageName)))
1677     Info.MangledNameOffset =
1678         Linker.StringPool.getStringOffset(Info.MangledName);
1679
1680   if (!Info.Name && (Info.Name = Die.getName(&U, DINameKind::ShortName)))
1681     Info.NameOffset = Linker.StringPool.getStringOffset(Info.Name);
1682
1683   return Info.Name || Info.MangledName;
1684 }
1685
1686 /// \brief Report a warning to the user, optionaly including
1687 /// information about a specific \p DIE related to the warning.
1688 void DwarfLinker::reportWarning(const Twine &Warning, const DWARFUnit *Unit,
1689                                 const DWARFDebugInfoEntryMinimal *DIE) const {
1690   StringRef Context = "<debug map>";
1691   if (CurrentDebugObject)
1692     Context = CurrentDebugObject->getObjectFilename();
1693   warn(Warning, Context);
1694
1695   if (!Options.Verbose || !DIE)
1696     return;
1697
1698   errs() << "    in DIE:\n";
1699   DIE->dump(errs(), const_cast<DWARFUnit *>(Unit), 0 /* RecurseDepth */,
1700             6 /* Indent */);
1701 }
1702
1703 bool DwarfLinker::createStreamer(Triple TheTriple, StringRef OutputFilename) {
1704   if (Options.NoOutput)
1705     return true;
1706
1707   Streamer = llvm::make_unique<DwarfStreamer>();
1708   return Streamer->init(TheTriple, OutputFilename);
1709 }
1710
1711 /// \brief Recursive helper to gather the child->parent relationships in the
1712 /// original compile unit.
1713 static void gatherDIEParents(const DWARFDebugInfoEntryMinimal *DIE,
1714                              unsigned ParentIdx, CompileUnit &CU,
1715                              DeclContext *CurrentDeclContext,
1716                              NonRelocatableStringpool &StringPool,
1717                              DeclContextTree &Contexts) {
1718   unsigned MyIdx = CU.getOrigUnit().getDIEIndex(DIE);
1719   CompileUnit::DIEInfo &Info = CU.getInfo(MyIdx);
1720
1721   Info.ParentIdx = ParentIdx;
1722   if (CU.hasODR()) {
1723     if (CurrentDeclContext) {
1724       auto PtrInvalidPair = Contexts.getChildDeclContext(*CurrentDeclContext,
1725                                                          DIE, CU, StringPool);
1726       CurrentDeclContext = PtrInvalidPair.getPointer();
1727       Info.Ctxt =
1728           PtrInvalidPair.getInt() ? nullptr : PtrInvalidPair.getPointer();
1729     } else
1730       Info.Ctxt = CurrentDeclContext = nullptr;
1731   }
1732
1733   if (DIE->hasChildren())
1734     for (auto *Child = DIE->getFirstChild(); Child && !Child->isNULL();
1735          Child = Child->getSibling())
1736       gatherDIEParents(Child, MyIdx, CU, CurrentDeclContext, StringPool,
1737                        Contexts);
1738 }
1739
1740 static bool dieNeedsChildrenToBeMeaningful(uint32_t Tag) {
1741   switch (Tag) {
1742   default:
1743     return false;
1744   case dwarf::DW_TAG_subprogram:
1745   case dwarf::DW_TAG_lexical_block:
1746   case dwarf::DW_TAG_subroutine_type:
1747   case dwarf::DW_TAG_structure_type:
1748   case dwarf::DW_TAG_class_type:
1749   case dwarf::DW_TAG_union_type:
1750     return true;
1751   }
1752   llvm_unreachable("Invalid Tag");
1753 }
1754
1755 static unsigned getRefAddrSize(const DWARFUnit &U) {
1756   if (U.getVersion() == 2)
1757     return U.getAddressByteSize();
1758   return 4;
1759 }
1760
1761 void DwarfLinker::startDebugObject(DWARFContext &Dwarf, DebugMapObject &Obj) {
1762   Units.reserve(Dwarf.getNumCompileUnits());
1763   // Iterate over the debug map entries and put all the ones that are
1764   // functions (because they have a size) into the Ranges map. This
1765   // map is very similar to the FunctionRanges that are stored in each
1766   // unit, with 2 notable differences:
1767   //  - obviously this one is global, while the other ones are per-unit.
1768   //  - this one contains not only the functions described in the DIE
1769   // tree, but also the ones that are only in the debug map.
1770   // The latter information is required to reproduce dsymutil's logic
1771   // while linking line tables. The cases where this information
1772   // matters look like bugs that need to be investigated, but for now
1773   // we need to reproduce dsymutil's behavior.
1774   // FIXME: Once we understood exactly if that information is needed,
1775   // maybe totally remove this (or try to use it to do a real
1776   // -gline-tables-only on Darwin.
1777   for (const auto &Entry : Obj.symbols()) {
1778     const auto &Mapping = Entry.getValue();
1779     if (Mapping.Size)
1780       Ranges[Mapping.ObjectAddress] = std::make_pair(
1781           Mapping.ObjectAddress + Mapping.Size,
1782           int64_t(Mapping.BinaryAddress) - Mapping.ObjectAddress);
1783   }
1784 }
1785
1786 void DwarfLinker::endDebugObject() {
1787   Units.clear();
1788   Ranges.clear();
1789
1790   for (auto I = DIEBlocks.begin(), E = DIEBlocks.end(); I != E; ++I)
1791     (*I)->~DIEBlock();
1792   for (auto I = DIELocs.begin(), E = DIELocs.end(); I != E; ++I)
1793     (*I)->~DIELoc();
1794
1795   DIEBlocks.clear();
1796   DIELocs.clear();
1797   DIEAlloc.Reset();
1798 }
1799
1800 /// \brief Iterate over the relocations of the given \p Section and
1801 /// store the ones that correspond to debug map entries into the
1802 /// ValidRelocs array.
1803 void DwarfLinker::RelocationManager::
1804 findValidRelocsMachO(const object::SectionRef &Section,
1805                      const object::MachOObjectFile &Obj,
1806                      const DebugMapObject &DMO) {
1807   StringRef Contents;
1808   Section.getContents(Contents);
1809   DataExtractor Data(Contents, Obj.isLittleEndian(), 0);
1810
1811   for (const object::RelocationRef &Reloc : Section.relocations()) {
1812     object::DataRefImpl RelocDataRef = Reloc.getRawDataRefImpl();
1813     MachO::any_relocation_info MachOReloc = Obj.getRelocation(RelocDataRef);
1814     unsigned RelocSize = 1 << Obj.getAnyRelocationLength(MachOReloc);
1815     uint64_t Offset64 = Reloc.getOffset();
1816     if ((RelocSize != 4 && RelocSize != 8)) {
1817       Linker.reportWarning(" unsupported relocation in debug_info section.");
1818       continue;
1819     }
1820     uint32_t Offset = Offset64;
1821     // Mach-o uses REL relocations, the addend is at the relocation offset.
1822     uint64_t Addend = Data.getUnsigned(&Offset, RelocSize);
1823
1824     auto Sym = Reloc.getSymbol();
1825     if (Sym != Obj.symbol_end()) {
1826       ErrorOr<StringRef> SymbolName = Sym->getName();
1827       if (!SymbolName) {
1828         Linker.reportWarning("error getting relocation symbol name.");
1829         continue;
1830       }
1831       if (const auto *Mapping = DMO.lookupSymbol(*SymbolName))
1832         ValidRelocs.emplace_back(Offset64, RelocSize, Addend, Mapping);
1833     } else if (const auto *Mapping = DMO.lookupObjectAddress(Addend)) {
1834       // Do not store the addend. The addend was the address of the
1835       // symbol in the object file, the address in the binary that is
1836       // stored in the debug map doesn't need to be offseted.
1837       ValidRelocs.emplace_back(Offset64, RelocSize, 0, Mapping);
1838     }
1839   }
1840 }
1841
1842 /// \brief Dispatch the valid relocation finding logic to the
1843 /// appropriate handler depending on the object file format.
1844 bool DwarfLinker::RelocationManager::findValidRelocs(
1845     const object::SectionRef &Section, const object::ObjectFile &Obj,
1846     const DebugMapObject &DMO) {
1847   // Dispatch to the right handler depending on the file type.
1848   if (auto *MachOObj = dyn_cast<object::MachOObjectFile>(&Obj))
1849     findValidRelocsMachO(Section, *MachOObj, DMO);
1850   else
1851     Linker.reportWarning(Twine("unsupported object file type: ") +
1852                          Obj.getFileName());
1853
1854   if (ValidRelocs.empty())
1855     return false;
1856
1857   // Sort the relocations by offset. We will walk the DIEs linearly in
1858   // the file, this allows us to just keep an index in the relocation
1859   // array that we advance during our walk, rather than resorting to
1860   // some associative container. See DwarfLinker::NextValidReloc.
1861   std::sort(ValidRelocs.begin(), ValidRelocs.end());
1862   return true;
1863 }
1864
1865 /// \brief Look for relocations in the debug_info section that match
1866 /// entries in the debug map. These relocations will drive the Dwarf
1867 /// link by indicating which DIEs refer to symbols present in the
1868 /// linked binary.
1869 /// \returns wether there are any valid relocations in the debug info.
1870 bool DwarfLinker::RelocationManager::
1871 findValidRelocsInDebugInfo(const object::ObjectFile &Obj,
1872                            const DebugMapObject &DMO) {
1873   // Find the debug_info section.
1874   for (const object::SectionRef &Section : Obj.sections()) {
1875     StringRef SectionName;
1876     Section.getName(SectionName);
1877     SectionName = SectionName.substr(SectionName.find_first_not_of("._"));
1878     if (SectionName != "debug_info")
1879       continue;
1880     return findValidRelocs(Section, Obj, DMO);
1881   }
1882   return false;
1883 }
1884
1885 /// \brief Checks that there is a relocation against an actual debug
1886 /// map entry between \p StartOffset and \p NextOffset.
1887 ///
1888 /// This function must be called with offsets in strictly ascending
1889 /// order because it never looks back at relocations it already 'went past'.
1890 /// \returns true and sets Info.InDebugMap if it is the case.
1891 bool DwarfLinker::RelocationManager::
1892 hasValidRelocation(uint32_t StartOffset, uint32_t EndOffset,
1893                    CompileUnit::DIEInfo &Info) {
1894   assert(NextValidReloc == 0 ||
1895          StartOffset > ValidRelocs[NextValidReloc - 1].Offset);
1896   if (NextValidReloc >= ValidRelocs.size())
1897     return false;
1898
1899   uint64_t RelocOffset = ValidRelocs[NextValidReloc].Offset;
1900
1901   // We might need to skip some relocs that we didn't consider. For
1902   // example the high_pc of a discarded DIE might contain a reloc that
1903   // is in the list because it actually corresponds to the start of a
1904   // function that is in the debug map.
1905   while (RelocOffset < StartOffset && NextValidReloc < ValidRelocs.size() - 1)
1906     RelocOffset = ValidRelocs[++NextValidReloc].Offset;
1907
1908   if (RelocOffset < StartOffset || RelocOffset >= EndOffset)
1909     return false;
1910
1911   const auto &ValidReloc = ValidRelocs[NextValidReloc++];
1912   const auto &Mapping = ValidReloc.Mapping->getValue();
1913   if (Linker.Options.Verbose)
1914     outs() << "Found valid debug map entry: " << ValidReloc.Mapping->getKey()
1915            << " " << format("\t%016" PRIx64 " => %016" PRIx64,
1916                             uint64_t(Mapping.ObjectAddress),
1917                             uint64_t(Mapping.BinaryAddress));
1918
1919   Info.AddrAdjust = int64_t(Mapping.BinaryAddress) + ValidReloc.Addend -
1920                     Mapping.ObjectAddress;
1921   Info.InDebugMap = true;
1922   return true;
1923 }
1924
1925 /// \brief Get the starting and ending (exclusive) offset for the
1926 /// attribute with index \p Idx descibed by \p Abbrev. \p Offset is
1927 /// supposed to point to the position of the first attribute described
1928 /// by \p Abbrev.
1929 /// \return [StartOffset, EndOffset) as a pair.
1930 static std::pair<uint32_t, uint32_t>
1931 getAttributeOffsets(const DWARFAbbreviationDeclaration *Abbrev, unsigned Idx,
1932                     unsigned Offset, const DWARFUnit &Unit) {
1933   DataExtractor Data = Unit.getDebugInfoExtractor();
1934
1935   for (unsigned i = 0; i < Idx; ++i)
1936     DWARFFormValue::skipValue(Abbrev->getFormByIndex(i), Data, &Offset, &Unit);
1937
1938   uint32_t End = Offset;
1939   DWARFFormValue::skipValue(Abbrev->getFormByIndex(Idx), Data, &End, &Unit);
1940
1941   return std::make_pair(Offset, End);
1942 }
1943
1944 /// \brief Check if a variable describing DIE should be kept.
1945 /// \returns updated TraversalFlags.
1946 unsigned DwarfLinker::shouldKeepVariableDIE(RelocationManager &RelocMgr,
1947                                             const DWARFDebugInfoEntryMinimal &DIE,
1948                                             CompileUnit &Unit,
1949                                             CompileUnit::DIEInfo &MyInfo,
1950                                             unsigned Flags) {
1951   const auto *Abbrev = DIE.getAbbreviationDeclarationPtr();
1952
1953   // Global variables with constant value can always be kept.
1954   if (!(Flags & TF_InFunctionScope) &&
1955       Abbrev->findAttributeIndex(dwarf::DW_AT_const_value) != -1U) {
1956     MyInfo.InDebugMap = true;
1957     return Flags | TF_Keep;
1958   }
1959
1960   uint32_t LocationIdx = Abbrev->findAttributeIndex(dwarf::DW_AT_location);
1961   if (LocationIdx == -1U)
1962     return Flags;
1963
1964   uint32_t Offset = DIE.getOffset() + getULEB128Size(Abbrev->getCode());
1965   const DWARFUnit &OrigUnit = Unit.getOrigUnit();
1966   uint32_t LocationOffset, LocationEndOffset;
1967   std::tie(LocationOffset, LocationEndOffset) =
1968       getAttributeOffsets(Abbrev, LocationIdx, Offset, OrigUnit);
1969
1970   // See if there is a relocation to a valid debug map entry inside
1971   // this variable's location. The order is important here. We want to
1972   // always check in the variable has a valid relocation, so that the
1973   // DIEInfo is filled. However, we don't want a static variable in a
1974   // function to force us to keep the enclosing function.
1975   if (!RelocMgr.hasValidRelocation(LocationOffset, LocationEndOffset, MyInfo) ||
1976       (Flags & TF_InFunctionScope))
1977     return Flags;
1978
1979   if (Options.Verbose)
1980     DIE.dump(outs(), const_cast<DWARFUnit *>(&OrigUnit), 0, 8 /* Indent */);
1981
1982   return Flags | TF_Keep;
1983 }
1984
1985 /// \brief Check if a function describing DIE should be kept.
1986 /// \returns updated TraversalFlags.
1987 unsigned DwarfLinker::shouldKeepSubprogramDIE(
1988     RelocationManager &RelocMgr,
1989     const DWARFDebugInfoEntryMinimal &DIE, CompileUnit &Unit,
1990     CompileUnit::DIEInfo &MyInfo, unsigned Flags) {
1991   const auto *Abbrev = DIE.getAbbreviationDeclarationPtr();
1992
1993   Flags |= TF_InFunctionScope;
1994
1995   uint32_t LowPcIdx = Abbrev->findAttributeIndex(dwarf::DW_AT_low_pc);
1996   if (LowPcIdx == -1U)
1997     return Flags;
1998
1999   uint32_t Offset = DIE.getOffset() + getULEB128Size(Abbrev->getCode());
2000   const DWARFUnit &OrigUnit = Unit.getOrigUnit();
2001   uint32_t LowPcOffset, LowPcEndOffset;
2002   std::tie(LowPcOffset, LowPcEndOffset) =
2003       getAttributeOffsets(Abbrev, LowPcIdx, Offset, OrigUnit);
2004
2005   uint64_t LowPc =
2006       DIE.getAttributeValueAsAddress(&OrigUnit, dwarf::DW_AT_low_pc, -1ULL);
2007   assert(LowPc != -1ULL && "low_pc attribute is not an address.");
2008   if (LowPc == -1ULL ||
2009       !RelocMgr.hasValidRelocation(LowPcOffset, LowPcEndOffset, MyInfo))
2010     return Flags;
2011
2012   if (Options.Verbose)
2013     DIE.dump(outs(), const_cast<DWARFUnit *>(&OrigUnit), 0, 8 /* Indent */);
2014
2015   Flags |= TF_Keep;
2016
2017   DWARFFormValue HighPcValue;
2018   if (!DIE.getAttributeValue(&OrigUnit, dwarf::DW_AT_high_pc, HighPcValue)) {
2019     reportWarning("Function without high_pc. Range will be discarded.\n",
2020                   &OrigUnit, &DIE);
2021     return Flags;
2022   }
2023
2024   uint64_t HighPc;
2025   if (HighPcValue.isFormClass(DWARFFormValue::FC_Address)) {
2026     HighPc = *HighPcValue.getAsAddress(&OrigUnit);
2027   } else {
2028     assert(HighPcValue.isFormClass(DWARFFormValue::FC_Constant));
2029     HighPc = LowPc + *HighPcValue.getAsUnsignedConstant();
2030   }
2031
2032   // Replace the debug map range with a more accurate one.
2033   Ranges[LowPc] = std::make_pair(HighPc, MyInfo.AddrAdjust);
2034   Unit.addFunctionRange(LowPc, HighPc, MyInfo.AddrAdjust);
2035   return Flags;
2036 }
2037
2038 /// \brief Check if a DIE should be kept.
2039 /// \returns updated TraversalFlags.
2040 unsigned DwarfLinker::shouldKeepDIE(RelocationManager &RelocMgr,
2041                                     const DWARFDebugInfoEntryMinimal &DIE,
2042                                     CompileUnit &Unit,
2043                                     CompileUnit::DIEInfo &MyInfo,
2044                                     unsigned Flags) {
2045   switch (DIE.getTag()) {
2046   case dwarf::DW_TAG_constant:
2047   case dwarf::DW_TAG_variable:
2048     return shouldKeepVariableDIE(RelocMgr, DIE, Unit, MyInfo, Flags);
2049   case dwarf::DW_TAG_subprogram:
2050     return shouldKeepSubprogramDIE(RelocMgr, DIE, Unit, MyInfo, Flags);
2051   case dwarf::DW_TAG_module:
2052   case dwarf::DW_TAG_imported_module:
2053   case dwarf::DW_TAG_imported_declaration:
2054   case dwarf::DW_TAG_imported_unit:
2055     // We always want to keep these.
2056     return Flags | TF_Keep;
2057   }
2058
2059   return Flags;
2060 }
2061
2062 /// \brief Mark the passed DIE as well as all the ones it depends on
2063 /// as kept.
2064 ///
2065 /// This function is called by lookForDIEsToKeep on DIEs that are
2066 /// newly discovered to be needed in the link. It recursively calls
2067 /// back to lookForDIEsToKeep while adding TF_DependencyWalk to the
2068 /// TraversalFlags to inform it that it's not doing the primary DIE
2069 /// tree walk.
2070 void DwarfLinker::keepDIEAndDenpendencies(RelocationManager &RelocMgr,
2071                                           const DWARFDebugInfoEntryMinimal &Die,
2072                                           CompileUnit::DIEInfo &MyInfo,
2073                                           const DebugMapObject &DMO,
2074                                           CompileUnit &CU, bool UseODR) {
2075   const DWARFUnit &Unit = CU.getOrigUnit();
2076   MyInfo.Keep = true;
2077
2078   // First mark all the parent chain as kept.
2079   unsigned AncestorIdx = MyInfo.ParentIdx;
2080   while (!CU.getInfo(AncestorIdx).Keep) {
2081     unsigned ODRFlag = UseODR ? TF_ODR : 0;
2082     lookForDIEsToKeep(RelocMgr, *Unit.getDIEAtIndex(AncestorIdx), DMO, CU,
2083                       TF_ParentWalk | TF_Keep | TF_DependencyWalk | ODRFlag);
2084     AncestorIdx = CU.getInfo(AncestorIdx).ParentIdx;
2085   }
2086
2087   // Then we need to mark all the DIEs referenced by this DIE's
2088   // attributes as kept.
2089   DataExtractor Data = Unit.getDebugInfoExtractor();
2090   const auto *Abbrev = Die.getAbbreviationDeclarationPtr();
2091   uint32_t Offset = Die.getOffset() + getULEB128Size(Abbrev->getCode());
2092
2093   // Mark all DIEs referenced through atttributes as kept.
2094   for (const auto &AttrSpec : Abbrev->attributes()) {
2095     DWARFFormValue Val(AttrSpec.Form);
2096
2097     if (!Val.isFormClass(DWARFFormValue::FC_Reference)) {
2098       DWARFFormValue::skipValue(AttrSpec.Form, Data, &Offset, &Unit);
2099       continue;
2100     }
2101
2102     Val.extractValue(Data, &Offset, &Unit);
2103     CompileUnit *ReferencedCU;
2104     if (const auto *RefDIE =
2105             resolveDIEReference(Val, Unit, Die, ReferencedCU)) {
2106       uint32_t RefIdx = ReferencedCU->getOrigUnit().getDIEIndex(RefDIE);
2107       CompileUnit::DIEInfo &Info = ReferencedCU->getInfo(RefIdx);
2108       // If the referenced DIE has a DeclContext that has already been
2109       // emitted, then do not keep the one in this CU. We'll link to
2110       // the canonical DIE in cloneDieReferenceAttribute.
2111       // FIXME: compatibility with dsymutil-classic. UseODR shouldn't
2112       // be necessary and could be advantageously replaced by
2113       // ReferencedCU->hasODR() && CU.hasODR().
2114       // FIXME: compatibility with dsymutil-classic. There is no
2115       // reason not to unique ref_addr references.
2116       if (AttrSpec.Form != dwarf::DW_FORM_ref_addr && UseODR && Info.Ctxt &&
2117           Info.Ctxt != ReferencedCU->getInfo(Info.ParentIdx).Ctxt &&
2118           Info.Ctxt->getCanonicalDIEOffset() && isODRAttribute(AttrSpec.Attr))
2119         continue;
2120
2121       unsigned ODRFlag = UseODR ? TF_ODR : 0;
2122       lookForDIEsToKeep(RelocMgr, *RefDIE, DMO, *ReferencedCU,
2123                         TF_Keep | TF_DependencyWalk | ODRFlag);
2124     }
2125   }
2126 }
2127
2128 /// \brief Recursively walk the \p DIE tree and look for DIEs to
2129 /// keep. Store that information in \p CU's DIEInfo.
2130 ///
2131 /// This function is the entry point of the DIE selection
2132 /// algorithm. It is expected to walk the DIE tree in file order and
2133 /// (though the mediation of its helper) call hasValidRelocation() on
2134 /// each DIE that might be a 'root DIE' (See DwarfLinker class
2135 /// comment).
2136 /// While walking the dependencies of root DIEs, this function is
2137 /// also called, but during these dependency walks the file order is
2138 /// not respected. The TF_DependencyWalk flag tells us which kind of
2139 /// traversal we are currently doing.
2140 void DwarfLinker::lookForDIEsToKeep(RelocationManager &RelocMgr,
2141                                     const DWARFDebugInfoEntryMinimal &Die,
2142                                     const DebugMapObject &DMO, CompileUnit &CU,
2143                                     unsigned Flags) {
2144   unsigned Idx = CU.getOrigUnit().getDIEIndex(&Die);
2145   CompileUnit::DIEInfo &MyInfo = CU.getInfo(Idx);
2146   bool AlreadyKept = MyInfo.Keep;
2147
2148   // If the Keep flag is set, we are marking a required DIE's
2149   // dependencies. If our target is already marked as kept, we're all
2150   // set.
2151   if ((Flags & TF_DependencyWalk) && AlreadyKept)
2152     return;
2153
2154   // We must not call shouldKeepDIE while called from keepDIEAndDenpendencies,
2155   // because it would screw up the relocation finding logic.
2156   if (!(Flags & TF_DependencyWalk))
2157     Flags = shouldKeepDIE(RelocMgr, Die, CU, MyInfo, Flags);
2158
2159   // If it is a newly kept DIE mark it as well as all its dependencies as kept.
2160   if (!AlreadyKept && (Flags & TF_Keep)) {
2161     bool UseOdr = (Flags & TF_DependencyWalk) ? (Flags & TF_ODR) : CU.hasODR();
2162     keepDIEAndDenpendencies(RelocMgr, Die, MyInfo, DMO, CU, UseOdr);
2163   }
2164   // The TF_ParentWalk flag tells us that we are currently walking up
2165   // the parent chain of a required DIE, and we don't want to mark all
2166   // the children of the parents as kept (consider for example a
2167   // DW_TAG_namespace node in the parent chain). There are however a
2168   // set of DIE types for which we want to ignore that directive and still
2169   // walk their children.
2170   if (dieNeedsChildrenToBeMeaningful(Die.getTag()))
2171     Flags &= ~TF_ParentWalk;
2172
2173   if (!Die.hasChildren() || (Flags & TF_ParentWalk))
2174     return;
2175
2176   for (auto *Child = Die.getFirstChild(); Child && !Child->isNULL();
2177        Child = Child->getSibling())
2178     lookForDIEsToKeep(RelocMgr, *Child, DMO, CU, Flags);
2179 }
2180
2181 /// \brief Assign an abbreviation numer to \p Abbrev.
2182 ///
2183 /// Our DIEs get freed after every DebugMapObject has been processed,
2184 /// thus the FoldingSet we use to unique DIEAbbrevs cannot refer to
2185 /// the instances hold by the DIEs. When we encounter an abbreviation
2186 /// that we don't know, we create a permanent copy of it.
2187 void DwarfLinker::AssignAbbrev(DIEAbbrev &Abbrev) {
2188   // Check the set for priors.
2189   FoldingSetNodeID ID;
2190   Abbrev.Profile(ID);
2191   void *InsertToken;
2192   DIEAbbrev *InSet = AbbreviationsSet.FindNodeOrInsertPos(ID, InsertToken);
2193
2194   // If it's newly added.
2195   if (InSet) {
2196     // Assign existing abbreviation number.
2197     Abbrev.setNumber(InSet->getNumber());
2198   } else {
2199     // Add to abbreviation list.
2200     Abbreviations.push_back(
2201         new DIEAbbrev(Abbrev.getTag(), Abbrev.hasChildren()));
2202     for (const auto &Attr : Abbrev.getData())
2203       Abbreviations.back()->AddAttribute(Attr.getAttribute(), Attr.getForm());
2204     AbbreviationsSet.InsertNode(Abbreviations.back(), InsertToken);
2205     // Assign the unique abbreviation number.
2206     Abbrev.setNumber(Abbreviations.size());
2207     Abbreviations.back()->setNumber(Abbreviations.size());
2208   }
2209 }
2210
2211 unsigned DwarfLinker::DIECloner::cloneStringAttribute(DIE &Die,
2212                                                       AttributeSpec AttrSpec,
2213                                                       const DWARFFormValue &Val,
2214                                                       const DWARFUnit &U) {
2215   // Switch everything to out of line strings.
2216   const char *String = *Val.getAsCString(&U);
2217   unsigned Offset = Linker.StringPool.getStringOffset(String);
2218   Die.addValue(DIEAlloc, dwarf::Attribute(AttrSpec.Attr), dwarf::DW_FORM_strp,
2219                DIEInteger(Offset));
2220   return 4;
2221 }
2222
2223 unsigned DwarfLinker::DIECloner::cloneDieReferenceAttribute(
2224     DIE &Die, const DWARFDebugInfoEntryMinimal &InputDIE,
2225     AttributeSpec AttrSpec, unsigned AttrSize, const DWARFFormValue &Val,
2226     CompileUnit &Unit) {
2227   const DWARFUnit &U = Unit.getOrigUnit();
2228   uint32_t Ref = *Val.getAsReference(&U);
2229   DIE *NewRefDie = nullptr;
2230   CompileUnit *RefUnit = nullptr;
2231   DeclContext *Ctxt = nullptr;
2232
2233   const DWARFDebugInfoEntryMinimal *RefDie =
2234       Linker.resolveDIEReference(Val, U, InputDIE, RefUnit);
2235
2236   // If the referenced DIE is not found,  drop the attribute.
2237   if (!RefDie)
2238     return 0;
2239
2240   unsigned Idx = RefUnit->getOrigUnit().getDIEIndex(RefDie);
2241   CompileUnit::DIEInfo &RefInfo = RefUnit->getInfo(Idx);
2242
2243   // If we already have emitted an equivalent DeclContext, just point
2244   // at it.
2245   if (isODRAttribute(AttrSpec.Attr)) {
2246     Ctxt = RefInfo.Ctxt;
2247     if (Ctxt && Ctxt->getCanonicalDIEOffset()) {
2248       DIEInteger Attr(Ctxt->getCanonicalDIEOffset());
2249       Die.addValue(DIEAlloc, dwarf::Attribute(AttrSpec.Attr),
2250                    dwarf::DW_FORM_ref_addr, Attr);
2251       return getRefAddrSize(U);
2252     }
2253   }
2254
2255   if (!RefInfo.Clone) {
2256     assert(Ref > InputDIE.getOffset());
2257     // We haven't cloned this DIE yet. Just create an empty one and
2258     // store it. It'll get really cloned when we process it.
2259     RefInfo.Clone = DIE::get(DIEAlloc, dwarf::Tag(RefDie->getTag()));
2260   }
2261   NewRefDie = RefInfo.Clone;
2262
2263   if (AttrSpec.Form == dwarf::DW_FORM_ref_addr ||
2264       (Unit.hasODR() && isODRAttribute(AttrSpec.Attr))) {
2265     // We cannot currently rely on a DIEEntry to emit ref_addr
2266     // references, because the implementation calls back to DwarfDebug
2267     // to find the unit offset. (We don't have a DwarfDebug)
2268     // FIXME: we should be able to design DIEEntry reliance on
2269     // DwarfDebug away.
2270     uint64_t Attr;
2271     if (Ref < InputDIE.getOffset()) {
2272       // We must have already cloned that DIE.
2273       uint32_t NewRefOffset =
2274           RefUnit->getStartOffset() + NewRefDie->getOffset();
2275       Attr = NewRefOffset;
2276       Die.addValue(DIEAlloc, dwarf::Attribute(AttrSpec.Attr),
2277                    dwarf::DW_FORM_ref_addr, DIEInteger(Attr));
2278     } else {
2279       // A forward reference. Note and fixup later.
2280       Attr = 0xBADDEF;
2281       Unit.noteForwardReference(
2282           NewRefDie, RefUnit, Ctxt,
2283           Die.addValue(DIEAlloc, dwarf::Attribute(AttrSpec.Attr),
2284                        dwarf::DW_FORM_ref_addr, DIEInteger(Attr)));
2285     }
2286     return getRefAddrSize(U);
2287   }
2288
2289   Die.addValue(DIEAlloc, dwarf::Attribute(AttrSpec.Attr),
2290                dwarf::Form(AttrSpec.Form), DIEEntry(*NewRefDie));
2291   return AttrSize;
2292 }
2293
2294 unsigned DwarfLinker::DIECloner::cloneBlockAttribute(DIE &Die,
2295                                                      AttributeSpec AttrSpec,
2296                                                      const DWARFFormValue &Val,
2297                                                      unsigned AttrSize) {
2298   DIEValueList *Attr;
2299   DIEValue Value;
2300   DIELoc *Loc = nullptr;
2301   DIEBlock *Block = nullptr;
2302   // Just copy the block data over.
2303   if (AttrSpec.Form == dwarf::DW_FORM_exprloc) {
2304     Loc = new (DIEAlloc) DIELoc;
2305     Linker.DIELocs.push_back(Loc);
2306   } else {
2307     Block = new (DIEAlloc) DIEBlock;
2308     Linker.DIEBlocks.push_back(Block);
2309   }
2310   Attr = Loc ? static_cast<DIEValueList *>(Loc)
2311              : static_cast<DIEValueList *>(Block);
2312
2313   if (Loc)
2314     Value = DIEValue(dwarf::Attribute(AttrSpec.Attr),
2315                      dwarf::Form(AttrSpec.Form), Loc);
2316   else
2317     Value = DIEValue(dwarf::Attribute(AttrSpec.Attr),
2318                      dwarf::Form(AttrSpec.Form), Block);
2319   ArrayRef<uint8_t> Bytes = *Val.getAsBlock();
2320   for (auto Byte : Bytes)
2321     Attr->addValue(DIEAlloc, static_cast<dwarf::Attribute>(0),
2322                    dwarf::DW_FORM_data1, DIEInteger(Byte));
2323   // FIXME: If DIEBlock and DIELoc just reuses the Size field of
2324   // the DIE class, this if could be replaced by
2325   // Attr->setSize(Bytes.size()).
2326   if (Linker.Streamer) {
2327     auto *AsmPrinter = &Linker.Streamer->getAsmPrinter();
2328     if (Loc)
2329       Loc->ComputeSize(AsmPrinter);
2330     else
2331       Block->ComputeSize(AsmPrinter);
2332   }
2333   Die.addValue(DIEAlloc, Value);
2334   return AttrSize;
2335 }
2336
2337 unsigned DwarfLinker::DIECloner::cloneAddressAttribute(
2338     DIE &Die, AttributeSpec AttrSpec, const DWARFFormValue &Val,
2339     const CompileUnit &Unit, AttributesInfo &Info) {
2340   uint64_t Addr = *Val.getAsAddress(&Unit.getOrigUnit());
2341   if (AttrSpec.Attr == dwarf::DW_AT_low_pc) {
2342     if (Die.getTag() == dwarf::DW_TAG_inlined_subroutine ||
2343         Die.getTag() == dwarf::DW_TAG_lexical_block)
2344       // The low_pc of a block or inline subroutine might get
2345       // relocated because it happens to match the low_pc of the
2346       // enclosing subprogram. To prevent issues with that, always use
2347       // the low_pc from the input DIE if relocations have been applied.
2348       Addr = (Info.OrigLowPc != UINT64_MAX ? Info.OrigLowPc : Addr) +
2349              Info.PCOffset;
2350     else if (Die.getTag() == dwarf::DW_TAG_compile_unit) {
2351       Addr = Unit.getLowPc();
2352       if (Addr == UINT64_MAX)
2353         return 0;
2354     }
2355     Info.HasLowPc = true;
2356   } else if (AttrSpec.Attr == dwarf::DW_AT_high_pc) {
2357     if (Die.getTag() == dwarf::DW_TAG_compile_unit) {
2358       if (uint64_t HighPc = Unit.getHighPc())
2359         Addr = HighPc;
2360       else
2361         return 0;
2362     } else
2363       // If we have a high_pc recorded for the input DIE, use
2364       // it. Otherwise (when no relocations where applied) just use the
2365       // one we just decoded.
2366       Addr = (Info.OrigHighPc ? Info.OrigHighPc : Addr) + Info.PCOffset;
2367   }
2368
2369   Die.addValue(DIEAlloc, static_cast<dwarf::Attribute>(AttrSpec.Attr),
2370                static_cast<dwarf::Form>(AttrSpec.Form), DIEInteger(Addr));
2371   return Unit.getOrigUnit().getAddressByteSize();
2372 }
2373
2374 unsigned DwarfLinker::DIECloner::cloneScalarAttribute(
2375     DIE &Die, const DWARFDebugInfoEntryMinimal &InputDIE, CompileUnit &Unit,
2376     AttributeSpec AttrSpec, const DWARFFormValue &Val, unsigned AttrSize,
2377     AttributesInfo &Info) {
2378   uint64_t Value;
2379   if (AttrSpec.Attr == dwarf::DW_AT_high_pc &&
2380       Die.getTag() == dwarf::DW_TAG_compile_unit) {
2381     if (Unit.getLowPc() == -1ULL)
2382       return 0;
2383     // Dwarf >= 4 high_pc is an size, not an address.
2384     Value = Unit.getHighPc() - Unit.getLowPc();
2385   } else if (AttrSpec.Form == dwarf::DW_FORM_sec_offset)
2386     Value = *Val.getAsSectionOffset();
2387   else if (AttrSpec.Form == dwarf::DW_FORM_sdata)
2388     Value = *Val.getAsSignedConstant();
2389   else if (auto OptionalValue = Val.getAsUnsignedConstant())
2390     Value = *OptionalValue;
2391   else {
2392     Linker.reportWarning(
2393         "Unsupported scalar attribute form. Dropping attribute.",
2394         &Unit.getOrigUnit(), &InputDIE);
2395     return 0;
2396   }
2397   PatchLocation Patch =
2398       Die.addValue(DIEAlloc, dwarf::Attribute(AttrSpec.Attr),
2399                    dwarf::Form(AttrSpec.Form), DIEInteger(Value));
2400   if (AttrSpec.Attr == dwarf::DW_AT_ranges)
2401     Unit.noteRangeAttribute(Die, Patch);
2402
2403   // A more generic way to check for location attributes would be
2404   // nice, but it's very unlikely that any other attribute needs a
2405   // location list.
2406   else if (AttrSpec.Attr == dwarf::DW_AT_location ||
2407            AttrSpec.Attr == dwarf::DW_AT_frame_base)
2408     Unit.noteLocationAttribute(Patch, Info.PCOffset);
2409   else if (AttrSpec.Attr == dwarf::DW_AT_declaration && Value)
2410     Info.IsDeclaration = true;
2411
2412   return AttrSize;
2413 }
2414
2415 /// \brief Clone \p InputDIE's attribute described by \p AttrSpec with
2416 /// value \p Val, and add it to \p Die.
2417 /// \returns the size of the cloned attribute.
2418 unsigned DwarfLinker::DIECloner::cloneAttribute(
2419     DIE &Die, const DWARFDebugInfoEntryMinimal &InputDIE, CompileUnit &Unit,
2420     const DWARFFormValue &Val, const AttributeSpec AttrSpec, unsigned AttrSize,
2421     AttributesInfo &Info) {
2422   const DWARFUnit &U = Unit.getOrigUnit();
2423
2424   switch (AttrSpec.Form) {
2425   case dwarf::DW_FORM_strp:
2426   case dwarf::DW_FORM_string:
2427     return cloneStringAttribute(Die, AttrSpec, Val, U);
2428   case dwarf::DW_FORM_ref_addr:
2429   case dwarf::DW_FORM_ref1:
2430   case dwarf::DW_FORM_ref2:
2431   case dwarf::DW_FORM_ref4:
2432   case dwarf::DW_FORM_ref8:
2433     return cloneDieReferenceAttribute(Die, InputDIE, AttrSpec, AttrSize, Val,
2434                                       Unit);
2435   case dwarf::DW_FORM_block:
2436   case dwarf::DW_FORM_block1:
2437   case dwarf::DW_FORM_block2:
2438   case dwarf::DW_FORM_block4:
2439   case dwarf::DW_FORM_exprloc:
2440     return cloneBlockAttribute(Die, AttrSpec, Val, AttrSize);
2441   case dwarf::DW_FORM_addr:
2442     return cloneAddressAttribute(Die, AttrSpec, Val, Unit, Info);
2443   case dwarf::DW_FORM_data1:
2444   case dwarf::DW_FORM_data2:
2445   case dwarf::DW_FORM_data4:
2446   case dwarf::DW_FORM_data8:
2447   case dwarf::DW_FORM_udata:
2448   case dwarf::DW_FORM_sdata:
2449   case dwarf::DW_FORM_sec_offset:
2450   case dwarf::DW_FORM_flag:
2451   case dwarf::DW_FORM_flag_present:
2452     return cloneScalarAttribute(Die, InputDIE, Unit, AttrSpec, Val, AttrSize,
2453                                 Info);
2454   default:
2455     Linker.reportWarning(
2456         "Unsupported attribute form in cloneAttribute. Dropping.", &U,
2457         &InputDIE);
2458   }
2459
2460   return 0;
2461 }
2462
2463 /// \brief Apply the valid relocations found by findValidRelocs() to
2464 /// the buffer \p Data, taking into account that Data is at \p BaseOffset
2465 /// in the debug_info section.
2466 ///
2467 /// Like for findValidRelocs(), this function must be called with
2468 /// monotonic \p BaseOffset values.
2469 ///
2470 /// \returns wether any reloc has been applied.
2471 bool DwarfLinker::RelocationManager::
2472 applyValidRelocs(MutableArrayRef<char> Data, uint32_t BaseOffset,
2473                  bool isLittleEndian) {
2474   assert((NextValidReloc == 0 ||
2475           BaseOffset > ValidRelocs[NextValidReloc - 1].Offset) &&
2476          "BaseOffset should only be increasing.");
2477   if (NextValidReloc >= ValidRelocs.size())
2478     return false;
2479
2480   // Skip relocs that haven't been applied.
2481   while (NextValidReloc < ValidRelocs.size() &&
2482          ValidRelocs[NextValidReloc].Offset < BaseOffset)
2483     ++NextValidReloc;
2484
2485   bool Applied = false;
2486   uint64_t EndOffset = BaseOffset + Data.size();
2487   while (NextValidReloc < ValidRelocs.size() &&
2488          ValidRelocs[NextValidReloc].Offset >= BaseOffset &&
2489          ValidRelocs[NextValidReloc].Offset < EndOffset) {
2490     const auto &ValidReloc = ValidRelocs[NextValidReloc++];
2491     assert(ValidReloc.Offset - BaseOffset < Data.size());
2492     assert(ValidReloc.Offset - BaseOffset + ValidReloc.Size <= Data.size());
2493     char Buf[8];
2494     uint64_t Value = ValidReloc.Mapping->getValue().BinaryAddress;
2495     Value += ValidReloc.Addend;
2496     for (unsigned i = 0; i != ValidReloc.Size; ++i) {
2497       unsigned Index = isLittleEndian ? i : (ValidReloc.Size - i - 1);
2498       Buf[i] = uint8_t(Value >> (Index * 8));
2499     }
2500     assert(ValidReloc.Size <= sizeof(Buf));
2501     memcpy(&Data[ValidReloc.Offset - BaseOffset], Buf, ValidReloc.Size);
2502     Applied = true;
2503   }
2504
2505   return Applied;
2506 }
2507
2508 static bool isTypeTag(uint16_t Tag) {
2509   switch (Tag) {
2510   case dwarf::DW_TAG_array_type:
2511   case dwarf::DW_TAG_class_type:
2512   case dwarf::DW_TAG_enumeration_type:
2513   case dwarf::DW_TAG_pointer_type:
2514   case dwarf::DW_TAG_reference_type:
2515   case dwarf::DW_TAG_string_type:
2516   case dwarf::DW_TAG_structure_type:
2517   case dwarf::DW_TAG_subroutine_type:
2518   case dwarf::DW_TAG_typedef:
2519   case dwarf::DW_TAG_union_type:
2520   case dwarf::DW_TAG_ptr_to_member_type:
2521   case dwarf::DW_TAG_set_type:
2522   case dwarf::DW_TAG_subrange_type:
2523   case dwarf::DW_TAG_base_type:
2524   case dwarf::DW_TAG_const_type:
2525   case dwarf::DW_TAG_constant:
2526   case dwarf::DW_TAG_file_type:
2527   case dwarf::DW_TAG_namelist:
2528   case dwarf::DW_TAG_packed_type:
2529   case dwarf::DW_TAG_volatile_type:
2530   case dwarf::DW_TAG_restrict_type:
2531   case dwarf::DW_TAG_interface_type:
2532   case dwarf::DW_TAG_unspecified_type:
2533   case dwarf::DW_TAG_shared_type:
2534     return true;
2535   default:
2536     break;
2537   }
2538   return false;
2539 }
2540
2541 static bool
2542 shouldSkipAttribute(DWARFAbbreviationDeclaration::AttributeSpec AttrSpec,
2543                     uint16_t Tag, bool InDebugMap, bool SkipPC,
2544                     bool InFunctionScope) {
2545   switch (AttrSpec.Attr) {
2546   default:
2547     return false;
2548   case dwarf::DW_AT_low_pc:
2549   case dwarf::DW_AT_high_pc:
2550   case dwarf::DW_AT_ranges:
2551     return SkipPC;
2552   case dwarf::DW_AT_location:
2553   case dwarf::DW_AT_frame_base:
2554     // FIXME: for some reason dsymutil-classic keeps the location
2555     // attributes when they are of block type (ie. not location
2556     // lists). This is totally wrong for globals where we will keep a
2557     // wrong address. It is mostly harmless for locals, but there is
2558     // no point in keeping these anyway when the function wasn't linked.
2559     return (SkipPC || (!InFunctionScope && Tag == dwarf::DW_TAG_variable &&
2560                        !InDebugMap)) &&
2561            !DWARFFormValue(AttrSpec.Form).isFormClass(DWARFFormValue::FC_Block);
2562   }
2563 }
2564
2565 DIE *DwarfLinker::DIECloner::cloneDIE(
2566     const DWARFDebugInfoEntryMinimal &InputDIE, CompileUnit &Unit,
2567     int64_t PCOffset, uint32_t OutOffset, unsigned Flags) {
2568   DWARFUnit &U = Unit.getOrigUnit();
2569   unsigned Idx = U.getDIEIndex(&InputDIE);
2570   CompileUnit::DIEInfo &Info = Unit.getInfo(Idx);
2571
2572   // Should the DIE appear in the output?
2573   if (!Unit.getInfo(Idx).Keep)
2574     return nullptr;
2575
2576   uint32_t Offset = InputDIE.getOffset();
2577   // The DIE might have been already created by a forward reference
2578   // (see cloneDieReferenceAttribute()).
2579   DIE *Die = Info.Clone;
2580   if (!Die)
2581     Die = Info.Clone = DIE::get(DIEAlloc, dwarf::Tag(InputDIE.getTag()));
2582   assert(Die->getTag() == InputDIE.getTag());
2583   Die->setOffset(OutOffset);
2584   if (Unit.hasODR() && Die->getTag() != dwarf::DW_TAG_namespace && Info.Ctxt &&
2585       Info.Ctxt != Unit.getInfo(Info.ParentIdx).Ctxt &&
2586       !Info.Ctxt->getCanonicalDIEOffset()) {
2587     // We are about to emit a DIE that is the root of its own valid
2588     // DeclContext tree. Make the current offset the canonical offset
2589     // for this context.
2590     Info.Ctxt->setCanonicalDIEOffset(OutOffset + Unit.getStartOffset());
2591   }
2592
2593   // Extract and clone every attribute.
2594   DataExtractor Data = U.getDebugInfoExtractor();
2595   uint32_t NextOffset = U.getDIEAtIndex(Idx + 1)->getOffset();
2596   AttributesInfo AttrInfo;
2597
2598   // We could copy the data only if we need to aply a relocation to
2599   // it. After testing, it seems there is no performance downside to
2600   // doing the copy unconditionally, and it makes the code simpler.
2601   SmallString<40> DIECopy(Data.getData().substr(Offset, NextOffset - Offset));
2602   Data = DataExtractor(DIECopy, Data.isLittleEndian(), Data.getAddressSize());
2603   // Modify the copy with relocated addresses.
2604   if (RelocMgr.applyValidRelocs(DIECopy, Offset, Data.isLittleEndian())) {
2605     // If we applied relocations, we store the value of high_pc that was
2606     // potentially stored in the input DIE. If high_pc is an address
2607     // (Dwarf version == 2), then it might have been relocated to a
2608     // totally unrelated value (because the end address in the object
2609     // file might be start address of another function which got moved
2610     // independantly by the linker). The computation of the actual
2611     // high_pc value is done in cloneAddressAttribute().
2612     AttrInfo.OrigHighPc =
2613         InputDIE.getAttributeValueAsAddress(&U, dwarf::DW_AT_high_pc, 0);
2614     // Also store the low_pc. It might get relocated in an
2615     // inline_subprogram that happens at the beginning of its
2616     // inlining function.
2617     AttrInfo.OrigLowPc =
2618         InputDIE.getAttributeValueAsAddress(&U, dwarf::DW_AT_low_pc, UINT64_MAX);
2619   }
2620
2621   // Reset the Offset to 0 as we will be working on the local copy of
2622   // the data.
2623   Offset = 0;
2624
2625   const auto *Abbrev = InputDIE.getAbbreviationDeclarationPtr();
2626   Offset += getULEB128Size(Abbrev->getCode());
2627
2628   // We are entering a subprogram. Get and propagate the PCOffset.
2629   if (Die->getTag() == dwarf::DW_TAG_subprogram)
2630     PCOffset = Info.AddrAdjust;
2631   AttrInfo.PCOffset = PCOffset;
2632
2633   if (Abbrev->getTag() == dwarf::DW_TAG_subprogram) {
2634     Flags |= TF_InFunctionScope;
2635     if (!Info.InDebugMap)
2636       Flags |= TF_SkipPC;
2637   }
2638
2639   bool Copied = false;
2640   for (const auto &AttrSpec : Abbrev->attributes()) {
2641     if (shouldSkipAttribute(AttrSpec, Die->getTag(), Info.InDebugMap,
2642                             Flags & TF_SkipPC, Flags & TF_InFunctionScope)) {
2643       DWARFFormValue::skipValue(AttrSpec.Form, Data, &Offset, &U);
2644       // FIXME: dsymutil-classic keeps the old abbreviation around
2645       // even if it's not used. We can remove this (and the copyAbbrev
2646       // helper) as soon as bit-for-bit compatibility is not a goal anymore.
2647       if (!Copied) {
2648         copyAbbrev(*InputDIE.getAbbreviationDeclarationPtr(), Unit.hasODR());
2649         Copied = true;
2650       }
2651       continue;
2652     }
2653
2654     DWARFFormValue Val(AttrSpec.Form);
2655     uint32_t AttrSize = Offset;
2656     Val.extractValue(Data, &Offset, &U);
2657     AttrSize = Offset - AttrSize;
2658
2659     OutOffset +=
2660         cloneAttribute(*Die, InputDIE, Unit, Val, AttrSpec, AttrSize, AttrInfo);
2661   }
2662
2663   // Look for accelerator entries.
2664   uint16_t Tag = InputDIE.getTag();
2665   // FIXME: This is slightly wrong. An inline_subroutine without a
2666   // low_pc, but with AT_ranges might be interesting to get into the
2667   // accelerator tables too. For now stick with dsymutil's behavior.
2668   if ((Info.InDebugMap || AttrInfo.HasLowPc) &&
2669       Tag != dwarf::DW_TAG_compile_unit &&
2670       getDIENames(InputDIE, Unit.getOrigUnit(), AttrInfo)) {
2671     if (AttrInfo.MangledName && AttrInfo.MangledName != AttrInfo.Name)
2672       Unit.addNameAccelerator(Die, AttrInfo.MangledName,
2673                               AttrInfo.MangledNameOffset,
2674                               Tag == dwarf::DW_TAG_inlined_subroutine);
2675     if (AttrInfo.Name)
2676       Unit.addNameAccelerator(Die, AttrInfo.Name, AttrInfo.NameOffset,
2677                               Tag == dwarf::DW_TAG_inlined_subroutine);
2678   } else if (isTypeTag(Tag) && !AttrInfo.IsDeclaration &&
2679              getDIENames(InputDIE, Unit.getOrigUnit(), AttrInfo)) {
2680     Unit.addTypeAccelerator(Die, AttrInfo.Name, AttrInfo.NameOffset);
2681   }
2682
2683   DIEAbbrev NewAbbrev = Die->generateAbbrev();
2684   // If a scope DIE is kept, we must have kept at least one child. If
2685   // it's not the case, we'll just be emitting one wasteful end of
2686   // children marker, but things won't break.
2687   if (InputDIE.hasChildren())
2688     NewAbbrev.setChildrenFlag(dwarf::DW_CHILDREN_yes);
2689   // Assign a permanent abbrev number
2690   Linker.AssignAbbrev(NewAbbrev);
2691   Die->setAbbrevNumber(NewAbbrev.getNumber());
2692
2693   // Add the size of the abbreviation number to the output offset.
2694   OutOffset += getULEB128Size(Die->getAbbrevNumber());
2695
2696   if (!Abbrev->hasChildren()) {
2697     // Update our size.
2698     Die->setSize(OutOffset - Die->getOffset());
2699     return Die;
2700   }
2701
2702   // Recursively clone children.
2703   for (auto *Child = InputDIE.getFirstChild(); Child && !Child->isNULL();
2704        Child = Child->getSibling()) {
2705     if (DIE *Clone = cloneDIE(*Child, Unit, PCOffset, OutOffset, Flags)) {
2706       Die->addChild(Clone);
2707       OutOffset = Clone->getOffset() + Clone->getSize();
2708     }
2709   }
2710
2711   // Account for the end of children marker.
2712   OutOffset += sizeof(int8_t);
2713   // Update our size.
2714   Die->setSize(OutOffset - Die->getOffset());
2715   return Die;
2716 }
2717
2718 /// \brief Patch the input object file relevant debug_ranges entries
2719 /// and emit them in the output file. Update the relevant attributes
2720 /// to point at the new entries.
2721 void DwarfLinker::patchRangesForUnit(const CompileUnit &Unit,
2722                                      DWARFContext &OrigDwarf) const {
2723   DWARFDebugRangeList RangeList;
2724   const auto &FunctionRanges = Unit.getFunctionRanges();
2725   unsigned AddressSize = Unit.getOrigUnit().getAddressByteSize();
2726   DataExtractor RangeExtractor(OrigDwarf.getRangeSection(),
2727                                OrigDwarf.isLittleEndian(), AddressSize);
2728   auto InvalidRange = FunctionRanges.end(), CurrRange = InvalidRange;
2729   DWARFUnit &OrigUnit = Unit.getOrigUnit();
2730   const auto *OrigUnitDie = OrigUnit.getUnitDIE(false);
2731   uint64_t OrigLowPc = OrigUnitDie->getAttributeValueAsAddress(
2732       &OrigUnit, dwarf::DW_AT_low_pc, -1ULL);
2733   // Ranges addresses are based on the unit's low_pc. Compute the
2734   // offset we need to apply to adapt to the the new unit's low_pc.
2735   int64_t UnitPcOffset = 0;
2736   if (OrigLowPc != -1ULL)
2737     UnitPcOffset = int64_t(OrigLowPc) - Unit.getLowPc();
2738
2739   for (const auto &RangeAttribute : Unit.getRangesAttributes()) {
2740     uint32_t Offset = RangeAttribute.get();
2741     RangeAttribute.set(Streamer->getRangesSectionSize());
2742     RangeList.extract(RangeExtractor, &Offset);
2743     const auto &Entries = RangeList.getEntries();
2744     if (!Entries.empty()) {
2745       const DWARFDebugRangeList::RangeListEntry &First = Entries.front();
2746
2747       if (CurrRange == InvalidRange ||
2748           First.StartAddress + OrigLowPc < CurrRange.start() ||
2749           First.StartAddress + OrigLowPc >= CurrRange.stop()) {
2750         CurrRange = FunctionRanges.find(First.StartAddress + OrigLowPc);
2751         if (CurrRange == InvalidRange ||
2752             CurrRange.start() > First.StartAddress + OrigLowPc) {
2753           reportWarning("no mapping for range.");
2754           continue;
2755         }
2756       }
2757     }
2758
2759     Streamer->emitRangesEntries(UnitPcOffset, OrigLowPc, CurrRange, Entries,
2760                                 AddressSize);
2761   }
2762 }
2763
2764 /// \brief Generate the debug_aranges entries for \p Unit and if the
2765 /// unit has a DW_AT_ranges attribute, also emit the debug_ranges
2766 /// contribution for this attribute.
2767 /// FIXME: this could actually be done right in patchRangesForUnit,
2768 /// but for the sake of initial bit-for-bit compatibility with legacy
2769 /// dsymutil, we have to do it in a delayed pass.
2770 void DwarfLinker::generateUnitRanges(CompileUnit &Unit) const {
2771   auto Attr = Unit.getUnitRangesAttribute();
2772   if (Attr)
2773     Attr->set(Streamer->getRangesSectionSize());
2774   Streamer->emitUnitRangesEntries(Unit, static_cast<bool>(Attr));
2775 }
2776
2777 /// \brief Insert the new line info sequence \p Seq into the current
2778 /// set of already linked line info \p Rows.
2779 static void insertLineSequence(std::vector<DWARFDebugLine::Row> &Seq,
2780                                std::vector<DWARFDebugLine::Row> &Rows) {
2781   if (Seq.empty())
2782     return;
2783
2784   if (!Rows.empty() && Rows.back().Address < Seq.front().Address) {
2785     Rows.insert(Rows.end(), Seq.begin(), Seq.end());
2786     Seq.clear();
2787     return;
2788   }
2789
2790   auto InsertPoint = std::lower_bound(
2791       Rows.begin(), Rows.end(), Seq.front(),
2792       [](const DWARFDebugLine::Row &LHS, const DWARFDebugLine::Row &RHS) {
2793         return LHS.Address < RHS.Address;
2794       });
2795
2796   // FIXME: this only removes the unneeded end_sequence if the
2797   // sequences have been inserted in order. using a global sort like
2798   // described in patchLineTableForUnit() and delaying the end_sequene
2799   // elimination to emitLineTableForUnit() we can get rid of all of them.
2800   if (InsertPoint != Rows.end() &&
2801       InsertPoint->Address == Seq.front().Address && InsertPoint->EndSequence) {
2802     *InsertPoint = Seq.front();
2803     Rows.insert(InsertPoint + 1, Seq.begin() + 1, Seq.end());
2804   } else {
2805     Rows.insert(InsertPoint, Seq.begin(), Seq.end());
2806   }
2807
2808   Seq.clear();
2809 }
2810
2811 static void patchStmtList(DIE &Die, DIEInteger Offset) {
2812   for (auto &V : Die.values())
2813     if (V.getAttribute() == dwarf::DW_AT_stmt_list) {
2814       V = DIEValue(V.getAttribute(), V.getForm(), Offset);
2815       return;
2816     }
2817
2818   llvm_unreachable("Didn't find DW_AT_stmt_list in cloned DIE!");
2819 }
2820
2821 /// \brief Extract the line table for \p Unit from \p OrigDwarf, and
2822 /// recreate a relocated version of these for the address ranges that
2823 /// are present in the binary.
2824 void DwarfLinker::patchLineTableForUnit(CompileUnit &Unit,
2825                                         DWARFContext &OrigDwarf) {
2826   const DWARFDebugInfoEntryMinimal *CUDie = Unit.getOrigUnit().getUnitDIE();
2827   uint64_t StmtList = CUDie->getAttributeValueAsSectionOffset(
2828       &Unit.getOrigUnit(), dwarf::DW_AT_stmt_list, -1ULL);
2829   if (StmtList == -1ULL)
2830     return;
2831
2832   // Update the cloned DW_AT_stmt_list with the correct debug_line offset.
2833   if (auto *OutputDIE = Unit.getOutputUnitDIE())
2834     patchStmtList(*OutputDIE, DIEInteger(Streamer->getLineSectionSize()));
2835
2836   // Parse the original line info for the unit.
2837   DWARFDebugLine::LineTable LineTable;
2838   uint32_t StmtOffset = StmtList;
2839   StringRef LineData = OrigDwarf.getLineSection().Data;
2840   DataExtractor LineExtractor(LineData, OrigDwarf.isLittleEndian(),
2841                               Unit.getOrigUnit().getAddressByteSize());
2842   LineTable.parse(LineExtractor, &OrigDwarf.getLineSection().Relocs,
2843                   &StmtOffset);
2844
2845   // This vector is the output line table.
2846   std::vector<DWARFDebugLine::Row> NewRows;
2847   NewRows.reserve(LineTable.Rows.size());
2848
2849   // Current sequence of rows being extracted, before being inserted
2850   // in NewRows.
2851   std::vector<DWARFDebugLine::Row> Seq;
2852   const auto &FunctionRanges = Unit.getFunctionRanges();
2853   auto InvalidRange = FunctionRanges.end(), CurrRange = InvalidRange;
2854
2855   // FIXME: This logic is meant to generate exactly the same output as
2856   // Darwin's classic dsynutil. There is a nicer way to implement this
2857   // by simply putting all the relocated line info in NewRows and simply
2858   // sorting NewRows before passing it to emitLineTableForUnit. This
2859   // should be correct as sequences for a function should stay
2860   // together in the sorted output. There are a few corner cases that
2861   // look suspicious though, and that required to implement the logic
2862   // this way. Revisit that once initial validation is finished.
2863
2864   // Iterate over the object file line info and extract the sequences
2865   // that correspond to linked functions.
2866   for (auto &Row : LineTable.Rows) {
2867     // Check wether we stepped out of the range. The range is
2868     // half-open, but consider accept the end address of the range if
2869     // it is marked as end_sequence in the input (because in that
2870     // case, the relocation offset is accurate and that entry won't
2871     // serve as the start of another function).
2872     if (CurrRange == InvalidRange || Row.Address < CurrRange.start() ||
2873         Row.Address > CurrRange.stop() ||
2874         (Row.Address == CurrRange.stop() && !Row.EndSequence)) {
2875       // We just stepped out of a known range. Insert a end_sequence
2876       // corresponding to the end of the range.
2877       uint64_t StopAddress = CurrRange != InvalidRange
2878                                  ? CurrRange.stop() + CurrRange.value()
2879                                  : -1ULL;
2880       CurrRange = FunctionRanges.find(Row.Address);
2881       bool CurrRangeValid =
2882           CurrRange != InvalidRange && CurrRange.start() <= Row.Address;
2883       if (!CurrRangeValid) {
2884         CurrRange = InvalidRange;
2885         if (StopAddress != -1ULL) {
2886           // Try harder by looking in the DebugMapObject function
2887           // ranges map. There are corner cases where this finds a
2888           // valid entry. It's unclear if this is right or wrong, but
2889           // for now do as dsymutil.
2890           // FIXME: Understand exactly what cases this addresses and
2891           // potentially remove it along with the Ranges map.
2892           auto Range = Ranges.lower_bound(Row.Address);
2893           if (Range != Ranges.begin() && Range != Ranges.end())
2894             --Range;
2895
2896           if (Range != Ranges.end() && Range->first <= Row.Address &&
2897               Range->second.first >= Row.Address) {
2898             StopAddress = Row.Address + Range->second.second;
2899           }
2900         }
2901       }
2902       if (StopAddress != -1ULL && !Seq.empty()) {
2903         // Insert end sequence row with the computed end address, but
2904         // the same line as the previous one.
2905         auto NextLine = Seq.back();
2906         NextLine.Address = StopAddress;
2907         NextLine.EndSequence = 1;
2908         NextLine.PrologueEnd = 0;
2909         NextLine.BasicBlock = 0;
2910         NextLine.EpilogueBegin = 0;
2911         Seq.push_back(NextLine);
2912         insertLineSequence(Seq, NewRows);
2913       }
2914
2915       if (!CurrRangeValid)
2916         continue;
2917     }
2918
2919     // Ignore empty sequences.
2920     if (Row.EndSequence && Seq.empty())
2921       continue;
2922
2923     // Relocate row address and add it to the current sequence.
2924     Row.Address += CurrRange.value();
2925     Seq.emplace_back(Row);
2926
2927     if (Row.EndSequence)
2928       insertLineSequence(Seq, NewRows);
2929   }
2930
2931   // Finished extracting, now emit the line tables.
2932   uint32_t PrologueEnd = StmtList + 10 + LineTable.Prologue.PrologueLength;
2933   // FIXME: LLVM hardcodes it's prologue values. We just copy the
2934   // prologue over and that works because we act as both producer and
2935   // consumer. It would be nicer to have a real configurable line
2936   // table emitter.
2937   if (LineTable.Prologue.Version != 2 ||
2938       LineTable.Prologue.DefaultIsStmt != DWARF2_LINE_DEFAULT_IS_STMT ||
2939       LineTable.Prologue.OpcodeBase > 13)
2940     reportWarning("line table paramters mismatch. Cannot emit.");
2941   else {
2942     MCDwarfLineTableParams Params;
2943     Params.DWARF2LineOpcodeBase = LineTable.Prologue.OpcodeBase;
2944     Params.DWARF2LineBase = LineTable.Prologue.LineBase;
2945     Params.DWARF2LineRange = LineTable.Prologue.LineRange;
2946     Streamer->emitLineTableForUnit(Params,
2947                                    LineData.slice(StmtList + 4, PrologueEnd),
2948                                    LineTable.Prologue.MinInstLength, NewRows,
2949                                    Unit.getOrigUnit().getAddressByteSize());
2950   }
2951 }
2952
2953 void DwarfLinker::emitAcceleratorEntriesForUnit(CompileUnit &Unit) {
2954   Streamer->emitPubNamesForUnit(Unit);
2955   Streamer->emitPubTypesForUnit(Unit);
2956 }
2957
2958 /// \brief Read the frame info stored in the object, and emit the
2959 /// patched frame descriptions for the linked binary.
2960 ///
2961 /// This is actually pretty easy as the data of the CIEs and FDEs can
2962 /// be considered as black boxes and moved as is. The only thing to do
2963 /// is to patch the addresses in the headers.
2964 void DwarfLinker::patchFrameInfoForObject(const DebugMapObject &DMO,
2965                                           DWARFContext &OrigDwarf,
2966                                           unsigned AddrSize) {
2967   StringRef FrameData = OrigDwarf.getDebugFrameSection();
2968   if (FrameData.empty())
2969     return;
2970
2971   DataExtractor Data(FrameData, OrigDwarf.isLittleEndian(), 0);
2972   uint32_t InputOffset = 0;
2973
2974   // Store the data of the CIEs defined in this object, keyed by their
2975   // offsets.
2976   DenseMap<uint32_t, StringRef> LocalCIES;
2977
2978   while (Data.isValidOffset(InputOffset)) {
2979     uint32_t EntryOffset = InputOffset;
2980     uint32_t InitialLength = Data.getU32(&InputOffset);
2981     if (InitialLength == 0xFFFFFFFF)
2982       return reportWarning("Dwarf64 bits no supported");
2983
2984     uint32_t CIEId = Data.getU32(&InputOffset);
2985     if (CIEId == 0xFFFFFFFF) {
2986       // This is a CIE, store it.
2987       StringRef CIEData = FrameData.substr(EntryOffset, InitialLength + 4);
2988       LocalCIES[EntryOffset] = CIEData;
2989       // The -4 is to account for the CIEId we just read.
2990       InputOffset += InitialLength - 4;
2991       continue;
2992     }
2993
2994     uint32_t Loc = Data.getUnsigned(&InputOffset, AddrSize);
2995
2996     // Some compilers seem to emit frame info that doesn't start at
2997     // the function entry point, thus we can't just lookup the address
2998     // in the debug map. Use the linker's range map to see if the FDE
2999     // describes something that we can relocate.
3000     auto Range = Ranges.upper_bound(Loc);
3001     if (Range != Ranges.begin())
3002       --Range;
3003     if (Range == Ranges.end() || Range->first > Loc ||
3004         Range->second.first <= Loc) {
3005       // The +4 is to account for the size of the InitialLength field itself.
3006       InputOffset = EntryOffset + InitialLength + 4;
3007       continue;
3008     }
3009
3010     // This is an FDE, and we have a mapping.
3011     // Have we already emitted a corresponding CIE?
3012     StringRef CIEData = LocalCIES[CIEId];
3013     if (CIEData.empty())
3014       return reportWarning("Inconsistent debug_frame content. Dropping.");
3015
3016     // Look if we already emitted a CIE that corresponds to the
3017     // referenced one (the CIE data is the key of that lookup).
3018     auto IteratorInserted = EmittedCIEs.insert(
3019         std::make_pair(CIEData, Streamer->getFrameSectionSize()));
3020     // If there is no CIE yet for this ID, emit it.
3021     if (IteratorInserted.second ||
3022         // FIXME: dsymutil-classic only caches the last used CIE for
3023         // reuse. Mimic that behavior for now. Just removing that
3024         // second half of the condition and the LastCIEOffset variable
3025         // makes the code DTRT.
3026         LastCIEOffset != IteratorInserted.first->getValue()) {
3027       LastCIEOffset = Streamer->getFrameSectionSize();
3028       IteratorInserted.first->getValue() = LastCIEOffset;
3029       Streamer->emitCIE(CIEData);
3030     }
3031
3032     // Emit the FDE with updated address and CIE pointer.
3033     // (4 + AddrSize) is the size of the CIEId + initial_location
3034     // fields that will get reconstructed by emitFDE().
3035     unsigned FDERemainingBytes = InitialLength - (4 + AddrSize);
3036     Streamer->emitFDE(IteratorInserted.first->getValue(), AddrSize,
3037                       Loc + Range->second.second,
3038                       FrameData.substr(InputOffset, FDERemainingBytes));
3039     InputOffset += FDERemainingBytes;
3040   }
3041 }
3042
3043 void DwarfLinker::DIECloner::copyAbbrev(
3044     const DWARFAbbreviationDeclaration &Abbrev, bool hasODR) {
3045   DIEAbbrev Copy(dwarf::Tag(Abbrev.getTag()),
3046                  dwarf::Form(Abbrev.hasChildren()));
3047
3048   for (const auto &Attr : Abbrev.attributes()) {
3049     uint16_t Form = Attr.Form;
3050     if (hasODR && isODRAttribute(Attr.Attr))
3051       Form = dwarf::DW_FORM_ref_addr;
3052     Copy.AddAttribute(dwarf::Attribute(Attr.Attr), dwarf::Form(Form));
3053   }
3054
3055   Linker.AssignAbbrev(Copy);
3056 }
3057
3058 ErrorOr<const object::ObjectFile &>
3059 DwarfLinker::loadObject(BinaryHolder &BinaryHolder, DebugMapObject &Obj,
3060                         const DebugMap &Map) {
3061   auto ErrOrObjs =
3062       BinaryHolder.GetObjectFiles(Obj.getObjectFilename(), Obj.getTimestamp());
3063   if (std::error_code EC = ErrOrObjs.getError()) {
3064     reportWarning(Twine(Obj.getObjectFilename()) + ": " + EC.message());
3065     return EC;
3066   }
3067   auto ErrOrObj = BinaryHolder.Get(Map.getTriple());
3068   if (std::error_code EC = ErrOrObj.getError())
3069     reportWarning(Twine(Obj.getObjectFilename()) + ": " + EC.message());
3070   return ErrOrObj;
3071 }
3072
3073 void DwarfLinker::DIECloner::cloneAllCompileUnits(
3074     DWARFContextInMemory &DwarfContext) {
3075   if (!Linker.Streamer)
3076     return;
3077
3078   for (auto &CurrentUnit : CompileUnits) {
3079     const auto *InputDIE = CurrentUnit.getOrigUnit().getUnitDIE();
3080     CurrentUnit.setStartOffset(Linker.OutputDebugInfoSize);
3081     DIE *OutputDIE = cloneDIE(*InputDIE, CurrentUnit, 0 /* PC offset */,
3082                               11 /* Unit Header size */, 0);
3083     CurrentUnit.setOutputUnitDIE(OutputDIE);
3084     Linker.OutputDebugInfoSize = CurrentUnit.computeNextUnitOffset();
3085     if (Linker.Options.NoOutput)
3086       continue;
3087     // FIXME: for compatibility with the classic dsymutil, we emit
3088     // an empty line table for the unit, even if the unit doesn't
3089     // actually exist in the DIE tree.
3090     Linker.patchLineTableForUnit(CurrentUnit, DwarfContext);
3091     if (!OutputDIE)
3092       continue;
3093     Linker.patchRangesForUnit(CurrentUnit, DwarfContext);
3094     Linker.Streamer->emitLocationsForUnit(CurrentUnit, DwarfContext);
3095     Linker.emitAcceleratorEntriesForUnit(CurrentUnit);
3096   }
3097
3098   if (Linker.Options.NoOutput)
3099     return;
3100
3101   // Emit all the compile unit's debug information.
3102   for (auto &CurrentUnit : CompileUnits) {
3103     Linker.generateUnitRanges(CurrentUnit);
3104     CurrentUnit.fixupForwardReferences();
3105     Linker.Streamer->emitCompileUnitHeader(CurrentUnit);
3106     if (!CurrentUnit.getOutputUnitDIE())
3107       continue;
3108     Linker.Streamer->emitDIE(*CurrentUnit.getOutputUnitDIE());
3109   }
3110 }
3111
3112 bool DwarfLinker::link(const DebugMap &Map) {
3113
3114   if (!createStreamer(Map.getTriple(), OutputFilename))
3115     return false;
3116
3117   // Size of the DIEs (and headers) generated for the linked output.
3118   OutputDebugInfoSize = 0;
3119   // A unique ID that identifies each compile unit.
3120   unsigned UnitID = 0;
3121   for (const auto &Obj : Map.objects()) {
3122     CurrentDebugObject = Obj.get();
3123
3124     if (Options.Verbose)
3125       outs() << "DEBUG MAP OBJECT: " << Obj->getObjectFilename() << "\n";
3126     auto ErrOrObj = loadObject(BinHolder, *Obj, Map);
3127     if (!ErrOrObj)
3128       continue;
3129
3130     // Look for relocations that correspond to debug map entries.
3131     RelocationManager RelocMgr(*this);
3132     if (!RelocMgr.findValidRelocsInDebugInfo(*ErrOrObj, *Obj)) {
3133       if (Options.Verbose)
3134         outs() << "No valid relocations found. Skipping.\n";
3135       continue;
3136     }
3137
3138     // Setup access to the debug info.
3139     DWARFContextInMemory DwarfContext(*ErrOrObj);
3140     startDebugObject(DwarfContext, *Obj);
3141
3142     // In a first phase, just read in the debug info and store the DIE
3143     // parent links that we will use during the next phase.
3144     for (const auto &CU : DwarfContext.compile_units()) {
3145       auto *CUDie = CU->getUnitDIE(false);
3146       if (Options.Verbose) {
3147         outs() << "Input compilation unit:";
3148         CUDie->dump(outs(), CU.get(), 0);
3149       }
3150       Units.emplace_back(*CU, UnitID++, !Options.NoODR);
3151       gatherDIEParents(CUDie, 0, Units.back(), &ODRContexts.getRoot(),
3152                        StringPool, ODRContexts);
3153     }
3154
3155     // Then mark all the DIEs that need to be present in the linked
3156     // output and collect some information about them. Note that this
3157     // loop can not be merged with the previous one becaue cross-cu
3158     // references require the ParentIdx to be setup for every CU in
3159     // the object file before calling this.
3160     for (auto &CurrentUnit : Units)
3161       lookForDIEsToKeep(RelocMgr, *CurrentUnit.getOrigUnit().getUnitDIE(), *Obj,
3162                         CurrentUnit, 0);
3163
3164     // The calls to applyValidRelocs inside cloneDIE will walk the
3165     // reloc array again (in the same way findValidRelocsInDebugInfo()
3166     // did). We need to reset the NextValidReloc index to the beginning.
3167     RelocMgr.resetValidRelocs();
3168     if (RelocMgr.hasValidRelocs())
3169       DIECloner(*this, RelocMgr, DIEAlloc, Units, Options)
3170           .cloneAllCompileUnits(DwarfContext);
3171     if (!Options.NoOutput && !Units.empty())
3172       patchFrameInfoForObject(*Obj, DwarfContext,
3173                               Units[0].getOrigUnit().getAddressByteSize());
3174
3175     // Clean-up before starting working on the next object.
3176     endDebugObject();
3177   }
3178
3179   // Emit everything that's global.
3180   if (!Options.NoOutput) {
3181     Streamer->emitAbbrevs(Abbreviations);
3182     Streamer->emitStrings(StringPool);
3183   }
3184
3185   return Options.NoOutput ? true : Streamer->finish(Map);
3186 }
3187 }
3188
3189 /// \brief Get the offset of string \p S in the string table. This
3190 /// can insert a new element or return the offset of a preexisitng
3191 /// one.
3192 uint32_t NonRelocatableStringpool::getStringOffset(StringRef S) {
3193   if (S.empty() && !Strings.empty())
3194     return 0;
3195
3196   std::pair<uint32_t, StringMapEntryBase *> Entry(0, nullptr);
3197   MapTy::iterator It;
3198   bool Inserted;
3199
3200   // A non-empty string can't be at offset 0, so if we have an entry
3201   // with a 0 offset, it must be a previously interned string.
3202   std::tie(It, Inserted) = Strings.insert(std::make_pair(S, Entry));
3203   if (Inserted || It->getValue().first == 0) {
3204     // Set offset and chain at the end of the entries list.
3205     It->getValue().first = CurrentEndOffset;
3206     CurrentEndOffset += S.size() + 1; // +1 for the '\0'.
3207     Last->getValue().second = &*It;
3208     Last = &*It;
3209   }
3210   return It->getValue().first;
3211 }
3212
3213 /// \brief Put \p S into the StringMap so that it gets permanent
3214 /// storage, but do not actually link it in the chain of elements
3215 /// that go into the output section. A latter call to
3216 /// getStringOffset() with the same string will chain it though.
3217 StringRef NonRelocatableStringpool::internString(StringRef S) {
3218   std::pair<uint32_t, StringMapEntryBase *> Entry(0, nullptr);
3219   auto InsertResult = Strings.insert(std::make_pair(S, Entry));
3220   return InsertResult.first->getKey();
3221 }
3222
3223 void warn(const Twine &Warning, const Twine &Context) {
3224   errs() << Twine("while processing ") + Context + ":\n";
3225   errs() << Twine("warning: ") + Warning + "\n";
3226 }
3227
3228 bool error(const Twine &Error, const Twine &Context) {
3229   errs() << Twine("while processing ") + Context + ":\n";
3230   errs() << Twine("error: ") + Error + "\n";
3231   return false;
3232 }
3233
3234 bool linkDwarf(StringRef OutputFilename, const DebugMap &DM,
3235                const LinkOptions &Options) {
3236   DwarfLinker Linker(OutputFilename, Options);
3237   return Linker.link(DM);
3238 }
3239 }
3240 }