find: improve usage text (Natanael Copa <natanael.copa@gmail.com>)
[oweals/busybox.git] / procps / kill.c
index da025fafc963f1099ff8ee9ffa35bd16da664a8a..9a6e936655fdb09e26e824d0155fea159cad4a91 100644 (file)
-#include "internal.h"
-#include <stdio.h>
-#include <stdlib.h>
-#include <unistd.h>
-#include <signal.h>
+/* vi: set sw=4 ts=4: */
+/*
+ * Mini kill/killall implementation for busybox
+ *
+ * Copyright (C) 1995, 1996 by Bruce Perens <bruce@pixar.com>.
+ * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
+ *
+ * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
+ */
 
-const char     kill_usage[] = "kill [-signal] process-id [process-id ...]\n";
+#include "busybox.h"
 
-struct signal_name {
-       const char *    name;
-       int                             number;
-};
+int kill_main(int argc, char **argv);
+int kill_main(int argc, char **argv)
+{
+       char *arg;
+       pid_t pid;
+       int signo = SIGTERM, errors = 0, quiet = 0;
+       const int killall = (ENABLE_KILLALL && applet_name[4] == 'a'
+                      && (!ENABLE_KILLALL5 || applet_name[7] != '5'));
+       const int killall5 = (ENABLE_KILLALL5 && applet_name[4] == 'a'
+                         && (!ENABLE_KILLALL || applet_name[7] == '5'));
 
-const struct signal_name signames[] = {
-       { "HUP",        SIGHUP },
-       { "INT",        SIGINT },
-       { "QUIT",       SIGQUIT },
-       { "ILL",        SIGILL },
-       { "TRAP",       SIGTRAP },
-       { "ABRT",       SIGABRT },
-#ifndef __alpha__
-       { "IOT",        SIGIOT },
-#endif
-#if defined(sparc) || defined(__alpha__)
-       { "EMT",        SIGEMT },
-#else
-       { "BUS",        SIGBUS },
-#endif
-       { "FPE",        SIGFPE },
-       { "KILL",       SIGKILL },
-#if defined(sparc) || defined(__alpha__)
-       { "BUS",        SIGBUS },
-#else
-       { "USR1",       SIGUSR1 },
-#endif
-       { "SEGV",       SIGSEGV },
-#if defined(sparc) || defined(__alpha__)
-       { "SYS",        SIGSYS },
-#else
-       { "USR2",       SIGUSR2 },
-#endif
-       { "PIPE",       SIGPIPE },
-       { "ALRM",       SIGALRM },
-       { "TERM",       SIGTERM },
-#if defined(sparc) || defined(__alpha__)
-       { "URG",        SIGURG },
-       { "STOP",       SIGSTOP },
-       { "TSTP",       SIGTSTP },
-       { "CONT",       SIGCONT },
-       { "CHLD",       SIGCHLD },
-       { "TTIN",       SIGTTIN },
-       { "TTOU",       SIGTTOU },
-       { "IO",         SIGIO },
-# ifndef __alpha__
-       { "POLL",       SIGIO },
-# endif
-       { "XCPU",       SIGXCPU },
-       { "XFSZ",       SIGXFSZ },
-       { "VTALRM",     SIGVTALRM },
-       { "PROF",       SIGPROF },
-       { "WINCH",      SIGWINCH },
-# ifdef __alpha__
-       { "INFO",       SIGINFO },
-# else
-       { "LOST",       SIGLOST },
-# endif
-       { "USR1",       SIGUSR1 },
-       { "USR2",       SIGUSR2 },
-#else
-       { "STKFLT",     SIGSTKFLT },
-       { "CHLD",       SIGCHLD },
-       { "CONT",       SIGCONT },
-       { "STOP",       SIGSTOP },
-       { "TSTP",       SIGTSTP },
-       { "TTIN",       SIGTTIN },
-       { "TTOU",       SIGTTOU },
-       { "URG",        SIGURG },
-       { "XCPU",       SIGXCPU },
-       { "XFSZ",       SIGXFSZ },
-       { "VTALRM",     SIGVTALRM },
-       { "PROF",       SIGPROF },
-       { "WINCH",      SIGWINCH },
-       { "IO",         SIGIO },
-       { "POLL",       SIGPOLL },
-       { "PWR",        SIGPWR },
-       { "UNUSED",     SIGUNUSED },
-#endif
-       { 0,            0               }
-};
+       /* Parse any options */
+       argc--;
+       arg = *++argv;
 
-extern int
-kill_main(struct FileInfo * i, int argc, char * * argv)
-{
-       int     had_error = 0;
-       int     sig = SIGTERM;
-       if ( argv[1][0] == '-' ) {
-               if ( argv[1][1] >= '0' && argv[1][1] <= '9' ) {
-                       sig = atoi(&argv[1][1]);
-                       if ( sig < 0 || sig >= NSIG ) {
-                               usage(kill_usage);
-                               exit(-1);
-                       }
-               }
-               else {
-                       const struct signal_name *      s = signames;
-                       for ( ; ; ) {
-                               if ( strcmp(s->name, &argv[1][1]) == 0 ) {
-                                       sig = s->number;
-                                       break;
+       if (argc < 1 || arg[0] != '-') {
+               goto do_it_now;
+       }
+
+       /* The -l option, which prints out signal names. */
+       if (arg[1] == 'l' && arg[2] == '\0') {
+               const char *name;
+               if (argc == 1) {
+                       /* Print the whole signal list */
+                       int col = 0;
+                       for (signo = 1; signo < 32; signo++) {
+                               name = get_signame(signo);
+                               if (isdigit(name[0])) continue;
+                               if (col > 66) {
+                                       puts("");
+                                       col = 0;
                                }
-                               s++;
-                               if ( s->name == 0 ) {
-                                       usage(kill_usage);
-                                       exit(-1);
+                               col += printf("%2d) %-6s", signo, name);
+                       }
+                       puts("");
+               } else { /* -l <sig list> */
+                       while ((arg = *++argv)) {
+                               if (isdigit(arg[0])) {
+                                       signo = xatoi_u(arg);
+                                       name = get_signame(signo);
+                               } else {
+                                       signo = get_signum(arg);
+                                       if (signo < 0)
+                                               bb_error_msg_and_die("unknown signal '%s'", arg);
+                                       name = get_signame(signo);
                                }
+                               printf("%2d) %s\n", signo, name);
                        }
                }
-               argv++;
+               /* If they specified -l, we are all done */
+               return EXIT_SUCCESS;
+       }
+
+       /* The -q quiet option */
+       if (killall && arg[1] == 'q' && arg[2] == '\0') {
+               quiet = 1;
+               arg = *++argv;
                argc--;
+               if (argc < 1) bb_show_usage();
+               if (arg[0] != '-') goto do_it_now;
+       }
+
+       /* -SIG */
+       signo = get_signum(&arg[1]);
+       if (signo < 0)
+               bb_error_msg_and_die("bad signal name '%s'", &arg[1]);
+       arg = *++argv;
+       argc--;
+
+do_it_now:
+
+       if (killall5) {
+               pid_t sid;
+               procps_status_t* p = NULL;
 
+// Cannot happen anyway? We don't TERM ourself, we STOP
+//             /* kill(-1, sig) on Linux (at least 2.1.x)
+//              * might send signal to the calling process too */
+//             signal(SIGTERM, SIG_IGN);
+               /* Now stop all processes */
+               kill(-1, SIGSTOP);
+               /* Find out our own session id */
+               pid = getpid();
+               sid = getsid(pid);
+               /* Now kill all processes except our session */
+               while ((p = procps_scan(p, PSSCAN_PID|PSSCAN_SID))) {
+                       if (p->sid != sid && p->pid != pid && p->pid != 1)
+                               kill(p->pid, signo);
+               }
+               /* And let them continue */
+               kill(-1, SIGCONT);
+               return 0;
        }
-       while ( argc > 1 ) {
-               int     pid;
-               if ( argv[1][0] < '0' || argv[1][0] > '9' ) {
-                       usage(kill_usage);
-                       exit(-1);
+
+       /* Pid or name required for kill/killall */
+       if (argc < 1)
+               bb_show_usage();
+
+       if (killall) {
+               /* Looks like they want to do a killall.  Do that */
+               pid = getpid();
+               while (arg) {
+                       pid_t* pidList;
+
+                       pidList = find_pid_by_name(arg);
+                       if (*pidList == 0) {
+                               errors++;
+                               if (!quiet)
+                                       bb_error_msg("%s: no process killed", arg);
+                       } else {
+                               pid_t *pl;
+
+                               for (pl = pidList; *pl; pl++) {
+                                       if (*pl == pid)
+                                               continue;
+                                       if (kill(*pl, signo) == 0)
+                                               continue;
+                                       errors++;
+                                       if (!quiet)
+                                               bb_perror_msg("cannot kill pid %u", (unsigned)*pl);
+                               }
+                       }
+                       free(pidList);
+                       arg = *++argv;
                }
-               pid = atoi(argv[1]);
-               if ( kill(pid, sig) != 0 ) {
-                       had_error = 1;
-                       perror(argv[1]);
+               return errors;
+       }
+
+       /* Looks like they want to do a kill. Do that */
+       while (arg) {
+               /* Huh?
+               if (!isdigit(arg[0]) && arg[0] != '-')
+                       bb_error_msg_and_die("bad pid '%s'", arg);
+               */
+               pid = xatou(arg);
+               /* FIXME: better overflow check? */
+               if (kill(pid, signo) != 0) {
+                       bb_perror_msg("cannot kill pid %u", (unsigned)pid);
+                       errors++;
                }
-               argv++;
-               argc--;
+               arg = *++argv;
        }
-       if ( had_error )
-               return -1;
-       else
-               return 0;
+       return errors;
 }