rpcb_clnt.c 26.0 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14
/*
 * In-kernel rpcbind client supporting versions 2, 3, and 4 of the rpcbind
 * protocol
 *
 * Based on RFC 1833: "Binding Protocols for ONC RPC Version 2" and
 * RFC 3530: "Network File System (NFS) version 4 Protocol"
 *
 * Original: Gilles Quillard, Bull Open Source, 2005 <gilles.quillard@bull.net>
 * Updated: Chuck Lever, Oracle Corporation, 2007 <chuck.lever@oracle.com>
 *
 * Descended from net/sunrpc/pmap_clnt.c,
 *  Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de>
 */

15 16
#include <linux/module.h>

17 18
#include <linux/types.h>
#include <linux/socket.h>
19 20
#include <linux/in.h>
#include <linux/in6.h>
21 22
#include <linux/kernel.h>
#include <linux/errno.h>
23
#include <linux/mutex.h>
24
#include <linux/slab.h>
25
#include <net/ipv6.h>
26 27 28

#include <linux/sunrpc/clnt.h>
#include <linux/sunrpc/sched.h>
29
#include <linux/sunrpc/xprtsock.h>
30 31 32 33 34 35 36 37

#ifdef RPC_DEBUG
# define RPCDBG_FACILITY	RPCDBG_BIND
#endif

#define RPCBIND_PROGRAM		(100000u)
#define RPCBIND_PORT		(111u)

38 39 40 41
#define RPCBVERS_2		(2u)
#define RPCBVERS_3		(3u)
#define RPCBVERS_4		(4u)

42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63
enum {
	RPCBPROC_NULL,
	RPCBPROC_SET,
	RPCBPROC_UNSET,
	RPCBPROC_GETPORT,
	RPCBPROC_GETADDR = 3,		/* alias for GETPORT */
	RPCBPROC_DUMP,
	RPCBPROC_CALLIT,
	RPCBPROC_BCAST = 5,		/* alias for CALLIT */
	RPCBPROC_GETTIME,
	RPCBPROC_UADDR2TADDR,
	RPCBPROC_TADDR2UADDR,
	RPCBPROC_GETVERSADDR,
	RPCBPROC_INDIRECT,
	RPCBPROC_GETADDRLIST,
	RPCBPROC_GETSTAT,
};

/*
 * r_owner
 *
 * The "owner" is allowed to unset a service in the rpcbind database.
64 65 66 67 68 69 70 71
 *
 * For AF_LOCAL SET/UNSET requests, rpcbind treats this string as a
 * UID which it maps to a local user name via a password lookup.
 * In all other cases it is ignored.
 *
 * For SET/UNSET requests, user space provides a value, even for
 * network requests, and GETADDR uses an empty string.  We follow
 * those precedents here.
72
 */
73
#define RPCB_OWNER_STRING	"0"
74 75
#define RPCB_MAXOWNERLEN	sizeof(RPCB_OWNER_STRING)

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 106
/*
 * XDR data type sizes
 */
#define RPCB_program_sz		(1)
#define RPCB_version_sz		(1)
#define RPCB_protocol_sz	(1)
#define RPCB_port_sz		(1)
#define RPCB_boolean_sz		(1)

#define RPCB_netid_sz		(1 + XDR_QUADLEN(RPCBIND_MAXNETIDLEN))
#define RPCB_addr_sz		(1 + XDR_QUADLEN(RPCBIND_MAXUADDRLEN))
#define RPCB_ownerstring_sz	(1 + XDR_QUADLEN(RPCB_MAXOWNERLEN))

/*
 * XDR argument and result sizes
 */
#define RPCB_mappingargs_sz	(RPCB_program_sz + RPCB_version_sz + \
				RPCB_protocol_sz + RPCB_port_sz)
#define RPCB_getaddrargs_sz	(RPCB_program_sz + RPCB_version_sz + \
				RPCB_netid_sz + RPCB_addr_sz + \
				RPCB_ownerstring_sz)

#define RPCB_getportres_sz	RPCB_port_sz
#define RPCB_setres_sz		RPCB_boolean_sz

/*
 * Note that RFC 1833 does not put any size restrictions on the
 * address string returned by the remote rpcbind database.
 */
#define RPCB_getaddrres_sz	RPCB_addr_sz

107
static void			rpcb_getport_done(struct rpc_task *, void *);
108
static void			rpcb_map_release(void *data);
109
static struct rpc_program	rpcb_program;
110

111 112 113
static struct rpc_clnt *	rpcb_local_clnt;
static struct rpc_clnt *	rpcb_local_clnt4;

114 115 116 117 118 119 120
struct rpcbind_args {
	struct rpc_xprt *	r_xprt;

	u32			r_prog;
	u32			r_vers;
	u32			r_prot;
	unsigned short		r_port;
121 122 123
	const char *		r_netid;
	const char *		r_addr;
	const char *		r_owner;
124 125

	int			r_status;
126 127 128 129
};

static struct rpc_procinfo rpcb_procedures2[];
static struct rpc_procinfo rpcb_procedures3[];
130
static struct rpc_procinfo rpcb_procedures4[];
131

