xprt.c 25.1 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 15
 *	the request struct, and calls xprt_transmit().
 *  -	xprt_transmit sends the message and installs the caller on the
 *	transport's wait list. At the same time, it installs a timer that
L
Linus Torvalds 已提交
16 17
 *	is run after the packet's timeout has expired.
 *  -	When a packet arrives, the data_ready handler walks the list of
18
 *	pending requests for that transport. If a matching XID is found, the
L
Linus Torvalds 已提交
19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
 *	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>
35 36
 *
 *  Transport switch API copyright (C) 2005, Chuck Lever <cel@netapp.com>
L
Linus Torvalds 已提交
37 38
 */

39 40
#include <linux/module.h>

L
Linus Torvalds 已提交
41
#include <linux/types.h>
42
#include <linux/interrupt.h>
L
Linus Torvalds 已提交
43 44 45
#include <linux/workqueue.h>
#include <linux/random.h>

46
#include <linux/sunrpc/clnt.h>
L
Linus Torvalds 已提交
47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66

/*
 * Local variables
 */

#ifdef RPC_DEBUG
# undef  RPC_DEBUG_DATA
# define RPCDBG_FACILITY	RPCDBG_XPRT
#endif

/*
 * Local functions
 */
static void	xprt_request_init(struct rpc_task *, struct rpc_xprt *);
static inline void	do_xprt_reserve(struct rpc_task *);
static void	xprt_connect_status(struct rpc_task *task);
static int      __xprt_get_cong(struct rpc_xprt *, struct rpc_task *);

static int	xprt_clear_backlog(struct rpc_xprt *xprt);

67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105
/**
 * xprt_reserve_xprt - serialize write access to transports
 * @task: task that is requesting access to the transport
 *
 * This prevents mixing the payload of separate requests, and prevents
 * transport connects from colliding with writes.  No congestion control
 * is provided.
 */
int xprt_reserve_xprt(struct rpc_task *task)
{
	struct rpc_xprt	*xprt = task->tk_xprt;
	struct rpc_rqst *req = task->tk_rqstp;

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

out_sleep:
	dprintk("RPC: %4d failed to lock transport %p\n",
			task->tk_pid, xprt);
	task->tk_timeout = 0;
	task->tk_status = -EAGAIN;
	if (req && req->rq_ntrans)
		rpc_sleep_on(&xprt->resend, task, NULL, NULL);
	else
		rpc_sleep_on(&xprt->sending, task, NULL, NULL);
	return 0;
}

L
Linus Torvalds 已提交
106
/*
107 108 109 110 111 112
 * 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 已提交
113
 */
114
int xprt_reserve_xprt_cong(struct rpc_task *task)
L
Linus Torvalds 已提交
115
{
116
	struct rpc_xprt	*xprt = task->tk_xprt;
L
Linus Torvalds 已提交
117 118
	struct rpc_rqst *req = task->tk_rqstp;

119
	if (test_and_set_bit(XPRT_LOCKED, &xprt->state)) {
L
Linus Torvalds 已提交
120 121 122 123
		if (task == xprt->snd_task)
			return 1;
		goto out_sleep;
	}
124
	if (__xprt_get_cong(xprt, task)) {
L
Linus Torvalds 已提交
125 126 127 128 129 130 131 132
		xprt->snd_task = task;
		if (req) {
			req->rq_bytes_sent = 0;
			req->rq_ntrans++;
		}
		return 1;
	}
	smp_mb__before_clear_bit();
133
	clear_bit(XPRT_LOCKED, &xprt->state);
L
Linus Torvalds 已提交
134 135
	smp_mb__after_clear_bit();
out_sleep:
136
	dprintk("RPC: %4d failed to lock transport %p\n", task->tk_pid, xprt);
L
Linus Torvalds 已提交
137 138 139 140 141 142 143 144 145
	task->tk_timeout = 0;
	task->tk_status = -EAGAIN;
	if (req && req->rq_ntrans)
		rpc_sleep_on(&xprt->resend, task, NULL, NULL);
	else
		rpc_sleep_on(&xprt->sending, task, NULL, NULL);
	return 0;
}

