clntproc.c 21.6 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9
/*
 * linux/fs/lockd/clntproc.c
 *
 * RPC procedures for the client side NLM implementation
 *
 * Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de>
 */

#include <linux/module.h>
10
#include <linux/slab.h>
L
Linus Torvalds 已提交
11 12 13 14 15
#include <linux/types.h>
#include <linux/errno.h>
#include <linux/fs.h>
#include <linux/nfs_fs.h>
#include <linux/utsname.h>
16
#include <linux/freezer.h>
L
Linus Torvalds 已提交
17 18 19 20 21 22
#include <linux/sunrpc/clnt.h>
#include <linux/sunrpc/svc.h>
#include <linux/lockd/lockd.h>

#define NLMDBG_FACILITY		NLMDBG_CLIENT
#define NLMCLNT_GRACE_WAIT	(5*HZ)
23
#define NLMCLNT_POLL_TIMEOUT	(30*HZ)
24
#define NLMCLNT_MAX_RETRIES	3
L
Linus Torvalds 已提交
25 26 27 28

static int	nlmclnt_test(struct nlm_rqst *, struct file_lock *);
static int	nlmclnt_lock(struct nlm_rqst *, struct file_lock *);
static int	nlmclnt_unlock(struct nlm_rqst *, struct file_lock *);
A
Al Viro 已提交
29
static int	nlm_stat_to_errno(__be32 stat);
L
Linus Torvalds 已提交
30
static void	nlmclnt_locks_init_private(struct file_lock *fl, struct nlm_host *host);
31
static int	nlmclnt_cancel(struct nlm_host *, int , struct file_lock *);
L
Linus Torvalds 已提交
32

33 34 35
static const struct rpc_call_ops nlmclnt_unlock_ops;
static const struct rpc_call_ops nlmclnt_cancel_ops;

L
Linus Torvalds 已提交
36 37 38
/*
 * Cookie counter for NLM requests
 */
39
static atomic_t	nlm_cookie = ATOMIC_INIT(0x1234);
L
Linus Torvalds 已提交
40

41
void nlmclnt_next_cookie(struct nlm_cookie *c)
L
Linus Torvalds 已提交
42
{
43 44 45
	u32	cookie = atomic_inc_return(&nlm_cookie);

	memcpy(c->data, &cookie, 4);
L
Linus Torvalds 已提交
46 47 48 49 50
	c->len=4;
}

static struct nlm_lockowner *nlm_get_lockowner(struct nlm_lockowner *lockowner)
{
51
	refcount_inc(&lockowner->count);
L
Linus Torvalds 已提交
52 53 54 55 56
	return lockowner;
}

static void nlm_put_lockowner(struct nlm_lockowner *lockowner)
{
57
	if (!refcount_dec_and_lock(&lockowner->count, &lockowner->host->h_lock))
L
Linus Torvalds 已提交
58 59 60
		return;
	list_del(&lockowner->list);
	spin_unlock(&lockowner->host->h_lock);
61
	nlmclnt_release_host(lockowner->host);
L
Linus Torvalds 已提交
62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102
	kfree(lockowner);
}

static inline int nlm_pidbusy(struct nlm_host *host, uint32_t pid)
{
	struct nlm_lockowner *lockowner;
	list_for_each_entry(lockowner, &host->h_lockowners, list) {
		if (lockowner->pid == pid)
			return -EBUSY;
	}
	return 0;
}

static inline uint32_t __nlm_alloc_pid(struct nlm_host *host)
{
	uint32_t res;
	do {
		res = host->h_pidcount++;
	} while (nlm_pidbusy(host, res) < 0);
	return res;
}

static struct nlm_lockowner *__nlm_find_lockowner(struct nlm_host *host, fl_owner_t owner)
{
	struct nlm_lockowner *lockowner;
	list_for_each_entry(lockowner, &host->h_lockowners, list) {
		if (lockowner->owner != owner)
			continue;
		return nlm_get_lockowner(lockowner);
	}
	return NULL;
}

static struct nlm_lockowner *nlm_find_lockowner(struct nlm_host *host, fl_owner_t owner)
{
	struct nlm_lockowner *res, *new = NULL;

	spin_lock(&host->h_lock);
	res = __nlm_find_lockowner(host, owner);
	if (res == NULL) {
		spin_unlock(&host->h_lock);
103
		new = kmalloc(sizeof(*new), GFP_KERNEL);
L
Linus Torvalds 已提交
104 105 106 107
		spin_lock(&host->h_lock);
		res = __nlm_find_lockowner(host, owner);
		if (res == NULL && new != NULL) {
			res = new;
108
			refcount_set(&new->count, 1);
L
Linus Torvalds 已提交
109 110 111 112 113 114 115 116
			new->owner = owner;
			new->pid = __nlm_alloc_pid(host);
			new->host = nlm_get_host(host);
			list_add(&new->list, &host->h_lockowners);
			new = NULL;
		}
	}
	spin_unlock(&host->h_lock);
J
Jesper Juhl 已提交
117
	kfree(new);
L
Linus Torvalds 已提交
118 119 120 121 122 123 124 125 126 127
	return res;
}

