IDA: Optimize function creation (#75)

This commit is contained in:
lifeengines
2020-09-22 02:25:35 +08:00
committed by GitHub
parent 2afdc8b421
commit db1805cb12
3 changed files with 22 additions and 5 deletions

View File

@@ -7,8 +7,16 @@ def SetName(addr, name):
new_name = name + '_' + str(addr)
ret = idc.set_name(addr, new_name, SN_NOWARN | SN_NOCHECK)
def MakeFunction(start):
ida_funcs.add_func(start)
def MakeFunction(start, name=None, addrMax=None):
ida_funcs.add_func(start)
#limit end function to maxAddr if any
if addrMax is None:
return
addrEnd = idc.get_func_attr(start,FUNCATTR_END)
if addrEnd == idaapi.BADADDR:
return
if addrEnd > addrMax:
idc.set_func_end(start,addrMax)
def MakeArray(addr, numItems, cppType):
SetType(addr, cppType)