From 6510a09c63bfdca3e231ac4356b517578703af6b Mon Sep 17 00:00:00 2001
From: Jing Ling <shmilylty@gmail.com>
Date: Wed, 2 Sep 2020 00:12:31 +0800
Subject: [PATCH] Fixed #139

---
 brute.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/brute.py b/brute.py
index a9450e4..04a7040 100644
--- a/brute.py
+++ b/brute.py
@@ -304,6 +304,7 @@ def collect_wildcard_record(domain, authoritative_ns):
     ips = set()
     ttl = int()
     ips_stat = dict()
+    ips_check = list()
     while True:
         token = secrets.token_hex(4)
         random_subdomain = f'{token}.{domain}'
@@ -314,7 +315,15 @@ def collect_wildcard_record(domain, authoritative_ns):
             logger.log('ALERT', f'Multiple query errors,'
                                 f'try to query a new random subdomain')
             continue
-        if ip is None:  # TODO 当权威服务器查询出的结果只有cname没有ip结果时会死循环
+        # 每5次查询检查结果列表 如果都没结果则结束查询
+        ips_check.append(ip)
+        if len(ips_check) == 5:
+            if not any(ips_check):
+                logger.log('ALERT', 'The query ends because there are '
+                                    'no results for 5 consecutive queries.')
+                break
+            ips_check = list()
+        if ip is None:
             continue
         ips.update(ip)
         # 统计每个泛解析IP出现次数