aboutsummaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
authorLoïc Hoguin <[email protected]>2011-10-20 01:28:50 +0200
committerLoïc Hoguin <[email protected]>2011-10-20 01:36:23 +0200
commit70d28ff64d107eb04903779573a4c6b4df18fdfa (patch)
treedc15d03a1061e9fa1de77393b9f750fcf38cdf86 /src
parentfe5f0ca539fe883f397c97a5793f86e2de4a34f3 (diff)
downloadcowboy-70d28ff64d107eb04903779573a4c6b4df18fdfa.tar.gz
cowboy-70d28ff64d107eb04903779573a4c6b4df18fdfa.tar.bz2
cowboy-70d28ff64d107eb04903779573a4c6b4df18fdfa.zip
Make sure the hixie-76 websocket code works properly with proxies
Diffstat (limited to 'src')
-rw-r--r--src/cowboy_http_websocket.erl28
1 files changed, 19 insertions, 9 deletions
diff --git a/src/cowboy_http_websocket.erl b/src/cowboy_http_websocket.erl
index d3c9c2d..59be2e5 100644
--- a/src/cowboy_http_websocket.erl
+++ b/src/cowboy_http_websocket.erl
@@ -50,7 +50,7 @@
version :: 0 | 7 | 8,
handler :: module(),
opts :: any(),
- challenge = undefined :: undefined | binary(),
+ challenge = undefined :: undefined | binary() | {binary(), binary()},
timeout = infinity :: timeout(),
timeout_ref = undefined :: undefined | reference(),
messages = undefined :: undefined | {atom(), atom(), atom()},
@@ -86,6 +86,11 @@ websocket_upgrade(State, Req) ->
-spec websocket_upgrade(undefined | <<_:8>>, #state{}, #http_req{})
-> {ok, #state{}, #http_req{}}.
%% No version given. Assuming hixie-76 draft.
+%%
+%% We need to wait to send a reply back before trying to read the
+%% third part of the challenge key, because proxies will wait for
+%% a reply before sending it. Therefore we calculate the challenge
+%% key only in websocket_handshake/3.
%% @todo Check Origin?
websocket_upgrade(undefined, State, Req) ->
{<<"WebSocket">>, Req2} = cowboy_http_req:header('Upgrade', Req),
@@ -93,11 +98,9 @@ websocket_upgrade(undefined, State, Req) ->
{Key1, Req4} = cowboy_http_req:header(<<"Sec-Websocket-Key1">>, Req3),
{Key2, Req5} = cowboy_http_req:header(<<"Sec-Websocket-Key2">>, Req4),
false = lists:member(undefined, [Origin, Key1, Key2]),
- {ok, Key3, Req6} = cowboy_http_req:body(8, Req5),
- Challenge = hixie76_challenge(Key1, Key2, Key3),
EOP = binary:compile_pattern(<< 255 >>),
- {ok, State#state{version=0, origin=Origin, challenge=Challenge,
- eop=EOP}, Req6};
+ {ok, State#state{version=0, origin=Origin, challenge={Key1, Key2},
+ eop=EOP}, Req5};
%% Versions 7 and 8. Implementation follows the hybi 7 through 10 drafts.
%% @todo We don't need Origin?
websocket_upgrade(<< Version >>, State, Req)
@@ -162,8 +165,9 @@ upgrade_terminate(#http_req{socket=Socket, transport=Transport}) ->
-spec websocket_handshake(#state{}, #http_req{}, any()) -> ok | none().
websocket_handshake(State=#state{version=0, origin=Origin,
- challenge=Challenge}, Req=#http_req{transport=Transport,
- raw_host=Host, port=Port, raw_path=Path, raw_qs=QS}, HandlerState) ->
+ challenge={Key1, Key2}}, Req=#http_req{socket=Socket,
+ transport=Transport, raw_host=Host, port=Port,
+ raw_path=Path, raw_qs=QS}, HandlerState) ->
Location = hixie76_location(Transport:name(), Host, Port, Path, QS),
{ok, Req2} = cowboy_http_req:reply(
<<"101 WebSocket Protocol Handshake">>,
@@ -171,9 +175,15 @@ websocket_handshake(State=#state{version=0, origin=Origin,
{<<"Upgrade">>, <<"WebSocket">>},
{<<"Sec-Websocket-Location">>, Location},
{<<"Sec-Websocket-Origin">>, Origin}],
- Challenge, Req#http_req{resp_state=waiting}),
+ [], Req#http_req{resp_state=waiting}),
+ %% We replied with a proper response. Proxies should be happy enough,
+ %% we can now read the 8 last bytes of the challenge keys and send
+ %% the challenge response directly to the socket.
+ {ok, Key3, Req3} = cowboy_http_req:body(8, Req2),
+ Challenge = hixie76_challenge(Key1, Key2, Key3),
+ Transport:send(Socket, Challenge),
handler_before_loop(State#state{messages=Transport:messages()},
- Req2, HandlerState, <<>>);
+ Req3, HandlerState, <<>>);
websocket_handshake(State=#state{challenge=Challenge},
Req=#http_req{transport=Transport}, HandlerState) ->
{ok, Req2} = cowboy_http_req:reply(