Fixed Error Handling For Shodan & Vulner

scanner.py

@@ -86,13 +86,18 @@ def port_discovery(self, url):
                     read = ports.readlines()
                     return read
             s.close()"""
-        SKey = self.params['s_key']
-        SApi = shodan.Shodan(SKey)
+        try:
+            SKey = self.params['s_key']
+            SApi = shodan.Shodan(SKey)
 
-        SInfo = SApi.host(ip)
-        ports = json.dumps(SInfo['ports'])
+            SInfo = SApi.host(ip)
+            ports = json.dumps(SInfo['ports'])
 
-        return ports
+            return ports
+        except shodan.exception.APITimeout:
+            return "Time Out! Something Went Wrong!"
+        except shodan.exception.APIError:
+            return "Shodan Key Not Authorized"
 
     def dns_record(self, url):
         domain = urlparse(url).netloc
@@ -203,17 +208,22 @@ def exploit_info(self, url):
         if server == '':
             return "No Server Version Found! Or Type!"
         else:
-            VKey = self.params["v_key"]  # get api key by going to https://vulners.com
-            VApi = vulners.Vulners(api_key=VKey)
-
-            search = VApi.searchExploit(server)
-            search = json.dumps(search, indent=2)
-
-            os.system(f"touch output_exploit_search.json")
-            with open(f"output_exploit_search.json", "w") as exploitResult:
-                exploitResult.writelines(search)
-
-            return f"Visit /output"
+            try:
+                VKey = self.params["v_key"]  # get api key by going to https://vulners.com
+                VApi = vulners.Vulners(api_key=VKey)
+
+                search = VApi.searchExploit(server)
+                search = json.dumps(search, indent=2)
+
+                os.system(f"touch output_exploit_search.json")
+                with open(f"output_exploit_search.json", "w") as exploitResult:
+                    exploitResult.writelines(search)
+
+                return f"Visit /output"
+            except requests.exceptions.HTTPError:
+                return "Something Went Wrong! But it works!"
+            except:
+                return "Something Went Wrong But it works!"
 
     def mitigation_info(self, url):
         payloadxss = "<script>document.write('xss');</script>"