[deliverable/linux.git] / fs / nfsd / nfs4idmap.c

/*
 *  fs/nfsd/nfs4idmap.c
 *
 *  Mapping of UID/GIDs to name and vice versa.
 *
 *  Copyright (c) 2002, 2003 The Regents of the University of
 *  Michigan.  All rights reserved.
 *
 *  Marius Aamodt Eriksen <marius@umich.edu>
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions
 *  are met:
 *
 *  1. Redistributions of source code must retain the above copyright
 *     notice, this list of conditions and the following disclaimer.
 *  2. Redistributions in binary form must reproduce the above copyright
 *     notice, this list of conditions and the following disclaimer in the
 *     documentation and/or other materials provided with the distribution.
 *  3. Neither the name of the University nor the names of its
 *     contributors may be used to endorse or promote products derived
 *     from this software without specific prior written permission.
 *
 *  THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED
 *  WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 *  MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 *  DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 *  FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 *  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 *  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
 *  BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 *  LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 *  NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 *  SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <linux/module.h>
#include <linux/init.h>

#include <linux/mm.h>
#include <linux/utsname.h>
#include <linux/errno.h>
#include <linux/string.h>
#include <linux/sunrpc/clnt.h>
#include <linux/nfs.h>
#include <linux/nfs4.h>
#include <linux/nfs_fs.h>
#include <linux/nfs_page.h>
#include <linux/sunrpc/cache.h>
#include <linux/nfsd_idmap.h>
#include <linux/list.h>
#include <linux/time.h>
#include <linux/seq_file.h>
#include <linux/sunrpc/svcauth.h>

/*
 * Cache entry
 */

/*
 * XXX we know that IDMAP_NAMESZ < PAGE_SIZE, but it's ugly to rely on
 * that.
 */

#define IDMAP_TYPE_USER  0
#define IDMAP_TYPE_GROUP 1

struct ent {
	struct cache_head h;
	int               type;		       /* User / Group */
	uid_t             id;
	char              name[IDMAP_NAMESZ];
	char              authname[IDMAP_NAMESZ];
};

/* Common entry handling */

#define ENT_HASHBITS          8
#define ENT_HASHMAX           (1 << ENT_HASHBITS)
#define ENT_HASHMASK          (ENT_HASHMAX - 1)

static void
ent_init(struct cache_head *cnew, struct cache_head *citm)
{
	struct ent *new = container_of(cnew, struct ent, h);
	struct ent *itm = container_of(citm, struct ent, h);

	new->id = itm->id;
	new->type = itm->type;

	strlcpy(new->name, itm->name, sizeof(new->name));
	strlcpy(new->authname, itm->authname, sizeof(new->name));
}

static void
ent_put(struct kref *ref)
{
	struct ent *map = container_of(ref, struct ent, h.ref);
	kfree(map);
}

static struct cache_head *
ent_alloc(void)
{
	struct ent *e = kmalloc(sizeof(*e), GFP_KERNEL);
	if (e)
		return &e->h;
	else
		return NULL;
}

/*
 * ID -> Name cache
 */

static struct cache_head *idtoname_table[ENT_HASHMAX];

static uint32_t
idtoname_hash(struct ent *ent)
{
	uint32_t hash;

	hash = hash_str(ent->authname, ENT_HASHBITS);
	hash = hash_long(hash ^ ent->id, ENT_HASHBITS);

	/* Flip LSB for user/group */
	if (ent->type == IDMAP_TYPE_GROUP)
		hash ^= 1;

	return hash;
}

static void
idtoname_request(struct cache_detail *cd, struct cache_head *ch, char **bpp,
    int *blen)
{
 	struct ent *ent = container_of(ch, struct ent, h);
	char idstr[11];

	qword_add(bpp, blen, ent->authname);
	snprintf(idstr, sizeof(idstr), "%u", ent->id);
	qword_add(bpp, blen, ent->type == IDMAP_TYPE_GROUP ? "group" : "user");
	qword_add(bpp, blen, idstr);

	(*bpp)[-1] = '\n';
}

static int
idtoname_upcall(struct cache_detail *cd, struct cache_head *ch)
{
	return sunrpc_cache_pipe_upcall(cd, ch, idtoname_request);
}

static int
idtoname_match(struct cache_head *ca, struct cache_head *cb)
{
	struct ent *a = container_of(ca, struct ent, h);
	struct ent *b = container_of(cb, struct ent, h);

	return (a->id == b->id && a->type == b->type &&
	    strcmp(a->authname, b->authname) == 0);
}

static int
idtoname_show(struct seq_file *m, struct cache_detail *cd, struct cache_head *h)
{
	struct ent *ent;

	if (h == NULL) {
		seq_puts(m, "#domain type id [name]\n");
		return 0;
	}
	ent = container_of(h, struct ent, h);
	seq_printf(m, "%s %s %u", ent->authname,
			ent->type == IDMAP_TYPE_GROUP ? "group" : "user",
			ent->id);
	if (test_bit(CACHE_VALID, &h->flags))
		seq_printf(m, " %s", ent->name);
	seq_printf(m, "\n");
	return 0;
}

static void
warn_no_idmapd(struct cache_detail *detail, int has_died)
{
	printk("nfsd: nfsv4 idmapping failing: has idmapd %s?\n",
			has_died ? "died" : "not been started");
}


static int         idtoname_parse(struct cache_detail *, char *, int);
static struct ent *idtoname_lookup(struct ent *);
static struct ent *idtoname_update(struct ent *, struct ent *);

