kconfig-hardened-check.py

   1 #!/usr/bin/python3
   2
   3 #
   4 # This script helps me to check the Linux kernel Kconfig option list
   5 # against my hardening preferences for X86_64, ARM64, X86_32, and ARM.
   6 # Let the computers do their job!
   7 #
   8 # Author: Alexander Popov <alex.popov@linux.com>
   9 #
  10 # Please don't cry if my Python code looks like C.
  11 #
  12 #
  13 # N.B Hardening command line parameters:
  14 #    page_poison=1
  15 #    slub_debug=FZP
  16 #    slab_nomerge
  17 #    pti=on
  18 #    kernel.kptr_restrict=1
  19 #    lockdown=1
  20 #
  21 #    spectre_v2=on
  22 #    pti=on
  23 #    spec_store_bypass_disable=on
  24 #    l1tf=full,force
  25 #
  26 #
  27 # N.B. Hardening sysctl's:
  28 #    net.core.bpf_jit_harden
  29 #    kptr_restrict=2
  30
  31 import sys
  32 from argparse import ArgumentParser
  33 from collections import OrderedDict
  34 import re
  35
  36 debug_mode = False  # set it to True to print the unknown options from the config
  37
  38 supported_archs = [ 'X86_64', 'X86_32', 'ARM64', 'ARM' ]
  39
  40 checklist = []
  41
  42
  43 class OptCheck:
  44     def __init__(self, name, expected, decision, reason):
  45         self.name = name
  46         self.expected = expected
  47         self.decision = decision
  48         self.reason = reason
  49         self.state = None
  50         self.result = None
  51
  52     def check(self):
  53         if self.expected == self.state:
  54             self.result = 'OK'
  55         elif self.state is None:
  56             if self.expected == 'is not set':
  57                 self.result = 'OK: not found'
  58             else:
  59                 self.result = 'FAIL: not found'
  60         else:
  61             self.result = 'FAIL: "' + self.state + '"'
  62
  63         if self.result.startswith('OK'):
  64             return True, self.result
  65         else:
  66             return False, self.result
  67
  68     def __repr__(self):
  69         return '{} = {}'.format(self.name, self.state)
  70
  71
  72 class ComplexOptCheck:
  73     def __init__(self, *opts):
  74         self.opts = opts
  75         self.result = None
  76
  77     @property
  78     def name(self):
  79         return self.opts[0].name
  80
  81     @property
  82     def expected(self):
  83         return self.opts[0].expected
  84
  85     @property
  86     def state(self):
  87         return self.opts[0].state
  88
  89     @property
  90     def decision(self):
  91         return self.opts[0].decision
  92
  93     @property
  94     def reason(self):
  95         return self.opts[0].reason
  96
  97
  98 class OR(ComplexOptCheck):
  99     # self.opts[0] is the option which this OR-check is about.
 100     # Use case:
 101     #     OR(<X_is_hardened>, <X_is_disabled>)
 102     #     OR(<X_is_hardened>, <X_is_hardened_old>)
 103
 104     def check(self):
 105         if not self.opts:
 106             sys.exit('[!] ERROR: invalid OR check')
 107
 108         for i, opt in enumerate(self.opts):
 109             ret, msg = opt.check()
 110             if ret:
 111                 if i == 0:
 112                     self.result = opt.result
 113                 else:
 114                     self.result = 'OK: CONFIG_{} "{}"'.format(opt.name, opt.expected)
 115                 return True, self.result
 116         self.result = self.opts[0].result
 117         return False, self.result
 118
 119
 120 class AND(ComplexOptCheck):
 121     # self.opts[0] is the option which this AND-check is about.
 122     # Use case: AND(<suboption>, <main_option>)
 123     # Suboption is not checked if checking of the main_option is failed.
 124
 125     def check(self):
 126         for i, opt in reversed(list(enumerate(self.opts))):
 127             ret, msg = opt.check()
 128             if i == 0:
 129                 self.result = opt.result
 130                 return ret, self.result
 131             elif not ret:
 132                 self.result = 'FAIL: CONFIG_{} is needed'.format(opt.name)
 133                 return False, self.result
 134
 135         sys.exit('[!] ERROR: invalid AND check')
 136
 137
 138 def detect_arch(fname):
 139     with open(fname, 'r') as f:
 140         arch_pattern = re.compile("CONFIG_[a-zA-Z0-9_]*=y")
 141         arch = None
 142         msg = None
 143         print('[+] Trying to detect architecture in "{}"...'.format(fname))
 144         for line in f.readlines():
 145             if arch_pattern.match(line):
 146                 option, value = line[7:].split('=', 1)
 147                 if option in supported_archs:
 148                     if not arch:
 149                         arch = option
 150                     else:
 151                         return None, 'more than one supported architecture is detected'
 152         if not arch:
 153             return None, 'failed to detect architecture'
 154         else:
 155             return arch, 'OK'
 156
 157
 158 def construct_checklist(arch):
 159     modules_not_set = OptCheck('MODULES',                'is not set', 'kspp', 'cut_attack_surface')
 160     devmem_not_set = OptCheck('DEVMEM',                  'is not set', 'kspp', 'cut_attack_surface') # refers to LOCK_DOWN_KERNEL
 161
 162     checklist.append(OptCheck('BUG',                         'y', 'defconfig', 'self_protection'))
 163     checklist.append(OR(OptCheck('STRICT_KERNEL_RWX',        'y', 'defconfig', 'self_protection'), \
 164                         OptCheck('DEBUG_RODATA',             'y', 'defconfig', 'self_protection'))) # before v4.11
 165     checklist.append(OR(OptCheck('STACKPROTECTOR_STRONG',    'y', 'defconfig', 'self_protection'), \
 166                         OptCheck('CC_STACKPROTECTOR_STRONG', 'y', 'defconfig', 'self_protection')))
 167     checklist.append(OptCheck('SLUB_DEBUG',                  'y', 'defconfig', 'self_protection'))
 168     checklist.append(OR(OptCheck('STRICT_MODULE_RWX',        'y', 'defconfig', 'self_protection'), \
 169                         OptCheck('DEBUG_SET_MODULE_RONX',    'y', 'defconfig', 'self_protection'), \
 170                         modules_not_set)) # DEBUG_SET_MODULE_RONX was before v4.11
 171     if debug_mode or arch == 'X86_64':
 172         checklist.append(OptCheck('PAGE_TABLE_ISOLATION',    'y', 'defconfig', 'self_protection'))
 173         checklist.append(OptCheck('RANDOMIZE_MEMORY',        'y', 'defconfig', 'self_protection'))
 174     if debug_mode or arch == 'X86_64' or arch == 'X86_32':
 175         checklist.append(OptCheck('RANDOMIZE_BASE',              'y', 'defconfig', 'self_protection'))
 176         checklist.append(OptCheck('RETPOLINE',                   'y', 'defconfig', 'self_protection'))
 177         checklist.append(OptCheck('X86_SMAP',                    'y', 'defconfig', 'self_protection'))
 178         checklist.append(OptCheck('X86_INTEL_UMIP',              'y', 'defconfig', 'self_protection'))
 179         checklist.append(OptCheck('SYN_COOKIES',                 'y', 'defconfig', 'self_protection')) # another reason?
 180     if debug_mode or arch == 'ARM64':
 181         checklist.append(OptCheck('UNMAP_KERNEL_AT_EL0',         'y', 'defconfig', 'self_protection'))
 182     if debug_mode or arch == 'X86_64' or arch == 'ARM64':
 183         checklist.append(OptCheck('VMAP_STACK',                  'y', 'defconfig', 'self_protection'))
 184     if debug_mode or arch == 'X86_64' or arch == 'ARM64' or arch == 'X86_32':
 185         checklist.append(OptCheck('THREAD_INFO_IN_TASK',         'y', 'defconfig', 'self_protection'))
 186     if debug_mode or arch == 'ARM':
 187         checklist.append(OptCheck('VMSPLIT_3G',                  'y', 'defconfig', 'self_protection'))
 188         checklist.append(OptCheck('CPU_SW_DOMAIN_PAN',           'y', 'defconfig', 'self_protection'))
 189     if debug_mode or arch == 'ARM64' or arch == 'ARM':
 190         checklist.append(OptCheck('REFCOUNT_FULL',               'y', 'defconfig', 'self_protection'))
 191
 192     checklist.append(OptCheck('BUG_ON_DATA_CORRUPTION',           'y', 'kspp', 'self_protection'))
 193     checklist.append(OptCheck('DEBUG_WX',                         'y', 'kspp', 'self_protection'))
 194     checklist.append(OptCheck('SCHED_STACK_END_CHECK',            'y', 'kspp', 'self_protection'))
 195     checklist.append(OptCheck('SLAB_FREELIST_HARDENED',           'y', 'kspp', 'self_protection'))
 196     checklist.append(OptCheck('SLAB_FREELIST_RANDOM',             'y', 'kspp', 'self_protection'))
 197     checklist.append(OptCheck('FORTIFY_SOURCE',                   'y', 'kspp', 'self_protection'))
 198     checklist.append(OptCheck('GCC_PLUGINS',                      'y', 'kspp', 'self_protection'))
 199     checklist.append(OptCheck('GCC_PLUGIN_RANDSTRUCT',            'y', 'kspp', 'self_protection'))
 200     checklist.append(OptCheck('GCC_PLUGIN_STRUCTLEAK',            'y', 'kspp', 'self_protection'))
 201     checklist.append(OptCheck('GCC_PLUGIN_STRUCTLEAK_BYREF_ALL',  'y', 'kspp', 'self_protection'))
 202     checklist.append(OptCheck('GCC_PLUGIN_LATENT_ENTROPY',        'y', 'kspp', 'self_protection'))
 203     checklist.append(OptCheck('DEBUG_LIST',                       'y', 'kspp', 'self_protection'))
 204     checklist.append(OptCheck('DEBUG_SG',                         'y', 'kspp', 'self_protection'))
 205     checklist.append(OptCheck('DEBUG_CREDENTIALS',                'y', 'kspp', 'self_protection'))
 206     checklist.append(OptCheck('DEBUG_NOTIFIERS',                  'y', 'kspp', 'self_protection'))
 207     page_poisoning_is_set = OptCheck('PAGE_POISONING',            'y', 'kspp', 'self_protection')
 208     checklist.append(page_poisoning_is_set)
 209     hardened_usercopy_is_set = OptCheck('HARDENED_USERCOPY',      'y', 'kspp', 'self_protection')
 210     checklist.append(hardened_usercopy_is_set)
 211     checklist.append(AND(OptCheck('HARDENED_USERCOPY_FALLBACK',   'is not set', 'kspp', 'self_protection'), \
 212                          hardened_usercopy_is_set))
 213     checklist.append(OR(OptCheck('MODULE_SIG',                    'y', 'kspp', 'self_protection'), \
 214                         modules_not_set))
 215     checklist.append(OR(OptCheck('MODULE_SIG_ALL',                'y', 'kspp', 'self_protection'), \
 216                         modules_not_set))
 217     checklist.append(OR(OptCheck('MODULE_SIG_SHA512',             'y', 'kspp', 'self_protection'), \
 218                         modules_not_set))
 219     checklist.append(OR(OptCheck('MODULE_SIG_FORCE',              'y', 'kspp', 'self_protection'), \
 220                         modules_not_set)) # refers to LOCK_DOWN_KERNEL
 221     if debug_mode or arch == 'X86_64' or arch == 'X86_32':
 222         checklist.append(OptCheck('DEFAULT_MMAP_MIN_ADDR',        '65536', 'kspp', 'self_protection'))
 223         checklist.append(OptCheck('REFCOUNT_FULL',                'y', 'kspp', 'self_protection'))
 224     if debug_mode or arch == 'X86_32':
 225         checklist.append(OptCheck('HIGHMEM64G',                   'y', 'kspp', 'self_protection'))
 226         checklist.append(OptCheck('X86_PAE',                      'y', 'kspp', 'self_protection'))
 227     if debug_mode or arch == 'ARM64':
 228         checklist.append(OptCheck('ARM64_SW_TTBR0_PAN',           'y', 'kspp', 'self_protection'))
 229         checklist.append(OptCheck('RANDOMIZE_BASE',               'y', 'kspp', 'self_protection'))
 230     if debug_mode or arch == 'ARM64' or arch == 'ARM':
 231         checklist.append(OptCheck('SYN_COOKIES',                  'y', 'kspp', 'self_protection')) # another reason?
 232         checklist.append(OptCheck('DEFAULT_MMAP_MIN_ADDR',        '32768', 'kspp', 'self_protection'))
 233
 234     if debug_mode or arch == 'X86_64' or arch == 'ARM64' or arch == 'X86_32':
 235         checklist.append(OptCheck('GCC_PLUGIN_STACKLEAK',         'y', 'my', 'self_protection'))
 236     checklist.append(OptCheck('LOCK_DOWN_KERNEL',                 'y', 'my', 'self_protection')) # remember about LOCK_DOWN_MANDATORY
 237     checklist.append(OptCheck('SLUB_DEBUG_ON',                    'y', 'my', 'self_protection'))
 238     checklist.append(OptCheck('SECURITY_DMESG_RESTRICT',          'y', 'my', 'self_protection'))
 239     checklist.append(OptCheck('STATIC_USERMODEHELPER',            'y', 'my', 'self_protection')) # needs userspace support (systemd)
 240     checklist.append(OptCheck('SECURITY_LOADPIN',                 'y', 'my', 'self_protection')) # needs userspace support
 241     checklist.append(OptCheck('RESET_ATTACK_MITIGATION',          'y', 'my', 'self_protection')) # needs userspace support (systemd)
 242     checklist.append(OptCheck('SLAB_MERGE_DEFAULT',               'is not set', 'my', 'self_protection')) # slab_nomerge
 243     checklist.append(AND(OptCheck('PAGE_POISONING_NO_SANITY',     'is not set', 'my', 'self_protection'), \
 244                          page_poisoning_is_set))
 245     checklist.append(AND(OptCheck('PAGE_POISONING_ZERO',          'is not set', 'my', 'self_protection'), \
 246                          page_poisoning_is_set))
 247     if debug_mode or arch == 'X86_32':
 248         checklist.append(OptCheck('PAGE_TABLE_ISOLATION',         'y', 'my', 'self_protection'))
 249     if debug_mode or arch == 'ARM':
 250         checklist.append(OptCheck('STACKPROTECTOR_PER_TASK',      'y', 'my', 'self_protection'))
 251
 252     if debug_mode or arch == 'X86_64' or arch == 'ARM64' or arch == 'X86_32':
 253         checklist.append(OptCheck('SECURITY',                'y', 'defconfig', 'security_policy'))
 254     if debug_mode or arch == 'ARM':
 255         checklist.append(OptCheck('SECURITY',                'y', 'kspp', 'security_policy'))
 256     checklist.append(OptCheck('SECURITY_YAMA',               'y', 'kspp', 'security_policy'))
 257     checklist.append(OptCheck('SECURITY_SELINUX_DISABLE',    'is not set', 'kspp', 'security_policy'))
 258
 259     checklist.append(OptCheck('SECCOMP',              'y', 'defconfig', 'cut_attack_surface'))
 260     checklist.append(OptCheck('SECCOMP_FILTER',       'y', 'defconfig', 'cut_attack_surface'))
 261     if debug_mode or arch == 'X86_64' or arch == 'ARM64' or arch == 'X86_32':
 262         checklist.append(OR(OptCheck('STRICT_DEVMEM',     'y', 'defconfig', 'cut_attack_surface'), \
 263                             devmem_not_set)) # refers to LOCK_DOWN_KERNEL
 264
 265     checklist.append(modules_not_set)
 266     checklist.append(devmem_not_set)
 267     checklist.append(OR(OptCheck('IO_STRICT_DEVMEM',  'y', 'kspp', 'cut_attack_surface'), \
 268                         devmem_not_set)) # refers to LOCK_DOWN_KERNEL
 269     if debug_mode or arch == 'ARM':
 270         checklist.append(OR(OptCheck('STRICT_DEVMEM', 'y', 'kspp', 'cut_attack_surface'), \
 271                             devmem_not_set)) # refers to LOCK_DOWN_KERNEL
 272     checklist.append(OptCheck('ACPI_CUSTOM_METHOD',   'is not set', 'kspp', 'cut_attack_surface')) # refers to LOCK_DOWN_KERNEL
 273     checklist.append(OptCheck('COMPAT_BRK',           'is not set', 'kspp', 'cut_attack_surface'))
 274     checklist.append(OptCheck('DEVKMEM',              'is not set', 'kspp', 'cut_attack_surface'))
 275     checklist.append(OptCheck('COMPAT_VDSO',          'is not set', 'kspp', 'cut_attack_surface'))
 276     checklist.append(OptCheck('BINFMT_MISC',          'is not set', 'kspp', 'cut_attack_surface'))
 277     checklist.append(OptCheck('INET_DIAG',            'is not set', 'kspp', 'cut_attack_surface'))
 278     checklist.append(OptCheck('KEXEC',                'is not set', 'kspp', 'cut_attack_surface')) # refers to LOCK_DOWN_KERNEL
 279     checklist.append(OptCheck('PROC_KCORE',           'is not set', 'kspp', 'cut_attack_surface')) # refers to LOCK_DOWN_KERNEL
 280     checklist.append(OptCheck('LEGACY_PTYS',          'is not set', 'kspp', 'cut_attack_surface'))
 281     checklist.append(OptCheck('HIBERNATION',          'is not set', 'kspp', 'cut_attack_surface')) # refers to LOCK_DOWN_KERNEL
 282     if debug_mode or arch == 'X86_64':
 283         checklist.append(OptCheck('LEGACY_VSYSCALL_NONE', 'y', 'kspp', 'cut_attack_surface')) # 'vsyscall=none'
 284         checklist.append(OptCheck('IA32_EMULATION',       'is not set', 'kspp', 'cut_attack_surface'))
 285         checklist.append(OptCheck('X86_X32',              'is not set', 'kspp', 'cut_attack_surface'))
 286         checklist.append(OptCheck('MODIFY_LDT_SYSCALL',   'is not set', 'kspp', 'cut_attack_surface'))
 287     if debug_mode or arch == 'ARM':
 288         checklist.append(OptCheck('OABI_COMPAT',          'is not set', 'kspp', 'cut_attack_surface'))
 289
 290     checklist.append(OptCheck('X86_PTDUMP',              'is not set', 'grsecurity', 'cut_attack_surface'))
 291     checklist.append(OptCheck('ZSMALLOC_STAT',           'is not set', 'grsecurity', 'cut_attack_surface'))
 292     checklist.append(OptCheck('PAGE_OWNER',              'is not set', 'grsecurity', 'cut_attack_surface'))
 293     checklist.append(OptCheck('DEBUG_KMEMLEAK',          'is not set', 'grsecurity', 'cut_attack_surface'))
 294     checklist.append(OptCheck('BINFMT_AOUT',             'is not set', 'grsecurity', 'cut_attack_surface'))
 295     checklist.append(OptCheck('KPROBES',                 'is not set', 'grsecurity', 'cut_attack_surface')) # refers to LOCK_DOWN_KERNEL
 296     checklist.append(OptCheck('UPROBES',                 'is not set', 'grsecurity', 'cut_attack_surface'))
 297     checklist.append(OptCheck('GENERIC_TRACER',          'is not set', 'grsecurity', 'cut_attack_surface'))
 298     checklist.append(OptCheck('PROC_VMCORE',             'is not set', 'grsecurity', 'cut_attack_surface'))
 299     checklist.append(OptCheck('PROC_PAGE_MONITOR',       'is not set', 'grsecurity', 'cut_attack_surface'))
 300     checklist.append(OptCheck('USELIB',                  'is not set', 'grsecurity', 'cut_attack_surface'))
 301     checklist.append(OptCheck('CHECKPOINT_RESTORE',      'is not set', 'grsecurity', 'cut_attack_surface'))
 302     checklist.append(OptCheck('USERFAULTFD',             'is not set', 'grsecurity', 'cut_attack_surface'))
 303     checklist.append(OptCheck('HWPOISON_INJECT',         'is not set', 'grsecurity', 'cut_attack_surface'))
 304     checklist.append(OptCheck('MEM_SOFT_DIRTY',          'is not set', 'grsecurity', 'cut_attack_surface'))
 305     checklist.append(OptCheck('DEVPORT',                 'is not set', 'grsecurity', 'cut_attack_surface')) # refers to LOCK_DOWN_KERNEL
 306     checklist.append(OptCheck('DEBUG_FS',                'is not set', 'grsecurity', 'cut_attack_surface')) # refers to LOCK_DOWN_KERNEL
 307     checklist.append(OptCheck('NOTIFIER_ERROR_INJECTION','is not set', 'grsecurity', 'cut_attack_surface'))
 308
 309     checklist.append(OptCheck('ACPI_TABLE_UPGRADE',   'is not set', 'lockdown', 'cut_attack_surface')) # refers to LOCK_DOWN_KERNEL
 310     checklist.append(OptCheck('ACPI_APEI_EINJ',       'is not set', 'lockdown', 'cut_attack_surface')) # refers to LOCK_DOWN_KERNEL
 311     checklist.append(OptCheck('PROFILING',            'is not set', 'lockdown', 'cut_attack_surface')) # refers to LOCK_DOWN_KERNEL
 312     checklist.append(OptCheck('BPF_SYSCALL',          'is not set', 'lockdown', 'cut_attack_surface')) # refers to LOCK_DOWN_KERNEL
 313     checklist.append(OptCheck('MMIOTRACE_TEST',       'is not set', 'lockdown', 'cut_attack_surface')) # refers to LOCK_DOWN_KERNEL
 314
 315     checklist.append(OptCheck('MMIOTRACE',            'is not set', 'my', 'cut_attack_surface')) # refers to LOCK_DOWN_KERNEL (permissive)
 316     checklist.append(OptCheck('KEXEC_FILE',           'is not set', 'my', 'cut_attack_surface')) # refers to LOCK_DOWN_KERNEL (permissive)
 317     checklist.append(OptCheck('LIVEPATCH',            'is not set', 'my', 'cut_attack_surface'))
 318     checklist.append(OptCheck('USER_NS',              'is not set', 'my', 'cut_attack_surface')) # user.max_user_namespaces=0
 319     checklist.append(OptCheck('IP_DCCP',              'is not set', 'my', 'cut_attack_surface'))
 320     checklist.append(OptCheck('IP_SCTP',              'is not set', 'my', 'cut_attack_surface'))
 321     checklist.append(OptCheck('FTRACE',               'is not set', 'my', 'cut_attack_surface'))
 322     checklist.append(OptCheck('BPF_JIT',              'is not set', 'my', 'cut_attack_surface'))
 323     if debug_mode or arch == 'X86_32':
 324         checklist.append(OptCheck('MODIFY_LDT_SYSCALL', 'is not set', 'my', 'cut_attack_surface'))
 325
 326     if debug_mode or arch == 'ARM64':
 327         checklist.append(OptCheck('ARM64_PTR_AUTH',       'y', 'defconfig', 'userspace_protection'))
 328     if debug_mode or arch == 'X86_64' or arch == 'ARM64':
 329         checklist.append(OptCheck('ARCH_MMAP_RND_BITS',   '32', 'my', 'userspace_protection'))
 330     if debug_mode or arch == 'X86_32' or arch == 'ARM':
 331         checklist.append(OptCheck('ARCH_MMAP_RND_BITS',   '16', 'my', 'userspace_protection'))
 332
 333 #   checklist.append(OptCheck('LKDTM',    'm', 'my', 'feature_test'))
 334
 335
 336 def print_checklist(arch):
 337     print('[+] Printing kernel hardening preferences for {}...'.format(arch))
 338     print('  {:<39}|{:^13}|{:^10}|{:^20}'.format(
 339         'option name', 'desired val', 'decision', 'reason'))
 340     print('  ' + '=' * 86)
 341     for opt in checklist:
 342         print('  CONFIG_{:<32}|{:^13}|{:^10}|{:^20}'.format(
 343             opt.name, opt.expected, opt.decision, opt.reason))
 344     print()
 345
 346
 347 def print_check_results():
 348     print('  {:<39}|{:^13}|{:^10}|{:^20}||{:^28}'.format(
 349         'option name', 'desired val', 'decision', 'reason', 'check result'))
 350     print('  ' + '=' * 115)
 351     for opt in checklist:
 352         print('  CONFIG_{:<32}|{:^13}|{:^10}|{:^20}||{:^28}'.format(
 353             opt.name, opt.expected, opt.decision, opt.reason, opt.result))
 354     print()
 355
 356
 357 def get_option_state(options, name):
 358     return options.get(name, None)
 359
 360
 361 def perform_checks(parsed_options):
 362     for opt in checklist:
 363         if hasattr(opt, 'opts'):
 364             for o in opt.opts:
 365                 o.state = get_option_state(parsed_options, o.name)
 366         else:
 367             opt.state = get_option_state(parsed_options, opt.name)
 368         opt.check()
 369
 370
 371 def check_config_file(fname):
 372     with open(fname, 'r') as f:
 373         parsed_options = OrderedDict()
 374         opt_is_on = re.compile("CONFIG_[a-zA-Z0-9_]*=[a-zA-Z0-9_\"]*")
 375         opt_is_off = re.compile("# CONFIG_[a-zA-Z0-9_]* is not set")
 376
 377         print('[+] Checking "{}" against hardening preferences...'.format(fname))
 378         for line in f.readlines():
 379             line = line.strip()
 380             option = None
 381             value = None
 382
 383             if opt_is_on.match(line):
 384                 option, value = line[7:].split('=', 1)
 385             elif opt_is_off.match(line):
 386                 option, value = line[9:].split(' ', 1)
 387                 if value != 'is not set':
 388                     sys.exit('[!] ERROR: bad disabled config option "{}"'.format(line))
 389
 390             if option in parsed_options:
 391                 sys.exit('[!] ERROR: config option "{}" exists multiple times'.format(line))
 392
 393             if option is not None:
 394                 parsed_options[option] = value
 395
 396         perform_checks(parsed_options)
 397
 398         if debug_mode:
 399             known_options = [opt.name for opt in checklist]
 400             for option, value in parsed_options.items():
 401                 if option not in known_options:
 402                     print("DEBUG: dunno about option {} ({})".format(option, value))
 403
 404         print_check_results()
 405
 406
 407 if __name__ == '__main__':
 408     parser = ArgumentParser(description='Checks the hardening options in the Linux kernel config')
 409     parser.add_argument('-p', '--print', choices=supported_archs,
 410                         help='print hardening preferences for selected architecture')
 411     parser.add_argument('-c', '--config',
 412                         help='check the config_file against these preferences')
 413     parser.add_argument('--debug', action='store_true',
 414                         help='enable internal debug mode')
 415     args = parser.parse_args()
 416
 417     if args.debug:
 418         debug_mode = True
 419
 420     if args.config:
 421         arch, msg = detect_arch(args.config)
 422         if not arch:
 423             sys.exit('[!] ERROR: {}'.format(msg))
 424         else:
 425             print('[+] Detected architecture: {}'.format(arch))
 426
 427         construct_checklist(arch)
 428         check_config_file(args.config)
 429         error_count = len(list(filter(lambda opt: opt.result.startswith('FAIL'), checklist)))
 430         ok_count = len(list(filter(lambda opt: opt.result.startswith('OK'), checklist)))
 431         if debug_mode:
 432             sys.exit(0)
 433         print('[+] config check is finished: \'OK\' - {} / \'FAIL\' - {}'.format(ok_count, error_count))
 434         sys.exit(0)
 435
 436     if args.print:
 437         arch = args.print
 438         construct_checklist(arch)
 439         print_checklist(arch)
 440         sys.exit(0)
 441
 442     parser.print_help()