From 8b02992e6abd63eab0aafe0d762c38dbbbb757b5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lo=C3=AFc=20Hoguin?= Date: Mon, 21 Mar 2011 17:26:00 +0100 Subject: Skip the request body if it hasn't been read by the handler. --- src/cowboy_http_protocol.erl | 28 +++++++++++++++++++++++----- 1 file changed, 23 insertions(+), 5 deletions(-) (limited to 'src/cowboy_http_protocol.erl') diff --git a/src/cowboy_http_protocol.erl b/src/cowboy_http_protocol.erl index 2b5292c..b02f471 100644 --- a/src/cowboy_http_protocol.erl +++ b/src/cowboy_http_protocol.erl @@ -154,16 +154,34 @@ handler_loop(HandlerState, Req, State=#state{handler={Handler, _Opts}}) -> -spec handler_terminate(HandlerState::term(), Req::#http_req{}, State::#state{}) -> ok. handler_terminate(HandlerState, Req, State=#state{handler={Handler, _Opts}}) -> - Res = (catch Handler:terminate( + HandlerRes = (catch Handler:terminate( Req#http_req{resp_state=locked}, HandlerState)), - %% @todo We must skip any body data from the request - %% before processing another. + BodyRes = ensure_body_processed(Req), ensure_response(Req, State), - case {Res, State#state.connection} of - {ok, keepalive} -> next_request(State); + case {HandlerRes, BodyRes, State#state.connection} of + {ok, ok, keepalive} -> next_request(State); _Closed -> terminate(State) end. +-spec ensure_body_processed(Req::#http_req{}) -> ok | close. +ensure_body_processed(#http_req{body_state=done}) -> + ok; +ensure_body_processed(Req=#http_req{body_state=waiting}) -> + {Length, Req2} = cowboy_http_req:header('Content-Length', Req), + case Length of + "" -> ok; + _Any -> + Length2 = list_to_integer(Length), + skip_body(Length2, Req2) + end. + +-spec skip_body(Length::non_neg_integer(), Req::#http_req{}) -> ok | close. +skip_body(Length, Req) -> + case cowboy_http_req:body(Length, Req) of + {error, _Reason} -> close; + _Any -> ok + end. + %% No response has been sent but everything apparently went fine. %% Reply with 204 No Content to indicate this. ensure_response(#http_req{resp_state=waiting}, State) -> -- cgit v1.2.3