/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#include <sys/types.h>
#include <sys/kstat.h>
#include "n2piupc_acc.h"
#include "n2piupc_tables.h"
#include "n2piupc.h"
#include "n2piupc_biterr.h"

#define	PIC_STR_LEN	5	/* Size of a PICx name string. */

static int n2piupc_create_name_kstat(n2piu_grp_t *grp);
static void n2piupc_delete_name_kstats(kstat_t **name_kstats_pp,
    int num_kstats);
static kstat_t *n2piupc_create_cntr_kstat(char *name, int dev_inst,
    int (*update)(kstat_t *, int), n2piu_ksinfo_t *ksinfop, int num_pics);
static int n2piupc_kstat_update(kstat_t *ksp, int rw);
static kstat_t *n2piupc_create_picN_kstat(char *mod_name, int pic,
    uint64_t mask, int num_ev, n2piu_event_t *ev_array);
static int n2piupc_write(n2piupc_t *n2piupc_p, int regid, uint64_t data);

/*
 * One-time initialization for this module.
 */
int
n2piupc_kstat_init()
{
	n2piu_grp_t **grp_pp;
	n2piu_grp_t *grp_p;

	N2PIUPC_DBG2("n2piupc: kstat_init: enter\n");

	/*
	 * Initialize the name kstats for each group, drawing upon the table
	 * for values.
	 */
	for (grp_pp = leaf_grps; *grp_pp != NULL; grp_pp++) {

		grp_p = *grp_pp;

		N2PIUPC_DBG2("Setting up group for %s\n", grp_p->grp_name);

		/* Create basic pic event-type pair. */
		grp_p->name_kstats_pp = kmem_zalloc((grp_p->num_counters *
		    sizeof (kstat_t)), KM_SLEEP);
		if (n2piupc_create_name_kstat(grp_p) != DDI_SUCCESS) {
			n2piupc_kstat_fini();
			N2PIUPC_DBG1("n2piupc: init: failure exit\n");
			return (DDI_FAILURE);
		}
	}

	N2PIUPC_DBG2("n2piupc: kstat_init: success exit\n");

	return (DDI_SUCCESS);
}

/*
 * Per-instance initialization for this module.
 */
int
n2piupc_kstat_attach(n2piupc_t *n2piupc_p)
{
	n2piu_grp_t **grp_pp;
	n2piu_grp_t *grp_p;
	n2piu_ksinfo_t *ksinfo_p;

	int i;

	N2PIUPC_DBG2("n2piupc: kstat_attach %d: enter\n",
	    ddi_get_instance(n2piupc_p->n2piupc_dip));

	/* Initialize biterr module.  Save opaque result. */
	if (n2piupc_biterr_attach(&n2piupc_p->n2piupc_biterr_p) != DDI_SUCCESS)
		goto err;

	/* Set up kstats for each group. */
	for (i = 0, grp_pp = leaf_grps; *grp_pp != NULL; i++, grp_pp++) {

		grp_p = *grp_pp;

		/*
		 * ksinfo_p keeps all info needed by n2piupc_kstat_update,
		 * which is fired off asynchronously on demand by the kstat
		 * framework.
		 */
		ksinfo_p = (n2piu_ksinfo_t *)kmem_zalloc(
		    sizeof (n2piu_ksinfo_t), KM_SLEEP);

		ksinfo_p->n2piupc_p = n2piupc_p;
		ksinfo_p->grp_p  = grp_p;

		/* Also save in state structure, for later cleanup. */
		n2piupc_p->n2piupc_ksinfo_p[i] = ksinfo_p;

		/* Create counter kstats */
		ksinfo_p->cntr_ksp = n2piupc_create_cntr_kstat(grp_p->grp_name,
		    ddi_get_instance(n2piupc_p->n2piupc_dip),
		    n2piupc_kstat_update, ksinfo_p, grp_p->num_counters);
		if (ksinfo_p->cntr_ksp == NULL)
			goto err;
	}

	/*
	 * Special treatment for bit err registers: enable them so they start
	 * counting now.
	 */
	if (n2piupc_write(n2piupc_p, leaf_grps[BIT_ERR_GRP]->regsel_p->regoff,
	    BTERR_CTR_ENABLE) != SUCCESS) {
		goto err;
	}

	N2PIUPC_DBG2("n2piupc: kstat_attach: success exit\n");
	return (DDI_SUCCESS);
err:
	n2piupc_kstat_detach(n2piupc_p);
	N2PIUPC_DBG2("n2piupc: kstat_attach: failure exit\n");
	return (DDI_FAILURE);
}

