123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269 |
- #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;
- void
- main(int argc, char *argv[])
- {
- char *user;
- int fd;
- char ctl[128];
- closefds();
- 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);
- }
- }
- void
- pass(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;
- void
- pinhead(void*, char *msg)
- {
- gotnote = 1;
- fprint(2, "init got note '%s'\n", msg);
- noted(NCONT);
- }
- void
- fexec(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;
- }
- }
- void
- rcexec(void)
- {
- if(cmd)
- execl("/bin/rc", "rc", "-c", cmd, 0);
- else if(manual || iscpu)
- execl("/bin/rc", "rc", 0);
- else if(strcmp(service, "terminal") == 0)
- execl("/bin/rc", "rc", "-c", ". /rc/bin/termrc; home=/usr/$user; cd; . lib/profile", 0);
- else
- execl("/bin/rc", "rc", 0);
- }
- void
- cpustart(void)
- {
- execl("/bin/rc", "rc", "-c", "/rc/bin/cpurc", 0);
- }
- 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;
- }
- void
- setenv(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);
- }
- }
- void
- cpenv(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
- */
- void
- closefds(void)
- {
- int i;
- for(i = 3; i < 30; i++)
- close(i);
- }
|