[deliverable/linux.git] / net / sunrpc / pmap_clnt.c

/*
 * linux/net/sunrpc/pmap.c
 *
 * Portmapper client.
 *
 * Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de>
 */

#include <linux/config.h>
#include <linux/types.h>
#include <linux/socket.h>
#include <linux/kernel.h>
#include <linux/errno.h>
#include <linux/uio.h>
#include <linux/in.h>
#include <linux/sunrpc/clnt.h>
#include <linux/sunrpc/xprt.h>
#include <linux/sunrpc/sched.h>

#ifdef RPC_DEBUG
# define RPCDBG_FACILITY	RPCDBG_PMAP
#endif

#define PMAP_SET		1
#define PMAP_UNSET		2
#define PMAP_GETPORT		3

static struct rpc_procinfo	pmap_procedures[];
static struct rpc_clnt *	pmap_create(char *, struct sockaddr_in *, int, int);
static void			pmap_getport_done(struct rpc_task *);
static struct rpc_program	pmap_program;
static DEFINE_SPINLOCK(pmap_lock);

/*
 * Obtain the port for a given RPC service on a given host. This one can
 * be called for an ongoing RPC request.
 */
void
rpc_getport(struct rpc_task *task, struct rpc_clnt *clnt)
{
	struct rpc_portmap *map = clnt->cl_pmap;
	struct sockaddr_in *sap = &clnt->cl_xprt->addr;
	struct rpc_message msg = {
		.rpc_proc	= &pmap_procedures[PMAP_GETPORT],
		.rpc_argp	= map,
		.rpc_resp	= &clnt->cl_port,
		.rpc_cred	= NULL
	};
	struct rpc_clnt	*pmap_clnt;
	struct rpc_task	*child;

	dprintk("RPC: %4d rpc_getport(%s, %d, %d, %d)\n",
			task->tk_pid, clnt->cl_server,
			map->pm_prog, map->pm_vers, map->pm_prot);

	/* Autobind on cloned rpc clients is discouraged */
	BUG_ON(clnt->cl_parent != clnt);

	spin_lock(&pmap_lock);
	if (map->pm_binding) {
		rpc_sleep_on(&map->pm_bindwait, task, NULL, NULL);
		spin_unlock(&pmap_lock);
		return;
	}
	map->pm_binding = 1;
	spin_unlock(&pmap_lock);

	pmap_clnt = pmap_create(clnt->cl_server, sap, map->pm_prot, 0);
	if (IS_ERR(pmap_clnt)) {
		task->tk_status = PTR_ERR(pmap_clnt);
		goto bailout;
	}
	task->tk_status = 0;

	/*
	 * Note: rpc_new_child will release client after a failure.
	 */
	if (!(child = rpc_new_child(pmap_clnt, task)))
		goto bailout;

	/* Setup the call info struct */
	rpc_call_setup(child, &msg, 0);

	/* ... and run the child task */
	rpc_run_child(task, child, pmap_getport_done);
	return;

bailout:
	spin_lock(&pmap_lock);
	map->pm_binding = 0;
	rpc_wake_up(&map->pm_bindwait);
	spin_unlock(&pmap_lock);
	task->tk_status = -EIO;
	task->tk_action = NULL;
}

#ifdef CONFIG_ROOT_NFS
int
rpc_getport_external(struct sockaddr_in *sin, __u32 prog, __u32 vers, int prot)
{
	struct rpc_portmap map = {
		.pm_prog	= prog,
		.pm_vers	= vers,
		.pm_prot	= prot,
		.pm_port	= 0
	};
	struct rpc_clnt	*pmap_clnt;
	char		hostname[32];
	int		status;

	dprintk("RPC:      rpc_getport_external(%u.%u.%u.%u, %d, %d, %d)\n",
			NIPQUAD(sin->sin_addr.s_addr), prog, vers, prot);

	sprintf(hostname, "%u.%u.%u.%u", NIPQUAD(sin->sin_addr.s_addr));
	pmap_clnt = pmap_create(hostname, sin, prot, 0);
	if (IS_ERR(pmap_clnt))
		return PTR_ERR(pmap_clnt);

	/* Setup the call info struct */
	status = rpc_call(pmap_clnt, PMAP_GETPORT, &map, &map.pm_port, 0);

	if (status >= 0) {
		if (map.pm_port != 0)
			return map.pm_port;
		status = -EACCES;
	}
	return status;
}
#endif

