public void ClearTest()
{
ReceiveStream target = new ReceiveStream(); // TODO: 初始化为适当的值
target.Write(new byte[] { 1, 2 }, 0, 2);
target.Clear();
Assert.IsTrue(target.Length == 0);
}
public void ClearTest1()
{
ReceiveStream target = new ReceiveStream(); // TODO: 初始化为适当的值
target.Write(new byte[] { 1, 2 }, 0, 2);
int count = 1; // TODO: 初始化为适当的值
target.Clear(count);
Assert.IsTrue(target.Length == 1 && target[0] == 2);
}
/// <summary>
/// 当接收到会话对象的数据时，将触发此方法
/// </summary>
/// <param name="session">会话对象</param>
/// <param name="buffer">接收到的历史数据</param>
/// <returns></returns>
protected sealed override void OnReceive(FastSession session, ReceiveStream buffer)
{
while (true)
{
FastPacket packet = null;
try
{
packet = FastPacket.From(buffer);
}
catch (Exception ex)
{
buffer.Clear();
this.OnException(session, ex);
}
if (packet == null)
{
break;
}
// 新线程处理业务内容
Task.Factory.StartNew(() => this.OnRecvPacket(session, packet));
}
}
/// <summary>
/// 从收到的数据流中解析请求体
/// </summary>
/// <param name="stream"></param>
/// <returns></returns>
public static TelnetRequest Parse(ReceiveStream stream)
{
stream.Position = 0;
var index = stream.IndexOf(Crlf); // 以CRLF为结束标识
if (index < 0)
{
return(null);
}
var input = stream.ReadString(index, Encoding.ASCII);
stream.Clear(index + Crlf.Length);
if (string.IsNullOrWhiteSpace(input) == true) // 空内容
{
return(TelnetRequest.Parse(stream));
}
var items = input.Split(new string[] { " " }, 2, StringSplitOptions.RemoveEmptyEntries);
var cmd = items.First();
var arg = items.Length > 1 ? items.Last() : null;
return(new TelnetRequest(cmd, arg));
}
/// <summary>
/// 解析一个数据包
/// 不足一个封包时返回null
/// </summary>
/// <param name="buffer">接收到的历史数据</param>
/// <exception cref="ProtocolException"></exception>
/// <returns></returns>
public static FastPacket From(ReceiveStream buffer)
{
if (buffer.Length < 4)
{
return(null);
}
buffer.Position = 0;
var totalBytes = buffer.ReadInt32();
const int packegMaxSize = 10 * 1204 * 1024; // 10M
if (totalBytes > packegMaxSize)
{
throw new ProtocolException();
}
// 少于15字节是异常数据，清除收到的所有数据
const int packetMinSize = 15;
if (totalBytes < packetMinSize)
{
throw new ProtocolException();
}
// 数据包未接收完整
if (buffer.Length < totalBytes)
{
return(null);
}
// api名称数据长度
var apiNameLength = buffer.ReadByte();
if (totalBytes < apiNameLength + packetMinSize)
{
throw new ProtocolException();
}
// api名称数据
var apiNameBytes = buffer.ReadArray(apiNameLength);
// 标识符
var id = buffer.ReadInt64();
// 是否为客户端封包
var isFromClient = buffer.ReadBoolean();
// 是否异常
var isException = buffer.ReadBoolean();
// 实体数据
var body = buffer.ReadArray(totalBytes - buffer.Position);
// 清空本条数据
buffer.Clear(totalBytes);
var apiName = Encoding.UTF8.GetString(apiNameBytes);
var packet = new FastPacket(apiName, id, isFromClient)
{
TotalBytes = totalBytes,
ApiNameLength = apiNameLength,
IsException = isException,
Body = body
};
return(packet);
}
protected override void OnReceive(SessionBase session, ReceiveStream buffer)
{
buffer.Clear();
}
/// <summary>
/// 解析连接请求信息
/// 如果数据未完整则返回null
/// </summary>
/// <param name="buffer">接收到的原始数量</param>
/// <param name="localEndpoint">服务器的本地终结点</param>
/// <param name="remoteEndpoint">远程端的IP和端口</param>
/// <exception cref="HttpException"></exception>
/// <returns></returns>
public static HttpRequest Parse(ReceiveStream buffer, IPEndPoint localEndpoint, IPEndPoint remoteEndpoint)
{
buffer.Position = 0;
var doubleCrlf = Encoding.ASCII.GetBytes("\r\n\r\n");
var headerIndex = buffer.IndexOf(doubleCrlf);
if (headerIndex < 0)
{
return null; // 数据未完整
}
var headerLength = headerIndex + doubleCrlf.Length;
var headerString = buffer.ReadString(headerLength, Encoding.ASCII);
const string pattern = @"^(?<method>[^\s]+)\s(?<path>[^\s]+)\sHTTP\/1\.1\r\n" +
@"((?<field_name>[^:\r\n]+):\s(?<field_value>[^\r\n]*)\r\n)+" +
@"\r\n";
var match = Regex.Match(headerString, pattern, RegexOptions.IgnoreCase);
if (match.Success == false)
{
throw new HttpException(400, "请求中有语法问题，或不能满足请求");
}
var httpMethod = GetHttpMethod(match.Groups["method"].Value);
var httpHeader = new HttpHeader(match.Groups["field_name"].Captures, match.Groups["field_value"].Captures);
var contentLength = httpHeader.TryGet<int>("Content-Length");
if (httpMethod == HttpMethod.POST && buffer.Length - headerLength < contentLength)
{
return null; // 数据未完整
}
var request = new HttpRequest
{
LocalEndPoint = localEndpoint,
RemoteEndPoint = remoteEndpoint,
HttpMethod = httpMethod,
Headers = httpHeader
};
request.Url = new Uri("http://localhost:" + localEndpoint.Port + match.Groups["path"].Value);
request.Path = request.Url.AbsolutePath;
request.Query = new HttpNameValueCollection(HttpUtility.UrlDecode(request.Url.Query.TrimStart('?')));
if (httpMethod == HttpMethod.GET)
{
request.InputStrem = new byte[0];
request.Form = new HttpNameValueCollection();
request.Files = new HttpFile[0];
}
else
{
request.InputStrem = buffer.ReadArray(contentLength);
buffer.Position = headerLength;
HttpRequest.GeneratePostFormAndFiles(request, buffer);
}
buffer.Clear(headerLength + contentLength);
return request;
}
/// <summary>
/// 解析请求的数据
/// 返回请求数据包
/// </summary>
/// <param name="buffer">所有收到的数据</param>
/// <returns></returns>
public static unsafe FrameRequest From(ReceiveStream buffer)
{
if (buffer.Length < 2)
{
return null;
}
ByteBits byte0 = buffer[0];
var fin = byte0[0];
var frameCode = (FrameCodes)(byte)byte0.Take(4, 4);
if (fin == false || frameCode == FrameCodes.Continuation)
{
return null;
}
var rsv = byte0.Take(1, 3);
ByteBits byte1 = buffer[1];
var mask = byte1[0];
if (mask == false || Enum.IsDefined(typeof(FrameCodes), frameCode) == false || rsv != 0)
{
return null;
}
var contentLength = (int)byte1.Take(1, 7);
buffer.Position = 2;
if (contentLength == 127)
{
contentLength = (int)buffer.ReadUInt64();
}
else if (contentLength == 126)
{
contentLength = (int)buffer.ReadUInt16();
}
var packetLength = 6 + contentLength;
if (buffer.Length < packetLength)
{
return null;
}
var maskingKey = buffer.ReadArray(4);
var content = buffer.ReadArray(contentLength);
buffer.Clear(packetLength);
if (contentLength > 0)
{
fixed (byte* pcontent = &content[0], pmask = &maskingKey[0])
{
for (var i = 0; i < contentLength; i++)
{
*(pcontent + i) = (byte)(*(pcontent + i) ^ *(pmask + i % 4));
}
}
}
return new FrameRequest
{
Fin = fin,
Rsv = rsv,
Mask = mask,
Frame = frameCode,
ContentLength = contentLength,
MaskingKey = maskingKey,
Content = content
};
}
/// <summary>
/// 解析连接请求信息
/// 如果数据未完整则返回null
/// </summary>
/// <param name="buffer">接收到的原始数量</param>
/// <param name="localEndpoint">服务器的本地终结点</param>
/// <param name="remoteEndpoint">远程端的IP和端口</param>
/// <exception cref="HttpException"></exception>
/// <returns></returns>
public static HttpRequest Parse(ReceiveStream buffer, IPEndPoint localEndpoint, IPEndPoint remoteEndpoint)
{
buffer.Position = 0;
var doubleCrlf = Encoding.ASCII.GetBytes("\r\n\r\n");
var headerIndex = buffer.IndexOf(doubleCrlf);
if (headerIndex < 0)
{
return(null); // 数据未完整
}
var headerLength = headerIndex + doubleCrlf.Length;
var headerString = buffer.ReadString(headerLength, Encoding.ASCII);
const string pattern = @"^(?<method>[^\s]+)\s(?<path>[^\s]+)\sHTTP\/1\.1\r\n" +
@"((?<field_name>[^:\r\n]+):\s(?<field_value>[^\r\n]*)\r\n)+" +
@"\r\n";
var match = Regex.Match(headerString, pattern, RegexOptions.IgnoreCase);
if (match.Success == false)
{
throw new HttpException(400, "请求中有语法问题，或不能满足请求");
}
var httpMethod = GetHttpMethod(match.Groups["method"].Value);
var httpHeader = new HttpHeader(match.Groups["field_name"].Captures, match.Groups["field_value"].Captures);
var contentLength = httpHeader.TryGet <int>("Content-Length");
if (httpMethod == HttpMethod.POST && buffer.Length - headerLength < contentLength)
{
return(null); // 数据未完整
}
var request = new HttpRequest
{
LocalEndPoint = localEndpoint,
RemoteEndPoint = remoteEndpoint,
HttpMethod = httpMethod,
Headers = httpHeader
};
request.Url = new Uri("http://localhost:" + localEndpoint.Port + match.Groups["path"].Value);
request.Path = request.Url.AbsolutePath;
request.Query = new HttpNameValueCollection(HttpUtility.UrlDecode(request.Url.Query.TrimStart('?')));
if (httpMethod == HttpMethod.GET)
{
request.InputStrem = new byte[0];
request.Form = new HttpNameValueCollection();
request.Files = new HttpFile[0];
}
else
{
request.InputStrem = buffer.ReadArray(contentLength);
buffer.Position = headerLength;
HttpRequest.GeneratePostFormAndFiles(request, buffer);
}
buffer.Clear(headerLength + contentLength);
return(request);
}
/// <summary>
/// 解析请求的数据
/// 返回请求数据包
/// </summary>
/// <param name="buffer">所有收到的数据</param>
/// <returns></returns>
public unsafe static FrameRequest From(ReceiveStream buffer)
{
if (buffer.Length < 2)
{
return(null);
}
ByteBits byte0 = buffer[0];
var fin = byte0[0];
var frameCode = (FrameCodes)(byte)byte0.Take(4, 4);
if (fin == false || frameCode == FrameCodes.Continuation)
{
return(null);
}
var rsv = byte0.Take(1, 3);
ByteBits byte1 = buffer[1];
var mask = byte1[0];
if (mask == false || Enum.IsDefined(typeof(FrameCodes), frameCode) == false || rsv != 0)
{
return(null);
}
var contentLength = (int)byte1.Take(1, 7);
buffer.Position = 2;
if (contentLength == 127)
{
contentLength = (int)buffer.ReadUInt64();
}
else if (contentLength == 126)
{
contentLength = (int)buffer.ReadUInt16();
}
var packetLength = 6 + contentLength;
if (buffer.Length < packetLength)
{
return(null);
}
var maskingKey = buffer.ReadArray(4);
var content = buffer.ReadArray(contentLength);
buffer.Clear(packetLength);
if (contentLength > 0)
{
fixed(byte *pcontent = &content[0], pmask = &maskingKey[0])
{
for (var i = 0; i < contentLength; i++)
{
*(pcontent + i) = (byte)(*(pcontent + i) ^ *(pmask + i % 4));
}
}
}
return(new FrameRequest
{
Fin = fin,
Rsv = rsv,
Mask = mask,
Frame = frameCode,
ContentLength = contentLength,
MaskingKey = maskingKey,
Content = content
});
}
/// <summary>
/// 解析一个数据包
/// 不足一个封包时返回null
/// </summary>
/// <param name="buffer">接收到的历史数据</param>
/// <exception cref="ProtocolException"></exception>
/// <returns></returns>
public static FastPacket From(ReceiveStream buffer)
{
if (buffer.Length < 4)
{
return null;
}
buffer.Position = 0;
var totalBytes = buffer.ReadInt32();
const int packegMaxSize = 10 * 1204 * 1024; // 10M
if (totalBytes > packegMaxSize)
{
throw new ProtocolException();
}
// 少于15字节是异常数据，清除收到的所有数据
const int packetMinSize = 15;
if (totalBytes < packetMinSize)
{
throw new ProtocolException();
}
// 数据包未接收完整
if (buffer.Length < totalBytes)
{
return null;
}
// api名称数据长度
var apiNameLength = buffer.ReadByte();
if (totalBytes < apiNameLength + packetMinSize)
{
throw new ProtocolException();
}
// api名称数据
var apiNameBytes = buffer.ReadArray(apiNameLength);
// 标识符
var id = buffer.ReadInt64();
// 是否为客户端封包
var isFromClient = buffer.ReadBoolean();
// 是否异常
var isException = buffer.ReadBoolean();
// 实体数据
var body = buffer.ReadArray(totalBytes - buffer.Position);
// 清空本条数据
buffer.Clear(totalBytes);
var apiName = Encoding.UTF8.GetString(apiNameBytes);
var packet = new FastPacket(apiName, id, isFromClient)
{
TotalBytes = totalBytes,
ApiNameLength = apiNameLength,
IsException = isException,
Body = body
};
return packet;
}