1 files changed, 62 insertions, 0 deletions

diff --git a/net/sctp/socket.c b/net/sctp/socket.c
index c03fe1b76706..c3167c43a9c1 100644
--- a/net/sctp/socket.c
+++ b/net/sctp/socket.c
@@ -6330,6 +6330,64 @@ out:
         return retval;
 }
 
+static int sctp_getsockopt_pr_assocstatus(struct sock *sk, int len,
+                                          char __user *optval,
+                                          int __user *optlen)
+{
+        struct sctp_prstatus params;
+        struct sctp_association *asoc;
+        int policy;
+        int retval = -EINVAL;
+
+        if (len < sizeof(params))
+                goto out;
+
+        len = sizeof(params);
+        if (copy_from_user(&params, optval, len)) {
+                retval = -EFAULT;
+                goto out;
+        }
+
+        policy = params.sprstat_policy;
+        if (policy & ~SCTP_PR_SCTP_MASK)
+                goto out;
+
+        asoc = sctp_id2assoc(sk, params.sprstat_assoc_id);
+        if (!asoc)
+                goto out;
+
+        if (policy == SCTP_PR_SCTP_NONE) {
+                params.sprstat_abandoned_unsent = 0;
+                params.sprstat_abandoned_sent = 0;
+                for (policy = 0; policy <= SCTP_PR_INDEX(MAX); policy++) {
+                        params.sprstat_abandoned_unsent +=
+                                asoc->abandoned_unsent[policy];
+                        params.sprstat_abandoned_sent +=
+                                asoc->abandoned_sent[policy];
+                }
+        } else {
+                params.sprstat_abandoned_unsent =
+                        asoc->abandoned_unsent[__SCTP_PR_INDEX(policy)];
+                params.sprstat_abandoned_sent =
+                        asoc->abandoned_sent[__SCTP_PR_INDEX(policy)];
+        }
+
+        if (put_user(len, optlen)) {
+                retval = -EFAULT;
+                goto out;
+        }
+
+        if (copy_to_user(optval, &params, len)) {
+                retval = -EFAULT;
+                goto out;
+        }
+
+        retval = 0;
+
+out:
+        return retval;
+}
+
 static int sctp_getsockopt(struct sock *sk, int level, int optname,
                            char __user *optval, int __user *optlen)
 {
@@ -6490,6 +6548,10 @@ static int sctp_getsockopt(struct sock *sk, int level, int optname,
                 retval = sctp_getsockopt_default_prinfo(sk, len, optval,
                                                         optlen);
                 break;
+        case SCTP_PR_ASSOC_STATUS:
+                retval = sctp_getsockopt_pr_assocstatus(sk, len, optval,
+                                                        optlen);
+                break;
         default:
                 retval = -ENOPROTOOPT;
                 break;