Adding missing err check

projectdiscovery · Mar 2, 2022 · 63c6d2f · 63c6d2f
1 parent 3ce99ec
commit 63c6d2f
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)