#define DEBUG_TYPE "dyld"
#include "RuntimeDyldImpl.h"
using namespace llvm;
using namespace llvm::object;
RTDyldMemoryManager::~RTDyldMemoryManager() {}
RuntimeDyldImpl::~RuntimeDyldImpl() {}
namespace llvm {
void RuntimeDyldImpl::extractFunction(StringRef Name, uint8_t *StartAddress,
uint8_t *EndAddress) {
uintptr_t Size = EndAddress - StartAddress + 1;
uintptr_t AllocSize = Size;
uint8_t *Mem = MemMgr->startFunctionBody(Name.data(), AllocSize);
assert(Size >= (uint64_t)(EndAddress - StartAddress + 1) &&
"Memory manager failed to allocate enough memory!");
memcpy(Mem, StartAddress, Size);
MemMgr->endFunctionBody(Name.data(), Mem, Mem + Size);
Functions[Name] = sys::MemoryBlock(Mem, Size);
SymbolTable[Name] = Mem;
DEBUG(dbgs() << " allocated to [" << Mem << ", " << Mem + Size << "]\n");
}
void RuntimeDyldImpl::resolveRelocations() {
StringMap<uint8_t*>::iterator i = SymbolTable.begin();
StringMap<uint8_t*>::iterator e = SymbolTable.end();
for (;i != e; ++i)
reassignSymbolAddress(i->getKey(), i->getValue());
}
RuntimeDyld::RuntimeDyld(RTDyldMemoryManager *mm) {
Dyld = 0;
MM = mm;
}
RuntimeDyld::~RuntimeDyld() {
delete Dyld;
}
bool RuntimeDyld::loadObject(MemoryBuffer *InputBuffer) {
if (!Dyld) {
if (RuntimeDyldMachO::isKnownFormat(InputBuffer))
Dyld = new RuntimeDyldMachO(MM);
else
report_fatal_error("Unknown object format!");
} else {
if(!Dyld->isCompatibleFormat(InputBuffer))
report_fatal_error("Incompatible object format!");
}
return Dyld->loadObject(InputBuffer);
}
void *RuntimeDyld::getSymbolAddress(StringRef Name) {
return Dyld->getSymbolAddress(Name);
}
void RuntimeDyld::resolveRelocations() {
Dyld->resolveRelocations();
}
void RuntimeDyld::reassignSymbolAddress(StringRef Name, uint8_t *Addr) {
Dyld->reassignSymbolAddress(Name, Addr);
}
StringRef RuntimeDyld::getErrorString() {
return Dyld->getErrorString();
}
}