/// <summary>收到客户端发来的数据</summary>
/// <param name="e"></param>
protected override void OnReceive(ReceivedEventArgs e)
{
if (e.Packet.Total == 0 /*|| !HttpBase.FastValidHeader(e.Packet)*/)
{
base.OnReceive(e);
return;
}
// WebSocket 数据
if (_websocket != null)
{
_websocket.Process(e.Packet);
base.OnReceive(e);
return;
}
var req = Request;
var request = new HttpRequest();
if (request.Parse(e.Packet))
{
req = Request = request;
WriteLog("{0} {1}", request.Method, request.Url);
_websocket = null;
OnNewRequest(request, e);
// 后面还有数据包，克隆缓冲区
if (req.IsCompleted)
{
_cache = null;
}
else
{
// 限制最大请求为1G
if (req.ContentLength > 1 * 1024 * 1024 * 1024)
{
var rs = new HttpResponse {
StatusCode = HttpStatusCode.RequestEntityTooLarge
};
Send(rs.Build());
Dispose();
return;
}
_cache = new MemoryStream(req.ContentLength);
req.Body.CopyTo(_cache);
//req.Body = req.Body.Clone();
}
}
else if (req != null)
{
// 链式数据包
//req.Body.Append(e.Packet.Clone());
e.Packet.CopyTo(_cache);
if (_cache.Length >= req.ContentLength)
{
_cache.Position = 0;
req.Body = new Packet(_cache);
_cache = null;
}
}
// 收到全部数据后，触发请求处理
if (req != null && req.IsCompleted)
{
var rs = ProcessRequest(req, e);
if (rs != null)
{
var server = (this as INetSession).Host as HttpServer;
if (!server.ServerName.IsNullOrEmpty() && !rs.Headers.ContainsKey("Server"))
{
rs.Headers["Server"] = server.ServerName;
}
var closing = !req.KeepAlive && _websocket == null;
if (closing && !rs.Headers.ContainsKey("Connection"))
{
rs.Headers["Connection"] = "close";
}
Send(rs.Build());
if (closing)
{
Dispose();
}
}
}
base.OnReceive(e);
}
/// <summary>处理Http请求</summary>
/// <param name="request"></param>
/// <param name="e"></param>
/// <returns></returns>
protected virtual HttpResponse ProcessRequest(HttpRequest request, ReceivedEventArgs e)
{
if (request?.Url == null)
{
return new HttpResponse {
StatusCode = HttpStatusCode.NotFound
}
}
;
// 匹配路由处理器
var server = (this as INetSession).Host as HttpServer;
var path = request.Url.OriginalString;
var p = path.IndexOf('?');
if (p > 0)
{
path = path.Substring(0, p);
}
// 路径安全检查，防止越界
if (path.Contains(".."))
{
return new HttpResponse {
StatusCode = HttpStatusCode.Forbidden
}
}
;
var handler = server.MatchHandler(path);
if (handler == null)
{
return new HttpResponse {
StatusCode = HttpStatusCode.NotFound
}
}
;
var context = new DefaultHttpContext
{
Connection = this,
Request = request,
Response = new HttpResponse(),
Path = path,
Handler = handler,
};
try
{
PrepareRequest(context);
// 处理 WebSocket 握手
if (_websocket == null)
{
_websocket = WebSocket.Handshake(context);
}
handler.ProcessRequest(context);
}
catch (Exception ex)
{
context.Response.SetResult(ex);
}
return(context.Response);
}
