/*
 * 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 2008 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#include <sys/types.h>
#include <sys/atomic.h>
#include <sys/systm.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <sys/modctl.h>
#include <sys/sunddi.h>
#include <ipp/ipp.h>
#include <ipp/ipp_config.h>
#include <inet/common.h>
#include <ipp/meters/meter_impl.h>

#define	D_SM_COMMENT	"IPP Single-Two Rate Token Meter"

/* DDI file for tokenmt ipp module */

/* Default DSCP to colour mapping for colour-aware meter */
enum meter_colour default_dscp_to_colour[64] = {
	TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN,
	TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN,
	TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN,
	TOKENMT_YELLOW, TOKENMT_GREEN, TOKENMT_RED, TOKENMT_GREEN,
	TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN,
	TOKENMT_YELLOW, TOKENMT_GREEN, TOKENMT_RED, TOKENMT_GREEN,
	TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN,
	TOKENMT_YELLOW, TOKENMT_GREEN, TOKENMT_RED, TOKENMT_GREEN,
	TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN,
	TOKENMT_YELLOW, TOKENMT_GREEN, TOKENMT_RED, TOKENMT_GREEN,
	TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN,
	TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN,
	TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN,
	TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN,
	TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN,
	TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN, TOKENMT_GREEN
};

static int tokenmt_create_action(ipp_action_id_t, nvlist_t **, ipp_flags_t);
static int tokenmt_modify_action(ipp_action_id_t, nvlist_t **, ipp_flags_t);
static int tokenmt_destroy_action(ipp_action_id_t, ipp_flags_t);
static int tokenmt_info(ipp_action_id_t, int (*)(nvlist_t *, void *), void *,
    ipp_flags_t);
static int tokenmt_invoke_action(ipp_action_id_t, ipp_packet_t *);

/* Initialize stats */
static int tokenmt_statinit(ipp_action_id_t, tokenmt_data_t *);

/* Stats callback function */
static int tokenmt_update_stats(ipp_stat_t *, void *, int);

ipp_ops_t tokenmt_ops = {
	IPPO_REV,
	tokenmt_create_action,	/* ippo_action_create */
	tokenmt_modify_action,	/* ippo_action_modify */
	tokenmt_destroy_action,	/* ippo_action_destroy */
	tokenmt_info,		/* ippo_action_info */
	tokenmt_invoke_action	/* ippo_action_invoke */
};

extern struct mod_ops mod_ippops;

/*
 * Module linkage information for the kernel.
 */
static struct modlipp modlipp = {
	&mod_ippops,
	D_SM_COMMENT,
	&tokenmt_ops
};

static struct modlinkage modlinkage = {
	MODREV_1,
	(void *)&modlipp,
	NULL
};


int
_init(void)
{
	return (mod_install(&modlinkage));
}

int
_fini(void)
{
	return (mod_remove(&modlinkage));
}

int
_info(struct modinfo *modinfop)
{
	return (mod_info(&modlinkage, modinfop));
}

