X-Git-Url: http://pileus.org/git/?a=blobdiff_plain;f=net%2Fipv4%2Finet_diag.c;h=56a964a553d2c739a03b880acd7158a4c9714b66;hb=c1d607cc4a8ea1ef89d7f6f5728112bc5a52f2f6;hp=41e1c3ea8b517778666fac9bf89cf903abf6f833;hpb=d7a20c86404bb423f843d4fe34cd1d11432892a0;p=~andy%2Flinux

diff --git a/net/ipv4/inet_diag.c b/net/ipv4/inet_diag.c
index 41e1c3ea8b5..56a964a553d 100644
--- a/net/ipv4/inet_diag.c
+++ b/net/ipv4/inet_diag.c
@@ -336,12 +336,9 @@ int inet_diag_dump_one_icsk(struct inet_hashinfo *hashinfo, struct sk_buff *in_s
 		err = 0;
 
 out:
-	if (sk) {
-		if (sk->sk_state == TCP_TIME_WAIT)
-			inet_twsk_put((struct inet_timewait_sock *)sk);
-		else
-			sock_put(sk);
-	}
+	if (sk)
+		sock_gen_put(sk);
+
 out_nosk:
 	return err;
 }