X-Git-Url: https://jxself.org/git/?a=blobdiff_plain;f=kernel_hardening_checker%2F__init__.py;h=91742c3923d78973a2d865b05adee609851c4568;hb=f6075e933d9ec15d4b0bc216f764acbdfb51235d;hp=182f76fd7fdf67083747ac340279b26d130408d6;hpb=24c2654fcbc7c0dcd9a1d51e1720f8160d4f5931;p=kconfig-hardened-check.git diff --git a/kernel_hardening_checker/__init__.py b/kernel_hardening_checker/__init__.py index 182f76f..91742c3 100644 --- a/kernel_hardening_checker/__init__.py +++ b/kernel_hardening_checker/__init__.py @@ -8,7 +8,7 @@ Author: Alexander Popov This module performs input/output. """ -# pylint: disable=missing-function-docstring,line-too-long,invalid-name,too-many-branches,too-many-statements +# pylint: disable=missing-function-docstring,line-too-long,too-many-branches,too-many-statements import gzip import sys @@ -16,19 +16,23 @@ from argparse import ArgumentParser from typing import List, Tuple, Dict, TextIO import re import json -from .__about__ import __version__ from .checks import add_kconfig_checks, add_cmdline_checks, normalize_cmdline_options, add_sysctl_checks -from .engine import StrOrNone, TupleOrNone, ChecklistObjType, print_unknown_options, populate_with_data, perform_checks, override_expected_value +from .engine import StrOrNone, TupleOrNone, ChecklistObjType +from .engine import print_unknown_options, populate_with_data, perform_checks, override_expected_value -def _open(file: str, *args, **kwargs) -> TextIO: +# kernel-hardening-checker version +__version__ = '0.6.6' + + +def _open(file: str) -> TextIO: if file.endswith('.gz'): - return gzip.open(file, *args, **kwargs) - return open(file, *args, **kwargs) + return gzip.open(file, 'rt', encoding='utf-8') + return open(file, 'rt', encoding='utf-8') def detect_arch(fname: str, archs: List[str]) -> Tuple[StrOrNone, str]: - with _open(fname, 'rt', encoding='utf-8') as f: + with _open(fname) as f: arch_pattern = re.compile(r"CONFIG_[a-zA-Z0-9_]+=y$") arch = None for line in f.readlines(): @@ -45,7 +49,7 @@ def detect_arch(fname: str, archs: List[str]) -> Tuple[StrOrNone, str]: def detect_kernel_version(fname: str) -> Tuple[TupleOrNone, str]: - with _open(fname, 'rt', encoding='utf-8') as f: + with _open(fname) as f: ver_pattern = re.compile(r"^# Linux/.+ Kernel Configuration$|^Linux version .+") for line in f.readlines(): if ver_pattern.match(line): @@ -54,7 +58,7 @@ def detect_kernel_version(fname: str) -> Tuple[TupleOrNone, str]: ver_str = parts[2].split('-', 1)[0] ver_numbers = ver_str.split('.') if len(ver_numbers) >= 3: - if all(map(lambda x: x.isdigit(), ver_numbers)): + if all(map(lambda x: x.isdecimal(), ver_numbers)): return tuple(map(int, ver_numbers)), 'OK' msg = f'failed to parse the version "{parts[2]}"' return None, msg @@ -64,7 +68,7 @@ def detect_kernel_version(fname: str) -> Tuple[TupleOrNone, str]: def detect_compiler(fname: str) -> Tuple[StrOrNone, str]: gcc_version = None clang_version = None - with _open(fname, 'rt', encoding='utf-8') as f: + with _open(fname) as f: for line in f.readlines(): if line.startswith('CONFIG_GCC_VERSION='): gcc_version = line[19:-1] @@ -132,7 +136,7 @@ def print_checklist(mode: StrOrNone, checklist: List[ChecklistObjType], with_res def parse_kconfig_file(_mode: StrOrNone, parsed_options: Dict[str, str], fname: str) -> None: - with _open(fname, 'rt', encoding='utf-8') as f: + with _open(fname) as f: opt_is_on = re.compile(r"CONFIG_[a-zA-Z0-9_]+=.+$") opt_is_off = re.compile(r"# CONFIG_[a-zA-Z0-9_]+ is not set$")