From 25ae2028d6a9ce516b01f0ec126abeab00eb329d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lo=C3=AFc=20Hoguin?= Date: Mon, 10 Oct 2011 09:09:15 +0200 Subject: Add {shutdown, Req} to websocket_init/3 to fail a websocket upgrade This change allows application developers to refuse websocket upgrades by returning {shutdown, Req}. The application can also send a reply with a custom error before returning from websocket_init/3, otherwise an error 400 is sent. Note that right now Cowboy closes the connection immediately. Also note that neither terminate/3 nor websocket_terminate/3 will be called when the connection is shutdown by websocket_init/3. --- src/cowboy_http_websocket.erl | 26 +++++++++++++++++++++++--- 1 file changed, 23 insertions(+), 3 deletions(-) (limited to 'src/cowboy_http_websocket.erl') diff --git a/src/cowboy_http_websocket.erl b/src/cowboy_http_websocket.erl index 61917b4..039e2b6 100644 --- a/src/cowboy_http_websocket.erl +++ b/src/cowboy_http_websocket.erl @@ -124,7 +124,9 @@ handler_init(State=#state{handler=Handler, opts=Opts}, Req2, HandlerState); {ok, Req2, HandlerState, Timeout, hibernate} -> websocket_handshake(State#state{timeout=Timeout, - hibernate=true}, Req2, HandlerState) + hibernate=true}, Req2, HandlerState); + {shutdown, Req2} -> + upgrade_denied(Req2) catch Class:Reason -> upgrade_error(Req), error_logger:error_msg( @@ -135,9 +137,27 @@ handler_init(State=#state{handler=Handler, opts=Opts}, end. -spec upgrade_error(#http_req{}) -> ok. -upgrade_error(Req=#http_req{socket=Socket, transport=Transport}) -> - {ok, _Req} = cowboy_http_req:reply(400, [], [], +upgrade_error(Req) -> + {ok, Req2} = cowboy_http_req:reply(400, [], [], Req#http_req{resp_state=waiting}), + upgrade_terminate(Req2). + +%% @see cowboy_http_protocol:ensure_response/1 +-spec upgrade_denied(#http_req{}) -> ok. +upgrade_denied(Req=#http_req{resp_state=done}) -> + upgrade_terminate(Req); +upgrade_denied(Req=#http_req{resp_state=waiting}) -> + {ok, Req2} = cowboy_http_req:reply(400, [], [], Req), + upgrade_terminate(Req2); +upgrade_denied(Req=#http_req{method='HEAD', resp_state=chunks}) -> + upgrade_terminate(Req); +upgrade_denied(Req=#http_req{socket=Socket, transport=Transport, + resp_state=chunks}) -> + Transport:send(Socket, <<"0\r\n\r\n">>), + upgrade_terminate(Req). + +-spec upgrade_terminate(#http_req{}) -> ok. +upgrade_terminate(#http_req{socket=Socket, transport=Transport}) -> Transport:close(Socket). -spec websocket_handshake(#state{}, #http_req{}, any()) -> ok. -- cgit v1.2.3