xprt.c 40.5 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11 12
/*
 *  linux/net/sunrpc/xprt.c
 *
 *  This is a generic RPC call interface supporting congestion avoidance,
 *  and asynchronous calls.
 *
 *  The interface works like this:
 *
 *  -	When a process places a call, it allocates a request slot if
 *	one is available. Otherwise, it sleeps on the backlog queue
 *	(xprt_reserve).
 *  -	Next, the caller puts together the RPC message, stuffs it into
13 14
 *	the request struct, and calls xprt_transmit().
 *  -	xprt_transmit sends the message and installs the caller on the
15 16 17
 *	transport's wait list. At the same time, if a reply is expected,
 *	it installs a timer that is run after the packet's timeout has
 *	expired.
L
Linus Torvalds 已提交
18
 *  -	When a packet arrives, the data_ready handler walks the list of
19
 *	pending requests for that transport. If a matching XID is found, the
L
Linus Torvalds 已提交
20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35
 *	caller is woken up, and the timer removed.
 *  -	When no reply arrives within the timeout interval, the timer is
 *	fired by the kernel and runs xprt_timer(). It either adjusts the
 *	timeout values (minor timeout) or wakes up the caller with a status
 *	of -ETIMEDOUT.
 *  -	When the caller receives a notification from RPC that a reply arrived,
 *	it should release the RPC slot, and process the reply.
 *	If the call timed out, it may choose to retry the operation by
 *	adjusting the initial timeout value, and simply calling rpc_call
 *	again.
 *
 *  Support for async RPC is done through a set of RPC-specific scheduling
 *  primitives that `transparently' work for processes as well as async
 *  tasks that rely on callbacks.
 *
 *  Copyright (C) 1995-1997, Olaf Kirch <okir@monad.swb.de>
36 37
 *
 *  Transport switch API copyright (C) 2005, Chuck Lever <cel@netapp.com>
L
Linus Torvalds 已提交
38 39
 */

40 41
#include <linux/module.h>

L
Linus Torvalds 已提交
42
#include <linux/types.h>
43
#include <linux/interrupt.h>
L
Linus Torvalds 已提交
44
#include <linux/workqueue.h>
45
#include <linux/net.h>
46
#include <linux/ktime.h>
L
Linus Torvalds 已提交
47

48
#include <linux/sunrpc/clnt.h>
49
#include <linux/sunrpc/metrics.h>
50
#include <linux/sunrpc/bc_xprt.h>
51
#include <linux/rcupdate.h>
L
Linus Torvalds 已提交
52

53 54
#include <trace/events/sunrpc.h>

55 56
#include "sunrpc.h"

L
Linus Torvalds 已提交
57 58 59 60
/*
 * Local variables
 */

J
Jeff Layton 已提交
61
#if IS_ENABLED(CONFIG_SUNRPC_DEBUG)
L
Linus Torvalds 已提交
62 63 64 65 66 67
# define RPCDBG_FACILITY	RPCDBG_XPRT
#endif

/*
 * Local functions
 */
68
static void	 xprt_init(struct rpc_xprt *xprt, struct net *net);
69
static __be32	xprt_alloc_xid(struct rpc_xprt *xprt);
L
Linus Torvalds 已提交
70 71
static void	xprt_connect_status(struct rpc_task *task);
static int      __xprt_get_cong(struct rpc_xprt *, struct rpc_task *);
72
static void     __xprt_put_cong(struct rpc_xprt *, struct rpc_rqst *);
73
static void	 xprt_destroy(struct rpc_xprt *xprt);
L
Linus Torvalds 已提交
74

J
Jiri Slaby 已提交
75
static DEFINE_SPINLOCK(xprt_list_lock);
76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98
static LIST_HEAD(xprt_list);

/**
 * xprt_register_transport - register a transport implementation
 * @transport: transport to register
 *
 * If a transport implementation is loaded as a kernel module, it can
 * call this interface to make itself known to the RPC client.
 *
 * Returns:
 * 0:		transport successfully registered
 * -EEXIST:	transport already registered
 * -EINVAL:	transport module being unloaded
 */
int xprt_register_transport(struct xprt_class *transport)
{
	struct xprt_class *t;
	int result;

	result = -EEXIST;
	spin_lock(&xprt_list_lock);
	list_for_each_entry(t, &xprt_list, list) {
		/* don't register the same transport class twice */
99
		if (t->ident == transport->ident)
100 101 102
			goto out;
	}

103 104 105 106
	list_add_tail(&transport->list, &xprt_list);
	printk(KERN_INFO "RPC: Registered %s transport module.\n",
	       transport->name);
	result = 0;
107 108 109 110 111 112 113 114 115

out:
	spin_unlock(&xprt_list_lock);
	return result;
}
EXPORT_SYMBOL_GPL(xprt_register_transport);

/**
 * xprt_unregister_transport - unregister a transport implementation
116
 * @transport: transport to unregister
117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145
 *
 * Returns:
 * 0:		transport successfully unregistered
 * -ENOENT:	transport never registered
 */
int xprt_unregister_transport(struct xprt_class *transport)
{
	struct xprt_class *t;
	int result;

	result = 0;
	spin_lock(&xprt_list_lock);
	list_for_each_entry(t, &xprt_list, list) {
		if (t == transport) {
			printk(KERN_INFO
				"RPC: Unregistered %s transport module.\n",
				transport->name);
			list_del_init(&transport->list);
			goto out;
		}
	}
	result = -ENOENT;

out:
	spin_unlock(&xprt_list_lock);
	return result;
}
EXPORT_SYMBOL_GPL(xprt_unregister_transport);

146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167
/**
 * xprt_load_transport - load a transport implementation
 * @transport_name: transport to load
 *
 * Returns:
 * 0:		transport successfully loaded
 * -ENOENT:	transport module not available
 */
int xprt_load_transport(const char *transport_name)
{
	struct xprt_class *t;
	int result;

	result = 0;
	spin_lock(&xprt_list_lock);
	list_for_each_entry(t, &xprt_list, list) {
		if (strcmp(t->name, transport_name) == 0) {
			spin_unlock(&xprt_list_lock);
			goto out;
		}
	}
	spin_unlock(&xprt_list_lock);
168
	result = request_module("xprt%s", transport_name);
169 170 171 172 173
out:
	return result;
}
EXPORT_SYMBOL_GPL(xprt_load_transport);

174 175 176
/**
 * xprt_reserve_xprt - serialize write access to transports
 * @task: task that is requesting access to the transport
177
 * @xprt: pointer to the target transport
178 179 180 181 182
 *
 * This prevents mixing the payload of separate requests, and prevents
 * transport connects from colliding with writes.  No congestion control
 * is provided.
 */
183
int xprt_reserve_xprt(struct rpc_xprt *xprt, struct rpc_task *task)
184 185
{
	struct rpc_rqst *req = task->tk_rqstp;
186
	int priority;
187 188 189 190 191 192 193

	if (test_and_set_bit(XPRT_LOCKED, &xprt->state)) {
		if (task == xprt->snd_task)
			return 1;
		goto out_sleep;
	}
	xprt->snd_task = task;
194
	if (req != NULL)
195
		req->rq_ntrans++;
196

197 198 199
	return 1;

out_sleep:
200
	dprintk("RPC: %5u failed to lock transport %p\n",
201 202 203
			task->tk_pid, xprt);
	task->tk_timeout = 0;
	task->tk_status = -EAGAIN;
204 205 206 207
	if (req == NULL)
		priority = RPC_PRIORITY_LOW;
	else if (!req->rq_ntrans)
		priority = RPC_PRIORITY_NORMAL;
208
	else
209 210
		priority = RPC_PRIORITY_HIGH;
	rpc_sleep_on_priority(&xprt->sending, task, NULL, priority);
211 212
	return 0;
}
213
EXPORT_SYMBOL_GPL(xprt_reserve_xprt);
214

215 216 217
static void xprt_clear_locked(struct rpc_xprt *xprt)
{
	xprt->snd_task = NULL;
218
	if (!test_bit(XPRT_CLOSE_WAIT, &xprt->state)) {
219
		smp_mb__before_atomic();
220
		clear_bit(XPRT_LOCKED, &xprt->state);
221
		smp_mb__after_atomic();
222
	} else
223
		queue_work(xprtiod_workqueue, &xprt->task_cleanup);
224 225
}

