return the target's address for the function-name feature

This commit is contained in:
Yacine Elhamer
2023-06-01 12:45:49 +01:00
parent f9291d4e50
commit 994edf66fe

View File

@@ -138,7 +138,7 @@ def extract_insn_api_features(fh: FunctionHandle, bb, ih: InsnHandle) -> Iterato
STT_FUNC = 0x2
if sym_value == target and sym_info & STT_FUNC != 0:
yield API(sym_name), ih.address
yield FunctionName(sym_name), ih.address
yield FunctionName(sym_name), target
if viv_utils.flirt.is_library_function(f.vw, target):
name = viv_utils.get_function_name(f.vw, target)