@feng3d/cts 0.0.6

package/dist/handlers/control-handler.js

@@ -0,0 +1,378 @@
+/**
+ * @module control-handler
+ * @description 控制通道处理器模块，负责处理客户端通过 WebSocket 控制通道发送的各类控制消息。
+ * 包括认证、端口注册/注销、心跳以及断开连接等消息的处理逻辑。
+ */
+import { WebSocket } from 'ws';
+import { MessageType, createMessage, logger, } from '@feng3d/chuantou-shared';
+/**
+ * 控制通道处理器
+ *
+ * 处理客户端通过 WebSocket 控制通道发送的各类控制消息，包括：
+ * - 客户端认证（AUTH）
+ * - 端口注册（REGISTER）与注销（UNREGISTER）
+ * - 心跳保活（HEARTBEAT）
+ * - 连接断开的清理工作
+ */
+export class ControlHandler {
+    /**
+     * 创建控制通道处理器实例
+     *
+     * @param sessionManager - 会话管理器，用于管理客户端会话
+     * @param config - 服务器配置，包含认证令牌等信息
+     * @param proxyHandler - 统一代理处理器，用于启停代理服务器
+     */
+    constructor(sessionManager, config, proxyHandler) {
+        this.sessionManager = sessionManager;
+        this.config = config;
+        this.proxyHandler = proxyHandler;
+    }
+    /**
+     * 处理新的 WebSocket 控制通道连接
+     *
+     * 为新连接创建会话，设置消息处理、关闭处理和错误处理回调，
+     * 并启动 30 秒认证超时计时器。
+     *
+     * @param socket - 新建立的 WebSocket 连接
+     */
+    handleConnection(socket) {
+        const clientId = this.sessionManager.createSession(socket);
+        logger.log(`新的控制连接来自客户端: ${clientId}`);
+        // 设置消息处理器
+        socket.on('message', (data) => {
+            this.handleMessage(clientId, socket, data).catch((error) => {
+                logger.error(`处理来自 ${clientId} 的消息时出错:`, error);
+                this.sendError(socket, `内部错误: ${error.message}`);
+            });
+        });
+        // 设置关闭处理器
+        socket.on('close', () => {
+            logger.log(`控制连接已关闭: ${clientId}`);
+            this.handleDisconnect(clientId);
+        });
+        // 设置错误处理器
+        socket.on('error', (error) => {
+            logger.error(`客户端 ${clientId} 的 Socket 错误:`, error);
+        });
+        // 设置认证超时
+        const authTimeout = setTimeout(() => {
+            const clientInfo = this.sessionManager.getClientInfo(clientId);
+            if (clientInfo && !clientInfo.authenticated) {
+                logger.log(`客户端 ${clientId} 认证超时`);
+                socket.close();
+            }
+        }, 30000); // 30秒认证超时
+        // 存储超时引用，以便在认证成功后清除
+        socket._authTimeout = authTimeout;
+    }
+    /**
+     * 处理收到的控制消息
+     *
+     * 解析 JSON 消息并根据消息类型分发到对应的处理方法。
+     * 支持的消息类型包括：AUTH、REGISTER、UNREGISTER、HEARTBEAT。
+     *
+     * @param clientId - 发送消息的客户端 ID
+     * @param socket - 客户端的 WebSocket 连接
+     * @param data - 收到的原始消息数据
+     */
+    async handleMessage(clientId, socket, data) {
+        try {
+            const message = JSON.parse(data.toString());
+            const msgType = message.type;
+            logger.log(`收到来自 ${clientId} 的消息: ${msgType}`);
+            switch (msgType) {
+                case MessageType.AUTH:
+                    await this.handleAuth(clientId, socket, message);
+                    break;
+                case MessageType.REGISTER:
+                    await this.handleRegister(clientId, socket, message);
+                    break;
+                case MessageType.UNREGISTER:
+                    await this.handleUnregister(clientId, socket, message);
+                    break;
+                case MessageType.HEARTBEAT:
+                    await this.handleHeartbeat(clientId, socket, message);
+                    break;
+                case 'http_response':
+                case 'http_response_headers':
+                case 'http_response_body':
+                    // 普通HTTP响应或流式响应头/体
+                    this.handleClientResponse(message.connectionId, message);
+                    break;
+                case 'http_response_data':
+                    // 流式响应数据块
+                    this.handleClientStreamData(message.connectionId, Buffer.from(message.data, 'base64'));
+                    break;
+                case 'http_response_end':
+                    // 流式响应结束
+                    this.handleClientStreamEnd(message.connectionId);
+                    break;
+                case 'connection_data':
+                    // 客户端发送的格式: { type: 'connection_data', connectionId, data }
+                    this.handleClientData(message.connectionId, Buffer.from(message.data, 'base64'));
+                    break;
+                case MessageType.CONNECTION_CLOSE:
+                    this.handleClientClose(message.payload.connectionId);
+                    break;
+                default:
+                    logger.warn(`未知的消息类型: ${msgType}`);
+                    this.sendError(socket, `未知的消息类型: ${msgType}`);
+            }
+        }
+        catch (error) {
+            logger.error(`解析来自 ${clientId} 的消息时出错:`, error);
+            this.sendError(socket, '无效的消息格式');
+        }
+    }
+    /**
+     * 处理认证消息
+     *
+     * 验证客户端提供的令牌是否在配置的有效令牌列表中。
+     * 认证成功后清除认证超时计时器；认证失败则关闭连接。
+     *
+     * @param clientId - 客户端唯一标识 ID
+     * @param socket - 客户端的 WebSocket 连接
+     * @param message - 收到的认证消息
+     */
+    async handleAuth(clientId, socket, message) {
+        const { token } = message.payload;
+        if (!token) {
+            this.sendMessage(socket, createMessage(MessageType.AUTH_RESP, {
+                success: false,
+                error: '令牌不能为空',
+            }, message.id));
+            socket.close();
+            return;
+        }
+        if (!this.config.authTokens.includes(token)) {
+            this.sendMessage(socket, createMessage(MessageType.AUTH_RESP, {
+                success: false,
+                error: '无效的令牌',
+            }, message.id));
+            socket.close();
+            return;
+        }
+        const authenticated = this.sessionManager.authenticateClient(clientId);
+        if (authenticated) {
+            // 清除认证超时
+            const authTimeout = socket._authTimeout;
+            if (authTimeout) {
+                clearTimeout(authTimeout);
+                delete socket._authTimeout;
+            }
+            this.sendMessage(socket, createMessage(MessageType.AUTH_RESP, {
+                success: true,
+            }, message.id));
+            logger.log(`客户端 ${clientId} 认证成功`);
+        }
+        else {
+            this.sendMessage(socket, createMessage(MessageType.AUTH_RESP, {
+                success: false,
+                error: '认证失败',
+            }, message.id));
+            socket.close();
+        }
+    }
+    /**
+     * 处理端口注册消息
+     *
+     * 验证客户端认证状态和端口范围（1024-65535），检查端口是否已被占用，
+     * 然后注册端口并启动代理服务器（同时支持 HTTP 和 WebSocket）。
+     * 若启动失败则自动回滚端口注册。
+     *
+     * @param clientId - 客户端唯一标识 ID
+     * @param socket - 客户端的 WebSocket 连接
+     * @param message - 收到的端口注册消息
+     */
+    async handleRegister(clientId, socket, message) {
+        const clientInfo = this.sessionManager.getClientInfo(clientId);
+        if (!clientInfo || !clientInfo.authenticated) {
+            this.sendMessage(socket, createMessage(MessageType.REGISTER_RESP, {
+                success: false,
+                error: '未认证',
+            }, message.id));
+            return;
+        }
+        const { remotePort, localPort, localHost } = message.payload;
+        // 验证端口
+        if (remotePort < 1024 || remotePort > 65535) {
+            this.sendMessage(socket, createMessage(MessageType.REGISTER_RESP, {
+                success: false,
+                error: '端口超出范围 (1024-65535)',
+            }, message.id));
+            return;
+        }
+        // 检查端口是否已被注册
+        const existingClientId = this.sessionManager.getClientByPort(remotePort);
+        if (existingClientId) {
+            this.sendMessage(socket, createMessage(MessageType.REGISTER_RESP, {
+                success: false,
+                error: '端口已被注册',
+            }, message.id));
+            return;
+        }
+        // 注册端口
+        const registered = this.sessionManager.registerPort(clientId, remotePort);
+        if (!registered) {
+            this.sendMessage(socket, createMessage(MessageType.REGISTER_RESP, {
+                success: false,
+                error: '端口注册失败',
+            }, message.id));
+            return;
+        }
+        // 启动代理服务器（同时支持 HTTP 和 WebSocket）
+        try {
+            await this.proxyHandler.startProxy(remotePort, clientId);
+            this.sendMessage(socket, createMessage(MessageType.REGISTER_RESP, {
+                success: true,
+                remotePort,
+                remoteUrl: `http://${this.config.host}:${remotePort}`,
+            }, message.id));
+            logger.log(`客户端 ${clientId} 注册了代理: 端口 ${remotePort} -> ${localHost || 'localhost'}:${localPort}`);
+        }
+        catch (error) {
+            // 启动代理失败，回滚端口注册
+            this.sessionManager.unregisterPort(clientId, remotePort);
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            this.sendMessage(socket, createMessage(MessageType.REGISTER_RESP, {
+                success: false,
+                error: `启动代理失败: ${errorMessage}`,
+            }, message.id));
+        }
+    }
+    /**
+     * 处理端口注销消息
+     *
+     * 验证客户端认证状态和端口归属，停止代理服务器并注销端口。
+     *
+     * @param clientId - 客户端唯一标识 ID
+     * @param socket - 客户端的 WebSocket 连接
+     * @param message - 收到的端口注销消息
+     */
+    async handleUnregister(clientId, socket, message) {
+        const clientInfo = this.sessionManager.getClientInfo(clientId);
+        if (!clientInfo || !clientInfo.authenticated) {
+            return;
+        }
+        const { remotePort } = message.payload;
+        // 检查端口是否属于该客户端
+        if (!clientInfo.registeredPorts.has(remotePort)) {
+            this.sendMessage(socket, createMessage(MessageType.REGISTER_RESP, {
+                success: false,
+                error: '该端口未被此客户端注册',
+            }, message.id));
+            return;
+        }
+        // 停止代理
+        await this.proxyHandler.stopProxy(remotePort);
+        // 注销端口
+        const unregistered = this.sessionManager.unregisterPort(clientId, remotePort);
+        if (unregistered) {
+            logger.log(`客户端 ${clientId} 注销了端口 ${remotePort}`);
+        }
+    }
+    /**
+     * 处理心跳消息
+     *
+     * 更新客户端的最后心跳时间并回复心跳响应。
+     *
+     * @param clientId - 客户端唯一标识 ID
+     * @param socket - 客户端的 WebSocket 连接
+     * @param message - 收到的心跳消息
+     */
+    async handleHeartbeat(clientId, socket, message) {
+        this.sessionManager.updateHeartbeat(clientId);
+        this.sendMessage(socket, createMessage(MessageType.HEARTBEAT_RESP, {
+            timestamp: Date.now(),
+        }, message.id));
+    }
+    /**
+     * 处理客户端断开连接
+     *
+     * 停止该客户端注册的所有代理服务器，并移除其会话。
+     *
+     * @param clientId - 断开连接的客户端唯一标识 ID
+     */
+    handleDisconnect(clientId) {
+        const clientInfo = this.sessionManager.getClientInfo(clientId);
+        if (clientInfo) {
+            // 停止所有代理
+            for (const port of clientInfo.registeredPorts) {
+                this.proxyHandler.stopProxy(port).catch(logger.error);
+            }
+        }
+        this.sessionManager.removeSession(clientId);
+    }
+    /**
+     * 通过 WebSocket 发送消息
+     *
+     * 将消息对象序列化为 JSON 字符串后发送，仅在连接处于 OPEN 状态时发送。
+     *
+     * @param socket - 目标 WebSocket 连接
+     * @param message - 需要发送的消息对象
+     */
+    sendMessage(socket, message) {
+        if (socket.readyState === WebSocket.OPEN) {
+            socket.send(JSON.stringify(message));
+        }
+    }
+    /**
+     * 发送错误消息
+     *
+     * 构造并发送一条 CONNECTION_ERROR 类型的错误消息给客户端。
+     *
+     * @param socket - 目标 WebSocket 连接
+     * @param error - 错误描述信息
+     */
+    sendError(socket, error) {
+        this.sendMessage(socket, createMessage(MessageType.CONNECTION_ERROR, {
+            connectionId: '',
+            error,
+        }));
+    }
+    /**
+     * 处理客户端返回的 HTTP 响应数据
+     *
+     * @param connectionId - 连接唯一标识 ID
+     * @param data - 客户端返回的 HTTP 响应数据
+     */
+    handleClientResponse(connectionId, data) {
+        this.proxyHandler.handleClientResponse(connectionId, data);
+    }
+    /**
+     * 处理来自客户端的 WebSocket 数据
+     *
+     * @param connectionId - 连接唯一标识 ID
+     * @param data - 客户端发送的数据
+     */
+    handleClientData(connectionId, data) {
+        this.proxyHandler.handleClientData(connectionId, data);
+    }
+    /**
+     * 处理来自客户端的连接关闭请求
+     *
+     * @param connectionId - 需要关闭的连接唯一标识 ID
+     * @param code - 可选的关闭状态码
+     */
+    handleClientClose(connectionId, code) {
+        this.proxyHandler.handleClientClose(connectionId, code);
+    }
+    /**
+     * 处理来自客户端的流式响应数据
+     *
+     * @param connectionId - 连接唯一标识 ID
+     * @param data - 客户端发送的流式数据
+     */
+    handleClientStreamData(connectionId, data) {
+        this.proxyHandler.handleClientStreamData(connectionId, data);
+    }
+    /**
+     * 处理来自客户端的流式响应结束通知
+     *
+     * @param connectionId - 连接唯一标识 ID
+     */
+    handleClientStreamEnd(connectionId) {
+        this.proxyHandler.handleClientStreamEnd(connectionId);
+    }
+}
+//# sourceMappingURL=control-handler.js.map