132
struct rpcb_info {
133
	u32			rpc_vers;
134
	struct rpc_procinfo *	rpc_proc;
135 136 137 138
};

static struct rpcb_info rpcb_next_version[];
static struct rpcb_info rpcb_next_version6[];
139 140 141 142 143 144 145 146 147 148 149 150

static const struct rpc_call_ops rpcb_getport_ops = {
	.rpc_call_done		= rpcb_getport_done,
	.rpc_release		= rpcb_map_release,
};

static void rpcb_wake_rpcbind_waiters(struct rpc_xprt *xprt, int status)
{
	xprt_clear_binding(xprt);
	rpc_wake_up_status(&xprt->binding, status);
}

151 152 153 154 155 156
static void rpcb_map_release(void *data)
{
	struct rpcbind_args *map = data;

	rpcb_wake_rpcbind_waiters(map->r_xprt, map->r_status);
	xprt_put(map->r_xprt);
157
	kfree(map->r_addr);
158 159 160
	kfree(map);
}

161 162 163 164 165 166
static const struct sockaddr_in rpcb_inaddr_loopback = {
	.sin_family		= AF_INET,
	.sin_addr.s_addr	= htonl(INADDR_LOOPBACK),
	.sin_port		= htons(RPCBIND_PORT),
};

167 168 169 170 171 172 173
static DEFINE_MUTEX(rpcb_create_local_mutex);

/*
 * Returns zero on success, otherwise a negative errno value
 * is returned.
 */
static int rpcb_create_local(void)
174 175
{
	struct rpc_create_args args = {
176
		.net		= &init_net,
177
		.protocol	= XPRT_TRANSPORT_TCP,
178 179
		.address	= (struct sockaddr *)&rpcb_inaddr_loopback,
		.addrsize	= sizeof(rpcb_inaddr_loopback),
180 181
		.servername	= "localhost",
		.program	= &rpcb_program,
182
		.version	= RPCBVERS_2,
183 184 185
		.authflavor	= RPC_AUTH_UNIX,
		.flags		= RPC_CLNT_CREATE_NOPING,
	};
186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202
	struct rpc_clnt *clnt, *clnt4;
	int result = 0;

	if (rpcb_local_clnt)
		return result;

	mutex_lock(&rpcb_create_local_mutex);
	if (rpcb_local_clnt)
		goto out;

	clnt = rpc_create(&args);
	if (IS_ERR(clnt)) {
		dprintk("RPC:       failed to create local rpcbind "
				"client (errno %ld).\n", PTR_ERR(clnt));
		result = -PTR_ERR(clnt);
		goto out;
	}
203

204 205 206 207 208 209 210
	/*
	 * This results in an RPC ping.  On systems running portmapper,
	 * the v4 ping will fail.  Proceed anyway, but disallow rpcb
	 * v4 upcalls.
	 */
	clnt4 = rpc_bind_new_program(clnt, &rpcb_program, RPCBVERS_4);
	if (IS_ERR(clnt4)) {
211 212 213
		dprintk("RPC:       failed to bind second program to "
				"rpcbind v4 client (errno %ld).\n",
				PTR_ERR(clnt4));
214 215 216 217 218 219 220 221 222
		clnt4 = NULL;
	}

	rpcb_local_clnt = clnt;
	rpcb_local_clnt4 = clnt4;

out:
	mutex_unlock(&rpcb_create_local_mutex);
	return result;
223 224
}

225
static struct rpc_clnt *rpcb_create(char *hostname, struct sockaddr *srvaddr,
226
				    size_t salen, int proto, u32 version)
227 228
{
	struct rpc_create_args args = {
229
		.net		= &init_net,
230 231
		.protocol	= proto,
		.address	= srvaddr,
232
		.addrsize	= salen,
233 234 235 236
		.servername	= hostname,
		.program	= &rpcb_program,
		.version	= version,
		.authflavor	= RPC_AUTH_UNIX,
237 238
		.flags		= (RPC_CLNT_CREATE_NOPING |
					RPC_CLNT_CREATE_NONPRIVPORT),
239 240
	};

241 242 243 244 245 246 247 248
	switch (srvaddr->sa_family) {
	case AF_INET:
		((struct sockaddr_in *)srvaddr)->sin_port = htons(RPCBIND_PORT);
		break;
	case AF_INET6:
		((struct sockaddr_in6 *)srvaddr)->sin6_port = htons(RPCBIND_PORT);
		break;
	default:
249
		return ERR_PTR(-EAFNOSUPPORT);
250 251
	}

252 253 254
	return rpc_create(&args);
}

255
static int rpcb_register_call(struct rpc_clnt *clnt, struct rpc_message *msg)
256
{
257
	int result, error = 0;
258

259
	msg->rpc_resp = &result;
260

261
	error = rpc_call_sync(clnt, msg, RPC_TASK_SOFTCONN);
262
	if (error < 0) {
263
		dprintk("RPC:       failed to contact local rpcbind "
264
				"server (errno %d).\n", -error);
265 266 267
		return error;
	}

268
	if (!result)
269 270
		return -EACCES;
	return 0;
271 272
}

