/*
 * 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 2010 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 *
 * Copyright 2015 Garrett D'Amore <garrett@damore.org>
 */

#include <stdlib.h>
#include <string.h>
#include <strings.h>
#include <sys/types.h>
#include <libdladm_impl.h>
#include <libdllink.h>
#include <libdlstat.h>
#include <libdlether.h>

/*
 * Ethernet administration library.
 */

/*
 * kstat names for extracting attributes.
 */
typedef struct ether_spdx_s {
        dladm_ether_spdx_t eth_spdx;
        char *eth_spdx_stat_name;
} ether_spdx_t;

static ether_spdx_t cap_spdx[] = {
        {{400000, LINK_DUPLEX_FULL}, "cap_400gfdx"},
        {{200000, LINK_DUPLEX_FULL}, "cap_200gfdx"},
        {{100000, LINK_DUPLEX_FULL}, "cap_100gfdx"},
        {{50000, LINK_DUPLEX_FULL}, "cap_50gfdx"},
        {{40000, LINK_DUPLEX_FULL}, "cap_40gfdx"},
        {{25000, LINK_DUPLEX_FULL}, "cap_25gfdx"},
        {{10000, LINK_DUPLEX_FULL}, "cap_10gfdx"},
        {{5000, LINK_DUPLEX_FULL}, "cap_5000fdx"},
        {{2500, LINK_DUPLEX_FULL}, "cap_2500fdx"},
        {{1000, LINK_DUPLEX_FULL}, "cap_1000fdx"},
        {{1000, LINK_DUPLEX_HALF}, "cap_1000hdx"},
        {{100, LINK_DUPLEX_FULL}, "cap_100fdx"},
        {{100, LINK_DUPLEX_HALF}, "cap_100hdx"},
        {{10, LINK_DUPLEX_FULL}, "cap_10fdx"},
        {{10, LINK_DUPLEX_HALF}, "cap_10hdx"},
        {{0, LINK_DUPLEX_UNKNOWN}, NULL}
};

static ether_spdx_t adv_cap_spdx[] = {
        {{400000, LINK_DUPLEX_FULL}, "adv_cap_400gfdx"},
        {{200000, LINK_DUPLEX_FULL}, "adv_cap_200gfdx"},
        {{100000, LINK_DUPLEX_FULL}, "adv_cap_100gfdx"},
        {{50000, LINK_DUPLEX_FULL}, "adv_cap_50gfdx"},
        {{40000, LINK_DUPLEX_FULL}, "adv_cap_40gfdx"},
        {{25000, LINK_DUPLEX_FULL}, "adv_cap_25gfdx"},
        {{10000, LINK_DUPLEX_FULL}, "adv_cap_10gfdx"},
        {{5000, LINK_DUPLEX_FULL}, "adv_cap_5000fdx"},
        {{2500, LINK_DUPLEX_FULL}, "adv_cap_2500fdx"},
        {{1000, LINK_DUPLEX_FULL}, "adv_cap_1000fdx"},
        {{1000, LINK_DUPLEX_HALF}, "adv_cap_1000hdx"},
        {{100, LINK_DUPLEX_FULL}, "adv_cap_100fdx"},
        {{100, LINK_DUPLEX_HALF}, "adv_cap_100hdx"},
        {{10, LINK_DUPLEX_FULL}, "adv_cap_10fdx"},
        {{10, LINK_DUPLEX_HALF}, "adv_cap_10hdx"},
        {{0, LINK_DUPLEX_UNKNOWN}, NULL}
};

