acceptor_SUITE.erl 69 KB

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