273 274 275 276
/**
 * rpcb_register - set or unset a port registration with the local rpcbind svc
 * @prog: RPC program number to bind
 * @vers: RPC version number to bind
277
 * @prot: transport protocol to register
278
 * @port: port value to register
279 280 281 282 283
 *
 * Returns zero if the registration request was dispatched successfully
 * and the rpcbind daemon returned success.  Otherwise, returns an errno
 * value that reflects the nature of the error (request could not be
 * dispatched, timed out, or rpcbind returned an error).
284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303
 *
 * RPC services invoke this function to advertise their contact
 * information via the system's rpcbind daemon.  RPC services
 * invoke this function once for each [program, version, transport]
 * tuple they wish to advertise.
 *
 * Callers may also unregister RPC services that are no longer
 * available by setting the passed-in port to zero.  This removes
 * all registered transports for [program, version] from the local
 * rpcbind database.
 *
 * This function uses rpcbind protocol version 2 to contact the
 * local rpcbind daemon.
 *
 * Registration works over both AF_INET and AF_INET6, and services
 * registered via this function are advertised as available for any
 * address.  If the local rpcbind daemon is listening on AF_INET6,
 * services registered via this function will be advertised on
 * IN6ADDR_ANY (ie available for all AF_INET and AF_INET6
 * addresses).
304
 */
305
int rpcb_register(u32 prog, u32 vers, int prot, unsigned short port)
306 307 308 309 310 311 312 313 314 315
{
	struct rpcbind_args map = {
		.r_prog		= prog,
		.r_vers		= vers,
		.r_prot		= prot,
		.r_port		= port,
	};
	struct rpc_message msg = {
		.rpc_argp	= &map,
	};
316 317 318 319 320
	int error;

	error = rpcb_create_local();
	if (error)
		return error;
321 322 323 324 325

	dprintk("RPC:       %sregistering (%u, %u, %d, %u) with local "
			"rpcbind\n", (port ? "" : "un"),
			prog, vers, prot, port);

326 327 328
	msg.rpc_proc = &rpcb_procedures2[RPCBPROC_UNSET];
	if (port)
		msg.rpc_proc = &rpcb_procedures2[RPCBPROC_SET];
329

330
	return rpcb_register_call(rpcb_local_clnt, &msg);
331 332
}

333 334 335
/*
 * Fill in AF_INET family-specific arguments to register
 */
336 337
static int rpcb_register_inet4(const struct sockaddr *sap,
			       struct rpc_message *msg)
338
{
339
	const struct sockaddr_in *sin = (const struct sockaddr_in *)sap;
340
	struct rpcbind_args *map = msg->rpc_argp;
341
	unsigned short port = ntohs(sin->sin_port);
342
	int result;
343

344
	map->r_addr = rpc_sockaddr2uaddr(sap);
345 346 347 348 349 350 351 352 353 354

	dprintk("RPC:       %sregistering [%u, %u, %s, '%s'] with "
		"local rpcbind\n", (port ? "" : "un"),
			map->r_prog, map->r_vers,
			map->r_addr, map->r_netid);

	msg->rpc_proc = &rpcb_procedures4[RPCBPROC_UNSET];
	if (port)
		msg->rpc_proc = &rpcb_procedures4[RPCBPROC_SET];

355
	result = rpcb_register_call(rpcb_local_clnt4, msg);
356 357
	kfree(map->r_addr);
	return result;
358 359 360 361 362
}

/*
 * Fill in AF_INET6 family-specific arguments to register
 */
363 364
static int rpcb_register_inet6(const struct sockaddr *sap,
			       struct rpc_message *msg)
365
{
366
	const struct sockaddr_in6 *sin6 = (const struct sockaddr_in6 *)sap;
367
	struct rpcbind_args *map = msg->rpc_argp;
368
	unsigned short port = ntohs(sin6->sin6_port);
369
	int result;
370

371
	map->r_addr = rpc_sockaddr2uaddr(sap);
372 373 374 375 376 377 378 379 380 381

	dprintk("RPC:       %sregistering [%u, %u, %s, '%s'] with "
		"local rpcbind\n", (port ? "" : "un"),
			map->r_prog, map->r_vers,
			map->r_addr, map->r_netid);

	msg->rpc_proc = &rpcb_procedures4[RPCBPROC_UNSET];
	if (port)
		msg->rpc_proc = &rpcb_procedures4[RPCBPROC_SET];

382
	result = rpcb_register_call(rpcb_local_clnt4, msg);
383 384
	kfree(map->r_addr);
	return result;
385 386
}

387 388 389 390 391 392 393 394 395 396 397
static int rpcb_unregister_all_protofamilies(struct rpc_message *msg)
{
	struct rpcbind_args *map = msg->rpc_argp;

	dprintk("RPC:       unregistering [%u, %u, '%s'] with "
		"local rpcbind\n",
			map->r_prog, map->r_vers, map->r_netid);

	map->r_addr = "";
	msg->rpc_proc = &rpcb_procedures4[RPCBPROC_UNSET];

398
	return rpcb_register_call(rpcb_local_clnt4, msg);
399 400
}

