clnt.c 36.8 KB
Newer Older
L
Linus Torvalds 已提交
1
/*
2
 *  linux/net/sunrpc/clnt.c
L
Linus Torvalds 已提交
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
 *
 *  This file contains the high-level RPC interface.
 *  It is modeled as a finite state machine to support both synchronous
 *  and asynchronous requests.
 *
 *  -	RPC header generation and argument serialization.
 *  -	Credential refresh.
 *  -	TCP connect handling.
 *  -	Retry of operation when it is suspected the operation failed because
 *	of uid squashing on the server, or when the credentials were stale
 *	and need to be refreshed, or when a packet was damaged in transit.
 *	This may be have to be moved to the VFS layer.
 *
 *  NB: BSD uses a more intelligent approach to guessing when a request
 *  or reply has been lost by keeping the RTO estimate for each procedure.
 *  We currently make do with a constant timeout value.
 *
 *  Copyright (C) 1992,1993 Rick Sladkey <jrs@world.std.com>
 *  Copyright (C) 1995,1996 Olaf Kirch <okir@monad.swb.de>
 */

#include <asm/system.h>

#include <linux/module.h>
#include <linux/types.h>
#include <linux/mm.h>
#include <linux/slab.h>
30
#include <linux/smp_lock.h>
L
Linus Torvalds 已提交
31
#include <linux/utsname.h>
32
#include <linux/workqueue.h>
L
Linus Torvalds 已提交
33 34 35

#include <linux/sunrpc/clnt.h>
#include <linux/sunrpc/rpc_pipe_fs.h>
36
#include <linux/sunrpc/metrics.h>
L
Linus Torvalds 已提交
37 38 39 40 41 42


#ifdef RPC_DEBUG
# define RPCDBG_FACILITY	RPCDBG_CALL
#endif

43 44 45 46
#define dprint_status(t)					\
	dprintk("RPC: %5u %s (status %d)\n", t->tk_pid,		\
			__FUNCTION__, t->tk_status)

47 48 49 50 51 52
/*
 * All RPC clients are linked into this list
 */
static LIST_HEAD(all_clients);
static DEFINE_SPINLOCK(rpc_client_lock);

L
Linus Torvalds 已提交
53 54 55 56 57 58 59 60 61 62
static DECLARE_WAIT_QUEUE_HEAD(destroy_wait);


static void	call_start(struct rpc_task *task);
static void	call_reserve(struct rpc_task *task);
static void	call_reserveresult(struct rpc_task *task);
static void	call_allocate(struct rpc_task *task);
static void	call_encode(struct rpc_task *task);
static void	call_decode(struct rpc_task *task);
static void	call_bind(struct rpc_task *task);
63
static void	call_bind_status(struct rpc_task *task);
L
Linus Torvalds 已提交
64 65
static void	call_transmit(struct rpc_task *task);
static void	call_status(struct rpc_task *task);
66
static void	call_transmit_status(struct rpc_task *task);
L
Linus Torvalds 已提交
67 68 69 70 71
static void	call_refresh(struct rpc_task *task);
static void	call_refreshresult(struct rpc_task *task);
static void	call_timeout(struct rpc_task *task);
static void	call_connect(struct rpc_task *task);
static void	call_connect_status(struct rpc_task *task);
72 73
static __be32 *	call_header(struct rpc_task *task);
static __be32 *	call_verify(struct rpc_task *task);
L
Linus Torvalds 已提交
74

T
Trond Myklebust 已提交
75 76
static int	rpc_ping(struct rpc_clnt *clnt, int flags);

77 78 79 80 81 82 83 84 85 86 87 88 89
static void rpc_register_client(struct rpc_clnt *clnt)
{
	spin_lock(&rpc_client_lock);
	list_add(&clnt->cl_clients, &all_clients);
	spin_unlock(&rpc_client_lock);
}

static void rpc_unregister_client(struct rpc_clnt *clnt)
{
	spin_lock(&rpc_client_lock);
	list_del(&clnt->cl_clients);
	spin_unlock(&rpc_client_lock);
}
L
Linus Torvalds 已提交
90 91 92 93

static int
rpc_setup_pipedir(struct rpc_clnt *clnt, char *dir_name)
{
94
	static uint32_t clntid;
L
Linus Torvalds 已提交
95 96
	int error;

97 98
	clnt->cl_vfsmnt = ERR_PTR(-ENOENT);
	clnt->cl_dentry = ERR_PTR(-ENOENT);
L
Linus Torvalds 已提交
99 100
	if (dir_name == NULL)
		return 0;
101 102 103 104 105

	clnt->cl_vfsmnt = rpc_get_mount();
	if (IS_ERR(clnt->cl_vfsmnt))
		return PTR_ERR(clnt->cl_vfsmnt);

106 107 108 109 110 111 112 113
	for (;;) {
		snprintf(clnt->cl_pathname, sizeof(clnt->cl_pathname),
				"%s/clnt%x", dir_name,
				(unsigned int)clntid++);
		clnt->cl_pathname[sizeof(clnt->cl_pathname) - 1] = '\0';
		clnt->cl_dentry = rpc_mkdir(clnt->cl_pathname, clnt);
		if (!IS_ERR(clnt->cl_dentry))
			return 0;
L
Linus Torvalds 已提交
114
		error = PTR_ERR(clnt->cl_dentry);
115 116 117
		if (error != -EEXIST) {
			printk(KERN_INFO "RPC: Couldn't create pipefs entry %s, error %d\n",
					clnt->cl_pathname, error);
118
			rpc_put_mount();
119 120
			return error;
		}
L
Linus Torvalds 已提交
121 122 123
	}
}

124
static struct rpc_clnt * rpc_new_client(struct rpc_xprt *xprt, char *servname, struct rpc_program *program, u32 vers, rpc_authflavor_t flavor)
L
Linus Torvalds 已提交
125 126 127
{
	struct rpc_version	*version;
	struct rpc_clnt		*clnt = NULL;
128
	struct rpc_auth		*auth;
L
Linus Torvalds 已提交
129 130 131
	int err;
	int len;

132 133
	dprintk("RPC:       creating %s client for %s (xprt %p)\n",
			program->name, servname, xprt);
L
Linus Torvalds 已提交
134

135 136 137
	err = rpciod_up();
	if (err)
		goto out_no_rpciod;
L
Linus Torvalds 已提交
138 139
	err = -EINVAL;
	if (!xprt)
140
		goto out_no_xprt;
L
Linus Torvalds 已提交
141 142 143 144
	if (vers >= program->nrvers || !(version = program->version[vers]))
		goto out_err;

	err = -ENOMEM;
145
	clnt = kzalloc(sizeof(*clnt), GFP_KERNEL);
L
Linus Torvalds 已提交
146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167
	if (!clnt)
		goto out_err;
	clnt->cl_parent = clnt;

	clnt->cl_server = clnt->cl_inline_name;
	len = strlen(servname) + 1;
	if (len > sizeof(clnt->cl_inline_name)) {
		char *buf = kmalloc(len, GFP_KERNEL);
		if (buf != 0)
			clnt->cl_server = buf;
		else
			len = sizeof(clnt->cl_inline_name);
	}
	strlcpy(clnt->cl_server, servname, len);

	clnt->cl_xprt     = xprt;
	clnt->cl_procinfo = version->procs;
	clnt->cl_maxproc  = version->nrprocs;
	clnt->cl_protname = program->name;
	clnt->cl_prog     = program->number;
	clnt->cl_vers     = version->number;
	clnt->cl_stats    = program->stats;
168
	clnt->cl_metrics  = rpc_alloc_iostats(clnt);
169 170 171
	err = -ENOMEM;
	if (clnt->cl_metrics == NULL)
		goto out_no_stats;
172
	clnt->cl_program  = program;
173
	INIT_LIST_HEAD(&clnt->cl_tasks);
174
	spin_lock_init(&clnt->cl_lock);
L
Linus Torvalds 已提交
175

176
	if (!xprt_bound(clnt->cl_xprt))
L
Linus Torvalds 已提交
177 178 179 180 181
		clnt->cl_autobind = 1;

	clnt->cl_rtt = &clnt->cl_rtt_default;
	rpc_init_rtt(&clnt->cl_rtt_default, xprt->timeout.to_initval);

182 183
	kref_init(&clnt->cl_kref);

L
Linus Torvalds 已提交
184 185 186 187
	err = rpc_setup_pipedir(clnt, program->pipe_dir_name);
	if (err < 0)
		goto out_no_path;

188 189
	auth = rpcauth_create(flavor, clnt);
	if (IS_ERR(auth)) {
L
Linus Torvalds 已提交
190 191
		printk(KERN_INFO "RPC: Couldn't create auth handle (flavor %u)\n",
				flavor);
192
		err = PTR_ERR(auth);
L
Linus Torvalds 已提交
193 194 195 196
		goto out_no_auth;
	}

	/* save the nodename */
197
	clnt->cl_nodelen = strlen(utsname()->nodename);
L
Linus Torvalds 已提交
198 199
	if (clnt->cl_nodelen > UNX_MAXNODENAME)
		clnt->cl_nodelen = UNX_MAXNODENAME;
200
	memcpy(clnt->cl_nodename, utsname()->nodename, clnt->cl_nodelen);
201
	rpc_register_client(clnt);
L
Linus Torvalds 已提交
202 203 204
	return clnt;

out_no_auth:
205
	if (!IS_ERR(clnt->cl_dentry)) {
T
Trond Myklebust 已提交
206
		rpc_rmdir(clnt->cl_dentry);
207 208
		rpc_put_mount();
	}
L
Linus Torvalds 已提交
209
out_no_path:
210 211
	rpc_free_iostats(clnt->cl_metrics);
out_no_stats:
L
Linus Torvalds 已提交
212 213 214 215
	if (clnt->cl_server != clnt->cl_inline_name)
		kfree(clnt->cl_server);
	kfree(clnt);
out_err:
216
	xprt_put(xprt);
217
out_no_xprt:
218 219
	rpciod_down();
out_no_rpciod:
L
Linus Torvalds 已提交
220 221 222
	return ERR_PTR(err);
}

