]> git.ipfire.org Git - thirdparty/suricata-update.git/commitdiff
Catch Keyboard Interrupt and exit cleanly.
authorVrinda Narayan <vrinda18120@iiitd.ac.in>
Wed, 13 Mar 2019 09:53:43 +0000 (15:23 +0530)
committerShivani Bhardwaj <shivanib134@gmail.com>
Mon, 5 Aug 2019 14:22:51 +0000 (19:52 +0530)
Optimization #2878 https://redmine.openinfosecfoundation.org/issues/2878
Can be done by importing python library signal, and defining a function
signal_handler which will print a message if keyboard interrupt is
detected and exit the program.

suricata/update/main.py

index ab995e3d10d3ab83fe3f2fcea9b1193ddfb42834..ac5ab1596a473e769506ff778aa0b59c81dc212a 100644 (file)
@@ -32,6 +32,7 @@ import shutil
 import glob
 import io
 import tempfile
+import signal
 
 try:
     # Python 3.
@@ -1472,7 +1473,12 @@ def _main():
 
     return 0
 
+def signal_handler(signal, frame):
+    print('Program interrupted. Aborting...')
+    sys.exit(1)
+
 def main():
+    signal.signal(signal.SIGINT, signal_handler)
     try:
         sys.exit(_main())
     except exceptions.ApplicationError as err: