net: sctp: implement rfc6458, 8.1.31. SCTP_DEFAULT_SNDINFO support

This patch implements section 8.1.31. of RFC6458, which adds support
for setting/retrieving SCTP_DEFAULT_SNDINFO:

  Applications that wish to use the sendto() system call may wish
  to specify a default set of parameters that would normally be
  supplied through the inclusion of ancillary data. This socket
  option allows such an application to set the default sctp_sndinfo
  structure. The application that wishes to use this socket option
  simply passes the sctp_sndinfo structure (defined in Section 5.3.4)
  to this call. The input parameters accepted by this call include
  snd_sid, snd_flags, snd_ppid, and snd_context. The snd_flags
  parameter is composed of a bitwise OR of SCTP_UNORDERED, SCTP_EOF,
  and SCTP_SENDALL. The snd_assoc_id field specifies the association
  to which to apply the parameters. For a one-to-many style socket,
  any of the predefined constants are also allowed in this field.
  The field is ignored for one-to-one style sockets.

Joint work with Daniel Borkmann.

Signed-off-by: Geir Ola Vaagland <geirola@gmail.com>
Signed-off-by: Daniel Borkmann <dborkman@redhat.com>
Signed-off-by: David S. Miller <davem@davemloft.net>

2 files changed, 99 insertions, 9 deletions

diff --git a/include/uapi/linux/sctp.h b/include/uapi/linux/sctp.h
index 222f82ffeca4..ce70fe6b45df 100644
--- a/include/uapi/linux/sctp.h
+++ b/include/uapi/linux/sctp.h
@@ -97,6 +97,7 @@ typedef __s32 sctp_assoc_t;
 #define SCTP_PEER_ADDR_THLDS    31
 #define SCTP_RECVRCVINFO        32
 #define SCTP_RECVNXTINFO        33
