Cryptojacker: Modify stubbed components to accept correct parameters

guardicore · Aug 9, 2023 · 13e2d7b · 13e2d7b
1 parent 2ce4461
commit 13e2d7b
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 4 deletions.
diff --git a/monkey/agent_plugins/payloads/cryptojacker/src/bitcoin_mining_network_traffic_simulator.py b/monkey/agent_plugins/payloads/cryptojacker/src/bitcoin_mining_network_traffic_simulator.py
@@ -1,7 +1,20 @@
 from typing import Optional
 
+from common.event_queue import IAgentEventPublisher
+from common.types import AgentID, SocketAddress
+
 
 class BitcoinMiningNetworkTrafficSimulator:
+    def __init__(
+        self,
+        island_server_address: SocketAddress,
+        agent_id: AgentID,
+        agent_event_publisher: IAgentEventPublisher,
+    ):
+        self._island_server_address = island_server_address
+        self._agent_id = agent_id
+        self._agent_event_publisher = agent_event_publisher
+
     def start(self):
         pass
 

diff --git a/monkey/agent_plugins/payloads/cryptojacker/src/cpu_utilizer.py b/monkey/agent_plugins/payloads/cryptojacker/src/cpu_utilizer.py
@@ -1,11 +1,19 @@
 from typing import Optional
 
-from common.types import PercentLimited
+from common.event_queue import IAgentEventPublisher
+from common.types import AgentID, PercentLimited
 
 
 class CPUUtilizer:
-    def __init__(self, cpu_utilization: PercentLimited):
+    def __init__(
+        self,
+        cpu_utilization: PercentLimited,
+        agent_id: AgentID,
+        agent_event_publisher: IAgentEventPublisher,
+    ):
         self._cpu_utilization = cpu_utilization
+        self._agent_id = agent_id
+        self._agent_event_publisher = agent_event_publisher
 
     def start(self):
         pass

diff --git a/monkey/agent_plugins/payloads/cryptojacker/src/memory_utilizer.py b/monkey/agent_plugins/payloads/cryptojacker/src/memory_utilizer.py
@@ -1,11 +1,19 @@
 from typing import Optional
 
-from common.types import PercentLimited
+from common.event_queue import IAgentEventPublisher
+from common.types import AgentID, PercentLimited
 
 
 class MemoryUtilizer:
-    def __init__(self, memory_utilization: PercentLimited):
+    def __init__(
+        self,
+        memory_utilization: PercentLimited,
+        agent_id: AgentID,
+        agent_event_publisher: IAgentEventPublisher,
+    ):
         self._memory_utilization = memory_utilization
+        self._agent_id = agent_id
+        self._agent_event_publisher = agent_event_publisher
 
     def start(self):
         pass