Merge pull request #497 from blacklanternsecurity/async-optimize

adding cleanup within nested module runs
blacklanternsecurity · Sep 30, 2024 · d378e95 · d378e95
2 parents 9c4b115 + 80ec979
commit d378e95
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/baddns/modules/references.py b/baddns/modules/references.py
@@ -166,6 +166,7 @@ async def process_cname_analysis(self, parsed_results):
                         "direct_mode": direct_mode,
                     }
                     cname_findings.append(finding)
+                cname_instance.cleanup()
         return cname_findings
 
     async def dispatch(self):

diff --git a/baddns/modules/txt.py b/baddns/modules/txt.py
@@ -60,6 +60,7 @@ async def dispatch(self):
                             "trigger": self.target_dnsmanager.target,
                         }
                     )
+                cname_instance_direct.cleanup()
 
                 cname_instance = BadDNS_cname(
                     host,
@@ -78,6 +79,7 @@ async def dispatch(self):
                             "trigger": self.target_dnsmanager.target,
                         }
                     )
+                cname_instance.cleanup()
 
         return True