#include "all.h"

/*
* BUG: global data
*/

static struct
{
	RWlock	uidlock;
	Iobuf*	uidbuf;
	int	flen;
	int	find;
} uidgc;

static int
byuid(const void *a1, const void *a2)
{
	Uid *u1, *u2;

	u1 = (Uid *)a1;
	u2 = (Uid *)a2;
	return u2->uid - u1->uid;
}

static int
byname(const void *a1, const void *a2)
{
	Uid *u1, *u2;

	u1 = (Uid *)a1;
	u2 = (Uid *)a2;
	return strcmp(uidspace+u2->offset, uidspace+u1->offset);
}

static int
fchar(void)
{

	if(uidgc.find >= uidgc.flen) {
		uidgc.find = 0;
		uidgc.flen = con_read(FID2, uidgc.uidbuf->iobuf,
			fscons.offset, BUFSIZE);
		if(uidgc.flen <= 0)
			return 0;
		fscons.offset += uidgc.flen;
	}
	return uidgc.uidbuf->iobuf[uidgc.find++];
}

static int
fname(char *name)
{
	int i, c;

	memset(name, 0, NAMELEN);
	for(i=0;; i++) {
		c = fchar();
		switch(c) {
		case ':':
		case '\n':
		case ',':
		case ' ':
		case '#':
		case 0:
			return c;
		}
		if(i < NAMELEN-1)
			name[i] = c;
	}
	return -1;
}

/*
	-1:adm:adm:
	0:none:adm:
	1:tor:tor:
	10000:sys::
	10001:map:map:
	10002:doc::
	10003:upas:upas:
	10004:font::
	10005:bootes:bootes:
*/

static struct {
	int	uid;
	char	*name;
	int	leader;
}
admusers[] = {
	-1,	"adm",	-1,
	 0,	"none",	-1,
	 1,	"tor",	1,
	10000,	"sys",	0,
	10001,	"map",	10001,
	10002,	"doc",	0,
	10003,	"upas",	10003,
	10004,	"font",	0,
	10005,	"bootes",	10005,
	10006,	"inferno",	10006,
	0,	0,	0,
};


