Merge pull request #749 from blacklanternsecurity/projectdiscovery-cu…

…stom-resolvers Reintroduce Custom DNS Resolvers for ProjectDiscovery Tools
blacklanternsecurity · Sep 22, 2023 · fba1e62 · fba1e62
2 parents 54fb4c3 + 02faebe
commit fba1e62
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/bbot/modules/deadly/nuclei.py b/bbot/modules/deadly/nuclei.py
@@ -185,10 +185,11 @@ async def execute_nuclei(self, nuclei_input):
             self.concurrency,
             "-disable-update-check",
             "-stats-json",
-            # "-r",
-            # self.helpers.resolver_file,
         ]
 
+        if self.helpers.system_resolvers:
+            command += ["-r", self.helpers.resolver_file]
+
         for cli_option in ("severity", "templates", "iserver", "itoken", "tags", "etags"):
             option = getattr(self, cli_option)
 

diff --git a/bbot/modules/httpx.py b/bbot/modules/httpx.py
@@ -98,9 +98,12 @@ async def handle_batch(self, *events):
             f"User-Agent: {self.scan.useragent}",
             "-response-size-to-read",
             f"{self.max_response_size}",
-            # "-r",
-            # self.helpers.resolver_file,
         ]
+
+        dns_resolvers = ",".join(self.helpers.system_resolvers)
+        if dns_resolvers:
+            command += ["-r", dns_resolvers]
+
         for hk, hv in self.scan.config.get("http_headers", {}).items():
             command += ["-header", f"{hk}: {hv}"]
         proxy = self.scan.config.get("http_proxy", "")