Merge tag 'ntb-5.20' of https://github.com/jonmason/ntb
[linux-2.6-microblaze.git] / fs / lockd / svclock.c
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * linux/fs/lockd/svclock.c
4  *
5  * Handling of server-side locks, mostly of the blocked variety.
6  * This is the ugliest part of lockd because we tread on very thin ice.
7  * GRANT and CANCEL calls may get stuck, meet in mid-flight, etc.
8  * IMNSHO introducing the grant callback into the NLM protocol was one
9  * of the worst ideas Sun ever had. Except maybe for the idea of doing
10  * NFS file locking at all.
11  *
12  * I'm trying hard to avoid race conditions by protecting most accesses
13  * to a file's list of blocked locks through a semaphore. The global
14  * list of blocked locks is not protected in this fashion however.
15  * Therefore, some functions (such as the RPC callback for the async grant
16  * call) move blocked locks towards the head of the list *while some other
17  * process might be traversing it*. This should not be a problem in
18  * practice, because this will only cause functions traversing the list
19  * to visit some blocks twice.
20  *
21  * Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de>
22  */
23
24 #include <linux/types.h>
25 #include <linux/slab.h>
26 #include <linux/errno.h>
27 #include <linux/kernel.h>
28 #include <linux/sched.h>
29 #include <linux/sunrpc/clnt.h>
30 #include <linux/sunrpc/svc_xprt.h>
31 #include <linux/lockd/nlm.h>
32 #include <linux/lockd/lockd.h>
33 #include <linux/kthread.h>
34 #include <linux/exportfs.h>
35
36 #define NLMDBG_FACILITY         NLMDBG_SVCLOCK
37
38 #ifdef CONFIG_LOCKD_V4
39 #define nlm_deadlock    nlm4_deadlock
40 #else
41 #define nlm_deadlock    nlm_lck_denied
42 #endif
43
44 static void nlmsvc_release_block(struct nlm_block *block);
45 static void     nlmsvc_insert_block(struct nlm_block *block, unsigned long);
46 static void     nlmsvc_remove_block(struct nlm_block *block);
47
48 static int nlmsvc_setgrantargs(struct nlm_rqst *call, struct nlm_lock *lock);
49 static void nlmsvc_freegrantargs(struct nlm_rqst *call);
50 static const struct rpc_call_ops nlmsvc_grant_ops;
51
52 /*
53  * The list of blocked locks to retry
54  */
55 static LIST_HEAD(nlm_blocked);
56 static DEFINE_SPINLOCK(nlm_blocked_lock);
57
58 #if IS_ENABLED(CONFIG_SUNRPC_DEBUG)
59 static const char *nlmdbg_cookie2a(const struct nlm_cookie *cookie)
60 {
61         /*
62          * We can get away with a static buffer because this is only called
63          * from lockd, which is single-threaded.
64          */
65         static char buf[2*NLM_MAXCOOKIELEN+1];
66         unsigned int i, len = sizeof(buf);
67         char *p = buf;
68
69         len--;  /* allow for trailing \0 */
70         if (len < 3)
71                 return "???";
72         for (i = 0 ; i < cookie->len ; i++) {
73                 if (len < 2) {
74                         strcpy(p-3, "...");
75                         break;
76                 }
77                 sprintf(p, "%02x", cookie->data[i]);
78                 p += 2;
79                 len -= 2;
80         }
81         *p = '\0';
82
83         return buf;
84 }
85 #endif
86
87 /*
88  * Insert a blocked lock into the global list
89  */
90 static void
91 nlmsvc_insert_block_locked(struct nlm_block *block, unsigned long when)
92 {
93         struct nlm_block *b;
94         struct list_head *pos;
95
96         dprintk("lockd: nlmsvc_insert_block(%p, %ld)\n", block, when);
97         if (list_empty(&block->b_list)) {
98                 kref_get(&block->b_count);
99         } else {
100                 list_del_init(&block->b_list);
101         }
102
103         pos = &nlm_blocked;
104         if (when != NLM_NEVER) {
105                 if ((when += jiffies) == NLM_NEVER)
106                         when ++;
107                 list_for_each(pos, &nlm_blocked) {
108                         b = list_entry(pos, struct nlm_block, b_list);
109                         if (time_after(b->b_when,when) || b->b_when == NLM_NEVER)
110                                 break;
111                 }
112                 /* On normal exit from the loop, pos == &nlm_blocked,
113                  * so we will be adding to the end of the list - good
114                  */
115         }
116
117         list_add_tail(&block->b_list, pos);
118         block->b_when = when;
119 }
120
121 static void nlmsvc_insert_block(struct nlm_block *block, unsigned long when)
122 {
123         spin_lock(&nlm_blocked_lock);
124         nlmsvc_insert_block_locked(block, when);
125         spin_unlock(&nlm_blocked_lock);
126 }
127
128 /*
129  * Remove a block from the global list
130  */
131 static inline void
132 nlmsvc_remove_block(struct nlm_block *block)
133 {
134         if (!list_empty(&block->b_list)) {
135                 spin_lock(&nlm_blocked_lock);
136                 list_del_init(&block->b_list);
137                 spin_unlock(&nlm_blocked_lock);
138                 nlmsvc_release_block(block);
139         }
140 }
141
142 /*
143  * Find a block for a given lock
144  */
145 static struct nlm_block *
146 nlmsvc_lookup_block(struct nlm_file *file, struct nlm_lock *lock)
147 {
148         struct nlm_block        *block;
149         struct file_lock        *fl;
150
151         dprintk("lockd: nlmsvc_lookup_block f=%p pd=%d %Ld-%Ld ty=%d\n",
152                                 file, lock->fl.fl_pid,
153                                 (long long)lock->fl.fl_start,
154                                 (long long)lock->fl.fl_end, lock->fl.fl_type);
155         list_for_each_entry(block, &nlm_blocked, b_list) {
156                 fl = &block->b_call->a_args.lock.fl;
157                 dprintk("lockd: check f=%p pd=%d %Ld-%Ld ty=%d cookie=%s\n",
158                                 block->b_file, fl->fl_pid,
159                                 (long long)fl->fl_start,
160                                 (long long)fl->fl_end, fl->fl_type,
161                                 nlmdbg_cookie2a(&block->b_call->a_args.cookie));
162                 if (block->b_file == file && nlm_compare_locks(fl, &lock->fl)) {
163                         kref_get(&block->b_count);
164                         return block;
165                 }
166         }
167
168         return NULL;
169 }
170
171 static inline int nlm_cookie_match(struct nlm_cookie *a, struct nlm_cookie *b)
172 {
173         if (a->len != b->len)
174                 return 0;
175         if (memcmp(a->data, b->data, a->len))
176                 return 0;
177         return 1;
178 }
179
180 /*
181  * Find a block with a given NLM cookie.
182  */
183 static inline struct nlm_block *
184 nlmsvc_find_block(struct nlm_cookie *cookie)
185 {
186         struct nlm_block *block;
187
188         list_for_each_entry(block, &nlm_blocked, b_list) {
189                 if (nlm_cookie_match(&block->b_call->a_args.cookie,cookie))
190                         goto found;
191         }
192
193         return NULL;
194
195 found:
196         dprintk("nlmsvc_find_block(%s): block=%p\n", nlmdbg_cookie2a(cookie), block);
197         kref_get(&block->b_count);
198         return block;
199 }
200
201 /*
202  * Create a block and initialize it.
203  *
204  * Note: we explicitly set the cookie of the grant reply to that of
205  * the blocked lock request. The spec explicitly mentions that the client
206  * should _not_ rely on the callback containing the same cookie as the
207  * request, but (as I found out later) that's because some implementations
208  * do just this. Never mind the standards comittees, they support our
209  * logging industries.
210  *
211  * 10 years later: I hope we can safely ignore these old and broken
212  * clients by now. Let's fix this so we can uniquely identify an incoming
213  * GRANTED_RES message by cookie, without having to rely on the client's IP
214  * address. --okir
215  */
216 static struct nlm_block *
217 nlmsvc_create_block(struct svc_rqst *rqstp, struct nlm_host *host,
218                     struct nlm_file *file, struct nlm_lock *lock,
219                     struct nlm_cookie *cookie)
220 {
221         struct nlm_block        *block;
222         struct nlm_rqst         *call = NULL;
223
224         call = nlm_alloc_call(host);
225         if (call == NULL)
226                 return NULL;
227
228         /* Allocate memory for block, and initialize arguments */
229         block = kzalloc(sizeof(*block), GFP_KERNEL);
230         if (block == NULL)
231                 goto failed;
232         kref_init(&block->b_count);
233         INIT_LIST_HEAD(&block->b_list);
234         INIT_LIST_HEAD(&block->b_flist);
235
236         if (!nlmsvc_setgrantargs(call, lock))
237                 goto failed_free;
238
239         /* Set notifier function for VFS, and init args */
240         call->a_args.lock.fl.fl_flags |= FL_SLEEP;
241         call->a_args.lock.fl.fl_lmops = &nlmsvc_lock_operations;
242         nlmclnt_next_cookie(&call->a_args.cookie);
243
244         dprintk("lockd: created block %p...\n", block);
245
246         /* Create and initialize the block */
247         block->b_daemon = rqstp->rq_server;
248         block->b_host   = host;
249         block->b_file   = file;
250         file->f_count++;
251
252         /* Add to file's list of blocks */
253         list_add(&block->b_flist, &file->f_blocks);
254
255         /* Set up RPC arguments for callback */
256         block->b_call = call;
257         call->a_flags   = RPC_TASK_ASYNC;
258         call->a_block = block;
259
260         return block;
261
262 failed_free:
263         kfree(block);
264 failed:
265         nlmsvc_release_call(call);
266         return NULL;
267 }
268
269 /*
270  * Delete a block.
271  * It is the caller's responsibility to check whether the file
272  * can be closed hereafter.
273  */
274 static int nlmsvc_unlink_block(struct nlm_block *block)
275 {
276         int status;
277         dprintk("lockd: unlinking block %p...\n", block);
278
279         /* Remove block from list */
280         status = locks_delete_block(&block->b_call->a_args.lock.fl);
281         nlmsvc_remove_block(block);
282         return status;
283 }
284
285 static void nlmsvc_free_block(struct kref *kref)
286 {
287         struct nlm_block *block = container_of(kref, struct nlm_block, b_count);
288         struct nlm_file         *file = block->b_file;
289
290         dprintk("lockd: freeing block %p...\n", block);
291
292         /* Remove block from file's list of blocks */
293         list_del_init(&block->b_flist);
294         mutex_unlock(&file->f_mutex);
295
296         nlmsvc_freegrantargs(block->b_call);
297         nlmsvc_release_call(block->b_call);
298         nlm_release_file(block->b_file);
299         kfree(block);
300 }
301
302 static void nlmsvc_release_block(struct nlm_block *block)
303 {
304         if (block != NULL)
305                 kref_put_mutex(&block->b_count, nlmsvc_free_block, &block->b_file->f_mutex);
306 }
307
308 /*
309  * Loop over all blocks and delete blocks held by
310  * a matching host.
311  */
312 void nlmsvc_traverse_blocks(struct nlm_host *host,
313                         struct nlm_file *file,
314                         nlm_host_match_fn_t match)
315 {
316         struct nlm_block *block, *next;
317
318 restart:
319         mutex_lock(&file->f_mutex);
320         list_for_each_entry_safe(block, next, &file->f_blocks, b_flist) {
321                 if (!match(block->b_host, host))
322                         continue;
323                 /* Do not destroy blocks that are not on
324                  * the global retry list - why? */
325                 if (list_empty(&block->b_list))
326                         continue;
327                 kref_get(&block->b_count);
328                 mutex_unlock(&file->f_mutex);
329                 nlmsvc_unlink_block(block);
330                 nlmsvc_release_block(block);
331                 goto restart;
332         }
333         mutex_unlock(&file->f_mutex);
334 }
335
336 static struct nlm_lockowner *
337 nlmsvc_get_lockowner(struct nlm_lockowner *lockowner)
338 {
339         refcount_inc(&lockowner->count);
340         return lockowner;
341 }
342
343 void nlmsvc_put_lockowner(struct nlm_lockowner *lockowner)
344 {
345         if (!refcount_dec_and_lock(&lockowner->count, &lockowner->host->h_lock))
346                 return;
347         list_del(&lockowner->list);
348         spin_unlock(&lockowner->host->h_lock);
349         nlmsvc_release_host(lockowner->host);
350         kfree(lockowner);
351 }
352
353 static struct nlm_lockowner *__nlmsvc_find_lockowner(struct nlm_host *host, pid_t pid)
354 {
355         struct nlm_lockowner *lockowner;
356         list_for_each_entry(lockowner, &host->h_lockowners, list) {
357                 if (lockowner->pid != pid)
358                         continue;
359                 return nlmsvc_get_lockowner(lockowner);
360         }
361         return NULL;
362 }
363
364 static struct nlm_lockowner *nlmsvc_find_lockowner(struct nlm_host *host, pid_t pid)
365 {
366         struct nlm_lockowner *res, *new = NULL;
367
368         spin_lock(&host->h_lock);
369         res = __nlmsvc_find_lockowner(host, pid);
370
371         if (res == NULL) {
372                 spin_unlock(&host->h_lock);
373                 new = kmalloc(sizeof(*res), GFP_KERNEL);
374                 spin_lock(&host->h_lock);
375                 res = __nlmsvc_find_lockowner(host, pid);
376                 if (res == NULL && new != NULL) {
377                         res = new;
378                         /* fs/locks.c will manage the refcount through lock_ops */
379                         refcount_set(&new->count, 1);
380                         new->pid = pid;
381                         new->host = nlm_get_host(host);
382                         list_add(&new->list, &host->h_lockowners);
383                         new = NULL;
384                 }
385         }
386
387         spin_unlock(&host->h_lock);
388         kfree(new);
389         return res;
390 }
391
392 void
393 nlmsvc_release_lockowner(struct nlm_lock *lock)
394 {
395         if (lock->fl.fl_owner)
396                 nlmsvc_put_lockowner(lock->fl.fl_owner);
397 }
398
399 void nlmsvc_locks_init_private(struct file_lock *fl, struct nlm_host *host,
400                                                 pid_t pid)
401 {
402         fl->fl_owner = nlmsvc_find_lockowner(host, pid);
403 }
404
405 /*
406  * Initialize arguments for GRANTED call. The nlm_rqst structure
407  * has been cleared already.
408  */
409 static int nlmsvc_setgrantargs(struct nlm_rqst *call, struct nlm_lock *lock)
410 {
411         locks_copy_lock(&call->a_args.lock.fl, &lock->fl);
412         memcpy(&call->a_args.lock.fh, &lock->fh, sizeof(call->a_args.lock.fh));
413         call->a_args.lock.caller = utsname()->nodename;
414         call->a_args.lock.oh.len = lock->oh.len;
415
416         /* set default data area */
417         call->a_args.lock.oh.data = call->a_owner;
418         call->a_args.lock.svid = ((struct nlm_lockowner *)lock->fl.fl_owner)->pid;
419
420         if (lock->oh.len > NLMCLNT_OHSIZE) {
421                 void *data = kmalloc(lock->oh.len, GFP_KERNEL);
422                 if (!data)
423                         return 0;
424                 call->a_args.lock.oh.data = (u8 *) data;
425         }
426
427         memcpy(call->a_args.lock.oh.data, lock->oh.data, lock->oh.len);
428         return 1;
429 }
430
431 static void nlmsvc_freegrantargs(struct nlm_rqst *call)
432 {
433         if (call->a_args.lock.oh.data != call->a_owner)
434                 kfree(call->a_args.lock.oh.data);
435
436         locks_release_private(&call->a_args.lock.fl);
437 }
438
439 /*
440  * Deferred lock request handling for non-blocking lock
441  */
442 static __be32
443 nlmsvc_defer_lock_rqst(struct svc_rqst *rqstp, struct nlm_block *block)
444 {
445         __be32 status = nlm_lck_denied_nolocks;
446
447         block->b_flags |= B_QUEUED;
448
449         nlmsvc_insert_block(block, NLM_TIMEOUT);
450
451         block->b_cache_req = &rqstp->rq_chandle;
452         if (rqstp->rq_chandle.defer) {
453                 block->b_deferred_req =
454                         rqstp->rq_chandle.defer(block->b_cache_req);
455                 if (block->b_deferred_req != NULL)
456                         status = nlm_drop_reply;
457         }
458         dprintk("lockd: nlmsvc_defer_lock_rqst block %p flags %d status %d\n",
459                 block, block->b_flags, ntohl(status));
460
461         return status;
462 }
463
464 /*
465  * Attempt to establish a lock, and if it can't be granted, block it
466  * if required.
467  */
468 __be32
469 nlmsvc_lock(struct svc_rqst *rqstp, struct nlm_file *file,
470             struct nlm_host *host, struct nlm_lock *lock, int wait,
471             struct nlm_cookie *cookie, int reclaim)
472 {
473 #if IS_ENABLED(CONFIG_SUNRPC_DEBUG)
474         struct inode            *inode = nlmsvc_file_inode(file);
475 #endif
476         struct nlm_block        *block = NULL;
477         int                     error;
478         int                     mode;
479         int                     async_block = 0;
480         __be32                  ret;
481
482         dprintk("lockd: nlmsvc_lock(%s/%ld, ty=%d, pi=%d, %Ld-%Ld, bl=%d)\n",
483                                 inode->i_sb->s_id, inode->i_ino,
484                                 lock->fl.fl_type, lock->fl.fl_pid,
485                                 (long long)lock->fl.fl_start,
486                                 (long long)lock->fl.fl_end,
487                                 wait);
488
489         if (nlmsvc_file_file(file)->f_op->lock) {
490                 async_block = wait;
491                 wait = 0;
492         }
493
494         /* Lock file against concurrent access */
495         mutex_lock(&file->f_mutex);
496         /* Get existing block (in case client is busy-waiting)
497          * or create new block
498          */
499         block = nlmsvc_lookup_block(file, lock);
500         if (block == NULL) {
501                 block = nlmsvc_create_block(rqstp, host, file, lock, cookie);
502                 ret = nlm_lck_denied_nolocks;
503                 if (block == NULL)
504                         goto out;
505                 lock = &block->b_call->a_args.lock;
506         } else
507                 lock->fl.fl_flags &= ~FL_SLEEP;
508
509         if (block->b_flags & B_QUEUED) {
510                 dprintk("lockd: nlmsvc_lock deferred block %p flags %d\n",
511                                                         block, block->b_flags);
512                 if (block->b_granted) {
513                         nlmsvc_unlink_block(block);
514                         ret = nlm_granted;
515                         goto out;
516                 }
517                 if (block->b_flags & B_TIMED_OUT) {
518                         nlmsvc_unlink_block(block);
519                         ret = nlm_lck_denied;
520                         goto out;
521                 }
522                 ret = nlm_drop_reply;
523                 goto out;
524         }
525
526         if (locks_in_grace(SVC_NET(rqstp)) && !reclaim) {
527                 ret = nlm_lck_denied_grace_period;
528                 goto out;
529         }
530         if (reclaim && !locks_in_grace(SVC_NET(rqstp))) {
531                 ret = nlm_lck_denied_grace_period;
532                 goto out;
533         }
534
535         if (!wait)
536                 lock->fl.fl_flags &= ~FL_SLEEP;
537         mode = lock_to_openmode(&lock->fl);
538         error = vfs_lock_file(file->f_file[mode], F_SETLK, &lock->fl, NULL);
539         lock->fl.fl_flags &= ~FL_SLEEP;
540
541         dprintk("lockd: vfs_lock_file returned %d\n", error);
542         switch (error) {
543                 case 0:
544                         ret = nlm_granted;
545                         goto out;
546                 case -EAGAIN:
547                         /*
548                          * If this is a blocking request for an
549                          * already pending lock request then we need
550                          * to put it back on lockd's block list
551                          */
552                         if (wait)
553                                 break;
554                         ret = async_block ? nlm_lck_blocked : nlm_lck_denied;
555                         goto out;
556                 case FILE_LOCK_DEFERRED:
557                         if (wait)
558                                 break;
559                         /* Filesystem lock operation is in progress
560                            Add it to the queue waiting for callback */
561                         ret = nlmsvc_defer_lock_rqst(rqstp, block);
562                         goto out;
563                 case -EDEADLK:
564                         ret = nlm_deadlock;
565                         goto out;
566                 default:                        /* includes ENOLCK */
567                         ret = nlm_lck_denied_nolocks;
568                         goto out;
569         }
570
571         ret = nlm_lck_blocked;
572
573         /* Append to list of blocked */
574         nlmsvc_insert_block(block, NLM_NEVER);
575 out:
576         mutex_unlock(&file->f_mutex);
577         nlmsvc_release_block(block);
578         dprintk("lockd: nlmsvc_lock returned %u\n", ret);
579         return ret;
580 }
581
582 /*
583  * Test for presence of a conflicting lock.
584  */
585 __be32
586 nlmsvc_testlock(struct svc_rqst *rqstp, struct nlm_file *file,
587                 struct nlm_host *host, struct nlm_lock *lock,
588                 struct nlm_lock *conflock, struct nlm_cookie *cookie)
589 {
590         int                     error;
591         int                     mode;
592         __be32                  ret;
593
594         dprintk("lockd: nlmsvc_testlock(%s/%ld, ty=%d, %Ld-%Ld)\n",
595                                 nlmsvc_file_inode(file)->i_sb->s_id,
596                                 nlmsvc_file_inode(file)->i_ino,
597                                 lock->fl.fl_type,
598                                 (long long)lock->fl.fl_start,
599                                 (long long)lock->fl.fl_end);
600
601         if (locks_in_grace(SVC_NET(rqstp))) {
602                 ret = nlm_lck_denied_grace_period;
603                 goto out;
604         }
605
606         mode = lock_to_openmode(&lock->fl);
607         error = vfs_test_lock(file->f_file[mode], &lock->fl);
608         if (error) {
609                 /* We can't currently deal with deferred test requests */
610                 if (error == FILE_LOCK_DEFERRED)
611                         WARN_ON_ONCE(1);
612
613                 ret = nlm_lck_denied_nolocks;
614                 goto out;
615         }
616
617         if (lock->fl.fl_type == F_UNLCK) {
618                 ret = nlm_granted;
619                 goto out;
620         }
621
622         dprintk("lockd: conflicting lock(ty=%d, %Ld-%Ld)\n",
623                 lock->fl.fl_type, (long long)lock->fl.fl_start,
624                 (long long)lock->fl.fl_end);
625         conflock->caller = "somehost";  /* FIXME */
626         conflock->len = strlen(conflock->caller);
627         conflock->oh.len = 0;           /* don't return OH info */
628         conflock->svid = lock->fl.fl_pid;
629         conflock->fl.fl_type = lock->fl.fl_type;
630         conflock->fl.fl_start = lock->fl.fl_start;
631         conflock->fl.fl_end = lock->fl.fl_end;
632         locks_release_private(&lock->fl);
633
634         ret = nlm_lck_denied;
635 out:
636         return ret;
637 }
638
639 /*
640  * Remove a lock.
641  * This implies a CANCEL call: We send a GRANT_MSG, the client replies
642  * with a GRANT_RES call which gets lost, and calls UNLOCK immediately
643  * afterwards. In this case the block will still be there, and hence
644  * must be removed.
645  */
646 __be32
647 nlmsvc_unlock(struct net *net, struct nlm_file *file, struct nlm_lock *lock)
648 {
649         int     error = 0;
650
651         dprintk("lockd: nlmsvc_unlock(%s/%ld, pi=%d, %Ld-%Ld)\n",
652                                 nlmsvc_file_inode(file)->i_sb->s_id,
653                                 nlmsvc_file_inode(file)->i_ino,
654                                 lock->fl.fl_pid,
655                                 (long long)lock->fl.fl_start,
656                                 (long long)lock->fl.fl_end);
657
658         /* First, cancel any lock that might be there */
659         nlmsvc_cancel_blocked(net, file, lock);
660
661         lock->fl.fl_type = F_UNLCK;
662         if (file->f_file[O_RDONLY])
663                 error = vfs_lock_file(file->f_file[O_RDONLY], F_SETLK,
664                                         &lock->fl, NULL);
665         if (file->f_file[O_WRONLY])
666                 error = vfs_lock_file(file->f_file[O_WRONLY], F_SETLK,
667                                         &lock->fl, NULL);
668
669         return (error < 0)? nlm_lck_denied_nolocks : nlm_granted;
670 }
671
672 /*
673  * Cancel a previously blocked request.
674  *
675  * A cancel request always overrides any grant that may currently
676  * be in progress.
677  * The calling procedure must check whether the file can be closed.
678  */
679 __be32
680 nlmsvc_cancel_blocked(struct net *net, struct nlm_file *file, struct nlm_lock *lock)
681 {
682         struct nlm_block        *block;
683         int status = 0;
684         int mode;
685
686         dprintk("lockd: nlmsvc_cancel(%s/%ld, pi=%d, %Ld-%Ld)\n",
687                                 nlmsvc_file_inode(file)->i_sb->s_id,
688                                 nlmsvc_file_inode(file)->i_ino,
689                                 lock->fl.fl_pid,
690                                 (long long)lock->fl.fl_start,
691                                 (long long)lock->fl.fl_end);
692
693         if (locks_in_grace(net))
694                 return nlm_lck_denied_grace_period;
695
696         mutex_lock(&file->f_mutex);
697         block = nlmsvc_lookup_block(file, lock);
698         mutex_unlock(&file->f_mutex);
699         if (block != NULL) {
700                 mode = lock_to_openmode(&lock->fl);
701                 vfs_cancel_lock(block->b_file->f_file[mode],
702                                 &block->b_call->a_args.lock.fl);
703                 status = nlmsvc_unlink_block(block);
704                 nlmsvc_release_block(block);
705         }
706         return status ? nlm_lck_denied : nlm_granted;
707 }
708
709 /*
710  * This is a callback from the filesystem for VFS file lock requests.
711  * It will be used if lm_grant is defined and the filesystem can not
712  * respond to the request immediately.
713  * For SETLK or SETLKW request it will get the local posix lock.
714  * In all cases it will move the block to the head of nlm_blocked q where
715  * nlmsvc_retry_blocked() can send back a reply for SETLKW or revisit the
716  * deferred rpc for GETLK and SETLK.
717  */
718 static void
719 nlmsvc_update_deferred_block(struct nlm_block *block, int result)
720 {
721         block->b_flags |= B_GOT_CALLBACK;
722         if (result == 0)
723                 block->b_granted = 1;
724         else
725                 block->b_flags |= B_TIMED_OUT;
726 }
727
728 static int nlmsvc_grant_deferred(struct file_lock *fl, int result)
729 {
730         struct nlm_block *block;
731         int rc = -ENOENT;
732
733         spin_lock(&nlm_blocked_lock);
734         list_for_each_entry(block, &nlm_blocked, b_list) {
735                 if (nlm_compare_locks(&block->b_call->a_args.lock.fl, fl)) {
736                         dprintk("lockd: nlmsvc_notify_blocked block %p flags %d\n",
737                                                         block, block->b_flags);
738                         if (block->b_flags & B_QUEUED) {
739                                 if (block->b_flags & B_TIMED_OUT) {
740                                         rc = -ENOLCK;
741                                         break;
742                                 }
743                                 nlmsvc_update_deferred_block(block, result);
744                         } else if (result == 0)
745                                 block->b_granted = 1;
746
747                         nlmsvc_insert_block_locked(block, 0);
748                         svc_wake_up(block->b_daemon);
749                         rc = 0;
750                         break;
751                 }
752         }
753         spin_unlock(&nlm_blocked_lock);
754         if (rc == -ENOENT)
755                 printk(KERN_WARNING "lockd: grant for unknown block\n");
756         return rc;
757 }
758
759 /*
760  * Unblock a blocked lock request. This is a callback invoked from the
761  * VFS layer when a lock on which we blocked is removed.
762  *
763  * This function doesn't grant the blocked lock instantly, but rather moves
764  * the block to the head of nlm_blocked where it can be picked up by lockd.
765  */
766 static void
767 nlmsvc_notify_blocked(struct file_lock *fl)
768 {
769         struct nlm_block        *block;
770
771         dprintk("lockd: VFS unblock notification for block %p\n", fl);
772         spin_lock(&nlm_blocked_lock);
773         list_for_each_entry(block, &nlm_blocked, b_list) {
774                 if (nlm_compare_locks(&block->b_call->a_args.lock.fl, fl)) {
775                         nlmsvc_insert_block_locked(block, 0);
776                         spin_unlock(&nlm_blocked_lock);
777                         svc_wake_up(block->b_daemon);
778                         return;
779                 }
780         }
781         spin_unlock(&nlm_blocked_lock);
782         printk(KERN_WARNING "lockd: notification for unknown block!\n");
783 }
784
785 static fl_owner_t nlmsvc_get_owner(fl_owner_t owner)
786 {
787         return nlmsvc_get_lockowner(owner);
788 }
789
790 static void nlmsvc_put_owner(fl_owner_t owner)
791 {
792         nlmsvc_put_lockowner(owner);
793 }
794
795 const struct lock_manager_operations nlmsvc_lock_operations = {
796         .lm_notify = nlmsvc_notify_blocked,
797         .lm_grant = nlmsvc_grant_deferred,
798         .lm_get_owner = nlmsvc_get_owner,
799         .lm_put_owner = nlmsvc_put_owner,
800 };
801
802 /*
803  * Try to claim a lock that was previously blocked.
804  *
805  * Note that we use both the RPC_GRANTED_MSG call _and_ an async
806  * RPC thread when notifying the client. This seems like overkill...
807  * Here's why:
808  *  -   we don't want to use a synchronous RPC thread, otherwise
809  *      we might find ourselves hanging on a dead portmapper.
810  *  -   Some lockd implementations (e.g. HP) don't react to
811  *      RPC_GRANTED calls; they seem to insist on RPC_GRANTED_MSG calls.
812  */
813 static void
814 nlmsvc_grant_blocked(struct nlm_block *block)
815 {
816         struct nlm_file         *file = block->b_file;
817         struct nlm_lock         *lock = &block->b_call->a_args.lock;
818         int                     mode;
819         int                     error;
820         loff_t                  fl_start, fl_end;
821
822         dprintk("lockd: grant blocked lock %p\n", block);
823
824         kref_get(&block->b_count);
825
826         /* Unlink block request from list */
827         nlmsvc_unlink_block(block);
828
829         /* If b_granted is true this means we've been here before.
830          * Just retry the grant callback, possibly refreshing the RPC
831          * binding */
832         if (block->b_granted) {
833                 nlm_rebind_host(block->b_host);
834                 goto callback;
835         }
836
837         /* Try the lock operation again */
838         /* vfs_lock_file() can mangle fl_start and fl_end, but we need
839          * them unchanged for the GRANT_MSG
840          */
841         lock->fl.fl_flags |= FL_SLEEP;
842         fl_start = lock->fl.fl_start;
843         fl_end = lock->fl.fl_end;
844         mode = lock_to_openmode(&lock->fl);
845         error = vfs_lock_file(file->f_file[mode], F_SETLK, &lock->fl, NULL);
846         lock->fl.fl_flags &= ~FL_SLEEP;
847         lock->fl.fl_start = fl_start;
848         lock->fl.fl_end = fl_end;
849
850         switch (error) {
851         case 0:
852                 break;
853         case FILE_LOCK_DEFERRED:
854                 dprintk("lockd: lock still blocked error %d\n", error);
855                 nlmsvc_insert_block(block, NLM_NEVER);
856                 nlmsvc_release_block(block);
857                 return;
858         default:
859                 printk(KERN_WARNING "lockd: unexpected error %d in %s!\n",
860                                 -error, __func__);
861                 nlmsvc_insert_block(block, 10 * HZ);
862                 nlmsvc_release_block(block);
863                 return;
864         }
865
866 callback:
867         /* Lock was granted by VFS. */
868         dprintk("lockd: GRANTing blocked lock.\n");
869         block->b_granted = 1;
870
871         /* keep block on the list, but don't reattempt until the RPC
872          * completes or the submission fails
873          */
874         nlmsvc_insert_block(block, NLM_NEVER);
875
876         /* Call the client -- use a soft RPC task since nlmsvc_retry_blocked
877          * will queue up a new one if this one times out
878          */
879         error = nlm_async_call(block->b_call, NLMPROC_GRANTED_MSG,
880                                 &nlmsvc_grant_ops);
881
882         /* RPC submission failed, wait a bit and retry */
883         if (error < 0)
884                 nlmsvc_insert_block(block, 10 * HZ);
885 }
886
887 /*
888  * This is the callback from the RPC layer when the NLM_GRANTED_MSG
889  * RPC call has succeeded or timed out.
890  * Like all RPC callbacks, it is invoked by the rpciod process, so it
891  * better not sleep. Therefore, we put the blocked lock on the nlm_blocked
892  * chain once more in order to have it removed by lockd itself (which can
893  * then sleep on the file semaphore without disrupting e.g. the nfs client).
894  */
895 static void nlmsvc_grant_callback(struct rpc_task *task, void *data)
896 {
897         struct nlm_rqst         *call = data;
898         struct nlm_block        *block = call->a_block;
899         unsigned long           timeout;
900
901         dprintk("lockd: GRANT_MSG RPC callback\n");
902
903         spin_lock(&nlm_blocked_lock);
904         /* if the block is not on a list at this point then it has
905          * been invalidated. Don't try to requeue it.
906          *
907          * FIXME: it's possible that the block is removed from the list
908          * after this check but before the nlmsvc_insert_block. In that
909          * case it will be added back. Perhaps we need better locking
910          * for nlm_blocked?
911          */
912         if (list_empty(&block->b_list))
913                 goto out;
914
915         /* Technically, we should down the file semaphore here. Since we
916          * move the block towards the head of the queue only, no harm
917          * can be done, though. */
918         if (task->tk_status < 0) {
919                 /* RPC error: Re-insert for retransmission */
920                 timeout = 10 * HZ;
921         } else {
922                 /* Call was successful, now wait for client callback */
923                 timeout = 60 * HZ;
924         }
925         nlmsvc_insert_block_locked(block, timeout);
926         svc_wake_up(block->b_daemon);
927 out:
928         spin_unlock(&nlm_blocked_lock);
929 }
930
931 /*
932  * FIXME: nlmsvc_release_block() grabs a mutex.  This is not allowed for an
933  * .rpc_release rpc_call_op
934  */
935 static void nlmsvc_grant_release(void *data)
936 {
937         struct nlm_rqst         *call = data;
938         nlmsvc_release_block(call->a_block);
939 }
940
941 static const struct rpc_call_ops nlmsvc_grant_ops = {
942         .rpc_call_done = nlmsvc_grant_callback,
943         .rpc_release = nlmsvc_grant_release,
944 };
945
946 /*
947  * We received a GRANT_RES callback. Try to find the corresponding
948  * block.
949  */
950 void
951 nlmsvc_grant_reply(struct nlm_cookie *cookie, __be32 status)
952 {
953         struct nlm_block        *block;
954
955         dprintk("grant_reply: looking for cookie %x, s=%d \n",
956                 *(unsigned int *)(cookie->data), status);
957         if (!(block = nlmsvc_find_block(cookie)))
958                 return;
959
960         if (status == nlm_lck_denied_grace_period) {
961                 /* Try again in a couple of seconds */
962                 nlmsvc_insert_block(block, 10 * HZ);
963         } else {
964                 /*
965                  * Lock is now held by client, or has been rejected.
966                  * In both cases, the block should be removed.
967                  */
968                 nlmsvc_unlink_block(block);
969         }
970         nlmsvc_release_block(block);
971 }
972
973 /* Helper function to handle retry of a deferred block.
974  * If it is a blocking lock, call grant_blocked.
975  * For a non-blocking lock or test lock, revisit the request.
976  */
977 static void
978 retry_deferred_block(struct nlm_block *block)
979 {
980         if (!(block->b_flags & B_GOT_CALLBACK))
981                 block->b_flags |= B_TIMED_OUT;
982         nlmsvc_insert_block(block, NLM_TIMEOUT);
983         dprintk("revisit block %p flags %d\n",  block, block->b_flags);
984         if (block->b_deferred_req) {
985                 block->b_deferred_req->revisit(block->b_deferred_req, 0);
986                 block->b_deferred_req = NULL;
987         }
988 }
989
990 /*
991  * Retry all blocked locks that have been notified. This is where lockd
992  * picks up locks that can be granted, or grant notifications that must
993  * be retransmitted.
994  */
995 unsigned long
996 nlmsvc_retry_blocked(void)
997 {
998         unsigned long   timeout = MAX_SCHEDULE_TIMEOUT;
999         struct nlm_block *block;
1000
1001         spin_lock(&nlm_blocked_lock);
1002         while (!list_empty(&nlm_blocked) && !kthread_should_stop()) {
1003                 block = list_entry(nlm_blocked.next, struct nlm_block, b_list);
1004
1005                 if (block->b_when == NLM_NEVER)
1006                         break;
1007                 if (time_after(block->b_when, jiffies)) {
1008                         timeout = block->b_when - jiffies;
1009                         break;
1010                 }
1011                 spin_unlock(&nlm_blocked_lock);
1012
1013                 dprintk("nlmsvc_retry_blocked(%p, when=%ld)\n",
1014                         block, block->b_when);
1015                 if (block->b_flags & B_QUEUED) {
1016                         dprintk("nlmsvc_retry_blocked delete block (%p, granted=%d, flags=%d)\n",
1017                                 block, block->b_granted, block->b_flags);
1018                         retry_deferred_block(block);
1019                 } else
1020                         nlmsvc_grant_blocked(block);
1021                 spin_lock(&nlm_blocked_lock);
1022         }
1023         spin_unlock(&nlm_blocked_lock);
1024
1025         return timeout;
1026 }