static struct cache_detail idtoname_cache = {
	.owner		= THIS_MODULE,
	.hash_size	= ENT_HASHMAX,
	.hash_table	= idtoname_table,
	.name		= "nfs4.idtoname",
	.cache_put	= ent_put,
	.cache_upcall	= idtoname_upcall,
	.cache_parse	= idtoname_parse,
	.cache_show	= idtoname_show,
	.warn_no_listener = warn_no_idmapd,
	.match		= idtoname_match,
	.init		= ent_init,
	.update		= ent_init,
	.alloc		= ent_alloc,
};

static int
idtoname_parse(struct cache_detail *cd, char *buf, int buflen)
{
	struct ent ent, *res;
	char *buf1, *bp;
	int len;
	int error = -EINVAL;

	if (buf[buflen - 1] != '\n')
		return (-EINVAL);
	buf[buflen - 1]= '\0';

	buf1 = kmalloc(PAGE_SIZE, GFP_KERNEL);
	if (buf1 == NULL)
		return (-ENOMEM);

	memset(&ent, 0, sizeof(ent));

	/* Authentication name */
	if (qword_get(&buf, buf1, PAGE_SIZE) <= 0)
		goto out;
	memcpy(ent.authname, buf1, sizeof(ent.authname));

	/* Type */
	if (qword_get(&buf, buf1, PAGE_SIZE) <= 0)
		goto out;
	ent.type = strcmp(buf1, "user") == 0 ?
		IDMAP_TYPE_USER : IDMAP_TYPE_GROUP;

	/* ID */
	if (qword_get(&buf, buf1, PAGE_SIZE) <= 0)
		goto out;
	ent.id = simple_strtoul(buf1, &bp, 10);
	if (bp == buf1)
		goto out;

	/* expiry */
	ent.h.expiry_time = get_expiry(&buf);
	if (ent.h.expiry_time == 0)
		goto out;

	error = -ENOMEM;
	res = idtoname_lookup(&ent);
	if (!res)
		goto out;

	/* Name */
	error = -EINVAL;
	len = qword_get(&buf, buf1, PAGE_SIZE);
	if (len < 0)
		goto out;
	if (len == 0)
		set_bit(CACHE_NEGATIVE, &ent.h.flags);
	else if (len >= IDMAP_NAMESZ)
		goto out;
	else
		memcpy(ent.name, buf1, sizeof(ent.name));
	error = -ENOMEM;
	res = idtoname_update(&ent, res);
	if (res == NULL)
		goto out;

	cache_put(&res->h, &idtoname_cache);

	error = 0;
out:
	kfree(buf1);

	return error;
}


static struct ent *
idtoname_lookup(struct ent *item)
{
	struct cache_head *ch = sunrpc_cache_lookup(&idtoname_cache,
						    &item->h,
						    idtoname_hash(item));
	if (ch)
		return container_of(ch, struct ent, h);
	else
		return NULL;
}

static struct ent *
idtoname_update(struct ent *new, struct ent *old)
{
	struct cache_head *ch = sunrpc_cache_update(&idtoname_cache,
						    &new->h, &old->h,
						    idtoname_hash(new));
	if (ch)
		return container_of(ch, struct ent, h);
	else
		return NULL;
}


/*
 * Name -> ID cache
 */

static struct cache_head *nametoid_table[ENT_HASHMAX];

static inline int
nametoid_hash(struct ent *ent)
{
	return hash_str(ent->name, ENT_HASHBITS);
}

static void
nametoid_request(struct cache_detail *cd, struct cache_head *ch, char **bpp,
    int *blen)
{
 	struct ent *ent = container_of(ch, struct ent, h);

	qword_add(bpp, blen, ent->authname);
	qword_add(bpp, blen, ent->type == IDMAP_TYPE_GROUP ? "group" : "user");
	qword_add(bpp, blen, ent->name);

	(*bpp)[-1] = '\n';
}

static int
nametoid_upcall(struct cache_detail *cd, struct cache_head *ch)
{
	return sunrpc_cache_pipe_upcall(cd, ch, nametoid_request);
}

static int
nametoid_match(struct cache_head *ca, struct cache_head *cb)
{
	struct ent *a = container_of(ca, struct ent, h);
	struct ent *b = container_of(cb, struct ent, h);

	return (a->type == b->type && strcmp(a->name, b->name) == 0 &&
	    strcmp(a->authname, b->authname) == 0);
}

static int
nametoid_show(struct seq_file *m, struct cache_detail *cd, struct cache_head *h)
{
	struct ent *ent;

	if (h == NULL) {
		seq_puts(m, "#domain type name [id]\n");
		return 0;
	}
	ent = container_of(h, struct ent, h);
	seq_printf(m, "%s %s %s", ent->authname,
			ent->type == IDMAP_TYPE_GROUP ? "group" : "user",
			ent->name);
	if (test_bit(CACHE_VALID, &h->flags))
		seq_printf(m, " %u", ent->id);
	seq_printf(m, "\n");
	return 0;
}

static struct ent *nametoid_lookup(struct ent *);
static struct ent *nametoid_update(struct ent *, struct ent *);
static int         nametoid_parse(struct cache_detail *, char *, int);

static struct cache_detail nametoid_cache = {
	.owner		= THIS_MODULE,
	.hash_size	= ENT_HASHMAX,
	.hash_table	= nametoid_table,
	.name		= "nfs4.nametoid",
	.cache_put	= ent_put,
	.cache_upcall	= nametoid_upcall,
	.cache_parse	= nametoid_parse,
	.cache_show	= nametoid_show,
	.warn_no_listener = warn_no_idmapd,
	.match		= nametoid_match,
	.init		= ent_init,
	.update		= ent_init,
	.alloc		= ent_alloc,
};

