1 //===- llvm/CodeGen/AsmPrinter/AccelTable.cpp - Accelerator Tables --------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // This file contains support for writing accelerator tables. 10 // 11 //===----------------------------------------------------------------------===// 12 13 #include "llvm/CodeGen/AccelTable.h" 14 #include "DwarfCompileUnit.h" 15 #include "llvm/ADT/STLExtras.h" 16 #include "llvm/ADT/StringMap.h" 17 #include "llvm/ADT/Twine.h" 18 #include "llvm/BinaryFormat/Dwarf.h" 19 #include "llvm/CodeGen/AsmPrinter.h" 20 #include "llvm/CodeGen/DIE.h" 21 #include "llvm/MC/MCExpr.h" 22 #include "llvm/MC/MCStreamer.h" 23 #include "llvm/MC/MCSymbol.h" 24 #include "llvm/Support/raw_ostream.h" 25 #include "llvm/Target/TargetLoweringObjectFile.h" 26 #include <algorithm> 27 #include <cstddef> 28 #include <cstdint> 29 #include <limits> 30 #include <vector> 31 32 using namespace llvm; 33 34 void AccelTableBase::computeBucketCount() { 35 // First get the number of unique hashes. 36 std::vector<uint32_t> Uniques; 37 Uniques.reserve(Entries.size()); 38 for (const auto &E : Entries) 39 Uniques.push_back(E.second.HashValue); 40 array_pod_sort(Uniques.begin(), Uniques.end()); 41 std::vector<uint32_t>::iterator P = 42 std::unique(Uniques.begin(), Uniques.end()); 43 44 UniqueHashCount = std::distance(Uniques.begin(), P); 45 46 if (UniqueHashCount > 1024) 47 BucketCount = UniqueHashCount / 4; 48 else if (UniqueHashCount > 16) 49 BucketCount = UniqueHashCount / 2; 50 else 51 BucketCount = std::max<uint32_t>(UniqueHashCount, 1); 52 } 53 54 void AccelTableBase::finalize(AsmPrinter *Asm, StringRef Prefix) { 55 // Create the individual hash data outputs. 56 for (auto &E : Entries) { 57 // Unique the entries. 58 llvm::stable_sort(E.second.Values, 59 [](const AccelTableData *A, const AccelTableData *B) { 60 return *A < *B; 61 }); 62 E.second.Values.erase( 63 std::unique(E.second.Values.begin(), E.second.Values.end()), 64 E.second.Values.end()); 65 } 66 67 // Figure out how many buckets we need, then compute the bucket contents and 68 // the final ordering. The hashes and offsets can be emitted by walking these 69 // data structures. We add temporary symbols to the data so they can be 70 // referenced when emitting the offsets. 71 computeBucketCount(); 72 73 // Compute bucket contents and final ordering. 74 Buckets.resize(BucketCount); 75 for (auto &E : Entries) { 76 uint32_t Bucket = E.second.HashValue % BucketCount; 77 Buckets[Bucket].push_back(&E.second); 78 E.second.Sym = Asm->createTempSymbol(Prefix); 79 } 80 81 // Sort the contents of the buckets by hash value so that hash collisions end 82 // up together. Stable sort makes testing easier and doesn't cost much more. 83 for (auto &Bucket : Buckets) 84 llvm::stable_sort(Bucket, [](HashData *LHS, HashData *RHS) { 85 return LHS->HashValue < RHS->HashValue; 86 }); 87 } 88 89 namespace { 90 /// Base class for writing out Accelerator tables. It holds the common 91 /// functionality for the two Accelerator table types. 92 class AccelTableWriter { 93 protected: 94 AsmPrinter *const Asm; ///< Destination. 95 const AccelTableBase &Contents; ///< Data to emit. 96 97 /// Controls whether to emit duplicate hash and offset table entries for names 98 /// with identical hashes. Apple tables don't emit duplicate entries, DWARF v5 99 /// tables do. 100 const bool SkipIdenticalHashes; 101 102 void emitHashes() const; 103 104 /// Emit offsets to lists of entries with identical names. The offsets are 105 /// relative to the Base argument. 106 void emitOffsets(const MCSymbol *Base) const; 107 108 public: 109 AccelTableWriter(AsmPrinter *Asm, const AccelTableBase &Contents, 110 bool SkipIdenticalHashes) 111 : Asm(Asm), Contents(Contents), SkipIdenticalHashes(SkipIdenticalHashes) { 112 } 113 }; 114 115 class AppleAccelTableWriter : public AccelTableWriter { 116 using Atom = AppleAccelTableData::Atom; 117 118 /// The fixed header of an Apple Accelerator Table. 119 struct Header { 120 uint32_t Magic = MagicHash; 121 uint16_t Version = 1; 122 uint16_t HashFunction = dwarf::DW_hash_function_djb; 123 uint32_t BucketCount; 124 uint32_t HashCount; 125 uint32_t HeaderDataLength; 126 127 /// 'HASH' magic value to detect endianness. 128 static const uint32_t MagicHash = 0x48415348; 129 130 Header(uint32_t BucketCount, uint32_t UniqueHashCount, uint32_t DataLength) 131 : BucketCount(BucketCount), HashCount(UniqueHashCount), 132 HeaderDataLength(DataLength) {} 133 134 void emit(AsmPrinter *Asm) const; 135 #ifndef NDEBUG 136 void print(raw_ostream &OS) const; 137 void dump() const { print(dbgs()); } 138 #endif 139 }; 140 141 /// The HeaderData describes the structure of an Apple accelerator table 142 /// through a list of Atoms. 143 struct HeaderData { 144 /// In the case of data that is referenced via DW_FORM_ref_* the offset 145 /// base is used to describe the offset for all forms in the list of atoms. 146 uint32_t DieOffsetBase; 147 148 const SmallVector<Atom, 4> Atoms; 149 150 HeaderData(ArrayRef<Atom> AtomList, uint32_t Offset = 0) 151 : DieOffsetBase(Offset), Atoms(AtomList.begin(), AtomList.end()) {} 152 153 void emit(AsmPrinter *Asm) const; 154 #ifndef NDEBUG 155 void print(raw_ostream &OS) const; 156 void dump() const { print(dbgs()); } 157 #endif 158 }; 159 160 Header Header; 161 HeaderData HeaderData; 162 const MCSymbol *SecBegin; 163 164 void emitBuckets() const; 165 void emitData() const; 166 167 public: 168 AppleAccelTableWriter(AsmPrinter *Asm, const AccelTableBase &Contents, 169 ArrayRef<Atom> Atoms, const MCSymbol *SecBegin) 170 : AccelTableWriter(Asm, Contents, true), 171 Header(Contents.getBucketCount(), Contents.getUniqueHashCount(), 172 8 + (Atoms.size() * 4)), 173 HeaderData(Atoms), SecBegin(SecBegin) {} 174 175 void emit() const; 176 177 #ifndef NDEBUG 178 void print(raw_ostream &OS) const; 179 void dump() const { print(dbgs()); } 180 #endif 181 }; 182 183 /// Class responsible for emitting a DWARF v5 Accelerator Table. The only 184 /// public function is emit(), which performs the actual emission. 185 /// 186 /// The class is templated in its data type. This allows us to emit both dyamic 187 /// and static data entries. A callback abstract the logic to provide a CU 188 /// index for a given entry, which is different per data type, but identical 189 /// for every entry in the same table. 190 template <typename DataT> 191 class Dwarf5AccelTableWriter : public AccelTableWriter { 192 struct Header { 193 uint16_t Version = 5; 194 uint16_t Padding = 0; 195 uint32_t CompUnitCount; 196 uint32_t LocalTypeUnitCount = 0; 197 uint32_t ForeignTypeUnitCount = 0; 198 uint32_t BucketCount; 199 uint32_t NameCount; 200 uint32_t AbbrevTableSize = 0; 201 uint32_t AugmentationStringSize = sizeof(AugmentationString); 202 char AugmentationString[8] = {'L', 'L', 'V', 'M', '0', '7', '0', '0'}; 203 204 Header(uint32_t CompUnitCount, uint32_t BucketCount, uint32_t NameCount) 205 : CompUnitCount(CompUnitCount), BucketCount(BucketCount), 206 NameCount(NameCount) {} 207 208 void emit(const Dwarf5AccelTableWriter &Ctx) const; 209 }; 210 struct AttributeEncoding { 211 dwarf::Index Index; 212 dwarf::Form Form; 213 }; 214 215 Header Header; 216 DenseMap<uint32_t, SmallVector<AttributeEncoding, 2>> Abbreviations; 217 ArrayRef<MCSymbol *> CompUnits; 218 llvm::function_ref<unsigned(const DataT &)> getCUIndexForEntry; 219 MCSymbol *ContributionStart = Asm->createTempSymbol("names_start"); 220 MCSymbol *ContributionEnd = Asm->createTempSymbol("names_end"); 221 MCSymbol *AbbrevStart = Asm->createTempSymbol("names_abbrev_start"); 222 MCSymbol *AbbrevEnd = Asm->createTempSymbol("names_abbrev_end"); 223 MCSymbol *EntryPool = Asm->createTempSymbol("names_entries"); 224 225 DenseSet<uint32_t> getUniqueTags() const; 226 227 // Right now, we emit uniform attributes for all tags. 228 SmallVector<AttributeEncoding, 2> getUniformAttributes() const; 229 230 void emitCUList() const; 231 void emitBuckets() const; 232 void emitStringOffsets() const; 233 void emitAbbrevs() const; 234 void emitEntry(const DataT &Entry) const; 235 void emitData() const; 236 237 public: 238 Dwarf5AccelTableWriter( 239 AsmPrinter *Asm, const AccelTableBase &Contents, 240 ArrayRef<MCSymbol *> CompUnits, 241 llvm::function_ref<unsigned(const DataT &)> GetCUIndexForEntry); 242 243 void emit() const; 244 }; 245 } // namespace 246 247 void AccelTableWriter::emitHashes() const { 248 uint64_t PrevHash = std::numeric_limits<uint64_t>::max(); 249 unsigned BucketIdx = 0; 250 for (auto &Bucket : Contents.getBuckets()) { 251 for (auto &Hash : Bucket) { 252 uint32_t HashValue = Hash->HashValue; 253 if (SkipIdenticalHashes && PrevHash == HashValue) 254 continue; 255 Asm->OutStreamer->AddComment("Hash in Bucket " + Twine(BucketIdx)); 256 Asm->emitInt32(HashValue); 257 PrevHash = HashValue; 258 } 259 BucketIdx++; 260 } 261 } 262 263 void AccelTableWriter::emitOffsets(const MCSymbol *Base) const { 264 const auto &Buckets = Contents.getBuckets(); 265 uint64_t PrevHash = std::numeric_limits<uint64_t>::max(); 266 for (size_t i = 0, e = Buckets.size(); i < e; ++i) { 267 for (auto *Hash : Buckets[i]) { 268 uint32_t HashValue = Hash->HashValue; 269 if (SkipIdenticalHashes && PrevHash == HashValue) 270 continue; 271 PrevHash = HashValue; 272 Asm->OutStreamer->AddComment("Offset in Bucket " + Twine(i)); 273 Asm->emitLabelDifference(Hash->Sym, Base, sizeof(uint32_t)); 274 } 275 } 276 } 277 278 void AppleAccelTableWriter::Header::emit(AsmPrinter *Asm) const { 279 Asm->OutStreamer->AddComment("Header Magic"); 280 Asm->emitInt32(Magic); 281 Asm->OutStreamer->AddComment("Header Version"); 282 Asm->emitInt16(Version); 283 Asm->OutStreamer->AddComment("Header Hash Function"); 284 Asm->emitInt16(HashFunction); 285 Asm->OutStreamer->AddComment("Header Bucket Count"); 286 Asm->emitInt32(BucketCount); 287 Asm->OutStreamer->AddComment("Header Hash Count"); 288 Asm->emitInt32(HashCount); 289 Asm->OutStreamer->AddComment("Header Data Length"); 290 Asm->emitInt32(HeaderDataLength); 291 } 292 293 void AppleAccelTableWriter::HeaderData::emit(AsmPrinter *Asm) const { 294 Asm->OutStreamer->AddComment("HeaderData Die Offset Base"); 295 Asm->emitInt32(DieOffsetBase); 296 Asm->OutStreamer->AddComment("HeaderData Atom Count"); 297 Asm->emitInt32(Atoms.size()); 298 299 for (const Atom &A : Atoms) { 300 Asm->OutStreamer->AddComment(dwarf::AtomTypeString(A.Type)); 301 Asm->emitInt16(A.Type); 302 Asm->OutStreamer->AddComment(dwarf::FormEncodingString(A.Form)); 303 Asm->emitInt16(A.Form); 304 } 305 } 306 307 void AppleAccelTableWriter::emitBuckets() const { 308 const auto &Buckets = Contents.getBuckets(); 309 unsigned index = 0; 310 for (size_t i = 0, e = Buckets.size(); i < e; ++i) { 311 Asm->OutStreamer->AddComment("Bucket " + Twine(i)); 312 if (!Buckets[i].empty()) 313 Asm->emitInt32(index); 314 else 315 Asm->emitInt32(std::numeric_limits<uint32_t>::max()); 316 // Buckets point in the list of hashes, not to the data. Do not increment 317 // the index multiple times in case of hash collisions. 318 uint64_t PrevHash = std::numeric_limits<uint64_t>::max(); 319 for (auto *HD : Buckets[i]) { 320 uint32_t HashValue = HD->HashValue; 321 if (PrevHash != HashValue) 322 ++index; 323 PrevHash = HashValue; 324 } 325 } 326 } 327 328 void AppleAccelTableWriter::emitData() const { 329 const auto &Buckets = Contents.getBuckets(); 330 for (size_t i = 0, e = Buckets.size(); i < e; ++i) { 331 uint64_t PrevHash = std::numeric_limits<uint64_t>::max(); 332 for (auto &Hash : Buckets[i]) { 333 // Terminate the previous entry if there is no hash collision with the 334 // current one. 335 if (PrevHash != std::numeric_limits<uint64_t>::max() && 336 PrevHash != Hash->HashValue) 337 Asm->emitInt32(0); 338 // Remember to emit the label for our offset. 339 Asm->OutStreamer->emitLabel(Hash->Sym); 340 Asm->OutStreamer->AddComment(Hash->Name.getString()); 341 Asm->emitDwarfStringOffset(Hash->Name); 342 Asm->OutStreamer->AddComment("Num DIEs"); 343 Asm->emitInt32(Hash->Values.size()); 344 for (const auto *V : Hash->Values) 345 static_cast<const AppleAccelTableData *>(V)->emit(Asm); 346 PrevHash = Hash->HashValue; 347 } 348 // Emit the final end marker for the bucket. 349 if (!Buckets[i].empty()) 350 Asm->emitInt32(0); 351 } 352 } 353 354 void AppleAccelTableWriter::emit() const { 355 Header.emit(Asm); 356 HeaderData.emit(Asm); 357 emitBuckets(); 358 emitHashes(); 359 emitOffsets(SecBegin); 360 emitData(); 361 } 362 363 template <typename DataT> 364 void Dwarf5AccelTableWriter<DataT>::Header::emit( 365 const Dwarf5AccelTableWriter &Ctx) const { 366 assert(CompUnitCount > 0 && "Index must have at least one CU."); 367 368 AsmPrinter *Asm = Ctx.Asm; 369 Asm->OutStreamer->AddComment("Header: unit length"); 370 Asm->emitLabelDifference(Ctx.ContributionEnd, Ctx.ContributionStart, 371 sizeof(uint32_t)); 372 Asm->OutStreamer->emitLabel(Ctx.ContributionStart); 373 Asm->OutStreamer->AddComment("Header: version"); 374 Asm->emitInt16(Version); 375 Asm->OutStreamer->AddComment("Header: padding"); 376 Asm->emitInt16(Padding); 377 Asm->OutStreamer->AddComment("Header: compilation unit count"); 378 Asm->emitInt32(CompUnitCount); 379 Asm->OutStreamer->AddComment("Header: local type unit count"); 380 Asm->emitInt32(LocalTypeUnitCount); 381 Asm->OutStreamer->AddComment("Header: foreign type unit count"); 382 Asm->emitInt32(ForeignTypeUnitCount); 383 Asm->OutStreamer->AddComment("Header: bucket count"); 384 Asm->emitInt32(BucketCount); 385 Asm->OutStreamer->AddComment("Header: name count"); 386 Asm->emitInt32(NameCount); 387 Asm->OutStreamer->AddComment("Header: abbreviation table size"); 388 Asm->emitLabelDifference(Ctx.AbbrevEnd, Ctx.AbbrevStart, sizeof(uint32_t)); 389 Asm->OutStreamer->AddComment("Header: augmentation string size"); 390 assert(AugmentationStringSize % 4 == 0); 391 Asm->emitInt32(AugmentationStringSize); 392 Asm->OutStreamer->AddComment("Header: augmentation string"); 393 Asm->OutStreamer->emitBytes({AugmentationString, AugmentationStringSize}); 394 } 395 396 template <typename DataT> 397 DenseSet<uint32_t> Dwarf5AccelTableWriter<DataT>::getUniqueTags() const { 398 DenseSet<uint32_t> UniqueTags; 399 for (auto &Bucket : Contents.getBuckets()) { 400 for (auto *Hash : Bucket) { 401 for (auto *Value : Hash->Values) { 402 unsigned Tag = static_cast<const DataT *>(Value)->getDieTag(); 403 UniqueTags.insert(Tag); 404 } 405 } 406 } 407 return UniqueTags; 408 } 409 410 template <typename DataT> 411 SmallVector<typename Dwarf5AccelTableWriter<DataT>::AttributeEncoding, 2> 412 Dwarf5AccelTableWriter<DataT>::getUniformAttributes() const { 413 SmallVector<AttributeEncoding, 2> UA; 414 if (CompUnits.size() > 1) { 415 size_t LargestCUIndex = CompUnits.size() - 1; 416 dwarf::Form Form = DIEInteger::BestForm(/*IsSigned*/ false, LargestCUIndex); 417 UA.push_back({dwarf::DW_IDX_compile_unit, Form}); 418 } 419 UA.push_back({dwarf::DW_IDX_die_offset, dwarf::DW_FORM_ref4}); 420 return UA; 421 } 422 423 template <typename DataT> 424 void Dwarf5AccelTableWriter<DataT>::emitCUList() const { 425 for (const auto &CU : enumerate(CompUnits)) { 426 Asm->OutStreamer->AddComment("Compilation unit " + Twine(CU.index())); 427 Asm->emitDwarfSymbolReference(CU.value()); 428 } 429 } 430 431 template <typename DataT> 432 void Dwarf5AccelTableWriter<DataT>::emitBuckets() const { 433 uint32_t Index = 1; 434 for (const auto &Bucket : enumerate(Contents.getBuckets())) { 435 Asm->OutStreamer->AddComment("Bucket " + Twine(Bucket.index())); 436 Asm->emitInt32(Bucket.value().empty() ? 0 : Index); 437 Index += Bucket.value().size(); 438 } 439 } 440 441 template <typename DataT> 442 void Dwarf5AccelTableWriter<DataT>::emitStringOffsets() const { 443 for (const auto &Bucket : enumerate(Contents.getBuckets())) { 444 for (auto *Hash : Bucket.value()) { 445 DwarfStringPoolEntryRef String = Hash->Name; 446 Asm->OutStreamer->AddComment("String in Bucket " + Twine(Bucket.index()) + 447 ": " + String.getString()); 448 Asm->emitDwarfStringOffset(String); 449 } 450 } 451 } 452 453 template <typename DataT> 454 void Dwarf5AccelTableWriter<DataT>::emitAbbrevs() const { 455 Asm->OutStreamer->emitLabel(AbbrevStart); 456 for (const auto &Abbrev : Abbreviations) { 457 Asm->OutStreamer->AddComment("Abbrev code"); 458 assert(Abbrev.first != 0); 459 Asm->emitULEB128(Abbrev.first); 460 Asm->OutStreamer->AddComment(dwarf::TagString(Abbrev.first)); 461 Asm->emitULEB128(Abbrev.first); 462 for (const auto &AttrEnc : Abbrev.second) { 463 Asm->emitULEB128(AttrEnc.Index, dwarf::IndexString(AttrEnc.Index).data()); 464 Asm->emitULEB128(AttrEnc.Form, 465 dwarf::FormEncodingString(AttrEnc.Form).data()); 466 } 467 Asm->emitULEB128(0, "End of abbrev"); 468 Asm->emitULEB128(0, "End of abbrev"); 469 } 470 Asm->emitULEB128(0, "End of abbrev list"); 471 Asm->OutStreamer->emitLabel(AbbrevEnd); 472 } 473 474 template <typename DataT> 475 void Dwarf5AccelTableWriter<DataT>::emitEntry(const DataT &Entry) const { 476 auto AbbrevIt = Abbreviations.find(Entry.getDieTag()); 477 assert(AbbrevIt != Abbreviations.end() && 478 "Why wasn't this abbrev generated?"); 479 480 Asm->emitULEB128(AbbrevIt->first, "Abbreviation code"); 481 for (const auto &AttrEnc : AbbrevIt->second) { 482 Asm->OutStreamer->AddComment(dwarf::IndexString(AttrEnc.Index)); 483 switch (AttrEnc.Index) { 484 case dwarf::DW_IDX_compile_unit: { 485 DIEInteger ID(getCUIndexForEntry(Entry)); 486 ID.emitValue(Asm, AttrEnc.Form); 487 break; 488 } 489 case dwarf::DW_IDX_die_offset: 490 assert(AttrEnc.Form == dwarf::DW_FORM_ref4); 491 Asm->emitInt32(Entry.getDieOffset()); 492 break; 493 default: 494 llvm_unreachable("Unexpected index attribute!"); 495 } 496 } 497 } 498 499 template <typename DataT> void Dwarf5AccelTableWriter<DataT>::emitData() const { 500 Asm->OutStreamer->emitLabel(EntryPool); 501 for (auto &Bucket : Contents.getBuckets()) { 502 for (auto *Hash : Bucket) { 503 // Remember to emit the label for our offset. 504 Asm->OutStreamer->emitLabel(Hash->Sym); 505 for (const auto *Value : Hash->Values) 506 emitEntry(*static_cast<const DataT *>(Value)); 507 Asm->OutStreamer->AddComment("End of list: " + Hash->Name.getString()); 508 Asm->emitInt32(0); 509 } 510 } 511 } 512 513 template <typename DataT> 514 Dwarf5AccelTableWriter<DataT>::Dwarf5AccelTableWriter( 515 AsmPrinter *Asm, const AccelTableBase &Contents, 516 ArrayRef<MCSymbol *> CompUnits, 517 llvm::function_ref<unsigned(const DataT &)> getCUIndexForEntry) 518 : AccelTableWriter(Asm, Contents, false), 519 Header(CompUnits.size(), Contents.getBucketCount(), 520 Contents.getUniqueNameCount()), 521 CompUnits(CompUnits), getCUIndexForEntry(std::move(getCUIndexForEntry)) { 522 DenseSet<uint32_t> UniqueTags = getUniqueTags(); 523 SmallVector<AttributeEncoding, 2> UniformAttributes = getUniformAttributes(); 524 525 Abbreviations.reserve(UniqueTags.size()); 526 for (uint32_t Tag : UniqueTags) 527 Abbreviations.try_emplace(Tag, UniformAttributes); 528 } 529 530 template <typename DataT> void Dwarf5AccelTableWriter<DataT>::emit() const { 531 Header.emit(*this); 532 emitCUList(); 533 emitBuckets(); 534 emitHashes(); 535 emitStringOffsets(); 536 emitOffsets(EntryPool); 537 emitAbbrevs(); 538 emitData(); 539 Asm->OutStreamer->emitValueToAlignment(4, 0); 540 Asm->OutStreamer->emitLabel(ContributionEnd); 541 } 542 543 void llvm::emitAppleAccelTableImpl(AsmPrinter *Asm, AccelTableBase &Contents, 544 StringRef Prefix, const MCSymbol *SecBegin, 545 ArrayRef<AppleAccelTableData::Atom> Atoms) { 546 Contents.finalize(Asm, Prefix); 547 AppleAccelTableWriter(Asm, Contents, Atoms, SecBegin).emit(); 548 } 549 550 void llvm::emitDWARF5AccelTable( 551 AsmPrinter *Asm, AccelTable<DWARF5AccelTableData> &Contents, 552 const DwarfDebug &DD, ArrayRef<std::unique_ptr<DwarfCompileUnit>> CUs) { 553 std::vector<MCSymbol *> CompUnits; 554 SmallVector<unsigned, 1> CUIndex(CUs.size()); 555 int Count = 0; 556 for (const auto &CU : enumerate(CUs)) { 557 if (CU.value()->getCUNode()->getNameTableKind() != 558 DICompileUnit::DebugNameTableKind::Default) 559 continue; 560 CUIndex[CU.index()] = Count++; 561 assert(CU.index() == CU.value()->getUniqueID()); 562 const DwarfCompileUnit *MainCU = 563 DD.useSplitDwarf() ? CU.value()->getSkeleton() : CU.value().get(); 564 CompUnits.push_back(MainCU->getLabelBegin()); 565 } 566 567 if (CompUnits.empty()) 568 return; 569 570 Asm->OutStreamer->SwitchSection( 571 Asm->getObjFileLowering().getDwarfDebugNamesSection()); 572 573 Contents.finalize(Asm, "names"); 574 Dwarf5AccelTableWriter<DWARF5AccelTableData>( 575 Asm, Contents, CompUnits, 576 [&](const DWARF5AccelTableData &Entry) { 577 const DIE *CUDie = Entry.getDie().getUnitDie(); 578 return CUIndex[DD.lookupCU(CUDie)->getUniqueID()]; 579 }) 580 .emit(); 581 } 582 583 void llvm::emitDWARF5AccelTable( 584 AsmPrinter *Asm, AccelTable<DWARF5AccelTableStaticData> &Contents, 585 ArrayRef<MCSymbol *> CUs, 586 llvm::function_ref<unsigned(const DWARF5AccelTableStaticData &)> 587 getCUIndexForEntry) { 588 Contents.finalize(Asm, "names"); 589 Dwarf5AccelTableWriter<DWARF5AccelTableStaticData>(Asm, Contents, CUs, 590 getCUIndexForEntry) 591 .emit(); 592 } 593 594 void AppleAccelTableOffsetData::emit(AsmPrinter *Asm) const { 595 Asm->emitInt32(Die.getDebugSectionOffset()); 596 } 597 598 void AppleAccelTableTypeData::emit(AsmPrinter *Asm) const { 599 Asm->emitInt32(Die.getDebugSectionOffset()); 600 Asm->emitInt16(Die.getTag()); 601 Asm->emitInt8(0); 602 } 603 604 void AppleAccelTableStaticOffsetData::emit(AsmPrinter *Asm) const { 605 Asm->emitInt32(Offset); 606 } 607 608 void AppleAccelTableStaticTypeData::emit(AsmPrinter *Asm) const { 609 Asm->emitInt32(Offset); 610 Asm->emitInt16(Tag); 611 Asm->emitInt8(ObjCClassIsImplementation ? dwarf::DW_FLAG_type_implementation 612 : 0); 613 Asm->emitInt32(QualifiedNameHash); 614 } 615 616 constexpr AppleAccelTableData::Atom AppleAccelTableTypeData::Atoms[]; 617 constexpr AppleAccelTableData::Atom AppleAccelTableOffsetData::Atoms[]; 618 constexpr AppleAccelTableData::Atom AppleAccelTableStaticOffsetData::Atoms[]; 619 constexpr AppleAccelTableData::Atom AppleAccelTableStaticTypeData::Atoms[]; 620 621 #ifndef NDEBUG 622 void AppleAccelTableWriter::Header::print(raw_ostream &OS) const { 623 OS << "Magic: " << format("0x%x", Magic) << "\n" 624 << "Version: " << Version << "\n" 625 << "Hash Function: " << HashFunction << "\n" 626 << "Bucket Count: " << BucketCount << "\n" 627 << "Header Data Length: " << HeaderDataLength << "\n"; 628 } 629 630 void AppleAccelTableData::Atom::print(raw_ostream &OS) const { 631 OS << "Type: " << dwarf::AtomTypeString(Type) << "\n" 632 << "Form: " << dwarf::FormEncodingString(Form) << "\n"; 633 } 634 635 void AppleAccelTableWriter::HeaderData::print(raw_ostream &OS) const { 636 OS << "DIE Offset Base: " << DieOffsetBase << "\n"; 637 for (auto Atom : Atoms) 638 Atom.print(OS); 639 } 640 641 void AppleAccelTableWriter::print(raw_ostream &OS) const { 642 Header.print(OS); 643 HeaderData.print(OS); 644 Contents.print(OS); 645 SecBegin->print(OS, nullptr); 646 } 647 648 void AccelTableBase::HashData::print(raw_ostream &OS) const { 649 OS << "Name: " << Name.getString() << "\n"; 650 OS << " Hash Value: " << format("0x%x", HashValue) << "\n"; 651 OS << " Symbol: "; 652 if (Sym) 653 OS << *Sym; 654 else 655 OS << "<none>"; 656 OS << "\n"; 657 for (auto *Value : Values) 658 Value->print(OS); 659 } 660 661 void AccelTableBase::print(raw_ostream &OS) const { 662 // Print Content. 663 OS << "Entries: \n"; 664 for (const auto &Entry : Entries) { 665 OS << "Name: " << Entry.first() << "\n"; 666 for (auto *V : Entry.second.Values) 667 V->print(OS); 668 } 669 670 OS << "Buckets and Hashes: \n"; 671 for (auto &Bucket : Buckets) 672 for (auto &Hash : Bucket) 673 Hash->print(OS); 674 675 OS << "Data: \n"; 676 for (auto &E : Entries) 677 E.second.print(OS); 678 } 679 680 void DWARF5AccelTableData::print(raw_ostream &OS) const { 681 OS << " Offset: " << getDieOffset() << "\n"; 682 OS << " Tag: " << dwarf::TagString(getDieTag()) << "\n"; 683 } 684 685 void DWARF5AccelTableStaticData::print(raw_ostream &OS) const { 686 OS << " Offset: " << getDieOffset() << "\n"; 687 OS << " Tag: " << dwarf::TagString(getDieTag()) << "\n"; 688 } 689 690 void AppleAccelTableOffsetData::print(raw_ostream &OS) const { 691 OS << " Offset: " << Die.getOffset() << "\n"; 692 } 693 694 void AppleAccelTableTypeData::print(raw_ostream &OS) const { 695 OS << " Offset: " << Die.getOffset() << "\n"; 696 OS << " Tag: " << dwarf::TagString(Die.getTag()) << "\n"; 697 } 698 699 void AppleAccelTableStaticOffsetData::print(raw_ostream &OS) const { 700 OS << " Static Offset: " << Offset << "\n"; 701 } 702 703 void AppleAccelTableStaticTypeData::print(raw_ostream &OS) const { 704 OS << " Static Offset: " << Offset << "\n"; 705 OS << " QualifiedNameHash: " << format("%x\n", QualifiedNameHash) << "\n"; 706 OS << " Tag: " << dwarf::TagString(Tag) << "\n"; 707 OS << " ObjCClassIsImplementation: " 708 << (ObjCClassIsImplementation ? "true" : "false"); 709 OS << "\n"; 710 } 711 #endif 712