/*
 * 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/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/dscpmk/dscpmk_impl.h>

#define D_SM_COMMENT    "IPP dscpmk marker module"

/* DDI file for dscpmk ipp module */

/* default dscp map - dscp unchanged */
uint8_t default_dscp_map[DSCPMK_ARRAY_COUNT] = {
        0,      1,      2,      3,
        4,      5,      6,      7,
        8,      9,      10,     11,
        12,     13,     14,     15,
        16,     17,     18,     19,
        20,     21,     22,     23,
        24,     25,     26,     27,
        28,     29,     30,     31,
        32,     33,     34,     35,
        36,     37,     38,     39,
        40,     41,     42,     43,
        44,     45,     46,     47,
        48,     49,     50,     51,
        52,     53,     54,     55,
        56,     57,     58,     59,
        60,     61,     62,     63
};

static int dscpmk_create_action(ipp_action_id_t, nvlist_t **, ipp_flags_t);
static int dscpmk_modify_action(ipp_action_id_t, nvlist_t **, ipp_flags_t);
static int dscpmk_destroy_action(ipp_action_id_t, ipp_flags_t);
static int dscpmk_info(ipp_action_id_t, int (*)(nvlist_t *, void *), void *,
    ipp_flags_t);
static int dscpmk_invoke_action(ipp_action_id_t, ipp_packet_t *);

/* Creating and updating summary stats */
static int dscpmk_summ_statinit(ipp_action_id_t, dscpmk_data_t *);
static int dscpmk_update_stats(ipp_stat_t *, void *, int);

/* Creating and updating per-dscp stats */
static int dscpmk_det_statinit(ipp_action_id_t, dscpmk_data_t *, int);
static int dscpmk_update_det_stats(ipp_stat_t *, void *, int);

/* Entry points for this IPP module */
ipp_ops_t dscpmk_ops = {
        IPPO_REV,
        dscpmk_create_action,   /* ippo_action_create */
        dscpmk_modify_action,   /* ippo_action_modify */
        dscpmk_destroy_action,  /* ippo_action_destroy */
        dscpmk_info,            /* ippo_action_info */
        dscpmk_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,
        &dscpmk_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));
}

