xref: /freebsd/usr.sbin/pmcstat/pmcstat.c (revision 53d0b9e438bc30ef12c24fddec13e97ade852164)
1ebccf1e3SJoseph Koshy /*-
24d846d26SWarner Losh  * SPDX-License-Identifier: BSD-2-Clause
31de7b4b8SPedro F. Giffuni  *
4165b10b0SJoseph Koshy  * Copyright (c) 2003-2008, Joseph Koshy
5b6010f9eSJoseph Koshy  * Copyright (c) 2007 The FreeBSD Foundation
6ebccf1e3SJoseph Koshy  * All rights reserved.
7ebccf1e3SJoseph Koshy  *
8b6010f9eSJoseph Koshy  * Portions of this software were developed by A. Joseph Koshy under
9b6010f9eSJoseph Koshy  * sponsorship from the FreeBSD Foundation and Google, Inc.
10b6010f9eSJoseph Koshy  *
11ebccf1e3SJoseph Koshy  * Redistribution and use in source and binary forms, with or without
12ebccf1e3SJoseph Koshy  * modification, are permitted provided that the following conditions
13ebccf1e3SJoseph Koshy  * are met:
14ebccf1e3SJoseph Koshy  * 1. Redistributions of source code must retain the above copyright
15ebccf1e3SJoseph Koshy  *    notice, this list of conditions and the following disclaimer.
16ebccf1e3SJoseph Koshy  * 2. Redistributions in binary form must reproduce the above copyright
17ebccf1e3SJoseph Koshy  *    notice, this list of conditions and the following disclaimer in the
18ebccf1e3SJoseph Koshy  *    documentation and/or other materials provided with the distribution.
19ebccf1e3SJoseph Koshy  *
20ebccf1e3SJoseph Koshy  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
21ebccf1e3SJoseph Koshy  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22ebccf1e3SJoseph Koshy  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23ebccf1e3SJoseph Koshy  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
24ebccf1e3SJoseph Koshy  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25ebccf1e3SJoseph Koshy  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26ebccf1e3SJoseph Koshy  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27ebccf1e3SJoseph Koshy  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28ebccf1e3SJoseph Koshy  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29ebccf1e3SJoseph Koshy  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30ebccf1e3SJoseph Koshy  * SUCH DAMAGE.
31ebccf1e3SJoseph Koshy  */
32ebccf1e3SJoseph Koshy 
33ebccf1e3SJoseph Koshy #include <sys/cdefs.h>
34ebccf1e3SJoseph Koshy __FBSDID("$FreeBSD$");
35ebccf1e3SJoseph Koshy 
36fbd8b34aSJoseph Koshy #include <sys/param.h>
37ea056888SAttilio Rao #include <sys/cpuset.h>
38ea056888SAttilio Rao #include <sys/event.h>
39ebccf1e3SJoseph Koshy #include <sys/queue.h>
4015139246SJoseph Koshy #include <sys/socket.h>
4115139246SJoseph Koshy #include <sys/stat.h>
42ee4910b1SJoseph Koshy #include <sys/sysctl.h>
43ebccf1e3SJoseph Koshy #include <sys/time.h>
44ebccf1e3SJoseph Koshy #include <sys/ttycom.h>
45fbd8b34aSJoseph Koshy #include <sys/user.h>
46ebccf1e3SJoseph Koshy #include <sys/wait.h>
47ebccf1e3SJoseph Koshy 
48ebccf1e3SJoseph Koshy #include <assert.h>
490b86b1bbSFabien Thomas #include <curses.h>
50ebccf1e3SJoseph Koshy #include <err.h>
51ebccf1e3SJoseph Koshy #include <errno.h>
52ebccf1e3SJoseph Koshy #include <fcntl.h>
53fbd8b34aSJoseph Koshy #include <kvm.h>
5449874f6eSJoseph Koshy #include <libgen.h>
55ebccf1e3SJoseph Koshy #include <limits.h>
56ebccf1e3SJoseph Koshy #include <math.h>
57ebccf1e3SJoseph Koshy #include <pmc.h>
58f263522aSJoseph Koshy #include <pmclog.h>
59fbd8b34aSJoseph Koshy #include <regex.h>
60ebccf1e3SJoseph Koshy #include <signal.h>
61ebccf1e3SJoseph Koshy #include <stdarg.h>
62b6e28991SAleksandr Rybalko #include <stdbool.h>
63ebccf1e3SJoseph Koshy #include <stdint.h>
6415139246SJoseph Koshy #include <stdio.h>
65ebccf1e3SJoseph Koshy #include <stdlib.h>
66ebccf1e3SJoseph Koshy #include <string.h>
67ebccf1e3SJoseph Koshy #include <sysexits.h>
68ebccf1e3SJoseph Koshy #include <unistd.h>
69ebccf1e3SJoseph Koshy 
70d27927f7SRuslan Bukin #include <libpmcstat.h>
71d27927f7SRuslan Bukin 
7215139246SJoseph Koshy #include "pmcstat.h"
7315139246SJoseph Koshy 
74f263522aSJoseph Koshy /*
75f263522aSJoseph Koshy  * A given invocation of pmcstat(8) can manage multiple PMCs of both
76f263522aSJoseph Koshy  * the system-wide and per-process variety.  Each of these could be in
77f263522aSJoseph Koshy  * 'counting mode' or in 'sampling mode'.
78f263522aSJoseph Koshy  *
79f263522aSJoseph Koshy  * For 'counting mode' PMCs, pmcstat(8) will periodically issue a
80f263522aSJoseph Koshy  * pmc_read() at the configured time interval and print out the value
81f263522aSJoseph Koshy  * of the requested PMCs.
82f263522aSJoseph Koshy  *
83f263522aSJoseph Koshy  * For 'sampling mode' PMCs it can log to a file for offline analysis,
84f263522aSJoseph Koshy  * or can analyse sampling data "on the fly", either by converting
85f263522aSJoseph Koshy  * samples to printed textual form or by creating gprof(1) compatible
86f263522aSJoseph Koshy  * profiles, one per program executed.  When creating gprof(1)
87f263522aSJoseph Koshy  * profiles it can optionally merge entries from multiple processes
88f263522aSJoseph Koshy  * for a given executable into a single profile file.
89ee4910b1SJoseph Koshy  *
90ee4910b1SJoseph Koshy  * pmcstat(8) can also execute a command line and attach PMCs to the
91ee4910b1SJoseph Koshy  * resulting child process.  The protocol used is as follows:
92ee4910b1SJoseph Koshy  *
93ee4910b1SJoseph Koshy  * - parent creates a socketpair for two way communication and
94ee4910b1SJoseph Koshy  *   fork()s.
95ee4910b1SJoseph Koshy  * - subsequently:
96ee4910b1SJoseph Koshy  *
97ee4910b1SJoseph Koshy  *   /Parent/				/Child/
98ee4910b1SJoseph Koshy  *
99ee4910b1SJoseph Koshy  *   - Wait for childs token.
100ee4910b1SJoseph Koshy  *					- Sends token.
101ee4910b1SJoseph Koshy  *					- Awaits signal to start.
102ee4910b1SJoseph Koshy  *  - Attaches PMCs to the child's pid
103ee4910b1SJoseph Koshy  *    and starts them. Sets up
104ee4910b1SJoseph Koshy  *    monitoring for the child.
105ee4910b1SJoseph Koshy  *  - Signals child to start.
1063df5ecacSUlrich Spörlein  *					- Receives signal, attempts exec().
107ee4910b1SJoseph Koshy  *
108ee4910b1SJoseph Koshy  * After this point normal processing can happen.
109f263522aSJoseph Koshy  */
110f263522aSJoseph Koshy 
11115139246SJoseph Koshy /* Globals */
112ebccf1e3SJoseph Koshy 
113ebccf1e3SJoseph Koshy int		pmcstat_displayheight = DEFAULT_DISPLAY_HEIGHT;
1140b86b1bbSFabien Thomas int		pmcstat_displaywidth  = DEFAULT_DISPLAY_WIDTH;
115bf70beceSEd Schouten static int	pmcstat_sockpair[NSOCKPAIRFD];
116bf70beceSEd Schouten static int	pmcstat_kq;
117bf70beceSEd Schouten static kvm_t	*pmcstat_kvm;
118bf70beceSEd Schouten static struct kinfo_proc *pmcstat_plist;
1190b86b1bbSFabien Thomas struct pmcstat_args args;
120b6e28991SAleksandr Rybalko static bool	libpmc_initialized = false;
121ebccf1e3SJoseph Koshy 
122ea056888SAttilio Rao static void
123ea056888SAttilio Rao pmcstat_get_cpumask(const char *cpuspec, cpuset_t *cpumask)
124ea056888SAttilio Rao {
125ea056888SAttilio Rao 	int cpu;
126ea056888SAttilio Rao 	const char *s;
127ea056888SAttilio Rao 	char *end;
128ea056888SAttilio Rao 
129ea056888SAttilio Rao 	CPU_ZERO(cpumask);
130ea056888SAttilio Rao 	s = cpuspec;
131ea056888SAttilio Rao 
132ea056888SAttilio Rao 	do {
133ea056888SAttilio Rao 		cpu = strtol(s, &end, 0);
134ea056888SAttilio Rao 		if (cpu < 0 || end == s)
13537d6f8a9SDavid E. O'Brien 			errx(EX_USAGE,
13637d6f8a9SDavid E. O'Brien 			    "ERROR: Illegal CPU specification \"%s\".",
13737d6f8a9SDavid E. O'Brien 			    cpuspec);
138ea056888SAttilio Rao 		CPU_SET(cpu, cpumask);
139ea056888SAttilio Rao 		s = end + strspn(end, ", \t");
140ea056888SAttilio Rao 	} while (*s);
141ef16b9abSJohn Baldwin 	assert(!CPU_EMPTY(cpumask));
142ea056888SAttilio Rao }
143ea056888SAttilio Rao 
144ee4910b1SJoseph Koshy void
1450b86b1bbSFabien Thomas pmcstat_cleanup(void)
146ebccf1e3SJoseph Koshy {
1473ad36a25SStephen J. Kiernan 	struct pmcstat_ev *ev;
148ebccf1e3SJoseph Koshy 
149ebccf1e3SJoseph Koshy 	/* release allocated PMCs. */
1503ad36a25SStephen J. Kiernan 	STAILQ_FOREACH(ev, &args.pa_events, ev_next)
151ebccf1e3SJoseph Koshy 		if (ev->ev_pmcid != PMC_ID_INVALID) {
152b8fbd97bSJoseph Koshy 			if (pmc_stop(ev->ev_pmcid) < 0)
1533ad36a25SStephen J. Kiernan 				err(EX_OSERR,
1543ad36a25SStephen J. Kiernan 				    "ERROR: cannot stop pmc 0x%x \"%s\"",
15537d6f8a9SDavid E. O'Brien 				    ev->ev_pmcid, ev->ev_name);
156ebccf1e3SJoseph Koshy 			if (pmc_release(ev->ev_pmcid) < 0)
1573ad36a25SStephen J. Kiernan 				err(EX_OSERR,
1583ad36a25SStephen J. Kiernan 				    "ERROR: cannot release pmc 0x%x \"%s\"",
15937d6f8a9SDavid E. O'Brien 				    ev->ev_pmcid, ev->ev_name);
160ebccf1e3SJoseph Koshy 		}
161f263522aSJoseph Koshy 
162dc1d9d2eSJoseph Koshy 	/* de-configure the log file if present. */
1630b86b1bbSFabien Thomas 	if (args.pa_flags & (FLAG_HAS_PIPE | FLAG_HAS_OUTPUT_LOGFILE))
164dc1d9d2eSJoseph Koshy 		(void) pmc_configure_logfile(-1);
165dc1d9d2eSJoseph Koshy 
1660b86b1bbSFabien Thomas 	if (args.pa_logparser) {
1670b86b1bbSFabien Thomas 		pmclog_close(args.pa_logparser);
1680b86b1bbSFabien Thomas 		args.pa_logparser = NULL;
169f263522aSJoseph Koshy 	}
17015139246SJoseph Koshy 
171d27927f7SRuslan Bukin 	pmcstat_log_shutdown_logging();
172ee4910b1SJoseph Koshy }
173ee4910b1SJoseph Koshy 
174fbd8b34aSJoseph Koshy void
1750b86b1bbSFabien Thomas pmcstat_find_targets(const char *spec)
176fbd8b34aSJoseph Koshy {
177fbd8b34aSJoseph Koshy 	int n, nproc, pid, rv;
178fbd8b34aSJoseph Koshy 	struct pmcstat_target *pt;
179fbd8b34aSJoseph Koshy 	char errbuf[_POSIX2_LINE_MAX], *end;
180fbd8b34aSJoseph Koshy 	static struct kinfo_proc *kp;
181fbd8b34aSJoseph Koshy 	regex_t reg;
182fbd8b34aSJoseph Koshy 	regmatch_t regmatch;
183fbd8b34aSJoseph Koshy 
184fbd8b34aSJoseph Koshy 	/* First check if we've been given a process id. */
185fbd8b34aSJoseph Koshy       	pid = strtol(spec, &end, 0);
186fbd8b34aSJoseph Koshy 	if (end != spec && pid >= 0) {
187fbd8b34aSJoseph Koshy 		if ((pt = malloc(sizeof(*pt))) == NULL)
188fbd8b34aSJoseph Koshy 			goto outofmemory;
189fbd8b34aSJoseph Koshy 		pt->pt_pid = pid;
1900b86b1bbSFabien Thomas 		SLIST_INSERT_HEAD(&args.pa_targets, pt, pt_next);
191fbd8b34aSJoseph Koshy 		return;
192fbd8b34aSJoseph Koshy 	}
193fbd8b34aSJoseph Koshy 
194fbd8b34aSJoseph Koshy 	/* Otherwise treat arg as a regular expression naming processes. */
195fbd8b34aSJoseph Koshy 	if (pmcstat_kvm == NULL) {
196fbd8b34aSJoseph Koshy 		if ((pmcstat_kvm = kvm_openfiles(NULL, "/dev/null", NULL, 0,
197fbd8b34aSJoseph Koshy 		    errbuf)) == NULL)
198fbd8b34aSJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot open kernel \"%s\"",
199fbd8b34aSJoseph Koshy 			    errbuf);
200fbd8b34aSJoseph Koshy 		if ((pmcstat_plist = kvm_getprocs(pmcstat_kvm, KERN_PROC_PROC,
201fbd8b34aSJoseph Koshy 		    0, &nproc)) == NULL)
202fbd8b34aSJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot get process list: %s",
203fbd8b34aSJoseph Koshy 			    kvm_geterr(pmcstat_kvm));
2045c15d3c8SFabien Thomas 	} else
2055c15d3c8SFabien Thomas 		nproc = 0;
206fbd8b34aSJoseph Koshy 
207fbd8b34aSJoseph Koshy 	if ((rv = regcomp(&reg, spec, REG_EXTENDED|REG_NOSUB)) != 0) {
208fbd8b34aSJoseph Koshy 		regerror(rv, &reg, errbuf, sizeof(errbuf));
209fbd8b34aSJoseph Koshy 		err(EX_DATAERR, "ERROR: Failed to compile regex \"%s\": %s",
210fbd8b34aSJoseph Koshy 		    spec, errbuf);
211fbd8b34aSJoseph Koshy 	}
212fbd8b34aSJoseph Koshy 
213fbd8b34aSJoseph Koshy 	for (n = 0, kp = pmcstat_plist; n < nproc; n++, kp++) {
214fbd8b34aSJoseph Koshy 		if ((rv = regexec(&reg, kp->ki_comm, 1, &regmatch, 0)) == 0) {
215fbd8b34aSJoseph Koshy 			if ((pt = malloc(sizeof(*pt))) == NULL)
216fbd8b34aSJoseph Koshy 				goto outofmemory;
217fbd8b34aSJoseph Koshy 			pt->pt_pid = kp->ki_pid;
2180b86b1bbSFabien Thomas 			SLIST_INSERT_HEAD(&args.pa_targets, pt, pt_next);
219fbd8b34aSJoseph Koshy 		} else if (rv != REG_NOMATCH) {
220fbd8b34aSJoseph Koshy 			regerror(rv, &reg, errbuf, sizeof(errbuf));
221fbd8b34aSJoseph Koshy 			errx(EX_SOFTWARE, "ERROR: Regex evalation failed: %s",
222fbd8b34aSJoseph Koshy 			    errbuf);
223fbd8b34aSJoseph Koshy 		}
224fbd8b34aSJoseph Koshy 	}
225fbd8b34aSJoseph Koshy 
226fbd8b34aSJoseph Koshy 	regfree(&reg);
227fbd8b34aSJoseph Koshy 
228fbd8b34aSJoseph Koshy 	return;
229fbd8b34aSJoseph Koshy 
230fbd8b34aSJoseph Koshy  outofmemory:
231fbd8b34aSJoseph Koshy 	errx(EX_SOFTWARE, "Out of memory.");
232fbd8b34aSJoseph Koshy 	/*NOTREACHED*/
233fbd8b34aSJoseph Koshy }
234fbd8b34aSJoseph Koshy 
235ee4910b1SJoseph Koshy void
2360b86b1bbSFabien Thomas pmcstat_kill_process(void)
237fbd8b34aSJoseph Koshy {
238fbd8b34aSJoseph Koshy 	struct pmcstat_target *pt;
239fbd8b34aSJoseph Koshy 
2400b86b1bbSFabien Thomas 	assert(args.pa_flags & FLAG_HAS_COMMANDLINE);
241fbd8b34aSJoseph Koshy 
242fbd8b34aSJoseph Koshy 	/*
243fbd8b34aSJoseph Koshy 	 * If a command line was specified, it would be the very first
244fbd8b34aSJoseph Koshy 	 * in the list, before any other processes specified by -t.
245fbd8b34aSJoseph Koshy 	 */
2460b86b1bbSFabien Thomas 	pt = SLIST_FIRST(&args.pa_targets);
247fbd8b34aSJoseph Koshy 	assert(pt != NULL);
248fbd8b34aSJoseph Koshy 
249fbd8b34aSJoseph Koshy 	if (kill(pt->pt_pid, SIGINT) != 0)
250fbd8b34aSJoseph Koshy 		err(EX_OSERR, "ERROR: cannot signal child process");
251fbd8b34aSJoseph Koshy }
252fbd8b34aSJoseph Koshy 
253fbd8b34aSJoseph Koshy void
2540b86b1bbSFabien Thomas pmcstat_start_pmcs(void)
255ebccf1e3SJoseph Koshy {
256ebccf1e3SJoseph Koshy 	struct pmcstat_ev *ev;
257ebccf1e3SJoseph Koshy 
258fbd8b34aSJoseph Koshy 	STAILQ_FOREACH(ev, &args.pa_events, ev_next) {
259ebccf1e3SJoseph Koshy 
260ebccf1e3SJoseph Koshy 	    assert(ev->ev_pmcid != PMC_ID_INVALID);
261ebccf1e3SJoseph Koshy 
262ebccf1e3SJoseph Koshy 	    if (pmc_start(ev->ev_pmcid) < 0) {
263f263522aSJoseph Koshy 	        warn("ERROR: Cannot start pmc 0x%x \"%s\"",
264ebccf1e3SJoseph Koshy 		    ev->ev_pmcid, ev->ev_name);
2650b86b1bbSFabien Thomas 		pmcstat_cleanup();
266f263522aSJoseph Koshy 		exit(EX_OSERR);
267ebccf1e3SJoseph Koshy 	    }
268ebccf1e3SJoseph Koshy 	}
269ebccf1e3SJoseph Koshy }
270ebccf1e3SJoseph Koshy 
271ebccf1e3SJoseph Koshy void
2720b86b1bbSFabien Thomas pmcstat_print_headers(void)
273ebccf1e3SJoseph Koshy {
274ebccf1e3SJoseph Koshy 	struct pmcstat_ev *ev;
275ee4910b1SJoseph Koshy 	int c, w;
276ebccf1e3SJoseph Koshy 
2770b86b1bbSFabien Thomas 	(void) fprintf(args.pa_printfile, PRINT_HEADER_PREFIX);
278ebccf1e3SJoseph Koshy 
2790b86b1bbSFabien Thomas 	STAILQ_FOREACH(ev, &args.pa_events, ev_next) {
280ebccf1e3SJoseph Koshy 		if (PMC_IS_SAMPLING_MODE(ev->ev_mode))
281ebccf1e3SJoseph Koshy 			continue;
282ebccf1e3SJoseph Koshy 
283ebccf1e3SJoseph Koshy 		c = PMC_IS_SYSTEM_MODE(ev->ev_mode) ? 's' : 'p';
284ebccf1e3SJoseph Koshy 
285ee4910b1SJoseph Koshy 		if (ev->ev_fieldskip != 0)
2860b86b1bbSFabien Thomas 			(void) fprintf(args.pa_printfile, "%*s",
287ee4910b1SJoseph Koshy 			    ev->ev_fieldskip, "");
288ee4910b1SJoseph Koshy 		w = ev->ev_fieldwidth - ev->ev_fieldskip - 2;
289ee4910b1SJoseph Koshy 
290ee4910b1SJoseph Koshy 		if (c == 's')
2910b86b1bbSFabien Thomas 			(void) fprintf(args.pa_printfile, "s/%02d/%-*s ",
292ee4910b1SJoseph Koshy 			    ev->ev_cpu, w-3, ev->ev_name);
293ee4910b1SJoseph Koshy 		else
2940b86b1bbSFabien Thomas 			(void) fprintf(args.pa_printfile, "p/%*s ", w,
295ebccf1e3SJoseph Koshy 			    ev->ev_name);
296ebccf1e3SJoseph Koshy 	}
297ebccf1e3SJoseph Koshy 
2980b86b1bbSFabien Thomas 	(void) fflush(args.pa_printfile);
299ebccf1e3SJoseph Koshy }
300ebccf1e3SJoseph Koshy 
301ebccf1e3SJoseph Koshy void
3020b86b1bbSFabien Thomas pmcstat_print_counters(void)
303ebccf1e3SJoseph Koshy {
304ebccf1e3SJoseph Koshy 	int extra_width;
305ebccf1e3SJoseph Koshy 	struct pmcstat_ev *ev;
306ebccf1e3SJoseph Koshy 	pmc_value_t value;
307ebccf1e3SJoseph Koshy 
308ebccf1e3SJoseph Koshy 	extra_width = sizeof(PRINT_HEADER_PREFIX) - 1;
309ebccf1e3SJoseph Koshy 
3100b86b1bbSFabien Thomas 	STAILQ_FOREACH(ev, &args.pa_events, ev_next) {
311ebccf1e3SJoseph Koshy 
312ebccf1e3SJoseph Koshy 		/* skip sampling mode counters */
313ebccf1e3SJoseph Koshy 		if (PMC_IS_SAMPLING_MODE(ev->ev_mode))
314ebccf1e3SJoseph Koshy 			continue;
315ebccf1e3SJoseph Koshy 
316ebccf1e3SJoseph Koshy 		if (pmc_read(ev->ev_pmcid, &value) < 0)
31737d6f8a9SDavid E. O'Brien 			err(EX_OSERR, "ERROR: Cannot read pmc \"%s\"",
31837d6f8a9SDavid E. O'Brien 			    ev->ev_name);
319ebccf1e3SJoseph Koshy 
3200b86b1bbSFabien Thomas 		(void) fprintf(args.pa_printfile, "%*ju ",
32115139246SJoseph Koshy 		    ev->ev_fieldwidth + extra_width,
32215139246SJoseph Koshy 		    (uintmax_t) ev->ev_cumulative ? value :
32315139246SJoseph Koshy 		    (value - ev->ev_saved));
32415139246SJoseph Koshy 
325ebccf1e3SJoseph Koshy 		if (ev->ev_cumulative == 0)
326ebccf1e3SJoseph Koshy 			ev->ev_saved = value;
327ebccf1e3SJoseph Koshy 		extra_width = 0;
328ebccf1e3SJoseph Koshy 	}
329ebccf1e3SJoseph Koshy 
3300b86b1bbSFabien Thomas 	(void) fflush(args.pa_printfile);
331ebccf1e3SJoseph Koshy }
332ebccf1e3SJoseph Koshy 
333ebccf1e3SJoseph Koshy /*
334ebccf1e3SJoseph Koshy  * Print output
335ebccf1e3SJoseph Koshy  */
336ebccf1e3SJoseph Koshy 
337ebccf1e3SJoseph Koshy void
3380b86b1bbSFabien Thomas pmcstat_print_pmcs(void)
339ebccf1e3SJoseph Koshy {
340ebccf1e3SJoseph Koshy 	static int linecount = 0;
341ebccf1e3SJoseph Koshy 
34215139246SJoseph Koshy 	/* check if we need to print a header line */
343ebccf1e3SJoseph Koshy 	if (++linecount > pmcstat_displayheight) {
3440b86b1bbSFabien Thomas 		(void) fprintf(args.pa_printfile, "\n");
345ebccf1e3SJoseph Koshy 		linecount = 1;
346ebccf1e3SJoseph Koshy 	}
347ebccf1e3SJoseph Koshy 	if (linecount == 1)
3480b86b1bbSFabien Thomas 		pmcstat_print_headers();
3490b86b1bbSFabien Thomas 	(void) fprintf(args.pa_printfile, "\n");
350ebccf1e3SJoseph Koshy 
3510b86b1bbSFabien Thomas 	pmcstat_print_counters();
352ebccf1e3SJoseph Koshy }
353ebccf1e3SJoseph Koshy 
354ebccf1e3SJoseph Koshy void
355ebccf1e3SJoseph Koshy pmcstat_show_usage(void)
356ebccf1e3SJoseph Koshy {
357ebccf1e3SJoseph Koshy 	errx(EX_USAGE,
358ebccf1e3SJoseph Koshy 	    "[options] [commandline]\n"
359ebccf1e3SJoseph Koshy 	    "\t Measure process and/or system performance using hardware\n"
360ebccf1e3SJoseph Koshy 	    "\t performance monitoring counters.\n"
361ebccf1e3SJoseph Koshy 	    "\t Options include:\n"
362f263522aSJoseph Koshy 	    "\t -C\t\t (toggle) show cumulative counts\n"
363f263522aSJoseph Koshy 	    "\t -D path\t create profiles in directory \"path\"\n"
364f263522aSJoseph Koshy 	    "\t -E\t\t (toggle) show counts at process exit\n"
3650b86b1bbSFabien Thomas 	    "\t -F file\t write a system-wide callgraph (Kcachegrind format)"
3660b86b1bbSFabien Thomas 		" to \"file\"\n"
367b6010f9eSJoseph Koshy 	    "\t -G file\t write a system-wide callgraph to \"file\"\n"
368959826caSMatt Macy 	    "\t -I\t\t don't resolve leaf function name, show address instead\n"
369959826caSMatt Macy 	    "\t -L\t\t list all counters available on this host\n"
37049874f6eSJoseph Koshy 	    "\t -M file\t print executable/gmon file map to \"file\"\n"
371b6010f9eSJoseph Koshy 	    "\t -N\t\t (toggle) capture callchains\n"
372f263522aSJoseph Koshy 	    "\t -O file\t send log output to \"file\"\n"
373f263522aSJoseph Koshy 	    "\t -P spec\t allocate a process-private sampling PMC\n"
374f263522aSJoseph Koshy 	    "\t -R file\t read events from \"file\"\n"
375f263522aSJoseph Koshy 	    "\t -S spec\t allocate a system-wide sampling PMC\n"
3760b86b1bbSFabien Thomas 	    "\t -T\t\t start in top mode\n"
377d3231fbdSMateusz Piotrowski 	    "\t -U \t\t merged user kernel stack capture\n"
378f263522aSJoseph Koshy 	    "\t -W\t\t (toggle) show counts per context switch\n"
3795ae1451dSScott Long 	    "\t -a file\t print sampled PCs and callgraph to \"file\"\n"
380ee4910b1SJoseph Koshy 	    "\t -c cpu-list\t set cpus for subsequent system-wide PMCs\n"
381f263522aSJoseph Koshy 	    "\t -d\t\t (toggle) track descendants\n"
382465dadb5SJonathan T. Looney 	    "\t -e\t\t use wide history counter for gprof(1) output\n"
3830b86b1bbSFabien Thomas 	    "\t -f spec\t pass \"spec\" to as plugin option\n"
384f263522aSJoseph Koshy 	    "\t -g\t\t produce gprof(1) compatible profiles\n"
385959826caSMatt Macy 	    "\t -i lwp\t\t filter on thread id \"lwp\" in post-processing\n"
38672fe532cSGeorge V. Neville-Neil 	    "\t -l secs\t set duration time\n"
38733b12a85SJim Harris 	    "\t -m file\t print sampled PCs to \"file\"\n"
388ebccf1e3SJoseph Koshy 	    "\t -n rate\t set sampling rate\n"
389ebccf1e3SJoseph Koshy 	    "\t -o file\t send print output to \"file\"\n"
390f263522aSJoseph Koshy 	    "\t -p spec\t allocate a process-private counting PMC\n"
39149874f6eSJoseph Koshy 	    "\t -q\t\t suppress verbosity\n"
39249874f6eSJoseph Koshy 	    "\t -r fsroot\t specify FS root directory\n"
393f263522aSJoseph Koshy 	    "\t -s spec\t allocate a system-wide counting PMC\n"
39467442c6eSJoseph Koshy 	    "\t -t process-spec attach to running processes matching "
39567442c6eSJoseph Koshy 		"\"process-spec\"\n"
396959826caSMatt Macy 	    "\t -u spec \t provide short description of counters matching spec\n"
39749874f6eSJoseph Koshy 	    "\t -v\t\t increase verbosity\n"
398b6010f9eSJoseph Koshy 	    "\t -w secs\t set printing time interval\n"
399b6010f9eSJoseph Koshy 	    "\t -z depth\t limit callchain display depth"
400ebccf1e3SJoseph Koshy 	);
401ebccf1e3SJoseph Koshy }
402ebccf1e3SJoseph Koshy 
403ebccf1e3SJoseph Koshy /*
4040b86b1bbSFabien Thomas  * At exit handler for top mode
4050b86b1bbSFabien Thomas  */
4060b86b1bbSFabien Thomas 
4070b86b1bbSFabien Thomas void
4080b86b1bbSFabien Thomas pmcstat_topexit(void)
4090b86b1bbSFabien Thomas {
4100b86b1bbSFabien Thomas 	if (!args.pa_toptty)
4110b86b1bbSFabien Thomas 		return;
4120b86b1bbSFabien Thomas 
4130b86b1bbSFabien Thomas 	/*
4140b86b1bbSFabien Thomas 	 * Shutdown ncurses.
4150b86b1bbSFabien Thomas 	 */
4160b86b1bbSFabien Thomas 	clrtoeol();
4170b86b1bbSFabien Thomas 	refresh();
4180b86b1bbSFabien Thomas 	endwin();
4190b86b1bbSFabien Thomas }
4200b86b1bbSFabien Thomas 
421b6e28991SAleksandr Rybalko static inline void
422b6e28991SAleksandr Rybalko libpmc_initialize(int *npmc)
423b6e28991SAleksandr Rybalko {
424b6e28991SAleksandr Rybalko 
425b6e28991SAleksandr Rybalko 	if (libpmc_initialized)
426b6e28991SAleksandr Rybalko 		return;
427b6e28991SAleksandr Rybalko 	if (pmc_init() < 0)
428b6e28991SAleksandr Rybalko 		err(EX_UNAVAILABLE, "ERROR: Initialization of the pmc(3)"
429b6e28991SAleksandr Rybalko 		    " library failed");
430b6e28991SAleksandr Rybalko 
431b6e28991SAleksandr Rybalko 	/* assume all CPUs are identical */
432b6e28991SAleksandr Rybalko 	if ((*npmc = pmc_npmc(0)) < 0)
433b6e28991SAleksandr Rybalko 		err(EX_OSERR, "ERROR: Cannot determine the number of PMCs on "
434b6e28991SAleksandr Rybalko 		    "CPU %d", 0);
435b6e28991SAleksandr Rybalko 	libpmc_initialized = true;
436b6e28991SAleksandr Rybalko }
4370b86b1bbSFabien Thomas /*
438ebccf1e3SJoseph Koshy  * Main
439ebccf1e3SJoseph Koshy  */
440ebccf1e3SJoseph Koshy 
441ebccf1e3SJoseph Koshy int
442ebccf1e3SJoseph Koshy main(int argc, char **argv)
443ebccf1e3SJoseph Koshy {
444b6e28991SAleksandr Rybalko 	cpuset_t cpumask, dommask, rootmask;
445ebccf1e3SJoseph Koshy 	double interval;
44672fe532cSGeorge V. Neville-Neil 	double duration;
447ef16b9abSJohn Baldwin 	int option, npmc;
448357405d9SMatt Macy 	int c, check_driver_stats;
449b6010f9eSJoseph Koshy 	int do_callchain, do_descendants, do_logproccsw, do_logprocexit;
450b6e28991SAleksandr Rybalko 	int do_print, do_read, do_listcounters, do_descr, domains;
451b6e28991SAleksandr Rybalko 	int do_userspace, i;
45238fd3834SJohn Baldwin 	size_t len;
453b6010f9eSJoseph Koshy 	int graphdepth;
454ac708fddSFabien Thomas 	int pipefd[2], rfd;
455f263522aSJoseph Koshy 	int use_cumulative_counts;
4560b86b1bbSFabien Thomas 	short cf, cb;
457357405d9SMatt Macy 	uint64_t current_sampling_count;
458*53d0b9e4SJessica Clarke 	char *end, *event;
459b6010f9eSJoseph Koshy 	const char *errmsg, *graphfilename;
460f263522aSJoseph Koshy 	enum pmcstat_state runstate;
4617fbf3da2SJoseph Koshy 	struct pmc_driverstats ds_start, ds_end;
462ebccf1e3SJoseph Koshy 	struct pmcstat_ev *ev;
463ebccf1e3SJoseph Koshy 	struct sigaction sa;
464ebccf1e3SJoseph Koshy 	struct kevent kev;
465ebccf1e3SJoseph Koshy 	struct winsize ws;
46615139246SJoseph Koshy 	struct stat sb;
467b6e28991SAleksandr Rybalko 	uint32_t caps;
468ebccf1e3SJoseph Koshy 
4697fbf3da2SJoseph Koshy 	check_driver_stats      = 0;
470e98bbcf9SMatt Macy 	current_sampling_count  = 0;
471b6010f9eSJoseph Koshy 	do_callchain		= 1;
472959826caSMatt Macy 	do_descr                = 0;
473ebccf1e3SJoseph Koshy 	do_descendants          = 0;
4745de96e33SMatt Macy 	do_userspace            = 0;
475f263522aSJoseph Koshy 	do_logproccsw           = 0;
476f263522aSJoseph Koshy 	do_logprocexit          = 0;
477959826caSMatt Macy 	do_listcounters         = 0;
478b6e28991SAleksandr Rybalko 	domains			= 0;
479ebccf1e3SJoseph Koshy 	use_cumulative_counts   = 0;
480b6010f9eSJoseph Koshy 	graphfilename		= "-";
481f263522aSJoseph Koshy 	args.pa_required	= 0;
482ebccf1e3SJoseph Koshy 	args.pa_flags		= 0;
48349874f6eSJoseph Koshy 	args.pa_verbosity	= 1;
48415139246SJoseph Koshy 	args.pa_logfd		= -1;
48549874f6eSJoseph Koshy 	args.pa_fsroot		= "";
48615139246SJoseph Koshy 	args.pa_samplesdir	= ".";
48715139246SJoseph Koshy 	args.pa_printfile	= stderr;
488b6010f9eSJoseph Koshy 	args.pa_graphdepth	= DEFAULT_CALLGRAPH_DEPTH;
489b6010f9eSJoseph Koshy 	args.pa_graphfile	= NULL;
490ebccf1e3SJoseph Koshy 	args.pa_interval	= DEFAULT_WAIT_INTERVAL;
49149874f6eSJoseph Koshy 	args.pa_mapfilename	= NULL;
492b6010f9eSJoseph Koshy 	args.pa_inputpath	= NULL;
493b6010f9eSJoseph Koshy 	args.pa_outputpath	= NULL;
4940b86b1bbSFabien Thomas 	args.pa_pplugin		= PMCSTAT_PL_NONE;
4950b86b1bbSFabien Thomas 	args.pa_plugin		= PMCSTAT_PL_NONE;
4960b86b1bbSFabien Thomas 	args.pa_ctdumpinstr	= 1;
4970b86b1bbSFabien Thomas 	args.pa_topmode		= PMCSTAT_TOP_DELTA;
4980b86b1bbSFabien Thomas 	args.pa_toptty		= 0;
4990b86b1bbSFabien Thomas 	args.pa_topcolor	= 0;
5000b86b1bbSFabien Thomas 	args.pa_mergepmc	= 0;
50172fe532cSGeorge V. Neville-Neil 	args.pa_duration	= 0.0;
502fbd8b34aSJoseph Koshy 	STAILQ_INIT(&args.pa_events);
503fbd8b34aSJoseph Koshy 	SLIST_INIT(&args.pa_targets);
504bea6aa93SJoseph Koshy 	bzero(&ds_start, sizeof(ds_start));
505bea6aa93SJoseph Koshy 	bzero(&ds_end, sizeof(ds_end));
506ebccf1e3SJoseph Koshy 	ev = NULL;
507ae266f9fSMatt Macy 	event = NULL;
508b6e28991SAleksandr Rybalko 	caps = 0;
509ea056888SAttilio Rao 	CPU_ZERO(&cpumask);
510ebccf1e3SJoseph Koshy 
511b6e28991SAleksandr Rybalko 	len = sizeof(domains);
512b6e28991SAleksandr Rybalko 	if (sysctlbyname("vm.ndomains", &domains, &len, NULL, 0) == -1)
513b6e28991SAleksandr Rybalko 		err(EX_OSERR, "ERROR: Cannot get number of domains");
51438fd3834SJohn Baldwin 
515b6010f9eSJoseph Koshy 	/*
516ef16b9abSJohn Baldwin 	 * The initial CPU mask specifies the root mask of this process
517ef16b9abSJohn Baldwin 	 * which is usually all CPUs in the system.
518b6010f9eSJoseph Koshy 	 */
519ef16b9abSJohn Baldwin 	if (cpuset_getaffinity(CPU_LEVEL_ROOT, CPU_WHICH_PID, -1,
520ef16b9abSJohn Baldwin 	    sizeof(rootmask), &rootmask) == -1)
521ef16b9abSJohn Baldwin 		err(EX_OSERR, "ERROR: Cannot determine the root set of CPUs");
522ef16b9abSJohn Baldwin 	CPU_COPY(&rootmask, &cpumask);
523ee4910b1SJoseph Koshy 
52449874f6eSJoseph Koshy 	while ((option = getopt(argc, argv,
52594e9ef85SMateusz Guzik 	    "ACD:EF:G:ILM:NO:P:R:S:TUWZa:c:def:gi:k:l:m:n:o:p:qr:s:t:u:vw:z:")) != -1)
526ebccf1e3SJoseph Koshy 		switch (option) {
52794e9ef85SMateusz Guzik 		case 'A':
52894e9ef85SMateusz Guzik 			args.pa_flags |= FLAG_SKIP_TOP_FN_RES;
52994e9ef85SMateusz Guzik 			break;
53094e9ef85SMateusz Guzik 
53185ec49f3SAdrian Chadd 		case 'a':	/* Annotate + callgraph */
53285ec49f3SAdrian Chadd 			args.pa_flags |= FLAG_DO_ANNOTATE;
53385ec49f3SAdrian Chadd 			args.pa_plugin = PMCSTAT_PL_ANNOTATE_CG;
53485ec49f3SAdrian Chadd 			graphfilename  = optarg;
53585ec49f3SAdrian Chadd 			break;
53685ec49f3SAdrian Chadd 
537ebccf1e3SJoseph Koshy 		case 'C':	/* cumulative values */
538ebccf1e3SJoseph Koshy 			use_cumulative_counts = !use_cumulative_counts;
539f263522aSJoseph Koshy 			args.pa_required |= FLAG_HAS_COUNTING_PMCS;
540ebccf1e3SJoseph Koshy 			break;
541ebccf1e3SJoseph Koshy 
542ebccf1e3SJoseph Koshy 		case 'c':	/* CPU */
543ef16b9abSJohn Baldwin 			if (optarg[0] == '*' && optarg[1] == '\0')
544ef16b9abSJohn Baldwin 				CPU_COPY(&rootmask, &cpumask);
545ef16b9abSJohn Baldwin 			else
546ea056888SAttilio Rao 				pmcstat_get_cpumask(optarg, &cpumask);
547ee4910b1SJoseph Koshy 
548e7ef1de5SFabien Thomas 			args.pa_flags	 |= FLAGS_HAS_CPUMASK;
549f263522aSJoseph Koshy 			args.pa_required |= FLAG_HAS_SYSTEM_PMCS;
550ebccf1e3SJoseph Koshy 			break;
551ebccf1e3SJoseph Koshy 
55215139246SJoseph Koshy 		case 'D':
55315139246SJoseph Koshy 			if (stat(optarg, &sb) < 0)
55415139246SJoseph Koshy 				err(EX_OSERR, "ERROR: Cannot stat \"%s\"",
55515139246SJoseph Koshy 				    optarg);
55615139246SJoseph Koshy 			if (!S_ISDIR(sb.st_mode))
55737d6f8a9SDavid E. O'Brien 				errx(EX_USAGE,
55837d6f8a9SDavid E. O'Brien 				    "ERROR: \"%s\" is not a directory.",
55937d6f8a9SDavid E. O'Brien 				    optarg);
56015139246SJoseph Koshy 			args.pa_samplesdir = optarg;
56115139246SJoseph Koshy 			args.pa_flags     |= FLAG_HAS_SAMPLESDIR;
56215139246SJoseph Koshy 			args.pa_required  |= FLAG_DO_GPROF;
56315139246SJoseph Koshy 			break;
56415139246SJoseph Koshy 
565ebccf1e3SJoseph Koshy 		case 'd':	/* toggle descendents */
566ebccf1e3SJoseph Koshy 			do_descendants = !do_descendants;
567f263522aSJoseph Koshy 			args.pa_required |= FLAG_HAS_PROCESS_PMCS;
568f263522aSJoseph Koshy 			break;
569f263522aSJoseph Koshy 
570959826caSMatt Macy 		case 'E':	/* log process exit */
571959826caSMatt Macy 			do_logprocexit = !do_logprocexit;
572959826caSMatt Macy 			args.pa_required |= (FLAG_HAS_PROCESS_PMCS |
573959826caSMatt Macy 			    FLAG_HAS_COUNTING_PMCS | FLAG_HAS_OUTPUT_LOGFILE);
574959826caSMatt Macy 			break;
575959826caSMatt Macy 
576465dadb5SJonathan T. Looney 		case 'e':	/* wide gprof metrics */
577465dadb5SJonathan T. Looney 			args.pa_flags |= FLAG_DO_WIDE_GPROF_HC;
578465dadb5SJonathan T. Looney 			break;
579465dadb5SJonathan T. Looney 
5800b86b1bbSFabien Thomas 		case 'F':	/* produce a system-wide calltree */
5810b86b1bbSFabien Thomas 			args.pa_flags |= FLAG_DO_CALLGRAPHS;
5820b86b1bbSFabien Thomas 			args.pa_plugin = PMCSTAT_PL_CALLTREE;
5830b86b1bbSFabien Thomas 			graphfilename = optarg;
5840b86b1bbSFabien Thomas 			break;
5850b86b1bbSFabien Thomas 
5860b86b1bbSFabien Thomas 		case 'f':	/* plugins options */
5870b86b1bbSFabien Thomas 			if (args.pa_plugin == PMCSTAT_PL_NONE)
5880b86b1bbSFabien Thomas 				err(EX_USAGE, "ERROR: Need -g/-G/-m/-T.");
5890b86b1bbSFabien Thomas 			pmcstat_pluginconfigure_log(optarg);
5900b86b1bbSFabien Thomas 			break;
5910b86b1bbSFabien Thomas 
592b6010f9eSJoseph Koshy 		case 'G':	/* produce a system-wide callgraph */
593b6010f9eSJoseph Koshy 			args.pa_flags |= FLAG_DO_CALLGRAPHS;
5940b86b1bbSFabien Thomas 			args.pa_plugin = PMCSTAT_PL_CALLGRAPH;
595b6010f9eSJoseph Koshy 			graphfilename = optarg;
596b6010f9eSJoseph Koshy 			break;
597b6010f9eSJoseph Koshy 
598f263522aSJoseph Koshy 		case 'g':	/* produce gprof compatible profiles */
599f263522aSJoseph Koshy 			args.pa_flags |= FLAG_DO_GPROF;
6000b86b1bbSFabien Thomas 			args.pa_pplugin = PMCSTAT_PL_CALLGRAPH;
6010b86b1bbSFabien Thomas 			args.pa_plugin	= PMCSTAT_PL_GPROF;
602f263522aSJoseph Koshy 			break;
603f263522aSJoseph Koshy 
604959826caSMatt Macy 		case 'i':
605959826caSMatt Macy 			args.pa_flags |= FLAG_FILTER_THREAD_ID;
606959826caSMatt Macy 			args.pa_tid = strtol(optarg, &end, 0);
607959826caSMatt Macy 			break;
608959826caSMatt Macy 
60994e9ef85SMateusz Guzik 		case 'I':
61094e9ef85SMateusz Guzik 			args.pa_flags |= FLAG_SHOW_OFFSET;
61194e9ef85SMateusz Guzik 			break;
61294e9ef85SMateusz Guzik 
61315139246SJoseph Koshy 		case 'k':	/* pathname to the kernel */
614*53d0b9e4SJessica Clarke 			warnx("WARNING: -k is obsolete, has no effect "
615*53d0b9e4SJessica Clarke 			    "and will be removed in FreeBSD 15.");
616f263522aSJoseph Koshy 			break;
617f263522aSJoseph Koshy 
6180b5dc7f6SMatt Macy 		case 'L':
619959826caSMatt Macy 			do_listcounters = 1;
6200b5dc7f6SMatt Macy 			break;
621959826caSMatt Macy 
62272fe532cSGeorge V. Neville-Neil 		case 'l':	/* time duration in seconds */
62372fe532cSGeorge V. Neville-Neil 			duration = strtod(optarg, &end);
62472fe532cSGeorge V. Neville-Neil 			if (*end != '\0' || duration <= 0)
62572fe532cSGeorge V. Neville-Neil 				errx(EX_USAGE, "ERROR: Illegal duration time "
62672fe532cSGeorge V. Neville-Neil 				    "value \"%s\".", optarg);
62772fe532cSGeorge V. Neville-Neil 			args.pa_flags |= FLAG_HAS_DURATION;
62872fe532cSGeorge V. Neville-Neil 			args.pa_duration = duration;
62972fe532cSGeorge V. Neville-Neil 			break;
63072fe532cSGeorge V. Neville-Neil 
631c0252222SAttilio Rao 		case 'm':
6320b86b1bbSFabien Thomas 			args.pa_flags |= FLAG_DO_ANNOTATE;
6330b86b1bbSFabien Thomas 			args.pa_plugin = PMCSTAT_PL_ANNOTATE;
634c0252222SAttilio Rao 			graphfilename  = optarg;
635c0252222SAttilio Rao 			break;
636c0252222SAttilio Rao 
63749874f6eSJoseph Koshy 		case 'M':	/* mapfile */
63849874f6eSJoseph Koshy 			args.pa_mapfilename = optarg;
63949874f6eSJoseph Koshy 			break;
64049874f6eSJoseph Koshy 
641b6010f9eSJoseph Koshy 		case 'N':
642b6010f9eSJoseph Koshy 			do_callchain = !do_callchain;
643b6010f9eSJoseph Koshy 			args.pa_required |= FLAG_HAS_SAMPLING_PMCS;
644b6010f9eSJoseph Koshy 			break;
645b6010f9eSJoseph Koshy 
646ebccf1e3SJoseph Koshy 		case 'p':	/* process virtual counting PMC */
647ebccf1e3SJoseph Koshy 		case 's':	/* system-wide counting PMC */
648ebccf1e3SJoseph Koshy 		case 'P':	/* process virtual sampling PMC */
649ebccf1e3SJoseph Koshy 		case 'S':	/* system-wide sampling PMC */
650b6e28991SAleksandr Rybalko 			caps = 0;
651ebccf1e3SJoseph Koshy 			if ((ev = malloc(sizeof(*ev))) == NULL)
652f263522aSJoseph Koshy 				errx(EX_SOFTWARE, "ERROR: Out of memory.");
653ebccf1e3SJoseph Koshy 
654ebccf1e3SJoseph Koshy 			switch (option) {
655ebccf1e3SJoseph Koshy 			case 'p': ev->ev_mode = PMC_MODE_TC; break;
656ebccf1e3SJoseph Koshy 			case 's': ev->ev_mode = PMC_MODE_SC; break;
657ebccf1e3SJoseph Koshy 			case 'P': ev->ev_mode = PMC_MODE_TS; break;
658ebccf1e3SJoseph Koshy 			case 'S': ev->ev_mode = PMC_MODE_SS; break;
659ebccf1e3SJoseph Koshy 			}
660ebccf1e3SJoseph Koshy 
661f263522aSJoseph Koshy 			if (option == 'P' || option == 'p') {
662f263522aSJoseph Koshy 				args.pa_flags |= FLAG_HAS_PROCESS_PMCS;
66315139246SJoseph Koshy 				args.pa_required |= (FLAG_HAS_COMMANDLINE |
664fbd8b34aSJoseph Koshy 				    FLAG_HAS_TARGET);
665f263522aSJoseph Koshy 			}
666ebccf1e3SJoseph Koshy 
667f263522aSJoseph Koshy 			if (option == 'P' || option == 'S') {
668f263522aSJoseph Koshy 				args.pa_flags |= FLAG_HAS_SAMPLING_PMCS;
66915139246SJoseph Koshy 				args.pa_required |= (FLAG_HAS_PIPE |
67015139246SJoseph Koshy 				    FLAG_HAS_OUTPUT_LOGFILE);
671f263522aSJoseph Koshy 			}
672ebccf1e3SJoseph Koshy 
673ebccf1e3SJoseph Koshy 			if (option == 'p' || option == 's')
674f263522aSJoseph Koshy 				args.pa_flags |= FLAG_HAS_COUNTING_PMCS;
675f263522aSJoseph Koshy 
676f263522aSJoseph Koshy 			if (option == 's' || option == 'S')
677f263522aSJoseph Koshy 				args.pa_flags |= FLAG_HAS_SYSTEM_PMCS;
678ebccf1e3SJoseph Koshy 
679ebccf1e3SJoseph Koshy 			ev->ev_spec = strdup(optarg);
6803ad36a25SStephen J. Kiernan 			if (ev->ev_spec == NULL)
6813ad36a25SStephen J. Kiernan 				errx(EX_SOFTWARE, "ERROR: Out of memory.");
682ebccf1e3SJoseph Koshy 
683ebccf1e3SJoseph Koshy 			if (option == 'S' || option == 'P')
684959826caSMatt Macy 				ev->ev_count = current_sampling_count ? current_sampling_count : pmc_pmu_sample_rate_get(ev->ev_spec);
685ebccf1e3SJoseph Koshy 			else
686edcf9e59SMitchell Horne 				ev->ev_count = 0;
687ebccf1e3SJoseph Koshy 
688ef16b9abSJohn Baldwin 			if (option == 'S' || option == 's')
6897816cb4aSJohn Baldwin 				ev->ev_cpu = CPU_FFS(&cpumask) - 1;
690ef16b9abSJohn Baldwin 			else
691ebccf1e3SJoseph Koshy 				ev->ev_cpu = PMC_CPU_ANY;
692ebccf1e3SJoseph Koshy 
693f263522aSJoseph Koshy 			ev->ev_flags = 0;
6945de96e33SMatt Macy 			if (do_callchain) {
695b6010f9eSJoseph Koshy 				ev->ev_flags |= PMC_F_CALLCHAIN;
6965de96e33SMatt Macy 				if (do_userspace)
6975de96e33SMatt Macy 					ev->ev_flags |= PMC_F_USERCALLCHAIN;
6985de96e33SMatt Macy 			}
699f263522aSJoseph Koshy 			if (do_descendants)
700f263522aSJoseph Koshy 				ev->ev_flags |= PMC_F_DESCENDANTS;
701f263522aSJoseph Koshy 			if (do_logprocexit)
702f263522aSJoseph Koshy 				ev->ev_flags |= PMC_F_LOG_PROCEXIT;
703f263522aSJoseph Koshy 			if (do_logproccsw)
704f263522aSJoseph Koshy 				ev->ev_flags |= PMC_F_LOG_PROCCSW;
705f263522aSJoseph Koshy 
706ebccf1e3SJoseph Koshy 			ev->ev_cumulative  = use_cumulative_counts;
707ebccf1e3SJoseph Koshy 
708ebccf1e3SJoseph Koshy 			ev->ev_saved = 0LL;
709ebccf1e3SJoseph Koshy 			ev->ev_pmcid = PMC_ID_INVALID;
710ebccf1e3SJoseph Koshy 
711ebccf1e3SJoseph Koshy 			/* extract event name */
712ebccf1e3SJoseph Koshy 			c = strcspn(optarg, ", \t");
713ebccf1e3SJoseph Koshy 			ev->ev_name = malloc(c + 1);
7143ad36a25SStephen J. Kiernan 			if (ev->ev_name == NULL)
7153ad36a25SStephen J. Kiernan 				errx(EX_SOFTWARE, "ERROR: Out of memory.");
716ebccf1e3SJoseph Koshy 			(void) strncpy(ev->ev_name, optarg, c);
717ebccf1e3SJoseph Koshy 			*(ev->ev_name + c) = '\0';
718b6e28991SAleksandr Rybalko 			libpmc_initialize(&npmc);
719b6e28991SAleksandr Rybalko 			if (args.pa_flags & FLAG_HAS_SYSTEM_PMCS) {
720b6e28991SAleksandr Rybalko 				if (pmc_allocate(ev->ev_spec, ev->ev_mode,
721b6e28991SAleksandr Rybalko 				    ev->ev_flags, ev->ev_cpu, &ev->ev_pmcid,
722b6e28991SAleksandr Rybalko 				    ev->ev_count) < 0)
723b6e28991SAleksandr Rybalko 					err(EX_OSERR, "ERROR: Cannot allocate "
724b6e28991SAleksandr Rybalko 					    "system-mode pmc with specification"
725b6e28991SAleksandr Rybalko 					    " \"%s\"", ev->ev_spec);
726b6e28991SAleksandr Rybalko 				if (pmc_capabilities(ev->ev_pmcid, &caps)) {
727b6e28991SAleksandr Rybalko 					pmc_release(ev->ev_pmcid);
728b6e28991SAleksandr Rybalko 					err(EX_OSERR, "ERROR: Cannot get pmc "
729b6e28991SAleksandr Rybalko 					    "capabilities");
730b6e28991SAleksandr Rybalko 				}
731b6e28991SAleksandr Rybalko 			}
732b6e28991SAleksandr Rybalko 
733ebccf1e3SJoseph Koshy 
734fbd8b34aSJoseph Koshy 			STAILQ_INSERT_TAIL(&args.pa_events, ev, ev_next);
735ebccf1e3SJoseph Koshy 
736b6e28991SAleksandr Rybalko 			if ((caps & PMC_CAP_SYSWIDE) == PMC_CAP_SYSWIDE)
737b6e28991SAleksandr Rybalko 				break;
738b6e28991SAleksandr Rybalko 			if ((caps & PMC_CAP_DOMWIDE) == PMC_CAP_DOMWIDE) {
739b6e28991SAleksandr Rybalko 				CPU_ZERO(&cpumask);
740b6e28991SAleksandr Rybalko 				/*
741b6e28991SAleksandr Rybalko 				 * Get number of domains and allocate one
742b6e28991SAleksandr Rybalko 				 * counter in each.
743b6e28991SAleksandr Rybalko 				 * First already allocated.
744b6e28991SAleksandr Rybalko 				 */
745b6e28991SAleksandr Rybalko 				for (i = 1; i < domains; i++) {
746b6e28991SAleksandr Rybalko 					CPU_ZERO(&dommask);
747b6e28991SAleksandr Rybalko 					cpuset_getaffinity(CPU_LEVEL_WHICH,
748b6e28991SAleksandr Rybalko 					    CPU_WHICH_DOMAIN, i, sizeof(dommask),
749b6e28991SAleksandr Rybalko 					    &dommask);
750b6e28991SAleksandr Rybalko 					CPU_SET(CPU_FFS(&dommask) - 1, &cpumask);
751b6e28991SAleksandr Rybalko 				}
752b6e28991SAleksandr Rybalko 				args.pa_flags |= FLAGS_HAS_CPUMASK;
753b6e28991SAleksandr Rybalko 			}
754ea056888SAttilio Rao 			if (option == 's' || option == 'S') {
755ea056888SAttilio Rao 				CPU_CLR(ev->ev_cpu, &cpumask);
756b6e28991SAleksandr Rybalko 				pmc_id_t saved_pmcid = ev->ev_pmcid;
757b6e28991SAleksandr Rybalko 				ev->ev_pmcid = PMC_ID_INVALID;
758d27927f7SRuslan Bukin 				pmcstat_clone_event_descriptor(ev, &cpumask, &args);
759b6e28991SAleksandr Rybalko 				ev->ev_pmcid = saved_pmcid;
760ea056888SAttilio Rao 				CPU_SET(ev->ev_cpu, &cpumask);
761ea056888SAttilio Rao 			}
762ee4910b1SJoseph Koshy 
763ebccf1e3SJoseph Koshy 			break;
764ebccf1e3SJoseph Koshy 
765ebccf1e3SJoseph Koshy 		case 'n':	/* sampling count */
766ebccf1e3SJoseph Koshy 			current_sampling_count = strtol(optarg, &end, 0);
767ebccf1e3SJoseph Koshy 			if (*end != '\0' || current_sampling_count <= 0)
768ebccf1e3SJoseph Koshy 				errx(EX_USAGE,
769f263522aSJoseph Koshy 				    "ERROR: Illegal count value \"%s\".",
770ebccf1e3SJoseph Koshy 				    optarg);
771f263522aSJoseph Koshy 			args.pa_required |= FLAG_HAS_SAMPLING_PMCS;
772ebccf1e3SJoseph Koshy 			break;
773ebccf1e3SJoseph Koshy 
774ebccf1e3SJoseph Koshy 		case 'o':	/* outputfile */
77566ce9936SFabien Thomas 			if (args.pa_printfile != NULL &&
77666ce9936SFabien Thomas 			    args.pa_printfile != stdout &&
77766ce9936SFabien Thomas 			    args.pa_printfile != stderr)
77815139246SJoseph Koshy 				(void) fclose(args.pa_printfile);
77915139246SJoseph Koshy 			if ((args.pa_printfile = fopen(optarg, "w")) == NULL)
78037d6f8a9SDavid E. O'Brien 				errx(EX_OSERR,
78137d6f8a9SDavid E. O'Brien 				    "ERROR: cannot open \"%s\" for writing.",
78237d6f8a9SDavid E. O'Brien 				    optarg);
78315139246SJoseph Koshy 			args.pa_flags |= FLAG_DO_PRINT;
784f263522aSJoseph Koshy 			break;
785ebccf1e3SJoseph Koshy 
786ebccf1e3SJoseph Koshy 		case 'O':	/* sampling output */
78715139246SJoseph Koshy 			if (args.pa_outputpath)
78837d6f8a9SDavid E. O'Brien 				errx(EX_USAGE,
78937d6f8a9SDavid E. O'Brien "ERROR: option -O may only be specified once.");
79015139246SJoseph Koshy 			args.pa_outputpath = optarg;
79115139246SJoseph Koshy 			args.pa_flags |= FLAG_HAS_OUTPUT_LOGFILE;
79215139246SJoseph Koshy 			break;
79315139246SJoseph Koshy 
79449874f6eSJoseph Koshy 		case 'q':	/* quiet mode */
79549874f6eSJoseph Koshy 			args.pa_verbosity = 0;
79649874f6eSJoseph Koshy 			break;
79749874f6eSJoseph Koshy 
79849874f6eSJoseph Koshy 		case 'r':	/* root FS path */
79949874f6eSJoseph Koshy 			args.pa_fsroot = optarg;
80049874f6eSJoseph Koshy 			break;
80149874f6eSJoseph Koshy 
80215139246SJoseph Koshy 		case 'R':	/* read an existing log file */
803b6010f9eSJoseph Koshy 			if (args.pa_inputpath != NULL)
80437d6f8a9SDavid E. O'Brien 				errx(EX_USAGE,
80537d6f8a9SDavid E. O'Brien "ERROR: option -R may only be specified once.");
80615139246SJoseph Koshy 			args.pa_inputpath = optarg;
80715139246SJoseph Koshy 			if (args.pa_printfile == stderr)
80815139246SJoseph Koshy 				args.pa_printfile = stdout;
80915139246SJoseph Koshy 			args.pa_flags |= FLAG_READ_LOGFILE;
810ebccf1e3SJoseph Koshy 			break;
811ebccf1e3SJoseph Koshy 
812fbd8b34aSJoseph Koshy 		case 't':	/* target pid or process name */
8130b86b1bbSFabien Thomas 			pmcstat_find_targets(optarg);
814ebccf1e3SJoseph Koshy 
815fbd8b34aSJoseph Koshy 			args.pa_flags |= FLAG_HAS_TARGET;
816f263522aSJoseph Koshy 			args.pa_required |= FLAG_HAS_PROCESS_PMCS;
817ebccf1e3SJoseph Koshy 			break;
818ebccf1e3SJoseph Koshy 
8190b86b1bbSFabien Thomas 		case 'T':	/* top mode */
8200b86b1bbSFabien Thomas 			args.pa_flags |= FLAG_DO_TOP;
8210b86b1bbSFabien Thomas 			args.pa_plugin = PMCSTAT_PL_CALLGRAPH;
8220b86b1bbSFabien Thomas 			args.pa_ctdumpinstr = 0;
8230b86b1bbSFabien Thomas 			args.pa_mergepmc = 1;
8240b86b1bbSFabien Thomas 			if (args.pa_printfile == stderr)
8250b86b1bbSFabien Thomas 				args.pa_printfile = stdout;
8260b86b1bbSFabien Thomas 			break;
8270b86b1bbSFabien Thomas 
828959826caSMatt Macy 		case 'u':
829959826caSMatt Macy 			do_descr = 1;
830959826caSMatt Macy 			event = optarg;
831959826caSMatt Macy 			break;
8325de96e33SMatt Macy 		case 'U':	/* toggle user-space callchain capture */
8335de96e33SMatt Macy 			do_userspace = !do_userspace;
8345de96e33SMatt Macy 			args.pa_required |= FLAG_HAS_SAMPLING_PMCS;
835959826caSMatt Macy 			break;
83649874f6eSJoseph Koshy 		case 'v':	/* verbose */
83749874f6eSJoseph Koshy 			args.pa_verbosity++;
83849874f6eSJoseph Koshy 			break;
83949874f6eSJoseph Koshy 
840ebccf1e3SJoseph Koshy 		case 'w':	/* wait interval */
841ebccf1e3SJoseph Koshy 			interval = strtod(optarg, &end);
842ebccf1e3SJoseph Koshy 			if (*end != '\0' || interval <= 0)
84337d6f8a9SDavid E. O'Brien 				errx(EX_USAGE,
84437d6f8a9SDavid E. O'Brien "ERROR: Illegal wait interval value \"%s\".",
84537d6f8a9SDavid E. O'Brien 				    optarg);
846ebccf1e3SJoseph Koshy 			args.pa_flags |= FLAG_HAS_WAIT_INTERVAL;
847ebccf1e3SJoseph Koshy 			args.pa_interval = interval;
848f263522aSJoseph Koshy 			break;
849ebccf1e3SJoseph Koshy 
850f263522aSJoseph Koshy 		case 'W':	/* toggle LOG_CSW */
851f263522aSJoseph Koshy 			do_logproccsw = !do_logproccsw;
852f263522aSJoseph Koshy 			args.pa_required |= (FLAG_HAS_PROCESS_PMCS |
85315139246SJoseph Koshy 			    FLAG_HAS_COUNTING_PMCS | FLAG_HAS_OUTPUT_LOGFILE);
854ebccf1e3SJoseph Koshy 			break;
855ebccf1e3SJoseph Koshy 
856b6010f9eSJoseph Koshy 		case 'z':
857b6010f9eSJoseph Koshy 			graphdepth = strtod(optarg, &end);
858b6010f9eSJoseph Koshy 			if (*end != '\0' || graphdepth <= 0)
85937d6f8a9SDavid E. O'Brien 				errx(EX_USAGE,
86037d6f8a9SDavid E. O'Brien 				    "ERROR: Illegal callchain depth \"%s\".",
86137d6f8a9SDavid E. O'Brien 				    optarg);
862b6010f9eSJoseph Koshy 			args.pa_graphdepth = graphdepth;
863b6010f9eSJoseph Koshy 			args.pa_required |= FLAG_DO_CALLGRAPHS;
864b6010f9eSJoseph Koshy 			break;
865b6010f9eSJoseph Koshy 
866ebccf1e3SJoseph Koshy 		case '?':
867ebccf1e3SJoseph Koshy 		default:
868ebccf1e3SJoseph Koshy 			pmcstat_show_usage();
869ebccf1e3SJoseph Koshy 			break;
870ebccf1e3SJoseph Koshy 
871ebccf1e3SJoseph Koshy 		}
8725de96e33SMatt Macy 	if ((do_listcounters | do_descr) &&
873959826caSMatt Macy 		pmc_pmu_enabled() == 0)
874959826caSMatt Macy 			errx(EX_USAGE, "pmu features not supported on host or hwpmc not loaded");
875959826caSMatt Macy 	if (do_listcounters) {
876fbf962e6SMatt Macy 		pmc_pmu_print_counters(NULL);
877959826caSMatt Macy 	} else if (do_descr) {
878959826caSMatt Macy 		pmc_pmu_print_counter_desc(event);
879959826caSMatt Macy 	}
8805de96e33SMatt Macy 	if (do_listcounters | do_descr)
881959826caSMatt Macy 		exit(0);
882ebccf1e3SJoseph Koshy 
883ebccf1e3SJoseph Koshy 	args.pa_argc = (argc -= optind);
884ebccf1e3SJoseph Koshy 	args.pa_argv = (argv += optind);
885ebccf1e3SJoseph Koshy 
886e7ef1de5SFabien Thomas 	/* If we read from logfile and no specified CPU mask use
887e7ef1de5SFabien Thomas 	 * the maximum CPU count.
888e7ef1de5SFabien Thomas 	 */
889e7ef1de5SFabien Thomas 	if ((args.pa_flags & FLAG_READ_LOGFILE) &&
890e7ef1de5SFabien Thomas 	    (args.pa_flags & FLAGS_HAS_CPUMASK) == 0)
891ea056888SAttilio Rao 		CPU_FILL(&cpumask);
892e7ef1de5SFabien Thomas 
893b6010f9eSJoseph Koshy 	args.pa_cpumask = cpumask; /* For selecting CPUs using -R. */
894b6010f9eSJoseph Koshy 
89515139246SJoseph Koshy 	if (argc)	/* command line present */
89615139246SJoseph Koshy 		args.pa_flags |= FLAG_HAS_COMMANDLINE;
897ebccf1e3SJoseph Koshy 
898c0252222SAttilio Rao 	if (args.pa_flags & (FLAG_DO_GPROF | FLAG_DO_CALLGRAPHS |
8990b86b1bbSFabien Thomas 	    FLAG_DO_ANNOTATE | FLAG_DO_TOP))
900b6010f9eSJoseph Koshy 		args.pa_flags |= FLAG_DO_ANALYSIS;
901b6010f9eSJoseph Koshy 
902ebccf1e3SJoseph Koshy 	/*
903ebccf1e3SJoseph Koshy 	 * Check invocation syntax.
904ebccf1e3SJoseph Koshy 	 */
905ebccf1e3SJoseph Koshy 
90615139246SJoseph Koshy 	/* disallow -O and -R together */
90715139246SJoseph Koshy 	if (args.pa_outputpath && args.pa_inputpath)
90837d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
90937d6f8a9SDavid E. O'Brien 		    "ERROR: options -O and -R are mutually exclusive.");
91015139246SJoseph Koshy 
91172fe532cSGeorge V. Neville-Neil 	/* disallow -T and -l together */
91272fe532cSGeorge V. Neville-Neil 	if ((args.pa_flags & FLAG_HAS_DURATION) &&
91372fe532cSGeorge V. Neville-Neil 	    (args.pa_flags & FLAG_DO_TOP))
91472fe532cSGeorge V. Neville-Neil 		errx(EX_USAGE, "ERROR: options -T and -l are mutually "
91572fe532cSGeorge V. Neville-Neil 		    "exclusive.");
91672fe532cSGeorge V. Neville-Neil 
91710813ab3SEric van Gyzen 	/* -a and -m require -R */
9180b86b1bbSFabien Thomas 	if (args.pa_flags & FLAG_DO_ANNOTATE && args.pa_inputpath == NULL)
91985ec49f3SAdrian Chadd 		errx(EX_USAGE, "ERROR: option %s requires an input file",
92085ec49f3SAdrian Chadd 		    args.pa_plugin == PMCSTAT_PL_ANNOTATE ? "-m" : "-a");
921c0252222SAttilio Rao 
922c0252222SAttilio Rao 	/* -m option is not allowed combined with -g or -G. */
9230b86b1bbSFabien Thomas 	if (args.pa_flags & FLAG_DO_ANNOTATE &&
924c0252222SAttilio Rao 	    args.pa_flags & (FLAG_DO_GPROF | FLAG_DO_CALLGRAPHS))
92537d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
92637d6f8a9SDavid E. O'Brien 		    "ERROR: option -m and -g | -G are mutually exclusive");
927c0252222SAttilio Rao 
92815139246SJoseph Koshy 	if (args.pa_flags & FLAG_READ_LOGFILE) {
929f263522aSJoseph Koshy 		errmsg = NULL;
93015139246SJoseph Koshy 		if (args.pa_flags & FLAG_HAS_COMMANDLINE)
931f263522aSJoseph Koshy 			errmsg = "a command line specification";
932fbd8b34aSJoseph Koshy 		else if (args.pa_flags & FLAG_HAS_TARGET)
933f263522aSJoseph Koshy 			errmsg = "option -t";
934fbd8b34aSJoseph Koshy 		else if (!STAILQ_EMPTY(&args.pa_events))
935f263522aSJoseph Koshy 			errmsg = "a PMC event specification";
936f263522aSJoseph Koshy 		if (errmsg)
93737d6f8a9SDavid E. O'Brien 			errx(EX_USAGE,
93837d6f8a9SDavid E. O'Brien 			    "ERROR: option -R may not be used with %s.",
93937d6f8a9SDavid E. O'Brien 			    errmsg);
940fbd8b34aSJoseph Koshy 	} else if (STAILQ_EMPTY(&args.pa_events))
94115139246SJoseph Koshy 		/* All other uses require a PMC spec. */
942ebccf1e3SJoseph Koshy 		pmcstat_show_usage();
943ebccf1e3SJoseph Koshy 
944f263522aSJoseph Koshy 	/* check for -t pid without a process PMC spec */
945c4473addSMitchell Horne 	if ((args.pa_flags & FLAG_HAS_TARGET) &&
946c4473addSMitchell Horne 	    (args.pa_required & FLAG_HAS_PROCESS_PMCS) &&
947f263522aSJoseph Koshy 	    (args.pa_flags & FLAG_HAS_PROCESS_PMCS) == 0)
94837d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
94937d6f8a9SDavid E. O'Brien "ERROR: option -t requires a process mode PMC to be specified."
95037d6f8a9SDavid E. O'Brien 		    );
951f263522aSJoseph Koshy 
952f263522aSJoseph Koshy 	/* check for process-mode options without a command or -t pid */
953f263522aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_PROCESS_PMCS) &&
954fbd8b34aSJoseph Koshy 	    (args.pa_flags & (FLAG_HAS_COMMANDLINE | FLAG_HAS_TARGET)) == 0)
95537d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
95637d6f8a9SDavid E. O'Brien "ERROR: options -d, -E, -p, -P, and -W require a command line or target process."
95737d6f8a9SDavid E. O'Brien 		    );
958f263522aSJoseph Koshy 
959f263522aSJoseph Koshy 	/* check for -p | -P without a target process of some sort */
960fbd8b34aSJoseph Koshy 	if ((args.pa_required & (FLAG_HAS_COMMANDLINE | FLAG_HAS_TARGET)) &&
961fbd8b34aSJoseph Koshy 	    (args.pa_flags & (FLAG_HAS_COMMANDLINE | FLAG_HAS_TARGET)) == 0)
96237d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
96337d6f8a9SDavid E. O'Brien "ERROR: options -P and -p require a target process or a command line."
96437d6f8a9SDavid E. O'Brien 		    );
965f263522aSJoseph Koshy 
966f263522aSJoseph Koshy 	/* check for process-mode options without a process-mode PMC */
967f263522aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_PROCESS_PMCS) &&
968f263522aSJoseph Koshy 	    (args.pa_flags & FLAG_HAS_PROCESS_PMCS) == 0)
96937d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
970bf661593SMitchell Horne "ERROR: options -d, -E, -t, and -W require a process mode PMC to be specified."
97137d6f8a9SDavid E. O'Brien 		    );
972f263522aSJoseph Koshy 
973b6010f9eSJoseph Koshy 	/* check for -c cpu with no system mode PMCs or logfile. */
974f263522aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_SYSTEM_PMCS) &&
975b6010f9eSJoseph Koshy 	    (args.pa_flags & FLAG_HAS_SYSTEM_PMCS) == 0 &&
976b6010f9eSJoseph Koshy 	    (args.pa_flags & FLAG_READ_LOGFILE) == 0)
97737d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
97837d6f8a9SDavid E. O'Brien "ERROR: option -c requires at least one system mode PMC to be specified."
97937d6f8a9SDavid E. O'Brien 		    );
980f263522aSJoseph Koshy 
981f263522aSJoseph Koshy 	/* check for counting mode options without a counting PMC */
982f263522aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_COUNTING_PMCS) &&
983f263522aSJoseph Koshy 	    (args.pa_flags & FLAG_HAS_COUNTING_PMCS) == 0)
98437d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
98537d6f8a9SDavid E. O'Brien "ERROR: options -C, -W and -o require at least one counting mode PMC to be specified."
98637d6f8a9SDavid E. O'Brien 		    );
987f263522aSJoseph Koshy 
988f263522aSJoseph Koshy 	/* check for sampling mode options without a sampling PMC spec */
989f263522aSJoseph Koshy 	if ((args.pa_required & FLAG_HAS_SAMPLING_PMCS) &&
990f263522aSJoseph Koshy 	    (args.pa_flags & FLAG_HAS_SAMPLING_PMCS) == 0)
99137d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
99237d6f8a9SDavid E. O'Brien "ERROR: options -N, -n and -O require at least one sampling mode PMC to be specified."
99337d6f8a9SDavid E. O'Brien 		    );
994f263522aSJoseph Koshy 
9950b86b1bbSFabien Thomas 	/* check if -g/-G/-m/-T are being used correctly */
996b6010f9eSJoseph Koshy 	if ((args.pa_flags & FLAG_DO_ANALYSIS) &&
99715139246SJoseph Koshy 	    !(args.pa_flags & (FLAG_HAS_SAMPLING_PMCS|FLAG_READ_LOGFILE)))
99837d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
99937d6f8a9SDavid E. O'Brien "ERROR: options -g/-G/-m/-T require sampling PMCs or -R to be specified."
100037d6f8a9SDavid E. O'Brien 		    );
100115139246SJoseph Koshy 
1002465dadb5SJonathan T. Looney 	/* check if -e was specified without -g */
1003465dadb5SJonathan T. Looney 	if ((args.pa_flags & FLAG_DO_WIDE_GPROF_HC) &&
1004465dadb5SJonathan T. Looney 	    !(args.pa_flags & FLAG_DO_GPROF))
1005465dadb5SJonathan T. Looney 		errx(EX_USAGE,
1006465dadb5SJonathan T. Looney "ERROR: option -e requires gprof mode to be specified."
1007465dadb5SJonathan T. Looney 		    );
1008465dadb5SJonathan T. Looney 
1009f263522aSJoseph Koshy 	/* check if -O was spuriously specified */
101015139246SJoseph Koshy 	if ((args.pa_flags & FLAG_HAS_OUTPUT_LOGFILE) &&
101115139246SJoseph Koshy 	    (args.pa_required & FLAG_HAS_OUTPUT_LOGFILE) == 0)
1012f263522aSJoseph Koshy 		errx(EX_USAGE,
101337d6f8a9SDavid E. O'Brien "ERROR: option -O is used only with options -E, -P, -S and -W."
101437d6f8a9SDavid E. O'Brien 		    );
1015f263522aSJoseph Koshy 
1016b6010f9eSJoseph Koshy 	/* -D only applies to gprof output mode (-g) */
101715139246SJoseph Koshy 	if ((args.pa_flags & FLAG_HAS_SAMPLESDIR) &&
1018b6010f9eSJoseph Koshy 	    (args.pa_flags & FLAG_DO_GPROF) == 0)
1019b6010f9eSJoseph Koshy 	    errx(EX_USAGE, "ERROR: option -D is only used with -g.");
102049874f6eSJoseph Koshy 
102149874f6eSJoseph Koshy 	/* -M mapfile requires -g or -R */
102249874f6eSJoseph Koshy 	if (args.pa_mapfilename != NULL &&
102349874f6eSJoseph Koshy 	    (args.pa_flags & FLAG_DO_GPROF) == 0 &&
102449874f6eSJoseph Koshy 	    (args.pa_flags & FLAG_READ_LOGFILE) == 0)
102549874f6eSJoseph Koshy 	    errx(EX_USAGE, "ERROR: option -M is only used with -g/-R.");
102615139246SJoseph Koshy 
102715139246SJoseph Koshy 	/*
102815139246SJoseph Koshy 	 * Disallow textual output of sampling PMCs if counting PMCs
102915139246SJoseph Koshy 	 * have also been asked for, mostly because the combined output
103015139246SJoseph Koshy 	 * is difficult to make sense of.
103115139246SJoseph Koshy 	 */
103215139246SJoseph Koshy 	if ((args.pa_flags & FLAG_HAS_COUNTING_PMCS) &&
103315139246SJoseph Koshy 	    (args.pa_flags & FLAG_HAS_SAMPLING_PMCS) &&
103400a22dc5SJoseph Koshy 	    ((args.pa_flags & FLAG_HAS_OUTPUT_LOGFILE) == 0))
103537d6f8a9SDavid E. O'Brien 		errx(EX_USAGE,
103637d6f8a9SDavid E. O'Brien "ERROR: option -O is required if counting and sampling PMCs are specified together."
103737d6f8a9SDavid E. O'Brien 		    );
103815139246SJoseph Koshy 
103949874f6eSJoseph Koshy 	/*
1040b6010f9eSJoseph Koshy 	 * If we have a callgraph be created, select the outputfile.
1041b6010f9eSJoseph Koshy 	 */
1042b6010f9eSJoseph Koshy 	if (args.pa_flags & FLAG_DO_CALLGRAPHS) {
1043b6010f9eSJoseph Koshy 		if (strcmp(graphfilename, "-") == 0)
1044b6010f9eSJoseph Koshy 		    args.pa_graphfile = args.pa_printfile;
1045b6010f9eSJoseph Koshy 		else {
1046b6010f9eSJoseph Koshy 			args.pa_graphfile = fopen(graphfilename, "w");
1047b6010f9eSJoseph Koshy 			if (args.pa_graphfile == NULL)
104837d6f8a9SDavid E. O'Brien 				err(EX_OSERR,
104937d6f8a9SDavid E. O'Brien 				    "ERROR: cannot open \"%s\" for writing",
105037d6f8a9SDavid E. O'Brien 				    graphfilename);
1051b6010f9eSJoseph Koshy 		}
1052b6010f9eSJoseph Koshy 	}
10530b86b1bbSFabien Thomas 	if (args.pa_flags & FLAG_DO_ANNOTATE) {
1054c0252222SAttilio Rao 		args.pa_graphfile = fopen(graphfilename, "w");
1055c0252222SAttilio Rao 		if (args.pa_graphfile == NULL)
1056c0252222SAttilio Rao 			err(EX_OSERR, "ERROR: cannot open \"%s\" for writing",
1057c0252222SAttilio Rao 			    graphfilename);
1058c0252222SAttilio Rao 	}
1059b6010f9eSJoseph Koshy 
1060ac708fddSFabien Thomas 	/* if we've been asked to process a log file, skip init */
1061b6e28991SAleksandr Rybalko 	if ((args.pa_flags & FLAG_READ_LOGFILE) == 0)
1062b6e28991SAleksandr Rybalko 		libpmc_initialize(&npmc);
1063ac708fddSFabien Thomas 
1064ac708fddSFabien Thomas 	/* Allocate a kqueue */
1065ac708fddSFabien Thomas 	if ((pmcstat_kq = kqueue()) < 0)
1066ac708fddSFabien Thomas 		err(EX_OSERR, "ERROR: Cannot allocate kqueue");
1067ac708fddSFabien Thomas 
1068ac708fddSFabien Thomas 	/* Setup the logfile as the source. */
106915139246SJoseph Koshy 	if (args.pa_flags & FLAG_READ_LOGFILE) {
107015139246SJoseph Koshy 		/*
107115139246SJoseph Koshy 		 * Print the log in textual form if we haven't been
1072b6010f9eSJoseph Koshy 		 * asked to generate profiling information.
107315139246SJoseph Koshy 		 */
1074b6010f9eSJoseph Koshy 		if ((args.pa_flags & FLAG_DO_ANALYSIS) == 0)
107515139246SJoseph Koshy 			args.pa_flags |= FLAG_DO_PRINT;
107615139246SJoseph Koshy 
1077d27927f7SRuslan Bukin 		pmcstat_log_initialize_logging();
1078ac708fddSFabien Thomas 		rfd = pmcstat_open_log(args.pa_inputpath,
1079302cbb90SJoseph Koshy 		    PMCSTAT_OPEN_FOR_READ);
1080ac708fddSFabien Thomas 		if ((args.pa_logparser = pmclog_open(rfd)) == NULL)
108115139246SJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot create parser");
1082ac708fddSFabien Thomas 		if (fcntl(rfd, F_SETFL, O_NONBLOCK) < 0)
1083ac708fddSFabien Thomas 			err(EX_OSERR, "ERROR: fcntl(2) failed");
1084ac708fddSFabien Thomas 		EV_SET(&kev, rfd, EVFILT_READ, EV_ADD,
1085ac708fddSFabien Thomas 		    0, 0, NULL);
1086ac708fddSFabien Thomas 		if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
1087ac708fddSFabien Thomas 			err(EX_OSERR, "ERROR: Cannot register kevent");
1088f263522aSJoseph Koshy 	}
108915139246SJoseph Koshy 	/*
109015139246SJoseph Koshy 	 * Configure the specified log file or setup a default log
109115139246SJoseph Koshy 	 * consumer via a pipe.
109215139246SJoseph Koshy 	 */
109315139246SJoseph Koshy 	if (args.pa_required & FLAG_HAS_OUTPUT_LOGFILE) {
1094302cbb90SJoseph Koshy 		if (args.pa_outputpath)
1095302cbb90SJoseph Koshy 			args.pa_logfd = pmcstat_open_log(args.pa_outputpath,
1096302cbb90SJoseph Koshy 			    PMCSTAT_OPEN_FOR_WRITE);
1097302cbb90SJoseph Koshy 		else {
109815139246SJoseph Koshy 			/*
109915139246SJoseph Koshy 			 * process the log on the fly by reading it in
110015139246SJoseph Koshy 			 * through a pipe.
110115139246SJoseph Koshy 			 */
110215139246SJoseph Koshy 			if (pipe(pipefd) < 0)
110315139246SJoseph Koshy 				err(EX_OSERR, "ERROR: pipe(2) failed");
110415139246SJoseph Koshy 
110515139246SJoseph Koshy 			if (fcntl(pipefd[READPIPEFD], F_SETFL, O_NONBLOCK) < 0)
110615139246SJoseph Koshy 				err(EX_OSERR, "ERROR: fcntl(2) failed");
110715139246SJoseph Koshy 
110815139246SJoseph Koshy 			EV_SET(&kev, pipefd[READPIPEFD], EVFILT_READ, EV_ADD,
110915139246SJoseph Koshy 			    0, 0, NULL);
111015139246SJoseph Koshy 
111115139246SJoseph Koshy 			if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
111215139246SJoseph Koshy 				err(EX_OSERR, "ERROR: Cannot register kevent");
111315139246SJoseph Koshy 
111415139246SJoseph Koshy 			args.pa_logfd = pipefd[WRITEPIPEFD];
111515139246SJoseph Koshy 
11160b86b1bbSFabien Thomas 			args.pa_flags |= FLAG_HAS_PIPE;
11170b86b1bbSFabien Thomas 			if ((args.pa_flags & FLAG_DO_TOP) == 0)
11180b86b1bbSFabien Thomas 				args.pa_flags |= FLAG_DO_PRINT;
111915139246SJoseph Koshy 			args.pa_logparser = pmclog_open(pipefd[READPIPEFD]);
112015139246SJoseph Koshy 		}
112115139246SJoseph Koshy 
112215139246SJoseph Koshy 		if (pmc_configure_logfile(args.pa_logfd) < 0)
112315139246SJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot configure log file");
112415139246SJoseph Koshy 	}
112515139246SJoseph Koshy 
11267fbf3da2SJoseph Koshy 	/* remember to check for driver errors if we are sampling or logging */
11277fbf3da2SJoseph Koshy 	check_driver_stats = (args.pa_flags & FLAG_HAS_SAMPLING_PMCS) ||
11287fbf3da2SJoseph Koshy 	    (args.pa_flags & FLAG_HAS_OUTPUT_LOGFILE);
11297fbf3da2SJoseph Koshy 
1130ebccf1e3SJoseph Koshy 	/*
1131ac708fddSFabien Thomas 	if (args.pa_flags & FLAG_READ_LOGFILE) {
1132ebccf1e3SJoseph Koshy 	 * Allocate PMCs.
1133ebccf1e3SJoseph Koshy 	 */
1134ebccf1e3SJoseph Koshy 
1135fbd8b34aSJoseph Koshy 	STAILQ_FOREACH(ev, &args.pa_events, ev_next) {
11360aa15077SAndrew Gallatin 		if (pmc_allocate(ev->ev_spec, ev->ev_mode,
1137b2ca2e50SMatt Macy 			ev->ev_flags, ev->ev_cpu, &ev->ev_pmcid,
1138b2ca2e50SMatt Macy 			ev->ev_count) < 0)
113937d6f8a9SDavid E. O'Brien 			err(EX_OSERR,
114037d6f8a9SDavid E. O'Brien "ERROR: Cannot allocate %s-mode pmc with specification \"%s\"",
114137d6f8a9SDavid E. O'Brien 			    PMC_IS_SYSTEM_MODE(ev->ev_mode) ?
114237d6f8a9SDavid E. O'Brien 			    "system" : "process", ev->ev_spec);
1143ebccf1e3SJoseph Koshy 
114415139246SJoseph Koshy 		if (PMC_IS_SAMPLING_MODE(ev->ev_mode) &&
114515139246SJoseph Koshy 		    pmc_set(ev->ev_pmcid, ev->ev_count) < 0)
114637d6f8a9SDavid E. O'Brien 			err(EX_OSERR,
114737d6f8a9SDavid E. O'Brien 			    "ERROR: Cannot set sampling count for PMC \"%s\"",
114837d6f8a9SDavid E. O'Brien 			    ev->ev_name);
114915139246SJoseph Koshy 	}
115015139246SJoseph Koshy 
1151ebccf1e3SJoseph Koshy 	/* compute printout widths */
1152fbd8b34aSJoseph Koshy 	STAILQ_FOREACH(ev, &args.pa_events, ev_next) {
1153c5153e19SJoseph Koshy 		int counter_width;
1154c5153e19SJoseph Koshy 		int display_width;
1155c5153e19SJoseph Koshy 		int header_width;
1156ebccf1e3SJoseph Koshy 
1157c5153e19SJoseph Koshy 		(void) pmc_width(ev->ev_pmcid, &counter_width);
1158ee4910b1SJoseph Koshy 		header_width = strlen(ev->ev_name) + 2; /* prefix '%c/' */
1159c5153e19SJoseph Koshy 		display_width = (int) floor(counter_width / 3.32193) + 1;
1160ebccf1e3SJoseph Koshy 
1161ee4910b1SJoseph Koshy 		if (PMC_IS_SYSTEM_MODE(ev->ev_mode))
1162ee4910b1SJoseph Koshy 			header_width += 3; /* 2 digit CPU number + '/' */
1163ee4910b1SJoseph Koshy 
1164c5153e19SJoseph Koshy 		if (header_width > display_width) {
1165ebccf1e3SJoseph Koshy 			ev->ev_fieldskip = 0;
1166c5153e19SJoseph Koshy 			ev->ev_fieldwidth = header_width;
1167ebccf1e3SJoseph Koshy 		} else {
1168c5153e19SJoseph Koshy 			ev->ev_fieldskip = display_width -
1169c5153e19SJoseph Koshy 			    header_width;
1170c5153e19SJoseph Koshy 			ev->ev_fieldwidth = display_width;
1171ebccf1e3SJoseph Koshy 		}
1172ebccf1e3SJoseph Koshy 	}
1173ebccf1e3SJoseph Koshy 
1174ebccf1e3SJoseph Koshy 	/*
1175ebccf1e3SJoseph Koshy 	 * If our output is being set to a terminal, register a handler
1176ebccf1e3SJoseph Koshy 	 * for window size changes.
1177ebccf1e3SJoseph Koshy 	 */
1178ebccf1e3SJoseph Koshy 
117915139246SJoseph Koshy 	if (isatty(fileno(args.pa_printfile))) {
1180ebccf1e3SJoseph Koshy 
118115139246SJoseph Koshy 		if (ioctl(fileno(args.pa_printfile), TIOCGWINSZ, &ws) < 0)
1182ebccf1e3SJoseph Koshy 			err(EX_OSERR, "ERROR: Cannot determine window size");
1183ebccf1e3SJoseph Koshy 
1184ebccf1e3SJoseph Koshy 		pmcstat_displayheight = ws.ws_row - 1;
11850b86b1bbSFabien Thomas 		pmcstat_displaywidth  = ws.ws_col - 1;
1186ebccf1e3SJoseph Koshy 
1187ebccf1e3SJoseph Koshy 		EV_SET(&kev, SIGWINCH, EVFILT_SIGNAL, EV_ADD, 0, 0, NULL);
1188ebccf1e3SJoseph Koshy 
1189ebccf1e3SJoseph Koshy 		if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
119037d6f8a9SDavid E. O'Brien 			err(EX_OSERR,
119137d6f8a9SDavid E. O'Brien 			    "ERROR: Cannot register kevent for SIGWINCH");
11920b86b1bbSFabien Thomas 
11930b86b1bbSFabien Thomas 		args.pa_toptty = 1;
11940b86b1bbSFabien Thomas 	}
11950b86b1bbSFabien Thomas 
11960b86b1bbSFabien Thomas 	/*
11970b86b1bbSFabien Thomas 	 * Listen to key input in top mode.
11980b86b1bbSFabien Thomas 	 */
11990b86b1bbSFabien Thomas 	if (args.pa_flags & FLAG_DO_TOP) {
12000b86b1bbSFabien Thomas 		EV_SET(&kev, fileno(stdin), EVFILT_READ, EV_ADD, 0, 0, NULL);
12010b86b1bbSFabien Thomas 		if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
12020b86b1bbSFabien Thomas 			err(EX_OSERR, "ERROR: Cannot register kevent");
1203ebccf1e3SJoseph Koshy 	}
1204ebccf1e3SJoseph Koshy 
1205ebccf1e3SJoseph Koshy 	EV_SET(&kev, SIGINT, EVFILT_SIGNAL, EV_ADD, 0, 0, NULL);
1206ebccf1e3SJoseph Koshy 	if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
1207ebccf1e3SJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot register kevent for SIGINT");
1208ebccf1e3SJoseph Koshy 
1209f263522aSJoseph Koshy 	EV_SET(&kev, SIGIO, EVFILT_SIGNAL, EV_ADD, 0, 0, NULL);
1210f263522aSJoseph Koshy 	if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
1211f263522aSJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot register kevent for SIGIO");
1212ebccf1e3SJoseph Koshy 
1213ebccf1e3SJoseph Koshy 	/*
1214f263522aSJoseph Koshy 	 * An exec() failure of a forked child is signalled by the
1215f263522aSJoseph Koshy 	 * child sending the parent a SIGCHLD.  We don't register an
1216f263522aSJoseph Koshy 	 * actual signal handler for SIGCHLD, but instead use our
1217f263522aSJoseph Koshy 	 * kqueue to pick up the signal.
1218ebccf1e3SJoseph Koshy 	 */
1219f263522aSJoseph Koshy 	EV_SET(&kev, SIGCHLD, EVFILT_SIGNAL, EV_ADD, 0, 0, NULL);
1220f263522aSJoseph Koshy 	if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
1221f263522aSJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot register kevent for SIGCHLD");
1222ebccf1e3SJoseph Koshy 
12230b86b1bbSFabien Thomas 	/*
12240b86b1bbSFabien Thomas 	 * Setup a timer if we have counting mode PMCs needing to be printed or
12250b86b1bbSFabien Thomas 	 * top mode plugin is active.
12260b86b1bbSFabien Thomas 	 */
12270b86b1bbSFabien Thomas 	if (((args.pa_flags & FLAG_HAS_COUNTING_PMCS) &&
12280b86b1bbSFabien Thomas 	     (args.pa_required & FLAG_HAS_OUTPUT_LOGFILE) == 0) ||
12290b86b1bbSFabien Thomas 	    (args.pa_flags & FLAG_DO_TOP)) {
1230ebccf1e3SJoseph Koshy 		EV_SET(&kev, 0, EVFILT_TIMER, EV_ADD, 0,
1231ebccf1e3SJoseph Koshy 		    args.pa_interval * 1000, NULL);
1232ebccf1e3SJoseph Koshy 
1233ebccf1e3SJoseph Koshy 		if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
123437d6f8a9SDavid E. O'Brien 			err(EX_OSERR,
123537d6f8a9SDavid E. O'Brien 			    "ERROR: Cannot register kevent for timer");
1236ebccf1e3SJoseph Koshy 	}
1237ebccf1e3SJoseph Koshy 
123872fe532cSGeorge V. Neville-Neil 	/*
123972fe532cSGeorge V. Neville-Neil 	 * Setup a duration timer if we have sampling mode PMCs and
124072fe532cSGeorge V. Neville-Neil 	 * a duration time is set
124172fe532cSGeorge V. Neville-Neil 	 */
124272fe532cSGeorge V. Neville-Neil 	if ((args.pa_flags & FLAG_HAS_SAMPLING_PMCS) &&
124372fe532cSGeorge V. Neville-Neil 	    (args.pa_flags & FLAG_HAS_DURATION)) {
124472fe532cSGeorge V. Neville-Neil 		EV_SET(&kev, 0, EVFILT_TIMER, EV_ADD, 0,
124572fe532cSGeorge V. Neville-Neil 		    args.pa_duration * 1000, NULL);
124672fe532cSGeorge V. Neville-Neil 
124772fe532cSGeorge V. Neville-Neil 		if (kevent(pmcstat_kq, &kev, 1, NULL, 0, NULL) < 0)
124872fe532cSGeorge V. Neville-Neil 			err(EX_OSERR, "ERROR: Cannot register kevent for "
124972fe532cSGeorge V. Neville-Neil 			    "time duration");
125072fe532cSGeorge V. Neville-Neil 	}
125172fe532cSGeorge V. Neville-Neil 
1252ebccf1e3SJoseph Koshy 	/* attach PMCs to the target process, starting it if specified */
1253ee4910b1SJoseph Koshy 	if (args.pa_flags & FLAG_HAS_COMMANDLINE)
1254d27927f7SRuslan Bukin 		pmcstat_create_process(pmcstat_sockpair, &args, pmcstat_kq);
1255ebccf1e3SJoseph Koshy 
12567fbf3da2SJoseph Koshy 	if (check_driver_stats && pmc_get_driver_stats(&ds_start) < 0)
12577fbf3da2SJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot retrieve driver statistics");
12587fbf3da2SJoseph Koshy 
1259ee4910b1SJoseph Koshy 	/* Attach process pmcs to the target process. */
1260165b10b0SJoseph Koshy 	if (args.pa_flags & (FLAG_HAS_TARGET | FLAG_HAS_COMMANDLINE)) {
1261fbd8b34aSJoseph Koshy 		if (SLIST_EMPTY(&args.pa_targets))
126237d6f8a9SDavid E. O'Brien 			errx(EX_DATAERR,
126337d6f8a9SDavid E. O'Brien 			    "ERROR: No matching target processes.");
12645477328dSJoseph Koshy 		if (args.pa_flags & FLAG_HAS_PROCESS_PMCS)
1265d27927f7SRuslan Bukin 			pmcstat_attach_pmcs(&args);
1266ee4910b1SJoseph Koshy 
1267fbd8b34aSJoseph Koshy 		if (pmcstat_kvm) {
1268fbd8b34aSJoseph Koshy 			kvm_close(pmcstat_kvm);
1269fbd8b34aSJoseph Koshy 			pmcstat_kvm = NULL;
1270fbd8b34aSJoseph Koshy 		}
1271fbd8b34aSJoseph Koshy 	}
1272fbd8b34aSJoseph Koshy 
1273ebccf1e3SJoseph Koshy 	/* start the pmcs */
12740b86b1bbSFabien Thomas 	pmcstat_start_pmcs();
1275ebccf1e3SJoseph Koshy 
1276ebccf1e3SJoseph Koshy 	/* start the (commandline) process if needed */
127715139246SJoseph Koshy 	if (args.pa_flags & FLAG_HAS_COMMANDLINE)
1278d27927f7SRuslan Bukin 		pmcstat_start_process(pmcstat_sockpair);
1279ebccf1e3SJoseph Koshy 
1280ac708fddSFabien Thomas 	/* initialize logging */
1281d27927f7SRuslan Bukin 	pmcstat_log_initialize_logging();
128215139246SJoseph Koshy 
1283ebccf1e3SJoseph Koshy 	/* Handle SIGINT using the kqueue loop */
1284ebccf1e3SJoseph Koshy 	sa.sa_handler = SIG_IGN;
1285ebccf1e3SJoseph Koshy 	sa.sa_flags   = 0;
1286ebccf1e3SJoseph Koshy 	(void) sigemptyset(&sa.sa_mask);
1287ebccf1e3SJoseph Koshy 
1288ebccf1e3SJoseph Koshy 	if (sigaction(SIGINT, &sa, NULL) < 0)
1289ebccf1e3SJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot install signal handler");
1290ebccf1e3SJoseph Koshy 
1291ebccf1e3SJoseph Koshy 	/*
12920b86b1bbSFabien Thomas 	 * Setup the top mode display.
12930b86b1bbSFabien Thomas 	 */
12940b86b1bbSFabien Thomas 	if (args.pa_flags & FLAG_DO_TOP) {
12950b86b1bbSFabien Thomas 		args.pa_flags &= ~FLAG_DO_PRINT;
12960b86b1bbSFabien Thomas 
12970b86b1bbSFabien Thomas 		if (args.pa_toptty) {
12980b86b1bbSFabien Thomas 			/*
12990b86b1bbSFabien Thomas 			 * Init ncurses.
13000b86b1bbSFabien Thomas 			 */
13010b86b1bbSFabien Thomas 			initscr();
13020b86b1bbSFabien Thomas 			if(has_colors() == TRUE) {
13030b86b1bbSFabien Thomas 				args.pa_topcolor = 1;
13040b86b1bbSFabien Thomas 				start_color();
13050b86b1bbSFabien Thomas 				use_default_colors();
13060b86b1bbSFabien Thomas 				pair_content(0, &cf, &cb);
13070b86b1bbSFabien Thomas 				init_pair(1, COLOR_RED, cb);
13080b86b1bbSFabien Thomas 				init_pair(2, COLOR_YELLOW, cb);
13090b86b1bbSFabien Thomas 				init_pair(3, COLOR_GREEN, cb);
13100b86b1bbSFabien Thomas 			}
13110b86b1bbSFabien Thomas 			cbreak();
13120b86b1bbSFabien Thomas 			noecho();
13130b86b1bbSFabien Thomas 			nonl();
13140b86b1bbSFabien Thomas 			nodelay(stdscr, 1);
13150b86b1bbSFabien Thomas 			intrflush(stdscr, FALSE);
13160b86b1bbSFabien Thomas 			keypad(stdscr, TRUE);
13170b86b1bbSFabien Thomas 			clear();
13182aa983e5SFabien Thomas 			/* Get terminal width / height with ncurses. */
131937d6f8a9SDavid E. O'Brien 			getmaxyx(stdscr,
132037d6f8a9SDavid E. O'Brien 			    pmcstat_displayheight, pmcstat_displaywidth);
13212aa983e5SFabien Thomas 			pmcstat_displayheight--; pmcstat_displaywidth--;
13220b86b1bbSFabien Thomas 			atexit(pmcstat_topexit);
13230b86b1bbSFabien Thomas 		}
13240b86b1bbSFabien Thomas 	}
13250b86b1bbSFabien Thomas 
13260b86b1bbSFabien Thomas 	/*
1327ebccf1e3SJoseph Koshy 	 * loop till either the target process (if any) exits, or we
132872fe532cSGeorge V. Neville-Neil 	 * are killed by a SIGINT or we reached the time duration.
1329ebccf1e3SJoseph Koshy 	 */
1330f263522aSJoseph Koshy 	runstate = PMCSTAT_RUNNING;
1331dceed24aSFabien Thomas 	do_print = do_read = 0;
1332ebccf1e3SJoseph Koshy 	do {
1333ebccf1e3SJoseph Koshy 		if ((c = kevent(pmcstat_kq, NULL, 0, &kev, 1, NULL)) <= 0) {
1334ebccf1e3SJoseph Koshy 			if (errno != EINTR)
1335ebccf1e3SJoseph Koshy 				err(EX_OSERR, "ERROR: kevent failed");
1336ebccf1e3SJoseph Koshy 			else
1337ebccf1e3SJoseph Koshy 				continue;
1338ebccf1e3SJoseph Koshy 		}
1339ebccf1e3SJoseph Koshy 
1340ebccf1e3SJoseph Koshy 		if (kev.flags & EV_ERROR)
1341ebccf1e3SJoseph Koshy 			errc(EX_OSERR, kev.data, "ERROR: kevent failed");
1342ebccf1e3SJoseph Koshy 
1343ebccf1e3SJoseph Koshy 		switch (kev.filter) {
1344f263522aSJoseph Koshy 		case EVFILT_PROC:  /* target has exited */
1345d27927f7SRuslan Bukin 			runstate = pmcstat_close_log(&args);
134638fc5f34SJoseph Koshy 			do_print = 1;
1347f263522aSJoseph Koshy 			break;
1348ebccf1e3SJoseph Koshy 
1349f263522aSJoseph Koshy 		case EVFILT_READ:  /* log file data is present */
1350ac708fddSFabien Thomas 			if (kev.ident == (unsigned)fileno(stdin) &&
1351ac708fddSFabien Thomas 			    (args.pa_flags & FLAG_DO_TOP)) {
13520b86b1bbSFabien Thomas 				if (pmcstat_keypress_log())
1353d27927f7SRuslan Bukin 					runstate = pmcstat_close_log(&args);
1354dceed24aSFabien Thomas 			} else {
1355dceed24aSFabien Thomas 				do_read = 0;
13560b86b1bbSFabien Thomas 				runstate = pmcstat_process_log();
1357dceed24aSFabien Thomas 			}
1358ebccf1e3SJoseph Koshy 			break;
1359ebccf1e3SJoseph Koshy 
1360ebccf1e3SJoseph Koshy 		case EVFILT_SIGNAL:
1361f263522aSJoseph Koshy 			if (kev.ident == SIGCHLD) {
1362f263522aSJoseph Koshy 				/*
1363f263522aSJoseph Koshy 				 * The child process sends us a
1364f263522aSJoseph Koshy 				 * SIGCHLD if its exec() failed.  We
1365f263522aSJoseph Koshy 				 * wait for it to exit and then exit
1366f263522aSJoseph Koshy 				 * ourselves.
1367f263522aSJoseph Koshy 				 */
1368f263522aSJoseph Koshy 				(void) wait(&c);
1369f263522aSJoseph Koshy 				runstate = PMCSTAT_FINISHED;
1370f263522aSJoseph Koshy 			} else if (kev.ident == SIGIO) {
1371f263522aSJoseph Koshy 				/*
1372f263522aSJoseph Koshy 				 * We get a SIGIO if a PMC loses all
1373f263522aSJoseph Koshy 				 * of its targets, or if logfile
1374f263522aSJoseph Koshy 				 * writes encounter an error.
1375f263522aSJoseph Koshy 				 */
1376d27927f7SRuslan Bukin 				runstate = pmcstat_close_log(&args);
1377f263522aSJoseph Koshy 				do_print = 1; /* print PMCs at exit */
1378f263522aSJoseph Koshy 			} else if (kev.ident == SIGINT) {
137915139246SJoseph Koshy 				/* Kill the child process if we started it */
138015139246SJoseph Koshy 				if (args.pa_flags & FLAG_HAS_COMMANDLINE)
13810b86b1bbSFabien Thomas 					pmcstat_kill_process();
1382d27927f7SRuslan Bukin 				runstate = pmcstat_close_log(&args);
1383ebccf1e3SJoseph Koshy 			} else if (kev.ident == SIGWINCH) {
138415139246SJoseph Koshy 				if (ioctl(fileno(args.pa_printfile),
1385ebccf1e3SJoseph Koshy 					TIOCGWINSZ, &ws) < 0)
138637d6f8a9SDavid E. O'Brien 				    err(EX_OSERR,
138737d6f8a9SDavid E. O'Brien 				        "ERROR: Cannot determine window size");
1388ebccf1e3SJoseph Koshy 				pmcstat_displayheight = ws.ws_row - 1;
13890b86b1bbSFabien Thomas 				pmcstat_displaywidth  = ws.ws_col - 1;
1390ebccf1e3SJoseph Koshy 			} else
1391ebccf1e3SJoseph Koshy 				assert(0);
1392ebccf1e3SJoseph Koshy 
1393ebccf1e3SJoseph Koshy 			break;
1394f263522aSJoseph Koshy 
139572fe532cSGeorge V. Neville-Neil 		case EVFILT_TIMER:
139672fe532cSGeorge V. Neville-Neil 			/* time duration reached, exit */
139772fe532cSGeorge V. Neville-Neil 			if (args.pa_flags & FLAG_HAS_DURATION) {
139872fe532cSGeorge V. Neville-Neil 				runstate = PMCSTAT_FINISHED;
139972fe532cSGeorge V. Neville-Neil 				break;
140072fe532cSGeorge V. Neville-Neil 			}
140172fe532cSGeorge V. Neville-Neil 			/* print out counting PMCs */
1402dceed24aSFabien Thomas 			if ((args.pa_flags & FLAG_DO_TOP) &&
1403cb43ed28SFabien Thomas 			    (args.pa_flags & FLAG_HAS_PIPE) &&
140466ce9936SFabien Thomas 			     pmc_flush_logfile() == 0)
1405dceed24aSFabien Thomas 				do_read = 1;
1406f263522aSJoseph Koshy 			do_print = 1;
1407f263522aSJoseph Koshy 			break;
1408f263522aSJoseph Koshy 
1409ebccf1e3SJoseph Koshy 		}
1410ebccf1e3SJoseph Koshy 
1411dceed24aSFabien Thomas 		if (do_print && !do_read) {
14120b86b1bbSFabien Thomas 			if ((args.pa_required & FLAG_HAS_OUTPUT_LOGFILE) == 0) {
14130b86b1bbSFabien Thomas 				pmcstat_print_pmcs();
141437d6f8a9SDavid E. O'Brien 				if (runstate == PMCSTAT_FINISHED &&
141537d6f8a9SDavid E. O'Brien 				    /* final newline */
141615139246SJoseph Koshy 				    (args.pa_flags & FLAG_DO_PRINT) == 0)
141715139246SJoseph Koshy 					(void) fprintf(args.pa_printfile, "\n");
14180b86b1bbSFabien Thomas 			}
14190b86b1bbSFabien Thomas 			if (args.pa_flags & FLAG_DO_TOP)
14200b86b1bbSFabien Thomas 				pmcstat_display_log();
1421f263522aSJoseph Koshy 			do_print = 0;
1422f263522aSJoseph Koshy 		}
1423f263522aSJoseph Koshy 
1424f263522aSJoseph Koshy 	} while (runstate != PMCSTAT_FINISHED);
1425f263522aSJoseph Koshy 
14260b86b1bbSFabien Thomas 	if ((args.pa_flags & FLAG_DO_TOP) && args.pa_toptty) {
14270b86b1bbSFabien Thomas 		pmcstat_topexit();
14280b86b1bbSFabien Thomas 		args.pa_toptty = 0;
14290b86b1bbSFabien Thomas 	}
14300b86b1bbSFabien Thomas 
1431f263522aSJoseph Koshy 	/* flush any pending log entries */
143215139246SJoseph Koshy 	if (args.pa_flags & (FLAG_HAS_OUTPUT_LOGFILE | FLAG_HAS_PIPE))
1433dceed24aSFabien Thomas 		pmc_close_logfile();
1434ebccf1e3SJoseph Koshy 
14350b86b1bbSFabien Thomas 	pmcstat_cleanup();
1436ebccf1e3SJoseph Koshy 
14377fbf3da2SJoseph Koshy 	/* check if the driver lost any samples or events */
14387fbf3da2SJoseph Koshy 	if (check_driver_stats) {
14397fbf3da2SJoseph Koshy 		if (pmc_get_driver_stats(&ds_end) < 0)
144037d6f8a9SDavid E. O'Brien 			err(EX_OSERR,
144137d6f8a9SDavid E. O'Brien 			    "ERROR: Cannot retrieve driver statistics");
144249874f6eSJoseph Koshy 		if (ds_start.pm_intr_bufferfull != ds_end.pm_intr_bufferfull &&
144349874f6eSJoseph Koshy 		    args.pa_verbosity > 0)
1444e64ba55dSJonathan T. Looney 			warnx(
1445e64ba55dSJonathan T. Looney "WARNING: sampling was paused at least %u time%s.\n"
1446e64ba55dSJonathan T. Looney "Please consider tuning the \"kern.hwpmc.nsamples\" tunable.",
1447e64ba55dSJonathan T. Looney 			    ds_end.pm_intr_bufferfull -
1448e64ba55dSJonathan T. Looney 			    ds_start.pm_intr_bufferfull,
1449e64ba55dSJonathan T. Looney 			    ((ds_end.pm_intr_bufferfull -
1450e64ba55dSJonathan T. Looney 			    ds_start.pm_intr_bufferfull) != 1) ? "s" : ""
145137d6f8a9SDavid E. O'Brien 			    );
14527fbf3da2SJoseph Koshy 		if (ds_start.pm_buffer_requests_failed !=
145349874f6eSJoseph Koshy 		    ds_end.pm_buffer_requests_failed &&
145449874f6eSJoseph Koshy 		    args.pa_verbosity > 0)
1455e64ba55dSJonathan T. Looney 			warnx(
1456e64ba55dSJonathan T. Looney "WARNING: at least %u event%s were discarded while running.\n"
14570939f965SPiotr Pawel Stefaniak "Please consider tuning the \"kern.hwpmc.nbuffers_pcpu\" tunable.",
1458e64ba55dSJonathan T. Looney 	 		    ds_end.pm_buffer_requests_failed -
1459e64ba55dSJonathan T. Looney 			    ds_start.pm_buffer_requests_failed,
1460e64ba55dSJonathan T. Looney 			    ((ds_end.pm_buffer_requests_failed -
1461e64ba55dSJonathan T. Looney 			    ds_start.pm_buffer_requests_failed) != 1) ? "s" : ""
146237d6f8a9SDavid E. O'Brien 			    );
14637fbf3da2SJoseph Koshy 	}
14647fbf3da2SJoseph Koshy 
14657fbf3da2SJoseph Koshy 	exit(EX_OK);
1466ebccf1e3SJoseph Koshy }
1467