mirror of
https://github.com/hedge-dev/XenonRecomp.git
synced 2025-04-23 01:01:26 +00:00
100 lines
2.6 KiB
C++
100 lines
2.6 KiB
C++
#include "image.h"
|
|
#include "elf.h"
|
|
#include "xex.h"
|
|
#include <cassert>
|
|
#include <cstring>
|
|
|
|
void Image::Map(const std::string_view& name, size_t base, uint32_t size, uint8_t flags, uint8_t* data)
|
|
{
|
|
sections.insert({ std::string(name), this->base + base,
|
|
size, static_cast<SectionFlags>(flags), data });
|
|
}
|
|
|
|
const void* Image::Find(size_t address) const
|
|
{
|
|
const auto section = std::prev(sections.upper_bound(address));
|
|
return section->data + (address - section->base);
|
|
}
|
|
|
|
const Section* Image::Find(const std::string_view& name) const
|
|
{
|
|
for (const auto& section : sections)
|
|
{
|
|
if (section.name == name)
|
|
{
|
|
return §ion;
|
|
}
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
Image Image::ParseImage(const uint8_t* data, size_t size)
|
|
{
|
|
if (data[0] == ELFMAG0 && data[1] == ELFMAG1 && data[2] == ELFMAG2 && data[3] == ELFMAG3)
|
|
{
|
|
return ElfLoadImage(data, size);
|
|
}
|
|
else if (data[0] == 'X' && data[1] == 'E' && data[2] == 'X' && data[3] == '2')
|
|
{
|
|
return Xex2LoadImage(data, size);
|
|
}
|
|
|
|
return {};
|
|
}
|
|
|
|
Image ElfLoadImage(const uint8_t* data, size_t size)
|
|
{
|
|
const auto* header = (elf32_hdr*)data;
|
|
assert(header->e_ident[EI_DATA] == 2);
|
|
|
|
Image image{};
|
|
image.size = size;
|
|
image.data = std::make_unique<uint8_t[]>(size);
|
|
image.entry_point = ByteSwap(header->e_entry);
|
|
memcpy(image.data.get(), data, size);
|
|
|
|
auto stringTableIndex = ByteSwap(header->e_shstrndx);
|
|
|
|
const auto numSections = ByteSwap(header->e_shnum);
|
|
const auto numpSections = ByteSwap(header->e_phnum);
|
|
|
|
const auto* sections = (elf32_shdr*)(data + ByteSwap(header->e_shoff));
|
|
const auto* psections = (elf32_phdr*)(data + ByteSwap(header->e_phoff));
|
|
|
|
for (size_t i = 0; i < numpSections; i++)
|
|
{
|
|
if (psections[i].p_type == ByteSwap((Elf32_Word)PT_LOAD))
|
|
{
|
|
image.base = ByteSwap(psections[i].p_vaddr);
|
|
break;
|
|
}
|
|
}
|
|
|
|
auto* stringTable = reinterpret_cast<const char*>(data + ByteSwap(sections[stringTableIndex].sh_offset));
|
|
|
|
for (size_t i = 0; i < numSections; i++)
|
|
{
|
|
const auto& section = sections[i];
|
|
if (section.sh_type == 0)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
uint8_t flags{};
|
|
|
|
if (section.sh_flags & ByteSwap(SHF_EXECINSTR))
|
|
{
|
|
flags |= SectionFlags_Code;
|
|
}
|
|
|
|
auto* name = section.sh_name != 0 ? stringTable + ByteSwap(section.sh_name) : nullptr;
|
|
const auto rva = ByteSwap(section.sh_addr) - image.base;
|
|
const auto size = ByteSwap(section.sh_size);
|
|
|
|
image.Map(name, rva, size, flags, image.data.get() + ByteSwap(section.sh_offset));
|
|
}
|
|
|
|
return image;
|
|
}
|