SUNRPC: Add RPC task and client level options to disable the resend timeout

Signed-off-by: Trond Myklebust <Trond.Myklebust@netapp.com>

2 files changed, 16 insertions, 4 deletions

diff --git a/net/sunrpc/clnt.c b/net/sunrpc/clnt.c
index fa50e42aabd3..b58525009e40 100644
--- a/net/sunrpc/clnt.c
+++ b/net/sunrpc/clnt.c
@@ -772,6 +772,8 @@ void rpc_task_set_client(struct rpc_task *task, struct rpc_clnt *clnt)
                 atomic_inc(&clnt->cl_count);
                 if (clnt->cl_softrtry)
                         task->tk_flags |= RPC_TASK_SOFT;
+                if (clnt->cl_noretranstimeo)
+                        task->tk_flags |= RPC_TASK_NO_RETRANS_TIMEOUT;
                 if (sk_memalloc_socks()) {
                         struct rpc_xprt *xprt;
 
@@ -1898,7 +1900,8 @@ call_status(struct rpc_task *task)
                 rpc_delay(task, 3*HZ);
         case -ETIMEDOUT:
                 task->tk_action = call_timeout;
-                if (task->tk_client->cl_discrtry)
+                if (!(task->tk_flags & RPC_TASK_NO_RETRANS_TIMEOUT)
+                    && task->tk_client->cl_discrtry)
                         xprt_conditional_disconnect(req->rq_xprt,
                                         req->rq_connect_cookie);
                 break;
diff --git a/net/sunrpc/xprt.c b/net/sunrpc/xprt.c
index 2326af57b9b9..d166d9947e36 100644
--- a/net/sunrpc/xprt.c
+++ b/net/sunrpc/xprt.c
@@ -873,9 +873,18 @@ bool xprt_prepare_transmit(struct rpc_task *task)
         dprintk("RPC: %5u xprt_prepare_transmit\n", task->tk_pid);
 
         spin_lock_bh(&xprt->transport_lock);
-        if (req->rq_reply_bytes_recvd && !req->rq_bytes_sent) {
-                task->tk_status = req->rq_reply_bytes_recvd;
-                goto out_unlock;
+        if (!req->rq_bytes_sent) {
+                if (req->rq_reply_bytes_recvd) {
+                        task->tk_status = req->rq_reply_bytes_recvd;
+                        goto out_unlock;
+                }
+                if ((task->tk_flags & RPC_TASK_NO_RETRANS_TIMEOUT)
+                    && xprt_connected(xprt)
+                    && req->rq_connect_cookie == xprt->connect_cookie) {
+                        xprt->ops->set_retrans_timeout(task);
+                        rpc_sleep_on(&xprt->pending, task, xprt_timer);
+                        goto out_unlock;
+                }
         }
         if (!xprt->ops->reserve_xprt(xprt, task)) {
                 task->tk_status = -EAGAIN;