Linux获取so/ko文件版本号教程

import os
import re
import platform

class ReadLibVersion:
    def __init__(self):
        so_version_file = "so_version.txt"
        ko_version_file = "ko_version.txt"
        self.so_version_file_obj = open(so_version_file,"w+")
        self.ko_version_file_obj = open(ko_version_file,"w+")
        system = platform.system()
        if "Windows" in system:
            self.path_split = "\\"
        else:
            self.path_split = "/"
        self.so_file_pattern = "\.so$"
        self.ko_file_pattern = "\.ko$"

    # 遍历要扫描的目录，寻找出目录下的所有so/ko文件
    def traversal_dir(self,dir):
        dir_contains = os.listdir(dir)
        for tmp in dir_contains:
            tmp_path = f"{dir}{self.path_split}{tmp}"
            if os.path.isfile(tmp_path):
                # 如果是so文件，则调用so版本号获取函数获取版本号
                if re.search(self.so_file_pattern,tmp_path) is not None:
                    # 首先从文件名获取版本号
                    so_version = self.read_so_version_by_name(tmp_path)
                    if so_version != "-":
                        self.so_version_file_obj.writelines(f"{tmp_path}\t\t{so_version}\r\n")
                    else:
                        # 如果从文件名获取不到版本号，那么就通过readelf读SONAME获取主版本号
                        so_version = self.read_so_version_by_readelf(tmp_path)
                        self.so_version_file_obj.writelines(f"{tmp_path}\t\t{so_version}\r\n")
                # 如果是ko文件，则通过modinfo读取version变量获取版本号
                elif re.search(self.ko_file_pattern,tmp_path) is not None:
                    ko_version = self.read_ko_version_by_modinfo(tmp_path)
                    self.ko_version_file_obj.writelines(f"{tmp_path}\t\t{ko_version}\r\n")
            elif os.path.isdir(tmp_path):
                sub_dir = tmp_path
                self.traversal_dir(sub_dir)

    # 从文件名获取so文件版本号
    def read_so_version_by_name(self,file_path):
        file = file_path.split(f"{self.path_split}")[-1]
        ver_pattern = "[.|\d]+\d"
        so_versions = re.findall(ver_pattern,file)
        if len(so_versions) >0:
            so_version = so_versions[0]
        else:
            so_version = "-"
        return so_version
        pass

    # 使用readelf读SONAME获取so文件主版本号
    def read_so_version_by_readelf(self,file_path):
        so_version = "-"
        readelf_result = os.popen(f"readelf -d  {file_path} |grep SONAME").read().strip()
        if readelf_result != "":
            so_version = readelf_result.split()[-1]
        return so_version
        pass

    # 使用modinfo读取version变量获取ko文件版本号
    def read_ko_version_by_modinfo(self,file_path):
        ko_version = "-"
        modinfo_result = os.popen(f"modinfo {file_path}|grep ^version:").read().strip()
        if modinfo_result != "":
            ko_version = modinfo_result.split()[-1]
        return ko_version
        pass

    def __del__(self):
        self.so_version_file_obj.close()
        self.ko_version_file_obj.close()
        pass


if __name__ == "__main__":
    # 要扫描的目录
    root_dir = "/usr/lib"
    read_lib_version_obj = ReadLibVersion()
    read_lib_version_obj.traversal_dir(root_dir)