/*
 * Initialize arguments for TEST/LOCK/UNLOCK/CANCEL calls
 */
static void nlmclnt_setlockargs(struct nlm_rqst *req, struct file_lock *fl)
{
	struct nlm_args	*argp = &req->a_args;
	struct nlm_lock	*lock = &argp->lock;
128
	char *nodename = req->a_host->h_rpcclnt->cl_nodename;
L
Linus Torvalds 已提交
129 130

	nlmclnt_next_cookie(&argp->cookie);
131
	memcpy(&lock->fh, NFS_FH(locks_inode(fl->fl_file)), sizeof(struct nfs_fh));
132
	lock->caller  = nodename;
L
Linus Torvalds 已提交
133
	lock->oh.data = req->a_owner;
134 135
	lock->oh.len  = snprintf(req->a_owner, sizeof(req->a_owner), "%u@%s",
				(unsigned int)fl->fl_u.nfs_fl.owner->pid,
136
				nodename);
137
	lock->svid = fl->fl_u.nfs_fl.owner->pid;
T
Trond Myklebust 已提交
138 139 140
	lock->fl.fl_start = fl->fl_start;
	lock->fl.fl_end = fl->fl_end;
	lock->fl.fl_type = fl->fl_type;
L
Linus Torvalds 已提交
141 142 143 144
}

static void nlmclnt_release_lockargs(struct nlm_rqst *req)
{
145
	WARN_ON_ONCE(req->a_args.lock.fl.fl_ops != NULL);
L
Linus Torvalds 已提交
146 147
}

148 149 150 151 152
/**
 * nlmclnt_proc - Perform a single client-side lock request
 * @host: address of a valid nlm_host context representing the NLM server
 * @cmd: fcntl-style file lock operation to perform
 * @fl: address of arguments for the lock operation
153
 * @data: address of data to be sent to callback operations
154
 *
L
Linus Torvalds 已提交
155
 */
156
int nlmclnt_proc(struct nlm_host *host, int cmd, struct file_lock *fl, void *data)
L
Linus Torvalds 已提交
157
{
T
Trond Myklebust 已提交
158
	struct nlm_rqst		*call;
159
	int			status;
160
	const struct nlmclnt_operations *nlmclnt_ops = host->h_nlmclnt_ops;
L
Linus Torvalds 已提交
161

T
Trond Myklebust 已提交
162 163 164
	call = nlm_alloc_call(host);
	if (call == NULL)
		return -ENOMEM;
L
Linus Torvalds 已提交
165

166 167 168
	if (nlmclnt_ops && nlmclnt_ops->nlmclnt_alloc_call)
		nlmclnt_ops->nlmclnt_alloc_call(data);

T
Trond Myklebust 已提交
169
	nlmclnt_locks_init_private(fl, host);
170 171 172 173 174
	if (!fl->fl_u.nfs_fl.owner) {
		/* lockowner allocation has failed */
		nlmclnt_release_call(call);
		return -ENOMEM;
	}
T
Trond Myklebust 已提交
175 176
	/* Set up the argument struct */
	nlmclnt_setlockargs(call, fl);
177
	call->a_callback_data = data;
L
Linus Torvalds 已提交
178 179 180 181 182 183 184 185 186 187 188

	if (IS_SETLK(cmd) || IS_SETLKW(cmd)) {
		if (fl->fl_type != F_UNLCK) {
			call->a_args.block = IS_SETLKW(cmd) ? 1 : 0;
			status = nlmclnt_lock(call, fl);
		} else
			status = nlmclnt_unlock(call, fl);
	} else if (IS_GETLK(cmd))
		status = nlmclnt_test(call, fl);
	else
		status = -EINVAL;
T
Trond Myklebust 已提交
189 190 191
	fl->fl_ops->fl_release_private(fl);
	fl->fl_ops = NULL;

L
Linus Torvalds 已提交
192 193 194
	dprintk("lockd: clnt proc returns %d\n", status);
	return status;
}
195
EXPORT_SYMBOL_GPL(nlmclnt_proc);
L
Linus Torvalds 已提交
196 197 198 199

/*
 * Allocate an NLM RPC call struct
 */
T
Trond Myklebust 已提交
200
struct nlm_rqst *nlm_alloc_call(struct nlm_host *host)
L
Linus Torvalds 已提交
201 202 203
{
	struct nlm_rqst	*call;

204 205 206
	for(;;) {
		call = kzalloc(sizeof(*call), GFP_KERNEL);
		if (call != NULL) {
207
			refcount_set(&call->a_count, 1);
L
Linus Torvalds 已提交
208 209
			locks_init_lock(&call->a_args.lock.fl);
			locks_init_lock(&call->a_res.lock.fl);
210
			call->a_host = nlm_get_host(host);
L
Linus Torvalds 已提交
211 212
			return call;
		}
213 214
		if (signalled())
			break;
T
Trond Myklebust 已提交
215
		printk("nlm_alloc_call: failed, waiting for memory\n");
216
		schedule_timeout_interruptible(5*HZ);
L
Linus Torvalds 已提交
217 218 219 220
	}
	return NULL;
}

C
Chuck Lever 已提交
221
void nlmclnt_release_call(struct nlm_rqst *call)
T
Trond Myklebust 已提交
222
{
223 224
	const struct nlmclnt_operations *nlmclnt_ops = call->a_host->h_nlmclnt_ops;

225
	if (!refcount_dec_and_test(&call->a_count))
226
		return;
227 228
	if (nlmclnt_ops && nlmclnt_ops->nlmclnt_release_call)
		nlmclnt_ops->nlmclnt_release_call(call->a_callback_data);
229
	nlmclnt_release_host(call->a_host);
T
Trond Myklebust 已提交
230 231 232 233 234 235
	nlmclnt_release_lockargs(call);
	kfree(call);
}

static void nlmclnt_rpc_release(void *data)
{
C
Chuck Lever 已提交
236
	nlmclnt_release_call(data);
T
Trond Myklebust 已提交
237 238
}

L
Linus Torvalds 已提交
239 240 241 242 243 244 245 246
static int nlm_wait_on_grace(wait_queue_head_t *queue)
{
	DEFINE_WAIT(wait);
	int status = -EINTR;

	prepare_to_wait(queue, &wait, TASK_INTERRUPTIBLE);
	if (!signalled ()) {
		schedule_timeout(NLMCLNT_GRACE_WAIT);
247
		try_to_freeze();
L
Linus Torvalds 已提交
248 249 250 251 252 253 254 255 256 257 258
		if (!signalled ())
			status = 0;
	}
	finish_wait(queue, &wait);
	return status;
}

/*
 * Generic NLM call
 */
static int
259
nlmclnt_call(struct rpc_cred *cred, struct nlm_rqst *req, u32 proc)
L
Linus Torvalds 已提交
260 261 262 263 264 265 266 267
{
	struct nlm_host	*host = req->a_host;
	struct rpc_clnt	*clnt;
	struct nlm_args	*argp = &req->a_args;
	struct nlm_res	*resp = &req->a_res;
	struct rpc_message msg = {
		.rpc_argp	= argp,
		.rpc_resp	= resp,
268
		.rpc_cred	= cred,
L
Linus Torvalds 已提交
269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303
	};
	int		status;

	dprintk("lockd: call procedure %d on %s\n",
			(int)proc, host->h_name);

	do {
		if (host->h_reclaiming && !argp->reclaim)
			goto in_grace_period;

		/* If we have no RPC client yet, create one. */
		if ((clnt = nlm_bind_host(host)) == NULL)
			return -ENOLCK;
		msg.rpc_proc = &clnt->cl_procinfo[proc];

		/* Perform the RPC call. If an error occurs, try again */
		if ((status = rpc_call_sync(clnt, &msg, 0)) < 0) {
			dprintk("lockd: rpc_call returned error %d\n", -status);
			switch (status) {
			case -EPROTONOSUPPORT:
				status = -EINVAL;
				break;
			case -ECONNREFUSED:
			case -ETIMEDOUT:
			case -ENOTCONN:
				nlm_rebind_host(host);
				status = -EAGAIN;
				break;
			case -ERESTARTSYS:
				return signalled () ? -EINTR : status;
			default:
				break;
			}
			break;
		} else
A
Al Viro 已提交
304
		if (resp->status == nlm_lck_denied_grace_period) {
L
Linus Torvalds 已提交
305 306 307 308 309 310 311 312 313 314 315
			dprintk("lockd: server in grace period\n");
			if (argp->reclaim) {
				printk(KERN_WARNING
				     "lockd: spurious grace period reject?!\n");
				return -ENOLCK;
			}
		} else {
			if (!argp->reclaim) {
				/* We appear to be out of the grace period */
				wake_up_all(&host->h_gracewait);
			}
316 317
			dprintk("lockd: server returns status %d\n",
				ntohl(resp->status));
L
Linus Torvalds 已提交
318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336
			return 0;	/* Okay, call complete */
		}

in_grace_period:
		/*
		 * The server has rebooted and appears to be in the grace
		 * period during which locks are only allowed to be
		 * reclaimed.
		 * We can only back off and try again later.
		 */
		status = nlm_wait_on_grace(&host->h_gracewait);
	} while (status == 0);

	return status;
}

