gnh1201/welsonjs
1
0
Fork 0
mirror of https://github.com/gnh1201/welsonjs.git synced 2025-10-30 20:41:18 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update WebSocketManager.cs

Browse Source
This commit is contained in:
Namhyeon Go 2025-08-14 03:18:14 +09:00 committed by GitHub
parent f1af7029fa
commit 8bca089976
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 25 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
WelsonJS.Toolkit/WelsonJS.Launcher/WebSocketManager.cs
View File

@ -1,10 +1,9 @@
// WebSocketManager.cs
// WebSocketManager.cs
// SPDX-License-Identifier: GPL-3.0-or-later
// SPDX-FileCopyrightText: 2025 Catswords OSS and WelsonJS Contributors
// https://github.com/gnh1201/welsonjs
//
using System;
using System.Buffers;
using System.Collections.Concurrent;
using System.IO;
using System.Net.WebSockets;
@ -146,45 +145,38 @@ namespace WelsonJS.Launcher
await entry.IoLock.WaitAsync(token);
try
{
// Send message (single-frame; can be split if needed)
// Send message
await sock.SendAsync(new ArraySegment<byte>(sendBuf), WebSocketMessageType.Text, true, token);
// Receive message until EndOfMessage is reached
var buffer = ArrayPool<byte>.Shared.Rent(8192);
try
// Receive message until EndOfMessage
byte[] buffer = new byte[8192];
using (var ms = new MemoryStream())
{
using (var ms = new MemoryStream())
while (true)
{
while (true)
var res = await sock.ReceiveAsync(new ArraySegment<byte>(buffer), token);
if (res.MessageType == WebSocketMessageType.Close)
{
var res = await sock.ReceiveAsync(new ArraySegment<byte>(buffer), token);
if (res.MessageType == WebSocketMessageType.Close)
{
// Server requested closure
try { await sock.CloseAsync(WebSocketCloseStatus.NormalClosure, "Closing as requested by server", token); } catch { }
throw new WebSocketException($"WebSocket closed by server: {sock.CloseStatus} {sock.CloseStatusDescription}");
}
if (res.Count > 0)
{
ms.Write(buffer, 0, res.Count);
if (ms.Length > maxMessageBytes)
throw new InvalidOperationException($"Received message exceeds limit ({maxMessageBytes} bytes).");
}
if (res.EndOfMessage)
break;
// Server requested closure
try { await sock.CloseAsync(WebSocketCloseStatus.NormalClosure, "Closing as requested by server", token); } catch { }
throw new WebSocketException($"WebSocket closed by server: {sock.CloseStatus} {sock.CloseStatusDescription}");
}
// Convert UTF-8 encoded text message to string
return Encoding.UTF8.GetString(ms.ToArray());
if (res.Count > 0)
{
ms.Write(buffer, 0, res.Count);
if (ms.Length > maxMessageBytes)
throw new InvalidOperationException($"Received message exceeds limit ({maxMessageBytes} bytes).");
}
if (res.EndOfMessage)
break;
}
}
finally
{
ArrayPool<byte>.Shared.Return(buffer);
// Convert UTF-8 encoded text message to string
return Encoding.UTF8.GetString(ms.ToArray());
}
}
finally