rpcb_clnt.c 27.6 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
#include <linux/un.h>
20 21
#include <linux/in.h>
#include <linux/in6.h>
22 23
#include <linux/kernel.h>
#include <linux/errno.h>
24
#include <linux/mutex.h>
25
#include <linux/slab.h>
26
#include <net/ipv6.h>
27 28 29

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

#ifdef RPC_DEBUG
# define RPCDBG_FACILITY	RPCDBG_BIND
#endif

36 37
#define RPCBIND_SOCK_PATHNAME	"/var/run/rpcbind.sock"

38 39 40
#define RPCBIND_PROGRAM		(100000u)
#define RPCBIND_PORT		(111u)

41 42 43 44
#define RPCBVERS_2		(2u)
#define RPCBVERS_3		(3u)
#define RPCBVERS_4		(4u)

45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66
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.
67 68 69 70 71 72 73 74
 *
 * 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.
75
 */
76
#define RPCB_OWNER_STRING	"0"
77 78
#define RPCB_MAXOWNERLEN	sizeof(RPCB_OWNER_STRING)

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 107 108 109
/*
 * 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

110
static void			rpcb_getport_done(struct rpc_task *, void *);
111
static void			rpcb_map_release(void *data);
112
static struct rpc_program	rpcb_program;
113

114 115 116
static struct rpc_clnt *	rpcb_local_clnt;
static struct rpc_clnt *	rpcb_local_clnt4;

117 118 119 120 121 122 123
struct rpcbind_args {
	struct rpc_xprt *	r_xprt;

	u32			r_prog;
	u32			r_vers;
	u32			r_prot;
	unsigned short		r_port;
124 125 126
	const char *		r_netid;
	const char *		r_addr;
	const char *		r_owner;
127 128

	int			r_status;
129 130 131 132
};

static struct rpc_procinfo rpcb_procedures2[];
static struct rpc_procinfo rpcb_procedures3[];
133
static struct rpc_procinfo rpcb_procedures4[];
134

135
struct rpcb_info {
136
	u32			rpc_vers;
137
	struct rpc_procinfo *	rpc_proc;
138 139 140 141
};

static struct rpcb_info rpcb_next_version[];
static struct rpcb_info rpcb_next_version6[];
142 143 144 145 146 147 148 149 150 151 152 153

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

154 155 156 157 158 159
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);
160
	kfree(map->r_addr);
161 162 163
	kfree(map);
}

164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210
/*
 * Returns zero on success, otherwise a negative errno value
 * is returned.
 */
static int rpcb_create_local_unix(void)
{
	static const struct sockaddr_un rpcb_localaddr_rpcbind = {
		.sun_family		= AF_LOCAL,
		.sun_path		= RPCBIND_SOCK_PATHNAME,
	};
	struct rpc_create_args args = {
		.net		= &init_net,
		.protocol	= XPRT_TRANSPORT_LOCAL,
		.address	= (struct sockaddr *)&rpcb_localaddr_rpcbind,
		.addrsize	= sizeof(rpcb_localaddr_rpcbind),
		.servername	= "localhost",
		.program	= &rpcb_program,
		.version	= RPCBVERS_2,
		.authflavor	= RPC_AUTH_NULL,
	};
	struct rpc_clnt *clnt, *clnt4;
	int result = 0;

	/*
	 * Because we requested an RPC PING at transport creation time,
	 * this works only if the user space portmapper is rpcbind, and
	 * it's listening on AF_LOCAL on the named socket.
	 */
	clnt = rpc_create(&args);
	if (IS_ERR(clnt)) {
		dprintk("RPC:       failed to create AF_LOCAL rpcbind "
				"client (errno %ld).\n", PTR_ERR(clnt));
		result = -PTR_ERR(clnt);
		goto out;
	}

	clnt4 = rpc_bind_new_program(clnt, &rpcb_program, RPCBVERS_4);
	if (IS_ERR(clnt4)) {
		dprintk("RPC:       failed to bind second program to "
				"rpcbind v4 client (errno %ld).\n",
				PTR_ERR(clnt4));
		clnt4 = NULL;
	}

	/* Protected by rpcb_create_local_mutex */
	rpcb_local_clnt = clnt;
	rpcb_local_clnt4 = clnt4;
211

212 213 214
out:
	return result;
}
215 216 217 218 219