L
Linus Torvalds 已提交
226
/*
227 228 229 230 231 232
 * xprt_reserve_xprt_cong - serialize write access to transports
 * @task: task that is requesting access to the transport
 *
 * Same as xprt_reserve_xprt, but Van Jacobson congestion control is
 * integrated into the decision of whether a request is allowed to be
 * woken up and given access to the transport.
L
Linus Torvalds 已提交
233
 */
234
int xprt_reserve_xprt_cong(struct rpc_xprt *xprt, struct rpc_task *task)
L
Linus Torvalds 已提交
235 236
{
	struct rpc_rqst *req = task->tk_rqstp;
237
	int priority;
L
Linus Torvalds 已提交
238

239
	if (test_and_set_bit(XPRT_LOCKED, &xprt->state)) {
L
Linus Torvalds 已提交
240 241 242 243
		if (task == xprt->snd_task)
			return 1;
		goto out_sleep;
	}
244 245 246 247
	if (req == NULL) {
		xprt->snd_task = task;
		return 1;
	}
248
	if (__xprt_get_cong(xprt, task)) {
L
Linus Torvalds 已提交
249
		xprt->snd_task = task;
250
		req->rq_ntrans++;
L
Linus Torvalds 已提交
251 252
		return 1;
	}
253
	xprt_clear_locked(xprt);
L
Linus Torvalds 已提交
254
out_sleep:
255 256
	if (req)
		__xprt_put_cong(xprt, req);
257
	dprintk("RPC: %5u failed to lock transport %p\n", task->tk_pid, xprt);
L
Linus Torvalds 已提交
258 259
	task->tk_timeout = 0;
	task->tk_status = -EAGAIN;
260 261 262 263
	if (req == NULL)
		priority = RPC_PRIORITY_LOW;
	else if (!req->rq_ntrans)
		priority = RPC_PRIORITY_NORMAL;
L
Linus Torvalds 已提交
264
	else
265 266
		priority = RPC_PRIORITY_HIGH;
	rpc_sleep_on_priority(&xprt->sending, task, NULL, priority);
L
Linus Torvalds 已提交
267 268
	return 0;
}
269
EXPORT_SYMBOL_GPL(xprt_reserve_xprt_cong);
L
Linus Torvalds 已提交
270

271
static inline int xprt_lock_write(struct rpc_xprt *xprt, struct rpc_task *task)
L
Linus Torvalds 已提交
272 273 274
{
	int retval;

C
Chuck Lever 已提交
275
	spin_lock_bh(&xprt->transport_lock);
276
	retval = xprt->ops->reserve_xprt(xprt, task);
C
Chuck Lever 已提交
277
	spin_unlock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
278 279 280
	return retval;
}

281
static bool __xprt_lock_write_func(struct rpc_task *task, void *data)
282
{
283
	struct rpc_xprt *xprt = data;
284 285 286 287
	struct rpc_rqst *req;

	req = task->tk_rqstp;
	xprt->snd_task = task;
288
	if (req)
289
		req->rq_ntrans++;
290 291
	return true;
}
292

293 294 295 296 297
static void __xprt_lock_write_next(struct rpc_xprt *xprt)
{
	if (test_and_set_bit(XPRT_LOCKED, &xprt->state))
		return;

298 299
	if (rpc_wake_up_first_on_wq(xprtiod_workqueue, &xprt->sending,
				__xprt_lock_write_func, xprt))
300
		return;
301
	xprt_clear_locked(xprt);
302 303
}

304
static bool __xprt_lock_write_cong_func(struct rpc_task *task, void *data)
L
Linus Torvalds 已提交
305
{
306
	struct rpc_xprt *xprt = data;
307
	struct rpc_rqst *req;
L
Linus Torvalds 已提交
308

309 310 311
	req = task->tk_rqstp;
	if (req == NULL) {
		xprt->snd_task = task;
312
		return true;
313
	}
314
	if (__xprt_get_cong(xprt, task)) {
L
Linus Torvalds 已提交
315
		xprt->snd_task = task;
316
		req->rq_ntrans++;
317
		return true;
L
Linus Torvalds 已提交
318
	}
319 320 321 322 323 324 325 326 327
	return false;
}

static void __xprt_lock_write_next_cong(struct rpc_xprt *xprt)
{
	if (test_and_set_bit(XPRT_LOCKED, &xprt->state))
		return;
	if (RPCXPRT_CONGESTED(xprt))
		goto out_unlock;
328 329
	if (rpc_wake_up_first_on_wq(xprtiod_workqueue, &xprt->sending,
				__xprt_lock_write_cong_func, xprt))
330
		return;
L
Linus Torvalds 已提交
331
out_unlock:
332
	xprt_clear_locked(xprt);
L
Linus Torvalds 已提交
333 334
}

335 336 337 338 339 340 341 342 343
static void xprt_task_clear_bytes_sent(struct rpc_task *task)
{
	if (task != NULL) {
		struct rpc_rqst *req = task->tk_rqstp;
		if (req != NULL)
			req->rq_bytes_sent = 0;
	}
}

344 345 346 347 348 349
/**
 * xprt_release_xprt - allow other requests to use a transport
 * @xprt: transport with other tasks potentially waiting
 * @task: task that is releasing access to the transport
 *
 * Note that "task" can be NULL.  No congestion control is provided.
L
Linus Torvalds 已提交
350
 */
351
void xprt_release_xprt(struct rpc_xprt *xprt, struct rpc_task *task)
L
Linus Torvalds 已提交
352 353
{
	if (xprt->snd_task == task) {
354
		xprt_task_clear_bytes_sent(task);
355
		xprt_clear_locked(xprt);
L
Linus Torvalds 已提交
356 357 358
		__xprt_lock_write_next(xprt);
	}
}
359
EXPORT_SYMBOL_GPL(xprt_release_xprt);
L
Linus Torvalds 已提交
360

361 362 363 364 365 366 367 368 369 370 371
/**
 * xprt_release_xprt_cong - allow other requests to use a transport
 * @xprt: transport with other tasks potentially waiting
 * @task: task that is releasing access to the transport
 *
 * Note that "task" can be NULL.  Another task is awoken to use the
 * transport if the transport's congestion window allows it.
 */
void xprt_release_xprt_cong(struct rpc_xprt *xprt, struct rpc_task *task)
{
	if (xprt->snd_task == task) {
372
		xprt_task_clear_bytes_sent(task);
373
		xprt_clear_locked(xprt);
374 375 376
		__xprt_lock_write_next_cong(xprt);
	}
}
377
EXPORT_SYMBOL_GPL(xprt_release_xprt_cong);
378 379

static inline void xprt_release_write(struct rpc_xprt *xprt, struct rpc_task *task)
L
Linus Torvalds 已提交
380
{
C
Chuck Lever 已提交
381
	spin_lock_bh(&xprt->transport_lock);
382
	xprt->ops->release_xprt(xprt, task);
C
Chuck Lever 已提交
383
	spin_unlock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
384 385 386 387 388 389 390 391 392 393 394 395 396
}

/*
 * Van Jacobson congestion avoidance. Check if the congestion window
 * overflowed. Put the task to sleep if this is the case.
 */
static int
__xprt_get_cong(struct rpc_xprt *xprt, struct rpc_task *task)
{
	struct rpc_rqst *req = task->tk_rqstp;

	if (req->rq_cong)
		return 1;
397
	dprintk("RPC: %5u xprt_cwnd_limited cong = %lu cwnd = %lu\n",
L
Linus Torvalds 已提交
398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416
			task->tk_pid, xprt->cong, xprt->cwnd);
	if (RPCXPRT_CONGESTED(xprt))
		return 0;
	req->rq_cong = 1;
	xprt->cong += RPC_CWNDSCALE;
	return 1;
}

/*
 * Adjust the congestion window, and wake up the next task
 * that has been sleeping due to congestion
 */
static void
__xprt_put_cong(struct rpc_xprt *xprt, struct rpc_rqst *req)
{
	if (!req->rq_cong)
		return;
	req->rq_cong = 0;
	xprt->cong -= RPC_CWNDSCALE;
417
	__xprt_lock_write_next_cong(xprt);
L
Linus Torvalds 已提交
418 419
}

