cdp-tunnel 2.5.3 → 2.5.5

package/extension-new/core/websocket.js

@@ -191,6 +191,7 @@ var WebSocketManager = (function() {
         Logger.info('[WS] Client connected, resuming event forwarding');
         State.setHasConnectedClient(true);
         State.addCDPClient(message.clientId, message.clientId);
+        startGroupMonitor();
         broadcastStateUpdate();
         break;
 
@@ -214,6 +215,7 @@ var WebSocketManager = (function() {
           State.removeCDPClient(discClientId);
           if (State.getCDPClients().length === 0) {
             State.setHasConnectedClient(false);
+            stopGroupMonitor();
           }
           broadcastStateUpdate();
         });
@@ -223,6 +225,11 @@ var WebSocketManager = (function() {
         Logger.info('[WS] Received client list:', message.clients);
         State.setCDPClients(message.clients || []);
         State.setHasConnectedClient((message.clients || []).length > 0);
+        if ((message.clients || []).length > 0) {
+          startGroupMonitor();
+        } else {
+          stopGroupMonitor();
+        }
         broadcastStateUpdate();
         break;
 
@@ -378,6 +385,49 @@ var WebSocketManager = (function() {
     });
   }
 
+  var _groupMonitorTimer = null;
+
+  function startGroupMonitor() {
+    Logger.info('[Monitor] Starting group monitor...');
+    if (_groupMonitorTimer) clearInterval(_groupMonitorTimer);
+    _groupMonitorTimer = setInterval(function() {
+      var attached = State.getAttachedTabIds();
+      attached.forEach(function(tabId) {
+        var clientId = State.getClientIdByTabId(tabId);
+        if (!clientId) return;
+        if (State.isPreExistingTab(tabId)) return;
+        chrome.tabs.get(tabId, function(tab) {
+          if (chrome.runtime.lastError || !tab) return;
+          if (tab.groupId) return;
+          Logger.info('[Monitor] Tab', tabId, 'has no group, regrouping for client:', clientId);
+          var baseName = CDPUtils.getGroupBaseName(clientId);
+          var groupId = State.getGroupIdForClient(clientId);
+          if (groupId) {
+            chrome.tabs.group({ tabIds: tabId, groupId: groupId }, function() {});
+          } else {
+            chrome.tabs.group({ tabIds: tabId }, function(newGroupId) {
+              if (chrome.runtime.lastError || !newGroupId) return;
+              chrome.tabGroups.update(newGroupId, {
+                title: baseName,
+                color: CDPUtils.getGroupColorForClient(clientId),
+                collapsed: true
+              }, function() {
+                State.setGroupIdForClient(clientId, newGroupId);
+              });
+            });
+          }
+        });
+      });
+    }, 5000);
+  }
+
+  function stopGroupMonitor() {
+    if (_groupMonitorTimer) {
+      clearInterval(_groupMonitorTimer);
+      _groupMonitorTimer = null;
+    }
+  }
+
   function handleServerRestart() {
     Logger.info('[WS] Server restarted, cleaning up all state...');
 

package/extension-new/manifest.json

@@ -1,7 +1,7 @@
 {
   "manifest_version": 3,
   "name": "CDP Bridge",
-  "version": "2.5.3",
+  "version": "2.5.5",
   "description": "Chrome DevTools Protocol Bridge for Playwright/Puppeteer automation",
   "permissions": [
     "debugger",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cdp-tunnel",
-  "version": "2.5.3",
+  "version": "2.5.5",
   "description": "Bridge Chrome's debugger API to WebSocket — control your existing browser with Playwright/Puppeteer via CDP",
   "main": "server/proxy-server.js",
   "bin": "./cli/index.js",

package/server/proxy-server.js

@@ -890,10 +890,12 @@ function handleClientConnection(ws, clientInfo, customClientId = null) {
 
         // 为每个请求分配全局唯一 ID，避免多客户端 ID 冲突
         let modifiedData = data;
+        let originalId = null;
+        let globalId = null;
         if (parsed && parsed.id !== undefined) {
-            const originalId = parsed.id;
+            originalId = parsed.id;
             globalRequestIdCounter++;
-            const globalId = globalRequestIdCounter;
+            globalId = globalRequestIdCounter;
             
             // 保存映射：全局ID -> {clientId, originalId, sessionId}
             // 如果请求有 sessionId，也保存它，用于响应路由
@@ -922,11 +924,36 @@ function handleClientConnection(ws, clientInfo, customClientId = null) {
             }
             console.log(`[PENDING CREATE TARGET] Request id=${parsed.id} from client=${id}`);
         }
-        if (parsed && parsed.method === 'Target.closeTarget' && parsed.id !== undefined) {
-            const currentMapping = globalRequestIdMap.get(parsed.id);
-            if (currentMapping) {
-                currentMapping.method = 'Target.closeTarget';
-                currentMapping.closeTargetId = parsed.params?.targetId;
+        if (parsed && parsed.id !== undefined) {
+            if (parsed.method === 'Target.closeTarget') {
+                const targetId = parsed.params?.targetId;
+                const ownerClient = targetId ? targetIdToClientId.get(targetId) : null;
+                if (ownerClient && ownerClient !== id) {
+                    console.log(`[BLOCKED] ${parsed.method} targetId=${targetId?.substring(0,8)} owner=${ownerClient?.substring(0,8)} requester=${id?.substring(0,8)} — not owner`);
+                    const errMsg = JSON.stringify({
+                        id: originalId,
+                        error: { code: -32000, message: 'Target is owned by another client' }
+                    });
+                    safeSend(ws, errMsg, 'client');
+                    return;
+                }
+                const currentMapping = globalRequestIdMap.get(parsed.id);
+                if (currentMapping) {
+                    currentMapping.method = 'Target.closeTarget';
+                    currentMapping.closeTargetId = targetId;
+                }
+            } else if (parsed.method === 'Target.attachToTarget') {
+                const targetId = parsed.params?.targetId;
+                const ownerClient = targetId ? targetIdToClientId.get(targetId) : null;
+                if (ownerClient && ownerClient !== id) {
+                    console.log(`[BLOCKED] ${parsed.method} targetId=${targetId?.substring(0,8)} owner=${ownerClient?.substring(0,8)} requester=${id?.substring(0,8)} — not owner`);
+                    const errMsg = JSON.stringify({
+                        id: originalId,
+                        error: { code: -32000, message: 'Target is owned by another client' }
+                    });
+                    safeSend(ws, errMsg, 'client');
+                    return;
+                }
             }
         }