ip6mr: Add sizeof verification to MRT6_ASSERT and MT6_PIM

Verify the length of the user-space arguments.

Signed-off-by: Joe Perches <joe@perches.com>
Signed-off-by: David S. Miller <davem@davemloft.net>

1 files changed, 6 insertions, 0 deletions

diff --git a/net/ipv6/ip6mr.c b/net/ipv6/ip6mr.c
index 79bb49061ef6..926ea544f499 100644
--- a/net/ipv6/ip6mr.c
+++ b/net/ipv6/ip6mr.c
@@ -1646,6 +1646,9 @@ int ip6_mroute_setsockopt(struct sock *sk, int optname, char __user *optval, uns
         case MRT6_ASSERT:
         {
                 int v;
+
+                if (optlen != sizeof(v))
+                        return -EINVAL;
                 if (get_user(v, (int __user *)optval))
                         return -EFAULT;
                 mrt->mroute_do_assert = v;
@@ -1656,6 +1659,9 @@ int ip6_mroute_setsockopt(struct sock *sk, int optname, char __user *optval, uns
         case MRT6_PIM:
         {
                 int v;
+
+                if (optlen != sizeof(v))
+                        return -EINVAL;
                 if (get_user(v, (int __user *)optval))
                         return -EFAULT;
                 v = !!v;