Add a check to `_open`
authorjvoisin <julien.voisin@dustri.org>
Sun, 12 May 2024 14:59:42 +0000 (16:59 +0200)
committerjvoisin <julien.voisin@dustri.org>
Sun, 12 May 2024 14:59:42 +0000 (16:59 +0200)
This shall transform ugly stacktraces into aesthetically pleasant error
messages.

kernel_hardening_checker/__init__.py

index 161fc9411fe1c79fc07555c0854cc60589c137a7..286f420ea5c016f72529647e44896223f65695de 100644 (file)
@@ -26,7 +26,11 @@ def _open(file: str, *args, **kwargs):
     if file.endswith('.gz'):
         open_method = gzip.open
 
-    return open_method(file, *args, **kwargs)
+    try:
+        return open_method(file, *args, **kwargs)
+    except FileNotFoundError:
+        sys.exit(f'[!] ERROR: unable to open {file}, are you sure it exists?')
+
 
 
 def detect_arch(fname, archs):