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

Agent: Handle window close event on Windows

This commit is contained in:
Mike Salvatore 2021-11-23 13:09:17 -05:00 committed by VakarisZ
parent 27ef06c546
commit 068307f0eb
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
monkey/infection_monkey/utils/signal_handler.py
View File

@ -12,7 +12,7 @@ class StopSignalHandler:
def __init__(self, master: IMaster): def __init__(self, master: IMaster):
self._master = master self._master = master
def __call__(self, _, __): def __call__(self, _, __=None):
self._master.terminate() self._master.terminate()
logger.debug("Some kind of interrupt signal was sent to the Monkey Agent") logger.debug("Some kind of interrupt signal was sent to the Monkey Agent")
raise PlannedShutdownException("Monkey Agent got an interrupt signal") raise PlannedShutdownException("Monkey Agent got an interrupt signal")
@ -24,4 +24,7 @@ def register_signal_handlers(master: IMaster):
signal.signal(signal.SIGTERM, stop_signal_handler) signal.signal(signal.SIGTERM, stop_signal_handler)
if is_windows_os(): if is_windows_os():
import win32api
signal.signal(signal.SIGBREAK, stop_signal_handler) signal.signal(signal.SIGBREAK, stop_signal_handler)
win32api.SetConsoleCtrlHandler(stop_signal_handler, True)