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" 183c9a918cSPavel Labath #include "llvm/Support/ScopedPrinter.h" 1982af9438SZachary Turner #include "llvm/Support/raw_ostream.h" 20e94042caSEugene Zelenko #include <cstddef> 21e94042caSEugene Zelenko #include <cstdint> 22e94042caSEugene Zelenko #include <utility> 2382af9438SZachary Turner 24e94042caSEugene Zelenko using namespace llvm; 2582af9438SZachary Turner 263c9a918cSPavel Labath namespace { 273c9a918cSPavel Labath struct DwarfConstant { 283c9a918cSPavel Labath StringRef (*StringFn)(unsigned); 293c9a918cSPavel Labath StringRef Type; 303c9a918cSPavel Labath unsigned Value; 313c9a918cSPavel Labath }; 323c9a918cSPavel Labath 333c9a918cSPavel Labath static raw_ostream &operator<<(raw_ostream &OS, const DwarfConstant &C) { 343c9a918cSPavel Labath StringRef Str = C.StringFn(C.Value); 353c9a918cSPavel Labath if (!Str.empty()) 363c9a918cSPavel Labath return OS << Str; 373c9a918cSPavel Labath return OS << "DW_" << C.Type << "_Unknown_0x" << format("%x", C.Value); 383c9a918cSPavel Labath } 393c9a918cSPavel Labath } // namespace 403c9a918cSPavel Labath 413c9a918cSPavel Labath static DwarfConstant formatTag(unsigned Tag) { 423c9a918cSPavel Labath return {dwarf::TagString, "TAG", Tag}; 433c9a918cSPavel Labath } 443c9a918cSPavel Labath 453c9a918cSPavel Labath static DwarfConstant formatForm(unsigned Form) { 463c9a918cSPavel Labath return {dwarf::FormEncodingString, "FORM", Form}; 473c9a918cSPavel Labath } 483c9a918cSPavel Labath 493c9a918cSPavel Labath static DwarfConstant formatIndex(unsigned Idx) { 503c9a918cSPavel Labath return {dwarf::IndexString, "IDX", Idx}; 513c9a918cSPavel Labath } 523c9a918cSPavel Labath 533c9a918cSPavel Labath static DwarfConstant formatAtom(unsigned Atom) { 543c9a918cSPavel Labath return {dwarf::AtomTypeString, "ATOM", Atom}; 553c9a918cSPavel Labath } 563c9a918cSPavel Labath 573c9a918cSPavel Labath DWARFAcceleratorTable::~DWARFAcceleratorTable() = default; 583c9a918cSPavel Labath 599b36fd25SPavel Labath llvm::Error AppleAcceleratorTable::extract() { 6082af9438SZachary Turner uint32_t Offset = 0; 6182af9438SZachary Turner 6282af9438SZachary Turner // Check that we can at least read the header. 6382af9438SZachary Turner if (!AccelSection.isValidOffset(offsetof(Header, HeaderDataLength)+4)) 64ba915897SJonas Devlieghere return make_error<StringError>("Section too small: cannot read header.", 65ba915897SJonas Devlieghere inconvertibleErrorCode()); 6682af9438SZachary Turner 6782af9438SZachary Turner Hdr.Magic = AccelSection.getU32(&Offset); 6882af9438SZachary Turner Hdr.Version = AccelSection.getU16(&Offset); 6982af9438SZachary Turner Hdr.HashFunction = AccelSection.getU16(&Offset); 70394e8056SPavel Labath Hdr.BucketCount = AccelSection.getU32(&Offset); 71394e8056SPavel Labath Hdr.HashCount = AccelSection.getU32(&Offset); 7282af9438SZachary Turner Hdr.HeaderDataLength = AccelSection.getU32(&Offset); 7382af9438SZachary Turner 7482af9438SZachary Turner // Check that we can read all the hashes and offsets from the 7582af9438SZachary Turner // section (see SourceLevelDebugging.rst for the structure of the index). 76ba915897SJonas Devlieghere // We need to substract one because we're checking for an *offset* which is 77ba915897SJonas Devlieghere // equal to the size for an empty table and hence pointer after the section. 7882af9438SZachary Turner if (!AccelSection.isValidOffset(sizeof(Hdr) + Hdr.HeaderDataLength + 79394e8056SPavel Labath Hdr.BucketCount * 4 + Hdr.HashCount * 8 - 1)) 80ba915897SJonas Devlieghere return make_error<StringError>( 81ba915897SJonas Devlieghere "Section too small: cannot read buckets and hashes.", 82ba915897SJonas Devlieghere inconvertibleErrorCode()); 8382af9438SZachary Turner 8482af9438SZachary Turner HdrData.DIEOffsetBase = AccelSection.getU32(&Offset); 8582af9438SZachary Turner uint32_t NumAtoms = AccelSection.getU32(&Offset); 8682af9438SZachary Turner 8782af9438SZachary Turner for (unsigned i = 0; i < NumAtoms; ++i) { 8882af9438SZachary Turner uint16_t AtomType = AccelSection.getU16(&Offset); 896c273763SGreg Clayton auto AtomForm = static_cast<dwarf::Form>(AccelSection.getU16(&Offset)); 9082af9438SZachary Turner HdrData.Atoms.push_back(std::make_pair(AtomType, AtomForm)); 9182af9438SZachary Turner } 9282af9438SZachary Turner 9399fdb9d9SAdrian Prantl IsValid = true; 94ba915897SJonas Devlieghere return Error::success(); 9582af9438SZachary Turner } 9682af9438SZachary Turner 97394e8056SPavel Labath uint32_t AppleAcceleratorTable::getNumBuckets() { return Hdr.BucketCount; } 98394e8056SPavel Labath uint32_t AppleAcceleratorTable::getNumHashes() { return Hdr.HashCount; } 999b36fd25SPavel Labath uint32_t AppleAcceleratorTable::getSizeHdr() { return sizeof(Hdr); } 1009b36fd25SPavel Labath uint32_t AppleAcceleratorTable::getHeaderDataLength() { 101e41823bbSSpyridoula Gravani return Hdr.HeaderDataLength; 102e41823bbSSpyridoula Gravani } 103e41823bbSSpyridoula Gravani 1049b36fd25SPavel Labath ArrayRef<std::pair<AppleAcceleratorTable::HeaderData::AtomType, 1059b36fd25SPavel Labath AppleAcceleratorTable::HeaderData::Form>> 1069b36fd25SPavel Labath AppleAcceleratorTable::getAtomsDesc() { 107837c110cSSpyridoula Gravani return HdrData.Atoms; 108837c110cSSpyridoula Gravani } 109837c110cSSpyridoula Gravani 1109b36fd25SPavel Labath bool AppleAcceleratorTable::validateForms() { 111837c110cSSpyridoula Gravani for (auto Atom : getAtomsDesc()) { 112837c110cSSpyridoula Gravani DWARFFormValue FormValue(Atom.second); 113837c110cSSpyridoula Gravani switch (Atom.first) { 114837c110cSSpyridoula Gravani case dwarf::DW_ATOM_die_offset: 11570d35e10SSpyridoula Gravani case dwarf::DW_ATOM_die_tag: 11670d35e10SSpyridoula Gravani case dwarf::DW_ATOM_type_flags: 117837c110cSSpyridoula Gravani if ((!FormValue.isFormClass(DWARFFormValue::FC_Constant) && 118837c110cSSpyridoula Gravani !FormValue.isFormClass(DWARFFormValue::FC_Flag)) || 119837c110cSSpyridoula Gravani FormValue.getForm() == dwarf::DW_FORM_sdata) 120837c110cSSpyridoula Gravani return false; 1210e6694d1SAdrian Prantl break; 122837c110cSSpyridoula Gravani default: 123837c110cSSpyridoula Gravani break; 124837c110cSSpyridoula Gravani } 125837c110cSSpyridoula Gravani } 126837c110cSSpyridoula Gravani return true; 127837c110cSSpyridoula Gravani } 128837c110cSSpyridoula Gravani 12970d35e10SSpyridoula Gravani std::pair<uint32_t, dwarf::Tag> 1309b36fd25SPavel Labath AppleAcceleratorTable::readAtoms(uint32_t &HashDataOffset) { 131837c110cSSpyridoula Gravani uint32_t DieOffset = dwarf::DW_INVALID_OFFSET; 13270d35e10SSpyridoula Gravani dwarf::Tag DieTag = dwarf::DW_TAG_null; 133e5400f8aSPaul Robinson DWARFFormParams FormParams = {Hdr.Version, 0, dwarf::DwarfFormat::DWARF32}; 134837c110cSSpyridoula Gravani 135837c110cSSpyridoula Gravani for (auto Atom : getAtomsDesc()) { 136837c110cSSpyridoula Gravani DWARFFormValue FormValue(Atom.second); 137e5400f8aSPaul Robinson FormValue.extractValue(AccelSection, &HashDataOffset, FormParams); 138837c110cSSpyridoula Gravani switch (Atom.first) { 139837c110cSSpyridoula Gravani case dwarf::DW_ATOM_die_offset: 140837c110cSSpyridoula Gravani DieOffset = *FormValue.getAsUnsignedConstant(); 141837c110cSSpyridoula Gravani break; 14270d35e10SSpyridoula Gravani case dwarf::DW_ATOM_die_tag: 14370d35e10SSpyridoula Gravani DieTag = (dwarf::Tag)*FormValue.getAsUnsignedConstant(); 14470d35e10SSpyridoula Gravani break; 145837c110cSSpyridoula Gravani default: 146837c110cSSpyridoula Gravani break; 147837c110cSSpyridoula Gravani } 148837c110cSSpyridoula Gravani } 14970d35e10SSpyridoula Gravani return {DieOffset, DieTag}; 150837c110cSSpyridoula Gravani } 151837c110cSSpyridoula Gravani 152394e8056SPavel Labath void AppleAcceleratorTable::Header::dump(ScopedPrinter &W) const { 153394e8056SPavel Labath DictScope HeaderScope(W, "Header"); 154394e8056SPavel Labath W.printHex("Magic", Magic); 155394e8056SPavel Labath W.printHex("Version", Version); 156394e8056SPavel Labath W.printHex("Hash function", HashFunction); 157394e8056SPavel Labath W.printNumber("Bucket count", BucketCount); 158394e8056SPavel Labath W.printNumber("Hashes count", HashCount); 159394e8056SPavel Labath W.printNumber("HeaderData length", HeaderDataLength); 160394e8056SPavel Labath } 161394e8056SPavel Labath 162394e8056SPavel Labath bool AppleAcceleratorTable::dumpName(ScopedPrinter &W, 163394e8056SPavel Labath SmallVectorImpl<DWARFFormValue> &AtomForms, 164394e8056SPavel Labath uint32_t *DataOffset) const { 165394e8056SPavel Labath DWARFFormParams FormParams = {Hdr.Version, 0, dwarf::DwarfFormat::DWARF32}; 166394e8056SPavel Labath uint32_t NameOffset = *DataOffset; 167394e8056SPavel Labath if (!AccelSection.isValidOffsetForDataOfSize(*DataOffset, 4)) { 168394e8056SPavel Labath W.printString("Incorrectly terminated list."); 169394e8056SPavel Labath return false; 170394e8056SPavel Labath } 171394e8056SPavel Labath unsigned StringOffset = AccelSection.getRelocatedValue(4, DataOffset); 172394e8056SPavel Labath if (!StringOffset) 173394e8056SPavel Labath return false; // End of list 174394e8056SPavel Labath 175394e8056SPavel Labath DictScope NameScope(W, ("Name@0x" + Twine::utohexstr(NameOffset)).str()); 176394e8056SPavel Labath W.startLine() << format("String: 0x%08x", StringOffset); 177394e8056SPavel Labath W.getOStream() << " \"" << StringSection.getCStr(&StringOffset) << "\"\n"; 178394e8056SPavel Labath 179394e8056SPavel Labath unsigned NumData = AccelSection.getU32(DataOffset); 180394e8056SPavel Labath for (unsigned Data = 0; Data < NumData; ++Data) { 181394e8056SPavel Labath ListScope DataScope(W, ("Data " + Twine(Data)).str()); 182394e8056SPavel Labath unsigned i = 0; 183394e8056SPavel Labath for (auto &Atom : AtomForms) { 184394e8056SPavel Labath W.startLine() << format("Atom[%d]: ", i++); 185394e8056SPavel Labath if (Atom.extractValue(AccelSection, DataOffset, FormParams)) 186394e8056SPavel Labath Atom.dump(W.getOStream()); 187394e8056SPavel Labath else 188394e8056SPavel Labath W.getOStream() << "Error extracting the value"; 189394e8056SPavel Labath W.getOStream() << "\n"; 190394e8056SPavel Labath } 191394e8056SPavel Labath } 192394e8056SPavel Labath return true; // more entries follow 193394e8056SPavel Labath } 194394e8056SPavel Labath 1959b36fd25SPavel Labath LLVM_DUMP_METHOD void AppleAcceleratorTable::dump(raw_ostream &OS) const { 19699fdb9d9SAdrian Prantl if (!IsValid) 19799fdb9d9SAdrian Prantl return; 19899fdb9d9SAdrian Prantl 199394e8056SPavel Labath ScopedPrinter W(OS); 20082af9438SZachary Turner 201394e8056SPavel Labath Hdr.dump(W); 202394e8056SPavel Labath 203394e8056SPavel Labath W.printNumber("DIE offset base", HdrData.DIEOffsetBase); 2043460957eSPavel Labath W.printNumber("Number of atoms", uint64_t(HdrData.Atoms.size())); 20582af9438SZachary Turner SmallVector<DWARFFormValue, 3> AtomForms; 206394e8056SPavel Labath { 207394e8056SPavel Labath ListScope AtomsScope(W, "Atoms"); 208394e8056SPavel Labath unsigned i = 0; 20982af9438SZachary Turner for (const auto &Atom : HdrData.Atoms) { 210394e8056SPavel Labath DictScope AtomScope(W, ("Atom " + Twine(i++)).str()); 211394e8056SPavel Labath W.startLine() << "Type: " << formatAtom(Atom.first) << '\n'; 212394e8056SPavel Labath W.startLine() << "Form: " << formatForm(Atom.second) << '\n'; 21382af9438SZachary Turner AtomForms.push_back(DWARFFormValue(Atom.second)); 21482af9438SZachary Turner } 215394e8056SPavel Labath } 21682af9438SZachary Turner 21782af9438SZachary Turner // Now go through the actual tables and dump them. 21882af9438SZachary Turner uint32_t Offset = sizeof(Hdr) + Hdr.HeaderDataLength; 219394e8056SPavel Labath unsigned HashesBase = Offset + Hdr.BucketCount * 4; 220394e8056SPavel Labath unsigned OffsetsBase = HashesBase + Hdr.HashCount * 4; 22182af9438SZachary Turner 222394e8056SPavel Labath for (unsigned Bucket = 0; Bucket < Hdr.BucketCount; ++Bucket) { 22382af9438SZachary Turner unsigned Index = AccelSection.getU32(&Offset); 22482af9438SZachary Turner 225394e8056SPavel Labath ListScope BucketScope(W, ("Bucket " + Twine(Bucket)).str()); 22682af9438SZachary Turner if (Index == UINT32_MAX) { 227394e8056SPavel Labath W.printString("EMPTY"); 22882af9438SZachary Turner continue; 22982af9438SZachary Turner } 23082af9438SZachary Turner 231394e8056SPavel Labath for (unsigned HashIdx = Index; HashIdx < Hdr.HashCount; ++HashIdx) { 23282af9438SZachary Turner unsigned HashOffset = HashesBase + HashIdx*4; 23382af9438SZachary Turner unsigned OffsetsOffset = OffsetsBase + HashIdx*4; 23482af9438SZachary Turner uint32_t Hash = AccelSection.getU32(&HashOffset); 23582af9438SZachary Turner 236394e8056SPavel Labath if (Hash % Hdr.BucketCount != Bucket) 23782af9438SZachary Turner break; 23882af9438SZachary Turner 23982af9438SZachary Turner unsigned DataOffset = AccelSection.getU32(&OffsetsOffset); 240394e8056SPavel Labath ListScope HashScope(W, ("Hash 0x" + Twine::utohexstr(Hash)).str()); 24182af9438SZachary Turner if (!AccelSection.isValidOffset(DataOffset)) { 242394e8056SPavel Labath W.printString("Invalid section offset"); 24382af9438SZachary Turner continue; 24482af9438SZachary Turner } 245394e8056SPavel Labath while (dumpName(W, AtomForms, &DataOffset)) 246394e8056SPavel Labath /*empty*/; 24782af9438SZachary Turner } 24882af9438SZachary Turner } 24982af9438SZachary Turner } 25099fdb9d9SAdrian Prantl 251*d99072bcSPavel Labath AppleAcceleratorTable::Entry::Entry( 252*d99072bcSPavel Labath const AppleAcceleratorTable::HeaderData &HdrData) 253*d99072bcSPavel Labath : HdrData(&HdrData) { 254*d99072bcSPavel Labath Values.reserve(HdrData.Atoms.size()); 255*d99072bcSPavel Labath for (const auto &Atom : HdrData.Atoms) 256*d99072bcSPavel Labath Values.push_back(DWARFFormValue(Atom.second)); 257*d99072bcSPavel Labath } 258*d99072bcSPavel Labath 259*d99072bcSPavel Labath void AppleAcceleratorTable::Entry::extract( 260*d99072bcSPavel Labath const AppleAcceleratorTable &AccelTable, uint32_t *Offset) { 261*d99072bcSPavel Labath 262*d99072bcSPavel Labath DWARFFormParams FormParams = {AccelTable.Hdr.Version, 0, 263*d99072bcSPavel Labath dwarf::DwarfFormat::DWARF32}; 264*d99072bcSPavel Labath for (auto &Atom : Values) 265*d99072bcSPavel Labath Atom.extractValue(AccelTable.AccelSection, Offset, FormParams); 266*d99072bcSPavel Labath } 267*d99072bcSPavel Labath 268*d99072bcSPavel Labath Optional<DWARFFormValue> 269*d99072bcSPavel Labath AppleAcceleratorTable::Entry::lookup(HeaderData::AtomType Atom) const { 270*d99072bcSPavel Labath assert(HdrData && "Dereferencing end iterator?"); 271*d99072bcSPavel Labath assert(HdrData->Atoms.size() == Values.size()); 272*d99072bcSPavel Labath for (const auto &Tuple : zip_first(HdrData->Atoms, Values)) { 273*d99072bcSPavel Labath if (std::get<0>(Tuple).first == Atom) 274*d99072bcSPavel Labath return std::get<1>(Tuple); 275*d99072bcSPavel Labath } 276*d99072bcSPavel Labath return None; 277*d99072bcSPavel Labath } 278*d99072bcSPavel Labath 279*d99072bcSPavel Labath Optional<uint64_t> AppleAcceleratorTable::Entry::getDIEOffset() const { 280*d99072bcSPavel Labath if (Optional<DWARFFormValue> Off = lookup(dwarf::DW_ATOM_die_offset)) 281*d99072bcSPavel Labath return Off->getAsSectionOffset(); 282*d99072bcSPavel Labath return None; 283*d99072bcSPavel Labath } 284*d99072bcSPavel Labath 285*d99072bcSPavel Labath Optional<uint64_t> AppleAcceleratorTable::Entry::getCUOffset() const { 286*d99072bcSPavel Labath if (Optional<DWARFFormValue> Off = lookup(dwarf::DW_ATOM_cu_offset)) 287*d99072bcSPavel Labath return Off->getAsSectionOffset(); 288*d99072bcSPavel Labath return None; 289*d99072bcSPavel Labath } 290*d99072bcSPavel Labath 291*d99072bcSPavel Labath Optional<dwarf::Tag> AppleAcceleratorTable::Entry::getTag() const { 292*d99072bcSPavel Labath Optional<DWARFFormValue> Tag = lookup(dwarf::DW_ATOM_die_tag); 293*d99072bcSPavel Labath if (!Tag) 294*d99072bcSPavel Labath return None; 295*d99072bcSPavel Labath if (Optional<uint64_t> Value = Tag->getAsUnsignedConstant()) 296*d99072bcSPavel Labath return dwarf::Tag(*Value); 297*d99072bcSPavel Labath return None; 298*d99072bcSPavel Labath } 299*d99072bcSPavel Labath 3009b36fd25SPavel Labath AppleAcceleratorTable::ValueIterator::ValueIterator( 3019b36fd25SPavel Labath const AppleAcceleratorTable &AccelTable, unsigned Offset) 302*d99072bcSPavel Labath : AccelTable(&AccelTable), Current(AccelTable.HdrData), DataOffset(Offset) { 30399fdb9d9SAdrian Prantl if (!AccelTable.AccelSection.isValidOffsetForDataOfSize(DataOffset, 4)) 30499fdb9d9SAdrian Prantl return; 30599fdb9d9SAdrian Prantl 30699fdb9d9SAdrian Prantl // Read the first entry. 30799fdb9d9SAdrian Prantl NumData = AccelTable.AccelSection.getU32(&DataOffset); 30899fdb9d9SAdrian Prantl Next(); 30999fdb9d9SAdrian Prantl } 31099fdb9d9SAdrian Prantl 3119b36fd25SPavel Labath void AppleAcceleratorTable::ValueIterator::Next() { 31299fdb9d9SAdrian Prantl assert(NumData > 0 && "attempted to increment iterator past the end"); 31399fdb9d9SAdrian Prantl auto &AccelSection = AccelTable->AccelSection; 31499fdb9d9SAdrian Prantl if (Data >= NumData || 31599fdb9d9SAdrian Prantl !AccelSection.isValidOffsetForDataOfSize(DataOffset, 4)) { 31699fdb9d9SAdrian Prantl NumData = 0; 31799fdb9d9SAdrian Prantl return; 31899fdb9d9SAdrian Prantl } 319*d99072bcSPavel Labath Current.extract(*AccelTable, &DataOffset); 32099fdb9d9SAdrian Prantl ++Data; 32199fdb9d9SAdrian Prantl } 32299fdb9d9SAdrian Prantl 3239b36fd25SPavel Labath iterator_range<AppleAcceleratorTable::ValueIterator> 3249b36fd25SPavel Labath AppleAcceleratorTable::equal_range(StringRef Key) const { 32599fdb9d9SAdrian Prantl if (!IsValid) 32699fdb9d9SAdrian Prantl return make_range(ValueIterator(), ValueIterator()); 32799fdb9d9SAdrian Prantl 32899fdb9d9SAdrian Prantl // Find the bucket. 32992ac9d3eSJonas Devlieghere unsigned HashValue = djbHash(Key); 330394e8056SPavel Labath unsigned Bucket = HashValue % Hdr.BucketCount; 33199fdb9d9SAdrian Prantl unsigned BucketBase = sizeof(Hdr) + Hdr.HeaderDataLength; 332394e8056SPavel Labath unsigned HashesBase = BucketBase + Hdr.BucketCount * 4; 333394e8056SPavel Labath unsigned OffsetsBase = HashesBase + Hdr.HashCount * 4; 33499fdb9d9SAdrian Prantl 33599fdb9d9SAdrian Prantl unsigned BucketOffset = BucketBase + Bucket * 4; 33699fdb9d9SAdrian Prantl unsigned Index = AccelSection.getU32(&BucketOffset); 33799fdb9d9SAdrian Prantl 33899fdb9d9SAdrian Prantl // Search through all hashes in the bucket. 339394e8056SPavel Labath for (unsigned HashIdx = Index; HashIdx < Hdr.HashCount; ++HashIdx) { 34099fdb9d9SAdrian Prantl unsigned HashOffset = HashesBase + HashIdx * 4; 34199fdb9d9SAdrian Prantl unsigned OffsetsOffset = OffsetsBase + HashIdx * 4; 34299fdb9d9SAdrian Prantl uint32_t Hash = AccelSection.getU32(&HashOffset); 34399fdb9d9SAdrian Prantl 344394e8056SPavel Labath if (Hash % Hdr.BucketCount != Bucket) 34599fdb9d9SAdrian Prantl // We are already in the next bucket. 34699fdb9d9SAdrian Prantl break; 34799fdb9d9SAdrian Prantl 34899fdb9d9SAdrian Prantl unsigned DataOffset = AccelSection.getU32(&OffsetsOffset); 34999fdb9d9SAdrian Prantl unsigned StringOffset = AccelSection.getRelocatedValue(4, &DataOffset); 35099fdb9d9SAdrian Prantl if (!StringOffset) 35199fdb9d9SAdrian Prantl break; 35299fdb9d9SAdrian Prantl 35399fdb9d9SAdrian Prantl // Finally, compare the key. 35499fdb9d9SAdrian Prantl if (Key == StringSection.getCStr(&StringOffset)) 35599fdb9d9SAdrian Prantl return make_range({*this, DataOffset}, ValueIterator()); 35699fdb9d9SAdrian Prantl } 35799fdb9d9SAdrian Prantl return make_range(ValueIterator(), ValueIterator()); 35899fdb9d9SAdrian Prantl } 3593c9a918cSPavel Labath 3603c9a918cSPavel Labath void DWARFDebugNames::Header::dump(ScopedPrinter &W) const { 3613c9a918cSPavel Labath DictScope HeaderScope(W, "Header"); 3623c9a918cSPavel Labath W.printHex("Length", UnitLength); 3633c9a918cSPavel Labath W.printNumber("Version", Version); 3643c9a918cSPavel Labath W.printHex("Padding", Padding); 3653c9a918cSPavel Labath W.printNumber("CU count", CompUnitCount); 3663c9a918cSPavel Labath W.printNumber("Local TU count", LocalTypeUnitCount); 3673c9a918cSPavel Labath W.printNumber("Foreign TU count", ForeignTypeUnitCount); 3683c9a918cSPavel Labath W.printNumber("Bucket count", BucketCount); 3693c9a918cSPavel Labath W.printNumber("Name count", NameCount); 3703c9a918cSPavel Labath W.printHex("Abbreviations table size", AbbrevTableSize); 3713c9a918cSPavel Labath W.startLine() << "Augmentation: '" << AugmentationString << "'\n"; 3723c9a918cSPavel Labath } 3733c9a918cSPavel Labath 3743c9a918cSPavel Labath llvm::Error DWARFDebugNames::Header::extract(const DWARFDataExtractor &AS, 3753c9a918cSPavel Labath uint32_t *Offset) { 3763c9a918cSPavel Labath // Check that we can read the fixed-size part. 377e7264106SPavel Labath if (!AS.isValidOffset(*Offset + sizeof(HeaderPOD) - 1)) 3783c9a918cSPavel Labath return make_error<StringError>("Section too small: cannot read header.", 3793c9a918cSPavel Labath inconvertibleErrorCode()); 3803c9a918cSPavel Labath 3813c9a918cSPavel Labath UnitLength = AS.getU32(Offset); 3823c9a918cSPavel Labath Version = AS.getU16(Offset); 3833c9a918cSPavel Labath Padding = AS.getU16(Offset); 3843c9a918cSPavel Labath CompUnitCount = AS.getU32(Offset); 3853c9a918cSPavel Labath LocalTypeUnitCount = AS.getU32(Offset); 3863c9a918cSPavel Labath ForeignTypeUnitCount = AS.getU32(Offset); 3873c9a918cSPavel Labath BucketCount = AS.getU32(Offset); 3883c9a918cSPavel Labath NameCount = AS.getU32(Offset); 3893c9a918cSPavel Labath AbbrevTableSize = AS.getU32(Offset); 3903c9a918cSPavel Labath AugmentationStringSize = AS.getU32(Offset); 3913c9a918cSPavel Labath 3923c9a918cSPavel Labath if (!AS.isValidOffsetForDataOfSize(*Offset, AugmentationStringSize)) 3933c9a918cSPavel Labath return make_error<StringError>( 3943c9a918cSPavel Labath "Section too small: cannot read header augmentation.", 3953c9a918cSPavel Labath inconvertibleErrorCode()); 3963c9a918cSPavel Labath AugmentationString.resize(AugmentationStringSize); 3973c9a918cSPavel Labath AS.getU8(Offset, reinterpret_cast<uint8_t *>(AugmentationString.data()), 3983c9a918cSPavel Labath AugmentationStringSize); 3993c9a918cSPavel Labath *Offset = alignTo(*Offset, 4); 4003c9a918cSPavel Labath return Error::success(); 4013c9a918cSPavel Labath } 4023c9a918cSPavel Labath 4033c9a918cSPavel Labath void DWARFDebugNames::Abbrev::dump(ScopedPrinter &W) const { 4043c9a918cSPavel Labath DictScope AbbrevScope(W, ("Abbreviation 0x" + Twine::utohexstr(Code)).str()); 4053c9a918cSPavel Labath W.startLine() << "Tag: " << formatTag(Tag) << '\n'; 4063c9a918cSPavel Labath 4073c9a918cSPavel Labath for (const auto &Attr : Attributes) { 4083c9a918cSPavel Labath W.startLine() << formatIndex(Attr.Index) << ": " << formatForm(Attr.Form) 4093c9a918cSPavel Labath << '\n'; 4103c9a918cSPavel Labath } 4113c9a918cSPavel Labath } 4123c9a918cSPavel Labath 4133c9a918cSPavel Labath static constexpr DWARFDebugNames::AttributeEncoding sentinelAttrEnc() { 4143c9a918cSPavel Labath return {dwarf::Index(0), dwarf::Form(0)}; 4153c9a918cSPavel Labath } 4163c9a918cSPavel Labath 4173c9a918cSPavel Labath static bool isSentinel(const DWARFDebugNames::AttributeEncoding &AE) { 4183c9a918cSPavel Labath return AE == sentinelAttrEnc(); 4193c9a918cSPavel Labath } 4203c9a918cSPavel Labath 4213c9a918cSPavel Labath static DWARFDebugNames::Abbrev sentinelAbbrev() { 4223c9a918cSPavel Labath return DWARFDebugNames::Abbrev(0, dwarf::Tag(0), {}); 4233c9a918cSPavel Labath } 4243c9a918cSPavel Labath 4253c9a918cSPavel Labath static bool isSentinel(const DWARFDebugNames::Abbrev &Abbr) { 4263c9a918cSPavel Labath return Abbr.Code == 0; 4273c9a918cSPavel Labath } 4283c9a918cSPavel Labath 4293c9a918cSPavel Labath DWARFDebugNames::Abbrev DWARFDebugNames::AbbrevMapInfo::getEmptyKey() { 4303c9a918cSPavel Labath return sentinelAbbrev(); 4313c9a918cSPavel Labath } 4323c9a918cSPavel Labath 4333c9a918cSPavel Labath DWARFDebugNames::Abbrev DWARFDebugNames::AbbrevMapInfo::getTombstoneKey() { 4343c9a918cSPavel Labath return DWARFDebugNames::Abbrev(~0, dwarf::Tag(0), {}); 4353c9a918cSPavel Labath } 4363c9a918cSPavel Labath 4373c9a918cSPavel Labath Expected<DWARFDebugNames::AttributeEncoding> 4383c9a918cSPavel Labath DWARFDebugNames::NameIndex::extractAttributeEncoding(uint32_t *Offset) { 4393c9a918cSPavel Labath if (*Offset >= EntriesBase) { 4403c9a918cSPavel Labath return make_error<StringError>("Incorrectly terminated abbreviation table.", 4413c9a918cSPavel Labath inconvertibleErrorCode()); 4423c9a918cSPavel Labath } 4433c9a918cSPavel Labath 4443c9a918cSPavel Labath uint32_t Index = Section.AccelSection.getULEB128(Offset); 4453c9a918cSPavel Labath uint32_t Form = Section.AccelSection.getULEB128(Offset); 4463c9a918cSPavel Labath return AttributeEncoding(dwarf::Index(Index), dwarf::Form(Form)); 4473c9a918cSPavel Labath } 4483c9a918cSPavel Labath 4493c9a918cSPavel Labath Expected<std::vector<DWARFDebugNames::AttributeEncoding>> 4503c9a918cSPavel Labath DWARFDebugNames::NameIndex::extractAttributeEncodings(uint32_t *Offset) { 4513c9a918cSPavel Labath std::vector<AttributeEncoding> Result; 4523c9a918cSPavel Labath for (;;) { 4533c9a918cSPavel Labath auto AttrEncOr = extractAttributeEncoding(Offset); 4543c9a918cSPavel Labath if (!AttrEncOr) 4553c9a918cSPavel Labath return AttrEncOr.takeError(); 4563c9a918cSPavel Labath if (isSentinel(*AttrEncOr)) 4573c9a918cSPavel Labath return std::move(Result); 4583c9a918cSPavel Labath 4593c9a918cSPavel Labath Result.emplace_back(*AttrEncOr); 4603c9a918cSPavel Labath } 4613c9a918cSPavel Labath } 4623c9a918cSPavel Labath 4633c9a918cSPavel Labath Expected<DWARFDebugNames::Abbrev> 4643c9a918cSPavel Labath DWARFDebugNames::NameIndex::extractAbbrev(uint32_t *Offset) { 4653c9a918cSPavel Labath if (*Offset >= EntriesBase) { 4663c9a918cSPavel Labath return make_error<StringError>("Incorrectly terminated abbreviation table.", 4673c9a918cSPavel Labath inconvertibleErrorCode()); 4683c9a918cSPavel Labath } 4693c9a918cSPavel Labath 4703c9a918cSPavel Labath uint32_t Code = Section.AccelSection.getULEB128(Offset); 4713c9a918cSPavel Labath if (Code == 0) 4723c9a918cSPavel Labath return sentinelAbbrev(); 4733c9a918cSPavel Labath 4743c9a918cSPavel Labath uint32_t Tag = Section.AccelSection.getULEB128(Offset); 4753c9a918cSPavel Labath auto AttrEncOr = extractAttributeEncodings(Offset); 4763c9a918cSPavel Labath if (!AttrEncOr) 4773c9a918cSPavel Labath return AttrEncOr.takeError(); 4783c9a918cSPavel Labath return Abbrev(Code, dwarf::Tag(Tag), std::move(*AttrEncOr)); 4793c9a918cSPavel Labath } 4803c9a918cSPavel Labath 4813c9a918cSPavel Labath Error DWARFDebugNames::NameIndex::extract() { 4823c9a918cSPavel Labath const DWARFDataExtractor &AS = Section.AccelSection; 4833c9a918cSPavel Labath uint32_t Offset = Base; 4843c9a918cSPavel Labath if (Error E = Hdr.extract(AS, &Offset)) 4853c9a918cSPavel Labath return E; 4863c9a918cSPavel Labath 4873c9a918cSPavel Labath CUsBase = Offset; 4883c9a918cSPavel Labath Offset += Hdr.CompUnitCount * 4; 4893c9a918cSPavel Labath Offset += Hdr.LocalTypeUnitCount * 4; 4903c9a918cSPavel Labath Offset += Hdr.ForeignTypeUnitCount * 8; 4913c9a918cSPavel Labath BucketsBase = Offset; 4923c9a918cSPavel Labath Offset += Hdr.BucketCount * 4; 4933c9a918cSPavel Labath HashesBase = Offset; 4943c9a918cSPavel Labath if (Hdr.BucketCount > 0) 4953c9a918cSPavel Labath Offset += Hdr.NameCount * 4; 4963c9a918cSPavel Labath StringOffsetsBase = Offset; 4973c9a918cSPavel Labath Offset += Hdr.NameCount * 4; 4983c9a918cSPavel Labath EntryOffsetsBase = Offset; 4993c9a918cSPavel Labath Offset += Hdr.NameCount * 4; 5003c9a918cSPavel Labath 5013c9a918cSPavel Labath if (!AS.isValidOffsetForDataOfSize(Offset, Hdr.AbbrevTableSize)) 5023c9a918cSPavel Labath return make_error<StringError>( 5033c9a918cSPavel Labath "Section too small: cannot read abbreviations.", 5043c9a918cSPavel Labath inconvertibleErrorCode()); 5053c9a918cSPavel Labath 5063c9a918cSPavel Labath EntriesBase = Offset + Hdr.AbbrevTableSize; 5073c9a918cSPavel Labath 5083c9a918cSPavel Labath for (;;) { 5093c9a918cSPavel Labath auto AbbrevOr = extractAbbrev(&Offset); 5103c9a918cSPavel Labath if (!AbbrevOr) 5113c9a918cSPavel Labath return AbbrevOr.takeError(); 5123c9a918cSPavel Labath if (isSentinel(*AbbrevOr)) 5133c9a918cSPavel Labath return Error::success(); 5143c9a918cSPavel Labath 5153c9a918cSPavel Labath if (!Abbrevs.insert(std::move(*AbbrevOr)).second) { 5163c9a918cSPavel Labath return make_error<StringError>("Duplicate abbreviation code.", 5173c9a918cSPavel Labath inconvertibleErrorCode()); 5183c9a918cSPavel Labath } 5193c9a918cSPavel Labath } 5203c9a918cSPavel Labath } 521*d99072bcSPavel Labath DWARFDebugNames::Entry::Entry(const NameIndex &NameIdx, const Abbrev &Abbr) 522*d99072bcSPavel Labath : NameIdx(&NameIdx), Abbr(&Abbr) { 5233c9a918cSPavel Labath // This merely creates form values. It is up to the caller 5243c9a918cSPavel Labath // (NameIndex::getEntry) to populate them. 5253c9a918cSPavel Labath Values.reserve(Abbr.Attributes.size()); 5263c9a918cSPavel Labath for (const auto &Attr : Abbr.Attributes) 5273c9a918cSPavel Labath Values.emplace_back(Attr.Form); 5283c9a918cSPavel Labath } 5293c9a918cSPavel Labath 530*d99072bcSPavel Labath Optional<DWARFFormValue> 531*d99072bcSPavel Labath DWARFDebugNames::Entry::lookup(dwarf::Index Index) const { 532*d99072bcSPavel Labath assert(Abbr->Attributes.size() == Values.size()); 533*d99072bcSPavel Labath for (const auto &Tuple : zip_first(Abbr->Attributes, Values)) { 534*d99072bcSPavel Labath if (std::get<0>(Tuple).Index == Index) 535*d99072bcSPavel Labath return std::get<1>(Tuple); 536*d99072bcSPavel Labath } 537*d99072bcSPavel Labath return None; 538*d99072bcSPavel Labath } 5393c9a918cSPavel Labath 540*d99072bcSPavel Labath Optional<uint64_t> DWARFDebugNames::Entry::getDIEOffset() const { 541*d99072bcSPavel Labath if (Optional<DWARFFormValue> Off = lookup(dwarf::DW_IDX_die_offset)) 542*d99072bcSPavel Labath return Off->getAsSectionOffset(); 543*d99072bcSPavel Labath return None; 544*d99072bcSPavel Labath } 545*d99072bcSPavel Labath 546*d99072bcSPavel Labath Optional<uint64_t> DWARFDebugNames::Entry::getCUIndex() const { 547*d99072bcSPavel Labath if (Optional<DWARFFormValue> Off = lookup(dwarf::DW_IDX_compile_unit)) 548*d99072bcSPavel Labath return Off->getAsUnsignedConstant(); 549*d99072bcSPavel Labath return None; 550*d99072bcSPavel Labath } 551*d99072bcSPavel Labath 552*d99072bcSPavel Labath Optional<uint64_t> DWARFDebugNames::Entry::getCUOffset() const { 553*d99072bcSPavel Labath Optional<uint64_t> Index = getCUIndex(); 554*d99072bcSPavel Labath if (!Index || *Index >= NameIdx->getCUCount()) 555*d99072bcSPavel Labath return None; 556*d99072bcSPavel Labath return NameIdx->getCUOffset(*Index); 557*d99072bcSPavel Labath } 558*d99072bcSPavel Labath 559*d99072bcSPavel Labath void DWARFDebugNames::Entry::dump(ScopedPrinter &W) const { 560*d99072bcSPavel Labath W.printHex("Abbrev", Abbr->Code); 561*d99072bcSPavel Labath W.startLine() << "Tag: " << formatTag(Abbr->Tag) << "\n"; 562*d99072bcSPavel Labath 563*d99072bcSPavel Labath assert(Abbr->Attributes.size() == Values.size()); 564*d99072bcSPavel Labath for (const auto &Tuple : zip_first(Abbr->Attributes, Values)) { 565*d99072bcSPavel Labath W.startLine() << formatIndex(std::get<0>(Tuple).Index) << ": "; 566*d99072bcSPavel Labath std::get<1>(Tuple).dump(W.getOStream()); 5673c9a918cSPavel Labath W.getOStream() << '\n'; 5683c9a918cSPavel Labath } 5693c9a918cSPavel Labath } 5703c9a918cSPavel Labath 5713c9a918cSPavel Labath char DWARFDebugNames::SentinelError::ID; 5723c9a918cSPavel Labath std::error_code DWARFDebugNames::SentinelError::convertToErrorCode() const { 5733c9a918cSPavel Labath return inconvertibleErrorCode(); 5743c9a918cSPavel Labath } 5753c9a918cSPavel Labath 5763c9a918cSPavel Labath uint32_t DWARFDebugNames::NameIndex::getCUOffset(uint32_t CU) const { 5773c9a918cSPavel Labath assert(CU < Hdr.CompUnitCount); 5783c9a918cSPavel Labath uint32_t Offset = CUsBase + 4 * CU; 5793c9a918cSPavel Labath return Section.AccelSection.getRelocatedValue(4, &Offset); 5803c9a918cSPavel Labath } 5813c9a918cSPavel Labath 5823c9a918cSPavel Labath uint32_t DWARFDebugNames::NameIndex::getLocalTUOffset(uint32_t TU) const { 5833c9a918cSPavel Labath assert(TU < Hdr.LocalTypeUnitCount); 5843c9a918cSPavel Labath uint32_t Offset = CUsBase + Hdr.CompUnitCount * 4; 5853c9a918cSPavel Labath return Section.AccelSection.getRelocatedValue(4, &Offset); 5863c9a918cSPavel Labath } 5873c9a918cSPavel Labath 588*d99072bcSPavel Labath uint64_t DWARFDebugNames::NameIndex::getForeignTUSignature(uint32_t TU) const { 5893c9a918cSPavel Labath assert(TU < Hdr.ForeignTypeUnitCount); 5903c9a918cSPavel Labath uint32_t Offset = CUsBase + (Hdr.CompUnitCount + Hdr.LocalTypeUnitCount) * 4; 5913c9a918cSPavel Labath return Section.AccelSection.getU64(&Offset); 5923c9a918cSPavel Labath } 5933c9a918cSPavel Labath 5943c9a918cSPavel Labath Expected<DWARFDebugNames::Entry> 5953c9a918cSPavel Labath DWARFDebugNames::NameIndex::getEntry(uint32_t *Offset) const { 5963c9a918cSPavel Labath const DWARFDataExtractor &AS = Section.AccelSection; 5973c9a918cSPavel Labath if (!AS.isValidOffset(*Offset)) 5983c9a918cSPavel Labath return make_error<StringError>("Incorrectly terminated entry list", 5993c9a918cSPavel Labath inconvertibleErrorCode()); 6003c9a918cSPavel Labath 6013c9a918cSPavel Labath uint32_t AbbrevCode = AS.getULEB128(Offset); 6023c9a918cSPavel Labath if (AbbrevCode == 0) 6033c9a918cSPavel Labath return make_error<SentinelError>(); 6043c9a918cSPavel Labath 6053c9a918cSPavel Labath const auto AbbrevIt = Abbrevs.find_as(AbbrevCode); 6063c9a918cSPavel Labath if (AbbrevIt == Abbrevs.end()) 6073c9a918cSPavel Labath return make_error<StringError>("Invalid abbreviation", 6083c9a918cSPavel Labath inconvertibleErrorCode()); 6093c9a918cSPavel Labath 610*d99072bcSPavel Labath Entry E(*this, *AbbrevIt); 6113c9a918cSPavel Labath 6123c9a918cSPavel Labath DWARFFormParams FormParams = {Hdr.Version, 0, dwarf::DwarfFormat::DWARF32}; 6133c9a918cSPavel Labath for (auto &Value : E.Values) { 6143c9a918cSPavel Labath if (!Value.extractValue(AS, Offset, FormParams)) 6153c9a918cSPavel Labath return make_error<StringError>("Error extracting index attribute values", 6163c9a918cSPavel Labath inconvertibleErrorCode()); 6173c9a918cSPavel Labath } 6183c9a918cSPavel Labath return std::move(E); 6193c9a918cSPavel Labath } 6203c9a918cSPavel Labath 6213c9a918cSPavel Labath DWARFDebugNames::NameTableEntry 6223c9a918cSPavel Labath DWARFDebugNames::NameIndex::getNameTableEntry(uint32_t Index) const { 6233c9a918cSPavel Labath assert(0 < Index && Index <= Hdr.NameCount); 6243c9a918cSPavel Labath uint32_t StringOffsetOffset = StringOffsetsBase + 4 * (Index - 1); 6253c9a918cSPavel Labath uint32_t EntryOffsetOffset = EntryOffsetsBase + 4 * (Index - 1); 6263c9a918cSPavel Labath const DWARFDataExtractor &AS = Section.AccelSection; 6273c9a918cSPavel Labath 6283c9a918cSPavel Labath uint32_t StringOffset = AS.getRelocatedValue(4, &StringOffsetOffset); 6293c9a918cSPavel Labath uint32_t EntryOffset = AS.getU32(&EntryOffsetOffset); 6303c9a918cSPavel Labath EntryOffset += EntriesBase; 6313c9a918cSPavel Labath return {StringOffset, EntryOffset}; 6323c9a918cSPavel Labath } 6333c9a918cSPavel Labath 6343c9a918cSPavel Labath uint32_t 6353c9a918cSPavel Labath DWARFDebugNames::NameIndex::getBucketArrayEntry(uint32_t Bucket) const { 6363c9a918cSPavel Labath assert(Bucket < Hdr.BucketCount); 6373c9a918cSPavel Labath uint32_t BucketOffset = BucketsBase + 4 * Bucket; 6383c9a918cSPavel Labath return Section.AccelSection.getU32(&BucketOffset); 6393c9a918cSPavel Labath } 6403c9a918cSPavel Labath 6413c9a918cSPavel Labath uint32_t DWARFDebugNames::NameIndex::getHashArrayEntry(uint32_t Index) const { 6423c9a918cSPavel Labath assert(0 < Index && Index <= Hdr.NameCount); 6433c9a918cSPavel Labath uint32_t HashOffset = HashesBase + 4 * (Index - 1); 6443c9a918cSPavel Labath return Section.AccelSection.getU32(&HashOffset); 6453c9a918cSPavel Labath } 6463c9a918cSPavel Labath 6473c9a918cSPavel Labath // Returns true if we should continue scanning for entries, false if this is the 6483c9a918cSPavel Labath // last (sentinel) entry). In case of a parsing error we also return false, as 6493c9a918cSPavel Labath // it's not possible to recover this entry list (but the other lists may still 6503c9a918cSPavel Labath // parse OK). 6513c9a918cSPavel Labath bool DWARFDebugNames::NameIndex::dumpEntry(ScopedPrinter &W, 6523c9a918cSPavel Labath uint32_t *Offset) const { 6533c9a918cSPavel Labath uint32_t EntryId = *Offset; 6543c9a918cSPavel Labath auto EntryOr = getEntry(Offset); 6553c9a918cSPavel Labath if (!EntryOr) { 6563c9a918cSPavel Labath handleAllErrors(EntryOr.takeError(), [](const SentinelError &) {}, 6573c9a918cSPavel Labath [&W](const ErrorInfoBase &EI) { EI.log(W.startLine()); }); 6583c9a918cSPavel Labath return false; 6593c9a918cSPavel Labath } 6603c9a918cSPavel Labath 6613c9a918cSPavel Labath DictScope EntryScope(W, ("Entry @ 0x" + Twine::utohexstr(EntryId)).str()); 6623c9a918cSPavel Labath EntryOr->dump(W); 6633c9a918cSPavel Labath return true; 6643c9a918cSPavel Labath } 6653c9a918cSPavel Labath 6663c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpName(ScopedPrinter &W, uint32_t Index, 6673c9a918cSPavel Labath Optional<uint32_t> Hash) const { 6683c9a918cSPavel Labath const DataExtractor &SS = Section.StringSection; 6693c9a918cSPavel Labath NameTableEntry NTE = getNameTableEntry(Index); 6703c9a918cSPavel Labath 6713c9a918cSPavel Labath DictScope NameScope(W, ("Name " + Twine(Index)).str()); 6723c9a918cSPavel Labath if (Hash) 6733c9a918cSPavel Labath W.printHex("Hash", *Hash); 6743c9a918cSPavel Labath 6753c9a918cSPavel Labath W.startLine() << format("String: 0x%08x", NTE.StringOffset); 6763c9a918cSPavel Labath W.getOStream() << " \"" << SS.getCStr(&NTE.StringOffset) << "\"\n"; 6773c9a918cSPavel Labath 6783c9a918cSPavel Labath while (dumpEntry(W, &NTE.EntryOffset)) 6793c9a918cSPavel Labath /*empty*/; 6803c9a918cSPavel Labath } 6813c9a918cSPavel Labath 6823c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpCUs(ScopedPrinter &W) const { 6833c9a918cSPavel Labath ListScope CUScope(W, "Compilation Unit offsets"); 6843c9a918cSPavel Labath for (uint32_t CU = 0; CU < Hdr.CompUnitCount; ++CU) 6853c9a918cSPavel Labath W.startLine() << format("CU[%u]: 0x%08x\n", CU, getCUOffset(CU)); 6863c9a918cSPavel Labath } 6873c9a918cSPavel Labath 6883c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpLocalTUs(ScopedPrinter &W) const { 6893c9a918cSPavel Labath if (Hdr.LocalTypeUnitCount == 0) 6903c9a918cSPavel Labath return; 6913c9a918cSPavel Labath 6923c9a918cSPavel Labath ListScope TUScope(W, "Local Type Unit offsets"); 6933c9a918cSPavel Labath for (uint32_t TU = 0; TU < Hdr.LocalTypeUnitCount; ++TU) 6943c9a918cSPavel Labath W.startLine() << format("LocalTU[%u]: 0x%08x\n", TU, getLocalTUOffset(TU)); 6953c9a918cSPavel Labath } 6963c9a918cSPavel Labath 6973c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpForeignTUs(ScopedPrinter &W) const { 6983c9a918cSPavel Labath if (Hdr.ForeignTypeUnitCount == 0) 6993c9a918cSPavel Labath return; 7003c9a918cSPavel Labath 7013c9a918cSPavel Labath ListScope TUScope(W, "Foreign Type Unit signatures"); 7023c9a918cSPavel Labath for (uint32_t TU = 0; TU < Hdr.ForeignTypeUnitCount; ++TU) { 7033c9a918cSPavel Labath W.startLine() << format("ForeignTU[%u]: 0x%016" PRIx64 "\n", TU, 704*d99072bcSPavel Labath getForeignTUSignature(TU)); 7053c9a918cSPavel Labath } 7063c9a918cSPavel Labath } 7073c9a918cSPavel Labath 7083c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpAbbreviations(ScopedPrinter &W) const { 7093c9a918cSPavel Labath ListScope AbbrevsScope(W, "Abbreviations"); 7103c9a918cSPavel Labath for (const auto &Abbr : Abbrevs) 7113c9a918cSPavel Labath Abbr.dump(W); 7123c9a918cSPavel Labath } 7133c9a918cSPavel Labath 7143c9a918cSPavel Labath void DWARFDebugNames::NameIndex::dumpBucket(ScopedPrinter &W, 7153c9a918cSPavel Labath uint32_t Bucket) const { 7163c9a918cSPavel Labath ListScope BucketScope(W, ("Bucket " + Twine(Bucket)).str()); 7173c9a918cSPavel Labath uint32_t Index = getBucketArrayEntry(Bucket); 7183c9a918cSPavel Labath if (Index == 0) { 7193c9a918cSPavel Labath W.printString("EMPTY"); 7203c9a918cSPavel Labath return; 7213c9a918cSPavel Labath } 7223c9a918cSPavel Labath if (Index > Hdr.NameCount) { 7233c9a918cSPavel Labath W.printString("Name index is invalid"); 7243c9a918cSPavel Labath return; 7253c9a918cSPavel Labath } 7263c9a918cSPavel Labath 7273c9a918cSPavel Labath for (; Index <= Hdr.NameCount; ++Index) { 7283c9a918cSPavel Labath uint32_t Hash = getHashArrayEntry(Index); 7293c9a918cSPavel Labath if (Hash % Hdr.BucketCount != Bucket) 7303c9a918cSPavel Labath break; 7313c9a918cSPavel Labath 7323c9a918cSPavel Labath dumpName(W, Index, Hash); 7333c9a918cSPavel Labath } 7343c9a918cSPavel Labath } 7353c9a918cSPavel Labath 7363c9a918cSPavel Labath LLVM_DUMP_METHOD void DWARFDebugNames::NameIndex::dump(ScopedPrinter &W) const { 7373c9a918cSPavel Labath DictScope UnitScope(W, ("Name Index @ 0x" + Twine::utohexstr(Base)).str()); 7383c9a918cSPavel Labath Hdr.dump(W); 7393c9a918cSPavel Labath dumpCUs(W); 7403c9a918cSPavel Labath dumpLocalTUs(W); 7413c9a918cSPavel Labath dumpForeignTUs(W); 7423c9a918cSPavel Labath dumpAbbreviations(W); 7433c9a918cSPavel Labath 7443c9a918cSPavel Labath if (Hdr.BucketCount > 0) { 7453c9a918cSPavel Labath for (uint32_t Bucket = 0; Bucket < Hdr.BucketCount; ++Bucket) 7463c9a918cSPavel Labath dumpBucket(W, Bucket); 7473c9a918cSPavel Labath return; 7483c9a918cSPavel Labath } 7493c9a918cSPavel Labath 7503c9a918cSPavel Labath W.startLine() << "Hash table not present\n"; 7513c9a918cSPavel Labath for (uint32_t Index = 1; Index <= Hdr.NameCount; ++Index) 7523c9a918cSPavel Labath dumpName(W, Index, None); 7533c9a918cSPavel Labath } 7543c9a918cSPavel Labath 7553c9a918cSPavel Labath llvm::Error DWARFDebugNames::extract() { 7563c9a918cSPavel Labath uint32_t Offset = 0; 7573c9a918cSPavel Labath while (AccelSection.isValidOffset(Offset)) { 7583c9a918cSPavel Labath NameIndex Next(*this, Offset); 7593c9a918cSPavel Labath if (llvm::Error E = Next.extract()) 7603c9a918cSPavel Labath return E; 7613c9a918cSPavel Labath Offset = Next.getNextUnitOffset(); 7623c9a918cSPavel Labath NameIndices.push_back(std::move(Next)); 7633c9a918cSPavel Labath } 7643c9a918cSPavel Labath return Error::success(); 7653c9a918cSPavel Labath } 7663c9a918cSPavel Labath 7673c9a918cSPavel Labath LLVM_DUMP_METHOD void DWARFDebugNames::dump(raw_ostream &OS) const { 7683c9a918cSPavel Labath ScopedPrinter W(OS); 7693c9a918cSPavel Labath for (const NameIndex &NI : NameIndices) 7703c9a918cSPavel Labath NI.dump(W); 7713c9a918cSPavel Labath } 772*d99072bcSPavel Labath 773*d99072bcSPavel Labath Optional<uint32_t> 774*d99072bcSPavel Labath DWARFDebugNames::ValueIterator::findEntryOffsetInCurrentIndex() { 775*d99072bcSPavel Labath const Header &Hdr = CurrentIndex->Hdr; 776*d99072bcSPavel Labath if (Hdr.BucketCount == 0) { 777*d99072bcSPavel Labath // No Hash Table, We need to search through all names in the Name Index. 778*d99072bcSPavel Labath for (uint32_t Index = 1; Index <= Hdr.NameCount; ++Index) { 779*d99072bcSPavel Labath NameTableEntry NTE = CurrentIndex->getNameTableEntry(Index); 780*d99072bcSPavel Labath if (CurrentIndex->Section.StringSection.getCStr(&NTE.StringOffset) == Key) 781*d99072bcSPavel Labath return NTE.EntryOffset; 782*d99072bcSPavel Labath } 783*d99072bcSPavel Labath return None; 784*d99072bcSPavel Labath } 785*d99072bcSPavel Labath 786*d99072bcSPavel Labath // The Name Index has a Hash Table, so use that to speed up the search. 787*d99072bcSPavel Labath // Compute the Key Hash, if it has not been done already. 788*d99072bcSPavel Labath if (!Hash) 789*d99072bcSPavel Labath Hash = caseFoldingDjbHash(Key); 790*d99072bcSPavel Labath uint32_t Bucket = *Hash % Hdr.BucketCount; 791*d99072bcSPavel Labath uint32_t Index = CurrentIndex->getBucketArrayEntry(Bucket); 792*d99072bcSPavel Labath if (Index == 0) 793*d99072bcSPavel Labath return None; // Empty bucket 794*d99072bcSPavel Labath 795*d99072bcSPavel Labath for (; Index <= Hdr.NameCount; ++Index) { 796*d99072bcSPavel Labath uint32_t Hash = CurrentIndex->getHashArrayEntry(Index); 797*d99072bcSPavel Labath if (Hash % Hdr.BucketCount != Bucket) 798*d99072bcSPavel Labath return None; // End of bucket 799*d99072bcSPavel Labath 800*d99072bcSPavel Labath NameTableEntry NTE = CurrentIndex->getNameTableEntry(Index); 801*d99072bcSPavel Labath if (CurrentIndex->Section.StringSection.getCStr(&NTE.StringOffset) == Key) 802*d99072bcSPavel Labath return NTE.EntryOffset; 803*d99072bcSPavel Labath } 804*d99072bcSPavel Labath return None; 805*d99072bcSPavel Labath } 806*d99072bcSPavel Labath 807*d99072bcSPavel Labath bool DWARFDebugNames::ValueIterator::getEntryAtCurrentOffset() { 808*d99072bcSPavel Labath auto EntryOr = CurrentIndex->getEntry(&DataOffset); 809*d99072bcSPavel Labath if (!EntryOr) { 810*d99072bcSPavel Labath consumeError(EntryOr.takeError()); 811*d99072bcSPavel Labath return false; 812*d99072bcSPavel Labath } 813*d99072bcSPavel Labath CurrentEntry = std::move(*EntryOr); 814*d99072bcSPavel Labath return true; 815*d99072bcSPavel Labath } 816*d99072bcSPavel Labath 817*d99072bcSPavel Labath bool DWARFDebugNames::ValueIterator::findInCurrentIndex() { 818*d99072bcSPavel Labath Optional<uint32_t> Offset = findEntryOffsetInCurrentIndex(); 819*d99072bcSPavel Labath if (!Offset) 820*d99072bcSPavel Labath return false; 821*d99072bcSPavel Labath DataOffset = *Offset; 822*d99072bcSPavel Labath return getEntryAtCurrentOffset(); 823*d99072bcSPavel Labath } 824*d99072bcSPavel Labath 825*d99072bcSPavel Labath void DWARFDebugNames::ValueIterator::searchFromStartOfCurrentIndex() { 826*d99072bcSPavel Labath for (const NameIndex *End = CurrentIndex->Section.NameIndices.end(); 827*d99072bcSPavel Labath CurrentIndex != End; ++CurrentIndex) { 828*d99072bcSPavel Labath if (findInCurrentIndex()) 829*d99072bcSPavel Labath return; 830*d99072bcSPavel Labath } 831*d99072bcSPavel Labath setEnd(); 832*d99072bcSPavel Labath } 833*d99072bcSPavel Labath 834*d99072bcSPavel Labath void DWARFDebugNames::ValueIterator::next() { 835*d99072bcSPavel Labath assert(CurrentIndex && "Incrementing an end() iterator?"); 836*d99072bcSPavel Labath 837*d99072bcSPavel Labath // First try the next entry in the current Index. 838*d99072bcSPavel Labath if (getEntryAtCurrentOffset()) 839*d99072bcSPavel Labath return; 840*d99072bcSPavel Labath 841*d99072bcSPavel Labath // Try the next Name Index. 842*d99072bcSPavel Labath ++CurrentIndex; 843*d99072bcSPavel Labath searchFromStartOfCurrentIndex(); 844*d99072bcSPavel Labath } 845*d99072bcSPavel Labath 846*d99072bcSPavel Labath DWARFDebugNames::ValueIterator::ValueIterator(const DWARFDebugNames &AccelTable, 847*d99072bcSPavel Labath StringRef Key) 848*d99072bcSPavel Labath : CurrentIndex(AccelTable.NameIndices.begin()), Key(Key) { 849*d99072bcSPavel Labath searchFromStartOfCurrentIndex(); 850*d99072bcSPavel Labath } 851*d99072bcSPavel Labath 852*d99072bcSPavel Labath iterator_range<DWARFDebugNames::ValueIterator> 853*d99072bcSPavel Labath DWARFDebugNames::equal_range(StringRef Key) const { 854*d99072bcSPavel Labath if (NameIndices.empty()) 855*d99072bcSPavel Labath return make_range(ValueIterator(), ValueIterator()); 856*d99072bcSPavel Labath return make_range(ValueIterator(*this, Key), ValueIterator()); 857*d99072bcSPavel Labath } 858