lib/runcmd.c

   1 /*
   2  *      UCW Library -- Running of Commands
   3  *
   4  *      (c) 2004 Martin Mares <mj@ucw.cz>
   5  *
   6  *      This software may be freely distributed and used according to the terms
   7  *      of the GNU Lesser General Public License.
   8  */
   9
  10 #include "lib/lib.h"
  11
  12 #include <stdlib.h>
  13 #include <string.h>
  14 #include <alloca.h>
  15 #include <unistd.h>
  16 #include <sys/wait.h>
  17
  18 void NONRET
  19 exec_command_v(byte *cmd, va_list args)
  20 {
  21   va_list cargs;
  22   va_copy(cargs, args);
  23   int cnt = 2;
  24   byte *arg;
  25   while (arg = va_arg(cargs, byte *))
  26     cnt++;
  27   char **argv = alloca(sizeof(byte *) * cnt);
  28   argv[0] = cmd;
  29   cnt = 1;
  30   va_copy(cargs, args);
  31   while (arg = va_arg(cargs, byte *))
  32     argv[cnt++] = arg;
  33   argv[cnt] = NULL;
  34   execv(cmd, argv);
  35   byte echo[256];
  36   echo_command_v(echo, sizeof(echo), cmd, args);
  37   log(L_ERROR, "Cannot execute %s: %m", echo);
  38   exit(255);
  39 }
  40
  41 int
  42 run_command_v(byte *cmd, va_list args)
  43 {
  44   pid_t p = fork();
  45   if (p < 0)
  46     {
  47       log(L_ERROR, "fork() failed: %m");
  48       return 0;
  49     }
  50   else if (!p)
  51     exec_command_v(cmd, args);
  52   else
  53     {
  54       int stat;
  55       byte msg[EXIT_STATUS_MSG_SIZE];
  56       p = waitpid(p, &stat, 0);
  57       if (p < 0)
  58         die("waitpid() failed: %m");
  59       if (format_exit_status(msg, stat))
  60         {
  61           byte echo[256];
  62           echo_command_v(echo, sizeof(echo), cmd, args);
  63           log(L_ERROR, "`%s' failed: %s", echo, msg);
  64           return 0;
  65         }
  66       return 1;
  67     }
  68 }
  69
  70 void
  71 echo_command_v(byte *buf, int size, byte *cmd, va_list args)
  72 {
  73   byte *limit = buf + size - 4;
  74   byte *p = buf;
  75   byte *arg = cmd;
  76   va_list cargs;
  77   va_copy(cargs, args);
  78   do
  79     {
  80       int l = strlen(arg);
  81       if (p != buf && p < limit)
  82         *p++ = ' ';
  83       if (p+l > limit)
  84         {
  85           memcpy(p, arg, limit-p);
  86           strcpy(limit, "...");
  87           return;
  88         }
  89       memcpy(p, arg, l);
  90       p += l;
  91     }
  92   while (arg = va_arg(cargs, byte *));
  93   *p = 0;
  94 }
  95
  96 int
  97 run_command(byte *cmd, ...)
  98 {
  99   va_list args;
 100   va_start(args, cmd);
 101   int e = run_command_v(cmd, args);
 102   va_end(args);
 103   return e;
 104 }
 105
 106 void NONRET
 107 exec_command(byte *cmd, ...)
 108 {
 109   va_list args;
 110   va_start(args, cmd);
 111   exec_command_v(cmd, args);
 112 }
 113
 114 void
 115 echo_command(byte *buf, int len, byte *cmd, ...)
 116 {
 117   va_list args;
 118   va_start(args, cmd);
 119   echo_command_v(buf, len, cmd, args);
 120   va_end(args);
 121 }
 122
 123 #ifdef TEST
 124
 125 int main(void)
 126 {
 127   byte msg[1024];
 128   echo_command(msg, sizeof(msg), "/bin/echo", "datel", "strakapoud", NULL);
 129   log(L_INFO, "Running <%s>", msg);
 130   run_command("/bin/echo", "datel", "strakapoud", NULL);
 131   return 0;
 132 }
 133
 134 #endif