2 files changed, 16 insertions, 5 deletions

diff --git a/net/ipv6/datagram.c b/net/ipv6/datagram.c
index 8561b9da6db6..7206747022fc 100644
--- a/net/ipv6/datagram.c
+++ b/net/ipv6/datagram.c
@@ -648,10 +648,13 @@ int datagram_send_ctl(struct msghdr *msg, struct flowi *fl,
 
                         rthdr = (struct ipv6_rt_hdr *)CMSG_DATA(cmsg);
 
-                        /*
-                         *      TYPE 0
-                         */
-                        if (rthdr->type) {
+                        switch (rthdr->type) {
+                        case IPV6_SRCRT_TYPE_0:
+#ifdef CONFIG_IPV6_MIP6
+                        case IPV6_SRCRT_TYPE_2:
+#endif
+                                break;
+                        default:
                                 err = -EINVAL;
                                 goto exit_f;
                         }
diff --git a/net/ipv6/ipv6_sockglue.c b/net/ipv6/ipv6_sockglue.c
index a5eaaf693abf..4f3bb7fcc8b5 100644
--- a/net/ipv6/ipv6_sockglue.c
+++ b/net/ipv6/ipv6_sockglue.c
@@ -407,8 +407,16 @@ static int do_ipv6_setsockopt(struct sock *sk, int level, int optname,
                 /* routing header option needs extra check */
                 if (optname == IPV6_RTHDR && opt->srcrt) {
                         struct ipv6_rt_hdr *rthdr = opt->srcrt;
-                        if (rthdr->type)
+                        switch (rthdr->type) {
+                        case IPV6_SRCRT_TYPE_0:
+#ifdef CONFIG_IPV6_MIP6
+                        case IPV6_SRCRT_TYPE_2:
+#endif
+                                break;
+                        default:
                                 goto sticky_done;
+                        }
+
                         if ((rthdr->hdrlen & 1) ||
                             (rthdr->hdrlen >> 1) != rthdr->segments_left)
                                 goto sticky_done;