p15693087
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests 1 Projects Releases Wiki Activity

Agent: Simplify relay disconnect logic in network/relay/utils.py

This commit is contained in:
Shreya Malviya 2022-09-27 13:06:29 +05:30
parent 441c14f15d
commit d1199fdab2
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
monkey/infection_monkey/network/relay/utils.py
View File

@ -79,18 +79,15 @@ def _check_if_island_server(
def send_remove_from_waitlist_control_message_to_relays(servers: Iterable[str]): def send_remove_from_waitlist_control_message_to_relays(servers: Iterable[str]):
for i, server in enumerate(servers, start=1): for i, server in enumerate(servers, start=1):
server_address = SocketAddress.from_string(server)
t = create_daemon_thread( t = create_daemon_thread(
target=_send_remove_from_waitlist_control_message_to_relay, target=notify_disconnect,
name=f"SendRemoveFromWaitlistControlMessageToRelaysThread-{i:02d}", name=f"SendRemoveFromWaitlistControlMessageToRelaysThread-{i:02d}",
args=(server,), args=(server_address,),
) )
t.start() t.start()
def _send_remove_from_waitlist_control_message_to_relay(server: str):
notify_disconnect(SocketAddress.from_string(server))
def notify_disconnect(server_address: SocketAddress): def notify_disconnect(server_address: SocketAddress):
""" """
Tell upstream relay that we no longer need the relay Tell upstream relay that we no longer need the relay