Update slowloris.py

Does a check if Socks library exists
biadabbbbb · Nov 30, 2016 · 29ebfbc · 29ebfbc
1 parent 354c0ce
commit 29ebfbc
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/slowloris.py b/slowloris.py
@@ -25,10 +25,12 @@
 
 if args.useproxy:
     print("Using SOCKS5 proxy for connecting...")
-    import socks
-    socks.setdefaultproxy(socks.PROXY_TYPE_SOCKS5, args.proxy_host, args.proxy_port)
-    socket.socket = socks.socksocket
-
+    try:
+        import socks
+        socks.setdefaultproxy(socks.PROXY_TYPE_SOCKS5, args.proxy_host, args.proxy_port)
+        socket.socket = socks.socksocket
+    except ImportError:
+        print("Socks Proxy Library Not Available!")
 if args.verbose == True:
     logging.basicConfig(format="[%(asctime)s] %(message)s", datefmt="%d-%m-%Y %H:%M:%S", level=logging.DEBUG)
 else: