mon.c 12.1 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11 12
/*
 * 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>
13
#include <linux/sunrpc/xprtsock.h>
L
Linus Torvalds 已提交
14 15 16 17 18 19
#include <linux/sunrpc/svc.h>
#include <linux/lockd/lockd.h>
#include <linux/lockd/sm_inter.h>


#define NLMDBG_FACILITY		NLMDBG_MONITOR
20 21 22 23 24 25 26 27 28 29 30 31
#define NSM_PROGRAM		100024
#define NSM_VERSION		1

enum {
	NSMPROC_NULL,
	NSMPROC_STAT,
	NSMPROC_MON,
	NSMPROC_UNMON,
	NSMPROC_UNMON_ALL,
	NSMPROC_SIMU_CRASH,
	NSMPROC_NOTIFY,
};
L
Linus Torvalds 已提交
32

33 34 35 36 37 38 39 40 41 42 43 44 45 46
struct nsm_args {
	__be32			addr;		/* remote address */
	u32			prog;		/* RPC callback info */
	u32			vers;
	u32			proc;

	char			*mon_name;
};

struct nsm_res {
	u32			status;
	u32			state;
};

L
Linus Torvalds 已提交
47 48 49
static struct rpc_clnt *	nsm_create(void);

static struct rpc_program	nsm_program;
50 51
static				LIST_HEAD(nsm_handles);
static				DEFINE_SPINLOCK(nsm_lock);
L
Linus Torvalds 已提交
52 53 54 55

/*
 * Local NSM state
 */
56
int				nsm_local_state;
L
Linus Torvalds 已提交
57

58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94
static void nsm_display_ipv4_address(const struct sockaddr *sap, char *buf,
				     const size_t len)
{
	const struct sockaddr_in *sin = (struct sockaddr_in *)sap;
	snprintf(buf, len, "%pI4", &sin->sin_addr.s_addr);
}

static void nsm_display_ipv6_address(const struct sockaddr *sap, char *buf,
				     const size_t len)
{
	const struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)sap;

	if (ipv6_addr_v4mapped(&sin6->sin6_addr))
		snprintf(buf, len, "%pI4", &sin6->sin6_addr.s6_addr32[3]);
	else if (sin6->sin6_scope_id != 0)
		snprintf(buf, len, "%pI6%%%u", &sin6->sin6_addr,
				sin6->sin6_scope_id);
	else
		snprintf(buf, len, "%pI6", &sin6->sin6_addr);
}

static void nsm_display_address(const struct sockaddr *sap,
				char *buf, const size_t len)
{
	switch (sap->sa_family) {
	case AF_INET:
		nsm_display_ipv4_address(sap, buf, len);
		break;
	case AF_INET6:
		nsm_display_ipv6_address(sap, buf, len);
		break;
	default:
		snprintf(buf, len, "unsupported address family");
		break;
	}
}

L
Linus Torvalds 已提交
95
/*
96
 * Common procedure for NSMPROC_MON/NSMPROC_UNMON calls
L
Linus Torvalds 已提交
97 98
 */
static int
99
nsm_mon_unmon(struct nsm_handle *nsm, u32 proc, struct nsm_res *res)
L
Linus Torvalds 已提交
100 101 102
{
	struct rpc_clnt	*clnt;
	int		status;
103 104 105 106 107
	struct nsm_args args = {
		.addr		= nsm_addr_in(nsm)->sin_addr.s_addr,
		.prog		= NLM_PROGRAM,
		.vers		= 3,
		.proc		= NLMPROC_NSM_NOTIFY,
108
		.mon_name	= nsm->sm_mon_name,
109
	};
110 111 112 113
	struct rpc_message msg = {
		.rpc_argp	= &args,
		.rpc_resp	= res,
	};
L
Linus Torvalds 已提交
114 115 116 117

	clnt = nsm_create();
	if (IS_ERR(clnt)) {
		status = PTR_ERR(clnt);
118 119
		dprintk("lockd: failed to create NSM upcall transport, "
				"status=%d\n", status);
L
Linus Torvalds 已提交
120 121 122 123 124
		goto out;
	}

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

125 126
	msg.rpc_proc = &clnt->cl_procinfo[proc];
	status = rpc_call_sync(clnt, &msg, 0);
L
Linus Torvalds 已提交
127
	if (status < 0)
128 129
		dprintk("lockd: NSM upcall RPC failed, status=%d\n",
				status);
L
Linus Torvalds 已提交
130 131
	else
		status = 0;
132
	rpc_shutdown_client(clnt);
L
Linus Torvalds 已提交
133 134 135 136
 out:
	return status;
}

