[net-next-2.6.git] / fs / nfs / callback.c

/*
 * linux/fs/nfs/callback.c
 *
 * Copyright (C) 2004 Trond Myklebust
 *
 * NFSv4 callback handling
 */

#include <linux/completion.h>
#include <linux/ip.h>
#include <linux/module.h>
#include <linux/smp_lock.h>
#include <linux/sunrpc/svc.h>
#include <linux/sunrpc/svcsock.h>
#include <linux/nfs_fs.h>
#include <linux/mutex.h>

#include <net/inet_sock.h>

#include "nfs4_fs.h"
#include "callback.h"
#include "internal.h"

#define NFSDBG_FACILITY NFSDBG_CALLBACK

struct nfs_callback_data {
	unsigned int users;
	struct svc_serv *serv;
	pid_t pid;
	struct completion started;
	struct completion stopped;
};

static struct nfs_callback_data nfs_callback_info;
static DEFINE_MUTEX(nfs_callback_mutex);
static struct svc_program nfs4_callback_program;

unsigned int nfs_callback_set_tcpport;
unsigned short nfs_callback_tcpport;
static const int nfs_set_port_min = 0;
static const int nfs_set_port_max = 65535;

static int param_set_port(const char *val, struct kernel_param *kp)
{
	char *endp;
	int num = simple_strtol(val, &endp, 0);
	if (endp == val || *endp || num < nfs_set_port_min || num > nfs_set_port_max)
		return -EINVAL;
	*((int *)kp->arg) = num;
	return 0;
}

module_param_call(callback_tcpport, param_set_port, param_get_int,
		 &nfs_callback_set_tcpport, 0644);

/*
 * This is the callback kernel thread.
 */
static void nfs_callback_svc(struct svc_rqst *rqstp)
{
	int err;

	__module_get(THIS_MODULE);
	lock_kernel();

	nfs_callback_info.pid = current->pid;
	daemonize("nfsv4-svc");
	/* Process request with signals blocked, but allow SIGKILL.  */
	allow_signal(SIGKILL);

	complete(&nfs_callback_info.started);

	for(;;) {
		if (signalled()) {
			if (nfs_callback_info.users == 0)
				break;
			flush_signals(current);
		}
		/*
		 * Listen for a request on the socket
		 */
		err = svc_recv(rqstp, MAX_SCHEDULE_TIMEOUT);
		if (err == -EAGAIN || err == -EINTR)
			continue;
		if (err < 0) {
			printk(KERN_WARNING
					"%s: terminating on error %d\n",
					__FUNCTION__, -err);
			break;
		}
		dprintk("%s: request from %u.%u.%u.%u\n", __FUNCTION__,
				NIPQUAD(rqstp->rq_addr.sin_addr.s_addr));
		svc_process(rqstp);
	}

	svc_exit_thread(rqstp);
	nfs_callback_info.pid = 0;
	complete(&nfs_callback_info.stopped);
	unlock_kernel();
	module_put_and_exit(0);
}

/*
 * Bring up the server process if it is not already up.
 */
int nfs_callback_up(void)
{
	struct svc_serv *serv;
	int ret = 0;

	lock_kernel();
	mutex_lock(&nfs_callback_mutex);
	if (nfs_callback_info.users++ || nfs_callback_info.pid != 0)
		goto out;
	init_completion(&nfs_callback_info.started);
	init_completion(&nfs_callback_info.stopped);
	serv = svc_create(&nfs4_callback_program, NFS4_CALLBACK_BUFSIZE, NULL);
	ret = -ENOMEM;
	if (!serv)
		goto out_err;

	ret = svc_makesock(serv, IPPROTO_TCP, nfs_callback_set_tcpport,
							SVC_SOCK_ANONYMOUS);
	if (ret <= 0)
		goto out_destroy;
	nfs_callback_tcpport = ret;
	dprintk("Callback port = 0x%x\n", nfs_callback_tcpport);

	ret = svc_create_thread(nfs_callback_svc, serv);
	if (ret < 0)
		goto out_destroy;
	nfs_callback_info.serv = serv;
	wait_for_completion(&nfs_callback_info.started);
out:
	mutex_unlock(&nfs_callback_mutex);
	unlock_kernel();
	return ret;
out_destroy:
	dprintk("Couldn't create callback socket or server thread; err = %d\n",
		ret);
	svc_destroy(serv);
out_err:
	nfs_callback_info.users--;
	goto out;
}

