1 //===- DWARFDie.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 "llvm/DebugInfo/DWARF/DWARFDie.h"
11 #include "SyntaxHighlighting.h"
12 #include "llvm/ADT/None.h"
13 #include "llvm/ADT/Optional.h"
14 #include "llvm/ADT/StringRef.h"
15 #include "llvm/BinaryFormat/Dwarf.h"
16 #include "llvm/DebugInfo/DWARF/DWARFAbbreviationDeclaration.h"
17 #include "llvm/DebugInfo/DWARF/DWARFContext.h"
18 #include "llvm/DebugInfo/DWARF/DWARFDebugRangeList.h"
19 #include "llvm/DebugInfo/DWARF/DWARFExpression.h"
20 #include "llvm/DebugInfo/DWARF/DWARFFormValue.h"
21 #include "llvm/DebugInfo/DWARF/DWARFUnit.h"
22 #include "llvm/Object/ObjectFile.h"
23 #include "llvm/Support/DataExtractor.h"
24 #include "llvm/Support/Format.h"
25 #include "llvm/Support/MathExtras.h"
26 #include "llvm/Support/raw_ostream.h"
27 #include <algorithm>
28 #include <cassert>
29 #include <cinttypes>
30 #include <cstdint>
31 #include <string>
32 #include <utility>
33 
34 using namespace llvm;
35 using namespace dwarf;
36 using namespace object;
37 using namespace syntax;
38 
39 static void dumpApplePropertyAttribute(raw_ostream &OS, uint64_t Val) {
40   OS << " (";
41   do {
42     uint64_t Shift = countTrailingZeros(Val);
43     assert(Shift < 64 && "undefined behavior");
44     uint64_t Bit = 1ULL << Shift;
45     auto PropName = ApplePropertyString(Bit);
46     if (!PropName.empty())
47       OS << PropName;
48     else
49       OS << format("DW_APPLE_PROPERTY_0x%" PRIx64, Bit);
50     if (!(Val ^= Bit))
51       break;
52     OS << ", ";
53   } while (true);
54   OS << ")";
55 }
56 
57 static void dumpRanges(const DWARFObject &Obj, raw_ostream &OS,
58                        const DWARFAddressRangesVector &Ranges,
59                        unsigned AddressSize, unsigned Indent,
60                        const DIDumpOptions &DumpOpts) {
61   ArrayRef<SectionName> SectionNames;
62   if (DumpOpts.Verbose)
63     SectionNames = Obj.getSectionNames();
64 
65   for (size_t I = 0; I < Ranges.size(); ++I) {
66     const DWARFAddressRange &R = Ranges[I];
67 
68     OS << '\n';
69     OS.indent(Indent);
70     OS << format("[0x%0*" PRIx64 " - 0x%0*" PRIx64 ")", AddressSize * 2,
71                  R.LowPC, AddressSize * 2, R.HighPC);
72 
73     if (SectionNames.empty() || R.SectionIndex == -1ULL)
74       continue;
75 
76     StringRef Name = SectionNames[R.SectionIndex].Name;
77     OS << " \"" << Name << '\"';
78 
79     // Print section index if name is not unique.
80     if (!SectionNames[R.SectionIndex].IsNameUnique)
81       OS << format(" [%" PRIu64 "]", R.SectionIndex);
82   }
83 }
84 
85 static void dumpLocation(raw_ostream &OS, DWARFFormValue &FormValue,
86                          DWARFUnit *U, unsigned Indent,
87                          DIDumpOptions DumpOpts) {
88   DWARFContext &Ctx = U->getContext();
89   const DWARFObject &Obj = Ctx.getDWARFObj();
90   const MCRegisterInfo *MRI = Ctx.getRegisterInfo();
91   if (FormValue.isFormClass(DWARFFormValue::FC_Block) ||
92       FormValue.isFormClass(DWARFFormValue::FC_Exprloc)) {
93     ArrayRef<uint8_t> Expr = *FormValue.getAsBlock();
94     DataExtractor Data(StringRef((const char *)Expr.data(), Expr.size()),
95                        Ctx.isLittleEndian(), 0);
96     DWARFExpression(Data, U->getVersion(), U->getAddressByteSize())
97         .print(OS, MRI);
98     return;
99   }
100 
101   FormValue.dump(OS, DumpOpts);
102   if (FormValue.isFormClass(DWARFFormValue::FC_SectionOffset)) {
103     const DWARFSection &LocSection = Obj.getLocSection();
104     const DWARFSection &LocDWOSection = Obj.getLocDWOSection();
105     uint32_t Offset = *FormValue.getAsSectionOffset();
106 
107     if (!LocSection.Data.empty()) {
108       DWARFDebugLoc DebugLoc;
109       DWARFDataExtractor Data(Obj, LocSection, Ctx.isLittleEndian(),
110                               Obj.getAddressSize());
111       auto LL = DebugLoc.parseOneLocationList(Data, &Offset);
112       if (LL)
113         LL->dump(OS, Ctx.isLittleEndian(), Obj.getAddressSize(), MRI, Indent);
114       else
115         OS << "error extracting location list.";
116     } else if (!LocDWOSection.Data.empty()) {
117       DataExtractor Data(LocDWOSection.Data, Ctx.isLittleEndian(), 0);
118       auto LL = DWARFDebugLocDWO::parseOneLocationList(Data, &Offset);
119       if (LL)
120         LL->dump(OS, Ctx.isLittleEndian(), Obj.getAddressSize(), MRI, Indent);
121       else
122         OS << "error extracting location list.";
123     }
124   }
125 }
126 
127 /// Dump the name encoded in the type tag.
128 static void dumpTypeTagName(raw_ostream &OS, dwarf::Tag T) {
129   StringRef TagStr = TagString(T);
130   if (!TagStr.startswith("DW_TAG_") || !TagStr.endswith("_type"))
131     return;
132   OS << TagStr.substr(7, TagStr.size() - 12) << " ";
133 }
134 
135 /// Recursively dump the DIE type name when applicable.
136 static void dumpTypeName(raw_ostream &OS, const DWARFDie &Die) {
137   DWARFDie D = Die.getAttributeValueAsReferencedDie(DW_AT_type);
138 
139   if (!D.isValid())
140     return;
141 
142   if (const char *Name = D.getName(DINameKind::LinkageName)) {
143     OS << Name;
144     return;
145   }
146 
147   // FIXME: We should have pretty printers per language. Currently we print
148   // everything as if it was C++ and fall back to the TAG type name.
149   const dwarf::Tag T = D.getTag();
150   switch (T) {
151   case DW_TAG_array_type:
152   case DW_TAG_pointer_type:
153   case DW_TAG_ptr_to_member_type:
154   case DW_TAG_reference_type:
155   case DW_TAG_rvalue_reference_type:
156     break;
157   default:
158     dumpTypeTagName(OS, T);
159   }
160 
161   // Follow the DW_AT_type if possible.
162   dumpTypeName(OS, D);
163 
164   switch (T) {
165   case DW_TAG_array_type:
166     OS << "[]";
167     break;
168   case DW_TAG_pointer_type:
169     OS << '*';
170     break;
171   case DW_TAG_ptr_to_member_type:
172     OS << '*';
173     break;
174   case DW_TAG_reference_type:
175     OS << '&';
176     break;
177   case DW_TAG_rvalue_reference_type:
178     OS << "&&";
179     break;
180   default:
181     break;
182   }
183 }
184 
185 static void dumpAttribute(raw_ostream &OS, const DWARFDie &Die,
186                           uint32_t *OffsetPtr, dwarf::Attribute Attr,
187                           dwarf::Form Form, unsigned Indent,
188                           DIDumpOptions DumpOpts) {
189   if (!Die.isValid())
190     return;
191   const char BaseIndent[] = "            ";
192   OS << BaseIndent;
193   OS.indent(Indent + 2);
194   auto attrString = AttributeString(Attr);
195   if (!attrString.empty())
196     WithColor(OS, syntax::Attribute) << attrString;
197   else
198     WithColor(OS, syntax::Attribute).get() << format("DW_AT_Unknown_%x", Attr);
199 
200   if (DumpOpts.Verbose || DumpOpts.ShowForm) {
201     auto formString = FormEncodingString(Form);
202     if (!formString.empty())
203       OS << " [" << formString << ']';
204     else
205       OS << format(" [DW_FORM_Unknown_%x]", Form);
206   }
207 
208   DWARFUnit *U = Die.getDwarfUnit();
209   DWARFFormValue formValue(Form);
210 
211   if (!formValue.extractValue(U->getDebugInfoExtractor(), OffsetPtr, U))
212     return;
213 
214   OS << "\t(";
215 
216   StringRef Name;
217   std::string File;
218   auto Color = syntax::Enumerator;
219   if (Attr == DW_AT_decl_file || Attr == DW_AT_call_file) {
220     Color = syntax::String;
221     if (const auto *LT = U->getContext().getLineTableForUnit(U))
222       if (LT->getFileNameByIndex(
223               formValue.getAsUnsignedConstant().getValue(),
224               U->getCompilationDir(),
225               DILineInfoSpecifier::FileLineInfoKind::AbsoluteFilePath, File)) {
226         File = '"' + File + '"';
227         Name = File;
228       }
229   } else if (Optional<uint64_t> Val = formValue.getAsUnsignedConstant())
230     Name = AttributeValueString(Attr, *Val);
231 
232   if (!Name.empty())
233     WithColor(OS, Color) << Name;
234   else if (Attr == DW_AT_decl_line || Attr == DW_AT_call_line)
235     OS << *formValue.getAsUnsignedConstant();
236   else if (Attr == DW_AT_location || Attr == DW_AT_frame_base ||
237            Attr == DW_AT_data_member_location)
238     dumpLocation(OS, formValue, U, sizeof(BaseIndent) + Indent + 4, DumpOpts);
239   else
240     formValue.dump(OS, DumpOpts);
241 
242   // We have dumped the attribute raw value. For some attributes
243   // having both the raw value and the pretty-printed value is
244   // interesting. These attributes are handled below.
245   if (Attr == DW_AT_specification || Attr == DW_AT_abstract_origin) {
246     if (const char *Name = Die.getAttributeValueAsReferencedDie(Attr).getName(
247             DINameKind::LinkageName))
248       OS << " \"" << Name << '\"';
249   } else if (Attr == DW_AT_type) {
250     OS << " \"";
251     dumpTypeName(OS, Die);
252     OS << '"';
253   } else if (Attr == DW_AT_APPLE_property_attribute) {
254     if (Optional<uint64_t> OptVal = formValue.getAsUnsignedConstant())
255       dumpApplePropertyAttribute(OS, *OptVal);
256   } else if (Attr == DW_AT_ranges) {
257     const DWARFObject &Obj = Die.getDwarfUnit()->getContext().getDWARFObj();
258     dumpRanges(Obj, OS, Die.getAddressRanges(), U->getAddressByteSize(),
259                sizeof(BaseIndent) + Indent + 4, DumpOpts);
260   }
261 
262   OS << ")\n";
263 }
264 
265 bool DWARFDie::isSubprogramDIE() const { return getTag() == DW_TAG_subprogram; }
266 
267 bool DWARFDie::isSubroutineDIE() const {
268   auto Tag = getTag();
269   return Tag == DW_TAG_subprogram || Tag == DW_TAG_inlined_subroutine;
270 }
271 
272 Optional<DWARFFormValue> DWARFDie::find(dwarf::Attribute Attr) const {
273   if (!isValid())
274     return None;
275   auto AbbrevDecl = getAbbreviationDeclarationPtr();
276   if (AbbrevDecl)
277     return AbbrevDecl->getAttributeValue(getOffset(), Attr, *U);
278   return None;
279 }
280 
281 Optional<DWARFFormValue>
282 DWARFDie::find(ArrayRef<dwarf::Attribute> Attrs) const {
283   if (!isValid())
284     return None;
285   auto AbbrevDecl = getAbbreviationDeclarationPtr();
286   if (AbbrevDecl) {
287     for (auto Attr : Attrs) {
288       if (auto Value = AbbrevDecl->getAttributeValue(getOffset(), Attr, *U))
289         return Value;
290     }
291   }
292   return None;
293 }
294 
295 Optional<DWARFFormValue>
296 DWARFDie::findRecursively(ArrayRef<dwarf::Attribute> Attrs) const {
297   if (!isValid())
298     return None;
299   auto Die = *this;
300   if (auto Value = Die.find(Attrs))
301     return Value;
302   if (auto D = Die.getAttributeValueAsReferencedDie(DW_AT_abstract_origin))
303     Die = D;
304   if (auto Value = Die.find(Attrs))
305     return Value;
306   if (auto D = Die.getAttributeValueAsReferencedDie(DW_AT_specification))
307     Die = D;
308   if (auto Value = Die.find(Attrs))
309     return Value;
310   return None;
311 }
312 
313 DWARFDie
314 DWARFDie::getAttributeValueAsReferencedDie(dwarf::Attribute Attr) const {
315   if (auto SpecRef = toReference(find(Attr))) {
316     if (auto SpecUnit = U->getUnitSection().getUnitForOffset(*SpecRef))
317       return SpecUnit->getDIEForOffset(*SpecRef);
318   }
319   return DWARFDie();
320 }
321 
322 Optional<uint64_t> DWARFDie::getRangesBaseAttribute() const {
323   return toSectionOffset(find({DW_AT_rnglists_base, DW_AT_GNU_ranges_base}));
324 }
325 
326 Optional<uint64_t> DWARFDie::getHighPC(uint64_t LowPC) const {
327   if (auto FormValue = find(DW_AT_high_pc)) {
328     if (auto Address = FormValue->getAsAddress()) {
329       // High PC is an address.
330       return Address;
331     }
332     if (auto Offset = FormValue->getAsUnsignedConstant()) {
333       // High PC is an offset from LowPC.
334       return LowPC + *Offset;
335     }
336   }
337   return None;
338 }
339 
340 bool DWARFDie::getLowAndHighPC(uint64_t &LowPC, uint64_t &HighPC,
341                                uint64_t &SectionIndex) const {
342   auto F = find(DW_AT_low_pc);
343   auto LowPcAddr = toAddress(F);
344   if (!LowPcAddr)
345     return false;
346   if (auto HighPcAddr = getHighPC(*LowPcAddr)) {
347     LowPC = *LowPcAddr;
348     HighPC = *HighPcAddr;
349     SectionIndex = F->getSectionIndex();
350     return true;
351   }
352   return false;
353 }
354 
355 DWARFAddressRangesVector DWARFDie::getAddressRanges() const {
356   if (isNULL())
357     return DWARFAddressRangesVector();
358   // Single range specified by low/high PC.
359   uint64_t LowPC, HighPC, Index;
360   if (getLowAndHighPC(LowPC, HighPC, Index))
361     return {{LowPC, HighPC, Index}};
362 
363   // Multiple ranges from .debug_ranges section.
364   auto RangesOffset = toSectionOffset(find(DW_AT_ranges));
365   if (RangesOffset) {
366     DWARFDebugRangeList RangeList;
367     if (U->extractRangeList(*RangesOffset, RangeList))
368       return RangeList.getAbsoluteRanges(U->getBaseAddress());
369   }
370   return DWARFAddressRangesVector();
371 }
372 
373 void DWARFDie::collectChildrenAddressRanges(
374     DWARFAddressRangesVector &Ranges) const {
375   if (isNULL())
376     return;
377   if (isSubprogramDIE()) {
378     const auto &DIERanges = getAddressRanges();
379     Ranges.insert(Ranges.end(), DIERanges.begin(), DIERanges.end());
380   }
381 
382   for (auto Child : children())
383     Child.collectChildrenAddressRanges(Ranges);
384 }
385 
386 bool DWARFDie::addressRangeContainsAddress(const uint64_t Address) const {
387   for (const auto &R : getAddressRanges()) {
388     if (R.LowPC <= Address && Address < R.HighPC)
389       return true;
390   }
391   return false;
392 }
393 
394 const char *DWARFDie::getSubroutineName(DINameKind Kind) const {
395   if (!isSubroutineDIE())
396     return nullptr;
397   return getName(Kind);
398 }
399 
400 const char *DWARFDie::getName(DINameKind Kind) const {
401   if (!isValid() || Kind == DINameKind::None)
402     return nullptr;
403   // Try to get mangled name only if it was asked for.
404   if (Kind == DINameKind::LinkageName) {
405     if (auto Name = dwarf::toString(
406             findRecursively({DW_AT_MIPS_linkage_name, DW_AT_linkage_name}),
407             nullptr))
408       return Name;
409   }
410   if (auto Name = dwarf::toString(findRecursively(DW_AT_name), nullptr))
411     return Name;
412   return nullptr;
413 }
414 
415 uint64_t DWARFDie::getDeclLine() const {
416   return toUnsigned(findRecursively(DW_AT_decl_line), 0);
417 }
418 
419 void DWARFDie::getCallerFrame(uint32_t &CallFile, uint32_t &CallLine,
420                               uint32_t &CallColumn,
421                               uint32_t &CallDiscriminator) const {
422   CallFile = toUnsigned(find(DW_AT_call_file), 0);
423   CallLine = toUnsigned(find(DW_AT_call_line), 0);
424   CallColumn = toUnsigned(find(DW_AT_call_column), 0);
425   CallDiscriminator = toUnsigned(find(DW_AT_GNU_discriminator), 0);
426 }
427 
428 /// Helper to dump a DIE with all of its parents, but no siblings.
429 static unsigned dumpParentChain(DWARFDie Die, raw_ostream &OS, unsigned Indent,
430                                 DIDumpOptions DumpOpts) {
431   if (!Die)
432     return Indent;
433   Indent = dumpParentChain(Die.getParent(), OS, Indent, DumpOpts);
434   Die.dump(OS, Indent, DumpOpts);
435   return Indent + 2;
436 }
437 
438 void DWARFDie::dump(raw_ostream &OS, unsigned Indent,
439                     DIDumpOptions DumpOpts) const {
440   if (!isValid())
441     return;
442   DWARFDataExtractor debug_info_data = U->getDebugInfoExtractor();
443   const uint32_t Offset = getOffset();
444   uint32_t offset = Offset;
445   if (DumpOpts.ShowParents) {
446     DumpOpts.ShowParents = false;
447     Indent = dumpParentChain(getParent(), OS, Indent, DumpOpts);
448   }
449 
450   if (debug_info_data.isValidOffset(offset)) {
451     uint32_t abbrCode = debug_info_data.getULEB128(&offset);
452     WithColor(OS, syntax::Address).get() << format("\n0x%8.8x: ", Offset);
453 
454     if (abbrCode) {
455       auto AbbrevDecl = getAbbreviationDeclarationPtr();
456       if (AbbrevDecl) {
457         auto tagString = TagString(getTag());
458         if (!tagString.empty())
459           WithColor(OS, syntax::Tag).get().indent(Indent) << tagString;
460         else
461           WithColor(OS, syntax::Tag).get().indent(Indent)
462               << format("DW_TAG_Unknown_%x", getTag());
463 
464         if (DumpOpts.Verbose)
465           OS << format(" [%u] %c", abbrCode,
466                        AbbrevDecl->hasChildren() ? '*' : ' ');
467         OS << '\n';
468 
469         // Dump all data in the DIE for the attributes.
470         for (const auto &AttrSpec : AbbrevDecl->attributes()) {
471           if (AttrSpec.Form == DW_FORM_implicit_const) {
472             // We are dumping .debug_info section ,
473             // implicit_const attribute values are not really stored here,
474             // but in .debug_abbrev section. So we just skip such attrs.
475             continue;
476           }
477           dumpAttribute(OS, *this, &offset, AttrSpec.Attr, AttrSpec.Form,
478                         Indent, DumpOpts);
479         }
480 
481         DWARFDie child = getFirstChild();
482         if (DumpOpts.RecurseDepth > 0 && child) {
483           DumpOpts.RecurseDepth--;
484           while (child) {
485             child.dump(OS, Indent + 2, DumpOpts);
486             child = child.getSibling();
487           }
488         }
489       } else {
490         OS << "Abbreviation code not found in 'debug_abbrev' class for code: "
491            << abbrCode << '\n';
492       }
493     } else {
494       OS.indent(Indent) << "NULL\n";
495     }
496   }
497 }
498 
499 LLVM_DUMP_METHOD void DWARFDie::dump() const { dump(llvm::errs(), 0); }
500 
501 DWARFDie DWARFDie::getParent() const {
502   if (isValid())
503     return U->getParent(Die);
504   return DWARFDie();
505 }
506 
507 DWARFDie DWARFDie::getSibling() const {
508   if (isValid())
509     return U->getSibling(Die);
510   return DWARFDie();
511 }
512 
513 iterator_range<DWARFDie::attribute_iterator> DWARFDie::attributes() const {
514   return make_range(attribute_iterator(*this, false),
515                     attribute_iterator(*this, true));
516 }
517 
518 DWARFDie::attribute_iterator::attribute_iterator(DWARFDie D, bool End)
519     : Die(D), AttrValue(0), Index(0) {
520   auto AbbrDecl = Die.getAbbreviationDeclarationPtr();
521   assert(AbbrDecl && "Must have abbreviation declaration");
522   if (End) {
523     // This is the end iterator so we set the index to the attribute count.
524     Index = AbbrDecl->getNumAttributes();
525   } else {
526     // This is the begin iterator so we extract the value for this->Index.
527     AttrValue.Offset = D.getOffset() + AbbrDecl->getCodeByteSize();
528     updateForIndex(*AbbrDecl, 0);
529   }
530 }
531 
532 void DWARFDie::attribute_iterator::updateForIndex(
533     const DWARFAbbreviationDeclaration &AbbrDecl, uint32_t I) {
534   Index = I;
535   // AbbrDecl must be valid before calling this function.
536   auto NumAttrs = AbbrDecl.getNumAttributes();
537   if (Index < NumAttrs) {
538     AttrValue.Attr = AbbrDecl.getAttrByIndex(Index);
539     // Add the previous byte size of any previous attribute value.
540     AttrValue.Offset += AttrValue.ByteSize;
541     AttrValue.Value.setForm(AbbrDecl.getFormByIndex(Index));
542     uint32_t ParseOffset = AttrValue.Offset;
543     auto U = Die.getDwarfUnit();
544     assert(U && "Die must have valid DWARF unit");
545     bool b = AttrValue.Value.extractValue(U->getDebugInfoExtractor(),
546                                           &ParseOffset, U);
547     (void)b;
548     assert(b && "extractValue cannot fail on fully parsed DWARF");
549     AttrValue.ByteSize = ParseOffset - AttrValue.Offset;
550   } else {
551     assert(Index == NumAttrs && "Indexes should be [0, NumAttrs) only");
552     AttrValue.clear();
553   }
554 }
555 
556 DWARFDie::attribute_iterator &DWARFDie::attribute_iterator::operator++() {
557   if (auto AbbrDecl = Die.getAbbreviationDeclarationPtr())
558     updateForIndex(*AbbrDecl, Index + 1);
559   return *this;
560 }
561