p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Agent: Fix control_client_object TODOs

This commit is contained in:
Ilija Lazoroski 2022-06-13 17:55:19 +02:00 committed by vakarisz
parent 7fe6c170cd
commit 17a0be2fa0
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
monkey/infection_monkey/control.py
View File

@ -25,8 +25,8 @@ PBA_FILE_DOWNLOAD = "https://%s/api/pba/download/%s"
class ControlClient:
# TODO Every telemetry should have its own control client
# for the moment that is a big refactor.
# TODO When we have mechanism that support telemetry messenger
# with control clients, then this needs to be removed
# Ref: infection_monkey.telemetry.base_telem.py
control_client_object = None

5
monkey/infection_monkey/monkey.py
View File

@ -10,8 +10,8 @@ import infection_monkey.tunnel as tunnel
from common.network.network_utils import address_to_ip_port
from common.utils.attack_utils import ScanStatus, UsageEnum
from common.version import get_version
from infection_monkey.control import ControlClient
from infection_monkey.config import GUID
from infection_monkey.control import ControlClient
from infection_monkey.credential_collectors import (
MimikatzCredentialCollector,
SSHCredentialCollector,
@ -90,7 +90,8 @@ class InfectionMonkey:
self._opts = self._get_arguments(args)
self._cmd_island_ip, self._cmd_island_port = address_to_ip_port(self._opts.server)
self._control_client = ControlClient(self._opts.server)
# TODO Refactor the BaseTelem to have its own control client
# TODO Refactor the telemetry messengers to accept control client
# and remove control_client_object
ControlClient.control_client_object = self._control_client
self._monkey_inbound_tunnel = None
self._telemetry_messenger = LegacyTelemetryMessengerAdapter()