// SPDX-License-Identifier: (GPL-2.0-only OR BSD-3-Clause)
/* QLogic qede NIC Driver
 * Copyright (c) 2015-2017  QLogic Corporation
 * Copyright (c) 2019-2020 Marvell International Ltd.
 */

#include <linux/array_size.h>
#include <linux/netdevice.h>
#include <linux/etherdevice.h>
#include <linux/skbuff.h>
#include <linux/bpf_trace.h>
#include <net/udp_tunnel.h>
#include <linux/ip.h>
#include <net/gro.h>
#include <net/ipv6.h>
#include <net/tcp.h>
#include <linux/if_ether.h>
#include <linux/if_vlan.h>
#include <net/ip6_checksum.h>
#include "qede_ptp.h"

#include <linux/qed/qed_if.h>
#include "qede.h"
/*********************************
 * Content also used by slowpath *
 *********************************/

int qede_alloc_rx_buffer(struct qede_rx_queue *rxq, bool allow_lazy)
{
        struct sw_rx_data *sw_rx_data;
        struct eth_rx_bd *rx_bd;
        dma_addr_t mapping;
        struct page *data;

        /* In case lazy-allocation is allowed, postpone allocation until the
         * end of the NAPI run. We'd still need to make sure the Rx ring has
         * sufficient buffers to guarantee an additional Rx interrupt.
         */
        if (allow_lazy && likely(rxq->filled_buffers > 12)) {
                rxq->filled_buffers--;
                return 0;
        }

        data = alloc_pages(GFP_ATOMIC, 0);
        if (unlikely(!data))
                return -ENOMEM;

        /* Map the entire page as it would be used
         * for multiple RX buffer segment size mapping.
         */
        mapping = dma_map_page(rxq->dev, data, 0,
                               PAGE_SIZE, rxq->data_direction);
        if (unlikely(dma_mapping_error(rxq->dev, mapping))) {
                __free_page(data);
                return -ENOMEM;
        }

        sw_rx_data = &rxq->sw_rx_ring[rxq->sw_rx_prod & NUM_RX_BDS_MAX];
        sw_rx_data->page_offset = 0;
        sw_rx_data->data = data;
        sw_rx_data->mapping = mapping;

        /* Advance PROD and get BD pointer */
        rx_bd = (struct eth_rx_bd *)qed_chain_produce(&rxq->rx_bd_ring);
        WARN_ON(!rx_bd);
        rx_bd->addr.hi = cpu_to_le32(upper_32_bits(mapping));
        rx_bd->addr.lo = cpu_to_le32(lower_32_bits(mapping) +
                                     rxq->rx_headroom);

        rxq->sw_rx_prod++;
        rxq->filled_buffers++;

        return 0;
}

/* Unmap the data and free skb */
int qede_free_tx_pkt(struct qede_dev *edev, struct qede_tx_queue *txq, int *len)
{
        u16 idx = txq->sw_tx_cons;
        struct sk_buff *skb = txq->sw_tx_ring.skbs[idx].skb;
        struct eth_tx_1st_bd *first_bd;
        struct eth_tx_bd *tx_data_bd;
        int bds_consumed = 0;
        int nbds;
        bool data_split = txq->sw_tx_ring.skbs[idx].flags & QEDE_TSO_SPLIT_BD;
        int i, split_bd_len = 0;

        if (unlikely(!skb)) {
                DP_ERR(edev,
                       "skb is null for txq idx=%d txq->sw_tx_cons=%d txq->sw_tx_prod=%d\n",
                       idx, txq->sw_tx_cons, txq->sw_tx_prod);
                return -1;
        }

        *len = skb->len;

        first_bd = (struct eth_tx_1st_bd *)qed_chain_consume(&txq->tx_pbl);

        bds_consumed++;

        nbds = first_bd->data.nbds;

        if (data_split) {
                struct eth_tx_bd *split = (struct eth_tx_bd *)
                        qed_chain_consume(&txq->tx_pbl);
                split_bd_len = BD_UNMAP_LEN(split);
                bds_consumed++;
        }
        dma_unmap_single(&edev->pdev->dev, BD_UNMAP_ADDR(first_bd),
                         BD_UNMAP_LEN(first_bd) + split_bd_len, DMA_TO_DEVICE);

        /* Unmap the data of the skb frags */
        for (i = 0; i < skb_shinfo(skb)->nr_frags; i++, bds_consumed++) {
                tx_data_bd = (struct eth_tx_bd *)
                        qed_chain_consume(&txq->tx_pbl);
                dma_unmap_page(&edev->pdev->dev, BD_UNMAP_ADDR(tx_data_bd),
                               BD_UNMAP_LEN(tx_data_bd), DMA_TO_DEVICE);
        }

        while (bds_consumed++ < nbds)
                qed_chain_consume(&txq->tx_pbl);

        /* Free skb */
        dev_kfree_skb_any(skb);
        txq->sw_tx_ring.skbs[idx].skb = NULL;
        txq->sw_tx_ring.skbs[idx].flags = 0;

        return 0;
}

/* Unmap the data and free skb when mapping failed during start_xmit */
static void qede_free_failed_tx_pkt(struct qede_tx_queue *txq,
                                    struct eth_tx_1st_bd *first_bd,
                                    int nbd, bool data_split)
{
        u16 idx = txq->sw_tx_prod;
        struct sk_buff *skb = txq->sw_tx_ring.skbs[idx].skb;
        struct eth_tx_bd *tx_data_bd;
        int i, split_bd_len = 0;

        /* Return prod to its position before this skb was handled */
        qed_chain_set_prod(&txq->tx_pbl,
                           le16_to_cpu(txq->tx_db.data.bd_prod), first_bd);

        first_bd = (struct eth_tx_1st_bd *)qed_chain_produce(&txq->tx_pbl);

        if (data_split) {
                struct eth_tx_bd *split = (struct eth_tx_bd *)
                                          qed_chain_produce(&txq->tx_pbl);
                split_bd_len = BD_UNMAP_LEN(split);
                nbd--;
        }

        dma_unmap_single(txq->dev, BD_UNMAP_ADDR(first_bd),
                         BD_UNMAP_LEN(first_bd) + split_bd_len, DMA_TO_DEVICE);

        /* Unmap the data of the skb frags */
        for (i = 0; i < nbd; i++) {
                tx_data_bd = (struct eth_tx_bd *)
                        qed_chain_produce(&txq->tx_pbl);
                if (tx_data_bd->nbytes)
                        dma_unmap_page(txq->dev,
                                       BD_UNMAP_ADDR(tx_data_bd),
                                       BD_UNMAP_LEN(tx_data_bd), DMA_TO_DEVICE);
        }

        /* Return again prod to its position before this skb was handled */
        qed_chain_set_prod(&txq->tx_pbl,
                           le16_to_cpu(txq->tx_db.data.bd_prod), first_bd);

        /* Free skb */
        dev_kfree_skb_any(skb);
        txq->sw_tx_ring.skbs[idx].skb = NULL;
        txq->sw_tx_ring.skbs[idx].flags = 0;
}

static u32 qede_xmit_type(struct sk_buff *skb, int *ipv6_ext)
{
        u32 rc = XMIT_L4_CSUM;
        __be16 l3_proto;

        if (skb->ip_summed != CHECKSUM_PARTIAL)
                return XMIT_PLAIN;

        l3_proto = vlan_get_protocol(skb);
        if (l3_proto == htons(ETH_P_IPV6) &&
            (ipv6_hdr(skb)->nexthdr == NEXTHDR_IPV6))
                *ipv6_ext = 1;

        if (skb->encapsulation) {
                rc |= XMIT_ENC;
                if (skb_is_gso(skb)) {
                        unsigned short gso_type = skb_shinfo(skb)->gso_type;

                        if ((gso_type & SKB_GSO_UDP_TUNNEL_CSUM) ||
                            (gso_type & SKB_GSO_GRE_CSUM))
                                rc |= XMIT_ENC_GSO_L4_CSUM;

                        rc |= XMIT_LSO;
                        return rc;
                }
        }

        if (skb_is_gso(skb))
                rc |= XMIT_LSO;

        return rc;
}

static void qede_set_params_for_ipv6_ext(struct sk_buff *skb,
                                         struct eth_tx_2nd_bd *second_bd,
                                         struct eth_tx_3rd_bd *third_bd)
{
        u8 l4_proto;
        u16 bd2_bits1 = 0, bd2_bits2 = 0;

        bd2_bits1 |= (1 << ETH_TX_DATA_2ND_BD_IPV6_EXT_SHIFT);

        bd2_bits2 |= ((skb_transport_offset(skb) >> 1) &
                     ETH_TX_DATA_2ND_BD_L4_HDR_START_OFFSET_W_MASK)
                    << ETH_TX_DATA_2ND_BD_L4_HDR_START_OFFSET_W_SHIFT;