420 421 422 423 424 425 426 427
/**
 * xprt_release_rqst_cong - housekeeping when request is complete
 * @task: RPC request that recently completed
 *
 * Useful for transports that require congestion control.
 */
void xprt_release_rqst_cong(struct rpc_task *task)
{
428 429 430
	struct rpc_rqst *req = task->tk_rqstp;

	__xprt_put_cong(req->rq_xprt, req);
431
}
432
EXPORT_SYMBOL_GPL(xprt_release_rqst_cong);
433

434 435
/**
 * xprt_adjust_cwnd - adjust transport congestion window
436
 * @xprt: pointer to xprt
437 438 439
 * @task: recently completed RPC request used to adjust window
 * @result: result code of completed RPC request
 *
440 441 442 443 444 445 446 447 448
 * The transport code maintains an estimate on the maximum number of out-
 * standing RPC requests, using a smoothed version of the congestion
 * avoidance implemented in 44BSD. This is basically the Van Jacobson
 * congestion algorithm: If a retransmit occurs, the congestion window is
 * halved; otherwise, it is incremented by 1/cwnd when
 *
 *	-	a reply is received and
 *	-	a full number of requests are outstanding and
 *	-	the congestion window hasn't been updated recently.
L
Linus Torvalds 已提交
449
 */
450
void xprt_adjust_cwnd(struct rpc_xprt *xprt, struct rpc_task *task, int result)
L
Linus Torvalds 已提交
451
{
452 453
	struct rpc_rqst *req = task->tk_rqstp;
	unsigned long cwnd = xprt->cwnd;
L
Linus Torvalds 已提交
454 455 456 457 458 459 460

	if (result >= 0 && cwnd <= xprt->cong) {
		/* The (cwnd >> 1) term makes sure
		 * the result gets rounded properly. */
		cwnd += (RPC_CWNDSCALE * RPC_CWNDSCALE + (cwnd >> 1)) / cwnd;
		if (cwnd > RPC_MAXCWND(xprt))
			cwnd = RPC_MAXCWND(xprt);
461
		__xprt_lock_write_next_cong(xprt);
L
Linus Torvalds 已提交
462 463 464 465 466
	} else if (result == -ETIMEDOUT) {
		cwnd >>= 1;
		if (cwnd < RPC_CWNDSCALE)
			cwnd = RPC_CWNDSCALE;
	}
467
	dprintk("RPC:       cong %ld, cwnd was %ld, now %ld\n",
L
Linus Torvalds 已提交
468 469
			xprt->cong, xprt->cwnd, cwnd);
	xprt->cwnd = cwnd;
470
	__xprt_put_cong(xprt, req);
L
Linus Torvalds 已提交
471
}
472
EXPORT_SYMBOL_GPL(xprt_adjust_cwnd);
L
Linus Torvalds 已提交
473

474 475 476 477 478 479 480 481 482 483 484 485 486
/**
 * xprt_wake_pending_tasks - wake all tasks on a transport's pending queue
 * @xprt: transport with waiting tasks
 * @status: result code to plant in each task before waking it
 *
 */
void xprt_wake_pending_tasks(struct rpc_xprt *xprt, int status)
{
	if (status < 0)
		rpc_wake_up_status(&xprt->pending, status);
	else
		rpc_wake_up(&xprt->pending);
}
487
EXPORT_SYMBOL_GPL(xprt_wake_pending_tasks);
488

489 490 491
/**
 * xprt_wait_for_buffer_space - wait for transport output buffer to clear
 * @task: task to be put to sleep
R
Randy Dunlap 已提交
492
 * @action: function pointer to be executed after wait
493 494 495 496
 *
 * Note that we only set the timer for the case of RPC_IS_SOFT(), since
 * we don't in general want to force a socket disconnection due to
 * an incomplete RPC call transmission.
497
 */
498
void xprt_wait_for_buffer_space(struct rpc_task *task, rpc_action action)
499 500 501 502
{
	struct rpc_rqst *req = task->tk_rqstp;
	struct rpc_xprt *xprt = req->rq_xprt;

503
	task->tk_timeout = RPC_IS_SOFT(task) ? req->rq_timeout : 0;
504
	rpc_sleep_on(&xprt->pending, task, action);
505
}
506
EXPORT_SYMBOL_GPL(xprt_wait_for_buffer_space);
507 508 509 510 511 512 513 514 515 516 517

/**
 * xprt_write_space - wake the task waiting for transport output buffer space
 * @xprt: transport with waiting tasks
 *
 * Can be called in a soft IRQ context, so xprt_write_space never sleeps.
 */
void xprt_write_space(struct rpc_xprt *xprt)
{
	spin_lock_bh(&xprt->transport_lock);
	if (xprt->snd_task) {
518 519
		dprintk("RPC:       write space: waking waiting task on "
				"xprt %p\n", xprt);
520 521
		rpc_wake_up_queued_task_on_wq(xprtiod_workqueue,
				&xprt->pending, xprt->snd_task);
522 523 524
	}
	spin_unlock_bh(&xprt->transport_lock);
}
525
EXPORT_SYMBOL_GPL(xprt_write_space);
526

527 528 529 530 531 532 533 534 535 536 537 538
/**
 * xprt_set_retrans_timeout_def - set a request's retransmit timeout
 * @task: task whose timeout is to be set
 *
 * Set a request's retransmit timeout based on the transport's
 * default timeout parameters.  Used by transports that don't adjust
 * the retransmit timeout based on round-trip time estimation.
 */
void xprt_set_retrans_timeout_def(struct rpc_task *task)
{
	task->tk_timeout = task->tk_rqstp->rq_timeout;
}
539
EXPORT_SYMBOL_GPL(xprt_set_retrans_timeout_def);
540

541
/**
542 543
 * xprt_set_retrans_timeout_rtt - set a request's retransmit timeout
 * @task: task whose timeout is to be set
544
 *
545 546 547 548 549
 * Set a request's retransmit timeout using the RTT estimator.
 */
void xprt_set_retrans_timeout_rtt(struct rpc_task *task)
{
	int timer = task->tk_msg.rpc_proc->p_timer;
550 551
	struct rpc_clnt *clnt = task->tk_client;
	struct rpc_rtt *rtt = clnt->cl_rtt;
552
	struct rpc_rqst *req = task->tk_rqstp;
553
	unsigned long max_timeout = clnt->cl_timeout->to_maxval;
554 555 556 557 558 559

	task->tk_timeout = rpc_calc_rto(rtt, timer);
	task->tk_timeout <<= rpc_ntimeo(rtt, timer) + req->rq_retries;
	if (task->tk_timeout > max_timeout || task->tk_timeout == 0)
		task->tk_timeout = max_timeout;
}
560
EXPORT_SYMBOL_GPL(xprt_set_retrans_timeout_rtt);
561

L
Linus Torvalds 已提交
562 563
static void xprt_reset_majortimeo(struct rpc_rqst *req)
{
564
	const struct rpc_timeout *to = req->rq_task->tk_client->cl_timeout;
L
Linus Torvalds 已提交
565 566 567 568 569 570 571 572 573 574 575

	req->rq_majortimeo = req->rq_timeout;
	if (to->to_exponential)
		req->rq_majortimeo <<= to->to_retries;
	else
		req->rq_majortimeo += to->to_increment * to->to_retries;
	if (req->rq_majortimeo > to->to_maxval || req->rq_majortimeo == 0)
		req->rq_majortimeo = to->to_maxval;
	req->rq_majortimeo += jiffies;
}

576 577 578 579
/**
 * xprt_adjust_timeout - adjust timeout values for next retransmit
 * @req: RPC request containing parameters to use for the adjustment
 *
L
Linus Torvalds 已提交
580 581 582 583
 */