223 224 225 226 227 228 229 230 231 232 233 234 235 236
/*
 * rpc_create - create an RPC client and transport with one call
 * @args: rpc_clnt create argument structure
 *
 * Creates and initializes an RPC transport and an RPC client.
 *
 * It can ping the server in order to determine if it is up, and to see if
 * it supports this program and version.  RPC_CLNT_CREATE_NOPING disables
 * this behavior so asynchronous tasks can also use rpc_create.
 */
struct rpc_clnt *rpc_create(struct rpc_create_args *args)
{
	struct rpc_xprt *xprt;
	struct rpc_clnt *clnt;
237 238
	struct rpc_xprtsock_create xprtargs = {
		.proto = args->protocol,
239
		.srcaddr = args->saddress,
240 241 242 243
		.dstaddr = args->address,
		.addrlen = args->addrsize,
		.timeout = args->timeout
	};
244
	char servername[20];
245

246
	xprt = xprt_create_transport(&xprtargs);
247 248 249
	if (IS_ERR(xprt))
		return (struct rpc_clnt *)xprt;

250 251 252 253 254 255 256 257 258 259 260 261
	/*
	 * If the caller chooses not to specify a hostname, whip
	 * up a string representation of the passed-in address.
	 */
	if (args->servername == NULL) {
		struct sockaddr_in *addr =
					(struct sockaddr_in *) &args->address;
		snprintf(servername, sizeof(servername), NIPQUAD_FMT,
			NIPQUAD(addr->sin_addr.s_addr));
		args->servername = servername;
	}

262 263 264 265 266 267 268 269 270 271 272
	/*
	 * By default, kernel RPC client connects from a reserved port.
	 * CAP_NET_BIND_SERVICE will not be set for unprivileged requesters,
	 * but it is always enabled for rpciod, which handles the connect
	 * operation.
	 */
	xprt->resvport = 1;
	if (args->flags & RPC_CLNT_CREATE_NONPRIVPORT)
		xprt->resvport = 0;

	dprintk("RPC:       creating %s client for %s (xprt %p)\n",
273
			args->program->name, args->servername, xprt);
274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295

	clnt = rpc_new_client(xprt, args->servername, args->program,
				args->version, args->authflavor);
	if (IS_ERR(clnt))
		return clnt;

	if (!(args->flags & RPC_CLNT_CREATE_NOPING)) {
		int err = rpc_ping(clnt, RPC_TASK_SOFT|RPC_TASK_NOINTR);
		if (err != 0) {
			rpc_shutdown_client(clnt);
			return ERR_PTR(err);
		}
	}

	clnt->cl_softrtry = 1;
	if (args->flags & RPC_CLNT_CREATE_HARDRTRY)
		clnt->cl_softrtry = 0;

	if (args->flags & RPC_CLNT_CREATE_INTR)
		clnt->cl_intr = 1;
	if (args->flags & RPC_CLNT_CREATE_AUTOBIND)
		clnt->cl_autobind = 1;
296 297
	if (args->flags & RPC_CLNT_CREATE_DISCRTRY)
		clnt->cl_discrtry = 1;
298 299 300

	return clnt;
}
301
EXPORT_SYMBOL_GPL(rpc_create);
302

L
Linus Torvalds 已提交
303 304 305 306 307 308 309 310 311
/*
 * This function clones the RPC client structure. It allows us to share the
 * same transport while varying parameters such as the authentication
 * flavour.
 */
struct rpc_clnt *
rpc_clone_client(struct rpc_clnt *clnt)
{
	struct rpc_clnt *new;
312
	int err = -ENOMEM;
L
Linus Torvalds 已提交
313

314
	new = kmemdup(clnt, sizeof(*new), GFP_KERNEL);
L
Linus Torvalds 已提交
315 316
	if (!new)
		goto out_no_clnt;
317 318 319 320 321 322
	new->cl_parent = clnt;
	/* Turn off autobind on clones */
	new->cl_autobind = 0;
	INIT_LIST_HEAD(&new->cl_tasks);
	spin_lock_init(&new->cl_lock);
	rpc_init_rtt(&new->cl_rtt_default, clnt->cl_xprt->timeout.to_initval);
323 324 325
	new->cl_metrics = rpc_alloc_iostats(clnt);
	if (new->cl_metrics == NULL)
		goto out_no_stats;
326
	kref_init(&new->cl_kref);
327 328 329
	err = rpc_setup_pipedir(new, clnt->cl_program->pipe_dir_name);
	if (err != 0)
		goto out_no_path;
L
Linus Torvalds 已提交
330 331
	if (new->cl_auth)
		atomic_inc(&new->cl_auth->au_count);
332 333
	xprt_get(clnt->cl_xprt);
	kref_get(&clnt->cl_kref);
334
	rpc_register_client(new);
335
	rpciod_up();
L
Linus Torvalds 已提交
336
	return new;
337 338
out_no_path:
	rpc_free_iostats(new->cl_metrics);
339 340
out_no_stats:
	kfree(new);
L
Linus Torvalds 已提交
341
out_no_clnt:
342
	dprintk("RPC:       %s: returned error %d\n", __FUNCTION__, err);
343
	return ERR_PTR(err);
L
Linus Torvalds 已提交
344 345 346 347
}

/*
 * Properly shut down an RPC client, terminating all outstanding
348
 * requests.
L
Linus Torvalds 已提交
349
 */
