Retries for _query_cvd_version_dns. Fixes #49.

Cisco-Talos · May 19, 2023 · ab30b5f · ab30b5f
1 parent 8ca1f23
commit ab30b5f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/cvdupdate/cvdupdate.py b/cvdupdate/cvdupdate.py
@@ -548,7 +548,12 @@ def _query_cvd_version_dns(self, db: str) -> int:
 
         if self.dns_version_tokens == []:
             # Query DNS if we haven't already
-            self._query_dns_txt_entry()
+            for _attempt in range(self.config['max retry']):
+                if self._query_dns_txt_entry():
+                    break
+                # Pause before next attempt.
+                time.sleep(0.1)
+
             if self.dns_version_tokens == []:
                 # Query failed. Bail out.
                 return version