mod_ssl: style

Author: devl00p

wapitiCore/attack/mod_ssl.py

@@ -234,8 +234,8 @@ def process_error(xml_file: str) -> str:
         if error:
             return error.text
         return ""
-    except Exception as exception:
-        return "Error parsing sslscan XML output" + str(exception)
+    except (ET.ParseError, FileNotFoundError, OSError, IOError) as exception:
+        return "Error parsing sslscan XML output:" + str(exception)
 
 
 async def process_vulnerabilities(xml_file: str) -> AsyncIterator[Tuple[int, str]]:
@@ -405,7 +405,12 @@ async def process_cert_features(self, target: str, port: int) -> AsyncIterator[T
     async def process_sslscan(self, hostname: str, port: int) -> AsyncIterator[Tuple[int, str]]:
         with tempfile.NamedTemporaryFile("r", suffix=".xml", delete=False) as temp_file:
             sslscan_command = ["sslscan", "--iana-names", "--ocsp", f"--xml={temp_file.name}", f"{hostname}:{port}"]
-            subprocess.run(sslscan_command, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+            try:
+                subprocess.run(sslscan_command, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL, check=True)
+            except subprocess.CalledProcessError as exception:
+                log_red("Error running sslscan: " + str(exception))
+                return
+
             error = process_error(temp_file.name)
             if error:
                 log_red(error)