350
void rpc_shutdown_client(struct rpc_clnt *clnt)
L
Linus Torvalds 已提交
351
{
352 353
	dprintk("RPC:       shutting down %s client for %s\n",
			clnt->cl_protname, clnt->cl_server);
L
Linus Torvalds 已提交
354

355
	while (!list_empty(&clnt->cl_tasks)) {
L
Linus Torvalds 已提交
356
		rpc_killall_tasks(clnt);
I
Ingo Molnar 已提交
357
		wait_event_timeout(destroy_wait,
358
			list_empty(&clnt->cl_tasks), 1*HZ);
L
Linus Torvalds 已提交
359 360
	}

361
	rpc_release_client(clnt);
L
Linus Torvalds 已提交
362 363 364
}

/*
365
 * Free an RPC client
L
Linus Torvalds 已提交
366
 */
367 368
static void
rpc_free_client(struct kref *kref)
L
Linus Torvalds 已提交
369
{
370
	struct rpc_clnt *clnt = container_of(kref, struct rpc_clnt, cl_kref);
L
Linus Torvalds 已提交
371

372
	dprintk("RPC:       destroying %s client for %s\n",
L
Linus Torvalds 已提交
373
			clnt->cl_protname, clnt->cl_server);
374
	if (!IS_ERR(clnt->cl_dentry)) {
T
Trond Myklebust 已提交
375
		rpc_rmdir(clnt->cl_dentry);
376 377
		rpc_put_mount();
	}
378
	if (clnt->cl_parent != clnt) {
379
		rpc_release_client(clnt->cl_parent);
380 381
		goto out_free;
	}
L
Linus Torvalds 已提交
382 383 384
	if (clnt->cl_server != clnt->cl_inline_name)
		kfree(clnt->cl_server);
out_free:
385
	rpc_unregister_client(clnt);
386 387
	rpc_free_iostats(clnt->cl_metrics);
	clnt->cl_metrics = NULL;
388
	xprt_put(clnt->cl_xprt);
389
	rpciod_down();
L
Linus Torvalds 已提交
390 391 392
	kfree(clnt);
}

393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416
/*
 * Free an RPC client
 */
static void
rpc_free_auth(struct kref *kref)
{
	struct rpc_clnt *clnt = container_of(kref, struct rpc_clnt, cl_kref);

	if (clnt->cl_auth == NULL) {
		rpc_free_client(kref);
		return;
	}

	/*
	 * Note: RPCSEC_GSS may need to send NULL RPC calls in order to
	 *       release remaining GSS contexts. This mechanism ensures
	 *       that it can do so safely.
	 */
	kref_init(kref);
	rpcauth_release(clnt->cl_auth);
	clnt->cl_auth = NULL;
	kref_put(kref, rpc_free_client);
}

L
Linus Torvalds 已提交
417
/*
418
 * Release reference to the RPC client
L
Linus Torvalds 已提交
419 420 421 422
 */
void
rpc_release_client(struct rpc_clnt *clnt)
{
423
	dprintk("RPC:       rpc_release_client(%p)\n", clnt);
L
Linus Torvalds 已提交
424

425 426
	if (list_empty(&clnt->cl_tasks))
		wake_up(&destroy_wait);
427
	kref_put(&clnt->cl_kref, rpc_free_auth);
428 429
}

430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463
/**
 * rpc_bind_new_program - bind a new RPC program to an existing client
 * @old - old rpc_client
 * @program - rpc program to set
 * @vers - rpc program version
 *
 * Clones the rpc client and sets up a new RPC program. This is mainly
 * of use for enabling different RPC programs to share the same transport.
 * The Sun NFSv2/v3 ACL protocol can do this.
 */
struct rpc_clnt *rpc_bind_new_program(struct rpc_clnt *old,
				      struct rpc_program *program,
				      int vers)
{
	struct rpc_clnt *clnt;
	struct rpc_version *version;
	int err;

	BUG_ON(vers >= program->nrvers || !program->version[vers]);
	version = program->version[vers];
	clnt = rpc_clone_client(old);
	if (IS_ERR(clnt))
		goto out;
	clnt->cl_procinfo = version->procs;
	clnt->cl_maxproc  = version->nrprocs;
	clnt->cl_protname = program->name;
	clnt->cl_prog     = program->number;
	clnt->cl_vers     = version->number;
	clnt->cl_stats    = program->stats;
	err = rpc_ping(clnt, RPC_TASK_SOFT|RPC_TASK_NOINTR);
	if (err != 0) {
		rpc_shutdown_client(clnt);
		clnt = ERR_PTR(err);
	}
464
out:
465 466 467
	return clnt;
}

L
Linus Torvalds 已提交
468 469 470 471
/*
 * Default callback for async RPC calls
 */
static void
472
rpc_default_callback(struct rpc_task *task, void *data)
L
Linus Torvalds 已提交
473 474 475
{
}

476 477 478 479
static const struct rpc_call_ops rpc_default_ops = {
	.rpc_call_done = rpc_default_callback,
};

L
Linus Torvalds 已提交
480
/*
481
 *	Export the signal mask handling for synchronous code that
L
Linus Torvalds 已提交
482 483
 *	sleeps on RPC calls
 */
484
#define RPC_INTR_SIGNALS (sigmask(SIGHUP) | sigmask(SIGINT) | sigmask(SIGQUIT) | sigmask(SIGTERM))
485

486 487
static void rpc_save_sigmask(sigset_t *oldset, int intr)
{
488
	unsigned long	sigallow = sigmask(SIGKILL);
489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507
	sigset_t sigmask;

	/* Block all signals except those listed in sigallow */
	if (intr)
		sigallow |= RPC_INTR_SIGNALS;
	siginitsetinv(&sigmask, sigallow);
	sigprocmask(SIG_BLOCK, &sigmask, oldset);
}

static inline void rpc_task_sigmask(struct rpc_task *task, sigset_t *oldset)
{
	rpc_save_sigmask(oldset, !RPC_TASK_UNINTERRUPTIBLE(task));
}

static inline void rpc_restore_sigmask(sigset_t *oldset)
{
	sigprocmask(SIG_SETMASK, oldset, NULL);
}

L
Linus Torvalds 已提交
508 509
void rpc_clnt_sigmask(struct rpc_clnt *clnt, sigset_t *oldset)
{
510
	rpc_save_sigmask(oldset, clnt->cl_intr);
L
Linus Torvalds 已提交
511 512 513 514
}

void rpc_clnt_sigunmask(struct rpc_clnt *clnt, sigset_t *oldset)
{
515
	rpc_restore_sigmask(oldset);
L
Linus Torvalds 已提交
516 517
}

518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556
static
struct rpc_task *rpc_do_run_task(struct rpc_clnt *clnt,
		struct rpc_message *msg,
		int flags,
		const struct rpc_call_ops *ops,
		void *data)
{
	struct rpc_task *task, *ret;
	sigset_t oldset;

	task = rpc_new_task(clnt, flags, ops, data);
	if (task == NULL) {
		rpc_release_calldata(ops, data);
		return ERR_PTR(-ENOMEM);
	}

	/* Mask signals on synchronous RPC calls and RPCSEC_GSS upcalls */
	rpc_task_sigmask(task, &oldset);
	if (msg != NULL) {
		rpc_call_setup(task, msg, 0);
		if (task->tk_status != 0) {
			ret = ERR_PTR(task->tk_status);
			rpc_put_task(task);
			goto out;
		}
	}
	atomic_inc(&task->tk_count);
	rpc_execute(task);
	ret = task;
out:
	rpc_restore_sigmask(&oldset);
	return ret;
}

/**
 * rpc_call_sync - Perform a synchronous RPC call
 * @clnt: pointer to RPC client
 * @msg: RPC call parameters
 * @flags: RPC call flags
L
Linus Torvalds 已提交
557 558 559 560
 */
int rpc_call_sync(struct rpc_clnt *clnt, struct rpc_message *msg, int flags)
{
	struct rpc_task	*task;
561
	int status;
L
Linus Torvalds 已提交
562 563 564

	BUG_ON(flags & RPC_TASK_ASYNC);

565 566 567
	task = rpc_do_run_task(clnt, msg, flags, &rpc_default_ops, NULL);
	if (IS_ERR(task))
		return PTR_ERR(task);
568
	status = task->tk_status;
569
	rpc_put_task(task);
L
Linus Torvalds 已提交
570 571 572
	return status;
}