/*
 * Create the name kstats for each group.
 */
static int
n2piupc_create_name_kstat(n2piu_grp_t *grp_p)
{
	int i;

	for (i = 0; i < grp_p->num_counters; i++) {
		grp_p->name_kstats_pp[i] = n2piupc_create_picN_kstat(
		    grp_p->grp_name, i,
		    grp_p->regsel_p->fields_p[i].event_offset,
		    grp_p->regsel_p->fields_p[i].num_events,
		    grp_p->regsel_p->fields_p[i].events_p);

		if (grp_p->name_kstats_pp[i] == NULL)
			return (DDI_FAILURE);
	}
	return (DDI_SUCCESS);
}

/*
 * Create the picN kstat. Returns a pointer to the
 * kstat which the driver must store to allow it
 * to be deleted when necessary.
 */
static kstat_t *
n2piupc_create_picN_kstat(char *mod_name, int pic, uint64_t ev_offset,
    int num_ev, n2piu_event_t *ev_array)
{
	int event;
	char pic_name[PIC_STR_LEN];
	kstat_t	*picN_ksp = NULL;
	struct kstat_named *pic_named_data;


	(void) snprintf(pic_name, PIC_STR_LEN, "pic%1d", pic);

	if ((picN_ksp = kstat_create(mod_name, 0, pic_name,
	    "bus", KSTAT_TYPE_NAMED, num_ev, 0)) == NULL) {
		cmn_err(CE_WARN, "%s %s : kstat create failed",
		    mod_name, pic_name);
		return (NULL);
	}

	/* NOTE: Number of events is assumed to always be non-zero. */

	pic_named_data = (struct kstat_named *)picN_ksp->ks_data;

	/*
	 * Fill up data section of the kstat
	 * Write event names and their associated pcr masks.
	 * num_ev - 1 is because CLEAR_PIC is added separately.
	 */
	for (event = 0; event < num_ev - 1; event++) {
		pic_named_data[event].value.ui64 =
		    ev_array[event].value << ev_offset;

		kstat_named_init(&pic_named_data[event],
		    ev_array[event].name, KSTAT_DATA_UINT64);
	}

	/*
	 * add the clear_pic entry
	 */
	pic_named_data[event].value.ui64 =
	    (uint64_t)~(ev_array[event].value << ev_offset);

	kstat_named_init(&pic_named_data[event], ev_array[event].name,
	    KSTAT_DATA_UINT64);

	kstat_install(picN_ksp);

	return (picN_ksp);
}

/*
 * Create the "counters" kstat.
 */
static kstat_t *
n2piupc_create_cntr_kstat(char *name, int dev_inst,
    int (*update)(kstat_t *, int), n2piu_ksinfo_t *ksinfop, int num_pics)
{
	int i;
	char pic_str[PIC_STR_LEN];
	struct kstat *counters_ksp;
	struct kstat_named *counters_named_data;

	N2PIUPC_DBG2("n2piupc_create_cntr_kstat: name: %s instance: %d\n",
	    name, dev_inst);

	/*
	 * Size of kstat is num_pics + 1. extra one for pcr.
	 */

	if ((counters_ksp = kstat_create(name, dev_inst, "counters", "bus",
	    KSTAT_TYPE_NAMED, num_pics + 1, KSTAT_FLAG_WRITABLE)) == NULL) {
		cmn_err(CE_WARN, "%s%d: kstat_create for %s counters failed",
		    NAMEINST(ksinfop->n2piupc_p->n2piupc_dip), name);
		return (NULL);
	}

	counters_named_data = (struct kstat_named *)(counters_ksp->ks_data);
	kstat_named_init(&counters_named_data[0], "pcr", KSTAT_DATA_UINT64);

	for (i = 0; i < num_pics; i++) {
		(void) snprintf(pic_str, PIC_STR_LEN, "pic%1d", i);

		kstat_named_init(&counters_named_data[i+1], pic_str,
		    KSTAT_DATA_UINT64);
	}

	/*
	 * Store the reg type and other info. in the kstat's private field
	 * so that they are available to the update function.
	 */
	counters_ksp->ks_private = (void *)ksinfop;
	counters_ksp->ks_update = update;

	kstat_install(counters_ksp);

	return (counters_ksp);
}

