create send_payload helper function

Author: bitterpanda63

aikido_zen/background_process/comms.py

@@ -6,7 +6,6 @@
 import multiprocessing.connection as con
 from threading import Thread
 
-from aikido_zen.background_process.commands.command import Payload
 from aikido_zen.helpers.logging import logger
 
 # pylint: disable=invalid-name # This variable does change
@@ -100,8 +99,3 @@ def target(address, key, receive, data, result_obj):
         if receive:
             return {"success": True, "data": result_obj[1]}
         return {"success": True, "data": None}
-
-    def send_payload(self, payload: Payload):
-        return self.send_data_to_bg_process(
-            payload.identifier, payload.request, payload.returns_data
-        )

aikido_zen/helpers/ipc/__init__.py

@@ -0,0 +1,8 @@
+from aikido_zen.background_process import AikidoIPCCommunications
+from aikido_zen.background_process.commands.command import Payload
+
+
+def send_payload(comms: AikidoIPCCommunications, payload: Payload):
+    return comms.send_data_to_bg_process(
+        payload.identifier, payload.request, payload.returns_data
+    )

aikido_zen/helpers/ipc/send_payload.py

@@ -31,6 +31,7 @@
 )
 from ..background_process.commands import PutEventCommand
 from ..helpers.create_detected_attack_api_event import create_detected_attack_api_event
+from ..helpers.ipc.send_payload import send_payload
 
 
 def run_vulnerability_scan(kind, op, args):
@@ -111,6 +112,6 @@ def run_vulnerability_scan(kind, op, args):
             injection_results, context, blocked, stack
         )
         if comms and event:
-            comms.send_payload(PutEventCommand.generate(event))
+            send_payload(comms, PutEventCommand.generate(event))
         if blocked:
             raise error_type(*error_args)