void
cmd_user(void)
{
	int c, n, o, u, g, i;
	char name[NAMELEN];

	wlock(&uidgc.uidlock);
	uidgc.uidbuf = getbuf(devnone, Cuidbuf, 0);

	memset(uid, 0, fsconf.nuid * sizeof(*uid));
	memset(uidspace, 0, fsconf.uidspace * sizeof(*uidspace));
	memset(gidspace, 0, fsconf.gidspace * sizeof(*gidspace));

	if(con_clone(FID1, FID2))
		goto initu;
	if(con_path(FID2, "/adm/users"))
		goto initu;
	if(con_open(FID2, 0))
		goto initu;

	uidgc.flen = 0;
	uidgc.find = 0;
	fscons.offset = 0;
	u = 0;
	o = fsconf.uidspace;

ul1:
	c = fname(name);
	if(c != ':')
		goto uskip;
	n = strtol(name, 0, 0);
	if(n == 0)
		goto uskip;
	c = fname(name);
	if(c != ':')
		goto uskip;
	o -= strlen(name)+1;
	if(o < 0) {
		cprint("fsconf.uidspace(%ld) too small\n", fsconf.uidspace);
		goto initu;
	}
	strcpy(uidspace+o, name);
	uid[u].uid = n;
	uid[u].offset = o;
	u++;
	if(u >= fsconf.nuid) {
		cprint("fsconf.nuid(%ld) too small\n", fsconf.nuid);
		goto initu;
	}

uskip:
	if(c == '\n')
		goto ul1;
	if(c) {
		c = fname(name);
		goto uskip;
	}
/*	cprint("%d uids read\n", u);/**/
	qsort(uid, u, sizeof(uid[0]), byuid);
	for(c=u-1; c>0; c--)
		if(uid[c].uid == uid[c-1].uid) {
			cprint("duplicate uids: %d\n", uid[c].uid);
			cprint("	%s", uidspace+uid[c].offset);
			cprint(" %s\n", uidspace+uid[c-1].offset);
		}
	qsort(uid, u, sizeof(uid[0]), byname);
	for(c=u-1; c>0; c--)
		if(!strcmp(uidspace+uid[c].offset,
		   uidspace+uid[c-1].offset)) {
			cprint("kfs: duplicate names: %s\n", uidspace+uid[c].offset);
			cprint("	%d", uid[c].uid);
			cprint(" %d\n", uid[c-1].uid);
		}
	if(fscons.flags & Fuid)
		for(c=0; c<u; c++)
			cprint("%6d %s\n", uid[c].uid, uidspace+uid[c].offset);

	uidgc.flen = 0;
	uidgc.find = 0;
	fscons.offset = 0;
	g = 0;

gl1:
	c = fname(name);
	if(c != ':')
		goto gskip;
	n = strtol(name, 0, 0);		/* number */
	if(n == 0)
		goto gskip;
	c = fname(name);	/* name */
	if(c != ':')
		goto gskip;
	c = fname(name);	/* leader */
	if(c != ':')
		goto gskip;
	o = strtouid1(name);
	if(o == 0 && strcmp(name, "") != 0)
		o = n;
	for(c=0; c<u; c++)
		if(uid[c].uid == n) {
			uid[c].lead = o;
			break;
		}
	c = fname(name);	/* list of members */
	if(c != ',' && c != '\n')
		goto gskip;
	if(!name[0])
		goto gskip;
	gidspace[g++] = n;
gl2:
	n = strtouid1(name);
	if(n)
		gidspace[g++] = n;
	if(g >= fsconf.gidspace-2) {
		cprint("fsconf.gidspace(%ld) too small\n", fsconf.gidspace);
		goto initu;
	}
	if(c == '\n')
		goto gl3;
	c = fname(name);
	if(c == ',' || c == '\n')
		goto gl2;
	cprint("gid truncated\n");

gl3:
	gidspace[g++] = 0;

gskip:
	if(c == '\n')
		goto gl1;
	if(c) {
		c = fname(name);
		goto gskip;
	}
	if(fscons.flags & Fuid) {
		o = 0;
		for(c=0; c<g; c++) {
			n = gidspace[c];
			if(n == 0) {
				o = 0;
				continue;
			}
			uidtostr1(name, n);
			if(o) {
				if(o > 6) {
					cprint("\n       %s", name);
					o = 1;
				} else
					cprint(" %s", name);
			} else
				cprint("\n%6s", name);
			o++;
		}
		cprint("\n");
	}
	goto out;

initu:
	cprint("initializing minimal user table\n");
	memset(uid, 0, fsconf.nuid * sizeof(*uid));
	memset(uidspace, 0, fsconf.uidspace * sizeof(*uidspace));
	memset(gidspace, 0, fsconf.gidspace * sizeof(*gidspace));
	o = fsconf.uidspace;
	u = 0;

	for(i=0; admusers[i].name; i++){
		o -= strlen(admusers[i].name)+1;
		strcpy(uidspace+o, admusers[i].name);
		uid[u].uid = admusers[i].uid;
		uid[u].lead = admusers[i].leader;
		uid[u].offset = o;
		u++;
	}

out:
	putbuf(uidgc.uidbuf);
	wunlock(&uidgc.uidlock);

}

void
uidtostr(char *name, int id)
{
	rlock(&uidgc.uidlock);
	uidtostr1(name, id);
	runlock(&uidgc.uidlock);
}

void
uidtostr1(char *name, int id)
{
	Uid *u;
	int i;

	if(id == 0){
		strncpy(name, "none", NAMELEN);
		return;
	}
	for(i=0, u=uid; i<fsconf.nuid; i++,u++) {
		if(u->uid == id) {
			strncpy(name, uidspace+u->offset, NAMELEN);
			return;
		}
	}
	strncpy(name, "none", NAMELEN);
}

int
strtouid(char *s)
{
	int i;

	rlock(&uidgc.uidlock);
	i = strtouid1(s);
	runlock(&uidgc.uidlock);
	return i;
}

int
strtouid1(char *s)
{
	Uid *u;
	int i;

	for(i=0, u=uid; i<fsconf.nuid; i++,u++)
		if(!strcmp(s, uidspace+u->offset))
			return u->uid;
	return 0;
}

int
ingroup(int u, int g)
{
	short *p;

	if(u == g)
		return 1;
	rlock(&uidgc.uidlock);
	for(p=gidspace; *p;) {
		if(*p != g) {
			for(p++; *p++;)
				;
			continue;
		}
		for(p++; *p; p++)
			if(*p == u) {
				runlock(&uidgc.uidlock);
				return 1;
			}
	}
	runlock(&uidgc.uidlock);
	return 0;
}

int
leadgroup(int ui, int gi)
{
	Uid *u;
	int i;

	rlock(&uidgc.uidlock);
	for(i=0, u=uid; i<fsconf.nuid; i++,u++) {
		if(u->uid == gi) {
			i = u->lead;
			runlock(&uidgc.uidlock);
			if(i == ui)
				return 1;
			if(i == 0)
				return ingroup(ui, gi);
			return 0;
		}
	}
	runlock(&uidgc.uidlock);
	return 0;
}