[people/ms/linux.git] / fs / lockd / svcproc.c

/*
 * linux/fs/lockd/svcproc.c
 *
 * Lockd server procedures. We don't implement the NLM_*_RES 
 * procedures because we don't use the async procedures.
 *
 * Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de>
 */

#include <linux/types.h>
#include <linux/time.h>
#include <linux/lockd/lockd.h>
#include <linux/lockd/share.h>

#define NLMDBG_FACILITY		NLMDBG_CLIENT

#ifdef CONFIG_LOCKD_V4
static __be32
cast_to_nlm(__be32 status, u32 vers)
{
	/* Note: status is assumed to be in network byte order !!! */
	if (vers != 4){
		switch (status) {
		case nlm_granted:
		case nlm_lck_denied:
		case nlm_lck_denied_nolocks:
		case nlm_lck_blocked:
		case nlm_lck_denied_grace_period:
		case nlm_drop_reply:
			break;
		case nlm4_deadlock:
			status = nlm_lck_denied;
			break;
		default:
			status = nlm_lck_denied_nolocks;
		}
	}

	return (status);
}
#define	cast_status(status) (cast_to_nlm(status, rqstp->rq_vers))
#else
#define cast_status(status) (status)
#endif

/*
 * Obtain client and file from arguments
 */
static __be32
nlmsvc_retrieve_args(struct svc_rqst *rqstp, struct nlm_args *argp,
			struct nlm_host **hostp, struct nlm_file **filp)
{
	struct nlm_host		*host = NULL;
	struct nlm_file		*file = NULL;
	struct nlm_lock		*lock = &argp->lock;
	__be32			error = 0;

	/* nfsd callbacks must have been installed for this procedure */
	if (!nlmsvc_ops)
		return nlm_lck_denied_nolocks;

	/* Obtain host handle */
	if (!(host = nlmsvc_lookup_host(rqstp, lock->caller, lock->len))
	 || (argp->monitor && nsm_monitor(host) < 0))
		goto no_locks;
	*hostp = host;

	/* Obtain file pointer. Not used by FREE_ALL call. */
	if (filp != NULL) {
		if ((error = nlm_lookup_file(rqstp, &file, &lock->fh)) != 0)
			goto no_locks;
		*filp = file;

		/* Set up the missing parts of the file_lock structure */
		lock->fl.fl_file  = file->f_file;
		lock->fl.fl_owner = (fl_owner_t) host;
		lock->fl.fl_lmops = &nlmsvc_lock_operations;
	}

	return 0;

no_locks:
	nlmsvc_release_host(host);
	if (error)
		return error;
	return nlm_lck_denied_nolocks;
}

/*
 * NULL: Test for presence of service
 */
static __be32
nlmsvc_proc_null(struct svc_rqst *rqstp, void *argp, void *resp)
{
	dprintk("lockd: NULL          called\n");
	return rpc_success;
}

/*
 * TEST: Check for conflicting lock
 */
static __be32
nlmsvc_proc_test(struct svc_rqst *rqstp, struct nlm_args *argp,
				         struct nlm_res  *resp)
{
	struct nlm_host	*host;
	struct nlm_file	*file;
	__be32 rc = rpc_success;

	dprintk("lockd: TEST          called\n");
	resp->cookie = argp->cookie;

	/* Obtain client and file */
	if ((resp->status = nlmsvc_retrieve_args(rqstp, argp, &host, &file)))
		return resp->status == nlm_drop_reply ? rpc_drop_reply :rpc_success;

	/* Now check for conflicting locks */
	resp->status = cast_status(nlmsvc_testlock(rqstp, file, host, &argp->lock, &resp->lock, &resp->cookie));
	if (resp->status == nlm_drop_reply)
		rc = rpc_drop_reply;
	else
		dprintk("lockd: TEST          status %d vers %d\n",
			ntohl(resp->status), rqstp->rq_vers);

	nlmsvc_release_host(host);
	nlm_release_file(file);
	return rc;
}

static __be32
nlmsvc_proc_lock(struct svc_rqst *rqstp, struct nlm_args *argp,
				         struct nlm_res  *resp)
{
	struct nlm_host	*host;
	struct nlm_file	*file;
	__be32 rc = rpc_success;

	dprintk("lockd: LOCK          called\n");

	resp->cookie = argp->cookie;

	/* Obtain client and file */
	if ((resp->status = nlmsvc_retrieve_args(rqstp, argp, &host, &file)))
		return resp->status == nlm_drop_reply ? rpc_drop_reply :rpc_success;

#if 0
	/* If supplied state doesn't match current state, we assume it's
	 * an old request that time-warped somehow. Any error return would
	 * do in this case because it's irrelevant anyway.
	 *
	 * NB: We don't retrieve the remote host's state yet.
	 */
	if (host->h_nsmstate && host->h_nsmstate != argp->state) {
		resp->status = nlm_lck_denied_nolocks;
	} else
#endif

	/* Now try to lock the file */
	resp->status = cast_status(nlmsvc_lock(rqstp, file, host, &argp->lock,
					       argp->block, &argp->cookie,
					       argp->reclaim));
	if (resp->status == nlm_drop_reply)
		rc = rpc_drop_reply;
	else
		dprintk("lockd: LOCK         status %d\n", ntohl(resp->status));

	nlmsvc_release_host(host);
	nlm_release_file(file);
	return rc;
}

static __be32
nlmsvc_proc_cancel(struct svc_rqst *rqstp, struct nlm_args *argp,
				           struct nlm_res  *resp)
{
	struct nlm_host	*host;
	struct nlm_file	*file;

