#include <u.h>
#include <libc.h>
#include <auth.h>
#include <fcall.h>
#include <thread.h>
#include <9p.h>

enum {
	Spec=		0xF800,		/* Unicode private space */
	PF=		Spec|0x20,	/* num pad function key */
	View=		Spec|0x00,	/* view (shift window up) */
	KF=		0xF000,		/* function key (begin Unicode private space) */
	Shift=		Spec|0x60,
	Break=		Spec|0x61,
	Ctrl=		Spec|0x62,
	Latin=		Spec|0x63,
	Caps=		Spec|0x64,
	Num=		Spec|0x65,
	Middle=		Spec|0x66,
	Altgr=		Spec|0x67,
	Kmouse=		Spec|0x100,
	No=		0x00,		/* peter */

	Home=		KF|13,
	Up=		KF|14,
	Pgup=		KF|15,
	Print=		KF|16,
	Left=		KF|17,
	Right=		KF|18,
	End=		KF|24,
	Down=		View,
	Pgdown=		KF|19,
	Ins=		KF|20,
	Del=		0x7F,
	Scroll=		KF|21,

	Nscan=	128,

	Int=	0,			/* scans indices */
	Ext,
	Nscans,

	Qroot=	0,
	Qkbd,
	Qkbin,
	Qkbmap,
	Qcons,
	Qconsctl,
	Nqid,

	Rawon=	0,
	Rawoff,
	Kbdflush,

	STACKSZ = 8*1024,
};

typedef struct Key Key;
typedef struct Scan Scan;

struct Key {
	int	down;
	int	c;
	Rune	r;
};

struct Scan {
	int	esc1;
	int	esc2;
	int	caps;
	int	num;
	int	shift;
	int	ctrl;
	int	latin;
	int	altgr;
	int	leds;
};

struct Qtab {
	char *name;
	int mode;
	int type;
} qtab[Nqid] = {
	"/",
		DMDIR|0555,
		QTDIR,

	"kbd",
		0666,
		0,

	"kbin",
		0222,	
		0,

	"kbmap",
		0666,	
		0,

	"cons",
		0666,	
		0,

	"consctl",
		0666,
		0,
};

char Eshort[] = "read count too small";
char Ebadarg[] = "invalid argument";
char Eperm[] = "permission denied";
char Enonexist[] = "file does not exist";
char Ebadspec[] = "bad attach specifier";
char Ewalk[] = "walk in non directory";
char Efront[] = "the front fell off";

int scanfd;
int ledsfd;

int consfd;
int echofd;

int kbdopen;
int consopen;

Channel *keychan;	/* Key */

Channel *reqchan;	/* Req* */
Channel *ctlchan;	/* int */

Channel *rawchan;	/* Rune */
Channel *linechan;	/* char * */
Channel *kbdchan;	/* char* */

/*
 * The codes at 0x79 and 0x7b are produced by the PFU Happy Hacking keyboard.
 * A 'standard' keyboard doesn't produce anything above 0x58.
 */
Rune kbtab[Nscan] = 
{
[0x00]	No,	0x1b,	'1',	'2',	'3',	'4',	'5',	'6',
[0x08]	'7',	'8',	'9',	'0',	'-',	'=',	'\b',	'\t',
[0x10]	'q',	'w',	'e',	'r',	't',	'y',	'u',	'i',
[0x18]	'o',	'p',	'[',	']',	'\n',	Ctrl,	'a',	's',
[0x20]	'd',	'f',	'g',	'h',	'j',	'k',	'l',	';',
[0x28]	'\'',	'`',	Shift,	'\\',	'z',	'x',	'c',	'v',
[0x30]	'b',	'n',	'm',	',',	'.',	'/',	Shift,	'*',
[0x38]	Latin,	' ',	Ctrl,	KF|1,	KF|2,	KF|3,	KF|4,	KF|5,
[0x40]	KF|6,	KF|7,	KF|8,	KF|9,	KF|10,	Num,	Scroll,	'7',
[0x48]	'8',	'9',	'-',	'4',	'5',	'6',	'+',	'1',
[0x50]	'2',	'3',	'0',	'.',	No,	No,	No,	KF|11,
[0x58]	KF|12,	No,	No,	No,	No,	No,	No,	No,
[0x60]	No,	No,	No,	No,	No,	No,	No,	No,
[0x68]	No,	No,	No,	No,	No,	No,	No,	No,
[0x70]	No,	No,	No,	No,	No,	No,	No,	No,
[0x78]	No,	View,	No,	Up,	No,	No,	No,	No,
};

