p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Agent: Change argument in InfectionMonkey._register_agent() to type SocketAddress

This commit is contained in:
Shreya Malviya 2022-09-26 17:16:17 +05:30
parent 6df2c29e30
commit ac633a6e75
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
monkey/infection_monkey/monkey.py
View File

@ -131,7 +131,7 @@ class InfectionMonkey:
server_address=server, island_api_client=self._island_api_client server_address=server, island_api_client=self._island_api_client
) )
self._control_channel = ControlChannel(server, get_agent_id(), self._island_api_client) self._control_channel = ControlChannel(server, get_agent_id(), self._island_api_client)
self._register_agent(server) self._register_agent(self._island_address)
# TODO Refactor the telemetry messengers to accept control client # TODO Refactor the telemetry messengers to accept control client
# and remove control_client_object # and remove control_client_object
@ -180,7 +180,7 @@ class InfectionMonkey:
return server, island_api_client return server, island_api_client
def _register_agent(self, server: str): def _register_agent(self, server: SocketAddress):
agent_registration_data = AgentRegistrationData( agent_registration_data = AgentRegistrationData(
id=get_agent_id(), id=get_agent_id(),
machine_hardware_id=get_machine_id(), machine_hardware_id=get_machine_id(),