	dprintk("lockd: CANCEL        called\n");

	resp->cookie = argp->cookie;

	/* Don't accept requests during grace period */
	if (locks_in_grace()) {
		resp->status = nlm_lck_denied_grace_period;
		return rpc_success;
	}

	/* Obtain client and file */
	if ((resp->status = nlmsvc_retrieve_args(rqstp, argp, &host, &file)))
		return resp->status == nlm_drop_reply ? rpc_drop_reply :rpc_success;

	/* Try to cancel request. */
	resp->status = cast_status(nlmsvc_cancel_blocked(file, &argp->lock));

	dprintk("lockd: CANCEL        status %d\n", ntohl(resp->status));
	nlmsvc_release_host(host);
	nlm_release_file(file);
	return rpc_success;
}

/*
 * UNLOCK: release a lock
 */
static __be32
nlmsvc_proc_unlock(struct svc_rqst *rqstp, struct nlm_args *argp,
				           struct nlm_res  *resp)
{
	struct nlm_host	*host;
	struct nlm_file	*file;

	dprintk("lockd: UNLOCK        called\n");

	resp->cookie = argp->cookie;

	/* Don't accept new lock requests during grace period */
	if (locks_in_grace()) {
		resp->status = nlm_lck_denied_grace_period;
		return rpc_success;
	}

	/* Obtain client and file */
	if ((resp->status = nlmsvc_retrieve_args(rqstp, argp, &host, &file)))
		return resp->status == nlm_drop_reply ? rpc_drop_reply :rpc_success;

	/* Now try to remove the lock */
	resp->status = cast_status(nlmsvc_unlock(file, &argp->lock));

	dprintk("lockd: UNLOCK        status %d\n", ntohl(resp->status));
	nlmsvc_release_host(host);
	nlm_release_file(file);
	return rpc_success;
}

/*
 * GRANTED: A server calls us to tell that a process' lock request
 * was granted
 */
static __be32
nlmsvc_proc_granted(struct svc_rqst *rqstp, struct nlm_args *argp,
				            struct nlm_res  *resp)
{
	resp->cookie = argp->cookie;

	dprintk("lockd: GRANTED       called\n");
	resp->status = nlmclnt_grant(svc_addr(rqstp), &argp->lock);
	dprintk("lockd: GRANTED       status %d\n", ntohl(resp->status));
	return rpc_success;
}

/*
 * This is the generic lockd callback for async RPC calls
 */
static void nlmsvc_callback_exit(struct rpc_task *task, void *data)
{
	dprintk("lockd: %5u callback returned %d\n", task->tk_pid,
			-task->tk_status);
}

void nlmsvc_release_call(struct nlm_rqst *call)
{
	if (!atomic_dec_and_test(&call->a_count))
		return;
	nlmsvc_release_host(call->a_host);
	kfree(call);
}

static void nlmsvc_callback_release(void *data)
{
	nlmsvc_release_call(data);
}

static const struct rpc_call_ops nlmsvc_callback_ops = {
	.rpc_call_done = nlmsvc_callback_exit,
	.rpc_release = nlmsvc_callback_release,
};

/*
 * `Async' versions of the above service routines. They aren't really,
 * because we send the callback before the reply proper. I hope this
 * doesn't break any clients.
 */
static __be32 nlmsvc_callback(struct svc_rqst *rqstp, u32 proc, struct nlm_args *argp,
		__be32 (*func)(struct svc_rqst *, struct nlm_args *, struct nlm_res  *))
{
	struct nlm_host	*host;
	struct nlm_rqst	*call;
	__be32 stat;

	host = nlmsvc_lookup_host(rqstp,
				  argp->lock.caller,
				  argp->lock.len);
	if (host == NULL)
		return rpc_system_err;

	call = nlm_alloc_call(host);
	if (call == NULL)
		return rpc_system_err;

	stat = func(rqstp, argp, &call->a_res);
	if (stat != 0) {
		nlmsvc_release_call(call);
		return stat;
	}

	call->a_flags = RPC_TASK_ASYNC;
	if (nlm_async_reply(call, proc, &nlmsvc_callback_ops) < 0)
		return rpc_system_err;
	return rpc_success;
}

static __be32 nlmsvc_proc_test_msg(struct svc_rqst *rqstp, struct nlm_args *argp,
					     void	     *resp)
{
	dprintk("lockd: TEST_MSG      called\n");
	return nlmsvc_callback(rqstp, NLMPROC_TEST_RES, argp, nlmsvc_proc_test);
}

static __be32 nlmsvc_proc_lock_msg(struct svc_rqst *rqstp, struct nlm_args *argp,
					     void	     *resp)
{
	dprintk("lockd: LOCK_MSG      called\n");
	return nlmsvc_callback(rqstp, NLMPROC_LOCK_RES, argp, nlmsvc_proc_lock);
}

static __be32 nlmsvc_proc_cancel_msg(struct svc_rqst *rqstp, struct nlm_args *argp,
					       void	       *resp)
{
	dprintk("lockd: CANCEL_MSG    called\n");
	return nlmsvc_callback(rqstp, NLMPROC_CANCEL_RES, argp, nlmsvc_proc_cancel);
}

static __be32
nlmsvc_proc_unlock_msg(struct svc_rqst *rqstp, struct nlm_args *argp,
                                               void            *resp)
{
	dprintk("lockd: UNLOCK_MSG    called\n");
	return nlmsvc_callback(rqstp, NLMPROC_UNLOCK_RES, argp, nlmsvc_proc_unlock);
}