static void
pmap_getport_done(struct rpc_task *task)
{
	struct rpc_clnt	*clnt = task->tk_client;
	struct rpc_portmap *map = clnt->cl_pmap;

	dprintk("RPC: %4d pmap_getport_done(status %d, port %d)\n",
			task->tk_pid, task->tk_status, clnt->cl_port);
	if (task->tk_status < 0) {
		/* Make the calling task exit with an error */
		task->tk_action = NULL;
	} else if (clnt->cl_port == 0) {
		/* Program not registered */
		task->tk_status = -EACCES;
		task->tk_action = NULL;
	} else {
		/* byte-swap port number first */
		clnt->cl_port = htons(clnt->cl_port);
		clnt->cl_xprt->addr.sin_port = clnt->cl_port;
	}
	spin_lock(&pmap_lock);
	map->pm_binding = 0;
	rpc_wake_up(&map->pm_bindwait);
	spin_unlock(&pmap_lock);
}

/*
 * Set or unset a port registration with the local portmapper.
 * port == 0 means unregister, port != 0 means register.
 */
int
rpc_register(u32 prog, u32 vers, int prot, unsigned short port, int *okay)
{
	struct sockaddr_in	sin;
	struct rpc_portmap	map;
	struct rpc_clnt		*pmap_clnt;
	int error = 0;

	dprintk("RPC: registering (%d, %d, %d, %d) with portmapper.\n",
			prog, vers, prot, port);

	sin.sin_family = AF_INET;
	sin.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
	pmap_clnt = pmap_create("localhost", &sin, IPPROTO_UDP, 1);
	if (IS_ERR(pmap_clnt)) {
		error = PTR_ERR(pmap_clnt);
		dprintk("RPC: couldn't create pmap client. Error = %d\n", error);
		return error;
	}

	map.pm_prog = prog;
	map.pm_vers = vers;
	map.pm_prot = prot;
	map.pm_port = port;

	error = rpc_call(pmap_clnt, port? PMAP_SET : PMAP_UNSET,
					&map, okay, 0);

	if (error < 0) {
		printk(KERN_WARNING
			"RPC: failed to contact portmap (errno %d).\n",
			error);
	}
	dprintk("RPC: registration status %d/%d\n", error, *okay);

	/* Client deleted automatically because cl_oneshot == 1 */
	return error;
}

static struct rpc_clnt *
pmap_create(char *hostname, struct sockaddr_in *srvaddr, int proto, int privileged)
{
	struct rpc_xprt	*xprt;
	struct rpc_clnt	*clnt;

	/* printk("pmap: create xprt\n"); */
	xprt = xprt_create_proto(proto, srvaddr, NULL);
	if (IS_ERR(xprt))
		return (struct rpc_clnt *)xprt;
	xprt->addr.sin_port = htons(RPC_PMAP_PORT);
	if (!privileged)
		xprt->resvport = 0;

	/* printk("pmap: create clnt\n"); */
	clnt = rpc_new_client(xprt, hostname,
				&pmap_program, RPC_PMAP_VERSION,
				RPC_AUTH_UNIX);
	if (!IS_ERR(clnt)) {
		clnt->cl_softrtry = 1;
		clnt->cl_chatty   = 1;
		clnt->cl_oneshot  = 1;
	}
	return clnt;
}

/*
 * XDR encode/decode functions for PMAP
 */
static int
xdr_encode_mapping(struct rpc_rqst *req, u32 *p, struct rpc_portmap *map)
{
	dprintk("RPC: xdr_encode_mapping(%d, %d, %d, %d)\n",
		map->pm_prog, map->pm_vers, map->pm_prot, map->pm_port);
	*p++ = htonl(map->pm_prog);
	*p++ = htonl(map->pm_vers);
	*p++ = htonl(map->pm_prot);
	*p++ = htonl(map->pm_port);

	req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
	return 0;
}

static int
xdr_decode_port(struct rpc_rqst *req, u32 *p, unsigned short *portp)
{
	*portp = (unsigned short) ntohl(*p++);
	return 0;
}

static int
xdr_decode_bool(struct rpc_rqst *req, u32 *p, unsigned int *boolp)
{
	*boolp = (unsigned int) ntohl(*p++);
	return 0;
}

