dnsforward: fix address

heyxkhoa · Jan 27, 2021 · ed47f26 · ed47f26
1 parent 98b222b
commit ed47f26
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/internal/dnsforward/stats.go b/internal/dnsforward/stats.go
@@ -11,6 +11,8 @@ import (
 	"github.com/miekg/dns"
 )
 
+// TODO(a.garipov): Cover with tests.
+
 // Write Stats data and logs
 func processQueryLogsAndStats(ctx *dnsContext) (rc resultCode) {
 	elapsed := time.Since(ctx.startTime)
@@ -77,8 +79,8 @@ func (s *Server) updateStats(ctx *dnsContext, elapsed time.Duration, res dnsfilt
 
 	if clientID := ctx.clientID; clientID != "" {
 		e.Client = clientID
-	} else if pctx.Addr != nil {
-		e.Client = pctx.Addr.String()
+	} else if ip := IPFromAddr(pctx.Addr); ip != nil {
+		e.Client = ip.String()
 	}
 
 	e.Time = uint32(elapsed / 1000)