1 files changed, 8 insertions, 3 deletions

diff --git a/net/ipv6/udp_offload.c b/net/ipv6/udp_offload.c
index 2b0fbe6929e8..5429f6bcf047 100644
--- a/net/ipv6/udp_offload.c
+++ b/net/ipv6/udp_offload.c
@@ -153,7 +153,7 @@ static struct sk_buff **udp6_gro_receive(struct sk_buff **head,
 
 skip:
         NAPI_GRO_CB(skb)->is_ipv6 = 1;
-        return udp_gro_receive(head, skb, uh);
+        return udp_gro_receive(head, skb, uh, udp6_lib_lookup_skb);
 
 flush:
         NAPI_GRO_CB(skb)->flush = 1;
@@ -173,7 +173,7 @@ static int udp6_gro_complete(struct sk_buff *skb, int nhoff)
                 skb_shinfo(skb)->gso_type |= SKB_GSO_UDP_TUNNEL;
         }
 
-        return udp_gro_complete(skb, nhoff);
+        return udp_gro_complete(skb, nhoff, udp6_lib_lookup_skb);
 }
 
 static const struct net_offload udpv6_offload = {
@@ -184,7 +184,12 @@ static const struct net_offload udpv6_offload = {
         },
 };
 
-int __init udp_offload_init(void)
+int udpv6_offload_init(void)
 {
         return inet6_add_offload(&udpv6_offload, IPPROTO_UDP);
 }
+
+int udpv6_offload_exit(void)
+{
+        return inet6_del_offload(&udpv6_offload, IPPROTO_UDP);
+}