libbb/run_parts.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * run command from specified directory
   4  *
   5  *
   6  * Copyright (C) 2001 by Emanuele Aina <emanuele.aina@tiscali.it>
   7  * rewrite to vfork usage by
   8  * Copyright (C) 2002 by Vladimir Oleynik <dzo@simtreas.ru>
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  *
  16  */
  17
  18
  19 #include <sys/types.h>
  20 #include <sys/wait.h>
  21 #include <stdlib.h>
  22 #include <dirent.h>
  23 #include <unistd.h>
  24 #include <ctype.h>
  25 #include <errno.h>
  26
  27 #include "libbb.h"
  28
  29 /* valid_name */
  30 /* True or false? Is this a valid filename (upper/lower alpha, digits,
  31  * underscores, and hyphens only?)
  32  */
  33 static int valid_name(const struct dirent *d)
  34 {
  35         char *c = d->d_name;
  36
  37         while (*c) {
  38                 if (!isalnum(*c) && (*c != '_') && (*c != '-')) {
  39                         return 0;
  40                 }
  41                 ++c;
  42         }
  43         return 1;
  44 }
  45
  46 /* run_parts */
  47 /* Find the parts to run & call run_part() */
  48 extern int run_parts(char **args, const unsigned char test_mode)
  49 {
  50         struct dirent **namelist = 0;
  51         struct stat st;
  52         char *filename;
  53         int entries;
  54         int i;
  55         int exitstatus = 0;
  56
  57 #if __GNUC__
  58         /* Avoid longjmp clobbering */
  59         (void) &i;
  60         (void) &exitstatus;
  61 #endif
  62         /* scandir() isn't POSIX, but it makes things easy. */
  63         entries = scandir(args[0], &namelist, valid_name, alphasort);
  64
  65         if (entries == -1) {
  66                 perror_msg_and_die("failed to open directory %s", args[0]);
  67         }
  68
  69         for (i = 0; i < entries; i++) {
  70
  71                 filename = concat_path_file(args[0], namelist[i]->d_name);
  72
  73                 if (stat(filename, &st) < 0) {
  74                         perror_msg_and_die("failed to stat component %s", filename);
  75                 }
  76                 if (S_ISREG(st.st_mode) && !access(filename, X_OK)) {
  77                         if (test_mode)
  78                                 printf("run-parts would run %s\n", filename);
  79                         else {
  80                                 /* exec_errno is common vfork variable */
  81                                 volatile int exec_errno = 0;
  82                                 int result;
  83                                 int pid;
  84
  85                                 if ((pid = vfork()) < 0) {
  86                                         perror_msg_and_die("failed to fork");
  87                                 } else if (!pid) {
  88                                         args[0] = filename;
  89                                         execv(filename, args);
  90                                         exec_errno = errno;
  91                                         _exit(1);
  92                                 }
  93
  94                                 waitpid(pid, &result, 0);
  95                                 if(exec_errno) {
  96                                         errno = exec_errno;
  97                                         perror_msg_and_die("failed to exec %s", filename);
  98                                 }
  99                                 if (WIFEXITED(result) && WEXITSTATUS(result)) {
 100                                         perror_msg("%s exited with return code %d", filename, WEXITSTATUS(result));
 101                                         exitstatus = 1;
 102                                 } else if (WIFSIGNALED(result)) {
 103                                         perror_msg("%s exited because of uncaught signal %d", filename, WTERMSIG(result));
 104                                         exitstatus = 1;
 105                                 }
 106                         }
 107                 }
 108                 else if (!S_ISDIR(st.st_mode)) {
 109                         error_msg("component %s is not an executable plain file", filename);
 110                         exitstatus = 1;
 111                 }
 112
 113                 free(namelist[i]);
 114                 free(filename);
 115         }
 116         free(namelist);
 117
 118         return(exitstatus);
 119 }