        bd2_bits1 |= (ETH_L4_PSEUDO_CSUM_CORRECT_LENGTH <<
                      ETH_TX_DATA_2ND_BD_L4_PSEUDO_CSUM_MODE_SHIFT);

        if (vlan_get_protocol(skb) == htons(ETH_P_IPV6))
                l4_proto = ipv6_hdr(skb)->nexthdr;
        else
                l4_proto = ip_hdr(skb)->protocol;

        if (l4_proto == IPPROTO_UDP)
                bd2_bits1 |= 1 << ETH_TX_DATA_2ND_BD_L4_UDP_SHIFT;

        if (third_bd)
                third_bd->data.bitfields |=
                        cpu_to_le16(((tcp_hdrlen(skb) / 4) &
                                ETH_TX_DATA_3RD_BD_TCP_HDR_LEN_DW_MASK) <<
                                ETH_TX_DATA_3RD_BD_TCP_HDR_LEN_DW_SHIFT);

        second_bd->data.bitfields1 = cpu_to_le16(bd2_bits1);
        second_bd->data.bitfields2 = cpu_to_le16(bd2_bits2);
}

static int map_frag_to_bd(struct qede_tx_queue *txq,
                          skb_frag_t *frag, struct eth_tx_bd *bd)
{
        dma_addr_t mapping;

        /* Map skb non-linear frag data for DMA */
        mapping = skb_frag_dma_map(txq->dev, frag, 0,
                                   skb_frag_size(frag), DMA_TO_DEVICE);
        if (unlikely(dma_mapping_error(txq->dev, mapping)))
                return -ENOMEM;

        /* Setup the data pointer of the frag data */
        BD_SET_UNMAP_ADDR_LEN(bd, mapping, skb_frag_size(frag));

        return 0;
}

static u16 qede_get_skb_hlen(struct sk_buff *skb, bool is_encap_pkt)
{
        if (is_encap_pkt)
                return skb_inner_tcp_all_headers(skb);

        return skb_tcp_all_headers(skb);
}

/* +2 for 1st BD for headers and 2nd BD for headlen (if required) */
#if ((MAX_SKB_FRAGS + 2) > ETH_TX_MAX_BDS_PER_NON_LSO_PACKET)
static bool qede_pkt_req_lin(struct sk_buff *skb, u8 xmit_type)
{
        int allowed_frags = ETH_TX_MAX_BDS_PER_NON_LSO_PACKET - 1;

        if (xmit_type & XMIT_LSO) {
                int hlen;

                hlen = qede_get_skb_hlen(skb, xmit_type & XMIT_ENC);

                /* linear payload would require its own BD */
                if (skb_headlen(skb) > hlen)
                        allowed_frags--;
        }

        return (skb_shinfo(skb)->nr_frags > allowed_frags);
}
#endif

static inline void qede_update_tx_producer(struct qede_tx_queue *txq)
{
        /* wmb makes sure that the BDs data is updated before updating the
         * producer, otherwise FW may read old data from the BDs.
         */
        wmb();
        barrier();
        writel(txq->tx_db.raw, txq->doorbell_addr);

        /* Fence required to flush the write combined buffer, since another
         * CPU may write to the same doorbell address and data may be lost
         * due to relaxed order nature of write combined bar.
         */
        wmb();
}

static int qede_xdp_xmit(struct qede_tx_queue *txq, dma_addr_t dma, u16 pad,
                         u16 len, struct page *page, struct xdp_frame *xdpf)
{
        struct eth_tx_1st_bd *bd;
        struct sw_tx_xdp *xdp;
        u16 val;

        if (unlikely(qed_chain_get_elem_used(&txq->tx_pbl) >=
                     txq->num_tx_buffers)) {
                txq->stopped_cnt++;
                return -ENOMEM;
        }

        bd = qed_chain_produce(&txq->tx_pbl);
        bd->data.nbds = 1;
        bd->data.bd_flags.bitfields = BIT(ETH_TX_1ST_BD_FLAGS_START_BD_SHIFT);

        val = (len & ETH_TX_DATA_1ST_BD_PKT_LEN_MASK) <<
               ETH_TX_DATA_1ST_BD_PKT_LEN_SHIFT;

        bd->data.bitfields = cpu_to_le16(val);

        /* We can safely ignore the offset, as it's 0 for XDP */
        BD_SET_UNMAP_ADDR_LEN(bd, dma + pad, len);

        xdp = txq->sw_tx_ring.xdp + txq->sw_tx_prod;
        xdp->mapping = dma;
        xdp->page = page;
        xdp->xdpf = xdpf;

        txq->sw_tx_prod = (txq->sw_tx_prod + 1) % txq->num_tx_buffers;

        return 0;
}

int qede_xdp_transmit(struct net_device *dev, int n_frames,
                      struct xdp_frame **frames, u32 flags)
{
        struct qede_dev *edev = netdev_priv(dev);
        struct device *dmadev = &edev->pdev->dev;
        struct qede_tx_queue *xdp_tx;
        struct xdp_frame *xdpf;
        dma_addr_t mapping;
        int i, nxmit = 0;
        u16 xdp_prod;

        if (unlikely(flags & ~XDP_XMIT_FLAGS_MASK))
                return -EINVAL;

        if (unlikely(!netif_running(dev)))
                return -ENETDOWN;

        i = smp_processor_id() % edev->total_xdp_queues;
        xdp_tx = edev->fp_array[i].xdp_tx;

        spin_lock(&xdp_tx->xdp_tx_lock);

        for (i = 0; i < n_frames; i++) {
                xdpf = frames[i];

                mapping = dma_map_single(dmadev, xdpf->data, xdpf->len,
                                         DMA_TO_DEVICE);
                if (unlikely(dma_mapping_error(dmadev, mapping)))
                        break;

                if (unlikely(qede_xdp_xmit(xdp_tx, mapping, 0, xdpf->len,
                                           NULL, xdpf)))
                        break;
                nxmit++;
        }

        if (flags & XDP_XMIT_FLUSH) {
                xdp_prod = qed_chain_get_prod_idx(&xdp_tx->tx_pbl);

                xdp_tx->tx_db.data.bd_prod = cpu_to_le16(xdp_prod);
                qede_update_tx_producer(xdp_tx);
        }

        spin_unlock(&xdp_tx->xdp_tx_lock);

        return nxmit;
}

int qede_txq_has_work(struct qede_tx_queue *txq)
{
        u16 hw_bd_cons;

        /* Tell compiler that consumer and producer can change */
        barrier();
        hw_bd_cons = le16_to_cpu(*txq->hw_cons_ptr);
        if (qed_chain_get_cons_idx(&txq->tx_pbl) == hw_bd_cons + 1)
                return 0;

        return hw_bd_cons != qed_chain_get_cons_idx(&txq->tx_pbl);
}

static void qede_xdp_tx_int(struct qede_dev *edev, struct qede_tx_queue *txq)
{
        struct sw_tx_xdp *xdp_info, *xdp_arr = txq->sw_tx_ring.xdp;
        struct device *dev = &edev->pdev->dev;
        struct xdp_frame *xdpf;
        u16 hw_bd_cons;

        hw_bd_cons = le16_to_cpu(*txq->hw_cons_ptr);
        barrier();

        while (hw_bd_cons != qed_chain_get_cons_idx(&txq->tx_pbl)) {
                xdp_info = xdp_arr + txq->sw_tx_cons;
                xdpf = xdp_info->xdpf;

                if (xdpf) {
                        dma_unmap_single(dev, xdp_info->mapping, xdpf->len,
                                         DMA_TO_DEVICE);
                        xdp_return_frame(xdpf);

                        xdp_info->xdpf = NULL;
                } else {
                        dma_unmap_page(dev, xdp_info->mapping, PAGE_SIZE,
                                       DMA_BIDIRECTIONAL);
                        __free_page(xdp_info->page);
                }

                qed_chain_consume(&txq->tx_pbl);
                txq->sw_tx_cons = (txq->sw_tx_cons + 1) % txq->num_tx_buffers;
                txq->xmit_pkts++;
        }
}

static int qede_tx_int(struct qede_dev *edev, struct qede_tx_queue *txq)
{
        unsigned int pkts_compl = 0, bytes_compl = 0;
        struct netdev_queue *netdev_txq;
        u16 hw_bd_cons;
        int rc;

        netdev_txq = netdev_get_tx_queue(edev->ndev, txq->ndev_txq_id);

        hw_bd_cons = le16_to_cpu(*txq->hw_cons_ptr);
        barrier();

        while (hw_bd_cons != qed_chain_get_cons_idx(&txq->tx_pbl)) {
                int len = 0;

                rc = qede_free_tx_pkt(edev, txq, &len);
                if (rc) {
                        DP_NOTICE(edev, "hw_bd_cons = %d, chain_cons=%d\n",
                                  hw_bd_cons,
                                  qed_chain_get_cons_idx(&txq->tx_pbl));
                        break;
                }

                bytes_compl += len;
                pkts_compl++;
                txq->sw_tx_cons = (txq->sw_tx_cons + 1) % txq->num_tx_buffers;
                txq->xmit_pkts++;
        }

        netdev_tx_completed_queue(netdev_txq, pkts_compl, bytes_compl);

        /* Need to make the tx_bd_cons update visible to start_xmit()
         * before checking for netif_tx_queue_stopped().  Without the
         * memory barrier, there is a small possibility that
         * start_xmit() will miss it and cause the queue to be stopped
         * forever.
         * On the other hand we need an rmb() here to ensure the proper
         * ordering of bit testing in the following
         * netif_tx_queue_stopped(txq) call.
         */
        smp_mb();

        if (unlikely(netif_tx_queue_stopped(netdev_txq))) {
                /* Taking tx_lock is needed to prevent reenabling the queue
                 * while it's empty. This could have happen if rx_action() gets
                 * suspended in qede_tx_int() after the condition before
                 * netif_tx_wake_queue(), while tx_action (qede_start_xmit()):
                 *
                 * stops the queue->sees fresh tx_bd_cons->releases the queue->
                 * sends some packets consuming the whole queue again->
                 * stops the queue
                 */

                __netif_tx_lock(netdev_txq, smp_processor_id());

                if ((netif_tx_queue_stopped(netdev_txq)) &&
                    (edev->state == QEDE_STATE_OPEN) &&
                    (qed_chain_get_elem_left(&txq->tx_pbl)
                      >= (MAX_SKB_FRAGS + 1))) {
                        netif_tx_wake_queue(netdev_txq);
                        DP_VERBOSE(edev, NETIF_MSG_TX_DONE,
                                   "Wake queue was called\n");
                }

                __netif_tx_unlock(netdev_txq);
        }

        return 0;
}

bool qede_has_rx_work(struct qede_rx_queue *rxq)
{
        u16 hw_comp_cons, sw_comp_cons;

        /* Tell compiler that status block fields can change */
        barrier();

        hw_comp_cons = le16_to_cpu(*rxq->hw_cons_ptr);
        sw_comp_cons = qed_chain_get_cons_idx(&rxq->rx_comp_ring);

        return hw_comp_cons != sw_comp_cons;
}

static inline void qede_rx_bd_ring_consume(struct qede_rx_queue *rxq)
{
        qed_chain_consume(&rxq->rx_bd_ring);
        rxq->sw_rx_cons++;
}

/* This function reuses the buffer(from an offset) from
 * consumer index to producer index in the bd ring
 */
static inline void qede_reuse_page(struct qede_rx_queue *rxq,
                                   struct sw_rx_data *curr_cons)
{
        struct eth_rx_bd *rx_bd_prod = qed_chain_produce(&rxq->rx_bd_ring);
        struct sw_rx_data *curr_prod;
        dma_addr_t new_mapping;

        curr_prod = &rxq->sw_rx_ring[rxq->sw_rx_prod & NUM_RX_BDS_MAX];
        *curr_prod = *curr_cons;

        new_mapping = curr_prod->mapping + curr_prod->page_offset;

        rx_bd_prod->addr.hi = cpu_to_le32(upper_32_bits(new_mapping));
        rx_bd_prod->addr.lo = cpu_to_le32(lower_32_bits(new_mapping) +
                                          rxq->rx_headroom);

        rxq->sw_rx_prod++;
        curr_cons->data = NULL;
}

/* In case of allocation failures reuse buffers
 * from consumer index to produce buffers for firmware
 */
void qede_recycle_rx_bd_ring(struct qede_rx_queue *rxq, u8 count)
{
        struct sw_rx_data *curr_cons;

        for (; count > 0; count--) {
                curr_cons = &rxq->sw_rx_ring[rxq->sw_rx_cons & NUM_RX_BDS_MAX];
                qede_reuse_page(rxq, curr_cons);
                qede_rx_bd_ring_consume(rxq);
        }
}

static inline int qede_realloc_rx_buffer(struct qede_rx_queue *rxq,
                                         struct sw_rx_data *curr_cons)
{
        /* Move to the next segment in the page */
        curr_cons->page_offset += rxq->rx_buf_seg_size;

        if (curr_cons->page_offset == PAGE_SIZE) {
                if (unlikely(qede_alloc_rx_buffer(rxq, true))) {
                        /* Since we failed to allocate new buffer
                         * current buffer can be used again.
                         */
                        curr_cons->page_offset -= rxq->rx_buf_seg_size;

                        return -ENOMEM;
                }

                dma_unmap_page(rxq->dev, curr_cons->mapping,
                               PAGE_SIZE, rxq->data_direction);
        } else {
                /* Increment refcount of the page as we don't want
                 * network stack to take the ownership of the page
                 * which can be recycled multiple times by the driver.
                 */
                page_ref_inc(curr_cons->data);
                qede_reuse_page(rxq, curr_cons);
        }

        return 0;
}

void qede_update_rx_prod(struct qede_dev *edev, struct qede_rx_queue *rxq)
{
        u16 bd_prod = qed_chain_get_prod_idx(&rxq->rx_bd_ring);
        u16 cqe_prod = qed_chain_get_prod_idx(&rxq->rx_comp_ring);
        struct eth_rx_prod_data rx_prods = {0};

        /* Update producers */
        rx_prods.bd_prod = cpu_to_le16(bd_prod);
        rx_prods.cqe_prod = cpu_to_le16(cqe_prod);

        /* Make sure that the BD and SGE data is updated before updating the
         * producers since FW might read the BD/SGE right after the producer
         * is updated.
         */
        wmb();

        internal_ram_wr(rxq->hw_rxq_prod_addr, sizeof(rx_prods),
                        (u32 *)&rx_prods);
}

static void qede_get_rxhash(struct sk_buff *skb, u8 bitfields, __le32 rss_hash)
{
        enum pkt_hash_types hash_type = PKT_HASH_TYPE_NONE;
        enum rss_hash_type htype;
        u32 hash = 0;

        htype = GET_FIELD(bitfields, ETH_FAST_PATH_RX_REG_CQE_RSS_HASH_TYPE);
        if (htype) {
                hash_type = ((htype == RSS_HASH_TYPE_IPV4) ||
                             (htype == RSS_HASH_TYPE_IPV6)) ?
                            PKT_HASH_TYPE_L3 : PKT_HASH_TYPE_L4;
                hash = le32_to_cpu(rss_hash);
        }
        skb_set_hash(skb, hash, hash_type);
}

static void qede_set_skb_csum(struct sk_buff *skb, u8 csum_flag)
{
        skb_checksum_none_assert(skb);

        if (csum_flag & QEDE_CSUM_UNNECESSARY)
                skb->ip_summed = CHECKSUM_UNNECESSARY;

        if (csum_flag & QEDE_TUNN_CSUM_UNNECESSARY) {
                skb->csum_level = 1;
                skb->encapsulation = 1;
        }
}

static inline void qede_skb_receive(struct qede_dev *edev,
                                    struct qede_fastpath *fp,
                                    struct qede_rx_queue *rxq,
                                    struct sk_buff *skb, u16 vlan_tag)
{
        if (vlan_tag)
                __vlan_hwaccel_put_tag(skb, htons(ETH_P_8021Q), vlan_tag);

        napi_gro_receive(&fp->napi, skb);
}

static void qede_set_gro_params(struct qede_dev *edev,
                                struct sk_buff *skb,
                                struct eth_fast_path_rx_tpa_start_cqe *cqe)
{
        u16 parsing_flags = le16_to_cpu(cqe->pars_flags.flags);

        if (((parsing_flags >> PARSING_AND_ERR_FLAGS_L3TYPE_SHIFT) &
            PARSING_AND_ERR_FLAGS_L3TYPE_MASK) == 2)
                skb_shinfo(skb)->gso_type = SKB_GSO_TCPV6;
        else
                skb_shinfo(skb)->gso_type = SKB_GSO_TCPV4;

        skb_shinfo(skb)->gso_size = __le16_to_cpu(cqe->len_on_first_bd) -
                                    cqe->header_len;
}