static int
nametoid_parse(struct cache_detail *cd, char *buf, int buflen)
{
	struct ent ent, *res;
	char *buf1;
	int error = -EINVAL;

	if (buf[buflen - 1] != '\n')
		return (-EINVAL);
	buf[buflen - 1]= '\0';

	buf1 = kmalloc(PAGE_SIZE, GFP_KERNEL);
	if (buf1 == NULL)
		return (-ENOMEM);

	memset(&ent, 0, sizeof(ent));

	/* Authentication name */
	if (qword_get(&buf, buf1, PAGE_SIZE) <= 0)
		goto out;
	memcpy(ent.authname, buf1, sizeof(ent.authname));

	/* Type */
	if (qword_get(&buf, buf1, PAGE_SIZE) <= 0)
		goto out;
	ent.type = strcmp(buf1, "user") == 0 ?
		IDMAP_TYPE_USER : IDMAP_TYPE_GROUP;

	/* Name */
	error = qword_get(&buf, buf1, PAGE_SIZE);
	if (error <= 0 || error >= IDMAP_NAMESZ)
		goto out;
	memcpy(ent.name, buf1, sizeof(ent.name));

	/* expiry */
	ent.h.expiry_time = get_expiry(&buf);
	if (ent.h.expiry_time == 0)
		goto out;

	/* ID */
	error = get_int(&buf, &ent.id);
	if (error == -EINVAL)
		goto out;
	if (error == -ENOENT)
		set_bit(CACHE_NEGATIVE, &ent.h.flags);

	error = -ENOMEM;
	res = nametoid_lookup(&ent);
	if (res == NULL)
		goto out;
	res = nametoid_update(&ent, res);
	if (res == NULL)
		goto out;

	cache_put(&res->h, &nametoid_cache);
	error = 0;
out:
	kfree(buf1);

	return (error);
}


static struct ent *
nametoid_lookup(struct ent *item)
{
	struct cache_head *ch = sunrpc_cache_lookup(&nametoid_cache,
						    &item->h,
						    nametoid_hash(item));
	if (ch)
		return container_of(ch, struct ent, h);
	else
		return NULL;
}

static struct ent *
nametoid_update(struct ent *new, struct ent *old)
{
	struct cache_head *ch = sunrpc_cache_update(&nametoid_cache,
						    &new->h, &old->h,
						    nametoid_hash(new));
	if (ch)
		return container_of(ch, struct ent, h);
	else
		return NULL;
}

/*
 * Exported API
 */

int
nfsd_idmap_init(void)
{
	int rv;

	rv = cache_register(&idtoname_cache);
	if (rv)
		return rv;
	rv = cache_register(&nametoid_cache);
	if (rv)
		cache_unregister(&idtoname_cache);
	return rv;
}

void
nfsd_idmap_shutdown(void)
{
	cache_unregister(&idtoname_cache);
	cache_unregister(&nametoid_cache);
}

/*
 * Deferred request handling
 */

struct idmap_defer_req {
       struct cache_req		req;
       struct cache_deferred_req deferred_req;
       wait_queue_head_t	waitq;
       atomic_t			count;
};

static inline void
put_mdr(struct idmap_defer_req *mdr)
{
	if (atomic_dec_and_test(&mdr->count))
		kfree(mdr);
}

static inline void
get_mdr(struct idmap_defer_req *mdr)
{
	atomic_inc(&mdr->count);
}

static void
idmap_revisit(struct cache_deferred_req *dreq, int toomany)
{
	struct idmap_defer_req *mdr =
		container_of(dreq, struct idmap_defer_req, deferred_req);

	wake_up(&mdr->waitq);
	put_mdr(mdr);
}

static struct cache_deferred_req *
idmap_defer(struct cache_req *req)
{
	struct idmap_defer_req *mdr =
		container_of(req, struct idmap_defer_req, req);

	mdr->deferred_req.revisit = idmap_revisit;
	get_mdr(mdr);
	return (&mdr->deferred_req);
}

static inline int
do_idmap_lookup(struct ent *(*lookup_fn)(struct ent *), struct ent *key,
		struct cache_detail *detail, struct ent **item,
		struct idmap_defer_req *mdr)
{
	*item = lookup_fn(key);
	if (!*item)
		return -ENOMEM;
	return cache_check(detail, &(*item)->h, &mdr->req);
}

static inline int
do_idmap_lookup_nowait(struct ent *(*lookup_fn)(struct ent *),
			struct ent *key, struct cache_detail *detail,
			struct ent **item)
{
	int ret = -ENOMEM;

	*item = lookup_fn(key);
	if (!*item)
		goto out_err;
	ret = -ETIMEDOUT;
	if (!test_bit(CACHE_VALID, &(*item)->h.flags)
			|| (*item)->h.expiry_time < get_seconds()
			|| detail->flush_time > (*item)->h.last_refresh)
		goto out_put;
	ret = -ENOENT;
	if (test_bit(CACHE_NEGATIVE, &(*item)->h.flags))
		goto out_put;
	return 0;
out_put:
	cache_put(&(*item)->h, detail);
out_err:
	*item = NULL;
	return ret;
}

static int
idmap_lookup(struct svc_rqst *rqstp,
		struct ent *(*lookup_fn)(struct ent *), struct ent *key,
		struct cache_detail *detail, struct ent **item)
{
	struct idmap_defer_req *mdr;
	int ret;

	mdr = kzalloc(sizeof(*mdr), GFP_KERNEL);
	if (!mdr)
		return -ENOMEM;
	atomic_set(&mdr->count, 1);
	init_waitqueue_head(&mdr->waitq);
	mdr->req.defer = idmap_defer;
	ret = do_idmap_lookup(lookup_fn, key, detail, item, mdr);
	if (ret == -EAGAIN) {
		wait_event_interruptible_timeout(mdr->waitq,
			test_bit(CACHE_VALID, &(*item)->h.flags), 1 * HZ);
		ret = do_idmap_lookup_nowait(lookup_fn, key, detail, item);
	}
	put_mdr(mdr);
	return ret;
}

