blob: 3ef28f1d4ae78c101ea4bf1317f24ab209def807 [file] [log] [blame]
Eric Andersen7d682902001-10-31 10:59:29 +00001/* vi: set sw=4 ts=4: */
2/*
3 * Mini run-parts implementation for busybox
4 *
Bernhard Reutner-Fischer6c4dade2008-09-25 12:13:34 +00005 * Copyright (C) 2007 Bernhard Reutner-Fischer
Eric Andersen7d682902001-10-31 10:59:29 +00006 *
Denys Vlasenko1d7ad7a2012-06-21 09:45:11 +02007 * Based on a older version that was in busybox which was 1k big.
Bernhard Reutner-Fischerb7cffd42007-03-28 20:35:13 +00008 * Copyright (C) 2001 by Emanuele Aina <emanuele.aina@tiscali.it>
Eric Andersen7d682902001-10-31 10:59:29 +00009 *
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>
Eric Andersenc7bda1c2004-03-15 08:29:22 +000013 *
Denys Vlasenko0ef64bd2010-08-16 20:14:46 +020014 * Licensed under GPLv2 or later, see file LICENSE in this source tree.
Eric Andersen7d682902001-10-31 10:59:29 +000015 */
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,
Denys Vlasenko1d7ad7a2012-06-21 09:45:11 +020021 * taken from debian-utils. I've only removed the long options and the
Eric Andersen7d682902001-10-31 10:59:29 +000022 * report mode. As the original run-parts support only long options, I've
Eric Andersenc7bda1c2004-03-15 08:29:22 +000023 * broken compatibility because the BusyBox policy doesn't allow them.
Eric Andersenfb74a452001-12-18 14:06:03 +000024 */
Denys Vlasenko28826ac2015-10-19 00:52:26 +020025//config:config RUN_PARTS
26//config: bool "run-parts"
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
Denys Vlasenko28826ac2015-10-19 00:52:26 +020044//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
Eric Andersen7d682902001-10-31 10:59:29 +000057
Pere Orga6a3e01d2011-04-01 22:56:30 +020058//usage:#define run_parts_trivial_usage
Denys Vlasenko1e43a382013-02-28 10:22:49 +010059//usage: "[-a ARG]... [-u UMASK] "
Peter Korsgaard5d1c5992013-02-28 12:25:49 +010060//usage: IF_FEATURE_RUN_PARTS_LONG_OPTIONS("[--reverse] [--test] [--exit-on-error] "IF_FEATURE_RUN_PARTS_FANCY("[--list] "))
Denys Vlasenko1e43a382013-02-28 10:22:49 +010061//usage: "DIRECTORY"
Pere Orga6a3e01d2011-04-01 22:56:30 +020062//usage:#define run_parts_full_usage "\n\n"
63//usage: "Run a bunch of scripts in DIRECTORY\n"
Denys Vlasenko1e43a382013-02-28 10:22:49 +010064//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(
Peter Korsgaard5d1c5992013-02-28 12:25:49 +010067//usage: "\n --reverse Reverse execution order"
Denys Vlasenko1e43a382013-02-28 10:22:49 +010068//usage: "\n --test Dry run"
Peter Korsgaard5d1c5992013-02-28 12:25:49 +010069//usage: "\n --exit-on-error Exit if a script exits with non-zero"
Pere Orga6a3e01d2011-04-01 22:56:30 +020070//usage: IF_FEATURE_RUN_PARTS_FANCY(
Denys Vlasenko1e43a382013-02-28 10:22:49 +010071//usage: "\n --list Print names of matching files even if they are not executable"
Pere Orga6a3e01d2011-04-01 22:56:30 +020072//usage: )
Denys Vlasenko1e43a382013-02-28 10:22:49 +010073//usage: )
Pere Orga6a3e01d2011-04-01 22:56:30 +020074//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
Denis Vlasenkob6adbf12007-05-26 19:00:18 +000089#include "libbb.h"
Denys Vlasenkoe6a2f4c2016-04-21 16:26:30 +020090#include "common_bufsiz.h"
Denis Vlasenkob6adbf12007-05-26 19:00:18 +000091
Denis Vlasenko2a1d0242007-09-23 18:34:49 +000092struct globals {
93 char **names;
94 int cur;
Denys Vlasenko5117eff2013-10-16 14:21:20 +020095 char *cmd[2 /* using 1 provokes compiler warning */];
Denys Vlasenko98a4c7c2010-02-04 15:00:15 +010096} FIX_ALIASING;
Denys Vlasenkoe6a2f4c2016-04-21 16:26:30 +020097#define G (*(struct globals*)bb_common_bufsiz1)
Denis Vlasenko2a1d0242007-09-23 18:34:49 +000098#define names (G.names)
99#define cur (G.cur )
100#define cmd (G.cmd )
Denys Vlasenko47cfbf32016-04-21 18:18:48 +0200101#define INIT_G() do { setup_common_bufsiz(); } while (0)
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000102
Denis Vlasenkod50dda82008-06-15 05:40:56 +0000103enum { NUM_CMD = (COMMON_BUFSIZE - sizeof(G)) / sizeof(cmd[0]) - 1 };
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000104
105enum {
Denys Vlasenko1e43a382013-02-28 10:22:49 +0100106 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,
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000113};
114
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000115/* Is this a valid filename (upper/lower alpha, digits,
116 * underscores, and hyphens only?)
117 */
118static bool invalid_name(const char *c)
119{
120 c = bb_basename(c);
121
122 while (*c && (isalnum(*c) || *c == '_' || *c == '-'))
123 c++;
124
125 return *c; /* TRUE (!0) if terminating NUL is not reached */
126}
127
128static int bb_alphasort(const void *p1, const void *p2)
129{
Denis Vlasenko19fb67e2008-02-28 21:30:22 +0000130 int r = strcmp(*(char **) p1, *(char **) p2);
131 return (option_mask32 & OPT_r) ? -r : r;
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000132}
133
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000134static int FAST_FUNC act(const char *file, struct stat *statbuf, void *args UNUSED_PARAM, int depth)
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000135{
136 if (depth == 1)
137 return TRUE;
138
139 if (depth == 2
140 && ( !(statbuf->st_mode & (S_IFREG | S_IFLNK))
141 || invalid_name(file)
Denys Vlasenko1e43a382013-02-28 10:22:49 +0100142 || (!(option_mask32 & OPT_l) && access(file, X_OK) != 0))
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000143 ) {
144 return SKIP;
145 }
146
Denis Vlasenkodeeed592008-07-08 05:14:36 +0000147 names = xrealloc_vector(names, 4, cur);
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000148 names[cur++] = xstrdup(file);
Denis Vlasenko27842282008-08-04 13:20:36 +0000149 /*names[cur] = NULL; - xrealloc_vector did it */
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000150
151 return TRUE;
152}
153
Bernhard Reutner-Fischerb7cffd42007-03-28 20:35:13 +0000154#if ENABLE_FEATURE_RUN_PARTS_LONG_OPTIONS
Denis Vlasenko6ca409e2007-08-12 20:58:27 +0000155static const char runparts_longopts[] ALIGN1 =
Denis Vlasenkobdc88fd2007-07-23 17:14:14 +0000156 "arg\0" Required_argument "a"
157 "umask\0" Required_argument "u"
Denis Vlasenkobdc88fd2007-07-23 17:14:14 +0000158//TODO: "verbose\0" No_argument "v"
Denys Vlasenko1e43a382013-02-28 10:22:49 +0100159 "reverse\0" No_argument "\xf0"
160 "test\0" No_argument "\xf1"
161 "exit-on-error\0" No_argument "\xf2"
162#if ENABLE_FEATURE_RUN_PARTS_FANCY
163 "list\0" No_argument "\xf3"
Bernhard Reutner-Fischerb7cffd42007-03-28 20:35:13 +0000164#endif
Denis Vlasenko990d0f62007-07-24 15:54:42 +0000165 ;
Bernhard Reutner-Fischerb7cffd42007-03-28 20:35:13 +0000166#endif
167
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +0000168int run_parts_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000169int run_parts_main(int argc UNUSED_PARAM, char **argv)
Eric Andersen7d682902001-10-31 10:59:29 +0000170{
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000171 const char *umask_p = "22";
Bernhard Reutner-Fischerb7cffd42007-03-28 20:35:13 +0000172 llist_t *arg_list = NULL;
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000173 unsigned n;
174 int ret;
Eric Andersen7d682902001-10-31 10:59:29 +0000175
Ian Wienand378ab682011-09-14 08:41:38 +0200176 INIT_G();
177
Bernhard Reutner-Fischerb7cffd42007-03-28 20:35:13 +0000178#if ENABLE_FEATURE_RUN_PARTS_LONG_OPTIONS
Denis Vlasenkobdc88fd2007-07-23 17:14:14 +0000179 applet_long_options = runparts_longopts;
Bernhard Reutner-Fischerb7cffd42007-03-28 20:35:13 +0000180#endif
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000181 /* We require exactly one argument: the directory name */
Denys Vlasenko237bedd2016-07-06 21:58:02 +0200182 opt_complementary = "=1";
183 getopt32(argv, "a:*u:", &arg_list, &umask_p);
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000184
185 umask(xstrtou_range(umask_p, 8, 0, 07777));
186
187 n = 1;
188 while (arg_list && n < NUM_CMD) {
Denis Vlasenkod50dda82008-06-15 05:40:56 +0000189 cmd[n++] = llist_pop(&arg_list);
Glenn L McGrath545106f2002-11-11 06:21:00 +0000190 }
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000191 /* cmd[n] = NULL; - is already zeroed out */
192
193 /* run-parts has to sort executables by name before running them */
194
195 recursive_action(argv[optind],
Denis Vlasenkobbd695d2007-04-08 10:52:28 +0000196 ACTION_RECURSE|ACTION_FOLLOWLINKS,
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000197 act, /* file action */
198 act, /* dir action */
199 NULL, /* user data */
200 1 /* depth */
201 );
202
203 if (!names)
204 return 0;
205
206 qsort(names, cur, sizeof(char *), bb_alphasort);
207
208 n = 0;
209 while (1) {
210 char *name = *names++;
211 if (!name)
212 break;
Denis Vlasenko19fb67e2008-02-28 21:30:22 +0000213 if (option_mask32 & (OPT_t | OPT_l)) {
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000214 puts(name);
215 continue;
216 }
217 cmd[0] = name;
Denys Vlasenko8531d762010-03-18 22:44:00 +0100218 ret = spawn_and_wait(cmd);
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000219 if (ret == 0)
220 continue;
221 n = 1;
222 if (ret < 0)
Denis Vlasenkof9d4fc32009-04-21 20:40:51 +0000223 bb_perror_msg("can't execute '%s'", name);
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000224 else /* ret > 0 */
Denys Vlasenko7f3a2a22015-10-08 11:24:44 +0200225 bb_error_msg("%s: exit status %u", name, ret & 0xff);
Peter Korsgaard0496e822013-02-28 09:59:23 +0100226
227 if (option_mask32 & OPT_e)
228 xfunc_die();
Denis Vlasenko2a1d0242007-09-23 18:34:49 +0000229 }
230
231 return n;
Eric Andersen7d682902001-10-31 10:59:29 +0000232}