146
static inline int xprt_lock_write(struct rpc_xprt *xprt, struct rpc_task *task)
L
Linus Torvalds 已提交
147 148 149
{
	int retval;

C
Chuck Lever 已提交
150
	spin_lock_bh(&xprt->transport_lock);
151
	retval = xprt->ops->reserve_xprt(task);
C
Chuck Lever 已提交
152
	spin_unlock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
153 154 155
	return retval;
}

156
static void __xprt_lock_write_next(struct rpc_xprt *xprt)
157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185
{
	struct rpc_task *task;
	struct rpc_rqst *req;

	if (test_and_set_bit(XPRT_LOCKED, &xprt->state))
		return;

	task = rpc_wake_up_next(&xprt->resend);
	if (!task) {
		task = rpc_wake_up_next(&xprt->sending);
		if (!task)
			goto out_unlock;
	}

	req = task->tk_rqstp;
	xprt->snd_task = task;
	if (req) {
		req->rq_bytes_sent = 0;
		req->rq_ntrans++;
	}
	return;

out_unlock:
	smp_mb__before_clear_bit();
	clear_bit(XPRT_LOCKED, &xprt->state);
	smp_mb__after_clear_bit();
}

static void __xprt_lock_write_next_cong(struct rpc_xprt *xprt)
L
Linus Torvalds 已提交
186 187 188
{
	struct rpc_task *task;

189
	if (test_and_set_bit(XPRT_LOCKED, &xprt->state))
L
Linus Torvalds 已提交
190
		return;
191
	if (RPCXPRT_CONGESTED(xprt))
L
Linus Torvalds 已提交
192 193 194 195 196 197 198
		goto out_unlock;
	task = rpc_wake_up_next(&xprt->resend);
	if (!task) {
		task = rpc_wake_up_next(&xprt->sending);
		if (!task)
			goto out_unlock;
	}
199
	if (__xprt_get_cong(xprt, task)) {
L
Linus Torvalds 已提交
200 201 202 203 204 205 206 207 208 209
		struct rpc_rqst *req = task->tk_rqstp;
		xprt->snd_task = task;
		if (req) {
			req->rq_bytes_sent = 0;
			req->rq_ntrans++;
		}
		return;
	}
out_unlock:
	smp_mb__before_clear_bit();
210
	clear_bit(XPRT_LOCKED, &xprt->state);
L
Linus Torvalds 已提交
211 212 213
	smp_mb__after_clear_bit();
}

214 215 216 217 218 219
/**
 * 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 已提交
220
 */
221
void xprt_release_xprt(struct rpc_xprt *xprt, struct rpc_task *task)
L
Linus Torvalds 已提交
222 223 224 225
{
	if (xprt->snd_task == task) {
		xprt->snd_task = NULL;
		smp_mb__before_clear_bit();
226
		clear_bit(XPRT_LOCKED, &xprt->state);
L
Linus Torvalds 已提交
227 228 229 230 231
		smp_mb__after_clear_bit();
		__xprt_lock_write_next(xprt);
	}
}

232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251
/**
 * 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) {
		xprt->snd_task = NULL;
		smp_mb__before_clear_bit();
		clear_bit(XPRT_LOCKED, &xprt->state);
		smp_mb__after_clear_bit();
		__xprt_lock_write_next_cong(xprt);
	}
}

static inline void xprt_release_write(struct rpc_xprt *xprt, struct rpc_task *task)
L
Linus Torvalds 已提交
252
{
C
Chuck Lever 已提交
253
	spin_lock_bh(&xprt->transport_lock);
254
	xprt->ops->release_xprt(xprt, task);
C
Chuck Lever 已提交
255
	spin_unlock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288
}

/*
 * 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;
	dprintk("RPC: %4d xprt_cwnd_limited cong = %ld cwnd = %ld\n",
			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;
289
	__xprt_lock_write_next_cong(xprt);
L
Linus Torvalds 已提交
290 291
}

292 293 294 295 296 297 298 299 300 301 302
/**
 * 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)
{
	__xprt_put_cong(task->tk_xprt, task->tk_rqstp);
}

303 304 305 306 307
/**
 * xprt_adjust_cwnd - adjust transport congestion window
 * @task: recently completed RPC request used to adjust window
 * @result: result code of completed RPC request
 *
L
Linus Torvalds 已提交
308 309
 * We use a time-smoothed congestion estimator to avoid heavy oscillation.
 */
