xref: /freebsd/usr.sbin/pmcstat/pmcstat.c (revision 165b10b07ba5fc59ae46b4bc60d355b2430fa948)
1ebccf1e3SJoseph Koshy /*-
2165b10b0SJoseph Koshy  * Copyright (c) 2003-2008, Joseph Koshy
3b6010f9eSJoseph Koshy  * Copyright (c) 2007 The FreeBSD Foundation
4ebccf1e3SJoseph Koshy  * All rights reserved.
5ebccf1e3SJoseph Koshy  *
6b6010f9eSJoseph Koshy  * Portions of this software were developed by A. Joseph Koshy under
7b6010f9eSJoseph Koshy  * sponsorship from the FreeBSD Foundation and Google, Inc.
8b6010f9eSJoseph Koshy  *
9ebccf1e3SJoseph Koshy  * Redistribution and use in source and binary forms, with or without
10ebccf1e3SJoseph Koshy  * modification, are permitted provided that the following conditions
11ebccf1e3SJoseph Koshy  * are met:
12ebccf1e3SJoseph Koshy  * 1. Redistributions of source code must retain the above copyright
13ebccf1e3SJoseph Koshy  *    notice, this list of conditions and the following disclaimer.
14ebccf1e3SJoseph Koshy  * 2. Redistributions in binary form must reproduce the above copyright
15ebccf1e3SJoseph Koshy  *    notice, this list of conditions and the following disclaimer in the
16ebccf1e3SJoseph Koshy  *    documentation and/or other materials provided with the distribution.
17ebccf1e3SJoseph Koshy  *
18ebccf1e3SJoseph Koshy  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
19ebccf1e3SJoseph Koshy  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20ebccf1e3SJoseph Koshy  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21ebccf1e3SJoseph Koshy  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
22ebccf1e3SJoseph Koshy  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23ebccf1e3SJoseph Koshy  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
24ebccf1e3SJoseph Koshy  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25ebccf1e3SJoseph Koshy  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26ebccf1e3SJoseph Koshy  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27ebccf1e3SJoseph Koshy  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28ebccf1e3SJoseph Koshy  * SUCH DAMAGE.
29ebccf1e3SJoseph Koshy  */
30ebccf1e3SJoseph Koshy 
31ebccf1e3SJoseph Koshy #include <sys/cdefs.h>
32ebccf1e3SJoseph Koshy __FBSDID("$FreeBSD$");
33ebccf1e3SJoseph Koshy 
34ebccf1e3SJoseph Koshy #include <sys/types.h>
35ebccf1e3SJoseph Koshy #include <sys/event.h>
36fbd8b34aSJoseph Koshy #include <sys/param.h>
37ebccf1e3SJoseph Koshy #include <sys/queue.h>
3815139246SJoseph Koshy #include <sys/socket.h>
3915139246SJoseph Koshy #include <sys/stat.h>
40ee4910b1SJoseph Koshy #include <sys/sysctl.h>
41ebccf1e3SJoseph Koshy #include <sys/time.h>
42ebccf1e3SJoseph Koshy #include <sys/ttycom.h>
43fbd8b34aSJoseph Koshy #include <sys/user.h>
44ebccf1e3SJoseph Koshy #include <sys/wait.h>
45ebccf1e3SJoseph Koshy 
46ebccf1e3SJoseph Koshy #include <assert.h>
47ebccf1e3SJoseph Koshy #include <err.h>
48ebccf1e3SJoseph Koshy #include <errno.h>
49ebccf1e3SJoseph Koshy #include <fcntl.h>
50fbd8b34aSJoseph Koshy #include <kvm.h>
5149874f6eSJoseph Koshy #include <libgen.h>
52ebccf1e3SJoseph Koshy #include <limits.h>
53ebccf1e3SJoseph Koshy #include <math.h>
54ebccf1e3SJoseph Koshy #include <pmc.h>
55f263522aSJoseph Koshy #include <pmclog.h>
56fbd8b34aSJoseph Koshy #include <regex.h>
57ebccf1e3SJoseph Koshy #include <signal.h>
58ebccf1e3SJoseph Koshy #include <stdarg.h>
59ebccf1e3SJoseph Koshy #include <stdint.h>
6015139246SJoseph Koshy #include <stdio.h>
61ebccf1e3SJoseph Koshy #include <stdlib.h>
62ebccf1e3SJoseph Koshy #include <string.h>
63ebccf1e3SJoseph Koshy #include <sysexits.h>
64ebccf1e3SJoseph Koshy #include <unistd.h>
65ebccf1e3SJoseph Koshy 
6615139246SJoseph Koshy #include "pmcstat.h"
6715139246SJoseph Koshy 
68f263522aSJoseph Koshy /*
69f263522aSJoseph Koshy  * A given invocation of pmcstat(8) can manage multiple PMCs of both
70f263522aSJoseph Koshy  * the system-wide and per-process variety.  Each of these could be in
71f263522aSJoseph Koshy  * 'counting mode' or in 'sampling mode'.
72f263522aSJoseph Koshy  *
73f263522aSJoseph Koshy  * For 'counting mode' PMCs, pmcstat(8) will periodically issue a
74f263522aSJoseph Koshy  * pmc_read() at the configured time interval and print out the value
75f263522aSJoseph Koshy  * of the requested PMCs.
76f263522aSJoseph Koshy  *
77f263522aSJoseph Koshy  * For 'sampling mode' PMCs it can log to a file for offline analysis,
78f263522aSJoseph Koshy  * or can analyse sampling data "on the fly", either by converting
79f263522aSJoseph Koshy  * samples to printed textual form or by creating gprof(1) compatible
80f263522aSJoseph Koshy  * profiles, one per program executed.  When creating gprof(1)
81f263522aSJoseph Koshy  * profiles it can optionally merge entries from multiple processes
82f263522aSJoseph Koshy  * for a given executable into a single profile file.
83ee4910b1SJoseph Koshy  *
84ee4910b1SJoseph Koshy  * pmcstat(8) can also execute a command line and attach PMCs to the
85ee4910b1SJoseph Koshy  * resulting child process.  The protocol used is as follows:
86ee4910b1SJoseph Koshy  *
87ee4910b1SJoseph Koshy  * - parent creates a socketpair for two way communication and
88ee4910b1SJoseph Koshy  *   fork()s.
89ee4910b1SJoseph Koshy  * - subsequently:
90ee4910b1SJoseph Koshy  *
91ee4910b1SJoseph Koshy  *   /Parent/				/Child/
92ee4910b1SJoseph Koshy  *
93ee4910b1SJoseph Koshy  *   - Wait for childs token.
94ee4910b1SJoseph Koshy  *					- Sends token.
95ee4910b1SJoseph Koshy  *					- Awaits signal to start.
96ee4910b1SJoseph Koshy  *  - Attaches PMCs to the child's pid
97ee4910b1SJoseph Koshy  *    and starts them. Sets up
98ee4910b1SJoseph Koshy  *    monitoring for the child.
99ee4910b1SJoseph Koshy  *  - Signals child to start.
100ee4910b1SJoseph Koshy  *					- Recieves signal, attempts exec().
101ee4910b1SJoseph Koshy  *
102ee4910b1SJoseph Koshy  * After this point normal processing can happen.
103f263522aSJoseph Koshy  */
104f263522aSJoseph Koshy 
10515139246SJoseph Koshy /* Globals */
106ebccf1e3SJoseph Koshy 
107ebccf1e3SJoseph Koshy int	pmcstat_interrupt = 0;
108ebccf1e3SJoseph Koshy int	pmcstat_displayheight = DEFAULT_DISPLAY_HEIGHT;
109ee4910b1SJoseph Koshy int	pmcstat_sockpair[NSOCKPAIRFD];
110ebccf1e3SJoseph Koshy int	pmcstat_kq;
111fbd8b34aSJoseph Koshy kvm_t	*pmcstat_kvm;
112fbd8b34aSJoseph Koshy struct kinfo_proc *pmcstat_plist;
113ebccf1e3SJoseph Koshy 
114ee4910b1SJoseph Koshy void
115ee4910b1SJoseph Koshy pmcstat_attach_pmcs(struct pmcstat_args *a)
116ee4910b1SJoseph Koshy {
117ee4910b1SJoseph Koshy 	struct pmcstat_ev *ev;
118fbd8b34aSJoseph Koshy 	struct pmcstat_target *pt;
119fbd8b34aSJoseph Koshy 	int count;
120ee4910b1SJoseph Koshy 
121fbd8b34aSJoseph Koshy 	/* Attach all process PMCs to target processes. */
122fbd8b34aSJoseph Koshy 	count = 0;
123fbd8b34aSJoseph Koshy 	STAILQ_FOREACH(ev, &a->pa_events, ev_next) {
124fbd8b34aSJoseph Koshy 		if (PMC_IS_SYSTEM_MODE(ev->ev_mode))
125fbd8b34aSJoseph Koshy 			continue;
126fbd8b34aSJoseph Koshy 		SLIST_FOREACH(pt, &a->pa_targets, pt_next)
127fbd8b34aSJoseph Koshy 			if (pmc_attach(ev->ev_pmcid, pt->pt_pid) == 0)
128fbd8b34aSJoseph Koshy 				count++;
129fbd8b34aSJoseph Koshy 			else if (errno != ESRCH)
130fbd8b34aSJoseph Koshy 				err(EX_OSERR, "ERROR: cannot attach pmc "
131fbd8b34aSJoseph Koshy 				    "\"%s\" to process %d", ev->ev_name,
132fbd8b34aSJoseph Koshy 				    (int) pt->pt_pid);
133fbd8b34aSJoseph Koshy 	}
134ee4910b1SJoseph Koshy 
135fbd8b34aSJoseph Koshy 	if (count == 0)
136fbd8b34aSJoseph Koshy 		errx(EX_DATAERR, "ERROR: No processes were attached to.");
137ee4910b1SJoseph Koshy }
138ee4910b1SJoseph Koshy 
139ebccf1e3SJoseph Koshy 
140ebccf1e3SJoseph Koshy void
141ebccf1e3SJoseph Koshy pmcstat_cleanup(struct pmcstat_args *a)
142ebccf1e3SJoseph Koshy {
143ebccf1e3SJoseph Koshy 	struct pmcstat_ev *ev, *tmp;
144ebccf1e3SJoseph Koshy 
145ebccf1e3SJoseph Koshy 	/* release allocated PMCs. */
146fbd8b34aSJoseph Koshy 	STAILQ_FOREACH_SAFE(ev, &a->pa_events, ev_next, tmp)
147ebccf1e3SJoseph Koshy 	    if (ev->ev_pmcid != PMC_ID_INVALID) {
148ebccf1e3SJoseph Koshy 		if (pmc_release(ev->ev_pmcid) < 0)
149ebccf1e3SJoseph Koshy 			err(EX_OSERR, "ERROR: cannot release pmc "
150f263522aSJoseph Koshy 			    "0x%x \"%s\"", ev->ev_pmcid, ev->ev_name);
151ebccf1e3SJoseph Koshy 		free(ev->ev_name);
152ebccf1e3SJoseph Koshy 		free(ev->ev_spec);
153fbd8b34aSJoseph Koshy 		STAILQ_REMOVE(&a->pa_events, ev, pmcstat_ev, ev_next);
154ebccf1e3SJoseph Koshy 		free(ev);
155ebccf1e3SJoseph Koshy 	    }
156f263522aSJoseph Koshy 
157dc1d9d2eSJoseph Koshy 	/* de-configure the log file if present. */
158dc1d9d2eSJoseph Koshy 	if (a->pa_flags & (FLAG_HAS_PIPE | FLAG_HAS_OUTPUT_LOGFILE))
159dc1d9d2eSJoseph Koshy 		(void) pmc_configure_logfile(-1);
160dc1d9d2eSJoseph Koshy 
161f263522aSJoseph Koshy 	if (a->pa_logparser) {
162f263522aSJoseph Koshy 		pmclog_close(a->pa_logparser);
163f263522aSJoseph Koshy 		a->pa_logparser = NULL;
164f263522aSJoseph Koshy 	}
16515139246SJoseph Koshy 
16615139246SJoseph Koshy 	if (a->pa_flags & (FLAG_HAS_PIPE | FLAG_HAS_OUTPUT_LOGFILE))
16749874f6eSJoseph Koshy 		pmcstat_shutdown_logging(a);
168ebccf1e3SJoseph Koshy }
169ebccf1e3SJoseph Koshy 
170ebccf1e3SJoseph Koshy void
171ee4910b1SJoseph Koshy pmcstat_clone_event_descriptor(struct pmcstat_args *a, struct pmcstat_ev *ev,
172ee4910b1SJoseph Koshy     uint32_t cpumask)
173ee4910b1SJoseph Koshy {
174ee4910b1SJoseph Koshy 	int cpu;
175ee4910b1SJoseph Koshy 	struct pmcstat_ev *ev_clone;
176ee4910b1SJoseph Koshy 
177ee4910b1SJoseph Koshy 	while ((cpu = ffs(cpumask)) > 0) {
178ee4910b1SJoseph Koshy 		cpu--;
179ee4910b1SJoseph Koshy 
180ee4910b1SJoseph Koshy 		if ((ev_clone = malloc(sizeof(*ev_clone))) == NULL)
181ee4910b1SJoseph Koshy 			errx(EX_SOFTWARE, "ERROR: Out of memory");
182ee4910b1SJoseph Koshy 		(void) memset(ev_clone, 0, sizeof(*ev_clone));
183ee4910b1SJoseph Koshy 
184ee4910b1SJoseph Koshy 		ev_clone->ev_count = ev->ev_count;
185ee4910b1SJoseph Koshy 		ev_clone->ev_cpu   = cpu;
186ee4910b1SJoseph Koshy 		ev_clone->ev_cumulative = ev->ev_cumulative;
187ee4910b1SJoseph Koshy 		ev_clone->ev_flags = ev->ev_flags;
188ee4910b1SJoseph Koshy 		ev_clone->ev_mode  = ev->ev_mode;
189ee4910b1SJoseph Koshy 		ev_clone->ev_name  = strdup(ev->ev_name);
190ee4910b1SJoseph Koshy 		ev_clone->ev_pmcid = ev->ev_pmcid;
191ee4910b1SJoseph Koshy 		ev_clone->ev_saved = ev->ev_saved;
192ee4910b1SJoseph Koshy 		ev_clone->ev_spec  = strdup(ev->ev_spec);
193ee4910b1SJoseph Koshy 
194fbd8b34aSJoseph Koshy 		STAILQ_INSERT_TAIL(&a->pa_events, ev_clone, ev_next);
195ee4910b1SJoseph Koshy 
196ee4910b1SJoseph Koshy 		cpumask &= ~(1 << cpu);
197ee4910b1SJoseph Koshy 	}
198ee4910b1SJoseph Koshy }
199ee4910b1SJoseph Koshy 
200ee4910b1SJoseph Koshy void
201ee4910b1SJoseph Koshy pmcstat_create_process(struct pmcstat_args *a)
202ee4910b1SJoseph Koshy {
203ee4910b1SJoseph Koshy 	char token;
204fbd8b34aSJoseph Koshy 	pid_t pid;
205ee4910b1SJoseph Koshy 	struct kevent kev;
206fbd8b34aSJoseph Koshy 	struct pmcstat_target *pt;
207ee4910b1SJoseph Koshy 
208ee4910b1SJoseph Koshy 	if (socketpair(AF_UNIX, SOCK_STREAM, 0, pmcstat_sockpair) < 0)
209ee4910b1SJoseph Koshy 		err(EX_OSERR, "ERROR: cannot create socket pair");
210ee4910b1SJoseph Koshy 
211fbd8b34aSJoseph Koshy 	switch (pid = fork()) {
212ee4910b1SJoseph Koshy 	case -1:
213ee4910b1SJoseph Koshy 		err(EX_OSERR, "ERROR: cannot fork");
214ee4910b1SJoseph Koshy 		/*NOTREACHED*/
215ee4910b1SJoseph Koshy 
216ee4910b1SJoseph Koshy 	case 0:		/* child */
217ee4910b1SJoseph Koshy 		(void) close(pmcstat_sockpair[PARENTSOCKET]);
218ee4910b1SJoseph Koshy 
219ee4910b1SJoseph Koshy 		/* Write a token to tell our parent we've started executing. */
220ee4910b1SJoseph Koshy 		if (write(pmcstat_sockpair[CHILDSOCKET], "+", 1) != 1)
221ee4910b1SJoseph Koshy 			err(EX_OSERR, "ERROR (child): cannot write token");
222ee4910b1SJoseph Koshy 
223ee4910b1SJoseph Koshy 		/* Wait for our parent to signal us to start. */
224ee4910b1SJoseph Koshy 		if (read(pmcstat_sockpair[CHILDSOCKET], &token, 1) < 0)
225ee4910b1SJoseph Koshy 			err(EX_OSERR, "ERROR (child): cannot read token");
226ee4910b1SJoseph Koshy 		(void) close(pmcstat_sockpair[CHILDSOCKET]);
227ee4910b1SJoseph Koshy 
228ee4910b1SJoseph Koshy 		/* exec() the program requested */
229ee4910b1SJoseph Koshy 		execvp(*a->pa_argv, a->pa_argv);
230ee4910b1SJoseph Koshy 		/* and if that fails, notify the parent */
231ee4910b1SJoseph Koshy 		kill(getppid(), SIGCHLD);
232ee4910b1SJoseph Koshy 		err(EX_OSERR, "ERROR: execvp \"%s\" failed", *a->pa_argv);
233ee4910b1SJoseph Koshy 		/*NOTREACHED*/
234ee4910b1SJoseph Koshy 
235ee4910b1SJoseph Koshy 	default:	/* parent */
236ee4910b1SJoseph Koshy 		(void) close(pmcstat_sockpair[CHILDSOCKET]);
237ee4910b1SJoseph Koshy 		break;
238ee4910b1SJoseph Koshy 	}
239ee4910b1SJoseph Koshy 
240ee4910b1SJoseph Koshy 	/* Ask to be notified via a kevent when the target process exits. */
241fbd8b34aSJoseph Koshy 	EV_SET(&kev, pid, EVFILT_PROC, EV_ADD|EV_ONESHOT, NOTE_EXIT, 0,
242ee4910b1SJoseph Koshy 	    NULL);
243ee4910b1SJoseph Koshy 	if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
244fbd8b34aSJoseph Koshy 		err(EX_OSERR, "ERROR: cannot monitor child process %d", pid);
245fbd8b34aSJoseph Koshy 
246fbd8b34aSJoseph Koshy 	if ((pt = malloc(sizeof(*pt))) == NULL)
247fbd8b34aSJoseph Koshy 		errx(EX_SOFTWARE, "ERROR: Out of memory.");
248fbd8b34aSJoseph Koshy 
249fbd8b34aSJoseph Koshy 	pt->pt_pid = pid;
250fbd8b34aSJoseph Koshy 	SLIST_INSERT_HEAD(&a->pa_targets, pt, pt_next);
251ee4910b1SJoseph Koshy 
252ee4910b1SJoseph Koshy 	/* Wait for the child to signal that its ready to go. */
253ee4910b1SJoseph Koshy 	if (read(pmcstat_sockpair[PARENTSOCKET], &token, 1) < 0)
254ee4910b1SJoseph Koshy 		err(EX_OSERR, "ERROR (parent): cannot read token");
255ee4910b1SJoseph Koshy 
256ee4910b1SJoseph Koshy 	return;
257ee4910b1SJoseph Koshy }
258ee4910b1SJoseph Koshy 
259fbd8b34aSJoseph Koshy void
260fbd8b34aSJoseph Koshy pmcstat_find_targets(struct pmcstat_args *a, const char *spec)
261fbd8b34aSJoseph Koshy {
262fbd8b34aSJoseph Koshy 	int n, nproc, pid, rv;
263fbd8b34aSJoseph Koshy 	struct pmcstat_target *pt;
264fbd8b34aSJoseph Koshy 	char errbuf[_POSIX2_LINE_MAX], *end;
265fbd8b34aSJoseph Koshy 	static struct kinfo_proc *kp;
266fbd8b34aSJoseph Koshy 	regex_t reg;
267fbd8b34aSJoseph Koshy 	regmatch_t regmatch;
268fbd8b34aSJoseph Koshy 
269fbd8b34aSJoseph Koshy 	/* First check if we've been given a process id. */
270fbd8b34aSJoseph Koshy       	pid = strtol(spec, &end, 0);
271fbd8b34aSJoseph Koshy 	if (end != spec && pid >= 0) {
272fbd8b34aSJoseph Koshy 		if ((pt = malloc(sizeof(*pt))) == NULL)
273fbd8b34aSJoseph Koshy 			goto outofmemory;
274fbd8b34aSJoseph Koshy 		pt->pt_pid = pid;
275fbd8b34aSJoseph Koshy 		SLIST_INSERT_HEAD(&a->pa_targets, pt, pt_next);
276fbd8b34aSJoseph Koshy 		return;
277fbd8b34aSJoseph Koshy 	}
278fbd8b34aSJoseph Koshy 
279fbd8b34aSJoseph Koshy 	/* Otherwise treat arg as a regular expression naming processes. */
280fbd8b34aSJoseph Koshy 	if (pmcstat_kvm == NULL) {
281fbd8b34aSJoseph Koshy 		if ((pmcstat_kvm = kvm_openfiles(NULL, "/dev/null", NULL, 0,
282fbd8b34aSJoseph Koshy 		    errbuf)) == NULL)
283fbd8b34aSJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot open kernel \"%s\"",
284fbd8b34aSJoseph Koshy 			    errbuf);
285fbd8b34aSJoseph Koshy 		if ((pmcstat_plist = kvm_getprocs(pmcstat_kvm, KERN_PROC_PROC,
286fbd8b34aSJoseph Koshy 		    0, &nproc)) == NULL)
287fbd8b34aSJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot get process list: %s",
288fbd8b34aSJoseph Koshy 			    kvm_geterr(pmcstat_kvm));
289fbd8b34aSJoseph Koshy 	}
290fbd8b34aSJoseph Koshy 
291fbd8b34aSJoseph Koshy 	if ((rv = regcomp(&reg, spec, REG_EXTENDED|REG_NOSUB)) != 0) {
292fbd8b34aSJoseph Koshy 		regerror(rv, &reg, errbuf, sizeof(errbuf));
293fbd8b34aSJoseph Koshy 		err(EX_DATAERR, "ERROR: Failed to compile regex \"%s\": %s",
294fbd8b34aSJoseph Koshy 		    spec, errbuf);
295fbd8b34aSJoseph Koshy 	}
296fbd8b34aSJoseph Koshy 
297fbd8b34aSJoseph Koshy 	for (n = 0, kp = pmcstat_plist; n < nproc; n++, kp++) {
298fbd8b34aSJoseph Koshy 		if ((rv = regexec(&reg, kp->ki_comm, 1, &regmatch, 0)) == 0) {
299fbd8b34aSJoseph Koshy 			if ((pt = malloc(sizeof(*pt))) == NULL)
300fbd8b34aSJoseph Koshy 				goto outofmemory;
301fbd8b34aSJoseph Koshy 			pt->pt_pid = kp->ki_pid;
302fbd8b34aSJoseph Koshy 			SLIST_INSERT_HEAD(&a->pa_targets, pt, pt_next);
303fbd8b34aSJoseph Koshy 		} else if (rv != REG_NOMATCH) {
304fbd8b34aSJoseph Koshy 			regerror(rv, &reg, errbuf, sizeof(errbuf));
305fbd8b34aSJoseph Koshy 			errx(EX_SOFTWARE, "ERROR: Regex evalation failed: %s",
306fbd8b34aSJoseph Koshy 			    errbuf);
307fbd8b34aSJoseph Koshy 		}
308fbd8b34aSJoseph Koshy 	}
309fbd8b34aSJoseph Koshy 
310fbd8b34aSJoseph Koshy 	regfree(&reg);
311fbd8b34aSJoseph Koshy 
312fbd8b34aSJoseph Koshy 	return;
313fbd8b34aSJoseph Koshy 
314fbd8b34aSJoseph Koshy  outofmemory:
315fbd8b34aSJoseph Koshy 	errx(EX_SOFTWARE, "Out of memory.");
316fbd8b34aSJoseph Koshy 	/*NOTREACHED*/
317fbd8b34aSJoseph Koshy }
318fbd8b34aSJoseph Koshy 
319ee4910b1SJoseph Koshy uint32_t
320ee4910b1SJoseph Koshy pmcstat_get_cpumask(const char *cpuspec)
321ee4910b1SJoseph Koshy {
322ee4910b1SJoseph Koshy 	uint32_t cpumask;
323ee4910b1SJoseph Koshy 	int cpu;
324ee4910b1SJoseph Koshy 	const char *s;
325ee4910b1SJoseph Koshy 	char *end;
326ee4910b1SJoseph Koshy 
327ee4910b1SJoseph Koshy 	s = cpuspec;
328ee4910b1SJoseph Koshy 	cpumask = 0ULL;
329ee4910b1SJoseph Koshy 
330ee4910b1SJoseph Koshy 	do {
331ee4910b1SJoseph Koshy 		cpu = strtol(s, &end, 0);
332ee4910b1SJoseph Koshy 		if (cpu < 0 || end == s)
333ee4910b1SJoseph Koshy 			errx(EX_USAGE, "ERROR: Illegal CPU specification "
334ee4910b1SJoseph Koshy 			    "\"%s\".", cpuspec);
335ee4910b1SJoseph Koshy 		cpumask |= (1 << cpu);
336ee4910b1SJoseph Koshy 		s = end + strspn(end, ", \t");
337ee4910b1SJoseph Koshy 	} while (*s);
338ee4910b1SJoseph Koshy 
339ee4910b1SJoseph Koshy 	return (cpumask);
340ee4910b1SJoseph Koshy }
341ee4910b1SJoseph Koshy 
342ee4910b1SJoseph Koshy void
343fbd8b34aSJoseph Koshy pmcstat_kill_process(struct pmcstat_args *a)
344fbd8b34aSJoseph Koshy {
345fbd8b34aSJoseph Koshy 	struct pmcstat_target *pt;
346fbd8b34aSJoseph Koshy 
347fbd8b34aSJoseph Koshy 	assert(a->pa_flags & FLAG_HAS_COMMANDLINE);
348fbd8b34aSJoseph Koshy 
349fbd8b34aSJoseph Koshy 	/*
350fbd8b34aSJoseph Koshy 	 * If a command line was specified, it would be the very first
351fbd8b34aSJoseph Koshy 	 * in the list, before any other processes specified by -t.
352fbd8b34aSJoseph Koshy 	 */
353fbd8b34aSJoseph Koshy 	pt = SLIST_FIRST(&a->pa_targets);
354fbd8b34aSJoseph Koshy 	assert(pt != NULL);
355fbd8b34aSJoseph Koshy 
356fbd8b34aSJoseph Koshy 	if (kill(pt->pt_pid, SIGINT) != 0)
357fbd8b34aSJoseph Koshy 		err(EX_OSERR, "ERROR: cannot signal child process");
358fbd8b34aSJoseph Koshy }
359fbd8b34aSJoseph Koshy 
360fbd8b34aSJoseph Koshy void
361ebccf1e3SJoseph Koshy pmcstat_start_pmcs(struct pmcstat_args *a)
362ebccf1e3SJoseph Koshy {
363ebccf1e3SJoseph Koshy 	struct pmcstat_ev *ev;
364ebccf1e3SJoseph Koshy 
365fbd8b34aSJoseph Koshy 	STAILQ_FOREACH(ev, &args.pa_events, ev_next) {
366ebccf1e3SJoseph Koshy 
367ebccf1e3SJoseph Koshy 	    assert(ev->ev_pmcid != PMC_ID_INVALID);
368ebccf1e3SJoseph Koshy 
369ebccf1e3SJoseph Koshy 	    if (pmc_start(ev->ev_pmcid) < 0) {
370f263522aSJoseph Koshy 	        warn("ERROR: Cannot start pmc 0x%x \"%s\"",
371ebccf1e3SJoseph Koshy 		    ev->ev_pmcid, ev->ev_name);
372ebccf1e3SJoseph Koshy 		pmcstat_cleanup(a);
373f263522aSJoseph Koshy 		exit(EX_OSERR);
374ebccf1e3SJoseph Koshy 	    }
375ebccf1e3SJoseph Koshy 	}
376ebccf1e3SJoseph Koshy 
377ebccf1e3SJoseph Koshy }
378ebccf1e3SJoseph Koshy 
379ebccf1e3SJoseph Koshy void
380ebccf1e3SJoseph Koshy pmcstat_print_headers(struct pmcstat_args *a)
381ebccf1e3SJoseph Koshy {
382ebccf1e3SJoseph Koshy 	struct pmcstat_ev *ev;
383ee4910b1SJoseph Koshy 	int c, w;
384ebccf1e3SJoseph Koshy 
38515139246SJoseph Koshy 	(void) fprintf(a->pa_printfile, PRINT_HEADER_PREFIX);
386ebccf1e3SJoseph Koshy 
387fbd8b34aSJoseph Koshy 	STAILQ_FOREACH(ev, &a->pa_events, ev_next) {
388ebccf1e3SJoseph Koshy 		if (PMC_IS_SAMPLING_MODE(ev->ev_mode))
389ebccf1e3SJoseph Koshy 			continue;
390ebccf1e3SJoseph Koshy 
391ebccf1e3SJoseph Koshy 		c = PMC_IS_SYSTEM_MODE(ev->ev_mode) ? 's' : 'p';
392ebccf1e3SJoseph Koshy 
393ee4910b1SJoseph Koshy 		if (ev->ev_fieldskip != 0)
394ee4910b1SJoseph Koshy 			(void) fprintf(a->pa_printfile, "%*s",
395ee4910b1SJoseph Koshy 			    ev->ev_fieldskip, "");
396ee4910b1SJoseph Koshy 		w = ev->ev_fieldwidth - ev->ev_fieldskip - 2;
397ee4910b1SJoseph Koshy 
398ee4910b1SJoseph Koshy 		if (c == 's')
399ee4910b1SJoseph Koshy 			(void) fprintf(a->pa_printfile, "s/%02d/%-*s ",
400ee4910b1SJoseph Koshy 			    ev->ev_cpu, w-3, ev->ev_name);
401ee4910b1SJoseph Koshy 		else
402ee4910b1SJoseph Koshy 			(void) fprintf(a->pa_printfile, "p/%*s ", w,
403ebccf1e3SJoseph Koshy 			    ev->ev_name);
404ebccf1e3SJoseph Koshy 	}
405ebccf1e3SJoseph Koshy 
40615139246SJoseph Koshy 	(void) fflush(a->pa_printfile);
407ebccf1e3SJoseph Koshy }
408ebccf1e3SJoseph Koshy 
409ebccf1e3SJoseph Koshy void
410ebccf1e3SJoseph Koshy pmcstat_print_counters(struct pmcstat_args *a)
411ebccf1e3SJoseph Koshy {
412ebccf1e3SJoseph Koshy 	int extra_width;
413ebccf1e3SJoseph Koshy 	struct pmcstat_ev *ev;
414ebccf1e3SJoseph Koshy 	pmc_value_t value;
415ebccf1e3SJoseph Koshy 
416ebccf1e3SJoseph Koshy 	extra_width = sizeof(PRINT_HEADER_PREFIX) - 1;
417ebccf1e3SJoseph Koshy 
418fbd8b34aSJoseph Koshy 	STAILQ_FOREACH(ev, &a->pa_events, ev_next) {
419ebccf1e3SJoseph Koshy 
420ebccf1e3SJoseph Koshy 		/* skip sampling mode counters */
421ebccf1e3SJoseph Koshy 		if (PMC_IS_SAMPLING_MODE(ev->ev_mode))
422ebccf1e3SJoseph Koshy 			continue;
423ebccf1e3SJoseph Koshy 
424ebccf1e3SJoseph Koshy 		if (pmc_read(ev->ev_pmcid, &value) < 0)
425ebccf1e3SJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot read pmc "
426ebccf1e3SJoseph Koshy 			    "\"%s\"", ev->ev_name);
427ebccf1e3SJoseph Koshy 
42815139246SJoseph Koshy 		(void) fprintf(a->pa_printfile, "%*ju ",
42915139246SJoseph Koshy 		    ev->ev_fieldwidth + extra_width,
43015139246SJoseph Koshy 		    (uintmax_t) ev->ev_cumulative ? value :
43115139246SJoseph Koshy 		    (value - ev->ev_saved));
43215139246SJoseph Koshy 
433ebccf1e3SJoseph Koshy 		if (ev->ev_cumulative == 0)
434ebccf1e3SJoseph Koshy 			ev->ev_saved = value;
435ebccf1e3SJoseph Koshy 		extra_width = 0;
436ebccf1e3SJoseph Koshy 	}
437ebccf1e3SJoseph Koshy 
43815139246SJoseph Koshy 	(void) fflush(a->pa_printfile);
439ebccf1e3SJoseph Koshy }
440ebccf1e3SJoseph Koshy 
441ebccf1e3SJoseph Koshy /*
442ebccf1e3SJoseph Koshy  * Print output
443ebccf1e3SJoseph Koshy  */
444ebccf1e3SJoseph Koshy 
445ebccf1e3SJoseph Koshy void
446ebccf1e3SJoseph Koshy pmcstat_print_pmcs(struct pmcstat_args *a)
447ebccf1e3SJoseph Koshy {
448ebccf1e3SJoseph Koshy 	static int linecount = 0;
449ebccf1e3SJoseph Koshy 
45015139246SJoseph Koshy 	/* check if we need to print a header line */
451ebccf1e3SJoseph Koshy 	if (++linecount > pmcstat_displayheight) {
45215139246SJoseph Koshy 		(void) fprintf(a->pa_printfile, "\n");
453ebccf1e3SJoseph Koshy 		linecount = 1;
454ebccf1e3SJoseph Koshy 	}
455ebccf1e3SJoseph Koshy 	if (linecount == 1)
456ebccf1e3SJoseph Koshy 		pmcstat_print_headers(a);
45715139246SJoseph Koshy 	(void) fprintf(a->pa_printfile, "\n");
458ebccf1e3SJoseph Koshy 
459ebccf1e3SJoseph Koshy 	pmcstat_print_counters(a);
460ebccf1e3SJoseph Koshy 
461ebccf1e3SJoseph Koshy 	return;
462ebccf1e3SJoseph Koshy }
463ebccf1e3SJoseph Koshy 
464ebccf1e3SJoseph Koshy /*
465ebccf1e3SJoseph Koshy  * Do process profiling
466ebccf1e3SJoseph Koshy  *
467ebccf1e3SJoseph Koshy  * If a pid was specified, attach each allocated PMC to the target
468ebccf1e3SJoseph Koshy  * process.  Otherwise, fork a child and attach the PMCs to the child,
469ebccf1e3SJoseph Koshy  * and have the child exec() the target program.
470ebccf1e3SJoseph Koshy  */
471ebccf1e3SJoseph Koshy 
472ebccf1e3SJoseph Koshy void
473ee4910b1SJoseph Koshy pmcstat_start_process(void)
474ebccf1e3SJoseph Koshy {
475ee4910b1SJoseph Koshy 	/* Signal the child to proceed. */
476ee4910b1SJoseph Koshy 	if (write(pmcstat_sockpair[PARENTSOCKET], "!", 1) != 1)
477ee4910b1SJoseph Koshy 		err(EX_OSERR, "ERROR (parent): write of token failed");
478ebccf1e3SJoseph Koshy 
479ee4910b1SJoseph Koshy 	(void) close(pmcstat_sockpair[PARENTSOCKET]);
480ebccf1e3SJoseph Koshy }
481ebccf1e3SJoseph Koshy 
482ebccf1e3SJoseph Koshy void
483ebccf1e3SJoseph Koshy pmcstat_show_usage(void)
484ebccf1e3SJoseph Koshy {
485ebccf1e3SJoseph Koshy 	errx(EX_USAGE,
486ebccf1e3SJoseph Koshy 	    "[options] [commandline]\n"
487ebccf1e3SJoseph Koshy 	    "\t Measure process and/or system performance using hardware\n"
488ebccf1e3SJoseph Koshy 	    "\t performance monitoring counters.\n"
489ebccf1e3SJoseph Koshy 	    "\t Options include:\n"
490f263522aSJoseph Koshy 	    "\t -C\t\t (toggle) show cumulative counts\n"
491f263522aSJoseph Koshy 	    "\t -D path\t create profiles in directory \"path\"\n"
492f263522aSJoseph Koshy 	    "\t -E\t\t (toggle) show counts at process exit\n"
493b6010f9eSJoseph Koshy 	    "\t -G file\t write a system-wide callgraph to \"file\"\n"
49449874f6eSJoseph Koshy 	    "\t -M file\t print executable/gmon file map to \"file\"\n"
495b6010f9eSJoseph Koshy 	    "\t -N\t\t (toggle) capture callchains\n"
496f263522aSJoseph Koshy 	    "\t -O file\t send log output to \"file\"\n"
497f263522aSJoseph Koshy 	    "\t -P spec\t allocate a process-private sampling PMC\n"
498f263522aSJoseph Koshy 	    "\t -R file\t read events from \"file\"\n"
499f263522aSJoseph Koshy 	    "\t -S spec\t allocate a system-wide sampling PMC\n"
500f263522aSJoseph Koshy 	    "\t -W\t\t (toggle) show counts per context switch\n"
501ee4910b1SJoseph Koshy 	    "\t -c cpu-list\t set cpus for subsequent system-wide PMCs\n"
502f263522aSJoseph Koshy 	    "\t -d\t\t (toggle) track descendants\n"
503f263522aSJoseph Koshy 	    "\t -g\t\t produce gprof(1) compatible profiles\n"
50424bffe06SRuslan Ermilov 	    "\t -k dir\t\t set the path to the kernel\n"
505ebccf1e3SJoseph Koshy 	    "\t -n rate\t set sampling rate\n"
506ebccf1e3SJoseph Koshy 	    "\t -o file\t send print output to \"file\"\n"
507f263522aSJoseph Koshy 	    "\t -p spec\t allocate a process-private counting PMC\n"
50849874f6eSJoseph Koshy 	    "\t -q\t\t suppress verbosity\n"
50949874f6eSJoseph Koshy 	    "\t -r fsroot\t specify FS root directory\n"
510f263522aSJoseph Koshy 	    "\t -s spec\t allocate a system-wide counting PMC\n"
51139f4e0fcSJoseph Koshy 	    "\t -t pid\t\t attach to running process with pid \"pid\"\n"
51249874f6eSJoseph Koshy 	    "\t -v\t\t increase verbosity\n"
513b6010f9eSJoseph Koshy 	    "\t -w secs\t set printing time interval\n"
514b6010f9eSJoseph Koshy 	    "\t -z depth\t limit callchain display depth"
515ebccf1e3SJoseph Koshy 	);
516ebccf1e3SJoseph Koshy }
517ebccf1e3SJoseph Koshy 
518ebccf1e3SJoseph Koshy /*
519ebccf1e3SJoseph Koshy  * Main
520ebccf1e3SJoseph Koshy  */
521ebccf1e3SJoseph Koshy 
522ebccf1e3SJoseph Koshy int
523ebccf1e3SJoseph Koshy main(int argc, char **argv)
524ebccf1e3SJoseph Koshy {
525ebccf1e3SJoseph Koshy 	double interval;
526b6010f9eSJoseph Koshy 	int option, npmc, ncpu, haltedcpus;
5277fbf3da2SJoseph Koshy 	int c, check_driver_stats, current_cpu, current_sampling_count;
528b6010f9eSJoseph Koshy 	int do_callchain, do_descendants, do_logproccsw, do_logprocexit;
529b6010f9eSJoseph Koshy 	int do_print;
530ee4910b1SJoseph Koshy 	size_t dummy;
531b6010f9eSJoseph Koshy 	int graphdepth;
532f263522aSJoseph Koshy 	int pipefd[2];
533f263522aSJoseph Koshy 	int use_cumulative_counts;
534ee4910b1SJoseph Koshy 	uint32_t cpumask;
53549874f6eSJoseph Koshy 	char *end, *tmp;
536b6010f9eSJoseph Koshy 	const char *errmsg, *graphfilename;
537f263522aSJoseph Koshy 	enum pmcstat_state runstate;
5387fbf3da2SJoseph Koshy 	struct pmc_driverstats ds_start, ds_end;
539ebccf1e3SJoseph Koshy 	struct pmcstat_ev *ev;
540ebccf1e3SJoseph Koshy 	struct sigaction sa;
541ebccf1e3SJoseph Koshy 	struct kevent kev;
542ebccf1e3SJoseph Koshy 	struct winsize ws;
54315139246SJoseph Koshy 	struct stat sb;
54449874f6eSJoseph Koshy 	char buffer[PATH_MAX];
545ebccf1e3SJoseph Koshy 
5467fbf3da2SJoseph Koshy 	check_driver_stats      = 0;
547ebccf1e3SJoseph Koshy 	current_cpu 		= 0;
548ebccf1e3SJoseph Koshy 	current_sampling_count  = DEFAULT_SAMPLE_COUNT;
549b6010f9eSJoseph Koshy 	do_callchain		= 1;
550ebccf1e3SJoseph Koshy 	do_descendants          = 0;
551f263522aSJoseph Koshy 	do_logproccsw           = 0;
552f263522aSJoseph Koshy 	do_logprocexit          = 0;
553ebccf1e3SJoseph Koshy 	use_cumulative_counts   = 0;
554b6010f9eSJoseph Koshy 	graphfilename		= "-";
555f263522aSJoseph Koshy 	args.pa_required	= 0;
556ebccf1e3SJoseph Koshy 	args.pa_flags		= 0;
55749874f6eSJoseph Koshy 	args.pa_verbosity	= 1;
55815139246SJoseph Koshy 	args.pa_logfd		= -1;
55949874f6eSJoseph Koshy 	args.pa_fsroot		= "";
56049874f6eSJoseph Koshy 	args.pa_kernel		= strdup("/boot/kernel");
56115139246SJoseph Koshy 	args.pa_samplesdir	= ".";
56215139246SJoseph Koshy 	args.pa_printfile	= stderr;
563b6010f9eSJoseph Koshy 	args.pa_graphdepth	= DEFAULT_CALLGRAPH_DEPTH;
564b6010f9eSJoseph Koshy 	args.pa_graphfile	= NULL;
565ebccf1e3SJoseph Koshy 	args.pa_interval	= DEFAULT_WAIT_INTERVAL;
56649874f6eSJoseph Koshy 	args.pa_mapfilename	= NULL;
567b6010f9eSJoseph Koshy 	args.pa_inputpath	= NULL;
568b6010f9eSJoseph Koshy 	args.pa_outputpath	= NULL;
569fbd8b34aSJoseph Koshy 	STAILQ_INIT(&args.pa_events);
570fbd8b34aSJoseph Koshy 	SLIST_INIT(&args.pa_targets);
571bea6aa93SJoseph Koshy 	bzero(&ds_start, sizeof(ds_start));
572bea6aa93SJoseph Koshy 	bzero(&ds_end, sizeof(ds_end));
573ebccf1e3SJoseph Koshy 	ev = NULL;
574ebccf1e3SJoseph Koshy 
575b6010f9eSJoseph Koshy 	/*
576b6010f9eSJoseph Koshy 	 * The initial CPU mask specifies all non-halted CPUS in the
577b6010f9eSJoseph Koshy 	 * system.
578b6010f9eSJoseph Koshy 	 */
579b6010f9eSJoseph Koshy 	dummy = sizeof(int);
580ee4910b1SJoseph Koshy 	if (sysctlbyname("hw.ncpu", &ncpu, &dummy, NULL, 0) < 0)
581b6010f9eSJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot determine the number of CPUs");
582ee4910b1SJoseph Koshy 	cpumask = (1 << ncpu) - 1;
583bcba261aSJung-uk Kim 	if (ncpu > 1) {
584b6010f9eSJoseph Koshy 		if (sysctlbyname("machdep.hlt_cpus", &haltedcpus, &dummy,
585b6010f9eSJoseph Koshy 		    NULL, 0) < 0)
586bcba261aSJung-uk Kim 			err(EX_OSERR, "ERROR: Cannot determine which CPUs are "
587bcba261aSJung-uk Kim 			    "halted");
588b6010f9eSJoseph Koshy 		cpumask &= ~haltedcpus;
589bcba261aSJung-uk Kim 	}
590ee4910b1SJoseph Koshy 
59149874f6eSJoseph Koshy 	while ((option = getopt(argc, argv,
592b6010f9eSJoseph Koshy 	    "CD:EG:M:NO:P:R:S:Wc:dgk:n:o:p:qr:s:t:vw:z:")) != -1)
593ebccf1e3SJoseph Koshy 		switch (option) {
594ebccf1e3SJoseph Koshy 		case 'C':	/* cumulative values */
595ebccf1e3SJoseph Koshy 			use_cumulative_counts = !use_cumulative_counts;
596f263522aSJoseph Koshy 			args.pa_required |= FLAG_HAS_COUNTING_PMCS;
597ebccf1e3SJoseph Koshy 			break;
598ebccf1e3SJoseph Koshy 
599ebccf1e3SJoseph Koshy 		case 'c':	/* CPU */
600ee4910b1SJoseph Koshy 
601ee4910b1SJoseph Koshy 			if (optarg[0] == '*' && optarg[1] == '\0')
602ee4910b1SJoseph Koshy 				cpumask = (1 << ncpu) - 1;
603ee4910b1SJoseph Koshy 			else
604ee4910b1SJoseph Koshy 				cpumask = pmcstat_get_cpumask(optarg);
605ee4910b1SJoseph Koshy 
606f263522aSJoseph Koshy 			args.pa_required |= FLAG_HAS_SYSTEM_PMCS;
607ebccf1e3SJoseph Koshy 			break;
608ebccf1e3SJoseph Koshy 
60915139246SJoseph Koshy 		case 'D':
61015139246SJoseph Koshy 			if (stat(optarg, &sb) < 0)
61115139246SJoseph Koshy 				err(EX_OSERR, "ERROR: Cannot stat \"%s\"",
61215139246SJoseph Koshy 				    optarg);
61315139246SJoseph Koshy 			if (!S_ISDIR(sb.st_mode))
61415139246SJoseph Koshy 				errx(EX_USAGE, "ERROR: \"%s\" is not a "
615ee4910b1SJoseph Koshy 				    "directory.", optarg);
61615139246SJoseph Koshy 			args.pa_samplesdir = optarg;
61715139246SJoseph Koshy 			args.pa_flags     |= FLAG_HAS_SAMPLESDIR;
61815139246SJoseph Koshy 			args.pa_required  |= FLAG_DO_GPROF;
61915139246SJoseph Koshy 			break;
62015139246SJoseph Koshy 
621ebccf1e3SJoseph Koshy 		case 'd':	/* toggle descendents */
622ebccf1e3SJoseph Koshy 			do_descendants = !do_descendants;
623f263522aSJoseph Koshy 			args.pa_required |= FLAG_HAS_PROCESS_PMCS;
624f263522aSJoseph Koshy 			break;
625f263522aSJoseph Koshy 
626b6010f9eSJoseph Koshy 		case 'G':	/* produce a system-wide callgraph */
627b6010f9eSJoseph Koshy 			args.pa_flags |= FLAG_DO_CALLGRAPHS;
628b6010f9eSJoseph Koshy 			graphfilename = optarg;
629b6010f9eSJoseph Koshy 			break;
630b6010f9eSJoseph Koshy 
631f263522aSJoseph Koshy 		case 'g':	/* produce gprof compatible profiles */
632f263522aSJoseph Koshy 			args.pa_flags |= FLAG_DO_GPROF;
633f263522aSJoseph Koshy 			break;
634f263522aSJoseph Koshy 
63515139246SJoseph Koshy 		case 'k':	/* pathname to the kernel */
63649874f6eSJoseph Koshy 			free(args.pa_kernel);
63749874f6eSJoseph Koshy 			args.pa_kernel = strdup(optarg);
638b6010f9eSJoseph Koshy 			args.pa_required |= FLAG_DO_ANALYSIS;
63915139246SJoseph Koshy 			args.pa_flags    |= FLAG_HAS_KERNELPATH;
640f263522aSJoseph Koshy 			break;
641f263522aSJoseph Koshy 
642f263522aSJoseph Koshy 		case 'E':	/* log process exit */
643f263522aSJoseph Koshy 			do_logprocexit = !do_logprocexit;
644f263522aSJoseph Koshy 			args.pa_required |= (FLAG_HAS_PROCESS_PMCS |
64515139246SJoseph Koshy 			    FLAG_HAS_COUNTING_PMCS | FLAG_HAS_OUTPUT_LOGFILE);
646ebccf1e3SJoseph Koshy 			break;
647ebccf1e3SJoseph Koshy 
64849874f6eSJoseph Koshy 		case 'M':	/* mapfile */
64949874f6eSJoseph Koshy 			args.pa_mapfilename = optarg;
65049874f6eSJoseph Koshy 			break;
65149874f6eSJoseph Koshy 
652b6010f9eSJoseph Koshy 		case 'N':
653b6010f9eSJoseph Koshy 			do_callchain = !do_callchain;
654b6010f9eSJoseph Koshy 			args.pa_required |= FLAG_HAS_SAMPLING_PMCS;
655b6010f9eSJoseph Koshy 			break;
656b6010f9eSJoseph Koshy 
657ebccf1e3SJoseph Koshy 		case 'p':	/* process virtual counting PMC */
658ebccf1e3SJoseph Koshy 		case 's':	/* system-wide counting PMC */
659ebccf1e3SJoseph Koshy 		case 'P':	/* process virtual sampling PMC */
660ebccf1e3SJoseph Koshy 		case 'S':	/* system-wide sampling PMC */
661ebccf1e3SJoseph Koshy 			if ((ev = malloc(sizeof(*ev))) == NULL)
662f263522aSJoseph Koshy 				errx(EX_SOFTWARE, "ERROR: Out of memory.");
663ebccf1e3SJoseph Koshy 
664ebccf1e3SJoseph Koshy 			switch (option) {
665ebccf1e3SJoseph Koshy 			case 'p': ev->ev_mode = PMC_MODE_TC; break;
666ebccf1e3SJoseph Koshy 			case 's': ev->ev_mode = PMC_MODE_SC; break;
667ebccf1e3SJoseph Koshy 			case 'P': ev->ev_mode = PMC_MODE_TS; break;
668ebccf1e3SJoseph Koshy 			case 'S': ev->ev_mode = PMC_MODE_SS; break;
669ebccf1e3SJoseph Koshy 			}
670ebccf1e3SJoseph Koshy 
671f263522aSJoseph Koshy 			if (option == 'P' || option == 'p') {
672f263522aSJoseph Koshy 				args.pa_flags |= FLAG_HAS_PROCESS_PMCS;
67315139246SJoseph Koshy 				args.pa_required |= (FLAG_HAS_COMMANDLINE |
674fbd8b34aSJoseph Koshy 				    FLAG_HAS_TARGET);
675f263522aSJoseph Koshy 			}
676ebccf1e3SJoseph Koshy 
677f263522aSJoseph Koshy 			if (option == 'P' || option == 'S') {
678f263522aSJoseph Koshy 				args.pa_flags |= FLAG_HAS_SAMPLING_PMCS;
67915139246SJoseph Koshy 				args.pa_required |= (FLAG_HAS_PIPE |
68015139246SJoseph Koshy 				    FLAG_HAS_OUTPUT_LOGFILE);
681f263522aSJoseph Koshy 			}
682ebccf1e3SJoseph Koshy 
683ebccf1e3SJoseph Koshy 			if (option == 'p' || option == 's')
684f263522aSJoseph Koshy 				args.pa_flags |= FLAG_HAS_COUNTING_PMCS;
685f263522aSJoseph Koshy 
686f263522aSJoseph Koshy 			if (option == 's' || option == 'S')
687f263522aSJoseph Koshy 				args.pa_flags |= FLAG_HAS_SYSTEM_PMCS;
688ebccf1e3SJoseph Koshy 
689ebccf1e3SJoseph Koshy 			ev->ev_spec  = strdup(optarg);
690ebccf1e3SJoseph Koshy 
691ebccf1e3SJoseph Koshy 			if (option == 'S' || option == 'P')
692ebccf1e3SJoseph Koshy 				ev->ev_count = current_sampling_count;
693ebccf1e3SJoseph Koshy 			else
694ebccf1e3SJoseph Koshy 				ev->ev_count = -1;
695ebccf1e3SJoseph Koshy 
696ebccf1e3SJoseph Koshy 			if (option == 'S' || option == 's')
697ee4910b1SJoseph Koshy 				ev->ev_cpu = ffs(cpumask) - 1;
698ebccf1e3SJoseph Koshy 			else
699ebccf1e3SJoseph Koshy 				ev->ev_cpu = PMC_CPU_ANY;
700ebccf1e3SJoseph Koshy 
701f263522aSJoseph Koshy 			ev->ev_flags = 0;
702b6010f9eSJoseph Koshy 			if (do_callchain)
703b6010f9eSJoseph Koshy 				ev->ev_flags |= PMC_F_CALLCHAIN;
704f263522aSJoseph Koshy 			if (do_descendants)
705f263522aSJoseph Koshy 				ev->ev_flags |= PMC_F_DESCENDANTS;
706f263522aSJoseph Koshy 			if (do_logprocexit)
707f263522aSJoseph Koshy 				ev->ev_flags |= PMC_F_LOG_PROCEXIT;
708f263522aSJoseph Koshy 			if (do_logproccsw)
709f263522aSJoseph Koshy 				ev->ev_flags |= PMC_F_LOG_PROCCSW;
710f263522aSJoseph Koshy 
711ebccf1e3SJoseph Koshy 			ev->ev_cumulative  = use_cumulative_counts;
712ebccf1e3SJoseph Koshy 
713ebccf1e3SJoseph Koshy 			ev->ev_saved = 0LL;
714ebccf1e3SJoseph Koshy 			ev->ev_pmcid = PMC_ID_INVALID;
715ebccf1e3SJoseph Koshy 
716ebccf1e3SJoseph Koshy 			/* extract event name */
717ebccf1e3SJoseph Koshy 			c = strcspn(optarg, ", \t");
718ebccf1e3SJoseph Koshy 			ev->ev_name = malloc(c + 1);
719ebccf1e3SJoseph Koshy 			(void) strncpy(ev->ev_name, optarg, c);
720ebccf1e3SJoseph Koshy 			*(ev->ev_name + c) = '\0';
721ebccf1e3SJoseph Koshy 
722fbd8b34aSJoseph Koshy 			STAILQ_INSERT_TAIL(&args.pa_events, ev, ev_next);
723ebccf1e3SJoseph Koshy 
724ee4910b1SJoseph Koshy 			if (option == 's' || option == 'S')
725ee4910b1SJoseph Koshy 				pmcstat_clone_event_descriptor(&args, ev,
726ee4910b1SJoseph Koshy 				    cpumask & ~(1 << ev->ev_cpu));
727ee4910b1SJoseph Koshy 
728ebccf1e3SJoseph Koshy 			break;
729ebccf1e3SJoseph Koshy 
730ebccf1e3SJoseph Koshy 		case 'n':	/* sampling count */
731ebccf1e3SJoseph Koshy 			current_sampling_count = strtol(optarg, &end, 0);
732ebccf1e3SJoseph Koshy 			if (*end != '\0' || current_sampling_count <= 0)
733ebccf1e3SJoseph Koshy 				errx(EX_USAGE,
734f263522aSJoseph Koshy 				    "ERROR: Illegal count value \"%s\".",
735ebccf1e3SJoseph Koshy 				    optarg);
736f263522aSJoseph Koshy 			args.pa_required |= FLAG_HAS_SAMPLING_PMCS;
737ebccf1e3SJoseph Koshy 			break;
738ebccf1e3SJoseph Koshy 
739ebccf1e3SJoseph Koshy 		case 'o':	/* outputfile */
74015139246SJoseph Koshy 			if (args.pa_printfile != NULL)
74115139246SJoseph Koshy 				(void) fclose(args.pa_printfile);
74215139246SJoseph Koshy 			if ((args.pa_printfile = fopen(optarg, "w")) == NULL)
743ebccf1e3SJoseph Koshy 				errx(EX_OSERR, "ERROR: cannot open \"%s\" for "
744f263522aSJoseph Koshy 				    "writing.", optarg);
74515139246SJoseph Koshy 			args.pa_flags |= FLAG_DO_PRINT;
746f263522aSJoseph Koshy 			break;
747ebccf1e3SJoseph Koshy 
748ebccf1e3SJoseph Koshy 		case 'O':	/* sampling output */
74915139246SJoseph Koshy 			if (args.pa_outputpath)
75015139246SJoseph Koshy 				errx(EX_USAGE, "ERROR: option -O may only be "
751f263522aSJoseph Koshy 				    "specified once.");
75215139246SJoseph Koshy 			args.pa_outputpath = optarg;
75315139246SJoseph Koshy 			args.pa_flags |= FLAG_HAS_OUTPUT_LOGFILE;
75415139246SJoseph Koshy 			break;
75515139246SJoseph Koshy 
75649874f6eSJoseph Koshy 		case 'q':	/* quiet mode */
75749874f6eSJoseph Koshy 			args.pa_verbosity = 0;
75849874f6eSJoseph Koshy 			break;
75949874f6eSJoseph Koshy 
76049874f6eSJoseph Koshy 		case 'r':	/* root FS path */
76149874f6eSJoseph Koshy 			args.pa_fsroot = optarg;
76249874f6eSJoseph Koshy 			break;
76349874f6eSJoseph Koshy 
76415139246SJoseph Koshy 		case 'R':	/* read an existing log file */
765b6010f9eSJoseph Koshy 			if (args.pa_inputpath != NULL)
76615139246SJoseph Koshy 				errx(EX_USAGE, "ERROR: option -R may only be "
76715139246SJoseph Koshy 				    "specified once.");
76815139246SJoseph Koshy 			args.pa_inputpath = optarg;
76915139246SJoseph Koshy 			if (args.pa_printfile == stderr)
77015139246SJoseph Koshy 				args.pa_printfile = stdout;
77115139246SJoseph Koshy 			args.pa_flags |= FLAG_READ_LOGFILE;
772ebccf1e3SJoseph Koshy 			break;
773ebccf1e3SJoseph Koshy 
774fbd8b34aSJoseph Koshy 		case 't':	/* target pid or process name */
775fbd8b34aSJoseph Koshy 			pmcstat_find_targets(&args, optarg);
776ebccf1e3SJoseph Koshy 
777fbd8b34aSJoseph Koshy 			args.pa_flags |= FLAG_HAS_TARGET;
778f263522aSJoseph Koshy 			args.pa_required |= FLAG_HAS_PROCESS_PMCS;
779ebccf1e3SJoseph Koshy 			break;
780ebccf1e3SJoseph Koshy 
78149874f6eSJoseph Koshy 		case 'v':	/* verbose */
78249874f6eSJoseph Koshy 			args.pa_verbosity++;
78349874f6eSJoseph Koshy 			break;
78449874f6eSJoseph Koshy 
785ebccf1e3SJoseph Koshy 		case 'w':	/* wait interval */
786ebccf1e3SJoseph Koshy 			interval = strtod(optarg, &end);
787ebccf1e3SJoseph Koshy 			if (*end != '\0' || interval <= 0)
788ebccf1e3SJoseph Koshy 				errx(EX_USAGE, "ERROR: Illegal wait interval "
789f263522aSJoseph Koshy 				    "value \"%s\".", optarg);
790ebccf1e3SJoseph Koshy 			args.pa_flags |= FLAG_HAS_WAIT_INTERVAL;
79115139246SJoseph Koshy 			args.pa_required |= FLAG_HAS_COUNTING_PMCS;
792ebccf1e3SJoseph Koshy 			args.pa_interval = interval;
793f263522aSJoseph Koshy 			break;
794ebccf1e3SJoseph Koshy 
795f263522aSJoseph Koshy 		case 'W':	/* toggle LOG_CSW */
796f263522aSJoseph Koshy 			do_logproccsw = !do_logproccsw;
797f263522aSJoseph Koshy 			args.pa_required |= (FLAG_HAS_PROCESS_PMCS |
79815139246SJoseph Koshy 			    FLAG_HAS_COUNTING_PMCS | FLAG_HAS_OUTPUT_LOGFILE);
799ebccf1e3SJoseph Koshy 			break;
800ebccf1e3SJoseph Koshy 
801b6010f9eSJoseph Koshy 		case 'z':
802b6010f9eSJoseph Koshy 			graphdepth = strtod(optarg, &end);
803b6010f9eSJoseph Koshy 			if (*end != '\0' || graphdepth <= 0)
804b6010f9eSJoseph Koshy 				errx(EX_USAGE, "ERROR: Illegal callchain "
805b6010f9eSJoseph Koshy 				    "depth \"%s\".", optarg);
806b6010f9eSJoseph Koshy 			args.pa_graphdepth = graphdepth;
807b6010f9eSJoseph Koshy 			args.pa_required |= FLAG_DO_CALLGRAPHS;
808b6010f9eSJoseph Koshy 			break;
809b6010f9eSJoseph Koshy 
810ebccf1e3SJoseph Koshy 		case '?':
811ebccf1e3SJoseph Koshy 		default:
812ebccf1e3SJoseph Koshy 			pmcstat_show_usage();
813ebccf1e3SJoseph Koshy 			break;
814ebccf1e3SJoseph Koshy 
815ebccf1e3SJoseph Koshy 		}
816ebccf1e3SJoseph Koshy 
817ebccf1e3SJoseph Koshy 	args.pa_argc = (argc -= optind);
818ebccf1e3SJoseph Koshy 	args.pa_argv = (argv += optind);
819ebccf1e3SJoseph Koshy 
820b6010f9eSJoseph Koshy 	args.pa_cpumask = cpumask; /* For selecting CPUs using -R. */
821b6010f9eSJoseph Koshy 
82215139246SJoseph Koshy 	if (argc)	/* command line present */
82315139246SJoseph Koshy 		args.pa_flags |= FLAG_HAS_COMMANDLINE;
824ebccf1e3SJoseph Koshy 
825b6010f9eSJoseph Koshy 	if (args.pa_flags & (FLAG_DO_GPROF | FLAG_DO_CALLGRAPHS))
826b6010f9eSJoseph Koshy 		args.pa_flags |= FLAG_DO_ANALYSIS;
827b6010f9eSJoseph Koshy 
828ebccf1e3SJoseph Koshy 	/*
829ebccf1e3SJoseph Koshy 	 * Check invocation syntax.
830ebccf1e3SJoseph Koshy 	 */
831ebccf1e3SJoseph Koshy 
83215139246SJoseph Koshy 	/* disallow -O and -R together */
83315139246SJoseph Koshy 	if (args.pa_outputpath && args.pa_inputpath)
83415139246SJoseph Koshy 		errx(EX_USAGE, "ERROR: options -O and -R are mutually "
83515139246SJoseph Koshy 		    "exclusive.");
83615139246SJoseph Koshy 
83715139246SJoseph Koshy 	if (args.pa_flags & FLAG_READ_LOGFILE) {
838f263522aSJoseph Koshy 		errmsg = NULL;
83915139246SJoseph Koshy 		if (args.pa_flags & FLAG_HAS_COMMANDLINE)
840f263522aSJoseph Koshy 			errmsg = "a command line specification";
841fbd8b34aSJoseph Koshy 		else if (args.pa_flags & FLAG_HAS_TARGET)
842f263522aSJoseph Koshy 			errmsg = "option -t";
843fbd8b34aSJoseph Koshy 		else if (!STAILQ_EMPTY(&args.pa_events))
844f263522aSJoseph Koshy 			errmsg = "a PMC event specification";
845f263522aSJoseph Koshy 		if (errmsg)
846f263522aSJoseph Koshy 			errx(EX_USAGE, "ERROR: option -R may not be used with "
847f263522aSJoseph Koshy 			    "%s.", errmsg);
848fbd8b34aSJoseph Koshy 	} else if (STAILQ_EMPTY(&args.pa_events))
84915139246SJoseph Koshy 		/* All other uses require a PMC spec. */
850ebccf1e3SJoseph Koshy 		pmcstat_show_usage();
851ebccf1e3SJoseph Koshy 
852f263522aSJoseph Koshy 	/* check for -t pid without a process PMC spec */
853fbd8b34aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_TARGET) &&
854f263522aSJoseph Koshy 	    (args.pa_flags & FLAG_HAS_PROCESS_PMCS) == 0)
855f263522aSJoseph Koshy 		errx(EX_USAGE, "ERROR: option -t requires a process mode PMC "
856f263522aSJoseph Koshy 		    "to be specified.");
857f263522aSJoseph Koshy 
858f263522aSJoseph Koshy 	/* check for process-mode options without a command or -t pid */
859f263522aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_PROCESS_PMCS) &&
860fbd8b34aSJoseph Koshy 	    (args.pa_flags & (FLAG_HAS_COMMANDLINE | FLAG_HAS_TARGET)) == 0)
86115139246SJoseph Koshy 		errx(EX_USAGE, "ERROR: options -d, -E, -p, -P, and -W require "
86215139246SJoseph Koshy 		    "a command line or target process.");
863f263522aSJoseph Koshy 
864f263522aSJoseph Koshy 	/* check for -p | -P without a target process of some sort */
865fbd8b34aSJoseph Koshy 	if ((args.pa_required & (FLAG_HAS_COMMANDLINE | FLAG_HAS_TARGET)) &&
866fbd8b34aSJoseph Koshy 	    (args.pa_flags & (FLAG_HAS_COMMANDLINE | FLAG_HAS_TARGET)) == 0)
86715139246SJoseph Koshy 		errx(EX_USAGE, "ERROR: options -P and -p require a "
868f263522aSJoseph Koshy 		    "target process or a command line.");
869f263522aSJoseph Koshy 
870f263522aSJoseph Koshy 	/* check for process-mode options without a process-mode PMC */
871f263522aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_PROCESS_PMCS) &&
872f263522aSJoseph Koshy 	    (args.pa_flags & FLAG_HAS_PROCESS_PMCS) == 0)
87315139246SJoseph Koshy 		errx(EX_USAGE, "ERROR: options -d, -E, and -W require a "
874f263522aSJoseph Koshy 		    "process mode PMC to be specified.");
875f263522aSJoseph Koshy 
876b6010f9eSJoseph Koshy 	/* check for -c cpu with no system mode PMCs or logfile. */
877f263522aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_SYSTEM_PMCS) &&
878b6010f9eSJoseph Koshy 	    (args.pa_flags & FLAG_HAS_SYSTEM_PMCS) == 0 &&
879b6010f9eSJoseph Koshy 	    (args.pa_flags & FLAG_READ_LOGFILE) == 0)
880f263522aSJoseph Koshy 		errx(EX_USAGE, "ERROR: option -c requires at least one "
881f263522aSJoseph Koshy 		    "system mode PMC to be specified.");
882f263522aSJoseph Koshy 
883f263522aSJoseph Koshy 	/* check for counting mode options without a counting PMC */
884f263522aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_COUNTING_PMCS) &&
885f263522aSJoseph Koshy 	    (args.pa_flags & FLAG_HAS_COUNTING_PMCS) == 0)
8864525db8cSJoseph Koshy 		errx(EX_USAGE, "ERROR: options -C, -W, -o and -w require at "
8874525db8cSJoseph Koshy 		    "least one counting mode PMC to be specified.");
888f263522aSJoseph Koshy 
889f263522aSJoseph Koshy 	/* check for sampling mode options without a sampling PMC spec */
890f263522aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_SAMPLING_PMCS) &&
891f263522aSJoseph Koshy 	    (args.pa_flags & FLAG_HAS_SAMPLING_PMCS) == 0)
892b6010f9eSJoseph Koshy 		errx(EX_USAGE, "ERROR: options -N, -n and -O require at "
893b6010f9eSJoseph Koshy 		    "least one sampling mode PMC to be specified.");
894f263522aSJoseph Koshy 
895b6010f9eSJoseph Koshy 	/* check if -g/-G are being used correctly */
896b6010f9eSJoseph Koshy 	if ((args.pa_flags & FLAG_DO_ANALYSIS) &&
89715139246SJoseph Koshy 	    !(args.pa_flags & (FLAG_HAS_SAMPLING_PMCS|FLAG_READ_LOGFILE)))
898b6010f9eSJoseph Koshy 		errx(EX_USAGE, "ERROR: options -g/-G require sampling PMCs "
899b6010f9eSJoseph Koshy 		    "or -R to be specified.");
90015139246SJoseph Koshy 
901f263522aSJoseph Koshy 	/* check if -O was spuriously specified */
90215139246SJoseph Koshy 	if ((args.pa_flags & FLAG_HAS_OUTPUT_LOGFILE) &&
90315139246SJoseph Koshy 	    (args.pa_required & FLAG_HAS_OUTPUT_LOGFILE) == 0)
904f263522aSJoseph Koshy 		errx(EX_USAGE,
905f263522aSJoseph Koshy 		    "ERROR: option -O is used only with options "
906f263522aSJoseph Koshy 		    "-E, -P, -S and -W.");
907f263522aSJoseph Koshy 
908b6010f9eSJoseph Koshy 	/* -k kernel path require -g/-G or -R */
90915139246SJoseph Koshy 	if ((args.pa_flags & FLAG_HAS_KERNELPATH) &&
910b6010f9eSJoseph Koshy 	    (args.pa_flags & FLAG_DO_ANALYSIS) == 0 &&
91149874f6eSJoseph Koshy 	    (args.pa_flags & FLAG_READ_LOGFILE) == 0)
91249874f6eSJoseph Koshy 	    errx(EX_USAGE, "ERROR: option -k is only used with -g/-R.");
91315139246SJoseph Koshy 
914b6010f9eSJoseph Koshy 	/* -D only applies to gprof output mode (-g) */
91515139246SJoseph Koshy 	if ((args.pa_flags & FLAG_HAS_SAMPLESDIR) &&
916b6010f9eSJoseph Koshy 	    (args.pa_flags & FLAG_DO_GPROF) == 0)
917b6010f9eSJoseph Koshy 	    errx(EX_USAGE, "ERROR: option -D is only used with -g.");
91849874f6eSJoseph Koshy 
91949874f6eSJoseph Koshy 	/* -M mapfile requires -g or -R */
92049874f6eSJoseph Koshy 	if (args.pa_mapfilename != NULL &&
92149874f6eSJoseph Koshy 	    (args.pa_flags & FLAG_DO_GPROF) == 0 &&
92249874f6eSJoseph Koshy 	    (args.pa_flags & FLAG_READ_LOGFILE) == 0)
92349874f6eSJoseph Koshy 	    errx(EX_USAGE, "ERROR: option -M is only used with -g/-R.");
92415139246SJoseph Koshy 
92515139246SJoseph Koshy 	/*
92615139246SJoseph Koshy 	 * Disallow textual output of sampling PMCs if counting PMCs
92715139246SJoseph Koshy 	 * have also been asked for, mostly because the combined output
92815139246SJoseph Koshy 	 * is difficult to make sense of.
92915139246SJoseph Koshy 	 */
93015139246SJoseph Koshy 	if ((args.pa_flags & FLAG_HAS_COUNTING_PMCS) &&
93115139246SJoseph Koshy 	    (args.pa_flags & FLAG_HAS_SAMPLING_PMCS) &&
93200a22dc5SJoseph Koshy 	    ((args.pa_flags & FLAG_HAS_OUTPUT_LOGFILE) == 0))
93315139246SJoseph Koshy 		errx(EX_USAGE, "ERROR: option -O is required if counting and "
93415139246SJoseph Koshy 		    "sampling PMCs are specified together.");
93515139246SJoseph Koshy 
93649874f6eSJoseph Koshy 	/*
937b6010f9eSJoseph Koshy 	 * Check if "-k kerneldir" was specified, and if whether
938b6010f9eSJoseph Koshy 	 * 'kerneldir' actually refers to a a file.  If so, use
939b6010f9eSJoseph Koshy 	 * `dirname path` to determine the kernel directory.
94049874f6eSJoseph Koshy 	 */
94149874f6eSJoseph Koshy 	if (args.pa_flags & FLAG_HAS_KERNELPATH) {
94249874f6eSJoseph Koshy 		(void) snprintf(buffer, sizeof(buffer), "%s%s", args.pa_fsroot,
94349874f6eSJoseph Koshy 		    args.pa_kernel);
94449874f6eSJoseph Koshy 		if (stat(buffer, &sb) < 0)
94549874f6eSJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot locate kernel \"%s\"",
94649874f6eSJoseph Koshy 			    buffer);
94749874f6eSJoseph Koshy 		if (!S_ISREG(sb.st_mode) && !S_ISDIR(sb.st_mode))
94849874f6eSJoseph Koshy 			errx(EX_USAGE, "ERROR: \"%s\": Unsupported file type.",
94949874f6eSJoseph Koshy 			    buffer);
95049874f6eSJoseph Koshy 		if (!S_ISDIR(sb.st_mode)) {
95149874f6eSJoseph Koshy 			tmp = args.pa_kernel;
95249874f6eSJoseph Koshy 			args.pa_kernel = strdup(dirname(args.pa_kernel));
95349874f6eSJoseph Koshy 			free(tmp);
95449874f6eSJoseph Koshy 			(void) snprintf(buffer, sizeof(buffer), "%s%s",
95549874f6eSJoseph Koshy 			    args.pa_fsroot, args.pa_kernel);
95649874f6eSJoseph Koshy 			if (stat(buffer, &sb) < 0)
95749874f6eSJoseph Koshy 				err(EX_OSERR, "ERROR: Cannot stat \"%s\"",
95849874f6eSJoseph Koshy 				    buffer);
95949874f6eSJoseph Koshy 			if (!S_ISDIR(sb.st_mode))
96049874f6eSJoseph Koshy 				errx(EX_USAGE, "ERROR: \"%s\" is not a "
96149874f6eSJoseph Koshy 				    "directory.", buffer);
96249874f6eSJoseph Koshy 		}
96349874f6eSJoseph Koshy 	}
96449874f6eSJoseph Koshy 
965b6010f9eSJoseph Koshy 	/*
966b6010f9eSJoseph Koshy 	 * If we have a callgraph be created, select the outputfile.
967b6010f9eSJoseph Koshy 	 */
968b6010f9eSJoseph Koshy 	if (args.pa_flags & FLAG_DO_CALLGRAPHS) {
969b6010f9eSJoseph Koshy 		if (strcmp(graphfilename, "-") == 0)
970b6010f9eSJoseph Koshy 		    args.pa_graphfile = args.pa_printfile;
971b6010f9eSJoseph Koshy 		else {
972b6010f9eSJoseph Koshy 			args.pa_graphfile = fopen(graphfilename, "w");
973b6010f9eSJoseph Koshy 			if (args.pa_graphfile == NULL)
974b6010f9eSJoseph Koshy 				err(EX_OSERR, "ERROR: cannot open \"%s\" "
975b6010f9eSJoseph Koshy 				    "for writing", graphfilename);
976b6010f9eSJoseph Koshy 		}
977b6010f9eSJoseph Koshy 	}
978b6010f9eSJoseph Koshy 
979f263522aSJoseph Koshy 	/* if we've been asked to process a log file, do that and exit */
98015139246SJoseph Koshy 	if (args.pa_flags & FLAG_READ_LOGFILE) {
98115139246SJoseph Koshy 		/*
98215139246SJoseph Koshy 		 * Print the log in textual form if we haven't been
983b6010f9eSJoseph Koshy 		 * asked to generate profiling information.
98415139246SJoseph Koshy 		 */
985b6010f9eSJoseph Koshy 		if ((args.pa_flags & FLAG_DO_ANALYSIS) == 0)
98615139246SJoseph Koshy 			args.pa_flags |= FLAG_DO_PRINT;
98715139246SJoseph Koshy 
98815139246SJoseph Koshy 		pmcstat_initialize_logging(&args);
989302cbb90SJoseph Koshy 		args.pa_logfd = pmcstat_open_log(args.pa_inputpath,
990302cbb90SJoseph Koshy 		    PMCSTAT_OPEN_FOR_READ);
99115139246SJoseph Koshy 		if ((args.pa_logparser = pmclog_open(args.pa_logfd)) == NULL)
99215139246SJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot create parser");
993f263522aSJoseph Koshy 		pmcstat_process_log(&args);
99449874f6eSJoseph Koshy 		pmcstat_shutdown_logging(&args);
995f263522aSJoseph Koshy 		exit(EX_OK);
996f263522aSJoseph Koshy 	}
997f263522aSJoseph Koshy 
998f263522aSJoseph Koshy 	/* otherwise, we've been asked to collect data */
999ebccf1e3SJoseph Koshy 	if (pmc_init() < 0)
1000ebccf1e3SJoseph Koshy 		err(EX_UNAVAILABLE,
1001ebccf1e3SJoseph Koshy 		    "ERROR: Initialization of the pmc(3) library failed");
1002ebccf1e3SJoseph Koshy 
1003ebccf1e3SJoseph Koshy 	if ((npmc = pmc_npmc(0)) < 0) /* assume all CPUs are identical */
1004ebccf1e3SJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot determine the number of PMCs "
1005ebccf1e3SJoseph Koshy 		    "on CPU %d", 0);
1006ebccf1e3SJoseph Koshy 
100715139246SJoseph Koshy 	/* Allocate a kqueue */
100815139246SJoseph Koshy 	if ((pmcstat_kq = kqueue()) < 0)
100915139246SJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot allocate kqueue");
101015139246SJoseph Koshy 
101115139246SJoseph Koshy 	/*
101215139246SJoseph Koshy 	 * Configure the specified log file or setup a default log
101315139246SJoseph Koshy 	 * consumer via a pipe.
101415139246SJoseph Koshy 	 */
101515139246SJoseph Koshy 	if (args.pa_required & FLAG_HAS_OUTPUT_LOGFILE) {
1016302cbb90SJoseph Koshy 		if (args.pa_outputpath)
1017302cbb90SJoseph Koshy 			args.pa_logfd = pmcstat_open_log(args.pa_outputpath,
1018302cbb90SJoseph Koshy 			    PMCSTAT_OPEN_FOR_WRITE);
1019302cbb90SJoseph Koshy 		else {
102015139246SJoseph Koshy 			/*
102115139246SJoseph Koshy 			 * process the log on the fly by reading it in
102215139246SJoseph Koshy 			 * through a pipe.
102315139246SJoseph Koshy 			 */
102415139246SJoseph Koshy 			if (pipe(pipefd) < 0)
102515139246SJoseph Koshy 				err(EX_OSERR, "ERROR: pipe(2) failed");
102615139246SJoseph Koshy 
102715139246SJoseph Koshy 			if (fcntl(pipefd[READPIPEFD], F_SETFL, O_NONBLOCK) < 0)
102815139246SJoseph Koshy 				err(EX_OSERR, "ERROR: fcntl(2) failed");
102915139246SJoseph Koshy 
103015139246SJoseph Koshy 			EV_SET(&kev, pipefd[READPIPEFD], EVFILT_READ, EV_ADD,
103115139246SJoseph Koshy 			    0, 0, NULL);
103215139246SJoseph Koshy 
103315139246SJoseph Koshy 			if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
103415139246SJoseph Koshy 				err(EX_OSERR, "ERROR: Cannot register kevent");
103515139246SJoseph Koshy 
103615139246SJoseph Koshy 			args.pa_logfd = pipefd[WRITEPIPEFD];
103715139246SJoseph Koshy 
103815139246SJoseph Koshy 			args.pa_flags |= (FLAG_HAS_PIPE | FLAG_DO_PRINT);
103915139246SJoseph Koshy 			args.pa_logparser = pmclog_open(pipefd[READPIPEFD]);
104015139246SJoseph Koshy 		}
104115139246SJoseph Koshy 
104215139246SJoseph Koshy 		if (pmc_configure_logfile(args.pa_logfd) < 0)
104315139246SJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot configure log file");
104415139246SJoseph Koshy 	}
104515139246SJoseph Koshy 
10467fbf3da2SJoseph Koshy 	/* remember to check for driver errors if we are sampling or logging */
10477fbf3da2SJoseph Koshy 	check_driver_stats = (args.pa_flags & FLAG_HAS_SAMPLING_PMCS) ||
10487fbf3da2SJoseph Koshy 	    (args.pa_flags & FLAG_HAS_OUTPUT_LOGFILE);
10497fbf3da2SJoseph Koshy 
1050ebccf1e3SJoseph Koshy 	/*
1051ebccf1e3SJoseph Koshy 	 * Allocate PMCs.
1052ebccf1e3SJoseph Koshy 	 */
1053ebccf1e3SJoseph Koshy 
1054fbd8b34aSJoseph Koshy 	STAILQ_FOREACH(ev, &args.pa_events, ev_next) {
1055ebccf1e3SJoseph Koshy 	    if (pmc_allocate(ev->ev_spec, ev->ev_mode,
1056f263522aSJoseph Koshy 		    ev->ev_flags, ev->ev_cpu, &ev->ev_pmcid) < 0)
1057ebccf1e3SJoseph Koshy 		    err(EX_OSERR, "ERROR: Cannot allocate %s-mode pmc with "
1058ebccf1e3SJoseph Koshy 			"specification \"%s\"",
1059ebccf1e3SJoseph Koshy 			PMC_IS_SYSTEM_MODE(ev->ev_mode) ? "system" : "process",
1060ebccf1e3SJoseph Koshy 			ev->ev_spec);
1061ebccf1e3SJoseph Koshy 
106215139246SJoseph Koshy 	    if (PMC_IS_SAMPLING_MODE(ev->ev_mode) &&
106315139246SJoseph Koshy 		pmc_set(ev->ev_pmcid, ev->ev_count) < 0)
106415139246SJoseph Koshy 		    err(EX_OSERR, "ERROR: Cannot set sampling count "
106515139246SJoseph Koshy 			"for PMC \"%s\"", ev->ev_name);
106615139246SJoseph Koshy 	}
106715139246SJoseph Koshy 
1068ebccf1e3SJoseph Koshy 	/* compute printout widths */
1069fbd8b34aSJoseph Koshy 	STAILQ_FOREACH(ev, &args.pa_events, ev_next) {
1070c5153e19SJoseph Koshy 		int counter_width;
1071c5153e19SJoseph Koshy 		int display_width;
1072c5153e19SJoseph Koshy 		int header_width;
1073ebccf1e3SJoseph Koshy 
1074c5153e19SJoseph Koshy 		(void) pmc_width(ev->ev_pmcid, &counter_width);
1075ee4910b1SJoseph Koshy 		header_width = strlen(ev->ev_name) + 2; /* prefix '%c/' */
1076c5153e19SJoseph Koshy 		display_width = (int) floor(counter_width / 3.32193) + 1;
1077ebccf1e3SJoseph Koshy 
1078ee4910b1SJoseph Koshy 		if (PMC_IS_SYSTEM_MODE(ev->ev_mode))
1079ee4910b1SJoseph Koshy 			header_width += 3; /* 2 digit CPU number + '/' */
1080ee4910b1SJoseph Koshy 
1081c5153e19SJoseph Koshy 		if (header_width > display_width) {
1082ebccf1e3SJoseph Koshy 			ev->ev_fieldskip = 0;
1083c5153e19SJoseph Koshy 			ev->ev_fieldwidth = header_width;
1084ebccf1e3SJoseph Koshy 		} else {
1085c5153e19SJoseph Koshy 			ev->ev_fieldskip = display_width -
1086c5153e19SJoseph Koshy 			    header_width;
1087c5153e19SJoseph Koshy 			ev->ev_fieldwidth = display_width;
1088ebccf1e3SJoseph Koshy 		}
1089ebccf1e3SJoseph Koshy 	}
1090ebccf1e3SJoseph Koshy 
1091ebccf1e3SJoseph Koshy 	/*
1092ebccf1e3SJoseph Koshy 	 * If our output is being set to a terminal, register a handler
1093ebccf1e3SJoseph Koshy 	 * for window size changes.
1094ebccf1e3SJoseph Koshy 	 */
1095ebccf1e3SJoseph Koshy 
109615139246SJoseph Koshy 	if (isatty(fileno(args.pa_printfile))) {
1097ebccf1e3SJoseph Koshy 
109815139246SJoseph Koshy 		if (ioctl(fileno(args.pa_printfile), TIOCGWINSZ, &ws) < 0)
1099ebccf1e3SJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot determine window size");
1100ebccf1e3SJoseph Koshy 
1101ebccf1e3SJoseph Koshy 		pmcstat_displayheight = ws.ws_row - 1;
1102ebccf1e3SJoseph Koshy 
1103ebccf1e3SJoseph Koshy 		EV_SET(&kev, SIGWINCH, EVFILT_SIGNAL, EV_ADD, 0, 0, NULL);
1104ebccf1e3SJoseph Koshy 
1105ebccf1e3SJoseph Koshy 		if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
1106ebccf1e3SJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot register kevent for "
1107ebccf1e3SJoseph Koshy 			    "SIGWINCH");
1108ebccf1e3SJoseph Koshy 	}
1109ebccf1e3SJoseph Koshy 
1110ebccf1e3SJoseph Koshy 	EV_SET(&kev, SIGINT, EVFILT_SIGNAL, EV_ADD, 0, 0, NULL);
1111ebccf1e3SJoseph Koshy 	if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
1112ebccf1e3SJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot register kevent for SIGINT");
1113ebccf1e3SJoseph Koshy 
1114f263522aSJoseph Koshy 	EV_SET(&kev, SIGIO, EVFILT_SIGNAL, EV_ADD, 0, 0, NULL);
1115f263522aSJoseph Koshy 	if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
1116f263522aSJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot register kevent for SIGIO");
1117ebccf1e3SJoseph Koshy 
1118ebccf1e3SJoseph Koshy 	/*
1119f263522aSJoseph Koshy 	 * An exec() failure of a forked child is signalled by the
1120f263522aSJoseph Koshy 	 * child sending the parent a SIGCHLD.  We don't register an
1121f263522aSJoseph Koshy 	 * actual signal handler for SIGCHLD, but instead use our
1122f263522aSJoseph Koshy 	 * kqueue to pick up the signal.
1123ebccf1e3SJoseph Koshy 	 */
1124f263522aSJoseph Koshy 	EV_SET(&kev, SIGCHLD, EVFILT_SIGNAL, EV_ADD, 0, 0, NULL);
1125f263522aSJoseph Koshy 	if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
1126f263522aSJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot register kevent for SIGCHLD");
1127ebccf1e3SJoseph Koshy 
112815139246SJoseph Koshy 	/* setup a timer if we have counting mode PMCs needing to be printed */
112915139246SJoseph Koshy 	if ((args.pa_flags & FLAG_HAS_COUNTING_PMCS) &&
113015139246SJoseph Koshy 	    (args.pa_required & FLAG_HAS_OUTPUT_LOGFILE) == 0) {
1131ebccf1e3SJoseph Koshy 		EV_SET(&kev, 0, EVFILT_TIMER, EV_ADD, 0,
1132ebccf1e3SJoseph Koshy 		    args.pa_interval * 1000, NULL);
1133ebccf1e3SJoseph Koshy 
1134ebccf1e3SJoseph Koshy 		if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
1135ebccf1e3SJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot register kevent for "
1136ebccf1e3SJoseph Koshy 			    "timer");
1137ebccf1e3SJoseph Koshy 	}
1138ebccf1e3SJoseph Koshy 
1139ebccf1e3SJoseph Koshy 	/* attach PMCs to the target process, starting it if specified */
1140ee4910b1SJoseph Koshy 	if (args.pa_flags & FLAG_HAS_COMMANDLINE)
1141ee4910b1SJoseph Koshy 		pmcstat_create_process(&args);
1142ebccf1e3SJoseph Koshy 
11437fbf3da2SJoseph Koshy 	if (check_driver_stats && pmc_get_driver_stats(&ds_start) < 0)
11447fbf3da2SJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot retrieve driver statistics");
11457fbf3da2SJoseph Koshy 
1146ee4910b1SJoseph Koshy 	/* Attach process pmcs to the target process. */
1147165b10b0SJoseph Koshy 	if (args.pa_flags & (FLAG_HAS_TARGET | FLAG_HAS_COMMANDLINE)) {
1148fbd8b34aSJoseph Koshy 		if (SLIST_EMPTY(&args.pa_targets))
1149fbd8b34aSJoseph Koshy 			errx(EX_DATAERR, "ERROR: No matching target "
1150fbd8b34aSJoseph Koshy 			    "processes.");
1151fbd8b34aSJoseph Koshy 		else
1152ee4910b1SJoseph Koshy 			pmcstat_attach_pmcs(&args);
1153ee4910b1SJoseph Koshy 
1154fbd8b34aSJoseph Koshy 		if (pmcstat_kvm) {
1155fbd8b34aSJoseph Koshy 			kvm_close(pmcstat_kvm);
1156fbd8b34aSJoseph Koshy 			pmcstat_kvm = NULL;
1157fbd8b34aSJoseph Koshy 		}
1158fbd8b34aSJoseph Koshy 	}
1159fbd8b34aSJoseph Koshy 
1160ebccf1e3SJoseph Koshy 	/* start the pmcs */
1161ebccf1e3SJoseph Koshy 	pmcstat_start_pmcs(&args);
1162ebccf1e3SJoseph Koshy 
1163ebccf1e3SJoseph Koshy 	/* start the (commandline) process if needed */
116415139246SJoseph Koshy 	if (args.pa_flags & FLAG_HAS_COMMANDLINE)
1165ee4910b1SJoseph Koshy 		pmcstat_start_process();
1166ebccf1e3SJoseph Koshy 
116715139246SJoseph Koshy 	/* initialize logging if printing the configured log */
116815139246SJoseph Koshy 	if ((args.pa_flags & FLAG_DO_PRINT) &&
116915139246SJoseph Koshy 	    (args.pa_flags & (FLAG_HAS_PIPE | FLAG_HAS_OUTPUT_LOGFILE)))
117015139246SJoseph Koshy 		pmcstat_initialize_logging(&args);
117115139246SJoseph Koshy 
1172ebccf1e3SJoseph Koshy 	/* Handle SIGINT using the kqueue loop */
1173ebccf1e3SJoseph Koshy 	sa.sa_handler = SIG_IGN;
1174ebccf1e3SJoseph Koshy 	sa.sa_flags   = 0;
1175ebccf1e3SJoseph Koshy 	(void) sigemptyset(&sa.sa_mask);
1176ebccf1e3SJoseph Koshy 
1177ebccf1e3SJoseph Koshy 	if (sigaction(SIGINT, &sa, NULL) < 0)
1178ebccf1e3SJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot install signal handler");
1179ebccf1e3SJoseph Koshy 
1180ebccf1e3SJoseph Koshy 	/*
1181ebccf1e3SJoseph Koshy 	 * loop till either the target process (if any) exits, or we
1182ebccf1e3SJoseph Koshy 	 * are killed by a SIGINT.
1183ebccf1e3SJoseph Koshy 	 */
1184f263522aSJoseph Koshy 	runstate = PMCSTAT_RUNNING;
1185f263522aSJoseph Koshy 	do_print = 0;
1186ebccf1e3SJoseph Koshy 	do {
1187ebccf1e3SJoseph Koshy 		if ((c = kevent(pmcstat_kq, NULL, 0, &kev, 1, NULL)) <= 0) {
1188ebccf1e3SJoseph Koshy 			if (errno != EINTR)
1189ebccf1e3SJoseph Koshy 				err(EX_OSERR, "ERROR: kevent failed");
1190ebccf1e3SJoseph Koshy 			else
1191ebccf1e3SJoseph Koshy 				continue;
1192ebccf1e3SJoseph Koshy 		}
1193ebccf1e3SJoseph Koshy 
1194ebccf1e3SJoseph Koshy 		if (kev.flags & EV_ERROR)
1195ebccf1e3SJoseph Koshy 			errc(EX_OSERR, kev.data, "ERROR: kevent failed");
1196ebccf1e3SJoseph Koshy 
1197ebccf1e3SJoseph Koshy 		switch (kev.filter) {
1198f263522aSJoseph Koshy 		case EVFILT_PROC:  /* target has exited */
119915139246SJoseph Koshy 			if (args.pa_flags & (FLAG_HAS_OUTPUT_LOGFILE |
120015139246SJoseph Koshy 				FLAG_HAS_PIPE))
1201f263522aSJoseph Koshy 				runstate = pmcstat_close_log(&args);
120238fc5f34SJoseph Koshy 			else
120338fc5f34SJoseph Koshy 				runstate = PMCSTAT_FINISHED;
120438fc5f34SJoseph Koshy 			do_print = 1;
1205f263522aSJoseph Koshy 			break;
1206ebccf1e3SJoseph Koshy 
1207f263522aSJoseph Koshy 		case EVFILT_READ:  /* log file data is present */
1208dc1d9d2eSJoseph Koshy 			runstate = pmcstat_process_log(&args);
1209ebccf1e3SJoseph Koshy 			break;
1210ebccf1e3SJoseph Koshy 
1211ebccf1e3SJoseph Koshy 		case EVFILT_SIGNAL:
1212f263522aSJoseph Koshy 			if (kev.ident == SIGCHLD) {
1213f263522aSJoseph Koshy 				/*
1214f263522aSJoseph Koshy 				 * The child process sends us a
1215f263522aSJoseph Koshy 				 * SIGCHLD if its exec() failed.  We
1216f263522aSJoseph Koshy 				 * wait for it to exit and then exit
1217f263522aSJoseph Koshy 				 * ourselves.
1218f263522aSJoseph Koshy 				 */
1219f263522aSJoseph Koshy 				(void) wait(&c);
1220f263522aSJoseph Koshy 				runstate = PMCSTAT_FINISHED;
1221f263522aSJoseph Koshy 			} else if (kev.ident == SIGIO) {
1222f263522aSJoseph Koshy 				/*
1223f263522aSJoseph Koshy 				 * We get a SIGIO if a PMC loses all
1224f263522aSJoseph Koshy 				 * of its targets, or if logfile
1225f263522aSJoseph Koshy 				 * writes encounter an error.
1226f263522aSJoseph Koshy 				 */
122715139246SJoseph Koshy 				if (args.pa_flags & (FLAG_HAS_OUTPUT_LOGFILE |
122815139246SJoseph Koshy 				    FLAG_HAS_PIPE)) {
1229f263522aSJoseph Koshy 					runstate = pmcstat_close_log(&args);
123015139246SJoseph Koshy 					if (args.pa_flags &
1231b6010f9eSJoseph Koshy 					    (FLAG_DO_PRINT|FLAG_DO_ANALYSIS))
123215139246SJoseph Koshy 						pmcstat_process_log(&args);
123315139246SJoseph Koshy 				}
1234f263522aSJoseph Koshy 				do_print = 1; /* print PMCs at exit */
1235f263522aSJoseph Koshy 				runstate = PMCSTAT_FINISHED;
1236f263522aSJoseph Koshy 			} else if (kev.ident == SIGINT) {
123715139246SJoseph Koshy 				/* Kill the child process if we started it */
123815139246SJoseph Koshy 				if (args.pa_flags & FLAG_HAS_COMMANDLINE)
1239fbd8b34aSJoseph Koshy 					pmcstat_kill_process(&args);
1240f263522aSJoseph Koshy 				runstate = PMCSTAT_FINISHED;
1241ebccf1e3SJoseph Koshy 			} else if (kev.ident == SIGWINCH) {
124215139246SJoseph Koshy 				if (ioctl(fileno(args.pa_printfile),
1243ebccf1e3SJoseph Koshy 					TIOCGWINSZ, &ws) < 0)
1244ebccf1e3SJoseph Koshy 				    err(EX_OSERR, "ERROR: Cannot determine "
1245ebccf1e3SJoseph Koshy 					"window size");
1246ebccf1e3SJoseph Koshy 				pmcstat_displayheight = ws.ws_row - 1;
1247ebccf1e3SJoseph Koshy 			} else
1248ebccf1e3SJoseph Koshy 				assert(0);
1249ebccf1e3SJoseph Koshy 
1250ebccf1e3SJoseph Koshy 			break;
1251f263522aSJoseph Koshy 
1252f263522aSJoseph Koshy 		case EVFILT_TIMER: /* print out counting PMCs */
1253f263522aSJoseph Koshy 			do_print = 1;
1254f263522aSJoseph Koshy 			break;
1255f263522aSJoseph Koshy 
1256ebccf1e3SJoseph Koshy 		}
1257ebccf1e3SJoseph Koshy 
125815139246SJoseph Koshy 		if (do_print &&
125915139246SJoseph Koshy 		    (args.pa_required & FLAG_HAS_OUTPUT_LOGFILE) == 0) {
1260f263522aSJoseph Koshy 			pmcstat_print_pmcs(&args);
126115139246SJoseph Koshy 			if (runstate == PMCSTAT_FINISHED && /* final newline */
126215139246SJoseph Koshy 			    (args.pa_flags & FLAG_DO_PRINT) == 0)
126315139246SJoseph Koshy 				(void) fprintf(args.pa_printfile, "\n");
1264f263522aSJoseph Koshy 			do_print = 0;
1265f263522aSJoseph Koshy 		}
1266f263522aSJoseph Koshy 
1267f263522aSJoseph Koshy 	} while (runstate != PMCSTAT_FINISHED);
1268f263522aSJoseph Koshy 
1269f263522aSJoseph Koshy 	/* flush any pending log entries */
127015139246SJoseph Koshy 	if (args.pa_flags & (FLAG_HAS_OUTPUT_LOGFILE | FLAG_HAS_PIPE))
1271f263522aSJoseph Koshy 		pmc_flush_logfile();
1272ebccf1e3SJoseph Koshy 
1273ebccf1e3SJoseph Koshy 	pmcstat_cleanup(&args);
1274ebccf1e3SJoseph Koshy 
127549874f6eSJoseph Koshy 	free(args.pa_kernel);
127649874f6eSJoseph Koshy 
12777fbf3da2SJoseph Koshy 	/* check if the driver lost any samples or events */
12787fbf3da2SJoseph Koshy 	if (check_driver_stats) {
12797fbf3da2SJoseph Koshy 		if (pmc_get_driver_stats(&ds_end) < 0)
12807fbf3da2SJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot retrieve driver "
12817fbf3da2SJoseph Koshy 			    "statistics");
128249874f6eSJoseph Koshy 		if (ds_start.pm_intr_bufferfull != ds_end.pm_intr_bufferfull &&
128349874f6eSJoseph Koshy 		    args.pa_verbosity > 0)
1284bea6aa93SJoseph Koshy 			warnx("WARNING: some samples were dropped.  Please "
12857fbf3da2SJoseph Koshy 			    "consider tuning the \"kern.hwpmc.nsamples\" "
12867fbf3da2SJoseph Koshy 			    "tunable.");
12877fbf3da2SJoseph Koshy 		if (ds_start.pm_buffer_requests_failed !=
128849874f6eSJoseph Koshy 		    ds_end.pm_buffer_requests_failed &&
128949874f6eSJoseph Koshy 		    args.pa_verbosity > 0)
1290bea6aa93SJoseph Koshy 			warnx("WARNING: some events were discarded.  Please "
12917fbf3da2SJoseph Koshy 			    "consider tuning the \"kern.hwpmc.nbuffers\" "
12927fbf3da2SJoseph Koshy 			    "tunable.");
12937fbf3da2SJoseph Koshy 	}
12947fbf3da2SJoseph Koshy 
12957fbf3da2SJoseph Koshy 	exit(EX_OK);
1296ebccf1e3SJoseph Koshy }
1297