Trailing whitespace removal over entire tree
[oweals/busybox.git] / procps / kill.c
index 8a99e0f9e99d016044a8c85974a91d3a787ee5e2..18121f06fbc3f267577793c765c59ed75c3d7f8a 100644 (file)
 /* vi: set sw=4 ts=4: */
 /*
- * Mini kill implementation for busybox
+ * 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>
  *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
- *
+ * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
  */
 
+#include "busybox.h"
 
-#include "internal.h"
-#include <stdio.h>
-#include <stdlib.h>
-#include <errno.h>
-#include <unistd.h>
-#include <signal.h>
-#include <ctype.h>
-#include <sys/stat.h>
-#include <unistd.h>
-
-static const char *kill_usage =
-       "kill [-signal] process-id [process-id ...]\n\n"
-       "Send a signal (default is SIGTERM) to the specified process(es).\n\n"
-       "Options:\n" "\t-l\tList all signal names and numbers.\n\n";
-
-static const char *killall_usage =
-       "killall [-signal] process-name [process-name ...]\n\n"
-       "Send a signal (default is SIGTERM) to the specified process(es).\n\n"
-       "Options:\n" "\t-l\tList all signal names and numbers.\n\n";
-
-
-#define KILL   0
-#define KILLALL        1
-
-struct signal_name {
-       const char *name;
-       int number;
-};
-
-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}
-};
-
-extern int kill_main(int argc, char **argv)
+int kill_main(int argc, char **argv)
 {
-       int whichApp, sig = SIGTERM;
-       const char *appUsage;
+       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'));
 
-       /* Figure out what we are trying to do here */
-       whichApp = (strcmp(*argv, "killall") == 0)? 
-               KILLALL : KILL; 
-       appUsage = (whichApp == KILLALL)?  killall_usage : kill_usage;
-
-       argc--;
-       argv++;
        /* Parse any options */
-       if (argc < 1)
-               usage(appUsage);
+       argc--;
+       arg = *++argv;
 
-       while (argc > 0 && **argv == '-') {
-               while (*++(*argv)) {
-                       switch (**argv) {
-                       case 'l':
-                               {
-                                       int col = 0;
-                                       const struct signal_name *s = signames;
+       if (argc < 1 || arg[0] != '-') {
+               goto do_it_now;
+       }
 
-                                       while (s->name != 0) {
-                                               col +=
-                                                       fprintf(stderr, "%2d) %-8s", s->number,
-                                                                       (s++)->name);
-                                               if (col > 60) {
-                                                       fprintf(stderr, "\n");
-                                                       col = 0;
-                                               }
-                                       }
-                                       fprintf(stderr, "\n\n");
-                                       exit(TRUE);
+       /* 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;
                                }
-                               break;
-                       case '-':
-                               usage(appUsage);
-                       default:
-                               {
-                                       if (isdigit(**argv)) {
-                                               sig = atoi(*argv);
-                                               if (sig < 0 || sig >= NSIG)
-                                                       goto end;
-                                               else {
-                                                       argc--;
-                                                       argv++;
-                                                       goto do_it_now;
-                                               }
-                                       } else {
-                                               const struct signal_name *s = signames;
-
-                                               while (s->name != 0) {
-                                                       if (strcasecmp(s->name, *argv) == 0) {
-                                                               sig = s->number;
-                                                               argc--;
-                                                               argv++;
-                                                               goto do_it_now;
-                                                       }
-                                                       s++;
-                                               }
-                                               if (s->name == 0)
-                                                       goto end;
-                                       }
+                               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);
                        }
-                       argc--;
-                       argv++;
                }
+               /* If they specified -l, we are all done */
+               return EXIT_SUCCESS;
        }
 
-  do_it_now:
+       /* 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;
+       }
 
-       if (whichApp == KILL) {
-               /* Looks like they want to do a kill. Do that */
-               while (--argc >= 0) {
-                       int pid;
+       /* -SIG */
+       signo = get_signum(&arg[1]);
+       if (signo < 0)
+               bb_error_msg_and_die("bad signal name '%s'", &arg[1]);
+       arg = *++argv;
+       argc--;
 
-                       if (!isdigit(**argv))
-                               fatalError( "Bad PID: %s\n", strerror(errno));
-                       pid = strtol(*argv, NULL, 0);
-                       if (kill(pid, sig) != 0) 
-                               fatalError( "Could not kill pid '%d': %s\n", pid, strerror(errno));
-                       argv++;
+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);
                }
-       } else {
-               /* Looks like they want to do a killall.  Do that */
-               while (--argc >= 0) {
-                       int pid;
+               /* And let them continue */
+               kill(-1, SIGCONT);
+               return 0;
+       }
+
+       /* Pid or name required for kill/killall */
+       if (argc < 1)
+               bb_show_usage();
 
-                       while((pid = findPidByName( *argv))) {
-                               if (kill(pid, sig) != 0) 
-                                       fatalError( "Could not kill pid '%d': %s\n", pid, strerror(errno));
+       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);
+                               }
                        }
-                       argv++;
+                       free(pidList);
+                       arg = *++argv;
                }
+               return errors;
        }
 
-       exit(TRUE);
-
-
-  end:
-       fatalError( "bad signal name: %s\n", *argv);
+       /* 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++;
+               }
+               arg = *++argv;
+       }
+       return errors;
 }