static ether_spdx_t lp_cap_spdx[] = {
        {{400000, LINK_DUPLEX_FULL}, "lp_cap_400gfdx"},
        {{200000, LINK_DUPLEX_FULL}, "lp_cap_200gfdx"},
        {{100000, LINK_DUPLEX_FULL}, "lp_cap_100gfdx"},
        {{50000, LINK_DUPLEX_FULL}, "lp_cap_50gfdx"},
        {{40000, LINK_DUPLEX_FULL}, "lp_cap_40gfdx"},
        {{25000, LINK_DUPLEX_FULL}, "lp_cap_25gfdx"},
        {{10000, LINK_DUPLEX_FULL}, "lp_cap_10gfdx"},
        {{5000, LINK_DUPLEX_FULL}, "lp_cap_5000fdx"},
        {{2500, LINK_DUPLEX_FULL}, "lp_cap_2500fdx"},
        {{1000, LINK_DUPLEX_FULL}, "lp_cap_1000fdx"},
        {{1000, LINK_DUPLEX_HALF}, "lp_cap_1000hdx"},
        {{100, LINK_DUPLEX_FULL}, "lp_cap_100fdx"},
        {{100, LINK_DUPLEX_HALF}, "lp_cap_100hdx"},
        {{10, LINK_DUPLEX_FULL}, "lp_cap_10fdx"},
        {{10, LINK_DUPLEX_HALF}, "lp_cap_10hdx"},
        {{0, LINK_DUPLEX_UNKNOWN}, NULL}
};

typedef struct attr_kstat_s {
        char *autoneg_stat;
        char *pause_stat;
        char *asmpause_stat;
        char *fault_stat;
        ether_spdx_t *spdx_stat;
} attr_kstat_t;

static attr_kstat_t attrstat[] =  {
        {"link_autoneg",        /* current */
            "link_pause",       "link_asmpause",        NULL,
            NULL},

        {"cap_autoneg",         /* capable */
            "cap_pause",        "cap_asmpause",         "cap_rem_fault",
            cap_spdx},

        {"adv_cap_autoneg",     /* advertised */
            "adv_cap_pause",    "adv_cap_asmpause",     "adv_rem_fault",
            adv_cap_spdx},

        {"lp_cap_autoneg",      /* peer advertised */
            "lp_cap_pause",     "lp_cap_asmpause",      "lp_rem_fault",
            lp_cap_spdx}
};

/*
 * Get the speed-duplex stats specified in the ether_spdx_t table passed in
 * by querying the appropriate kstat for each entry in the table.
 */
static dladm_status_t
i_dladm_get_spdx(dladm_handle_t handle, datalink_id_t linkid,
    dladm_ether_attr_t *eattr, ether_spdx_t *spdx_stat)
{
        int             i, nspdx = 0;
        uint32_t        speed;
        dladm_status_t  status;
        void            *ptr;

        eattr->le_spdx = NULL;
        for (i = 0; spdx_stat[i].eth_spdx_stat_name != NULL; i++) {
                if ((status = dladm_get_single_mac_stat(handle, linkid,
                    spdx_stat[i].eth_spdx_stat_name,
                    KSTAT_DATA_UINT32, &speed)) != DLADM_STATUS_OK) {

                        if (status == DLADM_STATUS_NOTFOUND) {
                                /*
                                 * Missing statistic.
                                 * Skip this one and try the rest.
                                 */
                                continue;
                        } else {
                                free(eattr->le_spdx);
                                eattr->le_num_spdx = 0;
                                return (status);
                        }
                }
                if (speed == 0)
                        continue;
                nspdx++;
                ptr = realloc(eattr->le_spdx,
                    nspdx * sizeof (dladm_ether_spdx_t));
                if (ptr != NULL) {
                        eattr->le_spdx = ptr;
                } else {
                        free(eattr->le_spdx);
                        eattr->le_num_spdx = 0;
                        return (DLADM_STATUS_NOMEM);
                }
                eattr->le_spdx[nspdx - 1] = spdx_stat[i].eth_spdx;
        }
        eattr->le_num_spdx = nspdx;
        return (DLADM_STATUS_OK);
}

/*
 * Returns "yes" or "no" based on the autonegotion capabilities
 * for the parameter type indicated by ptype. The permissible
 * values for ptype are CURRENT, CAPABLE, ADV, PEERADV.
 */
