初始提交

build/script/image_analysis/common_utils.py

@@ -0,0 +1,146 @@
+#!/usr/bin/env python3
+# coding=utf-8
+# Copyright (c) HiSilicon (Shanghai) Technologies Co., Ltd. 2021-2022. All rights reserved.
+
+from os import listdir
+from os.path import isdir, exists
+import re
+import subprocess
+import sys
+
+
+ELF_SYMBOL_RULE = r'^[0-9a-zA-Z]{8}'
+
+
+class CommonUtils:
+    symbol_key_arr = []
+    symbol_data = {}
+
+    @staticmethod
+    def get_prefix(path):
+        data = {
+            'prefix': '',
+            'environment': ''
+        }
+        if not exists(path) or not isdir(path):
+            print(f'compiler path not find or not a directory: {path}')
+            raise FileExistsError()
+        cp_files = listdir(path)
+
+        objdump_win_sign = '-objdump.exe'
+        objdump_linux_sign = '-objdump'
+        has_compiler = False
+
+        for file_name in cp_files:
+            if file_name.endswith(objdump_win_sign):
+                data['prefix'] = file_name[0: file_name.rindex(objdump_win_sign)]
+                data['environment'] = 'windows'
+                has_compiler = True
+                break
+            elif file_name.endswith(objdump_linux_sign):
+                data['prefix'] = file_name[0: file_name.rindex(objdump_linux_sign)]
+                data['environment'] = 'linux'
+                has_compiler = True
+                break
+        if not has_compiler:
+            print(f'Incorrect compiler path: {path}')
+            raise Exception()
+
+        return data
+
+    @staticmethod
+    def check_elf_file(file_path):
+        with open(file_path, 'rb') as f:
+            elf_ident = f.read(16)
+            elf_str = ''
+            for i in elf_ident:
+                elf_str += hex(i)[2:]
+            return elf_str.find('7f454c46') >= 0
+
+    @staticmethod
+    def get_obj_dump_info(obj_dump_path, arg, elf_file_path, is_return_arr=True, is_convert=True):
+        result = []
+        if not exists(obj_dump_path):
+            return result
+        cmd = [obj_dump_path, arg, elf_file_path]
+        data = CommonUtils.exec_cmd(cmd, is_convert)
+        if data:
+            if is_return_arr:
+                result = data.strip().split('\n')
+            else:
+                result.append(data)
+        return result
+
+    @staticmethod
+    def exec_cmd(cmd, is_convert=True):
+        try:
+            p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+            out, _ = p.communicate(timeout=120)
+            if is_convert:
+                out = out.decode()
+            return out
+        except subprocess.TimeoutExpired as e:
+            print('cmd timeouot')
+            raise SystemError() from e
+        except SystemError as e:
+            print('cmd error')
+            raise SystemError() from e
+
+    @staticmethod
+    def get_symbol_data(obj_dump_path='', elf_file_path=''):
+        if not obj_dump_path or not exists(obj_dump_path):
+            raise FileExistsError(f'objdump path not find: {obj_dump_path}')
+        if not elf_file_path or not exists(elf_file_path):
+            raise FileExistsError(f'elf path not find: {elf_file_path}')
+        CommonUtils.symbol_data = {}
+        CommonUtils.symbol_key_arr = []
+        res = CommonUtils.get_obj_dump_info(obj_dump_path, '-t', elf_file_path)
+        for item in res:
+            line = re.sub(r'\s+', ' ', item)
+            if re.search(ELF_SYMBOL_RULE, line):
+                data_arr = line.split(' ')
+                section_name = data_arr[3]
+                symbol_name = data_arr[-1]
+                type_name = data_arr[2]
+                if section_name != symbol_name and type_name == 'F':
+                    CommonUtils.symbol_data[data_arr[0]] = symbol_name
+        for keys in CommonUtils.symbol_data.keys():
+            hex_str = '0x' + keys
+            CommonUtils.symbol_key_arr.append(int(hex_str, 16))
+
+        if len(CommonUtils.symbol_key_arr) == 0:
+            return
+        CommonUtils.symbol_key_arr.sort(reverse=True)
+
+    @staticmethod
+    def get_function_name(num):
+        index = 0
+        d_value = sys.maxsize
+        if num.find('0x') < 0:
+            num = '0x' + num
+        for i, symbol_key_val in enumerate(CommonUtils.symbol_key_arr):
+            new_value = int(num, 16) - symbol_key_val
+            if d_value >= new_value >= 0:
+                if new_value == d_value and symbol_key_val < CommonUtils.symbol_key_arr[index]:
+                    continue
+                index = i
+                d_value = new_value
+                break
+        if CommonUtils.symbol_key_arr[index]:
+            addr = hex(CommonUtils.symbol_key_arr[index])
+            return {
+                'funcname': CommonUtils.symbol_data['%08x' % int(addr, 16)],
+                'funcaddr': '0x%08x' % int(addr, 16)
+            }
+        return {
+            'funcname': '',
+            'funcaddr': ''
+        }
+
+    @staticmethod
+    def read_uint16_le(data, offset=0):
+        return int.from_bytes(data[offset:offset + 2][::-1], byteorder='big')
+
+    @staticmethod
+    def read_uint32_le(data, offset=0):
+        return int.from_bytes(data[offset:offset + 4][::-1], byteorder='big')