/*
 * Copyright (c) 2015-2016, Mellanox Technologies. All rights reserved.
 *
 * This software is available to you under a choice of one of two
 * licenses.  You may choose to be licensed under the terms of the GNU
 * General Public License (GPL) Version 2, available from the file
 * COPYING in the main directory of this source tree, or the
 * OpenIB.org BSD license below:
 *
 *     Redistribution and use in source and binary forms, with or
 *     without modification, are permitted provided that the following
 *     conditions are met:
 *
 *      - Redistributions of source code must retain the above
 *        copyright notice, this list of conditions and the following
 *        disclaimer.
 *
 *      - 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.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#include <net/tc_act/tc_gact.h>
#include <net/pkt_cls.h>
#include <linux/mlx5/fs.h>
#include <net/vxlan.h>
#include <net/geneve.h>
#include <linux/bpf.h>
#include <linux/if_bridge.h>
#include <net/page_pool.h>
#include <net/xdp_sock.h>
#include "eswitch.h"
#include "en.h"
#include "en/txrx.h"
#include "en_tc.h"
#include "en_rep.h"
#include "en_accel/ipsec.h"
#include "en_accel/ipsec_rxtx.h"
#include "en_accel/en_accel.h"
#include "en_accel/tls.h"
#include "accel/ipsec.h"
#include "accel/tls.h"
#include "lib/vxlan.h"
#include "lib/clock.h"
#include "en/port.h"
#include "en/xdp.h"
#include "lib/eq.h"
#include "en/monitor_stats.h"
#include "en/reporter.h"
#include "en/params.h"
#include "en/xsk/umem.h"
#include "en/xsk/setup.h"
#include "en/xsk/rx.h"
#include "en/xsk/tx.h"


bool mlx5e_check_fragmented_striding_rq_cap(struct mlx5_core_dev *mdev)
{
        bool striding_rq_umr = MLX5_CAP_GEN(mdev, striding_rq) &&
                MLX5_CAP_GEN(mdev, umr_ptr_rlky) &&
                MLX5_CAP_ETH(mdev, reg_umr_sq);
        u16 max_wqe_sz_cap = MLX5_CAP_GEN(mdev, max_wqe_sz_sq);
        bool inline_umr = MLX5E_UMR_WQE_INLINE_SZ <= max_wqe_sz_cap;

        if (!striding_rq_umr)
                return false;
        if (!inline_umr) {
                mlx5_core_warn(mdev, "Cannot support Striding RQ: UMR WQE size (%d) exceeds maximum supported (%d).\n",
                               (int)MLX5E_UMR_WQE_INLINE_SZ, max_wqe_sz_cap);
                return false;
        }
        return true;
}

void mlx5e_init_rq_type_params(struct mlx5_core_dev *mdev,
                               struct mlx5e_params *params)
{
        params->log_rq_mtu_frames = is_kdump_kernel() ?
                MLX5E_PARAMS_MINIMUM_LOG_RQ_SIZE :
                MLX5E_PARAMS_DEFAULT_LOG_RQ_SIZE;

        mlx5_core_info(mdev, "MLX5E: StrdRq(%d) RqSz(%ld) StrdSz(%ld) RxCqeCmprss(%d)\n",
                       params->rq_wq_type == MLX5_WQ_TYPE_LINKED_LIST_STRIDING_RQ,
                       params->rq_wq_type == MLX5_WQ_TYPE_LINKED_LIST_STRIDING_RQ ?
                       BIT(mlx5e_mpwqe_get_log_rq_size(params, NULL)) :
                       BIT(params->log_rq_mtu_frames),
                       BIT(mlx5e_mpwqe_get_log_stride_size(mdev, params, NULL)),
                       MLX5E_GET_PFLAG(params, MLX5E_PFLAG_RX_CQE_COMPRESS));
}

bool mlx5e_striding_rq_possible(struct mlx5_core_dev *mdev,
                                struct mlx5e_params *params)
{
        if (!mlx5e_check_fragmented_striding_rq_cap(mdev))
                return false;

        if (MLX5_IPSEC_DEV(mdev))
                return false;

        if (params->xdp_prog) {
                /* XSK params are not considered here. If striding RQ is in use,
                 * and an XSK is being opened, mlx5e_rx_mpwqe_is_linear_skb will
                 * be called with the known XSK params.
                 */
                if (!mlx5e_rx_mpwqe_is_linear_skb(mdev, params, NULL))
                        return false;
        }

        return true;
}

void mlx5e_set_rq_type(struct mlx5_core_dev *mdev, struct mlx5e_params *params)
{
        params->rq_wq_type = mlx5e_striding_rq_possible(mdev, params) &&
                MLX5E_GET_PFLAG(params, MLX5E_PFLAG_RX_STRIDING_RQ) ?
                MLX5_WQ_TYPE_LINKED_LIST_STRIDING_RQ :
                MLX5_WQ_TYPE_CYCLIC;
}

void mlx5e_update_carrier(struct mlx5e_priv *priv)
{
        struct mlx5_core_dev *mdev = priv->mdev;
        u8 port_state;

        port_state = mlx5_query_vport_state(mdev,
                                            MLX5_VPORT_STATE_OP_MOD_VNIC_VPORT,
                                            0);

        if (port_state == VPORT_STATE_UP) {
                netdev_info(priv->netdev, "Link up\n");
                netif_carrier_on(priv->netdev);
        } else {
                netdev_info(priv->netdev, "Link down\n");
                netif_carrier_off(priv->netdev);
        }
}

static void mlx5e_update_carrier_work(struct work_struct *work)
{
        struct mlx5e_priv *priv = container_of(work, struct mlx5e_priv,
                                               update_carrier_work);

        mutex_lock(&priv->state_lock);
        if (test_bit(MLX5E_STATE_OPENED, &priv->state))
                if (priv->profile->update_carrier)
                        priv->profile->update_carrier(priv);
        mutex_unlock(&priv->state_lock);
}

void mlx5e_update_stats(struct mlx5e_priv *priv)
{
        int i;

        for (i = mlx5e_num_stats_grps - 1; i >= 0; i--)
                if (mlx5e_stats_grps[i].update_stats)
                        mlx5e_stats_grps[i].update_stats(priv);
}

void mlx5e_update_ndo_stats(struct mlx5e_priv *priv)
{
        int i;

        for (i = mlx5e_num_stats_grps - 1; i >= 0; i--)
                if (mlx5e_stats_grps[i].update_stats_mask &
                    MLX5E_NDO_UPDATE_STATS)
                        mlx5e_stats_grps[i].update_stats(priv);
}

static void mlx5e_update_stats_work(struct work_struct *work)
{
        struct mlx5e_priv *priv = container_of(work, struct mlx5e_priv,
                                               update_stats_work);

        mutex_lock(&priv->state_lock);
        priv->profile->update_stats(priv);
        mutex_unlock(&priv->state_lock);
}

void mlx5e_queue_update_stats(struct mlx5e_priv *priv)
{
        if (!priv->profile->update_stats)
                return;

        if (unlikely(test_bit(MLX5E_STATE_DESTROYING, &priv->state)))
                return;

        queue_work(priv->wq, &priv->update_stats_work);
}

static int async_event(struct notifier_block *nb, unsigned long event, void *data)
{
        struct mlx5e_priv *priv = container_of(nb, struct mlx5e_priv, events_nb);
        struct mlx5_eqe   *eqe = data;

        if (event != MLX5_EVENT_TYPE_PORT_CHANGE)
                return NOTIFY_DONE;

        switch (eqe->sub_type) {
        case MLX5_PORT_CHANGE_SUBTYPE_DOWN:
        case MLX5_PORT_CHANGE_SUBTYPE_ACTIVE:
                queue_work(priv->wq, &priv->update_carrier_work);
                break;
        default:
                return NOTIFY_DONE;
        }

        return NOTIFY_OK;
}

static void mlx5e_enable_async_events(struct mlx5e_priv *priv)
{
        priv->events_nb.notifier_call = async_event;
        mlx5_notifier_register(priv->mdev, &priv->events_nb);
}

static void mlx5e_disable_async_events(struct mlx5e_priv *priv)
{
        mlx5_notifier_unregister(priv->mdev, &priv->events_nb);
}

static inline void mlx5e_build_umr_wqe(struct mlx5e_rq *rq,
                                       struct mlx5e_icosq *sq,
                                       struct mlx5e_umr_wqe *wqe)
{
        struct mlx5_wqe_ctrl_seg      *cseg = &wqe->ctrl;
        struct mlx5_wqe_umr_ctrl_seg *ucseg = &wqe->uctrl;
        u8 ds_cnt = DIV_ROUND_UP(MLX5E_UMR_WQE_INLINE_SZ, MLX5_SEND_WQE_DS);

        cseg->qpn_ds    = cpu_to_be32((sq->sqn << MLX5_WQE_CTRL_QPN_SHIFT) |
                                      ds_cnt);
        cseg->fm_ce_se  = MLX5_WQE_CTRL_CQ_UPDATE;
        cseg->imm       = rq->mkey_be;

        ucseg->flags = MLX5_UMR_TRANSLATION_OFFSET_EN | MLX5_UMR_INLINE;
        ucseg->xlt_octowords =
                cpu_to_be16(MLX5_MTT_OCTW(MLX5_MPWRQ_PAGES_PER_WQE));
        ucseg->mkey_mask     = cpu_to_be64(MLX5_MKEY_MASK_FREE);
}

static u32 mlx5e_rqwq_get_size(struct mlx5e_rq *rq)
{
        switch (rq->wq_type) {
        case MLX5_WQ_TYPE_LINKED_LIST_STRIDING_RQ:
                return mlx5_wq_ll_get_size(&rq->mpwqe.wq);
        default:
                return mlx5_wq_cyc_get_size(&rq->wqe.wq);
        }
}

static u32 mlx5e_rqwq_get_cur_sz(struct mlx5e_rq *rq)
{
        switch (rq->wq_type) {
        case MLX5_WQ_TYPE_LINKED_LIST_STRIDING_RQ:
                return rq->mpwqe.wq.cur_sz;
        default:
                return rq->wqe.wq.cur_sz;
        }
}

static int mlx5e_rq_alloc_mpwqe_info(struct mlx5e_rq *rq,
                                     struct mlx5e_channel *c)
{
        int wq_sz = mlx5_wq_ll_get_size(&rq->mpwqe.wq);

        rq->mpwqe.info = kvzalloc_node(array_size(wq_sz,
                                                  sizeof(*rq->mpwqe.info)),
                                       GFP_KERNEL, cpu_to_node(c->cpu));
        if (!rq->mpwqe.info)
                return -ENOMEM;

        mlx5e_build_umr_wqe(rq, &c->icosq, &rq->mpwqe.umr_wqe);

        return 0;
}

static int mlx5e_create_umr_mkey(struct mlx5_core_dev *mdev,
                                 u64 npages, u8 page_shift,
                                 struct mlx5_core_mkey *umr_mkey)
{
        int inlen = MLX5_ST_SZ_BYTES(create_mkey_in);
        void *mkc;
        u32 *in;
        int err;

        in = kvzalloc(inlen, GFP_KERNEL);
        if (!in)
                return -ENOMEM;

        mkc = MLX5_ADDR_OF(create_mkey_in, in, memory_key_mkey_entry);

        MLX5_SET(mkc, mkc, free, 1);
        MLX5_SET(mkc, mkc, umr_en, 1);
        MLX5_SET(mkc, mkc, lw, 1);
        MLX5_SET(mkc, mkc, lr, 1);
        MLX5_SET(mkc, mkc, access_mode_1_0, MLX5_MKC_ACCESS_MODE_MTT);

        MLX5_SET(mkc, mkc, qpn, 0xffffff);
        MLX5_SET(mkc, mkc, pd, mdev->mlx5e_res.pdn);
        MLX5_SET64(mkc, mkc, len, npages << page_shift);
        MLX5_SET(mkc, mkc, translations_octword_size,
                 MLX5_MTT_OCTW(npages));
        MLX5_SET(mkc, mkc, log_page_size, page_shift);

        err = mlx5_core_create_mkey(mdev, umr_mkey, in, inlen);

        kvfree(in);
        return err;
}

static int mlx5e_create_rq_umr_mkey(struct mlx5_core_dev *mdev, struct mlx5e_rq *rq)
{
        u64 num_mtts = MLX5E_REQUIRED_MTTS(mlx5_wq_ll_get_size(&rq->mpwqe.wq));

        return mlx5e_create_umr_mkey(mdev, num_mtts, PAGE_SHIFT, &rq->umr_mkey);
}

static inline u64 mlx5e_get_mpwqe_offset(struct mlx5e_rq *rq, u16 wqe_ix)
{
        return (wqe_ix << MLX5E_LOG_ALIGNED_MPWQE_PPW) << PAGE_SHIFT;
}

static void mlx5e_init_frags_partition(struct mlx5e_rq *rq)
{
        struct mlx5e_wqe_frag_info next_frag = {};
        struct mlx5e_wqe_frag_info *prev = NULL;
        int i;

        next_frag.di = &rq->wqe.di[0];

        for (i = 0; i < mlx5_wq_cyc_get_size(&rq->wqe.wq); i++) {
                struct mlx5e_rq_frag_info *frag_info = &rq->wqe.info.arr[0];
                struct mlx5e_wqe_frag_info *frag =
                        &rq->wqe.frags[i << rq->wqe.info.log_num_frags];
                int f;

                for (f = 0; f < rq->wqe.info.num_frags; f++, frag++) {
                        if (next_frag.offset + frag_info[f].frag_stride > PAGE_SIZE) {
                                next_frag.di++;
                                next_frag.offset = 0;
                                if (prev)
                                        prev->last_in_page = true;
                        }
                        *frag = next_frag;

                        /* prepare next */
                        next_frag.offset += frag_info[f].frag_stride;
                        prev = frag;
                }
        }

        if (prev)
                prev->last_in_page = true;
}

static int mlx5e_init_di_list(struct mlx5e_rq *rq,
                              int wq_sz, int cpu)
{
        int len = wq_sz << rq->wqe.info.log_num_frags;

        rq->wqe.di = kvzalloc_node(array_size(len, sizeof(*rq->wqe.di)),
                                   GFP_KERNEL, cpu_to_node(cpu));
        if (!rq->wqe.di)
                return -ENOMEM;

        mlx5e_init_frags_partition(rq);

        return 0;
}

static void mlx5e_free_di_list(struct mlx5e_rq *rq)
{
        kvfree(rq->wqe.di);
}

static int mlx5e_alloc_rq(struct mlx5e_channel *c,
                          struct mlx5e_params *params,
                          struct mlx5e_xsk_param *xsk,
                          struct xdp_umem *umem,
                          struct mlx5e_rq_param *rqp,
                          struct mlx5e_rq *rq)
{
        struct page_pool_params pp_params = { 0 };
        struct mlx5_core_dev *mdev = c->mdev;
        void *rqc = rqp->rqc;
        void *rqc_wq = MLX5_ADDR_OF(rqc, rqc, wq);
        u32 num_xsk_frames = 0;
        u32 rq_xdp_ix;
        u32 pool_size;
        int wq_sz;
        int err;
        int i;

        rqp->wq.db_numa_node = cpu_to_node(c->cpu);

        rq->wq_type = params->rq_wq_type;
        rq->pdev    = c->pdev;
        rq->netdev  = c->netdev;
        rq->tstamp  = c->tstamp;
        rq->clock   = &mdev->clock;
        rq->channel = c;
        rq->ix      = c->ix;
        rq->mdev    = mdev;
        rq->hw_mtu  = MLX5E_SW2HW_MTU(params, params->sw_mtu);
        rq->xdpsq   = &c->rq_xdpsq;
        rq->umem    = umem;

        if (rq->umem)
                rq->stats = &c->priv->channel_stats[c->ix].xskrq;
        else
                rq->stats = &c->priv->channel_stats[c->ix].rq;

        rq->xdp_prog = params->xdp_prog ? bpf_prog_inc(params->xdp_prog) : NULL;
        if (IS_ERR(rq->xdp_prog)) {
                err = PTR_ERR(rq->xdp_prog);
                rq->xdp_prog = NULL;
                goto err_rq_wq_destroy;
        }

        rq_xdp_ix = rq->ix;
        if (xsk)
                rq_xdp_ix += params->num_channels * MLX5E_RQ_GROUP_XSK;
        err = xdp_rxq_info_reg(&rq->xdp_rxq, rq->netdev, rq_xdp_ix);
        if (err < 0)
                goto err_rq_wq_destroy;

        rq->buff.map_dir = rq->xdp_prog ? DMA_BIDIRECTIONAL : DMA_FROM_DEVICE;
        rq->buff.headroom = mlx5e_get_rq_headroom(mdev, params, xsk);
        rq->buff.umem_headroom = xsk ? xsk->headroom : 0;
        pool_size = 1 << params->log_rq_mtu_frames;

        switch (rq->wq_type) {
        case MLX5_WQ_TYPE_LINKED_LIST_STRIDING_RQ:
                err = mlx5_wq_ll_create(mdev, &rqp->wq, rqc_wq, &rq->mpwqe.wq,
                                        &rq->wq_ctrl);
                if (err)
                        return err;

                rq->mpwqe.wq.db = &rq->mpwqe.wq.db[MLX5_RCV_DBR];

                wq_sz = mlx5_wq_ll_get_size(&rq->mpwqe.wq);

                if (xsk)
                        num_xsk_frames = wq_sz <<
                                mlx5e_mpwqe_get_log_num_strides(mdev, params, xsk);

                pool_size = MLX5_MPWRQ_PAGES_PER_WQE <<
                        mlx5e_mpwqe_get_log_rq_size(params, xsk);

                rq->post_wqes = mlx5e_post_rx_mpwqes;
                rq->dealloc_wqe = mlx5e_dealloc_rx_mpwqe;

                rq->handle_rx_cqe = c->priv->profile->rx_handlers.handle_rx_cqe_mpwqe;
#ifdef CONFIG_MLX5_EN_IPSEC
                if (MLX5_IPSEC_DEV(mdev)) {
                        err = -EINVAL;
                        netdev_err(c->netdev, "MPWQE RQ with IPSec offload not supported\n");
                        goto err_rq_wq_destroy;
                }
#endif
                if (!rq->handle_rx_cqe) {
                        err = -EINVAL;
                        netdev_err(c->netdev, "RX handler of MPWQE RQ is not set, err %d\n", err);
                        goto err_rq_wq_destroy;
                }

                rq->mpwqe.skb_from_cqe_mpwrq = xsk ?
                        mlx5e_xsk_skb_from_cqe_mpwrq_linear :
                        mlx5e_rx_mpwqe_is_linear_skb(mdev, params, NULL) ?
                                mlx5e_skb_from_cqe_mpwrq_linear :
                                mlx5e_skb_from_cqe_mpwrq_nonlinear;

                rq->mpwqe.log_stride_sz = mlx5e_mpwqe_get_log_stride_size(mdev, params, xsk);
                rq->mpwqe.num_strides =
                        BIT(mlx5e_mpwqe_get_log_num_strides(mdev, params, xsk));

                err = mlx5e_create_rq_umr_mkey(mdev, rq);
                if (err)
                        goto err_rq_wq_destroy;
                rq->mkey_be = cpu_to_be32(rq->umr_mkey.key);

                err = mlx5e_rq_alloc_mpwqe_info(rq, c);
                if (err)
                        goto err_free;
                break;
        default: /* MLX5_WQ_TYPE_CYCLIC */
                err = mlx5_wq_cyc_create(mdev, &rqp->wq, rqc_wq, &rq->wqe.wq,
                                         &rq->wq_ctrl);
                if (err)
                        return err;

                rq->wqe.wq.db = &rq->wqe.wq.db[MLX5_RCV_DBR];

                wq_sz = mlx5_wq_cyc_get_size(&rq->wqe.wq);

                if (xsk)
                        num_xsk_frames = wq_sz << rq->wqe.info.log_num_frags;

                rq->wqe.info = rqp->frags_info;
                rq->wqe.frags =
                        kvzalloc_node(array_size(sizeof(*rq->wqe.frags),
                                        (wq_sz << rq->wqe.info.log_num_frags)),
                                      GFP_KERNEL, cpu_to_node(c->cpu));
                if (!rq->wqe.frags) {
                        err = -ENOMEM;
                        goto err_free;
                }

                err = mlx5e_init_di_list(rq, wq_sz, c->cpu);
                if (err)
                        goto err_free;

                rq->post_wqes = mlx5e_post_rx_wqes;
                rq->dealloc_wqe = mlx5e_dealloc_rx_wqe;

#ifdef CONFIG_MLX5_EN_IPSEC
                if (c->priv->ipsec)
                        rq->handle_rx_cqe = mlx5e_ipsec_handle_rx_cqe;
                else
#endif
                        rq->handle_rx_cqe = c->priv->profile->rx_handlers.handle_rx_cqe;
                if (!rq->handle_rx_cqe) {
                        err = -EINVAL;
                        netdev_err(c->netdev, "RX handler of RQ is not set, err %d\n", err);
                        goto err_free;
                }

                rq->wqe.skb_from_cqe = xsk ?
                        mlx5e_xsk_skb_from_cqe_linear :
                        mlx5e_rx_is_linear_skb(params, NULL) ?
                                mlx5e_skb_from_cqe_linear :
                                mlx5e_skb_from_cqe_nonlinear;
                rq->mkey_be = c->mkey_be;
        }

        if (xsk) {
                err = mlx5e_xsk_resize_reuseq(umem, num_xsk_frames);
                if (unlikely(err)) {
                        mlx5_core_err(mdev, "Unable to allocate the Reuse Ring for %u frames\n",
                                      num_xsk_frames);
                        goto err_free;
                }

                rq->zca.free = mlx5e_xsk_zca_free;
                err = xdp_rxq_info_reg_mem_model(&rq->xdp_rxq,
                                                 MEM_TYPE_ZERO_COPY,
                                                 &rq->zca);
        } else {
                /* Create a page_pool and register it with rxq */
                pp_params.order     = 0;
                pp_params.flags     = 0; /* No-internal DMA mapping in page_pool */
                pp_params.pool_size = pool_size;
                pp_params.nid       = cpu_to_node(c->cpu);
                pp_params.dev       = c->pdev;
                pp_params.dma_dir   = rq->buff.map_dir;

                /* page_pool can be used even when there is no rq->xdp_prog,
                 * given page_pool does not handle DMA mapping there is no
                 * required state to clear. And page_pool gracefully handle
                 * elevated refcnt.
                 */
                rq->page_pool = page_pool_create(&pp_params);
                if (IS_ERR(rq->page_pool)) {
                        err = PTR_ERR(rq->page_pool);
                        rq->page_pool = NULL;
                        goto err_free;
                }
                err = xdp_rxq_info_reg_mem_model(&rq->xdp_rxq,
                                                 MEM_TYPE_PAGE_POOL, rq->page_pool);
        }
        if (err)
                goto err_free;

        for (i = 0; i < wq_sz; i++) {
                if (rq->wq_type == MLX5_WQ_TYPE_LINKED_LIST_STRIDING_RQ) {
                        struct mlx5e_rx_wqe_ll *wqe =
                                mlx5_wq_ll_get_wqe(&rq->mpwqe.wq, i);
                        u32 byte_count =
                                rq->mpwqe.num_strides << rq->mpwqe.log_stride_sz;
                        u64 dma_offset = mlx5e_get_mpwqe_offset(rq, i);

                        wqe->data[0].addr = cpu_to_be64(dma_offset + rq->buff.headroom);
                        wqe->data[0].byte_count = cpu_to_be32(byte_count);
                        wqe->data[0].lkey = rq->mkey_be;
                } else {
                        struct mlx5e_rx_wqe_cyc *wqe =
                                mlx5_wq_cyc_get_wqe(&rq->wqe.wq, i);
                        int f;

                        for (f = 0; f < rq->wqe.info.num_frags; f++) {
                                u32 frag_size = rq->wqe.info.arr[f].frag_size |
                                        MLX5_HW_START_PADDING;

                                wqe->data[f].byte_count = cpu_to_be32(frag_size);
                                wqe->data[f].lkey = rq->mkey_be;
                        }
                        /* check if num_frags is not a pow of two */
                        if (rq->wqe.info.num_frags < (1 << rq->wqe.info.log_num_frags)) {
                                wqe->data[f].byte_count = 0;
                                wqe->data[f].lkey = cpu_to_be32(MLX5_INVALID_LKEY);
                                wqe->data[f].addr = 0;
                        }
                }
        }

        INIT_WORK(&rq->dim.work, mlx5e_rx_dim_work);

        switch (params->rx_cq_moderation.cq_period_mode) {
        case MLX5_CQ_PERIOD_MODE_START_FROM_CQE:
                rq->dim.mode = DIM_CQ_PERIOD_MODE_START_FROM_CQE;
                break;
        case MLX5_CQ_PERIOD_MODE_START_FROM_EQE:
        default:
                rq->dim.mode = DIM_CQ_PERIOD_MODE_START_FROM_EQE;
        }

        rq->page_cache.head = 0;
        rq->page_cache.tail = 0;

        return 0;

err_free:
        switch (rq->wq_type) {
        case MLX5_WQ_TYPE_LINKED_LIST_STRIDING_RQ:
                kvfree(rq->mpwqe.info);
                mlx5_core_destroy_mkey(mdev, &rq->umr_mkey);
                break;
        default: /* MLX5_WQ_TYPE_CYCLIC */
                kvfree(rq->wqe.frags);
                mlx5e_free_di_list(rq);
        }

err_rq_wq_destroy:
        if (rq->xdp_prog)
                bpf_prog_put(rq->xdp_prog);
        xdp_rxq_info_unreg(&rq->xdp_rxq);
        page_pool_destroy(rq->page_pool);
        mlx5_wq_destroy(&rq->wq_ctrl);

        return err;
}

static void mlx5e_free_rq(struct mlx5e_rq *rq)
{
        int i;

        if (rq->xdp_prog)
                bpf_prog_put(rq->xdp_prog);

        switch (rq->wq_type) {
        case MLX5_WQ_TYPE_LINKED_LIST_STRIDING_RQ:
                kvfree(rq->mpwqe.info);
                mlx5_core_destroy_mkey(rq->mdev, &rq->umr_mkey);
                break;
        default: /* MLX5_WQ_TYPE_CYCLIC */
                kvfree(rq->wqe.frags);
                mlx5e_free_di_list(rq);
        }

        for (i = rq->page_cache.head; i != rq->page_cache.tail;
             i = (i + 1) & (MLX5E_CACHE_SIZE - 1)) {
                struct mlx5e_dma_info *dma_info = &rq->page_cache.page_cache[i];

                /* With AF_XDP, page_cache is not used, so this loop is not
                 * entered, and it's safe to call mlx5e_page_release_dynamic
                 * directly.
                 */
                mlx5e_page_release_dynamic(rq, dma_info, false);
        }

        xdp_rxq_info_unreg(&rq->xdp_rxq);
        page_pool_destroy(rq->page_pool);
        mlx5_wq_destroy(&rq->wq_ctrl);
}

static int mlx5e_create_rq(struct mlx5e_rq *rq,
                           struct mlx5e_rq_param *param)
{
        struct mlx5_core_dev *mdev = rq->mdev;

        void *in;
        void *rqc;
        void *wq;
        int inlen;
        int err;

        inlen = MLX5_ST_SZ_BYTES(create_rq_in) +
                sizeof(u64) * rq->wq_ctrl.buf.npages;
        in = kvzalloc(inlen, GFP_KERNEL);
        if (!in)
                return -ENOMEM;

        rqc = MLX5_ADDR_OF(create_rq_in, in, ctx);
        wq  = MLX5_ADDR_OF(rqc, rqc, wq);

        memcpy(rqc, param->rqc, sizeof(param->rqc));

        MLX5_SET(rqc,  rqc, cqn,                rq->cq.mcq.cqn);
        MLX5_SET(rqc,  rqc, state,              MLX5_RQC_STATE_RST);
        MLX5_SET(wq,   wq,  log_wq_pg_sz,       rq->wq_ctrl.buf.page_shift -
                                                MLX5_ADAPTER_PAGE_SHIFT);
        MLX5_SET64(wq, wq,  dbr_addr,           rq->wq_ctrl.db.dma);

        mlx5_fill_page_frag_array(&rq->wq_ctrl.buf,
                                  (__be64 *)MLX5_ADDR_OF(wq, wq, pas));

        err = mlx5_core_create_rq(mdev, in, inlen, &rq->rqn);

        kvfree(in);

        return err;
}

static int mlx5e_modify_rq_state(struct mlx5e_rq *rq, int curr_state,
                                 int next_state)
{
        struct mlx5_core_dev *mdev = rq->mdev;

        void *in;
        void *rqc;
        int inlen;
        int err;

        inlen = MLX5_ST_SZ_BYTES(modify_rq_in);
        in = kvzalloc(inlen, GFP_KERNEL);
        if (!in)
                return -ENOMEM;

        rqc = MLX5_ADDR_OF(modify_rq_in, in, ctx);

        MLX5_SET(modify_rq_in, in, rq_state, curr_state);
        MLX5_SET(rqc, rqc, state, next_state);

        err = mlx5_core_modify_rq(mdev, rq->rqn, in, inlen);

        kvfree(in);

        return err;
}

static int mlx5e_modify_rq_scatter_fcs(struct mlx5e_rq *rq, bool enable)
{
        struct mlx5e_channel *c = rq->channel;
        struct mlx5e_priv *priv = c->priv;
        struct mlx5_core_dev *mdev = priv->mdev;

        void *in;
        void *rqc;
        int inlen;
        int err;

        inlen = MLX5_ST_SZ_BYTES(modify_rq_in);
        in = kvzalloc(inlen, GFP_KERNEL);
        if (!in)
                return -ENOMEM;

        rqc = MLX5_ADDR_OF(modify_rq_in, in, ctx);

        MLX5_SET(modify_rq_in, in, rq_state, MLX5_RQC_STATE_RDY);
        MLX5_SET64(modify_rq_in, in, modify_bitmask,
                   MLX5_MODIFY_RQ_IN_MODIFY_BITMASK_SCATTER_FCS);
        MLX5_SET(rqc, rqc, scatter_fcs, enable);
        MLX5_SET(rqc, rqc, state, MLX5_RQC_STATE_RDY);

        err = mlx5_core_modify_rq(mdev, rq->rqn, in, inlen);

        kvfree(in);

        return err;
}

static int mlx5e_modify_rq_vsd(struct mlx5e_rq *rq, bool vsd)
{
        struct mlx5e_channel *c = rq->channel;
        struct mlx5_core_dev *mdev = c->mdev;
        void *in;
        void *rqc;
        int inlen;
        int err;

        inlen = MLX5_ST_SZ_BYTES(modify_rq_in);
        in = kvzalloc(inlen, GFP_KERNEL);
        if (!in)
                return -ENOMEM;

        rqc = MLX5_ADDR_OF(modify_rq_in, in, ctx);

        MLX5_SET(modify_rq_in, in, rq_state, MLX5_RQC_STATE_RDY);
        MLX5_SET64(modify_rq_in, in, modify_bitmask,
                   MLX5_MODIFY_RQ_IN_MODIFY_BITMASK_VSD);
        MLX5_SET(rqc, rqc, vsd, vsd);
        MLX5_SET(rqc, rqc, state, MLX5_RQC_STATE_RDY);

        err = mlx5_core_modify_rq(mdev, rq->rqn, in, inlen);

        kvfree(in);

        return err;
}

static void mlx5e_destroy_rq(struct mlx5e_rq *rq)
{
        mlx5_core_destroy_rq(rq->mdev, rq->rqn);
}

int mlx5e_wait_for_min_rx_wqes(struct mlx5e_rq *rq, int wait_time)
{
        unsigned long exp_time = jiffies + msecs_to_jiffies(wait_time);
        struct mlx5e_channel *c = rq->channel;

        u16 min_wqes = mlx5_min_rx_wqes(rq->wq_type, mlx5e_rqwq_get_size(rq));

        do {
                if (mlx5e_rqwq_get_cur_sz(rq) >= min_wqes)
                        return 0;

                msleep(20);
        } while (time_before(jiffies, exp_time));

        netdev_warn(c->netdev, "Failed to get min RX wqes on Channel[%d] RQN[0x%x] wq cur_sz(%d) min_rx_wqes(%d)\n",
                    c->ix, rq->rqn, mlx5e_rqwq_get_cur_sz(rq), min_wqes);

        return -ETIMEDOUT;
}

static void mlx5e_free_rx_descs(struct mlx5e_rq *rq)
{
        __be16 wqe_ix_be;
        u16 wqe_ix;

        if (rq->wq_type == MLX5_WQ_TYPE_LINKED_LIST_STRIDING_RQ) {
                struct mlx5_wq_ll *wq = &rq->mpwqe.wq;
                u16 head = wq->head;
                int i;

                /* Outstanding UMR WQEs (in progress) start at wq->head */
                for (i = 0; i < rq->mpwqe.umr_in_progress; i++) {
                        rq->dealloc_wqe(rq, head);
                        head = mlx5_wq_ll_get_wqe_next_ix(wq, head);
                }

                while (!mlx5_wq_ll_is_empty(wq)) {
                        struct mlx5e_rx_wqe_ll *wqe;

                        wqe_ix_be = *wq->tail_next;
                        wqe_ix    = be16_to_cpu(wqe_ix_be);
                        wqe       = mlx5_wq_ll_get_wqe(wq, wqe_ix);
                        rq->dealloc_wqe(rq, wqe_ix);
                        mlx5_wq_ll_pop(wq, wqe_ix_be,
                                       &wqe->next.next_wqe_index);
                }
        } else {
                struct mlx5_wq_cyc *wq = &rq->wqe.wq;

                while (!mlx5_wq_cyc_is_empty(wq)) {
                        wqe_ix = mlx5_wq_cyc_get_tail(wq);
                        rq->dealloc_wqe(rq, wqe_ix);
                        mlx5_wq_cyc_pop(wq);
                }
        }

}

int mlx5e_open_rq(struct mlx5e_channel *c, struct mlx5e_params *params,
                  struct mlx5e_rq_param *param, struct mlx5e_xsk_param *xsk,
                  struct xdp_umem *umem, struct mlx5e_rq *rq)
{
        int err;

        err = mlx5e_alloc_rq(c, params, xsk, umem, param, rq);
        if (err)
                return err;

        err = mlx5e_create_rq(rq, param);
        if (err)
                goto err_free_rq;

        err = mlx5e_modify_rq_state(rq, MLX5_RQC_STATE_RST, MLX5_RQC_STATE_RDY);
        if (err)
                goto err_destroy_rq;

        if (MLX5_CAP_ETH(c->mdev, cqe_checksum_full))
                __set_bit(MLX5E_RQ_STATE_CSUM_FULL, &c->rq.state);

        if (params->rx_dim_enabled)
                __set_bit(MLX5E_RQ_STATE_AM, &c->rq.state);

        /* We disable csum_complete when XDP is enabled since
         * XDP programs might manipulate packets which will render
         * skb->checksum incorrect.
         */
        if (MLX5E_GET_PFLAG(params, MLX5E_PFLAG_RX_NO_CSUM_COMPLETE) || c->xdp)
                __set_bit(MLX5E_RQ_STATE_NO_CSUM_COMPLETE, &c->rq.state);

        return 0;

err_destroy_rq:
        mlx5e_destroy_rq(rq);
err_free_rq:
        mlx5e_free_rq(rq);

        return err;
}

static void mlx5e_activate_rq(struct mlx5e_rq *rq)
{
        set_bit(MLX5E_RQ_STATE_ENABLED, &rq->state);
        mlx5e_trigger_irq(&rq->channel->icosq);
}

void mlx5e_deactivate_rq(struct mlx5e_rq *rq)
{
        clear_bit(MLX5E_RQ_STATE_ENABLED, &rq->state);
        napi_synchronize(&rq->channel->napi); /* prevent mlx5e_post_rx_wqes */
}

void mlx5e_close_rq(struct mlx5e_rq *rq)
{
        cancel_work_sync(&rq->dim.work);
        mlx5e_destroy_rq(rq);
        mlx5e_free_rx_descs(rq);
        mlx5e_free_rq(rq);
}

static void mlx5e_free_xdpsq_db(struct mlx5e_xdpsq *sq)
{
        kvfree(sq->db.xdpi_fifo.xi);
        kvfree(sq->db.wqe_info);
}

static int mlx5e_alloc_xdpsq_fifo(struct mlx5e_xdpsq *sq, int numa)
{
        struct mlx5e_xdp_info_fifo *xdpi_fifo = &sq->db.xdpi_fifo;
        int wq_sz        = mlx5_wq_cyc_get_size(&sq->wq);
        int dsegs_per_wq = wq_sz * MLX5_SEND_WQEBB_NUM_DS;

        xdpi_fifo->xi = kvzalloc_node(sizeof(*xdpi_fifo->xi) * dsegs_per_wq,
                                      GFP_KERNEL, numa);
        if (!xdpi_fifo->xi)
                return -ENOMEM;

        xdpi_fifo->pc   = &sq->xdpi_fifo_pc;
        xdpi_fifo->cc   = &sq->xdpi_fifo_cc;
        xdpi_fifo->mask = dsegs_per_wq - 1;

        return 0;
}

static int mlx5e_alloc_xdpsq_db(struct mlx5e_xdpsq *sq, int numa)
{
        int wq_sz = mlx5_wq_cyc_get_size(&sq->wq);
        int err;

        sq->db.wqe_info = kvzalloc_node(sizeof(*sq->db.wqe_info) * wq_sz,
                                        GFP_KERNEL, numa);
        if (!sq->db.wqe_info)
                return -ENOMEM;

        err = mlx5e_alloc_xdpsq_fifo(sq, numa);
        if (err) {
                mlx5e_free_xdpsq_db(sq);
                return err;
        }

        return 0;
}

static int mlx5e_alloc_xdpsq(struct mlx5e_channel *c,
                             struct mlx5e_params *params,
                             struct xdp_umem *umem,
                             struct mlx5e_sq_param *param,
                             struct mlx5e_xdpsq *sq,
                             bool is_redirect)
{
        void *sqc_wq               = MLX5_ADDR_OF(sqc, param->sqc, wq);
        struct mlx5_core_dev *mdev = c->mdev;
        struct mlx5_wq_cyc *wq = &sq->wq;
        int err;

        sq->pdev      = c->pdev;
        sq->mkey_be   = c->mkey_be;
        sq->channel   = c;
        sq->uar_map   = mdev->mlx5e_res.bfreg.map;
        sq->min_inline_mode = params->tx_min_inline_mode;
        sq->hw_mtu    = MLX5E_SW2HW_MTU(params, params->sw_mtu);
        sq->umem      = umem;

        sq->stats = sq->umem ?
                &c->priv->channel_stats[c->ix].xsksq :
                is_redirect ?
                        &c->priv->channel_stats[c->ix].xdpsq :

                        &c->priv->channel_stats[c->ix].rq_xdpsq;

        param->wq.db_numa_node = cpu_to_node(c->cpu);
        err = mlx5_wq_cyc_create(mdev, &param->wq, sqc_wq, wq, &sq->wq_ctrl);
        if (err)
                return err;
        wq->db = &wq->db[MLX5_SND_DBR];

        err = mlx5e_alloc_xdpsq_db(sq, cpu_to_node(c->cpu));
        if (err)
                goto err_sq_wq_destroy;

        return 0;

err_sq_wq_destroy:
        mlx5_wq_destroy(&sq->wq_ctrl);

        return err;
}

static void mlx5e_free_xdpsq(struct mlx5e_xdpsq *sq)
{
        mlx5e_free_xdpsq_db(sq);
        mlx5_wq_destroy(&sq->wq_ctrl);
}

static void mlx5e_free_icosq_db(struct mlx5e_icosq *sq)
{
        kvfree(sq->db.ico_wqe);
}

static int mlx5e_alloc_icosq_db(struct mlx5e_icosq *sq, int numa)
{
        int wq_sz = mlx5_wq_cyc_get_size(&sq->wq);

        sq->db.ico_wqe = kvzalloc_node(array_size(wq_sz,
                                                  sizeof(*sq->db.ico_wqe)),
                                       GFP_KERNEL, numa);
        if (!sq->db.ico_wqe)
                return -ENOMEM;

        return 0;
}

static int mlx5e_alloc_icosq(struct mlx5e_channel *c,
                             struct mlx5e_sq_param *param,
                             struct mlx5e_icosq *sq)
{
        void *sqc_wq               = MLX5_ADDR_OF(sqc, param->sqc, wq);
        struct mlx5_core_dev *mdev = c->mdev;
        struct mlx5_wq_cyc *wq = &sq->wq;
        int err;

        sq->channel   = c;
        sq->uar_map   = mdev->mlx5e_res.bfreg.map;

        param->wq.db_numa_node = cpu_to_node(c->cpu);
        err = mlx5_wq_cyc_create(mdev, &param->wq, sqc_wq, wq, &sq->wq_ctrl);
        if (err)
                return err;
        wq->db = &wq->db[MLX5_SND_DBR];

        err = mlx5e_alloc_icosq_db(sq, cpu_to_node(c->cpu));
        if (err)
                goto err_sq_wq_destroy;

        return 0;

err_sq_wq_destroy:
        mlx5_wq_destroy(&sq->wq_ctrl);

        return err;
}

static void mlx5e_free_icosq(struct mlx5e_icosq *sq)
{
        mlx5e_free_icosq_db(sq);
        mlx5_wq_destroy(&sq->wq_ctrl);
}

static void mlx5e_free_txqsq_db(struct mlx5e_txqsq *sq)
{
        kvfree(sq->db.wqe_info);
        kvfree(sq->db.dma_fifo);
}

static int mlx5e_alloc_txqsq_db(struct mlx5e_txqsq *sq, int numa)
{
        int wq_sz = mlx5_wq_cyc_get_size(&sq->wq);
        int df_sz = wq_sz * MLX5_SEND_WQEBB_NUM_DS;

        sq->db.dma_fifo = kvzalloc_node(array_size(df_sz,
                                                   sizeof(*sq->db.dma_fifo)),
                                        GFP_KERNEL, numa);
        sq->db.wqe_info = kvzalloc_node(array_size(wq_sz,
                                                   sizeof(*sq->db.wqe_info)),
                                        GFP_KERNEL, numa);
        if (!sq->db.dma_fifo || !sq->db.wqe_info) {
                mlx5e_free_txqsq_db(sq);
                return -ENOMEM;
        }

        sq->dma_fifo_mask = df_sz - 1;

        return 0;
}

static void mlx5e_tx_err_cqe_work(struct work_struct *recover_work);
static int mlx5e_alloc_txqsq(struct mlx5e_channel *c,
                             int txq_ix,
                             struct mlx5e_params *params,
                             struct mlx5e_sq_param *param,
                             struct mlx5e_txqsq *sq,
                             int tc)
{
        void *sqc_wq               = MLX5_ADDR_OF(sqc, param->sqc, wq);
        struct mlx5_core_dev *mdev = c->mdev;
        struct mlx5_wq_cyc *wq = &sq->wq;
        int err;

        sq->pdev      = c->pdev;
        sq->tstamp    = c->tstamp;
        sq->clock     = &mdev->clock;
        sq->mkey_be   = c->mkey_be;
        sq->channel   = c;
        sq->ch_ix     = c->ix;
        sq->txq_ix    = txq_ix;
        sq->uar_map   = mdev->mlx5e_res.bfreg.map;
        sq->min_inline_mode = params->tx_min_inline_mode;
        sq->stats     = &c->priv->channel_stats[c->ix].sq[tc];
        sq->stop_room = MLX5E_SQ_STOP_ROOM;
        INIT_WORK(&sq->recover_work, mlx5e_tx_err_cqe_work);
        if (MLX5_IPSEC_DEV(c->priv->mdev))
                set_bit(MLX5E_SQ_STATE_IPSEC, &sq->state);
        if (mlx5_accel_is_tls_device(c->priv->mdev)) {
                set_bit(MLX5E_SQ_STATE_TLS, &sq->state);
                sq->stop_room += MLX5E_SQ_TLS_ROOM;
        }

        param->wq.db_numa_node = cpu_to_node(c->cpu);
        err = mlx5_wq_cyc_create(mdev, &param->wq, sqc_wq, wq, &sq->wq_ctrl);
        if (err)
                return err;
        wq->db    = &wq->db[MLX5_SND_DBR];

        err = mlx5e_alloc_txqsq_db(sq, cpu_to_node(c->cpu));
        if (err)
                goto err_sq_wq_destroy;

        INIT_WORK(&sq->dim.work, mlx5e_tx_dim_work);
        sq->dim.mode = params->tx_cq_moderation.cq_period_mode;

        return 0;

err_sq_wq_destroy:
        mlx5_wq_destroy(&sq->wq_ctrl);

        return err;
}

static void mlx5e_free_txqsq(struct mlx5e_txqsq *sq)
{
        mlx5e_free_txqsq_db(sq);
        mlx5_wq_destroy(&sq->wq_ctrl);
}

struct mlx5e_create_sq_param {
        struct mlx5_wq_ctrl        *wq_ctrl;
        u32                         cqn;
        u32                         tisn;
        u8                          tis_lst_sz;
        u8                          min_inline_mode;
};

static int mlx5e_create_sq(struct mlx5_core_dev *mdev,
                           struct mlx5e_sq_param *param,
                           struct mlx5e_create_sq_param *csp,
                           u32 *sqn)
{
        void *in;
        void *sqc;
        void *wq;
        int inlen;
        int err;

        inlen = MLX5_ST_SZ_BYTES(create_sq_in) +
                sizeof(u64) * csp->wq_ctrl->buf.npages;
        in = kvzalloc(inlen, GFP_KERNEL);
        if (!in)
                return -ENOMEM;

        sqc = MLX5_ADDR_OF(create_sq_in, in, ctx);
        wq = MLX5_ADDR_OF(sqc, sqc, wq);

        memcpy(sqc, param->sqc, sizeof(param->sqc));
        MLX5_SET(sqc,  sqc, tis_lst_sz, csp->tis_lst_sz);
        MLX5_SET(sqc,  sqc, tis_num_0, csp->tisn);
        MLX5_SET(sqc,  sqc, cqn, csp->cqn);

        if (MLX5_CAP_ETH(mdev, wqe_inline_mode) == MLX5_CAP_INLINE_MODE_VPORT_CONTEXT)
                MLX5_SET(sqc,  sqc, min_wqe_inline_mode, csp->min_inline_mode);

        MLX5_SET(sqc,  sqc, state, MLX5_SQC_STATE_RST);
        MLX5_SET(sqc,  sqc, flush_in_error_en, 1);

        MLX5_SET(wq,   wq, wq_type,       MLX5_WQ_TYPE_CYCLIC);
        MLX5_SET(wq,   wq, uar_page,      mdev->mlx5e_res.bfreg.index);
        MLX5_SET(wq,   wq, log_wq_pg_sz,  csp->wq_ctrl->buf.page_shift -
                                          MLX5_ADAPTER_PAGE_SHIFT);
        MLX5_SET64(wq, wq, dbr_addr,      csp->wq_ctrl->db.dma);

        mlx5_fill_page_frag_array(&csp->wq_ctrl->buf,
                                  (__be64 *)MLX5_ADDR_OF(wq, wq, pas));

        err = mlx5_core_create_sq(mdev, in, inlen, sqn);

        kvfree(in);

        return err;
}

int mlx5e_modify_sq(struct mlx5_core_dev *mdev, u32 sqn,
                    struct mlx5e_modify_sq_param *p)
{
        void *in;
        void *sqc;
        int inlen;
        int err;

        inlen = MLX5_ST_SZ_BYTES(modify_sq_in);
        in = kvzalloc(inlen, GFP_KERNEL);
        if (!in)
                return -ENOMEM;

        sqc = MLX5_ADDR_OF(modify_sq_in, in, ctx);

        MLX5_SET(modify_sq_in, in, sq_state, p->curr_state);
        MLX5_SET(sqc, sqc, state, p->next_state);
        if (p->rl_update && p->next_state == MLX5_SQC_STATE_RDY) {
                MLX5_SET64(modify_sq_in, in, modify_bitmask, 1);
                MLX5_SET(sqc,  sqc, packet_pacing_rate_limit_index, p->rl_index);
        }

        err = mlx5_core_modify_sq(mdev, sqn, in, inlen);

        kvfree(in);

        return err;
}

static void mlx5e_destroy_sq(struct mlx5_core_dev *mdev, u32 sqn)
{
        mlx5_core_destroy_sq(mdev, sqn);
}

static int mlx5e_create_sq_rdy(struct mlx5_core_dev *mdev,
                               struct mlx5e_sq_param *param,
                               struct mlx5e_create_sq_param *csp,
                               u32 *sqn)
{
        struct mlx5e_modify_sq_param msp = {0};
        int err;

        err = mlx5e_create_sq(mdev, param, csp, sqn);
        if (err)
                return err;

        msp.curr_state = MLX5_SQC_STATE_RST;
        msp.next_state = MLX5_SQC_STATE_RDY;
        err = mlx5e_modify_sq(mdev, *sqn, &msp);
        if (err)
                mlx5e_destroy_sq(mdev, *sqn);

        return err;
}

static int mlx5e_set_sq_maxrate(struct net_device *dev,
                                struct mlx5e_txqsq *sq, u32 rate);

static int mlx5e_open_txqsq(struct mlx5e_channel *c,
                            u32 tisn,
                            int txq_ix,
                            struct mlx5e_params *params,
                            struct mlx5e_sq_param *param,
                            struct mlx5e_txqsq *sq,
                            int tc)
{
        struct mlx5e_create_sq_param csp = {};
        u32 tx_rate;
        int err;

        err = mlx5e_alloc_txqsq(c, txq_ix, params, param, sq, tc);
        if (err)
                return err;

        csp.tisn            = tisn;
        csp.tis_lst_sz      = 1;
        csp.cqn             = sq->cq.mcq.cqn;
        csp.wq_ctrl         = &sq->wq_ctrl;
        csp.min_inline_mode = sq->min_inline_mode;
        err = mlx5e_create_sq_rdy(c->mdev, param, &csp, &sq->sqn);
        if (err)
                goto err_free_txqsq;

        tx_rate = c->priv->tx_rates[sq->txq_ix];
        if (tx_rate)
                mlx5e_set_sq_maxrate(c->netdev, sq, tx_rate);

        if (params->tx_dim_enabled)
                sq->state |= BIT(MLX5E_SQ_STATE_AM);

        return 0;

err_free_txqsq:
        clear_bit(MLX5E_SQ_STATE_ENABLED, &sq->state);
        mlx5e_free_txqsq(sq);

        return err;
}

void mlx5e_activate_txqsq(struct mlx5e_txqsq *sq)
{
        sq->txq = netdev_get_tx_queue(sq->channel->netdev, sq->txq_ix);
        set_bit(MLX5E_SQ_STATE_ENABLED, &sq->state);
        netdev_tx_reset_queue(sq->txq);
        netif_tx_start_queue(sq->txq);
}

void mlx5e_tx_disable_queue(struct netdev_queue *txq)
{
        __netif_tx_lock_bh(txq);
        netif_tx_stop_queue(txq);
        __netif_tx_unlock_bh(txq);
}

static void mlx5e_deactivate_txqsq(struct mlx5e_txqsq *sq)
{
        struct mlx5e_channel *c = sq->channel;
        struct mlx5_wq_cyc *wq = &sq->wq;

        clear_bit(MLX5E_SQ_STATE_ENABLED, &sq->state);
        /* prevent netif_tx_wake_queue */
        napi_synchronize(&c->napi);

        mlx5e_tx_disable_queue(sq->txq);

        /* last doorbell out, godspeed .. */
        if (mlx5e_wqc_has_room_for(wq, sq->cc, sq->pc, 1)) {
                u16 pi = mlx5_wq_cyc_ctr2ix(wq, sq->pc);
                struct mlx5e_tx_wqe *nop;

                sq->db.wqe_info[pi].skb = NULL;
                nop = mlx5e_post_nop(wq, sq->sqn, &sq->pc);
                mlx5e_notify_hw(wq, sq->pc, sq->uar_map, &nop->ctrl);
        }
}

static void mlx5e_close_txqsq(struct mlx5e_txqsq *sq)
{
        struct mlx5e_channel *c = sq->channel;
        struct mlx5_core_dev *mdev = c->mdev;
        struct mlx5_rate_limit rl = {0};

        cancel_work_sync(&sq->dim.work);
        cancel_work_sync(&sq->recover_work);
        mlx5e_destroy_sq(mdev, sq->sqn);
        if (sq->rate_limit) {
                rl.rate = sq->rate_limit;
                mlx5_rl_remove_rate(mdev, &rl);
        }
        mlx5e_free_txqsq_descs(sq);
        mlx5e_free_txqsq(sq);
}

static void mlx5e_tx_err_cqe_work(struct work_struct *recover_work)
{
        struct mlx5e_txqsq *sq = container_of(recover_work, struct mlx5e_txqsq,
                                              recover_work);

        mlx5e_tx_reporter_err_cqe(sq);
}

int mlx5e_open_icosq(struct mlx5e_channel *c, struct mlx5e_params *params,
                     struct mlx5e_sq_param *param, struct mlx5e_icosq *sq)
{
        struct mlx5e_create_sq_param csp = {};
        int err;

        err = mlx5e_alloc_icosq(c, param, sq);
        if (err)
                return err;

        csp.cqn             = sq->cq.mcq.cqn;
        csp.wq_ctrl         = &sq->wq_ctrl;
        csp.min_inline_mode = params->tx_min_inline_mode;
        set_bit(MLX5E_SQ_STATE_ENABLED, &sq->state);
        err = mlx5e_create_sq_rdy(c->mdev, param, &csp, &sq->sqn);
        if (err)
                goto err_free_icosq;

        return 0;

err_free_icosq:
        clear_bit(MLX5E_SQ_STATE_ENABLED, &sq->state);
        mlx5e_free_icosq(sq);

        return err;
}

void mlx5e_close_icosq(struct mlx5e_icosq *sq)
{
        struct mlx5e_channel *c = sq->channel;

        clear_bit(MLX5E_SQ_STATE_ENABLED, &sq->state);
        napi_synchronize(&c->napi);

        mlx5e_destroy_sq(c->mdev, sq->sqn);
        mlx5e_free_icosq(sq);
}

int mlx5e_open_xdpsq(struct mlx5e_channel *c, struct mlx5e_params *params,
                     struct mlx5e_sq_param *param, struct xdp_umem *umem,
                     struct mlx5e_xdpsq *sq, bool is_redirect)
{
        struct mlx5e_create_sq_param csp = {};
        int err;

        err = mlx5e_alloc_xdpsq(c, params, umem, param, sq, is_redirect);
        if (err)
                return err;

        csp.tis_lst_sz      = 1;
        csp.tisn            = c->priv->tisn[0]; /* tc = 0 */
        csp.cqn             = sq->cq.mcq.cqn;
        csp.wq_ctrl         = &sq->wq_ctrl;
        csp.min_inline_mode = sq->min_inline_mode;
        set_bit(MLX5E_SQ_STATE_ENABLED, &sq->state);
        err = mlx5e_create_sq_rdy(c->mdev, param, &csp, &sq->sqn);
        if (err)
                goto err_free_xdpsq;

        mlx5e_set_xmit_fp(sq, param->is_mpw);