/*
 * Returns zero on success, otherwise a negative errno value
 * is returned.
 */
220
static int rpcb_create_local_net(void)
221
{
222 223 224 225 226
	static const struct sockaddr_in rpcb_inaddr_loopback = {
		.sin_family		= AF_INET,
		.sin_addr.s_addr	= htonl(INADDR_LOOPBACK),
		.sin_port		= htons(RPCBIND_PORT),
	};
227
	struct rpc_create_args args = {
228
		.net		= &init_net,
229
		.protocol	= XPRT_TRANSPORT_TCP,
230 231
		.address	= (struct sockaddr *)&rpcb_inaddr_loopback,
		.addrsize	= sizeof(rpcb_inaddr_loopback),
232 233
		.servername	= "localhost",
		.program	= &rpcb_program,
234
		.version	= RPCBVERS_2,
235 236 237
		.authflavor	= RPC_AUTH_UNIX,
		.flags		= RPC_CLNT_CREATE_NOPING,
	};
238 239 240 241 242 243 244 245 246 247
	struct rpc_clnt *clnt, *clnt4;
	int result = 0;

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

249 250 251 252 253 254 255
	/*
	 * 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)) {
256 257 258
		dprintk("RPC:       failed to bind second program to "
				"rpcbind v4 client (errno %ld).\n",
				PTR_ERR(clnt4));
259 260 261
		clnt4 = NULL;
	}

262
	/* Protected by rpcb_create_local_mutex */
263 264 265
	rpcb_local_clnt = clnt;
	rpcb_local_clnt4 = clnt4;

266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288
out:
	return result;
}

/*
 * Returns zero on success, otherwise a negative errno value
 * is returned.
 */
static int rpcb_create_local(void)
{
	static DEFINE_MUTEX(rpcb_create_local_mutex);
	int result = 0;

	if (rpcb_local_clnt)
		return result;

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

	if (rpcb_create_local_unix() != 0)
		result = rpcb_create_local_net();

289 290 291
out:
	mutex_unlock(&rpcb_create_local_mutex);
	return result;
292 293
}

294
static struct rpc_clnt *rpcb_create(char *hostname, struct sockaddr *srvaddr,
295
				    size_t salen, int proto, u32 version)
296 297
{
	struct rpc_create_args args = {
298
		.net		= &init_net,
299 300
		.protocol	= proto,
		.address	= srvaddr,
301
		.addrsize	= salen,
302 303 304 305
		.servername	= hostname,
		.program	= &rpcb_program,
		.version	= version,
		.authflavor	= RPC_AUTH_UNIX,
306 307
		.flags		= (RPC_CLNT_CREATE_NOPING |
					RPC_CLNT_CREATE_NONPRIVPORT),
308 309
	};

310 311 312 313 314 315 316 317
	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:
318
		return ERR_PTR(-EAFNOSUPPORT);
319 320
	}

321 322 323
	return rpc_create(&args);
}

324
static int rpcb_register_call(struct rpc_clnt *clnt, struct rpc_message *msg)
325
{
326
	int result, error = 0;
327

328
	msg->rpc_resp = &result;
329

330
	error = rpc_call_sync(clnt, msg, RPC_TASK_SOFTCONN);
331
	if (error < 0) {
332
		dprintk("RPC:       failed to contact local rpcbind "
333
				"server (errno %d).\n", -error);
334 335 336
		return error;
	}

337
	if (!result)
338 339
		return -EACCES;
	return 0;
340 341
}

342 343 344 345
/**
 * 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
346
 * @prot: transport protocol to register
347
 * @port: port value to register
348 349 350 351 352
 *
 * 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).
353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372
 *
 * 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).
373
 */
374
int rpcb_register(u32 prog, u32 vers, int prot, unsigned short port)
375 376 377 378 379 380 381 382 383 384
{
	struct rpcbind_args map = {
		.r_prog		= prog,
		.r_vers		= vers,
		.r_prot		= prot,
		.r_port		= port,
	};
	struct rpc_message msg = {
		.rpc_argp	= &map,
	};
385 386 387 388 389
	int error;

	error = rpcb_create_local();
	if (error)
		return error;
390 391 392 393 394

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

395 396 397
	msg.rpc_proc = &rpcb_procedures2[RPCBPROC_UNSET];
	if (port)
		msg.rpc_proc = &rpcb_procedures2[RPCBPROC_SET];
398

399
	return rpcb_register_call(rpcb_local_clnt, &msg);
400 401
}

