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 | |
Denis Vlasenko | d18f52b | 2008-03-02 12:53:15 +0000 | [diff] [blame] | 28 | /* Busyboxed by Denys Vlasenko <vda.linux@googlemail.com> */ |
Denys Vlasenko | 0863e1a | 2015-10-19 00:41:28 +0200 | [diff] [blame] | 29 | |
| 30 | //config:config RUNSV |
Denys Vlasenko | b097a84 | 2018-12-28 03:20:17 +0100 | [diff] [blame] | 31 | //config: bool "runsv (7.8 kb)" |
Denys Vlasenko | 0863e1a | 2015-10-19 00:41:28 +0200 | [diff] [blame] | 32 | //config: default y |
| 33 | //config: help |
Denys Vlasenko | 72089cf | 2017-07-21 09:50:55 +0200 | [diff] [blame] | 34 | //config: runsv starts and monitors a service and optionally an appendant log |
| 35 | //config: service. |
Denys Vlasenko | 0863e1a | 2015-10-19 00:41:28 +0200 | [diff] [blame] | 36 | |
| 37 | //applet:IF_RUNSV(APPLET(runsv, BB_DIR_USR_BIN, BB_SUID_DROP)) |
| 38 | |
| 39 | //kbuild:lib-$(CONFIG_RUNSV) += runsv.o |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 40 | |
Pere Orga | 5bc8c00 | 2011-04-11 03:29:49 +0200 | [diff] [blame] | 41 | //usage:#define runsv_trivial_usage |
| 42 | //usage: "DIR" |
| 43 | //usage:#define runsv_full_usage "\n\n" |
| 44 | //usage: "Start and monitor a service and optionally an appendant log service" |
| 45 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 46 | #include <sys/file.h> |
Denis Vlasenko | b6adbf1 | 2007-05-26 19:00:18 +0000 | [diff] [blame] | 47 | #include "libbb.h" |
Denys Vlasenko | e6a2f4c | 2016-04-21 16:26:30 +0200 | [diff] [blame] | 48 | #include "common_bufsiz.h" |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 49 | #include "runit_lib.h" |
| 50 | |
Denis Vlasenko | 7bc5360 | 2007-08-31 21:45:52 +0000 | [diff] [blame] | 51 | #if ENABLE_MONOTONIC_SYSCALL |
| 52 | #include <sys/syscall.h> |
| 53 | |
Denis Vlasenko | 7bc5360 | 2007-08-31 21:45:52 +0000 | [diff] [blame] | 54 | static void gettimeofday_ns(struct timespec *ts) |
| 55 | { |
Denys Vlasenko | be5a505 | 2019-10-24 16:26:55 +0200 | [diff] [blame] | 56 | clock_gettime(CLOCK_REALTIME, ts); |
Denis Vlasenko | 7bc5360 | 2007-08-31 21:45:52 +0000 | [diff] [blame] | 57 | } |
| 58 | #else |
| 59 | static void gettimeofday_ns(struct timespec *ts) |
| 60 | { |
Denys Vlasenko | ab3964d | 2015-10-13 14:50:20 +0200 | [diff] [blame] | 61 | BUILD_BUG_ON(sizeof(struct timeval) != sizeof(struct timespec)); |
| 62 | BUILD_BUG_ON(sizeof(((struct timeval*)ts)->tv_usec) != sizeof(ts->tv_nsec)); |
| 63 | /* Cheat */ |
| 64 | gettimeofday((void*)ts, NULL); |
| 65 | ts->tv_nsec *= 1000; |
Denis Vlasenko | 7bc5360 | 2007-08-31 21:45:52 +0000 | [diff] [blame] | 66 | } |
| 67 | #endif |
| 68 | |
| 69 | /* Compare possibly overflowing unsigned counters */ |
| 70 | #define LESS(a,b) ((int)((unsigned)(b) - (unsigned)(a)) > 0) |
| 71 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 72 | /* state */ |
| 73 | #define S_DOWN 0 |
| 74 | #define S_RUN 1 |
| 75 | #define S_FINISH 2 |
| 76 | /* ctrl */ |
| 77 | #define C_NOOP 0 |
| 78 | #define C_TERM 1 |
| 79 | #define C_PAUSE 2 |
| 80 | /* want */ |
| 81 | #define W_UP 0 |
| 82 | #define W_DOWN 1 |
| 83 | #define W_EXIT 2 |
| 84 | |
| 85 | struct svdir { |
| 86 | int pid; |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 87 | smallint state; |
| 88 | smallint ctrl; |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 89 | smallint sd_want; |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 90 | smallint islog; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 91 | struct timespec start; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 92 | int fdlock; |
| 93 | int fdcontrol; |
| 94 | int fdcontrolwrite; |
Earl Chew | 80a3418 | 2009-08-02 02:23:27 +0200 | [diff] [blame] | 95 | int wstat; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 96 | }; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 97 | |
Denis Vlasenko | b925605 | 2007-09-28 10:29:17 +0000 | [diff] [blame] | 98 | struct globals { |
| 99 | smallint haslog; |
| 100 | smallint sigterm; |
| 101 | smallint pidchanged; |
Denis Vlasenko | 3718832 | 2008-02-16 13:20:56 +0000 | [diff] [blame] | 102 | struct fd_pair selfpipe; |
| 103 | struct fd_pair logpipe; |
Denis Vlasenko | b925605 | 2007-09-28 10:29:17 +0000 | [diff] [blame] | 104 | char *dir; |
| 105 | struct svdir svd[2]; |
Denys Vlasenko | 98a4c7c | 2010-02-04 15:00:15 +0100 | [diff] [blame] | 106 | } FIX_ALIASING; |
Denys Vlasenko | e6a2f4c | 2016-04-21 16:26:30 +0200 | [diff] [blame] | 107 | #define G (*(struct globals*)bb_common_bufsiz1) |
Denis Vlasenko | b925605 | 2007-09-28 10:29:17 +0000 | [diff] [blame] | 108 | #define haslog (G.haslog ) |
| 109 | #define sigterm (G.sigterm ) |
| 110 | #define pidchanged (G.pidchanged ) |
| 111 | #define selfpipe (G.selfpipe ) |
| 112 | #define logpipe (G.logpipe ) |
| 113 | #define dir (G.dir ) |
| 114 | #define svd (G.svd ) |
Denis Vlasenko | 7049ff8 | 2008-06-25 09:53:17 +0000 | [diff] [blame] | 115 | #define INIT_G() do { \ |
Denys Vlasenko | 47cfbf3 | 2016-04-21 18:18:48 +0200 | [diff] [blame] | 116 | setup_common_bufsiz(); \ |
Denis Vlasenko | 7049ff8 | 2008-06-25 09:53:17 +0000 | [diff] [blame] | 117 | pidchanged = 1; \ |
| 118 | } while (0) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 119 | |
Denis Vlasenko | ab2aea4 | 2007-01-29 22:51:58 +0000 | [diff] [blame] | 120 | static void fatal2_cannot(const char *m1, const char *m2) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 121 | { |
Denys Vlasenko | 7f3a2a2 | 2015-10-08 11:24:44 +0200 | [diff] [blame] | 122 | bb_perror_msg_and_die("%s: fatal: can't %s%s", dir, m1, m2); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 123 | /* was exiting 111 */ |
| 124 | } |
Denis Vlasenko | ab2aea4 | 2007-01-29 22:51:58 +0000 | [diff] [blame] | 125 | static void fatal_cannot(const char *m) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 126 | { |
| 127 | fatal2_cannot(m, ""); |
| 128 | /* was exiting 111 */ |
| 129 | } |
Denis Vlasenko | ab2aea4 | 2007-01-29 22:51:58 +0000 | [diff] [blame] | 130 | static void fatal2x_cannot(const char *m1, const char *m2) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 131 | { |
Denys Vlasenko | 7f3a2a2 | 2015-10-08 11:24:44 +0200 | [diff] [blame] | 132 | bb_error_msg_and_die("%s: fatal: can't %s%s", dir, m1, m2); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 133 | /* was exiting 111 */ |
| 134 | } |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 135 | static void warn2_cannot(const char *m1, const char *m2) |
| 136 | { |
| 137 | bb_perror_msg("%s: warning: can't %s%s", dir, m1, m2); |
| 138 | } |
Denis Vlasenko | ab2aea4 | 2007-01-29 22:51:58 +0000 | [diff] [blame] | 139 | static void warn_cannot(const char *m) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 140 | { |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 141 | warn2_cannot(m, ""); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 142 | } |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 143 | |
Denis Vlasenko | a60f84e | 2008-07-05 09:18:54 +0000 | [diff] [blame] | 144 | static void s_child(int sig_no UNUSED_PARAM) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 145 | { |
Denis Vlasenko | 3718832 | 2008-02-16 13:20:56 +0000 | [diff] [blame] | 146 | write(selfpipe.wr, "", 1); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 147 | } |
| 148 | |
Denis Vlasenko | a60f84e | 2008-07-05 09:18:54 +0000 | [diff] [blame] | 149 | static void s_term(int sig_no UNUSED_PARAM) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 150 | { |
| 151 | sigterm = 1; |
Denis Vlasenko | 3718832 | 2008-02-16 13:20:56 +0000 | [diff] [blame] | 152 | write(selfpipe.wr, "", 1); /* XXX */ |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 153 | } |
| 154 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 155 | static int open_trunc_or_warn(const char *name) |
| 156 | { |
Denys Vlasenko | e09bff3 | 2010-09-05 19:28:29 +0200 | [diff] [blame] | 157 | /* Why O_NDELAY? */ |
| 158 | int fd = open(name, O_WRONLY | O_NDELAY | O_TRUNC | O_CREAT, 0644); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 159 | if (fd < 0) |
| 160 | bb_perror_msg("%s: warning: cannot open %s", |
| 161 | dir, name); |
| 162 | return fd; |
| 163 | } |
| 164 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 165 | static void update_status(struct svdir *s) |
| 166 | { |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 167 | ssize_t sz; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 168 | int fd; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 169 | svstatus_t status; |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 170 | const char *fstatus ="log/supervise/status"; |
| 171 | const char *fstatusnew ="log/supervise/status.new"; |
| 172 | const char *f_stat ="log/supervise/stat"; |
| 173 | const char *fstatnew ="log/supervise/stat.new"; |
| 174 | const char *fpid ="log/supervise/pid"; |
| 175 | const char *fpidnew ="log/supervise/pid.new"; |
| 176 | |
| 177 | if (!s->islog) { |
| 178 | fstatus += 4; |
| 179 | fstatusnew += 4; |
| 180 | f_stat += 4; |
| 181 | fstatnew += 4; |
| 182 | fpid += 4; |
| 183 | fpidnew += 4; |
| 184 | } |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 185 | |
| 186 | /* pid */ |
| 187 | if (pidchanged) { |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 188 | fd = open_trunc_or_warn(fpidnew); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 189 | if (fd < 0) |
| 190 | return; |
| 191 | if (s->pid) { |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 192 | char spid[sizeof(int)*3 + 2]; |
| 193 | int size = sprintf(spid, "%u\n", (unsigned)s->pid); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 194 | write(fd, spid, size); |
| 195 | } |
| 196 | close(fd); |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 197 | if (rename_or_warn(fpidnew, fpid)) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 198 | return; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 199 | pidchanged = 0; |
| 200 | } |
| 201 | |
| 202 | /* stat */ |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 203 | fd = open_trunc_or_warn(fstatnew); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 204 | if (fd < -1) |
| 205 | return; |
| 206 | |
| 207 | { |
| 208 | char stat_buf[sizeof("finish, paused, got TERM, want down\n")]; |
| 209 | char *p = stat_buf; |
| 210 | switch (s->state) { |
| 211 | case S_DOWN: |
Dan Fandrich | dc50676 | 2011-02-12 22:26:57 -0800 | [diff] [blame] | 212 | p = stpcpy(p, "down"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 213 | break; |
| 214 | case S_RUN: |
Dan Fandrich | dc50676 | 2011-02-12 22:26:57 -0800 | [diff] [blame] | 215 | p = stpcpy(p, "run"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 216 | break; |
| 217 | case S_FINISH: |
Dan Fandrich | dc50676 | 2011-02-12 22:26:57 -0800 | [diff] [blame] | 218 | p = stpcpy(p, "finish"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 219 | break; |
| 220 | } |
Earl Chew | 80a3418 | 2009-08-02 02:23:27 +0200 | [diff] [blame] | 221 | if (s->ctrl & C_PAUSE) |
Dan Fandrich | dc50676 | 2011-02-12 22:26:57 -0800 | [diff] [blame] | 222 | p = stpcpy(p, ", paused"); |
Earl Chew | 80a3418 | 2009-08-02 02:23:27 +0200 | [diff] [blame] | 223 | if (s->ctrl & C_TERM) |
Dan Fandrich | dc50676 | 2011-02-12 22:26:57 -0800 | [diff] [blame] | 224 | p = stpcpy(p, ", got TERM"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 225 | if (s->state != S_DOWN) |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 226 | switch (s->sd_want) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 227 | case W_DOWN: |
Dan Fandrich | dc50676 | 2011-02-12 22:26:57 -0800 | [diff] [blame] | 228 | p = stpcpy(p, ", want down"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 229 | break; |
| 230 | case W_EXIT: |
Dan Fandrich | dc50676 | 2011-02-12 22:26:57 -0800 | [diff] [blame] | 231 | p = stpcpy(p, ", want exit"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 232 | break; |
| 233 | } |
| 234 | *p++ = '\n'; |
| 235 | write(fd, stat_buf, p - stat_buf); |
| 236 | close(fd); |
| 237 | } |
| 238 | |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 239 | rename_or_warn(fstatnew, f_stat); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 240 | |
| 241 | /* supervise compatibility */ |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 242 | memset(&status, 0, sizeof(status)); |
| 243 | status.time_be64 = SWAP_BE64(s->start.tv_sec + 0x400000000000000aULL); |
| 244 | status.time_nsec_be32 = SWAP_BE32(s->start.tv_nsec); |
| 245 | status.pid_le32 = SWAP_LE32(s->pid); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 246 | if (s->ctrl & C_PAUSE) |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 247 | status.paused = 1; |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 248 | if (s->sd_want == W_UP) |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 249 | status.want = 'u'; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 250 | else |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 251 | status.want = 'd'; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 252 | if (s->ctrl & C_TERM) |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 253 | status.got_term = 1; |
| 254 | status.run_or_finish = s->state; |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 255 | fd = open_trunc_or_warn(fstatusnew); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 256 | if (fd < 0) |
| 257 | return; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 258 | sz = write(fd, &status, sizeof(status)); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 259 | close(fd); |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 260 | if (sz != sizeof(status)) { |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 261 | warn2_cannot("write ", fstatusnew); |
| 262 | unlink(fstatusnew); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 263 | return; |
| 264 | } |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 265 | rename_or_warn(fstatusnew, fstatus); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 266 | } |
| 267 | |
| 268 | static unsigned custom(struct svdir *s, char c) |
| 269 | { |
Denis Vlasenko | 3854c5d | 2008-11-06 22:39:57 +0000 | [diff] [blame] | 270 | pid_t pid; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 271 | int w; |
| 272 | char a[10]; |
| 273 | struct stat st; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 274 | |
Denys Vlasenko | 8f24f98 | 2009-06-07 16:02:00 +0200 | [diff] [blame] | 275 | if (s->islog) |
| 276 | return 0; |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 277 | strcpy(a, "control/?"); |
Denis Vlasenko | 592d4fe | 2008-03-17 09:19:26 +0000 | [diff] [blame] | 278 | a[8] = c; /* replace '?' */ |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 279 | if (stat(a, &st) == 0) { |
| 280 | if (st.st_mode & S_IXUSR) { |
Denis Vlasenko | 592d4fe | 2008-03-17 09:19:26 +0000 | [diff] [blame] | 281 | pid = vfork(); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 282 | if (pid == -1) { |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 283 | warn2_cannot("vfork for ", a); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 284 | return 0; |
| 285 | } |
Earl Chew | 80a3418 | 2009-08-02 02:23:27 +0200 | [diff] [blame] | 286 | if (pid == 0) { |
Denis Vlasenko | 592d4fe | 2008-03-17 09:19:26 +0000 | [diff] [blame] | 287 | /* child */ |
Denis Vlasenko | 3718832 | 2008-02-16 13:20:56 +0000 | [diff] [blame] | 288 | if (haslog && dup2(logpipe.wr, 1) == -1) |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 289 | warn2_cannot("setup stdout for ", a); |
Denis Vlasenko | f09f4e0 | 2009-02-26 12:29:59 +0000 | [diff] [blame] | 290 | execl(a, a, (char *) NULL); |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 291 | fatal2_cannot("run ", a); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 292 | } |
Denis Vlasenko | 592d4fe | 2008-03-17 09:19:26 +0000 | [diff] [blame] | 293 | /* parent */ |
Denis Vlasenko | f09f4e0 | 2009-02-26 12:29:59 +0000 | [diff] [blame] | 294 | if (safe_waitpid(pid, &w, 0) == -1) { |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 295 | warn2_cannot("wait for child ", a); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 296 | return 0; |
| 297 | } |
Denys Vlasenko | 8f24f98 | 2009-06-07 16:02:00 +0200 | [diff] [blame] | 298 | return WEXITSTATUS(w) == 0; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 299 | } |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 300 | } else { |
| 301 | if (errno != ENOENT) |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 302 | warn2_cannot("stat ", a); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 303 | } |
| 304 | return 0; |
| 305 | } |
| 306 | |
| 307 | static void stopservice(struct svdir *s) |
| 308 | { |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 309 | if (s->pid && !custom(s, 't')) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 310 | kill(s->pid, SIGTERM); |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 311 | s->ctrl |= C_TERM; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 312 | update_status(s); |
| 313 | } |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 314 | if (s->sd_want == W_DOWN) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 315 | kill(s->pid, SIGCONT); |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 316 | custom(s, 'd'); |
| 317 | return; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 318 | } |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 319 | if (s->sd_want == W_EXIT) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 320 | kill(s->pid, SIGCONT); |
| 321 | custom(s, 'x'); |
| 322 | } |
| 323 | } |
| 324 | |
| 325 | static void startservice(struct svdir *s) |
| 326 | { |
| 327 | int p; |
Earl Chew | 80a3418 | 2009-08-02 02:23:27 +0200 | [diff] [blame] | 328 | const char *arg[4]; |
| 329 | char exitcode[sizeof(int)*3 + 2]; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 330 | |
Earl Chew | 80a3418 | 2009-08-02 02:23:27 +0200 | [diff] [blame] | 331 | if (s->state == S_FINISH) { |
| 332 | /* Two arguments are given to ./finish. The first one is ./run exit code, |
| 333 | * or -1 if ./run didnt exit normally. The second one is |
| 334 | * the least significant byte of the exit status as determined by waitpid; |
| 335 | * for instance it is 0 if ./run exited normally, and the signal number |
| 336 | * if ./run was terminated by a signal. If runsv cannot start ./run |
| 337 | * for some reason, the exit code is 111 and the status is 0. |
| 338 | */ |
| 339 | arg[0] = "./finish"; |
| 340 | arg[1] = "-1"; |
| 341 | if (WIFEXITED(s->wstat)) { |
Denys Vlasenko | 7bb346f | 2009-10-06 22:09:50 +0200 | [diff] [blame] | 342 | *utoa_to_buf(WEXITSTATUS(s->wstat), exitcode, sizeof(exitcode)) = '\0'; |
Earl Chew | 80a3418 | 2009-08-02 02:23:27 +0200 | [diff] [blame] | 343 | arg[1] = exitcode; |
| 344 | } |
| 345 | //arg[2] = "0"; |
| 346 | //if (WIFSIGNALED(s->wstat)) { |
Denys Vlasenko | 7bb346f | 2009-10-06 22:09:50 +0200 | [diff] [blame] | 347 | arg[2] = utoa(WTERMSIG(s->wstat)); |
Earl Chew | 80a3418 | 2009-08-02 02:23:27 +0200 | [diff] [blame] | 348 | //} |
| 349 | arg[3] = NULL; |
| 350 | } else { |
| 351 | arg[0] = "./run"; |
| 352 | arg[1] = NULL; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 353 | custom(s, 'u'); |
| 354 | } |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 355 | |
Denis Vlasenko | ff131b9 | 2007-04-10 15:42:06 +0000 | [diff] [blame] | 356 | if (s->pid != 0) |
| 357 | stopservice(s); /* should never happen */ |
Denis Vlasenko | 592d4fe | 2008-03-17 09:19:26 +0000 | [diff] [blame] | 358 | while ((p = vfork()) == -1) { |
| 359 | warn_cannot("vfork, sleeping"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 360 | sleep(5); |
| 361 | } |
| 362 | if (p == 0) { |
| 363 | /* child */ |
| 364 | if (haslog) { |
Denis Vlasenko | 3718832 | 2008-02-16 13:20:56 +0000 | [diff] [blame] | 365 | /* NB: bug alert! right order is close, then dup2 */ |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 366 | if (s->islog) { |
Denis Vlasenko | a27a11b | 2007-08-18 14:16:39 +0000 | [diff] [blame] | 367 | xchdir("./log"); |
Denis Vlasenko | 3718832 | 2008-02-16 13:20:56 +0000 | [diff] [blame] | 368 | close(logpipe.wr); |
| 369 | xdup2(logpipe.rd, 0); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 370 | } else { |
Denis Vlasenko | 3718832 | 2008-02-16 13:20:56 +0000 | [diff] [blame] | 371 | close(logpipe.rd); |
| 372 | xdup2(logpipe.wr, 1); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 373 | } |
| 374 | } |
Denis Vlasenko | 3fa36e2 | 2008-11-09 00:15:11 +0000 | [diff] [blame] | 375 | /* Non-ignored signals revert to SIG_DFL on exec anyway */ |
| 376 | /*bb_signals(0 |
Denis Vlasenko | 25591c3 | 2008-02-16 22:58:56 +0000 | [diff] [blame] | 377 | + (1 << SIGCHLD) |
| 378 | + (1 << SIGTERM) |
Denis Vlasenko | 3fa36e2 | 2008-11-09 00:15:11 +0000 | [diff] [blame] | 379 | , SIG_DFL);*/ |
Denis Vlasenko | 8c78395 | 2007-01-27 22:21:52 +0000 | [diff] [blame] | 380 | sig_unblock(SIGCHLD); |
Denis Vlasenko | 8c78395 | 2007-01-27 22:21:52 +0000 | [diff] [blame] | 381 | sig_unblock(SIGTERM); |
Earl Chew | 80a3418 | 2009-08-02 02:23:27 +0200 | [diff] [blame] | 382 | execv(arg[0], (char**) arg); |
| 383 | fatal2_cannot(s->islog ? "start log/" : "start ", arg[0]); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 384 | } |
Denis Vlasenko | 592d4fe | 2008-03-17 09:19:26 +0000 | [diff] [blame] | 385 | /* parent */ |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 386 | if (s->state != S_FINISH) { |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 387 | gettimeofday_ns(&s->start); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 388 | s->state = S_RUN; |
| 389 | } |
| 390 | s->pid = p; |
| 391 | pidchanged = 1; |
| 392 | s->ctrl = C_NOOP; |
| 393 | update_status(s); |
| 394 | } |
| 395 | |
| 396 | static int ctrl(struct svdir *s, char c) |
| 397 | { |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 398 | int sig; |
| 399 | |
Denis Vlasenko | bf0a201 | 2006-12-26 10:42:51 +0000 | [diff] [blame] | 400 | switch (c) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 401 | case 'd': /* down */ |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 402 | s->sd_want = W_DOWN; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 403 | update_status(s); |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 404 | if (s->state == S_RUN) |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 405 | stopservice(s); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 406 | break; |
| 407 | case 'u': /* up */ |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 408 | s->sd_want = W_UP; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 409 | update_status(s); |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 410 | if (s->state == S_DOWN) |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 411 | startservice(s); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 412 | break; |
| 413 | case 'x': /* exit */ |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 414 | if (s->islog) |
| 415 | break; |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 416 | s->sd_want = W_EXIT; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 417 | update_status(s); |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 418 | /* FALLTHROUGH */ |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 419 | case 't': /* sig term */ |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 420 | if (s->state == S_RUN) |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 421 | stopservice(s); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 422 | break; |
| 423 | case 'k': /* sig kill */ |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 424 | if ((s->state == S_RUN) && !custom(s, c)) |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 425 | kill(s->pid, SIGKILL); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 426 | s->state = S_DOWN; |
| 427 | break; |
| 428 | case 'p': /* sig pause */ |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 429 | if ((s->state == S_RUN) && !custom(s, c)) |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 430 | kill(s->pid, SIGSTOP); |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 431 | s->ctrl |= C_PAUSE; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 432 | update_status(s); |
| 433 | break; |
| 434 | case 'c': /* sig cont */ |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 435 | if ((s->state == S_RUN) && !custom(s, c)) |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 436 | kill(s->pid, SIGCONT); |
Denis Vlasenko | 31773b7 | 2009-02-26 12:38:01 +0000 | [diff] [blame] | 437 | s->ctrl &= ~C_PAUSE; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 438 | update_status(s); |
| 439 | break; |
| 440 | case 'o': /* once */ |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 441 | s->sd_want = W_DOWN; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 442 | update_status(s); |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 443 | if (s->state == S_DOWN) |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 444 | startservice(s); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 445 | break; |
| 446 | case 'a': /* sig alarm */ |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 447 | sig = SIGALRM; |
| 448 | goto sendsig; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 449 | case 'h': /* sig hup */ |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 450 | sig = SIGHUP; |
| 451 | goto sendsig; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 452 | case 'i': /* sig int */ |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 453 | sig = SIGINT; |
| 454 | goto sendsig; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 455 | case 'q': /* sig quit */ |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 456 | sig = SIGQUIT; |
| 457 | goto sendsig; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 458 | case '1': /* sig usr1 */ |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 459 | sig = SIGUSR1; |
| 460 | goto sendsig; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 461 | case '2': /* sig usr2 */ |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 462 | sig = SIGUSR2; |
| 463 | goto sendsig; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 464 | } |
| 465 | return 1; |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 466 | sendsig: |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 467 | if ((s->state == S_RUN) && !custom(s, c)) |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 468 | kill(s->pid, sig); |
| 469 | return 1; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 470 | } |
| 471 | |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 472 | static void open_control(const char *f, struct svdir *s) |
| 473 | { |
| 474 | struct stat st; |
| 475 | mkfifo(f, 0600); |
| 476 | if (stat(f, &st) == -1) |
| 477 | fatal2_cannot("stat ", f); |
| 478 | if (!S_ISFIFO(st.st_mode)) |
| 479 | bb_error_msg_and_die("%s: fatal: %s exists but is not a fifo", dir, f); |
| 480 | s->fdcontrol = xopen(f, O_RDONLY|O_NDELAY); |
| 481 | close_on_exec_on(s->fdcontrol); |
| 482 | s->fdcontrolwrite = xopen(f, O_WRONLY|O_NDELAY); |
| 483 | close_on_exec_on(s->fdcontrolwrite); |
| 484 | update_status(s); |
| 485 | } |
| 486 | |
Denis Vlasenko | 9b49a5e | 2007-10-11 10:05:36 +0000 | [diff] [blame] | 487 | int runsv_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
Denis Vlasenko | a60f84e | 2008-07-05 09:18:54 +0000 | [diff] [blame] | 488 | int runsv_main(int argc UNUSED_PARAM, char **argv) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 489 | { |
| 490 | struct stat s; |
| 491 | int fd; |
| 492 | int r; |
| 493 | char buf[256]; |
| 494 | |
Denis Vlasenko | b925605 | 2007-09-28 10:29:17 +0000 | [diff] [blame] | 495 | INIT_G(); |
| 496 | |
Denys Vlasenko | e992bae | 2009-11-28 15:18:53 +0100 | [diff] [blame] | 497 | dir = single_argv(argv); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 498 | |
Denis Vlasenko | 3718832 | 2008-02-16 13:20:56 +0000 | [diff] [blame] | 499 | xpiped_pair(selfpipe); |
| 500 | close_on_exec_on(selfpipe.rd); |
| 501 | close_on_exec_on(selfpipe.wr); |
| 502 | ndelay_on(selfpipe.rd); |
| 503 | ndelay_on(selfpipe.wr); |
Denis Vlasenko | f7996f3 | 2007-01-11 17:20:00 +0000 | [diff] [blame] | 504 | |
Denis Vlasenko | 8c78395 | 2007-01-27 22:21:52 +0000 | [diff] [blame] | 505 | sig_block(SIGCHLD); |
Denis Vlasenko | cab28aa | 2009-01-31 01:02:07 +0000 | [diff] [blame] | 506 | bb_signals_recursive_norestart(1 << SIGCHLD, s_child); |
Denis Vlasenko | 8c78395 | 2007-01-27 22:21:52 +0000 | [diff] [blame] | 507 | sig_block(SIGTERM); |
Denis Vlasenko | cab28aa | 2009-01-31 01:02:07 +0000 | [diff] [blame] | 508 | bb_signals_recursive_norestart(1 << SIGTERM, s_term); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 509 | |
| 510 | xchdir(dir); |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 511 | /* bss: svd[0].pid = 0; */ |
| 512 | if (S_DOWN) svd[0].state = S_DOWN; /* otherwise already 0 (bss) */ |
| 513 | if (C_NOOP) svd[0].ctrl = C_NOOP; |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 514 | if (W_UP) svd[0].sd_want = W_UP; |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 515 | /* bss: svd[0].islog = 0; */ |
| 516 | /* bss: svd[1].pid = 0; */ |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 517 | gettimeofday_ns(&svd[0].start); |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 518 | if (stat("down", &s) != -1) |
| 519 | svd[0].sd_want = W_DOWN; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 520 | |
| 521 | if (stat("log", &s) == -1) { |
| 522 | if (errno != ENOENT) |
| 523 | warn_cannot("stat ./log"); |
| 524 | } else { |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 525 | if (!S_ISDIR(s.st_mode)) { |
| 526 | errno = 0; |
| 527 | warn_cannot("stat log/down: log is not a directory"); |
| 528 | } else { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 529 | haslog = 1; |
| 530 | svd[1].state = S_DOWN; |
| 531 | svd[1].ctrl = C_NOOP; |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 532 | svd[1].sd_want = W_UP; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 533 | svd[1].islog = 1; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 534 | gettimeofday_ns(&svd[1].start); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 535 | if (stat("log/down", &s) != -1) |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 536 | svd[1].sd_want = W_DOWN; |
Denis Vlasenko | 3718832 | 2008-02-16 13:20:56 +0000 | [diff] [blame] | 537 | xpiped_pair(logpipe); |
| 538 | close_on_exec_on(logpipe.rd); |
| 539 | close_on_exec_on(logpipe.wr); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 540 | } |
| 541 | } |
| 542 | |
| 543 | if (mkdir("supervise", 0700) == -1) { |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 544 | r = readlink("supervise", buf, sizeof(buf)); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 545 | if (r != -1) { |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 546 | if (r == sizeof(buf)) |
| 547 | fatal2x_cannot("readlink ./supervise", ": name too long"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 548 | buf[r] = 0; |
| 549 | mkdir(buf, 0700); |
| 550 | } else { |
| 551 | if ((errno != ENOENT) && (errno != EINVAL)) |
| 552 | fatal_cannot("readlink ./supervise"); |
| 553 | } |
| 554 | } |
| 555 | svd[0].fdlock = xopen3("log/supervise/lock"+4, |
| 556 | O_WRONLY|O_NDELAY|O_APPEND|O_CREAT, 0600); |
Denys Vlasenko | 05e8605 | 2010-10-13 12:53:27 +0200 | [diff] [blame] | 557 | if (flock(svd[0].fdlock, LOCK_EX | LOCK_NB) == -1) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 558 | fatal_cannot("lock supervise/lock"); |
Denis Vlasenko | 96e1b38 | 2007-09-30 23:50:48 +0000 | [diff] [blame] | 559 | close_on_exec_on(svd[0].fdlock); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 560 | if (haslog) { |
| 561 | if (mkdir("log/supervise", 0700) == -1) { |
| 562 | r = readlink("log/supervise", buf, 256); |
| 563 | if (r != -1) { |
| 564 | if (r == 256) |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 565 | fatal2x_cannot("readlink ./log/supervise", ": name too long"); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 566 | buf[r] = 0; |
| 567 | fd = xopen(".", O_RDONLY|O_NDELAY); |
| 568 | xchdir("./log"); |
| 569 | mkdir(buf, 0700); |
| 570 | if (fchdir(fd) == -1) |
| 571 | fatal_cannot("change back to service directory"); |
| 572 | close(fd); |
| 573 | } |
| 574 | else { |
| 575 | if ((errno != ENOENT) && (errno != EINVAL)) |
| 576 | fatal_cannot("readlink ./log/supervise"); |
| 577 | } |
| 578 | } |
| 579 | svd[1].fdlock = xopen3("log/supervise/lock", |
| 580 | O_WRONLY|O_NDELAY|O_APPEND|O_CREAT, 0600); |
Denys Vlasenko | 05e8605 | 2010-10-13 12:53:27 +0200 | [diff] [blame] | 581 | if (flock(svd[1].fdlock, LOCK_EX) == -1) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 582 | fatal_cannot("lock log/supervise/lock"); |
Denis Vlasenko | 96e1b38 | 2007-09-30 23:50:48 +0000 | [diff] [blame] | 583 | close_on_exec_on(svd[1].fdlock); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 584 | } |
| 585 | |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 586 | open_control("log/supervise/control"+4, &svd[0]); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 587 | if (haslog) { |
Denys Vlasenko | bcb5764 | 2017-05-15 19:44:48 +0200 | [diff] [blame] | 588 | open_control("log/supervise/control", &svd[1]); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 589 | } |
Denis Vlasenko | e2473f8 | 2007-01-27 22:22:17 +0000 | [diff] [blame] | 590 | mkfifo("log/supervise/ok"+4, 0600); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 591 | fd = xopen("log/supervise/ok"+4, O_RDONLY|O_NDELAY); |
Denis Vlasenko | 96e1b38 | 2007-09-30 23:50:48 +0000 | [diff] [blame] | 592 | close_on_exec_on(fd); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 593 | if (haslog) { |
Denis Vlasenko | e2473f8 | 2007-01-27 22:22:17 +0000 | [diff] [blame] | 594 | mkfifo("log/supervise/ok", 0600); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 595 | fd = xopen("log/supervise/ok", O_RDONLY|O_NDELAY); |
Denis Vlasenko | 96e1b38 | 2007-09-30 23:50:48 +0000 | [diff] [blame] | 596 | close_on_exec_on(fd); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 597 | } |
| 598 | for (;;) { |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 599 | struct pollfd x[3]; |
| 600 | unsigned deadline; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 601 | char ch; |
| 602 | |
| 603 | if (haslog) |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 604 | if (!svd[1].pid && svd[1].sd_want == W_UP) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 605 | startservice(&svd[1]); |
| 606 | if (!svd[0].pid) |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 607 | if (svd[0].sd_want == W_UP || svd[0].state == S_FINISH) |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 608 | startservice(&svd[0]); |
| 609 | |
Denis Vlasenko | 3718832 | 2008-02-16 13:20:56 +0000 | [diff] [blame] | 610 | x[0].fd = selfpipe.rd; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 611 | x[0].events = POLLIN; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 612 | x[1].fd = svd[0].fdcontrol; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 613 | x[1].events = POLLIN; |
| 614 | /* x[2] is used only if haslog == 1 */ |
| 615 | x[2].fd = svd[1].fdcontrol; |
| 616 | x[2].events = POLLIN; |
Denis Vlasenko | 8c78395 | 2007-01-27 22:21:52 +0000 | [diff] [blame] | 617 | sig_unblock(SIGTERM); |
| 618 | sig_unblock(SIGCHLD); |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 619 | poll(x, 2 + haslog, 3600*1000); |
Denis Vlasenko | 8c78395 | 2007-01-27 22:21:52 +0000 | [diff] [blame] | 620 | sig_block(SIGTERM); |
| 621 | sig_block(SIGCHLD); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 622 | |
Denis Vlasenko | 3718832 | 2008-02-16 13:20:56 +0000 | [diff] [blame] | 623 | while (read(selfpipe.rd, &ch, 1) == 1) |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 624 | continue; |
| 625 | |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 626 | for (;;) { |
Denis Vlasenko | 3854c5d | 2008-11-06 22:39:57 +0000 | [diff] [blame] | 627 | pid_t child; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 628 | int wstat; |
Denis Vlasenko | f7996f3 | 2007-01-11 17:20:00 +0000 | [diff] [blame] | 629 | |
Denis Vlasenko | fb0eba7 | 2008-01-02 19:55:04 +0000 | [diff] [blame] | 630 | child = wait_any_nohang(&wstat); |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 631 | if (!child) |
| 632 | break; |
| 633 | if ((child == -1) && (errno != EINTR)) |
| 634 | break; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 635 | if (child == svd[0].pid) { |
Earl Chew | 80a3418 | 2009-08-02 02:23:27 +0200 | [diff] [blame] | 636 | svd[0].wstat = wstat; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 637 | svd[0].pid = 0; |
| 638 | pidchanged = 1; |
Earl Chew | 80a3418 | 2009-08-02 02:23:27 +0200 | [diff] [blame] | 639 | svd[0].ctrl &= ~C_TERM; |
Denis Vlasenko | d0762e3 | 2007-02-18 11:07:43 +0000 | [diff] [blame] | 640 | if (svd[0].state != S_FINISH) { |
Denys Vlasenko | 05e8605 | 2010-10-13 12:53:27 +0200 | [diff] [blame] | 641 | fd = open("finish", O_RDONLY|O_NDELAY); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 642 | if (fd != -1) { |
| 643 | close(fd); |
| 644 | svd[0].state = S_FINISH; |
| 645 | update_status(&svd[0]); |
| 646 | continue; |
| 647 | } |
Denis Vlasenko | d0762e3 | 2007-02-18 11:07:43 +0000 | [diff] [blame] | 648 | } |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 649 | svd[0].state = S_DOWN; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 650 | deadline = svd[0].start.tv_sec + 1; |
| 651 | gettimeofday_ns(&svd[0].start); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 652 | update_status(&svd[0]); |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 653 | if (LESS(svd[0].start.tv_sec, deadline)) |
Denys Vlasenko | ec16c03 | 2020-11-29 11:37:34 +0100 | [diff] [blame^] | 654 | sleep1(); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 655 | } |
| 656 | if (haslog) { |
| 657 | if (child == svd[1].pid) { |
Earl Chew | 80a3418 | 2009-08-02 02:23:27 +0200 | [diff] [blame] | 658 | svd[0].wstat = wstat; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 659 | svd[1].pid = 0; |
| 660 | pidchanged = 1; |
| 661 | svd[1].state = S_DOWN; |
Denis Vlasenko | 3aba666 | 2007-03-09 22:46:06 +0000 | [diff] [blame] | 662 | svd[1].ctrl &= ~C_TERM; |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 663 | deadline = svd[1].start.tv_sec + 1; |
| 664 | gettimeofday_ns(&svd[1].start); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 665 | update_status(&svd[1]); |
Denis Vlasenko | 45946f8 | 2007-08-20 17:27:40 +0000 | [diff] [blame] | 666 | if (LESS(svd[1].start.tv_sec, deadline)) |
Denys Vlasenko | ec16c03 | 2020-11-29 11:37:34 +0100 | [diff] [blame^] | 667 | sleep1(); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 668 | } |
| 669 | } |
Denis Vlasenko | 3718832 | 2008-02-16 13:20:56 +0000 | [diff] [blame] | 670 | } /* for (;;) */ |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 671 | if (read(svd[0].fdcontrol, &ch, 1) == 1) |
| 672 | ctrl(&svd[0], ch); |
| 673 | if (haslog) |
| 674 | if (read(svd[1].fdcontrol, &ch, 1) == 1) |
| 675 | ctrl(&svd[1], ch); |
| 676 | |
| 677 | if (sigterm) { |
| 678 | ctrl(&svd[0], 'x'); |
| 679 | sigterm = 0; |
| 680 | } |
| 681 | |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 682 | if (svd[0].sd_want == W_EXIT && svd[0].state == S_DOWN) { |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 683 | if (svd[1].pid == 0) |
Bernhard Reutner-Fischer | 636a1f8 | 2008-05-19 09:29:47 +0000 | [diff] [blame] | 684 | _exit(EXIT_SUCCESS); |
Denys Vlasenko | e46601d | 2009-07-15 00:21:49 +0200 | [diff] [blame] | 685 | if (svd[1].sd_want != W_EXIT) { |
| 686 | svd[1].sd_want = W_EXIT; |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 687 | /* stopservice(&svd[1]); */ |
| 688 | update_status(&svd[1]); |
Denis Vlasenko | 3718832 | 2008-02-16 13:20:56 +0000 | [diff] [blame] | 689 | close(logpipe.wr); |
| 690 | close(logpipe.rd); |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 691 | } |
| 692 | } |
Denis Vlasenko | 3718832 | 2008-02-16 13:20:56 +0000 | [diff] [blame] | 693 | } /* for (;;) */ |
Denis Vlasenko | 04c6386 | 2006-11-17 18:58:49 +0000 | [diff] [blame] | 694 | /* not reached */ |
| 695 | return 0; |
| 696 | } |