310
void xprt_adjust_cwnd(struct rpc_task *task, int result)
L
Linus Torvalds 已提交
311
{
312 313 314
	struct rpc_rqst *req = task->tk_rqstp;
	struct rpc_xprt *xprt = task->tk_xprt;
	unsigned long cwnd = xprt->cwnd;
L
Linus Torvalds 已提交
315 316 317 318 319 320 321

	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);
322
		__xprt_lock_write_next_cong(xprt);
L
Linus Torvalds 已提交
323 324 325 326 327 328 329 330
	} else if (result == -ETIMEDOUT) {
		cwnd >>= 1;
		if (cwnd < RPC_CWNDSCALE)
			cwnd = RPC_CWNDSCALE;
	}
	dprintk("RPC:      cong %ld, cwnd was %ld, now %ld\n",
			xprt->cong, xprt->cwnd, cwnd);
	xprt->cwnd = cwnd;
331
	__xprt_put_cong(xprt, req);
L
Linus Torvalds 已提交
332 333
}

334 335 336 337 338 339 340 341 342 343 344 345 346 347
/**
 * 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);
}

348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381
/**
 * xprt_wait_for_buffer_space - wait for transport output buffer to clear
 * @task: task to be put to sleep
 *
 */
void xprt_wait_for_buffer_space(struct rpc_task *task)
{
	struct rpc_rqst *req = task->tk_rqstp;
	struct rpc_xprt *xprt = req->rq_xprt;

	task->tk_timeout = req->rq_timeout;
	rpc_sleep_on(&xprt->pending, task, NULL, NULL);
}

/**
 * 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)
{
	if (unlikely(xprt->shutdown))
		return;

	spin_lock_bh(&xprt->transport_lock);
	if (xprt->snd_task) {
		dprintk("RPC:      write space: waking waiting task on xprt %p\n",
				xprt);
		rpc_wake_up_task(xprt->snd_task);
	}
	spin_unlock_bh(&xprt->transport_lock);
}

382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413
/**
 * 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;
}

/*
 * xprt_set_retrans_timeout_rtt - set a request's retransmit timeout
 * @task: task whose timeout is to be set
 * 
 * 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;
	struct rpc_rtt *rtt = task->tk_client->cl_rtt;
	struct rpc_rqst *req = task->tk_rqstp;
	unsigned long max_timeout = req->rq_xprt->timeout.to_maxval;

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

L
Linus Torvalds 已提交
414 415 416 417 418 419 420 421 422 423 424 425 426 427
static void xprt_reset_majortimeo(struct rpc_rqst *req)
{
	struct rpc_timeout *to = &req->rq_xprt->timeout;

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

428 429 430 431
/**
 * xprt_adjust_timeout - adjust timeout values for next retransmit
 * @req: RPC request containing parameters to use for the adjustment
 *
L
Linus Torvalds 已提交
432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452
 */
int xprt_adjust_timeout(struct rpc_rqst *req)
{
	struct rpc_xprt *xprt = req->rq_xprt;
	struct rpc_timeout *to = &xprt->timeout;
	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++;
		pprintk("RPC: %lu retrans\n", jiffies);
	} else {
		req->rq_timeout = to->to_initval;
		req->rq_retries = 0;
		xprt_reset_majortimeo(req);
		/* Reset the RTT counters == "slow start" */
C
Chuck Lever 已提交
453
		spin_lock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
454
		rpc_init_rtt(req->rq_task->tk_client->cl_rtt, to->to_initval);
C
Chuck Lever 已提交
455
		spin_unlock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
456 457 458 459 460 461 462 463 464 465 466
		pprintk("RPC: %lu timeout\n", jiffies);
		status = -ETIMEDOUT;
	}

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

467
static void xprt_autoclose(void *args)
L
Linus Torvalds 已提交
468 469 470 471
{
	struct rpc_xprt *xprt = (struct rpc_xprt *)args;

	xprt_disconnect(xprt);
472
	xprt->ops->close(xprt);
L
Linus Torvalds 已提交
473 474 475
	xprt_release_write(xprt, NULL);
}

