revert changes made to background_process & check_firewall_lists

Author: bitterpanda63

aikido_zen/background_process/cloud_connection_manager/__init__.py

@@ -21,10 +21,6 @@
 from .update_service_config import update_service_config
 from .on_start import on_start
 from .send_heartbeat import send_heartbeat
-from aikido_zen.vulnerabilities.attack_wave_detection.attack_wave_detector import (
-    AttackWaveDetector,
-)
-from aikido_zen.context import Context
 
 
 class CloudConnectionManager:
@@ -55,7 +51,6 @@ def __init__(self, block, api, token, serverless):
         self.statistics = Statistics()
         self.ai_stats = AIStatistics()
         self.middleware_installed = False
-        self.attack_wave_detector = AttackWaveDetector()
 
         if isinstance(serverless, str) and len(serverless) == 0:
             raise ValueError("Serverless cannot be an empty string")

aikido_zen/background_process/commands/check_firewall_lists.py

@@ -4,7 +4,7 @@
 def process_check_firewall_lists(connection_manager, data):
     """
     Checks whether an IP is blocked and whether there is an ongoing attack wave
-    data: {"ip": string, "user-agent": string, "is_attack_wave_request" ?: bool}
+    data: {"ip": string, "user-agent": string}
     returns -> CheckFirewallListsRes
     """
     ip = data["ip"]
@@ -24,17 +24,11 @@ def process_check_firewall_lists(connection_manager, data):
         if connection_manager.firewall_lists.is_user_agent_blocked(user_agent):
             return CheckFirewallListsRes(blocked=True, type="bot-blocking")
 
-    is_attack_wave_request = data.get("is_attack_wave_request", None)
-    if is_attack_wave_request and ip is not None:
-        if connection_manager.attack_wave_detector.is_attack_wave(ip):
-            return CheckFirewallListsRes(blocked=False, is_attack_wave=True)
-
     return CheckFirewallListsRes()
 
 
 class CheckFirewallListsRes:
-    def __init__(self, blocked=False, is_attack_wave=False, type=None, reason=None):
+    def __init__(self, blocked=False, type=None, reason=None):
         self.blocked = blocked
-        self.is_attack_wave = is_attack_wave
         self.type = type
         self.reason = reason