With tunnel4_input_afinfo added, IPIP tunnel processing in
ip_vti can be easily done with .cb_handler. So replace the
processing by calling ip_tunnel_rcv() with it.

Signed-off-by: Xin Long <lucien....@gmail.com>
---
 net/ipv4/ip_vti.c | 33 ++++++++-------------------------
 1 file changed, 8 insertions(+), 25 deletions(-)

diff --git a/net/ipv4/ip_vti.c b/net/ipv4/ip_vti.c
index 1d9c8cf..fd762d9 100644
--- a/net/ipv4/ip_vti.c
+++ b/net/ipv4/ip_vti.c
@@ -93,28 +93,10 @@ static int vti_rcv_proto(struct sk_buff *skb)
 
 static int vti_rcv_tunnel(struct sk_buff *skb)
 {
-       struct ip_tunnel_net *itn = net_generic(dev_net(skb->dev), vti_net_id);
-       const struct iphdr *iph = ip_hdr(skb);
-       struct ip_tunnel *tunnel;
-
-       tunnel = ip_tunnel_lookup(itn, skb->dev->ifindex, TUNNEL_NO_KEY,
-                                 iph->saddr, iph->daddr, 0);
-       if (tunnel) {
-               struct tnl_ptk_info tpi = {
-                       .proto = htons(ETH_P_IP),
-               };
-
-               if (!xfrm4_policy_check(NULL, XFRM_POLICY_IN, skb))
-                       goto drop;
-               if (iptunnel_pull_header(skb, 0, tpi.proto, false))
-                       goto drop;
-               return ip_tunnel_rcv(tunnel, skb, &tpi, NULL, false);
-       }
+       XFRM_SPI_SKB_CB(skb)->family = AF_INET;
+       XFRM_SPI_SKB_CB(skb)->daddroff = offsetof(struct iphdr, daddr);
 
-       return -EINVAL;
-drop:
-       kfree_skb(skb);
-       return 0;
+       return vti_input(skb, IPPROTO_IPIP, ip_hdr(skb)->saddr, 0, false);
 }
 
 static int vti_rcv_cb(struct sk_buff *skb, int err)
@@ -495,8 +477,9 @@ static struct xfrm4_protocol vti_ipcomp4_protocol 
__read_mostly = {
        .priority       =       100,
 };
 
-static struct xfrm_tunnel ipip_handler __read_mostly = {
+static struct xfrm_tunnel vti_ipip_handler __read_mostly = {
        .handler        =       vti_rcv_tunnel,
+       .cb_handler     =       vti_rcv_cb,
        .err_handler    =       vti4_err,
        .priority       =       0,
 };
@@ -670,7 +653,7 @@ static int __init vti_init(void)
                goto xfrm_proto_comp_failed;
 
        msg = "ipip tunnel";
-       err = xfrm4_tunnel_register(&ipip_handler, AF_INET);
+       err = xfrm4_tunnel_register(&vti_ipip_handler, AF_INET);
        if (err < 0)
                goto xfrm_tunnel_failed;
 
@@ -682,7 +665,7 @@ static int __init vti_init(void)
        return err;
 
 rtnl_link_failed:
-       xfrm4_tunnel_deregister(&ipip_handler, AF_INET);
+       xfrm4_tunnel_deregister(&vti_ipip_handler, AF_INET);
 xfrm_tunnel_failed:
        xfrm4_protocol_deregister(&vti_ipcomp4_protocol, IPPROTO_COMP);
 xfrm_proto_comp_failed:
@@ -699,7 +682,7 @@ static int __init vti_init(void)
 static void __exit vti_fini(void)
 {
        rtnl_link_unregister(&vti_link_ops);
-       xfrm4_tunnel_deregister(&ipip_handler, AF_INET);
+       xfrm4_tunnel_deregister(&vti_ipip_handler, AF_INET);
        xfrm4_protocol_deregister(&vti_ipcomp4_protocol, IPPROTO_COMP);
        xfrm4_protocol_deregister(&vti_ah4_protocol, IPPROTO_AH);
        xfrm4_protocol_deregister(&vti_esp4_protocol, IPPROTO_ESP);
-- 
2.1.0

Reply via email to