mirror of
https://github.com/shmilylty/OneForAll.git
synced 2026-08-26 12:57:50 +08:00
修复解析没有结果的问题
This commit is contained in:
@@ -148,7 +148,7 @@ def request_callback(future, index, datas):
|
|||||||
try:
|
try:
|
||||||
result = future.result()
|
result = future.result()
|
||||||
except BaseException as e:
|
except BaseException as e:
|
||||||
logger.log('DEBUG', e.args)
|
logger.log('TRACE', e.args)
|
||||||
datas[index]['reason'] = str(e.args)
|
datas[index]['reason'] = str(e.args)
|
||||||
datas[index]['valid'] = 0
|
datas[index]['valid'] = 0
|
||||||
else:
|
else:
|
||||||
|
|||||||
@@ -28,7 +28,7 @@ async def dns_query_a(hostname):
|
|||||||
"""
|
"""
|
||||||
resolver = dns_resolver()
|
resolver = dns_resolver()
|
||||||
try:
|
try:
|
||||||
answer = await resolver.query(hostname, 'A')
|
answer = resolver.query(hostname, 'A')
|
||||||
except BaseException as exception:
|
except BaseException as exception:
|
||||||
logger.log('TRACE', exception.args)
|
logger.log('TRACE', exception.args)
|
||||||
answer = exception
|
answer = exception
|
||||||
|
|||||||
@@ -50,8 +50,7 @@ class BruteSRV(Module):
|
|||||||
tasks = []
|
tasks = []
|
||||||
for name in names:
|
for name in names:
|
||||||
tasks.append(self.query(name))
|
tasks.append(self.query(name))
|
||||||
loop = asyncio.new_event_loop()
|
loop = asyncio.get_event_loop()
|
||||||
asyncio.set_event_loop(loop)
|
|
||||||
group = asyncio.gather(*tasks)
|
group = asyncio.gather(*tasks)
|
||||||
results = loop.run_until_complete(group)
|
results = loop.run_until_complete(group)
|
||||||
for answer in results:
|
for answer in results:
|
||||||
|
|||||||
Reference in New Issue
Block a user