From a1516535f9240734ddafd84d24328b36dd829747 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Wed, 28 Sep 2022 15:08:59 -0400 Subject: [PATCH] Agent: Add InfectionMonkey._agent_id attribute --- monkey/infection_monkey/monkey.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/monkey/infection_monkey/monkey.py b/monkey/infection_monkey/monkey.py index acd10ce95..e998854e8 100644 --- a/monkey/infection_monkey/monkey.py +++ b/monkey/infection_monkey/monkey.py @@ -111,6 +111,7 @@ class InfectionMonkey: self._singleton = SystemSingleton() self._opts = self._get_arguments(args) + self._agent_id = get_agent_id() self._agent_event_serializer_registry = self._setup_agent_event_serializers() @@ -122,7 +123,7 @@ class InfectionMonkey: server_address=self._island_address, island_api_client=self._island_api_client ) self._control_channel = ControlChannel( - str(self._island_address), get_agent_id(), self._island_api_client + str(self._island_address), self._agent_id, self._island_api_client ) self._register_agent() @@ -176,7 +177,7 @@ class InfectionMonkey: def _register_agent(self): agent_registration_data = AgentRegistrationData( - id=get_agent_id(), + id=self._agent_id, machine_hardware_id=get_machine_id(), start_time=agent_process.get_start_time(), # parent_id=parent,