mon.c 14.3 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
struct nsm_args {
34
	struct nsm_private	*priv;
35 36 37 38 39 40 41 42 43 44 45 46
	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
	struct nsm_args args = {
104
		.priv		= &nsm->sm_priv,
105 106 107
		.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
static struct nsm_handle *nsm_lookup_hostname(const char *hostname,
					      const size_t len)
{
	struct nsm_handle *nsm;

	list_for_each_entry(nsm, &nsm_handles, sm_link)
		if (strlen(nsm->sm_name) == len &&
		    memcmp(nsm->sm_name, hostname, len) == 0)
			return nsm;
	return NULL;
}

216 217 218 219 220 221 222 223 224 225
static struct nsm_handle *nsm_lookup_addr(const struct sockaddr *sap)
{
	struct nsm_handle *nsm;

	list_for_each_entry(nsm, &nsm_handles, sm_link)
		if (nlm_cmp_addr(nsm_addr(nsm), sap))
			return nsm;
	return NULL;
}

226 227 228 229 230 231 232 233 234 235 236
static struct nsm_handle *nsm_lookup_priv(const struct nsm_private *priv)
{
	struct nsm_handle *nsm;

	list_for_each_entry(nsm, &nsm_handles, sm_link)
		if (memcmp(nsm->sm_priv.data, priv->data,
					sizeof(priv->data)) == 0)
			return nsm;
	return NULL;
}

237 238 239 240 241 242 243 244 245 246 247 248 249 250 251
/*
 * Construct a unique cookie to match this nsm_handle to this monitored
 * host.  It is passed to the local rpc.statd via NSMPROC_MON, and
 * returned via NLMPROC_SM_NOTIFY, in the "priv" field of these
 * requests.
 *
 * Linux provides the raw IP address of the monitored host,
 * left in network byte order.
 */
static void nsm_init_private(struct nsm_handle *nsm)
{
	__be32 *p = (__be32 *)&nsm->sm_priv.data;
	*p = nsm_addr_in(nsm)->sin_addr.s_addr;
}

252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275
static struct nsm_handle *nsm_create_handle(const struct sockaddr *sap,
					    const size_t salen,
					    const char *hostname,
					    const size_t hostname_len)
{
	struct nsm_handle *new;

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

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

	return new;
}

276
/**
277
 * nsm_get_handle - Find or create a cached nsm_handle
278 279 280 281 282
 * @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
 *
283
 * Behavior is modulated by the global nsm_use_hostnames variable.
284
 *
285 286 287 288
 * Returns a cached nsm_handle after bumping its ref count, or
 * 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.
289
 */
290 291 292
struct nsm_handle *nsm_get_handle(const struct sockaddr *sap,
				  const size_t salen, const char *hostname,
				  const size_t hostname_len)
293
{
294
	struct nsm_handle *cached, *new = NULL;
295 296 297 298 299 300 301 302 303 304 305 306

	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);
307 308 309 310 311 312 313 314 315 316 317 318 319 320 321

	if (nsm_use_hostnames && hostname != NULL)
		cached = nsm_lookup_hostname(hostname, hostname_len);
	else
		cached = nsm_lookup_addr(sap);

	if (cached != NULL) {
		atomic_inc(&cached->sm_count);
		spin_unlock(&nsm_lock);
		kfree(new);
		dprintk("lockd: found nsm_handle for %s (%s), "
				"cnt %d\n", cached->sm_name,
				cached->sm_addrbuf,
				atomic_read(&cached->sm_count));
		return cached;
322
	}
323 324 325 326

	if (new != NULL) {
		list_add(&new->sm_link, &nsm_handles);
		spin_unlock(&nsm_lock);
327
		dprintk("lockd: created nsm_handle for %s (%s)\n",
328 329
				new->sm_name, new->sm_addrbuf);
		return new;
330
	}
331

332 333
	spin_unlock(&nsm_lock);

334 335
	new = nsm_create_handle(sap, salen, hostname, hostname_len);
	if (unlikely(new == NULL))
336 337 338 339
		return NULL;
	goto retry;
}

340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380
/**
 * nsm_reboot_lookup - match NLMPROC_SM_NOTIFY arguments to an nsm_handle
 * @info: pointer to NLMPROC_SM_NOTIFY arguments
 *
 * Returns a matching nsm_handle if found in the nsm cache; the returned
 * nsm_handle's reference count is bumped and sm_monitored is cleared.
 * Otherwise returns NULL if some error occurred.
 */
struct nsm_handle *nsm_reboot_lookup(const struct nlm_reboot *info)
{
	struct nsm_handle *cached;

