711 lines
		
	
	
		
			24 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			711 lines
		
	
	
		
			24 KiB
		
	
	
	
		
			C++
		
	
	
	
| //===- llvm/CodeGen/AsmPrinter/AccelTable.cpp - Accelerator Tables --------===//
 | |
| //
 | |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 | |
| // See https://llvm.org/LICENSE.txt for license information.
 | |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 | |
| //
 | |
| //===----------------------------------------------------------------------===//
 | |
| //
 | |
| // This file contains support for writing accelerator tables.
 | |
| //
 | |
| //===----------------------------------------------------------------------===//
 | |
| 
 | |
| #include "llvm/CodeGen/AccelTable.h"
 | |
| #include "DwarfCompileUnit.h"
 | |
| #include "llvm/ADT/STLExtras.h"
 | |
| #include "llvm/ADT/StringMap.h"
 | |
| #include "llvm/ADT/Twine.h"
 | |
| #include "llvm/BinaryFormat/Dwarf.h"
 | |
| #include "llvm/CodeGen/AsmPrinter.h"
 | |
| #include "llvm/CodeGen/DIE.h"
 | |
| #include "llvm/MC/MCStreamer.h"
 | |
| #include "llvm/MC/MCSymbol.h"
 | |
| #include "llvm/Support/raw_ostream.h"
 | |
| #include "llvm/Target/TargetLoweringObjectFile.h"
 | |
| #include <algorithm>
 | |
| #include <cstddef>
 | |
| #include <cstdint>
 | |
| #include <limits>
 | |
| #include <vector>
 | |
| 
 | |
| using namespace llvm;
 | |
| 
 | |
| void AccelTableBase::computeBucketCount() {
 | |
|   // First get the number of unique hashes.
 | |
|   std::vector<uint32_t> Uniques;
 | |
|   Uniques.reserve(Entries.size());
 | |
|   for (const auto &E : Entries)
 | |
|     Uniques.push_back(E.second.HashValue);
 | |
|   array_pod_sort(Uniques.begin(), Uniques.end());
 | |
|   std::vector<uint32_t>::iterator P =
 | |
|       std::unique(Uniques.begin(), Uniques.end());
 | |
| 
 | |
|   UniqueHashCount = std::distance(Uniques.begin(), P);
 | |
| 
 | |
|   if (UniqueHashCount > 1024)
 | |
|     BucketCount = UniqueHashCount / 4;
 | |
|   else if (UniqueHashCount > 16)
 | |
|     BucketCount = UniqueHashCount / 2;
 | |
|   else
 | |
|     BucketCount = std::max<uint32_t>(UniqueHashCount, 1);
 | |
| }
 | |
| 
 | |
| void AccelTableBase::finalize(AsmPrinter *Asm, StringRef Prefix) {
 | |
|   // Create the individual hash data outputs.
 | |
|   for (auto &E : Entries) {
 | |
|     // Unique the entries.
 | |
|     llvm::stable_sort(E.second.Values,
 | |
|                       [](const AccelTableData *A, const AccelTableData *B) {
 | |
|                         return *A < *B;
 | |
|                       });
 | |
|     E.second.Values.erase(
 | |
|         std::unique(E.second.Values.begin(), E.second.Values.end()),
 | |
|         E.second.Values.end());
 | |
|   }
 | |
| 
 | |
|   // Figure out how many buckets we need, then compute the bucket contents and
 | |
|   // the final ordering. The hashes and offsets can be emitted by walking these
 | |
|   // data structures. We add temporary symbols to the data so they can be
 | |
|   // referenced when emitting the offsets.
 | |
|   computeBucketCount();
 | |
| 
 | |
|   // Compute bucket contents and final ordering.
 | |
|   Buckets.resize(BucketCount);
 | |
|   for (auto &E : Entries) {
 | |
|     uint32_t Bucket = E.second.HashValue % BucketCount;
 | |
|     Buckets[Bucket].push_back(&E.second);
 | |
|     E.second.Sym = Asm->createTempSymbol(Prefix);
 | |
|   }
 | |
| 
 | |
|   // Sort the contents of the buckets by hash value so that hash collisions end
 | |
|   // up together. Stable sort makes testing easier and doesn't cost much more.
 | |
|   for (auto &Bucket : Buckets)
 | |
|     llvm::stable_sort(Bucket, [](HashData *LHS, HashData *RHS) {
 | |
|       return LHS->HashValue < RHS->HashValue;
 | |
|     });
 | |
| }
 | |
| 
 | |
