修复解析没有结果的问题

This commit is contained in:
shmilylty
2019-10-26 17:58:31 +08:00
parent e66c6d7e21
commit f30f48aea2
3 changed files with 3 additions and 4 deletions
+1 -1
View File
@@ -148,7 +148,7 @@ def request_callback(future, index, datas):
try:
result = future.result()
except BaseException as e:
logger.log('DEBUG', e.args)
logger.log('TRACE', e.args)
datas[index]['reason'] = str(e.args)
datas[index]['valid'] = 0
else:
+1 -1
View File
@@ -28,7 +28,7 @@ async def dns_query_a(hostname):
"""
resolver = dns_resolver()
try:
answer = await resolver.query(hostname, 'A')
answer = resolver.query(hostname, 'A')
except BaseException as exception:
logger.log('TRACE', exception.args)
answer = exception
+1 -2
View File
@@ -50,8 +50,7 @@ class BruteSRV(Module):
tasks = []
for name in names:
tasks.append(self.query(name))
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
loop = asyncio.get_event_loop()
group = asyncio.gather(*tasks)
results = loop.run_until_complete(group)
for answer in results: