/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (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 2002 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#include <sys/types.h>
#include <sys/conf.h>
#include <sys/atomic.h>
#include <sys/systm.h>
#include <sys/socket.h>
#include <sys/spl.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/flowacct/flowacct_impl.h>
#include <sys/ddi.h>

#define D_SM_COMMENT    "IPP Flow Accounting Module"

/* DDI file for flowacct ipp module */

static int flowacct_create_action(ipp_action_id_t, nvlist_t **, ipp_flags_t);
static int flowacct_modify_action(ipp_action_id_t, nvlist_t **, ipp_flags_t);
static int flowacct_destroy_action(ipp_action_id_t, ipp_flags_t);
static int flowacct_info(ipp_action_id_t, int (*)(nvlist_t *, void *), void *,
    ipp_flags_t);
static int flowacct_invoke_action(ipp_action_id_t, ipp_packet_t *);

static int update_flowacct_kstats(ipp_stat_t *, void *, int);

ipp_ops_t flowacct_ops = {
        IPPO_REV,
        flowacct_create_action,         /* ippo_action_create */
        flowacct_modify_action,         /* ippo_action_modify */
        flowacct_destroy_action,        /* ippo_action_destroy */
        flowacct_info,                  /* ippo_action_info */
        flowacct_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 " 1.12",
        &flowacct_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));
}

/* Update global stats */
static int
update_flowacct_kstats(ipp_stat_t *sp, void *arg, int rw)
{
        flowacct_data_t *flowacct_data = (flowacct_data_t *)arg;
        flowacct_stat_t *fl_stat  = (flowacct_stat_t *)sp->ipps_data;
        ASSERT((fl_stat != NULL) && (flowacct_data != 0));

        (void) ipp_stat_named_op(&fl_stat->nbytes, &flowacct_data->nbytes, rw);
        (void) ipp_stat_named_op(&fl_stat->tbytes, &flowacct_data->tbytes, rw);
        (void) ipp_stat_named_op(&fl_stat->nflows, &flowacct_data->nflows, rw);
        (void) ipp_stat_named_op(&fl_stat->usedmem, &flowacct_data->usedmem,
            rw);
        (void) ipp_stat_named_op(&fl_stat->npackets, &flowacct_data->npackets,
            rw);
        (void) ipp_stat_named_op(&fl_stat->epackets, &flowacct_data->epackets,
            rw);
        return (0);
}

/* Initialize global stats */
static int
global_statinit(ipp_action_id_t aid, flowacct_data_t *flowacct_data)
{
        flowacct_stat_t *flacct_stat;
        int err = 0;

        if ((err = ipp_stat_create(aid, FLOWACCT_STATS_STRING,
            FLOWACCT_STATS_COUNT, update_flowacct_kstats, flowacct_data,
            &flowacct_data->stats)) != 0) {
                flowacct0dbg(("global_statinit: error creating flowacct "\
                    "stats\n"));
                return (err);
        }
        flacct_stat = (flowacct_stat_t *)(flowacct_data->stats)->ipps_data;
        ASSERT(flacct_stat != NULL);

        if ((err = ipp_stat_named_init(flowacct_data->stats, "bytes_in_tbl",
            IPP_STAT_UINT64, &flacct_stat->tbytes)) != 0) {
                flowacct0dbg(("global_statinit: ipp_stat_named_init returned "\
                    "with error %d\n", err));
                return (err);
        }
        if ((err = ipp_stat_named_init(flowacct_data->stats, "nbytes",
            IPP_STAT_UINT64, &flacct_stat->nbytes)) != 0) {
                flowacct0dbg(("global_statinit: ipp_stat_named_init returned "\
                    "with error %d\n", err));
                return (err);
        }
        if ((err = ipp_stat_named_init(flowacct_data->stats, "npackets",
            IPP_STAT_UINT64, &flacct_stat->npackets)) != 0) {
                flowacct0dbg(("global_statinit:ipp_stat_named_init returned "\
                    "with error %d\n", err));
                return (err);
        }
        if ((err = ipp_stat_named_init(flowacct_data->stats, "usedmem",
            IPP_STAT_UINT64, &flacct_stat->usedmem)) != 0) {
                flowacct0dbg(("global_statinit:ipp_stat_named_init returned "\
                    "with error %d\n", err));
                return (err);
        }
        if ((err = ipp_stat_named_init(flowacct_data->stats, "flows_in_tbl",
            IPP_STAT_UINT32, &flacct_stat->nflows)) != 0) {
                flowacct0dbg(("global_statinit:ipp_stat_named_init returned "\
                    "with error %d\n", err));
                return (err);
        }
        if ((err = ipp_stat_named_init(flowacct_data->stats, "epackets",
            IPP_STAT_UINT64, &flacct_stat->epackets)) != 0) {
                flowacct0dbg(("global_statinit:ipp_stat_named_init returned "\
                    "with error %d\n", err));
                return (err);
        }
        ipp_stat_install(flowacct_data->stats);

        return (err);
}

