Now returns -1 when an uncaught exception is found.

jtesta · Jul 7, 2020 · f80e3f2 · f80e3f2
1 parent 49bd2c9
commit f80e3f2
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/ssh-audit.py b/ssh-audit.py
@@ -51,6 +51,7 @@
 PROGRAM_RETVAL_WARNING = 2
 PROGRAM_RETVAL_CONNECTION_ERROR = 1
 PROGRAM_RETVAL_GOOD = 0
+PROGRAM_RETVAL_UNKNOWN_ERROR = -1
 
 try:  # pragma: nocover
     from colorama import init as colorama_init
@@ -3516,5 +3517,12 @@ def main() -> int:
 
 
 if __name__ == '__main__':  # pragma: nocover
-    exit_code = main()
+    exit_code = PROGRAM_RETVAL_GOOD
+
+    try:
+        exit_code = main()
+    except Exception:
+        exit_code = PROGRAM_RETVAL_UNKNOWN_ERROR
+        print(traceback.format_exc())
+
     sys.exit(exit_code)