1/*-
2 * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
3 *
4 * Copyright (c) 2018, Matthew Macy
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 *    notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 *    notice, this list of conditions and the following disclaimer in the
13 *    documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 * SUCH DAMAGE.
26 *
27 * $FreeBSD$
28 *
29 */
30
31#include <sys/types.h>
32#include <sys/errno.h>
33#include <sys/sysctl.h>
34#include <stddef.h>
35#include <stdlib.h>
36#include <err.h>
37#include <limits.h>
38#include <string.h>
39#include <pmc.h>
40#include <pmclog.h>
41#include <libpmcstat.h>
42#include <sysexits.h>
43#include <unistd.h>
44
45#include <libpmcstat.h>
46#include "cmd_pmc.h"
47
48int	pmc_displayheight = DEFAULT_DISPLAY_HEIGHT;
49int	pmc_displaywidth = DEFAULT_DISPLAY_WIDTH;
50int	pmc_kq;
51struct pmcstat_args pmc_args;
52
53struct pmcstat_pmcs pmcstat_pmcs = LIST_HEAD_INITIALIZER(pmcstat_pmcs);
54
55struct pmcstat_image_hash_list pmcstat_image_hash[PMCSTAT_NHASH];
56
57struct pmcstat_process_hash_list pmcstat_process_hash[PMCSTAT_NHASH];
58
59struct cmd_handler {
60	const char *ch_name;
61	cmd_disp_t ch_fn;
62};
63
64static struct cmd_handler disp_table[] = {
65	{"stat", cmd_pmc_stat},
66	{"stat-system", cmd_pmc_stat_system},
67	{"list-events", cmd_pmc_list_events},
68	{"filter", cmd_pmc_filter},
69	{"summary", cmd_pmc_summary},
70	{NULL, NULL}
71};
72
73static void __dead2
74usage(void)
75{
76	errx(EX_USAGE,
77	    "\t pmc management utility\n"
78	    "\t stat <program> run program and print stats\n"
79		 "\t stat-system <program> run program and print system wide stats for duration of execution\n"
80		 "\t list-events list PMC events available on host\n"
81		 "\t filter filter records by lwp, pid, or event\n"
82	    );
83}
84
85static cmd_disp_t
86disp_lookup(char *name)
87{
88	struct cmd_handler *hnd;
89
90	for (hnd = disp_table; hnd->ch_name != NULL; hnd++)
91		if (strcmp(hnd->ch_name, name) == 0)
92			return (hnd->ch_fn);
93	return (NULL);
94}
95
96int
97main(int argc, char **argv)
98{
99	cmd_disp_t disp;
100
101	pmc_args.pa_printfile = stderr;
102	STAILQ_INIT(&pmc_args.pa_events);
103	SLIST_INIT(&pmc_args.pa_targets);
104	if (argc == 1)
105		usage();
106	if ((disp = disp_lookup(argv[1])) == NULL)
107		usage();
108	argc--;
109	argv++;
110
111	/* Allocate a kqueue */
112	if ((pmc_kq = kqueue()) < 0)
113		err(EX_OSERR, "ERROR: Cannot allocate kqueue");
114	if (pmc_init() < 0)
115		err(EX_UNAVAILABLE,
116		    "ERROR: Initialization of the pmc(3) library failed"
117		    );
118	return (disp(argc, argv));
119}
120