修复问题15 https://github.com/shmilylty/OneForAll/issues/15

oneforall/aiobrute.py

@@ -211,6 +211,8 @@ class AIOBrute(Module):
 
     def deal_results(self, results):
         for result in results:
+            if result is None:
+                continue
             if isinstance(result, Exception):
                 # logger.log('DEBUG', f'爆破{subdomain}时出错 {str(answers)}')
                 continue

oneforall/common/resolve.py

@@ -51,12 +51,20 @@ async def aiodns_query_a(hostname, semaphore=None):
     """
     if semaphore is None:
         resolver = aiodns_resolver()
-        answers = await resolver.query(hostname, 'A')
+        try:
+            answers = await resolver.query(hostname, 'A')
+        except BaseException as e:
+            logger.log('DEBUG', e.args)
+            answers = None
         return hostname, answers
     else:
         async with semaphore:
             resolver = aiodns_resolver()
-            answers = await resolver.query(hostname, 'A')
+            try:
+                answers = await resolver.query(hostname, 'A')
+            except BaseException as e:
+                logger.log('DEBUG', e.args)
+                answers = None
             return hostname, answers