        if (!param->is_mpw) {
                unsigned int ds_cnt = MLX5E_XDP_TX_DS_COUNT;
                unsigned int inline_hdr_sz = 0;
                int i;

                if (sq->min_inline_mode != MLX5_INLINE_MODE_NONE) {
                        inline_hdr_sz = MLX5E_XDP_MIN_INLINE;
                        ds_cnt++;
                }

                /* Pre initialize fixed WQE fields */
                for (i = 0; i < mlx5_wq_cyc_get_size(&sq->wq); i++) {
                        struct mlx5e_xdp_wqe_info *wi  = &sq->db.wqe_info[i];
                        struct mlx5e_tx_wqe      *wqe  = mlx5_wq_cyc_get_wqe(&sq->wq, i);
                        struct mlx5_wqe_ctrl_seg *cseg = &wqe->ctrl;
                        struct mlx5_wqe_eth_seg  *eseg = &wqe->eth;
                        struct mlx5_wqe_data_seg *dseg;

                        cseg->qpn_ds = cpu_to_be32((sq->sqn << 8) | ds_cnt);
                        eseg->inline_hdr.sz = cpu_to_be16(inline_hdr_sz);

                        dseg = (struct mlx5_wqe_data_seg *)cseg + (ds_cnt - 1);
                        dseg->lkey = sq->mkey_be;

                        wi->num_wqebbs = 1;
                        wi->num_pkts   = 1;
                }
        }

        return 0;

err_free_xdpsq:
        clear_bit(MLX5E_SQ_STATE_ENABLED, &sq->state);
        mlx5e_free_xdpsq(sq);

        return err;
}

void mlx5e_close_xdpsq(struct mlx5e_xdpsq *sq)
{
        struct mlx5e_channel *c = sq->channel;

        clear_bit(MLX5E_SQ_STATE_ENABLED, &sq->state);
        napi_synchronize(&c->napi);

        mlx5e_destroy_sq(c->mdev, sq->sqn);
        mlx5e_free_xdpsq_descs(sq);
        mlx5e_free_xdpsq(sq);
}

static int mlx5e_alloc_cq_common(struct mlx5_core_dev *mdev,
                                 struct mlx5e_cq_param *param,
                                 struct mlx5e_cq *cq)
{
        struct mlx5_core_cq *mcq = &cq->mcq;
        int eqn_not_used;
        unsigned int irqn;
        int err;
        u32 i;

        err = mlx5_vector2eqn(mdev, param->eq_ix, &eqn_not_used, &irqn);
        if (err)
                return err;

        err = mlx5_cqwq_create(mdev, &param->wq, param->cqc, &cq->wq,
                               &cq->wq_ctrl);
        if (err)
                return err;

        mcq->cqe_sz     = 64;
        mcq->set_ci_db  = cq->wq_ctrl.db.db;
        mcq->arm_db     = cq->wq_ctrl.db.db + 1;
        *mcq->set_ci_db = 0;
        *mcq->arm_db    = 0;
        mcq->vector     = param->eq_ix;
        mcq->comp       = mlx5e_completion_event;
        mcq->event      = mlx5e_cq_error_event;
        mcq->irqn       = irqn;

        for (i = 0; i < mlx5_cqwq_get_size(&cq->wq); i++) {
                struct mlx5_cqe64 *cqe = mlx5_cqwq_get_wqe(&cq->wq, i);

                cqe->op_own = 0xf1;
        }

        cq->mdev = mdev;

        return 0;
}

static int mlx5e_alloc_cq(struct mlx5e_channel *c,
                          struct mlx5e_cq_param *param,
                          struct mlx5e_cq *cq)
{
        struct mlx5_core_dev *mdev = c->priv->mdev;
        int err;

        param->wq.buf_numa_node = cpu_to_node(c->cpu);
        param->wq.db_numa_node  = cpu_to_node(c->cpu);
        param->eq_ix   = c->ix;

        err = mlx5e_alloc_cq_common(mdev, param, cq);

        cq->napi    = &c->napi;
        cq->channel = c;

        return err;
}

static void mlx5e_free_cq(struct mlx5e_cq *cq)
{
        mlx5_wq_destroy(&cq->wq_ctrl);
}

static int mlx5e_create_cq(struct mlx5e_cq *cq, struct mlx5e_cq_param *param)
{
        u32 out[MLX5_ST_SZ_DW(create_cq_out)];
        struct mlx5_core_dev *mdev = cq->mdev;
        struct mlx5_core_cq *mcq = &cq->mcq;

        void *in;
        void *cqc;
        int inlen;
        unsigned int irqn_not_used;
        int eqn;
        int err;

        err = mlx5_vector2eqn(mdev, param->eq_ix, &eqn, &irqn_not_used);
        if (err)
                return err;

        inlen = MLX5_ST_SZ_BYTES(create_cq_in) +
                sizeof(u64) * cq->wq_ctrl.buf.npages;
        in = kvzalloc(inlen, GFP_KERNEL);
        if (!in)
                return -ENOMEM;

        cqc = MLX5_ADDR_OF(create_cq_in, in, cq_context);

        memcpy(cqc, param->cqc, sizeof(param->cqc));

        mlx5_fill_page_frag_array(&cq->wq_ctrl.buf,
                                  (__be64 *)MLX5_ADDR_OF(create_cq_in, in, pas));

        MLX5_SET(cqc,   cqc, cq_period_mode, param->cq_period_mode);
        MLX5_SET(cqc,   cqc, c_eqn,         eqn);
        MLX5_SET(cqc,   cqc, uar_page,      mdev->priv.uar->index);
        MLX5_SET(cqc,   cqc, log_page_size, cq->wq_ctrl.buf.page_shift -
                                            MLX5_ADAPTER_PAGE_SHIFT);
        MLX5_SET64(cqc, cqc, dbr_addr,      cq->wq_ctrl.db.dma);

        err = mlx5_core_create_cq(mdev, mcq, in, inlen, out, sizeof(out));

        kvfree(in);

        if (err)
                return err;

        mlx5e_cq_arm(cq);

        return 0;
}

static void mlx5e_destroy_cq(struct mlx5e_cq *cq)
{
        mlx5_core_destroy_cq(cq->mdev, &cq->mcq);
}

int mlx5e_open_cq(struct mlx5e_channel *c, struct dim_cq_moder moder,
                  struct mlx5e_cq_param *param, struct mlx5e_cq *cq)
{
        struct mlx5_core_dev *mdev = c->mdev;
        int err;

        err = mlx5e_alloc_cq(c, param, cq);
        if (err)
                return err;

        err = mlx5e_create_cq(cq, param);
        if (err)
                goto err_free_cq;

        if (MLX5_CAP_GEN(mdev, cq_moderation))
                mlx5_core_modify_cq_moderation(mdev, &cq->mcq, moder.usec, moder.pkts);
        return 0;

err_free_cq:
        mlx5e_free_cq(cq);

        return err;
}

void mlx5e_close_cq(struct mlx5e_cq *cq)
{
        mlx5e_destroy_cq(cq);
        mlx5e_free_cq(cq);
}

static int mlx5e_open_tx_cqs(struct mlx5e_channel *c,
                             struct mlx5e_params *params,
                             struct mlx5e_channel_param *cparam)
{
        int err;
        int tc;

        for (tc = 0; tc < c->num_tc; tc++) {
                err = mlx5e_open_cq(c, params->tx_cq_moderation,
                                    &cparam->tx_cq, &c->sq[tc].cq);
                if (err)
                        goto err_close_tx_cqs;
        }

        return 0;

err_close_tx_cqs:
        for (tc--; tc >= 0; tc--)
                mlx5e_close_cq(&c->sq[tc].cq);

        return err;
}

static void mlx5e_close_tx_cqs(struct mlx5e_channel *c)
{
        int tc;

        for (tc = 0; tc < c->num_tc; tc++)
                mlx5e_close_cq(&c->sq[tc].cq);
}

static int mlx5e_open_sqs(struct mlx5e_channel *c,
                          struct mlx5e_params *params,
                          struct mlx5e_channel_param *cparam)
{
        struct mlx5e_priv *priv = c->priv;
        int err, tc;

        for (tc = 0; tc < params->num_tc; tc++) {
                int txq_ix = c->ix + tc * priv->max_nch;

                err = mlx5e_open_txqsq(c, c->priv->tisn[tc], txq_ix,
                                       params, &cparam->sq, &c->sq[tc], tc);
                if (err)
                        goto err_close_sqs;
        }

        return 0;

err_close_sqs:
        for (tc--; tc >= 0; tc--)
                mlx5e_close_txqsq(&c->sq[tc]);

        return err;
}

static void mlx5e_close_sqs(struct mlx5e_channel *c)
{
        int tc;

        for (tc = 0; tc < c->num_tc; tc++)
                mlx5e_close_txqsq(&c->sq[tc]);
}

static int mlx5e_set_sq_maxrate(struct net_device *dev,
                                struct mlx5e_txqsq *sq, u32 rate)
{
        struct mlx5e_priv *priv = netdev_priv(dev);
        struct mlx5_core_dev *mdev = priv->mdev;
        struct mlx5e_modify_sq_param msp = {0};
        struct mlx5_rate_limit rl = {0};
        u16 rl_index = 0;
        int err;

        if (rate == sq->rate_limit)
                /* nothing to do */
                return 0;

        if (sq->rate_limit) {
                rl.rate = sq->rate_limit;
                /* remove current rl index to free space to next ones */
                mlx5_rl_remove_rate(mdev, &rl);
        }

        sq->rate_limit = 0;

        if (rate) {
                rl.rate = rate;
                err = mlx5_rl_add_rate(mdev, &rl_index, &rl);
                if (err) {
                        netdev_err(dev, "Failed configuring rate %u: %d\n",
                                   rate, err);
                        return err;
                }
        }

        msp.curr_state = MLX5_SQC_STATE_RDY;
        msp.next_state = MLX5_SQC_STATE_RDY;
        msp.rl_index   = rl_index;
        msp.rl_update  = true;
        err = mlx5e_modify_sq(mdev, sq->sqn, &msp);
        if (err) {
                netdev_err(dev, "Failed configuring rate %u: %d\n",
                           rate, err);
                /* remove the rate from the table */
                if (rate)
                        mlx5_rl_remove_rate(mdev, &rl);
                return err;
        }

        sq->rate_limit = rate;
        return 0;
}

static int mlx5e_set_tx_maxrate(struct net_device *dev, int index, u32 rate)
{
        struct mlx5e_priv *priv = netdev_priv(dev);
        struct mlx5_core_dev *mdev = priv->mdev;
        struct mlx5e_txqsq *sq = priv->txq2sq[index];
        int err = 0;

        if (!mlx5_rl_is_supported(mdev)) {
                netdev_err(dev, "Rate limiting is not supported on this device\n");
                return -EINVAL;
        }

        /* rate is given in Mb/sec, HW config is in Kb/sec */
        rate = rate << 10;

        /* Check whether rate in valid range, 0 is always valid */
        if (rate && !mlx5_rl_is_in_range(mdev, rate)) {
                netdev_err(dev, "TX rate %u, is not in range\n", rate);
                return -ERANGE;
        }

        mutex_lock(&priv->state_lock);
        if (test_bit(MLX5E_STATE_OPENED, &priv->state))
                err = mlx5e_set_sq_maxrate(dev, sq, rate);
        if (!err)
                priv->tx_rates[index] = rate;
        mutex_unlock(&priv->state_lock);

        return err;
}

static int mlx5e_alloc_xps_cpumask(struct mlx5e_channel *c,
                                   struct mlx5e_params *params)
{
        int num_comp_vectors = mlx5_comp_vectors_count(c->mdev);
        int irq;

        if (!zalloc_cpumask_var(&c->xps_cpumask, GFP_KERNEL))
                return -ENOMEM;

        for (irq = c->ix; irq < num_comp_vectors; irq += params->num_channels) {
                int cpu = cpumask_first(mlx5_comp_irq_get_affinity_mask(c->mdev, irq));

                cpumask_set_cpu(cpu, c->xps_cpumask);
        }

        return 0;
}

static void mlx5e_free_xps_cpumask(struct mlx5e_channel *c)
{
        free_cpumask_var(c->xps_cpumask);
}

static int mlx5e_open_queues(struct mlx5e_channel *c,
                             struct mlx5e_params *params,
                             struct mlx5e_channel_param *cparam)
{
        struct dim_cq_moder icocq_moder = {0, 0};
        int err;

        err = mlx5e_open_cq(c, icocq_moder, &cparam->icosq_cq, &c->icosq.cq);
        if (err)
                return err;

        err = mlx5e_open_tx_cqs(c, params, cparam);
        if (err)
                goto err_close_icosq_cq;

        err = mlx5e_open_cq(c, params->tx_cq_moderation, &cparam->tx_cq, &c->xdpsq.cq);
        if (err)
                goto err_close_tx_cqs;

        err = mlx5e_open_cq(c, params->rx_cq_moderation, &cparam->rx_cq, &c->rq.cq);
        if (err)
                goto err_close_xdp_tx_cqs;

        /* XDP SQ CQ params are same as normal TXQ sq CQ params */
        err = c->xdp ? mlx5e_open_cq(c, params->tx_cq_moderation,
                                     &cparam->tx_cq, &c->rq_xdpsq.cq) : 0;
        if (err)
                goto err_close_rx_cq;

        napi_enable(&c->napi);

        err = mlx5e_open_icosq(c, params, &cparam->icosq, &c->icosq);
        if (err)
                goto err_disable_napi;

        err = mlx5e_open_sqs(c, params, cparam);
        if (err)
                goto err_close_icosq;

        if (c->xdp) {
                err = mlx5e_open_xdpsq(c, params, &cparam->xdp_sq, NULL,
                                       &c->rq_xdpsq, false);
                if (err)
                        goto err_close_sqs;
        }

        err = mlx5e_open_rq(c, params, &cparam->rq, NULL, NULL, &c->rq);
        if (err)
                goto err_close_xdp_sq;

        err = mlx5e_open_xdpsq(c, params, &cparam->xdp_sq, NULL, &c->xdpsq, true);
        if (err)
                goto err_close_rq;

        return 0;

err_close_rq:
        mlx5e_close_rq(&c->rq);

err_close_xdp_sq:
        if (c->xdp)
                mlx5e_close_xdpsq(&c->rq_xdpsq);

err_close_sqs:
        mlx5e_close_sqs(c);

err_close_icosq:
        mlx5e_close_icosq(&c->icosq);

err_disable_napi:
        napi_disable(&c->napi);

        if (c->xdp)
                mlx5e_close_cq(&c->rq_xdpsq.cq);

err_close_rx_cq:
        mlx5e_close_cq(&c->rq.cq);

err_close_xdp_tx_cqs:
        mlx5e_close_cq(&c->xdpsq.cq);

err_close_tx_cqs:
        mlx5e_close_tx_cqs(c);

err_close_icosq_cq:
        mlx5e_close_cq(&c->icosq.cq);

        return err;
}

static void mlx5e_close_queues(struct mlx5e_channel *c)
{
        mlx5e_close_xdpsq(&c->xdpsq);
        mlx5e_close_rq(&c->rq);
        if (c->xdp)
                mlx5e_close_xdpsq(&c->rq_xdpsq);
        mlx5e_close_sqs(c);
        mlx5e_close_icosq(&c->icosq);
        napi_disable(&c->napi);
        if (c->xdp)
                mlx5e_close_cq(&c->rq_xdpsq.cq);
        mlx5e_close_cq(&c->rq.cq);
        mlx5e_close_cq(&c->xdpsq.cq);
        mlx5e_close_tx_cqs(c);
        mlx5e_close_cq(&c->icosq.cq);
}

static int mlx5e_open_channel(struct mlx5e_priv *priv, int ix,
                              struct mlx5e_params *params,
                              struct mlx5e_channel_param *cparam,
                              struct xdp_umem *umem,
                              struct mlx5e_channel **cp)
{
        int cpu = cpumask_first(mlx5_comp_irq_get_affinity_mask(priv->mdev, ix));
        struct net_device *netdev = priv->netdev;
        struct mlx5e_xsk_param xsk;
        struct mlx5e_channel *c;
        unsigned int irq;
        int err;
        int eqn;

        err = mlx5_vector2eqn(priv->mdev, ix, &eqn, &irq);
        if (err)
                return err;

        c = kvzalloc_node(sizeof(*c), GFP_KERNEL, cpu_to_node(cpu));
        if (!c)
                return -ENOMEM;

        c->priv     = priv;
        c->mdev     = priv->mdev;
        c->tstamp   = &priv->tstamp;
        c->ix       = ix;
        c->cpu      = cpu;
        c->pdev     = priv->mdev->device;
        c->netdev   = priv->netdev;
        c->mkey_be  = cpu_to_be32(priv->mdev->mlx5e_res.mkey.key);
        c->num_tc   = params->num_tc;
        c->xdp      = !!params->xdp_prog;
        c->stats    = &priv->channel_stats[ix].ch;
        c->irq_desc = irq_to_desc(irq);

        err = mlx5e_alloc_xps_cpumask(c, params);
        if (err)
                goto err_free_channel;

        netif_napi_add(netdev, &c->napi, mlx5e_napi_poll, 64);

        err = mlx5e_open_queues(c, params, cparam);
        if (unlikely(err))
                goto err_napi_del;

        if (umem) {
                mlx5e_build_xsk_param(umem, &xsk);
                err = mlx5e_open_xsk(priv, params, &xsk, umem, c);
                if (unlikely(err))
                        goto err_close_queues;
        }

        *cp = c;

        return 0;

err_close_queues:
        mlx5e_close_queues(c);

err_napi_del:
        netif_napi_del(&c->napi);
        mlx5e_free_xps_cpumask(c);

err_free_channel:
        kvfree(c);

        return err;
}

static void mlx5e_activate_channel(struct mlx5e_channel *c)
{
        int tc;

        for (tc = 0; tc < c->num_tc; tc++)
                mlx5e_activate_txqsq(&c->sq[tc]);
        mlx5e_activate_rq(&c->rq);
        netif_set_xps_queue(c->netdev, c->xps_cpumask, c->ix);

        if (test_bit(MLX5E_CHANNEL_STATE_XSK, c->state))
                mlx5e_activate_xsk(c);
}

static void mlx5e_deactivate_channel(struct mlx5e_channel *c)
{

        int tc;

        if (test_bit(MLX5E_CHANNEL_STATE_XSK, c->state))
                mlx5e_deactivate_xsk(c);

        mlx5e_deactivate_rq(&c->rq);
        for (tc = 0; tc < c->num_tc; tc++)
                mlx5e_deactivate_txqsq(&c->sq[tc]);
}

static void mlx5e_close_channel(struct mlx5e_channel *c)
{
        if (test_bit(MLX5E_CHANNEL_STATE_XSK, c->state))
                mlx5e_close_xsk(c);
        mlx5e_close_queues(c);
        netif_napi_del(&c->napi);
        mlx5e_free_xps_cpumask(c);

        kvfree(c);
}

#define DEFAULT_FRAG_SIZE (2048)

static void mlx5e_build_rq_frags_info(struct mlx5_core_dev *mdev,
                                      struct mlx5e_params *params,
                                      struct mlx5e_xsk_param *xsk,
                                      struct mlx5e_rq_frags_info *info)
{
        u32 byte_count = MLX5E_SW2HW_MTU(params, params->sw_mtu);
        int frag_size_max = DEFAULT_FRAG_SIZE;
        u32 buf_size = 0;
        int i;

#ifdef CONFIG_MLX5_EN_IPSEC
        if (MLX5_IPSEC_DEV(mdev))
                byte_count += MLX5E_METADATA_ETHER_LEN;
#endif

        if (mlx5e_rx_is_linear_skb(params, xsk)) {
                int frag_stride;

                frag_stride = mlx5e_rx_get_linear_frag_sz(params, xsk);
                frag_stride = roundup_pow_of_two(frag_stride);

                info->arr[0].frag_size = byte_count;
                info->arr[0].frag_stride = frag_stride;
                info->num_frags = 1;
                info->wqe_bulk = PAGE_SIZE / frag_stride;
                goto out;
        }

        if (byte_count > PAGE_SIZE +
            (MLX5E_MAX_RX_FRAGS - 1) * frag_size_max)
                frag_size_max = PAGE_SIZE;

        i = 0;
        while (buf_size < byte_count) {
                int frag_size = byte_count - buf_size;

                if (i < MLX5E_MAX_RX_FRAGS - 1)
                        frag_size = min(frag_size, frag_size_max);

                info->arr[i].frag_size = frag_size;
                info->arr[i].frag_stride = roundup_pow_of_two(frag_size);

                buf_size += frag_size;
                i++;
        }
        info->num_frags = i;
        /* number of different wqes sharing a page */
        info->wqe_bulk = 1 + (info->num_frags % 2);

out:
        info->wqe_bulk = max_t(u8, info->wqe_bulk, 8);
        info->log_num_frags = order_base_2(info->num_frags);
}

static inline u8 mlx5e_get_rqwq_log_stride(u8 wq_type, int ndsegs)
{
        int sz = sizeof(struct mlx5_wqe_data_seg) * ndsegs;

        switch (wq_type) {
        case MLX5_WQ_TYPE_LINKED_LIST_STRIDING_RQ:
                sz += sizeof(struct mlx5e_rx_wqe_ll);
                break;
        default: /* MLX5_WQ_TYPE_CYCLIC */
                sz += sizeof(struct mlx5e_rx_wqe_cyc);
        }

        return order_base_2(sz);
}

static u8 mlx5e_get_rq_log_wq_sz(void *rqc)
{
        void *wq = MLX5_ADDR_OF(rqc, rqc, wq);

        return MLX5_GET(wq, wq, log_wq_sz);
}

void mlx5e_build_rq_param(struct mlx5e_priv *priv,
                          struct mlx5e_params *params,
                          struct mlx5e_xsk_param *xsk,
                          struct mlx5e_rq_param *param)
{
        struct mlx5_core_dev *mdev = priv->mdev;
        void *rqc = param->rqc;
        void *wq = MLX5_ADDR_OF(rqc, rqc, wq);
        int ndsegs = 1;

        switch (params->rq_wq_type) {
        case MLX5_WQ_TYPE_LINKED_LIST_STRIDING_RQ:
                MLX5_SET(wq, wq, log_wqe_num_of_strides,
                         mlx5e_mpwqe_get_log_num_strides(mdev, params, xsk) -
                         MLX5_MPWQE_LOG_NUM_STRIDES_BASE);
                MLX5_SET(wq, wq, log_wqe_stride_size,
                         mlx5e_mpwqe_get_log_stride_size(mdev, params, xsk) -
                         MLX5_MPWQE_LOG_STRIDE_SZ_BASE);
                MLX5_SET(wq, wq, log_wq_sz, mlx5e_mpwqe_get_log_rq_size(params, xsk));
                break;
        default: /* MLX5_WQ_TYPE_CYCLIC */
                MLX5_SET(wq, wq, log_wq_sz, params->log_rq_mtu_frames);
                mlx5e_build_rq_frags_info(mdev, params, xsk, &param->frags_info);
                ndsegs = param->frags_info.num_frags;
        }

        MLX5_SET(wq, wq, wq_type,          params->rq_wq_type);
        MLX5_SET(wq, wq, end_padding_mode, MLX5_WQ_END_PAD_MODE_ALIGN);
        MLX5_SET(wq, wq, log_wq_stride,
                 mlx5e_get_rqwq_log_stride(params->rq_wq_type, ndsegs));
        MLX5_SET(wq, wq, pd,               mdev->mlx5e_res.pdn);
        MLX5_SET(rqc, rqc, counter_set_id, priv->q_counter);
        MLX5_SET(rqc, rqc, vsd,            params->vlan_strip_disable);
        MLX5_SET(rqc, rqc, scatter_fcs,    params->scatter_fcs_en);

        param->wq.buf_numa_node = dev_to_node(mdev->device);
}

static void mlx5e_build_drop_rq_param(struct mlx5e_priv *priv,
                                      struct mlx5e_rq_param *param)
{
        struct mlx5_core_dev *mdev = priv->mdev;
        void *rqc = param->rqc;
        void *wq = MLX5_ADDR_OF(rqc, rqc, wq);

        MLX5_SET(wq, wq, wq_type, MLX5_WQ_TYPE_CYCLIC);
        MLX5_SET(wq, wq, log_wq_stride,
                 mlx5e_get_rqwq_log_stride(MLX5_WQ_TYPE_CYCLIC, 1));
        MLX5_SET(rqc, rqc, counter_set_id, priv->drop_rq_q_counter);

        param->wq.buf_numa_node = dev_to_node(mdev->device);
}

void mlx5e_build_sq_param_common(struct mlx5e_priv *priv,
                                 struct mlx5e_sq_param *param)
{
        void *sqc = param->sqc;
        void *wq = MLX5_ADDR_OF(sqc, sqc, wq);

        MLX5_SET(wq, wq, log_wq_stride, ilog2(MLX5_SEND_WQE_BB));
        MLX5_SET(wq, wq, pd,            priv->mdev->mlx5e_res.pdn);

        param->wq.buf_numa_node = dev_to_node(priv->mdev->device);
}

static void mlx5e_build_sq_param(struct mlx5e_priv *priv,
                                 struct mlx5e_params *params,
                                 struct mlx5e_sq_param *param)
{
        void *sqc = param->sqc;
        void *wq = MLX5_ADDR_OF(sqc, sqc, wq);
        bool allow_swp;

        allow_swp = mlx5_geneve_tx_allowed(priv->mdev) ||
                    !!MLX5_IPSEC_DEV(priv->mdev);
        mlx5e_build_sq_param_common(priv, param);
        MLX5_SET(wq, wq, log_wq_sz, params->log_sq_size);
        MLX5_SET(sqc, sqc, allow_swp, allow_swp);
}

static void mlx5e_build_common_cq_param(struct mlx5e_priv *priv,
                                        struct mlx5e_cq_param *param)
{
        void *cqc = param->cqc;

        MLX5_SET(cqc, cqc, uar_page, priv->mdev->priv.uar->index);
        if (MLX5_CAP_GEN(priv->mdev, cqe_128_always) && cache_line_size() >= 128)
                MLX5_SET(cqc, cqc, cqe_sz, CQE_STRIDE_128_PAD);
}

void mlx5e_build_rx_cq_param(struct mlx5e_priv *priv,
                             struct mlx5e_params *params,
                             struct mlx5e_xsk_param *xsk,
                             struct mlx5e_cq_param *param)
{
        struct mlx5_core_dev *mdev = priv->mdev;
        void *cqc = param->cqc;
        u8 log_cq_size;

        switch (params->rq_wq_type) {
        case MLX5_WQ_TYPE_LINKED_LIST_STRIDING_RQ:
                log_cq_size = mlx5e_mpwqe_get_log_rq_size(params, xsk) +
                        mlx5e_mpwqe_get_log_num_strides(mdev, params, xsk);
                break;
        default: /* MLX5_WQ_TYPE_CYCLIC */
                log_cq_size = params->log_rq_mtu_frames;
        }

        MLX5_SET(cqc, cqc, log_cq_size, log_cq_size);
        if (MLX5E_GET_PFLAG(params, MLX5E_PFLAG_RX_CQE_COMPRESS)) {
                MLX5_SET(cqc, cqc, mini_cqe_res_format, MLX5_CQE_FORMAT_CSUM);
                MLX5_SET(cqc, cqc, cqe_comp_en, 1);
        }

        mlx5e_build_common_cq_param(priv, param);
        param->cq_period_mode = params->rx_cq_moderation.cq_period_mode;
}

void mlx5e_build_tx_cq_param(struct mlx5e_priv *priv,
                             struct mlx5e_params *params,
                             struct mlx5e_cq_param *param)
{
        void *cqc = param->cqc;

        MLX5_SET(cqc, cqc, log_cq_size, params->log_sq_size);

        mlx5e_build_common_cq_param(priv, param);
        param->cq_period_mode = params->tx_cq_moderation.cq_period_mode;
}

void mlx5e_build_ico_cq_param(struct mlx5e_priv *priv,
                              u8 log_wq_size,
                              struct mlx5e_cq_param *param)
{
        void *cqc = param->cqc;

        MLX5_SET(cqc, cqc, log_cq_size, log_wq_size);

        mlx5e_build_common_cq_param(priv, param);

        param->cq_period_mode = DIM_CQ_PERIOD_MODE_START_FROM_EQE;
}

void mlx5e_build_icosq_param(struct mlx5e_priv *priv,
                             u8 log_wq_size,
                             struct mlx5e_sq_param *param)
{
        void *sqc = param->sqc;
        void *wq = MLX5_ADDR_OF(sqc, sqc, wq);

        mlx5e_build_sq_param_common(priv, param);

        MLX5_SET(wq, wq, log_wq_sz, log_wq_size);
        MLX5_SET(sqc, sqc, reg_umr, MLX5_CAP_ETH(priv->mdev, reg_umr_sq));
}

void mlx5e_build_xdpsq_param(struct mlx5e_priv *priv,
                             struct mlx5e_params *params,
                             struct mlx5e_sq_param *param)
{
        void *sqc = param->sqc;
        void *wq = MLX5_ADDR_OF(sqc, sqc, wq);

        mlx5e_build_sq_param_common(priv, param);
        MLX5_SET(wq, wq, log_wq_sz, params->log_sq_size);
        param->is_mpw = MLX5E_GET_PFLAG(params, MLX5E_PFLAG_XDP_TX_MPWQE);
}

static u8 mlx5e_build_icosq_log_wq_sz(struct mlx5e_params *params,
                                      struct mlx5e_rq_param *rqp)
{
        switch (params->rq_wq_type) {
        case MLX5_WQ_TYPE_LINKED_LIST_STRIDING_RQ:
                return order_base_2(MLX5E_UMR_WQEBBS) +
                        mlx5e_get_rq_log_wq_sz(rqp->rqc);
        default: /* MLX5_WQ_TYPE_CYCLIC */
                return MLX5E_PARAMS_MINIMUM_LOG_SQ_SIZE;
        }
}

static void mlx5e_build_channel_param(struct mlx5e_priv *priv,
                                      struct mlx5e_params *params,
                                      struct mlx5e_channel_param *cparam)
{
        u8 icosq_log_wq_sz;

        mlx5e_build_rq_param(priv, params, NULL, &cparam->rq);

        icosq_log_wq_sz = mlx5e_build_icosq_log_wq_sz(params, &cparam->rq);

        mlx5e_build_sq_param(priv, params, &cparam->sq);
        mlx5e_build_xdpsq_param(priv, params, &cparam->xdp_sq);
        mlx5e_build_icosq_param(priv, icosq_log_wq_sz, &cparam->icosq);
        mlx5e_build_rx_cq_param(priv, params, NULL, &cparam->rx_cq);
        mlx5e_build_tx_cq_param(priv, params, &cparam->tx_cq);
        mlx5e_build_ico_cq_param(priv, icosq_log_wq_sz, &cparam->icosq_cq);
}

int mlx5e_open_channels(struct mlx5e_priv *priv,
                        struct mlx5e_channels *chs)
{
        struct mlx5e_channel_param *cparam;
        int err = -ENOMEM;
        int i;

        chs->num = chs->params.num_channels;

        chs->c = kcalloc(chs->num, sizeof(struct mlx5e_channel *), GFP_KERNEL);
        cparam = kvzalloc(sizeof(struct mlx5e_channel_param), GFP_KERNEL);
        if (!chs->c || !cparam)
                goto err_free;

        mlx5e_build_channel_param(priv, &chs->params, cparam);
        for (i = 0; i < chs->num; i++) {
                struct xdp_umem *umem = NULL;

                if (chs->params.xdp_prog)
                        umem = mlx5e_xsk_get_umem(&chs->params, chs->params.xsk, i);

                err = mlx5e_open_channel(priv, i, &chs->params, cparam, umem, &chs->c[i]);
                if (err)
                        goto err_close_channels;
        }

        if (!IS_ERR_OR_NULL(priv->tx_reporter))
                devlink_health_reporter_state_update(priv->tx_reporter,
                                                     DEVLINK_HEALTH_REPORTER_STATE_HEALTHY);

        kvfree(cparam);
        return 0;

err_close_channels:
        for (i--; i >= 0; i--)
                mlx5e_close_channel(chs->c[i]);

err_free:
        kfree(chs->c);
        kvfree(cparam);
        chs->num = 0;
        return err;
}

static void mlx5e_activate_channels(struct mlx5e_channels *chs)
{
        int i;

        for (i = 0; i < chs->num; i++)
                mlx5e_activate_channel(chs->c[i]);
}

#define MLX5E_RQ_WQES_TIMEOUT 20000 /* msecs */

static int mlx5e_wait_channels_min_rx_wqes(struct mlx5e_channels *chs)
{
        int err = 0;
        int i;

        for (i = 0; i < chs->num; i++) {
                int timeout = err ? 0 : MLX5E_RQ_WQES_TIMEOUT;

                err |= mlx5e_wait_for_min_rx_wqes(&chs->c[i]->rq, timeout);

                /* Don't wait on the XSK RQ, because the newer xdpsock sample
                 * doesn't provide any Fill Ring entries at the setup stage.
                 */
        }

        return err ? -ETIMEDOUT : 0;
}

static void mlx5e_deactivate_channels(struct mlx5e_channels *chs)
{
        int i;

        for (i = 0; i < chs->num; i++)
                mlx5e_deactivate_channel(chs->c[i]);
}

void mlx5e_close_channels(struct mlx5e_channels *chs)
{
        int i;

        for (i = 0; i < chs->num; i++)
                mlx5e_close_channel(chs->c[i]);

        kfree(chs->c);
        chs->num = 0;
}

static int
mlx5e_create_rqt(struct mlx5e_priv *priv, int sz, struct mlx5e_rqt *rqt)
{
        struct mlx5_core_dev *mdev = priv->mdev;
        void *rqtc;
        int inlen;
        int err;
        u32 *in;
        int i;

        inlen = MLX5_ST_SZ_BYTES(create_rqt_in) + sizeof(u32) * sz;
        in = kvzalloc(inlen, GFP_KERNEL);
        if (!in)
                return -ENOMEM;

        rqtc = MLX5_ADDR_OF(create_rqt_in, in, rqt_context);

        MLX5_SET(rqtc, rqtc, rqt_actual_size, sz);
        MLX5_SET(rqtc, rqtc, rqt_max_size, sz);

        for (i = 0; i < sz; i++)
                MLX5_SET(rqtc, rqtc, rq_num[i], priv->drop_rq.rqn);

        err = mlx5_core_create_rqt(mdev, in, inlen, &rqt->rqtn);
        if (!err)
                rqt->enabled = true;

        kvfree(in);
        return err;
}

void mlx5e_destroy_rqt(struct mlx5e_priv *priv, struct mlx5e_rqt *rqt)
{
        rqt->enabled = false;
        mlx5_core_destroy_rqt(priv->mdev, rqt->rqtn);
}

int mlx5e_create_indirect_rqt(struct mlx5e_priv *priv)
{
        struct mlx5e_rqt *rqt = &priv->indir_rqt;
        int err;

        err = mlx5e_create_rqt(priv, MLX5E_INDIR_RQT_SIZE, rqt);
        if (err)
                mlx5_core_warn(priv->mdev, "create indirect rqts failed, %d\n", err);
        return err;
}

int mlx5e_create_direct_rqts(struct mlx5e_priv *priv, struct mlx5e_tir *tirs)
{
        int err;
        int ix;

        for (ix = 0; ix < priv->max_nch; ix++) {
                err = mlx5e_create_rqt(priv, 1 /*size */, &tirs[ix].rqt);
                if (unlikely(err))
                        goto err_destroy_rqts;
        }

        return 0;

err_destroy_rqts:
        mlx5_core_warn(priv->mdev, "create rqts failed, %d\n", err);
        for (ix--; ix >= 0; ix--)
                mlx5e_destroy_rqt(priv, &tirs[ix].rqt);

        return err;
}

void mlx5e_destroy_direct_rqts(struct mlx5e_priv *priv, struct mlx5e_tir *tirs)
{
        int i;

        for (i = 0; i < priv->max_nch; i++)
                mlx5e_destroy_rqt(priv, &tirs[i].rqt);
}

static int mlx5e_rx_hash_fn(int hfunc)
{
        return (hfunc == ETH_RSS_HASH_TOP) ?
               MLX5_RX_HASH_FN_TOEPLITZ :
               MLX5_RX_HASH_FN_INVERTED_XOR8;
}

int mlx5e_bits_invert(unsigned long a, int size)
{
        int inv = 0;
        int i;

        for (i = 0; i < size; i++)
                inv |= (test_bit(size - i - 1, &a) ? 1 : 0) << i;

        return inv;
}

static void mlx5e_fill_rqt_rqns(struct mlx5e_priv *priv, int sz,
                                struct mlx5e_redirect_rqt_param rrp, void *rqtc)
{
        int i;

        for (i = 0; i < sz; i++) {
                u32 rqn;

                if (rrp.is_rss) {
                        int ix = i;

                        if (rrp.rss.hfunc == ETH_RSS_HASH_XOR)
                                ix = mlx5e_bits_invert(i, ilog2(sz));

                        ix = priv->rss_params.indirection_rqt[ix];
                        rqn = rrp.rss.channels->c[ix]->rq.rqn;
                } else {
                        rqn = rrp.rqn;
                }
                MLX5_SET(rqtc, rqtc, rq_num[i], rqn);
        }
}

int mlx5e_redirect_rqt(struct mlx5e_priv *priv, u32 rqtn, int sz,
                       struct mlx5e_redirect_rqt_param rrp)
{
        struct mlx5_core_dev *mdev = priv->mdev;
        void *rqtc;
        int inlen;
        u32 *in;
        int err;

        inlen = MLX5_ST_SZ_BYTES(modify_rqt_in) + sizeof(u32) * sz;
        in = kvzalloc(inlen, GFP_KERNEL);
        if (!in)
                return -ENOMEM;

        rqtc = MLX5_ADDR_OF(modify_rqt_in, in, ctx);

        MLX5_SET(rqtc, rqtc, rqt_actual_size, sz);
        MLX5_SET(modify_rqt_in, in, bitmask.rqn_list, 1);
        mlx5e_fill_rqt_rqns(priv, sz, rrp, rqtc);
        err = mlx5_core_modify_rqt(mdev, rqtn, in, inlen);

        kvfree(in);
        return err;
}

static u32 mlx5e_get_direct_rqn(struct mlx5e_priv *priv, int ix,
                                struct mlx5e_redirect_rqt_param rrp)
{
        if (!rrp.is_rss)
                return rrp.rqn;

        if (ix >= rrp.rss.channels->num)
                return priv->drop_rq.rqn;

        return rrp.rss.channels->c[ix]->rq.rqn;
}

static void mlx5e_redirect_rqts(struct mlx5e_priv *priv,
                                struct mlx5e_redirect_rqt_param rrp)
{
        u32 rqtn;
        int ix;

        if (priv->indir_rqt.enabled) {
                /* RSS RQ table */
                rqtn = priv->indir_rqt.rqtn;
                mlx5e_redirect_rqt(priv, rqtn, MLX5E_INDIR_RQT_SIZE, rrp);
        }

        for (ix = 0; ix < priv->max_nch; ix++) {
                struct mlx5e_redirect_rqt_param direct_rrp = {
                        .is_rss = false,
                        {
                                .rqn    = mlx5e_get_direct_rqn(priv, ix, rrp)
                        },
                };

                /* Direct RQ Tables */
                if (!priv->direct_tir[ix].rqt.enabled)
                        continue;

                rqtn = priv->direct_tir[ix].rqt.rqtn;
                mlx5e_redirect_rqt(priv, rqtn, 1, direct_rrp);
        }
}

static void mlx5e_redirect_rqts_to_channels(struct mlx5e_priv *priv,
                                            struct mlx5e_channels *chs)
{
        struct mlx5e_redirect_rqt_param rrp = {
                .is_rss        = true,
                {
                        .rss = {
                                .channels  = chs,
                                .hfunc     = priv->rss_params.hfunc,
                        }
                },
        };

        mlx5e_redirect_rqts(priv, rrp);
}

static void mlx5e_redirect_rqts_to_drop(struct mlx5e_priv *priv)
{
        struct mlx5e_redirect_rqt_param drop_rrp = {
                .is_rss = false,
                {
                        .rqn = priv->drop_rq.rqn,
                },
        };

        mlx5e_redirect_rqts(priv, drop_rrp);
}

static const struct mlx5e_tirc_config tirc_default_config[MLX5E_NUM_INDIR_TIRS] = {
        [MLX5E_TT_IPV4_TCP] = { .l3_prot_type = MLX5_L3_PROT_TYPE_IPV4,
                                .l4_prot_type = MLX5_L4_PROT_TYPE_TCP,
                                .rx_hash_fields = MLX5_HASH_IP_L4PORTS,
        },
        [MLX5E_TT_IPV6_TCP] = { .l3_prot_type = MLX5_L3_PROT_TYPE_IPV6,
                                .l4_prot_type = MLX5_L4_PROT_TYPE_TCP,
                                .rx_hash_fields = MLX5_HASH_IP_L4PORTS,
        },
        [MLX5E_TT_IPV4_UDP] = { .l3_prot_type = MLX5_L3_PROT_TYPE_IPV4,
                                .l4_prot_type = MLX5_L4_PROT_TYPE_UDP,
                                .rx_hash_fields = MLX5_HASH_IP_L4PORTS,
        },
        [MLX5E_TT_IPV6_UDP] = { .l3_prot_type = MLX5_L3_PROT_TYPE_IPV6,
                                .l4_prot_type = MLX5_L4_PROT_TYPE_UDP,
                                .rx_hash_fields = MLX5_HASH_IP_L4PORTS,
        },
        [MLX5E_TT_IPV4_IPSEC_AH] = { .l3_prot_type = MLX5_L3_PROT_TYPE_IPV4,
                                     .l4_prot_type = 0,
                                     .rx_hash_fields = MLX5_HASH_IP_IPSEC_SPI,
        },
        [MLX5E_TT_IPV6_IPSEC_AH] = { .l3_prot_type = MLX5_L3_PROT_TYPE_IPV6,
                                     .l4_prot_type = 0,
                                     .rx_hash_fields = MLX5_HASH_IP_IPSEC_SPI,
        },
        [MLX5E_TT_IPV4_IPSEC_ESP] = { .l3_prot_type = MLX5_L3_PROT_TYPE_IPV4,
                                      .l4_prot_type = 0,
                                      .rx_hash_fields = MLX5_HASH_IP_IPSEC_SPI,
        },
        [MLX5E_TT_IPV6_IPSEC_ESP] = { .l3_prot_type = MLX5_L3_PROT_TYPE_IPV6,
                                      .l4_prot_type = 0,
                                      .rx_hash_fields = MLX5_HASH_IP_IPSEC_SPI,
        },
        [MLX5E_TT_IPV4] = { .l3_prot_type = MLX5_L3_PROT_TYPE_IPV4,
                            .l4_prot_type = 0,
                            .rx_hash_fields = MLX5_HASH_IP,
        },
        [MLX5E_TT_IPV6] = { .l3_prot_type = MLX5_L3_PROT_TYPE_IPV6,
                            .l4_prot_type = 0,
                            .rx_hash_fields = MLX5_HASH_IP,
        },
};

struct mlx5e_tirc_config mlx5e_tirc_get_default_config(enum mlx5e_traffic_types tt)
{
        return tirc_default_config[tt];
}

static void mlx5e_build_tir_ctx_lro(struct mlx5e_params *params, void *tirc)
{
        if (!params->lro_en)
                return;

#define ROUGH_MAX_L2_L3_HDR_SZ 256

        MLX5_SET(tirc, tirc, lro_enable_mask,
                 MLX5_TIRC_LRO_ENABLE_MASK_IPV4_LRO |
                 MLX5_TIRC_LRO_ENABLE_MASK_IPV6_LRO);
        MLX5_SET(tirc, tirc, lro_max_ip_payload_size,
                 (MLX5E_PARAMS_DEFAULT_LRO_WQE_SZ - ROUGH_MAX_L2_L3_HDR_SZ) >> 8);
        MLX5_SET(tirc, tirc, lro_timeout_period_usecs, params->lro_timeout);
}

void mlx5e_build_indir_tir_ctx_hash(struct mlx5e_rss_params *rss_params,
                                    const struct mlx5e_tirc_config *ttconfig,
                                    void *tirc, bool inner)
{
        void *hfso = inner ? MLX5_ADDR_OF(tirc, tirc, rx_hash_field_selector_inner) :
                             MLX5_ADDR_OF(tirc, tirc, rx_hash_field_selector_outer);

        MLX5_SET(tirc, tirc, rx_hash_fn, mlx5e_rx_hash_fn(rss_params->hfunc));
        if (rss_params->hfunc == ETH_RSS_HASH_TOP) {
                void *rss_key = MLX5_ADDR_OF(tirc, tirc,
                                             rx_hash_toeplitz_key);
                size_t len = MLX5_FLD_SZ_BYTES(tirc,
                                               rx_hash_toeplitz_key);

                MLX5_SET(tirc, tirc, rx_hash_symmetric, 1);
                memcpy(rss_key, rss_params->toeplitz_hash_key, len);
        }
        MLX5_SET(rx_hash_field_select, hfso, l3_prot_type,
                 ttconfig->l3_prot_type);
        MLX5_SET(rx_hash_field_select, hfso, l4_prot_type,
                 ttconfig->l4_prot_type);
        MLX5_SET(rx_hash_field_select, hfso, selected_fields,
                 ttconfig->rx_hash_fields);
}

static void mlx5e_update_rx_hash_fields(struct mlx5e_tirc_config *ttconfig,
                                        enum mlx5e_traffic_types tt,
                                        u32 rx_hash_fields)
{
        *ttconfig                = tirc_default_config[tt];
        ttconfig->rx_hash_fields = rx_hash_fields;
}

void mlx5e_modify_tirs_hash(struct mlx5e_priv *priv, void *in, int inlen)
{
        void *tirc = MLX5_ADDR_OF(modify_tir_in, in, ctx);
        struct mlx5e_rss_params *rss = &priv->rss_params;
        struct mlx5_core_dev *mdev = priv->mdev;
        int ctxlen = MLX5_ST_SZ_BYTES(tirc);
        struct mlx5e_tirc_config ttconfig;
        int tt;

        MLX5_SET(modify_tir_in, in, bitmask.hash, 1);

        for (tt = 0; tt < MLX5E_NUM_INDIR_TIRS; tt++) {
                memset(tirc, 0, ctxlen);
                mlx5e_update_rx_hash_fields(&ttconfig, tt,
                                            rss->rx_hash_fields[tt]);
                mlx5e_build_indir_tir_ctx_hash(rss, &ttconfig, tirc, false);
                mlx5_core_modify_tir(mdev, priv->indir_tir[tt].tirn, in, inlen);
        }

        if (!mlx5e_tunnel_inner_ft_supported(priv->mdev))
                return;

        for (tt = 0; tt < MLX5E_NUM_INDIR_TIRS; tt++) {
                memset(tirc, 0, ctxlen);
                mlx5e_update_rx_hash_fields(&ttconfig, tt,
                                            rss->rx_hash_fields[tt]);
                mlx5e_build_indir_tir_ctx_hash(rss, &ttconfig, tirc, true);
                mlx5_core_modify_tir(mdev, priv->inner_indir_tir[tt].tirn, in,
                                     inlen);
        }
}

static int mlx5e_modify_tirs_lro(struct mlx5e_priv *priv)
{
        struct mlx5_core_dev *mdev = priv->mdev;

        void *in;
        void *tirc;
        int inlen;
        int err;
        int tt;
        int ix;

        inlen = MLX5_ST_SZ_BYTES(modify_tir_in);
        in = kvzalloc(inlen, GFP_KERNEL);
        if (!in)
                return -ENOMEM;

        MLX5_SET(modify_tir_in, in, bitmask.lro, 1);
        tirc = MLX5_ADDR_OF(modify_tir_in, in, ctx);

        mlx5e_build_tir_ctx_lro(&priv->channels.params, tirc);

        for (tt = 0; tt < MLX5E_NUM_INDIR_TIRS; tt++) {
                err = mlx5_core_modify_tir(mdev, priv->indir_tir[tt].tirn, in,
                                           inlen);
                if (err)
                        goto free_in;
        }

        for (ix = 0; ix < priv->max_nch; ix++) {
                err = mlx5_core_modify_tir(mdev, priv->direct_tir[ix].tirn,
                                           in, inlen);
                if (err)
                        goto free_in;
        }

free_in:
        kvfree(in);

        return err;
}

static int mlx5e_set_mtu(struct mlx5_core_dev *mdev,
                         struct mlx5e_params *params, u16 mtu)
{
        u16 hw_mtu = MLX5E_SW2HW_MTU(params, mtu);
        int err;

        err = mlx5_set_port_mtu(mdev, hw_mtu, 1);
        if (err)
                return err;

        /* Update vport context MTU */
        mlx5_modify_nic_vport_mtu(mdev, hw_mtu);
        return 0;
}

static void mlx5e_query_mtu(struct mlx5_core_dev *mdev,
                            struct mlx5e_params *params, u16 *mtu)
{
        u16 hw_mtu = 0;
        int err;

        err = mlx5_query_nic_vport_mtu(mdev, &hw_mtu);
        if (err || !hw_mtu) /* fallback to port oper mtu */
                mlx5_query_port_oper_mtu(mdev, &hw_mtu, 1);

        *mtu = MLX5E_HW2SW_MTU(params, hw_mtu);
}

int mlx5e_set_dev_port_mtu(struct mlx5e_priv *priv)
{
        struct mlx5e_params *params = &priv->channels.params;
        struct net_device *netdev = priv->netdev;
        struct mlx5_core_dev *mdev = priv->mdev;
        u16 mtu;
        int err;

        err = mlx5e_set_mtu(mdev, params, params->sw_mtu);
        if (err)
                return err;

        mlx5e_query_mtu(mdev, params, &mtu);
        if (mtu != params->sw_mtu)
                netdev_warn(netdev, "%s: VPort MTU %d is different than netdev mtu %d\n",
                            __func__, mtu, params->sw_mtu);

        params->sw_mtu = mtu;
        return 0;
}

void mlx5e_set_netdev_mtu_boundaries(struct mlx5e_priv *priv)
{
        struct mlx5e_params *params = &priv->channels.params;
        struct net_device *netdev   = priv->netdev;
        struct mlx5_core_dev *mdev  = priv->mdev;
        u16 max_mtu;

        /* MTU range: 68 - hw-specific max */
        netdev->min_mtu = ETH_MIN_MTU;

        mlx5_query_port_max_mtu(mdev, &max_mtu, 1);
        netdev->max_mtu = min_t(unsigned int, MLX5E_HW2SW_MTU(params, max_mtu),
                                ETH_MAX_MTU);
}

static void mlx5e_netdev_set_tcs(struct net_device *netdev)
{
        struct mlx5e_priv *priv = netdev_priv(netdev);
        int nch = priv->channels.params.num_channels;
        int ntc = priv->channels.params.num_tc;
        int tc;

        netdev_reset_tc(netdev);

        if (ntc == 1)
                return;

        netdev_set_num_tc(netdev, ntc);

        /* Map netdev TCs to offset 0
         * We have our own UP to TXQ mapping for QoS
         */
        for (tc = 0; tc < ntc; tc++)
                netdev_set_tc_queue(netdev, tc, nch, 0);
}

static void mlx5e_build_tc2txq_maps(struct mlx5e_priv *priv)
{
        int i, tc;

        for (i = 0; i < priv->max_nch; i++)
                for (tc = 0; tc < priv->profile->max_tc; tc++)
                        priv->channel_tc2txq[i][tc] = i + tc * priv->max_nch;
}

static void mlx5e_build_tx2sq_maps(struct mlx5e_priv *priv)
{
        struct mlx5e_channel *c;
        struct mlx5e_txqsq *sq;
        int i, tc;

        for (i = 0; i < priv->channels.num; i++) {
                c = priv->channels.c[i];
                for (tc = 0; tc < c->num_tc; tc++) {
                        sq = &c->sq[tc];
                        priv->txq2sq[sq->txq_ix] = sq;
                }
        }
}

void mlx5e_activate_priv_channels(struct mlx5e_priv *priv)
{
        int num_txqs = priv->channels.num * priv->channels.params.num_tc;
        int num_rxqs = priv->channels.num * priv->profile->rq_groups;
        struct net_device *netdev = priv->netdev;

        mlx5e_netdev_set_tcs(netdev);
        netif_set_real_num_tx_queues(netdev, num_txqs);
        netif_set_real_num_rx_queues(netdev, num_rxqs);

        mlx5e_build_tx2sq_maps(priv);
        mlx5e_activate_channels(&priv->channels);
        mlx5e_xdp_tx_enable(priv);
        netif_tx_start_all_queues(priv->netdev);

        if (mlx5e_is_vport_rep(priv))
                mlx5e_add_sqs_fwd_rules(priv);

        mlx5e_wait_channels_min_rx_wqes(&priv->channels);
        mlx5e_redirect_rqts_to_channels(priv, &priv->channels);

        mlx5e_xsk_redirect_rqts_to_channels(priv, &priv->channels);
}

void mlx5e_deactivate_priv_channels(struct mlx5e_priv *priv)
{
        mlx5e_xsk_redirect_rqts_to_drop(priv, &priv->channels);

        mlx5e_redirect_rqts_to_drop(priv);

        if (mlx5e_is_vport_rep(priv))
                mlx5e_remove_sqs_fwd_rules(priv);

        /* FIXME: This is a W/A only for tx timeout watch dog false alarm when
         * polling for inactive tx queues.
         */
        netif_tx_stop_all_queues(priv->netdev);
        netif_tx_disable(priv->netdev);
        mlx5e_xdp_tx_disable(priv);
        mlx5e_deactivate_channels(&priv->channels);
}

static void mlx5e_switch_priv_channels(struct mlx5e_priv *priv,
                                       struct mlx5e_channels *new_chs,
                                       mlx5e_fp_hw_modify hw_modify)
{
        struct net_device *netdev = priv->netdev;
        int new_num_txqs;
        int carrier_ok;

        new_num_txqs = new_chs->num * new_chs->params.num_tc;

        carrier_ok = netif_carrier_ok(netdev);
        netif_carrier_off(netdev);

        if (new_num_txqs < netdev->real_num_tx_queues)
                netif_set_real_num_tx_queues(netdev, new_num_txqs);

        mlx5e_deactivate_priv_channels(priv);
        mlx5e_close_channels(&priv->channels);

        priv->channels = *new_chs;

        /* New channels are ready to roll, modify HW settings if needed */
        if (hw_modify)
                hw_modify(priv);

        priv->profile->update_rx(priv);
        mlx5e_activate_priv_channels(priv);

        /* return carrier back if needed */
        if (carrier_ok)
                netif_carrier_on(netdev);
}

int mlx5e_safe_switch_channels(struct mlx5e_priv *priv,
                               struct mlx5e_channels *new_chs,
                               mlx5e_fp_hw_modify hw_modify)
{
        int err;

        err = mlx5e_open_channels(priv, new_chs);
        if (err)
                return err;

        mlx5e_switch_priv_channels(priv, new_chs, hw_modify);
        return 0;
}

int mlx5e_safe_reopen_channels(struct mlx5e_priv *priv)
{
        struct mlx5e_channels new_channels = {};

        new_channels.params = priv->channels.params;
        return mlx5e_safe_switch_channels(priv, &new_channels, NULL);
}

void mlx5e_timestamp_init(struct mlx5e_priv *priv)
{
        priv->tstamp.tx_type   = HWTSTAMP_TX_OFF;
        priv->tstamp.rx_filter = HWTSTAMP_FILTER_NONE;
}

int mlx5e_open_locked(struct net_device *netdev)
{
        struct mlx5e_priv *priv = netdev_priv(netdev);
        bool is_xdp = priv->channels.params.xdp_prog;
        int err;

        set_bit(MLX5E_STATE_OPENED, &priv->state);
        if (is_xdp)
                mlx5e_xdp_set_open(priv);

        err = mlx5e_open_channels(priv, &priv->channels);
        if (err)
                goto err_clear_state_opened_flag;

        priv->profile->update_rx(priv);
        mlx5e_activate_priv_channels(priv);
        if (priv->profile->update_carrier)
                priv->profile->update_carrier(priv);