Commit fc0d026a authored by Eric Dumazet's avatar Eric Dumazet Committed by Pablo Neira Ayuso

netfilter: nf_nat_masquerade: add netns refcount tracker to masq_dev_work

If compiled with CONFIG_NET_NS_REFCNT_TRACKER=y,
using put_net_track() in iterate_cleanup_work()
and netns_tracker_alloc() in nf_nat_masq_schedule()
might help us finding netns refcount imbalances.
Signed-off-by: default avatarEric Dumazet <edumazet@google.com>
Signed-off-by: default avatarPablo Neira Ayuso <pablo@netfilter.org>
parent a9382d93
...@@ -12,6 +12,7 @@ ...@@ -12,6 +12,7 @@
struct masq_dev_work { struct masq_dev_work {
struct work_struct work; struct work_struct work;
struct net *net; struct net *net;
netns_tracker ns_tracker;
union nf_inet_addr addr; union nf_inet_addr addr;
int ifindex; int ifindex;
int (*iter)(struct nf_conn *i, void *data); int (*iter)(struct nf_conn *i, void *data);
...@@ -82,7 +83,7 @@ static void iterate_cleanup_work(struct work_struct *work) ...@@ -82,7 +83,7 @@ static void iterate_cleanup_work(struct work_struct *work)
nf_ct_iterate_cleanup_net(w->net, w->iter, (void *)w, 0, 0); nf_ct_iterate_cleanup_net(w->net, w->iter, (void *)w, 0, 0);
put_net(w->net); put_net_track(w->net, &w->ns_tracker);
kfree(w); kfree(w);
atomic_dec(&masq_worker_count); atomic_dec(&masq_worker_count);
module_put(THIS_MODULE); module_put(THIS_MODULE);
...@@ -119,6 +120,7 @@ static void nf_nat_masq_schedule(struct net *net, union nf_inet_addr *addr, ...@@ -119,6 +120,7 @@ static void nf_nat_masq_schedule(struct net *net, union nf_inet_addr *addr,
INIT_WORK(&w->work, iterate_cleanup_work); INIT_WORK(&w->work, iterate_cleanup_work);
w->ifindex = ifindex; w->ifindex = ifindex;
w->net = net; w->net = net;
netns_tracker_alloc(net, &w->ns_tracker, gfp_flags);
w->iter = iter; w->iter = iter;
if (addr) if (addr)
w->addr = *addr; w->addr = *addr;
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment