/*
 * 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/dlcosmk/dlcosmk_impl.h>

#define D_SM_COMMENT    "IPP dlcosmk marker module"

/* DDI file for dlcosmk ipp module */

static int dlcosmk_create_action(ipp_action_id_t, nvlist_t **, ipp_flags_t);
static int dlcosmk_modify_action(ipp_action_id_t, nvlist_t **, ipp_flags_t);
static int dlcosmk_destroy_action(ipp_action_id_t, ipp_flags_t);
static int dlcosmk_info(ipp_action_id_t, int (*)(nvlist_t *, void *), void *,
    ipp_flags_t);
static int dlcosmk_invoke_action(ipp_action_id_t, ipp_packet_t *);

static int dlcosmk_statinit(ipp_action_id_t, dlcosmk_data_t *);
static int dlcosmk_update_stats(ipp_stat_t *, void *, int);

/* Entry points for this IPP module */
ipp_ops_t dlcosmk_ops = {
        IPPO_REV,
        dlcosmk_create_action,  /* ippo_action_create */
        dlcosmk_modify_action,  /* ippo_action_modify */
        dlcosmk_destroy_action, /* ippo_action_destroy */
        dlcosmk_info,           /* ippo_action_info */
        dlcosmk_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,
        &dlcosmk_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
dlcosmk_create_action(ipp_action_id_t aid, nvlist_t **nvlpp,
    ipp_flags_t flags)
{
        nvlist_t *nvlp;
        dlcosmk_data_t *dlcosmk_data;
        char *next_action;
        int err;
        uint32_t bstats, param;

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

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

        if ((dlcosmk_data = kmem_zalloc(DLCOSMK_DATA_SZ, KM_NOSLEEP)) == NULL) {
                nvlist_free(nvlp);
                return (ENOMEM);
        }

        /* parse next action name */
        if ((err = nvlist_lookup_string(nvlp, DLCOSMK_NEXT_ACTION_NAME,
            &next_action)) != 0) {
                nvlist_free(nvlp);
                dlcosmk0dbg(("dlcosmk_create_action: invalid config, "\
                    "next_action name missing\n"));
                kmem_free(dlcosmk_data, DLCOSMK_DATA_SZ);
                return (err);
        }
        if ((dlcosmk_data->next_action =
            ipp_action_lookup(next_action)) == IPP_ACTION_INVAL) {
                nvlist_free(nvlp);
                dlcosmk0dbg(("dlcosmk_create_action: next_action invalid\n"));
                kmem_free(dlcosmk_data, DLCOSMK_DATA_SZ);
                return (EINVAL);
        }

        /* parse cos - from the config file */
        if ((err = nvlist_lookup_byte(nvlp, DLCOSMK_COS,
            &dlcosmk_data->usr_pri)) != 0) {
                nvlist_free(nvlp);
                dlcosmk0dbg(("dlcosmk_create_action: invalid config, "\
                    "cos missing\n"));
                kmem_free(dlcosmk_data, DLCOSMK_DATA_SZ);
                return (err);
        }

        /* parse b_band - mapped from cos */
        if ((err = nvlist_lookup_uint32(nvlp, DLCOSMK_BAND, &param)) != 0) {
                nvlist_free(nvlp);
                dlcosmk0dbg(("dlcosmk_create_action: invalid config, "\
                    "b_band missing\n"));
                kmem_free(dlcosmk_data, DLCOSMK_DATA_SZ);
                return (err);
        }
        dlcosmk_data->b_band = param;

        /* parse dl_priority.dl_max  - mapped from cos */
        if ((err = nvlist_lookup_uint32(nvlp, DLCOSMK_PRI, &param)) != 0) {
                nvlist_free(nvlp);
                dlcosmk0dbg(("dlcosmk_create_action: invalid config, "\
                    "dl_priority missing\n"));
                kmem_free(dlcosmk_data, DLCOSMK_DATA_SZ);
                return (err);
        }
        dlcosmk_data->dl_max = param;

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

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

        /* set action chain reference */
        if ((err = ipp_action_ref(aid, dlcosmk_data->next_action,
            flags)) != 0) {
                dlcosmk0dbg(("dlcosmk_create_action: ipp_action_ref " \
                    "returned with error %d\n", err));
                ipp_stat_destroy(dlcosmk_data->stats);
                kmem_free(dlcosmk_data, DLCOSMK_DATA_SZ);
                return (err);
        }

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

static int
dlcosmk_modify_action(ipp_action_id_t aid, nvlist_t **nvlpp, ipp_flags_t flags)
{
        nvlist_t *nvlp;
        int err = 0;
        uint32_t band, dlpri;
        uint8_t config_type;
        uint8_t cos;
        char *next_action_name;
        ipp_action_id_t next_action;
        dlcosmk_data_t *dlcosmk_data;
        uint32_t bstats;

        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);
                dlcosmk0dbg(("dlcosmk_modify_action: invalid configuration "\
                    "type\n"));
                return (err);
        }

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

        dlcosmk_data = (dlcosmk_data_t *)ipp_action_get_ptr(aid);
        ASSERT(dlcosmk_data != NULL);

        /* parse next action name, if present */
        if ((err = nvlist_lookup_string(nvlp, DLCOSMK_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);
                        dlcosmk0dbg(("dlcosmk_modify_action: next_action "\
                            "invalid\n"));
                        return (EINVAL);
                }
                /* reference new action */
                if ((err = ipp_action_ref(aid, next_action, flags)) != 0) {
                        nvlist_free(nvlp);
                        dlcosmk0dbg(("dlcosmk_modify_action: ipp_action_ref "\
                            "returned with error %d\n", err));
                        return (err);
                }
                /* unref old action */
                err = ipp_action_unref(aid, dlcosmk_data->next_action, flags);
                ASSERT(err == 0);
                dlcosmk_data->next_action = next_action;
        }

        /* parse cos, if present */
        if ((err = nvlist_lookup_byte(nvlp, DLCOSMK_COS, &cos)) == 0) {

                /* parse b_band, mapped from cos */
                if ((err = nvlist_lookup_uint32(nvlp, DLCOSMK_BAND,
                    &band)) != 0) {
                        nvlist_free(nvlp);
                        dlcosmk0dbg(("dlcosmk_modify_action: b_band not "\
                            "provided\n"));
                        return (err);
                }

                /* parse dl_priority, mapped from cos */
                if ((err = nvlist_lookup_uint32(nvlp, DLCOSMK_PRI,
                    &dlpri)) != 0) {
                        nvlist_free(nvlp);
                        dlcosmk0dbg(("dlcosmk_modify_action: dl_priority not "\
                            "provided\n"));
                        return (err);
                }

                /* Have all the three values, change them */
                dlcosmk_data->usr_pri = cos;
                dlcosmk_data->b_band = band;
                dlcosmk_data->dl_max = dlpri;
        }


        /* parse gather_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 (!dlcosmk_data->gather_stats && val) {
                        if ((err = dlcosmk_statinit(aid, dlcosmk_data)) != 0) {
                                nvlist_free(nvlp);
                                return (err);
                        }
                /* Turning off stats */
                } else if (!val && dlcosmk_data->gather_stats) {
                        ipp_stat_destroy(dlcosmk_data->stats);

                }
                dlcosmk_data->gather_stats = val;
        }

        /* Free thenvlist */
        nvlist_free(nvlp);
        return (0);
}