Rune kbtabshift[Nscan] =
{
[0x00]	No,	0x1b,	'!',	'@',	'#',	'$',	'%',	'^',
[0x08]	'&',	'*',	'(',	')',	'_',	'+',	'\b',	'\t',
[0x10]	'Q',	'W',	'E',	'R',	'T',	'Y',	'U',	'I',
[0x18]	'O',	'P',	'{',	'}',	'\n',	Ctrl,	'A',	'S',
[0x20]	'D',	'F',	'G',	'H',	'J',	'K',	'L',	':',
[0x28]	'"',	'~',	Shift,	'|',	'Z',	'X',	'C',	'V',
[0x30]	'B',	'N',	'M',	'<',	'>',	'?',	Shift,	'*',
[0x38]	Latin,	' ',	Ctrl,	KF|1,	KF|2,	KF|3,	KF|4,	KF|5,
[0x40]	KF|6,	KF|7,	KF|8,	KF|9,	KF|10,	Num,	Scroll,	'7',
[0x48]	'8',	'9',	'-',	'4',	'5',	'6',	'+',	'1',
[0x50]	'2',	'3',	'0',	'.',	No,	No,	No,	KF|11,
[0x58]	KF|12,	No,	No,	No,	No,	No,	No,	No,
[0x60]	No,	No,	No,	No,	No,	No,	No,	No,
[0x68]	No,	No,	No,	No,	No,	No,	No,	No,
[0x70]	No,	No,	No,	No,	No,	No,	No,	No,
[0x78]	No,	Up,	No,	Up,	No,	No,	No,	No,
};

Rune kbtabesc1[Nscan] =
{
[0x00]	No,	No,	No,	No,	No,	No,	No,	No,
[0x08]	No,	No,	No,	No,	No,	No,	No,	No,
[0x10]	No,	No,	No,	No,	No,	No,	No,	No,
[0x18]	No,	No,	No,	No,	'\n',	Ctrl,	No,	No,
[0x20]	No,	No,	No,	No,	No,	No,	No,	No,
[0x28]	No,	No,	Shift,	No,	No,	No,	No,	No,
[0x30]	No,	No,	No,	No,	No,	'/',	No,	Print,
[0x38]	Altgr,	No,	No,	No,	No,	No,	No,	No,
[0x40]	No,	No,	No,	No,	No,	No,	Break,	Home,
[0x48]	Up,	Pgup,	No,	Left,	No,	Right,	No,	End,
[0x50]	Down,	Pgdown,	Ins,	Del,	No,	No,	No,	No,
[0x58]	No,	No,	No,	No,	No,	No,	No,	No,
[0x60]	No,	No,	No,	No,	No,	No,	No,	No,
[0x68]	No,	No,	No,	No,	No,	No,	No,	No,
[0x70]	No,	No,	No,	No,	No,	No,	No,	No,
[0x78]	No,	Up,	No,	No,	No,	No,	No,	No,
};

Rune kbtabaltgr[Nscan] =
{
[0x00]	No,	No,	No,	No,	No,	No,	No,	No,
[0x08]	No,	No,	No,	No,	No,	No,	No,	No,
[0x10]	No,	No,	No,	No,	No,	No,	No,	No,
[0x18]	No,	No,	No,	No,	'\n',	Ctrl,	No,	No,
[0x20]	No,	No,	No,	No,	No,	No,	No,	No,
[0x28]	No,	No,	Shift,	No,	No,	No,	No,	No,
[0x30]	No,	No,	No,	No,	No,	'/',	No,	Print,
[0x38]	Altgr,	No,	No,	No,	No,	No,	No,	No,
[0x40]	No,	No,	No,	No,	No,	No,	Break,	Home,
[0x48]	Up,	Pgup,	No,	Left,	No,	Right,	No,	End,
[0x50]	Down,	Pgdown,	Ins,	Del,	No,	No,	No,	No,
[0x58]	No,	No,	No,	No,	No,	No,	No,	No,
[0x60]	No,	No,	No,	No,	No,	No,	No,	No,
[0x68]	No,	No,	No,	No,	No,	No,	No,	No,
[0x70]	No,	No,	No,	No,	No,	No,	No,	No,
[0x78]	No,	Up,	No,	No,	No,	No,	No,	No,
};

Rune kbtabctrl[Nscan] =
{
[0x00]	No,	'', 	'', 	'', 	'', 	'', 	'', 	'', 
[0x08]	'', 	'', 	'', 	'', 	'
', 	'', 	'\b',	'\t',
[0x10]	'', 	'', 	'', 	'', 	'', 	'', 	'', 	'\t',
[0x18]	'', 	'', 	'', 	'', 	'\n',	Ctrl,	'
', 	'', 
[0x20]	'', 	'', 	'', 	'\b',	'\n',	'', 	'', 	'', 
[0x28]	'', 	No, 	Shift,	'', 	'', 	'', 	'', 	'', 
[0x30]	'', 	'', 	'
', 	'', 	'', 	'', 	Shift,	'\n',
[0x38]	Latin,	No, 	Ctrl,	'', 	'', 	'', 	'', 	'', 
[0x40]	'', 	'', 	'', 	'
', 	'', 	'', 	'', 	'', 
[0x48]	'', 	'', 	'
', 	'', 	'', 	'', 	'', 	'', 
[0x50]	'', 	'', 	'', 	'', 	No,	No,	No,	'', 
[0x58]	'', 	No,	No,	No,	No,	No,	No,	No,
[0x60]	No,	No,	No,	No,	No,	No,	No,	No,
[0x68]	No,	No,	No,	No,	No,	No,	No,	No,
[0x70]	No,	No,	No,	No,	No,	No,	No,	No,
[0x78]	No,	'', 	No,	'\b',	No,	No,	No,	No,
};

void reboot(void);

/*
 * Scan code processing
 */
void
kbdputsc(Scan *scan, int c)
{
	Key key;

	/*
	 *  e0's is the first of a 2 character sequence, e1 the first
	 *  of a 3 character sequence (on the safari)
	 */
	if(c == 0xe0){
		scan->esc1 = 1;
		return;
	} else if(c == 0xe1){
		scan->esc2 = 2;
		return;
	}

	key.down = (c & 0x80) == 0;
	key.c = c & 0x7f;

	if(key.c >= Nscan)
		return;

	if(scan->esc1)
		key.r = kbtabesc1[key.c];
	else if(scan->shift)
		key.r = kbtabshift[key.c];
	else if(scan->altgr)
		key.r = kbtabaltgr[key.c];
	else if(scan->ctrl)
		key.r = kbtabctrl[key.c];
	else
		key.r = kbtab[key.c];

	if(scan->caps && key.r<='z' && key.r>='a')
		key.r += 'A' - 'a';

	if(scan->ctrl && scan->latin && key.r == Del)
		reboot();

	send(keychan, &key);

	if(scan->esc1)
		scan->esc1 = 0;
	else if(scan->esc2)
		scan->esc2--;

	switch(key.r){
	case Shift:
		scan->shift = key.down;
		break;
	case Ctrl:
		scan->ctrl = key.down;
		break;
	case Altgr:
		scan->altgr = key.down;
		break;
	case Latin:
		scan->latin = key.down;
		break;
	case Num:
		scan->num ^= key.down;
		break;
	case Caps:
		scan->caps ^= key.down;
		break;
	}
}

void
setleds(Scan *scan, int leds)
{
	char buf[8];

	if(ledsfd < 0 || scan->leds == leds)
		return;
	leds &= 7;
	snprint(buf, sizeof(buf), "%d", leds);
	pwrite(ledsfd, buf, strlen(buf), 0);
	scan->leds = leds;
}

/*
 * Read scan codes from scanfd
 */ 
void
scanproc(void *)
{
	uchar buf[64];
	Scan scan;
	int i, n;

	threadsetname("scanproc");

	memset(&scan, 0, sizeof scan);
	while((n = read(scanfd, buf, sizeof buf)) > 0){
		for(i=0; i<n; i++)
			kbdputsc(&scan, buf[i]);
		setleds(&scan, (scan.num<<1) | (scan.caps<<2));
	}
}

char*
utfconv(Rune *r, int n)
{
	char *s, *p;
	int l;

	l = runenlen(r, n) + 1;
	s = emalloc9p(l);
	for(p = s; n > 0; r++, n--)
		p += runetochar(p, r);
	*p = 0;
	return s;
}

/*
 * Read key events from keychan and produce characters to
 * rawchan and keystate in kbdchan. this way here is only
 * one global keystate even if multiple keyboards are used.
 */
void
keyproc(void *)
{
	int cb[Nscan];
	Rune rb[Nscan];
	Key key;
	int i, nb;
	char *s;

	threadsetname("keyproc");

	nb = 0;
	while(recv(keychan, &key) > 0){
		if(key.down){
			switch(key.r){
			case No:
			case Caps:
			case Num:
			case Shift:
			case Latin:
			case Ctrl:
			case Altgr:
			case Kmouse|1:
			case Kmouse|2:
			case Kmouse|3:
			case Kmouse|4:
			case Kmouse|5:
			case KF|11:
			case KF|12:
				break;
			default:
				nbsend(rawchan, &key.r);
			}
		}

		s = nil;
		for(i=0; i<nb && cb[i] != key.c; i++)
			;
		if(!key.down){
			if(i < nb){
				memmove(cb+i, cb+i+1, (nb-i+1) * sizeof(cb[0]));
				memmove(rb+i, rb+i+1, (nb-i+1) * sizeof(rb[0]));
				nb--;
				s = utfconv(rb, nb);
			}
		} else if(i == nb && nb < nelem(cb) && key.r){
			cb[nb] = key.c;
			rb[nb] = key.r;
			nb++;
			s = utfconv(rb, nb);
		}
		if(s && nbsendp(kbdchan, s) <= 0)
			free(s);
	}
}

/*
 * Read characters from consfd (serial console)
 */ 
void
consproc(void *)
{
	char *p, *e, *x, buf[64];
	Rune r;
	int n;

	threadsetname("consproc");

	p = buf;
	e = buf + sizeof(buf);
	while((n = read(consfd, p, e - p)) > 0){
		x = buf + n;
		while(p < x && fullrune(p, x - p)){
			p += chartorune(&r, p);
			send(rawchan, &r);
		}
		n = x - p;
		if(n > 0){
			memmove(buf, p, n);
			p = buf + n;
		}
	}
}

/*
 * Cook lines for cons
 */
void
lineproc(void *aux)
{
	char *l, *p, *e, *s;
	Channel *cook;
	int done;
	Rune r;

	cook = aux;

	threadsetname("lineproc");

	for(;;){
		l = emalloc9p(1024);
		p = s = l;
		e = l + 1024-(UTFmax+1);
		done = 0;
		do {
			recv(cook, &r);
			switch(r){
			case '\0':	/* flush */
				p = l;
				continue;
			case '\b':	/* backspace */
				while(p > l){
					--p;
					if(fullrune(p, s - p))
						break;
				}
				write(echofd, "\b", 1);
				continue;
			case 0x04:	/* eof */
				p = l;
				done = 1;
				break;
			case '\r':
				continue;
			case '\n':
				done = 1;
			default:
				p += runetochar(s = p, &r);
				write(echofd, s, p - s);
			}
		} while(!done && p < e);
		*p = 0;
		sendp(linechan, l);
	}
}