/* Higher-level register write, hides SW abstractions. */
static int
n2piupc_write(n2piupc_t *n2piupc_p, int regid, uint64_t data)
{
	int rval = SUCCESS;

	switch (regid) {
	case SW_N2PIU_BITERR_SEL:
	case SW_N2PIU_BITERR_CLR:
		rval = n2piupc_biterr_write(n2piupc_p, regid, data);
		break;

	default:
		if (n2piupc_set_perfreg(n2piupc_p->n2piupc_handle,
		    regid, data) != H_EOK)
			rval = EIO;
		break;
	}

	N2PIUPC_DBG1("n2piupc_write: status:%d\n", rval);
	return (rval);
}


/* Higher-level register read, hides SW abstractions. */
static int
n2piupc_read(n2piupc_t *n2piupc_p, int regid, uint64_t *data)
{
	int rval = SUCCESS;

	N2PIUPC_DBG2("n2piupc_read enter: regid:%d\n", regid);

	/* This "register" is a layered SW-implemented reg. */
	switch (regid) {
	case SW_N2PIU_BITERR_CNT1_DATA:
	case SW_N2PIU_BITERR_CNT2_DATA:
	case SW_N2PIU_BITERR_SEL:
		rval = n2piupc_biterr_read(n2piupc_p, regid, data);
		break;

	default:
		if (n2piupc_get_perfreg(n2piupc_p->n2piupc_handle,
		    regid, data) != H_EOK)
			rval = EIO;
		break;
	}

	N2PIUPC_DBG1("n2piupc_read exit: data:0x%lx, status:%d\n", *data,
	    rval);

	return (rval);
}


/*
 * Program a performance counter.
 *
 * reggroup is which type of counter.
 * counter is the counter number.
 * event is the event to program for that counter.
 */
static int
n2piupc_perfcnt_program(n2piupc_t *n2piupc_p, n2piu_grp_t *grp_p,
    uint64_t new_events)
{
	uint64_t old_events;
	int rval = SUCCESS;
	uint64_t event_mask;
	int counter;

	N2PIUPC_DBG1(
	    "n2piupc_perfcnt_program enter: new_events:0x%" PRIx64 "\n",
	    new_events);

	if ((rval = n2piupc_read(n2piupc_p, grp_p->regsel_p->regoff,
	    &old_events)) != SUCCESS) {
		N2PIUPC_DBG1(
		    "Read of old event data failed, select reg offset:%ld\n",
		    grp_p->regsel_p->regoff);
		goto done_pgm;
	}

	N2PIUPC_DBG1("  old_events:0x%" PRIx64 "\n", old_events);

	for (counter = 0; counter < grp_p->num_counters; counter++) {

		if (grp_p->counters_p[counter].zero_regoff == NO_REGISTER)
			continue;

		event_mask = grp_p->regsel_p->fields_p[counter].event_mask <<
		    grp_p->regsel_p->fields_p[counter].event_offset;

		N2PIUPC_DBG1(
		    "grp:%s, counter:%d, zero_regoff:0x%lx, "
		    "event_mask:0x%" PRIx64 ", old&mask:0x%lx, "
		    "new&mask:0x%lx\n",
		    grp_p->grp_name, counter,
		    grp_p->counters_p[counter].zero_regoff,
		    event_mask, old_events & event_mask,
		    new_events & event_mask);

		if ((old_events & event_mask) ==
		    (new_events & event_mask))
			continue;

		N2PIUPC_DBG1("Zeroing counter %d\n", counter);
		if ((rval = n2piupc_write(n2piupc_p,
		    grp_p->counters_p[counter].zero_regoff,
		    grp_p->counters_p[counter].zero_value)) != SUCCESS)
			goto done_pgm;
	}

	if (old_events != new_events) {
		N2PIUPC_DBG1("old != new, setting event reg %ld to 0x%lx\n",
		    grp_p->regsel_p->regoff, new_events);
		if ((rval = n2piupc_write(n2piupc_p, grp_p->regsel_p->regoff,
		    new_events)) != SUCCESS) {
			N2PIUPC_DBG1(
			    "Write of new event data failed, "
			    "select reg offset: %ld\n",
			    grp_p->regsel_p->regoff);
			goto done_pgm;
		}
	}
done_pgm:
	N2PIUPC_DBG1("n2piupc_perfcnt_program: returning status %d.\n", rval);
	return (rval);
}

