internal/dnsHandler.go

@@ -20,11 +20,10 @@ func (dm *DomainManager) ServeDNS(w dns.ResponseWriter, r *dns.Msg) {
 	start := time.Now()
 	q := r.Question[0]
 	responseMessage := new(dns.Msg)
-
 	ql := QueryLog{
 		Started:  start.UTC(),
 		Protocol: w.RemoteAddr().Network(),
-		ClientIP: w.RemoteAddr().String()[:strings.LastIndex(w.RemoteAddr().String(), ":")],
+		ClientIP: strings.Split(w.RemoteAddr().String(), ":")[0],
 		Domain:   q.Name,
 		Status:   NoAnswer,
 	}

main.go

@@ -13,10 +13,9 @@ import (
 )
 
 var (
-	dbPath          = flag.String("db-path", ".", "Directory to write database files to")
-	httpAddr        = flag.String("http-address", ":80", "Bind address for http server")
-	dnsAddr         = flag.String("dns-address", ":53", "Bind address for dns server")
-	defaultUpstream = flag.String("upstream", "1.1.1.1:53", "default upstream DNS server when no others are specified")
+	dbPath   = flag.String("db-path", ".", "Directory to write database files to")
+	httpAddr = flag.String("http-address", ":80", "Bind address for http server")
+	dnsAddr  = flag.String("dns-address", ":53", "Bind address for dns server")
 )
 
 func main() {
@@ -58,7 +57,7 @@ func main() {
 		Storage:       store,
 		RuleEvaluator: re,
 		Recursors: internal.Recursor{
-			Upstreams: []string{*defaultUpstream},
+			Upstreams: []string{"1.1.1.1:53"},
 			Client:    dnsClient,
 		},
 	}