sfp_tool_trufflehog: Enforce a timeout to prevent being blocked.

n3mas1s · Feb 7, 2023 · d8b02b1 · d8b02b1
1 parent 38a417e
commit d8b02b1
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/modules/sfp_tool_trufflehog.py b/modules/sfp_tool_trufflehog.py
@@ -14,7 +14,7 @@
 import sys
 import json
 import os
-from subprocess import PIPE, Popen
+from subprocess import PIPE, Popen, TimeoutExpired
 
 from spiderfoot import SpiderFootPlugin, SpiderFootEvent
 
@@ -135,8 +135,13 @@ def handleEvent(self, event):
         args.append(url)
         try:
             p = Popen(args, stdout=PIPE, stderr=PIPE)
-            out, _ = p.communicate(input=None)
+            out, _ = p.communicate(input=None, timeout=600)
             stdout = out.decode(sys.stdin.encoding)
+        except TimeoutExpired:
+            p.kill()
+            stdout, stderr = p.communicate()
+            self.debug(f"Timed out waiting for trufflehog to finish on {url}")
+            return
         except Exception as e:
             self.error(f"Unable to run trufflehog: {e}")
             return