From 1c412c62ba3be17b7a818f264049a7ee7942351e Mon Sep 17 00:00:00 2001
From: Micael Karlberg <bmk@erlang.org>
Date: Fri, 28 Sep 2018 18:40:08 +0200
Subject: [socket-nif] Add support for socket (level otp) buffer options

Add support for otp level socket options rcvbuf, rcvctrlbuf and
sndctrlbuf. These options define default sizes for these
buffers.
The 'rcvbuf' is used when receiving messages when calling
the recv, recvfrom and recvmsg functions.
The 'rcvctrlbuf' is used for the control message header info
when calling the recvmsg function.
The 'sndctrlbuf' is used for the control message header info
when calling the sendmsg function.

OTP-14831
---
 lib/kernel/test/socket_SUITE.erl | 112 ++++++++++++++++++++++++++++++++++++++-
 1 file changed, 111 insertions(+), 1 deletion(-)

(limited to 'lib/kernel')

diff --git a/lib/kernel/test/socket_SUITE.erl b/lib/kernel/test/socket_SUITE.erl
index 0255c8ef75..d4cd144168 100644
--- a/lib/kernel/test/socket_SUITE.erl
+++ b/lib/kernel/test/socket_SUITE.erl
@@ -38,6 +38,9 @@
          api_b_send_and_recv_tcp4/1,
          api_b_sendmsg_and_recvmsg_tcp4/1,
 
+         %% API Options
+         api_opt_simple_otp_options/1,
+
          %% API Operation Timeout
          api_to_connect_tcp4/1,
          api_to_connect_tcp6/1,
@@ -90,13 +93,15 @@ all() ->
 groups() -> 
     [{api,                 [], api_cases()},
      {api_basic,           [], api_basic_cases()},
-     {api_op_with_timeout, [], api_op_with_timeout_cases()}
+     {api_op_with_timeout, [], api_op_with_timeout_cases()},
+     {api_options,         [], api_options_cases()}
      %% {tickets,             [], ticket_cases()}
     ].
      
 api_cases() ->
     [
      {group, api_basic},
+     {group, api_options},
      {group, api_op_with_timeout}
     ].
 
@@ -110,6 +115,11 @@ api_basic_cases() ->
      api_b_sendmsg_and_recvmsg_tcp4
     ].
 
+api_options_cases() ->
+    [
+     api_opt_simple_otp_options
+    ].
+
 api_op_with_timeout_cases() ->
     [
      api_to_connect_tcp4,
@@ -431,6 +441,92 @@ api_b_send_and_recv_tcp(Domain, Send, Recv) ->
 
 
 
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+
+%% Perform some simple getopt and setopt with the level = otp options
+api_opt_simple_otp_options(suite) ->
+    [];
+api_opt_simple_otp_options(doc) ->
+    [];
+api_opt_simple_otp_options(_Config) when is_list(_Config) ->
+    tc_begin(api_opt_simple_otp_options),
+
+    p("Create sockets"),
+    S1 = sock_open(inet, stream, tcp),
+    S2 = sock_open(inet, dgram,  udp),
+    
+    Get = fun(S, Key) ->
+                  socket:getopt(S, otp, Key)
+          end,
+    Set = fun(S, Key, Val) ->
+                  socket:setopt(S, otp, Key, Val)
+          end,
+
+    p("Create dummy process"),
+    Pid = spawn_link(fun() -> 
+                             receive
+                                 die -> 
+                                     exit(normal) 
+                             end 
+                     end),
+
+    F = fun(Sock) ->
+                p("Test IOW"),
+                {ok, IOW}    = Get(Sock, iow),
+                NotIOW       = not IOW,
+                ok           = Set(Sock, iow, NotIOW),
+                {ok, NotIOW} = Get(Sock, iow),
+
+                p("Test rcvbuf"),
+                {ok, RcvBuf}  = Get(Sock, rcvbuf),
+                RcvBuf2       = RcvBuf*2,
+                ok            = Set(Sock, rcvbuf, RcvBuf2),
+                {ok, RcvBuf2} = Get(Sock, rcvbuf),
+                ok            = Set(Sock, rcvbuf, default),
+                {ok, RcvBuf}  = Get(Sock, rcvbuf),
+
+                p("Test rcvctrlbuf"),
+                {ok, RcvCtrlBuf}  = Get(Sock, rcvctrlbuf),
+                RcvCtrlBuf2       = RcvCtrlBuf*2,
+                ok                = Set(Sock, rcvctrlbuf, RcvCtrlBuf2),
+                {ok, RcvCtrlBuf2} = Get(Sock, rcvctrlbuf),
+                ok                = Set(Sock, rcvctrlbuf, default),
+                {ok, RcvCtrlBuf}  = Get(Sock, rcvctrlbuf),
+
+                p("Test sndctrlbuf"),
+                {ok, SndCtrlBuf}  = Get(Sock, sndctrlbuf),
+                SndCtrlBuf2       = SndCtrlBuf*2,
+                ok                = Set(Sock, sndctrlbuf, SndCtrlBuf2),
+                {ok, SndCtrlBuf2} = Get(Sock, sndctrlbuf),
+                ok                = Set(Sock, sndctrlbuf, default),
+                {ok, RcvCtrlBuf}  = Get(Sock, sndctrlbuf),
+
+                p("Test controlling-process"),
+                Self = self(),
+                {ok, Self}   = Get(Sock, controlling_process),
+                ok           = Set(Sock, controlling_process, Pid),
+                {ok, Pid}    = Get(Sock, controlling_process)
+
+        end,
+
+    p("Test stream/tcp "),
+    F(S1),
+
+    p("Test dgram/udp "),
+    F(S2),
+
+    p("kill dummy process"),
+    %% This will also close its sockets (S1 and S2),
+    %% This should really be tested explicitly...
+    Pid ! die,
+
+    %% p("close sockets"),
+    %% sock_close(S1),
+    %% sock_close(S2),
+
+    tc_end().
+
+
 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
 
 %% This test case is intended to test the connect timeout option
@@ -1067,6 +1163,20 @@ sock_accept(LSock) ->
     end.
 
 
+sock_close(Sock) ->
+    try socket:close(Sock) of
+        ok ->
+            ok;
+        {error, Reason} ->
+            p("sock_close -> error: ~p", [Reason]),
+            ?FAIL({close, Reason})
+    catch
+        C:E:S ->
+            p("sock_close -> failed: ~p, ~p, ~p", [C, E, S]),
+            ?FAIL({close, C, E, S})
+    end.
+
+
 
 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
 
-- 
cgit v1.2.3