// SPDX-License-Identifier: (GPL-2.0-or-later OR BSD-3-Clause)
/*
 * Copyright (c) 2014-2025, Advanced Micro Devices, Inc.
 * Copyright (c) 2014, Synopsys, Inc.
 * All rights reserved
 */

#include <linux/netdevice.h>
#include <net/dcbnl.h>

#include "xgbe.h"
#include "xgbe-common.h"

static int xgbe_dcb_ieee_getets(struct net_device *netdev,
                                struct ieee_ets *ets)
{
        struct xgbe_prv_data *pdata = netdev_priv(netdev);

        /* Set number of supported traffic classes */
        ets->ets_cap = pdata->hw_feat.tc_cnt;

        if (pdata->ets) {
                ets->cbs = pdata->ets->cbs;
                memcpy(ets->tc_tx_bw, pdata->ets->tc_tx_bw,
                       sizeof(ets->tc_tx_bw));
                memcpy(ets->tc_tsa, pdata->ets->tc_tsa,
                       sizeof(ets->tc_tsa));
                memcpy(ets->prio_tc, pdata->ets->prio_tc,
                       sizeof(ets->prio_tc));
        }

        return 0;
}

static int xgbe_dcb_ieee_setets(struct net_device *netdev,
                                struct ieee_ets *ets)
{
        struct xgbe_prv_data *pdata = netdev_priv(netdev);
        unsigned int i, tc_ets, tc_ets_weight;
        u8 max_tc = 0;

        tc_ets = 0;
        tc_ets_weight = 0;
        for (i = 0; i < IEEE_8021QAZ_MAX_TCS; i++) {
                netif_dbg(pdata, drv, netdev,
                          "TC%u: tx_bw=%hhu, rx_bw=%hhu, tsa=%hhu\n", i,
                          ets->tc_tx_bw[i], ets->tc_rx_bw[i],
                          ets->tc_tsa[i]);
                netif_dbg(pdata, drv, netdev, "PRIO%u: TC=%hhu\n", i,
                          ets->prio_tc[i]);

                max_tc = max_t(u8, max_tc, ets->prio_tc[i]);
                if ((ets->tc_tx_bw[i] || ets->tc_tsa[i]))
                        max_tc = max_t(u8, max_tc, i);

                switch (ets->tc_tsa[i]) {
                case IEEE_8021QAZ_TSA_STRICT:
                        break;
                case IEEE_8021QAZ_TSA_ETS:
                        tc_ets = 1;
                        tc_ets_weight += ets->tc_tx_bw[i];
                        break;
                default:
                        netif_err(pdata, drv, netdev,
                                  "unsupported TSA algorithm (%hhu)\n",
                                  ets->tc_tsa[i]);
                        return -EINVAL;
                }
        }

        /* Check maximum traffic class requested */
        if (max_tc >= pdata->hw_feat.tc_cnt) {
                netif_err(pdata, drv, netdev,
                          "exceeded number of supported traffic classes\n");
                return -EINVAL;
        }

        /* Weights must add up to 100% */
        if (tc_ets && (tc_ets_weight != 100)) {
                netif_err(pdata, drv, netdev,
                          "sum of ETS algorithm weights is not 100 (%u)\n",
                          tc_ets_weight);
                return -EINVAL;
        }

        if (!pdata->ets) {
                pdata->ets = devm_kzalloc(pdata->dev, sizeof(*pdata->ets),
                                          GFP_KERNEL);
                if (!pdata->ets)
                        return -ENOMEM;
        }

        pdata->num_tcs = max_tc + 1;
        memcpy(pdata->ets, ets, sizeof(*pdata->ets));

        pdata->hw_if.config_dcb_tc(pdata);

        return 0;
}

static int xgbe_dcb_ieee_getpfc(struct net_device *netdev,
                                struct ieee_pfc *pfc)
{
        struct xgbe_prv_data *pdata = netdev_priv(netdev);

        /* Set number of supported PFC traffic classes */
        pfc->pfc_cap = pdata->hw_feat.tc_cnt;

        if (pdata->pfc) {
                pfc->pfc_en = pdata->pfc->pfc_en;
                pfc->mbc = pdata->pfc->mbc;
                pfc->delay = pdata->pfc->delay;
        }

        return 0;
}

static int xgbe_dcb_ieee_setpfc(struct net_device *netdev,
                                struct ieee_pfc *pfc)
{
        struct xgbe_prv_data *pdata = netdev_priv(netdev);

        netif_dbg(pdata, drv, netdev,
                  "cap=%d, en=%#x, mbc=%d, delay=%d\n",
                  pfc->pfc_cap, pfc->pfc_en, pfc->mbc, pfc->delay);

        /* Check PFC for supported number of traffic classes */
        if (pfc->pfc_en & ~((1 << pdata->hw_feat.tc_cnt) - 1)) {
                netif_err(pdata, drv, netdev,
                          "PFC requested for unsupported traffic class\n");
                return -EINVAL;
        }

        if (!pdata->pfc) {
                pdata->pfc = devm_kzalloc(pdata->dev, sizeof(*pdata->pfc),
                                          GFP_KERNEL);
                if (!pdata->pfc)
                        return -ENOMEM;
        }

        memcpy(pdata->pfc, pfc, sizeof(*pdata->pfc));

        pdata->hw_if.config_dcb_pfc(pdata);

        return 0;
}

static u8 xgbe_dcb_getdcbx(struct net_device *netdev)
{
        return DCB_CAP_DCBX_HOST | DCB_CAP_DCBX_VER_IEEE;
}

static u8 xgbe_dcb_setdcbx(struct net_device *netdev, u8 dcbx)
{
        struct xgbe_prv_data *pdata = netdev_priv(netdev);
        u8 support = xgbe_dcb_getdcbx(netdev);

        netif_dbg(pdata, drv, netdev, "DCBX=%#hhx\n", dcbx);

        if (dcbx & ~support)
                return 1;

        if ((dcbx & support) != support)
                return 1;

        return 0;
}

static const struct dcbnl_rtnl_ops xgbe_dcbnl_ops = {
        /* IEEE 802.1Qaz std */
        .ieee_getets = xgbe_dcb_ieee_getets,
        .ieee_setets = xgbe_dcb_ieee_setets,
        .ieee_getpfc = xgbe_dcb_ieee_getpfc,
        .ieee_setpfc = xgbe_dcb_ieee_setpfc,

        /* DCBX configuration */
        .getdcbx     = xgbe_dcb_getdcbx,
        .setdcbx     = xgbe_dcb_setdcbx,
};

const struct dcbnl_rtnl_ops *xgbe_get_dcbnl_ops(void)
{
        return &xgbe_dcbnl_ops;
}