63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
# Copyright 2021 Dolphin Emulator Project
|
|
# Licensed under GPLv2+
|
|
# Refer to the LICENSES/GPL-2.0-or-later.txt file included.
|
|
|
|
from collections import namedtuple
|
|
|
|
|
|
DolphinSymbol = namedtuple("DolphinSymbol", [
|
|
"section", "addr", "size", "vaddr", "align", "name"
|
|
])
|
|
|
|
|
|
def load_dolphin_map(filepath):
|
|
with open(filepath, "r") as f:
|
|
section = ""
|
|
symbol_map = []
|
|
for line in f.readlines():
|
|
t = line.strip().split(" ", 4)
|
|
if len(t) == 3 and t[1] == "section" and t[2] == "layout":
|
|
section = t[0]
|
|
continue
|
|
if not section or len(t) != 5:
|
|
continue
|
|
symbol_map.append(DolphinSymbol(section, *t))
|
|
return symbol_map
|
|
|
|
|
|
def ida_main():
|
|
import idc
|
|
|
|
filepath = ida_kernwin.ask_file(0, "*.map", "Load a Dolphin emulator symbol map")
|
|
if filepath is None:
|
|
return
|
|
symbol_map = load_dolphin_map(filepath)
|
|
|
|
for symbol in symbol_map:
|
|
addr = int(symbol.vaddr, 16)
|
|
size = int(symbol.size, 16)
|
|
ida_bytes.del_items(addr, size, 0)
|
|
if symbol.section in [".init", ".text"]:
|
|
idc.create_insn(addr)
|
|
success = ida_funcs.add_func(
|
|
addr,
|
|
idc.BADADDR if not size else (addr+size)
|
|
)
|
|
else:
|
|
success = ida_bytes.create_data(addr, idc.FF_BYTE, size, 0)
|
|
|
|
if not success:
|
|
ida_kernwin.msg("Can't apply properties for symbol:"
|
|
" {0.vaddr} - {0.name}\n".format(symbol))
|
|
|
|
flags = idc.SN_NOCHECK | idc.SN_PUBLIC
|
|
if symbol.name.startswith("zz_"):
|
|
flags |= idc.SN_AUTO | idc.SN_WEAK
|
|
else:
|
|
flags |= idc.SN_NON_AUTO
|
|
idc.set_name(addr, symbol.name, flags)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
ida_main()
|