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_clear_port(struct sockaddr *sap)
45 switch (sap->sa_family) {
47 ((struct sockaddr_in *)sap)->sin_port = 0;
50 ((struct sockaddr_in6 *)sap)->sin6_port = 0;
55 static void nlm_display_address(const struct sockaddr *sap,
56 char *buf, const size_t len)
58 const struct sockaddr_in *sin = (struct sockaddr_in *)sap;
59 const struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)sap;
61 switch (sap->sa_family) {
63 snprintf(buf, len, "unspecified");
66 snprintf(buf, len, NIPQUAD_FMT, NIPQUAD(sin->sin_addr.s_addr));
69 if (ipv6_addr_v4mapped(&sin6->sin6_addr))
70 snprintf(buf, len, NIPQUAD_FMT,
71 NIPQUAD(sin6->sin6_addr.s6_addr32[3]));
73 snprintf(buf, len, NIP6_FMT, NIP6(sin6->sin6_addr));
76 snprintf(buf, len, "unsupported address family");
82 * Common host lookup routine for server & client
84 static struct nlm_host *nlm_lookup_host(int server,
85 const struct sockaddr_in *sin,
86 int proto, u32 version,
88 unsigned int hostname_len,
89 const struct sockaddr_in *ssin)
91 struct hlist_head *chain;
92 struct hlist_node *pos;
93 struct nlm_host *host;
94 struct nsm_handle *nsm = NULL;
97 dprintk("lockd: nlm_lookup_host(proto=%d, vers=%u,"
98 " my role is %s, hostname=%.*s)\n",
99 proto, version, server ? "server" : "client",
100 hostname_len, hostname ? hostname : "<none>");
102 hash = NLM_ADDRHASH(sin->sin_addr.s_addr);
104 /* Lock hash table */
105 mutex_lock(&nlm_host_mutex);
107 if (time_after_eq(jiffies, next_gc))
110 /* We may keep several nlm_host objects for a peer, because each
111 * nlm_host is identified by
112 * (address, protocol, version, server/client)
113 * We could probably simplify this a little by putting all those
114 * different NLM rpc_clients into one single nlm_host object.
115 * This would allow us to have one nlm_host per address.
117 chain = &nlm_hosts[hash];
118 hlist_for_each_entry(host, pos, chain, h_hash) {
119 if (!nlm_cmp_addr(&host->h_addr, sin))
122 /* See if we have an NSM handle for this client */
124 nsm = host->h_nsmhandle;
126 if (host->h_proto != proto)
128 if (host->h_version != version)
130 if (host->h_server != server)
132 if (!nlm_cmp_addr(&host->h_saddr, ssin))
135 /* Move to head of hash chain. */
136 hlist_del(&host->h_hash);
137 hlist_add_head(&host->h_hash, chain);
140 dprintk("lockd: nlm_lookup_host found host %s (%s)\n",
141 host->h_name, host->h_addrbuf);
146 * The host wasn't in our hash table. If we don't
147 * have an NSM handle for it yet, create one.
150 atomic_inc(&nsm->sm_count);
153 nsm = nsm_find(sin, hostname, hostname_len);
155 dprintk("lockd: nlm_lookup_host failed; "
161 host = kzalloc(sizeof(*host), GFP_KERNEL);
164 dprintk("lockd: nlm_lookup_host failed; no memory\n");
167 host->h_name = nsm->sm_name;
169 nlm_clear_port((struct sockaddr *)&host->h_addr);
170 host->h_saddr = *ssin;
171 host->h_version = version;
172 host->h_proto = proto;
173 host->h_rpcclnt = NULL;
174 mutex_init(&host->h_mutex);
175 host->h_nextrebind = jiffies + NLM_HOST_REBIND;
176 host->h_expires = jiffies + NLM_HOST_EXPIRE;
177 atomic_set(&host->h_count, 1);
178 init_waitqueue_head(&host->h_gracewait);
179 init_rwsem(&host->h_rwsem);
180 host->h_state = 0; /* pseudo NSM state */
181 host->h_nsmstate = 0; /* real NSM state */
182 host->h_nsmhandle = nsm;
183 host->h_server = server;
184 hlist_add_head(&host->h_hash, chain);
185 INIT_LIST_HEAD(&host->h_lockowners);
186 spin_lock_init(&host->h_lock);
187 INIT_LIST_HEAD(&host->h_granted);
188 INIT_LIST_HEAD(&host->h_reclaim);
192 nlm_display_address((struct sockaddr *)&host->h_addr,
193 host->h_addrbuf, sizeof(host->h_addrbuf));
194 nlm_display_address((struct sockaddr *)&host->h_saddr,
195 host->h_saddrbuf, sizeof(host->h_saddrbuf));
197 dprintk("lockd: nlm_lookup_host created host %s\n",
201 mutex_unlock(&nlm_host_mutex);
209 nlm_destroy_host(struct nlm_host *host)
211 struct rpc_clnt *clnt;
213 BUG_ON(!list_empty(&host->h_lockowners));
214 BUG_ON(atomic_read(&host->h_count));
217 * Release NSM handle and unmonitor host.
221 clnt = host->h_rpcclnt;
223 rpc_shutdown_client(clnt);
228 * Find an NLM server handle in the cache. If there is none, create it.
230 struct nlm_host *nlmclnt_lookup_host(const struct sockaddr_in *sin,
231 int proto, u32 version,
232 const char *hostname,
233 unsigned int hostname_len)
235 const struct sockaddr_in source = {
236 .sin_family = AF_UNSPEC,
239 return nlm_lookup_host(0, sin, proto, version,
240 hostname, hostname_len, &source);
244 * Find an NLM client handle in the cache. If there is none, create it.
247 nlmsvc_lookup_host(struct svc_rqst *rqstp,
248 const char *hostname, unsigned int hostname_len)
250 const struct sockaddr_in source = {
251 .sin_family = AF_INET,
252 .sin_addr = rqstp->rq_daddr.addr,
255 return nlm_lookup_host(1, svc_addr_in(rqstp),
256 rqstp->rq_prot, rqstp->rq_vers,
257 hostname, hostname_len, &source);
261 * Create the NLM RPC client for an NLM peer
264 nlm_bind_host(struct nlm_host *host)
266 struct rpc_clnt *clnt;
268 dprintk("lockd: nlm_bind_host %s (%s), my addr=%s\n",
269 host->h_name, host->h_addrbuf, host->h_saddrbuf);
271 /* Lock host handle */
272 mutex_lock(&host->h_mutex);
274 /* If we've already created an RPC client, check whether
275 * RPC rebind is required
277 if ((clnt = host->h_rpcclnt) != NULL) {
278 if (time_after_eq(jiffies, host->h_nextrebind)) {
279 rpc_force_rebind(clnt);
280 host->h_nextrebind = jiffies + NLM_HOST_REBIND;
281 dprintk("lockd: next rebind in %lu jiffies\n",
282 host->h_nextrebind - jiffies);
285 unsigned long increment = nlmsvc_timeout;
286 struct rpc_timeout timeparms = {
287 .to_initval = increment,
288 .to_increment = increment,
289 .to_maxval = increment * 6UL,
292 struct rpc_create_args args = {
293 .protocol = host->h_proto,
294 .address = (struct sockaddr *)&host->h_addr,
295 .addrsize = sizeof(host->h_addr),
296 .saddress = (struct sockaddr *)&host->h_saddr,
297 .timeout = &timeparms,
298 .servername = host->h_name,
299 .program = &nlm_program,
300 .version = host->h_version,
301 .authflavor = RPC_AUTH_UNIX,
302 .flags = (RPC_CLNT_CREATE_NOPING |
303 RPC_CLNT_CREATE_AUTOBIND),
307 * lockd retries server side blocks automatically so we want
308 * those to be soft RPC calls. Client side calls need to be
312 args.flags |= RPC_CLNT_CREATE_HARDRTRY;
314 clnt = rpc_create(&args);
316 host->h_rpcclnt = clnt;
318 printk("lockd: couldn't create RPC handle for %s\n", host->h_name);
323 mutex_unlock(&host->h_mutex);
328 * Force a portmap lookup of the remote lockd port
331 nlm_rebind_host(struct nlm_host *host)
333 dprintk("lockd: rebind host %s\n", host->h_name);
334 if (host->h_rpcclnt && time_after_eq(jiffies, host->h_nextrebind)) {
335 rpc_force_rebind(host->h_rpcclnt);
336 host->h_nextrebind = jiffies + NLM_HOST_REBIND;
341 * Increment NLM host count
343 struct nlm_host * nlm_get_host(struct nlm_host *host)
346 dprintk("lockd: get host %s\n", host->h_name);
347 atomic_inc(&host->h_count);
348 host->h_expires = jiffies + NLM_HOST_EXPIRE;
354 * Release NLM host after use
356 void nlm_release_host(struct nlm_host *host)
359 dprintk("lockd: release host %s\n", host->h_name);
360 BUG_ON(atomic_read(&host->h_count) < 0);
361 if (atomic_dec_and_test(&host->h_count)) {
362 BUG_ON(!list_empty(&host->h_lockowners));
363 BUG_ON(!list_empty(&host->h_granted));
364 BUG_ON(!list_empty(&host->h_reclaim));
370 * We were notified that the host indicated by address &sin
372 * Release all resources held by that peer.
374 void nlm_host_rebooted(const struct sockaddr_in *sin,
375 const char *hostname,
376 unsigned int hostname_len,
379 struct hlist_head *chain;
380 struct hlist_node *pos;
381 struct nsm_handle *nsm;
382 struct nlm_host *host;
384 /* Find the NSM handle for this peer */
385 nsm = __nsm_find(sin, hostname, hostname_len, 0);
387 dprintk("lockd: never saw rebooted peer '%.*s' before\n",
388 hostname_len, hostname);
392 dprintk("lockd: nlm_host_rebooted(%.*s, %s)\n",
393 hostname_len, hostname, nsm->sm_addrbuf);
395 /* When reclaiming locks on this peer, make sure that
396 * we set up a new notification */
397 nsm->sm_monitored = 0;
399 /* Mark all hosts tied to this NSM state as having rebooted.
400 * We run the loop repeatedly, because we drop the host table
402 * To avoid processing a host several times, we match the nsmstate.
404 again: mutex_lock(&nlm_host_mutex);
405 for (chain = nlm_hosts; chain < nlm_hosts + NLM_HOST_NRHASH; ++chain) {
406 hlist_for_each_entry(host, pos, chain, h_hash) {
407 if (host->h_nsmhandle == nsm
408 && host->h_nsmstate != new_state) {
409 host->h_nsmstate = new_state;
413 mutex_unlock(&nlm_host_mutex);
415 if (host->h_server) {
416 /* We're server for this guy, just ditch
417 * all the locks he held. */
418 nlmsvc_free_host_resources(host);
420 /* He's the server, initiate lock recovery. */
421 nlmclnt_recovery(host);
424 nlm_release_host(host);
430 mutex_unlock(&nlm_host_mutex);
434 * Shut down the hosts module.
435 * Note that this routine is called only at server shutdown time.
438 nlm_shutdown_hosts(void)
440 struct hlist_head *chain;
441 struct hlist_node *pos;
442 struct nlm_host *host;
444 dprintk("lockd: shutting down host module\n");
445 mutex_lock(&nlm_host_mutex);
447 /* First, make all hosts eligible for gc */
448 dprintk("lockd: nuking all hosts...\n");
449 for (chain = nlm_hosts; chain < nlm_hosts + NLM_HOST_NRHASH; ++chain) {
450 hlist_for_each_entry(host, pos, chain, h_hash) {
451 host->h_expires = jiffies - 1;
452 if (host->h_rpcclnt) {
453 rpc_shutdown_client(host->h_rpcclnt);
454 host->h_rpcclnt = NULL;
459 /* Then, perform a garbage collection pass */
461 mutex_unlock(&nlm_host_mutex);
463 /* complain if any hosts are left */
465 printk(KERN_WARNING "lockd: couldn't shutdown host module!\n");
466 dprintk("lockd: %d hosts left:\n", nrhosts);
467 for (chain = nlm_hosts; chain < nlm_hosts + NLM_HOST_NRHASH; ++chain) {
468 hlist_for_each_entry(host, pos, chain, h_hash) {
469 dprintk(" %s (cnt %d use %d exp %ld)\n",
470 host->h_name, atomic_read(&host->h_count),
471 host->h_inuse, host->h_expires);
478 * Garbage collect any unused NLM hosts.
479 * This GC combines reference counting for async operations with
480 * mark & sweep for resources held by remote clients.
485 struct hlist_head *chain;
486 struct hlist_node *pos, *next;
487 struct nlm_host *host;
489 dprintk("lockd: host garbage collection\n");
490 for (chain = nlm_hosts; chain < nlm_hosts + NLM_HOST_NRHASH; ++chain) {
491 hlist_for_each_entry(host, pos, chain, h_hash)
495 /* Mark all hosts that hold locks, blocks or shares */
496 nlmsvc_mark_resources();
498 for (chain = nlm_hosts; chain < nlm_hosts + NLM_HOST_NRHASH; ++chain) {
499 hlist_for_each_entry_safe(host, pos, next, chain, h_hash) {
500 if (atomic_read(&host->h_count) || host->h_inuse
501 || time_before(jiffies, host->h_expires)) {
502 dprintk("nlm_gc_hosts skipping %s (cnt %d use %d exp %ld)\n",
503 host->h_name, atomic_read(&host->h_count),
504 host->h_inuse, host->h_expires);
507 dprintk("lockd: delete host %s\n", host->h_name);
508 hlist_del_init(&host->h_hash);
510 nlm_destroy_host(host);
515 next_gc = jiffies + NLM_HOST_COLLECT;
522 static LIST_HEAD(nsm_handles);
523 static DEFINE_SPINLOCK(nsm_lock);
525 static struct nsm_handle *
526 __nsm_find(const struct sockaddr_in *sin,
527 const char *hostname, unsigned int hostname_len,
530 struct nsm_handle *nsm = NULL;
531 struct nsm_handle *pos;
536 if (hostname && memchr(hostname, '/', hostname_len) != NULL) {
537 if (printk_ratelimit()) {
538 printk(KERN_WARNING "Invalid hostname \"%.*s\" "
539 "in NFS lock request\n",
540 hostname_len, hostname);
546 spin_lock(&nsm_lock);
547 list_for_each_entry(pos, &nsm_handles, sm_link) {
549 if (hostname && nsm_use_hostnames) {
550 if (strlen(pos->sm_name) != hostname_len
551 || memcmp(pos->sm_name, hostname, hostname_len))
553 } else if (!nlm_cmp_addr(&pos->sm_addr, sin))
555 atomic_inc(&pos->sm_count);
561 list_add(&nsm->sm_link, &nsm_handles);
564 spin_unlock(&nsm_lock);
569 nsm = kzalloc(sizeof(*nsm) + hostname_len + 1, GFP_KERNEL);
574 nsm->sm_name = (char *) (nsm + 1);
575 memcpy(nsm->sm_name, hostname, hostname_len);
576 nsm->sm_name[hostname_len] = '\0';
577 nlm_display_address((struct sockaddr *)&nsm->sm_addr,
578 nsm->sm_addrbuf, sizeof(nsm->sm_addrbuf));
579 atomic_set(&nsm->sm_count, 1);
583 spin_unlock(&nsm_lock);
587 static struct nsm_handle *
588 nsm_find(const struct sockaddr_in *sin, const char *hostname,
589 unsigned int hostname_len)
591 return __nsm_find(sin, hostname, hostname_len, 1);
595 * Release an NSM handle
598 nsm_release(struct nsm_handle *nsm)
602 if (atomic_dec_and_lock(&nsm->sm_count, &nsm_lock)) {
603 list_del(&nsm->sm_link);
604 spin_unlock(&nsm_lock);