| namespace {
 | |
| /// Base class for writing out Accelerator tables. It holds the common
 | |
| /// functionality for the two Accelerator table types.
 | |
| class AccelTableWriter {
 | |
| protected:
 | |
|   AsmPrinter *const Asm;          ///< Destination.
 | |
|   const AccelTableBase &Contents; ///< Data to emit.
 | |
| 
 | |
|   /// Controls whether to emit duplicate hash and offset table entries for names
 | |
|   /// with identical hashes. Apple tables don't emit duplicate entries, DWARF v5
 | |
|   /// tables do.
 | |
|   const bool SkipIdenticalHashes;
 | |
| 
 | |
|   void emitHashes() const;
 | |
| 
 | |
|   /// Emit offsets to lists of entries with identical names. The offsets are
 | |
|   /// relative to the Base argument.
 | |
|   void emitOffsets(const MCSymbol *Base) const;
 | |
| 
 | |
| public:
 | |
|   AccelTableWriter(AsmPrinter *Asm, const AccelTableBase &Contents,
 | |
|                    bool SkipIdenticalHashes)
 | |
|       : Asm(Asm), Contents(Contents), SkipIdenticalHashes(SkipIdenticalHashes) {
 | |
|   }
 | |
| };
 | |
| 
 | |
| class AppleAccelTableWriter : public AccelTableWriter {
 | |
|   using Atom = AppleAccelTableData::Atom;
 | |
| 
 | |
|   /// The fixed header of an Apple Accelerator Table.
 | |
|   struct Header {
 | |
|     uint32_t Magic = MagicHash;
 | |
|     uint16_t Version = 1;
 | |
|     uint16_t HashFunction = dwarf::DW_hash_function_djb;
 | |
|     uint32_t BucketCount;
 | |
|     uint32_t HashCount;
 | |
|     uint32_t HeaderDataLength;
 | |
| 
 | |
|     /// 'HASH' magic value to detect endianness.
 | |
|     static const uint32_t MagicHash = 0x48415348;
 | |
| 
 | |
|     Header(uint32_t BucketCount, uint32_t UniqueHashCount, uint32_t DataLength)
 | |
|         : BucketCount(BucketCount), HashCount(UniqueHashCount),
 | |
|           HeaderDataLength(DataLength) {}
 | |
| 
 | |
|     void emit(AsmPrinter *Asm) const;
 | |
| #ifndef NDEBUG
 | |
|     void print(raw_ostream &OS) const;
 | |
|     void dump() const { print(dbgs()); }
 | |
| #endif
 | |
|   };
 | |
| 
 | |
|   /// The HeaderData describes the structure of an Apple accelerator table
 | |
|   /// through a list of Atoms.
 | |
|   struct HeaderData {
 | |
|     /// In the case of data that is referenced via DW_FORM_ref_* the offset
 | |
|     /// base is used to describe the offset for all forms in the list of atoms.
 | |
|     uint32_t DieOffsetBase;
 | |
| 
 | |
|     const SmallVector<Atom, 4> Atoms;
 | |
| 
 | |
|     HeaderData(ArrayRef<Atom> AtomList, uint32_t Offset = 0)
 | |
|         : DieOffsetBase(Offset), Atoms(AtomList.begin(), AtomList.end()) {}
 | |
| 
 | |
|     void emit(AsmPrinter *Asm) const;
 | |
| #ifndef NDEBUG
 | |
|     void print(raw_ostream &OS) const;
 | |
|     void dump() const { print(dbgs()); }
 | |
| #endif
 | |
|   };
 | |
| 
 | |
|   Header Header;
 | |
|   HeaderData HeaderData;
 | |
|   const MCSymbol *SecBegin;
 | |
| 
 | |
|   void emitBuckets() const;
 | |
|   void emitData() const;
 | |
| 
 | |
| public:
 | |
|   AppleAccelTableWriter(AsmPrinter *Asm, const AccelTableBase &Contents,
 | |
|                         ArrayRef<Atom> Atoms, const MCSymbol *SecBegin)
 | |
|       : AccelTableWriter(Asm, Contents, true),
 | |
|         Header(Contents.getBucketCount(), Contents.getUniqueHashCount(),
 | |
|                8 + (Atoms.size() * 4)),
 | |
|         HeaderData(Atoms), SecBegin(SecBegin) {}
 | |
| 
 | |
|   void emit() const;
 | |
| 
 | |
| #ifndef NDEBUG
 | |
|   void print(raw_ostream &OS) const;
 | |
|   void dump() const { print(dbgs()); }
 | |
| #endif
 | |
| };
 | |
| 
 | |
| /// Class responsible for emitting a DWARF v5 Accelerator Table. The only
 | |
| /// public function is emit(), which performs the actual emission.
 | |
| ///
 | |
| /// The class is templated in its data type. This allows us to emit both dyamic
 | |
| /// and static data entries. A callback abstract the logic to provide a CU
 | |
| /// index for a given entry, which is different per data type, but identical
 | |
| /// for every entry in the same table.
 | |
| template <typename DataT>
 | |
| class Dwarf5AccelTableWriter : public AccelTableWriter {
 | |
|   struct Header {
 | |
|     uint16_t Version = 5;
 | |
|     uint16_t Padding = 0;
 | |
|     uint32_t CompUnitCount;
 | |
|     uint32_t LocalTypeUnitCount = 0;
 | |
|     uint32_t ForeignTypeUnitCount = 0;
 | |
|     uint32_t BucketCount;
 | |
|     uint32_t NameCount;
 | |
|     uint32_t AbbrevTableSize = 0;
 | |
|     uint32_t AugmentationStringSize = sizeof(AugmentationString);
 | |
|     char AugmentationString[8] = {'L', 'L', 'V', 'M', '0', '7', '0', '0'};
 | |
| 
 | |
|     Header(uint32_t CompUnitCount, uint32_t BucketCount, uint32_t NameCount)
 | |
|         : CompUnitCount(CompUnitCount), BucketCount(BucketCount),
 | |
|           NameCount(NameCount) {}
 | |
| 
 | |
|     void emit(Dwarf5AccelTableWriter &Ctx);
 | |
|   };
 | |
|   struct AttributeEncoding {
 | |
|     dwarf::Index Index;
 | |
|     dwarf::Form Form;
 | |
|   };
 | |
| 
 | |
|   Header Header;
 | |
|   DenseMap<uint32_t, SmallVector<AttributeEncoding, 2>> Abbreviations;
 | |
|   ArrayRef<MCSymbol *> CompUnits;
 | |
|   llvm::function_ref<unsigned(const DataT &)> getCUIndexForEntry;
 | |
|   MCSymbol *ContributionEnd = nullptr;
 | |
|   MCSymbol *AbbrevStart = Asm->createTempSymbol("names_abbrev_start");
 | |
|   MCSymbol *AbbrevEnd = Asm->createTempSymbol("names_abbrev_end");
 | |
|   MCSymbol *EntryPool = Asm->createTempSymbol("names_entries");
 | |
| 
 | |
|   DenseSet<uint32_t> getUniqueTags() const;
 | |
| 
 | |
|   // Right now, we emit uniform attributes for all tags.
 | |
|   SmallVector<AttributeEncoding, 2> getUniformAttributes() const;
 | |
| 
 | |
|   void emitCUList() const;
 | |
|   void emitBuckets() const;
 | |
|   void emitStringOffsets() const;
 | |
|   void emitAbbrevs() const;
 | |
|   void emitEntry(const DataT &Entry) const;
 | |
|   void emitData() const;
 | |
| 
 | |
| public:
 | |
|   Dwarf5AccelTableWriter(
 | |
|       AsmPrinter *Asm, const AccelTableBase &Contents,
 | |
|       ArrayRef<MCSymbol *> CompUnits,
 | |
|       llvm::function_ref<unsigned(const DataT &)> GetCUIndexForEntry);
 | |
| 
 | |
|   void emit();
 | |
| };
 | |
| } // namespace
 | |
| 
 | |
| void AccelTableWriter::emitHashes() const {
 | |
|   uint64_t PrevHash = std::numeric_limits<uint64_t>::max();
 | |
|   unsigned BucketIdx = 0;
 | |
|   for (const auto &Bucket : Contents.getBuckets()) {
 | |
|     for (const auto &Hash : Bucket) {
 | |
|       uint32_t HashValue = Hash->HashValue;
 | |
|       if (SkipIdenticalHashes && PrevHash == HashValue)
 | |
|         continue;
 | |
|       Asm->OutStreamer->AddComment("Hash in Bucket " + Twine(BucketIdx));
 | |
|       Asm->emitInt32(HashValue);
 | |
|       PrevHash = HashValue;
 | |
|     }
 | |
|     BucketIdx++;
 | |
|   }
 | |
| }
 | |
| 
 | |
| void AccelTableWriter::emitOffsets(const MCSymbol *Base) const {
 | |
|   const auto &Buckets = Contents.getBuckets();
 | |
|   uint64_t PrevHash = std::numeric_limits<uint64_t>::max();
 | |
|   for (size_t i = 0, e = Buckets.size(); i < e; ++i) {
 | |
|     for (auto *Hash : Buckets[i]) {
 | |
|       uint32_t HashValue = Hash->HashValue;
 | |
|       if (SkipIdenticalHashes && PrevHash == HashValue)
 | |
|         continue;
 | |
|       PrevHash = HashValue;
 | |
|       Asm->OutStreamer->AddComment("Offset in Bucket " + Twine(i));
 | |
|       Asm->emitLabelDifference(Hash->Sym, Base, Asm->getDwarfOffsetByteSize());
 | |
|     }
 | |
|   }
 | |
| }
 | |
| 
 | |