/*
 * Generic NLM call, async version.
 */
337
static struct rpc_task *__nlm_async_call(struct nlm_rqst *req, u32 proc, struct rpc_message *msg, const struct rpc_call_ops *tk_ops)
L
Linus Torvalds 已提交
338 339 340
{
	struct nlm_host	*host = req->a_host;
	struct rpc_clnt	*clnt;
341 342 343 344 345 346
	struct rpc_task_setup task_setup_data = {
		.rpc_message = msg,
		.callback_ops = tk_ops,
		.callback_data = req,
		.flags = RPC_TASK_ASYNC,
	};
L
Linus Torvalds 已提交
347 348 349 350 351

	dprintk("lockd: call procedure %d on %s (async)\n",
			(int)proc, host->h_name);

	/* If we have no RPC client yet, create one. */
T
Trond Myklebust 已提交
352 353 354
	clnt = nlm_bind_host(host);
	if (clnt == NULL)
		goto out_err;
355
	msg->rpc_proc = &clnt->cl_procinfo[proc];
356
	task_setup_data.rpc_client = clnt;
L
Linus Torvalds 已提交
357 358

        /* bootstrap and kick off the async RPC call */
359
	return rpc_run_task(&task_setup_data);
T
Trond Myklebust 已提交
360
out_err:
361
	tk_ops->rpc_release(req);
362 363 364 365 366 367 368 369 370 371 372 373
	return ERR_PTR(-ENOLCK);
}

static int nlm_do_async_call(struct nlm_rqst *req, u32 proc, struct rpc_message *msg, const struct rpc_call_ops *tk_ops)
{
	struct rpc_task *task;

	task = __nlm_async_call(req, proc, msg, tk_ops);
	if (IS_ERR(task))
		return PTR_ERR(task);
	rpc_put_task(task);
	return 0;
L
Linus Torvalds 已提交
374 375
}

376 377 378
/*
 * NLM asynchronous call.
 */
379 380 381 382 383 384
int nlm_async_call(struct nlm_rqst *req, u32 proc, const struct rpc_call_ops *tk_ops)
{
	struct rpc_message msg = {
		.rpc_argp	= &req->a_args,
		.rpc_resp	= &req->a_res,
	};
385
	return nlm_do_async_call(req, proc, &msg, tk_ops);
386 387 388 389 390 391 392
}

int nlm_async_reply(struct nlm_rqst *req, u32 proc, const struct rpc_call_ops *tk_ops)
{
	struct rpc_message msg = {
		.rpc_argp	= &req->a_res,
	};
393 394 395 396 397 398 399 400 401 402 403
	return nlm_do_async_call(req, proc, &msg, tk_ops);
}

/*
 * NLM client asynchronous call.
 *
 * Note that although the calls are asynchronous, and are therefore
 *      guaranteed to complete, we still always attempt to wait for
 *      completion in order to be able to correctly track the lock
 *      state.
 */
404
static int nlmclnt_async_call(struct rpc_cred *cred, struct nlm_rqst *req, u32 proc, const struct rpc_call_ops *tk_ops)
405 406 407 408
{
	struct rpc_message msg = {
		.rpc_argp	= &req->a_args,
		.rpc_resp	= &req->a_res,
409
		.rpc_cred	= cred,
410 411 412 413 414 415 416 417 418 419
	};
	struct rpc_task *task;
	int err;

	task = __nlm_async_call(req, proc, &msg, tk_ops);
	if (IS_ERR(task))
		return PTR_ERR(task);
	err = rpc_wait_for_completion_task(task);
	rpc_put_task(task);
	return err;
420 421
}

L
Linus Torvalds 已提交
422 423 424 425 426 427 428 429
/*
 * TEST for the presence of a conflicting lock
 */
static int
nlmclnt_test(struct nlm_rqst *req, struct file_lock *fl)
{
	int	status;

430
	status = nlmclnt_call(nfs_file_cred(fl->fl_file), req, NLMPROC_TEST);
L
Linus Torvalds 已提交
431
	if (status < 0)
T
Trond Myklebust 已提交
432
		goto out;
L
Linus Torvalds 已提交
433

T
Trond Myklebust 已提交
434
	switch (req->a_res.status) {
A
Al Viro 已提交
435
		case nlm_granted:
T
Trond Myklebust 已提交
436 437
			fl->fl_type = F_UNLCK;
			break;
A
Al Viro 已提交
438
		case nlm_lck_denied:
T
Trond Myklebust 已提交
439 440 441 442
			/*
			 * Report the conflicting lock back to the application.
			 */
			fl->fl_start = req->a_res.lock.fl.fl_start;
443
			fl->fl_end = req->a_res.lock.fl.fl_end;
T
Trond Myklebust 已提交
444 445 446 447 448
			fl->fl_type = req->a_res.lock.fl.fl_type;
			fl->fl_pid = 0;
			break;
		default:
			status = nlm_stat_to_errno(req->a_res.status);
L
Linus Torvalds 已提交
449
	}
T
Trond Myklebust 已提交
450
out:
C
Chuck Lever 已提交
451
	nlmclnt_release_call(req);
T
Trond Myklebust 已提交
452
	return status;
L
Linus Torvalds 已提交
453 454 455 456
}

