bool
readSections(BigEndianView &in, Header &header, std::vector<Section> §ions)
{
sections.resize(header.shnum);
for (auto i = 0u; i < sections.size(); ++i) {
auto §ion = sections[i];
// Read section header
in.seek(header.shoff + header.shentsize * i);
readSectionHeader(in, section.header);
if ((section.header.type == SHT_NOBITS) || !section.header.size) {
continue;
}
// Read section data
if (section.header.flags & SHF_DEFLATED) {
auto stream = z_stream {};
auto ret = Z_OK;
// Read the original size
in.seek(section.header.offset);
section.data.resize(in.read<uint32_t>());
// Inflate
memset(&stream, 0, sizeof(stream));
stream.zalloc = Z_NULL;
stream.zfree = Z_NULL;
stream.opaque = Z_NULL;
ret = inflateInit(&stream);
if (ret != Z_OK) {
gLog->error("Couldn't decompress .rpx section because inflateInit returned {}", ret);
section.data.clear();
} else {
stream.avail_in = section.header.size;
stream.next_in = in.readRaw<Bytef>(section.header.size);
stream.avail_out = static_cast<uInt>(section.data.size());
stream.next_out = reinterpret_cast<Bytef*>(section.data.data());
ret = inflate(&stream, Z_FINISH);
if (ret != Z_OK && ret != Z_STREAM_END) {
gLog->error("Couldn't decompress .rpx section because inflate returned {}", ret);
section.data.clear();
}
inflateEnd(&stream);
}
} else {
section.data.resize(section.header.size);
in.seek(section.header.offset);
in.read(section.data.data(), section.header.size);
}
}
return true;
}
bool
readRelocationAddend(BigEndianView &in, Rela &rela)
{
in.read(rela.offset);
in.read(rela.info);
in.read(rela.addend);
return true;
}
bool
readSymbol(BigEndianView &in, Symbol &sym)
{
in.read(sym.name);
in.read(sym.value);
in.read(sym.size);
in.read(sym.info);
in.read(sym.other);
in.read(sym.shndx);
return true;
}
bool
readSectionData(BigEndianView &in, const SectionHeader& header, std::vector<uint8_t> &data)
{
if (header.type == SHT_NOBITS || header.size == 0) {
data.clear();
return true;
}
if (header.flags & SHF_DEFLATED) {
auto stream = z_stream{};
auto ret = Z_OK;
// Read the original size
in.seek(header.offset);
data.resize(in.read<uint32_t>());
// Inflate
memset(&stream, 0, sizeof(stream));
stream.zalloc = Z_NULL;
stream.zfree = Z_NULL;
stream.opaque = Z_NULL;
ret = inflateInit(&stream);
if (ret != Z_OK) {
gLog->error("Couldn't decompress .rpx section because inflateInit returned {}", ret);
data.clear();
} else {
stream.avail_in = header.size;
stream.next_in = const_cast<Bytef*>(in.readRaw<Bytef>(header.size));
stream.avail_out = static_cast<uInt>(data.size());
stream.next_out = reinterpret_cast<Bytef*>(data.data());
ret = inflate(&stream, Z_FINISH);
if (ret != Z_OK && ret != Z_STREAM_END) {
gLog->error("Couldn't decompress .rpx section because inflate returned {}", ret);
data.clear();
}
inflateEnd(&stream);
}
} else {
data.resize(header.size);
in.seek(header.offset);
in.read(data.data(), header.size);
}
return data.size() > 0;
}
// Get symbol at index
static elf::Symbol
getSymbol(BigEndianView &symSecView, uint32_t index)
{
elf::Symbol sym;
symSecView.seek(index * sizeof(elf::Symbol));
elf::readSymbol(symSecView, sym);
return sym;
}
bool
readSectionHeader(BigEndianView &in, SectionHeader &shdr)
{
in.read(shdr.name);
in.read(shdr.type);
in.read(shdr.flags);
in.read(shdr.addr);
in.read(shdr.offset);
in.read(shdr.size);
in.read(shdr.link);
in.read(shdr.info);
in.read(shdr.addralign);
in.read(shdr.entsize);
return true;
}
bool
readSectionHeaders(BigEndianView &in, Header &header, std::vector<XSection>& sections)
{
sections.resize(header.shnum);
for (auto i = 0u; i < sections.size(); ++i) {
auto §ionHeader = sections[i].header;
in.seek(header.shoff + header.shentsize * i);
if (!readSectionHeader(in, sectionHeader)) {
return false;
}
}
return true;
}
bool
readHeader(BigEndianView &in, Header &header)
{
in.read(header.magic);
in.read(header.fileClass);
in.read(header.encoding);
in.read(header.elfVersion);
in.read(header.abi);
in.read(header.pad);
in.read(header.type);
in.read(header.machine);
in.read(header.version);
in.read(header.entry);
in.read(header.phoff);
in.read(header.shoff);
in.read(header.flags);
in.read(header.ehsize);
in.read(header.phentsize);
in.read(header.phnum);
in.read(header.shentsize);
in.read(header.shnum);
in.read(header.shstrndx);
if (header.magic != Header::Magic) {
gLog->error("Unexpected elf magic, found {:08x} expected {:08x}", header.magic, Header::Magic);
return false;
}
if (header.fileClass != ELFCLASS32) {
gLog->error("Unexpected elf file class, found {:02x} expected {:02x}", header.fileClass, ELFCLASS32);
return false;
}
if (header.encoding != ELFDATA2MSB) {
gLog->error("Unexpected elf encoding, found {:02x} expected {:02x}", header.encoding, ELFDATA2MSB);
return false;
}
if (header.machine != EM_PPC) {
gLog->error("Unexpected elf machine, found {:04x} expected {:04x}", header.machine, EM_PPC);
return false;
}
if (header.elfVersion != EV_CURRENT) {
gLog->error("Unexpected elf version, found {:02x} expected {:02x}", header.elfVersion, EV_CURRENT);
return false;
}
assert(header.shentsize == sizeof(SectionHeader));
return true;
}
bool
readFileInfo(BigEndianView &in, elf::FileInfo &info)
{
in.read(info.minVersion);
in.read(info.textSize);
in.read(info.textAlign);
in.read(info.dataSize);
in.read(info.dataAlign);
in.read(info.loaderSize);
in.read(info.loaderInfo);
in.read(info.tempSize);
in.read(info.trampAdjust);
in.read(info.sdaBase);
in.read(info.sda2Base);
in.read(info.stackSize);
in.read(info.filename);
in.read(info.flags);
in.read(info.heapSize);
in.read(info.tags);
in.read(info.unk1);
in.read(info.compressionLevel);
in.read(info.unk2);
in.read(info.fileInfoPad);
in.read(info.cafeSdkVersion);
in.read(info.cafeSdkRevision);
in.read(info.unk3);
in.read(info.unk4);
return true;
}
bool parseGSH(BigEndianView &fh)
{
gsh::Header header;
fh.read(header.magic);
if (header.magic != gsh::Header::Magic) {
std::cout << "Unexpected gsh magic " << header.magic << std::endl;
return false;
}
fh.read(header.unk1);
fh.read(header.unk2);
fh.read(header.unk3);
fh.read(header.unk4);
fh.read(header.unk5);
fh.read(header.unk6);
fh.read(header.unk7);
while (!fh.eof()) {
gsh::Block block;
fh.read(block.magic);
if (block.magic != gsh::Block::Magic) {
std::cout << "Unexpected block magic " << header.magic << std::endl;
return false;
}
fh.read(block.unk1);
fh.read(block.unk2);
fh.read(block.unk3);
fh.read(block.type);
fh.read(block.dataLength);
fh.read(block.unk4);
fh.read(block.unk5);
block.data.resize(block.dataLength);
fh.read<uint8_t>(block.data);
if (block.type == gsh::Block::VertexShader || block.type == gsh::Block::PixelShader) {
std::string out;
latte::disassemble(out, block.data);
std::cout << "----------------------------------------------" << std::endl;
std::cout << " Disassembly " << std::endl;
std::cout << "----------------------------------------------" << std::endl;
std::cout << out << std::endl;
std::cout << std::endl;
latte::Shader shader;
latte::decode(shader, block.data);
std::cout << "----------------------------------------------" << std::endl;
std::cout << " Blocks " << std::endl;
std::cout << "----------------------------------------------" << std::endl;
latte::blockify(shader);
std::cout << std::endl;
std::string hlsl;
latte::generateHLSL(shader, hlsl);
std::cout << "----------------------------------------------" << std::endl;
std::cout << " HLSL " << std::endl;
std::cout << "----------------------------------------------" << std::endl;
std::cout << hlsl << std::endl;
std::cout << std::endl;
}
}
return true;
}
