Initial public busybox maemo commit, 3:1.10.2.legal-1osso12
[busybox4maemo] / procps / kill.c
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 the GPL v2 or later, see the file LICENSE in this tarball.
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) MAIN_EXTERNALLY_VISIBLE;
28 int kill_main(int argc, char **argv)
29 {
30         char *arg;
31         pid_t pid;
32         int signo = SIGTERM, errors = 0, quiet = 0;
33 #if !ENABLE_KILLALL && !ENABLE_KILLALL5
34 #define killall 0
35 #define killall5 0
36 #else
37 /* How to determine who we are? find 3rd char from the end:
38  * kill, killall, killall5
39  *  ^i       ^a        ^l  - it's unique
40  * (checking from the start is complicated by /bin/kill... case) */
41         const char char3 = argv[0][strlen(argv[0]) - 3];
42 #define killall (ENABLE_KILLALL && char3 == 'a')
43 #define killall5 (ENABLE_KILLALL5 && char3 == 'l')
44 #endif
45
46         /* Parse any options */
47         argc--;
48         arg = *++argv;
49
50         if (argc < 1 || arg[0] != '-') {
51                 goto do_it_now;
52         }
53
54         /* The -l option, which prints out signal names.
55          * Intended usage in shell:
56          * echo "Died of SIG`kill -l $?`"
57          * We try to mimic what kill from coreutils-6.8 does */
58         if (arg[1] == 'l' && arg[2] == '\0') {
59                 if (argc == 1) {
60                         /* Print the whole signal list */
61                         print_signames();
62                         return 0;
63                 }
64                 /* -l <sig list> */
65                 while ((arg = *++argv)) {
66                         if (isdigit(arg[0])) {
67                                 signo = bb_strtou(arg, NULL, 10);
68                                 if (errno) {
69                                         bb_error_msg("unknown signal '%s'", arg);
70                                         return EXIT_FAILURE;
71                                 }
72                                 /* Exitcodes >= 0x80 are to be treated
73                                  * as "killed by signal (exitcode & 0x7f)" */
74                                 puts(get_signame(signo & 0x7f));
75                                 /* TODO: 'bad' signal# - coreutils says:
76                                  * kill: 127: invalid signal
77                                  * we just print "127" instead */
78                         } else {
79                                 signo = get_signum(arg);
80                                 if (signo < 0) {
81                                         bb_error_msg("unknown signal '%s'", arg);
82                                         return EXIT_FAILURE;
83                                 }
84                                 printf("%d\n", signo);
85                         }
86                 }
87                 /* If they specified -l, we are all done */
88                 return EXIT_SUCCESS;
89         }
90
91         /* The -q quiet option */
92         if (killall && arg[1] == 'q' && arg[2] == '\0') {
93                 quiet = 1;
94                 arg = *++argv;
95                 argc--;
96                 if (argc < 1) bb_show_usage();
97                 if (arg[0] != '-') goto do_it_now;
98         }
99
100         /* -SIG */
101         signo = get_signum(&arg[1]);
102         if (signo < 0) { /* || signo > MAX_SIGNUM ? */
103                 bb_error_msg("bad signal name '%s'", &arg[1]);
104                 return EXIT_FAILURE;
105         }
106         arg = *++argv;
107         argc--;
108
109 do_it_now:
110
111         if (killall5) {
112                 pid_t sid;
113                 procps_status_t* p = NULL;
114
115                 /* Now stop all processes */
116                 kill(-1, SIGSTOP);
117                 /* Find out our own session id */
118                 pid = getpid();
119                 sid = getsid(pid);
120                 /* Now kill all processes except our session */
121                 while ((p = procps_scan(p, PSSCAN_PID|PSSCAN_SID))) {
122                         if (p->sid != sid && p->pid != pid && p->pid != 1)
123                                 kill(p->pid, signo);
124                 }
125                 /* And let them continue */
126                 kill(-1, SIGCONT);
127                 return 0;
128         }
129
130         /* Pid or name is required for kill/killall */
131         if (argc < 1) {
132                 bb_error_msg("you need to specify whom to kill");
133                 return EXIT_FAILURE;
134         }
135
136         if (killall) {
137                 /* Looks like they want to do a killall.  Do that */
138                 pid = getpid();
139                 while (arg) {
140                         pid_t* pidList;
141
142                         pidList = find_pid_by_name(arg);
143                         if (*pidList == 0) {
144                                 errors++;
145                                 if (!quiet)
146                                         bb_error_msg("%s: no process killed", arg);
147                         } else {
148                                 pid_t *pl;
149
150                                 for (pl = pidList; *pl; pl++) {
151                                         if (*pl == pid)
152                                                 continue;
153                                         if (kill(*pl, signo) == 0)
154                                                 continue;
155                                         errors++;
156                                         if (!quiet)
157                                                 bb_perror_msg("cannot kill pid %u", (unsigned)*pl);
158                                 }
159                         }
160                         free(pidList);
161                         arg = *++argv;
162                 }
163                 return errors;
164         }
165
166         /* Looks like they want to do a kill. Do that */
167         while (arg) {
168                 /* Support shell 'space' trick */
169                 if (arg[0] == ' ')
170                         arg++;
171                 pid = bb_strtoi(arg, NULL, 10);
172                 if (errno) {
173                         bb_error_msg("bad pid '%s'", arg);
174                         errors++;
175                 } else if (kill(pid, signo) != 0) {
176                         bb_perror_msg("cannot kill pid %d", (int)pid);
177                         errors++;
178                 }
179                 arg = *++argv;
180         }
181         return errors;
182 }