static int
dscpmk_create_action(ipp_action_id_t aid, nvlist_t **nvlpp, ipp_flags_t flags)
{
        nvlist_t *nvlp;
        dscpmk_data_t *dscpmk_data;
        char *next_action;
        int err, cnt;
        int32_t *tbl;
        uint_t nelem = DSCPMK_ARRAY_COUNT;
        uint32_t bstats;

        ASSERT((nvlpp != NULL) && (*nvlpp != NULL));

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

        if ((dscpmk_data = kmem_zalloc(DSCPMK_DATA_SZ, KM_NOSLEEP)) == NULL) {
                nvlist_free(nvlp);
                return (ENOMEM);
        }

        /* parse next action name */
        if ((err = nvlist_lookup_string(nvlp, DSCPMK_NEXT_ACTION_NAME,
            &next_action)) != 0) {
                nvlist_free(nvlp);
                dscpmk0dbg(("dscpmk_create_action: invalid config, " \
                    "next_action name missing\n"));
                kmem_free(dscpmk_data, DSCPMK_DATA_SZ);
                return (err);
        }

        if ((dscpmk_data->next_action = ipp_action_lookup(next_action))
            == IPP_ACTION_INVAL) {
                nvlist_free(nvlp);
                dscpmk0dbg(("dscpmk_create_action: next_action "\
                    "invalid\n"));
                kmem_free(dscpmk_data, DSCPMK_DATA_SZ);
                return (EINVAL);
        }

        /* Fill in the default value */
        bcopy(default_dscp_map, dscpmk_data->dscp_map,
            sizeof (default_dscp_map));
        /*
         * parse dscp_map, if present. Note that the module gets
         * the entire array with unchanged entries marked with -1.
         */
        if ((err = nvlist_lookup_int32_array(nvlp, DSCPMK_DSCP_MAP,
            &tbl, &nelem)) == 0) {
                for (cnt = 0; cnt < DSCPMK_ARRAY_COUNT; cnt++) {
                        if ((tbl[cnt] != DSCPMK_UNCHANGED_DSCP) && (tbl[cnt] !=
                            dscpmk_data->dscp_map[cnt])) {
                                dscpmk_data->dscp_map[cnt] = tbl[cnt];
                        }
                }
        }


        /* parse summary_stats boolean */
        if ((err = nvlist_lookup_uint32(nvlp, IPP_ACTION_STATS_ENABLE, &bstats))
            != 0) {
                dscpmk_data->summary_stats = B_FALSE;
        } else {
                dscpmk_data->summary_stats = (bstats != 0) ? B_TRUE : B_FALSE;
                /* If stats is needed, initialize the stats structure */
                if (dscpmk_data->summary_stats) {
                        if ((err = dscpmk_summ_statinit(aid, dscpmk_data))
                            != 0) {
                                nvlist_free(nvlp);
                                kmem_free(dscpmk_data, DSCPMK_DATA_SZ);
                                return (err);
                        }
                }
        }

        /*
         * Initialize per-dscp stats; B_FALSE in present indicates a dscp
         * with this value (count) is not present in the map.
         */
        for (cnt = 0; cnt < DSCPMK_ARRAY_COUNT; cnt++) {
                dscpmk_data->dscp_stats[cnt].present = B_FALSE;
                dscpmk_data->dscp_stats[cnt].npackets = 0;
        }

        /* parse detailed_stats boolean */
        if ((err = nvlist_lookup_uint32(nvlp, DSCPMK_DETAILED_STATS, &bstats))
            != 0) {
                dscpmk_data->detailed_stats = B_FALSE;
        } else {
                dscpmk_data->detailed_stats = (bstats != 0) ? B_TRUE : B_FALSE;
                /* If stats is needed, initialize the stats structure */
                if (dscpmk_data->detailed_stats) {
                        for (cnt = 0; cnt < DSCPMK_ARRAY_COUNT; cnt++) {
                                int val = dscpmk_data->dscp_map[cnt];
                                if (dscpmk_data->dscp_stats[val].present) {
                                        continue;
                                }
                                dscpmk_data->dscp_stats[val].present = B_TRUE;
                                if ((err = dscpmk_det_statinit(aid, dscpmk_data,
                                    val)) != 0) {
                                        nvlist_free(nvlp);
                                        kmem_free(dscpmk_data, DSCPMK_DATA_SZ);
                                        return (err);
                                }
                        }
                }
        }

        /* Free the nvlist */
        nvlist_free(nvlp);

        /* set action chain reference */
        if ((err = ipp_action_ref(aid, dscpmk_data->next_action, flags)) != 0) {
                dscpmk0dbg(("dscpmk_create_action: ipp_action_ref " \
                    "returned with error %d\n", err));
                if (dscpmk_data->summary_stats) {
                        ipp_stat_destroy(dscpmk_data->stats);
                }
                if (dscpmk_data->detailed_stats) {
                        for (cnt = 0; cnt < DSCPMK_ARRAY_COUNT; cnt++) {
                                if (dscpmk_data->dscp_stats[cnt].present) {
                                        ipp_stat_destroy(
                                            dscpmk_data->dscp_stats[cnt].stats);
                                }
                        }
                }
                kmem_free(dscpmk_data, DSCPMK_DATA_SZ);
                return (err);
        }

        ipp_action_set_ptr(aid, (void *)dscpmk_data);
        return (0);
}

