int main (int argc, char const *const *argv, char const *const *envp)
{
unsigned int len ;
int fd;
PROG = "cgjoin" ;
if (argc < 3) strerr_dieusage(100, USAGE) ;
len = strlen(argv[1]) ;
{
char path[len + PREFIXLEN + SUFFIXLEN + 1] ;
char pid[UINT_FMT + 1] ;
unsigned int k ;
// format this program's PID as a string
k = uint_fmt(pid, getpid()) ;
pid[k++] = 0 ;
// Build up the path to the cgrup filesystem
memcpy(path, PREFIX, PREFIXLEN);
memcpy(path + PREFIXLEN, argv[1], len);
memcpy(path + PREFIXLEN + len, SUFFIX, SUFFIXLEN);
path[PREFIXLEN + len + SUFFIXLEN] = 0 ;
doparents(path);
// Write the pid to the cgroup's tasks file
if ((fd = open_write(path)) < 0) strerr_dief2x(100, "cannot open cgroup tasks file: ", path) ;
fd_write(fd, pid, k) ;
close(fd) ;
}
pathexec_run(argv[2], argv+2, envp) ;
strerr_dieexec(111, argv[2]) ;
}
static void scanlist (genalloc *list, char const *s)
{
register unsigned int i = 0 ;
genalloc_setlen(diuint, list, 0) ;
while (s[i])
{
char const sep[4] = ", \t" ;
diuint iv ;
if (s[i] == '-') iv.left = 1 ;
else
{
unsigned int j = uint_scan(s+i, &iv.left) ;
if (!j || !iv.left) strerr_dief2x(100, "invalid list argument: ", s) ;
i += j ;
}
if (s[i] != '-') iv.right = iv.left ;
else
{
unsigned int j = uint_scan(s + ++i, &iv.right) ;
if (!j) iv.right = 0 ;
else if (iv.right < iv.left)
strerr_dief2x(100, "invalid list argument: ", s) ;
else i += j ;
}
switch (byte_chr(sep, 4, s[i]))
{
case 0 :
case 1 :
case 2 : i++ ;
case 3 : break ;
case 4 :
strerr_dief2x(100, "invalid list argument: ", s) ;
}
if (!genalloc_append(diuint, list, &iv))
strerr_diefu1sys(111, "build interval list") ;
}
}
int main (int argc, char const *const *argv)
{
tain_t deadline, tto ;
unsigned int dirlen ;
char const *live = S6RC_LIVE_BASE ;
char const *compiled = S6RC_COMPILED_BASE ;
PROG = "s6-rc-init" ;
{
unsigned int t = 0 ;
subgetopt_t l = SUBGETOPT_ZERO ;
for (;;)
{
register int opt = subgetopt_r(argc, argv, "c:l:t:", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 'c' : compiled = l.arg ; break ;
case 'l' : live = l.arg ; break ;
case 't' : if (!uint0_scan(l.arg, &t)) dieusage() ; break ;
default : dieusage() ;
}
}
argc -= l.ind ; argv += l.ind ;
if (t) tain_from_millisecs(&tto, t) ;
else tto = tain_infinite_relative ;
}
if (!argc) dieusage() ;
if (compiled[0] != '/')
strerr_dief2x(100, compiled, " is not an absolute path") ;
if (live[0] != '/')
strerr_dief2x(100, live, " is not an absolute path") ;
if (argv[0][0] != '/')
strerr_dief2x(100, argv[0], " is not an absolute path") ;
tain_now_g() ;
tain_add_g(&deadline, &tto) ;
if (!s6rc_sanitize_dir(&satmp, live, &dirlen)) dienomem() ;
llen = satmp.len ;
if (!stralloc_cats(&satmp, ":initial") || !stralloc_0(&satmp))
strerr_diefu1sys(111, "stralloc_catb") ;
{
int fdlock ;
int fdcompiled ;
int ok ;
s6rc_db_t db ;
unsigned int n ;
char lfn[llen + 13] ;
char cfn[llen + 23] ;
/* Create the real dir, lock it, symlink */
unlink(live) ;
rm_rf(satmp.s) ;
if (mkdir(satmp.s, 0755) < 0) strerr_diefu2sys(111, "mkdir ", satmp.s) ;
if (!s6rc_lock(satmp.s, 2, &fdlock, 0, 0, 0))
{
char tmp[satmp.len] ;
byte_copy(tmp, satmp.len, satmp.s) ;
cleanup() ;
strerr_diefu2sys(111, "take lock on ", tmp) ;
}
byte_copy(lfn, llen, satmp.s) ;
lfn[llen] = 0 ;
if (symlink(satmp.s + dirlen, lfn) < 0)
{
char tmp[satmp.len - dirlen] ;
byte_copy(tmp, satmp.len - dirlen, satmp.s + dirlen) ;
cleanup() ;
strerr_diefu4sys(111, "symlink ", tmp, " to ", lfn) ;
}
/* compiled */
fdcompiled = open_readb(compiled) ;
if (fdcompiled < 0)
{
cleanup() ;
strerr_diefu2sys(111, "open ", compiled) ;
}
byte_copy(lfn + llen, 10, "/compiled") ;
if (symlink(compiled, lfn) < 0)
{
cleanup() ;
strerr_diefu4sys(111, "symlink ", compiled, " to ", lfn) ;
}
/* scandir */
byte_copy(lfn + llen + 1, 8, "scandir") ;
if (symlink(argv[0], lfn) < 0)
{
cleanup() ;
strerr_diefu4sys(111, "symlink ", argv[0], " to ", lfn) ;
}
/* state */
byte_copy(lfn + llen + 1, 6, "state") ;
{
register int r = s6rc_db_read_sizes(fdcompiled, &db) ;
if (r <= 0)
{
cleanup() ;
if (r < 0) strerr_diefu2sys(111, "read database size in ", compiled) ;
else strerr_dief2x(4, "invalid database size in ", compiled) ;
}
close(fdcompiled) ;
n = db.nshort + db.nlong ;
{
char zero[n] ;
byte_zero(zero, n) ;
if (!openwritenclose_unsafe(lfn, zero, n))
{
cleanup() ;
strerr_diefu2sys(111, "write ", lfn) ;
}
}
}
/* servicedirs */
byte_copy(lfn + llen + 1, 12, "servicedirs") ;
byte_copy(cfn, llen + 1, lfn) ;
byte_copy(cfn + llen + 1, 21, "compiled/servicedirs") ;
if (!hiercopy(cfn, lfn))
{
cleanup() ;
strerr_diefu4sys(111, "recursively copy ", cfn, " to ", lfn) ;
}
/* start the supervisors */
lfn[llen] = 0 ;
ok = s6rc_servicedir_manage_g(lfn, &deadline) ;
if (!ok)
{
cleanup() ;
strerr_diefu3sys(111, "supervise service directories in ", lfn, "/servicedirs") ;
}
if (ok & 2)
strerr_warnw3x("s6-svscan not running on ", lfn, "/scandir") ;
}
return 0 ;
}