static char *
rqst_authname(struct svc_rqst *rqstp)
{
	struct auth_domain *clp;

	clp = rqstp->rq_gssclient ? rqstp->rq_gssclient : rqstp->rq_client;
	return clp->name;
}

static int
idmap_name_to_id(struct svc_rqst *rqstp, int type, const char *name, u32 namelen,
		uid_t *id)
{
	struct ent *item, key = {
		.type = type,
	};
	int ret;

	if (namelen + 1 > sizeof(key.name))
		return -EINVAL;
	memcpy(key.name, name, namelen);
	key.name[namelen] = '\0';
	strlcpy(key.authname, rqst_authname(rqstp), sizeof(key.authname));
	ret = idmap_lookup(rqstp, nametoid_lookup, &key, &nametoid_cache, &item);
	if (ret == -ENOENT)
		ret = -ESRCH; /* nfserr_badname */
	if (ret)
		return ret;
	*id = item->id;
	cache_put(&item->h, &nametoid_cache);
	return 0;
}

static int
idmap_id_to_name(struct svc_rqst *rqstp, int type, uid_t id, char *name)
{
	struct ent *item, key = {
		.id = id,
		.type = type,
	};
	int ret;

	strlcpy(key.authname, rqst_authname(rqstp), sizeof(key.authname));
	ret = idmap_lookup(rqstp, idtoname_lookup, &key, &idtoname_cache, &item);
	if (ret == -ENOENT)
		return sprintf(name, "%u", id);
	if (ret)
		return ret;
	ret = strlen(item->name);
	BUG_ON(ret > IDMAP_NAMESZ);
	memcpy(name, item->name, ret);
	cache_put(&item->h, &idtoname_cache);
	return ret;
}

int
nfsd_map_name_to_uid(struct svc_rqst *rqstp, const char *name, size_t namelen,
		__u32 *id)
{
	return idmap_name_to_id(rqstp, IDMAP_TYPE_USER, name, namelen, id);
}

int
nfsd_map_name_to_gid(struct svc_rqst *rqstp, const char *name, size_t namelen,
		__u32 *id)
{
	return idmap_name_to_id(rqstp, IDMAP_TYPE_GROUP, name, namelen, id);
}

int
nfsd_map_uid_to_name(struct svc_rqst *rqstp, __u32 id, char *name)
{
	return idmap_id_to_name(rqstp, IDMAP_TYPE_USER, id, name);
}

int
nfsd_map_gid_to_name(struct svc_rqst *rqstp, __u32 id, char *name)
{
	return idmap_id_to_name(rqstp, IDMAP_TYPE_GROUP, id, name);
}
Commit	Line	Data
1da177e4 LT	1	/*
	2	* fs/nfsd/nfs4idmap.c
	3	*
	4	* Mapping of UID/GIDs to name and vice versa.
	5	*
	6	* Copyright (c) 2002, 2003 The Regents of the University of
	7	* Michigan. All rights reserved.
	8	*
	9	* Marius Aamodt Eriksen <marius@umich.edu>
	10	*
	11	* Redistribution and use in source and binary forms, with or without
	12	* modification, are permitted provided that the following conditions
	13	* are met:
	14	*
	15	* 1. Redistributions of source code must retain the above copyright
	16	* notice, this list of conditions and the following disclaimer.
	17	* 2. Redistributions in binary form must reproduce the above copyright
	18	* notice, this list of conditions and the following disclaimer in the
	19	* documentation and/or other materials provided with the distribution.
	20	* 3. Neither the name of the University nor the names of its
	21	* contributors may be used to endorse or promote products derived
	22	* from this software without specific prior written permission.
	23	*
	24	* THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED
	25	* WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
	26	* MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
	27	* DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
	28	* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
	29	* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
	30	* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
	31	* BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
	32	* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
	33	* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
	34	* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
	35	*/
	36
1da177e4 LT	37	#include <linux/module.h>
	38	#include <linux/init.h>
	39
	40	#include <linux/mm.h>
	41	#include <linux/utsname.h>
	42	#include <linux/errno.h>
	43	#include <linux/string.h>
	44	#include <linux/sunrpc/clnt.h>
	45	#include <linux/nfs.h>
	46	#include <linux/nfs4.h>
	47	#include <linux/nfs_fs.h>
	48	#include <linux/nfs_page.h>
1da177e4 LT	49	#include <linux/sunrpc/cache.h>
	50	#include <linux/nfsd_idmap.h>
	51	#include <linux/list.h>
1da177e4 LT	52	#include <linux/time.h>
	53	#include <linux/seq_file.h>
	54	#include <linux/sunrpc/svcauth.h>
	55
	56	/*
	57	* Cache entry
	58	*/
	59
	60	/*
	61	* XXX we know that IDMAP_NAMESZ < PAGE_SIZE, but it's ugly to rely on
	62	* that.
	63	*/
	64
	65	#define IDMAP_TYPE_USER 0
	66	#define IDMAP_TYPE_GROUP 1
	67
	68	struct ent {
	69	struct cache_head h;
	70	int type; /* User / Group */
	71	uid_t id;
	72	char name[IDMAP_NAMESZ];
	73	char authname[IDMAP_NAMESZ];
	74	};
	75
1da177e4 LT	76	/* Common entry handling */
	77
	78	#define ENT_HASHBITS 8
	79	#define ENT_HASHMAX (1 << ENT_HASHBITS)
	80	#define ENT_HASHMASK (ENT_HASHMAX - 1)
	81
f9ecc921 N	82	static void
f9ecc921 N	83	ent_init(struct cache_head cnew, struct cache_head citm)
1da177e4	84	{
f9ecc921 N	85	struct ent *new = container_of(cnew, struct ent, h);
	86	struct ent *itm = container_of(citm, struct ent, h);
	87
1da177e4 LT	88	new->id = itm->id;
	89	new->type = itm->type;
	90
	91	strlcpy(new->name, itm->name, sizeof(new->name));
	92	strlcpy(new->authname, itm->authname, sizeof(new->name));
	93	}
	94
