Nixyri/ghidra
1
0
Fork 0
mirror of https://github.com/NationalSecurityAgency/ghidra.git synced 2024-11-22 04:05:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

GP-2567 IDA Python 3 plugins

Browse Source 
Updated IDA Pro plugins to be compatable with both python 2 and 3.
Installation instructions are the same, however idaxml can alternatively
be placed in the IDA python, IDA python/3, or IDA python/2 directories.
This commit is contained in:
ghidraGander 2022-10-19 20:35:02 -04:00
parent ff503596ac
commit e6a86e91ca
4 changed files with 70 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
GhidraBuild/IDAPro/Python/7xx/loaders/xml_loader.py
View File

@ -22,6 +22,7 @@ This file must be placed in the IDA loaders directory.
The file idaxml.py must be placed in the IDA python directory. The file idaxml.py must be placed in the IDA python directory.
""" """
from __future__ import print_function
import ida_idaapi import ida_idaapi
import ida_idp import ida_idp
import ida_kernwin import ida_kernwin
@ -30,6 +31,10 @@ import idaxml
import idc import idc
import sys import sys
if sys.version_info.major >= 3:
from idaxml import _exc_info
sys.exc_value = lambda: _exc_info()[1]
sys.exc_type = lambda: _exc_info()[0]
""" """
Loader functions Loader functions
@ -52,13 +57,13 @@ def accept_file(li, filename):
# read 16K bytes to allow for the DTD # read 16K bytes to allow for the DTD
data = li.read(0x4000) data = li.read(0x4000)
# look for start of <PROGRAM> element # look for start of <PROGRAM> element
start = data.find("<PROGRAM") start = data.find(b"<PROGRAM")
if start >= 0: if start >= 0:
s = data.find("<PROCESSOR ") s = data.find(b"<PROCESSOR ")
p = data[s+11:] p = data[s+11:]
e = p.find("/>") e = p.find(b"/>")
proc = p[:e] proc = p[:e]
ida_kernwin.info("Processor specified in the XML file is:\n" + proc + ida_kernwin.info("Processor specified in the XML file is:\n" + proc.decode() +
"\n\nYou must select and set the compatible " + "\n\nYou must select and set the compatible " +
"IDA processor type.") "IDA processor type.")
return { 'format': "XML PROGRAM file", 'options': 0x8001 } return { 'format': "XML PROGRAM file", 'options': 0x8001 }
@ -83,19 +88,19 @@ def load_file(li, neflags, format):
status = xml.import_xml() status = xml.import_xml()
except idaxml.Cancelled: except idaxml.Cancelled:
msg = "XML PROGRAM import cancelled!" msg = "XML PROGRAM import cancelled!"
print "\n" + msg print("\n" + msg)
idc.warning(msg) idc.warning(msg)
except idaxml.MultipleAddressSpacesNotSupported: except idaxml.MultipleAddressSpacesNotSupported:
msg = "XML Import cancelled!" msg = "XML Import cancelled!"
msg += "\n\nXML Import does not currently support" msg += "\n\nXML Import does not currently support"
msg += "\nimporting multiple address spaces." msg += "\nimporting multiple address spaces."
print "\n" + msg print("\n" + msg)
idc.warning(msg) idc.warning(msg)
except: except:
print "\nHouston, we have a problem!" print("\nHouston, we have a problem!")
msg = "***** Exception occurred: XML loader failed! *****" msg = "***** Exception occurred: XML loader failed! *****"
print "\n" + msg + "\n", sys.exc_type, sys.exc_value print("\n" + msg + "\n", sys.exc_type, sys.exc_value)
print event, element.tag, element.attrib print(event, element.tag, element.attrib)
idc.warning(msg) idc.warning(msg)
finally: finally:
idc.set_ida_state(st) idc.set_ida_state(st)

9
GhidraBuild/IDAPro/Python/7xx/plugins/xml_exporter.py
View File

@ -22,6 +22,7 @@ This file must be placed in the IDA plugins directory.
The file idaxml.py must be placed in the IDA python directory. The file idaxml.py must be placed in the IDA python directory.
""" """
from __future__ import print_function
import ida_auto import ida_auto
import ida_idaapi import ida_idaapi
import ida_kernwin import ida_kernwin
@ -29,6 +30,10 @@ import idaxml
import idc import idc
import sys import sys
if sys.version_info.major >= 3:
from idaxml import _exc_info
sys.exc_value = lambda: _exc_info()[1]
sys.exc_type = lambda: _exc_info()[0]
class XmlExporterPlugin(ida_idaapi.plugin_t): class XmlExporterPlugin(ida_idaapi.plugin_t):
""" """
@ -72,12 +77,12 @@ class XmlExporterPlugin(ida_idaapi.plugin_t):
except idaxml.Cancelled: except idaxml.Cancelled:
ida_kernwin.hide_wait_box() ida_kernwin.hide_wait_box()
msg = "XML Export cancelled!" msg = "XML Export cancelled!"
print "\n" + msg print("\n" + msg)
idc.warning(msg) idc.warning(msg)
except: except:
ida_kernwin.hide_wait_box() ida_kernwin.hide_wait_box()
msg = "***** Exception occurred: XML Exporter failed! *****" msg = "***** Exception occurred: XML Exporter failed! *****"
print "\n" + msg + "\n", sys.exc_type, sys.exc_value print("\n" + msg + "\n", sys.exc_type, sys.exc_value)
idc.warning(msg) idc.warning(msg)
finally: finally:
xml.cleanup() xml.cleanup()

