feat: custom DNS server (#682)

* feat: custom DNS server

main.go

@@ -38,6 +38,9 @@ var noWebService = flag.Bool("noweb", false, "不启动 web 服务")
 // 跳过验证证书
 var skipVerify = flag.Bool("skipVerify", false, "跳过验证证书, 适合不能升级的老系统")
 
+// 自定义 DNS 服务器
+var customDNSServer = flag.String("dns", "", "自定义 DNS 服务器（例如 1.1.1.1）")
+
 //go:embed static
 var staticEmbededFiles embed.FS
 
@@ -47,9 +50,6 @@ var faviconEmbededFile embed.FS
 // version
 var version = "DEV"
 
-// buildTime
-var buildTime = ""
-
 func main() {
 	flag.Parse()
 	if _, err := net.ResolveTCPAddr("tcp", *listen); err != nil {
@@ -63,6 +63,9 @@ func main() {
 	if *skipVerify {
 		os.Setenv(util.SkipVerifyENV, "true")
 	}
+	if *customDNSServer != "" {
+		os.Setenv(util.DNSServerEnv, *customDNSServer+":53")
+	}
 	switch *serviceType {
 	case "install":
 		installService()

util/dns.go

@@ -0,0 +1,23 @@
+package util
+
+import (
+	"context"
+	"net"
+	"os"
+)
+
+const DNSServerEnv = "DDNS_GO_DNS_SERVER"
+
+// customDNSResolver 当 DNSServerEnv 值不为空时，使用 Go 内置 DNS 解析器来解析其 DNS 服务器。
+func customDNSResolver() *net.Resolver {
+	s := os.Getenv(DNSServerEnv)
+	if s != "" {
+		return &net.Resolver{
+			PreferGo: true,
+			Dial: func(ctx context.Context, network, address string) (net.Conn, error) {
+				return net.Dial("udp", s)
+			},
+		}
+	}
+	return &net.Resolver{}
+}

util/dns_test.go

@@ -0,0 +1,16 @@
+package util
+
+import (
+	"context"
+	"os"
+	"testing"
+)
+
+// TestCustomDNSResolver 测试能否通过 DNSServerEnv 值的 DNS 服务器解析域名的 IP。
+func TestCustomDNSResolver(t *testing.T) {
+	os.Setenv(DNSServerEnv, "1.1.1.1:53")
+	_, err := customDNSResolver().LookupIP(context.Background(), "ip", "cloudflare.com")
+	if err != nil {
+		t.Errorf("Failed to lookup IP, err: %v", err)
+	}
+}

util/http_client_util.go

@@ -31,6 +31,7 @@ var defaultTransport = &http.Transport{
 
 // CreateHTTPClient Create Default HTTP Client
 func CreateHTTPClient() *http.Client {
+	dialer.Resolver = customDNSResolver()
 	// SkipVerfiry
 	defaultTransport.TLSClientConfig = &tls.Config{InsecureSkipVerify: os.Getenv(SkipVerifyENV) == "true"}
 	return &http.Client{
@@ -73,6 +74,7 @@ var noProxyTcp6Transport = &http.Transport{
 
 // CreateNoProxyHTTPClient Create NoProxy HTTP Client
 func CreateNoProxyHTTPClient(network string) *http.Client {
+	dialer.Resolver = customDNSResolver()
 	if network == "tcp6" {
 		// SkipVerfiry
 		noProxyTcp6Transport.TLSClientConfig = &tls.Config{InsecureSkipVerify: os.Getenv(SkipVerifyENV) == "true"}