xprt.c 39.3 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);
L
Linus Torvalds 已提交
69 70 71
static void	xprt_request_init(struct rpc_task *, struct rpc_xprt *);
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
		rpc_wake_up_queued_task(&xprt->pending, xprt->snd_task);
521 522 523
	}
	spin_unlock_bh(&xprt->transport_lock);
}
524
EXPORT_SYMBOL_GPL(xprt_write_space);
525

526 527 528 529 530 531 532 533 534 535 536 537
/**
 * 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;
}
538
EXPORT_SYMBOL_GPL(xprt_set_retrans_timeout_def);
539

540
/**
541 542
 * xprt_set_retrans_timeout_rtt - set a request's retransmit timeout
 * @task: task whose timeout is to be set
543
 *
544 545 546 547 548
 * 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;
549 550
	struct rpc_clnt *clnt = task->tk_client;
	struct rpc_rtt *rtt = clnt->cl_rtt;
551
	struct rpc_rqst *req = task->tk_rqstp;
552
	unsigned long max_timeout = clnt->cl_timeout->to_maxval;
553 554 555 556 557 558

	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;
}
559
EXPORT_SYMBOL_GPL(xprt_set_retrans_timeout_rtt);
560

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

	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;
}

575 576 577 578
/**
 * xprt_adjust_timeout - adjust timeout values for next retransmit
 * @req: RPC request containing parameters to use for the adjustment
 *
L
Linus Torvalds 已提交
579 580 581 582
 */
int xprt_adjust_timeout(struct rpc_rqst *req)
{
	struct rpc_xprt *xprt = req->rq_xprt;
583
	const struct rpc_timeout *to = req->rq_task->tk_client->cl_timeout;
L
Linus Torvalds 已提交
584 585 586 587 588 589 590 591 592 593 594 595 596 597 598
	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 已提交
599
		spin_lock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
600
		rpc_init_rtt(req->rq_task->tk_client->cl_rtt, to->to_initval);
C
Chuck Lever 已提交
601
		spin_unlock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
602 603 604 605 606 607 608 609 610 611
		status = -ETIMEDOUT;
	}

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

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

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

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

638 639 640 641 642 643 644 645 646 647 648 649
/**
 * 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)
650
		queue_work(xprtiod_workqueue, &xprt->task_cleanup);
651
	xprt_wake_pending_tasks(xprt, -EAGAIN);
652 653
	spin_unlock_bh(&xprt->transport_lock);
}
654
EXPORT_SYMBOL_GPL(xprt_force_disconnect);
655

656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672
/**
 * 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;
673
	if (test_bit(XPRT_CLOSING, &xprt->state))
674 675 676 677
		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)
678
		queue_work(xprtiod_workqueue, &xprt->task_cleanup);
679
	xprt_wake_pending_tasks(xprt, -EAGAIN);
680 681 682 683
out:
	spin_unlock_bh(&xprt->transport_lock);
}

684 685 686 687 688 689 690 691 692 693 694 695 696 697
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 已提交
698 699 700 701 702
static void
xprt_init_autodisconnect(unsigned long data)
{
	struct rpc_xprt *xprt = (struct rpc_xprt *)data;

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

717 718 719 720 721 722 723 724 725 726 727
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;
728
	xprt_task_clear_bytes_sent(task);
729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744
	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);
745
	xprt_schedule_autodisconnect(xprt);
746 747
out:
	spin_unlock_bh(&xprt->transport_lock);
748
	wake_up_bit(&xprt->state, XPRT_LOCKED);
749 750
}

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

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

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

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

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

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

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

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

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

823 824 825 826 827
/**
 * 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
 *
L
Linus Torvalds 已提交
828
 */
829
struct rpc_rqst *xprt_lookup_rqst(struct rpc_xprt *xprt, __be32 xid)
L
Linus Torvalds 已提交
830
{
831
	struct rpc_rqst *entry;
L
Linus Torvalds 已提交
832

833
	list_for_each_entry(entry, &xprt->recv, rq_list)
834 835
		if (entry->rq_xid == xid) {
			trace_xprt_lookup_rqst(xprt, xid, 0);
836
			return entry;
837
		}
838 839 840

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

847 848 849 850 851 852 853 854 855 856 857
/**
 * 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);
}
858
EXPORT_SYMBOL_GPL(xprt_pin_rqst);
859 860 861 862 863 864 865 866 867 868 869 870 871 872 873

/**
 * 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);
}
874
EXPORT_SYMBOL_GPL(xprt_unpin_rqst);
875 876

static void xprt_wait_on_pinned_rqst(struct rpc_rqst *req)
877
__must_hold(&req->rq_xprt->recv_lock)
878 879 880 881
{
	struct rpc_task *task = req->rq_task;
	
	if (task && test_bit(RPC_TASK_MSG_RECV, &task->tk_runstate)) {
882
		spin_unlock(&req->rq_xprt->recv_lock);
883 884 885 886
		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);
887
		spin_lock(&req->rq_xprt->recv_lock);
888 889 890
	}
}

891
static void xprt_update_rtt(struct rpc_task *task)
892 893 894
{
	struct rpc_rqst *req = task->tk_rqstp;
	struct rpc_rtt *rtt = task->tk_client->cl_rtt;
895
	unsigned int timer = task->tk_msg.rpc_proc->p_timer;
896
	long m = usecs_to_jiffies(ktime_to_us(req->rq_rtt));
897 898 899

	if (timer) {
		if (req->rq_ntrans == 1)
900
			rpc_update_rtt(rtt, timer, m);
901 902 903 904
		rpc_set_timeo(rtt, timer, req->rq_ntrans - 1);
	}
}

905 906
/**
 * xprt_complete_rqst - called when reply processing is complete
907
 * @task: RPC request that recently completed
908 909
 * @copied: actual number of bytes received from the transport
 *
910
 * Caller holds transport lock.
L
Linus Torvalds 已提交
911
 */
912
void xprt_complete_rqst(struct rpc_task *task, int copied)
L
Linus Torvalds 已提交
913
{
914
	struct rpc_rqst *req = task->tk_rqstp;
915
	struct rpc_xprt *xprt = req->rq_xprt;
L
Linus Torvalds 已提交
916

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

921
	xprt->stat.recvs++;
922
	req->rq_rtt = ktime_sub(ktime_get(), req->rq_xtime);
923 924
	if (xprt->ops->timer != NULL)
		xprt_update_rtt(task);
925

L
Linus Torvalds 已提交
926
	list_del_init(&req->rq_list);
927
	req->rq_private_buf.len = copied;
928 929
	/* Ensure all writes are done before we update */
	/* req->rq_reply_bytes_recvd */
930
	smp_wmb();
931
	req->rq_reply_bytes_recvd = copied;
932
	rpc_wake_up_queued_task(&xprt->pending, task);
L
Linus Torvalds 已提交
933
}
934
EXPORT_SYMBOL_GPL(xprt_complete_rqst);
L
Linus Torvalds 已提交
935

936
static void xprt_timer(struct rpc_task *task)
L
Linus Torvalds 已提交
937
{
938
	struct rpc_rqst *req = task->tk_rqstp;
L
Linus Torvalds 已提交
939 940
	struct rpc_xprt *xprt = req->rq_xprt;

941 942
	if (task->tk_status != -ETIMEDOUT)
		return;
943
	dprintk("RPC: %5u xprt_timer\n", task->tk_pid);
L
Linus Torvalds 已提交
944

945
	if (!req->rq_reply_bytes_recvd) {
946
		if (xprt->ops->timer)
947
			xprt->ops->timer(xprt, task);
948 949
	} else
		task->tk_status = 0;
L
Linus Torvalds 已提交
950 951
}

952 953 954 955
/**
 * xprt_prepare_transmit - reserve the transport before sending a request
 * @task: RPC task about to send a request
 *
L
Linus Torvalds 已提交
956
 */
957
bool xprt_prepare_transmit(struct rpc_task *task)
L
Linus Torvalds 已提交
958 959 960
{
	struct rpc_rqst	*req = task->tk_rqstp;
	struct rpc_xprt	*xprt = req->rq_xprt;
961
	bool ret = false;
L
Linus Torvalds 已提交
962

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

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

989
void xprt_end_transmit(struct rpc_task *task)
990
{
991
	xprt_release_write(task->tk_rqstp->rq_xprt, task);
992 993
}

994 995 996 997 998 999 1000
/**
 * 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 已提交
1001 1002 1003
{
	struct rpc_rqst	*req = task->tk_rqstp;
	struct rpc_xprt	*xprt = req->rq_xprt;
1004
	int status, numreqs;
L
Linus Torvalds 已提交
1005

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

1008
	if (!req->rq_reply_bytes_recvd) {
1009 1010 1011 1012
		if (list_empty(&req->rq_list) && rpc_reply_expected(task)) {
			/*
			 * Add to the list only if we're expecting a reply
			 */
L
Linus Torvalds 已提交
1013 1014 1015 1016
			/* 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 */
1017
			spin_lock(&xprt->recv_lock);
L
Linus Torvalds 已提交
1018
			list_add_tail(&req->rq_list, &xprt->recv);
1019
			spin_unlock(&xprt->recv_lock);
L
Linus Torvalds 已提交
1020
			xprt_reset_majortimeo(req);
1021 1022
			/* Turn off autodisconnect */
			del_singleshot_timer_sync(&xprt->timer);
L
Linus Torvalds 已提交
1023 1024 1025 1026
		}
	} else if (!req->rq_bytes_sent)
		return;

1027
	req->rq_xtime = ktime_get();
1028
	status = xprt->ops->send_request(task);
1029
	trace_xprt_transmit(xprt, req->rq_xid, status);
1030 1031 1032 1033
	if (status != 0) {
		task->tk_status = status;
		return;
	}
C
Chuck Lever 已提交
1034
	xprt_inject_disconnect(xprt);
1035

1036
	dprintk("RPC: %5u xmit complete\n", task->tk_pid);
1037
	task->tk_flags |= RPC_TASK_SENT;
1038
	spin_lock_bh(&xprt->transport_lock);
1039

1040
	xprt->ops->set_retrans_timeout(task);
1041

1042 1043 1044
	numreqs = atomic_read(&xprt->num_reqs);
	if (numreqs > xprt->stat.max_slots)
		xprt->stat.max_slots = numreqs;
1045 1046 1047
	xprt->stat.sends++;
	xprt->stat.req_u += xprt->stat.sends - xprt->stat.recvs;
	xprt->stat.bklog_u += xprt->backlog.qlen;
1048 1049
	xprt->stat.sending_u += xprt->sending.qlen;
	xprt->stat.pending_u += xprt->pending.qlen;
L
Linus Torvalds 已提交
1050

1051 1052 1053
	/* Don't race with disconnect */
	if (!xprt_connected(xprt))
		task->tk_status = -ENOTCONN;
1054
	else {
1055 1056 1057 1058
		/*
		 * Sleep on the pending queue since
		 * we're expecting a reply.
		 */
1059 1060 1061
		if (!req->rq_reply_bytes_recvd && rpc_reply_expected(task))
			rpc_sleep_on(&xprt->pending, task, xprt_timer);
		req->rq_connect_cookie = xprt->connect_cookie;
1062
	}
1063
	spin_unlock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
1064 1065
}

1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093
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;
}

1094
static struct rpc_rqst *xprt_dynamic_alloc_slot(struct rpc_xprt *xprt)
1095 1096 1097 1098 1099
{
	struct rpc_rqst *req = ERR_PTR(-EAGAIN);

	if (!atomic_add_unless(&xprt->num_reqs, 1, xprt->max_reqs))
		goto out;
1100 1101 1102
	spin_unlock(&xprt->reserve_lock);
	req = kzalloc(sizeof(struct rpc_rqst), GFP_NOFS);
	spin_lock(&xprt->reserve_lock);
1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119
	if (req != NULL)
		goto out;
	atomic_dec(&xprt->num_reqs);
	req = ERR_PTR(-ENOMEM);
out:
	return req;
}

static bool xprt_dynamic_free_slot(struct rpc_xprt *xprt, struct rpc_rqst *req)
{
	if (atomic_add_unless(&xprt->num_reqs, -1, xprt->min_reqs)) {
		kfree(req);
		return true;
	}
	return false;
}

1120
void xprt_alloc_slot(struct rpc_xprt *xprt, struct rpc_task *task)
L
Linus Torvalds 已提交
1121
{
1122
	struct rpc_rqst *req;
L
Linus Torvalds 已提交
1123

1124
	spin_lock(&xprt->reserve_lock);
L
Linus Torvalds 已提交
1125
	if (!list_empty(&xprt->free)) {
1126 1127 1128 1129
		req = list_entry(xprt->free.next, struct rpc_rqst, rq_list);
		list_del(&req->rq_list);
		goto out_init_req;
	}
1130
	req = xprt_dynamic_alloc_slot(xprt);
1131 1132 1133 1134 1135 1136
	if (!IS_ERR(req))
		goto out_init_req;
	switch (PTR_ERR(req)) {
	case -ENOMEM:
		dprintk("RPC:       dynamic allocation of request slot "
				"failed! Retrying\n");
1137
		task->tk_status = -ENOMEM;
1138 1139
		break;
	case -EAGAIN:
1140
		xprt_add_backlog(xprt, task);
1141
		dprintk("RPC:       waiting for request slot\n");
1142
		/* fall through */
1143 1144
	default:
		task->tk_status = -EAGAIN;
L
Linus Torvalds 已提交
1145
	}
1146
	spin_unlock(&xprt->reserve_lock);
1147 1148 1149 1150 1151
	return;
out_init_req:
	task->tk_status = 0;
	task->tk_rqstp = req;
	xprt_request_init(task, xprt);
1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166
	spin_unlock(&xprt->reserve_lock);
}
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 已提交
1167
}
1168
EXPORT_SYMBOL_GPL(xprt_lock_and_alloc_slot);
L
Linus Torvalds 已提交
1169

1170 1171 1172
static void xprt_free_slot(struct rpc_xprt *xprt, struct rpc_rqst *req)
{
	spin_lock(&xprt->reserve_lock);
1173 1174 1175 1176
	if (!xprt_dynamic_free_slot(xprt, req)) {
		memset(req, 0, sizeof(*req));	/* mark unused */
		list_add(&req->rq_list, &xprt->free);
	}
1177
	xprt_wake_up_backlog(xprt);
1178 1179 1180
	spin_unlock(&xprt->reserve_lock);
}

1181 1182 1183 1184 1185 1186 1187 1188 1189 1190
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);
	}
}

1191 1192 1193
struct rpc_xprt *xprt_alloc(struct net *net, size_t size,
		unsigned int num_prealloc,
		unsigned int max_alloc)
1194 1195
{
	struct rpc_xprt *xprt;
1196 1197
	struct rpc_rqst *req;
	int i;
1198 1199 1200 1201 1202

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

1203 1204 1205 1206 1207
	xprt_init(xprt, net);

	for (i = 0; i < num_prealloc; i++) {
		req = kzalloc(sizeof(struct rpc_rqst), GFP_KERNEL);
		if (!req)
1208
			goto out_free;
1209 1210
		list_add(&req->rq_list, &xprt->free);
	}
1211 1212 1213 1214 1215 1216
	if (max_alloc > num_prealloc)
		xprt->max_reqs = max_alloc;
	else
		xprt->max_reqs = num_prealloc;
	xprt->min_reqs = num_prealloc;
	atomic_set(&xprt->num_reqs, num_prealloc);
1217 1218 1219 1220

	return xprt;

out_free:
1221
	xprt_free(xprt);
1222 1223 1224 1225 1226
out:
	return NULL;
}
EXPORT_SYMBOL_GPL(xprt_alloc);

1227 1228
void xprt_free(struct rpc_xprt *xprt)
{
P
Pavel Emelyanov 已提交
1229
	put_net(xprt->xprt_net);
1230
	xprt_free_all_slots(xprt);
1231
	kfree_rcu(xprt, rcu);
1232 1233 1234
}
EXPORT_SYMBOL_GPL(xprt_free);

1235 1236 1237 1238
/**
 * xprt_reserve - allocate an RPC request slot
 * @task: RPC task requesting a slot allocation
 *
1239 1240
 * If the transport is marked as being congested, or if no more
 * slots are available, place the task on the transport's
1241 1242 1243
 * backlog queue.
 */
void xprt_reserve(struct rpc_task *task)
L
Linus Torvalds 已提交
1244
{
1245
	struct rpc_xprt *xprt = task->tk_xprt;
L
Linus Torvalds 已提交
1246

1247 1248 1249 1250 1251 1252
	task->tk_status = 0;
	if (task->tk_rqstp != NULL)
		return;

	task->tk_timeout = 0;
	task->tk_status = -EAGAIN;
1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267
	if (!xprt_throttle_congested(xprt, task))
		xprt->ops->alloc_slot(xprt, task);
}

/**
 * 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)
{
1268
	struct rpc_xprt *xprt = task->tk_xprt;
1269 1270 1271 1272 1273 1274 1275

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

	task->tk_timeout = 0;
	task->tk_status = -EAGAIN;
1276
	xprt->ops->alloc_slot(xprt, task);
L
Linus Torvalds 已提交
1277 1278
}

1279
static inline __be32 xprt_alloc_xid(struct rpc_xprt *xprt)
L
Linus Torvalds 已提交
1280
{
1281
	return (__force __be32)xprt->xid++;
L
Linus Torvalds 已提交
1282 1283 1284 1285
}

static inline void xprt_init_xid(struct rpc_xprt *xprt)
{
1286
	xprt->xid = prandom_u32();
L
Linus Torvalds 已提交
1287 1288
}

1289
static void xprt_request_init(struct rpc_task *task, struct rpc_xprt *xprt)
L
Linus Torvalds 已提交
1290 1291 1292
{
	struct rpc_rqst	*req = task->tk_rqstp;

1293
	INIT_LIST_HEAD(&req->rq_list);
1294
	req->rq_timeout = task->tk_client->cl_timeout->to_initval;
L
Linus Torvalds 已提交
1295 1296
	req->rq_task	= task;
	req->rq_xprt    = xprt;
1297
	req->rq_buffer  = NULL;
L
Linus Torvalds 已提交
1298
	req->rq_xid     = xprt_alloc_xid(xprt);
1299
	req->rq_connect_cookie = xprt->connect_cookie - 1;
1300 1301 1302 1303 1304
	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;
1305
	req->rq_release_snd_buf = NULL;
1306
	xprt_reset_majortimeo(req);
1307
	dprintk("RPC: %5u reserved req %p xid %08x\n", task->tk_pid,
L
Linus Torvalds 已提交
1308 1309 1310
			req, ntohl(req->rq_xid));
}

1311 1312 1313 1314
/**
 * xprt_release - release an RPC request slot
 * @task: task which is finished with the slot
 *
L
Linus Torvalds 已提交
1315
 */
1316
void xprt_release(struct rpc_task *task)
L
Linus Torvalds 已提交
1317
{
1318
	struct rpc_xprt	*xprt;
1319
	struct rpc_rqst	*req = task->tk_rqstp;
L
Linus Torvalds 已提交
1320

1321 1322
	if (req == NULL) {
		if (task->tk_client) {
1323
			xprt = task->tk_xprt;
1324 1325 1326
			if (xprt->snd_task == task)
				xprt_release_write(xprt, task);
		}
L
Linus Torvalds 已提交
1327
		return;
1328
	}
1329 1330

	xprt = req->rq_xprt;
1331 1332 1333 1334
	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);
1335 1336 1337 1338 1339 1340
	spin_lock(&xprt->recv_lock);
	if (!list_empty(&req->rq_list)) {
		list_del(&req->rq_list);
		xprt_wait_on_pinned_rqst(req);
	}
	spin_unlock(&xprt->recv_lock);
C
Chuck Lever 已提交
1341
	spin_lock_bh(&xprt->transport_lock);
1342
	xprt->ops->release_xprt(xprt, task);
1343 1344
	if (xprt->ops->release_request)
		xprt->ops->release_request(task);
L
Linus Torvalds 已提交
1345
	xprt->last_used = jiffies;
1346
	xprt_schedule_autodisconnect(xprt);
C
Chuck Lever 已提交
1347
	spin_unlock_bh(&xprt->transport_lock);
1348
	if (req->rq_buffer)
1349
		xprt->ops->buf_free(task);
C
Chuck Lever 已提交
1350
	xprt_inject_disconnect(xprt);
1351 1352
	if (req->rq_cred != NULL)
		put_rpccred(req->rq_cred);
L
Linus Torvalds 已提交
1353
	task->tk_rqstp = NULL;
1354 1355
	if (req->rq_release_snd_buf)
		req->rq_release_snd_buf(req);
1356

1357
	dprintk("RPC: %5u release request %p\n", task->tk_pid, req);
1358 1359 1360
	if (likely(!bc_prealloc(req)))
		xprt_free_slot(xprt, req);
	else
1361
		xprt_free_bc_request(req);
L
Linus Torvalds 已提交
1362 1363
}

