Don't try to retrieve a query that wasn't made

1 files changed, 7 insertions, 0 deletions

diff --git a/dnsviz/analysis/offline.py b/dnsviz/analysis/offline.py
index 6b7b6b2..afc257b 100644
--- a/dnsviz/analysis/offline.py
+++ b/dnsviz/analysis/offline.py
@@ -256,6 +256,13 @@ class OfflineDomainNameAnalysis(OnlineDomainNameAnalysis):
         return info_obj
 
     def _get_response_info(self, name, rdtype):
+        #XXX there are reasons for this (e.g., NXDOMAIN, after which no further
+        # queries are made), but it would be good to have a sanity check, so
+        # we don't simply produce an incomplete output.
+        # see also: dnsviz.viz.dnssec.graph_rrset_auth()
+        if (name, rdtype) not in self.queries:
+            return None
+
         query = self.queries[(name, rdtype)]
         name_to_info_mapping = {}
         rrset_to_cname_mapping = {}