[net-next-2.6.git] / fs / lockd / mon.c

/*
 * linux/fs/lockd/mon.c
 *
 * The kernel statd client.
 *
 * Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de>
 */

#include <linux/types.h>
#include <linux/utsname.h>
#include <linux/kernel.h>
#include <linux/sunrpc/clnt.h>
#include <linux/sunrpc/xprtsock.h>
#include <linux/sunrpc/svc.h>
#include <linux/lockd/lockd.h>
#include <linux/lockd/sm_inter.h>


#define NLMDBG_FACILITY		NLMDBG_MONITOR

static struct rpc_clnt *	nsm_create(void);

static struct rpc_program	nsm_program;

/*
 * Local NSM state
 */
int				nsm_local_state;

/*
 * Common procedure for SM_MON/SM_UNMON calls
 */
static int
nsm_mon_unmon(struct nsm_handle *nsm, u32 proc, struct nsm_res *res)
{
	struct rpc_clnt	*clnt;
	int		status;
	struct nsm_args args = {
		.addr		= nsm_addr_in(nsm)->sin_addr.s_addr,
		.prog		= NLM_PROGRAM,
		.vers		= 3,
		.proc		= NLMPROC_NSM_NOTIFY,
		.mon_name	= nsm->sm_mon_name,
	};
	struct rpc_message msg = {
		.rpc_argp	= &args,
		.rpc_resp	= res,
	};

	clnt = nsm_create();
	if (IS_ERR(clnt)) {
		status = PTR_ERR(clnt);
		dprintk("lockd: failed to create NSM upcall transport, "
				"status=%d\n", status);
		goto out;
	}

	memset(res, 0, sizeof(*res));

	msg.rpc_proc = &clnt->cl_procinfo[proc];
	status = rpc_call_sync(clnt, &msg, 0);
	if (status < 0)
		dprintk("lockd: NSM upcall RPC failed, status=%d\n",
				status);
	else
		status = 0;
	rpc_shutdown_client(clnt);
 out:
	return status;
}

/*
 * Set up monitoring of a remote host
 */
int
nsm_monitor(struct nlm_host *host)
{
	struct nsm_handle *nsm = host->h_nsmhandle;
	struct nsm_res	res;
	int		status;

	dprintk("lockd: nsm_monitor(%s)\n", host->h_name);
	BUG_ON(nsm == NULL);

	if (nsm->sm_monitored)
		return 0;

	/*
	 * Choose whether to record the caller_name or IP address of
	 * this peer in the local rpc.statd's database.
	 */
	nsm->sm_mon_name = nsm_use_hostnames ? nsm->sm_name : nsm->sm_addrbuf;

	status = nsm_mon_unmon(nsm, SM_MON, &res);

	if (status < 0 || res.status != 0)
		printk(KERN_NOTICE "lockd: cannot monitor %s\n", host->h_name);
	else
		nsm->sm_monitored = 1;
	return status;
}

/*
 * Cease to monitor remote host
 */
int
nsm_unmonitor(struct nlm_host *host)
{
	struct nsm_handle *nsm = host->h_nsmhandle;
	struct nsm_res	res;
	int		status = 0;

	if (nsm == NULL)
		return 0;
	host->h_nsmhandle = NULL;

	if (atomic_read(&nsm->sm_count) == 1
	 && nsm->sm_monitored && !nsm->sm_sticky) {
		dprintk("lockd: nsm_unmonitor(%s)\n", host->h_name);

		status = nsm_mon_unmon(nsm, SM_UNMON, &res);
		if (status < 0)
			printk(KERN_NOTICE "lockd: cannot unmonitor %s\n",
					host->h_name);
		else
			nsm->sm_monitored = 0;
	}
	nsm_release(nsm);
	return status;
}

/*
 * Create NSM client for the local host
 */
static struct rpc_clnt *
nsm_create(void)
{
	struct sockaddr_in	sin = {
		.sin_family	= AF_INET,
		.sin_addr.s_addr = htonl(INADDR_LOOPBACK),
		.sin_port	= 0,
	};
	struct rpc_create_args args = {
		.protocol	= XPRT_TRANSPORT_UDP,
		.address	= (struct sockaddr *)&sin,
		.addrsize	= sizeof(sin),
		.servername	= "localhost",
		.program	= &nsm_program,
		.version	= SM_VERSION,
		.authflavor	= RPC_AUTH_NULL,
	};

	return rpc_create(&args);
}

/*
 * XDR functions for NSM.
 *
 * See http://www.opengroup.org/ for details on the Network
 * Status Monitor wire protocol.
 */

