1/*-
2 * Copyright (c) 2004-2009, Jilles Tjoelker
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with
6 * or without modification, are permitted provided that the
7 * following conditions are met:
8 *
9 * 1. Redistributions of source code must retain the above
10 *    copyright notice, this list of conditions and the
11 *    following disclaimer.
12 * 2. Redistributions in binary form must reproduce the
13 *    above copyright notice, this list of conditions and
14 *    the following disclaimer in the documentation and/or
15 *    other materials provided with the distribution.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
18 * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED
19 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
20 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
21 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
22 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY
23 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
25 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
26 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
27 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
28 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
29 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
30 * USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
31 * OF SUCH DAMAGE.
32 */
33
34#include <sys/cdefs.h>
35__FBSDID("$FreeBSD$");
36
37#include <sys/types.h>
38#include <sys/event.h>
39#include <sys/time.h>
40#include <sys/wait.h>
41
42#include <err.h>
43#include <errno.h>
44#include <fcntl.h>
45#include <signal.h>
46#include <stdio.h>
47#include <stdlib.h>
48#include <string.h>
49#include <sysexits.h>
50#include <unistd.h>
51
52static void
53usage(void)
54{
55
56	fprintf(stderr, "usage: pwait [-v] pid ...\n");
57	exit(EX_USAGE);
58}
59
60/*
61 * pwait - wait for processes to terminate
62 */
63int
64main(int argc, char *argv[])
65{
66	int kq;
67	struct kevent *e;
68	int verbose = 0;
69	int opt, nleft, n, i, duplicate, status;
70	long pid;
71	char *s, *end;
72
73	while ((opt = getopt(argc, argv, "v")) != -1) {
74		switch (opt) {
75		case 'v':
76			verbose = 1;
77			break;
78		default:
79			usage();
80			/* NOTREACHED */
81		}
82	}
83
84	argc -= optind;
85	argv += optind;
86
87	if (argc == 0)
88		usage();
89
90	kq = kqueue();
91	if (kq == -1)
92		err(1, "kqueue");
93
94	e = malloc(argc * sizeof(struct kevent));
95	if (e == NULL)
96		err(1, "malloc");
97	nleft = 0;
98	for (n = 0; n < argc; n++) {
99		s = argv[n];
100		if (!strncmp(s, "/proc/", 6)) /* Undocumented Solaris compat */
101			s += 6;
102		errno = 0;
103		pid = strtol(s, &end, 10);
104		if (pid < 0 || *end != '\0' || errno != 0) {
105			warnx("%s: bad process id", s);
106			continue;
107		}
108		duplicate = 0;
109		for (i = 0; i < nleft; i++)
110			if (e[i].ident == (uintptr_t)pid)
111				duplicate = 1;
112		if (!duplicate) {
113			EV_SET(e + nleft, pid, EVFILT_PROC, EV_ADD, NOTE_EXIT,
114			    0, NULL);
115			if (kevent(kq, e + nleft, 1, NULL, 0, NULL) == -1)
116				warn("%ld", pid);
117			else
118				nleft++;
119		}
120	}
121
122	while (nleft > 0) {
123		n = kevent(kq, NULL, 0, e, nleft, NULL);
124		if (n == -1)
125			err(1, "kevent");
126		if (verbose)
127			for (i = 0; i < n; i++) {
128				status = e[i].data;
129				if (WIFEXITED(status))
130					printf("%ld: exited with status %d.\n",
131					    (long)e[i].ident,
132					    WEXITSTATUS(status));
133				else if (WIFSIGNALED(status))
134					printf("%ld: killed by signal %d.\n",
135					    (long)e[i].ident,
136					    WTERMSIG(status));
137				else
138					printf("%ld: terminated.\n",
139					    (long)e[i].ident);
140			}
141		nleft -= n;
142	}
143
144	exit(EX_OK);
145}
146