Add in context exists checks

Author: bitterpanda63

aikido_zen/vulnerabilities/__init__.py

@@ -37,6 +37,10 @@ def run_vulnerability_scan(kind, op, args):
     raises error if blocking is enabled, communicates it with connection_manager
     """
     context = get_current_context()
+    if not context and kind != "ssrf":
+        # Make a special exception for SSRF:
+        # For stored ssrf we don't need a context
+        return
     if should_skip_attack_scan(context) and kind != "ssrf":
         # Make a special exception for SSRF:
         # For stored ssrf we don't want to check bypassed IPs or protection forced off.

aikido_zen/vulnerabilities/ssrf/inspect_getaddrinfo_result.py

@@ -28,7 +28,7 @@ def inspect_getaddrinfo_result(dns_results, hostname, port):
         return
 
     context = get_current_context()
-    if should_skip_attack_scan(context):
+    if not context or should_skip_attack_scan(context):
         return
 
     # attack_findings is an object containing source, pathToPayload and payload.