#ifndef LLVM_LIB_CODEGEN_ASMPRINTER_DWARFACCELTABLE_H
#define LLVM_LIB_CODEGEN_ASMPRINTER_DWARFACCELTABLE_H
#include "llvm/ADT/ArrayRef.h"
#include "llvm/ADT/StringMap.h"
#include "llvm/CodeGen/DIE.h"
#include "llvm/MC/MCSymbol.h"
#include "llvm/Support/Compiler.h"
#include "llvm/Support/DataTypes.h"
#include "llvm/Support/Debug.h"
#include "llvm/Support/Dwarf.h"
#include "llvm/Support/ErrorHandling.h"
#include "llvm/Support/Format.h"
#include "llvm/Support/FormattedStream.h"
#include <vector>
namespace llvm {
class AsmPrinter;
class DwarfDebug;
class DwarfAccelTable {
public:
static uint32_t HashDJB(StringRef Str, uint32_t h = 5381) {
for (unsigned i = 0, e = Str.size(); i != e; ++i)
h = ((h << 5) + h) + (unsigned char)Str[i];
return h;
}
private:
void ComputeBucketCount(void);
struct TableHeader {
uint32_t magic; uint16_t version; uint16_t hash_function; uint32_t bucket_count; uint32_t hashes_count; uint32_t header_data_len;
static const uint32_t MagicHash = 0x48415348;
TableHeader(uint32_t data_len)
: magic(MagicHash), version(1),
hash_function(dwarf::DW_hash_function_djb), bucket_count(0),
hashes_count(0), header_data_len(data_len) {}
#ifndef NDEBUG
void print(raw_ostream &O) {
O << "Magic: " << format("0x%x", magic) << "\n"
<< "Version: " << version << "\n"
<< "Hash Function: " << hash_function << "\n"
<< "Bucket Count: " << bucket_count << "\n"
<< "Header Data Length: " << header_data_len << "\n";
}
void dump() { print(dbgs()); }
#endif
};
public:
struct Atom {
uint16_t type; uint16_t form;
LLVM_CONSTEXPR Atom(uint16_t type, uint16_t form)
: type(type), form(form) {}
#ifndef NDEBUG
void print(raw_ostream &O) {
O << "Type: " << dwarf::AtomTypeString(type) << "\n"
<< "Form: " << dwarf::FormEncodingString(form) << "\n";
}
void dump() { print(dbgs()); }
#endif
};
private:
struct TableHeaderData {
uint32_t die_offset_base;
SmallVector<Atom, 3> Atoms;
TableHeaderData(ArrayRef<Atom> AtomList, uint32_t offset = 0)
: die_offset_base(offset), Atoms(AtomList.begin(), AtomList.end()) {}
#ifndef NDEBUG
void print(raw_ostream &O) {
O << "die_offset_base: " << die_offset_base << "\n";
for (size_t i = 0; i < Atoms.size(); i++)
Atoms[i].print(O);
}
void dump() { print(dbgs()); }
#endif
};
public:
struct HashDataContents {
union {
const DIE *Die; MCSymbol *Strp; uint32_t DieOffset; };
char Flags;
HashDataContents(const DIE *D, char Flags = 0) : Die(D), Flags(Flags) {}
HashDataContents(uint32_t DieOffset) : DieOffset(DieOffset), Flags(0) {}
HashDataContents(MCSymbol *Strp) : Strp(Strp), Flags(-1) {}
bool isDIE() const { return Flags != -1; }
char *getAtomData() const {
return (char *)(void *)(this + 1);
}
template<typename AtomT>
AtomT getAtom(unsigned int Offset) const {
AtomT Atom;
memcpy(&Atom, getAtomData() + Offset, sizeof(AtomT));
return Atom;
}
#ifndef NDEBUG
void print(raw_ostream &O, const DwarfAccelTable &Table) const;
#endif
};
class HashDataContentsRef {
const HashDataContents &Contents; const DwarfAccelTable &Table; uint16_t AtomBytePos; uint8_t AtomPos;
public:
HashDataContentsRef(HashDataContents &Contents, DwarfAccelTable &Table)
: Contents(Contents), Table(Table), AtomBytePos(0), AtomPos(1) {}
template<typename AtomT>
HashDataContentsRef addAtom(AtomT Atom) {
assert(AtomBytePos + sizeof(AtomT) <= Table.getAtomsSize());
assert(Table.checkAtomSize(AtomPos, sizeof(AtomT)));
memcpy(Contents.getAtomData() + AtomBytePos, &Atom, sizeof(AtomT));
++AtomPos;
AtomBytePos += sizeof(AtomT);
return *this;
}
};
private:
struct DataArray {
union {
MCSymbol *StrSym; uint32_t StrOffset; };
std::vector<HashDataContents *> Values;
DataArray() : StrSym(nullptr) {}
};
friend struct HashData;
struct HashData {
StringRef Str;
uint32_t HashValue;
MCSymbol *Sym;
DwarfAccelTable::DataArray &Data; HashData(StringRef S, DwarfAccelTable::DataArray &Data)
: Str(S), Data(Data) {
HashValue = DwarfAccelTable::HashDJB(S);
}
#ifndef NDEBUG
void print(raw_ostream &O, DwarfAccelTable& Table) {
O << "Name: " << Str << "\n";
O << " Hash Value: " << format("0x%x", HashValue) << "\n";
O << " Symbol: ";
if (Sym)
Sym->print(O);
else
O << "<none>";
O << "\n";
for (HashDataContents *C : Data.Values)
C->print(O, Table);
}
void dump(DwarfAccelTable& Table) { print(dbgs(), Table); }
#endif
};
DwarfAccelTable(const DwarfAccelTable &) LLVM_DELETED_FUNCTION;
void operator=(const DwarfAccelTable &) LLVM_DELETED_FUNCTION;
void EmitHeader(AsmPrinter *);
void EmitBuckets(AsmPrinter *);
void EmitHashes(AsmPrinter *);
void emitOffsets(AsmPrinter *, const MCSymbol *);
void EmitData(AsmPrinter *, DwarfDebug *D);
BumpPtrAllocator Allocator;
TableHeader Header;
TableHeaderData HeaderData;
std::vector<HashData *> Data;
typedef StringMap<DataArray, BumpPtrAllocator &> StringEntries;
StringEntries Entries;
typedef std::vector<HashData *> HashList;
typedef std::vector<HashList> BucketList;
BucketList Buckets;
HashList Hashes;
unsigned AtomsSize;
bool UseDieOffsets;
bool UseStringOffsets;
public:
DwarfAccelTable(ArrayRef<DwarfAccelTable::Atom>, bool UseDieOffset = false,
bool UseStringOffsets = false);
HashDataContentsRef AddName(StringRef Name, MCSymbol *StrSym, const DIE *Die);
HashDataContentsRef AddName(StringRef Name, uint32_t StrOffset,
uint32_t DIeOffset);
void AddUID(StringRef UID, MCSymbol *UIDSym, MCSymbol *ModuleSym);
void FinalizeTable(AsmPrinter *, StringRef);
void emit(AsmPrinter *, const MCSymbol *, DwarfDebug *);
unsigned getAtomsSize() const { return AtomsSize; }
bool useDieOffssets() const { return UseDieOffsets; }
bool useStringOffssets() const { return UseStringOffsets; }
const TableHeaderData &getHeaderData() const { return HeaderData; }
#ifndef NDEBUG
bool checkAtomSize(unsigned Index, unsigned ByteSize) const;
void print(raw_ostream &O);
void dump() { print(dbgs()); }
#endif
};
}
#endif