401 402 403 404 405 406
/**
 * rpcb_v4_register - set or unset a port registration with the local rpcbind
 * @program: RPC program number of service to (un)register
 * @version: RPC version number of service to (un)register
 * @address: address family, IP address, and port to (un)register
 * @netid: netid of transport protocol to (un)register
407 408 409 410 411
 *
 * Returns zero if the registration request was dispatched successfully
 * and the rpcbind daemon returned success.  Otherwise, returns an errno
 * value that reflects the nature of the error (request could not be
 * dispatched, timed out, or rpcbind returned an error).
412 413 414 415 416 417
 *
 * RPC services invoke this function to advertise their contact
 * information via the system's rpcbind daemon.  RPC services
 * invoke this function once for each [program, version, address,
 * netid] tuple they wish to advertise.
 *
418 419 420 421 422
 * Callers may also unregister RPC services that are registered at a
 * specific address by setting the port number in @address to zero.
 * They may unregister all registered protocol families at once for
 * a service by passing a NULL @address argument.  If @netid is ""
 * then all netids for [program, version, address] are unregistered.
423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444
 *
 * This function uses rpcbind protocol version 4 to contact the
 * local rpcbind daemon.  The local rpcbind daemon must support
 * version 4 of the rpcbind protocol in order for these functions
 * to register a service successfully.
 *
 * Supported netids include "udp" and "tcp" for UDP and TCP over
 * IPv4, and "udp6" and "tcp6" for UDP and TCP over IPv6,
 * respectively.
 *
 * The contents of @address determine the address family and the
 * port to be registered.  The usual practice is to pass INADDR_ANY
 * as the raw address, but specifying a non-zero address is also
 * supported by this API if the caller wishes to advertise an RPC
 * service on a specific network interface.
 *
 * Note that passing in INADDR_ANY does not create the same service
 * registration as IN6ADDR_ANY.  The former advertises an RPC
 * service on any IPv4 address, but not on IPv6.  The latter
 * advertises the service on all IPv4 and IPv6 addresses.
 */
int rpcb_v4_register(const u32 program, const u32 version,
445
		     const struct sockaddr *address, const char *netid)
446 447 448 449 450 451 452 453 454 455
{
	struct rpcbind_args map = {
		.r_prog		= program,
		.r_vers		= version,
		.r_netid	= netid,
		.r_owner	= RPCB_OWNER_STRING,
	};
	struct rpc_message msg = {
		.rpc_argp	= &map,
	};
456 457 458 459 460 461 462
	int error;

	error = rpcb_create_local();
	if (error)
		return error;
	if (rpcb_local_clnt4 == NULL)
		return -EPROTONOSUPPORT;
463

464 465 466
	if (address == NULL)
		return rpcb_unregister_all_protofamilies(&msg);

467 468
	switch (address->sa_family) {
	case AF_INET:
469
		return rpcb_register_inet4(address, &msg);
470
	case AF_INET6:
471
		return rpcb_register_inet6(address, &msg);
472 473 474 475 476
	}

	return -EAFNOSUPPORT;
}

477
static struct rpc_task *rpcb_call_async(struct rpc_clnt *rpcb_clnt, struct rpcbind_args *map, struct rpc_procinfo *proc)
478 479
{
	struct rpc_message msg = {
480
		.rpc_proc = proc,
481
		.rpc_argp = map,
482
		.rpc_resp = map,
483 484 485 486 487 488
	};
	struct rpc_task_setup task_setup_data = {
		.rpc_client = rpcb_clnt,
		.rpc_message = &msg,
		.callback_ops = &rpcb_getport_ops,
		.callback_data = map,
489
		.flags = RPC_TASK_ASYNC | RPC_TASK_SOFTCONN,
490 491 492 493 494
	};

	return rpc_run_task(&task_setup_data);
}

495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516
/*
 * In the case where rpc clients have been cloned, we want to make
 * sure that we use the program number/version etc of the actual
 * owner of the xprt. To do so, we walk back up the tree of parents
 * to find whoever created the transport and/or whoever has the
 * autobind flag set.
 */
static struct rpc_clnt *rpcb_find_transport_owner(struct rpc_clnt *clnt)
{
	struct rpc_clnt *parent = clnt->cl_parent;

	while (parent != clnt) {
		if (parent->cl_xprt != clnt->cl_xprt)
			break;
		if (clnt->cl_autobind)
			break;
		clnt = parent;
		parent = parent->cl_parent;
	}
	return clnt;
}

517
/**
518
 * rpcb_getport_async - obtain the port for a given RPC service on a given host
519 520 521 522 523
 * @task: task that is waiting for portmapper request
 *
 * This one can be called for an ongoing RPC request, and can be used in
 * an async (rpciod) context.
 */
