Merge pull request #684 from AkikoOrenji/AkikoOrenji-nuclei.py-fixes-1

Update nuclei.py for usability and improve error handling

bbot/modules/deadly/nuclei.py

@@ -66,7 +66,7 @@ async def setup(self):
                 self.warning(f"Failure while updating nuclei templates: {update_results.stderr}")
         else:
             self.warning("Error running nuclei template update command")
-
+        self.proxy = self.scan.config.get("http_proxy", "")
         self.mode = self.config.get("mode", "severe")
         self.ratelimit = int(self.config.get("ratelimit", 150))
         self.concurrency = int(self.config.get("concurrency", 25))
@@ -135,6 +135,9 @@ async def handle_batch(self, *events):
             cleaned_host = temp_target.get(host)
             source_event = self.correlate_event(events, cleaned_host)
 
+            if not source_event:
+                continue
+
             if url == "":
                 url = str(source_event.data)
 
@@ -204,6 +207,10 @@ async def execute_nuclei(self, nuclei_input):
             command.append("-t")
             command.append(self.budget_templates_file)
 
+        if self.proxy:
+            command.append("-proxy")
+            command.append(f"{self.proxy}")
+
         stats_file = self.helpers.tempfile_tail(callback=self.log_nuclei_status)
         try:
             with open(stats_file, "w") as stats_fh: