xref: /dragonfly/usr.bin/killall/killall.c (revision 1de703da) 
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: src/usr.bin/killall/killall.c,v 1.5.2.4 2001/05/19 19:22:49 phk Exp $
28  * $DragonFly: src/usr.bin/killall/killall.c,v 1.2 2003/06/17 04:29:27 dillon Exp $
29  */
30 
31 #include <sys/cdefs.h>
32 #include <sys/param.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 
49 static char	*prog;
50 
51 static void __dead2
52 usage(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 
60 static char *
61 upper(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 
74 static void
75 printsig(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 
91 static void
92 nosig(char *name)
93 {
94 
95 	warnx("unknown signal %s; valid signals:", name);
96 	printsig(stderr);
97 	exit(1);
98 }
99 
100 int
101 main(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 		pw = getpwnam(user);
229 		if (pw == NULL)
230 			errx(1, "user %s does not exist", user);
231 		uid = pw->pw_uid;
232 		if (dflag)
233 			printf("uid:%d\n", uid);
234 	} else {
235 		uid = getuid();
236 		if (uid != 0) {
237 			pw = getpwuid(uid);
238 			if (pw)
239 				user = pw->pw_name;
240 			if (dflag)
241 				printf("uid:%d\n", uid);
242 		}
243 	}
244 	size = 0;
245 	mib[0] = CTL_KERN;
246 	mib[1] = KERN_PROC;
247 	mib[2] = KERN_PROC_ALL;
248 	mib[3] = 0;
249 	miblen = 3;
250 
251 	if (user && mib[2] == KERN_PROC_ALL) {
252 		mib[2] = KERN_PROC_RUID;
253 		mib[3] = uid;
254 		miblen = 4;
255 	}
256 	if (tty && mib[2] == KERN_PROC_ALL) {
257 		mib[2] = KERN_PROC_TTY;
258 		mib[3] = tdev;
259 		miblen = 4;
260 	}
261 
262 	st = sysctl(mib, miblen, NULL, &size, NULL, 0);
263 	do {
264 		size += size / 10;
265 		newprocs = realloc(procs, size);
266 		if (newprocs == 0) {
267 			if (procs)
268 				free(procs);
269 			errx(1, "could not reallocate memory");
270 		}
271 		procs = newprocs;
272 		st = sysctl(mib, miblen, procs, &size, NULL, 0);
273 	} while (st == -1 && errno == ENOMEM);
274 	if (st == -1)
275 		err(1, "could not sysctl(KERN_PROC)");
276 	if (size % sizeof(struct kinfo_proc) != 0) {
277 		fprintf(stderr, "proc size mismatch (%d total, %d chunks)\n",
278 			size, sizeof(struct kinfo_proc));
279 		fprintf(stderr, "userland out of sync with kernel, recompile libkvm etc\n");
280 		exit(1);
281 	}
282 	nprocs = size / sizeof(struct kinfo_proc);
283 	if (dflag)
284 		printf("nprocs %d\n", nprocs);
285 
286 	for (i = 0; i < nprocs; i++) {
287 		thispid = procs[i].kp_proc.p_pid;
288 		strncpy(thiscmd, procs[i].kp_proc.p_comm, MAXCOMLEN);
289 		thiscmd[MAXCOMLEN] = '\0';
290 		thistdev = procs[i].kp_eproc.e_tdev;
291 		thisuid = procs[i].kp_eproc.e_pcred.p_ruid;	/* real uid */
292 
293 		matched = 1;
294 		if (user) {
295 			if (thisuid != uid)
296 				matched = 0;
297 		}
298 		if (tty) {
299 			if (thistdev != tdev)
300 				matched = 0;
301 		}
302 		if (cmd) {
303 			if (mflag) {
304 				if (regcomp(&rgx, cmd,
305 				    REG_EXTENDED|REG_NOSUB) != 0) {
306 					mflag = 0;
307 					warnx("%s: illegal regexp", cmd);
308 				}
309 			}
310 			if (mflag) {
311 				pmatch.rm_so = 0;
312 				pmatch.rm_eo = strlen(thiscmd);
313 				if (regexec(&rgx, thiscmd, 0, &pmatch,
314 				    REG_STARTEND) != 0)
315 					matched = 0;
316 				regfree(&rgx);
317 			} else {
318 				if (strcmp(thiscmd, cmd) != 0)
319 					matched = 0;
320 			}
321 		}
322 		if (matched == 0)
323 			continue;
324 		matched = 0;
325 		for (j = 0; j < ac; j++) {
326 			if (mflag) {
327 				if (regcomp(&rgx, av[j],
328 				    REG_EXTENDED|REG_NOSUB) != 0) {
329 					mflag = 0;
330 					warnx("%s: illegal regexp", av[j]);
331 				}
332 			}
333 			if (mflag) {
334 				pmatch.rm_so = 0;
335 				pmatch.rm_eo = strlen(thiscmd);
336 				if (regexec(&rgx, thiscmd, 0, &pmatch,
337 				    REG_STARTEND) == 0)
338 					matched = 1;
339 				regfree(&rgx);
340 			} else {
341 				if (strcmp(thiscmd, av[j]) == 0)
342 					matched = 1;
343 			}
344 			if (matched)
345 				break;
346 		}
347 		if (matched == 0)
348 			continue;
349 		if (dflag)
350 			printf("sig:%d, cmd:%s, pid:%d, dev:0x%x uid:%d\n", sig,
351 			    thiscmd, thispid, thistdev, thisuid);
352 
353 		if (vflag || sflag)
354 			printf("kill -%s %d\n", upper(sys_signame[sig]),
355 			    thispid);
356 
357 		killed++;
358 		if (!dflag && !sflag) {
359 			if (kill(thispid, sig) < 0 /* && errno != ESRCH */ ) {
360 				warn("kill -%s %d", upper(sys_signame[sig]),
361 				    thispid);
362 				errors = 1;
363 			}
364 		}
365 	}
366 	if (killed == 0) {
367 		fprintf(stderr, "No matching processes %swere found\n",
368 		    getuid() != 0 ? "belonging to you " : "");
369 		errors = 1;
370 	}
371 	exit(errors);
372 }
373