p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Agent: add a comment warning that windows will terminate the process 5s after CTRL_CLOSE_EVENT signal 

The comment will warn us that in case that particular signal is raised, the cleanup shouldn't take longer than 5s
This commit is contained in:
VakarisZ 2021-11-24 10:40:05 +02:00
parent 73329e9729
commit 3f7c4a8859
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, signum, __=None): def __call__(self, signum, _=None):
logger.info(f"The Monkey Agent received signal {signum}") logger.info(f"The Monkey Agent received signal {signum}")
self._master.terminate() self._master.terminate()
raise PlannedShutdownException("Monkey Agent got an interrupt signal") raise PlannedShutdownException("Monkey Agent got an interrupt signal")
@ -27,4 +27,7 @@ def register_signal_handlers(master: IMaster):
import win32api import win32api
signal.signal(signal.SIGBREAK, stop_signal_handler) signal.signal(signal.SIGBREAK, stop_signal_handler)
# CTRL_CLOSE_EVENT signal has a timeout of 5000ms,
# after that OS will forcefully kill the process
win32api.SetConsoleCtrlHandler(stop_signal_handler, True) win32api.SetConsoleCtrlHandler(stop_signal_handler, True)