package/dist/handlers/control-handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"control-handler.js","sourceRoot":"","sources":["../../src/handlers/control-handler.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC/B,OAAO,EACL,WAAW,EACX,aAAa,EAMb,MAAM,GACP,MAAM,yBAAyB,CAAC;AAIjC;;;;;;;;GAQG;AACH,MAAM,OAAO,cAAc;IAQzB;;;;;;OAMG;IACH,YACE,cAA8B,EAC9B,MAAoB,EACpB,YAAiC;QAEjC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACnC,CAAC;IAED;;;;;;;OAOG;IACH,gBAAgB,CAAC,MAAiB;QAChC,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QAE3D,MAAM,CAAC,GAAG,CAAC,gBAAgB,QAAQ,EAAE,CAAC,CAAC;QAEvC,UAAU;QACV,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAY,EAAE,EAAE;YACpC,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACzD,MAAM,CAAC,KAAK,CAAC,QAAQ,QAAQ,UAAU,EAAE,KAAK,CAAC,CAAC;gBAChD,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,SAAS,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACnD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,UAAU;QACV,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACtB,MAAM,CAAC,GAAG,CAAC,YAAY,QAAQ,EAAE,CAAC,CAAC;YACnC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,UAAU;QACV,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YAC3B,MAAM,CAAC,KAAK,CAAC,OAAO,QAAQ,eAAe,EAAE,KAAK,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,SAAS;QACT,MAAM,WAAW,GAAG,UAAU,CAAC,GAAG,EAAE;YAClC,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;YAC/D,IAAI,UAAU,IAAI,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC;gBAC5C,MAAM,CAAC,GAAG,CAAC,OAAO,QAAQ,OAAO,CAAC,CAAC;gBACnC,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,CAAC;QACH,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,UAAU;QAErB,oBAAoB;QACnB,MAAc,CAAC,YAAY,GAAG,WAAW,CAAC;IAC7C,CAAC;IAED;;;;;;;;;OASG;IACK,KAAK,CAAC,aAAa,CAAC,QAAgB,EAAE,MAAiB,EAAE,IAAY;QAC3E,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;YAC5C,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;YAE7B,MAAM,CAAC,GAAG,CAAC,QAAQ,QAAQ,SAAS,OAAO,EAAE,CAAC,CAAC;YAE/C,QAAQ,OAAO,EAAE,CAAC;gBAChB,KAAK,WAAW,CAAC,IAAI;oBACnB,MAAM,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAsB,CAAC,CAAC;oBAChE,MAAM;gBAER,KAAK,WAAW,CAAC,QAAQ;oBACvB,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,OAA0B,CAAC,CAAC;oBACxE,MAAM;gBAER,KAAK,WAAW,CAAC,UAAU;oBACzB,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,MAAM,EAAE,OAA4B,CAAC,CAAC;oBAC5E,MAAM;gBAER,KAAK,WAAW,CAAC,SAAS;oBACxB,MAAM,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,MAAM,EAAE,OAA2B,CAAC,CAAC;oBAC1E,MAAM;gBAER,KAAK,eAAe,CAAC;gBACrB,KAAK,uBAAuB,CAAC;gBAC7B,KAAK,oBAAoB;oBACvB,mBAAmB;oBACnB,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;oBACzD,MAAM;gBAER,KAAK,oBAAoB;oBACvB,UAAU;oBACV,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;oBACvF,MAAM;gBAER,KAAK,mBAAmB;oBACtB,SAAS;oBACT,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;oBACjD,MAAM;gBAER,KAAK,iBAAiB;oBACpB,4DAA4D;oBAC5D,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;oBACjF,MAAM;gBAER,KAAK,WAAW,CAAC,gBAAgB;oBAC/B,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;oBACrD,MAAM;gBAER;oBACE,MAAM,CAAC,IAAI,CAAC,YAAY,OAAO,EAAE,CAAC,CAAC;oBACnC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,YAAY,OAAO,EAAE,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,QAAQ,QAAQ,UAAU,EAAE,KAAK,CAAC,CAAC;YAChD,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED;;;;;;;;;OASG;IACK,KAAK,CAAC,UAAU,CAAC,QAAgB,EAAE,MAAiB,EAAE,OAAoB;QAChF,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC;QAElC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,WAAW,CAAC,SAAS,EAAE;gBAC5D,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,QAAQ;aAChB,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;YAChB,MAAM,CAAC,KAAK,EAAE,CAAC;YACf,OAAO;QACT,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC5C,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,WAAW,CAAC,SAAS,EAAE;gBAC5D,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,OAAO;aACf,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;YAChB,MAAM,CAAC,KAAK,EAAE,CAAC;YACf,OAAO;QACT,CAAC;QAED,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QACvE,IAAI,aAAa,EAAE,CAAC;YAClB,SAAS;YACT,MAAM,WAAW,GAAI,MAAc,CAAC,YAAY,CAAC;YACjD,IAAI,WAAW,EAAE,CAAC;gBAChB,YAAY,CAAC,WAAW,CAAC,CAAC;gBAC1B,OAAQ,MAAc,CAAC,YAAY,CAAC;YACtC,CAAC;YAED,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,WAAW,CAAC,SAAS,EAAE;gBAC5D,OAAO,EAAE,IAAI;aACd,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;YAChB,MAAM,CAAC,GAAG,CAAC,OAAO,QAAQ,OAAO,CAAC,CAAC;QACrC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,WAAW,CAAC,SAAS,EAAE;gBAC5D,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,MAAM;aACd,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;YAChB,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,CAAC;IACH,CAAC;IAED;;;;;;;;;;OAUG;IACK,KAAK,CAAC,cAAc,CAAC,QAAgB,EAAE,MAAiB,EAAE,OAAwB;QACxF,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC/D,IAAI,CAAC,UAAU,IAAI,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC;YAC7C,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,WAAW,CAAC,aAAa,EAAE;gBAChE,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK;aACb,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;YAChB,OAAO;QACT,CAAC;QAED,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC;QAE7D,OAAO;QACP,IAAI,UAAU,GAAG,IAAI,IAAI,UAAU,GAAG,KAAK,EAAE,CAAC;YAC5C,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,WAAW,CAAC,aAAa,EAAE;gBAChE,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,qBAAqB;aAC7B,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;YAChB,OAAO;QACT,CAAC;QAED,aAAa;QACb,MAAM,gBAAgB,GAAG,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;QACzE,IAAI,gBAAgB,EAAE,CAAC;YACrB,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,WAAW,CAAC,aAAa,EAAE;gBAChE,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,QAAQ;aAChB,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;YAChB,OAAO;QACT,CAAC;QAED,OAAO;QACP,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAC1E,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,WAAW,CAAC,aAAa,EAAE;gBAChE,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,QAAQ;aAChB,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;YAChB,OAAO;QACT,CAAC;QAED,iCAAiC;QACjC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YAEzD,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,WAAW,CAAC,aAAa,EAAE;gBAChE,OAAO,EAAE,IAAI;gBACb,UAAU;gBACV,SAAS,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,UAAU,EAAE;aACtD,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;YAEhB,MAAM,CAAC,GAAG,CAAC,OAAO,QAAQ,cAAc,UAAU,OAAO,SAAS,IAAI,WAAW,IAAI,SAAS,EAAE,CAAC,CAAC;QACpG,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,gBAAgB;YAChB,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACzD,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,WAAW,CAAC,aAAa,EAAE;gBAChE,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,WAAW,YAAY,EAAE;aACjC,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC;IAED;;;;;;;;OAQG;IACK,KAAK,CAAC,gBAAgB,CAAC,QAAgB,EAAE,MAAiB,EAAE,OAA0B;QAC5F,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC/D,IAAI,CAAC,UAAU,IAAI,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC;YAC7C,OAAO;QACT,CAAC;QAED,MAAM,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC;QAEvC,eAAe;QACf,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;YAChD,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,WAAW,CAAC,aAAa,EAAE;gBAChE,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,aAAa;aACrB,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;YAChB,OAAO;QACT,CAAC;QAED,OAAO;QACP,MAAM,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAE9C,OAAO;QACP,MAAM,YAAY,GAAG,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAC9E,IAAI,YAAY,EAAE,CAAC;YACjB,MAAM,CAAC,GAAG,CAAC,OAAO,QAAQ,UAAU,UAAU,EAAE,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAED;;;;;;;;OAQG;IACK,KAAK,CAAC,eAAe,CAAC,QAAgB,EAAE,MAAiB,EAAE,OAAyB;QAC1F,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC9C,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,WAAW,CAAC,cAAc,EAAE;YACjE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;IAClB,CAAC;IAED;;;;;;OAMG;IACK,gBAAgB,CAAC,QAAgB;QACvC,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC/D,IAAI,UAAU,EAAE,CAAC;YACf,SAAS;YACT,KAAK,MAAM,IAAI,IAAI,UAAU,CAAC,eAAe,EAAE,CAAC;gBAC9C,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;QACD,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;IAC9C,CAAC;IAED;;;;;;;OAOG;IACK,WAAW,CAAC,MAAiB,EAAE,OAAY;QACjD,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YACzC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAED;;;;;;;OAOG;IACK,SAAS,CAAC,MAAiB,EAAE,KAAa;QAChD,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,WAAW,CAAC,gBAAgB,EAAE;YACnE,YAAY,EAAE,EAAE;YAChB,KAAK;SACN,CAAC,CAAC,CAAC;IACN,CAAC;IAED;;;;;OAKG;IACH,oBAAoB,CAAC,YAAoB,EAAE,IAAS;QAClD,IAAI,CAAC,YAAY,CAAC,oBAAoB,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;IAC7D,CAAC;IAED;;;;;OAKG;IACH,gBAAgB,CAAC,YAAoB,EAAE,IAAY;QACjD,IAAI,CAAC,YAAY,CAAC,gBAAgB,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;IACzD,CAAC;IAED;;;;;OAKG;IACH,iBAAiB,CAAC,YAAoB,EAAE,IAAa;QACnD,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;IAC1D,CAAC;IAED;;;;;OAKG;IACH,sBAAsB,CAAC,YAAoB,EAAE,IAAY;QACvD,IAAI,CAAC,YAAY,CAAC,sBAAsB,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;IAC/D,CAAC;IAED;;;;OAIG;IACH,qBAAqB,CAAC,YAAoB;QACxC,IAAI,CAAC,YAAY,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC;IACxD,CAAC;CACF"}

