Merge pull request #498 from blacklanternsecurity/async-optimize

missing await
blacklanternsecurity · Sep 30, 2024 · 6c8762c · 6c8762c
2 parents d378e95 + a0332c4
commit 6c8762c
Show file tree

Hide file tree

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

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