static int qede_fill_frag_skb(struct qede_dev *edev,
                              struct qede_rx_queue *rxq,
                              u8 tpa_agg_index, u16 len_on_bd)
{
        struct sw_rx_data *current_bd = &rxq->sw_rx_ring[rxq->sw_rx_cons &
                                                         NUM_RX_BDS_MAX];
        struct qede_agg_info *tpa_info = &rxq->tpa_info[tpa_agg_index];
        struct sk_buff *skb = tpa_info->skb;

        if (unlikely(tpa_info->state != QEDE_AGG_STATE_START))
                goto out;

        /* Add one frag and update the appropriate fields in the skb */
        skb_fill_page_desc(skb, tpa_info->frag_id++,
                           current_bd->data,
                           current_bd->page_offset + rxq->rx_headroom,
                           len_on_bd);

        if (unlikely(qede_realloc_rx_buffer(rxq, current_bd))) {
                /* Incr page ref count to reuse on allocation failure
                 * so that it doesn't get freed while freeing SKB.
                 */
                page_ref_inc(current_bd->data);
                goto out;
        }

        qede_rx_bd_ring_consume(rxq);

        skb->data_len += len_on_bd;
        skb->truesize += rxq->rx_buf_seg_size;
        skb->len += len_on_bd;

        return 0;

out:
        tpa_info->state = QEDE_AGG_STATE_ERROR;
        qede_recycle_rx_bd_ring(rxq, 1);

        return -ENOMEM;
}

static bool qede_tunn_exist(u16 flag)
{
        return !!(flag & (PARSING_AND_ERR_FLAGS_TUNNELEXIST_MASK <<
                          PARSING_AND_ERR_FLAGS_TUNNELEXIST_SHIFT));
}

static u8 qede_check_tunn_csum(u16 flag)
{
        u16 csum_flag = 0;
        u8 tcsum = 0;

        if (flag & (PARSING_AND_ERR_FLAGS_TUNNELL4CHKSMWASCALCULATED_MASK <<
                    PARSING_AND_ERR_FLAGS_TUNNELL4CHKSMWASCALCULATED_SHIFT))
                csum_flag |= PARSING_AND_ERR_FLAGS_TUNNELL4CHKSMERROR_MASK <<
                             PARSING_AND_ERR_FLAGS_TUNNELL4CHKSMERROR_SHIFT;

        if (flag & (PARSING_AND_ERR_FLAGS_L4CHKSMWASCALCULATED_MASK <<
                    PARSING_AND_ERR_FLAGS_L4CHKSMWASCALCULATED_SHIFT)) {
                csum_flag |= PARSING_AND_ERR_FLAGS_L4CHKSMERROR_MASK <<
                             PARSING_AND_ERR_FLAGS_L4CHKSMERROR_SHIFT;
                tcsum = QEDE_TUNN_CSUM_UNNECESSARY;
        }

        csum_flag |= PARSING_AND_ERR_FLAGS_TUNNELIPHDRERROR_MASK <<
                     PARSING_AND_ERR_FLAGS_TUNNELIPHDRERROR_SHIFT |
                     PARSING_AND_ERR_FLAGS_IPHDRERROR_MASK <<
                     PARSING_AND_ERR_FLAGS_IPHDRERROR_SHIFT;

        if (csum_flag & flag)
                return QEDE_CSUM_ERROR;

        return QEDE_CSUM_UNNECESSARY | tcsum;
}

static inline struct sk_buff *
qede_build_skb(struct qede_rx_queue *rxq,
               struct sw_rx_data *bd, u16 len, u16 pad)
{
        struct sk_buff *skb;
        void *buf;

        buf = page_address(bd->data) + bd->page_offset;
        skb = build_skb(buf, rxq->rx_buf_seg_size);

        if (unlikely(!skb))
                return NULL;

        skb_reserve(skb, pad);
        skb_put(skb, len);

        return skb;
}

static struct sk_buff *
qede_tpa_rx_build_skb(struct qede_dev *edev,
                      struct qede_rx_queue *rxq,
                      struct sw_rx_data *bd, u16 len, u16 pad,
                      bool alloc_skb)
{
        struct sk_buff *skb;

        skb = qede_build_skb(rxq, bd, len, pad);
        bd->page_offset += rxq->rx_buf_seg_size;

        if (bd->page_offset == PAGE_SIZE) {
                if (unlikely(qede_alloc_rx_buffer(rxq, true))) {
                        DP_NOTICE(edev,
                                  "Failed to allocate RX buffer for tpa start\n");
                        bd->page_offset -= rxq->rx_buf_seg_size;
                        page_ref_inc(bd->data);
                        dev_kfree_skb_any(skb);
                        return NULL;
                }
        } else {
                page_ref_inc(bd->data);
                qede_reuse_page(rxq, bd);
        }

        /* We've consumed the first BD and prepared an SKB */
        qede_rx_bd_ring_consume(rxq);

        return skb;
}

static struct sk_buff *
qede_rx_build_skb(struct qede_dev *edev,
                  struct qede_rx_queue *rxq,
                  struct sw_rx_data *bd, u16 len, u16 pad)
{
        struct sk_buff *skb = NULL;

        /* For smaller frames still need to allocate skb, memcpy
         * data and benefit in reusing the page segment instead of
         * un-mapping it.
         */
        if ((len + pad <= edev->rx_copybreak)) {
                unsigned int offset = bd->page_offset + pad;

                skb = netdev_alloc_skb(edev->ndev, QEDE_RX_HDR_SIZE);
                if (unlikely(!skb))
                        return NULL;

                skb_reserve(skb, pad);
                skb_put_data(skb, page_address(bd->data) + offset, len);
                qede_reuse_page(rxq, bd);
                goto out;
        }

        skb = qede_build_skb(rxq, bd, len, pad);

        if (unlikely(qede_realloc_rx_buffer(rxq, bd))) {
                /* Incr page ref count to reuse on allocation failure so
                 * that it doesn't get freed while freeing SKB [as its
                 * already mapped there].
                 */
                page_ref_inc(bd->data);
                dev_kfree_skb_any(skb);
                return NULL;
        }
out:
        /* We've consumed the first BD and prepared an SKB */
        qede_rx_bd_ring_consume(rxq);

        return skb;
}

static void qede_tpa_start(struct qede_dev *edev,
                           struct qede_rx_queue *rxq,
                           struct eth_fast_path_rx_tpa_start_cqe *cqe)
{
        struct qede_agg_info *tpa_info = &rxq->tpa_info[cqe->tpa_agg_index];
        struct sw_rx_data *sw_rx_data_cons;
        u16 pad;

        sw_rx_data_cons = &rxq->sw_rx_ring[rxq->sw_rx_cons & NUM_RX_BDS_MAX];
        pad = cqe->placement_offset + rxq->rx_headroom;

        tpa_info->skb = qede_tpa_rx_build_skb(edev, rxq, sw_rx_data_cons,
                                              le16_to_cpu(cqe->len_on_first_bd),
                                              pad, false);
        tpa_info->buffer.page_offset = sw_rx_data_cons->page_offset;
        tpa_info->buffer.mapping = sw_rx_data_cons->mapping;

        if (unlikely(!tpa_info->skb)) {
                DP_NOTICE(edev, "Failed to allocate SKB for gro\n");

                /* Consume from ring but do not produce since
                 * this might be used by FW still, it will be re-used
                 * at TPA end.
                 */
                tpa_info->tpa_start_fail = true;
                qede_rx_bd_ring_consume(rxq);
                tpa_info->state = QEDE_AGG_STATE_ERROR;
                goto cons_buf;
        }

        tpa_info->frag_id = 0;
        tpa_info->state = QEDE_AGG_STATE_START;

        if ((le16_to_cpu(cqe->pars_flags.flags) >>
             PARSING_AND_ERR_FLAGS_TAG8021QEXIST_SHIFT) &
            PARSING_AND_ERR_FLAGS_TAG8021QEXIST_MASK)
                tpa_info->vlan_tag = le16_to_cpu(cqe->vlan_tag);
        else
                tpa_info->vlan_tag = 0;

        qede_get_rxhash(tpa_info->skb, cqe->bitfields, cqe->rss_hash);

        /* This is needed in order to enable forwarding support */
        qede_set_gro_params(edev, tpa_info->skb, cqe);

cons_buf: /* We still need to handle bd_len_list to consume buffers */
        if (likely(cqe->bw_ext_bd_len_list[0]))
                qede_fill_frag_skb(edev, rxq, cqe->tpa_agg_index,
                                   le16_to_cpu(cqe->bw_ext_bd_len_list[0]));

        if (unlikely(cqe->bw_ext_bd_len_list[1])) {
                DP_ERR(edev,
                       "Unlikely - got a TPA aggregation with more than one bw_ext_bd_len_list entry in the TPA start\n");
                tpa_info->state = QEDE_AGG_STATE_ERROR;
        }
}

#ifdef CONFIG_INET
static void qede_gro_ip_csum(struct sk_buff *skb)
{
        const struct iphdr *iph = ip_hdr(skb);
        struct tcphdr *th;

        skb_set_transport_header(skb, sizeof(struct iphdr));
        th = tcp_hdr(skb);

        th->check = ~tcp_v4_check(skb->len - skb_transport_offset(skb),
                                  iph->saddr, iph->daddr, 0);

        tcp_gro_complete(skb);
}