524
void rpcb_getport_async(struct rpc_task *task)
525
{
526
	struct rpc_clnt *clnt;
527
	struct rpc_procinfo *proc;
528
	u32 bind_version;
529
	struct rpc_xprt *xprt;
530 531 532
	struct rpc_clnt	*rpcb_clnt;
	static struct rpcbind_args *map;
	struct rpc_task	*child;
533 534 535
	struct sockaddr_storage addr;
	struct sockaddr *sap = (struct sockaddr *)&addr;
	size_t salen;
536 537
	int status;

538 539 540
	clnt = rpcb_find_transport_owner(task->tk_client);
	xprt = clnt->cl_xprt;

541
	dprintk("RPC: %5u %s(%s, %u, %u, %d)\n",
542
		task->tk_pid, __func__,
543
		clnt->cl_server, clnt->cl_prog, clnt->cl_vers, xprt->prot);
544

545 546 547 548
	/* Put self on the wait queue to ensure we get notified if
	 * some other task is already attempting to bind the port */
	rpc_sleep_on(&xprt->binding, task, NULL);

549
	if (xprt_test_and_set_binding(xprt)) {
550
		dprintk("RPC: %5u %s: waiting for another binder\n",
551
			task->tk_pid, __func__);
552
		return;
553 554 555 556 557
	}

	/* Someone else may have bound if we slept */
	if (xprt_bound(xprt)) {
		status = 0;
558
		dprintk("RPC: %5u %s: already bound\n",
559
			task->tk_pid, __func__);
560 561 562
		goto bailout_nofree;
	}

563
	/* Parent transport's destination address */
564
	salen = rpc_peeraddr(clnt, sap, sizeof(addr));
565 566

	/* Don't ever use rpcbind v2 for AF_INET6 requests */
567
	switch (sap->sa_family) {
568
	case AF_INET:
569 570
		proc = rpcb_next_version[xprt->bind_index].rpc_proc;
		bind_version = rpcb_next_version[xprt->bind_index].rpc_vers;
571 572
		break;
	case AF_INET6:
573 574
		proc = rpcb_next_version6[xprt->bind_index].rpc_proc;
		bind_version = rpcb_next_version6[xprt->bind_index].rpc_vers;
575 576 577 578
		break;
	default:
		status = -EAFNOSUPPORT;
		dprintk("RPC: %5u %s: bad address family\n",
579
				task->tk_pid, __func__);
580 581
		goto bailout_nofree;
	}
582
	if (proc == NULL) {
583
		xprt->bind_index = 0;
584
		status = -EPFNOSUPPORT;
585
		dprintk("RPC: %5u %s: no more getport versions available\n",
586
			task->tk_pid, __func__);
587 588 589
		goto bailout_nofree;
	}

590
	dprintk("RPC: %5u %s: trying rpcbind version %u\n",
591
		task->tk_pid, __func__, bind_version);
592

593
	rpcb_clnt = rpcb_create(clnt->cl_server, sap, salen, xprt->prot,
594
				bind_version);
595 596 597
	if (IS_ERR(rpcb_clnt)) {
		status = PTR_ERR(rpcb_clnt);
		dprintk("RPC: %5u %s: rpcb_create failed, error %ld\n",
598
			task->tk_pid, __func__, PTR_ERR(rpcb_clnt));
599 600 601
		goto bailout_nofree;
	}

602 603 604
	map = kzalloc(sizeof(struct rpcbind_args), GFP_ATOMIC);
	if (!map) {
		status = -ENOMEM;
605
		dprintk("RPC: %5u %s: no memory available\n",
606
			task->tk_pid, __func__);
607
		goto bailout_release_client;
608 609 610 611 612 613
	}
	map->r_prog = clnt->cl_prog;
	map->r_vers = clnt->cl_vers;
	map->r_prot = xprt->prot;
	map->r_port = 0;
	map->r_xprt = xprt_get(xprt);
614
	map->r_status = -EIO;
615

616 617 618 619 620 621 622 623 624 625 626 627 628 629
	switch (bind_version) {
	case RPCBVERS_4:
	case RPCBVERS_3:
		map->r_netid = rpc_peeraddr2str(clnt, RPC_DISPLAY_NETID);
		map->r_addr = rpc_sockaddr2uaddr(sap);
		map->r_owner = "";
		break;
	case RPCBVERS_2:
		map->r_addr = NULL;
		break;
	default:
		BUG();
	}

630
	child = rpcb_call_async(rpcb_clnt, map, proc);
631
	rpc_release_client(rpcb_clnt);
632
	if (IS_ERR(child)) {
633
		/* rpcb_map_release() has freed the arguments */
634
		dprintk("RPC: %5u %s: rpc_run_task failed\n",
635
			task->tk_pid, __func__);
636
		return;
637 638
	}

639 640
	xprt->stat.bind_count++;
	rpc_put_task(child);
641 642
	return;

643 644
bailout_release_client:
	rpc_release_client(rpcb_clnt);
645 646 647 648
bailout_nofree:
	rpcb_wake_rpcbind_waiters(xprt, status);
	task->tk_status = status;
}
649
EXPORT_SYMBOL_GPL(rpcb_getport_async);
650 651 652 653 654 655 656 657 658 659