static struct rpc_procinfo	pmap_procedures[] = {
[PMAP_SET] = {
	  .p_proc		= PMAP_SET,
	  .p_encode		= (kxdrproc_t) xdr_encode_mapping,	
	  .p_decode		= (kxdrproc_t) xdr_decode_bool,
	  .p_bufsiz		= 4,
	  .p_count		= 1,
	},
[PMAP_UNSET] = {
	  .p_proc		= PMAP_UNSET,
	  .p_encode		= (kxdrproc_t) xdr_encode_mapping,	
	  .p_decode		= (kxdrproc_t) xdr_decode_bool,
	  .p_bufsiz		= 4,
	  .p_count		= 1,
	},
[PMAP_GETPORT] = {
	  .p_proc		= PMAP_GETPORT,
	  .p_encode		= (kxdrproc_t) xdr_encode_mapping,
	  .p_decode		= (kxdrproc_t) xdr_decode_port,
	  .p_bufsiz		= 4,
	  .p_count		= 1,
	},
};

static struct rpc_version	pmap_version2 = {
	.number		= 2,
	.nrprocs	= 4,
	.procs		= pmap_procedures
};

static struct rpc_version *	pmap_version[] = {
	NULL,
	NULL,
	&pmap_version2
};

static struct rpc_stat		pmap_stats;

static struct rpc_program	pmap_program = {
	.name		= "portmap",
	.number		= RPC_PMAP_PROGRAM,
	.nrvers		= ARRAY_SIZE(pmap_version),
	.version	= pmap_version,
	.stats		= &pmap_stats,
};
Commit	Line	Data
1da177e4 LT	1	/*
	2	* linux/net/sunrpc/pmap.c
	3	*
	4	* Portmapper client.
	5	*
	6	* Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de>
	7	*/
	8
	9	#include <linux/config.h>
	10	#include <linux/types.h>
	11	#include <linux/socket.h>
	12	#include <linux/kernel.h>
	13	#include <linux/errno.h>
	14	#include <linux/uio.h>
	15	#include <linux/in.h>
	16	#include <linux/sunrpc/clnt.h>
	17	#include <linux/sunrpc/xprt.h>
	18	#include <linux/sunrpc/sched.h>
	19
	20	#ifdef RPC_DEBUG
	21	# define RPCDBG_FACILITY RPCDBG_PMAP
	22	#endif
	23
	24	#define PMAP_SET 1
	25	#define PMAP_UNSET 2
	26	#define PMAP_GETPORT 3
	27
	28	static struct rpc_procinfo pmap_procedures[];