int xprt_adjust_timeout(struct rpc_rqst *req)
{
	struct rpc_xprt *xprt = req->rq_xprt;
584
	const struct rpc_timeout *to = req->rq_task->tk_client->cl_timeout;
L
Linus Torvalds 已提交
585 586 587 588 589 590 591 592 593 594 595 596 597 598 599
	int status = 0;

	if (time_before(jiffies, req->rq_majortimeo)) {
		if (to->to_exponential)
			req->rq_timeout <<= 1;
		else
			req->rq_timeout += to->to_increment;
		if (to->to_maxval && req->rq_timeout >= to->to_maxval)
			req->rq_timeout = to->to_maxval;
		req->rq_retries++;
	} else {
		req->rq_timeout = to->to_initval;
		req->rq_retries = 0;
		xprt_reset_majortimeo(req);
		/* Reset the RTT counters == "slow start" */
C
Chuck Lever 已提交
600
		spin_lock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
601
		rpc_init_rtt(req->rq_task->tk_client->cl_rtt, to->to_initval);
C
Chuck Lever 已提交
602
		spin_unlock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
603 604 605 606 607 608 609 610 611 612
		status = -ETIMEDOUT;
	}

	if (req->rq_timeout == 0) {
		printk(KERN_WARNING "xprt_adjust_timeout: rq_timeout = 0!\n");
		req->rq_timeout = 5 * HZ;
	}
	return status;
}

613
static void xprt_autoclose(struct work_struct *work)
L
Linus Torvalds 已提交
614
{
615 616
	struct rpc_xprt *xprt =
		container_of(work, struct rpc_xprt, task_cleanup);
L
Linus Torvalds 已提交
617

618
	clear_bit(XPRT_CLOSE_WAIT, &xprt->state);
619
	xprt->ops->close(xprt);
L
Linus Torvalds 已提交
620
	xprt_release_write(xprt, NULL);
621
	wake_up_bit(&xprt->state, XPRT_LOCKED);
L
Linus Torvalds 已提交
622 623
}

624
/**
625
 * xprt_disconnect_done - mark a transport as disconnected
626 627
 * @xprt: transport to flag for disconnect
 *
L
Linus Torvalds 已提交
628
 */
629
void xprt_disconnect_done(struct rpc_xprt *xprt)
L
Linus Torvalds 已提交
630
{
631
	dprintk("RPC:       disconnected transport %p\n", xprt);
C
Chuck Lever 已提交
632
	spin_lock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
633
	xprt_clear_connected(xprt);
634
	xprt_wake_pending_tasks(xprt, -EAGAIN);
C
Chuck Lever 已提交
635
	spin_unlock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
636
}
637
EXPORT_SYMBOL_GPL(xprt_disconnect_done);
L
Linus Torvalds 已提交
638

639 640 641 642 643 644 645 646 647 648 649 650
/**
 * xprt_force_disconnect - force a transport to disconnect
 * @xprt: transport to disconnect
 *
 */
void xprt_force_disconnect(struct rpc_xprt *xprt)
{
	/* Don't race with the test_bit() in xprt_clear_locked() */
	spin_lock_bh(&xprt->transport_lock);
	set_bit(XPRT_CLOSE_WAIT, &xprt->state);
	/* Try to schedule an autoclose RPC call */
	if (test_and_set_bit(XPRT_LOCKED, &xprt->state) == 0)
651
		queue_work(xprtiod_workqueue, &xprt->task_cleanup);
652
	xprt_wake_pending_tasks(xprt, -EAGAIN);
653 654
	spin_unlock_bh(&xprt->transport_lock);
}
655
EXPORT_SYMBOL_GPL(xprt_force_disconnect);
656

657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673
/**
 * xprt_conditional_disconnect - force a transport to disconnect
 * @xprt: transport to disconnect
 * @cookie: 'connection cookie'
 *
 * This attempts to break the connection if and only if 'cookie' matches
 * the current transport 'connection cookie'. It ensures that we don't
 * try to break the connection more than once when we need to retransmit
 * a batch of RPC requests.
 *
 */
void xprt_conditional_disconnect(struct rpc_xprt *xprt, unsigned int cookie)
{
	/* Don't race with the test_bit() in xprt_clear_locked() */
	spin_lock_bh(&xprt->transport_lock);
	if (cookie != xprt->connect_cookie)
		goto out;
674
	if (test_bit(XPRT_CLOSING, &xprt->state))
675 676 677 678
		goto out;
	set_bit(XPRT_CLOSE_WAIT, &xprt->state);
	/* Try to schedule an autoclose RPC call */
	if (test_and_set_bit(XPRT_LOCKED, &xprt->state) == 0)
679
		queue_work(xprtiod_workqueue, &xprt->task_cleanup);
680
	xprt_wake_pending_tasks(xprt, -EAGAIN);
681 682 683 684
out:
	spin_unlock_bh(&xprt->transport_lock);
}

685 686 687 688 689 690 691 692 693 694 695 696 697 698
static bool
xprt_has_timer(const struct rpc_xprt *xprt)
{
	return xprt->idle_timeout != 0;
}

static void
xprt_schedule_autodisconnect(struct rpc_xprt *xprt)
	__must_hold(&xprt->transport_lock)
{
	if (list_empty(&xprt->recv) && xprt_has_timer(xprt))
		mod_timer(&xprt->timer, xprt->last_used + xprt->idle_timeout);
}

L
Linus Torvalds 已提交
699
static void
700
xprt_init_autodisconnect(struct timer_list *t)
L
Linus Torvalds 已提交
701
{
702
	struct rpc_xprt *xprt = from_timer(xprt, t, timer);
L
Linus Torvalds 已提交
703

C
Chuck Lever 已提交
704
	spin_lock(&xprt->transport_lock);
705
	if (!list_empty(&xprt->recv))
L
Linus Torvalds 已提交
706
		goto out_abort;
707 708
	/* Reset xprt->last_used to avoid connect/autodisconnect cycling */
	xprt->last_used = jiffies;
709
	if (test_and_set_bit(XPRT_LOCKED, &xprt->state))
L
Linus Torvalds 已提交
710
		goto out_abort;
C
Chuck Lever 已提交
711
	spin_unlock(&xprt->transport_lock);
712
	queue_work(xprtiod_workqueue, &xprt->task_cleanup);
L
Linus Torvalds 已提交
713 714
	return;
out_abort:
C
Chuck Lever 已提交
715
	spin_unlock(&xprt->transport_lock);
L
Linus Torvalds 已提交
716 717
}

718 719 720 721 722 723 724 725 726 727 728
bool xprt_lock_connect(struct rpc_xprt *xprt,
		struct rpc_task *task,
		void *cookie)
{
	bool ret = false;

	spin_lock_bh(&xprt->transport_lock);
	if (!test_bit(XPRT_LOCKED, &xprt->state))
		goto out;
	if (xprt->snd_task != task)
		goto out;
729
	xprt_task_clear_bytes_sent(task);
730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745
	xprt->snd_task = cookie;
	ret = true;
out:
	spin_unlock_bh(&xprt->transport_lock);
	return ret;
}

void xprt_unlock_connect(struct rpc_xprt *xprt, void *cookie)
{
	spin_lock_bh(&xprt->transport_lock);
	if (xprt->snd_task != cookie)
		goto out;
	if (!test_bit(XPRT_LOCKED, &xprt->state))
		goto out;
	xprt->snd_task =NULL;
	xprt->ops->release_xprt(xprt, NULL);
746
	xprt_schedule_autodisconnect(xprt);
747 748
out:
	spin_unlock_bh(&xprt->transport_lock);
749
	wake_up_bit(&xprt->state, XPRT_LOCKED);
750 751
}

752 753 754
/**
 * xprt_connect - schedule a transport connect operation
 * @task: RPC task that is requesting the connect
L
Linus Torvalds 已提交
755 756 757 758
 *
 */
void xprt_connect(struct rpc_task *task)
{
759
	struct rpc_xprt	*xprt = task->tk_rqstp->rq_xprt;
L
Linus Torvalds 已提交
760

761
	dprintk("RPC: %5u xprt_connect xprt %p %s connected\n", task->tk_pid,
L
Linus Torvalds 已提交
762 763
			xprt, (xprt_connected(xprt) ? "is" : "is not"));

764
	if (!xprt_bound(xprt)) {
765
		task->tk_status = -EAGAIN;
L
Linus Torvalds 已提交
766 767 768 769
		return;
	}
	if (!xprt_lock_write(xprt, task))
		return;
770 771 772 773

	if (test_and_clear_bit(XPRT_CLOSE_WAIT, &xprt->state))
		xprt->ops->close(xprt);

774
	if (!xprt_connected(xprt)) {
775
		task->tk_rqstp->rq_bytes_sent = 0;
776
		task->tk_timeout = task->tk_rqstp->rq_timeout;
777
		task->tk_rqstp->rq_connect_cookie = xprt->connect_cookie;
778
		rpc_sleep_on(&xprt->pending, task, xprt_connect_status);
779 780 781 782 783

		if (test_bit(XPRT_CLOSING, &xprt->state))
			return;
		if (xprt_test_and_set_connecting(xprt))
			return;
784
		xprt->stat.connect_start = jiffies;
785
		xprt->ops->connect(xprt, task);
L
Linus Torvalds 已提交
786
	}
787
	xprt_release_write(xprt, task);
L
Linus Torvalds 已提交
788 789
}