static void nlmclnt_locks_copy_lock(struct file_lock *new, struct file_lock *fl)
{
457
	spin_lock(&fl->fl_u.nfs_fl.owner->host->h_lock);
458 459 460
	new->fl_u.nfs_fl.state = fl->fl_u.nfs_fl.state;
	new->fl_u.nfs_fl.owner = nlm_get_lockowner(fl->fl_u.nfs_fl.owner);
	list_add_tail(&new->fl_u.nfs_fl.list, &fl->fl_u.nfs_fl.owner->host->h_granted);
461
	spin_unlock(&fl->fl_u.nfs_fl.owner->host->h_lock);
L
Linus Torvalds 已提交
462 463 464 465
}

static void nlmclnt_locks_release_private(struct file_lock *fl)
{
466
	spin_lock(&fl->fl_u.nfs_fl.owner->host->h_lock);
467
	list_del(&fl->fl_u.nfs_fl.list);
468
	spin_unlock(&fl->fl_u.nfs_fl.owner->host->h_lock);
L
Linus Torvalds 已提交
469 470 471
	nlm_put_lockowner(fl->fl_u.nfs_fl.owner);
}

472
static const struct file_lock_operations nlmclnt_lock_ops = {
L
Linus Torvalds 已提交
473 474 475 476 477 478 479 480
	.fl_copy_lock = nlmclnt_locks_copy_lock,
	.fl_release_private = nlmclnt_locks_release_private,
};

static void nlmclnt_locks_init_private(struct file_lock *fl, struct nlm_host *host)
{
	fl->fl_u.nfs_fl.state = 0;
	fl->fl_u.nfs_fl.owner = nlm_find_lockowner(host, fl->fl_owner);
481
	INIT_LIST_HEAD(&fl->fl_u.nfs_fl.list);
L
Linus Torvalds 已提交
482 483 484
	fl->fl_ops = &nlmclnt_lock_ops;
}

485
static int do_vfs_lock(struct file_lock *fl)
L
Linus Torvalds 已提交
486
{
487
	return locks_lock_file_wait(fl->fl_file, fl);
L
Linus Torvalds 已提交
488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512
}

/*
 * LOCK: Try to create a lock
 *
 *			Programmer Harassment Alert
 *
 * When given a blocking lock request in a sync RPC call, the HPUX lockd
 * will faithfully return LCK_BLOCKED but never cares to notify us when
 * the lock could be granted. This way, our local process could hang
 * around forever waiting for the callback.
 *
 *  Solution A:	Implement busy-waiting
 *  Solution B: Use the async version of the call (NLM_LOCK_{MSG,RES})
 *
 * For now I am implementing solution A, because I hate the idea of
 * re-implementing lockd for a third time in two months. The async
 * calls shouldn't be too hard to do, however.
 *
 * This is one of the lovely things about standards in the NFS area:
 * they're so soft and squishy you can't really blame HP for doing this.
 */
