mirror of
https://github.com/cemu-project/idapython.git
synced 2024-12-28 02:31:53 +01:00
1258fab948
- IDA Pro 6.2 support
- added set_idc_func_ex(): it is now possible to add new IDC functions using Python
- added visit_patched_bytes() (see ex_patch.py)
- added support for the multiline text input control in the Form class
- added support for the editable/readonly dropdown list control in the Form class
- added execute_sync() to register a function call into the UI message queue
- added execute_ui_requests() / check ex_uirequests.py
- added add_hotkey() / del_hotkey() to bind Python methods to hotkeys
- added register_timer()/unregister_timer(). Check ex_timer.py
- added the IDC (Arrays) netnode manipulation layer into idc.py
- added idautils.Structs() and StructMembers() generator functions
- removed the "Run Python Statement" menu item. IDA now has a unified dialog.
Use RunPlugin("python", 0) to invoke it manually.
- better error messages for script plugins, loaders and processor modules
- bugfix: Dbg_Hooks.dbg_run_to() was receiving wrong input
- bugfix: A few Enum related functions were not properly working in idc.py
- bugfix: GetIdaDirectory() and GetProcessName() were broken in idc.py
- bugfix: idaapi.get_item_head() / idc.ItemHead() were not working
367 lines
11 KiB
Python
367 lines
11 KiB
Python
# -----------------------------------------------------------------------
|
|
# This is an example illustrating how to use the Form class
|
|
# (c) Hex-Rays
|
|
#
|
|
from idaapi import Form
|
|
|
|
#<pycode(ex_askusingform)>
|
|
# --------------------------------------------------------------------------
|
|
class TestEmbeddedChooserClass(Choose2):
|
|
"""
|
|
A simple chooser to be used as an embedded chooser
|
|
"""
|
|
def __init__(self, title, nb = 5, flags=0):
|
|
Choose2.__init__(self,
|
|
title,
|
|
[ ["Address", 10], ["Name", 30] ],
|
|
embedded=True, width=30, height=20, flags=flags)
|
|
self.n = 0
|
|
self.items = [ self.make_item() for x in xrange(0, nb+1) ]
|
|
self.icon = 5
|
|
self.selcount = 0
|
|
|
|
def make_item(self):
|
|
r = [str(self.n), "func_%04d" % self.n]
|
|
self.n += 1
|
|
return r
|
|
|
|
def OnClose(self):
|
|
pass
|
|
|
|
def OnGetLine(self, n):
|
|
print("getline %d" % n)
|
|
return self.items[n]
|
|
|
|
def OnGetSize(self):
|
|
n = len(self.items)
|
|
print("getsize -> %d" % n)
|
|
return n
|
|
|
|
# --------------------------------------------------------------------------
|
|
class MyForm(Form):
|
|
def __init__(self):
|
|
self.invert = False
|
|
self.EChooser = TestEmbeddedChooserClass("E1", flags=Choose2.CH_MULTI)
|
|
Form.__init__(self, r"""STARTITEM {id:rNormal}
|
|
BUTTON YES* Yeah
|
|
BUTTON NO Nope
|
|
BUTTON CANCEL Nevermind
|
|
Form Test
|
|
|
|
{FormChangeCb}
|
|
This is a string: +{cStr1}+
|
|
This is an address: +{cAddr1}+
|
|
|
|
Escape\{control}
|
|
This is a string: '{cStr2}'
|
|
This is a number: {cVal1}
|
|
|
|
<#Hint1#Enter name:{iStr1}>
|
|
<#Hint2#Select color:{iColor1}>
|
|
Browse test
|
|
<#Select a file to open#Browse to open:{iFileOpen}>
|
|
<#Select a file to save#Browse to save:{iFileSave}>
|
|
<#Select dir#Browse for dir:{iDir}>
|
|
Type
|
|
<#Select type#Write a type:{iType}>
|
|
Numbers
|
|
<##Enter a selector value:{iSegment}>
|
|
<##Enter a raw hex:{iRawHex}>
|
|
<##Enter a character:{iChar}>
|
|
<##Enter an address:{iAddr}>
|
|
Button test
|
|
<##Button1:{iButton1}> <##Button2:{iButton2}>
|
|
|
|
Check boxes:
|
|
<Error output:{rError}>
|
|
<Normal output:{rNormal}>
|
|
<Warnings:{rWarnings}>{cGroup1}>
|
|
|
|
Radio boxes:
|
|
<Green:{rGreen}>
|
|
<Red:{rRed}>
|
|
<Blue:{rBlue}>{cGroup2}>
|
|
<Embedded chooser:{cEChooser}>
|
|
The end!
|
|
""", {
|
|
'cStr1': Form.StringLabel("Hello"),
|
|
'cStr2': Form.StringLabel("StringTest"),
|
|
'cAddr1': Form.NumericLabel(0x401000, Form.FT_ADDR),
|
|
'cVal1' : Form.NumericLabel(99, Form.FT_HEX),
|
|
'iStr1': Form.StringInput(),
|
|
'iColor1': Form.ColorInput(),
|
|
'iFileOpen': Form.FileInput(open=True),
|
|
'iFileSave': Form.FileInput(save=True),
|
|
'iDir': Form.DirInput(),
|
|
'iType': Form.StringInput(tp=Form.FT_TYPE),
|
|
'iSegment': Form.NumericInput(tp=Form.FT_SEG),
|
|
'iRawHex': Form.NumericInput(tp=Form.FT_RAWHEX),
|
|
'iAddr': Form.NumericInput(tp=Form.FT_ADDR),
|
|
'iChar': Form.NumericInput(tp=Form.FT_CHAR),
|
|
'iButton1': Form.ButtonInput(self.OnButton1),
|
|
'iButton2': Form.ButtonInput(self.OnButton2),
|
|
'cGroup1': Form.ChkGroupControl(("rNormal", "rError", "rWarnings")),
|
|
'cGroup2': Form.RadGroupControl(("rRed", "rGreen", "rBlue")),
|
|
'FormChangeCb': Form.FormChangeCb(self.OnFormChange),
|
|
'cEChooser' : Form.EmbeddedChooserControl(self.EChooser)
|
|
})
|
|
|
|
|
|
def OnButton1(self, code=0):
|
|
print("Button1 pressed")
|
|
|
|
|
|
def OnButton2(self, code=0):
|
|
print("Button2 pressed")
|
|
|
|
|
|
def OnFormChange(self, fid):
|
|
if fid == self.iButton1.id:
|
|
print("Button1 fchg;inv=%s" % self.invert)
|
|
self.SetFocusedField(self.rNormal)
|
|
self.EnableField(self.rError, self.invert)
|
|
self.invert = not self.invert
|
|
elif fid == self.iButton2.id:
|
|
g1 = self.GetControlValue(self.cGroup1)
|
|
g2 = self.GetControlValue(self.cGroup2)
|
|
d = self.GetControlValue(self.iDir)
|
|
f = self.GetControlValue(self.iFileOpen)
|
|
print("cGroup2:%x;Dir=%s;fopen=%s;cGroup1:%x" % (g1, d, f, g2))
|
|
elif fid == self.cEChooser.id:
|
|
l = self.GetControlValue(self.cEChooser)
|
|
print("Chooser: %s" % l)
|
|
else:
|
|
print(">>fid:%d" % fid)
|
|
return 1
|
|
|
|
|
|
|
|
# --------------------------------------------------------------------------
|
|
def stdalone_main():
|
|
f = MyForm()
|
|
f, args = f.Compile()
|
|
print args[0]
|
|
print args[1:]
|
|
f.rNormal.checked = True
|
|
f.rWarnings.checked = True
|
|
print hex(f.cGroup1.value)
|
|
|
|
f.rGreen.selected = True
|
|
print f.cGroup2.value
|
|
print "Title: '%s'" % f.title
|
|
|
|
f.Free()
|
|
|
|
# --------------------------------------------------------------------------
|
|
def ida_main():
|
|
# Create form
|
|
global f
|
|
f = MyForm()
|
|
|
|
# Compile (in order to populate the controls)
|
|
f.Compile()
|
|
|
|
f.iColor1.value = 0x5bffff
|
|
f.iDir.value = os.getcwd()
|
|
f.rNormal.checked = True
|
|
f.rWarnings.checked = True
|
|
f.rGreen.selected = True
|
|
f.iStr1.value = "Hello"
|
|
f.iFileSave.value = "*.*"
|
|
f.iFileOpen.value = "*.*"
|
|
# Execute the form
|
|
ok = f.Execute()
|
|
print("r=%d" % ok)
|
|
if ok == 1:
|
|
print("f.str1=%s" % f.iStr1.value)
|
|
print("f.color1=%x" % f.iColor1.value)
|
|
print("f.openfile=%s" % f.iFileOpen.value)
|
|
print("f.savefile=%s" % f.iFileSave.value)
|
|
print("f.dir=%s" % f.iDir.value)
|
|
print("f.type=%s" % f.iType.value)
|
|
print("f.seg=%s" % f.iSegment.value)
|
|
print("f.rawhex=%x" % f.iRawHex.value)
|
|
print("f.char=%x" % f.iChar.value)
|
|
print("f.addr=%x" % f.iAddr.value)
|
|
print("f.cGroup1=%x" % f.cGroup1.value)
|
|
print("f.cGroup2=%x" % f.cGroup2.value)
|
|
|
|
sel = f.EChooser.GetEmbSelection()
|
|
if sel is None:
|
|
print("No selection")
|
|
else:
|
|
print("Selection: %s" % sel)
|
|
|
|
# Dispose the form
|
|
f.Free()
|
|
|
|
# --------------------------------------------------------------------------
|
|
def ida_main_legacy():
|
|
# Here we simply show how to use the old style form format using Python
|
|
|
|
# Sample form from kernwin.hpp
|
|
s = """Sample dialog box
|
|
|
|
|
|
This is sample dialog box for %A
|
|
using address %$
|
|
|
|
<~E~nter value:N:32:16::>
|
|
"""
|
|
|
|
# Use either StringArgument or NumericArgument to pass values to the function
|
|
num = Form.NumericArgument('N', value=123)
|
|
ok = idaapi.AskUsingForm(s,
|
|
Form.StringArgument("PyAskUsingForm").arg,
|
|
Form.NumericArgument('$', 0x401000).arg,
|
|
num.arg)
|
|
if ok == 1:
|
|
print("You entered: %x" % num.value)
|
|
|
|
# --------------------------------------------------------------------------
|
|
def test_multilinetext_legacy():
|
|
# Here we text the multi line text control in legacy mode
|
|
|
|
# Sample form from kernwin.hpp
|
|
s = """Sample dialog box
|
|
|
|
This is sample dialog box
|
|
<Enter multi line text:t40:80:50::>
|
|
"""
|
|
# Use either StringArgument or NumericArgument to pass values to the function
|
|
ti = textctrl_info_t("Some initial value")
|
|
ok = idaapi.AskUsingForm(s, pointer(c_void_p.from_address(ti.clink_ptr)))
|
|
if ok == 1:
|
|
print("You entered: %s" % ti.text)
|
|
|
|
del ti
|
|
|
|
# --------------------------------------------------------------------------
|
|
class MyForm2(Form):
|
|
"""Simple Form to test multilinetext and combo box controls"""
|
|
def __init__(self):
|
|
Form.__init__(self, r"""STARTITEM 0
|
|
BUTTON YES* Yeah
|
|
BUTTON NO Nope
|
|
BUTTON CANCEL NONE
|
|
Form Test
|
|
|
|
{FormChangeCb}
|
|
<Multilinetext:{txtMultiLineText}>
|
|
""", {
|
|
'txtMultiLineText': Form.MultiLineTextControl(text="Hello"),
|
|
'FormChangeCb': Form.FormChangeCb(self.OnFormChange),
|
|
})
|
|
|
|
|
|
def OnFormChange(self, fid):
|
|
if fid == self.txtMultiLineText.id:
|
|
pass
|
|
elif fid == -2:
|
|
ti = self.GetControlValue(self.txtMultiLineText)
|
|
print "ti.text = %s" % ti.text
|
|
else:
|
|
print(">>fid:%d" % fid)
|
|
return 1
|
|
|
|
# --------------------------------------------------------------------------
|
|
def test_multilinetext(execute=True):
|
|
"""Test the multilinetext and combobox controls"""
|
|
f = MyForm2()
|
|
f, args = f.Compile()
|
|
if execute:
|
|
ok = f.Execute()
|
|
else:
|
|
print args[0]
|
|
print args[1:]
|
|
ok = 0
|
|
|
|
if ok == 1:
|
|
assert f.txtMultiLineText.text == f.txtMultiLineText.value
|
|
print f.txtMultiLineText.text
|
|
|
|
f.Free()
|
|
|
|
# --------------------------------------------------------------------------
|
|
class MyForm3(Form):
|
|
"""Simple Form to test multilinetext and combo box controls"""
|
|
def __init__(self):
|
|
self.__n = 0
|
|
Form.__init__(self,
|
|
r"""BUTTON YES* Yeah
|
|
BUTTON NO Nope
|
|
BUTTON CANCEL NONE
|
|
Dropdown list test
|
|
|
|
{FormChangeCb}
|
|
<Dropdown list (readonly):{cbReadonly}> <Add element:{iButtonAddelement}> <Set index:{iButtonSetIndex}>
|
|
<Dropdown list (editable):{cbEditable}> <Set string:{iButtonSetString}>
|
|
""", {
|
|
'FormChangeCb': Form.FormChangeCb(self.OnFormChange),
|
|
'cbReadonly': Form.DropdownListControl(
|
|
items=["red", "green", "blue"],
|
|
readonly=True,
|
|
selval=1),
|
|
'cbEditable': Form.DropdownListControl(
|
|
items=["1MB", "2MB", "3MB", "4MB"],
|
|
readonly=False,
|
|
selval="4MB"),
|
|
'iButtonAddelement': Form.ButtonInput(self.OnButtonNop),
|
|
'iButtonSetIndex': Form.ButtonInput(self.OnButtonNop),
|
|
'iButtonSetString': Form.ButtonInput(self.OnButtonNop),
|
|
})
|
|
|
|
|
|
def OnButtonNop(self, code=0):
|
|
"""Do nothing, we will handle events in the form callback"""
|
|
pass
|
|
|
|
def OnFormChange(self, fid):
|
|
if fid == self.iButtonSetString.id:
|
|
s = idc.AskStr("none", "Enter value")
|
|
if s:
|
|
self.SetControlValue(self.cbEditable, s)
|
|
elif fid == self.iButtonSetIndex.id:
|
|
s = idc.AskStr("1", "Enter index value:")
|
|
if s:
|
|
try:
|
|
i = int(s)
|
|
except:
|
|
i = 0
|
|
self.SetControlValue(self.cbReadonly, i)
|
|
elif fid == self.iButtonAddelement.id:
|
|
# add a value to the string list
|
|
self.__n += 1
|
|
self.cbReadonly.add("some text #%d" % self.__n)
|
|
# Refresh the control
|
|
self.RefreshField(self.cbReadonly)
|
|
elif fid == -2:
|
|
s = self.GetControlValue(self.cbEditable)
|
|
print "user entered: %s" % s
|
|
sel_idx = self.GetControlValue(self.cbReadonly)
|
|
|
|
return 1
|
|
|
|
# --------------------------------------------------------------------------
|
|
def test_dropdown(execute=True):
|
|
"""Test the combobox controls"""
|
|
f = MyForm3()
|
|
f, args = f.Compile()
|
|
if execute:
|
|
ok = f.Execute()
|
|
else:
|
|
print args[0]
|
|
print args[1:]
|
|
ok = 0
|
|
|
|
if ok == 1:
|
|
print "Editable: %s" % f.cbEditable.value
|
|
print "Readonly: %s" % f.cbReadonly.value
|
|
|
|
f.Free()
|
|
|
|
#</pycode(ex_askusingform)>
|
|
|
|
|
|
# --------------------------------------------------------------------------
|
|
ida_main() |