/*-
 * Broadcom NetXtreme-C/E network driver.
 *
 * Copyright (c) 2024 Broadcom, All Rights Reserved.
 * The term Broadcom refers to Broadcom Limited and/or its subsidiaries
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS'
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
 * THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <sys/endian.h>
#include <linux/errno.h>
#include <linux/bitops.h>

#include "bnxt.h"
#include "bnxt_hwrm.h"
#include "bnxt_dcb.h"
#include "hsi_struct_def.h"

static int
bnxt_tx_queue_to_tc(struct bnxt_softc *softc, uint8_t queue_id)
{
        int i, j;

        for (i = 0; i < softc->max_tc; i++) {
                if (softc->tx_q_info[i].queue_id == queue_id) {
                        for (j = 0; j < softc->max_tc; j++) {
                                if (softc->tc_to_qidx[j] == i)
                                        return j;
                        }
                }
        }
        return -EINVAL;
}

static int
bnxt_hwrm_queue_pri2cos_cfg(struct bnxt_softc *softc,
                                       struct bnxt_ieee_ets *ets,
                                       uint32_t path_dir)
{
        struct hwrm_queue_pri2cos_cfg_input req = {0};
        struct bnxt_queue_info *q_info;
        uint8_t *pri2cos;
        int i;

        bnxt_hwrm_cmd_hdr_init(softc, &req, HWRM_QUEUE_PRI2COS_CFG);

        req.flags = htole32(path_dir | HWRM_QUEUE_PRI2COS_CFG_INPUT_FLAGS_IVLAN);
        if (path_dir == HWRM_QUEUE_PRI2COS_CFG_INPUT_FLAGS_PATH_BIDIR ||
            path_dir == HWRM_QUEUE_PRI2COS_CFG_INPUT_FLAGS_PATH_TX)
                q_info = softc->tx_q_info;
        else
                q_info = softc->rx_q_info;
        pri2cos = &req.pri0_cos_queue_id;
        for (i = 0; i < BNXT_IEEE_8021QAZ_MAX_TCS; i++) {
                uint8_t qidx;

                req.enables |= htole32(HWRM_QUEUE_PRI2COS_CFG_INPUT_ENABLES_PRI0_COS_QUEUE_ID << i);

                qidx = softc->tc_to_qidx[ets->prio_tc[i]];
                pri2cos[i] = q_info[qidx].queue_id;
        }
        return _hwrm_send_message(softc, &req, sizeof(req));
}

static int
bnxt_hwrm_queue_pri2cos_qcfg(struct bnxt_softc *softc, struct bnxt_ieee_ets *ets)
{
        struct hwrm_queue_pri2cos_qcfg_output *resp =
                (void *)softc->hwrm_cmd_resp.idi_vaddr;
        struct hwrm_queue_pri2cos_qcfg_input req = {0};
        int rc;

        bnxt_hwrm_cmd_hdr_init(softc, &req, HWRM_QUEUE_PRI2COS_QCFG);

        req.flags = htole32(HWRM_QUEUE_PRI2COS_QCFG_INPUT_FLAGS_IVLAN);
        rc = _hwrm_send_message(softc, &req, sizeof(req));
        if (!rc) {
                uint8_t *pri2cos = &resp->pri0_cos_queue_id;
                int i;

                for (i = 0; i < BNXT_IEEE_8021QAZ_MAX_TCS; i++) {
                        uint8_t queue_id = pri2cos[i];
                        int tc;

                        tc = bnxt_tx_queue_to_tc(softc, queue_id);
                        if (tc >= 0)
                                ets->prio_tc[i] = tc;
                }
        }
        return rc;
}

static int
bnxt_hwrm_queue_cos2bw_cfg(struct bnxt_softc *softc, struct bnxt_ieee_ets *ets,
                                      uint8_t max_tc)
{
        struct hwrm_queue_cos2bw_cfg_input req = {0};
        struct bnxt_cos2bw_cfg cos2bw;
        void *data;
        int i;

        bnxt_hwrm_cmd_hdr_init(softc, &req, HWRM_QUEUE_COS2BW_CFG);

        for (i = 0; i < max_tc; i++) {
                uint8_t qidx = softc->tc_to_qidx[i];

                req.enables |=
                        htole32(HWRM_QUEUE_COS2BW_CFG_INPUT_ENABLES_COS_QUEUE_ID0_VALID << qidx);

                memset(&cos2bw, 0, sizeof(cos2bw));
                cos2bw.queue_id = softc->tx_q_info[qidx].queue_id;
                if (ets->tc_tsa[i] == BNXT_IEEE_8021QAZ_TSA_STRICT) {
                        cos2bw.tsa =
                                HWRM_QUEUE_COS2BW_QCFG_OUTPUT_QUEUE_ID0_TSA_ASSIGN_SP;
                        cos2bw.pri_lvl = i;
                } else {
                        cos2bw.tsa =
                                HWRM_QUEUE_COS2BW_QCFG_OUTPUT_QUEUE_ID0_TSA_ASSIGN_ETS;
                        cos2bw.bw_weight = ets->tc_tx_bw[i];
                        /* older firmware requires min_bw to be set to the
                         * same weight value in percent.
                         */
                        if (BNXT_FW_MAJ(softc) < 218) {
                                cos2bw.min_bw =
                                        htole32((ets->tc_tx_bw[i] * 100) |
                                                    BW_VALUE_UNIT_PERCENT1_100);
                        }
                }
                data = &req.unused_0 + qidx * (sizeof(cos2bw) - 4);
                memcpy(data, &cos2bw.queue_id, sizeof(cos2bw) - 4);
                if (qidx == 0) {
                        req.queue_id0 = cos2bw.queue_id;
                        req.unused_0 = 0;
                }
        }
        return _hwrm_send_message(softc, &req, sizeof(req));
}

static int
bnxt_hwrm_queue_cos2bw_qcfg(struct bnxt_softc *softc, struct bnxt_ieee_ets *ets)
{
        struct hwrm_queue_cos2bw_qcfg_output *resp =
                (void *)softc->hwrm_cmd_resp.idi_vaddr;
        struct hwrm_queue_cos2bw_qcfg_input req = {0};
        struct bnxt_cos2bw_cfg cos2bw;
        uint8_t *data;
        int rc, i;

        bnxt_hwrm_cmd_hdr_init(softc, &req, HWRM_QUEUE_COS2BW_QCFG);

        rc = _hwrm_send_message(softc, &req, sizeof(req));
        if (rc) {
                return rc;
        }

        data = &resp->queue_id0 + offsetof(struct bnxt_cos2bw_cfg, queue_id);
        for (i = 0; i < softc->max_tc; i++, data += sizeof(cos2bw.cfg)) {
                int tc;

                memcpy(&cos2bw.cfg, data, sizeof(cos2bw.cfg));
                if (i == 0)
                        cos2bw.queue_id = resp->queue_id0;

                tc = bnxt_tx_queue_to_tc(softc, cos2bw.queue_id);
                if (tc < 0)
                        continue;

                if (cos2bw.tsa == HWRM_QUEUE_COS2BW_QCFG_OUTPUT_QUEUE_ID0_TSA_ASSIGN_SP) {
                        ets->tc_tsa[tc] = BNXT_IEEE_8021QAZ_TSA_STRICT;
                } else {
                        ets->tc_tsa[tc] = BNXT_IEEE_8021QAZ_TSA_ETS;
                        ets->tc_tx_bw[tc] = cos2bw.bw_weight;
                }
        }
        return 0;
}

static int
bnxt_queue_remap(struct bnxt_softc *softc, unsigned int lltc_mask)
{
        unsigned long qmap = 0;
        int max = softc->max_tc;
        int i, j, rc;

        /* Assign lossless TCs first */
        for (i = 0, j = 0; i < max; ) {
                if (lltc_mask & (1 << i)) {
                        if (BNXT_LLQ(softc->rx_q_info[j].queue_profile)) {
                                softc->tc_to_qidx[i] = j;
                                __set_bit(j, &qmap);
                                i++;
                        }
                        j++;
                        continue;
                }
                i++;
        }

        for (i = 0, j = 0; i < max; i++) {
                if (lltc_mask & (1 << i))
                        continue;
                j = find_next_zero_bit(&qmap, max, j);
                softc->tc_to_qidx[i] = j;
                __set_bit(j, &qmap);
                j++;
        }

        if (softc->ieee_ets) {
                rc = bnxt_hwrm_queue_cos2bw_cfg(softc, softc->ieee_ets, softc->max_tc);
                if (rc) {
                        device_printf(softc->dev, "failed to config BW, rc = %d\n", rc);
                        return rc;
                }
                rc = bnxt_hwrm_queue_pri2cos_cfg(softc, softc->ieee_ets,
                                                 HWRM_QUEUE_PRI2COS_CFG_INPUT_FLAGS_PATH_BIDIR);
                if (rc) {
                        device_printf(softc->dev, "failed to config prio, rc = %d\n", rc);
                        return rc;
                }
        }
        return 0;
}

static int
bnxt_hwrm_queue_pfc_cfg(struct bnxt_softc *softc, struct bnxt_ieee_pfc *pfc)
{
        struct hwrm_queue_pfcenable_cfg_input req = {0};
        struct bnxt_ieee_ets *my_ets = softc->ieee_ets;
        unsigned int tc_mask = 0, pri_mask = 0;
        uint8_t i, pri, lltc_count = 0;
        bool need_q_remap = false;

        if (!my_ets)
                return -EINVAL;

        for (i = 0; i < softc->max_tc; i++) {
                for (pri = 0; pri < BNXT_IEEE_8021QAZ_MAX_TCS; pri++) {
                        if ((pfc->pfc_en & (1 << pri)) &&
                            (my_ets->prio_tc[pri] == i)) {
                                pri_mask |= 1 << pri;
                                tc_mask |= 1 << i;
                        }
                }
                if (tc_mask & (1 << i))
                        lltc_count++;
        }

        if (lltc_count > softc->max_lltc) {
                device_printf(softc->dev,
                               "Hardware doesn't support %d lossless queues "
                               "to configure PFC (cap %d)\n", lltc_count, softc->max_lltc);
                return -EINVAL;
        }

        for (i = 0; i < softc->max_tc; i++) {
                if (tc_mask & (1 << i)) {
                        uint8_t qidx = softc->tc_to_qidx[i];

                        if (!BNXT_LLQ(softc->rx_q_info[qidx].queue_profile)) {
                                need_q_remap = true;
                                break;
                        }
                }
        }

        if (need_q_remap)
                bnxt_queue_remap(softc, tc_mask);

        bnxt_hwrm_cmd_hdr_init(softc, &req, HWRM_QUEUE_PFCENABLE_CFG);

        req.flags = htole32(pri_mask);
        return _hwrm_send_message(softc, &req, sizeof(req));
}

static int
bnxt_hwrm_queue_pfc_qcfg(struct bnxt_softc *softc, struct bnxt_ieee_pfc *pfc)
{
        struct hwrm_queue_pfcenable_qcfg_output *resp =
                (void *)softc->hwrm_cmd_resp.idi_vaddr;
        struct hwrm_queue_pfcenable_qcfg_input req = {0};
        uint8_t pri_mask;
        int rc;

        bnxt_hwrm_cmd_hdr_init(softc, &req, HWRM_QUEUE_PFCENABLE_QCFG);

        rc = _hwrm_send_message(softc, &req, sizeof(req));
        if (rc) {
                return rc;
        }

        pri_mask = le32toh(resp->flags);
        pfc->pfc_en = pri_mask;
        return 0;
}

static int
bnxt_hwrm_get_dcbx_app(struct bnxt_softc *softc, struct bnxt_dcb_app *app,
    size_t nitems, int *num_inputs)
{
        struct hwrm_fw_get_structured_data_input get = {0};
        struct hwrm_struct_data_dcbx_app *fw_app;
        struct hwrm_struct_hdr *data;
        struct iflib_dma_info dma_data;
        size_t data_len;
        int rc, n, i;

        if (softc->hwrm_spec_code < 0x10601)
                return 0;

        if (BNXT_VF(softc))
                return 0;

        bnxt_hwrm_cmd_hdr_init(softc, &get, HWRM_FW_GET_STRUCTURED_DATA);

        n = BNXT_IEEE_8021QAZ_MAX_TCS;
        data_len = sizeof(*data) + sizeof(*fw_app) * n;
        rc = iflib_dma_alloc(softc->ctx, data_len, &dma_data,
                             BUS_DMA_NOWAIT);
        if (rc)
                return ENOMEM;
        get.dest_data_addr = htole64(dma_data.idi_paddr);
        get.structure_id = htole16(HWRM_STRUCT_HDR_STRUCT_ID_DCBX_APP);
        get.subtype = htole16(HWRM_STRUCT_DATA_SUBTYPE_HOST_OPERATIONAL);
        get.count = 0;
        rc = _hwrm_send_message(softc, &get, sizeof(get));
        if (rc)
                goto set_app_exit;

        data = (void *)dma_data.idi_vaddr;
        fw_app = (struct hwrm_struct_data_dcbx_app *)(data + 1);

        if (data->struct_id != htole16(HWRM_STRUCT_HDR_STRUCT_ID_DCBX_APP)) {
                rc = -ENODEV;
                goto set_app_exit;
        }

        n = data->count;
        for (i = 0; i < n && *num_inputs < nitems; i++, fw_app++) {
                app[*num_inputs].priority = fw_app->priority;
                app[*num_inputs].protocol = htobe16(fw_app->protocol_id);
                app[*num_inputs].selector = fw_app->protocol_selector;
                (*num_inputs)++;
        }

set_app_exit:
        iflib_dma_free(&dma_data);
        return rc;
}

static int
bnxt_hwrm_set_dcbx_app(struct bnxt_softc *softc, struct bnxt_dcb_app *app,
                                  bool add)
{
        struct hwrm_fw_set_structured_data_input set = {0};
        struct hwrm_fw_get_structured_data_input get = {0};
        struct hwrm_struct_data_dcbx_app *fw_app;
        struct hwrm_struct_hdr *data;
        struct iflib_dma_info dma_data;
        size_t data_len;
        int rc, n, i;

        if (softc->hwrm_spec_code < 0x10601)
                return 0;

        bnxt_hwrm_cmd_hdr_init(softc, &get, HWRM_FW_GET_STRUCTURED_DATA);

        n = BNXT_IEEE_8021QAZ_MAX_TCS;
        data_len = sizeof(*data) + sizeof(*fw_app) * n;
        rc = iflib_dma_alloc(softc->ctx, data_len, &dma_data,
                             BUS_DMA_NOWAIT);
        if (rc)
                return ENOMEM;
        get.dest_data_addr = htole64(dma_data.idi_paddr);
        get.structure_id = htole16(HWRM_STRUCT_HDR_STRUCT_ID_DCBX_APP);
        get.subtype = htole16(HWRM_STRUCT_DATA_SUBTYPE_HOST_OPERATIONAL);
        get.count = 0;
        rc = _hwrm_send_message(softc, &get, sizeof(get));
        if (rc)
                goto set_app_exit;

        data = (void *)dma_data.idi_vaddr;
        fw_app = (struct hwrm_struct_data_dcbx_app *)(data + 1);

        if (data->struct_id != htole16(HWRM_STRUCT_HDR_STRUCT_ID_DCBX_APP)) {
                rc = -ENODEV;
                goto set_app_exit;
        }

        n = data->count;
        for (i = 0; i < n; i++, fw_app++) {
                if (fw_app->protocol_id == htobe16(app->protocol) &&
                    fw_app->protocol_selector == app->selector &&
                    fw_app->priority == app->priority) {
                        if (add)
                                goto set_app_exit;
                        else
                                break;
                }
        }
        if (add) {
                /* append */
                n++;
                fw_app->protocol_id = htobe16(app->protocol);
                fw_app->protocol_selector = app->selector;
                fw_app->priority = app->priority;
                fw_app->valid = 1;
        } else {
                size_t len = 0;

                /* not found, nothing to delete */
                if (n == i)
                        goto set_app_exit;

                len = (n - 1 - i) * sizeof(*fw_app);
                if (len)
                        memmove(fw_app, fw_app + 1, len);
                n--;
                memset(fw_app + n, 0, sizeof(*fw_app));
        }
        data->count = n;
        data->len = htole16(sizeof(*fw_app) * n);
        data->subtype = htole16(HWRM_STRUCT_DATA_SUBTYPE_HOST_OPERATIONAL);

        bnxt_hwrm_cmd_hdr_init(softc, &set, HWRM_FW_SET_STRUCTURED_DATA);

        set.src_data_addr = htole64(dma_data.idi_paddr);
        set.data_len = htole16(sizeof(*data) + sizeof(*fw_app) * n);
        set.hdr_cnt = 1;
        rc = _hwrm_send_message(softc, &set, sizeof(set));

set_app_exit:
        iflib_dma_free(&dma_data);
        return rc;
}