	spin_lock(&nsm_lock);

	if (nsm_use_hostnames && info->mon != NULL)
		cached = nsm_lookup_hostname(info->mon, info->len);
	else
		cached = nsm_lookup_priv(&info->priv);

	if (unlikely(cached == NULL)) {
		spin_unlock(&nsm_lock);
		dprintk("lockd: never saw rebooted peer '%.*s' before\n",
				info->len, info->mon);
		return cached;
	}

	atomic_inc(&cached->sm_count);
	spin_unlock(&nsm_lock);

	/*
	 * During subsequent lock activity, force a fresh
	 * notification to be set up for this host.
	 */
	cached->sm_monitored = 0;

	dprintk("lockd: host %s (%s) rebooted, cnt %d\n",
			cached->sm_name, cached->sm_addrbuf,
			atomic_read(&cached->sm_count));
	return cached;
}

381 382 383 384 385 386 387 388 389 390
/**
 * 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);
391 392
		dprintk("lockd: destroyed nsm_handle for %s (%s)\n",
				nsm->sm_name, nsm->sm_addrbuf);
393 394 395 396
		kfree(nsm);
	}
}

L
Linus Torvalds 已提交
397 398 399 400 401 402
/*
 * Create NSM client for the local host
 */
static struct rpc_clnt *
nsm_create(void)
{
403 404 405 406 407 408
	struct sockaddr_in	sin = {
		.sin_family	= AF_INET,
		.sin_addr.s_addr = htonl(INADDR_LOOPBACK),
		.sin_port	= 0,
	};
	struct rpc_create_args args = {
409
		.protocol	= XPRT_TRANSPORT_UDP,
410 411 412 413
		.address	= (struct sockaddr *)&sin,
		.addrsize	= sizeof(sin),
		.servername	= "localhost",
		.program	= &nsm_program,
414
		.version	= NSM_VERSION,
415 416 417 418
		.authflavor	= RPC_AUTH_NULL,
	};

	return rpc_create(&args);
L
Linus Torvalds 已提交
419 420 421 422
}

/*
 * XDR functions for NSM.
423 424 425
 *
 * See http://www.opengroup.org/ for details on the Network
 * Status Monitor wire protocol.
L
Linus Torvalds 已提交
426 427
 */

428
static int encode_nsm_string(struct xdr_stream *xdr, const char *string)
429
{
430 431 432 433 434 435 436 437 438 439
	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;
440 441
}

442 443 444
/*
 * "mon_name" specifies the host to be monitored.
 */
445
static int encode_mon_name(struct xdr_stream *xdr, const struct nsm_args *argp)
446
{
447
	return encode_nsm_string(xdr, argp->mon_name);
448 449
}

450 451 452
/*
 * The "my_id" argument specifies the hostname and RPC procedure
 * to be called when the status manager receives notification
453
 * (via the NLMPROC_SM_NOTIFY call) that the state of host "mon_name"
454 455
 * has changed.
 */
456
static int encode_my_id(struct xdr_stream *xdr, const struct nsm_args *argp)
457
{
458 459 460 461 462 463 464 465 466
	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;
467 468 469
	*p++ = htonl(argp->prog);
	*p++ = htonl(argp->vers);
	*p++ = htonl(argp->proc);
470
	return 0;
471 472
}

473 474
/*
 * The "mon_id" argument specifies the non-private arguments
475
 * of an NSMPROC_MON or NSMPROC_UNMON call.
476
 */
477
static int encode_mon_id(struct xdr_stream *xdr, const struct nsm_args *argp)
478
{
479
	int status;
480

481 482 483 484
	status = encode_mon_name(xdr, argp);
	if (unlikely(status != 0))
		return status;
	return encode_my_id(xdr, argp);
485 486
}

487 488
/*
 * The "priv" argument may contain private information required
489 490
 * by the NSMPROC_MON call. This information will be supplied in the
 * NLMPROC_SM_NOTIFY call.
491
 */
492
static int encode_priv(struct xdr_stream *xdr, const struct nsm_args *argp)
493
{
494 495 496 497 498
	__be32 *p;

	p = xdr_reserve_space(xdr, SM_PRIV_SIZE);
	if (unlikely(p == NULL))
		return -EIO;
499
	xdr_encode_opaque_fixed(p, argp->priv->data, SM_PRIV_SIZE);
500
	return 0;
501 502
}

503 504
static int xdr_enc_mon(struct rpc_rqst *req, __be32 *p,
		       const struct nsm_args *argp)
L
Linus Torvalds 已提交
505
{
506 507
	struct xdr_stream xdr;
	int status;
508

509 510 511 512 513
	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 已提交
514 515
}

516 517
static int xdr_enc_unmon(struct rpc_rqst *req, __be32 *p,
			 const struct nsm_args *argp)
L
Linus Torvalds 已提交
518
{
519 520 521 522
	struct xdr_stream xdr;

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

525 526
static int xdr_dec_stat_res(struct rpc_rqst *rqstp, __be32 *p,
			    struct nsm_res *resp)
L
Linus Torvalds 已提交
527
{
528 529 530 531 532 533
	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 已提交
534
	resp->status = ntohl(*p++);
535 536 537
	resp->state = ntohl(*p);

	dprintk("lockd: xdr_dec_stat_res status %d state %d\n",
L
Linus Torvalds 已提交
538 539 540 541
			resp->status, resp->state);
	return 0;
}

542 543
static int xdr_dec_stat(struct rpc_rqst *rqstp, __be32 *p,
			struct nsm_res *resp)
L
Linus Torvalds 已提交
544
{
545 546 547 548 549 550 551 552 553
	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 已提交
554 555 556 557
	return 0;
}

#define SM_my_name_sz	(1+XDR_QUADLEN(SM_MAXSTRLEN))
558 559 560
#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)
561 562
#define SM_priv_sz	(XDR_QUADLEN(SM_PRIV_SIZE))
#define SM_mon_sz	(SM_mon_id_sz+SM_priv_sz)
L
Linus Torvalds 已提交
563 564 565 566
#define SM_monres_sz	2
#define SM_unmonres_sz	1

static struct rpc_procinfo	nsm_procedures[] = {
567 568
[NSMPROC_MON] = {
		.p_proc		= NSMPROC_MON,
569 570
		.p_encode	= (kxdrproc_t)xdr_enc_mon,
		.p_decode	= (kxdrproc_t)xdr_dec_stat_res,
571 572
		.p_arglen	= SM_mon_sz,
		.p_replen	= SM_monres_sz,
573
		.p_statidx	= NSMPROC_MON,
574
		.p_name		= "MONITOR",
L
Linus Torvalds 已提交
575
	},
576 577
[NSMPROC_UNMON] = {
		.p_proc		= NSMPROC_UNMON,
578 579
		.p_encode	= (kxdrproc_t)xdr_enc_unmon,
		.p_decode	= (kxdrproc_t)xdr_dec_stat,
580 581
		.p_arglen	= SM_mon_id_sz,
		.p_replen	= SM_unmonres_sz,
582
		.p_statidx	= NSMPROC_UNMON,
583
		.p_name		= "UNMONITOR",
L
Linus Torvalds 已提交
584 585 586 587
	},
};

static struct rpc_version	nsm_version1 = {
588 589
		.number		= 1,
		.nrprocs	= ARRAY_SIZE(nsm_procedures),
L
Linus Torvalds 已提交
590 591 592 593 594 595 596 597 598 599 600
		.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",
601
		.number		= NSM_PROGRAM,
602
		.nrvers		= ARRAY_SIZE(nsm_version),
L
Linus Torvalds 已提交
603 604 605
		.version	= nsm_version,
		.stats		= &nsm_stats
};
新手
引导
客服 返回
顶部