/* ARGSUSED */
static int
tokenmt_create_action(ipp_action_id_t aid, nvlist_t **nvlpp, ipp_flags_t flags)
{
	nvlist_t *nvlp;
	tokenmt_data_t *tokenmt_data;
	char *next_action;
	tokenmt_cfg_t *cfg_parms;
	uint32_t mode;
	uint32_t bstats;
	int rc, rc2;
	int32_t *colour_tbl;
	uint_t nelem = 64;

	nvlp = *nvlpp;
	*nvlpp = NULL;		/* nvlist should be NULL on return */

	if ((cfg_parms = kmem_zalloc(TOKENMT_CFG_SZ, KM_NOSLEEP)) == NULL) {
		nvlist_free(nvlp);
		return (ENOMEM);
	}

	/* parse red next action name */
	if ((rc = nvlist_lookup_string(nvlp, TOKENMT_RED_ACTION_NAME,
	    &next_action)) != 0) {
		nvlist_free(nvlp);
		tokenmt0dbg(("tokenmt_create_action:invalid config, red "\
		    "action name missing\n"));
		kmem_free(cfg_parms, TOKENMT_CFG_SZ);
		return (rc);
	}
	if ((cfg_parms->red_action = ipp_action_lookup(next_action))
	    == IPP_ACTION_INVAL) {
		nvlist_free(nvlp);
		tokenmt0dbg(("tokenmt_create_action: red action invalid\n"));
		kmem_free(cfg_parms, TOKENMT_CFG_SZ);
		return (EINVAL);
	}

	/* parse yellow next action name, if present  this is Two Rate meter */
	if ((rc = nvlist_lookup_string(nvlp, TOKENMT_YELLOW_ACTION_NAME,
	    &next_action)) == 0) {
		if ((cfg_parms->yellow_action = ipp_action_lookup(next_action))
		    == IPP_ACTION_INVAL) {
			nvlist_free(nvlp);
			tokenmt0dbg(("tokenmt_create_action: yellow action "\
			    "invalid\n"));
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			return (EINVAL);
		}
	} else {
		cfg_parms->yellow_action = TOKENMT_NO_ACTION;
	}

	/* parse green next action name */
	if ((rc = nvlist_lookup_string(nvlp, TOKENMT_GREEN_ACTION_NAME,
	    &next_action)) != 0) {
		nvlist_free(nvlp);
		tokenmt0dbg(("tokenmt_create_action:invalid config, green " \
		    "action name missing\n"));
		kmem_free(cfg_parms, TOKENMT_CFG_SZ);
		return (rc);
	}
	if ((cfg_parms->green_action = ipp_action_lookup(next_action))
	    == IPP_ACTION_INVAL) {
		nvlist_free(nvlp);
		tokenmt0dbg(("tokenmt_create_action: green action invalid\n"));
		kmem_free(cfg_parms, TOKENMT_CFG_SZ);
		return (EINVAL);
	}

	/* parse committed rate  - in kilo bits / sec */
	if ((rc = nvlist_lookup_uint32(nvlp, TOKENMT_COMMITTED_RATE,
	    &cfg_parms->committed_rate)) != 0) {
		nvlist_free(nvlp);
		tokenmt0dbg(("tokenmt_create_action: invalid config, "\
		    " committed rate missing\n"));
		kmem_free(cfg_parms, TOKENMT_CFG_SZ);
		return (rc);
	}
	if (cfg_parms->committed_rate == 0) {
		nvlist_free(nvlp);
		tokenmt0dbg(("tokenmt_create_action: invalid committed rate, "\
		    "%u\n", cfg_parms->committed_rate));
		kmem_free(cfg_parms, TOKENMT_CFG_SZ);
		return (EINVAL);
	}

	/* parse committed burst in bits */
	if ((rc = nvlist_lookup_uint32(nvlp, TOKENMT_COMMITTED_BURST,
	    &cfg_parms->committed_burst)) != 0) {
		nvlist_free(nvlp);
		tokenmt0dbg(("tokenmt_create_action: invalid config, "\
		    " committed burst missing\n"));
		kmem_free(cfg_parms, TOKENMT_CFG_SZ);
		return (rc);
	}


	/*
	 * If the peak burst size is specified, make sure we have the
	 * yellow action.
	 */
	if ((rc = nvlist_lookup_uint32(nvlp, TOKENMT_PEAK_BURST,
	    &cfg_parms->peak_burst)) == 0) {
		if (cfg_parms->yellow_action == TOKENMT_NO_ACTION) {
			nvlist_free(nvlp);
			tokenmt0dbg(("tokenmt_create_action: peak burst "\
			    "specified without yellow action\n"));
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			return (EINVAL);
		}
	} else if (cfg_parms->yellow_action != TOKENMT_NO_ACTION) {
		nvlist_free(nvlp);
		tokenmt0dbg(("tokenmt_create_action: peak burst must be "\
		    "provided with yellow action\n"));
		kmem_free(cfg_parms, TOKENMT_CFG_SZ);
		return (EINVAL);
	}

	/* Check if we have a peak_rate */
	if ((rc = nvlist_lookup_uint32(nvlp, TOKENMT_PEAK_RATE,
	    &cfg_parms->peak_rate)) == 0) {
		if (cfg_parms->yellow_action == TOKENMT_NO_ACTION) {
			nvlist_free(nvlp);
			tokenmt0dbg(("tokenmt_create_action: peak rate "\
			    "specified without yellow action\n"));
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			return (EINVAL);
		} else if ((cfg_parms->peak_rate == 0) ||
		    (cfg_parms->peak_rate < cfg_parms->committed_rate)) {
			nvlist_free(nvlp);
			tokenmt0dbg(("tokenmt_create_action: invalid "\
			    "peak rate, %u\n", cfg_parms->peak_rate));
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			return (EINVAL);
		}
		cfg_parms->tokenmt_type = TRTCL_TOKENMT;
	} else {
		cfg_parms->tokenmt_type = SRTCL_TOKENMT;
	}

	/* Validate the committed and peak burst size */
	if (cfg_parms->tokenmt_type == SRTCL_TOKENMT) {
		if ((cfg_parms->committed_burst == 0) &&
		    (cfg_parms->peak_burst == 0)) {
			nvlist_free(nvlp);
			tokenmt0dbg(("tokenmt_create_action: at least one "\
			    "burst size must be non-zero\n"));
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			return (EINVAL);
		}
	} else {	/* TRTCL_TOKENMT */
		if ((cfg_parms->committed_burst == 0) ||
		    (cfg_parms->peak_burst == 0)) {
			nvlist_free(nvlp);
			tokenmt0dbg(("tokenmt_create_action: both the "\
			    "burst sizes must be non-zero\n"));
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			return (EINVAL);
		}
	}

	/* just copy default colour mapping */
	bcopy(default_dscp_to_colour, cfg_parms->dscp_to_colour,
	    sizeof (default_dscp_to_colour));

	/* parse mode, if present */
	if ((rc = nvlist_lookup_uint32(nvlp, TOKENMT_COLOUR_AWARE,
	    &mode)) != 0) {
		cfg_parms->colour_aware = B_FALSE;
	} else {
		cfg_parms->colour_aware = (mode == 0) ? B_FALSE : B_TRUE;
	}

	/* Get the dscp to colour mapping array */
	if (cfg_parms->colour_aware) {
		if ((rc = nvlist_lookup_int32_array(nvlp,
		    TOKENMT_COLOUR_MAP, &colour_tbl, &nelem)) == 0) {
			int count;
			for (count = 0; count < 64; count++) {
				if (colour_tbl[count] == -1)
					continue;
				cfg_parms->dscp_to_colour[count] =
				    colour_tbl[count];
			}
		}
	}

	/* parse stats */
	if ((rc = nvlist_lookup_uint32(nvlp, IPP_ACTION_STATS_ENABLE, &bstats))
	    != 0) {
		cfg_parms->stats = B_FALSE;
	} else {
		cfg_parms->stats = (bstats == 0) ? B_FALSE : B_TRUE;
	}

	nvlist_free(nvlp);

	/* Initialize other stuff */
	tokenmt_data = kmem_zalloc(TOKENMT_DATA_SZ, KM_NOSLEEP);
	if (tokenmt_data == NULL) {
		kmem_free(cfg_parms, TOKENMT_CFG_SZ);
		return (ENOMEM);
	}

	/* Initialize stats, if required */
	if (cfg_parms->stats) {
		if ((rc = tokenmt_statinit(aid, tokenmt_data)) != 0) {
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			kmem_free(tokenmt_data, TOKENMT_DATA_SZ);
			return (rc);
		}
	}

	/* set action chain reference */
	if ((rc = ipp_action_ref(aid, cfg_parms->red_action, flags)) != 0) {
		tokenmt0dbg(("tokenmt_create_action: ipp_action_ref " \
		    "returned with error %d", rc));
		goto cleanup;
	}
	if ((rc = ipp_action_ref(aid, cfg_parms->green_action, flags)) != 0) {
		tokenmt0dbg(("tokenmt_create_action: ipp_action_ref " \
		    "returned with error %d", rc));
		rc2 = ipp_action_unref(aid, cfg_parms->red_action, flags);
		ASSERT(rc2 == 0);
		goto cleanup;
	}

	if (cfg_parms->yellow_action != TOKENMT_NO_ACTION) {
		if ((rc = ipp_action_ref(aid, cfg_parms->yellow_action,
		    flags)) != 0) {
			tokenmt0dbg(("tokenmt_create_action: ipp_action_ref "\
			    "returned with error %d", rc));
			rc2 = ipp_action_unref(aid, cfg_parms->red_action,
			    flags);
			ASSERT(rc2 == 0);
			rc2 = ipp_action_unref(aid, cfg_parms->green_action,
			    flags);
			ASSERT(rc2 == 0);
			goto cleanup;
		}
	}


	tokenmt_data->cfg_parms = cfg_parms;

	tokenmt_data->committed_tokens = cfg_parms->committed_burst;
	tokenmt_data->peak_tokens = cfg_parms->peak_burst;
	tokenmt_data->last_seen = gethrtime();

	mutex_init(&tokenmt_data->tokenmt_lock, NULL, MUTEX_DEFAULT, 0);
	ipp_action_set_ptr(aid, (void *)tokenmt_data);
	return (0);

cleanup:
	if (cfg_parms->stats) {
		ipp_stat_destroy(tokenmt_data->stats);
	}
	kmem_free(cfg_parms, TOKENMT_CFG_SZ);
	kmem_free(tokenmt_data, TOKENMT_DATA_SZ);
	return (rc);
}

