raise Recoverable on RetryError

CybercentreCanada · Jul 14, 2021 · 3322813 · 3322813
1 parent 7864065
commit 3322813
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/suricata_/suricata_.py b/suricata_/suricata_.py
@@ -10,7 +10,7 @@
 
 from io import StringIO
 from pathlib import Path
-from retrying import retry
+from retrying import retry, RetryError
 
 from assemblyline.common.digests import get_sha256_for_file
 from assemblyline.common.exceptions import RecoverableError
@@ -159,7 +159,10 @@ def reload_rules(self):
 
     def start_suricata_if_necessary(self):
         if not self.suricata_running():
-            self.launch_or_load_suricata()
+            try:
+                self.launch_or_load_suricata()
+            except RetryError as e:
+                raise RecoverableError(e)
 
     # Try connecting to the Suricata socket
     def suricata_running(self):