cdp-tunnel 3.0.5 → 3.0.7

package/extension-new/manifest.json

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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cdp-tunnel",
-  "version": "3.0.5",
+  "version": "3.0.7",
   "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/modules/port-pool.js

@@ -64,28 +64,19 @@ class PortPoolManager {
     const session = new PortPoolManager.PortSession(portIndex, port);
     this.portSessions[portIndex] = session;
 
-    const server = http.createServer((req, res) => {
-      this._handleHttp(req, res, session);
-    });
-
-    // 每个 create 端口用独立的 wss（不复用主 proxy 的 wss）
-    const wss = new WebSocket.Server({ noServer: true });
-    this.createWss[portIndex] = wss;
+    const server = http.createServer();
+    this.createServers[portIndex] = server;
 
+    // 端口池端口复用主 proxy 的 wss——通过 localPort 区分
     server.on('upgrade', (req, socket, head) => {
-      const url = new URL(req.url, `http://localhost:${port}`);
-      const path = url.pathname;
-
-      // 只允许 client 连接（plugin 连的是主 proxy 的 9221）
-      if (path !== '/client' && !path.startsWith('/client/') &&
-          !path.startsWith('/devtools/browser/') && !path.startsWith('/devtools/page/')) {
-        socket.destroy();
-        return;
-      }
+      req._poolPortIndex = portIndex;
+      req._poolPort = port;
+      // 转发给主 proxy 的 wss 处理
+      this.mainProxy.handlePoolUpgrade(req, socket, head, portIndex, port);
+    });
 
-      wss.handleUpgrade(req, socket, head, (ws) => {
-        this._handleClientConnect(ws, req, session);
-      });
+    server.on('request', (req, res) => {
+      this._handleHttp(req, res, session);
     });
 
     server.on('error', (err) => {
@@ -101,7 +92,6 @@ class PortPoolManager {
     });
 
     this.createServers[portIndex] = server;
-    this.createWss[portIndex] = wss;
   }
 
   _startTakeoverPort() {
@@ -174,7 +164,6 @@ class PortPoolManager {
    */
   _handleClientConnect(ws, req, session) {
     session.clients.add(ws);
-    console.log(`[PORT ${session.port}] Client connected (total: ${session.clients.size})`);
 
     // 找到 plugin 连接（从主 proxy 获取）
     const pluginWs = this.mainProxy.getPluginConnection();
@@ -197,6 +186,10 @@ class PortPoolManager {
           await this._ensureVisible(session, pluginWs, msg.sessionId);
         }
 
+        if (msg.method === 'Network.enable' || msg.method === 'Page.startScreencast') {
+          // 已转发，继续
+        }
+
         // 命令：分配新 id，记录映射（含 method 名供响应后处理），转发给 plugin
         const newId = `pool${session.portIndex}_${msg.id}`;
         session.pendingRequests.set(newId, {

package/server/proxy-server.js

@@ -456,6 +456,15 @@ wss.on('connection', (ws, req) => {
     const path = url.pathname;
     const pathParts = path.split('/').filter(Boolean);
 
+    // v3.0 端口池连接：走 PortPoolManager 的隔离逻辑
+    if (req._poolPortIndex !== undefined && portPool) {
+        const session = portPool.portSessions[req._poolPortIndex];
+        if (session) {
+            portPool._handleClientConnect(ws, req, session);
+            return;
+        }
+    }
+
     const clientInfo = {
         ip: req.socket.remoteAddress,
         port: req.socket.remotePort
@@ -812,7 +821,7 @@ function handlePluginConnection(ws, clientInfo, request) {
 
         // v3.0 端口池 hook：先让 PortPoolManager 处理端口池的消息
         if (parsed && portPool && portPool.handlePluginMessage(parsed, ws)) {
-            return;  // 已被端口池处理
+            return;
         }
 
         // 处理 keepalive 消息
@@ -2130,6 +2139,12 @@ portPool = new PortPoolManager({
         for (const ws of pluginConnections) return ws;
         return null;
     },
+    handlePoolUpgrade: (req, socket, head, portIndex, port) => {
+        req._poolPortIndex = portIndex;
+        wss.handleUpgrade(req, socket, head, (ws) => {
+            wss.emit('connection', ws, req);
+        });
+    },
     getAllTargets: async (portIndex) => {
         const session = portPool.portSessions[portIndex];
         if (!session) return [];