static int
dscpmk_modify_action(ipp_action_id_t aid, nvlist_t **nvlpp, ipp_flags_t flags)
{
        nvlist_t *nvlp;
        int err = 0, cnt;
        uint8_t config_type;
        char *next_action_name;
        uint32_t bstats;
        uint_t nelem = DSCPMK_ARRAY_COUNT;
        int32_t *tbl;
        ipp_action_id_t next_action;
        dscpmk_data_t *dscpmk_data;

        ASSERT((nvlpp != NULL) && (*nvlpp != NULL));

        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);
                dscpmk0dbg(("dscpmk_modify_action: invalid cfg. type\n"));
                return (err);
        }

        if (config_type != IPP_SET) {
                nvlist_free(nvlp);
                dscpmk0dbg(("dscpmk_modify_action: invalid cfg. type " \
                    "%d\n", config_type));
                return (EINVAL);
        }

        dscpmk_data = (dscpmk_data_t *)ipp_action_get_ptr(aid);
        ASSERT(dscpmk_data != NULL);

        /* parse next action name, if present */
        if ((err = nvlist_lookup_string(nvlp, DSCPMK_NEXT_ACTION_NAME,
            &next_action_name)) == 0) {
                /* lookup action name to get action id */
                if ((next_action = ipp_action_lookup(next_action_name))
                    == IPP_ACTION_INVAL) {
                        nvlist_free(nvlp);
                        dscpmk0dbg(("dscpmk_modify_action: next_action "\
                            "invalid\n"));
                        return (EINVAL);
                }
                /* reference new action */
                if ((err = ipp_action_ref(aid, next_action, flags)) != 0) {
                        nvlist_free(nvlp);
                        dscpmk0dbg(("dscpmk_modify_action: ipp_action_ref " \
                            "returned with error %d\n", err));
                        return (err);
                }
                /* unref old action */
                err = ipp_action_unref(aid, dscpmk_data->next_action, flags);
                ASSERT(err == 0);
                dscpmk_data->next_action = next_action;
        }

        /*
         * parse dscp_map, if present. Note that the module gets
         * the entire array with unchanged entries marked with -1.
         * If this array is absent during modification, it means revert to
         * the default table.
         */
        if ((err = nvlist_lookup_int32_array(nvlp, DSCPMK_DSCP_MAP,
            &tbl, &nelem)) == 0) {
                for (cnt = 0; cnt < DSCPMK_ARRAY_COUNT; cnt++) {
                        if ((tbl[cnt] != DSCPMK_UNCHANGED_DSCP) && (tbl[cnt] !=
                            dscpmk_data->dscp_map[cnt])) {
                                dscpmk_data->dscp_map[cnt] = tbl[cnt];
                        }
                }
        } else {
                bcopy(default_dscp_map, dscpmk_data->dscp_map,
                    sizeof (default_dscp_map));
        }

        /* parse summary_stats boolean, if present */
        if ((err = nvlist_lookup_uint32(nvlp, IPP_ACTION_STATS_ENABLE, &bstats))
            == 0) {
                boolean_t val = (bstats != 0) ? B_TRUE : B_FALSE;
                /* Turning on stats */
                if (!dscpmk_data->summary_stats && val) {
                        if ((err = dscpmk_summ_statinit(aid, dscpmk_data))
                            != 0) {
                                nvlist_free(nvlp);
                                return (err);
                        }
                /* Turning off stats */
                } else if (!val && dscpmk_data->summary_stats) {
                        ipp_stat_destroy(dscpmk_data->stats);

                }
                dscpmk_data->summary_stats = val;
        }

        /* parse detailed_stats boolean */
        if ((err = nvlist_lookup_uint32(nvlp, DSCPMK_DETAILED_STATS, &bstats))
            == 0) {
                boolean_t val = (bstats != 0) ? B_TRUE : B_FALSE;
                if (dscpmk_data->detailed_stats && !val) {
                        for (cnt = 0; cnt < DSCPMK_ARRAY_COUNT; cnt++) {
                                if (dscpmk_data->dscp_stats[cnt].present) {
                                        dscpmk_data->dscp_stats[cnt].present =
                                            B_FALSE;
                                        ipp_stat_destroy(dscpmk_data->
                                            dscp_stats[cnt].stats);
                                }
                        }
                }
                dscpmk_data->detailed_stats = val;
        }

        /* The map might have changed */
        if (dscpmk_data->detailed_stats) {
                for (cnt = 0; cnt < DSCPMK_ARRAY_COUNT; cnt++) {
                        int val = dscpmk_data->dscp_map[cnt];
                        if (!dscpmk_data->dscp_stats[val].present) {
                                dscpmk_data->dscp_stats[val].present = B_TRUE;
                                if ((err = dscpmk_det_statinit(aid, dscpmk_data,
                                    val)) != 0) {
                                        nvlist_free(nvlp);
                                        return (err);
                                }
                        }
                }
        }

        /* Free the nvlist */
        nvlist_free(nvlp);
        return (0);
}