static int
tokenmt_modify_action(ipp_action_id_t aid, nvlist_t **nvlpp, ipp_flags_t flags)
{
	nvlist_t *nvlp;
	int err = 0, err2;
	uint8_t config_type;
	char *next_action_name;
	ipp_action_id_t next_action;
	uint32_t rate, cbs, pbs;
	tokenmt_cfg_t *cfg_parms, *old_cfg;
	tokenmt_data_t *tokenmt_data;
	uint32_t bstats, mode;
	int32_t *colour_tbl;
	uint_t nelem = 64;

	nvlp = *nvlpp;
	*nvlpp = NULL;		/* nvlist should be NULL when this returns */

	if ((err = nvlist_lookup_byte(nvlp, IPP_CONFIG_TYPE, &config_type))
	    != 0) {
		nvlist_free(nvlp);
		tokenmt0dbg(("tokenmt_modify_action: invalid configuration "\
		    "type"));
		return (err);
	}

	if (config_type != IPP_SET) {
		nvlist_free(nvlp);
		tokenmt0dbg(("tokenmt_modify_action: invalid configuration "\
		    "type %d", config_type));
		return (EINVAL);
	}

	tokenmt_data = (tokenmt_data_t *)ipp_action_get_ptr(aid);
	old_cfg = tokenmt_data->cfg_parms;

	cfg_parms = kmem_zalloc(TOKENMT_CFG_SZ, KM_NOSLEEP);
	if (cfg_parms == NULL) {
		nvlist_free(nvlp);
		tokenmt0dbg(("tokenmt_modify_action: memory allocation "\
		    "failure\n"));
		return (ENOMEM);
	}

	/* Just copy all and change as needed */
	bcopy(old_cfg, cfg_parms, TOKENMT_CFG_SZ);

	/* parse red action name, if present */
	if ((err = nvlist_lookup_string(nvlp, TOKENMT_RED_ACTION_NAME,
	    &next_action_name)) == 0) {
		/* Get action id */
		if ((next_action = ipp_action_lookup(next_action_name))
		    == IPP_ACTION_INVAL) {
			nvlist_free(nvlp);
			tokenmt0dbg(("tokenmt_modify_action: next_action "\
			    "invalid"));
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			return (EINVAL);
		}
		cfg_parms->red_action = next_action;
	}

	/* parse yellow action name, if present */
	if ((err = nvlist_lookup_string(nvlp, TOKENMT_YELLOW_ACTION_NAME,
	    &next_action_name)) == 0) {
		/* Get action id */
		if ((next_action = ipp_action_lookup(next_action_name))
		    == IPP_ACTION_INVAL) {
			nvlist_free(nvlp);
			tokenmt0dbg(("tokenmt_modify_action: next_action "\
			    "invalid"));
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			return (EINVAL);
		}
		cfg_parms->yellow_action = next_action;
	} else {
		cfg_parms->yellow_action = TOKENMT_NO_ACTION;
	}

	/* parse green action name, if present */
	if ((err = nvlist_lookup_string(nvlp, TOKENMT_GREEN_ACTION_NAME,
	    &next_action_name)) == 0) {
		/* Get action id */
		if ((next_action = ipp_action_lookup(next_action_name))
		    == IPP_ACTION_INVAL) {
			nvlist_free(nvlp);
			tokenmt0dbg(("tokenmt_modify_action: next_action "\
			    "invalid"));
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			return (EINVAL);
		}
		cfg_parms->green_action = next_action;
	}

	/* parse committed rate, if present */
	if ((err = nvlist_lookup_uint32(nvlp, TOKENMT_COMMITTED_RATE, &rate))
	    == 0) {
		if (rate == 0) {
			nvlist_free(nvlp);
			tokenmt0dbg(("tokenmt_modify_action: invalid "\
			    "committed rate %u\n", cfg_parms->committed_rate));
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			return (EINVAL);
		}
		cfg_parms->committed_rate = rate;
	}

	/* parse committed burst, if present */
	if (nvlist_lookup_uint32(nvlp, TOKENMT_COMMITTED_BURST, &cbs) == 0) {
		cfg_parms->committed_burst = cbs;
	}


	if (nvlist_lookup_uint32(nvlp, TOKENMT_PEAK_BURST, &pbs) == 0) {
		cfg_parms->peak_burst = pbs;
	} else {
		cfg_parms->peak_burst = 0;
	}

	/* If the peak rate is not specified, then it means single rate meter */
	if (nvlist_lookup_uint32(nvlp, TOKENMT_PEAK_RATE, &rate) == 0) {
		cfg_parms->peak_rate = rate;
		if ((rate == 0) || (rate < cfg_parms->committed_rate)) {
			nvlist_free(nvlp);
			tokenmt0dbg(("tokenmt_modify_action: invalid "\
			    "committed rate %u\n", cfg_parms->committed_rate));
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			return (EINVAL);
		}
		cfg_parms->tokenmt_type = TRTCL_TOKENMT;
	} else {
		cfg_parms->peak_rate = 0;
		cfg_parms->tokenmt_type = SRTCL_TOKENMT;
	}

	if (cfg_parms->yellow_action == TOKENMT_NO_ACTION) {
		if ((cfg_parms->peak_burst != 0) ||
		    (cfg_parms->tokenmt_type == TRTCL_TOKENMT)) {
			nvlist_free(nvlp);
			tokenmt0dbg(("tokenmt_modify_action: yellow action "\
			    "missing\n"));
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			return (EINVAL);
		}
	} else {
		if ((cfg_parms->tokenmt_type != TRTCL_TOKENMT) &&
		    (cfg_parms->peak_burst == 0)) {
			nvlist_free(nvlp);
			tokenmt0dbg(("tokenmt_modify_action: peak "\
			    "burst/rate missing\n"));
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			return (EINVAL);
		}
	}

	/* Validate the committed and peak burst size */
	if (cfg_parms->tokenmt_type == SRTCL_TOKENMT) {
		if ((cfg_parms->committed_burst == 0) &&
		    (cfg_parms->peak_burst == 0)) {
			nvlist_free(nvlp);
			tokenmt0dbg(("tokenmt_modify_action: at least one "\
			    "burst size must be non-zero\n"));
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			return (EINVAL);
		}
	} else {	/* TRTCL_TOKENMT */
		if ((cfg_parms->committed_burst == 0) ||
		    (cfg_parms->peak_burst == 0)) {
			nvlist_free(nvlp);
			tokenmt0dbg(("tokenmt_modify_action: both the "\
			    "burst sizes must be non-zero\n"));
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			return (EINVAL);
		}
	}

	/* parse mode */
	if (nvlist_lookup_uint32(nvlp, TOKENMT_COLOUR_AWARE, &mode) == 0) {
		cfg_parms->colour_aware = (mode == 0) ? B_FALSE : B_TRUE;
	} else {
		cfg_parms->colour_aware = B_FALSE;
	}

	if (cfg_parms->colour_aware) {
		if (nvlist_lookup_int32_array(nvlp, TOKENMT_COLOUR_MAP,
		    &colour_tbl, &nelem) == 0) {
			int count;
			for (count = 0; count < 64; count++) {
				if (colour_tbl[count] == -1)
					continue;
				cfg_parms->dscp_to_colour[count] =
				    colour_tbl[count];
			}
		} else {
			bcopy(default_dscp_to_colour, cfg_parms->dscp_to_colour,
			    sizeof (default_dscp_to_colour));
		}
	}

	/* parse stats, if present */
	if (nvlist_lookup_uint32(nvlp, IPP_ACTION_STATS_ENABLE, &bstats) == 0) {
		cfg_parms->stats = (bstats == 0) ? B_FALSE : B_TRUE;
		if (cfg_parms->stats && !old_cfg->stats) {
			if ((err = tokenmt_statinit(aid, tokenmt_data)) != 0) {
				nvlist_free(nvlp);
				kmem_free(cfg_parms, TOKENMT_CFG_SZ);
				return (err);
			}
		} else if (!cfg_parms->stats && old_cfg->stats) {
			ipp_stat_destroy(tokenmt_data->stats);
		}
	}

	/* Can we ref all the new actions? */
	if ((err = ipp_action_ref(aid, cfg_parms->red_action, flags)) != 0) {
		tokenmt0dbg(("tokenmt_modify_data: can't ref. red action\n"));
		kmem_free(cfg_parms, TOKENMT_CFG_SZ);
		return (err);
	}
	if ((err = ipp_action_ref(aid, cfg_parms->green_action, flags)) != 0) {
		tokenmt0dbg(("tokenmt_modify_data:can't ref. green action\n"));
		err2 = ipp_action_unref(aid, cfg_parms->red_action, flags);
		ASSERT(err2 == 0);
		kmem_free(cfg_parms, TOKENMT_CFG_SZ);
		return (err);
	}

	if (cfg_parms->yellow_action != TOKENMT_NO_ACTION) {
		if ((err = ipp_action_ref(aid, cfg_parms->yellow_action,
		    flags)) != 0) {
			tokenmt0dbg(("tokenmt_modify_data:can't ref. yellow "\
			    "action\n"));
			err2 = ipp_action_unref(aid, cfg_parms->red_action,
			    flags);
			ASSERT(err2 == 0);
			err2 = ipp_action_unref(aid, cfg_parms->green_action,
			    flags);
			ASSERT(err2 == 0);
			kmem_free(cfg_parms, TOKENMT_CFG_SZ);
			return (err);
		}
	}


	/* Actually modify the configuration */
	mutex_enter(&tokenmt_data->tokenmt_lock);
	tokenmt_data->cfg_parms = cfg_parms;
	mutex_exit(&tokenmt_data->tokenmt_lock);

	/* Un-ref the old actions */
	err = ipp_action_unref(aid, old_cfg->red_action, flags);
	ASSERT(err == 0);
	if (old_cfg->yellow_action != TOKENMT_NO_ACTION) {
		err = ipp_action_unref(aid, old_cfg->yellow_action, flags);
		ASSERT(err == 0);
	}
	err = ipp_action_unref(aid, old_cfg->green_action, flags);
	ASSERT(err == 0);

	/* Free the old configuration */
	kmem_free(old_cfg, TOKENMT_CFG_SZ);
	return (0);
}

