/* screenlock - lock a terminal */#include <u.h>#include <libc.h>#include <libsec.h>#include <draw.h>#include <thread.h>#include <auth.h> char pic[] = "/lib/bunny.bit"; int vgactl;int debug;int doblank;int chatty = 0; char user[256]; voidblankscreen(int blank){	if(vgactl < 0)		return;	seek(vgactl, 0, 0);	if(fprint(vgactl, blank? "blank": "unblank") < 0)		fprint(2, "blankscreen: can't blank: %r\n");} voiderror(char *fmt, ...){	Fmt f;	char buf[64];	va_list arg; 	fmtfdinit(&f, 1, buf, sizeof buf);	fmtprint(&f, "screenlock: ");	va_start(arg, fmt);	fmtvprint(&f, fmt, arg);	va_end(arg);	fmtprint(&f, "\n");	fmtfdflush(&f);	threadexitsall("fatal error");} voidusage(void){	fprint(2, "usage: %s\n", argv0);	exits("usage");}  voidreadfile(char *name, char *buf, int nbuf, int addnul){	int fd; 	fd = open(name, OREAD);	if(fd == -1)		error("%s - can't open: %r", name);	nbuf = read(fd, buf, nbuf-addnul);	close(fd);	if(nbuf == -1)		error("%s - can't can't read: %r", name);	if(addnul)		buf[nbuf] = '\0';} voidreadline(char *buf, int nbuf){	char c;	int i; 	i = 0;	while(i < nbuf-1)		if(read(0, &c, 1) != 1 || c == '\04' || c == '\177'){			i = 0;			break;		} else if(c == '\n')			break;		else if(c == '\b' && i > 0)			--i;		else if(c == ('u' & 037))			i = 0;		else			buf[i++] = c;	buf[i] = '\0';} voidcheckpassword(void){	int fd, consctl, must;	char buf[256];	AuthInfo *ai;	static int opened; 	must = 1;	if(!opened){		fd = open("/dev/cons", OREAD);		if(fd == -1)			error("can't open cons: %r");		dup(fd, 0);		close(fd);		fd = open("/dev/cons", OWRITE);		if(fd == -1)			error("can't open cons: %r");		dup(fd, 1);		dup(1, 2);		close(fd);		consctl = open("/dev/consctl", OWRITE);		if(consctl == -1)			error("can't open consctl: %r");		if(write(consctl, "rawon", 5) != 5)			error("can't turn off echo\n");		opened = 1;	} 	for(;;){		if(chatty || !must)			fprint(2, "%s's screenlock password: ", user);		memset(buf, 0, sizeof buf);		readline(buf, sizeof buf);		blankscreen(0);		if(chatty || !must)			fprint(2, "\n");		if(buf[0] == '\0' || buf[0] == '\04'){			if(must)				continue;			error("no password typed");		} 		/* authenticate */		ai = auth_userpasswd(user, buf);		if(ai != nil && ai->cap != nil)			break;		auth_freeAI(ai); 		if(chatty || !must)			fprint(2, "password mismatch\n");		doblank = 1;	}	memset(buf, 0, sizeof buf);	blankscreen(0);} voidblanker(void *){	int tics; 	tics = 0;	for(;;){		if(doblank > 0){			doblank = 0;			tics = 10;		}		if(tics > 0 && --tics == 0)			blankscreen(1);		sleep(1000);	}} voidgrabmouse(void*){	int fd, x, y;	char ibuf[256], obuf[256]; 	if(debug)		return;	fd = open("/dev/mouse", ORDWR);	if(fd < 0)		error("can't open /dev/mouse: %r"); 	snprint(obuf, sizeof obuf, "m %d %d",		screen->r.min.x + Dx(screen->r)/2,		screen->r.min.y + Dy(screen->r)/2);	while(read(fd, ibuf, sizeof ibuf) > 0){		ibuf[12] = 0;		ibuf[24] = 0;		x = atoi(ibuf+1);		y = atoi(ibuf+13);		if(x != screen->r.min.x + Dx(screen->r)/2 ||		   y != screen->r.min.y + Dy(screen->r)/2){			fprint(fd, "%s", obuf);			doblank = 1;		}	}} /* lay down text at `p' */static voidscreenstring(Point p, char *s){	string(screen, p, screen->display->white, ZP, font, s);	flushimage(display, 1);} voidlockscreen(void){	enum { Nfld = 5, Fldlen = 12, Cursorlen = 2*4 + 2*2*16, };	char *s;	char buf[Nfld*Fldlen], *flds[Nfld], newcmd[128], cbuf[Cursorlen];	int fd, dx, dy;	Image *i;	Point p;	Rectangle r;	Tm *tm; 	fd = open("/dev/screen", OREAD);	if(fd < 0)		error("can't open /dev/screen: %r");	if(read(fd, buf, Nfld*Fldlen) != Nfld*Fldlen)		error("can't read /dev/screen: %r");	close(fd);	buf[sizeof buf-1] = 0;	if(tokenize(buf, flds, Nfld) != Nfld)		error("can't tokenize /dev/screen header");	snprint(newcmd, sizeof newcmd, "-r %s %s %d %d",		flds[1], flds[2], atoi(flds[3]) - 1, atoi(flds[4]) - 1);	newwindow(newcmd);	if (initdraw(nil, nil, "screenlock") < 0)		sysfatal("initdraw failed");	if(display == nil)		error("no display"); 	/* screen is now open and covered.  grab mouse and hold on tight */	procrfork(grabmouse, nil, 4096, RFFDG);	procrfork(blanker, nil, 4096, RFFDG);	fd = open(pic, OREAD);	if(fd > 0){		i = readimage(display, fd, 0);		if(i){ 			r = screen->r;			p = Pt(r.max.x / 2, r.max.y * 2 / 3); 			dx = (Dx(screen->r) - Dx(i->r)) / 2;			r.min.x += dx;			r.max.x -= dx;			dy = (Dy(screen->r) - Dy(i->r)) / 2;			r.min.y += dy;			r.max.y -= dy;			draw(screen, screen->r, display->black, nil, ZP);			draw(screen, r, i, nil, i->r.min);			flushimage(display, 1);		}		close(fd); 		/* identify the user on screen, centered */		tm = localtime(time(0));		s = smprint("user %s at %d:%02.2d", getuser(), tm->hour, tm->min);		p = subpt(p, Pt(stringwidth(font, "m") * strlen(s) / 2, 0));		screenstring(p, s);	} 	/* clear the cursor */	fd = open("/dev/cursor", OWRITE);	if(fd > 0){		memset(cbuf, 0, sizeof cbuf);		write(fd, cbuf, sizeof cbuf);		/* leave it open */	}} voidthreadmain(int argc, char *argv[]){	readfile("#c/user", user, sizeof user, 1); 	if((vgactl = open("/dev/vgactl", OWRITE)) < 0)		vgactl = open("#v/vgactl", OWRITE); 	ARGBEGIN{	case 'd':		debug++;		break;	default:		usage();	}ARGEND 	if(argc != 0)		usage(); 	doblank = 1;	lockscreen();	checkpassword();	threadexitsall(nil);}