static int
dscpmk_destroy_action(ipp_action_id_t aid, ipp_flags_t flags)
{
        dscpmk_data_t *dscpmk_data;
        int err, cnt;

        dscpmk_data = (dscpmk_data_t *)ipp_action_get_ptr(aid);
        ASSERT(dscpmk_data != NULL);

        /* Destroy stats, if gathered */
        if (dscpmk_data->summary_stats) {
                ipp_stat_destroy(dscpmk_data->stats);
        }

        if (dscpmk_data->detailed_stats) {
                for (cnt = 0; cnt < DSCPMK_ARRAY_COUNT; cnt++) {
                        if (dscpmk_data->dscp_stats[cnt].present) {
                                ipp_stat_destroy(dscpmk_data->dscp_stats[cnt].
                                    stats);
                        }
                }
        }

        /* unreference the action */
        err = ipp_action_unref(aid, dscpmk_data->next_action, flags);
        ASSERT(err == 0);

        kmem_free(dscpmk_data, DSCPMK_DATA_SZ);
        return (0);
}

static int
dscpmk_invoke_action(ipp_action_id_t aid, ipp_packet_t *packet)
{
        dscpmk_data_t *dscpmk_data;
        mblk_t *mp = NULL;
        ip_priv_t *priv;
        int err;

        ASSERT(packet != NULL);

        /* get mblk from ipp_packet structure */
        mp = ipp_packet_get_data(packet);
        priv = (ip_priv_t *)ipp_packet_get_private(packet);

        dscpmk_data = (dscpmk_data_t *)ipp_action_get_ptr(aid);
        ASSERT(dscpmk_data != NULL);

        /* dscpmk packet as configured */
        if ((err = dscpmk_process(&mp, dscpmk_data, priv->proc)) != 0) {
                return (err);
        } else {
                /* return packet with next action set */
                return (ipp_packet_next(packet, dscpmk_data->next_action));
        }
}

static int
dscpmk_det_statinit(ipp_action_id_t aid, dscpmk_data_t *dscpmk_data, int val)
{
        int err = 0;
        dscpmk_dscp_stats_t *statp;
        char stats_string[15];

        (void) sprintf(stats_string, "dscpmk_dscp0x%x", val);

        /* install stats entry */
        if ((err = ipp_stat_create(aid, stats_string, DSCPMK_DSCP_STATS_COUNT,
            dscpmk_update_det_stats, dscpmk_data,
            &dscpmk_data->dscp_stats[val].stats)) != 0) {
                dscpmk0dbg(("dscpmk_det_statinit: ipp_stat_create returned "\
                    "with error %d\n", err));
                return (err);
        }

        statp = (dscpmk_dscp_stats_t *)
            (dscpmk_data->dscp_stats[val].stats)->ipps_data;
        ASSERT(statp != NULL);

        if ((err = ipp_stat_named_init(dscpmk_data->dscp_stats[val].stats,
            "dscp", IPP_STAT_UINT32, &statp->dscp)) != 0) {
                dscpmk0dbg(("dscpmk_det_statinit: ipp_stat_named_init "\
                    "returned with error %d\n", err));
                return (err);
        }

        if ((err = ipp_stat_named_init(dscpmk_data->dscp_stats[val].stats,
            "npackets", IPP_STAT_UINT64, &statp->npackets)) != 0) {
                dscpmk0dbg(("dscpmk_det_statinit: ipp_stat_named_init "\
                    "returned with error %d\n", err));
                return (err);
        }

        ipp_stat_install(dscpmk_data->dscp_stats[val].stats);
        return (0);
}


