From c9982ab6a25c0cd95bfc0867dc8cab991786ddb5 Mon Sep 17 00:00:00 2001
From: Sam Varshavchik
Date: Thu, 7 May 2020 22:49:32 -0400
Subject: More simultaneous DNS query support.

---
 rfc1035/rfc1035reply.c | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

(limited to 'rfc1035/rfc1035reply.c')

diff --git a/rfc1035/rfc1035reply.c b/rfc1035/rfc1035reply.c
index ac1e87e..ab43dfb 100644
--- a/rfc1035/rfc1035reply.c
+++ b/rfc1035/rfc1035reply.c
@@ -489,7 +489,10 @@ void rfc1035_rr_rand_ar(struct rfc1035_reply *rr)
 
 void rfc1035_rr_rand(struct rfc1035_reply *rr)
 {
-	rfc1035_rr_rand_an(rr);
-	rfc1035_rr_rand_ns(rr);
-	rfc1035_rr_rand_ar(rr);
+	if (rr->ancount > 1)
+		rfc1035_rr_rand_an(rr);
+	if (rr->nscount > 1)
+		rfc1035_rr_rand_ns(rr);
+	if (rr->arcount > 1)
+		rfc1035_rr_rand_ar(rr);
 }
-- 
cgit v1.2.3