790
static void xprt_connect_status(struct rpc_task *task)
L
Linus Torvalds 已提交
791
{
792
	struct rpc_xprt	*xprt = task->tk_rqstp->rq_xprt;
L
Linus Torvalds 已提交
793

794
	if (task->tk_status == 0) {
795 796
		xprt->stat.connect_count++;
		xprt->stat.connect_time += (long)jiffies - xprt->stat.connect_start;
797
		dprintk("RPC: %5u xprt_connect_status: connection established\n",
L
Linus Torvalds 已提交
798 799 800 801 802
				task->tk_pid);
		return;
	}

	switch (task->tk_status) {
803 804 805 806 807
	case -ECONNREFUSED:
	case -ECONNRESET:
	case -ECONNABORTED:
	case -ENETUNREACH:
	case -EHOSTUNREACH:
808
	case -EPIPE:
809 810
	case -EAGAIN:
		dprintk("RPC: %5u xprt_connect_status: retrying\n", task->tk_pid);
811
		break;
L
Linus Torvalds 已提交
812
	case -ETIMEDOUT:
813 814
		dprintk("RPC: %5u xprt_connect_status: connect attempt timed "
				"out\n", task->tk_pid);
L
Linus Torvalds 已提交
815 816
		break;
	default:
817 818
		dprintk("RPC: %5u xprt_connect_status: error %d connecting to "
				"server %s\n", task->tk_pid, -task->tk_status,
819
				xprt->servername);
820
		task->tk_status = -EIO;
L
Linus Torvalds 已提交
821 822 823
	}
}

824 825 826 827 828
/**
 * xprt_lookup_rqst - find an RPC request corresponding to an XID
 * @xprt: transport on which the original request was transmitted
 * @xid: RPC XID of incoming reply
 *
829
 * Caller holds xprt->recv_lock.
L
Linus Torvalds 已提交
830
 */
831
struct rpc_rqst *xprt_lookup_rqst(struct rpc_xprt *xprt, __be32 xid)
L
Linus Torvalds 已提交
832
{
833
	struct rpc_rqst *entry;
L
Linus Torvalds 已提交
834

835
	list_for_each_entry(entry, &xprt->recv, rq_list)
836 837
		if (entry->rq_xid == xid) {
			trace_xprt_lookup_rqst(xprt, xid, 0);
838
			entry->rq_rtt = ktime_sub(ktime_get(), entry->rq_xtime);
839
			return entry;
840
		}
841 842 843

	dprintk("RPC:       xprt_lookup_rqst did not find xid %08x\n",
			ntohl(xid));
844
	trace_xprt_lookup_rqst(xprt, xid, -ENOENT);
845 846
	xprt->stat.bad_xids++;
	return NULL;
L
Linus Torvalds 已提交
847
}
848
EXPORT_SYMBOL_GPL(xprt_lookup_rqst);
L
Linus Torvalds 已提交
849

850 851 852 853 854 855 856 857 858 859 860
/**
 * xprt_pin_rqst - Pin a request on the transport receive list
 * @req: Request to pin
 *
 * Caller must ensure this is atomic with the call to xprt_lookup_rqst()
 * so should be holding the xprt transport lock.
 */
void xprt_pin_rqst(struct rpc_rqst *req)
{
	set_bit(RPC_TASK_MSG_RECV, &req->rq_task->tk_runstate);
}
861
EXPORT_SYMBOL_GPL(xprt_pin_rqst);
862 863 864 865 866 867 868 869 870 871 872 873 874 875 876

/**
 * xprt_unpin_rqst - Unpin a request on the transport receive list
 * @req: Request to pin
 *
 * Caller should be holding the xprt transport lock.
 */
void xprt_unpin_rqst(struct rpc_rqst *req)
{
	struct rpc_task *task = req->rq_task;

	clear_bit(RPC_TASK_MSG_RECV, &task->tk_runstate);
	if (test_bit(RPC_TASK_MSG_RECV_WAIT, &task->tk_runstate))
		wake_up_bit(&task->tk_runstate, RPC_TASK_MSG_RECV);
}
877
EXPORT_SYMBOL_GPL(xprt_unpin_rqst);
878 879

static void xprt_wait_on_pinned_rqst(struct rpc_rqst *req)
880
__must_hold(&req->rq_xprt->recv_lock)
881 882
{
	struct rpc_task *task = req->rq_task;
S
Stephen Hemminger 已提交
883

884
	if (task && test_bit(RPC_TASK_MSG_RECV, &task->tk_runstate)) {
885
		spin_unlock(&req->rq_xprt->recv_lock);
886 887 888 889
		set_bit(RPC_TASK_MSG_RECV_WAIT, &task->tk_runstate);
		wait_on_bit(&task->tk_runstate, RPC_TASK_MSG_RECV,
				TASK_UNINTERRUPTIBLE);
		clear_bit(RPC_TASK_MSG_RECV_WAIT, &task->tk_runstate);
890
		spin_lock(&req->rq_xprt->recv_lock);
891 892 893
	}
}

894 895 896 897 898 899 900
/**
 * xprt_update_rtt - Update RPC RTT statistics
 * @task: RPC request that recently completed
 *
 * Caller holds xprt->recv_lock.
 */
void xprt_update_rtt(struct rpc_task *task)
901 902 903
{
	struct rpc_rqst *req = task->tk_rqstp;
	struct rpc_rtt *rtt = task->tk_client->cl_rtt;
904
	unsigned int timer = task->tk_msg.rpc_proc->p_timer;
905
	long m = usecs_to_jiffies(ktime_to_us(req->rq_rtt));
906 907 908

	if (timer) {
		if (req->rq_ntrans == 1)
909
			rpc_update_rtt(rtt, timer, m);
910 911 912
		rpc_set_timeo(rtt, timer, req->rq_ntrans - 1);
	}
}
913
EXPORT_SYMBOL_GPL(xprt_update_rtt);
914

915 916
/**
 * xprt_complete_rqst - called when reply processing is complete
917
 * @task: RPC request that recently completed
918 919
 * @copied: actual number of bytes received from the transport
 *
920
 * Caller holds xprt->recv_lock.
L
Linus Torvalds 已提交
921
 */
922
void xprt_complete_rqst(struct rpc_task *task, int copied)
L
Linus Torvalds 已提交
923
{
924
	struct rpc_rqst *req = task->tk_rqstp;
925
	struct rpc_xprt *xprt = req->rq_xprt;
L
Linus Torvalds 已提交
926

927 928
	dprintk("RPC: %5u xid %08x complete (%d bytes received)\n",
			task->tk_pid, ntohl(req->rq_xid), copied);
929
	trace_xprt_complete_rqst(xprt, req->rq_xid, copied);
L
Linus Torvalds 已提交
930

931
	xprt->stat.recvs++;
932

L
Linus Torvalds 已提交
933
	list_del_init(&req->rq_list);
934
	req->rq_private_buf.len = copied;
935 936
	/* Ensure all writes are done before we update */
	/* req->rq_reply_bytes_recvd */
937
	smp_wmb();
938
	req->rq_reply_bytes_recvd = copied;
939
	rpc_wake_up_queued_task(&xprt->pending, task);
L
Linus Torvalds 已提交
940
}
941
EXPORT_SYMBOL_GPL(xprt_complete_rqst);
L
Linus Torvalds 已提交
942

943
static void xprt_timer(struct rpc_task *task)
L
Linus Torvalds 已提交
944
{
945
	struct rpc_rqst *req = task->tk_rqstp;
L
Linus Torvalds 已提交
946 947
	struct rpc_xprt *xprt = req->rq_xprt;

948 949
	if (task->tk_status != -ETIMEDOUT)
		return;
L
Linus Torvalds 已提交
950

C
Chuck Lever 已提交
951
	trace_xprt_timer(xprt, req->rq_xid, task->tk_status);
952
	if (!req->rq_reply_bytes_recvd) {
953
		if (xprt->ops->timer)
954
			xprt->ops->timer(xprt, task);
955 956
	} else
		task->tk_status = 0;
L
Linus Torvalds 已提交
957 958
}