12
GhidraBuild/IDAPro/Python/7xx/plugins/xml_importer.py
View File

@ -22,12 +22,18 @@ This file must be placed in the IDA plugins directory.
The file idaxml.py must be placed in the IDA python directory. The file idaxml.py must be placed in the IDA python directory.
""" """
from __future__ import print_function
import ida_idaapi import ida_idaapi
import ida_pro import ida_pro
import idaxml import idaxml
import idc import idc
import sys import sys
if sys.version_info.major >= 3:
from idaxml import _exc_info
sys.exc_value = lambda: _exc_info()[1]
sys.exc_type = lambda: _exc_info()[0]
class XmlImporterPlugin(ida_idaapi.plugin_t): class XmlImporterPlugin(ida_idaapi.plugin_t):
""" """
XML Importer plugin class XML Importer plugin class
@ -68,17 +74,17 @@ class XmlImporterPlugin(ida_idaapi.plugin_t):
xml.import_xml() xml.import_xml()
except idaxml.Cancelled: except idaxml.Cancelled:
msg = "XML Import cancelled!" msg = "XML Import cancelled!"
print "\n" + msg print("\n" + msg)
idc.warning(msg) idc.warning(msg)
except idaxml.MultipleAddressSpacesNotSupported: except idaxml.MultipleAddressSpacesNotSupported:
msg = "XML Import cancelled!" msg = "XML Import cancelled!"
msg += "\n\nXML Import does not currently support" msg += "\n\nXML Import does not currently support"
msg += "\nimporting multiple address spaces." msg += "\nimporting multiple address spaces."
print "\n" + msg print("\n" + msg)
idc.warning(msg) idc.warning(msg)
except: except:
msg = "***** Exception occurred: XML Importer failed! *****" msg = "***** Exception occurred: XML Importer failed! *****"
print "\n" + msg + "\n", sys.exc_type, sys.exc_value print("\n" + msg + "\n", sys.exc_type, sys.exc_value)
idc.warning(msg) idc.warning(msg)
finally: finally:
xml.cleanup() xml.cleanup()

56
GhidraBuild/IDAPro/Python/7xx/python/idaxml.py
View File

