xref: /freebsd/usr.bin/killall/killall.c (revision 657729a89dd578d8cfc70d6616f5c65a48a8b33a)
1 /*-
2  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
3  *
4  * Copyright (c) 2000 Peter Wemm <peter@FreeBSD.org>
5  * Copyright (c) 2000 Paul Saab <ps@FreeBSD.org>
6  * All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  */
29 
30 #include <sys/cdefs.h>
31 __FBSDID("$FreeBSD$");
32 
33 #include <sys/param.h>
34 #include <sys/jail.h>
35 #include <sys/stat.h>
36 #include <sys/uio.h>
37 #include <sys/user.h>
38 #include <sys/sysctl.h>
39 #include <fcntl.h>
40 #include <dirent.h>
41 #include <jail.h>
42 #include <stdint.h>
43 #include <stdio.h>
44 #include <stdlib.h>
45 #include <string.h>
46 #include <pwd.h>
47 #include <signal.h>
48 #include <regex.h>
49 #include <ctype.h>
50 #include <err.h>
51 #include <errno.h>
52 #include <unistd.h>
53 #include <locale.h>
54 
55 static void __dead2
56 usage(void)
57 {
58 
59 	fprintf(stderr, "usage: killall [-delmsqvz] [-help] [-I] [-j jail]\n");
60 	fprintf(stderr,
61 	    "               [-u user] [-t tty] [-c cmd] [-SIGNAL] [cmd]...\n");
62 	fprintf(stderr, "At least one option or argument to specify processes must be given.\n");
63 	exit(1);
64 }
65 
66 
67 static void
68 printsig(FILE *fp)
69 {
70 	const char	*const * p;
71 	int		cnt;
72 	int		offset = 0;
73 
74 	for (cnt = NSIG, p = sys_signame + 1; --cnt; ++p) {
75 		offset += fprintf(fp, "%s ", *p);
76 		if (offset >= 75 && cnt > 1) {
77 			offset = 0;
78 			fprintf(fp, "\n");
79 		}
80 	}
81 	fprintf(fp, "\n");
82 }
83 
84 static void
85 nosig(char *name)
86 {
87 
88 	warnx("unknown signal %s; valid signals:", name);
89 	printsig(stderr);
90 	exit(1);
91 }
92 
93 int
94 main(int ac, char **av)
95 {
96 	char		**saved_av;
97 	struct kinfo_proc *procs, *newprocs;
98 	struct stat	sb;
99 	struct passwd	*pw;
100 	regex_t		rgx;
101 	regmatch_t	pmatch;
102 	int		i, j, ch;
103 	char		buf[256];
104 	char		first;
105 	char		*user = NULL;
106 	char		*tty = NULL;
107 	char		*cmd = NULL;
108 	int		qflag = 0;
109 	int		vflag = 0;
110 	int		sflag = 0;
111 	int		dflag = 0;
112 	int		eflag = 0;
113 	int		Iflag = 0;
114 	int		jflag = 0;
115 	int		mflag = 0;
116 	int		zflag = 0;
117 	uid_t		uid = 0;
118 	dev_t		tdev = 0;
119 	pid_t		mypid;
120 	char		thiscmd[MAXCOMLEN + 1];
121 	pid_t		thispid;
122 	uid_t		thisuid;
123 	dev_t		thistdev;
124 	int		sig = SIGTERM;
125 	const char *const *p;
126 	char		*ep;
127 	int		errors = 0;
128 	int		jid;
129 	int		mib[4];
130 	size_t		miblen;
131 	int		st, nprocs;
132 	size_t		size;
133 	int		matched;
134 	int		killed = 0;
135 
136 	setlocale(LC_ALL, "");
137 
138 	av++;
139 	ac--;
140 
141 	while (ac > 0) {
142 		if (strcmp(*av, "-l") == 0) {
143 			printsig(stdout);
144 			exit(0);
145 		}
146 		if (strcmp(*av, "-help") == 0)
147 			usage();
148 		if (**av == '-') {
149 			++*av;
150 			switch (**av) {
151 			case 'j':
152 				++*av;
153 				if (**av == '\0') {
154 					++av;
155 					--ac;
156 				}
157 				jflag++;
158 				if (*av == NULL)
159 				    	errx(1, "must specify jail");
160 				jid = jail_getid(*av);
161 				if (jid < 0)
162 					errx(1, "%s", jail_errmsg);
163 				if (jail_attach(jid) == -1)
164 					err(1, "jail_attach(%d)", jid);
165 				break;
166 			case 'u':
167 				++*av;
168 				if (**av == '\0') {
169 					++av;
170 					--ac;
171 				}
172 				if (*av == NULL)
173 				    	errx(1, "must specify user");
174 				user = *av;
175 				break;
176 			case 't':
177 				++*av;
178 				if (**av == '\0') {
179 					++av;
180 					--ac;
181 				}
182 				if (*av == NULL)
183 				    	errx(1, "must specify tty");
184 				tty = *av;
185 				break;
186 			case 'c':
187 				++*av;
188 				if (**av == '\0') {
189 					++av;
190 					--ac;
191 				}
192 				if (*av == NULL)
193 				    	errx(1, "must specify procname");
194 				cmd = *av;
195 				break;
196 			case 'q':
197 				qflag++;
198 				break;
199 			case 'v':
200 				vflag++;
201 				break;
202 			case 's':
203 				sflag++;
204 				break;
205 			case 'd':
206 				dflag++;
207 				break;
208 			case 'e':
209 				eflag++;
210 				break;
211 			case 'm':
212 				mflag++;
213 				break;
214 			case 'z':
215 				zflag++;
216 				break;
217 			default:
218 				saved_av = av;
219 				if (isalpha((unsigned char)**av)) {
220 					if (strncasecmp(*av, "SIG", 3) == 0)
221 						*av += 3;
222 					for (sig = NSIG, p = sys_signame + 1;
223 					     --sig; ++p)
224 						if (strcasecmp(*p, *av) == 0) {
225 							sig = p - sys_signame;
226 							break;
227 						}
228 					if (!sig) {
229 						if (**saved_av == 'I') {
230 							av = saved_av;
231 							Iflag = 1;
232 							break;
233 						} else
234 							nosig(*av);
235 					}
236 				} else if (isdigit((unsigned char)**av)) {
237 					sig = strtol(*av, &ep, 10);
238 					if (!*av || *ep)
239 						errx(1, "illegal signal number: %s", *av);
240 					if (sig < 0 || sig >= NSIG)
241 						nosig(*av);
242 				} else
243 					nosig(*av);
244 			}
245 			++av;
246 			--ac;
247 		} else {
248 			break;
249 		}
250 	}
251 
252 	if (user == NULL && tty == NULL && cmd == NULL && !jflag && ac == 0)
253 		usage();
254 
255 	if (tty) {
256 		if (strncmp(tty, "/dev/", 5) == 0)
257 			snprintf(buf, sizeof(buf), "%s", tty);
258 		else if (strncmp(tty, "tty", 3) == 0 ||
259 		    strncmp(tty, "pts/", 4) == 0)
260 			snprintf(buf, sizeof(buf), "/dev/%s", tty);
261 		else
262 			snprintf(buf, sizeof(buf), "/dev/tty%s", tty);
263 		if (stat(buf, &sb) < 0)
264 			err(1, "stat(%s)", buf);
265 		if (!S_ISCHR(sb.st_mode))
266 			errx(1, "%s: not a character device", buf);
267 		tdev = sb.st_rdev;
268 		if (dflag)
269 			printf("ttydev:0x%jx\n", (uintmax_t)tdev);
270 	}
271 	if (user) {
272 		uid = strtol(user, &ep, 10);
273 		if (*user == '\0' || *ep != '\0') { /* was it a number? */
274 			pw = getpwnam(user);
275 			if (pw == NULL)
276 				errx(1, "user %s does not exist", user);
277 			uid = pw->pw_uid;
278 			if (dflag)
279 				printf("uid:%d\n", uid);
280 		}
281 	} else {
282 		uid = getuid();
283 		if (uid != 0) {
284 			pw = getpwuid(uid);
285 			if (pw)
286 				user = pw->pw_name;
287 			if (dflag)
288 				printf("uid:%d\n", uid);
289 		}
290 	}
291 	size = 0;
292 	mib[0] = CTL_KERN;
293 	mib[1] = KERN_PROC;
294 
295 	if (user) {
296 		mib[2] = eflag ? KERN_PROC_UID : KERN_PROC_RUID;
297 		mib[3] = uid;
298 		miblen = 4;
299 	} else if (tty) {
300 		mib[2] = KERN_PROC_TTY;
301 		mib[3] = tdev;
302 		miblen = 4;
303 	} else {
304 		mib[2] = KERN_PROC_PROC;
305 		mib[3] = 0;
306 		miblen = 3;
307 	}
308 
309 	procs = NULL;
310 	st = sysctl(mib, miblen, NULL, &size, NULL, 0);
311 	do {
312 		size += size / 10;
313 		newprocs = realloc(procs, size);
314 		if (newprocs == NULL) {
315 			free(procs);
316 			err(1, "could not reallocate memory");
317 		}
318 		procs = newprocs;
319 		st = sysctl(mib, miblen, procs, &size, NULL, 0);
320 	} while (st == -1 && errno == ENOMEM);
321 	if (st == -1)
322 		err(1, "could not sysctl(KERN_PROC)");
323 	if (size % sizeof(struct kinfo_proc) != 0) {
324 		fprintf(stderr, "proc size mismatch (%zu total, %zu chunks)\n",
325 			size, sizeof(struct kinfo_proc));
326 		fprintf(stderr, "userland out of sync with kernel\n");
327 		exit(1);
328 	}
329 	nprocs = size / sizeof(struct kinfo_proc);
330 	if (dflag)
331 		printf("nprocs %d\n", nprocs);
332 	mypid = getpid();
333 
334 	for (i = 0; i < nprocs; i++) {
335 		if (procs[i].ki_stat == SZOMB && !zflag)
336 			continue;
337 		thispid = procs[i].ki_pid;
338 		strlcpy(thiscmd, procs[i].ki_comm, sizeof(thiscmd));
339 		thistdev = procs[i].ki_tdev;
340 		if (eflag)
341 			thisuid = procs[i].ki_uid;	/* effective uid */
342 		else
343 			thisuid = procs[i].ki_ruid;	/* real uid */
344 
345 		if (thispid == mypid)
346 			continue;
347 		matched = 1;
348 		if (user) {
349 			if (thisuid != uid)
350 				matched = 0;
351 		}
352 		if (tty) {
353 			if (thistdev != tdev)
354 				matched = 0;
355 		}
356 		if (cmd) {
357 			if (mflag) {
358 				if (regcomp(&rgx, cmd,
359 				    REG_EXTENDED|REG_NOSUB) != 0) {
360 					mflag = 0;
361 					warnx("%s: illegal regexp", cmd);
362 				}
363 			}
364 			if (mflag) {
365 				pmatch.rm_so = 0;
366 				pmatch.rm_eo = strlen(thiscmd);
367 				if (regexec(&rgx, thiscmd, 0, &pmatch,
368 				    REG_STARTEND) != 0)
369 					matched = 0;
370 				regfree(&rgx);
371 			} else {
372 				if (strncmp(thiscmd, cmd, MAXCOMLEN) != 0)
373 					matched = 0;
374 			}
375 		}
376 		if (jflag && thispid == getpid())
377 			matched = 0;
378 		if (matched == 0)
379 			continue;
380 		if (ac > 0)
381 			matched = 0;
382 		for (j = 0; j < ac; j++) {
383 			if (mflag) {
384 				if (regcomp(&rgx, av[j],
385 				    REG_EXTENDED|REG_NOSUB) != 0) {
386 					mflag = 0;
387 					warnx("%s: illegal regexp", av[j]);
388 				}
389 			}
390 			if (mflag) {
391 				pmatch.rm_so = 0;
392 				pmatch.rm_eo = strlen(thiscmd);
393 				if (regexec(&rgx, thiscmd, 0, &pmatch,
394 				    REG_STARTEND) == 0)
395 					matched = 1;
396 				regfree(&rgx);
397 			} else {
398 				if (strcmp(thiscmd, av[j]) == 0)
399 					matched = 1;
400 			}
401 			if (matched)
402 				break;
403 		}
404 		if (matched != 0 && Iflag) {
405 			printf("Send signal %d to %s (pid %d uid %d)? ",
406 				sig, thiscmd, thispid, thisuid);
407 			fflush(stdout);
408 			first = ch = getchar();
409 			while (ch != '\n' && ch != EOF)
410 				ch = getchar();
411 			if (first != 'y' && first != 'Y')
412 				matched = 0;
413 		}
414 		if (matched == 0)
415 			continue;
416 		if (dflag)
417 			printf("sig:%d, cmd:%s, pid:%d, dev:0x%jx uid:%d\n",
418 			    sig, thiscmd, thispid, (uintmax_t)thistdev,
419 			    thisuid);
420 
421 		if (vflag || sflag)
422 			printf("kill -%s %d\n", sys_signame[sig], thispid);
423 
424 		killed++;
425 		if (!dflag && !sflag) {
426 			if (kill(thispid, sig) < 0 /* && errno != ESRCH */ ) {
427 				warn("warning: kill -%s %d",
428 				    sys_signame[sig], thispid);
429 				errors = 1;
430 			}
431 		}
432 	}
433 	if (killed == 0) {
434 		if (!qflag)
435 			fprintf(stderr, "No matching processes %swere found\n",
436 			    getuid() != 0 ? "belonging to you " : "");
437 		errors = 1;
438 	}
439 	exit(errors);
440 }
441