static __be32 *xdr_encode_nsm_string(__be32 *p, char *string)
{
	size_t len = strlen(string);

	if (len > SM_MAXSTRLEN)
		len = SM_MAXSTRLEN;
	return xdr_encode_opaque(p, string, len);
}

/*
 * "mon_name" specifies the host to be monitored.
 */
static __be32 *xdr_encode_mon_name(__be32 *p, struct nsm_args *argp)
{
	return xdr_encode_nsm_string(p, argp->mon_name);
}

/*
 * The "my_id" argument specifies the hostname and RPC procedure
 * to be called when the status manager receives notification
 * (via the SM_NOTIFY call) that the state of host "mon_name"
 * has changed.
 */
static __be32 *xdr_encode_my_id(__be32 *p, struct nsm_args *argp)
{
	p = xdr_encode_nsm_string(p, utsname()->nodename);
	if (!p)
		return ERR_PTR(-EIO);

	*p++ = htonl(argp->prog);
	*p++ = htonl(argp->vers);
	*p++ = htonl(argp->proc);

	return p;
}

/*
 * The "mon_id" argument specifies the non-private arguments
 * of an SM_MON or SM_UNMON call.
 */
static __be32 *xdr_encode_mon_id(__be32 *p, struct nsm_args *argp)
{
	p = xdr_encode_mon_name(p, argp);
	if (!p)
		return ERR_PTR(-EIO);

	return xdr_encode_my_id(p, argp);
}

/*
 * The "priv" argument may contain private information required
 * by the SM_MON call. This information will be supplied in the
 * SM_NOTIFY call.
 *
 * Linux provides the raw IP address of the monitored host,
 * left in network byte order.
 */
static __be32 *xdr_encode_priv(__be32 *p, struct nsm_args *argp)
{
	*p++ = argp->addr;
	*p++ = 0;
	*p++ = 0;
	*p++ = 0;

	return p;
}

