Cleanup patch from Tito.
[oweals/busybox.git] / debianutils / run_parts.c
index 3ec4b9d10ee9cc172ae8cb86011712e93d711e08..6205595bf892c417eb5d0dec6b261069581fe627 100644 (file)
@@ -8,7 +8,7 @@
  * 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>
- *   
+ *
  *
  * 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
  * 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 
+ * 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: 
+ * 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 
+ * -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 
+/* TODO
  * done - convert calls to error in perror... and remove error()
- * done - convert malloc/realloc to their x... counterparts 
+ * done - convert malloc/realloc to their x... counterparts
  * done - remove catch_sigchld
- * done - use bb's concat_path_file() 
+ * done - use bb's concat_path_file()
  * done - declare run_parts_main() as extern and any other function as static?
  */
 
-#include <stdio.h>
-#include <stdarg.h>
-#include <stdlib.h>
-/* #include <sys/types.h> */
-#include <sys/wait.h>
-#include <dirent.h>
-#include <sys/stat.h>
-#include <unistd.h>
 #include <getopt.h>
-#include <string.h>
-#include <errno.h>
-#include <ctype.h>
-
-#include "busybox.h"
-
-static int test_mode = 0;
-static int exitstatus = 0;
-
-static int argcount = 0, argsize = 0;
-static char **args = 0;
-
-
-/* set_umask */
-/* Check and set the umask of the program executed. As stated in the original
- * run-parts, the octal conversion in libc is not foolproof; it will take the 
- * 8 and 9 digits under some circumstances. We'll just have to live with it.
- */
-
-static void set_umask (void)
-{
-  int mask, result;
-
-  /*TODO
-   * We must substitute sscanf, according to bb's style guide? */
-  result = sscanf (optarg, "%o", &mask);
-  if ((result != 1) || (mask > 07777) || (mask < 0)) {
-         perror_msg_and_die ("bad umask value");
-  }
-
-  umask (mask);
-}
-
-/* add_argument */
-/* Add an argument to the commands that we will call. Called once for
-   every argument. */
-static void add_argument (char *newarg)
-{
-  if (argcount+1 >= argsize) {
-    argsize = argsize ? argsize*2 : 4;
-       /*TODO if we convert to xrealloc we lose the verbose error message */
-       args = realloc(args, argsize * (sizeof(char*)));
-    if (!args) {
-               perror_msg_and_die ("failed to reallocate memory for arguments");
-       }
-  }
-  args[argcount++] = newarg;
-  args[argcount] = 0;
-}
-
-/* valid_name */
-/* True or false? Is this a valid filename (upper/lower alpha, digits,
- * underscores, and hyphens only?)
- */
-
-static int valid_name (const struct dirent *d)
-{
-       char *c = d->d_name;
-       while (*c) {
-               if (!isalnum(*c) && *c!='_' && *c!='-') {
-                       return 0;
-               }
-               ++c;
-       }
-       return 1;
-}
-
-
-/* run_part */
-/* Execute a file */
-
-static void run_part (char *progname)
-{
-  int result;
-  int pid;
-
-  
-  if ((pid=fork()) < 0) {
-    perror_msg_and_die ("failed to fork");
-  }
-  else if (!pid) {
-    args[0] = progname;
-    execv (progname, args);
-    perror_msg_and_die ("failed to exec %s", progname);
-  }
-
-  if (0) {
-                 
-  } else {
-                 
-                 waitpid(pid, &result, 0);
-  }
-
-  if (WIFEXITED (result) && WEXITSTATUS(result)) {
-                 perror_msg ("%s exited with return code %d", progname, WEXITSTATUS(result));
-                 exitstatus = 1;
-  }
-  else if (WIFSIGNALED (result)) {
-                 perror_msg ("%s exited because of uncaught signal %d", progname,
-                                                 WTERMSIG(result));
-                 exitstatus = 1;
-  }
-}
-
-/* run_parts */
-/* Find the parts to run & call run_part() */
-
-static void run_parts (char *dir_name)
-{
-  struct dirent **namelist = 0;
-  char *filename;
-  int entries, i;
-  struct stat st;
-
-  /* -- */
-  
-  /* scandir() isn't POSIX, but it makes things easy. */
-  entries = scandir (dir_name, &namelist, valid_name, alphasort);
-
-  if (entries < 0) {
-    perror_msg_and_die ("failed to open directory %s", dir_name);
-  }
-  
-  for (i = 0; i < entries; i++) {
+#include <stdlib.h>
 
-         /* -- */
+#include "libbb.h"
 
-         filename = concat_path_file (dir_name, namelist[i]->d_name);
-         
-         if (stat (filename, &st) < 0) {
-                 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 {
-                         run_part (filename);
-                 }
-         }
+static const struct option runparts_long_options[] = {
+       { "test",               0,              NULL,           't' },
+       { "umask",              1,              NULL,           'u' },
+       { "arg",                1,              NULL,           'a' },
+       { 0,                    0,              0,                      0 }
+};
 
-         else if (!S_ISDIR(st.st_mode)) {
-                 error_msg ("component %s is not an executable plain file",
-                                 filename);
-                 exitstatus = 1;
-         }
-         
-         free (namelist[i]);
-         free (filename);
-  }
-  free (namelist);
-}
+extern char **environ;
 
 /* run_parts_main */
 /* Process options */
-int run_parts_main (int argc, char *argv[])
+int run_parts_main(int argc, char **argv)
 {
-  umask (022);
-  add_argument(0);
-  
-  for (;;) {
-    int c;
+       char **args = xmalloc(2 * sizeof(char *));
+       unsigned char test_mode = 0;
+       unsigned short argcount = 1;
+       int opt;
+
+       umask(022);
+
+       while ((opt = getopt_long (argc, argv, "tu:a:",
+                                       runparts_long_options, NULL)) > 0)
+       {
+               switch (opt) {
+                       /* Enable test mode */
+                       case 't':
+                               test_mode++;
+                               break;
+                       /* Set the umask of the programs executed */
+                       case 'u':
+                               /* Check and set the umask of the program executed. As stated in the original
+                                * run-parts, the octal conversion in libc is not foolproof; it will take the
+                                * 8 and 9 digits under some circumstances. We'll just have to live with it.
+                                */
+                               umask(bb_xgetlarg(optarg, 8, 0, 07777));
+                               break;
+                       /* Pass an argument to the programs */
+                       case 'a':
+                               /* Add an argument to the commands that we will call.
+                                * Called once for every argument. */
+                               args = xrealloc(args, (argcount + 2) * (sizeof(char *)));
+                               args[argcount++] = optarg;
+                               break;
+                       default:
+                               bb_show_usage();
+               }
+       }
 
-    opterr = 0;
-    c = getopt(argc, argv, "tu:a:");
-       
-    if (c == EOF)
-      break;
-    switch (c) {
-       case 't':               /* Enable test mode */
-               test_mode = 1;
-               break;                  
-       case 'u':               /* Set the umask of the programs executed */
-               set_umask ();
-               break;
-    case 'a':               /* Pass an argument to the programs */
-               add_argument (optarg);
-               break;
-    default:
-               show_usage();
-    }
-  }
-  
-  /* We require exactly one argument: the directory name */
-  if (optind != (argc - 1)) {
-         show_usage();
-  }
+       /* We require exactly one argument: the directory name */
+       if (optind != (argc - 1)) {
+               bb_show_usage();
+       }
 
-  run_parts (argv[optind]);
+       args[0] = argv[optind];
+       args[argcount] = 0;
 
-  return exitstatus;
+       return(run_parts(args, test_mode, environ));
 }