bulk upload of source

1 files changed, 177 insertions, 0 deletions

diff --git a/run.c b/run.c
new file mode 100644
index 0000000..7a642ad
--- /dev/null
+++ b/run.c
@@ -0,0 +1,177 @@
+#include <unistd.h>
+#include <fcntl.h>
+#include <errno.h>
+#include <alloca.h>
+#include <sys/ioctl.h>
+#include <sys/wait.h>
+#include <grp.h>
+#include "ninitfeatures.h"
+#include "uid.h"
+
+static char *service, *home, *sysdir;
+
+static void exec_setup(char *s) {
+  if (access(s, X_OK) == 0) {
+    char *x = fu(getpid());
+    pid_t pid=fork();
+    if (pid==-1) return;
+    if (pid==0) {
+      char *qq[6];
+      INIT_ARGS6(qq, s,service,home,sysdir,x,0);
+      execve(s,qq,environ);
+      _exit(1);
+    }
+    waitpid(pid,0,0);
+  }
+}
+
+static struct error_table table[] = {
+  {ENOEXEC,	"Exec format error"},
+  {ENOENT,	"No such file"},
+  {EACCES,	"Permission denied"},
+  {ENOMEM,	"Out of memory"},
+  {0,0}
+};
+
+static void die_exec() {
+  char *x=0, *y = ": ";
+  struct error_table *t=table;
+  for (; t->s; t++) 
+    if (t->n == errno) { x = t->s; break; }
+  if (x==0) { x = fu(errno); y = ": errno="; }
+  errmsg_iam(0);
+  carp(home,"/",sysdir,"/run",": Could not start ",
+       home,"/",service,"/run",y,x);
+  _exit(1); 
+}
+
+#define NI_TEXT_FILES	2
+#define MAX_groups 33
+enum { NI_params,NI_pidfile };
+
+int main(int Argc, char **Argv) {
+  char **argv,*argv_0,**xx,*s;
+  char *tmp[10];
+  static char **aa[NI_TEXT_FILES];
+  static unsigned long uid[MAX_groups + 2];
+  unsigned long ul[1];
+  int i,fd,len;
+  
+  errmsg_iam("run");
+  if (Argc<3) 
+    { carp("usage: ",errmsg_argv0," service home sys"); _exit(1); }
+  
+  service=Argv[1];
+  home=   Argv[2];
+  sysdir= Argv[3];
+  
+  if (chdir(home) || chdir(service)) die_exec();
+ 
+  if (Argv[0][0] != '-') {
+    if (getppid() == 1) {
+      ioctl(0, TIOCNOTTY, 0);
+      setsid();
+      opendevconsole();
+      // ioctl(0, TIOCSCTTY, 1);
+      // tcsetpgrp(0, getpgrp());
+    }
+
+    if (Argv[0][0] == 'e') { /* "end" */
+      execve(Argv[0],Argv,environ);
+      _exit(100);
+    }
+
+    exec_setup("sys-rsetup");
+    
+    if (!access("wait", R_OK) ||
+	!access("environ", R_OK) ||
+	!access("softlimit", R_OK) ||
+	!access("cron", R_OK)) {
+      Argv[0] = "run-wait";
+      
+      if (chdir(home) || chdir(sysdir)) return 1;
+      execve("./run-wait",Argv, environ);
+      if (chdir(home) || chdir(service)) return 1;  /* exec fails; go on */
+    }
+  }
+  
+  exec_setup("rsetup");
+  
+  INIT_ARGS2(tmp, "params","pidfile");
+  for (i=0; i < NI_TEXT_FILES; i++) {
+    int offset=6;
+    if (i) offset=0;
+    
+    if ((fd = open(tmp[i], O_RDONLY)) >=0) {
+      if (GLOBAL_READ(fd,s, len,12000)) _exit(1);
+      close(fd);
+      s[len]=0;
+      
+      len = splitmem(0,s,'\n');
+      xx=alloca((len+offset+1) * sizeof(char*));  if (xx==0) _exit(1);
+      xx += offset;
+      aa[i] = xx;
+      
+      splitmem(xx,s,'\n');
+      if (i==0 && xx[--len][0]==0) xx[len]=0;
+      skip_comments(xx);
+    }
+  }
+  
+  if ((xx=aa[NI_params])) argv = xx;
+  else { argv = tmp+7; *argv = 0; }
+  
+  for (len=129; ; len *= 2) {
+    argv_0=alloca(len);		if (!argv_0) _exit(1);
+    if ((i=readlink("run",argv_0,len))<0) {
+      if (errno!=EINVAL) die_exec();	/* not a symbolic link */
+      argv_0="./run";
+      break;
+    }
+    if (i<len) { argv_0[i]=0; break; }
+  }
+  
+  if ((xx=aa[NI_pidfile])) {
+    argv -= 5;
+    INIT_ARGS5(argv, service, xx[0], "-H", home, argv_0);
+    argv_0  = "/sbin/ninit-pidfile";
+  }
+  
+  s = argv_0 + str_rchr(argv_0, '/');
+  if (s[0]) ++s;
+  else s = argv_0;
+  *--argv = s;
+  
+  dup2_inout("in", 0,O_RDONLY);
+  dup2_inout("out",1,O_WRONLY);
+  
+  if (read_ulongs("sleep",ul,1))
+    nano_sleep(*ul, 200000000);
+  if ((s=read_header("nice"))) {
+    if (*s == '-') { i=-1; s++; } else i=1;
+    if (scan_ulong(s, ul)) { i *= (int)ul[0]; nice(i); }
+  }
+
+  len = read_ulongs("uid",uid,33);
+#ifdef RUN_WANT_GID_FILE
+  if (len < 2 && read_ulongs("gid",uid+1,1)) len=2;
+#endif
+
+  if (len > 1) {
+    gid__t g[len];
+    for (i=0; i<len; i++) g[i] = uid[i];
+    SYS_setgroups(len-1,g+1);
+  }
+  if (uid[1]) setgid(uid[1]);
+  if (uid[0]) setuid(uid[0]);
+
+  exec_setup("setup");
+  if (read_ulongs("alarm",ul,1)) alarm(*ul);
+
+  for (fd=3; fd<1024; ++fd) close(fd);
+  
+  if (!access("pause", R_OK)) kill(getpid(), SIGSTOP);
+  execve(argv_0,argv,environ);
+  die_exec();
+  return 1;
+}