2 * linux/fs/lockd/host.c
4 * Management for NLM peer hosts. The nlm_host struct is shared
5 * between client and server implementation. The only reason to
6 * do so is to reduce code bloat.
8 * Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de>
11 #include <linux/types.h>
12 #include <linux/slab.h>
14 #include <linux/in6.h>
15 #include <linux/sunrpc/clnt.h>
16 #include <linux/sunrpc/svc.h>
17 #include <linux/lockd/lockd.h>
18 #include <linux/lockd/sm_inter.h>
19 #include <linux/mutex.h>
23 #define NLMDBG_FACILITY NLMDBG_HOSTCACHE
24 #define NLM_HOST_NRHASH 32
25 #define NLM_ADDRHASH(addr) (ntohl(addr) & (NLM_HOST_NRHASH-1))
26 #define NLM_HOST_REBIND (60 * HZ)
27 #define NLM_HOST_EXPIRE (300 * HZ)
28 #define NLM_HOST_COLLECT (120 * HZ)
30 static struct hlist_head nlm_hosts[NLM_HOST_NRHASH];
31 static unsigned long next_gc;
33 static DEFINE_MUTEX(nlm_host_mutex);
36 static void nlm_gc_hosts(void);
37 static struct nsm_handle * __nsm_find(const struct sockaddr_in *,
38 const char *, unsigned int, int);
39 static struct nsm_handle * nsm_find(const struct sockaddr_in *sin,
41 unsigned int hostname_len);
43 static void nlm_display_address(const struct sockaddr *sap,
44 char *buf, const size_t len)
46 const struct sockaddr_in *sin = (struct sockaddr_in *)sap;
47 const struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)sap;
49 switch (sap->sa_family) {
51 snprintf(buf, len, "unspecified");
54 snprintf(buf, len, NIPQUAD_FMT, NIPQUAD(sin->sin_addr.s_addr));
57 if (ipv6_addr_v4mapped(&sin6->sin6_addr))
58 snprintf(buf, len, NIPQUAD_FMT,
59 NIPQUAD(sin6->sin6_addr.s6_addr32[3]));
61 snprintf(buf, len, NIP6_FMT, NIP6(sin6->sin6_addr));
64 snprintf(buf, len, "unsupported address family");
70 * Common host lookup routine for server & client
72 static struct nlm_host *nlm_lookup_host(int server,
73 const struct sockaddr_in *sin,
74 int proto, u32 version,
76 unsigned int hostname_len,
77 const struct sockaddr_in *ssin)
79 struct hlist_head *chain;
80 struct hlist_node *pos;
81 struct nlm_host *host;
82 struct nsm_handle *nsm = NULL;
85 dprintk("lockd: nlm_lookup_host(proto=%d, vers=%u,"
86 " my role is %s, hostname=%.*s)\n",
87 proto, version, server ? "server" : "client",
88 hostname_len, hostname ? hostname : "<none>");
90 hash = NLM_ADDRHASH(sin->sin_addr.s_addr);
93 mutex_lock(&nlm_host_mutex);
95 if (time_after_eq(jiffies, next_gc))
98 /* We may keep several nlm_host objects for a peer, because each
99 * nlm_host is identified by
100 * (address, protocol, version, server/client)
101 * We could probably simplify this a little by putting all those
102 * different NLM rpc_clients into one single nlm_host object.
103 * This would allow us to have one nlm_host per address.
105 chain = &nlm_hosts[hash];
106 hlist_for_each_entry(host, pos, chain, h_hash) {
107 if (!nlm_cmp_addr(&host->h_addr, sin))
110 /* See if we have an NSM handle for this client */
112 nsm = host->h_nsmhandle;
114 if (host->h_proto != proto)
116 if (host->h_version != version)
118 if (host->h_server != server)
120 if (!nlm_cmp_addr(&host->h_saddr, ssin))
123 /* Move to head of hash chain. */
124 hlist_del(&host->h_hash);
125 hlist_add_head(&host->h_hash, chain);
128 dprintk("lockd: nlm_lookup_host found host %s (%s)\n",
129 host->h_name, host->h_addrbuf);
134 * The host wasn't in our hash table. If we don't
135 * have an NSM handle for it yet, create one.
138 atomic_inc(&nsm->sm_count);
141 nsm = nsm_find(sin, hostname, hostname_len);
143 dprintk("lockd: nlm_lookup_host failed; "
149 host = kzalloc(sizeof(*host), GFP_KERNEL);
152 dprintk("lockd: nlm_lookup_host failed; no memory\n");
155 host->h_name = nsm->sm_name;
157 host->h_addr.sin_port = 0; /* ouch! */
158 host->h_saddr = *ssin;
159 host->h_version = version;
160 host->h_proto = proto;
161 host->h_rpcclnt = NULL;
162 mutex_init(&host->h_mutex);
163 host->h_nextrebind = jiffies + NLM_HOST_REBIND;
164 host->h_expires = jiffies + NLM_HOST_EXPIRE;
165 atomic_set(&host->h_count, 1);
166 init_waitqueue_head(&host->h_gracewait);
167 init_rwsem(&host->h_rwsem);
168 host->h_state = 0; /* pseudo NSM state */
169 host->h_nsmstate = 0; /* real NSM state */
170 host->h_nsmhandle = nsm;
171 host->h_server = server;
172 hlist_add_head(&host->h_hash, chain);
173 INIT_LIST_HEAD(&host->h_lockowners);
174 spin_lock_init(&host->h_lock);
175 INIT_LIST_HEAD(&host->h_granted);
176 INIT_LIST_HEAD(&host->h_reclaim);
180 nlm_display_address((struct sockaddr *)&host->h_addr,
181 host->h_addrbuf, sizeof(host->h_addrbuf));
182 nlm_display_address((struct sockaddr *)&host->h_saddr,
183 host->h_saddrbuf, sizeof(host->h_saddrbuf));
185 dprintk("lockd: nlm_lookup_host created host %s\n",
189 mutex_unlock(&nlm_host_mutex);
197 nlm_destroy_host(struct nlm_host *host)
199 struct rpc_clnt *clnt;
201 BUG_ON(!list_empty(&host->h_lockowners));
202 BUG_ON(atomic_read(&host->h_count));
205 * Release NSM handle and unmonitor host.
209 clnt = host->h_rpcclnt;
211 rpc_shutdown_client(clnt);
216 * Find an NLM server handle in the cache. If there is none, create it.
218 struct nlm_host *nlmclnt_lookup_host(const struct sockaddr_in *sin,
219 int proto, u32 version,
220 const char *hostname,
221 unsigned int hostname_len)
223 const struct sockaddr_in source = {
224 .sin_family = AF_UNSPEC,
227 return nlm_lookup_host(0, sin, proto, version,
228 hostname, hostname_len, &source);
232 * Find an NLM client handle in the cache. If there is none, create it.
235 nlmsvc_lookup_host(struct svc_rqst *rqstp,
236 const char *hostname, unsigned int hostname_len)
238 const struct sockaddr_in source = {
239 .sin_family = AF_INET,
240 .sin_addr = rqstp->rq_daddr.addr,
243 return nlm_lookup_host(1, svc_addr_in(rqstp),
244 rqstp->rq_prot, rqstp->rq_vers,
245 hostname, hostname_len, &source);
249 * Create the NLM RPC client for an NLM peer
252 nlm_bind_host(struct nlm_host *host)
254 struct rpc_clnt *clnt;
256 dprintk("lockd: nlm_bind_host %s (%s), my addr=%s\n",
257 host->h_name, host->h_addrbuf, host->h_saddrbuf);
259 /* Lock host handle */
260 mutex_lock(&host->h_mutex);
262 /* If we've already created an RPC client, check whether
263 * RPC rebind is required
265 if ((clnt = host->h_rpcclnt) != NULL) {
266 if (time_after_eq(jiffies, host->h_nextrebind)) {
267 rpc_force_rebind(clnt);
268 host->h_nextrebind = jiffies + NLM_HOST_REBIND;
269 dprintk("lockd: next rebind in %lu jiffies\n",
270 host->h_nextrebind - jiffies);
273 unsigned long increment = nlmsvc_timeout;
274 struct rpc_timeout timeparms = {
275 .to_initval = increment,
276 .to_increment = increment,
277 .to_maxval = increment * 6UL,
280 struct rpc_create_args args = {
281 .protocol = host->h_proto,
282 .address = (struct sockaddr *)&host->h_addr,
283 .addrsize = sizeof(host->h_addr),
284 .saddress = (struct sockaddr *)&host->h_saddr,
285 .timeout = &timeparms,
286 .servername = host->h_name,
287 .program = &nlm_program,
288 .version = host->h_version,
289 .authflavor = RPC_AUTH_UNIX,
290 .flags = (RPC_CLNT_CREATE_NOPING |
291 RPC_CLNT_CREATE_AUTOBIND),
295 * lockd retries server side blocks automatically so we want
296 * those to be soft RPC calls. Client side calls need to be
300 args.flags |= RPC_CLNT_CREATE_HARDRTRY;
302 clnt = rpc_create(&args);
304 host->h_rpcclnt = clnt;
306 printk("lockd: couldn't create RPC handle for %s\n", host->h_name);
311 mutex_unlock(&host->h_mutex);
316 * Force a portmap lookup of the remote lockd port
319 nlm_rebind_host(struct nlm_host *host)
321 dprintk("lockd: rebind host %s\n", host->h_name);
322 if (host->h_rpcclnt && time_after_eq(jiffies, host->h_nextrebind)) {
323 rpc_force_rebind(host->h_rpcclnt);
324 host->h_nextrebind = jiffies + NLM_HOST_REBIND;
329 * Increment NLM host count
331 struct nlm_host * nlm_get_host(struct nlm_host *host)
334 dprintk("lockd: get host %s\n", host->h_name);
335 atomic_inc(&host->h_count);
336 host->h_expires = jiffies + NLM_HOST_EXPIRE;
342 * Release NLM host after use
344 void nlm_release_host(struct nlm_host *host)
347 dprintk("lockd: release host %s\n", host->h_name);
348 BUG_ON(atomic_read(&host->h_count) < 0);
349 if (atomic_dec_and_test(&host->h_count)) {
350 BUG_ON(!list_empty(&host->h_lockowners));
351 BUG_ON(!list_empty(&host->h_granted));
352 BUG_ON(!list_empty(&host->h_reclaim));
358 * We were notified that the host indicated by address &sin
360 * Release all resources held by that peer.
362 void nlm_host_rebooted(const struct sockaddr_in *sin,
363 const char *hostname,
364 unsigned int hostname_len,
367 struct hlist_head *chain;
368 struct hlist_node *pos;
369 struct nsm_handle *nsm;
370 struct nlm_host *host;
372 /* Find the NSM handle for this peer */
373 nsm = __nsm_find(sin, hostname, hostname_len, 0);
375 dprintk("lockd: never saw rebooted peer '%.*s' before\n",
376 hostname_len, hostname);
380 dprintk("lockd: nlm_host_rebooted(%.*s, %s)\n",
381 hostname_len, hostname, nsm->sm_addrbuf);
383 /* When reclaiming locks on this peer, make sure that
384 * we set up a new notification */
385 nsm->sm_monitored = 0;
387 /* Mark all hosts tied to this NSM state as having rebooted.
388 * We run the loop repeatedly, because we drop the host table
390 * To avoid processing a host several times, we match the nsmstate.
392 again: mutex_lock(&nlm_host_mutex);
393 for (chain = nlm_hosts; chain < nlm_hosts + NLM_HOST_NRHASH; ++chain) {
394 hlist_for_each_entry(host, pos, chain, h_hash) {
395 if (host->h_nsmhandle == nsm
396 && host->h_nsmstate != new_state) {
397 host->h_nsmstate = new_state;
401 mutex_unlock(&nlm_host_mutex);
403 if (host->h_server) {
404 /* We're server for this guy, just ditch
405 * all the locks he held. */
406 nlmsvc_free_host_resources(host);
408 /* He's the server, initiate lock recovery. */
409 nlmclnt_recovery(host);
412 nlm_release_host(host);
418 mutex_unlock(&nlm_host_mutex);
422 * Shut down the hosts module.
423 * Note that this routine is called only at server shutdown time.
426 nlm_shutdown_hosts(void)
428 struct hlist_head *chain;
429 struct hlist_node *pos;
430 struct nlm_host *host;
432 dprintk("lockd: shutting down host module\n");
433 mutex_lock(&nlm_host_mutex);
435 /* First, make all hosts eligible for gc */
436 dprintk("lockd: nuking all hosts...\n");
437 for (chain = nlm_hosts; chain < nlm_hosts + NLM_HOST_NRHASH; ++chain) {
438 hlist_for_each_entry(host, pos, chain, h_hash) {
439 host->h_expires = jiffies - 1;
440 if (host->h_rpcclnt) {
441 rpc_shutdown_client(host->h_rpcclnt);
442 host->h_rpcclnt = NULL;
447 /* Then, perform a garbage collection pass */
449 mutex_unlock(&nlm_host_mutex);
451 /* complain if any hosts are left */
453 printk(KERN_WARNING "lockd: couldn't shutdown host module!\n");
454 dprintk("lockd: %d hosts left:\n", nrhosts);
455 for (chain = nlm_hosts; chain < nlm_hosts + NLM_HOST_NRHASH; ++chain) {
456 hlist_for_each_entry(host, pos, chain, h_hash) {
457 dprintk(" %s (cnt %d use %d exp %ld)\n",
458 host->h_name, atomic_read(&host->h_count),
459 host->h_inuse, host->h_expires);
466 * Garbage collect any unused NLM hosts.
467 * This GC combines reference counting for async operations with
468 * mark & sweep for resources held by remote clients.
473 struct hlist_head *chain;
474 struct hlist_node *pos, *next;
475 struct nlm_host *host;
477 dprintk("lockd: host garbage collection\n");
478 for (chain = nlm_hosts; chain < nlm_hosts + NLM_HOST_NRHASH; ++chain) {
479 hlist_for_each_entry(host, pos, chain, h_hash)
483 /* Mark all hosts that hold locks, blocks or shares */
484 nlmsvc_mark_resources();
486 for (chain = nlm_hosts; chain < nlm_hosts + NLM_HOST_NRHASH; ++chain) {
487 hlist_for_each_entry_safe(host, pos, next, chain, h_hash) {
488 if (atomic_read(&host->h_count) || host->h_inuse
489 || time_before(jiffies, host->h_expires)) {
490 dprintk("nlm_gc_hosts skipping %s (cnt %d use %d exp %ld)\n",
491 host->h_name, atomic_read(&host->h_count),
492 host->h_inuse, host->h_expires);
495 dprintk("lockd: delete host %s\n", host->h_name);
496 hlist_del_init(&host->h_hash);
498 nlm_destroy_host(host);
503 next_gc = jiffies + NLM_HOST_COLLECT;
510 static LIST_HEAD(nsm_handles);
511 static DEFINE_SPINLOCK(nsm_lock);
513 static struct nsm_handle *
514 __nsm_find(const struct sockaddr_in *sin,
515 const char *hostname, unsigned int hostname_len,
518 struct nsm_handle *nsm = NULL;
519 struct nsm_handle *pos;
524 if (hostname && memchr(hostname, '/', hostname_len) != NULL) {
525 if (printk_ratelimit()) {
526 printk(KERN_WARNING "Invalid hostname \"%.*s\" "
527 "in NFS lock request\n",
528 hostname_len, hostname);
534 spin_lock(&nsm_lock);
535 list_for_each_entry(pos, &nsm_handles, sm_link) {
537 if (hostname && nsm_use_hostnames) {
538 if (strlen(pos->sm_name) != hostname_len
539 || memcmp(pos->sm_name, hostname, hostname_len))
541 } else if (!nlm_cmp_addr(&pos->sm_addr, sin))
543 atomic_inc(&pos->sm_count);
549 list_add(&nsm->sm_link, &nsm_handles);
552 spin_unlock(&nsm_lock);
557 nsm = kzalloc(sizeof(*nsm) + hostname_len + 1, GFP_KERNEL);
562 nsm->sm_name = (char *) (nsm + 1);
563 memcpy(nsm->sm_name, hostname, hostname_len);
564 nsm->sm_name[hostname_len] = '\0';
565 nlm_display_address((struct sockaddr *)&nsm->sm_addr,
566 nsm->sm_addrbuf, sizeof(nsm->sm_addrbuf));
567 atomic_set(&nsm->sm_count, 1);
571 spin_unlock(&nsm_lock);
575 static struct nsm_handle *
576 nsm_find(const struct sockaddr_in *sin, const char *hostname,
577 unsigned int hostname_len)
579 return __nsm_find(sin, hostname, hostname_len, 1);
583 * Release an NSM handle
586 nsm_release(struct nsm_handle *nsm)
590 if (atomic_dec_and_lock(&nsm->sm_count, &nsm_lock)) {
591 list_del(&nsm->sm_link);
592 spin_unlock(&nsm_lock);