static void qede_gro_ipv6_csum(struct sk_buff *skb)
{
        struct ipv6hdr *iph = ipv6_hdr(skb);
        struct tcphdr *th;

        skb_set_transport_header(skb, sizeof(struct ipv6hdr));
        th = tcp_hdr(skb);

        th->check = ~tcp_v6_check(skb->len - skb_transport_offset(skb),
                                  &iph->saddr, &iph->daddr, 0);
        tcp_gro_complete(skb);
}
#endif

static void qede_gro_receive(struct qede_dev *edev,
                             struct qede_fastpath *fp,
                             struct sk_buff *skb,
                             u16 vlan_tag)
{
        /* FW can send a single MTU sized packet from gro flow
         * due to aggregation timeout/last segment etc. which
         * is not expected to be a gro packet. If a skb has zero
         * frags then simply push it in the stack as non gso skb.
         */
        if (unlikely(!skb->data_len)) {
                skb_shinfo(skb)->gso_type = 0;
                skb_shinfo(skb)->gso_size = 0;
                goto send_skb;
        }

#ifdef CONFIG_INET
        if (skb_shinfo(skb)->gso_size) {
                skb_reset_network_header(skb);

                switch (skb->protocol) {
                case htons(ETH_P_IP):
                        qede_gro_ip_csum(skb);
                        break;
                case htons(ETH_P_IPV6):
                        qede_gro_ipv6_csum(skb);
                        break;
                default:
                        DP_ERR(edev,
                               "Error: FW GRO supports only IPv4/IPv6, not 0x%04x\n",
                               ntohs(skb->protocol));
                }
        }
#endif

send_skb:
        skb_record_rx_queue(skb, fp->rxq->rxq_id);
        qede_skb_receive(edev, fp, fp->rxq, skb, vlan_tag);
}

static inline void qede_tpa_cont(struct qede_dev *edev,
                                 struct qede_rx_queue *rxq,
                                 struct eth_fast_path_rx_tpa_cont_cqe *cqe)
{
        int i;

        for (i = 0; cqe->len_list[i] && i < ARRAY_SIZE(cqe->len_list); i++)
                qede_fill_frag_skb(edev, rxq, cqe->tpa_agg_index,
                                   le16_to_cpu(cqe->len_list[i]));

        if (unlikely(i > 1))
                DP_ERR(edev,
                       "Strange - TPA cont with more than a single len_list entry\n");
}

static int qede_tpa_end(struct qede_dev *edev,
                        struct qede_fastpath *fp,
                        struct eth_fast_path_rx_tpa_end_cqe *cqe)
{
        struct qede_rx_queue *rxq = fp->rxq;
        struct qede_agg_info *tpa_info;
        struct sk_buff *skb;
        int i;

        tpa_info = &rxq->tpa_info[cqe->tpa_agg_index];
        skb = tpa_info->skb;

        if (tpa_info->buffer.page_offset == PAGE_SIZE)
                dma_unmap_page(rxq->dev, tpa_info->buffer.mapping,
                               PAGE_SIZE, rxq->data_direction);

        for (i = 0; cqe->len_list[i] && i < ARRAY_SIZE(cqe->len_list); i++)
                qede_fill_frag_skb(edev, rxq, cqe->tpa_agg_index,
                                   le16_to_cpu(cqe->len_list[i]));
        if (unlikely(i > 1))
                DP_ERR(edev,
                       "Strange - TPA emd with more than a single len_list entry\n");

        if (unlikely(tpa_info->state != QEDE_AGG_STATE_START))
                goto err;

        /* Sanity */
        if (unlikely(cqe->num_of_bds != tpa_info->frag_id + 1))
                DP_ERR(edev,
                       "Strange - TPA had %02x BDs, but SKB has only %d frags\n",
                       cqe->num_of_bds, tpa_info->frag_id);
        if (unlikely(skb->len != le16_to_cpu(cqe->total_packet_len)))
                DP_ERR(edev,
                       "Strange - total packet len [cqe] is %4x but SKB has len %04x\n",
                       le16_to_cpu(cqe->total_packet_len), skb->len);

        /* Finalize the SKB */
        skb->protocol = eth_type_trans(skb, edev->ndev);
        skb->ip_summed = CHECKSUM_UNNECESSARY;

        /* tcp_gro_complete() will copy NAPI_GRO_CB(skb)->count
         * to skb_shinfo(skb)->gso_segs
         */
        NAPI_GRO_CB(skb)->count = le16_to_cpu(cqe->num_of_coalesced_segs);

        qede_gro_receive(edev, fp, skb, tpa_info->vlan_tag);

        tpa_info->state = QEDE_AGG_STATE_NONE;

        return 1;
err:
        tpa_info->state = QEDE_AGG_STATE_NONE;

        if (tpa_info->tpa_start_fail) {
                qede_reuse_page(rxq, &tpa_info->buffer);
                tpa_info->tpa_start_fail = false;
        }

        dev_kfree_skb_any(tpa_info->skb);
        tpa_info->skb = NULL;
        return 0;
}

static u8 qede_check_notunn_csum(u16 flag)
{
        u16 csum_flag = 0;
        u8 csum = 0;

        if (flag & (PARSING_AND_ERR_FLAGS_L4CHKSMWASCALCULATED_MASK <<
                    PARSING_AND_ERR_FLAGS_L4CHKSMWASCALCULATED_SHIFT)) {
                csum_flag |= PARSING_AND_ERR_FLAGS_L4CHKSMERROR_MASK <<
                             PARSING_AND_ERR_FLAGS_L4CHKSMERROR_SHIFT;
                csum = QEDE_CSUM_UNNECESSARY;
        }

        csum_flag |= PARSING_AND_ERR_FLAGS_IPHDRERROR_MASK <<
                     PARSING_AND_ERR_FLAGS_IPHDRERROR_SHIFT;

        if (csum_flag & flag)
                return QEDE_CSUM_ERROR;

        return csum;
}

static u8 qede_check_csum(u16 flag)
{
        if (!qede_tunn_exist(flag))
                return qede_check_notunn_csum(flag);
        else
                return qede_check_tunn_csum(flag);
}

static bool qede_pkt_is_ip_fragmented(struct eth_fast_path_rx_reg_cqe *cqe,
                                      u16 flag)
{
        u8 tun_pars_flg = cqe->tunnel_pars_flags.flags;

        if ((tun_pars_flg & (ETH_TUNNEL_PARSING_FLAGS_IPV4_FRAGMENT_MASK <<
                             ETH_TUNNEL_PARSING_FLAGS_IPV4_FRAGMENT_SHIFT)) ||
            (flag & (PARSING_AND_ERR_FLAGS_IPV4FRAG_MASK <<
                     PARSING_AND_ERR_FLAGS_IPV4FRAG_SHIFT)))
                return true;

        return false;
}

/* Return true iff packet is to be passed to stack */
static bool qede_rx_xdp(struct qede_dev *edev,
                        struct qede_fastpath *fp,
                        struct qede_rx_queue *rxq,
                        struct bpf_prog *prog,
                        struct sw_rx_data *bd,
                        struct eth_fast_path_rx_reg_cqe *cqe,
                        u16 *data_offset, u16 *len)
{
        struct xdp_buff xdp;
        enum xdp_action act;

        xdp_init_buff(&xdp, rxq->rx_buf_seg_size, &rxq->xdp_rxq);
        xdp_prepare_buff(&xdp, page_address(bd->data), *data_offset,
                         *len, false);

        act = bpf_prog_run_xdp(prog, &xdp);

        /* Recalculate, as XDP might have changed the headers */
        *data_offset = xdp.data - xdp.data_hard_start;
        *len = xdp.data_end - xdp.data;

        if (act == XDP_PASS)
                return true;

        /* Count number of packets not to be passed to stack */
        rxq->xdp_no_pass++;

        switch (act) {
        case XDP_TX:
                /* We need the replacement buffer before transmit. */
                if (unlikely(qede_alloc_rx_buffer(rxq, true))) {
                        qede_recycle_rx_bd_ring(rxq, 1);

                        trace_xdp_exception(edev->ndev, prog, act);
                        break;
                }

                /* Now if there's a transmission problem, we'd still have to
                 * throw current buffer, as replacement was already allocated.
                 */
                if (unlikely(qede_xdp_xmit(fp->xdp_tx, bd->mapping,
                                           *data_offset, *len, bd->data,
                                           NULL))) {
                        dma_unmap_page(rxq->dev, bd->mapping, PAGE_SIZE,
                                       rxq->data_direction);
                        __free_page(bd->data);

                        trace_xdp_exception(edev->ndev, prog, act);
                } else {
                        dma_sync_single_for_device(rxq->dev,
                                                   bd->mapping + *data_offset,
                                                   *len, rxq->data_direction);
                        fp->xdp_xmit |= QEDE_XDP_TX;
                }

                /* Regardless, we've consumed an Rx BD */
                qede_rx_bd_ring_consume(rxq);
                break;
        case XDP_REDIRECT:
                /* We need the replacement buffer before transmit. */
                if (unlikely(qede_alloc_rx_buffer(rxq, true))) {
                        qede_recycle_rx_bd_ring(rxq, 1);

                        trace_xdp_exception(edev->ndev, prog, act);
                        break;
                }

                dma_unmap_page(rxq->dev, bd->mapping, PAGE_SIZE,
                               rxq->data_direction);

                if (unlikely(xdp_do_redirect(edev->ndev, &xdp, prog)))
                        DP_NOTICE(edev, "Failed to redirect the packet\n");
                else
                        fp->xdp_xmit |= QEDE_XDP_REDIRECT;

                qede_rx_bd_ring_consume(rxq);
                break;
        default:
                bpf_warn_invalid_xdp_action(edev->ndev, prog, act);
                fallthrough;
        case XDP_ABORTED:
                trace_xdp_exception(edev->ndev, prog, act);
                fallthrough;
        case XDP_DROP:
                qede_recycle_rx_bd_ring(rxq, cqe->bd_num);
        }

        return false;
}

static int qede_rx_build_jumbo(struct qede_dev *edev,
                               struct qede_rx_queue *rxq,
                               struct sk_buff *skb,
                               struct eth_fast_path_rx_reg_cqe *cqe,
                               u16 first_bd_len)
{
        u16 pkt_len = le16_to_cpu(cqe->pkt_len);
        struct sw_rx_data *bd;
        u16 bd_cons_idx;
        u8 num_frags;

        pkt_len -= first_bd_len;

        /* We've already used one BD for the SKB. Now take care of the rest */
        for (num_frags = cqe->bd_num - 1; num_frags > 0; num_frags--) {
                u16 cur_size = pkt_len > rxq->rx_buf_size ? rxq->rx_buf_size :
                    pkt_len;

                if (unlikely(!cur_size)) {
                        DP_ERR(edev,
                               "Still got %d BDs for mapping jumbo, but length became 0\n",
                               num_frags);
                        goto out;
                }

                /* We need a replacement buffer for each BD */
                if (unlikely(qede_alloc_rx_buffer(rxq, true)))
                        goto out;

                /* Now that we've allocated the replacement buffer,
                 * we can safely consume the next BD and map it to the SKB.
                 */
                bd_cons_idx = rxq->sw_rx_cons & NUM_RX_BDS_MAX;
                bd = &rxq->sw_rx_ring[bd_cons_idx];
                qede_rx_bd_ring_consume(rxq);

                dma_unmap_page(rxq->dev, bd->mapping,
                               PAGE_SIZE, DMA_FROM_DEVICE);

                skb_add_rx_frag(skb, skb_shinfo(skb)->nr_frags, bd->data,
                                rxq->rx_headroom, cur_size, PAGE_SIZE);

                pkt_len -= cur_size;
        }

        if (unlikely(pkt_len))
                DP_ERR(edev,
                       "Mapped all BDs of jumbo, but still have %d bytes\n",
                       pkt_len);

out:
        return num_frags;
}

static int qede_rx_process_tpa_cqe(struct qede_dev *edev,
                                   struct qede_fastpath *fp,
                                   struct qede_rx_queue *rxq,
                                   union eth_rx_cqe *cqe,
                                   enum eth_rx_cqe_type type)
{
        switch (type) {
        case ETH_RX_CQE_TYPE_TPA_START:
                qede_tpa_start(edev, rxq, &cqe->fast_path_tpa_start);
                return 0;
        case ETH_RX_CQE_TYPE_TPA_CONT:
                qede_tpa_cont(edev, rxq, &cqe->fast_path_tpa_cont);
                return 0;
        case ETH_RX_CQE_TYPE_TPA_END:
                return qede_tpa_end(edev, fp, &cqe->fast_path_tpa_end);
        default:
                return 0;
        }
}

static int qede_rx_process_cqe(struct qede_dev *edev,
                               struct qede_fastpath *fp,
                               struct qede_rx_queue *rxq)
{
        struct bpf_prog *xdp_prog = READ_ONCE(rxq->xdp_prog);
        struct eth_fast_path_rx_reg_cqe *fp_cqe;
        u16 len, pad, bd_cons_idx, parse_flag;
        enum eth_rx_cqe_type cqe_type;
        union eth_rx_cqe *cqe;
        struct sw_rx_data *bd;
        struct sk_buff *skb;
        __le16 flags;
        u8 csum_flag;

        /* Get the CQE from the completion ring */
        cqe = (union eth_rx_cqe *)qed_chain_consume(&rxq->rx_comp_ring);
        cqe_type = cqe->fast_path_regular.type;

        /* Process an unlikely slowpath event */
        if (unlikely(cqe_type == ETH_RX_CQE_TYPE_SLOW_PATH)) {
                struct eth_slow_path_rx_cqe *sp_cqe;

                sp_cqe = (struct eth_slow_path_rx_cqe *)cqe;
                edev->ops->eth_cqe_completion(edev->cdev, fp->id, sp_cqe);
                return 0;
        }

        /* Handle TPA cqes */
        if (cqe_type != ETH_RX_CQE_TYPE_REGULAR)
                return qede_rx_process_tpa_cqe(edev, fp, rxq, cqe, cqe_type);

        /* Get the data from the SW ring; Consume it only after it's evident
         * we wouldn't recycle it.
         */
        bd_cons_idx = rxq->sw_rx_cons & NUM_RX_BDS_MAX;
        bd = &rxq->sw_rx_ring[bd_cons_idx];

        fp_cqe = &cqe->fast_path_regular;
        len = le16_to_cpu(fp_cqe->len_on_first_bd);
        pad = fp_cqe->placement_offset + rxq->rx_headroom;

        /* Run eBPF program if one is attached */
        if (xdp_prog)
                if (!qede_rx_xdp(edev, fp, rxq, xdp_prog, bd, fp_cqe,
                                 &pad, &len))
                        return 0;

        /* If this is an error packet then drop it */
        flags = cqe->fast_path_regular.pars_flags.flags;
        parse_flag = le16_to_cpu(flags);

        csum_flag = qede_check_csum(parse_flag);
        if (unlikely(csum_flag == QEDE_CSUM_ERROR)) {
                if (qede_pkt_is_ip_fragmented(fp_cqe, parse_flag))
                        rxq->rx_ip_frags++;
                else
                        rxq->rx_hw_errors++;
        }

        /* Basic validation passed; Need to prepare an SKB. This would also
         * guarantee to finally consume the first BD upon success.
         */
        skb = qede_rx_build_skb(edev, rxq, bd, len, pad);
        if (!skb) {
                rxq->rx_alloc_errors++;
                qede_recycle_rx_bd_ring(rxq, fp_cqe->bd_num);
                return 0;
        }

        /* In case of Jumbo packet, several PAGE_SIZEd buffers will be pointed
         * by a single cqe.
         */
        if (fp_cqe->bd_num > 1) {
                u16 unmapped_frags = qede_rx_build_jumbo(edev, rxq, skb,
                                                         fp_cqe, len);

                if (unlikely(unmapped_frags > 0)) {
                        qede_recycle_rx_bd_ring(rxq, unmapped_frags);
                        dev_kfree_skb_any(skb);
                        return 0;
                }
        }

        /* The SKB contains all the data. Now prepare meta-magic */
        skb->protocol = eth_type_trans(skb, edev->ndev);
        qede_get_rxhash(skb, fp_cqe->bitfields, fp_cqe->rss_hash);
        qede_set_skb_csum(skb, csum_flag);
        skb_record_rx_queue(skb, rxq->rxq_id);
        qede_ptp_record_rx_ts(edev, cqe, skb);

        /* SKB is prepared - pass it to stack */
        qede_skb_receive(edev, fp, rxq, skb, le16_to_cpu(fp_cqe->vlan_tag));

        return 1;
}

static int qede_rx_int(struct qede_fastpath *fp, int budget)
{
        struct qede_rx_queue *rxq = fp->rxq;
        struct qede_dev *edev = fp->edev;
        int work_done = 0, rcv_pkts = 0;
        u16 hw_comp_cons, sw_comp_cons;

        hw_comp_cons = le16_to_cpu(*rxq->hw_cons_ptr);
        sw_comp_cons = qed_chain_get_cons_idx(&rxq->rx_comp_ring);

        /* Memory barrier to prevent the CPU from doing speculative reads of CQE
         * / BD in the while-loop before reading hw_comp_cons. If the CQE is
         * read before it is written by FW, then FW writes CQE and SB, and then
         * the CPU reads the hw_comp_cons, it will use an old CQE.
         */
        rmb();

        /* Loop to complete all indicated BDs */
        while ((sw_comp_cons != hw_comp_cons) && (work_done < budget)) {
                rcv_pkts += qede_rx_process_cqe(edev, fp, rxq);
                qed_chain_recycle_consumed(&rxq->rx_comp_ring);
                sw_comp_cons = qed_chain_get_cons_idx(&rxq->rx_comp_ring);
                work_done++;
        }

        rxq->rcv_pkts += rcv_pkts;

        /* Allocate replacement buffers */
        while (rxq->num_rx_buffers - rxq->filled_buffers)
                if (qede_alloc_rx_buffer(rxq, false))
                        break;

        /* Update producers */
        qede_update_rx_prod(edev, rxq);

        return work_done;
}