476 477 478 479
/**
 * xprt_disconnect - mark a transport as disconnected
 * @xprt: transport to flag for disconnect
 *
L
Linus Torvalds 已提交
480
 */
481
void xprt_disconnect(struct rpc_xprt *xprt)
L
Linus Torvalds 已提交
482 483
{
	dprintk("RPC:      disconnected transport %p\n", xprt);
C
Chuck Lever 已提交
484
	spin_lock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
485
	xprt_clear_connected(xprt);
486
	xprt_wake_pending_tasks(xprt, -ENOTCONN);
C
Chuck Lever 已提交
487
	spin_unlock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
488 489 490 491 492 493 494
}

static void
xprt_init_autodisconnect(unsigned long data)
{
	struct rpc_xprt *xprt = (struct rpc_xprt *)data;

C
Chuck Lever 已提交
495
	spin_lock(&xprt->transport_lock);
L
Linus Torvalds 已提交
496 497
	if (!list_empty(&xprt->recv) || xprt->shutdown)
		goto out_abort;
498
	if (test_and_set_bit(XPRT_LOCKED, &xprt->state))
L
Linus Torvalds 已提交
499
		goto out_abort;
C
Chuck Lever 已提交
500
	spin_unlock(&xprt->transport_lock);
501
	if (xprt_connecting(xprt))
L
Linus Torvalds 已提交
502 503 504 505 506
		xprt_release_write(xprt, NULL);
	else
		schedule_work(&xprt->task_cleanup);
	return;
out_abort:
C
Chuck Lever 已提交
507
	spin_unlock(&xprt->transport_lock);
L
Linus Torvalds 已提交
508 509
}

510 511 512
/**
 * xprt_connect - schedule a transport connect operation
 * @task: RPC task that is requesting the connect
L
Linus Torvalds 已提交
513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532
 *
 */
void xprt_connect(struct rpc_task *task)
{
	struct rpc_xprt	*xprt = task->tk_xprt;

	dprintk("RPC: %4d xprt_connect xprt %p %s connected\n", task->tk_pid,
			xprt, (xprt_connected(xprt) ? "is" : "is not"));

	if (xprt->shutdown) {
		task->tk_status = -EIO;
		return;
	}
	if (!xprt->addr.sin_port) {
		task->tk_status = -EIO;
		return;
	}
	if (!xprt_lock_write(xprt, task))
		return;
	if (xprt_connected(xprt))
533 534 535 536
		xprt_release_write(xprt, task);
	else {
		if (task->tk_rqstp)
			task->tk_rqstp->rq_bytes_sent = 0;
L
Linus Torvalds 已提交
537

538 539 540
		task->tk_timeout = RPC_CONNECT_TIMEOUT;
		rpc_sleep_on(&xprt->pending, task, xprt_connect_status, NULL);
		xprt->ops->connect(task);
L
Linus Torvalds 已提交
541 542 543 544
	}
	return;
}

545
static void xprt_connect_status(struct rpc_task *task)
L
Linus Torvalds 已提交
546 547 548 549 550 551 552 553 554 555 556 557
{
	struct rpc_xprt	*xprt = task->tk_xprt;

	if (task->tk_status >= 0) {
		dprintk("RPC: %4d xprt_connect_status: connection established\n",
				task->tk_pid);
		return;
	}

	switch (task->tk_status) {
	case -ECONNREFUSED:
	case -ECONNRESET:
558 559 560
		dprintk("RPC: %4d xprt_connect_status: server %s refused connection\n",
				task->tk_pid, task->tk_client->cl_server);
		break;
L
Linus Torvalds 已提交
561
	case -ENOTCONN:
562 563 564
		dprintk("RPC: %4d xprt_connect_status: connection broken\n",
				task->tk_pid);
		break;
L
Linus Torvalds 已提交
565
	case -ETIMEDOUT:
566
		dprintk("RPC: %4d xprt_connect_status: connect attempt timed out\n",
L
Linus Torvalds 已提交
567 568 569
				task->tk_pid);
		break;
	default:
570 571 572 573 574 575 576 577 578 579 580
		dprintk("RPC: %4d xprt_connect_status: error %d connecting to server %s\n",
				task->tk_pid, -task->tk_status, task->tk_client->cl_server);
		xprt_release_write(xprt, task);
		task->tk_status = -EIO;
		return;
	}

	/* if soft mounted, just cause this RPC to fail */
	if (RPC_IS_SOFT(task)) {
		xprt_release_write(xprt, task);
		task->tk_status = -EIO;
L
Linus Torvalds 已提交
581 582 583
	}
}