402 403 404
/*
 * Fill in AF_INET family-specific arguments to register
 */
405 406
static int rpcb_register_inet4(const struct sockaddr *sap,
			       struct rpc_message *msg)
407
{
408
	const struct sockaddr_in *sin = (const struct sockaddr_in *)sap;
409
	struct rpcbind_args *map = msg->rpc_argp;
410
	unsigned short port = ntohs(sin->sin_port);
411
	int result;
412

413
	map->r_addr = rpc_sockaddr2uaddr(sap);
414 415 416 417 418 419 420 421 422 423

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

424
	result = rpcb_register_call(rpcb_local_clnt4, msg);
425 426
	kfree(map->r_addr);
	return result;
427 428 429 430 431
}

/*
 * Fill in AF_INET6 family-specific arguments to register
 */
432 433
static int rpcb_register_inet6(const struct sockaddr *sap,
			       struct rpc_message *msg)
434
{
435
	const struct sockaddr_in6 *sin6 = (const struct sockaddr_in6 *)sap;
436
	struct rpcbind_args *map = msg->rpc_argp;
437
	unsigned short port = ntohs(sin6->sin6_port);
438
	int result;
439

440
	map->r_addr = rpc_sockaddr2uaddr(sap);
441 442 443 444 445 446 447 448 449 450

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

451
	result = rpcb_register_call(rpcb_local_clnt4, msg);
452 453
	kfree(map->r_addr);
	return result;
454 455
}

456 457 458 459 460 461 462 463 464 465 466
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];

467
	return rpcb_register_call(rpcb_local_clnt4, msg);
468 469
}

470 471 472 473 474 475
/**
 * 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
476 477 478 479 480
 *
 * 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).
481 482 483 484 485 486
 *
 * 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.
 *
487 488 489 490 491
 * 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.
492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513
 *
 * 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,
514
		     const struct sockaddr *address, const char *netid)
515 516 517 518 519 520 521 522 523 524
{
	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,
	};
525 526 527 528 529 530 531
	int error;

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

533 534 535
	if (address == NULL)
		return rpcb_unregister_all_protofamilies(&msg);

536 537
	switch (address->sa_family) {
	case AF_INET:
538
		return rpcb_register_inet4(address, &msg);
539
	case AF_INET6:
540
		return rpcb_register_inet6(address, &msg);
541 542 543 544 545
	}

	return -EAFNOSUPPORT;
}

546
static struct rpc_task *rpcb_call_async(struct rpc_clnt *rpcb_clnt, struct rpcbind_args *map, struct rpc_procinfo *proc)
547 548
{
	struct rpc_message msg = {
549
		.rpc_proc = proc,
550
		.rpc_argp = map,
551
		.rpc_resp = map,
552 553 554 555 556 557
	};
	struct rpc_task_setup task_setup_data = {
		.rpc_client = rpcb_clnt,
		.rpc_message = &msg,
		.callback_ops = &rpcb_getport_ops,
		.callback_data = map,
558
		.flags = RPC_TASK_ASYNC | RPC_TASK_SOFTCONN,
559 560 561 562 563
	};

	return rpc_run_task(&task_setup_data);
}

564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585
/*
 * 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;
}

586
/**
587
 * rpcb_getport_async - obtain the port for a given RPC service on a given host
588 589 590 591 592
 * @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.
 */
