/test/ws_SUITE.erl

https://github.com/fgallaire/cowboy · Erlang · 710 lines · 629 code · 43 blank · 38 comment · 2 complexity · 70f502aa88ca66a2ae25d51e3a56f487 MD5 · raw file 

    

  1. %% Copyright (c) 2011-2014, Loïc Hoguin <essen@ninenines.eu>
    2. 

  2. %%
    3. 

  3. %% Permission to use, copy, modify, and/or distribute this software for any
    4. 

  4. %% purpose with or without fee is hereby granted, provided that the above
    5. 

  5. %% copyright notice and this permission notice appear in all copies.
    6. 

  6. %%
    7. 

  7. %% THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
    8. 

  8. %% WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
    9. 

  9. %% MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
    10. 

  10. %% ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
    11. 

  11. %% WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
    12. 

  12. %% ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
    13. 

  13. %% OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
    14. 

  14. 

  15. -module(ws_SUITE).
    16. 

  16. -compile(export_all).
    17. 

  17. 

  18. -import(cowboy_test, [config/2]).
    19. 

  19. 

  20. %% ct.
    21. 

  21. 

  22. all() ->
    23. 

  23. 	[{group, autobahn}, {group, ws}].
    24. 

  24. 

  25. groups() ->
    26. 

  26. 	BaseTests = cowboy_test:all(?MODULE) -- [autobahn_fuzzingclient],
    27. 

  27. 	[{autobahn, [], [autobahn_fuzzingclient]}, {ws, [parallel], BaseTests}].
    28. 

  28. 

  29. init_per_suite(Config) ->
    30. 

  30. 	Config.
    31. 

  31. 

  32. init_per_group(Name = autobahn, Config) ->
    33. 

  33. 	%% Some systems have it named pip2.
    34. 

  34. 	Out = os:cmd("pip show autobahntestsuite ; pip2 show autobahntestsuite"),
    35. 

  35. 	case string:str(Out, "autobahntestsuite") of
    36. 

  36. 		0 ->
    37. 

  37. 			ct:print("Skipping the autobahn group because the "
    38. 

  38. 				"Autobahn Test Suite is not installed.~nTo install it, "
    39. 

  39. 				"please follow the instructions on this page:~n~n    "
    40. 

  40. 				"http://autobahn.ws/testsuite/installation.html"),
    41. 

  41. 			{skip, "Autobahn Test Suite not installed."};
    42. 

  42. 		_ ->
    43. 

  43. 			{ok, _} = cowboy:start_http(Name, 100, [{port, 33080}], [
    44. 

  44. 				{env, [{dispatch, init_dispatch()}]}]),
    45. 

  45. 			Config
    46. 

  46. 	end;
    47. 

  47. init_per_group(Name = ws, Config) ->
    48. 

  48. 	cowboy_test:init_http(Name, [
    49. 

  49. 		{env, [{dispatch, init_dispatch()}]},
    50. 

  50. 		{compress, true}
    51. 

  51. 	], Config).
    52. 

  52. 

  53. end_per_group(Listener, _Config) ->
    54. 

  54. 	cowboy:stop_listener(Listener).
    55. 

  55. 

  56. %% Dispatch configuration.
    57. 

  57. 

  58. init_dispatch() ->
    59. 

  59. 	cowboy_router:compile([
    60. 

  60. 		{"localhost", [
    61. 

  61. 			{"/ws_echo", ws_echo, []},
    62. 

  62. 			{"/ws_echo_timer", ws_echo_timer, []},
    63. 

  63. 			{"/ws_init_shutdown", ws_init_shutdown, []},
    64. 

  64. 			{"/ws_send_many", ws_send_many, [
    65. 

  65. 				{sequence, [
    66. 

  66. 					{text, <<"one">>},
    67. 

  67. 					{text, <<"two">>},
    68. 

  68. 					{text, <<"seven!">>}]}
    69. 

  69. 			]},
    70. 

  70. 			{"/ws_send_close", ws_send_many, [
    71. 

  71. 				{sequence, [
    72. 

  72. 					{text, <<"send">>},
    73. 

  73. 					close,
    74. 

  74. 					{text, <<"won't be received">>}]}
    75. 

  75. 			]},
    76. 

  76. 			{"/ws_send_close_payload", ws_send_many, [
    77. 

  77. 				{sequence, [
    78. 

  78. 					{text, <<"send">>},
    79. 

  79. 					{close, 1001, <<"some text!">>},
    80. 

  80. 					{text, <<"won't be received">>}]}
    81. 

  81. 			]},
    82. 

  82. 			{"/ws_timeout_hibernate", ws_timeout_hibernate, []},
    83. 

  83. 			{"/ws_timeout_cancel", ws_timeout_cancel, []},
    84. 

  84. 			{"/ws_upgrade_with_opts", ws_upgrade_with_opts,
    85. 

  85. 				<<"failure">>}
    86. 

  86. 		]}
    87. 

  87. 	]).
    88. 

  88. 

  89. %% Tests.
    90. 

  90. 

  91. autobahn_fuzzingclient(Config) ->
    92. 

  92. 	Out = os:cmd("cd " ++ config(priv_dir, Config)
    93. 

  93. 		++ " && wstest -m fuzzingclient -s "
    94. 

  94. 		++ config(data_dir, Config) ++ "client.json"),
    95. 

  95. 	Report = config(priv_dir, Config) ++ "reports/servers/index.html",
    96. 

  96. 	ct:log("<h2><a href=\"~s\">Full report</a></h2>~n", [Report]),
    97. 

  97. 	ct:print("Autobahn Test Suite report: file://~s~n", [Report]),
    98. 

  98. 	ct:log("~s~n", [Out]),
    99. 

  99. 	{ok, HTML} = file:read_file(Report),
    100. 

  100. 	case length(binary:matches(HTML, <<"case_failed">>)) > 2 of
    101. 

  101. 		true -> error(failed);
    102. 

  102. 		false -> ok
    103. 

  103. 	end.
    104. 

  104. 

  105. %% We do not support hixie76 anymore.
    106. 

  106. ws0(Config) ->
    107. 

  107. 	{port, Port} = lists:keyfind(port, 1, Config),
    108. 

  108. 	{ok, Socket} = gen_tcp:connect("localhost", Port,
    109. 

  109. 		[binary, {active, false}, {packet, raw}]),
    110. 

  110. 	ok = gen_tcp:send(Socket,
    111. 

  111. 		"GET /ws_echo_timer HTTP/1.1\r\n"
    112. 

  112. 		"Host: localhost\r\n"
    113. 

  113. 		"Connection: Upgrade\r\n"
    114. 

  114. 		"Upgrade: WebSocket\r\n"
    115. 

  115. 		"Origin: http://localhost\r\n"
    116. 

  116. 		"Sec-Websocket-Key1: Y\" 4 1Lj!957b8@0H756!i\r\n"
    117. 

  117. 		"Sec-Websocket-Key2: 1711 M;4\\74  80<6\r\n"
    118. 

  118. 		"\r\n"),
    119. 

  119. 	{ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
    120. 

  120. 	{ok, {http_response, {1, 1}, 400, _}, _}
    121. 

  121. 		= erlang:decode_packet(http, Handshake, []).
    122. 

  122. 

  123. ws8(Config) ->
    124. 

  124. 	{port, Port} = lists:keyfind(port, 1, Config),
    125. 

  125. 	{ok, Socket} = gen_tcp:connect("localhost", Port,
    126. 

  126. 		[binary, {active, false}, {packet, raw}]),
    127. 

  127. 	ok = gen_tcp:send(Socket, [
    128. 

  128. 		"GET /ws_echo_timer HTTP/1.1\r\n"
    129. 

  129. 		"Host: localhost\r\n"
    130. 

  130. 		"Connection: Upgrade\r\n"
    131. 

  131. 		"Upgrade: websocket\r\n"
    132. 

  132. 		"Sec-WebSocket-Origin: http://localhost\r\n"
    133. 

  133. 		"Sec-WebSocket-Version: 8\r\n"
    134. 

  134. 		"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
    135. 

  135. 		"\r\n"]),
    136. 

  136. 	{ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
    137. 

  137. 	{ok, {http_response, {1, 1}, 101, "Switching Protocols"}, Rest}
    138. 

  138. 		= erlang:decode_packet(http, Handshake, []),
    139. 

  139. 	[Headers, <<>>] = do_decode_headers(
    140. 

  140. 		erlang:decode_packet(httph, Rest, []), []),
    141. 

  141. 	{'Connection', "Upgrade"} = lists:keyfind('Connection', 1, Headers),
    142. 

  142. 	{'Upgrade', "websocket"} = lists:keyfind('Upgrade', 1, Headers),
    143. 

  143. 	{"sec-websocket-accept", "s3pPLMBiTxaQ9kYGzzhZRbK+xOo="}
    144. 

  144. 		= lists:keyfind("sec-websocket-accept", 1, Headers),
    145. 

  145. 	ok = gen_tcp:send(Socket, << 16#81, 16#85, 16#37, 16#fa, 16#21, 16#3d,
    146. 

  146. 		16#7f, 16#9f, 16#4d, 16#51, 16#58 >>),
    147. 

  147. 	{ok, << 1:1, 0:3, 1:4, 0:1, 5:7, "Hello" >>}
    148. 

  148. 		= gen_tcp:recv(Socket, 0, 6000),
    149. 

  149. 	{ok, << 1:1, 0:3, 1:4, 0:1, 14:7, "websocket_init" >>}
    150. 

  150. 		= gen_tcp:recv(Socket, 0, 6000),
    151. 

  151. 	{ok, << 1:1, 0:3, 1:4, 0:1, 16:7, "websocket_handle" >>}
    152. 

  152. 		= gen_tcp:recv(Socket, 0, 6000),
    153. 

  153. 	{ok, << 1:1, 0:3, 1:4, 0:1, 16:7, "websocket_handle" >>}
    154. 

  154. 		= gen_tcp:recv(Socket, 0, 6000),
    155. 

  155. 	{ok, << 1:1, 0:3, 1:4, 0:1, 16:7, "websocket_handle" >>}
    156. 

  156. 		= gen_tcp:recv(Socket, 0, 6000),
    157. 

  157. 	ok = gen_tcp:send(Socket, << 1:1, 0:3, 9:4, 1:1, 0:7, 0:32 >>), %% ping
    158. 

  158. 	{ok, << 1:1, 0:3, 10:4, 0:8 >>} = gen_tcp:recv(Socket, 0, 6000), %% pong
    159. 

  159. 	ok = gen_tcp:send(Socket, << 1:1, 0:3, 8:4, 1:1, 0:7, 0:32 >>), %% close
    160. 

  160. 	{ok, << 1:1, 0:3, 8:4, 0:8 >>} = gen_tcp:recv(Socket, 0, 6000),
    161. 

  161. 	{error, closed} = gen_tcp:recv(Socket, 0, 6000),
    162. 

  162. 	ok.
    163. 

  163. 

  164. ws8_init_shutdown(Config) ->
    165. 

  165. 	{port, Port} = lists:keyfind(port, 1, Config),
    166. 

  166. 	{ok, Socket} = gen_tcp:connect("localhost", Port,
    167. 

  167. 		[binary, {active, false}, {packet, raw}]),
    168. 

  168. 	ok = gen_tcp:send(Socket, [
    169. 

  169. 		"GET /ws_init_shutdown HTTP/1.1\r\n"
    170. 

  170. 		"Host: localhost\r\n"
    171. 

  171. 		"Connection: Upgrade\r\n"
    172. 

  172. 		"Upgrade: websocket\r\n"
    173. 

  173. 		"Sec-WebSocket-Origin: http://localhost\r\n"
    174. 

  174. 		"Sec-WebSocket-Version: 8\r\n"
    175. 

  175. 		"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
    176. 

  176. 		"\r\n"]),
    177. 

  177. 	{ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
    178. 

  178. 	{ok, {http_response, {1, 1}, 403, "Forbidden"}, _Rest}
    179. 

  179. 		= erlang:decode_packet(http, Handshake, []),
    180. 

  180. 	{error, closed} = gen_tcp:recv(Socket, 0, 6000),
    181. 

  181. 	ok.
    182. 

  182. 

  183. ws8_single_bytes(Config) ->
    184. 

  184. 	{port, Port} = lists:keyfind(port, 1, Config),
    185. 

  185. 	{ok, Socket} = gen_tcp:connect("localhost", Port,
    186. 

  186. 		[binary, {active, false}, {packet, raw}]),
    187. 

  187. 	ok = gen_tcp:send(Socket, [
    188. 

  188. 		"GET /ws_echo_timer HTTP/1.1\r\n"
    189. 

  189. 		"Host: localhost\r\n"
    190. 

  190. 		"Connection: Upgrade\r\n"
    191. 

  191. 		"Upgrade: websocket\r\n"
    192. 

  192. 		"Sec-WebSocket-Origin: http://localhost\r\n"
    193. 

  193. 		"Sec-WebSocket-Version: 8\r\n"
    194. 

  194. 		"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
    195. 

  195. 		"\r\n"]),
    196. 

  196. 	{ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
    197. 

  197. 	{ok, {http_response, {1, 1}, 101, "Switching Protocols"}, Rest}
    198. 

  198. 		= erlang:decode_packet(http, Handshake, []),
    199. 

  199. 	[Headers, <<>>] = do_decode_headers(
    200. 

  200. 		erlang:decode_packet(httph, Rest, []), []),
    201. 

  201. 	{'Connection', "Upgrade"} = lists:keyfind('Connection', 1, Headers),
    202. 

  202. 	{'Upgrade', "websocket"} = lists:keyfind('Upgrade', 1, Headers),
    203. 

  203. 	{"sec-websocket-accept", "s3pPLMBiTxaQ9kYGzzhZRbK+xOo="}
    204. 

  204. 		= lists:keyfind("sec-websocket-accept", 1, Headers),
    205. 

  205. 	ok = gen_tcp:send(Socket, << 16#81 >>), %% send one byte
    206. 

  206. 	ok = timer:sleep(100), %% sleep for a period
    207. 

  207. 	ok = gen_tcp:send(Socket, << 16#85 >>), %% send another and so on
    208. 

  208. 	ok = timer:sleep(100),
    209. 

  209. 	ok = gen_tcp:send(Socket, << 16#37 >>),
    210. 

  210. 	ok = timer:sleep(100),
    211. 

  211. 	ok = gen_tcp:send(Socket, << 16#fa >>),
    212. 

  212. 	ok = timer:sleep(100),
    213. 

  213. 	ok = gen_tcp:send(Socket, << 16#21 >>),
    214. 

  214. 	ok = timer:sleep(100),
    215. 

  215. 	ok = gen_tcp:send(Socket, << 16#3d >>),
    216. 

  216. 	ok = timer:sleep(100),
    217. 

  217. 	ok = gen_tcp:send(Socket, << 16#7f >>),
    218. 

  218. 	ok = timer:sleep(100),
    219. 

  219. 	ok = gen_tcp:send(Socket, << 16#9f >>),
    220. 

  220. 	ok = timer:sleep(100),
    221. 

  221. 	ok = gen_tcp:send(Socket, << 16#4d >>),
    222. 

  222. 	ok = timer:sleep(100),
    223. 

  223. 	ok = gen_tcp:send(Socket, << 16#51 >>),
    224. 

  224. 	ok = timer:sleep(100),
    225. 

  225. 	ok = gen_tcp:send(Socket, << 16#58 >>),
    226. 

  226. 	{ok, << 1:1, 0:3, 1:4, 0:1, 14:7, "websocket_init" >>}
    227. 

  227. 		= gen_tcp:recv(Socket, 0, 6000),
    228. 

  228. 	{ok, << 1:1, 0:3, 1:4, 0:1, 5:7, "Hello" >>}
    229. 

  229. 		= gen_tcp:recv(Socket, 0, 6000),
    230. 

  230. 	{ok, << 1:1, 0:3, 1:4, 0:1, 16:7, "websocket_handle" >>}
    231. 

  231. 		= gen_tcp:recv(Socket, 0, 6000),
    232. 

  232. 	{ok, << 1:1, 0:3, 1:4, 0:1, 16:7, "websocket_handle" >>}
    233. 

  233. 		= gen_tcp:recv(Socket, 0, 6000),
    234. 

  234. 	{ok, << 1:1, 0:3, 1:4, 0:1, 16:7, "websocket_handle" >>}
    235. 

  235. 		= gen_tcp:recv(Socket, 0, 6000),
    236. 

  236. 	ok = gen_tcp:send(Socket, << 1:1, 0:3, 9:4, 1:1, 0:7, 0:32 >>), %% ping
    237. 

  237. 	{ok, << 1:1, 0:3, 10:4, 0:8 >>} = gen_tcp:recv(Socket, 0, 6000), %% pong
    238. 

  238. 	ok = gen_tcp:send(Socket, << 1:1, 0:3, 8:4, 1:1, 0:7, 0:32 >>), %% close
    239. 

  239. 	{ok, << 1:1, 0:3, 8:4, 0:8 >>} = gen_tcp:recv(Socket, 0, 6000),
    240. 

  240. 	{error, closed} = gen_tcp:recv(Socket, 0, 6000),
    241. 

  241. 	ok.
    242. 

  242. 

  243. ws13(Config) ->
    244. 

  244. 	{port, Port} = lists:keyfind(port, 1, Config),
    245. 

  245. 	{ok, Socket} = gen_tcp:connect("localhost", Port,
    246. 

  246. 		[binary, {active, false}, {packet, raw}]),
    247. 

  247. 	ok = gen_tcp:send(Socket, [
    248. 

  248. 		"GET /ws_echo_timer HTTP/1.1\r\n"
    249. 

  249. 		"Host: localhost\r\n"
    250. 

  250. 		"Connection: Upgrade\r\n"
    251. 

  251. 		"Origin: http://localhost\r\n"
    252. 

  252. 		"Sec-WebSocket-Version: 13\r\n"
    253. 

  253. 		"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
    254. 

  254. 		"Upgrade: websocket\r\n"
    255. 

  255. 		"\r\n"]),
    256. 

  256. 	{ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
    257. 

  257. 	{ok, {http_response, {1, 1}, 101, "Switching Protocols"}, Rest}
    258. 

  258. 		= erlang:decode_packet(http, Handshake, []),
    259. 

  259. 	[Headers, <<>>] = do_decode_headers(
    260. 

  260. 		erlang:decode_packet(httph, Rest, []), []),
    261. 

  261. 	{'Connection', "Upgrade"} = lists:keyfind('Connection', 1, Headers),
    262. 

  262. 	{'Upgrade', "websocket"} = lists:keyfind('Upgrade', 1, Headers),
    263. 

  263. 	{"sec-websocket-accept", "s3pPLMBiTxaQ9kYGzzhZRbK+xOo="}
    264. 

  264. 		= lists:keyfind("sec-websocket-accept", 1, Headers),
    265. 

  265. 	%% text
    266. 

  266. 	ok = gen_tcp:send(Socket, << 16#81, 16#85, 16#37, 16#fa, 16#21, 16#3d,
    267. 

  267. 		16#7f, 16#9f, 16#4d, 16#51, 16#58 >>),
    268. 

  268. 	{ok, << 1:1, 0:3, 1:4, 0:1, 5:7, "Hello" >>}
    269. 

  269. 		= gen_tcp:recv(Socket, 0, 6000),
    270. 

  270. 	%% binary (empty)
    271. 

  271. 	ok = gen_tcp:send(Socket, << 1:1, 0:3, 2:4, 1:1, 0:7, 0:32 >>),
    272. 

  272. 	{ok, << 1:1, 0:3, 2:4, 0:8 >>} = gen_tcp:recv(Socket, 0, 6000),
    273. 

  273. 	%% binary
    274. 

  274. 	ok = gen_tcp:send(Socket, << 16#82, 16#85, 16#37, 16#fa, 16#21, 16#3d,
    275. 

  275. 		16#7f, 16#9f, 16#4d, 16#51, 16#58 >>),
    276. 

  276. 	{ok, << 1:1, 0:3, 2:4, 0:1, 5:7, "Hello" >>}
    277. 

  277. 		= gen_tcp:recv(Socket, 0, 6000),
    278. 

  278. 	%% Receives.
    279. 

  279. 	{ok, << 1:1, 0:3, 1:4, 0:1, 14:7, "websocket_init" >>}
    280. 

  280. 		= gen_tcp:recv(Socket, 0, 6000),
    281. 

  281. 	{ok, << 1:1, 0:3, 1:4, 0:1, 16:7, "websocket_handle" >>}
    282. 

  282. 		= gen_tcp:recv(Socket, 0, 6000),
    283. 

  283. 	{ok, << 1:1, 0:3, 1:4, 0:1, 16:7, "websocket_handle" >>}
    284. 

  284. 		= gen_tcp:recv(Socket, 0, 6000),
    285. 

  285. 	{ok, << 1:1, 0:3, 1:4, 0:1, 16:7, "websocket_handle" >>}
    286. 

  286. 		= gen_tcp:recv(Socket, 0, 6000),
    287. 

  287. 	ok = gen_tcp:send(Socket, << 1:1, 0:3, 9:4, 1:1, 0:7, 0:32 >>), %% ping
    288. 

  288. 	{ok, << 1:1, 0:3, 10:4, 0:8 >>} = gen_tcp:recv(Socket, 0, 6000), %% pong
    289. 

  289. 	ok = gen_tcp:send(Socket, << 1:1, 0:3, 8:4, 1:1, 0:7, 0:32 >>), %% close
    290. 

  290. 	{ok, << 1:1, 0:3, 8:4, 0:8 >>} = gen_tcp:recv(Socket, 0, 6000),
    291. 

  291. 	{error, closed} = gen_tcp:recv(Socket, 0, 6000),
    292. 

  292. 	ok.
    293. 

  293. 

  294. ws_deflate(Config) ->
    295. 

  295. 	{port, Port} = lists:keyfind(port, 1, Config),
    296. 

  296. 	{ok, Socket} = gen_tcp:connect("localhost", Port,
    297. 

  297. 		[binary, {active, false}, {packet, raw}, {nodelay, true}]),
    298. 

  298. 	ok = gen_tcp:send(Socket, [
    299. 

  299. 		"GET /ws_echo HTTP/1.1\r\n"
    300. 

  300. 		"Host: localhost\r\n"
    301. 

  301. 		"Connection: Upgrade\r\n"
    302. 

  302. 		"Upgrade: websocket\r\n"
    303. 

  303. 		"Sec-WebSocket-Origin: http://localhost\r\n"
    304. 

  304. 		"Sec-WebSocket-Version: 8\r\n"
    305. 

  305. 		"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
    306. 

  306. 		"Sec-WebSocket-Extensions: x-webkit-deflate-frame\r\n"
    307. 

  307. 		"\r\n"]),
    308. 

  308. 	{ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
    309. 

  309. 	{ok, {http_response, {1, 1}, 101, "Switching Protocols"}, Rest}
    310. 

  310. 		= erlang:decode_packet(http, Handshake, []),
    311. 

  311. 	[Headers, <<>>] = do_decode_headers(
    312. 

  312. 		erlang:decode_packet(httph, Rest, []), []),
    313. 

  313. 	{'Connection', "Upgrade"} = lists:keyfind('Connection', 1, Headers),
    314. 

  314. 	{'Upgrade', "websocket"} = lists:keyfind('Upgrade', 1, Headers),
    315. 

  315. 	{"sec-websocket-accept", "s3pPLMBiTxaQ9kYGzzhZRbK+xOo="}
    316. 

  316. 		= lists:keyfind("sec-websocket-accept", 1, Headers),
    317. 

  317. 	{"sec-websocket-extensions", "x-webkit-deflate-frame"}
    318. 

  318. 		= lists:keyfind("sec-websocket-extensions", 1, Headers),
    319. 

  319. 

  320. 	Mask = 16#11223344,
    321. 

  321. 	Hello = << 242, 72, 205, 201, 201, 7, 0 >>,
    322. 

  322. 	MaskedHello = do_mask(Hello, Mask, <<>>),
    323. 

  323. 

  324. 	% send compressed text frame containing the Hello string
    325. 

  325. 	ok = gen_tcp:send(Socket, << 1:1, 1:1, 0:2, 1:4, 1:1, 7:7, Mask:32,
    326. 

  326. 		MaskedHello/binary >>),
    327. 

  327. 	% receive compressed text frame containing the Hello string
    328. 

  328. 	{ok, << 1:1, 1:1, 0:2, 1:4, 0:1, 7:7, Hello/binary >>}
    329. 

  329. 		= gen_tcp:recv(Socket, 0, 6000),
    330. 

  330. 

  331. 	ok = gen_tcp:send(Socket, << 1:1, 0:3, 8:4, 1:1, 0:7, 0:32 >>), %% close
    332. 

  332. 	{ok, << 1:1, 0:3, 8:4, 0:8 >>} = gen_tcp:recv(Socket, 0, 6000),
    333. 

  333. 	{error, closed} = gen_tcp:recv(Socket, 0, 6000),
    334. 

  334. 	ok.
    335. 

  335. 

  336. ws_deflate_chunks(Config) ->
    337. 

  337. 	{port, Port} = lists:keyfind(port, 1, Config),
    338. 

  338. 	{ok, Socket} = gen_tcp:connect("localhost", Port,
    339. 

  339. 		[binary, {active, false}, {packet, raw}, {nodelay, true}]),
    340. 

  340. 	ok = gen_tcp:send(Socket, [
    341. 

  341. 		"GET /ws_echo HTTP/1.1\r\n"
    342. 

  342. 		"Host: localhost\r\n"
    343. 

  343. 		"Connection: Upgrade\r\n"
    344. 

  344. 		"Upgrade: websocket\r\n"
    345. 

  345. 		"Sec-WebSocket-Origin: http://localhost\r\n"
    346. 

  346. 		"Sec-WebSocket-Version: 8\r\n"
    347. 

  347. 		"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
    348. 

  348. 		"Sec-WebSocket-Extensions: x-webkit-deflate-frame\r\n"
    349. 

  349. 		"\r\n"]),
    350. 

  350. 	{ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
    351. 

  351. 	{ok, {http_response, {1, 1}, 101, "Switching Protocols"}, Rest}
    352. 

  352. 		= erlang:decode_packet(http, Handshake, []),
    353. 

  353. 	[Headers, <<>>] = do_decode_headers(
    354. 

  354. 		erlang:decode_packet(httph, Rest, []), []),
    355. 

  355. 	{'Connection', "Upgrade"} = lists:keyfind('Connection', 1, Headers),
    356. 

  356. 	{'Upgrade', "websocket"} = lists:keyfind('Upgrade', 1, Headers),
    357. 

  357. 	{"sec-websocket-accept", "s3pPLMBiTxaQ9kYGzzhZRbK+xOo="}
    358. 

  358. 		= lists:keyfind("sec-websocket-accept", 1, Headers),
    359. 

  359. 	{"sec-websocket-extensions", "x-webkit-deflate-frame"}
    360. 

  360. 		= lists:keyfind("sec-websocket-extensions", 1, Headers),
    361. 

  361. 

  362. 	Mask = 16#11223344,
    363. 

  363. 	Hello = << 242, 72, 205, 201, 201, 7, 0 >>,
    364. 

  364. 	MaskedHello = do_mask(Hello, Mask, <<>>),
    365. 

  365. 

  366. 	% send compressed text frame containing the Hello string
    367. 

  367. 	ok = gen_tcp:send(Socket, << 1:1, 1:1, 0:2, 1:4, 1:1, 7:7, Mask:32,
    368. 

  368. 		(binary:part(MaskedHello, 0, 4))/binary >>),
    369. 

  369. 	ok = timer:sleep(100),
    370. 

  370. 	ok = gen_tcp:send(Socket, binary:part(MaskedHello, 4, 3)),
    371. 

  371. 

  372. 	% receive compressed text frame containing the Hello string
    373. 

  373. 	{ok, << 1:1, 1:1, 0:2, 1:4, 0:1, 7:7, Hello/binary >>}
    374. 

  374. 		= gen_tcp:recv(Socket, 0, 6000),
    375. 

  375. 

  376. 	ok = gen_tcp:send(Socket, << 1:1, 0:3, 8:4, 1:1, 0:7, 0:32 >>), %% close
    377. 

  377. 	{ok, << 1:1, 0:3, 8:4, 0:8 >>} = gen_tcp:recv(Socket, 0, 6000),
    378. 

  378. 	{error, closed} = gen_tcp:recv(Socket, 0, 6000),
    379. 

  379. 	ok.
    380. 

  380. 

  381. ws_deflate_fragments(Config) ->
    382. 

  382. 	{port, Port} = lists:keyfind(port, 1, Config),
    383. 

  383. 	{ok, Socket} = gen_tcp:connect("localhost", Port,
    384. 

  384. 		[binary, {active, false}, {packet, raw}, {nodelay, true}]),
    385. 

  385. 	ok = gen_tcp:send(Socket, [
    386. 

  386. 		"GET /ws_echo HTTP/1.1\r\n"
    387. 

  387. 		"Host: localhost\r\n"
    388. 

  388. 		"Connection: Upgrade\r\n"
    389. 

  389. 		"Upgrade: websocket\r\n"
    390. 

  390. 		"Sec-WebSocket-Origin: http://localhost\r\n"
    391. 

  391. 		"Sec-WebSocket-Version: 8\r\n"
    392. 

  392. 		"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
    393. 

  393. 		"Sec-WebSocket-Extensions: x-webkit-deflate-frame\r\n"
    394. 

  394. 		"\r\n"]),
    395. 

  395. 	{ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
    396. 

  396. 	{ok, {http_response, {1, 1}, 101, "Switching Protocols"}, Rest}
    397. 

  397. 		= erlang:decode_packet(http, Handshake, []),
    398. 

  398. 	[Headers, <<>>] = do_decode_headers(
    399. 

  399. 		erlang:decode_packet(httph, Rest, []), []),
    400. 

  400. 	{'Connection', "Upgrade"} = lists:keyfind('Connection', 1, Headers),
    401. 

  401. 	{'Upgrade', "websocket"} = lists:keyfind('Upgrade', 1, Headers),
    402. 

  402. 	{"sec-websocket-accept", "s3pPLMBiTxaQ9kYGzzhZRbK+xOo="}
    403. 

  403. 		= lists:keyfind("sec-websocket-accept", 1, Headers),
    404. 

  404. 	{"sec-websocket-extensions", "x-webkit-deflate-frame"}
    405. 

  405. 		= lists:keyfind("sec-websocket-extensions", 1, Headers),
    406. 

  406. 

  407. 	Mask = 16#11223344,
    408. 

  408. 	Hello = << 242, 72, 205, 201, 201, 7, 0 >>,
    409. 

  409. 

  410. 	% send compressed text frame containing the Hello string
    411. 

  411. 	% as 2 separate fragments
    412. 

  412. 	ok = gen_tcp:send(Socket, << 0:1, 1:1, 0:2, 1:4, 1:1, 4:7, Mask:32,
    413. 

  413. 		(do_mask(binary:part(Hello, 0, 4), Mask, <<>>))/binary >>),
    414. 

  414. 	ok = gen_tcp:send(Socket, << 1:1, 1:1, 0:2, 0:4, 1:1, 3:7, Mask:32,
    415. 

  415. 		(do_mask(binary:part(Hello, 4, 3), Mask, <<>>))/binary >>),
    416. 

  416. 	% receive compressed text frame containing the Hello string
    417. 

  417. 	{ok, << 1:1, 1:1, 0:2, 1:4, 0:1, 7:7, Hello/binary >>}
    418. 

  418. 		= gen_tcp:recv(Socket, 0, 6000),
    419. 

  419. 

  420. 	ok = gen_tcp:send(Socket, << 1:1, 0:3, 8:4, 1:1, 0:7, 0:32 >>), %% close
    421. 

  421. 	{ok, << 1:1, 0:3, 8:4, 0:8 >>} = gen_tcp:recv(Socket, 0, 6000),
    422. 

  422. 	{error, closed} = gen_tcp:recv(Socket, 0, 6000),
    423. 

  423. 	ok.
    424. 

  424. 

  425. ws_send_close(Config) ->
    426. 

  426. 	{port, Port} = lists:keyfind(port, 1, Config),
    427. 

  427. 	{ok, Socket} = gen_tcp:connect("localhost", Port,
    428. 

  428. 		[binary, {active, false}, {packet, raw}]),
    429. 

  429. 	ok = gen_tcp:send(Socket, [
    430. 

  430. 		"GET /ws_send_close HTTP/1.1\r\n"
    431. 

  431. 		"Host: localhost\r\n"
    432. 

  432. 		"Connection: Upgrade\r\n"
    433. 

  433. 		"Upgrade: websocket\r\n"
    434. 

  434. 		"Sec-WebSocket-Origin: http://localhost\r\n"
    435. 

  435. 		"Sec-WebSocket-Version: 8\r\n"
    436. 

  436. 		"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
    437. 

  437. 		"\r\n"]),
    438. 

  438. 	{ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
    439. 

  439. 	{ok, {http_response, {1, 1}, 101, "Switching Protocols"}, Rest}
    440. 

  440. 		= erlang:decode_packet(http, Handshake, []),
    441. 

  441. 	[Headers, <<>>] = do_decode_headers(
    442. 

  442. 		erlang:decode_packet(httph, Rest, []), []),
    443. 

  443. 	{'Connection', "Upgrade"} = lists:keyfind('Connection', 1, Headers),
    444. 

  444. 	{'Upgrade', "websocket"} = lists:keyfind('Upgrade', 1, Headers),
    445. 

  445. 	{"sec-websocket-accept", "s3pPLMBiTxaQ9kYGzzhZRbK+xOo="}
    446. 

  446. 		= lists:keyfind("sec-websocket-accept", 1, Headers),
    447. 

  447. 	%% We catch all frames at once and check them directly.
    448. 

  448. 	{ok, Many} = gen_tcp:recv(Socket, 8, 6000),
    449. 

  449. 	<< 1:1, 0:3, 1:4, 0:1, 4:7, "send",
    450. 

  450. 		1:1, 0:3, 8:4, 0:8 >> = Many,
    451. 

  451. 	{error, closed} = gen_tcp:recv(Socket, 0, 6000),
    452. 

  452. 	ok.
    453. 

  453. 

  454. ws_send_close_payload(Config) ->
    455. 

  455. 	{port, Port} = lists:keyfind(port, 1, Config),
    456. 

  456. 	{ok, Socket} = gen_tcp:connect("localhost", Port,
    457. 

  457. 		[binary, {active, false}, {packet, raw}]),
    458. 

  458. 	ok = gen_tcp:send(Socket, [
    459. 

  459. 		"GET /ws_send_close_payload HTTP/1.1\r\n"
    460. 

  460. 		"Host: localhost\r\n"
    461. 

  461. 		"Connection: Upgrade\r\n"
    462. 

  462. 		"Upgrade: websocket\r\n"
    463. 

  463. 		"Sec-WebSocket-Origin: http://localhost\r\n"
    464. 

  464. 		"Sec-WebSocket-Version: 8\r\n"
    465. 

  465. 		"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
    466. 

  466. 		"\r\n"]),
    467. 

  467. 	{ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
    468. 

  468. 	{ok, {http_response, {1, 1}, 101, "Switching Protocols"}, Rest}
    469. 

  469. 		= erlang:decode_packet(http, Handshake, []),
    470. 

  470. 	[Headers, <<>>] = do_decode_headers(
    471. 

  471. 		erlang:decode_packet(httph, Rest, []), []),
    472. 

  472. 	{'Connection', "Upgrade"} = lists:keyfind('Connection', 1, Headers),
    473. 

  473. 	{'Upgrade', "websocket"} = lists:keyfind('Upgrade', 1, Headers),
    474. 

  474. 	{"sec-websocket-accept", "s3pPLMBiTxaQ9kYGzzhZRbK+xOo="}
    475. 

  475. 		= lists:keyfind("sec-websocket-accept", 1, Headers),
    476. 

  476. 	%% We catch all frames at once and check them directly.
    477. 

  477. 	{ok, Many} = gen_tcp:recv(Socket, 20, 6000),
    478. 

  478. 	<< 1:1, 0:3, 1:4, 0:1, 4:7, "send",
    479. 

  479. 		1:1, 0:3, 8:4, 0:1, 12:7, 1001:16, "some text!" >> = Many,
    480. 

  480. 	{error, closed} = gen_tcp:recv(Socket, 0, 6000),
    481. 

  481. 	ok.
    482. 

  482. 

  483. ws_send_many(Config) ->
    484. 

  484. 	{port, Port} = lists:keyfind(port, 1, Config),
    485. 

  485. 	{ok, Socket} = gen_tcp:connect("localhost", Port,
    486. 

  486. 		[binary, {active, false}, {packet, raw}]),
    487. 

  487. 	ok = gen_tcp:send(Socket, [
    488. 

  488. 		"GET /ws_send_many HTTP/1.1\r\n"
    489. 

  489. 		"Host: localhost\r\n"
    490. 

  490. 		"Connection: Upgrade\r\n"
    491. 

  491. 		"Upgrade: websocket\r\n"
    492. 

  492. 		"Sec-WebSocket-Origin: http://localhost\r\n"
    493. 

  493. 		"Sec-WebSocket-Version: 8\r\n"
    494. 

  494. 		"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
    495. 

  495. 		"\r\n"]),
    496. 

  496. 	{ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
    497. 

  497. 	{ok, {http_response, {1, 1}, 101, "Switching Protocols"}, Rest}
    498. 

  498. 		= erlang:decode_packet(http, Handshake, []),
    499. 

  499. 	[Headers, <<>>] = do_decode_headers(
    500. 

  500. 		erlang:decode_packet(httph, Rest, []), []),
    501. 

  501. 	{'Connection', "Upgrade"} = lists:keyfind('Connection', 1, Headers),
    502. 

  502. 	{'Upgrade', "websocket"} = lists:keyfind('Upgrade', 1, Headers),
    503. 

  503. 	{"sec-websocket-accept", "s3pPLMBiTxaQ9kYGzzhZRbK+xOo="}
    504. 

  504. 		= lists:keyfind("sec-websocket-accept", 1, Headers),
    505. 

  505. 	%% We catch all frames at once and check them directly.
    506. 

  506. 	{ok, Many} = gen_tcp:recv(Socket, 18, 6000),
    507. 

  507. 	<< 1:1, 0:3, 1:4, 0:1, 3:7, "one",
    508. 

  508. 		1:1, 0:3, 1:4, 0:1, 3:7, "two",
    509. 

  509. 		1:1, 0:3, 1:4, 0:1, 6:7, "seven!" >> = Many,
    510. 

  510. 	ok = gen_tcp:send(Socket, << 1:1, 0:3, 8:4, 1:1, 0:7, 0:32 >>), %% close
    511. 

  511. 	{ok, << 1:1, 0:3, 8:4, 0:8 >>} = gen_tcp:recv(Socket, 0, 6000),
    512. 

  512. 	{error, closed} = gen_tcp:recv(Socket, 0, 6000),
    513. 

  513. 	ok.
    514. 

  514. 

  515. ws_text_fragments(Config) ->
    516. 

  516. 	{port, Port} = lists:keyfind(port, 1, Config),
    517. 

  517. 	{ok, Socket} = gen_tcp:connect("localhost", Port,
    518. 

  518. 		[binary, {active, false}, {packet, raw}]),
    519. 

  519. 	ok = gen_tcp:send(Socket, [
    520. 

  520. 		"GET /ws_echo HTTP/1.1\r\n"
    521. 

  521. 		"Host: localhost\r\n"
    522. 

  522. 		"Connection: Upgrade\r\n"
    523. 

  523. 		"Upgrade: websocket\r\n"
    524. 

  524. 		"Sec-WebSocket-Origin: http://localhost\r\n"
    525. 

  525. 		"Sec-WebSocket-Version: 8\r\n"
    526. 

  526. 		"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
    527. 

  527. 		"\r\n"]),
    528. 

  528. 	{ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
    529. 

  529. 	{ok, {http_response, {1, 1}, 101, "Switching Protocols"}, Rest}
    530. 

  530. 		= erlang:decode_packet(http, Handshake, []),
    531. 

  531. 	[Headers, <<>>] = do_decode_headers(
    532. 

  532. 		erlang:decode_packet(httph, Rest, []), []),
    533. 

  533. 	{'Connection', "Upgrade"} = lists:keyfind('Connection', 1, Headers),
    534. 

  534. 	{'Upgrade', "websocket"} = lists:keyfind('Upgrade', 1, Headers),
    535. 

  535. 	{"sec-websocket-accept", "s3pPLMBiTxaQ9kYGzzhZRbK+xOo="}
    536. 

  536. 		= lists:keyfind("sec-websocket-accept", 1, Headers),
    537. 

  537. 

  538. 	ok = gen_tcp:send(Socket, [
    539. 

  539. 		<< 0:1, 0:3, 1:4, 1:1, 5:7 >>,
    540. 

  540. 		<< 16#37 >>, << 16#fa >>, << 16#21 >>, << 16#3d >>, << 16#7f >>,
    541. 

  541. 		<< 16#9f >>, << 16#4d >>, << 16#51 >>, << 16#58 >>]),
    542. 

  542. 	ok = gen_tcp:send(Socket, [
    543. 

  543. 		<< 1:1, 0:3, 0:4, 1:1, 5:7 >>,
    544. 

  544. 		<< 16#37 >>, << 16#fa >>, << 16#21 >>, << 16#3d >>, << 16#7f >>,
    545. 

  545. 		<< 16#9f >>, << 16#4d >>, << 16#51 >>, << 16#58 >>]),
    546. 

  546. 	{ok, << 1:1, 0:3, 1:4, 0:1, 10:7, "HelloHello" >>}
    547. 

  547. 		= gen_tcp:recv(Socket, 0, 6000),
    548. 

  548. 

  549. 	ok = gen_tcp:send(Socket, [
    550. 

  550. 		%% #1
    551. 

  551. 		<< 0:1, 0:3, 1:4, 1:1, 5:7 >>,
    552. 

  552. 		<< 16#37 >>, << 16#fa >>, << 16#21 >>, << 16#3d >>, << 16#7f >>,
    553. 

  553. 		<< 16#9f >>, << 16#4d >>, << 16#51 >>, << 16#58 >>,
    554. 

  554. 		%% #2
    555. 

  555. 		<< 0:1, 0:3, 0:4, 1:1, 5:7 >>,
    556. 

  556. 		<< 16#37 >>, << 16#fa >>, << 16#21 >>, << 16#3d >>, << 16#7f >>,
    557. 

  557. 		<< 16#9f >>, << 16#4d >>, << 16#51 >>, << 16#58 >>,
    558. 

  558. 		%% #3
    559. 

  559. 		<< 1:1, 0:3, 0:4, 1:1, 5:7 >>,
    560. 

  560. 		<< 16#37 >>, << 16#fa >>, << 16#21 >>, << 16#3d >>, << 16#7f >>,
    561. 

  561. 		<< 16#9f >>, << 16#4d >>, << 16#51 >>, << 16#58 >>]),
    562. 

  562. 	{ok, << 1:1, 0:3, 1:4, 0:1, 15:7, "HelloHelloHello" >>}
    563. 

  563. 		= gen_tcp:recv(Socket, 0, 6000),
    564. 

  564. 	ok = gen_tcp:send(Socket, << 1:1, 0:3, 8:4, 1:1, 0:7, 0:32 >>), %% close
    565. 

  565. 	{ok, << 1:1, 0:3, 8:4, 0:8 >>} = gen_tcp:recv(Socket, 0, 6000),
    566. 

  566. 	{error, closed} = gen_tcp:recv(Socket, 0, 6000),
    567. 

  567. 	ok.
    568. 

  568. 

  569. ws_timeout_hibernate(Config) ->
    570. 

  570. 	{port, Port} = lists:keyfind(port, 1, Config),
    571. 

  571. 	{ok, Socket} = gen_tcp:connect("localhost", Port,
    572. 

  572. 		[binary, {active, false}, {packet, raw}]),
    573. 

  573. 	ok = gen_tcp:send(Socket, [
    574. 

  574. 		"GET /ws_timeout_hibernate HTTP/1.1\r\n"
    575. 

  575. 		"Host: localhost\r\n"
    576. 

  576. 		"Connection: Upgrade\r\n"
    577. 

  577. 		"Upgrade: websocket\r\n"
    578. 

  578. 		"Sec-WebSocket-Origin: http://localhost\r\n"
    579. 

  579. 		"Sec-WebSocket-Version: 8\r\n"
    580. 

  580. 		"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
    581. 

  581. 		"\r\n"]),
    582. 

  582. 	{ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
    583. 

  583. 	{ok, {http_response, {1, 1}, 101, "Switching Protocols"}, Rest}
    584. 

  584. 		= erlang:decode_packet(http, Handshake, []),
    585. 

  585. 	[Headers, <<>>] = do_decode_headers(
    586. 

  586. 		erlang:decode_packet(httph, Rest, []), []),
    587. 

  587. 	{'Connection', "Upgrade"} = lists:keyfind('Connection', 1, Headers),
    588. 

  588. 	{'Upgrade', "websocket"} = lists:keyfind('Upgrade', 1, Headers),
    589. 

  589. 	{"sec-websocket-accept", "s3pPLMBiTxaQ9kYGzzhZRbK+xOo="}
    590. 

  590. 		= lists:keyfind("sec-websocket-accept", 1, Headers),
    591. 

  591. 	{ok, << 1:1, 0:3, 8:4, 0:1, 2:7, 1000:16 >>} = gen_tcp:recv(Socket, 0, 6000),
    592. 

  592. 	{error, closed} = gen_tcp:recv(Socket, 0, 6000),
    593. 

  593. 	ok.
    594. 

  594. 

  595. ws_timeout_cancel(Config) ->
    596. 

  596. 	%% Erlang messages to a socket should not cancel the timeout	
    597. 

  597. 	{port, Port} = lists:keyfind(port, 1, Config),
    598. 

  598. 	{ok, Socket} = gen_tcp:connect("localhost", Port,
    599. 

  599. 		[binary, {active, false}, {packet, raw}]),
    600. 

  600. 	ok = gen_tcp:send(Socket, [
    601. 

  601. 		"GET /ws_timeout_cancel HTTP/1.1\r\n"
    602. 

  602. 		"Host: localhost\r\n"
    603. 

  603. 		"Connection: Upgrade\r\n"
    604. 

  604. 		"Upgrade: websocket\r\n"
    605. 

  605. 		"Sec-WebSocket-Origin: http://localhost\r\n"
    606. 

  606. 		"Sec-WebSocket-Version: 8\r\n"
    607. 

  607. 		"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
    608. 

  608. 		"\r\n"]),
    609. 

  609. 	{ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
    610. 

  610. 	{ok, {http_response, {1, 1}, 101, "Switching Protocols"}, Rest}
    611. 

  611. 		= erlang:decode_packet(http, Handshake, []),
    612. 

  612. 	[Headers, <<>>] = do_decode_headers(
    613. 

  613. 		erlang:decode_packet(httph, Rest, []), []),
    614. 

  614. 	{'Connection', "Upgrade"} = lists:keyfind('Connection', 1, Headers),
    615. 

  615. 	{'Upgrade', "websocket"} = lists:keyfind('Upgrade', 1, Headers),
    616. 

  616. 	{"sec-websocket-accept", "s3pPLMBiTxaQ9kYGzzhZRbK+xOo="}
    617. 

  617. 		= lists:keyfind("sec-websocket-accept", 1, Headers),
    618. 

  618. 	{ok, << 1:1, 0:3, 8:4, 0:1, 2:7, 1000:16 >>} = gen_tcp:recv(Socket, 0, 6000),
    619. 

  619. 	{error, closed} = gen_tcp:recv(Socket, 0, 6000),
    620. 

  620. 	ok.
    621. 

  621. 

  622. ws_timeout_reset(Config) ->
    623. 

  623. 	%% Erlang messages across a socket should reset the timeout	
    624. 

  624. 	{port, Port} = lists:keyfind(port, 1, Config),
    625. 

  625. 	{ok, Socket} = gen_tcp:connect("localhost", Port,
    626. 

  626. 		[binary, {active, false}, {packet, raw}]),
    627. 

  627. 	ok = gen_tcp:send(Socket, [
    628. 

  628. 		"GET /ws_timeout_cancel HTTP/1.1\r\n"
    629. 

  629. 		"Host: localhost\r\n"
    630. 

  630. 		"Connection: Upgrade\r\n"
    631. 

  631. 		"Upgrade: websocket\r\n"
    632. 

  632. 		"Sec-WebSocket-Origin: http://localhost\r\n"
    633. 

  633. 		"Sec-Websocket-Version: 13\r\n"
    634. 

  634. 		"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
    635. 

  635. 		"\r\n"]),
    636. 

  636. 	{ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
    637. 

  637. 	{ok, {http_response, {1, 1}, 101, "Switching Protocols"}, Rest}
    638. 

  638. 		= erlang:decode_packet(http, Handshake, []),
    639. 

  639. 	[Headers, <<>>] = do_decode_headers(
    640. 

  640. 		erlang:decode_packet(httph, Rest, []), []),
    641. 

  641. 	{'Connection', "Upgrade"} = lists:keyfind('Connection', 1, Headers),
    642. 

  642. 	{'Upgrade', "websocket"} = lists:keyfind('Upgrade', 1, Headers),
    643. 

  643. 	{"sec-websocket-accept", "s3pPLMBiTxaQ9kYGzzhZRbK+xOo="}
    644. 

  644. 		= lists:keyfind("sec-websocket-accept", 1, Headers),
    645. 

  645. 	[begin
    646. 

  646. 		ok = gen_tcp:send(Socket, << 16#81, 16#85, 16#37, 16#fa, 16#21, 16#3d,
    647. 

  647. 			16#7f, 16#9f, 16#4d, 16#51, 16#58 >>),
    648. 

  648. 		{ok, << 1:1, 0:3, 1:4, 0:1, 5:7, "Hello" >>}
    649. 

  649. 			= gen_tcp:recv(Socket, 0, 6000),
    650. 

  650. 		ok = timer:sleep(500)
    651. 

  651. 	end || _ <- [1, 2, 3, 4]],
    652. 

  652. 	{ok, << 1:1, 0:3, 8:4, 0:1, 2:7, 1000:16 >>} = gen_tcp:recv(Socket, 0, 6000),
    653. 

  653. 	{error, closed} = gen_tcp:recv(Socket, 0, 6000),
    654. 

  654. 	ok.
    655. 

  655. 

  656. ws_upgrade_with_opts(Config) ->
    657. 

  657. 	{port, Port} = lists:keyfind(port, 1, Config),
    658. 

  658. 	{ok, Socket} = gen_tcp:connect("localhost", Port,
    659. 

  659. 		[binary, {active, false}, {packet, raw}]),
    660. 

  660. 	ok = gen_tcp:send(Socket, [
    661. 

  661. 		"GET /ws_upgrade_with_opts HTTP/1.1\r\n"
    662. 

  662. 		"Host: localhost\r\n"
    663. 

  663. 		"Connection: Upgrade\r\n"
    664. 

  664. 		"Upgrade: websocket\r\n"
    665. 

  665. 		"Sec-WebSocket-Origin: http://localhost\r\n"
    666. 

  666. 		"Sec-WebSocket-Version: 8\r\n"
    667. 

  667. 		"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
    668. 

  668. 		"\r\n"]),
    669. 

  669. 	{ok, Handshake} = gen_tcp:recv(Socket, 0, 6000),
    670. 

  670. 	{ok, {http_response, {1, 1}, 101, "Switching Protocols"}, Rest}
    671. 

  671. 		= erlang:decode_packet(http, Handshake, []),
    672. 

  672. 	[Headers, <<>>] = do_decode_headers(
    673. 

  673. 		erlang:decode_packet(httph, Rest, []), []),
    674. 

  674. 	{'Connection', "Upgrade"} = lists:keyfind('Connection', 1, Headers),
    675. 

  675. 	{'Upgrade', "websocket"} = lists:keyfind('Upgrade', 1, Headers),
    676. 

  676. 	{"sec-websocket-accept", "s3pPLMBiTxaQ9kYGzzhZRbK+xOo="}
    677. 

  677. 		= lists:keyfind("sec-websocket-accept", 1, Headers),
    678. 

  678. 	{ok, Response} = gen_tcp:recv(Socket, 9, 6000),
    679. 

  679. 	<< 1:1, 0:3, 1:4, 0:1, 7:7, "success" >> = Response,
    680. 

  680. 	ok = gen_tcp:send(Socket, << 1:1, 0:3, 8:4, 1:1, 0:7, 0:32 >>), %% close
    681. 

  681. 	{ok, << 1:1, 0:3, 8:4, 0:8 >>} = gen_tcp:recv(Socket, 0, 6000),
    682. 

  682. 	{error, closed} = gen_tcp:recv(Socket, 0, 6000),
    683. 

  683. 	ok.
    684. 

  684. 

  685. %% Internal.
    686. 

  686. 

  687. do_decode_headers({ok, http_eoh, Rest}, Acc) ->
    688. 

  688. 	[Acc, Rest];
    689. 

  689. do_decode_headers({ok, {http_header, _I, Key, _R, Value}, Rest}, Acc) ->
    690. 

  690. 	F = fun(S) when is_atom(S) -> S; (S) -> string:to_lower(S) end,
    691. 

  691. 	do_decode_headers(erlang:decode_packet(httph, Rest, []),
    692. 

  692. 		[{F(Key), Value}|Acc]).
    693. 

  693. 

  694. do_mask(<<>>, _, Acc) ->
    695. 

  695. 	Acc;
    696. 

  696. do_mask(<< O:32, Rest/bits >>, MaskKey, Acc) ->
    697. 

  697. 	T = O bxor MaskKey,
    698. 

  698. 	do_mask(Rest, MaskKey, << Acc/binary, T:32 >>);
    699. 

  699. do_mask(<< O:24 >>, MaskKey, Acc) ->
    700. 

  700. 	<< MaskKey2:24, _:8 >> = << MaskKey:32 >>,
    701. 

  701. 	T = O bxor MaskKey2,
    702. 

  702. 	<< Acc/binary, T:24 >>;
    703. 

  703. do_mask(<< O:16 >>, MaskKey, Acc) ->
    704. 

  704. 	<< MaskKey2:16, _:16 >> = << MaskKey:32 >>,
    705. 

  705. 	T = O bxor MaskKey2,
    706. 

  706. 	<< Acc/binary, T:16 >>;
    707. 

  707. do_mask(<< O:8 >>, MaskKey, Acc) ->
    708. 

  708. 	<< MaskKey2:8, _:24 >> = << MaskKey:32 >>,
    709. 

  709. 	T = O bxor MaskKey2,
    710. 

  710. 	<< Acc/binary, T:8 >>.
    711.