573 574 575 576 577 578 579
/**
 * rpc_call_async - Perform an asynchronous RPC call
 * @clnt: pointer to RPC client
 * @msg: RPC call parameters
 * @flags: RPC call flags
 * @ops: RPC call ops
 * @data: user call data
L
Linus Torvalds 已提交
580 581 582
 */
int
rpc_call_async(struct rpc_clnt *clnt, struct rpc_message *msg, int flags,
583
	       const struct rpc_call_ops *tk_ops, void *data)
L
Linus Torvalds 已提交
584 585 586
{
	struct rpc_task	*task;

587 588 589 590 591
	task = rpc_do_run_task(clnt, msg, flags|RPC_TASK_ASYNC, tk_ops, data);
	if (IS_ERR(task))
		return PTR_ERR(task);
	rpc_put_task(task);
	return 0;
L
Linus Torvalds 已提交
592 593
}

594 595 596 597 598 599 600 601 602 603 604 605 606 607
/**
 * rpc_run_task - Allocate a new RPC task, then run rpc_execute against it
 * @clnt: pointer to RPC client
 * @flags: RPC flags
 * @ops: RPC call ops
 * @data: user call data
 */
struct rpc_task *rpc_run_task(struct rpc_clnt *clnt, int flags,
					const struct rpc_call_ops *tk_ops,
					void *data)
{
	return rpc_do_run_task(clnt, NULL, flags, tk_ops, data);
}
EXPORT_SYMBOL(rpc_run_task);
L
Linus Torvalds 已提交
608 609 610 611 612 613 614 615 616 617 618 619 620 621 622

void
rpc_call_setup(struct rpc_task *task, struct rpc_message *msg, int flags)
{
	task->tk_msg   = *msg;
	task->tk_flags |= flags;
	/* Bind the user cred */
	if (task->tk_msg.rpc_cred != NULL)
		rpcauth_holdcred(task);
	else
		rpcauth_bindcred(task);

	if (task->tk_status == 0)
		task->tk_action = call_start;
	else
T
Trond Myklebust 已提交
623
		task->tk_action = rpc_exit_task;
L
Linus Torvalds 已提交
624 625
}

626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642
/**
 * rpc_peeraddr - extract remote peer address from clnt's xprt
 * @clnt: RPC client structure
 * @buf: target buffer
 * @size: length of target buffer
 *
 * Returns the number of bytes that are actually in the stored address.
 */
size_t rpc_peeraddr(struct rpc_clnt *clnt, struct sockaddr *buf, size_t bufsize)
{
	size_t bytes;
	struct rpc_xprt *xprt = clnt->cl_xprt;

	bytes = sizeof(xprt->addr);
	if (bytes > bufsize)
		bytes = bufsize;
	memcpy(buf, &clnt->cl_xprt->addr, bytes);
643
	return xprt->addrlen;
644
}
645
EXPORT_SYMBOL_GPL(rpc_peeraddr);
646

647 648 649 650 651 652 653 654 655
/**
 * rpc_peeraddr2str - return remote peer address in printable format
 * @clnt: RPC client structure
 * @format: address format
 *
 */
char *rpc_peeraddr2str(struct rpc_clnt *clnt, enum rpc_display_format_t format)
{
	struct rpc_xprt *xprt = clnt->cl_xprt;
656 657 658 659 660

	if (xprt->address_strings[format] != NULL)
		return xprt->address_strings[format];
	else
		return "unprintable";
661
}
662
EXPORT_SYMBOL_GPL(rpc_peeraddr2str);
663

L
Linus Torvalds 已提交
664 665 666 667
void
rpc_setbufsize(struct rpc_clnt *clnt, unsigned int sndsize, unsigned int rcvsize)
{
	struct rpc_xprt *xprt = clnt->cl_xprt;
668 669
	if (xprt->ops->set_buffer_size)
		xprt->ops->set_buffer_size(xprt, sndsize, rcvsize);
L
Linus Torvalds 已提交
670 671 672 673 674 675 676 677 678 679 680 681 682 683
}

/*
 * Return size of largest payload RPC client can support, in bytes
 *
 * For stream transports, this is one RPC record fragment (see RFC
 * 1831), as we don't support multi-record requests yet.  For datagram
 * transports, this is the size of an IP packet minus the IP, UDP, and
 * RPC header sizes.
 */
size_t rpc_max_payload(struct rpc_clnt *clnt)
{
	return clnt->cl_xprt->max_payload;
}
684
EXPORT_SYMBOL_GPL(rpc_max_payload);
L
Linus Torvalds 已提交
685

686 687 688 689 690 691 692 693
/**
 * rpc_force_rebind - force transport to check that remote port is unchanged
 * @clnt: client to rebind
 *
 */
void rpc_force_rebind(struct rpc_clnt *clnt)
{
	if (clnt->cl_autobind)
694
		xprt_clear_bound(clnt->cl_xprt);
695
}
696
EXPORT_SYMBOL_GPL(rpc_force_rebind);
697

L
Linus Torvalds 已提交
698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721
/*
 * Restart an (async) RPC call. Usually called from within the
 * exit handler.
 */
void
rpc_restart_call(struct rpc_task *task)
{
	if (RPC_ASSASSINATED(task))
		return;

	task->tk_action = call_start;
}

/*
 * 0.  Initial state
 *
 *     Other FSM states can be visited zero or more times, but
 *     this state is visited exactly once for each RPC.
 */
static void
call_start(struct rpc_task *task)
{
	struct rpc_clnt	*clnt = task->tk_client;

722 723 724 725
	dprintk("RPC: %5u call_start %s%d proc %d (%s)\n", task->tk_pid,
			clnt->cl_protname, clnt->cl_vers,
			task->tk_msg.rpc_proc->p_proc,
			(RPC_IS_ASYNC(task) ? "async" : "sync"));
L
Linus Torvalds 已提交
726 727 728 729 730 731 732 733 734 735 736 737 738

	/* Increment call count */
	task->tk_msg.rpc_proc->p_count++;
	clnt->cl_stats->rpccnt++;
	task->tk_action = call_reserve;
}

/*
 * 1.	Reserve an RPC call slot
 */
static void
call_reserve(struct rpc_task *task)
{
739
	dprint_status(task);
L
Linus Torvalds 已提交
740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758

	if (!rpcauth_uptodatecred(task)) {
		task->tk_action = call_refresh;
		return;
	}

	task->tk_status  = 0;
	task->tk_action  = call_reserveresult;
	xprt_reserve(task);
}

/*
 * 1b.	Grok the result of xprt_reserve()
 */
static void
call_reserveresult(struct rpc_task *task)
{
	int status = task->tk_status;

759
	dprint_status(task);
L
Linus Torvalds 已提交
760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803

	/*
	 * After a call to xprt_reserve(), we must have either
	 * a request slot or else an error status.
	 */
	task->tk_status = 0;
	if (status >= 0) {
		if (task->tk_rqstp) {
			task->tk_action = call_allocate;
			return;
		}

		printk(KERN_ERR "%s: status=%d, but no request slot, exiting\n",
				__FUNCTION__, status);
		rpc_exit(task, -EIO);
		return;
	}

	/*
	 * Even though there was an error, we may have acquired
	 * a request slot somehow.  Make sure not to leak it.
	 */
	if (task->tk_rqstp) {
		printk(KERN_ERR "%s: status=%d, request allocated anyway\n",
				__FUNCTION__, status);
		xprt_release(task);
	}

	switch (status) {
	case -EAGAIN:	/* woken up; retry */
		task->tk_action = call_reserve;
		return;
	case -EIO:	/* probably a shutdown */
		break;
	default:
		printk(KERN_ERR "%s: unrecognized error %d, exiting\n",
				__FUNCTION__, status);
		break;
	}
	rpc_exit(task, status);
}

