From fbfc39421fd61b0a18eede931b166aaf5d9609ce Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=80=81=E4=B9=9D?= <366193849@qq.com>
Date: Tue, 21 Sep 2021 14:04:41 +0800
Subject: [PATCH] =?UTF-8?q?=E7=A7=BB=E9=99=A4forceTcp=E5=8F=82=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 FastGithub.DomainResolve/DnsClient.cs      | 5 ++---
 FastGithub.DomainResolve/DomainResolver.cs | 4 ++--
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/FastGithub.DomainResolve/DnsClient.cs b/FastGithub.DomainResolve/DnsClient.cs
index 97ea09e..a734674 100644
--- a/FastGithub.DomainResolve/DnsClient.cs
+++ b/FastGithub.DomainResolve/DnsClient.cs
@@ -23,11 +23,10 @@ namespace FastGithub.DomainResolve
         /// DNS客户端
         /// </summary>
         /// <param name="dns"></param>
-        /// <param name="forceTcp"></param>
-        public DnsClient(IPEndPoint dns, bool forceTcp)
+        public DnsClient(IPEndPoint dns)
         {
             this.dns = dns;
-            this.resolver = forceTcp
+            this.resolver = dns.Port == 53
                 ? new TcpRequestResolver(dns)
                 : new UdpRequestResolver(dns, new TcpRequestResolver(dns));
         }
diff --git a/FastGithub.DomainResolve/DomainResolver.cs b/FastGithub.DomainResolve/DomainResolver.cs
index 3c8777d..f66ea7c 100644
--- a/FastGithub.DomainResolve/DomainResolver.cs
+++ b/FastGithub.DomainResolve/DomainResolver.cs
@@ -145,7 +145,7 @@ namespace FastGithub.DomainResolve
                 return null;
             }
 
-            var dnsClient = new DnsClient(dns, forceTcp: false);
+            var dnsClient = new DnsClient(dns);
             var address = await this.LookupAsync(dnsClient, domain, cancellationToken);
             return address ?? await this.LookupAsync(dnsClient, domain, cancellationToken);
         }
@@ -161,7 +161,7 @@ namespace FastGithub.DomainResolve
         {
             foreach (var dns in this.fastGithubConfig.FallbackDns)
             {
-                var dnsClient = new DnsClient(dns, forceTcp: true);
+                var dnsClient = new DnsClient(dns);
                 var address = await this.LookupAsync(dnsClient, domain, cancellationToken);
                 if (address != null)
                 {