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

Agent: Rework send_control_relay_message a bit

This commit is contained in:
Ilija Lazoroski 2022-09-07 19:39:14 +02:00 committed by Mike Salvatore
parent 60f9aa6a4e
commit 20172230f1
1 changed files with 6 additions and 3 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

@ -52,10 +52,13 @@ def _open_socket_to_server(server: str):
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as d_socket:
d_socket.settimeout(MEDIUM_REQUEST_TIMEOUT)
ip, port = address_to_ip_port(server)
logger.info(f"Control message was sent to the server/relay {server}")
try:
address, port = address_to_ip_port(server)
d_socket.connect((address, int(port)))
d_socket.connect((ip, int(port)))
d_socket.send(RELAY_CONTROL_MESSAGE)
logger.info(f"Control message was sent to the server/relay {server}")
except OSError as err:
logger.error(f"Error connecting to socket {server}: {err}")
except TimeoutError as err:
logger.error(f"Timed out while connecting to socket {server}: {err}")