/*
 * 2.	Allocate the buffer. For details, see sched.c:rpc_malloc.
804
 *	(Note: buffer memory is freed in xprt_release).
L
Linus Torvalds 已提交
805 806 807 808
 */
static void
call_allocate(struct rpc_task *task)
{
809
	unsigned int slack = task->tk_msg.rpc_cred->cr_auth->au_cslack;
810 811
	struct rpc_rqst *req = task->tk_rqstp;
	struct rpc_xprt *xprt = task->tk_xprt;
812
	struct rpc_procinfo *proc = task->tk_msg.rpc_proc;
L
Linus Torvalds 已提交
813

814 815
	dprint_status(task);

816
	task->tk_status = 0;
L
Linus Torvalds 已提交
817
	task->tk_action = call_bind;
818

819
	if (req->rq_buffer)
L
Linus Torvalds 已提交
820 821
		return;

822 823 824 825 826
	if (proc->p_proc != 0) {
		BUG_ON(proc->p_arglen == 0);
		if (proc->p_decode != NULL)
			BUG_ON(proc->p_replen == 0);
	}
L
Linus Torvalds 已提交
827

828 829 830 831 832 833 834 835 836 837
	/*
	 * Calculate the size (in quads) of the RPC call
	 * and reply headers, and convert both values
	 * to byte sizes.
	 */
	req->rq_callsize = RPC_CALLHDRSIZE + (slack << 1) + proc->p_arglen;
	req->rq_callsize <<= 2;
	req->rq_rcvsize = RPC_REPHDRSIZE + slack + proc->p_replen;
	req->rq_rcvsize <<= 2;

838 839
	req->rq_buffer = xprt->ops->buf_alloc(task,
					req->rq_callsize + req->rq_rcvsize);
840
	if (req->rq_buffer != NULL)
L
Linus Torvalds 已提交
841
		return;
842 843

	dprintk("RPC: %5u rpc_buffer allocation failed\n", task->tk_pid);
L
Linus Torvalds 已提交
844

845
	if (RPC_IS_ASYNC(task) || !signalled()) {
L
Linus Torvalds 已提交
846 847 848 849 850 851 852 853 854
		xprt_release(task);
		task->tk_action = call_reserve;
		rpc_delay(task, HZ>>4);
		return;
	}

	rpc_exit(task, -ERESTARTSYS);
}

855 856 857 858 859 860 861 862 863 864 865 866
static inline int
rpc_task_need_encode(struct rpc_task *task)
{
	return task->tk_rqstp->rq_snd_buf.len == 0;
}

static inline void
rpc_task_force_reencode(struct rpc_task *task)
{
	task->tk_rqstp->rq_snd_buf.len = 0;
}

867 868 869 870 871 872 873 874 875 876 877
static inline void
rpc_xdr_buf_init(struct xdr_buf *buf, void *start, size_t len)
{
	buf->head[0].iov_base = start;
	buf->head[0].iov_len = len;
	buf->tail[0].iov_len = 0;
	buf->page_len = 0;
	buf->len = 0;
	buf->buflen = len;
}

L
Linus Torvalds 已提交
878 879 880 881 882 883 884 885
/*
 * 3.	Encode arguments of an RPC call
 */
static void
call_encode(struct rpc_task *task)
{
	struct rpc_rqst	*req = task->tk_rqstp;
	kxdrproc_t	encode;
886
	__be32		*p;
L
Linus Torvalds 已提交
887

888
	dprint_status(task);
L
Linus Torvalds 已提交
889

890 891 892 893 894 895
	rpc_xdr_buf_init(&req->rq_snd_buf,
			 req->rq_buffer,
			 req->rq_callsize);
	rpc_xdr_buf_init(&req->rq_rcv_buf,
			 (char *)req->rq_buffer + req->rq_callsize,
			 req->rq_rcvsize);
L
Linus Torvalds 已提交
896 897 898 899 900 901 902 903

	/* Encode header and provided arguments */
	encode = task->tk_msg.rpc_proc->p_encode;
	if (!(p = call_header(task))) {
		printk(KERN_INFO "RPC: call_header failed, exit EIO\n");
		rpc_exit(task, -EIO);
		return;
	}
904 905 906
	if (encode == NULL)
		return;

907
	lock_kernel();
908 909
	task->tk_status = rpcauth_wrap_req(task, encode, req, p,
			task->tk_msg.rpc_argp);
910
	unlock_kernel();
911 912 913 914 915
	if (task->tk_status == -ENOMEM) {
		/* XXX: Is this sane? */
		rpc_delay(task, 3*HZ);
		task->tk_status = -EAGAIN;
	}
L
Linus Torvalds 已提交
916 917 918 919 920 921 922 923
}

/*
 * 4.	Get the server port number if not yet set
 */
static void
call_bind(struct rpc_task *task)
{
924
	struct rpc_xprt *xprt = task->tk_xprt;
L
Linus Torvalds 已提交
925

926
	dprint_status(task);
L
Linus Torvalds 已提交
927

928
	task->tk_action = call_connect;
929
	if (!xprt_bound(xprt)) {
930
		task->tk_action = call_bind_status;
931
		task->tk_timeout = xprt->bind_timeout;
932
		xprt->ops->rpcbind(task);
L
Linus Torvalds 已提交
933 934 935 936
	}
}

/*
937 938 939 940 941 942 943 944
 * 4a.	Sort out bind result
 */
static void
call_bind_status(struct rpc_task *task)
{
	int status = -EACCES;

	if (task->tk_status >= 0) {
945
		dprint_status(task);
946 947 948 949 950 951 952
		task->tk_status = 0;
		task->tk_action = call_connect;
		return;
	}

	switch (task->tk_status) {
	case -EACCES:
953 954
		dprintk("RPC: %5u remote rpcbind: RPC program/version "
				"unavailable\n", task->tk_pid);
955
		rpc_delay(task, 3*HZ);
956
		goto retry_timeout;
957
	case -ETIMEDOUT:
958
		dprintk("RPC: %5u rpcbind request timed out\n",
959
				task->tk_pid);
960
		goto retry_timeout;
961
	case -EPFNOSUPPORT:
962
		dprintk("RPC: %5u remote rpcbind service unavailable\n",
963 964 965
				task->tk_pid);
		break;
	case -EPROTONOSUPPORT:
966
		dprintk("RPC: %5u remote rpcbind version unavailable, retrying\n",
967
				task->tk_pid);
968 969 970
		task->tk_status = 0;
		task->tk_action = call_bind;
		return;
971
	default:
972
		dprintk("RPC: %5u unrecognized rpcbind error (%d)\n",
973 974 975 976 977 978 979
				task->tk_pid, -task->tk_status);
		status = -EIO;
	}

	rpc_exit(task, status);
	return;

980 981
retry_timeout:
	task->tk_action = call_timeout;
982 983 984 985
}

/*
 * 4b.	Connect to the RPC server
L
Linus Torvalds 已提交
986 987 988 989
 */
static void
call_connect(struct rpc_task *task)
{
990
	struct rpc_xprt *xprt = task->tk_xprt;
L
Linus Torvalds 已提交
991

992
	dprintk("RPC: %5u call_connect xprt %p %s connected\n",
993 994
			task->tk_pid, xprt,
			(xprt_connected(xprt) ? "is" : "is not"));
L
Linus Torvalds 已提交
995

996 997 998 999 1000 1001
	task->tk_action = call_transmit;
	if (!xprt_connected(xprt)) {
		task->tk_action = call_connect_status;
		if (task->tk_status < 0)
			return;
		xprt_connect(task);
L
Linus Torvalds 已提交
1002 1003 1004 1005
	}
}

/*
1006
 * 4c.	Sort out connect result
L
Linus Torvalds 已提交
1007 1008 1009 1010 1011 1012 1013
 */
