netns xfrm: policy walking in netns

Signed-off-by: Alexey Dobriyan <adobriyan@gmail.com>
Signed-off-by: David S. Miller <davem@davemloft.net>

2 files changed, 4 insertions, 4 deletions

diff --git a/net/xfrm/xfrm_policy.c b/net/xfrm/xfrm_policy.c
index 96895ef61858..6165218fd7c2 100644
--- a/net/xfrm/xfrm_policy.c
+++ b/net/xfrm/xfrm_policy.c
@@ -851,7 +851,7 @@ out:
 }
 EXPORT_SYMBOL(xfrm_policy_flush);
 
-int xfrm_policy_walk(struct xfrm_policy_walk *walk,
+int xfrm_policy_walk(struct net *net, struct xfrm_policy_walk *walk,
                      int (*func)(struct xfrm_policy *, int, int, void*),
                      void *data)
 {
@@ -868,10 +868,10 @@ int xfrm_policy_walk(struct xfrm_policy_walk *walk,
 
         write_lock_bh(&xfrm_policy_lock);
         if (list_empty(&walk->walk.all))
-                x = list_first_entry(&init_net.xfrm.policy_all, struct xfrm_policy_walk_entry, all);
+                x = list_first_entry(&net->xfrm.policy_all, struct xfrm_policy_walk_entry, all);
         else
                 x = list_entry(&walk->walk.all, struct xfrm_policy_walk_entry, all);
-        list_for_each_entry_from(x, &init_net.xfrm.policy_all, all) {
+        list_for_each_entry_from(x, &net->xfrm.policy_all, all) {
                 if (x->dead)
                         continue;
                 pol = container_of(x, struct xfrm_policy, walk);
diff --git a/net/xfrm/xfrm_user.c b/net/xfrm/xfrm_user.c
index efd6ab5c0aca..f6e02726cf1b 100644
--- a/net/xfrm/xfrm_user.c
+++ b/net/xfrm/xfrm_user.c
@@ -1279,7 +1279,7 @@ static int xfrm_dump_policy(struct sk_buff *skb, struct netlink_callback *cb)
                 xfrm_policy_walk_init(walk, XFRM_POLICY_TYPE_ANY);
         }
 
-        (void) xfrm_policy_walk(walk, dump_one_policy, &info);
+        (void) xfrm_policy_walk(&init_net, walk, dump_one_policy, &info);
 
         return skb->len;
 }