| void AppleAccelTableWriter::Header::emit(AsmPrinter *Asm) const {
 | |
|   Asm->OutStreamer->AddComment("Header Magic");
 | |
|   Asm->emitInt32(Magic);
 | |
|   Asm->OutStreamer->AddComment("Header Version");
 | |
|   Asm->emitInt16(Version);
 | |
|   Asm->OutStreamer->AddComment("Header Hash Function");
 | |
|   Asm->emitInt16(HashFunction);
 | |
|   Asm->OutStreamer->AddComment("Header Bucket Count");
 | |
|   Asm->emitInt32(BucketCount);
 | |
|   Asm->OutStreamer->AddComment("Header Hash Count");
 | |
|   Asm->emitInt32(HashCount);
 | |
|   Asm->OutStreamer->AddComment("Header Data Length");
 | |
|   Asm->emitInt32(HeaderDataLength);
 | |
| }
 | |
| 
 | |
| void AppleAccelTableWriter::HeaderData::emit(AsmPrinter *Asm) const {
 | |
|   Asm->OutStreamer->AddComment("HeaderData Die Offset Base");
 | |
|   Asm->emitInt32(DieOffsetBase);
 | |
|   Asm->OutStreamer->AddComment("HeaderData Atom Count");
 | |
|   Asm->emitInt32(Atoms.size());
 | |
| 
 | |
|   for (const Atom &A : Atoms) {
 | |
|     Asm->OutStreamer->AddComment(dwarf::AtomTypeString(A.Type));
 | |
|     Asm->emitInt16(A.Type);
 | |
|     Asm->OutStreamer->AddComment(dwarf::FormEncodingString(A.Form));
 | |
|     Asm->emitInt16(A.Form);
 | |
|   }
 | |
| }
 | |
| 
 | |
| void AppleAccelTableWriter::emitBuckets() const {
 | |
|   const auto &Buckets = Contents.getBuckets();
 | |
|   unsigned index = 0;
 | |
|   for (size_t i = 0, e = Buckets.size(); i < e; ++i) {
 | |
|     Asm->OutStreamer->AddComment("Bucket " + Twine(i));
 | |
|     if (!Buckets[i].empty())
 | |
|       Asm->emitInt32(index);
 | |
|     else
 | |
|       Asm->emitInt32(std::numeric_limits<uint32_t>::max());
 | |
|     // Buckets point in the list of hashes, not to the data. Do not increment
 | |
|     // the index multiple times in case of hash collisions.
 | |
|     uint64_t PrevHash = std::numeric_limits<uint64_t>::max();
 | |
|     for (auto *HD : Buckets[i]) {
 | |
|       uint32_t HashValue = HD->HashValue;
 | |
|       if (PrevHash != HashValue)
 | |
|         ++index;
 | |
|       PrevHash = HashValue;
 | |
|     }
 | |
|   }
 | |
| }
 | |
| 
 | |
| void AppleAccelTableWriter::emitData() const {
 | |
|   const auto &Buckets = Contents.getBuckets();
 | |
|   for (const AccelTableBase::HashList &Bucket : Buckets) {
 | |
|     uint64_t PrevHash = std::numeric_limits<uint64_t>::max();
 | |
|     for (const auto &Hash : Bucket) {
 | |
|       // Terminate the previous entry if there is no hash collision with the
 | |
|       // current one.
 | |
|       if (PrevHash != std::numeric_limits<uint64_t>::max() &&
 | |
|           PrevHash != Hash->HashValue)
 | |
|         Asm->emitInt32(0);
 | |
|       // Remember to emit the label for our offset.
 | |
|       Asm->OutStreamer->emitLabel(Hash->Sym);
 | |
|       Asm->OutStreamer->AddComment(Hash->Name.getString());
 | |
|       Asm->emitDwarfStringOffset(Hash->Name);
 | |
|       Asm->OutStreamer->AddComment("Num DIEs");
 | |
|       Asm->emitInt32(Hash->Values.size());
 | |
|       for (const auto *V : Hash->Values)
 | |
|         static_cast<const AppleAccelTableData *>(V)->emit(Asm);
 | |
|       PrevHash = Hash->HashValue;
 | |
|     }
 | |
|     // Emit the final end marker for the bucket.
 | |
|     if (!Bucket.empty())
 | |
|       Asm->emitInt32(0);
 | |
|   }
 | |
| }
 | |
| 
 | |
| void AppleAccelTableWriter::emit() const {
 | |
|   Header.emit(Asm);
 | |
|   HeaderData.emit(Asm);
 | |
|   emitBuckets();
 | |
|   emitHashes();
 | |
|   emitOffsets(SecBegin);
 | |
|   emitData();
 | |
| }
 | |
| 
 | |
| template <typename DataT>
 | |
