/*-
 * 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 <libelf.h>
#include <libpmcstat.h>
#include <sysexits.h>
#include <unistd.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;
	const char *ch_desc;
};

static struct cmd_handler disp_table[] = {
	{ "filter", cmd_pmc_filter, NULL },
	{ "frontend", cmd_pmc_frontend, "Analyze front-end stalls" },
	{ "info", cmd_pmc_info, "Display high level information about a log" },
	{ "list-events", cmd_pmc_list_events, "List available PMC events" },
	{ "record", cmd_pmc_record, "Record a performance sample" },
	{ "stat", cmd_pmc_stat, NULL },
	{ "stat-system", cmd_pmc_stat_system, NULL },
	{ "summary", cmd_pmc_summary, NULL },
	{ NULL, NULL, NULL }
};

static void
usage(void)
{
	int i;

	fprintf(stderr, "Usage: pmc <COMMAND> [OPTIONS] ...\n\n");
	fprintf(stderr, "PMC tool\n\n");

	fprintf(stderr, "Commands:\n");
	for (i = 0; disp_table[i].ch_name != NULL; i++) {
		// Hide stuff we plan to deprecate
		if (disp_table[i].ch_desc)
			fprintf(stderr, "    %-12s    %s\n",
			    disp_table[i].ch_name, disp_table[i].ch_desc);
	}

	fprintf(stderr, "\nEnvironment Variables:\n");
	fprintf(stderr, "    CLICOLOR        Enables color graphs\n");
	fprintf(stderr, "    SYSROOT         Path to system root\n");
	fprintf(stderr, "    SYMROOT         Path to debug symbols\n");
	fprintf(stderr, "    TERM            Controls symbols and color palettes\n");

	exit(EX_USAGE);
}

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"
		    );

	if (elf_version(EV_CURRENT) == EV_NONE)
		err(EX_UNAVAILABLE, "ERROR: ELF library version too old");

	return (disp(argc, argv));
}