Revert "Do not error when src_data is not present"

This reverts commit c7a6099.
thinkst · Nov 21, 2024 · 8386dc1 · 8386dc1
1 parent 98f5b53
commit 8386dc1
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/canarytokens/channel_dns.py b/canarytokens/channel_dns.py
@@ -252,9 +252,7 @@ def query(self, query: Query, src_ip: str):  # noqa C901
                 return defer.succeed(self._do_dynamic_response(name=query.name.name))
 
         if canarydrop.type == TokenTypes.WINDOWS_FAKE_FS:
-            invocation_id = src_data.get("src_data", {}).get(
-                "windows_fake_fs_invocation_id"
-            )
+            invocation_id = src_data["src_data"].get("windows_fake_fs_invocation_id")
             if (invocation_id is not None) and any(
                 hit.src_data.get("windows_fake_fs_invocation_id") == invocation_id
                 for hit in canarydrop.triggered_details.hits