package/dist/handlers/unified-proxy.d.ts

@@ -0,0 +1,207 @@
+/**
+ * @module unified-proxy
+ * @description 统一代理处理器模块，在指定端口上同时支持 HTTP 和 WebSocket 代理。
+ * 将外部 HTTP/WebSocket 请求转发给穿透客户端，实现双向数据转发。
+ */
+import { SessionManager } from '../session-manager.js';
+/**
+ * 统一代理处理器
+ *
+ * 管理多个代理服务器实例，每个代理监听一个独立端口，
+ * 同时支持 HTTP 请求和 WebSocket 升级请求。
+ * 当外部请求到达代理端口时，将请求信息通过控制通道转发给对应的穿透客户端。
+ */
+export declare class UnifiedProxyHandler {
+    /** 会话管理器实例 */
+    private sessionManager;
+    /** 代理服务器映射表，键为端口号，值为 HTTP 服务器实例 */
+    private proxies;
+    /** WebSocket 服务器映射表，用于处理 upgrade 事件 */
+    private wsServers;
+    /**
+     * 待处理响应映射表
+     *
+     * 存储等待客户端响应的 Promise 回调和超时定时器，键为连接 ID。
+     */
+    private pendingResponses;
+    /**
+     * 用户 WebSocket 连接映射表
+     *
+     * 存储外部用户的 WebSocket 连接，键为连接 ID，值为用户的 WebSocket 实例。
+     */
+    private userConnections;
+    /**
+     * 流式 HTTP 响应映射表
+     *
+     * 存储正在进行的流式响应（如 SSE）的响应对象。
+     */
+    private streamingResponses;
+    /**
+     * 创建统一代理处理器实例
+     *
+     * @param sessionManager - 会话管理器，用于获取客户端连接和管理连接记录
+     */
+    constructor(sessionManager: SessionManager);
+    /**
+     * 启动代理服务器
+     *
+     * 在指定端口创建并启动一个 HTTP 服务器，该端口同时支持 HTTP 请求和 WebSocket 升级。
+     * 若该端口已存在代理或端口被占用，将抛出异常。
+     *
+     * @param port - 代理监听的端口号
+     * @param clientId - 绑定的客户端唯一标识 ID
+     * @returns 代理服务器启动完成的 Promise
+     * @throws 端口已存在代理或端口被占用时抛出错误
+     */
+    startProxy(port: number, clientId: string): Promise<void>;
+    /**
+     * 处理外部 HTTP 请求
+     *
+     * 将外部 HTTP 请求封装为消息发送给穿透客户端，等待客户端处理后返回响应。
+     * 支持流式响应（如 SSE）和普通 HTTP 响应。
+     *
+     * @param clientId - 处理该请求的客户端唯一标识 ID
+     * @param req - 收到的 HTTP 请求对象
+     * @param res - 用于发送响应的 HTTP 响应对象
+     */
+    private handleHttpRequest;
+    /**
+     * 处理 WebSocket 升级请求
+     *
+     * 将外部 WebSocket 连接桥接到穿透客户端，实现双向实时数据转发。
+     *
+     * @param clientId - 处理该连接的客户端唯一标识 ID
+     * @param req - HTTP 升级请求对象
+     * @param socket - 底层 socket 连接
+     * @param head - 升级请求的头部 buffer
+     */
+    private handleWebSocketUpgrade;
+    /**
+     * 处理 WebSocket 连接
+     *
+     * 当外部用户连接到代理端口时，记录连接信息，通知穿透客户端有新连接到来，
+     * 并设置消息转发、关闭和错误的事件处理。
+     *
+     * @param clientId - 处理该连接的客户端唯一标识 ID
+     * @param connectionId - 连接唯一标识 ID
+     * @param userWs - 外部用户的 WebSocket 连接
+     * @param req - HTTP 升级请求对象
+     */
+    private handleWebSocketConnection;
+    /**
+     * 将用户消息转发到穿透客户端
+     *
+     * 将外部用户发送的 WebSocket 数据编码为 Base64 后，通过控制通道发送给穿透客户端。
+     *
+     * @param clientId - 目标客户端唯一标识 ID
+     * @param connectionId - 连接唯一标识 ID
+     * @param data - 用户发送的原始数据
+     */
+    private forwardToClient;
+    /**
+     * 通知穿透客户端连接已关闭
+     *
+     * 当外部用户的 WebSocket 连接关闭时，向穿透客户端发送连接关闭通知消息。
+     *
+     * @param clientId - 目标客户端唯一标识 ID
+     * @param connectionId - 已关闭的连接唯一标识 ID
+     * @param code - WebSocket 关闭状态码
+     */
+    private notifyClientClose;
+    /**
+     * 等待客户端响应
+     *
+     * 创建一个 Promise，等待穿透客户端处理完请求并返回响应数据。
+     * 若超过 30 秒未收到响应，将自动超时并拒绝 Promise。
+     *
+     * @param connectionId - 连接唯一标识 ID
+     * @param clientId - 客户端唯一标识 ID
+     * @returns 客户端返回的 HTTP 响应数据
+     */
+    private waitForResponse;
+    /**
+     * 读取 HTTP 请求体
+     *
+     * 从 IncomingMessage 流中读取完整的请求体数据。
+     *
+     * @param req - HTTP 请求对象
+     * @returns 包含完整请求体的 Buffer
+     */
+    private readRequestBody;
+    /**
+     * 处理穿透客户端返回的响应数据
+     *
+     * 接收客户端处理完请求后返回的 HTTP 响应数据，清除超时定时器并解析对应的待处理 Promise。
+     *
+     * @param connectionId - 连接唯一标识 ID
+     * @param data - 客户端返回的原始响应消息对象，包含 statusCode, headers, body 等字段
+     */
+    handleClientResponse(connectionId: string, data: any): void;
+    /**
+     * 处理来自穿透客户端的数据
+     *
+     * 将穿透客户端发送的数据转发给对应的外部用户 WebSocket 连接。
+     *
+     * @param connectionId - 连接唯一标识 ID
+     * @param data - 穿透客户端发送的数据
+     */
+    handleClientData(connectionId: string, data: Buffer): void;
+    /**
+     * 处理来自穿透客户端的关闭请求
+     *
+     * 当穿透客户端请求关闭某个连接时，关闭对应的外部用户 WebSocket 连接并清理资源。
+     *
+     * @param connectionId - 需要关闭的连接唯一标识 ID
+     * @param code - 可选的 WebSocket 关闭状态码，默认为 1000（正常关闭）
+     */
+    handleClientClose(connectionId: string, code?: number): void;
+    /**
+     * 清理连接资源
+     *
+     * 从用户连接映射表和会话管理器中移除指定连接的记录。
+     *
+     * @param connectionId - 需要清理的连接唯一标识 ID
+     */
+    private cleanupConnection;
+    /**
+     * 处理来自客户端的流式响应数据
+     *
+     * 将客户端发送的流式数据（如 SSE 事件）转发给对应的外部 HTTP 连接。
+     *
+     * @param connectionId - 连接唯一标识 ID
+     * @param data - 客户端发送的流式数据
+     */
+    handleClientStreamData(connectionId: string, data: Buffer): void;
+    /**
+     * 处理来自客户端的流式响应结束通知
+     *
+     * 当客户端通知流式响应结束时，关闭外部 HTTP 连接并清理资源。
+     *
+     * @param connectionId - 连接唯一标识 ID
+     */
+    handleClientStreamEnd(connectionId: string): void;
+    /**
+     * 停止指定端口的代理服务器
+     *
+     * 关闭指定端口上的代理服务器并从映射表中移除。
+     *
+     * @param port - 需要停止的代理端口号
+     * @returns 代理服务器关闭完成的 Promise
+     */
+    stopProxy(port: number): Promise<void>;
+    /**
+     * 停止所有代理服务器
+     *
+     * 并行关闭所有已启动的代理服务器。
+     *
+     * @returns 所有代理服务器关闭完成的 Promise
+     */
+    stopAll(): Promise<void>;
+    /**
+     * 获取所有活跃代理的端口列表
+     *
+     * @returns 当前正在运行的所有代理端口号数组
+     */
+    getActivePorts(): number[];
+}
+//# sourceMappingURL=unified-proxy.d.ts.map

