0
Fork 0
mirror of https://github.com/ninenines/cowboy.git synced 2025-07-14 12:20:24 +00:00

Replace cowboy_req:host/1 with cowboy_req:raw_host/1

The latter is much more useful than the former, which ends up
being removed.
This commit is contained in:
Loïc Hoguin 2012-08-30 23:11:24 +02:00
parent e17e18668d
commit 6fa734b487
5 changed files with 30 additions and 32 deletions

View file

@ -25,9 +25,8 @@
version = {1, 1} :: cowboy_http:version(),
peer = undefined :: undefined |
{inet:ip_address(), inet:port_number()},
host = undefined :: undefined | cowboy_dispatcher:tokens(),
host = undefined :: undefined | binary(),
host_info = undefined :: undefined | cowboy_dispatcher:tokens(),
raw_host = undefined :: undefined | binary(),
port = undefined :: undefined | inet:port_number(),
path = undefined :: undefined | '*' | cowboy_dispatcher:tokens(),
path_info = undefined :: undefined | cowboy_dispatcher:tokens(),

View file

@ -62,6 +62,7 @@
max_line_length :: integer(),
timeout :: timeout(),
buffer = <<>> :: binary(),
host_tokens = undefined :: undefined | cowboy_dispatcher:tokens(),
hibernate = false :: boolean(),
loop_timeout = infinity :: timeout(),
loop_timeout_ref :: undefined | reference()
@ -184,18 +185,20 @@ wait_header(Req, State=#state{socket=Socket,
-spec header({http_header, integer(), cowboy_http:header(), any(), binary()}
| http_eoh, #http_req{}, #state{}) -> ok.
header({http_header, _I, 'Host', _R, RawHost}, Req=#http_req{
transport=Transport, host=undefined}, State) ->
transport=Transport}, State=#state{host_tokens=undefined}) ->
RawHost2 = cowboy_bstr:to_lower(RawHost),
case catch cowboy_dispatcher:split_host(RawHost2) of
{Host, RawHost3, undefined} ->
{HostTokens, RawHost3, undefined} ->
Port = default_port(Transport:name()),
parse_header(Req#http_req{
host=Host, raw_host=RawHost3, port=Port,
headers=[{'Host', RawHost}|Req#http_req.headers]}, State);
{Host, RawHost3, Port} ->
host=RawHost3, port=Port,
headers=[{'Host', RawHost}|Req#http_req.headers]},
State#state{host_tokens=HostTokens});
{HostTokens, RawHost3, Port} ->
parse_header(Req#http_req{
host=Host, raw_host=RawHost3, port=Port,
headers=[{'Host', RawHost}|Req#http_req.headers]}, State);
host=RawHost3, port=Port,
headers=[{'Host', RawHost}|Req#http_req.headers]},
State#state{host_tokens=HostTokens});
{'EXIT', _Reason} ->
error_terminate(400, State)
end;
@ -216,14 +219,15 @@ header({http_header, _I, Field, _R, Value}, Req, State) ->
parse_header(Req#http_req{headers=[{Field2, Value}|Req#http_req.headers]},
State);
%% The Host header is required in HTTP/1.1.
header(http_eoh, #http_req{version={1, 1}, host=undefined}, State) ->
header(http_eoh, #http_req{version={1, 1}},
State=#state{host_tokens=undefined}) ->
error_terminate(400, State);
%% It is however optional in HTTP/1.0.
header(http_eoh, Req=#http_req{version={1, 0}, transport=Transport,
host=undefined}, State=#state{buffer=Buffer}) ->
header(http_eoh, Req=#http_req{version={1, 0}, transport=Transport},
State=#state{buffer=Buffer, host_tokens=undefined}) ->
Port = default_port(Transport:name()),
onrequest(Req#http_req{host=[], raw_host= <<>>,
port=Port, buffer=Buffer}, State#state{buffer= <<>>});
onrequest(Req#http_req{host= <<>>, port=Port, buffer=Buffer},
State#state{buffer= <<>>, host_tokens=[]});
header(http_eoh, Req, State=#state{buffer=Buffer}) ->
onrequest(Req#http_req{buffer=Buffer}, State#state{buffer= <<>>});
header(_Any, _Req, State) ->
@ -244,12 +248,13 @@ onrequest(Req, State=#state{onrequest=OnRequest}) ->
end.
-spec dispatch(#http_req{}, #state{}) -> ok.
dispatch(Req=#http_req{host=Host, path=Path},
State=#state{dispatch=Dispatch}) ->
case cowboy_dispatcher:match(Host, Path, Dispatch) of
dispatch(Req=#http_req{path=Path},
State=#state{dispatch=Dispatch, host_tokens=HostTokens}) ->
case cowboy_dispatcher:match(HostTokens, Path, Dispatch) of
{ok, Handler, Opts, Binds, HostInfo, PathInfo} ->
handler_init(Req#http_req{host_info=HostInfo, path_info=PathInfo,
bindings=Binds}, State#state{handler={Handler, Opts}});
bindings=Binds}, State#state{handler={Handler, Opts},
host_tokens=undefined});
{error, notfound, host} ->
error_terminate(400, State);
{error, notfound, path} ->
@ -403,8 +408,8 @@ next_request(Req=#http_req{connection=Conn}, State=#state{
case {HandlerRes, BodyRes, RespRes, Conn} of
{ok, ok, ok, keepalive} ->
?MODULE:parse_request(State#state{
buffer=Buffer, req_empty_lines=0,
req_keepalive=Keepalive + 1});
buffer=Buffer, host_tokens=undefined,
req_empty_lines=0, req_keepalive=Keepalive + 1});
_Closed ->
terminate(State)
end.

View file

@ -28,7 +28,6 @@
-export([peer_addr/1]).
-export([host/1]).
-export([host_info/1]).
-export([raw_host/1]).
-export([port/1]).
-export([path/1]).
-export([path_info/1]).
@ -131,8 +130,8 @@ peer_addr(Req = #http_req{}) ->
end,
{PeerAddr, Req3}.
%% @doc Return the tokens for the hostname requested.
-spec host(Req) -> {cowboy_dispatcher:tokens(), Req} when Req::req().
%% @doc Return the host binary string.
-spec host(Req) -> {binary(), Req} when Req::req().
host(Req) ->
{Req#http_req.host, Req}.
@ -143,11 +142,6 @@ host(Req) ->
host_info(Req) ->
{Req#http_req.host_info, Req}.
%% @doc Return the raw host directly taken from the request.
-spec raw_host(Req) -> {binary(), Req} when Req::req().
raw_host(Req) ->
{Req#http_req.raw_host, Req}.
%% @doc Return the port used for this request.
-spec port(Req) -> {inet:port_number(), Req} when Req::req().
port(Req) ->
@ -826,12 +820,12 @@ upgrade_reply(Status, Headers, Req=#http_req{
%% @doc Compact the request data by removing all non-system information.
%%
%% This essentially removes the host, path, query string, bindings and headers.
%% This essentially removes the path, query string, bindings and headers.
%% Use it when you really need to save up memory, for example when having
%% many concurrent long-running connections.
-spec compact(Req) -> Req when Req::req().
compact(Req) ->
Req#http_req{host=undefined, host_info=undefined, path=undefined,
Req#http_req{host_info=undefined, path=undefined,
path_info=undefined, qs_vals=undefined,
bindings=undefined, headers=[],
p_headers=[], cookies=[]}.

View file

@ -672,7 +672,7 @@ create_path(Req=#http_req{meta=Meta}, State) ->
State2, 303)
end.
create_path_location(#http_req{transport=Transport, raw_host=Host,
create_path_location(#http_req{transport=Transport, host=Host,
port=Port}, Path) ->
TransportName = Transport:name(),
<< (create_path_location_protocol(TransportName))/binary, "://",

View file

@ -155,7 +155,7 @@ upgrade_denied(#http_req{socket=Socket, transport=Transport,
-spec websocket_handshake(#state{}, #http_req{}, any()) -> closed.
websocket_handshake(State=#state{version=0, origin=Origin,
challenge={Key1, Key2}}, Req=#http_req{socket=Socket,
transport=Transport, raw_host=Host, port=Port,
transport=Transport, host=Host, port=Port,
raw_path=Path, raw_qs=QS}, HandlerState) ->
Location = hixie76_location(Transport:name(), Host, Port, Path, QS),
{ok, Req2} = cowboy_req:upgrade_reply(