static bool qede_poll_is_more_work(struct qede_fastpath *fp)
{
        qed_sb_update_sb_idx(fp->sb_info);

        /* *_has_*_work() reads the status block, thus we need to ensure that
         * status block indices have been actually read (qed_sb_update_sb_idx)
         * prior to this check (*_has_*_work) so that we won't write the
         * "newer" value of the status block to HW (if there was a DMA right
         * after qede_has_rx_work and if there is no rmb, the memory reading
         * (qed_sb_update_sb_idx) may be postponed to right before *_ack_sb).
         * In this case there will never be another interrupt until there is
         * another update of the status block, while there is still unhandled
         * work.
         */
        rmb();

        if (likely(fp->type & QEDE_FASTPATH_RX))
                if (qede_has_rx_work(fp->rxq))
                        return true;

        if (fp->type & QEDE_FASTPATH_XDP)
                if (qede_txq_has_work(fp->xdp_tx))
                        return true;

        if (likely(fp->type & QEDE_FASTPATH_TX)) {
                int cos;

                for_each_cos_in_txq(fp->edev, cos) {
                        if (qede_txq_has_work(&fp->txq[cos]))
                                return true;
                }
        }

        return false;
}

/*********************
 * NDO & API related *
 *********************/
int qede_poll(struct napi_struct *napi, int budget)
{
        struct qede_fastpath *fp = container_of(napi, struct qede_fastpath,
                                                napi);
        struct qede_dev *edev = fp->edev;
        int rx_work_done = 0;
        u16 xdp_prod;

        fp->xdp_xmit = 0;

        if (likely(fp->type & QEDE_FASTPATH_TX)) {
                int cos;

                for_each_cos_in_txq(fp->edev, cos) {
                        if (qede_txq_has_work(&fp->txq[cos]))
                                qede_tx_int(edev, &fp->txq[cos]);
                }
        }

        if ((fp->type & QEDE_FASTPATH_XDP) && qede_txq_has_work(fp->xdp_tx))
                qede_xdp_tx_int(edev, fp->xdp_tx);

        rx_work_done = (likely(fp->type & QEDE_FASTPATH_RX) &&
                        qede_has_rx_work(fp->rxq)) ?
                        qede_rx_int(fp, budget) : 0;

        if (fp->xdp_xmit & QEDE_XDP_REDIRECT)
                xdp_do_flush();

        /* Handle case where we are called by netpoll with a budget of 0 */
        if (rx_work_done < budget || !budget) {
                if (!qede_poll_is_more_work(fp)) {
                        napi_complete_done(napi, rx_work_done);

                        /* Update and reenable interrupts */
                        qed_sb_ack(fp->sb_info, IGU_INT_ENABLE, 1);
                } else {
                        rx_work_done = budget;
                }
        }

        if (fp->xdp_xmit & QEDE_XDP_TX) {
                xdp_prod = qed_chain_get_prod_idx(&fp->xdp_tx->tx_pbl);

                fp->xdp_tx->tx_db.data.bd_prod = cpu_to_le16(xdp_prod);
                qede_update_tx_producer(fp->xdp_tx);
        }

        return rx_work_done;
}

irqreturn_t qede_msix_fp_int(int irq, void *fp_cookie)
{
        struct qede_fastpath *fp = fp_cookie;

        qed_sb_ack(fp->sb_info, IGU_INT_DISABLE, 0 /*do not update*/);

        napi_schedule_irqoff(&fp->napi);
        return IRQ_HANDLED;
}

