Merge pull request #47 from projectdiscovery/bug-truncate-err-check

Adding missing err check on truncate exchange
projectdiscovery · Mar 3, 2022 · 5a9c5bd · 5a9c5bd
2 parents 3ce99ec + 63c6d2f
commit 5a9c5bd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/client.go b/client.go
@@ -262,7 +262,10 @@ func (c *Client) QueryMultiple(host string, requestTypes []uint16) (*DNSData, er
 
 			// https://github.com/projectdiscovery/retryabledns/issues/25
 			if resp.Truncated && c.TCPFallback {
-				resp, _, _ = c.tcpClient.Exchange(msg, resolver.String())
+				resp, _, err = c.tcpClient.Exchange(msg, resolver.String())
+				if err != nil || resp == nil {
+					continue
+				}
 			}
 
 			err = dnsdata.ParseFromMsg(resp)