idaapi的方法可能变动，导致在IDA9无法使用（加了下自己的patch）

[hahbiubiubiu]

Traceback (most recent call last):
File ".../IDA/plugins/finger_plugin.py", line 20, in recognize_function
func_feat = ida_func.get_func_feature(start_ea)
File "...\Python\lib\site-packages\finger_sdk\ida_func.py", line 106, in get_func_feature
arch, endian = Func.get_file_structure()
File "...\Python\lib\site-packages\finger_sdk\ida_func.py", line 23, in get_file_structure
info = idaapi.get_inf_structure()
AttributeError: module 'idaapi' has no attribute 'get_inf_structure'

[-]sub_3D5CA4 recognize failed

[hahbiubiubiu]

修改了一下api，适配了一下ida9的api。

# ...\finger_sdk\ida_func.py
import idc
import idaapi
import idautils
import ida_loader
import ida_ida
import ida_kernwin
import re


class FuncSigFeature:
    def __init__(self):
        self.file_path = idc.get_input_file_path()

        self.code_list = ["",".text",".plt",".got","extern",".pdata",".bss"]

        self.control_ins_list = ["call","jc","jnc","jz","jnz","js","jns","jo","jno","jp",
                                "jpe","jnp","jpo","ja","jnbe","jae","jnb","jb","jnae","jbe",
                                "jna","je","jne","jg","jnle","jge","jnl","jl","jnge","jle","jng"]
       
        self.string_list = dict()
        self.is_patched = False
        self.check_ida_version()
        for s in idautils.Strings():
            self.string_list[str(s)] = s.ea

    def check_ida_version(self):
        print("checking ida version")
        kv = ida_kernwin.get_kernel_version().split(".")
        if (int(kv[0]) == 9) or (int(kv[1]) == 0):
            print("IDA version is 9.0. Using the patched version of the plugin.")
            self.is_patched = True

    def get_file_structure(self):
        if self.is_patched:
            return self.get_file_structure_patched()
        else:
            return self.get_file_structure_nopatched()

    def get_file_type(self):
        if self.is_patched:
            return self.get_file_type_patched()
        else:
            return self.get_file_type_nopatched()

    def get_file_structure_nopatched(self):
        info = idaapi.get_inf_structure()
        arch = info.procname
        if info.is_be():
            endian = "MSB"
        else:
            endian = "LSB"
        return arch, endian
    
    def get_file_type_nopatched(self):
        file_format = ""
        file_type = ""
        info = idaapi.get_inf_structure()
        if info.is_64bit():
            file_format = "64"
        elif info.is_32bit():
            file_format = "32"
        if info.filetype == idaapi.f_PE:
            file_type = "PE"
        elif info.filetype == idaapi.f_ELF:
            file_type = "ELF"
        return file_format, file_type

    def get_file_structure_patched(self):
        arch = ida_ida.inf_get_procname()
        if ida_ida.inf_is_be():
            endian = "MSB"
        else:
            endian = "LSB"
        return arch, endian
    
    def get_file_type_patched(self):
        file_format = ""
        file_type = ""
        if ida_ida.inf_is_64bit():
            file_format = "64"
        elif ida_ida.idainfo_is_32bit():
            file_format = "32"
        if 'PE' in ida_loader.get_file_type_name(): 
            file_type = "PE"
        elif 'ELF' in ida_loader.get_file_type_name():
            file_type = "ELF"
        return file_format, file_type


    def get_module_info(self):
        module_info = ""
        if len(idc.ARGV) == 2:
            module_info = idc.ARGV[1]
        return module_info


    def byte2str(self, l):
        if "bytes" in str(type(l)):
            l = l.decode()
        return l


    def extract_const(self, ins_addr):
        const_str = ""
        op_str = idc.print_insn_mnem(ins_addr)
        if op_str not in self.control_ins_list:
            for i in range(2):
                operand_type = idc.get_operand_type(ins_addr, i)
                if operand_type == idc.o_mem:
                    const_addr = idc.get_operand_value(ins_addr, i)
                    if idc.get_segm_name(const_addr) not in self.code_list:
                        str_const = idc.get_strlit_contents(const_addr)
                        if str_const:
                            str_const = self.byte2str(str_const)
                            if (str_const in self.string_list) and (const_addr == self.string_list[str_const]):
                                const_str += str_const
                                break
        return const_str


    def get_ins_feature(self, start_ea):
        ins_str_list = list()
        ins_bytes_list = list()
        ins_list = list(idautils.FuncItems(start_ea))
        for ins_addr in ins_list:
            ins_bytes = idc.get_bytes(ins_addr, idc.get_item_size(ins_addr))
            ins_bytes_list.append(ins_bytes)
            ins_str = self.extract_const(ins_addr)
            ins_str_list.append(ins_str)
        return ins_bytes_list, ins_str_list

    
    def filter_segment(self, func_addr):
        ignore_list = ["extern",".plt",".got",".idata"]
        if idc.get_segm_name(func_addr) in ignore_list:
            return True
        else:
            return False


def get_func_feature(ea):
    content = dict()
    pfn = idaapi.get_func(ea)
    if pfn:
        func_addr = pfn.start_ea
        Func = FuncSigFeature()
        if Func.filter_segment(func_addr):
            return None
        arch, endian = Func.get_file_structure()
        file_format, file_type = Func.get_file_type()
        module_info = Func.get_module_info()
        ins_bytes_list, ins_str_list = Func.get_ins_feature(func_addr)
        content["extmsg"] = [arch, endian, file_format, file_type, module_info]
        content["ins_bytes"] = ins_bytes_list
        content["ins_str"] = ins_str_list
        content["func_name"] = idaapi.get_func_name(func_addr)
        return content
    else:
        return None