1 /* vi: set sw=4 ts=4: */
2 /*
3 * Mini run-parts implementation for busybox
4 *
5 * Copyright (C) 2007 Bernhard Reutner-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 * Licensed under GPLv2 or later, see file LICENSE in this source tree.
15 */
16
17 /* This is my first attempt to write a program in C (well, this is my first
18 * attempt to write a program! :-) . */
19
20 /* This piece of code is heavily based on the original version of run-parts,
21 * taken from debian-utils. I've only removed the long options and the
22 * report mode. As the original run-parts support only long options, I've
23 * broken compatibility because the BusyBox policy doesn't allow them.
24 */
25 //config:config RUN_PARTS
26 //config: bool "run-parts (6.1 kb)"
27 //config: default y
28 //config: help
29 //config: run-parts is a utility designed to run all the scripts in a directory.
30 //config:
31 //config: It is useful to set up a directory like cron.daily, where you need to
32 //config: execute all the scripts in that directory.
33 //config:
34 //config: In this implementation of run-parts some features (such as report
35 //config: mode) are not implemented.
36 //config:
37 //config: Unless you know that run-parts is used in some of your scripts
38 //config: you can safely say N here.
39 //config:
40 //config:config FEATURE_RUN_PARTS_LONG_OPTIONS
41 //config: bool "Enable long options"
42 //config: default y
43 //config: depends on RUN_PARTS && LONG_OPTS
44 //config:
45 //config:config FEATURE_RUN_PARTS_FANCY
46 //config: bool "Support additional arguments"
47 //config: default y
48 //config: depends on RUN_PARTS
49 //config: help
50 //config: Support additional options:
51 //config: -l --list print the names of the all matching files (not
52 //config: limited to executables), but don't actually run them.
53
54 //applet:IF_RUN_PARTS(APPLET_ODDNAME(run-parts, run_parts, BB_DIR_BIN, BB_SUID_DROP, run_parts))
55
56 //kbuild:lib-$(CONFIG_RUN_PARTS) += run_parts.o
57
58 //usage:#define run_parts_trivial_usage
59 //usage: "[-a ARG]... [-u UMASK] "
60 //usage: IF_FEATURE_RUN_PARTS_LONG_OPTIONS("[--reverse] [--test] [--exit-on-error] "IF_FEATURE_RUN_PARTS_FANCY("[--list] "))
61 //usage: "DIRECTORY"
62 //usage:#define run_parts_full_usage "\n\n"
63 //usage: "Run a bunch of scripts in DIRECTORY\n"
64 //usage: "\n -a ARG Pass ARG as argument to scripts"
65 //usage: "\n -u UMASK Set UMASK before running scripts"
66 //usage: IF_FEATURE_RUN_PARTS_LONG_OPTIONS(
67 //usage: "\n --reverse Reverse execution order"
68 //usage: "\n --test Dry run"
69 //usage: "\n --exit-on-error Exit if a script exits with non-zero"
70 //usage: IF_FEATURE_RUN_PARTS_FANCY(
71 //usage: "\n --list Print names of matching files even if they are not executable"
72 //usage: )
73 //usage: )
74 //usage:
75 //usage:#define run_parts_example_usage
76 //usage: "$ run-parts -a start /etc/init.d\n"
77 //usage: "$ run-parts -a stop=now /etc/init.d\n\n"
78 //usage: "Let's assume you have a script foo/dosomething:\n"
79 //usage: "#!/bin/sh\n"
80 //usage: "for i in $*; do eval $i; done; unset i\n"
81 //usage: "case \"$1\" in\n"
82 //usage: "start*) echo starting something;;\n"
83 //usage: "stop*) set -x; shutdown -h $stop;;\n"
84 //usage: "esac\n\n"
85 //usage: "Running this yields:\n"
86 //usage: "$run-parts -a stop=+4m foo/\n"
87 //usage: "+ shutdown -h +4m"
88
89 #include "libbb.h"
90 #include "common_bufsiz.h"
91
92 struct globals {
93 char **names;
94 int cur;
95 char *cmd[2 /* using 1 provokes compiler warning */];
96 } FIX_ALIASING;
97 #define G (*(struct globals*)bb_common_bufsiz1)
98 #define names (G.names)
99 #define cur (G.cur )
100 #define cmd (G.cmd )
101 #define INIT_G() do { setup_common_bufsiz(); } while (0)
102
103 enum { NUM_CMD = (COMMON_BUFSIZE - sizeof(G)) / sizeof(cmd[0]) - 1 };
104
105 enum {
106 OPT_a = (1 << 0),
107 OPT_u = (1 << 1),
108 OPT_r = (1 << 2) * ENABLE_FEATURE_RUN_PARTS_LONG_OPTIONS,
109 OPT_t = (1 << 3) * ENABLE_FEATURE_RUN_PARTS_LONG_OPTIONS,
110 OPT_e = (1 << 4) * ENABLE_FEATURE_RUN_PARTS_LONG_OPTIONS,
111 OPT_l = (1 << 5) * ENABLE_FEATURE_RUN_PARTS_LONG_OPTIONS
112 * ENABLE_FEATURE_RUN_PARTS_FANCY,
113 };
114
115 /* Is this a valid filename (upper/lower alpha, digits,
116 * underscores, hyphens, and non-leading dots only?)
117 */
invalid_name(const char * c)118 static bool invalid_name(const char *c)
119 {
120 c = bb_basename(c);
121
122 if (*c == '.')
123 return *c;
124
125 /* Debian run-parts 4.8.3, manpage:
126 * "...the names must consist entirely of ASCII letters,
127 * ASCII digits, ASCII underscores, and ASCII minus-hyphens.
128 * However, the name must not begin with a period."
129 * The last sentence is a giveaway that something is fishy
130 * (why mention leading dot if dots are not allowed anyway?).
131 * Yes, you guessed it right: in fact non-leading dots ARE allowed.
132 */
133 while (isalnum(*c) || *c == '_' || *c == '-' || *c == '.')
134 c++;
135
136 return *c; /* TRUE (!0) if terminating NUL is not reached */
137 }
138
bb_alphasort(const void * p1,const void * p2)139 static int bb_alphasort(const void *p1, const void *p2)
140 {
141 int r = strcmp(*(char **) p1, *(char **) p2);
142 return (option_mask32 & OPT_r) ? -r : r;
143 }
144
act(struct recursive_state * state,const char * file,struct stat * statbuf)145 static int FAST_FUNC act(struct recursive_state *state,
146 const char *file, struct stat *statbuf)
147 {
148 if (state->depth == 0)
149 return TRUE;
150
151 if (state->depth == 1
152 && ( !(statbuf->st_mode & (S_IFREG | S_IFLNK))
153 || invalid_name(file)
154 || (!(option_mask32 & OPT_l) && access(file, X_OK) != 0))
155 ) {
156 return SKIP;
157 }
158
159 names = xrealloc_vector(names, 4, cur);
160 names[cur++] = xstrdup(file);
161 /*names[cur] = NULL; - xrealloc_vector did it */
162
163 return TRUE;
164 }
165
166 #if ENABLE_FEATURE_RUN_PARTS_LONG_OPTIONS
167 static const char runparts_longopts[] ALIGN1 =
168 "arg\0" Required_argument "a"
169 "umask\0" Required_argument "u"
170 //TODO: "verbose\0" No_argument "v"
171 "reverse\0" No_argument "\xf0"
172 "test\0" No_argument "\xf1"
173 "exit-on-error\0" No_argument "\xf2"
174 # if ENABLE_FEATURE_RUN_PARTS_FANCY
175 "list\0" No_argument "\xf3"
176 # endif
177 ;
178 # define GETOPT32 getopt32long
179 # define LONGOPTS ,runparts_longopts
180 #else
181 # define GETOPT32 getopt32
182 # define LONGOPTS
183 #endif
184
185 int run_parts_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
run_parts_main(int argc UNUSED_PARAM,char ** argv)186 int run_parts_main(int argc UNUSED_PARAM, char **argv)
187 {
188 const char *umask_p = "22";
189 llist_t *arg_list = NULL;
190 unsigned n;
191 int ret;
192
193 INIT_G();
194
195 /* We require exactly one argument: the directory name */
196 GETOPT32(argv, "^" "a:*u:" "\0" "=1" LONGOPTS,
197 &arg_list, &umask_p
198 );
199
200 umask(xstrtou_range(umask_p, 8, 0, 07777));
201
202 n = 1;
203 while (arg_list && n < NUM_CMD) {
204 cmd[n++] = llist_pop(&arg_list);
205 }
206 /* cmd[n] = NULL; - is already zeroed out */
207
208 /* run-parts has to sort executables by name before running them */
209
210 recursive_action(argv[optind],
211 ACTION_RECURSE|ACTION_FOLLOWLINKS,
212 act, /* file action */
213 act, /* dir action */
214 NULL /* user data */
215 );
216
217 if (!names)
218 return 0;
219
220 qsort(names, cur, sizeof(char *), bb_alphasort);
221
222 n = 0;
223 while (1) {
224 char *name = *names++;
225 if (!name)
226 break;
227 if (option_mask32 & (OPT_t | OPT_l)) {
228 puts(name);
229 continue;
230 }
231 cmd[0] = name;
232 ret = spawn_and_wait(cmd);
233 if (ret == 0)
234 continue;
235 n = 1;
236 if (ret < 0)
237 bb_perror_msg("can't execute '%s'", name);
238 else /* ret > 0 */
239 bb_error_msg("%s: exit status %u", name, ret & 0xff);
240
241 if (option_mask32 & OPT_e)
242 xfunc_die();
243 }
244
245 return n;
246 }
247