char *
dladm_ether_autoneg2str(char *buf, size_t buflen, dladm_ether_info_t *eattr,
    int ptype)
{
        boolean_t autoneg = eattr->lei_attr[ptype].le_autoneg;

        (void) strlcpy(buf, (autoneg ? "yes" : "no"), buflen);
        return (buf);
}

/*
 * Returns {"bi", "tx", "none"} based on the flow-control capabilities
 * for the parameter type indicated by ptype. The permissible
 * values for ptype are CURRENT, CAPABLE, ADV, PEERADV.
 */
char *
dladm_ether_pause2str(char *buf, size_t buflen, dladm_ether_info_t *eattr,
    int ptype)
{
        boolean_t pause = eattr->lei_attr[ptype].le_pause;
        boolean_t asmpause = eattr->lei_attr[ptype].le_asmpause;

        if (pause)
                (void) strlcpy(buf, "bi", buflen);
        else if (asmpause)
                (void) strlcpy(buf, "tx", buflen);
        else
                (void) strlcpy(buf, "none", buflen);
        return (buf);
}

/*
 * For a given param type, parse the list of speed-duplex pairs in
 * the dladm_ether_info_t and return a  comma-separated string formatted
 * as <speed><speed-unit-char>-<duplex-chars> where <speed> is the value of
 * speed, in units specifid by the <speed-unit-char> which is one
 * of 'M' (Mbits/sec) or 'G' (Gigabits/sec).  The permissible values of
 * <duplex-chars> are 'u' (indicating duplex is "unknown") or one/both of
 * 'f', 'h' (indicating full-duplex and half-duplex respectively)
 */
extern char *
dladm_ether_spdx2str(char *buf, size_t buflen, dladm_ether_info_t *eattr,
    int ptype)
{
        uint_t          i, j;
        boolean_t       is_full, is_half;
        int             speed;
        char            speed_unit;
        char            tmpbuf[DLADM_STRSIZE];
        dladm_ether_spdx_t *spdx;
        uint32_t        nspdx;

        spdx = eattr->lei_attr[ptype].le_spdx;
        nspdx = eattr->lei_attr[ptype].le_num_spdx;
        for (i = 0; i < nspdx; i++) {

                speed = spdx[i].lesd_speed;

                /*
                 * if we have already covered this speed for
                 * the <other>-duplex case before this, skip it
                 */
                for (j = 0; j < i; j++) {
                        if (speed == spdx[j].lesd_speed)
                                break;
                }
                if (j < i)
                        continue;

                if ((speed % 1000) == 0) {
                        speed = speed/1000;
                        speed_unit = 'G';
                } else {
                        speed_unit = 'M';
                }
                (void) snprintf(tmpbuf, DLADM_STRSIZE, "%d%c",
                    speed, speed_unit);
                if (i > 0)
                        (void) strncat(buf, ",", buflen);
                (void) strncat(buf, tmpbuf, buflen);

                is_full = is_half = B_FALSE;
                /*
                 * Find all the supported duplex values for this speed.
                 */
                for (j = 0; j < nspdx; j++) {
                        if (spdx[j].lesd_speed != spdx[i].lesd_speed)
                                continue;
                        if (spdx[j].lesd_duplex == LINK_DUPLEX_FULL)
                                is_full = B_TRUE;
                        if (spdx[j].lesd_duplex == LINK_DUPLEX_HALF)
                                is_half = B_TRUE;
                }
                if (is_full && is_half)
                        (void) strncat(buf, "-fh", buflen);
                else if (is_full)
                        (void) strncat(buf, "-f", buflen);
                else if (is_half)
                        (void) strncat(buf, "-h", buflen);
        }
        return (buf);
}

/*
 * Extract Ethernet attributes of the link specified by linkid.
 * Information for the CURRENT, CAPABLE, ADV and PEERADV parameter
 * types is extracted into the lei_attr[] entries in the dladm_ether_info_t.
 * On succesful return, the memory allocated in this function should be
 * freed by calling dladm_ether_info_done().
 */
