xref: /freebsd/usr.sbin/pmcstat/pmcstat.c (revision 465dadb59bade7a5c8acb28460341ede115f8e30)
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 
34fbd8b34aSJoseph Koshy #include <sys/param.h>
35ea056888SAttilio Rao #include <sys/cpuset.h>
36ea056888SAttilio Rao #include <sys/event.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>
470b86b1bbSFabien Thomas #include <curses.h>
48ebccf1e3SJoseph Koshy #include <err.h>
49ebccf1e3SJoseph Koshy #include <errno.h>
50ebccf1e3SJoseph Koshy #include <fcntl.h>
51fbd8b34aSJoseph Koshy #include <kvm.h>
5249874f6eSJoseph Koshy #include <libgen.h>
53ebccf1e3SJoseph Koshy #include <limits.h>
54ebccf1e3SJoseph Koshy #include <math.h>
55ebccf1e3SJoseph Koshy #include <pmc.h>
56f263522aSJoseph Koshy #include <pmclog.h>
57fbd8b34aSJoseph Koshy #include <regex.h>
58ebccf1e3SJoseph Koshy #include <signal.h>
59ebccf1e3SJoseph Koshy #include <stdarg.h>
60ebccf1e3SJoseph Koshy #include <stdint.h>
6115139246SJoseph Koshy #include <stdio.h>
62ebccf1e3SJoseph Koshy #include <stdlib.h>
63ebccf1e3SJoseph Koshy #include <string.h>
64ebccf1e3SJoseph Koshy #include <sysexits.h>
65ebccf1e3SJoseph Koshy #include <unistd.h>
66ebccf1e3SJoseph Koshy 
6715139246SJoseph Koshy #include "pmcstat.h"
6815139246SJoseph Koshy 
69f263522aSJoseph Koshy /*
70f263522aSJoseph Koshy  * A given invocation of pmcstat(8) can manage multiple PMCs of both
71f263522aSJoseph Koshy  * the system-wide and per-process variety.  Each of these could be in
72f263522aSJoseph Koshy  * 'counting mode' or in 'sampling mode'.
73f263522aSJoseph Koshy  *
74f263522aSJoseph Koshy  * For 'counting mode' PMCs, pmcstat(8) will periodically issue a
75f263522aSJoseph Koshy  * pmc_read() at the configured time interval and print out the value
76f263522aSJoseph Koshy  * of the requested PMCs.
77f263522aSJoseph Koshy  *
78f263522aSJoseph Koshy  * For 'sampling mode' PMCs it can log to a file for offline analysis,
79f263522aSJoseph Koshy  * or can analyse sampling data "on the fly", either by converting
80f263522aSJoseph Koshy  * samples to printed textual form or by creating gprof(1) compatible
81f263522aSJoseph Koshy  * profiles, one per program executed.  When creating gprof(1)
82f263522aSJoseph Koshy  * profiles it can optionally merge entries from multiple processes
83f263522aSJoseph Koshy  * for a given executable into a single profile file.
84ee4910b1SJoseph Koshy  *
85ee4910b1SJoseph Koshy  * pmcstat(8) can also execute a command line and attach PMCs to the
86ee4910b1SJoseph Koshy  * resulting child process.  The protocol used is as follows:
87ee4910b1SJoseph Koshy  *
88ee4910b1SJoseph Koshy  * - parent creates a socketpair for two way communication and
89ee4910b1SJoseph Koshy  *   fork()s.
90ee4910b1SJoseph Koshy  * - subsequently:
91ee4910b1SJoseph Koshy  *
92ee4910b1SJoseph Koshy  *   /Parent/				/Child/
93ee4910b1SJoseph Koshy  *
94ee4910b1SJoseph Koshy  *   - Wait for childs token.
95ee4910b1SJoseph Koshy  *					- Sends token.
96ee4910b1SJoseph Koshy  *					- Awaits signal to start.
97ee4910b1SJoseph Koshy  *  - Attaches PMCs to the child's pid
98ee4910b1SJoseph Koshy  *    and starts them. Sets up
99ee4910b1SJoseph Koshy  *    monitoring for the child.
100ee4910b1SJoseph Koshy  *  - Signals child to start.
1013df5ecacSUlrich Spörlein  *					- Receives signal, attempts exec().
102ee4910b1SJoseph Koshy  *
103ee4910b1SJoseph Koshy  * After this point normal processing can happen.
104f263522aSJoseph Koshy  */
105f263522aSJoseph Koshy 
10615139246SJoseph Koshy /* Globals */
107ebccf1e3SJoseph Koshy 
108ebccf1e3SJoseph Koshy int		pmcstat_displayheight = DEFAULT_DISPLAY_HEIGHT;
1090b86b1bbSFabien Thomas int		pmcstat_displaywidth  = DEFAULT_DISPLAY_WIDTH;
110bf70beceSEd Schouten static int	pmcstat_sockpair[NSOCKPAIRFD];
111bf70beceSEd Schouten static int	pmcstat_kq;
112bf70beceSEd Schouten static kvm_t	*pmcstat_kvm;
113bf70beceSEd Schouten static struct kinfo_proc *pmcstat_plist;
1140b86b1bbSFabien Thomas struct pmcstat_args args;
115ebccf1e3SJoseph Koshy 
116ea056888SAttilio Rao static void
117ea056888SAttilio Rao pmcstat_clone_event_descriptor(struct pmcstat_ev *ev, const cpuset_t *cpumask)
118ea056888SAttilio Rao {
119ef16b9abSJohn Baldwin 	int cpu;
120ea056888SAttilio Rao 	struct pmcstat_ev *ev_clone;
121ea056888SAttilio Rao 
122ef16b9abSJohn Baldwin 	for (cpu = 0; cpu < CPU_SETSIZE; cpu++) {
123ea056888SAttilio Rao 		if (!CPU_ISSET(cpu, cpumask))
124ea056888SAttilio Rao 			continue;
125ea056888SAttilio Rao 
126ea056888SAttilio Rao 		if ((ev_clone = malloc(sizeof(*ev_clone))) == NULL)
127ea056888SAttilio Rao 			errx(EX_SOFTWARE, "ERROR: Out of memory");
128ea056888SAttilio Rao 		(void) memset(ev_clone, 0, sizeof(*ev_clone));
129ea056888SAttilio Rao 
130ea056888SAttilio Rao 		ev_clone->ev_count = ev->ev_count;
131ea056888SAttilio Rao 		ev_clone->ev_cpu   = cpu;
132ea056888SAttilio Rao 		ev_clone->ev_cumulative = ev->ev_cumulative;
133ea056888SAttilio Rao 		ev_clone->ev_flags = ev->ev_flags;
134ea056888SAttilio Rao 		ev_clone->ev_mode  = ev->ev_mode;
135ea056888SAttilio Rao 		ev_clone->ev_name  = strdup(ev->ev_name);
136ea056888SAttilio Rao 		ev_clone->ev_pmcid = ev->ev_pmcid;
137ea056888SAttilio Rao 		ev_clone->ev_saved = ev->ev_saved;
138ea056888SAttilio Rao 		ev_clone->ev_spec  = strdup(ev->ev_spec);
139ea056888SAttilio Rao 
140ea056888SAttilio Rao 		STAILQ_INSERT_TAIL(&args.pa_events, ev_clone, ev_next);
141ea056888SAttilio Rao 	}
142ea056888SAttilio Rao }
143ea056888SAttilio Rao 
144ea056888SAttilio Rao static void
145ea056888SAttilio Rao pmcstat_get_cpumask(const char *cpuspec, cpuset_t *cpumask)
146ea056888SAttilio Rao {
147ea056888SAttilio Rao 	int cpu;
148ea056888SAttilio Rao 	const char *s;
149ea056888SAttilio Rao 	char *end;
150ea056888SAttilio Rao 
151ea056888SAttilio Rao 	CPU_ZERO(cpumask);
152ea056888SAttilio Rao 	s = cpuspec;
153ea056888SAttilio Rao 
154ea056888SAttilio Rao 	do {
155ea056888SAttilio Rao 		cpu = strtol(s, &end, 0);
156ea056888SAttilio Rao 		if (cpu < 0 || end == s)
15737d6f8a9SDavid E. O'Brien 			errx(EX_USAGE,
15837d6f8a9SDavid E. O'Brien 			    "ERROR: Illegal CPU specification \"%s\".",
15937d6f8a9SDavid E. O'Brien 			    cpuspec);
160ea056888SAttilio Rao 		CPU_SET(cpu, cpumask);
161ea056888SAttilio Rao 		s = end + strspn(end, ", \t");
162ea056888SAttilio Rao 	} while (*s);
163ef16b9abSJohn Baldwin 	assert(!CPU_EMPTY(cpumask));
164ea056888SAttilio Rao }
165ea056888SAttilio Rao 
166ee4910b1SJoseph Koshy void
1670b86b1bbSFabien Thomas pmcstat_attach_pmcs(void)
168ee4910b1SJoseph Koshy {
169ee4910b1SJoseph Koshy 	struct pmcstat_ev *ev;
170fbd8b34aSJoseph Koshy 	struct pmcstat_target *pt;
171fbd8b34aSJoseph Koshy 	int count;
172ee4910b1SJoseph Koshy 
173fbd8b34aSJoseph Koshy 	/* Attach all process PMCs to target processes. */
174fbd8b34aSJoseph Koshy 	count = 0;
1750b86b1bbSFabien Thomas 	STAILQ_FOREACH(ev, &args.pa_events, ev_next) {
176fbd8b34aSJoseph Koshy 		if (PMC_IS_SYSTEM_MODE(ev->ev_mode))
177fbd8b34aSJoseph Koshy 			continue;
1780b86b1bbSFabien Thomas 		SLIST_FOREACH(pt, &args.pa_targets, pt_next)
179fbd8b34aSJoseph Koshy 			if (pmc_attach(ev->ev_pmcid, pt->pt_pid) == 0)
180fbd8b34aSJoseph Koshy 				count++;
181fbd8b34aSJoseph Koshy 			else if (errno != ESRCH)
18237d6f8a9SDavid E. O'Brien 				err(EX_OSERR,
18337d6f8a9SDavid E. O'Brien "ERROR: cannot attach pmc \"%s\" to process %d",
18437d6f8a9SDavid E. O'Brien 				    ev->ev_name, (int)pt->pt_pid);
185fbd8b34aSJoseph Koshy 	}
186ee4910b1SJoseph Koshy 
187fbd8b34aSJoseph Koshy 	if (count == 0)
188fbd8b34aSJoseph Koshy 		errx(EX_DATAERR, "ERROR: No processes were attached to.");
189ee4910b1SJoseph Koshy }
190ee4910b1SJoseph Koshy 
191ebccf1e3SJoseph Koshy 
192ebccf1e3SJoseph Koshy void
1930b86b1bbSFabien Thomas pmcstat_cleanup(void)
194ebccf1e3SJoseph Koshy {
195ebccf1e3SJoseph Koshy 	struct pmcstat_ev *ev, *tmp;
196ebccf1e3SJoseph Koshy 
197ebccf1e3SJoseph Koshy 	/* release allocated PMCs. */
1980b86b1bbSFabien Thomas 	STAILQ_FOREACH_SAFE(ev, &args.pa_events, ev_next, tmp)
199ebccf1e3SJoseph Koshy 	    if (ev->ev_pmcid != PMC_ID_INVALID) {
200b8fbd97bSJoseph Koshy 		if (pmc_stop(ev->ev_pmcid) < 0)
20137d6f8a9SDavid E. O'Brien 			err(EX_OSERR, "ERROR: cannot stop pmc 0x%x \"%s\"",
20237d6f8a9SDavid E. O'Brien 			    ev->ev_pmcid, ev->ev_name);
203ebccf1e3SJoseph Koshy 		if (pmc_release(ev->ev_pmcid) < 0)
20437d6f8a9SDavid E. O'Brien 			err(EX_OSERR, "ERROR: cannot release pmc 0x%x \"%s\"",
20537d6f8a9SDavid E. O'Brien 			    ev->ev_pmcid, ev->ev_name);
206ebccf1e3SJoseph Koshy 		free(ev->ev_name);
207ebccf1e3SJoseph Koshy 		free(ev->ev_spec);
2080b86b1bbSFabien Thomas 		STAILQ_REMOVE(&args.pa_events, ev, pmcstat_ev, ev_next);
209ebccf1e3SJoseph Koshy 		free(ev);
210ebccf1e3SJoseph Koshy 	    }
211f263522aSJoseph Koshy 
212dc1d9d2eSJoseph Koshy 	/* de-configure the log file if present. */
2130b86b1bbSFabien Thomas 	if (args.pa_flags & (FLAG_HAS_PIPE | FLAG_HAS_OUTPUT_LOGFILE))
214dc1d9d2eSJoseph Koshy 		(void) pmc_configure_logfile(-1);
215dc1d9d2eSJoseph Koshy 
2160b86b1bbSFabien Thomas 	if (args.pa_logparser) {
2170b86b1bbSFabien Thomas 		pmclog_close(args.pa_logparser);
2180b86b1bbSFabien Thomas 		args.pa_logparser = NULL;
219f263522aSJoseph Koshy 	}
22015139246SJoseph Koshy 
2210b86b1bbSFabien Thomas 	pmcstat_shutdown_logging();
222ebccf1e3SJoseph Koshy }
223ebccf1e3SJoseph Koshy 
224ebccf1e3SJoseph Koshy void
2250b86b1bbSFabien Thomas pmcstat_create_process(void)
226ee4910b1SJoseph Koshy {
227ee4910b1SJoseph Koshy 	char token;
228fbd8b34aSJoseph Koshy 	pid_t pid;
229ee4910b1SJoseph Koshy 	struct kevent kev;
230fbd8b34aSJoseph Koshy 	struct pmcstat_target *pt;
231ee4910b1SJoseph Koshy 
232ee4910b1SJoseph Koshy 	if (socketpair(AF_UNIX, SOCK_STREAM, 0, pmcstat_sockpair) < 0)
233ee4910b1SJoseph Koshy 		err(EX_OSERR, "ERROR: cannot create socket pair");
234ee4910b1SJoseph Koshy 
235fbd8b34aSJoseph Koshy 	switch (pid = fork()) {
236ee4910b1SJoseph Koshy 	case -1:
237ee4910b1SJoseph Koshy 		err(EX_OSERR, "ERROR: cannot fork");
238ee4910b1SJoseph Koshy 		/*NOTREACHED*/
239ee4910b1SJoseph Koshy 
240ee4910b1SJoseph Koshy 	case 0:		/* child */
241ee4910b1SJoseph Koshy 		(void) close(pmcstat_sockpair[PARENTSOCKET]);
242ee4910b1SJoseph Koshy 
243ee4910b1SJoseph Koshy 		/* Write a token to tell our parent we've started executing. */
244ee4910b1SJoseph Koshy 		if (write(pmcstat_sockpair[CHILDSOCKET], "+", 1) != 1)
245ee4910b1SJoseph Koshy 			err(EX_OSERR, "ERROR (child): cannot write token");
246ee4910b1SJoseph Koshy 
247ee4910b1SJoseph Koshy 		/* Wait for our parent to signal us to start. */
248ee4910b1SJoseph Koshy 		if (read(pmcstat_sockpair[CHILDSOCKET], &token, 1) < 0)
249ee4910b1SJoseph Koshy 			err(EX_OSERR, "ERROR (child): cannot read token");
250ee4910b1SJoseph Koshy 		(void) close(pmcstat_sockpair[CHILDSOCKET]);
251ee4910b1SJoseph Koshy 
252ee4910b1SJoseph Koshy 		/* exec() the program requested */
2530b86b1bbSFabien Thomas 		execvp(*args.pa_argv, args.pa_argv);
254ee4910b1SJoseph Koshy 		/* and if that fails, notify the parent */
255ee4910b1SJoseph Koshy 		kill(getppid(), SIGCHLD);
2560b86b1bbSFabien Thomas 		err(EX_OSERR, "ERROR: execvp \"%s\" failed", *args.pa_argv);
257ee4910b1SJoseph Koshy 		/*NOTREACHED*/
258ee4910b1SJoseph Koshy 
259ee4910b1SJoseph Koshy 	default:	/* parent */
260ee4910b1SJoseph Koshy 		(void) close(pmcstat_sockpair[CHILDSOCKET]);
261ee4910b1SJoseph Koshy 		break;
262ee4910b1SJoseph Koshy 	}
263ee4910b1SJoseph Koshy 
264ee4910b1SJoseph Koshy 	/* Ask to be notified via a kevent when the target process exits. */
265fbd8b34aSJoseph Koshy 	EV_SET(&kev, pid, EVFILT_PROC, EV_ADD|EV_ONESHOT, NOTE_EXIT, 0,
266ee4910b1SJoseph Koshy 	    NULL);
267ee4910b1SJoseph Koshy 	if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
268fbd8b34aSJoseph Koshy 		err(EX_OSERR, "ERROR: cannot monitor child process %d", pid);
269fbd8b34aSJoseph Koshy 
270fbd8b34aSJoseph Koshy 	if ((pt = malloc(sizeof(*pt))) == NULL)
271fbd8b34aSJoseph Koshy 		errx(EX_SOFTWARE, "ERROR: Out of memory.");
272fbd8b34aSJoseph Koshy 
273fbd8b34aSJoseph Koshy 	pt->pt_pid = pid;
2740b86b1bbSFabien Thomas 	SLIST_INSERT_HEAD(&args.pa_targets, pt, pt_next);
275ee4910b1SJoseph Koshy 
276ee4910b1SJoseph Koshy 	/* Wait for the child to signal that its ready to go. */
277ee4910b1SJoseph Koshy 	if (read(pmcstat_sockpair[PARENTSOCKET], &token, 1) < 0)
278ee4910b1SJoseph Koshy 		err(EX_OSERR, "ERROR (parent): cannot read token");
279ee4910b1SJoseph Koshy 
280ee4910b1SJoseph Koshy 	return;
281ee4910b1SJoseph Koshy }
282ee4910b1SJoseph Koshy 
283fbd8b34aSJoseph Koshy void
2840b86b1bbSFabien Thomas pmcstat_find_targets(const char *spec)
285fbd8b34aSJoseph Koshy {
286fbd8b34aSJoseph Koshy 	int n, nproc, pid, rv;
287fbd8b34aSJoseph Koshy 	struct pmcstat_target *pt;
288fbd8b34aSJoseph Koshy 	char errbuf[_POSIX2_LINE_MAX], *end;
289fbd8b34aSJoseph Koshy 	static struct kinfo_proc *kp;
290fbd8b34aSJoseph Koshy 	regex_t reg;
291fbd8b34aSJoseph Koshy 	regmatch_t regmatch;
292fbd8b34aSJoseph Koshy 
293fbd8b34aSJoseph Koshy 	/* First check if we've been given a process id. */
294fbd8b34aSJoseph Koshy       	pid = strtol(spec, &end, 0);
295fbd8b34aSJoseph Koshy 	if (end != spec && pid >= 0) {
296fbd8b34aSJoseph Koshy 		if ((pt = malloc(sizeof(*pt))) == NULL)
297fbd8b34aSJoseph Koshy 			goto outofmemory;
298fbd8b34aSJoseph Koshy 		pt->pt_pid = pid;
2990b86b1bbSFabien Thomas 		SLIST_INSERT_HEAD(&args.pa_targets, pt, pt_next);
300fbd8b34aSJoseph Koshy 		return;
301fbd8b34aSJoseph Koshy 	}
302fbd8b34aSJoseph Koshy 
303fbd8b34aSJoseph Koshy 	/* Otherwise treat arg as a regular expression naming processes. */
304fbd8b34aSJoseph Koshy 	if (pmcstat_kvm == NULL) {
305fbd8b34aSJoseph Koshy 		if ((pmcstat_kvm = kvm_openfiles(NULL, "/dev/null", NULL, 0,
306fbd8b34aSJoseph Koshy 		    errbuf)) == NULL)
307fbd8b34aSJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot open kernel \"%s\"",
308fbd8b34aSJoseph Koshy 			    errbuf);
309fbd8b34aSJoseph Koshy 		if ((pmcstat_plist = kvm_getprocs(pmcstat_kvm, KERN_PROC_PROC,
310fbd8b34aSJoseph Koshy 		    0, &nproc)) == NULL)
311fbd8b34aSJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot get process list: %s",
312fbd8b34aSJoseph Koshy 			    kvm_geterr(pmcstat_kvm));
3135c15d3c8SFabien Thomas 	} else
3145c15d3c8SFabien Thomas 		nproc = 0;
315fbd8b34aSJoseph Koshy 
316fbd8b34aSJoseph Koshy 	if ((rv = regcomp(&reg, spec, REG_EXTENDED|REG_NOSUB)) != 0) {
317fbd8b34aSJoseph Koshy 		regerror(rv, &reg, errbuf, sizeof(errbuf));
318fbd8b34aSJoseph Koshy 		err(EX_DATAERR, "ERROR: Failed to compile regex \"%s\": %s",
319fbd8b34aSJoseph Koshy 		    spec, errbuf);
320fbd8b34aSJoseph Koshy 	}
321fbd8b34aSJoseph Koshy 
322fbd8b34aSJoseph Koshy 	for (n = 0, kp = pmcstat_plist; n < nproc; n++, kp++) {
323fbd8b34aSJoseph Koshy 		if ((rv = regexec(&reg, kp->ki_comm, 1, &regmatch, 0)) == 0) {
324fbd8b34aSJoseph Koshy 			if ((pt = malloc(sizeof(*pt))) == NULL)
325fbd8b34aSJoseph Koshy 				goto outofmemory;
326fbd8b34aSJoseph Koshy 			pt->pt_pid = kp->ki_pid;
3270b86b1bbSFabien Thomas 			SLIST_INSERT_HEAD(&args.pa_targets, pt, pt_next);
328fbd8b34aSJoseph Koshy 		} else if (rv != REG_NOMATCH) {
329fbd8b34aSJoseph Koshy 			regerror(rv, &reg, errbuf, sizeof(errbuf));
330fbd8b34aSJoseph Koshy 			errx(EX_SOFTWARE, "ERROR: Regex evalation failed: %s",
331fbd8b34aSJoseph Koshy 			    errbuf);
332fbd8b34aSJoseph Koshy 		}
333fbd8b34aSJoseph Koshy 	}
334fbd8b34aSJoseph Koshy 
335fbd8b34aSJoseph Koshy 	regfree(&reg);
336fbd8b34aSJoseph Koshy 
337fbd8b34aSJoseph Koshy 	return;
338fbd8b34aSJoseph Koshy 
339fbd8b34aSJoseph Koshy  outofmemory:
340fbd8b34aSJoseph Koshy 	errx(EX_SOFTWARE, "Out of memory.");
341fbd8b34aSJoseph Koshy 	/*NOTREACHED*/
342fbd8b34aSJoseph Koshy }
343fbd8b34aSJoseph Koshy 
344ee4910b1SJoseph Koshy void
3450b86b1bbSFabien Thomas pmcstat_kill_process(void)
346fbd8b34aSJoseph Koshy {
347fbd8b34aSJoseph Koshy 	struct pmcstat_target *pt;
348fbd8b34aSJoseph Koshy 
3490b86b1bbSFabien Thomas 	assert(args.pa_flags & FLAG_HAS_COMMANDLINE);
350fbd8b34aSJoseph Koshy 
351fbd8b34aSJoseph Koshy 	/*
352fbd8b34aSJoseph Koshy 	 * If a command line was specified, it would be the very first
353fbd8b34aSJoseph Koshy 	 * in the list, before any other processes specified by -t.
354fbd8b34aSJoseph Koshy 	 */
3550b86b1bbSFabien Thomas 	pt = SLIST_FIRST(&args.pa_targets);
356fbd8b34aSJoseph Koshy 	assert(pt != NULL);
357fbd8b34aSJoseph Koshy 
358fbd8b34aSJoseph Koshy 	if (kill(pt->pt_pid, SIGINT) != 0)
359fbd8b34aSJoseph Koshy 		err(EX_OSERR, "ERROR: cannot signal child process");
360fbd8b34aSJoseph Koshy }
361fbd8b34aSJoseph Koshy 
362fbd8b34aSJoseph Koshy void
3630b86b1bbSFabien Thomas pmcstat_start_pmcs(void)
364ebccf1e3SJoseph Koshy {
365ebccf1e3SJoseph Koshy 	struct pmcstat_ev *ev;
366ebccf1e3SJoseph Koshy 
367fbd8b34aSJoseph Koshy 	STAILQ_FOREACH(ev, &args.pa_events, ev_next) {
368ebccf1e3SJoseph Koshy 
369ebccf1e3SJoseph Koshy 	    assert(ev->ev_pmcid != PMC_ID_INVALID);
370ebccf1e3SJoseph Koshy 
371ebccf1e3SJoseph Koshy 	    if (pmc_start(ev->ev_pmcid) < 0) {
372f263522aSJoseph Koshy 	        warn("ERROR: Cannot start pmc 0x%x \"%s\"",
373ebccf1e3SJoseph Koshy 		    ev->ev_pmcid, ev->ev_name);
3740b86b1bbSFabien Thomas 		pmcstat_cleanup();
375f263522aSJoseph Koshy 		exit(EX_OSERR);
376ebccf1e3SJoseph Koshy 	    }
377ebccf1e3SJoseph Koshy 	}
378ebccf1e3SJoseph Koshy 
379ebccf1e3SJoseph Koshy }
380ebccf1e3SJoseph Koshy 
381ebccf1e3SJoseph Koshy void
3820b86b1bbSFabien Thomas pmcstat_print_headers(void)
383ebccf1e3SJoseph Koshy {
384ebccf1e3SJoseph Koshy 	struct pmcstat_ev *ev;
385ee4910b1SJoseph Koshy 	int c, w;
386ebccf1e3SJoseph Koshy 
3870b86b1bbSFabien Thomas 	(void) fprintf(args.pa_printfile, PRINT_HEADER_PREFIX);
388ebccf1e3SJoseph Koshy 
3890b86b1bbSFabien Thomas 	STAILQ_FOREACH(ev, &args.pa_events, ev_next) {
390ebccf1e3SJoseph Koshy 		if (PMC_IS_SAMPLING_MODE(ev->ev_mode))
391ebccf1e3SJoseph Koshy 			continue;
392ebccf1e3SJoseph Koshy 
393ebccf1e3SJoseph Koshy 		c = PMC_IS_SYSTEM_MODE(ev->ev_mode) ? 's' : 'p';
394ebccf1e3SJoseph Koshy 
395ee4910b1SJoseph Koshy 		if (ev->ev_fieldskip != 0)
3960b86b1bbSFabien Thomas 			(void) fprintf(args.pa_printfile, "%*s",
397ee4910b1SJoseph Koshy 			    ev->ev_fieldskip, "");
398ee4910b1SJoseph Koshy 		w = ev->ev_fieldwidth - ev->ev_fieldskip - 2;
399ee4910b1SJoseph Koshy 
400ee4910b1SJoseph Koshy 		if (c == 's')
4010b86b1bbSFabien Thomas 			(void) fprintf(args.pa_printfile, "s/%02d/%-*s ",
402ee4910b1SJoseph Koshy 			    ev->ev_cpu, w-3, ev->ev_name);
403ee4910b1SJoseph Koshy 		else
4040b86b1bbSFabien Thomas 			(void) fprintf(args.pa_printfile, "p/%*s ", w,
405ebccf1e3SJoseph Koshy 			    ev->ev_name);
406ebccf1e3SJoseph Koshy 	}
407ebccf1e3SJoseph Koshy 
4080b86b1bbSFabien Thomas 	(void) fflush(args.pa_printfile);
409ebccf1e3SJoseph Koshy }
410ebccf1e3SJoseph Koshy 
411ebccf1e3SJoseph Koshy void
4120b86b1bbSFabien Thomas pmcstat_print_counters(void)
413ebccf1e3SJoseph Koshy {
414ebccf1e3SJoseph Koshy 	int extra_width;
415ebccf1e3SJoseph Koshy 	struct pmcstat_ev *ev;
416ebccf1e3SJoseph Koshy 	pmc_value_t value;
417ebccf1e3SJoseph Koshy 
418ebccf1e3SJoseph Koshy 	extra_width = sizeof(PRINT_HEADER_PREFIX) - 1;
419ebccf1e3SJoseph Koshy 
4200b86b1bbSFabien Thomas 	STAILQ_FOREACH(ev, &args.pa_events, ev_next) {
421ebccf1e3SJoseph Koshy 
422ebccf1e3SJoseph Koshy 		/* skip sampling mode counters */
423ebccf1e3SJoseph Koshy 		if (PMC_IS_SAMPLING_MODE(ev->ev_mode))
424ebccf1e3SJoseph Koshy 			continue;
425ebccf1e3SJoseph Koshy 
426ebccf1e3SJoseph Koshy 		if (pmc_read(ev->ev_pmcid, &value) < 0)
42737d6f8a9SDavid E. O'Brien 			err(EX_OSERR, "ERROR: Cannot read pmc \"%s\"",
42837d6f8a9SDavid E. O'Brien 			    ev->ev_name);
429ebccf1e3SJoseph Koshy 
4300b86b1bbSFabien Thomas 		(void) fprintf(args.pa_printfile, "%*ju ",
43115139246SJoseph Koshy 		    ev->ev_fieldwidth + extra_width,
43215139246SJoseph Koshy 		    (uintmax_t) ev->ev_cumulative ? value :
43315139246SJoseph Koshy 		    (value - ev->ev_saved));
43415139246SJoseph Koshy 
435ebccf1e3SJoseph Koshy 		if (ev->ev_cumulative == 0)
436ebccf1e3SJoseph Koshy 			ev->ev_saved = value;
437ebccf1e3SJoseph Koshy 		extra_width = 0;
438ebccf1e3SJoseph Koshy 	}
439ebccf1e3SJoseph Koshy 
4400b86b1bbSFabien Thomas 	(void) fflush(args.pa_printfile);
441ebccf1e3SJoseph Koshy }
442ebccf1e3SJoseph Koshy 
443ebccf1e3SJoseph Koshy /*
444ebccf1e3SJoseph Koshy  * Print output
445ebccf1e3SJoseph Koshy  */
446ebccf1e3SJoseph Koshy 
447ebccf1e3SJoseph Koshy void
4480b86b1bbSFabien Thomas pmcstat_print_pmcs(void)
449ebccf1e3SJoseph Koshy {
450ebccf1e3SJoseph Koshy 	static int linecount = 0;
451ebccf1e3SJoseph Koshy 
45215139246SJoseph Koshy 	/* check if we need to print a header line */
453ebccf1e3SJoseph Koshy 	if (++linecount > pmcstat_displayheight) {
4540b86b1bbSFabien Thomas 		(void) fprintf(args.pa_printfile, "\n");
455ebccf1e3SJoseph Koshy 		linecount = 1;
456ebccf1e3SJoseph Koshy 	}
457ebccf1e3SJoseph Koshy 	if (linecount == 1)
4580b86b1bbSFabien Thomas 		pmcstat_print_headers();
4590b86b1bbSFabien Thomas 	(void) fprintf(args.pa_printfile, "\n");
460ebccf1e3SJoseph Koshy 
4610b86b1bbSFabien Thomas 	pmcstat_print_counters();
462ebccf1e3SJoseph Koshy 
463ebccf1e3SJoseph Koshy 	return;
464ebccf1e3SJoseph Koshy }
465ebccf1e3SJoseph Koshy 
466ebccf1e3SJoseph Koshy /*
467ebccf1e3SJoseph Koshy  * Do process profiling
468ebccf1e3SJoseph Koshy  *
469ebccf1e3SJoseph Koshy  * If a pid was specified, attach each allocated PMC to the target
470ebccf1e3SJoseph Koshy  * process.  Otherwise, fork a child and attach the PMCs to the child,
471ebccf1e3SJoseph Koshy  * and have the child exec() the target program.
472ebccf1e3SJoseph Koshy  */
473ebccf1e3SJoseph Koshy 
474ebccf1e3SJoseph Koshy void
475ee4910b1SJoseph Koshy pmcstat_start_process(void)
476ebccf1e3SJoseph Koshy {
477ee4910b1SJoseph Koshy 	/* Signal the child to proceed. */
478ee4910b1SJoseph Koshy 	if (write(pmcstat_sockpair[PARENTSOCKET], "!", 1) != 1)
479ee4910b1SJoseph Koshy 		err(EX_OSERR, "ERROR (parent): write of token failed");
480ebccf1e3SJoseph Koshy 
481ee4910b1SJoseph Koshy 	(void) close(pmcstat_sockpair[PARENTSOCKET]);
482ebccf1e3SJoseph Koshy }
483ebccf1e3SJoseph Koshy 
484ebccf1e3SJoseph Koshy void
485ebccf1e3SJoseph Koshy pmcstat_show_usage(void)
486ebccf1e3SJoseph Koshy {
487ebccf1e3SJoseph Koshy 	errx(EX_USAGE,
488ebccf1e3SJoseph Koshy 	    "[options] [commandline]\n"
489ebccf1e3SJoseph Koshy 	    "\t Measure process and/or system performance using hardware\n"
490ebccf1e3SJoseph Koshy 	    "\t performance monitoring counters.\n"
491ebccf1e3SJoseph Koshy 	    "\t Options include:\n"
492f263522aSJoseph Koshy 	    "\t -C\t\t (toggle) show cumulative counts\n"
493f263522aSJoseph Koshy 	    "\t -D path\t create profiles in directory \"path\"\n"
494f263522aSJoseph Koshy 	    "\t -E\t\t (toggle) show counts at process exit\n"
4950b86b1bbSFabien Thomas 	    "\t -F file\t write a system-wide callgraph (Kcachegrind format)"
4960b86b1bbSFabien Thomas 		" to \"file\"\n"
497b6010f9eSJoseph Koshy 	    "\t -G file\t write a system-wide callgraph to \"file\"\n"
49849874f6eSJoseph Koshy 	    "\t -M file\t print executable/gmon file map to \"file\"\n"
499b6010f9eSJoseph Koshy 	    "\t -N\t\t (toggle) capture callchains\n"
500f263522aSJoseph Koshy 	    "\t -O file\t send log output to \"file\"\n"
501f263522aSJoseph Koshy 	    "\t -P spec\t allocate a process-private sampling PMC\n"
502f263522aSJoseph Koshy 	    "\t -R file\t read events from \"file\"\n"
503f263522aSJoseph Koshy 	    "\t -S spec\t allocate a system-wide sampling PMC\n"
5040b86b1bbSFabien Thomas 	    "\t -T\t\t start in top mode\n"
505f263522aSJoseph Koshy 	    "\t -W\t\t (toggle) show counts per context switch\n"
5065ae1451dSScott Long 	    "\t -a file\t print sampled PCs and callgraph to \"file\"\n"
507ee4910b1SJoseph Koshy 	    "\t -c cpu-list\t set cpus for subsequent system-wide PMCs\n"
508f263522aSJoseph Koshy 	    "\t -d\t\t (toggle) track descendants\n"
509*465dadb5SJonathan T. Looney 	    "\t -e\t\t use wide history counter for gprof(1) output\n"
5100b86b1bbSFabien Thomas 	    "\t -f spec\t pass \"spec\" to as plugin option\n"
511f263522aSJoseph Koshy 	    "\t -g\t\t produce gprof(1) compatible profiles\n"
51224bffe06SRuslan Ermilov 	    "\t -k dir\t\t set the path to the kernel\n"
51372fe532cSGeorge V. Neville-Neil 	    "\t -l secs\t set duration time\n"
51433b12a85SJim Harris 	    "\t -m file\t print sampled PCs to \"file\"\n"
515ebccf1e3SJoseph Koshy 	    "\t -n rate\t set sampling rate\n"
516ebccf1e3SJoseph Koshy 	    "\t -o file\t send print output to \"file\"\n"
517f263522aSJoseph Koshy 	    "\t -p spec\t allocate a process-private counting PMC\n"
51849874f6eSJoseph Koshy 	    "\t -q\t\t suppress verbosity\n"
51949874f6eSJoseph Koshy 	    "\t -r fsroot\t specify FS root directory\n"
520f263522aSJoseph Koshy 	    "\t -s spec\t allocate a system-wide counting PMC\n"
52167442c6eSJoseph Koshy 	    "\t -t process-spec attach to running processes matching "
52267442c6eSJoseph Koshy 		"\"process-spec\"\n"
52349874f6eSJoseph Koshy 	    "\t -v\t\t increase verbosity\n"
524b6010f9eSJoseph Koshy 	    "\t -w secs\t set printing time interval\n"
525b6010f9eSJoseph Koshy 	    "\t -z depth\t limit callchain display depth"
526ebccf1e3SJoseph Koshy 	);
527ebccf1e3SJoseph Koshy }
528ebccf1e3SJoseph Koshy 
529ebccf1e3SJoseph Koshy /*
5300b86b1bbSFabien Thomas  * At exit handler for top mode
5310b86b1bbSFabien Thomas  */
5320b86b1bbSFabien Thomas 
5330b86b1bbSFabien Thomas void
5340b86b1bbSFabien Thomas pmcstat_topexit(void)
5350b86b1bbSFabien Thomas {
5360b86b1bbSFabien Thomas 	if (!args.pa_toptty)
5370b86b1bbSFabien Thomas 		return;
5380b86b1bbSFabien Thomas 
5390b86b1bbSFabien Thomas 	/*
5400b86b1bbSFabien Thomas 	 * Shutdown ncurses.
5410b86b1bbSFabien Thomas 	 */
5420b86b1bbSFabien Thomas 	clrtoeol();
5430b86b1bbSFabien Thomas 	refresh();
5440b86b1bbSFabien Thomas 	endwin();
5450b86b1bbSFabien Thomas }
5460b86b1bbSFabien Thomas 
5470b86b1bbSFabien Thomas /*
548ebccf1e3SJoseph Koshy  * Main
549ebccf1e3SJoseph Koshy  */
550ebccf1e3SJoseph Koshy 
551ebccf1e3SJoseph Koshy int
552ebccf1e3SJoseph Koshy main(int argc, char **argv)
553ebccf1e3SJoseph Koshy {
554ef16b9abSJohn Baldwin 	cpuset_t cpumask, rootmask;
555ebccf1e3SJoseph Koshy 	double interval;
55672fe532cSGeorge V. Neville-Neil 	double duration;
557ef16b9abSJohn Baldwin 	int option, npmc;
558831b9630SEitan Adler 	int c, check_driver_stats, current_sampling_count;
559b6010f9eSJoseph Koshy 	int do_callchain, do_descendants, do_logproccsw, do_logprocexit;
560dceed24aSFabien Thomas 	int do_print, do_read;
56138fd3834SJohn Baldwin 	size_t len;
562b6010f9eSJoseph Koshy 	int graphdepth;
563ac708fddSFabien Thomas 	int pipefd[2], rfd;
564f263522aSJoseph Koshy 	int use_cumulative_counts;
5650b86b1bbSFabien Thomas 	short cf, cb;
56649874f6eSJoseph Koshy 	char *end, *tmp;
567b6010f9eSJoseph Koshy 	const char *errmsg, *graphfilename;
568f263522aSJoseph Koshy 	enum pmcstat_state runstate;
5697fbf3da2SJoseph Koshy 	struct pmc_driverstats ds_start, ds_end;
570ebccf1e3SJoseph Koshy 	struct pmcstat_ev *ev;
571ebccf1e3SJoseph Koshy 	struct sigaction sa;
572ebccf1e3SJoseph Koshy 	struct kevent kev;
573ebccf1e3SJoseph Koshy 	struct winsize ws;
57415139246SJoseph Koshy 	struct stat sb;
57549874f6eSJoseph Koshy 	char buffer[PATH_MAX];
576ebccf1e3SJoseph Koshy 
5777fbf3da2SJoseph Koshy 	check_driver_stats      = 0;
578ebccf1e3SJoseph Koshy 	current_sampling_count  = DEFAULT_SAMPLE_COUNT;
579b6010f9eSJoseph Koshy 	do_callchain		= 1;
580ebccf1e3SJoseph Koshy 	do_descendants          = 0;
581f263522aSJoseph Koshy 	do_logproccsw           = 0;
582f263522aSJoseph Koshy 	do_logprocexit          = 0;
583ebccf1e3SJoseph Koshy 	use_cumulative_counts   = 0;
584b6010f9eSJoseph Koshy 	graphfilename		= "-";
585f263522aSJoseph Koshy 	args.pa_required	= 0;
586ebccf1e3SJoseph Koshy 	args.pa_flags		= 0;
58749874f6eSJoseph Koshy 	args.pa_verbosity	= 1;
58815139246SJoseph Koshy 	args.pa_logfd		= -1;
58949874f6eSJoseph Koshy 	args.pa_fsroot		= "";
59015139246SJoseph Koshy 	args.pa_samplesdir	= ".";
59115139246SJoseph Koshy 	args.pa_printfile	= stderr;
592b6010f9eSJoseph Koshy 	args.pa_graphdepth	= DEFAULT_CALLGRAPH_DEPTH;
593b6010f9eSJoseph Koshy 	args.pa_graphfile	= NULL;
594ebccf1e3SJoseph Koshy 	args.pa_interval	= DEFAULT_WAIT_INTERVAL;
59549874f6eSJoseph Koshy 	args.pa_mapfilename	= NULL;
596b6010f9eSJoseph Koshy 	args.pa_inputpath	= NULL;
597b6010f9eSJoseph Koshy 	args.pa_outputpath	= NULL;
5980b86b1bbSFabien Thomas 	args.pa_pplugin		= PMCSTAT_PL_NONE;
5990b86b1bbSFabien Thomas 	args.pa_plugin		= PMCSTAT_PL_NONE;
6000b86b1bbSFabien Thomas 	args.pa_ctdumpinstr	= 1;
6010b86b1bbSFabien Thomas 	args.pa_topmode		= PMCSTAT_TOP_DELTA;
6020b86b1bbSFabien Thomas 	args.pa_toptty		= 0;
6030b86b1bbSFabien Thomas 	args.pa_topcolor	= 0;
6040b86b1bbSFabien Thomas 	args.pa_mergepmc	= 0;
60572fe532cSGeorge V. Neville-Neil 	args.pa_duration	= 0.0;
606fbd8b34aSJoseph Koshy 	STAILQ_INIT(&args.pa_events);
607fbd8b34aSJoseph Koshy 	SLIST_INIT(&args.pa_targets);
608bea6aa93SJoseph Koshy 	bzero(&ds_start, sizeof(ds_start));
609bea6aa93SJoseph Koshy 	bzero(&ds_end, sizeof(ds_end));
610ebccf1e3SJoseph Koshy 	ev = NULL;
611ea056888SAttilio Rao 	CPU_ZERO(&cpumask);
612ebccf1e3SJoseph Koshy 
61338fd3834SJohn Baldwin 	/* Default to using the running system kernel. */
61438fd3834SJohn Baldwin 	len = 0;
61538fd3834SJohn Baldwin 	if (sysctlbyname("kern.bootfile", NULL, &len, NULL, 0) == -1)
61638fd3834SJohn Baldwin 		err(EX_OSERR, "ERROR: Cannot determine path of running kernel");
61738fd3834SJohn Baldwin 	args.pa_kernel = malloc(len + 1);
61838fd3834SJohn Baldwin 	if (sysctlbyname("kern.bootfile", args.pa_kernel, &len, NULL, 0) == -1)
61938fd3834SJohn Baldwin 		err(EX_OSERR, "ERROR: Cannot determine path of running kernel");
62038fd3834SJohn Baldwin 
621b6010f9eSJoseph Koshy 	/*
622ef16b9abSJohn Baldwin 	 * The initial CPU mask specifies the root mask of this process
623ef16b9abSJohn Baldwin 	 * which is usually all CPUs in the system.
624b6010f9eSJoseph Koshy 	 */
625ef16b9abSJohn Baldwin 	if (cpuset_getaffinity(CPU_LEVEL_ROOT, CPU_WHICH_PID, -1,
626ef16b9abSJohn Baldwin 	    sizeof(rootmask), &rootmask) == -1)
627ef16b9abSJohn Baldwin 		err(EX_OSERR, "ERROR: Cannot determine the root set of CPUs");
628ef16b9abSJohn Baldwin 	CPU_COPY(&rootmask, &cpumask);
629ee4910b1SJoseph Koshy 
63049874f6eSJoseph Koshy 	while ((option = getopt(argc, argv,
631*465dadb5SJonathan T. Looney 	    "CD:EF:G:M:NO:P:R:S:TWa:c:def:gk:l:m:n:o:p:qr:s:t:vw:z:")) != -1)
632ebccf1e3SJoseph Koshy 		switch (option) {
63385ec49f3SAdrian Chadd 		case 'a':	/* Annotate + callgraph */
63485ec49f3SAdrian Chadd 			args.pa_flags |= FLAG_DO_ANNOTATE;
63585ec49f3SAdrian Chadd 			args.pa_plugin = PMCSTAT_PL_ANNOTATE_CG;
63685ec49f3SAdrian Chadd 			graphfilename  = optarg;
63785ec49f3SAdrian Chadd 			break;
63885ec49f3SAdrian Chadd 
639ebccf1e3SJoseph Koshy 		case 'C':	/* cumulative values */
640ebccf1e3SJoseph Koshy 			use_cumulative_counts = !use_cumulative_counts;
641f263522aSJoseph Koshy 			args.pa_required |= FLAG_HAS_COUNTING_PMCS;
642ebccf1e3SJoseph Koshy 			break;
643ebccf1e3SJoseph Koshy 
644ebccf1e3SJoseph Koshy 		case 'c':	/* CPU */
645ef16b9abSJohn Baldwin 			if (optarg[0] == '*' && optarg[1] == '\0')
646ef16b9abSJohn Baldwin 				CPU_COPY(&rootmask, &cpumask);
647ef16b9abSJohn Baldwin 			else
648ea056888SAttilio Rao 				pmcstat_get_cpumask(optarg, &cpumask);
649ee4910b1SJoseph Koshy 
650e7ef1de5SFabien Thomas 			args.pa_flags	 |= FLAGS_HAS_CPUMASK;
651f263522aSJoseph Koshy 			args.pa_required |= FLAG_HAS_SYSTEM_PMCS;
652ebccf1e3SJoseph Koshy 			break;
653ebccf1e3SJoseph Koshy 
65415139246SJoseph Koshy 		case 'D':
65515139246SJoseph Koshy 			if (stat(optarg, &sb) < 0)
65615139246SJoseph Koshy 				err(EX_OSERR, "ERROR: Cannot stat \"%s\"",
65715139246SJoseph Koshy 				    optarg);
65815139246SJoseph Koshy 			if (!S_ISDIR(sb.st_mode))
65937d6f8a9SDavid E. O'Brien 				errx(EX_USAGE,
66037d6f8a9SDavid E. O'Brien 				    "ERROR: \"%s\" is not a directory.",
66137d6f8a9SDavid E. O'Brien 				    optarg);
66215139246SJoseph Koshy 			args.pa_samplesdir = optarg;
66315139246SJoseph Koshy 			args.pa_flags     |= FLAG_HAS_SAMPLESDIR;
66415139246SJoseph Koshy 			args.pa_required  |= FLAG_DO_GPROF;
66515139246SJoseph Koshy 			break;
66615139246SJoseph Koshy 
667ebccf1e3SJoseph Koshy 		case 'd':	/* toggle descendents */
668ebccf1e3SJoseph Koshy 			do_descendants = !do_descendants;
669f263522aSJoseph Koshy 			args.pa_required |= FLAG_HAS_PROCESS_PMCS;
670f263522aSJoseph Koshy 			break;
671f263522aSJoseph Koshy 
672*465dadb5SJonathan T. Looney 		case 'e':	/* wide gprof metrics */
673*465dadb5SJonathan T. Looney 			args.pa_flags |= FLAG_DO_WIDE_GPROF_HC;
674*465dadb5SJonathan T. Looney 			break;
675*465dadb5SJonathan T. Looney 
6760b86b1bbSFabien Thomas 		case 'F':	/* produce a system-wide calltree */
6770b86b1bbSFabien Thomas 			args.pa_flags |= FLAG_DO_CALLGRAPHS;
6780b86b1bbSFabien Thomas 			args.pa_plugin = PMCSTAT_PL_CALLTREE;
6790b86b1bbSFabien Thomas 			graphfilename = optarg;
6800b86b1bbSFabien Thomas 			break;
6810b86b1bbSFabien Thomas 
6820b86b1bbSFabien Thomas 		case 'f':	/* plugins options */
6830b86b1bbSFabien Thomas 			if (args.pa_plugin == PMCSTAT_PL_NONE)
6840b86b1bbSFabien Thomas 				err(EX_USAGE, "ERROR: Need -g/-G/-m/-T.");
6850b86b1bbSFabien Thomas 			pmcstat_pluginconfigure_log(optarg);
6860b86b1bbSFabien Thomas 			break;
6870b86b1bbSFabien Thomas 
688b6010f9eSJoseph Koshy 		case 'G':	/* produce a system-wide callgraph */
689b6010f9eSJoseph Koshy 			args.pa_flags |= FLAG_DO_CALLGRAPHS;
6900b86b1bbSFabien Thomas 			args.pa_plugin = PMCSTAT_PL_CALLGRAPH;
691b6010f9eSJoseph Koshy 			graphfilename = optarg;
692b6010f9eSJoseph Koshy 			break;
693b6010f9eSJoseph Koshy 
694f263522aSJoseph Koshy 		case 'g':	/* produce gprof compatible profiles */
695f263522aSJoseph Koshy 			args.pa_flags |= FLAG_DO_GPROF;
6960b86b1bbSFabien Thomas 			args.pa_pplugin = PMCSTAT_PL_CALLGRAPH;
6970b86b1bbSFabien Thomas 			args.pa_plugin	= PMCSTAT_PL_GPROF;
698f263522aSJoseph Koshy 			break;
699f263522aSJoseph Koshy 
70015139246SJoseph Koshy 		case 'k':	/* pathname to the kernel */
70149874f6eSJoseph Koshy 			free(args.pa_kernel);
70249874f6eSJoseph Koshy 			args.pa_kernel = strdup(optarg);
703b6010f9eSJoseph Koshy 			args.pa_required |= FLAG_DO_ANALYSIS;
70415139246SJoseph Koshy 			args.pa_flags    |= FLAG_HAS_KERNELPATH;
705f263522aSJoseph Koshy 			break;
706f263522aSJoseph Koshy 
70772fe532cSGeorge V. Neville-Neil 		case 'l':	/* time duration in seconds */
70872fe532cSGeorge V. Neville-Neil 			duration = strtod(optarg, &end);
70972fe532cSGeorge V. Neville-Neil 			if (*end != '\0' || duration <= 0)
71072fe532cSGeorge V. Neville-Neil 				errx(EX_USAGE, "ERROR: Illegal duration time "
71172fe532cSGeorge V. Neville-Neil 				    "value \"%s\".", optarg);
71272fe532cSGeorge V. Neville-Neil 			args.pa_flags |= FLAG_HAS_DURATION;
71372fe532cSGeorge V. Neville-Neil 			args.pa_duration = duration;
71472fe532cSGeorge V. Neville-Neil 			break;
71572fe532cSGeorge V. Neville-Neil 
716c0252222SAttilio Rao 		case 'm':
7170b86b1bbSFabien Thomas 			args.pa_flags |= FLAG_DO_ANNOTATE;
7180b86b1bbSFabien Thomas 			args.pa_plugin = PMCSTAT_PL_ANNOTATE;
719c0252222SAttilio Rao 			graphfilename  = optarg;
720c0252222SAttilio Rao 			break;
721c0252222SAttilio Rao 
722f263522aSJoseph Koshy 		case 'E':	/* log process exit */
723f263522aSJoseph Koshy 			do_logprocexit = !do_logprocexit;
724f263522aSJoseph Koshy 			args.pa_required |= (FLAG_HAS_PROCESS_PMCS |
72515139246SJoseph Koshy 			    FLAG_HAS_COUNTING_PMCS | FLAG_HAS_OUTPUT_LOGFILE);
726ebccf1e3SJoseph Koshy 			break;
727ebccf1e3SJoseph Koshy 
72849874f6eSJoseph Koshy 		case 'M':	/* mapfile */
72949874f6eSJoseph Koshy 			args.pa_mapfilename = optarg;
73049874f6eSJoseph Koshy 			break;
73149874f6eSJoseph Koshy 
732b6010f9eSJoseph Koshy 		case 'N':
733b6010f9eSJoseph Koshy 			do_callchain = !do_callchain;
734b6010f9eSJoseph Koshy 			args.pa_required |= FLAG_HAS_SAMPLING_PMCS;
735b6010f9eSJoseph Koshy 			break;
736b6010f9eSJoseph Koshy 
737ebccf1e3SJoseph Koshy 		case 'p':	/* process virtual counting PMC */
738ebccf1e3SJoseph Koshy 		case 's':	/* system-wide counting PMC */
739ebccf1e3SJoseph Koshy 		case 'P':	/* process virtual sampling PMC */
740ebccf1e3SJoseph Koshy 		case 'S':	/* system-wide sampling PMC */
741ebccf1e3SJoseph Koshy 			if ((ev = malloc(sizeof(*ev))) == NULL)
742f263522aSJoseph Koshy 				errx(EX_SOFTWARE, "ERROR: Out of memory.");
743ebccf1e3SJoseph Koshy 
744ebccf1e3SJoseph Koshy 			switch (option) {
745ebccf1e3SJoseph Koshy 			case 'p': ev->ev_mode = PMC_MODE_TC; break;
746ebccf1e3SJoseph Koshy 			case 's': ev->ev_mode = PMC_MODE_SC; break;
747ebccf1e3SJoseph Koshy 			case 'P': ev->ev_mode = PMC_MODE_TS; break;
748ebccf1e3SJoseph Koshy 			case 'S': ev->ev_mode = PMC_MODE_SS; break;
749ebccf1e3SJoseph Koshy 			}
750ebccf1e3SJoseph Koshy 
751f263522aSJoseph Koshy 			if (option == 'P' || option == 'p') {
752f263522aSJoseph Koshy 				args.pa_flags |= FLAG_HAS_PROCESS_PMCS;
75315139246SJoseph Koshy 				args.pa_required |= (FLAG_HAS_COMMANDLINE |
754fbd8b34aSJoseph Koshy 				    FLAG_HAS_TARGET);
755f263522aSJoseph Koshy 			}
756ebccf1e3SJoseph Koshy 
757f263522aSJoseph Koshy 			if (option == 'P' || option == 'S') {
758f263522aSJoseph Koshy 				args.pa_flags |= FLAG_HAS_SAMPLING_PMCS;
75915139246SJoseph Koshy 				args.pa_required |= (FLAG_HAS_PIPE |
76015139246SJoseph Koshy 				    FLAG_HAS_OUTPUT_LOGFILE);
761f263522aSJoseph Koshy 			}
762ebccf1e3SJoseph Koshy 
763ebccf1e3SJoseph Koshy 			if (option == 'p' || option == 's')
764f263522aSJoseph Koshy 				args.pa_flags |= FLAG_HAS_COUNTING_PMCS;
765f263522aSJoseph Koshy 
766f263522aSJoseph Koshy 			if (option == 's' || option == 'S')
767f263522aSJoseph Koshy 				args.pa_flags |= FLAG_HAS_SYSTEM_PMCS;
768ebccf1e3SJoseph Koshy 
769ebccf1e3SJoseph Koshy 			ev->ev_spec  = strdup(optarg);
770ebccf1e3SJoseph Koshy 
771ebccf1e3SJoseph Koshy 			if (option == 'S' || option == 'P')
772ebccf1e3SJoseph Koshy 				ev->ev_count = current_sampling_count;
773ebccf1e3SJoseph Koshy 			else
774ebccf1e3SJoseph Koshy 				ev->ev_count = -1;
775ebccf1e3SJoseph Koshy 
776ef16b9abSJohn Baldwin 			if (option == 'S' || option == 's')
7777816cb4aSJohn Baldwin 				ev->ev_cpu = CPU_FFS(&cpumask) - 1;
778ef16b9abSJohn Baldwin 			else
779ebccf1e3SJoseph Koshy 				ev->ev_cpu = PMC_CPU_ANY;
780ebccf1e3SJoseph Koshy 
781f263522aSJoseph Koshy 			ev->ev_flags = 0;
782b6010f9eSJoseph Koshy 			if (do_callchain)
783b6010f9eSJoseph Koshy 				ev->ev_flags |= PMC_F_CALLCHAIN;
784f263522aSJoseph Koshy 			if (do_descendants)
785f263522aSJoseph Koshy 				ev->ev_flags |= PMC_F_DESCENDANTS;
786f263522aSJoseph Koshy 			if (do_logprocexit)
787f263522aSJoseph Koshy 				ev->ev_flags |= PMC_F_LOG_PROCEXIT;
788f263522aSJoseph Koshy 			if (do_logproccsw)
789f263522aSJoseph Koshy 				ev->ev_flags |= PMC_F_LOG_PROCCSW;
790f263522aSJoseph Koshy 
791ebccf1e3SJoseph Koshy 			ev->ev_cumulative  = use_cumulative_counts;
792ebccf1e3SJoseph Koshy 
793ebccf1e3SJoseph Koshy 			ev->ev_saved = 0LL;
794ebccf1e3SJoseph Koshy 			ev->ev_pmcid = PMC_ID_INVALID;
795ebccf1e3SJoseph Koshy 
796ebccf1e3SJoseph Koshy 			/* extract event name */
797ebccf1e3SJoseph Koshy 			c = strcspn(optarg, ", \t");
798ebccf1e3SJoseph Koshy 			ev->ev_name = malloc(c + 1);
799ebccf1e3SJoseph Koshy 			(void) strncpy(ev->ev_name, optarg, c);
800ebccf1e3SJoseph Koshy 			*(ev->ev_name + c) = '\0';
801ebccf1e3SJoseph Koshy 
802fbd8b34aSJoseph Koshy 			STAILQ_INSERT_TAIL(&args.pa_events, ev, ev_next);
803ebccf1e3SJoseph Koshy 
804ea056888SAttilio Rao 			if (option == 's' || option == 'S') {
805ea056888SAttilio Rao 				CPU_CLR(ev->ev_cpu, &cpumask);
806ea056888SAttilio Rao 				pmcstat_clone_event_descriptor(ev, &cpumask);
807ea056888SAttilio Rao 				CPU_SET(ev->ev_cpu, &cpumask);
808ea056888SAttilio Rao 			}
809ee4910b1SJoseph Koshy 
810ebccf1e3SJoseph Koshy 			break;
811ebccf1e3SJoseph Koshy 
812ebccf1e3SJoseph Koshy 		case 'n':	/* sampling count */
813ebccf1e3SJoseph Koshy 			current_sampling_count = strtol(optarg, &end, 0);
814ebccf1e3SJoseph Koshy 			if (*end != '\0' || current_sampling_count <= 0)
815ebccf1e3SJoseph Koshy 				errx(EX_USAGE,
816f263522aSJoseph Koshy 				    "ERROR: Illegal count value \"%s\".",
817ebccf1e3SJoseph Koshy 				    optarg);
818f263522aSJoseph Koshy 			args.pa_required |= FLAG_HAS_SAMPLING_PMCS;
819ebccf1e3SJoseph Koshy 			break;
820ebccf1e3SJoseph Koshy 
821ebccf1e3SJoseph Koshy 		case 'o':	/* outputfile */
82266ce9936SFabien Thomas 			if (args.pa_printfile != NULL &&
82366ce9936SFabien Thomas 			    args.pa_printfile != stdout &&
82466ce9936SFabien Thomas 			    args.pa_printfile != stderr)
82515139246SJoseph Koshy 				(void) fclose(args.pa_printfile);
82615139246SJoseph Koshy 			if ((args.pa_printfile = fopen(optarg, "w")) == NULL)
82737d6f8a9SDavid E. O'Brien 				errx(EX_OSERR,
82837d6f8a9SDavid E. O'Brien 				    "ERROR: cannot open \"%s\" for writing.",
82937d6f8a9SDavid E. O'Brien 				    optarg);
83015139246SJoseph Koshy 			args.pa_flags |= FLAG_DO_PRINT;
831f263522aSJoseph Koshy 			break;
832ebccf1e3SJoseph Koshy 
833ebccf1e3SJoseph Koshy 		case 'O':	/* sampling output */
83415139246SJoseph Koshy 			if (args.pa_outputpath)
83537d6f8a9SDavid E. O'Brien 				errx(EX_USAGE,
83637d6f8a9SDavid E. O'Brien "ERROR: option -O may only be specified once.");
83715139246SJoseph Koshy 			args.pa_outputpath = optarg;
83815139246SJoseph Koshy 			args.pa_flags |= FLAG_HAS_OUTPUT_LOGFILE;
83915139246SJoseph Koshy 			break;
84015139246SJoseph Koshy 
84149874f6eSJoseph Koshy 		case 'q':	/* quiet mode */
84249874f6eSJoseph Koshy 			args.pa_verbosity = 0;
84349874f6eSJoseph Koshy 			break;
84449874f6eSJoseph Koshy 
84549874f6eSJoseph Koshy 		case 'r':	/* root FS path */
84649874f6eSJoseph Koshy 			args.pa_fsroot = optarg;
84749874f6eSJoseph Koshy 			break;
84849874f6eSJoseph Koshy 
84915139246SJoseph Koshy 		case 'R':	/* read an existing log file */
850b6010f9eSJoseph Koshy 			if (args.pa_inputpath != NULL)
85137d6f8a9SDavid E. O'Brien 				errx(EX_USAGE,
85237d6f8a9SDavid E. O'Brien "ERROR: option -R may only be specified once.");
85315139246SJoseph Koshy 			args.pa_inputpath = optarg;
85415139246SJoseph Koshy 			if (args.pa_printfile == stderr)
85515139246SJoseph Koshy 				args.pa_printfile = stdout;
85615139246SJoseph Koshy 			args.pa_flags |= FLAG_READ_LOGFILE;
857ebccf1e3SJoseph Koshy 			break;
858ebccf1e3SJoseph Koshy 
859fbd8b34aSJoseph Koshy 		case 't':	/* target pid or process name */
8600b86b1bbSFabien Thomas 			pmcstat_find_targets(optarg);
861ebccf1e3SJoseph Koshy 
862fbd8b34aSJoseph Koshy 			args.pa_flags |= FLAG_HAS_TARGET;
863f263522aSJoseph Koshy 			args.pa_required |= FLAG_HAS_PROCESS_PMCS;
864ebccf1e3SJoseph Koshy 			break;
865ebccf1e3SJoseph Koshy 
8660b86b1bbSFabien Thomas 		case 'T':	/* top mode */
8670b86b1bbSFabien Thomas 			args.pa_flags |= FLAG_DO_TOP;
8680b86b1bbSFabien Thomas 			args.pa_plugin = PMCSTAT_PL_CALLGRAPH;
8690b86b1bbSFabien Thomas 			args.pa_ctdumpinstr = 0;
8700b86b1bbSFabien Thomas 			args.pa_mergepmc = 1;
8710b86b1bbSFabien Thomas 			if (args.pa_printfile == stderr)
8720b86b1bbSFabien Thomas 				args.pa_printfile = stdout;
8730b86b1bbSFabien Thomas 			break;
8740b86b1bbSFabien Thomas 
87549874f6eSJoseph Koshy 		case 'v':	/* verbose */
87649874f6eSJoseph Koshy 			args.pa_verbosity++;
87749874f6eSJoseph Koshy 			break;
87849874f6eSJoseph Koshy 
879ebccf1e3SJoseph Koshy 		case 'w':	/* wait interval */
880ebccf1e3SJoseph Koshy 			interval = strtod(optarg, &end);
881ebccf1e3SJoseph Koshy 			if (*end != '\0' || interval <= 0)
88237d6f8a9SDavid E. O'Brien 				errx(EX_USAGE,
88337d6f8a9SDavid E. O'Brien "ERROR: Illegal wait interval value \"%s\".",
88437d6f8a9SDavid E. O'Brien 				    optarg);
885ebccf1e3SJoseph Koshy 			args.pa_flags |= FLAG_HAS_WAIT_INTERVAL;
886ebccf1e3SJoseph Koshy 			args.pa_interval = interval;
887f263522aSJoseph Koshy 			break;
888ebccf1e3SJoseph Koshy 
889f263522aSJoseph Koshy 		case 'W':	/* toggle LOG_CSW */
890f263522aSJoseph Koshy 			do_logproccsw = !do_logproccsw;
891f263522aSJoseph Koshy 			args.pa_required |= (FLAG_HAS_PROCESS_PMCS |
89215139246SJoseph Koshy 			    FLAG_HAS_COUNTING_PMCS | FLAG_HAS_OUTPUT_LOGFILE);
893ebccf1e3SJoseph Koshy 			break;
894ebccf1e3SJoseph Koshy 
895b6010f9eSJoseph Koshy 		case 'z':
896b6010f9eSJoseph Koshy 			graphdepth = strtod(optarg, &end);
897b6010f9eSJoseph Koshy 			if (*end != '\0' || graphdepth <= 0)
89837d6f8a9SDavid E. O'Brien 				errx(EX_USAGE,
89937d6f8a9SDavid E. O'Brien 				    "ERROR: Illegal callchain depth \"%s\".",
90037d6f8a9SDavid E. O'Brien 				    optarg);
901b6010f9eSJoseph Koshy 			args.pa_graphdepth = graphdepth;
902b6010f9eSJoseph Koshy 			args.pa_required |= FLAG_DO_CALLGRAPHS;
903b6010f9eSJoseph Koshy 			break;
904b6010f9eSJoseph Koshy 
905ebccf1e3SJoseph Koshy 		case '?':
906ebccf1e3SJoseph Koshy 		default:
907ebccf1e3SJoseph Koshy 			pmcstat_show_usage();
908ebccf1e3SJoseph Koshy 			break;
909ebccf1e3SJoseph Koshy 
910ebccf1e3SJoseph Koshy 		}
911ebccf1e3SJoseph Koshy 
912ebccf1e3SJoseph Koshy 	args.pa_argc = (argc -= optind);
913ebccf1e3SJoseph Koshy 	args.pa_argv = (argv += optind);
914ebccf1e3SJoseph Koshy 
915e7ef1de5SFabien Thomas 	/* If we read from logfile and no specified CPU mask use
916e7ef1de5SFabien Thomas 	 * the maximum CPU count.
917e7ef1de5SFabien Thomas 	 */
918e7ef1de5SFabien Thomas 	if ((args.pa_flags & FLAG_READ_LOGFILE) &&
919e7ef1de5SFabien Thomas 	    (args.pa_flags & FLAGS_HAS_CPUMASK) == 0)
920ea056888SAttilio Rao 		CPU_FILL(&cpumask);
921e7ef1de5SFabien Thomas 
922b6010f9eSJoseph Koshy 	args.pa_cpumask = cpumask; /* For selecting CPUs using -R. */
923b6010f9eSJoseph Koshy 
92415139246SJoseph Koshy 	if (argc)	/* command line present */
92515139246SJoseph Koshy 		args.pa_flags |= FLAG_HAS_COMMANDLINE;
926ebccf1e3SJoseph Koshy 
927c0252222SAttilio Rao 	if (args.pa_flags & (FLAG_DO_GPROF | FLAG_DO_CALLGRAPHS |
9280b86b1bbSFabien Thomas 	    FLAG_DO_ANNOTATE | FLAG_DO_TOP))
929b6010f9eSJoseph Koshy 		args.pa_flags |= FLAG_DO_ANALYSIS;
930b6010f9eSJoseph Koshy 
931ebccf1e3SJoseph Koshy 	/*
932ebccf1e3SJoseph Koshy 	 * Check invocation syntax.
933ebccf1e3SJoseph Koshy 	 */
934ebccf1e3SJoseph Koshy 
93515139246SJoseph Koshy 	/* disallow -O and -R together */
93615139246SJoseph Koshy 	if (args.pa_outputpath && args.pa_inputpath)
93737d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
93837d6f8a9SDavid E. O'Brien 		    "ERROR: options -O and -R are mutually exclusive.");
93915139246SJoseph Koshy 
94072fe532cSGeorge V. Neville-Neil 	/* disallow -T and -l together */
94172fe532cSGeorge V. Neville-Neil 	if ((args.pa_flags & FLAG_HAS_DURATION) &&
94272fe532cSGeorge V. Neville-Neil 	    (args.pa_flags & FLAG_DO_TOP))
94372fe532cSGeorge V. Neville-Neil 		errx(EX_USAGE, "ERROR: options -T and -l are mutually "
94472fe532cSGeorge V. Neville-Neil 		    "exclusive.");
94572fe532cSGeorge V. Neville-Neil 
94610813ab3SEric van Gyzen 	/* -a and -m require -R */
9470b86b1bbSFabien Thomas 	if (args.pa_flags & FLAG_DO_ANNOTATE && args.pa_inputpath == NULL)
94885ec49f3SAdrian Chadd 		errx(EX_USAGE, "ERROR: option %s requires an input file",
94985ec49f3SAdrian Chadd 		    args.pa_plugin == PMCSTAT_PL_ANNOTATE ? "-m" : "-a");
950c0252222SAttilio Rao 
951c0252222SAttilio Rao 	/* -m option is not allowed combined with -g or -G. */
9520b86b1bbSFabien Thomas 	if (args.pa_flags & FLAG_DO_ANNOTATE &&
953c0252222SAttilio Rao 	    args.pa_flags & (FLAG_DO_GPROF | FLAG_DO_CALLGRAPHS))
95437d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
95537d6f8a9SDavid E. O'Brien 		    "ERROR: option -m and -g | -G are mutually exclusive");
956c0252222SAttilio Rao 
95715139246SJoseph Koshy 	if (args.pa_flags & FLAG_READ_LOGFILE) {
958f263522aSJoseph Koshy 		errmsg = NULL;
95915139246SJoseph Koshy 		if (args.pa_flags & FLAG_HAS_COMMANDLINE)
960f263522aSJoseph Koshy 			errmsg = "a command line specification";
961fbd8b34aSJoseph Koshy 		else if (args.pa_flags & FLAG_HAS_TARGET)
962f263522aSJoseph Koshy 			errmsg = "option -t";
963fbd8b34aSJoseph Koshy 		else if (!STAILQ_EMPTY(&args.pa_events))
964f263522aSJoseph Koshy 			errmsg = "a PMC event specification";
965f263522aSJoseph Koshy 		if (errmsg)
96637d6f8a9SDavid E. O'Brien 			errx(EX_USAGE,
96737d6f8a9SDavid E. O'Brien 			    "ERROR: option -R may not be used with %s.",
96837d6f8a9SDavid E. O'Brien 			    errmsg);
969fbd8b34aSJoseph Koshy 	} else if (STAILQ_EMPTY(&args.pa_events))
97015139246SJoseph Koshy 		/* All other uses require a PMC spec. */
971ebccf1e3SJoseph Koshy 		pmcstat_show_usage();
972ebccf1e3SJoseph Koshy 
973f263522aSJoseph Koshy 	/* check for -t pid without a process PMC spec */
974fbd8b34aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_TARGET) &&
975f263522aSJoseph Koshy 	    (args.pa_flags & FLAG_HAS_PROCESS_PMCS) == 0)
97637d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
97737d6f8a9SDavid E. O'Brien "ERROR: option -t requires a process mode PMC to be specified."
97837d6f8a9SDavid E. O'Brien 		    );
979f263522aSJoseph Koshy 
980f263522aSJoseph Koshy 	/* check for process-mode options without a command or -t pid */
981f263522aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_PROCESS_PMCS) &&
982fbd8b34aSJoseph Koshy 	    (args.pa_flags & (FLAG_HAS_COMMANDLINE | FLAG_HAS_TARGET)) == 0)
98337d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
98437d6f8a9SDavid E. O'Brien "ERROR: options -d, -E, -p, -P, and -W require a command line or target process."
98537d6f8a9SDavid E. O'Brien 		    );
986f263522aSJoseph Koshy 
987f263522aSJoseph Koshy 	/* check for -p | -P without a target process of some sort */
988fbd8b34aSJoseph Koshy 	if ((args.pa_required & (FLAG_HAS_COMMANDLINE | FLAG_HAS_TARGET)) &&
989fbd8b34aSJoseph Koshy 	    (args.pa_flags & (FLAG_HAS_COMMANDLINE | FLAG_HAS_TARGET)) == 0)
99037d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
99137d6f8a9SDavid E. O'Brien "ERROR: options -P and -p require a target process or a command line."
99237d6f8a9SDavid E. O'Brien 		    );
993f263522aSJoseph Koshy 
994f263522aSJoseph Koshy 	/* check for process-mode options without a process-mode PMC */
995f263522aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_PROCESS_PMCS) &&
996f263522aSJoseph Koshy 	    (args.pa_flags & FLAG_HAS_PROCESS_PMCS) == 0)
99737d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
99837d6f8a9SDavid E. O'Brien "ERROR: options -d, -E, and -W require a process mode PMC to be specified."
99937d6f8a9SDavid E. O'Brien 		    );
1000f263522aSJoseph Koshy 
1001b6010f9eSJoseph Koshy 	/* check for -c cpu with no system mode PMCs or logfile. */
1002f263522aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_SYSTEM_PMCS) &&
1003b6010f9eSJoseph Koshy 	    (args.pa_flags & FLAG_HAS_SYSTEM_PMCS) == 0 &&
1004b6010f9eSJoseph Koshy 	    (args.pa_flags & FLAG_READ_LOGFILE) == 0)
100537d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
100637d6f8a9SDavid E. O'Brien "ERROR: option -c requires at least one system mode PMC to be specified."
100737d6f8a9SDavid E. O'Brien 		    );
1008f263522aSJoseph Koshy 
1009f263522aSJoseph Koshy 	/* check for counting mode options without a counting PMC */
1010f263522aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_COUNTING_PMCS) &&
1011f263522aSJoseph Koshy 	    (args.pa_flags & FLAG_HAS_COUNTING_PMCS) == 0)
101237d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
101337d6f8a9SDavid E. O'Brien "ERROR: options -C, -W and -o require at least one counting mode PMC to be specified."
101437d6f8a9SDavid E. O'Brien 		    );
1015f263522aSJoseph Koshy 
1016f263522aSJoseph Koshy 	/* check for sampling mode options without a sampling PMC spec */
1017f263522aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_SAMPLING_PMCS) &&
1018f263522aSJoseph Koshy 	    (args.pa_flags & FLAG_HAS_SAMPLING_PMCS) == 0)
101937d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
102037d6f8a9SDavid E. O'Brien "ERROR: options -N, -n and -O require at least one sampling mode PMC to be specified."
102137d6f8a9SDavid E. O'Brien 		    );
1022f263522aSJoseph Koshy 
10230b86b1bbSFabien Thomas 	/* check if -g/-G/-m/-T are being used correctly */
1024b6010f9eSJoseph Koshy 	if ((args.pa_flags & FLAG_DO_ANALYSIS) &&
102515139246SJoseph Koshy 	    !(args.pa_flags & (FLAG_HAS_SAMPLING_PMCS|FLAG_READ_LOGFILE)))
102637d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
102737d6f8a9SDavid E. O'Brien "ERROR: options -g/-G/-m/-T require sampling PMCs or -R to be specified."
102837d6f8a9SDavid E. O'Brien 		    );
102915139246SJoseph Koshy 
1030*465dadb5SJonathan T. Looney 	/* check if -e was specified without -g */
1031*465dadb5SJonathan T. Looney 	if ((args.pa_flags & FLAG_DO_WIDE_GPROF_HC) &&
1032*465dadb5SJonathan T. Looney 	    !(args.pa_flags & FLAG_DO_GPROF))
1033*465dadb5SJonathan T. Looney 		errx(EX_USAGE,
1034*465dadb5SJonathan T. Looney "ERROR: option -e requires gprof mode to be specified."
1035*465dadb5SJonathan T. Looney 		    );
1036*465dadb5SJonathan T. Looney 
1037f263522aSJoseph Koshy 	/* check if -O was spuriously specified */
103815139246SJoseph Koshy 	if ((args.pa_flags & FLAG_HAS_OUTPUT_LOGFILE) &&
103915139246SJoseph Koshy 	    (args.pa_required & FLAG_HAS_OUTPUT_LOGFILE) == 0)
1040f263522aSJoseph Koshy 		errx(EX_USAGE,
104137d6f8a9SDavid E. O'Brien "ERROR: option -O is used only with options -E, -P, -S and -W."
104237d6f8a9SDavid E. O'Brien 		    );
1043f263522aSJoseph Koshy 
10440b86b1bbSFabien Thomas 	/* -k kernel path require -g/-G/-m/-T or -R */
104515139246SJoseph Koshy 	if ((args.pa_flags & FLAG_HAS_KERNELPATH) &&
1046b6010f9eSJoseph Koshy 	    (args.pa_flags & FLAG_DO_ANALYSIS) == 0 &&
104749874f6eSJoseph Koshy 	    (args.pa_flags & FLAG_READ_LOGFILE) == 0)
10480b86b1bbSFabien Thomas 	    errx(EX_USAGE, "ERROR: option -k is only used with -g/-R/-m/-T.");
104915139246SJoseph Koshy 
1050b6010f9eSJoseph Koshy 	/* -D only applies to gprof output mode (-g) */
105115139246SJoseph Koshy 	if ((args.pa_flags & FLAG_HAS_SAMPLESDIR) &&
1052b6010f9eSJoseph Koshy 	    (args.pa_flags & FLAG_DO_GPROF) == 0)
1053b6010f9eSJoseph Koshy 	    errx(EX_USAGE, "ERROR: option -D is only used with -g.");
105449874f6eSJoseph Koshy 
105549874f6eSJoseph Koshy 	/* -M mapfile requires -g or -R */
105649874f6eSJoseph Koshy 	if (args.pa_mapfilename != NULL &&
105749874f6eSJoseph Koshy 	    (args.pa_flags & FLAG_DO_GPROF) == 0 &&
105849874f6eSJoseph Koshy 	    (args.pa_flags & FLAG_READ_LOGFILE) == 0)
105949874f6eSJoseph Koshy 	    errx(EX_USAGE, "ERROR: option -M is only used with -g/-R.");
106015139246SJoseph Koshy 
106115139246SJoseph Koshy 	/*
106215139246SJoseph Koshy 	 * Disallow textual output of sampling PMCs if counting PMCs
106315139246SJoseph Koshy 	 * have also been asked for, mostly because the combined output
106415139246SJoseph Koshy 	 * is difficult to make sense of.
106515139246SJoseph Koshy 	 */
106615139246SJoseph Koshy 	if ((args.pa_flags & FLAG_HAS_COUNTING_PMCS) &&
106715139246SJoseph Koshy 	    (args.pa_flags & FLAG_HAS_SAMPLING_PMCS) &&
106800a22dc5SJoseph Koshy 	    ((args.pa_flags & FLAG_HAS_OUTPUT_LOGFILE) == 0))
106937d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
107037d6f8a9SDavid E. O'Brien "ERROR: option -O is required if counting and sampling PMCs are specified together."
107137d6f8a9SDavid E. O'Brien 		    );
107215139246SJoseph Koshy 
107349874f6eSJoseph Koshy 	/*
107438fd3834SJohn Baldwin 	 * Check if 'kerneldir' refers to a file rather than a
107538fd3834SJohn Baldwin 	 * directory.  If so, use `dirname path` to determine the
107638fd3834SJohn Baldwin 	 * kernel directory.
107749874f6eSJoseph Koshy 	 */
107849874f6eSJoseph Koshy 	(void) snprintf(buffer, sizeof(buffer), "%s%s", args.pa_fsroot,
107949874f6eSJoseph Koshy 	    args.pa_kernel);
108049874f6eSJoseph Koshy 	if (stat(buffer, &sb) < 0)
108149874f6eSJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot locate kernel \"%s\"",
108249874f6eSJoseph Koshy 		    buffer);
108349874f6eSJoseph Koshy 	if (!S_ISREG(sb.st_mode) && !S_ISDIR(sb.st_mode))
108449874f6eSJoseph Koshy 		errx(EX_USAGE, "ERROR: \"%s\": Unsupported file type.",
108549874f6eSJoseph Koshy 		    buffer);
108649874f6eSJoseph Koshy 	if (!S_ISDIR(sb.st_mode)) {
108749874f6eSJoseph Koshy 		tmp = args.pa_kernel;
108849874f6eSJoseph Koshy 		args.pa_kernel = strdup(dirname(args.pa_kernel));
108949874f6eSJoseph Koshy 		free(tmp);
109049874f6eSJoseph Koshy 		(void) snprintf(buffer, sizeof(buffer), "%s%s",
109149874f6eSJoseph Koshy 		    args.pa_fsroot, args.pa_kernel);
109249874f6eSJoseph Koshy 		if (stat(buffer, &sb) < 0)
109349874f6eSJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot stat \"%s\"",
109449874f6eSJoseph Koshy 			    buffer);
109549874f6eSJoseph Koshy 		if (!S_ISDIR(sb.st_mode))
109637d6f8a9SDavid E. O'Brien 			errx(EX_USAGE,
109737d6f8a9SDavid E. O'Brien 			    "ERROR: \"%s\" is not a directory.",
109837d6f8a9SDavid E. O'Brien 			    buffer);
109949874f6eSJoseph Koshy 	}
110049874f6eSJoseph Koshy 
1101b6010f9eSJoseph Koshy 	/*
1102b6010f9eSJoseph Koshy 	 * If we have a callgraph be created, select the outputfile.
1103b6010f9eSJoseph Koshy 	 */
1104b6010f9eSJoseph Koshy 	if (args.pa_flags & FLAG_DO_CALLGRAPHS) {
1105b6010f9eSJoseph Koshy 		if (strcmp(graphfilename, "-") == 0)
1106b6010f9eSJoseph Koshy 		    args.pa_graphfile = args.pa_printfile;
1107b6010f9eSJoseph Koshy 		else {
1108b6010f9eSJoseph Koshy 			args.pa_graphfile = fopen(graphfilename, "w");
1109b6010f9eSJoseph Koshy 			if (args.pa_graphfile == NULL)
111037d6f8a9SDavid E. O'Brien 				err(EX_OSERR,
111137d6f8a9SDavid E. O'Brien 				    "ERROR: cannot open \"%s\" for writing",
111237d6f8a9SDavid E. O'Brien 				    graphfilename);
1113b6010f9eSJoseph Koshy 		}
1114b6010f9eSJoseph Koshy 	}
11150b86b1bbSFabien Thomas 	if (args.pa_flags & FLAG_DO_ANNOTATE) {
1116c0252222SAttilio Rao 		args.pa_graphfile = fopen(graphfilename, "w");
1117c0252222SAttilio Rao 		if (args.pa_graphfile == NULL)
1118c0252222SAttilio Rao 			err(EX_OSERR, "ERROR: cannot open \"%s\" for writing",
1119c0252222SAttilio Rao 			    graphfilename);
1120c0252222SAttilio Rao 	}
1121b6010f9eSJoseph Koshy 
1122ac708fddSFabien Thomas 	/* if we've been asked to process a log file, skip init */
1123ac708fddSFabien Thomas 	if ((args.pa_flags & FLAG_READ_LOGFILE) == 0) {
1124ac708fddSFabien Thomas 		if (pmc_init() < 0)
1125ac708fddSFabien Thomas 			err(EX_UNAVAILABLE,
112637d6f8a9SDavid E. O'Brien 			    "ERROR: Initialization of the pmc(3) library failed"
112737d6f8a9SDavid E. O'Brien 			    );
1128ac708fddSFabien Thomas 
1129ac708fddSFabien Thomas 		if ((npmc = pmc_npmc(0)) < 0) /* assume all CPUs are identical */
113037d6f8a9SDavid E. O'Brien 			err(EX_OSERR,
113137d6f8a9SDavid E. O'Brien "ERROR: Cannot determine the number of PMCs on CPU %d",
113237d6f8a9SDavid E. O'Brien 			    0);
1133ac708fddSFabien Thomas 	}
1134ac708fddSFabien Thomas 
1135ac708fddSFabien Thomas 	/* Allocate a kqueue */
1136ac708fddSFabien Thomas 	if ((pmcstat_kq = kqueue()) < 0)
1137ac708fddSFabien Thomas 		err(EX_OSERR, "ERROR: Cannot allocate kqueue");
1138ac708fddSFabien Thomas 
1139ac708fddSFabien Thomas 	/* Setup the logfile as the source. */
114015139246SJoseph Koshy 	if (args.pa_flags & FLAG_READ_LOGFILE) {
114115139246SJoseph Koshy 		/*
114215139246SJoseph Koshy 		 * Print the log in textual form if we haven't been
1143b6010f9eSJoseph Koshy 		 * asked to generate profiling information.
114415139246SJoseph Koshy 		 */
1145b6010f9eSJoseph Koshy 		if ((args.pa_flags & FLAG_DO_ANALYSIS) == 0)
114615139246SJoseph Koshy 			args.pa_flags |= FLAG_DO_PRINT;
114715139246SJoseph Koshy 
11480b86b1bbSFabien Thomas 		pmcstat_initialize_logging();
1149ac708fddSFabien Thomas 		rfd = pmcstat_open_log(args.pa_inputpath,
1150302cbb90SJoseph Koshy 		    PMCSTAT_OPEN_FOR_READ);
1151ac708fddSFabien Thomas 		if ((args.pa_logparser = pmclog_open(rfd)) == NULL)
115215139246SJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot create parser");
1153ac708fddSFabien Thomas 		if (fcntl(rfd, F_SETFL, O_NONBLOCK) < 0)
1154ac708fddSFabien Thomas 			err(EX_OSERR, "ERROR: fcntl(2) failed");
1155ac708fddSFabien Thomas 		EV_SET(&kev, rfd, EVFILT_READ, EV_ADD,
1156ac708fddSFabien Thomas 		    0, 0, NULL);
1157ac708fddSFabien Thomas 		if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
1158ac708fddSFabien Thomas 			err(EX_OSERR, "ERROR: Cannot register kevent");
1159f263522aSJoseph Koshy 	}
116015139246SJoseph Koshy 	/*
116115139246SJoseph Koshy 	 * Configure the specified log file or setup a default log
116215139246SJoseph Koshy 	 * consumer via a pipe.
116315139246SJoseph Koshy 	 */
116415139246SJoseph Koshy 	if (args.pa_required & FLAG_HAS_OUTPUT_LOGFILE) {
1165302cbb90SJoseph Koshy 		if (args.pa_outputpath)
1166302cbb90SJoseph Koshy 			args.pa_logfd = pmcstat_open_log(args.pa_outputpath,
1167302cbb90SJoseph Koshy 			    PMCSTAT_OPEN_FOR_WRITE);
1168302cbb90SJoseph Koshy 		else {
116915139246SJoseph Koshy 			/*
117015139246SJoseph Koshy 			 * process the log on the fly by reading it in
117115139246SJoseph Koshy 			 * through a pipe.
117215139246SJoseph Koshy 			 */
117315139246SJoseph Koshy 			if (pipe(pipefd) < 0)
117415139246SJoseph Koshy 				err(EX_OSERR, "ERROR: pipe(2) failed");
117515139246SJoseph Koshy 
117615139246SJoseph Koshy 			if (fcntl(pipefd[READPIPEFD], F_SETFL, O_NONBLOCK) < 0)
117715139246SJoseph Koshy 				err(EX_OSERR, "ERROR: fcntl(2) failed");
117815139246SJoseph Koshy 
117915139246SJoseph Koshy 			EV_SET(&kev, pipefd[READPIPEFD], EVFILT_READ, EV_ADD,
118015139246SJoseph Koshy 			    0, 0, NULL);
118115139246SJoseph Koshy 
118215139246SJoseph Koshy 			if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
118315139246SJoseph Koshy 				err(EX_OSERR, "ERROR: Cannot register kevent");
118415139246SJoseph Koshy 
118515139246SJoseph Koshy 			args.pa_logfd = pipefd[WRITEPIPEFD];
118615139246SJoseph Koshy 
11870b86b1bbSFabien Thomas 			args.pa_flags |= FLAG_HAS_PIPE;
11880b86b1bbSFabien Thomas 			if ((args.pa_flags & FLAG_DO_TOP) == 0)
11890b86b1bbSFabien Thomas 				args.pa_flags |= FLAG_DO_PRINT;
119015139246SJoseph Koshy 			args.pa_logparser = pmclog_open(pipefd[READPIPEFD]);
119115139246SJoseph Koshy 		}
119215139246SJoseph Koshy 
119315139246SJoseph Koshy 		if (pmc_configure_logfile(args.pa_logfd) < 0)
119415139246SJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot configure log file");
119515139246SJoseph Koshy 	}
119615139246SJoseph Koshy 
11977fbf3da2SJoseph Koshy 	/* remember to check for driver errors if we are sampling or logging */
11987fbf3da2SJoseph Koshy 	check_driver_stats = (args.pa_flags & FLAG_HAS_SAMPLING_PMCS) ||
11997fbf3da2SJoseph Koshy 	    (args.pa_flags & FLAG_HAS_OUTPUT_LOGFILE);
12007fbf3da2SJoseph Koshy 
1201ebccf1e3SJoseph Koshy 	/*
1202ac708fddSFabien Thomas 	if (args.pa_flags & FLAG_READ_LOGFILE) {
1203ebccf1e3SJoseph Koshy 	 * Allocate PMCs.
1204ebccf1e3SJoseph Koshy 	 */
1205ebccf1e3SJoseph Koshy 
1206fbd8b34aSJoseph Koshy 	STAILQ_FOREACH(ev, &args.pa_events, ev_next) {
1207ebccf1e3SJoseph Koshy 		if (pmc_allocate(ev->ev_spec, ev->ev_mode,
1208f263522aSJoseph Koshy 		    ev->ev_flags, ev->ev_cpu, &ev->ev_pmcid) < 0)
120937d6f8a9SDavid E. O'Brien 			err(EX_OSERR,
121037d6f8a9SDavid E. O'Brien "ERROR: Cannot allocate %s-mode pmc with specification \"%s\"",
121137d6f8a9SDavid E. O'Brien 			    PMC_IS_SYSTEM_MODE(ev->ev_mode) ?
121237d6f8a9SDavid E. O'Brien 			    "system" : "process", ev->ev_spec);
1213ebccf1e3SJoseph Koshy 
121415139246SJoseph Koshy 		if (PMC_IS_SAMPLING_MODE(ev->ev_mode) &&
121515139246SJoseph Koshy 		    pmc_set(ev->ev_pmcid, ev->ev_count) < 0)
121637d6f8a9SDavid E. O'Brien 			err(EX_OSERR,
121737d6f8a9SDavid E. O'Brien 			    "ERROR: Cannot set sampling count for PMC \"%s\"",
121837d6f8a9SDavid E. O'Brien 			    ev->ev_name);
121915139246SJoseph Koshy 	}
122015139246SJoseph Koshy 
1221ebccf1e3SJoseph Koshy 	/* compute printout widths */
1222fbd8b34aSJoseph Koshy 	STAILQ_FOREACH(ev, &args.pa_events, ev_next) {
1223c5153e19SJoseph Koshy 		int counter_width;
1224c5153e19SJoseph Koshy 		int display_width;
1225c5153e19SJoseph Koshy 		int header_width;
1226ebccf1e3SJoseph Koshy 
1227c5153e19SJoseph Koshy 		(void) pmc_width(ev->ev_pmcid, &counter_width);
1228ee4910b1SJoseph Koshy 		header_width = strlen(ev->ev_name) + 2; /* prefix '%c/' */
1229c5153e19SJoseph Koshy 		display_width = (int) floor(counter_width / 3.32193) + 1;
1230ebccf1e3SJoseph Koshy 
1231ee4910b1SJoseph Koshy 		if (PMC_IS_SYSTEM_MODE(ev->ev_mode))
1232ee4910b1SJoseph Koshy 			header_width += 3; /* 2 digit CPU number + '/' */
1233ee4910b1SJoseph Koshy 
1234c5153e19SJoseph Koshy 		if (header_width > display_width) {
1235ebccf1e3SJoseph Koshy 			ev->ev_fieldskip = 0;
1236c5153e19SJoseph Koshy 			ev->ev_fieldwidth = header_width;
1237ebccf1e3SJoseph Koshy 		} else {
1238c5153e19SJoseph Koshy 			ev->ev_fieldskip = display_width -
1239c5153e19SJoseph Koshy 			    header_width;
1240c5153e19SJoseph Koshy 			ev->ev_fieldwidth = display_width;
1241ebccf1e3SJoseph Koshy 		}
1242ebccf1e3SJoseph Koshy 	}
1243ebccf1e3SJoseph Koshy 
1244ebccf1e3SJoseph Koshy 	/*
1245ebccf1e3SJoseph Koshy 	 * If our output is being set to a terminal, register a handler
1246ebccf1e3SJoseph Koshy 	 * for window size changes.
1247ebccf1e3SJoseph Koshy 	 */
1248ebccf1e3SJoseph Koshy 
124915139246SJoseph Koshy 	if (isatty(fileno(args.pa_printfile))) {
1250ebccf1e3SJoseph Koshy 
125115139246SJoseph Koshy 		if (ioctl(fileno(args.pa_printfile), TIOCGWINSZ, &ws) < 0)
1252ebccf1e3SJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot determine window size");
1253ebccf1e3SJoseph Koshy 
1254ebccf1e3SJoseph Koshy 		pmcstat_displayheight = ws.ws_row - 1;
12550b86b1bbSFabien Thomas 		pmcstat_displaywidth  = ws.ws_col - 1;
1256ebccf1e3SJoseph Koshy 
1257ebccf1e3SJoseph Koshy 		EV_SET(&kev, SIGWINCH, EVFILT_SIGNAL, EV_ADD, 0, 0, NULL);
1258ebccf1e3SJoseph Koshy 
1259ebccf1e3SJoseph Koshy 		if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
126037d6f8a9SDavid E. O'Brien 			err(EX_OSERR,
126137d6f8a9SDavid E. O'Brien 			    "ERROR: Cannot register kevent for SIGWINCH");
12620b86b1bbSFabien Thomas 
12630b86b1bbSFabien Thomas 		args.pa_toptty = 1;
12640b86b1bbSFabien Thomas 	}
12650b86b1bbSFabien Thomas 
12660b86b1bbSFabien Thomas 	/*
12670b86b1bbSFabien Thomas 	 * Listen to key input in top mode.
12680b86b1bbSFabien Thomas 	 */
12690b86b1bbSFabien Thomas 	if (args.pa_flags & FLAG_DO_TOP) {
12700b86b1bbSFabien Thomas 		EV_SET(&kev, fileno(stdin), EVFILT_READ, EV_ADD, 0, 0, NULL);
12710b86b1bbSFabien Thomas 		if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
12720b86b1bbSFabien Thomas 			err(EX_OSERR, "ERROR: Cannot register kevent");
1273ebccf1e3SJoseph Koshy 	}
1274ebccf1e3SJoseph Koshy 
1275ebccf1e3SJoseph Koshy 	EV_SET(&kev, SIGINT, EVFILT_SIGNAL, EV_ADD, 0, 0, NULL);
1276ebccf1e3SJoseph Koshy 	if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
1277ebccf1e3SJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot register kevent for SIGINT");
1278ebccf1e3SJoseph Koshy 
1279f263522aSJoseph Koshy 	EV_SET(&kev, SIGIO, EVFILT_SIGNAL, EV_ADD, 0, 0, NULL);
1280f263522aSJoseph Koshy 	if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
1281f263522aSJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot register kevent for SIGIO");
1282ebccf1e3SJoseph Koshy 
1283ebccf1e3SJoseph Koshy 	/*
1284f263522aSJoseph Koshy 	 * An exec() failure of a forked child is signalled by the
1285f263522aSJoseph Koshy 	 * child sending the parent a SIGCHLD.  We don't register an
1286f263522aSJoseph Koshy 	 * actual signal handler for SIGCHLD, but instead use our
1287f263522aSJoseph Koshy 	 * kqueue to pick up the signal.
1288ebccf1e3SJoseph Koshy 	 */
1289f263522aSJoseph Koshy 	EV_SET(&kev, SIGCHLD, EVFILT_SIGNAL, EV_ADD, 0, 0, NULL);
1290f263522aSJoseph Koshy 	if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
1291f263522aSJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot register kevent for SIGCHLD");
1292ebccf1e3SJoseph Koshy 
12930b86b1bbSFabien Thomas 	/*
12940b86b1bbSFabien Thomas 	 * Setup a timer if we have counting mode PMCs needing to be printed or
12950b86b1bbSFabien Thomas 	 * top mode plugin is active.
12960b86b1bbSFabien Thomas 	 */
12970b86b1bbSFabien Thomas 	if (((args.pa_flags & FLAG_HAS_COUNTING_PMCS) &&
12980b86b1bbSFabien Thomas 	     (args.pa_required & FLAG_HAS_OUTPUT_LOGFILE) == 0) ||
12990b86b1bbSFabien Thomas 	    (args.pa_flags & FLAG_DO_TOP)) {
1300ebccf1e3SJoseph Koshy 		EV_SET(&kev, 0, EVFILT_TIMER, EV_ADD, 0,
1301ebccf1e3SJoseph Koshy 		    args.pa_interval * 1000, NULL);
1302ebccf1e3SJoseph Koshy 
1303ebccf1e3SJoseph Koshy 		if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
130437d6f8a9SDavid E. O'Brien 			err(EX_OSERR,
130537d6f8a9SDavid E. O'Brien 			    "ERROR: Cannot register kevent for timer");
1306ebccf1e3SJoseph Koshy 	}
1307ebccf1e3SJoseph Koshy 
130872fe532cSGeorge V. Neville-Neil 	/*
130972fe532cSGeorge V. Neville-Neil 	 * Setup a duration timer if we have sampling mode PMCs and
131072fe532cSGeorge V. Neville-Neil 	 * a duration time is set
131172fe532cSGeorge V. Neville-Neil 	 */
131272fe532cSGeorge V. Neville-Neil 	if ((args.pa_flags & FLAG_HAS_SAMPLING_PMCS) &&
131372fe532cSGeorge V. Neville-Neil 	    (args.pa_flags & FLAG_HAS_DURATION)) {
131472fe532cSGeorge V. Neville-Neil 		EV_SET(&kev, 0, EVFILT_TIMER, EV_ADD, 0,
131572fe532cSGeorge V. Neville-Neil 		    args.pa_duration * 1000, NULL);
131672fe532cSGeorge V. Neville-Neil 
131772fe532cSGeorge V. Neville-Neil 		if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
131872fe532cSGeorge V. Neville-Neil 			err(EX_OSERR, "ERROR: Cannot register kevent for "
131972fe532cSGeorge V. Neville-Neil 			    "time duration");
132072fe532cSGeorge V. Neville-Neil 	}
132172fe532cSGeorge V. Neville-Neil 
1322ebccf1e3SJoseph Koshy 	/* attach PMCs to the target process, starting it if specified */
1323ee4910b1SJoseph Koshy 	if (args.pa_flags & FLAG_HAS_COMMANDLINE)
13240b86b1bbSFabien Thomas 		pmcstat_create_process();
1325ebccf1e3SJoseph Koshy 
13267fbf3da2SJoseph Koshy 	if (check_driver_stats && pmc_get_driver_stats(&ds_start) < 0)
13277fbf3da2SJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot retrieve driver statistics");
13287fbf3da2SJoseph Koshy 
1329ee4910b1SJoseph Koshy 	/* Attach process pmcs to the target process. */
1330165b10b0SJoseph Koshy 	if (args.pa_flags & (FLAG_HAS_TARGET | FLAG_HAS_COMMANDLINE)) {
1331fbd8b34aSJoseph Koshy 		if (SLIST_EMPTY(&args.pa_targets))
133237d6f8a9SDavid E. O'Brien 			errx(EX_DATAERR,
133337d6f8a9SDavid E. O'Brien 			    "ERROR: No matching target processes.");
13345477328dSJoseph Koshy 		if (args.pa_flags & FLAG_HAS_PROCESS_PMCS)
13350b86b1bbSFabien Thomas 			pmcstat_attach_pmcs();
1336ee4910b1SJoseph Koshy 
1337fbd8b34aSJoseph Koshy 		if (pmcstat_kvm) {
1338fbd8b34aSJoseph Koshy 			kvm_close(pmcstat_kvm);
1339fbd8b34aSJoseph Koshy 			pmcstat_kvm = NULL;
1340fbd8b34aSJoseph Koshy 		}
1341fbd8b34aSJoseph Koshy 	}
1342fbd8b34aSJoseph Koshy 
1343ebccf1e3SJoseph Koshy 	/* start the pmcs */
13440b86b1bbSFabien Thomas 	pmcstat_start_pmcs();
1345ebccf1e3SJoseph Koshy 
1346ebccf1e3SJoseph Koshy 	/* start the (commandline) process if needed */
134715139246SJoseph Koshy 	if (args.pa_flags & FLAG_HAS_COMMANDLINE)
1348ee4910b1SJoseph Koshy 		pmcstat_start_process();
1349ebccf1e3SJoseph Koshy 
1350ac708fddSFabien Thomas 	/* initialize logging */
13510b86b1bbSFabien Thomas 	pmcstat_initialize_logging();
135215139246SJoseph Koshy 
1353ebccf1e3SJoseph Koshy 	/* Handle SIGINT using the kqueue loop */
1354ebccf1e3SJoseph Koshy 	sa.sa_handler = SIG_IGN;
1355ebccf1e3SJoseph Koshy 	sa.sa_flags   = 0;
1356ebccf1e3SJoseph Koshy 	(void) sigemptyset(&sa.sa_mask);
1357ebccf1e3SJoseph Koshy 
1358ebccf1e3SJoseph Koshy 	if (sigaction(SIGINT, &sa, NULL) < 0)
1359ebccf1e3SJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot install signal handler");
1360ebccf1e3SJoseph Koshy 
1361ebccf1e3SJoseph Koshy 	/*
13620b86b1bbSFabien Thomas 	 * Setup the top mode display.
13630b86b1bbSFabien Thomas 	 */
13640b86b1bbSFabien Thomas 	if (args.pa_flags & FLAG_DO_TOP) {
13650b86b1bbSFabien Thomas 		args.pa_flags &= ~FLAG_DO_PRINT;
13660b86b1bbSFabien Thomas 
13670b86b1bbSFabien Thomas 		if (args.pa_toptty) {
13680b86b1bbSFabien Thomas 			/*
13690b86b1bbSFabien Thomas 			 * Init ncurses.
13700b86b1bbSFabien Thomas 			 */
13710b86b1bbSFabien Thomas 			initscr();
13720b86b1bbSFabien Thomas 			if(has_colors() == TRUE) {
13730b86b1bbSFabien Thomas 				args.pa_topcolor = 1;
13740b86b1bbSFabien Thomas 				start_color();
13750b86b1bbSFabien Thomas 				use_default_colors();
13760b86b1bbSFabien Thomas 				pair_content(0, &cf, &cb);
13770b86b1bbSFabien Thomas 				init_pair(1, COLOR_RED, cb);
13780b86b1bbSFabien Thomas 				init_pair(2, COLOR_YELLOW, cb);
13790b86b1bbSFabien Thomas 				init_pair(3, COLOR_GREEN, cb);
13800b86b1bbSFabien Thomas 			}
13810b86b1bbSFabien Thomas 			cbreak();
13820b86b1bbSFabien Thomas 			noecho();
13830b86b1bbSFabien Thomas 			nonl();
13840b86b1bbSFabien Thomas 			nodelay(stdscr, 1);
13850b86b1bbSFabien Thomas 			intrflush(stdscr, FALSE);
13860b86b1bbSFabien Thomas 			keypad(stdscr, TRUE);
13870b86b1bbSFabien Thomas 			clear();
13882aa983e5SFabien Thomas 			/* Get terminal width / height with ncurses. */
138937d6f8a9SDavid E. O'Brien 			getmaxyx(stdscr,
139037d6f8a9SDavid E. O'Brien 			    pmcstat_displayheight, pmcstat_displaywidth);
13912aa983e5SFabien Thomas 			pmcstat_displayheight--; pmcstat_displaywidth--;
13920b86b1bbSFabien Thomas 			atexit(pmcstat_topexit);
13930b86b1bbSFabien Thomas 		}
13940b86b1bbSFabien Thomas 	}
13950b86b1bbSFabien Thomas 
13960b86b1bbSFabien Thomas 	/*
1397ebccf1e3SJoseph Koshy 	 * loop till either the target process (if any) exits, or we
139872fe532cSGeorge V. Neville-Neil 	 * are killed by a SIGINT or we reached the time duration.
1399ebccf1e3SJoseph Koshy 	 */
1400f263522aSJoseph Koshy 	runstate = PMCSTAT_RUNNING;
1401dceed24aSFabien Thomas 	do_print = do_read = 0;
1402ebccf1e3SJoseph Koshy 	do {
1403ebccf1e3SJoseph Koshy 		if ((c = kevent(pmcstat_kq, NULL, 0, &kev, 1, NULL)) <= 0) {
1404ebccf1e3SJoseph Koshy 			if (errno != EINTR)
1405ebccf1e3SJoseph Koshy 				err(EX_OSERR, "ERROR: kevent failed");
1406ebccf1e3SJoseph Koshy 			else
1407ebccf1e3SJoseph Koshy 				continue;
1408ebccf1e3SJoseph Koshy 		}
1409ebccf1e3SJoseph Koshy 
1410ebccf1e3SJoseph Koshy 		if (kev.flags & EV_ERROR)
1411ebccf1e3SJoseph Koshy 			errc(EX_OSERR, kev.data, "ERROR: kevent failed");
1412ebccf1e3SJoseph Koshy 
1413ebccf1e3SJoseph Koshy 		switch (kev.filter) {
1414f263522aSJoseph Koshy 		case EVFILT_PROC:  /* target has exited */
14150b86b1bbSFabien Thomas 			runstate = pmcstat_close_log();
141638fc5f34SJoseph Koshy 			do_print = 1;
1417f263522aSJoseph Koshy 			break;
1418ebccf1e3SJoseph Koshy 
1419f263522aSJoseph Koshy 		case EVFILT_READ:  /* log file data is present */
1420ac708fddSFabien Thomas 			if (kev.ident == (unsigned)fileno(stdin) &&
1421ac708fddSFabien Thomas 			    (args.pa_flags & FLAG_DO_TOP)) {
14220b86b1bbSFabien Thomas 				if (pmcstat_keypress_log())
14230b86b1bbSFabien Thomas 					runstate = pmcstat_close_log();
1424dceed24aSFabien Thomas 			} else {
1425dceed24aSFabien Thomas 				do_read = 0;
14260b86b1bbSFabien Thomas 				runstate = pmcstat_process_log();
1427dceed24aSFabien Thomas 			}
1428ebccf1e3SJoseph Koshy 			break;
1429ebccf1e3SJoseph Koshy 
1430ebccf1e3SJoseph Koshy 		case EVFILT_SIGNAL:
1431f263522aSJoseph Koshy 			if (kev.ident == SIGCHLD) {
1432f263522aSJoseph Koshy 				/*
1433f263522aSJoseph Koshy 				 * The child process sends us a
1434f263522aSJoseph Koshy 				 * SIGCHLD if its exec() failed.  We
1435f263522aSJoseph Koshy 				 * wait for it to exit and then exit
1436f263522aSJoseph Koshy 				 * ourselves.
1437f263522aSJoseph Koshy 				 */
1438f263522aSJoseph Koshy 				(void) wait(&c);
1439f263522aSJoseph Koshy 				runstate = PMCSTAT_FINISHED;
1440f263522aSJoseph Koshy 			} else if (kev.ident == SIGIO) {
1441f263522aSJoseph Koshy 				/*
1442f263522aSJoseph Koshy 				 * We get a SIGIO if a PMC loses all
1443f263522aSJoseph Koshy 				 * of its targets, or if logfile
1444f263522aSJoseph Koshy 				 * writes encounter an error.
1445f263522aSJoseph Koshy 				 */
14460b86b1bbSFabien Thomas 				runstate = pmcstat_close_log();
1447f263522aSJoseph Koshy 				do_print = 1; /* print PMCs at exit */
1448f263522aSJoseph Koshy 			} else if (kev.ident == SIGINT) {
144915139246SJoseph Koshy 				/* Kill the child process if we started it */
145015139246SJoseph Koshy 				if (args.pa_flags & FLAG_HAS_COMMANDLINE)
14510b86b1bbSFabien Thomas 					pmcstat_kill_process();
1452ac708fddSFabien Thomas 				runstate = pmcstat_close_log();
1453ebccf1e3SJoseph Koshy 			} else if (kev.ident == SIGWINCH) {
145415139246SJoseph Koshy 				if (ioctl(fileno(args.pa_printfile),
1455ebccf1e3SJoseph Koshy 					TIOCGWINSZ, &ws) < 0)
145637d6f8a9SDavid E. O'Brien 				    err(EX_OSERR,
145737d6f8a9SDavid E. O'Brien 				        "ERROR: Cannot determine window size");
1458ebccf1e3SJoseph Koshy 				pmcstat_displayheight = ws.ws_row - 1;
14590b86b1bbSFabien Thomas 				pmcstat_displaywidth  = ws.ws_col - 1;
1460ebccf1e3SJoseph Koshy 			} else
1461ebccf1e3SJoseph Koshy 				assert(0);
1462ebccf1e3SJoseph Koshy 
1463ebccf1e3SJoseph Koshy 			break;
1464f263522aSJoseph Koshy 
146572fe532cSGeorge V. Neville-Neil 		case EVFILT_TIMER:
146672fe532cSGeorge V. Neville-Neil 			/* time duration reached, exit */
146772fe532cSGeorge V. Neville-Neil 			if (args.pa_flags & FLAG_HAS_DURATION) {
146872fe532cSGeorge V. Neville-Neil 				runstate = PMCSTAT_FINISHED;
146972fe532cSGeorge V. Neville-Neil 				break;
147072fe532cSGeorge V. Neville-Neil 			}
147172fe532cSGeorge V. Neville-Neil 			/* print out counting PMCs */
1472dceed24aSFabien Thomas 			if ((args.pa_flags & FLAG_DO_TOP) &&
147366ce9936SFabien Thomas 			     pmc_flush_logfile() == 0)
1474dceed24aSFabien Thomas 				do_read = 1;
1475f263522aSJoseph Koshy 			do_print = 1;
1476f263522aSJoseph Koshy 			break;
1477f263522aSJoseph Koshy 
1478ebccf1e3SJoseph Koshy 		}
1479ebccf1e3SJoseph Koshy 
1480dceed24aSFabien Thomas 		if (do_print && !do_read) {
14810b86b1bbSFabien Thomas 			if ((args.pa_required & FLAG_HAS_OUTPUT_LOGFILE) == 0) {
14820b86b1bbSFabien Thomas 				pmcstat_print_pmcs();
148337d6f8a9SDavid E. O'Brien 				if (runstate == PMCSTAT_FINISHED &&
148437d6f8a9SDavid E. O'Brien 				    /* final newline */
148515139246SJoseph Koshy 				    (args.pa_flags & FLAG_DO_PRINT) == 0)
148615139246SJoseph Koshy 					(void) fprintf(args.pa_printfile, "\n");
14870b86b1bbSFabien Thomas 			}
14880b86b1bbSFabien Thomas 			if (args.pa_flags & FLAG_DO_TOP)
14890b86b1bbSFabien Thomas 				pmcstat_display_log();
1490f263522aSJoseph Koshy 			do_print = 0;
1491f263522aSJoseph Koshy 		}
1492f263522aSJoseph Koshy 
1493f263522aSJoseph Koshy 	} while (runstate != PMCSTAT_FINISHED);
1494f263522aSJoseph Koshy 
14950b86b1bbSFabien Thomas 	if ((args.pa_flags & FLAG_DO_TOP) && args.pa_toptty) {
14960b86b1bbSFabien Thomas 		pmcstat_topexit();
14970b86b1bbSFabien Thomas 		args.pa_toptty = 0;
14980b86b1bbSFabien Thomas 	}
14990b86b1bbSFabien Thomas 
1500f263522aSJoseph Koshy 	/* flush any pending log entries */
150115139246SJoseph Koshy 	if (args.pa_flags & (FLAG_HAS_OUTPUT_LOGFILE | FLAG_HAS_PIPE))
1502dceed24aSFabien Thomas 		pmc_close_logfile();
1503ebccf1e3SJoseph Koshy 
15040b86b1bbSFabien Thomas 	pmcstat_cleanup();
1505ebccf1e3SJoseph Koshy 
150649874f6eSJoseph Koshy 	free(args.pa_kernel);
150749874f6eSJoseph Koshy 
15087fbf3da2SJoseph Koshy 	/* check if the driver lost any samples or events */
15097fbf3da2SJoseph Koshy 	if (check_driver_stats) {
15107fbf3da2SJoseph Koshy 		if (pmc_get_driver_stats(&ds_end) < 0)
151137d6f8a9SDavid E. O'Brien 			err(EX_OSERR,
151237d6f8a9SDavid E. O'Brien 			    "ERROR: Cannot retrieve driver statistics");
151349874f6eSJoseph Koshy 		if (ds_start.pm_intr_bufferfull != ds_end.pm_intr_bufferfull &&
151449874f6eSJoseph Koshy 		    args.pa_verbosity > 0)
1515e64ba55dSJonathan T. Looney 			warnx(
1516e64ba55dSJonathan T. Looney "WARNING: sampling was paused at least %u time%s.\n"
1517e64ba55dSJonathan T. Looney "Please consider tuning the \"kern.hwpmc.nsamples\" tunable.",
1518e64ba55dSJonathan T. Looney 			    ds_end.pm_intr_bufferfull -
1519e64ba55dSJonathan T. Looney 			    ds_start.pm_intr_bufferfull,
1520e64ba55dSJonathan T. Looney 			    ((ds_end.pm_intr_bufferfull -
1521e64ba55dSJonathan T. Looney 			    ds_start.pm_intr_bufferfull) != 1) ? "s" : ""
152237d6f8a9SDavid E. O'Brien 			    );
15237fbf3da2SJoseph Koshy 		if (ds_start.pm_buffer_requests_failed !=
152449874f6eSJoseph Koshy 		    ds_end.pm_buffer_requests_failed &&
152549874f6eSJoseph Koshy 		    args.pa_verbosity > 0)
1526e64ba55dSJonathan T. Looney 			warnx(
1527e64ba55dSJonathan T. Looney "WARNING: at least %u event%s were discarded while running.\n"
1528e64ba55dSJonathan T. Looney "Please consider tuning the \"kern.hwpmc.nbuffers\" tunable.",
1529e64ba55dSJonathan T. Looney 	 		    ds_end.pm_buffer_requests_failed -
1530e64ba55dSJonathan T. Looney 			    ds_start.pm_buffer_requests_failed,
1531e64ba55dSJonathan T. Looney 			    ((ds_end.pm_buffer_requests_failed -
1532e64ba55dSJonathan T. Looney 			    ds_start.pm_buffer_requests_failed) != 1) ? "s" : ""
153337d6f8a9SDavid E. O'Brien 			    );
15347fbf3da2SJoseph Koshy 	}
15357fbf3da2SJoseph Koshy 
15367fbf3da2SJoseph Koshy 	exit(EX_OK);
1537ebccf1e3SJoseph Koshy }
1538