def get_sptr_by_member_id(mid):
if mid is None or mid == BADADDR:
return None
mptr = get_mptr_by_member_id(mid)
if not mptr:
return None
return ida_struct.get_member_struc(ida_struct.get_member_fullname(mptr.id))
def get_member_by_id(mid):
# Replacement for the ida_struct.get_member_by_id(),
# which in IDA7.0 returns incorrect sptr
res = ida_struct.get_member_by_id(mid)
if not res:
return None
mptr, member_fullname, _ = res
sptr = ida_struct.get_member_struc(ida_struct.get_member_fullname(mptr.id))
return mptr, member_fullname, sptr
def get_variable_name(self, ea_or_stack_tuple):
"""
Returns the name of the variable for the given ea or stack tuple.
:param ea_or_stack_tuple: ea address or tuple containing: (frame_id, stack_offset)
:return: string of name or None
"""
if isinstance(ea_or_stack_tuple, tuple):
frame_id, stack_offset = ea_or_stack_tuple
member_id = idc.get_member_id(frame_id, stack_offset)
return ida_struct.get_member_fullname(member_id)
else:
ea = ea_or_stack_tuple
name = idc.get_name(ea)
if name:
return name
_, original_location = self.get_original_location(ea)
if original_location:
return self.get_variable_name(original_location)
def get_full_name(self):
return ida_struct.get_member_fullname(self.mem.id)
def expand_struct(struct_id, new_size):
struct = ida_struct.get_struc(struct_id)
if struct is None:
log.warning("Struct id 0x%X wasn't found", struct_id)
return
log.debug(
"Expanding struc %s, size: 0x%X -> 0x%X",
ida_struct.get_struc_name(struct_id),
ida_struct.get_struc_size(struct_id),
new_size,
)
if ida_struct.get_struc_size(struct_id) > new_size - WORD_LEN:
return
fix_list = []
xrefs = idautils.XrefsTo(struct.id)
for xref in xrefs:
if xref.type == ida_xref.dr_R and xref.user == 0 and xref.iscode == 0:
res = ida_struct.get_member_by_id(xref.frm)
if not res or not res[0]:
log.warning("Xref from %08X wasn't struct_member", xref.frm)
continue
member = res[0]
x_struct = ida_struct.get_member_struc(ida_struct.get_member_fullname(member.id))
assert x_struct
old_name = ida_struct.get_member_name(member.id)
offset = member.soff
# FIXME: why use random here?
marker_name = "marker_%d" % random.randint(0, 0xFFFFFF)
# FIXME: check if add_struc_member actually added a member
idc.add_struc_member(
x_struct.id,
marker_name,
member.soff + new_size,
idaapi.FF_DATA | idaapi.FF_BYTE,
-1,
0,
)
log.debug(
"Delete member (0x%X-0x%X)",
member.soff,
member.soff + new_size - 1,
)
# FIXME: check if struc member actually deleted
ida_struct.del_struc_members(x_struct, member.soff, member.soff + new_size - 1)
fix_list.append(
[
x_struct.id,
old_name,
offset,
idaapi.FF_STRUCT | idaapi.FF_DATA,
struct_id,
new_size,
]
)
ret = add_to_struct(ida_struct.get_struc(struct_id), None, None, new_size - WORD_LEN)
log.debug("Now fix args:")
for fix_args in fix_list:
ret = idc.add_struc_member(*fix_args)
log.debug("%s = %d", fix_args, ret)
x_struct_id = fix_args[0]
idc.del_struc_member(x_struct_id, ida_struct.get_struc_size(x_struct_id))