diff options
author | Guillaume Rousse <guillomovitch@mandriva.org> | 2007-09-11 12:39:51 +0000 |
---|---|---|
committer | Guillaume Rousse <guillomovitch@mandriva.org> | 2007-09-11 12:39:51 +0000 |
commit | 25e03a3bb2c7bfbc1f3c177ee71c7f75683ebd9d (patch) | |
tree | 9afa5ad3fecc7aca66222da768ba0245ae479d65 /utils/run-parts.c | |
download | setup-25e03a3bb2c7bfbc1f3c177ee71c7f75683ebd9d.tar setup-25e03a3bb2c7bfbc1f3c177ee71c7f75683ebd9d.tar.gz setup-25e03a3bb2c7bfbc1f3c177ee71c7f75683ebd9d.tar.bz2 setup-25e03a3bb2c7bfbc1f3c177ee71c7f75683ebd9d.tar.xz setup-25e03a3bb2c7bfbc1f3c177ee71c7f75683ebd9d.zip |
move everything under trunk directory
Diffstat (limited to 'utils/run-parts.c')
-rw-r--r-- | utils/run-parts.c | 425 |
1 files changed, 425 insertions, 0 deletions
diff --git a/utils/run-parts.c b/utils/run-parts.c new file mode 100644 index 0000000..5f29a9f --- /dev/null +++ b/utils/run-parts.c @@ -0,0 +1,425 @@ +/* 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> + * + * 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> +#include <getopt.h> +#include <string.h> +#include <errno.h> +#include <ctype.h> +#include <signal.h> +#include <sys/time.h> + +#define VERSION "1.15" + +int test_mode = 0; +int verbose_mode = 0; +int report_mode = 0; +int exitstatus = 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"); +} + + +static void version (void) +{ + fprintf (stderr, "Mandriva Linux run-parts program, version " VERSION "\n" + "Copyright (C) 1994 Ian Jackson, Copyright (C) 1996 Jeff Noxon.\n" + "Copyright (C) 1996,1997,1998,1999 Guy Maor\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); +} + + +static void usage (void) +{ + fprintf (stderr, "Usage: run-parts [OPTION]... DIRECTORY\n" + " --test print script names which would run, but don't run them.\n" + " --verbose print script names before running them.\n" + " --report print script names if they produce output.\n" + " --umask=UMASK sets umask to UMASK (octal), default is 022.\n" + " --arg=ARGUMENT pass ARGUMENT to scripts, use once for each argument.\n" + " --version output version information and exit.\n" + " --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. + */ +static void set_umask (void) +{ + 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 (upper/lower alpha, digits, + * underscores, and hyphens only?) + */ +int valid_name (const struct dirent *d) +{ + char *c = d->d_name; + int filenamelen; + + filenamelen=strlen(c); + if( (strncmp(".rpmsave",&(c[filenamelen-8]),9)==0) + || (strncmp(".rpmorig",&(c[filenamelen-8]),9)==0) + || (strncmp(".rpmnew",&(c[filenamelen-7]),8)==0) ) + return 0; + + while (*c) { + if (!isalnum(*c) && *c!='_' && *c!='-' && *c!='.') + return 0; + ++c; + } + return 1; +} + + +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. + */ +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) { + 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); + /* Try with sh */ + if (verbose_mode) fprintf(stderr, "run-parts: executing %s with /bin/sh\n", progname); + execl("/bin/sh", "/bin/sh", "-c", progname, args, (char *)0); + error ("failed to exec %s: %s", progname, strerror (errno)); + exit (1); + } + + if (report_mode) { + fd_set set; + int max, r, w, printflag, errno_select; + ssize_t c; + char buf[4096]; + struct timeval zerotime, *timeout; + struct sigaction sa, sold; + + close(pout[1]); + close(perr[1]); + 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; + zerotime.tv_usec = 0; + zerotime.tv_sec = 0; + timeout = NULL; + sigaction(SIGCHLD, NULL, &sa); + sa.sa_handler = catch_sigchld; + + for (;;) { + FD_ZERO(&set); + if (pout[0] >= 0) FD_SET(pout[0], &set); + if (perr[0] >= 0) FD_SET(perr[0], &set); + + /* select() call may be interrupted by SIGCHLD */ + sigaction(SIGCHLD, &sa, &sold); + r = select(max, &set, 0, 0, timeout); + errno_select = errno; + sigaction(SIGCHLD, &sold, NULL); + + 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 if (r < 0) { + /* select() was interrupted so check if we got a SIGCHLD + * probably from our child (check this later). + */ + if (errno_select != EINTR) { + error("select: %s", strerror(errno_select)); + exit(1); + } + } + else if (timeout == NULL) { /* should never happen */ + error("internal error: select() returned 0 !"); + exit(1); + } + if (timeout == NULL) { + w = waitpid(pid, &result, WNOHANG); + if (w > 0) { + /* Our child has died so prepare to flush the pipes */ + timeout = &zerotime; + } + } + else { + if (pout[0]) close(pout[0]); + if (perr[0]) close(perr[0]); + break; + } + } + } + else { + 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); + } + + for (i = 0; i < entries; 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 = stat (filename, &st); + if (result < 0) { + error ("failed to stat component %s: %s", filename, + strerror (errno)); + exit (1); + } + if (S_ISREG(st.st_mode) && !access (filename, X_OK)) { + if (test_mode) + printf ("run-parts would run %s\n", filename); + else { + if (verbose_mode) + fprintf(stderr, "run-parts: executing %s\n", filename); + run_part (filename); + } + } + else if (!S_ISDIR(st.st_mode)) { + printf ("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}, + {"verbose", 0, &verbose_mode, 1}, + {"report", 0, &report_mode, 1}, + {"umask", 1, 0, 'u'}, + {"arg", 1, 0, 'a'}, + {"help", 0, 0, 'h'}, + {"version", 0, 0, 'v'}, + {0, 0, 0, 0} + }; + + opterr = 0; + c = getopt_long (argc, argv, "", 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': + version (); + break; + default: + error("unrecognized option `%s'", argv[optind-1]); + 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); + } + + run_parts (argv[optind]); + + return exitstatus; +} |