+#define SCTP_DEFAULT_SNDINFO    34
 
 /* Internal Socket Options. Some of the sctp library functions are
  * implemented using these socket options.
diff --git a/net/sctp/socket.c b/net/sctp/socket.c
index 9bca87ee5152..d95a50c013c9 100644
--- a/net/sctp/socket.c
+++ b/net/sctp/socket.c
@@ -2770,19 +2770,22 @@ static int sctp_setsockopt_default_send_param(struct sock *sk,
                                               char __user *optval,
                                               unsigned int optlen)
 {
-        struct sctp_sndrcvinfo info;
-        struct sctp_association *asoc;
         struct sctp_sock *sp = sctp_sk(sk);
+        struct sctp_association *asoc;
+        struct sctp_sndrcvinfo info;
 
-        if (optlen != sizeof(struct sctp_sndrcvinfo))
+        if (optlen != sizeof(info))
                 return -EINVAL;
         if (copy_from_user(&info, optval, optlen))
                 return -EFAULT;
+        if (info.sinfo_flags &
+            ~(SCTP_UNORDERED | SCTP_ADDR_OVER |
+              SCTP_ABORT | SCTP_EOF))
+                return -EINVAL;
 
         asoc = sctp_id2assoc(sk, info.sinfo_assoc_id);
         if (!asoc && info.sinfo_assoc_id && sctp_style(sk, UDP))
                 return -EINVAL;
-
         if (asoc) {
                 asoc->default_stream = info.sinfo_stream;
                 asoc->default_flags = info.sinfo_flags;
@@ -2800,6 +2803,44 @@ static int sctp_setsockopt_default_send_param(struct sock *sk,
         return 0;
 }
 
+/* RFC6458, Section 8.1.31. Set/get Default Send Parameters
+ * (SCTP_DEFAULT_SNDINFO)
+ */
+static int sctp_setsockopt_default_sndinfo(struct sock *sk,
+                                           char __user *optval,
+                                           unsigned int optlen)
+{
+        struct sctp_sock *sp = sctp_sk(sk);
+        struct sctp_association *asoc;
+        struct sctp_sndinfo info;
+
+        if (optlen != sizeof(info))
+                return -EINVAL;
+        if (copy_from_user(&info, optval, optlen))
+                return -EFAULT;
+        if (info.snd_flags &
+            ~(SCTP_UNORDERED | SCTP_ADDR_OVER |
+              SCTP_ABORT | SCTP_EOF))
+                return -EINVAL;
+
+        asoc = sctp_id2assoc(sk, info.snd_assoc_id);
+        if (!asoc && info.snd_assoc_id && sctp_style(sk, UDP))
+                return -EINVAL;
+        if (asoc) {
+                asoc->default_stream = info.snd_sid;
+                asoc->default_flags = info.snd_flags;
+                asoc->default_ppid = info.snd_ppid;
+                asoc->default_context = info.snd_context;
+        } else {
+                sp->default_stream = info.snd_sid;
+                sp->default_flags = info.snd_flags;
+                sp->default_ppid = info.snd_ppid;
+                sp->default_context = info.snd_context;
+        }
+
+        return 0;
+}
+
 /* 7.1.10 Set Primary Address (SCTP_PRIMARY_ADDR)
  *
  * Requests that the local SCTP stack use the enclosed peer address as
@@ -3725,6 +3766,9 @@ static int sctp_setsockopt(struct sock *sk, int level, int optname,
                 retval = sctp_setsockopt_default_send_param(sk, optval,
                                                             optlen);
                 break;
+        case SCTP_DEFAULT_SNDINFO:
+                retval = sctp_setsockopt_default_sndinfo(sk, optval, optlen);
+                break;
         case SCTP_PRIMARY_ADDR:
                 retval = sctp_setsockopt_primary_addr(sk, optval, optlen);
                 break;
@@ -5027,14 +5071,14 @@ static int sctp_getsockopt_default_send_param(struct sock *sk,
                                         int len, char __user *optval,
                                         int __user *optlen)
 {
-        struct sctp_sndrcvinfo info;
-        struct sctp_association *asoc;
         struct sctp_sock *sp = sctp_sk(sk);
+        struct sctp_association *asoc;
+        struct sctp_sndrcvinfo info;
 
-        if (len < sizeof(struct sctp_sndrcvinfo))
+        if (len < sizeof(info))
                 return -EINVAL;
 
-        len = sizeof(struct sctp_sndrcvinfo);
+        len = sizeof(info);
 
         if (copy_from_user(&info, optval, len))
                 return -EFAULT;
@@ -5042,7 +5086,6 @@ static int sctp_getsockopt_default_send_param(struct sock *sk,
         asoc = sctp_id2assoc(sk, info.sinfo_assoc_id);
         if (!asoc && info.sinfo_assoc_id && sctp_style(sk, UDP))
                 return -EINVAL;
-
         if (asoc) {
                 info.sinfo_stream = asoc->default_stream;
                 info.sinfo_flags = asoc->default_flags;
@@ -5065,6 +5108,48 @@ static int sctp_getsockopt_default_send_param(struct sock *sk,
         return 0;
 }
 
+/* RFC6458, Section 8.1.31. Set/get Default Send Parameters
+ * (SCTP_DEFAULT_SNDINFO)
+ */
+static int sctp_getsockopt_default_sndinfo(struct sock *sk, int len,
+                                           char __user *optval,
+                                           int __user *optlen)
+{
+        struct sctp_sock *sp = sctp_sk(sk);
+        struct sctp_association *asoc;
+        struct sctp_sndinfo info;
+
+        if (len < sizeof(info))
+                return -EINVAL;
+
+        len = sizeof(info);
+
+        if (copy_from_user(&info, optval, len))
+                return -EFAULT;
+
+        asoc = sctp_id2assoc(sk, info.snd_assoc_id);
+        if (!asoc && info.snd_assoc_id && sctp_style(sk, UDP))
+                return -EINVAL;
+        if (asoc) {
+                info.snd_sid = asoc->default_stream;
+                info.snd_flags = asoc->default_flags;
+                info.snd_ppid = asoc->default_ppid;
+                info.snd_context = asoc->default_context;
+        } else {
+                info.snd_sid = sp->default_stream;
+                info.snd_flags = sp->default_flags;
+                info.snd_ppid = sp->default_ppid;
+                info.snd_context = sp->default_context;
+        }
+
+        if (put_user(len, optlen))
+                return -EFAULT;
+        if (copy_to_user(optval, &info, len))
+                return -EFAULT;
+
+        return 0;
+}
+
 /*
  *
  * 7.1.5 SCTP_NODELAY
@@ -5924,6 +6009,10 @@ static int sctp_getsockopt(struct sock *sk, int level, int optname,