[3320] | 1 | /* vi: set sw=4 ts=4: */
|
---|
| 2 | /*
|
---|
| 3 | * Mini kill/killall[5] implementation for busybox
|
---|
| 4 | *
|
---|
| 5 | * Copyright (C) 1995, 1996 by Bruce Perens <bruce@pixar.com>.
|
---|
| 6 | * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
|
---|
| 7 | *
|
---|
| 8 | * Licensed under GPLv2 or later, see file LICENSE in this source tree.
|
---|
| 9 | */
|
---|
| 10 |
|
---|
| 11 | #include "libbb.h"
|
---|
| 12 |
|
---|
| 13 | /* Note: kill_main is directly called from shell in order to implement
|
---|
| 14 | * kill built-in. Shell substitutes job ids with process groups first.
|
---|
| 15 | *
|
---|
| 16 | * This brings some complications:
|
---|
| 17 | *
|
---|
| 18 | * + we can't use xfunc here
|
---|
| 19 | * + we can't use applet_name
|
---|
| 20 | * + we can't use bb_show_usage
|
---|
| 21 | * (Above doesn't apply for killall[5] cases)
|
---|
| 22 | *
|
---|
| 23 | * kill %n gets translated into kill ' -<process group>' by shell (note space!)
|
---|
| 24 | * This is needed to avoid collision with kill -9 ... syntax
|
---|
| 25 | */
|
---|
| 26 |
|
---|
| 27 | int kill_main(int argc, char **argv)
|
---|
| 28 | {
|
---|
| 29 | char *arg;
|
---|
| 30 | pid_t pid;
|
---|
| 31 | int signo = SIGTERM, errors = 0, quiet = 0;
|
---|
| 32 | #if !ENABLE_KILLALL && !ENABLE_KILLALL5
|
---|
| 33 | #define killall 0
|
---|
| 34 | #define killall5 0
|
---|
| 35 | #else
|
---|
| 36 | /* How to determine who we are? find 3rd char from the end:
|
---|
| 37 | * kill, killall, killall5
|
---|
| 38 | * ^i ^a ^l - it's unique
|
---|
| 39 | * (checking from the start is complicated by /bin/kill... case) */
|
---|
| 40 | const char char3 = argv[0][strlen(argv[0]) - 3];
|
---|
| 41 | #define killall (ENABLE_KILLALL && char3 == 'a')
|
---|
| 42 | #define killall5 (ENABLE_KILLALL5 && char3 == 'l')
|
---|
| 43 | #endif
|
---|
| 44 |
|
---|
| 45 | /* Parse any options */
|
---|
| 46 | argc--;
|
---|
| 47 | arg = *++argv;
|
---|
| 48 |
|
---|
| 49 | if (argc < 1 || arg[0] != '-') {
|
---|
| 50 | goto do_it_now;
|
---|
| 51 | }
|
---|
| 52 |
|
---|
| 53 | /* The -l option, which prints out signal names.
|
---|
| 54 | * Intended usage in shell:
|
---|
| 55 | * echo "Died of SIG`kill -l $?`"
|
---|
| 56 | * We try to mimic what kill from coreutils-6.8 does */
|
---|
| 57 | if (arg[1] == 'l' && arg[2] == '\0') {
|
---|
| 58 | if (argc == 1) {
|
---|
| 59 | /* Print the whole signal list */
|
---|
| 60 | print_signames();
|
---|
| 61 | return 0;
|
---|
| 62 | }
|
---|
| 63 | /* -l <sig list> */
|
---|
| 64 | while ((arg = *++argv)) {
|
---|
| 65 | if (isdigit(arg[0])) {
|
---|
| 66 | signo = bb_strtou(arg, NULL, 10);
|
---|
| 67 | if (errno) {
|
---|
| 68 | bb_error_msg("unknown signal '%s'", arg);
|
---|
| 69 | return EXIT_FAILURE;
|
---|
| 70 | }
|
---|
| 71 | /* Exitcodes >= 0x80 are to be treated
|
---|
| 72 | * as "killed by signal (exitcode & 0x7f)" */
|
---|
| 73 | puts(get_signame(signo & 0x7f));
|
---|
| 74 | /* TODO: 'bad' signal# - coreutils says:
|
---|
| 75 | * kill: 127: invalid signal
|
---|
| 76 | * we just print "127" instead */
|
---|
| 77 | } else {
|
---|
| 78 | signo = get_signum(arg);
|
---|
| 79 | if (signo < 0) {
|
---|
| 80 | bb_error_msg("unknown signal '%s'", arg);
|
---|
| 81 | return EXIT_FAILURE;
|
---|
| 82 | }
|
---|
| 83 | printf("%d\n", signo);
|
---|
| 84 | }
|
---|
| 85 | }
|
---|
| 86 | /* If they specified -l, we are all done */
|
---|
| 87 | return EXIT_SUCCESS;
|
---|
| 88 | }
|
---|
| 89 |
|
---|
| 90 | /* The -q quiet option */
|
---|
| 91 | if (killall && arg[1] == 'q' && arg[2] == '\0') {
|
---|
| 92 | quiet = 1;
|
---|
| 93 | arg = *++argv;
|
---|
| 94 | argc--;
|
---|
| 95 | if (argc < 1)
|
---|
| 96 | bb_show_usage();
|
---|
| 97 | if (arg[0] != '-')
|
---|
| 98 | goto do_it_now;
|
---|
| 99 | }
|
---|
| 100 |
|
---|
| 101 | arg++; /* skip '-' */
|
---|
| 102 |
|
---|
| 103 | /* -o PID? (if present, it always is at the end of command line) */
|
---|
| 104 | if (killall5 && arg[0] == 'o')
|
---|
| 105 | goto do_it_now;
|
---|
| 106 |
|
---|
| 107 | if (argc > 1 && arg[0] == 's' && arg[1] == '\0') { /* -s SIG? */
|
---|
| 108 | argc--;
|
---|
| 109 | arg = *++argv;
|
---|
| 110 | } /* else it must be -SIG */
|
---|
| 111 | signo = get_signum(arg);
|
---|
| 112 | if (signo < 0) { /* || signo > MAX_SIGNUM ? */
|
---|
| 113 | bb_error_msg("bad signal name '%s'", arg);
|
---|
| 114 | return EXIT_FAILURE;
|
---|
| 115 | }
|
---|
| 116 | arg = *++argv;
|
---|
| 117 | argc--;
|
---|
| 118 |
|
---|
| 119 | do_it_now:
|
---|
| 120 | pid = getpid();
|
---|
| 121 |
|
---|
| 122 | if (killall5) {
|
---|
| 123 | pid_t sid;
|
---|
| 124 | procps_status_t* p = NULL;
|
---|
| 125 | int ret = 0;
|
---|
| 126 |
|
---|
| 127 | /* Find out our session id */
|
---|
| 128 | sid = getsid(pid);
|
---|
| 129 | /* Stop all processes */
|
---|
| 130 | kill(-1, SIGSTOP);
|
---|
| 131 | /* Signal all processes except those in our session */
|
---|
| 132 | while ((p = procps_scan(p, PSSCAN_PID|PSSCAN_SID))) {
|
---|
| 133 | int i;
|
---|
| 134 |
|
---|
| 135 | if (p->sid == (unsigned)sid
|
---|
| 136 | || p->pid == (unsigned)pid
|
---|
| 137 | || p->pid == 1)
|
---|
| 138 | continue;
|
---|
| 139 |
|
---|
| 140 | /* All remaining args must be -o PID options.
|
---|
| 141 | * Check p->pid against them. */
|
---|
| 142 | for (i = 0; i < argc; i++) {
|
---|
| 143 | pid_t omit;
|
---|
| 144 |
|
---|
| 145 | arg = argv[i];
|
---|
| 146 | if (arg[0] != '-' || arg[1] != 'o') {
|
---|
| 147 | bb_error_msg("bad option '%s'", arg);
|
---|
| 148 | ret = 1;
|
---|
| 149 | goto resume;
|
---|
| 150 | }
|
---|
| 151 | arg += 2;
|
---|
| 152 | if (!arg[0] && argv[++i])
|
---|
| 153 | arg = argv[i];
|
---|
| 154 | omit = bb_strtoi(arg, NULL, 10);
|
---|
| 155 | if (errno) {
|
---|
| 156 | bb_error_msg("invalid number '%s'", arg);
|
---|
| 157 | ret = 1;
|
---|
| 158 | goto resume;
|
---|
| 159 | }
|
---|
| 160 | if (p->pid == omit)
|
---|
| 161 | goto dont_kill;
|
---|
| 162 | }
|
---|
| 163 | kill(p->pid, signo);
|
---|
| 164 | dont_kill: ;
|
---|
| 165 | }
|
---|
| 166 | resume:
|
---|
| 167 | /* And let them continue */
|
---|
| 168 | kill(-1, SIGCONT);
|
---|
| 169 | return ret;
|
---|
| 170 | }
|
---|
| 171 |
|
---|
| 172 | /* Pid or name is required for kill/killall */
|
---|
| 173 | if (argc < 1) {
|
---|
| 174 | bb_error_msg("you need to specify whom to kill");
|
---|
| 175 | return EXIT_FAILURE;
|
---|
| 176 | }
|
---|
| 177 |
|
---|
| 178 | if (killall) {
|
---|
| 179 | /* Looks like they want to do a killall. Do that */
|
---|
| 180 | while (arg) {
|
---|
| 181 | pid_t* pidList;
|
---|
| 182 |
|
---|
| 183 | pidList = find_pid_by_name(arg);
|
---|
| 184 | if (*pidList == 0) {
|
---|
| 185 | errors++;
|
---|
| 186 | if (!quiet)
|
---|
| 187 | bb_error_msg("%s: no process killed", arg);
|
---|
| 188 | } else {
|
---|
| 189 | pid_t *pl;
|
---|
| 190 |
|
---|
| 191 | for (pl = pidList; *pl; pl++) {
|
---|
| 192 | if (*pl == pid)
|
---|
| 193 | continue;
|
---|
| 194 | if (kill(*pl, signo) == 0)
|
---|
| 195 | continue;
|
---|
| 196 | errors++;
|
---|
| 197 | if (!quiet)
|
---|
| 198 | bb_perror_msg("can't kill pid %d", (int)*pl);
|
---|
| 199 | }
|
---|
| 200 | }
|
---|
| 201 | free(pidList);
|
---|
| 202 | arg = *++argv;
|
---|
| 203 | }
|
---|
| 204 | return errors;
|
---|
| 205 | }
|
---|
| 206 |
|
---|
| 207 | /* Looks like they want to do a kill. Do that */
|
---|
| 208 | while (arg) {
|
---|
| 209 | /* Support shell 'space' trick */
|
---|
| 210 | if (arg[0] == ' ')
|
---|
| 211 | arg++;
|
---|
| 212 | pid = bb_strtoi(arg, NULL, 10);
|
---|
| 213 | if (errno) {
|
---|
| 214 | bb_error_msg("invalid number '%s'", arg);
|
---|
| 215 | errors++;
|
---|
| 216 | } else if (kill(pid, signo) != 0) {
|
---|
| 217 | bb_perror_msg("can't kill pid %d", (int)pid);
|
---|
| 218 | errors++;
|
---|
| 219 | }
|
---|
| 220 | arg = *++argv;
|
---|
| 221 | }
|
---|
| 222 | return errors;
|
---|
| 223 | }
|
---|