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

#include "ice.h"
#include "ice_base.h"
#include "ice_flow.h"
#include "ice_lib.h"
#include "ice_fltr.h"
#include "ice_dcb_lib.h"
#include "ice_devlink.h"

/**
 * ice_vsi_type_str - maps VSI type enum to string equivalents
 * @vsi_type: VSI type enum
 */
const char *ice_vsi_type_str(enum ice_vsi_type vsi_type)
{
        switch (vsi_type) {
        case ICE_VSI_PF:
                return "ICE_VSI_PF";
        case ICE_VSI_VF:
                return "ICE_VSI_VF";
        case ICE_VSI_CTRL:
                return "ICE_VSI_CTRL";
        case ICE_VSI_LB:
                return "ICE_VSI_LB";
        default:
                return "unknown";
        }
}

/**
 * ice_vsi_ctrl_all_rx_rings - Start or stop a VSI's Rx rings
 * @vsi: the VSI being configured
 * @ena: start or stop the Rx rings
 *
 * First enable/disable all of the Rx rings, flush any remaining writes, and
 * then verify that they have all been enabled/disabled successfully. This will
 * let all of the register writes complete when enabling/disabling the Rx rings
 * before waiting for the change in hardware to complete.
 */
static int ice_vsi_ctrl_all_rx_rings(struct ice_vsi *vsi, bool ena)
{
        int ret = 0;
        u16 i;

        for (i = 0; i < vsi->num_rxq; i++)
                ice_vsi_ctrl_one_rx_ring(vsi, ena, i, false);

        ice_flush(&vsi->back->hw);

        for (i = 0; i < vsi->num_rxq; i++) {
                ret = ice_vsi_wait_one_rx_ring(vsi, ena, i);
                if (ret)
                        break;
        }

        return ret;
}

/**
 * ice_vsi_alloc_arrays - Allocate queue and vector pointer arrays for the VSI
 * @vsi: VSI pointer
 *
 * On error: returns error code (negative)
 * On success: returns 0
 */
static int ice_vsi_alloc_arrays(struct ice_vsi *vsi)
{
        struct ice_pf *pf = vsi->back;
        struct device *dev;

        dev = ice_pf_to_dev(pf);

        /* allocate memory for both Tx and Rx ring pointers */
        vsi->tx_rings = devm_kcalloc(dev, vsi->alloc_txq,
                                     sizeof(*vsi->tx_rings), GFP_KERNEL);
        if (!vsi->tx_rings)
                return -ENOMEM;

        vsi->rx_rings = devm_kcalloc(dev, vsi->alloc_rxq,
                                     sizeof(*vsi->rx_rings), GFP_KERNEL);
        if (!vsi->rx_rings)
                goto err_rings;

        /* XDP will have vsi->alloc_txq Tx queues as well, so double the size */
        vsi->txq_map = devm_kcalloc(dev, (2 * vsi->alloc_txq),
                                    sizeof(*vsi->txq_map), GFP_KERNEL);

        if (!vsi->txq_map)
                goto err_txq_map;

        vsi->rxq_map = devm_kcalloc(dev, vsi->alloc_rxq,
                                    sizeof(*vsi->rxq_map), GFP_KERNEL);
        if (!vsi->rxq_map)
                goto err_rxq_map;

        /* There is no need to allocate q_vectors for a loopback VSI. */
        if (vsi->type == ICE_VSI_LB)
                return 0;

        /* allocate memory for q_vector pointers */
        vsi->q_vectors = devm_kcalloc(dev, vsi->num_q_vectors,
                                      sizeof(*vsi->q_vectors), GFP_KERNEL);
        if (!vsi->q_vectors)
                goto err_vectors;

        vsi->af_xdp_zc_qps = bitmap_zalloc(max_t(int, vsi->alloc_txq, vsi->alloc_rxq), GFP_KERNEL);
        if (!vsi->af_xdp_zc_qps)
                goto err_zc_qps;

        return 0;

err_zc_qps:
        devm_kfree(dev, vsi->q_vectors);
err_vectors:
        devm_kfree(dev, vsi->rxq_map);
err_rxq_map:
        devm_kfree(dev, vsi->txq_map);
err_txq_map:
        devm_kfree(dev, vsi->rx_rings);
err_rings:
        devm_kfree(dev, vsi->tx_rings);
        return -ENOMEM;
}

/**
 * ice_vsi_set_num_desc - Set number of descriptors for queues on this VSI
 * @vsi: the VSI being configured
 */
static void ice_vsi_set_num_desc(struct ice_vsi *vsi)
{
        switch (vsi->type) {
        case ICE_VSI_PF:
        case ICE_VSI_CTRL:
        case ICE_VSI_LB:
                /* a user could change the values of num_[tr]x_desc using
                 * ethtool -G so we should keep those values instead of
                 * overwriting them with the defaults.
                 */
                if (!vsi->num_rx_desc)
                        vsi->num_rx_desc = ICE_DFLT_NUM_RX_DESC;
                if (!vsi->num_tx_desc)
                        vsi->num_tx_desc = ICE_DFLT_NUM_TX_DESC;
                break;
        default:
                dev_dbg(ice_pf_to_dev(vsi->back), "Not setting number of Tx/Rx descriptors for VSI type %d\n",
                        vsi->type);
                break;
        }
}

/**
 * ice_vsi_set_num_qs - Set number of queues, descriptors and vectors for a VSI
 * @vsi: the VSI being configured
 * @vf_id: ID of the VF being configured
 *
 * Return 0 on success and a negative value on error
 */
static void ice_vsi_set_num_qs(struct ice_vsi *vsi, u16 vf_id)
{
        struct ice_pf *pf = vsi->back;
        struct ice_vf *vf = NULL;

        if (vsi->type == ICE_VSI_VF)
                vsi->vf_id = vf_id;
        else
                vsi->vf_id = ICE_INVAL_VFID;

        switch (vsi->type) {
        case ICE_VSI_PF:
                if (vsi->req_txq) {
                        vsi->alloc_txq = vsi->req_txq;
                        vsi->num_txq = vsi->req_txq;
                } else {
                        vsi->alloc_txq = min3(pf->num_lan_msix,
                                              ice_get_avail_txq_count(pf),
                                              (u16)num_online_cpus());
                }

                pf->num_lan_tx = vsi->alloc_txq;

                /* only 1 Rx queue unless RSS is enabled */
                if (!test_bit(ICE_FLAG_RSS_ENA, pf->flags)) {
                        vsi->alloc_rxq = 1;
                } else {
                        if (vsi->req_rxq) {
                                vsi->alloc_rxq = vsi->req_rxq;
                                vsi->num_rxq = vsi->req_rxq;
                        } else {
                                vsi->alloc_rxq = min3(pf->num_lan_msix,
                                                      ice_get_avail_rxq_count(pf),
                                                      (u16)num_online_cpus());
                        }
                }

                pf->num_lan_rx = vsi->alloc_rxq;

                vsi->num_q_vectors = min_t(int, pf->num_lan_msix,
                                           max_t(int, vsi->alloc_rxq,
                                                 vsi->alloc_txq));
                break;
        case ICE_VSI_VF:
                vf = &pf->vf[vsi->vf_id];
                if (vf->num_req_qs)
                        vf->num_vf_qs = vf->num_req_qs;
                vsi->alloc_txq = vf->num_vf_qs;
                vsi->alloc_rxq = vf->num_vf_qs;
                /* pf->num_msix_per_vf includes (VF miscellaneous vector +
                 * data queue interrupts). Since vsi->num_q_vectors is number
                 * of queues vectors, subtract 1 (ICE_NONQ_VECS_VF) from the
                 * original vector count
                 */
                vsi->num_q_vectors = pf->num_msix_per_vf - ICE_NONQ_VECS_VF;
                break;
        case ICE_VSI_CTRL:
                vsi->alloc_txq = 1;
                vsi->alloc_rxq = 1;
                vsi->num_q_vectors = 1;
                break;
        case ICE_VSI_LB:
                vsi->alloc_txq = 1;
                vsi->alloc_rxq = 1;
                break;
        default:
                dev_warn(ice_pf_to_dev(pf), "Unknown VSI type %d\n", vsi->type);
                break;
        }

        ice_vsi_set_num_desc(vsi);
}

/**
 * ice_get_free_slot - get the next non-NULL location index in array
 * @array: array to search
 * @size: size of the array
 * @curr: last known occupied index to be used as a search hint
 *
 * void * is being used to keep the functionality generic. This lets us use this
 * function on any array of pointers.
 */
static int ice_get_free_slot(void *array, int size, int curr)
{
        int **tmp_array = (int **)array;
        int next;

        if (curr < (size - 1) && !tmp_array[curr + 1]) {
                next = curr + 1;
        } else {
                int i = 0;

                while ((i < size) && (tmp_array[i]))
                        i++;
                if (i == size)
                        next = ICE_NO_VSI;
                else
                        next = i;
        }
        return next;
}

/**
 * ice_vsi_delete - delete a VSI from the switch
 * @vsi: pointer to VSI being removed
 */
static void ice_vsi_delete(struct ice_vsi *vsi)
{
        struct ice_pf *pf = vsi->back;
        struct ice_vsi_ctx *ctxt;
        enum ice_status status;

        ctxt = kzalloc(sizeof(*ctxt), GFP_KERNEL);
        if (!ctxt)
                return;

        if (vsi->type == ICE_VSI_VF)
                ctxt->vf_num = vsi->vf_id;
        ctxt->vsi_num = vsi->vsi_num;

        memcpy(&ctxt->info, &vsi->info, sizeof(ctxt->info));

        status = ice_free_vsi(&pf->hw, vsi->idx, ctxt, false, NULL);
        if (status)
                dev_err(ice_pf_to_dev(pf), "Failed to delete VSI %i in FW - error: %s\n",
                        vsi->vsi_num, ice_stat_str(status));

        kfree(ctxt);
}

/**
 * ice_vsi_free_arrays - De-allocate queue and vector pointer arrays for the VSI
 * @vsi: pointer to VSI being cleared
 */
static void ice_vsi_free_arrays(struct ice_vsi *vsi)
{
        struct ice_pf *pf = vsi->back;
        struct device *dev;

        dev = ice_pf_to_dev(pf);

        if (vsi->af_xdp_zc_qps) {
                bitmap_free(vsi->af_xdp_zc_qps);
                vsi->af_xdp_zc_qps = NULL;
        }
        /* free the ring and vector containers */
        if (vsi->q_vectors) {
                devm_kfree(dev, vsi->q_vectors);
                vsi->q_vectors = NULL;
        }
        if (vsi->tx_rings) {
                devm_kfree(dev, vsi->tx_rings);
                vsi->tx_rings = NULL;
        }
        if (vsi->rx_rings) {
                devm_kfree(dev, vsi->rx_rings);
                vsi->rx_rings = NULL;
        }
        if (vsi->txq_map) {
                devm_kfree(dev, vsi->txq_map);
                vsi->txq_map = NULL;
        }
        if (vsi->rxq_map) {
                devm_kfree(dev, vsi->rxq_map);
                vsi->rxq_map = NULL;
        }
}

/**
 * ice_vsi_clear - clean up and deallocate the provided VSI
 * @vsi: pointer to VSI being cleared
 *
 * This deallocates the VSI's queue resources, removes it from the PF's
 * VSI array if necessary, and deallocates the VSI
 *
 * Returns 0 on success, negative on failure
 */
static int ice_vsi_clear(struct ice_vsi *vsi)
{
        struct ice_pf *pf = NULL;
        struct device *dev;

        if (!vsi)
                return 0;

        if (!vsi->back)
                return -EINVAL;

        pf = vsi->back;
        dev = ice_pf_to_dev(pf);

        if (!pf->vsi[vsi->idx] || pf->vsi[vsi->idx] != vsi) {
                dev_dbg(dev, "vsi does not exist at pf->vsi[%d]\n", vsi->idx);
                return -EINVAL;
        }

        mutex_lock(&pf->sw_mutex);
        /* updates the PF for this cleared VSI */

        pf->vsi[vsi->idx] = NULL;
        if (vsi->idx < pf->next_vsi && vsi->type != ICE_VSI_CTRL)
                pf->next_vsi = vsi->idx;
        if (vsi->idx < pf->next_vsi && vsi->type == ICE_VSI_CTRL &&
            vsi->vf_id != ICE_INVAL_VFID)
                pf->next_vsi = vsi->idx;

        ice_vsi_free_arrays(vsi);
        mutex_unlock(&pf->sw_mutex);
        devm_kfree(dev, vsi);

        return 0;
}

/**
 * ice_msix_clean_ctrl_vsi - MSIX mode interrupt handler for ctrl VSI
 * @irq: interrupt number
 * @data: pointer to a q_vector
 */
static irqreturn_t ice_msix_clean_ctrl_vsi(int __always_unused irq, void *data)
{
        struct ice_q_vector *q_vector = (struct ice_q_vector *)data;

        if (!q_vector->tx.ring)
                return IRQ_HANDLED;

#define FDIR_RX_DESC_CLEAN_BUDGET 64
        ice_clean_rx_irq(q_vector->rx.ring, FDIR_RX_DESC_CLEAN_BUDGET);
        ice_clean_ctrl_tx_irq(q_vector->tx.ring);

        return IRQ_HANDLED;
}

/**
 * ice_msix_clean_rings - MSIX mode Interrupt Handler
 * @irq: interrupt number
 * @data: pointer to a q_vector
 */
static irqreturn_t ice_msix_clean_rings(int __always_unused irq, void *data)
{
        struct ice_q_vector *q_vector = (struct ice_q_vector *)data;

        if (!q_vector->tx.ring && !q_vector->rx.ring)
                return IRQ_HANDLED;

        q_vector->total_events++;

        napi_schedule(&q_vector->napi);

        return IRQ_HANDLED;
}

/**
 * ice_vsi_alloc - Allocates the next available struct VSI in the PF
 * @pf: board private structure
 * @vsi_type: type of VSI
 * @vf_id: ID of the VF being configured
 *
 * returns a pointer to a VSI on success, NULL on failure.
 */
static struct ice_vsi *
ice_vsi_alloc(struct ice_pf *pf, enum ice_vsi_type vsi_type, u16 vf_id)
{
        struct device *dev = ice_pf_to_dev(pf);
        struct ice_vsi *vsi = NULL;

        /* Need to protect the allocation of the VSIs at the PF level */
        mutex_lock(&pf->sw_mutex);

        /* If we have already allocated our maximum number of VSIs,
         * pf->next_vsi will be ICE_NO_VSI. If not, pf->next_vsi index
         * is available to be populated
         */
        if (pf->next_vsi == ICE_NO_VSI) {
                dev_dbg(dev, "out of VSI slots!\n");
                goto unlock_pf;
        }

        vsi = devm_kzalloc(dev, sizeof(*vsi), GFP_KERNEL);
        if (!vsi)
                goto unlock_pf;

        vsi->type = vsi_type;
        vsi->back = pf;
        set_bit(ICE_VSI_DOWN, vsi->state);

        if (vsi_type == ICE_VSI_VF)
                ice_vsi_set_num_qs(vsi, vf_id);
        else
                ice_vsi_set_num_qs(vsi, ICE_INVAL_VFID);

        switch (vsi->type) {
        case ICE_VSI_PF:
                if (ice_vsi_alloc_arrays(vsi))
                        goto err_rings;

                /* Setup default MSIX irq handler for VSI */
                vsi->irq_handler = ice_msix_clean_rings;
                break;
        case ICE_VSI_CTRL:
                if (ice_vsi_alloc_arrays(vsi))
                        goto err_rings;

                /* Setup ctrl VSI MSIX irq handler */
                vsi->irq_handler = ice_msix_clean_ctrl_vsi;
                break;
        case ICE_VSI_VF:
                if (ice_vsi_alloc_arrays(vsi))
                        goto err_rings;
                break;
        case ICE_VSI_LB:
                if (ice_vsi_alloc_arrays(vsi))
                        goto err_rings;
                break;
        default:
                dev_warn(dev, "Unknown VSI type %d\n", vsi->type);
                goto unlock_pf;
        }

        if (vsi->type == ICE_VSI_CTRL && vf_id == ICE_INVAL_VFID) {
                /* Use the last VSI slot as the index for PF control VSI */
                vsi->idx = pf->num_alloc_vsi - 1;
                pf->ctrl_vsi_idx = vsi->idx;
                pf->vsi[vsi->idx] = vsi;
        } else {
                /* fill slot and make note of the index */
                vsi->idx = pf->next_vsi;
                pf->vsi[pf->next_vsi] = vsi;

                /* prepare pf->next_vsi for next use */
                pf->next_vsi = ice_get_free_slot(pf->vsi, pf->num_alloc_vsi,
                                                 pf->next_vsi);
        }

        if (vsi->type == ICE_VSI_CTRL && vf_id != ICE_INVAL_VFID)
                pf->vf[vf_id].ctrl_vsi_idx = vsi->idx;
        goto unlock_pf;

err_rings:
        devm_kfree(dev, vsi);
        vsi = NULL;
unlock_pf:
        mutex_unlock(&pf->sw_mutex);
        return vsi;
}

/**
 * ice_alloc_fd_res - Allocate FD resource for a VSI
 * @vsi: pointer to the ice_vsi
 *
 * This allocates the FD resources
 *
 * Returns 0 on success, -EPERM on no-op or -EIO on failure
 */
static int ice_alloc_fd_res(struct ice_vsi *vsi)
{
        struct ice_pf *pf = vsi->back;
        u32 g_val, b_val;

        /* Flow Director filters are only allocated/assigned to the PF VSI which
         * passes the traffic. The CTRL VSI is only used to add/delete filters
         * so we don't allocate resources to it
         */

        /* FD filters from guaranteed pool per VSI */
        g_val = pf->hw.func_caps.fd_fltr_guar;
        if (!g_val)
                return -EPERM;

        /* FD filters from best effort pool */
        b_val = pf->hw.func_caps.fd_fltr_best_effort;
        if (!b_val)
                return -EPERM;

        if (!(vsi->type == ICE_VSI_PF || vsi->type == ICE_VSI_VF))
                return -EPERM;

        if (!test_bit(ICE_FLAG_FD_ENA, pf->flags))
                return -EPERM;

        vsi->num_gfltr = g_val / pf->num_alloc_vsi;

        /* each VSI gets same "best_effort" quota */
        vsi->num_bfltr = b_val;

        if (vsi->type == ICE_VSI_VF) {
                vsi->num_gfltr = 0;

                /* each VSI gets same "best_effort" quota */
                vsi->num_bfltr = b_val;
        }

        return 0;
}

/**
 * ice_vsi_get_qs - Assign queues from PF to VSI
 * @vsi: the VSI to assign queues to
 *
 * Returns 0 on success and a negative value on error
 */
static int ice_vsi_get_qs(struct ice_vsi *vsi)
{
        struct ice_pf *pf = vsi->back;
        struct ice_qs_cfg tx_qs_cfg = {
                .qs_mutex = &pf->avail_q_mutex,
                .pf_map = pf->avail_txqs,
                .pf_map_size = pf->max_pf_txqs,
                .q_count = vsi->alloc_txq,
                .scatter_count = ICE_MAX_SCATTER_TXQS,
                .vsi_map = vsi->txq_map,
                .vsi_map_offset = 0,
                .mapping_mode = ICE_VSI_MAP_CONTIG
        };
        struct ice_qs_cfg rx_qs_cfg = {
                .qs_mutex = &pf->avail_q_mutex,
                .pf_map = pf->avail_rxqs,
                .pf_map_size = pf->max_pf_rxqs,
                .q_count = vsi->alloc_rxq,
                .scatter_count = ICE_MAX_SCATTER_RXQS,
                .vsi_map = vsi->rxq_map,
                .vsi_map_offset = 0,
                .mapping_mode = ICE_VSI_MAP_CONTIG
        };
        int ret;

        ret = __ice_vsi_get_qs(&tx_qs_cfg);
        if (ret)
                return ret;
        vsi->tx_mapping_mode = tx_qs_cfg.mapping_mode;

        ret = __ice_vsi_get_qs(&rx_qs_cfg);
        if (ret)
                return ret;
        vsi->rx_mapping_mode = rx_qs_cfg.mapping_mode;

        return 0;
}

/**
 * ice_vsi_put_qs - Release queues from VSI to PF
 * @vsi: the VSI that is going to release queues
 */
static void ice_vsi_put_qs(struct ice_vsi *vsi)
{
        struct ice_pf *pf = vsi->back;
        int i;

        mutex_lock(&pf->avail_q_mutex);

        for (i = 0; i < vsi->alloc_txq; i++) {
                clear_bit(vsi->txq_map[i], pf->avail_txqs);
                vsi->txq_map[i] = ICE_INVAL_Q_INDEX;
        }

        for (i = 0; i < vsi->alloc_rxq; i++) {
                clear_bit(vsi->rxq_map[i], pf->avail_rxqs);
                vsi->rxq_map[i] = ICE_INVAL_Q_INDEX;
        }

        mutex_unlock(&pf->avail_q_mutex);
}

/**
 * ice_is_safe_mode
 * @pf: pointer to the PF struct
 *
 * returns true if driver is in safe mode, false otherwise
 */
bool ice_is_safe_mode(struct ice_pf *pf)
{
        return !test_bit(ICE_FLAG_ADV_FEATURES, pf->flags);
}

/**
 * ice_is_aux_ena
 * @pf: pointer to the PF struct
 *
 * returns true if AUX devices/drivers are supported, false otherwise
 */
bool ice_is_aux_ena(struct ice_pf *pf)
{
        return test_bit(ICE_FLAG_AUX_ENA, pf->flags);
}

/**
 * ice_vsi_clean_rss_flow_fld - Delete RSS configuration
 * @vsi: the VSI being cleaned up
 *
 * This function deletes RSS input set for all flows that were configured
 * for this VSI
 */
static void ice_vsi_clean_rss_flow_fld(struct ice_vsi *vsi)
{
        struct ice_pf *pf = vsi->back;
        enum ice_status status;

        if (ice_is_safe_mode(pf))
                return;

        status = ice_rem_vsi_rss_cfg(&pf->hw, vsi->idx);
        if (status)
                dev_dbg(ice_pf_to_dev(pf), "ice_rem_vsi_rss_cfg failed for vsi = %d, error = %s\n",
                        vsi->vsi_num, ice_stat_str(status));
}

/**
 * ice_rss_clean - Delete RSS related VSI structures and configuration
 * @vsi: the VSI being removed
 */
static void ice_rss_clean(struct ice_vsi *vsi)
{
        struct ice_pf *pf = vsi->back;
        struct device *dev;

        dev = ice_pf_to_dev(pf);

        if (vsi->rss_hkey_user)
                devm_kfree(dev, vsi->rss_hkey_user);
        if (vsi->rss_lut_user)
                devm_kfree(dev, vsi->rss_lut_user);

        ice_vsi_clean_rss_flow_fld(vsi);
        /* remove RSS replay list */
        if (!ice_is_safe_mode(pf))
                ice_rem_vsi_rss_list(&pf->hw, vsi->idx);
}

/**
 * ice_vsi_set_rss_params - Setup RSS capabilities per VSI type
 * @vsi: the VSI being configured
 */
static void ice_vsi_set_rss_params(struct ice_vsi *vsi)
{
        struct ice_hw_common_caps *cap;
        struct ice_pf *pf = vsi->back;

        if (!test_bit(ICE_FLAG_RSS_ENA, pf->flags)) {
                vsi->rss_size = 1;
                return;
        }

        cap = &pf->hw.func_caps.common_cap;
        switch (vsi->type) {
        case ICE_VSI_PF:
                /* PF VSI will inherit RSS instance of PF */
                vsi->rss_table_size = (u16)cap->rss_table_size;
                vsi->rss_size = min_t(u16, num_online_cpus(),
                                      BIT(cap->rss_table_entry_width));
                vsi->rss_lut_type = ICE_AQC_GSET_RSS_LUT_TABLE_TYPE_PF;
                break;
        case ICE_VSI_VF:
                /* VF VSI will get a small RSS table.
                 * For VSI_LUT, LUT size should be set to 64 bytes.
                 */
                vsi->rss_table_size = ICE_VSIQF_HLUT_ARRAY_SIZE;
                vsi->rss_size = ICE_MAX_RSS_QS_PER_VF;
                vsi->rss_lut_type = ICE_AQC_GSET_RSS_LUT_TABLE_TYPE_VSI;
                break;
        case ICE_VSI_LB:
                break;
        default:
                dev_dbg(ice_pf_to_dev(pf), "Unsupported VSI type %s\n",
                        ice_vsi_type_str(vsi->type));
                break;
        }
}

/**
 * ice_set_dflt_vsi_ctx - Set default VSI context before adding a VSI
 * @ctxt: the VSI context being set
 *
 * This initializes a default VSI context for all sections except the Queues.
 */
static void ice_set_dflt_vsi_ctx(struct ice_vsi_ctx *ctxt)
{
        u32 table = 0;

        memset(&ctxt->info, 0, sizeof(ctxt->info));
        /* VSI's should be allocated from shared pool */
        ctxt->alloc_from_pool = true;
        /* Src pruning enabled by default */
        ctxt->info.sw_flags = ICE_AQ_VSI_SW_FLAG_SRC_PRUNE;
        /* Traffic from VSI can be sent to LAN */
        ctxt->info.sw_flags2 = ICE_AQ_VSI_SW_FLAG_LAN_ENA;
        /* By default bits 3 and 4 in vlan_flags are 0's which results in legacy
         * behavior (show VLAN, DEI, and UP) in descriptor. Also, allow all
         * packets untagged/tagged.
         */
        ctxt->info.vlan_flags = ((ICE_AQ_VSI_VLAN_MODE_ALL &
                                  ICE_AQ_VSI_VLAN_MODE_M) >>
                                 ICE_AQ_VSI_VLAN_MODE_S);
        /* Have 1:1 UP mapping for both ingress/egress tables */
        table |= ICE_UP_TABLE_TRANSLATE(0, 0);
        table |= ICE_UP_TABLE_TRANSLATE(1, 1);
        table |= ICE_UP_TABLE_TRANSLATE(2, 2);
        table |= ICE_UP_TABLE_TRANSLATE(3, 3);
        table |= ICE_UP_TABLE_TRANSLATE(4, 4);
        table |= ICE_UP_TABLE_TRANSLATE(5, 5);
        table |= ICE_UP_TABLE_TRANSLATE(6, 6);
        table |= ICE_UP_TABLE_TRANSLATE(7, 7);
        ctxt->info.ingress_table = cpu_to_le32(table);
        ctxt->info.egress_table = cpu_to_le32(table);
        /* Have 1:1 UP mapping for outer to inner UP table */
        ctxt->info.outer_up_table = cpu_to_le32(table);
        /* No Outer tag support outer_tag_flags remains to zero */
}

/**
 * ice_vsi_setup_q_map - Setup a VSI queue map
 * @vsi: the VSI being configured
 * @ctxt: VSI context structure
 */