584 585 586 587 588
/**
 * 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 已提交
589
 */
590
struct rpc_rqst *xprt_lookup_rqst(struct rpc_xprt *xprt, u32 xid)
L
Linus Torvalds 已提交
591 592 593 594 595 596 597 598 599 600 601 602 603 604
{
	struct list_head *pos;
	struct rpc_rqst	*req = NULL;

	list_for_each(pos, &xprt->recv) {
		struct rpc_rqst *entry = list_entry(pos, struct rpc_rqst, rq_list);
		if (entry->rq_xid == xid) {
			req = entry;
			break;
		}
	}
	return req;
}

605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623
/**
 * xprt_update_rtt - update an RPC client's RTT state after receiving a reply
 * @task: RPC request that recently completed
 *
 */
void xprt_update_rtt(struct rpc_task *task)
{
	struct rpc_rqst *req = task->tk_rqstp;
	struct rpc_rtt *rtt = task->tk_client->cl_rtt;
	unsigned timer = task->tk_msg.rpc_proc->p_timer;

	if (timer) {
		if (req->rq_ntrans == 1)
			rpc_update_rtt(rtt, timer,
					(long)jiffies - req->rq_xtime);
		rpc_set_timeo(rtt, timer, req->rq_ntrans - 1);
	}
}

624 625
/**
 * xprt_complete_rqst - called when reply processing is complete
626
 * @task: RPC request that recently completed
627 628
 * @copied: actual number of bytes received from the transport
 *
629
 * Caller holds transport lock.
L
Linus Torvalds 已提交
630
 */
631
void xprt_complete_rqst(struct rpc_task *task, int copied)
L
Linus Torvalds 已提交
632
{
633
	struct rpc_rqst *req = task->tk_rqstp;
L
Linus Torvalds 已提交
634

635 636
	dprintk("RPC: %5u xid %08x complete (%d bytes received)\n",
			task->tk_pid, ntohl(req->rq_xid), copied);
L
Linus Torvalds 已提交
637 638 639 640 641 642

	list_del_init(&req->rq_list);
	req->rq_received = req->rq_private_buf.len = copied;
	rpc_wake_up_task(task);
}

643
static void xprt_timer(struct rpc_task *task)
L
Linus Torvalds 已提交
644
{
645
	struct rpc_rqst *req = task->tk_rqstp;
L
Linus Torvalds 已提交
646 647
	struct rpc_xprt *xprt = req->rq_xprt;

648
	dprintk("RPC: %4d xprt_timer\n", task->tk_pid);
L
Linus Torvalds 已提交
649

650 651 652 653 654 655
	spin_lock(&xprt->transport_lock);
	if (!req->rq_received) {
		if (xprt->ops->timer)
			xprt->ops->timer(task);
		task->tk_status = -ETIMEDOUT;
	}
L
Linus Torvalds 已提交
656 657
	task->tk_timeout = 0;
	rpc_wake_up_task(task);
C
Chuck Lever 已提交
658
	spin_unlock(&xprt->transport_lock);
L
Linus Torvalds 已提交
659 660
}

661 662 663 664
/**
 * xprt_prepare_transmit - reserve the transport before sending a request
 * @task: RPC task about to send a request
 *
L
Linus Torvalds 已提交
665
 */
666
int xprt_prepare_transmit(struct rpc_task *task)
L
Linus Torvalds 已提交
667 668 669 670 671 672 673 674 675 676
{
	struct rpc_rqst	*req = task->tk_rqstp;
	struct rpc_xprt	*xprt = req->rq_xprt;
	int err = 0;

	dprintk("RPC: %4d xprt_prepare_transmit\n", task->tk_pid);

	if (xprt->shutdown)
		return -EIO;

C
Chuck Lever 已提交
677
	spin_lock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
678 679 680 681
	if (req->rq_received && !req->rq_bytes_sent) {
		err = req->rq_received;
		goto out_unlock;
	}
682
	if (!xprt->ops->reserve_xprt(task)) {
L
Linus Torvalds 已提交
683 684 685 686 687 688 689 690 691
		err = -EAGAIN;
		goto out_unlock;
	}

	if (!xprt_connected(xprt)) {
		err = -ENOTCONN;
		goto out_unlock;
	}
out_unlock:
C
Chuck Lever 已提交
692
	spin_unlock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
693 694 695
	return err;
}

