All url DNS use 8.8.8.8

dns_resolver.py

@@ -9,53 +9,48 @@ class DnsManager:
 
     def __new__(cls, *args, **kwargs):
         if not cls._instance:
+            # 使用 super() 调用 object 的 __new__ 来创建单例内存
             cls._instance = super(DnsManager, cls).__new__(cls)
         return cls._instance
 
     def __init__(self, cache_ttl=600):
-        # 避免单例模式下重复初始化
         if hasattr(self, '_initialized'):
             return
         self.cache_ttl = cache_ttl
-        self.cache = {}  # 格式: {"hostname": {"ip": "xxx", "expiry": 0}}
-        # 覆盖 Discord 机器人运行所需的所有关键域名
-        self.target_hosts = [
-            'discord.com', 
-            'gateway.discord.gg', 
-            'cdn.discordapp.com', 
-            'discordapp.com'
-        ]
+        self.cache = {}  # 缓存: {"hostname": {"ip": "xxx", "expiry": 0}}
         self.logger = logging.getLogger("DnsManager")
         self._initialized = True
 
     def _fetch_ip_from_http(self, hostname):
-        """通过 HTTPS (TCP 443) 绕过系统 UDP DNS 限制获取 IP"""
+        """核心：通过 HTTPS (TCP 443) 绕过系统 UDP DNS 限制获取 IP"""
         try:
-            # 使用 Google DNS API
+            # 注意：这里直接使用 Google DNS 的 API 地址
             url = f"https://dns.google/resolve?name={hostname}&type=A"
             req = urllib.request.Request(url, headers={"Accept": "application/dns-json"})
             with urllib.request.urlopen(req, timeout=5) as response:
                 data = json.loads(response.read().decode())
                 if "Answer" in data:
-                    # 找到第一个 A 记录 (IPv4)
                     for answer in data['Answer']:
-                        if answer['type'] == 1: # A record
+                        if answer['type'] == 1: # 只要 A 记录 (IPv4)
                             return answer['data']
         except Exception as e:
-            self.logger.warning(f"无法通过 HTTP 解析 {hostname}: {e}")
+            self.logger.warning(f"DoH 解析失败 {hostname}: {e}")
         return None
 
     def get_ip(self, hostname):
-        """获取 IP，优先使用缓存，过期则自动更新"""
+        """获取 IP 逻辑：缓存 -> DoH -> 保底"""
+        # 1. 如果是纯 IP 地址，直接返回
+        if hostname.replace('.', '').isdigit():
+            return hostname
+
         current_time = time.time()
-        
+        # 2. 检查缓存
         if hostname in self.cache:
             if current_time < self.cache[hostname]['expiry']:
                 return self.cache[hostname]['ip']
 
-        # 尝试更新 IP
+        # 3. HTTP 解析
         new_ip = self._fetch_ip_from_http(hostname)
-        
         if new_ip:
             self.cache[hostname] = {
                 "ip": new_ip,
@@ -64,32 +59,35 @@ class DnsManager:
             print(f"🔄 [DNS库] 动态解析成功: {hostname} -> {new_ip}")
             return new_ip
         
-        # 如果获取失败，且缓存里有旧的，则延长旧缓存寿命并继续使用
+        # 4. 保底 IP (Discord)
         if hostname in self.cache:
-            self.logger.warning(f"解析失败，正在使用 {hostname} 的过期缓存")
             return self.cache[hostname]['ip']
-        
-        # 最后的保底 hardcode IP (Discord 常用 IP)
         return "162.159.137.232"
 
     def patch_socket(self):
-        """核心方法：将补丁注入到 Python 全局 socket 层"""
+        """注入补丁：除了 dns.google，全部强行接管"""
         original_getaddrinfo = socket.getaddrinfo
 
         def patched_getaddrinfo(*args, **kwargs):
             host = args[0]
-            # 只有在目标域名列表里且不是正在请求 DNS 接口本身时才拦截
-            if host in self.target_hosts:
+            port = args[1]
+
+            # 关键排除逻辑：
+            # 如果请求的是 dns.google 本身，必须走原始解析，否则会陷入死循环（无限递归）
+            if host == "dns.google":
+                return original_getaddrinfo(*args, **kwargs)
+            
+            # 其他所有请求（Discord, MongoDB 等）全部走补丁
+            try:
                 latest_ip = self.get_ip(host)
-                port = args[1]
-                # 强制返回 IPv4 地址族 (AF_INET)，避免 IPv6 解析错误
+                # 强制返回 IPv4 结构体给系统
                 return [(socket.AF_INET, socket.SOCK_STREAM, 6, '', (latest_ip, port))]
-            
-            # 其余请求（如访问 MongoDB 或 Google DNS 接口）走原始解析
-            return original_getaddrinfo(*args, **kwargs)
+            except Exception:
+                # 最后的最后，如果补丁崩了，尝试退回原始解析
+                return original_getaddrinfo(*args, **kwargs)
 
         socket.getaddrinfo = patched_getaddrinfo
-        print("💉 [DNS库] 全局 Socket 补丁已加载，已绕过系统 DNS")
+        print("💉 [DNS库] 全局覆盖模式已加载：除 dns.google 外，所有域名解析已托管至 DoH")
 
-# 导出实例
+# 导出单例实例
 dns_manager = DnsManager()