static int
xdr_encode_mon(struct rpc_rqst *rqstp, __be32 *p, struct nsm_args *argp)
{
	p = xdr_encode_mon_id(p, argp);
	if (IS_ERR(p))
		return PTR_ERR(p);

	p = xdr_encode_priv(p, argp);
	if (IS_ERR(p))
		return PTR_ERR(p);

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

static int
xdr_encode_unmon(struct rpc_rqst *rqstp, __be32 *p, struct nsm_args *argp)
{
	p = xdr_encode_mon_id(p, argp);
	if (IS_ERR(p))
		return PTR_ERR(p);
	rqstp->rq_slen = xdr_adjust_iovec(rqstp->rq_svec, p);
	return 0;
}

static int
xdr_decode_stat_res(struct rpc_rqst *rqstp, __be32 *p, struct nsm_res *resp)
{
	resp->status = ntohl(*p++);
	resp->state = ntohl(*p++);
	dprintk("nsm: xdr_decode_stat_res status %d state %d\n",
			resp->status, resp->state);
	return 0;
}

static int
xdr_decode_stat(struct rpc_rqst *rqstp, __be32 *p, struct nsm_res *resp)
{
	resp->state = ntohl(*p++);
	return 0;
}

#define SM_my_name_sz	(1+XDR_QUADLEN(SM_MAXSTRLEN))
#define SM_my_id_sz	(SM_my_name_sz+3)
#define SM_mon_name_sz	(1+XDR_QUADLEN(SM_MAXSTRLEN))
#define SM_mon_id_sz	(SM_mon_name_sz+SM_my_id_sz)
#define SM_priv_sz	(XDR_QUADLEN(SM_PRIV_SIZE))
#define SM_mon_sz	(SM_mon_id_sz+SM_priv_sz)
#define SM_monres_sz	2
#define SM_unmonres_sz	1

static struct rpc_procinfo	nsm_procedures[] = {
[SM_MON] = {
		.p_proc		= SM_MON,
		.p_encode	= (kxdrproc_t) xdr_encode_mon,
		.p_decode	= (kxdrproc_t) xdr_decode_stat_res,
		.p_arglen	= SM_mon_sz,
		.p_replen	= SM_monres_sz,
		.p_statidx	= SM_MON,
		.p_name		= "MONITOR",
	},
[SM_UNMON] = {
		.p_proc		= SM_UNMON,
		.p_encode	= (kxdrproc_t) xdr_encode_unmon,
		.p_decode	= (kxdrproc_t) xdr_decode_stat,
		.p_arglen	= SM_mon_id_sz,
		.p_replen	= SM_unmonres_sz,
		.p_statidx	= SM_UNMON,
		.p_name		= "UNMONITOR",
	},
};

static struct rpc_version	nsm_version1 = {
		.number		= 1,
		.nrprocs	= ARRAY_SIZE(nsm_procedures),
		.procs		= nsm_procedures
};

static struct rpc_version *	nsm_version[] = {
	[1] = &nsm_version1,
};

static struct rpc_stat		nsm_stats;

static struct rpc_program	nsm_program = {
		.name		= "statd",
		.number		= SM_PROGRAM,
		.nrvers		= ARRAY_SIZE(nsm_version),
		.version	= nsm_version,
		.stats		= &nsm_stats
};
Commit	Line	Data
1da177e4 LT	1	/*
	2	* linux/fs/lockd/mon.c
	3	*
	4	* The kernel statd client.
	5	*
	6	* Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de>
	7	*/
	8
	9	#include <linux/types.h>
	10	#include <linux/utsname.h>
	11	#include <linux/kernel.h>
	12	#include <linux/sunrpc/clnt.h>
0896a725	13	#include <linux/sunrpc/xprtsock.h>
1da177e4 LT	14	#include <linux/sunrpc/svc.h>
	15	#include <linux/lockd/lockd.h>
	16	#include <linux/lockd/sm_inter.h>
	17
	18
	19	#define NLMDBG_FACILITY NLMDBG_MONITOR
	20
	21	static struct rpc_clnt * nsm_create(void);
	22
	23	static struct rpc_program nsm_program;
	24
	25	/*
	26	* Local NSM state
	27	*/
460f5cac	28	int nsm_local_state;
1da177e4 LT	29
	30	/*
	31	* Common procedure for SM_MON/SM_UNMON calls
	32	*/
	33	static int
9502c522	34	nsm_mon_unmon(struct nsm_handle nsm, u32 proc, struct nsm_res res)
1da177e4 LT	35	{
	36	struct rpc_clnt *clnt;
	37	int status;
a4846750 CL	38	struct nsm_args args = {
	39	.addr = nsm_addr_in(nsm)->sin_addr.s_addr,
	40	.prog = NLM_PROGRAM,
	41	.vers = 3,
	42	.proc = NLMPROC_NSM_NOTIFY,
29ed1407	43	.mon_name = nsm->sm_mon_name,
a4846750	44	};
dead28da CL	45	struct rpc_message msg = {
	46	.rpc_argp = &args,
	47	.rpc_resp = res,
	48	};
1da177e4 LT	49
	50	clnt = nsm_create();
	51	if (IS_ERR(clnt)) {
	52	status = PTR_ERR(clnt);
5acf4315 CL	53	dprintk("lockd: failed to create NSM upcall transport, "
5acf4315 CL	54	"status=%d\n", status);
1da177e4 LT	55	goto out;
	56	}
	57
1da177e4 LT	58	memset(res, 0, sizeof(*res));
1da177e4 LT	59
dead28da CL	60	msg.rpc_proc = &clnt->cl_procinfo[proc];
dead28da CL	61	status = rpc_call_sync(clnt, &msg, 0);
1da177e4	62	if (status < 0)
5acf4315 CL	63	dprintk("lockd: NSM upcall RPC failed, status=%d\n",
5acf4315 CL	64	status);
1da177e4 LT	65	else
1da177e4 LT	66	status = 0;
90c5755f	67	rpc_shutdown_client(clnt);
1da177e4 LT	68	out:
	69	return status;
	70	}
	71
	72	/*
	73	* Set up monitoring of a remote host
	74	*/
	75	int
	76	nsm_monitor(struct nlm_host *host)
	77	{
8dead0db	78	struct nsm_handle *nsm = host->h_nsmhandle;
1da177e4 LT	79	struct nsm_res res;
	80	int status;
	81
	82	dprintk("lockd: nsm_monitor(%s)\n", host->h_name);
8dead0db OK	83	BUG_ON(nsm == NULL);
	84
	85	if (nsm->sm_monitored)
977faf39	86	return 0;
1da177e4	87
29ed1407 CL	88	/*
	89	* Choose whether to record the caller_name or IP address of
	90	* this peer in the local rpc.statd's database.
	91	*/
	92	nsm->sm_mon_name = nsm_use_hostnames ? nsm->sm_name : nsm->sm_addrbuf;
	93
9502c522	94	status = nsm_mon_unmon(nsm, SM_MON, &res);
1da177e4 LT	95
	96	if (status < 0 \|\| res.status != 0)
	97	printk(KERN_NOTICE "lockd: cannot monitor %s\n", host->h_name);
	98	else
8dead0db	99	nsm->sm_monitored = 1;
1da177e4 LT	100	return status;
	101	}
	102
	103	/*
	104	* Cease to monitor remote host
	105	*/
	106	int
	107	nsm_unmonitor(struct nlm_host *host)
	108	{
8dead0db	109	struct nsm_handle *nsm = host->h_nsmhandle;
1da177e4	110	struct nsm_res res;
977faf39	111	int status = 0;
1da177e4	112
8dead0db	113	if (nsm == NULL)
977faf39	114	return 0;
8dead0db	115	host->h_nsmhandle = NULL;
977faf39	116
9502c522 OK	117	if (atomic_read(&nsm->sm_count) == 1
	118	&& nsm->sm_monitored && !nsm->sm_sticky) {
	119	dprintk("lockd: nsm_unmonitor(%s)\n", host->h_name);
	120
	121	status = nsm_mon_unmon(nsm, SM_UNMON, &res);
977faf39	122	if (status < 0)
9502c522 OK	123	printk(KERN_NOTICE "lockd: cannot unmonitor %s\n",
	124	host->h_name);
	125	else
	126	nsm->sm_monitored = 0;
977faf39	127	}
8dead0db	128	nsm_release(nsm);
1da177e4 LT	129	return status;
	130	}
	131
	132	/*
	133	* Create NSM client for the local host
	134	*/
	135	static struct rpc_clnt *
	136	nsm_create(void)
	137	{
e1ec7892 CL	138	struct sockaddr_in sin = {
	139	.sin_family = AF_INET,
	140	.sin_addr.s_addr = htonl(INADDR_LOOPBACK),
	141	.sin_port = 0,
	142	};
	143	struct rpc_create_args args = {
0896a725	144	.protocol = XPRT_TRANSPORT_UDP,
e1ec7892 CL	145	.address = (struct sockaddr *)&sin,
	146	.addrsize = sizeof(sin),
	147	.servername = "localhost",
	148	.program = &nsm_program,
	149	.version = SM_VERSION,
	150	.authflavor = RPC_AUTH_NULL,
e1ec7892 CL	151	};
	152
	153	return rpc_create(&args);
1da177e4 LT	154	}
	155
	156	/*
	157	* XDR functions for NSM.
2ca7754d CL	158	*
	159	* See http://www.opengroup.org/ for details on the Network
	160	* Status Monitor wire protocol.
1da177e4 LT	161	*/
1da177e4 LT	162
099bd05f CL	163	static __be32 xdr_encode_nsm_string(__be32 p, char *string)
	164	{
	165	size_t len = strlen(string);
	166
	167	if (len > SM_MAXSTRLEN)
	168	len = SM_MAXSTRLEN;
	169	return xdr_encode_opaque(p, string, len);
	170	}
	171
49695174 CL	172	/*
49695174 CL	173	* "mon_name" specifies the host to be monitored.
49695174 CL	174	*/
	175	static __be32 xdr_encode_mon_name(__be32 p, struct nsm_args *argp)
	176	{
29ed1407	177	return xdr_encode_nsm_string(p, argp->mon_name);
49695174 CL	178	}
49695174 CL	179
850c95fd CL	180	/*
	181	* The "my_id" argument specifies the hostname and RPC procedure
	182	* to be called when the status manager receives notification
	183	* (via the SM_NOTIFY call) that the state of host "mon_name"
	184	* has changed.
	185	*/
	186	static __be32 xdr_encode_my_id(__be32 p, struct nsm_args *argp)
	187	{
	188	p = xdr_encode_nsm_string(p, utsname()->nodename);
	189	if (!p)
	190	return ERR_PTR(-EIO);
	191
	192	*p++ = htonl(argp->prog);
	193	*p++ = htonl(argp->vers);
	194	*p++ = htonl(argp->proc);
	195
	196	return p;
	197	}
	198
ea72a7f1 CL	199	/*
	200	* The "mon_id" argument specifies the non-private arguments
	201	* of an SM_MON or SM_UNMON call.
	202	*/
	203	static __be32 xdr_encode_mon_id(__be32 p, struct nsm_args *argp)
	204	{
	205	p = xdr_encode_mon_name(p, argp);
	206	if (!p)
	207	return ERR_PTR(-EIO);
	208
	209	return xdr_encode_my_id(p, argp);
	210	}
	211
0490a54a CL	212	/*
	213	* The "priv" argument may contain private information required
	214	* by the SM_MON call. This information will be supplied in the
	215	* SM_NOTIFY call.
	216	*
	217	* Linux provides the raw IP address of the monitored host,
	218	* left in network byte order.
	219	*/
	220	static __be32 xdr_encode_priv(__be32 p, struct nsm_args *argp)
	221	{
	222	*p++ = argp->addr;
	223	*p++ = 0;
	224	*p++ = 0;
	225	*p++ = 0;
	226
	227	return p;
	228	}
	229
1da177e4	230	static int
52921e02	231	xdr_encode_mon(struct rpc_rqst rqstp, __be32 p, struct nsm_args *argp)
1da177e4	232	{
2ca7754d	233	p = xdr_encode_mon_id(p, argp);
1da177e4 LT	234	if (IS_ERR(p))
1da177e4 LT	235	return PTR_ERR(p);
9502c522	236
0490a54a CL	237	p = xdr_encode_priv(p, argp);
	238	if (IS_ERR(p))
	239	return PTR_ERR(p);
	240
1da177e4 LT	241	rqstp->rq_slen = xdr_adjust_iovec(rqstp->rq_svec, p);
	242	return 0;
	243	}
	244
	245	static int
52921e02	246	xdr_encode_unmon(struct rpc_rqst rqstp, __be32 p, struct nsm_args *argp)
1da177e4	247	{
2ca7754d	248	p = xdr_encode_mon_id(p, argp);
1da177e4 LT	249	if (IS_ERR(p))
	250	return PTR_ERR(p);
	251	rqstp->rq_slen = xdr_adjust_iovec(rqstp->rq_svec, p);
	252	return 0;
	253	}
	254
	255	static int
52921e02	256	xdr_decode_stat_res(struct rpc_rqst rqstp, __be32 p, struct nsm_res *resp)
1da177e4 LT	257	{
	258	resp->status = ntohl(*p++);
	259	resp->state = ntohl(*p++);
	260	dprintk("nsm: xdr_decode_stat_res status %d state %d\n",
	261	resp->status, resp->state);
	262	return 0;
	263	}
	264
	265	static int
52921e02	266	xdr_decode_stat(struct rpc_rqst rqstp, __be32 p, struct nsm_res *resp)
1da177e4 LT	267	{
	268	resp->state = ntohl(*p++);
	269	return 0;
	270	}
	271
	272	#define SM_my_name_sz (1+XDR_QUADLEN(SM_MAXSTRLEN))
2ca7754d CL	273	#define SM_my_id_sz (SM_my_name_sz+3)
	274	#define SM_mon_name_sz (1+XDR_QUADLEN(SM_MAXSTRLEN))
	275	#define SM_mon_id_sz (SM_mon_name_sz+SM_my_id_sz)
0490a54a CL	276	#define SM_priv_sz (XDR_QUADLEN(SM_PRIV_SIZE))
0490a54a CL	277	#define SM_mon_sz (SM_mon_id_sz+SM_priv_sz)
1da177e4 LT	278	#define SM_monres_sz 2
	279	#define SM_unmonres_sz 1
	280
1da177e4 LT	281	static struct rpc_procinfo nsm_procedures[] = {
	282	[SM_MON] = {
	283	.p_proc = SM_MON,
	284	.p_encode = (kxdrproc_t) xdr_encode_mon,
	285	.p_decode = (kxdrproc_t) xdr_decode_stat_res,
2bea90d4 CL	286	.p_arglen = SM_mon_sz,
2bea90d4 CL	287	.p_replen = SM_monres_sz,
cc0175c1 CL	288	.p_statidx = SM_MON,
cc0175c1 CL	289	.p_name = "MONITOR",
1da177e4 LT	290	},
	291	[SM_UNMON] = {
	292	.p_proc = SM_UNMON,
	293	.p_encode = (kxdrproc_t) xdr_encode_unmon,
	294	.p_decode = (kxdrproc_t) xdr_decode_stat,
2bea90d4 CL	295	.p_arglen = SM_mon_id_sz,
2bea90d4 CL	296	.p_replen = SM_unmonres_sz,
cc0175c1 CL	297	.p_statidx = SM_UNMON,
cc0175c1 CL	298	.p_name = "UNMONITOR",
1da177e4 LT	299	},
	300	};
	301
	302	static struct rpc_version nsm_version1 = {
e8c96f8c TK	303	.number = 1,
e8c96f8c TK	304	.nrprocs = ARRAY_SIZE(nsm_procedures),
1da177e4 LT	305	.procs = nsm_procedures
	306	};
	307
	308	static struct rpc_version * nsm_version[] = {
	309	[1] = &nsm_version1,
	310	};
	311
	312	static struct rpc_stat nsm_stats;
	313
	314	static struct rpc_program nsm_program = {
	315	.name = "statd",
	316	.number = SM_PROGRAM,
e8c96f8c	317	.nrvers = ARRAY_SIZE(nsm_version),
1da177e4 LT	318	.version = nsm_version,
	319	.stats = &nsm_stats
	320	};