From: jvoisin <julien.voisin@dustri.org>
Date: Sun, 12 May 2024 14:59:42 +0000 (+0200)
Subject: Add a check to `_open`
X-Git-Url: https://jxself.org/git/?a=commitdiff_plain;h=771ea4774d92d62c70c5e35657abc3e297d0551d;p=kconfig-hardened-check.git

Add a check to `_open`

This shall transform ugly stacktraces into aesthetically pleasant error
messages.
---

diff --git a/kernel_hardening_checker/__init__.py b/kernel_hardening_checker/__init__.py
index 161fc94..286f420 100644
--- a/kernel_hardening_checker/__init__.py
+++ b/kernel_hardening_checker/__init__.py
@@ -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):