summaryrefslogtreecommitdiff
path: root/run-parts.c
diff options
context:
space:
mode:
authorClint Adams <schizo@debian.org>2005-06-21 23:30:25 +0000
committerClint Adams <schizo@debian.org>2005-06-21 23:30:25 +0000
commitc2a1c435ef5e7d590328802ce333de410ce40cb6 (patch)
tree6095e344242e76daf58d6d42afc5ca29673d2a0a /run-parts.c
downloaddebianutils-c2a1c435ef5e7d590328802ce333de410ce40cb6.tar.gz
tag of schizo@debian.org--2004-primary/debianutils--etch--0--version-0
(automatically generated log message) git-archimport-id: schizo@debian.org--etch/debianutils--etch--0--base-0
Diffstat (limited to 'run-parts.c')
-rw-r--r--run-parts.c472
1 files changed, 472 insertions, 0 deletions
diff --git a/run-parts.c b/run-parts.c
new file mode 100644
index 0000000..b599046
--- /dev/null
+++ b/run-parts.c
@@ -0,0 +1,472 @@
+/* run-parts: run a bunch of scripts in a directory
+ *
+ * Debian run-parts program
+ * Copyright (C) 1996 Jeff Noxon <jeff@router.patch.net>,
+ * Copyright (C) 1996-1999 Guy Maor <maor@debian.org>
+ * Copyright (C) 2002, 2003, 2004, 2005 Clint Adams <schizo@debian.org>
+ *
+ * This is free software; see the GNU General Public License version 2
+ * or later for copying conditions. There is NO warranty.
+ *
+ * Based on run-parts.pl version 0.2, Copyright (C) 1994 Ian Jackson.
+ *
+ */
+
+#include <stdio.h>
+#include <stdarg.h>
+#include <stdlib.h>
+#include <sys/types.h>
+#include <sys/wait.h>
+#include <dirent.h>
+#include <sys/stat.h>
+#include <unistd.h>
+#include <fcntl.h>
+#ifdef HAVE_GETOPT_H
+#include <getopt.h>
+#endif /* HAVE_GETOPT_H */
+#include <string.h>
+#include <errno.h>
+#include <ctype.h>
+#include <signal.h>
+#include <sys/time.h>
+#include <regex.h>
+
+int test_mode = 0;
+int list_mode = 0;
+int verbose_mode = 0;
+int report_mode = 0;
+int reverse_mode = 0;
+int exitstatus = 0;
+int lsbsysinit_mode = 0;
+int exit_on_error_mode = 0;
+
+int argcount = 0, argsize = 0;
+char **args = 0;
+
+void error(char *format, ...)
+{
+ va_list ap;
+
+ fprintf(stderr, "run-parts: ");
+
+ va_start(ap, format);
+ vfprintf(stderr, format, ap);
+ va_end(ap);
+
+ fprintf(stderr, "\n");
+}
+
+
+void version()
+{
+ fprintf(stderr, "Debian run-parts program, version " PACKAGE_VERSION
+ "\nCopyright (C) 1994 Ian Jackson, Copyright (C) 1996 Jeff Noxon.\n"
+ "Copyright (C) 1996,1997,1998,1999 Guy Maor\n"
+ "Copyright (C) 2002, 2003, 2004, 2005 Clint Adams\n"
+ "This is free software; see the GNU General Public License version 2\n"
+ "or later for copying conditions. There is NO warranty.\n");
+ exit(0);
+}
+
+
+void usage()
+{
+ fprintf(stderr, "Usage: run-parts [OPTION]... DIRECTORY\n"
+ " --test print script names which would run, but don't run them.\n"
+ " --list print names of all valid files (can not be used with\n"
+ " --test)\n"
+ " -v, --verbose print script names before running them.\n"
+ " --report print script names if they produce output.\n"
+ " --reverse reverse execution order of scripts.\n"
+ " --exit-on-error exit as soon as a script returns with a non-zero exit\n"
+ " code.\n"
+ " --lsbsysinit validate filenames based on LSB sysinit specs.\n"
+ " -u, --umask=UMASK sets umask to UMASK (octal), default is 022.\n"
+ " -a, --arg=ARGUMENT pass ARGUMENT to scripts, use once for each argument.\n"
+ " -V, --version output version information and exit.\n"
+ " -h, --help display this help and exit.\n");
+ exit(0);
+}
+
+
+/* The octal conversion in libc is not foolproof; it will take the 8 and 9
+ * digits under some circumstances. We'll just have to live with it.
+ */
+void set_umask()
+{
+ int mask, result;
+
+ result = sscanf(optarg, "%o", &mask);
+ if ((result != 1) || (mask > 07777) || (mask < 0)) {
+ error("bad umask value");
+ exit(1);
+ }
+
+ umask(mask);
+}
+
+/* Add an argument to the commands that we will call. Called once for
+ every argument. */
+void add_argument(char *newarg)
+{
+ if (argcount + 1 >= argsize) {
+ argsize = argsize ? argsize * 2 : 4;
+ args = realloc(args, argsize * (sizeof(char *)));
+ if (!args) {
+ error("failed to reallocate memory for arguments: %s", strerror(errno));
+ exit(1);
+ }
+ }
+ args[argcount++] = newarg;
+ args[argcount] = 0;
+}
+
+/* True or false? Is this a valid filename? */
+int valid_name(const struct dirent *d)
+{
+ char *c = d->d_name;
+ regex_t hierre, tradre, excsre, classicalre;
+
+ /* The regcomps should be moved to program init */
+
+ if (lsbsysinit_mode) {
+
+ if (regcomp
+ (&hierre, "^_?([a-z0-9_.]+-)+[a-z0-9]+$", REG_EXTENDED | REG_NOSUB)
+ || regcomp(&excsre, "^[a-z0-9-].*dpkg-(old|dist|new|tmp)$",
+ REG_EXTENDED | REG_NOSUB)
+ || regcomp(&tradre, "^[a-z0-9][a-z0-9-]*$", REG_NOSUB)) {
+ error("regex failure");
+ exit(1);
+ }
+
+ if (!regexec(&hierre, c, 0, NULL, 0))
+ return regexec(&excsre, c, 0, NULL, 0);
+
+ return !regexec(&tradre, c, 0, NULL, 0);
+ }
+ else {
+ if (regcomp(&classicalre, "^[a-zA-Z0-9_-]+$", REG_EXTENDED | REG_NOSUB)) {
+ error("regex failure");
+ exit(1);
+ }
+ return !regexec(&classicalre, c, 0, NULL, 0);
+ }
+
+}
+
+
+void set_fl(int fd, int flags)
+{
+ int val;
+ if ((val = fcntl(fd, F_GETFL, 0)) < 0) {
+ error("fcntl F_GETFL: %s", strerror(errno));
+ exit(1);
+ }
+ val |= flags;
+ if (fcntl(fd, F_SETFL, val) < 0) {
+ error("fcntl F_SETFL: %s", strerror(errno));
+ exit(1);
+ }
+}
+
+
+/* We have to abort the select() call in run_part when we receive a
+ * SIGCHLD signal. We can't simply ignore it, so we do nothing.
+ */
+
+/* itz Tue Nov 26 13:23:27 PST 2002 why "We have to abort?"
+ just leave the default and reap child at the end
+ */
+void catch_sigchld(int sig)
+{
+}
+
+
+/* Execute a file */
+void run_part(char *progname)
+{
+ int result;
+ int pid;
+ int pout[2], perr[2];
+
+ if (report_mode && (pipe(pout) || pipe(perr))) {
+ error("pipe: %s", strerror(errno));
+ exit(1);
+ }
+ if ((pid = fork()) < 0) {
+ error("failed to fork: %s", strerror(errno));
+ exit(1);
+ }
+ else if (!pid) {
+ setsid();
+ if (report_mode) {
+ if (dup2(pout[1], STDOUT_FILENO) == -1 ||
+ dup2(perr[1], STDERR_FILENO) == -1) {
+ error("dup2: %s", strerror(errno));
+ exit(1);
+ }
+ close(pout[0]);
+ close(perr[0]);
+ close(pout[1]);
+ close(perr[1]);
+ }
+ args[0] = progname;
+ execv(progname, args);
+ error("failed to exec %s: %s", progname, strerror(errno));
+ exit(1);
+ }
+
+ if (report_mode) {
+ fd_set set;
+ int max, r, printflag;
+ ssize_t c;
+ char buf[4096];
+
+ close(pout[1]);
+ close(perr[1]);
+ /* itz Tue Nov 26 13:26:09 PST 2002 why is this necessary? */
+ /* set_fl(pout[0], O_NONBLOCK); */
+ /* set_fl(perr[0], O_NONBLOCK); */
+ max = pout[0] > perr[0] ? pout[0] + 1 : perr[0] + 1;
+ printflag = 0;
+
+ while (pout[0] >= 0 || perr[0] >= 0) {
+
+ FD_ZERO(&set);
+ if (pout[0] >= 0)
+ FD_SET(pout[0], &set);
+ if (perr[0] >= 0)
+ FD_SET(perr[0], &set);
+ r = select(max, &set, 0, 0, 0);
+ while (r < 0 && errno == EINTR) {
+ FD_ZERO(&set);
+ if (pout[0] >= 0)
+ FD_SET(pout[0], &set);
+ if (perr[0] >= 0)
+ FD_SET(perr[0], &set);
+ r = select(max, &set, 0, 0, 0);
+ } /*while */
+
+ if (r < 0) {
+ /* assert(errno != EINTR) */
+ error("select: %s", strerror(errno));
+ exit(1);
+ }
+ else if (r > 0) {
+ if (pout[0] >= 0 && FD_ISSET(pout[0], &set)) {
+ c = read(pout[0], buf, sizeof(buf));
+ if (c > 0) {
+ if (!printflag) {
+ printf("%s:\n", progname);
+ fflush(stdout);
+ printflag = 1;
+ }
+ write(STDOUT_FILENO, buf, c);
+ }
+ else if (c == 0) {
+ close(pout[0]);
+ pout[0] = -1;
+ }
+ }
+ if (perr[0] >= 0 && FD_ISSET(perr[0], &set)) {
+ c = read(perr[0], buf, sizeof(buf));
+ if (c > 0) {
+ if (!printflag) {
+ fprintf(stderr, "%s:\n", progname);
+ fflush(stderr);
+ printflag = 1;
+ }
+ write(STDERR_FILENO, buf, c);
+ }
+ else if (c == 0) {
+ close(perr[0]);
+ perr[0] = -1;
+ }
+ }
+ }
+ else {
+ /* assert(FALSE): select was called with infinite timeout, so
+ it either returns successfully or is interrupted */
+ } /*if */
+ } /*while */
+ }
+
+ waitpid(pid, &result, 0);
+
+ if (WIFEXITED(result) && WEXITSTATUS(result)) {
+ error("%s exited with return code %d", progname, WEXITSTATUS(result));
+ exitstatus = 1;
+ }
+ else if (WIFSIGNALED(result)) {
+ error("%s exited because of uncaught signal %d", progname,
+ WTERMSIG(result));
+ exitstatus = 1;
+ }
+}
+
+
+/* Find the parts to run & call run_part() */
+void run_parts(char *dirname)
+{
+ struct dirent **namelist;
+ char *filename;
+ size_t filename_length, dirname_length;
+ int entries, i, result;
+ struct stat st;
+
+ /* dirname + "/" */
+ dirname_length = strlen(dirname) + 1;
+ /* dirname + "/" + ".." + "\0" (This will save one realloc.) */
+ filename_length = dirname_length + 2 + 1;
+ if (!(filename = malloc(filename_length))) {
+ error("failed to allocate memory for path: %s", strerror(errno));
+ exit(1);
+ }
+ strcpy(filename, dirname);
+ strcat(filename, "/");
+
+ /* scandir() isn't POSIX, but it makes things easy. */
+ entries = scandir(dirname, &namelist, valid_name, alphasort);
+ if (entries < 0) {
+ error("failed to open directory %s: %s", dirname, strerror(errno));
+ exit(1);
+ }
+
+ i = reverse_mode ? 0 : entries;
+ for (i = reverse_mode ? (entries - 1) : 0;
+ reverse_mode ? (i >= 0) : (i < entries); reverse_mode ? i-- : i++) {
+ if (filename_length < dirname_length + strlen(namelist[i]->d_name) + 1) {
+ filename_length = dirname_length + strlen(namelist[i]->d_name) + 1;
+ if (!(filename = realloc(filename, filename_length))) {
+ error("failed to reallocate memory for path: %s", strerror(errno));
+ exit(1);
+ }
+ }
+ strcpy(filename + dirname_length, namelist[i]->d_name);
+
+ strcpy(filename, dirname);
+ strcat(filename, "/");
+ strcat(filename, namelist[i]->d_name);
+
+ result = lstat(filename, &st);
+ if (result < 0) {
+ if (exit_on_error_mode) {
+ error("failed to lstat component %s: %s", filename, strerror(errno));
+ exit(1);
+ }
+ else
+ return;
+ }
+
+ if (S_ISREG(st.st_mode) || S_ISLNK(st.st_mode)) {
+ if (!access(filename, X_OK)) {
+ if (test_mode) {
+ printf("%s\n", filename);
+ }
+ else if (list_mode) {
+ if (!access(filename, R_OK))
+ printf("%s\n", filename);
+ }
+ else {
+ if (verbose_mode)
+ fprintf(stderr, "run-parts: executing %s\n", filename);
+ run_part(filename);
+ if (exitstatus != 0 && exit_on_error_mode) return;
+ }
+ }
+ else if (!access(filename, R_OK)) {
+ if (list_mode) {
+ printf("%s\n", filename);
+ }
+ }
+ else if (S_ISLNK(st.st_mode)) {
+ if (!list_mode) {
+ error("run-parts: component %s is a broken symbolic link\n",filename);
+ exitstatus = 1;
+ }
+ }
+ }
+ else if (!S_ISDIR(st.st_mode)) {
+ if (!list_mode) {
+ error("run-parts: component %s is not an executable plain file\n",
+ filename);
+ exitstatus = 1;
+ }
+ }
+
+ free(namelist[i]);
+ }
+ free(namelist);
+ free(filename);
+}
+
+/* Process options */
+int main(int argc, char *argv[])
+{
+ umask(022);
+ add_argument(0);
+
+ for (;;) {
+ int c;
+ int option_index = 0;
+
+ static struct option long_options[] = {
+ {"test", 0, &test_mode, 1},
+ {"list", 0, &list_mode, 1},
+ {"verbose", 0, 0, 'v'},
+ {"report", 0, &report_mode, 1},
+ {"reverse", 0, &reverse_mode, 1},
+ {"umask", 1, 0, 'u'},
+ {"arg", 1, 0, 'a'},
+ {"help", 0, 0, 'h'},
+ {"version", 0, 0, 'V'},
+ {"lsbsysinit", 0, &lsbsysinit_mode, 1},
+ {"exit-on-error", 0, &exit_on_error_mode, 1},
+ {0, 0, 0, 0}
+ };
+
+ c = getopt_long(argc, argv, "u:ha:vV", long_options, &option_index);
+ if (c == EOF)
+ break;
+ switch (c) {
+ case 0:
+ break;
+ case 'u':
+ set_umask();
+ break;
+ case 'a':
+ add_argument(optarg);
+ break;
+ case 'h':
+ usage();
+ break;
+ case 'v':
+ verbose_mode = 1;
+ break;
+ case 'V':
+ version();
+ break;
+ default:
+ fprintf(stderr, "Try `run-parts --help' for more information.\n");
+ exit(1);
+ }
+ }
+
+ /* We require exactly one argument: the directory name */
+ if (optind != (argc - 1)) {
+ error("missing operand");
+ fprintf(stderr, "Try `run-parts --help' for more information.\n");
+ exit(1);
+ }
+
+ if (list_mode && test_mode) {
+ error("--list and --test can not be used together");
+ fprintf(stderr, "Try `run-parts --help' for more information.\n");
+ exit(1);
+ }
+
+ run_parts(argv[optind]);
+
+ return exitstatus;
+}