/*
 * Rpcbind child task calls this callback via tk_exit.
 */
static void rpcb_getport_done(struct rpc_task *child, void *data)
{
	struct rpcbind_args *map = data;
	struct rpc_xprt *xprt = map->r_xprt;
	int status = child->tk_status;

C
Chuck Lever 已提交
660 661 662 663
	/* Garbage reply: retry with a lesser rpcbind version */
	if (status == -EIO)
		status = -EPROTONOSUPPORT;

664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684
	/* rpcbind server doesn't support this rpcbind protocol version */
	if (status == -EPROTONOSUPPORT)
		xprt->bind_index++;

	if (status < 0) {
		/* rpcbind server not available on remote host? */
		xprt->ops->set_port(xprt, 0);
	} else if (map->r_port == 0) {
		/* Requested RPC service wasn't registered on remote host */
		xprt->ops->set_port(xprt, 0);
		status = -EACCES;
	} else {
		/* Succeeded */
		xprt->ops->set_port(xprt, map->r_port);
		xprt_set_bound(xprt);
		status = 0;
	}

	dprintk("RPC: %5u rpcb_getport_done(status %d, port %u)\n",
			child->tk_pid, status, map->r_port);

685
	map->r_status = status;
686 687
}

688 689 690 691
/*
 * XDR functions for rpcbind
 */

692 693
static void rpcb_enc_mapping(struct rpc_rqst *req, struct xdr_stream *xdr,
			     const struct rpcbind_args *rpcb)
694 695
{
	struct rpc_task *task = req->rq_task;
696
	__be32 *p;
697 698 699 700 701

	dprintk("RPC: %5u encoding PMAP_%s call (%u, %u, %d, %u)\n",
			task->tk_pid, task->tk_msg.rpc_proc->p_name,
			rpcb->r_prog, rpcb->r_vers, rpcb->r_prot, rpcb->r_port);

702
	p = xdr_reserve_space(xdr, RPCB_mappingargs_sz << 2);
703 704 705 706
	*p++ = cpu_to_be32(rpcb->r_prog);
	*p++ = cpu_to_be32(rpcb->r_vers);
	*p++ = cpu_to_be32(rpcb->r_prot);
	*p   = cpu_to_be32(rpcb->r_port);
707 708
}

709 710 711 712 713 714 715 716 717 718 719
static int rpcb_dec_getport(struct rpc_rqst *req, __be32 *p,
			    struct rpcbind_args *rpcb)
{
	struct rpc_task *task = req->rq_task;
	struct xdr_stream xdr;
	unsigned long port;

	xdr_init_decode(&xdr, &req->rq_rcv_buf, p);

	rpcb->r_port = 0;

720
	p = xdr_inline_decode(&xdr, 4);
721 722 723
	if (unlikely(p == NULL))
		return -EIO;

724
	port = be32_to_cpup(p);
725 726
	dprintk("RPC: %5u PMAP_%s result: %lu\n", task->tk_pid,
			task->tk_msg.rpc_proc->p_name, port);
727
	if (unlikely(port > USHRT_MAX))
728 729 730 731 732 733
		return -EIO;

	rpcb->r_port = port;
	return 0;
}

734 735 736 737 738 739 740 741
static int rpcb_dec_set(struct rpc_rqst *req, __be32 *p,
			unsigned int *boolp)
{
	struct rpc_task *task = req->rq_task;
	struct xdr_stream xdr;

	xdr_init_decode(&xdr, &req->rq_rcv_buf, p);

742
	p = xdr_inline_decode(&xdr, 4);
743 744 745 746 747 748 749 750 751 752 753 754 755
	if (unlikely(p == NULL))
		return -EIO;

	*boolp = 0;
	if (*p)
		*boolp = 1;

	dprintk("RPC: %5u RPCB_%s call %s\n",
			task->tk_pid, task->tk_msg.rpc_proc->p_name,
			(*boolp ? "succeeded" : "failed"));
	return 0;
}

756 757
static void encode_rpcb_string(struct xdr_stream *xdr, const char *string,
			       const u32 maxstrlen)
758 759
{
	__be32 *p;
760
	u32 len;
761 762

	len = strlen(string);
763 764
	BUG_ON(len > maxstrlen);
	p = xdr_reserve_space(xdr, 4 + len);
765 766 767
	xdr_encode_opaque(p, string, len);
}

768 769
static void rpcb_enc_getaddr(struct rpc_rqst *req, struct xdr_stream *xdr,
			     const struct rpcbind_args *rpcb)
