fix query error handling

Matchlighter · Feb 16, 2022 · c405060 · c405060
1 parent ca2fdc4
commit c405060
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/cmd/sparse_dns/main.go b/cmd/sparse_dns/main.go
@@ -132,16 +132,15 @@ func syncResolve(req *dns.Msg) (*dns.Msg, time.Duration, error) {
 		if err != nil || (resolved != nil && resolved.Truncated) {
 			resolved, rtt, err = tcpClient.Exchange(req, addr)
 		}
-		if (dns.RcodeToString[resolved.Rcode] != "NOERROR") {
-			continue
+		if (err != nil) {
+			return nil, 0, err
 		}
-		if err == nil {
+		if (dns.RcodeToString[resolved.Rcode] == "SERVFAIL") {
+			break
+		}
+		if (dns.RcodeToString[resolved.Rcode] == "NOERROR") {
 			break
 		}
-	}
-
-	if err != nil {
-		return nil, 0, err
 	}
 
 	return resolved, rtt, nil