jehanne/sys/src/cmd/init.c

268 lines
4.1 KiB
C
Raw Normal View History

#include <u.h>
#include <lib9.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);
int procopen(int pid, char *name, int mode);
void printfile(int fd);
char *service;
char *cmd;
char *cpu;
char *systemname;
int manual;
int iscpu;
void
main(int argc, char *argv[])
{
char *user;
int fd;
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;
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("/cfg/timezone", "#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 /cmd/rc\n");
fexec(rcexec);
manual = 1;
cmd = 0;
sleep(1000);
}
}
void
printfile(int fd)
{
int n;
char buf[256];
if(fd >= 0){
n = fd2path(fd, buf, 256);
if(n < 0){
fprint(2, "printfile(%d): fd2path: %r\n", fd);
return;
}
print("%s:\n", buf);
while((n = read(fd, buf, 256)) > 0){
write(1, buf, n);
sleep(500);
}
print("\n");
}
}
static int gotnote;
void
pinhead(void *c, 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("/cmd/rc", "rc", "-c", cmd, nil);
else if(manual || iscpu){
execl("/cmd/rc", "rc", "-m/arch/rc/lib/rcmain", "-i", nil);
}else if(strcmp(service, "terminal") == 0)
execl("/cmd/rc", "rc", "-c", ". /sys/lib/rc/startup/terminal; home=/usr/$user; cd && . lib/profile", nil);
else
execl("/cmd/rc", "rc", nil);
}
void
cpustart(void)
{
execl("/cmd/rc", "rc", "-c", "/cfg/startup", nil);
}
char*
readfile(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 nil;
}
d = dirfstat(f);
if(d == nil){
print("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){
print("init: can't malloc %s: %r\n", name);
close(f);
return nil;
}
len = read(f, val, len);
close(f);
if(len < 0){
print("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;
}
void
setenv(char *name, char *val)
{
int fd;
fd = ocreate(name, OWRITE, 0644);
if(fd < 0)
fprint(2, "init: can't create %s: %r\n", name);
else{
write(fd, val, strlen(val));
close(fd);
}
}
void
cpenv(char *from, char *to)
{
char *val;
val = readfile(from);
if(val != nil){
setenv(to, val);
free(val);
}
}
/*
* clean up after /boot
*/
void
closefds(void)
{
int i;
for(i = 3; i < 30; i++)
close(i);
}
int
procopen(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;
}