acceptor_SUITE.erl 64 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771
  1. %% Copyright (c) 2011-2020, Loïc Hoguin <essen@ninenines.eu>
  2. %% Copyright (c) 2020, Jan Uhlig <j.uhlig@mailingwork.de>
  3. %%
  4. %% Permission to use, copy, modify, and/or distribute this software for any
  5. %% purpose with or without fee is hereby granted, provided that the above
  6. %% copyright notice and this permission notice appear in all copies.
  7. %%
  8. %% THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9. %% WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10. %% MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11. %% ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12. %% WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13. %% ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14. %% OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15. -module(acceptor_SUITE).
  16. -compile(export_all).
  17. -compile(nowarn_export_all).
  18. -dialyzer({nowarn_function, misc_wait_for_connections/1}).
  19. %% @todo Remove when specs in ssl are updated to accept local addresses.
  20. -dialyzer({nowarn_function, do_ssl_local_echo/0}).
  21. -import(ct_helper, [doc/1]).
  22. -import(ct_helper, [name/0]).
  23. %% ct.
  24. all() ->
  25. [{group, tcp}, {group, ssl}, {group, misc}, {group, supervisor}].
  26. groups() ->
  27. [{tcp, [
  28. tcp_active_echo,
  29. tcp_active_n_echo,
  30. tcp_echo,
  31. tcp_local_echo,
  32. tcp_graceful,
  33. tcp_inherit_options,
  34. tcp_max_connections,
  35. tcp_max_connections_and_beyond,
  36. tcp_max_connections_infinity,
  37. tcp_remove_connections,
  38. tcp_remove_connections_acceptor_wakeup,
  39. tcp_set_max_connections,
  40. tcp_set_max_connections_clean,
  41. tcp_getopts_capability,
  42. tcp_getstat_capability,
  43. tcp_upgrade,
  44. tcp_10_acceptors_10_listen_sockets,
  45. tcp_many_listen_sockets_no_reuseport,
  46. tcp_error_eaddrinuse,
  47. tcp_error_eacces
  48. ]}, {ssl, [
  49. ssl_accept_error,
  50. ssl_active_echo,
  51. ssl_active_n_echo,
  52. ssl_echo,
  53. ssl_local_echo,
  54. ssl_graceful,
  55. ssl_handshake,
  56. ssl_sni_echo,
  57. ssl_sni_fail,
  58. ssl_tls_psk,
  59. ssl_tls_psk_fail,
  60. ssl_upgrade_from_tcp,
  61. ssl_getopts_capability,
  62. ssl_getstat_capability,
  63. ssl_10_acceptors_10_listen_sockets,
  64. ssl_many_listen_sockets_no_reuseport,
  65. ssl_error_eaddrinuse,
  66. ssl_error_no_cert,
  67. ssl_error_eacces,
  68. ssl_unsupported_tlsv13_options
  69. ]}, {misc, [
  70. misc_bad_transport,
  71. misc_bad_transport_options,
  72. misc_repeated_remove,
  73. misc_info,
  74. misc_info_embedded,
  75. misc_metrics,
  76. misc_opts_logger,
  77. misc_post_listen_callback,
  78. misc_post_listen_callback_error,
  79. misc_set_transport_options,
  80. misc_wait_for_connections,
  81. misc_multiple_ip_local_socket_opts
  82. ]}, {supervisor, [
  83. connection_type_supervisor,
  84. connection_type_supervisor_separate_from_connection,
  85. supervisor_10_acceptors_1_conns_sup,
  86. supervisor_9_acceptors_4_conns_sups,
  87. supervisor_10_acceptors_10_conns_sups,
  88. supervisor_1_acceptor_10_conns_sups,
  89. supervisor_changed_options_restart,
  90. supervisor_clean_child_restart,
  91. supervisor_clean_restart,
  92. supervisor_conns_alive,
  93. supervisor_embedded_ranch_server_crash,
  94. supervisor_protocol_start_link_crash,
  95. supervisor_server_recover_state,
  96. supervisor_unexpected_message
  97. ]}].
  98. %% misc.
  99. misc_bad_transport(_) ->
  100. doc("Reject invalid transport modules."),
  101. {error, {bad_transport, invalid_transport}} = ranch:start_listener(misc_bad_transport,
  102. invalid_transport, #{},
  103. echo_protocol, []),
  104. ok.
  105. misc_bad_transport_options(_) ->
  106. doc("Ignore invalid transport options."),
  107. {ok, _} = ranch:start_listener(misc_bad_transport_options,
  108. ranch_tcp, [binary, {packet, 4}, <<"garbage">>, raw, backlog],
  109. echo_protocol, []),
  110. ok.
  111. misc_info(_) ->
  112. doc("Information about listeners."),
  113. %% Open a listener with a few connections.
  114. {ok, Pid1} = ranch:start_listener({misc_info, tcp},
  115. ranch_tcp, #{num_acceptors => 1},
  116. remove_conn_and_wait_protocol, [{remove, true, 2500}]),
  117. Port1 = ranch:get_port({misc_info, tcp}),
  118. %% Open a few more listeners with different arguments.
  119. {ok, Pid2} = ranch:start_listener({misc_info, act},
  120. ranch_tcp, #{num_acceptors => 2},
  121. active_echo_protocol, {}),
  122. Port2 = ranch:get_port({misc_info, act}),
  123. ranch:set_max_connections({misc_info, act}, infinity),
  124. Opts = ct_helper:get_certs_from_ets(),
  125. {ok, Pid3} = ranch:start_listener({misc_info, ssl},
  126. ranch_ssl, #{num_acceptors => 3, socket_opts => Opts},
  127. echo_protocol, [{}]),
  128. Port3 = ranch:get_port({misc_info, ssl}),
  129. %% Open 5 connections, 3 removed from the count.
  130. {ok, _} = gen_tcp:connect("localhost", Port1, [binary, {active, false}, {packet, raw}]),
  131. {ok, _} = gen_tcp:connect("localhost", Port1, [binary, {active, false}, {packet, raw}]),
  132. {ok, _} = gen_tcp:connect("localhost", Port1, [binary, {active, false}, {packet, raw}]),
  133. receive after 250 -> ok end,
  134. ranch:set_protocol_options({misc_info, tcp}, [{remove, false, 2500}]),
  135. receive after 250 -> ok end,
  136. {ok, _} = gen_tcp:connect("localhost", Port1, [binary, {active, false}, {packet, raw}]),
  137. {ok, _} = gen_tcp:connect("localhost", Port1, [binary, {active, false}, {packet, raw}]),
  138. receive after 250 -> ok end,
  139. %% Confirm the info returned by Ranch is correct.
  140. #{
  141. {misc_info, act} := #{
  142. pid := Pid2,
  143. port := Port2,
  144. max_connections := infinity, %% Option was modified.
  145. active_connections := 0,
  146. all_connections := 0,
  147. transport := ranch_tcp,
  148. transport_options := #{num_acceptors := 2},
  149. protocol := active_echo_protocol,
  150. protocol_options := {}
  151. },
  152. {misc_info, ssl} := #{
  153. pid := Pid3,
  154. port := Port3,
  155. max_connections := 1024,
  156. active_connections := 0,
  157. all_connections := 0,
  158. transport := ranch_ssl,
  159. transport_options := #{num_acceptors := 3, socket_opts := Opts},
  160. protocol := echo_protocol,
  161. protocol_options := [{}]
  162. },
  163. {misc_info, tcp} := #{
  164. pid := Pid1,
  165. port := Port1,
  166. max_connections := 1024,
  167. active_connections := 2,
  168. all_connections := 5,
  169. transport := ranch_tcp,
  170. transport_options := #{num_acceptors := 1},
  171. protocol := remove_conn_and_wait_protocol,
  172. protocol_options := [{remove, false, 2500}] %% Option was modified.
  173. }
  174. } = ranch:info(),
  175. %% Get acceptors.
  176. [_] = ranch:procs({misc_info, tcp}, acceptors),
  177. [_, _] = ranch:procs({misc_info, act}, acceptors),
  178. [_, _, _] = ranch:procs({misc_info, ssl}, acceptors),
  179. %% Get connections.
  180. [_, _, _, _, _] = ranch:procs({misc_info, tcp}, connections),
  181. [] = ranch:procs({misc_info, act}, connections),
  182. [] = ranch:procs({misc_info, ssl}, connections),
  183. ok.
  184. misc_info_embedded(_) ->
  185. doc("Information about listeners in embedded mode."),
  186. {ok, SupPid} = embedded_sup:start_link(),
  187. %% Open a listener with a few connections.
  188. {ok, EmbeddedSupPid1} = embedded_sup:start_listener(SupPid, {misc_info_embedded, tcp},
  189. ranch_tcp, #{num_acceptors => 1},
  190. remove_conn_and_wait_protocol, [{remove, true, 2500}]),
  191. {_, Pid1, _, _} = lists:keyfind({ranch_listener_sup, {misc_info_embedded, tcp}}, 1,
  192. supervisor:which_children(EmbeddedSupPid1)),
  193. Port1 = ranch:get_port({misc_info_embedded, tcp}),
  194. %% Open a few more listeners with different arguments.
  195. {ok, EmbeddedSupPid2} = embedded_sup:start_listener(SupPid, {misc_info_embedded, act},
  196. ranch_tcp, #{num_acceptors => 2},
  197. active_echo_protocol, {}),
  198. {_, Pid2, _, _} = lists:keyfind({ranch_listener_sup, {misc_info_embedded, act}}, 1,
  199. supervisor:which_children(EmbeddedSupPid2)),
  200. Port2 = ranch:get_port({misc_info_embedded, act}),
  201. ranch:set_max_connections({misc_info_embedded, act}, infinity),
  202. Opts = ct_helper:get_certs_from_ets(),
  203. {ok, EmbeddedSupPid3} = embedded_sup:start_listener(SupPid, {misc_info_embedded, ssl},
  204. ranch_ssl, #{num_acceptors => 3, socket_opts => Opts},
  205. echo_protocol, [{}]),
  206. {_, Pid3, _, _} = lists:keyfind({ranch_listener_sup, {misc_info_embedded, ssl}}, 1,
  207. supervisor:which_children(EmbeddedSupPid3)),
  208. Port3 = ranch:get_port({misc_info_embedded, ssl}),
  209. %% Open 5 connections, 3 removed from the count.
  210. {ok, _} = gen_tcp:connect("localhost", Port1, [binary, {active, false}, {packet, raw}]),
  211. {ok, _} = gen_tcp:connect("localhost", Port1, [binary, {active, false}, {packet, raw}]),
  212. {ok, _} = gen_tcp:connect("localhost", Port1, [binary, {active, false}, {packet, raw}]),
  213. receive after 250 -> ok end,
  214. ranch:set_protocol_options({misc_info_embedded, tcp}, [{remove, false, 2500}]),
  215. receive after 250 -> ok end,
  216. {ok, _} = gen_tcp:connect("localhost", Port1, [binary, {active, false}, {packet, raw}]),
  217. {ok, _} = gen_tcp:connect("localhost", Port1, [binary, {active, false}, {packet, raw}]),
  218. receive after 250 -> ok end,
  219. %% Confirm the info returned by Ranch is correct.
  220. #{
  221. {misc_info_embedded, act} := #{
  222. pid := Pid2,
  223. port := Port2,
  224. max_connections := infinity, %% Option was modified.
  225. active_connections := 0,
  226. all_connections := 0,
  227. transport := ranch_tcp,
  228. transport_options := #{num_acceptors := 2},
  229. protocol := active_echo_protocol,
  230. protocol_options := {}
  231. },
  232. {misc_info_embedded, ssl} := #{
  233. pid := Pid3,
  234. port := Port3,
  235. max_connections := 1024,
  236. active_connections := 0,
  237. all_connections := 0,
  238. transport := ranch_ssl,
  239. transport_options := #{num_acceptors := 3, socket_opts := Opts},
  240. protocol := echo_protocol,
  241. protocol_options := [{}]
  242. },
  243. {misc_info_embedded, tcp} := #{
  244. pid := Pid1,
  245. port := Port1,
  246. max_connections := 1024,
  247. active_connections := 2,
  248. all_connections := 5,
  249. transport := ranch_tcp,
  250. transport_options := #{num_acceptors := 1},
  251. protocol := remove_conn_and_wait_protocol,
  252. protocol_options := [{remove, false, 2500}] %% Option was modified.
  253. }
  254. } = ranch:info(),
  255. %% Get acceptors.
  256. [_] = ranch:procs({misc_info_embedded, tcp}, acceptors),
  257. [_, _] = ranch:procs({misc_info_embedded, act}, acceptors),
  258. [_, _, _] = ranch:procs({misc_info_embedded, ssl}, acceptors),
  259. %% Get connections.
  260. [_, _, _, _, _] = ranch:procs({misc_info_embedded, tcp}, connections),
  261. [] = ranch:procs({misc_info_embedded, act}, connections),
  262. [] = ranch:procs({misc_info_embedded, ssl}, connections),
  263. %% Stop embedded tcp listener and ensure it is gone.
  264. ok = embedded_sup:stop_listener(SupPid, {misc_info_embedded, tcp}),
  265. timer:sleep(500),
  266. false = maps:is_key({misc_info_embedded, tcp}, ranch:info()),
  267. %% Stop embedded act listener and ensure it is gone.
  268. ok = embedded_sup:stop_listener(SupPid, {misc_info_embedded, act}),
  269. timer:sleep(500),
  270. false = maps:is_key({misc_info_embedded, act}, ranch:info()),
  271. %% Stop embedded ssl listener and ensure it is gone.
  272. ok = embedded_sup:stop_listener(SupPid, {misc_info_embedded, ssl}),
  273. timer:sleep(500),
  274. false = maps:is_key({misc_info_embedded, ssl}, ranch:info()),
  275. %% Stop embedded supervisor.
  276. embedded_sup:stop(SupPid),
  277. ok.
  278. misc_metrics(_) ->
  279. doc("Confirm accept/terminate metrics are correct."),
  280. Name = name(),
  281. {ok, _} = ranch:start_listener(Name, ranch_tcp, #{},
  282. notify_and_wait_protocol, #{pid => self()}),
  283. Port = ranch:get_port(Name),
  284. %% Start 10 connections.
  285. ok = connect_loop(Port, 10, 0),
  286. {10, ConnPids1} = receive_loop(connected, 400),
  287. #{metrics := Metrics1} = ranch:info(Name),
  288. {10, 0} = do_accumulate_metrics(Metrics1),
  289. %% Start 10 more connections.
  290. ok = connect_loop(Port, 10, 0),
  291. {10, ConnPids2} = receive_loop(connected, 400),
  292. #{metrics := Metrics2} = ranch:info(Name),
  293. {20, 0} = do_accumulate_metrics(Metrics2),
  294. %% Terminate 10 connections.
  295. ok = terminate_loop(stop, ConnPids2),
  296. timer:sleep(100),
  297. #{metrics := Metrics3} = ranch:info(Name),
  298. {20, 10} = do_accumulate_metrics(Metrics3),
  299. %% Terminate 10 more connections.
  300. ok = terminate_loop(stop, ConnPids1),
  301. timer:sleep(100),
  302. #{metrics := Metrics4} = ranch:info(Name),
  303. {20, 20} = do_accumulate_metrics(Metrics4),
  304. ok = ranch:stop_listener(Name),
  305. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  306. ok.
  307. do_accumulate_metrics(Metrics) ->
  308. maps:fold(
  309. fun
  310. ({conns_sup, _, accept}, N, {Accepts, Terminates}) ->
  311. {Accepts+N, Terminates};
  312. ({conns_sup, _, terminate}, N, {Accepts, Terminates}) ->
  313. {Accepts, Terminates+N}
  314. end,
  315. {0, 0},
  316. Metrics
  317. ).
  318. misc_opts_logger(_) ->
  319. doc("Confirm that messages are sent via the configured logger module."),
  320. register(misc_opts_logger, self()),
  321. {ok, _} = ranch:start_listener(name(),
  322. ranch_tcp, #{logger => ?MODULE, socket_opts => [<<"garbage">>]},
  323. echo_protocol, []),
  324. receive
  325. {warning, "Transport option " ++ _, [<<"garbage">>]} ->
  326. ok
  327. after 1000 ->
  328. error(timeout)
  329. end.
  330. warning(Format, Args) ->
  331. misc_opts_logger ! {warning, Format, Args}.
  332. misc_post_listen_callback(_) ->
  333. doc("Ensure that the post-listen callback works."),
  334. Name = name(),
  335. Self = self(),
  336. Ref = make_ref(),
  337. PostListenCb = fun (LSock) ->
  338. ok = ranch_tcp:setopts(LSock, [{send_timeout, 1000}]),
  339. Self ! {post_listen, Ref, LSock},
  340. ok
  341. end,
  342. {ok, _} = ranch:start_listener(Name,
  343. ranch_tcp, #{post_listen_callback => PostListenCb,
  344. socket_opts => [{send_timeout, infinity}]},
  345. echo_protocol, []),
  346. receive
  347. {post_listen, Ref, LSock} ->
  348. {ok, [{send_timeout, 1000}]} = ranch_tcp:getopts(LSock, [send_timeout]),
  349. ok
  350. after 1000 ->
  351. error(timeout)
  352. end,
  353. Port = ranch:get_port(Name),
  354. {ok, S} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  355. ok = gen_tcp:send(S, <<"Test">>),
  356. {ok, <<"Test">>} = gen_tcp:recv(S, 4, 1000),
  357. ok = ranch:stop_listener(Name),
  358. {error, closed} = gen_tcp:recv(S, 0, 1000),
  359. %% Make sure the listener stopped.
  360. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  361. ok.
  362. misc_post_listen_callback_error(_) ->
  363. doc("Ensure that starting a listener fails when the post-listen callback returns an error."),
  364. Name = name(),
  365. PostListenCb = fun (_) -> {error, test} end,
  366. {error, _} = ranch:start_listener(Name,
  367. ranch_tcp, #{post_listen_callback => PostListenCb},
  368. echo_protocol, []),
  369. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  370. ok.
  371. misc_repeated_remove(_) ->
  372. doc("Ensure repeated removal of connection does not crash the connection supervisor."),
  373. Name = name(),
  374. {ok, _} = ranch:start_listener(Name,
  375. ranch_tcp, #{},
  376. remove_conn_and_wait_protocol, [{remove, 5, 0}]),
  377. Port = ranch:get_port(Name),
  378. ConnsSups = lists:sort(ranch_server:get_connections_sups(Name)),
  379. {ok, _} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  380. timer:sleep(1000),
  381. ConnsSups = lists:sort(ranch_server:get_connections_sups(Name)),
  382. true = lists:all(fun ({_, ConnsSup}) -> erlang:is_process_alive(ConnsSup) end, ConnsSups),
  383. ok = ranch:stop_listener(Name).
  384. misc_set_transport_options(_) ->
  385. doc(""),
  386. Name = name(),
  387. {ok, ListenerSupPid} = ranch:start_listener(Name, ranch_tcp, #{max_connections => 10,
  388. handshake_timeout => 5000, shutdown => 1000, num_acceptors => 1,
  389. socket_opts => [{send_timeout, 5000}]}, echo_protocol, []),
  390. ok = ranch:set_transport_options(Name, #{max_connections => 20, handshake_timeout => 5001,
  391. num_acceptors => 2, shutdown => 1001, socket_opts => [{send_timeout, 5002}]}),
  392. ConnsSups = [ConnsSup || {_, ConnsSup} <- ranch_server:get_connections_sups(Name)],
  393. _ = [begin
  394. {State, _, _, _} = sys:get_state(ConnsSup),
  395. 20 = element(11, State),
  396. 5001 = element(10, State),
  397. 1001 = element(6, State)
  398. end || ConnsSup <- ConnsSups],
  399. ok = ranch:suspend_listener(Name),
  400. ok = ranch:resume_listener(Name),
  401. 2 = length(ranch:procs(Name, acceptors)),
  402. LSocket = do_get_listener_socket(ListenerSupPid),
  403. {ok, [{send_timeout, 5002}]} = ranch_tcp:getopts(LSocket, [send_timeout]),
  404. ok = ranch:stop_listener(Name).
  405. misc_wait_for_connections(_) ->
  406. doc("Ensure wait for connections works."),
  407. Name = name(),
  408. Self = self(),
  409. %% Ensure invalid arguments are rejected.
  410. {'EXIT', {badarg, _}} = begin catch ranch:wait_for_connections(Name, 'foo', 0) end,
  411. {'EXIT', {badarg, _}} = begin catch ranch:wait_for_connections(Name, '==', -1) end,
  412. {'EXIT', {badarg, _}} = begin catch ranch:wait_for_connections(Name, '==', 0, -1) end,
  413. {'EXIT', {badarg, _}} = begin catch ranch:wait_for_connections(Name, '<', 0) end,
  414. %% Create waiters for increasing number of connections.
  415. Pid1GT = do_create_waiter(Self, Name, '>', 0),
  416. Pid1GE = do_create_waiter(Self, Name, '>=', 1),
  417. Pid1EQ = do_create_waiter(Self, Name, '==', 1),
  418. Pid2GT = do_create_waiter(Self, Name, '>', 1),
  419. Pid2GE = do_create_waiter(Self, Name, '>=', 2),
  420. Pid2EQ = do_create_waiter(Self, Name, '==', 2),
  421. {ok, _} = ranch:start_listener(Name,
  422. ranch_tcp, #{num_acceptors => 1},
  423. echo_protocol, []),
  424. Port = ranch:get_port(Name),
  425. %% Create some connections, ensure that waiters respond.
  426. {ok, Sock1} = gen_tcp:connect("localhost", Port, []),
  427. ok = do_expect_waiter(Pid1GT),
  428. ok = do_expect_waiter(Pid1GE),
  429. ok = do_expect_waiter(Pid1EQ),
  430. ok = do_expect_waiter(undefined),
  431. {ok, Sock2} = gen_tcp:connect("localhost", Port, []),
  432. ok = do_expect_waiter(Pid2GT),
  433. ok = do_expect_waiter(Pid2GE),
  434. ok = do_expect_waiter(Pid2EQ),
  435. ok = do_expect_waiter(undefined),
  436. %% Create waiters for decreasing number of connections.
  437. Pid3LT = do_create_waiter(Self, Name, '<', 2),
  438. Pid3LE = do_create_waiter(Self, Name, '=<', 1),
  439. Pid3EQ = do_create_waiter(Self, Name, '==', 1),
  440. Pid4LT = do_create_waiter(Self, Name, '<', 1),
  441. Pid4LE = do_create_waiter(Self, Name, '=<', 0),
  442. Pid4EQ = do_create_waiter(Self, Name, '==', 0),
  443. %% Close connections, ensure that waiters respond.
  444. ok = gen_tcp:close(Sock1),
  445. ok = do_expect_waiter(Pid3LT),
  446. ok = do_expect_waiter(Pid3LE),
  447. ok = do_expect_waiter(Pid3EQ),
  448. ok = do_expect_waiter(undefined),
  449. ok = gen_tcp:close(Sock2),
  450. ok = do_expect_waiter(Pid4LT),
  451. ok = do_expect_waiter(Pid4LE),
  452. ok = do_expect_waiter(Pid4EQ),
  453. ok = do_expect_waiter(undefined),
  454. ok = ranch:stop_listener(Name),
  455. %% Make sure the listener stopped.
  456. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  457. ok.
  458. do_create_waiter(ReplyTo, Ref, Op, NumConns) ->
  459. spawn(fun () -> ok = ranch:wait_for_connections(Ref, Op, NumConns, 100),
  460. ReplyTo ! {wait_connections, self()} end).
  461. do_expect_waiter(WaiterPid) ->
  462. receive
  463. {wait_connections, _} when WaiterPid=:=undefined ->
  464. error;
  465. {wait_connections, Pid} when Pid=:=WaiterPid ->
  466. ok
  467. after 1000 ->
  468. case WaiterPid of
  469. undefined ->
  470. ok;
  471. _ ->
  472. timeout
  473. end
  474. end.
  475. misc_multiple_ip_local_socket_opts(_) ->
  476. case do_os_supports_local_sockets() of
  477. true ->
  478. do_misc_multiple_ip_local_socket_opts();
  479. false ->
  480. {skip, "No local socket support."}
  481. end.
  482. do_misc_multiple_ip_local_socket_opts() ->
  483. doc("Ensure that a listener uses the expected ip option if multiple are given."),
  484. Name = name(),
  485. SockFile1 = do_tempname(),
  486. SockFile2 = do_tempname(),
  487. Opts = [{ip, {local, SockFile1}}, {ip, {local, SockFile2}}],
  488. {ok, _} = ranch:start_listener(Name, ranch_tcp, #{socket_opts => Opts}, echo_protocol, []),
  489. {local, SockFile2} = ranch:get_addr(Name),
  490. %% Make sure the socket file from the ignored ip option
  491. %% has not been created.
  492. {error, enoent} = file:read_file_info(SockFile1),
  493. ok = ranch:stop_listener(Name),
  494. %% Make sure the socket file from the accepted ip option
  495. %% is removed.
  496. {error, enoent} = file:read_file_info(SockFile2),
  497. ok.
  498. %% ssl.
  499. ssl_accept_error(_) ->
  500. doc("Acceptor must not crash if client disconnects in the middle of SSL handshake."),
  501. Name = name(),
  502. Opts = ct_helper:get_certs_from_ets(),
  503. {ok, ListenerSup} = ranch:start_listener(Name,
  504. ranch_ssl, #{num_acceptors => 1, socket_opts => Opts},
  505. echo_protocol, []),
  506. Port = ranch:get_port(Name),
  507. ListenerSupChildren = supervisor:which_children(ListenerSup),
  508. {_, AcceptorsSup, _, _} = lists:keyfind(ranch_acceptors_sup, 1, ListenerSupChildren),
  509. [{{acceptor, _, _}, AcceptorPid, _, _}] = supervisor:which_children(AcceptorsSup),
  510. true = is_process_alive(AcceptorPid),
  511. {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  512. ok = gen_tcp:close(Socket),
  513. receive after 500 -> ok end,
  514. true = is_process_alive(AcceptorPid),
  515. ok = ranch:stop_listener(Name).
  516. ssl_10_acceptors_10_listen_sockets(_) ->
  517. case do_os_supports_reuseport() of
  518. true ->
  519. ok = do_ssl_10_acceptors_10_listen_sockets();
  520. false ->
  521. {skip, "No SO_REUSEPORT support."}
  522. end.
  523. do_ssl_10_acceptors_10_listen_sockets() ->
  524. doc("Ensure that we can use 10 listen sockets across 10 acceptors with SSL."),
  525. Name = name(),
  526. Opts = ct_helper:get_certs_from_ets(),
  527. {ok, ListenerSupPid} = ranch:start_listener(Name,
  528. ranch_ssl, #{
  529. num_acceptors => 10,
  530. num_listen_sockets => 10,
  531. socket_opts => [{raw, 1, 15, <<1:32/native>>}|Opts]},
  532. echo_protocol, []),
  533. 10 = length(do_get_listener_sockets(ListenerSupPid)),
  534. ok = ranch:stop_listener(Name),
  535. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  536. ok.
  537. ssl_many_listen_sockets_no_reuseport(_) ->
  538. case do_os_supports_reuseport() of
  539. true ->
  540. ok = do_ssl_many_listen_sockets_no_reuseport();
  541. false ->
  542. {skip, "No SO_REUSEPORT support."}
  543. end.
  544. do_ssl_many_listen_sockets_no_reuseport() ->
  545. doc("Confirm that ranch:start_listener/5 fails when SO_REUSEPORT is not available with SSL."),
  546. Name = name(),
  547. Opts = ct_helper:get_certs_from_ets(),
  548. {error, eaddrinuse} = ranch:start_listener(Name,
  549. ranch_ssl, #{
  550. num_acceptors => 10,
  551. num_listen_sockets => 10,
  552. socket_opts => [{raw, 1, 15, <<0:32/native>>}|Opts]},
  553. echo_protocol, []),
  554. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  555. ok.
  556. ssl_active_echo(_) ->
  557. doc("Ensure that active mode works with SSL transport."),
  558. Name = name(),
  559. Opts = ct_helper:get_certs_from_ets(),
  560. {ok, _} = ranch:start_listener(Name,
  561. ranch_ssl, Opts,
  562. active_echo_protocol, []),
  563. Port = ranch:get_port(Name),
  564. {ok, Socket} = ssl:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  565. ok = ssl:send(Socket, <<"SSL Ranch is working!">>),
  566. {ok, <<"SSL Ranch is working!">>} = ssl:recv(Socket, 21, 1000),
  567. ok = ranch:stop_listener(Name),
  568. {error, closed} = ssl:recv(Socket, 0, 1000),
  569. %% Make sure the listener stopped.
  570. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  571. ok.
  572. ssl_active_n_echo(_) ->
  573. case do_get_ssl_version() >= {9, 2, 0} of
  574. true ->
  575. do_ssl_active_n_echo();
  576. false ->
  577. {skip, "No Active N support."}
  578. end.
  579. do_ssl_active_n_echo() ->
  580. doc("Ensure that active N mode works with SSL transport."),
  581. Name = name(),
  582. Opts = ct_helper:get_certs_from_ets(),
  583. {ok, _} = ranch:start_listener(Name,
  584. ranch_ssl, Opts,
  585. batch_echo_protocol, [{batch_size, 3}]),
  586. Port = ranch:get_port(Name),
  587. {ok, Socket} = ssl:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  588. ok = ssl:send(Socket, <<"One">>),
  589. {ok, <<"OK">>} = ssl:recv(Socket, 2, 1000),
  590. ok = ssl:send(Socket, <<"Two">>),
  591. {ok, <<"OK">>} = ssl:recv(Socket, 2, 1000),
  592. ok = ssl:send(Socket, <<"Three">>),
  593. {ok, <<"OK">>} = ssl:recv(Socket, 2, 1000),
  594. {ok, <<"OneTwoThree">>} = ssl:recv(Socket, 11, 1000),
  595. ok = ranch:stop_listener(Name),
  596. {error, closed} = ssl:recv(Socket, 0, 1000),
  597. %% Make sure the listener stopped.
  598. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  599. ok.
  600. ssl_echo(_) ->
  601. doc("Ensure that passive mode works with SSL transport."),
  602. Name = name(),
  603. Opts = ct_helper:get_certs_from_ets(),
  604. {ok, _} = ranch:start_listener(Name,
  605. ranch_ssl, Opts,
  606. echo_protocol, []),
  607. Port = ranch:get_port(Name),
  608. {ok, Socket} = ssl:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  609. ok = ssl:send(Socket, <<"SSL Ranch is working!">>),
  610. {ok, <<"SSL Ranch is working!">>} = ssl:recv(Socket, 21, 1000),
  611. ok = ranch:stop_listener(Name),
  612. {error, closed} = ssl:recv(Socket, 0, 1000),
  613. %% Make sure the listener stopped.
  614. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  615. ok.
  616. ssl_handshake(_) ->
  617. doc("Ensure that multiple steps handshake works with SSL transport."),
  618. Name = name(),
  619. {CaCert1, Cert1, Key1} = ct_helper:make_certs(),
  620. {CaCert2, Cert2, Key2} = ct_helper:make_certs(),
  621. Opts1 = [{cert, Cert1}, {key, Key1}, {cacerts, [CaCert1]}, {verify, verify_peer}],
  622. Opts2 = [{cert, Cert2}, {key, Key2}, {cacerts, [CaCert2]}, {verify, verify_peer}],
  623. DefaultOpts = ct_helper:get_certs_from_ets(),
  624. {ok, _} = ranch:start_listener(Name,
  625. ranch_ssl, [{handshake, hello}|DefaultOpts],
  626. handshake_protocol, #{"ranch1" => Opts1, "ranch2" => Opts2}),
  627. Port = ranch:get_port(Name),
  628. {ok, Socket1} = ssl:connect("localhost", Port, [binary, {active, false}, {packet, raw},
  629. {server_name_indication, "ranch1"}], 5000),
  630. {ok, Cert1} = ssl:peercert(Socket1),
  631. ok = ssl:send(Socket1, <<"SSL Ranch is working!">>),
  632. {ok, <<"SSL Ranch is working!">>} = ssl:recv(Socket1, 21, 1000),
  633. {ok, Socket2} = ssl:connect("localhost", Port, [binary, {active, false}, {packet, raw},
  634. {server_name_indication, "ranch2"}], 5000),
  635. {ok, Cert2} = ssl:peercert(Socket2),
  636. ok = ssl:send(Socket2, <<"SSL Ranch is working!">>),
  637. {ok, <<"SSL Ranch is working!">>} = ssl:recv(Socket2, 21, 1000),
  638. ok = ranch:stop_listener(Name),
  639. {error, closed} = ssl:recv(Socket1, 0, 1000),
  640. {error, closed} = ssl:recv(Socket2, 0, 1000),
  641. %% Make sure the listener stopped.
  642. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  643. ok.
  644. ssl_local_echo(_) ->
  645. case do_os_supports_local_sockets() of
  646. true ->
  647. do_ssl_local_echo();
  648. false ->
  649. {skip, "No local socket support."}
  650. end.
  651. do_ssl_local_echo() ->
  652. doc("Ensure that listening on a local socket works with SSL transport."),
  653. SockFile = do_tempname(),
  654. try
  655. Name = name(),
  656. Opts = ct_helper:get_certs_from_ets(),
  657. {ok, _} = ranch:start_listener(Name,
  658. ranch_ssl, #{socket_opts => [{ip, {local, SockFile}}|Opts]},
  659. echo_protocol, []),
  660. undefined = ranch:get_port(Name),
  661. {ok, Socket} = ssl:connect({local, SockFile}, 0, [binary, {active, false}, {packet, raw}]),
  662. ok = ssl:send(Socket, <<"SSL Ranch is working!">>),
  663. {ok, <<"SSL Ranch is working!">>} = ssl:recv(Socket, 21, 1000),
  664. ok = ranch:stop_listener(Name),
  665. {error, closed} = ssl:recv(Socket, 0, 1000),
  666. %% Make sure the listener stopped.
  667. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  668. %% Make sure the socket file is removed.
  669. {error, enoent} = file:read_file_info(SockFile),
  670. ok
  671. after
  672. file:delete(SockFile)
  673. end.
  674. ssl_sni_echo(_) ->
  675. doc("Ensure that SNI works with SSL transport."),
  676. Name = name(),
  677. Opts = ct_helper:get_certs_from_ets(),
  678. {ok, _} = ranch:start_listener(Name,
  679. ranch_ssl, [{sni_hosts, [{"localhost", Opts}]}],
  680. echo_protocol, []),
  681. Port = ranch:get_port(Name),
  682. %% We stick to TLS 1.2 because there seems to be a bug in OTP-23.0rc2
  683. %% that leads to a malformed_handshake_data error.
  684. {ok, Socket} = ssl:connect("localhost", Port,
  685. [binary, {active, false}, {packet, raw}, {versions, ['tlsv1.2']}]),
  686. ok = ssl:send(Socket, <<"SSL Ranch is working!">>),
  687. {ok, <<"SSL Ranch is working!">>} = ssl:recv(Socket, 21, 1000),
  688. ok = ranch:stop_listener(Name),
  689. {error, closed} = ssl:recv(Socket, 0, 1000),
  690. %% Make sure the listener stopped.
  691. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  692. ok.
  693. ssl_sni_fail(_) ->
  694. doc("Ensure that connection fails when host is not in SNI list."),
  695. Name = name(),
  696. Opts = ct_helper:get_certs_from_ets(),
  697. {ok, _} = ranch:start_listener(Name,
  698. ranch_ssl, [{sni_hosts, [{"pouet", Opts}]}],
  699. echo_protocol, []),
  700. Port = ranch:get_port(Name),
  701. %% We stick to TLS 1.2 because there seems to be a bug in OTP-23.0rc2
  702. %% that leads to a malformed_handshake_data error.
  703. {error, _} = ssl:connect("localhost", Port,
  704. [binary, {active, false}, {packet, raw}, {versions, ['tlsv1.2']}]),
  705. ok = ranch:stop_listener(Name),
  706. %% Make sure the listener stopped.
  707. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  708. ok.
  709. ssl_tls_psk(_) ->
  710. doc("Ensure that TLS-PSK works without certificate."),
  711. Name = name(),
  712. Ciphers = [#{cipher => aes_256_gcm, key_exchange => psk, mac => aead, prf => sha384}],
  713. LookupFun = {fun psk_lookup_helper/3, <<"shared_secret">>},
  714. {ok, _} = ranch:start_listener(Name,
  715. ranch_ssl, [{ciphers, Ciphers}, {user_lookup_fun, LookupFun}, {versions, ['tlsv1.2']}],
  716. echo_protocol, []),
  717. Port = ranch:get_port(Name),
  718. {ok, Socket} = ssl:connect("localhost", Port, [
  719. binary, {active, false}, {ciphers, Ciphers},
  720. {user_lookup_fun, LookupFun}, {versions, ['tlsv1.2']}
  721. ]),
  722. ok = ssl:send(Socket, <<"SSL Ranch is working!">>),
  723. {ok, <<"SSL Ranch is working!">>} = ssl:recv(Socket, 21, 1000),
  724. ok = ranch:stop_listener(Name),
  725. {error, closed} = ssl:recv(Socket, 0, 1000),
  726. %% Make sure the listener stopped.
  727. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  728. ok.
  729. ssl_tls_psk_fail(_) ->
  730. doc("Ensure that TLS-PSK filed for different shared keys."),
  731. Name = name(),
  732. Ciphers = [#{cipher => aes_256_gcm, key_exchange => psk, mac => aead, prf => sha384}],
  733. ServerLookupFun = {fun psk_lookup_helper/3, <<"server_secret">>},
  734. ClientLookupFun = {fun psk_lookup_helper/3, <<"client_secret">>},
  735. {ok, _} = ranch:start_listener(Name,
  736. ranch_ssl, [{ciphers, Ciphers}, {user_lookup_fun, ServerLookupFun}, {versions, ['tlsv1.2']}],
  737. echo_protocol, []),
  738. Port = ranch:get_port(Name),
  739. {error, _} = ssl:connect("localhost", Port, [
  740. binary, {active, false}, {ciphers, Ciphers},
  741. {user_lookup_fun, ClientLookupFun}, {versions, ['tlsv1.2']}
  742. ]),
  743. ok = ranch:stop_listener(Name),
  744. %% Make sure the listener stopped.
  745. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  746. ok.
  747. psk_lookup_helper(psk, _PskIdentity, UserState) ->
  748. {ok, UserState}.
  749. ssl_upgrade_from_tcp(_) ->
  750. doc("Ensure a TCP socket can be upgraded to SSL"),
  751. Name = name(),
  752. {ok, _} = ranch:start_listener(Name,
  753. ranch_tcp, #{},
  754. ssl_upgrade_protocol, []),
  755. Port = ranch:get_port(Name),
  756. {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  757. ok = gen_tcp:send(Socket, <<"ECHO Before upgrading to SSL">>),
  758. {ok, <<"Before upgrading to SSL">>} = gen_tcp:recv(Socket, 23, 1000),
  759. ok = gen_tcp:send(Socket, <<"UPGRADE">>),
  760. {ok, <<"READY">>} = gen_tcp:recv(Socket, 5, 1000),
  761. {ok, SslSocket} = ssl:connect(Socket, [{verify, verify_none}], 5000),
  762. ok = ssl:send(SslSocket, <<"ECHO After upgrading to SSL">>),
  763. {ok, <<"After upgrading to SSL">>} = ssl:recv(SslSocket, 22, 1000),
  764. ok = ranch:stop_listener(Name),
  765. {error, closed} = ssl:recv(SslSocket, 0, 1000),
  766. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  767. ok.
  768. ssl_graceful(_) ->
  769. doc("Ensure suspending and resuming of listeners does not kill active connections."),
  770. Name = name(),
  771. Opts = ct_helper:get_certs_from_ets(),
  772. {ok, _} = ranch:start_listener(Name,
  773. ranch_ssl, Opts,
  774. echo_protocol, []),
  775. Port = ranch:get_port(Name),
  776. %% Make sure connections with a fresh listener work.
  777. running = ranch:get_status(Name),
  778. {ok, Socket1} = ssl:connect("localhost", Port,
  779. [binary, {active, false}, {packet, raw}]),
  780. ok = ssl:send(Socket1, <<"SSL with fresh listener">>),
  781. {ok, <<"SSL with fresh listener">>} = ssl:recv(Socket1, 23, 1000),
  782. %% Suspend listener, make sure established connections keep running.
  783. ok = ranch:suspend_listener(Name),
  784. suspended = ranch:get_status(Name),
  785. ok = ssl:send(Socket1, <<"SSL with suspended listener">>),
  786. {ok, <<"SSL with suspended listener">>} = ssl:recv(Socket1, 27, 1000),
  787. %% Make sure new connections are refused on the suspended listener.
  788. {error, econnrefused} = ssl:connect("localhost", Port,
  789. [binary, {active, false}, {packet, raw}]),
  790. %% Make sure transport options can be changed when listener is suspended.
  791. ok = ranch:set_transport_options(Name, #{socket_opts => [{port, Port}|Opts]}),
  792. %% Resume listener, make sure connections can be established again.
  793. ok = ranch:resume_listener(Name),
  794. running = ranch:get_status(Name),
  795. {ok, Socket2} = ssl:connect("localhost", Port,
  796. [binary, {active, false}, {packet, raw}]),
  797. ok = ssl:send(Socket2, <<"SSL with resumed listener">>),
  798. {ok, <<"SSL with resumed listener">>} = ssl:recv(Socket2, 25, 1000),
  799. ok = ranch:stop_listener(Name),
  800. {error, closed} = ssl:recv(Socket1, 0, 1000),
  801. {error, closed} = ssl:recv(Socket2, 0, 1000),
  802. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  803. ok.
  804. ssl_getopts_capability(_) ->
  805. doc("Ensure getopts/2 capability."),
  806. Name=name(),
  807. Opts=ct_helper:get_certs_from_ets(),
  808. {ok, _} = ranch:start_listener(Name,
  809. ranch_ssl, Opts,
  810. transport_capabilities_protocol, []),
  811. Port=ranch:get_port(Name),
  812. {ok, Socket}=ssl:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  813. ok=ssl:send(Socket, <<"getopts/2">>),
  814. {ok, <<"OK">>}=ssl:recv(Socket, 0, 1000),
  815. ok=ranch:stop_listener(Name),
  816. {error, closed}=ssl:recv(Socket, 0, 1000),
  817. {'EXIT', _}=begin catch ranch:get_port(Name) end,
  818. ok.
  819. ssl_getstat_capability(_) ->
  820. doc("Ensure getstat/1,2 capability."),
  821. Name=name(),
  822. Opts=ct_helper:get_certs_from_ets(),
  823. {ok, _} = ranch:start_listener(Name,
  824. ranch_ssl, Opts,
  825. transport_capabilities_protocol, []),
  826. Port=ranch:get_port(Name),
  827. {ok, Socket}=ssl:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  828. ok=ssl:send(Socket, <<"getstat/1">>),
  829. {ok, <<"OK">>}=ssl:recv(Socket, 0, 1000),
  830. ok=ssl:send(Socket, <<"getstat/2">>),
  831. {ok, <<"OK">>}=ssl:recv(Socket, 0, 1000),
  832. ok=ranch:stop_listener(Name),
  833. {error, closed}=ssl:recv(Socket, 0, 1000),
  834. {'EXIT', _}=begin catch ranch:get_port(Name) end,
  835. ok.
  836. ssl_error_eaddrinuse(_) ->
  837. doc("Ensure that failure due to an eaddrinuse returns a compact readable error."),
  838. Name = name(),
  839. Opts = ct_helper:get_certs_from_ets(),
  840. {ok, _} = ranch:start_listener(Name,
  841. ranch_ssl, Opts,
  842. active_echo_protocol, []),
  843. Port = ranch:get_port(Name),
  844. {error, eaddrinuse} = ranch:start_listener({Name, fails},
  845. ranch_ssl, [{port, Port}|Opts],
  846. active_echo_protocol, []),
  847. ok = ranch:stop_listener(Name),
  848. %% Make sure the listener stopped.
  849. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  850. ok.
  851. ssl_error_no_cert(_) ->
  852. doc("Ensure that failure due to missing certificate returns a compact readable error."),
  853. {error, no_cert} = ranch:start_listener(name(),
  854. ranch_ssl, #{},
  855. active_echo_protocol, []),
  856. ok.
  857. ssl_error_eacces(_) ->
  858. case os:type() of
  859. {win32, nt} ->
  860. {skip, "No privileged ports."};
  861. {unix, darwin} ->
  862. {skip, "No privileged ports."};
  863. _ ->
  864. doc("Ensure that failure due to an eacces returns a compact readable error."),
  865. Name = name(),
  866. Opts = ct_helper:get_certs_from_ets(),
  867. {error, eacces} = ranch:start_listener(Name,
  868. ranch_ssl, [{port, 283}|Opts],
  869. active_echo_protocol, []),
  870. ok
  871. end.
  872. ssl_unsupported_tlsv13_options(_) ->
  873. {available, Versions} = lists:keyfind(available, 1, ssl:versions()),
  874. case {lists:member('tlsv1.3', Versions), do_get_ssl_version() >= {10, 0, 0}} of
  875. {true, true} ->
  876. do_ssl_unsupported_tlsv13_options();
  877. {false, _} ->
  878. {skip, "No TLSv1.3 support."};
  879. {_, false} ->
  880. {skip, "No TLSv1.3 option dependency checking."}
  881. end.
  882. do_ssl_unsupported_tlsv13_options() ->
  883. doc("Ensure that a listener can be started when TLSv1.3 is "
  884. "the only protocol and unsupported options are present."),
  885. CheckOpts = [
  886. {beast_mitigation, one_n_minus_one},
  887. {client_renegotiation, true},
  888. {next_protocols_advertised, [<<"dummy">>]},
  889. {padding_check, true},
  890. {psk_identity, "dummy"},
  891. {secure_renegotiate, true},
  892. {reuse_session, fun (_, _, _, _) -> true end},
  893. {reuse_sessions, true},
  894. {user_lookup_fun, {fun (_, _, _) -> error end, <<"dummy">>}}
  895. ],
  896. Name = name(),
  897. Opts = ct_helper:get_certs_from_ets() ++ [{versions, ['tlsv1.3']}],
  898. ok = lists:foreach(
  899. fun (CheckOpt) ->
  900. Opts1 = Opts ++ [CheckOpt],
  901. {error, {options, dependency, _}} = ssl:listen(0, Opts1),
  902. {ok, _} = ranch:start_listener(Name,
  903. ranch_ssl, #{socket_opts => Opts1},
  904. echo_protocol, []),
  905. ok = ranch:stop_listener(Name)
  906. end,
  907. CheckOpts
  908. ),
  909. ok.
  910. %% tcp.
  911. tcp_10_acceptors_10_listen_sockets(_) ->
  912. case do_os_supports_reuseport() of
  913. true ->
  914. ok = do_tcp_10_acceptors_10_listen_sockets();
  915. false ->
  916. {skip, "No SO_REUSEPORT support."}
  917. end.
  918. do_tcp_10_acceptors_10_listen_sockets() ->
  919. doc("Ensure that we can use 10 listen sockets across 10 acceptors with TCP."),
  920. Name = name(),
  921. {ok, ListenerSupPid} = ranch:start_listener(Name,
  922. ranch_tcp, #{
  923. num_acceptors => 10,
  924. num_listen_sockets => 10,
  925. socket_opts => [{raw, 1, 15, <<1:32/native>>}]},
  926. echo_protocol, []),
  927. 10 = length(do_get_listener_sockets(ListenerSupPid)),
  928. ok = ranch:stop_listener(Name),
  929. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  930. ok.
  931. tcp_many_listen_sockets_no_reuseport(_) ->
  932. case do_os_supports_reuseport() of
  933. true ->
  934. ok = do_tcp_many_listen_sockets_no_reuseport();
  935. false ->
  936. {skip, "No SO_REUSEPORT support."}
  937. end.
  938. do_tcp_many_listen_sockets_no_reuseport() ->
  939. doc("Confirm that ranch:start_listener/5 fails when SO_REUSEPORT is not available with TCP."),
  940. Name = name(),
  941. {error, eaddrinuse} = ranch:start_listener(Name,
  942. ranch_tcp, #{
  943. num_acceptors => 10,
  944. num_listen_sockets => 10,
  945. socket_opts => [{raw, 1, 15, <<0:32/native>>}]},
  946. echo_protocol, []),
  947. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  948. ok.
  949. tcp_active_echo(_) ->
  950. doc("Ensure that active mode works with TCP transport."),
  951. Name = name(),
  952. {ok, _} = ranch:start_listener(Name,
  953. ranch_tcp, #{},
  954. active_echo_protocol, []),
  955. Port = ranch:get_port(Name),
  956. {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  957. ok = gen_tcp:send(Socket, <<"TCP Ranch is working!">>),
  958. {ok, <<"TCP Ranch is working!">>} = gen_tcp:recv(Socket, 21, 1000),
  959. ok = ranch:stop_listener(Name),
  960. {error, closed} = gen_tcp:recv(Socket, 0, 1000),
  961. %% Make sure the listener stopped.
  962. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  963. ok.
  964. tcp_active_n_echo(_) ->
  965. doc("Ensure that active N mode works with TCP transport."),
  966. Name = name(),
  967. {ok, _} = ranch:start_listener(Name,
  968. ranch_tcp, #{},
  969. batch_echo_protocol, [{batch_size, 3}]),
  970. Port = ranch:get_port(Name),
  971. {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  972. ok = gen_tcp:send(Socket, <<"One">>),
  973. {ok, <<"OK">>} = gen_tcp:recv(Socket, 2, 1000),
  974. ok = gen_tcp:send(Socket, <<"Two">>),
  975. {ok, <<"OK">>} = gen_tcp:recv(Socket, 2, 1000),
  976. ok = gen_tcp:send(Socket, <<"Three">>),
  977. {ok, <<"OK">>} = gen_tcp:recv(Socket, 2, 1000),
  978. {ok, <<"OneTwoThree">>} = gen_tcp:recv(Socket, 11, 1000),
  979. ok = ranch:stop_listener(Name),
  980. {error, closed} = gen_tcp:recv(Socket, 0, 1000),
  981. %% Make sure the listener stopped.
  982. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  983. ok.
  984. tcp_echo(_) ->
  985. doc("Ensure that passive mode works with TCP transport."),
  986. Name = name(),
  987. {ok, _} = ranch:start_listener(Name,
  988. ranch_tcp, #{},
  989. echo_protocol, []),
  990. Port = ranch:get_port(Name),
  991. {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  992. ok = gen_tcp:send(Socket, <<"TCP Ranch is working!">>),
  993. {ok, <<"TCP Ranch is working!">>} = gen_tcp:recv(Socket, 21, 1000),
  994. ok = ranch:stop_listener(Name),
  995. {error, closed} = gen_tcp:recv(Socket, 0, 1000),
  996. %% Make sure the listener stopped.
  997. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  998. ok.
  999. tcp_local_echo(_) ->
  1000. case do_os_supports_local_sockets() of
  1001. true ->
  1002. do_tcp_local_echo();
  1003. false ->
  1004. {skip, "No local socket support."}
  1005. end.
  1006. do_tcp_local_echo() ->
  1007. doc("Ensure that listening on a local socket works with TCP transport."),
  1008. SockFile = do_tempname(),
  1009. try
  1010. Name = name(),
  1011. {ok, _} = ranch:start_listener(Name,
  1012. ranch_tcp, #{socket_opts => [{ip, {local, SockFile}}]},
  1013. echo_protocol, []),
  1014. undefined = ranch:get_port(Name),
  1015. {ok, Socket} = gen_tcp:connect({local, SockFile}, 0, [binary, {active, false}, {packet, raw}]),
  1016. ok = gen_tcp:send(Socket, <<"TCP Ranch is working!">>),
  1017. {ok, <<"TCP Ranch is working!">>} = gen_tcp:recv(Socket, 21, 1000),
  1018. ok = ranch:stop_listener(Name),
  1019. {error, closed} = gen_tcp:recv(Socket, 0, 1000),
  1020. %% Make sure the listener stopped.
  1021. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  1022. %% Make sure the socket file is removed.
  1023. {error, enoent} = file:read_file_info(SockFile),
  1024. ok
  1025. after
  1026. file:delete(SockFile)
  1027. end.
  1028. tcp_graceful(_) ->
  1029. doc("Ensure suspending and resuming of listeners does not kill active connections."),
  1030. Name = name(),
  1031. {ok, _} = ranch:start_listener(Name,
  1032. ranch_tcp, #{},
  1033. echo_protocol, []),
  1034. Port = ranch:get_port(Name),
  1035. %% Make sure connections with a fresh listener work.
  1036. running = ranch:get_status(Name),
  1037. {ok, Socket1} = gen_tcp:connect("localhost", Port,
  1038. [binary, {active, false}, {packet, raw}]),
  1039. ok = gen_tcp:send(Socket1, <<"TCP with fresh listener">>),
  1040. {ok, <<"TCP with fresh listener">>} = gen_tcp:recv(Socket1, 23, 1000),
  1041. %% Suspend listener, make sure established connections keep running.
  1042. ok = ranch:suspend_listener(Name),
  1043. suspended = ranch:get_status(Name),
  1044. ok = gen_tcp:send(Socket1, <<"TCP with suspended listener">>),
  1045. {ok, <<"TCP with suspended listener">>} = gen_tcp:recv(Socket1, 27, 1000),
  1046. %% Make sure new connections are refused on the suspended listener.
  1047. {error, econnrefused} = gen_tcp:connect("localhost", Port,
  1048. [binary, {active, false}, {packet, raw}]),
  1049. %% Make sure transport options can be changed when listener is suspended.
  1050. ok = ranch:set_transport_options(Name, [{port, Port}]),
  1051. %% Resume listener, make sure connections can be established again.
  1052. ok = ranch:resume_listener(Name),
  1053. running = ranch:get_status(Name),
  1054. {ok, Socket2} = gen_tcp:connect("localhost", Port,
  1055. [binary, {active, false}, {packet, raw}]),
  1056. ok = gen_tcp:send(Socket2, <<"TCP with resumed listener">>),
  1057. {ok, <<"TCP with resumed listener">>} = gen_tcp:recv(Socket2, 25, 1000),
  1058. ok = ranch:stop_listener(Name),
  1059. {error, closed} = gen_tcp:recv(Socket1, 0, 1000),
  1060. {error, closed} = gen_tcp:recv(Socket2, 0, 1000),
  1061. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  1062. ok.
  1063. tcp_inherit_options(_) ->
  1064. doc("Ensure TCP options are inherited in the protocol."),
  1065. Name = name(),
  1066. Opts = [{nodelay, false}, {send_timeout_close, false}],
  1067. {ok, _} = ranch:start_listener(Name,
  1068. ranch_tcp, Opts,
  1069. check_tcp_options, [{pid, self()} | Opts]),
  1070. Port = ranch:get_port(Name),
  1071. {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, true}, {packet, raw}]),
  1072. receive checked -> ok after 1000 -> error(timeout) end,
  1073. ok = gen_tcp:close(Socket),
  1074. ok = ranch:stop_listener(Name).
  1075. tcp_max_connections(_) ->
  1076. doc("Ensure the max_connections option actually limits connections."),
  1077. Name = name(),
  1078. {ok, _} = ranch:start_listener(Name,
  1079. ranch_tcp, #{max_connections => 10, num_acceptors => 1},
  1080. notify_and_wait_protocol, #{pid => self()}),
  1081. Port = ranch:get_port(Name),
  1082. ok = connect_loop(Port, 11, 150),
  1083. 10 = ranch_server:count_connections(Name),
  1084. {10, Pids1} = receive_loop(connected, 400),
  1085. ok = terminate_loop(stop, Pids1),
  1086. {1, Pids2} = receive_loop(connected, 1000),
  1087. ok = terminate_loop(stop, Pids2),
  1088. ok = ranch:stop_listener(Name).
  1089. tcp_max_connections_and_beyond(_) ->
  1090. doc("Ensure the max_connections option works when connections are removed from the count."),
  1091. Name = name(),
  1092. {ok, _} = ranch:start_listener(Name,
  1093. ranch_tcp, #{max_connections => 10, num_acceptors => 1},
  1094. remove_conn_and_wait_protocol, [{remove, true, 2500}]),
  1095. Port = ranch:get_port(Name),
  1096. ok = connect_loop(Port, 10, 0),
  1097. receive after 250 -> ok end,
  1098. 0 = ranch_server:count_connections(Name),
  1099. 10 = length(do_conns_which_children(Name)),
  1100. Counts = do_conns_count_children(Name),
  1101. {_, 1} = lists:keyfind(specs, 1, Counts),
  1102. {_, 0} = lists:keyfind(supervisors, 1, Counts),
  1103. {_, 10} = lists:keyfind(active, 1, Counts),
  1104. {_, 10} = lists:keyfind(workers, 1, Counts),
  1105. ranch:set_protocol_options(Name, [{remove, false, 2500}]),
  1106. receive after 250 -> ok end,
  1107. ok = connect_loop(Port, 10, 0),
  1108. receive after 250 -> ok end,
  1109. 10 = ranch_server:count_connections(Name),
  1110. 20 = length(do_conns_which_children(Name)),
  1111. Counts2 = do_conns_count_children(Name),
  1112. {_, 20} = lists:keyfind(active, 1, Counts2),
  1113. {_, 20} = lists:keyfind(workers, 1, Counts2),
  1114. ok = ranch:stop_listener(Name).
  1115. tcp_max_connections_infinity(_) ->
  1116. doc("Set the max_connections option from 10 to infinity and back to 10."),
  1117. Name = name(),
  1118. {ok, _} = ranch:start_listener(Name,
  1119. ranch_tcp, #{max_connections => 10, num_acceptors => 1},
  1120. notify_and_wait_protocol, #{pid => self()}),
  1121. Port = ranch:get_port(Name),
  1122. ok = connect_loop(Port, 20, 0),
  1123. 10 = ranch_server:count_connections(Name),
  1124. {10, Pids1} = receive_loop(connected, 1000),
  1125. 10 = ranch_server:count_connections(Name),
  1126. 10 = ranch:get_max_connections(Name),
  1127. ranch:set_max_connections(Name, infinity),
  1128. receive after 250 -> ok end,
  1129. 20 = ranch_server:count_connections(Name),
  1130. infinity = ranch:get_max_connections(Name),
  1131. ranch:set_max_connections(Name, 10),
  1132. 20 = ranch_server:count_connections(Name),
  1133. {10, Pids2} = receive_loop(connected, 1000),
  1134. ok = terminate_loop(stop, Pids1 ++ Pids2),
  1135. ok = ranch:stop_listener(Name).
  1136. tcp_remove_connections(_) ->
  1137. doc("Ensure that removed connections are only removed once."),
  1138. Name = name(),
  1139. {ok, _} = ranch:start_listener(Name,
  1140. ranch_tcp, #{},
  1141. remove_conn_and_wait_protocol, [{remove, true, 0}]),
  1142. Port = ranch:get_port(Name),
  1143. ok = connect_loop(Port, 10, 0),
  1144. receive after 250 -> ok end,
  1145. 0 = ranch_server:count_connections(Name),
  1146. ok = ranch:stop_listener(Name).
  1147. tcp_remove_connections_acceptor_wakeup(_) ->
  1148. doc("Ensure that removed connections wake up acceptors."),
  1149. Name = name(),
  1150. {ok, _} = ranch:start_listener(Name,
  1151. ranch_tcp, #{max_connections => 1, num_acceptors => 1},
  1152. remove_conn_and_wait_protocol, [{remove, true, infinity}]),
  1153. Port = ranch:get_port(Name),
  1154. ConnectOptions = [binary, {active, false}],
  1155. Localhost = "localhost",
  1156. {ok, Socket1} = gen_tcp:connect(Localhost, Port, ConnectOptions),
  1157. {ok, Socket2} = gen_tcp:connect(Localhost, Port, ConnectOptions),
  1158. {ok, Socket3} = gen_tcp:connect(Localhost, Port, ConnectOptions),
  1159. ok = gen_tcp:send(Socket3, <<"bye">>),
  1160. true = maps:get(all_connections, ranch:info(Name)) >= 2,
  1161. ok = gen_tcp:send(Socket1, <<"bye">>),
  1162. ok = gen_tcp:send(Socket2, <<"bye">>),
  1163. ok = ranch:stop_listener(Name).
  1164. tcp_set_max_connections(_) ->
  1165. doc("Ensure that changing the max_connections option to a larger value allows for more connections."),
  1166. Name = name(),
  1167. {ok, _} = ranch:start_listener(Name,
  1168. ranch_tcp, #{max_connections => 10, num_acceptors => 1},
  1169. notify_and_wait_protocol, #{pid => self()}),
  1170. Port = ranch:get_port(Name),
  1171. ok = connect_loop(Port, 20, 0),
  1172. 10 = ranch_server:count_connections(Name),
  1173. {10, Pids1} = receive_loop(connected, 1000),
  1174. 10 = ranch:get_max_connections(Name),
  1175. ranch:set_max_connections(Name, 20),
  1176. {10, Pids2} = receive_loop(connected, 1000),
  1177. 20 = ranch:get_max_connections(Name),
  1178. ok = terminate_loop(stop, Pids1 ++ Pids2),
  1179. ok = ranch:stop_listener(Name).
  1180. tcp_set_max_connections_clean(_) ->
  1181. doc("Ensure that setting max_connections does not crash any process."),
  1182. Name = name(),
  1183. {ok, ListSupPid} = ranch:start_listener(Name,
  1184. ranch_tcp, #{max_connections => 4},
  1185. notify_and_wait_protocol, #{pid => self()}),
  1186. Children = supervisor:which_children(ListSupPid),
  1187. {_, AccSupPid, _, _} = lists:keyfind(ranch_acceptors_sup, 1, Children),
  1188. 1 = erlang:trace(ListSupPid, true, [procs]),
  1189. 1 = erlang:trace(AccSupPid, true, [procs]),
  1190. Port = ranch:get_port(Name),
  1191. N = 20,
  1192. ok = connect_loop(Port, N*5, 0),
  1193. %% Randomly set max_connections.
  1194. [spawn(ranch, set_max_connections, [Name, Max]) ||
  1195. Max <- lists:flatten(lists:duplicate(N, [6, 4, 8, infinity]))],
  1196. receive
  1197. {trace, _, spawn, _, _} ->
  1198. error(dirty_set_max_connections)
  1199. after
  1200. 2000 -> ok
  1201. end,
  1202. _ = erlang:trace(all, false, [all]),
  1203. ok = clean_traces(),
  1204. ok = ranch:stop_listener(Name).
  1205. tcp_getopts_capability(_) ->
  1206. doc("Ensure getopts/2 capability."),
  1207. Name=name(),
  1208. {ok, _}=ranch:start_listener(Name,
  1209. ranch_tcp, #{},
  1210. transport_capabilities_protocol, []),
  1211. Port=ranch:get_port(Name),
  1212. {ok, Socket}=gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  1213. ok=gen_tcp:send(Socket, <<"getopts/2">>),
  1214. {ok, <<"OK">>}=gen_tcp:recv(Socket, 0, 1000),
  1215. ok=ranch:stop_listener(Name),
  1216. {error, closed}=gen_tcp:recv(Socket, 0, 1000),
  1217. {'EXIT', _}=begin catch ranch:get_port(Name) end,
  1218. ok.
  1219. tcp_getstat_capability(_) ->
  1220. doc("Ensure getstat/1,2 capability."),
  1221. Name=name(),
  1222. {ok, _}=ranch:start_listener(Name,
  1223. ranch_tcp, #{},
  1224. transport_capabilities_protocol, []),
  1225. Port=ranch:get_port(Name),
  1226. {ok, Socket}=gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  1227. ok=gen_tcp:send(Socket, <<"getstat/1">>),
  1228. {ok, <<"OK">>}=gen_tcp:recv(Socket, 0, 1000),
  1229. ok=gen_tcp:send(Socket, <<"getstat/2">>),
  1230. {ok, <<"OK">>}=gen_tcp:recv(Socket, 0, 1000),
  1231. ok=ranch:stop_listener(Name),
  1232. {error, closed}=gen_tcp:recv(Socket, 0, 1000),
  1233. {'EXIT', _}=begin catch ranch:get_port(Name) end,
  1234. ok.
  1235. tcp_upgrade(_) ->
  1236. doc("Ensure that protocol options can be updated."),
  1237. Name = name(),
  1238. {ok, _} = ranch:start_listener(Name,
  1239. ranch_tcp, #{},
  1240. notify_and_wait_protocol, #{pid => self()}),
  1241. Port = ranch:get_port(Name),
  1242. ok = connect_loop(Port, 1, 0),
  1243. {1, Pids1} = receive_loop(connected, 1000),
  1244. ranch:set_protocol_options(Name, #{msg => upgraded, pid => self()}),
  1245. ok = connect_loop(Port, 1, 0),
  1246. {1, Pids2} = receive_loop(upgraded, 1000),
  1247. ok = terminate_loop(stop, Pids1 ++ Pids2),
  1248. ok = ranch:stop_listener(Name).
  1249. tcp_error_eaddrinuse(_) ->
  1250. doc("Ensure that failure due to an eaddrinuse returns a compact readable error."),
  1251. Name = name(),
  1252. {ok, _} = ranch:start_listener(Name,
  1253. ranch_tcp, #{},
  1254. active_echo_protocol, []),
  1255. Port = ranch:get_port(Name),
  1256. {error, eaddrinuse} = ranch:start_listener({Name, fails},
  1257. ranch_tcp, [{port, Port}],
  1258. active_echo_protocol, []),
  1259. ok = ranch:stop_listener(Name),
  1260. %% Make sure the listener stopped.
  1261. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  1262. ok.
  1263. tcp_error_eacces(_) ->
  1264. case os:type() of
  1265. {win32, nt} ->
  1266. {skip, "No privileged ports."};
  1267. {unix, darwin} ->
  1268. {skip, "No privileged ports."};
  1269. _ ->
  1270. doc("Ensure that failure due to an eacces returns a compact readable error."),
  1271. Name = name(),
  1272. {error, eacces} = ranch:start_listener(Name,
  1273. ranch_tcp, [{port, 283}],
  1274. active_echo_protocol, []),
  1275. ok
  1276. end.
  1277. %% Supervisor tests
  1278. connection_type_supervisor(_) ->
  1279. doc("The supervisor connection type must be reflected in the specifications."),
  1280. Name = name(),
  1281. {ok, _} = ranch:start_listener(Name,
  1282. ranch_tcp, #{connection_type => supervisor},
  1283. echo_protocol, []),
  1284. Port = ranch:get_port(Name),
  1285. {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  1286. ok = gen_tcp:send(Socket, <<"TCP Ranch is working!">>),
  1287. {ok, <<"TCP Ranch is working!">>} = gen_tcp:recv(Socket, 21, 1000),
  1288. [{echo_protocol, _, supervisor, [echo_protocol]}] = do_conns_which_children(Name),
  1289. ok = ranch:stop_listener(Name),
  1290. {error, closed} = gen_tcp:recv(Socket, 0, 1000),
  1291. %% Make sure the listener stopped.
  1292. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  1293. ok.
  1294. connection_type_supervisor_separate_from_connection(_) ->
  1295. doc("The supervisor connection type allows separate supervised and connection processes."),
  1296. Name = name(),
  1297. {ok, _} = ranch:start_listener(Name,
  1298. ranch_tcp, #{connection_type => supervisor},
  1299. supervisor_separate, []),
  1300. Port = ranch:get_port(Name),
  1301. {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  1302. ok = gen_tcp:send(Socket, <<"TCP Ranch is working!">>),
  1303. {ok, <<"TCP Ranch is working!">>} = gen_tcp:recv(Socket, 21, 1000),
  1304. [{supervisor_separate, _, supervisor, [supervisor_separate]}] = do_conns_which_children(Name),
  1305. ok = ranch:stop_listener(Name),
  1306. {error, closed} = gen_tcp:recv(Socket, 0, 1000),
  1307. %% Make sure the listener stopped.
  1308. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  1309. ok.
  1310. supervisor_10_acceptors_1_conns_sup(_) ->
  1311. doc("Ensure that using 10 acceptors and 1 connection supervisor works."),
  1312. ok = do_supervisor_n_acceptors_m_conns_sups(10, 1).
  1313. supervisor_9_acceptors_4_conns_sups(_) ->
  1314. doc("Ensure that using 9 acceptors and 4 connection supervisors works."),
  1315. ok = do_supervisor_n_acceptors_m_conns_sups(9, 4).
  1316. supervisor_10_acceptors_10_conns_sups(_) ->
  1317. doc("Ensure that using 10 acceptors and 10 connection supervisors works."),
  1318. ok = do_supervisor_n_acceptors_m_conns_sups(10, 10).
  1319. supervisor_1_acceptor_10_conns_sups(_) ->
  1320. doc("Ensure that using 1 acceptor and 10 connection supervisors works."),
  1321. ok = do_supervisor_n_acceptors_m_conns_sups(1, 10).
  1322. do_supervisor_n_acceptors_m_conns_sups(NumAcceptors, NumConnsSups) ->
  1323. Name = name(),
  1324. {ok, Pid} = ranch:start_listener(Name,
  1325. ranch_tcp, #{num_conns_sups => NumConnsSups, num_acceptors => NumAcceptors},
  1326. notify_and_wait_protocol, #{pid => self()}),
  1327. Port = ranch:get_port(Name),
  1328. ConnsSups = [ConnsSup || {_, ConnsSup} <- ranch_server:get_connections_sups(Name)],
  1329. NumConnsSups = length(ConnsSups),
  1330. {ranch_acceptors_sup, AcceptorsSup, supervisor, _} =
  1331. lists:keyfind(ranch_acceptors_sup, 1, supervisor:which_children(Pid)),
  1332. AcceptorIds = [AcceptorId ||
  1333. {{acceptor, _, AcceptorId}, _, worker, _} <- supervisor:which_children(AcceptorsSup)],
  1334. NumAcceptors = length(AcceptorIds),
  1335. AcceptorConnsSups0 = [ranch_server:get_connections_sup(Name, AcceptorId) ||
  1336. AcceptorId <- AcceptorIds],
  1337. AcceptorConnsSups1 = lists:usort(AcceptorConnsSups0),
  1338. if
  1339. NumAcceptors > NumConnsSups ->
  1340. NumConnsSups = length(AcceptorConnsSups1),
  1341. [] = ConnsSups -- AcceptorConnsSups1;
  1342. NumAcceptors < NumConnsSups ->
  1343. NumAcceptors = length(AcceptorConnsSups1),
  1344. [] = AcceptorConnsSups1 -- ConnsSups;
  1345. NumAcceptors =:= NumConnsSups ->
  1346. NumConnsSups = length(AcceptorConnsSups1),
  1347. NumAcceptors = length(AcceptorConnsSups1),
  1348. [] = ConnsSups -- AcceptorConnsSups1,
  1349. [] = AcceptorConnsSups1 -- ConnsSups
  1350. end,
  1351. ok = connect_loop(Port, 100, 0),
  1352. {100, Pids} = receive_loop(connected, 1000),
  1353. 100 = ranch_server:count_connections(Name),
  1354. ok = terminate_loop(stop, Pids),
  1355. ok = ranch:stop_listener(Name),
  1356. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  1357. ok.
  1358. supervisor_changed_options_restart(_) ->
  1359. doc("Ensure that a listener is restarted with changed transport options."),
  1360. Name = name(),
  1361. %% Start a listener using send_timeout as option change marker.
  1362. {ok, ListenerSupPid1} = ranch:start_listener(Name,
  1363. ranch_tcp, [{send_timeout, 300000}],
  1364. echo_protocol, []),
  1365. %% Ensure send_timeout is really set to initial value.
  1366. {ok, [{send_timeout, 300000}]}
  1367. = inet:getopts(do_get_listener_socket(ListenerSupPid1), [send_timeout]),
  1368. %% Change send_timeout option.
  1369. ok = ranch:suspend_listener(Name),
  1370. ok = ranch:set_transport_options(Name, [{send_timeout, 300001}]),
  1371. ok = ranch:resume_listener(Name),
  1372. %% Ensure send_timeout is really set to the changed value.
  1373. {ok, [{send_timeout, 300001}]}
  1374. = inet:getopts(do_get_listener_socket(ListenerSupPid1), [send_timeout]),
  1375. %% Crash the listener_sup process, allow a short time for restart to succeed.
  1376. %% We silence the expected log events coming from the relevant supervisors.
  1377. ListenerChilds = [ChildPid || {_, ChildPid, _, _} <- supervisor:which_children(ListenerSupPid1)],
  1378. FilterFun = fun (#{meta := #{pid := EventPid}}, _) ->
  1379. case lists:member(EventPid, ListenerChilds) of
  1380. true -> stop;
  1381. false -> ignore
  1382. end
  1383. end,
  1384. ok = logger:add_primary_filter(?MODULE, {FilterFun, undefined}),
  1385. try
  1386. exit(ListenerSupPid1, kill),
  1387. timer:sleep(1000)
  1388. after
  1389. ok = logger:remove_primary_filter(?MODULE)
  1390. end,
  1391. %% Obtain pid of restarted listener_sup process.
  1392. [ListenerSupPid2] = [Pid || {{ranch_listener_sup, Ref}, Pid, supervisor, _}
  1393. <- supervisor:which_children(ranch_sup), Ref =:= Name],
  1394. %% Ensure send_timeout is still set to the changed value.
  1395. {ok, [{send_timeout, 300001}]}
  1396. = inet:getopts(do_get_listener_socket(ListenerSupPid2), [send_timeout]),
  1397. ok = ranch:stop_listener(Name),
  1398. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  1399. ok.
  1400. supervisor_clean_child_restart(_) ->
  1401. doc("Verify that only the relevant parts of the supervision tree restarted "
  1402. "when the listening socket is closed."),
  1403. Name = name(),
  1404. %% Trace socket allocations.
  1405. {module, ranch_tcp} = code:ensure_loaded(ranch_tcp),
  1406. _ = erlang:trace(new, true, [call]),
  1407. 1 = erlang:trace_pattern({ranch_tcp, listen, 1},
  1408. [{'_', [], [{return_trace}]}], [global]),
  1409. {ok, Pid} = ranch:start_listener(Name,
  1410. ranch_tcp, #{num_acceptors => 1},
  1411. echo_protocol, []),
  1412. %% Trace supervisor spawns.
  1413. 1 = erlang:trace(Pid, true, [procs, set_on_spawn]),
  1414. ConnsSups = ranch_server:get_connections_sups(Name),
  1415. %% Manually shut the listening socket down.
  1416. LSocket = receive
  1417. {trace, _, return_from, {ranch_tcp, listen, 1}, {ok, Socket}} ->
  1418. Socket
  1419. after 0 ->
  1420. error(lsocket_unknown)
  1421. end,
  1422. ok = gen_tcp:close(LSocket),
  1423. receive after 1000 -> ok end,
  1424. %% Verify that supervisor and its first two children are alive.
  1425. true = is_process_alive(Pid),
  1426. true = lists:all(fun erlang:is_process_alive/1, [ConnsSup || {_, ConnsSup} <- ConnsSups]),
  1427. %% Check that acceptors_sup is restarted properly.
  1428. AccSupPid = receive {trace, Pid, spawn, Pid1, _} -> Pid1 end,
  1429. receive {trace, AccSupPid, spawn, _, _} -> ok end,
  1430. %% No more traces then.
  1431. receive
  1432. {trace, _, spawn, _, _} -> error(invalid_restart)
  1433. after 1000 -> ok end,
  1434. %% Verify that children still registered right.
  1435. ConnsSups = ranch_server:get_connections_sups(Name),
  1436. _ = erlang:trace_pattern({ranch_tcp, listen, 1}, false, []),
  1437. _ = erlang:trace(all, false, [all]),
  1438. ok = clean_traces(),
  1439. ok = ranch:stop_listener(Name).
  1440. supervisor_clean_restart(_) ->
  1441. doc("Verify that killing ranch_conns_sup does not crash everything "
  1442. "and that it restarts properly."),
  1443. Name = name(),
  1444. NumAcc = 4,
  1445. {ok, Pid} = ranch:start_listener(Name,
  1446. ranch_tcp, #{num_acceptors => NumAcc},
  1447. echo_protocol, []),
  1448. %% Trace supervisor spawns.
  1449. 1 = erlang:trace(Pid, true, [procs, set_on_spawn]),
  1450. {_, ConnsSupSup0, _, _} = lists:keyfind(ranch_conns_sup_sup, 1, supervisor:which_children(Pid)),
  1451. exit(ConnsSupSup0, kill),
  1452. receive after 1000 -> ok end,
  1453. %% Verify that supervisor is alive
  1454. true = is_process_alive(Pid),
  1455. %% ...but children are dead.
  1456. false = is_process_alive(ConnsSupSup0),
  1457. %% Receive traces from newly started children
  1458. ConnsSupSup = receive {trace, Pid, spawn, Pid2, _} -> Pid2 end,
  1459. [receive {trace, ConnsSupSup, spawn, _Pid, _} -> ok end ||
  1460. _ <- lists:seq(1, NumAcc)],
  1461. AccSupPid = receive {trace, Pid, spawn, Pid3, _} -> Pid3 end,
  1462. %% ...and its acceptors.
  1463. [receive {trace, AccSupPid, spawn, _Pid, _} -> ok end ||
  1464. _ <- lists:seq(1, NumAcc)],
  1465. %% No more traces then.
  1466. receive
  1467. {trace, EPid, spawn, _, _} when EPid == Pid; EPid == AccSupPid ->
  1468. error(invalid_restart)
  1469. after 1000 -> ok end,
  1470. %% Verify that new children registered themselves properly.
  1471. _ = erlang:trace(all, false, [all]),
  1472. ok = clean_traces(),
  1473. ok = ranch:stop_listener(Name).
  1474. supervisor_conns_alive(_) ->
  1475. doc("Ensure that active connections stay open when the listening socket gets closed."),
  1476. Name = name(),
  1477. {module, ranch_tcp} = code:ensure_loaded(ranch_tcp),
  1478. _ = erlang:trace(new, true, [call]),
  1479. 1 = erlang:trace_pattern({ranch_tcp, listen, 1},
  1480. [{'_', [], [{return_trace}]}], [global]),
  1481. {ok, _} = ranch:start_listener(Name,
  1482. ranch_tcp, #{},
  1483. remove_conn_and_wait_protocol, [{remove, false, 2500}]),
  1484. %% Get the listener socket
  1485. LSocket = receive
  1486. {trace, _, return_from, {ranch_tcp, listen, 1}, {ok, S}} ->
  1487. S
  1488. after 500 ->
  1489. error(lsocket_unknown)
  1490. end,
  1491. TcpPort = ranch:get_port(Name),
  1492. {ok, Socket} = gen_tcp:connect("localhost", TcpPort,
  1493. [binary, {active, true}, {packet, raw}]),
  1494. receive after 500 -> ok end,
  1495. %% Shut the socket down
  1496. ok = gen_tcp:close(LSocket),
  1497. %% Assert that client is still viable.
  1498. receive {tcp_closed, _} -> error(closed) after 1500 -> ok end,
  1499. ok = gen_tcp:send(Socket, <<"poke">>),
  1500. receive {tcp_closed, _} -> ok end,
  1501. _ = erlang:trace(all, false, [all]),
  1502. ok = clean_traces(),
  1503. ok = ranch:stop_listener(Name).
  1504. supervisor_embedded_ranch_server_crash(_) ->
  1505. doc("Ensure that restarting ranch_server also restarts embedded listeners."),
  1506. Name = name(),
  1507. {ok, SupPid} = embedded_sup:start_link(),
  1508. {ok, EmbeddedSupPid} = embedded_sup:start_listener(SupPid, Name,
  1509. ranch_tcp, #{},
  1510. echo_protocol, []),
  1511. [{{ranch_listener_sup, Name}, ListenerPid, supervisor, _},
  1512. {ranch_server_proxy, ProxyPid, worker, _}] = supervisor:which_children(EmbeddedSupPid),
  1513. ProxyMonitor = monitor(process, ProxyPid),
  1514. ListenerMonitor = monitor(process, ListenerPid),
  1515. ok = supervisor:terminate_child(ranch_sup, ranch_server),
  1516. receive {'DOWN', ProxyMonitor, process, ProxyPid, shutdown} -> ok after 1000 -> exit(timeout) end,
  1517. receive {'DOWN', ListenerMonitor, process, ListenerPid, shutdown} -> ok after 1000 -> exit(timeout) end,
  1518. {ok, _} = supervisor:restart_child(ranch_sup, ranch_server),
  1519. receive after 1000 -> ok end,
  1520. [{{ranch_listener_sup, Name}, _, supervisor, _},
  1521. {ranch_server_proxy, _, worker, _}] = supervisor:which_children(EmbeddedSupPid),
  1522. embedded_sup:stop_listener(SupPid, Name),
  1523. embedded_sup:stop(SupPid),
  1524. ok.
  1525. supervisor_protocol_start_link_crash(_) ->
  1526. doc("Ensure a protocol start crash does not kill all connections."),
  1527. Name = name(),
  1528. {ok, _} = ranch:start_listener(Name,
  1529. ranch_tcp, #{},
  1530. crash_protocol, []),
  1531. ConnsSups = ranch_server:get_connections_sups(Name),
  1532. Port = ranch:get_port(Name),
  1533. {ok, _} = gen_tcp:connect("localhost", Port, [binary, {active, true}, {packet, raw}]),
  1534. receive after 500 -> ok end,
  1535. ConnsSups = ranch_server:get_connections_sups(Name),
  1536. ok = ranch:stop_listener(Name).
  1537. supervisor_server_recover_state(_) ->
  1538. doc("Ensure that when ranch_server crashes and restarts, it recovers "
  1539. "its state and continues monitoring the same processes."),
  1540. Name = name(),
  1541. _ = erlang:trace(new, true, [call]),
  1542. 1 = erlang:trace_pattern({ranch_server, init, 1},
  1543. [{'_', [], [{return_trace}]}], [global]),
  1544. {ok, _} = ranch:start_listener(Name,
  1545. ranch_tcp, #{},
  1546. echo_protocol, []),
  1547. ConnsSups = ranch_server:get_connections_sups(Name),
  1548. ServerPid = erlang:whereis(ranch_server),
  1549. {monitors, Monitors} = erlang:process_info(ServerPid, monitors),
  1550. erlang:exit(ServerPid, kill),
  1551. receive
  1552. {trace, ServerPid2, return_from, {ranch_server, init, 1}, _Result} ->
  1553. {monitors, Monitors2} = erlang:process_info(ServerPid2, monitors),
  1554. %% Check that ranch_server is monitoring the same processes.
  1555. true = (lists:usort(Monitors) == lists:usort(Monitors2))
  1556. after
  1557. 1000 ->
  1558. error(timeout)
  1559. end,
  1560. ConnsSups = ranch_server:get_connections_sups(Name),
  1561. ok = ranch:stop_listener(Name),
  1562. %% Check ranch_server has removed the ranch_conns_sup.
  1563. [] = (catch ranch_server:get_connections_sups(Name)),
  1564. _ = erlang:trace(all, false, [all]),
  1565. ok = clean_traces().
  1566. supervisor_unexpected_message(_) ->
  1567. doc("Ensure the connections supervisor stays alive when it receives "
  1568. "an unexpected message."),
  1569. Name = name(),
  1570. {ok, _} = ranch:start_listener(Name,
  1571. ranch_tcp, #{},
  1572. echo_protocol, []),
  1573. Port = ranch:get_port(Name),
  1574. {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]),
  1575. ok = gen_tcp:send(Socket, <<"TCP Ranch is working!">>),
  1576. {ok, <<"TCP Ranch is working!">>} = gen_tcp:recv(Socket, 21, 1000),
  1577. %% Send the unexpected message to all ranch_conns_sups.
  1578. _ = [ConnSup ! hello || {_, ConnSup} <- ranch_server:get_connections_sups(Name)],
  1579. %% Connection is still up.
  1580. ok = gen_tcp:send(Socket, <<"TCP Ranch is working!">>),
  1581. {ok, <<"TCP Ranch is working!">>} = gen_tcp:recv(Socket, 21, 1000),
  1582. ok = ranch:stop_listener(Name),
  1583. {error, closed} = gen_tcp:recv(Socket, 0, 1000),
  1584. %% Make sure the listener stopped.
  1585. {'EXIT', _} = begin catch ranch:get_port(Name) end,
  1586. ok.
  1587. %% Utility functions.
  1588. connect_loop(_, 0, _) ->
  1589. ok;
  1590. connect_loop(Port, N, Sleep) ->
  1591. {ok, _} = gen_tcp:connect("localhost", Port,
  1592. [binary, {active, false}, {packet, raw}]),
  1593. receive after Sleep -> ok end,
  1594. connect_loop(Port, N - 1, Sleep).
  1595. receive_loop(Message, Timeout) ->
  1596. receive_loop(Message, Timeout, 0, []).
  1597. receive_loop(Message, Timeout, N, Acc) ->
  1598. receive {Pid, Message} ->
  1599. receive_loop(Message, Timeout, N + 1, [Pid|Acc])
  1600. after Timeout ->
  1601. {N, Acc}
  1602. end.
  1603. terminate_loop(_, []) ->
  1604. ok;
  1605. terminate_loop(Message, [Pid|Pids]) ->
  1606. Pid ! Message,
  1607. terminate_loop(Message, Pids).
  1608. clean_traces() ->
  1609. receive
  1610. {trace, _, _, _} ->
  1611. clean_traces();
  1612. {trace, _, _, _, _} ->
  1613. clean_traces()
  1614. after 0 ->
  1615. ok
  1616. end.
  1617. do_get_listener_socket(ListenerSupPid) ->
  1618. [LSocket] = do_get_listener_sockets(ListenerSupPid),
  1619. LSocket.
  1620. do_get_listener_sockets(ListenerSupPid) ->
  1621. [AcceptorsSupPid] = [Pid || {ranch_acceptors_sup, Pid, supervisor, _}
  1622. <- supervisor:which_children(ListenerSupPid)],
  1623. {links, Links} = erlang:process_info(AcceptorsSupPid, links),
  1624. [P || P <- Links, is_port(P)].
  1625. do_conns_which_children(Name) ->
  1626. Conns = [supervisor:which_children(ConnsSup) ||
  1627. {_, ConnsSup} <- ranch_server:get_connections_sups(Name)],
  1628. lists:flatten(Conns).
  1629. do_conns_count_children(Name) ->
  1630. lists:foldl(
  1631. fun
  1632. (Stats, undefined) ->
  1633. Stats;
  1634. (Stats, Acc) ->
  1635. lists:zipwith(
  1636. fun ({K, V1}, {K, V2}) -> {K, V1+V2} end,
  1637. Acc,
  1638. Stats
  1639. )
  1640. end,
  1641. undefined,
  1642. [supervisor:count_children(ConnsSup) ||
  1643. {_, ConnsSup} <- ranch_server:get_connections_sups(Name)]
  1644. ).
  1645. do_os_supports_reuseport() ->
  1646. case {os:type(), os:version()} of
  1647. {{unix, linux}, {Major, _, _}} when Major > 3 -> true;
  1648. {{unix, linux}, {3, Minor, _}} when Minor >= 9 -> true;
  1649. _ -> false
  1650. end.
  1651. do_os_supports_local_sockets() ->
  1652. case os:type() of
  1653. {unix, _} -> true;
  1654. _ -> false
  1655. end.
  1656. do_tempname() ->
  1657. list_to_binary(lists:droplast(os:cmd("mktemp -u"))).
  1658. do_get_ssl_version() ->
  1659. {ok, Vsn} = application:get_key(ssl, vsn),
  1660. Vsns0 = re:split(Vsn, "\\D+", [{return, list}]),
  1661. Vsns1 = lists:map(fun list_to_integer/1, Vsns0),
  1662. case Vsns1 of
  1663. [] -> {0, 0, 0};
  1664. [Major] -> {Major, 0, 0};
  1665. [Major, Minor] -> {Major, Minor, 0};
  1666. [Major, Minor, Patch|_] -> {Major, Minor, Patch}
  1667. end.