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):
self._master = master
def __call__(self, _, __):
def __call__(self, _, __=None):
self._master.terminate()
logger.debug("Some kind of interrupt signal was sent to the Monkey Agent")
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)
if is_windows_os():
import win32api
signal.signal(signal.SIGBREAK, stop_signal_handler)
win32api.SetConsoleCtrlHandler(stop_signal_handler, True)