959 960 961 962
/**
 * xprt_prepare_transmit - reserve the transport before sending a request
 * @task: RPC task about to send a request
 *
L
Linus Torvalds 已提交
963
 */
964
bool xprt_prepare_transmit(struct rpc_task *task)
L
Linus Torvalds 已提交
965 966 967
{
	struct rpc_rqst	*req = task->tk_rqstp;
	struct rpc_xprt	*xprt = req->rq_xprt;
968
	bool ret = false;
L
Linus Torvalds 已提交
969

970
	dprintk("RPC: %5u xprt_prepare_transmit\n", task->tk_pid);
L
Linus Torvalds 已提交
971

C
Chuck Lever 已提交
972
	spin_lock_bh(&xprt->transport_lock);
973 974 975 976 977 978 979 980 981 982 983 984
	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;
		}
L
Linus Torvalds 已提交
985
	}
986 987 988 989 990
	if (!xprt->ops->reserve_xprt(xprt, task)) {
		task->tk_status = -EAGAIN;
		goto out_unlock;
	}
	ret = true;
L
Linus Torvalds 已提交
991
out_unlock:
C
Chuck Lever 已提交
992
	spin_unlock_bh(&xprt->transport_lock);
993
	return ret;
L
Linus Torvalds 已提交
994 995
}

996
void xprt_end_transmit(struct rpc_task *task)
997
{
998
	xprt_release_write(task->tk_rqstp->rq_xprt, task);
999 1000
}

1001 1002 1003 1004 1005 1006 1007
/**
 * xprt_transmit - send an RPC request on a transport
 * @task: controlling RPC task
 *
 * We have to copy the iovec because sendmsg fiddles with its contents.
 */
void xprt_transmit(struct rpc_task *task)
L
Linus Torvalds 已提交
1008 1009 1010
{
	struct rpc_rqst	*req = task->tk_rqstp;
	struct rpc_xprt	*xprt = req->rq_xprt;
1011
	unsigned int connect_cookie;
1012
	int status;
L
Linus Torvalds 已提交
1013

1014
	dprintk("RPC: %5u xprt_transmit(%u)\n", task->tk_pid, req->rq_slen);
L
Linus Torvalds 已提交
1015

1016
	if (!req->rq_reply_bytes_recvd) {
1017 1018 1019 1020 1021 1022 1023

		/* Verify that our message lies in the RPCSEC_GSS window */
		if (!req->rq_bytes_sent && rpcauth_xmit_need_reencode(task)) {
			task->tk_status = -EBADMSG;
			return;
		}

1024 1025 1026 1027
		if (list_empty(&req->rq_list) && rpc_reply_expected(task)) {
			/*
			 * Add to the list only if we're expecting a reply
			 */
L
Linus Torvalds 已提交
1028 1029 1030 1031
			/* Update the softirq receive buffer */
			memcpy(&req->rq_private_buf, &req->rq_rcv_buf,
					sizeof(req->rq_private_buf));
			/* Add request to the receive list */
1032
			spin_lock(&xprt->recv_lock);
L
Linus Torvalds 已提交
1033
			list_add_tail(&req->rq_list, &xprt->recv);
1034
			spin_unlock(&xprt->recv_lock);
L
Linus Torvalds 已提交
1035
			xprt_reset_majortimeo(req);
1036 1037
			/* Turn off autodisconnect */
			del_singleshot_timer_sync(&xprt->timer);
L
Linus Torvalds 已提交
1038 1039 1040 1041
		}
	} else if (!req->rq_bytes_sent)
		return;

1042
	connect_cookie = xprt->connect_cookie;
1043
	status = xprt->ops->send_request(task);
1044
	trace_xprt_transmit(xprt, req->rq_xid, status);
1045 1046 1047 1048
	if (status != 0) {
		task->tk_status = status;
		return;
	}
C
Chuck Lever 已提交
1049
	xprt_inject_disconnect(xprt);
1050

1051
	dprintk("RPC: %5u xmit complete\n", task->tk_pid);
1052
	task->tk_flags |= RPC_TASK_SENT;
1053
	spin_lock_bh(&xprt->transport_lock);
1054

1055
	xprt->ops->set_retrans_timeout(task);
1056

1057 1058 1059
	xprt->stat.sends++;
	xprt->stat.req_u += xprt->stat.sends - xprt->stat.recvs;
	xprt->stat.bklog_u += xprt->backlog.qlen;
1060 1061
	xprt->stat.sending_u += xprt->sending.qlen;
	xprt->stat.pending_u += xprt->pending.qlen;
1062
	spin_unlock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
1063

1064 1065
	req->rq_connect_cookie = connect_cookie;
	if (rpc_reply_expected(task) && !READ_ONCE(req->rq_reply_bytes_recvd)) {
1066
		/*
1067 1068 1069
		 * Sleep on the pending queue if we're expecting a reply.
		 * The spinlock ensures atomicity between the test of
		 * req->rq_reply_bytes_recvd, and the call to rpc_sleep_on().
1070
		 */
1071 1072
		spin_lock(&xprt->recv_lock);
		if (!req->rq_reply_bytes_recvd) {
1073
			rpc_sleep_on(&xprt->pending, task, xprt_timer);
1074 1075 1076 1077 1078 1079 1080 1081 1082
			/*
			 * Send an extra queue wakeup call if the
			 * connection was dropped in case the call to
			 * rpc_sleep_on() raced.
			 */
			if (!xprt_connected(xprt))
				xprt_wake_pending_tasks(xprt, -ENOTCONN);
		}
		spin_unlock(&xprt->recv_lock);
1083
	}
L
Linus Torvalds 已提交
1084 1085
}

1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113
static void xprt_add_backlog(struct rpc_xprt *xprt, struct rpc_task *task)
{
	set_bit(XPRT_CONGESTED, &xprt->state);
	rpc_sleep_on(&xprt->backlog, task, NULL);
}

static void xprt_wake_up_backlog(struct rpc_xprt *xprt)
{
	if (rpc_wake_up_next(&xprt->backlog) == NULL)
		clear_bit(XPRT_CONGESTED, &xprt->state);
}

static bool xprt_throttle_congested(struct rpc_xprt *xprt, struct rpc_task *task)
{
	bool ret = false;

	if (!test_bit(XPRT_CONGESTED, &xprt->state))
		goto out;
	spin_lock(&xprt->reserve_lock);
	if (test_bit(XPRT_CONGESTED, &xprt->state)) {
		rpc_sleep_on(&xprt->backlog, task, NULL);
		ret = true;
	}
	spin_unlock(&xprt->reserve_lock);
out:
	return ret;
}

1114
static struct rpc_rqst *xprt_dynamic_alloc_slot(struct rpc_xprt *xprt)
1115 1116 1117
{
	struct rpc_rqst *req = ERR_PTR(-EAGAIN);

1118
	if (xprt->num_reqs >= xprt->max_reqs)
1119
		goto out;
1120
	++xprt->num_reqs;
1121 1122 1123
	spin_unlock(&xprt->reserve_lock);
	req = kzalloc(sizeof(struct rpc_rqst), GFP_NOFS);
	spin_lock(&xprt->reserve_lock);
1124 1125
	if (req != NULL)
		goto out;
1126
	--xprt->num_reqs;
1127 1128 1129 1130 1131 1132 1133
	req = ERR_PTR(-ENOMEM);
out:
	return req;
}

static bool xprt_dynamic_free_slot(struct rpc_xprt *xprt, struct rpc_rqst *req)
{
1134 1135
	if (xprt->num_reqs > xprt->min_reqs) {
		--xprt->num_reqs;
1136 1137 1138 1139 1140 1141
		kfree(req);
		return true;
	}
	return false;
}