enum {
	Areq,
	Actl,
	Araw,
	Aline,
	Akbd,
	Aend,
};

/*
 * Queue reads to cons and kbd, flushing and
 * relay data between 9p and rawchan / kbdchan.
 */
void
ctlproc(void *)
{
	struct {
		Req *h;
		Req **t;
	} qcons, qkbd, *q;
	Alt a[Aend+1];
	Req *req;
	Fid *fid;
	Rune r;
	char *s, *b, *p, *e;
	int c, n, raw;
	Channel *cook;

	threadsetname("ctlproc");

	cook = chancreate(sizeof(Rune), 0);

	if(scanfd >= 0)
		proccreate(scanproc, nil, STACKSZ);
	if(consfd >= 0)
		proccreate(consproc, nil, STACKSZ);

	threadcreate(keyproc, nil, STACKSZ);
	threadcreate(lineproc, cook, STACKSZ);

	raw = 0;

	b = p = e = nil;

	qcons.h = nil;
	qcons.t = &qcons.h;
	qkbd.h = nil;
	qkbd.t = &qkbd.h;

	memset(a, 0, sizeof a);

	a[Areq].c = reqchan;
	a[Areq].v = &req;
	a[Areq].op = CHANRCV;

	a[Actl].c = ctlchan;
	a[Actl].v = &c;
	a[Actl].op = CHANRCV;

	a[Araw].c = rawchan;
	a[Araw].v = &r;
	a[Araw].op = CHANRCV;

	a[Aline].c = linechan;
	a[Aline].v = &s;
	a[Aline].op = CHANRCV;

	a[Akbd].c = kbdchan;
	a[Akbd].v = &s;
	a[Akbd].op = CHANRCV;

	a[Aend].op = CHANEND;

	for(;;){
		s = nil;

		a[Araw].op = (b == nil) ? CHANRCV : CHANNOP;
		a[Aline].op = (b == nil) ? CHANRCV : CHANNOP;
		a[Akbd].op = qkbd.h || !kbdopen ? CHANRCV : CHANNOP;

		switch(alt(a)){
		case Areq:
			fid = req->fid;
			if(req->ifcall.type == Tflush){
				Req **rr;

				fid = req->oldreq->fid;
				q = fid->qid.path == Qcons ? &qcons : &qkbd;
				for(rr = &q->h; *rr && *rr != req->oldreq; rr = &((*rr)->aux))
					;
				if(*rr == req->oldreq){
					if((*rr = req->oldreq->aux) == nil)
						q->t = rr;
					req->oldreq->aux = nil;
					respond(req->oldreq, "interrupted");
				}
				respond(req, nil);
			} else if(req->ifcall.type == Tread){
				q = fid->qid.path == Qcons ? &qcons : &qkbd;
				req->aux = nil;
				*q->t = req;
				q->t = &req->aux;
				goto Havereq;
			} else
				respond(req, Efront);
			break;

		case Actl:
			switch(c){
			case Rawoff:
			case Rawon:
				raw = (c == Rawon);
				if(raw){
					while(s = nbrecvp(linechan))
						free(s);
					r = '\0';
					send(cook, &r);
					free(b);
					b = nil;
				}
				break;
			case Kbdflush:
				while(s = nbrecvp(kbdchan))
					free(s);
				break;
			}
			break;

		case Araw:
			if(raw){
				s = emalloc9p(UTFmax+1);
				s[runetochar(s, &r)] = 0;
			} else {
				send(cook, &r);
				break;
			}
			/* no break */

		case Aline:
			b = s;
			p = s;
			e = s + strlen(s);

		Havereq:
			while(b && (req = qcons.h)){
				if((qcons.h = req->aux) == nil)
					qcons.t = &qcons.h;
				n = e - p;
				if(req->ifcall.count < n)
					n = req->ifcall.count;
				req->ofcall.count = n;
				memmove(req->ofcall.data, p, n);
				respond(req, nil);
				p += n;
				if(p >= e){
					free(b);
					b = nil;
				}
			}
			break;

		case Akbd:
			if(req = qkbd.h){
				if((qkbd.h = req->aux) == nil)
					qkbd.t = &qkbd.h;
				n = strlen(s) + 1;
				if(n > req->ifcall.count)
					respond(req, Eshort);
				else {
					req->ofcall.count = n;
					memmove(req->ofcall.data, s, n);
					respond(req, nil);
				}
			}
			break;
		}
	}
}

