sfp_shodan: Set error state when reaching API limits.

n3mas1s · Nov 2, 2022 · ab1362d · ab1362d
1 parent 3b8bdbb
commit ab1362d
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/modules/sfp_shodan.py b/modules/sfp_shodan.py
@@ -96,6 +96,11 @@ def queryHost(self, qry):
         )
         time.sleep(1)
 
+        if res['code'] in ["403", "401"]:
+            self.error("SHODAN API key seems to have been rejected or you have exceeded usage limits.")
+            self.errorState = True
+            return None
+
         if res['content'] is None:
             self.info(f"No SHODAN info found for {qry}")
             return None
@@ -124,6 +129,12 @@ def searchHosts(self, qry):
             useragent="SpiderFoot"
         )
         time.sleep(1)
+
+        if res['code'] in ["403", "401"]:
+            self.error("SHODAN API key seems to have been rejected or you have exceeded usage limits.")
+            self.errorState = True
+            return None
+
         if res['content'] is None:
             self.info(f"No SHODAN info found for {qry}")
             return None
@@ -152,6 +163,12 @@ def searchHtml(self, qry):
             useragent="SpiderFoot"
         )
         time.sleep(1)
+
+        if res['code'] in ["403", "401"]:
+            self.error("SHODAN API key seems to have been rejected or you have exceeded usage limits.")
+            self.errorState = True
+            return None
+
         if res['content'] is None:
             self.info(f"No SHODAN info found for {qry}")
             return None