Commit 0cc764f5 by Alexander Popov

Refactoring of the argument parsing

parent db71a9e2
...@@ -165,14 +165,14 @@ def parse_kconfig_file(parsed_options, fname): ...@@ -165,14 +165,14 @@ def parse_kconfig_file(parsed_options, fname):
if opt_is_on.match(line): if opt_is_on.match(line):
option, value = line.split('=', 1) option, value = line.split('=', 1)
if value == 'is not set': if value == 'is not set':
sys.exit(f'[!] ERROR: bad enabled kconfig option "{line}"') sys.exit(f'[!] ERROR: bad enabled Kconfig option "{line}"')
elif opt_is_off.match(line): elif opt_is_off.match(line):
option, value = line[2:].split(' ', 1) option, value = line[2:].split(' ', 1)
if value != 'is not set': if value != 'is not set':
sys.exit(f'[!] ERROR: bad disabled kconfig option "{line}"') sys.exit(f'[!] ERROR: bad disabled Kconfig option "{line}"')
if option in parsed_options: if option in parsed_options:
sys.exit(f'[!] ERROR: kconfig option "{line}" exists multiple times') sys.exit(f'[!] ERROR: Kconfig option "{line}" exists multiple times')
if option: if option:
parsed_options[option] = value parsed_options[option] = value
...@@ -200,7 +200,7 @@ def parse_cmdline_file(parsed_options, fname): ...@@ -200,7 +200,7 @@ def parse_cmdline_file(parsed_options, fname):
def main(): def main():
# Report modes: # Report modes:
# * verbose mode for # * verbose mode for
# - reporting about unknown kernel options in the kconfig # - reporting about unknown kernel options in the Kconfig
# - verbose printing of ComplexOptCheck items # - verbose printing of ComplexOptCheck items
# * json mode for printing the results in JSON format # * json mode for printing the results in JSON format
report_modes = ['verbose', 'json', 'show_ok', 'show_fail'] report_modes = ['verbose', 'json', 'show_ok', 'show_fail']
...@@ -208,14 +208,14 @@ def main(): ...@@ -208,14 +208,14 @@ def main():
parser = ArgumentParser(prog='kconfig-hardened-check', parser = ArgumentParser(prog='kconfig-hardened-check',
description='A tool for checking the security hardening options of the Linux kernel') description='A tool for checking the security hardening options of the Linux kernel')
parser.add_argument('--version', action='version', version='%(prog)s ' + __version__) parser.add_argument('--version', action='version', version='%(prog)s ' + __version__)
parser.add_argument('-p', '--print', choices=supported_archs, parser.add_argument('-m', '--mode', choices=report_modes,
help='print the security hardening recommendations for the selected microarchitecture') help='choose the report mode')
parser.add_argument('-c', '--config', parser.add_argument('-c', '--config',
help='check the security hardening options in the kernel kconfig file (also supports *.gz files)') help='check the security hardening options in the kernel Kconfig file (also supports *.gz files)')
parser.add_argument('-l', '--cmdline', parser.add_argument('-l', '--cmdline',
help='check the security hardening options in the kernel cmdline file') help='check the security hardening options in the kernel cmdline file')
parser.add_argument('-m', '--mode', choices=report_modes, parser.add_argument('-p', '--print', choices=supported_archs,
help='choose the report mode') help='print the security hardening recommendations for the selected microarchitecture')
args = parser.parse_args() args = parser.parse_args()
mode = None mode = None
...@@ -254,14 +254,14 @@ def main(): ...@@ -254,14 +254,14 @@ def main():
else: else:
print(f'[-] Can\'t detect the compiler: {msg}') print(f'[-] Can\'t detect the compiler: {msg}')
# add relevant kconfig checks to the checklist # add relevant Kconfig checks to the checklist
add_kconfig_checks(config_checklist, arch) add_kconfig_checks(config_checklist, arch)
if args.cmdline: if args.cmdline:
# add relevant cmdline checks to the checklist # add relevant cmdline checks to the checklist
add_cmdline_checks(config_checklist, arch) add_cmdline_checks(config_checklist, arch)
# populate the checklist with the parsed kconfig data # populate the checklist with the parsed Kconfig data
parsed_kconfig_options = OrderedDict() parsed_kconfig_options = OrderedDict()
parse_kconfig_file(parsed_kconfig_options, args.config) parse_kconfig_file(parsed_kconfig_options, args.config)
populate_with_data(config_checklist, parsed_kconfig_options, 'kconfig') populate_with_data(config_checklist, parsed_kconfig_options, 'kconfig')
...@@ -295,10 +295,11 @@ def main(): ...@@ -295,10 +295,11 @@ def main():
sys.exit(0) sys.exit(0)
elif args.cmdline: elif args.cmdline:
sys.exit('[!] ERROR: checking cmdline doesn\'t work without checking kconfig') sys.exit('[!] ERROR: checking cmdline doesn\'t work without checking Kconfig')
if args.print: if args.print:
if mode in ('show_ok', 'show_fail'): assert(args.config is None and args.cmdline is None), 'unexpected args'
if mode and mode not in ('verbose', 'json'):
sys.exit(f'[!] ERROR: wrong mode "{mode}" for --print') sys.exit(f'[!] ERROR: wrong mode "{mode}" for --print')
arch = args.print arch = args.print
add_kconfig_checks(config_checklist, arch) add_kconfig_checks(config_checklist, arch)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment