#ifndef LLVM_CODEGEN_BINARYOBJECT_H
#define LLVM_CODEGEN_BINARYOBJECT_H
#include "llvm/CodeGen/MachineRelocation.h"
#include "llvm/System/DataTypes.h"
#include <string>
#include <vector>
namespace llvm {
typedef std::vector<uint8_t> BinaryData;
class BinaryObject {
protected:
std::string Name;
bool IsLittleEndian;
bool Is64Bit;
BinaryData Data;
std::vector<MachineRelocation> Relocations;
public:
BinaryObject() {}
BinaryObject(bool isLittleEndian, bool is64Bit)
: IsLittleEndian(isLittleEndian), Is64Bit(is64Bit) {}
BinaryObject(const std::string &name, bool isLittleEndian, bool is64Bit)
: Name(name), IsLittleEndian(isLittleEndian), Is64Bit(is64Bit) {}
~BinaryObject() {}
inline std::string getName() const { return Name; }
size_t size() const {
return Data.size();
}
BinaryData& getData() {
return Data;
}
const std::vector<MachineRelocation>& getRelocations() const {
return Relocations;
}
bool hasRelocations() const {
return !Relocations.empty();
}
inline void emitZeros(unsigned Size) {
for (unsigned i=0; i < Size; ++i)
emitByte(0);
}
inline void emitByte(uint8_t B) {
Data.push_back(B);
}
inline void emitWord16(uint16_t W) {
if (IsLittleEndian)
emitWord16LE(W);
else
emitWord16BE(W);
}
inline void emitWord16LE(uint16_t W) {
Data.push_back((uint8_t)(W >> 0));
Data.push_back((uint8_t)(W >> 8));
}
inline void emitWord16BE(uint16_t W) {
Data.push_back((uint8_t)(W >> 8));
Data.push_back((uint8_t)(W >> 0));
}
inline void emitWord(uint64_t W) {
if (!Is64Bit)
emitWord32(W);
else
emitWord64(W);
}
inline void emitWord32(uint32_t W) {
if (IsLittleEndian)
emitWordLE(W);
else
emitWordBE(W);
}
inline void emitWord64(uint64_t W) {
if (IsLittleEndian)
emitDWordLE(W);
else
emitDWordBE(W);
}
inline void emitWordFP80(const uint64_t *W, unsigned PadSize) {
if (IsLittleEndian) {
emitWord64(W[0]);
emitWord16(W[1]);
} else {
emitWord16(W[1]);
emitWord64(W[0]);
}
emitZeros(PadSize);
}
inline void emitWordLE(uint32_t W) {
Data.push_back((uint8_t)(W >> 0));
Data.push_back((uint8_t)(W >> 8));
Data.push_back((uint8_t)(W >> 16));
Data.push_back((uint8_t)(W >> 24));
}
inline void emitWordBE(uint32_t W) {
Data.push_back((uint8_t)(W >> 24));
Data.push_back((uint8_t)(W >> 16));
Data.push_back((uint8_t)(W >> 8));
Data.push_back((uint8_t)(W >> 0));
}
inline void emitDWordLE(uint64_t W) {
Data.push_back((uint8_t)(W >> 0));
Data.push_back((uint8_t)(W >> 8));
Data.push_back((uint8_t)(W >> 16));
Data.push_back((uint8_t)(W >> 24));
Data.push_back((uint8_t)(W >> 32));
Data.push_back((uint8_t)(W >> 40));
Data.push_back((uint8_t)(W >> 48));
Data.push_back((uint8_t)(W >> 56));
}
inline void emitDWordBE(uint64_t W) {
Data.push_back((uint8_t)(W >> 56));
Data.push_back((uint8_t)(W >> 48));
Data.push_back((uint8_t)(W >> 40));
Data.push_back((uint8_t)(W >> 32));
Data.push_back((uint8_t)(W >> 24));
Data.push_back((uint8_t)(W >> 16));
Data.push_back((uint8_t)(W >> 8));
Data.push_back((uint8_t)(W >> 0));
}
inline void fixByte(uint8_t B, uint32_t offset) {
Data[offset] = B;
}
inline void fixWord16(uint16_t W, uint32_t offset) {
if (IsLittleEndian)
fixWord16LE(W, offset);
else
fixWord16BE(W, offset);
}
inline void fixWord16LE(uint16_t W, uint32_t offset) {
Data[offset] = (uint8_t)(W >> 0);
Data[++offset] = (uint8_t)(W >> 8);
}
inline void fixWord16BE(uint16_t W, uint32_t offset) {
Data[offset] = (uint8_t)(W >> 8);
Data[++offset] = (uint8_t)(W >> 0);
}
inline void fixWord(uint64_t W, uint32_t offset) {
if (!Is64Bit)
fixWord32(W, offset);
else
fixWord64(W, offset);
}
inline void fixWord32(uint32_t W, uint32_t offset) {
if (IsLittleEndian)
fixWord32LE(W, offset);
else
fixWord32BE(W, offset);
}
inline void fixWord32LE(uint32_t W, uint32_t offset) {
Data[offset] = (uint8_t)(W >> 0);
Data[++offset] = (uint8_t)(W >> 8);
Data[++offset] = (uint8_t)(W >> 16);
Data[++offset] = (uint8_t)(W >> 24);
}
inline void fixWord32BE(uint32_t W, uint32_t offset) {
Data[offset] = (uint8_t)(W >> 24);
Data[++offset] = (uint8_t)(W >> 16);
Data[++offset] = (uint8_t)(W >> 8);
Data[++offset] = (uint8_t)(W >> 0);
}
inline void fixWord64(uint64_t W, uint32_t offset) {
if (IsLittleEndian)
fixWord64LE(W, offset);
else
fixWord64BE(W, offset);
}
inline void fixWord64LE(uint64_t W, uint32_t offset) {
Data[offset] = (uint8_t)(W >> 0);
Data[++offset] = (uint8_t)(W >> 8);
Data[++offset] = (uint8_t)(W >> 16);
Data[++offset] = (uint8_t)(W >> 24);
Data[++offset] = (uint8_t)(W >> 32);
Data[++offset] = (uint8_t)(W >> 40);
Data[++offset] = (uint8_t)(W >> 48);
Data[++offset] = (uint8_t)(W >> 56);
}
inline void fixWord64BE(uint64_t W, uint32_t offset) {
Data[offset] = (uint8_t)(W >> 56);
Data[++offset] = (uint8_t)(W >> 48);
Data[++offset] = (uint8_t)(W >> 40);
Data[++offset] = (uint8_t)(W >> 32);
Data[++offset] = (uint8_t)(W >> 24);
Data[++offset] = (uint8_t)(W >> 16);
Data[++offset] = (uint8_t)(W >> 8);
Data[++offset] = (uint8_t)(W >> 0);
}
void emitAlignment(unsigned Alignment, uint8_t fill = 0) {
if (Alignment <= 1) return;
unsigned PadSize = -Data.size() & (Alignment-1);
for (unsigned i = 0; i<PadSize; ++i)
Data.push_back(fill);
}
void emitULEB128Bytes(uint64_t Value) {
do {
uint8_t Byte = (uint8_t)(Value & 0x7f);
Value >>= 7;
if (Value) Byte |= 0x80;
emitByte(Byte);
} while (Value);
}
void emitSLEB128Bytes(int64_t Value) {
int Sign = Value >> (8 * sizeof(Value) - 1);
bool IsMore;
do {
uint8_t Byte = (uint8_t)(Value & 0x7f);
Value >>= 7;
IsMore = Value != Sign || ((Byte ^ Sign) & 0x40) != 0;
if (IsMore) Byte |= 0x80;
emitByte(Byte);
} while (IsMore);
}
void emitString(const std::string &String) {
for (unsigned i = 0, N = static_cast<unsigned>(String.size()); i<N; ++i) {
unsigned char C = String[i];
emitByte(C);
}
emitByte(0);
}
uintptr_t getCurrentPCOffset() const {
return Data.size();
}
void addRelocation(const MachineRelocation& relocation) {
Relocations.push_back(relocation);
}
};
}
#endif