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 | |
Denys Vlasenko | 95f7953 | 2017-08-02 14:26:33 +0200 | [diff] [blame] | 16 | THIS SOFTWARE IS PROVIDED BY THE AUTHOR ''AS IS'' AND ANY EXPRESS OR IMPLIED |
Denis Vlasenko | 8a16405 | 2007-03-12 23:34:52 +0000 | [diff] [blame] | 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 | |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 28 | /* Taken from http://smarden.org/runit/sv.8.html: |
Denis Vlasenko | 8a16405 | 2007-03-12 23:34:52 +0000 | [diff] [blame] | 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 |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 39 | service used by runsv(8). If service doesn't start with a dot or slash and |
| 40 | doesn't end with a slash, it is searched in the default services directory |
| 41 | /var/service/, otherwise relative to the current directory. |
Denis Vlasenko | 8a16405 | 2007-03-12 23:34:52 +0000 | [diff] [blame] | 42 | |
| 43 | command is one of up, down, status, once, pause, cont, hup, alarm, interrupt, |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 44 | 1, 2, term, kill, or exit, or start, stop, reload, restart, shutdown, |
| 45 | force-stop, force-reload, force-restart, force-shutdown, try-restart. |
Denis Vlasenko | 8a16405 | 2007-03-12 23:34:52 +0000 | [diff] [blame] | 46 | |
| 47 | status |
| 48 | Report the current status of the service, and the appendant log service |
| 49 | if available, to standard output. |
| 50 | up |
| 51 | If the service is not running, start it. If the service stops, restart it. |
| 52 | down |
| 53 | If the service is running, send it the TERM signal, and the CONT signal. |
| 54 | If ./run exits, start ./finish if it exists. After it stops, do not |
| 55 | restart service. |
| 56 | once |
| 57 | If the service is not running, start it. Do not restart it if it stops. |
| 58 | pause cont hup alarm interrupt quit 1 2 term kill |
| 59 | If the service is running, send it the STOP, CONT, HUP, ALRM, INT, QUIT, |
| 60 | USR1, USR2, TERM, or KILL signal respectively. |
| 61 | exit |
| 62 | If the service is running, send it the TERM signal, and the CONT signal. |
| 63 | Do not restart the service. If the service is down, and no log service |
| 64 | exists, runsv(8) exits. If the service is down and a log service exists, |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 65 | runsv(8) closes the standard input of the log service and waits for it to |
| 66 | terminate. If the log service is down, runsv(8) exits. This command is |
| 67 | ignored if it is given to an appendant log service. |
Denis Vlasenko | 8a16405 | 2007-03-12 23:34:52 +0000 | [diff] [blame] | 68 | |
| 69 | sv actually looks only at the first character of above commands. |
| 70 | |
Denis Vlasenko | f223efb | 2007-08-03 10:58:12 +0000 | [diff] [blame] | 71 | Commands compatible to LSB init script actions: |
| 72 | |
Denis Vlasenko | 8a16405 | 2007-03-12 23:34:52 +0000 | [diff] [blame] | 73 | status |
| 74 | Same as status. |
| 75 | start |
| 76 | Same as up, but wait up to 7 seconds for the command to take effect. |
| 77 | Then report the status or timeout. If the script ./check exists in |
| 78 | the service directory, sv runs this script to check whether the service |
| 79 | is up and available; it's considered to be available if ./check exits |
| 80 | with 0. |
| 81 | stop |
| 82 | Same as down, but wait up to 7 seconds for the service to become down. |
| 83 | Then report the status or timeout. |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 84 | reload |
| 85 | Same as hup, and additionally report the status afterwards. |
Denis Vlasenko | 8a16405 | 2007-03-12 23:34:52 +0000 | [diff] [blame] | 86 | restart |
| 87 | Send the commands term, cont, and up to the service, and wait up to |
| 88 | 7 seconds for the service to restart. Then report the status or timeout. |
| 89 | If the script ./check exists in the service directory, sv runs this script |
| 90 | to check whether the service is up and available again; it's considered |
| 91 | to be available if ./check exits with 0. |
| 92 | shutdown |
| 93 | Same as exit, but wait up to 7 seconds for the runsv(8) process |
| 94 | to terminate. Then report the status or timeout. |
| 95 | force-stop |
| 96 | Same as down, but wait up to 7 seconds for the service to become down. |
| 97 | Then report the status, and on timeout send the service the kill command. |
| 98 | force-reload |
| 99 | Send the service the term and cont commands, and wait up to |
| 100 | 7 seconds for the service to restart. Then report the status, |
| 101 | and on timeout send the service the kill command. |
| 102 | force-restart |
| 103 | Send the service the term, cont and up commands, and wait up to |
| 104 | 7 seconds for the service to restart. Then report the status, and |
| 105 | on timeout send the service the kill command. If the script ./check |
| 106 | exists in the service directory, sv runs this script to check whether |
Denis Vlasenko | f223efb | 2007-08-03 10:58:12 +0000 | [diff] [blame] | 107 | 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] | 108 | if ./check exits with 0. |
| 109 | force-shutdown |
| 110 | Same as exit, but wait up to 7 seconds for the runsv(8) process to |
| 111 | terminate. Then report the status, and on timeout send the service |
| 112 | the kill command. |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 113 | try-restart |
| 114 | if the service is running, send it the term and cont commands, and wait up to |
| 115 | 7 seconds for the service to restart. Then report the status or timeout. |
Denis Vlasenko | 8a16405 | 2007-03-12 23:34:52 +0000 | [diff] [blame] | 116 | |
| 117 | Additional Commands |
| 118 | |
| 119 | check |
| 120 | Check for the service to be in the state that's been requested. Wait up to |
| 121 | 7 seconds for the service to reach the requested state, then report |
| 122 | the status or timeout. If the requested state of the service is up, |
| 123 | and the script ./check exists in the service directory, sv runs |
| 124 | this script to check whether the service is up and running; |
| 125 | it's considered to be up if ./check exits with 0. |
| 126 | |
| 127 | Options |
| 128 | |
| 129 | -v |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 130 | If the command is up, down, term, once, cont, or exit, then wait up to 7 |
| 131 | seconds for the command to take effect. Then report the status or timeout. |
Denis Vlasenko | 8a16405 | 2007-03-12 23:34:52 +0000 | [diff] [blame] | 132 | -w sec |
| 133 | Override the default timeout of 7 seconds with sec seconds. Implies -v. |
| 134 | |
| 135 | Environment |
| 136 | |
| 137 | SVDIR |
| 138 | The environment variable $SVDIR overrides the default services directory |
| 139 | /var/service. |
| 140 | SVWAIT |
| 141 | The environment variable $SVWAIT overrides the default 7 seconds to wait |
| 142 | for a command to take effect. It is overridden by the -w option. |
| 143 | |
| 144 | Exit Codes |
| 145 | sv exits 0, if the command was successfully sent to all services, and, |
| 146 | if it was told to wait, the command has taken effect to all services. |
| 147 | |
| 148 | For each service that caused an error (e.g. the directory is not |
| 149 | controlled by a runsv(8) process, or sv timed out while waiting), |
| 150 | sv increases the exit code by one and exits non zero. The maximum |
| 151 | is 99. sv exits 100 on error. |
| 152 | */ |
| 153 | |
Denis Vlasenko | d18f52b | 2008-03-02 12:53:15 +0000 | [diff] [blame] | 154 | /* Busyboxed by Denys Vlasenko <vda.linux@googlemail.com> */ |
Denys Vlasenko | 0863e1a | 2015-10-19 00:41:28 +0200 | [diff] [blame] | 155 | |
| 156 | //config:config SV |
Denys Vlasenko | 4eed2c6 | 2017-07-18 22:01:24 +0200 | [diff] [blame] | 157 | //config: bool "sv (7.8 kb)" |
Denys Vlasenko | 0863e1a | 2015-10-19 00:41:28 +0200 | [diff] [blame] | 158 | //config: default y |
| 159 | //config: help |
Denys Vlasenko | 72089cf | 2017-07-21 09:50:55 +0200 | [diff] [blame] | 160 | //config: sv reports the current status and controls the state of services |
| 161 | //config: monitored by the runsv supervisor. |
Denys Vlasenko | 0863e1a | 2015-10-19 00:41:28 +0200 | [diff] [blame] | 162 | //config: |
| 163 | //config:config SV_DEFAULT_SERVICE_DIR |
| 164 | //config: string "Default directory for services" |
| 165 | //config: default "/var/service" |
Denys Vlasenko | 0d79d77 | 2018-03-30 20:02:33 +0200 | [diff] [blame^] | 166 | //config: depends on SV || SVC || SVOK |
Denys Vlasenko | 0863e1a | 2015-10-19 00:41:28 +0200 | [diff] [blame] | 167 | //config: help |
Denys Vlasenko | 72089cf | 2017-07-21 09:50:55 +0200 | [diff] [blame] | 168 | //config: Default directory for services. |
| 169 | //config: Defaults to "/var/service" |
Denys Vlasenko | 4051a99 | 2016-12-05 13:56:40 +0100 | [diff] [blame] | 170 | //config: |
| 171 | //config:config SVC |
Denys Vlasenko | 4eed2c6 | 2017-07-18 22:01:24 +0200 | [diff] [blame] | 172 | //config: bool "svc (7.8 kb)" |
Denys Vlasenko | 4051a99 | 2016-12-05 13:56:40 +0100 | [diff] [blame] | 173 | //config: default y |
| 174 | //config: help |
Denys Vlasenko | 72089cf | 2017-07-21 09:50:55 +0200 | [diff] [blame] | 175 | //config: svc controls the state of services monitored by the runsv supervisor. |
Denys Vlasenko | 0d79d77 | 2018-03-30 20:02:33 +0200 | [diff] [blame^] | 176 | //config: It is compatible with daemontools command with the same name. |
| 177 | //config: |
| 178 | //config:config SVOK |
| 179 | //config: bool "svok" |
| 180 | //config: default y |
| 181 | //config: help |
| 182 | //config: svok checks whether runsv supervisor is running. |
| 183 | //config: It is compatible with daemontools command with the same name. |
Denys Vlasenko | 0863e1a | 2015-10-19 00:41:28 +0200 | [diff] [blame] | 184 | |
Denys Vlasenko | 0d79d77 | 2018-03-30 20:02:33 +0200 | [diff] [blame^] | 185 | //applet:IF_SV( APPLET_NOEXEC(sv, sv, BB_DIR_USR_BIN, BB_SUID_DROP, sv )) |
| 186 | //applet:IF_SVC( APPLET_NOEXEC(svc, svc, BB_DIR_USR_BIN, BB_SUID_DROP, svc )) |
| 187 | //applet:IF_SVOK(APPLET_NOEXEC(svok, svok, BB_DIR_USR_BIN, BB_SUID_DROP, svok)) |
Denys Vlasenko | 0863e1a | 2015-10-19 00:41:28 +0200 | [diff] [blame] | 188 | |
| 189 | //kbuild:lib-$(CONFIG_SV) += sv.o |
Denys Vlasenko | 4051a99 | 2016-12-05 13:56:40 +0100 | [diff] [blame] | 190 | //kbuild:lib-$(CONFIG_SVC) += sv.o |
Denys Vlasenko | 0d79d77 | 2018-03-30 20:02:33 +0200 | [diff] [blame^] | 191 | //kbuild:lib-$(CONFIG_SVOK) += sv.o |
Pere Orga | 5bc8c00 | 2011-04-11 03:29:49 +0200 | [diff] [blame] | 192 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 193 | #include <sys/file.h> |
Denis Vlasenko | b6adbf1 | 2007-05-26 19:00:18 +0000 | [diff] [blame] | 194 | #include "libbb.h" |
Denys Vlasenko | e6a2f4c | 2016-04-21 16:26:30 +0200 | [diff] [blame] | 195 | #include "common_bufsiz.h" |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 196 | #include "runit_lib.h" |
| 197 | |
Denis Vlasenko | b925605 | 2007-09-28 10:29:17 +0000 | [diff] [blame] | 198 | struct globals { |
| 199 | const char *acts; |
| 200 | char **service; |
| 201 | unsigned rc; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 202 | /* "Bernstein" time format: unix + 0x400000000000000aULL */ |
Denis Vlasenko | b925605 | 2007-09-28 10:29:17 +0000 | [diff] [blame] | 203 | uint64_t tstart, tnow; |
| 204 | svstatus_t svstatus; |
Denys Vlasenko | b9be780 | 2017-08-06 21:23:03 +0200 | [diff] [blame] | 205 | smallint islog; |
Denys Vlasenko | 98a4c7c | 2010-02-04 15:00:15 +0100 | [diff] [blame] | 206 | } FIX_ALIASING; |
Denys Vlasenko | e6a2f4c | 2016-04-21 16:26:30 +0200 | [diff] [blame] | 207 | #define G (*(struct globals*)bb_common_bufsiz1) |
Denis Vlasenko | b925605 | 2007-09-28 10:29:17 +0000 | [diff] [blame] | 208 | #define acts (G.acts ) |
| 209 | #define service (G.service ) |
| 210 | #define rc (G.rc ) |
| 211 | #define tstart (G.tstart ) |
| 212 | #define tnow (G.tnow ) |
| 213 | #define svstatus (G.svstatus ) |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 214 | #define islog (G.islog ) |
Denys Vlasenko | b9be780 | 2017-08-06 21:23:03 +0200 | [diff] [blame] | 215 | #define INIT_G() do { \ |
| 216 | setup_common_bufsiz(); \ |
Denys Vlasenko | 7389662 | 2017-08-06 21:29:51 +0200 | [diff] [blame] | 217 | /* need to zero out, svc calls sv() repeatedly */ \ |
| 218 | memset(&G, 0, sizeof(G)); \ |
Denys Vlasenko | b9be780 | 2017-08-06 21:23:03 +0200 | [diff] [blame] | 219 | } while (0) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 220 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 221 | |
Denys Vlasenko | 1d3a04a | 2016-11-28 01:22:57 +0100 | [diff] [blame] | 222 | #define str_equal(s,t) (strcmp((s), (t)) == 0) |
Denys Vlasenko | 05e8605 | 2010-10-13 12:53:27 +0200 | [diff] [blame] | 223 | |
| 224 | |
Denis Vlasenko | a60f84e | 2008-07-05 09:18:54 +0000 | [diff] [blame] | 225 | static void fatal_cannot(const char *m1) NORETURN; |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 226 | static void fatal_cannot(const char *m1) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 227 | { |
Bernhard Reutner-Fischer | a53de7f | 2008-07-21 13:46:54 +0000 | [diff] [blame] | 228 | bb_perror_msg("fatal: can't %s", m1); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 229 | _exit(151); |
| 230 | } |
| 231 | |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 232 | static void out(const char *p, const char *m1) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 233 | { |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 234 | printf("%s%s%s: %s", p, *service, islog ? "/log" : "", m1); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 235 | if (errno) { |
Denys Vlasenko | 6f97b30 | 2017-09-29 18:17:25 +0200 | [diff] [blame] | 236 | printf(": "STRERROR_FMT STRERROR_ERRNO); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 237 | } |
Denis Vlasenko | 4daad90 | 2007-09-27 10:20:47 +0000 | [diff] [blame] | 238 | bb_putchar('\n'); /* will also flush the output */ |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 239 | } |
| 240 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 241 | #define WARN "warning: " |
| 242 | #define OK "ok: " |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 243 | |
Denis Vlasenko | ac678ec | 2007-04-16 22:32:04 +0000 | [diff] [blame] | 244 | static void fail(const char *m1) |
| 245 | { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 246 | ++rc; |
| 247 | out("fail: ", m1); |
| 248 | } |
Denis Vlasenko | ac678ec | 2007-04-16 22:32:04 +0000 | [diff] [blame] | 249 | static void failx(const char *m1) |
| 250 | { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 251 | errno = 0; |
| 252 | fail(m1); |
| 253 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 254 | static void warn(const char *m1) |
Denis Vlasenko | ac678ec | 2007-04-16 22:32:04 +0000 | [diff] [blame] | 255 | { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 256 | ++rc; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 257 | /* "warning: <service>: <m1>\n" */ |
| 258 | out("warning: ", m1); |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 259 | } |
Denis Vlasenko | ac678ec | 2007-04-16 22:32:04 +0000 | [diff] [blame] | 260 | static void ok(const char *m1) |
| 261 | { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 262 | errno = 0; |
| 263 | out(OK, m1); |
| 264 | } |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 265 | |
| 266 | static int svstatus_get(void) |
| 267 | { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 268 | int fd, r; |
| 269 | |
Denys Vlasenko | 05e8605 | 2010-10-13 12:53:27 +0200 | [diff] [blame] | 270 | fd = open("supervise/ok", O_WRONLY|O_NDELAY); |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 271 | if (fd == -1) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 272 | if (errno == ENODEV) { |
| 273 | *acts == 'x' ? ok("runsv not running") |
| 274 | : failx("runsv not running"); |
| 275 | return 0; |
| 276 | } |
Denys Vlasenko | 6331cf0 | 2009-11-13 09:08:27 +0100 | [diff] [blame] | 277 | warn("can't open supervise/ok"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 278 | return -1; |
| 279 | } |
| 280 | close(fd); |
Denys Vlasenko | 05e8605 | 2010-10-13 12:53:27 +0200 | [diff] [blame] | 281 | fd = open("supervise/status", O_RDONLY|O_NDELAY); |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 282 | if (fd == -1) { |
Denys Vlasenko | 6331cf0 | 2009-11-13 09:08:27 +0100 | [diff] [blame] | 283 | warn("can't open supervise/status"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 284 | return -1; |
| 285 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 286 | r = read(fd, &svstatus, 20); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 287 | close(fd); |
Denis Vlasenko | bf0a201 | 2006-12-26 10:42:51 +0000 | [diff] [blame] | 288 | switch (r) { |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 289 | case 20: |
| 290 | break; |
| 291 | case -1: |
Denys Vlasenko | 6331cf0 | 2009-11-13 09:08:27 +0100 | [diff] [blame] | 292 | warn("can't read supervise/status"); |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 293 | return -1; |
| 294 | default: |
| 295 | errno = 0; |
Denys Vlasenko | 6331cf0 | 2009-11-13 09:08:27 +0100 | [diff] [blame] | 296 | warn("can't read supervise/status: bad format"); |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 297 | return -1; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 298 | } |
| 299 | return 1; |
| 300 | } |
| 301 | |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 302 | static unsigned svstatus_print(const char *m) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 303 | { |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 304 | int diff; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 305 | int pid; |
| 306 | int normallyup = 0; |
| 307 | struct stat s; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 308 | uint64_t timestamp; |
Denis Vlasenko | f7996f3 | 2007-01-11 17:20:00 +0000 | [diff] [blame] | 309 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 310 | if (stat("down", &s) == -1) { |
| 311 | if (errno != ENOENT) { |
Denys Vlasenko | 6331cf0 | 2009-11-13 09:08:27 +0100 | [diff] [blame] | 312 | bb_perror_msg(WARN"can't stat %s/down", *service); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 313 | return 0; |
| 314 | } |
| 315 | normallyup = 1; |
| 316 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 317 | pid = SWAP_LE32(svstatus.pid_le32); |
| 318 | timestamp = SWAP_BE64(svstatus.time_be64); |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 319 | switch (svstatus.run_or_finish) { |
| 320 | case 0: printf("down: "); break; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 321 | case 1: printf("run: "); break; |
| 322 | case 2: printf("finish: "); break; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 323 | } |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 324 | printf("%s: ", m); |
| 325 | if (svstatus.run_or_finish) |
| 326 | printf("(pid %d) ", pid); |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 327 | diff = tnow - timestamp; |
| 328 | printf("%us", (diff < 0 ? 0 : diff)); |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 329 | if (pid) { |
| 330 | if (!normallyup) printf(", normally down"); |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 331 | if (svstatus.paused) printf(", paused"); |
| 332 | if (svstatus.want == 'd') printf(", want down"); |
| 333 | if (svstatus.got_term) printf(", got TERM"); |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 334 | } else { |
| 335 | if (normallyup) printf(", normally up"); |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 336 | if (svstatus.want == 'u') printf(", want up"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 337 | } |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 338 | return pid ? 1 : 2; |
| 339 | } |
| 340 | |
Denis Vlasenko | a60f84e | 2008-07-05 09:18:54 +0000 | [diff] [blame] | 341 | static int status(const char *unused UNUSED_PARAM) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 342 | { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 343 | int r; |
| 344 | |
Denys Vlasenko | b3ba9e2 | 2009-07-15 00:24:08 +0200 | [diff] [blame] | 345 | if (svstatus_get() <= 0) |
| 346 | return 0; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 347 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 348 | r = svstatus_print(*service); |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 349 | islog = 1; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 350 | if (chdir("log") == -1) { |
| 351 | if (errno != ENOENT) { |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 352 | printf("; "); |
| 353 | warn("can't change directory"); |
| 354 | } else |
| 355 | bb_putchar('\n'); |
| 356 | } else { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 357 | printf("; "); |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 358 | if (svstatus_get()) { |
| 359 | r = svstatus_print("log"); |
| 360 | bb_putchar('\n'); |
| 361 | } |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 362 | } |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 363 | islog = 0; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 364 | return r; |
| 365 | } |
| 366 | |
| 367 | static int checkscript(void) |
| 368 | { |
| 369 | char *prog[2]; |
| 370 | struct stat s; |
| 371 | int pid, w; |
| 372 | |
| 373 | if (stat("check", &s) == -1) { |
| 374 | if (errno == ENOENT) return 1; |
Denys Vlasenko | 6331cf0 | 2009-11-13 09:08:27 +0100 | [diff] [blame] | 375 | bb_perror_msg(WARN"can't stat %s/check", *service); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 376 | return 0; |
| 377 | } |
| 378 | /* if (!(s.st_mode & S_IXUSR)) return 1; */ |
Denis Vlasenko | 53091ec | 2007-03-26 13:35:09 +0000 | [diff] [blame] | 379 | prog[0] = (char*)"./check"; |
| 380 | prog[1] = NULL; |
| 381 | pid = spawn(prog); |
| 382 | if (pid <= 0) { |
Denys Vlasenko | 6331cf0 | 2009-11-13 09:08:27 +0100 | [diff] [blame] | 383 | bb_perror_msg(WARN"can't %s child %s/check", "run", *service); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 384 | return 0; |
| 385 | } |
Denis Vlasenko | fb0eba7 | 2008-01-02 19:55:04 +0000 | [diff] [blame] | 386 | while (safe_waitpid(pid, &w, 0) == -1) { |
Denys Vlasenko | 6331cf0 | 2009-11-13 09:08:27 +0100 | [diff] [blame] | 387 | bb_perror_msg(WARN"can't %s child %s/check", "wait for", *service); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 388 | return 0; |
| 389 | } |
Denys Vlasenko | 8f24f98 | 2009-06-07 16:02:00 +0200 | [diff] [blame] | 390 | return WEXITSTATUS(w) == 0; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 391 | } |
| 392 | |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 393 | static int check(const char *a) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 394 | { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 395 | int r; |
Denys Vlasenko | b3ba9e2 | 2009-07-15 00:24:08 +0200 | [diff] [blame] | 396 | unsigned pid_le32; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 397 | uint64_t timestamp; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 398 | |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 399 | r = svstatus_get(); |
| 400 | if (r == -1) |
| 401 | return -1; |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 402 | while (*a) { |
| 403 | if (r == 0) { |
| 404 | if (*a == 'x') |
| 405 | return 1; |
| 406 | return -1; |
| 407 | } |
| 408 | pid_le32 = svstatus.pid_le32; |
| 409 | switch (*a) { |
| 410 | case 'x': |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 411 | return 0; |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 412 | case 'u': |
| 413 | if (!pid_le32 || svstatus.run_or_finish != 1) |
| 414 | return 0; |
| 415 | if (!checkscript()) |
| 416 | return 0; |
| 417 | break; |
| 418 | case 'd': |
| 419 | if (pid_le32 || svstatus.run_or_finish != 0) |
| 420 | return 0; |
| 421 | break; |
| 422 | case 'C': |
| 423 | if (pid_le32 && !checkscript()) |
| 424 | return 0; |
| 425 | break; |
| 426 | case 't': |
| 427 | case 'k': |
| 428 | if (!pid_le32 && svstatus.want == 'd') |
| 429 | break; |
| 430 | timestamp = SWAP_BE64(svstatus.time_be64); |
| 431 | if ((tstart > timestamp) || !pid_le32 || svstatus.got_term || !checkscript()) |
| 432 | return 0; |
| 433 | break; |
| 434 | case 'o': |
| 435 | timestamp = SWAP_BE64(svstatus.time_be64); |
| 436 | if ((!pid_le32 && tstart > timestamp) || (pid_le32 && svstatus.want != 'd')) |
| 437 | return 0; |
| 438 | break; |
| 439 | case 'p': |
| 440 | if (pid_le32 && !svstatus.paused) |
| 441 | return 0; |
| 442 | break; |
| 443 | case 'c': |
| 444 | if (pid_le32 && svstatus.paused) |
| 445 | return 0; |
| 446 | break; |
| 447 | } |
| 448 | ++a; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 449 | } |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 450 | printf(OK); |
| 451 | svstatus_print(*service); |
Denis Vlasenko | 4daad90 | 2007-09-27 10:20:47 +0000 | [diff] [blame] | 452 | bb_putchar('\n'); /* will also flush the output */ |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 453 | return 1; |
| 454 | } |
| 455 | |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 456 | static int control(const char *a) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 457 | { |
Denys Vlasenko | b3ba9e2 | 2009-07-15 00:24:08 +0200 | [diff] [blame] | 458 | int fd, r, l; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 459 | |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 460 | if (svstatus_get() <= 0) |
| 461 | return -1; |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 462 | if (svstatus.want == *a && (*a != 'd' || svstatus.got_term == 1)) |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 463 | return 0; |
Denys Vlasenko | 05e8605 | 2010-10-13 12:53:27 +0200 | [diff] [blame] | 464 | fd = open("supervise/control", O_WRONLY|O_NDELAY); |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 465 | if (fd == -1) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 466 | if (errno != ENODEV) |
Denys Vlasenko | 6331cf0 | 2009-11-13 09:08:27 +0100 | [diff] [blame] | 467 | warn("can't open supervise/control"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 468 | else |
| 469 | *a == 'x' ? ok("runsv not running") : failx("runsv not running"); |
| 470 | return -1; |
| 471 | } |
Denys Vlasenko | b3ba9e2 | 2009-07-15 00:24:08 +0200 | [diff] [blame] | 472 | l = strlen(a); |
| 473 | r = write(fd, a, l); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 474 | close(fd); |
Denys Vlasenko | b3ba9e2 | 2009-07-15 00:24:08 +0200 | [diff] [blame] | 475 | if (r != l) { |
Denys Vlasenko | 6331cf0 | 2009-11-13 09:08:27 +0100 | [diff] [blame] | 476 | warn("can't write to supervise/control"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 477 | return -1; |
| 478 | } |
| 479 | return 1; |
| 480 | } |
| 481 | |
Denys Vlasenko | 4051a99 | 2016-12-05 13:56:40 +0100 | [diff] [blame] | 482 | //usage:#define sv_trivial_usage |
| 483 | //usage: "[-v] [-w SEC] CMD SERVICE_DIR..." |
| 484 | //usage:#define sv_full_usage "\n\n" |
| 485 | //usage: "Control services monitored by runsv supervisor.\n" |
| 486 | //usage: "Commands (only first character is enough):\n" |
| 487 | //usage: "\n" |
| 488 | //usage: "status: query service status\n" |
| 489 | //usage: "up: if service isn't running, start it. If service stops, restart it\n" |
| 490 | //usage: "once: like 'up', but if service stops, don't restart it\n" |
| 491 | //usage: "down: send TERM and CONT signals. If ./run exits, start ./finish\n" |
| 492 | //usage: " if it exists. After it stops, don't restart service\n" |
| 493 | //usage: "exit: send TERM and CONT signals to service and log service. If they exit,\n" |
| 494 | //usage: " runsv exits too\n" |
| 495 | //usage: "pause, cont, hup, alarm, interrupt, quit, 1, 2, term, kill: send\n" |
| 496 | //usage: "STOP, CONT, HUP, ALRM, INT, QUIT, USR1, USR2, TERM, KILL signal to service" |
| 497 | static int sv(char **argv) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 498 | { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 499 | char *x; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 500 | char *action; |
Denis Vlasenko | f068b3e | 2008-11-06 23:07:42 +0000 | [diff] [blame] | 501 | const char *varservice = CONFIG_SV_DEFAULT_SERVICE_DIR; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 502 | unsigned waitsec = 7; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 503 | smallint kll = 0; |
Denis Vlasenko | 1d42665 | 2008-03-17 09:09:09 +0000 | [diff] [blame] | 504 | int verbose = 0; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 505 | int (*act)(const char*); |
| 506 | int (*cbk)(const char*); |
| 507 | int curdir; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 508 | |
Denis Vlasenko | b925605 | 2007-09-28 10:29:17 +0000 | [diff] [blame] | 509 | INIT_G(); |
| 510 | |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 511 | xfunc_error_retval = 100; |
| 512 | |
| 513 | x = getenv("SVDIR"); |
| 514 | if (x) varservice = x; |
| 515 | x = getenv("SVWAIT"); |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 516 | if (x) waitsec = xatou(x); |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 517 | |
Denys Vlasenko | 22542ec | 2017-08-08 21:55:02 +0200 | [diff] [blame] | 518 | getopt32(argv, "^" "w:+v" "\0" "vv" /* -w N, -v is a counter */, |
| 519 | &waitsec, &verbose |
| 520 | ); |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 521 | argv += optind; |
| 522 | action = *argv++; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 523 | if (!action || !*argv) bb_show_usage(); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 524 | |
Denis Vlasenko | 04158e0 | 2009-02-02 10:48:06 +0000 | [diff] [blame] | 525 | tnow = time(NULL) + 0x400000000000000aULL; |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 526 | tstart = tnow; |
Denys Vlasenko | 05e8605 | 2010-10-13 12:53:27 +0200 | [diff] [blame] | 527 | curdir = open(".", O_RDONLY|O_NDELAY); |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 528 | if (curdir == -1) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 529 | fatal_cannot("open current directory"); |
| 530 | |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 531 | act = &control; |
| 532 | acts = "s"; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 533 | cbk = ✓ |
| 534 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 535 | switch (*action) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 536 | case 'x': |
| 537 | case 'e': |
| 538 | acts = "x"; |
| 539 | if (!verbose) cbk = NULL; |
| 540 | break; |
| 541 | case 'X': |
| 542 | case 'E': |
| 543 | acts = "x"; |
| 544 | kll = 1; |
| 545 | break; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 546 | case 'D': |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 547 | acts = "d"; |
| 548 | kll = 1; |
| 549 | break; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 550 | case 'T': |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 551 | acts = "tc"; |
| 552 | kll = 1; |
| 553 | break; |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 554 | case 't': |
| 555 | if (str_equal(action, "try-restart")) { |
| 556 | acts = "tc"; |
| 557 | break; |
| 558 | } |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 559 | case 'c': |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 560 | if (str_equal(action, "check")) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 561 | act = NULL; |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 562 | acts = "C"; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 563 | break; |
| 564 | } |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 565 | case 'u': case 'd': case 'o': case 'p': case 'h': |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 566 | case 'a': case 'i': case 'k': case 'q': case '1': case '2': |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 567 | action[1] = '\0'; |
| 568 | acts = action; |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 569 | if (!verbose) |
| 570 | cbk = NULL; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 571 | break; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 572 | case 's': |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 573 | if (str_equal(action, "shutdown")) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 574 | acts = "x"; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 575 | break; |
| 576 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 577 | if (str_equal(action, "start")) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 578 | acts = "u"; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 579 | break; |
| 580 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 581 | if (str_equal(action, "stop")) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 582 | acts = "d"; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 583 | break; |
| 584 | } |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 585 | /* "status" */ |
Denis Vlasenko | 322661d | 2007-01-29 23:43:52 +0000 | [diff] [blame] | 586 | act = &status; |
| 587 | cbk = NULL; |
| 588 | break; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 589 | case 'r': |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 590 | if (str_equal(action, "restart")) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 591 | acts = "tcu"; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 592 | break; |
| 593 | } |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 594 | if (str_equal(action, "reload")) { |
| 595 | acts = "h"; |
| 596 | break; |
| 597 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 598 | bb_show_usage(); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 599 | case 'f': |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 600 | if (str_equal(action, "force-reload")) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 601 | acts = "tc"; |
| 602 | kll = 1; |
| 603 | break; |
| 604 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 605 | if (str_equal(action, "force-restart")) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 606 | acts = "tcu"; |
| 607 | kll = 1; |
| 608 | break; |
| 609 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 610 | if (str_equal(action, "force-shutdown")) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 611 | acts = "x"; |
| 612 | kll = 1; |
| 613 | break; |
| 614 | } |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 615 | if (str_equal(action, "force-stop")) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 616 | acts = "d"; |
| 617 | kll = 1; |
| 618 | break; |
| 619 | } |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 620 | default: |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 621 | bb_show_usage(); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 622 | } |
| 623 | |
Denys Vlasenko | b3ba9e2 | 2009-07-15 00:24:08 +0200 | [diff] [blame] | 624 | service = argv; |
| 625 | while ((x = *service) != NULL) { |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 626 | if (x[0] != '/' && x[0] != '.' |
Denys Vlasenko | 0d79d77 | 2018-03-30 20:02:33 +0200 | [diff] [blame^] | 627 | && !last_char_is(x, '/') |
James Byrne | 0c63299 | 2017-05-15 21:39:51 +0200 | [diff] [blame] | 628 | ) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 629 | if (chdir(varservice) == -1) |
| 630 | goto chdir_failed_0; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 631 | } |
Denys Vlasenko | b3ba9e2 | 2009-07-15 00:24:08 +0200 | [diff] [blame] | 632 | if (chdir(x) == -1) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 633 | chdir_failed_0: |
Denys Vlasenko | 6331cf0 | 2009-11-13 09:08:27 +0100 | [diff] [blame] | 634 | fail("can't change to service directory"); |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 635 | goto nullify_service_0; |
| 636 | } |
| 637 | if (act && (act(acts) == -1)) { |
| 638 | nullify_service_0: |
Denys Vlasenko | b3ba9e2 | 2009-07-15 00:24:08 +0200 | [diff] [blame] | 639 | *service = (char*) -1L; /* "dead" */ |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 640 | } |
| 641 | if (fchdir(curdir) == -1) |
| 642 | fatal_cannot("change to original directory"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 643 | service++; |
| 644 | } |
| 645 | |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 646 | if (cbk) while (1) { |
Denys Vlasenko | b3ba9e2 | 2009-07-15 00:24:08 +0200 | [diff] [blame] | 647 | int want_exit; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 648 | int diff; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 649 | |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 650 | diff = tnow - tstart; |
Denys Vlasenko | b3ba9e2 | 2009-07-15 00:24:08 +0200 | [diff] [blame] | 651 | service = argv; |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 652 | want_exit = 1; |
Denys Vlasenko | b3ba9e2 | 2009-07-15 00:24:08 +0200 | [diff] [blame] | 653 | while ((x = *service) != NULL) { |
| 654 | if (x == (char*) -1L) /* "dead" */ |
| 655 | goto next; |
| 656 | if (x[0] != '/' && x[0] != '.') { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 657 | if (chdir(varservice) == -1) |
| 658 | goto chdir_failed; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 659 | } |
Denys Vlasenko | b3ba9e2 | 2009-07-15 00:24:08 +0200 | [diff] [blame] | 660 | if (chdir(x) == -1) { |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 661 | chdir_failed: |
Denys Vlasenko | 6331cf0 | 2009-11-13 09:08:27 +0100 | [diff] [blame] | 662 | fail("can't change to service directory"); |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 663 | goto nullify_service; |
| 664 | } |
| 665 | if (cbk(acts) != 0) |
| 666 | goto nullify_service; |
| 667 | want_exit = 0; |
| 668 | if (diff >= waitsec) { |
| 669 | printf(kll ? "kill: " : "timeout: "); |
| 670 | if (svstatus_get() > 0) { |
Denys Vlasenko | b3ba9e2 | 2009-07-15 00:24:08 +0200 | [diff] [blame] | 671 | svstatus_print(x); |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 672 | ++rc; |
| 673 | } |
Denis Vlasenko | 4daad90 | 2007-09-27 10:20:47 +0000 | [diff] [blame] | 674 | bb_putchar('\n'); /* will also flush the output */ |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 675 | if (kll) |
| 676 | control("k"); |
| 677 | nullify_service: |
Denys Vlasenko | b3ba9e2 | 2009-07-15 00:24:08 +0200 | [diff] [blame] | 678 | *service = (char*) -1L; /* "dead" */ |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 679 | } |
| 680 | if (fchdir(curdir) == -1) |
| 681 | fatal_cannot("change to original directory"); |
Denys Vlasenko | b3ba9e2 | 2009-07-15 00:24:08 +0200 | [diff] [blame] | 682 | next: |
| 683 | service++; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 684 | } |
Denis Vlasenko | 7fca91a3 | 2007-02-02 01:19:09 +0000 | [diff] [blame] | 685 | if (want_exit) break; |
| 686 | usleep(420000); |
Denis Vlasenko | 04158e0 | 2009-02-02 10:48:06 +0000 | [diff] [blame] | 687 | tnow = time(NULL) + 0x400000000000000aULL; |
Denis Vlasenko | 8c78395 | 2007-01-27 22:21:52 +0000 | [diff] [blame] | 688 | } |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 689 | return rc > 99 ? 99 : rc; |
| 690 | } |
Denys Vlasenko | 4051a99 | 2016-12-05 13:56:40 +0100 | [diff] [blame] | 691 | |
| 692 | #if ENABLE_SV |
| 693 | int sv_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
| 694 | int sv_main(int argc UNUSED_PARAM, char **argv) |
| 695 | { |
| 696 | return sv(argv); |
| 697 | } |
| 698 | #endif |
| 699 | |
| 700 | //usage:#define svc_trivial_usage |
| 701 | //usage: "[-udopchaitkx] SERVICE_DIR..." |
| 702 | //usage:#define svc_full_usage "\n\n" |
| 703 | //usage: "Control services monitored by runsv supervisor" |
| 704 | //usage: "\n" |
| 705 | //usage: "\n"" -u If service is not running, start it; restart if it stops" |
| 706 | //usage: "\n"" -d If service is running, send TERM+CONT signals; do not restart it" |
| 707 | //usage: "\n"" -o Once: if service is not running, start it; do not restart it" |
| 708 | //usage: "\n"" -pchaitk Send STOP, CONT, HUP, ALRM, INT, TERM, KILL signal to service" |
| 709 | //usage: "\n"" -x Exit: runsv will exit as soon as the service is down" |
| 710 | #if ENABLE_SVC |
| 711 | int svc_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
| 712 | int svc_main(int argc UNUSED_PARAM, char **argv) |
| 713 | { |
| 714 | char command[2]; |
| 715 | const char *optstring; |
| 716 | unsigned opts; |
| 717 | |
Denys Vlasenko | 4051a99 | 2016-12-05 13:56:40 +0100 | [diff] [blame] | 718 | optstring = "udopchaitkx"; |
| 719 | opts = getopt32(argv, optstring); |
| 720 | argv += optind; |
| 721 | if (!argv[0] || !opts) |
| 722 | bb_show_usage(); |
| 723 | |
| 724 | argv -= 2; |
| 725 | if (optind > 2) { |
| 726 | argv--; |
| 727 | argv[2] = (char*)"--"; |
| 728 | } |
| 729 | argv[0] = (char*)"sv"; |
| 730 | argv[1] = command; |
| 731 | command[1] = '\0'; |
| 732 | |
Denys Vlasenko | 4051a99 | 2016-12-05 13:56:40 +0100 | [diff] [blame] | 733 | do { |
| 734 | if (opts & 1) { |
| 735 | int r; |
Denys Vlasenko | 754e9f9 | 2017-08-05 01:38:55 +0200 | [diff] [blame] | 736 | |
Denys Vlasenko | 4051a99 | 2016-12-05 13:56:40 +0100 | [diff] [blame] | 737 | command[0] = *optstring; |
Denys Vlasenko | 754e9f9 | 2017-08-05 01:38:55 +0200 | [diff] [blame] | 738 | |
| 739 | /* getopt() was already called by getopt32(): |
| 740 | * reset the libc getopt() function's internal state. |
| 741 | */ |
| 742 | GETOPT_RESET(); |
Denys Vlasenko | 4051a99 | 2016-12-05 13:56:40 +0100 | [diff] [blame] | 743 | r = sv(argv); |
| 744 | if (r) |
| 745 | return 1; |
| 746 | } |
| 747 | optstring++; |
| 748 | opts >>= 1; |
| 749 | } while (opts); |
| 750 | |
| 751 | return 0; |
| 752 | } |
| 753 | #endif |
Denys Vlasenko | 0d79d77 | 2018-03-30 20:02:33 +0200 | [diff] [blame^] | 754 | |
| 755 | //usage:#define svok_trivial_usage |
| 756 | //usage: "SERVICE_DIR" |
| 757 | //usage:#define svok_full_usage "\n\n" |
| 758 | //usage: "Check whether runsv supervisor is running.\n" |
| 759 | //usage: "Exit code is 0 if it does, 100 if it does not,\n" |
| 760 | //usage: "111 (with error message) if SERVICE_DIR does not exist." |
| 761 | #if ENABLE_SVOK |
| 762 | int svok_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
| 763 | int svok_main(int argc UNUSED_PARAM, char **argv) |
| 764 | { |
| 765 | const char *dir = argv[1]; |
| 766 | |
| 767 | if (!dir) |
| 768 | bb_show_usage(); |
| 769 | |
| 770 | xfunc_error_retval = 111; |
| 771 | |
| 772 | /* |
| 773 | * daemontools has no concept of "default service dir", runit does. |
| 774 | * Let's act as runit. |
| 775 | */ |
| 776 | if (dir[0] != '/' && dir[0] != '.' |
| 777 | && !last_char_is(dir, '/') |
| 778 | ) { |
| 779 | xchdir(CONFIG_SV_DEFAULT_SERVICE_DIR); |
| 780 | } |
| 781 | |
| 782 | xchdir(dir); |
| 783 | if (open("supervise/ok", O_WRONLY) < 0) { |
| 784 | if (errno == ENOENT || errno == ENXIO) |
| 785 | return 100; |
| 786 | bb_perror_msg_and_die("can't open '%s'", "supervise/ok"); |
| 787 | } |
| 788 | |
| 789 | return 0; |
| 790 | } |
| 791 | #endif |