static int
bnxt_hwrm_queue_dscp_qcaps(struct bnxt_softc *softc)
{
        struct hwrm_queue_dscp_qcaps_output *resp =
                (void *)softc->hwrm_cmd_resp.idi_vaddr;
        struct hwrm_queue_dscp_qcaps_input req = {0};
        int rc;

        softc->max_dscp_value = 0;
        if (softc->hwrm_spec_code < 0x10800 || BNXT_VF(softc))
                return 0;

        bnxt_hwrm_cmd_hdr_init(softc, &req, HWRM_QUEUE_DSCP_QCAPS);

        rc = _hwrm_send_message(softc, &req, sizeof(req));
        if (!rc) {
                softc->max_dscp_value = (1 << resp->num_dscp_bits) - 1;
                if (softc->max_dscp_value < 0x3f)
                        softc->max_dscp_value = 0;
        }
        return rc;
}

static int
bnxt_hwrm_queue_dscp2pri_qcfg(struct bnxt_softc *softc, struct bnxt_dcb_app *app,
    size_t nitems, int *num_inputs)
{
        struct hwrm_queue_dscp2pri_qcfg_input req = {0};
        struct hwrm_queue_dscp2pri_qcfg_output *resp =
                (void *)softc->hwrm_cmd_resp.idi_vaddr;
        struct bnxt_dscp2pri_entry *dscp2pri;
        struct iflib_dma_info dma_data;
        int rc, entry_cnt;
        int i;

        if (softc->hwrm_spec_code < 0x10800)
                return 0;

        rc = iflib_dma_alloc(softc->ctx, sizeof(*dscp2pri) * 128, &dma_data,
                             BUS_DMA_NOWAIT);
        if (rc)
                return ENOMEM;

        dscp2pri = (void *)dma_data.idi_vaddr;

        bnxt_hwrm_cmd_hdr_init(softc, &req, HWRM_QUEUE_DSCP2PRI_QCFG);

        req.dest_data_addr = htole64(dma_data.idi_paddr);
        req.dest_data_buffer_size = htole16(sizeof(*dscp2pri) * 64);
        req.port_id = htole16(softc->pf.port_id);
        rc = _hwrm_send_message(softc, &req, sizeof(req));

        if (rc)
                goto end;

        entry_cnt =  le16toh(resp->entry_cnt);
        for (i = 0; i < entry_cnt && *num_inputs < nitems; i++) {
                app[*num_inputs].priority = dscp2pri[i].pri;
                app[*num_inputs].protocol = dscp2pri[i].dscp;
                app[*num_inputs].selector = BNXT_IEEE_8021QAZ_APP_SEL_DSCP;
                (*num_inputs)++;
        }

end:
        iflib_dma_free(&dma_data);
        return rc;
}

static int
bnxt_hwrm_queue_dscp2pri_cfg(struct bnxt_softc *softc, struct bnxt_dcb_app *app,
                             bool add)
{
        struct hwrm_queue_dscp2pri_cfg_input req = {0};
        struct bnxt_dscp2pri_entry *dscp2pri;
        struct iflib_dma_info dma_data;
        int rc;

        if (softc->hwrm_spec_code < 0x10800)
                return 0;

        rc = iflib_dma_alloc(softc->ctx, sizeof(*dscp2pri), &dma_data,
                             BUS_DMA_NOWAIT);
        if (rc)
                return ENOMEM;

        bnxt_hwrm_cmd_hdr_init(softc, &req, HWRM_QUEUE_DSCP2PRI_CFG);

        req.src_data_addr = htole64(dma_data.idi_paddr);
        dscp2pri = (void *)dma_data.idi_vaddr;
        dscp2pri->dscp = app->protocol;
        if (add)
                dscp2pri->mask = 0x3f;
        else
                dscp2pri->mask = 0;
        dscp2pri->pri = app->priority;
        req.entry_cnt = htole16(1);
        req.port_id = htole16(softc->pf.port_id);
        rc = _hwrm_send_message(softc, &req, sizeof(req));

        iflib_dma_free(&dma_data);
        return rc;
}