6cd7525a	29	static struct rpc_clnt * pmap_create(char , struct sockaddr_in , int, int);
1da177e4 LT	30	static void pmap_getport_done(struct rpc_task *);
	31	static struct rpc_program pmap_program;
	32	static DEFINE_SPINLOCK(pmap_lock);
	33
	34	/*
	35	* Obtain the port for a given RPC service on a given host. This one can
	36	* be called for an ongoing RPC request.
	37	*/
	38	void
	39	rpc_getport(struct rpc_task task, struct rpc_clnt clnt)
	40	{
	41	struct rpc_portmap *map = clnt->cl_pmap;
	42	struct sockaddr_in *sap = &clnt->cl_xprt->addr;
	43	struct rpc_message msg = {
	44	.rpc_proc = &pmap_procedures[PMAP_GETPORT],
	45	.rpc_argp = map,
	46	.rpc_resp = &clnt->cl_port,
	47	.rpc_cred = NULL
	48	};
	49	struct rpc_clnt *pmap_clnt;
	50	struct rpc_task *child;
	51
	52	dprintk("RPC: %4d rpc_getport(%s, %d, %d, %d)\n",
	53	task->tk_pid, clnt->cl_server,
	54	map->pm_prog, map->pm_vers, map->pm_prot);
	55
007e251f AG	56	/* Autobind on cloned rpc clients is discouraged */
	57	BUG_ON(clnt->cl_parent != clnt);
	58
1da177e4 LT	59	spin_lock(&pmap_lock);
	60	if (map->pm_binding) {
	61	rpc_sleep_on(&map->pm_bindwait, task, NULL, NULL);
	62	spin_unlock(&pmap_lock);
	63	return;
	64	}
	65	map->pm_binding = 1;
	66	spin_unlock(&pmap_lock);
	67
6cd7525a	68	pmap_clnt = pmap_create(clnt->cl_server, sap, map->pm_prot, 0);
1da177e4 LT	69	if (IS_ERR(pmap_clnt)) {
	70	task->tk_status = PTR_ERR(pmap_clnt);
	71	goto bailout;
	72	}
	73	task->tk_status = 0;
	74
	75	/*
	76	* Note: rpc_new_child will release client after a failure.
	77	*/
	78	if (!(child = rpc_new_child(pmap_clnt, task)))
	79	goto bailout;
	80
	81	/* Setup the call info struct */
	82	rpc_call_setup(child, &msg, 0);
	83
	84	/* ... and run the child task */
	85	rpc_run_child(task, child, pmap_getport_done);
	86	return;
	87
	88	bailout:
	89	spin_lock(&pmap_lock);
	90	map->pm_binding = 0;
	91	rpc_wake_up(&map->pm_bindwait);
	92	spin_unlock(&pmap_lock);
	93	task->tk_status = -EIO;
	94	task->tk_action = NULL;
	95	}
	96
	97	#ifdef CONFIG_ROOT_NFS
	98	int
	99	rpc_getport_external(struct sockaddr_in *sin, __u32 prog, __u32 vers, int prot)
	100	{
	101	struct rpc_portmap map = {
	102	.pm_prog = prog,
	103	.pm_vers = vers,
	104	.pm_prot = prot,
	105	.pm_port = 0
	106	};
	107	struct rpc_clnt *pmap_clnt;
	108	char hostname[32];
	109	int status;
	110
	111	dprintk("RPC: rpc_getport_external(%u.%u.%u.%u, %d, %d, %d)\n",
	112	NIPQUAD(sin->sin_addr.s_addr), prog, vers, prot);
	113
	114	sprintf(hostname, "%u.%u.%u.%u", NIPQUAD(sin->sin_addr.s_addr));
6cd7525a	115	pmap_clnt = pmap_create(hostname, sin, prot, 0);
1da177e4 LT	116	if (IS_ERR(pmap_clnt))
	117	return PTR_ERR(pmap_clnt);
	118
	119	/* Setup the call info struct */
	120	status = rpc_call(pmap_clnt, PMAP_GETPORT, &map, &map.pm_port, 0);
	121
	122	if (status >= 0) {
	123	if (map.pm_port != 0)
	124	return map.pm_port;
	125	status = -EACCES;
	126	}
	127	return status;
	128	}
	129	#endif
	130
	131	static void
	132	pmap_getport_done(struct rpc_task *task)
	133	{
	134	struct rpc_clnt *clnt = task->tk_client;
	135	struct rpc_portmap *map = clnt->cl_pmap;
	136
	137	dprintk("RPC: %4d pmap_getport_done(status %d, port %d)\n",
	138	task->tk_pid, task->tk_status, clnt->cl_port);
	139	if (task->tk_status < 0) {
	140	/* Make the calling task exit with an error */
	141	task->tk_action = NULL;
	142	} else if (clnt->cl_port == 0) {
	143	/* Program not registered */
	144	task->tk_status = -EACCES;
	145	task->tk_action = NULL;
	146	} else {
	147	/* byte-swap port number first */
	148	clnt->cl_port = htons(clnt->cl_port);
	149	clnt->cl_xprt->addr.sin_port = clnt->cl_port;
	150	}
	151	spin_lock(&pmap_lock);
	152	map->pm_binding = 0;
	153	rpc_wake_up(&map->pm_bindwait);
	154	spin_unlock(&pmap_lock);
	155	}
	156
	157	/*
	158	* Set or unset a port registration with the local portmapper.
	159	* port == 0 means unregister, port != 0 means register.
	160	*/
	161	int
	162	rpc_register(u32 prog, u32 vers, int prot, unsigned short port, int *okay)
	163	{
	164	struct sockaddr_in sin;
	165	struct rpc_portmap map;
	166	struct rpc_clnt *pmap_clnt;
	167	int error = 0;
	168
	169	dprintk("RPC: registering (%d, %d, %d, %d) with portmapper.\n",
	170	prog, vers, prot, port);
	171
	172	sin.sin_family = AF_INET;
	173	sin.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
6cd7525a	174	pmap_clnt = pmap_create("localhost", &sin, IPPROTO_UDP, 1);
1da177e4 LT	175	if (IS_ERR(pmap_clnt)) {
	176	error = PTR_ERR(pmap_clnt);
	177	dprintk("RPC: couldn't create pmap client. Error = %d\n", error);
	178	return error;
	179	}
	180
	181	map.pm_prog = prog;
	182	map.pm_vers = vers;
	183	map.pm_prot = prot;
	184	map.pm_port = port;
	185
	186	error = rpc_call(pmap_clnt, port? PMAP_SET : PMAP_UNSET,
	187	&map, okay, 0);
	188
	189	if (error < 0) {
	190	printk(KERN_WARNING
	191	"RPC: failed to contact portmap (errno %d).\n",
	192	error);
	193	}
	194	dprintk("RPC: registration status %d/%d\n", error, *okay);
	195
	196	/* Client deleted automatically because cl_oneshot == 1 */
	197	return error;
	198	}
	199
	200	static struct rpc_clnt *