/*
 * Keyboard layout maps
 */

Rune*
kbmapent(int t, int sc)
{
	if(sc < 0 || sc >= Nscan)
		return nil;
	switch(t){
	default:
		return nil;
	case 0:
		return &kbtab[sc];
	case 1:
		return &kbtabshift[sc];
	case 2:
		return &kbtabesc1[sc];
	case 3:
		return &kbtabaltgr[sc];
	case 4:
		return &kbtabctrl[sc];
	}
}

void
kbmapread(Req *req)
{
	char tmp[3*12+1];
	int t, sc, off, n;
	Rune *rp;

	off = req->ifcall.offset/(sizeof(tmp)-1);
	t = off/Nscan;
	sc = off%Nscan;
	if(rp = kbmapent(t, sc))
		sprint(tmp, "%11d %11d %11d\n", t, sc, *rp);
	else
		*tmp = 0;
	n = strlen(tmp);
	if(req->ifcall.count < n)
		n = req->ifcall.count;
	req->ofcall.count = n;
	memmove(req->ofcall.data, tmp, n);
	respond(req, nil);
}

void
kbmapwrite(Req *req)
{
	char line[100], *lp, *b;
	Rune r, *rp;
	int sc, t, l;
	Fid *f;

	f = req->fid;
	b = req->ifcall.data;
	l = req->ifcall.count;
	lp = line;
	if(f->aux){
		strcpy(line, f->aux);
		lp = line+strlen(line);
		free(f->aux);
		f->aux = nil;
	}
	while(--l >= 0) {
		*lp++  = *b++;
		if(lp[-1] == '\n' || lp == &line[sizeof(line)-1]) {
			*lp = 0;
			if(*line == 0){
			Badarg:
				respond(req, Ebadarg);
				return;
			}
			if(*line == '\n' || *line == '#'){
				lp = line;
				continue;
			}
			lp = line;
			while(*lp == ' ' || *lp == '\t')
				lp++;
			t = strtoul(line, &lp, 0);
			sc = strtoul(lp, &lp, 0);
			while(*lp == ' ' || *lp == '\t')
				lp++;
			if((rp = kbmapent(t, sc)) == nil)
				goto Badarg;
			r = 0;
			if(*lp == '\'' && lp[1])
				chartorune(&r, lp+1);
			else if(*lp == '^' && lp[1]){
				chartorune(&r, lp+1);
				if(0x40 <= r && r < 0x60)
					r -= 0x40;
				else
					goto Badarg;
			}else if(*lp == 'M' && ('1' <= lp[1] && lp[1] <= '5'))
				r = 0xF900+lp[1]-'0';
			else if(*lp>='0' && *lp<='9') /* includes 0x... */
				r = strtoul(lp, &lp, 0);
			else
				goto Badarg;
			*rp = r;
			lp = line;
		}
	}
	if(lp != line){
		l = lp-line;
		f->aux = lp = emalloc9p(l+1);
		memmove(lp, line, l);
		lp[l] = 0;
	}
	req->ofcall.count = req->ifcall.count;
	respond(req, nil);
}

/*
 * Filesystem
 */

static int
fillstat(ulong qid, Dir *d)
{
	struct Qtab *t;

	memset(d, 0, sizeof *d);
	d->uid = "kbd";
	d->gid = "kbd";
	d->muid = "";
	d->qid = (Qid){qid, 0, 0};
	d->atime = time(0);
	t = qtab + qid;
	d->name = t->name;
	d->qid.type = t->type;
	d->mode = t->mode;
	return 1;
}