770 771
{
	struct rpc_task *task = req->rq_task;
772
	__be32 *p;
773 774 775 776 777 778

	dprintk("RPC: %5u encoding RPCB_%s call (%u, %u, '%s', '%s')\n",
			task->tk_pid, task->tk_msg.rpc_proc->p_name,
			rpcb->r_prog, rpcb->r_vers,
			rpcb->r_netid, rpcb->r_addr);

779
	p = xdr_reserve_space(xdr, (RPCB_program_sz + RPCB_version_sz) << 2);
780 781
	*p++ = cpu_to_be32(rpcb->r_prog);
	*p = cpu_to_be32(rpcb->r_vers);
782

783 784 785
	encode_rpcb_string(xdr, rpcb->r_netid, RPCBIND_MAXNETIDLEN);
	encode_rpcb_string(xdr, rpcb->r_addr, RPCBIND_MAXUADDRLEN);
	encode_rpcb_string(xdr, rpcb->r_owner, RPCB_MAXOWNERLEN);
786 787
}

788 789 790 791 792 793 794 795 796 797 798 799 800
static int rpcb_dec_getaddr(struct rpc_rqst *req, __be32 *p,
			    struct rpcbind_args *rpcb)
{
	struct sockaddr_storage address;
	struct sockaddr *sap = (struct sockaddr *)&address;
	struct rpc_task *task = req->rq_task;
	struct xdr_stream xdr;
	u32 len;

	rpcb->r_port = 0;

	xdr_init_decode(&xdr, &req->rq_rcv_buf, p);

801
	p = xdr_inline_decode(&xdr, 4);
802 803
	if (unlikely(p == NULL))
		goto out_fail;
804
	len = be32_to_cpup(p);
805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836

	/*
	 * If the returned universal address is a null string,
	 * the requested RPC service was not registered.
	 */
	if (len == 0) {
		dprintk("RPC: %5u RPCB reply: program not registered\n",
				task->tk_pid);
		return 0;
	}

	if (unlikely(len > RPCBIND_MAXUADDRLEN))
		goto out_fail;

	p = xdr_inline_decode(&xdr, len);
	if (unlikely(p == NULL))
		goto out_fail;
	dprintk("RPC: %5u RPCB_%s reply: %s\n", task->tk_pid,
			task->tk_msg.rpc_proc->p_name, (char *)p);

	if (rpc_uaddr2sockaddr((char *)p, len, sap, sizeof(address)) == 0)
		goto out_fail;
	rpcb->r_port = rpc_get_port(sap);

	return 0;

out_fail:
	dprintk("RPC: %5u malformed RPCB_%s reply\n",
			task->tk_pid, task->tk_msg.rpc_proc->p_name);
	return -EIO;
}

837 838 839 840
/*
 * Not all rpcbind procedures described in RFC 1833 are implemented
 * since the Linux kernel RPC code requires only these.
 */
841

842
static struct rpc_procinfo rpcb_procedures2[] = {
843 844
	[RPCBPROC_SET] = {
		.p_proc		= RPCBPROC_SET,
845
		.p_encode	= (kxdreproc_t)rpcb_enc_mapping,
846 847 848 849 850 851 852 853 854
		.p_decode	= (kxdrproc_t)rpcb_dec_set,
		.p_arglen	= RPCB_mappingargs_sz,
		.p_replen	= RPCB_setres_sz,
		.p_statidx	= RPCBPROC_SET,
		.p_timer	= 0,
		.p_name		= "SET",
	},
	[RPCBPROC_UNSET] = {
		.p_proc		= RPCBPROC_UNSET,
855
		.p_encode	= (kxdreproc_t)rpcb_enc_mapping,
856 857 858 859 860 861 862 863 864
		.p_decode	= (kxdrproc_t)rpcb_dec_set,
		.p_arglen	= RPCB_mappingargs_sz,
		.p_replen	= RPCB_setres_sz,
		.p_statidx	= RPCBPROC_UNSET,
		.p_timer	= 0,
		.p_name		= "UNSET",
	},
	[RPCBPROC_GETPORT] = {
		.p_proc		= RPCBPROC_GETPORT,
865
		.p_encode	= (kxdreproc_t)rpcb_enc_mapping,
866 867 868 869 870 871 872
		.p_decode	= (kxdrproc_t)rpcb_dec_getport,
		.p_arglen	= RPCB_mappingargs_sz,
		.p_replen	= RPCB_getportres_sz,
		.p_statidx	= RPCBPROC_GETPORT,
		.p_timer	= 0,
		.p_name		= "GETPORT",
	},
873 874 875
};

static struct rpc_procinfo rpcb_procedures3[] = {
876 877
	[RPCBPROC_SET] = {
		.p_proc		= RPCBPROC_SET,
878
		.p_encode	= (kxdreproc_t)rpcb_enc_getaddr,
879 880 881 882 883 884 885 886 887
		.p_decode	= (kxdrproc_t)rpcb_dec_set,
		.p_arglen	= RPCB_getaddrargs_sz,
		.p_replen	= RPCB_setres_sz,
		.p_statidx	= RPCBPROC_SET,
		.p_timer	= 0,
		.p_name		= "SET",
	},
	[RPCBPROC_UNSET] = {
		.p_proc		= RPCBPROC_UNSET,
888
		.p_encode	= (kxdreproc_t)rpcb_enc_getaddr,
889 890 891 892 893 894 895 896 897
		.p_decode	= (kxdrproc_t)rpcb_dec_set,
		.p_arglen	= RPCB_getaddrargs_sz,
		.p_replen	= RPCB_setres_sz,
		.p_statidx	= RPCBPROC_UNSET,
		.p_timer	= 0,
		.p_name		= "UNSET",
	},
	[RPCBPROC_GETADDR] = {
		.p_proc		= RPCBPROC_GETADDR,
898
		.p_encode	= (kxdreproc_t)rpcb_enc_getaddr,
899 900 901 902 903 904 905
		.p_decode	= (kxdrproc_t)rpcb_dec_getaddr,
		.p_arglen	= RPCB_getaddrargs_sz,
		.p_replen	= RPCB_getaddrres_sz,
		.p_statidx	= RPCBPROC_GETADDR,
		.p_timer	= 0,
		.p_name		= "GETADDR",
	},
906 907 908
};

