mirror of
https://github.com/idapython/src
synced 2026-06-08 14:47:00 +00:00
Arnaud Diederen
200 lines
6.0 KiB
Python
200 lines
6.0 KiB
Python
#<pycode(py_ida)>
|
|
import sys
|
|
|
|
def __make_idainfo_bound(func, attr):
|
|
def __func(self, *args):
|
|
return func(*args)
|
|
setattr(idainfo, attr, __func)
|
|
|
|
_NO_SETTER = "<nosetter>"
|
|
def __make_idainfo_accessors(
|
|
attr,
|
|
getter_name=None,
|
|
setter_name=None):
|
|
if getter_name is None:
|
|
getter_name = attr
|
|
getter = globals()["idainfo_%s" % getter_name]
|
|
__make_idainfo_bound(getter, getter_name)
|
|
if setter_name != _NO_SETTER:
|
|
if setter_name is None:
|
|
setter_name = "set_%s" % attr
|
|
setter = globals()["idainfo_%s" % setter_name]
|
|
__make_idainfo_bound(setter, setter_name)
|
|
|
|
def __make_idainfo_getter(name):
|
|
return __make_idainfo_accessors(None, getter_name=name, setter_name=_NO_SETTER)
|
|
|
|
|
|
idainfo_big_arg_align = inf_big_arg_align
|
|
__make_idainfo_getter("big_arg_align")
|
|
|
|
idainfo_gen_null = inf_gen_null
|
|
idainfo_set_gen_null = inf_set_gen_null
|
|
__make_idainfo_accessors("gen_null")
|
|
|
|
idainfo_gen_lzero = inf_gen_lzero
|
|
idainfo_set_gen_lzero = inf_set_gen_lzero
|
|
__make_idainfo_accessors("gen_lzero")
|
|
|
|
idainfo_gen_tryblks = inf_gen_tryblks
|
|
idainfo_set_gen_tryblks = inf_set_gen_tryblks
|
|
__make_idainfo_accessors("gen_tryblks")
|
|
|
|
idainfo_get_demname_form = inf_get_demname_form
|
|
__make_idainfo_getter("get_demname_form")
|
|
|
|
idainfo_get_pack_mode = inf_get_pack_mode
|
|
idainfo_set_pack_mode = inf_set_pack_mode
|
|
__make_idainfo_accessors(None, "get_pack_mode", "set_pack_mode")
|
|
|
|
def idainfo_is_32bit(): return not inf_is_16bit() # in reality this means "is 32bit or higher"
|
|
__make_idainfo_getter("is_32bit")
|
|
|
|
idainfo_is_64bit = inf_is_64bit
|
|
idainfo_set_64bit = inf_set_64bit
|
|
__make_idainfo_accessors(None, "is_64bit", "set_64bit")
|
|
|
|
idainfo_is_auto_enabled = inf_is_auto_enabled
|
|
idainfo_set_auto_enabled = inf_set_auto_enabled
|
|
__make_idainfo_accessors(None, "is_auto_enabled", "set_auto_enabled")
|
|
|
|
idainfo_is_be = inf_is_be
|
|
idainfo_set_be = inf_set_be
|
|
__make_idainfo_accessors(None, "is_be", "set_be")
|
|
|
|
idainfo_is_dll = inf_is_dll
|
|
__make_idainfo_getter("is_dll")
|
|
|
|
idainfo_is_flat_off32 = inf_is_flat_off32
|
|
__make_idainfo_getter("is_flat_off32")
|
|
|
|
idainfo_is_graph_view = inf_is_graph_view
|
|
idainfo_set_graph_view = inf_set_graph_view
|
|
__make_idainfo_accessors(None, "is_graph_view", "set_graph_view")
|
|
|
|
idainfo_is_hard_float = inf_is_hard_float
|
|
__make_idainfo_getter("is_hard_float")
|
|
|
|
idainfo_is_kernel_mode = inf_is_kernel_mode
|
|
__make_idainfo_getter("is_kernel_mode")
|
|
|
|
idainfo_is_mem_aligned4 = inf_is_mem_aligned4
|
|
__make_idainfo_getter("is_mem_aligned4")
|
|
|
|
idainfo_is_snapshot = inf_is_snapshot
|
|
__make_idainfo_getter("is_snapshot")
|
|
|
|
idainfo_is_wide_high_byte_first = inf_is_wide_high_byte_first
|
|
idainfo_set_wide_high_byte_first = inf_set_wide_high_byte_first
|
|
__make_idainfo_accessors(None, "is_wide_high_byte_first", "set_wide_high_byte_first")
|
|
|
|
idainfo_like_binary = inf_like_binary
|
|
__make_idainfo_getter("like_binary")
|
|
|
|
idainfo_line_pref_with_seg = inf_line_pref_with_seg
|
|
idainfo_set_line_pref_with_seg = inf_set_line_pref_with_seg
|
|
__make_idainfo_accessors("line_pref_with_seg")
|
|
|
|
idainfo_show_auto = inf_show_auto
|
|
idainfo_set_show_auto = inf_set_show_auto
|
|
__make_idainfo_accessors("show_auto")
|
|
|
|
idainfo_show_line_pref = inf_show_line_pref
|
|
idainfo_set_show_line_pref = inf_set_show_line_pref
|
|
__make_idainfo_accessors("show_line_pref")
|
|
|
|
idainfo_show_void = inf_show_void
|
|
idainfo_set_show_void = inf_set_show_void
|
|
__make_idainfo_accessors("show_void")
|
|
|
|
idainfo_loading_idc = inf_loading_idc
|
|
__make_idainfo_getter("loading_idc")
|
|
|
|
idainfo_map_stkargs = inf_map_stkargs
|
|
__make_idainfo_getter("map_stkargs")
|
|
|
|
idainfo_pack_stkargs = inf_pack_stkargs
|
|
__make_idainfo_getter("pack_stkargs")
|
|
|
|
idainfo_readonly_idb = inf_readonly_idb
|
|
__make_idainfo_getter("readonly_idb")
|
|
|
|
idainfo_set_store_user_info = lambda *args: not inf_set_store_user_info()
|
|
|
|
idainfo_stack_ldbl = inf_stack_ldbl
|
|
__make_idainfo_getter("stack_ldbl")
|
|
|
|
idainfo_stack_varargs = inf_stack_varargs
|
|
__make_idainfo_getter("stack_varargs")
|
|
|
|
idainfo_use_allasm = inf_use_allasm
|
|
__make_idainfo_getter("use_allasm")
|
|
|
|
idainfo_use_gcc_layout = inf_use_gcc_layout
|
|
__make_idainfo_getter("use_gcc_layout")
|
|
|
|
macros_enabled = inf_macros_enabled
|
|
should_create_stkvars = inf_should_create_stkvars
|
|
should_trace_sp = inf_should_trace_sp
|
|
show_all_comments = inf_show_all_comments
|
|
show_comments = lambda *args: not inf_hide_comments()
|
|
show_repeatables = inf_show_repeatables
|
|
|
|
inf_get_comment = inf_get_cmt_indent
|
|
inf_set_comment = inf_set_cmt_indent
|
|
idainfo_comment_get = inf_get_cmt_indent
|
|
idainfo_comment_set = inf_set_cmt_indent
|
|
|
|
__make_idainfo_accessors(None, "is_graph_view", "set_graph_view")
|
|
|
|
def __wrap_hooks_callback(klass, new_name, old_name, do_call):
|
|
bkp_name = "__real_%s" % new_name
|
|
def __wrapper(self, *args):
|
|
rc = getattr(self, bkp_name)(*args)
|
|
cb = getattr(self, old_name, None)
|
|
if cb:
|
|
rc = do_call(cb, *args)
|
|
return rc
|
|
|
|
new_cb = getattr(klass, new_name)
|
|
__wrapper.__doc__ = new_cb.__doc__
|
|
setattr(klass, bkp_name, new_cb)
|
|
setattr(__wrapper, "__trampoline", True)
|
|
setattr(klass, new_name, __wrapper)
|
|
return __wrapper
|
|
|
|
|
|
def __set_module_dynattrs(modname, pdict):
|
|
import types
|
|
class _module_wrapper_t(types.ModuleType):
|
|
def __init__(self, orig):
|
|
self.orig = orig
|
|
def __getattribute__(self, name):
|
|
if name in pdict:
|
|
return pdict[name][0]()
|
|
elif name == "orig":
|
|
return types.ModuleType.__getattribute__(self, name)
|
|
elif name == "__dict__":
|
|
d = self.orig.__dict__
|
|
d = d.copy()
|
|
for name in pdict:
|
|
d[name] = pdict[name][0]()
|
|
return d
|
|
else:
|
|
return getattr(self.orig, name)
|
|
def __setattr__(self, name, value):
|
|
if name == "orig":
|
|
types.ModuleType.__setattr__(self, name, value)
|
|
else:
|
|
return setattr(self.orig, name, value)
|
|
|
|
sys.modules[modname] = _module_wrapper_t(sys.modules[modname])
|
|
|
|
__set_module_dynattrs(
|
|
__name__,
|
|
{
|
|
"MAXADDR" : (lambda: inf_get_privrange_start_ea(), None),
|
|
})
|
|
|
|
#</pycode(py_ida)>
|