1 //===-- DWARFUnit.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 "DWARFUnit.h"
10 
11 #include "lldb/Core/Module.h"
12 #include "lldb/Host/StringConvert.h"
13 #include "lldb/Symbol/ObjectFile.h"
14 #include "lldb/Utility/LLDBAssert.h"
15 #include "lldb/Utility/StreamString.h"
16 #include "lldb/Utility/Timer.h"
17 #include "llvm/Object/Error.h"
18 
19 #include "DWARFCompileUnit.h"
20 #include "DWARFDebugAranges.h"
21 #include "DWARFDebugInfo.h"
22 #include "DWARFTypeUnit.h"
23 #include "LogChannelDWARF.h"
24 #include "SymbolFileDWARFDwo.h"
25 
26 using namespace lldb;
27 using namespace lldb_private;
28 using namespace std;
29 
30 extern int g_verbose;
31 
32 DWARFUnit::DWARFUnit(SymbolFileDWARF &dwarf, lldb::user_id_t uid,
33                      const DWARFUnitHeader &header,
34                      const DWARFAbbreviationDeclarationSet &abbrevs,
35                      DIERef::Section section, bool is_dwo)
36     : UserID(uid), m_dwarf(dwarf), m_header(header), m_abbrevs(&abbrevs),
37       m_cancel_scopes(false), m_section(section), m_is_dwo(is_dwo),
38       m_has_parsed_non_skeleton_unit(false), m_dwo_id(header.GetDWOId()) {}
39 
40 DWARFUnit::~DWARFUnit() = default;
41 
42 // Parses first DIE of a compile unit, excluding DWO.
43 void DWARFUnit::ExtractUnitDIENoDwoIfNeeded() {
44   {
45     llvm::sys::ScopedReader lock(m_first_die_mutex);
46     if (m_first_die)
47       return; // Already parsed
48   }
49   llvm::sys::ScopedWriter lock(m_first_die_mutex);
50   if (m_first_die)
51     return; // Already parsed
52 
53   LLDB_SCOPED_TIMERF("%8.8x: DWARFUnit::ExtractUnitDIENoDwoIfNeeded()",
54                      GetOffset());
55 
56   // Set the offset to that of the first DIE and calculate the start of the
57   // next compilation unit header.
58   lldb::offset_t offset = GetFirstDIEOffset();
59 
60   // We are in our compile unit, parse starting at the offset we were told to
61   // parse
62   const DWARFDataExtractor &data = GetData();
63   if (offset < GetNextUnitOffset() &&
64       m_first_die.Extract(data, this, &offset)) {
65     AddUnitDIE(m_first_die);
66     return;
67   }
68 }
69 
70 // Parses first DIE of a compile unit including DWO.
71 void DWARFUnit::ExtractUnitDIEIfNeeded() {
72   ExtractUnitDIENoDwoIfNeeded();
73 
74   if (m_has_parsed_non_skeleton_unit)
75     return;
76 
77   m_has_parsed_non_skeleton_unit = true;
78 
79   std::shared_ptr<SymbolFileDWARFDwo> dwo_symbol_file =
80       m_dwarf.GetDwoSymbolFileForCompileUnit(*this, m_first_die);
81   if (!dwo_symbol_file)
82     return;
83 
84   DWARFUnit *dwo_cu = dwo_symbol_file->GetDWOCompileUnitForHash(m_dwo_id);
85 
86   if (!dwo_cu)
87     return; // Can't fetch the compile unit from the dwo file.
88   dwo_cu->SetUserData(this);
89 
90   DWARFBaseDIE dwo_cu_die = dwo_cu->GetUnitDIEOnly();
91   if (!dwo_cu_die.IsValid())
92     return; // Can't fetch the compile unit DIE from the dwo file.
93 
94   // Here for DWO CU we want to use the address base set in the skeleton unit
95   // (DW_AT_addr_base) if it is available and use the DW_AT_GNU_addr_base
96   // otherwise. We do that because pre-DWARF v5 could use the DW_AT_GNU_*
97   // attributes which were applicable to the DWO units. The corresponding
98   // DW_AT_* attributes standardized in DWARF v5 are also applicable to the
99   // main unit in contrast.
100   if (m_addr_base)
101     dwo_cu->SetAddrBase(*m_addr_base);
102   else if (m_gnu_addr_base)
103     dwo_cu->SetAddrBase(*m_gnu_addr_base);
104 
105   if (GetVersion() <= 4 && m_gnu_ranges_base)
106     dwo_cu->SetRangesBase(*m_gnu_ranges_base);
107   else if (dwo_symbol_file->GetDWARFContext()
108                .getOrLoadRngListsData()
109                .GetByteSize() > 0)
110     dwo_cu->SetRangesBase(llvm::DWARFListTableHeader::getHeaderSize(DWARF32));
111 
112   if (GetVersion() >= 5 &&
113       dwo_symbol_file->GetDWARFContext().getOrLoadLocListsData().GetByteSize() >
114           0)
115     dwo_cu->SetLoclistsBase(llvm::DWARFListTableHeader::getHeaderSize(DWARF32));
116 
117   dwo_cu->SetBaseAddress(GetBaseAddress());
118 
119   m_dwo = std::shared_ptr<DWARFUnit>(std::move(dwo_symbol_file), dwo_cu);
120 }
121 
122 // Parses a compile unit and indexes its DIEs if it hasn't already been done.
123 // It will leave this compile unit extracted forever.
124 void DWARFUnit::ExtractDIEsIfNeeded() {
125   m_cancel_scopes = true;
126 
127   {
128     llvm::sys::ScopedReader lock(m_die_array_mutex);
129     if (!m_die_array.empty())
130       return; // Already parsed
131   }
132   llvm::sys::ScopedWriter lock(m_die_array_mutex);
133   if (!m_die_array.empty())
134     return; // Already parsed
135 
136   ExtractDIEsRWLocked();
137 }
138 
139 // Parses a compile unit and indexes its DIEs if it hasn't already been done.
140 // It will clear this compile unit after returned instance gets out of scope,
141 // no other ScopedExtractDIEs instance is running for this compile unit
142 // and no ExtractDIEsIfNeeded() has been executed during this ScopedExtractDIEs
143 // lifetime.
144 DWARFUnit::ScopedExtractDIEs DWARFUnit::ExtractDIEsScoped() {
145   ScopedExtractDIEs scoped(*this);
146 
147   {
148     llvm::sys::ScopedReader lock(m_die_array_mutex);
149     if (!m_die_array.empty())
150       return scoped; // Already parsed
151   }
152   llvm::sys::ScopedWriter lock(m_die_array_mutex);
153   if (!m_die_array.empty())
154     return scoped; // Already parsed
155 
156   // Otherwise m_die_array would be already populated.
157   lldbassert(!m_cancel_scopes);
158 
159   ExtractDIEsRWLocked();
160   scoped.m_clear_dies = true;
161   return scoped;
162 }
163 
164 DWARFUnit::ScopedExtractDIEs::ScopedExtractDIEs(DWARFUnit &cu) : m_cu(&cu) {
165   m_cu->m_die_array_scoped_mutex.lock_shared();
166 }
167 
168 DWARFUnit::ScopedExtractDIEs::~ScopedExtractDIEs() {
169   if (!m_cu)
170     return;
171   m_cu->m_die_array_scoped_mutex.unlock_shared();
172   if (!m_clear_dies || m_cu->m_cancel_scopes)
173     return;
174   // Be sure no other ScopedExtractDIEs is running anymore.
175   llvm::sys::ScopedWriter lock_scoped(m_cu->m_die_array_scoped_mutex);
176   llvm::sys::ScopedWriter lock(m_cu->m_die_array_mutex);
177   if (m_cu->m_cancel_scopes)
178     return;
179   m_cu->ClearDIEsRWLocked();
180 }
181 
182 DWARFUnit::ScopedExtractDIEs::ScopedExtractDIEs(ScopedExtractDIEs &&rhs)
183     : m_cu(rhs.m_cu), m_clear_dies(rhs.m_clear_dies) {
184   rhs.m_cu = nullptr;
185 }
186 
187 DWARFUnit::ScopedExtractDIEs &DWARFUnit::ScopedExtractDIEs::operator=(
188     DWARFUnit::ScopedExtractDIEs &&rhs) {
189   m_cu = rhs.m_cu;
190   rhs.m_cu = nullptr;
191   m_clear_dies = rhs.m_clear_dies;
192   return *this;
193 }
194 
195 // Parses a compile unit and indexes its DIEs, m_die_array_mutex must be
196 // held R/W and m_die_array must be empty.
197 void DWARFUnit::ExtractDIEsRWLocked() {
198   llvm::sys::ScopedWriter first_die_lock(m_first_die_mutex);
199 
200   LLDB_SCOPED_TIMERF("%8.8x: DWARFUnit::ExtractDIEsIfNeeded()", GetOffset());
201 
202   // Set the offset to that of the first DIE and calculate the start of the
203   // next compilation unit header.
204   lldb::offset_t offset = GetFirstDIEOffset();
205   lldb::offset_t next_cu_offset = GetNextUnitOffset();
206 
207   DWARFDebugInfoEntry die;
208 
209   uint32_t depth = 0;
210   // We are in our compile unit, parse starting at the offset we were told to
211   // parse
212   const DWARFDataExtractor &data = GetData();
213   std::vector<uint32_t> die_index_stack;
214   die_index_stack.reserve(32);
215   die_index_stack.push_back(0);
216   bool prev_die_had_children = false;
217   while (offset < next_cu_offset && die.Extract(data, this, &offset)) {
218     const bool null_die = die.IsNULL();
219     if (depth == 0) {
220       assert(m_die_array.empty() && "Compile unit DIE already added");
221 
222       // The average bytes per DIE entry has been seen to be around 14-20 so
223       // lets pre-reserve half of that since we are now stripping the NULL
224       // tags.
225 
226       // Only reserve the memory if we are adding children of the main
227       // compile unit DIE. The compile unit DIE is always the first entry, so
228       // if our size is 1, then we are adding the first compile unit child
229       // DIE and should reserve the memory.
230       m_die_array.reserve(GetDebugInfoSize() / 24);
231       m_die_array.push_back(die);
232 
233       if (!m_first_die)
234         AddUnitDIE(m_die_array.front());
235 
236       // With -fsplit-dwarf-inlining, clang will emit non-empty skeleton compile
237       // units. We are not able to access these DIE *and* the dwo file
238       // simultaneously. We also don't need to do that as the dwo file will
239       // contain a superset of information. So, we don't even attempt to parse
240       // any remaining DIEs.
241       if (m_dwo) {
242         m_die_array.front().SetHasChildren(false);
243         break;
244       }
245 
246     } else {
247       if (null_die) {
248         if (prev_die_had_children) {
249           // This will only happen if a DIE says is has children but all it
250           // contains is a NULL tag. Since we are removing the NULL DIEs from
251           // the list (saves up to 25% in C++ code), we need a way to let the
252           // DIE know that it actually doesn't have children.
253           if (!m_die_array.empty())
254             m_die_array.back().SetHasChildren(false);
255         }
256       } else {
257         die.SetParentIndex(m_die_array.size() - die_index_stack[depth - 1]);
258 
259         if (die_index_stack.back())
260           m_die_array[die_index_stack.back()].SetSiblingIndex(
261               m_die_array.size() - die_index_stack.back());
262 
263         // Only push the DIE if it isn't a NULL DIE
264         m_die_array.push_back(die);
265       }
266     }
267 
268     if (null_die) {
269       // NULL DIE.
270       if (!die_index_stack.empty())
271         die_index_stack.pop_back();
272 
273       if (depth > 0)
274         --depth;
275       prev_die_had_children = false;
276     } else {
277       die_index_stack.back() = m_die_array.size() - 1;
278       // Normal DIE
279       const bool die_has_children = die.HasChildren();
280       if (die_has_children) {
281         die_index_stack.push_back(0);
282         ++depth;
283       }
284       prev_die_had_children = die_has_children;
285     }
286 
287     if (depth == 0)
288       break; // We are done with this compile unit!
289   }
290 
291   if (!m_die_array.empty()) {
292     // The last die cannot have children (if it did, it wouldn't be the last one).
293     // This only makes a difference for malformed dwarf that does not have a
294     // terminating null die.
295     m_die_array.back().SetHasChildren(false);
296 
297     if (m_first_die) {
298       // Only needed for the assertion.
299       m_first_die.SetHasChildren(m_die_array.front().HasChildren());
300       lldbassert(m_first_die == m_die_array.front());
301     }
302     m_first_die = m_die_array.front();
303   }
304 
305   m_die_array.shrink_to_fit();
306 
307   if (m_dwo)
308     m_dwo->ExtractDIEsIfNeeded();
309 }
310 
311 // This is used when a split dwarf is enabled.
312 // A skeleton compilation unit may contain the DW_AT_str_offsets_base attribute
313 // that points to the first string offset of the CU contribution to the
314 // .debug_str_offsets. At the same time, the corresponding split debug unit also
315 // may use DW_FORM_strx* forms pointing to its own .debug_str_offsets.dwo and
316 // for that case, we should find the offset (skip the section header).
317 void DWARFUnit::SetDwoStrOffsetsBase() {
318   lldb::offset_t baseOffset = 0;
319 
320   if (const llvm::DWARFUnitIndex::Entry *entry = m_header.GetIndexEntry()) {
321     if (const auto *contribution =
322             entry->getContribution(llvm::DW_SECT_STR_OFFSETS))
323       baseOffset = contribution->Offset;
324     else
325       return;
326   }
327 
328   if (GetVersion() >= 5) {
329     const DWARFDataExtractor &strOffsets =
330         GetSymbolFileDWARF().GetDWARFContext().getOrLoadStrOffsetsData();
331     uint64_t length = strOffsets.GetU32(&baseOffset);
332     if (length == 0xffffffff)
333       length = strOffsets.GetU64(&baseOffset);
334 
335     // Check version.
336     if (strOffsets.GetU16(&baseOffset) < 5)
337       return;
338 
339     // Skip padding.
340     baseOffset += 2;
341   }
342 
343   SetStrOffsetsBase(baseOffset);
344 }
345 
346 uint64_t DWARFUnit::GetDWOId() {
347   ExtractUnitDIENoDwoIfNeeded();
348   return m_dwo_id;
349 }
350 
351 // m_die_array_mutex must be already held as read/write.
352 void DWARFUnit::AddUnitDIE(const DWARFDebugInfoEntry &cu_die) {
353   DWARFAttributes attributes;
354   size_t num_attributes = cu_die.GetAttributes(this, attributes);
355 
356   // Extract DW_AT_addr_base first, as other attributes may need it.
357   for (size_t i = 0; i < num_attributes; ++i) {
358     if (attributes.AttributeAtIndex(i) != DW_AT_addr_base)
359       continue;
360     DWARFFormValue form_value;
361     if (attributes.ExtractFormValueAtIndex(i, form_value)) {
362       SetAddrBase(form_value.Unsigned());
363       break;
364     }
365   }
366 
367   for (size_t i = 0; i < num_attributes; ++i) {
368     dw_attr_t attr = attributes.AttributeAtIndex(i);
369     DWARFFormValue form_value;
370     if (!attributes.ExtractFormValueAtIndex(i, form_value))
371       continue;
372     switch (attr) {
373     case DW_AT_loclists_base:
374       SetLoclistsBase(form_value.Unsigned());
375       break;
376     case DW_AT_rnglists_base:
377       SetRangesBase(form_value.Unsigned());
378       break;
379     case DW_AT_str_offsets_base:
380       SetStrOffsetsBase(form_value.Unsigned());
381       break;
382     case DW_AT_low_pc:
383       SetBaseAddress(form_value.Address());
384       break;
385     case DW_AT_entry_pc:
386       // If the value was already set by DW_AT_low_pc, don't update it.
387       if (m_base_addr == LLDB_INVALID_ADDRESS)
388         SetBaseAddress(form_value.Address());
389       break;
390     case DW_AT_stmt_list:
391       m_line_table_offset = form_value.Unsigned();
392       break;
393     case DW_AT_GNU_addr_base:
394       m_gnu_addr_base = form_value.Unsigned();
395       break;
396     case DW_AT_GNU_ranges_base:
397       m_gnu_ranges_base = form_value.Unsigned();
398       break;
399     case DW_AT_GNU_dwo_id:
400       m_dwo_id = form_value.Unsigned();
401       break;
402     }
403   }
404 
405   if (m_is_dwo) {
406     m_has_parsed_non_skeleton_unit = true;
407     SetDwoStrOffsetsBase();
408     return;
409   }
410 }
411 
412 size_t DWARFUnit::GetDebugInfoSize() const {
413   return GetLengthByteSize() + GetLength() - GetHeaderByteSize();
414 }
415 
416 const DWARFAbbreviationDeclarationSet *DWARFUnit::GetAbbreviations() const {
417   return m_abbrevs;
418 }
419 
420 dw_offset_t DWARFUnit::GetAbbrevOffset() const {
421   return m_abbrevs ? m_abbrevs->GetOffset() : DW_INVALID_OFFSET;
422 }
423 
424 dw_offset_t DWARFUnit::GetLineTableOffset() {
425   ExtractUnitDIENoDwoIfNeeded();
426   return m_line_table_offset;
427 }
428 
429 void DWARFUnit::SetAddrBase(dw_addr_t addr_base) { m_addr_base = addr_base; }
430 
431 // Parse the rangelist table header, including the optional array of offsets
432 // following it (DWARF v5 and later).
433 template <typename ListTableType>
434 static llvm::Expected<ListTableType>
435 ParseListTableHeader(const llvm::DWARFDataExtractor &data, uint64_t offset,
436                      DwarfFormat format) {
437   // We are expected to be called with Offset 0 or pointing just past the table
438   // header. Correct Offset in the latter case so that it points to the start
439   // of the header.
440   if (offset == 0) {
441     // This means DW_AT_rnglists_base is missing and therefore DW_FORM_rnglistx
442     // cannot be handled. Returning a default-constructed ListTableType allows
443     // DW_FORM_sec_offset to be supported.
444     return ListTableType();
445   }
446 
447   uint64_t HeaderSize = llvm::DWARFListTableHeader::getHeaderSize(format);
448   if (offset < HeaderSize)
449     return llvm::createStringError(errc::invalid_argument,
450                                    "did not detect a valid"
451                                    " list table with base = 0x%" PRIx64 "\n",
452                                    offset);
453   offset -= HeaderSize;
454   ListTableType Table;
455   if (llvm::Error E = Table.extractHeaderAndOffsets(data, &offset))
456     return std::move(E);
457   return Table;
458 }
459 
460 void DWARFUnit::SetLoclistsBase(dw_addr_t loclists_base) {
461   uint64_t offset = 0;
462   if (const llvm::DWARFUnitIndex::Entry *entry = m_header.GetIndexEntry()) {
463     const auto *contribution = entry->getContribution(llvm::DW_SECT_LOCLISTS);
464     if (!contribution) {
465       GetSymbolFileDWARF().GetObjectFile()->GetModule()->ReportError(
466           "Failed to find location list contribution for CU with DWO Id "
467           "0x%" PRIx64,
468           this->GetDWOId());
469       return;
470     }
471     offset += contribution->Offset;
472   }
473   m_loclists_base = loclists_base;
474 
475   uint64_t header_size = llvm::DWARFListTableHeader::getHeaderSize(DWARF32);
476   if (loclists_base < header_size)
477     return;
478 
479   m_loclist_table_header.emplace(".debug_loclists", "locations");
480   offset += loclists_base - header_size;
481   if (llvm::Error E = m_loclist_table_header->extract(
482           m_dwarf.GetDWARFContext().getOrLoadLocListsData().GetAsLLVM(),
483           &offset)) {
484     GetSymbolFileDWARF().GetObjectFile()->GetModule()->ReportError(
485         "Failed to extract location list table at offset 0x%" PRIx64
486         " (location list base: 0x%" PRIx64 "): %s",
487         offset, loclists_base, toString(std::move(E)).c_str());
488   }
489 }
490 
491 std::unique_ptr<llvm::DWARFLocationTable>
492 DWARFUnit::GetLocationTable(const DataExtractor &data) const {
493   llvm::DWARFDataExtractor llvm_data(
494       data.GetData(), data.GetByteOrder() == lldb::eByteOrderLittle,
495       data.GetAddressByteSize());
496 
497   if (m_is_dwo || GetVersion() >= 5)
498     return std::make_unique<llvm::DWARFDebugLoclists>(llvm_data, GetVersion());
499   return std::make_unique<llvm::DWARFDebugLoc>(llvm_data);
500 }
501 
502 DWARFDataExtractor DWARFUnit::GetLocationData() const {
503   DWARFContext &Ctx = GetSymbolFileDWARF().GetDWARFContext();
504   const DWARFDataExtractor &data =
505       GetVersion() >= 5 ? Ctx.getOrLoadLocListsData() : Ctx.getOrLoadLocData();
506   if (const llvm::DWARFUnitIndex::Entry *entry = m_header.GetIndexEntry()) {
507     if (const auto *contribution = entry->getContribution(
508             GetVersion() >= 5 ? llvm::DW_SECT_LOCLISTS : llvm::DW_SECT_EXT_LOC))
509       return DWARFDataExtractor(data, contribution->Offset,
510                                 contribution->Length);
511     return DWARFDataExtractor();
512   }
513   return data;
514 }
515 
516 void DWARFUnit::SetRangesBase(dw_addr_t ranges_base) {
517   lldbassert(!m_rnglist_table_done);
518 
519   m_ranges_base = ranges_base;
520 }
521 
522 const llvm::Optional<llvm::DWARFDebugRnglistTable> &
523 DWARFUnit::GetRnglistTable() {
524   if (GetVersion() >= 5 && !m_rnglist_table_done) {
525     m_rnglist_table_done = true;
526     if (auto table_or_error =
527             ParseListTableHeader<llvm::DWARFDebugRnglistTable>(
528                 m_dwarf.GetDWARFContext().getOrLoadRngListsData().GetAsLLVM(),
529                 m_ranges_base, DWARF32))
530       m_rnglist_table = std::move(table_or_error.get());
531     else
532       GetSymbolFileDWARF().GetObjectFile()->GetModule()->ReportError(
533           "Failed to extract range list table at offset 0x%" PRIx64 ": %s",
534           m_ranges_base, toString(table_or_error.takeError()).c_str());
535   }
536   return m_rnglist_table;
537 }
538 
539 // This function is called only for DW_FORM_rnglistx.
540 llvm::Expected<uint64_t> DWARFUnit::GetRnglistOffset(uint32_t Index) {
541   if (!GetRnglistTable())
542     return llvm::createStringError(errc::invalid_argument,
543                                    "missing or invalid range list table");
544   if (!m_ranges_base)
545     return llvm::createStringError(errc::invalid_argument,
546                                    "DW_FORM_rnglistx cannot be used without "
547                                    "DW_AT_rnglists_base for CU at 0x%8.8x",
548                                    GetOffset());
549   if (llvm::Optional<uint64_t> off = GetRnglistTable()->getOffsetEntry(
550           m_dwarf.GetDWARFContext().getOrLoadRngListsData().GetAsLLVM(), Index))
551     return *off + m_ranges_base;
552   return llvm::createStringError(
553       errc::invalid_argument,
554       "invalid range list table index %u; OffsetEntryCount is %u, "
555       "DW_AT_rnglists_base is %" PRIu64,
556       Index, GetRnglistTable()->getOffsetEntryCount(), m_ranges_base);
557 }
558 
559 void DWARFUnit::SetStrOffsetsBase(dw_offset_t str_offsets_base) {
560   m_str_offsets_base = str_offsets_base;
561 }
562 
563 // It may be called only with m_die_array_mutex held R/W.
564 void DWARFUnit::ClearDIEsRWLocked() {
565   m_die_array.clear();
566   m_die_array.shrink_to_fit();
567 
568   if (m_dwo)
569     m_dwo->ClearDIEsRWLocked();
570 }
571 
572 lldb::ByteOrder DWARFUnit::GetByteOrder() const {
573   return m_dwarf.GetObjectFile()->GetByteOrder();
574 }
575 
576 void DWARFUnit::SetBaseAddress(dw_addr_t base_addr) { m_base_addr = base_addr; }
577 
578 // Compare function DWARFDebugAranges::Range structures
579 static bool CompareDIEOffset(const DWARFDebugInfoEntry &die,
580                              const dw_offset_t die_offset) {
581   return die.GetOffset() < die_offset;
582 }
583 
584 // GetDIE()
585 //
586 // Get the DIE (Debug Information Entry) with the specified offset by first
587 // checking if the DIE is contained within this compile unit and grabbing the
588 // DIE from this compile unit. Otherwise we grab the DIE from the DWARF file.
589 DWARFDIE
590 DWARFUnit::GetDIE(dw_offset_t die_offset) {
591   if (die_offset == DW_INVALID_OFFSET)
592     return DWARFDIE(); // Not found
593 
594   if (!ContainsDIEOffset(die_offset)) {
595     GetSymbolFileDWARF().GetObjectFile()->GetModule()->ReportError(
596         "GetDIE for DIE 0x%" PRIx32 " is outside of its CU 0x%" PRIx32,
597         die_offset, GetOffset());
598     return DWARFDIE(); // Not found
599   }
600 
601   ExtractDIEsIfNeeded();
602   DWARFDebugInfoEntry::const_iterator end = m_die_array.cend();
603   DWARFDebugInfoEntry::const_iterator pos =
604       lower_bound(m_die_array.cbegin(), end, die_offset, CompareDIEOffset);
605 
606   if (pos != end && die_offset == (*pos).GetOffset())
607     return DWARFDIE(this, &(*pos));
608   return DWARFDIE(); // Not found
609 }
610 
611 DWARFUnit &DWARFUnit::GetNonSkeletonUnit() {
612   ExtractUnitDIEIfNeeded();
613   if (m_dwo)
614     return *m_dwo;
615   return *this;
616 }
617 
618 uint8_t DWARFUnit::GetAddressByteSize(const DWARFUnit *cu) {
619   if (cu)
620     return cu->GetAddressByteSize();
621   return DWARFUnit::GetDefaultAddressSize();
622 }
623 
624 uint8_t DWARFUnit::GetDefaultAddressSize() { return 4; }
625 
626 void *DWARFUnit::GetUserData() const { return m_user_data; }
627 
628 void DWARFUnit::SetUserData(void *d) { m_user_data = d; }
629 
630 bool DWARFUnit::Supports_DW_AT_APPLE_objc_complete_type() {
631   return GetProducer() != eProducerLLVMGCC;
632 }
633 
634 bool DWARFUnit::DW_AT_decl_file_attributes_are_invalid() {
635   // llvm-gcc makes completely invalid decl file attributes and won't ever be
636   // fixed, so we need to know to ignore these.
637   return GetProducer() == eProducerLLVMGCC;
638 }
639 
640 bool DWARFUnit::Supports_unnamed_objc_bitfields() {
641   if (GetProducer() == eProducerClang) {
642     const uint32_t major_version = GetProducerVersionMajor();
643     return major_version > 425 ||
644            (major_version == 425 && GetProducerVersionUpdate() >= 13);
645   }
646   return true; // Assume all other compilers didn't have incorrect ObjC bitfield
647                // info
648 }
649 
650 void DWARFUnit::ParseProducerInfo() {
651   m_producer_version_major = UINT32_MAX;
652   m_producer_version_minor = UINT32_MAX;
653   m_producer_version_update = UINT32_MAX;
654 
655   const DWARFDebugInfoEntry *die = GetUnitDIEPtrOnly();
656   if (die) {
657 
658     const char *producer_cstr =
659         die->GetAttributeValueAsString(this, DW_AT_producer, nullptr);
660     if (producer_cstr) {
661       RegularExpression llvm_gcc_regex(
662           llvm::StringRef("^4\\.[012]\\.[01] \\(Based on Apple "
663                           "Inc\\. build [0-9]+\\) \\(LLVM build "
664                           "[\\.0-9]+\\)$"));
665       if (llvm_gcc_regex.Execute(llvm::StringRef(producer_cstr))) {
666         m_producer = eProducerLLVMGCC;
667       } else if (strstr(producer_cstr, "clang")) {
668         static RegularExpression g_clang_version_regex(
669             llvm::StringRef("clang-([0-9]+)\\.([0-9]+)\\.([0-9]+)"));
670         llvm::SmallVector<llvm::StringRef, 4> matches;
671         if (g_clang_version_regex.Execute(llvm::StringRef(producer_cstr),
672                                           &matches)) {
673           m_producer_version_major =
674               StringConvert::ToUInt32(matches[1].str().c_str(), UINT32_MAX, 10);
675           m_producer_version_minor =
676               StringConvert::ToUInt32(matches[2].str().c_str(), UINT32_MAX, 10);
677           m_producer_version_update =
678               StringConvert::ToUInt32(matches[3].str().c_str(), UINT32_MAX, 10);
679         }
680         m_producer = eProducerClang;
681       } else if (strstr(producer_cstr, "GNU"))
682         m_producer = eProducerGCC;
683     }
684   }
685   if (m_producer == eProducerInvalid)
686     m_producer = eProcucerOther;
687 }
688 
689 DWARFProducer DWARFUnit::GetProducer() {
690   if (m_producer == eProducerInvalid)
691     ParseProducerInfo();
692   return m_producer;
693 }
694 
695 uint32_t DWARFUnit::GetProducerVersionMajor() {
696   if (m_producer_version_major == 0)
697     ParseProducerInfo();
698   return m_producer_version_major;
699 }
700 
701 uint32_t DWARFUnit::GetProducerVersionMinor() {
702   if (m_producer_version_minor == 0)
703     ParseProducerInfo();
704   return m_producer_version_minor;
705 }
706 
707 uint32_t DWARFUnit::GetProducerVersionUpdate() {
708   if (m_producer_version_update == 0)
709     ParseProducerInfo();
710   return m_producer_version_update;
711 }
712 
713 uint64_t DWARFUnit::GetDWARFLanguageType() {
714   if (m_language_type)
715     return *m_language_type;
716 
717   const DWARFDebugInfoEntry *die = GetUnitDIEPtrOnly();
718   if (!die)
719     m_language_type = 0;
720   else
721     m_language_type = die->GetAttributeValueAsUnsigned(this, DW_AT_language, 0);
722   return *m_language_type;
723 }
724 
725 bool DWARFUnit::GetIsOptimized() {
726   if (m_is_optimized == eLazyBoolCalculate) {
727     const DWARFDebugInfoEntry *die = GetUnitDIEPtrOnly();
728     if (die) {
729       m_is_optimized = eLazyBoolNo;
730       if (die->GetAttributeValueAsUnsigned(this, DW_AT_APPLE_optimized, 0) ==
731           1) {
732         m_is_optimized = eLazyBoolYes;
733       }
734     }
735   }
736   return m_is_optimized == eLazyBoolYes;
737 }
738 
739 FileSpec::Style DWARFUnit::GetPathStyle() {
740   if (!m_comp_dir)
741     ComputeCompDirAndGuessPathStyle();
742   return m_comp_dir->GetPathStyle();
743 }
744 
745 const FileSpec &DWARFUnit::GetCompilationDirectory() {
746   if (!m_comp_dir)
747     ComputeCompDirAndGuessPathStyle();
748   return *m_comp_dir;
749 }
750 
751 const FileSpec &DWARFUnit::GetAbsolutePath() {
752   if (!m_file_spec)
753     ComputeAbsolutePath();
754   return *m_file_spec;
755 }
756 
757 FileSpec DWARFUnit::GetFile(size_t file_idx) {
758   return m_dwarf.GetFile(*this, file_idx);
759 }
760 
761 // DWARF2/3 suggests the form hostname:pathname for compilation directory.
762 // Remove the host part if present.
763 static llvm::StringRef
764 removeHostnameFromPathname(llvm::StringRef path_from_dwarf) {
765   if (!path_from_dwarf.contains(':'))
766     return path_from_dwarf;
767   llvm::StringRef host, path;
768   std::tie(host, path) = path_from_dwarf.split(':');
769 
770   if (host.contains('/'))
771     return path_from_dwarf;
772 
773   // check whether we have a windows path, and so the first character is a
774   // drive-letter not a hostname.
775   if (host.size() == 1 && llvm::isAlpha(host[0]) && path.startswith("\\"))
776     return path_from_dwarf;
777 
778   return path;
779 }
780 
781 void DWARFUnit::ComputeCompDirAndGuessPathStyle() {
782   m_comp_dir = FileSpec();
783   const DWARFDebugInfoEntry *die = GetUnitDIEPtrOnly();
784   if (!die)
785     return;
786 
787   llvm::StringRef comp_dir = removeHostnameFromPathname(
788       die->GetAttributeValueAsString(this, DW_AT_comp_dir, nullptr));
789   if (!comp_dir.empty()) {
790     FileSpec::Style comp_dir_style =
791         FileSpec::GuessPathStyle(comp_dir).getValueOr(FileSpec::Style::native);
792     m_comp_dir = FileSpec(comp_dir, comp_dir_style);
793   } else {
794     // Try to detect the style based on the DW_AT_name attribute, but just store
795     // the detected style in the m_comp_dir field.
796     const char *name =
797         die->GetAttributeValueAsString(this, DW_AT_name, nullptr);
798     m_comp_dir = FileSpec(
799         "", FileSpec::GuessPathStyle(name).getValueOr(FileSpec::Style::native));
800   }
801 }
802 
803 void DWARFUnit::ComputeAbsolutePath() {
804   m_file_spec = FileSpec();
805   const DWARFDebugInfoEntry *die = GetUnitDIEPtrOnly();
806   if (!die)
807     return;
808 
809   m_file_spec =
810       FileSpec(die->GetAttributeValueAsString(this, DW_AT_name, nullptr),
811                GetPathStyle());
812 
813   if (m_file_spec->IsRelative())
814     m_file_spec->MakeAbsolute(GetCompilationDirectory());
815 }
816 
817 SymbolFileDWARFDwo *DWARFUnit::GetDwoSymbolFile() {
818   ExtractUnitDIEIfNeeded();
819   if (m_dwo)
820     return &llvm::cast<SymbolFileDWARFDwo>(m_dwo->GetSymbolFileDWARF());
821   return nullptr;
822 }
823 
824 const DWARFDebugAranges &DWARFUnit::GetFunctionAranges() {
825   if (m_func_aranges_up == nullptr) {
826     m_func_aranges_up = std::make_unique<DWARFDebugAranges>();
827     const DWARFDebugInfoEntry *die = DIEPtr();
828     if (die)
829       die->BuildFunctionAddressRangeTable(this, m_func_aranges_up.get());
830 
831     if (m_dwo) {
832       const DWARFDebugInfoEntry *dwo_die = m_dwo->DIEPtr();
833       if (dwo_die)
834         dwo_die->BuildFunctionAddressRangeTable(m_dwo.get(),
835                                                 m_func_aranges_up.get());
836     }
837 
838     const bool minimize = false;
839     m_func_aranges_up->Sort(minimize);
840   }
841   return *m_func_aranges_up;
842 }
843 
844 llvm::Expected<DWARFUnitHeader>
845 DWARFUnitHeader::extract(const DWARFDataExtractor &data,
846                          DIERef::Section section,
847                          lldb_private::DWARFContext &context,
848                          lldb::offset_t *offset_ptr) {
849   DWARFUnitHeader header;
850   header.m_offset = *offset_ptr;
851   header.m_length = data.GetDWARFInitialLength(offset_ptr);
852   header.m_version = data.GetU16(offset_ptr);
853   if (header.m_version == 5) {
854     header.m_unit_type = data.GetU8(offset_ptr);
855     header.m_addr_size = data.GetU8(offset_ptr);
856     header.m_abbr_offset = data.GetDWARFOffset(offset_ptr);
857     if (header.m_unit_type == llvm::dwarf::DW_UT_skeleton ||
858         header.m_unit_type == llvm::dwarf::DW_UT_split_compile)
859       header.m_dwo_id = data.GetU64(offset_ptr);
860   } else {
861     header.m_abbr_offset = data.GetDWARFOffset(offset_ptr);
862     header.m_addr_size = data.GetU8(offset_ptr);
863     header.m_unit_type =
864         section == DIERef::Section::DebugTypes ? DW_UT_type : DW_UT_compile;
865   }
866 
867   if (context.isDwo()) {
868     if (header.IsTypeUnit()) {
869       header.m_index_entry =
870           context.GetAsLLVM().getTUIndex().getFromOffset(header.m_offset);
871     } else {
872       header.m_index_entry =
873           context.GetAsLLVM().getCUIndex().getFromOffset(header.m_offset);
874     }
875   }
876 
877   if (header.m_index_entry) {
878     if (header.m_abbr_offset) {
879       return llvm::createStringError(
880           llvm::inconvertibleErrorCode(),
881           "Package unit with a non-zero abbreviation offset");
882     }
883     auto *unit_contrib = header.m_index_entry->getContribution();
884     if (!unit_contrib || unit_contrib->Length != header.m_length + 4) {
885       return llvm::createStringError(llvm::inconvertibleErrorCode(),
886                                      "Inconsistent DWARF package unit index");
887     }
888     auto *abbr_entry =
889         header.m_index_entry->getContribution(llvm::DW_SECT_ABBREV);
890     if (!abbr_entry) {
891       return llvm::createStringError(
892           llvm::inconvertibleErrorCode(),
893           "DWARF package index missing abbreviation column");
894     }
895     header.m_abbr_offset = abbr_entry->Offset;
896   }
897   if (header.IsTypeUnit()) {
898     header.m_type_hash = data.GetU64(offset_ptr);
899     header.m_type_offset = data.GetDWARFOffset(offset_ptr);
900   }
901 
902   bool length_OK = data.ValidOffset(header.GetNextUnitOffset() - 1);
903   bool version_OK = SymbolFileDWARF::SupportedVersion(header.m_version);
904   bool addr_size_OK = (header.m_addr_size == 4) || (header.m_addr_size == 8);
905   bool type_offset_OK =
906       !header.IsTypeUnit() || (header.m_type_offset <= header.GetLength());
907 
908   if (!length_OK)
909     return llvm::make_error<llvm::object::GenericBinaryError>(
910         "Invalid unit length");
911   if (!version_OK)
912     return llvm::make_error<llvm::object::GenericBinaryError>(
913         "Unsupported unit version");
914   if (!addr_size_OK)
915     return llvm::make_error<llvm::object::GenericBinaryError>(
916         "Invalid unit address size");
917   if (!type_offset_OK)
918     return llvm::make_error<llvm::object::GenericBinaryError>(
919         "Type offset out of range");
920 
921   return header;
922 }
923 
924 llvm::Expected<DWARFUnitSP>
925 DWARFUnit::extract(SymbolFileDWARF &dwarf, user_id_t uid,
926                    const DWARFDataExtractor &debug_info,
927                    DIERef::Section section, lldb::offset_t *offset_ptr) {
928   assert(debug_info.ValidOffset(*offset_ptr));
929 
930   auto expected_header = DWARFUnitHeader::extract(
931       debug_info, section, dwarf.GetDWARFContext(), offset_ptr);
932   if (!expected_header)
933     return expected_header.takeError();
934 
935   const DWARFDebugAbbrev *abbr = dwarf.DebugAbbrev();
936   if (!abbr)
937     return llvm::make_error<llvm::object::GenericBinaryError>(
938         "No debug_abbrev data");
939 
940   bool abbr_offset_OK =
941       dwarf.GetDWARFContext().getOrLoadAbbrevData().ValidOffset(
942           expected_header->GetAbbrOffset());
943   if (!abbr_offset_OK)
944     return llvm::make_error<llvm::object::GenericBinaryError>(
945         "Abbreviation offset for unit is not valid");
946 
947   const DWARFAbbreviationDeclarationSet *abbrevs =
948       abbr->GetAbbreviationDeclarationSet(expected_header->GetAbbrOffset());
949   if (!abbrevs)
950     return llvm::make_error<llvm::object::GenericBinaryError>(
951         "No abbrev exists at the specified offset.");
952 
953   bool is_dwo = dwarf.GetDWARFContext().isDwo();
954   if (expected_header->IsTypeUnit())
955     return DWARFUnitSP(new DWARFTypeUnit(dwarf, uid, *expected_header, *abbrevs,
956                                          section, is_dwo));
957   return DWARFUnitSP(new DWARFCompileUnit(dwarf, uid, *expected_header,
958                                           *abbrevs, section, is_dwo));
959 }
960 
961 const lldb_private::DWARFDataExtractor &DWARFUnit::GetData() const {
962   return m_section == DIERef::Section::DebugTypes
963              ? m_dwarf.GetDWARFContext().getOrLoadDebugTypesData()
964              : m_dwarf.GetDWARFContext().getOrLoadDebugInfoData();
965 }
966 
967 uint32_t DWARFUnit::GetHeaderByteSize() const {
968   switch (m_header.GetUnitType()) {
969   case llvm::dwarf::DW_UT_compile:
970   case llvm::dwarf::DW_UT_partial:
971     return GetVersion() < 5 ? 11 : 12;
972   case llvm::dwarf::DW_UT_skeleton:
973   case llvm::dwarf::DW_UT_split_compile:
974     return 20;
975   case llvm::dwarf::DW_UT_type:
976   case llvm::dwarf::DW_UT_split_type:
977     return GetVersion() < 5 ? 23 : 24;
978   }
979   llvm_unreachable("invalid UnitType.");
980 }
981 
982 llvm::Optional<uint64_t>
983 DWARFUnit::GetStringOffsetSectionItem(uint32_t index) const {
984   offset_t offset = GetStrOffsetsBase() + index * 4;
985   return m_dwarf.GetDWARFContext().getOrLoadStrOffsetsData().GetU32(&offset);
986 }
987 
988 llvm::Expected<DWARFRangeList>
989 DWARFUnit::FindRnglistFromOffset(dw_offset_t offset) {
990   if (GetVersion() <= 4) {
991     const DWARFDebugRanges *debug_ranges = m_dwarf.GetDebugRanges();
992     if (!debug_ranges)
993       return llvm::make_error<llvm::object::GenericBinaryError>(
994           "No debug_ranges section");
995     DWARFRangeList ranges;
996     debug_ranges->FindRanges(this, offset, ranges);
997     return ranges;
998   }
999 
1000   if (!GetRnglistTable())
1001     return llvm::createStringError(errc::invalid_argument,
1002                                    "missing or invalid range list table");
1003 
1004   llvm::DWARFDataExtractor data =
1005       m_dwarf.GetDWARFContext().getOrLoadRngListsData().GetAsLLVM();
1006 
1007   // As DW_AT_rnglists_base may be missing we need to call setAddressSize.
1008   data.setAddressSize(m_header.GetAddressByteSize());
1009   auto range_list_or_error = GetRnglistTable()->findList(data, offset);
1010   if (!range_list_or_error)
1011     return range_list_or_error.takeError();
1012 
1013   llvm::Expected<llvm::DWARFAddressRangesVector> llvm_ranges =
1014       range_list_or_error->getAbsoluteRanges(
1015           llvm::object::SectionedAddress{GetBaseAddress()},
1016           GetAddressByteSize(), [&](uint32_t index) {
1017             uint32_t index_size = GetAddressByteSize();
1018             dw_offset_t addr_base = GetAddrBase();
1019             lldb::offset_t offset = addr_base + index * index_size;
1020             return llvm::object::SectionedAddress{
1021                 m_dwarf.GetDWARFContext().getOrLoadAddrData().GetMaxU64(
1022                     &offset, index_size)};
1023           });
1024   if (!llvm_ranges)
1025     return llvm_ranges.takeError();
1026 
1027   DWARFRangeList ranges;
1028   for (const llvm::DWARFAddressRange &llvm_range : *llvm_ranges) {
1029     ranges.Append(DWARFRangeList::Entry(llvm_range.LowPC,
1030                                         llvm_range.HighPC - llvm_range.LowPC));
1031   }
1032   return ranges;
1033 }
1034 
1035 llvm::Expected<DWARFRangeList>
1036 DWARFUnit::FindRnglistFromIndex(uint32_t index) {
1037   llvm::Expected<uint64_t> maybe_offset = GetRnglistOffset(index);
1038   if (!maybe_offset)
1039     return maybe_offset.takeError();
1040   return FindRnglistFromOffset(*maybe_offset);
1041 }
1042