mirror of
https://github.com/encounter/decomp.me.git
synced 2026-03-30 11:06:27 -07:00
Mark Street
73 lines
1.9 KiB
Python
73 lines
1.9 KiB
Python
import contextlib
|
|
import io
|
|
import logging
|
|
|
|
from m2c.main import parse_flags, run
|
|
|
|
from coreapp.compilers import Compiler
|
|
|
|
from coreapp.sandbox import Sandbox
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class M2CError(Exception):
|
|
pass
|
|
|
|
|
|
class M2CWrapper:
|
|
@staticmethod
|
|
def get_triple(compiler: Compiler, arch: str) -> str:
|
|
if "mipse" in arch:
|
|
t_arch = "mipsel"
|
|
elif "mips" in arch:
|
|
t_arch = "mips"
|
|
elif "ppc" in arch:
|
|
t_arch = "ppc"
|
|
else:
|
|
raise M2CError(f"Unsupported arch '{arch}'")
|
|
|
|
if compiler.is_ido:
|
|
t_compiler = "ido"
|
|
elif compiler.is_gcc:
|
|
t_compiler = "gcc"
|
|
elif compiler.is_mwcc:
|
|
t_compiler = "mwcc"
|
|
else:
|
|
raise M2CError(f"Unsupported compiler '{compiler}'")
|
|
|
|
return f"{t_arch}-{t_compiler}"
|
|
|
|
@staticmethod
|
|
def decompile(asm: str, context: str, compiler: Compiler, arch: str) -> str:
|
|
with Sandbox() as sandbox:
|
|
flags = ["--stop-on-error", "--pointer-style=left"]
|
|
|
|
flags.append(f"--target={M2CWrapper.get_triple(compiler, arch)}")
|
|
|
|
# Create temp asm file
|
|
asm_path = sandbox.path / "asm.s"
|
|
asm_path.write_text(asm)
|
|
|
|
if context:
|
|
# Create temp context file
|
|
ctx_path = sandbox.path / "ctx.c"
|
|
ctx_path.write_text(context)
|
|
|
|
flags.append("--context")
|
|
flags.append(str(ctx_path))
|
|
|
|
flags.append(str(asm_path))
|
|
options = parse_flags(flags)
|
|
|
|
out_string = io.StringIO()
|
|
with contextlib.redirect_stdout(out_string):
|
|
returncode = run(options)
|
|
|
|
out_text = out_string.getvalue()
|
|
|
|
if returncode == 0:
|
|
return out_text
|
|
else:
|
|
raise M2CError(out_text)
|