killall.c revision 78144 
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 * $FreeBSD: head/usr.bin/killall/killall.c 78144 2001-06-12 15:58:31Z obrien $
28 */
29
30#include <sys/cdefs.h>
31#include <sys/param.h>
32#include <sys/lock.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
49static char	*prog;
50
51static void __dead2
52usage(void)
53{
54
55	fprintf(stderr, "usage: %s [-l] [-v] [-m] [-sig] [-u user] [-t tty] [-c cmd] [cmd]...\n", prog);
56	fprintf(stderr, "At least one option or argument to specify processes must be given.\n");
57	exit(1);
58}
59
60static char *
61upper(const char *str)
62{
63	static char buf[80];
64	char *s;
65
66	strncpy(buf, str, sizeof(buf));
67	buf[sizeof(buf) - 1] = '\0';
68	for (s = buf; *s; s++)
69		*s = toupper(*s);
70	return buf;
71}
72
73
74static void
75printsig(FILE *fp)
76{
77	const char	*const * p;
78	int		cnt;
79	int		offset = 0;
80
81	for (cnt = NSIG, p = sys_signame + 1; --cnt; ++p) {
82		offset += fprintf(fp, "%s ", upper(*p));
83		if (offset >= 75 && cnt > 1) {
84			offset = 0;
85			fprintf(fp, "\n");
86		}
87	}
88	fprintf(fp, "\n");
89}
90
91static void
92nosig(char *name)
93{
94
95	warnx("unknown signal %s; valid signals:", name);
96	printsig(stderr);
97	exit(1);
98}
99
100int
101main(int ac, char **av)
102{
103	struct kinfo_proc *procs = NULL, *newprocs;
104	struct stat	sb;
105	struct passwd	*pw;
106	regex_t		rgx;
107	regmatch_t	pmatch;
108	int		i, j;
109	char		buf[256];
110	char		*user = NULL;
111	char		*tty = NULL;
112	char		*cmd = NULL;
113	int		vflag = 0;
114	int		sflag = 0;
115	int		dflag = 0;
116	int		mflag = 0;
117	uid_t		uid = 0;
118	dev_t		tdev = 0;
119	char		thiscmd[MAXCOMLEN + 1];
120	pid_t		thispid;
121	uid_t		thisuid;
122	dev_t		thistdev;
123	int		sig = SIGTERM;
124	const char *const *p;
125	char		*ep;
126	int		errors = 0;
127	int		mib[4];
128	size_t		miblen;
129	int		st, nprocs;
130	size_t		size;
131	int		matched;
132	int		killed = 0;
133
134	prog = av[0];
135	av++;
136	ac--;
137
138	while (ac > 0) {
139		if (strcmp(*av, "-l") == 0) {
140			printsig(stdout);
141			exit(0);
142		}
143		if (strcmp(*av, "-help") == 0)
144			usage();
145		if (**av == '-') {
146			++*av;
147			switch (**av) {
148			case 'u':
149				++*av;
150				if (**av == '\0')
151					++av;
152				--ac;
153				user = *av;
154				break;
155			case 't':
156				++*av;
157				if (**av == '\0')
158					++av;
159				--ac;
160				tty = *av;
161				break;
162			case 'c':
163				++*av;
164				if (**av == '\0')
165					++av;
166				--ac;
167				cmd = *av;
168				break;
169			case 'v':
170				vflag++;
171				break;
172			case 's':
173				sflag++;
174				break;
175			case 'd':
176				dflag++;
177				break;
178			case 'm':
179				mflag++;
180				break;
181			default:
182				if (isalpha(**av)) {
183					if (strncasecmp(*av, "sig", 3) == 0)
184						*av += 3;
185					for (sig = NSIG, p = sys_signame + 1;
186					     --sig; ++p)
187						if (strcasecmp(*p, *av) == 0) {
188							sig = p - sys_signame;
189							break;
190						}
191					if (!sig)
192						nosig(*av);
193				} else if (isdigit(**av)) {
194					sig = strtol(*av, &ep, 10);
195					if (!*av || *ep)
196						errx(1, "illegal signal number: %s", *av);
197					if (sig < 0 || sig > NSIG)
198						nosig(*av);
199				} else
200					nosig(*av);
201			}
202			++av;
203			--ac;
204		} else {
205			break;
206		}
207	}
208
209	if (user == NULL && tty == NULL && cmd == NULL && ac == 0)
210		usage();
211
212	if (tty) {
213		if (strncmp(tty, "/dev/", 5) == 0)
214			snprintf(buf, sizeof(buf), "%s", tty);
215		else if (strncmp(tty, "tty", 3) == 0)
216			snprintf(buf, sizeof(buf), "/dev/%s", tty);
217		else
218			snprintf(buf, sizeof(buf), "/dev/tty%s", tty);
219		if (stat(buf, &sb) < 0)
220			err(1, "stat(%s)", buf);
221		if (!S_ISCHR(sb.st_mode))
222			errx(1, "%s: not a character device", buf);
223		tdev = sb.st_rdev;
224		if (dflag)
225			printf("ttydev:0x%x\n", tdev);
226	}
227	if (user) {
228		uid = strtol(user, &ep, 10);
229		if ((ep - user) < strlen(user)) {
230			pw = getpwnam(user);
231			if (pw == NULL)
232				errx(1, "user %s does not exist", user);
233			uid = pw->pw_uid;
234			if (dflag)
235				printf("uid:%d\n", uid);
236		}
237	} else {
238		uid = getuid();
239		if (uid != 0) {
240			pw = getpwuid(uid);
241			if (pw)
242				user = pw->pw_name;
243			if (dflag)
244				printf("uid:%d\n", uid);
245		}
246	}
247	size = 0;
248	mib[0] = CTL_KERN;
249	mib[1] = KERN_PROC;
250	mib[2] = KERN_PROC_ALL;
251	mib[3] = 0;
252	miblen = 3;
253
254	if (user && mib[2] == KERN_PROC_ALL) {
255		mib[2] = KERN_PROC_RUID;
256		mib[3] = uid;
257		miblen = 4;
258	}
259	if (tty && mib[2] == KERN_PROC_ALL) {
260		mib[2] = KERN_PROC_TTY;
261		mib[3] = tdev;
262		miblen = 4;
263	}
264
265	st = sysctl(mib, miblen, NULL, &size, NULL, 0);
266	do {
267		size += size / 10;
268		newprocs = realloc(procs, size);
269		if (newprocs == 0) {
270			if (procs)
271				free(procs);
272			errx(1, "could not reallocate memory");
273		}
274		procs = newprocs;
275		st = sysctl(mib, miblen, procs, &size, NULL, 0);
276	} while (st == -1 && errno == ENOMEM);
277	if (st == -1)
278		err(1, "could not sysctl(KERN_PROC)");
279	if (size % sizeof(struct kinfo_proc) != 0) {
280		fprintf(stderr, "proc size mismatch (%d total, %d chunks)\n",
281			size, sizeof(struct kinfo_proc));
282		fprintf(stderr, "userland out of sync with kernel, recompile libkvm etc\n");
283		exit(1);
284	}
285	nprocs = size / sizeof(struct kinfo_proc);
286	if (dflag)
287		printf("nprocs %d\n", nprocs);
288
289	for (i = 0; i < nprocs; i++) {
290		thispid = procs[i].ki_pid;
291		strncpy(thiscmd, procs[i].ki_comm, MAXCOMLEN);
292		thiscmd[MAXCOMLEN] = '\0';
293		thistdev = procs[i].ki_tdev;
294		thisuid = procs[i].ki_ruid;	/* real uid */
295
296		matched = 1;
297		if (user) {
298			if (thisuid != uid)
299				matched = 0;
300		}
301		if (tty) {
302			if (thistdev != tdev)
303				matched = 0;
304		}
305		if (cmd) {
306			if (mflag) {
307				if (regcomp(&rgx, cmd,
308				    REG_EXTENDED|REG_NOSUB) != 0) {
309					mflag = 0;
310					warnx("%s: illegal regexp", cmd);
311				}
312			}
313			if (mflag) {
314				pmatch.rm_so = 0;
315				pmatch.rm_eo = strlen(thiscmd);
316				if (regexec(&rgx, thiscmd, 0, &pmatch,
317				    REG_STARTEND) != 0)
318					matched = 0;
319				regfree(&rgx);
320			} else {
321				if (strncmp(thiscmd, cmd, MAXCOMLEN) != 0)
322					matched = 0;
323			}
324		}
325		if (matched == 0)
326			continue;
327		if (ac > 0)
328			matched = 0;
329		for (j = 0; j < ac; j++) {
330			if (mflag) {
331				if (regcomp(&rgx, av[j],
332				    REG_EXTENDED|REG_NOSUB) != 0) {
333					mflag = 0;
334					warnx("%s: illegal regexp", av[j]);
335				}
336			}
337			if (mflag) {
338				pmatch.rm_so = 0;
339				pmatch.rm_eo = strlen(thiscmd);
340				if (regexec(&rgx, thiscmd, 0, &pmatch,
341				    REG_STARTEND) == 0)
342					matched = 1;
343				regfree(&rgx);
344			} else {
345				if (strcmp(thiscmd, av[j]) == 0)
346					matched = 1;
347			}
348			if (matched)
349				break;
350		}
351		if (matched == 0)
352			continue;
353		if (dflag)
354			printf("sig:%d, cmd:%s, pid:%d, dev:0x%x uid:%d\n", sig,
355			    thiscmd, thispid, thistdev, thisuid);
356
357		if (vflag || sflag)
358			printf("kill -%s %d\n", upper(sys_signame[sig]),
359			    thispid);
360
361		killed++;
362		if (!dflag && !sflag) {
363			if (kill(thispid, sig) < 0 /* && errno != ESRCH */ ) {
364				warn("kill -%s %d", upper(sys_signame[sig]),
365				    thispid);
366				errors = 1;
367			}
368		}
369	}
370	if (killed == 0) {
371		fprintf(stderr, "No matching processes %swere found\n",
372		    getuid() != 0 ? "belonging to you " : "");
373		errors = 1;
374	}
375	exit(errors);
376}
377