- 2.6.17 port work build breaks, but the patch set is relativly stable
[linux-flexiantxendom0-3.2.10.git] / fs / lockd / svclock.c
index 4a6f55a..3ef7391 100644 (file)
 #define nlm_deadlock   nlm_lck_denied
 #endif
 
+static void nlmsvc_release_block(struct nlm_block *block);
 static void    nlmsvc_insert_block(struct nlm_block *block, unsigned long);
-static void    nlmsvc_remove_block(struct nlm_block *block);
+static int     nlmsvc_remove_block(struct nlm_block *block);
 
+static int nlmsvc_setgrantargs(struct nlm_rqst *call, struct nlm_lock *lock);
+static void nlmsvc_freegrantargs(struct nlm_rqst *call);
 static const struct rpc_call_ops nlmsvc_grant_ops;
 
 /*
  * The list of blocked locks to retry
  */
-static LIST_HEAD(nlm_blocked);
+static struct nlm_block *      nlm_blocked;
 
 /*
  * Insert a blocked lock into the global list
@@ -55,60 +58,72 @@ static LIST_HEAD(nlm_blocked);
 static void
 nlmsvc_insert_block(struct nlm_block *block, unsigned long when)
 {
-       struct nlm_block *b;
-       struct list_head *pos;
+       struct nlm_block **bp, *b;
 
        dprintk("lockd: nlmsvc_insert_block(%p, %ld)\n", block, when);
-       list_del_init(&block->b_list);
-
-       pos = &nlm_blocked;
+       kref_get(&block->b_count);
+       if (block->b_queued)
+               nlmsvc_remove_block(block);
+       bp = &nlm_blocked;
        if (when != NLM_NEVER) {
                if ((when += jiffies) == NLM_NEVER)
                        when ++;
-               list_for_each(pos, &nlm_blocked) {
-                       b = list_entry(pos, struct nlm_block, b_list);
-                       if (time_after(b->b_when,when) || b->b_when == NLM_NEVER)
-                               break;
-               }
-       }
+               while ((b = *bp) && time_before_eq(b->b_when,when) && b->b_when != NLM_NEVER)
+                       bp = &b->b_next;
+       } else
+               while ((b = *bp) != 0)
+                       bp = &b->b_next;
 
-       list_add_tail(&block->b_list, pos);
+       block->b_queued = 1;
        block->b_when = when;
+       block->b_next = b;
+       *bp = block;
 }
 
 /*
  * Remove a block from the global list
  */
-static inline void
+static int
 nlmsvc_remove_block(struct nlm_block *block)
 {
-       list_del_init(&block->b_list);
+       struct nlm_block **bp, *b;
+
+       if (!block->b_queued)
+               return 1;
+       for (bp = &nlm_blocked; (b = *bp) != 0; bp = &b->b_next) {
+               if (b == block) {
+                       *bp = block->b_next;
+                       block->b_queued = 0;
+                       nlmsvc_release_block(block);
+                       return 1;
+               }
+       }
+
+       return 0;
 }
 
 /*
- * Find a block for a given lock and optionally remove it from
- * the list.
+ * Find a block for a given lock
  */
 static struct nlm_block *