/*
 * Kill the server process if it is not already up.
 */
void nfs_callback_down(void)
{
	lock_kernel();
	mutex_lock(&nfs_callback_mutex);
	nfs_callback_info.users--;
	do {
		if (nfs_callback_info.users != 0 || nfs_callback_info.pid == 0)
			break;
		if (kill_proc(nfs_callback_info.pid, SIGKILL, 1) < 0)
			break;
	} while (wait_for_completion_timeout(&nfs_callback_info.stopped, 5*HZ) == 0);
	mutex_unlock(&nfs_callback_mutex);
	unlock_kernel();
}

static int nfs_callback_authenticate(struct svc_rqst *rqstp)
{
	struct sockaddr_in *addr = &rqstp->rq_addr;
	struct nfs_client *clp;

	/* Don't talk to strangers */
	clp = nfs_find_client(addr, 4);
	if (clp == NULL)
		return SVC_DROP;
	dprintk("%s: %u.%u.%u.%u NFSv4 callback!\n", __FUNCTION__, NIPQUAD(addr->sin_addr));
	nfs_put_client(clp);
	switch (rqstp->rq_authop->flavour) {
		case RPC_AUTH_NULL:
			if (rqstp->rq_proc != CB_NULL)
				return SVC_DENIED;
			break;
		case RPC_AUTH_UNIX:
			break;
		case RPC_AUTH_GSS:
			/* FIXME: RPCSEC_GSS handling? */
		default:
			return SVC_DENIED;
	}
	return SVC_OK;
}

/*
 * Define NFS4 callback program
 */
static struct svc_version *nfs4_callback_version[] = {
	[1] = &nfs4_callback_version1,
};

static struct svc_stat nfs4_callback_stats;

static struct svc_program nfs4_callback_program = {
	.pg_prog = NFS4_CALLBACK,			/* RPC service number */
	.pg_nvers = ARRAY_SIZE(nfs4_callback_version),	/* Number of entries */
	.pg_vers = nfs4_callback_version,		/* version table */
	.pg_name = "NFSv4 callback",			/* service name */
	.pg_class = "nfs",				/* authentication class */
	.pg_stats = &nfs4_callback_stats,
	.pg_authenticate = nfs_callback_authenticate,
};
Commit	Line	Data
1da177e4 LT	1	/*
	2	* linux/fs/nfs/callback.c
	3	*
	4	* Copyright (C) 2004 Trond Myklebust
	5	*
	6	* NFSv4 callback handling
	7	*/
	8
1da177e4 LT	9	#include <linux/completion.h>
	10	#include <linux/ip.h>
	11	#include <linux/module.h>
	12	#include <linux/smp_lock.h>
	13	#include <linux/sunrpc/svc.h>
	14	#include <linux/sunrpc/svcsock.h>
	15	#include <linux/nfs_fs.h>
353ab6e9	16	#include <linux/mutex.h>
14c85021 ACM	17
	18	#include <net/inet_sock.h>
	19
4ce79717	20	#include "nfs4_fs.h"
1da177e4	21	#include "callback.h"
24c8dbbb	22	#include "internal.h"
1da177e4 LT	23
	24	#define NFSDBG_FACILITY NFSDBG_CALLBACK
	25
	26	struct nfs_callback_data {
	27	unsigned int users;
	28	struct svc_serv *serv;
	29	pid_t pid;
	30	struct completion started;
	31	struct completion stopped;
	32	};
	33
	34	static struct nfs_callback_data nfs_callback_info;
