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