static int
tokenmt_destroy_action(ipp_action_id_t aid, ipp_flags_t flags)
{
	tokenmt_data_t *tokenmt_data;
	tokenmt_cfg_t *cfg_parms;
	int rc;

	tokenmt_data = (tokenmt_data_t *)ipp_action_get_ptr(aid);
	ASSERT(tokenmt_data != NULL);

	cfg_parms = tokenmt_data->cfg_parms;

	if (cfg_parms->stats) {
		ipp_stat_destroy(tokenmt_data->stats);
	}

	/* unreference the action */
	rc = ipp_action_unref(aid, cfg_parms->red_action, flags);
	ASSERT(rc == 0);
	if (cfg_parms->yellow_action != TOKENMT_NO_ACTION) {
		rc = ipp_action_unref(aid, cfg_parms->yellow_action, flags);
		ASSERT(rc == 0);
	}
	rc = ipp_action_unref(aid, cfg_parms->green_action, flags);
	ASSERT(rc == 0);

	mutex_destroy(&tokenmt_data->tokenmt_lock);
	kmem_free(cfg_parms, TOKENMT_CFG_SZ);
	kmem_free(tokenmt_data, TOKENMT_DATA_SZ);
	return (0);
}

static int
tokenmt_invoke_action(ipp_action_id_t aid, ipp_packet_t *packet)
{
	tokenmt_data_t *tokenmt_data;
	ipp_action_id_t next_action;
	mblk_t *mp = NULL;
	int rc;

	/* get mblk from ipp_packet structure */
	mp = ipp_packet_get_data(packet);
	tokenmt_data = (tokenmt_data_t *)ipp_action_get_ptr(aid);
	ASSERT(tokenmt_data != NULL);

	/* meter packet as configured */
	if ((rc = tokenmt_process(&mp, tokenmt_data, &next_action)) != 0) {
		return (rc);
	} else {
		return (ipp_packet_next(packet, next_action));
	}
}