static __be32
nlmsvc_proc_granted_msg(struct svc_rqst *rqstp, struct nlm_args *argp,
                                                void            *resp)
{
	dprintk("lockd: GRANTED_MSG   called\n");
	return nlmsvc_callback(rqstp, NLMPROC_GRANTED_RES, argp, nlmsvc_proc_granted);
}

/*
 * SHARE: create a DOS share or alter existing share.
 */
static __be32
nlmsvc_proc_share(struct svc_rqst *rqstp, struct nlm_args *argp,
				          struct nlm_res  *resp)
{
	struct nlm_host	*host;
	struct nlm_file	*file;

	dprintk("lockd: SHARE         called\n");

	resp->cookie = argp->cookie;

	/* Don't accept new lock requests during grace period */
	if (locks_in_grace() && !argp->reclaim) {
		resp->status = nlm_lck_denied_grace_period;
		return rpc_success;
	}

	/* Obtain client and file */
	if ((resp->status = nlmsvc_retrieve_args(rqstp, argp, &host, &file)))
		return resp->status == nlm_drop_reply ? rpc_drop_reply :rpc_success;

	/* Now try to create the share */
	resp->status = cast_status(nlmsvc_share_file(host, file, argp));

	dprintk("lockd: SHARE         status %d\n", ntohl(resp->status));
	nlmsvc_release_host(host);
	nlm_release_file(file);
	return rpc_success;
}

/*
 * UNSHARE: Release a DOS share.
 */
static __be32
nlmsvc_proc_unshare(struct svc_rqst *rqstp, struct nlm_args *argp,
				            struct nlm_res  *resp)
{
	struct nlm_host	*host;
	struct nlm_file	*file;

	dprintk("lockd: UNSHARE       called\n");

	resp->cookie = argp->cookie;

	/* Don't accept requests during grace period */
	if (locks_in_grace()) {
		resp->status = nlm_lck_denied_grace_period;
		return rpc_success;
	}

	/* Obtain client and file */
	if ((resp->status = nlmsvc_retrieve_args(rqstp, argp, &host, &file)))
		return resp->status == nlm_drop_reply ? rpc_drop_reply :rpc_success;

	/* Now try to unshare the file */
	resp->status = cast_status(nlmsvc_unshare_file(host, file, argp));

	dprintk("lockd: UNSHARE       status %d\n", ntohl(resp->status));
	nlmsvc_release_host(host);
	nlm_release_file(file);
	return rpc_success;
}

/*
 * NM_LOCK: Create an unmonitored lock
 */
static __be32
nlmsvc_proc_nm_lock(struct svc_rqst *rqstp, struct nlm_args *argp,
				            struct nlm_res  *resp)
{
	dprintk("lockd: NM_LOCK       called\n");

	argp->monitor = 0;		/* just clean the monitor flag */
	return nlmsvc_proc_lock(rqstp, argp, resp);
}

/*
 * FREE_ALL: Release all locks and shares held by client
 */
static __be32
nlmsvc_proc_free_all(struct svc_rqst *rqstp, struct nlm_args *argp,
					     void            *resp)
{
	struct nlm_host	*host;

	/* Obtain client */
	if (nlmsvc_retrieve_args(rqstp, argp, &host, NULL))
		return rpc_success;

	nlmsvc_free_host_resources(host);
	nlmsvc_release_host(host);
	return rpc_success;
}

/*
 * SM_NOTIFY: private callback from statd (not part of official NLM proto)
 */
static __be32
nlmsvc_proc_sm_notify(struct svc_rqst *rqstp, struct nlm_reboot *argp,
					      void	        *resp)
{
	dprintk("lockd: SM_NOTIFY     called\n");

	if (!nlm_privileged_requester(rqstp)) {
		char buf[RPC_MAX_ADDRBUFLEN];
		printk(KERN_WARNING "lockd: rejected NSM callback from %s\n",
				svc_print_addr(rqstp, buf, sizeof(buf)));
		return rpc_system_err;
	}

	nlm_host_rebooted(argp);
	return rpc_success;
}

/*
 * client sent a GRANTED_RES, let's remove the associated block
 */
static __be32
nlmsvc_proc_granted_res(struct svc_rqst *rqstp, struct nlm_res  *argp,
                                                void            *resp)
{
	if (!nlmsvc_ops)
		return rpc_success;

	dprintk("lockd: GRANTED_RES   called\n");

	nlmsvc_grant_reply(&argp->cookie, argp->status);
	return rpc_success;
}

/*
 * NLM Server procedures.
 */

#define nlmsvc_encode_norep	nlmsvc_encode_void
#define nlmsvc_decode_norep	nlmsvc_decode_void
#define nlmsvc_decode_testres	nlmsvc_decode_void
#define nlmsvc_decode_lockres	nlmsvc_decode_void
#define nlmsvc_decode_unlockres	nlmsvc_decode_void
#define nlmsvc_decode_cancelres	nlmsvc_decode_void
#define nlmsvc_decode_grantedres	nlmsvc_decode_void

#define nlmsvc_proc_none	nlmsvc_proc_null
#define nlmsvc_proc_test_res	nlmsvc_proc_null
#define nlmsvc_proc_lock_res	nlmsvc_proc_null
#define nlmsvc_proc_cancel_res	nlmsvc_proc_null
#define nlmsvc_proc_unlock_res	nlmsvc_proc_null