@ -19,6 +19,7 @@
""" """
""" """
from __future__ import print_function
import ida_auto import ida_auto
import ida_bytes import ida_bytes
import ida_diskio import ida_diskio
@ -52,10 +53,16 @@ import sys
import time import time
from xml.etree import cElementTree from xml.etree import cElementTree
if sys.version_info.major >= 3:
import copy
time.clock = time.perf_counter
_exc_info = copy.copy(sys.exc_info)
sys.exc_value = lambda: _exc_info()[1]
sys.exc_type = lambda: _exc_info()[0]
DEBUG = False # print debug statements DEBUG = False # print debug statements
IDAXML_VERSION = "5.0.1" IDAXML_VERSION = "5.0.2"
BASELINE_IDA_VERSION = 700 BASELINE_IDA_VERSION = 700
BASELINE_STR = '7.00' BASELINE_STR = '7.00'
IDA_SDK_VERSION = ida_pro.IDA_SDK_VERSION IDA_SDK_VERSION = ida_pro.IDA_SDK_VERSION
@ -166,7 +173,19 @@ class IdaXml:
Args: Args:
what: String indicating Exporter, Importer, or Loader what: String indicating Exporter, Importer, or Loader
""" """
f = ida_diskio.idadir('python') + '/idaxml.py' if os.path.isfile(os.path.join(ida_diskio.idadir('python'), 'idaxml.py')):
f = os.path.join(ida_diskio.idadir('python'), 'idaxml.py')
elif os.path.isfile(os.path.join(ida_diskio.idadir('python'), '3', 'idaxml.py')):
f = os.path.join(ida_diskio.idadir('python'), '3', 'idaxml.py')
elif os.path.isfile(os.path.join(ida_diskio.idadir('python'), '2', 'idaxml.py')):
f = os.path.join(ida_diskio.idadir('python'), '2', 'idaxml.py')
elif os.path.isfile(os.path.join(ida_diskio.get_user_idadir(), 'python', 'idaxml.py')):
f = os.path.join(ida_diskio.get_user_idadir(), 'python', 'idaxml.py')
else:
msg = "Error opening file " + os.path.join(ida_diskio.idadir('python'), 'idaxml.py') + " !\n"
idc.msg(msg)
raise FileError
ftime = time.localtime(os.path.getmtime(f)) ftime = time.localtime(os.path.getmtime(f))
ts = time.strftime('%b %d %Y %H:%M:%S', ftime) ts = time.strftime('%b %d %Y %H:%M:%S', ftime)
version = "\nXML " + what + " v" + IDAXML_VERSION version = "\nXML " + what + " v" + IDAXML_VERSION
@ -254,7 +273,7 @@ class XmlExporter(IdaXml):
self.inf = ida_idaapi.get_inf_structure() self.inf = ida_idaapi.get_inf_structure()
self.min_ea = self.inf.min_ea self.min_ea = self.inf.min_ea
self.max_ea = self.inf.max_ea self.max_ea = self.inf.max_ea
self.cbsize = (ida_idp.ph_get_cnbits()+7)/8 self.cbsize = (ida_idp.ph_get_cnbits()+7)//8
self.processor = str.upper(ida_idp.get_idp_name()) self.processor = str.upper(ida_idp.get_idp_name())
self.batch = ida_kernwin.cvar.batch self.batch = ida_kernwin.cvar.batch
@ -373,6 +392,8 @@ class XmlExporter(IdaXml):
String containing either the character or the entity String containing either the character or the entity
substition string. substition string.
""" """
if isinstance(ch, int):
ch = chr(ch)
if ((ord(ch) < 0x20) and (ord(ch) != 0x09 and if ((ord(ch) < 0x20) and (ord(ch) != 0x09 and
ord(ch) != 0x0A and ord(ch) != 0x0D)): return '' ord(ch) != 0x0A and ord(ch) != 0x0D)): return ''
elif ch == '&' : return '&amp;' elif ch == '&' : return '&amp;'
@ -647,7 +668,10 @@ class XmlExporter(IdaXml):
# tag_remove seems to be losing last character # tag_remove seems to be losing last character
# work around is to add a space # work around is to add a space
cmt_text = ida_lines.tag_remove(cmt + ' ') cmt_text = ida_lines.tag_remove(cmt + ' ')
self.write_text(cmt_text.decode('utf-8')) if sys.version_info.major >= 3:
self.write_text(cmt_text)
else:
self.write_text(cmt_text.decode('utf-8'))
self.end_element(COMMENT, False) self.end_element(COMMENT, False)
@ -714,7 +738,7 @@ class XmlExporter(IdaXml):
if msize == 0: if msize == 0:
msize = 1 msize = 1
if idc.is_strlit(f) == False and size != msize: if idc.is_strlit(f) == False and size != msize:
dtype = "%s[%d]" % (dtype, size/msize) dtype = "%s[%d]" % (dtype, size//msize)
self.start_element(DEFINED_DATA) self.start_element(DEFINED_DATA)
self.write_address_attribute(ADDRESS, addr) self.write_address_attribute(ADDRESS, addr)
self.write_attribute(DATATYPE, dtype) self.write_attribute(DATATYPE, dtype)
@ -1108,7 +1132,7 @@ class XmlExporter(IdaXml):
if size < msize: size = msize if size < msize: size = msize
if (size != msize): if (size != msize):
arraytype = self.get_member_type(m) arraytype = self.get_member_type(m)
dtype = "%s[%d]" % (arraytype, size/msize) dtype = "%s[%d]" % (arraytype, size//msize)
self.write_attribute(DATATYPE, dtype) self.write_attribute(DATATYPE, dtype)
self.write_numeric_attribute(SIZE, size*self.cbsize) self.write_numeric_attribute(SIZE, size*self.cbsize)
regcmt = ida_struct.get_member_cmt(m.id, False) regcmt = ida_struct.get_member_cmt(m.id, False)
@ -1544,7 +1568,7 @@ class XmlExporter(IdaXml):
if size < msize: size = msize if size < msize: size = msize
if (idc.is_strlit(f) == False and ida_bytes.is_align(f) == False if (idc.is_strlit(f) == False and ida_bytes.is_align(f) == False
and size != msize): and size != msize):
mtype = "%s[%d]" % (mtype, size/msize) mtype = "%s[%d]" % (mtype, size//msize)
self.write_attribute(DATATYPE, mtype) self.write_attribute(DATATYPE, mtype)
self.write_numeric_attribute(SIZE, size*self.cbsize) self.write_numeric_attribute(SIZE, size*self.cbsize)
regcmt = ida_struct.get_member_cmt(member.id, False) regcmt = ida_struct.get_member_cmt(member.id, False)
@ -1781,7 +1805,7 @@ class XmlExporter(IdaXml):
return idc.get_struc_name(opnd.tid) return idc.get_struc_name(opnd.tid)
if idc.is_strlit(f) == True: if idc.is_strlit(f) == True:
str_type = idc.get_str_type(addr) str_type = idc.get_str_type(addr)
#print ida_bytes.print_strlit_type(str_type) #print(ida_bytes.print_strlit_type(str_type))
if str_type == ida_nalt.STRTYPE_TERMCHR: return "string" if str_type == ida_nalt.STRTYPE_TERMCHR: return "string"
if str_type == ida_nalt.STRTYPE_PASCAL: return "string1" if str_type == ida_nalt.STRTYPE_PASCAL: return "string1"
if str_type == ida_nalt.STRTYPE_LEN2: return "string2" if str_type == ida_nalt.STRTYPE_LEN2: return "string2"
@ -2451,7 +2475,7 @@ class XmlImporter(IdaXml):
Integer representing the number of 8-bit bytes in an Integer representing the number of 8-bit bytes in an
addressable codebyte. addressable codebyte.
""" """
return (ida_idp.ph_get_cnbits()+7)/8 return (ida_idp.ph_get_cnbits()+7)//8
def get_datatype_flags(self, datatype, size): def get_datatype_flags(self, datatype, size):
@ -2637,7 +2661,7 @@ class XmlImporter(IdaXml):
if idc.is_mapped(addr) == False: if idc.is_mapped(addr) == False:
msg = ("import_bookmark: address %X not enabled in database" msg = ("import_bookmark: address %X not enabled in database"
% addr) % addr)
print msg print(msg)
return return
self.update_counter(BOOKMARK) self.update_counter(BOOKMARK)
for slot in range(ida_moves.MAX_MARK_SLOT): for slot in range(ida_moves.MAX_MARK_SLOT):
@ -2647,7 +2671,7 @@ class XmlImporter(IdaXml):
break break
except: except:
msg = "** Exception occurred in import_bookmark **" msg = "** Exception occurred in import_bookmark **"
print "\n" + msg + "\n", sys.exc_type, sys.exc_value print("\n" + msg + "\n", sys.exc_type, sys.exc_value)
def import_cmts(self, element, sid, typ): def import_cmts(self, element, sid, typ):
@ -2777,7 +2801,7 @@ class XmlImporter(IdaXml):
""" """
self.update_counter(DESCRIPTION) self.update_counter(DESCRIPTION)
# TODO: import_description: decide what to do with DESCRIPTION # TODO: import_description: decide what to do with DESCRIPTION
# print description.text # print(description.text)
def import_enum(self, enum): def import_enum(self, enum):
@ -2946,7 +2970,7 @@ class XmlImporter(IdaXml):
if idc.is_mapped(entry_point) == False: if idc.is_mapped(entry_point) == False:
msg = ("import_function: address %X not enabled in database" msg = ("import_function: address %X not enabled in database"
% entry_point) % entry_point)
print msg print(msg)
return return
idc.add_func(entry_point, BADADDR) idc.add_func(entry_point, BADADDR)
self.update_counter(FUNCTION) self.update_counter(FUNCTION)
@ -2981,7 +3005,7 @@ class XmlImporter(IdaXml):
self.import_register_var(register_var, func) self.import_register_var(register_var, func)
except: except:
msg = "** Exception occurred in import_function **" msg = "** Exception occurred in import_function **"
print "\n" + msg + "\n", sys.exc_type, sys.exc_value print("\n" + msg + "\n", sys.exc_type, sys.exc_value)
def import_function_def(self, function_def): def import_function_def(self, function_def):
@ -3205,7 +3229,7 @@ class XmlImporter(IdaXml):
# TODO: import_memory_reference: store refs? maybe only user-defined? # TODO: import_memory_reference: store refs? maybe only user-defined?
''' '''
if user == 'y': if user == 'y':
#print "%08X %08X" % (addr, to_addr), op, primary #print("%08X %08X" % (addr, to_addr), op, primary)
pass pass
''' '''
@ -3232,7 +3256,7 @@ class XmlImporter(IdaXml):
seg_str = '' seg_str = ''
if '::' in addrstr: if '::' in addrstr:
# overlay - skip for now # overlay - skip for now
print ' ** Overlayed memory block %s skipped ** ' % name print(' ** Overlayed memory block %s skipped ** ' % name)
msg = 'Overlayed memory block %s skipped!' % name msg = 'Overlayed memory block %s skipped!' % name
msg += "\n\nXML Import does not currently support" msg += "\n\nXML Import does not currently support"
msg += "\noverlayed memory blocks." msg += "\noverlayed memory blocks."