static void
call_connect_status(struct rpc_task *task)
{
	struct rpc_clnt *clnt = task->tk_client;
	int status = task->tk_status;

1014
	dprint_status(task);
1015

L
Linus Torvalds 已提交
1016 1017 1018 1019 1020 1021 1022
	task->tk_status = 0;
	if (status >= 0) {
		clnt->cl_stats->netreconn++;
		task->tk_action = call_transmit;
		return;
	}

1023
	/* Something failed: remote service port may have changed */
1024
	rpc_force_rebind(clnt);
1025

L
Linus Torvalds 已提交
1026 1027 1028
	switch (status) {
	case -ENOTCONN:
	case -EAGAIN:
1029
		task->tk_action = call_bind;
1030 1031 1032 1033 1034 1035
		if (!RPC_IS_SOFT(task))
			return;
		/* if soft mounted, test if we've timed out */
	case -ETIMEDOUT:
		task->tk_action = call_timeout;
		return;
L
Linus Torvalds 已提交
1036
	}
1037
	rpc_exit(task, -EIO);
L
Linus Torvalds 已提交
1038 1039 1040 1041 1042 1043 1044 1045
}

/*
 * 5.	Transmit the RPC request, and wait for reply
 */
static void
call_transmit(struct rpc_task *task)
{
1046
	dprint_status(task);
L
Linus Torvalds 已提交
1047 1048 1049 1050 1051 1052 1053

	task->tk_action = call_status;
	if (task->tk_status < 0)
		return;
	task->tk_status = xprt_prepare_transmit(task);
	if (task->tk_status != 0)
		return;
1054
	task->tk_action = call_transmit_status;
L
Linus Torvalds 已提交
1055
	/* Encode here so that rpcsec_gss can use correct sequence number. */
1056
	if (rpc_task_need_encode(task)) {
1057
		BUG_ON(task->tk_rqstp->rq_bytes_sent != 0);
L
Linus Torvalds 已提交
1058
		call_encode(task);
1059 1060
		/* Did the encode result in an error condition? */
		if (task->tk_status != 0)
1061
			return;
1062
	}
L
Linus Torvalds 已提交
1063 1064 1065
	xprt_transmit(task);
	if (task->tk_status < 0)
		return;
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
	/*
	 * On success, ensure that we call xprt_end_transmit() before sleeping
	 * in order to allow access to the socket to other RPC requests.
	 */
	call_transmit_status(task);
	if (task->tk_msg.rpc_proc->p_decode != NULL)
		return;
	task->tk_action = rpc_exit_task;
	rpc_wake_up_task(task);
}

/*
 * 5a.	Handle cleanup after a transmission
 */
static void
call_transmit_status(struct rpc_task *task)
{
	task->tk_action = call_status;
	/*
	 * Special case: if we've been waiting on the socket's write_space()
	 * callback, then don't call xprt_end_transmit().
	 */
	if (task->tk_status == -EAGAIN)
		return;
	xprt_end_transmit(task);
1091
	rpc_task_force_reencode(task);
L
Linus Torvalds 已提交
1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106
}

/*
 * 6.	Sort out the RPC call status
 */
static void
call_status(struct rpc_task *task)
{
	struct rpc_clnt	*clnt = task->tk_client;
	struct rpc_rqst	*req = task->tk_rqstp;
	int		status;

	if (req->rq_received > 0 && !req->rq_bytes_sent)
		task->tk_status = req->rq_received;

1107
	dprint_status(task);
L
Linus Torvalds 已提交
1108 1109 1110 1111 1112 1113 1114 1115 1116

	status = task->tk_status;
	if (status >= 0) {
		task->tk_action = call_decode;
		return;
	}

	task->tk_status = 0;
	switch(status) {
1117 1118 1119 1120 1121 1122 1123 1124
	case -EHOSTDOWN:
	case -EHOSTUNREACH:
	case -ENETUNREACH:
		/*
		 * Delay any retries for 3 seconds, then handle as if it
		 * were a timeout.
		 */
		rpc_delay(task, 3*HZ);
L
Linus Torvalds 已提交
1125 1126
	case -ETIMEDOUT:
		task->tk_action = call_timeout;
1127 1128
		if (task->tk_client->cl_discrtry)
			xprt_disconnect(task->tk_xprt);
L
Linus Torvalds 已提交
1129 1130 1131
		break;
	case -ECONNREFUSED:
	case -ENOTCONN:
1132
		rpc_force_rebind(clnt);
L
Linus Torvalds 已提交
1133 1134 1135 1136 1137 1138 1139 1140 1141 1142
		task->tk_action = call_bind;
		break;
	case -EAGAIN:
		task->tk_action = call_transmit;
		break;
	case -EIO:
		/* shutdown or soft timeout */
		rpc_exit(task, status);
		break;
	default:
C
Chuck Lever 已提交
1143
		printk("%s: RPC call returned error %d\n",
L
Linus Torvalds 已提交
1144 1145 1146 1147 1148 1149
			       clnt->cl_protname, -status);
		rpc_exit(task, status);
	}
}

/*
1150
 * 6a.	Handle RPC timeout
L
Linus Torvalds 已提交
1151 1152 1153 1154 1155 1156 1157 1158 1159
 * 	We do not release the request slot, so we keep using the
 *	same XID for all retransmits.
 */
static void
call_timeout(struct rpc_task *task)
{
	struct rpc_clnt	*clnt = task->tk_client;

	if (xprt_adjust_timeout(task->tk_rqstp) == 0) {
1160
		dprintk("RPC: %5u call_timeout (minor)\n", task->tk_pid);
L
Linus Torvalds 已提交
1161 1162 1163
		goto retry;
	}

1164
	dprintk("RPC: %5u call_timeout (major)\n", task->tk_pid);
1165 1166
	task->tk_timeouts++;

L
Linus Torvalds 已提交
1167
	if (RPC_IS_SOFT(task)) {
C
Chuck Lever 已提交
1168
		printk(KERN_NOTICE "%s: server %s not responding, timed out\n",
L
Linus Torvalds 已提交
1169 1170 1171 1172 1173
				clnt->cl_protname, clnt->cl_server);
		rpc_exit(task, -EIO);
		return;
	}

C
Chuck Lever 已提交
1174
	if (!(task->tk_flags & RPC_CALL_MAJORSEEN)) {
L
Linus Torvalds 已提交
1175 1176 1177 1178
		task->tk_flags |= RPC_CALL_MAJORSEEN;
		printk(KERN_NOTICE "%s: server %s not responding, still trying\n",
			clnt->cl_protname, clnt->cl_server);
	}
1179
	rpc_force_rebind(clnt);
L
Linus Torvalds 已提交
1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195

retry:
	clnt->cl_stats->rpcretrans++;
	task->tk_action = call_bind;
	task->tk_status = 0;
}

/*
 * 7.	Decode the RPC reply
 */
