1 files changed, 117 insertions, 0 deletions
diff --git a/usr.sbin/pmc/pmc.c b/usr.sbin/pmc/pmc.c
new file mode 100644
index 000000000000..4b4d113e2bd4
--- /dev/null
+++ b/usr.sbin/pmc/pmc.c
@@ -0,0 +1,117 @@
+/*-
+ * SPDX-License-Identifier: BSD-2-Clause
+ *
+ * Copyright (c) 2018, Matthew Macy
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ *
+ */
+
+#include <sys/types.h>
+#include <sys/errno.h>
+#include <sys/sysctl.h>
+#include <stddef.h>
+#include <stdlib.h>
+#include <err.h>
+#include <limits.h>
+#include <string.h>
+#include <pmc.h>
+#include <pmclog.h>
+#include <libpmcstat.h>
+#include <sysexits.h>
+#include <unistd.h>
+
+#include <libpmcstat.h>
+#include "cmd_pmc.h"
+
+int	pmc_displayheight = DEFAULT_DISPLAY_HEIGHT;
+int	pmc_displaywidth = DEFAULT_DISPLAY_WIDTH;
+int	pmc_kq;
+struct pmcstat_args pmc_args;
+
+struct pmcstat_pmcs pmcstat_pmcs = LIST_HEAD_INITIALIZER(pmcstat_pmcs);
+
+struct pmcstat_image_hash_list pmcstat_image_hash[PMCSTAT_NHASH];
+
+struct pmcstat_process_hash_list pmcstat_process_hash[PMCSTAT_NHASH];
+
+struct cmd_handler {
+	const char *ch_name;
+	cmd_disp_t ch_fn;
+};
+
+static struct cmd_handler disp_table[] = {
+	{"stat", cmd_pmc_stat},
+	{"stat-system", cmd_pmc_stat_system},
+	{"list-events", cmd_pmc_list_events},
+	{"filter", cmd_pmc_filter},
+	{"summary", cmd_pmc_summary},
+	{NULL, NULL}
+};
+
+static void __dead2
+usage(void)
+{
+	errx(EX_USAGE,
+	    "\t pmc management utility\n"
+	    "\t stat <program> run program and print stats\n"
+		 "\t stat-system <program> run program and print system wide stats for duration of execution\n"
+		 "\t list-events list PMC events available on host\n"
+		 "\t filter filter records by lwp, pid, or event\n"
+	    );
+}
+
+static cmd_disp_t
+disp_lookup(char *name)
+{
+	struct cmd_handler *hnd;
+
+	for (hnd = disp_table; hnd->ch_name != NULL; hnd++)
+		if (strcmp(hnd->ch_name, name) == 0)
+			return (hnd->ch_fn);
+	return (NULL);
+}
+
+int
+main(int argc, char **argv)
+{
+	cmd_disp_t disp;
+
+	pmc_args.pa_printfile = stderr;
+	STAILQ_INIT(&pmc_args.pa_events);
+	SLIST_INIT(&pmc_args.pa_targets);
+	if (argc == 1)
+		usage();
+	if ((disp = disp_lookup(argv[1])) == NULL)
+		usage();
+	argc--;
+	argv++;
+
+	/* Allocate a kqueue */
+	if ((pmc_kq = kqueue()) < 0)
+		err(EX_OSERR, "ERROR: Cannot allocate kqueue");
+	if (pmc_init() < 0)
+		err(EX_UNAVAILABLE,
+		    "ERROR: Initialization of the pmc(3) library failed"
+		    );
+	return (disp(argc, argv));
+}