This patch is just preparation for MPLS support in ip6_tunnel

Signed-off-by: Vadim Fedorenko <vfedore...@novek.ru>
---
 net/ipv6/tunnel6.c | 96 +++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 92 insertions(+), 4 deletions(-)

diff --git a/net/ipv6/tunnel6.c b/net/ipv6/tunnel6.c
index 21e7b95..1baef7b 100644
--- a/net/ipv6/tunnel6.c
+++ b/net/ipv6/tunnel6.c
@@ -21,6 +21,9 @@
 
 static struct xfrm6_tunnel __rcu *tunnel6_handlers __read_mostly;
 static struct xfrm6_tunnel __rcu *tunnel46_handlers __read_mostly;
+#if IS_ENABLED(CONFIG_MPLS)
+static struct xfrm6_tunnel __rcu *tunnelmpls6_handlers __read_mostly;
+#endif
 static DEFINE_MUTEX(tunnel6_mutex);
 
 int xfrm6_tunnel_register(struct xfrm6_tunnel *handler, unsigned short family)
@@ -32,8 +35,23 @@ int xfrm6_tunnel_register(struct xfrm6_tunnel *handler, 
unsigned short family)
 
        mutex_lock(&tunnel6_mutex);
 
-       for (pprev = (family == AF_INET6) ? &tunnel6_handlers : 
&tunnel46_handlers;
-            (t = rcu_dereference_protected(*pprev,
+       switch (family) {
+       case AF_INET6:
+               pprev = &tunnel6_handlers;
+               break;
+       case AF_INET:
+               pprev = &tunnel46_handlers;
+               break;
+#if IS_ENABLED(CONFIG_MPLS)
+       case AF_MPLS:
+               pprev = &tunnelmpls6_handlers;
+               break;
+#endif
+       default:
+               goto err;
+       }
+
+       for (; (t = rcu_dereference_protected(*pprev,
                        lockdep_is_held(&tunnel6_mutex))) != NULL;
             pprev = &t->next) {
                if (t->priority > priority)
@@ -62,8 +80,23 @@ int xfrm6_tunnel_deregister(struct xfrm6_tunnel *handler, 
unsigned short family)
 
        mutex_lock(&tunnel6_mutex);
 
-       for (pprev = (family == AF_INET6) ? &tunnel6_handlers : 
&tunnel46_handlers;
-            (t = rcu_dereference_protected(*pprev,
+       switch (family) {
+       case AF_INET6:
+               pprev = &tunnel6_handlers;
+               break;
+       case AF_INET:
+               pprev = &tunnel46_handlers;
+               break;
+#if IS_ENABLED(CONFIG_MPLS)
+       case AF_MPLS:
+               pprev = &tunnelmpls6_handlers;
+               break;
+#endif
+       default:
+               goto err;
+       }
+
+       for (; (t = rcu_dereference_protected(*pprev,
                        lockdep_is_held(&tunnel6_mutex))) != NULL;
             pprev = &t->next) {
                if (t == handler) {
@@ -73,6 +106,7 @@ int xfrm6_tunnel_deregister(struct xfrm6_tunnel *handler, 
unsigned short family)
                }
        }
 
+err:
        mutex_unlock(&tunnel6_mutex);
 
        synchronize_net();
@@ -86,6 +120,26 @@ int xfrm6_tunnel_deregister(struct xfrm6_tunnel *handler, 
unsigned short family)
             handler != NULL;                           \
             handler = rcu_dereference(handler->next))  \
 
+#if IS_ENABLED(CONFIG_MPLS)
+static int tunnelmpls6_rcv(struct sk_buff *skb)
+{
+       struct xfrm6_tunnel *handler;
+
+       if (!pskb_may_pull(skb, sizeof(struct ipv6hdr)))
+               goto drop;
+
+       for_each_tunnel_rcu(tunnelmpls6_handlers, handler)
+               if (!handler->handler(skb))
+                       return 0;
+
+       icmpv6_send(skb, ICMPV6_DEST_UNREACH, ICMPV6_PORT_UNREACH, 0);
+
+drop:
+       kfree_skb(skb);
+       return 0;
+}
+#endif
+
 static int tunnel6_rcv(struct sk_buff *skb)
 {
        struct xfrm6_tunnel *handler;
@@ -146,6 +200,20 @@ static int tunnel46_err(struct sk_buff *skb, struct 
inet6_skb_parm *opt,
        return -ENOENT;
 }
 
+#if IS_ENABLED(CONFIG_MPLS)
+static int tunnelmpls6_err(struct sk_buff *skb, struct inet6_skb_parm *opt,
+                          u8 type, u8 code, int offset, __be32 info)
+{
+       struct xfrm6_tunnel *handler;
+
+       for_each_tunnel_rcu(tunnelmpls6_handlers, handler)
+               if (!handler->err_handler(skb, opt, type, code, offset, info))
+                       return 0;
+
+       return -ENOENT;
+}
+#endif
+
 static const struct inet6_protocol tunnel6_protocol = {
        .handler        = tunnel6_rcv,
        .err_handler    = tunnel6_err,
@@ -158,6 +226,14 @@ static int tunnel46_err(struct sk_buff *skb, struct 
inet6_skb_parm *opt,
        .flags          = INET6_PROTO_NOPOLICY|INET6_PROTO_FINAL,
 };
 
+#if IS_ENABLED(CONFIG_MPLS)
+static const struct inet6_protocol tunnelmpls6_protocol = {
+       .handler        = tunnelmpls6_rcv,
+       .err_handler    = tunnelmpls6_err,
+       .flags          = INET6_PROTO_NOPOLICY|INET6_PROTO_FINAL,
+};
+#endif
+
 static int __init tunnel6_init(void)
 {
        if (inet6_add_protocol(&tunnel6_protocol, IPPROTO_IPV6)) {
@@ -169,6 +245,14 @@ static int __init tunnel6_init(void)
                inet6_del_protocol(&tunnel6_protocol, IPPROTO_IPV6);
                return -EAGAIN;
        }
+#if IS_ENABLED(CONFIG_MPLS)
+       if (inet6_add_protocol(&tunnelmpls6_protocol, IPPROTO_MPLS)) {
+               pr_err("%s: can't add protocol\n", __func__);
+               inet6_del_protocol(&tunnel6_protocol, IPPROTO_IPV6);
+               inet6_del_protocol(&tunnel46_protocol, IPPROTO_IPIP);
+               return -EAGAIN;
+       }
+#endif
        return 0;
 }
 
@@ -178,6 +262,10 @@ static void __exit tunnel6_fini(void)
                pr_err("%s: can't remove protocol\n", __func__);
        if (inet6_del_protocol(&tunnel6_protocol, IPPROTO_IPV6))
                pr_err("%s: can't remove protocol\n", __func__);
+#if IS_ENABLED(CONFIG_MPLS)
+       if (inet6_del_protocol(&tunnelmpls6_protocol, IPPROTO_MPLS))
+               pr_err("%s: can't remove protocol\n", __func__);
+#endif
 }
 
 module_init(tunnel6_init);
-- 
1.8.3.1

Reply via email to