示例#1
0
        /// <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);
        }
示例#2
0
        /// <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);
        }