static void
call_decode(struct rpc_task *task)
{
	struct rpc_clnt	*clnt = task->tk_client;
	struct rpc_rqst	*req = task->tk_rqstp;
	kxdrproc_t	decode = task->tk_msg.rpc_proc->p_decode;
1196
	__be32		*p;
L
Linus Torvalds 已提交
1197

1198 1199
	dprintk("RPC: %5u call_decode (status %d)\n",
			task->tk_pid, task->tk_status);
L
Linus Torvalds 已提交
1200

C
Chuck Lever 已提交
1201
	if (task->tk_flags & RPC_CALL_MAJORSEEN) {
L
Linus Torvalds 已提交
1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212
		printk(KERN_NOTICE "%s: server %s OK\n",
			clnt->cl_protname, clnt->cl_server);
		task->tk_flags &= ~RPC_CALL_MAJORSEEN;
	}

	if (task->tk_status < 12) {
		if (!RPC_IS_SOFT(task)) {
			task->tk_action = call_bind;
			clnt->cl_stats->rpcretrans++;
			goto out_retry;
		}
1213 1214
		dprintk("RPC:       %s: too small RPC reply size (%d bytes)\n",
				clnt->cl_protname, task->tk_status);
1215 1216
		task->tk_action = call_timeout;
		goto out_retry;
L
Linus Torvalds 已提交
1217 1218
	}

1219 1220 1221 1222 1223
	/*
	 * Ensure that we see all writes made by xprt_complete_rqst()
	 * before it changed req->rq_received.
	 */
	smp_rmb();
L
Linus Torvalds 已提交
1224 1225 1226 1227 1228 1229 1230
	req->rq_rcv_buf.len = req->rq_private_buf.len;

	/* Check that the softirq receive buffer is valid */
	WARN_ON(memcmp(&req->rq_rcv_buf, &req->rq_private_buf,
				sizeof(req->rq_rcv_buf)) != 0);

	/* Verify the RPC header */
T
Trond Myklebust 已提交
1231 1232 1233 1234 1235
	p = call_verify(task);
	if (IS_ERR(p)) {
		if (p == ERR_PTR(-EAGAIN))
			goto out_retry;
		return;
L
Linus Torvalds 已提交
1236 1237
	}

T
Trond Myklebust 已提交
1238
	task->tk_action = rpc_exit_task;
L
Linus Torvalds 已提交
1239

1240 1241
	if (decode) {
		lock_kernel();
L
Linus Torvalds 已提交
1242 1243
		task->tk_status = rpcauth_unwrap_resp(task, decode, req, p,
						      task->tk_msg.rpc_resp);
1244 1245
		unlock_kernel();
	}
1246 1247
	dprintk("RPC: %5u call_decode result %d\n", task->tk_pid,
			task->tk_status);
L
Linus Torvalds 已提交
1248 1249 1250 1251
	return;
out_retry:
	req->rq_received = req->rq_private_buf.len = 0;
	task->tk_status = 0;
1252 1253
	if (task->tk_client->cl_discrtry)
		xprt_disconnect(task->tk_xprt);
L
Linus Torvalds 已提交
1254 1255 1256 1257 1258 1259 1260 1261
}

/*
 * 8.	Refresh the credentials if rejected by the server
 */
static void
call_refresh(struct rpc_task *task)
{
1262
	dprint_status(task);
L
Linus Torvalds 已提交
1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277

	xprt_release(task);	/* Must do to obtain new XID */
	task->tk_action = call_refreshresult;
	task->tk_status = 0;
	task->tk_client->cl_stats->rpcauthrefresh++;
	rpcauth_refreshcred(task);
}

/*
 * 8a.	Process the results of a credential refresh
 */
static void
call_refreshresult(struct rpc_task *task)
{
	int status = task->tk_status;
1278 1279

	dprint_status(task);
L
Linus Torvalds 已提交
1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297

	task->tk_status = 0;
	task->tk_action = call_reserve;
	if (status >= 0 && rpcauth_uptodatecred(task))
		return;
	if (status == -EACCES) {
		rpc_exit(task, -EACCES);
		return;
	}
	task->tk_action = call_refresh;
	if (status != -ETIMEDOUT)
		rpc_delay(task, 3*HZ);
	return;
}

/*
 * Call header serialization
 */
1298
static __be32 *
L
Linus Torvalds 已提交
1299 1300 1301 1302
call_header(struct rpc_task *task)
{
	struct rpc_clnt *clnt = task->tk_client;
	struct rpc_rqst	*req = task->tk_rqstp;
1303
	__be32		*p = req->rq_svec[0].iov_base;
L
Linus Torvalds 已提交
1304 1305

	/* FIXME: check buffer size? */
1306 1307

	p = xprt_skip_transport_header(task->tk_xprt, p);
L
Linus Torvalds 已提交
1308 1309 1310 1311 1312 1313
	*p++ = req->rq_xid;		/* XID */
	*p++ = htonl(RPC_CALL);		/* CALL */
	*p++ = htonl(RPC_VERSION);	/* RPC version */
	*p++ = htonl(clnt->cl_prog);	/* program number */
	*p++ = htonl(clnt->cl_vers);	/* program version */
	*p++ = htonl(task->tk_msg.rpc_proc->p_proc);	/* procedure */
1314 1315 1316
	p = rpcauth_marshcred(task, p);
	req->rq_slen = xdr_adjust_iovec(&req->rq_svec[0], p);
	return p;
L
Linus Torvalds 已提交
1317 1318 1319 1320 1321
}

/*
 * Reply header verification
 */
