[TCP]: Replace struct net on tcp_iter_state with seq_net_private.

Signed-off-by: Denis V. Lunev <den@openvz.org>
Signed-off-by: David S. Miller <davem@davemloft.net>

2 files changed, 6 insertions, 7 deletions

diff --git a/include/net/tcp.h b/include/net/tcp.h
index 7b41bb962b9c..f5b61e0f01f2 100644
--- a/include/net/tcp.h
+++ b/include/net/tcp.h
@@ -1333,7 +1333,7 @@ struct tcp_seq_afinfo {
 };
 
 struct tcp_iter_state {
-        struct net              *net;
+        struct seq_net_private  p;
         sa_family_t             family;
         enum tcp_seq_states     state;
         struct sock             *syn_wait_sk;
diff --git a/net/ipv4/tcp_ipv4.c b/net/ipv4/tcp_ipv4.c
index 52e3ae603ca9..86148cdfb21f 100644
--- a/net/ipv4/tcp_ipv4.c
+++ b/net/ipv4/tcp_ipv4.c
@@ -1954,7 +1954,7 @@ static void *listening_get_next(struct seq_file *seq, void *cur)
         struct hlist_node *node;
         struct sock *sk = cur;
         struct tcp_iter_state* st = seq->private;
-        struct net *net = st->net;
+        struct net *net = seq_file_net(seq);
 
         if (!sk) {
                 st->bucket = 0;
@@ -2035,7 +2035,7 @@ static void *listening_get_idx(struct seq_file *seq, loff_t *pos)
 static void *established_get_first(struct seq_file *seq)
 {
         struct tcp_iter_state* st = seq->private;
-        struct net *net = st->net;
+        struct net *net = seq_file_net(seq);
         void *rc = NULL;
 
         for (st->bucket = 0; st->bucket < tcp_hashinfo.ehash_size; ++st->bucket) {
@@ -2076,7 +2076,7 @@ static void *established_get_next(struct seq_file *seq, void *cur)
         struct inet_timewait_sock *tw;
         struct hlist_node *node;
         struct tcp_iter_state* st = seq->private;
-        struct net *net = st->net;
+        struct net *net = seq_file_net(seq);
 
         ++st->num;
 
@@ -2233,7 +2233,7 @@ static int tcp_seq_open(struct inode *inode, struct file *file)
         s->seq_ops.next         = tcp_seq_next;
         s->seq_ops.show         = afinfo->seq_show;
         s->seq_ops.stop         = tcp_seq_stop;
-        s->net                  = net;
+        s->p.net                = net;
 
         rc = seq_open(file, &s->seq_ops);
         if (rc)
@@ -2252,9 +2252,8 @@ out_kfree:
 static int tcp_seq_release(struct inode *inode, struct file *file)
 {
         struct seq_file *seq = file->private_data;
-        struct tcp_iter_state *s = seq->private;
 
-        put_net(s->net);
+        put_net(seq_file_net(seq));
         seq_release_private(inode, file);
         return 0;
 }