xref: /freebsd/contrib/llvm-project/llvm/lib/DebugInfo/DWARF/DWARFContext.cpp (revision dd41de95a84d979615a2ef11df6850622bf6184e)
1 //===- DWARFContext.cpp ---------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include "llvm/DebugInfo/DWARF/DWARFContext.h"
10 #include "llvm/ADT/STLExtras.h"
11 #include "llvm/ADT/SmallString.h"
12 #include "llvm/ADT/SmallVector.h"
13 #include "llvm/ADT/StringRef.h"
14 #include "llvm/ADT/StringSwitch.h"
15 #include "llvm/BinaryFormat/Dwarf.h"
16 #include "llvm/DebugInfo/DWARF/DWARFAcceleratorTable.h"
17 #include "llvm/DebugInfo/DWARF/DWARFCompileUnit.h"
18 #include "llvm/DebugInfo/DWARF/DWARFDebugAbbrev.h"
19 #include "llvm/DebugInfo/DWARF/DWARFDebugAddr.h"
20 #include "llvm/DebugInfo/DWARF/DWARFDebugArangeSet.h"
21 #include "llvm/DebugInfo/DWARF/DWARFDebugAranges.h"
22 #include "llvm/DebugInfo/DWARF/DWARFDebugFrame.h"
23 #include "llvm/DebugInfo/DWARF/DWARFDebugLine.h"
24 #include "llvm/DebugInfo/DWARF/DWARFDebugLoc.h"
25 #include "llvm/DebugInfo/DWARF/DWARFDebugMacro.h"
26 #include "llvm/DebugInfo/DWARF/DWARFDebugPubTable.h"
27 #include "llvm/DebugInfo/DWARF/DWARFDebugRangeList.h"
28 #include "llvm/DebugInfo/DWARF/DWARFDebugRnglists.h"
29 #include "llvm/DebugInfo/DWARF/DWARFDie.h"
30 #include "llvm/DebugInfo/DWARF/DWARFFormValue.h"
31 #include "llvm/DebugInfo/DWARF/DWARFGdbIndex.h"
32 #include "llvm/DebugInfo/DWARF/DWARFSection.h"
33 #include "llvm/DebugInfo/DWARF/DWARFUnitIndex.h"
34 #include "llvm/DebugInfo/DWARF/DWARFVerifier.h"
35 #include "llvm/MC/MCRegisterInfo.h"
36 #include "llvm/Object/Decompressor.h"
37 #include "llvm/Object/MachO.h"
38 #include "llvm/Object/ObjectFile.h"
39 #include "llvm/Object/RelocationResolver.h"
40 #include "llvm/Support/Casting.h"
41 #include "llvm/Support/DataExtractor.h"
42 #include "llvm/Support/Error.h"
43 #include "llvm/Support/Format.h"
44 #include "llvm/Support/LEB128.h"
45 #include "llvm/Support/MemoryBuffer.h"
46 #include "llvm/Support/Path.h"
47 #include "llvm/Support/TargetRegistry.h"
48 #include "llvm/Support/raw_ostream.h"
49 #include <algorithm>
50 #include <cstdint>
51 #include <deque>
52 #include <map>
53 #include <string>
54 #include <utility>
55 #include <vector>
56 
57 using namespace llvm;
58 using namespace dwarf;
59 using namespace object;
60 
61 #define DEBUG_TYPE "dwarf"
62 
63 using DWARFLineTable = DWARFDebugLine::LineTable;
64 using FileLineInfoKind = DILineInfoSpecifier::FileLineInfoKind;
65 using FunctionNameKind = DILineInfoSpecifier::FunctionNameKind;
66 
67 DWARFContext::DWARFContext(std::unique_ptr<const DWARFObject> DObj,
68                            std::string DWPName,
69                            std::function<void(Error)> RecoverableErrorHandler,
70                            std::function<void(Error)> WarningHandler)
71     : DIContext(CK_DWARF), DWPName(std::move(DWPName)),
72       RecoverableErrorHandler(RecoverableErrorHandler),
73       WarningHandler(WarningHandler), DObj(std::move(DObj)) {}
74 
75 DWARFContext::~DWARFContext() = default;
76 
77 /// Dump the UUID load command.
78 static void dumpUUID(raw_ostream &OS, const ObjectFile &Obj) {
79   auto *MachO = dyn_cast<MachOObjectFile>(&Obj);
80   if (!MachO)
81     return;
82   for (auto LC : MachO->load_commands()) {
83     raw_ostream::uuid_t UUID;
84     if (LC.C.cmd == MachO::LC_UUID) {
85       if (LC.C.cmdsize < sizeof(UUID) + sizeof(LC.C)) {
86         OS << "error: UUID load command is too short.\n";
87         return;
88       }
89       OS << "UUID: ";
90       memcpy(&UUID, LC.Ptr+sizeof(LC.C), sizeof(UUID));
91       OS.write_uuid(UUID);
92       Triple T = MachO->getArchTriple();
93       OS << " (" << T.getArchName() << ')';
94       OS << ' ' << MachO->getFileName() << '\n';
95     }
96   }
97 }
98 
99 using ContributionCollection =
100     std::vector<Optional<StrOffsetsContributionDescriptor>>;
101 
102 // Collect all the contributions to the string offsets table from all units,
103 // sort them by their starting offsets and remove duplicates.
104 static ContributionCollection
105 collectContributionData(DWARFContext::unit_iterator_range Units) {
106   ContributionCollection Contributions;
107   for (const auto &U : Units)
108     if (const auto &C = U->getStringOffsetsTableContribution())
109       Contributions.push_back(C);
110   // Sort the contributions so that any invalid ones are placed at
111   // the start of the contributions vector. This way they are reported
112   // first.
113   llvm::sort(Contributions,
114              [](const Optional<StrOffsetsContributionDescriptor> &L,
115                 const Optional<StrOffsetsContributionDescriptor> &R) {
116                if (L && R)
117                  return L->Base < R->Base;
118                return R.hasValue();
119              });
120 
121   // Uniquify contributions, as it is possible that units (specifically
122   // type units in dwo or dwp files) share contributions. We don't want
123   // to report them more than once.
124   Contributions.erase(
125       std::unique(Contributions.begin(), Contributions.end(),
126                   [](const Optional<StrOffsetsContributionDescriptor> &L,
127                      const Optional<StrOffsetsContributionDescriptor> &R) {
128                     if (L && R)
129                       return L->Base == R->Base && L->Size == R->Size;
130                     return false;
131                   }),
132       Contributions.end());
133   return Contributions;
134 }
135 
136 // Dump a DWARF string offsets section. This may be a DWARF v5 formatted
137 // string offsets section, where each compile or type unit contributes a
138 // number of entries (string offsets), with each contribution preceded by
139 // a header containing size and version number. Alternatively, it may be a
140 // monolithic series of string offsets, as generated by the pre-DWARF v5
141 // implementation of split DWARF; however, in that case we still need to
142 // collect contributions of units because the size of the offsets (4 or 8
143 // bytes) depends on the format of the referencing unit (DWARF32 or DWARF64).
144 static void dumpStringOffsetsSection(raw_ostream &OS, DIDumpOptions DumpOpts,
145                                      StringRef SectionName,
146                                      const DWARFObject &Obj,
147                                      const DWARFSection &StringOffsetsSection,
148                                      StringRef StringSection,
149                                      DWARFContext::unit_iterator_range Units,
150                                      bool LittleEndian) {
151   auto Contributions = collectContributionData(Units);
152   DWARFDataExtractor StrOffsetExt(Obj, StringOffsetsSection, LittleEndian, 0);
153   DataExtractor StrData(StringSection, LittleEndian, 0);
154   uint64_t SectionSize = StringOffsetsSection.Data.size();
155   uint64_t Offset = 0;
156   for (auto &Contribution : Contributions) {
157     // Report an ill-formed contribution.
158     if (!Contribution) {
159       OS << "error: invalid contribution to string offsets table in section ."
160          << SectionName << ".\n";
161       return;
162     }
163 
164     dwarf::DwarfFormat Format = Contribution->getFormat();
165     int OffsetDumpWidth = 2 * dwarf::getDwarfOffsetByteSize(Format);
166     uint16_t Version = Contribution->getVersion();
167     uint64_t ContributionHeader = Contribution->Base;
168     // In DWARF v5 there is a contribution header that immediately precedes
169     // the string offsets base (the location we have previously retrieved from
170     // the CU DIE's DW_AT_str_offsets attribute). The header is located either
171     // 8 or 16 bytes before the base, depending on the contribution's format.
172     if (Version >= 5)
173       ContributionHeader -= Format == DWARF32 ? 8 : 16;
174 
175     // Detect overlapping contributions.
176     if (Offset > ContributionHeader) {
177       DumpOpts.RecoverableErrorHandler(createStringError(
178           errc::invalid_argument,
179           "overlapping contributions to string offsets table in section .%s.",
180           SectionName.data()));
181     }
182     // Report a gap in the table.
183     if (Offset < ContributionHeader) {
184       OS << format("0x%8.8" PRIx64 ": Gap, length = ", Offset);
185       OS << (ContributionHeader - Offset) << "\n";
186     }
187     OS << format("0x%8.8" PRIx64 ": ", ContributionHeader);
188     // In DWARF v5 the contribution size in the descriptor does not equal
189     // the originally encoded length (it does not contain the length of the
190     // version field and the padding, a total of 4 bytes). Add them back in
191     // for reporting.
192     OS << "Contribution size = " << (Contribution->Size + (Version < 5 ? 0 : 4))
193        << ", Format = " << dwarf::FormatString(Format)
194        << ", Version = " << Version << "\n";
195 
196     Offset = Contribution->Base;
197     unsigned EntrySize = Contribution->getDwarfOffsetByteSize();
198     while (Offset - Contribution->Base < Contribution->Size) {
199       OS << format("0x%8.8" PRIx64 ": ", Offset);
200       uint64_t StringOffset =
201           StrOffsetExt.getRelocatedValue(EntrySize, &Offset);
202       OS << format("%0*" PRIx64 " ", OffsetDumpWidth, StringOffset);
203       const char *S = StrData.getCStr(&StringOffset);
204       if (S)
205         OS << format("\"%s\"", S);
206       OS << "\n";
207     }
208   }
209   // Report a gap at the end of the table.
210   if (Offset < SectionSize) {
211     OS << format("0x%8.8" PRIx64 ": Gap, length = ", Offset);
212     OS << (SectionSize - Offset) << "\n";
213   }
214 }
215 
216 // Dump the .debug_addr section.
217 static void dumpAddrSection(raw_ostream &OS, DWARFDataExtractor &AddrData,
218                             DIDumpOptions DumpOpts, uint16_t Version,
219                             uint8_t AddrSize) {
220   uint64_t Offset = 0;
221   while (AddrData.isValidOffset(Offset)) {
222     DWARFDebugAddrTable AddrTable;
223     uint64_t TableOffset = Offset;
224     if (Error Err = AddrTable.extract(AddrData, &Offset, Version, AddrSize,
225                                       DumpOpts.WarningHandler)) {
226       DumpOpts.RecoverableErrorHandler(std::move(Err));
227       // Keep going after an error, if we can, assuming that the length field
228       // could be read. If it couldn't, stop reading the section.
229       if (auto TableLength = AddrTable.getFullLength()) {
230         Offset = TableOffset + *TableLength;
231         continue;
232       }
233       break;
234     }
235     AddrTable.dump(OS, DumpOpts);
236   }
237 }
238 
239 // Dump the .debug_rnglists or .debug_rnglists.dwo section (DWARF v5).
240 static void dumpRnglistsSection(
241     raw_ostream &OS, DWARFDataExtractor &rnglistData,
242     llvm::function_ref<Optional<object::SectionedAddress>(uint32_t)>
243         LookupPooledAddress,
244     DIDumpOptions DumpOpts) {
245   uint64_t Offset = 0;
246   while (rnglistData.isValidOffset(Offset)) {
247     llvm::DWARFDebugRnglistTable Rnglists;
248     uint64_t TableOffset = Offset;
249     if (Error Err = Rnglists.extract(rnglistData, &Offset)) {
250       DumpOpts.RecoverableErrorHandler(std::move(Err));
251       uint64_t Length = Rnglists.length();
252       // Keep going after an error, if we can, assuming that the length field
253       // could be read. If it couldn't, stop reading the section.
254       if (Length == 0)
255         break;
256       Offset = TableOffset + Length;
257     } else {
258       Rnglists.dump(OS, LookupPooledAddress, DumpOpts);
259     }
260   }
261 }
262 
263 std::unique_ptr<DWARFDebugMacro>
264 DWARFContext::parseMacroOrMacinfo(MacroSecType SectionType) {
265   auto Macro = std::make_unique<DWARFDebugMacro>();
266   auto ParseAndDump = [&](DWARFDataExtractor &Data, bool IsMacro) {
267     if (Error Err = IsMacro ? Macro->parseMacro(SectionType == MacroSection
268                                                     ? compile_units()
269                                                     : dwo_compile_units(),
270                                                 SectionType == MacroSection
271                                                     ? getStringExtractor()
272                                                     : getStringDWOExtractor(),
273                                                 Data)
274                             : Macro->parseMacinfo(Data)) {
275       RecoverableErrorHandler(std::move(Err));
276       Macro = nullptr;
277     }
278   };
279   switch (SectionType) {
280   case MacinfoSection: {
281     DWARFDataExtractor Data(DObj->getMacinfoSection(), isLittleEndian(), 0);
282     ParseAndDump(Data, /*IsMacro=*/false);
283     break;
284   }
285   case MacinfoDwoSection: {
286     DWARFDataExtractor Data(DObj->getMacinfoDWOSection(), isLittleEndian(), 0);
287     ParseAndDump(Data, /*IsMacro=*/false);
288     break;
289   }
290   case MacroSection: {
291     DWARFDataExtractor Data(*DObj, DObj->getMacroSection(), isLittleEndian(),
292                             0);
293     ParseAndDump(Data, /*IsMacro=*/true);
294     break;
295   }
296   case MacroDwoSection: {
297     DWARFDataExtractor Data(DObj->getMacroDWOSection(), isLittleEndian(), 0);
298     ParseAndDump(Data, /*IsMacro=*/true);
299     break;
300   }
301   }
302   return Macro;
303 }
304 
305 static void dumpLoclistsSection(raw_ostream &OS, DIDumpOptions DumpOpts,
306                                 DWARFDataExtractor Data,
307                                 const MCRegisterInfo *MRI,
308                                 const DWARFObject &Obj,
309                                 Optional<uint64_t> DumpOffset) {
310   uint64_t Offset = 0;
311 
312   while (Data.isValidOffset(Offset)) {
313     DWARFListTableHeader Header(".debug_loclists", "locations");
314     if (Error E = Header.extract(Data, &Offset)) {
315       DumpOpts.RecoverableErrorHandler(std::move(E));
316       return;
317     }
318 
319     Header.dump(OS, DumpOpts);
320 
321     uint64_t EndOffset = Header.length() + Header.getHeaderOffset();
322     Data.setAddressSize(Header.getAddrSize());
323     DWARFDebugLoclists Loc(Data, Header.getVersion());
324     if (DumpOffset) {
325       if (DumpOffset >= Offset && DumpOffset < EndOffset) {
326         Offset = *DumpOffset;
327         Loc.dumpLocationList(&Offset, OS, /*BaseAddr=*/None, MRI, Obj, nullptr,
328                              DumpOpts, /*Indent=*/0);
329         OS << "\n";
330         return;
331       }
332     } else {
333       Loc.dumpRange(Offset, EndOffset - Offset, OS, MRI, Obj, DumpOpts);
334     }
335     Offset = EndOffset;
336   }
337 }
338 
339 static void dumpPubTableSection(raw_ostream &OS, DIDumpOptions DumpOpts,
340                                 DWARFDataExtractor Data, bool GnuStyle) {
341   DWARFDebugPubTable Table;
342   Table.extract(Data, GnuStyle, DumpOpts.RecoverableErrorHandler);
343   Table.dump(OS);
344 }
345 
346 void DWARFContext::dump(
347     raw_ostream &OS, DIDumpOptions DumpOpts,
348     std::array<Optional<uint64_t>, DIDT_ID_Count> DumpOffsets) {
349   uint64_t DumpType = DumpOpts.DumpType;
350 
351   StringRef Extension = sys::path::extension(DObj->getFileName());
352   bool IsDWO = (Extension == ".dwo") || (Extension == ".dwp");
353 
354   // Print UUID header.
355   const auto *ObjFile = DObj->getFile();
356   if (DumpType & DIDT_UUID)
357     dumpUUID(OS, *ObjFile);
358 
359   // Print a header for each explicitly-requested section.
360   // Otherwise just print one for non-empty sections.
361   // Only print empty .dwo section headers when dumping a .dwo file.
362   bool Explicit = DumpType != DIDT_All && !IsDWO;
363   bool ExplicitDWO = Explicit && IsDWO;
364   auto shouldDump = [&](bool Explicit, const char *Name, unsigned ID,
365                         StringRef Section) -> Optional<uint64_t> * {
366     unsigned Mask = 1U << ID;
367     bool Should = (DumpType & Mask) && (Explicit || !Section.empty());
368     if (!Should)
369       return nullptr;
370     OS << "\n" << Name << " contents:\n";
371     return &DumpOffsets[ID];
372   };
373 
374   // Dump individual sections.
375   if (shouldDump(Explicit, ".debug_abbrev", DIDT_ID_DebugAbbrev,
376                  DObj->getAbbrevSection()))
377     getDebugAbbrev()->dump(OS);
378   if (shouldDump(ExplicitDWO, ".debug_abbrev.dwo", DIDT_ID_DebugAbbrev,
379                  DObj->getAbbrevDWOSection()))
380     getDebugAbbrevDWO()->dump(OS);
381 
382   auto dumpDebugInfo = [&](const char *Name, unit_iterator_range Units) {
383     OS << '\n' << Name << " contents:\n";
384     if (auto DumpOffset = DumpOffsets[DIDT_ID_DebugInfo])
385       for (const auto &U : Units)
386         U->getDIEForOffset(DumpOffset.getValue())
387             .dump(OS, 0, DumpOpts.noImplicitRecursion());
388     else
389       for (const auto &U : Units)
390         U->dump(OS, DumpOpts);
391   };
392   if ((DumpType & DIDT_DebugInfo)) {
393     if (Explicit || getNumCompileUnits())
394       dumpDebugInfo(".debug_info", info_section_units());
395     if (ExplicitDWO || getNumDWOCompileUnits())
396       dumpDebugInfo(".debug_info.dwo", dwo_info_section_units());
397   }
398 
399   auto dumpDebugType = [&](const char *Name, unit_iterator_range Units) {
400     OS << '\n' << Name << " contents:\n";
401     for (const auto &U : Units)
402       if (auto DumpOffset = DumpOffsets[DIDT_ID_DebugTypes])
403         U->getDIEForOffset(*DumpOffset)
404             .dump(OS, 0, DumpOpts.noImplicitRecursion());
405       else
406         U->dump(OS, DumpOpts);
407   };
408   if ((DumpType & DIDT_DebugTypes)) {
409     if (Explicit || getNumTypeUnits())
410       dumpDebugType(".debug_types", types_section_units());
411     if (ExplicitDWO || getNumDWOTypeUnits())
412       dumpDebugType(".debug_types.dwo", dwo_types_section_units());
413   }
414 
415   DIDumpOptions LLDumpOpts = DumpOpts;
416   if (LLDumpOpts.Verbose)
417     LLDumpOpts.DisplayRawContents = true;
418 
419   if (const auto *Off = shouldDump(Explicit, ".debug_loc", DIDT_ID_DebugLoc,
420                                    DObj->getLocSection().Data)) {
421     getDebugLoc()->dump(OS, getRegisterInfo(), *DObj, LLDumpOpts, *Off);
422   }
423   if (const auto *Off =
424           shouldDump(Explicit, ".debug_loclists", DIDT_ID_DebugLoclists,
425                      DObj->getLoclistsSection().Data)) {
426     DWARFDataExtractor Data(*DObj, DObj->getLoclistsSection(), isLittleEndian(),
427                             0);
428     dumpLoclistsSection(OS, LLDumpOpts, Data, getRegisterInfo(), *DObj, *Off);
429   }
430   if (const auto *Off =
431           shouldDump(ExplicitDWO, ".debug_loclists.dwo", DIDT_ID_DebugLoclists,
432                      DObj->getLoclistsDWOSection().Data)) {
433     DWARFDataExtractor Data(*DObj, DObj->getLoclistsDWOSection(),
434                             isLittleEndian(), 0);
435     dumpLoclistsSection(OS, LLDumpOpts, Data, getRegisterInfo(), *DObj, *Off);
436   }
437 
438   if (const auto *Off =
439           shouldDump(ExplicitDWO, ".debug_loc.dwo", DIDT_ID_DebugLoc,
440                      DObj->getLocDWOSection().Data)) {
441     DWARFDataExtractor Data(*DObj, DObj->getLocDWOSection(), isLittleEndian(),
442                             4);
443     DWARFDebugLoclists Loc(Data, /*Version=*/4);
444     if (*Off) {
445       uint64_t Offset = **Off;
446       Loc.dumpLocationList(&Offset, OS,
447                            /*BaseAddr=*/None, getRegisterInfo(), *DObj, nullptr,
448                            LLDumpOpts, /*Indent=*/0);
449       OS << "\n";
450     } else {
451       Loc.dumpRange(0, Data.getData().size(), OS, getRegisterInfo(), *DObj,
452                     LLDumpOpts);
453     }
454   }
455 
456   if (const Optional<uint64_t> *Off =
457           shouldDump(Explicit, ".debug_frame", DIDT_ID_DebugFrame,
458                      DObj->getFrameSection().Data)) {
459     if (Expected<const DWARFDebugFrame *> DF = getDebugFrame())
460       (*DF)->dump(OS, getRegisterInfo(), *Off);
461     else
462       RecoverableErrorHandler(DF.takeError());
463   }
464 
465   if (const Optional<uint64_t> *Off =
466           shouldDump(Explicit, ".eh_frame", DIDT_ID_DebugFrame,
467                      DObj->getEHFrameSection().Data)) {
468     if (Expected<const DWARFDebugFrame *> DF = getEHFrame())
469       (*DF)->dump(OS, getRegisterInfo(), *Off);
470     else
471       RecoverableErrorHandler(DF.takeError());
472   }
473 
474   if (shouldDump(Explicit, ".debug_macro", DIDT_ID_DebugMacro,
475                  DObj->getMacroSection().Data)) {
476     if (auto Macro = getDebugMacro())
477       Macro->dump(OS);
478   }
479 
480   if (shouldDump(Explicit, ".debug_macro.dwo", DIDT_ID_DebugMacro,
481                  DObj->getMacroDWOSection())) {
482     if (auto MacroDWO = getDebugMacroDWO())
483       MacroDWO->dump(OS);
484   }
485 
486   if (shouldDump(Explicit, ".debug_macinfo", DIDT_ID_DebugMacro,
487                  DObj->getMacinfoSection())) {
488     if (auto Macinfo = getDebugMacinfo())
489       Macinfo->dump(OS);
490   }
491 
492   if (shouldDump(Explicit, ".debug_macinfo.dwo", DIDT_ID_DebugMacro,
493                  DObj->getMacinfoDWOSection())) {
494     if (auto MacinfoDWO = getDebugMacinfoDWO())
495       MacinfoDWO->dump(OS);
496   }
497 
498   if (shouldDump(Explicit, ".debug_aranges", DIDT_ID_DebugAranges,
499                  DObj->getArangesSection())) {
500     uint64_t offset = 0;
501     DWARFDataExtractor arangesData(DObj->getArangesSection(), isLittleEndian(),
502                                    0);
503     DWARFDebugArangeSet set;
504     while (arangesData.isValidOffset(offset)) {
505       if (Error E = set.extract(arangesData, &offset)) {
506         RecoverableErrorHandler(std::move(E));
507         break;
508       }
509       set.dump(OS);
510     }
511   }
512 
513   auto DumpLineSection = [&](DWARFDebugLine::SectionParser Parser,
514                              DIDumpOptions DumpOpts,
515                              Optional<uint64_t> DumpOffset) {
516     while (!Parser.done()) {
517       if (DumpOffset && Parser.getOffset() != *DumpOffset) {
518         Parser.skip(DumpOpts.WarningHandler, DumpOpts.WarningHandler);
519         continue;
520       }
521       OS << "debug_line[" << format("0x%8.8" PRIx64, Parser.getOffset())
522          << "]\n";
523       Parser.parseNext(DumpOpts.WarningHandler, DumpOpts.WarningHandler, &OS,
524                        DumpOpts.Verbose);
525     }
526   };
527 
528   if (const auto *Off = shouldDump(Explicit, ".debug_line", DIDT_ID_DebugLine,
529                                    DObj->getLineSection().Data)) {
530     DWARFDataExtractor LineData(*DObj, DObj->getLineSection(), isLittleEndian(),
531                                 0);
532     DWARFDebugLine::SectionParser Parser(LineData, *this, compile_units(),
533                                          type_units());
534     DumpLineSection(Parser, DumpOpts, *Off);
535   }
536 
537   if (const auto *Off =
538           shouldDump(ExplicitDWO, ".debug_line.dwo", DIDT_ID_DebugLine,
539                      DObj->getLineDWOSection().Data)) {
540     DWARFDataExtractor LineData(*DObj, DObj->getLineDWOSection(),
541                                 isLittleEndian(), 0);
542     DWARFDebugLine::SectionParser Parser(LineData, *this, dwo_compile_units(),
543                                          dwo_type_units());
544     DumpLineSection(Parser, DumpOpts, *Off);
545   }
546 
547   if (shouldDump(Explicit, ".debug_cu_index", DIDT_ID_DebugCUIndex,
548                  DObj->getCUIndexSection())) {
549     getCUIndex().dump(OS);
550   }
551 
552   if (shouldDump(Explicit, ".debug_tu_index", DIDT_ID_DebugTUIndex,
553                  DObj->getTUIndexSection())) {
554     getTUIndex().dump(OS);
555   }
556 
557   if (shouldDump(Explicit, ".debug_str", DIDT_ID_DebugStr,
558                  DObj->getStrSection())) {
559     DataExtractor strData(DObj->getStrSection(), isLittleEndian(), 0);
560     uint64_t offset = 0;
561     uint64_t strOffset = 0;
562     while (const char *s = strData.getCStr(&offset)) {
563       OS << format("0x%8.8" PRIx64 ": \"%s\"\n", strOffset, s);
564       strOffset = offset;
565     }
566   }
567   if (shouldDump(ExplicitDWO, ".debug_str.dwo", DIDT_ID_DebugStr,
568                  DObj->getStrDWOSection())) {
569     DataExtractor strDWOData(DObj->getStrDWOSection(), isLittleEndian(), 0);
570     uint64_t offset = 0;
571     uint64_t strDWOOffset = 0;
572     while (const char *s = strDWOData.getCStr(&offset)) {
573       OS << format("0x%8.8" PRIx64 ": \"%s\"\n", strDWOOffset, s);
574       strDWOOffset = offset;
575     }
576   }
577   if (shouldDump(Explicit, ".debug_line_str", DIDT_ID_DebugLineStr,
578                  DObj->getLineStrSection())) {
579     DataExtractor strData(DObj->getLineStrSection(), isLittleEndian(), 0);
580     uint64_t offset = 0;
581     uint64_t strOffset = 0;
582     while (const char *s = strData.getCStr(&offset)) {
583       OS << format("0x%8.8" PRIx64 ": \"", strOffset);
584       OS.write_escaped(s);
585       OS << "\"\n";
586       strOffset = offset;
587     }
588   }
589 
590   if (shouldDump(Explicit, ".debug_addr", DIDT_ID_DebugAddr,
591                  DObj->getAddrSection().Data)) {
592     DWARFDataExtractor AddrData(*DObj, DObj->getAddrSection(),
593                                    isLittleEndian(), 0);
594     dumpAddrSection(OS, AddrData, DumpOpts, getMaxVersion(), getCUAddrSize());
595   }
596 
597   if (shouldDump(Explicit, ".debug_ranges", DIDT_ID_DebugRanges,
598                  DObj->getRangesSection().Data)) {
599     uint8_t savedAddressByteSize = getCUAddrSize();
600     DWARFDataExtractor rangesData(*DObj, DObj->getRangesSection(),
601                                   isLittleEndian(), savedAddressByteSize);
602     uint64_t offset = 0;
603     DWARFDebugRangeList rangeList;
604     while (rangesData.isValidOffset(offset)) {
605       if (Error E = rangeList.extract(rangesData, &offset)) {
606         DumpOpts.RecoverableErrorHandler(std::move(E));
607         break;
608       }
609       rangeList.dump(OS);
610     }
611   }
612 
613   auto LookupPooledAddress = [&](uint32_t Index) -> Optional<SectionedAddress> {
614     const auto &CUs = compile_units();
615     auto I = CUs.begin();
616     if (I == CUs.end())
617       return None;
618     return (*I)->getAddrOffsetSectionItem(Index);
619   };
620 
621   if (shouldDump(Explicit, ".debug_rnglists", DIDT_ID_DebugRnglists,
622                  DObj->getRnglistsSection().Data)) {
623     DWARFDataExtractor RnglistData(*DObj, DObj->getRnglistsSection(),
624                                    isLittleEndian(), 0);
625     dumpRnglistsSection(OS, RnglistData, LookupPooledAddress, DumpOpts);
626   }
627 
628   if (shouldDump(ExplicitDWO, ".debug_rnglists.dwo", DIDT_ID_DebugRnglists,
629                  DObj->getRnglistsDWOSection().Data)) {
630     DWARFDataExtractor RnglistData(*DObj, DObj->getRnglistsDWOSection(),
631                                    isLittleEndian(), 0);
632     dumpRnglistsSection(OS, RnglistData, LookupPooledAddress, DumpOpts);
633   }
634 
635   if (shouldDump(Explicit, ".debug_pubnames", DIDT_ID_DebugPubnames,
636                  DObj->getPubnamesSection().Data)) {
637     DWARFDataExtractor PubTableData(*DObj, DObj->getPubnamesSection(),
638                                     isLittleEndian(), 0);
639     dumpPubTableSection(OS, DumpOpts, PubTableData, /*GnuStyle=*/false);
640   }
641 
642   if (shouldDump(Explicit, ".debug_pubtypes", DIDT_ID_DebugPubtypes,
643                  DObj->getPubtypesSection().Data)) {
644     DWARFDataExtractor PubTableData(*DObj, DObj->getPubtypesSection(),
645                                     isLittleEndian(), 0);
646     dumpPubTableSection(OS, DumpOpts, PubTableData, /*GnuStyle=*/false);
647   }
648 
649   if (shouldDump(Explicit, ".debug_gnu_pubnames", DIDT_ID_DebugGnuPubnames,
650                  DObj->getGnuPubnamesSection().Data)) {
651     DWARFDataExtractor PubTableData(*DObj, DObj->getGnuPubnamesSection(),
652                                     isLittleEndian(), 0);
653     dumpPubTableSection(OS, DumpOpts, PubTableData, /*GnuStyle=*/true);
654   }
655 
656   if (shouldDump(Explicit, ".debug_gnu_pubtypes", DIDT_ID_DebugGnuPubtypes,
657                  DObj->getGnuPubtypesSection().Data)) {
658     DWARFDataExtractor PubTableData(*DObj, DObj->getGnuPubtypesSection(),
659                                     isLittleEndian(), 0);
660     dumpPubTableSection(OS, DumpOpts, PubTableData, /*GnuStyle=*/true);
661   }
662 
663   if (shouldDump(Explicit, ".debug_str_offsets", DIDT_ID_DebugStrOffsets,
664                  DObj->getStrOffsetsSection().Data))
665     dumpStringOffsetsSection(
666         OS, DumpOpts, "debug_str_offsets", *DObj, DObj->getStrOffsetsSection(),
667         DObj->getStrSection(), normal_units(), isLittleEndian());
668   if (shouldDump(ExplicitDWO, ".debug_str_offsets.dwo", DIDT_ID_DebugStrOffsets,
669                  DObj->getStrOffsetsDWOSection().Data))
670     dumpStringOffsetsSection(OS, DumpOpts, "debug_str_offsets.dwo", *DObj,
671                              DObj->getStrOffsetsDWOSection(),
672                              DObj->getStrDWOSection(), dwo_units(),
673                              isLittleEndian());
674 
675   if (shouldDump(Explicit, ".gdb_index", DIDT_ID_GdbIndex,
676                  DObj->getGdbIndexSection())) {
677     getGdbIndex().dump(OS);
678   }
679 
680   if (shouldDump(Explicit, ".apple_names", DIDT_ID_AppleNames,
681                  DObj->getAppleNamesSection().Data))
682     getAppleNames().dump(OS);
683 
684   if (shouldDump(Explicit, ".apple_types", DIDT_ID_AppleTypes,
685                  DObj->getAppleTypesSection().Data))
686     getAppleTypes().dump(OS);
687 
688   if (shouldDump(Explicit, ".apple_namespaces", DIDT_ID_AppleNamespaces,
689                  DObj->getAppleNamespacesSection().Data))
690     getAppleNamespaces().dump(OS);
691 
692   if (shouldDump(Explicit, ".apple_objc", DIDT_ID_AppleObjC,
693                  DObj->getAppleObjCSection().Data))
694     getAppleObjC().dump(OS);
695   if (shouldDump(Explicit, ".debug_names", DIDT_ID_DebugNames,
696                  DObj->getNamesSection().Data))
697     getDebugNames().dump(OS);
698 }
699 
700 DWARFCompileUnit *DWARFContext::getDWOCompileUnitForHash(uint64_t Hash) {
701   parseDWOUnits(LazyParse);
702 
703   if (const auto &CUI = getCUIndex()) {
704     if (const auto *R = CUI.getFromHash(Hash))
705       return dyn_cast_or_null<DWARFCompileUnit>(
706           DWOUnits.getUnitForIndexEntry(*R));
707     return nullptr;
708   }
709 
710   // If there's no index, just search through the CUs in the DWO - there's
711   // probably only one unless this is something like LTO - though an in-process
712   // built/cached lookup table could be used in that case to improve repeated
713   // lookups of different CUs in the DWO.
714   for (const auto &DWOCU : dwo_compile_units()) {
715     // Might not have parsed DWO ID yet.
716     if (!DWOCU->getDWOId()) {
717       if (Optional<uint64_t> DWOId =
718           toUnsigned(DWOCU->getUnitDIE().find(DW_AT_GNU_dwo_id)))
719         DWOCU->setDWOId(*DWOId);
720       else
721         // No DWO ID?
722         continue;
723     }
724     if (DWOCU->getDWOId() == Hash)
725       return dyn_cast<DWARFCompileUnit>(DWOCU.get());
726   }
727   return nullptr;
728 }
729 
730 DWARFDie DWARFContext::getDIEForOffset(uint64_t Offset) {
731   parseNormalUnits();
732   if (auto *CU = NormalUnits.getUnitForOffset(Offset))
733     return CU->getDIEForOffset(Offset);
734   return DWARFDie();
735 }
736 
737 bool DWARFContext::verify(raw_ostream &OS, DIDumpOptions DumpOpts) {
738   bool Success = true;
739   DWARFVerifier verifier(OS, *this, DumpOpts);
740 
741   Success &= verifier.handleDebugAbbrev();
742   if (DumpOpts.DumpType & DIDT_DebugInfo)
743     Success &= verifier.handleDebugInfo();
744   if (DumpOpts.DumpType & DIDT_DebugLine)
745     Success &= verifier.handleDebugLine();
746   Success &= verifier.handleAccelTables();
747   return Success;
748 }
749 
750 const DWARFUnitIndex &DWARFContext::getCUIndex() {
751   if (CUIndex)
752     return *CUIndex;
753 
754   DataExtractor CUIndexData(DObj->getCUIndexSection(), isLittleEndian(), 0);
755 
756   CUIndex = std::make_unique<DWARFUnitIndex>(DW_SECT_INFO);
757   CUIndex->parse(CUIndexData);
758   return *CUIndex;
759 }
760 
761 const DWARFUnitIndex &DWARFContext::getTUIndex() {
762   if (TUIndex)
763     return *TUIndex;
764 
765   DataExtractor TUIndexData(DObj->getTUIndexSection(), isLittleEndian(), 0);
766 
767   TUIndex = std::make_unique<DWARFUnitIndex>(DW_SECT_EXT_TYPES);
768   TUIndex->parse(TUIndexData);
769   return *TUIndex;
770 }
771 
772 DWARFGdbIndex &DWARFContext::getGdbIndex() {
773   if (GdbIndex)
774     return *GdbIndex;
775 
776   DataExtractor GdbIndexData(DObj->getGdbIndexSection(), true /*LE*/, 0);
777   GdbIndex = std::make_unique<DWARFGdbIndex>();
778   GdbIndex->parse(GdbIndexData);
779   return *GdbIndex;
780 }
781 
782 const DWARFDebugAbbrev *DWARFContext::getDebugAbbrev() {
783   if (Abbrev)
784     return Abbrev.get();
785 
786   DataExtractor abbrData(DObj->getAbbrevSection(), isLittleEndian(), 0);
787 
788   Abbrev.reset(new DWARFDebugAbbrev());
789   Abbrev->extract(abbrData);
790   return Abbrev.get();
791 }
792 
793 const DWARFDebugAbbrev *DWARFContext::getDebugAbbrevDWO() {
794   if (AbbrevDWO)
795     return AbbrevDWO.get();
796 
797   DataExtractor abbrData(DObj->getAbbrevDWOSection(), isLittleEndian(), 0);
798   AbbrevDWO.reset(new DWARFDebugAbbrev());
799   AbbrevDWO->extract(abbrData);
800   return AbbrevDWO.get();
801 }
802 
803 const DWARFDebugLoc *DWARFContext::getDebugLoc() {
804   if (Loc)
805     return Loc.get();
806 
807   // Assume all units have the same address byte size.
808   auto LocData =
809       getNumCompileUnits()
810           ? DWARFDataExtractor(*DObj, DObj->getLocSection(), isLittleEndian(),
811                                getUnitAtIndex(0)->getAddressByteSize())
812           : DWARFDataExtractor("", isLittleEndian(), 0);
813   Loc.reset(new DWARFDebugLoc(std::move(LocData)));
814   return Loc.get();
815 }
816 
817 const DWARFDebugAranges *DWARFContext::getDebugAranges() {
818   if (Aranges)
819     return Aranges.get();
820 
821   Aranges.reset(new DWARFDebugAranges());
822   Aranges->generate(this);
823   return Aranges.get();
824 }
825 
826 Expected<const DWARFDebugFrame *> DWARFContext::getDebugFrame() {
827   if (DebugFrame)
828     return DebugFrame.get();
829 
830   // There's a "bug" in the DWARFv3 standard with respect to the target address
831   // size within debug frame sections. While DWARF is supposed to be independent
832   // of its container, FDEs have fields with size being "target address size",
833   // which isn't specified in DWARF in general. It's only specified for CUs, but
834   // .eh_frame can appear without a .debug_info section. Follow the example of
835   // other tools (libdwarf) and extract this from the container (ObjectFile
836   // provides this information). This problem is fixed in DWARFv4
837   // See this dwarf-discuss discussion for more details:
838   // http://lists.dwarfstd.org/htdig.cgi/dwarf-discuss-dwarfstd.org/2011-December/001173.html
839   DWARFDataExtractor debugFrameData(*DObj, DObj->getFrameSection(),
840                                     isLittleEndian(), DObj->getAddressSize());
841   auto DF = std::make_unique<DWARFDebugFrame>(getArch(), /*IsEH=*/false);
842   if (Error E = DF->parse(debugFrameData))
843     return std::move(E);
844 
845   DebugFrame.swap(DF);
846   return DebugFrame.get();
847 }
848 
849 Expected<const DWARFDebugFrame *> DWARFContext::getEHFrame() {
850   if (EHFrame)
851     return EHFrame.get();
852 
853   DWARFDataExtractor debugFrameData(*DObj, DObj->getEHFrameSection(),
854                                     isLittleEndian(), DObj->getAddressSize());
855 
856   auto DF = std::make_unique<DWARFDebugFrame>(getArch(), /*IsEH=*/true);
857   if (Error E = DF->parse(debugFrameData))
858     return std::move(E);
859   DebugFrame.swap(DF);
860   return DebugFrame.get();
861 }
862 
863 const DWARFDebugMacro *DWARFContext::getDebugMacro() {
864   if (!Macro)
865     Macro = parseMacroOrMacinfo(MacroSection);
866   return Macro.get();
867 }
868 
869 const DWARFDebugMacro *DWARFContext::getDebugMacroDWO() {
870   if (!MacroDWO)
871     MacroDWO = parseMacroOrMacinfo(MacroDwoSection);
872   return MacroDWO.get();
873 }
874 
875 const DWARFDebugMacro *DWARFContext::getDebugMacinfo() {
876   if (!Macinfo)
877     Macinfo = parseMacroOrMacinfo(MacinfoSection);
878   return Macinfo.get();
879 }
880 
881 const DWARFDebugMacro *DWARFContext::getDebugMacinfoDWO() {
882   if (!MacinfoDWO)
883     MacinfoDWO = parseMacroOrMacinfo(MacinfoDwoSection);
884   return MacinfoDWO.get();
885 }
886 
887 template <typename T>
888 static T &getAccelTable(std::unique_ptr<T> &Cache, const DWARFObject &Obj,
889                         const DWARFSection &Section, StringRef StringSection,
890                         bool IsLittleEndian) {
891   if (Cache)
892     return *Cache;
893   DWARFDataExtractor AccelSection(Obj, Section, IsLittleEndian, 0);
894   DataExtractor StrData(StringSection, IsLittleEndian, 0);
895   Cache.reset(new T(AccelSection, StrData));
896   if (Error E = Cache->extract())
897     llvm::consumeError(std::move(E));
898   return *Cache;
899 }
900 
901 const DWARFDebugNames &DWARFContext::getDebugNames() {
902   return getAccelTable(Names, *DObj, DObj->getNamesSection(),
903                        DObj->getStrSection(), isLittleEndian());
904 }
905 
906 const AppleAcceleratorTable &DWARFContext::getAppleNames() {
907   return getAccelTable(AppleNames, *DObj, DObj->getAppleNamesSection(),
908                        DObj->getStrSection(), isLittleEndian());
909 }
910 
911 const AppleAcceleratorTable &DWARFContext::getAppleTypes() {
912   return getAccelTable(AppleTypes, *DObj, DObj->getAppleTypesSection(),
913                        DObj->getStrSection(), isLittleEndian());
914 }
915 
916 const AppleAcceleratorTable &DWARFContext::getAppleNamespaces() {
917   return getAccelTable(AppleNamespaces, *DObj,
918                        DObj->getAppleNamespacesSection(),
919                        DObj->getStrSection(), isLittleEndian());
920 }
921 
922 const AppleAcceleratorTable &DWARFContext::getAppleObjC() {
923   return getAccelTable(AppleObjC, *DObj, DObj->getAppleObjCSection(),
924                        DObj->getStrSection(), isLittleEndian());
925 }
926 
927 const DWARFDebugLine::LineTable *
928 DWARFContext::getLineTableForUnit(DWARFUnit *U) {
929   Expected<const DWARFDebugLine::LineTable *> ExpectedLineTable =
930       getLineTableForUnit(U, WarningHandler);
931   if (!ExpectedLineTable) {
932     WarningHandler(ExpectedLineTable.takeError());
933     return nullptr;
934   }
935   return *ExpectedLineTable;
936 }
937 
938 Expected<const DWARFDebugLine::LineTable *> DWARFContext::getLineTableForUnit(
939     DWARFUnit *U, function_ref<void(Error)> RecoverableErrorHandler) {
940   if (!Line)
941     Line.reset(new DWARFDebugLine);
942 
943   auto UnitDIE = U->getUnitDIE();
944   if (!UnitDIE)
945     return nullptr;
946 
947   auto Offset = toSectionOffset(UnitDIE.find(DW_AT_stmt_list));
948   if (!Offset)
949     return nullptr; // No line table for this compile unit.
950 
951   uint64_t stmtOffset = *Offset + U->getLineTableOffset();
952   // See if the line table is cached.
953   if (const DWARFLineTable *lt = Line->getLineTable(stmtOffset))
954     return lt;
955 
956   // Make sure the offset is good before we try to parse.
957   if (stmtOffset >= U->getLineSection().Data.size())
958     return nullptr;
959 
960   // We have to parse it first.
961   DWARFDataExtractor lineData(*DObj, U->getLineSection(), isLittleEndian(),
962                               U->getAddressByteSize());
963   return Line->getOrParseLineTable(lineData, stmtOffset, *this, U,
964                                    RecoverableErrorHandler);
965 }
966 
967 void DWARFContext::parseNormalUnits() {
968   if (!NormalUnits.empty())
969     return;
970   DObj->forEachInfoSections([&](const DWARFSection &S) {
971     NormalUnits.addUnitsForSection(*this, S, DW_SECT_INFO);
972   });
973   NormalUnits.finishedInfoUnits();
974   DObj->forEachTypesSections([&](const DWARFSection &S) {
975     NormalUnits.addUnitsForSection(*this, S, DW_SECT_EXT_TYPES);
976   });
977 }
978 
979 void DWARFContext::parseDWOUnits(bool Lazy) {
980   if (!DWOUnits.empty())
981     return;
982   DObj->forEachInfoDWOSections([&](const DWARFSection &S) {
983     DWOUnits.addUnitsForDWOSection(*this, S, DW_SECT_INFO, Lazy);
984   });
985   DWOUnits.finishedInfoUnits();
986   DObj->forEachTypesDWOSections([&](const DWARFSection &S) {
987     DWOUnits.addUnitsForDWOSection(*this, S, DW_SECT_EXT_TYPES, Lazy);
988   });
989 }
990 
991 DWARFCompileUnit *DWARFContext::getCompileUnitForOffset(uint64_t Offset) {
992   parseNormalUnits();
993   return dyn_cast_or_null<DWARFCompileUnit>(
994       NormalUnits.getUnitForOffset(Offset));
995 }
996 
997 DWARFCompileUnit *DWARFContext::getCompileUnitForAddress(uint64_t Address) {
998   // First, get the offset of the compile unit.
999   uint64_t CUOffset = getDebugAranges()->findAddress(Address);
1000   // Retrieve the compile unit.
1001   return getCompileUnitForOffset(CUOffset);
1002 }
1003 
1004 DWARFContext::DIEsForAddress DWARFContext::getDIEsForAddress(uint64_t Address) {
1005   DIEsForAddress Result;
1006 
1007   DWARFCompileUnit *CU = getCompileUnitForAddress(Address);
1008   if (!CU)
1009     return Result;
1010 
1011   Result.CompileUnit = CU;
1012   Result.FunctionDIE = CU->getSubroutineForAddress(Address);
1013 
1014   std::vector<DWARFDie> Worklist;
1015   Worklist.push_back(Result.FunctionDIE);
1016   while (!Worklist.empty()) {
1017     DWARFDie DIE = Worklist.back();
1018     Worklist.pop_back();
1019 
1020     if (!DIE.isValid())
1021       continue;
1022 
1023     if (DIE.getTag() == DW_TAG_lexical_block &&
1024         DIE.addressRangeContainsAddress(Address)) {
1025       Result.BlockDIE = DIE;
1026       break;
1027     }
1028 
1029     for (auto Child : DIE)
1030       Worklist.push_back(Child);
1031   }
1032 
1033   return Result;
1034 }
1035 
1036 /// TODO: change input parameter from "uint64_t Address"
1037 ///       into "SectionedAddress Address"
1038 static bool getFunctionNameAndStartLineForAddress(DWARFCompileUnit *CU,
1039                                                   uint64_t Address,
1040                                                   FunctionNameKind Kind,
1041                                                   std::string &FunctionName,
1042                                                   uint32_t &StartLine) {
1043   // The address may correspond to instruction in some inlined function,
1044   // so we have to build the chain of inlined functions and take the
1045   // name of the topmost function in it.
1046   SmallVector<DWARFDie, 4> InlinedChain;
1047   CU->getInlinedChainForAddress(Address, InlinedChain);
1048   if (InlinedChain.empty())
1049     return false;
1050 
1051   const DWARFDie &DIE = InlinedChain[0];
1052   bool FoundResult = false;
1053   const char *Name = nullptr;
1054   if (Kind != FunctionNameKind::None && (Name = DIE.getSubroutineName(Kind))) {
1055     FunctionName = Name;
1056     FoundResult = true;
1057   }
1058   if (auto DeclLineResult = DIE.getDeclLine()) {
1059     StartLine = DeclLineResult;
1060     FoundResult = true;
1061   }
1062 
1063   return FoundResult;
1064 }
1065 
1066 static Optional<uint64_t> getTypeSize(DWARFDie Type, uint64_t PointerSize) {
1067   if (auto SizeAttr = Type.find(DW_AT_byte_size))
1068     if (Optional<uint64_t> Size = SizeAttr->getAsUnsignedConstant())
1069       return Size;
1070 
1071   switch (Type.getTag()) {
1072   case DW_TAG_pointer_type:
1073   case DW_TAG_reference_type:
1074   case DW_TAG_rvalue_reference_type:
1075     return PointerSize;
1076   case DW_TAG_ptr_to_member_type: {
1077     if (DWARFDie BaseType = Type.getAttributeValueAsReferencedDie(DW_AT_type))
1078       if (BaseType.getTag() == DW_TAG_subroutine_type)
1079         return 2 * PointerSize;
1080     return PointerSize;
1081   }
1082   case DW_TAG_const_type:
1083   case DW_TAG_volatile_type:
1084   case DW_TAG_restrict_type:
1085   case DW_TAG_typedef: {
1086     if (DWARFDie BaseType = Type.getAttributeValueAsReferencedDie(DW_AT_type))
1087       return getTypeSize(BaseType, PointerSize);
1088     break;
1089   }
1090   case DW_TAG_array_type: {
1091     DWARFDie BaseType = Type.getAttributeValueAsReferencedDie(DW_AT_type);
1092     if (!BaseType)
1093       return Optional<uint64_t>();
1094     Optional<uint64_t> BaseSize = getTypeSize(BaseType, PointerSize);
1095     if (!BaseSize)
1096       return Optional<uint64_t>();
1097     uint64_t Size = *BaseSize;
1098     for (DWARFDie Child : Type) {
1099       if (Child.getTag() != DW_TAG_subrange_type)
1100         continue;
1101 
1102       if (auto ElemCountAttr = Child.find(DW_AT_count))
1103         if (Optional<uint64_t> ElemCount =
1104                 ElemCountAttr->getAsUnsignedConstant())
1105           Size *= *ElemCount;
1106       if (auto UpperBoundAttr = Child.find(DW_AT_upper_bound))
1107         if (Optional<int64_t> UpperBound =
1108                 UpperBoundAttr->getAsSignedConstant()) {
1109           int64_t LowerBound = 0;
1110           if (auto LowerBoundAttr = Child.find(DW_AT_lower_bound))
1111             LowerBound = LowerBoundAttr->getAsSignedConstant().getValueOr(0);
1112           Size *= *UpperBound - LowerBound + 1;
1113         }
1114     }
1115     return Size;
1116   }
1117   default:
1118     break;
1119   }
1120   return Optional<uint64_t>();
1121 }
1122 
1123 static Optional<int64_t>
1124 getExpressionFrameOffset(ArrayRef<uint8_t> Expr,
1125                          Optional<unsigned> FrameBaseReg) {
1126   if (!Expr.empty() &&
1127       (Expr[0] == DW_OP_fbreg ||
1128        (FrameBaseReg && Expr[0] == DW_OP_breg0 + *FrameBaseReg))) {
1129     unsigned Count;
1130     int64_t Offset = decodeSLEB128(Expr.data() + 1, &Count, Expr.end());
1131     // A single DW_OP_fbreg or DW_OP_breg.
1132     if (Expr.size() == Count + 1)
1133       return Offset;
1134     // Same + DW_OP_deref (Fortran arrays look like this).
1135     if (Expr.size() == Count + 2 && Expr[Count + 1] == DW_OP_deref)
1136       return Offset;
1137     // Fallthrough. Do not accept ex. (DW_OP_breg W29, DW_OP_stack_value)
1138   }
1139   return None;
1140 }
1141 
1142 void DWARFContext::addLocalsForDie(DWARFCompileUnit *CU, DWARFDie Subprogram,
1143                                    DWARFDie Die, std::vector<DILocal> &Result) {
1144   if (Die.getTag() == DW_TAG_variable ||
1145       Die.getTag() == DW_TAG_formal_parameter) {
1146     DILocal Local;
1147     if (const char *Name = Subprogram.getSubroutineName(DINameKind::ShortName))
1148       Local.FunctionName = Name;
1149 
1150     Optional<unsigned> FrameBaseReg;
1151     if (auto FrameBase = Subprogram.find(DW_AT_frame_base))
1152       if (Optional<ArrayRef<uint8_t>> Expr = FrameBase->getAsBlock())
1153         if (!Expr->empty() && (*Expr)[0] >= DW_OP_reg0 &&
1154             (*Expr)[0] <= DW_OP_reg31) {
1155           FrameBaseReg = (*Expr)[0] - DW_OP_reg0;
1156         }
1157 
1158     if (Expected<std::vector<DWARFLocationExpression>> Loc =
1159             Die.getLocations(DW_AT_location)) {
1160       for (const auto &Entry : *Loc) {
1161         if (Optional<int64_t> FrameOffset =
1162                 getExpressionFrameOffset(Entry.Expr, FrameBaseReg)) {
1163           Local.FrameOffset = *FrameOffset;
1164           break;
1165         }
1166       }
1167     } else {
1168       // FIXME: missing DW_AT_location is OK here, but other errors should be
1169       // reported to the user.
1170       consumeError(Loc.takeError());
1171     }
1172 
1173     if (auto TagOffsetAttr = Die.find(DW_AT_LLVM_tag_offset))
1174       Local.TagOffset = TagOffsetAttr->getAsUnsignedConstant();
1175 
1176     if (auto Origin =
1177             Die.getAttributeValueAsReferencedDie(DW_AT_abstract_origin))
1178       Die = Origin;
1179     if (auto NameAttr = Die.find(DW_AT_name))
1180       if (Optional<const char *> Name = NameAttr->getAsCString())
1181         Local.Name = *Name;
1182     if (auto Type = Die.getAttributeValueAsReferencedDie(DW_AT_type))
1183       Local.Size = getTypeSize(Type, getCUAddrSize());
1184     if (auto DeclFileAttr = Die.find(DW_AT_decl_file)) {
1185       if (const auto *LT = CU->getContext().getLineTableForUnit(CU))
1186         LT->getFileNameByIndex(
1187             DeclFileAttr->getAsUnsignedConstant().getValue(),
1188             CU->getCompilationDir(),
1189             DILineInfoSpecifier::FileLineInfoKind::AbsoluteFilePath,
1190             Local.DeclFile);
1191     }
1192     if (auto DeclLineAttr = Die.find(DW_AT_decl_line))
1193       Local.DeclLine = DeclLineAttr->getAsUnsignedConstant().getValue();
1194 
1195     Result.push_back(Local);
1196     return;
1197   }
1198 
1199   if (Die.getTag() == DW_TAG_inlined_subroutine)
1200     if (auto Origin =
1201             Die.getAttributeValueAsReferencedDie(DW_AT_abstract_origin))
1202       Subprogram = Origin;
1203 
1204   for (auto Child : Die)
1205     addLocalsForDie(CU, Subprogram, Child, Result);
1206 }
1207 
1208 std::vector<DILocal>
1209 DWARFContext::getLocalsForAddress(object::SectionedAddress Address) {
1210   std::vector<DILocal> Result;
1211   DWARFCompileUnit *CU = getCompileUnitForAddress(Address.Address);
1212   if (!CU)
1213     return Result;
1214 
1215   DWARFDie Subprogram = CU->getSubroutineForAddress(Address.Address);
1216   if (Subprogram.isValid())
1217     addLocalsForDie(CU, Subprogram, Subprogram, Result);
1218   return Result;
1219 }
1220 
1221 DILineInfo DWARFContext::getLineInfoForAddress(object::SectionedAddress Address,
1222                                                DILineInfoSpecifier Spec) {
1223   DILineInfo Result;
1224 
1225   DWARFCompileUnit *CU = getCompileUnitForAddress(Address.Address);
1226   if (!CU)
1227     return Result;
1228 
1229   getFunctionNameAndStartLineForAddress(CU, Address.Address, Spec.FNKind,
1230                                         Result.FunctionName, Result.StartLine);
1231   if (Spec.FLIKind != FileLineInfoKind::None) {
1232     if (const DWARFLineTable *LineTable = getLineTableForUnit(CU)) {
1233       LineTable->getFileLineInfoForAddress(
1234           {Address.Address, Address.SectionIndex}, CU->getCompilationDir(),
1235           Spec.FLIKind, Result);
1236     }
1237   }
1238   return Result;
1239 }
1240 
1241 DILineInfoTable DWARFContext::getLineInfoForAddressRange(
1242     object::SectionedAddress Address, uint64_t Size, DILineInfoSpecifier Spec) {
1243   DILineInfoTable  Lines;
1244   DWARFCompileUnit *CU = getCompileUnitForAddress(Address.Address);
1245   if (!CU)
1246     return Lines;
1247 
1248   uint32_t StartLine = 0;
1249   std::string FunctionName(DILineInfo::BadString);
1250   getFunctionNameAndStartLineForAddress(CU, Address.Address, Spec.FNKind,
1251                                         FunctionName, StartLine);
1252 
1253   // If the Specifier says we don't need FileLineInfo, just
1254   // return the top-most function at the starting address.
1255   if (Spec.FLIKind == FileLineInfoKind::None) {
1256     DILineInfo Result;
1257     Result.FunctionName = FunctionName;
1258     Result.StartLine = StartLine;
1259     Lines.push_back(std::make_pair(Address.Address, Result));
1260     return Lines;
1261   }
1262 
1263   const DWARFLineTable *LineTable = getLineTableForUnit(CU);
1264 
1265   // Get the index of row we're looking for in the line table.
1266   std::vector<uint32_t> RowVector;
1267   if (!LineTable->lookupAddressRange({Address.Address, Address.SectionIndex},
1268                                      Size, RowVector)) {
1269     return Lines;
1270   }
1271 
1272   for (uint32_t RowIndex : RowVector) {
1273     // Take file number and line/column from the row.
1274     const DWARFDebugLine::Row &Row = LineTable->Rows[RowIndex];
1275     DILineInfo Result;
1276     LineTable->getFileNameByIndex(Row.File, CU->getCompilationDir(),
1277                                   Spec.FLIKind, Result.FileName);
1278     Result.FunctionName = FunctionName;
1279     Result.Line = Row.Line;
1280     Result.Column = Row.Column;
1281     Result.StartLine = StartLine;
1282     Lines.push_back(std::make_pair(Row.Address.Address, Result));
1283   }
1284 
1285   return Lines;
1286 }
1287 
1288 DIInliningInfo
1289 DWARFContext::getInliningInfoForAddress(object::SectionedAddress Address,
1290                                         DILineInfoSpecifier Spec) {
1291   DIInliningInfo InliningInfo;
1292 
1293   DWARFCompileUnit *CU = getCompileUnitForAddress(Address.Address);
1294   if (!CU)
1295     return InliningInfo;
1296 
1297   const DWARFLineTable *LineTable = nullptr;
1298   SmallVector<DWARFDie, 4> InlinedChain;
1299   CU->getInlinedChainForAddress(Address.Address, InlinedChain);
1300   if (InlinedChain.size() == 0) {
1301     // If there is no DIE for address (e.g. it is in unavailable .dwo file),
1302     // try to at least get file/line info from symbol table.
1303     if (Spec.FLIKind != FileLineInfoKind::None) {
1304       DILineInfo Frame;
1305       LineTable = getLineTableForUnit(CU);
1306       if (LineTable && LineTable->getFileLineInfoForAddress(
1307                            {Address.Address, Address.SectionIndex},
1308                            CU->getCompilationDir(), Spec.FLIKind, Frame))
1309         InliningInfo.addFrame(Frame);
1310     }
1311     return InliningInfo;
1312   }
1313 
1314   uint32_t CallFile = 0, CallLine = 0, CallColumn = 0, CallDiscriminator = 0;
1315   for (uint32_t i = 0, n = InlinedChain.size(); i != n; i++) {
1316     DWARFDie &FunctionDIE = InlinedChain[i];
1317     DILineInfo Frame;
1318     // Get function name if necessary.
1319     if (const char *Name = FunctionDIE.getSubroutineName(Spec.FNKind))
1320       Frame.FunctionName = Name;
1321     if (auto DeclLineResult = FunctionDIE.getDeclLine())
1322       Frame.StartLine = DeclLineResult;
1323     if (Spec.FLIKind != FileLineInfoKind::None) {
1324       if (i == 0) {
1325         // For the topmost frame, initialize the line table of this
1326         // compile unit and fetch file/line info from it.
1327         LineTable = getLineTableForUnit(CU);
1328         // For the topmost routine, get file/line info from line table.
1329         if (LineTable)
1330           LineTable->getFileLineInfoForAddress(
1331               {Address.Address, Address.SectionIndex}, CU->getCompilationDir(),
1332               Spec.FLIKind, Frame);
1333       } else {
1334         // Otherwise, use call file, call line and call column from
1335         // previous DIE in inlined chain.
1336         if (LineTable)
1337           LineTable->getFileNameByIndex(CallFile, CU->getCompilationDir(),
1338                                         Spec.FLIKind, Frame.FileName);
1339         Frame.Line = CallLine;
1340         Frame.Column = CallColumn;
1341         Frame.Discriminator = CallDiscriminator;
1342       }
1343       // Get call file/line/column of a current DIE.
1344       if (i + 1 < n) {
1345         FunctionDIE.getCallerFrame(CallFile, CallLine, CallColumn,
1346                                    CallDiscriminator);
1347       }
1348     }
1349     InliningInfo.addFrame(Frame);
1350   }
1351   return InliningInfo;
1352 }
1353 
1354 std::shared_ptr<DWARFContext>
1355 DWARFContext::getDWOContext(StringRef AbsolutePath) {
1356   if (auto S = DWP.lock()) {
1357     DWARFContext *Ctxt = S->Context.get();
1358     return std::shared_ptr<DWARFContext>(std::move(S), Ctxt);
1359   }
1360 
1361   std::weak_ptr<DWOFile> *Entry = &DWOFiles[AbsolutePath];
1362 
1363   if (auto S = Entry->lock()) {
1364     DWARFContext *Ctxt = S->Context.get();
1365     return std::shared_ptr<DWARFContext>(std::move(S), Ctxt);
1366   }
1367 
1368   Expected<OwningBinary<ObjectFile>> Obj = [&] {
1369     if (!CheckedForDWP) {
1370       SmallString<128> DWPName;
1371       auto Obj = object::ObjectFile::createObjectFile(
1372           this->DWPName.empty()
1373               ? (DObj->getFileName() + ".dwp").toStringRef(DWPName)
1374               : StringRef(this->DWPName));
1375       if (Obj) {
1376         Entry = &DWP;
1377         return Obj;
1378       } else {
1379         CheckedForDWP = true;
1380         // TODO: Should this error be handled (maybe in a high verbosity mode)
1381         // before falling back to .dwo files?
1382         consumeError(Obj.takeError());
1383       }
1384     }
1385 
1386     return object::ObjectFile::createObjectFile(AbsolutePath);
1387   }();
1388 
1389   if (!Obj) {
1390     // TODO: Actually report errors helpfully.
1391     consumeError(Obj.takeError());
1392     return nullptr;
1393   }
1394 
1395   auto S = std::make_shared<DWOFile>();
1396   S->File = std::move(Obj.get());
1397   S->Context = DWARFContext::create(*S->File.getBinary());
1398   *Entry = S;
1399   auto *Ctxt = S->Context.get();
1400   return std::shared_ptr<DWARFContext>(std::move(S), Ctxt);
1401 }
1402 
1403 static Error createError(const Twine &Reason, llvm::Error E) {
1404   return make_error<StringError>(Reason + toString(std::move(E)),
1405                                  inconvertibleErrorCode());
1406 }
1407 
1408 /// SymInfo contains information about symbol: it's address
1409 /// and section index which is -1LL for absolute symbols.
1410 struct SymInfo {
1411   uint64_t Address;
1412   uint64_t SectionIndex;
1413 };
1414 
1415 /// Returns the address of symbol relocation used against and a section index.
1416 /// Used for futher relocations computation. Symbol's section load address is
1417 static Expected<SymInfo> getSymbolInfo(const object::ObjectFile &Obj,
1418                                        const RelocationRef &Reloc,
1419                                        const LoadedObjectInfo *L,
1420                                        std::map<SymbolRef, SymInfo> &Cache) {
1421   SymInfo Ret = {0, (uint64_t)-1LL};
1422   object::section_iterator RSec = Obj.section_end();
1423   object::symbol_iterator Sym = Reloc.getSymbol();
1424 
1425   std::map<SymbolRef, SymInfo>::iterator CacheIt = Cache.end();
1426   // First calculate the address of the symbol or section as it appears
1427   // in the object file
1428   if (Sym != Obj.symbol_end()) {
1429     bool New;
1430     std::tie(CacheIt, New) = Cache.insert({*Sym, {0, 0}});
1431     if (!New)
1432       return CacheIt->second;
1433 
1434     Expected<uint64_t> SymAddrOrErr = Sym->getAddress();
1435     if (!SymAddrOrErr)
1436       return createError("failed to compute symbol address: ",
1437                          SymAddrOrErr.takeError());
1438 
1439     // Also remember what section this symbol is in for later
1440     auto SectOrErr = Sym->getSection();
1441     if (!SectOrErr)
1442       return createError("failed to get symbol section: ",
1443                          SectOrErr.takeError());
1444 
1445     RSec = *SectOrErr;
1446     Ret.Address = *SymAddrOrErr;
1447   } else if (auto *MObj = dyn_cast<MachOObjectFile>(&Obj)) {
1448     RSec = MObj->getRelocationSection(Reloc.getRawDataRefImpl());
1449     Ret.Address = RSec->getAddress();
1450   }
1451 
1452   if (RSec != Obj.section_end())
1453     Ret.SectionIndex = RSec->getIndex();
1454 
1455   // If we are given load addresses for the sections, we need to adjust:
1456   // SymAddr = (Address of Symbol Or Section in File) -
1457   //           (Address of Section in File) +
1458   //           (Load Address of Section)
1459   // RSec is now either the section being targeted or the section
1460   // containing the symbol being targeted. In either case,
1461   // we need to perform the same computation.
1462   if (L && RSec != Obj.section_end())
1463     if (uint64_t SectionLoadAddress = L->getSectionLoadAddress(*RSec))
1464       Ret.Address += SectionLoadAddress - RSec->getAddress();
1465 
1466   if (CacheIt != Cache.end())
1467     CacheIt->second = Ret;
1468 
1469   return Ret;
1470 }
1471 
1472 static bool isRelocScattered(const object::ObjectFile &Obj,
1473                              const RelocationRef &Reloc) {
1474   const MachOObjectFile *MachObj = dyn_cast<MachOObjectFile>(&Obj);
1475   if (!MachObj)
1476     return false;
1477   // MachO also has relocations that point to sections and
1478   // scattered relocations.
1479   auto RelocInfo = MachObj->getRelocation(Reloc.getRawDataRefImpl());
1480   return MachObj->isRelocationScattered(RelocInfo);
1481 }
1482 
1483 namespace {
1484 struct DWARFSectionMap final : public DWARFSection {
1485   RelocAddrMap Relocs;
1486 };
1487 
1488 class DWARFObjInMemory final : public DWARFObject {
1489   bool IsLittleEndian;
1490   uint8_t AddressSize;
1491   StringRef FileName;
1492   const object::ObjectFile *Obj = nullptr;
1493   std::vector<SectionName> SectionNames;
1494 
1495   using InfoSectionMap = MapVector<object::SectionRef, DWARFSectionMap,
1496                                    std::map<object::SectionRef, unsigned>>;
1497 
1498   InfoSectionMap InfoSections;
1499   InfoSectionMap TypesSections;
1500   InfoSectionMap InfoDWOSections;
1501   InfoSectionMap TypesDWOSections;
1502 
1503   DWARFSectionMap LocSection;
1504   DWARFSectionMap LoclistsSection;
1505   DWARFSectionMap LoclistsDWOSection;
1506   DWARFSectionMap LineSection;
1507   DWARFSectionMap RangesSection;
1508   DWARFSectionMap RnglistsSection;
1509   DWARFSectionMap StrOffsetsSection;
1510   DWARFSectionMap LineDWOSection;
1511   DWARFSectionMap FrameSection;
1512   DWARFSectionMap EHFrameSection;
1513   DWARFSectionMap LocDWOSection;
1514   DWARFSectionMap StrOffsetsDWOSection;
1515   DWARFSectionMap RangesDWOSection;
1516   DWARFSectionMap RnglistsDWOSection;
1517   DWARFSectionMap AddrSection;
1518   DWARFSectionMap AppleNamesSection;
1519   DWARFSectionMap AppleTypesSection;
1520   DWARFSectionMap AppleNamespacesSection;
1521   DWARFSectionMap AppleObjCSection;
1522   DWARFSectionMap NamesSection;
1523   DWARFSectionMap PubnamesSection;
1524   DWARFSectionMap PubtypesSection;
1525   DWARFSectionMap GnuPubnamesSection;
1526   DWARFSectionMap GnuPubtypesSection;
1527   DWARFSectionMap MacroSection;
1528 
1529   DWARFSectionMap *mapNameToDWARFSection(StringRef Name) {
1530     return StringSwitch<DWARFSectionMap *>(Name)
1531         .Case("debug_loc", &LocSection)
1532         .Case("debug_loclists", &LoclistsSection)
1533         .Case("debug_loclists.dwo", &LoclistsDWOSection)
1534         .Case("debug_line", &LineSection)
1535         .Case("debug_frame", &FrameSection)
1536         .Case("eh_frame", &EHFrameSection)
1537         .Case("debug_str_offsets", &StrOffsetsSection)
1538         .Case("debug_ranges", &RangesSection)
1539         .Case("debug_rnglists", &RnglistsSection)
1540         .Case("debug_loc.dwo", &LocDWOSection)
1541         .Case("debug_line.dwo", &LineDWOSection)
1542         .Case("debug_names", &NamesSection)
1543         .Case("debug_rnglists.dwo", &RnglistsDWOSection)
1544         .Case("debug_str_offsets.dwo", &StrOffsetsDWOSection)
1545         .Case("debug_addr", &AddrSection)
1546         .Case("apple_names", &AppleNamesSection)
1547         .Case("debug_pubnames", &PubnamesSection)
1548         .Case("debug_pubtypes", &PubtypesSection)
1549         .Case("debug_gnu_pubnames", &GnuPubnamesSection)
1550         .Case("debug_gnu_pubtypes", &GnuPubtypesSection)
1551         .Case("apple_types", &AppleTypesSection)
1552         .Case("apple_namespaces", &AppleNamespacesSection)
1553         .Case("apple_namespac", &AppleNamespacesSection)
1554         .Case("apple_objc", &AppleObjCSection)
1555         .Case("debug_macro", &MacroSection)
1556         .Default(nullptr);
1557   }
1558 
1559   StringRef AbbrevSection;
1560   StringRef ArangesSection;
1561   StringRef StrSection;
1562   StringRef MacinfoSection;
1563   StringRef MacinfoDWOSection;
1564   StringRef MacroDWOSection;
1565   StringRef AbbrevDWOSection;
1566   StringRef StrDWOSection;
1567   StringRef CUIndexSection;
1568   StringRef GdbIndexSection;
1569   StringRef TUIndexSection;
1570   StringRef LineStrSection;
1571 
1572   // A deque holding section data whose iterators are not invalidated when
1573   // new decompressed sections are inserted at the end.
1574   std::deque<SmallString<0>> UncompressedSections;
1575 
1576   StringRef *mapSectionToMember(StringRef Name) {
1577     if (DWARFSection *Sec = mapNameToDWARFSection(Name))
1578       return &Sec->Data;
1579     return StringSwitch<StringRef *>(Name)
1580         .Case("debug_abbrev", &AbbrevSection)
1581         .Case("debug_aranges", &ArangesSection)
1582         .Case("debug_str", &StrSection)
1583         .Case("debug_macinfo", &MacinfoSection)
1584         .Case("debug_macinfo.dwo", &MacinfoDWOSection)
1585         .Case("debug_macro.dwo", &MacroDWOSection)
1586         .Case("debug_abbrev.dwo", &AbbrevDWOSection)
1587         .Case("debug_str.dwo", &StrDWOSection)
1588         .Case("debug_cu_index", &CUIndexSection)
1589         .Case("debug_tu_index", &TUIndexSection)
1590         .Case("gdb_index", &GdbIndexSection)
1591         .Case("debug_line_str", &LineStrSection)
1592         // Any more debug info sections go here.
1593         .Default(nullptr);
1594   }
1595 
1596   /// If Sec is compressed section, decompresses and updates its contents
1597   /// provided by Data. Otherwise leaves it unchanged.
1598   Error maybeDecompress(const object::SectionRef &Sec, StringRef Name,
1599                         StringRef &Data) {
1600     if (!Decompressor::isCompressed(Sec))
1601       return Error::success();
1602 
1603     Expected<Decompressor> Decompressor =
1604         Decompressor::create(Name, Data, IsLittleEndian, AddressSize == 8);
1605     if (!Decompressor)
1606       return Decompressor.takeError();
1607 
1608     SmallString<0> Out;
1609     if (auto Err = Decompressor->resizeAndDecompress(Out))
1610       return Err;
1611 
1612     UncompressedSections.push_back(std::move(Out));
1613     Data = UncompressedSections.back();
1614 
1615     return Error::success();
1616   }
1617 
1618 public:
1619   DWARFObjInMemory(const StringMap<std::unique_ptr<MemoryBuffer>> &Sections,
1620                    uint8_t AddrSize, bool IsLittleEndian)
1621       : IsLittleEndian(IsLittleEndian) {
1622     for (const auto &SecIt : Sections) {
1623       if (StringRef *SectionData = mapSectionToMember(SecIt.first()))
1624         *SectionData = SecIt.second->getBuffer();
1625       else if (SecIt.first() == "debug_info")
1626         // Find debug_info and debug_types data by section rather than name as
1627         // there are multiple, comdat grouped, of these sections.
1628         InfoSections[SectionRef()].Data = SecIt.second->getBuffer();
1629       else if (SecIt.first() == "debug_info.dwo")
1630         InfoDWOSections[SectionRef()].Data = SecIt.second->getBuffer();
1631       else if (SecIt.first() == "debug_types")
1632         TypesSections[SectionRef()].Data = SecIt.second->getBuffer();
1633       else if (SecIt.first() == "debug_types.dwo")
1634         TypesDWOSections[SectionRef()].Data = SecIt.second->getBuffer();
1635     }
1636   }
1637   DWARFObjInMemory(const object::ObjectFile &Obj, const LoadedObjectInfo *L,
1638                    function_ref<void(Error)> HandleError, function_ref<void(Error)> HandleWarning )
1639       : IsLittleEndian(Obj.isLittleEndian()),
1640         AddressSize(Obj.getBytesInAddress()), FileName(Obj.getFileName()),
1641         Obj(&Obj) {
1642 
1643     StringMap<unsigned> SectionAmountMap;
1644     for (const SectionRef &Section : Obj.sections()) {
1645       StringRef Name;
1646       if (auto NameOrErr = Section.getName())
1647         Name = *NameOrErr;
1648       else
1649         consumeError(NameOrErr.takeError());
1650 
1651       ++SectionAmountMap[Name];
1652       SectionNames.push_back({ Name, true });
1653 
1654       // Skip BSS and Virtual sections, they aren't interesting.
1655       if (Section.isBSS() || Section.isVirtual())
1656         continue;
1657 
1658       // Skip sections stripped by dsymutil.
1659       if (Section.isStripped())
1660         continue;
1661 
1662       StringRef Data;
1663       Expected<section_iterator> SecOrErr = Section.getRelocatedSection();
1664       if (!SecOrErr) {
1665         HandleError(createError("failed to get relocated section: ",
1666                                 SecOrErr.takeError()));
1667         continue;
1668       }
1669 
1670       // Try to obtain an already relocated version of this section.
1671       // Else use the unrelocated section from the object file. We'll have to
1672       // apply relocations ourselves later.
1673       section_iterator RelocatedSection = *SecOrErr;
1674       if (!L || !L->getLoadedSectionContents(*RelocatedSection, Data)) {
1675         Expected<StringRef> E = Section.getContents();
1676         if (E)
1677           Data = *E;
1678         else
1679           // maybeDecompress below will error.
1680           consumeError(E.takeError());
1681       }
1682 
1683       if (auto Err = maybeDecompress(Section, Name, Data)) {
1684         HandleError(createError("failed to decompress '" + Name + "', ",
1685                                 std::move(Err)));
1686         continue;
1687       }
1688 
1689       // Compressed sections names in GNU style starts from ".z",
1690       // at this point section is decompressed and we drop compression prefix.
1691       Name = Name.substr(
1692           Name.find_first_not_of("._z")); // Skip ".", "z" and "_" prefixes.
1693 
1694       // Map platform specific debug section names to DWARF standard section
1695       // names.
1696       Name = Obj.mapDebugSectionName(Name);
1697 
1698       if (StringRef *SectionData = mapSectionToMember(Name)) {
1699         *SectionData = Data;
1700         if (Name == "debug_ranges") {
1701           // FIXME: Use the other dwo range section when we emit it.
1702           RangesDWOSection.Data = Data;
1703         }
1704       } else if (Name == "debug_info") {
1705         // Find debug_info and debug_types data by section rather than name as
1706         // there are multiple, comdat grouped, of these sections.
1707         InfoSections[Section].Data = Data;
1708       } else if (Name == "debug_info.dwo") {
1709         InfoDWOSections[Section].Data = Data;
1710       } else if (Name == "debug_types") {
1711         TypesSections[Section].Data = Data;
1712       } else if (Name == "debug_types.dwo") {
1713         TypesDWOSections[Section].Data = Data;
1714       }
1715 
1716       if (RelocatedSection == Obj.section_end())
1717         continue;
1718 
1719       StringRef RelSecName;
1720       if (auto NameOrErr = RelocatedSection->getName())
1721         RelSecName = *NameOrErr;
1722       else
1723         consumeError(NameOrErr.takeError());
1724 
1725       // If the section we're relocating was relocated already by the JIT,
1726       // then we used the relocated version above, so we do not need to process
1727       // relocations for it now.
1728       StringRef RelSecData;
1729       if (L && L->getLoadedSectionContents(*RelocatedSection, RelSecData))
1730         continue;
1731 
1732       // In Mach-o files, the relocations do not need to be applied if
1733       // there is no load offset to apply. The value read at the
1734       // relocation point already factors in the section address
1735       // (actually applying the relocations will produce wrong results
1736       // as the section address will be added twice).
1737       if (!L && isa<MachOObjectFile>(&Obj))
1738         continue;
1739 
1740       RelSecName = RelSecName.substr(
1741           RelSecName.find_first_not_of("._z")); // Skip . and _ prefixes.
1742 
1743       // TODO: Add support for relocations in other sections as needed.
1744       // Record relocations for the debug_info and debug_line sections.
1745       DWARFSectionMap *Sec = mapNameToDWARFSection(RelSecName);
1746       RelocAddrMap *Map = Sec ? &Sec->Relocs : nullptr;
1747       if (!Map) {
1748         // Find debug_info and debug_types relocs by section rather than name
1749         // as there are multiple, comdat grouped, of these sections.
1750         if (RelSecName == "debug_info")
1751           Map = &static_cast<DWARFSectionMap &>(InfoSections[*RelocatedSection])
1752                      .Relocs;
1753         else if (RelSecName == "debug_info.dwo")
1754           Map = &static_cast<DWARFSectionMap &>(
1755                      InfoDWOSections[*RelocatedSection])
1756                      .Relocs;
1757         else if (RelSecName == "debug_types")
1758           Map =
1759               &static_cast<DWARFSectionMap &>(TypesSections[*RelocatedSection])
1760                    .Relocs;
1761         else if (RelSecName == "debug_types.dwo")
1762           Map = &static_cast<DWARFSectionMap &>(
1763                      TypesDWOSections[*RelocatedSection])
1764                      .Relocs;
1765         else
1766           continue;
1767       }
1768 
1769       if (Section.relocation_begin() == Section.relocation_end())
1770         continue;
1771 
1772       // Symbol to [address, section index] cache mapping.
1773       std::map<SymbolRef, SymInfo> AddrCache;
1774       bool (*Supports)(uint64_t);
1775       RelocationResolver Resolver;
1776       std::tie(Supports, Resolver) = getRelocationResolver(Obj);
1777       for (const RelocationRef &Reloc : Section.relocations()) {
1778         // FIXME: it's not clear how to correctly handle scattered
1779         // relocations.
1780         if (isRelocScattered(Obj, Reloc))
1781           continue;
1782 
1783         Expected<SymInfo> SymInfoOrErr =
1784             getSymbolInfo(Obj, Reloc, L, AddrCache);
1785         if (!SymInfoOrErr) {
1786           HandleError(SymInfoOrErr.takeError());
1787           continue;
1788         }
1789 
1790         // Check if Resolver can handle this relocation type early so as not to
1791         // handle invalid cases in DWARFDataExtractor.
1792         //
1793         // TODO Don't store Resolver in every RelocAddrEntry.
1794         if (Supports && Supports(Reloc.getType())) {
1795           auto I = Map->try_emplace(
1796               Reloc.getOffset(),
1797               RelocAddrEntry{SymInfoOrErr->SectionIndex, Reloc,
1798                              SymInfoOrErr->Address,
1799                              Optional<object::RelocationRef>(), 0, Resolver});
1800           // If we didn't successfully insert that's because we already had a
1801           // relocation for that offset. Store it as a second relocation in the
1802           // same RelocAddrEntry instead.
1803           if (!I.second) {
1804             RelocAddrEntry &entry = I.first->getSecond();
1805             if (entry.Reloc2) {
1806               HandleError(createError(
1807                   "At most two relocations per offset are supported"));
1808             }
1809             entry.Reloc2 = Reloc;
1810             entry.SymbolValue2 = SymInfoOrErr->Address;
1811           }
1812         } else {
1813           SmallString<32> Type;
1814           Reloc.getTypeName(Type);
1815           // FIXME: Support more relocations & change this to an error
1816           HandleWarning(
1817               createError("failed to compute relocation: " + Type + ", ",
1818                           errorCodeToError(object_error::parse_failed)));
1819         }
1820       }
1821     }
1822 
1823     for (SectionName &S : SectionNames)
1824       if (SectionAmountMap[S.Name] > 1)
1825         S.IsNameUnique = false;
1826   }
1827 
1828   Optional<RelocAddrEntry> find(const DWARFSection &S,
1829                                 uint64_t Pos) const override {
1830     auto &Sec = static_cast<const DWARFSectionMap &>(S);
1831     RelocAddrMap::const_iterator AI = Sec.Relocs.find(Pos);
1832     if (AI == Sec.Relocs.end())
1833       return None;
1834     return AI->second;
1835   }
1836 
1837   const object::ObjectFile *getFile() const override { return Obj; }
1838 
1839   ArrayRef<SectionName> getSectionNames() const override {
1840     return SectionNames;
1841   }
1842 
1843   bool isLittleEndian() const override { return IsLittleEndian; }
1844   StringRef getAbbrevDWOSection() const override { return AbbrevDWOSection; }
1845   const DWARFSection &getLineDWOSection() const override {
1846     return LineDWOSection;
1847   }
1848   const DWARFSection &getLocDWOSection() const override {
1849     return LocDWOSection;
1850   }
1851   StringRef getStrDWOSection() const override { return StrDWOSection; }
1852   const DWARFSection &getStrOffsetsDWOSection() const override {
1853     return StrOffsetsDWOSection;
1854   }
1855   const DWARFSection &getRangesDWOSection() const override {
1856     return RangesDWOSection;
1857   }
1858   const DWARFSection &getRnglistsDWOSection() const override {
1859     return RnglistsDWOSection;
1860   }
1861   const DWARFSection &getLoclistsDWOSection() const override {
1862     return LoclistsDWOSection;
1863   }
1864   const DWARFSection &getAddrSection() const override { return AddrSection; }
1865   StringRef getCUIndexSection() const override { return CUIndexSection; }
1866   StringRef getGdbIndexSection() const override { return GdbIndexSection; }
1867   StringRef getTUIndexSection() const override { return TUIndexSection; }
1868 
1869   // DWARF v5
1870   const DWARFSection &getStrOffsetsSection() const override {
1871     return StrOffsetsSection;
1872   }
1873   StringRef getLineStrSection() const override { return LineStrSection; }
1874 
1875   // Sections for DWARF5 split dwarf proposal.
1876   void forEachInfoDWOSections(
1877       function_ref<void(const DWARFSection &)> F) const override {
1878     for (auto &P : InfoDWOSections)
1879       F(P.second);
1880   }
1881   void forEachTypesDWOSections(
1882       function_ref<void(const DWARFSection &)> F) const override {
1883     for (auto &P : TypesDWOSections)
1884       F(P.second);
1885   }
1886 
1887   StringRef getAbbrevSection() const override { return AbbrevSection; }
1888   const DWARFSection &getLocSection() const override { return LocSection; }
1889   const DWARFSection &getLoclistsSection() const override { return LoclistsSection; }
1890   StringRef getArangesSection() const override { return ArangesSection; }
1891   const DWARFSection &getFrameSection() const override {
1892     return FrameSection;
1893   }
1894   const DWARFSection &getEHFrameSection() const override {
1895     return EHFrameSection;
1896   }
1897   const DWARFSection &getLineSection() const override { return LineSection; }
1898   StringRef getStrSection() const override { return StrSection; }
1899   const DWARFSection &getRangesSection() const override { return RangesSection; }
1900   const DWARFSection &getRnglistsSection() const override {
1901     return RnglistsSection;
1902   }
1903   const DWARFSection &getMacroSection() const override { return MacroSection; }
1904   StringRef getMacroDWOSection() const override { return MacroDWOSection; }
1905   StringRef getMacinfoSection() const override { return MacinfoSection; }
1906   StringRef getMacinfoDWOSection() const override { return MacinfoDWOSection; }
1907   const DWARFSection &getPubnamesSection() const override { return PubnamesSection; }
1908   const DWARFSection &getPubtypesSection() const override { return PubtypesSection; }
1909   const DWARFSection &getGnuPubnamesSection() const override {
1910     return GnuPubnamesSection;
1911   }
1912   const DWARFSection &getGnuPubtypesSection() const override {
1913     return GnuPubtypesSection;
1914   }
1915   const DWARFSection &getAppleNamesSection() const override {
1916     return AppleNamesSection;
1917   }
1918   const DWARFSection &getAppleTypesSection() const override {
1919     return AppleTypesSection;
1920   }
1921   const DWARFSection &getAppleNamespacesSection() const override {
1922     return AppleNamespacesSection;
1923   }
1924   const DWARFSection &getAppleObjCSection() const override {
1925     return AppleObjCSection;
1926   }
1927   const DWARFSection &getNamesSection() const override {
1928     return NamesSection;
1929   }
1930 
1931   StringRef getFileName() const override { return FileName; }
1932   uint8_t getAddressSize() const override { return AddressSize; }
1933   void forEachInfoSections(
1934       function_ref<void(const DWARFSection &)> F) const override {
1935     for (auto &P : InfoSections)
1936       F(P.second);
1937   }
1938   void forEachTypesSections(
1939       function_ref<void(const DWARFSection &)> F) const override {
1940     for (auto &P : TypesSections)
1941       F(P.second);
1942   }
1943 };
1944 } // namespace
1945 
1946 std::unique_ptr<DWARFContext>
1947 DWARFContext::create(const object::ObjectFile &Obj, const LoadedObjectInfo *L,
1948                      std::string DWPName,
1949                      std::function<void(Error)> RecoverableErrorHandler,
1950                      std::function<void(Error)> WarningHandler) {
1951   auto DObj =
1952       std::make_unique<DWARFObjInMemory>(Obj, L, RecoverableErrorHandler, WarningHandler);
1953   return std::make_unique<DWARFContext>(std::move(DObj), std::move(DWPName),
1954                                         RecoverableErrorHandler,
1955                                         WarningHandler);
1956 }
1957 
1958 std::unique_ptr<DWARFContext>
1959 DWARFContext::create(const StringMap<std::unique_ptr<MemoryBuffer>> &Sections,
1960                      uint8_t AddrSize, bool isLittleEndian,
1961                      std::function<void(Error)> RecoverableErrorHandler,
1962                      std::function<void(Error)> WarningHandler) {
1963   auto DObj =
1964       std::make_unique<DWARFObjInMemory>(Sections, AddrSize, isLittleEndian);
1965   return std::make_unique<DWARFContext>(
1966       std::move(DObj), "", RecoverableErrorHandler, WarningHandler);
1967 }
1968 
1969 Error DWARFContext::loadRegisterInfo(const object::ObjectFile &Obj) {
1970   // Detect the architecture from the object file. We usually don't need OS
1971   // info to lookup a target and create register info.
1972   Triple TT;
1973   TT.setArch(Triple::ArchType(Obj.getArch()));
1974   TT.setVendor(Triple::UnknownVendor);
1975   TT.setOS(Triple::UnknownOS);
1976   std::string TargetLookupError;
1977   const Target *TheTarget =
1978       TargetRegistry::lookupTarget(TT.str(), TargetLookupError);
1979   if (!TargetLookupError.empty())
1980     return createStringError(errc::invalid_argument,
1981                              TargetLookupError.c_str());
1982   RegInfo.reset(TheTarget->createMCRegInfo(TT.str()));
1983   return Error::success();
1984 }
1985 
1986 uint8_t DWARFContext::getCUAddrSize() {
1987   // In theory, different compile units may have different address byte
1988   // sizes, but for simplicity we just use the address byte size of the
1989   // first compile unit. In practice the address size field is repeated across
1990   // various DWARF headers (at least in version 5) to make it easier to dump
1991   // them independently, not to enable varying the address size.
1992   unit_iterator_range CUs = compile_units();
1993   return CUs.empty() ? 0 : (*CUs.begin())->getAddressByteSize();
1994 }
1995