static int
flowacct_create_action(ipp_action_id_t aid, nvlist_t **nvlpp, ipp_flags_t flags)
{
        nvlist_t *nvlp;
        flowacct_data_t *flowacct_data;
        char *next_action;
        int rc, flow_count;
        list_head_t *head;
        uint32_t bstats;
        uint32_t timeout = FLOWACCT_DEF_TIMEOUT;
        uint32_t timer = FLOWACCT_DEF_TIMER;

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

        if ((flowacct_data = kmem_zalloc(FLOWACCT_DATA_SZ, KM_NOSLEEP))
            == NULL) {
                nvlist_free(nvlp);
                return (ENOMEM);
        }

        /* parse next action name */
        if ((rc = nvlist_lookup_string(nvlp, FLOWACCT_NEXT_ACTION_NAME,
            &next_action)) != 0) {
                nvlist_free(nvlp);
                kmem_free(flowacct_data, FLOWACCT_DATA_SZ);
                flowacct0dbg(("flowacct_create_action: invalid config, "\
                    "next_action missing\n"));
                return (rc);
        }
        if ((flowacct_data->next_action = ipp_action_lookup(next_action))
            == IPP_ACTION_INVAL) {
                nvlist_free(nvlp);
                flowacct0dbg(("flowacct_create_action: invalid next_action\n"));
                kmem_free(flowacct_data, FLOWACCT_DATA_SZ);
                return (EINVAL);
        }

        if ((rc = ipp_action_name(aid, &flowacct_data->act_name)) != 0) {
                nvlist_free(nvlp);
                flowacct0dbg(("flowacct_create_action: invalid next aid\n"));
                kmem_free(flowacct_data, FLOWACCT_DATA_SZ);
                return (EINVAL);
        }

        /* parse flow timeout - in millisec, if present */
        (void) nvlist_lookup_uint32(nvlp, FLOWACCT_TIMEOUT, &timeout);

        /* Convert to FLOWACCT_MSEC_TO_NSEC */
        flowacct_data->timeout = (uint64_t)timeout * FLOWACCT_MSEC_TO_NSEC;

        /* parse flow timer - in millisec, if present  */
        (void) nvlist_lookup_uint32(nvlp, FLOWACCT_TIMER, &timer);

        /* Convert to FLOWACCT_MSEC_TO_USEC */
        flowacct_data->timer = (uint64_t)timer * FLOWACCT_MSEC_TO_USEC;

        if ((rc = nvlist_lookup_uint32(nvlp, FLOWACCT_MAX_LIMIT,
            &flowacct_data->max_limit)) != 0) {
                nvlist_free(nvlp);
                flowacct0dbg(("flowacct_create_action: invalid config, "\
                    "max_limit missing\n"));
                kmem_free(flowacct_data, FLOWACCT_DATA_SZ);
                return (rc);
        }

        if ((rc = nvlist_lookup_uint32(nvlp, IPP_ACTION_STATS_ENABLE,
            &bstats)) != 0) {
                flowacct_data->global_stats = B_FALSE;
        } else {
                flowacct_data->global_stats = (boolean_t)bstats;
                if (flowacct_data->global_stats) {
                        if ((rc = global_statinit(aid, flowacct_data)) != 0) {
                                kmem_free(flowacct_data, FLOWACCT_DATA_SZ);
                                return (rc);
                        }
                }
        }

        nvlist_free(nvlp);

        /* set action chain reference */
        if ((rc = ipp_action_ref(aid, flowacct_data->next_action,
            flags)) != 0) {
                flowacct0dbg(("flowacct_create_action: ipp_action_ref " \
                    "returned with error %d\n", rc));
                if (flowacct_data->stats != NULL) {
                        ipp_stat_destroy(flowacct_data->stats);
                }
                kmem_free(flowacct_data, FLOWACCT_DATA_SZ);
                return (rc);
        }

        /* Initialize locks */
        for (flow_count = 0, head = flowacct_data->flows_tbl;
            flow_count < (FLOW_TBL_COUNT + 1); flow_count++, head++) {
                mutex_init(&head->lock, NULL, MUTEX_DEFAULT, 0);
        }

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

static int
flowacct_modify_action(ipp_action_id_t aid, nvlist_t **nvlpp, ipp_flags_t flags)
{
        nvlist_t *nvlp;
        int rc = 0;
        uint8_t config_type;
        char *next_action_name, *act_name;
        ipp_action_id_t next_action;
        uint32_t timeout, timer, bstats, max_limit;
        flowacct_data_t *flowacct_data;

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

        if ((rc = nvlist_lookup_byte(nvlp, IPP_CONFIG_TYPE, &config_type))
            != 0) {
                nvlist_free(nvlp);
                flowacct0dbg(("flowacct_modify_action: invalid configuration "\
                    "type\n"));
                return (rc);
        }

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

        flowacct_data = (flowacct_data_t *)ipp_action_get_ptr(aid);

        /* parse next action name, if present */
        if ((rc = nvlist_lookup_string(nvlp, FLOWACCT_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);
                        flowacct0dbg(("flowacct_modify_action: next_action "\
                            "invalid\n"));
                        return (EINVAL);
                }
                /* reference new action */
                if ((rc = ipp_action_ref(aid, next_action, flags)) != 0) {
                        nvlist_free(nvlp);
                        flowacct0dbg(("flowacct_modify_action: "\
                            "ipp_action_ref returned with error %d\n", rc));
                        return (rc);
                }

                if ((rc = ipp_action_name(aid, &act_name)) != 0) {
                        nvlist_free(nvlp);
                        flowacct0dbg(("flowacct_modify_action: invalid next "\
                            "aid\n"));
                        return (EINVAL);
                }

                /* unref old action */
                rc = ipp_action_unref(aid, flowacct_data->next_action, flags);
                ASSERT(rc == 0);
                flowacct_data->next_action = next_action;
                kmem_free(flowacct_data->act_name,
                    (strlen(flowacct_data->act_name) + 1));
                flowacct_data->act_name = act_name;
        }

        /* parse timeout, if present */
        if ((rc = nvlist_lookup_uint32(nvlp, FLOWACCT_TIMEOUT, &timeout))
            == 0) {
                flowacct_data->timeout = (uint64_t)timeout *
                    FLOWACCT_MSEC_TO_NSEC;
        }

        /* parse timer, if present */
        if ((rc = nvlist_lookup_uint32(nvlp, FLOWACCT_TIMER, &timer)) == 0) {
                flowacct_data->timer = (uint64_t)timer * FLOWACCT_MSEC_TO_USEC;
        }

        /* parse max_flow, if present */
        if ((rc = nvlist_lookup_uint32(nvlp, FLOWACCT_MAX_LIMIT, &max_limit))
            == 0) {
                flowacct_data->max_limit = max_limit;
        }

        /* parse gather_stats boolean, if present */
        if ((rc = nvlist_lookup_uint32(nvlp, IPP_ACTION_STATS_ENABLE, &bstats))
            == 0) {
                boolean_t new_val = (boolean_t)bstats;

                /* Turning global stats on */
                if (new_val && !flowacct_data->global_stats) {
                        rc = global_statinit(aid, flowacct_data);
                        if (rc == 0) {
                                flowacct_data->global_stats = new_val;
                        } else {
                                flowacct0dbg(("flowacct_modify_action: error "\
                                    "enabling stats\n"));
                        }
                } else if (!new_val && flowacct_data->global_stats) {
                        flowacct_data->global_stats = new_val;
                        ipp_stat_destroy(flowacct_data->stats);
                }
        }
        return (0);
}

static int
flowacct_destroy_action(ipp_action_id_t aid, ipp_flags_t flags)
{
        flowacct_data_t *flowacct_data;
        int rc, flow_count;
        list_head_t *head;

        flowacct_data = (flowacct_data_t *)ipp_action_get_ptr(aid);
        ASSERT(flowacct_data != NULL);

        while (flowacct_data->flow_tid != 0) {
                timeout_id_t tid = flowacct_data->flow_tid;
                flowacct_data->flow_tid = 0;
                (void) untimeout(tid);
        }

        if (flowacct_data->stats != NULL) {
                ipp_stat_destroy(flowacct_data->stats);
        }

        /* Dump all the flows to the file */
        flowacct_timer(FLOWACCT_PURGE_FLOW, flowacct_data);

        kmem_free(flowacct_data->act_name, (strlen(flowacct_data->act_name)
            + 1));

        /* Destroy the locks */
        for (flow_count = 0, head = flowacct_data->flows_tbl;
            flow_count < FLOW_TBL_COUNT; flow_count++, head++) {
                mutex_destroy(&head->lock);
        }
        /* unreference the action */
        rc = ipp_action_unref(aid, flowacct_data->next_action, flags);
        ASSERT(rc == 0);


        kmem_free(flowacct_data, FLOWACCT_DATA_SZ);
        return (0);
}

static int
flowacct_invoke_action(ipp_action_id_t aid, ipp_packet_t *packet)
{
        flowacct_data_t *flowacct_data;
        mblk_t *mp = NULL;
        int rc;

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

        /* flowacct packet as configured */
        if ((rc = flowacct_process(&mp, flowacct_data)) != 0) {
                return (rc);
        } else {
                /* return packet with next action set */
                return (ipp_packet_next(packet, flowacct_data->next_action));
        }
}

/* ARGSUSED */
static int
flowacct_info(ipp_action_id_t aid, int (*fn)(nvlist_t *, void *), void *arg,
    ipp_flags_t flags)
{
        nvlist_t *nvlp;
        flowacct_data_t *flowacct_data;
        char *next_action;
        uint32_t param;
        int rc;

        flowacct_data = (flowacct_data_t *)ipp_action_get_ptr(aid);
        ASSERT(flowacct_data != NULL);
        ASSERT(fn != NULL);

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

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

        /* add next action name */
        if ((rc = nvlist_add_string(nvlp, FLOWACCT_NEXT_ACTION_NAME,
            next_action)) != 0) {
                flowacct0dbg(("flowacct_info: error adding next action\n"));
                nvlist_free(nvlp);
                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) {
                flowacct0dbg(("flowacct_info: error adding config type\n"));
                nvlist_free(nvlp);
                return (rc);
        }

        /* add timer */
        param = flowacct_data->timer / FLOWACCT_MSEC_TO_USEC;
        if ((rc = nvlist_add_uint32(nvlp, FLOWACCT_TIMER, param)) != 0) {
                flowacct0dbg(("flowacct_info: error adding timer info.\n"));
                nvlist_free(nvlp);
                return (rc);
        }

        /* add max_limit */
        if ((rc = nvlist_add_uint32(nvlp, FLOWACCT_MAX_LIMIT,
            flowacct_data->max_limit)) != 0) {
                flowacct0dbg(("flowacct_info: error adding max_flow info.\n"));
                nvlist_free(nvlp);
                return (rc);
        }


        param = flowacct_data->timeout / FLOWACCT_MSEC_TO_NSEC;
        /* add timeout */
        if ((rc = nvlist_add_uint32(nvlp, FLOWACCT_TIMEOUT, param)) != 0) {
                flowacct0dbg(("flowacct_info: error adding timeout info.\n"));
                nvlist_free(nvlp);
                return (rc);
        }

        /* add global stats boolean */
        if ((rc = nvlist_add_uint32(nvlp, IPP_ACTION_STATS_ENABLE,
            (uint32_t)flowacct_data->global_stats)) != 0) {
                flowacct0dbg(("flowacct_info: error adding global stats "\
                    "info.\n"));
                nvlist_free(nvlp);
                return (rc);
        }

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

        nvlist_free(nvlp);
        return (rc);
}