mirror of
https://github.com/cemu-project/idapython.git
synced 2024-11-28 03:54:18 +01:00
c88a77e809
- added support for 'long' addresses - updated copyright notice in hexrays.i - added methods to remove or insert elements in the qlist<cinsn_t> list - added support for ctree_visitor_t / ctree_parentee_t / cfunc_parentee_t / user_lvar_visitor_t - updated vds3 sample - added vds4 and vds7 python samples (ported from their C++ counter parts)
124 lines
4.1 KiB
Python
124 lines
4.1 KiB
Python
""" Print user-defined details to the output window.
|
|
|
|
Author: EiNSTeiN_ <einstein@g3nius.org>
|
|
|
|
This is a rewrite in Python of the vds4 example that comes with hexrays sdk.
|
|
"""
|
|
|
|
import idautils
|
|
import idaapi
|
|
import idc
|
|
|
|
import traceback
|
|
|
|
def run():
|
|
|
|
cfunc = idaapi.decompile(idaapi.get_screen_ea())
|
|
if not cfunc:
|
|
print 'Please move the cursor into a function.'
|
|
return
|
|
|
|
entry_ea = cfunc.entry_ea
|
|
print "Dump of user-defined information for function at %x" % (entry_ea, )
|
|
|
|
# Display user defined labels.
|
|
labels = idaapi.restore_user_labels(entry_ea);
|
|
if labels is not None:
|
|
print "------- %u user defined labels" % (len(labels), )
|
|
for org_label, name in labels.iteritems():
|
|
print "Label %d: %s" % (org_label, str(name))
|
|
idaapi.user_labels_free(labels)
|
|
|
|
# Display user defined comments
|
|
cmts = idaapi.restore_user_cmts(entry_ea);
|
|
if cmts is not None:
|
|
print "------- %u user defined comments" % (len(cmts), )
|
|
for tl, cmt in cmts.iteritems():
|
|
print "Comment at %x, preciser %x:\n%s\n" % (tl.ea, tl.itp, str(cmt))
|
|
idaapi.user_cmts_free(cmts)
|
|
|
|
# Display user defined citem iflags
|
|
iflags = idaapi.restore_user_iflags(entry_ea)
|
|
if iflags is not None:
|
|
print "------- %u user defined citem iflags" % (len(iflags), )
|
|
for cl, t in iflags.iteritems():
|
|
print "%a(%d): %08X%s" % (cl.ea, cl.op, f, " CIT_COLLAPSED" if f & CIT_COLLAPSED else "")
|
|
idaapi.user_iflags_free(iflags)
|
|
|
|
# Display user defined number formats
|
|
numforms = idaapi.restore_user_numforms(entry_ea)
|
|
if numforms is not None:
|
|
print "------- %u user defined number formats" % (len(numforms), )
|
|
for ol, nf in numforms.iteritems():
|
|
|
|
print "Number format at %a, operand %d: %s" % (ol.ea, ol.opnum, "negated " if (nf.props & NF_NEGATE) != 0 else "")
|
|
|
|
if nf.isEnum():
|
|
print "enum %s (serial %d)" % (str(nf.type_name), nf.serial)
|
|
|
|
elif nf.isChar():
|
|
print "char"
|
|
|
|
elif nf.isStroff():
|
|
print "struct offset %s" % (str(nf.type_name), )
|
|
|
|
else:
|
|
print "number base=%d" % (idaapi.getRadix(nf.flags, ol.opnum), )
|
|
|
|
idaapi.user_numforms_free(numforms)
|
|
|
|
# Display user-defined local variable information
|
|
# First defined the visitor class
|
|
class dump_lvar_info_t(idaapi.user_lvar_visitor_t):
|
|
|
|
def __init__(self):
|
|
idaapi.user_lvar_visitor_t.__init__(self)
|
|
self.displayed_header = False
|
|
return
|
|
|
|
def get_info_qty_for_saving(self):
|
|
return 0
|
|
|
|
def get_info_for_saving(self, lv):
|
|
return False
|
|
|
|
def handle_retrieved_info(self, lv):
|
|
|
|
try:
|
|
if not self.displayed_header:
|
|
self.displayed_header = True;
|
|
print "------- User defined local variable information"
|
|
|
|
print "Lvar defined at %x" % (lv.ll.defea, )
|
|
|
|
if len(str(lv.name)):
|
|
print " Name: %s" % (str(lv.name), )
|
|
|
|
if len(str(lv.type)):
|
|
#~ print_type_to_one_line(buf, sizeof(buf), idati, .c_str());
|
|
print " Type: %s" % (str(lv.type), )
|
|
|
|
if len(str(lv.cmt)):
|
|
print " Comment: %s" % (str(lv.cmt), )
|
|
except:
|
|
traceback.print_exc()
|
|
return 0
|
|
|
|
def handle_retrieved_mapping(self, lm):
|
|
return 0
|
|
|
|
def get_info_mapping_for_saving(self):
|
|
return None
|
|
|
|
# Now iterate over all user definitions
|
|
dli = dump_lvar_info_t();
|
|
idaapi.restore_user_lvar_settings(entry_ea, dli)
|
|
|
|
return
|
|
|
|
|
|
if idaapi.init_hexrays_plugin():
|
|
run()
|
|
else:
|
|
print 'dump user info: hexrays is not available.'
|