fd39ca9a	95	static void
baab935f	96	ent_put(struct kref *ref)
1da177e4	97	{
baab935f N	98	struct ent *map = container_of(ref, struct ent, h.ref);
baab935f N	99	kfree(map);
1da177e4 LT	100	}
1da177e4 LT	101
f9ecc921 N	102	static struct cache_head *
	103	ent_alloc(void)
	104	{
	105	struct ent e = kmalloc(sizeof(e), GFP_KERNEL);
	106	if (e)
	107	return &e->h;
	108	else
	109	return NULL;
	110	}
	111
1da177e4 LT	112	/*
	113	* ID -> Name cache
	114	*/
	115
	116	static struct cache_head *idtoname_table[ENT_HASHMAX];
	117
	118	static uint32_t
	119	idtoname_hash(struct ent *ent)
	120	{
	121	uint32_t hash;
	122
	123	hash = hash_str(ent->authname, ENT_HASHBITS);
	124	hash = hash_long(hash ^ ent->id, ENT_HASHBITS);
	125
	126	/* Flip LSB for user/group */
	127	if (ent->type == IDMAP_TYPE_GROUP)
	128	hash ^= 1;
	129
	130	return hash;
	131	}
	132
	133	static void
	134	idtoname_request(struct cache_detail cd, struct cache_head ch, char **bpp,
	135	int *blen)
	136	{
	137	struct ent *ent = container_of(ch, struct ent, h);
	138	char idstr[11];
	139
	140	qword_add(bpp, blen, ent->authname);
0a725fc4	141	snprintf(idstr, sizeof(idstr), "%u", ent->id);
1da177e4 LT	142	qword_add(bpp, blen, ent->type == IDMAP_TYPE_GROUP ? "group" : "user");
	143	qword_add(bpp, blen, idstr);
	144
	145	(*bpp)[-1] = '\n';
	146	}
	147
bc74b4f5 TM	148	static int
	149	idtoname_upcall(struct cache_detail cd, struct cache_head ch)
	150	{
	151	return sunrpc_cache_pipe_upcall(cd, ch, idtoname_request);
	152	}
	153
f9ecc921 N	154	static int
f9ecc921 N	155	idtoname_match(struct cache_head ca, struct cache_head cb)
1da177e4	156	{
f9ecc921 N	157	struct ent *a = container_of(ca, struct ent, h);
	158	struct ent *b = container_of(cb, struct ent, h);
	159
1da177e4 LT	160	return (a->id == b->id && a->type == b->type &&
	161	strcmp(a->authname, b->authname) == 0);
	162	}
	163
	164	static int
	165	idtoname_show(struct seq_file m, struct cache_detail cd, struct cache_head *h)
	166	{
	167	struct ent *ent;
	168
	169	if (h == NULL) {
	170	seq_puts(m, "#domain type id [name]\n");
	171	return 0;
	172	}
	173	ent = container_of(h, struct ent, h);
0a725fc4	174	seq_printf(m, "%s %s %u", ent->authname,
1da177e4 LT	175	ent->type == IDMAP_TYPE_GROUP ? "group" : "user",
	176	ent->id);
	177	if (test_bit(CACHE_VALID, &h->flags))
	178	seq_printf(m, " %s", ent->name);
	179	seq_printf(m, "\n");
	180	return 0;
	181	}
	182
	183	static void
2da8ca26	184	warn_no_idmapd(struct cache_detail *detail, int has_died)
1da177e4 LT	185	{
1da177e4 LT	186	printk("nfsd: nfsv4 idmapping failing: has idmapd %s?\n",
2da8ca26	187	has_died ? "died" : "not been started");
1da177e4 LT	188	}
	189
	190
	191	static int idtoname_parse(struct cache_detail , char , int);
