src/vwait.c

   1 // $Id: vwait.c 2403 2006-11-24 23:06:08Z dhozac $    --*- c -*--
   2
   3 // Copyright (C) 2005 Enrico Scholz <enrico.scholz@informatik.tu-chemnitz.de>
   4 //
   5 // This program is free software; you can redistribute it and/or modify
   6 // it under the terms of the GNU General Public License as published by
   7 // the Free Software Foundation; version 2 of the License.
   8 //
   9 // This program is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU General Public License
  15 // along with this program; if not, write to the Free Software
  16 // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  17
  18
  19 #ifdef HAVE_CONFIG_H
  20 #  include <config.h>
  21 #endif
  22
  23 #include "lib/vserver.h"
  24 #include "lib/internal.h"
  25 #include "util.h"
  26
  27 #include <getopt.h>
  28 #include <signal.h>
  29 #include <time.h>
  30 #include <errno.h>
  31 #include <libgen.h>
  32 #include <assert.h>
  33
  34 #define ENSC_WRAPPERS_PREFIX    "vwait: "
  35 #define ENSC_WRAPPERS_STDLIB    1
  36 #define ENSC_WRAPPERS_UNISTD    1
  37 #define ENSC_WRAPPERS_VSERVER   1
  38 #include <wrappers.h>
  39
  40 #define CMD_HELP        0x8000
  41 #define CMD_VERSION     0x8001
  42
  43 #define CMD_TIMEOUT     0x4000
  44 #define CMD_TERMINATE   0x4001
  45 #define CMD_STATUS_FD   0x4002
  46
  47 static struct option const
  48 CMDLINE_OPTIONS[] = {
  49   { "help",       no_argument,        0, CMD_HELP },
  50   { "version",    no_argument,        0, CMD_VERSION },
  51   { "timeout",    required_argument,  0, CMD_TIMEOUT },
  52   { "terminate",  no_argument,        0, CMD_TERMINATE },
  53   { "status-fd",  required_argument,  0, CMD_STATUS_FD },
  54   { 0,0,0,0 }
  55 };
  56
  57 int                     wrapper_exit_code = 1;
  58 static sig_atomic_t     aborted = 0;
  59
  60 struct StatusType {
  61     enum {stERROR, stFINISHED, stKILLED,
  62           stTIMEOUT}                            status;
  63     int                                         rc;
  64 };
  65
  66 struct Arguments
  67 {
  68     xid_t               xid;
  69     int                 timeout;
  70     int                 status_fd;
  71     bool                do_terminate;
  72 };
  73
  74 static void
  75 showHelp(char const *cmd)
  76 {
  77   VSERVER_DECLARE_CMD(cmd);
  78
  79   WRITE_MSG(1, "Usage:  ");
  80   WRITE_STR(1, cmd);
  81   WRITE_STR(1,
  82             " [--timeout <timeout>] [--terminate] [--status-fd <fd>] [--] <xid>\n"
  83             "\n"
  84             "Please report bugs to " PACKAGE_BUGREPORT "\n");
  85   exit(0);
  86 }
  87
  88 static void
  89 showVersion()
  90 {
  91   WRITE_MSG(1,
  92             "vwait " VERSION " -- waits for a context to finish\n"
  93             "This program is part of " PACKAGE_STRING "\n\n"
  94             "Copyright (C) 2005 Enrico Scholz\n"
  95             VERSION_COPYRIGHT_DISCLAIMER);
  96   exit(0);
  97 }
  98
  99 static void
 100 handler(int UNUSED num)
 101 {
 102   aborted = 1;
 103 }
 104
 105 static struct StatusType
 106 doit(struct Arguments const *args)
 107 {
 108   time_t                        end_time = 0, now = 0;
 109   struct StatusType             res;
 110
 111   if (args->timeout>0) {
 112     end_time = time(0) + args->timeout;
 113     siginterrupt(SIGALRM, 1);
 114     signal(SIGALRM, handler);
 115     alarm(args->timeout);
 116   }
 117
 118   for (;;) {
 119     res.rc = vc_wait_exit(args->xid);
 120
 121     if      (res.rc==-1 && errno!=EAGAIN && errno!=EINTR) {
 122         // the error-case
 123       res.rc     = errno;
 124       res.status = stERROR;
 125       perror(ENSC_WRAPPERS_PREFIX "vc_wait_exit()");
 126     }
 127     else if (res.rc==-1 && args->timeout>0 && (now=time(0))>=end_time) {
 128         // an EINTR or EAGAIN signal was delivered, a timeout was set and
 129         // reached
 130       if (!args->do_terminate)
 131         res.status = stTIMEOUT;
 132       else {
 133         vc_ctx_kill(args->xid, 1, 9);
 134         vc_ctx_kill(args->xid, 0, 9);
 135         res.status = stKILLED;
 136       }
 137     }
 138     else if (res.rc==-1) {
 139         // an EINTR or EAGAIN signal was delivered but the timeout not set or
 140         // not reached yet
 141
 142         // we are here, when args->timeout==0 or 'now' was initialized (and
 143         // compared with 'end_time'). So, 'now' can be used below.
 144       assert(args->timeout<=0 || (now < end_time));
 145
 146       if (args->timeout>0)      // (re)set the alarm-clock
 147         alarm(end_time-now);
 148
 149       continue;
 150     }
 151     else
 152         // vc_wait_exit(2) finished successfully
 153       res.status = stFINISHED;
 154
 155     break;
 156   }
 157
 158   alarm(0);
 159   return res;
 160 }
 161
 162 static void
 163 writeStatus(int fd, char const *str, int const *rc, int exit_code)
 164 {
 165   if (fd==-1) exit(exit_code);
 166
 167   WRITE_STR(fd, str);
 168   if (rc) {
 169     char                buf[sizeof(*rc)*3 + 2];
 170     size_t              len = utilvserver_fmt_long(buf, *rc);
 171     WRITE_MSG(fd, " ");
 172     Vwrite   (fd, buf, len);
 173   }
 174   WRITE_MSG(fd, "\n");
 175
 176   exit(exit_code);
 177 }
 178
 179 int main(int argc, char *argv[])
 180 {
 181   struct StatusType     res;
 182   struct Arguments      args = {
 183     .xid          = VC_NOCTX,
 184     .timeout      = -1,
 185     .status_fd    = -1,
 186     .do_terminate = false,
 187   };
 188
 189   while (1) {
 190     int         c = getopt_long(argc, argv, "c:", CMDLINE_OPTIONS, 0);
 191     if (c==-1) break;
 192
 193     switch (c) {
 194       case CMD_HELP             :  showHelp(argv[0]);
 195       case CMD_VERSION          :  showVersion();
 196       case CMD_TERMINATE        :  args.do_terminate = true;         break;
 197       case CMD_TIMEOUT          :  args.timeout      = atoi(optarg); break;
 198       case CMD_STATUS_FD        :  args.status_fd    = atoi(optarg); break;
 199       default           :
 200         WRITE_MSG(2, "Try '");
 201         WRITE_STR(2, argv[0]);
 202         WRITE_MSG(2, " --help' for more information.\n");
 203         return EXIT_FAILURE;
 204         break;
 205     }
 206   }
 207
 208   if (optind+1 > argc) {
 209     WRITE_MSG(2, ENSC_WRAPPERS_PREFIX "no context specified; try '--help' for more information\n");
 210     exit(wrapper_exit_code);
 211   }
 212
 213   if (optind+1 < argc) {
 214     WRITE_MSG(2, ENSC_WRAPPERS_PREFIX "can not wait for more than one context; try '--help' for more information\n");
 215     exit(wrapper_exit_code);
 216   }
 217
 218   args.xid = Evc_xidopt2xid(argv[optind], true);
 219
 220   if (args.xid==VC_NOCTX) {
 221     WRITE_MSG(2, ENSC_WRAPPERS_PREFIX "invalid context specified; try '--help' for more information\n");
 222     exit(wrapper_exit_code);
 223   }
 224
 225   res = doit(&args);
 226
 227   switch (res.status) {
 228     case stERROR        :  writeStatus(args.status_fd, "ERROR",    &res.rc, 127);
 229     case stFINISHED     :  writeStatus(args.status_fd, "FINISHED", &res.rc,   0);
 230     case stKILLED       :  writeStatus(args.status_fd, "KILLED",         0,   1);
 231     case stTIMEOUT      :  writeStatus(args.status_fd, "TIMEOUT",        0,   2);
 232     default             :  writeStatus(args.status_fd, "???",      &res.rc, 126);
 233   }
 234 }