ws 帧合并判断处理

BarrageGrab/Proxy/TitaniumProxy.cs

@@ -103,20 +103,46 @@ namespace BarrageGrab.Proxy
 
             var processid = args.HttpClient.ProcessId.Value;
 
-            var frames = args.WebSocketDecoderReceive.Decode(e.Buffer, e.Offset, e.Count).ToList();
+            List<byte> messageData = new List<byte>();
 
-            foreach (var frame in frames)
+            foreach (var frame in args.WebSocketDecoderReceive.Decode(e.Buffer, e.Offset, e.Count))
             {
+                if (frame.OpCode == WebsocketOpCode.Continuation)
+                {
+                    messageData.AddRange(frame.Data.ToArray());
+                    continue;
+                }
+                else
+                {
+                    byte[] payload;
+                    if (messageData.Count > 0)
+                    {
+                        messageData.AddRange(frame.Data.ToArray());
+                        payload = messageData.ToArray();
+                        messageData.Clear();
+                    }
+                    else
+                    {
+                        payload = frame.Data.ToArray();
+                    }
+
                     base.FireWsEvent(new WsMessageEventArgs()
                     {
                         ProcessID = processid,
                         HostName = hostname,
-                    Payload = frame.Data.ToArray(),
+                        Payload = payload,
                         ProcessName = base.GetProcessName(processid)
                     });
                 }
             }
 
+            if (messageData.Count > 0)
+            {
+                // 没有收到 WebSocket 帧的结束帧，抛出异常或者进行处理
+            }
+
+        }
+
         /// <summary>
         /// 释放资源，关闭系统代理
         /// </summary>