#include "libbb.h"
+static const struct option runparts_long_options[] = {
+ { "test", 0, NULL, 't' },
+ { "umask", 1, NULL, 'u' },
+ { "arg", 1, NULL, 'a' },
+ { 0, 0, 0, 0 }
+};
+
/* run_parts_main */
/* Process options */
int run_parts_main(int argc, char **argv)
umask(022);
- while ((opt = getopt(argc, argv, "tu:a:")) != -1) {
+ while ((opt = getopt_long (argc, argv, "tu:a:",
+ runparts_long_options, NULL)) > 0)
+ {
switch (opt) {
- case 't': /* Enable test mode */
- test_mode = 1;
- break;
- case 'u': /* Set the umask of the programs executed */
- /* Check and set the umask of the program executed. As stated in the original
- * run-parts, 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.
- */
- {
- const unsigned int mask = (unsigned int) strtol(optarg, NULL, 8);
- if (mask > 07777) {
- bb_perror_msg_and_die("bad umask value");
- }
- umask(mask);
- }
- break;
- case 'a': /* Pass an argument to the programs */
- /* Add an argument to the commands that we will call.
- * Called once for every argument. */
- args = xrealloc(args, (argcount + 2) * (sizeof(char *)));
- args[argcount++] = optarg;
- break;
- default:
- bb_show_usage();
+ /* Enable test mode */
+ case 't':
+ test_mode++;
+ break;
+ /* Set the umask of the programs executed */
+ case 'u':
+ /* Check and set the umask of the program executed. As stated in the original
+ * run-parts, 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.
+ */
+ umask(bb_xgetlarg(optarg, 8, 0, 07777));
+ break;
+ /* Pass an argument to the programs */
+ case 'a':
+ /* Add an argument to the commands that we will call.
+ * Called once for every argument. */
+ args = xrealloc(args, (argcount + 2) * (sizeof(char *)));
+ args[argcount++] = optarg;
+ break;
+ default:
+ bb_show_usage();
}
}
+++ /dev/null
-/* vi: set sw=4 ts=4: */
-/*
- * Mini start-stop-daemon implementation(s) for busybox
- *
- * Written by Marek Michalkiewicz <marekm@i17linuxb.ists.pwr.wroc.pl>,
- * public domain.
- * Adapted for busybox David Kimdon <dwhedon@gordian.com>
- */
-
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <stdarg.h>
-#include <signal.h>
-#include <errno.h>
-#include <sys/stat.h>
-#include <dirent.h>
-#include <unistd.h>
-#include <getopt.h>
-
-#include "busybox.h"
-#include "pwd_.h"
-
-static int start = 0;
-static int stop = 0;
-static int fork_before_exec = 0;
-static int signal_nr = 15;
-static int user_id = -1;
-static char *userspec = NULL;
-static char *cmdname = NULL;
-static char *execname = NULL;
-static char *startas = NULL;
-
-typedef struct pid_list {
- struct pid_list *next;
- int pid;
-} pid_list;
-
-static pid_list *found = NULL;
-
-static inline void
-push(int pid)
-{
- pid_list *p;
-
- p = xmalloc(sizeof(*p));
- p->next = found;
- p->pid = pid;
- found = p;
-}
-
-static int
-pid_is_exec(int pid, const char *exec)
-{
- char buf[PATH_MAX];
- FILE *fp;
-
- sprintf(buf, "/proc/%d/cmdline", pid);
- fp = fopen(buf, "r");
- if (fp && fgets (buf, sizeof (buf), fp) ) {
- fclose(fp);
- if (strncmp (buf, exec, strlen(exec)) == 0)
- return 1;
- }
- return 0;
-}
-
-
-static int
-pid_is_user(int pid, int uid)
-{
- struct stat sb;
- char buf[32];
-
- sprintf(buf, "/proc/%d", pid);
- if (stat(buf, &sb) != 0)
- return 0;
- return (sb.st_uid == uid);
-}
-
-
-static int
-pid_is_cmd(int pid, const char *name)
-{
- char buf[32];
- FILE *f;
- int c;
-
- sprintf(buf, "/proc/%d/stat", pid);
- f = fopen(buf, "r");
- if (!f)
- return 0;
- while ((c = getc(f)) != EOF && c != '(')
- ;
- if (c != '(') {
- fclose(f);
- return 0;
- }
- /* this hopefully handles command names containing ')' */
- while ((c = getc(f)) != EOF && c == *name)
- name++;
- fclose(f);
- return (c == ')' && *name == '\0');
-}
-
-
-static void
-check(int pid)
-{
- if (execname && !pid_is_exec(pid, execname)) {
- return;
- }
- if (userspec && !pid_is_user(pid, user_id)) {
- return;
- }
- if (cmdname && !pid_is_cmd(pid, cmdname)) {
- return;
- }
- push(pid);
-}
-
-
-
-static void
-do_procfs(void)
-{
- DIR *procdir;
- struct dirent *entry;
- int foundany, pid;
-
- procdir = opendir("/proc");
- if (!procdir)
- bb_perror_msg_and_die ("opendir /proc");
-
- foundany = 0;
- while ((entry = readdir(procdir)) != NULL) {
- if (sscanf(entry->d_name, "%d", &pid) != 1)
- continue;
- foundany++;
- check(pid);
- }
- closedir(procdir);
- if (!foundany)
- bb_error_msg_and_die ("nothing in /proc - not mounted?");
-}
-
-
-static void
-do_stop(void)
-{
- char what[1024];
- pid_list *p;
- int killed = 0;
-
- if (cmdname)
- strcpy(what, cmdname);
- else if (execname)
- strcpy(what, execname);
- else if (userspec)
- sprintf(what, "process(es) owned by `%s'", userspec);
- else
- bb_error_msg_and_die ("internal error, please report");
-
- if (!found) {
- printf("no %s found; none killed.\n", what);
- return;
- }
- for (p = found; p; p = p->next) {
- if (kill(p->pid, signal_nr) == 0) {
- p->pid = -p->pid;
- killed++;
- } else {
- bb_perror_msg("warning: failed to kill %d:", p->pid);
- }
- }
- if (killed) {
- printf("stopped %s (pid", what);
- for (p = found; p; p = p->next)
- if(p->pid < 0)
- printf(" %d", -p->pid);
- printf(").\n");
- }
-}
-
-
-static const struct option ssd_long_options[] = {
- { "stop", 0, NULL, 'K' },
- { "start", 0, NULL, 'S' },
- { "background", 0, NULL, 'b' },
- { "startas", 1, NULL, 'a' },
- { "name", 1, NULL, 'n' },
- { "signal", 1, NULL, 's' },
- { "user", 1, NULL, 'u' },
- { "exec", 1, NULL, 'x' },
- { 0, 0, 0, 0 }
-};
-
-int
-start_stop_daemon_main(int argc, char **argv)
-{
- int flags;
- char *signame;
- bb_applet_long_options = ssd_long_options;
-
- flags = bb_getopt_ulflags(argc, argv, "KSba:n:s:u:x:",
- &startas, &cmdname, &signame, &userspec, &execname);
-
- /* Be sneaky and avoid branching */
- stop = (flags & 1);
- start = (flags & 2);
- fork_before_exec = (flags & 4);
-
- signal_nr = bb_xgetlarg(signame, 10, 0, NSIG);
-
- if (start == stop)
- bb_error_msg_and_die ("need exactly one of -S or -K");
-
- if (!execname && !userspec)
- bb_error_msg_and_die ("need at least one of -x or -u");
-
- if (!startas)
- startas = execname;
-
- if (start && !startas)
- bb_error_msg_and_die ("-S needs -x or -a");
-
- argc -= optind;
- argv += optind;
-
- if (userspec && sscanf(userspec, "%d", &user_id) != 1)
- user_id = my_getpwnam(userspec);
-
- do_procfs();
-
- if (stop) {
- do_stop();
- return EXIT_SUCCESS;
- }
-
- if (found) {
- printf("%s already running.\n%d\n", execname ,found->pid);
- return EXIT_SUCCESS;
- }
- *--argv = startas;
- if (fork_before_exec) {
- if (daemon(0, 0) == -1)
- bb_perror_msg_and_die ("unable to fork");
- }
- setsid();
- execv(startas, argv);
- bb_perror_msg_and_die ("unable to start %s", startas);
-}
-
projects / oweals / busybox.git / commitdiff