593
void rpcb_getport_async(struct rpc_task *task)
594
{
595
	struct rpc_clnt *clnt;
596
	struct rpc_procinfo *proc;
597
	u32 bind_version;
598
	struct rpc_xprt *xprt;
599 600 601
	struct rpc_clnt	*rpcb_clnt;
	static struct rpcbind_args *map;
	struct rpc_task	*child;
602 603 604
	struct sockaddr_storage addr;
	struct sockaddr *sap = (struct sockaddr *)&addr;
	size_t salen;
605 606
	int status;

607 608 609
	clnt = rpcb_find_transport_owner(task->tk_client);
	xprt = clnt->cl_xprt;

610
	dprintk("RPC: %5u %s(%s, %u, %u, %d)\n",
611
		task->tk_pid, __func__,
612
		clnt->cl_server, clnt->cl_prog, clnt->cl_vers, xprt->prot);
613

614 615 616 617
	/* 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);

618
	if (xprt_test_and_set_binding(xprt)) {
619
		dprintk("RPC: %5u %s: waiting for another binder\n",
620
			task->tk_pid, __func__);
621
		return;
622 623 624 625 626
	}

	/* Someone else may have bound if we slept */
	if (xprt_bound(xprt)) {
		status = 0;
627
		dprintk("RPC: %5u %s: already bound\n",
628
			task->tk_pid, __func__);
629 630 631
		goto bailout_nofree;
	}

632
	/* Parent transport's destination address */
633
	salen = rpc_peeraddr(clnt, sap, sizeof(addr));
634 635

	/* Don't ever use rpcbind v2 for AF_INET6 requests */
636
	switch (sap->sa_family) {
637
	case AF_INET:
638 639
		proc = rpcb_next_version[xprt->bind_index].rpc_proc;
		bind_version = rpcb_next_version[xprt->bind_index].rpc_vers;
640 641
		break;
	case AF_INET6:
642 643
		proc = rpcb_next_version6[xprt->bind_index].rpc_proc;
		bind_version = rpcb_next_version6[xprt->bind_index].rpc_vers;
644 645 646 647
		break;
	default:
		status = -EAFNOSUPPORT;
		dprintk("RPC: %5u %s: bad address family\n",
648
				task->tk_pid, __func__);
649 650
		goto bailout_nofree;
	}
651
	if (proc == NULL) {
652
		xprt->bind_index = 0;
653
		status = -EPFNOSUPPORT;
654
		dprintk("RPC: %5u %s: no more getport versions available\n",
655
			task->tk_pid, __func__);
656 657 658
		goto bailout_nofree;
	}

659
	dprintk("RPC: %5u %s: trying rpcbind version %u\n",
660
		task->tk_pid, __func__, bind_version);
661

662
	rpcb_clnt = rpcb_create(clnt->cl_server, sap, salen, xprt->prot,
663
				bind_version);
664 665 666
	if (IS_ERR(rpcb_clnt)) {
		status = PTR_ERR(rpcb_clnt);
		dprintk("RPC: %5u %s: rpcb_create failed, error %ld\n",
667
			task->tk_pid, __func__, PTR_ERR(rpcb_clnt));
668 669 670
		goto bailout_nofree;
	}

671 672 673
	map = kzalloc(sizeof(struct rpcbind_args), GFP_ATOMIC);
	if (!map) {
		status = -ENOMEM;
674
		dprintk("RPC: %5u %s: no memory available\n",
675
			task->tk_pid, __func__);
676
		goto bailout_release_client;
677 678 679 680 681 682
	}
	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);
683
	map->r_status = -EIO;
684

685 686 687 688 689 690 691 692 693 694 695 696 697 698
	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();
	}

699
	child = rpcb_call_async(rpcb_clnt, map, proc);
700
	rpc_release_client(rpcb_clnt);
701
	if (IS_ERR(child)) {
702
		/* rpcb_map_release() has freed the arguments */
703
		dprintk("RPC: %5u %s: rpc_run_task failed\n",
704
			task->tk_pid, __func__);
705
		return;
706 707
	}

708 709
	xprt->stat.bind_count++;
	rpc_put_task(child);
710 711
	return;

712 713
bailout_release_client:
	rpc_release_client(rpcb_clnt);
714 715 716 717
bailout_nofree:
	rpcb_wake_rpcbind_waiters(xprt, status);
	task->tk_status = status;
}
718
EXPORT_SYMBOL_GPL(rpcb_getport_async);
719 720 721 722 723 724 725 726 727 728

/*
 * 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 已提交
729 730 731 732
	/* Garbage reply: retry with a lesser rpcbind version */
	if (status == -EIO)
		status = -EPROTONOSUPPORT;

733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753
	/* 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);

754
	map->r_status = status;
755 756
}

757 758 759 760
/*
 * XDR functions for rpcbind
 */

761 762
static void rpcb_enc_mapping(struct rpc_rqst *req, struct xdr_stream *xdr,
			     const struct rpcbind_args *rpcb)
763 764
{
	struct rpc_task *task = req->rq_task;
765
	__be32 *p;
766 767 768 769 770

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

771
	p = xdr_reserve_space(xdr, RPCB_mappingargs_sz << 2);
772 773 774 775
	*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);
