Merge pull request #93 from AikidoSec/AIK-3348

Don't start heartbeat and realtime polling if token invalid

aikido_firewall/background_process/reporter/__init__.py

@@ -6,6 +6,7 @@
 from aikido_firewall.background_process.heartbeats import send_heartbeats_every_x_secs
 from aikido_firewall.background_process.routes import Routes
 from aikido_firewall.ratelimiting.rate_limiter import RateLimiter
+from aikido_firewall.helpers.logging import logger
 from ..service_config import ServiceConfig
 from ..users import Users
 from ..hostnames import Hostnames
@@ -48,7 +49,12 @@ def __init__(self, block, api, token, serverless):
 
     def start(self, event_scheduler):
         """Send out start event and add heartbeats"""
-        self.on_start()
+        res = self.on_start()
+        if res.get("error", None) == "invalid_token":
+            logger.info(
+                "Token was invalid, not starting heartbeats and realtime polling."
+            )
+            return
         send_heartbeats_every_x_secs(self, self.heartbeat_secs, event_scheduler)
         start_polling_for_changes(
             self.update_service_config, self.serverless, self.token, event_scheduler

aikido_firewall/background_process/reporter/on_start.py

@@ -24,3 +24,4 @@ def on_start(reporter):
     else:
         reporter.update_service_config(res)
         logger.info("Established connection with Aikido Server")
+    return res