Add otpbp_gen_sctp with connect/3, connect_init/3

Ledest · Oct 26, 2023 · 17659c9 · 17659c9
1 parent 2921864
commit 17659c9
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 0 deletions.
diff --git a/rebar.config.script b/rebar.config.script
@@ -191,6 +191,8 @@ Funs = [{application, get_supervisor, 1},
         {gen_event, start_monitor, 0},
         {gen_event, start_monitor, 1},
         {gen_event, start_monitor, 2},
+        {gen_sctp, connect, 3},
+        {gen_sctp, connect_init, 3},
         {gen_server, start_monitor, 3},
         {gen_server, start_monitor, 4},
         {gen_statem, enter_loop, 4},
@@ -205,6 +207,7 @@ Funs = [{application, get_supervisor, 1},
         {httpd_util, integer_to_hexlist, 1},
         {httpd_util, strip, 1},
         {httpd_util, suffix, 1},
+        {inet, ensure_sockaddr, 1},
         {inet, info, 1},
         {inet, ipv4_mapped_ipv6_address, 1},
         {io_lib, format, 3},
@@ -272,6 +275,7 @@ Funs = [{application, get_supervisor, 1},
         {pg2, start, 0}, % OTP < 24.0
         {pg2, start_link, 0}, % OTP < 24.0
         {pg2, which_groups, 0}, % OTP < 24.0
+        {prim_socket, enc_sockaddr, 1},
         {proc_lib, init_fail, 2},
         {proc_lib, init_fail, 3},
         {proc_lib, start_monitor, 3},

diff --git a/src/otpbp_gen_sctp.erl b/src/otpbp_gen_sctp.erl
@@ -0,0 +1,31 @@
+-module(otpbp_gen_sctp).
+
+-compile({parse_transform, otpbp_pt}).
+
+-ifndef(HAVE_gen_sctp__connect_3).
+% OTP 24.3
+-export([connect/3]).
+-endif.
+-ifndef(HAVE_gen_sctp__connect_init_3).
+% OTP 24.3
+-export([connect_init/3]).
+-endif.
+
+-ifndef(HAVE_gen_sctp__connect_3).
+connect(S, SockAddr, Opts) when is_port(S), is_map(SockAddr), is_list(Opts) ->
+    case inet_db:lookup_socket(S) of
+        {ok, Mod} -> Mod:connect(S, inet:ensure_sockaddr(SockAddr), Opts, false);
+        Error -> Error
+    end;
+connect(S, SockAddr, Opts) -> error(badarg, [S, SockAddr, Opts, infinity]).
+-endif.
+
+-ifndef(HAVE_gen_sctp__connect_init_3).
+connect_init(S, SockAddr, Opts) when is_port(S), is_map(SockAddr), is_list(Opts) ->
+    case inet_db:lookup_socket(S) of
+        {ok, Mod} -> Mod:connect(S, inet:ensure_sockaddr(SockAddr), Opts, nowait);
+        Error -> Error
+    end;
+connect_init(S, SockAddr, Opts) -> error(badarg, [S, SockAddr, Opts, infinity]).
+-endif.
+
diff --git a/src/otpbp_pt.erl b/src/otpbp_pt.erl
@@ -137,6 +137,7 @@
                               {{gen, debug_options, 2}, otpbp_gen},
                               {{gen_event, [start, start_link], 2}, otpbp_gen_event},
                               {{gen_event, start_monitor, [0, 1, 2]}, otpbp_gen_event},
+                              {{geb_sctp, [connect, connect_init], 3}, otpbp_gen_sctp},
                               {{gen_server, start_monitor, [3, 4]}, otpbp_gen_server},
                               {{gen_statem, enter_loop, 4}, otpbp_gen_statem},
                               {{inet, [ensure_sockaddr, info, ipv4_mapped_ipv6_address], 1}, otpbp_inet},