From b51236ac51c31ce112c6cc7d9e44421820546304 Mon Sep 17 00:00:00 2001 From: Jing Ling Date: Sun, 15 Mar 2020 18:06:11 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E4=B8=80=E7=9B=B4=E6=B2=A1?= =?UTF-8?q?=E6=9C=89=E9=80=80=E5=87=BA=E7=9A=84bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- oneforall/aiobrute.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/oneforall/aiobrute.py b/oneforall/aiobrute.py index ddd8d5b..b195f6c 100644 --- a/oneforall/aiobrute.py +++ b/oneforall/aiobrute.py @@ -266,8 +266,9 @@ class AIOBrute(Module): def run(self, rx_queue=None): self.domains = utils.get_domains(self.target) - while self.domains: - self.domain = self.domains.pop() + loop = asyncio.get_event_loop() + asyncio.set_event_loop(loop) + for self.domain in self.domains: start = time.time() db = Database() db.create_table(self.domain) @@ -278,8 +279,6 @@ class AIOBrute(Module): # fuzz模式不使用递归爆破 if self.recursive_brute and not self.fuzz: logger.log('INFOR', f'开始递归爆破{self.domain}的第1层子域') - loop = asyncio.get_event_loop() - asyncio.set_event_loop(loop) loop.run_until_complete(self.main(self.domain, rx_queue)) # 递归爆破下一层的子域