mirror of
https://github.com/pmret/papermario.git
synced 2024-11-10 04:52:34 +01:00
66 lines
1.4 KiB
Python
66 lines
1.4 KiB
Python
|
#! /usr/bin/env python3
|
||
|
|
||
|
from capstone import *
|
||
|
from capstone.mips import *
|
||
|
|
||
|
import argparse
|
||
|
from util import rominfo
|
||
|
from segtypes.code import N64SegCode
|
||
|
|
||
|
parser = argparse.ArgumentParser(description="Create a splat config from a rom")
|
||
|
parser.add_argument("rom", help="path to a .z64 rom")
|
||
|
|
||
|
|
||
|
def main(rom_path):
|
||
|
rom = rominfo.get_info(rom_path)
|
||
|
basename = rom.name.replace(" ", "").lower()
|
||
|
|
||
|
header = \
|
||
|
"""name: {0} ({1})
|
||
|
basename: {2}
|
||
|
options:
|
||
|
find_file_boundaries: True
|
||
|
compiler: "IDO"
|
||
|
""".format(rom.name.title(), rom.get_country_name(), basename)
|
||
|
|
||
|
with open(rom_path, "rb") as f:
|
||
|
fbytes = f.read()
|
||
|
|
||
|
rom_addr = 0x1000
|
||
|
|
||
|
md = Cs(CS_ARCH_MIPS, CS_MODE_MIPS64 + CS_MODE_BIG_ENDIAN)
|
||
|
for insn in md.disasm(fbytes[rom_addr:], rom.entry_point):
|
||
|
rom_addr += 4
|
||
|
|
||
|
segments = \
|
||
|
"""segments:
|
||
|
- name: header
|
||
|
type: header
|
||
|
start: 0x0
|
||
|
vram: 0
|
||
|
files:
|
||
|
- [0x0, header, header]
|
||
|
- name: boot
|
||
|
type: bin
|
||
|
start: 0x40
|
||
|
- name: main
|
||
|
type: code
|
||
|
start: 0x1000
|
||
|
vram: 0x{:X}
|
||
|
files:
|
||
|
- [0x1000, asm]
|
||
|
- type: bin
|
||
|
start: 0x{:X}
|
||
|
- [0x{:X}]
|
||
|
""".format(rom.entry_point, rom_addr, rom.size)
|
||
|
|
||
|
outstr = header + segments
|
||
|
|
||
|
outname = rom.name.replace(" ", "").lower()
|
||
|
with open(outname + ".yaml", "w", newline="\n") as f:
|
||
|
f.write(outstr)
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
args = parser.parse_args()
|
||
|
main(args.rom)
|