busybox/init/run_parts.c

/* vi: set sw=4 ts=4: */
/*
 * Mini run-parts implementation for busybox
 *
 *
 * 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>
 *   
 *
 * 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 <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++) {

	  /* -- */

	  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);
		  }
	  }

	  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);
}

/* run_parts_main */
/* Process options */
int run_parts_main (int argc, char *argv[])
{
  umask (022);
  add_argument(0);
  
  for (;;) {
    int c;

    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();
  }

  run_parts (argv[optind]);

  return exitstatus;
}