diff options
| author | Anthony Ramine <[email protected]> | 2011-11-04 11:42:36 +0100 |
|---|---|---|
| committer | Anthony Ramine <[email protected]> | 2011-11-04 12:21:01 +0100 |
| commit | 04bcbc444d507f6936bfee18347ec292efe0ae67 (patch) | |
| tree | a8016247d2f71cf55cdcbbf2aeede60def5b468e /src/cowboy_http_req.erl | |
| parent | 68c1d886e5c4cb974b62b9c5fe7f21e6bbc5d557 (diff) | |
| download | cowboy-04bcbc444d507f6936bfee18347ec292efe0ae67.tar.gz cowboy-04bcbc444d507f6936bfee18347ec292efe0ae67.tar.bz2 cowboy-04bcbc444d507f6936bfee18347ec292efe0ae67.zip | |
Support 'Content-Length' in parse_header/2
Diffstat (limited to 'src/cowboy_http_req.erl')
| -rw-r--r-- | src/cowboy_http_req.erl | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/src/cowboy_http_req.erl b/src/cowboy_http_req.erl index 3a3dd2f..7a09bcb 100644 --- a/src/cowboy_http_req.erl +++ b/src/cowboy_http_req.erl @@ -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. |