Merge branch 'upstream-jeff' of git://git.kernel.org/pub/scm/linux/kernel/git/romieu...
[sfrench/cifs-2.6.git] / net / ipv4 / tcp_diag.c
index c148c1081880a03aa1a7f24a692cb439f7ee3d70..2fbcc7d1b1a057dcba3326e88468455d9a1b778f 100644 (file)
@@ -11,7 +11,6 @@
  *      2 of the License, or (at your option) any later version.
  */
 
-#include <linux/config.h>
 
 #include <linux/module.h>
 #include <linux/inet_diag.h>
@@ -26,8 +25,13 @@ static void tcp_diag_get_info(struct sock *sk, struct inet_diag_msg *r,
        const struct tcp_sock *tp = tcp_sk(sk);
        struct tcp_info *info = _info;
 
-       r->idiag_rqueue = tp->rcv_nxt - tp->copied_seq;
-       r->idiag_wqueue = tp->write_seq - tp->snd_una;
+       if (sk->sk_state == TCP_LISTEN) {
+               r->idiag_rqueue = sk->sk_ack_backlog;
+               r->idiag_wqueue = sk->sk_max_ack_backlog;
+       } else {
+               r->idiag_rqueue = tp->rcv_nxt - tp->copied_seq;
+               r->idiag_wqueue = tp->write_seq - tp->snd_una;
+       }
        if (info != NULL)
                tcp_get_info(sk, info);
 }
@@ -52,3 +56,4 @@ static void __exit tcp_diag_exit(void)
 module_init(tcp_diag_init);
 module_exit(tcp_diag_exit);
 MODULE_LICENSE("GPL");
+MODULE_ALIAS_NET_PF_PROTO_TYPE(PF_NETLINK, NETLINK_INET_DIAG, TCPDIAG_GETSOCK);