static int
dscpmk_summ_statinit(ipp_action_id_t aid, dscpmk_data_t *dscpmk_data)
{
        int err = 0;
        dscpmk_stat_t *statp;

        /* install stats entry */
        if ((err = ipp_stat_create(aid, DSCPMK_STATS_STRING, DSCPMK_STATS_COUNT,
            dscpmk_update_stats, dscpmk_data, &dscpmk_data->stats)) != 0) {
                dscpmk0dbg(("dscpmk_create_action: ipp_stat_create returned " \
                    "with error %d\n", err));
                return (err);
        }

        statp = (dscpmk_stat_t *)(dscpmk_data->stats)->ipps_data;
        ASSERT(statp != NULL);

        if ((err = ipp_stat_named_init(dscpmk_data->stats, "npackets",
            IPP_STAT_UINT64, &statp->npackets)) != 0) {
                dscpmk0dbg(("dscpmk_summ_statinit: ipp_stat_named_init " \
                    "returned with error %d\n", err));
                return (err);
        }

        if ((err = ipp_stat_named_init(dscpmk_data->stats, "dscp_changed",
            IPP_STAT_UINT64, &statp->dscp_changed)) != 0) {
                dscpmk0dbg(("dscpmk_summ_statinit: ipp_stat_named_init " \
                    "returned with error %d\n", err));
                return (err);
        }

        if ((err = ipp_stat_named_init(dscpmk_data->stats, "dscp_unchanged",
            IPP_STAT_UINT64, &statp->dscp_unchanged)) != 0) {
                dscpmk0dbg(("dscpmk_summ_statinit: ipp_stat_named_init " \
                    "returned with error %d\n", err));
                return (err);
        }

        if ((err = ipp_stat_named_init(dscpmk_data->stats, "ipackets",
            IPP_STAT_UINT64, &statp->ipackets)) != 0) {
                dscpmk0dbg(("dscpmk_summ_statinit: ipp_stat_named_init " \
                    "returned with error %d\n", err));
                return (err);
        }

        if ((err = ipp_stat_named_init(dscpmk_data->stats, "epackets",
            IPP_STAT_UINT64, &statp->epackets)) != 0) {
                dscpmk0dbg(("dscpmk_summ_statinit: ipp_stat_named_init " \
                    "returned with error %d\n", err));
                return (err);
        }

        ipp_stat_install(dscpmk_data->stats);
        return (0);
}

/* ARGSUSED */
static int
dscpmk_update_det_stats(ipp_stat_t *sp, void *arg, int rw)
{
        dscpmk_data_t *dscpmk_data = (dscpmk_data_t *)arg;
        dscpmk_dscp_stats_t *statp;
        uint32_t count;

        for (count = 0; count < DSCPMK_ARRAY_COUNT; count++) {
                if (!dscpmk_data->dscp_stats[count].present)
                        continue;
                statp = (dscpmk_dscp_stats_t *)
                    (dscpmk_data->dscp_stats[count].stats)->ipps_data;
                ASSERT(statp != NULL);
                (void) ipp_stat_named_op(&statp->npackets,
                    &dscpmk_data->dscp_stats[count].npackets, rw);
                (void) ipp_stat_named_op(&statp->dscp, &count, rw);
        }
        return (0);
}