f9ecc921 N	192	static struct ent idtoname_lookup(struct ent );
f9ecc921 N	193	static struct ent idtoname_update(struct ent , struct ent *);
1da177e4	194
fd39ca9a	195	static struct cache_detail idtoname_cache = {
f35279d3	196	.owner = THIS_MODULE,
1da177e4 LT	197	.hash_size = ENT_HASHMAX,
	198	.hash_table = idtoname_table,
	199	.name = "nfs4.idtoname",
	200	.cache_put = ent_put,
bc74b4f5	201	.cache_upcall = idtoname_upcall,
1da177e4 LT	202	.cache_parse = idtoname_parse,
	203	.cache_show = idtoname_show,
	204	.warn_no_listener = warn_no_idmapd,
f9ecc921 N	205	.match = idtoname_match,
	206	.init = ent_init,
	207	.update = ent_init,
	208	.alloc = ent_alloc,
1da177e4 LT	209	};
1da177e4 LT	210
a254b246	211	static int
1da177e4 LT	212	idtoname_parse(struct cache_detail cd, char buf, int buflen)
	213	{
	214	struct ent ent, *res;
	215	char buf1, bp;
c9b6cbe5	216	int len;
1da177e4 LT	217	int error = -EINVAL;
	218
	219	if (buf[buflen - 1] != '\n')
	220	return (-EINVAL);
	221	buf[buflen - 1]= '\0';
	222
	223	buf1 = kmalloc(PAGE_SIZE, GFP_KERNEL);
	224	if (buf1 == NULL)
	225	return (-ENOMEM);
	226
	227	memset(&ent, 0, sizeof(ent));
	228
	229	/* Authentication name */
	230	if (qword_get(&buf, buf1, PAGE_SIZE) <= 0)
	231	goto out;
	232	memcpy(ent.authname, buf1, sizeof(ent.authname));
	233
	234	/* Type */
	235	if (qword_get(&buf, buf1, PAGE_SIZE) <= 0)
	236	goto out;
	237	ent.type = strcmp(buf1, "user") == 0 ?
	238	IDMAP_TYPE_USER : IDMAP_TYPE_GROUP;
	239
	240	/* ID */
	241	if (qword_get(&buf, buf1, PAGE_SIZE) <= 0)
	242	goto out;
	243	ent.id = simple_strtoul(buf1, &bp, 10);
	244	if (bp == buf1)
	245	goto out;
	246
	247	/* expiry */
	248	ent.h.expiry_time = get_expiry(&buf);
	249	if (ent.h.expiry_time == 0)
	250	goto out;
	251
f9ecc921 N	252	error = -ENOMEM;
	253	res = idtoname_lookup(&ent);
	254	if (!res)
	255	goto out;
	256
1da177e4	257	/* Name */
c9b6cbe5 BF	258	error = -EINVAL;
	259	len = qword_get(&buf, buf1, PAGE_SIZE);
	260	if (len < 0)
1da177e4	261	goto out;
c9b6cbe5	262	if (len == 0)
1da177e4	263	set_bit(CACHE_NEGATIVE, &ent.h.flags);
d4395e03 BF	264	else if (len >= IDMAP_NAMESZ)
	265	goto out;
	266	else
1da177e4	267	memcpy(ent.name, buf1, sizeof(ent.name));
1da177e4	268	error = -ENOMEM;
f9ecc921 N	269	res = idtoname_update(&ent, res);
f9ecc921 N	270	if (res == NULL)
1da177e4 LT	271	goto out;
1da177e4 LT	272
baab935f	273	cache_put(&res->h, &idtoname_cache);
1da177e4 LT	274
	275	error = 0;
	276	out:
	277	kfree(buf1);
	278
	279	return error;
	280	}
	281
f9ecc921 N	282
	283	static struct ent *
	284	idtoname_lookup(struct ent *item)
	285	{
	286	struct cache_head *ch = sunrpc_cache_lookup(&idtoname_cache,
	287	&item->h,
	288	idtoname_hash(item));
	289	if (ch)
	290	return container_of(ch, struct ent, h);
	291	else
	292	return NULL;
	293	}
	294
	295	static struct ent *
	296	idtoname_update(struct ent new, struct ent old)
	297	{
	298	struct cache_head *ch = sunrpc_cache_update(&idtoname_cache,
	299	&new->h, &old->h,
	300	idtoname_hash(new));
	301	if (ch)
	302	return container_of(ch, struct ent, h);
	303	else
	304	return NULL;
	305	}
	306
1da177e4 LT	307
	308	/*
	309	* Name -> ID cache
	310	*/
	311
	312	static struct cache_head *nametoid_table[ENT_HASHMAX];
	313
	314	static inline int
	315	nametoid_hash(struct ent *ent)
	316	{
	317	return hash_str(ent->name, ENT_HASHBITS);
	318	}
	319
fd39ca9a	320	static void
1da177e4 LT	321	nametoid_request(struct cache_detail cd, struct cache_head ch, char **bpp,
	322	int *blen)
	323	{
	324	struct ent *ent = container_of(ch, struct ent, h);
	325
	326	qword_add(bpp, blen, ent->authname);
	327	qword_add(bpp, blen, ent->type == IDMAP_TYPE_GROUP ? "group" : "user");
	328	qword_add(bpp, blen, ent->name);
	329
	330	(*bpp)[-1] = '\n';
	331	}
	332
bc74b4f5 TM	333	static int
	334	nametoid_upcall(struct cache_detail cd, struct cache_head ch)
	335	{
	336	return sunrpc_cache_pipe_upcall(cd, ch, nametoid_request);
	337	}
	338
f9ecc921 N	339	static int
f9ecc921 N	340	nametoid_match(struct cache_head ca, struct cache_head cb)
1da177e4	341	{
f9ecc921 N	342	struct ent *a = container_of(ca, struct ent, h);
	343	struct ent *b = container_of(cb, struct ent, h);
	344
1da177e4 LT	345	return (a->type == b->type && strcmp(a->name, b->name) == 0 &&
	346	strcmp(a->authname, b->authname) == 0);
	347	}
	348
	349	static int
	350	nametoid_show(struct seq_file m, struct cache_detail cd, struct cache_head *h)
	351	{
	352	struct ent *ent;
	353
	354	if (h == NULL) {
	355	seq_puts(m, "#domain type name [id]\n");
	356	return 0;
	357	}
	358	ent = container_of(h, struct ent, h);
	359	seq_printf(m, "%s %s %s", ent->authname,
	360	ent->type == IDMAP_TYPE_GROUP ? "group" : "user",
	361	ent->name);
	362	if (test_bit(CACHE_VALID, &h->flags))
0a725fc4	363	seq_printf(m, " %u", ent->id);
1da177e4 LT	364	seq_printf(m, "\n");
	365	return 0;
	366	}
	367
