#ifndef CODEGEN_ELF_H
#define CODEGEN_ELF_H
#include "llvm/CodeGen/BinaryObject.h"
#include "llvm/CodeGen/MachineRelocation.h"
#include "llvm/Support/ELF.h"
#include "llvm/Support/DataTypes.h"
namespace llvm {
class GlobalValue;
struct ELFSym {
union {
const GlobalValue *GV; const char *Ext; } Source;
enum {
isGV, isExtSym, isOther };
unsigned SourceType;
bool isGlobalValue() const { return SourceType == isGV; }
bool isExternalSym() const { return SourceType == isExtSym; }
const GlobalValue *getGlobalValue() const {
assert(SourceType == isGV && "This is not a global value");
return Source.GV;
}
const char *getExternalSymbol() const {
assert(SourceType == isExtSym && "This is not an external symbol");
return Source.Ext;
}
static ELFSym *getGV(const GlobalValue *GV, unsigned Bind,
unsigned Type, unsigned Visibility) {
ELFSym *Sym = new ELFSym();
Sym->Source.GV = GV;
Sym->setBind(Bind);
Sym->setType(Type);
Sym->setVisibility(Visibility);
Sym->SourceType = isGV;
return Sym;
}
static ELFSym *getExtSym(const char *Ext) {
ELFSym *Sym = new ELFSym();
Sym->Source.Ext = Ext;
Sym->setBind(ELF::STB_GLOBAL);
Sym->setType(ELF::STT_NOTYPE);
Sym->setVisibility(ELF::STV_DEFAULT);
Sym->SourceType = isExtSym;
return Sym;
}
static ELFSym *getSectionSym() {
ELFSym *Sym = new ELFSym();
Sym->setBind(ELF::STB_LOCAL);
Sym->setType(ELF::STT_SECTION);
Sym->setVisibility(ELF::STV_DEFAULT);
Sym->SourceType = isOther;
return Sym;
}
static ELFSym *getFileSym() {
ELFSym *Sym = new ELFSym();
Sym->setBind(ELF::STB_LOCAL);
Sym->setType(ELF::STT_FILE);
Sym->setVisibility(ELF::STV_DEFAULT);
Sym->SectionIdx = 0xfff1; Sym->SourceType = isOther;
return Sym;
}
static ELFSym *getUndefGV(const GlobalValue *GV, unsigned Bind) {
ELFSym *Sym = new ELFSym();
Sym->Source.GV = GV;
Sym->setBind(Bind);
Sym->setType(ELF::STT_NOTYPE);
Sym->setVisibility(ELF::STV_DEFAULT);
Sym->SectionIdx = 0; Sym->SourceType = isGV;
return Sym;
}
unsigned NameIdx; uint64_t Value;
unsigned Size;
uint8_t Info;
uint8_t Other;
unsigned short SectionIdx;
unsigned SymTabIdx;
ELFSym() : SourceType(isOther), NameIdx(0), Value(0),
Size(0), Info(0), Other(ELF::STV_DEFAULT), SectionIdx(0),
SymTabIdx(0) {}
unsigned getBind() const { return (Info >> 4) & 0xf; }
unsigned getType() const { return Info & 0xf; }
bool isLocalBind() const { return getBind() == ELF::STB_LOCAL; }
bool isFileType() const { return getType() == ELF::STT_FILE; }
void setBind(unsigned X) {
assert(X == (X & 0xF) && "Bind value out of range!");
Info = (Info & 0x0F) | (X << 4);
}
void setType(unsigned X) {
assert(X == (X & 0xF) && "Type value out of range!");
Info = (Info & 0xF0) | X;
}
void setVisibility(unsigned V) {
assert(V == (V & 0x3) && "Visibility value out of range!");
Other = V;
}
};
class ELFSection : public BinaryObject {
public:
unsigned NameIdx; unsigned Type; unsigned Flags; uint64_t Addr; unsigned Offset; unsigned Size; unsigned Link; unsigned Info; unsigned Align; unsigned EntSize;
unsigned short SectionIdx;
ELFSym *Sym;
unsigned getSymbolTableIndex() const {
assert(Sym && "section not present in the symbol table");
return Sym->SymTabIdx;
}
ELFSection(const std::string &name, bool isLittleEndian, bool is64Bit)
: BinaryObject(name, isLittleEndian, is64Bit), Type(0), Flags(0), Addr(0),
Offset(0), Size(0), Link(0), Info(0), Align(0), EntSize(0), Sym(0) {}
};
class ELFRelocation {
uint64_t r_offset; uint32_t r_symidx; uint32_t r_type; int64_t r_add; bool r_rela; public:
uint64_t getInfo(bool is64Bit) const {
if (is64Bit)
return ((uint64_t)r_symidx << 32) + ((uint64_t)r_type & 0xFFFFFFFFL);
else
return (r_symidx << 8) + (r_type & 0xFFL);
}
uint64_t getOffset() const { return r_offset; }
int64_t getAddend() const { return r_add; }
ELFRelocation(uint64_t off, uint32_t sym, uint32_t type,
bool rela = true, int64_t addend = 0) :
r_offset(off), r_symidx(sym), r_type(type),
r_add(addend), r_rela(rela) {}
};
}
#endif