From e6b2be55702aebfa848d28890db94bd108a0b697 Mon Sep 17 00:00:00 2001
From: Shulhan <ms@kilabit.info>
Date: Thu, 30 Aug 2018 12:18:32 +0700
Subject: cmd/resolver: fix counting number of minimum dots

---
 cmd/resolver/main.go | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'cmd')

diff --git a/cmd/resolver/main.go b/cmd/resolver/main.go
index 14f5f3c..9a4918f 100644
--- a/cmd/resolver/main.go
+++ b/cmd/resolver/main.go
@@ -31,9 +31,16 @@ func parseNameServers(nameservers []string) (udpAddrs []*net.UDPAddr) {
 }
 
 func populateQueries(cr *libnet.ResolvConf, qname string) (queries []string) {
-	names := strings.Split(qname, ".")
+	ndots := 0
 
-	if len(names) == cr.NDots+1 {
+	for _, c := range qname {
+		if c == '.' {
+			ndots++
+			continue
+		}
+	}
+
+	if ndots >= cr.NDots {
 		queries = append(queries, qname)
 	} else {
 		if len(cr.Domain) > 0 {
-- 
cgit v1.3-5-g9baa