/*
* Dump crash to file
*/
static int dump_elf(Elf32_Ehdr *elfhdr_addr, char *filename)
{
int fd;
int rc;
printf("Crash dump to %s\n", filename);
fd = open_create(filename);
printf("open_create return %d\n", fd);
if (fd < 0)
return 1;
rc = write_elf(elfhdr_addr, fd);
close(fd);
return rc;
}
static error_t
open_hook (struct trivfs_peropen *peropen)
{
error_t err = 0;
struct dev *const dev = peropen->cntl->hook;
if (dev->store)
{
mutex_lock (&dev->lock);
if (dev->nperopens++ == 0)
err = store_clear_flags (dev->store, STORE_INACTIVE);
mutex_unlock (&dev->lock);
if (!err)
err = open_create (dev, (struct open **)&peropen->hook);
}
return err;
}
int main (int argc, char const *const *argv, char const *const *envp)
{
unsigned int nb = 0, ex = 1 ;
unsigned int timeout = 0 ;
PROG = "s6-setlock" ;
for (;;)
{
register int opt = subgetopt(argc, argv, "nNrwt:") ;
if (opt == -1) break ;
switch (opt)
{
case 'n' : nb = 1 ; break ;
case 'N' : nb = 0 ; break ;
case 'r' : ex = 0 ; break ;
case 'w' : ex = 1 ; break ;
case 't' : if (!uint0_scan(subgetopt_here.arg, &timeout)) dieusage() ;
nb = 2 ; break ;
default : dieusage() ;
}
}
argc -= subgetopt_here.ind ; argv += subgetopt_here.ind ;
if (argc < 2) dieusage() ;
if (nb < 2)
{
int fd = open_create(argv[0]) ;
if (fd == -1) strerr_diefu2sys(111, "open_create ", argv[0]) ;
if ((*f[ex][nb])(fd) == -1) strerr_diefu2sys(1, "lock ", argv[0]) ;
}
else
{
char const *cargv[3] = { "s6lockd-helper", argv[0], 0 } ;
char const *cenvp[2] = { ex ? "S6LOCK_EX=1" : 0, 0 } ;
iopause_fd x = { .events = IOPAUSE_READ } ;
tain_t deadline ;
int p[2] ;
unsigned int pid ;
char c ;
tain_now_g() ;
tain_from_millisecs(&deadline, timeout) ;
tain_add_g(&deadline, &deadline) ;
pid = child_spawn(S6_LIBEXECPREFIX "s6lockd-helper", cargv, cenvp, p, 2) ;
if (!pid) strerr_diefu2sys(111, "spawn ", S6_LIBEXECPREFIX "s6lockd-helper") ;
x.fd = p[0] ;
for (;;)
{
register int r = iopause_g(&x, 1, &deadline) ;
if (r < 0) strerr_diefu1sys(111, "iopause") ;
if (!r)
{
kill(pid, SIGTERM) ;
errno = ETIMEDOUT ;
strerr_diefu1sys(1, "acquire lock") ;
}
r = sanitize_read(fd_read(p[0], &c, 1)) ;
if (r < 0) strerr_diefu1sys(111, "read ack from helper") ;
if (r) break ;
}
if (c != '!') strerr_dief1x(111, "helper sent garbage ack") ;
fd_close(p[0]) ;
if (uncoe(p[1]) < 0) strerr_diefu1sys(111, "uncoe fd to helper") ;
}
pathexec_run(argv[1], argv+1, envp) ;
strerr_dieexec(111, argv[1]) ;
}