static int
nlmclnt_lock(struct nlm_rqst *req, struct file_lock *fl)
{
513
	struct rpc_cred *cred = nfs_file_cred(fl->fl_file);
L
Linus Torvalds 已提交
514 515
	struct nlm_host	*host = req->a_host;
	struct nlm_res	*resp = &req->a_res;
T
Trond Myklebust 已提交
516
	struct nlm_wait *block = NULL;
517
	unsigned char fl_flags = fl->fl_flags;
518
	unsigned char fl_type;
T
Trond Myklebust 已提交
519
	int status = -ENOLCK;
L
Linus Torvalds 已提交
520

521
	if (nsm_monitor(host) < 0)
L
Linus Torvalds 已提交
522
		goto out;
523
	req->a_args.state = nsm_local_state;
524

525 526
	fl->fl_flags |= FL_ACCESS;
	status = do_vfs_lock(fl);
527
	fl->fl_flags = fl_flags;
528 529
	if (status < 0)
		goto out;
L
Linus Torvalds 已提交
530

T
Trond Myklebust 已提交
531
	block = nlmclnt_prepare_block(host, fl);
T
Trond Myklebust 已提交
532
again:
533 534 535 536 537
	/*
	 * Initialise resp->status to a valid non-zero value,
	 * since 0 == nlm_lck_granted
	 */
	resp->status = nlm_lck_blocked;
538
	for(;;) {
T
Trond Myklebust 已提交
539 540
		/* Reboot protection */
		fl->fl_u.nfs_fl.state = host->h_state;
541
		status = nlmclnt_call(cred, req, NLMPROC_LOCK);
542 543 544
		if (status < 0)
			break;
		/* Did a reclaimer thread notify us of a server reboot? */
A
Al Viro 已提交
545
		if (resp->status ==  nlm_lck_denied_grace_period)
546
			continue;
A
Al Viro 已提交
547
		if (resp->status != nlm_lck_blocked)
548
			break;
T
Trond Myklebust 已提交
549 550 551
		/* Wait on an NLM blocking lock */
		status = nlmclnt_block(block, req, NLMCLNT_POLL_TIMEOUT);
		if (status < 0)
552
			break;
A
Al Viro 已提交
553
		if (resp->status != nlm_lck_blocked)
T
Trond Myklebust 已提交
554
			break;
555
	}
L
Linus Torvalds 已提交
556

557 558 559 560 561 562 563 564 565 566
	/* if we were interrupted while blocking, then cancel the lock request
	 * and exit
	 */
	if (resp->status == nlm_lck_blocked) {
		if (!req->a_args.block)
			goto out_unlock;
		if (nlmclnt_cancel(host, req->a_args.block, fl) == 0)
			goto out_unblock;
	}

A
Al Viro 已提交
567
	if (resp->status == nlm_granted) {
T
Trond Myklebust 已提交
568 569 570 571 572 573
		down_read(&host->h_rwsem);
		/* Check whether or not the server has rebooted */
		if (fl->fl_u.nfs_fl.state != host->h_state) {
			up_read(&host->h_rwsem);
			goto again;
		}
574
		/* Ensure the resulting lock will get added to granted list */
575
		fl->fl_flags |= FL_SLEEP;
576
		if (do_vfs_lock(fl) < 0)
577
			printk(KERN_WARNING "%s: VFS is out of sync with lock manager!\n", __func__);
T
Trond Myklebust 已提交
578
		up_read(&host->h_rwsem);
579
		fl->fl_flags = fl_flags;
580
		status = 0;
L
Linus Torvalds 已提交
581
	}
582 583
	if (status < 0)
		goto out_unlock;
584 585 586 587 588 589 590 591 592
	/*
	 * EAGAIN doesn't make sense for sleeping locks, and in some
	 * cases NLM_LCK_DENIED is returned for a permanent error.  So
	 * turn it into an ENOLCK.
	 */
	if (resp->status == nlm_lck_denied && (fl_flags & FL_SLEEP))
		status = -ENOLCK;
	else
		status = nlm_stat_to_errno(resp->status);
593
out_unblock:
T
Trond Myklebust 已提交
594
	nlmclnt_finish_block(block);
L
Linus Torvalds 已提交
595
out:
C
Chuck Lever 已提交
596
	nlmclnt_release_call(req);
L
Linus Torvalds 已提交
597
	return status;
598 599 600 601 602 603 604 605 606 607 608 609
out_unlock:
	/* Fatal error: ensure that we remove the lock altogether */
	dprintk("lockd: lock attempt ended in fatal error.\n"
		"       Attempting to unlock.\n");
	nlmclnt_finish_block(block);
	fl_type = fl->fl_type;
	fl->fl_type = F_UNLCK;
	down_read(&host->h_rwsem);
	do_vfs_lock(fl);
	up_read(&host->h_rwsem);
	fl->fl_type = fl_type;
	fl->fl_flags = fl_flags;
610
	nlmclnt_async_call(cred, req, NLMPROC_UNLOCK, &nlmclnt_unlock_ops);
611
	return status;
L
Linus Torvalds 已提交
612 613 614 615 616 617
}

/*
 * RECLAIM: Try to reclaim a lock
 */
int
618 619
nlmclnt_reclaim(struct nlm_host *host, struct file_lock *fl,
		struct nlm_rqst *req)
L
Linus Torvalds 已提交
620 621 622 623 624 625 626 627 628 629 630 631
{
	int		status;

	memset(req, 0, sizeof(*req));
	locks_init_lock(&req->a_args.lock.fl);
	locks_init_lock(&req->a_res.lock.fl);
	req->a_host  = host;

	/* Set up the argument struct */
	nlmclnt_setlockargs(req, fl);
	req->a_args.reclaim = 1;

632 633
	status = nlmclnt_call(nfs_file_cred(fl->fl_file), req, NLMPROC_LOCK);
	if (status >= 0 && req->a_res.status == nlm_granted)
L
Linus Torvalds 已提交
634 635 636 637
		return 0;

	printk(KERN_WARNING "lockd: failed to reclaim lock for pid %d "
				"(errno %d, status %d)\n", fl->fl_pid,
A
Al Viro 已提交
638
				status, ntohl(req->a_res.status));
L
Linus Torvalds 已提交
639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660

	/*
	 * FIXME: This is a serious failure. We can
	 *
	 *  a.	Ignore the problem
	 *  b.	Send the owning process some signal (Linux doesn't have
	 *	SIGLOST, though...)
	 *  c.	Retry the operation
	 *
	 * Until someone comes up with a simple implementation
	 * for b or c, I'll choose option a.
	 */

	return -ENOLCK;
}