f9ecc921 N	368	static struct ent nametoid_lookup(struct ent );
f9ecc921 N	369	static struct ent nametoid_update(struct ent , struct ent *);
fd39ca9a	370	static int nametoid_parse(struct cache_detail , char , int);
1da177e4	371
fd39ca9a	372	static struct cache_detail nametoid_cache = {
f35279d3	373	.owner = THIS_MODULE,
1da177e4 LT	374	.hash_size = ENT_HASHMAX,
	375	.hash_table = nametoid_table,
	376	.name = "nfs4.nametoid",
	377	.cache_put = ent_put,
bc74b4f5	378	.cache_upcall = nametoid_upcall,
1da177e4 LT	379	.cache_parse = nametoid_parse,
	380	.cache_show = nametoid_show,
	381	.warn_no_listener = warn_no_idmapd,
f9ecc921 N	382	.match = nametoid_match,
	383	.init = ent_init,
	384	.update = ent_init,
	385	.alloc = ent_alloc,
1da177e4 LT	386	};
1da177e4 LT	387
fd39ca9a	388	static int
1da177e4 LT	389	nametoid_parse(struct cache_detail cd, char buf, int buflen)
	390	{
	391	struct ent ent, *res;
	392	char *buf1;
	393	int error = -EINVAL;
	394
	395	if (buf[buflen - 1] != '\n')
	396	return (-EINVAL);
	397	buf[buflen - 1]= '\0';
	398
	399	buf1 = kmalloc(PAGE_SIZE, GFP_KERNEL);
	400	if (buf1 == NULL)
	401	return (-ENOMEM);
	402
	403	memset(&ent, 0, sizeof(ent));
	404
	405	/* Authentication name */
	406	if (qword_get(&buf, buf1, PAGE_SIZE) <= 0)
	407	goto out;
	408	memcpy(ent.authname, buf1, sizeof(ent.authname));
	409
	410	/* Type */
	411	if (qword_get(&buf, buf1, PAGE_SIZE) <= 0)
	412	goto out;
	413	ent.type = strcmp(buf1, "user") == 0 ?
	414	IDMAP_TYPE_USER : IDMAP_TYPE_GROUP;
	415
	416	/* Name */
	417	error = qword_get(&buf, buf1, PAGE_SIZE);
	418	if (error <= 0 \|\| error >= IDMAP_NAMESZ)
	419	goto out;
	420	memcpy(ent.name, buf1, sizeof(ent.name));
	421
	422	/* expiry */
	423	ent.h.expiry_time = get_expiry(&buf);
	424	if (ent.h.expiry_time == 0)
	425	goto out;
	426
	427	/* ID */
	428	error = get_int(&buf, &ent.id);
	429	if (error == -EINVAL)
	430	goto out;
	431	if (error == -ENOENT)
	432	set_bit(CACHE_NEGATIVE, &ent.h.flags);
	433
	434	error = -ENOMEM;
f9ecc921 N	435	res = nametoid_lookup(&ent);
	436	if (res == NULL)
	437	goto out;
	438	res = nametoid_update(&ent, res);
	439	if (res == NULL)
1da177e4 LT	440	goto out;
1da177e4 LT	441
baab935f	442	cache_put(&res->h, &nametoid_cache);
1da177e4 LT	443	error = 0;
	444	out:
	445	kfree(buf1);
	446
	447	return (error);
	448	}
	449
f9ecc921 N	450
	451	static struct ent *
	452	nametoid_lookup(struct ent *item)
	453	{
	454	struct cache_head *ch = sunrpc_cache_lookup(&nametoid_cache,
	455	&item->h,
	456	nametoid_hash(item));
	457	if (ch)
	458	return container_of(ch, struct ent, h);
	459	else
	460	return NULL;
	461	}
	462
	463	static struct ent *
	464	nametoid_update(struct ent new, struct ent old)
	465	{
	466	struct cache_head *ch = sunrpc_cache_update(&nametoid_cache,
	467	&new->h, &old->h,
	468	nametoid_hash(new));
	469	if (ch)
	470	return container_of(ch, struct ent, h);
	471	else
	472	return NULL;
	473	}
1da177e4 LT	474
	475	/*
	476	* Exported API
	477	*/
	478
dbf847ec	479	int
1da177e4 LT	480	nfsd_idmap_init(void)
1da177e4 LT	481	{
dbf847ec BF	482	int rv;
	483
	484	rv = cache_register(&idtoname_cache);
	485	if (rv)
	486	return rv;
	487	rv = cache_register(&nametoid_cache);
	488	if (rv)
	489	cache_unregister(&idtoname_cache);
	490	return rv;
1da177e4 LT	491	}
	492
	493	void
	494	nfsd_idmap_shutdown(void)
	495	{
df95a9d4 BF	496	cache_unregister(&idtoname_cache);
df95a9d4 BF	497	cache_unregister(&nametoid_cache);
1da177e4 LT	498	}
	499
	500	/*
	501	* Deferred request handling
	502	*/
	503
	504	struct idmap_defer_req {
	505	struct cache_req req;
	506	struct cache_deferred_req deferred_req;
	507	wait_queue_head_t waitq;
	508	atomic_t count;
	509	};
	510
	511	static inline void
	512	put_mdr(struct idmap_defer_req *mdr)
	513	{
	514	if (atomic_dec_and_test(&mdr->count))
	515	kfree(mdr);
	516	}
	517
	518	static inline void
	519	get_mdr(struct idmap_defer_req *mdr)
	520	{
	521	atomic_inc(&mdr->count);
	522	}
	523
	524	static void
	525	idmap_revisit(struct cache_deferred_req *dreq, int toomany)
	526	{
	527	struct idmap_defer_req *mdr =
	528	container_of(dreq, struct idmap_defer_req, deferred_req);
	529
	530	wake_up(&mdr->waitq);
	531	put_mdr(mdr);
	532	}
	533
	534	static struct cache_deferred_req *
	535	idmap_defer(struct cache_req *req)
	536	{
	537	struct idmap_defer_req *mdr =
	538	container_of(req, struct idmap_defer_req, req);
	539
	540	mdr->deferred_req.revisit = idmap_revisit;
	541	get_mdr(mdr);
	542	return (&mdr->deferred_req);
	543	}
	544
	545	static inline int
