From 46fa8f74806ca0254cd46da1c2ab48b06fc0a8e4 Mon Sep 17 00:00:00 2001
From: Loyalsoldier <10487845+Loyalsoldier@users.noreply.github.com>
Date: Sun, 29 Aug 2021 16:24:10 +0800
Subject: [PATCH] Fix: DNS return value (#1242)

---
 app/dns/nameserver_doh.go  | 4 ++++
 app/dns/nameserver_quic.go | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/app/dns/nameserver_doh.go b/app/dns/nameserver_doh.go
index db61e831a..ec86884a1 100644
--- a/app/dns/nameserver_doh.go
+++ b/app/dns/nameserver_doh.go
@@ -325,6 +325,10 @@ func (s *DoHNameServer) findIPsForDomain(domain string, option dns_feature.IPOpt
 		return nil, err6
 	}
 
+	if (option.IPv4Enable && record.A != nil) || (option.IPv6Enable && record.AAAA != nil) {
+		return nil, dns_feature.ErrEmptyResponse
+	}
+
 	return nil, errRecordNotFound
 }
 
diff --git a/app/dns/nameserver_quic.go b/app/dns/nameserver_quic.go
index b0618ecee..565da06ef 100644
--- a/app/dns/nameserver_quic.go
+++ b/app/dns/nameserver_quic.go
@@ -262,6 +262,10 @@ func (s *QUICNameServer) findIPsForDomain(domain string, option dns_feature.IPOp
 		return nil, err6
 	}
 
+	if (option.IPv4Enable && record.A != nil) || (option.IPv6Enable && record.AAAA != nil) {
+		return nil, dns_feature.ErrEmptyResponse
+	}
+
 	return nil, errRecordNotFound
 }