175 lines
4.3 KiB
C
175 lines
4.3 KiB
C
/* vi: set sw=4 ts=4: */
|
|
/*
|
|
* Mini run-parts implementation for busybox
|
|
*
|
|
* Copyright (C) 2007 Bernhard Fischer
|
|
*
|
|
* Based on a older version that was in busybox which was 1k big..
|
|
* 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>
|
|
*
|
|
*
|
|
* Licensed under GPL v2 or later, see file LICENSE in this tarball for details.
|
|
*/
|
|
|
|
/* 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.
|
|
*/
|
|
|
|
#include <getopt.h>
|
|
|
|
#include "libbb.h"
|
|
|
|
struct globals {
|
|
char **names;
|
|
int cur;
|
|
char *cmd[1];
|
|
};
|
|
#define G (*(struct globals*)&bb_common_bufsiz1)
|
|
#define names (G.names)
|
|
#define cur (G.cur )
|
|
#define cmd (G.cmd )
|
|
|
|
enum { NUM_CMD = (COMMON_BUFSIZE - sizeof(struct globals)) / sizeof(cmd[0]) };
|
|
|
|
enum {
|
|
RUN_PARTS_OPT_a = (1 << 0),
|
|
RUN_PARTS_OPT_u = (1 << 1),
|
|
RUN_PARTS_OPT_t = (1 << 2),
|
|
RUN_PARTS_OPT_l = (1 << 3) * ENABLE_FEATURE_RUN_PARTS_FANCY,
|
|
};
|
|
|
|
#if ENABLE_FEATURE_RUN_PARTS_FANCY
|
|
#define list_mode (option_mask32 & RUN_PARTS_OPT_l)
|
|
#else
|
|
#define list_mode 0
|
|
#endif
|
|
|
|
/* Is this a valid filename (upper/lower alpha, digits,
|
|
* underscores, and hyphens only?)
|
|
*/
|
|
static bool invalid_name(const char *c)
|
|
{
|
|
c = bb_basename(c);
|
|
|
|
while (*c && (isalnum(*c) || *c == '_' || *c == '-'))
|
|
c++;
|
|
|
|
return *c; /* TRUE (!0) if terminating NUL is not reached */
|
|
}
|
|
|
|
static int bb_alphasort(const void *p1, const void *p2)
|
|
{
|
|
return strcmp(*(char **) p1, *(char **) p2);
|
|
}
|
|
|
|
static int act(const char *file, struct stat *statbuf, void *args, int depth)
|
|
{
|
|
if (depth == 1)
|
|
return TRUE;
|
|
|
|
if (depth == 2
|
|
&& ( !(statbuf->st_mode & (S_IFREG | S_IFLNK))
|
|
|| invalid_name(file)
|
|
|| (!list_mode && access(file, X_OK) != 0))
|
|
) {
|
|
return SKIP;
|
|
}
|
|
|
|
names = xrealloc(names, (cur + 2) * sizeof(names[0]));
|
|
names[cur++] = xstrdup(file);
|
|
names[cur] = NULL;
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
#if ENABLE_FEATURE_RUN_PARTS_LONG_OPTIONS
|
|
static const char runparts_longopts[] ALIGN1 =
|
|
"arg\0" Required_argument "a"
|
|
"umask\0" Required_argument "u"
|
|
"test\0" No_argument "t"
|
|
#if ENABLE_FEATURE_RUN_PARTS_FANCY
|
|
"list\0" No_argument "l"
|
|
//TODO: "reverse\0" No_argument "r"
|
|
//TODO: "verbose\0" No_argument "v"
|
|
#endif
|
|
;
|
|
#endif
|
|
|
|
int run_parts_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
|
|
int run_parts_main(int argc, char **argv)
|
|
{
|
|
const char *umask_p = "22";
|
|
llist_t *arg_list = NULL;
|
|
unsigned n;
|
|
int ret;
|
|
|
|
#if ENABLE_FEATURE_RUN_PARTS_LONG_OPTIONS
|
|
applet_long_options = runparts_longopts;
|
|
#endif
|
|
/* We require exactly one argument: the directory name */
|
|
opt_complementary = "=1:a::";
|
|
getopt32(argv, "a:u:t"USE_FEATURE_RUN_PARTS_FANCY("l"), &arg_list, &umask_p);
|
|
|
|
umask(xstrtou_range(umask_p, 8, 0, 07777));
|
|
|
|
n = 1;
|
|
while (arg_list && n < NUM_CMD) {
|
|
cmd[n] = arg_list->data;
|
|
arg_list = arg_list->link;
|
|
n++;
|
|
}
|
|
/* cmd[n] = NULL; - is already zeroed out */
|
|
|
|
/* run-parts has to sort executables by name before running them */
|
|
|
|
recursive_action(argv[optind],
|
|
ACTION_RECURSE|ACTION_FOLLOWLINKS,
|
|
act, /* file action */
|
|
act, /* dir action */
|
|
NULL, /* user data */
|
|
1 /* depth */
|
|
);
|
|
|
|
if (!names)
|
|
return 0;
|
|
|
|
qsort(names, cur, sizeof(char *), bb_alphasort);
|
|
|
|
n = 0;
|
|
while (1) {
|
|
char *name = *names++;
|
|
if (!name)
|
|
break;
|
|
if (option_mask32 & (RUN_PARTS_OPT_t | RUN_PARTS_OPT_l)) {
|
|
puts(name);
|
|
continue;
|
|
}
|
|
cmd[0] = name;
|
|
ret = wait4pid(spawn(cmd));
|
|
if (ret == 0)
|
|
continue;
|
|
n = 1;
|
|
if (ret < 0)
|
|
bb_perror_msg("failed to exec %s", name);
|
|
else /* ret > 0 */
|
|
bb_error_msg("%s exited with return code %d", name, ret);
|
|
}
|
|
|
|
return n;
|
|
}
|