Fix: should check all ips need to fallback (#2915)

yuhaya · Sep 2, 2023 · db2b5db · db2b5db
1 parent 1855e18
commit db2b5db
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dns/resolver.go b/dns/resolver.go
@@ -265,7 +265,10 @@ func (r *Resolver) ipExchange(ctx context.Context, m *D.Msg) (msg *D.Msg, err er
 	res := <-msgCh
 	if res.Error == nil {
 		if ips := msgToIP(res.Msg); len(ips) != 0 {
-			if !r.shouldIPFallback(ips[0]) {
+			shouldNotFallback := lo.EveryBy(ips, func(ip net.IP) bool {
+				return !r.shouldIPFallback(ip)
+			})
+			if shouldNotFallback {
 				msg = res.Msg // no need to wait for fallback result
 				err = res.Error
 				return msg, err