| void Dwarf5AccelTableWriter<DataT>::Header::emit(Dwarf5AccelTableWriter &Ctx) {
 | |
|   assert(CompUnitCount > 0 && "Index must have at least one CU.");
 | |
| 
 | |
|   AsmPrinter *Asm = Ctx.Asm;
 | |
|   Ctx.ContributionEnd =
 | |
|       Asm->emitDwarfUnitLength("names", "Header: unit length");
 | |
|   Asm->OutStreamer->AddComment("Header: version");
 | |
|   Asm->emitInt16(Version);
 | |
|   Asm->OutStreamer->AddComment("Header: padding");
 | |
|   Asm->emitInt16(Padding);
 | |
|   Asm->OutStreamer->AddComment("Header: compilation unit count");
 | |
|   Asm->emitInt32(CompUnitCount);
 | |
|   Asm->OutStreamer->AddComment("Header: local type unit count");
 | |
|   Asm->emitInt32(LocalTypeUnitCount);
 | |
|   Asm->OutStreamer->AddComment("Header: foreign type unit count");
 | |
|   Asm->emitInt32(ForeignTypeUnitCount);
 | |
|   Asm->OutStreamer->AddComment("Header: bucket count");
 | |
|   Asm->emitInt32(BucketCount);
 | |
|   Asm->OutStreamer->AddComment("Header: name count");
 | |
|   Asm->emitInt32(NameCount);
 | |
|   Asm->OutStreamer->AddComment("Header: abbreviation table size");
 | |
|   Asm->emitLabelDifference(Ctx.AbbrevEnd, Ctx.AbbrevStart, sizeof(uint32_t));
 | |
|   Asm->OutStreamer->AddComment("Header: augmentation string size");
 | |
|   assert(AugmentationStringSize % 4 == 0);
 | |
|   Asm->emitInt32(AugmentationStringSize);
 | |
|   Asm->OutStreamer->AddComment("Header: augmentation string");
 | |
|   Asm->OutStreamer->emitBytes({AugmentationString, AugmentationStringSize});
 | |
| }
 | |
| 
 | |
| template <typename DataT>
 | |
| DenseSet<uint32_t> Dwarf5AccelTableWriter<DataT>::getUniqueTags() const {
 | |
|   DenseSet<uint32_t> UniqueTags;
 | |
|   for (auto &Bucket : Contents.getBuckets()) {
 | |
|     for (auto *Hash : Bucket) {
 | |
|       for (auto *Value : Hash->Values) {
 | |
|         unsigned Tag = static_cast<const DataT *>(Value)->getDieTag();
 | |
|         UniqueTags.insert(Tag);
 | |
|       }
 | |
|     }
 | |
|   }
 | |
|   return UniqueTags;
 | |
| }
 | |
| 
 | |
| template <typename DataT>
 | |
| SmallVector<typename Dwarf5AccelTableWriter<DataT>::AttributeEncoding, 2>
 | |
| Dwarf5AccelTableWriter<DataT>::getUniformAttributes() const {
 | |
|   SmallVector<AttributeEncoding, 2> UA;
 | |
|   if (CompUnits.size() > 1) {
 | |
|     size_t LargestCUIndex = CompUnits.size() - 1;
 | |
|     dwarf::Form Form = DIEInteger::BestForm(/*IsSigned*/ false, LargestCUIndex);
 | |
|     UA.push_back({dwarf::DW_IDX_compile_unit, Form});
 | |
|   }
 | |
|   UA.push_back({dwarf::DW_IDX_die_offset, dwarf::DW_FORM_ref4});
 | |
|   return UA;
 | |
| }
 | |
| 
 | |
| template <typename DataT>
 | |
| void Dwarf5AccelTableWriter<DataT>::emitCUList() const {
 | |
|   for (const auto &CU : enumerate(CompUnits)) {
 | |
|     Asm->OutStreamer->AddComment("Compilation unit " + Twine(CU.index()));
 | |
|     Asm->emitDwarfSymbolReference(CU.value());
 | |
|   }
 | |
| }
 | |
| 
 | |
| template <typename DataT>
 | |
| void Dwarf5AccelTableWriter<DataT>::emitBuckets() const {
 | |
|   uint32_t Index = 1;
 | |
|   for (const auto &Bucket : enumerate(Contents.getBuckets())) {
 | |
|     Asm->OutStreamer->AddComment("Bucket " + Twine(Bucket.index()));
 | |
|     Asm->emitInt32(Bucket.value().empty() ? 0 : Index);
 | |
|     Index += Bucket.value().size();
 | |
|   }
 | |
| }
 | |
| 
 | |
| template <typename DataT>
 | |
| void Dwarf5AccelTableWriter<DataT>::emitStringOffsets() const {
 | |
|   for (const auto &Bucket : enumerate(Contents.getBuckets())) {
 | |
|     for (auto *Hash : Bucket.value()) {
 | |
|       DwarfStringPoolEntryRef String = Hash->Name;
 | |
|       Asm->OutStreamer->AddComment("String in Bucket " + Twine(Bucket.index()) +
 | |
|                                    ": " + String.getString());
 | |
|       Asm->emitDwarfStringOffset(String);
 | |
|     }
 | |
|   }
 | |
| }
 | |
| 
 | |
| template <typename DataT>
 | |
