Add a smart resolver

Author: mohanson

cmd/daze/main.go

@@ -79,14 +79,7 @@ func main() {
 		log.Println("main: server cipher is", *flCipher)
 		log.Println("main: protocol is used", *flProtoc)
 		if *flDnserv != "" {
-			switch {
-			case strings.HasSuffix(*flDnserv, ":53"):
-				net.DefaultResolver = daze.ResolverDns(*flDnserv)
-			case strings.HasSuffix(*flDnserv, ":853"):
-				net.DefaultResolver = daze.ResolverDot(*flDnserv)
-			case strings.HasPrefix(*flDnserv, "https://"):
-				net.DefaultResolver = daze.ResolverDoh(*flDnserv)
-			}
+			net.DefaultResolver = daze.ResolverAny(*flDnserv)
 			log.Println("main: domain server is", *flDnserv)
 		}
 		if *flLimits != "" {
@@ -151,14 +144,7 @@ func main() {
 		log.Println("main: client cipher is", *flCipher)
 		log.Println("main: protocol is used", *flProtoc)
 		if *flDnserv != "" {
-			switch {
-			case strings.HasSuffix(*flDnserv, ":53"):
-				net.DefaultResolver = daze.ResolverDns(*flDnserv)
-			case strings.HasSuffix(*flDnserv, ":853"):
-				net.DefaultResolver = daze.ResolverDot(*flDnserv)
-			case strings.HasPrefix(*flDnserv, "https://"):
-				net.DefaultResolver = daze.ResolverDoh(*flDnserv)
-			}
+			net.DefaultResolver = daze.ResolverAny(*flDnserv)
 			log.Println("main: domain server is", *flDnserv)
 		}
 		if *flLimits != "" {

daze.go

@@ -176,6 +176,24 @@ func ResolverDoh(addr string) *net.Resolver {
 	}
 }
 
+// ResolverDot returns a DoT resolver. Depending on the addr entered, the dns, dot or doh protocol will be used
+// automatically.
+//
+// Dns: 1.1.1.1:53
+// Dot: 1.1.1.1:853
+// Doh: https://1.1.1.1/dns-query
+func ResolverAny(addr string) *net.Resolver {
+	switch {
+	case strings.HasSuffix(addr, ":53"):
+		return ResolverDns(addr)
+	case strings.HasSuffix(addr, ":853"):
+		return ResolverDot(addr)
+	case strings.HasPrefix(addr, "https://"):
+		return ResolverDoh(addr)
+	}
+	panic("unreachable")
+}
+
 // Link copies from src to dst and dst to src until either EOF is reached.
 func Link(a, b io.ReadWriteCloser) {
 	w := sync.WaitGroup{}