mon.c 5.6 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26
/*
 * 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/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
 */
27
int				nsm_local_state;
L
Linus Torvalds 已提交
28 29 30 31 32

/*
 * Common procedure for SM_MON/SM_UNMON calls
 */
static int
33
nsm_mon_unmon(struct nsm_handle *nsm, u32 proc, struct nsm_res *res)
L
Linus Torvalds 已提交
34 35 36 37
{
	struct rpc_clnt	*clnt;
	int		status;
	struct nsm_args	args;
38 39 40 41
	struct rpc_message msg = {
		.rpc_argp	= &args,
		.rpc_resp	= res,
	};
L
Linus Torvalds 已提交
42 43 44 45 46 47 48

	clnt = nsm_create();
	if (IS_ERR(clnt)) {
		status = PTR_ERR(clnt);
		goto out;
	}

49
	memset(&args, 0, sizeof(args));
50
	args.mon_name = nsm->sm_name;
51
	args.addr = nsm->sm_addr.sin_addr.s_addr;
L
Linus Torvalds 已提交
52
	args.prog = NLM_PROGRAM;
53
	args.vers = 3;
L
Linus Torvalds 已提交
54 55 56
	args.proc = NLMPROC_NSM_NOTIFY;
	memset(res, 0, sizeof(*res));

57 58
	msg.rpc_proc = &clnt->cl_procinfo[proc];
	status = rpc_call_sync(clnt, &msg, 0);
L
Linus Torvalds 已提交
59 60 61 62 63 64 65 66 67 68 69 70 71 72 73
	if (status < 0)
		printk(KERN_DEBUG "nsm_mon_unmon: rpc failed, status=%d\n",
			status);
	else
		status = 0;
 out:
	return status;
}

/*
 * Set up monitoring of a remote host
 */
int
nsm_monitor(struct nlm_host *host)
{
74
	struct nsm_handle *nsm = host->h_nsmhandle;
L
Linus Torvalds 已提交
75 76 77 78
	struct nsm_res	res;
	int		status;

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

	if (nsm->sm_monitored)
82
		return 0;
L
Linus Torvalds 已提交
83

84
	status = nsm_mon_unmon(nsm, SM_MON, &res);
L
Linus Torvalds 已提交
85 86 87 88

	if (status < 0 || res.status != 0)
		printk(KERN_NOTICE "lockd: cannot monitor %s\n", host->h_name);
	else
89
		nsm->sm_monitored = 1;
L
Linus Torvalds 已提交
90 91 92 93 94 95 96 97 98
	return status;
}

/*
 * Cease to monitor remote host
 */
int
nsm_unmonitor(struct nlm_host *host)
{
99
	struct nsm_handle *nsm = host->h_nsmhandle;
L
Linus Torvalds 已提交
100
	struct nsm_res	res;
101
	int		status = 0;
L
Linus Torvalds 已提交
102

103
	if (nsm == NULL)
104
		return 0;
105
	host->h_nsmhandle = NULL;
106

107 108 109 110 111
	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);
112
		if (status < 0)
113 114 115 116
			printk(KERN_NOTICE "lockd: cannot unmonitor %s\n",
					host->h_name);
		else
			nsm->sm_monitored = 0;
117
	}
118
	nsm_release(nsm);
L
Linus Torvalds 已提交
119 120 121 122 123 124 125 126 127
	return status;
}

/*
 * Create NSM client for the local host
 */
static struct rpc_clnt *
nsm_create(void)
{
128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144
	struct sockaddr_in	sin = {
		.sin_family	= AF_INET,
		.sin_addr.s_addr = htonl(INADDR_LOOPBACK),
		.sin_port	= 0,
	};
	struct rpc_create_args args = {
		.protocol	= IPPROTO_UDP,
		.address	= (struct sockaddr *)&sin,
		.addrsize	= sizeof(sin),
		.servername	= "localhost",
		.program	= &nsm_program,
		.version	= SM_VERSION,
		.authflavor	= RPC_AUTH_NULL,
		.flags		= (RPC_CLNT_CREATE_ONESHOT),
	};

	return rpc_create(&args);
L
Linus Torvalds 已提交
145 146 147 148 149 150 151 152 153
}

