diff options
| author | Loïc Hoguin <[email protected]> | 2012-09-17 00:30:54 +0200 |
|---|---|---|
| committer | Loïc Hoguin <[email protected]> | 2012-09-17 13:57:28 +0200 |
| commit | 5b56dd780bfeddac96ae8360bd5b352e427e1b3d (patch) | |
| tree | ee3480737879d3237f9f032b740cca9a3250ed21 /src | |
| parent | fc02b7f767516116167fb98b28d69805e48dbd23 (diff) | |
| download | cowboy-5b56dd780bfeddac96ae8360bd5b352e427e1b3d.tar.gz cowboy-5b56dd780bfeddac96ae8360bd5b352e427e1b3d.tar.bz2 cowboy-5b56dd780bfeddac96ae8360bd5b352e427e1b3d.zip | |
Add the private set_buffer/2 function used by cowboy_protocol
Diffstat (limited to 'src')
| -rw-r--r-- | src/cowboy_protocol.erl | 6 | ||||
| -rw-r--r-- | src/cowboy_req.erl | 6 |
2 files changed, 10 insertions, 2 deletions
diff --git a/src/cowboy_protocol.erl b/src/cowboy_protocol.erl index 9520667..bcbdf57 100644 --- a/src/cowboy_protocol.erl +++ b/src/cowboy_protocol.erl @@ -214,11 +214,13 @@ header(http_eoh, Req, State=#state{host_tokens=undefined, error_terminate(400, State); {{1, 0}, Req2} -> Port = default_port(Transport:name()), - onrequest(Req2#http_req{host= <<>>, port=Port, buffer=Buffer}, + onrequest( + cowboy_req:set_buffer(Buffer, + cowboy_req:set_host(<<>>, Port, <<>>, Req2)), State#state{buffer= <<>>, host_tokens=[]}) end; header(http_eoh, Req, State=#state{buffer=Buffer}) -> - onrequest(Req#http_req{buffer=Buffer}, State#state{buffer= <<>>}); + onrequest(cowboy_req:set_buffer(Buffer, Req), State#state{buffer= <<>>}); header(_Any, _Req, State) -> error_terminate(400, State). diff --git a/src/cowboy_req.erl b/src/cowboy_req.erl index d074efa..aad6a97 100644 --- a/src/cowboy_req.erl +++ b/src/cowboy_req.erl @@ -106,6 +106,7 @@ -export([set_host/4]). -export([set_connection/2]). -export([add_header/3]). +-export([set_buffer/2]). %% Misc API. -export([compact/1]). @@ -935,6 +936,11 @@ set_connection(RawConnection, Req=#http_req{headers=Headers}) -> add_header(Name, Value, Req=#http_req{headers=Headers}) -> Req#http_req{headers=[{Name, Value}|Headers]}. +%% @private +-spec set_buffer(binary(), Req) -> Req when Req::req(). +set_buffer(Buffer, Req) -> + Req#http_req{buffer=Buffer}. + %% Misc API. %% @doc Compact the request data by removing all non-system information. |