f9ecc921	546	do_idmap_lookup(struct ent (lookup_fn)(struct ent ), struct ent key,
1da177e4 LT	547	struct cache_detail detail, struct ent *item,
	548	struct idmap_defer_req *mdr)
	549	{
f9ecc921	550	*item = lookup_fn(key);
1da177e4 LT	551	if (!*item)
	552	return -ENOMEM;
	553	return cache_check(detail, &(*item)->h, &mdr->req);
	554	}
	555
	556	static inline int
f9ecc921	557	do_idmap_lookup_nowait(struct ent (lookup_fn)(struct ent *),
1da177e4 LT	558	struct ent key, struct cache_detail detail,
	559	struct ent **item)
	560	{
	561	int ret = -ENOMEM;
	562
f9ecc921	563	*item = lookup_fn(key);
1da177e4 LT	564	if (!*item)
	565	goto out_err;
	566	ret = -ETIMEDOUT;
	567	if (!test_bit(CACHE_VALID, &(*item)->h.flags)
	568	\|\| (*item)->h.expiry_time < get_seconds()
	569	\|\| detail->flush_time > (*item)->h.last_refresh)
	570	goto out_put;
	571	ret = -ENOENT;
	572	if (test_bit(CACHE_NEGATIVE, &(*item)->h.flags))
	573	goto out_put;
	574	return 0;
	575	out_put:
baab935f	576	cache_put(&(*item)->h, detail);
1da177e4 LT	577	out_err:
	578	*item = NULL;
	579	return ret;
	580	}
	581
	582	static int
	583	idmap_lookup(struct svc_rqst *rqstp,
f9ecc921	584	struct ent (lookup_fn)(struct ent ), struct ent key,
1da177e4 LT	585	struct cache_detail detail, struct ent *item)
	586	{
	587	struct idmap_defer_req *mdr;
	588	int ret;
	589
f8314dc6	590	mdr = kzalloc(sizeof(*mdr), GFP_KERNEL);
1da177e4 LT	591	if (!mdr)
1da177e4 LT	592	return -ENOMEM;
1da177e4 LT	593	atomic_set(&mdr->count, 1);
	594	init_waitqueue_head(&mdr->waitq);
	595	mdr->req.defer = idmap_defer;
	596	ret = do_idmap_lookup(lookup_fn, key, detail, item, mdr);
	597	if (ret == -EAGAIN) {
	598	wait_event_interruptible_timeout(mdr->waitq,
	599	test_bit(CACHE_VALID, &(item)->h.flags), 1 HZ);
	600	ret = do_idmap_lookup_nowait(lookup_fn, key, detail, item);
	601	}
	602	put_mdr(mdr);
	603	return ret;
	604	}
	605
3ab4d8b1 BF	606	static char *
	607	rqst_authname(struct svc_rqst *rqstp)
	608	{
	609	struct auth_domain *clp;
	610
	611	clp = rqstp->rq_gssclient ? rqstp->rq_gssclient : rqstp->rq_client;
	612	return clp->name;
	613	}
	614
1da177e4 LT	615	static int
	616	idmap_name_to_id(struct svc_rqst rqstp, int type, const char name, u32 namelen,
	617	uid_t *id)
	618	{
	619	struct ent *item, key = {
	620	.type = type,
	621	};
	622	int ret;
	623
	624	if (namelen + 1 > sizeof(key.name))
	625	return -EINVAL;
	626	memcpy(key.name, name, namelen);
	627	key.name[namelen] = '\0';
3ab4d8b1	628	strlcpy(key.authname, rqst_authname(rqstp), sizeof(key.authname));
1da177e4 LT	629	ret = idmap_lookup(rqstp, nametoid_lookup, &key, &nametoid_cache, &item);
	630	if (ret == -ENOENT)
	631	ret = -ESRCH; /* nfserr_badname */
	632	if (ret)
	633	return ret;
	634	*id = item->id;
baab935f	635	cache_put(&item->h, &nametoid_cache);
1da177e4 LT	636	return 0;
	637	}
	638
	639	static int
	640	idmap_id_to_name(struct svc_rqst rqstp, int type, uid_t id, char name)
	641	{
	642	struct ent *item, key = {
	643	.id = id,
	644	.type = type,
	645	};
	646	int ret;
	647
3ab4d8b1	648	strlcpy(key.authname, rqst_authname(rqstp), sizeof(key.authname));
1da177e4 LT	649	ret = idmap_lookup(rqstp, idtoname_lookup, &key, &idtoname_cache, &item);
	650	if (ret == -ENOENT)
	651	return sprintf(name, "%u", id);
	652	if (ret)
	653	return ret;
	654	ret = strlen(item->name);
	655	BUG_ON(ret > IDMAP_NAMESZ);
	656	memcpy(name, item->name, ret);
baab935f	657	cache_put(&item->h, &idtoname_cache);
1da177e4 LT	658	return ret;
	659	}
	660
	661	int
	662	nfsd_map_name_to_uid(struct svc_rqst rqstp, const char name, size_t namelen,
	663	__u32 *id)
	664	{
	665	return idmap_name_to_id(rqstp, IDMAP_TYPE_USER, name, namelen, id);
	666	}
	667
	668	int
	669	nfsd_map_name_to_gid(struct svc_rqst rqstp, const char name, size_t namelen,
	670	__u32 *id)
	671	{
	672	return idmap_name_to_id(rqstp, IDMAP_TYPE_GROUP, name, namelen, id);
	673	}
	674
	675	int
	676	nfsd_map_uid_to_name(struct svc_rqst rqstp, __u32 id, char name)
	677	{
	678	return idmap_id_to_name(rqstp, IDMAP_TYPE_USER, id, name);
	679	}
	680
	681	int
	682	nfsd_map_gid_to_name(struct svc_rqst rqstp, __u32 id, char name)
	683	{
	684	return idmap_id_to_name(rqstp, IDMAP_TYPE_GROUP, id, name);
	685	}