From f0029a0d824a327402bab1d3417f7b9e7cc3c1fa Mon Sep 17 00:00:00 2001 From: vakaris_zilius Date: Wed, 22 Jun 2022 13:13:32 +0000 Subject: [PATCH] Agent: Close inbound tunnel if one was opened --- monkey/infection_monkey/monkey.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/monkey/infection_monkey/monkey.py b/monkey/infection_monkey/monkey.py index d91f28018..53d8f8662 100644 --- a/monkey/infection_monkey/monkey.py +++ b/monkey/infection_monkey/monkey.py @@ -99,6 +99,7 @@ class InfectionMonkey: self._telemetry_messenger = LegacyTelemetryMessengerAdapter() self._current_depth = self._opts.depth self._master = None + self._inbound_tunnel_opened = False @staticmethod def _get_arguments(args): @@ -172,6 +173,7 @@ class InfectionMonkey: self._control_client.server_address, GUID, self._control_client.proxies ).get_config() if self._monkey_inbound_tunnel and should_propagate(config, self._current_depth): + self._inbound_tunnel_opened = True self._monkey_inbound_tunnel.start() StateTelem(is_done=False, version=get_version()).send() @@ -358,7 +360,7 @@ class InfectionMonkey: reset_signal_handlers() - if self._monkey_inbound_tunnel: + if self._inbound_tunnel_opened: self._monkey_inbound_tunnel.stop() self._monkey_inbound_tunnel.join()