diff options
Diffstat (limited to 'debianutils/run_parts.c')
-rw-r--r-- | debianutils/run_parts.c | 191 |
1 files changed, 191 insertions, 0 deletions
diff --git a/debianutils/run_parts.c b/debianutils/run_parts.c new file mode 100644 index 000000000..2b3b18854 --- /dev/null +++ b/debianutils/run_parts.c | |||
@@ -0,0 +1,191 @@ | |||
1 | /* vi: set sw=4 ts=4: */ | ||
2 | /* | ||
3 | * Mini run-parts implementation for busybox | ||
4 | * | ||
5 | * | ||
6 | * Copyright (C) 2001 by Emanuele Aina <emanuele.aina@tiscali.it> | ||
7 | * | ||
8 | * Based on the Debian run-parts program, version 1.15 | ||
9 | * Copyright (C) 1996 Jeff Noxon <jeff@router.patch.net>, | ||
10 | * Copyright (C) 1996-1999 Guy Maor <maor@debian.org> | ||
11 | * | ||
12 | * | ||
13 | * Licensed under GPL v2 or later, see file LICENSE in this tarball for details. | ||
14 | */ | ||
15 | |||
16 | /* This is my first attempt to write a program in C (well, this is my first | ||
17 | * attempt to write a program! :-) . */ | ||
18 | |||
19 | /* This piece of code is heavily based on the original version of run-parts, | ||
20 | * taken from debian-utils. I've only removed the long options and a the | ||
21 | * report mode. As the original run-parts support only long options, I've | ||
22 | * broken compatibility because the BusyBox policy doesn't allow them. | ||
23 | * The supported options are: | ||
24 | * -t test. Print the name of the files to be executed, without | ||
25 | * execute them. | ||
26 | * -a ARG argument. Pass ARG as an argument the program executed. It can | ||
27 | * be repeated to pass multiple arguments. | ||
28 | * -u MASK umask. Set the umask of the program executed to MASK. */ | ||
29 | |||
30 | /* TODO | ||
31 | * done - convert calls to error in perror... and remove error() | ||
32 | * done - convert malloc/realloc to their x... counterparts | ||
33 | * done - remove catch_sigchld | ||
34 | * done - use bb's concat_path_file() | ||
35 | * done - declare run_parts_main() as extern and any other function as static? | ||
36 | */ | ||
37 | |||
38 | #include "busybox.h" | ||
39 | #include <getopt.h> | ||
40 | |||
41 | static const struct option runparts_long_options[] = { | ||
42 | { "test", 0, NULL, 't' }, | ||
43 | { "umask", 1, NULL, 'u' }, | ||
44 | { "arg", 1, NULL, 'a' }, | ||
45 | { 0, 0, 0, 0 } | ||
46 | }; | ||
47 | |||
48 | /* valid_name */ | ||
49 | /* True or false? Is this a valid filename (upper/lower alpha, digits, | ||
50 | * underscores, and hyphens only?) | ||
51 | */ | ||
52 | static int valid_name(const struct dirent *d) | ||
53 | { | ||
54 | const char *c = d->d_name; | ||
55 | |||
56 | while (*c) { | ||
57 | if (!isalnum(*c) && (*c != '_') && (*c != '-')) { | ||
58 | return 0; | ||
59 | } | ||
60 | ++c; | ||
61 | } | ||
62 | return 1; | ||
63 | } | ||
64 | |||
65 | /* test mode = 1 is the same as official run_parts | ||
66 | * test_mode = 2 means to fail silently on missing directories | ||
67 | */ | ||
68 | static int run_parts(char **args, const unsigned char test_mode) | ||
69 | { | ||
70 | struct dirent **namelist = 0; | ||
71 | struct stat st; | ||
72 | char *filename; | ||
73 | char *arg0 = args[0]; | ||
74 | int entries; | ||
75 | int i; | ||
76 | int exitstatus = 0; | ||
77 | |||
78 | #if __GNUC__ | ||
79 | /* Avoid longjmp clobbering */ | ||
80 | (void) &i; | ||
81 | (void) &exitstatus; | ||
82 | #endif | ||
83 | /* scandir() isn't POSIX, but it makes things easy. */ | ||
84 | entries = scandir(arg0, &namelist, valid_name, alphasort); | ||
85 | |||
86 | if (entries == -1) { | ||
87 | if (test_mode & 2) { | ||
88 | return 2; | ||
89 | } | ||
90 | bb_perror_msg_and_die("cannot open '%s'", arg0); | ||
91 | } | ||
92 | |||
93 | for (i = 0; i < entries; i++) { | ||
94 | filename = concat_path_file(arg0, namelist[i]->d_name); | ||
95 | |||
96 | xstat(filename, &st); | ||
97 | if (S_ISREG(st.st_mode) && !access(filename, X_OK)) { | ||
98 | if (test_mode) { | ||
99 | puts(filename); | ||
100 | } else { | ||
101 | /* exec_errno is common vfork variable */ | ||
102 | volatile int exec_errno = 0; | ||
103 | int result; | ||
104 | int pid; | ||
105 | |||
106 | if ((pid = vfork()) < 0) { | ||
107 | bb_perror_msg_and_die("failed to fork"); | ||
108 | } else if (!pid) { | ||
109 | args[0] = filename; | ||
110 | execve(filename, args, environ); | ||
111 | exec_errno = errno; | ||
112 | _exit(1); | ||
113 | } | ||
114 | |||
115 | waitpid(pid, &result, 0); | ||
116 | if (exec_errno) { | ||
117 | errno = exec_errno; | ||
118 | bb_perror_msg("failed to exec %s", filename); | ||
119 | exitstatus = 1; | ||
120 | } | ||
121 | if (WIFEXITED(result) && WEXITSTATUS(result)) { | ||
122 | bb_perror_msg("%s exited with return code %d", filename, WEXITSTATUS(result)); | ||
123 | exitstatus = 1; | ||
124 | } else if (WIFSIGNALED(result)) { | ||
125 | bb_perror_msg("%s exited because of uncaught signal %d", filename, WTERMSIG(result)); | ||
126 | exitstatus = 1; | ||
127 | } | ||
128 | } | ||
129 | } else if (!S_ISDIR(st.st_mode)) { | ||
130 | bb_error_msg("component %s is not an executable plain file", filename); | ||
131 | exitstatus = 1; | ||
132 | } | ||
133 | |||
134 | free(namelist[i]); | ||
135 | free(filename); | ||
136 | } | ||
137 | free(namelist); | ||
138 | |||
139 | return exitstatus; | ||
140 | } | ||
141 | |||
142 | |||
143 | /* run_parts_main */ | ||
144 | /* Process options */ | ||
145 | int run_parts_main(int argc, char **argv) | ||
146 | { | ||
147 | char **args = xmalloc(2 * sizeof(char *)); | ||
148 | unsigned char test_mode = 0; | ||
149 | unsigned short argcount = 1; | ||
150 | int opt; | ||
151 | |||
152 | umask(022); | ||
153 | |||
154 | while ((opt = getopt_long(argc, argv, "tu:a:", | ||
155 | runparts_long_options, NULL)) > 0) | ||
156 | { | ||
157 | switch (opt) { | ||
158 | /* Enable test mode */ | ||
159 | case 't': | ||
160 | test_mode++; | ||
161 | break; | ||
162 | /* Set the umask of the programs executed */ | ||
163 | case 'u': | ||
164 | /* Check and set the umask of the program executed. As stated in the original | ||
165 | * run-parts, the octal conversion in libc is not foolproof; it will take the | ||
166 | * 8 and 9 digits under some circumstances. We'll just have to live with it. | ||
167 | */ | ||
168 | umask(xstrtoul_range(optarg, 8, 0, 07777)); | ||
169 | break; | ||
170 | /* Pass an argument to the programs */ | ||
171 | case 'a': | ||
172 | /* Add an argument to the commands that we will call. | ||
173 | * Called once for every argument. */ | ||
174 | args = xrealloc(args, (argcount + 2) * (sizeof(char *))); | ||
175 | args[argcount++] = optarg; | ||
176 | break; | ||
177 | default: | ||
178 | bb_show_usage(); | ||
179 | } | ||
180 | } | ||
181 | |||
182 | /* We require exactly one argument: the directory name */ | ||
183 | if (optind != (argc - 1)) { | ||
184 | bb_show_usage(); | ||
185 | } | ||
186 | |||
187 | args[0] = argv[optind]; | ||
188 | args[argcount] = 0; | ||
189 | |||
190 | return run_parts(args, test_mode); | ||
191 | } | ||