712 lines
		
	
	
		
			24 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			712 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/MCExpr.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 (auto &Bucket : Contents.getBuckets()) {
 | 
						|
    for (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 (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 (auto &Bucket : Buckets)
 | 
						|
    for (auto &Hash : Bucket)
 | 
						|
      Hash->print(OS);
 | 
						|
 | 
						|
  OS << "Data: \n";
 | 
						|
  for (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
 |