static struct rpc_procinfo rpcb_procedures4[] = {
909 910
	[RPCBPROC_SET] = {
		.p_proc		= RPCBPROC_SET,
911
		.p_encode	= (kxdreproc_t)rpcb_enc_getaddr,
912 913 914 915 916 917 918 919 920
		.p_decode	= (kxdrproc_t)rpcb_dec_set,
		.p_arglen	= RPCB_getaddrargs_sz,
		.p_replen	= RPCB_setres_sz,
		.p_statidx	= RPCBPROC_SET,
		.p_timer	= 0,
		.p_name		= "SET",
	},
	[RPCBPROC_UNSET] = {
		.p_proc		= RPCBPROC_UNSET,
921
		.p_encode	= (kxdreproc_t)rpcb_enc_getaddr,
922 923 924 925 926 927 928 929 930
		.p_decode	= (kxdrproc_t)rpcb_dec_set,
		.p_arglen	= RPCB_getaddrargs_sz,
		.p_replen	= RPCB_setres_sz,
		.p_statidx	= RPCBPROC_UNSET,
		.p_timer	= 0,
		.p_name		= "UNSET",
	},
	[RPCBPROC_GETADDR] = {
		.p_proc		= RPCBPROC_GETADDR,
931
		.p_encode	= (kxdreproc_t)rpcb_enc_getaddr,
932 933 934 935 936 937 938
		.p_decode	= (kxdrproc_t)rpcb_dec_getaddr,
		.p_arglen	= RPCB_getaddrargs_sz,
		.p_replen	= RPCB_getaddrres_sz,
		.p_statidx	= RPCBPROC_GETADDR,
		.p_timer	= 0,
		.p_name		= "GETADDR",
	},
939 940 941
};

static struct rpcb_info rpcb_next_version[] = {
942 943 944 945 946 947 948
	{
		.rpc_vers	= RPCBVERS_2,
		.rpc_proc	= &rpcb_procedures2[RPCBPROC_GETPORT],
	},
	{
		.rpc_proc	= NULL,
	},
949 950
};

951
static struct rpcb_info rpcb_next_version6[] = {
952 953
	{
		.rpc_vers	= RPCBVERS_4,
954
		.rpc_proc	= &rpcb_procedures4[RPCBPROC_GETADDR],
955 956 957 958 959 960 961 962
	},
	{
		.rpc_vers	= RPCBVERS_3,
		.rpc_proc	= &rpcb_procedures3[RPCBPROC_GETADDR],
	},
	{
		.rpc_proc	= NULL,
	},
963 964
};

965
static struct rpc_version rpcb_version2 = {
966
	.number		= RPCBVERS_2,
967
	.nrprocs	= ARRAY_SIZE(rpcb_procedures2),
968 969 970 971
	.procs		= rpcb_procedures2
};

static struct rpc_version rpcb_version3 = {
972
	.number		= RPCBVERS_3,
973
	.nrprocs	= ARRAY_SIZE(rpcb_procedures3),
974 975 976 977
	.procs		= rpcb_procedures3
};

static struct rpc_version rpcb_version4 = {
978
	.number		= RPCBVERS_4,
979
	.nrprocs	= ARRAY_SIZE(rpcb_procedures4),
980 981 982 983 984 985 986 987 988 989 990 991 992
	.procs		= rpcb_procedures4
};

static struct rpc_version *rpcb_version[] = {
	NULL,
	NULL,
	&rpcb_version2,
	&rpcb_version3,
	&rpcb_version4
};

static struct rpc_stat rpcb_stats;

993
static struct rpc_program rpcb_program = {
994 995 996 997 998 999
	.name		= "rpcbind",
	.number		= RPCBIND_PROGRAM,
	.nrvers		= ARRAY_SIZE(rpcb_version),
	.version	= rpcb_version,
	.stats		= &rpcb_stats,
};
1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011

/**
 * cleanup_rpcb_clnt - remove xprtsock's sysctls, unregister
 *
 */
void cleanup_rpcb_clnt(void)
{
	if (rpcb_local_clnt4)
		rpc_shutdown_client(rpcb_local_clnt4);
	if (rpcb_local_clnt)
		rpc_shutdown_client(rpcb_local_clnt);
}