1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
/* ISC license. */
#include <string.h>
#include <strings.h>
#include <stdint.h>
#include <skalibs/types.h>
#include <skalibs/sgetopt.h>
#include <skalibs/strerr2.h>
#include <skalibs/djbunix.h>
#include <s6-rc/config.h>
#include <s6-rc/s6rc.h>
#define USAGE "s6-rc-oneshot-run [ -l live ] [ -b ] up|down servicenumber"
#define dieusage() strerr_dieusage(100, USAGE)
int main (int argc, char const *const *argv, char const *const *envp)
{
char const *live = S6RC_LIVE_BASE ;
unsigned int number ;
int up ;
int blocking = 0 ;
PROG = "s6-rc-oneshot-run" ;
{
subgetopt_t l = SUBGETOPT_ZERO ;
for (;;)
{
int opt = subgetopt_r(argc, argv, "l:b", &l) ;
if (opt == -1) break ;
switch (opt)
{
case 'l' : live = l.arg ; break ;
case 'b' : blocking = 1 ; break ;
default : dieusage() ;
}
}
argc -= l.ind ; argv += l.ind ;
}
if (argc < 2) dieusage() ;
if (!strcasecmp(argv[0], "up")) up = 1 ;
else if (!strcasecmp(argv[0], "down")) up = 0 ;
else dieusage() ;
if (!uint0_scan(argv[1], &number)) dieusage() ;
{
size_t livelen = strlen(live) ;
int fdcompiled, compiledlock ;
s6rc_db_t db ;
char compiled[livelen + 10] ;
memcpy(compiled, live, livelen) ;
memcpy(compiled + livelen, "/compiled", 10) ;
if (!s6rc_lock(0, 0, 0, compiled, 1, &compiledlock, blocking))
strerr_diefu2sys(111, "take lock on ", compiled) ;
fdcompiled = open_readb(compiled) ;
if (fdcompiled < 0)
strerr_diefu2sys(111, "open ", compiled) ;
/* Read the sizes of the compiled db */
fdcompiled = open_readb(compiled) ;
if (!s6rc_db_read_sizes(fdcompiled, &db))
strerr_diefu3sys(111, "read ", compiled, "/n") ;
if (number < db.nlong || number >= db.nlong + db.nshort)
strerr_dief1x(3, "invalid service number") ;
/* Allocate enough stack for the db */
{
s6rc_service_t serviceblob[db.nshort + db.nlong] ;
char const *argvblob[db.nargvs] ;
uint32_t depsblob[db.ndeps << 1] ;
char stringblob[db.stringlen] ;
int r ;
db.services = serviceblob ;
db.argvs = argvblob ;
db.deps = depsblob ;
db.string = stringblob ;
/* Read the db from the file */
r = s6rc_db_read(fdcompiled, &db) ;
if (r < 0) strerr_diefu3sys(111, "read ", compiled, "/db") ;
if (!r) strerr_dief3x(4, "invalid service database in ", compiled, "/db") ;
/* Run the script */
{
unsigned int sargc = db.services[number].x.oneshot.argc[up] ;
char const *const *sargv = db.argvs + db.services[number].x.oneshot.argv[up] ;
char const *newargv[sargc + 1] ;
char const **p = newargv ;
while (sargc--) *p++ = *sargv++ ;
*p = 0 ;
pathexec0_run(newargv, envp) ;
strerr_dieexec(111, newargv[0]) ;
}
}
}
}
|
