1855fc3bbSJonas Devlieghere //===- llvm/CodeGen/AsmPrinter/AccelTable.cpp - Accelerator Tables --------===//
2855fc3bbSJonas Devlieghere //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6855fc3bbSJonas Devlieghere //
7855fc3bbSJonas Devlieghere //===----------------------------------------------------------------------===//
8855fc3bbSJonas Devlieghere //
9855fc3bbSJonas Devlieghere // This file contains support for writing accelerator tables.
10855fc3bbSJonas Devlieghere //
11855fc3bbSJonas Devlieghere //===----------------------------------------------------------------------===//
12855fc3bbSJonas Devlieghere 
13855fc3bbSJonas Devlieghere #include "llvm/CodeGen/AccelTable.h"
146088c234SPavel Labath #include "DwarfCompileUnit.h"
15855fc3bbSJonas Devlieghere #include "llvm/ADT/STLExtras.h"
16855fc3bbSJonas Devlieghere #include "llvm/ADT/StringMap.h"
17855fc3bbSJonas Devlieghere #include "llvm/ADT/Twine.h"
18855fc3bbSJonas Devlieghere #include "llvm/BinaryFormat/Dwarf.h"
19855fc3bbSJonas Devlieghere #include "llvm/CodeGen/AsmPrinter.h"
20855fc3bbSJonas Devlieghere #include "llvm/CodeGen/DIE.h"
21855fc3bbSJonas Devlieghere #include "llvm/MC/MCStreamer.h"
2298062cb3SJonas Devlieghere #include "llvm/MC/MCSymbol.h"
23855fc3bbSJonas Devlieghere #include "llvm/Support/raw_ostream.h"
2466cf14d0SDavid Blaikie #include "llvm/Target/TargetLoweringObjectFile.h"
25855fc3bbSJonas Devlieghere #include <algorithm>
26855fc3bbSJonas Devlieghere #include <cstddef>
27855fc3bbSJonas Devlieghere #include <cstdint>
28855fc3bbSJonas Devlieghere #include <limits>
29855fc3bbSJonas Devlieghere #include <vector>
30855fc3bbSJonas Devlieghere 
31855fc3bbSJonas Devlieghere using namespace llvm;
32855fc3bbSJonas Devlieghere 
computeBucketCount()33a7c457d2SPavel Labath void AccelTableBase::computeBucketCount() {
34a7c457d2SPavel Labath   // First get the number of unique hashes.
35a7c457d2SPavel Labath   std::vector<uint32_t> Uniques;
36a7c457d2SPavel Labath   Uniques.reserve(Entries.size());
37a7c457d2SPavel Labath   for (const auto &E : Entries)
38a7c457d2SPavel Labath     Uniques.push_back(E.second.HashValue);
39a7c457d2SPavel Labath   array_pod_sort(Uniques.begin(), Uniques.end());
40a7c457d2SPavel Labath   std::vector<uint32_t>::iterator P =
41a7c457d2SPavel Labath       std::unique(Uniques.begin(), Uniques.end());
42a7c457d2SPavel Labath 
43a7c457d2SPavel Labath   UniqueHashCount = std::distance(Uniques.begin(), P);
44a7c457d2SPavel Labath 
45a7c457d2SPavel Labath   if (UniqueHashCount > 1024)
46a7c457d2SPavel Labath     BucketCount = UniqueHashCount / 4;
47a7c457d2SPavel Labath   else if (UniqueHashCount > 16)
48a7c457d2SPavel Labath     BucketCount = UniqueHashCount / 2;
49a7c457d2SPavel Labath   else
50a7c457d2SPavel Labath     BucketCount = std::max<uint32_t>(UniqueHashCount, 1);
51a7c457d2SPavel Labath }
52a7c457d2SPavel Labath 
finalize(AsmPrinter * Asm,StringRef Prefix)53a7c457d2SPavel Labath void AccelTableBase::finalize(AsmPrinter *Asm, StringRef Prefix) {
54a7c457d2SPavel Labath   // Create the individual hash data outputs.
55a7c457d2SPavel Labath   for (auto &E : Entries) {
56a7c457d2SPavel Labath     // Unique the entries.
57efd94c56SFangrui Song     llvm::stable_sort(E.second.Values,
58a7c457d2SPavel Labath                       [](const AccelTableData *A, const AccelTableData *B) {
59a7c457d2SPavel Labath                         return *A < *B;
60a7c457d2SPavel Labath                       });
61a7c457d2SPavel Labath     E.second.Values.erase(
62a7c457d2SPavel Labath         std::unique(E.second.Values.begin(), E.second.Values.end()),
63a7c457d2SPavel Labath         E.second.Values.end());
64a7c457d2SPavel Labath   }
65a7c457d2SPavel Labath 
66a7c457d2SPavel Labath   // Figure out how many buckets we need, then compute the bucket contents and
67a7c457d2SPavel Labath   // the final ordering. The hashes and offsets can be emitted by walking these
68a7c457d2SPavel Labath   // data structures. We add temporary symbols to the data so they can be
69a7c457d2SPavel Labath   // referenced when emitting the offsets.
70a7c457d2SPavel Labath   computeBucketCount();
71a7c457d2SPavel Labath 
72a7c457d2SPavel Labath   // Compute bucket contents and final ordering.
73a7c457d2SPavel Labath   Buckets.resize(BucketCount);
74a7c457d2SPavel Labath   for (auto &E : Entries) {
75a7c457d2SPavel Labath     uint32_t Bucket = E.second.HashValue % BucketCount;
76a7c457d2SPavel Labath     Buckets[Bucket].push_back(&E.second);
77a7c457d2SPavel Labath     E.second.Sym = Asm->createTempSymbol(Prefix);
78a7c457d2SPavel Labath   }
79a7c457d2SPavel Labath 
80a7c457d2SPavel Labath   // Sort the contents of the buckets by hash value so that hash collisions end
81a7c457d2SPavel Labath   // up together. Stable sort makes testing easier and doesn't cost much more.
82a7c457d2SPavel Labath   for (auto &Bucket : Buckets)
83efd94c56SFangrui Song     llvm::stable_sort(Bucket, [](HashData *LHS, HashData *RHS) {
84a7c457d2SPavel Labath       return LHS->HashValue < RHS->HashValue;
85a7c457d2SPavel Labath     });
86a7c457d2SPavel Labath }
87a7c457d2SPavel Labath 
88a7c457d2SPavel Labath namespace {
896088c234SPavel Labath /// Base class for writing out Accelerator tables. It holds the common
906088c234SPavel Labath /// functionality for the two Accelerator table types.
91e60ca777SJonas Devlieghere class AccelTableWriter {
92a7c457d2SPavel Labath protected:
93a7c457d2SPavel Labath   AsmPrinter *const Asm;          ///< Destination.
94a7c457d2SPavel Labath   const AccelTableBase &Contents; ///< Data to emit.
95a7c457d2SPavel Labath 
96a7c457d2SPavel Labath   /// Controls whether to emit duplicate hash and offset table entries for names
97a7c457d2SPavel Labath   /// with identical hashes. Apple tables don't emit duplicate entries, DWARF v5
98a7c457d2SPavel Labath   /// tables do.
99a7c457d2SPavel Labath   const bool SkipIdenticalHashes;
100a7c457d2SPavel Labath 
101a7c457d2SPavel Labath   void emitHashes() const;
102a7c457d2SPavel Labath 
103a7c457d2SPavel Labath   /// Emit offsets to lists of entries with identical names. The offsets are
104a7c457d2SPavel Labath   /// relative to the Base argument.
105a7c457d2SPavel Labath   void emitOffsets(const MCSymbol *Base) const;
106a7c457d2SPavel Labath 
107a7c457d2SPavel Labath public:
AccelTableWriter(AsmPrinter * Asm,const AccelTableBase & Contents,bool SkipIdenticalHashes)108e60ca777SJonas Devlieghere   AccelTableWriter(AsmPrinter *Asm, const AccelTableBase &Contents,
109a7c457d2SPavel Labath                    bool SkipIdenticalHashes)
110a7c457d2SPavel Labath       : Asm(Asm), Contents(Contents), SkipIdenticalHashes(SkipIdenticalHashes) {
111a7c457d2SPavel Labath   }
112a7c457d2SPavel Labath };
113a7c457d2SPavel Labath 
114e60ca777SJonas Devlieghere class AppleAccelTableWriter : public AccelTableWriter {
115a7c457d2SPavel Labath   using Atom = AppleAccelTableData::Atom;
116a7c457d2SPavel Labath 
117a7c457d2SPavel Labath   /// The fixed header of an Apple Accelerator Table.
118a7c457d2SPavel Labath   struct Header {
119a7c457d2SPavel Labath     uint32_t Magic = MagicHash;
120a7c457d2SPavel Labath     uint16_t Version = 1;
121a7c457d2SPavel Labath     uint16_t HashFunction = dwarf::DW_hash_function_djb;
122a7c457d2SPavel Labath     uint32_t BucketCount;
123a7c457d2SPavel Labath     uint32_t HashCount;
124a7c457d2SPavel Labath     uint32_t HeaderDataLength;
125a7c457d2SPavel Labath 
126a7c457d2SPavel Labath     /// 'HASH' magic value to detect endianness.
127a7c457d2SPavel Labath     static const uint32_t MagicHash = 0x48415348;
128a7c457d2SPavel Labath 
Header__anon796e91580311::AppleAccelTableWriter::Header129a7c457d2SPavel Labath     Header(uint32_t BucketCount, uint32_t UniqueHashCount, uint32_t DataLength)
130a7c457d2SPavel Labath         : BucketCount(BucketCount), HashCount(UniqueHashCount),
131a7c457d2SPavel Labath           HeaderDataLength(DataLength) {}
132a7c457d2SPavel Labath 
133a7c457d2SPavel Labath     void emit(AsmPrinter *Asm) const;
134a7c457d2SPavel Labath #ifndef NDEBUG
135a7c457d2SPavel Labath     void print(raw_ostream &OS) const;
dump__anon796e91580311::AppleAccelTableWriter::Header136a7c457d2SPavel Labath     void dump() const { print(dbgs()); }
137a7c457d2SPavel Labath #endif
138a7c457d2SPavel Labath   };
139a7c457d2SPavel Labath 
140a7c457d2SPavel Labath   /// The HeaderData describes the structure of an Apple accelerator table
141a7c457d2SPavel Labath   /// through a list of Atoms.
142a7c457d2SPavel Labath   struct HeaderData {
143a7c457d2SPavel Labath     /// In the case of data that is referenced via DW_FORM_ref_* the offset
144a7c457d2SPavel Labath     /// base is used to describe the offset for all forms in the list of atoms.
145a7c457d2SPavel Labath     uint32_t DieOffsetBase;
146a7c457d2SPavel Labath 
147a7c457d2SPavel Labath     const SmallVector<Atom, 4> Atoms;
148a7c457d2SPavel Labath 
HeaderData__anon796e91580311::AppleAccelTableWriter::HeaderData149a7c457d2SPavel Labath     HeaderData(ArrayRef<Atom> AtomList, uint32_t Offset = 0)
150a7c457d2SPavel Labath         : DieOffsetBase(Offset), Atoms(AtomList.begin(), AtomList.end()) {}
151a7c457d2SPavel Labath 
152a7c457d2SPavel Labath     void emit(AsmPrinter *Asm) const;
153a7c457d2SPavel Labath #ifndef NDEBUG
154a7c457d2SPavel Labath     void print(raw_ostream &OS) const;
dump__anon796e91580311::AppleAccelTableWriter::HeaderData155a7c457d2SPavel Labath     void dump() const { print(dbgs()); }
156a7c457d2SPavel Labath #endif
157a7c457d2SPavel Labath   };
158a7c457d2SPavel Labath 
159a7c457d2SPavel Labath   Header Header;
160a7c457d2SPavel Labath   HeaderData HeaderData;
161a7c457d2SPavel Labath   const MCSymbol *SecBegin;
162a7c457d2SPavel Labath 
163a7c457d2SPavel Labath   void emitBuckets() const;
164a7c457d2SPavel Labath   void emitData() const;
165a7c457d2SPavel Labath 
166a7c457d2SPavel Labath public:
AppleAccelTableWriter(AsmPrinter * Asm,const AccelTableBase & Contents,ArrayRef<Atom> Atoms,const MCSymbol * SecBegin)167e60ca777SJonas Devlieghere   AppleAccelTableWriter(AsmPrinter *Asm, const AccelTableBase &Contents,
168a7c457d2SPavel Labath                         ArrayRef<Atom> Atoms, const MCSymbol *SecBegin)
169e60ca777SJonas Devlieghere       : AccelTableWriter(Asm, Contents, true),
170a7c457d2SPavel Labath         Header(Contents.getBucketCount(), Contents.getUniqueHashCount(),
171a7c457d2SPavel Labath                8 + (Atoms.size() * 4)),
172a7c457d2SPavel Labath         HeaderData(Atoms), SecBegin(SecBegin) {}
173a7c457d2SPavel Labath 
174a7c457d2SPavel Labath   void emit() const;
175a7c457d2SPavel Labath 
176a7c457d2SPavel Labath #ifndef NDEBUG
177a7c457d2SPavel Labath   void print(raw_ostream &OS) const;
dump() const178a7c457d2SPavel Labath   void dump() const { print(dbgs()); }
179a7c457d2SPavel Labath #endif
180a7c457d2SPavel Labath };
1816088c234SPavel Labath 
18298062cb3SJonas Devlieghere /// Class responsible for emitting a DWARF v5 Accelerator Table. The only
18398062cb3SJonas Devlieghere /// public function is emit(), which performs the actual emission.
18498062cb3SJonas Devlieghere ///
18598062cb3SJonas Devlieghere /// The class is templated in its data type. This allows us to emit both dyamic
18698062cb3SJonas Devlieghere /// and static data entries. A callback abstract the logic to provide a CU
18798062cb3SJonas Devlieghere /// index for a given entry, which is different per data type, but identical
18898062cb3SJonas Devlieghere /// for every entry in the same table.
18998062cb3SJonas Devlieghere template <typename DataT>
190e60ca777SJonas Devlieghere class Dwarf5AccelTableWriter : public AccelTableWriter {
1916088c234SPavel Labath   struct Header {
1926088c234SPavel Labath     uint16_t Version = 5;
1936088c234SPavel Labath     uint16_t Padding = 0;
1946088c234SPavel Labath     uint32_t CompUnitCount;
1956088c234SPavel Labath     uint32_t LocalTypeUnitCount = 0;
1966088c234SPavel Labath     uint32_t ForeignTypeUnitCount = 0;
1976088c234SPavel Labath     uint32_t BucketCount;
1986088c234SPavel Labath     uint32_t NameCount;
1996088c234SPavel Labath     uint32_t AbbrevTableSize = 0;
2006088c234SPavel Labath     uint32_t AugmentationStringSize = sizeof(AugmentationString);
2016088c234SPavel Labath     char AugmentationString[8] = {'L', 'L', 'V', 'M', '0', '7', '0', '0'};
2026088c234SPavel Labath 
Header__anon796e91580311::Dwarf5AccelTableWriter::Header2036088c234SPavel Labath     Header(uint32_t CompUnitCount, uint32_t BucketCount, uint32_t NameCount)
2046088c234SPavel Labath         : CompUnitCount(CompUnitCount), BucketCount(BucketCount),
2056088c234SPavel Labath           NameCount(NameCount) {}
2066088c234SPavel Labath 
207d39bc36bSChen Zheng     void emit(Dwarf5AccelTableWriter &Ctx);
2086088c234SPavel Labath   };
2096088c234SPavel Labath   struct AttributeEncoding {
2106088c234SPavel Labath     dwarf::Index Index;
2116088c234SPavel Labath     dwarf::Form Form;
2126088c234SPavel Labath   };
2136088c234SPavel Labath 
2146088c234SPavel Labath   Header Header;
2156088c234SPavel Labath   DenseMap<uint32_t, SmallVector<AttributeEncoding, 2>> Abbreviations;
21698062cb3SJonas Devlieghere   ArrayRef<MCSymbol *> CompUnits;
21798062cb3SJonas Devlieghere   llvm::function_ref<unsigned(const DataT &)> getCUIndexForEntry;
218d39bc36bSChen Zheng   MCSymbol *ContributionEnd = nullptr;
2196088c234SPavel Labath   MCSymbol *AbbrevStart = Asm->createTempSymbol("names_abbrev_start");
2206088c234SPavel Labath   MCSymbol *AbbrevEnd = Asm->createTempSymbol("names_abbrev_end");
2216088c234SPavel Labath   MCSymbol *EntryPool = Asm->createTempSymbol("names_entries");
2226088c234SPavel Labath 
2236088c234SPavel Labath   DenseSet<uint32_t> getUniqueTags() const;
2246088c234SPavel Labath 
2256088c234SPavel Labath   // Right now, we emit uniform attributes for all tags.
2266088c234SPavel Labath   SmallVector<AttributeEncoding, 2> getUniformAttributes() const;
2276088c234SPavel Labath 
2286088c234SPavel Labath   void emitCUList() const;
2296088c234SPavel Labath   void emitBuckets() const;
2306088c234SPavel Labath   void emitStringOffsets() const;
2316088c234SPavel Labath   void emitAbbrevs() const;
232cb0bab86SFangrui Song   void emitEntry(const DataT &Entry) const;
2336088c234SPavel Labath   void emitData() const;
2346088c234SPavel Labath 
2356088c234SPavel Labath public:
23698062cb3SJonas Devlieghere   Dwarf5AccelTableWriter(
23798062cb3SJonas Devlieghere       AsmPrinter *Asm, const AccelTableBase &Contents,
23898062cb3SJonas Devlieghere       ArrayRef<MCSymbol *> CompUnits,
23998062cb3SJonas Devlieghere       llvm::function_ref<unsigned(const DataT &)> GetCUIndexForEntry);
2406088c234SPavel Labath 
241d39bc36bSChen Zheng   void emit();
2426088c234SPavel Labath };
243a7c457d2SPavel Labath } // namespace
244a7c457d2SPavel Labath 
emitHashes() const245e60ca777SJonas Devlieghere void AccelTableWriter::emitHashes() const {
246a7c457d2SPavel Labath   uint64_t PrevHash = std::numeric_limits<uint64_t>::max();
247a7c457d2SPavel Labath   unsigned BucketIdx = 0;
248*9e6d1f4bSKazu Hirata   for (const auto &Bucket : Contents.getBuckets()) {
249*9e6d1f4bSKazu Hirata     for (const auto &Hash : Bucket) {
250a7c457d2SPavel Labath       uint32_t HashValue = Hash->HashValue;
251a7c457d2SPavel Labath       if (SkipIdenticalHashes && PrevHash == HashValue)
252a7c457d2SPavel Labath         continue;
253a7c457d2SPavel Labath       Asm->OutStreamer->AddComment("Hash in Bucket " + Twine(BucketIdx));
2544b4d85fdSRafael Espindola       Asm->emitInt32(HashValue);
255a7c457d2SPavel Labath       PrevHash = HashValue;
256a7c457d2SPavel Labath     }
257a7c457d2SPavel Labath     BucketIdx++;
258a7c457d2SPavel Labath   }
259a7c457d2SPavel Labath }
260a7c457d2SPavel Labath 
emitOffsets(const MCSymbol * Base) const261e60ca777SJonas Devlieghere void AccelTableWriter::emitOffsets(const MCSymbol *Base) const {
262a7c457d2SPavel Labath   const auto &Buckets = Contents.getBuckets();
263a7c457d2SPavel Labath   uint64_t PrevHash = std::numeric_limits<uint64_t>::max();
264a7c457d2SPavel Labath   for (size_t i = 0, e = Buckets.size(); i < e; ++i) {
265a7c457d2SPavel Labath     for (auto *Hash : Buckets[i]) {
266a7c457d2SPavel Labath       uint32_t HashValue = Hash->HashValue;
267a7c457d2SPavel Labath       if (SkipIdenticalHashes && PrevHash == HashValue)
268a7c457d2SPavel Labath         continue;
269a7c457d2SPavel Labath       PrevHash = HashValue;
270a7c457d2SPavel Labath       Asm->OutStreamer->AddComment("Offset in Bucket " + Twine(i));
271a93dd26dSIgor Kudrin       Asm->emitLabelDifference(Hash->Sym, Base, Asm->getDwarfOffsetByteSize());
272a7c457d2SPavel Labath     }
273a7c457d2SPavel Labath   }
274a7c457d2SPavel Labath }
275a7c457d2SPavel Labath 
emit(AsmPrinter * Asm) const276e60ca777SJonas Devlieghere void AppleAccelTableWriter::Header::emit(AsmPrinter *Asm) const {
277855fc3bbSJonas Devlieghere   Asm->OutStreamer->AddComment("Header Magic");
2784b4d85fdSRafael Espindola   Asm->emitInt32(Magic);
279855fc3bbSJonas Devlieghere   Asm->OutStreamer->AddComment("Header Version");
2804b4d85fdSRafael Espindola   Asm->emitInt16(Version);
281855fc3bbSJonas Devlieghere   Asm->OutStreamer->AddComment("Header Hash Function");
2824b4d85fdSRafael Espindola   Asm->emitInt16(HashFunction);
283855fc3bbSJonas Devlieghere   Asm->OutStreamer->AddComment("Header Bucket Count");
2844b4d85fdSRafael Espindola   Asm->emitInt32(BucketCount);
285855fc3bbSJonas Devlieghere   Asm->OutStreamer->AddComment("Header Hash Count");
2864b4d85fdSRafael Espindola   Asm->emitInt32(HashCount);
287855fc3bbSJonas Devlieghere   Asm->OutStreamer->AddComment("Header Data Length");
2884b4d85fdSRafael Espindola   Asm->emitInt32(HeaderDataLength);
289a7c457d2SPavel Labath }
290855fc3bbSJonas Devlieghere 
emit(AsmPrinter * Asm) const291e60ca777SJonas Devlieghere void AppleAccelTableWriter::HeaderData::emit(AsmPrinter *Asm) const {
292855fc3bbSJonas Devlieghere   Asm->OutStreamer->AddComment("HeaderData Die Offset Base");
2934b4d85fdSRafael Espindola   Asm->emitInt32(DieOffsetBase);
294855fc3bbSJonas Devlieghere   Asm->OutStreamer->AddComment("HeaderData Atom Count");
2954b4d85fdSRafael Espindola   Asm->emitInt32(Atoms.size());
296855fc3bbSJonas Devlieghere 
297a7c457d2SPavel Labath   for (const Atom &A : Atoms) {
298855fc3bbSJonas Devlieghere     Asm->OutStreamer->AddComment(dwarf::AtomTypeString(A.Type));
2994b4d85fdSRafael Espindola     Asm->emitInt16(A.Type);
300855fc3bbSJonas Devlieghere     Asm->OutStreamer->AddComment(dwarf::FormEncodingString(A.Form));
3014b4d85fdSRafael Espindola     Asm->emitInt16(A.Form);
302855fc3bbSJonas Devlieghere   }
303855fc3bbSJonas Devlieghere }
304855fc3bbSJonas Devlieghere 
emitBuckets() const305e60ca777SJonas Devlieghere void AppleAccelTableWriter::emitBuckets() const {
306a7c457d2SPavel Labath   const auto &Buckets = Contents.getBuckets();
307855fc3bbSJonas Devlieghere   unsigned index = 0;
308855fc3bbSJonas Devlieghere   for (size_t i = 0, e = Buckets.size(); i < e; ++i) {
309855fc3bbSJonas Devlieghere     Asm->OutStreamer->AddComment("Bucket " + Twine(i));
310855fc3bbSJonas Devlieghere     if (!Buckets[i].empty())
3114b4d85fdSRafael Espindola       Asm->emitInt32(index);
312855fc3bbSJonas Devlieghere     else
3134b4d85fdSRafael Espindola       Asm->emitInt32(std::numeric_limits<uint32_t>::max());
314855fc3bbSJonas Devlieghere     // Buckets point in the list of hashes, not to the data. Do not increment
315855fc3bbSJonas Devlieghere     // the index multiple times in case of hash collisions.
316855fc3bbSJonas Devlieghere     uint64_t PrevHash = std::numeric_limits<uint64_t>::max();
317855fc3bbSJonas Devlieghere     for (auto *HD : Buckets[i]) {
318855fc3bbSJonas Devlieghere       uint32_t HashValue = HD->HashValue;
319855fc3bbSJonas Devlieghere       if (PrevHash != HashValue)
320855fc3bbSJonas Devlieghere         ++index;
321855fc3bbSJonas Devlieghere       PrevHash = HashValue;
322855fc3bbSJonas Devlieghere     }
323855fc3bbSJonas Devlieghere   }
324855fc3bbSJonas Devlieghere }
325855fc3bbSJonas Devlieghere 
emitData() const326e60ca777SJonas Devlieghere void AppleAccelTableWriter::emitData() const {
327a7c457d2SPavel Labath   const auto &Buckets = Contents.getBuckets();
328c5e90a88SKazu Hirata   for (const AccelTableBase::HashList &Bucket : Buckets) {
329855fc3bbSJonas Devlieghere     uint64_t PrevHash = std::numeric_limits<uint64_t>::max();
330*9e6d1f4bSKazu Hirata     for (const auto &Hash : Bucket) {
331855fc3bbSJonas Devlieghere       // Terminate the previous entry if there is no hash collision with the
332855fc3bbSJonas Devlieghere       // current one.
333855fc3bbSJonas Devlieghere       if (PrevHash != std::numeric_limits<uint64_t>::max() &&
334855fc3bbSJonas Devlieghere           PrevHash != Hash->HashValue)
3354b4d85fdSRafael Espindola         Asm->emitInt32(0);
336855fc3bbSJonas Devlieghere       // Remember to emit the label for our offset.
3376d2d589bSFangrui Song       Asm->OutStreamer->emitLabel(Hash->Sym);
338062eb537SPavel Labath       Asm->OutStreamer->AddComment(Hash->Name.getString());
339062eb537SPavel Labath       Asm->emitDwarfStringOffset(Hash->Name);
340855fc3bbSJonas Devlieghere       Asm->OutStreamer->AddComment("Num DIEs");
3414b4d85fdSRafael Espindola       Asm->emitInt32(Hash->Values.size());
342a7c457d2SPavel Labath       for (const auto *V : Hash->Values)
343a7c457d2SPavel Labath         static_cast<const AppleAccelTableData *>(V)->emit(Asm);
344855fc3bbSJonas Devlieghere       PrevHash = Hash->HashValue;
345855fc3bbSJonas Devlieghere     }
346855fc3bbSJonas Devlieghere     // Emit the final end marker for the bucket.
347c5e90a88SKazu Hirata     if (!Bucket.empty())
3484b4d85fdSRafael Espindola       Asm->emitInt32(0);
349855fc3bbSJonas Devlieghere   }
350855fc3bbSJonas Devlieghere }
351855fc3bbSJonas Devlieghere 
emit() const352e60ca777SJonas Devlieghere void AppleAccelTableWriter::emit() const {
353a7c457d2SPavel Labath   Header.emit(Asm);
354a7c457d2SPavel Labath   HeaderData.emit(Asm);
355a7c457d2SPavel Labath   emitBuckets();
356a7c457d2SPavel Labath   emitHashes();
357a7c457d2SPavel Labath   emitOffsets(SecBegin);
358a7c457d2SPavel Labath   emitData();
359855fc3bbSJonas Devlieghere }
360855fc3bbSJonas Devlieghere 
36198062cb3SJonas Devlieghere template <typename DataT>
emit(Dwarf5AccelTableWriter & Ctx)362d39bc36bSChen Zheng void Dwarf5AccelTableWriter<DataT>::Header::emit(Dwarf5AccelTableWriter &Ctx) {
363eadfac87SPavel Labath   assert(CompUnitCount > 0 && "Index must have at least one CU.");
364eadfac87SPavel Labath 
3656088c234SPavel Labath   AsmPrinter *Asm = Ctx.Asm;
366d39bc36bSChen Zheng   Ctx.ContributionEnd =
367d39bc36bSChen Zheng       Asm->emitDwarfUnitLength("names", "Header: unit length");
3686088c234SPavel Labath   Asm->OutStreamer->AddComment("Header: version");
3696088c234SPavel Labath   Asm->emitInt16(Version);
3706088c234SPavel Labath   Asm->OutStreamer->AddComment("Header: padding");
3716088c234SPavel Labath   Asm->emitInt16(Padding);
3726088c234SPavel Labath   Asm->OutStreamer->AddComment("Header: compilation unit count");
3736088c234SPavel Labath   Asm->emitInt32(CompUnitCount);
3746088c234SPavel Labath   Asm->OutStreamer->AddComment("Header: local type unit count");
3756088c234SPavel Labath   Asm->emitInt32(LocalTypeUnitCount);
3766088c234SPavel Labath   Asm->OutStreamer->AddComment("Header: foreign type unit count");
3776088c234SPavel Labath   Asm->emitInt32(ForeignTypeUnitCount);
3786088c234SPavel Labath   Asm->OutStreamer->AddComment("Header: bucket count");
3796088c234SPavel Labath   Asm->emitInt32(BucketCount);
3806088c234SPavel Labath   Asm->OutStreamer->AddComment("Header: name count");
3816088c234SPavel Labath   Asm->emitInt32(NameCount);
3826088c234SPavel Labath   Asm->OutStreamer->AddComment("Header: abbreviation table size");
3830bc77a0fSFangrui Song   Asm->emitLabelDifference(Ctx.AbbrevEnd, Ctx.AbbrevStart, sizeof(uint32_t));
3846088c234SPavel Labath   Asm->OutStreamer->AddComment("Header: augmentation string size");
3856088c234SPavel Labath   assert(AugmentationStringSize % 4 == 0);
3866088c234SPavel Labath   Asm->emitInt32(AugmentationStringSize);
3876088c234SPavel Labath   Asm->OutStreamer->AddComment("Header: augmentation string");
388a55daa14SFangrui Song   Asm->OutStreamer->emitBytes({AugmentationString, AugmentationStringSize});
3896088c234SPavel Labath }
3906088c234SPavel Labath 
39198062cb3SJonas Devlieghere template <typename DataT>
getUniqueTags() const39298062cb3SJonas Devlieghere DenseSet<uint32_t> Dwarf5AccelTableWriter<DataT>::getUniqueTags() const {
3936088c234SPavel Labath   DenseSet<uint32_t> UniqueTags;
3946088c234SPavel Labath   for (auto &Bucket : Contents.getBuckets()) {
3956088c234SPavel Labath     for (auto *Hash : Bucket) {
3966088c234SPavel Labath       for (auto *Value : Hash->Values) {
39798062cb3SJonas Devlieghere         unsigned Tag = static_cast<const DataT *>(Value)->getDieTag();
39898062cb3SJonas Devlieghere         UniqueTags.insert(Tag);
3996088c234SPavel Labath       }
4006088c234SPavel Labath     }
4016088c234SPavel Labath   }
4026088c234SPavel Labath   return UniqueTags;
4036088c234SPavel Labath }
4046088c234SPavel Labath 
40598062cb3SJonas Devlieghere template <typename DataT>
40698062cb3SJonas Devlieghere SmallVector<typename Dwarf5AccelTableWriter<DataT>::AttributeEncoding, 2>
getUniformAttributes() const40798062cb3SJonas Devlieghere Dwarf5AccelTableWriter<DataT>::getUniformAttributes() const {
4086088c234SPavel Labath   SmallVector<AttributeEncoding, 2> UA;
4093192e35bSJonas Devlieghere   if (CompUnits.size() > 1) {
4103192e35bSJonas Devlieghere     size_t LargestCUIndex = CompUnits.size() - 1;
4116088c234SPavel Labath     dwarf::Form Form = DIEInteger::BestForm(/*IsSigned*/ false, LargestCUIndex);
4126088c234SPavel Labath     UA.push_back({dwarf::DW_IDX_compile_unit, Form});
4136088c234SPavel Labath   }
4146088c234SPavel Labath   UA.push_back({dwarf::DW_IDX_die_offset, dwarf::DW_FORM_ref4});
4156088c234SPavel Labath   return UA;
4166088c234SPavel Labath }
4176088c234SPavel Labath 
41898062cb3SJonas Devlieghere template <typename DataT>
emitCUList() const41998062cb3SJonas Devlieghere void Dwarf5AccelTableWriter<DataT>::emitCUList() const {
4203192e35bSJonas Devlieghere   for (const auto &CU : enumerate(CompUnits)) {
4213192e35bSJonas Devlieghere     Asm->OutStreamer->AddComment("Compilation unit " + Twine(CU.index()));
42298062cb3SJonas Devlieghere     Asm->emitDwarfSymbolReference(CU.value());
4236088c234SPavel Labath   }
4246088c234SPavel Labath }
4256088c234SPavel Labath 
42698062cb3SJonas Devlieghere template <typename DataT>
emitBuckets() const42798062cb3SJonas Devlieghere void Dwarf5AccelTableWriter<DataT>::emitBuckets() const {
4286088c234SPavel Labath   uint32_t Index = 1;
4296088c234SPavel Labath   for (const auto &Bucket : enumerate(Contents.getBuckets())) {
4306088c234SPavel Labath     Asm->OutStreamer->AddComment("Bucket " + Twine(Bucket.index()));
4316088c234SPavel Labath     Asm->emitInt32(Bucket.value().empty() ? 0 : Index);
4326088c234SPavel Labath     Index += Bucket.value().size();
4336088c234SPavel Labath   }
4346088c234SPavel Labath }
4356088c234SPavel Labath 
43698062cb3SJonas Devlieghere template <typename DataT>
emitStringOffsets() const43798062cb3SJonas Devlieghere void Dwarf5AccelTableWriter<DataT>::emitStringOffsets() const {
4386088c234SPavel Labath   for (const auto &Bucket : enumerate(Contents.getBuckets())) {
4396088c234SPavel Labath     for (auto *Hash : Bucket.value()) {
4406088c234SPavel Labath       DwarfStringPoolEntryRef String = Hash->Name;
4416088c234SPavel Labath       Asm->OutStreamer->AddComment("String in Bucket " + Twine(Bucket.index()) +
4426088c234SPavel Labath                                    ": " + String.getString());
4436088c234SPavel Labath       Asm->emitDwarfStringOffset(String);
4446088c234SPavel Labath     }
4456088c234SPavel Labath   }
4466088c234SPavel Labath }
4476088c234SPavel Labath 
44898062cb3SJonas Devlieghere template <typename DataT>
emitAbbrevs() const44998062cb3SJonas Devlieghere void Dwarf5AccelTableWriter<DataT>::emitAbbrevs() const {
4506d2d589bSFangrui Song   Asm->OutStreamer->emitLabel(AbbrevStart);
4516088c234SPavel Labath   for (const auto &Abbrev : Abbreviations) {
4526088c234SPavel Labath     Asm->OutStreamer->AddComment("Abbrev code");
4536088c234SPavel Labath     assert(Abbrev.first != 0);
4540bc77a0fSFangrui Song     Asm->emitULEB128(Abbrev.first);
4556088c234SPavel Labath     Asm->OutStreamer->AddComment(dwarf::TagString(Abbrev.first));
4560bc77a0fSFangrui Song     Asm->emitULEB128(Abbrev.first);
4576088c234SPavel Labath     for (const auto &AttrEnc : Abbrev.second) {
4580bc77a0fSFangrui Song       Asm->emitULEB128(AttrEnc.Index, dwarf::IndexString(AttrEnc.Index).data());
4590bc77a0fSFangrui Song       Asm->emitULEB128(AttrEnc.Form,
4606088c234SPavel Labath                        dwarf::FormEncodingString(AttrEnc.Form).data());
4616088c234SPavel Labath     }
4620bc77a0fSFangrui Song     Asm->emitULEB128(0, "End of abbrev");
4630bc77a0fSFangrui Song     Asm->emitULEB128(0, "End of abbrev");
4646088c234SPavel Labath   }
4650bc77a0fSFangrui Song   Asm->emitULEB128(0, "End of abbrev list");
4666d2d589bSFangrui Song   Asm->OutStreamer->emitLabel(AbbrevEnd);
4676088c234SPavel Labath }
4686088c234SPavel Labath 
46998062cb3SJonas Devlieghere template <typename DataT>
emitEntry(const DataT & Entry) const47098062cb3SJonas Devlieghere void Dwarf5AccelTableWriter<DataT>::emitEntry(const DataT &Entry) const {
47198062cb3SJonas Devlieghere   auto AbbrevIt = Abbreviations.find(Entry.getDieTag());
4726088c234SPavel Labath   assert(AbbrevIt != Abbreviations.end() &&
4736088c234SPavel Labath          "Why wasn't this abbrev generated?");
4746088c234SPavel Labath 
4750bc77a0fSFangrui Song   Asm->emitULEB128(AbbrevIt->first, "Abbreviation code");
4766088c234SPavel Labath   for (const auto &AttrEnc : AbbrevIt->second) {
4776088c234SPavel Labath     Asm->OutStreamer->AddComment(dwarf::IndexString(AttrEnc.Index));
4786088c234SPavel Labath     switch (AttrEnc.Index) {
4796088c234SPavel Labath     case dwarf::DW_IDX_compile_unit: {
48098062cb3SJonas Devlieghere       DIEInteger ID(getCUIndexForEntry(Entry));
48177497103SFangrui Song       ID.emitValue(Asm, AttrEnc.Form);
4826088c234SPavel Labath       break;
4836088c234SPavel Labath     }
4846088c234SPavel Labath     case dwarf::DW_IDX_die_offset:
4856088c234SPavel Labath       assert(AttrEnc.Form == dwarf::DW_FORM_ref4);
48698062cb3SJonas Devlieghere       Asm->emitInt32(Entry.getDieOffset());
4876088c234SPavel Labath       break;
4886088c234SPavel Labath     default:
4896088c234SPavel Labath       llvm_unreachable("Unexpected index attribute!");
4906088c234SPavel Labath     }
4916088c234SPavel Labath   }
4926088c234SPavel Labath }
4936088c234SPavel Labath 
emitData() const49498062cb3SJonas Devlieghere template <typename DataT> void Dwarf5AccelTableWriter<DataT>::emitData() const {
4956d2d589bSFangrui Song   Asm->OutStreamer->emitLabel(EntryPool);
4966088c234SPavel Labath   for (auto &Bucket : Contents.getBuckets()) {
4976088c234SPavel Labath     for (auto *Hash : Bucket) {
4986088c234SPavel Labath       // Remember to emit the label for our offset.
4996d2d589bSFangrui Song       Asm->OutStreamer->emitLabel(Hash->Sym);
5006088c234SPavel Labath       for (const auto *Value : Hash->Values)
50198062cb3SJonas Devlieghere         emitEntry(*static_cast<const DataT *>(Value));
5026088c234SPavel Labath       Asm->OutStreamer->AddComment("End of list: " + Hash->Name.getString());
5033445ec9bSIgor Kudrin       Asm->emitInt8(0);
5046088c234SPavel Labath     }
5056088c234SPavel Labath   }
5066088c234SPavel Labath }
5076088c234SPavel Labath 
50898062cb3SJonas Devlieghere template <typename DataT>
Dwarf5AccelTableWriter(AsmPrinter * Asm,const AccelTableBase & Contents,ArrayRef<MCSymbol * > CompUnits,llvm::function_ref<unsigned (const DataT &)> getCUIndexForEntry)50998062cb3SJonas Devlieghere Dwarf5AccelTableWriter<DataT>::Dwarf5AccelTableWriter(
51098062cb3SJonas Devlieghere     AsmPrinter *Asm, const AccelTableBase &Contents,
51198062cb3SJonas Devlieghere     ArrayRef<MCSymbol *> CompUnits,
51298062cb3SJonas Devlieghere     llvm::function_ref<unsigned(const DataT &)> getCUIndexForEntry)
513e60ca777SJonas Devlieghere     : AccelTableWriter(Asm, Contents, false),
5143192e35bSJonas Devlieghere       Header(CompUnits.size(), Contents.getBucketCount(),
5156088c234SPavel Labath              Contents.getUniqueNameCount()),
51698062cb3SJonas Devlieghere       CompUnits(CompUnits), getCUIndexForEntry(std::move(getCUIndexForEntry)) {
5176088c234SPavel Labath   DenseSet<uint32_t> UniqueTags = getUniqueTags();
5186088c234SPavel Labath   SmallVector<AttributeEncoding, 2> UniformAttributes = getUniformAttributes();
5196088c234SPavel Labath 
5206088c234SPavel Labath   Abbreviations.reserve(UniqueTags.size());
5216088c234SPavel Labath   for (uint32_t Tag : UniqueTags)
5226088c234SPavel Labath     Abbreviations.try_emplace(Tag, UniformAttributes);
5236088c234SPavel Labath }
5246088c234SPavel Labath 
emit()525d39bc36bSChen Zheng template <typename DataT> void Dwarf5AccelTableWriter<DataT>::emit() {
5266088c234SPavel Labath   Header.emit(*this);
5276088c234SPavel Labath   emitCUList();
5286088c234SPavel Labath   emitBuckets();
5296088c234SPavel Labath   emitHashes();
5306088c234SPavel Labath   emitStringOffsets();
5316088c234SPavel Labath   emitOffsets(EntryPool);
5326088c234SPavel Labath   emitAbbrevs();
5336088c234SPavel Labath   emitData();
5346d2d589bSFangrui Song   Asm->OutStreamer->emitValueToAlignment(4, 0);
5356d2d589bSFangrui Song   Asm->OutStreamer->emitLabel(ContributionEnd);
5366088c234SPavel Labath }
5376088c234SPavel Labath 
emitAppleAccelTableImpl(AsmPrinter * Asm,AccelTableBase & Contents,StringRef Prefix,const MCSymbol * SecBegin,ArrayRef<AppleAccelTableData::Atom> Atoms)538a7c457d2SPavel Labath void llvm::emitAppleAccelTableImpl(AsmPrinter *Asm, AccelTableBase &Contents,
539a7c457d2SPavel Labath                                    StringRef Prefix, const MCSymbol *SecBegin,
540a7c457d2SPavel Labath                                    ArrayRef<AppleAccelTableData::Atom> Atoms) {
541a7c457d2SPavel Labath   Contents.finalize(Asm, Prefix);
542e60ca777SJonas Devlieghere   AppleAccelTableWriter(Asm, Contents, Atoms, SecBegin).emit();
543855fc3bbSJonas Devlieghere }
544855fc3bbSJonas Devlieghere 
emitDWARF5AccelTable(AsmPrinter * Asm,AccelTable<DWARF5AccelTableData> & Contents,const DwarfDebug & DD,ArrayRef<std::unique_ptr<DwarfCompileUnit>> CUs)5456088c234SPavel Labath void llvm::emitDWARF5AccelTable(
5466088c234SPavel Labath     AsmPrinter *Asm, AccelTable<DWARF5AccelTableData> &Contents,
5476088c234SPavel Labath     const DwarfDebug &DD, ArrayRef<std::unique_ptr<DwarfCompileUnit>> CUs) {
54898062cb3SJonas Devlieghere   std::vector<MCSymbol *> CompUnits;
5496dd452b5SDavid Blaikie   SmallVector<unsigned, 1> CUIndex(CUs.size());
5506dd452b5SDavid Blaikie   int Count = 0;
55198062cb3SJonas Devlieghere   for (const auto &CU : enumerate(CUs)) {
5522f511762SDavid Blaikie     if (CU.value()->getCUNode()->getNameTableKind() !=
5532f511762SDavid Blaikie         DICompileUnit::DebugNameTableKind::Default)
55466cf14d0SDavid Blaikie       continue;
5556dd452b5SDavid Blaikie     CUIndex[CU.index()] = Count++;
55698062cb3SJonas Devlieghere     assert(CU.index() == CU.value()->getUniqueID());
55798062cb3SJonas Devlieghere     const DwarfCompileUnit *MainCU =
55898062cb3SJonas Devlieghere         DD.useSplitDwarf() ? CU.value()->getSkeleton() : CU.value().get();
55998062cb3SJonas Devlieghere     CompUnits.push_back(MainCU->getLabelBegin());
56098062cb3SJonas Devlieghere   }
56198062cb3SJonas Devlieghere 
56266cf14d0SDavid Blaikie   if (CompUnits.empty())
56366cf14d0SDavid Blaikie     return;
56466cf14d0SDavid Blaikie 
565adf4142fSFangrui Song   Asm->OutStreamer->switchSection(
56666cf14d0SDavid Blaikie       Asm->getObjFileLowering().getDwarfDebugNamesSection());
56766cf14d0SDavid Blaikie 
5686088c234SPavel Labath   Contents.finalize(Asm, "names");
56998062cb3SJonas Devlieghere   Dwarf5AccelTableWriter<DWARF5AccelTableData>(
57098062cb3SJonas Devlieghere       Asm, Contents, CompUnits,
5716dd452b5SDavid Blaikie       [&](const DWARF5AccelTableData &Entry) {
57298062cb3SJonas Devlieghere         const DIE *CUDie = Entry.getDie().getUnitDie();
5736dd452b5SDavid Blaikie         return CUIndex[DD.lookupCU(CUDie)->getUniqueID()];
57498062cb3SJonas Devlieghere       })
57598062cb3SJonas Devlieghere       .emit();
57698062cb3SJonas Devlieghere }
57798062cb3SJonas Devlieghere 
emitDWARF5AccelTable(AsmPrinter * Asm,AccelTable<DWARF5AccelTableStaticData> & Contents,ArrayRef<MCSymbol * > CUs,llvm::function_ref<unsigned (const DWARF5AccelTableStaticData &)> getCUIndexForEntry)57898062cb3SJonas Devlieghere void llvm::emitDWARF5AccelTable(
57998062cb3SJonas Devlieghere     AsmPrinter *Asm, AccelTable<DWARF5AccelTableStaticData> &Contents,
58098062cb3SJonas Devlieghere     ArrayRef<MCSymbol *> CUs,
58198062cb3SJonas Devlieghere     llvm::function_ref<unsigned(const DWARF5AccelTableStaticData &)>
58298062cb3SJonas Devlieghere         getCUIndexForEntry) {
58398062cb3SJonas Devlieghere   Contents.finalize(Asm, "names");
58498062cb3SJonas Devlieghere   Dwarf5AccelTableWriter<DWARF5AccelTableStaticData>(Asm, Contents, CUs,
58598062cb3SJonas Devlieghere                                                      getCUIndexForEntry)
58698062cb3SJonas Devlieghere       .emit();
5876088c234SPavel Labath }
5886088c234SPavel Labath 
emit(AsmPrinter * Asm) const589855fc3bbSJonas Devlieghere void AppleAccelTableOffsetData::emit(AsmPrinter *Asm) const {
590a845ebd6SIgor Kudrin   assert(Die.getDebugSectionOffset() <= UINT32_MAX &&
591a845ebd6SIgor Kudrin          "The section offset exceeds the limit.");
5927f7e6069SPavel Labath   Asm->emitInt32(Die.getDebugSectionOffset());
593855fc3bbSJonas Devlieghere }
594855fc3bbSJonas Devlieghere 
emit(AsmPrinter * Asm) const595855fc3bbSJonas Devlieghere void AppleAccelTableTypeData::emit(AsmPrinter *Asm) const {
596a845ebd6SIgor Kudrin   assert(Die.getDebugSectionOffset() <= UINT32_MAX &&
597a845ebd6SIgor Kudrin          "The section offset exceeds the limit.");
5987f7e6069SPavel Labath   Asm->emitInt32(Die.getDebugSectionOffset());
5997f7e6069SPavel Labath   Asm->emitInt16(Die.getTag());
6004b4d85fdSRafael Espindola   Asm->emitInt8(0);
601855fc3bbSJonas Devlieghere }
6025ead3a2bSJonas Devlieghere 
emit(AsmPrinter * Asm) const6035ead3a2bSJonas Devlieghere void AppleAccelTableStaticOffsetData::emit(AsmPrinter *Asm) const {
6044b4d85fdSRafael Espindola   Asm->emitInt32(Offset);
6055ead3a2bSJonas Devlieghere }
6065ead3a2bSJonas Devlieghere 
emit(AsmPrinter * Asm) const6075ead3a2bSJonas Devlieghere void AppleAccelTableStaticTypeData::emit(AsmPrinter *Asm) const {
6084b4d85fdSRafael Espindola   Asm->emitInt32(Offset);
6094b4d85fdSRafael Espindola   Asm->emitInt16(Tag);
6104b4d85fdSRafael Espindola   Asm->emitInt8(ObjCClassIsImplementation ? dwarf::DW_FLAG_type_implementation
6115ead3a2bSJonas Devlieghere                                           : 0);
6124b4d85fdSRafael Espindola   Asm->emitInt32(QualifiedNameHash);
6135ead3a2bSJonas Devlieghere }
614073971b2SJonas Devlieghere 
615a7c457d2SPavel Labath constexpr AppleAccelTableData::Atom AppleAccelTableTypeData::Atoms[];
616a7c457d2SPavel Labath constexpr AppleAccelTableData::Atom AppleAccelTableOffsetData::Atoms[];
617a7c457d2SPavel Labath constexpr AppleAccelTableData::Atom AppleAccelTableStaticOffsetData::Atoms[];
618a7c457d2SPavel Labath constexpr AppleAccelTableData::Atom AppleAccelTableStaticTypeData::Atoms[];
6191ce64dccSJonas Devlieghere 
6201ce64dccSJonas Devlieghere #ifndef NDEBUG
print(raw_ostream & OS) const621e60ca777SJonas Devlieghere void AppleAccelTableWriter::Header::print(raw_ostream &OS) const {
6221ce64dccSJonas Devlieghere   OS << "Magic: " << format("0x%x", Magic) << "\n"
6231ce64dccSJonas Devlieghere      << "Version: " << Version << "\n"
6241ce64dccSJonas Devlieghere      << "Hash Function: " << HashFunction << "\n"
6251ce64dccSJonas Devlieghere      << "Bucket Count: " << BucketCount << "\n"
6261ce64dccSJonas Devlieghere      << "Header Data Length: " << HeaderDataLength << "\n";
6271ce64dccSJonas Devlieghere }
6281ce64dccSJonas Devlieghere 
print(raw_ostream & OS) const629a7c457d2SPavel Labath void AppleAccelTableData::Atom::print(raw_ostream &OS) const {
6301ce64dccSJonas Devlieghere   OS << "Type: " << dwarf::AtomTypeString(Type) << "\n"
6311ce64dccSJonas Devlieghere      << "Form: " << dwarf::FormEncodingString(Form) << "\n";
6321ce64dccSJonas Devlieghere }
6331ce64dccSJonas Devlieghere 
print(raw_ostream & OS) const634e60ca777SJonas Devlieghere void AppleAccelTableWriter::HeaderData::print(raw_ostream &OS) const {
6351ce64dccSJonas Devlieghere   OS << "DIE Offset Base: " << DieOffsetBase << "\n";
6361ce64dccSJonas Devlieghere   for (auto Atom : Atoms)
6371ce64dccSJonas Devlieghere     Atom.print(OS);
6381ce64dccSJonas Devlieghere }
6391ce64dccSJonas Devlieghere 
print(raw_ostream & OS) const640e60ca777SJonas Devlieghere void AppleAccelTableWriter::print(raw_ostream &OS) const {
6411ce64dccSJonas Devlieghere   Header.print(OS);
6421ce64dccSJonas Devlieghere   HeaderData.print(OS);
643a7c457d2SPavel Labath   Contents.print(OS);
644a7c457d2SPavel Labath   SecBegin->print(OS, nullptr);
6451ce64dccSJonas Devlieghere }
6461ce64dccSJonas Devlieghere 
print(raw_ostream & OS) const647a7c457d2SPavel Labath void AccelTableBase::HashData::print(raw_ostream &OS) const {
648062eb537SPavel Labath   OS << "Name: " << Name.getString() << "\n";
6491ce64dccSJonas Devlieghere   OS << "  Hash Value: " << format("0x%x", HashValue) << "\n";
6501ce64dccSJonas Devlieghere   OS << "  Symbol: ";
6511ce64dccSJonas Devlieghere   if (Sym)
6521ce64dccSJonas Devlieghere     OS << *Sym;
6531ce64dccSJonas Devlieghere   else
6541ce64dccSJonas Devlieghere     OS << "<none>";
6551ce64dccSJonas Devlieghere   OS << "\n";
656062eb537SPavel Labath   for (auto *Value : Values)
6571ce64dccSJonas Devlieghere     Value->print(OS);
6581ce64dccSJonas Devlieghere }
6591ce64dccSJonas Devlieghere 
print(raw_ostream & OS) const660a7c457d2SPavel Labath void AccelTableBase::print(raw_ostream &OS) const {
6611ce64dccSJonas Devlieghere   // Print Content.
6621ce64dccSJonas Devlieghere   OS << "Entries: \n";
6631ce64dccSJonas Devlieghere   for (const auto &Entry : Entries) {
6641ce64dccSJonas Devlieghere     OS << "Name: " << Entry.first() << "\n";
6651ce64dccSJonas Devlieghere     for (auto *V : Entry.second.Values)
6661ce64dccSJonas Devlieghere       V->print(OS);
6671ce64dccSJonas Devlieghere   }
6681ce64dccSJonas Devlieghere 
6691ce64dccSJonas Devlieghere   OS << "Buckets and Hashes: \n";
670*9e6d1f4bSKazu Hirata   for (const auto &Bucket : Buckets)
671*9e6d1f4bSKazu Hirata     for (const auto &Hash : Bucket)
6721ce64dccSJonas Devlieghere       Hash->print(OS);
6731ce64dccSJonas Devlieghere 
6741ce64dccSJonas Devlieghere   OS << "Data: \n";
675*9e6d1f4bSKazu Hirata   for (const auto &E : Entries)
676062eb537SPavel Labath     E.second.print(OS);
6771ce64dccSJonas Devlieghere }
6781ce64dccSJonas Devlieghere 
print(raw_ostream & OS) const6796088c234SPavel Labath void DWARF5AccelTableData::print(raw_ostream &OS) const {
68098062cb3SJonas Devlieghere   OS << "  Offset: " << getDieOffset() << "\n";
68198062cb3SJonas Devlieghere   OS << "  Tag: " << dwarf::TagString(getDieTag()) << "\n";
68298062cb3SJonas Devlieghere }
68398062cb3SJonas Devlieghere 
print(raw_ostream & OS) const68498062cb3SJonas Devlieghere void DWARF5AccelTableStaticData::print(raw_ostream &OS) const {
68598062cb3SJonas Devlieghere   OS << "  Offset: " << getDieOffset() << "\n";
68698062cb3SJonas Devlieghere   OS << "  Tag: " << dwarf::TagString(getDieTag()) << "\n";
6876088c234SPavel Labath }
6886088c234SPavel Labath 
print(raw_ostream & OS) const6891ce64dccSJonas Devlieghere void AppleAccelTableOffsetData::print(raw_ostream &OS) const {
6900120691fSPavel Labath   OS << "  Offset: " << Die.getOffset() << "\n";
6911ce64dccSJonas Devlieghere }
6921ce64dccSJonas Devlieghere 
print(raw_ostream & OS) const6931ce64dccSJonas Devlieghere void AppleAccelTableTypeData::print(raw_ostream &OS) const {
6940120691fSPavel Labath   OS << "  Offset: " << Die.getOffset() << "\n";
6950120691fSPavel Labath   OS << "  Tag: " << dwarf::TagString(Die.getTag()) << "\n";
6961ce64dccSJonas Devlieghere }
6971ce64dccSJonas Devlieghere 
print(raw_ostream & OS) const6981ce64dccSJonas Devlieghere void AppleAccelTableStaticOffsetData::print(raw_ostream &OS) const {
6991ce64dccSJonas Devlieghere   OS << "  Static Offset: " << Offset << "\n";
7001ce64dccSJonas Devlieghere }
7011ce64dccSJonas Devlieghere 
print(raw_ostream & OS) const7021ce64dccSJonas Devlieghere void AppleAccelTableStaticTypeData::print(raw_ostream &OS) const {
7031ce64dccSJonas Devlieghere   OS << "  Static Offset: " << Offset << "\n";
7041ce64dccSJonas Devlieghere   OS << "  QualifiedNameHash: " << format("%x\n", QualifiedNameHash) << "\n";
7051ce64dccSJonas Devlieghere   OS << "  Tag: " << dwarf::TagString(Tag) << "\n";
7061ce64dccSJonas Devlieghere   OS << "  ObjCClassIsImplementation: "
7071ce64dccSJonas Devlieghere      << (ObjCClassIsImplementation ? "true" : "false");
7081ce64dccSJonas Devlieghere   OS << "\n";
7091ce64dccSJonas Devlieghere }
7101ce64dccSJonas Devlieghere #endif
711