776 777
}

778
static int rpcb_dec_getport(struct rpc_rqst *req, struct xdr_stream *xdr,
779 780 781 782
			    struct rpcbind_args *rpcb)
{
	struct rpc_task *task = req->rq_task;
	unsigned long port;
783
	__be32 *p;
784 785 786

	rpcb->r_port = 0;

787
	p = xdr_inline_decode(xdr, 4);
788 789 790
	if (unlikely(p == NULL))
		return -EIO;

791
	port = be32_to_cpup(p);
792 793
	dprintk("RPC: %5u PMAP_%s result: %lu\n", task->tk_pid,
			task->tk_msg.rpc_proc->p_name, port);
794
	if (unlikely(port > USHRT_MAX))
795 796 797 798 799 800
		return -EIO;

	rpcb->r_port = port;
	return 0;
}

801
static int rpcb_dec_set(struct rpc_rqst *req, struct xdr_stream *xdr,
802 803 804
			unsigned int *boolp)
{
	struct rpc_task *task = req->rq_task;
805
	__be32 *p;
806

807
	p = xdr_inline_decode(xdr, 4);
808 809 810 811
	if (unlikely(p == NULL))
		return -EIO;

	*boolp = 0;
812
	if (*p != xdr_zero)
813 814 815 816 817 818 819 820
		*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;
}

821 822
static void encode_rpcb_string(struct xdr_stream *xdr, const char *string,
			       const u32 maxstrlen)
823 824
{
	__be32 *p;
825
	u32 len;
826 827

	len = strlen(string);
828 829
	BUG_ON(len > maxstrlen);
	p = xdr_reserve_space(xdr, 4 + len);
830 831 832
	xdr_encode_opaque(p, string, len);
}

833 834
static void rpcb_enc_getaddr(struct rpc_rqst *req, struct xdr_stream *xdr,
			     const struct rpcbind_args *rpcb)
835 836
{
	struct rpc_task *task = req->rq_task;
837
	__be32 *p;
838 839 840 841 842 843

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

844
	p = xdr_reserve_space(xdr, (RPCB_program_sz + RPCB_version_sz) << 2);
845 846
	*p++ = cpu_to_be32(rpcb->r_prog);
	*p = cpu_to_be32(rpcb->r_vers);
847

848 849 850
	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);
851 852
}

853
static int rpcb_dec_getaddr(struct rpc_rqst *req, struct xdr_stream *xdr,
854 855 856 857 858
			    struct rpcbind_args *rpcb)
{
	struct sockaddr_storage address;
	struct sockaddr *sap = (struct sockaddr *)&address;
	struct rpc_task *task = req->rq_task;
859
	__be32 *p;
860 861 862 863
	u32 len;

	rpcb->r_port = 0;

864
	p = xdr_inline_decode(xdr, 4);
865 866
	if (unlikely(p == NULL))
		goto out_fail;
867
	len = be32_to_cpup(p);
868 869 870 871 872 873 874 875 876 877 878 879 880 881

