From: Eric W Biederman <ebied...@xmission.com>

Inspired-by: Pablo Neira Ayuso <pa...@netfilter.org>
Signed-off-by: Eric W Biederman <ebied...@xmission.com>
---
 net/ipv4/netfilter/ipt_SYNPROXY.c  | 24 ++++++++++++++++++++----
 net/ipv6/netfilter/ip6t_SYNPROXY.c | 24 ++++++++++++++++++++----
 2 files changed, 40 insertions(+), 8 deletions(-)

diff --git a/net/ipv4/netfilter/ipt_SYNPROXY.c 
b/net/ipv4/netfilter/ipt_SYNPROXY.c
index 301bb886a289..cb84e4e24175 100644
--- a/net/ipv4/netfilter/ipt_SYNPROXY.c
+++ b/net/ipv4/netfilter/ipt_SYNPROXY.c
@@ -446,12 +446,28 @@ static struct nf_hook_ops ipv4_synproxy_ops[] 
__read_mostly = {
        },
 };
 
+static int synproxy_tg4_net_init(struct net *net)
+{
+       return nf_register_hooks(net, ipv4_synproxy_ops,
+                                ARRAY_SIZE(ipv4_synproxy_ops));
+}
+
+static void synproxy_tg4_net_exit(struct net *net)
+{
+       nf_unregister_hooks(net, ipv4_synproxy_ops,
+                           ARRAY_SIZE(ipv4_synproxy_ops));
+}
+
+static struct pernet_operations synproxy_tg4_net_ops = {
+       .init = synproxy_tg4_net_init,
+       .exit = synproxy_tg4_net_exit,
+};
+
 static int __init synproxy_tg4_init(void)
 {
        int err;
 
-       err = nf_register_hooks(&init_net, ipv4_synproxy_ops,
-                               ARRAY_SIZE(ipv4_synproxy_ops));
+       err = register_pernet_subsys(&synproxy_tg4_net_ops);
        if (err < 0)
                goto err1;
 
@@ -462,7 +478,7 @@ static int __init synproxy_tg4_init(void)
        return 0;
 
 err2:
-       nf_unregister_hooks(&init_net, ipv4_synproxy_ops, 
ARRAY_SIZE(ipv4_synproxy_ops));
+       unregister_pernet_subsys(&synproxy_tg4_net_ops);
 err1:
        return err;
 }
@@ -470,7 +486,7 @@ err1:
 static void __exit synproxy_tg4_exit(void)
 {
        xt_unregister_target(&synproxy_tg4_reg);
-       nf_unregister_hooks(&init_net, ipv4_synproxy_ops, 
ARRAY_SIZE(ipv4_synproxy_ops));
+       unregister_pernet_subsys(&synproxy_tg4_net_ops);
 }
 
 module_init(synproxy_tg4_init);
diff --git a/net/ipv6/netfilter/ip6t_SYNPROXY.c 
b/net/ipv6/netfilter/ip6t_SYNPROXY.c
index 320521086b5c..0acc786fd3f0 100644
--- a/net/ipv6/netfilter/ip6t_SYNPROXY.c
+++ b/net/ipv6/netfilter/ip6t_SYNPROXY.c
@@ -469,12 +469,28 @@ static struct nf_hook_ops ipv6_synproxy_ops[] 
__read_mostly = {
        },
 };
 
+static int synproxy_tg6_net_init(struct net *net)
+{
+       return nf_register_hooks(net, ipv6_synproxy_ops,
+                                ARRAY_SIZE(ipv6_synproxy_ops));
+}
+
+static void synproxy_tg6_net_exit(struct net *net)
+{
+       nf_unregister_hooks(net, ipv6_synproxy_ops,
+                           ARRAY_SIZE(ipv6_synproxy_ops));
+}
+
+static struct pernet_operations synproxy_tg6_net_ops = {
+       .init = synproxy_tg6_net_init,
+       .exit = synproxy_tg6_net_exit,
+};
+
 static int __init synproxy_tg6_init(void)
 {
        int err;
 
-       err = nf_register_hooks(&init_net, ipv6_synproxy_ops,
-                               ARRAY_SIZE(ipv6_synproxy_ops));
+       err = register_pernet_subsys(&synproxy_tg6_net_ops);
        if (err < 0)
                goto err1;
 
@@ -485,7 +501,7 @@ static int __init synproxy_tg6_init(void)
        return 0;
 
 err2:
-       nf_unregister_hooks(&init_net, ipv6_synproxy_ops, 
ARRAY_SIZE(ipv6_synproxy_ops));
+       unregister_pernet_subsys(&synproxy_tg6_net_ops);
 err1:
        return err;
 }
@@ -493,7 +509,7 @@ err1:
 static void __exit synproxy_tg6_exit(void)
 {
        xt_unregister_target(&synproxy_tg6_reg);
-       nf_unregister_hooks(&init_net, ipv6_synproxy_ops, 
ARRAY_SIZE(ipv6_synproxy_ops));
+       unregister_pernet_subsys(&synproxy_tg6_net_ops);
 }
 
 module_init(synproxy_tg6_init);
-- 
2.2.1

--
To unsubscribe from this list: send the line "unsubscribe netdev" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html

Reply via email to