package/dist/handlers/unified-proxy.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"unified-proxy.d.ts","sourceRoot":"","sources":["../../src/handlers/unified-proxy.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAMH,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAEvD;;;;;;GAMG;AACH,qBAAa,mBAAmB;IAC9B,cAAc;IACd,OAAO,CAAC,cAAc,CAAiB;IACvC,mCAAmC;IACnC,OAAO,CAAC,OAAO,CAA0B;IACzC,uCAAuC;IACvC,OAAO,CAAC,SAAS,CAA+B;IAEhD;;;;OAIG;IACH,OAAO,CAAC,gBAAgB,CAOT;IAEf;;;;OAIG;IACH,OAAO,CAAC,eAAe,CAAqC;IAE5D;;;;OAIG;IACH,OAAO,CAAC,kBAAkB,CAA0C;IAEpE;;;;OAIG;gBACS,cAAc,EAAE,cAAc;IAM1C;;;;;;;;;;OAUG;IACG,UAAU,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAsC/D;;;;;;;;;OASG;YACW,iBAAiB;IAoG/B;;;;;;;;;OASG;YACW,sBAAsB;IAuBpC;;;;;;;;;;OAUG;IACH,OAAO,CAAC,yBAAyB;IAkDjC;;;;;;;;OAQG;IACH,OAAO,CAAC,eAAe;IAWvB;;;;;;;;OAQG;IACH,OAAO,CAAC,iBAAiB;IAUzB;;;;;;;;;OASG;IACH,OAAO,CAAC,eAAe;IAWvB;;;;;;;OAOG;IACH,OAAO,CAAC,eAAe;IAavB;;;;;;;OAOG;IACH,oBAAoB,CAAC,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,IAAI;IAe3D;;;;;;;OAOG;IACH,gBAAgB,CAAC,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI;IAO1D;;;;;;;OAOG;IACH,iBAAiB,CAAC,YAAY,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,GAAG,IAAI;IAQ5D;;;;;;OAMG;IACH,OAAO,CAAC,iBAAiB;IAMzB;;;;;;;OAOG;IACH,sBAAsB,CAAC,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI;IAOhE;;;;;;OAMG;IACH,qBAAqB,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI;IASjD;;;;;;;OAOG;IACG,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAa5C;;;;;;OAMG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAQ9B;;;;OAIG;IACH,cAAc,IAAI,MAAM,EAAE;CAG3B"}