137 138 139 140 141 142 143 144 145 146
/**
 * nsm_monitor - Notify a peer in case we reboot
 * @host: pointer to nlm_host of peer to notify
 *
 * If this peer is not already monitored, this function sends an
 * upcall to the local rpc.statd to record the name/address of
 * the peer to notify in case we reboot.
 *
 * Returns zero if the peer is monitored by the local rpc.statd;
 * otherwise a negative errno value is returned.
L
Linus Torvalds 已提交
147
 */
148
int nsm_monitor(const struct nlm_host *host)
L
Linus Torvalds 已提交
149
{
150
	struct nsm_handle *nsm = host->h_nsmhandle;
L
Linus Torvalds 已提交
151 152 153
	struct nsm_res	res;
	int		status;

154
	dprintk("lockd: nsm_monitor(%s)\n", nsm->sm_name);
155 156

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

159 160 161 162 163 164
	/*
	 * 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;

165
	status = nsm_mon_unmon(nsm, NSMPROC_MON, &res);
166 167 168
	if (res.status != 0)
		status = -EIO;
	if (status < 0)
169
		printk(KERN_NOTICE "lockd: cannot monitor %s\n", nsm->sm_name);
L
Linus Torvalds 已提交
170
	else
171
		nsm->sm_monitored = 1;
L
Linus Torvalds 已提交
172 173 174
	return status;
}

175 176 177 178 179 180 181
/**
 * nsm_unmonitor - Unregister peer notification
 * @host: pointer to nlm_host of peer to stop monitoring
 *
 * If this peer is monitored, this function sends an upcall to
 * tell the local rpc.statd not to send this peer a notification
 * when we reboot.
L
Linus Torvalds 已提交
182
 */
183
void nsm_unmonitor(const struct nlm_host *host)
L
Linus Torvalds 已提交
184
{
185
	struct nsm_handle *nsm = host->h_nsmhandle;
L
Linus Torvalds 已提交
186
	struct nsm_res	res;
187
	int status;
L
Linus Torvalds 已提交
188

189 190
	if (atomic_read(&nsm->sm_count) == 1
	 && nsm->sm_monitored && !nsm->sm_sticky) {
191
		dprintk("lockd: nsm_unmonitor(%s)\n", nsm->sm_name);
192

193
		status = nsm_mon_unmon(nsm, NSMPROC_UNMON, &res);
194 195
		if (res.status != 0)
			status = -EIO;
196
		if (status < 0)
197
			printk(KERN_NOTICE "lockd: cannot unmonitor %s\n",
198
					nsm->sm_name);
199 200
		else
			nsm->sm_monitored = 0;
201
	}
L
Linus Torvalds 已提交
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 238 239 240 241 242 243 244 245 246 247 248
/**
 * nsm_find - Find or create a cached nsm_handle
 * @sap: pointer to socket address of handle to find
 * @salen: length of socket address
 * @hostname: pointer to C string containing hostname to find
 * @hostname_len: length of C string
 * @create: one means create new handle if not found in cache
 *
 * Behavior is modulated by the global nsm_use_hostnames variable
 * and by the @create argument.
 *
 * Returns a cached nsm_handle after bumping its ref count, or if
 * @create is set, returns a fresh nsm_handle if a handle that
 * matches @sap and/or @hostname cannot be found in the handle cache.
 * Returns NULL if an error occurs.
 */
struct nsm_handle *nsm_find(const struct sockaddr *sap, const size_t salen,
			    const char *hostname, const size_t hostname_len,
			    const int create)
{
	struct nsm_handle *nsm = NULL;
	struct nsm_handle *pos;

	if (hostname && memchr(hostname, '/', hostname_len) != NULL) {
		if (printk_ratelimit()) {
			printk(KERN_WARNING "Invalid hostname \"%.*s\" "
					    "in NFS lock request\n",
				(int)hostname_len, hostname);
		}
		return NULL;
	}

retry:
	spin_lock(&nsm_lock);
	list_for_each_entry(pos, &nsm_handles, sm_link) {

		if (hostname && nsm_use_hostnames) {
			if (strlen(pos->sm_name) != hostname_len
			 || memcmp(pos->sm_name, hostname, hostname_len))
				continue;
		} else if (!nlm_cmp_addr(nsm_addr(pos), sap))
			continue;
		atomic_inc(&pos->sm_count);
		kfree(nsm);
		nsm = pos;
249 250 251
		dprintk("lockd: found nsm_handle for %s (%s), cnt %d\n",
				pos->sm_name, pos->sm_addrbuf,
				atomic_read(&pos->sm_count));
252 253 254 255
		goto found;
	}
	if (nsm) {
		list_add(&nsm->sm_link, &nsm_handles);
256 257
		dprintk("lockd: created nsm_handle for %s (%s)\n",
				nsm->sm_name, nsm->sm_addrbuf);
258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293
		goto found;
	}
	spin_unlock(&nsm_lock);

	if (!create)
		return NULL;

	nsm = kzalloc(sizeof(*nsm) + hostname_len + 1, GFP_KERNEL);
	if (nsm == NULL)
		return NULL;

	memcpy(nsm_addr(nsm), sap, salen);
	nsm->sm_addrlen = salen;
	nsm->sm_name = (char *) (nsm + 1);
	memcpy(nsm->sm_name, hostname, hostname_len);
	nsm->sm_name[hostname_len] = '\0';
	nsm_display_address((struct sockaddr *)&nsm->sm_addr,
				nsm->sm_addrbuf, sizeof(nsm->sm_addrbuf));
	atomic_set(&nsm->sm_count, 1);
	goto retry;

found:
	spin_unlock(&nsm_lock);
	return nsm;
}

/**
 * nsm_release - Release an NSM handle
 * @nsm: pointer to handle to be released
 *
 */
void nsm_release(struct nsm_handle *nsm)
{
	if (atomic_dec_and_lock(&nsm->sm_count, &nsm_lock)) {
		list_del(&nsm->sm_link);
		spin_unlock(&nsm_lock);
294 295
		dprintk("lockd: destroyed nsm_handle for %s (%s)\n",
				nsm->sm_name, nsm->sm_addrbuf);
296 297 298 299
		kfree(nsm);
	}
}

L
Linus Torvalds 已提交
300 301 302 303 304 305
/*
 * Create NSM client for the local host
 */
static struct rpc_clnt *
nsm_create(void)
{
306 307 308 309 310 311
	struct sockaddr_in	sin = {
		.sin_family	= AF_INET,
		.sin_addr.s_addr = htonl(INADDR_LOOPBACK),
		.sin_port	= 0,
	};
	struct rpc_create_args args = {
312
		.protocol	= XPRT_TRANSPORT_UDP,
313 314 315 316
		.address	= (struct sockaddr *)&sin,
		.addrsize	= sizeof(sin),
		.servername	= "localhost",
		.program	= &nsm_program,
317
		.version	= NSM_VERSION,
318 319 320 321
		.authflavor	= RPC_AUTH_NULL,
	};

	return rpc_create(&args);
L
Linus Torvalds 已提交
322 323 324 325
}

/*
 * XDR functions for NSM.
326 327 328
 *
 * See http://www.opengroup.org/ for details on the Network
 * Status Monitor wire protocol.
L
Linus Torvalds 已提交
329 330
 */

331
static int encode_nsm_string(struct xdr_stream *xdr, const char *string)
332
{
333 334 335 336 337 338 339 340 341 342
	const u32 len = strlen(string);
	__be32 *p;

	if (unlikely(len > SM_MAXSTRLEN))
		return -EIO;
	p = xdr_reserve_space(xdr, sizeof(u32) + len);
	if (unlikely(p == NULL))
		return -EIO;
	xdr_encode_opaque(p, string, len);
	return 0;
343 344
}

345 346 347
/*
 * "mon_name" specifies the host to be monitored.
 */
348
static int encode_mon_name(struct xdr_stream *xdr, const struct nsm_args *argp)
349
{
350
	return encode_nsm_string(xdr, argp->mon_name);
351 352
}

353 354 355
/*
 * The "my_id" argument specifies the hostname and RPC procedure
 * to be called when the status manager receives notification
356
 * (via the NLMPROC_SM_NOTIFY call) that the state of host "mon_name"
357 358
 * has changed.
 */
359
static int encode_my_id(struct xdr_stream *xdr, const struct nsm_args *argp)
360
{
361 362 363 364 365 366 367 368 369
	int status;
	__be32 *p;

	status = encode_nsm_string(xdr, utsname()->nodename);
	if (unlikely(status != 0))
		return status;
	p = xdr_reserve_space(xdr, 3 * sizeof(u32));
	if (unlikely(p == NULL))
		return -EIO;
370 371 372
	*p++ = htonl(argp->prog);
	*p++ = htonl(argp->vers);
	*p++ = htonl(argp->proc);
373
	return 0;
374 375
}

376 377
/*
 * The "mon_id" argument specifies the non-private arguments
378
 * of an NSMPROC_MON or NSMPROC_UNMON call.
379
 */
380
static int encode_mon_id(struct xdr_stream *xdr, const struct nsm_args *argp)
381
{
382
	int status;
383

384 385 386 387
	status = encode_mon_name(xdr, argp);
	if (unlikely(status != 0))
		return status;
	return encode_my_id(xdr, argp);
388 389
}

390 391
/*
 * The "priv" argument may contain private information required
392 393
 * by the NSMPROC_MON call. This information will be supplied in the
 * NLMPROC_SM_NOTIFY call.
394 395 396 397
 *
 * Linux provides the raw IP address of the monitored host,
 * left in network byte order.
 */
398
static int encode_priv(struct xdr_stream *xdr, const struct nsm_args *argp)
399
{
400 401 402 403 404
	__be32 *p;

	p = xdr_reserve_space(xdr, SM_PRIV_SIZE);
	if (unlikely(p == NULL))
		return -EIO;
405 406 407 408
	*p++ = argp->addr;
	*p++ = 0;
	*p++ = 0;
	*p++ = 0;
409
	return 0;
410 411
}

412 413
static int xdr_enc_mon(struct rpc_rqst *req, __be32 *p,
		       const struct nsm_args *argp)
L
Linus Torvalds 已提交
414
{
415 416
	struct xdr_stream xdr;
	int status;
417

418 419 420 421 422
	xdr_init_encode(&xdr, &req->rq_snd_buf, p);
	status = encode_mon_id(&xdr, argp);
	if (unlikely(status))
		return status;
	return encode_priv(&xdr, argp);
L
Linus Torvalds 已提交
423 424
}

425 426
static int xdr_enc_unmon(struct rpc_rqst *req, __be32 *p,
			 const struct nsm_args *argp)
L
Linus Torvalds 已提交
427
{
428 429 430 431
	struct xdr_stream xdr;

	xdr_init_encode(&xdr, &req->rq_snd_buf, p);
	return encode_mon_id(&xdr, argp);
L
Linus Torvalds 已提交
432 433
}

434 435
static int xdr_dec_stat_res(struct rpc_rqst *rqstp, __be32 *p,
			    struct nsm_res *resp)
L
Linus Torvalds 已提交
436
{
437 438 439 440 441 442
	struct xdr_stream xdr;

	xdr_init_decode(&xdr, &rqstp->rq_rcv_buf, p);
	p = xdr_inline_decode(&xdr, 2 * sizeof(u32));
	if (unlikely(p == NULL))
		return -EIO;
L
Linus Torvalds 已提交
443
	resp->status = ntohl(*p++);
444 445 446
	resp->state = ntohl(*p);

	dprintk("lockd: xdr_dec_stat_res status %d state %d\n",
L
Linus Torvalds 已提交
447 448 449 450
			resp->status, resp->state);
	return 0;
}

451 452
static int xdr_dec_stat(struct rpc_rqst *rqstp, __be32 *p,
			struct nsm_res *resp)
L
Linus Torvalds 已提交
453
{
454 455 456 457 458 459 460 461 462
	struct xdr_stream xdr;

	xdr_init_decode(&xdr, &rqstp->rq_rcv_buf, p);
	p = xdr_inline_decode(&xdr, sizeof(u32));
	if (unlikely(p == NULL))
		return -EIO;
	resp->state = ntohl(*p);

	dprintk("lockd: xdr_dec_stat state %d\n", resp->state);
L
Linus Torvalds 已提交
463 464 465 466
	return 0;
}

#define SM_my_name_sz	(1+XDR_QUADLEN(SM_MAXSTRLEN))
467 468 469
#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)
470 471
#define SM_priv_sz	(XDR_QUADLEN(SM_PRIV_SIZE))
#define SM_mon_sz	(SM_mon_id_sz+SM_priv_sz)
L
Linus Torvalds 已提交
472 473 474 475
#define SM_monres_sz	2
#define SM_unmonres_sz	1

static struct rpc_procinfo	nsm_procedures[] = {
476 477
[NSMPROC_MON] = {
		.p_proc		= NSMPROC_MON,
478 479
		.p_encode	= (kxdrproc_t)xdr_enc_mon,
		.p_decode	= (kxdrproc_t)xdr_dec_stat_res,
480 481
		.p_arglen	= SM_mon_sz,
		.p_replen	= SM_monres_sz,
482
		.p_statidx	= NSMPROC_MON,
483
		.p_name		= "MONITOR",
L
Linus Torvalds 已提交
484
	},
485 486
[NSMPROC_UNMON] = {
		.p_proc		= NSMPROC_UNMON,
487 488
		.p_encode	= (kxdrproc_t)xdr_enc_unmon,
		.p_decode	= (kxdrproc_t)xdr_dec_stat,
489 490
		.p_arglen	= SM_mon_id_sz,
		.p_replen	= SM_unmonres_sz,
491
		.p_statidx	= NSMPROC_UNMON,
492
		.p_name		= "UNMONITOR",
L
Linus Torvalds 已提交
493 494 495 496
	},
};

static struct rpc_version	nsm_version1 = {
497 498
		.number		= 1,
		.nrprocs	= ARRAY_SIZE(nsm_procedures),
L
Linus Torvalds 已提交
499 500 501 502 503 504 505 506 507 508 509
		.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",
510
		.number		= NSM_PROGRAM,
511
		.nrvers		= ARRAY_SIZE(nsm_version),
L
Linus Torvalds 已提交
512 513 514
		.version	= nsm_version,
		.stats		= &nsm_stats
};
反馈
建议
客服 返回
顶部