1142
void xprt_alloc_slot(struct rpc_xprt *xprt, struct rpc_task *task)
L
Linus Torvalds 已提交
1143
{
1144
	struct rpc_rqst *req;
L
Linus Torvalds 已提交
1145

1146
	spin_lock(&xprt->reserve_lock);
L
Linus Torvalds 已提交
1147
	if (!list_empty(&xprt->free)) {
1148 1149 1150 1151
		req = list_entry(xprt->free.next, struct rpc_rqst, rq_list);
		list_del(&req->rq_list);
		goto out_init_req;
	}
1152
	req = xprt_dynamic_alloc_slot(xprt);
1153 1154 1155 1156 1157 1158
	if (!IS_ERR(req))
		goto out_init_req;
	switch (PTR_ERR(req)) {
	case -ENOMEM:
		dprintk("RPC:       dynamic allocation of request slot "
				"failed! Retrying\n");
1159
		task->tk_status = -ENOMEM;
1160 1161
		break;
	case -EAGAIN:
1162
		xprt_add_backlog(xprt, task);
1163
		dprintk("RPC:       waiting for request slot\n");
1164
		/* fall through */
1165 1166
	default:
		task->tk_status = -EAGAIN;
L
Linus Torvalds 已提交
1167
	}
1168
	spin_unlock(&xprt->reserve_lock);
1169 1170
	return;
out_init_req:
1171 1172
	xprt->stat.max_slots = max_t(unsigned int, xprt->stat.max_slots,
				     xprt->num_reqs);
1173 1174
	spin_unlock(&xprt->reserve_lock);

1175 1176
	task->tk_status = 0;
	task->tk_rqstp = req;
1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190
}
EXPORT_SYMBOL_GPL(xprt_alloc_slot);

void xprt_lock_and_alloc_slot(struct rpc_xprt *xprt, struct rpc_task *task)
{
	/* Note: grabbing the xprt_lock_write() ensures that we throttle
	 * new slot allocation if the transport is congested (i.e. when
	 * reconnecting a stream transport or when out of socket write
	 * buffer space).
	 */
	if (xprt_lock_write(xprt, task)) {
		xprt_alloc_slot(xprt, task);
		xprt_release_write(xprt, task);
	}
L
Linus Torvalds 已提交
1191
}
1192
EXPORT_SYMBOL_GPL(xprt_lock_and_alloc_slot);
L
Linus Torvalds 已提交
1193

1194
void xprt_free_slot(struct rpc_xprt *xprt, struct rpc_rqst *req)
1195 1196
{
	spin_lock(&xprt->reserve_lock);
1197 1198 1199 1200
	if (!xprt_dynamic_free_slot(xprt, req)) {
		memset(req, 0, sizeof(*req));	/* mark unused */
		list_add(&req->rq_list, &xprt->free);
	}
1201
	xprt_wake_up_backlog(xprt);
1202 1203
	spin_unlock(&xprt->reserve_lock);
}
1204
EXPORT_SYMBOL_GPL(xprt_free_slot);
1205

1206 1207 1208 1209 1210 1211 1212 1213 1214 1215
static void xprt_free_all_slots(struct rpc_xprt *xprt)
{
	struct rpc_rqst *req;
	while (!list_empty(&xprt->free)) {
		req = list_first_entry(&xprt->free, struct rpc_rqst, rq_list);
		list_del(&req->rq_list);
		kfree(req);
	}
}

1216 1217 1218
struct rpc_xprt *xprt_alloc(struct net *net, size_t size,
		unsigned int num_prealloc,
		unsigned int max_alloc)
1219 1220
{
	struct rpc_xprt *xprt;
1221 1222
	struct rpc_rqst *req;
	int i;
1223 1224 1225 1226 1227

	xprt = kzalloc(size, GFP_KERNEL);
	if (xprt == NULL)
		goto out;

1228 1229 1230 1231 1232
	xprt_init(xprt, net);

	for (i = 0; i < num_prealloc; i++) {
		req = kzalloc(sizeof(struct rpc_rqst), GFP_KERNEL);
		if (!req)
1233
			goto out_free;
1234 1235
		list_add(&req->rq_list, &xprt->free);
	}
1236 1237 1238 1239 1240
	if (max_alloc > num_prealloc)
		xprt->max_reqs = max_alloc;
	else
		xprt->max_reqs = num_prealloc;
	xprt->min_reqs = num_prealloc;
1241
	xprt->num_reqs = num_prealloc;
1242 1243 1244 1245

	return xprt;

out_free:
1246
	xprt_free(xprt);
1247 1248 1249 1250 1251
out:
	return NULL;
}
EXPORT_SYMBOL_GPL(xprt_alloc);

1252 1253
void xprt_free(struct rpc_xprt *xprt)
{
P
Pavel Emelyanov 已提交
1254
	put_net(xprt->xprt_net);
1255
	xprt_free_all_slots(xprt);
1256
	kfree_rcu(xprt, rcu);
1257 1258 1259
}
EXPORT_SYMBOL_GPL(xprt_free);

1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308
static __be32
xprt_alloc_xid(struct rpc_xprt *xprt)
{
	__be32 xid;

	spin_lock(&xprt->reserve_lock);
	xid = (__force __be32)xprt->xid++;
	spin_unlock(&xprt->reserve_lock);
	return xid;
}

static void
xprt_init_xid(struct rpc_xprt *xprt)
{
	xprt->xid = prandom_u32();
}

static void
xprt_request_init(struct rpc_task *task)
{
	struct rpc_xprt *xprt = task->tk_xprt;
	struct rpc_rqst	*req = task->tk_rqstp;

	INIT_LIST_HEAD(&req->rq_list);
	req->rq_timeout = task->tk_client->cl_timeout->to_initval;
	req->rq_task	= task;
	req->rq_xprt    = xprt;
	req->rq_buffer  = NULL;
	req->rq_xid	= xprt_alloc_xid(xprt);
	req->rq_connect_cookie = xprt->connect_cookie - 1;
	req->rq_bytes_sent = 0;
	req->rq_snd_buf.len = 0;
	req->rq_snd_buf.buflen = 0;
	req->rq_rcv_buf.len = 0;
	req->rq_rcv_buf.buflen = 0;
	req->rq_release_snd_buf = NULL;
	xprt_reset_majortimeo(req);
	dprintk("RPC: %5u reserved req %p xid %08x\n", task->tk_pid,
			req, ntohl(req->rq_xid));
}

static void
xprt_do_reserve(struct rpc_xprt *xprt, struct rpc_task *task)
{
	xprt->ops->alloc_slot(xprt, task);
	if (task->tk_rqstp != NULL)
		xprt_request_init(task);
}

1309 1310 1311 1312
/**
 * xprt_reserve - allocate an RPC request slot
 * @task: RPC task requesting a slot allocation
 *
1313 1314
 * If the transport is marked as being congested, or if no more
 * slots are available, place the task on the transport's
1315 1316 1317
 * backlog queue.
 */
void xprt_reserve(struct rpc_task *task)
L
Linus Torvalds 已提交
1318
{
1319
	struct rpc_xprt *xprt = task->tk_xprt;
L
Linus Torvalds 已提交
1320

1321 1322 1323 1324 1325 1326
	task->tk_status = 0;
	if (task->tk_rqstp != NULL)
		return;

	task->tk_timeout = 0;
	task->tk_status = -EAGAIN;
1327
	if (!xprt_throttle_congested(xprt, task))
1328
		xprt_do_reserve(xprt, task);
1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341
}

/**
 * xprt_retry_reserve - allocate an RPC request slot
 * @task: RPC task requesting a slot allocation
 *
 * If no more slots are available, place the task on the transport's
 * backlog queue.
 * Note that the only difference with xprt_reserve is that we now
 * ignore the value of the XPRT_CONGESTED flag.
 */
void xprt_retry_reserve(struct rpc_task *task)
{
1342
	struct rpc_xprt *xprt = task->tk_xprt;
1343 1344 1345 1346 1347 1348 1349

	task->tk_status = 0;
	if (task->tk_rqstp != NULL)
		return;

	task->tk_timeout = 0;
	task->tk_status = -EAGAIN;
1350
	xprt_do_reserve(xprt, task);
L
Linus Torvalds 已提交
1351 1352
}

1353 1354 1355 1356
/**
 * xprt_release - release an RPC request slot
 * @task: task which is finished with the slot
 *
L
Linus Torvalds 已提交
1357
 */
