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 index 9b4f45931bf..b10bb2a6cb2 100644 --- 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 index 74ff6e6865a..3ad935e86ff 100644 --- 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 index a4abbf3d19f..c93649683b1 100644 --- 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