应用安全 --- IDAPro脚本 之 导出所有函数地址

import idautils
import idc
import csvdef export_frida_functions_csv():"""导出适合Frida使用的函数列表"""with open("D:/frida_functions.csv", "w", newline='', encoding='utf-8-sig') as f:writer = csv.writer(f)# Frida专用表头writer.writerow(["Module", "Function_Name", "Address", "Size","Export_Type"])count = 0# 获取当前模块名称module_name = idc.get_input_file_path().split('\\')[-1].split('/')[-1]for func_ea in idautils.Functions():func_name = idc.get_func_name(func_ea)start_ea = func_eaend_ea = idc.find_func_end(func_ea)# 计算函数大小if end_ea != idc.BADADDR:func_size = end_ea - start_eaelse:func_size = 0# 判断导出类型flags = idc.get_func_attr(func_ea, idc.FUNCATTR_FLAGS)if flags & idc.FUNC_LIB:export_type = "Import"else:export_type = "Export"# 写入CSV行writer.writerow([module_name,func_name, "0x{:X}".format(start_ea), func_size,export_type])count += 1print("[+] {} functions exported to D:/frida_functions.csv for Frida usage".format(count))export_frida_functions_csv()