Hello_User/XenonRecomp
1
0
Fork 0
mirror of https://github.com/hedge-dev/XenonRecomp.git synced 2025-07-23 05:23:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Function branch analysis

Browse Source
This commit is contained in:
Sajid
2024-09-09 21:09:37 +06:00
parent 25ede377a1
commit 60edcad576
6 changed files with 310 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

191
PowerAnalyse/main.cpp
View File

@@ -1,90 +1,167 @@
#include <cassert>
#include <file.h>
#include <disasm.h>
#include <image.h>
#include <format>
#include "function.h"
#include <print>
#include <ppc.h>
int main()
{
const auto file = LoadFile("add.elf");
const auto file = LoadFile("cond.elf");
auto image = Image::ParseImage(file.data(), file.size()).value();
FILE* f = fopen("add.elf.cpp", "w");
for (const auto& section : image.sections)
{
image.symbols.emplace(section.name, section.base, section.size, Symbol_Section);
}
// image.symbols.emplace("_start", image.entry_point, 0x30, Symbol_Function);
//ppc_insn insn;
//uint8_t c[4] = { 0x10, 0x00, 0x59, 0xC3 };
//ppc::Disassemble(c, 0x831D6C64, insn);
//std::println("{:20}{}", insn.opcode->name, insn.op_str);
std::vector<function> functions;
for (const auto& section : image.sections)
{
if (!(section.flags & SectionFlags_Code))
{
continue;
}
size_t base = section.base;
uint8_t* data = section.data;
uint8_t* dataEnd = section.data + section.size;
while (data < dataEnd)
{
if (*(uint32_t*)data == 0)
{
data += 4;
base += 4;
continue;
}
const auto& fn = functions.emplace_back(function::Analyze(data, dataEnd - data, base));
data += fn.size;
base += fn.size;
image.symbols.emplace(std::format("sub_{:X}", fn.base), fn.base, fn.size, Symbol_Function);
}
}
const auto entrySymbol = image.symbols.find(image.entry_point);
assert(entrySymbol != image.symbols.end());
const auto entrySize = entrySymbol->size;
image.symbols.erase(entrySymbol);
image.symbols.emplace("_start", image.entry_point, entrySize, Symbol_Function);
std::println("FUNCTIONS");
for (const auto& fn : functions)
{
std::println("\tsub_{:X}", fn.base);
}
std::println("");
std::println("SECTIONS");
for (const auto& section : image.sections)
{
std::printf("Section %.8s\n", section.name.c_str());
std::printf("\t%X-%X\n", section.base, section.base + section.size);
}
auto* data = (uint32_t*)section.data;
auto base = section.base;
const auto end = section.base + section.size;
std::println("");
if (section.flags & SectionFlags_Code)
FILE* f = fopen("add.elf.cpp", "w");
for (const auto& fn : functions)
{
auto base = fn.base;
auto end = base + fn.size;
auto* data = (uint32_t*)image.Find(base);
std::string name = "";
auto symbol = image.symbols.find(base);
if (symbol != image.symbols.end())
{
ppc_insn insn;
while(base < end)
{
ppc::Disassemble(data, 4, base, insn);
name = symbol->name;
}
else
{
name = std::format("sub_{:X}", base);
}
base += 4;
++data;
if (insn.opcode == nullptr)
std::println(f, "void {}() {{", name);
ppc_insn insn;
while (base < end)
{
ppc::Disassemble(data, 4, base, insn);
base += 4;
++data;
if (insn.opcode == nullptr)
{
std::println(f, "\t// {:x} {}", base - 4, insn.op_str);
}
else
{
std::println(f, "\t// {:x} {} {}", base - 4, insn.opcode->name, insn.op_str);
switch (insn.opcode->id)
{
std::println(f, "// {:x} {}", base - 4, insn.op_str);
}
else
{
std::println(f, "// {:x} {} {}", base - 4, insn.opcode->name, insn.op_str);
switch (insn.opcode->id)
case PPC_INST_ADD:
std::println(f, "\tr{} = r{} + r{};", insn.operands[0], insn.operands[1], insn.operands[2]);
break;
case PPC_INST_ADDI:
std::println(f, "\tr{} = r{} + {};", insn.operands[0], insn.operands[1], insn.operands[2]);
break;
case PPC_INST_STWU:
std::println(f, "\tea = r{} + {};", insn.operands[2], static_cast<int32_t>(insn.operands[1]));
std::println(f, "\t*ea = byteswap(r{});", insn.operands[0]);
std::println(f, "\tr{} = ea;", insn.operands[2]);
break;
case PPC_INST_STW:
std::println(f, "\t*(r{} + {}) = byteswap(r{});", insn.operands[2], static_cast<int32_t>(insn.operands[1]), insn.operands[0]);
break;
case PPC_INST_MR:
std::println(f, "\tr{} = r{};", insn.operands[0], insn.operands[1]);
break;
case PPC_INST_LWZ:
std::println(f, "\tr{} = *(r{} + {});", insn.operands[0], insn.operands[2], insn.operands[1]);
break;
case PPC_INST_LI:
std::println(f, "\tr{} = {};", insn.operands[0], insn.operands[1]);
break;
case PPC_INST_MFLR:
std::println(f, "\tr{} = lr;", insn.operands[0]);
break;
case PPC_INST_MTLR:
std::println(f, "\tlr = r{};", insn.operands[0]);
break;
case PPC_INST_BLR:
std::println(f, "\treturn;");
break;
case PPC_INST_BL:
{
case PPC_INST_ADD:
std::println(f, "r{} = r{} + r{};", insn.operands[0], insn.operands[1], insn.operands[2]);
break;
case PPC_INST_ADDI:
std::println(f, "r{} = r{} + {};", insn.operands[0], insn.operands[1], insn.operands[2]);
break;
case PPC_INST_STWU:
std::println(f, "ea = r{} + {};", insn.operands[2], static_cast<int32_t>(insn.operands[1]));
std::println(f, "*ea = byteswap(r{});", insn.operands[0]);
std::println(f, "r{} = ea;", insn.operands[2]);
break;
case PPC_INST_STW:
std::println(f, "*(r{} + {}) = byteswap(r{});", insn.operands[2], static_cast<int32_t>(insn.operands[1]), insn.operands[0]);
break;
case PPC_INST_MR:
std::println(f, "r{} = r{};", insn.operands[0], insn.operands[1]);
break;
case PPC_INST_LWZ:
std::println(f, "r{} = *(r{} + {});", insn.operands[0], insn.operands[2], insn.operands[1]);
break;
case PPC_INST_LI:
std::println(f, "r{} = {};", insn.operands[0], insn.operands[1]);
break;
case PPC_INST_MFLR:
std::println(f, "r{} = lr;", insn.operands[0]);
break;
case PPC_INST_MTLR:
std::println(f, "lr = r{};", insn.operands[0]);
break;
case PPC_INST_BLR:
std::println(f, "return;");
break;
case PPC_INST_BL:
std::println(f, "lr = 0x{:x};", insn.operands[0]);
std::println(f, "sub_{:x}();", insn.operands[0]);
std::string targetName = "";
auto targetSymbol = image.symbols.find(insn.operands[0]);
if (targetSymbol != image.symbols.end() && targetSymbol->type == Symbol_Function)
{
targetName = targetSymbol->name;
}
else
{
targetName = std::format("sub_{:X}", insn.operands[0]);
}
std::println(f, "\tlr = 0x{:x};", base);
std::println(f, "\t{}();", targetName);
break;
}
}
}
}
std::println(f, "}}\n");
}
fclose(f);