static int
bnxt_ets_validate(struct bnxt_softc *softc, struct bnxt_ieee_ets *ets, uint8_t *tc)
{
        int total_ets_bw = 0;
        bool zero = false;
        uint8_t max_tc = 0;
        int i;

        for (i = 0; i < BNXT_IEEE_8021QAZ_MAX_TCS; i++) {
                if (ets->prio_tc[i] > softc->max_tc) {
                        device_printf(softc->dev, "priority to TC mapping exceeds TC count %d\n",
                                   ets->prio_tc[i]);
                        return -EINVAL;
                }
                if (ets->prio_tc[i] > max_tc)
                        max_tc = ets->prio_tc[i];

                if ((ets->tc_tx_bw[i] || ets->tc_tsa[i]) && i > softc->max_tc)
                        return -EINVAL;

                switch (ets->tc_tsa[i]) {
                case BNXT_IEEE_8021QAZ_TSA_STRICT:
                        break;
                case BNXT_IEEE_8021QAZ_TSA_ETS:
                        total_ets_bw += ets->tc_tx_bw[i];
                        zero = zero || !ets->tc_tx_bw[i];
                        break;
                default:
                        return -ENOTSUPP;
                }
        }
        if (total_ets_bw > 100) {
                device_printf(softc->dev, "rejecting ETS config exceeding available bandwidth\n");
                return -EINVAL;
        }
        if (zero && total_ets_bw == 100) {
                device_printf(softc->dev, "rejecting ETS config starving a TC\n");
                return -EINVAL;
        }

        if (max_tc >= softc->max_tc)
                *tc = softc->max_tc;
        else
                *tc = max_tc + 1;
        return 0;
}

int
bnxt_dcb_ieee_getets(struct bnxt_softc *softc, struct bnxt_ieee_ets *ets)
{
        struct bnxt_ieee_ets *my_ets = softc->ieee_ets;
        int rc;

        if (!my_ets)
                return 0;

        rc = bnxt_hwrm_queue_cos2bw_qcfg(softc, my_ets);
        if (rc)
                goto error;
        rc = bnxt_hwrm_queue_pri2cos_qcfg(softc, my_ets);
        if (rc)
                goto error;

        if (ets) {
                ets->cbs = my_ets->cbs;
                ets->ets_cap = softc->max_tc;
                memcpy(ets->tc_tx_bw, my_ets->tc_tx_bw, sizeof(ets->tc_tx_bw));
                memcpy(ets->tc_rx_bw, my_ets->tc_rx_bw, sizeof(ets->tc_rx_bw));
                memcpy(ets->tc_tsa, my_ets->tc_tsa, sizeof(ets->tc_tsa));
                memcpy(ets->prio_tc, my_ets->prio_tc, sizeof(ets->prio_tc));
        }
        return 0;
error:
        return rc;
}

