Denis Vlasenko | 8a16405 | 2007-03-12 23:34:52 +0000 | [diff] [blame] | 1 | /* |
| 2 | Copyright (c) 2001-2006, Gerrit Pape |
| 3 | All rights reserved. |
| 4 | |
| 5 | Redistribution and use in source and binary forms, with or without |
| 6 | modification, are permitted provided that the following conditions are met: |
| 7 | |
| 8 | 1. Redistributions of source code must retain the above copyright notice, |
| 9 | this list of conditions and the following disclaimer. |
| 10 | 2. Redistributions in binary form must reproduce the above copyright |
| 11 | notice, this list of conditions and the following disclaimer in the |
| 12 | documentation and/or other materials provided with the distribution. |
| 13 | 3. The name of the author may not be used to endorse or promote products |
| 14 | derived from this software without specific prior written permission. |
| 15 | |
| 16 | THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED |
| 17 | WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF |
| 18 | MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO |
| 19 | EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| 20 | SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
| 21 | PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; |
| 22 | OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, |
| 23 | WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR |
| 24 | OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF |
| 25 | ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 26 | */ |
| 27 | |
| 28 | /* Taken from http://smarden.sunsite.dk/runit/sv.8.html: |
| 29 | |
| 30 | sv - control and manage services monitored by runsv |
| 31 | |
| 32 | sv [-v] [-w sec] command services |
| 33 | /etc/init.d/service [-w sec] command |
| 34 | |
| 35 | The sv program reports the current status and controls the state of services |
| 36 | monitored by the runsv(8) supervisor. |
| 37 | |
| 38 | services consists of one or more arguments, each argument naming a directory |
Denis Vlasenko | f223efb | 2007-08-03 10:58:12 +0000 | [diff] [blame] | 39 | service used by runsv(8). If service doesn't start with a dot or slash, |
Denis Vlasenko | 8a16405 | 2007-03-12 23:34:52 +0000 | [diff] [blame] | 40 | it is searched in the default services directory /var/service/, otherwise |
| 41 | relative to the current directory. |
| 42 | |
| 43 | command is one of up, down, status, once, pause, cont, hup, alarm, interrupt, |
| 44 | 1, 2, term, kill, or exit, or start, stop, restart, shutdown, force-stop, |
| 45 | force-reload, force-restart, force-shutdown. |
| 46 | |
| 47 | The sv program can be sym-linked to /etc/init.d/ to provide an LSB init |
| 48 | script interface. The service to be controlled then is specified by the |
| 49 | base name of the "init script". |
| 50 | |
| 51 | status |
| 52 | Report the current status of the service, and the appendant log service |
| 53 | if available, to standard output. |
| 54 | up |
| 55 | If the service is not running, start it. If the service stops, restart it. |
| 56 | down |
| 57 | If the service is running, send it the TERM signal, and the CONT signal. |
| 58 | If ./run exits, start ./finish if it exists. After it stops, do not |
| 59 | restart service. |
| 60 | once |
| 61 | If the service is not running, start it. Do not restart it if it stops. |
| 62 | pause cont hup alarm interrupt quit 1 2 term kill |
| 63 | If the service is running, send it the STOP, CONT, HUP, ALRM, INT, QUIT, |
| 64 | USR1, USR2, TERM, or KILL signal respectively. |
| 65 | exit |
| 66 | If the service is running, send it the TERM signal, and the CONT signal. |
| 67 | Do not restart the service. If the service is down, and no log service |
| 68 | exists, runsv(8) exits. If the service is down and a log service exists, |
| 69 | send the TERM signal to the log service. If the log service is down, |
| 70 | runsv(8) exits. This command is ignored if it is given to an appendant |
| 71 | log service. |
| 72 | |
| 73 | sv actually looks only at the first character of above commands. |
| 74 | |
Denis Vlasenko | f223efb | 2007-08-03 10:58:12 +0000 | [diff] [blame] | 75 | Commands compatible to LSB init script actions: |
| 76 | |
Denis Vlasenko | 8a16405 | 2007-03-12 23:34:52 +0000 | [diff] [blame] | 77 | status |
| 78 | Same as status. |
| 79 | start |
| 80 | Same as up, but wait up to 7 seconds for the command to take effect. |
| 81 | Then report the status or timeout. If the script ./check exists in |
| 82 | the service directory, sv runs this script to check whether the service |
| 83 | is up and available; it's considered to be available if ./check exits |
| 84 | with 0. |
| 85 | stop |
| 86 | Same as down, but wait up to 7 seconds for the service to become down. |
| 87 | Then report the status or timeout. |
| 88 | restart |
| 89 | Send the commands term, cont, and up to the service, and wait up to |
| 90 | 7 seconds for the service to restart. Then report the status or timeout. |
| 91 | If the script ./check exists in the service directory, sv runs this script |
| 92 | to check whether the service is up and available again; it's considered |
| 93 | to be available if ./check exits with 0. |
| 94 | shutdown |
| 95 | Same as exit, but wait up to 7 seconds for the runsv(8) process |
| 96 | to terminate. Then report the status or timeout. |
| 97 | force-stop |
| 98 | Same as down, but wait up to 7 seconds for the service to become down. |
| 99 | Then report the status, and on timeout send the service the kill command. |
| 100 | force-reload |
| 101 | Send the service the term and cont commands, and wait up to |
| 102 | 7 seconds for the service to restart. Then report the status, |
| 103 | and on timeout send the service the kill command. |
| 104 | force-restart |
| 105 | Send the service the term, cont and up commands, and wait up to |
| 106 | 7 seconds for the service to restart. Then report the status, and |
| 107 | on timeout send the service the kill command. If the script ./check |
| 108 | exists in the service directory, sv runs this script to check whether |
Denis Vlasenko | f223efb | 2007-08-03 10:58:12 +0000 | [diff] [blame] | 109 | the service is up and available again; it's considered to be available |
Denis Vlasenko | 8a16405 | 2007-03-12 23:34:52 +0000 | [diff] [blame] | 110 | if ./check exits with 0. |
| 111 | force-shutdown |
| 112 | Same as exit, but wait up to 7 seconds for the runsv(8) process to |
| 113 | terminate. Then report the status, and on timeout send the service |
| 114 | the kill command. |
| 115 | |
| 116 | Additional Commands |
| 117 | |
| 118 | check |
| 119 | Check for the service to be in the state that's been requested. Wait up to |
| 120 | 7 seconds for the service to reach the requested state, then report |
| 121 | the status or timeout. If the requested state of the service is up, |
| 122 | and the script ./check exists in the service directory, sv runs |
| 123 | this script to check whether the service is up and running; |
| 124 | it's considered to be up if ./check exits with 0. |
| 125 | |
| 126 | Options |
| 127 | |
| 128 | -v |
| 129 | wait up to 7 seconds for the command to take effect. |
| 130 | Then report the status or timeout. |
| 131 | -w sec |
| 132 | Override the default timeout of 7 seconds with sec seconds. Implies -v. |
| 133 | |
| 134 | Environment |
| 135 | |
| 136 | SVDIR |
| 137 | The environment variable $SVDIR overrides the default services directory |
| 138 | /var/service. |
| 139 | SVWAIT |
| 140 | The environment variable $SVWAIT overrides the default 7 seconds to wait |
| 141 | for a command to take effect. It is overridden by the -w option. |
| 142 | |
| 143 | Exit Codes |
| 144 | sv exits 0, if the command was successfully sent to all services, and, |
| 145 | if it was told to wait, the command has taken effect to all services. |
| 146 | |
| 147 | For each service that caused an error (e.g. the directory is not |
| 148 | controlled by a runsv(8) process, or sv timed out while waiting), |
| 149 | sv increases the exit code by one and exits non zero. The maximum |
| 150 | is 99. sv exits 100 on error. |
| 151 | */ |
| 152 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 153 | /* Busyboxed by Denis Vlasenko <vda.linux@googlemail.com> */ |
| 154 | /* TODO: depends on runit_lib.c - review and reduce/eliminate */ |
| 155 | |
| 156 | #include <sys/poll.h> |
| 157 | #include <sys/file.h> |
Denis Vlasenko | b6adbf1 | 2007-05-26 19:00:18 +0000 | [diff] [blame] | 158 | #include "libbb.h" |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 159 | #include "runit_lib.h" |
| 160 | |
Denis Vlasenko | b925605 | 2007-09-28 10:29:17 +0000 | [diff] [blame] | 161 | struct globals { |
| 162 | const char *acts; |
| 163 | char **service; |
| 164 | unsigned rc; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 165 | /* "Bernstein" time format: unix + 0x400000000000000aULL */ |
Denis Vlasenko | b925605 | 2007-09-28 10:29:17 +0000 | [diff] [blame] | 166 | uint64_t tstart, tnow; |
| 167 | svstatus_t svstatus; |
| 168 | }; |
| 169 | #define G (*(struct globals*)&bb_common_bufsiz1) |
| 170 | #define acts (G.acts ) |
| 171 | #define service (G.service ) |
| 172 | #define rc (G.rc ) |
| 173 | #define tstart (G.tstart ) |
| 174 | #define tnow (G.tnow ) |
| 175 | #define svstatus (G.svstatus ) |
| 176 | #define INIT_G() do { } while (0) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 177 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 178 | |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 179 | static void fatal_cannot(const char *m1) ATTRIBUTE_NORETURN; |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 180 | static void fatal_cannot(const char *m1) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 181 | { |
| 182 | bb_perror_msg("fatal: cannot %s", m1); |
| 183 | _exit(151); |
| 184 | } |
| 185 | |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 186 | static void out(const char *p, const char *m1) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 187 | { |
| 188 | printf("%s%s: %s", p, *service, m1); |
| 189 | if (errno) { |
| 190 | printf(": %s", strerror(errno)); |
| 191 | } |
Denis Vlasenko | 4daad90 | 2007-09-27 10:20:47 +0000 | [diff] [blame] | 192 | bb_putchar('\n'); /* will also flush the output */ |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 193 | } |
| 194 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 195 | #define WARN "warning: " |
| 196 | #define OK "ok: " |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 197 | |
Denis Vlasenko | ac678ec | 2007-04-16 22:32:04 +0000 | [diff] [blame] | 198 | static void fail(const char *m1) |
| 199 | { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 200 | ++rc; |
| 201 | out("fail: ", m1); |
| 202 | } |
Denis Vlasenko | ac678ec | 2007-04-16 22:32:04 +0000 | [diff] [blame] | 203 | static void failx(const char *m1) |
| 204 | { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 205 | errno = 0; |
| 206 | fail(m1); |
| 207 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 208 | static void warn(const char *m1) |
Denis Vlasenko | ac678ec | 2007-04-16 22:32:04 +0000 | [diff] [blame] | 209 | { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 210 | ++rc; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 211 | /* "warning: <service>: <m1>\n" */ |
| 212 | out("warning: ", m1); |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 213 | } |
Denis Vlasenko | ac678ec | 2007-04-16 22:32:04 +0000 | [diff] [blame] | 214 | static void ok(const char *m1) |
| 215 | { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 216 | errno = 0; |
| 217 | out(OK, m1); |
| 218 | } |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 219 | |
| 220 | static int svstatus_get(void) |
| 221 | { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 222 | int fd, r; |
| 223 | |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 224 | fd = open_write("supervise/ok"); |
| 225 | if (fd == -1) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 226 | if (errno == ENODEV) { |
| 227 | *acts == 'x' ? ok("runsv not running") |
| 228 | : failx("runsv not running"); |
| 229 | return 0; |
| 230 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 231 | warn("cannot open supervise/ok"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 232 | return -1; |
| 233 | } |
| 234 | close(fd); |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 235 | fd = open_read("supervise/status"); |
| 236 | if (fd == -1) { |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 237 | warn("cannot open supervise/status"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 238 | return -1; |
| 239 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 240 | r = read(fd, &svstatus, 20); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 241 | close(fd); |
Denis Vlasenko | bf0a201 | 2006-12-26 10:42:51 +0000 | [diff] [blame] | 242 | switch (r) { |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 243 | case 20: |
| 244 | break; |
| 245 | case -1: |
| 246 | warn("cannot read supervise/status"); |
| 247 | return -1; |
| 248 | default: |
| 249 | errno = 0; |
| 250 | warn("cannot read supervise/status: bad format"); |
| 251 | return -1; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 252 | } |
| 253 | return 1; |
| 254 | } |
| 255 | |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 256 | static unsigned svstatus_print(const char *m) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 257 | { |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 258 | int diff; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 259 | int pid; |
| 260 | int normallyup = 0; |
| 261 | struct stat s; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 262 | uint64_t timestamp; |
Denis Vlasenko | f7996f3 | 2007-01-11 17:20:00 +0000 | [diff] [blame] | 263 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 264 | if (stat("down", &s) == -1) { |
| 265 | if (errno != ENOENT) { |
| 266 | bb_perror_msg(WARN"cannot stat %s/down", *service); |
| 267 | return 0; |
| 268 | } |
| 269 | normallyup = 1; |
| 270 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 271 | pid = SWAP_LE32(svstatus.pid_le32); |
| 272 | timestamp = SWAP_BE64(svstatus.time_be64); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 273 | if (pid) { |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 274 | switch (svstatus.run_or_finish) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 275 | case 1: printf("run: "); break; |
| 276 | case 2: printf("finish: "); break; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 277 | } |
| 278 | printf("%s: (pid %d) ", m, pid); |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 279 | } else { |
| 280 | printf("down: %s: ", m); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 281 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 282 | diff = tnow - timestamp; |
| 283 | printf("%us", (diff < 0 ? 0 : diff)); |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 284 | if (pid) { |
| 285 | if (!normallyup) printf(", normally down"); |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 286 | if (svstatus.paused) printf(", paused"); |
| 287 | if (svstatus.want == 'd') printf(", want down"); |
| 288 | if (svstatus.got_term) printf(", got TERM"); |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 289 | } else { |
| 290 | if (normallyup) printf(", normally up"); |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 291 | if (svstatus.want == 'u') printf(", want up"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 292 | } |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 293 | return pid ? 1 : 2; |
| 294 | } |
| 295 | |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 296 | static int status(const char *unused) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 297 | { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 298 | int r; |
| 299 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 300 | r = svstatus_get(); |
Denis Vlasenko | bf0a201 | 2006-12-26 10:42:51 +0000 | [diff] [blame] | 301 | switch (r) { case -1: case 0: return 0; } |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 302 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 303 | r = svstatus_print(*service); |
| 304 | if (chdir("log") == -1) { |
| 305 | if (errno != ENOENT) { |
| 306 | printf("; log: "WARN"cannot change to log service directory: %s", |
| 307 | strerror(errno)); |
| 308 | } |
| 309 | } else if (svstatus_get()) { |
| 310 | printf("; "); |
| 311 | svstatus_print("log"); |
| 312 | } |
Denis Vlasenko | 4daad90 | 2007-09-27 10:20:47 +0000 | [diff] [blame] | 313 | bb_putchar('\n'); /* will also flush the output */ |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 314 | return r; |
| 315 | } |
| 316 | |
| 317 | static int checkscript(void) |
| 318 | { |
| 319 | char *prog[2]; |
| 320 | struct stat s; |
| 321 | int pid, w; |
| 322 | |
| 323 | if (stat("check", &s) == -1) { |
| 324 | if (errno == ENOENT) return 1; |
| 325 | bb_perror_msg(WARN"cannot stat %s/check", *service); |
| 326 | return 0; |
| 327 | } |
| 328 | /* if (!(s.st_mode & S_IXUSR)) return 1; */ |
Denis Vlasenko | 53091ec | 2007-03-26 13:35:09 +0000 | [diff] [blame] | 329 | prog[0] = (char*)"./check"; |
| 330 | prog[1] = NULL; |
| 331 | pid = spawn(prog); |
| 332 | if (pid <= 0) { |
| 333 | bb_perror_msg(WARN"cannot %s child %s/check", "run", *service); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 334 | return 0; |
| 335 | } |
Denis Vlasenko | fb0eba7 | 2008-01-02 19:55:04 +0000 | [diff] [blame] | 336 | while (safe_waitpid(pid, &w, 0) == -1) { |
Denis Vlasenko | 53091ec | 2007-03-26 13:35:09 +0000 | [diff] [blame] | 337 | bb_perror_msg(WARN"cannot %s child %s/check", "wait for", *service); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 338 | return 0; |
| 339 | } |
| 340 | return !wait_exitcode(w); |
| 341 | } |
| 342 | |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 343 | static int check(const char *a) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 344 | { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 345 | int r; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 346 | unsigned pid; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 347 | uint64_t timestamp; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 348 | |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 349 | r = svstatus_get(); |
| 350 | if (r == -1) |
| 351 | return -1; |
| 352 | if (r == 0) { |
| 353 | if (*a == 'x') |
| 354 | return 1; |
| 355 | return -1; |
| 356 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 357 | pid = SWAP_LE32(svstatus.pid_le32); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 358 | switch (*a) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 359 | case 'x': |
| 360 | return 0; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 361 | case 'u': |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 362 | if (!pid || svstatus.run_or_finish != 1) return 0; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 363 | if (!checkscript()) return 0; |
| 364 | break; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 365 | case 'd': |
| 366 | if (pid) return 0; |
| 367 | break; |
| 368 | case 'c': |
| 369 | if (pid && !checkscript()) return 0; |
| 370 | break; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 371 | case 't': |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 372 | if (!pid && svstatus.want == 'd') break; |
| 373 | timestamp = SWAP_BE64(svstatus.time_be64); |
| 374 | if ((tstart > timestamp) || !pid || svstatus.got_term || !checkscript()) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 375 | return 0; |
| 376 | break; |
| 377 | case 'o': |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 378 | timestamp = SWAP_BE64(svstatus.time_be64); |
| 379 | if ((!pid && tstart > timestamp) || (pid && svstatus.want != 'd')) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 380 | return 0; |
| 381 | } |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 382 | printf(OK); |
| 383 | svstatus_print(*service); |
Denis Vlasenko | 4daad90 | 2007-09-27 10:20:47 +0000 | [diff] [blame] | 384 | bb_putchar('\n'); /* will also flush the output */ |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 385 | return 1; |
| 386 | } |
| 387 | |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 388 | static int control(const char *a) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 389 | { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 390 | int fd, r; |
| 391 | |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 392 | if (svstatus_get() <= 0) |
| 393 | return -1; |
| 394 | if (svstatus.want == *a) |
| 395 | return 0; |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 396 | fd = open_write("supervise/control"); |
| 397 | if (fd == -1) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 398 | if (errno != ENODEV) |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 399 | warn("cannot open supervise/control"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 400 | else |
| 401 | *a == 'x' ? ok("runsv not running") : failx("runsv not running"); |
| 402 | return -1; |
| 403 | } |
| 404 | r = write(fd, a, strlen(a)); |
| 405 | close(fd); |
| 406 | if (r != strlen(a)) { |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 407 | warn("cannot write to supervise/control"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 408 | return -1; |
| 409 | } |
| 410 | return 1; |
| 411 | } |
| 412 | |
Denis Vlasenko | 9b49a5e | 2007-10-11 10:05:36 +0000 | [diff] [blame] | 413 | int sv_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 414 | int sv_main(int argc, char **argv) |
| 415 | { |
| 416 | unsigned opt; |
| 417 | unsigned i, want_exit; |
| 418 | char *x; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 419 | char *action; |
| 420 | const char *varservice = "/var/service/"; |
| 421 | unsigned services; |
| 422 | char **servicex; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 423 | unsigned waitsec = 7; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 424 | smallint kll = 0; |
| 425 | smallint verbose = 0; |
| 426 | int (*act)(const char*); |
| 427 | int (*cbk)(const char*); |
| 428 | int curdir; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 429 | |
Denis Vlasenko | b925605 | 2007-09-28 10:29:17 +0000 | [diff] [blame] | 430 | INIT_G(); |
| 431 | |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 432 | xfunc_error_retval = 100; |
| 433 | |
| 434 | x = getenv("SVDIR"); |
| 435 | if (x) varservice = x; |
| 436 | x = getenv("SVWAIT"); |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 437 | if (x) waitsec = xatou(x); |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 438 | |
Denis Vlasenko | fe7cd64 | 2007-08-18 15:32:12 +0000 | [diff] [blame] | 439 | opt = getopt32(argv, "w:v", &x); |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 440 | if (opt & 1) waitsec = xatou(x); // -w |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 441 | if (opt & 2) verbose = 1; // -v |
| 442 | argc -= optind; |
| 443 | argv += optind; |
| 444 | action = *argv++; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 445 | if (!action || !*argv) bb_show_usage(); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 446 | service = argv; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 447 | services = argc - 1; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 448 | |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 449 | tnow = time(0) + 0x400000000000000aULL; |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 450 | tstart = tnow; |
| 451 | curdir = open_read("."); |
| 452 | if (curdir == -1) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 453 | fatal_cannot("open current directory"); |
| 454 | |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 455 | act = &control; |
| 456 | acts = "s"; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 457 | cbk = ✓ |
| 458 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 459 | switch (*action) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 460 | case 'x': |
| 461 | case 'e': |
| 462 | acts = "x"; |
| 463 | if (!verbose) cbk = NULL; |
| 464 | break; |
| 465 | case 'X': |
| 466 | case 'E': |
| 467 | acts = "x"; |
| 468 | kll = 1; |
| 469 | break; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 470 | case 'D': |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 471 | acts = "d"; |
| 472 | kll = 1; |
| 473 | break; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 474 | case 'T': |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 475 | acts = "tc"; |
| 476 | kll = 1; |
| 477 | break; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 478 | case 'c': |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 479 | if (str_equal(action, "check")) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 480 | act = NULL; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 481 | acts = "c"; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 482 | break; |
| 483 | } |
| 484 | case 'u': case 'd': case 'o': case 't': case 'p': case 'h': |
| 485 | case 'a': case 'i': case 'k': case 'q': case '1': case '2': |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 486 | action[1] = '\0'; |
| 487 | acts = action; |
| 488 | if (!verbose) cbk = NULL; |
| 489 | break; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 490 | case 's': |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 491 | if (str_equal(action, "shutdown")) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 492 | acts = "x"; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 493 | break; |
| 494 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 495 | if (str_equal(action, "start")) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 496 | acts = "u"; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 497 | break; |
| 498 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 499 | if (str_equal(action, "stop")) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 500 | acts = "d"; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 501 | break; |
| 502 | } |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 503 | /* "status" */ |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 504 | act = &status; |
| 505 | cbk = NULL; |
| 506 | break; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 507 | case 'r': |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 508 | if (str_equal(action, "restart")) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 509 | acts = "tcu"; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 510 | break; |
| 511 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 512 | bb_show_usage(); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 513 | case 'f': |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 514 | if (str_equal(action, "force-reload")) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 515 | acts = "tc"; |
| 516 | kll = 1; |
| 517 | break; |
| 518 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 519 | if (str_equal(action, "force-restart")) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 520 | acts = "tcu"; |
| 521 | kll = 1; |
| 522 | break; |
| 523 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 524 | if (str_equal(action, "force-shutdown")) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 525 | acts = "x"; |
| 526 | kll = 1; |
| 527 | break; |
| 528 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 529 | if (str_equal(action, "force-stop")) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 530 | acts = "d"; |
| 531 | kll = 1; |
| 532 | break; |
| 533 | } |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 534 | default: |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 535 | bb_show_usage(); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 536 | } |
| 537 | |
| 538 | servicex = service; |
| 539 | for (i = 0; i < services; ++i) { |
| 540 | if ((**service != '/') && (**service != '.')) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 541 | if (chdir(varservice) == -1) |
| 542 | goto chdir_failed_0; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 543 | } |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 544 | if (chdir(*service) == -1) { |
| 545 | chdir_failed_0: |
| 546 | fail("cannot change to service directory"); |
| 547 | goto nullify_service_0; |
| 548 | } |
| 549 | if (act && (act(acts) == -1)) { |
| 550 | nullify_service_0: |
| 551 | *service = NULL; |
| 552 | } |
| 553 | if (fchdir(curdir) == -1) |
| 554 | fatal_cannot("change to original directory"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 555 | service++; |
| 556 | } |
| 557 | |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 558 | if (cbk) while (1) { |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 559 | int diff; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 560 | |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 561 | diff = tnow - tstart; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 562 | service = servicex; |
| 563 | want_exit = 1; |
| 564 | for (i = 0; i < services; ++i, ++service) { |
| 565 | if (!*service) |
| 566 | continue; |
| 567 | if ((**service != '/') && (**service != '.')) { |
| 568 | if (chdir(varservice) == -1) |
| 569 | goto chdir_failed; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 570 | } |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 571 | if (chdir(*service) == -1) { |
| 572 | chdir_failed: |
| 573 | fail("cannot change to service directory"); |
| 574 | goto nullify_service; |
| 575 | } |
| 576 | if (cbk(acts) != 0) |
| 577 | goto nullify_service; |
| 578 | want_exit = 0; |
| 579 | if (diff >= waitsec) { |
| 580 | printf(kll ? "kill: " : "timeout: "); |
| 581 | if (svstatus_get() > 0) { |
| 582 | svstatus_print(*service); |
| 583 | ++rc; |
| 584 | } |
Denis Vlasenko | 4daad90 | 2007-09-27 10:20:47 +0000 | [diff] [blame] | 585 | bb_putchar('\n'); /* will also flush the output */ |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 586 | if (kll) |
| 587 | control("k"); |
| 588 | nullify_service: |
| 589 | *service = NULL; |
| 590 | } |
| 591 | if (fchdir(curdir) == -1) |
| 592 | fatal_cannot("change to original directory"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 593 | } |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 594 | if (want_exit) break; |
| 595 | usleep(420000); |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 596 | tnow = time(0) + 0x400000000000000aULL; |
Denis Vlasenko | 8c78395 | 2007-01-27 22:21:52 +0000 | [diff] [blame] | 597 | } |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 598 | return rc > 99 ? 99 : rc; |
| 599 | } |