Merge pull request #12 from CybercentreCanada/hotfix/retry_on_retryfail

raise Recoverable on RetryError
CybercentreCanada · Jul 14, 2021 · 283d254 · 283d254
2 parents 7864065 + 3322813
commit 283d254
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):