696 697 698 699 700 701 702
/**
 * 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 已提交
703 704 705
{
	struct rpc_rqst	*req = task->tk_rqstp;
	struct rpc_xprt	*xprt = req->rq_xprt;
706
	int status;
L
Linus Torvalds 已提交
707 708 709 710 711 712

	dprintk("RPC: %4d xprt_transmit(%u)\n", task->tk_pid, req->rq_slen);

	smp_rmb();
	if (!req->rq_received) {
		if (list_empty(&req->rq_list)) {
C
Chuck Lever 已提交
713
			spin_lock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
714 715 716 717 718
			/* 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 */
			list_add_tail(&req->rq_list, &xprt->recv);
C
Chuck Lever 已提交
719
			spin_unlock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
720
			xprt_reset_majortimeo(req);
721 722
			/* Turn off autodisconnect */
			del_singleshot_timer_sync(&xprt->timer);
L
Linus Torvalds 已提交
723 724 725 726
		}
	} else if (!req->rq_bytes_sent)
		return;

727
	status = xprt->ops->send_request(task);
728 729 730 731 732 733 734 735 736
	if (status == 0) {
		dprintk("RPC: %4d xmit complete\n", task->tk_pid);
		spin_lock_bh(&xprt->transport_lock);
		xprt->ops->set_retrans_timeout(task);
		/* Don't race with disconnect */
		if (!xprt_connected(xprt))
			task->tk_status = -ENOTCONN;
		else if (!req->rq_received)
			rpc_sleep_on(&xprt->pending, task, NULL, xprt_timer);
737
		xprt->ops->release_xprt(xprt, task);
738 739 740
		spin_unlock_bh(&xprt->transport_lock);
		return;
	}
L
Linus Torvalds 已提交
741 742 743 744 745 746 747 748 749 750 751

	/* Note: at this point, task->tk_sleeping has not yet been set,
	 *	 hence there is no danger of the waking up task being put on
	 *	 schedq, and being picked up by a parallel run of rpciod().
	 */
	task->tk_status = status;

	switch (status) {
	case -ECONNREFUSED:
		task->tk_timeout = RPC_REESTABLISH_TIMEOUT;
		rpc_sleep_on(&xprt->sending, task, NULL, NULL);
752
	case -EAGAIN:
L
Linus Torvalds 已提交
753 754 755
	case -ENOTCONN:
		return;
	default:
756
		break;
L
Linus Torvalds 已提交
757 758 759 760 761
	}
	xprt_release_write(xprt, task);
	return;
}

762
static inline void do_xprt_reserve(struct rpc_task *task)
L
Linus Torvalds 已提交
763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781
{
	struct rpc_xprt	*xprt = task->tk_xprt;

	task->tk_status = 0;
	if (task->tk_rqstp)
		return;
	if (!list_empty(&xprt->free)) {
		struct rpc_rqst	*req = list_entry(xprt->free.next, struct rpc_rqst, rq_list);
		list_del_init(&req->rq_list);
		task->tk_rqstp = req;
		xprt_request_init(task, xprt);
		return;
	}
	dprintk("RPC:      waiting for request slot\n");
	task->tk_status = -EAGAIN;
	task->tk_timeout = 0;
	rpc_sleep_on(&xprt->backlog, task, NULL, NULL);
}

782 783 784 785 786 787 788 789
/**
 * xprt_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.
 */
void xprt_reserve(struct rpc_task *task)
L
Linus Torvalds 已提交
790 791 792 793 794
{
	struct rpc_xprt	*xprt = task->tk_xprt;

	task->tk_status = -EIO;
	if (!xprt->shutdown) {
C
Chuck Lever 已提交
795
		spin_lock(&xprt->reserve_lock);
L
Linus Torvalds 已提交
796
		do_xprt_reserve(task);
C
Chuck Lever 已提交
797
		spin_unlock(&xprt->reserve_lock);
L
Linus Torvalds 已提交
798 799 800 801 802 803 804 805 806 807 808 809 810
	}
}

