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

Agent: Use a finally clause to ensure singleton is always unlocked

This commit is contained in:
Mike Salvatore 2022-09-07 19:45:12 -04:00
parent 6c2cab9950
commit 1d58ce36e7
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
monkey/infection_monkey/monkey.py
View File

@ -415,11 +415,12 @@ class InfectionMonkey:
).send() # Signal the server (before closing the tunnel) ).send() # Signal the server (before closing the tunnel)
self._close_tunnel() self._close_tunnel()
self._singleton.unlock()
except Exception as e: except Exception as e:
logger.error(f"An error occurred while cleaning up the monkey agent: {e}") logger.error(f"An error occurred while cleaning up the monkey agent: {e}")
if deleted is None: if deleted is None:
InfectionMonkey._self_delete() InfectionMonkey._self_delete()
finally:
self._singleton.unlock()
logger.info("Monkey is shutting down") logger.info("Monkey is shutting down")