WebSocketDataHandlingAdapter.cs 9.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290
  1. //------------------------------------------------------------------------------
  2. // 此代码版权(除特别声明或在XREF结尾的命名空间的代码)归作者本人若汝棋茗所有
  3. // 源代码使用协议遵循本仓库的开源协议及附加协议,若本仓库没有设置,则按MIT开源协议授权
  4. // CSDN博客:https://blog.csdn.net/qq_40374647
  5. // 哔哩哔哩视频:https://space.bilibili.com/94253567
  6. // Gitee源代码仓库:https://gitee.com/RRQM_Home
  7. // Github源代码仓库:https://github.com/RRQM
  8. // API首页:https://www.yuque.com/rrqm/touchsocket/index
  9. // 交流QQ群:234762506
  10. // 感谢您的下载和使用
  11. //------------------------------------------------------------------------------
  12. //------------------------------------------------------------------------------
  13. using System;
  14. using System.Collections.Generic;
  15. using TouchSocket.Core;
  16. using TouchSocket.Sockets;
  17. namespace TouchSocket.Http.WebSockets
  18. {
  19. /// <summary>
  20. /// WebSocket适配器
  21. /// </summary>
  22. public class WebSocketDataHandlingAdapter : DataHandlingAdapter
  23. {
  24. private WSDataFrame m_dataFrameTemp;
  25. /// <summary>
  26. /// 数据包剩余长度
  27. /// </summary>
  28. private int m_surPlusLength = 0;
  29. /// <summary>
  30. /// 临时包
  31. /// </summary>
  32. private ByteBlock m_tempByteBlock;
  33. /// <summary>
  34. /// <inheritdoc/>
  35. /// </summary>
  36. public override bool CanSplicingSend => false;
  37. /// <summary>
  38. /// <inheritdoc/>
  39. /// </summary>
  40. public override bool CanSendRequestInfo => false;
  41. /// <summary>
  42. /// 解码
  43. /// </summary>
  44. /// <param name="dataBuffer"></param>
  45. /// <param name="offset"></param>
  46. /// <param name="length"></param>
  47. /// <param name="dataFrame"></param>
  48. /// <returns></returns>
  49. public FilterResult DecodingFromBytes(byte[] dataBuffer, ref int offset, int length, out WSDataFrame dataFrame)
  50. {
  51. int index = offset;
  52. dataFrame = new WSDataFrame();
  53. dataFrame.RSV1 = dataBuffer[offset].GetBit(6) == 1;
  54. dataFrame.RSV2 = dataBuffer[offset].GetBit(5) == 1;
  55. dataFrame.RSV3 = dataBuffer[offset].GetBit(4) == 1;
  56. dataFrame.FIN = (dataBuffer[offset] >> 7) == 1;
  57. dataFrame.Opcode = (WSDataType)(dataBuffer[offset] & 0xf);
  58. dataFrame.Mask = (dataBuffer[++offset] >> 7) == 1;
  59. int payloadLength = dataBuffer[offset] & 0x7f;
  60. if (payloadLength < 126)
  61. {
  62. offset++;
  63. }
  64. else if (payloadLength == 126)
  65. {
  66. if (length < 4)
  67. {
  68. offset = index;
  69. return FilterResult.Cache;
  70. }
  71. payloadLength = TouchSocketBitConverter.BigEndian.ToUInt16(dataBuffer, ++offset);
  72. offset += 2;
  73. }
  74. else if (payloadLength == 127)
  75. {
  76. if (length < 12)
  77. {
  78. if (m_tempByteBlock == null)
  79. {
  80. m_tempByteBlock = new ByteBlock();
  81. }
  82. m_tempByteBlock.Write(dataBuffer, index, length);
  83. offset = index;
  84. return FilterResult.GoOn;
  85. }
  86. payloadLength = (int)TouchSocketBitConverter.BigEndian.ToUInt64(dataBuffer, ++offset);
  87. offset += 8;
  88. }
  89. dataFrame.PayloadLength = payloadLength;
  90. if (dataFrame.Mask)
  91. {
  92. if (length < (offset - index) + 4)
  93. {
  94. if (m_tempByteBlock == null)
  95. {
  96. m_tempByteBlock = new ByteBlock();
  97. }
  98. m_tempByteBlock.Write(dataBuffer, index, length);
  99. offset = index;
  100. return FilterResult.GoOn;
  101. }
  102. dataFrame.MaskingKey = new byte[4];
  103. dataFrame.MaskingKey[0] = dataBuffer[offset++];
  104. dataFrame.MaskingKey[1] = dataBuffer[offset++];
  105. dataFrame.MaskingKey[2] = dataBuffer[offset++];
  106. dataFrame.MaskingKey[3] = dataBuffer[offset++];
  107. }
  108. ByteBlock byteBlock = new ByteBlock(payloadLength);
  109. dataFrame.PayloadData = byteBlock;
  110. int surlen = length - (offset - index);
  111. if (payloadLength <= surlen)
  112. {
  113. byteBlock.Write(dataBuffer, offset, payloadLength);
  114. offset += payloadLength;
  115. }
  116. else
  117. {
  118. byteBlock.Write(dataBuffer, offset, surlen);
  119. offset += surlen;
  120. }
  121. return FilterResult.Success;
  122. }
  123. /// <summary>
  124. /// 当接收到数据时处理数据
  125. /// </summary>
  126. /// <param name="byteBlock">数据流</param>
  127. protected override void PreviewReceived(ByteBlock byteBlock)
  128. {
  129. byte[] buffer = byteBlock.Buffer;
  130. int r = byteBlock.Len;
  131. if (m_tempByteBlock != null)
  132. {
  133. m_tempByteBlock.Write(buffer, 0, r);
  134. buffer = m_tempByteBlock.ToArray();
  135. r = m_tempByteBlock.Pos;
  136. m_tempByteBlock.Dispose();
  137. m_tempByteBlock = null;
  138. }
  139. if (m_dataFrameTemp == null)
  140. {
  141. SplitPackage(buffer, 0, r);
  142. }
  143. else
  144. {
  145. if (m_surPlusLength == r)
  146. {
  147. m_dataFrameTemp.PayloadData.Write(buffer, 0, m_surPlusLength);
  148. PreviewHandle(m_dataFrameTemp);
  149. m_dataFrameTemp = null;
  150. m_surPlusLength = 0;
  151. }
  152. else if (m_surPlusLength < r)
  153. {
  154. m_dataFrameTemp.PayloadData.Write(buffer, 0, m_surPlusLength);
  155. PreviewHandle(m_dataFrameTemp);
  156. m_dataFrameTemp = null;
  157. SplitPackage(buffer, m_surPlusLength, r);
  158. }
  159. else
  160. {
  161. m_dataFrameTemp.PayloadData.Write(buffer, 0, r);
  162. m_surPlusLength -= r;
  163. }
  164. }
  165. }
  166. /// <summary>
  167. /// 当发送数据前处理数据
  168. /// </summary>
  169. /// <param name="buffer"></param>
  170. /// <param name="offset"></param>
  171. /// <param name="length"></param>
  172. protected override void PreviewSend(byte[] buffer, int offset, int length)
  173. {
  174. GoSend(buffer, offset, length);
  175. }
  176. /// <summary>
  177. /// <inheritdoc/>
  178. /// </summary>
  179. /// <param name="transferBytes"></param>
  180. protected override void PreviewSend(IList<ArraySegment<byte>> transferBytes)
  181. {
  182. throw new System.NotImplementedException();//因为设置了不支持拼接发送,所以该方法可以不实现。
  183. }
  184. /// <summary>
  185. /// <inheritdoc/>
  186. /// </summary>
  187. /// <param name="requestInfo"></param>
  188. protected override void PreviewSend(IRequestInfo requestInfo)
  189. {
  190. throw new NotImplementedException();
  191. }
  192. /// <summary>
  193. /// <inheritdoc/>
  194. /// </summary>
  195. protected override void Reset()
  196. {
  197. m_tempByteBlock = null;
  198. m_dataFrameTemp = null;
  199. m_surPlusLength = 0;
  200. base.Reset();
  201. }
  202. private void PreviewHandle(WSDataFrame dataFrame)
  203. {
  204. try
  205. {
  206. if (dataFrame.Mask)
  207. {
  208. WSTools.DoMask(dataFrame.PayloadData.Buffer, 0, dataFrame.PayloadData.Buffer, 0, dataFrame.PayloadData.Len, dataFrame.MaskingKey);
  209. }
  210. GoReceived(null, dataFrame);
  211. }
  212. finally
  213. {
  214. dataFrame.Dispose();
  215. }
  216. }
  217. /// <summary>
  218. /// 分解包
  219. /// </summary>
  220. /// <param name="dataBuffer"></param>
  221. /// <param name="offset"></param>
  222. /// <param name="length"></param>
  223. private void SplitPackage(byte[] dataBuffer, int offset, int length)
  224. {
  225. while (offset < length)
  226. {
  227. if (length - offset < 2)
  228. {
  229. if (m_tempByteBlock == null)
  230. {
  231. m_tempByteBlock = new ByteBlock();
  232. }
  233. m_tempByteBlock.Write(dataBuffer, offset, length - offset);
  234. return;
  235. }
  236. switch (DecodingFromBytes(dataBuffer, ref offset, length - offset, out WSDataFrame dataFrame))
  237. {
  238. case FilterResult.Cache:
  239. {
  240. if (m_tempByteBlock == null)
  241. {
  242. m_tempByteBlock = new ByteBlock();
  243. }
  244. m_tempByteBlock.Write(dataBuffer, offset, length - offset);
  245. return;
  246. }
  247. case FilterResult.Success:
  248. {
  249. if (dataFrame.PayloadLength == dataFrame.PayloadData.Len)
  250. {
  251. PreviewHandle(dataFrame);
  252. }
  253. else
  254. {
  255. m_surPlusLength = dataFrame.PayloadLength - dataFrame.PayloadData.Len;
  256. m_dataFrameTemp = dataFrame;
  257. }
  258. }
  259. break;
  260. case FilterResult.GoOn:
  261. default:
  262. return;
  263. }
  264. }
  265. }
  266. }
  267. }