/*
 * UNLOCK: remove an existing lock
 */
static int
nlmclnt_unlock(struct nlm_rqst *req, struct file_lock *fl)
{
T
Trond Myklebust 已提交
661
	struct nlm_host	*host = req->a_host;
L
Linus Torvalds 已提交
662
	struct nlm_res	*resp = &req->a_res;
663 664
	int status;
	unsigned char fl_flags = fl->fl_flags;
L
Linus Torvalds 已提交
665

666 667 668 669 670
	/*
	 * Note: the server is supposed to either grant us the unlock
	 * request, or to deny it with NLM_LCK_DENIED_GRACE_PERIOD. In either
	 * case, we want to unlock.
	 */
671
	fl->fl_flags |= FL_EXISTS;
T
Trond Myklebust 已提交
672
	down_read(&host->h_rwsem);
673 674 675 676 677
	status = do_vfs_lock(fl);
	up_read(&host->h_rwsem);
	fl->fl_flags = fl_flags;
	if (status == -ENOENT) {
		status = 0;
678 679
		goto out;
	}
680

681
	refcount_inc(&req->a_count);
682 683
	status = nlmclnt_async_call(nfs_file_cred(fl->fl_file), req,
			NLMPROC_UNLOCK, &nlmclnt_unlock_ops);
L
Linus Torvalds 已提交
684
	if (status < 0)
T
Trond Myklebust 已提交
685
		goto out;
L
Linus Torvalds 已提交
686

A
Al Viro 已提交
687
	if (resp->status == nlm_granted)
T
Trond Myklebust 已提交
688
		goto out;
L
Linus Torvalds 已提交
689

A
Al Viro 已提交
690
	if (resp->status != nlm_lck_denied_nolocks)
691 692
		printk("lockd: unexpected unlock status: %d\n",
			ntohl(resp->status));
L
Linus Torvalds 已提交
693
	/* What to do now? I'm out of my depth... */
T
Trond Myklebust 已提交
694 695
	status = -ENOLCK;
out:
C
Chuck Lever 已提交
696
	nlmclnt_release_call(req);
T
Trond Myklebust 已提交
697
	return status;
L
Linus Torvalds 已提交
698 699
}

700 701 702 703 704 705 706 707 708 709 710 711 712
static void nlmclnt_unlock_prepare(struct rpc_task *task, void *data)
{
	struct nlm_rqst	*req = data;
	const struct nlmclnt_operations *nlmclnt_ops = req->a_host->h_nlmclnt_ops;
	bool defer_call = false;

	if (nlmclnt_ops && nlmclnt_ops->nlmclnt_unlock_prepare)
		defer_call = nlmclnt_ops->nlmclnt_unlock_prepare(task, req->a_callback_data);

	if (!defer_call)
		rpc_call_start(task);
}

713
static void nlmclnt_unlock_callback(struct rpc_task *task, void *data)
L
Linus Torvalds 已提交
714
{
715
	struct nlm_rqst	*req = data;
A
Al Viro 已提交
716
	u32 status = ntohl(req->a_res.status);
L
Linus Torvalds 已提交
717 718 719 720 721 722

	if (RPC_ASSASSINATED(task))
		goto die;

	if (task->tk_status < 0) {
		dprintk("lockd: unlock failed (err = %d)\n", -task->tk_status);
723 724 725 726 727 728 729
		switch (task->tk_status) {
		case -EACCES:
		case -EIO:
			goto die;
		default:
			goto retry_rebind;
		}
L
Linus Torvalds 已提交
730 731 732 733 734 735 736 737 738 739 740 741 742 743 744
	}
	if (status == NLM_LCK_DENIED_GRACE_PERIOD) {
		rpc_delay(task, NLMCLNT_GRACE_WAIT);
		goto retry_unlock;
	}
	if (status != NLM_LCK_GRANTED)
		printk(KERN_WARNING "lockd: unexpected unlock status: %d\n", status);
die:
	return;
 retry_rebind:
	nlm_rebind_host(req->a_host);
 retry_unlock:
	rpc_restart_call(task);
}

745
static const struct rpc_call_ops nlmclnt_unlock_ops = {
746
	.rpc_call_prepare = nlmclnt_unlock_prepare,
747
	.rpc_call_done = nlmclnt_unlock_callback,
T
Trond Myklebust 已提交
748
	.rpc_release = nlmclnt_rpc_release,
749 750
};

L
Linus Torvalds 已提交
751 752 753 754 755
/*
 * Cancel a blocked lock request.
 * We always use an async RPC call for this in order not to hang a
 * process that has been Ctrl-C'ed.
 */
