aboutsummaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
authorJeremy Ong <[email protected]>2012-12-19 11:34:44 -0800
committerLoïc Hoguin <[email protected]>2012-12-22 18:32:15 +0100
commit5f122d9fa670f9c239f3f68b4053b35884704f9e (patch)
tree2e08bf0a6117767e02eaa465382864e4dede69b7 /test
parenta07d063fd8fc8a1cfadcb35cc89e190695773fe8 (diff)
downloadcowboy-5f122d9fa670f9c239f3f68b4053b35884704f9e.tar.gz
cowboy-5f122d9fa670f9c239f3f68b4053b35884704f9e.tar.bz2
cowboy-5f122d9fa670f9c239f3f68b4053b35884704f9e.zip
Cancel timer only on websocket_data receives or sends
This commit prevents erlang messages from keeping a websocket connection alive. Previously, the timer was canceled upon any activity. Now, the timeout is only canceled when actual data is sent from the client. The handler_loop_timeout/1 function is called from websocket_data/4 instead of handler_before_loop/4. It is also called after every successful reply in handler_call/4.
Diffstat (limited to 'test')
-rw-r--r--test/ws_SUITE.erl79
-rw-r--r--test/ws_timeout_cancel_handler.erl33
2 files changed, 112 insertions, 0 deletions
diff --git a/test/ws_SUITE.erl b/test/ws_SUITE.erl
index 1a0583d..34befda 100644
--- a/test/ws_SUITE.erl
+++ b/test/ws_SUITE.erl
@@ -35,6 +35,8 @@
-export([ws_send_many/1]).
-export([ws_text_fragments/1]).
-export([ws_timeout_hibernate/1]).
+-export([ws_timeout_cancel/1]).
+-export([ws_timeout_reset/1]).
-export([ws_upgrade_with_opts/1]).
%% ct.
@@ -54,6 +56,8 @@ groups() ->
ws_send_many,
ws_text_fragments,
ws_timeout_hibernate,
+ ws_timeout_cancel,
+ ws_timeout_reset,
ws_upgrade_with_opts
],
[{ws, [], BaseTests}].
@@ -110,6 +114,7 @@ init_dispatch() ->
{text, <<"won't be received">>}]}
]},
{[<<"ws_timeout_hibernate">>], ws_timeout_hibernate_handler, []},
+ {[<<"ws_timeout_cancel">>], ws_timeout_cancel_handler, []},
{[<<"ws_upgrade_with_opts">>], ws_upgrade_with_opts_handler,
<<"failure">>}
]}
@@ -506,6 +511,80 @@ ws_timeout_hibernate(Config) ->
{error, closed} = gen_tcp:recv(Socket, 0, 6000),
ok.
+ws_timeout_cancel(Config) ->
+ %% Erlang messages to a socket should not cancel the timeout
+ {port, Port} = lists:keyfind(port, 1, Config),
+ {ok, Socket} = gen_tcp:connect("localhost", Port,
+ [binary, {active, false}, {packet, raw}]),
+ ok = gen_tcp:send(Socket, [
+ "GET /ws_timeout_cancel HTTP/1.1\r\n"
+ "Host: localhost\r\n"
+ "Connection: Upgrade\r\n"
+ "Upgrade: websocket\r\n"
+ "Sec-WebSocket-Origin: http://localhost\r\n"
+ "Sec-WebSocket-Version: 8\r\n"
+ "Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
+ "\r\n"]),
+ {ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
+ {ok, {http_response, {1, 1}, 101, "Switching Protocols"}, Rest}
+ = erlang:decode_packet(http, Handshake, []),
+ [Headers, <<>>] = websocket_headers(
+ erlang:decode_packet(httph, Rest, []), []),
+ {'Connection', "Upgrade"} = lists:keyfind('Connection', 1, Headers),
+ {'Upgrade', "websocket"} = lists:keyfind('Upgrade', 1, Headers),
+ {"sec-websocket-accept", "s3pPLMBiTxaQ9kYGzzhZRbK+xOo="}
+ = lists:keyfind("sec-websocket-accept", 1, Headers),
+ {ok, << 1:1, 0:3, 8:4, 0:8 >>} = gen_tcp:recv(Socket, 0, 6000),
+ {error, closed} = gen_tcp:recv(Socket, 0, 6000),
+ ok.
+
+ws_timeout_reset(Config) ->
+ %% Erlang messages across a socket should reset the timeout
+ {port, Port} = lists:keyfind(port, 1, Config),
+ {ok, Socket} = gen_tcp:connect("localhost", Port,
+ [binary, {active, false}, {packet, raw}]),
+ ok = gen_tcp:send(Socket, [
+ "GET /ws_timeout_cancel HTTP/1.1\r\n"
+ "Host: localhost\r\n"
+ "Connection: Upgrade\r\n"
+ "Upgrade: WebSocket\r\n"
+ "Origin: http://localhost\r\n"
+ "Sec-Websocket-Key1: Y\" 4 [email protected]!i\r\n"
+ "Sec-Websocket-Key2: 1711 M;4\\74 80<6\r\n"
+ "\r\n"]),
+ {ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
+ {ok, {http_response, {1, 1}, 101, "WebSocket Protocol Handshake"}, Rest}
+ = erlang:decode_packet(http, Handshake, []),
+ [Headers, <<>>] = websocket_headers(
+ erlang:decode_packet(httph, Rest, []), []),
+ {'Connection', "Upgrade"} = lists:keyfind('Connection', 1, Headers),
+ {'Upgrade', "WebSocket"} = lists:keyfind('Upgrade', 1, Headers),
+ {"sec-websocket-location", "ws://localhost/ws_timeout_cancel"}
+ = lists:keyfind("sec-websocket-location", 1, Headers),
+ {"sec-websocket-origin", "http://localhost"}
+ = lists:keyfind("sec-websocket-origin", 1, Headers),
+ ok = gen_tcp:send(Socket, <<15,245,8,18,2,204,133,33>>),
+ {ok, Body} = gen_tcp:recv(Socket, 0, 6000),
+ <<169,244,191,103,146,33,149,59,74,104,67,5,99,118,171,236>> = Body,
+ ok = gen_tcp:send(Socket, << 0, "msg sent", 255 >>),
+ {ok, << 0, "msg sent", 255 >>}
+ = gen_tcp:recv(Socket, 0, 6000),
+ ok = timer:sleep(500),
+ ok = gen_tcp:send(Socket, << 0, "msg sent", 255 >>),
+ {ok, << 0, "msg sent", 255 >>}
+ = gen_tcp:recv(Socket, 0, 6000),
+ ok = timer:sleep(500),
+ ok = gen_tcp:send(Socket, << 0, "msg sent", 255 >>),
+ {ok, << 0, "msg sent", 255 >>}
+ = gen_tcp:recv(Socket, 0, 6000),
+ ok = timer:sleep(500),
+ ok = gen_tcp:send(Socket, << 0, "msg sent", 255 >>),
+ {ok, << 0, "msg sent", 255 >>}
+ = gen_tcp:recv(Socket, 0, 6000),
+ {ok, << 255, 0 >>} = gen_tcp:recv(Socket, 0, 6000),
+ {error, closed} = gen_tcp:recv(Socket, 0, 6000),
+ ok.
+
ws_upgrade_with_opts(Config) ->
{port, Port} = lists:keyfind(port, 1, Config),
{ok, Socket} = gen_tcp:connect("localhost", Port,
diff --git a/test/ws_timeout_cancel_handler.erl b/test/ws_timeout_cancel_handler.erl
new file mode 100644
index 0000000..ee75d9b
--- /dev/null
+++ b/test/ws_timeout_cancel_handler.erl
@@ -0,0 +1,33 @@
+%% Feel free to use, reuse and abuse the code in this file.
+
+-module(ws_timeout_cancel_handler).
+-behaviour(cowboy_http_handler).
+-behaviour(cowboy_websocket_handler).
+-export([init/3, handle/2, terminate/2]).
+-export([websocket_init/3, websocket_handle/3,
+ websocket_info/3, websocket_terminate/3]).
+
+init(_Any, _Req, _Opts) ->
+ {upgrade, protocol, cowboy_websocket}.
+
+handle(_Req, _State) ->
+ exit(badarg).
+
+terminate(_Req, _State) ->
+ exit(badarg).
+
+websocket_init(_TransportName, Req, _Opts) ->
+ erlang:start_timer(500, self(), should_not_cancel_timer),
+ {ok, Req, undefined, 1000}.
+
+websocket_handle({text, Data}, Req, State) ->
+ {reply, {text, Data}, Req, State};
+websocket_handle({binary, Data}, Req, State) ->
+ {reply, {binary, Data}, Req, State}.
+
+websocket_info(_Info, Req, State) ->
+ erlang:start_timer(500, self(), should_not_cancel_timer),
+ {ok, Req, State}.
+
+websocket_terminate(_Reason, _Req, _State) ->
+ ok.