353ab6e9	35	static DEFINE_MUTEX(nfs_callback_mutex);
1da177e4 LT	36	static struct svc_program nfs4_callback_program;
1da177e4 LT	37
a72b4422	38	unsigned int nfs_callback_set_tcpport;
1da177e4	39	unsigned short nfs_callback_tcpport;
7d4e2747 DH	40	static const int nfs_set_port_min = 0;
	41	static const int nfs_set_port_max = 65535;
	42
	43	static int param_set_port(const char val, struct kernel_param kp)
	44	{
	45	char *endp;
	46	int num = simple_strtol(val, &endp, 0);
	47	if (endp == val \|\| *endp \|\| num < nfs_set_port_min \|\| num > nfs_set_port_max)
	48	return -EINVAL;
	49	((int )kp->arg) = num;
	50	return 0;
	51	}
	52
	53	module_param_call(callback_tcpport, param_set_port, param_get_int,
	54	&nfs_callback_set_tcpport, 0644);
1da177e4 LT	55
	56	/*
	57	* This is the callback kernel thread.
	58	*/
	59	static void nfs_callback_svc(struct svc_rqst *rqstp)
	60	{
1da177e4 LT	61	int err;
	62
	63	__module_get(THIS_MODULE);
	64	lock_kernel();
	65
	66	nfs_callback_info.pid = current->pid;
	67	daemonize("nfsv4-svc");
	68	/* Process request with signals blocked, but allow SIGKILL. */
	69	allow_signal(SIGKILL);
	70
	71	complete(&nfs_callback_info.started);
	72
1dd761e9 TM	73	for(;;) {
	74	if (signalled()) {
	75	if (nfs_callback_info.users == 0)
	76	break;
	77	flush_signals(current);
	78	}
1da177e4 LT	79	/*
	80	* Listen for a request on the socket
	81	*/
6fb2b47f	82	err = svc_recv(rqstp, MAX_SCHEDULE_TIMEOUT);
1da177e4 LT	83	if (err == -EAGAIN \|\| err == -EINTR)
	84	continue;
	85	if (err < 0) {
	86	printk(KERN_WARNING
	87	"%s: terminating on error %d\n",
	88	__FUNCTION__, -err);
	89	break;
	90	}
	91	dprintk("%s: request from %u.%u.%u.%u\n", __FUNCTION__,
	92	NIPQUAD(rqstp->rq_addr.sin_addr.s_addr));
6fb2b47f	93	svc_process(rqstp);
1da177e4 LT	94	}
1da177e4 LT	95
f25bc349	96	svc_exit_thread(rqstp);
1da177e4 LT	97	nfs_callback_info.pid = 0;
	98	complete(&nfs_callback_info.stopped);
	99	unlock_kernel();
	100	module_put_and_exit(0);
	101	}
	102
	103	/*
	104	* Bring up the server process if it is not already up.
	105	*/
	106	int nfs_callback_up(void)
	107	{
	108	struct svc_serv *serv;
1da177e4 LT	109	int ret = 0;
	110
	111	lock_kernel();
353ab6e9	112	mutex_lock(&nfs_callback_mutex);
1da177e4 LT	113	if (nfs_callback_info.users++ \|\| nfs_callback_info.pid != 0)
	114	goto out;
	115	init_completion(&nfs_callback_info.started);
	116	init_completion(&nfs_callback_info.stopped);
bc591ccf	117	serv = svc_create(&nfs4_callback_program, NFS4_CALLBACK_BUFSIZE, NULL);
1da177e4 LT	118	ret = -ENOMEM;
	119	if (!serv)
	120	goto out_err;
482fb94e CL	121
	122	ret = svc_makesock(serv, IPPROTO_TCP, nfs_callback_set_tcpport,
	123	SVC_SOCK_ANONYMOUS);
	124	if (ret <= 0)
1da177e4	125	goto out_destroy;
482fb94e CL	126	nfs_callback_tcpport = ret;
	127	dprintk("Callback port = 0x%x\n", nfs_callback_tcpport);
	128
1da177e4 LT	129	ret = svc_create_thread(nfs_callback_svc, serv);
	130	if (ret < 0)
	131	goto out_destroy;
	132	nfs_callback_info.serv = serv;
	133	wait_for_completion(&nfs_callback_info.started);
	134	out:
353ab6e9	135	mutex_unlock(&nfs_callback_mutex);
1da177e4 LT	136	unlock_kernel();
	137	return ret;
	138	out_destroy:
482fb94e CL	139	dprintk("Couldn't create callback socket or server thread; err = %d\n",
482fb94e CL	140	ret);
1da177e4 LT	141	svc_destroy(serv);
	142	out_err:
	143	nfs_callback_info.users--;
	144	goto out;
	145	}
	146
	147	/*
	148	* Kill the server process if it is not already up.
	149	*/
