mirror of
https://github.com/idapython/src
synced 2026-06-08 14:47:00 +00:00
Arnaud Diederen
77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
from __future__ import print_function
|
|
# -----------------------------------------------------------------------
|
|
# This is an example illustrating how to use the execute_ui_requests()
|
|
# and the idautils.ProcessUiActions()
|
|
# (c) Hex-Rays
|
|
#
|
|
import idaapi
|
|
import idautils
|
|
import idc
|
|
|
|
# --------------------------------------------------------------------------
|
|
class __process_ui_actions_helper(object):
|
|
def __init__(self, actions, flags = 0):
|
|
"""Expect a list or a string with a list of actions"""
|
|
if isinstance(actions, str):
|
|
lst = actions.split(";")
|
|
elif isinstance(actions, (list, tuple)):
|
|
lst = actions
|
|
else:
|
|
raise ValueError("Must pass a string, list or a tuple")
|
|
|
|
# Remember the action list and the flags
|
|
self.__action_list = lst
|
|
self.__flags = flags
|
|
|
|
# Reset action index
|
|
self.__idx = 0
|
|
|
|
def __len__(self):
|
|
return len(self.__action_list)
|
|
|
|
def __call__(self):
|
|
if self.__idx >= len(self.__action_list):
|
|
return False
|
|
|
|
# Execute one action
|
|
idaapi.process_ui_action(
|
|
self.__action_list[self.__idx],
|
|
self.__flags)
|
|
|
|
# Move to next action
|
|
self.__idx += 1
|
|
print("index=%d" % self.__idx)
|
|
|
|
# Reschedule
|
|
return True
|
|
|
|
# --------------------------------------------------------------------------
|
|
def ProcessUiActions(actions, flags=0):
|
|
"""
|
|
@param actions: A string containing a list of actions separated by semicolon, a list or a tuple
|
|
@param flags: flags to be passed to process_ui_action()
|
|
@return: Boolean. Returns False if the action list was empty or execute_ui_requests() failed.
|
|
"""
|
|
|
|
# Instantiate a helper
|
|
helper = __process_ui_actions_helper(actions, flags)
|
|
return False if len(helper) < 1 else idaapi.execute_ui_requests((helper,))
|
|
|
|
|
|
# --------------------------------------------------------------------------
|
|
class print_req_t(object):
|
|
def __init__(self, s):
|
|
self.s = s
|
|
def __call__(self):
|
|
idaapi.msg("%s" % self.s)
|
|
return False # Don't reschedule
|
|
|
|
|
|
|
|
if idc.ask_yn(1,("HIDECANCEL\nDo you want to run execute_ui_requests() example?\n"
|
|
"Press NO to execute ProcessUiActions() example\n")):
|
|
idaapi.execute_ui_requests(
|
|
(print_req_t("Hello"), print_req_t(" world\n")) )
|
|
else:
|
|
ProcessUiActions("JumpQ;JumpName")
|