acceptor_SUITE.erl 20 KB

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