umgeher/cowboy
0
Fork 0
mirror of https://github.com/ninenines/cowboy.git synced 2025-07-16 05:00:24 +00:00
Code Issues Wiki Activity

Support 'Content-Length' in parse_header/2

Browse source
This commit is contained in:
Anthony Ramine 2011-11-04 11:42:36 +01:00
parent 68c1d886e5
commit 04bcbc444d
2 changed files with 49 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/cowboy_http_req.erl
View file

@ -231,6 +231,11 @@ parse_header(Name, Req, Default) when Name =:= 'Connection' ->
fun (Value) ->
cowboy_http:nonempty_list(Value, fun cowboy_http:token_ci/2)
end);
parse_header(Name, Req, Default) when Name =:= 'Content-Length' ->
parse_header(Name, Req, Default,
fun (Value) ->
cowboy_http:digits(Value)
end);
parse_header(Name, Req, Default) ->
{Value, Req2} = header(Name, Req, Default),
{undefined, Value, Req2}.
@ -292,12 +297,12 @@ cookies(Req=#http_req{cookies=Cookies}) ->
%% @todo We probably want to allow a max length.
-spec body(#http_req{}) -> {ok, binary(), #http_req{}} | {error, atom()}.
body(Req) ->
{Length, Req2} = cowboy_http_req:header('Content-Length', Req),
{Length, Req2} = cowboy_http_req:parse_header('Content-Length', Req),
case Length of
undefined -> {error, badarg};
{error, badarg} -> {error, badarg};
_Any ->
Length2 = list_to_integer(binary_to_list(Length)),
body(Length2, Req2)
body(Length, Req2)
end.
%% @doc Return <em>Length</em> bytes of the request body.