river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-08-26 12:57:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复一直没有退出的bug

Browse Source
This commit is contained in:
Jing Ling
2020-03-15 18:06:11 +08:00
parent 6868e32a84
commit b51236ac51
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
oneforall/aiobrute.py
+3 -4
View File
@@ -266,8 +266,9 @@ class AIOBrute(Module):
def run(self, rx_queue=None): def run(self, rx_queue=None):
self.domains = utils.get_domains(self.target) self.domains = utils.get_domains(self.target)
while self.domains: loop = asyncio.get_event_loop()
self.domain = self.domains.pop() asyncio.set_event_loop(loop)
for self.domain in self.domains:
start = time.time() start = time.time()
db = Database() db = Database()
db.create_table(self.domain) db.create_table(self.domain)
@@ -278,8 +279,6 @@ class AIOBrute(Module):
# fuzz模式不使用递归爆破 # fuzz模式不使用递归爆破
if self.recursive_brute and not self.fuzz: if self.recursive_brute and not self.fuzz:
logger.log('INFOR', f'开始递归爆破{self.domain}的第1层子域') 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)) loop.run_until_complete(self.main(self.domain, rx_queue))
# 递归爆破下一层的子域 # 递归爆破下一层的子域