fix: add process whitelist to _free_port, add worker startup interval config

Author: Mag1cFall

src/manager/service.py

@@ -76,6 +76,7 @@ def load_config(self) -> Dict[str, Any]:
             "launch_mode": "headless",
             "script_injection_enabled": False,
             "worker_mode_enabled": False,
+            "worker_startup_interval": 5,
             "log_enabled": True,
         }
 
@@ -450,7 +451,7 @@ async def _start_worker_mode(
             started_count += 1
             logger.info(f"启动Worker {worker_id} (端口:{worker.port})")
             if index < len(worker_ids) - 1:
-                await asyncio.sleep(15)
+                await asyncio.sleep(config.get("worker_startup_interval", 5))
 
         if self.stop_event.is_set():
             self.service_status = "stopped"

src/static/dashboard.html

@@ -410,6 +410,13 @@ <h2 class="text-2xl font-bold mb-6 text-white border-b border-[#30363d] pb-2">{{
                         </div>
                     </div>
 
+                    <div v-if="config.worker_mode_enabled" class="bg-[#161b22] p-4 rounded-lg border border-[#30363d]">
+                        <label class="block text-sm font-medium text-gray-300 mb-2">{{ t('config.workerStartupInterval') }}</label>
+                        <input v-model.number="config.worker_startup_interval" type="number" min="1" max="120"
+                            class="w-full bg-[#0d1117] border border-[#30363d] rounded px-3 py-2 text-white focus:border-blue-500 focus:outline-none transition">
+                        <p class="text-xs text-gray-600 mt-1">{{ t('config.workerStartupIntervalDesc') }}</p>
+                    </div>
+
                     <div class="flex justify-end pt-4">
                         <button @click="saveConfig"
                             class="px-6 py-2 bg-blue-600 hover:bg-blue-500 text-white rounded font-medium transition shadow-lg">{{
@@ -1047,7 +1054,11 @@ <h3 class="text-lg font-bold text-white mb-4">{{ currentPrompt.text }}</h3>
 
                 const fetchConfig = async () => {
                     const res = await fetch('/api/config');
-                    config.value = await res.json();
+                    const data = await res.json();
+                    if (data.worker_startup_interval === undefined) {
+                        data.worker_startup_interval = 5;
+                    }
+                    config.value = data;
                 };
 
                 const saveConfig = async () => {

src/static/i18n.js

@@ -57,6 +57,8 @@ const locales = {
             scriptInjectionDesc: '启用后可添加 AI Studio 未列出的模型（已被弃用）',
             logEnabled: '启用日志',
             logEnabledDesc: '禁用日志可提升性能（需重启服务生效）',
+            workerStartupInterval: 'Worker 启动间隔（秒）',
+            workerStartupIntervalDesc: '多个 Worker 之间依次启动的等待时间，默认 5 秒',
             save: '保存配置'
         },
         auth: {
@@ -209,6 +211,8 @@ const locales = {
             scriptInjectionDesc: '啟用後可添加 AI Studio 未列出的模型（已被棄用）',
             logEnabled: '啟用日誌',
             logEnabledDesc: '禁用日誌可提升性能（需重啟服務生效）',
+            workerStartupInterval: 'Worker 啟動間隔（秒）',
+            workerStartupIntervalDesc: '多個 Worker 之間依次啟動的等待時間，預設 5 秒',
             save: '儲存設定'
         },
         auth: {
@@ -361,6 +365,8 @@ const locales = {
             scriptInjectionDesc: 'Enable to add unlisted models in AI Studio (Deprecated)',
             logEnabled: 'Enable Logging',
             logEnabledDesc: 'Disabling logs improves performance (requires service restart)',
+            workerStartupInterval: 'Worker Startup Interval (seconds)',
+            workerStartupIntervalDesc: 'Delay between starting each Worker, default 5 seconds',
             save: 'Save Config'
         },
         auth: {
@@ -513,6 +519,8 @@ const locales = {
             scriptInjectionDesc: '有効にするとAI Studioに未掲載のモデルを追加できます（非推奨）',
             logEnabled: 'ログを有効にする',
             logEnabledDesc: 'ログを無効にするとパフォーマンスが向上します（サービス再起動が必要）',
+            workerStartupInterval: 'Worker起動間隔（秒）',
+            workerStartupIntervalDesc: '各Workerの起動間隔、デフォルト5秒',
             save: '設定を保存'
         },
         auth: {
@@ -665,6 +673,8 @@ const locales = {
             scriptInjectionDesc: '활성화하면 AI Studio에 나열되지 않은 모델 추가 가능 (더 이상 사용되지 않음)',
             logEnabled: '로깅 활성화',
             logEnabledDesc: '로그 비활성화 시 성능 향상 (서비스 재시작 필요)',
+            workerStartupInterval: 'Worker 시작 간격 (초)',
+            workerStartupIntervalDesc: '각 Worker 시작 사이의 대기 시간, 기본 5초',
             save: '설정 저장'
         },
         auth: {

src/worker/pool.py

@@ -244,8 +244,9 @@ def _cancel_restart(self, worker_id: str):
         if task and not task.done():
             task.cancel()
 
+    _SAFE_PROCESS_NAMES = {"python", "python3", "pythonw", "camoufox", "firefox", "uv"}
+
     def _free_port(self, port: int) -> None:
-        """Kill any process occupying the given TCP port."""
         try:
             if platform.system() == "Windows":
                 result = subprocess.run(
@@ -257,20 +258,36 @@ def _free_port(self, port: int) -> None:
                         parts = line.split()
                         pid = parts[-1]
                         if pid.isdigit():
-                            subprocess.run(
-                                ["taskkill", "/PID", pid, "/F"],
-                                capture_output=True
+                            proc_check = subprocess.run(
+                                ["tasklist", "/FI", f"PID eq {pid}", "/FO", "CSV", "/NH"],
+                                capture_output=True, text=True
                             )
-                            logger.info(f"Freed port {port} (killed PID {pid})")
+                            proc_name = proc_check.stdout.strip().split(",")[0].strip('"').lower().replace(".exe", "")
+                            if proc_name in self._SAFE_PROCESS_NAMES:
+                                subprocess.run(
+                                    ["taskkill", "/PID", pid, "/F"],
+                                    capture_output=True
+                                )
+                                logger.info(f"Freed port {port} (killed {proc_name} PID {pid})")
+                            else:
+                                logger.warning(f"Port {port} occupied by {proc_name} (PID {pid}), skipping")
             else:
                 result = subprocess.run(
                     ["lsof", "-ti", f"tcp:{port}"],
                     capture_output=True, text=True
                 )
                 for pid in result.stdout.strip().splitlines():
                     if pid.isdigit():
-                        subprocess.run(["kill", "-9", pid], capture_output=True)
-                        logger.info(f"Freed port {port} (killed PID {pid})")
+                        proc_check = subprocess.run(
+                            ["ps", "-p", pid, "-o", "comm="],
+                            capture_output=True, text=True
+                        )
+                        proc_name = proc_check.stdout.strip().lower()
+                        if any(safe in proc_name for safe in self._SAFE_PROCESS_NAMES):
+                            subprocess.run(["kill", "-9", pid], capture_output=True)
+                            logger.info(f"Freed port {port} (killed {proc_name} PID {pid})")
+                        else:
+                            logger.warning(f"Port {port} occupied by {proc_name} (PID {pid}), skipping")
         except Exception as e:
             logger.warning(f"Failed to free port {port}: {e}")