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