The-Homebrew-Cloud/cemu-idapython
2
0
Fork 0
mirror of https://github.com/cemu-project/idapython.git synced 2024-11-28 03:54:18 +01:00
Code Issues Packages Projects Releases Wiki Activity

idc.py: AddStrucMember() and SetMemberType() switched to native IDC implementation

Browse Source
This commit is contained in:
gergely.erdelyi 2009-07-17 17:58:57 +00:00
parent 09ead310b7
commit b0d8723ccb
1 changed files with 12 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
python/idc.py
View File

@ -5005,48 +5005,7 @@ def SetStrucComment(sid, comment, repeatable):
return idaapi.set_struc_cmt(sid, comment, repeatable) return idaapi.set_struc_cmt(sid, comment, repeatable)
def _IDC_PrepareStrucMemberTypeinfo(flag, typeid, target=None, tdelta=None, reftype=None): def AddStrucMember(sid, name, offset, flag, typeid, nbytes, target=-1, tdelta=0, reftype=REF_OFF32):
""" Internal function to prepare typeinfo_t for adding/setting structure members """
if idaapi.isOff0(flag):
ti = idaapi.typeinfo_t()
ri = idaapi.refinfo_t()
ri.base = typeid
if target != None:
ri.target = target
ri.tdelta = 0
ri.flags = reftype
else:
ri.target = BADADDR
ri.tdelta = 0
if (flag & FF_WORD):
ri.flags = REF_OFF16
elif (flag & FF_BYTE):
ri.flags = REF_OFF8
elif (flag & FF_QWRD):
ri.flags = REF_OFF64
else:
ri.flags = REF_OFF32
ti.ri = ri
elif idaapi.isEnum0(flag):
ti = idaapi.typeinfo_t()
ec = idaapi.enum_const_t()
ec.tid = typeid
ec.serial = 0
ti.ec = ec
elif idaapi.isStroff0(flag):
ti = idaapi.typeinfo_t()
ti.path.len = 1
target_struct = idaapi.get_struc(typeid)
assert target_struct, "Target structure is invalid"
ti.path.ids = [ typeid ]
else:
ti = idaapi.typeinfo_t()
ti.tid = typeid
return ti
def AddStrucMember(sid, name, offset, flag, typeid, nbytes, target=None, tdelta=None, reftype=None):
""" """
Add structure member Add structure member
@ -5075,10 +5034,11 @@ def AddStrucMember(sid, name, offset, flag, typeid, nbytes, target=None, tdelta=
@return: 0 - ok, otherwise error code (one of STRUC_ERROR_*) @return: 0 - ok, otherwise error code (one of STRUC_ERROR_*)
""" """
struc = idaapi.get_struc(sid) if isOff0(flag):
assert struct, "get_struc() failed" return Eval('AddStrucMember(%d, "%s", %d, %d, %d, %d, %d, %d, %d);' % (sid, name, offset, flag, typeid, nbytes,
ti = _IDC_PrepareStrucMemberTypeinfo(flag, typeid, target, tdelta, reftype) target, tdelta, reftype))
return idaapi.add_struc_member(struc, name, offset, flag, ti, nbytes) else:
return Eval('AddStrucMember(%d, "%s", %d, %d, %d, %d);' % (sid, name, offset, flag, typeid, nbytes))
STRUC_ERROR_MEMBER_NAME = -1 # already has member with this name (bad name) STRUC_ERROR_MEMBER_NAME = -1 # already has member with this name (bad name)
@ -5127,7 +5087,7 @@ def SetMemberName(sid, member_offset, name):
return idaapi.set_member_name(s, member_offset, name) return idaapi.set_member_name(s, member_offset, name)
def SetMemberType(sid, member_offset, flag, typeid, nitems, target=None, tdelta=None, reftype=None): def SetMemberType(sid, member_offset, flag, typeid, nitems, target=-1, tdelta=0, reftype=REF_OFF32):
""" """
Change structure member type Change structure member type
@ -5153,9 +5113,11 @@ def SetMemberType(sid, member_offset, flag, typeid, nitems, target=None, tdelta=
@return: !=0 - ok. @return: !=0 - ok.
""" """
ti = _IDC_PrepareStrucMemberTypeinfo(flag, typeid, target, tdelta, reftype) if isOff0(flag):
elsize = idaapi.get_data_elsize(BADADDR, flag, ti); return Eval('SetMemberType(%d, %d, %d, %d, %d, %d, %d, %d);' % (sid, member_offset, flag, typeid, nitems,
return idaapi.set_member_type(idaapi.get_struc(sid), member_offset, flag, ti, elsize*nitems) target, tdelta, reftype))
else:
return Eval('SetMemberType(%d, %d, %d, %d, %d);' % (sid, member_offset, flag, typeid, nitems))
def SetMemberComment(sid, member_offset, comment, repeatable): def SetMemberComment(sid, member_offset, comment, repeatable):