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; 31599fdb9d9SAdrian Prantl return; 31699fdb9d9SAdrian Prantl } 317d99072bcSPavel Labath Current.extract(*AccelTable, &DataOffset); 31899fdb9d9SAdrian Prantl ++Data; 31999fdb9d9SAdrian Prantl } 32099fdb9d9SAdrian Prantl 3219b36fd25SPavel Labath iterator_range<AppleAcceleratorTable::ValueIterator> 3229b36fd25SPavel Labath AppleAcceleratorTable::equal_range(StringRef Key) const { 32399fdb9d9SAdrian Prantl if (!IsValid) 32499fdb9d9SAdrian Prantl return make_range(ValueIterator(), ValueIterator()); 32599fdb9d9SAdrian Prantl 32699fdb9d9SAdrian Prantl // Find the bucket. 32792ac9d3eSJonas Devlieghere unsigned HashValue = djbHash(Key); 328394e8056SPavel Labath unsigned Bucket = HashValue % Hdr.BucketCount; 32999fdb9d9SAdrian Prantl unsigned BucketBase = sizeof(Hdr) + Hdr.HeaderDataLength; 330394e8056SPavel Labath unsigned HashesBase = BucketBase + Hdr.BucketCount * 4; 331394e8056SPavel Labath unsigned OffsetsBase = HashesBase + Hdr.HashCount * 4; 33299fdb9d9SAdrian Prantl 33399fdb9d9SAdrian Prantl unsigned BucketOffset = BucketBase + Bucket * 4; 33499fdb9d9SAdrian Prantl unsigned Index = AccelSection.getU32(&BucketOffset); 33599fdb9d9SAdrian Prantl 33699fdb9d9SAdrian Prantl // Search through all hashes in the bucket. 337394e8056SPavel Labath for (unsigned HashIdx = Index; HashIdx < Hdr.HashCount; ++HashIdx) { 33899fdb9d9SAdrian Prantl unsigned HashOffset = HashesBase + HashIdx * 4; 33999fdb9d9SAdrian Prantl unsigned OffsetsOffset = OffsetsBase + HashIdx * 4; 34099fdb9d9SAdrian Prantl uint32_t Hash = AccelSection.getU32(&HashOffset); 34199fdb9d9SAdrian Prantl 342394e8056SPavel Labath if (Hash % Hdr.BucketCount != Bucket) 34399fdb9d9SAdrian Prantl // We are already in the next bucket. 34499fdb9d9SAdrian Prantl break; 34599fdb9d9SAdrian Prantl 34699fdb9d9SAdrian Prantl unsigned DataOffset = AccelSection.getU32(&OffsetsOffset); 34799fdb9d9SAdrian Prantl unsigned StringOffset = AccelSection.getRelocatedValue(4, &DataOffset); 34899fdb9d9SAdrian Prantl if (!StringOffset) 34999fdb9d9SAdrian Prantl break; 35099fdb9d9SAdrian Prantl 35199fdb9d9SAdrian Prantl // Finally, compare the key. 35299fdb9d9SAdrian Prantl if (Key == StringSection.getCStr(&StringOffset)) 35399fdb9d9SAdrian Prantl return make_range({*this, DataOffset}, ValueIterator()); 35499fdb9d9SAdrian Prantl } 35599fdb9d9SAdrian Prantl return make_range(ValueIterator(), ValueIterator()); 35699fdb9d9SAdrian Prantl } 3573c9a918cSPavel Labath 3583c9a918cSPavel Labath void DWARFDebugNames::Header::dump(ScopedPrinter &W) const { 3593c9a918cSPavel Labath DictScope HeaderScope(W, "Header"); 3603c9a918cSPavel Labath W.printHex("Length", UnitLength); 3613c9a918cSPavel Labath W.printNumber("Version", Version); 3623c9a918cSPavel Labath W.printHex("Padding", Padding); 3633c9a918cSPavel Labath W.printNumber("CU count", CompUnitCount); 3643c9a918cSPavel Labath W.printNumber("Local TU count", LocalTypeUnitCount); 3653c9a918cSPavel Labath W.printNumber("Foreign TU count", ForeignTypeUnitCount); 3663c9a918cSPavel Labath W.printNumber("Bucket count", BucketCount); 3673c9a918cSPavel Labath W.printNumber("Name count", NameCount); 3683c9a918cSPavel Labath W.printHex("Abbreviations table size", AbbrevTableSize); 3693c9a918cSPavel Labath W.startLine() << "Augmentation: '" << AugmentationString << "'\n"; 3703c9a918cSPavel Labath } 3713c9a918cSPavel Labath 3723c9a918cSPavel Labath llvm::Error DWARFDebugNames::Header::extract(const DWARFDataExtractor &AS, 3733c9a918cSPavel Labath uint32_t *Offset) { 3743c9a918cSPavel Labath // Check that we can read the fixed-size part. 375e7264106SPavel Labath if (!AS.isValidOffset(*Offset + sizeof(HeaderPOD) - 1)) 3763c9a918cSPavel Labath return make_error<StringError>("Section too small: cannot read header.", 3773c9a918cSPavel Labath inconvertibleErrorCode()); 3783c9a918cSPavel Labath 3793c9a918cSPavel Labath UnitLength = AS.getU32(Offset); 3803c9a918cSPavel Labath Version = AS.getU16(Offset); 3813c9a918cSPavel Labath Padding = AS.getU16(Offset); 3823c9a918cSPavel Labath CompUnitCount = AS.getU32(Offset); 3833c9a918cSPavel Labath LocalTypeUnitCount = AS.getU32(Offset); 3843c9a918cSPavel Labath ForeignTypeUnitCount = AS.getU32(Offset); 3853c9a918cSPavel Labath BucketCount = AS.getU32(Offset); 3863c9a918cSPavel Labath NameCount = AS.getU32(Offset); 3873c9a918cSPavel Labath AbbrevTableSize = AS.getU32(Offset); 388*ea0f841cSPavel Labath AugmentationStringSize = alignTo(AS.getU32(Offset), 4); 3893c9a918cSPavel Labath 3903c9a918cSPavel Labath if (!AS.isValidOffsetForDataOfSize(*Offset, AugmentationStringSize)) 3913c9a918cSPavel Labath return make_error<StringError>( 3923c9a918cSPavel Labath "Section too small: cannot read header augmentation.", 3933c9a918cSPavel Labath inconvertibleErrorCode()); 3943c9a918cSPavel Labath AugmentationString.resize(AugmentationStringSize); 3953c9a918cSPavel Labath AS.getU8(Offset, reinterpret_cast<uint8_t *>(AugmentationString.data()), 3963c9a918cSPavel Labath AugmentationStringSize); 3973c9a918cSPavel Labath return Error::success(); 3983c9a918cSPavel Labath } 3993c9a918cSPavel Labath 4003c9a918cSPavel Labath void DWARFDebugNames::Abbrev::dump(ScopedPrinter &W) const { 4013c9a918cSPavel Labath DictScope AbbrevScope(W, ("Abbreviation 0x" + Twine::utohexstr(Code)).str()); 4029025f955SPavel Labath W.startLine() << formatv("Tag: {0}\n", Tag); 4033c9a918cSPavel Labath 4049025f955SPavel Labath for (const auto &Attr : Attributes) 4059025f955SPavel Labath W.startLine() << formatv("{0}: {1}\n", Attr.Index, Attr.Form); 4063c9a918cSPavel Labath } 4073c9a918cSPavel Labath 4083c9a918cSPavel Labath static constexpr DWARFDebugNames::AttributeEncoding sentinelAttrEnc() { 4093c9a918cSPavel Labath return {dwarf::Index(0), dwarf::Form(0)}; 4103c9a918cSPavel Labath } 4113c9a918cSPavel Labath 4123c9a918cSPavel Labath static bool isSentinel(const DWARFDebugNames::AttributeEncoding &AE) { 4133c9a918cSPavel Labath return AE == sentinelAttrEnc(); 4143c9a918cSPavel Labath } 4153c9a918cSPavel Labath 4163c9a918cSPavel Labath static DWARFDebugNames::Abbrev sentinelAbbrev() { 4173c9a918cSPavel Labath return DWARFDebugNames::Abbrev(0, dwarf::Tag(0), {}); 4183c9a918cSPavel Labath } 4193c9a918cSPavel Labath 4203c9a918cSPavel Labath static bool isSentinel(const DWARFDebugNames::Abbrev &Abbr) { 4213c9a918cSPavel Labath return Abbr.Code == 0; 4223c9a918cSPavel Labath } 4233c9a918cSPavel Labath 4243c9a918cSPavel Labath DWARFDebugNames::Abbrev DWARFDebugNames::AbbrevMapInfo::getEmptyKey() { 4253c9a918cSPavel Labath return sentinelAbbrev(); 4263c9a918cSPavel Labath } 4273c9a918cSPavel Labath 4283c9a918cSPavel Labath DWARFDebugNames::Abbrev DWARFDebugNames::AbbrevMapInfo::getTombstoneKey() { 4293c9a918cSPavel Labath return DWARFDebugNames::Abbrev(~0, dwarf::Tag(0), {}); 4303c9a918cSPavel Labath } 4313c9a918cSPavel Labath 4323c9a918cSPavel Labath Expected<DWARFDebugNames::AttributeEncoding> 4333c9a918cSPavel Labath DWARFDebugNames::NameIndex::extractAttributeEncoding(uint32_t *Offset) { 4343c9a918cSPavel Labath if (*Offset >= EntriesBase) { 4353c9a918cSPavel Labath return make_error<StringError>("Incorrectly terminated abbreviation table.", 4363c9a918cSPavel Labath inconvertibleErrorCode()); 4373c9a918cSPavel Labath } 4383c9a918cSPavel Labath 4393c9a918cSPavel Labath uint32_t Index = Section.AccelSection.getULEB128(Offset); 4403c9a918cSPavel Labath uint32_t Form = Section.AccelSection.getULEB128(Offset); 4413c9a918cSPavel Labath return AttributeEncoding(dwarf::Index(Index), dwarf::Form(Form)); 4423c9a918cSPavel Labath } 4433c9a918cSPavel Labath 4443c9a918cSPavel Labath Expected<std::vector<DWARFDebugNames::AttributeEncoding>> 4453c9a918cSPavel Labath DWARFDebugNames::NameIndex::extractAttributeEncodings(uint32_t *Offset) { 4463c9a918cSPavel Labath std::vector<AttributeEncoding> Result; 4473c9a918cSPavel Labath for (;;) { 4483c9a918cSPavel Labath auto AttrEncOr = extractAttributeEncoding(Offset); 4493c9a918cSPavel Labath if (!AttrEncOr) 4503c9a918cSPavel Labath return AttrEncOr.takeError(); 4513c9a918cSPavel Labath if (isSentinel(*AttrEncOr)) 4523c9a918cSPavel Labath return std::move(Result); 4533c9a918cSPavel Labath 4543c9a918cSPavel Labath Result.emplace_back(*AttrEncOr); 4553c9a918cSPavel Labath } 4563c9a918cSPavel Labath } 4573c9a918cSPavel Labath 4583c9a918cSPavel Labath Expected<DWARFDebugNames::Abbrev> 4593c9a918cSPavel Labath DWARFDebugNames::NameIndex::extractAbbrev(uint32_t *Offset) { 4603c9a918cSPavel Labath if (*Offset >= EntriesBase) { 4613c9a918cSPavel Labath return make_error<StringError>("Incorrectly terminated abbreviation table.", 4623c9a918cSPavel Labath inconvertibleErrorCode()); 4633c9a918cSPavel Labath } 4643c9a918cSPavel Labath 4653c9a918cSPavel Labath uint32_t Code = Section.AccelSection.getULEB128(Offset); 4663c9a918cSPavel Labath if (Code == 0) 4673c9a918cSPavel Labath return sentinelAbbrev(); 4683c9a918cSPavel Labath 4693c9a918cSPavel Labath uint32_t Tag = Section.AccelSection.getULEB128(Offset); 4703c9a918cSPavel Labath auto AttrEncOr = extractAttributeEncodings(Offset); 4713c9a918cSPavel Labath if (!AttrEncOr) 4723c9a918cSPavel Labath return AttrEncOr.takeError(); 4733c9a918cSPavel Labath return Abbrev(Code, dwarf::Tag(Tag), std::move(*AttrEncOr)); 4743c9a918cSPavel Labath } 4753c9a918cSPavel Labath 4763c9a918cSPavel Labath Error DWARFDebugNames::NameIndex::extract() { 4773c9a918cSPavel Labath const DWARFDataExtractor &AS = Section.AccelSection; 4783c9a918cSPavel Labath uint32_t Offset = Base; 4793c9a918cSPavel Labath if (Error E = Hdr.extract(AS, &Offset)) 4803c9a918cSPavel Labath return E; 4813c9a918cSPavel Labath 4823c9a918cSPavel Labath CUsBase = Offset; 4833c9a918cSPavel Labath Offset += Hdr.CompUnitCount * 4; 4843c9a918cSPavel Labath Offset += Hdr.LocalTypeUnitCount * 4; 4853c9a918cSPavel Labath Offset += Hdr.ForeignTypeUnitCount * 8; 4863c9a918cSPavel Labath BucketsBase = Offset; 4873c9a918cSPavel Labath Offset += Hdr.BucketCount * 4; 4883c9a918cSPavel Labath HashesBase = Offset; 4893c9a918cSPavel Labath if (Hdr.BucketCount > 0) 4903c9a918cSPavel Labath Offset += Hdr.NameCount * 4; 4913c9a918cSPavel Labath StringOffsetsBase = Offset; 4923c9a918cSPavel Labath Offset += Hdr.NameCount * 4; 4933c9a918cSPavel Labath EntryOffsetsBase = Offset; 4943c9a918cSPavel Labath Offset += Hdr.NameCount * 4; 4953c9a918cSPavel Labath 4963c9a918cSPavel Labath if (!AS.isValidOffsetForDataOfSize(Offset, Hdr.AbbrevTableSize)) 4973c9a918cSPavel Labath return make_error<StringError>( 4983c9a918cSPavel Labath "Section too small: cannot read abbreviations.", 4993c9a918cSPavel Labath inconvertibleErrorCode()); 5003c9a918cSPavel Labath 5013c9a918cSPavel Labath EntriesBase = Offset + Hdr.AbbrevTableSize; 5023c9a918cSPavel Labath 5033c9a918cSPavel Labath for (;;) { 5043c9a918cSPavel Labath auto AbbrevOr = extractAbbrev(&Offset); 5053c9a918cSPavel Labath if (!AbbrevOr) 5063c9a918cSPavel Labath return AbbrevOr.takeError(); 5073c9a918cSPavel Labath if (isSentinel(*AbbrevOr)) 5083c9a918cSPavel Labath return Error::success(); 5093c9a918cSPavel Labath 5103c9a918cSPavel Labath if (!Abbrevs.insert(std::move(*AbbrevOr)).second) { 5113c9a918cSPavel Labath return make_error<StringError>("Duplicate abbreviation code.", 5123c9a918cSPavel Labath inconvertibleErrorCode()); 5133c9a918cSPavel Labath } 5143c9a918cSPavel Labath } 5153c9a918cSPavel Labath } 516d99072bcSPavel Labath DWARFDebugNames::Entry::Entry(const NameIndex &NameIdx, const Abbrev &Abbr) 517d99072bcSPavel Labath : NameIdx(&NameIdx), Abbr(&Abbr) { 5183c9a918cSPavel Labath // This merely creates form values. It is up to the caller 5193c9a918cSPavel Labath // (NameIndex::getEntry) to populate them. 5203c9a918cSPavel Labath Values.reserve(Abbr.Attributes.size()); 5213c9a918cSPavel Labath for (const auto &Attr : Abbr.Attributes) 5223c9a918cSPavel Labath Values.emplace_back(Attr.Form); 5233c9a918cSPavel Labath } 5243c9a918cSPavel Labath 525d99072bcSPavel Labath Optional<DWARFFormValue> 526d99072bcSPavel Labath DWARFDebugNames::Entry::lookup(dwarf::Index Index) const { 527d99072bcSPavel Labath assert(Abbr->Attributes.size() == Values.size()); 528d99072bcSPavel Labath for (const auto &Tuple : zip_first(Abbr->Attributes, Values)) { 529d99072bcSPavel Labath if (std::get<0>(Tuple).Index == Index) 530d99072bcSPavel Labath return std::get<1>(Tuple); 531d99072bcSPavel Labath } 532d99072bcSPavel Labath return None; 533d99072bcSPavel Labath } 5343c9a918cSPavel Labath 53547c3472cSPavel Labath Optional<uint64_t> DWARFDebugNames::Entry::getDIEUnitOffset() const { 536d99072bcSPavel Labath if (Optional<DWARFFormValue> Off = lookup(dwarf::DW_IDX_die_offset)) 5372d1fc437SPavel Labath return Off->getAsReferenceUVal(); 538d99072bcSPavel Labath return None; 539d99072bcSPavel Labath } 540d99072bcSPavel Labath 541d99072bcSPavel Labath Optional<uint64_t> DWARFDebugNames::Entry::getCUIndex() const { 542d99072bcSPavel Labath if (Optional<DWARFFormValue> Off = lookup(dwarf::DW_IDX_compile_unit)) 543d99072bcSPavel Labath return Off->getAsUnsignedConstant(); 54447c3472cSPavel Labath // In a per-CU index, the entries without a DW_IDX_compile_unit attribute 54547c3472cSPavel Labath // implicitly refer to the single CU. 54647c3472cSPavel Labath if (NameIdx->getCUCount() == 1) 54747c3472cSPavel Labath return 0; 548d99072bcSPavel Labath return None; 549d99072bcSPavel Labath } 550d99072bcSPavel Labath 551d99072bcSPavel Labath Optional<uint64_t> DWARFDebugNames::Entry::getCUOffset() const { 552d99072bcSPavel Labath Optional<uint64_t> Index = getCUIndex(); 553d99072bcSPavel Labath if (!Index || *Index >= NameIdx->getCUCount()) 554d99072bcSPavel Labath return None; 555d99072bcSPavel Labath return NameIdx->getCUOffset(*Index); 556d99072bcSPavel Labath } 557d99072bcSPavel Labath 55847c3472cSPavel Labath Optional<uint64_t> DWARFDebugNames::Entry::getDIESectionOffset() const { 55947c3472cSPavel Labath Optional<uint64_t> CUOff = getCUOffset(); 56047c3472cSPavel Labath Optional<uint64_t> DIEOff = getDIEUnitOffset(); 56147c3472cSPavel Labath if (CUOff && DIEOff) 56247c3472cSPavel Labath return *CUOff + *DIEOff; 56347c3472cSPavel Labath return None; 56447c3472cSPavel Labath } 56547c3472cSPavel Labath 566d99072bcSPavel Labath void DWARFDebugNames::Entry::dump(ScopedPrinter &W) const { 567d99072bcSPavel Labath W.printHex("Abbrev", Abbr->Code); 5689025f955SPavel Labath W.startLine() << formatv("Tag: {0}\n", Abbr->Tag); 569d99072bcSPavel Labath assert(Abbr->Attributes.size() == Values.size()); 570d99072bcSPavel Labath for (const auto &Tuple : zip_first(Abbr->Attributes, Values)) { 5719025f955SPavel Labath W.startLine() << formatv("{0}: ", std::get<0>(Tuple).Index); 572d99072bcSPavel Labath std::get<1>(Tuple).dump(W.getOStream()); 5733c9a918cSPavel Labath W.getOStream() << '\n'; 5743c9a918cSPavel Labath } 5753c9a918cSPavel Labath } 5763c9a918cSPavel Labath 5773c9a918cSPavel Labath char DWARFDebugNames::SentinelError::ID; 5783c9a918cSPavel Labath std::error_code DWARFDebugNames::SentinelError::convertToErrorCode() const { 5793c9a918cSPavel Labath return inconvertibleErrorCode(); 5803c9a918cSPavel Labath } 5813c9a918cSPavel Labath 5823c9a918cSPavel Labath uint32_t DWARFDebugNames::NameIndex::getCUOffset(uint32_t CU) const { 5833c9a918cSPavel Labath assert(CU < Hdr.CompUnitCount); 5843c9a918cSPavel Labath uint32_t Offset = CUsBase + 4 * CU; 5853c9a918cSPavel Labath return Section.AccelSection.getRelocatedValue(4, &Offset); 5863c9a918cSPavel Labath } 5873c9a918cSPavel Labath 5883c9a918cSPavel Labath uint32_t DWARFDebugNames::NameIndex::getLocalTUOffset(uint32_t TU) const { 5893c9a918cSPavel Labath assert(TU < Hdr.LocalTypeUnitCount); 5903c9a918cSPavel Labath uint32_t Offset = CUsBase + Hdr.CompUnitCount * 4; 5913c9a918cSPavel Labath return Section.AccelSection.getRelocatedValue(4, &Offset); 5923c9a918cSPavel Labath } 5933c9a918cSPavel Labath 594d99072bcSPavel Labath uint64_t DWARFDebugNames::NameIndex::getForeignTUSignature(uint32_t TU) const { 5953c9a918cSPavel Labath assert(TU < Hdr.ForeignTypeUnitCount); 5963c9a918cSPavel Labath uint32_t Offset = CUsBase + (Hdr.CompUnitCount + Hdr.LocalTypeUnitCount) * 4; 5973c9a918cSPavel Labath return Section.AccelSection.getU64(&Offset); 5983c9a918cSPavel Labath } 5993c9a918cSPavel Labath 6003c9a918cSPavel Labath Expected<DWARFDebugNames::Entry> 6013c9a918cSPavel Labath DWARFDebugNames::NameIndex::getEntry(uint32_t *Offset) const { 6023c9a918cSPavel Labath const DWARFDataExtractor &AS = Section.AccelSection; 6033c9a918cSPavel Labath if (!AS.isValidOffset(*Offset)) 6043c9a918cSPavel Labath return make_error<StringError>("Incorrectly terminated entry list", 6053c9a918cSPavel Labath inconvertibleErrorCode()); 6063c9a918cSPavel Labath 6073c9a918cSPavel Labath uint32_t AbbrevCode = AS.getULEB128(Offset); 6083c9a918cSPavel Labath if (AbbrevCode == 0) 6093c9a918cSPavel Labath return make_error<SentinelError>(); 6103c9a918cSPavel Labath 6113c9a918cSPavel Labath const auto AbbrevIt = Abbrevs.find_as(AbbrevCode); 6123c9a918cSPavel Labath if (AbbrevIt == Abbrevs.end()) 6133c9a918cSPavel Labath return make_error<StringError>("Invalid abbreviation", 6143c9a918cSPavel Labath inconvertibleErrorCode()); 6153c9a918cSPavel Labath 616d99072bcSPavel Labath Entry E(*this, *AbbrevIt); 6173c9a918cSPavel Labath 618322711f5SPavel Labath dwarf::FormParams FormParams = {Hdr.Version, 0, dwarf::DwarfFormat::DWARF32}; 6193c9a918cSPavel Labath for (auto &Value : E.Values) { 6203c9a918cSPavel Labath if (!Value.extractValue(AS, Offset, FormParams)) 6213c9a918cSPavel Labath return make_error<StringError>("Error extracting index attribute values", 6223c9a918cSPavel Labath inconvertibleErrorCode()); 6233c9a918cSPavel Labath } 6243c9a918cSPavel Labath return std::move(E); 6253c9a918cSPavel Labath } 6263c9a918cSPavel Labath 6273c9a918cSPavel Labath DWARFDebugNames::NameTableEntry 6283c9a918cSPavel Labath DWARFDebugNames::NameIndex::getNameTableEntry(uint32_t Index) const { 6293c9a918cSPavel Labath assert(0 < Index && Index <= Hdr.NameCount); 6303c9a918cSPavel Labath uint32_t StringOffsetOffset = StringOffsetsBase + 4 * (Index - 1); 6313c9a918cSPavel Labath uint32_t EntryOffsetOffset = EntryOffsetsBase + 4 * (Index - 1); 6323c9a918cSPavel Labath const DWARFDataExtractor &AS = Section.AccelSection; 6333c9a918cSPavel Labath 6343c9a918cSPavel Labath uint32_t StringOffset = AS.getRelocatedValue(4, &StringOffsetOffset); 6353c9a918cSPavel Labath uint32_t EntryOffset = AS.getU32(&EntryOffsetOffset); 6363c9a918cSPavel Labath EntryOffset += EntriesBase; 6373c9a918cSPavel Labath return {StringOffset, EntryOffset}; 6383c9a918cSPavel Labath } 6393c9a918cSPavel Labath 6403c9a918cSPavel Labath uint32_t 6413c9a918cSPavel Labath DWARFDebugNames::NameIndex::getBucketArrayEntry(uint32_t Bucket) const { 6423c9a918cSPavel Labath assert(Bucket < Hdr.BucketCount); 6433c9a918cSPavel Labath uint32_t BucketOffset = BucketsBase + 4 * Bucket; 6443c9a918cSPavel Labath return Section.AccelSection.getU32(&BucketOffset); 6453c9a918cSPavel Labath } 6463c9a918cSPavel Labath 6473c9a918cSPavel Labath uint32_t DWARFDebugNames::NameIndex::getHashArrayEntry(uint32_t Index) const { 6483c9a918cSPavel Labath assert(0 < Index && Index <= Hdr.NameCount); 6493c9a918cSPavel Labath uint32_t HashOffset = HashesBase + 4 * (Index - 1); 6503c9a918cSPavel Labath return Section.AccelSection.getU32(&HashOffset); 6513c9a918cSPavel Labath } 6523c9a918cSPavel Labath 6533c9a918cSPavel Labath // Returns true if we should continue scanning for entries, false if this is the 6543c9a918cSPavel Labath // last (sentinel) entry). In case of a parsing error we also return false, as 6553c9a918cSPavel Labath // it's not possible to recover this entry list (but the other lists may still 6563c9a918cSPavel Labath // parse OK). 6573c9a918cSPavel Labath bool DWARFDebugNames::NameIndex::dumpEntry(ScopedPrinter &W, 6583c9a918cSPavel Labath uint32_t *Offset) const { 6593c9a918cSPavel Labath uint32_t EntryId = *Offset; 6603c9a918cSPavel Labath auto EntryOr = getEntry(Offset); 6613c9a918cSPavel Labath if (!EntryOr) { 6623c9a918cSPavel Labath handleAllErrors(EntryOr.takeError(), [](const SentinelError &) {}, 6633c9a918cSPavel Labath [&W](const ErrorInfoBase &EI) { EI.log(W.startLine()); }); 6643c9a918cSPavel Labath return false; 6653c9a918cSPavel Labath } 6663c9a918cSPavel Labath 6673c9a918cSPavel Labath DictScope EntryScope(W, ("Entry @ 0x" + Twine::utohexstr(EntryId)).str()); 6683c9a918cSPavel Labath EntryOr->dump(W); 6693c9a918cSPavel Labath return true; 6703c9a918cSPavel Labath } 6713c9a918cSPavel Labath 6723c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpName(ScopedPrinter &W, uint32_t Index, 6733c9a918cSPavel Labath Optional<uint32_t> Hash) const { 6743c9a918cSPavel Labath const DataExtractor &SS = Section.StringSection; 6753c9a918cSPavel Labath NameTableEntry NTE = getNameTableEntry(Index); 6763c9a918cSPavel Labath 6773c9a918cSPavel Labath DictScope NameScope(W, ("Name " + Twine(Index)).str()); 6783c9a918cSPavel Labath if (Hash) 6793c9a918cSPavel Labath W.printHex("Hash", *Hash); 6803c9a918cSPavel Labath 6813c9a918cSPavel Labath W.startLine() << format("String: 0x%08x", NTE.StringOffset); 6823c9a918cSPavel Labath W.getOStream() << " \"" << SS.getCStr(&NTE.StringOffset) << "\"\n"; 6833c9a918cSPavel Labath 6843c9a918cSPavel Labath while (dumpEntry(W, &NTE.EntryOffset)) 6853c9a918cSPavel Labath /*empty*/; 6863c9a918cSPavel Labath } 6873c9a918cSPavel Labath 6883c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpCUs(ScopedPrinter &W) const { 6893c9a918cSPavel Labath ListScope CUScope(W, "Compilation Unit offsets"); 6903c9a918cSPavel Labath for (uint32_t CU = 0; CU < Hdr.CompUnitCount; ++CU) 6913c9a918cSPavel Labath W.startLine() << format("CU[%u]: 0x%08x\n", CU, getCUOffset(CU)); 6923c9a918cSPavel Labath } 6933c9a918cSPavel Labath 6943c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpLocalTUs(ScopedPrinter &W) const { 6953c9a918cSPavel Labath if (Hdr.LocalTypeUnitCount == 0) 6963c9a918cSPavel Labath return; 6973c9a918cSPavel Labath 6983c9a918cSPavel Labath ListScope TUScope(W, "Local Type Unit offsets"); 6993c9a918cSPavel Labath for (uint32_t TU = 0; TU < Hdr.LocalTypeUnitCount; ++TU) 7003c9a918cSPavel Labath W.startLine() << format("LocalTU[%u]: 0x%08x\n", TU, getLocalTUOffset(TU)); 7013c9a918cSPavel Labath } 7023c9a918cSPavel Labath 7033c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpForeignTUs(ScopedPrinter &W) const { 7043c9a918cSPavel Labath if (Hdr.ForeignTypeUnitCount == 0) 7053c9a918cSPavel Labath return; 7063c9a918cSPavel Labath 7073c9a918cSPavel Labath ListScope TUScope(W, "Foreign Type Unit signatures"); 7083c9a918cSPavel Labath for (uint32_t TU = 0; TU < Hdr.ForeignTypeUnitCount; ++TU) { 7093c9a918cSPavel Labath W.startLine() << format("ForeignTU[%u]: 0x%016" PRIx64 "\n", TU, 710d99072bcSPavel Labath getForeignTUSignature(TU)); 7113c9a918cSPavel Labath } 7123c9a918cSPavel Labath } 7133c9a918cSPavel Labath 7143c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpAbbreviations(ScopedPrinter &W) const { 7153c9a918cSPavel Labath ListScope AbbrevsScope(W, "Abbreviations"); 7163c9a918cSPavel Labath for (const auto &Abbr : Abbrevs) 7173c9a918cSPavel Labath Abbr.dump(W); 7183c9a918cSPavel Labath } 7193c9a918cSPavel Labath 7203c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpBucket(ScopedPrinter &W, 7213c9a918cSPavel Labath uint32_t Bucket) const { 7223c9a918cSPavel Labath ListScope BucketScope(W, ("Bucket " + Twine(Bucket)).str()); 7233c9a918cSPavel Labath uint32_t Index = getBucketArrayEntry(Bucket); 7243c9a918cSPavel Labath if (Index == 0) { 7253c9a918cSPavel Labath W.printString("EMPTY"); 7263c9a918cSPavel Labath return; 7273c9a918cSPavel Labath } 7283c9a918cSPavel Labath if (Index > Hdr.NameCount) { 7293c9a918cSPavel Labath W.printString("Name index is invalid"); 7303c9a918cSPavel Labath return; 7313c9a918cSPavel Labath } 7323c9a918cSPavel Labath 7333c9a918cSPavel Labath for (; Index <= Hdr.NameCount; ++Index) { 7343c9a918cSPavel Labath uint32_t Hash = getHashArrayEntry(Index); 7353c9a918cSPavel Labath if (Hash % Hdr.BucketCount != Bucket) 7363c9a918cSPavel Labath break; 7373c9a918cSPavel Labath 7383c9a918cSPavel Labath dumpName(W, Index, Hash); 7393c9a918cSPavel Labath } 7403c9a918cSPavel Labath } 7413c9a918cSPavel Labath 7423c9a918cSPavel Labath LLVM_DUMP_METHOD void DWARFDebugNames::NameIndex::dump(ScopedPrinter &W) const { 7433c9a918cSPavel Labath DictScope UnitScope(W, ("Name Index @ 0x" + Twine::utohexstr(Base)).str()); 7443c9a918cSPavel Labath Hdr.dump(W); 7453c9a918cSPavel Labath dumpCUs(W); 7463c9a918cSPavel Labath dumpLocalTUs(W); 7473c9a918cSPavel Labath dumpForeignTUs(W); 7483c9a918cSPavel Labath dumpAbbreviations(W); 7493c9a918cSPavel Labath 7503c9a918cSPavel Labath if (Hdr.BucketCount > 0) { 7513c9a918cSPavel Labath for (uint32_t Bucket = 0; Bucket < Hdr.BucketCount; ++Bucket) 7523c9a918cSPavel Labath dumpBucket(W, Bucket); 7533c9a918cSPavel Labath return; 7543c9a918cSPavel Labath } 7553c9a918cSPavel Labath 7563c9a918cSPavel Labath W.startLine() << "Hash table not present\n"; 7573c9a918cSPavel Labath for (uint32_t Index = 1; Index <= Hdr.NameCount; ++Index) 7583c9a918cSPavel Labath dumpName(W, Index, None); 7593c9a918cSPavel Labath } 7603c9a918cSPavel Labath 7613c9a918cSPavel Labath llvm::Error DWARFDebugNames::extract() { 7623c9a918cSPavel Labath uint32_t Offset = 0; 7633c9a918cSPavel Labath while (AccelSection.isValidOffset(Offset)) { 7643c9a918cSPavel Labath NameIndex Next(*this, Offset); 7653c9a918cSPavel Labath if (llvm::Error E = Next.extract()) 7663c9a918cSPavel Labath return E; 7673c9a918cSPavel Labath Offset = Next.getNextUnitOffset(); 7683c9a918cSPavel Labath NameIndices.push_back(std::move(Next)); 7693c9a918cSPavel Labath } 7703c9a918cSPavel Labath return Error::success(); 7713c9a918cSPavel Labath } 7723c9a918cSPavel Labath 7733c9a918cSPavel Labath LLVM_DUMP_METHOD void DWARFDebugNames::dump(raw_ostream &OS) const { 7743c9a918cSPavel Labath ScopedPrinter W(OS); 7753c9a918cSPavel Labath for (const NameIndex &NI : NameIndices) 7763c9a918cSPavel Labath NI.dump(W); 7773c9a918cSPavel Labath } 778d99072bcSPavel Labath 779d99072bcSPavel Labath Optional<uint32_t> 780d99072bcSPavel Labath DWARFDebugNames::ValueIterator::findEntryOffsetInCurrentIndex() { 781d99072bcSPavel Labath const Header &Hdr = CurrentIndex->Hdr; 782d99072bcSPavel Labath if (Hdr.BucketCount == 0) { 783d99072bcSPavel Labath // No Hash Table, We need to search through all names in the Name Index. 784d99072bcSPavel Labath for (uint32_t Index = 1; Index <= Hdr.NameCount; ++Index) { 785d99072bcSPavel Labath NameTableEntry NTE = CurrentIndex->getNameTableEntry(Index); 786d99072bcSPavel Labath if (CurrentIndex->Section.StringSection.getCStr(&NTE.StringOffset) == Key) 787d99072bcSPavel Labath return NTE.EntryOffset; 788d99072bcSPavel Labath } 789d99072bcSPavel Labath return None; 790d99072bcSPavel Labath } 791d99072bcSPavel Labath 792d99072bcSPavel Labath // The Name Index has a Hash Table, so use that to speed up the search. 793d99072bcSPavel Labath // Compute the Key Hash, if it has not been done already. 794d99072bcSPavel Labath if (!Hash) 795d99072bcSPavel Labath Hash = caseFoldingDjbHash(Key); 796d99072bcSPavel Labath uint32_t Bucket = *Hash % Hdr.BucketCount; 797d99072bcSPavel Labath uint32_t Index = CurrentIndex->getBucketArrayEntry(Bucket); 798d99072bcSPavel Labath if (Index == 0) 799d99072bcSPavel Labath return None; // Empty bucket 800d99072bcSPavel Labath 801d99072bcSPavel Labath for (; Index <= Hdr.NameCount; ++Index) { 802d99072bcSPavel Labath uint32_t Hash = CurrentIndex->getHashArrayEntry(Index); 803d99072bcSPavel Labath if (Hash % Hdr.BucketCount != Bucket) 804d99072bcSPavel Labath return None; // End of bucket 805d99072bcSPavel Labath 806d99072bcSPavel Labath NameTableEntry NTE = CurrentIndex->getNameTableEntry(Index); 807d99072bcSPavel Labath if (CurrentIndex->Section.StringSection.getCStr(&NTE.StringOffset) == Key) 808d99072bcSPavel Labath return NTE.EntryOffset; 809d99072bcSPavel Labath } 810d99072bcSPavel Labath return None; 811d99072bcSPavel Labath } 812d99072bcSPavel Labath 813d99072bcSPavel Labath bool DWARFDebugNames::ValueIterator::getEntryAtCurrentOffset() { 814d99072bcSPavel Labath auto EntryOr = CurrentIndex->getEntry(&DataOffset); 815d99072bcSPavel Labath if (!EntryOr) { 816d99072bcSPavel Labath consumeError(EntryOr.takeError()); 817d99072bcSPavel Labath return false; 818d99072bcSPavel Labath } 819d99072bcSPavel Labath CurrentEntry = std::move(*EntryOr); 820d99072bcSPavel Labath return true; 821d99072bcSPavel Labath } 822d99072bcSPavel Labath 823d99072bcSPavel Labath bool DWARFDebugNames::ValueIterator::findInCurrentIndex() { 824d99072bcSPavel Labath Optional<uint32_t> Offset = findEntryOffsetInCurrentIndex(); 825d99072bcSPavel Labath if (!Offset) 826d99072bcSPavel Labath return false; 827d99072bcSPavel Labath DataOffset = *Offset; 828d99072bcSPavel Labath return getEntryAtCurrentOffset(); 829d99072bcSPavel Labath } 830d99072bcSPavel Labath 831d99072bcSPavel Labath void DWARFDebugNames::ValueIterator::searchFromStartOfCurrentIndex() { 832d99072bcSPavel Labath for (const NameIndex *End = CurrentIndex->Section.NameIndices.end(); 833d99072bcSPavel Labath CurrentIndex != End; ++CurrentIndex) { 834d99072bcSPavel Labath if (findInCurrentIndex()) 835d99072bcSPavel Labath return; 836d99072bcSPavel Labath } 837d99072bcSPavel Labath setEnd(); 838d99072bcSPavel Labath } 839d99072bcSPavel Labath 840d99072bcSPavel Labath void DWARFDebugNames::ValueIterator::next() { 841d99072bcSPavel Labath assert(CurrentIndex && "Incrementing an end() iterator?"); 842d99072bcSPavel Labath 843d99072bcSPavel Labath // First try the next entry in the current Index. 844d99072bcSPavel Labath if (getEntryAtCurrentOffset()) 845d99072bcSPavel Labath return; 846d99072bcSPavel Labath 847d99072bcSPavel Labath // Try the next Name Index. 848d99072bcSPavel Labath ++CurrentIndex; 849d99072bcSPavel Labath searchFromStartOfCurrentIndex(); 850d99072bcSPavel Labath } 851d99072bcSPavel Labath 852d99072bcSPavel Labath DWARFDebugNames::ValueIterator::ValueIterator(const DWARFDebugNames &AccelTable, 853d99072bcSPavel Labath StringRef Key) 854d99072bcSPavel Labath : CurrentIndex(AccelTable.NameIndices.begin()), Key(Key) { 855d99072bcSPavel Labath searchFromStartOfCurrentIndex(); 856d99072bcSPavel Labath } 857d99072bcSPavel Labath 858d99072bcSPavel Labath iterator_range<DWARFDebugNames::ValueIterator> 859d99072bcSPavel Labath DWARFDebugNames::equal_range(StringRef Key) const { 860d99072bcSPavel Labath if (NameIndices.empty()) 861d99072bcSPavel Labath return make_range(ValueIterator(), ValueIterator()); 862d99072bcSPavel Labath return make_range(ValueIterator(*this, Key), ValueIterator()); 863d99072bcSPavel Labath } 864