#include <u.h>#include <libc.h>#include <auth.h> char*	readfile(char*);char*	readenv(char*);void	setenv(char*, char*);void	cpenv(char*, char*);void	closefds(void);int	procopen(int, char*, int);void	fexec(void(*)(void));void	rcexec(void);void	cpustart(void); char	*service;char	*cmd;char	*cpu;char	*systemname;int	manual;int	iscpu; voidmain(int argc, char *argv[]){	char *user;	int fd; 	closefds(); 	service = "cpu";	manual = 0;	ARGBEGIN{	case 'c':		service = "cpu";		break;	case 'm':		manual = 1;		break;	case 't':		service = "terminal";		break;	}ARGEND	cmd = *argv; 	fd = procopen(getpid(), "ctl", OWRITE);	if(fd >= 0){		if(write(fd, "pri 10", 6) != 6)			fprint(2, "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(;;){		fprint(2, "\ninit: starting /bin/rc\n");		fexec(rcexec);		manual = 1;		cmd = 0;		sleep(1000);	}} static int gotnote;static int interrupted; voidpinhead(void*, char *msg){	gotnote = 1;	if(strcmp(msg, "interrupt") == 0)		interrupted = 1;	else		fprint(2, "init got note '%s'\n", msg);	noted(NCONT);} voidfexec(void (*execfn)(void)){	Waitmsg *w;	int fd, pid; 	switch(pid=fork()){	case 0:		rfork(RFNOTEG);		(*execfn)();		fprint(2, "init: exec error: %r\n");		exits("exec");	case -1:		fprint(2, "init: fork error: %r\n");		return;	default:		fd = procopen(pid, "notepg", OWRITE);	casedefault:		notify(pinhead);		interrupted = 0;		gotnote = 0;		w = wait();		if(w == nil){			if(interrupted && fd >= 0)				write(fd, "interrupt", 9);			if(gotnote)				goto casedefault;			fprint(2, "init: wait error: %r\n");			break;		}		if(w->pid != pid){			free(w);			goto casedefault;		}		if(strstr(w->msg, "exec error") != 0){			fprint(2, "init: exit string %s\n", w->msg);			fprint(2, "init: sleeping because exec failed\n");			for(;;)				sleep(1000);		}		if(w->msg[0])			fprint(2, "init: rc exit status: %s\n", w->msg);		free(w);		break;	}	if(fd >= 0)		close(fd);} 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*readfile(char *name){	int f, len;	Dir *d;	char *val; 	f = open(name, OREAD);	if(f < 0){		fprint(2, "init: can't open %s: %r\n", name);		return nil;		}	d = dirfstat(f);	if(d == nil){		fprint(2, "init: can't stat %s: %r\n", name);		close(f);		return nil;	}	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){		fprint(2, "init: can't malloc %s: %r\n", name);		close(f);		return nil;	}	len = read(f, val, len);	close(f);	if(len < 0){		fprint(2, "init: can't read %s: %r\n", name);		return nil;	}else		val[len] = '\0';	return val;} char*readenv(char *name){	char *val; 	val = readfile(name);	if(val == nil)		val = "*unknown*";	return val;} voidsetenv(char *name, char *val){	int fd; 	fd = create(name, OWRITE, 0644);	if(fd < 0)		fprint(2, "init: can't create %s: %r\n", name);	else{		write(fd, val, strlen(val));		close(fd);	}} voidcpenv(char *from, char *to){	char *val; 	val = readfile(from);	if(val != nil){		setenv(to, val);		free(val);	}} /* *  clean up after /boot */voidclosefds(void){	int i; 	for(i = 3; i < 30; i++)		close(i);} intprocopen(int pid, char *name, int mode){	char buf[128];	int fd; 	snprint(buf, sizeof(buf), "#p/%d/%s", pid, name);	fd = open(buf, mode);	if(fd < 0)		fprint(2, "init: warning: can't open %s: %r\n", name);	return fd;}