SUNRPC: Fix up socket polling

Ensure that we do not exit the socket read callback without clearing
XPRT_SOCK_DATA_READY.

Signed-off-by: Trond Myklebust <trond.myklebust@hammerspace.com>

1 files changed, 2 insertions, 2 deletions

diff --git a/net/sunrpc/xprtsock.c b/net/sunrpc/xprtsock.c
index fc6d129401ba..92d08be2384d 100644
--- a/net/sunrpc/xprtsock.c
+++ b/net/sunrpc/xprtsock.c
@@ -649,9 +649,9 @@ static void xs_stream_data_receive(struct sock_xprt *transport)
         ssize_t ret = 0;
 
         mutex_lock(&transport->recv_mutex);
+        clear_bit(XPRT_SOCK_DATA_READY, &transport->sock_state);
         if (transport->sock == NULL)
                 goto out;
-        clear_bit(XPRT_SOCK_DATA_READY, &transport->sock_state);
         for (;;) {
                 ret = xs_read_stream(transport, MSG_DONTWAIT);
                 if (ret < 0)
@@ -1346,10 +1346,10 @@ static void xs_udp_data_receive(struct sock_xprt *transport)
         int err;
 
         mutex_lock(&transport->recv_mutex);
+        clear_bit(XPRT_SOCK_DATA_READY, &transport->sock_state);
         sk = transport->inet;
         if (sk == NULL)
                 goto out;
-        clear_bit(XPRT_SOCK_DATA_READY, &transport->sock_state);
         for (;;) {
                 skb = skb_recv_udp(sk, 0, 1, &err);
                 if (skb == NULL)