static void
fsattach(Req *r)
{
	char *spec;

	spec = r->ifcall.aname;
	if(spec && spec[0]){
		respond(r, Ebadspec);
		return;
	}
	r->fid->qid = (Qid){Qroot, 0, QTDIR};
	r->ofcall.qid = r->fid->qid;
	respond(r, nil);
}

static void
fsstat(Req *r)
{
	fillstat((ulong)r->fid->qid.path, &r->d);
	r->d.name = estrdup9p(r->d.name);
	r->d.uid = estrdup9p(r->d.uid);
	r->d.gid = estrdup9p(r->d.gid);
	r->d.muid = estrdup9p(r->d.muid);
	respond(r, nil);
}

static char*
fswalk1(Fid *fid, char *name, Qid *qid)
{
	int i;
	ulong path;

	path = fid->qid.path;
	switch(path){
	case Qroot:
		if (strcmp(name, "..") == 0) {
			*qid = (Qid){Qroot, 0, QTDIR};
			fid->qid = *qid;
			return nil;
		}
		for(i = fid->qid.path; i<Nqid; i++){
			if(strcmp(name, qtab[i].name) != 0)
				continue;
			*qid = (Qid){i, 0, 0};
			fid->qid = *qid;
			return nil;
		}
		return Enonexist;
		
	default:
		return Ewalk;
	}
}

static void
fsopen(Req *r)
{
	Fid *f;
	static int need[4] = { 4, 2, 6, 1 };
	struct Qtab *t;
	int n;

	f = r->fid;
	t = qtab + f->qid.path;
	n = need[r->ifcall.mode & 3];
	if((n & t->mode) != n)
		respond(r, Eperm);
	else{
		f->aux = nil;
		switch((ulong)f->qid.path){
		case Qkbd:
			if(kbdopen++ == 0)
				sendul(ctlchan, Kbdflush);
			break;
		case Qcons:
		case Qconsctl:
			if(consopen++ == 0)
				sendul(ctlchan, Rawoff);
			break;
		}
		respond(r, nil);
	}
}

static int
readtopdir(Fid*, uchar *buf, long off, int cnt, int blen)
{
	int i, m, n;
	long pos;
	Dir d;

	n = 0;
	pos = 0;
	for (i = 1; i < Nqid; i++){
		fillstat(i, &d);
		m = convD2M(&d, &buf[n], blen-n);
		if(off <= pos){
			if(m <= BIT16SZ || m > cnt)
				break;
			n += m;
			cnt -= m;
		}
		pos += m;
	}
	return n;
}

static void
fsread(Req *r)
{
	Fid *f;

	f = r->fid;
	switch((ulong)f->qid.path){
	default:
		respond(r, Efront);
		return;

	case Qroot:
		r->ofcall.count = readtopdir(f, (void*)r->ofcall.data, r->ifcall.offset,
			r->ifcall.count, r->ifcall.count);
		break;

	case Qkbd:
	case Qcons:
		sendp(reqchan, r);
		return;

	case Qkbmap:
		kbmapread(r);
		return;
	}
	respond(r, nil);
}

static void
fswrite(Req *r)
{
	Fid *f;
	char *p;
	int n, i;

	f = r->fid;
	switch((ulong)f->qid.path){
	default:
		respond(r, Efront);
		return;

	case Qcons:
		n = r->ifcall.count;
		if(write(echofd, r->ifcall.data, n) != n){
			responderror(r);
			return;
		}
		r->ofcall.count = n;
		break;

	case Qconsctl:
		p = r->ifcall.data;
		n = r->ifcall.count;
		if(n >= 5 && memcmp(p, "rawon", 5) == 0)
			sendul(ctlchan, Rawon);
		else if(n >= 6 && memcmp(p, "rawoff", 6) == 0)
			sendul(ctlchan, Rawoff);
		else {
			respond(r, Ebadarg);
			return;
		}
		r->ofcall.count = n;
		break;

	case Qkbin:
		if(f->aux == nil){
			f->aux = emalloc9p(sizeof(Scan));
			memset(f->aux, 0, sizeof(Scan));
		}
		for(i=0; i<r->ifcall.count; i++)
			kbdputsc((Scan*)f->aux, (uchar)r->ifcall.data[i]);
		r->ofcall.count = i;
		break;

	case Qkbmap:
		kbmapwrite(r);
		return;

	}
	respond(r, nil);
}

