mirror of
https://github.com/snltty/linker.git
synced 2025-10-31 20:43:00 +08:00
优化内存分享，修复一些问题，新增答题投票功能
This commit is contained in:
snltty
649
cmonitor/api/websocket/WebSocketParser.cs
Normal file
649
cmonitor/api/websocket/WebSocketParser.cs
Normal file
@@ -0,0 +1,649 @@
|
||||
using common.libs;
|
||||
using System.Buffers;
|
||||
using System.Buffers.Binary;
|
||||
using System.Net;
|
||||
using System.Security.Cryptography;
|
||||
using System.Text;
|
||||
|
||||
namespace cmonitor.api.websocket
|
||||
{
|
||||
/// <summary>
|
||||
/// websocket解析器
|
||||
/// </summary>
|
||||
public static class WebSocketParser
|
||||
{
|
||||
private readonly static SHA1 sha1 = SHA1.Create();
|
||||
private readonly static Memory<byte> magicCode = Encoding.ASCII.GetBytes("258EAFA5-E914-47DA-95CA-C5AB0DC85B11");
|
||||
/// <summary>
|
||||
/// 构建连接数据
|
||||
/// </summary>
|
||||
/// <param name="header"></param>
|
||||
/// <returns></returns>
|
||||
public static byte[] BuildConnectData(WebsocketHeaderInfo header)
|
||||
{
|
||||
string path = header.Path.Length == 0 ? "/" : Encoding.UTF8.GetString(header.Path.Span);
|
||||
|
||||
StringBuilder sb = new StringBuilder(10);
|
||||
sb.Append($"GET {path} HTTP/1.1\r\n");
|
||||
sb.Append($"Upgrade: websocket\r\n");
|
||||
sb.Append($"Connection: Upgrade\r\n");
|
||||
sb.Append($"Sec-WebSocket-Version: 13\r\n");
|
||||
sb.Append($"Sec-WebSocket-Key: {Encoding.UTF8.GetString(header.SecWebSocketKey.Span)}\r\n");
|
||||
if (header.SecWebSocketProtocol.Length > 0)
|
||||
{
|
||||
sb.Append($"Sec-WebSocket-Protocol: {Encoding.UTF8.GetString(header.SecWebSocketProtocol.Span)}\r\n");
|
||||
}
|
||||
if (header.SecWebSocketExtensions.Length > 0)
|
||||
{
|
||||
sb.Append($"Sec-WebSocket-Extensions: {Encoding.UTF8.GetString(header.SecWebSocketExtensions.Span)}\r\n");
|
||||
}
|
||||
sb.Append("\r\n");
|
||||
|
||||
return Encoding.UTF8.GetBytes(sb.ToString());
|
||||
}
|
||||
/// <summary>
|
||||
/// 构建连接回应数据
|
||||
/// </summary>
|
||||
/// <param name="header"></param>
|
||||
/// <returns></returns>
|
||||
public static byte[] BuildConnectResponseData(WebsocketHeaderInfo header)
|
||||
{
|
||||
string acceptStr = BuildSecWebSocketAccept(header.SecWebSocketKey);
|
||||
|
||||
StringBuilder sb = new StringBuilder(10);
|
||||
sb.Append($"HTTP/1.1 {(int)header.StatusCode} {AddSpace(header.StatusCode)}\r\n");
|
||||
sb.Append($"Sec-WebSocket-Accept: {acceptStr}\r\n");
|
||||
if (header.Connection.Length > 0)
|
||||
{
|
||||
sb.Append($"Connection: {Encoding.UTF8.GetString(header.Connection.Span)}\r\n");
|
||||
}
|
||||
if (header.Upgrade.Length > 0)
|
||||
{
|
||||
sb.Append($"Upgrade: {Encoding.UTF8.GetString(header.Upgrade.Span)}\r\n");
|
||||
}
|
||||
if (header.SecWebSocketVersion.Length > 0)
|
||||
{
|
||||
sb.Append($"Sec-WebSocket-Version: {Encoding.UTF8.GetString(header.SecWebSocketVersion.Span)}\r\n");
|
||||
}
|
||||
if (header.SecWebSocketProtocol.Length > 0)
|
||||
{
|
||||
sb.Append($"Sec-WebSocket-Protocol: {Encoding.UTF8.GetString(header.SecWebSocketProtocol.Span)}\r\n");
|
||||
}
|
||||
if (header.SecWebSocketExtensions.Length > 0)
|
||||
{
|
||||
sb.Append($"Sec-WebSocket-Extensions: {Encoding.UTF8.GetString(header.SecWebSocketExtensions.Span)}\r\n");
|
||||
}
|
||||
sb.Append("\r\n");
|
||||
|
||||
return Encoding.UTF8.GetBytes(sb.ToString());
|
||||
}
|
||||
/// <summary>
|
||||
/// 生成随机key
|
||||
/// </summary>
|
||||
/// <returns></returns>
|
||||
public static byte[] BuildSecWebSocketKey()
|
||||
{
|
||||
byte[] bytes = new byte[16];
|
||||
Random random = new Random(DateTime.Now.Ticks.GetHashCode());
|
||||
for (int i = 0; i < bytes.Length; i++)
|
||||
{
|
||||
bytes[i] = (byte)random.Next(0, 255);
|
||||
}
|
||||
byte[] res = Encoding.UTF8.GetBytes(Convert.ToBase64String(bytes));
|
||||
return res;
|
||||
}
|
||||
/// <summary>
|
||||
/// 构建mask数据
|
||||
/// </summary>
|
||||
/// <returns></returns>
|
||||
public static byte[] BuildMaskKey()
|
||||
{
|
||||
byte[] bytes = new byte[4];
|
||||
|
||||
Random random = new Random(DateTime.Now.Ticks.GetHashCode());
|
||||
for (int i = 0; i < bytes.Length; i++)
|
||||
{
|
||||
bytes[i] = (byte)random.Next(0, 255);
|
||||
}
|
||||
|
||||
return bytes;
|
||||
}
|
||||
/// <summary>
|
||||
/// 生成accept回应
|
||||
/// </summary>
|
||||
/// <param name="key"></param>
|
||||
/// <returns></returns>
|
||||
private static string BuildSecWebSocketAccept(Memory<byte> key)
|
||||
{
|
||||
int keyLength = key.Length + magicCode.Length;
|
||||
byte[] acceptBytes = new byte[keyLength];
|
||||
|
||||
key.CopyTo(acceptBytes);
|
||||
magicCode.CopyTo(acceptBytes.AsMemory(key.Length));
|
||||
|
||||
string acceptStr = Convert.ToBase64String(sha1.ComputeHash(acceptBytes, 0, keyLength));
|
||||
|
||||
return acceptStr;
|
||||
}
|
||||
/// <summary>
|
||||
/// 验证回应的accept
|
||||
/// </summary>
|
||||
/// <param name="key"></param>
|
||||
/// <param name="accept"></param>
|
||||
/// <returns></returns>
|
||||
public static bool VerifySecWebSocketAccept(Memory<byte> key, Memory<byte> accept)
|
||||
{
|
||||
string acceptStr = BuildSecWebSocketAccept(key);
|
||||
return acceptStr == Encoding.UTF8.GetString(accept.Span);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 构建ping帧
|
||||
/// </summary>
|
||||
/// <returns></returns>
|
||||
public static byte[] BuildPingData()
|
||||
{
|
||||
return new byte[]
|
||||
{
|
||||
(byte)WebSocketFrameInfo.EnumFin.Fin | (byte)WebSocketFrameInfo.EnumOpcode.Ping, //fin + ping
|
||||
(byte)WebSocketFrameInfo.EnumMask.None | 0, //没有 mask 和 payload length
|
||||
};
|
||||
}
|
||||
/// <summary>
|
||||
/// 构建pong帧
|
||||
/// </summary>
|
||||
/// <returns></returns>
|
||||
public static byte[] BuildPongData()
|
||||
{
|
||||
return new byte[]
|
||||
{
|
||||
(byte)WebSocketFrameInfo.EnumFin.Fin | (byte)WebSocketFrameInfo.EnumOpcode.Pong, //fin + pong
|
||||
(byte)WebSocketFrameInfo.EnumMask.None | 0, //没有 mask 和 payload length
|
||||
};
|
||||
}
|
||||
/// <summary>
|
||||
/// 构建数据帧
|
||||
/// </summary>
|
||||
/// <param name="remark"></param>
|
||||
/// <returns></returns>
|
||||
/// <exception cref="Exception"></exception>
|
||||
public static byte[] BuildFrameData(WebSocketFrameRemarkInfo remark, out int length)
|
||||
{
|
||||
if (remark.Mask > 0 && remark.MaskData.Length != 4)
|
||||
{
|
||||
throw new Exception("mask data just 4byte");
|
||||
}
|
||||
|
||||
length = 1 + 1 + remark.Data.Length;
|
||||
int index = 2;
|
||||
if (remark.Mask == WebSocketFrameInfo.EnumMask.Mask)
|
||||
{
|
||||
length += 4;
|
||||
}
|
||||
|
||||
byte payloadLength;
|
||||
int dataLength = remark.Data.Length;
|
||||
if (dataLength > ushort.MaxValue)
|
||||
{
|
||||
length += 8;
|
||||
payloadLength = 127;
|
||||
}
|
||||
else if (dataLength > 125)
|
||||
{
|
||||
length += 2;
|
||||
payloadLength = 126;
|
||||
}
|
||||
else
|
||||
{
|
||||
payloadLength = (byte)dataLength;
|
||||
}
|
||||
|
||||
|
||||
byte[] bytes = ArrayPool<byte>.Shared.Rent(length);
|
||||
var memory = bytes.AsMemory();
|
||||
bytes[0] = (byte)((byte)remark.Fin | (byte)remark.Rsv1 | (byte)remark.Rsv2 | (byte)remark.Rsv3 | (byte)remark.Opcode);
|
||||
bytes[1] = (byte)((byte)remark.Mask | payloadLength);
|
||||
|
||||
|
||||
if (dataLength > ushort.MaxValue)
|
||||
{
|
||||
BinaryPrimitives.WriteUInt64BigEndian(memory.Slice(index).Span, (ulong)dataLength);
|
||||
index += 8;
|
||||
}
|
||||
else if (dataLength > 125)
|
||||
{
|
||||
BinaryPrimitives.WriteUInt16BigEndian(memory.Slice(index).Span, (ushort)dataLength);
|
||||
index += 2;
|
||||
}
|
||||
|
||||
if (remark.Mask == WebSocketFrameInfo.EnumMask.Mask)
|
||||
{
|
||||
remark.MaskData.CopyTo(bytes.AsMemory(index, remark.MaskData.Length));
|
||||
index += remark.MaskData.Length;
|
||||
}
|
||||
|
||||
if (remark.Data.Length > 0)
|
||||
{
|
||||
remark.Data.CopyTo(bytes.AsMemory(index));
|
||||
}
|
||||
|
||||
return bytes;
|
||||
}
|
||||
public static void Return(byte[] bytes)
|
||||
{
|
||||
ArrayPool<byte>.Shared.Return(bytes);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 给每个大写字母前加一个空格，例如ProxyAuthenticationRequired 变成Proxy Authentication Required
|
||||
/// </summary>
|
||||
/// <param name="statusCode"></param>
|
||||
/// <returns></returns>
|
||||
private static string AddSpace(HttpStatusCode statusCode)
|
||||
{
|
||||
ReadOnlySpan<char> span = statusCode.ToString().AsSpan();
|
||||
|
||||
int totalLength = span.Length * 2;
|
||||
|
||||
char[] result = new char[totalLength];
|
||||
Span<char> resultSpan = result.AsSpan(0, totalLength);
|
||||
span.CopyTo(resultSpan);
|
||||
|
||||
int length = 0;
|
||||
for (int i = 0; i < span.Length; i++)
|
||||
{
|
||||
if (i > 0 && span[i] >= 65 && span[i] <= 90)
|
||||
{
|
||||
resultSpan.Slice(i + length, totalLength - (length + i) - 1).CopyTo(resultSpan.Slice(i + length + 1));
|
||||
resultSpan[i + length] = (char)32;
|
||||
length++;
|
||||
}
|
||||
}
|
||||
|
||||
string resultStr = resultSpan.Slice(0, span.Length + length).ToString();
|
||||
|
||||
return resultStr;
|
||||
}
|
||||
}
|
||||
|
||||
public sealed class WebSocketFrameRemarkInfo
|
||||
{
|
||||
/// <summary>
|
||||
/// 是否是结束帧，如果只有一帧，那必定是结束帧
|
||||
/// </summary>
|
||||
public WebSocketFrameInfo.EnumFin Fin { get; set; } = WebSocketFrameInfo.EnumFin.Fin;
|
||||
/// <summary>
|
||||
/// 保留位1
|
||||
/// </summary>
|
||||
public WebSocketFrameInfo.EnumRsv1 Rsv1 { get; set; } = WebSocketFrameInfo.EnumRsv1.None;
|
||||
/// <summary>
|
||||
/// 保留位2
|
||||
/// </summary>
|
||||
public WebSocketFrameInfo.EnumRsv2 Rsv2 { get; set; } = WebSocketFrameInfo.EnumRsv2.None;
|
||||
/// <summary>
|
||||
/// 保留位3
|
||||
/// </summary>
|
||||
public WebSocketFrameInfo.EnumRsv3 Rsv3 { get; set; } = WebSocketFrameInfo.EnumRsv3.None;
|
||||
/// <summary>
|
||||
/// 数据描述，默认TEXT数据
|
||||
/// </summary>
|
||||
public WebSocketFrameInfo.EnumOpcode Opcode { get; set; } = WebSocketFrameInfo.EnumOpcode.Text;
|
||||
|
||||
/// <summary>
|
||||
/// 是否有掩码
|
||||
/// </summary>
|
||||
public WebSocketFrameInfo.EnumMask Mask { get; set; } = WebSocketFrameInfo.EnumMask.None;
|
||||
/// <summary>
|
||||
/// 掩码key 4字节
|
||||
/// </summary>
|
||||
public Memory<byte> MaskData { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// payload data
|
||||
/// </summary>
|
||||
public Memory<byte> Data { get; set; }
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 数据帧解析
|
||||
/// </summary>
|
||||
public sealed class WebSocketFrameInfo
|
||||
{
|
||||
/// <summary>
|
||||
/// 是否是结束帧
|
||||
/// </summary>
|
||||
public EnumFin Fin { get; set; }
|
||||
/// <summary>
|
||||
/// 保留位
|
||||
/// </summary>
|
||||
public EnumRsv1 Rsv1 { get; set; }
|
||||
/// <summary>
|
||||
///
|
||||
/// </summary>
|
||||
public EnumRsv2 Rsv2 { get; set; }
|
||||
/// <summary>
|
||||
///
|
||||
/// </summary>
|
||||
public EnumRsv3 Rsv3 { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// 操作码 0附加数据，1文本，2二进制，3-7为非控制保留，8关闭，9ping，a pong，b-f 为控制保留
|
||||
/// </summary>
|
||||
public EnumOpcode Opcode { get; set; }
|
||||
/// <summary>
|
||||
/// 掩码
|
||||
/// </summary>
|
||||
public EnumMask Mask { get; set; }
|
||||
/// <summary>
|
||||
/// 总长度
|
||||
/// </summary>
|
||||
public int TotalLength { get; set; }
|
||||
/// <summary>
|
||||
/// 数据 如果OPCODE是 EnumOpcode.Close 则数据的前2字节为关闭状态码，余下的为其它描述数据
|
||||
/// </summary>
|
||||
public Memory<byte> PayloadData { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// 解析帧，如果false解析失败，则应该把data缓存起来，等待下次来数据后，拼接起来再次解析
|
||||
/// </summary>
|
||||
/// <param name="data"></param>
|
||||
/// <param name="frameInfo"></param>
|
||||
/// <returns></returns>
|
||||
public static bool TryParse(Memory<byte> data, out WebSocketFrameInfo frameInfo)
|
||||
{
|
||||
frameInfo = null;
|
||||
|
||||
//小于2字节不可解析
|
||||
if (data.Length < 2)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
Span<byte> span = data.Span;
|
||||
int index = 2;
|
||||
|
||||
//第2字节
|
||||
//1位 是否mask
|
||||
EnumMask mask = (EnumMask)(span[1] & (byte)EnumMask.Mask);
|
||||
int payloadLength = (span[1] & 0b01111111);
|
||||
if (payloadLength == 126)
|
||||
{
|
||||
payloadLength = BinaryPrimitives.ReadUInt16BigEndian(span.Slice(2, 2));
|
||||
index += 2;
|
||||
}
|
||||
else if (payloadLength == 127)
|
||||
{
|
||||
payloadLength = (int)BinaryPrimitives.ReadUInt64BigEndian(span.Slice(2, 8));
|
||||
index += 8;
|
||||
}
|
||||
//数据长+头长 大于 整个数据长，则不是一个完整的包
|
||||
if (data.Length < payloadLength + index + (mask == EnumMask.Mask ? 4 : 0))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
//第1字节
|
||||
//1位 是否是结束帧
|
||||
EnumFin fin = (EnumFin)(byte)(span[0] & (byte)EnumFin.Fin);
|
||||
//2 3 4 保留
|
||||
EnumRsv1 rsv1 = (EnumRsv1)(byte)(span[0] & (byte)EnumRsv1.Rsv);
|
||||
EnumRsv2 rsv2 = (EnumRsv2)(byte)(span[0] & (byte)EnumRsv2.Rsv);
|
||||
EnumRsv3 rsv3 = (EnumRsv3)(byte)(span[0] & (byte)EnumRsv3.Rsv);
|
||||
//5 6 7 8 操作码
|
||||
EnumOpcode opcode = (EnumOpcode)(byte)(span[0] & (byte)EnumOpcode.Last);
|
||||
|
||||
|
||||
Span<byte> maskKey = Helper.EmptyArray;
|
||||
if (mask == EnumMask.Mask)
|
||||
{
|
||||
//mask掩码key 用来解码数据
|
||||
maskKey = span.Slice(index, 4);
|
||||
index += 4;
|
||||
}
|
||||
|
||||
//数据
|
||||
Memory<byte> payloadData = data.Slice(index, payloadLength);
|
||||
if (mask == EnumMask.Mask)
|
||||
{
|
||||
//解码
|
||||
Span<byte> payloadDataSpan = payloadData.Span;
|
||||
for (int i = 0; i < payloadDataSpan.Length; i++)
|
||||
{
|
||||
payloadDataSpan[i] = (byte)(payloadDataSpan[i] ^ maskKey[3 & i]);
|
||||
}
|
||||
}
|
||||
|
||||
frameInfo = new WebSocketFrameInfo
|
||||
{
|
||||
Fin = fin,
|
||||
Rsv1 = rsv1,
|
||||
Rsv2 = rsv2,
|
||||
Rsv3 = rsv3,
|
||||
Opcode = opcode,
|
||||
Mask = mask,
|
||||
PayloadData = payloadData,
|
||||
TotalLength = index + payloadLength
|
||||
};
|
||||
return true;
|
||||
}
|
||||
public enum EnumFin : byte
|
||||
{
|
||||
None = 0x0,
|
||||
Fin = 0b10000000,
|
||||
}
|
||||
public enum EnumMask : byte
|
||||
{
|
||||
None = 0x0,
|
||||
Mask = 0b10000000,
|
||||
}
|
||||
public enum EnumRsv1 : byte
|
||||
{
|
||||
None = 0x0,
|
||||
Rsv = 0b01000000,
|
||||
}
|
||||
public enum EnumRsv2 : byte
|
||||
{
|
||||
None = 0x0,
|
||||
Rsv = 0b00100000,
|
||||
}
|
||||
public enum EnumRsv3 : byte
|
||||
{
|
||||
None = 0x0,
|
||||
Rsv = 0b00010000,
|
||||
}
|
||||
public enum EnumOpcode : byte
|
||||
{
|
||||
Data = 0x0,
|
||||
Text = 0x1,
|
||||
Binary = 0x2,
|
||||
UnControll3 = 0x3,
|
||||
UnControll4 = 0x4,
|
||||
UnControll5 = 0x5,
|
||||
UnControll6 = 0x6,
|
||||
UnControll7 = 0x7,
|
||||
Close = 0x8,
|
||||
Ping = 0x9,
|
||||
Pong = 0xa,
|
||||
Controll11 = 0xb,
|
||||
Controll12 = 0xc,
|
||||
Controll13 = 0xd,
|
||||
Controll14 = 0xe,
|
||||
Controll15 = 0xf,
|
||||
Last = 0xf,
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 关闭的状态码
|
||||
/// </summary>
|
||||
public enum EnumCloseStatus : ushort
|
||||
{
|
||||
/// <summary>
|
||||
/// 正常关闭
|
||||
/// </summary>
|
||||
Normal = 1000,
|
||||
/// <summary>
|
||||
/// 正在离开
|
||||
/// </summary>
|
||||
Leaving = 1001,
|
||||
/// <summary>
|
||||
/// 协议错误
|
||||
/// </summary>
|
||||
ProtocolError = 1002,
|
||||
/// <summary>
|
||||
/// 只能接收TEXT数据
|
||||
/// </summary>
|
||||
TextOnly = 1003,
|
||||
/// <summary>
|
||||
/// 保留
|
||||
/// </summary>
|
||||
None1004 = 1004,
|
||||
/// <summary>
|
||||
/// 保留
|
||||
/// </summary>
|
||||
None1005 = 1005,
|
||||
/// <summary>
|
||||
/// 保留
|
||||
/// </summary>
|
||||
None1006 = 1006,
|
||||
/// <summary>
|
||||
/// 消息类型不一致
|
||||
/// </summary>
|
||||
DataTypeError = 1007,
|
||||
/// <summary>
|
||||
/// 通用状态码，没有别的合适的状态码时，用这个
|
||||
/// </summary>
|
||||
PublicError = 1008,
|
||||
/// <summary>
|
||||
/// 数据太大，无法处理
|
||||
/// </summary>
|
||||
DataTooBig = 1009,
|
||||
/// <summary>
|
||||
/// 扩展错误
|
||||
/// </summary>
|
||||
ExtendsError = 1010,//正常关闭
|
||||
/// <summary>
|
||||
/// 意外情况
|
||||
/// </summary>
|
||||
Unexpected = 1011,
|
||||
/// <summary>
|
||||
/// TLS握手失败
|
||||
/// </summary>
|
||||
TLSError = 1015
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 请求头解析
|
||||
/// </summary>
|
||||
public sealed class WebsocketHeaderInfo
|
||||
{
|
||||
static byte[][] bytes = new byte[][] {
|
||||
Encoding.ASCII.GetBytes("Connection: "),
|
||||
Encoding.ASCII.GetBytes("Upgrade: "),
|
||||
Encoding.ASCII.GetBytes("Origin: "),
|
||||
Encoding.ASCII.GetBytes("Sec-WebSocket-Version: "),
|
||||
Encoding.ASCII.GetBytes("Sec-WebSocket-Key: "),
|
||||
Encoding.ASCII.GetBytes("Sec-WebSocket-Extensions: "),
|
||||
Encoding.ASCII.GetBytes("Sec-WebSocket-Protocol: "),
|
||||
Encoding.ASCII.GetBytes("Sec-WebSocket-Accept: "),
|
||||
};
|
||||
static byte[] httpBytes = Encoding.UTF8.GetBytes("HTTP/");
|
||||
|
||||
public HttpStatusCode StatusCode { get; set; } = HttpStatusCode.SwitchingProtocols;
|
||||
public Memory<byte> Method { get; private set; }
|
||||
|
||||
private string _pathSet { get; set; }
|
||||
/// <summary>
|
||||
/// 用这个设置path值
|
||||
/// </summary>
|
||||
public string PathSet
|
||||
{
|
||||
get
|
||||
{
|
||||
return _pathSet;
|
||||
}
|
||||
set
|
||||
{
|
||||
_pathSet = value;
|
||||
Path = Encoding.UTF8.GetBytes(_pathSet);
|
||||
}
|
||||
}
|
||||
/// <summary>
|
||||
/// 如果 仅1个字符，那就是 /
|
||||
/// </summary>
|
||||
public Memory<byte> Path { get; private set; }
|
||||
public Memory<byte> Connection { get; private set; }
|
||||
public Memory<byte> Upgrade { get; private set; }
|
||||
public Memory<byte> Origin { get; private set; }
|
||||
public Memory<byte> SecWebSocketVersion { get; private set; }
|
||||
public Memory<byte> SecWebSocketKey { get; set; }
|
||||
public Memory<byte> SecWebSocketExtensions { get; set; }
|
||||
public Memory<byte> SecWebSocketProtocol { get; set; }
|
||||
public Memory<byte> SecWebSocketAccept { get; set; }
|
||||
public static WebsocketHeaderInfo Parse(Memory<byte> header)
|
||||
{
|
||||
Span<byte> span = header.Span;
|
||||
int flag = 0xff;
|
||||
int bit = 0x01;
|
||||
|
||||
ulong[] res = new ulong[bytes.Length];
|
||||
|
||||
for (int i = 0, len = span.Length; i < len; i++)
|
||||
{
|
||||
if (span[i] == 13 && span[i + 1] == 10 && span[i + 2] == 13 && span[i + 3] == 10)
|
||||
{
|
||||
break;
|
||||
}
|
||||
if (span[i] == 13 && span[i + 1] == 10)
|
||||
{
|
||||
int startIndex = i + 2;
|
||||
for (int k = 0; k < bytes.Length; k++)
|
||||
{
|
||||
if ((flag >> k & 1) == 1 && span[startIndex] == bytes[k][0])
|
||||
{
|
||||
if (span.Slice(startIndex, bytes[k].Length).SequenceEqual(bytes[k]))
|
||||
{
|
||||
int index = span.Slice(startIndex).IndexOf((byte)13);
|
||||
flag &= ~(bit << k);
|
||||
|
||||
#pragma warning disable CS0675 // 对进行了带符号扩展的操作数使用了按位或运算符
|
||||
res[k] = ((ulong)(startIndex + bytes[k].Length) << 32) | (ulong)(index - bytes[k].Length);
|
||||
#pragma warning restore CS0675 // 对进行了带符号扩展的操作数使用了按位或运算符
|
||||
|
||||
i += index + 1;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
WebsocketHeaderInfo headerInfo = new WebsocketHeaderInfo
|
||||
{
|
||||
Connection = header.Slice((int)(res[0] >> 32), (int)(res[0] & 0xffffffff)),
|
||||
Upgrade = header.Slice((int)(res[1] >> 32), (int)(res[1] & 0xffffffff)),
|
||||
Origin = header.Slice((int)(res[2] >> 32), (int)(res[2] & 0xffffffff)),
|
||||
SecWebSocketVersion = header.Slice((int)(res[3] >> 32), (int)(res[3] & 0xffffffff)),
|
||||
SecWebSocketKey = header.Slice((int)(res[4] >> 32), (int)(res[4] & 0xffffffff)),
|
||||
SecWebSocketExtensions = header.Slice((int)(res[5] >> 32), (int)(res[5] & 0xffffffff)),
|
||||
SecWebSocketProtocol = header.Slice((int)(res[6] >> 32), (int)(res[6] & 0xffffffff)),
|
||||
SecWebSocketAccept = header.Slice((int)(res[7] >> 32), (int)(res[7] & 0xffffffff)),
|
||||
};
|
||||
|
||||
int pathIndex = span.IndexOf((byte)32);
|
||||
int pathIndex1 = span.Slice(pathIndex + 1).IndexOf((byte)32);
|
||||
if (header.Slice(0, httpBytes.Length).Span.SequenceEqual(httpBytes))
|
||||
{
|
||||
int code = int.Parse(Encoding.UTF8.GetString(header.Slice(pathIndex + 1, pathIndex1).Span));
|
||||
headerInfo.StatusCode = (HttpStatusCode)code;
|
||||
}
|
||||
else
|
||||
{
|
||||
headerInfo.Path = header.Slice(pathIndex + 1, pathIndex1);
|
||||
headerInfo.Method = header.Slice(0, pathIndex);
|
||||
}
|
||||
|
||||
return headerInfo;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user