| 1 | /* vi: set sw=4 ts=4: */
|
|---|
| 2 | /*
|
|---|
| 3 | * Mini run-parts implementation for busybox
|
|---|
| 4 | *
|
|---|
| 5 | * Copyright (C) 2007 Bernhard Fischer
|
|---|
| 6 | *
|
|---|
| 7 | * Based on a older version that was in busybox which was 1k big..
|
|---|
| 8 | * Copyright (C) 2001 by Emanuele Aina <emanuele.aina@tiscali.it>
|
|---|
| 9 | *
|
|---|
| 10 | * Based on the Debian run-parts program, version 1.15
|
|---|
| 11 | * Copyright (C) 1996 Jeff Noxon <jeff@router.patch.net>,
|
|---|
| 12 | * Copyright (C) 1996-1999 Guy Maor <maor@debian.org>
|
|---|
| 13 | *
|
|---|
| 14 | *
|
|---|
| 15 | * Licensed under GPL v2 or later, see file LICENSE in this tarball for details.
|
|---|
| 16 | */
|
|---|
| 17 |
|
|---|
| 18 | /* This is my first attempt to write a program in C (well, this is my first
|
|---|
| 19 | * attempt to write a program! :-) . */
|
|---|
| 20 |
|
|---|
| 21 | /* This piece of code is heavily based on the original version of run-parts,
|
|---|
| 22 | * taken from debian-utils. I've only removed the long options and a the
|
|---|
| 23 | * report mode. As the original run-parts support only long options, I've
|
|---|
| 24 | * broken compatibility because the BusyBox policy doesn't allow them.
|
|---|
| 25 | * The supported options are:
|
|---|
| 26 | * -t test. Print the name of the files to be executed, without
|
|---|
| 27 | * execute them.
|
|---|
| 28 | * -a ARG argument. Pass ARG as an argument the program executed. It can
|
|---|
| 29 | * be repeated to pass multiple arguments.
|
|---|
| 30 | * -u MASK umask. Set the umask of the program executed to MASK.
|
|---|
| 31 | */
|
|---|
| 32 |
|
|---|
| 33 | #include <getopt.h>
|
|---|
| 34 |
|
|---|
| 35 | #include "libbb.h"
|
|---|
| 36 |
|
|---|
| 37 | #if ENABLE_FEATURE_RUN_PARTS_LONG_OPTIONS
|
|---|
| 38 | static const char runparts_longopts[] ALIGN1 =
|
|---|
| 39 | "arg\0" Required_argument "a"
|
|---|
| 40 | "umask\0" Required_argument "u"
|
|---|
| 41 | "test\0" No_argument "t"
|
|---|
| 42 | #if ENABLE_FEATURE_RUN_PARTS_FANCY
|
|---|
| 43 | "list\0" No_argument "l"
|
|---|
| 44 | //TODO: "reverse\0" No_argument "r"
|
|---|
| 45 | //TODO: "verbose\0" No_argument "v"
|
|---|
| 46 | #endif
|
|---|
| 47 | ;
|
|---|
| 48 | #endif
|
|---|
| 49 |
|
|---|
| 50 | struct globals {
|
|---|
| 51 | smalluint mode;
|
|---|
| 52 | char *cmd[10]; /* merely arbitrary arg count */
|
|---|
| 53 | };
|
|---|
| 54 | #define G (*(struct globals*)&bb_common_bufsiz1)
|
|---|
| 55 |
|
|---|
| 56 | /* valid_name */
|
|---|
| 57 | /* True or false? Is this a valid filename (upper/lower alpha, digits,
|
|---|
| 58 | * underscores, and hyphens only?)
|
|---|
| 59 | */
|
|---|
| 60 | static bool invalid_name(const char *c)
|
|---|
| 61 | {
|
|---|
| 62 | c = bb_basename(c);
|
|---|
| 63 |
|
|---|
| 64 | while (*c && (isalnum(*c) || *c == '_' || *c == '-'))
|
|---|
| 65 | c++;
|
|---|
| 66 |
|
|---|
| 67 | return *c; /* TRUE (!0) if terminating NUL is not reached */
|
|---|
| 68 | }
|
|---|
| 69 |
|
|---|
| 70 | #define RUN_PARTS_OPT_a (1<<0)
|
|---|
| 71 | #define RUN_PARTS_OPT_u (1<<1)
|
|---|
| 72 | #define RUN_PARTS_OPT_t (1<<2)
|
|---|
| 73 | #if ENABLE_FEATURE_RUN_PARTS_FANCY
|
|---|
| 74 | #define RUN_PARTS_OPT_l (1<<3)
|
|---|
| 75 | #endif
|
|---|
| 76 |
|
|---|
| 77 | #define test_mode (G.mode & RUN_PARTS_OPT_t)
|
|---|
| 78 | #if ENABLE_FEATURE_RUN_PARTS_FANCY
|
|---|
| 79 | #define list_mode (G.mode & RUN_PARTS_OPT_l)
|
|---|
| 80 | #else
|
|---|
| 81 | #define list_mode (0)
|
|---|
| 82 | #endif
|
|---|
| 83 |
|
|---|
| 84 | static int act(const char *file, struct stat *statbuf, void *args, int depth)
|
|---|
| 85 | {
|
|---|
| 86 | int ret;
|
|---|
| 87 |
|
|---|
| 88 | if (depth == 1)
|
|---|
| 89 | return TRUE;
|
|---|
| 90 |
|
|---|
| 91 | if (depth == 2 &&
|
|---|
| 92 | ((!list_mode && access(file, X_OK)) ||
|
|---|
| 93 | invalid_name(file) ||
|
|---|
| 94 | !(statbuf->st_mode & (S_IFREG | S_IFLNK))) )
|
|---|
| 95 | return SKIP;
|
|---|
| 96 |
|
|---|
| 97 | if (test_mode || list_mode) {
|
|---|
| 98 | puts(file);
|
|---|
| 99 | return TRUE;
|
|---|
| 100 | }
|
|---|
| 101 | G.cmd[0] = (char*)file;
|
|---|
| 102 | ret = wait4pid(spawn(G.cmd));
|
|---|
| 103 | if (ret < 0) {
|
|---|
| 104 | bb_perror_msg("failed to exec %s", file);
|
|---|
| 105 | } else if (ret > 0) {
|
|---|
| 106 | bb_error_msg("%s exited with return code %d", file, ret);
|
|---|
| 107 | }
|
|---|
| 108 | return !ret;
|
|---|
| 109 | }
|
|---|
| 110 |
|
|---|
| 111 | int run_parts_main(int argc, char **argv);
|
|---|
| 112 | int run_parts_main(int argc, char **argv)
|
|---|
| 113 | {
|
|---|
| 114 | char *umask_p;
|
|---|
| 115 | llist_t *arg_list = NULL;
|
|---|
| 116 | unsigned tmp;
|
|---|
| 117 |
|
|---|
| 118 | umask(022);
|
|---|
| 119 | /* We require exactly one argument: the directory name */
|
|---|
| 120 | opt_complementary = "=1:a::";
|
|---|
| 121 | #if ENABLE_FEATURE_RUN_PARTS_LONG_OPTIONS
|
|---|
| 122 | applet_long_options = runparts_longopts;
|
|---|
| 123 | #endif
|
|---|
| 124 | tmp = getopt32(argv, "a:u:t"USE_FEATURE_RUN_PARTS_FANCY("l"), &arg_list, &umask_p);
|
|---|
| 125 | G.mode = tmp &~ (RUN_PARTS_OPT_a|RUN_PARTS_OPT_u);
|
|---|
| 126 | if (tmp & RUN_PARTS_OPT_u) {
|
|---|
| 127 | /* Check and set the umask of the program executed.
|
|---|
| 128 | * As stated in the original run-parts, the octal conversion in
|
|---|
| 129 | * libc is not foolproof; it will take the 8 and 9 digits under
|
|---|
| 130 | * some circumstances. We'll just have to live with it.
|
|---|
| 131 | */
|
|---|
| 132 | umask(xstrtoul_range(umask_p, 8, 0, 07777));
|
|---|
| 133 | }
|
|---|
| 134 | for (tmp = 1; arg_list; arg_list = arg_list->link, tmp++)
|
|---|
| 135 | G.cmd[tmp] = arg_list->data;
|
|---|
| 136 | /* G.cmd[tmp] = NULL; - G is already zeroed out */
|
|---|
| 137 | if (!recursive_action(argv[argc - 1],
|
|---|
| 138 | ACTION_RECURSE|ACTION_FOLLOWLINKS,
|
|---|
| 139 | act, /* file action */
|
|---|
| 140 | act, /* dir action */
|
|---|
| 141 | NULL, /* user data */
|
|---|
| 142 | 1 /* depth */
|
|---|
| 143 | ))
|
|---|
| 144 | return EXIT_FAILURE;
|
|---|
| 145 | return EXIT_SUCCESS;
|
|---|
| 146 | }
|
|---|