5ae1fbce	150	void nfs_callback_down(void)
1da177e4	151	{
1da177e4	152	lock_kernel();
353ab6e9	153	mutex_lock(&nfs_callback_mutex);
1dd761e9 TM	154	nfs_callback_info.users--;
	155	do {
	156	if (nfs_callback_info.users != 0 \|\| nfs_callback_info.pid == 0)
	157	break;
	158	if (kill_proc(nfs_callback_info.pid, SIGKILL, 1) < 0)
	159	break;
	160	} while (wait_for_completion_timeout(&nfs_callback_info.stopped, 5*HZ) == 0);
353ab6e9	161	mutex_unlock(&nfs_callback_mutex);
1da177e4	162	unlock_kernel();
1da177e4 LT	163	}
	164
	165	static int nfs_callback_authenticate(struct svc_rqst *rqstp)
	166	{
24c8dbbb	167	struct sockaddr_in *addr = &rqstp->rq_addr;
adfa6f98	168	struct nfs_client *clp;
1da177e4 LT	169
1da177e4 LT	170	/* Don't talk to strangers */
24c8dbbb	171	clp = nfs_find_client(addr, 4);
1da177e4 LT	172	if (clp == NULL)
1da177e4 LT	173	return SVC_DROP;
24c8dbbb DH	174	dprintk("%s: %u.%u.%u.%u NFSv4 callback!\n", __FUNCTION__, NIPQUAD(addr->sin_addr));
24c8dbbb DH	175	nfs_put_client(clp);
1da177e4 LT	176	switch (rqstp->rq_authop->flavour) {
	177	case RPC_AUTH_NULL:
	178	if (rqstp->rq_proc != CB_NULL)
	179	return SVC_DENIED;
	180	break;
	181	case RPC_AUTH_UNIX:
	182	break;
	183	case RPC_AUTH_GSS:
	184	/* FIXME: RPCSEC_GSS handling? */
	185	default:
	186	return SVC_DENIED;
	187	}
	188	return SVC_OK;
	189	}
	190
	191	/*
	192	* Define NFS4 callback program
	193	*/
1da177e4 LT	194	static struct svc_version *nfs4_callback_version[] = {
	195	[1] = &nfs4_callback_version1,
	196	};
	197
	198	static struct svc_stat nfs4_callback_stats;
	199
	200	static struct svc_program nfs4_callback_program = {
	201	.pg_prog = NFS4_CALLBACK, /* RPC service number */
	202	.pg_nvers = ARRAY_SIZE(nfs4_callback_version), /* Number of entries */
	203	.pg_vers = nfs4_callback_version, /* version table */
	204	.pg_name = "NFSv4 callback", /* service name */
	205	.pg_class = "nfs", /* authentication class */
	206	.pg_stats = &nfs4_callback_stats,
	207	.pg_authenticate = nfs_callback_authenticate,
	208	};