| void Dwarf5AccelTableWriter<DataT>::emitAbbrevs() const {
 | |
|   Asm->OutStreamer->emitLabel(AbbrevStart);
 | |
|   for (const auto &Abbrev : Abbreviations) {
 | |
|     Asm->OutStreamer->AddComment("Abbrev code");
 | |
|     assert(Abbrev.first != 0);
 | |
|     Asm->emitULEB128(Abbrev.first);
 | |
|     Asm->OutStreamer->AddComment(dwarf::TagString(Abbrev.first));
 | |
|     Asm->emitULEB128(Abbrev.first);
 | |
|     for (const auto &AttrEnc : Abbrev.second) {
 | |
|       Asm->emitULEB128(AttrEnc.Index, dwarf::IndexString(AttrEnc.Index).data());
 | |
|       Asm->emitULEB128(AttrEnc.Form,
 | |
|                        dwarf::FormEncodingString(AttrEnc.Form).data());
 | |
|     }
 | |
|     Asm->emitULEB128(0, "End of abbrev");
 | |
|     Asm->emitULEB128(0, "End of abbrev");
 | |
|   }
 | |
|   Asm->emitULEB128(0, "End of abbrev list");
 | |
|   Asm->OutStreamer->emitLabel(AbbrevEnd);
 | |
| }
 | |
| 
 | |
| template <typename DataT>
 | |
| void Dwarf5AccelTableWriter<DataT>::emitEntry(const DataT &Entry) const {
 | |
|   auto AbbrevIt = Abbreviations.find(Entry.getDieTag());
 | |
|   assert(AbbrevIt != Abbreviations.end() &&
 | |
|          "Why wasn't this abbrev generated?");
 | |
| 
 | |
|   Asm->emitULEB128(AbbrevIt->first, "Abbreviation code");
 | |
|   for (const auto &AttrEnc : AbbrevIt->second) {
 | |
|     Asm->OutStreamer->AddComment(dwarf::IndexString(AttrEnc.Index));
 | |
|     switch (AttrEnc.Index) {
 | |
|     case dwarf::DW_IDX_compile_unit: {
 | |
|       DIEInteger ID(getCUIndexForEntry(Entry));
 | |
|       ID.emitValue(Asm, AttrEnc.Form);
 | |
|       break;
 | |
|     }
 | |
|     case dwarf::DW_IDX_die_offset:
 | |
|       assert(AttrEnc.Form == dwarf::DW_FORM_ref4);
 | |
|       Asm->emitInt32(Entry.getDieOffset());
 | |
|       break;
 | |
|     default:
 | |
|       llvm_unreachable("Unexpected index attribute!");
 | |
|     }
 | |
|   }
 | |
| }
 | |
| 
 | |
| template <typename DataT> void Dwarf5AccelTableWriter<DataT>::emitData() const {
 | |
|   Asm->OutStreamer->emitLabel(EntryPool);
 | |
|   for (auto &Bucket : Contents.getBuckets()) {
 | |
|     for (auto *Hash : Bucket) {
 | |
|       // Remember to emit the label for our offset.
 | |
|       Asm->OutStreamer->emitLabel(Hash->Sym);
 | |
|       for (const auto *Value : Hash->Values)
 | |
|         emitEntry(*static_cast<const DataT *>(Value));
 | |
|       Asm->OutStreamer->AddComment("End of list: " + Hash->Name.getString());
 | |
|       Asm->emitInt8(0);
 | |
|     }
 | |
|   }
 | |
| }
 | |
| 
 | |
| template <typename DataT>
 | |
| Dwarf5AccelTableWriter<DataT>::Dwarf5AccelTableWriter(
 | |
|     AsmPrinter *Asm, const AccelTableBase &Contents,
 | |
|     ArrayRef<MCSymbol *> CompUnits,
 | |
|     llvm::function_ref<unsigned(const DataT &)> getCUIndexForEntry)
 | |
|     : AccelTableWriter(Asm, Contents, false),
 | |
|       Header(CompUnits.size(), Contents.getBucketCount(),
 | |
|              Contents.getUniqueNameCount()),
 | |
|       CompUnits(CompUnits), getCUIndexForEntry(std::move(getCUIndexForEntry)) {
 | |
|   DenseSet<uint32_t> UniqueTags = getUniqueTags();
 | |
|   SmallVector<AttributeEncoding, 2> UniformAttributes = getUniformAttributes();
 | |
| 
 | |
|   Abbreviations.reserve(UniqueTags.size());
 | |
|   for (uint32_t Tag : UniqueTags)
 | |
|     Abbreviations.try_emplace(Tag, UniformAttributes);
 | |
| }
 | |
| 
 | |
| template <typename DataT> void Dwarf5AccelTableWriter<DataT>::emit() {
 | |
|   Header.emit(*this);
 | |
|   emitCUList();
 | |
|   emitBuckets();
 | |
|   emitHashes();
 | |
|   emitStringOffsets();
 | |
|   emitOffsets(EntryPool);
 | |
|   emitAbbrevs();
 | |
|   emitData();
 | |
|   Asm->OutStreamer->emitValueToAlignment(4, 0);
 | |
|   Asm->OutStreamer->emitLabel(ContributionEnd);
 | |
| }
 | |
