mirror of
https://github.com/hedge-dev/XenonRecomp.git
synced 2025-04-23 01:01:26 +00:00
475 lines
16 KiB
C++
475 lines
16 KiB
C++
#include <file.h>
|
|
#include <image.h>
|
|
#include <function.h>
|
|
#include <format>
|
|
#include <print>
|
|
#include <disasm.h>
|
|
#include <filesystem>
|
|
|
|
#define TEST_FILE "add.elf"
|
|
|
|
int main()
|
|
{
|
|
const auto file = LoadFile(TEST_FILE).value();
|
|
auto image = Image::ParseImage(file.data(), file.size()).value();
|
|
|
|
// TODO: Load functions from an existing database
|
|
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);
|
|
}
|
|
}
|
|
|
|
std::filesystem::create_directory("out");
|
|
FILE* f = fopen("out/" TEST_FILE ".cpp", "w");
|
|
std::println(f, "#include <ppc_context.h>\n");
|
|
|
|
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())
|
|
{
|
|
name = symbol->name;
|
|
}
|
|
else
|
|
{
|
|
name = std::format("sub_{:X}", base);
|
|
}
|
|
|
|
std::println(f, "void {}(PPCContext& __restrict ctx, uint8_t* __restrict base) {{", name);
|
|
std::println(f, "\tuint32_t ea;\n");
|
|
|
|
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)
|
|
{
|
|
case PPC_INST_ADD:
|
|
// TODO: . variant
|
|
std::println(f, "\tctx.r{}.u64 = ctx.r{}.u64 + ctx.r{}.u64;", insn.operands[0], insn.operands[1], insn.operands[2]);
|
|
break;
|
|
|
|
case PPC_INST_ADDI:
|
|
std::println(f, "\tctx.r{}.u64 = ctx.r{}.u64 + {};", insn.operands[0], insn.operands[1], insn.operands[2]);
|
|
break;
|
|
|
|
case PPC_INST_ADDIC:
|
|
case PPC_INST_ADDIS:
|
|
case PPC_INST_ADDZE:
|
|
case PPC_INST_AND:
|
|
case PPC_INST_ANDC:
|
|
case PPC_INST_ANDI:
|
|
case PPC_INST_ANDIS:
|
|
case PPC_INST_ATTN:
|
|
case PPC_INST_B:
|
|
case PPC_INST_BCTR:
|
|
case PPC_INST_BCTRL:
|
|
case PPC_INST_BDNZ:
|
|
case PPC_INST_BDNZF:
|
|
case PPC_INST_BEQ:
|
|
case PPC_INST_BEQLR:
|
|
case PPC_INST_BGE:
|
|
case PPC_INST_BGELR:
|
|
case PPC_INST_BGT:
|
|
case PPC_INST_BGTLR:
|
|
break;
|
|
|
|
case PPC_INST_BL:
|
|
{
|
|
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, "\tctx.lr = 0x{:X};", base);
|
|
std::println(f, "\t{}(ctx, base);", targetName);
|
|
break;
|
|
}
|
|
|
|
case PPC_INST_BLE:
|
|
case PPC_INST_BLELR:
|
|
break;
|
|
|
|
case PPC_INST_BLR:
|
|
// TODO: do they use this for anything other than a return?
|
|
std::println(f, "\treturn;");
|
|
break;
|
|
|
|
case PPC_INST_BLRL:
|
|
case PPC_INST_BLT:
|
|
case PPC_INST_BLTLR:
|
|
case PPC_INST_BNE:
|
|
case PPC_INST_BNECTR:
|
|
case PPC_INST_BNELR:
|
|
case PPC_INST_CCTPL:
|
|
case PPC_INST_CCTPM:
|
|
case PPC_INST_CLRLDI:
|
|
case PPC_INST_CLRLWI:
|
|
case PPC_INST_CMPD:
|
|
case PPC_INST_CMPDI:
|
|
case PPC_INST_CMPLD:
|
|
case PPC_INST_CMPLDI:
|
|
case PPC_INST_CMPLW:
|
|
case PPC_INST_CMPLWI:
|
|
case PPC_INST_CMPW:
|
|
case PPC_INST_CMPWI:
|
|
case PPC_INST_CNTLZD:
|
|
case PPC_INST_CNTLZW:
|
|
case PPC_INST_DB16CYC:
|
|
case PPC_INST_DCBF:
|
|
case PPC_INST_DCBT:
|
|
case PPC_INST_DCBTST:
|
|
case PPC_INST_DCBZ:
|
|
case PPC_INST_DCBZL:
|
|
case PPC_INST_DIVD:
|
|
case PPC_INST_DIVDU:
|
|
case PPC_INST_DIVW:
|
|
case PPC_INST_DIVWU:
|
|
case PPC_INST_EIEIO:
|
|
case PPC_INST_EXTSB:
|
|
case PPC_INST_EXTSH:
|
|
case PPC_INST_EXTSW:
|
|
case PPC_INST_FABS:
|
|
case PPC_INST_FADD:
|
|
case PPC_INST_FADDS:
|
|
case PPC_INST_FCFID:
|
|
case PPC_INST_FCMPU:
|
|
case PPC_INST_FCTID:
|
|
case PPC_INST_FCTIDZ:
|
|
case PPC_INST_FCTIWZ:
|
|
case PPC_INST_FDIV:
|
|
case PPC_INST_FDIVS:
|
|
case PPC_INST_FMADD:
|
|
case PPC_INST_FMADDS:
|
|
case PPC_INST_FMR:
|
|
case PPC_INST_FMSUB:
|
|
case PPC_INST_FMSUBS:
|
|
case PPC_INST_FMUL:
|
|
case PPC_INST_FMULS:
|
|
case PPC_INST_FNABS:
|
|
case PPC_INST_FNEG:
|
|
case PPC_INST_FNMADDS:
|
|
case PPC_INST_FNMSUB:
|
|
case PPC_INST_FNMSUBS:
|
|
case PPC_INST_FRES:
|
|
case PPC_INST_FRSP:
|
|
case PPC_INST_FSEL:
|
|
case PPC_INST_FSQRT:
|
|
case PPC_INST_FSQRTS:
|
|
case PPC_INST_FSUB:
|
|
case PPC_INST_FSUBS:
|
|
case PPC_INST_LBZ:
|
|
case PPC_INST_LBZU:
|
|
case PPC_INST_LBZX:
|
|
case PPC_INST_LD:
|
|
case PPC_INST_LDARX:
|
|
case PPC_INST_LDU:
|
|
case PPC_INST_LDX:
|
|
case PPC_INST_LFD:
|
|
case PPC_INST_LFDX:
|
|
case PPC_INST_LFS:
|
|
case PPC_INST_LFSX:
|
|
case PPC_INST_LHA:
|
|
case PPC_INST_LHAX:
|
|
case PPC_INST_LHZ:
|
|
case PPC_INST_LHZX:
|
|
break;
|
|
|
|
case PPC_INST_LI:
|
|
std::println(f, "\tctx.r{}.u64 = {};", insn.operands[0], insn.operands[1]);
|
|
break;
|
|
|
|
case PPC_INST_LIS:
|
|
case PPC_INST_LVEWX:
|
|
case PPC_INST_LVEWX128:
|
|
case PPC_INST_LVLX:
|
|
case PPC_INST_LVLX128:
|
|
case PPC_INST_LVRX:
|
|
case PPC_INST_LVRX128:
|
|
case PPC_INST_LVSL:
|
|
case PPC_INST_LVSR:
|
|
case PPC_INST_LVX:
|
|
case PPC_INST_LVX128:
|
|
case PPC_INST_LWA:
|
|
case PPC_INST_LWARX:
|
|
case PPC_INST_LWAX:
|
|
case PPC_INST_LWBRX:
|
|
case PPC_INST_LWSYNC:
|
|
break;
|
|
|
|
case PPC_INST_LWZ:
|
|
std::println(f, "\tctx.r{}.u64 = PPC_LOAD_U32({} + ctx.r{}.u32);", insn.operands[0], int32_t(insn.operands[1]), insn.operands[2]);
|
|
break;
|
|
|
|
case PPC_INST_LWZU:
|
|
case PPC_INST_LWZX:
|
|
case PPC_INST_MFCR:
|
|
case PPC_INST_MFFS:
|
|
break;
|
|
|
|
case PPC_INST_MFLR:
|
|
std::println(f, "\tctx.lr = ctx.r{}.u64;", insn.operands[0]);
|
|
break;
|
|
|
|
case PPC_INST_MFMSR:
|
|
case PPC_INST_MFOCRF:
|
|
case PPC_INST_MFTB:
|
|
break;
|
|
|
|
case PPC_INST_MR:
|
|
// TODO: . variant
|
|
std::println(f, "\tctx.r{}.u64 = ctx.r{}.u64;", insn.operands[0], insn.operands[1]);
|
|
break;
|
|
|
|
case PPC_INST_MTCR:
|
|
case PPC_INST_MTCTR:
|
|
case PPC_INST_MTFSF:
|
|
break;
|
|
|
|
case PPC_INST_MTLR:
|
|
std::println(f, "\tctx.r{}.u64 = ctx.lr;", insn.operands[0]);
|
|
break;
|
|
|
|
case PPC_INST_MTMSRD:
|
|
case PPC_INST_MTXER:
|
|
case PPC_INST_MULCHWU:
|
|
case PPC_INST_MULHHW:
|
|
case PPC_INST_MULHW:
|
|
case PPC_INST_MULHWU:
|
|
case PPC_INST_MULLD:
|
|
case PPC_INST_MULLI:
|
|
case PPC_INST_MULLW:
|
|
case PPC_INST_NAND:
|
|
case PPC_INST_NEG:
|
|
case PPC_INST_NOP:
|
|
case PPC_INST_NOR:
|
|
case PPC_INST_NOT:
|
|
case PPC_INST_OR:
|
|
case PPC_INST_ORC:
|
|
case PPC_INST_ORI:
|
|
case PPC_INST_ORIS:
|
|
case PPC_INST_RLDICL:
|
|
case PPC_INST_RLDICR:
|
|
case PPC_INST_RLDIMI:
|
|
case PPC_INST_RLWIMI:
|
|
case PPC_INST_RLWINM:
|
|
case PPC_INST_ROTLDI:
|
|
case PPC_INST_ROTLW:
|
|
case PPC_INST_ROTLWI:
|
|
case PPC_INST_SLD:
|
|
case PPC_INST_SLW:
|
|
case PPC_INST_SRAD:
|
|
case PPC_INST_SRADI:
|
|
case PPC_INST_SRAW:
|
|
case PPC_INST_SRAWI:
|
|
case PPC_INST_SRD:
|
|
case PPC_INST_SRW:
|
|
case PPC_INST_STB:
|
|
case PPC_INST_STBU:
|
|
case PPC_INST_STBX:
|
|
case PPC_INST_STD:
|
|
case PPC_INST_STDCX:
|
|
case PPC_INST_STDU:
|
|
case PPC_INST_STDX:
|
|
case PPC_INST_STFD:
|
|
case PPC_INST_STFDX:
|
|
case PPC_INST_STFIWX:
|
|
case PPC_INST_STFS:
|
|
case PPC_INST_STFSX:
|
|
case PPC_INST_STH:
|
|
case PPC_INST_STHBRX:
|
|
case PPC_INST_STHX:
|
|
case PPC_INST_STVEHX:
|
|
case PPC_INST_STVEWX:
|
|
case PPC_INST_STVEWX128:
|
|
case PPC_INST_STVLX:
|
|
case PPC_INST_STVLX128:
|
|
case PPC_INST_STVRX:
|
|
case PPC_INST_STVRX128:
|
|
case PPC_INST_STVX:
|
|
case PPC_INST_STVX128:
|
|
break;
|
|
|
|
case PPC_INST_STW:
|
|
std::println(f, "\tPPC_STORE_U32({} + ctx.r{}.u32, ctx.r{}.u32);", int32_t(insn.operands[1]), insn.operands[2], insn.operands[0]);
|
|
break;
|
|
|
|
case PPC_INST_STWBRX:
|
|
case PPC_INST_STWCX:
|
|
break;
|
|
|
|
case PPC_INST_STWU:
|
|
std::println(f, "\tea = {} + ctx.r{}.u32;", int32_t(insn.operands[1]), insn.operands[2]);
|
|
std::println(f, "\tPPC_STORE_U32(ea, ctx.r{}.u32);", insn.operands[0]);
|
|
std::println(f, "\tctx.r{}.u64 = ea;", insn.operands[0]);
|
|
break;
|
|
|
|
case PPC_INST_STWUX:
|
|
case PPC_INST_STWX:
|
|
case PPC_INST_SUBF:
|
|
case PPC_INST_SUBFC:
|
|
case PPC_INST_SUBFE:
|
|
case PPC_INST_SUBFIC:
|
|
case PPC_INST_SYNC:
|
|
case PPC_INST_TDLGEI:
|
|
case PPC_INST_TDLLEI:
|
|
case PPC_INST_TWI:
|
|
case PPC_INST_TWLGEI:
|
|
case PPC_INST_TWLLEI:
|
|
case PPC_INST_VADDFP:
|
|
case PPC_INST_VADDFP128:
|
|
case PPC_INST_VADDSHS:
|
|
case PPC_INST_VADDUBM:
|
|
case PPC_INST_VADDUBS:
|
|
case PPC_INST_VADDUHM:
|
|
case PPC_INST_VADDUWM:
|
|
case PPC_INST_VADDUWS:
|
|
case PPC_INST_VAND:
|
|
case PPC_INST_VAND128:
|
|
case PPC_INST_VANDC128:
|
|
case PPC_INST_VAVGSB:
|
|
case PPC_INST_VAVGSH:
|
|
case PPC_INST_VAVGUB:
|
|
case PPC_INST_VCFPSXWS128:
|
|
case PPC_INST_VCFSX:
|
|
case PPC_INST_VCFUX:
|
|
case PPC_INST_VCMPBFP128:
|
|
case PPC_INST_VCMPEQFP:
|
|
case PPC_INST_VCMPEQFP128:
|
|
case PPC_INST_VCMPEQUB:
|
|
case PPC_INST_VCMPEQUW:
|
|
case PPC_INST_VCMPEQUW128:
|
|
case PPC_INST_VCMPGEFP:
|
|
case PPC_INST_VCMPGEFP128:
|
|
case PPC_INST_VCMPGTFP:
|
|
case PPC_INST_VCMPGTFP128:
|
|
case PPC_INST_VCMPGTUB:
|
|
case PPC_INST_VCMPGTUH:
|
|
case PPC_INST_VCSXWFP128:
|
|
case PPC_INST_VCTSXS:
|
|
case PPC_INST_VCUXWFP128:
|
|
case PPC_INST_VEXPTEFP128:
|
|
case PPC_INST_VLOGEFP128:
|
|
case PPC_INST_VMADDCFP128:
|
|
case PPC_INST_VMADDFP:
|
|
case PPC_INST_VMADDFP128:
|
|
case PPC_INST_VMAXFP:
|
|
case PPC_INST_VMAXFP128:
|
|
case PPC_INST_VMAXSW:
|
|
case PPC_INST_VMINFP:
|
|
case PPC_INST_VMINFP128:
|
|
case PPC_INST_VMRGHB:
|
|
case PPC_INST_VMRGHH:
|
|
case PPC_INST_VMRGHW:
|
|
case PPC_INST_VMRGHW128:
|
|
case PPC_INST_VMRGLB:
|
|
case PPC_INST_VMRGLH:
|
|
case PPC_INST_VMRGLW:
|
|
case PPC_INST_VMRGLW128:
|
|
case PPC_INST_VMSUM3FP128:
|
|
case PPC_INST_VMSUM4FP128:
|
|
case PPC_INST_VMULFP128:
|
|
case PPC_INST_VNMSUBFP:
|
|
case PPC_INST_VNMSUBFP128:
|
|
case PPC_INST_VOR:
|
|
case PPC_INST_VOR128:
|
|
case PPC_INST_VPERM:
|
|
case PPC_INST_VPERM128:
|
|
case PPC_INST_VPERMWI128:
|
|
case PPC_INST_VPKD3D128:
|
|
case PPC_INST_VPKSHUS:
|
|
case PPC_INST_VREFP:
|
|
case PPC_INST_VREFP128:
|
|
case PPC_INST_VRFIM128:
|
|
case PPC_INST_VRFIN:
|
|
case PPC_INST_VRFIN128:
|
|
case PPC_INST_VRFIZ128:
|
|
case PPC_INST_VRLIMI128:
|
|
case PPC_INST_VRSQRTEFP:
|
|
case PPC_INST_VRSQRTEFP128:
|
|
case PPC_INST_VSEL:
|
|
case PPC_INST_VSLB:
|
|
case PPC_INST_VSLDOI:
|
|
case PPC_INST_VSLDOI128:
|
|
case PPC_INST_VSLW128:
|
|
case PPC_INST_VSPLTH:
|
|
case PPC_INST_VSPLTISB:
|
|
case PPC_INST_VSPLTISW:
|
|
case PPC_INST_VSPLTISW128:
|
|
case PPC_INST_VSPLTW:
|
|
case PPC_INST_VSPLTW128:
|
|
case PPC_INST_VSR:
|
|
case PPC_INST_VSRAW128:
|
|
case PPC_INST_VSRW:
|
|
case PPC_INST_VSRW128:
|
|
case PPC_INST_VSUBFP:
|
|
case PPC_INST_VSUBFP128:
|
|
case PPC_INST_VSUBSWS:
|
|
case PPC_INST_VSUBUBS:
|
|
case PPC_INST_VSUBUHM:
|
|
case PPC_INST_VUPKD3D128:
|
|
case PPC_INST_VUPKHSB128:
|
|
case PPC_INST_VUPKHSH:
|
|
case PPC_INST_VUPKLSB128:
|
|
case PPC_INST_VUPKLSH:
|
|
case PPC_INST_VXOR:
|
|
case PPC_INST_VXOR128:
|
|
case PPC_INST_XOR:
|
|
case PPC_INST_XORI:
|
|
case PPC_INST_XORIS:
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
std::println(f, "}}\n");
|
|
}
|
|
|
|
fclose(f);
|
|
|
|
return 0;
|
|
}
|