mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-09-29 16:21:29 +00:00
118 lines
2.5 KiB
C++
118 lines
2.5 KiB
C++
#include "ExecSpace.h"
|
|
#include "ELFLoader.h"
|
|
#include <AK/TemporaryFile.h>
|
|
#include <AK/Types.h>
|
|
|
|
//#define EXECSPACE_DEBUG
|
|
|
|
ExecSpace::ExecSpace()
|
|
{
|
|
initializeBuiltins();
|
|
}
|
|
|
|
ExecSpace::~ExecSpace()
|
|
{
|
|
if (!hookableAlloc) {
|
|
for (auto& area : m_areas)
|
|
kfree(area->memory);
|
|
}
|
|
}
|
|
|
|
#ifdef SERENITY
|
|
int puts(const char* str)
|
|
{
|
|
kprintf("%s\n", str);
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
void ExecSpace::initializeBuiltins()
|
|
{
|
|
#ifndef SERENITY
|
|
m_symbols.set("puts", { (char*)puts, 0 });
|
|
#endif
|
|
}
|
|
|
|
#ifdef SERENITY
|
|
bool ExecSpace::loadELF(ByteBuffer&& file)
|
|
#else
|
|
bool ExecSpace::loadELF(MappedFile&& file)
|
|
#endif
|
|
{
|
|
ELFLoader loader(*this, move(file));
|
|
if (!loader.load())
|
|
return false;
|
|
#ifdef EXECSPACE_DEBUG
|
|
kprintf("[ExecSpace] ELF loaded, symbol map now:\n");
|
|
for (auto& s : m_symbols) {
|
|
kprintf("> %p: %s (%u)\n",
|
|
s.value.ptr,
|
|
s.key.characters(),
|
|
s.value.size);
|
|
}
|
|
#endif
|
|
return true;
|
|
}
|
|
|
|
#ifdef EXECSPACE_DEBUG
|
|
static void disassemble(const char* data, size_t length)
|
|
{
|
|
if (!length)
|
|
return;
|
|
|
|
#ifdef SERENITY
|
|
for (unsigned i = 0; i < length; ++i) {
|
|
kprintf("%b ", (unsigned char)data[i]);
|
|
}
|
|
kprintf("\n");
|
|
#else
|
|
TemporaryFile temp;
|
|
if (!temp.isValid()) {
|
|
fprintf(stderr, "Unable to create temp file for disassembly.\n");
|
|
return;
|
|
}
|
|
fprintf(temp.stream(), "db ");
|
|
for (unsigned i = 0; i < length; ++i) {
|
|
fprintf(temp.stream(), "0x%02x, ", (unsigned char)data[i]);
|
|
}
|
|
fprintf(temp.stream(), "\n");
|
|
temp.sync();
|
|
|
|
char cmdbuf[128];
|
|
ksprintf(cmdbuf, "nasm -f bin -o /dev/stdout %s | ndisasm -b32 -", temp.fileName().characters());
|
|
system(cmdbuf);
|
|
#endif
|
|
}
|
|
#endif
|
|
|
|
char* ExecSpace::symbolPtr(const char* name)
|
|
{
|
|
if (auto it = m_symbols.find(name); it != m_symbols.end()) {
|
|
auto& symbol = (*it).value;
|
|
#ifdef EXECSPACE_DEBUG
|
|
kprintf("[ELFLoader] symbolPtr(%s) dump:\n", name);
|
|
disassemble(symbol.ptr, symbol.size);
|
|
#endif
|
|
return symbol.ptr;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
char* ExecSpace::allocateArea(String&& name, unsigned size)
|
|
{
|
|
char* ptr;
|
|
if (hookableAlloc)
|
|
ptr = static_cast<char*>(hookableAlloc(name, size));
|
|
else
|
|
ptr = static_cast<char*>(kmalloc(size));
|
|
if (size)
|
|
ASSERT(ptr);
|
|
m_areas.append(make<Area>(move(name), ptr, size));
|
|
return ptr;
|
|
}
|
|
|
|
void ExecSpace::addSymbol(String&& name, char* ptr, unsigned size)
|
|
{
|
|
m_symbols.set(move(name), { ptr, size });
|
|
}
|