xref: /freebsd/usr.sbin/pmcstat/pmcstat_log.c (revision ea056888f5d0cb97b47fa0eb52e06cb389762480)
115139246SJoseph Koshy /*-
2b6010f9eSJoseph Koshy  * Copyright (c) 2005-2007, Joseph Koshy
3b6010f9eSJoseph Koshy  * Copyright (c) 2007 The FreeBSD Foundation
415139246SJoseph Koshy  * All rights reserved.
515139246SJoseph 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  *
915139246SJoseph Koshy  * Redistribution and use in source and binary forms, with or without
1015139246SJoseph Koshy  * modification, are permitted provided that the following conditions
1115139246SJoseph Koshy  * are met:
1215139246SJoseph Koshy  * 1. Redistributions of source code must retain the above copyright
1315139246SJoseph Koshy  *    notice, this list of conditions and the following disclaimer.
1415139246SJoseph Koshy  * 2. Redistributions in binary form must reproduce the above copyright
1515139246SJoseph Koshy  *    notice, this list of conditions and the following disclaimer in the
1615139246SJoseph Koshy  *    documentation and/or other materials provided with the distribution.
1715139246SJoseph Koshy  *
1815139246SJoseph Koshy  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
1915139246SJoseph Koshy  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
2015139246SJoseph Koshy  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
2115139246SJoseph Koshy  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
2215139246SJoseph Koshy  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
2315139246SJoseph Koshy  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
2415139246SJoseph Koshy  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
2515139246SJoseph Koshy  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
2615139246SJoseph Koshy  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
2715139246SJoseph Koshy  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
2815139246SJoseph Koshy  * SUCH DAMAGE.
2915139246SJoseph Koshy  */
3015139246SJoseph Koshy 
3149874f6eSJoseph Koshy /*
3249874f6eSJoseph Koshy  * Transform a hwpmc(4) log into human readable form, and into
3349874f6eSJoseph Koshy  * gprof(1) compatible profiles.
3449874f6eSJoseph Koshy  */
3549874f6eSJoseph Koshy 
3615139246SJoseph Koshy #include <sys/cdefs.h>
3715139246SJoseph Koshy __FBSDID("$FreeBSD$");
3815139246SJoseph Koshy 
3915139246SJoseph Koshy #include <sys/param.h>
4015139246SJoseph Koshy #include <sys/endian.h>
41*ea056888SAttilio Rao #include <sys/cpuset.h>
4215139246SJoseph Koshy #include <sys/gmon.h>
4315139246SJoseph Koshy #include <sys/imgact_aout.h>
4415139246SJoseph Koshy #include <sys/imgact_elf.h>
4515139246SJoseph Koshy #include <sys/mman.h>
4615139246SJoseph Koshy #include <sys/pmc.h>
4715139246SJoseph Koshy #include <sys/queue.h>
48302cbb90SJoseph Koshy #include <sys/socket.h>
4915139246SJoseph Koshy #include <sys/stat.h>
5015139246SJoseph Koshy #include <sys/wait.h>
5115139246SJoseph Koshy 
5215139246SJoseph Koshy #include <netinet/in.h>
5315139246SJoseph Koshy 
5415139246SJoseph Koshy #include <assert.h>
550b86b1bbSFabien Thomas #include <curses.h>
5615139246SJoseph Koshy #include <err.h>
57302cbb90SJoseph Koshy #include <errno.h>
5815139246SJoseph Koshy #include <fcntl.h>
59b6010f9eSJoseph Koshy #include <gelf.h>
6015139246SJoseph Koshy #include <libgen.h>
6115139246SJoseph Koshy #include <limits.h>
62302cbb90SJoseph Koshy #include <netdb.h>
6315139246SJoseph Koshy #include <pmc.h>
6415139246SJoseph Koshy #include <pmclog.h>
6515139246SJoseph Koshy #include <sysexits.h>
6615139246SJoseph Koshy #include <stdint.h>
6715139246SJoseph Koshy #include <stdio.h>
6815139246SJoseph Koshy #include <stdlib.h>
6915139246SJoseph Koshy #include <string.h>
7015139246SJoseph Koshy #include <unistd.h>
7115139246SJoseph Koshy 
7215139246SJoseph Koshy #include "pmcstat.h"
730b86b1bbSFabien Thomas #include "pmcstat_log.h"
740b86b1bbSFabien Thomas #include "pmcstat_top.h"
7515139246SJoseph Koshy 
76b6010f9eSJoseph Koshy #define	PMCSTAT_ALLOCATE		1
77b6010f9eSJoseph Koshy 
7815139246SJoseph Koshy /*
7949874f6eSJoseph Koshy  * PUBLIC INTERFACES
8049874f6eSJoseph Koshy  *
8149874f6eSJoseph Koshy  * pmcstat_initialize_logging()	initialize this module, called first
8249874f6eSJoseph Koshy  * pmcstat_shutdown_logging()		orderly shutdown, called last
8349874f6eSJoseph Koshy  * pmcstat_open_log()			open an eventlog for processing
8449874f6eSJoseph Koshy  * pmcstat_process_log()		print/convert an event log
850b86b1bbSFabien Thomas  * pmcstat_display_log()		top mode display for the log
8649874f6eSJoseph Koshy  * pmcstat_close_log()			finish processing an event log
8749874f6eSJoseph Koshy  *
88b6010f9eSJoseph Koshy  * IMPLEMENTATION NOTES
8949874f6eSJoseph Koshy  *
90b6010f9eSJoseph Koshy  * We correlate each 'callchain' or 'sample' entry seen in the event
91b6010f9eSJoseph Koshy  * log back to an executable object in the system. Executable objects
92b6010f9eSJoseph Koshy  * include:
9349874f6eSJoseph Koshy  * 	- program executables,
9449874f6eSJoseph Koshy  *	- shared libraries loaded by the runtime loader,
9549874f6eSJoseph Koshy  *	- dlopen()'ed objects loaded by the program,
9649874f6eSJoseph Koshy  *	- the runtime loader itself,
9749874f6eSJoseph Koshy  *	- the kernel and kernel modules.
9849874f6eSJoseph Koshy  *
9949874f6eSJoseph Koshy  * Each process that we know about is treated as a set of regions that
10049874f6eSJoseph Koshy  * map to executable objects.  Processes are described by
10149874f6eSJoseph Koshy  * 'pmcstat_process' structures.  Executable objects are tracked by
10249874f6eSJoseph Koshy  * 'pmcstat_image' structures.  The kernel and kernel modules are
10349874f6eSJoseph Koshy  * common to all processes (they reside at the same virtual addresses
10449874f6eSJoseph Koshy  * for all processes).  Individual processes can have their text
10549874f6eSJoseph Koshy  * segments and shared libraries loaded at process-specific locations.
10649874f6eSJoseph Koshy  *
10749874f6eSJoseph Koshy  * A given executable object can be in use by multiple processes
10849874f6eSJoseph Koshy  * (e.g., libc.so) and loaded at a different address in each.
10949874f6eSJoseph Koshy  * pmcstat_pcmap structures track per-image mappings.
11049874f6eSJoseph Koshy  *
11149874f6eSJoseph Koshy  * The sample log could have samples from multiple PMCs; we
11249874f6eSJoseph Koshy  * generate one 'gmon.out' profile per PMC.
113b6010f9eSJoseph Koshy  *
114b6010f9eSJoseph Koshy  * IMPLEMENTATION OF GMON OUTPUT
115b6010f9eSJoseph Koshy  *
116b6010f9eSJoseph Koshy  * Each executable object gets one 'gmon.out' profile, per PMC in
117b6010f9eSJoseph Koshy  * use.  Creation of 'gmon.out' profiles is done lazily.  The
118b6010f9eSJoseph Koshy  * 'gmon.out' profiles generated for a given sampling PMC are
119b6010f9eSJoseph Koshy  * aggregates of all the samples for that particular executable
120b6010f9eSJoseph Koshy  * object.
121b6010f9eSJoseph Koshy  *
122b6010f9eSJoseph Koshy  * IMPLEMENTATION OF SYSTEM-WIDE CALLGRAPH OUTPUT
123b6010f9eSJoseph Koshy  *
124b6010f9eSJoseph Koshy  * Each active pmcid has its own callgraph structure, described by a
125b6010f9eSJoseph Koshy  * 'struct pmcstat_callgraph'.  Given a process id and a list of pc
126b6010f9eSJoseph Koshy  * values, we map each pc value to a tuple (image, symbol), where
127b6010f9eSJoseph Koshy  * 'image' denotes an executable object and 'symbol' is the closest
128b6010f9eSJoseph Koshy  * symbol that precedes the pc value.  Each pc value in the list is
129b6010f9eSJoseph Koshy  * also given a 'rank' that reflects its depth in the call stack.
13015139246SJoseph Koshy  */
13115139246SJoseph Koshy 
1320b86b1bbSFabien Thomas struct pmcstat_pmcs pmcstat_pmcs = LIST_HEAD_INITIALIZER(pmcstat_pmcs);
13315139246SJoseph Koshy 
13449874f6eSJoseph Koshy /*
13549874f6eSJoseph Koshy  * All image descriptors are kept in a hash table.
13649874f6eSJoseph Koshy  */
1370b86b1bbSFabien Thomas struct pmcstat_image_hash_list pmcstat_image_hash[PMCSTAT_NHASH];
13815139246SJoseph Koshy 
13949874f6eSJoseph Koshy /*
14049874f6eSJoseph Koshy  * All process descriptors are kept in a hash table.
14149874f6eSJoseph Koshy  */
1420b86b1bbSFabien Thomas struct pmcstat_process_hash_list pmcstat_process_hash[PMCSTAT_NHASH];
14315139246SJoseph Koshy 
1440b86b1bbSFabien Thomas struct pmcstat_stats pmcstat_stats; /* statistics */
145ac708fddSFabien Thomas int ps_samples_period; /* samples count between top refresh. */
14615139246SJoseph Koshy 
1470b86b1bbSFabien Thomas struct pmcstat_process *pmcstat_kernproc; /* kernel 'process' */
148b6010f9eSJoseph Koshy 
1490b86b1bbSFabien Thomas #include "pmcpl_gprof.h"
1500b86b1bbSFabien Thomas #include "pmcpl_callgraph.h"
1510b86b1bbSFabien Thomas #include "pmcpl_annotate.h"
1520b86b1bbSFabien Thomas #include "pmcpl_calltree.h"
1530b86b1bbSFabien Thomas 
1540b86b1bbSFabien Thomas struct pmc_plugins  {
1550b86b1bbSFabien Thomas 	const char 	*pl_name;	/* name */
1560b86b1bbSFabien Thomas 
1570b86b1bbSFabien Thomas 	/* configure */
1580b86b1bbSFabien Thomas 	int (*pl_configure)(char *opt);
1590b86b1bbSFabien Thomas 
1600b86b1bbSFabien Thomas 	/* init and shutdown */
1610b86b1bbSFabien Thomas 	int (*pl_init)(void);
1620b86b1bbSFabien Thomas 	void (*pl_shutdown)(FILE *mf);
1630b86b1bbSFabien Thomas 
1640b86b1bbSFabien Thomas 	/* sample processing */
1650b86b1bbSFabien Thomas 	void (*pl_process)(struct pmcstat_process *pp,
1660b86b1bbSFabien Thomas 	    struct pmcstat_pmcrecord *pmcr, uint32_t nsamples,
1670b86b1bbSFabien Thomas 	    uintfptr_t *cc, int usermode, uint32_t cpu);
1680b86b1bbSFabien Thomas 
1690b86b1bbSFabien Thomas 	/* image */
1700b86b1bbSFabien Thomas 	void (*pl_initimage)(struct pmcstat_image *pi);
1710b86b1bbSFabien Thomas 	void (*pl_shutdownimage)(struct pmcstat_image *pi);
1720b86b1bbSFabien Thomas 
1730b86b1bbSFabien Thomas 	/* pmc */
1740b86b1bbSFabien Thomas 	void (*pl_newpmc)(pmcstat_interned_string ps,
1750b86b1bbSFabien Thomas 		struct pmcstat_pmcrecord *pr);
1760b86b1bbSFabien Thomas 
1770b86b1bbSFabien Thomas 	/* top display */
1780b86b1bbSFabien Thomas 	void (*pl_topdisplay)(void);
1790b86b1bbSFabien Thomas 
1800b86b1bbSFabien Thomas 	/* top keypress */
1810b86b1bbSFabien Thomas 	int (*pl_topkeypress)(int c, WINDOW *w);
1820b86b1bbSFabien Thomas 
1830b86b1bbSFabien Thomas } plugins[] = {
1840b86b1bbSFabien Thomas 	{
1850b86b1bbSFabien Thomas 		.pl_name		= "none",
1860b86b1bbSFabien Thomas 	},
1870b86b1bbSFabien Thomas 	{
1880b86b1bbSFabien Thomas 		.pl_name		= "callgraph",
1890b86b1bbSFabien Thomas 		.pl_init		= pmcpl_cg_init,
1900b86b1bbSFabien Thomas 		.pl_shutdown		= pmcpl_cg_shutdown,
1910b86b1bbSFabien Thomas 		.pl_process		= pmcpl_cg_process,
1920b86b1bbSFabien Thomas 		.pl_topkeypress		= pmcpl_cg_topkeypress,
1930b86b1bbSFabien Thomas 		.pl_topdisplay		= pmcpl_cg_topdisplay
1940b86b1bbSFabien Thomas 	},
1950b86b1bbSFabien Thomas 	{
1960b86b1bbSFabien Thomas 		.pl_name		= "gprof",
1970b86b1bbSFabien Thomas 		.pl_shutdown		= pmcpl_gmon_shutdown,
1980b86b1bbSFabien Thomas 		.pl_process		= pmcpl_gmon_process,
1990b86b1bbSFabien Thomas 		.pl_initimage		= pmcpl_gmon_initimage,
2000b86b1bbSFabien Thomas 		.pl_shutdownimage	= pmcpl_gmon_shutdownimage,
2010b86b1bbSFabien Thomas 		.pl_newpmc		= pmcpl_gmon_newpmc
2020b86b1bbSFabien Thomas 	},
2030b86b1bbSFabien Thomas 	{
2040b86b1bbSFabien Thomas 		.pl_name		= "annotate",
2050b86b1bbSFabien Thomas 		.pl_process		= pmcpl_annotate_process
2060b86b1bbSFabien Thomas 	},
2070b86b1bbSFabien Thomas 	{
2080b86b1bbSFabien Thomas 		.pl_name		= "calltree",
2090b86b1bbSFabien Thomas 		.pl_configure		= pmcpl_ct_configure,
2100b86b1bbSFabien Thomas 		.pl_init		= pmcpl_ct_init,
2110b86b1bbSFabien Thomas 		.pl_shutdown		= pmcpl_ct_shutdown,
2120b86b1bbSFabien Thomas 		.pl_process		= pmcpl_ct_process,
2130b86b1bbSFabien Thomas 		.pl_topkeypress		= pmcpl_ct_topkeypress,
2140b86b1bbSFabien Thomas 		.pl_topdisplay		= pmcpl_ct_topdisplay
2150b86b1bbSFabien Thomas 	},
2160b86b1bbSFabien Thomas 	{
2170b86b1bbSFabien Thomas 		.pl_name		= NULL
2180b86b1bbSFabien Thomas 	}
219b6010f9eSJoseph Koshy };
220b6010f9eSJoseph Koshy 
2210b86b1bbSFabien Thomas int pmcstat_mergepmc;
222b6010f9eSJoseph Koshy 
2230b86b1bbSFabien Thomas int pmcstat_pmcinfilter = 0; /* PMC filter for top mode. */
2240b86b1bbSFabien Thomas float pmcstat_threshold = 0.5; /* Cost filter for top mode. */
225b6010f9eSJoseph Koshy 
22615139246SJoseph Koshy /*
22715139246SJoseph Koshy  * Prototypes
22815139246SJoseph Koshy  */
22915139246SJoseph Koshy 
23049874f6eSJoseph Koshy static struct pmcstat_image *pmcstat_image_from_path(pmcstat_interned_string
23149874f6eSJoseph Koshy     _path, int _iskernelmodule);
2320b86b1bbSFabien Thomas static void pmcstat_image_get_aout_params(struct pmcstat_image *_image);
2330b86b1bbSFabien Thomas static void pmcstat_image_get_elf_params(struct pmcstat_image *_image);
23415139246SJoseph Koshy static void	pmcstat_image_link(struct pmcstat_process *_pp,
23549874f6eSJoseph Koshy     struct pmcstat_image *_i, uintfptr_t _lpc);
23615139246SJoseph Koshy 
23749874f6eSJoseph Koshy static void	pmcstat_pmcid_add(pmc_id_t _pmcid,
2380b86b1bbSFabien Thomas     pmcstat_interned_string _name);
23915139246SJoseph Koshy 
24049874f6eSJoseph Koshy static void	pmcstat_process_aout_exec(struct pmcstat_process *_pp,
2410b86b1bbSFabien Thomas     struct pmcstat_image *_image, uintfptr_t _entryaddr);
24249874f6eSJoseph Koshy static void	pmcstat_process_elf_exec(struct pmcstat_process *_pp,
2430b86b1bbSFabien Thomas     struct pmcstat_image *_image, uintfptr_t _entryaddr);
24460f918efSJoseph Koshy static void	pmcstat_process_exec(struct pmcstat_process *_pp,
2450b86b1bbSFabien Thomas     pmcstat_interned_string _path, uintfptr_t _entryaddr);
24649874f6eSJoseph Koshy static struct pmcstat_process *pmcstat_process_lookup(pid_t _pid,
24749874f6eSJoseph Koshy     int _allocate);
24815139246SJoseph Koshy static int	pmcstat_string_compute_hash(const char *_string);
24949874f6eSJoseph Koshy static void pmcstat_string_initialize(void);
25049874f6eSJoseph Koshy static int	pmcstat_string_lookup_hash(pmcstat_interned_string _is);
25149874f6eSJoseph Koshy static void pmcstat_string_shutdown(void);
252ac708fddSFabien Thomas static void pmcstat_stats_reset(int _reset_global);
25315139246SJoseph Koshy 
25415139246SJoseph Koshy /*
25549874f6eSJoseph Koshy  * A simple implementation of interned strings.  Each interned string
25649874f6eSJoseph Koshy  * is assigned a unique address, so that subsequent string compares
25749874f6eSJoseph Koshy  * can be done by a simple pointer comparision instead of using
25849874f6eSJoseph Koshy  * strcmp().  This speeds up hash table lookups and saves memory if
25949874f6eSJoseph Koshy  * duplicate strings are the norm.
26049874f6eSJoseph Koshy  */
26149874f6eSJoseph Koshy struct pmcstat_string {
26249874f6eSJoseph Koshy 	LIST_ENTRY(pmcstat_string)	ps_next;	/* hash link */
26349874f6eSJoseph Koshy 	int		ps_len;
26449874f6eSJoseph Koshy 	int		ps_hash;
26549874f6eSJoseph Koshy 	char		*ps_string;
26649874f6eSJoseph Koshy };
26749874f6eSJoseph Koshy 
26849874f6eSJoseph Koshy static LIST_HEAD(,pmcstat_string)	pmcstat_string_hash[PMCSTAT_NHASH];
26949874f6eSJoseph Koshy 
27049874f6eSJoseph Koshy /*
2710b86b1bbSFabien Thomas  * PMC count.
2720b86b1bbSFabien Thomas  */
2730b86b1bbSFabien Thomas int pmcstat_npmcs;
2740b86b1bbSFabien Thomas 
2750b86b1bbSFabien Thomas /*
2760b86b1bbSFabien Thomas  * PMC Top mode pause state.
2770b86b1bbSFabien Thomas  */
2780b86b1bbSFabien Thomas int pmcstat_pause;
2790b86b1bbSFabien Thomas 
280c86819ecSFabien Thomas static void
281ac708fddSFabien Thomas pmcstat_stats_reset(int reset_global)
282c86819ecSFabien Thomas {
283c86819ecSFabien Thomas 	struct pmcstat_pmcrecord *pr;
284c86819ecSFabien Thomas 
285c86819ecSFabien Thomas 	/* Flush PMCs stats. */
286c86819ecSFabien Thomas 	LIST_FOREACH(pr, &pmcstat_pmcs, pr_next) {
287c86819ecSFabien Thomas 		pr->pr_samples = 0;
288c86819ecSFabien Thomas 		pr->pr_dubious_frames = 0;
289c86819ecSFabien Thomas 	}
290ac708fddSFabien Thomas 	ps_samples_period = 0;
291c86819ecSFabien Thomas 
292c86819ecSFabien Thomas 	/* Flush global stats. */
293ac708fddSFabien Thomas 	if (reset_global)
294c86819ecSFabien Thomas 		bzero(&pmcstat_stats, sizeof(struct pmcstat_stats));
295c86819ecSFabien Thomas }
296c86819ecSFabien Thomas 
2970b86b1bbSFabien Thomas /*
29849874f6eSJoseph Koshy  * Compute a 'hash' value for a string.
29949874f6eSJoseph Koshy  */
30049874f6eSJoseph Koshy 
30149874f6eSJoseph Koshy static int
30249874f6eSJoseph Koshy pmcstat_string_compute_hash(const char *s)
30349874f6eSJoseph Koshy {
30449874f6eSJoseph Koshy 	int hash;
30549874f6eSJoseph Koshy 
30649874f6eSJoseph Koshy 	for (hash = 0; *s; s++)
30749874f6eSJoseph Koshy 		hash ^= *s;
30849874f6eSJoseph Koshy 
30949874f6eSJoseph Koshy 	return (hash & PMCSTAT_HASH_MASK);
31049874f6eSJoseph Koshy }
31149874f6eSJoseph Koshy 
31249874f6eSJoseph Koshy /*
31349874f6eSJoseph Koshy  * Intern a copy of string 's', and return a pointer to the
31449874f6eSJoseph Koshy  * interned structure.
31549874f6eSJoseph Koshy  */
31649874f6eSJoseph Koshy 
3170b86b1bbSFabien Thomas pmcstat_interned_string
31849874f6eSJoseph Koshy pmcstat_string_intern(const char *s)
31949874f6eSJoseph Koshy {
32049874f6eSJoseph Koshy 	struct pmcstat_string *ps;
32149874f6eSJoseph Koshy 	const struct pmcstat_string *cps;
32249874f6eSJoseph Koshy 	int hash, len;
32349874f6eSJoseph Koshy 
32449874f6eSJoseph Koshy 	if ((cps = pmcstat_string_lookup(s)) != NULL)
32549874f6eSJoseph Koshy 		return (cps);
32649874f6eSJoseph Koshy 
32749874f6eSJoseph Koshy 	hash = pmcstat_string_compute_hash(s);
32849874f6eSJoseph Koshy 	len  = strlen(s);
32949874f6eSJoseph Koshy 
33049874f6eSJoseph Koshy 	if ((ps = malloc(sizeof(*ps))) == NULL)
33149874f6eSJoseph Koshy 		err(EX_OSERR, "ERROR: Could not intern string");
33249874f6eSJoseph Koshy 	ps->ps_len = len;
33349874f6eSJoseph Koshy 	ps->ps_hash = hash;
33449874f6eSJoseph Koshy 	ps->ps_string = strdup(s);
33549874f6eSJoseph Koshy 	LIST_INSERT_HEAD(&pmcstat_string_hash[hash], ps, ps_next);
33649874f6eSJoseph Koshy 	return ((pmcstat_interned_string) ps);
33749874f6eSJoseph Koshy }
33849874f6eSJoseph Koshy 
3390b86b1bbSFabien Thomas const char *
34049874f6eSJoseph Koshy pmcstat_string_unintern(pmcstat_interned_string str)
34149874f6eSJoseph Koshy {
34249874f6eSJoseph Koshy 	const char *s;
34349874f6eSJoseph Koshy 
34449874f6eSJoseph Koshy 	s = ((const struct pmcstat_string *) str)->ps_string;
34549874f6eSJoseph Koshy 	return (s);
34649874f6eSJoseph Koshy }
34749874f6eSJoseph Koshy 
3480b86b1bbSFabien Thomas pmcstat_interned_string
34949874f6eSJoseph Koshy pmcstat_string_lookup(const char *s)
35049874f6eSJoseph Koshy {
35149874f6eSJoseph Koshy 	struct pmcstat_string *ps;
35249874f6eSJoseph Koshy 	int hash, len;
35349874f6eSJoseph Koshy 
35449874f6eSJoseph Koshy 	hash = pmcstat_string_compute_hash(s);
35549874f6eSJoseph Koshy 	len = strlen(s);
35649874f6eSJoseph Koshy 
35749874f6eSJoseph Koshy 	LIST_FOREACH(ps, &pmcstat_string_hash[hash], ps_next)
35849874f6eSJoseph Koshy 	    if (ps->ps_len == len && ps->ps_hash == hash &&
35949874f6eSJoseph Koshy 		strcmp(ps->ps_string, s) == 0)
36049874f6eSJoseph Koshy 		    return (ps);
36149874f6eSJoseph Koshy 	return (NULL);
36249874f6eSJoseph Koshy }
36349874f6eSJoseph Koshy 
36449874f6eSJoseph Koshy static int
36549874f6eSJoseph Koshy pmcstat_string_lookup_hash(pmcstat_interned_string s)
36649874f6eSJoseph Koshy {
36749874f6eSJoseph Koshy 	const struct pmcstat_string *ps;
36849874f6eSJoseph Koshy 
36949874f6eSJoseph Koshy 	ps = (const struct pmcstat_string *) s;
37049874f6eSJoseph Koshy 	return (ps->ps_hash);
37149874f6eSJoseph Koshy }
37249874f6eSJoseph Koshy 
37349874f6eSJoseph Koshy /*
37449874f6eSJoseph Koshy  * Initialize the string interning facility.
37549874f6eSJoseph Koshy  */
37649874f6eSJoseph Koshy 
37749874f6eSJoseph Koshy static void
37849874f6eSJoseph Koshy pmcstat_string_initialize(void)
37949874f6eSJoseph Koshy {
38049874f6eSJoseph Koshy 	int i;
38149874f6eSJoseph Koshy 
38249874f6eSJoseph Koshy 	for (i = 0; i < PMCSTAT_NHASH; i++)
38349874f6eSJoseph Koshy 		LIST_INIT(&pmcstat_string_hash[i]);
38449874f6eSJoseph Koshy }
38549874f6eSJoseph Koshy 
38649874f6eSJoseph Koshy /*
38749874f6eSJoseph Koshy  * Destroy the string table, free'ing up space.
38849874f6eSJoseph Koshy  */
38949874f6eSJoseph Koshy 
39049874f6eSJoseph Koshy static void
39149874f6eSJoseph Koshy pmcstat_string_shutdown(void)
39249874f6eSJoseph Koshy {
39349874f6eSJoseph Koshy 	int i;
39449874f6eSJoseph Koshy 	struct pmcstat_string *ps, *pstmp;
39549874f6eSJoseph Koshy 
39649874f6eSJoseph Koshy 	for (i = 0; i < PMCSTAT_NHASH; i++)
39749874f6eSJoseph Koshy 		LIST_FOREACH_SAFE(ps, &pmcstat_string_hash[i], ps_next,
39849874f6eSJoseph Koshy 		    pstmp) {
39949874f6eSJoseph Koshy 			LIST_REMOVE(ps, ps_next);
40049874f6eSJoseph Koshy 			free(ps->ps_string);
40149874f6eSJoseph Koshy 			free(ps);
40249874f6eSJoseph Koshy 		}
40349874f6eSJoseph Koshy }
40449874f6eSJoseph Koshy 
40549874f6eSJoseph Koshy /*
40649874f6eSJoseph Koshy  * Determine whether a given executable image is an A.OUT object, and
40749874f6eSJoseph Koshy  * if so, fill in its parameters from the text file.
40849874f6eSJoseph Koshy  * Sets image->pi_type.
40949874f6eSJoseph Koshy  */
41049874f6eSJoseph Koshy 
41115139246SJoseph Koshy static void
4120b86b1bbSFabien Thomas pmcstat_image_get_aout_params(struct pmcstat_image *image)
41349874f6eSJoseph Koshy {
41449874f6eSJoseph Koshy 	int fd;
41549874f6eSJoseph Koshy 	ssize_t nbytes;
41649874f6eSJoseph Koshy 	struct exec ex;
41749874f6eSJoseph Koshy 	const char *path;
41849874f6eSJoseph Koshy 	char buffer[PATH_MAX];
41949874f6eSJoseph Koshy 
42049874f6eSJoseph Koshy 	path = pmcstat_string_unintern(image->pi_execpath);
42149874f6eSJoseph Koshy 	assert(path != NULL);
42249874f6eSJoseph Koshy 
42349874f6eSJoseph Koshy 	if (image->pi_iskernelmodule)
42449874f6eSJoseph Koshy 		errx(EX_SOFTWARE, "ERROR: a.out kernel modules are "
42549874f6eSJoseph Koshy 		    "unsupported \"%s\"", path);
42649874f6eSJoseph Koshy 
42749874f6eSJoseph Koshy 	(void) snprintf(buffer, sizeof(buffer), "%s%s",
4280b86b1bbSFabien Thomas 	    args.pa_fsroot, path);
42949874f6eSJoseph Koshy 
43049874f6eSJoseph Koshy 	if ((fd = open(buffer, O_RDONLY, 0)) < 0 ||
43149874f6eSJoseph Koshy 	    (nbytes = read(fd, &ex, sizeof(ex))) < 0) {
43249874f6eSJoseph Koshy 		warn("WARNING: Cannot determine type of \"%s\"", path);
43349874f6eSJoseph Koshy 		image->pi_type = PMCSTAT_IMAGE_INDETERMINABLE;
43449874f6eSJoseph Koshy 		if (fd != -1)
43549874f6eSJoseph Koshy 			(void) close(fd);
43649874f6eSJoseph Koshy 		return;
43749874f6eSJoseph Koshy 	}
43849874f6eSJoseph Koshy 
43949874f6eSJoseph Koshy 	(void) close(fd);
44049874f6eSJoseph Koshy 
44149874f6eSJoseph Koshy 	if ((unsigned) nbytes != sizeof(ex) ||
44249874f6eSJoseph Koshy 	    N_BADMAG(ex))
44349874f6eSJoseph Koshy 		return;
44449874f6eSJoseph Koshy 
44549874f6eSJoseph Koshy 	image->pi_type = PMCSTAT_IMAGE_AOUT;
44649874f6eSJoseph Koshy 
44749874f6eSJoseph Koshy 	/* TODO: the rest of a.out processing */
44849874f6eSJoseph Koshy 
44949874f6eSJoseph Koshy 	return;
45049874f6eSJoseph Koshy }
45149874f6eSJoseph Koshy 
45249874f6eSJoseph Koshy /*
453b6010f9eSJoseph Koshy  * Helper function.
454b6010f9eSJoseph Koshy  */
455b6010f9eSJoseph Koshy 
456b6010f9eSJoseph Koshy static int
457b6010f9eSJoseph Koshy pmcstat_symbol_compare(const void *a, const void *b)
458b6010f9eSJoseph Koshy {
459b6010f9eSJoseph Koshy 	const struct pmcstat_symbol *sym1, *sym2;
460b6010f9eSJoseph Koshy 
461b6010f9eSJoseph Koshy 	sym1 = (const struct pmcstat_symbol *) a;
462b6010f9eSJoseph Koshy 	sym2 = (const struct pmcstat_symbol *) b;
463b6010f9eSJoseph Koshy 
464b6010f9eSJoseph Koshy 	if (sym1->ps_end <= sym2->ps_start)
465b6010f9eSJoseph Koshy 		return (-1);
466b6010f9eSJoseph Koshy 	if (sym1->ps_start >= sym2->ps_end)
467b6010f9eSJoseph Koshy 		return (1);
468b6010f9eSJoseph Koshy 	return (0);
469b6010f9eSJoseph Koshy }
470b6010f9eSJoseph Koshy 
471b6010f9eSJoseph Koshy /*
472b6010f9eSJoseph Koshy  * Map an address to a symbol in an image.
473b6010f9eSJoseph Koshy  */
474b6010f9eSJoseph Koshy 
4750b86b1bbSFabien Thomas struct pmcstat_symbol *
476b6010f9eSJoseph Koshy pmcstat_symbol_search(struct pmcstat_image *image, uintfptr_t addr)
477b6010f9eSJoseph Koshy {
478b6010f9eSJoseph Koshy 	struct pmcstat_symbol sym;
479b6010f9eSJoseph Koshy 
480b6010f9eSJoseph Koshy 	if (image->pi_symbols == NULL)
481b6010f9eSJoseph Koshy 		return (NULL);
482b6010f9eSJoseph Koshy 
483b6010f9eSJoseph Koshy 	sym.ps_name  = NULL;
484b6010f9eSJoseph Koshy 	sym.ps_start = addr;
485b6010f9eSJoseph Koshy 	sym.ps_end   = addr + 1;
486b6010f9eSJoseph Koshy 
487b6010f9eSJoseph Koshy 	return (bsearch((void *) &sym, image->pi_symbols,
488b6010f9eSJoseph Koshy 		    image->pi_symcount, sizeof(struct pmcstat_symbol),
489b6010f9eSJoseph Koshy 		    pmcstat_symbol_compare));
490b6010f9eSJoseph Koshy }
491b6010f9eSJoseph Koshy 
492b6010f9eSJoseph Koshy /*
493b6010f9eSJoseph Koshy  * Add the list of symbols in the given section to the list associated
494b6010f9eSJoseph Koshy  * with the object.
495b6010f9eSJoseph Koshy  */
496b6010f9eSJoseph Koshy static void
497b6010f9eSJoseph Koshy pmcstat_image_add_symbols(struct pmcstat_image *image, Elf *e,
498b6010f9eSJoseph Koshy     Elf_Scn *scn, GElf_Shdr *sh)
499b6010f9eSJoseph Koshy {
500b6010f9eSJoseph Koshy 	int firsttime;
501b6010f9eSJoseph Koshy 	size_t n, newsyms, nshsyms, nfuncsyms;
502b6010f9eSJoseph Koshy 	struct pmcstat_symbol *symptr;
503b6010f9eSJoseph Koshy 	char *fnname;
504b6010f9eSJoseph Koshy 	GElf_Sym sym;
505b6010f9eSJoseph Koshy 	Elf_Data *data;
506b6010f9eSJoseph Koshy 
507b6010f9eSJoseph Koshy 	if ((data = elf_getdata(scn, NULL)) == NULL)
508b6010f9eSJoseph Koshy 		return;
509b6010f9eSJoseph Koshy 
510b6010f9eSJoseph Koshy 	/*
511b6010f9eSJoseph Koshy 	 * Determine the number of functions named in this
512b6010f9eSJoseph Koshy 	 * section.
513b6010f9eSJoseph Koshy 	 */
514b6010f9eSJoseph Koshy 
515b6010f9eSJoseph Koshy 	nshsyms = sh->sh_size / sh->sh_entsize;
516b6010f9eSJoseph Koshy 	for (n = nfuncsyms = 0; n < nshsyms; n++) {
517b6010f9eSJoseph Koshy 		if (gelf_getsym(data, (int) n, &sym) != &sym)
518b6010f9eSJoseph Koshy 			return;
519b6010f9eSJoseph Koshy 		if (GELF_ST_TYPE(sym.st_info) == STT_FUNC)
520b6010f9eSJoseph Koshy 			nfuncsyms++;
521b6010f9eSJoseph Koshy 	}
522b6010f9eSJoseph Koshy 
523b6010f9eSJoseph Koshy 	if (nfuncsyms == 0)
524b6010f9eSJoseph Koshy 		return;
525b6010f9eSJoseph Koshy 
526b6010f9eSJoseph Koshy 	/*
527b6010f9eSJoseph Koshy 	 * Allocate space for the new entries.
528b6010f9eSJoseph Koshy 	 */
529b6010f9eSJoseph Koshy 	firsttime = image->pi_symbols == NULL;
530b6010f9eSJoseph Koshy 	symptr = realloc(image->pi_symbols,
531b6010f9eSJoseph Koshy 	    sizeof(*symptr) * (image->pi_symcount + nfuncsyms));
532b6010f9eSJoseph Koshy 	if (symptr == image->pi_symbols) /* realloc() failed. */
533b6010f9eSJoseph Koshy 		return;
534b6010f9eSJoseph Koshy 	image->pi_symbols = symptr;
535b6010f9eSJoseph Koshy 
536b6010f9eSJoseph Koshy 	/*
537b6010f9eSJoseph Koshy 	 * Append new symbols to the end of the current table.
538b6010f9eSJoseph Koshy 	 */
539b6010f9eSJoseph Koshy 	symptr += image->pi_symcount;
540b6010f9eSJoseph Koshy 
541b6010f9eSJoseph Koshy 	for (n = newsyms = 0; n < nshsyms; n++) {
542b6010f9eSJoseph Koshy 		if (gelf_getsym(data, (int) n, &sym) != &sym)
543b6010f9eSJoseph Koshy 			return;
544b6010f9eSJoseph Koshy 		if (GELF_ST_TYPE(sym.st_info) != STT_FUNC)
545b6010f9eSJoseph Koshy 			continue;
546176755d3SFabien Thomas 		if (sym.st_shndx == STN_UNDEF)
547176755d3SFabien Thomas 			continue;
548b6010f9eSJoseph Koshy 
549b6010f9eSJoseph Koshy 		if (!firsttime && pmcstat_symbol_search(image, sym.st_value))
550b6010f9eSJoseph Koshy 			continue; /* We've seen this symbol already. */
551b6010f9eSJoseph Koshy 
552b6010f9eSJoseph Koshy 		if ((fnname = elf_strptr(e, sh->sh_link, sym.st_name))
553b6010f9eSJoseph Koshy 		    == NULL)
554b6010f9eSJoseph Koshy 			continue;
555b6010f9eSJoseph Koshy 
556b6010f9eSJoseph Koshy 		symptr->ps_name  = pmcstat_string_intern(fnname);
557b6010f9eSJoseph Koshy 		symptr->ps_start = sym.st_value - image->pi_vaddr;
558b6010f9eSJoseph Koshy 		symptr->ps_end   = symptr->ps_start + sym.st_size;
559b6010f9eSJoseph Koshy 		symptr++;
560b6010f9eSJoseph Koshy 
561b6010f9eSJoseph Koshy 		newsyms++;
562b6010f9eSJoseph Koshy 	}
563b6010f9eSJoseph Koshy 
564b6010f9eSJoseph Koshy 	image->pi_symcount += newsyms;
565b6010f9eSJoseph Koshy 
566b6010f9eSJoseph Koshy 	assert(newsyms <= nfuncsyms);
567b6010f9eSJoseph Koshy 
568b6010f9eSJoseph Koshy 	/*
569b6010f9eSJoseph Koshy 	 * Return space to the system if there were duplicates.
570b6010f9eSJoseph Koshy 	 */
571b6010f9eSJoseph Koshy 	if (newsyms < nfuncsyms)
572b6010f9eSJoseph Koshy 		image->pi_symbols = realloc(image->pi_symbols,
573b6010f9eSJoseph Koshy 		    sizeof(*symptr) * image->pi_symcount);
574b6010f9eSJoseph Koshy 
575b6010f9eSJoseph Koshy 	/*
576b6010f9eSJoseph Koshy 	 * Keep the list of symbols sorted.
577b6010f9eSJoseph Koshy 	 */
578b6010f9eSJoseph Koshy 	qsort(image->pi_symbols, image->pi_symcount, sizeof(*symptr),
579b6010f9eSJoseph Koshy 	    pmcstat_symbol_compare);
580b6010f9eSJoseph Koshy 
581b6010f9eSJoseph Koshy 	/*
582b6010f9eSJoseph Koshy 	 * Deal with function symbols that have a size of 'zero' by
583b6010f9eSJoseph Koshy 	 * making them extend to the next higher address.  These
584b6010f9eSJoseph Koshy 	 * symbols are usually defined in assembly code.
585b6010f9eSJoseph Koshy 	 */
586b6010f9eSJoseph Koshy 	for (symptr = image->pi_symbols;
587b6010f9eSJoseph Koshy 	     symptr < image->pi_symbols + (image->pi_symcount - 1);
588b6010f9eSJoseph Koshy 	     symptr++)
589b6010f9eSJoseph Koshy 		if (symptr->ps_start == symptr->ps_end)
590b6010f9eSJoseph Koshy 			symptr->ps_end = (symptr+1)->ps_start;
591b6010f9eSJoseph Koshy }
592b6010f9eSJoseph Koshy 
593b6010f9eSJoseph Koshy /*
59449874f6eSJoseph Koshy  * Examine an ELF file to determine the size of its text segment.
59549874f6eSJoseph Koshy  * Sets image->pi_type if anything conclusive can be determined about
59649874f6eSJoseph Koshy  * this image.
59749874f6eSJoseph Koshy  */
59849874f6eSJoseph Koshy 
59949874f6eSJoseph Koshy static void
6000b86b1bbSFabien Thomas pmcstat_image_get_elf_params(struct pmcstat_image *image)
60115139246SJoseph Koshy {
602b6010f9eSJoseph Koshy 	int fd;
603b6010f9eSJoseph Koshy 	size_t i, nph, nsh;
604b6010f9eSJoseph Koshy 	const char *path, *elfbase;
6050b86b1bbSFabien Thomas 	char *p, *endp;
60615139246SJoseph Koshy 	uintfptr_t minva, maxva;
607b6010f9eSJoseph Koshy 	Elf *e;
608b6010f9eSJoseph Koshy 	Elf_Scn *scn;
609b6010f9eSJoseph Koshy 	GElf_Ehdr eh;
610b6010f9eSJoseph Koshy 	GElf_Phdr ph;
611b6010f9eSJoseph Koshy 	GElf_Shdr sh;
61249874f6eSJoseph Koshy 	enum pmcstat_image_type image_type;
613e7ef1de5SFabien Thomas 	char buffer[PATH_MAX];
61415139246SJoseph Koshy 
61560f918efSJoseph Koshy 	assert(image->pi_type == PMCSTAT_IMAGE_UNKNOWN);
61660f918efSJoseph Koshy 
617b6010f9eSJoseph Koshy 	image->pi_start = minva = ~(uintfptr_t) 0;
618b6010f9eSJoseph Koshy 	image->pi_end = maxva = (uintfptr_t) 0;
619b6010f9eSJoseph Koshy 	image->pi_type = image_type = PMCSTAT_IMAGE_INDETERMINABLE;
620b6010f9eSJoseph Koshy 	image->pi_isdynamic = 0;
621b6010f9eSJoseph Koshy 	image->pi_dynlinkerpath = NULL;
622b6010f9eSJoseph Koshy 	image->pi_vaddr = 0;
62315139246SJoseph Koshy 
624b6010f9eSJoseph Koshy 	path = pmcstat_string_unintern(image->pi_execpath);
62549874f6eSJoseph Koshy 	assert(path != NULL);
62615139246SJoseph Koshy 
62749874f6eSJoseph Koshy 	/*
62849874f6eSJoseph Koshy 	 * Look for kernel modules under FSROOT/KERNELPATH/NAME,
62949874f6eSJoseph Koshy 	 * and user mode executable objects under FSROOT/PATHNAME.
63049874f6eSJoseph Koshy 	 */
63149874f6eSJoseph Koshy 	if (image->pi_iskernelmodule)
63249874f6eSJoseph Koshy 		(void) snprintf(buffer, sizeof(buffer), "%s%s/%s",
6330b86b1bbSFabien Thomas 		    args.pa_fsroot, args.pa_kernel, path);
63449874f6eSJoseph Koshy 	else
63549874f6eSJoseph Koshy 		(void) snprintf(buffer, sizeof(buffer), "%s%s",
6360b86b1bbSFabien Thomas 		    args.pa_fsroot, path);
63715139246SJoseph Koshy 
638b6010f9eSJoseph Koshy 	e = NULL;
63949874f6eSJoseph Koshy 	if ((fd = open(buffer, O_RDONLY, 0)) < 0 ||
640b6010f9eSJoseph Koshy 	    (e = elf_begin(fd, ELF_C_READ, NULL)) == NULL ||
641b6010f9eSJoseph Koshy 	    (elf_kind(e) != ELF_K_ELF)) {
642b6010f9eSJoseph Koshy 		warnx("WARNING: Cannot determine the type of \"%s\".",
643b6010f9eSJoseph Koshy 		    buffer);
644b6010f9eSJoseph Koshy 		goto done;
64549874f6eSJoseph Koshy 	}
64615139246SJoseph Koshy 
647b6010f9eSJoseph Koshy 	if (gelf_getehdr(e, &eh) != &eh) {
648b6010f9eSJoseph Koshy 		warnx("WARNING: Cannot retrieve the ELF Header for "
649b6010f9eSJoseph Koshy 		    "\"%s\": %s.", buffer, elf_errmsg(-1));
650b6010f9eSJoseph Koshy 		goto done;
651b6010f9eSJoseph Koshy 	}
65215139246SJoseph Koshy 
653b6010f9eSJoseph Koshy 	if (eh.e_type != ET_EXEC && eh.e_type != ET_DYN &&
654b6010f9eSJoseph Koshy 	    !(image->pi_iskernelmodule && eh.e_type == ET_REL)) {
655b6010f9eSJoseph Koshy 		warnx("WARNING: \"%s\" is of an unsupported ELF type.",
656b6010f9eSJoseph Koshy 		    buffer);
657b6010f9eSJoseph Koshy 		goto done;
658b6010f9eSJoseph Koshy 	}
659b6010f9eSJoseph Koshy 
660b6010f9eSJoseph Koshy 	image_type = eh.e_ident[EI_CLASS] == ELFCLASS32 ?
661b6010f9eSJoseph Koshy 	    PMCSTAT_IMAGE_ELF32 : PMCSTAT_IMAGE_ELF64;
66215139246SJoseph Koshy 
66349874f6eSJoseph Koshy 	/*
664b6010f9eSJoseph Koshy 	 * Determine the virtual address where an executable would be
665b6010f9eSJoseph Koshy 	 * loaded.  Additionally, for dynamically linked executables,
666b6010f9eSJoseph Koshy 	 * save the pathname to the runtime linker.
66749874f6eSJoseph Koshy 	 */
668b6010f9eSJoseph Koshy 	if (eh.e_type == ET_EXEC) {
669b6010f9eSJoseph Koshy 		if (elf_getphnum(e, &nph) == 0) {
670b6010f9eSJoseph Koshy 			warnx("WARNING: Could not determine the number of "
671b6010f9eSJoseph Koshy 			    "program headers in \"%s\": %s.", buffer,
672b6010f9eSJoseph Koshy 			    elf_errmsg(-1));
673b6010f9eSJoseph Koshy 			goto done;
674887a8d04SJoseph Koshy 		}
675b6010f9eSJoseph Koshy 		for (i = 0; i < eh.e_phnum; i++) {
676b6010f9eSJoseph Koshy 			if (gelf_getphdr(e, i, &ph) != &ph) {
677b6010f9eSJoseph Koshy 				warnx("WARNING: Retrieval of PHDR entry #%ju "
678b6010f9eSJoseph Koshy 				    "in \"%s\" failed: %s.", (uintmax_t) i,
679b6010f9eSJoseph Koshy 				    buffer, elf_errmsg(-1));
680b6010f9eSJoseph Koshy 				goto done;
681b6010f9eSJoseph Koshy 			}
682b6010f9eSJoseph Koshy 			switch (ph.p_type) {
683b6010f9eSJoseph Koshy 			case PT_DYNAMIC:
684b6010f9eSJoseph Koshy 				image->pi_isdynamic = 1;
685887a8d04SJoseph Koshy 				break;
686b6010f9eSJoseph Koshy 			case PT_INTERP:
687b6010f9eSJoseph Koshy 				if ((elfbase = elf_rawfile(e, NULL)) == NULL) {
688b6010f9eSJoseph Koshy 					warnx("WARNING: Cannot retrieve the "
689b6010f9eSJoseph Koshy 					    "interpreter for \"%s\": %s.",
690b6010f9eSJoseph Koshy 					    buffer, elf_errmsg(-1));
691b6010f9eSJoseph Koshy 					goto done;
692887a8d04SJoseph Koshy 				}
693b6010f9eSJoseph Koshy 				image->pi_dynlinkerpath =
694e7ef1de5SFabien Thomas 				    pmcstat_string_intern(elfbase +
695e7ef1de5SFabien Thomas 				        ph.p_offset);
696b6010f9eSJoseph Koshy 				break;
697b6010f9eSJoseph Koshy 			case PT_LOAD:
698b6010f9eSJoseph Koshy 				if (ph.p_offset == 0)
699b6010f9eSJoseph Koshy 					image->pi_vaddr = ph.p_vaddr;
70015139246SJoseph Koshy 				break;
70115139246SJoseph Koshy 			}
702b6010f9eSJoseph Koshy 		}
703b6010f9eSJoseph Koshy 	}
704887a8d04SJoseph Koshy 
705b6010f9eSJoseph Koshy 	/*
706b6010f9eSJoseph Koshy 	 * Get the min and max VA associated with this ELF object.
707b6010f9eSJoseph Koshy 	 */
708b6010f9eSJoseph Koshy 	if (elf_getshnum(e, &nsh) == 0) {
709b6010f9eSJoseph Koshy 		warnx("WARNING: Could not determine the number of sections "
710b6010f9eSJoseph Koshy 		    "for \"%s\": %s.", buffer, elf_errmsg(-1));
711b6010f9eSJoseph Koshy 		goto done;
712b6010f9eSJoseph Koshy 	}
713b6010f9eSJoseph Koshy 
714b6010f9eSJoseph Koshy 	for (i = 0; i < nsh; i++) {
715b6010f9eSJoseph Koshy 		if ((scn = elf_getscn(e, i)) == NULL ||
716b6010f9eSJoseph Koshy 		    gelf_getshdr(scn, &sh) != &sh) {
717b6010f9eSJoseph Koshy 			warnx("WARNING: Could not retrieve section header "
718b6010f9eSJoseph Koshy 			    "#%ju in \"%s\": %s.", (uintmax_t) i, buffer,
719b6010f9eSJoseph Koshy 			    elf_errmsg(-1));
720b6010f9eSJoseph Koshy 			goto done;
721b6010f9eSJoseph Koshy 		}
722b6010f9eSJoseph Koshy 		if (sh.sh_flags & SHF_EXECINSTR) {
723b6010f9eSJoseph Koshy 			minva = min(minva, sh.sh_addr);
724b6010f9eSJoseph Koshy 			maxva = max(maxva, sh.sh_addr + sh.sh_size);
725b6010f9eSJoseph Koshy 		}
726b6010f9eSJoseph Koshy 		if (sh.sh_type == SHT_SYMTAB || sh.sh_type == SHT_DYNSYM)
727b6010f9eSJoseph Koshy 			pmcstat_image_add_symbols(image, e, scn, &sh);
728b6010f9eSJoseph Koshy 	}
729887a8d04SJoseph Koshy 
730887a8d04SJoseph Koshy 	image->pi_start = minva;
731887a8d04SJoseph Koshy 	image->pi_end   = maxva;
73249874f6eSJoseph Koshy 	image->pi_type  = image_type;
733b6010f9eSJoseph Koshy 	image->pi_fullpath = pmcstat_string_intern(buffer);
73415139246SJoseph Koshy 
7350b86b1bbSFabien Thomas 	/* Build display name
7360b86b1bbSFabien Thomas 	 */
7370b86b1bbSFabien Thomas 	endp = buffer;
7380b86b1bbSFabien Thomas 	for (p = buffer; *p; p++)
7390b86b1bbSFabien Thomas 		if (*p == '/')
7400b86b1bbSFabien Thomas 			endp = p+1;
7410b86b1bbSFabien Thomas 	image->pi_name = pmcstat_string_intern(endp);
7420b86b1bbSFabien Thomas 
743b6010f9eSJoseph Koshy  done:
744b6010f9eSJoseph Koshy 	(void) elf_end(e);
745b6010f9eSJoseph Koshy 	if (fd >= 0)
746b6010f9eSJoseph Koshy 		(void) close(fd);
74749874f6eSJoseph Koshy 	return;
74849874f6eSJoseph Koshy }
74915139246SJoseph Koshy 
75049874f6eSJoseph Koshy /*
75149874f6eSJoseph Koshy  * Given an image descriptor, determine whether it is an ELF, or AOUT.
75249874f6eSJoseph Koshy  * If no handler claims the image, set its type to 'INDETERMINABLE'.
75349874f6eSJoseph Koshy  */
75449874f6eSJoseph Koshy 
7550b86b1bbSFabien Thomas void
7560b86b1bbSFabien Thomas pmcstat_image_determine_type(struct pmcstat_image *image)
75749874f6eSJoseph Koshy {
75849874f6eSJoseph Koshy 	assert(image->pi_type == PMCSTAT_IMAGE_UNKNOWN);
75949874f6eSJoseph Koshy 
76049874f6eSJoseph Koshy 	/* Try each kind of handler in turn */
76149874f6eSJoseph Koshy 	if (image->pi_type == PMCSTAT_IMAGE_UNKNOWN)
7620b86b1bbSFabien Thomas 		pmcstat_image_get_elf_params(image);
76349874f6eSJoseph Koshy 	if (image->pi_type == PMCSTAT_IMAGE_UNKNOWN)
7640b86b1bbSFabien Thomas 		pmcstat_image_get_aout_params(image);
76549874f6eSJoseph Koshy 
76649874f6eSJoseph Koshy 	/*
76749874f6eSJoseph Koshy 	 * Otherwise, remember that we tried to determine
76849874f6eSJoseph Koshy 	 * the object's type and had failed.
76949874f6eSJoseph Koshy 	 */
77049874f6eSJoseph Koshy 	if (image->pi_type == PMCSTAT_IMAGE_UNKNOWN)
77149874f6eSJoseph Koshy 		image->pi_type = PMCSTAT_IMAGE_INDETERMINABLE;
77215139246SJoseph Koshy }
77315139246SJoseph Koshy 
77415139246SJoseph Koshy /*
77560f918efSJoseph Koshy  * Locate an image descriptor given an interned path, adding a fresh
77660f918efSJoseph Koshy  * descriptor to the cache if necessary.  This function also finds a
77760f918efSJoseph Koshy  * suitable name for this image's sample file.
77849874f6eSJoseph Koshy  *
77949874f6eSJoseph Koshy  * We defer filling in the file format specific parts of the image
78049874f6eSJoseph Koshy  * structure till the time we actually see a sample that would fall
78149874f6eSJoseph Koshy  * into this image.
78215139246SJoseph Koshy  */
78315139246SJoseph Koshy 
78415139246SJoseph Koshy static struct pmcstat_image *
78549874f6eSJoseph Koshy pmcstat_image_from_path(pmcstat_interned_string internedpath,
78649874f6eSJoseph Koshy     int iskernelmodule)
78715139246SJoseph Koshy {
7880b86b1bbSFabien Thomas 	int hash;
78915139246SJoseph Koshy 	struct pmcstat_image *pi;
79015139246SJoseph Koshy 
79149874f6eSJoseph Koshy 	hash = pmcstat_string_lookup_hash(internedpath);
79215139246SJoseph Koshy 
79349874f6eSJoseph Koshy 	/* First, look for an existing entry. */
79415139246SJoseph Koshy 	LIST_FOREACH(pi, &pmcstat_image_hash[hash], pi_next)
79549874f6eSJoseph Koshy 	    if (pi->pi_execpath == internedpath &&
796b6010f9eSJoseph Koshy 		  pi->pi_iskernelmodule == iskernelmodule)
79749874f6eSJoseph Koshy 		    return (pi);
79815139246SJoseph Koshy 
79915139246SJoseph Koshy 	/*
800b6010f9eSJoseph Koshy 	 * Allocate a new entry and place it at the head of the hash
801b6010f9eSJoseph Koshy 	 * and LRU lists.
80215139246SJoseph Koshy 	 */
80315139246SJoseph Koshy 	pi = malloc(sizeof(*pi));
80415139246SJoseph Koshy 	if (pi == NULL)
80549874f6eSJoseph Koshy 		return (NULL);
80615139246SJoseph Koshy 
80715139246SJoseph Koshy 	pi->pi_type = PMCSTAT_IMAGE_UNKNOWN;
80849874f6eSJoseph Koshy 	pi->pi_execpath = internedpath;
80915139246SJoseph Koshy 	pi->pi_start = ~0;
81015139246SJoseph Koshy 	pi->pi_end = 0;
811b6010f9eSJoseph Koshy 	pi->pi_entry = 0;
812b6010f9eSJoseph Koshy 	pi->pi_vaddr = 0;
813b6010f9eSJoseph Koshy 	pi->pi_isdynamic = 0;
81449874f6eSJoseph Koshy 	pi->pi_iskernelmodule = iskernelmodule;
815b6010f9eSJoseph Koshy 	pi->pi_dynlinkerpath = NULL;
816b6010f9eSJoseph Koshy 	pi->pi_symbols = NULL;
817b6010f9eSJoseph Koshy 	pi->pi_symcount = 0;
8180b86b1bbSFabien Thomas 	pi->pi_addr2line = NULL;
81915139246SJoseph Koshy 
8200b86b1bbSFabien Thomas 	if (plugins[args.pa_pplugin].pl_initimage != NULL)
8210b86b1bbSFabien Thomas 		plugins[args.pa_pplugin].pl_initimage(pi);
8220b86b1bbSFabien Thomas 	if (plugins[args.pa_plugin].pl_initimage != NULL)
8230b86b1bbSFabien Thomas 		plugins[args.pa_plugin].pl_initimage(pi);
82415139246SJoseph Koshy 
82515139246SJoseph Koshy 	LIST_INSERT_HEAD(&pmcstat_image_hash[hash], pi, pi_next);
82615139246SJoseph Koshy 
82749874f6eSJoseph Koshy 	return (pi);
82815139246SJoseph Koshy }
82915139246SJoseph Koshy 
83015139246SJoseph Koshy /*
83149874f6eSJoseph Koshy  * Record the fact that PC values from 'start' to 'end' come from
83215139246SJoseph Koshy  * image 'image'.
83315139246SJoseph Koshy  */
83415139246SJoseph Koshy 
83515139246SJoseph Koshy static void
83615139246SJoseph Koshy pmcstat_image_link(struct pmcstat_process *pp, struct pmcstat_image *image,
83749874f6eSJoseph Koshy     uintfptr_t start)
83815139246SJoseph Koshy {
83915139246SJoseph Koshy 	struct pmcstat_pcmap *pcm, *pcmnew;
84049874f6eSJoseph Koshy 	uintfptr_t offset;
84149874f6eSJoseph Koshy 
84249874f6eSJoseph Koshy 	assert(image->pi_type != PMCSTAT_IMAGE_UNKNOWN &&
84349874f6eSJoseph Koshy 	    image->pi_type != PMCSTAT_IMAGE_INDETERMINABLE);
84415139246SJoseph Koshy 
84515139246SJoseph Koshy 	if ((pcmnew = malloc(sizeof(*pcmnew))) == NULL)
84649874f6eSJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot create a map entry");
84715139246SJoseph Koshy 
84849874f6eSJoseph Koshy 	/*
84949874f6eSJoseph Koshy 	 * Adjust the map entry to only cover the text portion
85049874f6eSJoseph Koshy 	 * of the object.
85149874f6eSJoseph Koshy 	 */
85249874f6eSJoseph Koshy 
85349874f6eSJoseph Koshy 	offset = start - image->pi_vaddr;
85449874f6eSJoseph Koshy 	pcmnew->ppm_lowpc  = image->pi_start + offset;
85549874f6eSJoseph Koshy 	pcmnew->ppm_highpc = image->pi_end + offset;
85615139246SJoseph Koshy 	pcmnew->ppm_image  = image;
85715139246SJoseph Koshy 
85849874f6eSJoseph Koshy 	assert(pcmnew->ppm_lowpc < pcmnew->ppm_highpc);
85949874f6eSJoseph Koshy 
86049874f6eSJoseph Koshy 	/* Overlapped mmap()'s are assumed to never occur. */
86115139246SJoseph Koshy 	TAILQ_FOREACH(pcm, &pp->pp_map, ppm_next)
86249874f6eSJoseph Koshy 	    if (pcm->ppm_lowpc >= pcmnew->ppm_highpc)
86315139246SJoseph Koshy 		    break;
86415139246SJoseph Koshy 
86515139246SJoseph Koshy 	if (pcm == NULL)
86615139246SJoseph Koshy 		TAILQ_INSERT_TAIL(&pp->pp_map, pcmnew, ppm_next);
86715139246SJoseph Koshy 	else
86815139246SJoseph Koshy 		TAILQ_INSERT_BEFORE(pcm, pcmnew, ppm_next);
86915139246SJoseph Koshy }
87015139246SJoseph Koshy 
87115139246SJoseph Koshy /*
87249874f6eSJoseph Koshy  * Unmap images in the range [start..end) associated with process
87349874f6eSJoseph Koshy  * 'pp'.
87449874f6eSJoseph Koshy  */
87549874f6eSJoseph Koshy 
87649874f6eSJoseph Koshy static void
87749874f6eSJoseph Koshy pmcstat_image_unmap(struct pmcstat_process *pp, uintfptr_t start,
87849874f6eSJoseph Koshy     uintfptr_t end)
87949874f6eSJoseph Koshy {
88049874f6eSJoseph Koshy 	struct pmcstat_pcmap *pcm, *pcmtmp, *pcmnew;
88149874f6eSJoseph Koshy 
88249874f6eSJoseph Koshy 	assert(pp != NULL);
88349874f6eSJoseph Koshy 	assert(start < end);
88449874f6eSJoseph Koshy 
88549874f6eSJoseph Koshy 	/*
88649874f6eSJoseph Koshy 	 * Cases:
88749874f6eSJoseph Koshy 	 * - we could have the range completely in the middle of an
88849874f6eSJoseph Koshy 	 *   existing pcmap; in this case we have to split the pcmap
88949874f6eSJoseph Koshy 	 *   structure into two (i.e., generate a 'hole').
89049874f6eSJoseph Koshy 	 * - we could have the range covering multiple pcmaps; these
89149874f6eSJoseph Koshy 	 *   will have to be removed.
89249874f6eSJoseph Koshy 	 * - we could have either 'start' or 'end' falling in the
89349874f6eSJoseph Koshy 	 *   middle of a pcmap; in this case shorten the entry.
89449874f6eSJoseph Koshy 	 */
89549874f6eSJoseph Koshy 	TAILQ_FOREACH_SAFE(pcm, &pp->pp_map, ppm_next, pcmtmp) {
89649874f6eSJoseph Koshy 		assert(pcm->ppm_lowpc < pcm->ppm_highpc);
89749874f6eSJoseph Koshy 		if (pcm->ppm_highpc <= start)
89849874f6eSJoseph Koshy 			continue;
899b6010f9eSJoseph Koshy 		if (pcm->ppm_lowpc >= end)
90049874f6eSJoseph Koshy 			return;
90149874f6eSJoseph Koshy 		if (pcm->ppm_lowpc >= start && pcm->ppm_highpc <= end) {
90249874f6eSJoseph Koshy 			/*
90349874f6eSJoseph Koshy 			 * The current pcmap is completely inside the
90449874f6eSJoseph Koshy 			 * unmapped range: remove it entirely.
90549874f6eSJoseph Koshy 			 */
90649874f6eSJoseph Koshy 			TAILQ_REMOVE(&pp->pp_map, pcm, ppm_next);
90749874f6eSJoseph Koshy 			free(pcm);
90849874f6eSJoseph Koshy 		} else if (pcm->ppm_lowpc < start && pcm->ppm_highpc > end) {
90949874f6eSJoseph Koshy 			/*
91049874f6eSJoseph Koshy 			 * Split this pcmap into two; curtail the
91149874f6eSJoseph Koshy 			 * current map to end at [start-1], and start
91249874f6eSJoseph Koshy 			 * the new one at [end].
91349874f6eSJoseph Koshy 			 */
91449874f6eSJoseph Koshy 			if ((pcmnew = malloc(sizeof(*pcmnew))) == NULL)
91549874f6eSJoseph Koshy 				err(EX_OSERR, "ERROR: Cannot split a map "
91649874f6eSJoseph Koshy 				    "entry");
91749874f6eSJoseph Koshy 
91849874f6eSJoseph Koshy 			pcmnew->ppm_image = pcm->ppm_image;
91949874f6eSJoseph Koshy 
92049874f6eSJoseph Koshy 			pcmnew->ppm_lowpc = end;
92149874f6eSJoseph Koshy 			pcmnew->ppm_highpc = pcm->ppm_highpc;
92249874f6eSJoseph Koshy 
92349874f6eSJoseph Koshy 			pcm->ppm_highpc = start;
92449874f6eSJoseph Koshy 
92549874f6eSJoseph Koshy 			TAILQ_INSERT_AFTER(&pp->pp_map, pcm, pcmnew, ppm_next);
92649874f6eSJoseph Koshy 
92749874f6eSJoseph Koshy 			return;
928b6010f9eSJoseph Koshy 		} else if (pcm->ppm_lowpc < start && pcm->ppm_highpc <= end)
929b6010f9eSJoseph Koshy 			pcm->ppm_highpc = start;
930b6010f9eSJoseph Koshy 		else if (pcm->ppm_lowpc >= start && pcm->ppm_highpc > end)
931b6010f9eSJoseph Koshy 			pcm->ppm_lowpc = end;
93249874f6eSJoseph Koshy 		else
93349874f6eSJoseph Koshy 			assert(0);
93449874f6eSJoseph Koshy 	}
93549874f6eSJoseph Koshy }
93649874f6eSJoseph Koshy 
93749874f6eSJoseph Koshy /*
9380b86b1bbSFabien Thomas  * Resolve file name and line number for the given address.
9390b86b1bbSFabien Thomas  */
9400b86b1bbSFabien Thomas int
9410b86b1bbSFabien Thomas pmcstat_image_addr2line(struct pmcstat_image *image, uintfptr_t addr,
9420b86b1bbSFabien Thomas     char *sourcefile, size_t sourcefile_len, unsigned *sourceline,
9430b86b1bbSFabien Thomas     char *funcname, size_t funcname_len)
9440b86b1bbSFabien Thomas {
9450b86b1bbSFabien Thomas 	static int addr2line_warn = 0;
9460b86b1bbSFabien Thomas 
9470b86b1bbSFabien Thomas 	char *sep, cmdline[PATH_MAX], imagepath[PATH_MAX];
9480b86b1bbSFabien Thomas 	int fd;
9490b86b1bbSFabien Thomas 
9500b86b1bbSFabien Thomas 	if (image->pi_addr2line == NULL) {
951ac708fddSFabien Thomas 		snprintf(imagepath, sizeof(imagepath), "%s%s.symbols",
952ac708fddSFabien Thomas 		    args.pa_fsroot,
9530b86b1bbSFabien Thomas 		    pmcstat_string_unintern(image->pi_fullpath));
9540b86b1bbSFabien Thomas 		fd = open(imagepath, O_RDONLY);
9550b86b1bbSFabien Thomas 		if (fd < 0) {
956ac708fddSFabien Thomas 			snprintf(imagepath, sizeof(imagepath), "%s%s",
957ac708fddSFabien Thomas 			    args.pa_fsroot,
9580b86b1bbSFabien Thomas 			    pmcstat_string_unintern(image->pi_fullpath));
9590b86b1bbSFabien Thomas 		} else
9600b86b1bbSFabien Thomas 			close(fd);
9610b86b1bbSFabien Thomas 		snprintf(cmdline, sizeof(cmdline), "addr2line -Cfe \"%s\"",
9620b86b1bbSFabien Thomas 		    imagepath);
9630b86b1bbSFabien Thomas 		image->pi_addr2line = popen(cmdline, "r+");
9640b86b1bbSFabien Thomas 		if (image->pi_addr2line == NULL) {
9650b86b1bbSFabien Thomas 			if (!addr2line_warn) {
9660b86b1bbSFabien Thomas 				addr2line_warn = 1;
9670b86b1bbSFabien Thomas 				warnx("WARNING: addr2line is needed"
9680b86b1bbSFabien Thomas 				    "for source code information.");
9690b86b1bbSFabien Thomas 			}
9700b86b1bbSFabien Thomas 			return (0);
9710b86b1bbSFabien Thomas 		}
9720b86b1bbSFabien Thomas 	}
9730b86b1bbSFabien Thomas 
9740b86b1bbSFabien Thomas 	if (feof(image->pi_addr2line) || ferror(image->pi_addr2line)) {
9750b86b1bbSFabien Thomas 		warnx("WARNING: addr2line pipe error");
9760b86b1bbSFabien Thomas 		pclose(image->pi_addr2line);
9770b86b1bbSFabien Thomas 		image->pi_addr2line = NULL;
9780b86b1bbSFabien Thomas 		return (0);
9790b86b1bbSFabien Thomas 	}
9800b86b1bbSFabien Thomas 
9810b86b1bbSFabien Thomas 	fprintf(image->pi_addr2line, "%p\n", (void *)addr);
9820b86b1bbSFabien Thomas 
9830b86b1bbSFabien Thomas 	if (fgets(funcname, funcname_len, image->pi_addr2line) == NULL) {
9840b86b1bbSFabien Thomas 		warnx("WARNING: addr2line function name read error");
9850b86b1bbSFabien Thomas 		return (0);
9860b86b1bbSFabien Thomas 	}
9870b86b1bbSFabien Thomas 	sep = strchr(funcname, '\n');
9880b86b1bbSFabien Thomas 	if (sep != NULL)
9890b86b1bbSFabien Thomas 		*sep = '\0';
9900b86b1bbSFabien Thomas 
9910b86b1bbSFabien Thomas 	if (fgets(sourcefile, sourcefile_len, image->pi_addr2line) == NULL) {
9920b86b1bbSFabien Thomas 		warnx("WARNING: addr2line source file read error");
9930b86b1bbSFabien Thomas 		return (0);
9940b86b1bbSFabien Thomas 	}
9950b86b1bbSFabien Thomas 	sep = strchr(sourcefile, ':');
9960b86b1bbSFabien Thomas 	if (sep == NULL) {
9970b86b1bbSFabien Thomas 		warnx("WARNING: addr2line source line separator missing");
9980b86b1bbSFabien Thomas 		return (0);
9990b86b1bbSFabien Thomas 	}
10000b86b1bbSFabien Thomas 	*sep = '\0';
10010b86b1bbSFabien Thomas 	*sourceline = atoi(sep+1);
10020b86b1bbSFabien Thomas 	if (*sourceline == 0)
10030b86b1bbSFabien Thomas 		return (0);
10040b86b1bbSFabien Thomas 
10050b86b1bbSFabien Thomas 	return (1);
10060b86b1bbSFabien Thomas }
10070b86b1bbSFabien Thomas 
10080b86b1bbSFabien Thomas /*
100915139246SJoseph Koshy  * Add a {pmcid,name} mapping.
101015139246SJoseph Koshy  */
101115139246SJoseph Koshy 
101215139246SJoseph Koshy static void
10130b86b1bbSFabien Thomas pmcstat_pmcid_add(pmc_id_t pmcid, pmcstat_interned_string ps)
101415139246SJoseph Koshy {
10150b86b1bbSFabien Thomas 	struct pmcstat_pmcrecord *pr, *prm;
101615139246SJoseph Koshy 
1017b6010f9eSJoseph Koshy 	/* Replace an existing name for the PMC. */
10180b86b1bbSFabien Thomas 	prm = NULL;
101915139246SJoseph Koshy 	LIST_FOREACH(pr, &pmcstat_pmcs, pr_next)
102015139246SJoseph Koshy 		if (pr->pr_pmcid == pmcid) {
102149874f6eSJoseph Koshy 			pr->pr_pmcname = ps;
102215139246SJoseph Koshy 			return;
10230b86b1bbSFabien Thomas 		} else if (pr->pr_pmcname == ps)
10240b86b1bbSFabien Thomas 			prm = pr;
102515139246SJoseph Koshy 
1026b6010f9eSJoseph Koshy 	/*
10270b86b1bbSFabien Thomas 	 * Otherwise, allocate a new descriptor and call the
10280b86b1bbSFabien Thomas 	 * plugins hook.
1029b6010f9eSJoseph Koshy 	 */
103015139246SJoseph Koshy 	if ((pr = malloc(sizeof(*pr))) == NULL)
103115139246SJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot allocate pmc record");
103215139246SJoseph Koshy 
103315139246SJoseph Koshy 	pr->pr_pmcid = pmcid;
103449874f6eSJoseph Koshy 	pr->pr_pmcname = ps;
10350b86b1bbSFabien Thomas 	pr->pr_pmcin = pmcstat_npmcs++;
1036c86819ecSFabien Thomas 	pr->pr_samples = 0;
1037c86819ecSFabien Thomas 	pr->pr_dubious_frames = 0;
10380b86b1bbSFabien Thomas 	pr->pr_merge = prm == NULL ? pr : prm;
10390b86b1bbSFabien Thomas 
104015139246SJoseph Koshy 	LIST_INSERT_HEAD(&pmcstat_pmcs, pr, pr_next);
104115139246SJoseph Koshy 
10420b86b1bbSFabien Thomas 	if (plugins[args.pa_pplugin].pl_newpmc != NULL)
10430b86b1bbSFabien Thomas 		plugins[args.pa_pplugin].pl_newpmc(ps, pr);
10440b86b1bbSFabien Thomas 	if (plugins[args.pa_plugin].pl_newpmc != NULL)
10450b86b1bbSFabien Thomas 		plugins[args.pa_plugin].pl_newpmc(ps, pr);
104615139246SJoseph Koshy }
104715139246SJoseph Koshy 
104815139246SJoseph Koshy /*
104949874f6eSJoseph Koshy  * Given a pmcid in use, find its human-readable name.
105015139246SJoseph Koshy  */
105115139246SJoseph Koshy 
10520b86b1bbSFabien Thomas const char *
105315139246SJoseph Koshy pmcstat_pmcid_to_name(pmc_id_t pmcid)
105415139246SJoseph Koshy {
105515139246SJoseph Koshy 	struct pmcstat_pmcrecord *pr;
105615139246SJoseph Koshy 
105715139246SJoseph Koshy 	LIST_FOREACH(pr, &pmcstat_pmcs, pr_next)
105815139246SJoseph Koshy 	    if (pr->pr_pmcid == pmcid)
105949874f6eSJoseph Koshy 		    return (pmcstat_string_unintern(pr->pr_pmcname));
106015139246SJoseph Koshy 
10610b86b1bbSFabien Thomas 	return NULL;
10620b86b1bbSFabien Thomas }
106315139246SJoseph Koshy 
10640b86b1bbSFabien Thomas /*
10650b86b1bbSFabien Thomas  * Convert PMC index to name.
10660b86b1bbSFabien Thomas  */
106715139246SJoseph Koshy 
10680b86b1bbSFabien Thomas const char *
10690b86b1bbSFabien Thomas pmcstat_pmcindex_to_name(int pmcin)
10700b86b1bbSFabien Thomas {
10710b86b1bbSFabien Thomas 	struct pmcstat_pmcrecord *pr;
107215139246SJoseph Koshy 
10730b86b1bbSFabien Thomas 	LIST_FOREACH(pr, &pmcstat_pmcs, pr_next)
10740b86b1bbSFabien Thomas 		if (pr->pr_pmcin == pmcin)
10750b86b1bbSFabien Thomas 			return pmcstat_string_unintern(pr->pr_pmcname);
10760b86b1bbSFabien Thomas 
10770b86b1bbSFabien Thomas 	return NULL;
10780b86b1bbSFabien Thomas }
10790b86b1bbSFabien Thomas 
10800b86b1bbSFabien Thomas /*
10810b86b1bbSFabien Thomas  * Return PMC record with given index.
10820b86b1bbSFabien Thomas  */
10830b86b1bbSFabien Thomas 
10840b86b1bbSFabien Thomas struct pmcstat_pmcrecord *
10850b86b1bbSFabien Thomas pmcstat_pmcindex_to_pmcr(int pmcin)
10860b86b1bbSFabien Thomas {
10870b86b1bbSFabien Thomas 	struct pmcstat_pmcrecord *pr;
10880b86b1bbSFabien Thomas 
10890b86b1bbSFabien Thomas 	LIST_FOREACH(pr, &pmcstat_pmcs, pr_next)
10900b86b1bbSFabien Thomas 		if (pr->pr_pmcin == pmcin)
10910b86b1bbSFabien Thomas 			return pr;
10920b86b1bbSFabien Thomas 
10930b86b1bbSFabien Thomas 	return NULL;
10940b86b1bbSFabien Thomas }
10950b86b1bbSFabien Thomas 
10960b86b1bbSFabien Thomas /*
10970b86b1bbSFabien Thomas  * Get PMC record by id, apply merge policy.
10980b86b1bbSFabien Thomas  */
10990b86b1bbSFabien Thomas 
11000b86b1bbSFabien Thomas static struct pmcstat_pmcrecord *
11010b86b1bbSFabien Thomas pmcstat_lookup_pmcid(pmc_id_t pmcid)
11020b86b1bbSFabien Thomas {
11030b86b1bbSFabien Thomas 	struct pmcstat_pmcrecord *pr;
11040b86b1bbSFabien Thomas 
11050b86b1bbSFabien Thomas 	LIST_FOREACH(pr, &pmcstat_pmcs, pr_next) {
11060b86b1bbSFabien Thomas 		if (pr->pr_pmcid == pmcid) {
11070b86b1bbSFabien Thomas 			if (pmcstat_mergepmc)
11080b86b1bbSFabien Thomas 				return pr->pr_merge;
11090b86b1bbSFabien Thomas 			return pr;
11100b86b1bbSFabien Thomas 		}
11110b86b1bbSFabien Thomas 	}
11120b86b1bbSFabien Thomas 
11130b86b1bbSFabien Thomas 	return NULL;
111415139246SJoseph Koshy }
111515139246SJoseph Koshy 
111615139246SJoseph Koshy /*
111749874f6eSJoseph Koshy  * Associate an AOUT image with a process.
111815139246SJoseph Koshy  */
111915139246SJoseph Koshy 
112015139246SJoseph Koshy static void
112149874f6eSJoseph Koshy pmcstat_process_aout_exec(struct pmcstat_process *pp,
11220b86b1bbSFabien Thomas     struct pmcstat_image *image, uintfptr_t entryaddr)
112315139246SJoseph Koshy {
112449874f6eSJoseph Koshy 	(void) pp;
112549874f6eSJoseph Koshy 	(void) image;
112649874f6eSJoseph Koshy 	(void) entryaddr;
112749874f6eSJoseph Koshy 	/* TODO Implement a.out handling */
112849874f6eSJoseph Koshy }
112949874f6eSJoseph Koshy 
113049874f6eSJoseph Koshy /*
113149874f6eSJoseph Koshy  * Associate an ELF image with a process.
113249874f6eSJoseph Koshy  */
113349874f6eSJoseph Koshy 
113449874f6eSJoseph Koshy static void
113549874f6eSJoseph Koshy pmcstat_process_elf_exec(struct pmcstat_process *pp,
11360b86b1bbSFabien Thomas     struct pmcstat_image *image, uintfptr_t entryaddr)
113749874f6eSJoseph Koshy {
113815139246SJoseph Koshy 	uintmax_t libstart;
113949874f6eSJoseph Koshy 	struct pmcstat_image *rtldimage;
114015139246SJoseph Koshy 
114149874f6eSJoseph Koshy 	assert(image->pi_type == PMCSTAT_IMAGE_ELF32 ||
114249874f6eSJoseph Koshy 	    image->pi_type == PMCSTAT_IMAGE_ELF64);
114315139246SJoseph Koshy 
114460f918efSJoseph Koshy 	/* Create a map entry for the base executable. */
114549874f6eSJoseph Koshy 	pmcstat_image_link(pp, image, image->pi_vaddr);
114615139246SJoseph Koshy 
114760f918efSJoseph Koshy 	/*
1148b6010f9eSJoseph Koshy 	 * For dynamically linked executables we need to determine
1149b6010f9eSJoseph Koshy 	 * where the dynamic linker was mapped to for this process,
1150b6010f9eSJoseph Koshy 	 * Subsequent executable objects that are mapped in by the
1151b6010f9eSJoseph Koshy 	 * dynamic linker will be tracked by log events of type
1152b6010f9eSJoseph Koshy 	 * PMCLOG_TYPE_MAP_IN.
115360f918efSJoseph Koshy 	 */
115449874f6eSJoseph Koshy 
115515139246SJoseph Koshy 	if (image->pi_isdynamic) {
115615139246SJoseph Koshy 
115760f918efSJoseph Koshy 		/*
115860f918efSJoseph Koshy 		 * The runtime loader gets loaded just after the maximum
115960f918efSJoseph Koshy 		 * possible heap address.  Like so:
116060f918efSJoseph Koshy 		 *
116160f918efSJoseph Koshy 		 * [  TEXT DATA BSS HEAP -->*RTLD  SHLIBS   <--STACK]
116260f918efSJoseph Koshy 		 * ^					            ^
116360f918efSJoseph Koshy 		 * 0				   VM_MAXUSER_ADDRESS
116449874f6eSJoseph Koshy 
116560f918efSJoseph Koshy 		 *
116660f918efSJoseph Koshy 		 * The exact address where the loader gets mapped in
116760f918efSJoseph Koshy 		 * will vary according to the size of the executable
116860f918efSJoseph Koshy 		 * and the limits on the size of the process'es data
116960f918efSJoseph Koshy 		 * segment at the time of exec().  The entry address
117060f918efSJoseph Koshy 		 * recorded at process exec time corresponds to the
117160f918efSJoseph Koshy 		 * 'start' address inside the dynamic linker.  From
117260f918efSJoseph Koshy 		 * this we can figure out the address where the
117360f918efSJoseph Koshy 		 * runtime loader's file object had been mapped to.
117460f918efSJoseph Koshy 		 */
11750b86b1bbSFabien Thomas 		rtldimage = pmcstat_image_from_path(image->pi_dynlinkerpath, 0);
117649874f6eSJoseph Koshy 		if (rtldimage == NULL) {
117749874f6eSJoseph Koshy 			warnx("WARNING: Cannot find image for \"%s\".",
117849874f6eSJoseph Koshy 			    pmcstat_string_unintern(image->pi_dynlinkerpath));
117949874f6eSJoseph Koshy 			pmcstat_stats.ps_exec_errors++;
118049874f6eSJoseph Koshy 			return;
118149874f6eSJoseph Koshy 		}
118249874f6eSJoseph Koshy 
118360f918efSJoseph Koshy 		if (rtldimage->pi_type == PMCSTAT_IMAGE_UNKNOWN)
11840b86b1bbSFabien Thomas 			pmcstat_image_get_elf_params(rtldimage);
118549874f6eSJoseph Koshy 
118649874f6eSJoseph Koshy 		if (rtldimage->pi_type != PMCSTAT_IMAGE_ELF32 &&
118749874f6eSJoseph Koshy 		    rtldimage->pi_type != PMCSTAT_IMAGE_ELF64) {
118849874f6eSJoseph Koshy 			warnx("WARNING: rtld not an ELF object \"%s\".",
118949874f6eSJoseph Koshy 			    pmcstat_string_unintern(image->pi_dynlinkerpath));
119049874f6eSJoseph Koshy 			return;
119149874f6eSJoseph Koshy 		}
119260f918efSJoseph Koshy 
119360f918efSJoseph Koshy 		libstart = entryaddr - rtldimage->pi_entry;
119449874f6eSJoseph Koshy 		pmcstat_image_link(pp, rtldimage, libstart);
119515139246SJoseph Koshy 	}
119615139246SJoseph Koshy }
119715139246SJoseph Koshy 
119815139246SJoseph Koshy /*
119915139246SJoseph Koshy  * Find the process descriptor corresponding to a PID.  If 'allocate'
120015139246SJoseph Koshy  * is zero, we return a NULL if a pid descriptor could not be found or
120115139246SJoseph Koshy  * a process descriptor process.  If 'allocate' is non-zero, then we
120215139246SJoseph Koshy  * will attempt to allocate a fresh process descriptor.  Zombie
120315139246SJoseph Koshy  * process descriptors are only removed if a fresh allocation for the
120415139246SJoseph Koshy  * same PID is requested.
120515139246SJoseph Koshy  */
120615139246SJoseph Koshy 
120715139246SJoseph Koshy static struct pmcstat_process *
120815139246SJoseph Koshy pmcstat_process_lookup(pid_t pid, int allocate)
120915139246SJoseph Koshy {
121015139246SJoseph Koshy 	uint32_t hash;
121115139246SJoseph Koshy 	struct pmcstat_pcmap *ppm, *ppmtmp;
121215139246SJoseph Koshy 	struct pmcstat_process *pp, *pptmp;
121315139246SJoseph Koshy 
121415139246SJoseph Koshy 	hash = (uint32_t) pid & PMCSTAT_HASH_MASK;	/* simplicity wins */
121515139246SJoseph Koshy 
121615139246SJoseph Koshy 	LIST_FOREACH_SAFE(pp, &pmcstat_process_hash[hash], pp_next, pptmp)
121715139246SJoseph Koshy 	    if (pp->pp_pid == pid) {
121815139246SJoseph Koshy 		    /* Found a descriptor, check and process zombies */
121949874f6eSJoseph Koshy 		    if (allocate && pp->pp_isactive == 0) {
122015139246SJoseph Koshy 			    /* remove maps */
122115139246SJoseph Koshy 			    TAILQ_FOREACH_SAFE(ppm, &pp->pp_map, ppm_next,
122215139246SJoseph Koshy 				ppmtmp) {
122315139246SJoseph Koshy 				    TAILQ_REMOVE(&pp->pp_map, ppm, ppm_next);
122415139246SJoseph Koshy 				    free(ppm);
122515139246SJoseph Koshy 			    }
122615139246SJoseph Koshy 			    /* remove process entry */
122715139246SJoseph Koshy 			    LIST_REMOVE(pp, pp_next);
122815139246SJoseph Koshy 			    free(pp);
122915139246SJoseph Koshy 			    break;
123015139246SJoseph Koshy 		    }
123149874f6eSJoseph Koshy 		    return (pp);
123215139246SJoseph Koshy 	    }
123315139246SJoseph Koshy 
123415139246SJoseph Koshy 	if (!allocate)
123549874f6eSJoseph Koshy 		return (NULL);
123615139246SJoseph Koshy 
123715139246SJoseph Koshy 	if ((pp = malloc(sizeof(*pp))) == NULL)
123815139246SJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot allocate pid descriptor");
123915139246SJoseph Koshy 
124015139246SJoseph Koshy 	pp->pp_pid = pid;
124115139246SJoseph Koshy 	pp->pp_isactive = 1;
124215139246SJoseph Koshy 
124315139246SJoseph Koshy 	TAILQ_INIT(&pp->pp_map);
124415139246SJoseph Koshy 
124515139246SJoseph Koshy 	LIST_INSERT_HEAD(&pmcstat_process_hash[hash], pp, pp_next);
124649874f6eSJoseph Koshy 	return (pp);
124715139246SJoseph Koshy }
124815139246SJoseph Koshy 
124915139246SJoseph Koshy /*
125015139246SJoseph Koshy  * Associate an image and a process.
125115139246SJoseph Koshy  */
125215139246SJoseph Koshy 
125315139246SJoseph Koshy static void
125449874f6eSJoseph Koshy pmcstat_process_exec(struct pmcstat_process *pp,
12550b86b1bbSFabien Thomas     pmcstat_interned_string path, uintfptr_t entryaddr)
125615139246SJoseph Koshy {
125715139246SJoseph Koshy 	struct pmcstat_image *image;
125815139246SJoseph Koshy 
125949874f6eSJoseph Koshy 	if ((image = pmcstat_image_from_path(path, 0)) == NULL) {
126049874f6eSJoseph Koshy 		pmcstat_stats.ps_exec_errors++;
126115139246SJoseph Koshy 		return;
126249874f6eSJoseph Koshy 	}
126315139246SJoseph Koshy 
126415139246SJoseph Koshy 	if (image->pi_type == PMCSTAT_IMAGE_UNKNOWN)
12650b86b1bbSFabien Thomas 		pmcstat_image_determine_type(image);
126615139246SJoseph Koshy 
126749874f6eSJoseph Koshy 	assert(image->pi_type != PMCSTAT_IMAGE_UNKNOWN);
126849874f6eSJoseph Koshy 
126949874f6eSJoseph Koshy 	switch (image->pi_type) {
127049874f6eSJoseph Koshy 	case PMCSTAT_IMAGE_ELF32:
127149874f6eSJoseph Koshy 	case PMCSTAT_IMAGE_ELF64:
127249874f6eSJoseph Koshy 		pmcstat_stats.ps_exec_elf++;
12730b86b1bbSFabien Thomas 		pmcstat_process_elf_exec(pp, image, entryaddr);
127415139246SJoseph Koshy 		break;
127515139246SJoseph Koshy 
127615139246SJoseph Koshy 	case PMCSTAT_IMAGE_AOUT:
127749874f6eSJoseph Koshy 		pmcstat_stats.ps_exec_aout++;
12780b86b1bbSFabien Thomas 		pmcstat_process_aout_exec(pp, image, entryaddr);
127949874f6eSJoseph Koshy 		break;
128049874f6eSJoseph Koshy 
128149874f6eSJoseph Koshy 	case PMCSTAT_IMAGE_INDETERMINABLE:
128249874f6eSJoseph Koshy 		pmcstat_stats.ps_exec_indeterminable++;
128315139246SJoseph Koshy 		break;
128415139246SJoseph Koshy 
128515139246SJoseph Koshy 	default:
128660f918efSJoseph Koshy 		err(EX_SOFTWARE, "ERROR: Unsupported executable type for "
128749874f6eSJoseph Koshy 		    "\"%s\"", pmcstat_string_unintern(path));
128815139246SJoseph Koshy 	}
128915139246SJoseph Koshy }
129015139246SJoseph Koshy 
129115139246SJoseph Koshy 
129260f918efSJoseph Koshy /*
129360f918efSJoseph Koshy  * Find the map entry associated with process 'p' at PC value 'pc'.
129460f918efSJoseph Koshy  */
129560f918efSJoseph Koshy 
12960b86b1bbSFabien Thomas struct pmcstat_pcmap *
129760f918efSJoseph Koshy pmcstat_process_find_map(struct pmcstat_process *p, uintfptr_t pc)
129860f918efSJoseph Koshy {
129960f918efSJoseph Koshy 	struct pmcstat_pcmap *ppm;
130060f918efSJoseph Koshy 
130149874f6eSJoseph Koshy 	TAILQ_FOREACH(ppm, &p->pp_map, ppm_next) {
130260f918efSJoseph Koshy 		if (pc >= ppm->ppm_lowpc && pc < ppm->ppm_highpc)
130349874f6eSJoseph Koshy 			return (ppm);
130449874f6eSJoseph Koshy 		if (pc < ppm->ppm_lowpc)
130549874f6eSJoseph Koshy 			return (NULL);
130649874f6eSJoseph Koshy 	}
130760f918efSJoseph Koshy 
130849874f6eSJoseph Koshy 	return (NULL);
130960f918efSJoseph Koshy }
131060f918efSJoseph Koshy 
1311b6010f9eSJoseph Koshy /*
1312b6010f9eSJoseph Koshy  * Convert a hwpmc(4) log to profile information.  A system-wide
1313b6010f9eSJoseph Koshy  * callgraph is generated if FLAG_DO_CALLGRAPHS is set.  gmon.out
1314b6010f9eSJoseph Koshy  * files usable by gprof(1) are created if FLAG_DO_GPROF is set.
1315b6010f9eSJoseph Koshy  */
1316b6010f9eSJoseph Koshy static int
13170b86b1bbSFabien Thomas pmcstat_analyze_log(void)
1318b6010f9eSJoseph Koshy {
1319b6010f9eSJoseph Koshy 	uint32_t cpu, cpuflags;
13200b86b1bbSFabien Thomas 	uintfptr_t pc;
132149874f6eSJoseph Koshy 	pid_t pid;
132249874f6eSJoseph Koshy 	struct pmcstat_image *image;
132315139246SJoseph Koshy 	struct pmcstat_process *pp, *ppnew;
132415139246SJoseph Koshy 	struct pmcstat_pcmap *ppm, *ppmtmp;
132515139246SJoseph Koshy 	struct pmclog_ev ev;
13260b86b1bbSFabien Thomas 	struct pmcstat_pmcrecord *pmcr;
132749874f6eSJoseph Koshy 	pmcstat_interned_string image_path;
132815139246SJoseph Koshy 
13290b86b1bbSFabien Thomas 	assert(args.pa_flags & FLAG_DO_ANALYSIS);
1330b6010f9eSJoseph Koshy 
1331b6010f9eSJoseph Koshy 	if (elf_version(EV_CURRENT) == EV_NONE)
1332b6010f9eSJoseph Koshy 		err(EX_UNAVAILABLE, "Elf library intialization failed");
1333b6010f9eSJoseph Koshy 
13340b86b1bbSFabien Thomas 	while (pmclog_read(args.pa_logparser, &ev) == 0) {
133515139246SJoseph Koshy 		assert(ev.pl_state == PMCLOG_OK);
133615139246SJoseph Koshy 
133715139246SJoseph Koshy 		switch (ev.pl_type) {
133849874f6eSJoseph Koshy 		case PMCLOG_TYPE_INITIALIZE:
133949874f6eSJoseph Koshy 			if ((ev.pl_u.pl_i.pl_version & 0xFF000000) !=
13400b86b1bbSFabien Thomas 			    PMC_VERSION_MAJOR << 24 && args.pa_verbosity > 0)
134149874f6eSJoseph Koshy 				warnx("WARNING: Log version 0x%x does not "
134249874f6eSJoseph Koshy 				    "match compiled version 0x%x.",
134349874f6eSJoseph Koshy 				    ev.pl_u.pl_i.pl_version,
134449874f6eSJoseph Koshy 				    PMC_VERSION_MAJOR);
134549874f6eSJoseph Koshy 			break;
1346b6010f9eSJoseph Koshy 
134749874f6eSJoseph Koshy 		case PMCLOG_TYPE_MAP_IN:
134815139246SJoseph Koshy 			/*
134915139246SJoseph Koshy 			 * Introduce an address range mapping for a
135049874f6eSJoseph Koshy 			 * userland process or the kernel (pid == -1).
135149874f6eSJoseph Koshy 			 *
135249874f6eSJoseph Koshy 			 * We always allocate a process descriptor so
135349874f6eSJoseph Koshy 			 * that subsequent samples seen for this
135449874f6eSJoseph Koshy 			 * address range are mapped to the current
135549874f6eSJoseph Koshy 			 * object being mapped in.
135615139246SJoseph Koshy 			 */
135749874f6eSJoseph Koshy 			pid = ev.pl_u.pl_mi.pl_pid;
135849874f6eSJoseph Koshy 			if (pid == -1)
135949874f6eSJoseph Koshy 				pp = pmcstat_kernproc;
136049874f6eSJoseph Koshy 			else
136149874f6eSJoseph Koshy 				pp = pmcstat_process_lookup(pid,
136249874f6eSJoseph Koshy 				    PMCSTAT_ALLOCATE);
136349874f6eSJoseph Koshy 
136449874f6eSJoseph Koshy 			assert(pp != NULL);
136549874f6eSJoseph Koshy 
136649874f6eSJoseph Koshy 			image_path = pmcstat_string_intern(ev.pl_u.pl_mi.
136749874f6eSJoseph Koshy 			    pl_pathname);
136849874f6eSJoseph Koshy 			image = pmcstat_image_from_path(image_path, pid == -1);
136949874f6eSJoseph Koshy 			if (image->pi_type == PMCSTAT_IMAGE_UNKNOWN)
13700b86b1bbSFabien Thomas 				pmcstat_image_determine_type(image);
137149874f6eSJoseph Koshy 			if (image->pi_type != PMCSTAT_IMAGE_INDETERMINABLE)
137249874f6eSJoseph Koshy 				pmcstat_image_link(pp, image,
137349874f6eSJoseph Koshy 				    ev.pl_u.pl_mi.pl_start);
137449874f6eSJoseph Koshy 			break;
137549874f6eSJoseph Koshy 
137649874f6eSJoseph Koshy 		case PMCLOG_TYPE_MAP_OUT:
137749874f6eSJoseph Koshy 			/*
137849874f6eSJoseph Koshy 			 * Remove an address map.
137949874f6eSJoseph Koshy 			 */
138049874f6eSJoseph Koshy 			pid = ev.pl_u.pl_mo.pl_pid;
138149874f6eSJoseph Koshy 			if (pid == -1)
138249874f6eSJoseph Koshy 				pp = pmcstat_kernproc;
138349874f6eSJoseph Koshy 			else
138449874f6eSJoseph Koshy 				pp = pmcstat_process_lookup(pid, 0);
138549874f6eSJoseph Koshy 
138649874f6eSJoseph Koshy 			if (pp == NULL)	/* unknown process */
138749874f6eSJoseph Koshy 				break;
138849874f6eSJoseph Koshy 
138949874f6eSJoseph Koshy 			pmcstat_image_unmap(pp, ev.pl_u.pl_mo.pl_start,
139049874f6eSJoseph Koshy 			    ev.pl_u.pl_mo.pl_end);
139115139246SJoseph Koshy 			break;
139215139246SJoseph Koshy 
139315139246SJoseph Koshy 		case PMCLOG_TYPE_PCSAMPLE:
1394b6010f9eSJoseph Koshy 			/*
1395b6010f9eSJoseph Koshy 			 * Note: the `PCSAMPLE' log entry is not
1396b6010f9eSJoseph Koshy 			 * generated by hpwmc(4) after version 2.
1397b6010f9eSJoseph Koshy 			 */
139815139246SJoseph Koshy 
139915139246SJoseph Koshy 			/*
140015139246SJoseph Koshy 			 * We bring in the gmon file for the image
140115139246SJoseph Koshy 			 * currently associated with the PMC & pid
140215139246SJoseph Koshy 			 * pair and increment the appropriate entry
140315139246SJoseph Koshy 			 * bin inside this.
140415139246SJoseph Koshy 			 */
140549874f6eSJoseph Koshy 			pmcstat_stats.ps_samples_total++;
1406ac708fddSFabien Thomas 			ps_samples_period++;
140749874f6eSJoseph Koshy 
140815139246SJoseph Koshy 			pc = ev.pl_u.pl_s.pl_pc;
140949874f6eSJoseph Koshy 			pp = pmcstat_process_lookup(ev.pl_u.pl_s.pl_pid,
141049874f6eSJoseph Koshy 			    PMCSTAT_ALLOCATE);
141115139246SJoseph Koshy 
14120b86b1bbSFabien Thomas 			/* Get PMC record. */
14130b86b1bbSFabien Thomas 			pmcr = pmcstat_lookup_pmcid(ev.pl_u.pl_s.pl_pmcid);
14140b86b1bbSFabien Thomas 			assert(pmcr != NULL);
1415c86819ecSFabien Thomas 			pmcr->pr_samples++;
141615139246SJoseph Koshy 
14170b86b1bbSFabien Thomas 			/*
14180b86b1bbSFabien Thomas 			 * Call the plugins processing
14190b86b1bbSFabien Thomas 			 * TODO: move pmcstat_process_find_map inside plugins
14200b86b1bbSFabien Thomas 			 */
14210b86b1bbSFabien Thomas 
14220b86b1bbSFabien Thomas 			if (plugins[args.pa_pplugin].pl_process != NULL)
14230b86b1bbSFabien Thomas 				plugins[args.pa_pplugin].pl_process(
14240b86b1bbSFabien Thomas 				    pp, pmcr, 1, &pc,
14250b86b1bbSFabien Thomas 				    pmcstat_process_find_map(pp, pc) != NULL, 0);
14260b86b1bbSFabien Thomas 			plugins[args.pa_plugin].pl_process(
14270b86b1bbSFabien Thomas 			    pp, pmcr, 1, &pc,
14280b86b1bbSFabien Thomas 			    pmcstat_process_find_map(pp, pc) != NULL, 0);
142915139246SJoseph Koshy 			break;
143015139246SJoseph Koshy 
1431b6010f9eSJoseph Koshy 		case PMCLOG_TYPE_CALLCHAIN:
1432b6010f9eSJoseph Koshy 			pmcstat_stats.ps_samples_total++;
1433ac708fddSFabien Thomas 			ps_samples_period++;
1434b6010f9eSJoseph Koshy 
1435b6010f9eSJoseph Koshy 			cpuflags = ev.pl_u.pl_cc.pl_cpuflags;
1436b6010f9eSJoseph Koshy 			cpu = PMC_CALLCHAIN_CPUFLAGS_TO_CPU(cpuflags);
1437b6010f9eSJoseph Koshy 
1438b6010f9eSJoseph Koshy 			/* Filter on the CPU id. */
1439*ea056888SAttilio Rao 			if (!CPU_ISSET(cpu, &(args.pa_cpumask))) {
1440b6010f9eSJoseph Koshy 				pmcstat_stats.ps_samples_skipped++;
1441b6010f9eSJoseph Koshy 				break;
1442b6010f9eSJoseph Koshy 			}
1443b6010f9eSJoseph Koshy 
1444b6010f9eSJoseph Koshy 			pp = pmcstat_process_lookup(ev.pl_u.pl_cc.pl_pid,
1445b6010f9eSJoseph Koshy 			    PMCSTAT_ALLOCATE);
1446b6010f9eSJoseph Koshy 
14470b86b1bbSFabien Thomas 			/* Get PMC record. */
14480b86b1bbSFabien Thomas 			pmcr = pmcstat_lookup_pmcid(ev.pl_u.pl_cc.pl_pmcid);
14490b86b1bbSFabien Thomas 			assert(pmcr != NULL);
1450c86819ecSFabien Thomas 			pmcr->pr_samples++;
1451b6010f9eSJoseph Koshy 
14520b86b1bbSFabien Thomas 			/*
14530b86b1bbSFabien Thomas 			 * Call the plugins processing
14540b86b1bbSFabien Thomas 			 */
1455b6010f9eSJoseph Koshy 
14560b86b1bbSFabien Thomas 			if (plugins[args.pa_pplugin].pl_process != NULL)
14570b86b1bbSFabien Thomas 				plugins[args.pa_pplugin].pl_process(
14580b86b1bbSFabien Thomas 				    pp, pmcr,
14590b86b1bbSFabien Thomas 				    ev.pl_u.pl_cc.pl_npc,
14600b86b1bbSFabien Thomas 				    ev.pl_u.pl_cc.pl_pc,
14610b86b1bbSFabien Thomas 				    PMC_CALLCHAIN_CPUFLAGS_TO_USERMODE(cpuflags),
14620b86b1bbSFabien Thomas 				    cpu);
14630b86b1bbSFabien Thomas 			plugins[args.pa_plugin].pl_process(
14640b86b1bbSFabien Thomas 			    pp, pmcr,
14650b86b1bbSFabien Thomas 			    ev.pl_u.pl_cc.pl_npc,
14660b86b1bbSFabien Thomas 			    ev.pl_u.pl_cc.pl_pc,
14670b86b1bbSFabien Thomas 			    PMC_CALLCHAIN_CPUFLAGS_TO_USERMODE(cpuflags),
14680b86b1bbSFabien Thomas 			    cpu);
1469b6010f9eSJoseph Koshy 			break;
1470b6010f9eSJoseph Koshy 
147115139246SJoseph Koshy 		case PMCLOG_TYPE_PMCALLOCATE:
147215139246SJoseph Koshy 			/*
147315139246SJoseph Koshy 			 * Record the association pmc id between this
147415139246SJoseph Koshy 			 * PMC and its name.
147515139246SJoseph Koshy 			 */
147615139246SJoseph Koshy 			pmcstat_pmcid_add(ev.pl_u.pl_a.pl_pmcid,
14770b86b1bbSFabien Thomas 			    pmcstat_string_intern(ev.pl_u.pl_a.pl_evname));
147815139246SJoseph Koshy 			break;
147915139246SJoseph Koshy 
148015139246SJoseph Koshy 		case PMCLOG_TYPE_PROCEXEC:
148115139246SJoseph Koshy 
148215139246SJoseph Koshy 			/*
148315139246SJoseph Koshy 			 * Change the executable image associated with
148415139246SJoseph Koshy 			 * a process.
148515139246SJoseph Koshy 			 */
148649874f6eSJoseph Koshy 			pp = pmcstat_process_lookup(ev.pl_u.pl_x.pl_pid,
148749874f6eSJoseph Koshy 			    PMCSTAT_ALLOCATE);
148815139246SJoseph Koshy 
148915139246SJoseph Koshy 			/* delete the current process map */
149015139246SJoseph Koshy 			TAILQ_FOREACH_SAFE(ppm, &pp->pp_map, ppm_next, ppmtmp) {
149115139246SJoseph Koshy 				TAILQ_REMOVE(&pp->pp_map, ppm, ppm_next);
149215139246SJoseph Koshy 				free(ppm);
149315139246SJoseph Koshy 			}
149415139246SJoseph Koshy 
149549874f6eSJoseph Koshy 			/* associate this process  image */
149615139246SJoseph Koshy 			image_path = pmcstat_string_intern(
149715139246SJoseph Koshy 				ev.pl_u.pl_x.pl_pathname);
149849874f6eSJoseph Koshy 			assert(image_path != NULL);
149960f918efSJoseph Koshy 			pmcstat_process_exec(pp, image_path,
15000b86b1bbSFabien Thomas 			    ev.pl_u.pl_x.pl_entryaddr);
150115139246SJoseph Koshy 			break;
150215139246SJoseph Koshy 
150315139246SJoseph Koshy 		case PMCLOG_TYPE_PROCEXIT:
150415139246SJoseph Koshy 
150515139246SJoseph Koshy 			/*
150615139246SJoseph Koshy 			 * Due to the way the log is generated, the
150715139246SJoseph Koshy 			 * last few samples corresponding to a process
150815139246SJoseph Koshy 			 * may appear in the log after the process
150915139246SJoseph Koshy 			 * exit event is recorded.  Thus we keep the
151015139246SJoseph Koshy 			 * process' descriptor and associated data
151115139246SJoseph Koshy 			 * structures around, but mark the process as
151215139246SJoseph Koshy 			 * having exited.
151315139246SJoseph Koshy 			 */
151415139246SJoseph Koshy 			pp = pmcstat_process_lookup(ev.pl_u.pl_e.pl_pid, 0);
151515139246SJoseph Koshy 			if (pp == NULL)
151615139246SJoseph Koshy 				break;
151749874f6eSJoseph Koshy 			pp->pp_isactive = 0;	/* mark as a zombie */
151815139246SJoseph Koshy 			break;
151915139246SJoseph Koshy 
152015139246SJoseph Koshy 		case PMCLOG_TYPE_SYSEXIT:
152115139246SJoseph Koshy 			pp = pmcstat_process_lookup(ev.pl_u.pl_se.pl_pid, 0);
152215139246SJoseph Koshy 			if (pp == NULL)
152315139246SJoseph Koshy 				break;
152415139246SJoseph Koshy 			pp->pp_isactive = 0;	/* make a zombie */
152515139246SJoseph Koshy 			break;
152615139246SJoseph Koshy 
152715139246SJoseph Koshy 		case PMCLOG_TYPE_PROCFORK:
152815139246SJoseph Koshy 
152915139246SJoseph Koshy 			/*
153049874f6eSJoseph Koshy 			 * Allocate a process descriptor for the new
153149874f6eSJoseph Koshy 			 * (child) process.
153249874f6eSJoseph Koshy 			 */
153349874f6eSJoseph Koshy 			ppnew =
153449874f6eSJoseph Koshy 			    pmcstat_process_lookup(ev.pl_u.pl_f.pl_newpid,
153549874f6eSJoseph Koshy 				PMCSTAT_ALLOCATE);
153649874f6eSJoseph Koshy 
153749874f6eSJoseph Koshy 			/*
153849874f6eSJoseph Koshy 			 * If we had been tracking the parent, clone
153949874f6eSJoseph Koshy 			 * its address maps.
154015139246SJoseph Koshy 			 */
154115139246SJoseph Koshy 			pp = pmcstat_process_lookup(ev.pl_u.pl_f.pl_oldpid, 0);
154215139246SJoseph Koshy 			if (pp == NULL)
154315139246SJoseph Koshy 				break;
154415139246SJoseph Koshy 			TAILQ_FOREACH(ppm, &pp->pp_map, ppm_next)
154515139246SJoseph Koshy 			    pmcstat_image_link(ppnew, ppm->ppm_image,
154649874f6eSJoseph Koshy 				ppm->ppm_lowpc);
154715139246SJoseph Koshy 			break;
154815139246SJoseph Koshy 
154915139246SJoseph Koshy 		default:	/* other types of entries are not relevant */
155015139246SJoseph Koshy 			break;
155115139246SJoseph Koshy 		}
155215139246SJoseph Koshy 	}
155315139246SJoseph Koshy 
155415139246SJoseph Koshy 	if (ev.pl_state == PMCLOG_EOF)
155549874f6eSJoseph Koshy 		return (PMCSTAT_FINISHED);
155615139246SJoseph Koshy 	else if (ev.pl_state == PMCLOG_REQUIRE_DATA)
155749874f6eSJoseph Koshy 		return (PMCSTAT_RUNNING);
155815139246SJoseph Koshy 
155915139246SJoseph Koshy 	err(EX_DATAERR, "ERROR: event parsing failed (record %jd, "
156015139246SJoseph Koshy 	    "offset 0x%jx)", (uintmax_t) ev.pl_count + 1, ev.pl_offset);
156115139246SJoseph Koshy }
156215139246SJoseph Koshy 
156315139246SJoseph Koshy /*
156415139246SJoseph Koshy  * Print log entries as text.
156515139246SJoseph Koshy  */
156615139246SJoseph Koshy 
156749874f6eSJoseph Koshy static int
15680b86b1bbSFabien Thomas pmcstat_print_log(void)
156915139246SJoseph Koshy {
157015139246SJoseph Koshy 	struct pmclog_ev ev;
1571b6010f9eSJoseph Koshy 	uint32_t npc;
157215139246SJoseph Koshy 
15730b86b1bbSFabien Thomas 	while (pmclog_read(args.pa_logparser, &ev) == 0) {
157415139246SJoseph Koshy 		assert(ev.pl_state == PMCLOG_OK);
157515139246SJoseph Koshy 		switch (ev.pl_type) {
1576b6010f9eSJoseph Koshy 		case PMCLOG_TYPE_CALLCHAIN:
15770b86b1bbSFabien Thomas 			PMCSTAT_PRINT_ENTRY("callchain",
1578b6010f9eSJoseph Koshy 			    "%d 0x%x %d %d %c", ev.pl_u.pl_cc.pl_pid,
1579b6010f9eSJoseph Koshy 			    ev.pl_u.pl_cc.pl_pmcid,
1580b6010f9eSJoseph Koshy 			    PMC_CALLCHAIN_CPUFLAGS_TO_CPU(ev.pl_u.pl_cc. \
1581b6010f9eSJoseph Koshy 				pl_cpuflags), ev.pl_u.pl_cc.pl_npc,
1582b6010f9eSJoseph Koshy 			    PMC_CALLCHAIN_CPUFLAGS_TO_USERMODE(ev.pl_u.pl_cc.\
1583b6010f9eSJoseph Koshy 			        pl_cpuflags) ? 'u' : 's');
1584b6010f9eSJoseph Koshy 			for (npc = 0; npc < ev.pl_u.pl_cc.pl_npc; npc++)
15850b86b1bbSFabien Thomas 				PMCSTAT_PRINT_ENTRY("...", "%p",
1586b6010f9eSJoseph Koshy 				    (void *) ev.pl_u.pl_cc.pl_pc[npc]);
1587b6010f9eSJoseph Koshy 			break;
158815139246SJoseph Koshy 		case PMCLOG_TYPE_CLOSELOG:
15890b86b1bbSFabien Thomas 			PMCSTAT_PRINT_ENTRY("closelog",);
159015139246SJoseph Koshy 			break;
159115139246SJoseph Koshy 		case PMCLOG_TYPE_DROPNOTIFY:
15920b86b1bbSFabien Thomas 			PMCSTAT_PRINT_ENTRY("drop",);
159315139246SJoseph Koshy 			break;
159415139246SJoseph Koshy 		case PMCLOG_TYPE_INITIALIZE:
15950b86b1bbSFabien Thomas 			PMCSTAT_PRINT_ENTRY("initlog","0x%x \"%s\"",
159615139246SJoseph Koshy 			    ev.pl_u.pl_i.pl_version,
159715139246SJoseph Koshy 			    pmc_name_of_cputype(ev.pl_u.pl_i.pl_arch));
15983ab065b6SJoseph Koshy 			if ((ev.pl_u.pl_i.pl_version & 0xFF000000) !=
15990b86b1bbSFabien Thomas 			    PMC_VERSION_MAJOR << 24 && args.pa_verbosity > 0)
16003ab065b6SJoseph Koshy 				warnx("WARNING: Log version 0x%x != expected "
16013ab065b6SJoseph Koshy 				    "version 0x%x.", ev.pl_u.pl_i.pl_version,
16023ab065b6SJoseph Koshy 				    PMC_VERSION);
160315139246SJoseph Koshy 			break;
160449874f6eSJoseph Koshy 		case PMCLOG_TYPE_MAP_IN:
16050b86b1bbSFabien Thomas 			PMCSTAT_PRINT_ENTRY("map-in","%d %p \"%s\"",
160649874f6eSJoseph Koshy 			    ev.pl_u.pl_mi.pl_pid,
160749874f6eSJoseph Koshy 			    (void *) ev.pl_u.pl_mi.pl_start,
160849874f6eSJoseph Koshy 			    ev.pl_u.pl_mi.pl_pathname);
160949874f6eSJoseph Koshy 			break;
161049874f6eSJoseph Koshy 		case PMCLOG_TYPE_MAP_OUT:
16110b86b1bbSFabien Thomas 			PMCSTAT_PRINT_ENTRY("map-out","%d %p %p",
161249874f6eSJoseph Koshy 			    ev.pl_u.pl_mo.pl_pid,
161349874f6eSJoseph Koshy 			    (void *) ev.pl_u.pl_mo.pl_start,
161449874f6eSJoseph Koshy 			    (void *) ev.pl_u.pl_mo.pl_end);
161515139246SJoseph Koshy 			break;
161615139246SJoseph Koshy 		case PMCLOG_TYPE_PCSAMPLE:
16170b86b1bbSFabien Thomas 			PMCSTAT_PRINT_ENTRY("sample","0x%x %d %p %c",
161815139246SJoseph Koshy 			    ev.pl_u.pl_s.pl_pmcid,
161915139246SJoseph Koshy 			    ev.pl_u.pl_s.pl_pid,
162015139246SJoseph Koshy 			    (void *) ev.pl_u.pl_s.pl_pc,
162115139246SJoseph Koshy 			    ev.pl_u.pl_s.pl_usermode ? 'u' : 's');
162215139246SJoseph Koshy 			break;
162315139246SJoseph Koshy 		case PMCLOG_TYPE_PMCALLOCATE:
16240b86b1bbSFabien Thomas 			PMCSTAT_PRINT_ENTRY("allocate","0x%x \"%s\" 0x%x",
162515139246SJoseph Koshy 			    ev.pl_u.pl_a.pl_pmcid,
162615139246SJoseph Koshy 			    ev.pl_u.pl_a.pl_evname,
162715139246SJoseph Koshy 			    ev.pl_u.pl_a.pl_flags);
162815139246SJoseph Koshy 			break;
162915139246SJoseph Koshy 		case PMCLOG_TYPE_PMCATTACH:
16300b86b1bbSFabien Thomas 			PMCSTAT_PRINT_ENTRY("attach","0x%x %d \"%s\"",
163115139246SJoseph Koshy 			    ev.pl_u.pl_t.pl_pmcid,
163215139246SJoseph Koshy 			    ev.pl_u.pl_t.pl_pid,
163315139246SJoseph Koshy 			    ev.pl_u.pl_t.pl_pathname);
163415139246SJoseph Koshy 			break;
163515139246SJoseph Koshy 		case PMCLOG_TYPE_PMCDETACH:
16360b86b1bbSFabien Thomas 			PMCSTAT_PRINT_ENTRY("detach","0x%x %d",
163715139246SJoseph Koshy 			    ev.pl_u.pl_d.pl_pmcid,
163815139246SJoseph Koshy 			    ev.pl_u.pl_d.pl_pid);
163915139246SJoseph Koshy 			break;
164015139246SJoseph Koshy 		case PMCLOG_TYPE_PROCCSW:
16410b86b1bbSFabien Thomas 			PMCSTAT_PRINT_ENTRY("cswval","0x%x %d %jd",
164215139246SJoseph Koshy 			    ev.pl_u.pl_c.pl_pmcid,
164315139246SJoseph Koshy 			    ev.pl_u.pl_c.pl_pid,
164415139246SJoseph Koshy 			    ev.pl_u.pl_c.pl_value);
164515139246SJoseph Koshy 			break;
164615139246SJoseph Koshy 		case PMCLOG_TYPE_PROCEXEC:
16470b86b1bbSFabien Thomas 			PMCSTAT_PRINT_ENTRY("exec","0x%x %d %p \"%s\"",
164815139246SJoseph Koshy 			    ev.pl_u.pl_x.pl_pmcid,
164915139246SJoseph Koshy 			    ev.pl_u.pl_x.pl_pid,
165015139246SJoseph Koshy 			    (void *) ev.pl_u.pl_x.pl_entryaddr,
165115139246SJoseph Koshy 			    ev.pl_u.pl_x.pl_pathname);
165215139246SJoseph Koshy 			break;
165315139246SJoseph Koshy 		case PMCLOG_TYPE_PROCEXIT:
16540b86b1bbSFabien Thomas 			PMCSTAT_PRINT_ENTRY("exitval","0x%x %d %jd",
165515139246SJoseph Koshy 			    ev.pl_u.pl_e.pl_pmcid,
165615139246SJoseph Koshy 			    ev.pl_u.pl_e.pl_pid,
165715139246SJoseph Koshy 			    ev.pl_u.pl_e.pl_value);
165815139246SJoseph Koshy 			break;
165915139246SJoseph Koshy 		case PMCLOG_TYPE_PROCFORK:
16600b86b1bbSFabien Thomas 			PMCSTAT_PRINT_ENTRY("fork","%d %d",
166115139246SJoseph Koshy 			    ev.pl_u.pl_f.pl_oldpid,
166215139246SJoseph Koshy 			    ev.pl_u.pl_f.pl_newpid);
166315139246SJoseph Koshy 			break;
166415139246SJoseph Koshy 		case PMCLOG_TYPE_USERDATA:
16650b86b1bbSFabien Thomas 			PMCSTAT_PRINT_ENTRY("userdata","0x%x",
166615139246SJoseph Koshy 			    ev.pl_u.pl_u.pl_userdata);
166715139246SJoseph Koshy 			break;
166815139246SJoseph Koshy 		case PMCLOG_TYPE_SYSEXIT:
16690b86b1bbSFabien Thomas 			PMCSTAT_PRINT_ENTRY("exit","%d",
167015139246SJoseph Koshy 			    ev.pl_u.pl_se.pl_pid);
167115139246SJoseph Koshy 			break;
167215139246SJoseph Koshy 		default:
16730b86b1bbSFabien Thomas 			fprintf(args.pa_printfile, "unknown event (type %d).\n",
167415139246SJoseph Koshy 			    ev.pl_type);
167515139246SJoseph Koshy 		}
167615139246SJoseph Koshy 	}
167715139246SJoseph Koshy 
167815139246SJoseph Koshy 	if (ev.pl_state == PMCLOG_EOF)
167949874f6eSJoseph Koshy 		return (PMCSTAT_FINISHED);
168015139246SJoseph Koshy 	else if (ev.pl_state ==  PMCLOG_REQUIRE_DATA)
168149874f6eSJoseph Koshy 		return (PMCSTAT_RUNNING);
168215139246SJoseph Koshy 
16833ab065b6SJoseph Koshy 	errx(EX_DATAERR, "ERROR: event parsing failed "
16843ab065b6SJoseph Koshy 	    "(record %jd, offset 0x%jx).",
168515139246SJoseph Koshy 	    (uintmax_t) ev.pl_count + 1, ev.pl_offset);
168615139246SJoseph Koshy 	/*NOTREACHED*/
168715139246SJoseph Koshy }
168815139246SJoseph Koshy 
168915139246SJoseph Koshy /*
169049874f6eSJoseph Koshy  * Public Interfaces.
169149874f6eSJoseph Koshy  */
169249874f6eSJoseph Koshy 
169349874f6eSJoseph Koshy /*
169449874f6eSJoseph Koshy  * Close a logfile, after first flushing all in-module queued data.
169549874f6eSJoseph Koshy  */
169649874f6eSJoseph Koshy 
169749874f6eSJoseph Koshy int
16980b86b1bbSFabien Thomas pmcstat_close_log(void)
169949874f6eSJoseph Koshy {
1700ac708fddSFabien Thomas 	/* If a local logfile is configured ask the kernel to stop
1701ac708fddSFabien Thomas 	 * and flush data. Kernel will close the file when data is flushed
1702ac708fddSFabien Thomas 	 * so keep the status to EXITING.
1703ac708fddSFabien Thomas 	 */
1704ac708fddSFabien Thomas 	if (args.pa_logfd != -1) {
1705b44906e5SFabien Thomas 		if (pmc_flush_logfile() < 0)
170649874f6eSJoseph Koshy 			err(EX_OSERR, "ERROR: logging failed");
1707ac708fddSFabien Thomas 	}
1708ac708fddSFabien Thomas 
17090b86b1bbSFabien Thomas 	return (args.pa_flags & FLAG_HAS_PIPE ? PMCSTAT_EXITING :
171049874f6eSJoseph Koshy 	    PMCSTAT_FINISHED);
171149874f6eSJoseph Koshy }
171249874f6eSJoseph Koshy 
171349874f6eSJoseph Koshy 
171449874f6eSJoseph Koshy 
171549874f6eSJoseph Koshy /*
171649874f6eSJoseph Koshy  * Open a log file, for reading or writing.
171749874f6eSJoseph Koshy  *
171849874f6eSJoseph Koshy  * The function returns the fd of a successfully opened log or -1 in
171949874f6eSJoseph Koshy  * case of failure.
172049874f6eSJoseph Koshy  */
172149874f6eSJoseph Koshy 
172249874f6eSJoseph Koshy int
172349874f6eSJoseph Koshy pmcstat_open_log(const char *path, int mode)
172449874f6eSJoseph Koshy {
1725ac708fddSFabien Thomas 	int error, fd, cfd;
1726302cbb90SJoseph Koshy 	size_t hlen;
1727302cbb90SJoseph Koshy 	const char *p, *errstr;
1728302cbb90SJoseph Koshy 	struct addrinfo hints, *res, *res0;
1729302cbb90SJoseph Koshy 	char hostname[MAXHOSTNAMELEN];
1730302cbb90SJoseph Koshy 
1731302cbb90SJoseph Koshy 	errstr = NULL;
1732302cbb90SJoseph Koshy 	fd = -1;
173349874f6eSJoseph Koshy 
173449874f6eSJoseph Koshy 	/*
173549874f6eSJoseph Koshy 	 * If 'path' is "-" then open one of stdin or stdout depending
1736302cbb90SJoseph Koshy 	 * on the value of 'mode'.
1737302cbb90SJoseph Koshy 	 *
1738302cbb90SJoseph Koshy 	 * If 'path' contains a ':' and does not start with a '/' or '.',
1739302cbb90SJoseph Koshy 	 * and is being opened for writing, treat it as a "host:port"
1740302cbb90SJoseph Koshy 	 * specification and open a network socket.
1741302cbb90SJoseph Koshy 	 *
1742302cbb90SJoseph Koshy 	 * Otherwise, treat 'path' as a file name and open that.
174349874f6eSJoseph Koshy 	 */
174449874f6eSJoseph Koshy 	if (path[0] == '-' && path[1] == '\0')
174549874f6eSJoseph Koshy 		fd = (mode == PMCSTAT_OPEN_FOR_READ) ? 0 : 1;
1746ac708fddSFabien Thomas 	else if (path[0] != '/' &&
1747302cbb90SJoseph Koshy 	    path[0] != '.' && strchr(path, ':') != NULL) {
1748302cbb90SJoseph Koshy 
1749302cbb90SJoseph Koshy 		p = strrchr(path, ':');
1750302cbb90SJoseph Koshy 		hlen = p - path;
1751302cbb90SJoseph Koshy 		if (p == path || hlen >= sizeof(hostname)) {
1752302cbb90SJoseph Koshy 			errstr = strerror(EINVAL);
1753302cbb90SJoseph Koshy 			goto done;
1754302cbb90SJoseph Koshy 		}
1755302cbb90SJoseph Koshy 
1756302cbb90SJoseph Koshy 		assert(hlen < sizeof(hostname));
1757302cbb90SJoseph Koshy 		(void) strncpy(hostname, path, hlen);
1758302cbb90SJoseph Koshy 		hostname[hlen] = '\0';
1759302cbb90SJoseph Koshy 
1760302cbb90SJoseph Koshy 		(void) memset(&hints, 0, sizeof(hints));
1761302cbb90SJoseph Koshy 		hints.ai_family = AF_UNSPEC;
1762302cbb90SJoseph Koshy 		hints.ai_socktype = SOCK_STREAM;
1763302cbb90SJoseph Koshy 		if ((error = getaddrinfo(hostname, p+1, &hints, &res0)) != 0) {
1764302cbb90SJoseph Koshy 			errstr = gai_strerror(error);
1765302cbb90SJoseph Koshy 			goto done;
1766302cbb90SJoseph Koshy 		}
1767302cbb90SJoseph Koshy 
1768302cbb90SJoseph Koshy 		fd = -1;
1769302cbb90SJoseph Koshy 		for (res = res0; res; res = res->ai_next) {
1770302cbb90SJoseph Koshy 			if ((fd = socket(res->ai_family, res->ai_socktype,
1771302cbb90SJoseph Koshy 			    res->ai_protocol)) < 0) {
1772302cbb90SJoseph Koshy 				errstr = strerror(errno);
1773302cbb90SJoseph Koshy 				continue;
1774302cbb90SJoseph Koshy 			}
1775ac708fddSFabien Thomas 			if (mode == PMCSTAT_OPEN_FOR_READ) {
1776ac708fddSFabien Thomas 				if (bind(fd, res->ai_addr, res->ai_addrlen) < 0) {
1777ac708fddSFabien Thomas 					errstr = strerror(errno);
1778ac708fddSFabien Thomas 					(void) close(fd);
1779ac708fddSFabien Thomas 					fd = -1;
1780ac708fddSFabien Thomas 					continue;
1781ac708fddSFabien Thomas 				}
1782ac708fddSFabien Thomas 				listen(fd, 1);
1783ac708fddSFabien Thomas 				cfd = accept(fd, NULL, NULL);
1784ac708fddSFabien Thomas 				(void) close(fd);
1785ac708fddSFabien Thomas 				if (cfd < 0) {
1786ac708fddSFabien Thomas 					errstr = strerror(errno);
1787ac708fddSFabien Thomas 					fd = -1;
1788ac708fddSFabien Thomas 					break;
1789ac708fddSFabien Thomas 				}
1790ac708fddSFabien Thomas 				fd = cfd;
1791ac708fddSFabien Thomas 			} else {
1792302cbb90SJoseph Koshy 				if (connect(fd, res->ai_addr, res->ai_addrlen) < 0) {
1793302cbb90SJoseph Koshy 					errstr = strerror(errno);
1794302cbb90SJoseph Koshy 					(void) close(fd);
1795302cbb90SJoseph Koshy 					fd = -1;
1796302cbb90SJoseph Koshy 					continue;
1797302cbb90SJoseph Koshy 				}
1798ac708fddSFabien Thomas 			}
1799302cbb90SJoseph Koshy 			errstr = NULL;
1800302cbb90SJoseph Koshy 			break;
1801302cbb90SJoseph Koshy 		}
1802302cbb90SJoseph Koshy 		freeaddrinfo(res0);
1803302cbb90SJoseph Koshy 
1804302cbb90SJoseph Koshy 	} else if ((fd = open(path, mode == PMCSTAT_OPEN_FOR_READ ?
180549874f6eSJoseph Koshy 		    O_RDONLY : (O_WRONLY|O_CREAT|O_TRUNC),
1806302cbb90SJoseph Koshy 		    S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH)) < 0)
1807302cbb90SJoseph Koshy 			errstr = strerror(errno);
1808302cbb90SJoseph Koshy 
1809302cbb90SJoseph Koshy   done:
1810302cbb90SJoseph Koshy 	if (errstr)
1811302cbb90SJoseph Koshy 		errx(EX_OSERR, "ERROR: Cannot open \"%s\" for %s: %s.", path,
1812302cbb90SJoseph Koshy 		    (mode == PMCSTAT_OPEN_FOR_READ ? "reading" : "writing"),
1813302cbb90SJoseph Koshy 		    errstr);
181449874f6eSJoseph Koshy 
181549874f6eSJoseph Koshy 	return (fd);
181649874f6eSJoseph Koshy }
181749874f6eSJoseph Koshy 
181849874f6eSJoseph Koshy /*
181915139246SJoseph Koshy  * Process a log file in offline analysis mode.
182015139246SJoseph Koshy  */
182115139246SJoseph Koshy 
1822dc1d9d2eSJoseph Koshy int
18230b86b1bbSFabien Thomas pmcstat_process_log(void)
182415139246SJoseph Koshy {
182515139246SJoseph Koshy 
182615139246SJoseph Koshy 	/*
1827b6010f9eSJoseph Koshy 	 * If analysis has not been asked for, just print the log to
1828b6010f9eSJoseph Koshy 	 * the current output file.
182915139246SJoseph Koshy 	 */
18300b86b1bbSFabien Thomas 	if (args.pa_flags & FLAG_DO_PRINT)
18310b86b1bbSFabien Thomas 		return (pmcstat_print_log());
183215139246SJoseph Koshy 	else
18330b86b1bbSFabien Thomas 		return (pmcstat_analyze_log());
18340b86b1bbSFabien Thomas }
18350b86b1bbSFabien Thomas 
18360b86b1bbSFabien Thomas /*
18370b86b1bbSFabien Thomas  * Refresh top display.
18380b86b1bbSFabien Thomas  */
18390b86b1bbSFabien Thomas 
18400b86b1bbSFabien Thomas static void
18410b86b1bbSFabien Thomas pmcstat_refresh_top(void)
18420b86b1bbSFabien Thomas {
1843c86819ecSFabien Thomas 	int v_attrs;
1844c86819ecSFabien Thomas 	float v;
18450b86b1bbSFabien Thomas 	char pmcname[40];
1846c86819ecSFabien Thomas 	struct pmcstat_pmcrecord *pmcpr;
18470b86b1bbSFabien Thomas 
18480b86b1bbSFabien Thomas 	/* If in pause mode do not refresh display. */
18490b86b1bbSFabien Thomas 	if (pmcstat_pause)
18500b86b1bbSFabien Thomas 		return;
18510b86b1bbSFabien Thomas 
18529ba84a88SFabien Thomas 	/* Wait until PMC pop in the log. */
1853c86819ecSFabien Thomas 	pmcpr = pmcstat_pmcindex_to_pmcr(pmcstat_pmcinfilter);
1854c86819ecSFabien Thomas 	if (pmcpr == NULL)
18559ba84a88SFabien Thomas 		return;
18569ba84a88SFabien Thomas 
18570b86b1bbSFabien Thomas 	/* Format PMC name. */
18580b86b1bbSFabien Thomas 	if (pmcstat_mergepmc)
1859c86819ecSFabien Thomas 		snprintf(pmcname, sizeof(pmcname), "[%s]",
1860c86819ecSFabien Thomas 		    pmcstat_string_unintern(pmcpr->pr_pmcname));
18610b86b1bbSFabien Thomas 	else
18620b86b1bbSFabien Thomas 		snprintf(pmcname, sizeof(pmcname), "%s.%d",
1863c86819ecSFabien Thomas 		    pmcstat_string_unintern(pmcpr->pr_pmcname),
1864c86819ecSFabien Thomas 		    pmcstat_pmcinfilter);
1865c86819ecSFabien Thomas 
1866c86819ecSFabien Thomas 	/* Format samples count. */
1867ac708fddSFabien Thomas 	if (ps_samples_period > 0)
1868ac708fddSFabien Thomas 		v = (pmcpr->pr_samples * 100.0) / ps_samples_period;
1869c86819ecSFabien Thomas 	else
1870c86819ecSFabien Thomas 		v = 0.;
1871c86819ecSFabien Thomas 	v_attrs = PMCSTAT_ATTRPERCENT(v);
18720b86b1bbSFabien Thomas 
18730b86b1bbSFabien Thomas 	PMCSTAT_PRINTBEGIN();
1874c86819ecSFabien Thomas 	PMCSTAT_PRINTW("PMC: %s Samples: %u ",
18750b86b1bbSFabien Thomas 	    pmcname,
1876c86819ecSFabien Thomas 	    pmcpr->pr_samples);
1877c86819ecSFabien Thomas 	PMCSTAT_ATTRON(v_attrs);
1878c86819ecSFabien Thomas 	PMCSTAT_PRINTW("(%.1f%%) ", v);
1879c86819ecSFabien Thomas 	PMCSTAT_ATTROFF(v_attrs);
1880c86819ecSFabien Thomas 	PMCSTAT_PRINTW(", %u unresolved\n\n",
1881c86819ecSFabien Thomas 	    pmcpr->pr_dubious_frames);
18820b86b1bbSFabien Thomas 	if (plugins[args.pa_plugin].pl_topdisplay != NULL)
18830b86b1bbSFabien Thomas 		plugins[args.pa_plugin].pl_topdisplay();
18840b86b1bbSFabien Thomas 	PMCSTAT_PRINTEND();
18850b86b1bbSFabien Thomas }
18860b86b1bbSFabien Thomas 
18870b86b1bbSFabien Thomas /*
18880b86b1bbSFabien Thomas  * Find the next pmc index to display.
18890b86b1bbSFabien Thomas  */
18900b86b1bbSFabien Thomas 
18910b86b1bbSFabien Thomas static void
18920b86b1bbSFabien Thomas pmcstat_changefilter(void)
18930b86b1bbSFabien Thomas {
18940b86b1bbSFabien Thomas 	int pmcin;
18950b86b1bbSFabien Thomas 	struct pmcstat_pmcrecord *pmcr;
18960b86b1bbSFabien Thomas 
18970b86b1bbSFabien Thomas 	/*
18980b86b1bbSFabien Thomas 	 * Find the next merge target.
18990b86b1bbSFabien Thomas 	 */
19000b86b1bbSFabien Thomas 	if (pmcstat_mergepmc) {
19010b86b1bbSFabien Thomas 		pmcin = pmcstat_pmcinfilter;
19020b86b1bbSFabien Thomas 
19030b86b1bbSFabien Thomas 		do {
19040b86b1bbSFabien Thomas 			pmcr = pmcstat_pmcindex_to_pmcr(pmcstat_pmcinfilter);
1905ac708fddSFabien Thomas 			if (pmcr == NULL || pmcr == pmcr->pr_merge)
19060b86b1bbSFabien Thomas 				break;
19070b86b1bbSFabien Thomas 
19080b86b1bbSFabien Thomas 			pmcstat_pmcinfilter++;
19090b86b1bbSFabien Thomas 			if (pmcstat_pmcinfilter >= pmcstat_npmcs)
19100b86b1bbSFabien Thomas 				pmcstat_pmcinfilter = 0;
19110b86b1bbSFabien Thomas 
19120b86b1bbSFabien Thomas 		} while (pmcstat_pmcinfilter != pmcin);
19130b86b1bbSFabien Thomas 	}
19140b86b1bbSFabien Thomas }
19150b86b1bbSFabien Thomas 
19160b86b1bbSFabien Thomas /*
19170b86b1bbSFabien Thomas  * Top mode keypress.
19180b86b1bbSFabien Thomas  */
19190b86b1bbSFabien Thomas 
19200b86b1bbSFabien Thomas int
19210b86b1bbSFabien Thomas pmcstat_keypress_log(void)
19220b86b1bbSFabien Thomas {
19230b86b1bbSFabien Thomas 	int c, ret = 0;
19240b86b1bbSFabien Thomas 	WINDOW *w;
19250b86b1bbSFabien Thomas 
19260b86b1bbSFabien Thomas 	w = newwin(1, 0, 1, 0);
19270b86b1bbSFabien Thomas 	c = wgetch(w);
19280b86b1bbSFabien Thomas 	wprintw(w, "Key: %c => ", c);
19290b86b1bbSFabien Thomas 	switch (c) {
19300b86b1bbSFabien Thomas 	case 'c':
19310b86b1bbSFabien Thomas 		wprintw(w, "enter mode 'd' or 'a' => ");
19320b86b1bbSFabien Thomas 		c = wgetch(w);
19330b86b1bbSFabien Thomas 		if (c == 'd') {
19340b86b1bbSFabien Thomas 			args.pa_topmode = PMCSTAT_TOP_DELTA;
19350b86b1bbSFabien Thomas 			wprintw(w, "switching to delta mode");
19360b86b1bbSFabien Thomas 		} else {
19370b86b1bbSFabien Thomas 			args.pa_topmode = PMCSTAT_TOP_ACCUM;
19380b86b1bbSFabien Thomas 			wprintw(w, "switching to accumulation mode");
19390b86b1bbSFabien Thomas 		}
19400b86b1bbSFabien Thomas 		break;
19410b86b1bbSFabien Thomas 	case 'm':
19420b86b1bbSFabien Thomas 		pmcstat_mergepmc = !pmcstat_mergepmc;
19430b86b1bbSFabien Thomas 		/*
19440b86b1bbSFabien Thomas 		 * Changing merge state require data reset.
19450b86b1bbSFabien Thomas 		 */
19460b86b1bbSFabien Thomas 		if (plugins[args.pa_plugin].pl_shutdown != NULL)
19470b86b1bbSFabien Thomas 			plugins[args.pa_plugin].pl_shutdown(NULL);
1948ac708fddSFabien Thomas 		pmcstat_stats_reset(0);
19490b86b1bbSFabien Thomas 		if (plugins[args.pa_plugin].pl_init != NULL)
19500b86b1bbSFabien Thomas 			plugins[args.pa_plugin].pl_init();
19510b86b1bbSFabien Thomas 
19520b86b1bbSFabien Thomas 		/* Update filter to be on a merge target. */
19530b86b1bbSFabien Thomas 		pmcstat_changefilter();
19540b86b1bbSFabien Thomas 		wprintw(w, "merge PMC %s", pmcstat_mergepmc ? "on" : "off");
19550b86b1bbSFabien Thomas 		break;
19560b86b1bbSFabien Thomas 	case 'n':
19570b86b1bbSFabien Thomas 		/* Close current plugin. */
19580b86b1bbSFabien Thomas 		if (plugins[args.pa_plugin].pl_shutdown != NULL)
19590b86b1bbSFabien Thomas 			plugins[args.pa_plugin].pl_shutdown(NULL);
19600b86b1bbSFabien Thomas 
19610b86b1bbSFabien Thomas 		/* Find next top display available. */
19620b86b1bbSFabien Thomas 		do {
19630b86b1bbSFabien Thomas 			args.pa_plugin++;
19640b86b1bbSFabien Thomas 			if (plugins[args.pa_plugin].pl_name == NULL)
19650b86b1bbSFabien Thomas 				args.pa_plugin = 0;
19660b86b1bbSFabien Thomas 		} while (plugins[args.pa_plugin].pl_topdisplay == NULL);
19670b86b1bbSFabien Thomas 
19680b86b1bbSFabien Thomas 		/* Open new plugin. */
1969ac708fddSFabien Thomas 		pmcstat_stats_reset(0);
19700b86b1bbSFabien Thomas 		if (plugins[args.pa_plugin].pl_init != NULL)
19710b86b1bbSFabien Thomas 			plugins[args.pa_plugin].pl_init();
19720b86b1bbSFabien Thomas 		wprintw(w, "switching to plugin %s",
19730b86b1bbSFabien Thomas 		    plugins[args.pa_plugin].pl_name);
19740b86b1bbSFabien Thomas 		break;
19750b86b1bbSFabien Thomas 	case 'p':
19760b86b1bbSFabien Thomas 		pmcstat_pmcinfilter++;
19770b86b1bbSFabien Thomas 		if (pmcstat_pmcinfilter >= pmcstat_npmcs)
19780b86b1bbSFabien Thomas 			pmcstat_pmcinfilter = 0;
19790b86b1bbSFabien Thomas 		pmcstat_changefilter();
19800b86b1bbSFabien Thomas 		wprintw(w, "switching to PMC %s.%d",
19810b86b1bbSFabien Thomas 		    pmcstat_pmcindex_to_name(pmcstat_pmcinfilter),
19820b86b1bbSFabien Thomas 		    pmcstat_pmcinfilter);
19830b86b1bbSFabien Thomas 		break;
19840b86b1bbSFabien Thomas 	case ' ':
19850b86b1bbSFabien Thomas 		pmcstat_pause = !pmcstat_pause;
19860b86b1bbSFabien Thomas 		if (pmcstat_pause)
19870b86b1bbSFabien Thomas 			wprintw(w, "pause => press space again to continue");
19880b86b1bbSFabien Thomas 		break;
19890b86b1bbSFabien Thomas 	case 'q':
19900b86b1bbSFabien Thomas 		wprintw(w, "exiting...");
19910b86b1bbSFabien Thomas 		ret = 1;
19925c15d3c8SFabien Thomas 		break;
19930b86b1bbSFabien Thomas 	default:
19940b86b1bbSFabien Thomas 		if (plugins[args.pa_plugin].pl_topkeypress != NULL)
19950b86b1bbSFabien Thomas 			if (plugins[args.pa_plugin].pl_topkeypress(c, w))
19960b86b1bbSFabien Thomas 				ret = 1;
19970b86b1bbSFabien Thomas 	}
19980b86b1bbSFabien Thomas 
19990b86b1bbSFabien Thomas 	wrefresh(w);
20000b86b1bbSFabien Thomas 	delwin(w);
20010b86b1bbSFabien Thomas 	return ret;
20020b86b1bbSFabien Thomas }
20030b86b1bbSFabien Thomas 
20040b86b1bbSFabien Thomas 
20050b86b1bbSFabien Thomas /*
20060b86b1bbSFabien Thomas  * Top mode display.
20070b86b1bbSFabien Thomas  */
20080b86b1bbSFabien Thomas 
20090b86b1bbSFabien Thomas void
20100b86b1bbSFabien Thomas pmcstat_display_log(void)
20110b86b1bbSFabien Thomas {
20120b86b1bbSFabien Thomas 
20130b86b1bbSFabien Thomas 	pmcstat_refresh_top();
20140b86b1bbSFabien Thomas 
20150b86b1bbSFabien Thomas 	/* Reset everythings if delta mode. */
20160b86b1bbSFabien Thomas 	if (args.pa_topmode == PMCSTAT_TOP_DELTA) {
20170b86b1bbSFabien Thomas 		if (plugins[args.pa_plugin].pl_shutdown != NULL)
20180b86b1bbSFabien Thomas 			plugins[args.pa_plugin].pl_shutdown(NULL);
2019ac708fddSFabien Thomas 		pmcstat_stats_reset(0);
20200b86b1bbSFabien Thomas 		if (plugins[args.pa_plugin].pl_init != NULL)
20210b86b1bbSFabien Thomas 			plugins[args.pa_plugin].pl_init();
20220b86b1bbSFabien Thomas 	}
20230b86b1bbSFabien Thomas 
20240b86b1bbSFabien Thomas }
20250b86b1bbSFabien Thomas 
20260b86b1bbSFabien Thomas /*
20270b86b1bbSFabien Thomas  * Configure a plugins.
20280b86b1bbSFabien Thomas  */
20290b86b1bbSFabien Thomas 
20300b86b1bbSFabien Thomas void
20310b86b1bbSFabien Thomas pmcstat_pluginconfigure_log(char *opt)
20320b86b1bbSFabien Thomas {
20330b86b1bbSFabien Thomas 
20340b86b1bbSFabien Thomas 	if (strncmp(opt, "threshold=", 10) == 0) {
20350b86b1bbSFabien Thomas 		pmcstat_threshold = atof(opt+10);
20360b86b1bbSFabien Thomas 	} else {
20370b86b1bbSFabien Thomas 		if (plugins[args.pa_plugin].pl_configure != NULL) {
20380b86b1bbSFabien Thomas 			if (!plugins[args.pa_plugin].pl_configure(opt))
20390b86b1bbSFabien Thomas 				err(EX_USAGE,
20400b86b1bbSFabien Thomas 				    "ERROR: unknown option <%s>.", opt);
20410b86b1bbSFabien Thomas 		}
20420b86b1bbSFabien Thomas 	}
204315139246SJoseph Koshy }
204415139246SJoseph Koshy 
204549874f6eSJoseph Koshy /*
204649874f6eSJoseph Koshy  * Initialize module.
204749874f6eSJoseph Koshy  */
204849874f6eSJoseph Koshy 
204915139246SJoseph Koshy void
20500b86b1bbSFabien Thomas pmcstat_initialize_logging(void)
205115139246SJoseph Koshy {
205260f918efSJoseph Koshy 	int i;
205349874f6eSJoseph Koshy 
205415139246SJoseph Koshy 	/* use a convenient format for 'ldd' output */
2055f3c8200eSJoseph Koshy 	if (setenv("LD_TRACE_LOADED_OBJECTS_FMT1","%o \"%p\" %x\n",1) != 0)
205649874f6eSJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot setenv");
205715139246SJoseph Koshy 
205815139246SJoseph Koshy 	/* Initialize hash tables */
205949874f6eSJoseph Koshy 	pmcstat_string_initialize();
206015139246SJoseph Koshy 	for (i = 0; i < PMCSTAT_NHASH; i++) {
206115139246SJoseph Koshy 		LIST_INIT(&pmcstat_image_hash[i]);
206215139246SJoseph Koshy 		LIST_INIT(&pmcstat_process_hash[i]);
206315139246SJoseph Koshy 	}
206415139246SJoseph Koshy 
206549874f6eSJoseph Koshy 	/*
206649874f6eSJoseph Koshy 	 * Create a fake 'process' entry for the kernel with pid -1.
206749874f6eSJoseph Koshy 	 * hwpmc(4) will subsequently inform us about where the kernel
206849874f6eSJoseph Koshy 	 * and any loaded kernel modules are mapped.
206949874f6eSJoseph Koshy 	 */
207049874f6eSJoseph Koshy 	if ((pmcstat_kernproc = pmcstat_process_lookup((pid_t) -1,
207149874f6eSJoseph Koshy 		 PMCSTAT_ALLOCATE)) == NULL)
207215139246SJoseph Koshy 		err(EX_OSERR, "ERROR: Cannot initialize logging");
20730b86b1bbSFabien Thomas 
20740b86b1bbSFabien Thomas 	/* PMC count. */
20750b86b1bbSFabien Thomas 	pmcstat_npmcs = 0;
20760b86b1bbSFabien Thomas 
20770b86b1bbSFabien Thomas 	/* Merge PMC with same name. */
20780b86b1bbSFabien Thomas 	pmcstat_mergepmc = args.pa_mergepmc;
20790b86b1bbSFabien Thomas 
20800b86b1bbSFabien Thomas 	/*
20810b86b1bbSFabien Thomas 	 * Initialize plugins
20820b86b1bbSFabien Thomas 	 */
20830b86b1bbSFabien Thomas 
20840b86b1bbSFabien Thomas 	if (plugins[args.pa_pplugin].pl_init != NULL)
20850b86b1bbSFabien Thomas 		plugins[args.pa_pplugin].pl_init();
20860b86b1bbSFabien Thomas 	if (plugins[args.pa_plugin].pl_init != NULL)
20870b86b1bbSFabien Thomas 		plugins[args.pa_plugin].pl_init();
208815139246SJoseph Koshy }
208915139246SJoseph Koshy 
209049874f6eSJoseph Koshy /*
209149874f6eSJoseph Koshy  * Shutdown module.
209249874f6eSJoseph Koshy  */
209349874f6eSJoseph Koshy 
209415139246SJoseph Koshy void
20950b86b1bbSFabien Thomas pmcstat_shutdown_logging(void)
209615139246SJoseph Koshy {
209715139246SJoseph Koshy 	int i;
209849874f6eSJoseph Koshy 	FILE *mf;
209915139246SJoseph Koshy 	struct pmcstat_image *pi, *pitmp;
210015139246SJoseph Koshy 	struct pmcstat_process *pp, *pptmp;
21010b86b1bbSFabien Thomas 	struct pmcstat_pcmap *ppm, *ppmtmp;
210249874f6eSJoseph Koshy 
210349874f6eSJoseph Koshy 	/* determine where to send the map file */
210449874f6eSJoseph Koshy 	mf = NULL;
21050b86b1bbSFabien Thomas 	if (args.pa_mapfilename != NULL)
21060b86b1bbSFabien Thomas 		mf = (strcmp(args.pa_mapfilename, "-") == 0) ?
21070b86b1bbSFabien Thomas 		    args.pa_printfile : fopen(args.pa_mapfilename, "w");
210849874f6eSJoseph Koshy 
21090b86b1bbSFabien Thomas 	if (mf == NULL && args.pa_flags & FLAG_DO_GPROF &&
21100b86b1bbSFabien Thomas 	    args.pa_verbosity >= 2)
21110b86b1bbSFabien Thomas 		mf = args.pa_printfile;
211249874f6eSJoseph Koshy 
211349874f6eSJoseph Koshy 	if (mf)
211449874f6eSJoseph Koshy 		(void) fprintf(mf, "MAP:\n");
211515139246SJoseph Koshy 
2116b6010f9eSJoseph Koshy 	/*
21170b86b1bbSFabien Thomas 	 * Shutdown the plugins
2118b6010f9eSJoseph Koshy 	 */
21192a6d2e9cSJoseph Koshy 
21200b86b1bbSFabien Thomas 	if (plugins[args.pa_plugin].pl_shutdown != NULL)
21210b86b1bbSFabien Thomas 		plugins[args.pa_plugin].pl_shutdown(mf);
21220b86b1bbSFabien Thomas 	if (plugins[args.pa_pplugin].pl_shutdown != NULL)
21230b86b1bbSFabien Thomas 		plugins[args.pa_pplugin].pl_shutdown(mf);
2124b6010f9eSJoseph Koshy 
2125b6010f9eSJoseph Koshy 	for (i = 0; i < PMCSTAT_NHASH; i++) {
21260b86b1bbSFabien Thomas 		LIST_FOREACH_SAFE(pi, &pmcstat_image_hash[i], pi_next,
21270b86b1bbSFabien Thomas 		    pitmp) {
21280b86b1bbSFabien Thomas 			if (plugins[args.pa_plugin].pl_shutdownimage != NULL)
21290b86b1bbSFabien Thomas 				plugins[args.pa_plugin].pl_shutdownimage(pi);
21300b86b1bbSFabien Thomas 			if (plugins[args.pa_pplugin].pl_shutdownimage != NULL)
21310b86b1bbSFabien Thomas 				plugins[args.pa_pplugin].pl_shutdownimage(pi);
21320b86b1bbSFabien Thomas 
2133b6010f9eSJoseph Koshy 			free(pi->pi_symbols);
21340b86b1bbSFabien Thomas 			if (pi->pi_addr2line != NULL)
21350b86b1bbSFabien Thomas 				pclose(pi->pi_addr2line);
213615139246SJoseph Koshy 			LIST_REMOVE(pi, pi_next);
213715139246SJoseph Koshy 			free(pi);
213815139246SJoseph Koshy 		}
2139b6010f9eSJoseph Koshy 
214015139246SJoseph Koshy 		LIST_FOREACH_SAFE(pp, &pmcstat_process_hash[i], pp_next,
214115139246SJoseph Koshy 		    pptmp) {
21420b86b1bbSFabien Thomas 			TAILQ_FOREACH_SAFE(ppm, &pp->pp_map, ppm_next, ppmtmp) {
21430b86b1bbSFabien Thomas 				TAILQ_REMOVE(&pp->pp_map, ppm, ppm_next);
21440b86b1bbSFabien Thomas 				free(ppm);
21450b86b1bbSFabien Thomas 			}
214615139246SJoseph Koshy 			LIST_REMOVE(pp, pp_next);
214715139246SJoseph Koshy 			free(pp);
214815139246SJoseph Koshy 		}
214915139246SJoseph Koshy 	}
215049874f6eSJoseph Koshy 
215149874f6eSJoseph Koshy 	pmcstat_string_shutdown();
215249874f6eSJoseph Koshy 
215349874f6eSJoseph Koshy 	/*
215449874f6eSJoseph Koshy 	 * Print errors unless -q was specified.  Print all statistics
215549874f6eSJoseph Koshy 	 * if verbosity > 1.
215649874f6eSJoseph Koshy 	 */
21570b86b1bbSFabien Thomas #define	PRINT(N,V) do {							\
21580b86b1bbSFabien Thomas 		if (pmcstat_stats.ps_##V || args.pa_verbosity >= 2)	\
21590b86b1bbSFabien Thomas 			(void) fprintf(args.pa_printfile, " %-40s %d\n",\
216049874f6eSJoseph Koshy 			    N, pmcstat_stats.ps_##V);			\
216149874f6eSJoseph Koshy 	} while (0)
216249874f6eSJoseph Koshy 
2163ac708fddSFabien Thomas 	if (args.pa_verbosity >= 1 && (args.pa_flags & FLAG_DO_ANALYSIS)) {
21640b86b1bbSFabien Thomas 		(void) fprintf(args.pa_printfile, "CONVERSION STATISTICS:\n");
21650b86b1bbSFabien Thomas 		PRINT("#exec/a.out", exec_aout);
21660b86b1bbSFabien Thomas 		PRINT("#exec/elf", exec_elf);
21670b86b1bbSFabien Thomas 		PRINT("#exec/unknown", exec_indeterminable);
21680b86b1bbSFabien Thomas 		PRINT("#exec handling errors", exec_errors);
21690b86b1bbSFabien Thomas 		PRINT("#samples/total", samples_total);
21700b86b1bbSFabien Thomas 		PRINT("#samples/unclaimed", samples_unknown_offset);
21710b86b1bbSFabien Thomas 		PRINT("#samples/unknown-object", samples_indeterminable);
2172eb707d60SFabien Thomas 		PRINT("#samples/unknown-function", samples_unknown_function);
21730b86b1bbSFabien Thomas 		PRINT("#callchain/dubious-frames", callchain_dubious_frames);
217415139246SJoseph Koshy 	}
217549874f6eSJoseph Koshy 
217649874f6eSJoseph Koshy 	if (mf)
217749874f6eSJoseph Koshy 		(void) fclose(mf);
217815139246SJoseph Koshy }
2179