1358
void xprt_release(struct rpc_task *task)
L
Linus Torvalds 已提交
1359
{
1360
	struct rpc_xprt	*xprt;
1361
	struct rpc_rqst	*req = task->tk_rqstp;
L
Linus Torvalds 已提交
1362

1363 1364
	if (req == NULL) {
		if (task->tk_client) {
1365
			xprt = task->tk_xprt;
1366 1367 1368
			if (xprt->snd_task == task)
				xprt_release_write(xprt, task);
		}
L
Linus Torvalds 已提交
1369
		return;
1370
	}
1371 1372

	xprt = req->rq_xprt;
1373 1374 1375 1376
	if (task->tk_ops->rpc_count_stats != NULL)
		task->tk_ops->rpc_count_stats(task, task->tk_calldata);
	else if (task->tk_client)
		rpc_count_iostats(task, task->tk_client->cl_metrics);
1377 1378
	spin_lock(&xprt->recv_lock);
	if (!list_empty(&req->rq_list)) {
1379
		list_del_init(&req->rq_list);
1380 1381 1382
		xprt_wait_on_pinned_rqst(req);
	}
	spin_unlock(&xprt->recv_lock);
C
Chuck Lever 已提交
1383
	spin_lock_bh(&xprt->transport_lock);
1384
	xprt->ops->release_xprt(xprt, task);
1385 1386
	if (xprt->ops->release_request)
		xprt->ops->release_request(task);
L
Linus Torvalds 已提交
1387
	xprt->last_used = jiffies;
1388
	xprt_schedule_autodisconnect(xprt);
C
Chuck Lever 已提交
1389
	spin_unlock_bh(&xprt->transport_lock);
1390
	if (req->rq_buffer)
1391
		xprt->ops->buf_free(task);
C
Chuck Lever 已提交
1392
	xprt_inject_disconnect(xprt);
1393 1394
	if (req->rq_cred != NULL)
		put_rpccred(req->rq_cred);
L
Linus Torvalds 已提交
1395
	task->tk_rqstp = NULL;
1396 1397
	if (req->rq_release_snd_buf)
		req->rq_release_snd_buf(req);
1398

1399
	dprintk("RPC: %5u release request %p\n", task->tk_pid, req);
1400
	if (likely(!bc_prealloc(req)))
1401
		xprt->ops->free_slot(xprt, req);
1402
	else
1403
		xprt_free_bc_request(req);
L
Linus Torvalds 已提交
1404 1405
}

1406
static void xprt_init(struct rpc_xprt *xprt, struct net *net)
1407
{
1408
	kref_init(&xprt->kref);
1409 1410 1411

	spin_lock_init(&xprt->transport_lock);
	spin_lock_init(&xprt->reserve_lock);
1412
	spin_lock_init(&xprt->recv_lock);
1413 1414 1415

	INIT_LIST_HEAD(&xprt->free);
	INIT_LIST_HEAD(&xprt->recv);
1416
#if defined(CONFIG_SUNRPC_BACKCHANNEL)
1417 1418
	spin_lock_init(&xprt->bc_pa_lock);
	INIT_LIST_HEAD(&xprt->bc_pa_list);
1419
#endif /* CONFIG_SUNRPC_BACKCHANNEL */
1420
	INIT_LIST_HEAD(&xprt->xprt_switch);
1421

1422 1423
	xprt->last_used = jiffies;
	xprt->cwnd = RPC_INITCWND;
1424
	xprt->bind_index = 0;
1425 1426 1427

	rpc_init_wait_queue(&xprt->binding, "xprt_binding");
	rpc_init_wait_queue(&xprt->pending, "xprt_pending");
1428
	rpc_init_priority_wait_queue(&xprt->sending, "xprt_sending");
1429 1430 1431 1432
	rpc_init_priority_wait_queue(&xprt->backlog, "xprt_backlog");

	xprt_init_xid(xprt);

1433
	xprt->xprt_net = get_net(net);
1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453
}

/**
 * xprt_create_transport - create an RPC transport
 * @args: rpc transport creation arguments
 *
 */
struct rpc_xprt *xprt_create_transport(struct xprt_create *args)
{
	struct rpc_xprt	*xprt;
	struct xprt_class *t;

	spin_lock(&xprt_list_lock);
	list_for_each_entry(t, &xprt_list, list) {
		if (t->ident == args->ident) {
			spin_unlock(&xprt_list_lock);
			goto found;
		}
	}
	spin_unlock(&xprt_list_lock);
1454
	dprintk("RPC: transport (%d) not supported\n", args->ident);
1455 1456 1457 1458 1459 1460 1461
	return ERR_PTR(-EIO);

found:
	xprt = t->setup(args);
	if (IS_ERR(xprt)) {
		dprintk("RPC:       xprt_create_transport: failed, %ld\n",
				-PTR_ERR(xprt));
1462
		goto out;
1463
	}
1464 1465
	if (args->flags & XPRT_CREATE_NO_IDLE_TIMEOUT)
		xprt->idle_timeout = 0;
1466 1467
	INIT_WORK(&xprt->task_cleanup, xprt_autoclose);
	if (xprt_has_timer(xprt))
1468
		timer_setup(&xprt->timer, xprt_init_autodisconnect, 0);
1469
	else
1470
		timer_setup(&xprt->timer, NULL, 0);
1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481

	if (strlen(args->servername) > RPC_MAXNETNAMELEN) {
		xprt_destroy(xprt);
		return ERR_PTR(-EINVAL);
	}
	xprt->servername = kstrdup(args->servername, GFP_KERNEL);
	if (xprt->servername == NULL) {
		xprt_destroy(xprt);
		return ERR_PTR(-ENOMEM);
	}

1482
	rpc_xprt_debugfs_register(xprt);
1483

1484
	dprintk("RPC:       created transport %p with %u slots\n", xprt,
1485
			xprt->max_reqs);
1486
out:
1487 1488 1489
	return xprt;
}

1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506
static void xprt_destroy_cb(struct work_struct *work)
{
	struct rpc_xprt *xprt =
		container_of(work, struct rpc_xprt, task_cleanup);

	rpc_xprt_debugfs_unregister(xprt);
	rpc_destroy_wait_queue(&xprt->binding);
	rpc_destroy_wait_queue(&xprt->pending);
	rpc_destroy_wait_queue(&xprt->sending);
	rpc_destroy_wait_queue(&xprt->backlog);
	kfree(xprt->servername);
	/*
	 * Tear down transport state and free the rpc_xprt
	 */
	xprt->ops->destroy(xprt);
}

1507 1508
/**
 * xprt_destroy - destroy an RPC transport, killing off all requests.
1509
 * @xprt: transport to destroy
1510
 *
L
Linus Torvalds 已提交
1511
 */
1512
static void xprt_destroy(struct rpc_xprt *xprt)
L
Linus Torvalds 已提交
1513
{
1514
	dprintk("RPC:       destroying transport %p\n", xprt);
1515

1516 1517 1518
	/*
	 * Exclude transport connect/disconnect handlers and autoclose
	 */
1519 1520
	wait_on_bit_lock(&xprt->state, XPRT_LOCKED, TASK_UNINTERRUPTIBLE);

1521
	del_timer_sync(&xprt->timer);
1522 1523

	/*
1524 1525
	 * Destroy sockets etc from the system workqueue so they can
	 * safely flush receive work running on rpciod.
1526
	 */
1527 1528
	INIT_WORK(&xprt->task_cleanup, xprt_destroy_cb);
	schedule_work(&xprt->task_cleanup);
1529
}
L
Linus Torvalds 已提交
1530

1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548
static void xprt_destroy_kref(struct kref *kref)
{
	xprt_destroy(container_of(kref, struct rpc_xprt, kref));
}

/**
 * xprt_get - return a reference to an RPC transport.
 * @xprt: pointer to the transport
 *
 */
struct rpc_xprt *xprt_get(struct rpc_xprt *xprt)
{
	if (xprt != NULL && kref_get_unless_zero(&xprt->kref))
		return xprt;
	return NULL;
}
EXPORT_SYMBOL_GPL(xprt_get);

1549 1550 1551 1552 1553 1554 1555
/**
 * xprt_put - release a reference to an RPC transport.
 * @xprt: pointer to the transport
 *
 */
void xprt_put(struct rpc_xprt *xprt)
{
1556 1557
	if (xprt != NULL)
		kref_put(&xprt->kref, xprt_destroy_kref);
1558
}
1559
EXPORT_SYMBOL_GPL(xprt_put);