/*
 * XDR functions for NSM.
 */

static u32 *
xdr_encode_common(struct rpc_rqst *rqstp, u32 *p, struct nsm_args *argp)
{
154
	char	buffer[20], *name;
L
Linus Torvalds 已提交
155 156 157 158 159 160 161

	/*
	 * Use the dotted-quad IP address of the remote host as
	 * identifier. Linux statd always looks up the canonical
	 * hostname first for whatever remote hostname it receives,
	 * so this works alright.
	 */
162 163 164 165 166 167 168
	if (nsm_use_hostnames) {
		name = argp->mon_name;
	} else {
		sprintf(buffer, "%u.%u.%u.%u", NIPQUAD(argp->addr));
		name = buffer;
	}
	if (!(p = xdr_encode_string(p, name))
169
	 || !(p = xdr_encode_string(p, utsname()->nodename)))
L
Linus Torvalds 已提交
170 171 172 173 174 175 176 177 178 179 180 181 182 183
		return ERR_PTR(-EIO);
	*p++ = htonl(argp->prog);
	*p++ = htonl(argp->vers);
	*p++ = htonl(argp->proc);

	return p;
}

static int
xdr_encode_mon(struct rpc_rqst *rqstp, u32 *p, struct nsm_args *argp)
{
	p = xdr_encode_common(rqstp, p, argp);
	if (IS_ERR(p))
		return PTR_ERR(p);
184 185

	/* Surprise - there may even be room for an IPv6 address now */
L
Linus Torvalds 已提交
186
	*p++ = argp->addr;
187 188
	*p++ = 0;
	*p++ = 0;
L
Linus Torvalds 已提交
189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237
	*p++ = 0;
	rqstp->rq_slen = xdr_adjust_iovec(rqstp->rq_svec, p);
	return 0;
}

static int
xdr_encode_unmon(struct rpc_rqst *rqstp, u32 *p, struct nsm_args *argp)
{
	p = xdr_encode_common(rqstp, 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, u32 *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, u32 *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	(3+1+SM_my_name_sz)
#define SM_mon_id_sz	(1+XDR_QUADLEN(20)+SM_my_id_sz)
#define SM_mon_sz	(SM_mon_id_sz+4)
#define SM_monres_sz	2
#define SM_unmonres_sz	1

#ifndef MAX
# define MAX(a, b)	(((a) > (b))? (a) : (b))
#endif

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_bufsiz	= MAX(SM_mon_sz, SM_monres_sz) << 2,
238 239
		.p_statidx	= SM_MON,
		.p_name		= "MONITOR",
L
Linus Torvalds 已提交
240 241 242 243 244 245
	},
[SM_UNMON] = {
		.p_proc		= SM_UNMON,
		.p_encode	= (kxdrproc_t) xdr_encode_unmon,
		.p_decode	= (kxdrproc_t) xdr_decode_stat,
		.p_bufsiz	= MAX(SM_mon_id_sz, SM_unmonres_sz) << 2,
246 247
		.p_statidx	= SM_UNMON,
		.p_name		= "UNMONITOR",
L
Linus Torvalds 已提交
248 249 250 251
	},
};

static struct rpc_version	nsm_version1 = {
252 253
		.number		= 1,
		.nrprocs	= ARRAY_SIZE(nsm_procedures),
L
Linus Torvalds 已提交
254 255 256 257 258 259 260 261 262 263 264 265
		.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,
266
		.nrvers		= ARRAY_SIZE(nsm_version),
L
Linus Torvalds 已提交
267 268 269
		.version	= nsm_version,
		.stats		= &nsm_stats
};
新手
引导
客服 返回
顶部