static int
dscpmk_update_stats(ipp_stat_t *sp, void *arg, int rw)
{
        dscpmk_data_t *dscpmk_data = (dscpmk_data_t *)arg;
        dscpmk_stat_t *snames = (dscpmk_stat_t *)sp->ipps_data;
        ASSERT(dscpmk_data != NULL);
        ASSERT(snames != NULL);

        (void) ipp_stat_named_op(&snames->npackets, &dscpmk_data->npackets, rw);
        (void) ipp_stat_named_op(&snames->dscp_changed, &dscpmk_data->changed,
            rw);
        (void) ipp_stat_named_op(&snames->dscp_unchanged,
            &dscpmk_data->unchanged, rw);
        (void) ipp_stat_named_op(&snames->ipackets, &dscpmk_data->ipackets, rw);
        (void) ipp_stat_named_op(&snames->epackets, &dscpmk_data->epackets, rw);

        return (0);
}

/* ARGSUSED */
static int
dscpmk_info(ipp_action_id_t aid, int (*fn)(nvlist_t *, void *), void *arg,
    ipp_flags_t flags)
{
        nvlist_t *nvlp;
        dscpmk_data_t *dscpmk_data;
        char *next_action;
        int err, cnt;
        int32_t dscp_map[DSCPMK_ARRAY_COUNT];

        ASSERT(fn != NULL);

        dscpmk_data = (dscpmk_data_t *)ipp_action_get_ptr(aid);
        ASSERT(dscpmk_data != NULL);

        /* allocate nvlist to be passed back */
        if ((err = nvlist_alloc(&nvlp, NV_UNIQUE_NAME, KM_NOSLEEP)) != 0) {
                dscpmk0dbg(("dscpmk_info: error allocating memory\n"));
                return (err);
        }

        /* look up next action with the next action id */
        if ((err = ipp_action_name(dscpmk_data->next_action,
            &next_action)) != 0) {
                dscpmk0dbg(("dscpmk_info: next action not available\n"));
                nvlist_free(nvlp);
                return (err);
        }

        /* add next action name */
        if ((err = nvlist_add_string(nvlp, DSCPMK_NEXT_ACTION_NAME,
            next_action)) != 0) {
                dscpmk0dbg(("dscpmk_info: error adding next action\n"));
                nvlist_free(nvlp);
                kmem_free(next_action, (strlen(next_action) + 1));
                return (err);
        }

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

        /* add config type */
        if ((err = nvlist_add_byte(nvlp, IPP_CONFIG_TYPE, IPP_SET)) != 0) {
                dscpmk0dbg(("dscpmk_info: error adding config type\n"));
                nvlist_free(nvlp);
                return (err);
        }

        /* add dscp map */
        bcopy(dscpmk_data->dscp_map, dscp_map, sizeof (dscp_map));
        for (cnt = 0; cnt < DSCPMK_ARRAY_COUNT; cnt++) {
                dscp_map[cnt] = dscpmk_data->dscp_map[cnt];
        }
        if ((err = nvlist_add_int32_array(nvlp, DSCPMK_DSCP_MAP,
            dscp_map, DSCPMK_ARRAY_COUNT)) != 0) {
                dscpmk0dbg(("dscpmk_info: error adding dscp map\n"));
                nvlist_free(nvlp);
                return (err);
        }

        /* add summary stats boolean */
        if ((err = nvlist_add_uint32(nvlp, IPP_ACTION_STATS_ENABLE,
            (dscpmk_data->summary_stats ? 1 : 0))) != 0) {
                dscpmk0dbg(("dscpmk_info: error adding stats status\n"));
                nvlist_free(nvlp);
                return (err);
        }

        /* add detailed stats boolean */
        if ((err = nvlist_add_uint32(nvlp, DSCPMK_DETAILED_STATS,
            (dscpmk_data->detailed_stats ? 1 : 0))) != 0) {
                dscpmk0dbg(("dscpmk_info: error adding det stats status\n"));
                nvlist_free(nvlp);
                return (err);
        }

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

        nvlist_free(nvlp);
        return (err);
}