[PATCH net 2/6] mptcp: fix poll after shutdown

[Mat Martineau]

From: Paolo Abeni <pab...@redhat.com>

The current mptcp_poll() implementation gives unexpected
results after shutdown(SEND_SHUTDOWN) and when the msk
status is TCP_CLOSE.

Set the correct mask.

Fixes: 8edf08649eed ("mptcp: rework poll+nospace handling")
Signed-off-by: Paolo Abeni <pab...@redhat.com>
Signed-off-by: Mat Martineau <mathew.j.martin...@linux.intel.com>
---
 net/mptcp/protocol.c | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/net/mptcp/protocol.c b/net/mptcp/protocol.c
index 9eecd1383d24..42ca49954bdd 100644
--- a/net/mptcp/protocol.c
+++ b/net/mptcp/protocol.c
@@ -3321,7 +3321,7 @@ static __poll_t mptcp_check_writeable(struct mptcp_sock 
*msk)
        struct sock *sk = (struct sock *)msk;
 
        if (unlikely(sk->sk_shutdown & SEND_SHUTDOWN))
-               return 0;
+               return EPOLLOUT | EPOLLWRNORM;
 
        if (sk_stream_is_writeable(sk))
                return EPOLLOUT | EPOLLWRNORM;
@@ -3354,6 +3354,8 @@ static __poll_t mptcp_poll(struct file *file, struct 
socket *sock,
                mask |= mptcp_check_readable(msk);
                mask |= mptcp_check_writeable(msk);
        }
+       if (sk->sk_shutdown == SHUTDOWN_MASK || state == TCP_CLOSE)
+               mask |= EPOLLHUP;
        if (sk->sk_shutdown & RCV_SHUTDOWN)
                mask |= EPOLLIN | EPOLLRDNORM | EPOLLRDHUP;
 
-- 
2.30.1