acceptor_SUITE.erl 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526
  1. %% Copyright (c) 2011-2015, Loïc Hoguin <essen@ninenines.eu>
  2. %%
  3. %% Permission to use, copy, modify, and/or distribute this software for any
  4. %% purpose with or without fee is hereby granted, provided that the above
  5. %% copyright notice and this permission notice appear in all copies.
  6. %%
  7. %% THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  8. %% WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  9. %% MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  10. %% ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  11. %% WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  12. %% ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  13. %% OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  14. -module(acceptor_SUITE).
  15. -compile(export_all).
  16. -import(ct_helper, [doc/1]).
  17. -import(ct_helper, [name/0]).
  18. %% ct.
  19. all() ->
  20. [{group, tcp}, {group, ssl}, {group, misc}, {group, supervisor}].
  21. groups() ->
  22. [{tcp, [
  23. tcp_accept_socket,
  24. tcp_active_echo,
  25. tcp_echo,
  26. tcp_inherit_options,
  27. tcp_max_connections,
  28. tcp_max_connections_and_beyond,
  29. tcp_max_connections_infinity,
  30. tcp_set_max_connections,
  31. tcp_set_max_connections_clean,
  32. tcp_upgrade
  33. ]}, {ssl, [
  34. ssl_accept_error,
  35. ssl_accept_socket,
  36. ssl_active_echo,
  37. ssl_echo
  38. ]}, {misc, [
  39. misc_bad_transport
  40. ]}, {supervisor, [
  41. connection_type_supervisor,
  42. connection_type_supervisor_separate_from_connection,
  43. supervisor_clean_child_restart,
  44. supervisor_clean_conns_sup_restart,
  45. supervisor_clean_restart,
  46. supervisor_conns_alive,
  47. supervisor_protocol_start_link_crash,
  48. supervisor_server_recover_state
  49. ]}].
  50. %% misc.
  51. misc_bad_transport(_) ->
  52. doc("Reject invalid transport modules."),
  53. {error, badarg} = ranch:start_listener(misc_bad_transport, 1,
  54. bad_transport, [], echo_protocol, []),
  55. ok.
  56. %% ssl.
  57. ssl_accept_error(_) ->
  58. doc("Acceptor must not crash if client disconnects in the middle of SSL handshake."),
  59. Name = name(),
  60. Opts = ct_helper:get_certs_from_ets(),
  61. {ok, ListenerSup} = ranch:start_listener(Name, 1, ranch_ssl, Opts, echo_protocol, []),
  62. Port = ranch:get_port(Name),
  63. ListenerSupChildren = supervisor:which_children(ListenerSup),
  64. {_, AcceptorsSup, _, _} = lists:keyfind(ranch_acceptors_sup, 1, ListenerSupChildren),
  65. [{{acceptor, _, _}, AcceptorPid, _, _}] = supervisor:which_children(AcceptorsSup),
  66. true = is_process_alive(AcceptorPid),
  67. {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  68. ok = gen_tcp:close(Socket),
  69. receive after 500 -> ok end,
  70. true = is_process_alive(AcceptorPid),
  71. ok = ranch:stop_listener(Name).
  72. ssl_accept_socket(_) ->
  73. doc("Ensure that listener can use an externally opened SSL listen socket."),
  74. Name = name(),
  75. Opts = ct_helper:get_certs_from_ets(),
  76. {ok, S} = ssl:listen(0, [binary, {active, false}, {packet, raw}, {reuseaddr, true}|Opts]),
  77. {ok, _} = ranch:start_listener(Name, 1, ranch_ssl, [{socket, S}], echo_protocol, []),
  78. Port = ranch:get_port(Name),
  79. {ok, Socket} = ssl:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  80. ok = ssl:send(Socket, <<"TCP Ranch is working!">>),
  81. {ok, <<"TCP Ranch is working!">>} = ssl:recv(Socket, 21, 1000),
  82. ok = ranch:stop_listener(Name),
  83. {error, closed} = ssl:recv(Socket, 0, 1000),
  84. %% Make sure the listener stopped.
  85. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  86. ok.
  87. ssl_active_echo(_) ->
  88. doc("Ensure that active mode works with SSL transport."),
  89. Name = name(),
  90. Opts = ct_helper:get_certs_from_ets(),
  91. {ok, _} = ranch:start_listener(Name, 1, ranch_ssl, Opts, active_echo_protocol, []),
  92. Port = ranch:get_port(Name),
  93. {ok, Socket} = ssl:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  94. ok = ssl:send(Socket, <<"SSL Ranch is working!">>),
  95. {ok, <<"SSL Ranch is working!">>} = ssl:recv(Socket, 21, 1000),
  96. ok = ranch:stop_listener(Name),
  97. {error, closed} = ssl:recv(Socket, 0, 1000),
  98. %% Make sure the listener stopped.
  99. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  100. ok.
  101. ssl_echo(_) ->
  102. doc("Ensure that passive mode works with SSL transport."),
  103. Name = name(),
  104. Opts = ct_helper:get_certs_from_ets(),
  105. {ok, _} = ranch:start_listener(Name, 1, ranch_ssl, Opts, echo_protocol, []),
  106. Port = ranch:get_port(Name),
  107. {ok, Socket} = ssl:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  108. ok = ssl:send(Socket, <<"SSL Ranch is working!">>),
  109. {ok, <<"SSL Ranch is working!">>} = ssl:recv(Socket, 21, 1000),
  110. ok = ranch:stop_listener(Name),
  111. {error, closed} = ssl:recv(Socket, 0, 1000),
  112. %% Make sure the listener stopped.
  113. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  114. ok.
  115. %% tcp.
  116. tcp_accept_socket(_) ->
  117. doc("Ensure that listener can use an externally opened TCP listen socket."),
  118. Name = name(),
  119. {ok, S} = gen_tcp:listen(0, [binary, {active, false}, {packet, raw}, {reuseaddr, true}]),
  120. {ok, _} = ranch:start_listener(Name, 1, ranch_tcp, [{socket, S}], echo_protocol, []),
  121. Port = ranch:get_port(Name),
  122. {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  123. ok = gen_tcp:send(Socket, <<"TCP Ranch is working!">>),
  124. {ok, <<"TCP Ranch is working!">>} = gen_tcp:recv(Socket, 21, 1000),
  125. ok = ranch:stop_listener(Name),
  126. {error, closed} = gen_tcp:recv(Socket, 0, 1000),
  127. %% Make sure the listener stopped.
  128. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  129. ok.
  130. tcp_active_echo(_) ->
  131. doc("Ensure that active mode works with TCP transport."),
  132. Name = name(),
  133. {ok, _} = ranch:start_listener(Name, 1, ranch_tcp, [], active_echo_protocol, []),
  134. Port = ranch:get_port(Name),
  135. {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  136. ok = gen_tcp:send(Socket, <<"TCP Ranch is working!">>),
  137. {ok, <<"TCP Ranch is working!">>} = gen_tcp:recv(Socket, 21, 1000),
  138. ok = ranch:stop_listener(Name),
  139. {error, closed} = gen_tcp:recv(Socket, 0, 1000),
  140. %% Make sure the listener stopped.
  141. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  142. ok.
  143. tcp_echo(_) ->
  144. doc("Ensure that passive mode works with TCP transport."),
  145. Name = name(),
  146. {ok, _} = ranch:start_listener(Name, 1, ranch_tcp, [], echo_protocol, []),
  147. Port = ranch:get_port(Name),
  148. {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  149. ok = gen_tcp:send(Socket, <<"TCP Ranch is working!">>),
  150. {ok, <<"TCP Ranch is working!">>} = gen_tcp:recv(Socket, 21, 1000),
  151. ok = ranch:stop_listener(Name),
  152. {error, closed} = gen_tcp:recv(Socket, 0, 1000),
  153. %% Make sure the listener stopped.
  154. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  155. ok.
  156. tcp_inherit_options(_) ->
  157. doc("Ensure TCP options are inherited in the protocol."),
  158. Name = name(),
  159. Opts = [{nodelay, false}, {send_timeout_close, false}],
  160. {ok, _} = ranch:start_listener(Name, 4, ranch_tcp, Opts, check_tcp_options, [{pid, self()} | Opts]),
  161. Port = ranch:get_port(Name),
  162. {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, true}, {packet, raw}]),
  163. receive checked -> ok after 1000 -> error(timeout) end,
  164. ok = gen_tcp:close(Socket),
  165. ok = ranch:stop_listener(Name).
  166. tcp_max_connections(_) ->
  167. doc("Ensure the max_connections option actually limits connections."),
  168. Name = name(),
  169. {ok, _} = ranch:start_listener(Name, 1,
  170. ranch_tcp, [{max_connections, 10}],
  171. notify_and_wait_protocol, [{msg, connected}, {pid, self()}]),
  172. Port = ranch:get_port(Name),
  173. ok = connect_loop(Port, 11, 150),
  174. 10 = ranch_server:count_connections(Name),
  175. 10 = receive_loop(connected, 400),
  176. 1 = receive_loop(connected, 1000),
  177. ok = ranch:stop_listener(Name).
  178. tcp_max_connections_and_beyond(_) ->
  179. doc("Ensure the max_connections option works when connections are removed from the count."),
  180. Name = name(),
  181. {ok, _} = ranch:start_listener(Name, 1,
  182. ranch_tcp, [{max_connections, 10}],
  183. remove_conn_and_wait_protocol, [{remove, true}]),
  184. Port = ranch:get_port(Name),
  185. ok = connect_loop(Port, 10, 0),
  186. receive after 250 -> ok end,
  187. 0 = ranch_server:count_connections(Name),
  188. 10 = length(supervisor:which_children(ranch_server:get_connections_sup(Name))),
  189. Counts = supervisor:count_children(ranch_server:get_connections_sup(Name)),
  190. {_, 1} = lists:keyfind(specs, 1, Counts),
  191. {_, 0} = lists:keyfind(supervisors, 1, Counts),
  192. {_, 10} = lists:keyfind(active, 1, Counts),
  193. {_, 10} = lists:keyfind(workers, 1, Counts),
  194. ranch:set_protocol_options(Name, [{remove, false}]),
  195. receive after 250 -> ok end,
  196. ok = connect_loop(Port, 10, 0),
  197. receive after 250 -> ok end,
  198. 10 = ranch_server:count_connections(Name),
  199. 20 = length(supervisor:which_children(ranch_server:get_connections_sup(Name))),
  200. Counts2 = supervisor:count_children(ranch_server:get_connections_sup(Name)),
  201. {_, 20} = lists:keyfind(active, 1, Counts2),
  202. {_, 20} = lists:keyfind(workers, 1, Counts2),
  203. ok = ranch:stop_listener(Name).
  204. tcp_max_connections_infinity(_) ->
  205. doc("Set the max_connections option from 10 to infinity and back to 10."),
  206. Name = name(),
  207. {ok, _} = ranch:start_listener(Name, 1,
  208. ranch_tcp, [{max_connections, 10}],
  209. notify_and_wait_protocol, [{msg, connected}, {pid, self()}]),
  210. Port = ranch:get_port(Name),
  211. ok = connect_loop(Port, 20, 0),
  212. 10 = ranch_server:count_connections(Name),
  213. 10 = receive_loop(connected, 1000),
  214. 10 = ranch_server:count_connections(Name),
  215. 10 = ranch:get_max_connections(Name),
  216. ranch:set_max_connections(Name, infinity),
  217. receive after 250 -> ok end,
  218. 20 = ranch_server:count_connections(Name),
  219. infinity = ranch:get_max_connections(Name),
  220. ranch:set_max_connections(Name, 10),
  221. 20 = ranch_server:count_connections(Name),
  222. 10 = receive_loop(connected, 1000),
  223. ok = ranch:stop_listener(Name).
  224. tcp_set_max_connections(_) ->
  225. doc("Ensure that changing the max_connections option to a larger value allows for more connections."),
  226. Name = name(),
  227. {ok, _} = ranch:start_listener(Name, 1,
  228. ranch_tcp, [{max_connections, 10}],
  229. notify_and_wait_protocol, [{msg, connected}, {pid, self()}]),
  230. Port = ranch:get_port(Name),
  231. ok = connect_loop(Port, 20, 0),
  232. 10 = ranch_server:count_connections(Name),
  233. 10 = receive_loop(connected, 1000),
  234. 10 = ranch:get_max_connections(Name),
  235. ranch:set_max_connections(Name, 20),
  236. 10 = receive_loop(connected, 1000),
  237. 20 = ranch:get_max_connections(Name),
  238. ok = ranch:stop_listener(Name).
  239. tcp_set_max_connections_clean(_) ->
  240. doc("Ensure that setting max_connections does not crash any process."),
  241. Name = name(),
  242. {ok, ListSupPid} = ranch:start_listener(Name, 4, ranch_tcp,
  243. [{max_connections, 4}],
  244. notify_and_wait_protocol, [{msg, connected}, {pid, self()}]),
  245. Children = supervisor:which_children(ListSupPid),
  246. {_, AccSupPid, _, _} = lists:keyfind(ranch_acceptors_sup, 1, Children),
  247. 1 = erlang:trace(ListSupPid, true, [procs]),
  248. 1 = erlang:trace(AccSupPid, true, [procs]),
  249. Port = ranch:get_port(Name),
  250. N = 20,
  251. ok = connect_loop(Port, N*5, 0),
  252. %% Randomly set max_connections.
  253. [spawn(ranch, set_max_connections, [Name, Max]) ||
  254. Max <- lists:flatten(lists:duplicate(N, [6, 4, 8, infinity]))],
  255. receive
  256. {trace, _, spawn, _, _} ->
  257. error(dirty_set_max_connections)
  258. after
  259. 2000 -> ok
  260. end,
  261. _ = erlang:trace(all, false, [all]),
  262. ok = clean_traces(),
  263. ok = ranch:stop_listener(Name).
  264. tcp_upgrade(_) ->
  265. doc("Ensure that protocol options can be updated."),
  266. Name = name(),
  267. {ok, _} = ranch:start_listener(Name, 1,
  268. ranch_tcp, [],
  269. notify_and_wait_protocol, [{msg, connected}, {pid, self()}]),
  270. Port = ranch:get_port(Name),
  271. ok = connect_loop(Port, 1, 0),
  272. receive connected -> ok after 1000 -> error(timeout) end,
  273. ranch:set_protocol_options(Name, [{msg, upgraded}, {pid, self()}]),
  274. ok = connect_loop(Port, 1, 0),
  275. receive upgraded -> ok after 1000 -> error(timeout) end,
  276. ok = ranch:stop_listener(Name).
  277. %% Supervisor tests
  278. connection_type_supervisor(_) ->
  279. doc("The supervisor connection type must be reflected in the specifications."),
  280. Name = name(),
  281. {ok, _} = ranch:start_listener(Name, 1,
  282. ranch_tcp, [{connection_type, supervisor}],
  283. echo_protocol, []),
  284. Port = ranch:get_port(Name),
  285. {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  286. ok = gen_tcp:send(Socket, <<"TCP Ranch is working!">>),
  287. {ok, <<"TCP Ranch is working!">>} = gen_tcp:recv(Socket, 21, 1000),
  288. ConnsSup = ranch_server:get_connections_sup(Name),
  289. [{echo_protocol, _, supervisor, [echo_protocol]}] = supervisor:which_children(ConnsSup),
  290. ok = ranch:stop_listener(Name),
  291. {error, closed} = gen_tcp:recv(Socket, 0, 1000),
  292. %% Make sure the listener stopped.
  293. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  294. ok.
  295. connection_type_supervisor_separate_from_connection(_) ->
  296. doc("The supervisor connection type allows separate supervised and connection processes."),
  297. Name = name(),
  298. {ok, _} = ranch:start_listener(Name, 1,
  299. ranch_tcp, [{connection_type, supervisor}],
  300. supervisor_separate, []),
  301. Port = ranch:get_port(Name),
  302. {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  303. ok = gen_tcp:send(Socket, <<"TCP Ranch is working!">>),
  304. {ok, <<"TCP Ranch is working!">>} = gen_tcp:recv(Socket, 21, 1000),
  305. ConnsSup = ranch_server:get_connections_sup(Name),
  306. [{supervisor_separate, _, supervisor, [supervisor_separate]}] = supervisor:which_children(ConnsSup),
  307. ok = ranch:stop_listener(Name),
  308. {error, closed} = gen_tcp:recv(Socket, 0, 1000),
  309. %% Make sure the listener stopped.
  310. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  311. ok.
  312. supervisor_clean_child_restart(_) ->
  313. doc("Verify that only the relevant parts of the supervision tree restarted "
  314. "when the listening socket is closed."),
  315. Name = name(),
  316. %% Trace socket allocations.
  317. _ = erlang:trace(new, true, [call]),
  318. 1 = erlang:trace_pattern({ranch_tcp, listen, 1},
  319. [{'_', [], [{return_trace}]}], [global]),
  320. {ok, Pid} = ranch:start_listener(Name,
  321. 1, ranch_tcp, [], echo_protocol, []),
  322. %% Trace supervisor spawns.
  323. 1 = erlang:trace(Pid, true, [procs, set_on_spawn]),
  324. ConnsSup = ranch_server:get_connections_sup(Name),
  325. %% Manually shut the listening socket down.
  326. LSocket = receive
  327. {trace, _, return_from, {ranch_tcp, listen, 1}, {ok, Socket}} ->
  328. Socket
  329. after 0 ->
  330. error(lsocket_unknown)
  331. end,
  332. ok = gen_tcp:close(LSocket),
  333. receive after 1000 -> ok end,
  334. %% Verify that supervisor and its first two children are alive.
  335. true = is_process_alive(Pid),
  336. true = is_process_alive(ConnsSup),
  337. %% Check that acceptors_sup is restarted properly.
  338. AccSupPid = receive {trace, Pid, spawn, Pid1, _} -> Pid1 end,
  339. receive {trace, AccSupPid, spawn, _, _} -> ok end,
  340. %% No more traces then.
  341. receive
  342. {trace, _, spawn, _, _} -> error(invalid_restart)
  343. after 1000 -> ok end,
  344. %% Verify that children still registered right.
  345. ConnsSup = ranch_server:get_connections_sup(Name),
  346. _ = erlang:trace_pattern({ranch_tcp, listen, 1}, false, []),
  347. _ = erlang:trace(all, false, [all]),
  348. ok = clean_traces(),
  349. ok = ranch:stop_listener(Name).
  350. supervisor_clean_conns_sup_restart(_) ->
  351. doc("Verify that a conns_sup can not register with the same name as an already "
  352. "registered ranch_conns_sup that is still alive. Make sure this does not crash "
  353. "the ranch_server process."),
  354. Name = name(),
  355. {ok, _} = ranch:start_listener(Name,
  356. 1, ranch_tcp, [], echo_protocol, []),
  357. Server = erlang:whereis(ranch_server),
  358. ServerMonRef = erlang:monitor(process, Server),
  359. %% Exit because Name already registered and is alive.
  360. {'EXIT', _} = (catch ranch_server:set_connections_sup(Name, self())),
  361. receive
  362. {'DOWN', ServerMonRef, process, Server, _} ->
  363. error(ranch_server_down)
  364. after
  365. 1000 ->
  366. ok
  367. end,
  368. ok = ranch:stop_listener(Name).
  369. supervisor_clean_restart(_) ->
  370. doc("Verify that killing ranch_conns_sup does not crash everything "
  371. "and that it restarts properly."),
  372. Name = name(),
  373. NbAcc = 4,
  374. {ok, Pid} = ranch:start_listener(Name, NbAcc, ranch_tcp, [], echo_protocol, []),
  375. %% Trace supervisor spawns.
  376. 1 = erlang:trace(Pid, true, [procs, set_on_spawn]),
  377. ConnsSup0 = ranch_server:get_connections_sup(Name),
  378. erlang:exit(ConnsSup0, kill),
  379. receive after 1000 -> ok end,
  380. %% Verify that supervisor is alive
  381. true = is_process_alive(Pid),
  382. %% ...but children are dead.
  383. false = is_process_alive(ConnsSup0),
  384. %% Receive traces from newly started children
  385. ConnsSup = receive {trace, Pid, spawn, Pid2, _} -> Pid2 end,
  386. AccSupPid = receive {trace, Pid, spawn, Pid3, _} -> Pid3 end,
  387. %% ...and its acceptors.
  388. [receive {trace, AccSupPid, spawn, _Pid, _} -> ok end ||
  389. _ <- lists:seq(1, NbAcc)],
  390. %% No more traces then.
  391. receive
  392. {trace, EPid, spawn, _, _} when EPid == Pid; EPid == AccSupPid ->
  393. error(invalid_restart)
  394. after 1000 -> ok end,
  395. %% Verify that new children registered themselves properly.
  396. ConnsSup = ranch_server:get_connections_sup(Name),
  397. _ = erlang:trace(all, false, [all]),
  398. ok = clean_traces(),
  399. ok = ranch:stop_listener(Name).
  400. supervisor_conns_alive(_) ->
  401. doc("Ensure that active connections stay open when the listening socket gets closed."),
  402. Name = name(),
  403. _ = erlang:trace(new, true, [call]),
  404. 1 = erlang:trace_pattern({ranch_tcp, listen, 1},
  405. [{'_', [], [{return_trace}]}], [global]),
  406. {ok, _} = ranch:start_listener(Name, 1,
  407. ranch_tcp, [],
  408. remove_conn_and_wait_protocol, [{remove, false}]),
  409. %% Get the listener socket
  410. LSocket = receive
  411. {trace, _, return_from, {ranch_tcp, listen, 1}, {ok, S}} ->
  412. S
  413. after 500 ->
  414. error(lsocket_unknown)
  415. end,
  416. TcpPort = ranch:get_port(Name),
  417. {ok, Socket} = gen_tcp:connect("localhost", TcpPort,
  418. [binary, {active, true}, {packet, raw}]),
  419. receive after 500 -> ok end,
  420. %% Shut the socket down
  421. ok = gen_tcp:close(LSocket),
  422. %% Assert that client is still viable.
  423. receive {tcp_closed, _} -> error(closed) after 1500 -> ok end,
  424. ok = gen_tcp:send(Socket, <<"poke">>),
  425. receive {tcp_closed, _} -> ok end,
  426. _ = erlang:trace(all, false, [all]),
  427. ok = clean_traces(),
  428. ok = ranch:stop_listener(Name).
  429. supervisor_protocol_start_link_crash(_) ->
  430. doc("Ensure a protocol start crash does not kill all connections."),
  431. Name = name(),
  432. {ok, _} = ranch:start_listener(Name, 1, ranch_tcp, [], crash_protocol, []),
  433. ConnsSup = ranch_server:get_connections_sup(Name),
  434. Port = ranch:get_port(Name),
  435. {ok, _} = gen_tcp:connect("localhost", Port, [binary, {active, true}, {packet, raw}]),
  436. receive after 500 -> ok end,
  437. ConnsSup = ranch_server:get_connections_sup(Name),
  438. ok = ranch:stop_listener(Name).
  439. supervisor_server_recover_state(_) ->
  440. doc("Ensure that when ranch_server crashes and restarts, it recovers "
  441. "its state and continues monitoring the same processes."),
  442. Name = name(),
  443. {ok, _} = ranch:start_listener(Name, 1, ranch_tcp, [], echo_protocol, []),
  444. _ = erlang:trace(new, true, [call]),
  445. 1 = erlang:trace_pattern({ranch_server, init, 1},
  446. [{'_', [], [{return_trace}]}], [global]),
  447. ConnsSup = ranch_server:get_connections_sup(Name),
  448. ServerPid = erlang:whereis(ranch_server),
  449. {monitors, Monitors} = erlang:process_info(ServerPid, monitors),
  450. erlang:exit(ServerPid, kill),
  451. receive
  452. {trace, ServerPid2, return_from, {ranch_server, init, 1}, _Result} ->
  453. {monitors, Monitors2} = erlang:process_info(ServerPid2, monitors),
  454. %% Check that ranch_server is monitoring the same processes.
  455. true = (lists:usort(Monitors) == lists:usort(Monitors2))
  456. after
  457. 1000 ->
  458. error(timeout)
  459. end,
  460. ConnsSup = ranch_server:get_connections_sup(Name),
  461. ok = ranch:stop_listener(Name),
  462. %% Check ranch_server has removed the ranch_conns_sup.
  463. {'EXIT', {badarg, _}} = (catch ranch_server:get_connections_sup(Name)),
  464. _ = erlang:trace(all, false, [all]),
  465. ok = clean_traces().
  466. %% Utility functions.
  467. connect_loop(_, 0, _) ->
  468. ok;
  469. connect_loop(Port, N, Sleep) ->
  470. {ok, _} = gen_tcp:connect("localhost", Port,
  471. [binary, {active, false}, {packet, raw}]),
  472. receive after Sleep -> ok end,
  473. connect_loop(Port, N - 1, Sleep).
  474. receive_loop(Message, Timeout) ->
  475. receive_loop(Message, Timeout, 0).
  476. receive_loop(Message, Timeout, N) ->
  477. receive Message ->
  478. receive_loop(Message, Timeout, N + 1)
  479. after Timeout ->
  480. N
  481. end.
  482. clean_traces() ->
  483. receive
  484. {trace, _, _, _} ->
  485. clean_traces();
  486. {trace, _, _, _, _} ->
  487. clean_traces()
  488. after 0 ->
  489. ok
  490. end.