address Mike's feedback

This commit is contained in:
Moritz Raabe
2021-06-29 08:57:43 +02:00
parent 18c87e4e55
commit 04b5949a05

View File

@@ -101,7 +101,7 @@ def extract_insn_api_features(f, bb, insn):
yield API(name), insn.ea
# extract IDA/FLIRT recognized API functions
targets = list(idautils.CodeRefsFrom(insn.ea, False))
targets = tuple(idautils.CodeRefsFrom(insn.ea, False))
if not targets:
return
@@ -111,8 +111,8 @@ def extract_insn_api_features(f, bb, insn):
# not a function (start)
return
if idaapi.get_func(target).flags & idaapi.FUNC_LIB:
name = idaapi.get_name(target)
if target_func.flags & idaapi.FUNC_LIB:
name = idaapi.get_name(target_func.start_ea)
yield API(name), insn.ea