static int
dlcosmk_destroy_action(ipp_action_id_t aid, ipp_flags_t flags)
{
        dlcosmk_data_t *dlcosmk_data;
        int err;

        dlcosmk_data = (dlcosmk_data_t *)ipp_action_get_ptr(aid);
        ASSERT(dlcosmk_data != NULL);

        /* Destroy stats, if gathered */
        if (dlcosmk_data->gather_stats) {
                ipp_stat_destroy(dlcosmk_data->stats);
        }

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

        kmem_free(dlcosmk_data, DLCOSMK_DATA_SZ);
        return (0);
}

static int
dlcosmk_invoke_action(ipp_action_id_t aid, ipp_packet_t *packet)
{
        dlcosmk_data_t *dlcosmk_data;
        mblk_t *mp = NULL;
        int err;
        ip_priv_t *priv;

        ASSERT(packet != NULL);

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

        dlcosmk_data = (dlcosmk_data_t *)ipp_action_get_ptr(aid);
        ASSERT(dlcosmk_data != NULL);

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

static int
dlcosmk_statinit(ipp_action_id_t aid, dlcosmk_data_t *dlcosmk_data)
{
        int err;
        dlcosmk_stat_t *statp;

        /* install stats entry */
        if ((err = ipp_stat_create(aid, DLCOSMK_STATS_STRING,
            DLCOSMK_STATS_COUNT, dlcosmk_update_stats, dlcosmk_data,
            &dlcosmk_data->stats)) != 0) {
                dlcosmk0dbg(("dlcosmk_create_action: ipp_stat_create " \
                    "returned with error %d\n", err));
                return (err);
        }

        statp = (dlcosmk_stat_t *)(dlcosmk_data->stats)->ipps_data;
        ASSERT(statp != NULL);

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

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

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

        if ((err = ipp_stat_named_init(dlcosmk_data->stats, "usr_pri",
            IPP_STAT_INT32, &statp->usr_pri)) != 0) {
                dlcosmk0dbg(("dlcosmk_create_action: ipp_stat_named_init " \
                    "returned with error %d", err));
                return (err);
        }

        if ((err = ipp_stat_named_init(dlcosmk_data->stats, "b_band",
            IPP_STAT_INT32, &statp->b_band)) != 0) {
                dlcosmk0dbg(("dlcosmk_create_action: ipp_stat_named_init " \
                    "returned with error %d\n", err));
                return (err);
        }

        if ((err = ipp_stat_named_init(dlcosmk_data->stats, "dl_max",
            IPP_STAT_INT32, &statp->dl_max)) != 0) {
                dlcosmk0dbg(("dlcosmk_create_action: ipp_stat_named_init " \
                    "returned with error %d\n", err));
                return (err);
        }

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

static int
dlcosmk_update_stats(ipp_stat_t *sp, void *arg, int rw)
{
        dlcosmk_data_t *dlcosmk_data = (dlcosmk_data_t *)arg;
        dlcosmk_stat_t *snames = (dlcosmk_stat_t *)sp->ipps_data;
        uint32_t upri, bband;

        ASSERT(dlcosmk_data != NULL);
        ASSERT(snames != NULL);

        upri = dlcosmk_data->usr_pri;
        bband = dlcosmk_data->b_band;

        (void) ipp_stat_named_op(&snames->npackets, &dlcosmk_data->npackets,
            rw);
        (void) ipp_stat_named_op(&snames->ipackets, &dlcosmk_data->ipackets,
            rw);
        (void) ipp_stat_named_op(&snames->epackets, &dlcosmk_data->epackets,
            rw);
        (void) ipp_stat_named_op(&snames->usr_pri, &upri, rw);
        (void) ipp_stat_named_op(&snames->b_band, &bband, rw);
        (void) ipp_stat_named_op(&snames->dl_max, &dlcosmk_data->dl_max, rw);

        return (0);
}

/* ARGSUSED */
static int
dlcosmk_info(ipp_action_id_t aid, int (*fn)(nvlist_t *, void *), void *arg,
    ipp_flags_t flags)
{
        nvlist_t *nvlp;
        dlcosmk_data_t *dlcosmk_data;
        char *next_action;
        int err;

        ASSERT(fn != NULL);

        dlcosmk_data = (dlcosmk_data_t *)ipp_action_get_ptr(aid);
        ASSERT(dlcosmk_data != NULL);

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

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

        /* add next action name */
        if ((err = nvlist_add_string(nvlp, DLCOSMK_NEXT_ACTION_NAME,
            next_action)) != 0) {
                dlcosmk0dbg(("dlcosmk_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) {
                dlcosmk0dbg(("dlcosmk_info: error adding config. type\n"));
                nvlist_free(nvlp);
                return (err);
        }

        /* just give the cos, since that is what is provided in the config */
        if ((err = nvlist_add_byte(nvlp, DLCOSMK_COS, dlcosmk_data->usr_pri))
            != 0) {
                dlcosmk0dbg(("dlcosmk_info: error adding cos\n"));
                nvlist_free(nvlp);
                return (err);
        }

        /* add gather stats boolean */
        if ((err = nvlist_add_uint32(nvlp, IPP_ACTION_STATS_ENABLE,
            (dlcosmk_data->gather_stats ? 1 : 0))) != 0) {
                dlcosmk0dbg(("dlcosmk_info: error adding stats status\n"));
                nvlist_free(nvlp);
                return (err);
        }

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

        nvlist_free(nvlp);
        return (err);
}