static void ice_vsi_setup_q_map(struct ice_vsi *vsi, struct ice_vsi_ctx *ctxt)
{
        u16 offset = 0, qmap = 0, tx_count = 0, pow = 0;
        u16 num_txq_per_tc, num_rxq_per_tc;
        u16 qcount_tx = vsi->alloc_txq;
        u16 qcount_rx = vsi->alloc_rxq;
        bool ena_tc0 = false;
        u8 netdev_tc = 0;
        int i;

        /* at least TC0 should be enabled by default */
        if (vsi->tc_cfg.numtc) {
                if (!(vsi->tc_cfg.ena_tc & BIT(0)))
                        ena_tc0 = true;
        } else {
                ena_tc0 = true;
        }

        if (ena_tc0) {
                vsi->tc_cfg.numtc++;
                vsi->tc_cfg.ena_tc |= 1;
        }

        num_rxq_per_tc = min_t(u16, qcount_rx / vsi->tc_cfg.numtc, ICE_MAX_RXQS_PER_TC);
        if (!num_rxq_per_tc)
                num_rxq_per_tc = 1;
        num_txq_per_tc = qcount_tx / vsi->tc_cfg.numtc;
        if (!num_txq_per_tc)
                num_txq_per_tc = 1;

        /* find the (rounded up) power-of-2 of qcount */
        pow = (u16)order_base_2(num_rxq_per_tc);

        /* TC mapping is a function of the number of Rx queues assigned to the
         * VSI for each traffic class and the offset of these queues.
         * The first 10 bits are for queue offset for TC0, next 4 bits for no:of
         * queues allocated to TC0. No:of queues is a power-of-2.
         *
         * If TC is not enabled, the queue offset is set to 0, and allocate one
         * queue, this way, traffic for the given TC will be sent to the default
         * queue.
         *
         * Setup number and offset of Rx queues for all TCs for the VSI
         */
        ice_for_each_traffic_class(i) {
                if (!(vsi->tc_cfg.ena_tc & BIT(i))) {
                        /* TC is not enabled */
                        vsi->tc_cfg.tc_info[i].qoffset = 0;
                        vsi->tc_cfg.tc_info[i].qcount_rx = 1;
                        vsi->tc_cfg.tc_info[i].qcount_tx = 1;
                        vsi->tc_cfg.tc_info[i].netdev_tc = 0;
                        ctxt->info.tc_mapping[i] = 0;
                        continue;
                }

                /* TC is enabled */
                vsi->tc_cfg.tc_info[i].qoffset = offset;
                vsi->tc_cfg.tc_info[i].qcount_rx = num_rxq_per_tc;
                vsi->tc_cfg.tc_info[i].qcount_tx = num_txq_per_tc;
                vsi->tc_cfg.tc_info[i].netdev_tc = netdev_tc++;

                qmap = ((offset << ICE_AQ_VSI_TC_Q_OFFSET_S) &
                        ICE_AQ_VSI_TC_Q_OFFSET_M) |
                        ((pow << ICE_AQ_VSI_TC_Q_NUM_S) &
                         ICE_AQ_VSI_TC_Q_NUM_M);
                offset += num_rxq_per_tc;
                tx_count += num_txq_per_tc;
                ctxt->info.tc_mapping[i] = cpu_to_le16(qmap);
        }

        /* if offset is non-zero, means it is calculated correctly based on
         * enabled TCs for a given VSI otherwise qcount_rx will always
         * be correct and non-zero because it is based off - VSI's
         * allocated Rx queues which is at least 1 (hence qcount_tx will be
         * at least 1)
         */
        if (offset)
                vsi->num_rxq = offset;
        else
                vsi->num_rxq = num_rxq_per_tc;

        vsi->num_txq = tx_count;

        if (vsi->type == ICE_VSI_VF && vsi->num_txq != vsi->num_rxq) {
                dev_dbg(ice_pf_to_dev(vsi->back), "VF VSI should have same number of Tx and Rx queues. Hence making them equal\n");
                /* since there is a chance that num_rxq could have been changed
                 * in the above for loop, make num_txq equal to num_rxq.
                 */
                vsi->num_txq = vsi->num_rxq;
        }

        /* Rx queue mapping */
        ctxt->info.mapping_flags |= cpu_to_le16(ICE_AQ_VSI_Q_MAP_CONTIG);
        /* q_mapping buffer holds the info for the first queue allocated for
         * this VSI in the PF space and also the number of queues associated
         * with this VSI.
         */
        ctxt->info.q_mapping[0] = cpu_to_le16(vsi->rxq_map[0]);
        ctxt->info.q_mapping[1] = cpu_to_le16(vsi->num_rxq);
}

/**
 * ice_set_fd_vsi_ctx - Set FD VSI context before adding a VSI
 * @ctxt: the VSI context being set
 * @vsi: the VSI being configured
 */
static void ice_set_fd_vsi_ctx(struct ice_vsi_ctx *ctxt, struct ice_vsi *vsi)
{
        u8 dflt_q_group, dflt_q_prio;
        u16 dflt_q, report_q, val;

        if (vsi->type != ICE_VSI_PF && vsi->type != ICE_VSI_CTRL &&
            vsi->type != ICE_VSI_VF)
                return;

        val = ICE_AQ_VSI_PROP_FLOW_DIR_VALID;
        ctxt->info.valid_sections |= cpu_to_le16(val);
        dflt_q = 0;
        dflt_q_group = 0;
        report_q = 0;
        dflt_q_prio = 0;

        /* enable flow director filtering/programming */
        val = ICE_AQ_VSI_FD_ENABLE | ICE_AQ_VSI_FD_PROG_ENABLE;
        ctxt->info.fd_options = cpu_to_le16(val);
        /* max of allocated flow director filters */
        ctxt->info.max_fd_fltr_dedicated =
                        cpu_to_le16(vsi->num_gfltr);
        /* max of shared flow director filters any VSI may program */
        ctxt->info.max_fd_fltr_shared =
                        cpu_to_le16(vsi->num_bfltr);
        /* default queue index within the VSI of the default FD */
        val = ((dflt_q << ICE_AQ_VSI_FD_DEF_Q_S) &
               ICE_AQ_VSI_FD_DEF_Q_M);
        /* target queue or queue group to the FD filter */
        val |= ((dflt_q_group << ICE_AQ_VSI_FD_DEF_GRP_S) &
                ICE_AQ_VSI_FD_DEF_GRP_M);
        ctxt->info.fd_def_q = cpu_to_le16(val);
        /* queue index on which FD filter completion is reported */
        val = ((report_q << ICE_AQ_VSI_FD_REPORT_Q_S) &
               ICE_AQ_VSI_FD_REPORT_Q_M);
        /* priority of the default qindex action */
        val |= ((dflt_q_prio << ICE_AQ_VSI_FD_DEF_PRIORITY_S) &
                ICE_AQ_VSI_FD_DEF_PRIORITY_M);
        ctxt->info.fd_report_opt = cpu_to_le16(val);
}

/**
 * ice_set_rss_vsi_ctx - Set RSS VSI context before adding a VSI
 * @ctxt: the VSI context being set
 * @vsi: the VSI being configured
 */
static void ice_set_rss_vsi_ctx(struct ice_vsi_ctx *ctxt, struct ice_vsi *vsi)
{
        u8 lut_type, hash_type;
        struct device *dev;
        struct ice_pf *pf;

        pf = vsi->back;
        dev = ice_pf_to_dev(pf);

        switch (vsi->type) {
        case ICE_VSI_PF:
                /* PF VSI will inherit RSS instance of PF */
                lut_type = ICE_AQ_VSI_Q_OPT_RSS_LUT_PF;
                hash_type = ICE_AQ_VSI_Q_OPT_RSS_TPLZ;
                break;
        case ICE_VSI_VF:
                /* VF VSI will gets a small RSS table which is a VSI LUT type */
                lut_type = ICE_AQ_VSI_Q_OPT_RSS_LUT_VSI;
                hash_type = ICE_AQ_VSI_Q_OPT_RSS_TPLZ;
                break;
        default:
                dev_dbg(dev, "Unsupported VSI type %s\n",
                        ice_vsi_type_str(vsi->type));
                return;
        }

        ctxt->info.q_opt_rss = ((lut_type << ICE_AQ_VSI_Q_OPT_RSS_LUT_S) &
                                ICE_AQ_VSI_Q_OPT_RSS_LUT_M) |
                                ((hash_type << ICE_AQ_VSI_Q_OPT_RSS_HASH_S) &
                                 ICE_AQ_VSI_Q_OPT_RSS_HASH_M);
}

/**
 * ice_vsi_init - Create and initialize a VSI
 * @vsi: the VSI being configured
 * @init_vsi: is this call creating a VSI
 *
 * This initializes a VSI context depending on the VSI type to be added and
 * passes it down to the add_vsi aq command to create a new VSI.
 */
static int ice_vsi_init(struct ice_vsi *vsi, bool init_vsi)
{
        struct ice_pf *pf = vsi->back;
        struct ice_hw *hw = &pf->hw;
        struct ice_vsi_ctx *ctxt;
        struct device *dev;
        int ret = 0;

        dev = ice_pf_to_dev(pf);
        ctxt = kzalloc(sizeof(*ctxt), GFP_KERNEL);
        if (!ctxt)
                return -ENOMEM;

        switch (vsi->type) {
        case ICE_VSI_CTRL:
        case ICE_VSI_LB:
        case ICE_VSI_PF:
                ctxt->flags = ICE_AQ_VSI_TYPE_PF;
                break;
        case ICE_VSI_VF:
                ctxt->flags = ICE_AQ_VSI_TYPE_VF;
                /* VF number here is the absolute VF number (0-255) */
                ctxt->vf_num = vsi->vf_id + hw->func_caps.vf_base_id;
                break;
        default:
                ret = -ENODEV;
                goto out;
        }

        ice_set_dflt_vsi_ctx(ctxt);
        if (test_bit(ICE_FLAG_FD_ENA, pf->flags))
                ice_set_fd_vsi_ctx(ctxt, vsi);
        /* if the switch is in VEB mode, allow VSI loopback */
        if (vsi->vsw->bridge_mode == BRIDGE_MODE_VEB)
                ctxt->info.sw_flags |= ICE_AQ_VSI_SW_FLAG_ALLOW_LB;

        /* Set LUT type and HASH type if RSS is enabled */

        if (test_bit(ICE_FLAG_RSS_ENA, pf->flags) &&
            vsi->type != ICE_VSI_CTRL) {
                ice_set_rss_vsi_ctx(ctxt, vsi);
                /* if updating VSI context, make sure to set valid_section:
                 * to indicate which section of VSI context being updated
                 */
                if (!init_vsi)
                        ctxt->info.valid_sections |=
                                cpu_to_le16(ICE_AQ_VSI_PROP_Q_OPT_VALID);
        }

        ctxt->info.sw_id = vsi->port_info->sw_id;
        ice_vsi_setup_q_map(vsi, ctxt);
        if (!init_vsi) /* means VSI being updated */
                /* must to indicate which section of VSI context are
                 * being modified
                 */
                ctxt->info.valid_sections |=
                        cpu_to_le16(ICE_AQ_VSI_PROP_RXQ_MAP_VALID);

        /* enable/disable MAC and VLAN anti-spoof when spoofchk is on/off
         * respectively
         */
        if (vsi->type == ICE_VSI_VF) {
                ctxt->info.valid_sections |=
                        cpu_to_le16(ICE_AQ_VSI_PROP_SECURITY_VALID);
                if (pf->vf[vsi->vf_id].spoofchk) {
                        ctxt->info.sec_flags |=
                                ICE_AQ_VSI_SEC_FLAG_ENA_MAC_ANTI_SPOOF |
                                (ICE_AQ_VSI_SEC_TX_VLAN_PRUNE_ENA <<
                                 ICE_AQ_VSI_SEC_TX_PRUNE_ENA_S);
                } else {
                        ctxt->info.sec_flags &=
                                ~(ICE_AQ_VSI_SEC_FLAG_ENA_MAC_ANTI_SPOOF |
                                  (ICE_AQ_VSI_SEC_TX_VLAN_PRUNE_ENA <<
                                   ICE_AQ_VSI_SEC_TX_PRUNE_ENA_S));
                }
        }

        /* Allow control frames out of main VSI */
        if (vsi->type == ICE_VSI_PF) {
                ctxt->info.sec_flags |= ICE_AQ_VSI_SEC_FLAG_ALLOW_DEST_OVRD;
                ctxt->info.valid_sections |=
                        cpu_to_le16(ICE_AQ_VSI_PROP_SECURITY_VALID);
        }

        if (init_vsi) {
                ret = ice_add_vsi(hw, vsi->idx, ctxt, NULL);
                if (ret) {
                        dev_err(dev, "Add VSI failed, err %d\n", ret);
                        ret = -EIO;
                        goto out;
                }
        } else {
                ret = ice_update_vsi(hw, vsi->idx, ctxt, NULL);
                if (ret) {
                        dev_err(dev, "Update VSI failed, err %d\n", ret);
                        ret = -EIO;
                        goto out;
                }
        }

        /* keep context for update VSI operations */
        vsi->info = ctxt->info;

        /* record VSI number returned */
        vsi->vsi_num = ctxt->vsi_num;

out:
        kfree(ctxt);
        return ret;
}

/**
 * ice_free_res - free a block of resources
 * @res: pointer to the resource
 * @index: starting index previously returned by ice_get_res
 * @id: identifier to track owner
 *
 * Returns number of resources freed
 */
int ice_free_res(struct ice_res_tracker *res, u16 index, u16 id)
{
        int count = 0;
        int i;

        if (!res || index >= res->end)
                return -EINVAL;

        id |= ICE_RES_VALID_BIT;
        for (i = index; i < res->end && res->list[i] == id; i++) {
                res->list[i] = 0;
                count++;
        }

        return count;
}

/**
 * ice_search_res - Search the tracker for a block of resources
 * @res: pointer to the resource
 * @needed: size of the block needed
 * @id: identifier to track owner
 *
 * Returns the base item index of the block, or -ENOMEM for error
 */
static int ice_search_res(struct ice_res_tracker *res, u16 needed, u16 id)
{
        u16 start = 0, end = 0;

        if (needed > res->end)
                return -ENOMEM;

        id |= ICE_RES_VALID_BIT;

        do {
                /* skip already allocated entries */
                if (res->list[end++] & ICE_RES_VALID_BIT) {
                        start = end;
                        if ((start + needed) > res->end)
                                break;
                }

                if (end == (start + needed)) {
                        int i = start;

                        /* there was enough, so assign it to the requestor */
                        while (i != end)
                                res->list[i++] = id;

                        return start;
                }
        } while (end < res->end);

        return -ENOMEM;
}

/**
 * ice_get_free_res_count - Get free count from a resource tracker
 * @res: Resource tracker instance
 */
static u16 ice_get_free_res_count(struct ice_res_tracker *res)
{
        u16 i, count = 0;

        for (i = 0; i < res->end; i++)
                if (!(res->list[i] & ICE_RES_VALID_BIT))
                        count++;

        return count;
}

/**
 * ice_get_res - get a block of resources
 * @pf: board private structure
 * @res: pointer to the resource
 * @needed: size of the block needed
 * @id: identifier to track owner
 *
 * Returns the base item index of the block, or negative for error
 */
int
ice_get_res(struct ice_pf *pf, struct ice_res_tracker *res, u16 needed, u16 id)
{
        if (!res || !pf)
                return -EINVAL;

        if (!needed || needed > res->num_entries || id >= ICE_RES_VALID_BIT) {
                dev_err(ice_pf_to_dev(pf), "param err: needed=%d, num_entries = %d id=0x%04x\n",
                        needed, res->num_entries, id);
                return -EINVAL;
        }

        return ice_search_res(res, needed, id);
}

/**
 * ice_vsi_setup_vector_base - Set up the base vector for the given VSI
 * @vsi: ptr to the VSI
 *
 * This should only be called after ice_vsi_alloc() which allocates the
 * corresponding SW VSI structure and initializes num_queue_pairs for the
 * newly allocated VSI.
 *
 * Returns 0 on success or negative on failure
 */
static int ice_vsi_setup_vector_base(struct ice_vsi *vsi)
{
        struct ice_pf *pf = vsi->back;
        struct device *dev;
        u16 num_q_vectors;
        int base;

        dev = ice_pf_to_dev(pf);
        /* SRIOV doesn't grab irq_tracker entries for each VSI */
        if (vsi->type == ICE_VSI_VF)
                return 0;

        if (vsi->base_vector) {
                dev_dbg(dev, "VSI %d has non-zero base vector %d\n",
                        vsi->vsi_num, vsi->base_vector);
                return -EEXIST;
        }

        num_q_vectors = vsi->num_q_vectors;
        /* reserve slots from OS requested IRQs */
        if (vsi->type == ICE_VSI_CTRL && vsi->vf_id != ICE_INVAL_VFID) {
                int i;

                ice_for_each_vf(pf, i) {
                        struct ice_vf *vf = &pf->vf[i];

                        if (i != vsi->vf_id && vf->ctrl_vsi_idx != ICE_NO_VSI) {
                                base = pf->vsi[vf->ctrl_vsi_idx]->base_vector;
                                break;
                        }
                }
                if (i == pf->num_alloc_vfs)
                        base = ice_get_res(pf, pf->irq_tracker, num_q_vectors,
                                           ICE_RES_VF_CTRL_VEC_ID);
        } else {
                base = ice_get_res(pf, pf->irq_tracker, num_q_vectors,
                                   vsi->idx);
        }

        if (base < 0) {
                dev_err(dev, "%d MSI-X interrupts available. %s %d failed to get %d MSI-X vectors\n",
                        ice_get_free_res_count(pf->irq_tracker),
                        ice_vsi_type_str(vsi->type), vsi->idx, num_q_vectors);
                return -ENOENT;
        }
        vsi->base_vector = (u16)base;
        pf->num_avail_sw_msix -= num_q_vectors;

        return 0;
}

/**
 * ice_vsi_clear_rings - Deallocates the Tx and Rx rings for VSI
 * @vsi: the VSI having rings deallocated
 */
static void ice_vsi_clear_rings(struct ice_vsi *vsi)
{
        int i;

        /* Avoid stale references by clearing map from vector to ring */
        if (vsi->q_vectors) {
                ice_for_each_q_vector(vsi, i) {
                        struct ice_q_vector *q_vector = vsi->q_vectors[i];

                        if (q_vector) {
                                q_vector->tx.ring = NULL;
                                q_vector->rx.ring = NULL;
                        }
                }
        }

        if (vsi->tx_rings) {
                for (i = 0; i < vsi->alloc_txq; i++) {
                        if (vsi->tx_rings[i]) {
                                kfree_rcu(vsi->tx_rings[i], rcu);
                                WRITE_ONCE(vsi->tx_rings[i], NULL);
                        }
                }
        }
        if (vsi->rx_rings) {
                for (i = 0; i < vsi->alloc_rxq; i++) {
                        if (vsi->rx_rings[i]) {
                                kfree_rcu(vsi->rx_rings[i], rcu);
                                WRITE_ONCE(vsi->rx_rings[i], NULL);
                        }
                }
        }
}

/**
 * ice_vsi_alloc_rings - Allocates Tx and Rx rings for the VSI
 * @vsi: VSI which is having rings allocated
 */
static int ice_vsi_alloc_rings(struct ice_vsi *vsi)
{
        struct ice_pf *pf = vsi->back;
        struct device *dev;
        u16 i;

        dev = ice_pf_to_dev(pf);
        /* Allocate Tx rings */
        for (i = 0; i < vsi->alloc_txq; i++) {
                struct ice_ring *ring;

                /* allocate with kzalloc(), free with kfree_rcu() */
                ring = kzalloc(sizeof(*ring), GFP_KERNEL);

                if (!ring)
                        goto err_out;

                ring->q_index = i;
                ring->reg_idx = vsi->txq_map[i];
                ring->ring_active = false;
                ring->vsi = vsi;
                ring->tx_tstamps = &pf->ptp.port.tx;
                ring->dev = dev;
                ring->count = vsi->num_tx_desc;
                WRITE_ONCE(vsi->tx_rings[i], ring);
        }

        /* Allocate Rx rings */
        for (i = 0; i < vsi->alloc_rxq; i++) {
                struct ice_ring *ring;

                /* allocate with kzalloc(), free with kfree_rcu() */
                ring = kzalloc(sizeof(*ring), GFP_KERNEL);
                if (!ring)
                        goto err_out;

                ring->q_index = i;
                ring->reg_idx = vsi->rxq_map[i];
                ring->ring_active = false;
                ring->vsi = vsi;
                ring->netdev = vsi->netdev;
                ring->dev = dev;
                ring->count = vsi->num_rx_desc;
                WRITE_ONCE(vsi->rx_rings[i], ring);
        }

        return 0;

err_out:
        ice_vsi_clear_rings(vsi);
        return -ENOMEM;
}

/**
 * ice_vsi_manage_rss_lut - disable/enable RSS
 * @vsi: the VSI being changed
 * @ena: boolean value indicating if this is an enable or disable request
 *
 * In the event of disable request for RSS, this function will zero out RSS
 * LUT, while in the event of enable request for RSS, it will reconfigure RSS
 * LUT.
 */
void ice_vsi_manage_rss_lut(struct ice_vsi *vsi, bool ena)
{
        u8 *lut;

        lut = kzalloc(vsi->rss_table_size, GFP_KERNEL);
        if (!lut)
                return;

        if (ena) {
                if (vsi->rss_lut_user)
                        memcpy(lut, vsi->rss_lut_user, vsi->rss_table_size);
                else
                        ice_fill_rss_lut(lut, vsi->rss_table_size,
                                         vsi->rss_size);
        }

        ice_set_rss_lut(vsi, lut, vsi->rss_table_size);
        kfree(lut);
}

/**
 * ice_vsi_cfg_rss_lut_key - Configure RSS params for a VSI
 * @vsi: VSI to be configured
 */
static int ice_vsi_cfg_rss_lut_key(struct ice_vsi *vsi)
{
        struct ice_pf *pf = vsi->back;
        struct device *dev;
        u8 *lut, *key;
        int err;

        dev = ice_pf_to_dev(pf);
        vsi->rss_size = min_t(u16, vsi->rss_size, vsi->num_rxq);

        lut = kzalloc(vsi->rss_table_size, GFP_KERNEL);
        if (!lut)
                return -ENOMEM;

        if (vsi->rss_lut_user)
                memcpy(lut, vsi->rss_lut_user, vsi->rss_table_size);
        else
                ice_fill_rss_lut(lut, vsi->rss_table_size, vsi->rss_size);

        err = ice_set_rss_lut(vsi, lut, vsi->rss_table_size);
        if (err) {
                dev_err(dev, "set_rss_lut failed, error %d\n", err);
                goto ice_vsi_cfg_rss_exit;
        }

        key = kzalloc(ICE_GET_SET_RSS_KEY_EXTEND_KEY_SIZE, GFP_KERNEL);
        if (!key) {
                err = -ENOMEM;
                goto ice_vsi_cfg_rss_exit;
        }

        if (vsi->rss_hkey_user)
                memcpy(key, vsi->rss_hkey_user, ICE_GET_SET_RSS_KEY_EXTEND_KEY_SIZE);
        else
                netdev_rss_key_fill((void *)key, ICE_GET_SET_RSS_KEY_EXTEND_KEY_SIZE);

        err = ice_set_rss_key(vsi, key);
        if (err)
                dev_err(dev, "set_rss_key failed, error %d\n", err);

        kfree(key);
ice_vsi_cfg_rss_exit:
        kfree(lut);
        return err;
}

/**
 * ice_vsi_set_vf_rss_flow_fld - Sets VF VSI RSS input set for different flows
 * @vsi: VSI to be configured
 *
 * This function will only be called during the VF VSI setup. Upon successful
 * completion of package download, this function will configure default RSS
 * input sets for VF VSI.
 */
static void ice_vsi_set_vf_rss_flow_fld(struct ice_vsi *vsi)
{
        struct ice_pf *pf = vsi->back;
        enum ice_status status;
        struct device *dev;

        dev = ice_pf_to_dev(pf);
        if (ice_is_safe_mode(pf)) {
                dev_dbg(dev, "Advanced RSS disabled. Package download failed, vsi num = %d\n",
                        vsi->vsi_num);
                return;
        }

        status = ice_add_avf_rss_cfg(&pf->hw, vsi->idx, ICE_DEFAULT_RSS_HENA);
        if (status)
                dev_dbg(dev, "ice_add_avf_rss_cfg failed for vsi = %d, error = %s\n",
                        vsi->vsi_num, ice_stat_str(status));
}

/**
 * ice_vsi_set_rss_flow_fld - Sets RSS input set for different flows
 * @vsi: VSI to be configured
 *
 * This function will only be called after successful download package call
 * during initialization of PF. Since the downloaded package will erase the
 * RSS section, this function will configure RSS input sets for different
 * flow types. The last profile added has the highest priority, therefore 2
 * tuple profiles (i.e. IPv4 src/dst) are added before 4 tuple profiles
 * (i.e. IPv4 src/dst TCP src/dst port).
 */
static void ice_vsi_set_rss_flow_fld(struct ice_vsi *vsi)
{
        u16 vsi_handle = vsi->idx, vsi_num = vsi->vsi_num;
        struct ice_pf *pf = vsi->back;
        struct ice_hw *hw = &pf->hw;
        enum ice_status status;
        struct device *dev;

        dev = ice_pf_to_dev(pf);
        if (ice_is_safe_mode(pf)) {
                dev_dbg(dev, "Advanced RSS disabled. Package download failed, vsi num = %d\n",
                        vsi_num);
                return;
        }
        /* configure RSS for IPv4 with input set IP src/dst */
        status = ice_add_rss_cfg(hw, vsi_handle, ICE_FLOW_HASH_IPV4,
                                 ICE_FLOW_SEG_HDR_IPV4);
        if (status)
                dev_dbg(dev, "ice_add_rss_cfg failed for ipv4 flow, vsi = %d, error = %s\n",
                        vsi_num, ice_stat_str(status));

        /* configure RSS for IPv6 with input set IPv6 src/dst */
        status = ice_add_rss_cfg(hw, vsi_handle, ICE_FLOW_HASH_IPV6,
                                 ICE_FLOW_SEG_HDR_IPV6);
        if (status)
                dev_dbg(dev, "ice_add_rss_cfg failed for ipv6 flow, vsi = %d, error = %s\n",
                        vsi_num, ice_stat_str(status));

        /* configure RSS for tcp4 with input set IP src/dst, TCP src/dst */
        status = ice_add_rss_cfg(hw, vsi_handle, ICE_HASH_TCP_IPV4,
                                 ICE_FLOW_SEG_HDR_TCP | ICE_FLOW_SEG_HDR_IPV4);
        if (status)
                dev_dbg(dev, "ice_add_rss_cfg failed for tcp4 flow, vsi = %d, error = %s\n",
                        vsi_num, ice_stat_str(status));

        /* configure RSS for udp4 with input set IP src/dst, UDP src/dst */
        status = ice_add_rss_cfg(hw, vsi_handle, ICE_HASH_UDP_IPV4,
                                 ICE_FLOW_SEG_HDR_UDP | ICE_FLOW_SEG_HDR_IPV4);
        if (status)
                dev_dbg(dev, "ice_add_rss_cfg failed for udp4 flow, vsi = %d, error = %s\n",
                        vsi_num, ice_stat_str(status));

        /* configure RSS for sctp4 with input set IP src/dst */
        status = ice_add_rss_cfg(hw, vsi_handle, ICE_FLOW_HASH_IPV4,
                                 ICE_FLOW_SEG_HDR_SCTP | ICE_FLOW_SEG_HDR_IPV4);
        if (status)
                dev_dbg(dev, "ice_add_rss_cfg failed for sctp4 flow, vsi = %d, error = %s\n",
                        vsi_num, ice_stat_str(status));

        /* configure RSS for tcp6 with input set IPv6 src/dst, TCP src/dst */
        status = ice_add_rss_cfg(hw, vsi_handle, ICE_HASH_TCP_IPV6,
                                 ICE_FLOW_SEG_HDR_TCP | ICE_FLOW_SEG_HDR_IPV6);
        if (status)
                dev_dbg(dev, "ice_add_rss_cfg failed for tcp6 flow, vsi = %d, error = %s\n",
                        vsi_num, ice_stat_str(status));

        /* configure RSS for udp6 with input set IPv6 src/dst, UDP src/dst */
        status = ice_add_rss_cfg(hw, vsi_handle, ICE_HASH_UDP_IPV6,
                                 ICE_FLOW_SEG_HDR_UDP | ICE_FLOW_SEG_HDR_IPV6);
        if (status)
                dev_dbg(dev, "ice_add_rss_cfg failed for udp6 flow, vsi = %d, error = %s\n",
                        vsi_num, ice_stat_str(status));

        /* configure RSS for sctp6 with input set IPv6 src/dst */
        status = ice_add_rss_cfg(hw, vsi_handle, ICE_FLOW_HASH_IPV6,
                                 ICE_FLOW_SEG_HDR_SCTP | ICE_FLOW_SEG_HDR_IPV6);
        if (status)
                dev_dbg(dev, "ice_add_rss_cfg failed for sctp6 flow, vsi = %d, error = %s\n",
                        vsi_num, ice_stat_str(status));
}

/**
 * ice_pf_state_is_nominal - checks the PF for nominal state
 * @pf: pointer to PF to check
 *
 * Check the PF's state for a collection of bits that would indicate
 * the PF is in a state that would inhibit normal operation for
 * driver functionality.
 *
 * Returns true if PF is in a nominal state, false otherwise
 */
bool ice_pf_state_is_nominal(struct ice_pf *pf)
{
        DECLARE_BITMAP(check_bits, ICE_STATE_NBITS) = { 0 };

        if (!pf)
                return false;

        bitmap_set(check_bits, 0, ICE_STATE_NOMINAL_CHECK_BITS);
        if (bitmap_intersects(pf->state, check_bits, ICE_STATE_NBITS))
                return false;

        return true;
}

/**
 * ice_update_eth_stats - Update VSI-specific ethernet statistics counters
 * @vsi: the VSI to be updated
 */
void ice_update_eth_stats(struct ice_vsi *vsi)
{
        struct ice_eth_stats *prev_es, *cur_es;
        struct ice_hw *hw = &vsi->back->hw;
        u16 vsi_num = vsi->vsi_num;    /* HW absolute index of a VSI */

        prev_es = &vsi->eth_stats_prev;
        cur_es = &vsi->eth_stats;

        ice_stat_update40(hw, GLV_GORCL(vsi_num), vsi->stat_offsets_loaded,
                          &prev_es->rx_bytes, &cur_es->rx_bytes);

        ice_stat_update40(hw, GLV_UPRCL(vsi_num), vsi->stat_offsets_loaded,
                          &prev_es->rx_unicast, &cur_es->rx_unicast);

        ice_stat_update40(hw, GLV_MPRCL(vsi_num), vsi->stat_offsets_loaded,
                          &prev_es->rx_multicast, &cur_es->rx_multicast);

        ice_stat_update40(hw, GLV_BPRCL(vsi_num), vsi->stat_offsets_loaded,
                          &prev_es->rx_broadcast, &cur_es->rx_broadcast);

        ice_stat_update32(hw, GLV_RDPC(vsi_num), vsi->stat_offsets_loaded,
                          &prev_es->rx_discards, &cur_es->rx_discards);

        ice_stat_update40(hw, GLV_GOTCL(vsi_num), vsi->stat_offsets_loaded,
                          &prev_es->tx_bytes, &cur_es->tx_bytes);

        ice_stat_update40(hw, GLV_UPTCL(vsi_num), vsi->stat_offsets_loaded,
                          &prev_es->tx_unicast, &cur_es->tx_unicast);

        ice_stat_update40(hw, GLV_MPTCL(vsi_num), vsi->stat_offsets_loaded,
                          &prev_es->tx_multicast, &cur_es->tx_multicast);

        ice_stat_update40(hw, GLV_BPTCL(vsi_num), vsi->stat_offsets_loaded,
                          &prev_es->tx_broadcast, &cur_es->tx_broadcast);

        ice_stat_update32(hw, GLV_TEPC(vsi_num), vsi->stat_offsets_loaded,
                          &prev_es->tx_errors, &cur_es->tx_errors);

        vsi->stat_offsets_loaded = true;
}

/**
 * ice_vsi_add_vlan - Add VSI membership for given VLAN
 * @vsi: the VSI being configured
 * @vid: VLAN ID to be added
 * @action: filter action to be performed on match
 */
int
ice_vsi_add_vlan(struct ice_vsi *vsi, u16 vid, enum ice_sw_fwd_act_type action)
{
        struct ice_pf *pf = vsi->back;
        struct device *dev;
        int err = 0;

        dev = ice_pf_to_dev(pf);

        if (!ice_fltr_add_vlan(vsi, vid, action)) {
                vsi->num_vlan++;
        } else {
                err = -ENODEV;
                dev_err(dev, "Failure Adding VLAN %d on VSI %i\n", vid,
                        vsi->vsi_num);
        }

        return err;
}

/**
 * ice_vsi_kill_vlan - Remove VSI membership for a given VLAN
 * @vsi: the VSI being configured
 * @vid: VLAN ID to be removed
 *
 * Returns 0 on success and negative on failure
 */
int ice_vsi_kill_vlan(struct ice_vsi *vsi, u16 vid)
{
        struct ice_pf *pf = vsi->back;
        enum ice_status status;
        struct device *dev;
        int err = 0;

        dev = ice_pf_to_dev(pf);

        status = ice_fltr_remove_vlan(vsi, vid, ICE_FWD_TO_VSI);
        if (!status) {
                vsi->num_vlan--;
        } else if (status == ICE_ERR_DOES_NOT_EXIST) {
                dev_dbg(dev, "Failed to remove VLAN %d on VSI %i, it does not exist, status: %s\n",
                        vid, vsi->vsi_num, ice_stat_str(status));
        } else {
                dev_err(dev, "Error removing VLAN %d on vsi %i error: %s\n",
                        vid, vsi->vsi_num, ice_stat_str(status));
                err = -EIO;
        }

        return err;
}

/**
 * ice_vsi_cfg_frame_size - setup max frame size and Rx buffer length
 * @vsi: VSI
 */
void ice_vsi_cfg_frame_size(struct ice_vsi *vsi)
{
        if (!vsi->netdev || test_bit(ICE_FLAG_LEGACY_RX, vsi->back->flags)) {
                vsi->max_frame = ICE_AQ_SET_MAC_FRAME_SIZE_MAX;
                vsi->rx_buf_len = ICE_RXBUF_2048;
#if (PAGE_SIZE < 8192)
        } else if (!ICE_2K_TOO_SMALL_WITH_PADDING &&
                   (vsi->netdev->mtu <= ETH_DATA_LEN)) {
                vsi->max_frame = ICE_RXBUF_1536 - NET_IP_ALIGN;
                vsi->rx_buf_len = ICE_RXBUF_1536 - NET_IP_ALIGN;
#endif
        } else {
                vsi->max_frame = ICE_AQ_SET_MAC_FRAME_SIZE_MAX;
#if (PAGE_SIZE < 8192)
                vsi->rx_buf_len = ICE_RXBUF_3072;
#else
                vsi->rx_buf_len = ICE_RXBUF_2048;
#endif
        }
}

/**
 * ice_write_qrxflxp_cntxt - write/configure QRXFLXP_CNTXT register
 * @hw: HW pointer
 * @pf_q: index of the Rx queue in the PF's queue space
 * @rxdid: flexible descriptor RXDID
 * @prio: priority for the RXDID for this queue
 * @ena_ts: true to enable timestamp and false to disable timestamp
 */
void
ice_write_qrxflxp_cntxt(struct ice_hw *hw, u16 pf_q, u32 rxdid, u32 prio,
                        bool ena_ts)
{
        int regval = rd32(hw, QRXFLXP_CNTXT(pf_q));

        /* clear any previous values */
        regval &= ~(QRXFLXP_CNTXT_RXDID_IDX_M |
                    QRXFLXP_CNTXT_RXDID_PRIO_M |
                    QRXFLXP_CNTXT_TS_M);

        regval |= (rxdid << QRXFLXP_CNTXT_RXDID_IDX_S) &
                QRXFLXP_CNTXT_RXDID_IDX_M;

        regval |= (prio << QRXFLXP_CNTXT_RXDID_PRIO_S) &
                QRXFLXP_CNTXT_RXDID_PRIO_M;

        if (ena_ts)
                /* Enable TimeSync on this queue */
                regval |= QRXFLXP_CNTXT_TS_M;

        wr32(hw, QRXFLXP_CNTXT(pf_q), regval);
}

int ice_vsi_cfg_single_rxq(struct ice_vsi *vsi, u16 q_idx)
{
        if (q_idx >= vsi->num_rxq)
                return -EINVAL;

        return ice_vsi_cfg_rxq(vsi->rx_rings[q_idx]);
}

int ice_vsi_cfg_single_txq(struct ice_vsi *vsi, struct ice_ring **tx_rings, u16 q_idx)
{
        struct ice_aqc_add_tx_qgrp *qg_buf;
        int err;

        if (q_idx >= vsi->alloc_txq || !tx_rings || !tx_rings[q_idx])
                return -EINVAL;

        qg_buf = kzalloc(struct_size(qg_buf, txqs, 1), GFP_KERNEL);
        if (!qg_buf)
                return -ENOMEM;

        qg_buf->num_txqs = 1;

        err = ice_vsi_cfg_txq(vsi, tx_rings[q_idx], qg_buf);
        kfree(qg_buf);
        return err;
}

/**
 * ice_vsi_cfg_rxqs - Configure the VSI for Rx
 * @vsi: the VSI being configured
 *
 * Return 0 on success and a negative value on error
 * Configure the Rx VSI for operation.
 */
int ice_vsi_cfg_rxqs(struct ice_vsi *vsi)
{
        u16 i;

        if (vsi->type == ICE_VSI_VF)
                goto setup_rings;

        ice_vsi_cfg_frame_size(vsi);
setup_rings:
        /* set up individual rings */
        ice_for_each_rxq(vsi, i) {
                int err = ice_vsi_cfg_rxq(vsi->rx_rings[i]);

                if (err)
                        return err;
        }

        return 0;
}

/**
 * ice_vsi_cfg_txqs - Configure the VSI for Tx
 * @vsi: the VSI being configured
 * @rings: Tx ring array to be configured
 * @count: number of Tx ring array elements
 *
 * Return 0 on success and a negative value on error
 * Configure the Tx VSI for operation.
 */
static int
ice_vsi_cfg_txqs(struct ice_vsi *vsi, struct ice_ring **rings, u16 count)
{
        struct ice_aqc_add_tx_qgrp *qg_buf;
        u16 q_idx = 0;
        int err = 0;

        qg_buf = kzalloc(struct_size(qg_buf, txqs, 1), GFP_KERNEL);
        if (!qg_buf)
                return -ENOMEM;

        qg_buf->num_txqs = 1;

        for (q_idx = 0; q_idx < count; q_idx++) {
                err = ice_vsi_cfg_txq(vsi, rings[q_idx], qg_buf);
                if (err)
                        goto err_cfg_txqs;
        }

err_cfg_txqs:
        kfree(qg_buf);
        return err;
}

/**
 * ice_vsi_cfg_lan_txqs - Configure the VSI for Tx
 * @vsi: the VSI being configured
 *
 * Return 0 on success and a negative value on error
 * Configure the Tx VSI for operation.
 */
int ice_vsi_cfg_lan_txqs(struct ice_vsi *vsi)
{
        return ice_vsi_cfg_txqs(vsi, vsi->tx_rings, vsi->num_txq);
}

/**
 * ice_vsi_cfg_xdp_txqs - Configure Tx queues dedicated for XDP in given VSI
 * @vsi: the VSI being configured
 *
 * Return 0 on success and a negative value on error
 * Configure the Tx queues dedicated for XDP in given VSI for operation.
 */
int ice_vsi_cfg_xdp_txqs(struct ice_vsi *vsi)
{
        int ret;
        int i;

        ret = ice_vsi_cfg_txqs(vsi, vsi->xdp_rings, vsi->num_xdp_txq);
        if (ret)
                return ret;

        for (i = 0; i < vsi->num_xdp_txq; i++)
                vsi->xdp_rings[i]->xsk_pool = ice_xsk_pool(vsi->xdp_rings[i]);

        return ret;
}

/**
 * ice_intrl_usec_to_reg - convert interrupt rate limit to register value
 * @intrl: interrupt rate limit in usecs
 * @gran: interrupt rate limit granularity in usecs
 *
 * This function converts a decimal interrupt rate limit in usecs to the format
 * expected by firmware.
 */
static u32 ice_intrl_usec_to_reg(u8 intrl, u8 gran)
{
        u32 val = intrl / gran;

        if (val)
                return val | GLINT_RATE_INTRL_ENA_M;
        return 0;
}

/**
 * ice_write_intrl - write throttle rate limit to interrupt specific register
 * @q_vector: pointer to interrupt specific structure
 * @intrl: throttle rate limit in microseconds to write
 */
void ice_write_intrl(struct ice_q_vector *q_vector, u8 intrl)
{
        struct ice_hw *hw = &q_vector->vsi->back->hw;

        wr32(hw, GLINT_RATE(q_vector->reg_idx),
             ice_intrl_usec_to_reg(intrl, ICE_INTRL_GRAN_ABOVE_25));
}

/**
 * __ice_write_itr - write throttle rate to register
 * @q_vector: pointer to interrupt data structure
 * @rc: pointer to ring container
 * @itr: throttle rate in microseconds to write
 */
static void __ice_write_itr(struct ice_q_vector *q_vector,
                            struct ice_ring_container *rc, u16 itr)
{
        struct ice_hw *hw = &q_vector->vsi->back->hw;

        wr32(hw, GLINT_ITR(rc->itr_idx, q_vector->reg_idx),
             ITR_REG_ALIGN(itr) >> ICE_ITR_GRAN_S);
}

/**
 * ice_write_itr - write throttle rate to queue specific register
 * @rc: pointer to ring container
 * @itr: throttle rate in microseconds to write
 */
void ice_write_itr(struct ice_ring_container *rc, u16 itr)
{
        struct ice_q_vector *q_vector;

        if (!rc->ring)
                return;

        q_vector = rc->ring->q_vector;

        __ice_write_itr(q_vector, rc, itr);
}

/**
 * ice_vsi_cfg_msix - MSIX mode Interrupt Config in the HW
 * @vsi: the VSI being configured
 *
 * This configures MSIX mode interrupts for the PF VSI, and should not be used
 * for the VF VSI.
 */
void ice_vsi_cfg_msix(struct ice_vsi *vsi)
{
        struct ice_pf *pf = vsi->back;
        struct ice_hw *hw = &pf->hw;
        u16 txq = 0, rxq = 0;
        int i, q;

        for (i = 0; i < vsi->num_q_vectors; i++) {
                struct ice_q_vector *q_vector = vsi->q_vectors[i];
                u16 reg_idx = q_vector->reg_idx;

                ice_cfg_itr(hw, q_vector);

                /* Both Transmit Queue Interrupt Cause Control register
                 * and Receive Queue Interrupt Cause control register
                 * expects MSIX_INDX field to be the vector index
                 * within the function space and not the absolute
                 * vector index across PF or across device.
                 * For SR-IOV VF VSIs queue vector index always starts
                 * with 1 since first vector index(0) is used for OICR
                 * in VF space. Since VMDq and other PF VSIs are within
                 * the PF function space, use the vector index that is
                 * tracked for this PF.
                 */
                for (q = 0; q < q_vector->num_ring_tx; q++) {
                        ice_cfg_txq_interrupt(vsi, txq, reg_idx,
                                              q_vector->tx.itr_idx);
                        txq++;
                }

                for (q = 0; q < q_vector->num_ring_rx; q++) {
                        ice_cfg_rxq_interrupt(vsi, rxq, reg_idx,
                                              q_vector->rx.itr_idx);
                        rxq++;
                }
        }
}

/**
 * ice_vsi_manage_vlan_insertion - Manage VLAN insertion for the VSI for Tx
 * @vsi: the VSI being changed
 */
int ice_vsi_manage_vlan_insertion(struct ice_vsi *vsi)
{
        struct ice_hw *hw = &vsi->back->hw;
        struct ice_vsi_ctx *ctxt;
        enum ice_status status;
        int ret = 0;

        ctxt = kzalloc(sizeof(*ctxt), GFP_KERNEL);
        if (!ctxt)
                return -ENOMEM;

        /* Here we are configuring the VSI to let the driver add VLAN tags by
         * setting vlan_flags to ICE_AQ_VSI_VLAN_MODE_ALL. The actual VLAN tag
         * insertion happens in the Tx hot path, in ice_tx_map.
         */
        ctxt->info.vlan_flags = ICE_AQ_VSI_VLAN_MODE_ALL;

        /* Preserve existing VLAN strip setting */
        ctxt->info.vlan_flags |= (vsi->info.vlan_flags &
                                  ICE_AQ_VSI_VLAN_EMOD_M);

        ctxt->info.valid_sections = cpu_to_le16(ICE_AQ_VSI_PROP_VLAN_VALID);

        status = ice_update_vsi(hw, vsi->idx, ctxt, NULL);
        if (status) {
                dev_err(ice_pf_to_dev(vsi->back), "update VSI for VLAN insert failed, err %s aq_err %s\n",
                        ice_stat_str(status),
                        ice_aq_str(hw->adminq.sq_last_status));
                ret = -EIO;
                goto out;
        }

        vsi->info.vlan_flags = ctxt->info.vlan_flags;
out:
        kfree(ctxt);
        return ret;
}

/**
 * ice_vsi_manage_vlan_stripping - Manage VLAN stripping for the VSI for Rx
 * @vsi: the VSI being changed
 * @ena: boolean value indicating if this is a enable or disable request
 */
int ice_vsi_manage_vlan_stripping(struct ice_vsi *vsi, bool ena)
{
        struct ice_hw *hw = &vsi->back->hw;
        struct ice_vsi_ctx *ctxt;
        enum ice_status status;
        int ret = 0;

        /* do not allow modifying VLAN stripping when a port VLAN is configured
         * on this VSI
         */
        if (vsi->info.pvid)
                return 0;

        ctxt = kzalloc(sizeof(*ctxt), GFP_KERNEL);
        if (!ctxt)
                return -ENOMEM;

        /* Here we are configuring what the VSI should do with the VLAN tag in
         * the Rx packet. We can either leave the tag in the packet or put it in
         * the Rx descriptor.
         */