static int
tokenmt_statinit(ipp_action_id_t aid, tokenmt_data_t *tokenmt_data) {

	int rc = 0;
	meter_stat_t *statsp;

	/* install stats entry */
	if ((rc = ipp_stat_create(aid, TOKENMT_STATS_STRING, METER_STATS_COUNT,
	    tokenmt_update_stats, tokenmt_data, &tokenmt_data->stats)) != 0) {
		tokenmt0dbg(("tokenmt_statinit: ipp_stat_create failed "\
		    " with %d\n", rc));
		return (rc);
	}

	statsp = (meter_stat_t *)(tokenmt_data->stats)->ipps_data;
	ASSERT(statsp != NULL);

	if ((rc = ipp_stat_named_init(tokenmt_data->stats, "red_packets",
	    IPP_STAT_UINT64, &statsp->red_packets)) != 0) {
		tokenmt0dbg(("tokenmt_statinit:ipp_stat_named_init failed "\
		    " with %d\n", rc));
		return (rc);
	}
	if ((rc = ipp_stat_named_init(tokenmt_data->stats, "yellow_packets",
	    IPP_STAT_UINT64, &statsp->yellow_packets)) != 0) {
		tokenmt0dbg(("tokenmt_statinit:ipp_stat_named_init failed "\
		    " with %d\n", rc));
		return (rc);
	}
	if ((rc = ipp_stat_named_init(tokenmt_data->stats, "green_packets",
	    IPP_STAT_UINT64, &statsp->green_packets)) != 0) {
		tokenmt0dbg(("tokenmt_statinit:ipp_stat_named_init failed "\
		    " with %d\n", rc));
		return (rc);
	}
	if ((rc = ipp_stat_named_init(tokenmt_data->stats, "red_bits",
	    IPP_STAT_UINT64, &statsp->red_bits)) != 0) {
		tokenmt0dbg(("tokenmt_statinit:ipp_stat_named_init failed "\
		    " with %d\n", rc));
		return (rc);
	}
	if ((rc = ipp_stat_named_init(tokenmt_data->stats, "yellow_bits",
	    IPP_STAT_UINT64, &statsp->yellow_bits)) != 0) {
		tokenmt0dbg(("tokenmt_statinit:ipp_stat_named_init failed "\
		    " with %d\n", rc));
		return (rc);
	}
	if ((rc = ipp_stat_named_init(tokenmt_data->stats, "green_bits",
	    IPP_STAT_UINT64, &statsp->green_bits)) != 0) {
		tokenmt0dbg(("tokenmt_statinit:ipp_stat_named_init failed "\
		    " with %d\n", rc));
		return (rc);
	}
	if ((rc = ipp_stat_named_init(tokenmt_data->stats, "epackets",
	    IPP_STAT_UINT64, &statsp->epackets)) != 0) {
		tokenmt0dbg(("tokenmt_statinit:ipp_stat_named_init failed "\
		    " with %d\n", rc));
		return (rc);
	}

	ipp_stat_install(tokenmt_data->stats);

	return (rc);
}