-nlmsvc_lookup_block(struct nlm_file *file, struct nlm_lock *lock, int remove)
+nlmsvc_lookup_block(struct nlm_file *file, struct nlm_lock *lock)
 {
-       struct nlm_block        *block;
+       struct nlm_block        **head, *block;
        struct file_lock        *fl;
 
        dprintk("lockd: nlmsvc_lookup_block f=%p pd=%d %Ld-%Ld ty=%d\n",
                                file, lock->fl.fl_pid,
                                (long long)lock->fl.fl_start,
                                (long long)lock->fl.fl_end, lock->fl.fl_type);
-
-       list_for_each_entry(block, &nlm_blocked, b_list) {
-               fl = &block->b_call.a_args.lock.fl;
+       for (head = &nlm_blocked; (block = *head) != 0; head = &block->b_next) {
+               fl = &block->b_call->a_args.lock.fl;
                dprintk("lockd: check f=%p pd=%d %Ld-%Ld ty=%d cookie=%s\n",
                                block->b_file, fl->fl_pid,
                                (long long)fl->fl_start,
                                (long long)fl->fl_end, fl->fl_type,
-                               nlmdbg_cookie2a(&block->b_call.a_args.cookie));
+                               nlmdbg_cookie2a(&block->b_call->a_args.cookie));
                if (block->b_file == file && nlm_compare_locks(fl, &lock->fl)) {
-                       list_del_init(&block->b_list);
+                       kref_get(&block->b_count);
                        return block;
                }
        }
@@ -129,19 +144,21 @@ static inline int nlm_cookie_match(struct nlm_cookie *a, struct nlm_cookie *b)
  * Find a block with a given NLM cookie.
  */
 static inline struct nlm_block *
-nlmsvc_find_block(struct nlm_cookie *cookie)
+nlmsvc_find_block(struct nlm_cookie *cookie,  struct sockaddr_in *sin)
 {
        struct nlm_block *block;
 
-       list_for_each_entry(block, &nlm_blocked, b_list)
-               if (nlm_cookie_match(&block->b_call.a_args.cookie, cookie)) {
-                       dprintk("nlmsvc_find_block(%s): block=%p\n",
-                               nlmdbg_cookie2a(cookie), block);
-                       return block;
-               }
+       for (block = nlm_blocked; block; block = block->b_next) {
+               dprintk("cookie: head of blocked queue %p, block %p\n", 
+                       nlm_blocked, block);
+               if (nlm_cookie_match(&block->b_call->a_args.cookie,cookie)
+                               && nlm_cmp_addr(sin, &block->b_host->h_addr))
+                       break;
+       }
 
-       dprintk("nlmsvc_find_block(%s): block=NULL\n", nlmdbg_cookie2a(cookie));
-       return NULL;
+       if (block != NULL)
+               kref_get(&block->b_count);
+       return block;
 }
 
 /*
@@ -153,11 +170,6 @@ nlmsvc_find_block(struct nlm_cookie *cookie)
  * request, but (as I found out later) that's because some implementations
  * do just this. Never mind the standards comittees, they support our
  * logging industries.
- *
- * 10 years later: I hope can can safely ignore these old and broken
- * clients by now. Let's fix this so we can uniquely identify an incoming
- * GRANTED_RES message by cookie, without having to rely on the client's IP
- * address. --okir
  */
 static inline struct nlm_block *
 nlmsvc_create_block(struct svc_rqst *rqstp, struct nlm_file *file,
@@ -165,28 +177,30 @@ nlmsvc_create_block(struct svc_rqst *rqstp, struct nlm_file *file,
 {
        struct nlm_block        *block;
        struct nlm_host         *host;
-       struct nlm_rqst         *call;
+       struct nlm_rqst         *call = NULL;
 
        /* Create host handle for callback */
-       host = nlmclnt_lookup_host(&rqstp->rq_addr,
-                               rqstp->rq_prot, rqstp->rq_vers,
-                               lock->caller, lock->len);
+       host = nlmsvc_lookup_host(rqstp);
        if (host == NULL)
                return NULL;
 
+       call = nlm_alloc_call(host);
+       if (call == NULL)
+               return NULL;
+
        /* Allocate memory for block, and initialize arguments */
-       if (!(block = (struct nlm_block *) kzalloc(sizeof(*block), GFP_KERNEL)))
+       block = kzalloc(sizeof(*block), GFP_KERNEL);
+       if (block == NULL)
                goto failed;
-       locks_init_lock(&block->b_call.a_args.lock.fl);
-       locks_init_lock(&block->b_call.a_res.lock.fl);
-       INIT_LIST_HEAD(&block->b_list);
+       kref_init(&block->b_count);
 
-       if (!nlmclnt_setgrantargs(&block->b_call, lock))
+       if (!nlmsvc_setgrantargs(call, lock))
                goto failed_free;
 
        /* Set notifier function for VFS, and init args */
-       block->b_call.a_args.lock.fl.fl_lmops = &nlmsvc_lock_operations;
-       nlmclnt_next_cookie(&block->b_call.a_args.cookie);
+       call->a_args.lock.fl.fl_flags |= FL_SLEEP;
+       call->a_args.lock.fl.fl_lmops = &nlmsvc_lock_operations;
+       call->a_args.cookie = *cookie;  /* see above */
 
        dprintk("lockd: created block %p...\n", block);
 
@@ -197,19 +211,20 @@ nlmsvc_create_block(struct svc_rqst *rqstp, struct nlm_file *file,
        file->f_count++;
 
        /* Add to file's list of blocks */
-       list_add(&block->b_flist, &file->f_blocks);
+       block->b_fnext  = file->f_blocks;
+       file->f_blocks  = block;
 
        /* Set up RPC arguments for callback */
-       call = &block->b_call;
-       call->a_host    = host;
+       block->b_call = call;
        call->a_flags   = RPC_TASK_ASYNC;
+       call->a_block = block;
 
        return block;
 
 failed_free:
        kfree(block);
 failed:
-       nlm_release_host(host);
+       nlm_release_call(call);
        return NULL;
 }
 
@@ -219,57 +234,120 @@ failed:
  * It is the caller's responsibility to check whether the file
  * can be closed hereafter.
  */
-static int
-nlmsvc_delete_block(struct nlm_block *block, int unlock)
+static int nlmsvc_unlink_block(struct nlm_block *block)
 {
-       struct file_lock        *fl = &block->b_call.a_args.lock.fl;
-       struct nlm_file         *file = block->b_file;
-       int status = 0;
-
-       dprintk("lockd: deleting block %p...\n", block);
+       int status;
+       dprintk("lockd: unlinking block %p...\n", block);
 
        /* Remove block from list */
+       status = posix_unblock_lock(block->b_file->f_file, &block->b_call->a_args.lock.fl);
        nlmsvc_remove_block(block);
-       if (unlock)
-               status = posix_unblock_lock(file->f_file, fl);
+       return status;
+}
 
-       /* If the block is in the middle of a GRANT callback,
-        * don't kill it yet. */
-       if (block->b_incall) {
-               nlmsvc_insert_block(block, NLM_NEVER);
-               block->b_done = 1;
-               return status;
-       }
+static void nlmsvc_free_block(struct kref *kref)
+{
+       struct nlm_block *block = container_of(kref, struct nlm_block, b_count);
+       struct nlm_file         *file = block->b_file;
+       struct nlm_block        **bp;
 
+       dprintk("lockd: freeing block %p...\n", block);
+
+       down(&file->f_sema);
        /* Remove block from file's list of blocks */
-       list_del_init(&block->b_flist);
+       for (bp = &file->f_blocks; *bp; bp = &(*bp)->b_fnext) {
+               if (*bp == block) {
+                       *bp = block->b_fnext;
+                       break;
+               }
+       }
+       up(&file->f_sema);
 
-       if (block->b_host)
-               nlm_release_host(block->b_host);
-       nlmclnt_freegrantargs(&block->b_call);
-       nlm_release_file(file);
+       nlmsvc_freegrantargs(block->b_call);
+       nlm_release_call(block->b_call);
+       nlm_release_file(block->b_file);
        kfree(block);
-       return status;
+}
+
+static void nlmsvc_release_block(struct nlm_block *block)
+{
+       if (block != NULL)
+               kref_put(&block->b_count, nlmsvc_free_block);
+}
+
+static void nlmsvc_act_mark(struct nlm_host *host, struct nlm_file *file)
+{
+       struct nlm_block *block;
+
+       down(&file->f_sema);
+       for (block = file->f_blocks; block != NULL; block = block->b_fnext)
+               block->b_host->h_inuse = 1;
+       up(&file->f_sema);
+}
+
+static void nlmsvc_act_unlock(struct nlm_host *host, struct nlm_file *file)
+{
+       struct nlm_block *block;
+
+restart:
+       down(&file->f_sema);
+       for (block = file->f_blocks; block != NULL; block = block->b_fnext) {
+               if (host != NULL && host != block->b_host)
+                       continue;
+               if (!block->b_queued)
+                       continue;
+               kref_get(&block->b_count);
+               up(&file->f_sema);
+               nlmsvc_unlink_block(block);
+               nlmsvc_release_block(block);
+               goto restart;
+       }
+       up(&file->f_sema);
 }
 
 /*
  * Loop over all blocks and perform the action specified.
  * (NLM_ACT_CHECK handled by nlmsvc_inspect_file).
  */
-int
-nlmsvc_traverse_blocks(struct nlm_host *host, struct nlm_file *file,
-                       nlm_host_visitor_t fn)
+void
+nlmsvc_traverse_blocks(struct nlm_host *host, struct nlm_file *file, int action)
 {
-       struct nlm_block        *block, *next;
-       /* XXX: Will everything get cleaned up if we don't unlock here? */
+       if (action == NLM_ACT_MARK)
+               nlmsvc_act_mark(host, file);
+       else
+               nlmsvc_act_unlock(host, file);
+}
 
-       down(&file->f_sema);
-       list_for_each_entry_safe(block, next, &file->f_blocks, b_flist) {
-               if (fn && fn(block->b_host, host))
-                       nlmsvc_delete_block(block, 1);
+/*
+ * Initialize arguments for GRANTED call. The nlm_rqst structure
+ * has been cleared already.
+ */
+static int nlmsvc_setgrantargs(struct nlm_rqst *call, struct nlm_lock *lock)
+{
+       locks_copy_lock(&call->a_args.lock.fl, &lock->fl);
+       memcpy(&call->a_args.lock.fh, &lock->fh, sizeof(call->a_args.lock.fh));
+       call->a_args.lock.caller = system_utsname.nodename;
+       call->a_args.lock.oh.len = lock->oh.len;
+
+       /* set default data area */
+       call->a_args.lock.oh.data = call->a_owner;
+       call->a_args.lock.svid = lock->fl.fl_pid;
+
+       if (lock->oh.len > NLMCLNT_OHSIZE) {
+               void *data = kmalloc(lock->oh.len, GFP_KERNEL);
+               if (!data)
+                       return 0;
+               call->a_args.lock.oh.data = (u8 *) data;
        }
-       up(&file->f_sema);
-       return 0;
+
+       memcpy(call->a_args.lock.oh.data, lock->oh.data, lock->oh.len);
+       return 1;
+}
+
+static void nlmsvc_freegrantargs(struct nlm_rqst *call)
+{
+       if (call->a_args.lock.oh.data != call->a_owner)
+               kfree(call->a_args.lock.oh.data);
 }
 
 /*
@@ -280,9 +358,9 @@ u32
 nlmsvc_lock(struct svc_rqst *rqstp, struct nlm_file *file,
                        struct nlm_lock *lock, int wait, struct nlm_cookie *cookie)
 {
-       struct file_lock        *conflock;
-       struct nlm_block        *block;
+       struct nlm_block        *block, *newblock = NULL;
        int                     error;
+       u32                     ret;
 
        dprintk("lockd: nlmsvc_lock(%s/%ld, ty=%d, pi=%d, %Ld-%Ld, bl=%d)\n",
                                file->f_file->f_dentry->d_inode->i_sb->s_id,
@@ -293,69 +371,65 @@ nlmsvc_lock(struct svc_rqst *rqstp, struct nlm_file *file,
                                wait);
 
 
-       /* Get existing block (in case client is busy-waiting) */
-       block = nlmsvc_lookup_block(file, lock, 0);
-
-       lock->fl.fl_flags |= FL_LOCKD;
-
+       lock->fl.fl_flags &= ~FL_SLEEP;
 again:
        /* Lock file against concurrent access */
        down(&file->f_sema);
+       /* Get existing block (in case client is busy-waiting) */
+       block = nlmsvc_lookup_block(file, lock);
+       if (block == NULL) {
+               if (newblock != NULL)
+                       lock = &newblock->b_call->a_args.lock;
+       } else
+               lock = &block->b_call->a_args.lock;
 
-       if (!(conflock = posix_test_lock(file->f_file, &lock->fl))) {
-               error = posix_lock_file(file->f_file, &lock->fl);
+       error = posix_lock_file(file->f_file, &lock->fl);
+       lock->fl.fl_flags &= ~FL_SLEEP;
 
-               if (block)
-                       nlmsvc_delete_block(block, 0);
-               up(&file->f_sema);
+       dprintk("lockd: posix_lock_file returned %d\n", error);
 
-               dprintk("lockd: posix_lock_file returned %d\n", -error);
-               switch(-error) {
+       switch(error) {
                case 0:
-                       return nlm_granted;
-               case EDEADLK:
-                       return nlm_deadlock;
-               case EAGAIN:
-                       return nlm_lck_denied;
+                       ret = nlm_granted;
+                       goto out;
+               case -EAGAIN:
+                       break;
+               case -EDEADLK:
+                       ret = nlm_deadlock;
+                       goto out;
                default:                        /* includes ENOLCK */
-                       return nlm_lck_denied_nolocks;
-               }
+                       ret = nlm_lck_denied_nolocks;
+                       goto out;
        }
 
-       if (!wait) {
-               up(&file->f_sema);
-               return nlm_lck_denied;
-       }
+       ret = nlm_lck_denied;
+       if (!wait)
+               goto out;
 
-       if (posix_locks_deadlock(&lock->fl, conflock)) {
-               up(&file->f_sema);
-               return nlm_deadlock;
-       }
+       ret = nlm_lck_blocked;
+       if (block != NULL)
+               goto out;
 
        /* If we don't have a block, create and initialize it. Then
         * retry because we may have slept in kmalloc. */
        /* We have to release f_sema as nlmsvc_create_block may try to
         * to claim it while doing host garbage collection */
-       if (block == NULL) {
+       if (newblock == NULL) {
                up(&file->f_sema);
                dprintk("lockd: blocking on this lock (allocating).\n");
-               if (!(block = nlmsvc_create_block(rqstp, file, lock, cookie)))
+               if (!(newblock = nlmsvc_create_block(rqstp, file, lock, cookie)))
                        return nlm_lck_denied_nolocks;
                goto again;
        }
 
        /* Append to list of blocked */
-       nlmsvc_insert_block(block, NLM_NEVER);
-
-       if (list_empty(&block->b_call.a_args.lock.fl.fl_block)) {
-               /* Now add block to block list of the conflicting lock
-                  if we haven't done so. */
-               dprintk("lockd: blocking on this lock.\n");
-               posix_block_lock(conflock, &block->b_call.a_args.lock.fl);
-       }
-
+       nlmsvc_insert_block(newblock, NLM_NEVER);
+out:
        up(&file->f_sema);
-       return nlm_lck_blocked;
+       nlmsvc_release_block(newblock);
+       nlmsvc_release_block(block);
+       dprintk("lockd: nlmsvc_lock returned %u\n", ret);
+       return ret;
 }
 
 /*
@@ -365,8 +439,6 @@ u32
 nlmsvc_testlock(struct nlm_file *file, struct nlm_lock *lock,
                                       struct nlm_lock *conflock)
 {
-       struct file_lock        *fl;
-
        dprintk("lockd: nlmsvc_testlock(%s/%ld, ty=%d, %Ld-%Ld)\n",
                                file->f_file->f_dentry->d_inode->i_sb->s_id,
                                file->f_file->f_dentry->d_inode->i_ino,
@@ -374,14 +446,14 @@ nlmsvc_testlock(struct nlm_file *file, struct nlm_lock *lock,
                                (long long)lock->fl.fl_start,
                                (long long)lock->fl.fl_end);
 
-       if ((fl = posix_test_lock(file->f_file, &lock->fl)) != NULL) {
+       if (posix_test_lock(file->f_file, &lock->fl, &conflock->fl)) {
                dprintk("lockd: conflicting lock(ty=%d, %Ld-%Ld)\n",
-                               fl->fl_type, (long long)fl->fl_start,
-                               (long long)fl->fl_end);
+                               conflock->fl.fl_type,
+                               (long long)conflock->fl.fl_start,
+                               (long long)conflock->fl.fl_end);
                conflock->caller = "somehost";  /* FIXME */
-               conflock->len = strlen(conflock->caller);
                conflock->oh.len = 0;           /* don't return OH info */
-               conflock->fl = *fl;
+               conflock->svid = conflock->fl.fl_pid;
                return nlm_lck_denied;
        }
 
@@ -437,9 +509,12 @@ nlmsvc_cancel_blocked(struct nlm_file *file, struct nlm_lock *lock)
                                (long long)lock->fl.fl_end);
 
        down(&file->f_sema);
-       if ((block = nlmsvc_lookup_block(file, lock, 1)) != NULL)
-               status = nlmsvc_delete_block(block, 1);
+       block = nlmsvc_lookup_block(file, lock);
        up(&file->f_sema);
+       if (block != NULL) {
+               status = nlmsvc_unlink_block(block);
+               nlmsvc_release_block(block);
+       }
        return status ? nlm_lck_denied : nlm_granted;
 }
 
@@ -453,11 +528,11 @@ nlmsvc_cancel_blocked(struct nlm_file *file, struct nlm_lock *lock)
 static void
 nlmsvc_notify_blocked(struct file_lock *fl)
 {
-       struct nlm_block        *block;
+       struct nlm_block        **bp, *block;
 
        dprintk("lockd: VFS unblock notification for block %p\n", fl);
-       list_for_each_entry(block, &nlm_blocked, b_list) {
-               if (nlm_compare_locks(&block->b_call.a_args.lock.fl, fl)) {
+       for (bp = &nlm_blocked; (block = *bp) != 0; bp = &block->b_next) {
+               if (nlm_compare_locks(&block->b_call->a_args.lock.fl, fl)) {
                        nlmsvc_insert_block(block, 0);
                        svc_wake_up(block->b_daemon);
                        return;
@@ -492,17 +567,13 @@ static void
 nlmsvc_grant_blocked(struct nlm_block *block)
 {
        struct nlm_file         *file = block->b_file;
-       struct nlm_lock         *lock = &block->b_call.a_args.lock;
-       struct file_lock        *conflock;
+       struct nlm_lock         *lock = &block->b_call->a_args.lock;
        int                     error;
 
        dprintk("lockd: grant blocked lock %p\n", block);
 
-       /* First thing is lock the file */
-       down(&file->f_sema);
-
        /* Unlink block request from list */
-       nlmsvc_remove_block(block);
+       nlmsvc_unlink_block(block);
 
        /* If b_granted is true this means we've been here before.
         * Just retry the grant callback, possibly refreshing the RPC
@@ -513,24 +584,21 @@ nlmsvc_grant_blocked(struct nlm_block *block)
        }
 
        /* Try the lock operation again */
-       if ((conflock = posix_test_lock(file->f_file, &lock->fl)) != NULL) {
-               /* Bummer, we blocked again */
+       lock->fl.fl_flags |= FL_SLEEP;
+       error = posix_lock_file(file->f_file, &lock->fl);
+       lock->fl.fl_flags &= ~FL_SLEEP;
+
+       switch (error) {
+       case 0:
+               break;
+       case -EAGAIN:
                dprintk("lockd: lock still blocked\n");
                nlmsvc_insert_block(block, NLM_NEVER);
-               posix_block_lock(conflock, &lock->fl);
-               up(&file->f_sema);
                return;
-       }
-
-       /* Alright, no conflicting lock. Now lock it for real. If the
-        * following yields an error, this is most probably due to low
-        * memory. Retry the lock in a few seconds.
-        */
-       if ((error = posix_lock_file(file->f_file, &lock->fl)) < 0) {
+       default:
                printk(KERN_WARNING "lockd: unexpected error %d in %s!\n",
                                -error, __FUNCTION__);
                nlmsvc_insert_block(block, 10 * HZ);
-               up(&file->f_sema);
                return;
        }
 
@@ -538,17 +606,15 @@ callback:
        /* Lock was granted by VFS. */
        dprintk("lockd: GRANTing blocked lock.\n");
        block->b_granted = 1;
-       block->b_incall  = 1;
 
        /* Schedule next grant callback in 30 seconds */
        nlmsvc_insert_block(block, 30 * HZ);
 
        /* Call the client */
-       nlm_get_host(block->b_call.a_host);
-       if (nlmsvc_async_call(&block->b_call, NLMPROC_GRANTED_MSG,
+       kref_get(&block->b_count);
+       if (nlm_async_call(block->b_call, NLMPROC_GRANTED_MSG,
                                                &nlmsvc_grant_ops) < 0)
-               nlm_release_host(block->b_call.a_host);
-       up(&file->f_sema);
+               nlmsvc_release_block(block);
 }
 
 /*
@@ -562,14 +628,10 @@ callback:
 static void nlmsvc_grant_callback(struct rpc_task *task, void *data)
 {
        struct nlm_rqst         *call = data;
-       struct nlm_block        *block;
+       struct nlm_block        *block = call->a_block;
        unsigned long           timeout;
-       struct sockaddr_in      *peer_addr = RPC_PEERADDR(task->tk_client);
 
-       dprintk("lockd: GRANT_MSG RPC callback from %u.%u.%u.%u\n",
-                       NIPQUAD(peer_addr->sin_addr.s_addr));
-       if (!(block = nlmsvc_find_block(&call->a_args.cookie)))
-               return;
+       dprintk("lockd: GRANT_MSG RPC callback\n");
 
        /* Technically, we should down the file semaphore here. Since we
         * move the block towards the head of the queue only, no harm
@@ -586,13 +648,18 @@ static void nlmsvc_grant_callback(struct rpc_task *task, void *data)
        }
        nlmsvc_insert_block(block, timeout);
        svc_wake_up(block->b_daemon);
-       block->b_incall = 0;
+}
 
-       nlm_release_host(call->a_host);
+static void nlmsvc_grant_release(void *data)
+{
+       struct nlm_rqst         *call = data;
+
+       nlmsvc_release_block(call->a_block);
 }
 
 static const struct rpc_call_ops nlmsvc_grant_ops = {
        .rpc_call_done = nlmsvc_grant_callback,
+       .rpc_release = nlmsvc_grant_release,
 };
 
 /*
@@ -605,31 +672,24 @@ nlmsvc_grant_reply(struct svc_rqst *rqstp, struct nlm_cookie *cookie, u32 status
        struct nlm_block        *block;
        struct nlm_file         *file;
 
-       dprintk("lockd: GRANT_RES RPC callback from %u.%u.%u.%u; status=%u\n",
-                       NIPQUAD(rqstp->rq_addr.sin_addr.s_addr), status);
-       if (!(block = nlmsvc_find_block(cookie)))
+       dprintk("grant_reply: looking for cookie %x, host (%08x), s=%d \n", 
+               *(unsigned int *)(cookie->data), 
+               ntohl(rqstp->rq_addr.sin_addr.s_addr), status);
+       if (!(block = nlmsvc_find_block(cookie, &rqstp->rq_addr)))
                return;
        file = block->b_file;
 
-       file->f_count++;
-       down(&file->f_sema);
-       block = nlmsvc_find_block(cookie);
        if (block) {
                if (status == NLM_LCK_DENIED_GRACE_PERIOD) {
                        /* Try again in a couple of seconds */
                        nlmsvc_insert_block(block, 10 * HZ);
-                       up(&file->f_sema);
                } else {
                        /* Lock is now held by client, or has been rejected.
                         * In both cases, the block should be removed. */
-                       up(&file->f_sema);
-                       if (status == NLM_LCK_GRANTED)
-                               nlmsvc_delete_block(block, 0);
-                       else
-                               nlmsvc_delete_block(block, 1);
+                       nlmsvc_unlink_block(block);
                }
        }
-       nlm_release_file(file);
+       nlmsvc_release_block(block);
 }
 
 /*
@@ -640,26 +700,28 @@ nlmsvc_grant_reply(struct svc_rqst *rqstp, struct nlm_cookie *cookie, u32 status
 unsigned long
 nlmsvc_retry_blocked(void)
 {
-       unsigned long   timeout = MAX_SCHEDULE_TIMEOUT;
-       struct nlm_block *block;
-
-       while (!list_empty(&nlm_blocked)) {
-               block = list_entry(nlm_blocked.next, struct nlm_block, b_list);
+       struct nlm_block        *block;
 
+       dprintk("nlmsvc_retry_blocked(%p, when=%ld)\n",
+                       nlm_blocked,
+                       nlm_blocked? nlm_blocked->b_when : 0);
+       while ((block = nlm_blocked) != 0) {
                if (block->b_when == NLM_NEVER)
                        break;
-               if (time_after(block->b_when,jiffies)) {
-                       timeout = block->b_when - jiffies;
+               if (time_after(block->b_when,jiffies))
                        break;
-               }
-
                dprintk("nlmsvc_retry_blocked(%p, when=%ld, done=%d)\n",
                        block, block->b_when, block->b_done);
+               kref_get(&block->b_count);
                if (block->b_done)
-                       nlmsvc_delete_block(block, 0);
+                       nlmsvc_unlink_block(block);
                else
                        nlmsvc_grant_blocked(block);
+               nlmsvc_release_block(block);
        }
 
-       return timeout;
+       if ((block = nlm_blocked) && block->b_when != NLM_NEVER)
+               return (block->b_when - jiffies);
+
+       return MAX_SCHEDULE_TIMEOUT;
 }