6cd7525a	201	pmap_create(char hostname, struct sockaddr_in srvaddr, int proto, int privileged)
1da177e4 LT	202	{
	203	struct rpc_xprt *xprt;
	204	struct rpc_clnt *clnt;
	205
	206	/* printk("pmap: create xprt\n"); */
	207	xprt = xprt_create_proto(proto, srvaddr, NULL);
	208	if (IS_ERR(xprt))
	209	return (struct rpc_clnt *)xprt;
	210	xprt->addr.sin_port = htons(RPC_PMAP_PORT);
6cd7525a CL	211	if (!privileged)
6cd7525a CL	212	xprt->resvport = 0;
1da177e4 LT	213
1da177e4 LT	214	/* printk("pmap: create clnt\n"); */
5ee0ed7d	215	clnt = rpc_new_client(xprt, hostname,
1da177e4 LT	216	&pmap_program, RPC_PMAP_VERSION,
1da177e4 LT	217	RPC_AUTH_UNIX);
5b616f5d	218	if (!IS_ERR(clnt)) {
1da177e4 LT	219	clnt->cl_softrtry = 1;
	220	clnt->cl_chatty = 1;
	221	clnt->cl_oneshot = 1;
	222	}
	223	return clnt;
	224	}
	225
	226	/*
	227	* XDR encode/decode functions for PMAP
	228	*/
	229	static int
	230	xdr_encode_mapping(struct rpc_rqst req, u32 p, struct rpc_portmap *map)
	231	{
	232	dprintk("RPC: xdr_encode_mapping(%d, %d, %d, %d)\n",
	233	map->pm_prog, map->pm_vers, map->pm_prot, map->pm_port);
	234	*p++ = htonl(map->pm_prog);
	235	*p++ = htonl(map->pm_vers);
	236	*p++ = htonl(map->pm_prot);
	237	*p++ = htonl(map->pm_port);
	238
	239	req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
	240	return 0;
	241	}
	242
	243	static int
	244	xdr_decode_port(struct rpc_rqst req, u32 p, unsigned short *portp)
	245	{
	246	portp = (unsigned short) ntohl(p++);
	247	return 0;
	248	}
	249
	250	static int
	251	xdr_decode_bool(struct rpc_rqst req, u32 p, unsigned int *boolp)
	252	{
	253	boolp = (unsigned int) ntohl(p++);
	254	return 0;
	255	}
	256
	257	static struct rpc_procinfo pmap_procedures[] = {
	258	[PMAP_SET] = {
	259	.p_proc = PMAP_SET,
	260	.p_encode = (kxdrproc_t) xdr_encode_mapping,
	261	.p_decode = (kxdrproc_t) xdr_decode_bool,
	262	.p_bufsiz = 4,
	263	.p_count = 1,
	264	},
	265	[PMAP_UNSET] = {
	266	.p_proc = PMAP_UNSET,
	267	.p_encode = (kxdrproc_t) xdr_encode_mapping,
	268	.p_decode = (kxdrproc_t) xdr_decode_bool,
	269	.p_bufsiz = 4,
	270	.p_count = 1,
	271	},
	272	[PMAP_GETPORT] = {
	273	.p_proc = PMAP_GETPORT,
	274	.p_encode = (kxdrproc_t) xdr_encode_mapping,
	275	.p_decode = (kxdrproc_t) xdr_decode_port,
	276	.p_bufsiz = 4,
	277	.p_count = 1,
	278	},
	279	};
	280
	281	static struct rpc_version pmap_version2 = {
	282	.number = 2,
283	.nrprocs = 4,
284	.procs = pmap_procedures
285	};
286
287	static struct rpc_version * pmap_version[] = {
288	NULL,
289	NULL,
290	&pmap_version2
291	};
292
293	static struct rpc_stat pmap_stats;
294
295	static struct rpc_program pmap_program = {
296	.name = "portmap",
297	.number = RPC_PMAP_PROGRAM,
298	.nrvers = ARRAY_SIZE(pmap_version),
299	.version = pmap_version,
300	.stats = &pmap_stats,
301	};