int
bnxt_dcb_ieee_setets(struct bnxt_softc *softc, struct bnxt_ieee_ets *ets)
{
        uint8_t max_tc = 0;
        int rc;

        if (!(softc->dcbx_cap & BNXT_DCB_CAP_DCBX_VER_IEEE) ||
            !(softc->dcbx_cap & BNXT_DCB_CAP_DCBX_HOST))
                return -EINVAL;

        rc = bnxt_ets_validate(softc, ets, &max_tc);
        if (rc)
                return rc;

        rc = bnxt_hwrm_queue_cos2bw_cfg(softc, ets, max_tc);
        if (rc)
                goto error;

        if (!softc->is_asym_q) {
                rc = bnxt_hwrm_queue_pri2cos_cfg(softc, ets,
                                                 HWRM_QUEUE_PRI2COS_CFG_INPUT_FLAGS_PATH_BIDIR);
                if (rc)
                        goto error;
        } else {
                rc = bnxt_hwrm_queue_pri2cos_cfg(softc, ets,
                                                 HWRM_QUEUE_PRI2COS_QCFG_INPUT_FLAGS_PATH_TX);
                if (rc)
                        goto error;

                rc = bnxt_hwrm_queue_pri2cos_cfg(softc, ets,
                                                 HWRM_QUEUE_PRI2COS_QCFG_INPUT_FLAGS_PATH_RX);
                if (rc)
                        goto error;
        }

        memcpy(softc->ieee_ets, ets, sizeof(*ets));
        return 0;
error:
        return rc;
}

int
bnxt_dcb_ieee_getpfc(struct bnxt_softc *softc, struct bnxt_ieee_pfc *pfc)
{
        struct bnxt_ieee_pfc *my_pfc = softc->ieee_pfc;
        int rc;

        if (!my_pfc)
                return -1;

        pfc->pfc_cap = softc->max_lltc;

        rc = bnxt_hwrm_queue_pfc_qcfg(softc, my_pfc);
        if (rc)
                return 0;

        pfc->pfc_en = my_pfc->pfc_en;
        pfc->mbc = my_pfc->mbc;
        pfc->delay = my_pfc->delay;

        return 0;
}

int
bnxt_dcb_ieee_setpfc(struct bnxt_softc *softc, struct bnxt_ieee_pfc *pfc)
{
        struct bnxt_ieee_pfc *my_pfc = softc->ieee_pfc;
        int rc;

        if (!my_pfc)
                return -1;

        if (!(softc->dcbx_cap & BNXT_DCB_CAP_DCBX_VER_IEEE) ||
            !(softc->dcbx_cap & BNXT_DCB_CAP_DCBX_HOST) ||
            (softc->phy_flags & BNXT_PHY_FL_NO_PAUSE))
                return -EINVAL;

        rc = bnxt_hwrm_queue_pfc_cfg(softc, pfc);
        if (!rc)
                memcpy(my_pfc, pfc, sizeof(*my_pfc));

        return rc;
}

static int
bnxt_dcb_ieee_dscp_app_prep(struct bnxt_softc *softc, struct bnxt_dcb_app *app)
{
        if (app->selector == BNXT_IEEE_8021QAZ_APP_SEL_DSCP) {
                if (!softc->max_dscp_value)
                        return -ENOTSUPP;
                if (app->protocol > softc->max_dscp_value)
                        return -EINVAL;
        }
        return 0;
}

int
bnxt_dcb_ieee_setapp(struct bnxt_softc *softc, struct bnxt_dcb_app *app)
{
        int rc;


        if (!(softc->dcbx_cap & BNXT_DCB_CAP_DCBX_VER_IEEE) ||
            !(softc->dcbx_cap & BNXT_DCB_CAP_DCBX_HOST))
                return -EINVAL;

        rc = bnxt_dcb_ieee_dscp_app_prep(softc, app);
        if (rc)
                return rc;

        if ((app->selector == BNXT_IEEE_8021QAZ_APP_SEL_ETHERTYPE &&
             app->protocol == ETH_P_ROCE) ||
            (app->selector == BNXT_IEEE_8021QAZ_APP_SEL_DGRAM &&
             app->protocol == ROCE_V2_UDP_DPORT))
                rc = bnxt_hwrm_set_dcbx_app(softc, app, true);

        if (app->selector == BNXT_IEEE_8021QAZ_APP_SEL_DSCP)
                rc = bnxt_hwrm_queue_dscp2pri_cfg(softc, app, true);

        return rc;
}