static void
fsflush(Req *r)
{
	switch((ulong)r->oldreq->fid->qid.path) {
	case Qkbd:
	case Qcons:
		sendp(reqchan, r);
		return;
	}
	respond(r, nil);
}

static void
fsdestroyfid(Fid *f)
{
	void *p;

	if(f->omode != -1)
		switch((ulong)f->qid.path){
		case Qkbin:
		case Qkbmap:
			if(p = f->aux){
				f->aux = nil;
				free(p);
			}
			break;
		case Qkbd:
			kbdopen--;
			break;
		case Qcons:
		case Qconsctl:
			consopen--;
			break;
		}
}

static void
fsend(Srv*)
{
	threadexitsall(nil);
}

Srv fs = {
	.attach=			fsattach,
	.walk1=			fswalk1,
	.open=			fsopen,
	.read=			fsread,
	.write=			fswrite,
	.stat=			fsstat,
	.flush=			fsflush,
	.destroyfid=		fsdestroyfid,
	.end=			fsend,
};

void
reboot(void)
{
	int fd;

	if((fd = open("/dev/reboot", OWRITE)) < 0){
		fprint(2, "can't open /dev/reboot: %r\n");
		return;
	}
	fprint(fd, "reboot\n");
	close(fd);
}

void
elevate(void)
{
	char buf[128];
	Dir *d, nd;
	int fd;

	snprint(buf, sizeof(buf), "/proc/%d/ctl", getpid());
	if((fd = open(buf, OWRITE)) < 0){
		fprint(2, "can't open %s: %r\n", buf);
		return;
	}

	/* get higher than normal priority */
	fprint(fd, "pri 16");

	/* dont let anybody kill us */
	if(d = dirfstat(fd)){
		nulldir(&nd);
		nd.mode = d->mode & ~0222;
		dirfwstat(fd, &nd);
		free(d);
	}

	close(fd);
	
}

void
usage(void)
{
	fprint(2, "usage: kbdfs [-m mntpnt]\n");
	exits("usage");
}

void
threadmain(int argc, char** argv)
{
	char *mtpt = "/dev";
	char *srv = nil;
	char *cons = nil;

	consfd = -1;
	echofd = 1;

	ARGBEGIN{
	case 'D':
		chatty9p++;
		break;
	case 'm':
		mtpt = EARGF(usage());
		break;
	case 's':
		srv = EARGF(usage());
		break;
	case 'c':
		cons = EARGF(usage());
		break;
	default:
		usage();
	}ARGEND

	if((scanfd = open("/dev/scancode", OREAD)) < 0)
		fprint(2, "can't open /dev/scancode: %r\n");
	if((ledsfd = open("/dev/leds", OWRITE)) < 0)
		fprint(2, "can't open /dev/leds: %r\n");

	if(cons){
		if((consfd = open(cons, ORDWR)) < 0)
			fprint(2, "can't open %s: %r\n", cons);
		else
			echofd = consfd;
	}	
	
	keychan = chancreate(sizeof(Key), 8);
	reqchan = chancreate(sizeof(Req*), 0);
	ctlchan = chancreate(sizeof(int), 0);
	rawchan = chancreate(sizeof(Rune), 32);
	linechan = chancreate(sizeof(char*), 16);
	kbdchan = chancreate(sizeof(char*), 16);

	elevate();

	procrfork(ctlproc, nil, STACKSZ, RFNAMEG|RFNOTEG);

	threadpostmountsrv(&fs, srv, mtpt, MBEFORE);
}