/* Main transmit function */
netdev_tx_t qede_start_xmit(struct sk_buff *skb, struct net_device *ndev)
{
        struct qede_dev *edev = netdev_priv(ndev);
        struct netdev_queue *netdev_txq;
        struct qede_tx_queue *txq;
        struct eth_tx_1st_bd *first_bd;
        struct eth_tx_2nd_bd *second_bd = NULL;
        struct eth_tx_3rd_bd *third_bd = NULL;
        struct eth_tx_bd *tx_data_bd = NULL;
        u16 txq_index, val = 0;
        u8 nbd = 0;
        dma_addr_t mapping;
        int rc, frag_idx = 0, ipv6_ext = 0;
        u8 xmit_type;
        u16 idx;
        u16 hlen;
        bool data_split = false;

        /* Get tx-queue context and netdev index */
        txq_index = skb_get_queue_mapping(skb);
        WARN_ON(txq_index >= QEDE_TSS_COUNT(edev) * edev->dev_info.num_tc);
        txq = QEDE_NDEV_TXQ_ID_TO_TXQ(edev, txq_index);
        netdev_txq = netdev_get_tx_queue(ndev, txq_index);

        WARN_ON(qed_chain_get_elem_left(&txq->tx_pbl) < (MAX_SKB_FRAGS + 1));

        xmit_type = qede_xmit_type(skb, &ipv6_ext);

#if ((MAX_SKB_FRAGS + 2) > ETH_TX_MAX_BDS_PER_NON_LSO_PACKET)
        if (qede_pkt_req_lin(skb, xmit_type)) {
                if (skb_linearize(skb)) {
                        txq->tx_mem_alloc_err++;

                        dev_kfree_skb_any(skb);
                        return NETDEV_TX_OK;
                }
        }
#endif

        /* Fill the entry in the SW ring and the BDs in the FW ring */
        idx = txq->sw_tx_prod;
        txq->sw_tx_ring.skbs[idx].skb = skb;
        first_bd = (struct eth_tx_1st_bd *)
                   qed_chain_produce(&txq->tx_pbl);
        memset(first_bd, 0, sizeof(*first_bd));
        first_bd->data.bd_flags.bitfields =
                1 << ETH_TX_1ST_BD_FLAGS_START_BD_SHIFT;

        if (unlikely(skb_shinfo(skb)->tx_flags & SKBTX_HW_TSTAMP))
                qede_ptp_tx_ts(edev, skb);

        /* Map skb linear data for DMA and set in the first BD */
        mapping = dma_map_single(txq->dev, skb->data,
                                 skb_headlen(skb), DMA_TO_DEVICE);
        if (unlikely(dma_mapping_error(txq->dev, mapping))) {
                DP_NOTICE(edev, "SKB mapping failed\n");
                qede_free_failed_tx_pkt(txq, first_bd, 0, false);
                qede_update_tx_producer(txq);
                return NETDEV_TX_OK;
        }
        nbd++;
        BD_SET_UNMAP_ADDR_LEN(first_bd, mapping, skb_headlen(skb));

        /* In case there is IPv6 with extension headers or LSO we need 2nd and
         * 3rd BDs.
         */
        if (unlikely((xmit_type & XMIT_LSO) | ipv6_ext)) {
                second_bd = (struct eth_tx_2nd_bd *)
                        qed_chain_produce(&txq->tx_pbl);
                memset(second_bd, 0, sizeof(*second_bd));

                nbd++;
                third_bd = (struct eth_tx_3rd_bd *)
                        qed_chain_produce(&txq->tx_pbl);
                memset(third_bd, 0, sizeof(*third_bd));

                nbd++;
                /* We need to fill in additional data in second_bd... */
                tx_data_bd = (struct eth_tx_bd *)second_bd;
        }

        if (skb_vlan_tag_present(skb)) {
                first_bd->data.vlan = cpu_to_le16(skb_vlan_tag_get(skb));
                first_bd->data.bd_flags.bitfields |=
                        1 << ETH_TX_1ST_BD_FLAGS_VLAN_INSERTION_SHIFT;
        }

        /* Fill the parsing flags & params according to the requested offload */
        if (xmit_type & XMIT_L4_CSUM) {
                /* We don't re-calculate IP checksum as it is already done by
                 * the upper stack
                 */
                first_bd->data.bd_flags.bitfields |=
                        1 << ETH_TX_1ST_BD_FLAGS_L4_CSUM_SHIFT;

                if (xmit_type & XMIT_ENC) {
                        first_bd->data.bd_flags.bitfields |=
                                1 << ETH_TX_1ST_BD_FLAGS_IP_CSUM_SHIFT;

                        val |= (1 << ETH_TX_DATA_1ST_BD_TUNN_FLAG_SHIFT);
                }

                /* Legacy FW had flipped behavior in regard to this bit -
                 * I.e., needed to set to prevent FW from touching encapsulated
                 * packets when it didn't need to.
                 */
                if (unlikely(txq->is_legacy))
                        val ^= (1 << ETH_TX_DATA_1ST_BD_TUNN_FLAG_SHIFT);

                /* If the packet is IPv6 with extension header, indicate that
                 * to FW and pass few params, since the device cracker doesn't
                 * support parsing IPv6 with extension header/s.
                 */
                if (unlikely(ipv6_ext))
                        qede_set_params_for_ipv6_ext(skb, second_bd, third_bd);
        }

        if (xmit_type & XMIT_LSO) {
                first_bd->data.bd_flags.bitfields |=
                        (1 << ETH_TX_1ST_BD_FLAGS_LSO_SHIFT);
                third_bd->data.lso_mss =
                        cpu_to_le16(skb_shinfo(skb)->gso_size);

                if (unlikely(xmit_type & XMIT_ENC)) {
                        first_bd->data.bd_flags.bitfields |=
                                1 << ETH_TX_1ST_BD_FLAGS_TUNN_IP_CSUM_SHIFT;

                        if (xmit_type & XMIT_ENC_GSO_L4_CSUM) {
                                u8 tmp = ETH_TX_1ST_BD_FLAGS_TUNN_L4_CSUM_SHIFT;

                                first_bd->data.bd_flags.bitfields |= 1 << tmp;
                        }
                        hlen = qede_get_skb_hlen(skb, true);
                } else {
                        first_bd->data.bd_flags.bitfields |=
                                1 << ETH_TX_1ST_BD_FLAGS_IP_CSUM_SHIFT;
                        hlen = qede_get_skb_hlen(skb, false);
                }

                /* @@@TBD - if will not be removed need to check */
                third_bd->data.bitfields |=
                        cpu_to_le16(1 << ETH_TX_DATA_3RD_BD_HDR_NBD_SHIFT);

                /* Make life easier for FW guys who can't deal with header and
                 * data on same BD. If we need to split, use the second bd...
                 */
                if (unlikely(skb_headlen(skb) > hlen)) {
                        DP_VERBOSE(edev, NETIF_MSG_TX_QUEUED,
                                   "TSO split header size is %d (%x:%x)\n",
                                   first_bd->nbytes, first_bd->addr.hi,
                                   first_bd->addr.lo);

                        mapping = HILO_U64(le32_to_cpu(first_bd->addr.hi),
                                           le32_to_cpu(first_bd->addr.lo)) +
                                           hlen;

                        BD_SET_UNMAP_ADDR_LEN(tx_data_bd, mapping,
                                              le16_to_cpu(first_bd->nbytes) -
                                              hlen);

                        /* this marks the BD as one that has no
                         * individual mapping
                         */
                        txq->sw_tx_ring.skbs[idx].flags |= QEDE_TSO_SPLIT_BD;

                        first_bd->nbytes = cpu_to_le16(hlen);

                        tx_data_bd = (struct eth_tx_bd *)third_bd;
                        data_split = true;
                }
        } else {
                if (unlikely(skb->len > ETH_TX_MAX_NON_LSO_PKT_LEN)) {
                        DP_ERR(edev, "Unexpected non LSO skb length = 0x%x\n", skb->len);
                        qede_free_failed_tx_pkt(txq, first_bd, 0, false);
                        qede_update_tx_producer(txq);
                        return NETDEV_TX_OK;
                }

                val |= ((skb->len & ETH_TX_DATA_1ST_BD_PKT_LEN_MASK) <<
                         ETH_TX_DATA_1ST_BD_PKT_LEN_SHIFT);
        }

        first_bd->data.bitfields = cpu_to_le16(val);

        /* Handle fragmented skb */
        /* special handle for frags inside 2nd and 3rd bds.. */
        while (tx_data_bd && frag_idx < skb_shinfo(skb)->nr_frags) {
                rc = map_frag_to_bd(txq,
                                    &skb_shinfo(skb)->frags[frag_idx],
                                    tx_data_bd);
                if (rc) {
                        qede_free_failed_tx_pkt(txq, first_bd, nbd, data_split);
                        qede_update_tx_producer(txq);
                        return NETDEV_TX_OK;
                }

                if (tx_data_bd == (struct eth_tx_bd *)second_bd)
                        tx_data_bd = (struct eth_tx_bd *)third_bd;
                else
                        tx_data_bd = NULL;

                frag_idx++;
        }

        /* map last frags into 4th, 5th .... */
        for (; frag_idx < skb_shinfo(skb)->nr_frags; frag_idx++, nbd++) {
                tx_data_bd = (struct eth_tx_bd *)
                             qed_chain_produce(&txq->tx_pbl);

                memset(tx_data_bd, 0, sizeof(*tx_data_bd));

                rc = map_frag_to_bd(txq,
                                    &skb_shinfo(skb)->frags[frag_idx],
                                    tx_data_bd);
                if (rc) {
                        qede_free_failed_tx_pkt(txq, first_bd, nbd, data_split);
                        qede_update_tx_producer(txq);
                        return NETDEV_TX_OK;
                }
        }

        /* update the first BD with the actual num BDs */
        first_bd->data.nbds = nbd;

        netdev_tx_sent_queue(netdev_txq, skb->len);

        skb_tx_timestamp(skb);

        /* Advance packet producer only before sending the packet since mapping
         * of pages may fail.
         */
        txq->sw_tx_prod = (txq->sw_tx_prod + 1) % txq->num_tx_buffers;

        /* 'next page' entries are counted in the producer value */
        txq->tx_db.data.bd_prod =
                cpu_to_le16(qed_chain_get_prod_idx(&txq->tx_pbl));

        if (!netdev_xmit_more() || netif_xmit_stopped(netdev_txq))
                qede_update_tx_producer(txq);

        if (unlikely(qed_chain_get_elem_left(&txq->tx_pbl)
                      < (MAX_SKB_FRAGS + 1))) {
                if (netdev_xmit_more())
                        qede_update_tx_producer(txq);

                netif_tx_stop_queue(netdev_txq);
                txq->stopped_cnt++;
                DP_VERBOSE(edev, NETIF_MSG_TX_QUEUED,
                           "Stop queue was called\n");
                /* paired memory barrier is in qede_tx_int(), we have to keep
                 * ordering of set_bit() in netif_tx_stop_queue() and read of
                 * fp->bd_tx_cons
                 */
                smp_mb();

                if ((qed_chain_get_elem_left(&txq->tx_pbl) >=
                     (MAX_SKB_FRAGS + 1)) &&
                    (edev->state == QEDE_STATE_OPEN)) {
                        netif_tx_wake_queue(netdev_txq);
                        DP_VERBOSE(edev, NETIF_MSG_TX_QUEUED,
                                   "Wake queue was called\n");
                }
        }

        return NETDEV_TX_OK;
}

u16 qede_select_queue(struct net_device *dev, struct sk_buff *skb,
                      struct net_device *sb_dev)
{
        struct qede_dev *edev = netdev_priv(dev);
        int total_txq;

        total_txq = QEDE_TSS_COUNT(edev) * edev->dev_info.num_tc;

        return QEDE_TSS_COUNT(edev) ?
                netdev_pick_tx(dev, skb, NULL) % total_txq :  0;
}

/* 8B udp header + 8B base tunnel header + 32B option length */
#define QEDE_MAX_TUN_HDR_LEN 48

netdev_features_t qede_features_check(struct sk_buff *skb,
                                      struct net_device *dev,
                                      netdev_features_t features)
{
        if (skb->encapsulation) {
                u8 l4_proto = 0;

                switch (vlan_get_protocol(skb)) {
                case htons(ETH_P_IP):
                        l4_proto = ip_hdr(skb)->protocol;
                        break;
                case htons(ETH_P_IPV6):
                        l4_proto = ipv6_hdr(skb)->nexthdr;
                        break;
                default:
                        return features;
                }

                /* Disable offloads for geneve tunnels, as HW can't parse
                 * the geneve header which has option length greater than 32b
                 * and disable offloads for the ports which are not offloaded.
                 */
                if (l4_proto == IPPROTO_UDP) {
                        struct qede_dev *edev = netdev_priv(dev);
                        u16 hdrlen, vxln_port, gnv_port;

                        hdrlen = QEDE_MAX_TUN_HDR_LEN;
                        vxln_port = edev->vxlan_dst_port;
                        gnv_port = edev->geneve_dst_port;

                        if ((skb_inner_mac_header(skb) -
                             skb_transport_header(skb)) > hdrlen ||
                             (ntohs(udp_hdr(skb)->dest) != vxln_port &&
                              ntohs(udp_hdr(skb)->dest) != gnv_port))
                                return features & ~(NETIF_F_CSUM_MASK |
                                                    NETIF_F_GSO_MASK);
                } else if (l4_proto == IPPROTO_IPIP) {
                        /* IPIP tunnels are unknown to the device or at least unsupported natively,
                         * offloads for them can't be done trivially, so disable them for such skb.
                         */
                        return features & ~(NETIF_F_CSUM_MASK | NETIF_F_GSO_MASK);
                }
        }

        return features;
}