increase max dnscommonsrv handlers, small masscan bugfix

blacklanternsecurity · Feb 2, 2024 · 2692914 · 2692914
1 parent 4a55633
commit 2692914
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/bbot/modules/dnscommonsrv.py b/bbot/modules/dnscommonsrv.py
@@ -94,7 +94,7 @@ class dnscommonsrv(BaseModule):
     produced_events = ["DNS_NAME"]
     flags = ["subdomain-enum", "passive", "safe"]
     meta = {"description": "Check for common SRV records"}
-    _max_event_handlers = 5
+    _max_event_handlers = 10
 
     async def filter_event(self, event):
         # skip SRV wildcards

diff --git a/bbot/modules/masscan.py b/bbot/modules/masscan.py
@@ -241,9 +241,11 @@ async def emit_from_cache(self):
                 await self.emit_event(line, "OPEN_TCP_PORT", source=source_event)
 
     def get_source_event(self, host):
-        source_event = self.scan.whitelist.get(host)
+        source_event = self.scan.target.get(host)
         if source_event is None:
-            source_event = self.scan.root_event
+            source_event = self.scan.whitelist.get(host)
+            if source_event is None:
+                source_event = self.scan.root_event
         return source_event
 
     async def cleanup(self):