// SPDX-License-Identifier: GPL-2.0
/* Copyright(c) 2013 - 2019 Intel Corporation. */

#include "fm10k.h"

/**
 * fm10k_dcbnl_ieee_getets - get the ETS configuration for the device
 * @dev: netdev interface for the device
 * @ets: ETS structure to push configuration to
 **/
static int fm10k_dcbnl_ieee_getets(struct net_device *dev, struct ieee_ets *ets)
{
        int i;

        /* we support 8 TCs in all modes */
        ets->ets_cap = IEEE_8021QAZ_MAX_TCS;
        ets->cbs = 0;

        /* we only support strict priority and cannot do traffic shaping */
        memset(ets->tc_tx_bw, 0, sizeof(ets->tc_tx_bw));
        memset(ets->tc_rx_bw, 0, sizeof(ets->tc_rx_bw));
        memset(ets->tc_tsa, IEEE_8021QAZ_TSA_STRICT, sizeof(ets->tc_tsa));

        /* populate the prio map based on the netdev */
        for (i = 0; i < IEEE_8021QAZ_MAX_TCS; i++)
                ets->prio_tc[i] = netdev_get_prio_tc_map(dev, i);

        return 0;
}

/**
 * fm10k_dcbnl_ieee_setets - set the ETS configuration for the device
 * @dev: netdev interface for the device
 * @ets: ETS structure to pull configuration from
 **/
static int fm10k_dcbnl_ieee_setets(struct net_device *dev, struct ieee_ets *ets)
{
        u8 num_tc = 0;
        int i;

        /* verify type and determine num_tcs needed */
        for (i = 0; i < IEEE_8021QAZ_MAX_TCS; i++) {
                if (ets->tc_tx_bw[i] || ets->tc_rx_bw[i])
                        return -EINVAL;
                if (ets->tc_tsa[i] != IEEE_8021QAZ_TSA_STRICT)
                        return -EINVAL;
                if (ets->prio_tc[i] > num_tc)
                        num_tc = ets->prio_tc[i];
        }

        /* if requested TC is greater than 0 then num_tcs is max + 1 */
        if (num_tc)
                num_tc++;

        if (num_tc > IEEE_8021QAZ_MAX_TCS)
                return -EINVAL;

        /* update TC hardware mapping if necessary */
        if (num_tc != netdev_get_num_tc(dev)) {
                int err = fm10k_setup_tc(dev, num_tc);
                if (err)
                        return err;
        }

        /* update priority mapping */
        for (i = 0; i < IEEE_8021QAZ_MAX_TCS; i++)
                netdev_set_prio_tc_map(dev, i, ets->prio_tc[i]);

        return 0;
}

/**
 * fm10k_dcbnl_ieee_getpfc - get the PFC configuration for the device
 * @dev: netdev interface for the device
 * @pfc: PFC structure to push configuration to
 **/
static int fm10k_dcbnl_ieee_getpfc(struct net_device *dev, struct ieee_pfc *pfc)
{
        struct fm10k_intfc *interface = netdev_priv(dev);

        /* record flow control max count and state of TCs */
        pfc->pfc_cap = IEEE_8021QAZ_MAX_TCS;
        pfc->pfc_en = interface->pfc_en;

        return 0;
}

/**
 * fm10k_dcbnl_ieee_setpfc - set the PFC configuration for the device
 * @dev: netdev interface for the device
 * @pfc: PFC structure to pull configuration from
 **/
static int fm10k_dcbnl_ieee_setpfc(struct net_device *dev, struct ieee_pfc *pfc)
{
        struct fm10k_intfc *interface = netdev_priv(dev);

        /* record PFC configuration to interface */
        interface->pfc_en = pfc->pfc_en;

        /* if we are running update the drop_en state for all queues */
        if (netif_running(dev))
                fm10k_update_rx_drop_en(interface);

        return 0;
}

/**
 * fm10k_dcbnl_getdcbx - get the DCBX configuration for the device
 * @dev: netdev interface for the device
 *
 * Returns that we support only IEEE DCB for this interface
 **/
static u8 fm10k_dcbnl_getdcbx(struct net_device __always_unused *dev)
{
        return DCB_CAP_DCBX_HOST | DCB_CAP_DCBX_VER_IEEE;
}

/**
 * fm10k_dcbnl_setdcbx - get the DCBX configuration for the device
 * @dev: netdev interface for the device
 * @mode: new mode for this device
 *
 * Returns error on attempt to enable anything but IEEE DCB for this interface
 **/
static u8 fm10k_dcbnl_setdcbx(struct net_device __always_unused *dev, u8 mode)
{
        return (mode != (DCB_CAP_DCBX_HOST | DCB_CAP_DCBX_VER_IEEE)) ? 1 : 0;
}

static const struct dcbnl_rtnl_ops fm10k_dcbnl_ops = {
        .ieee_getets    = fm10k_dcbnl_ieee_getets,
        .ieee_setets    = fm10k_dcbnl_ieee_setets,
        .ieee_getpfc    = fm10k_dcbnl_ieee_getpfc,
        .ieee_setpfc    = fm10k_dcbnl_ieee_setpfc,

        .getdcbx        = fm10k_dcbnl_getdcbx,
        .setdcbx        = fm10k_dcbnl_setdcbx,
};

/**
 * fm10k_dcbnl_set_ops - Configures dcbnl ops pointer for netdev
 * @dev: netdev interface for the device
 *
 * Enables PF for DCB by assigning DCBNL ops pointer.
 **/
void fm10k_dcbnl_set_ops(struct net_device *dev)
{
        struct fm10k_intfc *interface = netdev_priv(dev);
        struct fm10k_hw *hw = &interface->hw;

        if (hw->mac.type == fm10k_mac_pf)
                dev->dcbnl_ops = &fm10k_dcbnl_ops;
}