Fix dns lookup

Author: duyleekun

util/wg_pool.go

@@ -47,6 +47,11 @@ func (w *WGPool) AddPoolPeerByPubKey(publicKey wgtypes.Key) {
 	}
 }
 
+func (w *WGPool) HasSubdomain(subdomain string) bool {
+	_, ok := w.subdomains[subdomain]
+	return ok
+}
+
 func (w *WGPool) GetPeerAddressBySubdomain(subdomain string) net.IP {
 	return w.subdomains[subdomain].address
 }

wgapi/myhttpserver/server.go

@@ -43,9 +43,10 @@ func (h *HTTPServer) StartServer() chan error {
 		}
 
 		var rr dns.RR
+		var subdomain = strings.Split(question.Name, ".")[0]
 		switch question.Qtype {
 		case dns.TypeA:
-			if val := h.server.GetPeerAddressFor(strings.ToLower(question.Name)); val != nil {
+			if val := h.server.GetPeerAddressFor(strings.ToLower(subdomain)); val != nil {
 				rr = &dns.A{
 					Hdr: dns.RR_Header{
 						Name:     question.Name,

wgapi/mywgserver/server.go

@@ -217,7 +217,12 @@ func (s *WGServer) StopServer() {
 }
 
 func (s *WGServer) GetPeerAddressFor(subdomain string) net.IP {
-	return s.wgPool.GetPeerAddressBySubdomain(subdomain)
+	s.logger.Verbosef("GetPeerAddressFor %s", subdomain)
+	if s.wgPool.HasSubdomain(subdomain) {
+		return s.wgPool.GetPeerAddressBySubdomain(subdomain)
+	} else {
+		return nil
+	}
 }
 
 func checkIsRoot(logger *device.Logger) (bool, error) {