1/*-
2 * Copyright (c) 2005-2007, Joseph Koshy
3 * Copyright (c) 2007 The FreeBSD Foundation
4 * All rights reserved.
5 *
6 * Portions of this software were developed by A. Joseph Koshy under
7 * sponsorship from the FreeBSD Foundation and Google, Inc.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
11 * are met:
12 * 1. Redistributions of source code must retain the above copyright
13 *    notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 *    notice, this list of conditions and the following disclaimer in the
16 *    documentation and/or other materials provided with the distribution.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
19 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
22 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
24 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28 * SUCH DAMAGE.
29 */
30
31#ifndef	_LIBPMCSTAT_H_
32#define	_LIBPMCSTAT_H_
33
34#include <sys/_cpuset.h>
35#include <sys/queue.h>
36
37#include <stdio.h>
38#include <gelf.h>
39
40#define	PMCSTAT_ALLOCATE		1
41
42#define	NSOCKPAIRFD			2
43#define	PARENTSOCKET			0
44#define	CHILDSOCKET			1
45
46#define	PMCSTAT_OPEN_FOR_READ		0
47#define	PMCSTAT_OPEN_FOR_WRITE		1
48#define	READPIPEFD			0
49#define	WRITEPIPEFD			1
50#define	NPIPEFD				2
51
52#define	PMCSTAT_NHASH			256
53#define	PMCSTAT_HASH_MASK		0xFF
54#define	DEFAULT_SAMPLE_COUNT		65536
55
56typedef const void *pmcstat_interned_string;
57struct pmc_plugins;
58
59enum pmcstat_state {
60	PMCSTAT_FINISHED = 0,
61	PMCSTAT_EXITING  = 1,
62	PMCSTAT_RUNNING  = 2
63};
64
65struct pmcstat_ev {
66	STAILQ_ENTRY(pmcstat_ev) ev_next;
67	int		ev_count; /* associated count if in sampling mode */
68	uint32_t	ev_cpu;	  /* cpus for this event */
69	int		ev_cumulative;  /* show cumulative counts */
70	int		ev_flags; /* PMC_F_* */
71	int		ev_fieldskip;   /* #leading spaces */
72	int		ev_fieldwidth;  /* print width */
73	enum pmc_mode	ev_mode;  /* desired mode */
74	char	       *ev_name;  /* (derived) event name */
75	pmc_id_t	ev_pmcid; /* allocated ID */
76	pmc_value_t	ev_saved; /* for incremental counts */
77	char	       *ev_spec;  /* event specification */
78};
79
80struct pmcstat_target {
81	SLIST_ENTRY(pmcstat_target) pt_next;
82	pid_t		pt_pid;
83};
84
85struct pmcstat_args {
86	int	pa_flags;		/* argument flags */
87#define	FLAG_HAS_TARGET			0x00000001	/* process target */
88#define	FLAG_HAS_WAIT_INTERVAL		0x00000002	/* -w secs */
89#define	FLAG_HAS_OUTPUT_LOGFILE		0x00000004	/* -O file or pipe */
90#define	FLAG_HAS_COMMANDLINE		0x00000008	/* command */
91#define	FLAG_HAS_SAMPLING_PMCS		0x00000010	/* -S or -P */
92#define	FLAG_HAS_COUNTING_PMCS		0x00000020	/* -s or -p */
93#define	FLAG_HAS_PROCESS_PMCS		0x00000040	/* -P or -p */
94#define	FLAG_HAS_SYSTEM_PMCS		0x00000080	/* -S or -s */
95#define	FLAG_HAS_PIPE			0x00000100	/* implicit log */
96#define	FLAG_READ_LOGFILE		0x00000200	/* -R file */
97#define	FLAG_DO_GPROF			0x00000400	/* -g */
98#define	FLAG_HAS_SAMPLESDIR		0x00000800	/* -D dir */
99/* was FLAG_HAS_KERNELPATH		0x00001000 */
100#define	FLAG_DO_PRINT			0x00002000	/* -o */
101#define	FLAG_DO_CALLGRAPHS		0x00004000	/* -G or -F */
102#define	FLAG_DO_ANNOTATE		0x00008000	/* -m */
103#define	FLAG_DO_TOP			0x00010000	/* -T */
104#define	FLAG_DO_ANALYSIS		0x00020000	/* -g or -G or -m or -T */
105#define	FLAGS_HAS_CPUMASK		0x00040000	/* -c */
106#define	FLAG_HAS_DURATION		0x00080000	/* -l secs */
107#define	FLAG_DO_WIDE_GPROF_HC		0x00100000	/* -e */
108#define	FLAG_SKIP_TOP_FN_RES		0x00200000	/* -A */
109#define	FLAG_FILTER_THREAD_ID		0x00400000	/* -L */
110#define	FLAG_SHOW_OFFSET		0x00800000	/* -I */
111
112	int	pa_required;		/* required features */
113	int	pa_pplugin;		/* pre-processing plugin */
114	int	pa_plugin;		/* analysis plugin */
115	int	pa_verbosity;		/* verbosity level */
116	FILE	*pa_printfile;		/* where to send printed output */
117	int	pa_logfd;		/* output log file */
118	char	*pa_inputpath;		/* path to input log */
119	char	*pa_outputpath;		/* path to output log */
120	void	*pa_logparser;		/* log file parser */
121	const char	*pa_fsroot;	/* FS root where executables reside */
122	const char	*pa_samplesdir;	/* directory for profile files */
123	const char	*pa_mapfilename;/* mapfile name */
124	FILE	*pa_graphfile;		/* where to send the callgraph */
125	int	pa_graphdepth;		/* print depth for callgraphs */
126	double	pa_interval;		/* printing interval in seconds */
127	cpuset_t	pa_cpumask;	/* filter for CPUs analysed */
128	int	pa_ctdumpinstr;		/* dump instructions with calltree */
129	int	pa_topmode;		/* delta or accumulative */
130	int	pa_toptty;		/* output to tty or file */
131	int	pa_topcolor;		/* terminal support color */
132	int	pa_mergepmc;		/* merge PMC with same name */
133	double	pa_duration;		/* time duration */
134	uint32_t pa_tid;
135	int	pa_argc;
136	char	**pa_argv;
137	STAILQ_HEAD(, pmcstat_ev) pa_events;
138	SLIST_HEAD(, pmcstat_target) pa_targets;
139};
140
141/*
142 * Each function symbol tracked by pmcstat(8).
143 */
144
145struct pmcstat_symbol {
146	pmcstat_interned_string ps_name;
147	uint64_t	ps_start;
148	uint64_t	ps_end;
149};
150
151/*
152 * A 'pmcstat_image' structure describes an executable program on
153 * disk.  'pi_execpath' is a cookie representing the pathname of
154 * the executable.  'pi_start' and 'pi_end' are the least and greatest
155 * virtual addresses for the text segments in the executable.
156 * 'pi_gmonlist' contains a linked list of gmon.out files associated
157 * with this image.
158 */
159
160enum pmcstat_image_type {
161	PMCSTAT_IMAGE_UNKNOWN = 0,	/* never looked at the image */
162	PMCSTAT_IMAGE_INDETERMINABLE,	/* can't tell what the image is */
163	PMCSTAT_IMAGE_ELF32,		/* ELF 32 bit object */
164	PMCSTAT_IMAGE_ELF64,		/* ELF 64 bit object */
165	PMCSTAT_IMAGE_AOUT		/* AOUT object */
166};
167
168struct pmcstat_image {
169	LIST_ENTRY(pmcstat_image) pi_next;	/* hash link */
170	pmcstat_interned_string	pi_execpath;    /* cookie */
171	pmcstat_interned_string pi_samplename;  /* sample path name */
172	pmcstat_interned_string pi_fullpath;    /* path to FS object */
173	pmcstat_interned_string pi_name;	/* display name */
174
175	enum pmcstat_image_type pi_type;	/* executable type */
176
177	/*
178	 * Executables have pi_start and pi_end; these are zero
179	 * for shared libraries.
180	 */
181	uintfptr_t	pi_start;	/* start address (inclusive) */
182	uintfptr_t	pi_end;		/* end address (exclusive) */
183	uintfptr_t	pi_entry;	/* entry address */
184	uintfptr_t	pi_vaddr;	/* virtual address where loaded */
185	int		pi_isdynamic;	/* whether a dynamic object */
186	int		pi_iskernelmodule;
187	pmcstat_interned_string pi_dynlinkerpath; /* path in .interp */
188
189	/* All symbols associated with this object. */
190	struct pmcstat_symbol *pi_symbols;
191	size_t		pi_symcount;
192
193	/* Handle to addr2line for this image. */
194	FILE *pi_addr2line;
195
196	/*
197	 * Plugins private data
198	 */
199
200	/* gprof:
201	 * An image can be associated with one or more gmon.out files;
202	 * one per PMC.
203	 */
204	LIST_HEAD(,pmcstat_gmonfile) pi_gmlist;
205};
206
207extern LIST_HEAD(pmcstat_image_hash_list, pmcstat_image) pmcstat_image_hash[PMCSTAT_NHASH];
208
209/*
210 * A simple implementation of interned strings.  Each interned string
211 * is assigned a unique address, so that subsequent string compares
212 * can be done by a simple pointer comparison instead of using
213 * strcmp().  This speeds up hash table lookups and saves memory if
214 * duplicate strings are the norm.
215 */
216struct pmcstat_string {
217	LIST_ENTRY(pmcstat_string)	ps_next;	/* hash link */
218	int		ps_len;
219	int		ps_hash;
220	char		*ps_string;
221};
222
223/*
224 * A 'pmcstat_pcmap' structure maps a virtual address range to an
225 * underlying 'pmcstat_image' descriptor.
226 */
227struct pmcstat_pcmap {
228	TAILQ_ENTRY(pmcstat_pcmap) ppm_next;
229	uintfptr_t	ppm_lowpc;
230	uintfptr_t	ppm_highpc;
231	struct pmcstat_image *ppm_image;
232};
233
234/*
235 * A 'pmcstat_process' structure models processes.  Each process is
236 * associated with a set of pmcstat_pcmap structures that map
237 * addresses inside it to executable objects.  This set is implemented
238 * as a list, kept sorted in ascending order of mapped addresses.
239 *
240 * 'pp_pid' holds the pid of the process.  When a process exits, the
241 * 'pp_isactive' field is set to zero, but the process structure is
242 * not immediately reclaimed because there may still be samples in the
243 * log for this process.
244 */
245
246struct pmcstat_process {
247	LIST_ENTRY(pmcstat_process) pp_next;	/* hash-next */
248	pid_t			pp_pid;		/* associated pid */
249	int			pp_isactive;	/* whether active */
250	uintfptr_t		pp_entryaddr;	/* entry address */
251	TAILQ_HEAD(,pmcstat_pcmap) pp_map;	/* address range map */
252};
253extern LIST_HEAD(pmcstat_process_hash_list, pmcstat_process) pmcstat_process_hash[PMCSTAT_NHASH];
254
255/*
256 * 'pmcstat_pmcrecord' is a mapping from PMC ids to human-readable
257 * names.
258 */
259
260struct pmcstat_pmcrecord {
261	LIST_ENTRY(pmcstat_pmcrecord)	pr_next;
262	pmc_id_t			pr_pmcid;
263	int				pr_pmcin;
264	pmcstat_interned_string		pr_pmcname;
265	int				pr_samples;
266	int				pr_dubious_frames;
267	struct pmcstat_pmcrecord	*pr_merge;
268};
269extern LIST_HEAD(pmcstat_pmcs, pmcstat_pmcrecord) pmcstat_pmcs; /* PMC list */
270
271struct pmc_plugins {
272	const char *pl_name;
273
274	/* configure */
275	int (*pl_configure)(char *opt);
276
277	/* init and shutdown */
278	int (*pl_init)(void);
279	void (*pl_shutdown)(FILE *mf);
280
281	/* sample processing */
282	void (*pl_process)(struct pmcstat_process *pp,
283	    struct pmcstat_pmcrecord *pmcr, uint32_t nsamples,
284	    uintfptr_t *cc, int usermode, uint32_t cpu);
285
286	/* image */
287	void (*pl_initimage)(struct pmcstat_image *pi);
288	void (*pl_shutdownimage)(struct pmcstat_image *pi);
289
290	/* pmc */
291	void (*pl_newpmc)(pmcstat_interned_string ps,
292		struct pmcstat_pmcrecord *pr);
293
294	/* top display */
295	void (*pl_topdisplay)(void);
296
297	/* top keypress */
298	int (*pl_topkeypress)(int c, void *w);
299};
300
301/*
302 * Misc. statistics
303 */
304struct pmcstat_stats {
305	int ps_exec_aout;	/* # a.out executables seen */
306	int ps_exec_elf;	/* # elf executables seen */
307	int ps_exec_errors;	/* # errors processing executables */
308	int ps_exec_indeterminable; /* # unknown executables seen */
309	int ps_samples_total;	/* total number of samples processed */
310	int ps_samples_skipped; /* #samples filtered out for any reason */
311	int ps_samples_unknown_offset;	/* #samples of rank 0 not in a map */
312	int ps_samples_indeterminable;	/* #samples in indeterminable images */
313	int ps_samples_unknown_function;/* #samples with unknown function at offset */
314	int ps_callchain_dubious_frames;/* #dubious frame pointers seen */
315};
316
317__BEGIN_DECLS
318int pmcstat_symbol_compare(const void *a, const void *b);
319struct pmcstat_symbol *pmcstat_symbol_search(struct pmcstat_image *image,
320    uintfptr_t addr);
321void pmcstat_image_add_symbols(struct pmcstat_image *image, Elf *e,
322    Elf_Scn *scn, GElf_Shdr *sh);
323
324const char *pmcstat_string_unintern(pmcstat_interned_string _is);
325pmcstat_interned_string pmcstat_string_intern(const char *_s);
326int pmcstat_string_compute_hash(const char *s);
327pmcstat_interned_string pmcstat_string_lookup(const char *_s);
328void pmcstat_image_get_elf_params(struct pmcstat_image *image, struct pmcstat_args *args);
329
330struct pmcstat_image *
331    pmcstat_image_from_path(pmcstat_interned_string internedpath,
332    int iskernelmodule, struct pmcstat_args *args,
333    struct pmc_plugins *plugins);
334int pmcstat_string_lookup_hash(pmcstat_interned_string _is);
335
336void pmcstat_process_elf_exec(struct pmcstat_process *_pp,
337    struct pmcstat_image *_image, uintptr_t _baseaddr, uintptr_t _dynaddr,
338    struct pmcstat_args *args, struct pmc_plugins *plugins,
339    struct pmcstat_stats *pmcstat_stats);
340
341void pmcstat_image_link(struct pmcstat_process *_pp,
342    struct pmcstat_image *_i, uintfptr_t _lpc);
343
344void pmcstat_process_aout_exec(struct pmcstat_process *_pp,
345    struct pmcstat_image *_image, uintptr_t _baseaddr);
346void pmcstat_process_exec(struct pmcstat_process *_pp,
347    pmcstat_interned_string _path, uintptr_t _baseaddr, uintptr_t _dynaddr,
348    struct pmcstat_args *args, struct pmc_plugins *plugins,
349    struct pmcstat_stats *pmcstat_stats);
350void pmcstat_image_determine_type(struct pmcstat_image *_image, struct pmcstat_args *args);
351void pmcstat_image_get_aout_params(struct pmcstat_image *_image, struct pmcstat_args *args);
352struct pmcstat_pcmap *pmcstat_process_find_map(struct pmcstat_process *_p,
353    uintfptr_t _pc);
354void pmcstat_initialize_logging(struct pmcstat_process **pmcstat_kernproc,
355    struct pmcstat_args *args, struct pmc_plugins *plugins,
356    int *pmcstat_npmcs, int *pmcstat_mergepmc);
357void pmcstat_shutdown_logging(struct pmcstat_args *args,
358    struct pmc_plugins *plugins,
359    struct pmcstat_stats *pmcstat_stats);
360struct pmcstat_process *pmcstat_process_lookup(pid_t _pid, int _allocate);
361void pmcstat_clone_event_descriptor(struct pmcstat_ev *ev, const cpuset_t *cpumask, struct pmcstat_args *args);
362
363void pmcstat_create_process(int *pmcstat_sockpair, struct pmcstat_args *args, int pmcstat_kq);
364void pmcstat_start_process(int *pmcstat_sockpair);
365
366void pmcstat_attach_pmcs(struct pmcstat_args *args);
367struct pmcstat_symbol *pmcstat_symbol_search_by_name(struct pmcstat_process *pp, const char *pi_name, const char *name, uintptr_t *, uintptr_t *);
368
369void pmcstat_string_initialize(void);
370void pmcstat_string_shutdown(void);
371
372int pmcstat_analyze_log(struct pmcstat_args *args,
373    struct pmc_plugins *plugins,
374    struct pmcstat_stats *pmcstat_stats,
375    struct pmcstat_process *pmcstat_kernproc,
376    int pmcstat_mergepmc,
377    int *pmcstat_npmcs,
378    int *ps_samples_period);
379
380int pmcstat_open_log(const char *_p, int _mode);
381int pmcstat_close_log(struct pmcstat_args *args);
382
383__END_DECLS
384
385#endif /* !_LIBPMCSTAT_H_ */
386