Add DRM_LEGACY, FB, and VT checks
[kconfig-hardened-check.git] / kconfig_hardened_check / __init__.py
1 #!/usr/bin/python3
2
3 #
4 # This tool 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 #    slub_debug=FZP
15 #    slab_nomerge
16 #    page_alloc.shuffle=1
17 #    iommu=force (does it help against DMA attacks?)
18 #    page_poison=1 (if enabled)
19 #    init_on_alloc=1
20 #    init_on_free=1
21 #    loadpin.enforce=1
22 #
23 #    Mitigations of CPU vulnerabilities:
24 #       Аrch-independent:
25 #           mitigations=auto,nosmt
26 #       X86:
27 #           spectre_v2=on
28 #           pti=on
29 #           spec_store_bypass_disable=on
30 #           l1tf=full,force
31 #           mds=full,nosmt
32 #           tsx=off
33 #       ARM64:
34 #           kpti=on
35 #           ssbd=force-on
36 #
37 # N.B. Hardening sysctls:
38 #    kernel.kptr_restrict=2
39 #    kernel.dmesg_restrict=1
40 #    kernel.perf_event_paranoid=3
41 #    kernel.kexec_load_disabled=1
42 #    kernel.yama.ptrace_scope=3
43 #    user.max_user_namespaces=0
44 #    kernel.unprivileged_bpf_disabled=1
45 #    net.core.bpf_jit_harden=2
46 #
47 #    vm.unprivileged_userfaultfd=0
48 #
49 #    dev.tty.ldisc_autoload=0
50 #    fs.protected_symlinks=1
51 #    fs.protected_hardlinks=1
52 #    fs.protected_fifos=2
53 #    fs.protected_regular=2
54 #    fs.suid_dumpable=0
55 #    kernel.modules_disabled=1
56
57 import sys
58 from argparse import ArgumentParser
59 from collections import OrderedDict
60 import re
61 import json
62
63 # debug_mode enables:
64 #    - reporting about unknown kernel options in the config,
65 #    - verbose printing of ComplexOptChecks (OR, AND).
66 debug_mode = False
67
68 # json_mode is for printing results in JSON format
69 json_mode = False
70
71 supported_archs = [ 'X86_64', 'X86_32', 'ARM64', 'ARM' ]
72
73 kernel_version = None
74
75
76 class OptCheck:
77     def __init__(self, name, expected, decision, reason):
78         self.name = name
79         self.expected = expected
80         self.decision = decision
81         self.reason = reason
82         self.state = None
83         self.result = None
84
85     def check(self):
86         if self.expected == self.state:
87             self.result = 'OK'
88         elif self.state is None:
89             if self.expected == 'is not set':
90                 self.result = 'OK: not found'
91             else:
92                 self.result = 'FAIL: not found'
93         else:
94             self.result = 'FAIL: "' + self.state + '"'
95
96         if self.result.startswith('OK'):
97             return True, self.result
98         else:
99             return False, self.result
100
101     def table_print(self, with_results):
102         print('CONFIG_{:<38}|{:^13}|{:^10}|{:^20}'.format(self.name, self.expected, self.decision, self.reason), end='')
103         if with_results:
104             print('|   {}'.format(self.result), end='')
105
106
107 class VerCheck:
108     def __init__(self, ver_expected):
109         self.ver_expected = ver_expected
110         self.result = None
111
112     def check(self):
113         if kernel_version[0] > self.ver_expected[0]:
114             self.result = 'OK: version >= ' + str(self.ver_expected[0]) + '.' + str(self.ver_expected[1])
115             return True, self.result
116         if kernel_version[0] < self.ver_expected[0]:
117             self.result = 'FAIL: version < ' + str(self.ver_expected[0]) + '.' + str(self.ver_expected[1])
118             return False, self.result
119         if kernel_version[1] >= self.ver_expected[1]:
120             self.result = 'OK: version >= ' + str(self.ver_expected[0]) + '.' + str(self.ver_expected[1])
121             return True, self.result
122         else:
123             self.result = 'FAIL: version < ' + str(self.ver_expected[0]) + '.' + str(self.ver_expected[1])
124             return False, self.result
125
126     def table_print(self, with_results):
127         ver_req = 'kernel version >= ' + str(self.ver_expected[0]) + '.' + str(self.ver_expected[1])
128         print('{:<91}'.format(ver_req), end='')
129         if with_results:
130             print('|   {}'.format(self.result), end='')
131
132
133 class PresenceCheck:
134     def __init__(self, name):
135         self.name = name
136         self.state = None
137         self.result = None
138
139     def check(self):
140         if self.state is None:
141             self.result = 'FAIL: not present'
142             return False, self.result
143         else:
144             self.result = 'OK: is present'
145             return True, self.result
146
147     def table_print(self, with_results):
148         print('CONFIG_{:<84}'.format(self.name + ' is present'), end='')
149         if with_results:
150             print('|   {}'.format(self.result), end='')
151
152
153 class ComplexOptCheck:
154     def __init__(self, *opts):
155         self.opts = opts
156         self.result = None
157
158     @property
159     def name(self):
160         return self.opts[0].name
161
162     @property
163     def expected(self):
164         return self.opts[0].expected
165
166     @property
167     def state(self):
168         return self.opts[0].state
169
170     @property
171     def decision(self):
172         return self.opts[0].decision
173
174     @property
175     def reason(self):
176         return self.opts[0].reason
177
178     def table_print(self, with_results):
179         if debug_mode:
180             print('    {:87}'.format('<<< ' + self.__class__.__name__ + ' >>>'), end='')
181             if with_results:
182                 print('|   {}'.format(self.result), end='')
183             for o in self.opts:
184                 print()
185                 o.table_print(with_results)
186         else:
187             o = self.opts[0]
188             o.table_print(False)
189             if with_results:
190                 print('|   {}'.format(self.result), end='')
191
192
193 class OR(ComplexOptCheck):
194     # self.opts[0] is the option that this OR-check is about.
195     # Use case:
196     #     OR(<X_is_hardened>, <X_is_disabled>)
197     #     OR(<X_is_hardened>, <X_is_hardened_old>)
198
199     def check(self):
200         if not self.opts:
201             sys.exit('[!] ERROR: invalid OR check')
202
203         for i, opt in enumerate(self.opts):
204             ret, msg = opt.check()
205             if ret:
206                 if i == 0 or not hasattr(opt, 'expected'):
207                     self.result = opt.result
208                 else:
209                     self.result = 'OK: CONFIG_{} "{}"'.format(opt.name, opt.expected)
210                 return True, self.result
211         self.result = self.opts[0].result
212         return False, self.result
213
214
215 class AND(ComplexOptCheck):
216     # self.opts[0] is the option that this AND-check is about.
217     # Use case: AND(<suboption>, <main_option>)
218     # Suboption is not checked if checking of the main_option is failed.
219
220     def check(self):
221         for i, opt in reversed(list(enumerate(self.opts))):
222             ret, msg = opt.check()
223             if i == 0:
224                 self.result = opt.result
225                 return ret, self.result
226             elif not ret:
227                 if hasattr(opt, 'expected'):
228                     self.result = 'FAIL: CONFIG_{} is needed'.format(opt.name)
229                 else:
230                     self.result = opt.result
231                 return False, self.result
232
233         sys.exit('[!] ERROR: invalid AND check')
234
235
236 def detect_arch(fname):
237     with open(fname, 'r') as f:
238         arch_pattern = re.compile("CONFIG_[a-zA-Z0-9_]*=y")
239         arch = None
240         if not json_mode:
241             print('[+] Trying to detect architecture in "{}"...'.format(fname))
242         for line in f.readlines():
243             if arch_pattern.match(line):
244                 option, value = line[7:].split('=', 1)
245                 if option in supported_archs:
246                     if not arch:
247                         arch = option
248                     else:
249                         return None, 'more than one supported architecture is detected'
250         if not arch:
251             return None, 'failed to detect architecture'
252         else:
253             return arch, 'OK'
254
255
256 def detect_version(fname):
257     with open(fname, 'r') as f:
258         ver_pattern = re.compile("# Linux/.* Kernel Configuration")
259         if not json_mode:
260             print('[+] Trying to detect kernel version in "{}"...'.format(fname))
261         for line in f.readlines():
262             if ver_pattern.match(line):
263                 line = line.strip()
264                 if not json_mode:
265                     print('[+] Found version line: "{}"'.format(line))
266                 parts = line.split()
267                 ver_str = parts[2]
268                 ver_numbers = ver_str.split('.')
269                 if len(ver_numbers) < 3 or not ver_numbers[0].isdigit() or not ver_numbers[1].isdigit():
270                     msg = 'failed to parse the version "' + ver_str + '"'
271                     return None, msg
272                 else:
273                     return (int(ver_numbers[0]), int(ver_numbers[1])), None
274         return None, 'no kernel version detected'
275
276
277 def construct_checklist(checklist, arch):
278     modules_not_set = OptCheck('MODULES',     'is not set', 'kspp', 'cut_attack_surface')
279     devmem_not_set = OptCheck('DEVMEM',       'is not set', 'kspp', 'cut_attack_surface') # refers to LOCKDOWN
280
281     checklist.append(OptCheck('BUG',                         'y', 'defconfig', 'self_protection'))
282     checklist.append(OR(OptCheck('STRICT_KERNEL_RWX',        'y', 'defconfig', 'self_protection'), \
283                         OptCheck('DEBUG_RODATA',             'y', 'defconfig', 'self_protection'))) # before v4.11
284     checklist.append(OR(OptCheck('STACKPROTECTOR_STRONG',    'y', 'defconfig', 'self_protection'), \
285                         OptCheck('CC_STACKPROTECTOR_STRONG', 'y', 'defconfig', 'self_protection')))
286     checklist.append(OptCheck('SLUB_DEBUG',                  'y', 'defconfig', 'self_protection'))
287     checklist.append(OR(OptCheck('STRICT_MODULE_RWX',        'y', 'defconfig', 'self_protection'), \
288                         OptCheck('DEBUG_SET_MODULE_RONX',    'y', 'defconfig', 'self_protection'), \
289                         modules_not_set)) # DEBUG_SET_MODULE_RONX was before v4.11
290     checklist.append(OptCheck('GCC_PLUGINS',                 'y', 'defconfig', 'self_protection'))
291     checklist.append(OR(OptCheck('REFCOUNT_FULL',            'y', 'defconfig', 'self_protection'), \
292                         VerCheck((5, 5)))) # REFCOUNT_FULL is enabled by default since v5.5
293     iommu_support_is_set = OptCheck('IOMMU_SUPPORT',         'y', 'defconfig', 'self_protection') # is needed for mitigating DMA attacks
294     checklist.append(iommu_support_is_set)
295     if arch == 'X86_64' or arch == 'X86_32':
296         checklist.append(OptCheck('MICROCODE',                   'y', 'defconfig', 'self_protection')) # is needed for mitigating CPU bugs
297         checklist.append(OptCheck('RETPOLINE',                   'y', 'defconfig', 'self_protection'))
298         checklist.append(OptCheck('X86_SMAP',                    'y', 'defconfig', 'self_protection'))
299         checklist.append(OR(OptCheck('X86_UMIP',                 'y', 'defconfig', 'self_protection'), \
300                             OptCheck('X86_INTEL_UMIP',           'y', 'defconfig', 'self_protection')))
301         checklist.append(OptCheck('SYN_COOKIES',                 'y', 'defconfig', 'self_protection')) # another reason?
302     if arch == 'X86_64':
303         checklist.append(OptCheck('PAGE_TABLE_ISOLATION',        'y', 'defconfig', 'self_protection'))
304         checklist.append(OptCheck('RANDOMIZE_MEMORY',            'y', 'defconfig', 'self_protection'))
305         checklist.append(AND(OptCheck('INTEL_IOMMU',             'y', 'defconfig', 'self_protection'), \
306                              iommu_support_is_set))
307         checklist.append(AND(OptCheck('AMD_IOMMU',               'y', 'defconfig', 'self_protection'), \
308                              iommu_support_is_set))
309     if arch == 'ARM64':
310         checklist.append(OptCheck('UNMAP_KERNEL_AT_EL0',         'y', 'defconfig', 'self_protection'))
311         checklist.append(OptCheck('HARDEN_EL2_VECTORS',          'y', 'defconfig', 'self_protection'))
312         checklist.append(OptCheck('RODATA_FULL_DEFAULT_ENABLED', 'y', 'defconfig', 'self_protection'))
313     if arch == 'X86_64' or arch == 'ARM64':
314         checklist.append(OptCheck('VMAP_STACK',                  'y', 'defconfig', 'self_protection'))
315     if arch == 'X86_64' or arch == 'ARM64' or arch == 'X86_32':
316         checklist.append(OptCheck('RANDOMIZE_BASE',              'y', 'defconfig', 'self_protection'))
317         checklist.append(OptCheck('THREAD_INFO_IN_TASK',         'y', 'defconfig', 'self_protection'))
318     if arch == 'ARM':
319         checklist.append(OptCheck('CPU_SW_DOMAIN_PAN',           'y', 'defconfig', 'self_protection'))
320         checklist.append(OptCheck('STACKPROTECTOR_PER_TASK',     'y', 'defconfig', 'self_protection'))
321     if arch == 'ARM64' or arch == 'ARM':
322         checklist.append(OptCheck('HARDEN_BRANCH_PREDICTOR',     'y', 'defconfig', 'self_protection'))
323
324     checklist.append(OptCheck('BUG_ON_DATA_CORRUPTION',           'y', 'kspp', 'self_protection'))
325     checklist.append(OptCheck('DEBUG_WX',                         'y', 'kspp', 'self_protection'))
326     checklist.append(OptCheck('SCHED_STACK_END_CHECK',            'y', 'kspp', 'self_protection'))
327     checklist.append(OptCheck('SLAB_FREELIST_HARDENED',           'y', 'kspp', 'self_protection'))
328     checklist.append(OptCheck('SLAB_FREELIST_RANDOM',             'y', 'kspp', 'self_protection'))
329     checklist.append(OptCheck('SHUFFLE_PAGE_ALLOCATOR',           'y', 'kspp', 'self_protection'))
330     checklist.append(OptCheck('FORTIFY_SOURCE',                   'y', 'kspp', 'self_protection'))
331     randstruct_is_set = OptCheck('GCC_PLUGIN_RANDSTRUCT',         'y', 'kspp', 'self_protection')
332     checklist.append(randstruct_is_set)
333     checklist.append(OptCheck('GCC_PLUGIN_LATENT_ENTROPY',        'y', 'kspp', 'self_protection'))
334     checklist.append(OptCheck('DEBUG_LIST',                       'y', 'kspp', 'self_protection'))
335     checklist.append(OptCheck('DEBUG_SG',                         'y', 'kspp', 'self_protection'))
336     checklist.append(OptCheck('DEBUG_CREDENTIALS',                'y', 'kspp', 'self_protection'))
337     checklist.append(OptCheck('DEBUG_NOTIFIERS',                  'y', 'kspp', 'self_protection'))
338     hardened_usercopy_is_set = OptCheck('HARDENED_USERCOPY',      'y', 'kspp', 'self_protection')
339     checklist.append(hardened_usercopy_is_set)
340     checklist.append(AND(OptCheck('HARDENED_USERCOPY_FALLBACK',   'is not set', 'kspp', 'self_protection'), \
341                          hardened_usercopy_is_set))
342     checklist.append(OR(OptCheck('MODULE_SIG',                    'y', 'kspp', 'self_protection'), \
343                         modules_not_set))
344     checklist.append(OR(OptCheck('MODULE_SIG_ALL',                'y', 'kspp', 'self_protection'), \
345                         modules_not_set))
346     checklist.append(OR(OptCheck('MODULE_SIG_SHA512',             'y', 'kspp', 'self_protection'), \
347                         modules_not_set))
348     checklist.append(OR(OptCheck('MODULE_SIG_FORCE',              'y', 'kspp', 'self_protection'), \
349                         modules_not_set)) # refers to LOCKDOWN
350     checklist.append(OR(OptCheck('INIT_STACK_ALL',                'y', 'kspp', 'self_protection'), \
351                       OptCheck('GCC_PLUGIN_STRUCTLEAK_BYREF_ALL', 'y', 'kspp', 'self_protection')))
352     checklist.append(OptCheck('INIT_ON_ALLOC_DEFAULT_ON',         'y', 'kspp', 'self_protection'))
353     checklist.append(OR(OptCheck('INIT_ON_FREE_DEFAULT_ON',       'y', 'kspp', 'self_protection'), \
354                         OptCheck('PAGE_POISONING',                'y', 'kspp', 'self_protection'))) # before v5.3
355     if arch == 'X86_64' or arch == 'ARM64' or arch == 'X86_32':
356         stackleak_is_set = OptCheck('GCC_PLUGIN_STACKLEAK',       'y', 'kspp', 'self_protection')
357         checklist.append(stackleak_is_set)
358         checklist.append(AND(OptCheck('STACKLEAK_METRICS',         'is not set', 'clipos', 'self_protection'), \
359                              stackleak_is_set))
360         checklist.append(AND(OptCheck('STACKLEAK_RUNTIME_DISABLE', 'is not set', 'clipos', 'self_protection'), \
361                              stackleak_is_set))
362     if arch == 'X86_64' or arch == 'X86_32':
363         checklist.append(OptCheck('DEFAULT_MMAP_MIN_ADDR',            '65536', 'kspp', 'self_protection'))
364     if arch == 'X86_32':
365         checklist.append(OptCheck('PAGE_TABLE_ISOLATION',             'y', 'kspp', 'self_protection'))
366         checklist.append(OptCheck('HIGHMEM64G',                       'y', 'kspp', 'self_protection'))
367         checklist.append(OptCheck('X86_PAE',                          'y', 'kspp', 'self_protection'))
368     if arch == 'ARM64':
369         checklist.append(OptCheck('ARM64_SW_TTBR0_PAN',               'y', 'kspp', 'self_protection'))
370     if arch == 'ARM64' or arch == 'ARM':
371         checklist.append(OptCheck('SYN_COOKIES',                      'y', 'kspp', 'self_protection')) # another reason?
372         checklist.append(OptCheck('DEFAULT_MMAP_MIN_ADDR',            '32768', 'kspp', 'self_protection'))
373
374     checklist.append(OptCheck('SECURITY_DMESG_RESTRICT',               'y', 'clipos', 'self_protection'))
375     checklist.append(OptCheck('DEBUG_VIRTUAL',                         'y', 'clipos', 'self_protection'))
376     checklist.append(OptCheck('STATIC_USERMODEHELPER',                 'y', 'clipos', 'self_protection')) # needs userspace support (systemd)
377     checklist.append(OptCheck('SLAB_MERGE_DEFAULT',                    'is not set', 'clipos', 'self_protection')) # slab_nomerge
378     checklist.append(AND(OptCheck('GCC_PLUGIN_RANDSTRUCT_PERFORMANCE', 'is not set', 'clipos', 'self_protection'), \
379                          randstruct_is_set))
380     checklist.append(OptCheck('RANDOM_TRUST_BOOTLOADER',               'is not set', 'clipos', 'self_protection'))
381     checklist.append(OptCheck('RANDOM_TRUST_CPU',                      'is not set', 'clipos', 'self_protection'))
382     if arch == 'X86_64' or arch == 'X86_32':
383         checklist.append(AND(OptCheck('INTEL_IOMMU_SVM',                   'y', 'clipos', 'self_protection'), \
384                              iommu_support_is_set))
385         checklist.append(AND(OptCheck('INTEL_IOMMU_DEFAULT_ON',            'y', 'clipos', 'self_protection'), \
386                              iommu_support_is_set))
387     if arch == 'X86_32':
388         checklist.append(AND(OptCheck('INTEL_IOMMU',                       'y', 'clipos', 'self_protection'), \
389                              iommu_support_is_set))
390
391     checklist.append(OptCheck('SLUB_DEBUG_ON',                      'y', 'my', 'self_protection'))
392     checklist.append(OptCheck('RESET_ATTACK_MITIGATION',            'y', 'my', 'self_protection')) # needs userspace support (systemd)
393     if arch == 'X86_64':
394         checklist.append(AND(OptCheck('AMD_IOMMU_V2',                   'y', 'my', 'self_protection'), \
395                              iommu_support_is_set))
396
397     if arch == 'X86_64' or arch == 'ARM64' or arch == 'X86_32':
398         checklist.append(OptCheck('SECURITY',                               'y', 'defconfig', 'security_policy')) # and choose your favourite LSM
399     if arch == 'ARM':
400         checklist.append(OptCheck('SECURITY',                               'y', 'kspp', 'security_policy')) # and choose your favourite LSM
401     checklist.append(OptCheck('SECURITY_YAMA',                          'y', 'kspp', 'security_policy'))
402     checklist.append(OR(OptCheck('SECURITY_WRITABLE_HOOKS',             'is not set', 'my', 'security_policy'), \
403                         OptCheck('SECURITY_SELINUX_DISABLE',            'is not set', 'kspp', 'security_policy')))
404     checklist.append(OptCheck('SECURITY_LOCKDOWN_LSM',                  'y', 'clipos', 'security_policy'))
405     checklist.append(OptCheck('SECURITY_LOCKDOWN_LSM_EARLY',            'y', 'clipos', 'security_policy'))
406     checklist.append(OptCheck('LOCK_DOWN_KERNEL_FORCE_CONFIDENTIALITY', 'y', 'clipos', 'security_policy'))
407     loadpin_is_set = OptCheck('SECURITY_LOADPIN',                       'y', 'my', 'security_policy') # needs userspace support
408     checklist.append(loadpin_is_set)
409     checklist.append(AND(OptCheck('SECURITY_LOADPIN_ENFORCE',           'y', 'my', 'security_policy'), \
410                          loadpin_is_set))
411     checklist.append(OptCheck('SECURITY_SAFESETID',                     'y', 'my', 'security_policy'))
412
413     checklist.append(OptCheck('SECCOMP',              'y', 'defconfig', 'cut_attack_surface'))
414     checklist.append(OptCheck('SECCOMP_FILTER',       'y', 'defconfig', 'cut_attack_surface'))
415     if arch == 'X86_64' or arch == 'ARM64' or arch == 'X86_32':
416         checklist.append(OR(OptCheck('STRICT_DEVMEM',     'y', 'defconfig', 'cut_attack_surface'), \
417                             devmem_not_set)) # refers to LOCKDOWN
418
419     checklist.append(modules_not_set)
420     checklist.append(devmem_not_set)
421     checklist.append(OR(OptCheck('IO_STRICT_DEVMEM',  'y', 'kspp', 'cut_attack_surface'), \
422                         devmem_not_set)) # refers to LOCKDOWN
423     if arch == 'ARM':
424         checklist.append(OR(OptCheck('STRICT_DEVMEM',     'y', 'kspp', 'cut_attack_surface'), \
425                             devmem_not_set)) # refers to LOCKDOWN
426     if arch == 'X86_64':
427         checklist.append(OptCheck('LEGACY_VSYSCALL_NONE', 'y', 'kspp', 'cut_attack_surface')) # 'vsyscall=none'
428     checklist.append(OptCheck('ACPI_CUSTOM_METHOD',   'is not set', 'kspp', 'cut_attack_surface')) # refers to LOCKDOWN
429     checklist.append(OptCheck('COMPAT_BRK',           'is not set', 'kspp', 'cut_attack_surface'))
430     checklist.append(OptCheck('DEVKMEM',              'is not set', 'kspp', 'cut_attack_surface')) # refers to LOCKDOWN
431     checklist.append(OptCheck('COMPAT_VDSO',          'is not set', 'kspp', 'cut_attack_surface'))
432     checklist.append(OptCheck('BINFMT_MISC',          'is not set', 'kspp', 'cut_attack_surface'))
433     checklist.append(OptCheck('INET_DIAG',            'is not set', 'kspp', 'cut_attack_surface'))
434     checklist.append(OptCheck('KEXEC',                'is not set', 'kspp', 'cut_attack_surface')) # refers to LOCKDOWN
435     checklist.append(OptCheck('PROC_KCORE',           'is not set', 'kspp', 'cut_attack_surface')) # refers to LOCKDOWN
436     checklist.append(OptCheck('LEGACY_PTYS',          'is not set', 'kspp', 'cut_attack_surface'))
437     checklist.append(OptCheck('HIBERNATION',          'is not set', 'kspp', 'cut_attack_surface')) # refers to LOCKDOWN
438     checklist.append(OptCheck('IA32_EMULATION',       'is not set', 'kspp', 'cut_attack_surface'))
439     checklist.append(OptCheck('X86_X32',              'is not set', 'kspp', 'cut_attack_surface'))
440     checklist.append(OptCheck('MODIFY_LDT_SYSCALL',   'is not set', 'kspp', 'cut_attack_surface'))
441     checklist.append(OptCheck('OABI_COMPAT',          'is not set', 'kspp', 'cut_attack_surface'))
442
443     checklist.append(OptCheck('X86_PTDUMP',              'is not set', 'grsecurity', 'cut_attack_surface'))
444     checklist.append(OptCheck('ZSMALLOC_STAT',           'is not set', 'grsecurity', 'cut_attack_surface'))
445     checklist.append(OptCheck('PAGE_OWNER',              'is not set', 'grsecurity', 'cut_attack_surface'))
446     checklist.append(OptCheck('DEBUG_KMEMLEAK',          'is not set', 'grsecurity', 'cut_attack_surface'))
447     checklist.append(OptCheck('BINFMT_AOUT',             'is not set', 'grsecurity', 'cut_attack_surface'))
448     checklist.append(OptCheck('KPROBES',                 'is not set', 'grsecurity', 'cut_attack_surface')) # refers to LOCKDOWN
449     checklist.append(OptCheck('UPROBES',                 'is not set', 'grsecurity', 'cut_attack_surface'))
450     checklist.append(OptCheck('GENERIC_TRACER',          'is not set', 'grsecurity', 'cut_attack_surface')) # refers to LOCKDOWN
451     checklist.append(OptCheck('PROC_VMCORE',             'is not set', 'grsecurity', 'cut_attack_surface'))
452     checklist.append(OptCheck('PROC_PAGE_MONITOR',       'is not set', 'grsecurity', 'cut_attack_surface'))
453     checklist.append(OptCheck('USELIB',                  'is not set', 'grsecurity', 'cut_attack_surface'))
454     checklist.append(OptCheck('CHECKPOINT_RESTORE',      'is not set', 'grsecurity', 'cut_attack_surface'))
455     checklist.append(OptCheck('USERFAULTFD',             'is not set', 'grsecurity', 'cut_attack_surface'))
456     checklist.append(OptCheck('HWPOISON_INJECT',         'is not set', 'grsecurity', 'cut_attack_surface'))
457     checklist.append(OptCheck('MEM_SOFT_DIRTY',          'is not set', 'grsecurity', 'cut_attack_surface'))
458     checklist.append(OptCheck('DEVPORT',                 'is not set', 'grsecurity', 'cut_attack_surface')) # refers to LOCKDOWN
459     checklist.append(OptCheck('DEBUG_FS',                'is not set', 'grsecurity', 'cut_attack_surface')) # refers to LOCKDOWN
460     checklist.append(OptCheck('NOTIFIER_ERROR_INJECTION','is not set', 'grsecurity', 'cut_attack_surface'))
461
462     checklist.append(OptCheck('DRM_LEGACY',     'is not set', 'maintainer', 'cut_attack_surface'))
463     checklist.append(OptCheck('FB',             'is not set', 'maintainer', 'cut_attack_surface'))
464     checklist.append(OptCheck('VT',             'is not set', 'maintainer', 'cut_attack_surface'))
465
466     checklist.append(OptCheck('ACPI_TABLE_UPGRADE',   'is not set', 'lockdown', 'cut_attack_surface')) # refers to LOCKDOWN
467     checklist.append(OptCheck('X86_IOPL_IOPERM',      'is not set', 'lockdown', 'cut_attack_surface')) # refers to LOCKDOWN
468     checklist.append(OptCheck('EFI_TEST',             'is not set', 'lockdown', 'cut_attack_surface')) # refers to LOCKDOWN
469     checklist.append(OptCheck('BPF_SYSCALL',          'is not set', 'lockdown', 'cut_attack_surface')) # refers to LOCKDOWN
470     checklist.append(OptCheck('MMIOTRACE_TEST',       'is not set', 'lockdown', 'cut_attack_surface')) # refers to LOCKDOWN
471
472     if arch == 'X86_64' or arch == 'X86_32':
473         checklist.append(OptCheck('X86_INTEL_TSX_MODE_OFF',   'y', 'clipos', 'cut_attack_surface')) # tsx=off
474     checklist.append(OptCheck('STAGING',                  'is not set', 'clipos', 'cut_attack_surface'))
475     checklist.append(OptCheck('KSM',                      'is not set', 'clipos', 'cut_attack_surface')) # to prevent FLUSH+RELOAD attack
476 #   checklist.append(OptCheck('IKCONFIG',                 'is not set', 'clipos', 'cut_attack_surface')) # no, this info is needed for this check :)
477     checklist.append(OptCheck('KALLSYMS',                 'is not set', 'clipos', 'cut_attack_surface'))
478     checklist.append(OptCheck('X86_VSYSCALL_EMULATION',   'is not set', 'clipos', 'cut_attack_surface'))
479     checklist.append(OptCheck('MAGIC_SYSRQ',              'is not set', 'clipos', 'cut_attack_surface'))
480     checklist.append(OptCheck('KEXEC_FILE',               'is not set', 'clipos', 'cut_attack_surface')) # refers to LOCKDOWN (permissive)
481     checklist.append(OptCheck('USER_NS',                  'is not set', 'clipos', 'cut_attack_surface')) # user.max_user_namespaces=0
482     checklist.append(OptCheck('X86_MSR',                  'is not set', 'clipos', 'cut_attack_surface')) # refers to LOCKDOWN
483     checklist.append(OptCheck('X86_CPUID',                'is not set', 'clipos', 'cut_attack_surface'))
484     checklist.append(AND(OptCheck('LDISC_AUTOLOAD',           'is not set', 'clipos', 'cut_attack_surface'), \
485                          PresenceCheck('LDISC_AUTOLOAD')))
486
487     checklist.append(OptCheck('AIO',                  'is not set', 'grapheneos', 'cut_attack_surface'))
488
489     checklist.append(OptCheck('MMIOTRACE',            'is not set', 'my', 'cut_attack_surface')) # refers to LOCKDOWN (permissive)
490     checklist.append(OptCheck('LIVEPATCH',            'is not set', 'my', 'cut_attack_surface'))
491     checklist.append(OptCheck('IP_DCCP',              'is not set', 'my', 'cut_attack_surface'))
492     checklist.append(OptCheck('IP_SCTP',              'is not set', 'my', 'cut_attack_surface'))
493     checklist.append(OptCheck('FTRACE',               'is not set', 'my', 'cut_attack_surface')) # refers to LOCKDOWN
494     checklist.append(OptCheck('BPF_JIT',              'is not set', 'my', 'cut_attack_surface'))
495     checklist.append(OptCheck('VIDEO_VIVID',          'is not set', 'my', 'cut_attack_surface'))
496
497     checklist.append(OptCheck('INTEGRITY',       'y', 'defconfig', 'userspace_hardening'))
498     if arch == 'ARM64':
499         checklist.append(OptCheck('ARM64_PTR_AUTH',       'y', 'defconfig', 'userspace_hardening'))
500     if arch == 'ARM' or  arch == 'X86_32':
501         checklist.append(OptCheck('VMSPLIT_3G',           'y', 'defconfig', 'userspace_hardening'))
502     if arch == 'X86_64' or arch == 'ARM64':
503         checklist.append(OptCheck('ARCH_MMAP_RND_BITS',   '32', 'clipos', 'userspace_hardening'))
504     if arch == 'X86_32' or arch == 'ARM':
505         checklist.append(OptCheck('ARCH_MMAP_RND_BITS',   '16', 'my', 'userspace_hardening'))
506
507 #   checklist.append(OptCheck('LKDTM',    'm', 'my', 'feature_test'))
508
509
510 def print_checklist(checklist, with_results):
511     if json_mode:
512         opts = []
513         for o in checklist:
514             opt = ['CONFIG_'+o.name, o.expected, o.decision, o.reason]
515             if with_results:
516                 opt.append(o.result)
517             opts.append(opt)
518         print(json.dumps(opts))
519         return
520
521     # table header
522     sep_line_len = 91
523     if with_results:
524         sep_line_len += 30
525     print('=' * sep_line_len)
526     print('{:^45}|{:^13}|{:^10}|{:^20}'.format('option name', 'desired val', 'decision', 'reason'), end='')
527     if with_results:
528         print('|   {}'.format('check result'), end='')
529     print()
530     print('=' * sep_line_len)
531
532     # table contents
533     for opt in checklist:
534         opt.table_print(with_results)
535         print()
536         if debug_mode:
537             print('-' * sep_line_len)
538     print()
539
540
541 def perform_checks(checklist, parsed_options):
542     for opt in checklist:
543         if hasattr(opt, 'opts'):
544             # prepare ComplexOptCheck
545             for o in opt.opts:
546                 if hasattr(o, 'state'):
547                     o.state = parsed_options.get(o.name, None)
548         else:
549             # prepare simple check
550             if not hasattr(opt, 'state'):
551                 sys.exit('[!] ERROR: bad simple check {}'.format(vars(opt)))
552             opt.state = parsed_options.get(opt.name, None)
553         opt.check()
554
555
556 def check_config_file(checklist, fname, arch):
557     with open(fname, 'r') as f:
558         parsed_options = OrderedDict()
559         opt_is_on = re.compile("CONFIG_[a-zA-Z0-9_]*=[a-zA-Z0-9_\"]*")
560         opt_is_off = re.compile("# CONFIG_[a-zA-Z0-9_]* is not set")
561
562         if not json_mode:
563             print('[+] Checking "{}" against {} hardening preferences...'.format(fname, arch))
564         for line in f.readlines():
565             line = line.strip()
566             option = None
567             value = None
568
569             if opt_is_on.match(line):
570                 option, value = line[7:].split('=', 1)
571             elif opt_is_off.match(line):
572                 option, value = line[9:].split(' ', 1)
573                 if value != 'is not set':
574                     sys.exit('[!] ERROR: bad disabled config option "{}"'.format(line))
575
576             if option in parsed_options:
577                 sys.exit('[!] ERROR: config option "{}" exists multiple times'.format(line))
578
579             if option is not None:
580                 parsed_options[option] = value
581
582         perform_checks(checklist, parsed_options)
583
584         if debug_mode:
585             known_options = []
586             for opt in checklist:
587                 if hasattr(opt, 'opts'):
588                     for o in opt.opts:
589                         if hasattr(o, 'name'):
590                             known_options.append(o.name)
591                 else:
592                     known_options.append(opt.name)
593             for option, value in parsed_options.items():
594                 if option not in known_options:
595                     print('DEBUG: dunno about option {} ({})'.format(option, value))
596
597         print_checklist(checklist, True)
598
599 def main():
600     global debug_mode
601     global json_mode
602     global kernel_version
603
604     config_checklist = []
605
606     parser = ArgumentParser(description='Checks the hardening options in the Linux kernel config')
607     parser.add_argument('-p', '--print', choices=supported_archs,
608                         help='print hardening preferences for selected architecture')
609     parser.add_argument('-c', '--config',
610                         help='check the config_file against these preferences')
611     parser.add_argument('--debug', action='store_true',
612                         help='enable verbose debug mode')
613     parser.add_argument('--json', action='store_true',
614                         help='print results in JSON format')
615     args = parser.parse_args()
616
617     if args.debug:
618         debug_mode = True
619         print('[!] WARNING: debug mode is enabled')
620     if args.json:
621         json_mode = True
622     if debug_mode and json_mode:
623         sys.exit('[!] ERROR: options --debug and --json cannot be used simultaneously')
624
625     if args.config:
626         arch, msg = detect_arch(args.config)
627         if not arch:
628             sys.exit('[!] ERROR: {}'.format(msg))
629         elif not json_mode:
630             print('[+] Detected architecture: {}'.format(arch))
631
632         kernel_version, msg = detect_version(args.config)
633         if not kernel_version:
634             sys.exit('[!] ERROR: {}'.format(msg))
635         elif not json_mode:
636             print('[+] Detected kernel version: {}.{}'.format(kernel_version[0], kernel_version[1]))
637
638         construct_checklist(config_checklist, arch)
639         check_config_file(config_checklist, args.config, arch)
640         error_count = len(list(filter(lambda opt: opt.result.startswith('FAIL'), config_checklist)))
641         ok_count = len(list(filter(lambda opt: opt.result.startswith('OK'), config_checklist)))
642         if not debug_mode and not json_mode:
643             print('[+] config check is finished: \'OK\' - {} / \'FAIL\' - {}'.format(ok_count, error_count))
644         sys.exit(0)
645
646     if args.print:
647         arch = args.print
648         construct_checklist(config_checklist, arch)
649         if not json_mode:
650             print('[+] Printing kernel hardening preferences for {}...'.format(arch))
651         print_checklist(config_checklist, False)
652         sys.exit(0)
653
654     parser.print_help()
655     sys.exit(0)
656
657 if __name__ == '__main__':
658     main()