Agent: Pass Island address to payload plugins

Issue #3411 PR #3556
guardicore · Aug 7, 2023 · 00fa994 · 00fa994
1 parent bd3f643
commit 00fa994
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/monkey/infection_monkey/monkey.py b/monkey/infection_monkey/monkey.py
@@ -430,7 +430,7 @@ def _build_puppet(self, operating_system: OperatingSystem) -> IPuppet:
                 create_plugin,
             ),
             AgentPluginType.PAYLOAD: PayloadPluginFactory(
-                self._agent_id, self._agent_event_publisher, create_plugin
+                self._agent_id, self._agent_event_publisher, self._island_address, create_plugin
             ),
         }
         plugin_registry = PluginRegistry(

diff --git a/monkey/infection_monkey/plugin/payload_plugin_factory.py b/monkey/infection_monkey/plugin/payload_plugin_factory.py
@@ -3,7 +3,7 @@
 from serpentarium import SingleUsePlugin
 
 from common.event_queue import IAgentEventPublisher
-from common.types import AgentID
+from common.types import AgentID, SocketAddress
 
 from .i_plugin_factory import IPluginFactory
 
@@ -13,9 +13,11 @@ def __init__(
         self,
         agent_id: AgentID,
         agent_event_publisher: IAgentEventPublisher,
+        island_server_address: SocketAddress,
         create_plugin: Callable[..., SingleUsePlugin],
     ):
         self._agent_id = agent_id
+        self._island_server_address = island_server_address
         self._agent_event_publisher = agent_event_publisher
         self._create_plugin = create_plugin
 
@@ -24,4 +26,5 @@ def create(self, plugin_name: str) -> SingleUsePlugin:
             plugin_name=plugin_name,
             agent_id=self._agent_id,
             agent_event_publisher=self._agent_event_publisher,
+            island_server_address=self._island_server_address,
         )