static inline u32 xprt_alloc_xid(struct rpc_xprt *xprt)
{
	return xprt->xid++;
}

static inline void xprt_init_xid(struct rpc_xprt *xprt)
{
	get_random_bytes(&xprt->xid, sizeof(xprt->xid));
}

811
static void xprt_request_init(struct rpc_task *task, struct rpc_xprt *xprt)
L
Linus Torvalds 已提交
812 813 814 815 816 817 818 819 820 821 822
{
	struct rpc_rqst	*req = task->tk_rqstp;

	req->rq_timeout = xprt->timeout.to_initval;
	req->rq_task	= task;
	req->rq_xprt    = xprt;
	req->rq_xid     = xprt_alloc_xid(xprt);
	dprintk("RPC: %4d reserved req %p xid %08x\n", task->tk_pid,
			req, ntohl(req->rq_xid));
}

823 824 825 826
/**
 * xprt_release - release an RPC request slot
 * @task: task which is finished with the slot
 *
L
Linus Torvalds 已提交
827
 */
828
void xprt_release(struct rpc_task *task)
L
Linus Torvalds 已提交
829 830 831 832 833 834
{
	struct rpc_xprt	*xprt = task->tk_xprt;
	struct rpc_rqst	*req;

	if (!(req = task->tk_rqstp))
		return;
C
Chuck Lever 已提交
835
	spin_lock_bh(&xprt->transport_lock);
836
	xprt->ops->release_xprt(xprt, task);
837 838
	if (xprt->ops->release_request)
		xprt->ops->release_request(task);
L
Linus Torvalds 已提交
839 840 841 842
	if (!list_empty(&req->rq_list))
		list_del(&req->rq_list);
	xprt->last_used = jiffies;
	if (list_empty(&xprt->recv) && !xprt->shutdown)
843 844
		mod_timer(&xprt->timer,
				xprt->last_used + RPC_IDLE_DISCONNECT_TIMEOUT);
C
Chuck Lever 已提交
845
	spin_unlock_bh(&xprt->transport_lock);
L
Linus Torvalds 已提交
846 847 848 849 850
	task->tk_rqstp = NULL;
	memset(req, 0, sizeof(*req));	/* mark unused */

	dprintk("RPC: %4d release request %p\n", task->tk_pid, req);

C
Chuck Lever 已提交
851
	spin_lock(&xprt->reserve_lock);
L
Linus Torvalds 已提交
852 853
	list_add(&req->rq_list, &xprt->free);
	xprt_clear_backlog(xprt);
C
Chuck Lever 已提交
854
	spin_unlock(&xprt->reserve_lock);
L
Linus Torvalds 已提交
855 856
}

857 858 859 860 861 862
/**
 * xprt_set_timeout - set constant RPC timeout
 * @to: RPC timeout parameters to set up
 * @retr: number of retries
 * @incr: amount of increase after each retry
 *
L
Linus Torvalds 已提交
863
 */
864
void xprt_set_timeout(struct rpc_timeout *to, unsigned int retr, unsigned long incr)
L
Linus Torvalds 已提交
865 866 867
{
	to->to_initval   = 
	to->to_increment = incr;
868
	to->to_maxval    = to->to_initval + (incr * retr);
L
Linus Torvalds 已提交
869 870 871 872
	to->to_retries   = retr;
	to->to_exponential = 0;
}

