1e94042caSEugene Zelenko //===- DWARFAcceleratorTable.cpp ------------------------------------------===//
282af9438SZachary Turner //
382af9438SZachary Turner //                     The LLVM Compiler Infrastructure
482af9438SZachary Turner //
582af9438SZachary Turner // This file is distributed under the University of Illinois Open Source
682af9438SZachary Turner // License. See LICENSE.TXT for details.
782af9438SZachary Turner //
882af9438SZachary Turner //===----------------------------------------------------------------------===//
982af9438SZachary Turner 
1082af9438SZachary Turner #include "llvm/DebugInfo/DWARF/DWARFAcceleratorTable.h"
11264b5d9eSZachary Turner 
126bda14b3SChandler Carruth #include "llvm/ADT/SmallVector.h"
13264b5d9eSZachary Turner #include "llvm/BinaryFormat/Dwarf.h"
14e94042caSEugene Zelenko #include "llvm/DebugInfo/DWARF/DWARFRelocMap.h"
15e94042caSEugene Zelenko #include "llvm/Support/Compiler.h"
1692ac9d3eSJonas Devlieghere #include "llvm/Support/DJB.h"
1782af9438SZachary Turner #include "llvm/Support/Format.h"
189025f955SPavel Labath #include "llvm/Support/FormatVariadic.h"
193c9a918cSPavel Labath #include "llvm/Support/ScopedPrinter.h"
2082af9438SZachary Turner #include "llvm/Support/raw_ostream.h"
21e94042caSEugene Zelenko #include <cstddef>
22e94042caSEugene Zelenko #include <cstdint>
23e94042caSEugene Zelenko #include <utility>
2482af9438SZachary Turner 
25e94042caSEugene Zelenko using namespace llvm;
2682af9438SZachary Turner 
273c9a918cSPavel Labath namespace {
289025f955SPavel Labath struct Atom {
293c9a918cSPavel Labath   unsigned Value;
303c9a918cSPavel Labath };
313c9a918cSPavel Labath 
329025f955SPavel Labath static raw_ostream &operator<<(raw_ostream &OS, const Atom &A) {
339025f955SPavel Labath   StringRef Str = dwarf::AtomTypeString(A.Value);
343c9a918cSPavel Labath   if (!Str.empty())
353c9a918cSPavel Labath     return OS << Str;
369025f955SPavel Labath   return OS << "DW_ATOM_unknown_" << format("%x", A.Value);
373c9a918cSPavel Labath }
383c9a918cSPavel Labath } // namespace
393c9a918cSPavel Labath 
409025f955SPavel Labath static Atom formatAtom(unsigned Atom) { return {Atom}; }
413c9a918cSPavel Labath 
423c9a918cSPavel Labath DWARFAcceleratorTable::~DWARFAcceleratorTable() = default;
433c9a918cSPavel Labath 
449b36fd25SPavel Labath llvm::Error AppleAcceleratorTable::extract() {
4582af9438SZachary Turner   uint32_t Offset = 0;
4682af9438SZachary Turner 
4782af9438SZachary Turner   // Check that we can at least read the header.
4882af9438SZachary Turner   if (!AccelSection.isValidOffset(offsetof(Header, HeaderDataLength)+4))
49ba915897SJonas Devlieghere     return make_error<StringError>("Section too small: cannot read header.",
50ba915897SJonas Devlieghere                                    inconvertibleErrorCode());
5182af9438SZachary Turner 
5282af9438SZachary Turner   Hdr.Magic = AccelSection.getU32(&Offset);
5382af9438SZachary Turner   Hdr.Version = AccelSection.getU16(&Offset);
5482af9438SZachary Turner   Hdr.HashFunction = AccelSection.getU16(&Offset);
55394e8056SPavel Labath   Hdr.BucketCount = AccelSection.getU32(&Offset);
56394e8056SPavel Labath   Hdr.HashCount = AccelSection.getU32(&Offset);
5782af9438SZachary Turner   Hdr.HeaderDataLength = AccelSection.getU32(&Offset);
5882af9438SZachary Turner 
5982af9438SZachary Turner   // Check that we can read all the hashes and offsets from the
6082af9438SZachary Turner   // section (see SourceLevelDebugging.rst for the structure of the index).
61ba915897SJonas Devlieghere   // We need to substract one because we're checking for an *offset* which is
62ba915897SJonas Devlieghere   // equal to the size for an empty table and hence pointer after the section.
6382af9438SZachary Turner   if (!AccelSection.isValidOffset(sizeof(Hdr) + Hdr.HeaderDataLength +
64394e8056SPavel Labath                                   Hdr.BucketCount * 4 + Hdr.HashCount * 8 - 1))
65ba915897SJonas Devlieghere     return make_error<StringError>(
66ba915897SJonas Devlieghere         "Section too small: cannot read buckets and hashes.",
67ba915897SJonas Devlieghere         inconvertibleErrorCode());
6882af9438SZachary Turner 
6982af9438SZachary Turner   HdrData.DIEOffsetBase = AccelSection.getU32(&Offset);
7082af9438SZachary Turner   uint32_t NumAtoms = AccelSection.getU32(&Offset);
7182af9438SZachary Turner 
7282af9438SZachary Turner   for (unsigned i = 0; i < NumAtoms; ++i) {
7382af9438SZachary Turner     uint16_t AtomType = AccelSection.getU16(&Offset);
746c273763SGreg Clayton     auto AtomForm = static_cast<dwarf::Form>(AccelSection.getU16(&Offset));
7582af9438SZachary Turner     HdrData.Atoms.push_back(std::make_pair(AtomType, AtomForm));
7682af9438SZachary Turner   }
7782af9438SZachary Turner 
7899fdb9d9SAdrian Prantl   IsValid = true;
79ba915897SJonas Devlieghere   return Error::success();
8082af9438SZachary Turner }
8182af9438SZachary Turner 
82394e8056SPavel Labath uint32_t AppleAcceleratorTable::getNumBuckets() { return Hdr.BucketCount; }
83394e8056SPavel Labath uint32_t AppleAcceleratorTable::getNumHashes() { return Hdr.HashCount; }
849b36fd25SPavel Labath uint32_t AppleAcceleratorTable::getSizeHdr() { return sizeof(Hdr); }
859b36fd25SPavel Labath uint32_t AppleAcceleratorTable::getHeaderDataLength() {
86e41823bbSSpyridoula Gravani   return Hdr.HeaderDataLength;
87e41823bbSSpyridoula Gravani }
88e41823bbSSpyridoula Gravani 
899b36fd25SPavel Labath ArrayRef<std::pair<AppleAcceleratorTable::HeaderData::AtomType,
909b36fd25SPavel Labath                    AppleAcceleratorTable::HeaderData::Form>>
919b36fd25SPavel Labath AppleAcceleratorTable::getAtomsDesc() {
92837c110cSSpyridoula Gravani   return HdrData.Atoms;
93837c110cSSpyridoula Gravani }
94837c110cSSpyridoula Gravani 
959b36fd25SPavel Labath bool AppleAcceleratorTable::validateForms() {
96837c110cSSpyridoula Gravani   for (auto Atom : getAtomsDesc()) {
97837c110cSSpyridoula Gravani     DWARFFormValue FormValue(Atom.second);
98837c110cSSpyridoula Gravani     switch (Atom.first) {
99837c110cSSpyridoula Gravani     case dwarf::DW_ATOM_die_offset:
10070d35e10SSpyridoula Gravani     case dwarf::DW_ATOM_die_tag:
10170d35e10SSpyridoula Gravani     case dwarf::DW_ATOM_type_flags:
102837c110cSSpyridoula Gravani       if ((!FormValue.isFormClass(DWARFFormValue::FC_Constant) &&
103837c110cSSpyridoula Gravani            !FormValue.isFormClass(DWARFFormValue::FC_Flag)) ||
104837c110cSSpyridoula Gravani           FormValue.getForm() == dwarf::DW_FORM_sdata)
105837c110cSSpyridoula Gravani         return false;
1060e6694d1SAdrian Prantl       break;
107837c110cSSpyridoula Gravani     default:
108837c110cSSpyridoula Gravani       break;
109837c110cSSpyridoula Gravani     }
110837c110cSSpyridoula Gravani   }
111837c110cSSpyridoula Gravani   return true;
112837c110cSSpyridoula Gravani }
113837c110cSSpyridoula Gravani 
11470d35e10SSpyridoula Gravani std::pair<uint32_t, dwarf::Tag>
1159b36fd25SPavel Labath AppleAcceleratorTable::readAtoms(uint32_t &HashDataOffset) {
116837c110cSSpyridoula Gravani   uint32_t DieOffset = dwarf::DW_INVALID_OFFSET;
11770d35e10SSpyridoula Gravani   dwarf::Tag DieTag = dwarf::DW_TAG_null;
118322711f5SPavel Labath   dwarf::FormParams FormParams = {Hdr.Version, 0, dwarf::DwarfFormat::DWARF32};
119837c110cSSpyridoula Gravani 
120837c110cSSpyridoula Gravani   for (auto Atom : getAtomsDesc()) {
121837c110cSSpyridoula Gravani     DWARFFormValue FormValue(Atom.second);
122e5400f8aSPaul Robinson     FormValue.extractValue(AccelSection, &HashDataOffset, FormParams);
123837c110cSSpyridoula Gravani     switch (Atom.first) {
124837c110cSSpyridoula Gravani     case dwarf::DW_ATOM_die_offset:
125837c110cSSpyridoula Gravani       DieOffset = *FormValue.getAsUnsignedConstant();
126837c110cSSpyridoula Gravani       break;
12770d35e10SSpyridoula Gravani     case dwarf::DW_ATOM_die_tag:
12870d35e10SSpyridoula Gravani       DieTag = (dwarf::Tag)*FormValue.getAsUnsignedConstant();
12970d35e10SSpyridoula Gravani       break;
130837c110cSSpyridoula Gravani     default:
131837c110cSSpyridoula Gravani       break;
132837c110cSSpyridoula Gravani     }
133837c110cSSpyridoula Gravani   }
13470d35e10SSpyridoula Gravani   return {DieOffset, DieTag};
135837c110cSSpyridoula Gravani }
136837c110cSSpyridoula Gravani 
137394e8056SPavel Labath void AppleAcceleratorTable::Header::dump(ScopedPrinter &W) const {
138394e8056SPavel Labath   DictScope HeaderScope(W, "Header");
139394e8056SPavel Labath   W.printHex("Magic", Magic);
140394e8056SPavel Labath   W.printHex("Version", Version);
141394e8056SPavel Labath   W.printHex("Hash function", HashFunction);
142394e8056SPavel Labath   W.printNumber("Bucket count", BucketCount);
143394e8056SPavel Labath   W.printNumber("Hashes count", HashCount);
144394e8056SPavel Labath   W.printNumber("HeaderData length", HeaderDataLength);
145394e8056SPavel Labath }
146394e8056SPavel Labath 
14747c3472cSPavel Labath Optional<uint64_t> AppleAcceleratorTable::HeaderData::extractOffset(
14847c3472cSPavel Labath     Optional<DWARFFormValue> Value) const {
14947c3472cSPavel Labath   if (!Value)
15047c3472cSPavel Labath     return None;
15147c3472cSPavel Labath 
15247c3472cSPavel Labath   switch (Value->getForm()) {
15347c3472cSPavel Labath   case dwarf::DW_FORM_ref1:
15447c3472cSPavel Labath   case dwarf::DW_FORM_ref2:
15547c3472cSPavel Labath   case dwarf::DW_FORM_ref4:
15647c3472cSPavel Labath   case dwarf::DW_FORM_ref8:
15747c3472cSPavel Labath   case dwarf::DW_FORM_ref_udata:
15847c3472cSPavel Labath     return Value->getRawUValue() + DIEOffsetBase;
15947c3472cSPavel Labath   default:
16047c3472cSPavel Labath     return Value->getAsSectionOffset();
16147c3472cSPavel Labath   }
16247c3472cSPavel Labath }
16347c3472cSPavel Labath 
164394e8056SPavel Labath bool AppleAcceleratorTable::dumpName(ScopedPrinter &W,
165394e8056SPavel Labath                                      SmallVectorImpl<DWARFFormValue> &AtomForms,
166394e8056SPavel Labath                                      uint32_t *DataOffset) const {
167322711f5SPavel Labath   dwarf::FormParams FormParams = {Hdr.Version, 0, dwarf::DwarfFormat::DWARF32};
168394e8056SPavel Labath   uint32_t NameOffset = *DataOffset;
169394e8056SPavel Labath   if (!AccelSection.isValidOffsetForDataOfSize(*DataOffset, 4)) {
170394e8056SPavel Labath     W.printString("Incorrectly terminated list.");
171394e8056SPavel Labath     return false;
172394e8056SPavel Labath   }
173394e8056SPavel Labath   unsigned StringOffset = AccelSection.getRelocatedValue(4, DataOffset);
174394e8056SPavel Labath   if (!StringOffset)
175394e8056SPavel Labath     return false; // End of list
176394e8056SPavel Labath 
177394e8056SPavel Labath   DictScope NameScope(W, ("Name@0x" + Twine::utohexstr(NameOffset)).str());
178394e8056SPavel Labath   W.startLine() << format("String: 0x%08x", StringOffset);
179394e8056SPavel Labath   W.getOStream() << " \"" << StringSection.getCStr(&StringOffset) << "\"\n";
180394e8056SPavel Labath 
181394e8056SPavel Labath   unsigned NumData = AccelSection.getU32(DataOffset);
182394e8056SPavel Labath   for (unsigned Data = 0; Data < NumData; ++Data) {
183394e8056SPavel Labath     ListScope DataScope(W, ("Data " + Twine(Data)).str());
184394e8056SPavel Labath     unsigned i = 0;
185394e8056SPavel Labath     for (auto &Atom : AtomForms) {
186394e8056SPavel Labath       W.startLine() << format("Atom[%d]: ", i++);
187394e8056SPavel Labath       if (Atom.extractValue(AccelSection, DataOffset, FormParams))
188394e8056SPavel Labath         Atom.dump(W.getOStream());
189394e8056SPavel Labath       else
190394e8056SPavel Labath         W.getOStream() << "Error extracting the value";
191394e8056SPavel Labath       W.getOStream() << "\n";
192394e8056SPavel Labath     }
193394e8056SPavel Labath   }
194394e8056SPavel Labath   return true; // more entries follow
195394e8056SPavel Labath }
196394e8056SPavel Labath 
1979b36fd25SPavel Labath LLVM_DUMP_METHOD void AppleAcceleratorTable::dump(raw_ostream &OS) const {
19899fdb9d9SAdrian Prantl   if (!IsValid)
19999fdb9d9SAdrian Prantl     return;
20099fdb9d9SAdrian Prantl 
201394e8056SPavel Labath   ScopedPrinter W(OS);
20282af9438SZachary Turner 
203394e8056SPavel Labath   Hdr.dump(W);
204394e8056SPavel Labath 
205394e8056SPavel Labath   W.printNumber("DIE offset base", HdrData.DIEOffsetBase);
2063460957eSPavel Labath   W.printNumber("Number of atoms", uint64_t(HdrData.Atoms.size()));
20782af9438SZachary Turner   SmallVector<DWARFFormValue, 3> AtomForms;
208394e8056SPavel Labath   {
209394e8056SPavel Labath     ListScope AtomsScope(W, "Atoms");
210394e8056SPavel Labath     unsigned i = 0;
21182af9438SZachary Turner     for (const auto &Atom : HdrData.Atoms) {
212394e8056SPavel Labath       DictScope AtomScope(W, ("Atom " + Twine(i++)).str());
213394e8056SPavel Labath       W.startLine() << "Type: " << formatAtom(Atom.first) << '\n';
2149025f955SPavel Labath       W.startLine() << "Form: " << formatv("{0}", Atom.second) << '\n';
21582af9438SZachary Turner       AtomForms.push_back(DWARFFormValue(Atom.second));
21682af9438SZachary Turner     }
217394e8056SPavel Labath   }
21882af9438SZachary Turner 
21982af9438SZachary Turner   // Now go through the actual tables and dump them.
22082af9438SZachary Turner   uint32_t Offset = sizeof(Hdr) + Hdr.HeaderDataLength;
221394e8056SPavel Labath   unsigned HashesBase = Offset + Hdr.BucketCount * 4;
222394e8056SPavel Labath   unsigned OffsetsBase = HashesBase + Hdr.HashCount * 4;
22382af9438SZachary Turner 
224394e8056SPavel Labath   for (unsigned Bucket = 0; Bucket < Hdr.BucketCount; ++Bucket) {
22582af9438SZachary Turner     unsigned Index = AccelSection.getU32(&Offset);
22682af9438SZachary Turner 
227394e8056SPavel Labath     ListScope BucketScope(W, ("Bucket " + Twine(Bucket)).str());
22882af9438SZachary Turner     if (Index == UINT32_MAX) {
229394e8056SPavel Labath       W.printString("EMPTY");
23082af9438SZachary Turner       continue;
23182af9438SZachary Turner     }
23282af9438SZachary Turner 
233394e8056SPavel Labath     for (unsigned HashIdx = Index; HashIdx < Hdr.HashCount; ++HashIdx) {
23482af9438SZachary Turner       unsigned HashOffset = HashesBase + HashIdx*4;
23582af9438SZachary Turner       unsigned OffsetsOffset = OffsetsBase + HashIdx*4;
23682af9438SZachary Turner       uint32_t Hash = AccelSection.getU32(&HashOffset);
23782af9438SZachary Turner 
238394e8056SPavel Labath       if (Hash % Hdr.BucketCount != Bucket)
23982af9438SZachary Turner         break;
24082af9438SZachary Turner 
24182af9438SZachary Turner       unsigned DataOffset = AccelSection.getU32(&OffsetsOffset);
242394e8056SPavel Labath       ListScope HashScope(W, ("Hash 0x" + Twine::utohexstr(Hash)).str());
24382af9438SZachary Turner       if (!AccelSection.isValidOffset(DataOffset)) {
244394e8056SPavel Labath         W.printString("Invalid section offset");
24582af9438SZachary Turner         continue;
24682af9438SZachary Turner       }
247394e8056SPavel Labath       while (dumpName(W, AtomForms, &DataOffset))
248394e8056SPavel Labath         /*empty*/;
24982af9438SZachary Turner     }
25082af9438SZachary Turner   }
25182af9438SZachary Turner }
25299fdb9d9SAdrian Prantl 
253d99072bcSPavel Labath AppleAcceleratorTable::Entry::Entry(
254d99072bcSPavel Labath     const AppleAcceleratorTable::HeaderData &HdrData)
255d99072bcSPavel Labath     : HdrData(&HdrData) {
256d99072bcSPavel Labath   Values.reserve(HdrData.Atoms.size());
257d99072bcSPavel Labath   for (const auto &Atom : HdrData.Atoms)
258d99072bcSPavel Labath     Values.push_back(DWARFFormValue(Atom.second));
259d99072bcSPavel Labath }
260d99072bcSPavel Labath 
261d99072bcSPavel Labath void AppleAcceleratorTable::Entry::extract(
262d99072bcSPavel Labath     const AppleAcceleratorTable &AccelTable, uint32_t *Offset) {
263d99072bcSPavel Labath 
264322711f5SPavel Labath   dwarf::FormParams FormParams = {AccelTable.Hdr.Version, 0,
265d99072bcSPavel Labath                                   dwarf::DwarfFormat::DWARF32};
266d99072bcSPavel Labath   for (auto &Atom : Values)
267d99072bcSPavel Labath     Atom.extractValue(AccelTable.AccelSection, Offset, FormParams);
268d99072bcSPavel Labath }
269d99072bcSPavel Labath 
270d99072bcSPavel Labath Optional<DWARFFormValue>
271d99072bcSPavel Labath AppleAcceleratorTable::Entry::lookup(HeaderData::AtomType Atom) const {
272d99072bcSPavel Labath   assert(HdrData && "Dereferencing end iterator?");
273d99072bcSPavel Labath   assert(HdrData->Atoms.size() == Values.size());
274d99072bcSPavel Labath   for (const auto &Tuple : zip_first(HdrData->Atoms, Values)) {
275d99072bcSPavel Labath     if (std::get<0>(Tuple).first == Atom)
276d99072bcSPavel Labath       return std::get<1>(Tuple);
277d99072bcSPavel Labath   }
278d99072bcSPavel Labath   return None;
279d99072bcSPavel Labath }
280d99072bcSPavel Labath 
28147c3472cSPavel Labath Optional<uint64_t> AppleAcceleratorTable::Entry::getDIESectionOffset() const {
28247c3472cSPavel Labath   return HdrData->extractOffset(lookup(dwarf::DW_ATOM_die_offset));
283d99072bcSPavel Labath }
284d99072bcSPavel Labath 
285d99072bcSPavel Labath Optional<uint64_t> AppleAcceleratorTable::Entry::getCUOffset() const {
28647c3472cSPavel Labath   return HdrData->extractOffset(lookup(dwarf::DW_ATOM_cu_offset));
287d99072bcSPavel Labath }
288d99072bcSPavel Labath 
289d99072bcSPavel Labath Optional<dwarf::Tag> AppleAcceleratorTable::Entry::getTag() const {
290d99072bcSPavel Labath   Optional<DWARFFormValue> Tag = lookup(dwarf::DW_ATOM_die_tag);
291d99072bcSPavel Labath   if (!Tag)
292d99072bcSPavel Labath     return None;
293d99072bcSPavel Labath   if (Optional<uint64_t> Value = Tag->getAsUnsignedConstant())
294d99072bcSPavel Labath     return dwarf::Tag(*Value);
295d99072bcSPavel Labath   return None;
296d99072bcSPavel Labath }
297d99072bcSPavel Labath 
2989b36fd25SPavel Labath AppleAcceleratorTable::ValueIterator::ValueIterator(
2999b36fd25SPavel Labath     const AppleAcceleratorTable &AccelTable, unsigned Offset)
300d99072bcSPavel Labath     : AccelTable(&AccelTable), Current(AccelTable.HdrData), DataOffset(Offset) {
30199fdb9d9SAdrian Prantl   if (!AccelTable.AccelSection.isValidOffsetForDataOfSize(DataOffset, 4))
30299fdb9d9SAdrian Prantl     return;
30399fdb9d9SAdrian Prantl 
30499fdb9d9SAdrian Prantl   // Read the first entry.
30599fdb9d9SAdrian Prantl   NumData = AccelTable.AccelSection.getU32(&DataOffset);
30699fdb9d9SAdrian Prantl   Next();
30799fdb9d9SAdrian Prantl }
30899fdb9d9SAdrian Prantl 
3099b36fd25SPavel Labath void AppleAcceleratorTable::ValueIterator::Next() {
31099fdb9d9SAdrian Prantl   assert(NumData > 0 && "attempted to increment iterator past the end");
31199fdb9d9SAdrian Prantl   auto &AccelSection = AccelTable->AccelSection;
31299fdb9d9SAdrian Prantl   if (Data >= NumData ||
31399fdb9d9SAdrian Prantl       !AccelSection.isValidOffsetForDataOfSize(DataOffset, 4)) {
31499fdb9d9SAdrian Prantl     NumData = 0;
315*59870af6SPavel Labath     DataOffset = 0;
31699fdb9d9SAdrian Prantl     return;
31799fdb9d9SAdrian Prantl   }
318d99072bcSPavel Labath   Current.extract(*AccelTable, &DataOffset);
31999fdb9d9SAdrian Prantl   ++Data;
32099fdb9d9SAdrian Prantl }
32199fdb9d9SAdrian Prantl 
3229b36fd25SPavel Labath iterator_range<AppleAcceleratorTable::ValueIterator>
3239b36fd25SPavel Labath AppleAcceleratorTable::equal_range(StringRef Key) const {
32499fdb9d9SAdrian Prantl   if (!IsValid)
32599fdb9d9SAdrian Prantl     return make_range(ValueIterator(), ValueIterator());
32699fdb9d9SAdrian Prantl 
32799fdb9d9SAdrian Prantl   // Find the bucket.
32892ac9d3eSJonas Devlieghere   unsigned HashValue = djbHash(Key);
329394e8056SPavel Labath   unsigned Bucket = HashValue % Hdr.BucketCount;
33099fdb9d9SAdrian Prantl   unsigned BucketBase = sizeof(Hdr) + Hdr.HeaderDataLength;
331394e8056SPavel Labath   unsigned HashesBase = BucketBase + Hdr.BucketCount * 4;
332394e8056SPavel Labath   unsigned OffsetsBase = HashesBase + Hdr.HashCount * 4;
33399fdb9d9SAdrian Prantl 
33499fdb9d9SAdrian Prantl   unsigned BucketOffset = BucketBase + Bucket * 4;
33599fdb9d9SAdrian Prantl   unsigned Index = AccelSection.getU32(&BucketOffset);
33699fdb9d9SAdrian Prantl 
33799fdb9d9SAdrian Prantl   // Search through all hashes in the bucket.
338394e8056SPavel Labath   for (unsigned HashIdx = Index; HashIdx < Hdr.HashCount; ++HashIdx) {
33999fdb9d9SAdrian Prantl     unsigned HashOffset = HashesBase + HashIdx * 4;
34099fdb9d9SAdrian Prantl     unsigned OffsetsOffset = OffsetsBase + HashIdx * 4;
34199fdb9d9SAdrian Prantl     uint32_t Hash = AccelSection.getU32(&HashOffset);
34299fdb9d9SAdrian Prantl 
343394e8056SPavel Labath     if (Hash % Hdr.BucketCount != Bucket)
34499fdb9d9SAdrian Prantl       // We are already in the next bucket.
34599fdb9d9SAdrian Prantl       break;
34699fdb9d9SAdrian Prantl 
34799fdb9d9SAdrian Prantl     unsigned DataOffset = AccelSection.getU32(&OffsetsOffset);
34899fdb9d9SAdrian Prantl     unsigned StringOffset = AccelSection.getRelocatedValue(4, &DataOffset);
34999fdb9d9SAdrian Prantl     if (!StringOffset)
35099fdb9d9SAdrian Prantl       break;
35199fdb9d9SAdrian Prantl 
35299fdb9d9SAdrian Prantl     // Finally, compare the key.
35399fdb9d9SAdrian Prantl     if (Key == StringSection.getCStr(&StringOffset))
35499fdb9d9SAdrian Prantl       return make_range({*this, DataOffset}, ValueIterator());
35599fdb9d9SAdrian Prantl   }
35699fdb9d9SAdrian Prantl   return make_range(ValueIterator(), ValueIterator());
35799fdb9d9SAdrian Prantl }
3583c9a918cSPavel Labath 
3593c9a918cSPavel Labath void DWARFDebugNames::Header::dump(ScopedPrinter &W) const {
3603c9a918cSPavel Labath   DictScope HeaderScope(W, "Header");
3613c9a918cSPavel Labath   W.printHex("Length", UnitLength);
3623c9a918cSPavel Labath   W.printNumber("Version", Version);
3633c9a918cSPavel Labath   W.printHex("Padding", Padding);
3643c9a918cSPavel Labath   W.printNumber("CU count", CompUnitCount);
3653c9a918cSPavel Labath   W.printNumber("Local TU count", LocalTypeUnitCount);
3663c9a918cSPavel Labath   W.printNumber("Foreign TU count", ForeignTypeUnitCount);
3673c9a918cSPavel Labath   W.printNumber("Bucket count", BucketCount);
3683c9a918cSPavel Labath   W.printNumber("Name count", NameCount);
3693c9a918cSPavel Labath   W.printHex("Abbreviations table size", AbbrevTableSize);
3703c9a918cSPavel Labath   W.startLine() << "Augmentation: '" << AugmentationString << "'\n";
3713c9a918cSPavel Labath }
3723c9a918cSPavel Labath 
3733c9a918cSPavel Labath llvm::Error DWARFDebugNames::Header::extract(const DWARFDataExtractor &AS,
3743c9a918cSPavel Labath                                              uint32_t *Offset) {
3753c9a918cSPavel Labath   // Check that we can read the fixed-size part.
376e7264106SPavel Labath   if (!AS.isValidOffset(*Offset + sizeof(HeaderPOD) - 1))
3773c9a918cSPavel Labath     return make_error<StringError>("Section too small: cannot read header.",
3783c9a918cSPavel Labath                                    inconvertibleErrorCode());
3793c9a918cSPavel Labath 
3803c9a918cSPavel Labath   UnitLength = AS.getU32(Offset);
3813c9a918cSPavel Labath   Version = AS.getU16(Offset);
3823c9a918cSPavel Labath   Padding = AS.getU16(Offset);
3833c9a918cSPavel Labath   CompUnitCount = AS.getU32(Offset);
3843c9a918cSPavel Labath   LocalTypeUnitCount = AS.getU32(Offset);
3853c9a918cSPavel Labath   ForeignTypeUnitCount = AS.getU32(Offset);
3863c9a918cSPavel Labath   BucketCount = AS.getU32(Offset);
3873c9a918cSPavel Labath   NameCount = AS.getU32(Offset);
3883c9a918cSPavel Labath   AbbrevTableSize = AS.getU32(Offset);
389ea0f841cSPavel Labath   AugmentationStringSize = alignTo(AS.getU32(Offset), 4);
3903c9a918cSPavel Labath 
3913c9a918cSPavel Labath   if (!AS.isValidOffsetForDataOfSize(*Offset, AugmentationStringSize))
3923c9a918cSPavel Labath     return make_error<StringError>(
3933c9a918cSPavel Labath         "Section too small: cannot read header augmentation.",
3943c9a918cSPavel Labath         inconvertibleErrorCode());
3953c9a918cSPavel Labath   AugmentationString.resize(AugmentationStringSize);
3963c9a918cSPavel Labath   AS.getU8(Offset, reinterpret_cast<uint8_t *>(AugmentationString.data()),
3973c9a918cSPavel Labath            AugmentationStringSize);
3983c9a918cSPavel Labath   return Error::success();
3993c9a918cSPavel Labath }
4003c9a918cSPavel Labath 
4013c9a918cSPavel Labath void DWARFDebugNames::Abbrev::dump(ScopedPrinter &W) const {
4023c9a918cSPavel Labath   DictScope AbbrevScope(W, ("Abbreviation 0x" + Twine::utohexstr(Code)).str());
4039025f955SPavel Labath   W.startLine() << formatv("Tag: {0}\n", Tag);
4043c9a918cSPavel Labath 
4059025f955SPavel Labath   for (const auto &Attr : Attributes)
4069025f955SPavel Labath     W.startLine() << formatv("{0}: {1}\n", Attr.Index, Attr.Form);
4073c9a918cSPavel Labath }
4083c9a918cSPavel Labath 
4093c9a918cSPavel Labath static constexpr DWARFDebugNames::AttributeEncoding sentinelAttrEnc() {
4103c9a918cSPavel Labath   return {dwarf::Index(0), dwarf::Form(0)};
4113c9a918cSPavel Labath }
4123c9a918cSPavel Labath 
4133c9a918cSPavel Labath static bool isSentinel(const DWARFDebugNames::AttributeEncoding &AE) {
4143c9a918cSPavel Labath   return AE == sentinelAttrEnc();
4153c9a918cSPavel Labath }
4163c9a918cSPavel Labath 
4173c9a918cSPavel Labath static DWARFDebugNames::Abbrev sentinelAbbrev() {
4183c9a918cSPavel Labath   return DWARFDebugNames::Abbrev(0, dwarf::Tag(0), {});
4193c9a918cSPavel Labath }
4203c9a918cSPavel Labath 
4213c9a918cSPavel Labath static bool isSentinel(const DWARFDebugNames::Abbrev &Abbr) {
4223c9a918cSPavel Labath   return Abbr.Code == 0;
4233c9a918cSPavel Labath }
4243c9a918cSPavel Labath 
4253c9a918cSPavel Labath DWARFDebugNames::Abbrev DWARFDebugNames::AbbrevMapInfo::getEmptyKey() {
4263c9a918cSPavel Labath   return sentinelAbbrev();
4273c9a918cSPavel Labath }
4283c9a918cSPavel Labath 
4293c9a918cSPavel Labath DWARFDebugNames::Abbrev DWARFDebugNames::AbbrevMapInfo::getTombstoneKey() {
4303c9a918cSPavel Labath   return DWARFDebugNames::Abbrev(~0, dwarf::Tag(0), {});
4313c9a918cSPavel Labath }
4323c9a918cSPavel Labath 
4333c9a918cSPavel Labath Expected<DWARFDebugNames::AttributeEncoding>
4343c9a918cSPavel Labath DWARFDebugNames::NameIndex::extractAttributeEncoding(uint32_t *Offset) {
4353c9a918cSPavel Labath   if (*Offset >= EntriesBase) {
4363c9a918cSPavel Labath     return make_error<StringError>("Incorrectly terminated abbreviation table.",
4373c9a918cSPavel Labath                                    inconvertibleErrorCode());
4383c9a918cSPavel Labath   }
4393c9a918cSPavel Labath 
4403c9a918cSPavel Labath   uint32_t Index = Section.AccelSection.getULEB128(Offset);
4413c9a918cSPavel Labath   uint32_t Form = Section.AccelSection.getULEB128(Offset);
4423c9a918cSPavel Labath   return AttributeEncoding(dwarf::Index(Index), dwarf::Form(Form));
4433c9a918cSPavel Labath }
4443c9a918cSPavel Labath 
4453c9a918cSPavel Labath Expected<std::vector<DWARFDebugNames::AttributeEncoding>>
4463c9a918cSPavel Labath DWARFDebugNames::NameIndex::extractAttributeEncodings(uint32_t *Offset) {
4473c9a918cSPavel Labath   std::vector<AttributeEncoding> Result;
4483c9a918cSPavel Labath   for (;;) {
4493c9a918cSPavel Labath     auto AttrEncOr = extractAttributeEncoding(Offset);
4503c9a918cSPavel Labath     if (!AttrEncOr)
4513c9a918cSPavel Labath       return AttrEncOr.takeError();
4523c9a918cSPavel Labath     if (isSentinel(*AttrEncOr))
4533c9a918cSPavel Labath       return std::move(Result);
4543c9a918cSPavel Labath 
4553c9a918cSPavel Labath     Result.emplace_back(*AttrEncOr);
4563c9a918cSPavel Labath   }
4573c9a918cSPavel Labath }
4583c9a918cSPavel Labath 
4593c9a918cSPavel Labath Expected<DWARFDebugNames::Abbrev>
4603c9a918cSPavel Labath DWARFDebugNames::NameIndex::extractAbbrev(uint32_t *Offset) {
4613c9a918cSPavel Labath   if (*Offset >= EntriesBase) {
4623c9a918cSPavel Labath     return make_error<StringError>("Incorrectly terminated abbreviation table.",
4633c9a918cSPavel Labath                                    inconvertibleErrorCode());
4643c9a918cSPavel Labath   }
4653c9a918cSPavel Labath 
4663c9a918cSPavel Labath   uint32_t Code = Section.AccelSection.getULEB128(Offset);
4673c9a918cSPavel Labath   if (Code == 0)
4683c9a918cSPavel Labath     return sentinelAbbrev();
4693c9a918cSPavel Labath 
4703c9a918cSPavel Labath   uint32_t Tag = Section.AccelSection.getULEB128(Offset);
4713c9a918cSPavel Labath   auto AttrEncOr = extractAttributeEncodings(Offset);
4723c9a918cSPavel Labath   if (!AttrEncOr)
4733c9a918cSPavel Labath     return AttrEncOr.takeError();
4743c9a918cSPavel Labath   return Abbrev(Code, dwarf::Tag(Tag), std::move(*AttrEncOr));
4753c9a918cSPavel Labath }
4763c9a918cSPavel Labath 
4773c9a918cSPavel Labath Error DWARFDebugNames::NameIndex::extract() {
4783c9a918cSPavel Labath   const DWARFDataExtractor &AS = Section.AccelSection;
4793c9a918cSPavel Labath   uint32_t Offset = Base;
4803c9a918cSPavel Labath   if (Error E = Hdr.extract(AS, &Offset))
4813c9a918cSPavel Labath     return E;
4823c9a918cSPavel Labath 
4833c9a918cSPavel Labath   CUsBase = Offset;
4843c9a918cSPavel Labath   Offset += Hdr.CompUnitCount * 4;
4853c9a918cSPavel Labath   Offset += Hdr.LocalTypeUnitCount * 4;
4863c9a918cSPavel Labath   Offset += Hdr.ForeignTypeUnitCount * 8;
4873c9a918cSPavel Labath   BucketsBase = Offset;
4883c9a918cSPavel Labath   Offset += Hdr.BucketCount * 4;
4893c9a918cSPavel Labath   HashesBase = Offset;
4903c9a918cSPavel Labath   if (Hdr.BucketCount > 0)
4913c9a918cSPavel Labath     Offset += Hdr.NameCount * 4;
4923c9a918cSPavel Labath   StringOffsetsBase = Offset;
4933c9a918cSPavel Labath   Offset += Hdr.NameCount * 4;
4943c9a918cSPavel Labath   EntryOffsetsBase = Offset;
4953c9a918cSPavel Labath   Offset += Hdr.NameCount * 4;
4963c9a918cSPavel Labath 
4973c9a918cSPavel Labath   if (!AS.isValidOffsetForDataOfSize(Offset, Hdr.AbbrevTableSize))
4983c9a918cSPavel Labath     return make_error<StringError>(
4993c9a918cSPavel Labath         "Section too small: cannot read abbreviations.",
5003c9a918cSPavel Labath         inconvertibleErrorCode());
5013c9a918cSPavel Labath 
5023c9a918cSPavel Labath   EntriesBase = Offset + Hdr.AbbrevTableSize;
5033c9a918cSPavel Labath 
5043c9a918cSPavel Labath   for (;;) {
5053c9a918cSPavel Labath     auto AbbrevOr = extractAbbrev(&Offset);
5063c9a918cSPavel Labath     if (!AbbrevOr)
5073c9a918cSPavel Labath       return AbbrevOr.takeError();
5083c9a918cSPavel Labath     if (isSentinel(*AbbrevOr))
5093c9a918cSPavel Labath       return Error::success();
5103c9a918cSPavel Labath 
5113c9a918cSPavel Labath     if (!Abbrevs.insert(std::move(*AbbrevOr)).second) {
5123c9a918cSPavel Labath       return make_error<StringError>("Duplicate abbreviation code.",
5133c9a918cSPavel Labath                                      inconvertibleErrorCode());
5143c9a918cSPavel Labath     }
5153c9a918cSPavel Labath   }
5163c9a918cSPavel Labath }
517d99072bcSPavel Labath DWARFDebugNames::Entry::Entry(const NameIndex &NameIdx, const Abbrev &Abbr)
518d99072bcSPavel Labath     : NameIdx(&NameIdx), Abbr(&Abbr) {
5193c9a918cSPavel Labath   // This merely creates form values. It is up to the caller
5203c9a918cSPavel Labath   // (NameIndex::getEntry) to populate them.
5213c9a918cSPavel Labath   Values.reserve(Abbr.Attributes.size());
5223c9a918cSPavel Labath   for (const auto &Attr : Abbr.Attributes)
5233c9a918cSPavel Labath     Values.emplace_back(Attr.Form);
5243c9a918cSPavel Labath }
5253c9a918cSPavel Labath 
526d99072bcSPavel Labath Optional<DWARFFormValue>
527d99072bcSPavel Labath DWARFDebugNames::Entry::lookup(dwarf::Index Index) const {
528d99072bcSPavel Labath   assert(Abbr->Attributes.size() == Values.size());
529d99072bcSPavel Labath   for (const auto &Tuple : zip_first(Abbr->Attributes, Values)) {
530d99072bcSPavel Labath     if (std::get<0>(Tuple).Index == Index)
531d99072bcSPavel Labath       return std::get<1>(Tuple);
532d99072bcSPavel Labath   }
533d99072bcSPavel Labath   return None;
534d99072bcSPavel Labath }
5353c9a918cSPavel Labath 
53647c3472cSPavel Labath Optional<uint64_t> DWARFDebugNames::Entry::getDIEUnitOffset() const {
537d99072bcSPavel Labath   if (Optional<DWARFFormValue> Off = lookup(dwarf::DW_IDX_die_offset))
5382d1fc437SPavel Labath     return Off->getAsReferenceUVal();
539d99072bcSPavel Labath   return None;
540d99072bcSPavel Labath }
541d99072bcSPavel Labath 
542d99072bcSPavel Labath Optional<uint64_t> DWARFDebugNames::Entry::getCUIndex() const {
543d99072bcSPavel Labath   if (Optional<DWARFFormValue> Off = lookup(dwarf::DW_IDX_compile_unit))
544d99072bcSPavel Labath     return Off->getAsUnsignedConstant();
54547c3472cSPavel Labath   // In a per-CU index, the entries without a DW_IDX_compile_unit attribute
54647c3472cSPavel Labath   // implicitly refer to the single CU.
54747c3472cSPavel Labath   if (NameIdx->getCUCount() == 1)
54847c3472cSPavel Labath     return 0;
549d99072bcSPavel Labath   return None;
550d99072bcSPavel Labath }
551d99072bcSPavel Labath 
552d99072bcSPavel Labath Optional<uint64_t> DWARFDebugNames::Entry::getCUOffset() const {
553d99072bcSPavel Labath   Optional<uint64_t> Index = getCUIndex();
554d99072bcSPavel Labath   if (!Index || *Index >= NameIdx->getCUCount())
555d99072bcSPavel Labath     return None;
556d99072bcSPavel Labath   return NameIdx->getCUOffset(*Index);
557d99072bcSPavel Labath }
558d99072bcSPavel Labath 
55947c3472cSPavel Labath Optional<uint64_t> DWARFDebugNames::Entry::getDIESectionOffset() const {
56047c3472cSPavel Labath   Optional<uint64_t> CUOff = getCUOffset();
56147c3472cSPavel Labath   Optional<uint64_t> DIEOff = getDIEUnitOffset();
56247c3472cSPavel Labath   if (CUOff && DIEOff)
56347c3472cSPavel Labath     return *CUOff + *DIEOff;
56447c3472cSPavel Labath   return None;
56547c3472cSPavel Labath }
56647c3472cSPavel Labath 
567d99072bcSPavel Labath void DWARFDebugNames::Entry::dump(ScopedPrinter &W) const {
568d99072bcSPavel Labath   W.printHex("Abbrev", Abbr->Code);
5699025f955SPavel Labath   W.startLine() << formatv("Tag: {0}\n", Abbr->Tag);
570d99072bcSPavel Labath   assert(Abbr->Attributes.size() == Values.size());
571d99072bcSPavel Labath   for (const auto &Tuple : zip_first(Abbr->Attributes, Values)) {
5729025f955SPavel Labath     W.startLine() << formatv("{0}: ", std::get<0>(Tuple).Index);
573d99072bcSPavel Labath     std::get<1>(Tuple).dump(W.getOStream());
5743c9a918cSPavel Labath     W.getOStream() << '\n';
5753c9a918cSPavel Labath   }
5763c9a918cSPavel Labath }
5773c9a918cSPavel Labath 
5783c9a918cSPavel Labath char DWARFDebugNames::SentinelError::ID;
5793c9a918cSPavel Labath std::error_code DWARFDebugNames::SentinelError::convertToErrorCode() const {
5803c9a918cSPavel Labath   return inconvertibleErrorCode();
5813c9a918cSPavel Labath }
5823c9a918cSPavel Labath 
5833c9a918cSPavel Labath uint32_t DWARFDebugNames::NameIndex::getCUOffset(uint32_t CU) const {
5843c9a918cSPavel Labath   assert(CU < Hdr.CompUnitCount);
5853c9a918cSPavel Labath   uint32_t Offset = CUsBase + 4 * CU;
5863c9a918cSPavel Labath   return Section.AccelSection.getRelocatedValue(4, &Offset);
5873c9a918cSPavel Labath }
5883c9a918cSPavel Labath 
5893c9a918cSPavel Labath uint32_t DWARFDebugNames::NameIndex::getLocalTUOffset(uint32_t TU) const {
5903c9a918cSPavel Labath   assert(TU < Hdr.LocalTypeUnitCount);
5913c9a918cSPavel Labath   uint32_t Offset = CUsBase + Hdr.CompUnitCount * 4;
5923c9a918cSPavel Labath   return Section.AccelSection.getRelocatedValue(4, &Offset);
5933c9a918cSPavel Labath }
5943c9a918cSPavel Labath 
595d99072bcSPavel Labath uint64_t DWARFDebugNames::NameIndex::getForeignTUSignature(uint32_t TU) const {
5963c9a918cSPavel Labath   assert(TU < Hdr.ForeignTypeUnitCount);
5973c9a918cSPavel Labath   uint32_t Offset = CUsBase + (Hdr.CompUnitCount + Hdr.LocalTypeUnitCount) * 4;
5983c9a918cSPavel Labath   return Section.AccelSection.getU64(&Offset);
5993c9a918cSPavel Labath }
6003c9a918cSPavel Labath 
6013c9a918cSPavel Labath Expected<DWARFDebugNames::Entry>
6023c9a918cSPavel Labath DWARFDebugNames::NameIndex::getEntry(uint32_t *Offset) const {
6033c9a918cSPavel Labath   const DWARFDataExtractor &AS = Section.AccelSection;
6043c9a918cSPavel Labath   if (!AS.isValidOffset(*Offset))
605c9f07b06SPavel Labath     return make_error<StringError>("Incorrectly terminated entry list.",
6063c9a918cSPavel Labath                                    inconvertibleErrorCode());
6073c9a918cSPavel Labath 
6083c9a918cSPavel Labath   uint32_t AbbrevCode = AS.getULEB128(Offset);
6093c9a918cSPavel Labath   if (AbbrevCode == 0)
6103c9a918cSPavel Labath     return make_error<SentinelError>();
6113c9a918cSPavel Labath 
6123c9a918cSPavel Labath   const auto AbbrevIt = Abbrevs.find_as(AbbrevCode);
6133c9a918cSPavel Labath   if (AbbrevIt == Abbrevs.end())
614c9f07b06SPavel Labath     return make_error<StringError>("Invalid abbreviation.",
6153c9a918cSPavel Labath                                    inconvertibleErrorCode());
6163c9a918cSPavel Labath 
617d99072bcSPavel Labath   Entry E(*this, *AbbrevIt);
6183c9a918cSPavel Labath 
619322711f5SPavel Labath   dwarf::FormParams FormParams = {Hdr.Version, 0, dwarf::DwarfFormat::DWARF32};
6203c9a918cSPavel Labath   for (auto &Value : E.Values) {
6213c9a918cSPavel Labath     if (!Value.extractValue(AS, Offset, FormParams))
622c9f07b06SPavel Labath       return make_error<StringError>("Error extracting index attribute values.",
6233c9a918cSPavel Labath                                      inconvertibleErrorCode());
6243c9a918cSPavel Labath   }
6253c9a918cSPavel Labath   return std::move(E);
6263c9a918cSPavel Labath }
6273c9a918cSPavel Labath 
6283c9a918cSPavel Labath DWARFDebugNames::NameTableEntry
6293c9a918cSPavel Labath DWARFDebugNames::NameIndex::getNameTableEntry(uint32_t Index) const {
6303c9a918cSPavel Labath   assert(0 < Index && Index <= Hdr.NameCount);
6313c9a918cSPavel Labath   uint32_t StringOffsetOffset = StringOffsetsBase + 4 * (Index - 1);
6323c9a918cSPavel Labath   uint32_t EntryOffsetOffset = EntryOffsetsBase + 4 * (Index - 1);
6333c9a918cSPavel Labath   const DWARFDataExtractor &AS = Section.AccelSection;
6343c9a918cSPavel Labath 
6353c9a918cSPavel Labath   uint32_t StringOffset = AS.getRelocatedValue(4, &StringOffsetOffset);
6363c9a918cSPavel Labath   uint32_t EntryOffset = AS.getU32(&EntryOffsetOffset);
6373c9a918cSPavel Labath   EntryOffset += EntriesBase;
6383c9a918cSPavel Labath   return {StringOffset, EntryOffset};
6393c9a918cSPavel Labath }
6403c9a918cSPavel Labath 
6413c9a918cSPavel Labath uint32_t
6423c9a918cSPavel Labath DWARFDebugNames::NameIndex::getBucketArrayEntry(uint32_t Bucket) const {
6433c9a918cSPavel Labath   assert(Bucket < Hdr.BucketCount);
6443c9a918cSPavel Labath   uint32_t BucketOffset = BucketsBase + 4 * Bucket;
6453c9a918cSPavel Labath   return Section.AccelSection.getU32(&BucketOffset);
6463c9a918cSPavel Labath }
6473c9a918cSPavel Labath 
6483c9a918cSPavel Labath uint32_t DWARFDebugNames::NameIndex::getHashArrayEntry(uint32_t Index) const {
6493c9a918cSPavel Labath   assert(0 < Index && Index <= Hdr.NameCount);
6503c9a918cSPavel Labath   uint32_t HashOffset = HashesBase + 4 * (Index - 1);
6513c9a918cSPavel Labath   return Section.AccelSection.getU32(&HashOffset);
6523c9a918cSPavel Labath }
6533c9a918cSPavel Labath 
6543c9a918cSPavel Labath // Returns true if we should continue scanning for entries, false if this is the
6553c9a918cSPavel Labath // last (sentinel) entry). In case of a parsing error we also return false, as
6563c9a918cSPavel Labath // it's not possible to recover this entry list (but the other lists may still
6573c9a918cSPavel Labath // parse OK).
6583c9a918cSPavel Labath bool DWARFDebugNames::NameIndex::dumpEntry(ScopedPrinter &W,
6593c9a918cSPavel Labath                                            uint32_t *Offset) const {
6603c9a918cSPavel Labath   uint32_t EntryId = *Offset;
6613c9a918cSPavel Labath   auto EntryOr = getEntry(Offset);
6623c9a918cSPavel Labath   if (!EntryOr) {
6633c9a918cSPavel Labath     handleAllErrors(EntryOr.takeError(), [](const SentinelError &) {},
6643c9a918cSPavel Labath                     [&W](const ErrorInfoBase &EI) { EI.log(W.startLine()); });
6653c9a918cSPavel Labath     return false;
6663c9a918cSPavel Labath   }
6673c9a918cSPavel Labath 
6683c9a918cSPavel Labath   DictScope EntryScope(W, ("Entry @ 0x" + Twine::utohexstr(EntryId)).str());
6693c9a918cSPavel Labath   EntryOr->dump(W);
6703c9a918cSPavel Labath   return true;
6713c9a918cSPavel Labath }
6723c9a918cSPavel Labath 
6733c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpName(ScopedPrinter &W, uint32_t Index,
6743c9a918cSPavel Labath                                           Optional<uint32_t> Hash) const {
6753c9a918cSPavel Labath   const DataExtractor &SS = Section.StringSection;
6763c9a918cSPavel Labath   NameTableEntry NTE = getNameTableEntry(Index);
6773c9a918cSPavel Labath 
6783c9a918cSPavel Labath   DictScope NameScope(W, ("Name " + Twine(Index)).str());
6793c9a918cSPavel Labath   if (Hash)
6803c9a918cSPavel Labath     W.printHex("Hash", *Hash);
6813c9a918cSPavel Labath 
6823c9a918cSPavel Labath   W.startLine() << format("String: 0x%08x", NTE.StringOffset);
6833c9a918cSPavel Labath   W.getOStream() << " \"" << SS.getCStr(&NTE.StringOffset) << "\"\n";
6843c9a918cSPavel Labath 
6853c9a918cSPavel Labath   while (dumpEntry(W, &NTE.EntryOffset))
6863c9a918cSPavel Labath     /*empty*/;
6873c9a918cSPavel Labath }
6883c9a918cSPavel Labath 
6893c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpCUs(ScopedPrinter &W) const {
6903c9a918cSPavel Labath   ListScope CUScope(W, "Compilation Unit offsets");
6913c9a918cSPavel Labath   for (uint32_t CU = 0; CU < Hdr.CompUnitCount; ++CU)
6923c9a918cSPavel Labath     W.startLine() << format("CU[%u]: 0x%08x\n", CU, getCUOffset(CU));
6933c9a918cSPavel Labath }
6943c9a918cSPavel Labath 
6953c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpLocalTUs(ScopedPrinter &W) const {
6963c9a918cSPavel Labath   if (Hdr.LocalTypeUnitCount == 0)
6973c9a918cSPavel Labath     return;
6983c9a918cSPavel Labath 
6993c9a918cSPavel Labath   ListScope TUScope(W, "Local Type Unit offsets");
7003c9a918cSPavel Labath   for (uint32_t TU = 0; TU < Hdr.LocalTypeUnitCount; ++TU)
7013c9a918cSPavel Labath     W.startLine() << format("LocalTU[%u]: 0x%08x\n", TU, getLocalTUOffset(TU));
7023c9a918cSPavel Labath }
7033c9a918cSPavel Labath 
7043c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpForeignTUs(ScopedPrinter &W) const {
7053c9a918cSPavel Labath   if (Hdr.ForeignTypeUnitCount == 0)
7063c9a918cSPavel Labath     return;
7073c9a918cSPavel Labath 
7083c9a918cSPavel Labath   ListScope TUScope(W, "Foreign Type Unit signatures");
7093c9a918cSPavel Labath   for (uint32_t TU = 0; TU < Hdr.ForeignTypeUnitCount; ++TU) {
7103c9a918cSPavel Labath     W.startLine() << format("ForeignTU[%u]: 0x%016" PRIx64 "\n", TU,
711d99072bcSPavel Labath                             getForeignTUSignature(TU));
7123c9a918cSPavel Labath   }
7133c9a918cSPavel Labath }
7143c9a918cSPavel Labath 
7153c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpAbbreviations(ScopedPrinter &W) const {
7163c9a918cSPavel Labath   ListScope AbbrevsScope(W, "Abbreviations");
7173c9a918cSPavel Labath   for (const auto &Abbr : Abbrevs)
7183c9a918cSPavel Labath     Abbr.dump(W);
7193c9a918cSPavel Labath }
7203c9a918cSPavel Labath 
7213c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpBucket(ScopedPrinter &W,
7223c9a918cSPavel Labath                                             uint32_t Bucket) const {
7233c9a918cSPavel Labath   ListScope BucketScope(W, ("Bucket " + Twine(Bucket)).str());
7243c9a918cSPavel Labath   uint32_t Index = getBucketArrayEntry(Bucket);
7253c9a918cSPavel Labath   if (Index == 0) {
7263c9a918cSPavel Labath     W.printString("EMPTY");
7273c9a918cSPavel Labath     return;
7283c9a918cSPavel Labath   }
7293c9a918cSPavel Labath   if (Index > Hdr.NameCount) {
7303c9a918cSPavel Labath     W.printString("Name index is invalid");
7313c9a918cSPavel Labath     return;
7323c9a918cSPavel Labath   }
7333c9a918cSPavel Labath 
7343c9a918cSPavel Labath   for (; Index <= Hdr.NameCount; ++Index) {
7353c9a918cSPavel Labath     uint32_t Hash = getHashArrayEntry(Index);
7363c9a918cSPavel Labath     if (Hash % Hdr.BucketCount != Bucket)
7373c9a918cSPavel Labath       break;
7383c9a918cSPavel Labath 
7393c9a918cSPavel Labath     dumpName(W, Index, Hash);
7403c9a918cSPavel Labath   }
7413c9a918cSPavel Labath }
7423c9a918cSPavel Labath 
7433c9a918cSPavel Labath LLVM_DUMP_METHOD void DWARFDebugNames::NameIndex::dump(ScopedPrinter &W) const {
7443c9a918cSPavel Labath   DictScope UnitScope(W, ("Name Index @ 0x" + Twine::utohexstr(Base)).str());
7453c9a918cSPavel Labath   Hdr.dump(W);
7463c9a918cSPavel Labath   dumpCUs(W);
7473c9a918cSPavel Labath   dumpLocalTUs(W);
7483c9a918cSPavel Labath   dumpForeignTUs(W);
7493c9a918cSPavel Labath   dumpAbbreviations(W);
7503c9a918cSPavel Labath 
7513c9a918cSPavel Labath   if (Hdr.BucketCount > 0) {
7523c9a918cSPavel Labath     for (uint32_t Bucket = 0; Bucket < Hdr.BucketCount; ++Bucket)
7533c9a918cSPavel Labath       dumpBucket(W, Bucket);
7543c9a918cSPavel Labath     return;
7553c9a918cSPavel Labath   }
7563c9a918cSPavel Labath 
7573c9a918cSPavel Labath   W.startLine() << "Hash table not present\n";
7583c9a918cSPavel Labath   for (uint32_t Index = 1; Index <= Hdr.NameCount; ++Index)
7593c9a918cSPavel Labath     dumpName(W, Index, None);
7603c9a918cSPavel Labath }
7613c9a918cSPavel Labath 
7623c9a918cSPavel Labath llvm::Error DWARFDebugNames::extract() {
7633c9a918cSPavel Labath   uint32_t Offset = 0;
7643c9a918cSPavel Labath   while (AccelSection.isValidOffset(Offset)) {
7653c9a918cSPavel Labath     NameIndex Next(*this, Offset);
7663c9a918cSPavel Labath     if (llvm::Error E = Next.extract())
7673c9a918cSPavel Labath       return E;
7683c9a918cSPavel Labath     Offset = Next.getNextUnitOffset();
7693c9a918cSPavel Labath     NameIndices.push_back(std::move(Next));
7703c9a918cSPavel Labath   }
7713c9a918cSPavel Labath   return Error::success();
7723c9a918cSPavel Labath }
7733c9a918cSPavel Labath 
77480827f10SPavel Labath iterator_range<DWARFDebugNames::ValueIterator>
77580827f10SPavel Labath DWARFDebugNames::NameIndex::equal_range(StringRef Key) const {
77680827f10SPavel Labath   return make_range(ValueIterator(*this, Key), ValueIterator());
77780827f10SPavel Labath }
77880827f10SPavel Labath 
7793c9a918cSPavel Labath LLVM_DUMP_METHOD void DWARFDebugNames::dump(raw_ostream &OS) const {
7803c9a918cSPavel Labath   ScopedPrinter W(OS);
7813c9a918cSPavel Labath   for (const NameIndex &NI : NameIndices)
7823c9a918cSPavel Labath     NI.dump(W);
7833c9a918cSPavel Labath }
784d99072bcSPavel Labath 
785d99072bcSPavel Labath Optional<uint32_t>
786d99072bcSPavel Labath DWARFDebugNames::ValueIterator::findEntryOffsetInCurrentIndex() {
787d99072bcSPavel Labath   const Header &Hdr = CurrentIndex->Hdr;
788d99072bcSPavel Labath   if (Hdr.BucketCount == 0) {
789d99072bcSPavel Labath     // No Hash Table, We need to search through all names in the Name Index.
790d99072bcSPavel Labath     for (uint32_t Index = 1; Index <= Hdr.NameCount; ++Index) {
791d99072bcSPavel Labath       NameTableEntry NTE = CurrentIndex->getNameTableEntry(Index);
792d99072bcSPavel Labath       if (CurrentIndex->Section.StringSection.getCStr(&NTE.StringOffset) == Key)
793d99072bcSPavel Labath         return NTE.EntryOffset;
794d99072bcSPavel Labath     }
795d99072bcSPavel Labath     return None;
796d99072bcSPavel Labath   }
797d99072bcSPavel Labath 
798d99072bcSPavel Labath   // The Name Index has a Hash Table, so use that to speed up the search.
799d99072bcSPavel Labath   // Compute the Key Hash, if it has not been done already.
800d99072bcSPavel Labath   if (!Hash)
801d99072bcSPavel Labath     Hash = caseFoldingDjbHash(Key);
802d99072bcSPavel Labath   uint32_t Bucket = *Hash % Hdr.BucketCount;
803d99072bcSPavel Labath   uint32_t Index = CurrentIndex->getBucketArrayEntry(Bucket);
804d99072bcSPavel Labath   if (Index == 0)
805d99072bcSPavel Labath     return None; // Empty bucket
806d99072bcSPavel Labath 
807d99072bcSPavel Labath   for (; Index <= Hdr.NameCount; ++Index) {
808d99072bcSPavel Labath     uint32_t Hash = CurrentIndex->getHashArrayEntry(Index);
809d99072bcSPavel Labath     if (Hash % Hdr.BucketCount != Bucket)
810d99072bcSPavel Labath       return None; // End of bucket
811d99072bcSPavel Labath 
812d99072bcSPavel Labath     NameTableEntry NTE = CurrentIndex->getNameTableEntry(Index);
813d99072bcSPavel Labath     if (CurrentIndex->Section.StringSection.getCStr(&NTE.StringOffset) == Key)
814d99072bcSPavel Labath       return NTE.EntryOffset;
815d99072bcSPavel Labath   }
816d99072bcSPavel Labath   return None;
817d99072bcSPavel Labath }
818d99072bcSPavel Labath 
819d99072bcSPavel Labath bool DWARFDebugNames::ValueIterator::getEntryAtCurrentOffset() {
820d99072bcSPavel Labath   auto EntryOr = CurrentIndex->getEntry(&DataOffset);
821d99072bcSPavel Labath   if (!EntryOr) {
822d99072bcSPavel Labath     consumeError(EntryOr.takeError());
823d99072bcSPavel Labath     return false;
824d99072bcSPavel Labath   }
825d99072bcSPavel Labath   CurrentEntry = std::move(*EntryOr);
826d99072bcSPavel Labath   return true;
827d99072bcSPavel Labath }
828d99072bcSPavel Labath 
829d99072bcSPavel Labath bool DWARFDebugNames::ValueIterator::findInCurrentIndex() {
830d99072bcSPavel Labath   Optional<uint32_t> Offset = findEntryOffsetInCurrentIndex();
831d99072bcSPavel Labath   if (!Offset)
832d99072bcSPavel Labath     return false;
833d99072bcSPavel Labath   DataOffset = *Offset;
834d99072bcSPavel Labath   return getEntryAtCurrentOffset();
835d99072bcSPavel Labath }
836d99072bcSPavel Labath 
837d99072bcSPavel Labath void DWARFDebugNames::ValueIterator::searchFromStartOfCurrentIndex() {
838d99072bcSPavel Labath   for (const NameIndex *End = CurrentIndex->Section.NameIndices.end();
839d99072bcSPavel Labath        CurrentIndex != End; ++CurrentIndex) {
840d99072bcSPavel Labath     if (findInCurrentIndex())
841d99072bcSPavel Labath       return;
842d99072bcSPavel Labath   }
843d99072bcSPavel Labath   setEnd();
844d99072bcSPavel Labath }
845d99072bcSPavel Labath 
846d99072bcSPavel Labath void DWARFDebugNames::ValueIterator::next() {
847d99072bcSPavel Labath   assert(CurrentIndex && "Incrementing an end() iterator?");
848d99072bcSPavel Labath 
849d99072bcSPavel Labath   // First try the next entry in the current Index.
850d99072bcSPavel Labath   if (getEntryAtCurrentOffset())
851d99072bcSPavel Labath     return;
852d99072bcSPavel Labath 
853*59870af6SPavel Labath   // If we're a local iterator or we have reached the last Index, we're done.
854*59870af6SPavel Labath   if (IsLocal || CurrentIndex == &CurrentIndex->Section.NameIndices.back()) {
85580827f10SPavel Labath     setEnd();
85680827f10SPavel Labath     return;
85780827f10SPavel Labath   }
85880827f10SPavel Labath 
85980827f10SPavel Labath   // Otherwise, try the next index.
860d99072bcSPavel Labath   ++CurrentIndex;
861d99072bcSPavel Labath   searchFromStartOfCurrentIndex();
862d99072bcSPavel Labath }
863d99072bcSPavel Labath 
864d99072bcSPavel Labath DWARFDebugNames::ValueIterator::ValueIterator(const DWARFDebugNames &AccelTable,
865d99072bcSPavel Labath                                               StringRef Key)
86680827f10SPavel Labath     : CurrentIndex(AccelTable.NameIndices.begin()), IsLocal(false), Key(Key) {
867d99072bcSPavel Labath   searchFromStartOfCurrentIndex();
868d99072bcSPavel Labath }
869d99072bcSPavel Labath 
87080827f10SPavel Labath DWARFDebugNames::ValueIterator::ValueIterator(
87180827f10SPavel Labath     const DWARFDebugNames::NameIndex &NI, StringRef Key)
87280827f10SPavel Labath     : CurrentIndex(&NI), IsLocal(true), Key(Key) {
87380827f10SPavel Labath   if (!findInCurrentIndex())
87480827f10SPavel Labath     setEnd();
87580827f10SPavel Labath }
87680827f10SPavel Labath 
877d99072bcSPavel Labath iterator_range<DWARFDebugNames::ValueIterator>
878d99072bcSPavel Labath DWARFDebugNames::equal_range(StringRef Key) const {
879d99072bcSPavel Labath   if (NameIndices.empty())
880d99072bcSPavel Labath     return make_range(ValueIterator(), ValueIterator());
881d99072bcSPavel Labath   return make_range(ValueIterator(*this, Key), ValueIterator());
882d99072bcSPavel Labath }
88380827f10SPavel Labath 
88480827f10SPavel Labath const DWARFDebugNames::NameIndex *
88580827f10SPavel Labath DWARFDebugNames::getCUNameIndex(uint32_t CUOffset) {
88680827f10SPavel Labath   if (CUToNameIndex.size() == 0 && NameIndices.size() > 0) {
88780827f10SPavel Labath     for (const auto &NI : *this) {
88880827f10SPavel Labath       for (uint32_t CU = 0; CU < NI.getCUCount(); ++CU)
88980827f10SPavel Labath         CUToNameIndex.try_emplace(NI.getCUOffset(CU), &NI);
89080827f10SPavel Labath     }
89180827f10SPavel Labath   }
89280827f10SPavel Labath   return CUToNameIndex.lookup(CUOffset);
89380827f10SPavel Labath }
894