extern dladm_status_t
dladm_ether_info(dladm_handle_t handle, datalink_id_t linkid,
    dladm_ether_info_t *eattr)
{
        uint32_t        autoneg, pause, asmpause, fault;
        uint64_t        sp64;
        dladm_status_t  status;
        int             i;
        link_duplex_t   link_duplex;

        bzero(eattr, sizeof (*eattr));
        status = dladm_datalink_id2info(handle, linkid, NULL, NULL, NULL,
            eattr->lei_linkname, sizeof (eattr->lei_linkname));
        if (status != DLADM_STATUS_OK)
                goto bail;

        /* get current values of speed, duplex, state of link */
        eattr->lei_attr[CURRENT].le_num_spdx = 1;
        eattr->lei_attr[CURRENT].le_spdx = malloc(sizeof (dladm_ether_spdx_t));
        if (eattr->lei_attr[CURRENT].le_spdx == NULL) {
                status = DLADM_STATUS_NOMEM;
                goto bail;
        }

        if ((status = dladm_get_single_mac_stat(handle, linkid, "ifspeed",
            KSTAT_DATA_UINT64, &sp64)) != DLADM_STATUS_OK)
                goto bail;

        if ((status = dladm_get_single_mac_stat(handle, linkid, "link_duplex",
            KSTAT_DATA_UINT32, &link_duplex)) != DLADM_STATUS_OK)
                goto bail;

        eattr->lei_attr[CURRENT].le_spdx->lesd_speed = (int)(sp64/1000000ull);
        eattr->lei_attr[CURRENT].le_spdx->lesd_duplex = link_duplex;

        status = dladm_get_state(handle, linkid, &eattr->lei_state);
        if (status != DLADM_STATUS_OK)
                goto bail;

        /* get the auto, pause, asmpause, fault values */
        for (i = CURRENT; i <= PEERADV; i++)  {

                status = dladm_get_single_mac_stat(handle, linkid,
                    attrstat[i].autoneg_stat, KSTAT_DATA_UINT32, &autoneg);
                if (status != DLADM_STATUS_OK)
                        goto bail;

                status = dladm_get_single_mac_stat(handle, linkid,
                    attrstat[i].pause_stat, KSTAT_DATA_UINT32, &pause);
                if (status != DLADM_STATUS_OK)
                        goto bail;

                status = dladm_get_single_mac_stat(handle, linkid,
                    attrstat[i].asmpause_stat, KSTAT_DATA_UINT32, &asmpause);
                if (status != DLADM_STATUS_OK)
                        goto bail;

                eattr->lei_attr[i].le_autoneg = (autoneg != 0);
                eattr->lei_attr[i].le_pause = (pause != 0);
                eattr->lei_attr[i].le_asmpause = (asmpause != 0);

                if (i == CURRENT)
                        continue;
                status = dladm_get_single_mac_stat(handle, linkid,
                    attrstat[i].fault_stat, KSTAT_DATA_UINT32, &fault);
                if (status != DLADM_STATUS_OK)
                        goto bail;
                eattr->lei_attr[i].le_fault = (pause != 0);

                /* get all the supported speed/duplex values */
                status = i_dladm_get_spdx(handle, linkid, &eattr->lei_attr[i],
                    attrstat[i].spdx_stat);
                if (status != DLADM_STATUS_OK)
                        goto bail;
        }
        eattr->lei_attr[CURRENT].le_fault =
            eattr->lei_attr[ADV].le_fault || eattr->lei_attr[PEERADV].le_fault;
bail:
        if (status != DLADM_STATUS_OK)
                dladm_ether_info_done(eattr);
        return (status);
}

extern void
dladm_ether_info_done(dladm_ether_info_t *eattr)
{
        int i;

        for (i = CURRENT; i <= PEERADV; i++) {
                free(eattr->lei_attr[i].le_spdx);
                eattr->lei_attr[i].le_spdx = NULL;
        }
}