| 
 | |
| void llvm::emitAppleAccelTableImpl(AsmPrinter *Asm, AccelTableBase &Contents,
 | |
|                                    StringRef Prefix, const MCSymbol *SecBegin,
 | |
|                                    ArrayRef<AppleAccelTableData::Atom> Atoms) {
 | |
|   Contents.finalize(Asm, Prefix);
 | |
|   AppleAccelTableWriter(Asm, Contents, Atoms, SecBegin).emit();
 | |
| }
 | |
| 
 | |
| void llvm::emitDWARF5AccelTable(
 | |
|     AsmPrinter *Asm, AccelTable<DWARF5AccelTableData> &Contents,
 | |
|     const DwarfDebug &DD, ArrayRef<std::unique_ptr<DwarfCompileUnit>> CUs) {
 | |
|   std::vector<MCSymbol *> CompUnits;
 | |
|   SmallVector<unsigned, 1> CUIndex(CUs.size());
 | |
|   int Count = 0;
 | |
|   for (const auto &CU : enumerate(CUs)) {
 | |
|     if (CU.value()->getCUNode()->getNameTableKind() !=
 | |
|         DICompileUnit::DebugNameTableKind::Default)
 | |
|       continue;
 | |
|     CUIndex[CU.index()] = Count++;
 | |
|     assert(CU.index() == CU.value()->getUniqueID());
 | |
|     const DwarfCompileUnit *MainCU =
 | |
|         DD.useSplitDwarf() ? CU.value()->getSkeleton() : CU.value().get();
 | |
|     CompUnits.push_back(MainCU->getLabelBegin());
 | |
|   }
 | |
| 
 | |
|   if (CompUnits.empty())
 | |
|     return;
 | |
| 
 | |
|   Asm->OutStreamer->switchSection(
 | |
|       Asm->getObjFileLowering().getDwarfDebugNamesSection());
 | |
| 
 | |
|   Contents.finalize(Asm, "names");
 | |
|   Dwarf5AccelTableWriter<DWARF5AccelTableData>(
 | |
|       Asm, Contents, CompUnits,
 | |
|       [&](const DWARF5AccelTableData &Entry) {
 | |
|         const DIE *CUDie = Entry.getDie().getUnitDie();
 | |
|         return CUIndex[DD.lookupCU(CUDie)->getUniqueID()];
 | |
|       })
 | |
|       .emit();
 | |
| }
 | |
| 
 | |
| void llvm::emitDWARF5AccelTable(
 | |
|     AsmPrinter *Asm, AccelTable<DWARF5AccelTableStaticData> &Contents,
 | |
|     ArrayRef<MCSymbol *> CUs,
 | |
|     llvm::function_ref<unsigned(const DWARF5AccelTableStaticData &)>
 | |
|         getCUIndexForEntry) {
 | |
|   Contents.finalize(Asm, "names");
 | |
|   Dwarf5AccelTableWriter<DWARF5AccelTableStaticData>(Asm, Contents, CUs,
 | |
|                                                      getCUIndexForEntry)
 | |
|       .emit();
 | |
| }
 | |
| 
 | |
| void AppleAccelTableOffsetData::emit(AsmPrinter *Asm) const {
 | |
|   assert(Die.getDebugSectionOffset() <= UINT32_MAX &&
 | |
|          "The section offset exceeds the limit.");
 | |
|   Asm->emitInt32(Die.getDebugSectionOffset());
 | |
| }
 | |
| 
 | |
| void AppleAccelTableTypeData::emit(AsmPrinter *Asm) const {
 | |
|   assert(Die.getDebugSectionOffset() <= UINT32_MAX &&
 | |
|          "The section offset exceeds the limit.");
 | |
|   Asm->emitInt32(Die.getDebugSectionOffset());
 | |
|   Asm->emitInt16(Die.getTag());
 | |
|   Asm->emitInt8(0);
 | |
| }
 | |
| 
 | |
| void AppleAccelTableStaticOffsetData::emit(AsmPrinter *Asm) const {
 | |
|   Asm->emitInt32(Offset);
 | |
| }
 | |
| 
 | |
| void AppleAccelTableStaticTypeData::emit(AsmPrinter *Asm) const {
 | |
|   Asm->emitInt32(Offset);
 | |
|   Asm->emitInt16(Tag);
 | |
|   Asm->emitInt8(ObjCClassIsImplementation ? dwarf::DW_FLAG_type_implementation
 | |
|                                           : 0);
 | |
|   Asm->emitInt32(QualifiedNameHash);
 | |
| }
 | |
| 
 | |
| constexpr AppleAccelTableData::Atom AppleAccelTableTypeData::Atoms[];
 | |
| constexpr AppleAccelTableData::Atom AppleAccelTableOffsetData::Atoms[];
 | |
