From f8b094244a27af98707dd42cbf6b18b7d00b45c4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E8=89=AF?= <841369634@qq.com>
Date: Tue, 20 Aug 2024 15:53:18 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BB=A3=E7=90=86=E8=BF=9E=E6=8E=A5=E8=B6=85?=
 =?UTF-8?q?=E6=97=B6=EF=BC=8C=E4=B9=9F=E8=AE=B0=E5=BD=95=E5=A4=B1=E8=B4=A5?=
 =?UTF-8?q?=E6=AC=A1=E6=95=B0=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../src/lib/proxy/mitmproxy/createConnectHandler.js         | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/packages/mitmproxy/src/lib/proxy/mitmproxy/createConnectHandler.js b/packages/mitmproxy/src/lib/proxy/mitmproxy/createConnectHandler.js
index ba95563..6eeedc9 100644
--- a/packages/mitmproxy/src/lib/proxy/mitmproxy/createConnectHandler.js
+++ b/packages/mitmproxy/src/lib/proxy/mitmproxy/createConnectHandler.js
@@ -88,6 +88,12 @@ function connect (req, cltSocket, head, hostname, port, dnsConfig/* , sniRegexpM
       log.error(errorMsg)
 
       cltSocket.destroy()
+
+      if (isDnsIntercept && isDnsIntercept.dns && isDnsIntercept.ip !== isDnsIntercept.hostname) {
+        const { dns, ip, hostname } = isDnsIntercept
+        dns.count(hostname, ip, true)
+        log.error(`记录ip失败次数，用于优选ip！ hostname: ${hostname}, ip: ${ip}, reason: ${errorMsg}, dns: ${dns.name}`)
+      }
     })
     proxySocket.on('error', (e) => {
       // 连接失败，可能被GFW拦截，或者服务端拥挤