mirror of
https://github.com/cemu-project/idapython.git
synced 2024-11-24 10:09:20 +01:00
84 lines
2.2 KiB
Python
84 lines
2.2 KiB
Python
import idaapi
|
|
from idaapi import Choose2
|
|
|
|
class MyChoose2(Choose2):
|
|
|
|
def __init__(self, title, nb = 5):
|
|
Choose2.__init__(self, title, [ ["Address", 10], ["Name", 30] ])
|
|
self.n = 0
|
|
self.items = [ self.make_item() for x in xrange(0, nb+1) ]
|
|
self.icon = 5
|
|
self.selcount = 0
|
|
self.popup_names = ["Inzert", "Del leet", "Ehdeet", "Ree frech"]
|
|
print "created", str(self)
|
|
|
|
def OnClose(self):
|
|
print "closed", str(self)
|
|
|
|
def OnEditLine(self, n):
|
|
self.items[n][1] = self.items[n][1] + "*"
|
|
print "editing", str(n)
|
|
|
|
def OnInsertLine(self):
|
|
self.items.append(self.make_item())
|
|
print "insert line"
|
|
|
|
def OnSelectLine(self, n):
|
|
self.selcount += 1
|
|
Warning("[%02d] selectline '%s'" % (self.selcount, n))
|
|
|
|
def OnGetLine(self, n):
|
|
print "getline", str(n)
|
|
return self.items[n]
|
|
|
|
def OnGetSize(self):
|
|
print "getsize"
|
|
return len(self.items)
|
|
|
|
def OnDeleteLine(self, n):
|
|
print "del ",str(n)
|
|
del self.items[n]
|
|
return n
|
|
|
|
def OnRefresh(self, n):
|
|
print "refresh", n
|
|
return n
|
|
|
|
def OnCommand(self, n, cmd_id):
|
|
if cmd_id == self.cmd_a:
|
|
print "command A selected @", n
|
|
elif cmd_id == self.cmd_b:
|
|
print "command B selected @", n
|
|
else:
|
|
print "Unknown command:", cmd_id, "@", n
|
|
return 1
|
|
|
|
def OnGetIcon(self, n):
|
|
r = self.items[n]
|
|
t = self.icon + r[1].count("*")
|
|
print "geticon", n, t
|
|
return t
|
|
|
|
def show(self):
|
|
t = self.Show()
|
|
if t < 0:
|
|
return False
|
|
self.cmd_a = self.AddCommand("command A")
|
|
self.cmd_b = self.AddCommand("command B")
|
|
return True
|
|
|
|
def make_item(self):
|
|
r = [str(self.n), "func_%04d" % self.n]
|
|
self.n += 1
|
|
return r
|
|
|
|
def OnGetLineAttr(self, n):
|
|
print "getlineattr", n
|
|
if n == 1:
|
|
return [0xFF0000, 0]
|
|
|
|
for i in xrange(1, 5+1):
|
|
c = MyChoose2("choose2 - sample %d" % i, i*2)
|
|
r = c.show()
|
|
print r
|
|
|