1364
static void xprt_init(struct rpc_xprt *xprt, struct net *net)
1365
{
1366
	kref_init(&xprt->kref);
1367 1368 1369

	spin_lock_init(&xprt->transport_lock);
	spin_lock_init(&xprt->reserve_lock);
1370
	spin_lock_init(&xprt->recv_lock);
1371 1372 1373

	INIT_LIST_HEAD(&xprt->free);
	INIT_LIST_HEAD(&xprt->recv);
1374
#if defined(CONFIG_SUNRPC_BACKCHANNEL)
1375 1376
	spin_lock_init(&xprt->bc_pa_lock);
	INIT_LIST_HEAD(&xprt->bc_pa_list);
1377
#endif /* CONFIG_SUNRPC_BACKCHANNEL */
1378
	INIT_LIST_HEAD(&xprt->xprt_switch);
1379

1380 1381
	xprt->last_used = jiffies;
	xprt->cwnd = RPC_INITCWND;
1382
	xprt->bind_index = 0;
1383 1384 1385

	rpc_init_wait_queue(&xprt->binding, "xprt_binding");
	rpc_init_wait_queue(&xprt->pending, "xprt_pending");
1386
	rpc_init_priority_wait_queue(&xprt->sending, "xprt_sending");
1387 1388 1389 1390
	rpc_init_priority_wait_queue(&xprt->backlog, "xprt_backlog");

	xprt_init_xid(xprt);

1391
	xprt->xprt_net = get_net(net);
1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411
}

/**
 * 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);
1412
	dprintk("RPC: transport (%d) not supported\n", args->ident);
1413 1414 1415 1416 1417 1418 1419
	return ERR_PTR(-EIO);

found:
	xprt = t->setup(args);
	if (IS_ERR(xprt)) {
		dprintk("RPC:       xprt_create_transport: failed, %ld\n",
				-PTR_ERR(xprt));
1420
		goto out;
1421
	}
1422 1423
	if (args->flags & XPRT_CREATE_NO_IDLE_TIMEOUT)
		xprt->idle_timeout = 0;
1424 1425 1426 1427 1428 1429
	INIT_WORK(&xprt->task_cleanup, xprt_autoclose);
	if (xprt_has_timer(xprt))
		setup_timer(&xprt->timer, xprt_init_autodisconnect,
			    (unsigned long)xprt);
	else
		init_timer(&xprt->timer);
1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440

	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);
	}

1441
	rpc_xprt_debugfs_register(xprt);
1442

1443
	dprintk("RPC:       created transport %p with %u slots\n", xprt,
1444
			xprt->max_reqs);
1445
out:
1446 1447 1448
	return xprt;
}

1449 1450
/**
 * xprt_destroy - destroy an RPC transport, killing off all requests.
1451
 * @xprt: transport to destroy
1452
 *
L
Linus Torvalds 已提交
1453
 */
1454
static void xprt_destroy(struct rpc_xprt *xprt)
L
Linus Torvalds 已提交
1455
{
1456
	dprintk("RPC:       destroying transport %p\n", xprt);
1457 1458 1459 1460

	/* Exclude transport connect/disconnect handlers */
	wait_on_bit_lock(&xprt->state, XPRT_LOCKED, TASK_UNINTERRUPTIBLE);

1461
	del_timer_sync(&xprt->timer);
1462

1463
	rpc_xprt_debugfs_unregister(xprt);
1464 1465 1466 1467
	rpc_destroy_wait_queue(&xprt->binding);
	rpc_destroy_wait_queue(&xprt->pending);
	rpc_destroy_wait_queue(&xprt->sending);
	rpc_destroy_wait_queue(&xprt->backlog);
1468
	cancel_work_sync(&xprt->task_cleanup);
1469
	kfree(xprt->servername);
1470 1471 1472
	/*
	 * Tear down transport state and free the rpc_xprt
	 */
1473
	xprt->ops->destroy(xprt);
1474
}
L
Linus Torvalds 已提交
1475

1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493
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);

1494 1495 1496 1497 1498 1499 1500
/**
 * xprt_put - release a reference to an RPC transport.
 * @xprt: pointer to the transport
 *
 */
void xprt_put(struct rpc_xprt *xprt)
{
1501 1502
	if (xprt != NULL)
		kref_put(&xprt->kref, xprt_destroy_kref);
1503
}
1504
EXPORT_SYMBOL_GPL(xprt_put);