From 64aaaf8e47e4cb5e783ec501e8f9b8adf2a472b0 Mon Sep 17 00:00:00 2001
From: Baobhan Sith <80159437+Heavrnl@users.noreply.github.com>
Date: Tue, 29 Apr 2025 06:58:13 +0800
Subject: [PATCH] Update websocket.ts

---
 packages/backend/src/websocket.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/backend/src/websocket.ts b/packages/backend/src/websocket.ts
index 5c5c846..29d39c1 100644
--- a/packages/backend/src/websocket.ts
+++ b/packages/backend/src/websocket.ts
@@ -418,7 +418,7 @@ export const initializeWebSocket = async (server: http.Server, sessionParser: Re
             console.log(`WebSocket 认证成功 (Path: ${pathname})：用户 ${request.session.username} (ID: ${request.session.userId})`);
 
             // --- 根据路径处理升级 ---
-            if (pathname === '/rdp-proxy') {
+            if (pathname === '/rdp-proxy' || pathname === '/ws/rdp-proxy') {
                 // RDP 代理路径 - 直接处理升级，连接逻辑在 'connection' 事件中处理
                 console.log(`WebSocket: Handling RDP proxy upgrade for user ${request.session.username}`);
                 wss.handleUpgrade(request, socket, head, (ws) => {