/*
 * kstat update function. Handles reads/writes
 * from/to kstat.
 */
static int
n2piupc_kstat_update(kstat_t *ksp, int rw)
{
	struct kstat_named *data_p;
	int counter;
	n2piu_ksinfo_t *ksinfop = ksp->ks_private;
	n2piu_grp_t *grp_p = ksinfop->grp_p;
	n2piupc_t *n2piupc_p = ksinfop->n2piupc_p;

	data_p = (struct kstat_named *)ksp->ks_data;

	if (rw == KSTAT_WRITE) {

		N2PIUPC_DBG2("n2piupc_kstat_update: wr %ld\n",
		    data_p[0].value.ui64);

		/*
		 * Fields without programmable events won't be zeroed as
		 * n2piupc_perfcnt_program is what zeros them.
		 */

		/* This group has programmable events. */
		if (grp_p->regsel_p->regoff != NO_REGISTER) {

			N2PIUPC_DBG2("write: regoff has valid register\n");
			if (n2piupc_perfcnt_program(n2piupc_p, grp_p,
			    data_p[0].value.ui64) != SUCCESS)
				return (EIO);
		}

	} else {	/* Read the event register and all of the counters. */

		/* This group has programmable events. */
		if (grp_p->regsel_p->regoff != NO_REGISTER) {

			N2PIUPC_DBG2("read: regoff has valid register\n");
			if (n2piupc_read(n2piupc_p, grp_p->regsel_p->regoff,
			    &data_p[0].value.ui64) != SUCCESS)
				return (EIO);
		} else
			data_p[0].value.ui64 = 0ull;

		N2PIUPC_DBG2("n2piupc_kstat_update: rd event %ld",
		    data_p[0].value.ui64);

		for (counter = 0; counter < grp_p->num_counters; counter++) {
			if (n2piupc_read(n2piupc_p,
			    grp_p->counters_p[counter].regoff,
			    &data_p[counter + 1].value.ui64) != SUCCESS)
				return (EIO);

			N2PIUPC_DBG2("cntr%d, off:0x%lx, val:0x%ld", counter,
			    grp_p->counters_p[counter].regoff,
			    data_p[counter + 1].value.ui64);
		}
	}
	return (SUCCESS);
}

void
n2piupc_kstat_fini()
{
	n2piu_grp_t **grp_pp;
	n2piu_grp_t *grp_p;
	int j;

	N2PIUPC_DBG2("n2piupc_kstat_fini called\n");

	for (j = 0, grp_pp = leaf_grps; *grp_pp != NULL; j++, grp_pp++) {
		grp_p = *grp_pp;
		if (grp_p->name_kstats_pp != NULL) {
			n2piupc_delete_name_kstats(grp_p->name_kstats_pp,
			    grp_p->num_counters);
			kmem_free(grp_p->name_kstats_pp,
			    grp_p->num_counters * sizeof (kstat_t));
			grp_p->name_kstats_pp = NULL;
		}
	}
}

static void
n2piupc_delete_name_kstats(kstat_t **name_kstats_pp, int num_kstats)
{
	int i;

	if (name_kstats_pp != NULL) {
		for (i = 0; i < num_kstats; i++) {
			if (name_kstats_pp[i] != NULL)
				kstat_delete(name_kstats_pp[i]);
		}
	}
}

void
n2piupc_kstat_detach(n2piupc_t *n2piupc_p)
{
	n2piu_grp_t **grp_pp;
	int i;

	N2PIUPC_DBG2("n2piupc_kstat_detach called\n");

	for (i = 0, grp_pp = leaf_grps; *grp_pp != NULL; i++, grp_pp++) {
		if (n2piupc_p->n2piupc_ksinfo_p[i] != NULL) {
			if (n2piupc_p->n2piupc_ksinfo_p[i]->cntr_ksp != NULL)
				kstat_delete(
				    n2piupc_p->n2piupc_ksinfo_p[i]->cntr_ksp);
			kmem_free(n2piupc_p->n2piupc_ksinfo_p[i],
			    sizeof (n2piu_ksinfo_t));
		}

	}

	n2piupc_biterr_detach(n2piupc_p->n2piupc_biterr_p);
}