#include <u.h>#include <libc.h>#include <auth.h>#include <authsrv.h> char*	readenv(char*);void	setenv(char*, char*);void	cpenv(char*, char*);void	closefds(void);void	fexec(void(*)(void));void	rcexec(void);void	cpustart(void);void	pass(int); char	*service;char	*cmd;char	*cpu;char	*systemname;int	manual;int	iscpu; voidmain(int argc, char *argv[]){	char *user;	int fd;	char ctl[128]; 	closefds();	alarm(0); 	service = "cpu";	manual = 0;	ARGBEGIN{	case 'c':		service = "cpu";		break;	case 'm':		manual = 1;		break;	case 't':		service = "terminal";		break;	}ARGEND	cmd = *argv; 	snprint(ctl, sizeof(ctl), "#p/%d/ctl", getpid());	fd = open(ctl, OWRITE);	if(fd < 0)		print("init: warning: can't open %s: %r\n", ctl);	else		if(write(fd, "pri 10", 6) != 6)			print("init: warning: can't set priority: %r\n");	close(fd); 	cpu = readenv("#e/cputype");	setenv("#e/objtype", cpu);	setenv("#e/service", service);	cpenv("/adm/timezone/local", "#e/timezone");	user = readenv("#c/user");	systemname = readenv("#c/sysname"); 	newns(user, 0);	iscpu = strcmp(service, "cpu")==0; 	if(iscpu && manual == 0)		fexec(cpustart); 	for(;;){		print("\ninit: starting /bin/rc\n");		fexec(rcexec);		manual = 1;		cmd = 0;		sleep(1000);	}} voidpass(int fd){	char key[DESKEYLEN];	char typed[32];	char crypted[DESKEYLEN];	int i; 	for(;;){		print("\n%s password:", systemname);		for(i=0; i<sizeof typed; i++){			if(read(0, typed+i, 1) != 1){				print("init: can't read password; insecure\n");				return;			}			if(typed[i] == '\n'){				typed[i] = 0;				break;			}		}		if(i == sizeof typed)			continue;		if(passtokey(crypted, typed) == 0)			continue;		seek(fd, 0, 0);		if(read(fd, key, DESKEYLEN) != DESKEYLEN){			print("init: can't read key; insecure\n");			return;		}		if(memcmp(crypted, key, sizeof key))			continue;		/* clean up memory */		memset(crypted, 0, sizeof crypted);		memset(key, 0, sizeof key);		return;	}} static int gotnote; voidpinhead(void*, char *msg){	gotnote = 1;	fprint(2, "init got note '%s'\n", msg);	noted(NCONT);} voidfexec(void (*execfn)(void)){	Waitmsg *w;	int pid; 	switch(pid=fork()){	case 0:		rfork(RFNOTEG);		(*execfn)();		print("init: exec error: %r\n");		exits("exec");	case -1:		print("init: fork error: %r\n");		exits("fork");	default:	casedefault:		notify(pinhead);		gotnote = 0;		w = wait();		if(w == nil){			if(gotnote)				goto casedefault;			print("init: wait error: %r\n");			break;		}		if(w->pid != pid){			free(w);			goto casedefault;		}		if(strstr(w->msg, "exec error") != 0){			print("init: exit string %s\n", w->msg);			print("init: sleeping because exec failed\n");			free(w);			for(;;)				sleep(1000);		}		if(w->msg[0])			print("init: rc exit status: %s\n", w->msg);		free(w);		break;	}} voidrcexec(void){	if(cmd)		execl("/bin/rc", "rc", "-c", cmd, nil);	else if(manual || iscpu)		execl("/bin/rc", "rc", nil);	else if(strcmp(service, "terminal") == 0)		execl("/bin/rc", "rc", "-c", ". /rc/bin/termrc; home=/usr/$user; cd; . lib/profile", nil);	else		execl("/bin/rc", "rc", nil);} voidcpustart(void){	execl("/bin/rc", "rc", "-c", "/rc/bin/cpurc", nil);} char*readenv(char *name){	int f, len;	Dir *d;	char *val; 	f = open(name, OREAD);	if(f < 0){		print("init: can't open %s: %r\n", name);		return "*unknown*";		}	d = dirfstat(f);	if(d == nil){		print("init: can't stat %s: %r\n", name);		return "*unknown*";	}	len = d->length;	free(d);	if(len == 0)	/* device files can be zero length but have contents */		len = 64;	val = malloc(len+1);	if(val == nil){		print("init: can't malloc %s: %r\n", name);		return "*unknown*";	}	len = read(f, val, len);	close(f);	if(len < 0){		print("init: can't read %s: %r\n", name);		return "*unknown*";	}else		val[len] = '\0';	return val;} voidsetenv(char *var, char *val){	int fd; 	fd = create(var, OWRITE, 0644);	if(fd < 0)		print("init: can't open %s\n", var);	else{		fprint(fd, val);		close(fd);	}} voidcpenv(char *file, char *var){	int i, fd;	char buf[8192]; 	fd = open(file, OREAD);	if(fd < 0)		print("init: can't open %s\n", file);	else{		i = read(fd, buf, sizeof(buf)-1);		if(i <= 0)			print("init: can't read %s: %r\n", file);		else{			close(fd);			buf[i] = 0;			setenv(var, buf);		}	}} /* *  clean up after /boot */voidclosefds(void){	int i; 	for(i = 3; i < 30; i++)		close(i);}