|
@@ -965,10 +965,20 @@ out:
|
|
|
}
|
|
|
EXPORT_SYMBOL_GPL(rpc_bind_new_program);
|
|
|
|
|
|
+void rpc_task_release_transport(struct rpc_task *task)
|
|
|
+{
|
|
|
+ struct rpc_xprt *xprt = task->tk_xprt;
|
|
|
+
|
|
|
+ if (xprt) {
|
|
|
+ task->tk_xprt = NULL;
|
|
|
+ xprt_put(xprt);
|
|
|
+ }
|
|
|
+}
|
|
|
+EXPORT_SYMBOL_GPL(rpc_task_release_transport);
|
|
|
+
|
|
|
void rpc_task_release_client(struct rpc_task *task)
|
|
|
{
|
|
|
struct rpc_clnt *clnt = task->tk_client;
|
|
|
- struct rpc_xprt *xprt = task->tk_xprt;
|
|
|
|
|
|
if (clnt != NULL) {
|
|
|
/* Remove from client task list */
|
|
@@ -979,12 +989,14 @@ void rpc_task_release_client(struct rpc_task *task)
|
|
|
|
|
|
rpc_release_client(clnt);
|
|
|
}
|
|
|
+ rpc_task_release_transport(task);
|
|
|
+}
|
|
|
|
|
|
- if (xprt != NULL) {
|
|
|
- task->tk_xprt = NULL;
|
|
|
-
|
|
|
- xprt_put(xprt);
|
|
|
- }
|
|
|
+static
|
|
|
+void rpc_task_set_transport(struct rpc_task *task, struct rpc_clnt *clnt)
|
|
|
+{
|
|
|
+ if (!task->tk_xprt)
|
|
|
+ task->tk_xprt = xprt_iter_get_next(&clnt->cl_xpi);
|
|
|
}
|
|
|
|
|
|
static
|
|
@@ -992,8 +1004,7 @@ void rpc_task_set_client(struct rpc_task *task, struct rpc_clnt *clnt)
|
|
|
{
|
|
|
|
|
|
if (clnt != NULL) {
|
|
|
- if (task->tk_xprt == NULL)
|
|
|
- task->tk_xprt = xprt_iter_get_next(&clnt->cl_xpi);
|
|
|
+ rpc_task_set_transport(task, clnt);
|
|
|
task->tk_client = clnt;
|
|
|
atomic_inc(&clnt->cl_count);
|
|
|
if (clnt->cl_softrtry)
|
|
@@ -1512,6 +1523,7 @@ call_start(struct rpc_task *task)
|
|
|
clnt->cl_program->version[clnt->cl_vers]->counts[idx]++;
|
|
|
clnt->cl_stats->rpccnt++;
|
|
|
task->tk_action = call_reserve;
|
|
|
+ rpc_task_set_transport(task, clnt);
|
|
|
}
|
|
|
|
|
|
/*
|