756
static int nlmclnt_cancel(struct nlm_host *host, int block, struct file_lock *fl)
L
Linus Torvalds 已提交
757 758
{
	struct nlm_rqst	*req;
759 760 761 762
	int status;

	dprintk("lockd: blocking lock attempt was interrupted by a signal.\n"
		"       Attempting to cancel lock.\n");
L
Linus Torvalds 已提交
763

764
	req = nlm_alloc_call(host);
L
Linus Torvalds 已提交
765 766 767 768 769
	if (!req)
		return -ENOMEM;
	req->a_flags = RPC_TASK_ASYNC;

	nlmclnt_setlockargs(req, fl);
770
	req->a_args.block = block;
L
Linus Torvalds 已提交
771

772
	refcount_inc(&req->a_count);
773 774
	status = nlmclnt_async_call(nfs_file_cred(fl->fl_file), req,
			NLMPROC_CANCEL, &nlmclnt_cancel_ops);
775 776
	if (status == 0 && req->a_res.status == nlm_lck_denied)
		status = -ENOLCK;
C
Chuck Lever 已提交
777
	nlmclnt_release_call(req);
778
	return status;
L
Linus Torvalds 已提交
779 780
}

781
static void nlmclnt_cancel_callback(struct rpc_task *task, void *data)
L
Linus Torvalds 已提交
782
{
783
	struct nlm_rqst	*req = data;
A
Al Viro 已提交
784
	u32 status = ntohl(req->a_res.status);
L
Linus Torvalds 已提交
785 786 787 788 789 790 791 792 793 794

	if (RPC_ASSASSINATED(task))
		goto die;

	if (task->tk_status < 0) {
		dprintk("lockd: CANCEL call error %d, retrying.\n",
					task->tk_status);
		goto retry_cancel;
	}

C
Chuck Lever 已提交
795
	dprintk("lockd: cancel status %u (task %u)\n",
A
Al Viro 已提交
796
			status, task->tk_pid);
L
Linus Torvalds 已提交
797

A
Al Viro 已提交
798
	switch (status) {
L
Linus Torvalds 已提交
799 800
	case NLM_LCK_GRANTED:
	case NLM_LCK_DENIED_GRACE_PERIOD:
801
	case NLM_LCK_DENIED:
L
Linus Torvalds 已提交
802 803 804 805 806 807 808
		/* Everything's good */
		break;
	case NLM_LCK_DENIED_NOLOCKS:
		dprintk("lockd: CANCEL failed (server has no locks)\n");
		goto retry_cancel;
	default:
		printk(KERN_NOTICE "lockd: weird return %d for CANCEL call\n",
A
Al Viro 已提交
809
			status);
L
Linus Torvalds 已提交
810 811 812 813 814 815
	}

die:
	return;

retry_cancel:
816 817 818
	/* Don't ever retry more than 3 times */
	if (req->a_retries++ >= NLMCLNT_MAX_RETRIES)
		goto die;
L
Linus Torvalds 已提交
819 820 821 822 823
	nlm_rebind_host(req->a_host);
	rpc_restart_call(task);
	rpc_delay(task, 30 * HZ);
}

824 825
static const struct rpc_call_ops nlmclnt_cancel_ops = {
	.rpc_call_done = nlmclnt_cancel_callback,
T
Trond Myklebust 已提交
826
	.rpc_release = nlmclnt_rpc_release,
827 828
};

L
Linus Torvalds 已提交
829 830 831 832
/*
 * Convert an NLM status code to a generic kernel errno
 */
static int
A
Al Viro 已提交
833
nlm_stat_to_errno(__be32 status)
L
Linus Torvalds 已提交
834
{
A
Al Viro 已提交
835
	switch(ntohl(status)) {
L
Linus Torvalds 已提交
836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858
	case NLM_LCK_GRANTED:
		return 0;
	case NLM_LCK_DENIED:
		return -EAGAIN;
	case NLM_LCK_DENIED_NOLOCKS:
	case NLM_LCK_DENIED_GRACE_PERIOD:
		return -ENOLCK;
	case NLM_LCK_BLOCKED:
		printk(KERN_NOTICE "lockd: unexpected status NLM_BLOCKED\n");
		return -ENOLCK;
#ifdef CONFIG_LOCKD_V4
	case NLM_DEADLCK:
		return -EDEADLK;
	case NLM_ROFS:
		return -EROFS;
	case NLM_STALE_FH:
		return -ESTALE;
	case NLM_FBIG:
		return -EOVERFLOW;
	case NLM_FAILED:
		return -ENOLCK;
#endif
	}
859 860
	printk(KERN_NOTICE "lockd: unexpected server status %d\n",
		 ntohl(status));
L
Linus Torvalds 已提交
861 862
	return -ENOLCK;
}