1 files changed, 19 insertions, 8 deletions

diff --git a/net/ipv6/udplite.c b/net/ipv6/udplite.c
index 87d4202522ee..706c5c375a01 100644
--- a/net/ipv6/udplite.c
+++ b/net/ipv6/udplite.c
@@ -35,11 +35,6 @@ static struct inet6_protocol udplitev6_protocol = {
         .flags          =       INET6_PROTO_NOPOLICY|INET6_PROTO_FINAL,
 };
 
-static int udplite_v6_get_port(struct sock *sk, unsigned short snum)
-{
-        return udplite_get_port(sk, snum, ipv6_rcv_saddr_equal);
-}
-
 DEFINE_PROTO_INUSE(udplitev6)
 
 struct proto udplitev6_prot = {
@@ -58,8 +53,9 @@ struct proto udplitev6_prot = {
         .backlog_rcv       = udpv6_queue_rcv_skb,
         .hash              = udp_lib_hash,
         .unhash            = udp_lib_unhash,
-        .get_port          = udplite_v6_get_port,
+        .get_port          = udp_v6_get_port,
         .obj_size          = sizeof(struct udp6_sock),
+        .h.udp_hash        = udplite_hash,
 #ifdef CONFIG_COMPAT
         .compat_setsockopt = compat_udpv6_setsockopt,
         .compat_getsockopt = compat_udpv6_getsockopt,
@@ -113,13 +109,28 @@ static struct udp_seq_afinfo udplite6_seq_afinfo = {
         .seq_fops       = &udplite6_seq_fops,
 };
 
+static int udplite6_proc_init_net(struct net *net)
+{
+        return udp_proc_register(net, &udplite6_seq_afinfo);
+}
+
+static void udplite6_proc_exit_net(struct net *net)
+{
+        udp_proc_unregister(net, &udplite6_seq_afinfo);
+}
+
+static struct pernet_operations udplite6_net_ops = {
+        .init = udplite6_proc_init_net,
+        .exit = udplite6_proc_exit_net,
+};
+
 int __init udplite6_proc_init(void)
 {
-        return udp_proc_register(&udplite6_seq_afinfo);
+        return register_pernet_subsys(&udplite6_net_ops);
 }
 
 void udplite6_proc_exit(void)
 {
-        udp_proc_unregister(&udplite6_seq_afinfo);
+        unregister_pernet_subsys(&udplite6_net_ops);
 }
 #endif