static int
tokenmt_update_stats(ipp_stat_t *sp, void *args, int rw)
{
	tokenmt_data_t *tokenmt_data = (tokenmt_data_t *)args;
	meter_stat_t *stats = (meter_stat_t *)sp->ipps_data;

	ASSERT((tokenmt_data != NULL) && (stats != NULL));

	(void) ipp_stat_named_op(&stats->red_packets,
	    &tokenmt_data->red_packets, rw);
	(void) ipp_stat_named_op(&stats->yellow_packets,
	    &tokenmt_data->yellow_packets, rw);
	(void) ipp_stat_named_op(&stats->green_packets,
	    &tokenmt_data->green_packets, rw);
	(void) ipp_stat_named_op(&stats->red_bits,
	    &tokenmt_data->red_bits, rw);
	(void) ipp_stat_named_op(&stats->yellow_bits,
	    &tokenmt_data->yellow_bits, rw);
	(void) ipp_stat_named_op(&stats->green_bits,
	    &tokenmt_data->green_bits, rw);
	(void) ipp_stat_named_op(&stats->epackets, &tokenmt_data->epackets,
	    rw);

	return (0);
}

/* ARGSUSED */
static int
tokenmt_info(ipp_action_id_t aid, int (*fn)(nvlist_t *, void *), void *arg,
    ipp_flags_t flags)
{
	nvlist_t *nvlp;
	tokenmt_data_t *tokenmt_data;
	tokenmt_cfg_t *cfg_parms;
	char *next_action;
	int32_t dscp_to_colour[64];
	int rc;

	tokenmt_data = (tokenmt_data_t *)ipp_action_get_ptr(aid);
	ASSERT(tokenmt_data != NULL);

	cfg_parms = tokenmt_data->cfg_parms;

	/* allocate nvlist to be passed back */
	if ((rc = nvlist_alloc(&nvlp, NV_UNIQUE_NAME, KM_NOSLEEP)) != 0) {
		tokenmt0dbg(("tokenmt_info: memory allocation failure\n"));
		return (rc);
	}

	/* look up red next action with the next action id */
	if ((rc = ipp_action_name(cfg_parms->red_action, &next_action)) != 0) {
		tokenmt0dbg(("tokenmt_info: red_action not available\n"));
		nvlist_free(nvlp);
		return (rc);
	}

	/* add next action name */
	if ((rc = nvlist_add_string(nvlp, TOKENMT_RED_ACTION_NAME,
	    next_action)) != 0) {
		nvlist_free(nvlp);
		tokenmt0dbg(("tokenmt_info: error adding red_action\n"));
		kmem_free(next_action, (strlen(next_action) + 1));
		return (rc);
	}

	/* free action name */
	kmem_free(next_action, (strlen(next_action) + 1));


	/* look up yellow next action with the next action id */
	if (cfg_parms->yellow_action != TOKENMT_NO_ACTION) {
		if ((rc = ipp_action_name(cfg_parms->yellow_action,
		    &next_action)) != 0) {
			tokenmt0dbg(("tokenmt_info: yellow_action not "\
			    "available\n"));
			nvlist_free(nvlp);
			return (rc);
		}
		/* add next action name */
		if ((rc = nvlist_add_string(nvlp, TOKENMT_YELLOW_ACTION_NAME,
		    next_action)) != 0) {
			nvlist_free(nvlp);
			tokenmt0dbg(("tokenmt_info: error adding "\
			    "yellow_action\n"));
			kmem_free(next_action, (strlen(next_action) + 1));
			return (rc);
		}
		/* free action name */
		kmem_free(next_action, (strlen(next_action) + 1));
	}

	/* look up green next action with the next action id */
	if ((rc = ipp_action_name(cfg_parms->green_action,
	    &next_action)) != 0) {
		tokenmt0dbg(("tokenmt_info: green_action not available\n"));
		nvlist_free(nvlp);
		return (rc);
	}

	/* add next action name */
	if ((rc = nvlist_add_string(nvlp, TOKENMT_GREEN_ACTION_NAME,
	    next_action)) != 0) {
		nvlist_free(nvlp);
		tokenmt0dbg(("tokenmt_info: error adding green_action\n"));
		kmem_free(next_action, (strlen(next_action) + 1));
		return (rc);
	}

	/* free action name */
	kmem_free(next_action, (strlen(next_action) + 1));

	/* add config type */
	if ((rc = nvlist_add_byte(nvlp, IPP_CONFIG_TYPE, IPP_SET)) != 0) {
		tokenmt0dbg(("tokenmt_info: error adding config_type\n"));
		nvlist_free(nvlp);
		return (rc);
	}

	/* add committed_rate  */
	if ((rc = nvlist_add_uint32(nvlp, TOKENMT_COMMITTED_RATE,
	    cfg_parms->committed_rate)) != 0) {
		tokenmt0dbg(("tokenmt_info: error adding committed_rate\n"));
		nvlist_free(nvlp);
		return (rc);
	}

	if (cfg_parms->tokenmt_type == TRTCL_TOKENMT) {
		/* add peak  rate */
		if ((rc = nvlist_add_uint32(nvlp, TOKENMT_PEAK_RATE,
		    cfg_parms->peak_rate)) != 0) {
			tokenmt0dbg(("tokenmt_info: error adding peak_rate\n"));
			nvlist_free(nvlp);
			return (rc);
		}
	}

	/* add committed_burst  */
	if ((rc = nvlist_add_uint32(nvlp, TOKENMT_COMMITTED_BURST,
	    cfg_parms->committed_burst)) != 0) {
		tokenmt0dbg(("tokenmt_info: error adding committed_burst\n"));
		nvlist_free(nvlp);
		return (rc);
	}

	/* add peak_burst  */
	if (cfg_parms->peak_burst != 0) {
		if ((rc = nvlist_add_uint32(nvlp, TOKENMT_PEAK_BURST,
		    cfg_parms->peak_burst)) != 0) {
			tokenmt0dbg(("tokenmt_info: error adding peak "\
			    "burst\n"));
			nvlist_free(nvlp);
			return (rc);
		}
	}

	/* add colour aware  */
	if ((rc = nvlist_add_uint32(nvlp, TOKENMT_COLOUR_AWARE,
	    cfg_parms->colour_aware)) != 0) {
		tokenmt0dbg(("tokenmt_info: error adding mode\n"));
		nvlist_free(nvlp);
		return (rc);
	}

	if (cfg_parms->colour_aware) {
		bcopy(cfg_parms->dscp_to_colour, dscp_to_colour,
		    sizeof (cfg_parms->dscp_to_colour));
		if ((rc = nvlist_add_int32_array(nvlp, TOKENMT_COLOUR_MAP,
		    dscp_to_colour, 64)) != 0) {
			tokenmt0dbg(("tokenmt_info: error adding colour "\
			    "array\n"));
			nvlist_free(nvlp);
			return (rc);
		}
	}

	if ((rc = nvlist_add_uint32(nvlp, IPP_ACTION_STATS_ENABLE,
	    (uint32_t)cfg_parms->stats)) != 0) {
		tokenmt0dbg(("tokenmt_info: error adding stats status\n"));
		nvlist_free(nvlp);
		return (rc);
	}

	/* call back with nvlist */
	rc = fn(nvlp, arg);

	nvlist_free(nvlp);
	return (rc);
}