find:: get rid of nested function (it's a gcc-ism)
[oweals/busybox.git] / debianutils / start_stop_daemon.c
index f10572ddf58c27680be95b26cfc003941b9fca3c..7dadc3c9eec0df0ee7c8e99e271de3a970d7fa4a 100644 (file)
@@ -5,7 +5,7 @@
  * Written by Marek Michalkiewicz <marekm@i17linuxb.ists.pwr.wroc.pl>,
  * Adapted for busybox David Kimdon <dwhedon@gordian.com>
  *
- * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
+ * Licensed under GPLv2 or later, see file LICENSE in this source tree.
  */
 
 /*
@@ -31,7 +31,8 @@ Options controlling process matching
 [TODO: can PROCESS_NAME be a full pathname? Should we require full match then
 with /proc/$PID/exe or argv[0] (comm can't be matched, it never contains path)]
         -x,--exec EXECUTABLE    Look for processes that were started with this
-                                command in /proc/$PID/cmdline.
+                                command in /proc/$PID/exe and /proc/$PID/cmdline
+                                (/proc/$PID/cmdline is a bbox extension)
                                 Unlike -n, we match against the full path:
                                 "ntpd" != "./ntpd" != "/path/to/ntpd"
         -p,--pidfile PID_FILE   Look for processes with PID from this file
@@ -56,6 +57,69 @@ Misc options:
         -v,--verbose            Verbose
 */
 
+//usage:#define start_stop_daemon_trivial_usage
+//usage:       "[OPTIONS] [-S|-K] ... [-- ARGS...]"
+//usage:#define start_stop_daemon_full_usage "\n\n"
+//usage:       "Search for matching processes, and then\n"
+//usage:       "-K: stop all matching processes.\n"
+//usage:       "-S: start a process unless a matching process is found.\n"
+//usage:       IF_FEATURE_START_STOP_DAEMON_LONG_OPTIONS(
+//usage:     "\nProcess matching:"
+//usage:     "\n       -u,--user USERNAME|UID  Match only this user's processes"
+//usage:     "\n       -n,--name NAME          Match processes with NAME"
+//usage:     "\n                               in comm field in /proc/PID/stat"
+//usage:     "\n       -x,--exec EXECUTABLE    Match processes with this command"
+//usage:     "\n                               in /proc/PID/{exe,cmdline}"
+//usage:     "\n       -p,--pidfile FILE       Match a process with PID from the file"
+//usage:     "\n       All specified conditions must match"
+//usage:     "\n-S only:"
+//usage:     "\n       -x,--exec EXECUTABLE    Program to run"
+//usage:     "\n       -a,--startas NAME       Zeroth argument"
+//usage:     "\n       -b,--background         Background"
+//usage:       IF_FEATURE_START_STOP_DAEMON_FANCY(
+//usage:     "\n       -N,--nicelevel N        Change nice level"
+//usage:       )
+//usage:     "\n       -c,--chuid USER[:[GRP]] Change to user/group"
+//usage:     "\n       -m,--make-pidfile       Write PID to the pidfile specified by -p"
+//usage:     "\n-K only:"
+//usage:     "\n       -s,--signal SIG         Signal to send"
+//usage:     "\n       -t,--test               Match only, exit with 0 if a process is found"
+//usage:     "\nOther:"
+//usage:       IF_FEATURE_START_STOP_DAEMON_FANCY(
+//usage:     "\n       -o,--oknodo             Exit with status 0 if nothing is done"
+//usage:     "\n       -v,--verbose            Verbose"
+//usage:       )
+//usage:     "\n       -q,--quiet              Quiet"
+//usage:       )
+//usage:       IF_NOT_FEATURE_START_STOP_DAEMON_LONG_OPTIONS(
+//usage:     "\nProcess matching:"
+//usage:     "\n       -u USERNAME|UID Match only this user's processes"
+//usage:     "\n       -n NAME         Match processes with NAME"
+//usage:     "\n                       in comm field in /proc/PID/stat"
+//usage:     "\n       -x EXECUTABLE   Match processes with this command"
+//usage:     "\n                       command in /proc/PID/cmdline"
+//usage:     "\n       -p FILE         Match a process with PID from the file"
+//usage:     "\n       All specified conditions must match"
+//usage:     "\n-S only:"
+//usage:     "\n       -x EXECUTABLE   Program to run"
+//usage:     "\n       -a NAME         Zeroth argument"
+//usage:     "\n       -b              Background"
+//usage:       IF_FEATURE_START_STOP_DAEMON_FANCY(
+//usage:     "\n       -N N            Change nice level"
+//usage:       )
+//usage:     "\n       -c USER[:[GRP]] Change to user/group"
+//usage:     "\n       -m              Write PID to the pidfile specified by -p"
+//usage:     "\n-K only:"
+//usage:     "\n       -s SIG          Signal to send"
+//usage:     "\n       -t              Match only, exit with 0 if a process is found"
+//usage:     "\nOther:"
+//usage:       IF_FEATURE_START_STOP_DAEMON_FANCY(
+//usage:     "\n       -o              Exit with status 0 if nothing is done"
+//usage:     "\n       -v              Verbose"
+//usage:       )
+//usage:     "\n       -q              Quiet"
+//usage:       )
+
 #include <sys/resource.h>
 
 /* Override ENABLE_FEATURE_PIDFILE */
@@ -89,16 +153,17 @@ enum {
 #define TEST  (option_mask32 & OPT_TEST)
 
 struct globals {
-       struct pid_list *found;
+       struct pid_list *found_procs;
        char *userspec;
        char *cmdname;
        char *execname;
        char *pidfile;
+       char *execname_cmpbuf;
+       unsigned execname_sizeof;
        int user_id;
        smallint signal_nr;
-};
+} FIX_ALIASING;
 #define G (*(struct globals*)&bb_common_bufsiz1)
-#define found             (G.found               )
 #define userspec          (G.userspec            )
 #define cmdname           (G.cmdname             )
 #define execname          (G.execname            )
@@ -118,7 +183,7 @@ struct globals {
 static int pid_is_exec(pid_t pid)
 {
        struct stat st;
-       char buf[sizeof("/proc//exe") + sizeof(int)*3];
+       char buf[sizeof("/proc/%u/exe") + sizeof(int)*3];
 
        sprintf(buf, "/proc/%u/exe", (unsigned)pid);
        if (stat(buf, &st) < 0)
@@ -133,13 +198,23 @@ static int pid_is_exec(pid_t pid)
 static int pid_is_exec(pid_t pid)
 {
        ssize_t bytes;
-       char buf[PATH_MAX];
+       char buf[sizeof("/proc/%u/cmdline") + sizeof(int)*3];
+       char *procname, *exelink;
+       int match;
+
+       procname = buf + sprintf(buf, "/proc/%u/exe", (unsigned)pid) - 3;
+
+       exelink = xmalloc_readlink(buf);
+       match = (exelink && strcmp(execname, exelink) == 0);
+       free(exelink);
+       if (match)
+               return match;
 
-       sprintf(buf, "/proc/%u/cmdline", (unsigned)pid);
-       bytes = open_read_close(buf, buf, sizeof(buf) - 1);
+       strcpy(procname, "cmdline");
+       bytes = open_read_close(buf, G.execname_cmpbuf, G.execname_sizeof);
        if (bytes > 0) {
-               buf[bytes] = '\0';
-               return strcmp(buf, execname) == 0;
+               G.execname_cmpbuf[bytes] = '\0';
+               return strcmp(execname, G.execname_cmpbuf) == 0;
        }
        return 0;
 }
@@ -194,9 +269,9 @@ static void check(int pid)
                return;
        }
        p = xmalloc(sizeof(*p));
-       p->next = found;
+       p->next = G.found_procs;
        p->pid = pid;
-       found = p;
+       G.found_procs = p;
 }
 
 static void do_pidfile(void)
@@ -204,7 +279,7 @@ static void do_pidfile(void)
        FILE *f;
        unsigned pid;
 
-       f = fopen(pidfile, "r");
+       f = fopen_for_read(pidfile);
        if (f) {
                if (fscanf(f, "%u", &pid) == 1)
                        check(pid);
@@ -266,23 +341,29 @@ static int do_stop(void)
                bb_error_msg_and_die("internal error, please report");
        }
 
-       if (!found) {
+       if (!G.found_procs) {
                if (!QUIET)
                        printf("no %s found; none killed\n", what);
                killed = -1;
                goto ret;
        }
-       for (p = found; p; p = p->next) {
-               if (TEST || kill(p->pid, signal_nr) == 0) {
+       for (p = G.found_procs; p; p = p->next) {
+               if (kill(p->pid, TEST ? 0 : signal_nr) == 0) {
                        killed++;
                } else {
-                       p->pid = 0;
                        bb_perror_msg("warning: killing process %u", (unsigned)p->pid);
+                       p->pid = 0;
+                       if (TEST) {
+                               /* Example: -K --test --pidfile PIDFILE detected
+                                * that PIDFILE's pid doesn't exist */
+                               killed = -1;
+                               goto ret;
+                       }
                }
        }
        if (!QUIET && killed) {
                printf("stopped %s (pid", what);
-               for (p = found; p; p = p->next)
+               for (p = G.found_procs; p; p = p->next)
                        if (p->pid)
                                printf(" %u", (unsigned)p->pid);
                puts(")");
@@ -320,7 +401,7 @@ static const char start_stop_daemon_longopts[] ALIGN1 =
 #endif
 
 int start_stop_daemon_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
-int start_stop_daemon_main(int argc ATTRIBUTE_UNUSED, char **argv)
+int start_stop_daemon_main(int argc UNUSED_PARAM, char **argv)
 {
        unsigned opt;
        char *signame;
@@ -347,13 +428,13 @@ int start_stop_daemon_main(int argc ATTRIBUTE_UNUSED, char **argv)
        /* -xa (at least one) is required if -S is given */
        /* -q turns off -v */
        opt_complementary = "K:S:K--S:S--K:m?p:K?xpun:S?xa"
-               USE_FEATURE_START_STOP_DAEMON_FANCY("q-v");
+               IF_FEATURE_START_STOP_DAEMON_FANCY("q-v");
        opt = getopt32(argv, "KSbqtma:n:s:u:c:x:p:"
-               USE_FEATURE_START_STOP_DAEMON_FANCY("ovN:"),
-//             USE_FEATURE_START_STOP_DAEMON_FANCY("ovN:R:"),
+               IF_FEATURE_START_STOP_DAEMON_FANCY("ovN:R:"),
                &startas, &cmdname, &signame, &userspec, &chuid, &execname, &pidfile
-               USE_FEATURE_START_STOP_DAEMON_FANCY(,&opt_N)
-//             USE_FEATURE_START_STOP_DAEMON_FANCY(,&retry_arg)
+               IF_FEATURE_START_STOP_DAEMON_FANCY(,&opt_N)
+               /* We accept and ignore -R <param> / --retry <param> */
+               IF_FEATURE_START_STOP_DAEMON_FANCY(,NULL)
        );
 
        if (opt & OPT_s) {
@@ -365,10 +446,14 @@ int start_stop_daemon_main(int argc ATTRIBUTE_UNUSED, char **argv)
                startas = execname;
        if (!execname) /* in case -a is given and -x is not */
                execname = startas;
+       if (execname) {
+               G.execname_sizeof = strlen(execname) + 1;
+               G.execname_cmpbuf = xmalloc(G.execname_sizeof + 1);
+       }
 
-//     USE_FEATURE_START_STOP_DAEMON_FANCY(
+//     IF_FEATURE_START_STOP_DAEMON_FANCY(
 //             if (retry_arg)
-//                     retries = xatoi_u(retry_arg);
+//                     retries = xatoi_positive(retry_arg);
 //     )
        //argc -= optind;
        argv += optind;
@@ -386,9 +471,9 @@ int start_stop_daemon_main(int argc ATTRIBUTE_UNUSED, char **argv)
                return (opt & OPT_OKNODO) ? 0 : (i <= 0);
        }
 
-       if (found) {
+       if (G.found_procs) {
                if (!QUIET)
-                       printf("%s is already running\n%u\n", execname, (unsigned)found->pid);
+                       printf("%s is already running\n%u\n", execname, (unsigned)G.found_procs->pid);
                return !(opt & OPT_OKNODO);
        }
 
@@ -400,7 +485,7 @@ int start_stop_daemon_main(int argc ATTRIBUTE_UNUSED, char **argv)
        *--argv = startas;
        if (opt & OPT_BACKGROUND) {
 #if BB_MMU
-               bb_daemonize(DAEMON_DEVNULL_STDIO + DAEMON_CLOSE_EXTRA_FDS);
+               bb_daemonize(DAEMON_DEVNULL_STDIO + DAEMON_CLOSE_EXTRA_FDS + DAEMON_DOUBLE_FORK);
                /* DAEMON_DEVNULL_STDIO is superfluous -
                 * it's always done by bb_daemonize() */
 #else
@@ -426,10 +511,18 @@ int start_stop_daemon_main(int argc ATTRIBUTE_UNUSED, char **argv)
                write_pidfile(pidfile);
        }
        if (opt & OPT_c) {
-               struct bb_uidgid_t ugid;
+               struct bb_uidgid_t ugid = { -1, -1 };
                parse_chown_usergroup_or_die(&ugid, chuid);
-               if (ugid.gid != (gid_t) -1) xsetgid(ugid.gid);
-               if (ugid.uid != (uid_t) -1) xsetuid(ugid.uid);
+               if (ugid.uid != (uid_t) -1) {
+                       struct passwd *pw = xgetpwuid(ugid.uid);
+                       if (ugid.gid != (gid_t) -1)
+                               pw->pw_gid = ugid.gid;
+                       /* initgroups, setgid, setuid: */
+                       change_identity(pw);
+               } else if (ugid.gid != (gid_t) -1) {
+                       xsetgid(ugid.gid);
+                       setgroups(1, &ugid.gid);
+               }
        }
 #if ENABLE_FEATURE_START_STOP_DAEMON_FANCY
        if (opt & OPT_NICELEVEL) {
@@ -441,5 +534,5 @@ int start_stop_daemon_main(int argc ATTRIBUTE_UNUSED, char **argv)
        }
 #endif
        execvp(startas, argv);
-       bb_perror_msg_and_die("cannot start %s", startas);
+       bb_perror_msg_and_die("can't execute '%s'", startas);
 }