873
static struct rpc_xprt *xprt_setup(int proto, struct sockaddr_in *ap, struct rpc_timeout *to)
L
Linus Torvalds 已提交
874
{
875
	int result;
L
Linus Torvalds 已提交
876 877 878 879 880 881 882 883
	struct rpc_xprt	*xprt;
	struct rpc_rqst	*req;

	if ((xprt = kmalloc(sizeof(struct rpc_xprt), GFP_KERNEL)) == NULL)
		return ERR_PTR(-ENOMEM);
	memset(xprt, 0, sizeof(*xprt)); /* Nnnngh! */

	xprt->addr = *ap;
884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900

	switch (proto) {
	case IPPROTO_UDP:
		result = xs_setup_udp(xprt, to);
		break;
	case IPPROTO_TCP:
		result = xs_setup_tcp(xprt, to);
		break;
	default:
		printk(KERN_ERR "RPC: unrecognized transport protocol: %d\n",
				proto);
		result = -EIO;
		break;
	}
	if (result) {
		kfree(xprt);
		return ERR_PTR(result);
L
Linus Torvalds 已提交
901
	}
902

C
Chuck Lever 已提交
903
	spin_lock_init(&xprt->transport_lock);
C
Chuck Lever 已提交
904
	spin_lock_init(&xprt->reserve_lock);
L
Linus Torvalds 已提交
905 906 907 908
	init_waitqueue_head(&xprt->cong_wait);

	INIT_LIST_HEAD(&xprt->free);
	INIT_LIST_HEAD(&xprt->recv);
909
	INIT_WORK(&xprt->task_cleanup, xprt_autoclose, xprt);
L
Linus Torvalds 已提交
910 911 912 913 914 915 916 917 918 919 920
	init_timer(&xprt->timer);
	xprt->timer.function = xprt_init_autodisconnect;
	xprt->timer.data = (unsigned long) xprt;
	xprt->last_used = jiffies;

	rpc_init_wait_queue(&xprt->pending, "xprt_pending");
	rpc_init_wait_queue(&xprt->sending, "xprt_sending");
	rpc_init_wait_queue(&xprt->resend, "xprt_resend");
	rpc_init_priority_wait_queue(&xprt->backlog, "xprt_backlog");

	/* initialize free list */
921
	for (req = &xprt->slot[xprt->max_reqs-1]; req >= &xprt->slot[0]; req--)
L
Linus Torvalds 已提交
922 923 924 925 926 927 928 929 930 931
		list_add(&req->rq_list, &xprt->free);

	xprt_init_xid(xprt);

	dprintk("RPC:      created transport %p with %u slots\n", xprt,
			xprt->max_reqs);
	
	return xprt;
}

932 933 934 935 936 937
/**
 * xprt_create_proto - create an RPC client transport
 * @proto: requested transport protocol
 * @sap: remote peer's address
 * @to: timeout parameters for new transport
 *
L
Linus Torvalds 已提交
938
 */
939
struct rpc_xprt *xprt_create_proto(int proto, struct sockaddr_in *sap, struct rpc_timeout *to)
L
Linus Torvalds 已提交
940 941 942 943 944 945 946 947 948 949 950
{
	struct rpc_xprt	*xprt;

	xprt = xprt_setup(proto, sap, to);
	if (IS_ERR(xprt))
		dprintk("RPC:      xprt_create_proto failed\n");
	else
		dprintk("RPC:      xprt_create_proto created xprt %p\n", xprt);
	return xprt;
}

951
static void xprt_shutdown(struct rpc_xprt *xprt)
L
Linus Torvalds 已提交
952 953 954 955
{
	xprt->shutdown = 1;
	rpc_wake_up(&xprt->sending);
	rpc_wake_up(&xprt->resend);
956
	xprt_wake_pending_tasks(xprt, -EIO);
L
Linus Torvalds 已提交
957 958 959 960 961
	rpc_wake_up(&xprt->backlog);
	wake_up(&xprt->cong_wait);
	del_timer_sync(&xprt->timer);
}

962
static int xprt_clear_backlog(struct rpc_xprt *xprt) {
L
Linus Torvalds 已提交
963 964 965 966 967
	rpc_wake_up_next(&xprt->backlog);
	wake_up(&xprt->cong_wait);
	return 1;
}

968 969 970 971
/**
 * xprt_destroy - destroy an RPC transport, killing off all requests.
 * @xprt: transport to destroy
 *
L
Linus Torvalds 已提交
972
 */
973
int xprt_destroy(struct rpc_xprt *xprt)
L
Linus Torvalds 已提交
974 975 976
{
	dprintk("RPC:      destroying transport %p\n", xprt);
	xprt_shutdown(xprt);
977
	xprt->ops->destroy(xprt);
L
Linus Torvalds 已提交
978 979 980 981
	kfree(xprt);

	return 0;
}