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