[apple/shell_cmds.git] / killall / killall.c

/*-
 * Copyright (c) 2000 Peter Wemm <peter@FreeBSD.org>
 * Copyright (c) 2000 Paul Saab <ps@FreeBSD.org>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 * $FreeBSD: src/usr.bin/killall/killall.c,v 1.15 2001/10/10 17:48:44 bde Exp $
 */

#include <sys/cdefs.h>
#include <sys/param.h>
#include <sys/stat.h>
#include <sys/user.h>
#include <sys/sysctl.h>
#include <fcntl.h>
#include <dirent.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <pwd.h>
#include <signal.h>
#include <regex.h>
#include <ctype.h>
#include <err.h>
#include <errno.h>
#include <unistd.h>

static char	*prog;

static void 
usage(void)
{

	fprintf(stderr, "usage: %s [-l] [-v] [-m] [-sig] [-u user] [-t tty] [-c cmd] [cmd]...\n", prog);
	fprintf(stderr, "At least one option or argument to specify processes must be given.\n");
	exit(1);
}

static char *
upper(const char *str)
{
	static char buf[80];
	char *s;

	strncpy(buf, str, sizeof(buf));
	buf[sizeof(buf) - 1] = '\0';
	for (s = buf; *s; s++)
		*s = toupper(*s);
	return buf;
}


static void
printsig(FILE *fp)
{
	const char	*const * p;
	int		cnt;
	int		offset = 0;

	for (cnt = NSIG, p = sys_signame + 1; --cnt; ++p) {
		offset += fprintf(fp, "%s ", upper(*p));
		if (offset >= 75 && cnt > 1) {
			offset = 0;
			fprintf(fp, "\n");
		}
	}
	fprintf(fp, "\n");
}

static void
nosig(char *name)
{

	warnx("unknown signal %s; valid signals:", name);
	printsig(stderr);
	exit(1);
}

int
main(int ac, char **av)
{
	struct kinfo_proc *procs = NULL, *newprocs;
	struct stat	sb;
	struct passwd	*pw;
	regex_t		rgx;
	regmatch_t	pmatch;
	int		i, j;
	char		buf[256];
	char		*user = NULL;
	char		*tty = NULL;
	char		*cmd = NULL;
	int		vflag = 0;
	int		sflag = 0;
	int		dflag = 0;
	int		mflag = 0;
	uid_t		uid = 0;
	dev_t		tdev = 0;
	char		thiscmd[MAXCOMLEN + 1];
	pid_t		thispid;
	uid_t		thisuid;
	dev_t		thistdev;
	int		sig = SIGTERM;
	const char *const *p;
	char		*ep;
	int		errors = 0;
	int		mib[4];
	size_t		miblen;
	int		st, nprocs;
	size_t		size;
	int		matched;
	int		killed = 0;

	prog = av[0];
	av++;
	ac--;

	while (ac > 0) {
		if (strcmp(*av, "-l") == 0) {
			printsig(stdout);
			exit(0);
		}
		if (strcmp(*av, "-help") == 0)
			usage();
		if (**av == '-') {
			++*av;
			switch (**av) {
			case 'u':
				++*av;
				if (**av == '\0')
					++av;
				--ac;
				user = *av;
				break;
			case 't':
				++*av;
				if (**av == '\0')
					++av;
				--ac;
				tty = *av;
				break;
			case 'c':
				++*av;
				if (**av == '\0')
					++av;
				--ac;
				cmd = *av;
				break;
			case 'v':
				vflag++;
				break;
			case 's':
				sflag++;
				break;
			case 'd':
				dflag++;
				break;
			case 'm':
				mflag++;
				break;
			default:
				if (isalpha(**av)) {
					if (strncasecmp(*av, "sig", 3) == 0)
						*av += 3;
					for (sig = NSIG, p = sys_signame + 1;
					     --sig; ++p)
						if (strcasecmp(*p, *av) == 0) {
							sig = p - sys_signame;
							break;
						}
					if (!sig)
						nosig(*av);
				} else if (isdigit(**av)) {
					sig = strtol(*av, &ep, 10);
					if (!*av || *ep)
						errx(1, "illegal signal number: %s", *av);
					if (sig < 0 || sig > NSIG)
						nosig(*av);
				} else
					nosig(*av);
			}
			++av;
			--ac;
		} else {
			break;
		}
	}

	if (user == NULL && tty == NULL && cmd == NULL && ac == 0)
		usage();

	if (tty) {
		if (strncmp(tty, "/dev/", 5) == 0)
			snprintf(buf, sizeof(buf), "%s", tty);
		else if (strncmp(tty, "tty", 3) == 0)
			snprintf(buf, sizeof(buf), "/dev/%s", tty);
		else
			snprintf(buf, sizeof(buf), "/dev/tty%s", tty);
		if (stat(buf, &sb) < 0)
			err(1, "stat(%s)", buf);
		if (!S_ISCHR(sb.st_mode))
			errx(1, "%s: not a character device", buf);
		tdev = sb.st_rdev;
		if (dflag)
			printf("ttydev:0x%x\n", tdev);
	}
	if (user) {
		uid = strtol(user, &ep, 10);
		if ((ep - user) < strlen(user)) {
			pw = getpwnam(user);
			if (pw == NULL)
				errx(1, "user %s does not exist", user);
			uid = pw->pw_uid;
			if (dflag)
				printf("uid:%d\n", uid);
		}
	} else {
		uid = getuid();
		if (uid != 0) {
			pw = getpwuid(uid);
			if (pw)
				user = pw->pw_name;
			if (dflag)
				printf("uid:%d\n", uid);
		}
	}
	size = 0;
	mib[0] = CTL_KERN;
	mib[1] = KERN_PROC;
	mib[2] = KERN_PROC_ALL;
	mib[3] = 0;
	miblen = 3;

	if (user && mib[2] == KERN_PROC_ALL) {
		mib[2] = KERN_PROC_RUID;
		mib[3] = uid;
		miblen = 4;
	}
	if (tty && mib[2] == KERN_PROC_ALL) {
		mib[2] = KERN_PROC_TTY;
		mib[3] = tdev;
		miblen = 4;
	}

	st = sysctl(mib, miblen, NULL, &size, NULL, 0);
	do {
		size += size / 10;
		newprocs = realloc(procs, size);
		if (newprocs == 0) {
			if (procs)
				free(procs);
			errx(1, "could not reallocate memory");
		}
		procs = newprocs;
		st = sysctl(mib, miblen, procs, &size, NULL, 0);
	} while (st == -1 && errno == ENOMEM);
	if (st == -1)
		err(1, "could not sysctl(KERN_PROC)");
	if (size % sizeof(struct kinfo_proc) != 0) {
		fprintf(stderr, "proc size mismatch (%d total, %d chunks)\n",
			size, sizeof(struct kinfo_proc));
		fprintf(stderr, "userland out of sync with kernel, recompile libkvm etc\n");
		exit(1);
	}
	nprocs = size / sizeof(struct kinfo_proc);
	if (dflag)
		printf("nprocs %d\n", nprocs);

	for (i = 0; i < nprocs; i++) {
		thispid = procs[i].kp_proc.p_pid;
		strncpy(thiscmd, procs[i].kp_proc.p_comm, MAXCOMLEN);
		thiscmd[MAXCOMLEN] = '\0';
		thistdev = procs[i].kp_eproc.e_tdev;
		thisuid = procs[i].kp_eproc.e_pcred.p_ruid;	/* real uid */

		matched = 1;
		if (user) {
			if (thisuid != uid)
				matched = 0;
		}
		if (tty) {
			if (thistdev != tdev)
				matched = 0;
		}
		if (cmd) {
			if (mflag) {
				if (regcomp(&rgx, cmd,
				    REG_EXTENDED|REG_NOSUB) != 0) {
					mflag = 0;
					warnx("%s: illegal regexp", cmd);
				}
			}
			if (mflag) {
				pmatch.rm_so = 0;
				pmatch.rm_eo = strlen(thiscmd);
				if (regexec(&rgx, thiscmd, 0, &pmatch,
				    REG_STARTEND) != 0)
					matched = 0;
				regfree(&rgx);
			} else {
				if (strncmp(thiscmd, cmd, MAXCOMLEN) != 0)
					matched = 0;
			}
		}
		if (matched == 0)
			continue;
		if (ac > 0)
			matched = 0;
		for (j = 0; j < ac; j++) {
			if (mflag) {
				if (regcomp(&rgx, av[j],
				    REG_EXTENDED|REG_NOSUB) != 0) {
					mflag = 0;
					warnx("%s: illegal regexp", av[j]);
				}
			}
			if (mflag) {
				pmatch.rm_so = 0;
				pmatch.rm_eo = strlen(thiscmd);
				if (regexec(&rgx, thiscmd, 0, &pmatch,
				    REG_STARTEND) == 0)
					matched = 1;
				regfree(&rgx);
			} else {
				if (strcmp(thiscmd, av[j]) == 0)
					matched = 1;
			}
			if (matched)
				break;
		}
		if (matched == 0)
			continue;
		if (dflag)
			printf("sig:%d, cmd:%s, pid:%d, dev:0x%x uid:%d\n", sig,
			    thiscmd, thispid, thistdev, thisuid);

		if (vflag || sflag)
			printf("kill -%s %d\n", upper(sys_signame[sig]),
			    thispid);

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