feat: support for custom dns port (#942)

jeessy2 · Dec 19, 2023 · a36c4c1 · a36c4c1
1 parent ebf78ee
commit a36c4c1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -11,6 +11,7 @@ import (
 	"os/exec"
 	"path/filepath"
 	"strconv"
+	"strings"
 	"time"
 
 	"github.com/jeessy2/ddns-go/v5/config"
@@ -83,7 +84,11 @@ func main() {
 		util.SetInsecureSkipVerify()
 	}
 	if *customDNSServer != "" {
-		util.NewDialerResolver(*customDNSServer + ":53")
+		if !strings.Contains(*customDNSServer, ":") {
+			util.NewDialerResolver(*customDNSServer + ":53")
+		} else {
+			util.NewDialerResolver(*customDNSServer)
+		}
 	}
 	os.Setenv(util.IPCacheTimesENV, strconv.Itoa(*ipCacheTimes))
 	switch *serviceType {