Fixed copy-pasta bug about state telemetry.

This commit is contained in:
Shay Nehmad 2019-08-26 14:08:45 +03:00
parent f865c4b4b9
commit fbb82f412b
1 changed files with 2 additions and 2 deletions

View File

@ -113,7 +113,7 @@ class InfectionMonkey(object):
if monkey_tunnel: if monkey_tunnel:
monkey_tunnel.start() monkey_tunnel.start()
StateTelem(False).send() StateTelem(is_done=False).send()
TunnelTelem().send() TunnelTelem().send()
if WormConfiguration.collect_system_info: if WormConfiguration.collect_system_info:
@ -225,7 +225,7 @@ class InfectionMonkey(object):
InfectionMonkey.close_tunnel() InfectionMonkey.close_tunnel()
firewall.close() firewall.close()
else: else:
StateTelem(False).send() # Signal the server (before closing the tunnel) StateTelem(is_done=True).send() # Signal the server (before closing the tunnel)
InfectionMonkey.close_tunnel() InfectionMonkey.close_tunnel()
firewall.close() firewall.close()
if WormConfiguration.send_log_to_server: if WormConfiguration.send_log_to_server: