Try to make indent formatting less horrible
[oweals/busybox.git] / libbb / run_parts.c
index bf906a399ac2d65163187becae7c8b8a64e25f2a..0eb766086bc77a54154a810c43214d6a5c15837c 100644 (file)
@@ -1,53 +1,20 @@
 /* vi: set sw=4 ts=4: */
 /*
- * Mini run-parts implementation for busybox
+ * run command from specified directory
  *
  *
  * Copyright (C) 2001 by Emanuele Aina <emanuele.aina@tiscali.it>
- *
- * Based on the Debian run-parts program, version 1.15
- *   Copyright (C) 1996 Jeff Noxon <jeff@router.patch.net>,
- *   Copyright (C) 1996-1999 Guy Maor <maor@debian.org>
- *   
+ * rewrite to vfork usage by
+ * Copyright (C) 2002 by Vladimir Oleynik <dzo@simtreas.ru>
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
  * the Free Software Foundation; either version 2 of the License, or
  * (at your option) any later version.
  *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
- * 02111-1307 USA
  *
  */
 
-/* This is my first attempt to write a program in C (well, this is my first
- * attempt to write a program! :-) . */
-
-/* This piece of code is heavily based on the original version of run-parts,
- * taken from debian-utils. I've only removed the long options and a the 
- * report mode. As the original run-parts support only long options, I've
- * broken compatibility because the BusyBox policy doesn't allow them. 
- * The supported options are: 
- * -t                  test. Print the name of the files to be executed, without
- *                             execute them.
- * -a ARG              argument. Pass ARG as an argument the program executed. It can 
- *                             be repeated to pass multiple arguments.
- * -u MASK             umask. Set the umask of the program executed to MASK. */
-
-/* TODO 
- * done - convert calls to error in perror... and remove error()
- * done - convert malloc/realloc to their x... counterparts 
- * done - remove catch_sigchld
- * done - use bb's concat_path_file() 
- * done - declare run_parts_main() as extern and any other function as static?
- */
 
 #include <sys/types.h>
 #include <sys/wait.h>
@@ -55,6 +22,7 @@
 #include <dirent.h>
 #include <unistd.h>
 #include <ctype.h>
+#include <errno.h>
 
 #include "libbb.h"
 
@@ -75,58 +43,77 @@ static int valid_name(const struct dirent *d)
        return 1;
 }
 
-/* run_parts */
-/* Find the parts to run & call run_part() */
-extern int run_parts(char **args, const unsigned char test_mode)
+/* test mode = 1 is the same as offical run_parts
+ * test_mode = 2 means to fail siliently on missing directories
+ */
+
+extern int run_parts(char **args, const unsigned char test_mode, char **env)
 {
        struct dirent **namelist = 0;
        struct stat st;
        char *filename;
+       char *arg0 = args[0];
        int entries;
        int i;
        int exitstatus = 0;
 
+#if __GNUC__
+       /* Avoid longjmp clobbering */
+       (void) &i;
+       (void) &exitstatus;
+#endif
        /* scandir() isn't POSIX, but it makes things easy. */
-       entries = scandir(args[0], &namelist, valid_name, alphasort);
+       entries = scandir(arg0, &namelist, valid_name, alphasort);
 
        if (entries == -1) {
-               perror_msg_and_die("failed to open directory %s", args[0]);
+               if (test_mode & 2) {
+                       return(2);
+               }
+               bb_perror_msg_and_die("failed to open directory %s", arg0);
        }
 
        for (i = 0; i < entries; i++) {
 
-               filename = concat_path_file(args[0], namelist[i]->d_name);
+               filename = concat_path_file(arg0, namelist[i]->d_name);
 
                if (stat(filename, &st) < 0) {
-                       perror_msg_and_die("failed to stat component %s", filename);
+                       bb_perror_msg_and_die("failed to stat component %s", filename);
                }
                if (S_ISREG(st.st_mode) && !access(filename, X_OK)) {
-                       if (test_mode)
-                               printf("run-parts would run %s\n", filename);
-                       else {
+                       if (test_mode) {
+                               puts(filename);
+                       } else {
+                               /* exec_errno is common vfork variable */
+                               volatile int exec_errno = 0;
                                int result;
                                int pid;
 
-                               if ((pid = fork()) < 0) {
-                                       perror_msg_and_die("failed to fork");
+                               if ((pid = vfork()) < 0) {
+                                       bb_perror_msg_and_die("failed to fork");
                                } else if (!pid) {
-                                       execv(args[0], args);
-                                       perror_msg_and_die("failed to exec %s", args[0]);
+                                       args[0] = filename;
+                                       execve(filename, args, env);
+                                       exec_errno = errno;
+                                       _exit(1);
                                }
 
                                waitpid(pid, &result, 0);
-
+                               if(exec_errno) {
+                                       errno = exec_errno;
+                                       bb_perror_msg("failed to exec %s", filename);
+                                       exitstatus = 1;
+                               }
                                if (WIFEXITED(result) && WEXITSTATUS(result)) {
-                                       perror_msg("%s exited with return code %d", args[0], WEXITSTATUS(result));
+                                       bb_perror_msg("%s exited with return code %d", filename, WEXITSTATUS(result));
                                        exitstatus = 1;
                                } else if (WIFSIGNALED(result)) {
-                                       perror_msg("%s exited because of uncaught signal %d", args[0], WTERMSIG(result));
+                                       bb_perror_msg("%s exited because of uncaught signal %d", filename, WTERMSIG(result));
                                        exitstatus = 1;
                                }
                        }
                } 
                else if (!S_ISDIR(st.st_mode)) {
-                       error_msg("component %s is not an executable plain file", filename);
+                       bb_error_msg("component %s is not an executable plain file", filename);
                        exitstatus = 1;
                }