def __call__(self):
from .core import Core
dll = Core.get_ida_dll()
get_idati = dll.get_idati
get_idati.argtypes = []
get_idati.restype = ctypes.c_void_p
set_numbered_type = dll.set_numbered_type
set_numbered_type.argtypes = [
ctypes.c_void_p,
ctypes.c_uint32,
ctypes.c_int,
ctypes.c_char_p,
ctypes.c_char_p,
ctypes.c_char_p,
ctypes.c_char_p,
ctypes.c_char_p,
ctypes.c_int,
]
set_numbered_type.restype = ctypes.c_int
py_ti = ida_typeinf.get_idati()
ordinal_qty = ida_typeinf.get_ordinal_qty(py_ti) - 1
last_ordinal = self.local_types[-1][0]
if ordinal_qty < last_ordinal:
ida_typeinf.alloc_type_ordinals(py_ti, last_ordinal - ordinal_qty)
else:
for py_ordinal in range(last_ordinal + 1, ordinal_qty + 1):
ida_typeinf.del_numbered_type(py_ti, py_ordinal)
local_types = self.local_types
for py_ord, name, type, fields, cmt, fieldcmts, sclass in local_types:
if type:
ti = get_idati()
ordinal = ctypes.c_uint32(py_ord)
ntf_flags = ctypes.c_int(ida_typeinf.NTF_REPLACE)
name = ctypes.c_char_p(name)
type = ctypes.c_char_p(type)
fields = ctypes.c_char_p(fields)
cmt = ctypes.c_char_p(cmt)
fieldcmts = ctypes.c_char_p(fieldcmts)
sclass = ctypes.c_int(sclass)
set_numbered_type(
ti,
ordinal,
ntf_flags,
name,
type,
fields,
cmt,
fieldcmts,
sclass,
)
else:
ida_typeinf.del_numbered_type(py_ti, py_ord)
ida_kernwin.request_refresh(ida_kernwin.IWID_LOCTYPS)
def __call__(self):
ti = ida_typeinf.get_idati()
ordinal_qty = ida_typeinf.get_ordinal_qty(ti)
last_ordinal = self.local_types[-1][0]
if ordinal_qty < last_ordinal:
ida_typeinf.alloc_type_ordinals(ti, last_ordinal - ordinal_qty)
else:
for ordinal in range(last_ordinal + 1, ordinal_qty + 1):
ida_typeinf.del_numbered_type(ti, ordinal)
for ordinal, name, ret in self.local_types:
name = Event.encode_bytes(name)
type, fields, fieldcmts = ret
type = Event.encode_bytes(type)
fields = Event.encode_bytes(fields)
fieldcmts = Event.encode_bytes(fieldcmts)
type_info = ida_typeinf.tinfo_t()
type_info.deserialize(ti, type, fields, fieldcmts)
type_info.set_numbered_type(ti, ordinal, 0, name)
ida_kernwin.request_refresh(ida_kernwin.IWID_LOCTYPS)
def __call__(self):
local_type = []
for t in self.local_type:
if t is not None:
local_type.append((
Event.encode_bytes(t[0]),
Event.encode_bytes(t[1]),
Event.encode_bytes(t[2]),
))
else:
local_type.append(None)
missing_ord = len(local_type) - ida_typeinf.get_ordinal_qty(None) + 1
if missing_ord > 0:
ida_typeinf.alloc_type_ordinals(None, missing_ord)
for i, t in enumerate(local_type):
if t is not None:
cur_tinfo = ida_typeinf.tinfo_t()
cur_tinfo.deserialize(None, t[0], t[1])
cur_tinfo.set_numbered_type(None, i + 1, 0, t[2])
else:
ida_typeinf.del_numbered_type(None, i + 1)
ida_kernwin.request_refresh(ida_kernwin.IWID_LOCTYPS)
def InsertType(type_obj, fReplace=False):
# print("Insert type %s." % type_obj.name)
wrapperTypeString = b'\x0d\x01\x01'
if getTypeOrdinal(type_obj.name) != 0:
idx = getTypeOrdinal(type_obj.name)
t = ImportLocalType(idx)
if (t.TypeFields is None or t.TypeFields == "") and t.is_sue():
fReplace = True
if t.isEqual(type_obj) or type_obj.TypeString == wrapperTypeString:
return 1
if not fReplace:
type_obj = DuplicateResolver(t, type_obj, False)
else:
idx = ida_typeinf.alloc_type_ordinals(idaapi.get_idati(), 1)
tif = ida_typeinf.tinfo_t()
ret = tif.deserialize(ida_typeinf.get_idati(), type_obj.GetTypeString(),
type_obj.TypeFields, type_obj.fieldcmts)
if not ret:
idaapi.warning(
"Error on tinfo deserilization, type name = %s, ret = %d" %
(type_obj.name, ret))
ret = -1
else:
ret = tif.set_numbered_type(idaapi.get_idati(), idx, 0x4,
type_obj.name)
del tif
# ret = idaapi.set_numbered_type(
# my_ti,
# idx,
# 0x4,
# type_obj.name,
# type_obj.GetTypeString(),
# type_obj.TypeFields,
# type_obj.cmt,
# type_obj.fieldcmts
# )
# print "Insert type %s. ret = %d"%(type_obj.name,ret)
if (ida_pro.IDA_SDK_VERSION < 700
and ret != 1) or (ida_pro.IDA_SDK_VERSION >= 700 and ret != 0):
print("bad insert: %s; ret = %d" % (type_obj.name, ret))
return ret
def alloc_oridinal(target_ordinal):
# Get_ordinal_qty() will return (current max ordinal + 1)
missing_ord = (target_ordinal - ida_typeinf.get_ordinal_qty(None) +
1)
if missing_ord > 0:
ida_typeinf.alloc_type_ordinals(None, missing_ord)
