handle stop signal to be able restart process without breaking current jobs

Author: doomedraven

changelog.md

@@ -1,3 +1,7 @@
+### [29-02-2020]
+* SIGHUP handling to stop submitting tasks and stop cuckoo.py, useful for when you need to reload it without breaking running jobs
+    * `ps aux|grep "python3 cuckoo.py"|cut -d" " -f 5| xargs kill -1`
+
 ### [22-01-2020]
 * Add qemu.py with support for x64/x86/MIPS/MIPSEL/ARM/ARMWRT/ARM64/PowerPC/PowerPC64/Sparc/Sparc64
 * Basic linux integration is done thanks to @enzok

lib/cuckoo/core/scheduler.py

@@ -7,6 +7,7 @@
 import gc
 import time
 import shutil
+import signal
 import logging
 import threading
 import queue
@@ -624,6 +625,10 @@ def __init__(self, maxcount=None):
         self.maxcount = maxcount
         self.total_analysis_count = 0
 
+    def set_stop_analyzing(self, signum, stack):
+        self.running = False
+        log.info("Now wait till all running tasks are completed")
+
     def initialize(self):
         """Initialize the machine manager."""
         global machinery, machine_lock
@@ -715,6 +720,9 @@ def start(self):
 
         log.info("Waiting for analysis tasks.")
 
+        #To handle stop analyzing when we need to restart process without break tasks
+        signal.signal(signal.SIGHUP, self.set_stop_analyzing)
+
         # Message queue with threads to transmit exceptions (used as IPC).
         errors = queue.Queue()