int
bnxt_dcb_ieee_delapp(struct bnxt_softc *softc, struct bnxt_dcb_app *app)
{
        int rc;

        if (!(softc->dcbx_cap & BNXT_DCB_CAP_DCBX_VER_IEEE) ||
            !(softc->dcbx_cap & BNXT_DCB_CAP_DCBX_HOST))
                return -EINVAL;

        rc = bnxt_dcb_ieee_dscp_app_prep(softc, app);
        if (rc)
                return rc;

        if ((app->selector == BNXT_IEEE_8021QAZ_APP_SEL_ETHERTYPE &&
             app->protocol == ETH_P_ROCE) ||
            (app->selector == BNXT_IEEE_8021QAZ_APP_SEL_DGRAM &&
             app->protocol == ROCE_V2_UDP_DPORT))
                rc = bnxt_hwrm_set_dcbx_app(softc, app, false);

        if (app->selector == BNXT_IEEE_8021QAZ_APP_SEL_DSCP)
                rc = bnxt_hwrm_queue_dscp2pri_cfg(softc, app, false);

        return rc;
}

int
bnxt_dcb_ieee_listapp(struct bnxt_softc *softc, struct bnxt_dcb_app *app,
    size_t nitems, int *num_inputs)
{
        bnxt_hwrm_get_dcbx_app(softc, app, nitems, num_inputs);
        bnxt_hwrm_queue_dscp2pri_qcfg(softc, app, nitems, num_inputs);

        return 0;
}

uint8_t
bnxt_dcb_getdcbx(struct bnxt_softc *softc)
{
        return softc->dcbx_cap;
}

uint8_t
bnxt_dcb_setdcbx(struct bnxt_softc *softc, uint8_t mode)
{
        /* All firmware DCBX settings are set in NVRAM */
        if (softc->dcbx_cap & BNXT_DCB_CAP_DCBX_LLD_MANAGED)
                return 1;

        /*
         * Do't allow editing CAP_DCBX_LLD_MANAGED since it is driven
         * based on FUNC_QCFG_OUTPUT_FLAGS_FW_DCBX_AGENT_ENABLED
         */
        if ((softc->dcbx_cap & BNXT_DCB_CAP_DCBX_LLD_MANAGED) !=
            (mode & BNXT_DCB_CAP_DCBX_LLD_MANAGED))
                return 1;

        if (mode & BNXT_DCB_CAP_DCBX_HOST) {
                if (BNXT_VF(softc) || (softc->fw_cap & BNXT_FW_CAP_LLDP_AGENT))
                        return 1;

                /* only support BNXT_IEEE */
                if ((mode & BNXT_DCB_CAP_DCBX_VER_CEE) ||
                    !(mode & BNXT_DCB_CAP_DCBX_VER_IEEE))
                        return 1;
        }

        if (mode == softc->dcbx_cap)
                return 0;

        softc->dcbx_cap = mode;
        return 0;
}

void
bnxt_dcb_init(struct bnxt_softc *softc)
{
        struct bnxt_ieee_ets ets = {0};
        struct bnxt_ieee_pfc pfc = {0};

        softc->dcbx_cap = 0;

        if (softc->hwrm_spec_code < 0x10501)
                return;

        softc->ieee_ets = malloc(sizeof(struct bnxt_ieee_ets), M_DEVBUF, M_NOWAIT | M_ZERO);
        if (!softc->ieee_ets)
                return;

        softc->ieee_pfc = malloc(sizeof(struct bnxt_ieee_pfc), M_DEVBUF, M_NOWAIT | M_ZERO);
        if (!softc->ieee_pfc)
                return;

        bnxt_hwrm_queue_dscp_qcaps(softc);
        softc->dcbx_cap = BNXT_DCB_CAP_DCBX_VER_IEEE;
        if (BNXT_PF(softc) && !(softc->fw_cap & BNXT_FW_CAP_LLDP_AGENT))
                softc->dcbx_cap |= BNXT_DCB_CAP_DCBX_HOST;
        else if (softc->fw_cap & BNXT_FW_CAP_DCBX_AGENT)
                softc->dcbx_cap |= BNXT_DCB_CAP_DCBX_LLD_MANAGED;

        bnxt_dcb_ieee_setets(softc, &ets);
        bnxt_dcb_ieee_setpfc(softc, &pfc);

}

void
bnxt_dcb_free(struct bnxt_softc *softc)
{
        free(softc->ieee_ets, M_DEVBUF);
        softc->ieee_ets = NULL;
        free(softc->ieee_pfc, M_DEVBUF);
        softc->ieee_pfc = NULL;
}