mirror of
https://github.com/idapython/src
synced 2026-06-08 14:47:00 +00:00
Arnaud Diederen
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
|
|
import ida_dbg
|
|
import ida_idaapi
|
|
import ida_idd
|
|
import ida_kernwin
|
|
import ida_typeinf
|
|
import ida_name
|
|
|
|
def log(msg):
|
|
print(">>> %s" % msg)
|
|
|
|
class appcall_hooks_t(ida_dbg.DBG_Hooks):
|
|
def __init__(self, name_funcs=[]):
|
|
ida_dbg.DBG_Hooks.__init__(self) # important
|
|
|
|
for ea, func_name in name_funcs:
|
|
log("Renaming 0x%08x to \"%s\"" % (ea, func_name))
|
|
ida_name.set_name(ea, func_name)
|
|
|
|
for func_name, func_proto in [
|
|
("ref4", "int ref4(int *);"),
|
|
("ref8", "int ref8(long long int *);"),
|
|
]:
|
|
log("Setting '%s's prototype" % func_name)
|
|
func_ea = ida_name.get_name_ea(ida_idaapi.BADADDR, func_name)
|
|
assert(ida_typeinf.apply_cdecl(None, func_ea, func_proto))
|
|
|
|
|
|
def dbg_run_to(self, pid, tid, ea):
|
|
log("'run_to' reached its target location. Performing appcalls.")
|
|
|
|
for func_name in ["ref4", "ref8"]:
|
|
int_value = ida_idd.Appcall.int64(5)
|
|
int_ptr = ida_idd.Appcall.byref(int_value)
|
|
if ida_idd.Appcall[func_name](int_ptr):
|
|
log("Appcall (%s) succeeded: int_value.value=%s, int_ptr.value=%s" % (
|
|
func_name,
|
|
int_value.value,
|
|
int_ptr.value))
|
|
else:
|
|
log("Appcall (%s) failed" % func_name)
|
|
|
|
|
|
def run(self):
|
|
log("Running program up to current address, and letting the hooks do the rest")
|
|
assert(ida_dbg.run_to(ida_kernwin.get_screen_ea()))
|