	/*
	 * 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;

882
	p = xdr_inline_decode(xdr, len);
883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899
	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;
}

900 901 902 903
/*
 * Not all rpcbind procedures described in RFC 1833 are implemented
 * since the Linux kernel RPC code requires only these.
 */
904

905
static struct rpc_procinfo rpcb_procedures2[] = {
906 907
	[RPCBPROC_SET] = {
		.p_proc		= RPCBPROC_SET,
908
		.p_encode	= (kxdreproc_t)rpcb_enc_mapping,
909
		.p_decode	= (kxdrdproc_t)rpcb_dec_set,
910 911 912 913 914 915 916 917
		.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,
918
		.p_encode	= (kxdreproc_t)rpcb_enc_mapping,
919
		.p_decode	= (kxdrdproc_t)rpcb_dec_set,
920 921 922 923 924 925 926 927
		.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,
928
		.p_encode	= (kxdreproc_t)rpcb_enc_mapping,
929
		.p_decode	= (kxdrdproc_t)rpcb_dec_getport,
930 931 932 933 934 935
		.p_arglen	= RPCB_mappingargs_sz,
		.p_replen	= RPCB_getportres_sz,
		.p_statidx	= RPCBPROC_GETPORT,
		.p_timer	= 0,
		.p_name		= "GETPORT",
	},
936 937 938
};

static struct rpc_procinfo rpcb_procedures3[] = {
939 940
	[RPCBPROC_SET] = {
		.p_proc		= RPCBPROC_SET,
941
		.p_encode	= (kxdreproc_t)rpcb_enc_getaddr,
942
		.p_decode	= (kxdrdproc_t)rpcb_dec_set,
943 944 945 946 947 948 949 950
		.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,
951
		.p_encode	= (kxdreproc_t)rpcb_enc_getaddr,
952
		.p_decode	= (kxdrdproc_t)rpcb_dec_set,
953 954 955 956 957 958 959 960
		.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,
961
		.p_encode	= (kxdreproc_t)rpcb_enc_getaddr,
962
		.p_decode	= (kxdrdproc_t)rpcb_dec_getaddr,
963 964 965 966 967 968
		.p_arglen	= RPCB_getaddrargs_sz,
		.p_replen	= RPCB_getaddrres_sz,
		.p_statidx	= RPCBPROC_GETADDR,
		.p_timer	= 0,
		.p_name		= "GETADDR",
	},
969 970 971
};

static struct rpc_procinfo rpcb_procedures4[] = {
972 973
	[RPCBPROC_SET] = {
		.p_proc		= RPCBPROC_SET,
974
		.p_encode	= (kxdreproc_t)rpcb_enc_getaddr,
975
		.p_decode	= (kxdrdproc_t)rpcb_dec_set,
976 977 978 979 980 981 982 983
		.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,
984
		.p_encode	= (kxdreproc_t)rpcb_enc_getaddr,
985
		.p_decode	= (kxdrdproc_t)rpcb_dec_set,
986 987 988 989 990 991 992 993
		.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,
994
		.p_encode	= (kxdreproc_t)rpcb_enc_getaddr,
995
		.p_decode	= (kxdrdproc_t)rpcb_dec_getaddr,
996 997 998 999 1000 1001
		.p_arglen	= RPCB_getaddrargs_sz,
		.p_replen	= RPCB_getaddrres_sz,
		.p_statidx	= RPCBPROC_GETADDR,
		.p_timer	= 0,
		.p_name		= "GETADDR",
	},
1002 1003 1004
};

static struct rpcb_info rpcb_next_version[] = {
1005 1006 1007 1008 1009 1010 1011
	{
		.rpc_vers	= RPCBVERS_2,
		.rpc_proc	= &rpcb_procedures2[RPCBPROC_GETPORT],
	},
	{
		.rpc_proc	= NULL,
	},
1012 1013
};

1014
static struct rpcb_info rpcb_next_version6[] = {
1015 1016
	{
		.rpc_vers	= RPCBVERS_4,
1017
		.rpc_proc	= &rpcb_procedures4[RPCBPROC_GETADDR],
1018 1019 1020 1021 1022 1023 1024 1025
	},
	{
		.rpc_vers	= RPCBVERS_3,
		.rpc_proc	= &rpcb_procedures3[RPCBPROC_GETADDR],
	},
	{
		.rpc_proc	= NULL,
	},
1026 1027
};

1028
static struct rpc_version rpcb_version2 = {
1029
	.number		= RPCBVERS_2,
1030
	.nrprocs	= ARRAY_SIZE(rpcb_procedures2),
1031 1032 1033 1034
	.procs		= rpcb_procedures2
};

static struct rpc_version rpcb_version3 = {
1035
	.number		= RPCBVERS_3,
1036
	.nrprocs	= ARRAY_SIZE(rpcb_procedures3),
1037 1038 1039 1040
	.procs		= rpcb_procedures3
};

static struct rpc_version rpcb_version4 = {
1041
	.number		= RPCBVERS_4,
1042
	.nrprocs	= ARRAY_SIZE(rpcb_procedures4),
1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055
	.procs		= rpcb_procedures4
};

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

static struct rpc_stat rpcb_stats;

1056
static struct rpc_program rpcb_program = {
1057 1058 1059 1060 1061 1062
	.name		= "rpcbind",
	.number		= RPCBIND_PROGRAM,
	.nrvers		= ARRAY_SIZE(rpcb_version),
	.version	= rpcb_version,
	.stats		= &rpcb_stats,
};
1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074

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