1322
static __be32 *
L
Linus Torvalds 已提交
1323 1324 1325 1326
call_verify(struct rpc_task *task)
{
	struct kvec *iov = &task->tk_rqstp->rq_rcv_buf.head[0];
	int len = task->tk_rqstp->rq_rcv_buf.len >> 2;
1327 1328
	__be32	*p = iov->iov_base;
	u32 n;
L
Linus Torvalds 已提交
1329 1330
	int error = -EACCES;

1331 1332 1333 1334 1335 1336
	if ((task->tk_rqstp->rq_rcv_buf.len & 3) != 0) {
		/* RFC-1014 says that the representation of XDR data must be a
		 * multiple of four bytes
		 * - if it isn't pointer subtraction in the NFS client may give
		 *   undefined results
		 */
1337 1338 1339
		dprintk("RPC: %5u %s: XDR representation not a multiple of"
		       " 4 bytes: 0x%x\n", task->tk_pid, __FUNCTION__,
		       task->tk_rqstp->rq_rcv_buf.len);
1340 1341
		goto out_eio;
	}
L
Linus Torvalds 已提交
1342 1343 1344 1345 1346
	if ((len -= 3) < 0)
		goto out_overflow;
	p += 1;	/* skip XID */

	if ((n = ntohl(*p++)) != RPC_REPLY) {
1347 1348
		dprintk("RPC: %5u %s: not an RPC reply: %x\n",
				task->tk_pid, __FUNCTION__, n);
T
Trond Myklebust 已提交
1349
		goto out_garbage;
L
Linus Torvalds 已提交
1350 1351 1352 1353 1354 1355 1356 1357
	}
	if ((n = ntohl(*p++)) != RPC_MSG_ACCEPTED) {
		if (--len < 0)
			goto out_overflow;
		switch ((n = ntohl(*p++))) {
			case RPC_AUTH_ERROR:
				break;
			case RPC_MISMATCH:
1358 1359 1360
				dprintk("RPC: %5u %s: RPC call version "
						"mismatch!\n",
						task->tk_pid, __FUNCTION__);
1361 1362
				error = -EPROTONOSUPPORT;
				goto out_err;
L
Linus Torvalds 已提交
1363
			default:
1364 1365 1366
				dprintk("RPC: %5u %s: RPC call rejected, "
						"unknown error: %x\n",
						task->tk_pid, __FUNCTION__, n);
L
Linus Torvalds 已提交
1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378
				goto out_eio;
		}
		if (--len < 0)
			goto out_overflow;
		switch ((n = ntohl(*p++))) {
		case RPC_AUTH_REJECTEDCRED:
		case RPC_AUTH_REJECTEDVERF:
		case RPCSEC_GSS_CREDPROBLEM:
		case RPCSEC_GSS_CTXPROBLEM:
			if (!task->tk_cred_retry)
				break;
			task->tk_cred_retry--;
1379 1380
			dprintk("RPC: %5u %s: retry stale creds\n",
					task->tk_pid, __FUNCTION__);
L
Linus Torvalds 已提交
1381 1382
			rpcauth_invalcred(task);
			task->tk_action = call_refresh;
T
Trond Myklebust 已提交
1383
			goto out_retry;
L
Linus Torvalds 已提交
1384 1385 1386 1387 1388 1389
		case RPC_AUTH_BADCRED:
		case RPC_AUTH_BADVERF:
			/* possibly garbled cred/verf? */
			if (!task->tk_garb_retry)
				break;
			task->tk_garb_retry--;
1390 1391
			dprintk("RPC: %5u %s: retry garbled creds\n",
					task->tk_pid, __FUNCTION__);
L
Linus Torvalds 已提交
1392
			task->tk_action = call_bind;
T
Trond Myklebust 已提交
1393
			goto out_retry;
L
Linus Torvalds 已提交
1394
		case RPC_AUTH_TOOWEAK:
1395 1396
			printk(KERN_NOTICE "call_verify: server %s requires stronger "
			       "authentication.\n", task->tk_client->cl_server);
L
Linus Torvalds 已提交
1397 1398
			break;
		default:
1399 1400
			dprintk("RPC: %5u %s: unknown auth error: %x\n",
					task->tk_pid, __FUNCTION__, n);
L
Linus Torvalds 已提交
1401 1402
			error = -EIO;
		}
1403 1404
		dprintk("RPC: %5u %s: call rejected %d\n",
				task->tk_pid, __FUNCTION__, n);
L
Linus Torvalds 已提交
1405 1406 1407
		goto out_err;
	}
	if (!(p = rpcauth_checkverf(task, p))) {
1408 1409
		dprintk("RPC: %5u %s: auth check failed\n",
				task->tk_pid, __FUNCTION__);
T
Trond Myklebust 已提交
1410
		goto out_garbage;		/* bad verifier, retry */
L
Linus Torvalds 已提交
1411
	}
1412
	len = p - (__be32 *)iov->iov_base - 1;
L
Linus Torvalds 已提交
1413 1414 1415 1416 1417 1418
	if (len < 0)
		goto out_overflow;
	switch ((n = ntohl(*p++))) {
	case RPC_SUCCESS:
		return p;
	case RPC_PROG_UNAVAIL:
1419 1420
		dprintk("RPC: %5u %s: program %u is unsupported by server %s\n",
				task->tk_pid, __FUNCTION__,
L
Linus Torvalds 已提交
1421 1422
				(unsigned int)task->tk_client->cl_prog,
				task->tk_client->cl_server);
1423 1424
		error = -EPFNOSUPPORT;
		goto out_err;
L
Linus Torvalds 已提交
1425
	case RPC_PROG_MISMATCH:
1426 1427
		dprintk("RPC: %5u %s: program %u, version %u unsupported by "
				"server %s\n", task->tk_pid, __FUNCTION__,
L
Linus Torvalds 已提交
1428 1429 1430
				(unsigned int)task->tk_client->cl_prog,
				(unsigned int)task->tk_client->cl_vers,
				task->tk_client->cl_server);
1431 1432
		error = -EPROTONOSUPPORT;
		goto out_err;
L
Linus Torvalds 已提交
1433
	case RPC_PROC_UNAVAIL:
1434 1435 1436
		dprintk("RPC: %5u %s: proc %p unsupported by program %u, "
				"version %u on server %s\n",
				task->tk_pid, __FUNCTION__,
L
Linus Torvalds 已提交
1437 1438 1439 1440
				task->tk_msg.rpc_proc,
				task->tk_client->cl_prog,
				task->tk_client->cl_vers,
				task->tk_client->cl_server);
1441 1442
		error = -EOPNOTSUPP;
		goto out_err;
L
Linus Torvalds 已提交
1443
	case RPC_GARBAGE_ARGS:
1444 1445
		dprintk("RPC: %5u %s: server saw garbage\n",
				task->tk_pid, __FUNCTION__);
L
Linus Torvalds 已提交
1446 1447
		break;			/* retry */
	default:
1448 1449
		dprintk("RPC: %5u %s: server accept status: %x\n",
				task->tk_pid, __FUNCTION__, n);
L
Linus Torvalds 已提交
1450 1451 1452
		/* Also retry */
	}

T
Trond Myklebust 已提交
1453
out_garbage:
L
Linus Torvalds 已提交
1454 1455 1456
	task->tk_client->cl_stats->rpcgarbage++;
	if (task->tk_garb_retry) {
		task->tk_garb_retry--;
1457 1458
		dprintk("RPC: %5u %s: retrying\n",
				task->tk_pid, __FUNCTION__);
L
Linus Torvalds 已提交
1459
		task->tk_action = call_bind;
T
Trond Myklebust 已提交
1460 1461
out_retry:
		return ERR_PTR(-EAGAIN);
L
Linus Torvalds 已提交
1462 1463 1464 1465 1466
	}
out_eio:
	error = -EIO;
out_err:
	rpc_exit(task, error);
1467 1468
	dprintk("RPC: %5u %s: call failed with error %d\n", task->tk_pid,
			__FUNCTION__, error);
T
Trond Myklebust 已提交
1469
	return ERR_PTR(error);
L
Linus Torvalds 已提交
1470
out_overflow:
1471 1472
	dprintk("RPC: %5u %s: server reply was truncated.\n", task->tk_pid,
			__FUNCTION__);
T
Trond Myklebust 已提交
1473
	goto out_garbage;
L
Linus Torvalds 已提交
1474
}
1475

1476
static int rpcproc_encode_null(void *rqstp, __be32 *data, void *obj)
1477 1478 1479 1480
{
	return 0;
}

1481
static int rpcproc_decode_null(void *rqstp, __be32 *data, void *obj)
1482 1483 1484 1485 1486 1487 1488 1489 1490
{
	return 0;
}

static struct rpc_procinfo rpcproc_null = {
	.p_encode = rpcproc_encode_null,
	.p_decode = rpcproc_decode_null,
};

T
Trond Myklebust 已提交
1491
static int rpc_ping(struct rpc_clnt *clnt, int flags)
1492 1493 1494 1495 1496 1497 1498 1499 1500 1501
{
	struct rpc_message msg = {
		.rpc_proc = &rpcproc_null,
	};
	int err;
	msg.rpc_cred = authnull_ops.lookup_cred(NULL, NULL, 0);
	err = rpc_call_sync(clnt, &msg, flags);
	put_rpccred(msg.rpc_cred);
	return err;
}
1502

1503 1504 1505 1506 1507 1508 1509 1510 1511 1512
struct rpc_task *rpc_call_null(struct rpc_clnt *clnt, struct rpc_cred *cred, int flags)
{
	struct rpc_message msg = {
		.rpc_proc = &rpcproc_null,
		.rpc_cred = cred,
	};
	return rpc_do_run_task(clnt, &msg, flags, &rpc_default_ops, NULL);
}
EXPORT_SYMBOL(rpc_call_null);

1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549
#ifdef RPC_DEBUG
void rpc_show_tasks(void)
{
	struct rpc_clnt *clnt;
	struct rpc_task *t;

	spin_lock(&rpc_client_lock);
	if (list_empty(&all_clients))
		goto out;
	printk("-pid- proc flgs status -client- -prog- --rqstp- -timeout "
		"-rpcwait -action- ---ops--\n");
	list_for_each_entry(clnt, &all_clients, cl_clients) {
		if (list_empty(&clnt->cl_tasks))
			continue;
		spin_lock(&clnt->cl_lock);
		list_for_each_entry(t, &clnt->cl_tasks, tk_task) {
			const char *rpc_waitq = "none";

			if (RPC_IS_QUEUED(t))
				rpc_waitq = rpc_qname(t->u.tk_wait.rpc_waitq);

			printk("%5u %04d %04x %6d %8p %6d %8p %8ld %8s %8p %8p\n",
				t->tk_pid,
				(t->tk_msg.rpc_proc ? t->tk_msg.rpc_proc->p_proc : -1),
				t->tk_flags, t->tk_status,
				t->tk_client,
				(t->tk_client ? t->tk_client->cl_prog : 0),
				t->tk_rqstp, t->tk_timeout,
				rpc_waitq,
				t->tk_action, t->tk_ops);
		}
		spin_unlock(&clnt->cl_lock);
	}
out:
	spin_unlock(&rpc_client_lock);
}
#endif