struct nlm_void			{ int dummy; };

#define PROC(name, xargt, xrest, argt, rest, respsize)	\
 { .pc_func	= (svc_procfunc) nlmsvc_proc_##name,	\
   .pc_decode	= (kxdrproc_t) nlmsvc_decode_##xargt,	\
   .pc_encode	= (kxdrproc_t) nlmsvc_encode_##xrest,	\
   .pc_release	= NULL,					\
   .pc_argsize	= sizeof(struct nlm_##argt),		\
   .pc_ressize	= sizeof(struct nlm_##rest),		\
   .pc_xdrressize = respsize,				\
 }

#define	Ck	(1+XDR_QUADLEN(NLM_MAXCOOKIELEN))	/* cookie */
#define	St	1				/* status */
#define	No	(1+1024/4)			/* Net Obj */
#define	Rg	2				/* range - offset + size */

struct svc_procedure		nlmsvc_procedures[] = {
  PROC(null,		void,		void,		void,	void, 1),
  PROC(test,		testargs,	testres,	args,	res, Ck+St+2+No+Rg),
  PROC(lock,		lockargs,	res,		args,	res, Ck+St),
  PROC(cancel,		cancargs,	res,		args,	res, Ck+St),
  PROC(unlock,		unlockargs,	res,		args,	res, Ck+St),
  PROC(granted,		testargs,	res,		args,	res, Ck+St),
  PROC(test_msg,	testargs,	norep,		args,	void, 1),
  PROC(lock_msg,	lockargs,	norep,		args,	void, 1),
  PROC(cancel_msg,	cancargs,	norep,		args,	void, 1),
  PROC(unlock_msg,	unlockargs,	norep,		args,	void, 1),
  PROC(granted_msg,	testargs,	norep,		args,	void, 1),
  PROC(test_res,	testres,	norep,		res,	void, 1),
  PROC(lock_res,	lockres,	norep,		res,	void, 1),
  PROC(cancel_res,	cancelres,	norep,		res,	void, 1),
  PROC(unlock_res,	unlockres,	norep,		res,	void, 1),
  PROC(granted_res,	res,		norep,		res,	void, 1),
  /* statd callback */
  PROC(sm_notify,	reboot,		void,		reboot,	void, 1),
  PROC(none,		void,		void,		void,	void, 1),
  PROC(none,		void,		void,		void,	void, 1),
  PROC(none,		void,		void,		void,	void, 1),
  PROC(share,		shareargs,	shareres,	args,	res, Ck+St+1),
  PROC(unshare,		shareargs,	shareres,	args,	res, Ck+St+1),
  PROC(nm_lock,		lockargs,	res,		args,	res, Ck+St),
  PROC(free_all,	notify,		void,		args,	void, 0),

};
Commit	Line	Data
1da177e4 LT	1	/*
	2	* linux/fs/lockd/svcproc.c
	3	*
	4	* Lockd server procedures. We don't implement the NLM_*_RES
	5	* procedures because we don't use the async procedures.
	6	*
	7	* Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de>
	8	*/
	9
1da177e4 LT	10	#include <linux/types.h>
1da177e4 LT	11	#include <linux/time.h>
1da177e4 LT	12	#include <linux/lockd/lockd.h>
1da177e4 LT	13	#include <linux/lockd/share.h>
1da177e4 LT	14
	15	#define NLMDBG_FACILITY NLMDBG_CLIENT
	16
1da177e4	17	#ifdef CONFIG_LOCKD_V4
52921e02 AV	18	static __be32
52921e02 AV	19	cast_to_nlm(__be32 status, u32 vers)
1da177e4 LT	20	{
	21	/* Note: status is assumed to be in network byte order !!! */
	22	if (vers != 4){
	23	switch (status) {
	24	case nlm_granted:
	25	case nlm_lck_denied:
	26	case nlm_lck_denied_nolocks:
	27	case nlm_lck_blocked:
	28	case nlm_lck_denied_grace_period:
5ea0d750	29	case nlm_drop_reply:
1da177e4 LT	30	break;
	31	case nlm4_deadlock:
	32	status = nlm_lck_denied;
	33	break;
	34	default:
	35	status = nlm_lck_denied_nolocks;
	36	}
	37	}
	38
	39	return (status);
	40	}
	41	#define cast_status(status) (cast_to_nlm(status, rqstp->rq_vers))
	42	#else
	43	#define cast_status(status) (status)
	44	#endif
	45
	46	/*
	47	* Obtain client and file from arguments
	48	*/
52921e02	49	static __be32
1da177e4 LT	50	nlmsvc_retrieve_args(struct svc_rqst rqstp, struct nlm_args argp,
	51	struct nlm_host hostp, struct nlm_file filp)
	52	{
	53	struct nlm_host *host = NULL;
	54	struct nlm_file *file = NULL;
	55	struct nlm_lock *lock = &argp->lock;
52921e02	56	__be32 error = 0;
1da177e4 LT	57
	58	/* nfsd callbacks must have been installed for this procedure */
	59	if (!nlmsvc_ops)
	60	return nlm_lck_denied_nolocks;
	61
	62	/* Obtain host handle */
db4e4c9a	63	if (!(host = nlmsvc_lookup_host(rqstp, lock->caller, lock->len))
977faf39	64	\|\| (argp->monitor && nsm_monitor(host) < 0))
1da177e4 LT	65	goto no_locks;
	66	*hostp = host;
	67
	68	/* Obtain file pointer. Not used by FREE_ALL call. */
	69	if (filp != NULL) {
	70	if ((error = nlm_lookup_file(rqstp, &file, &lock->fh)) != 0)
	71	goto no_locks;
	72	*filp = file;
	73
	74	/* Set up the missing parts of the file_lock structure */
	75	lock->fl.fl_file = file->f_file;
	76	lock->fl.fl_owner = (fl_owner_t) host;
	77	lock->fl.fl_lmops = &nlmsvc_lock_operations;
	78	}
	79
	80	return 0;
	81
	82	no_locks:
67216b94	83	nlmsvc_release_host(host);
d343fce1 N	84	if (error)
d343fce1 N	85	return error;
1da177e4 LT	86	return nlm_lck_denied_nolocks;
	87	}
	88
	89	/*
	90	* NULL: Test for presence of service
	91	*/
7111c66e	92	static __be32
1da177e4 LT	93	nlmsvc_proc_null(struct svc_rqst rqstp, void argp, void *resp)
	94	{
	95	dprintk("lockd: NULL called\n");
	96	return rpc_success;
	97	}
	98
	99	/*
	100	* TEST: Check for conflicting lock
	101	*/
7111c66e	102	static __be32
1da177e4 LT	103	nlmsvc_proc_test(struct svc_rqst rqstp, struct nlm_args argp,
	104	struct nlm_res *resp)
	105	{
	106	struct nlm_host *host;
	107	struct nlm_file *file;
317602f3	108	__be32 rc = rpc_success;
1da177e4 LT	109
	110	dprintk("lockd: TEST called\n");
	111	resp->cookie = argp->cookie;
	112
1da177e4 LT	113	/* Obtain client and file */
1da177e4 LT	114	if ((resp->status = nlmsvc_retrieve_args(rqstp, argp, &host, &file)))
d343fce1	115	return resp->status == nlm_drop_reply ? rpc_drop_reply :rpc_success;
1da177e4 LT	116
1da177e4 LT	117	/* Now check for conflicting locks */
8f920d5e	118	resp->status = cast_status(nlmsvc_testlock(rqstp, file, host, &argp->lock, &resp->lock, &resp->cookie));
5ea0d750	119	if (resp->status == nlm_drop_reply)
b7e6b869 OD	120	rc = rpc_drop_reply;
	121	else
	122	dprintk("lockd: TEST status %d vers %d\n",
	123	ntohl(resp->status), rqstp->rq_vers);
1da177e4	124
67216b94	125	nlmsvc_release_host(host);
1da177e4	126	nlm_release_file(file);
b7e6b869	127	return rc;
1da177e4 LT	128	}
1da177e4 LT	129
7111c66e	130	static __be32
1da177e4 LT	131	nlmsvc_proc_lock(struct svc_rqst rqstp, struct nlm_args argp,
	132	struct nlm_res *resp)
	133	{
	134	struct nlm_host *host;
	135	struct nlm_file *file;
317602f3	136	__be32 rc = rpc_success;
1da177e4 LT	137
	138	dprintk("lockd: LOCK called\n");
	139
	140	resp->cookie = argp->cookie;
	141
1da177e4 LT	142	/* Obtain client and file */
1da177e4 LT	143	if ((resp->status = nlmsvc_retrieve_args(rqstp, argp, &host, &file)))
d343fce1	144	return resp->status == nlm_drop_reply ? rpc_drop_reply :rpc_success;
1da177e4 LT	145
	146	#if 0
	147	/* If supplied state doesn't match current state, we assume it's
	148	* an old request that time-warped somehow. Any error return would
	149	* do in this case because it's irrelevant anyway.
	150	*
	151	* NB: We don't retrieve the remote host's state yet.
	152	*/
	153	if (host->h_nsmstate && host->h_nsmstate != argp->state) {
	154	resp->status = nlm_lck_denied_nolocks;
	155	} else
	156	#endif
	157
	158	/* Now try to lock the file */
6cde4de8	159	resp->status = cast_status(nlmsvc_lock(rqstp, file, host, &argp->lock,
b2b50289 BF	160	argp->block, &argp->cookie,
b2b50289 BF	161	argp->reclaim));
1a8322b2	162	if (resp->status == nlm_drop_reply)
b7e6b869 OD	163	rc = rpc_drop_reply;
	164	else
	165	dprintk("lockd: LOCK status %d\n", ntohl(resp->status));
1da177e4	166
67216b94	167	nlmsvc_release_host(host);
1da177e4	168	nlm_release_file(file);
b7e6b869	169	return rc;
1da177e4 LT	170	}
1da177e4 LT	171
7111c66e	172	static __be32
1da177e4 LT	173	nlmsvc_proc_cancel(struct svc_rqst rqstp, struct nlm_args argp,
	174	struct nlm_res *resp)
	175	{
	176	struct nlm_host *host;
	177	struct nlm_file *file;
	178
	179	dprintk("lockd: CANCEL called\n");
	180
	181	resp->cookie = argp->cookie;
	182
	183	/* Don't accept requests during grace period */
af558e33	184	if (locks_in_grace()) {
1da177e4 LT	185	resp->status = nlm_lck_denied_grace_period;
	186	return rpc_success;
	187	}
	188
	189	/* Obtain client and file */
	190	if ((resp->status = nlmsvc_retrieve_args(rqstp, argp, &host, &file)))
d343fce1	191	return resp->status == nlm_drop_reply ? rpc_drop_reply :rpc_success;
1da177e4 LT	192
	193	/* Try to cancel request. */
	194	resp->status = cast_status(nlmsvc_cancel_blocked(file, &argp->lock));
	195
	196	dprintk("lockd: CANCEL status %d\n", ntohl(resp->status));
67216b94	197	nlmsvc_release_host(host);
1da177e4 LT	198	nlm_release_file(file);
	199	return rpc_success;
	200	}
	201
	202	/*
	203	* UNLOCK: release a lock
	204	*/
7111c66e	205	static __be32
1da177e4 LT	206	nlmsvc_proc_unlock(struct svc_rqst rqstp, struct nlm_args argp,
	207	struct nlm_res *resp)
	208	{
	209	struct nlm_host *host;
	210	struct nlm_file *file;
	211
	212	dprintk("lockd: UNLOCK called\n");
	213
	214	resp->cookie = argp->cookie;
	215
	216	/* Don't accept new lock requests during grace period */
af558e33	217	if (locks_in_grace()) {
1da177e4 LT	218	resp->status = nlm_lck_denied_grace_period;
	219	return rpc_success;
	220	}
	221
	222	/* Obtain client and file */
	223	if ((resp->status = nlmsvc_retrieve_args(rqstp, argp, &host, &file)))
d343fce1	224	return resp->status == nlm_drop_reply ? rpc_drop_reply :rpc_success;
1da177e4 LT	225
	226	/* Now try to remove the lock */
	227	resp->status = cast_status(nlmsvc_unlock(file, &argp->lock));
	228
	229	dprintk("lockd: UNLOCK status %d\n", ntohl(resp->status));
67216b94	230	nlmsvc_release_host(host);
1da177e4 LT	231	nlm_release_file(file);
	232	return rpc_success;
	233	}
	234
	235	/*
	236	* GRANTED: A server calls us to tell that a process' lock request
	237	* was granted
	238	*/
7111c66e	239	static __be32
1da177e4 LT	240	nlmsvc_proc_granted(struct svc_rqst rqstp, struct nlm_args argp,
	241	struct nlm_res *resp)
	242	{
	243	resp->cookie = argp->cookie;
	244
	245	dprintk("lockd: GRANTED called\n");
dcff09f1	246	resp->status = nlmclnt_grant(svc_addr(rqstp), &argp->lock);
1da177e4 LT	247	dprintk("lockd: GRANTED status %d\n", ntohl(resp->status));
	248	return rpc_success;
	249	}
	250
d4716624 TM	251	/*
	252	* This is the generic lockd callback for async RPC calls
	253	*/
	254	static void nlmsvc_callback_exit(struct rpc_task task, void data)
	255	{
c041b5ff	256	dprintk("lockd: %5u callback returned %d\n", task->tk_pid,
d4716624 TM	257	-task->tk_status);
	258	}
	259
7db836d4 CL	260	void nlmsvc_release_call(struct nlm_rqst *call)
	261	{
	262	if (!atomic_dec_and_test(&call->a_count))
	263	return;
67216b94	264	nlmsvc_release_host(call->a_host);
7db836d4 CL	265	kfree(call);
	266	}
	267
d4716624 TM	268	static void nlmsvc_callback_release(void *data)
d4716624 TM	269	{
7db836d4	270	nlmsvc_release_call(data);
d4716624 TM	271	}
	272
	273	static const struct rpc_call_ops nlmsvc_callback_ops = {
	274	.rpc_call_done = nlmsvc_callback_exit,
	275	.rpc_release = nlmsvc_callback_release,
	276	};
	277
1da177e4 LT	278	/*
	279	* `Async' versions of the above service routines. They aren't really,
	280	* because we send the callback before the reply proper. I hope this
	281	* doesn't break any clients.
	282	*/
7111c66e AV	283	static __be32 nlmsvc_callback(struct svc_rqst rqstp, u32 proc, struct nlm_args argp,
7111c66e AV	284	__be32 (func)(struct svc_rqst , struct nlm_args , struct nlm_res ))
1da177e4	285	{
d4716624 TM	286	struct nlm_host *host;
d4716624 TM	287	struct nlm_rqst *call;
7111c66e	288	__be32 stat;
1da177e4	289
db4e4c9a OK	290	host = nlmsvc_lookup_host(rqstp,
	291	argp->lock.caller,
	292	argp->lock.len);
d4716624 TM	293	if (host == NULL)
	294	return rpc_system_err;
	295
	296	call = nlm_alloc_call(host);
	297	if (call == NULL)
	298	return rpc_system_err;
	299
	300	stat = func(rqstp, argp, &call->a_res);
	301	if (stat != 0) {
7db836d4	302	nlmsvc_release_call(call);
d4716624 TM	303	return stat;
d4716624 TM	304	}
1da177e4	305
d4716624 TM	306	call->a_flags = RPC_TASK_ASYNC;
	307	if (nlm_async_reply(call, proc, &nlmsvc_callback_ops) < 0)
	308	return rpc_system_err;
	309	return rpc_success;
1da177e4 LT	310	}
1da177e4 LT	311
7111c66e	312	static __be32 nlmsvc_proc_test_msg(struct svc_rqst rqstp, struct nlm_args argp,
1da177e4 LT	313	void *resp)
1da177e4 LT	314	{
d4716624 TM	315	dprintk("lockd: TEST_MSG called\n");
	316	return nlmsvc_callback(rqstp, NLMPROC_TEST_RES, argp, nlmsvc_proc_test);
	317	}
1da177e4	318
7111c66e	319	static __be32 nlmsvc_proc_lock_msg(struct svc_rqst rqstp, struct nlm_args argp,
d4716624 TM	320	void *resp)
d4716624 TM	321	{
1da177e4	322	dprintk("lockd: LOCK_MSG called\n");
d4716624	323	return nlmsvc_callback(rqstp, NLMPROC_LOCK_RES, argp, nlmsvc_proc_lock);
1da177e4 LT	324	}
1da177e4 LT	325
7111c66e	326	static __be32 nlmsvc_proc_cancel_msg(struct svc_rqst rqstp, struct nlm_args argp,
1da177e4 LT	327	void *resp)
1da177e4 LT	328	{
1da177e4	329	dprintk("lockd: CANCEL_MSG called\n");
d4716624	330	return nlmsvc_callback(rqstp, NLMPROC_CANCEL_RES, argp, nlmsvc_proc_cancel);
1da177e4 LT	331	}
1da177e4 LT	332
7111c66e	333	static __be32
1da177e4 LT	334	nlmsvc_proc_unlock_msg(struct svc_rqst rqstp, struct nlm_args argp,
	335	void *resp)
	336	{
1da177e4	337	dprintk("lockd: UNLOCK_MSG called\n");
d4716624	338	return nlmsvc_callback(rqstp, NLMPROC_UNLOCK_RES, argp, nlmsvc_proc_unlock);
1da177e4 LT	339	}
1da177e4 LT	340
7111c66e	341	static __be32
1da177e4 LT	342	nlmsvc_proc_granted_msg(struct svc_rqst rqstp, struct nlm_args argp,
	343	void *resp)
	344	{
1da177e4	345	dprintk("lockd: GRANTED_MSG called\n");
d4716624	346	return nlmsvc_callback(rqstp, NLMPROC_GRANTED_RES, argp, nlmsvc_proc_granted);
1da177e4 LT	347	}
	348
	349	/*
	350	* SHARE: create a DOS share or alter existing share.
	351	*/
7111c66e	352	static __be32
1da177e4 LT	353	nlmsvc_proc_share(struct svc_rqst rqstp, struct nlm_args argp,
	354	struct nlm_res *resp)
	355	{
	356	struct nlm_host *host;
	357	struct nlm_file *file;
	358
	359	dprintk("lockd: SHARE called\n");
	360
	361	resp->cookie = argp->cookie;
	362
	363	/* Don't accept new lock requests during grace period */
af558e33	364	if (locks_in_grace() && !argp->reclaim) {
1da177e4 LT	365	resp->status = nlm_lck_denied_grace_period;
	366	return rpc_success;
	367	}
	368
	369	/* Obtain client and file */
	370	if ((resp->status = nlmsvc_retrieve_args(rqstp, argp, &host, &file)))
d343fce1	371	return resp->status == nlm_drop_reply ? rpc_drop_reply :rpc_success;
1da177e4 LT	372
	373	/* Now try to create the share */
	374	resp->status = cast_status(nlmsvc_share_file(host, file, argp));
	375
	376	dprintk("lockd: SHARE status %d\n", ntohl(resp->status));
67216b94	377	nlmsvc_release_host(host);
1da177e4 LT	378	nlm_release_file(file);
	379	return rpc_success;
	380	}
	381
	382	/*
	383	* UNSHARE: Release a DOS share.
	384	*/
7111c66e	385	static __be32
1da177e4 LT	386	nlmsvc_proc_unshare(struct svc_rqst rqstp, struct nlm_args argp,
	387	struct nlm_res *resp)
	388	{
	389	struct nlm_host *host;
	390	struct nlm_file *file;
	391
	392	dprintk("lockd: UNSHARE called\n");
	393
	394	resp->cookie = argp->cookie;
	395
	396	/* Don't accept requests during grace period */
af558e33	397	if (locks_in_grace()) {
1da177e4 LT	398	resp->status = nlm_lck_denied_grace_period;
	399	return rpc_success;
	400	}
	401
	402	/* Obtain client and file */
	403	if ((resp->status = nlmsvc_retrieve_args(rqstp, argp, &host, &file)))
d343fce1	404	return resp->status == nlm_drop_reply ? rpc_drop_reply :rpc_success;
1da177e4 LT	405
	406	/* Now try to unshare the file */
	407	resp->status = cast_status(nlmsvc_unshare_file(host, file, argp));
	408
	409	dprintk("lockd: UNSHARE status %d\n", ntohl(resp->status));
67216b94	410	nlmsvc_release_host(host);
1da177e4 LT	411	nlm_release_file(file);
	412	return rpc_success;
	413	}
	414
	415	/*
	416	* NM_LOCK: Create an unmonitored lock
	417	*/
7111c66e	418	static __be32
1da177e4 LT	419	nlmsvc_proc_nm_lock(struct svc_rqst rqstp, struct nlm_args argp,
	420	struct nlm_res *resp)
	421	{
	422	dprintk("lockd: NM_LOCK called\n");
	423
	424	argp->monitor = 0; /* just clean the monitor flag */
	425	return nlmsvc_proc_lock(rqstp, argp, resp);
	426	}
	427
	428	/*
	429	* FREE_ALL: Release all locks and shares held by client
	430	*/
7111c66e	431	static __be32
1da177e4 LT	432	nlmsvc_proc_free_all(struct svc_rqst rqstp, struct nlm_args argp,
	433	void *resp)
	434	{
	435	struct nlm_host *host;
	436
	437	/* Obtain client */
	438	if (nlmsvc_retrieve_args(rqstp, argp, &host, NULL))
	439	return rpc_success;
	440
	441	nlmsvc_free_host_resources(host);
67216b94	442	nlmsvc_release_host(host);
1da177e4 LT	443	return rpc_success;
	444	}
	445
	446	/*
	447	* SM_NOTIFY: private callback from statd (not part of official NLM proto)
	448	*/
7111c66e	449	static __be32
1da177e4 LT	450	nlmsvc_proc_sm_notify(struct svc_rqst rqstp, struct nlm_reboot argp,
	451	void *resp)
	452	{
1da177e4	453	dprintk("lockd: SM_NOTIFY called\n");
b85e4676 CL	454
b85e4676 CL	455	if (!nlm_privileged_requester(rqstp)) {
ad06e4bd CL	456	char buf[RPC_MAX_ADDRBUFLEN];
	457	printk(KERN_WARNING "lockd: rejected NSM callback from %s\n",
	458	svc_print_addr(rqstp, buf, sizeof(buf)));
1da177e4 LT	459	return rpc_system_err;
	460	}
	461
7fefc9cb	462	nlm_host_rebooted(argp);
1da177e4 LT	463	return rpc_success;
	464	}
	465
	466	/*
	467	* client sent a GRANTED_RES, let's remove the associated block
	468	*/
7111c66e	469	static __be32
1da177e4 LT	470	nlmsvc_proc_granted_res(struct svc_rqst rqstp, struct nlm_res argp,
	471	void *resp)
	472	{
	473	if (!nlmsvc_ops)
	474	return rpc_success;
	475
	476	dprintk("lockd: GRANTED_RES called\n");
	477
39be4502	478	nlmsvc_grant_reply(&argp->cookie, argp->status);
1da177e4 LT	479	return rpc_success;
	480	}
	481
1da177e4 LT	482	/*
	483	* NLM Server procedures.
	484	*/
	485
	486	#define nlmsvc_encode_norep nlmsvc_encode_void
	487	#define nlmsvc_decode_norep nlmsvc_decode_void
	488	#define nlmsvc_decode_testres nlmsvc_decode_void
	489	#define nlmsvc_decode_lockres nlmsvc_decode_void
	490	#define nlmsvc_decode_unlockres nlmsvc_decode_void
	491	#define nlmsvc_decode_cancelres nlmsvc_decode_void
	492	#define nlmsvc_decode_grantedres nlmsvc_decode_void
	493
	494	#define nlmsvc_proc_none nlmsvc_proc_null
	495	#define nlmsvc_proc_test_res nlmsvc_proc_null
	496	#define nlmsvc_proc_lock_res nlmsvc_proc_null
	497	#define nlmsvc_proc_cancel_res nlmsvc_proc_null
	498	#define nlmsvc_proc_unlock_res nlmsvc_proc_null
	499
	500	struct nlm_void { int dummy; };
	501
	502	#define PROC(name, xargt, xrest, argt, rest, respsize) \
	503	{ .pc_func = (svc_procfunc) nlmsvc_proc_##name, \
	504	.pc_decode = (kxdrproc_t) nlmsvc_decode_##xargt, \
	505	.pc_encode = (kxdrproc_t) nlmsvc_encode_##xrest, \
	506	.pc_release = NULL, \
	507	.pc_argsize = sizeof(struct nlm_##argt), \
	508	.pc_ressize = sizeof(struct nlm_##rest), \
	509	.pc_xdrressize = respsize, \
	510	}
	511
	512	#define Ck (1+XDR_QUADLEN(NLM_MAXCOOKIELEN)) /* cookie */
	513	#define St 1 /* status */
	514	#define No (1+1024/4) /* Net Obj */
	515	#define Rg 2 /* range - offset + size */
	516
	517	struct svc_procedure nlmsvc_procedures[] = {
	518	PROC(null, void, void, void, void, 1),
	519	PROC(test, testargs, testres, args, res, Ck+St+2+No+Rg),
	520	PROC(lock, lockargs, res, args, res, Ck+St),
	521	PROC(cancel, cancargs, res, args, res, Ck+St),
	522	PROC(unlock, unlockargs, res, args, res, Ck+St),
	523	PROC(granted, testargs, res, args, res, Ck+St),
	524	PROC(test_msg, testargs, norep, args, void, 1),
	525	PROC(lock_msg, lockargs, norep, args, void, 1),
	526	PROC(cancel_msg, cancargs, norep, args, void, 1),
	527	PROC(unlock_msg, unlockargs, norep, args, void, 1),
	528	PROC(granted_msg, testargs, norep, args, void, 1),
	529	PROC(test_res, testres, norep, res, void, 1),
	530	PROC(lock_res, lockres, norep, res, void, 1),
	531	PROC(cancel_res, cancelres, norep, res, void, 1),
	532	PROC(unlock_res, unlockres, norep, res, void, 1),
	533	PROC(granted_res, res, norep, res, void, 1),
	534	/* statd callback */
	535	PROC(sm_notify, reboot, void, reboot, void, 1),
	536	PROC(none, void, void, void, void, 1),
	537	PROC(none, void, void, void, void, 1),
	538	PROC(none, void, void, void, void, 1),
	539	PROC(share, shareargs, shareres, args, res, Ck+St+1),
	540	PROC(unshare, shareargs, shareres, args, res, Ck+St+1),
	541	PROC(nm_lock, lockargs, res, args, res, Ck+St),
	542	PROC(free_all, notify, void, args, void, 0),
	543
	544	};