50ec4d3cb4766abdf05ee8be6c36cc37bf9e7111
[oota-llvm.git] / lib / DebugInfo / DWARFContext.cpp
1 //===-- DWARFContext.cpp --------------------------------------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9
10 #include "DWARFContext.h"
11 #include "llvm/ADT/SmallString.h"
12 #include "llvm/ADT/StringSwitch.h"
13 #include "llvm/Support/Compression.h"
14 #include "llvm/Support/Dwarf.h"
15 #include "llvm/Support/Format.h"
16 #include "llvm/Support/Path.h"
17 #include "llvm/Support/raw_ostream.h"
18 #include <algorithm>
19 using namespace llvm;
20 using namespace dwarf;
21 using namespace object;
22
23 typedef DWARFDebugLine::LineTable DWARFLineTable;
24
25 static void dumpPubSection(raw_ostream &OS, StringRef Name, StringRef Data,
26                            bool LittleEndian, bool GnuStyle) {
27   OS << "\n." << Name << " contents:\n";
28   DataExtractor pubNames(Data, LittleEndian, 0);
29   uint32_t offset = 0;
30   while (pubNames.isValidOffset(offset)) {
31     OS << "length = " << format("0x%08x", pubNames.getU32(&offset));
32     OS << " version = " << format("0x%04x", pubNames.getU16(&offset));
33     OS << " unit_offset = " << format("0x%08x", pubNames.getU32(&offset));
34     OS << " unit_size = " << format("0x%08x", pubNames.getU32(&offset)) << '\n';
35     if (GnuStyle)
36       OS << "Offset     Linkage  Kind     Name\n";
37     else
38       OS << "Offset     Name\n";
39
40     while (offset < Data.size()) {
41       uint32_t dieRef = pubNames.getU32(&offset);
42       if (dieRef == 0)
43         break;
44       OS << format("0x%8.8x ", dieRef);
45       if (GnuStyle) {
46         PubIndexEntryDescriptor desc(pubNames.getU8(&offset));
47         OS << format("%-8s", dwarf::GDBIndexEntryLinkageString(desc.Linkage))
48            << ' ' << format("%-8s", dwarf::GDBIndexEntryKindString(desc.Kind))
49            << ' ';
50       }
51       OS << '\"' << pubNames.getCStr(&offset) << "\"\n";
52     }
53   }
54 }
55
56 void DWARFContext::dump(raw_ostream &OS, DIDumpType DumpType) {
57   if (DumpType == DIDT_All || DumpType == DIDT_Abbrev) {
58     OS << ".debug_abbrev contents:\n";
59     getDebugAbbrev()->dump(OS);
60   }
61
62   if (DumpType == DIDT_All || DumpType == DIDT_AbbrevDwo)
63     if (const DWARFDebugAbbrev *D = getDebugAbbrevDWO()) {
64       OS << "\n.debug_abbrev.dwo contents:\n";
65       D->dump(OS);
66     }
67
68   if (DumpType == DIDT_All || DumpType == DIDT_Info) {
69     OS << "\n.debug_info contents:\n";
70     for (const auto &CU : compile_units())
71       CU->dump(OS);
72   }
73
74   if ((DumpType == DIDT_All || DumpType == DIDT_InfoDwo) &&
75       getNumDWOCompileUnits()) {
76     OS << "\n.debug_info.dwo contents:\n";
77     for (const auto &DWOCU : dwo_compile_units())
78       DWOCU->dump(OS);
79   }
80
81   if ((DumpType == DIDT_All || DumpType == DIDT_Types) && getNumTypeUnits()) {
82     OS << "\n.debug_types contents:\n";
83     for (const auto &TU : type_units())
84       TU->dump(OS);
85   }
86
87   if ((DumpType == DIDT_All || DumpType == DIDT_TypesDwo) &&
88       getNumDWOTypeUnits()) {
89     OS << "\n.debug_types.dwo contents:\n";
90     for (const auto &DWOTU : dwo_type_units())
91       DWOTU->dump(OS);
92   }
93
94   if (DumpType == DIDT_All || DumpType == DIDT_Loc) {
95     OS << "\n.debug_loc contents:\n";
96     getDebugLoc()->dump(OS);
97   }
98
99   if (DumpType == DIDT_All || DumpType == DIDT_LocDwo) {
100     OS << "\n.debug_loc.dwo contents:\n";
101     getDebugLocDWO()->dump(OS);
102   }
103
104   if (DumpType == DIDT_All || DumpType == DIDT_Frames) {
105     OS << "\n.debug_frame contents:\n";
106     getDebugFrame()->dump(OS);
107   }
108
109   uint32_t offset = 0;
110   if (DumpType == DIDT_All || DumpType == DIDT_Aranges) {
111     OS << "\n.debug_aranges contents:\n";
112     DataExtractor arangesData(getARangeSection(), isLittleEndian(), 0);
113     DWARFDebugArangeSet set;
114     while (set.extract(arangesData, &offset))
115       set.dump(OS);
116   }
117
118   uint8_t savedAddressByteSize = 0;
119   if (DumpType == DIDT_All || DumpType == DIDT_Line) {
120     OS << "\n.debug_line contents:\n";
121     for (const auto &CU : compile_units()) {
122       savedAddressByteSize = CU->getAddressByteSize();
123       unsigned stmtOffset =
124           CU->getCompileUnitDIE()->getAttributeValueAsSectionOffset(
125               CU.get(), DW_AT_stmt_list, -1U);
126       if (stmtOffset != -1U) {
127         DataExtractor lineData(getLineSection().Data, isLittleEndian(),
128                                savedAddressByteSize);
129         DWARFDebugLine::DumpingState state(OS);
130         DWARFDebugLine::parseStatementTable(lineData, &getLineSection().Relocs, &stmtOffset, state);
131       }
132     }
133   }
134
135   if (DumpType == DIDT_All || DumpType == DIDT_LineDwo) {
136     OS << "\n.debug_line.dwo contents:\n";
137     unsigned stmtOffset = 0;
138     DataExtractor lineData(getLineDWOSection().Data, isLittleEndian(),
139                            savedAddressByteSize);
140     DWARFDebugLine::DumpingState state(OS);
141     while (DWARFDebugLine::parsePrologue(lineData, &stmtOffset, &state.Prologue))
142       state.finalize();
143   }
144
145   if (DumpType == DIDT_All || DumpType == DIDT_Str) {
146     OS << "\n.debug_str contents:\n";
147     DataExtractor strData(getStringSection(), isLittleEndian(), 0);
148     offset = 0;
149     uint32_t strOffset = 0;
150     while (const char *s = strData.getCStr(&offset)) {
151       OS << format("0x%8.8x: \"%s\"\n", strOffset, s);
152       strOffset = offset;
153     }
154   }
155
156   if ((DumpType == DIDT_All || DumpType == DIDT_StrDwo) &&
157       !getStringDWOSection().empty()) {
158     OS << "\n.debug_str.dwo contents:\n";
159     DataExtractor strDWOData(getStringDWOSection(), isLittleEndian(), 0);
160     offset = 0;
161     uint32_t strDWOOffset = 0;
162     while (const char *s = strDWOData.getCStr(&offset)) {
163       OS << format("0x%8.8x: \"%s\"\n", strDWOOffset, s);
164       strDWOOffset = offset;
165     }
166   }
167
168   if (DumpType == DIDT_All || DumpType == DIDT_Ranges) {
169     OS << "\n.debug_ranges contents:\n";
170     // In fact, different compile units may have different address byte
171     // sizes, but for simplicity we just use the address byte size of the last
172     // compile unit (there is no easy and fast way to associate address range
173     // list and the compile unit it describes).
174     DataExtractor rangesData(getRangeSection(), isLittleEndian(),
175                              savedAddressByteSize);
176     offset = 0;
177     DWARFDebugRangeList rangeList;
178     while (rangeList.extract(rangesData, &offset))
179       rangeList.dump(OS);
180   }
181
182   if (DumpType == DIDT_All || DumpType == DIDT_Pubnames)
183     dumpPubSection(OS, "debug_pubnames", getPubNamesSection(),
184                    isLittleEndian(), false);
185
186   if (DumpType == DIDT_All || DumpType == DIDT_Pubtypes)
187     dumpPubSection(OS, "debug_pubtypes", getPubTypesSection(),
188                    isLittleEndian(), false);
189
190   if (DumpType == DIDT_All || DumpType == DIDT_GnuPubnames)
191     dumpPubSection(OS, "debug_gnu_pubnames", getGnuPubNamesSection(),
192                    isLittleEndian(), true /* GnuStyle */);
193
194   if (DumpType == DIDT_All || DumpType == DIDT_GnuPubtypes)
195     dumpPubSection(OS, "debug_gnu_pubtypes", getGnuPubTypesSection(),
196                    isLittleEndian(), true /* GnuStyle */);
197
198   if ((DumpType == DIDT_All || DumpType == DIDT_StrOffsetsDwo) &&
199       !getStringOffsetDWOSection().empty()) {
200     OS << "\n.debug_str_offsets.dwo contents:\n";
201     DataExtractor strOffsetExt(getStringOffsetDWOSection(), isLittleEndian(),
202                                0);
203     offset = 0;
204     uint64_t size = getStringOffsetDWOSection().size();
205     while (offset < size) {
206       OS << format("0x%8.8x: ", offset);
207       OS << format("%8.8x\n", strOffsetExt.getU32(&offset));
208     }
209   }
210 }
211
212 const DWARFDebugAbbrev *DWARFContext::getDebugAbbrev() {
213   if (Abbrev)
214     return Abbrev.get();
215
216   DataExtractor abbrData(getAbbrevSection(), isLittleEndian(), 0);
217
218   Abbrev.reset(new DWARFDebugAbbrev());
219   Abbrev->parse(abbrData);
220   return Abbrev.get();
221 }
222
223 const DWARFDebugAbbrev *DWARFContext::getDebugAbbrevDWO() {
224   if (AbbrevDWO)
225     return AbbrevDWO.get();
226
227   DataExtractor abbrData(getAbbrevDWOSection(), isLittleEndian(), 0);
228   AbbrevDWO.reset(new DWARFDebugAbbrev());
229   AbbrevDWO->parse(abbrData);
230   return AbbrevDWO.get();
231 }
232
233 const DWARFDebugLoc *DWARFContext::getDebugLoc() {
234   if (Loc)
235     return Loc.get();
236
237   DataExtractor LocData(getLocSection().Data, isLittleEndian(), 0);
238   Loc.reset(new DWARFDebugLoc(getLocSection().Relocs));
239   // assume all compile units have the same address byte size
240   if (getNumCompileUnits())
241     Loc->parse(LocData, getCompileUnitAtIndex(0)->getAddressByteSize());
242   return Loc.get();
243 }
244
245 const DWARFDebugLocDWO *DWARFContext::getDebugLocDWO() {
246   if (LocDWO)
247     return LocDWO.get();
248
249   DataExtractor LocData(getLocDWOSection().Data, isLittleEndian(), 0);
250   LocDWO.reset(new DWARFDebugLocDWO());
251   LocDWO->parse(LocData);
252   return LocDWO.get();
253 }
254
255 const DWARFDebugAranges *DWARFContext::getDebugAranges() {
256   if (Aranges)
257     return Aranges.get();
258
259   Aranges.reset(new DWARFDebugAranges());
260   Aranges->generate(this);
261   return Aranges.get();
262 }
263
264 const DWARFDebugFrame *DWARFContext::getDebugFrame() {
265   if (DebugFrame)
266     return DebugFrame.get();
267
268   // There's a "bug" in the DWARFv3 standard with respect to the target address
269   // size within debug frame sections. While DWARF is supposed to be independent
270   // of its container, FDEs have fields with size being "target address size",
271   // which isn't specified in DWARF in general. It's only specified for CUs, but
272   // .eh_frame can appear without a .debug_info section. Follow the example of
273   // other tools (libdwarf) and extract this from the container (ObjectFile
274   // provides this information). This problem is fixed in DWARFv4
275   // See this dwarf-discuss discussion for more details:
276   // http://lists.dwarfstd.org/htdig.cgi/dwarf-discuss-dwarfstd.org/2011-December/001173.html
277   DataExtractor debugFrameData(getDebugFrameSection(), isLittleEndian(),
278                                getAddressSize());
279   DebugFrame.reset(new DWARFDebugFrame());
280   DebugFrame->parse(debugFrameData);
281   return DebugFrame.get();
282 }
283
284 const DWARFLineTable *
285 DWARFContext::getLineTableForCompileUnit(DWARFCompileUnit *cu) {
286   if (!Line)
287     Line.reset(new DWARFDebugLine(&getLineSection().Relocs));
288
289   unsigned stmtOffset =
290       cu->getCompileUnitDIE()->getAttributeValueAsSectionOffset(
291           cu, DW_AT_stmt_list, -1U);
292   if (stmtOffset == -1U)
293     return nullptr; // No line table for this compile unit.
294
295   // See if the line table is cached.
296   if (const DWARFLineTable *lt = Line->getLineTable(stmtOffset))
297     return lt;
298
299   // We have to parse it first.
300   DataExtractor lineData(getLineSection().Data, isLittleEndian(),
301                          cu->getAddressByteSize());
302   return Line->getOrParseLineTable(lineData, stmtOffset);
303 }
304
305 void DWARFContext::parseCompileUnits() {
306   if (!CUs.empty())
307     return;
308   uint32_t offset = 0;
309   const DataExtractor &DIData = DataExtractor(getInfoSection().Data,
310                                               isLittleEndian(), 0);
311   while (DIData.isValidOffset(offset)) {
312     std::unique_ptr<DWARFCompileUnit> CU(new DWARFCompileUnit(
313         getDebugAbbrev(), getInfoSection().Data, getAbbrevSection(),
314         getRangeSection(), getStringSection(), StringRef(), getAddrSection(),
315         &getInfoSection().Relocs, isLittleEndian()));
316     if (!CU->extract(DIData, &offset)) {
317       break;
318     }
319     CUs.push_back(std::move(CU));
320     offset = CUs.back()->getNextUnitOffset();
321   }
322 }
323
324 void DWARFContext::parseTypeUnits() {
325   if (!TUs.empty())
326     return;
327   for (const auto &I : getTypesSections()) {
328     uint32_t offset = 0;
329     const DataExtractor &DIData =
330         DataExtractor(I.second.Data, isLittleEndian(), 0);
331     while (DIData.isValidOffset(offset)) {
332       std::unique_ptr<DWARFTypeUnit> TU(new DWARFTypeUnit(
333           getDebugAbbrev(), I.second.Data, getAbbrevSection(),
334           getRangeSection(), getStringSection(), StringRef(), getAddrSection(),
335           &I.second.Relocs, isLittleEndian()));
336       if (!TU->extract(DIData, &offset))
337         break;
338       TUs.push_back(std::move(TU));
339       offset = TUs.back()->getNextUnitOffset();
340     }
341   }
342 }
343
344 void DWARFContext::parseDWOCompileUnits() {
345   if (!DWOCUs.empty())
346     return;
347   uint32_t offset = 0;
348   const DataExtractor &DIData =
349       DataExtractor(getInfoDWOSection().Data, isLittleEndian(), 0);
350   while (DIData.isValidOffset(offset)) {
351     std::unique_ptr<DWARFCompileUnit> DWOCU(new DWARFCompileUnit(
352         getDebugAbbrevDWO(), getInfoDWOSection().Data, getAbbrevDWOSection(),
353         getRangeDWOSection(), getStringDWOSection(),
354         getStringOffsetDWOSection(), getAddrSection(),
355         &getInfoDWOSection().Relocs, isLittleEndian()));
356     if (!DWOCU->extract(DIData, &offset)) {
357       break;
358     }
359     DWOCUs.push_back(std::move(DWOCU));
360     offset = DWOCUs.back()->getNextUnitOffset();
361   }
362 }
363
364 void DWARFContext::parseDWOTypeUnits() {
365   if (!DWOTUs.empty())
366     return;
367   for (const auto &I : getTypesDWOSections()) {
368     uint32_t offset = 0;
369     const DataExtractor &DIData =
370         DataExtractor(I.second.Data, isLittleEndian(), 0);
371     while (DIData.isValidOffset(offset)) {
372       std::unique_ptr<DWARFTypeUnit> TU(new DWARFTypeUnit(
373           getDebugAbbrevDWO(), I.second.Data, getAbbrevDWOSection(),
374           getRangeDWOSection(), getStringDWOSection(),
375           getStringOffsetDWOSection(), getAddrSection(), &I.second.Relocs,
376           isLittleEndian()));
377       if (!TU->extract(DIData, &offset))
378         break;
379       DWOTUs.push_back(std::move(TU));
380       offset = DWOTUs.back()->getNextUnitOffset();
381     }
382   }
383 }
384
385 namespace {
386   struct OffsetComparator {
387
388     bool operator()(const std::unique_ptr<DWARFCompileUnit> &LHS,
389                     const std::unique_ptr<DWARFCompileUnit> &RHS) const {
390       return LHS->getOffset() < RHS->getOffset();
391     }
392     bool operator()(const std::unique_ptr<DWARFCompileUnit> &LHS,
393                     uint32_t RHS) const {
394       return LHS->getOffset() < RHS;
395     }
396     bool operator()(uint32_t LHS,
397                     const std::unique_ptr<DWARFCompileUnit> &RHS) const {
398       return LHS < RHS->getOffset();
399     }
400   };
401 }
402
403 DWARFCompileUnit *DWARFContext::getCompileUnitForOffset(uint32_t Offset) {
404   parseCompileUnits();
405
406   std::unique_ptr<DWARFCompileUnit> *CU =
407       std::lower_bound(CUs.begin(), CUs.end(), Offset, OffsetComparator());
408   if (CU != CUs.end()) {
409     return CU->get();
410   }
411   return nullptr;
412 }
413
414 DWARFCompileUnit *DWARFContext::getCompileUnitForAddress(uint64_t Address) {
415   // First, get the offset of the compile unit.
416   uint32_t CUOffset = getDebugAranges()->findAddress(Address);
417   // Retrieve the compile unit.
418   return getCompileUnitForOffset(CUOffset);
419 }
420
421 static bool getFileNameForCompileUnit(DWARFCompileUnit *CU,
422                                       const DWARFLineTable *LineTable,
423                                       uint64_t FileIndex,
424                                       bool NeedsAbsoluteFilePath,
425                                       std::string &FileName) {
426   if (!CU || !LineTable ||
427       !LineTable->getFileNameByIndex(FileIndex, NeedsAbsoluteFilePath,
428                                      FileName))
429     return false;
430   if (NeedsAbsoluteFilePath && sys::path::is_relative(FileName)) {
431     // We may still need to append compilation directory of compile unit.
432     SmallString<16> AbsolutePath;
433     if (const char *CompilationDir = CU->getCompilationDir()) {
434       sys::path::append(AbsolutePath, CompilationDir);
435     }
436     sys::path::append(AbsolutePath, FileName);
437     FileName = AbsolutePath.str();
438   }
439   return true;
440 }
441
442 static bool getFileLineInfoForCompileUnit(DWARFCompileUnit *CU,
443                                           const DWARFLineTable *LineTable,
444                                           uint64_t Address,
445                                           bool NeedsAbsoluteFilePath,
446                                           DILineInfo &Result) {
447   if (!CU || !LineTable)
448     return false;
449   // Get the index of row we're looking for in the line table.
450   uint32_t RowIndex = LineTable->lookupAddress(Address);
451   if (RowIndex == -1U)
452     return false;
453   // Take file number and line/column from the row.
454   const DWARFDebugLine::Row &Row = LineTable->Rows[RowIndex];
455   if (!getFileNameForCompileUnit(CU, LineTable, Row.File, NeedsAbsoluteFilePath,
456                                  Result.FileName))
457     return false;
458   Result.Line = Row.Line;
459   Result.Column = Row.Column;
460   return true;
461 }
462
463 static bool getFunctionNameForAddress(DWARFCompileUnit *CU, uint64_t Address,
464                                       std::string &FunctionName) {
465   // The address may correspond to instruction in some inlined function,
466   // so we have to build the chain of inlined functions and take the
467   // name of the topmost function in it.
468   const DWARFDebugInfoEntryInlinedChain &InlinedChain =
469       CU->getInlinedChainForAddress(Address);
470   if (InlinedChain.DIEs.size() == 0)
471     return false;
472   const DWARFDebugInfoEntryMinimal &TopFunctionDIE = InlinedChain.DIEs[0];
473   if (const char *Name = TopFunctionDIE.getSubroutineName(InlinedChain.U)) {
474     FunctionName = Name;
475     return true;
476   }
477   return false;
478 }
479
480 DILineInfo DWARFContext::getLineInfoForAddress(uint64_t Address,
481     DILineInfoSpecifier Specifier) {
482   DILineInfo Result;
483
484   DWARFCompileUnit *CU = getCompileUnitForAddress(Address);
485   if (!CU)
486     return Result;
487   if (Specifier.needs(DILineInfoSpecifier::FunctionName)) {
488     getFunctionNameForAddress(CU, Address, Result.FunctionName);
489   }
490   if (Specifier.needs(DILineInfoSpecifier::FileLineInfo)) {
491     const DWARFLineTable *LineTable = getLineTableForCompileUnit(CU);
492     const bool NeedsAbsoluteFilePath =
493         Specifier.needs(DILineInfoSpecifier::AbsoluteFilePath);
494     getFileLineInfoForCompileUnit(CU, LineTable, Address, NeedsAbsoluteFilePath,
495                                   Result);
496   }
497   return Result;
498 }
499
500 DILineInfoTable DWARFContext::getLineInfoForAddressRange(uint64_t Address,
501     uint64_t Size,
502     DILineInfoSpecifier Specifier) {
503   DILineInfoTable  Lines;
504   DWARFCompileUnit *CU = getCompileUnitForAddress(Address);
505   if (!CU)
506     return Lines;
507
508   std::string FunctionName = "<invalid>";
509   if (Specifier.needs(DILineInfoSpecifier::FunctionName)) {
510     getFunctionNameForAddress(CU, Address, FunctionName);
511   }
512
513   // If the Specifier says we don't need FileLineInfo, just
514   // return the top-most function at the starting address.
515   if (!Specifier.needs(DILineInfoSpecifier::FileLineInfo)) {
516     DILineInfo Result;
517     Result.FunctionName = FunctionName;
518     Lines.push_back(std::make_pair(Address, Result));
519     return Lines;
520   }
521
522   const DWARFLineTable *LineTable = getLineTableForCompileUnit(CU);
523   const bool NeedsAbsoluteFilePath =
524       Specifier.needs(DILineInfoSpecifier::AbsoluteFilePath);
525
526   // Get the index of row we're looking for in the line table.
527   std::vector<uint32_t> RowVector;
528   if (!LineTable->lookupAddressRange(Address, Size, RowVector))
529     return Lines;
530
531   for (uint32_t RowIndex : RowVector) {
532     // Take file number and line/column from the row.
533     const DWARFDebugLine::Row &Row = LineTable->Rows[RowIndex];
534     DILineInfo Result;
535     getFileNameForCompileUnit(CU, LineTable, Row.File, NeedsAbsoluteFilePath,
536                               Result.FileName);
537     Result.FunctionName = FunctionName;
538     Result.Line = Row.Line;
539     Result.Column = Row.Column;
540     Lines.push_back(std::make_pair(Row.Address, Result));
541   }
542
543   return Lines;
544 }
545
546 DIInliningInfo DWARFContext::getInliningInfoForAddress(uint64_t Address,
547     DILineInfoSpecifier Specifier) {
548   DWARFCompileUnit *CU = getCompileUnitForAddress(Address);
549   if (!CU)
550     return DIInliningInfo();
551
552   const DWARFDebugInfoEntryInlinedChain &InlinedChain =
553       CU->getInlinedChainForAddress(Address);
554   if (InlinedChain.DIEs.size() == 0)
555     return DIInliningInfo();
556
557   DIInliningInfo InliningInfo;
558   uint32_t CallFile = 0, CallLine = 0, CallColumn = 0;
559   const DWARFLineTable *LineTable = nullptr;
560   for (uint32_t i = 0, n = InlinedChain.DIEs.size(); i != n; i++) {
561     const DWARFDebugInfoEntryMinimal &FunctionDIE = InlinedChain.DIEs[i];
562     DILineInfo Frame;
563     // Get function name if necessary.
564     if (Specifier.needs(DILineInfoSpecifier::FunctionName)) {
565       if (const char *Name = FunctionDIE.getSubroutineName(InlinedChain.U))
566         Frame.FunctionName = Name;
567     }
568     if (Specifier.needs(DILineInfoSpecifier::FileLineInfo)) {
569       const bool NeedsAbsoluteFilePath =
570           Specifier.needs(DILineInfoSpecifier::AbsoluteFilePath);
571       if (i == 0) {
572         // For the topmost frame, initialize the line table of this
573         // compile unit and fetch file/line info from it.
574         LineTable = getLineTableForCompileUnit(CU);
575         // For the topmost routine, get file/line info from line table.
576         getFileLineInfoForCompileUnit(CU, LineTable, Address,
577                                       NeedsAbsoluteFilePath,
578                                       Frame);
579       } else {
580         // Otherwise, use call file, call line and call column from
581         // previous DIE in inlined chain.
582         getFileNameForCompileUnit(CU, LineTable, CallFile,
583                                   NeedsAbsoluteFilePath, Frame.FileName);
584         Frame.Line = CallLine;
585         Frame.Column = CallColumn;
586       }
587       // Get call file/line/column of a current DIE.
588       if (i + 1 < n) {
589         FunctionDIE.getCallerFrame(InlinedChain.U, CallFile, CallLine,
590                                    CallColumn);
591       }
592     }
593     InliningInfo.addFrame(Frame);
594   }
595   return InliningInfo;
596 }
597
598 static bool consumeCompressedDebugSectionHeader(StringRef &data,
599                                                 uint64_t &OriginalSize) {
600   // Consume "ZLIB" prefix.
601   if (!data.startswith("ZLIB"))
602     return false;
603   data = data.substr(4);
604   // Consume uncompressed section size (big-endian 8 bytes).
605   DataExtractor extractor(data, false, 8);
606   uint32_t Offset = 0;
607   OriginalSize = extractor.getU64(&Offset);
608   if (Offset == 0)
609     return false;
610   data = data.substr(Offset);
611   return true;
612 }
613
614 DWARFContextInMemory::DWARFContextInMemory(object::ObjectFile *Obj)
615     : IsLittleEndian(Obj->isLittleEndian()),
616       AddressSize(Obj->getBytesInAddress()) {
617   for (const SectionRef &Section : Obj->sections()) {
618     StringRef name;
619     Section.getName(name);
620     StringRef data;
621     Section.getContents(data);
622
623     name = name.substr(name.find_first_not_of("._")); // Skip . and _ prefixes.
624
625     // Check if debug info section is compressed with zlib.
626     if (name.startswith("zdebug_")) {
627       uint64_t OriginalSize;
628       if (!zlib::isAvailable() ||
629           !consumeCompressedDebugSectionHeader(data, OriginalSize))
630         continue;
631       UncompressedSections.resize(UncompressedSections.size() + 1);
632       if (zlib::uncompress(data, UncompressedSections.back(), OriginalSize) !=
633           zlib::StatusOK) {
634         UncompressedSections.pop_back();
635         continue;
636       }
637       // Make data point to uncompressed section contents and save its contents.
638       name = name.substr(1);
639       data = UncompressedSections.back();
640     }
641
642     StringRef *SectionData =
643         StringSwitch<StringRef *>(name)
644             .Case("debug_info", &InfoSection.Data)
645             .Case("debug_abbrev", &AbbrevSection)
646             .Case("debug_loc", &LocSection.Data)
647             .Case("debug_line", &LineSection.Data)
648             .Case("debug_aranges", &ARangeSection)
649             .Case("debug_frame", &DebugFrameSection)
650             .Case("debug_str", &StringSection)
651             .Case("debug_ranges", &RangeSection)
652             .Case("debug_pubnames", &PubNamesSection)
653             .Case("debug_pubtypes", &PubTypesSection)
654             .Case("debug_gnu_pubnames", &GnuPubNamesSection)
655             .Case("debug_gnu_pubtypes", &GnuPubTypesSection)
656             .Case("debug_info.dwo", &InfoDWOSection.Data)
657             .Case("debug_abbrev.dwo", &AbbrevDWOSection)
658             .Case("debug_loc.dwo", &LocDWOSection.Data)
659             .Case("debug_line.dwo", &LineDWOSection.Data)
660             .Case("debug_str.dwo", &StringDWOSection)
661             .Case("debug_str_offsets.dwo", &StringOffsetDWOSection)
662             .Case("debug_addr", &AddrSection)
663             // Any more debug info sections go here.
664             .Default(nullptr);
665     if (SectionData) {
666       *SectionData = data;
667       if (name == "debug_ranges") {
668         // FIXME: Use the other dwo range section when we emit it.
669         RangeDWOSection = data;
670       }
671     } else if (name == "debug_types") {
672       // Find debug_types data by section rather than name as there are
673       // multiple, comdat grouped, debug_types sections.
674       TypesSections[Section].Data = data;
675     } else if (name == "debug_types.dwo") {
676       TypesDWOSections[Section].Data = data;
677     }
678
679     section_iterator RelocatedSection = Section.getRelocatedSection();
680     if (RelocatedSection == Obj->section_end())
681       continue;
682
683     StringRef RelSecName;
684     RelocatedSection->getName(RelSecName);
685     RelSecName = RelSecName.substr(
686         RelSecName.find_first_not_of("._")); // Skip . and _ prefixes.
687
688     // TODO: Add support for relocations in other sections as needed.
689     // Record relocations for the debug_info and debug_line sections.
690     RelocAddrMap *Map = StringSwitch<RelocAddrMap*>(RelSecName)
691         .Case("debug_info", &InfoSection.Relocs)
692         .Case("debug_loc", &LocSection.Relocs)
693         .Case("debug_info.dwo", &InfoDWOSection.Relocs)
694         .Case("debug_line", &LineSection.Relocs)
695         .Default(nullptr);
696     if (!Map) {
697       // Find debug_types relocs by section rather than name as there are
698       // multiple, comdat grouped, debug_types sections.
699       if (RelSecName == "debug_types")
700         Map = &TypesSections[*RelocatedSection].Relocs;
701       else if (RelSecName == "debug_types.dwo")
702         Map = &TypesDWOSections[*RelocatedSection].Relocs;
703       else
704         continue;
705     }
706
707     if (Section.relocation_begin() != Section.relocation_end()) {
708       uint64_t SectionSize;
709       RelocatedSection->getSize(SectionSize);
710       for (const RelocationRef &Reloc : Section.relocations()) {
711         uint64_t Address;
712         Reloc.getOffset(Address);
713         uint64_t Type;
714         Reloc.getType(Type);
715         uint64_t SymAddr = 0;
716         // ELF relocations may need the symbol address
717         if (Obj->isELF()) {
718           object::symbol_iterator Sym = Reloc.getSymbol();
719           Sym->getAddress(SymAddr);
720         }
721
722         object::RelocVisitor V(Obj->getFileFormatName());
723         // The section address is always 0 for debug sections.
724         object::RelocToApply R(V.visit(Type, Reloc, 0, SymAddr));
725         if (V.error()) {
726           SmallString<32> Name;
727           error_code ec(Reloc.getTypeName(Name));
728           if (ec) {
729             errs() << "Aaaaaa! Nameless relocation! Aaaaaa!\n";
730           }
731           errs() << "error: failed to compute relocation: "
732                  << Name << "\n";
733           continue;
734         }
735
736         if (Address + R.Width > SectionSize) {
737           errs() << "error: " << R.Width << "-byte relocation starting "
738                  << Address << " bytes into section " << name << " which is "
739                  << SectionSize << " bytes long.\n";
740           continue;
741         }
742         if (R.Width > 8) {
743           errs() << "error: can't handle a relocation of more than 8 bytes at "
744                     "a time.\n";
745           continue;
746         }
747         DEBUG(dbgs() << "Writing " << format("%p", R.Value)
748                      << " at " << format("%p", Address)
749                      << " with width " << format("%d", R.Width)
750                      << "\n");
751         Map->insert(std::make_pair(Address, std::make_pair(R.Width, R.Value)));
752       }
753     }
754   }
755 }
756
757 void DWARFContextInMemory::anchor() { }