acceptor_SUITE.erl 70 KB

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