| constexpr AppleAccelTableData::Atom AppleAccelTableStaticOffsetData::Atoms[];
 | |
| constexpr AppleAccelTableData::Atom AppleAccelTableStaticTypeData::Atoms[];
 | |
| 
 | |
| #ifndef NDEBUG
 | |
| void AppleAccelTableWriter::Header::print(raw_ostream &OS) const {
 | |
|   OS << "Magic: " << format("0x%x", Magic) << "\n"
 | |
|      << "Version: " << Version << "\n"
 | |
|      << "Hash Function: " << HashFunction << "\n"
 | |
|      << "Bucket Count: " << BucketCount << "\n"
 | |
|      << "Header Data Length: " << HeaderDataLength << "\n";
 | |
| }
 | |
| 
 | |
| void AppleAccelTableData::Atom::print(raw_ostream &OS) const {
 | |
|   OS << "Type: " << dwarf::AtomTypeString(Type) << "\n"
 | |
|      << "Form: " << dwarf::FormEncodingString(Form) << "\n";
 | |
| }
 | |
| 
 | |
| void AppleAccelTableWriter::HeaderData::print(raw_ostream &OS) const {
 | |
|   OS << "DIE Offset Base: " << DieOffsetBase << "\n";
 | |
|   for (auto Atom : Atoms)
 | |
|     Atom.print(OS);
 | |
| }
 | |
| 
 | |
| void AppleAccelTableWriter::print(raw_ostream &OS) const {
 | |
|   Header.print(OS);
 | |
|   HeaderData.print(OS);
 | |
|   Contents.print(OS);
 | |
|   SecBegin->print(OS, nullptr);
 | |
| }
 | |
| 
 | |
| void AccelTableBase::HashData::print(raw_ostream &OS) const {
 | |
|   OS << "Name: " << Name.getString() << "\n";
 | |
|   OS << "  Hash Value: " << format("0x%x", HashValue) << "\n";
 | |
|   OS << "  Symbol: ";
 | |
|   if (Sym)
 | |
|     OS << *Sym;
 | |
|   else
 | |
|     OS << "<none>";
 | |
|   OS << "\n";
 | |
|   for (auto *Value : Values)
 | |
|     Value->print(OS);
 | |
| }
 | |
| 
 | |
| void AccelTableBase::print(raw_ostream &OS) const {
 | |
|   // Print Content.
 | |
|   OS << "Entries: \n";
 | |
|   for (const auto &Entry : Entries) {
 | |
|     OS << "Name: " << Entry.first() << "\n";
 | |
|     for (auto *V : Entry.second.Values)
 | |
|       V->print(OS);
 | |
|   }
 | |
| 
 | |
|   OS << "Buckets and Hashes: \n";
 | |
|   for (const auto &Bucket : Buckets)
 | |
|     for (const auto &Hash : Bucket)
 | |
|       Hash->print(OS);
 | |
| 
 | |
|   OS << "Data: \n";
 | |
|   for (const auto &E : Entries)
 | |
|     E.second.print(OS);
 | |
| }
 | |
| 
 | |
| void DWARF5AccelTableData::print(raw_ostream &OS) const {
 | |
|   OS << "  Offset: " << getDieOffset() << "\n";
 | |
|   OS << "  Tag: " << dwarf::TagString(getDieTag()) << "\n";
 | |
| }
 | |
| 
 | |
| void DWARF5AccelTableStaticData::print(raw_ostream &OS) const {
 | |
|   OS << "  Offset: " << getDieOffset() << "\n";
 | |
|   OS << "  Tag: " << dwarf::TagString(getDieTag()) << "\n";
 | |
| }
 | |
| 
 | |
| void AppleAccelTableOffsetData::print(raw_ostream &OS) const {
 | |
|   OS << "  Offset: " << Die.getOffset() << "\n";
 | |
| }
 | |
| 
 | |
| void AppleAccelTableTypeData::print(raw_ostream &OS) const {
 | |
|   OS << "  Offset: " << Die.getOffset() << "\n";
 | |
|   OS << "  Tag: " << dwarf::TagString(Die.getTag()) << "\n";
 | |
| }
 | |
| 
 | |
| void AppleAccelTableStaticOffsetData::print(raw_ostream &OS) const {
 | |
|   OS << "  Static Offset: " << Offset << "\n";
 | |
| }
 | |
| 
 | |
| void AppleAccelTableStaticTypeData::print(raw_ostream &OS) const {
 | |
|   OS << "  Static Offset: " << Offset << "\n";
 | |
|   OS << "  QualifiedNameHash: " << format("%x\n", QualifiedNameHash) << "\n";
 | |
|   OS << "  Tag: " << dwarf::TagString(Tag) << "\n";
 | |
|   OS << "  ObjCClassIsImplementation: "
 | |
|      << (ObjCClassIsImplementation ? "true" : "false");
 | |
|   OS << "\n";
 | |
| }
 | |
| #endif
 |