/* Broadcom NetXtreme-C/E network driver.
 *
 * Copyright (c) 2014-2016 Broadcom Corporation
 * Copyright (c) 2016-2018 Broadcom Limited
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation.
 */

#include <linux/module.h>

#include <linux/stringify.h>
#include <linux/kernel.h>
#include <linux/timer.h>
#include <linux/errno.h>
#include <linux/ioport.h>
#include <linux/slab.h>
#include <linux/vmalloc.h>
#include <linux/interrupt.h>
#include <linux/pci.h>
#include <linux/netdevice.h>
#include <linux/etherdevice.h>
#include <linux/skbuff.h>
#include <linux/dma-mapping.h>
#include <linux/bitops.h>
#include <linux/io.h>
#include <linux/irq.h>
#include <linux/delay.h>
#include <asm/byteorder.h>
#include <asm/page.h>
#include <linux/time.h>
#include <linux/mii.h>
#include <linux/if.h>
#include <linux/if_vlan.h>
#include <linux/if_bridge.h>
#include <linux/rtc.h>
#include <linux/bpf.h>
#include <net/ip.h>
#include <net/tcp.h>
#include <net/udp.h>
#include <net/checksum.h>
#include <net/ip6_checksum.h>
#include <net/udp_tunnel.h>
#include <linux/workqueue.h>
#include <linux/prefetch.h>
#include <linux/cache.h>
#include <linux/log2.h>
#include <linux/aer.h>
#include <linux/bitmap.h>
#include <linux/cpu_rmap.h>
#include <linux/cpumask.h>
#include <net/pkt_cls.h>
#include <linux/hwmon.h>
#include <linux/hwmon-sysfs.h>

#include "bnxt_hsi.h"
#include "bnxt.h"
#include "bnxt_ulp.h"
#include "bnxt_sriov.h"
#include "bnxt_ethtool.h"
#include "bnxt_dcb.h"
#include "bnxt_xdp.h"
#include "bnxt_vfr.h"
#include "bnxt_tc.h"
#include "bnxt_devlink.h"
#include "bnxt_debugfs.h"

#define BNXT_TX_TIMEOUT         (5 * HZ)

static const char version[] =
        "Broadcom NetXtreme-C/E driver " DRV_MODULE_NAME " v" DRV_MODULE_VERSION "\n";

MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("Broadcom BCM573xx network driver");
MODULE_VERSION(DRV_MODULE_VERSION);

#define BNXT_RX_OFFSET (NET_SKB_PAD + NET_IP_ALIGN)
#define BNXT_RX_DMA_OFFSET NET_SKB_PAD
#define BNXT_RX_COPY_THRESH 256

#define BNXT_TX_PUSH_THRESH 164

enum board_idx {
        BCM57301,
        BCM57302,
        BCM57304,
        BCM57417_NPAR,
        BCM58700,
        BCM57311,
        BCM57312,
        BCM57402,
        BCM57404,
        BCM57406,
        BCM57402_NPAR,
        BCM57407,
        BCM57412,
        BCM57414,
        BCM57416,
        BCM57417,
        BCM57412_NPAR,
        BCM57314,
        BCM57417_SFP,
        BCM57416_SFP,
        BCM57404_NPAR,
        BCM57406_NPAR,
        BCM57407_SFP,
        BCM57407_NPAR,
        BCM57414_NPAR,
        BCM57416_NPAR,
        BCM57452,
        BCM57454,
        BCM5745x_NPAR,
        BCM57508,
        BCM58802,
        BCM58804,
        BCM58808,
        NETXTREME_E_VF,
        NETXTREME_C_VF,
        NETXTREME_S_VF,
        NETXTREME_E_P5_VF,
};

/* indexed by enum above */
static const struct {
        char *name;
} board_info[] = {
        [BCM57301] = { "Broadcom BCM57301 NetXtreme-C 10Gb Ethernet" },
        [BCM57302] = { "Broadcom BCM57302 NetXtreme-C 10Gb/25Gb Ethernet" },
        [BCM57304] = { "Broadcom BCM57304 NetXtreme-C 10Gb/25Gb/40Gb/50Gb Ethernet" },
        [BCM57417_NPAR] = { "Broadcom BCM57417 NetXtreme-E Ethernet Partition" },
        [BCM58700] = { "Broadcom BCM58700 Nitro 1Gb/2.5Gb/10Gb Ethernet" },
        [BCM57311] = { "Broadcom BCM57311 NetXtreme-C 10Gb Ethernet" },
        [BCM57312] = { "Broadcom BCM57312 NetXtreme-C 10Gb/25Gb Ethernet" },
        [BCM57402] = { "Broadcom BCM57402 NetXtreme-E 10Gb Ethernet" },
        [BCM57404] = { "Broadcom BCM57404 NetXtreme-E 10Gb/25Gb Ethernet" },
        [BCM57406] = { "Broadcom BCM57406 NetXtreme-E 10GBase-T Ethernet" },
        [BCM57402_NPAR] = { "Broadcom BCM57402 NetXtreme-E Ethernet Partition" },
        [BCM57407] = { "Broadcom BCM57407 NetXtreme-E 10GBase-T Ethernet" },
        [BCM57412] = { "Broadcom BCM57412 NetXtreme-E 10Gb Ethernet" },
        [BCM57414] = { "Broadcom BCM57414 NetXtreme-E 10Gb/25Gb Ethernet" },
        [BCM57416] = { "Broadcom BCM57416 NetXtreme-E 10GBase-T Ethernet" },
        [BCM57417] = { "Broadcom BCM57417 NetXtreme-E 10GBase-T Ethernet" },
        [BCM57412_NPAR] = { "Broadcom BCM57412 NetXtreme-E Ethernet Partition" },
        [BCM57314] = { "Broadcom BCM57314 NetXtreme-C 10Gb/25Gb/40Gb/50Gb Ethernet" },
        [BCM57417_SFP] = { "Broadcom BCM57417 NetXtreme-E 10Gb/25Gb Ethernet" },
        [BCM57416_SFP] = { "Broadcom BCM57416 NetXtreme-E 10Gb Ethernet" },
        [BCM57404_NPAR] = { "Broadcom BCM57404 NetXtreme-E Ethernet Partition" },
        [BCM57406_NPAR] = { "Broadcom BCM57406 NetXtreme-E Ethernet Partition" },
        [BCM57407_SFP] = { "Broadcom BCM57407 NetXtreme-E 25Gb Ethernet" },
        [BCM57407_NPAR] = { "Broadcom BCM57407 NetXtreme-E Ethernet Partition" },
        [BCM57414_NPAR] = { "Broadcom BCM57414 NetXtreme-E Ethernet Partition" },
        [BCM57416_NPAR] = { "Broadcom BCM57416 NetXtreme-E Ethernet Partition" },
        [BCM57452] = { "Broadcom BCM57452 NetXtreme-E 10Gb/25Gb/40Gb/50Gb Ethernet" },
        [BCM57454] = { "Broadcom BCM57454 NetXtreme-E 10Gb/25Gb/40Gb/50Gb/100Gb Ethernet" },
        [BCM5745x_NPAR] = { "Broadcom BCM5745x NetXtreme-E Ethernet Partition" },
        [BCM57508] = { "Broadcom BCM57508 NetXtreme-E 10Gb/25Gb/50Gb/100Gb/200Gb Ethernet" },
        [BCM58802] = { "Broadcom BCM58802 NetXtreme-S 10Gb/25Gb/40Gb/50Gb Ethernet" },
        [BCM58804] = { "Broadcom BCM58804 NetXtreme-S 10Gb/25Gb/40Gb/50Gb/100Gb Ethernet" },
        [BCM58808] = { "Broadcom BCM58808 NetXtreme-S 10Gb/25Gb/40Gb/50Gb/100Gb Ethernet" },
        [NETXTREME_E_VF] = { "Broadcom NetXtreme-E Ethernet Virtual Function" },
        [NETXTREME_C_VF] = { "Broadcom NetXtreme-C Ethernet Virtual Function" },
        [NETXTREME_S_VF] = { "Broadcom NetXtreme-S Ethernet Virtual Function" },
        [NETXTREME_E_P5_VF] = { "Broadcom BCM5750X NetXtreme-E Ethernet Virtual Function" },
};

static const struct pci_device_id bnxt_pci_tbl[] = {
        { PCI_VDEVICE(BROADCOM, 0x1604), .driver_data = BCM5745x_NPAR },
        { PCI_VDEVICE(BROADCOM, 0x1605), .driver_data = BCM5745x_NPAR },
        { PCI_VDEVICE(BROADCOM, 0x1614), .driver_data = BCM57454 },
        { PCI_VDEVICE(BROADCOM, 0x16c0), .driver_data = BCM57417_NPAR },
        { PCI_VDEVICE(BROADCOM, 0x16c8), .driver_data = BCM57301 },
        { PCI_VDEVICE(BROADCOM, 0x16c9), .driver_data = BCM57302 },
        { PCI_VDEVICE(BROADCOM, 0x16ca), .driver_data = BCM57304 },
        { PCI_VDEVICE(BROADCOM, 0x16cc), .driver_data = BCM57417_NPAR },
        { PCI_VDEVICE(BROADCOM, 0x16cd), .driver_data = BCM58700 },
        { PCI_VDEVICE(BROADCOM, 0x16ce), .driver_data = BCM57311 },
        { PCI_VDEVICE(BROADCOM, 0x16cf), .driver_data = BCM57312 },
        { PCI_VDEVICE(BROADCOM, 0x16d0), .driver_data = BCM57402 },
        { PCI_VDEVICE(BROADCOM, 0x16d1), .driver_data = BCM57404 },
        { PCI_VDEVICE(BROADCOM, 0x16d2), .driver_data = BCM57406 },
        { PCI_VDEVICE(BROADCOM, 0x16d4), .driver_data = BCM57402_NPAR },
        { PCI_VDEVICE(BROADCOM, 0x16d5), .driver_data = BCM57407 },
        { PCI_VDEVICE(BROADCOM, 0x16d6), .driver_data = BCM57412 },
        { PCI_VDEVICE(BROADCOM, 0x16d7), .driver_data = BCM57414 },
        { PCI_VDEVICE(BROADCOM, 0x16d8), .driver_data = BCM57416 },
        { PCI_VDEVICE(BROADCOM, 0x16d9), .driver_data = BCM57417 },
        { PCI_VDEVICE(BROADCOM, 0x16de), .driver_data = BCM57412_NPAR },
        { PCI_VDEVICE(BROADCOM, 0x16df), .driver_data = BCM57314 },
        { PCI_VDEVICE(BROADCOM, 0x16e2), .driver_data = BCM57417_SFP },
        { PCI_VDEVICE(BROADCOM, 0x16e3), .driver_data = BCM57416_SFP },
        { PCI_VDEVICE(BROADCOM, 0x16e7), .driver_data = BCM57404_NPAR },
        { PCI_VDEVICE(BROADCOM, 0x16e8), .driver_data = BCM57406_NPAR },
        { PCI_VDEVICE(BROADCOM, 0x16e9), .driver_data = BCM57407_SFP },
        { PCI_VDEVICE(BROADCOM, 0x16ea), .driver_data = BCM57407_NPAR },
        { PCI_VDEVICE(BROADCOM, 0x16eb), .driver_data = BCM57412_NPAR },
        { PCI_VDEVICE(BROADCOM, 0x16ec), .driver_data = BCM57414_NPAR },
        { PCI_VDEVICE(BROADCOM, 0x16ed), .driver_data = BCM57414_NPAR },
        { PCI_VDEVICE(BROADCOM, 0x16ee), .driver_data = BCM57416_NPAR },
        { PCI_VDEVICE(BROADCOM, 0x16ef), .driver_data = BCM57416_NPAR },
        { PCI_VDEVICE(BROADCOM, 0x16f0), .driver_data = BCM58808 },
        { PCI_VDEVICE(BROADCOM, 0x16f1), .driver_data = BCM57452 },
        { PCI_VDEVICE(BROADCOM, 0x1750), .driver_data = BCM57508 },
        { PCI_VDEVICE(BROADCOM, 0xd802), .driver_data = BCM58802 },
        { PCI_VDEVICE(BROADCOM, 0xd804), .driver_data = BCM58804 },
#ifdef CONFIG_BNXT_SRIOV
        { PCI_VDEVICE(BROADCOM, 0x1606), .driver_data = NETXTREME_E_VF },
        { PCI_VDEVICE(BROADCOM, 0x1609), .driver_data = NETXTREME_E_VF },
        { PCI_VDEVICE(BROADCOM, 0x16c1), .driver_data = NETXTREME_E_VF },
        { PCI_VDEVICE(BROADCOM, 0x16cb), .driver_data = NETXTREME_C_VF },
        { PCI_VDEVICE(BROADCOM, 0x16d3), .driver_data = NETXTREME_E_VF },
        { PCI_VDEVICE(BROADCOM, 0x16dc), .driver_data = NETXTREME_E_VF },
        { PCI_VDEVICE(BROADCOM, 0x16e1), .driver_data = NETXTREME_C_VF },
        { PCI_VDEVICE(BROADCOM, 0x16e5), .driver_data = NETXTREME_C_VF },
        { PCI_VDEVICE(BROADCOM, 0x1807), .driver_data = NETXTREME_E_P5_VF },
        { PCI_VDEVICE(BROADCOM, 0xd800), .driver_data = NETXTREME_S_VF },
#endif
        { 0 }
};

MODULE_DEVICE_TABLE(pci, bnxt_pci_tbl);

static const u16 bnxt_vf_req_snif[] = {
        HWRM_FUNC_CFG,
        HWRM_FUNC_VF_CFG,
        HWRM_PORT_PHY_QCFG,
        HWRM_CFA_L2_FILTER_ALLOC,
};

static const u16 bnxt_async_events_arr[] = {
        ASYNC_EVENT_CMPL_EVENT_ID_LINK_STATUS_CHANGE,
        ASYNC_EVENT_CMPL_EVENT_ID_PF_DRVR_UNLOAD,
        ASYNC_EVENT_CMPL_EVENT_ID_PORT_CONN_NOT_ALLOWED,
        ASYNC_EVENT_CMPL_EVENT_ID_VF_CFG_CHANGE,
        ASYNC_EVENT_CMPL_EVENT_ID_LINK_SPEED_CFG_CHANGE,
};

static struct workqueue_struct *bnxt_pf_wq;

static bool bnxt_vf_pciid(enum board_idx idx)
{
        return (idx == NETXTREME_C_VF || idx == NETXTREME_E_VF ||
                idx == NETXTREME_S_VF || idx == NETXTREME_E_P5_VF);
}

#define DB_CP_REARM_FLAGS       (DB_KEY_CP | DB_IDX_VALID)
#define DB_CP_FLAGS             (DB_KEY_CP | DB_IDX_VALID | DB_IRQ_DIS)
#define DB_CP_IRQ_DIS_FLAGS     (DB_KEY_CP | DB_IRQ_DIS)

#define BNXT_CP_DB_IRQ_DIS(db)                                          \
                writel(DB_CP_IRQ_DIS_FLAGS, db)

#define BNXT_DB_CQ(db, idx)                                             \
        writel(DB_CP_FLAGS | RING_CMP(idx), (db)->doorbell)

#define BNXT_DB_NQ_P5(db, idx)                                          \
        writeq((db)->db_key64 | DBR_TYPE_NQ | RING_CMP(idx), (db)->doorbell)

#define BNXT_DB_CQ_ARM(db, idx)                                         \
        writel(DB_CP_REARM_FLAGS | RING_CMP(idx), (db)->doorbell)

#define BNXT_DB_NQ_ARM_P5(db, idx)                                      \
        writeq((db)->db_key64 | DBR_TYPE_NQ_ARM | RING_CMP(idx), (db)->doorbell)

static void bnxt_db_nq(struct bnxt *bp, struct bnxt_db_info *db, u32 idx)
{
        if (bp->flags & BNXT_FLAG_CHIP_P5)
                BNXT_DB_NQ_P5(db, idx);
        else
                BNXT_DB_CQ(db, idx);
}

static void bnxt_db_nq_arm(struct bnxt *bp, struct bnxt_db_info *db, u32 idx)
{
        if (bp->flags & BNXT_FLAG_CHIP_P5)
                BNXT_DB_NQ_ARM_P5(db, idx);
        else
                BNXT_DB_CQ_ARM(db, idx);
}

static void bnxt_db_cq(struct bnxt *bp, struct bnxt_db_info *db, u32 idx)
{
        if (bp->flags & BNXT_FLAG_CHIP_P5)
                writeq(db->db_key64 | DBR_TYPE_CQ_ARMALL | RING_CMP(idx),
                       db->doorbell);
        else
                BNXT_DB_CQ(db, idx);
}

const u16 bnxt_lhint_arr[] = {
        TX_BD_FLAGS_LHINT_512_AND_SMALLER,
        TX_BD_FLAGS_LHINT_512_TO_1023,
        TX_BD_FLAGS_LHINT_1024_TO_2047,
        TX_BD_FLAGS_LHINT_1024_TO_2047,
        TX_BD_FLAGS_LHINT_2048_AND_LARGER,
        TX_BD_FLAGS_LHINT_2048_AND_LARGER,
        TX_BD_FLAGS_LHINT_2048_AND_LARGER,
        TX_BD_FLAGS_LHINT_2048_AND_LARGER,
        TX_BD_FLAGS_LHINT_2048_AND_LARGER,
        TX_BD_FLAGS_LHINT_2048_AND_LARGER,
        TX_BD_FLAGS_LHINT_2048_AND_LARGER,
        TX_BD_FLAGS_LHINT_2048_AND_LARGER,
        TX_BD_FLAGS_LHINT_2048_AND_LARGER,
        TX_BD_FLAGS_LHINT_2048_AND_LARGER,
        TX_BD_FLAGS_LHINT_2048_AND_LARGER,
        TX_BD_FLAGS_LHINT_2048_AND_LARGER,
        TX_BD_FLAGS_LHINT_2048_AND_LARGER,
        TX_BD_FLAGS_LHINT_2048_AND_LARGER,
        TX_BD_FLAGS_LHINT_2048_AND_LARGER,
};

static u16 bnxt_xmit_get_cfa_action(struct sk_buff *skb)
{
        struct metadata_dst *md_dst = skb_metadata_dst(skb);

        if (!md_dst || md_dst->type != METADATA_HW_PORT_MUX)
                return 0;

        return md_dst->u.port_info.port_id;
}

static netdev_tx_t bnxt_start_xmit(struct sk_buff *skb, struct net_device *dev)
{
        struct bnxt *bp = netdev_priv(dev);
        struct tx_bd *txbd;
        struct tx_bd_ext *txbd1;
        struct netdev_queue *txq;
        int i;
        dma_addr_t mapping;
        unsigned int length, pad = 0;
        u32 len, free_size, vlan_tag_flags, cfa_action, flags;
        u16 prod, last_frag;
        struct pci_dev *pdev = bp->pdev;
        struct bnxt_tx_ring_info *txr;
        struct bnxt_sw_tx_bd *tx_buf;

        i = skb_get_queue_mapping(skb);
        if (unlikely(i >= bp->tx_nr_rings)) {
                dev_kfree_skb_any(skb);
                return NETDEV_TX_OK;
        }

        txq = netdev_get_tx_queue(dev, i);
        txr = &bp->tx_ring[bp->tx_ring_map[i]];
        prod = txr->tx_prod;

        free_size = bnxt_tx_avail(bp, txr);
        if (unlikely(free_size < skb_shinfo(skb)->nr_frags + 2)) {
                netif_tx_stop_queue(txq);
                return NETDEV_TX_BUSY;
        }

        length = skb->len;
        len = skb_headlen(skb);
        last_frag = skb_shinfo(skb)->nr_frags;

        txbd = &txr->tx_desc_ring[TX_RING(prod)][TX_IDX(prod)];

        txbd->tx_bd_opaque = prod;

        tx_buf = &txr->tx_buf_ring[prod];
        tx_buf->skb = skb;
        tx_buf->nr_frags = last_frag;

        vlan_tag_flags = 0;
        cfa_action = bnxt_xmit_get_cfa_action(skb);
        if (skb_vlan_tag_present(skb)) {
                vlan_tag_flags = TX_BD_CFA_META_KEY_VLAN |
                                 skb_vlan_tag_get(skb);
                /* Currently supports 8021Q, 8021AD vlan offloads
                 * QINQ1, QINQ2, QINQ3 vlan headers are deprecated
                 */
                if (skb->vlan_proto == htons(ETH_P_8021Q))
                        vlan_tag_flags |= 1 << TX_BD_CFA_META_TPID_SHIFT;
        }

        if (free_size == bp->tx_ring_size && length <= bp->tx_push_thresh) {
                struct tx_push_buffer *tx_push_buf = txr->tx_push;
                struct tx_push_bd *tx_push = &tx_push_buf->push_bd;
                struct tx_bd_ext *tx_push1 = &tx_push->txbd2;
                void __iomem *db = txr->tx_db.doorbell;
                void *pdata = tx_push_buf->data;
                u64 *end;
                int j, push_len;

                /* Set COAL_NOW to be ready quickly for the next push */
                tx_push->tx_bd_len_flags_type =
                        cpu_to_le32((length << TX_BD_LEN_SHIFT) |
                                        TX_BD_TYPE_LONG_TX_BD |
                                        TX_BD_FLAGS_LHINT_512_AND_SMALLER |
                                        TX_BD_FLAGS_COAL_NOW |
                                        TX_BD_FLAGS_PACKET_END |
                                        (2 << TX_BD_FLAGS_BD_CNT_SHIFT));

                if (skb->ip_summed == CHECKSUM_PARTIAL)
                        tx_push1->tx_bd_hsize_lflags =
                                        cpu_to_le32(TX_BD_FLAGS_TCP_UDP_CHKSUM);
                else
                        tx_push1->tx_bd_hsize_lflags = 0;

                tx_push1->tx_bd_cfa_meta = cpu_to_le32(vlan_tag_flags);
                tx_push1->tx_bd_cfa_action =
                        cpu_to_le32(cfa_action << TX_BD_CFA_ACTION_SHIFT);

                end = pdata + length;
                end = PTR_ALIGN(end, 8) - 1;
                *end = 0;

                skb_copy_from_linear_data(skb, pdata, len);
                pdata += len;
                for (j = 0; j < last_frag; j++) {
                        skb_frag_t *frag = &skb_shinfo(skb)->frags[j];
                        void *fptr;

                        fptr = skb_frag_address_safe(frag);
                        if (!fptr)
                                goto normal_tx;

                        memcpy(pdata, fptr, skb_frag_size(frag));
                        pdata += skb_frag_size(frag);
                }

                txbd->tx_bd_len_flags_type = tx_push->tx_bd_len_flags_type;
                txbd->tx_bd_haddr = txr->data_mapping;
                prod = NEXT_TX(prod);
                txbd = &txr->tx_desc_ring[TX_RING(prod)][TX_IDX(prod)];
                memcpy(txbd, tx_push1, sizeof(*txbd));
                prod = NEXT_TX(prod);
                tx_push->doorbell =
                        cpu_to_le32(DB_KEY_TX_PUSH | DB_LONG_TX_PUSH | prod);
                txr->tx_prod = prod;

                tx_buf->is_push = 1;
                netdev_tx_sent_queue(txq, skb->len);
                wmb();  /* Sync is_push and byte queue before pushing data */

                push_len = (length + sizeof(*tx_push) + 7) / 8;
                if (push_len > 16) {
                        __iowrite64_copy(db, tx_push_buf, 16);
                        __iowrite32_copy(db + 4, tx_push_buf + 1,
                                         (push_len - 16) << 1);
                } else {
                        __iowrite64_copy(db, tx_push_buf, push_len);
                }

                goto tx_done;
        }

normal_tx:
        if (length < BNXT_MIN_PKT_SIZE) {
                pad = BNXT_MIN_PKT_SIZE - length;
                if (skb_pad(skb, pad)) {
                        /* SKB already freed. */
                        tx_buf->skb = NULL;
                        return NETDEV_TX_OK;
                }
                length = BNXT_MIN_PKT_SIZE;
        }

        mapping = dma_map_single(&pdev->dev, skb->data, len, DMA_TO_DEVICE);

        if (unlikely(dma_mapping_error(&pdev->dev, mapping))) {
                dev_kfree_skb_any(skb);
                tx_buf->skb = NULL;
                return NETDEV_TX_OK;
        }

        dma_unmap_addr_set(tx_buf, mapping, mapping);
        flags = (len << TX_BD_LEN_SHIFT) | TX_BD_TYPE_LONG_TX_BD |
                ((last_frag + 2) << TX_BD_FLAGS_BD_CNT_SHIFT);

        txbd->tx_bd_haddr = cpu_to_le64(mapping);

        prod = NEXT_TX(prod);
        txbd1 = (struct tx_bd_ext *)
                &txr->tx_desc_ring[TX_RING(prod)][TX_IDX(prod)];

        txbd1->tx_bd_hsize_lflags = 0;
        if (skb_is_gso(skb)) {
                u32 hdr_len;

                if (skb->encapsulation)
                        hdr_len = skb_inner_network_offset(skb) +
                                skb_inner_network_header_len(skb) +
                                inner_tcp_hdrlen(skb);
                else
                        hdr_len = skb_transport_offset(skb) +
                                tcp_hdrlen(skb);

                txbd1->tx_bd_hsize_lflags = cpu_to_le32(TX_BD_FLAGS_LSO |
                                        TX_BD_FLAGS_T_IPID |
                                        (hdr_len << (TX_BD_HSIZE_SHIFT - 1)));
                length = skb_shinfo(skb)->gso_size;
                txbd1->tx_bd_mss = cpu_to_le32(length);
                length += hdr_len;
        } else if (skb->ip_summed == CHECKSUM_PARTIAL) {
                txbd1->tx_bd_hsize_lflags =
                        cpu_to_le32(TX_BD_FLAGS_TCP_UDP_CHKSUM);
                txbd1->tx_bd_mss = 0;
        }

        length >>= 9;
        if (unlikely(length >= ARRAY_SIZE(bnxt_lhint_arr))) {
                dev_warn_ratelimited(&pdev->dev, "Dropped oversize %d bytes TX packet.\n",
                                     skb->len);
                i = 0;
                goto tx_dma_error;
        }
        flags |= bnxt_lhint_arr[length];
        txbd->tx_bd_len_flags_type = cpu_to_le32(flags);

        txbd1->tx_bd_cfa_meta = cpu_to_le32(vlan_tag_flags);
        txbd1->tx_bd_cfa_action =
                        cpu_to_le32(cfa_action << TX_BD_CFA_ACTION_SHIFT);
        for (i = 0; i < last_frag; i++) {
                skb_frag_t *frag = &skb_shinfo(skb)->frags[i];

                prod = NEXT_TX(prod);
                txbd = &txr->tx_desc_ring[TX_RING(prod)][TX_IDX(prod)];

                len = skb_frag_size(frag);
                mapping = skb_frag_dma_map(&pdev->dev, frag, 0, len,
                                           DMA_TO_DEVICE);

                if (unlikely(dma_mapping_error(&pdev->dev, mapping)))
                        goto tx_dma_error;

                tx_buf = &txr->tx_buf_ring[prod];
                dma_unmap_addr_set(tx_buf, mapping, mapping);

                txbd->tx_bd_haddr = cpu_to_le64(mapping);

                flags = len << TX_BD_LEN_SHIFT;
                txbd->tx_bd_len_flags_type = cpu_to_le32(flags);
        }

        flags &= ~TX_BD_LEN;
        txbd->tx_bd_len_flags_type =
                cpu_to_le32(((len + pad) << TX_BD_LEN_SHIFT) | flags |
                            TX_BD_FLAGS_PACKET_END);

        netdev_tx_sent_queue(txq, skb->len);

        /* Sync BD data before updating doorbell */
        wmb();

        prod = NEXT_TX(prod);
        txr->tx_prod = prod;

        if (!skb->xmit_more || netif_xmit_stopped(txq))
                bnxt_db_write(bp, &txr->tx_db, prod);

tx_done:

        mmiowb();

        if (unlikely(bnxt_tx_avail(bp, txr) <= MAX_SKB_FRAGS + 1)) {
                if (skb->xmit_more && !tx_buf->is_push)
                        bnxt_db_write(bp, &txr->tx_db, prod);

                netif_tx_stop_queue(txq);

                /* netif_tx_stop_queue() must be done before checking
                 * tx index in bnxt_tx_avail() below, because in
                 * bnxt_tx_int(), we update tx index before checking for
                 * netif_tx_queue_stopped().
                 */
                smp_mb();
                if (bnxt_tx_avail(bp, txr) > bp->tx_wake_thresh)
                        netif_tx_wake_queue(txq);
        }
        return NETDEV_TX_OK;

tx_dma_error:
        last_frag = i;

        /* start back at beginning and unmap skb */
        prod = txr->tx_prod;
        tx_buf = &txr->tx_buf_ring[prod];
        tx_buf->skb = NULL;
        dma_unmap_single(&pdev->dev, dma_unmap_addr(tx_buf, mapping),
                         skb_headlen(skb), PCI_DMA_TODEVICE);
        prod = NEXT_TX(prod);

        /* unmap remaining mapped pages */
        for (i = 0; i < last_frag; i++) {
                prod = NEXT_TX(prod);
                tx_buf = &txr->tx_buf_ring[prod];
                dma_unmap_page(&pdev->dev, dma_unmap_addr(tx_buf, mapping),
                               skb_frag_size(&skb_shinfo(skb)->frags[i]),
                               PCI_DMA_TODEVICE);
        }

        dev_kfree_skb_any(skb);
        return NETDEV_TX_OK;
}

static void bnxt_tx_int(struct bnxt *bp, struct bnxt_napi *bnapi, int nr_pkts)
{
        struct bnxt_tx_ring_info *txr = bnapi->tx_ring;
        struct netdev_queue *txq = netdev_get_tx_queue(bp->dev, txr->txq_index);
        u16 cons = txr->tx_cons;
        struct pci_dev *pdev = bp->pdev;
        int i;
        unsigned int tx_bytes = 0;

        for (i = 0; i < nr_pkts; i++) {
                struct bnxt_sw_tx_bd *tx_buf;
                struct sk_buff *skb;
                int j, last;

                tx_buf = &txr->tx_buf_ring[cons];
                cons = NEXT_TX(cons);
                skb = tx_buf->skb;
                tx_buf->skb = NULL;

                if (tx_buf->is_push) {
                        tx_buf->is_push = 0;
                        goto next_tx_int;
                }

                dma_unmap_single(&pdev->dev, dma_unmap_addr(tx_buf, mapping),
                                 skb_headlen(skb), PCI_DMA_TODEVICE);
                last = tx_buf->nr_frags;

                for (j = 0; j < last; j++) {
                        cons = NEXT_TX(cons);
                        tx_buf = &txr->tx_buf_ring[cons];
                        dma_unmap_page(
                                &pdev->dev,
                                dma_unmap_addr(tx_buf, mapping),
                                skb_frag_size(&skb_shinfo(skb)->frags[j]),
                                PCI_DMA_TODEVICE);
                }

next_tx_int:
                cons = NEXT_TX(cons);

                tx_bytes += skb->len;
                dev_kfree_skb_any(skb);
        }

        netdev_tx_completed_queue(txq, nr_pkts, tx_bytes);
        txr->tx_cons = cons;

        /* Need to make the tx_cons update visible to bnxt_start_xmit()
         * before checking for netif_tx_queue_stopped().  Without the
         * memory barrier, there is a small possibility that bnxt_start_xmit()
         * will miss it and cause the queue to be stopped forever.
         */
        smp_mb();

        if (unlikely(netif_tx_queue_stopped(txq)) &&
            (bnxt_tx_avail(bp, txr) > bp->tx_wake_thresh)) {
                __netif_tx_lock(txq, smp_processor_id());
                if (netif_tx_queue_stopped(txq) &&
                    bnxt_tx_avail(bp, txr) > bp->tx_wake_thresh &&
                    txr->dev_state != BNXT_DEV_STATE_CLOSING)
                        netif_tx_wake_queue(txq);
                __netif_tx_unlock(txq);
        }
}

static struct page *__bnxt_alloc_rx_page(struct bnxt *bp, dma_addr_t *mapping,
                                         gfp_t gfp)
{
        struct device *dev = &bp->pdev->dev;
        struct page *page;

        page = alloc_page(gfp);
        if (!page)
                return NULL;

        *mapping = dma_map_page_attrs(dev, page, 0, PAGE_SIZE, bp->rx_dir,
                                      DMA_ATTR_WEAK_ORDERING);
        if (dma_mapping_error(dev, *mapping)) {
                __free_page(page);
                return NULL;
        }
        *mapping += bp->rx_dma_offset;
        return page;
}

static inline u8 *__bnxt_alloc_rx_data(struct bnxt *bp, dma_addr_t *mapping,
                                       gfp_t gfp)
{
        u8 *data;
        struct pci_dev *pdev = bp->pdev;

        data = kmalloc(bp->rx_buf_size, gfp);
        if (!data)
                return NULL;

        *mapping = dma_map_single_attrs(&pdev->dev, data + bp->rx_dma_offset,
                                        bp->rx_buf_use_size, bp->rx_dir,
                                        DMA_ATTR_WEAK_ORDERING);

        if (dma_mapping_error(&pdev->dev, *mapping)) {
                kfree(data);
                data = NULL;
        }
        return data;
}

int bnxt_alloc_rx_data(struct bnxt *bp, struct bnxt_rx_ring_info *rxr,
                       u16 prod, gfp_t gfp)
{
        struct rx_bd *rxbd = &rxr->rx_desc_ring[RX_RING(prod)][RX_IDX(prod)];
        struct bnxt_sw_rx_bd *rx_buf = &rxr->rx_buf_ring[prod];
        dma_addr_t mapping;

        if (BNXT_RX_PAGE_MODE(bp)) {
                struct page *page = __bnxt_alloc_rx_page(bp, &mapping, gfp);

                if (!page)
                        return -ENOMEM;

                rx_buf->data = page;
                rx_buf->data_ptr = page_address(page) + bp->rx_offset;
        } else {
                u8 *data = __bnxt_alloc_rx_data(bp, &mapping, gfp);

                if (!data)
                        return -ENOMEM;

                rx_buf->data = data;
                rx_buf->data_ptr = data + bp->rx_offset;
        }
        rx_buf->mapping = mapping;

        rxbd->rx_bd_haddr = cpu_to_le64(mapping);
        return 0;
}

void bnxt_reuse_rx_data(struct bnxt_rx_ring_info *rxr, u16 cons, void *data)
{
        u16 prod = rxr->rx_prod;
        struct bnxt_sw_rx_bd *cons_rx_buf, *prod_rx_buf;
        struct rx_bd *cons_bd, *prod_bd;

        prod_rx_buf = &rxr->rx_buf_ring[prod];
        cons_rx_buf = &rxr->rx_buf_ring[cons];

        prod_rx_buf->data = data;
        prod_rx_buf->data_ptr = cons_rx_buf->data_ptr;

        prod_rx_buf->mapping = cons_rx_buf->mapping;

        prod_bd = &rxr->rx_desc_ring[RX_RING(prod)][RX_IDX(prod)];
        cons_bd = &rxr->rx_desc_ring[RX_RING(cons)][RX_IDX(cons)];

        prod_bd->rx_bd_haddr = cons_bd->rx_bd_haddr;
}

static inline u16 bnxt_find_next_agg_idx(struct bnxt_rx_ring_info *rxr, u16 idx)
{
        u16 next, max = rxr->rx_agg_bmap_size;

        next = find_next_zero_bit(rxr->rx_agg_bmap, max, idx);
        if (next >= max)
                next = find_first_zero_bit(rxr->rx_agg_bmap, max);
        return next;
}

static inline int bnxt_alloc_rx_page(struct bnxt *bp,
                                     struct bnxt_rx_ring_info *rxr,
                                     u16 prod, gfp_t gfp)
{
        struct rx_bd *rxbd =
                &rxr->rx_agg_desc_ring[RX_RING(prod)][RX_IDX(prod)];
        struct bnxt_sw_rx_agg_bd *rx_agg_buf;
        struct pci_dev *pdev = bp->pdev;
        struct page *page;
        dma_addr_t mapping;
        u16 sw_prod = rxr->rx_sw_agg_prod;
        unsigned int offset = 0;

        if (PAGE_SIZE > BNXT_RX_PAGE_SIZE) {
                page = rxr->rx_page;
                if (!page) {
                        page = alloc_page(gfp);
                        if (!page)
                                return -ENOMEM;
                        rxr->rx_page = page;
                        rxr->rx_page_offset = 0;
                }
                offset = rxr->rx_page_offset;
                rxr->rx_page_offset += BNXT_RX_PAGE_SIZE;
                if (rxr->rx_page_offset == PAGE_SIZE)
                        rxr->rx_page = NULL;
                else
                        get_page(page);
        } else {
                page = alloc_page(gfp);
                if (!page)
                        return -ENOMEM;
        }

        mapping = dma_map_page_attrs(&pdev->dev, page, offset,
                                     BNXT_RX_PAGE_SIZE, PCI_DMA_FROMDEVICE,
                                     DMA_ATTR_WEAK_ORDERING);
        if (dma_mapping_error(&pdev->dev, mapping)) {
                __free_page(page);
                return -EIO;
        }

        if (unlikely(test_bit(sw_prod, rxr->rx_agg_bmap)))
                sw_prod = bnxt_find_next_agg_idx(rxr, sw_prod);

        __set_bit(sw_prod, rxr->rx_agg_bmap);
        rx_agg_buf = &rxr->rx_agg_ring[sw_prod];
        rxr->rx_sw_agg_prod = NEXT_RX_AGG(sw_prod);

        rx_agg_buf->page = page;
        rx_agg_buf->offset = offset;
        rx_agg_buf->mapping = mapping;
        rxbd->rx_bd_haddr = cpu_to_le64(mapping);
        rxbd->rx_bd_opaque = sw_prod;
        return 0;
}

static void bnxt_reuse_rx_agg_bufs(struct bnxt_cp_ring_info *cpr, u16 cp_cons,
                                   u32 agg_bufs)
{
        struct bnxt_napi *bnapi = cpr->bnapi;
        struct bnxt *bp = bnapi->bp;
        struct bnxt_rx_ring_info *rxr = bnapi->rx_ring;
        u16 prod = rxr->rx_agg_prod;
        u16 sw_prod = rxr->rx_sw_agg_prod;
        u32 i;

        for (i = 0; i < agg_bufs; i++) {
                u16 cons;
                struct rx_agg_cmp *agg;
                struct bnxt_sw_rx_agg_bd *cons_rx_buf, *prod_rx_buf;
                struct rx_bd *prod_bd;
                struct page *page;

                agg = (struct rx_agg_cmp *)
                        &cpr->cp_desc_ring[CP_RING(cp_cons)][CP_IDX(cp_cons)];
                cons = agg->rx_agg_cmp_opaque;
                __clear_bit(cons, rxr->rx_agg_bmap);

                if (unlikely(test_bit(sw_prod, rxr->rx_agg_bmap)))
                        sw_prod = bnxt_find_next_agg_idx(rxr, sw_prod);

                __set_bit(sw_prod, rxr->rx_agg_bmap);
                prod_rx_buf = &rxr->rx_agg_ring[sw_prod];
                cons_rx_buf = &rxr->rx_agg_ring[cons];

                /* It is possible for sw_prod to be equal to cons, so
                 * set cons_rx_buf->page to NULL first.
                 */
                page = cons_rx_buf->page;
                cons_rx_buf->page = NULL;
                prod_rx_buf->page = page;
                prod_rx_buf->offset = cons_rx_buf->offset;

                prod_rx_buf->mapping = cons_rx_buf->mapping;

                prod_bd = &rxr->rx_agg_desc_ring[RX_RING(prod)][RX_IDX(prod)];

                prod_bd->rx_bd_haddr = cpu_to_le64(cons_rx_buf->mapping);
                prod_bd->rx_bd_opaque = sw_prod;

                prod = NEXT_RX_AGG(prod);
                sw_prod = NEXT_RX_AGG(sw_prod);
                cp_cons = NEXT_CMP(cp_cons);
        }
        rxr->rx_agg_prod = prod;
        rxr->rx_sw_agg_prod = sw_prod;
}

static struct sk_buff *bnxt_rx_page_skb(struct bnxt *bp,
                                        struct bnxt_rx_ring_info *rxr,
                                        u16 cons, void *data, u8 *data_ptr,
                                        dma_addr_t dma_addr,
                                        unsigned int offset_and_len)
{
        unsigned int payload = offset_and_len >> 16;
        unsigned int len = offset_and_len & 0xffff;
        struct skb_frag_struct *frag;
        struct page *page = data;
        u16 prod = rxr->rx_prod;
        struct sk_buff *skb;
        int off, err;

        err = bnxt_alloc_rx_data(bp, rxr, prod, GFP_ATOMIC);
        if (unlikely(err)) {
                bnxt_reuse_rx_data(rxr, cons, data);
                return NULL;
        }
        dma_addr -= bp->rx_dma_offset;
        dma_unmap_page_attrs(&bp->pdev->dev, dma_addr, PAGE_SIZE, bp->rx_dir,
                             DMA_ATTR_WEAK_ORDERING);

        if (unlikely(!payload))
                payload = eth_get_headlen(data_ptr, len);

        skb = napi_alloc_skb(&rxr->bnapi->napi, payload);
        if (!skb) {
                __free_page(page);
                return NULL;
        }

        off = (void *)data_ptr - page_address(page);
        skb_add_rx_frag(skb, 0, page, off, len, PAGE_SIZE);
        memcpy(skb->data - NET_IP_ALIGN, data_ptr - NET_IP_ALIGN,
               payload + NET_IP_ALIGN);

        frag = &skb_shinfo(skb)->frags[0];
        skb_frag_size_sub(frag, payload);
        frag->page_offset += payload;
        skb->data_len -= payload;
        skb->tail += payload;

        return skb;
}

static struct sk_buff *bnxt_rx_skb(struct bnxt *bp,
                                   struct bnxt_rx_ring_info *rxr, u16 cons,
                                   void *data, u8 *data_ptr,
                                   dma_addr_t dma_addr,
                                   unsigned int offset_and_len)
{
        u16 prod = rxr->rx_prod;
        struct sk_buff *skb;
        int err;

        err = bnxt_alloc_rx_data(bp, rxr, prod, GFP_ATOMIC);
        if (unlikely(err)) {
                bnxt_reuse_rx_data(rxr, cons, data);
                return NULL;
        }

        skb = build_skb(data, 0);
        dma_unmap_single_attrs(&bp->pdev->dev, dma_addr, bp->rx_buf_use_size,
                               bp->rx_dir, DMA_ATTR_WEAK_ORDERING);
        if (!skb) {
                kfree(data);
                return NULL;
        }

        skb_reserve(skb, bp->rx_offset);
        skb_put(skb, offset_and_len & 0xffff);
        return skb;
}

static struct sk_buff *bnxt_rx_pages(struct bnxt *bp,
                                     struct bnxt_cp_ring_info *cpr,
                                     struct sk_buff *skb, u16 cp_cons,
                                     u32 agg_bufs)
{
        struct bnxt_napi *bnapi = cpr->bnapi;
        struct pci_dev *pdev = bp->pdev;
        struct bnxt_rx_ring_info *rxr = bnapi->rx_ring;
        u16 prod = rxr->rx_agg_prod;
        u32 i;

        for (i = 0; i < agg_bufs; i++) {
                u16 cons, frag_len;
                struct rx_agg_cmp *agg;
                struct bnxt_sw_rx_agg_bd *cons_rx_buf;
                struct page *page;
                dma_addr_t mapping;

                agg = (struct rx_agg_cmp *)
                        &cpr->cp_desc_ring[CP_RING(cp_cons)][CP_IDX(cp_cons)];
                cons = agg->rx_agg_cmp_opaque;
                frag_len = (le32_to_cpu(agg->rx_agg_cmp_len_flags_type) &
                            RX_AGG_CMP_LEN) >> RX_AGG_CMP_LEN_SHIFT;

                cons_rx_buf = &rxr->rx_agg_ring[cons];
                skb_fill_page_desc(skb, i, cons_rx_buf->page,
                                   cons_rx_buf->offset, frag_len);
                __clear_bit(cons, rxr->rx_agg_bmap);

                /* It is possible for bnxt_alloc_rx_page() to allocate
                 * a sw_prod index that equals the cons index, so we
                 * need to clear the cons entry now.
                 */
                mapping = cons_rx_buf->mapping;
                page = cons_rx_buf->page;
                cons_rx_buf->page = NULL;

                if (bnxt_alloc_rx_page(bp, rxr, prod, GFP_ATOMIC) != 0) {
                        struct skb_shared_info *shinfo;
                        unsigned int nr_frags;

                        shinfo = skb_shinfo(skb);
                        nr_frags = --shinfo->nr_frags;
                        __skb_frag_set_page(&shinfo->frags[nr_frags], NULL);

                        dev_kfree_skb(skb);

                        cons_rx_buf->page = page;

                        /* Update prod since possibly some pages have been
                         * allocated already.
                         */

                        rxr->rx_agg_prod = prod;
                        bnxt_reuse_rx_agg_bufs(cpr, cp_cons, agg_bufs - i);
                        return NULL;
                }

                dma_unmap_page_attrs(&pdev->dev, mapping, BNXT_RX_PAGE_SIZE,
                                     PCI_DMA_FROMDEVICE,
                                     DMA_ATTR_WEAK_ORDERING);

                skb->data_len += frag_len;
                skb->len += frag_len;
                skb->truesize += PAGE_SIZE;

                prod = NEXT_RX_AGG(prod);
                cp_cons = NEXT_CMP(cp_cons);
        }
        rxr->rx_agg_prod = prod;
        return skb;
}

static int bnxt_agg_bufs_valid(struct bnxt *bp, struct bnxt_cp_ring_info *cpr,
                               u8 agg_bufs, u32 *raw_cons)
{
        u16 last;
        struct rx_agg_cmp *agg;

        *raw_cons = ADV_RAW_CMP(*raw_cons, agg_bufs);
        last = RING_CMP(*raw_cons);
        agg = (struct rx_agg_cmp *)
                &cpr->cp_desc_ring[CP_RING(last)][CP_IDX(last)];
        return RX_AGG_CMP_VALID(agg, *raw_cons);
}

static inline struct sk_buff *bnxt_copy_skb(struct bnxt_napi *bnapi, u8 *data,
                                            unsigned int len,
                                            dma_addr_t mapping)
{
        struct bnxt *bp = bnapi->bp;
        struct pci_dev *pdev = bp->pdev;
        struct sk_buff *skb;

        skb = napi_alloc_skb(&bnapi->napi, len);
        if (!skb)
                return NULL;

        dma_sync_single_for_cpu(&pdev->dev, mapping, bp->rx_copy_thresh,
                                bp->rx_dir);

        memcpy(skb->data - NET_IP_ALIGN, data - NET_IP_ALIGN,
               len + NET_IP_ALIGN);

        dma_sync_single_for_device(&pdev->dev, mapping, bp->rx_copy_thresh,
                                   bp->rx_dir);

        skb_put(skb, len);
        return skb;
}

static int bnxt_discard_rx(struct bnxt *bp, struct bnxt_cp_ring_info *cpr,
                           u32 *raw_cons, void *cmp)
{
        struct rx_cmp *rxcmp = cmp;
        u32 tmp_raw_cons = *raw_cons;
        u8 cmp_type, agg_bufs = 0;

        cmp_type = RX_CMP_TYPE(rxcmp);

        if (cmp_type == CMP_TYPE_RX_L2_CMP) {
                agg_bufs = (le32_to_cpu(rxcmp->rx_cmp_misc_v1) &
                            RX_CMP_AGG_BUFS) >>
                           RX_CMP_AGG_BUFS_SHIFT;
        } else if (cmp_type == CMP_TYPE_RX_L2_TPA_END_CMP) {
                struct rx_tpa_end_cmp *tpa_end = cmp;

                agg_bufs = (le32_to_cpu(tpa_end->rx_tpa_end_cmp_misc_v1) &
                            RX_TPA_END_CMP_AGG_BUFS) >>
                           RX_TPA_END_CMP_AGG_BUFS_SHIFT;
        }

        if (agg_bufs) {
                if (!bnxt_agg_bufs_valid(bp, cpr, agg_bufs, &tmp_raw_cons))
                        return -EBUSY;
        }
        *raw_cons = tmp_raw_cons;
        return 0;
}

static void bnxt_queue_sp_work(struct bnxt *bp)
{
        if (BNXT_PF(bp))
                queue_work(bnxt_pf_wq, &bp->sp_task);
        else
                schedule_work(&bp->sp_task);
}

static void bnxt_cancel_sp_work(struct bnxt *bp)
{
        if (BNXT_PF(bp))
                flush_workqueue(bnxt_pf_wq);
        else
                cancel_work_sync(&bp->sp_task);
}

static void bnxt_sched_reset(struct bnxt *bp, struct bnxt_rx_ring_info *rxr)
{
        if (!rxr->bnapi->in_reset) {
                rxr->bnapi->in_reset = true;
                set_bit(BNXT_RESET_TASK_SP_EVENT, &bp->sp_event);
                bnxt_queue_sp_work(bp);
        }
        rxr->rx_next_cons = 0xffff;
}

static void bnxt_tpa_start(struct bnxt *bp, struct bnxt_rx_ring_info *rxr,
                           struct rx_tpa_start_cmp *tpa_start,
                           struct rx_tpa_start_cmp_ext *tpa_start1)
{
        u8 agg_id = TPA_START_AGG_ID(tpa_start);
        u16 cons, prod;
        struct bnxt_tpa_info *tpa_info;
        struct bnxt_sw_rx_bd *cons_rx_buf, *prod_rx_buf;
        struct rx_bd *prod_bd;
        dma_addr_t mapping;

        cons = tpa_start->rx_tpa_start_cmp_opaque;
        prod = rxr->rx_prod;
        cons_rx_buf = &rxr->rx_buf_ring[cons];
        prod_rx_buf = &rxr->rx_buf_ring[prod];
        tpa_info = &rxr->rx_tpa[agg_id];

        if (unlikely(cons != rxr->rx_next_cons)) {
                bnxt_sched_reset(bp, rxr);
                return;
        }
        /* Store cfa_code in tpa_info to use in tpa_end
         * completion processing.
         */
        tpa_info->cfa_code = TPA_START_CFA_CODE(tpa_start1);
        prod_rx_buf->data = tpa_info->data;
        prod_rx_buf->data_ptr = tpa_info->data_ptr;

        mapping = tpa_info->mapping;
        prod_rx_buf->mapping = mapping;

        prod_bd = &rxr->rx_desc_ring[RX_RING(prod)][RX_IDX(prod)];

        prod_bd->rx_bd_haddr = cpu_to_le64(mapping);

        tpa_info->data = cons_rx_buf->data;
        tpa_info->data_ptr = cons_rx_buf->data_ptr;
        cons_rx_buf->data = NULL;
        tpa_info->mapping = cons_rx_buf->mapping;

        tpa_info->len =
                le32_to_cpu(tpa_start->rx_tpa_start_cmp_len_flags_type) >>
                                RX_TPA_START_CMP_LEN_SHIFT;
        if (likely(TPA_START_HASH_VALID(tpa_start))) {
                u32 hash_type = TPA_START_HASH_TYPE(tpa_start);

                tpa_info->hash_type = PKT_HASH_TYPE_L4;
                tpa_info->gso_type = SKB_GSO_TCPV4;
                /* RSS profiles 1 and 3 with extract code 0 for inner 4-tuple */
                if (hash_type == 3 || TPA_START_IS_IPV6(tpa_start1))
                        tpa_info->gso_type = SKB_GSO_TCPV6;
                tpa_info->rss_hash =
                        le32_to_cpu(tpa_start->rx_tpa_start_cmp_rss_hash);
        } else {
                tpa_info->hash_type = PKT_HASH_TYPE_NONE;
                tpa_info->gso_type = 0;
                if (netif_msg_rx_err(bp))
                        netdev_warn(bp->dev, "TPA packet without valid hash\n");
        }
        tpa_info->flags2 = le32_to_cpu(tpa_start1->rx_tpa_start_cmp_flags2);
        tpa_info->metadata = le32_to_cpu(tpa_start1->rx_tpa_start_cmp_metadata);
        tpa_info->hdr_info = le32_to_cpu(tpa_start1->rx_tpa_start_cmp_hdr_info);

        rxr->rx_prod = NEXT_RX(prod);
        cons = NEXT_RX(cons);
        rxr->rx_next_cons = NEXT_RX(cons);
        cons_rx_buf = &rxr->rx_buf_ring[cons];

        bnxt_reuse_rx_data(rxr, cons, cons_rx_buf->data);
        rxr->rx_prod = NEXT_RX(rxr->rx_prod);
        cons_rx_buf->data = NULL;
}

static void bnxt_abort_tpa(struct bnxt_cp_ring_info *cpr, u16 cp_cons,
                           u32 agg_bufs)
{
        if (agg_bufs)
                bnxt_reuse_rx_agg_bufs(cpr, cp_cons, agg_bufs);
}

static struct sk_buff *bnxt_gro_func_5731x(struct bnxt_tpa_info *tpa_info,
                                           int payload_off, int tcp_ts,
                                           struct sk_buff *skb)
{
#ifdef CONFIG_INET
        struct tcphdr *th;
        int len, nw_off;
        u16 outer_ip_off, inner_ip_off, inner_mac_off;
        u32 hdr_info = tpa_info->hdr_info;
        bool loopback = false;

        inner_ip_off = BNXT_TPA_INNER_L3_OFF(hdr_info);
        inner_mac_off = BNXT_TPA_INNER_L2_OFF(hdr_info);
        outer_ip_off = BNXT_TPA_OUTER_L3_OFF(hdr_info);

        /* If the packet is an internal loopback packet, the offsets will
         * have an extra 4 bytes.
         */
        if (inner_mac_off == 4) {
                loopback = true;
        } else if (inner_mac_off > 4) {
                __be16 proto = *((__be16 *)(skb->data + inner_ip_off -
                                            ETH_HLEN - 2));

                /* We only support inner iPv4/ipv6.  If we don't see the
                 * correct protocol ID, it must be a loopback packet where
                 * the offsets are off by 4.
                 */
                if (proto != htons(ETH_P_IP) && proto != htons(ETH_P_IPV6))
                        loopback = true;
        }
        if (loopback) {
                /* internal loopback packet, subtract all offsets by 4 */
                inner_ip_off -= 4;
                inner_mac_off -= 4;
                outer_ip_off -= 4;
        }

        nw_off = inner_ip_off - ETH_HLEN;
        skb_set_network_header(skb, nw_off);
        if (tpa_info->flags2 & RX_TPA_START_CMP_FLAGS2_IP_TYPE) {
                struct ipv6hdr *iph = ipv6_hdr(skb);

                skb_set_transport_header(skb, nw_off + sizeof(struct ipv6hdr));
                len = skb->len - skb_transport_offset(skb);
                th = tcp_hdr(skb);
                th->check = ~tcp_v6_check(len, &iph->saddr, &iph->daddr, 0);
        } else {
                struct iphdr *iph = ip_hdr(skb);

                skb_set_transport_header(skb, nw_off + sizeof(struct iphdr));
                len = skb->len - skb_transport_offset(skb);
                th = tcp_hdr(skb);
                th->check = ~tcp_v4_check(len, iph->saddr, iph->daddr, 0);
        }

        if (inner_mac_off) { /* tunnel */
                struct udphdr *uh = NULL;
                __be16 proto = *((__be16 *)(skb->data + outer_ip_off -
                                            ETH_HLEN - 2));

                if (proto == htons(ETH_P_IP)) {
                        struct iphdr *iph = (struct iphdr *)skb->data;

                        if (iph->protocol == IPPROTO_UDP)
                                uh = (struct udphdr *)(iph + 1);
                } else {
                        struct ipv6hdr *iph = (struct ipv6hdr *)skb->data;

                        if (iph->nexthdr == IPPROTO_UDP)
                                uh = (struct udphdr *)(iph + 1);
                }
                if (uh) {
                        if (uh->check)
                                skb_shinfo(skb)->gso_type |=
                                        SKB_GSO_UDP_TUNNEL_CSUM;
                        else
                                skb_shinfo(skb)->gso_type |= SKB_GSO_UDP_TUNNEL;
                }
        }
#endif
        return skb;
}

#define BNXT_IPV4_HDR_SIZE      (sizeof(struct iphdr) + sizeof(struct tcphdr))
#define BNXT_IPV6_HDR_SIZE      (sizeof(struct ipv6hdr) + sizeof(struct tcphdr))

static struct sk_buff *bnxt_gro_func_5730x(struct bnxt_tpa_info *tpa_info,
                                           int payload_off, int tcp_ts,
                                           struct sk_buff *skb)
{
#ifdef CONFIG_INET
        struct tcphdr *th;
        int len, nw_off, tcp_opt_len = 0;

        if (tcp_ts)
                tcp_opt_len = 12;

        if (tpa_info->gso_type == SKB_GSO_TCPV4) {
                struct iphdr *iph;

                nw_off = payload_off - BNXT_IPV4_HDR_SIZE - tcp_opt_len -
                         ETH_HLEN;
                skb_set_network_header(skb, nw_off);
                iph = ip_hdr(skb);
                skb_set_transport_header(skb, nw_off + sizeof(struct iphdr));
                len = skb->len - skb_transport_offset(skb);
                th = tcp_hdr(skb);
                th->check = ~tcp_v4_check(len, iph->saddr, iph->daddr, 0);
        } else if (tpa_info->gso_type == SKB_GSO_TCPV6) {
                struct ipv6hdr *iph;

                nw_off = payload_off - BNXT_IPV6_HDR_SIZE - tcp_opt_len -
                         ETH_HLEN;
                skb_set_network_header(skb, nw_off);
                iph = ipv6_hdr(skb);
                skb_set_transport_header(skb, nw_off + sizeof(struct ipv6hdr));
                len = skb->len - skb_transport_offset(skb);
                th = tcp_hdr(skb);
                th->check = ~tcp_v6_check(len, &iph->saddr, &iph->daddr, 0);
        } else {
                dev_kfree_skb_any(skb);
                return NULL;
        }

        if (nw_off) { /* tunnel */
                struct udphdr *uh = NULL;

                if (skb->protocol == htons(ETH_P_IP)) {
                        struct iphdr *iph = (struct iphdr *)skb->data;

                        if (iph->protocol == IPPROTO_UDP)
                                uh = (struct udphdr *)(iph + 1);
                } else {
                        struct ipv6hdr *iph = (struct ipv6hdr *)skb->data;

                        if (iph->nexthdr == IPPROTO_UDP)
                                uh = (struct udphdr *)(iph + 1);
                }
                if (uh) {
                        if (uh->check)
                                skb_shinfo(skb)->gso_type |=
                                        SKB_GSO_UDP_TUNNEL_CSUM;
                        else
                                skb_shinfo(skb)->gso_type |= SKB_GSO_UDP_TUNNEL;
                }
        }
#endif
        return skb;
}

static inline struct sk_buff *bnxt_gro_skb(struct bnxt *bp,
                                           struct bnxt_tpa_info *tpa_info,
                                           struct rx_tpa_end_cmp *tpa_end,
                                           struct rx_tpa_end_cmp_ext *tpa_end1,
                                           struct sk_buff *skb)
{
#ifdef CONFIG_INET
        int payload_off;
        u16 segs;

        segs = TPA_END_TPA_SEGS(tpa_end);
        if (segs == 1)
                return skb;

        NAPI_GRO_CB(skb)->count = segs;
        skb_shinfo(skb)->gso_size =
                le32_to_cpu(tpa_end1->rx_tpa_end_cmp_seg_len);
        skb_shinfo(skb)->gso_type = tpa_info->gso_type;
        payload_off = (le32_to_cpu(tpa_end->rx_tpa_end_cmp_misc_v1) &
                       RX_TPA_END_CMP_PAYLOAD_OFFSET) >>
                      RX_TPA_END_CMP_PAYLOAD_OFFSET_SHIFT;
        skb = bp->gro_func(tpa_info, payload_off, TPA_END_GRO_TS(tpa_end), skb);
        if (likely(skb))
                tcp_gro_complete(skb);
#endif
        return skb;
}

/* Given the cfa_code of a received packet determine which
 * netdev (vf-rep or PF) the packet is destined to.
 */
static struct net_device *bnxt_get_pkt_dev(struct bnxt *bp, u16 cfa_code)
{
        struct net_device *dev = bnxt_get_vf_rep(bp, cfa_code);

        /* if vf-rep dev is NULL, the must belongs to the PF */
        return dev ? dev : bp->dev;
}

static inline struct sk_buff *bnxt_tpa_end(struct bnxt *bp,
                                           struct bnxt_cp_ring_info *cpr,
                                           u32 *raw_cons,
                                           struct rx_tpa_end_cmp *tpa_end,
                                           struct rx_tpa_end_cmp_ext *tpa_end1,
                                           u8 *event)
{
        struct bnxt_napi *bnapi = cpr->bnapi;
        struct bnxt_rx_ring_info *rxr = bnapi->rx_ring;
        u8 agg_id = TPA_END_AGG_ID(tpa_end);
        u8 *data_ptr, agg_bufs;
        u16 cp_cons = RING_CMP(*raw_cons);
        unsigned int len;
        struct bnxt_tpa_info *tpa_info;
        dma_addr_t mapping;
        struct sk_buff *skb;
        void *data;

        if (unlikely(bnapi->in_reset)) {
                int rc = bnxt_discard_rx(bp, cpr, raw_cons, tpa_end);

                if (rc < 0)
                        return ERR_PTR(-EBUSY);
                return NULL;
        }

        tpa_info = &rxr->rx_tpa[agg_id];
        data = tpa_info->data;
        data_ptr = tpa_info->data_ptr;
        prefetch(data_ptr);
        len = tpa_info->len;
        mapping = tpa_info->mapping;

        agg_bufs = (le32_to_cpu(tpa_end->rx_tpa_end_cmp_misc_v1) &
                    RX_TPA_END_CMP_AGG_BUFS) >> RX_TPA_END_CMP_AGG_BUFS_SHIFT;

        if (agg_bufs) {
                if (!bnxt_agg_bufs_valid(bp, cpr, agg_bufs, raw_cons))
                        return ERR_PTR(-EBUSY);

                *event |= BNXT_AGG_EVENT;
                cp_cons = NEXT_CMP(cp_cons);
        }

        if (unlikely(agg_bufs > MAX_SKB_FRAGS || TPA_END_ERRORS(tpa_end1))) {
                bnxt_abort_tpa(cpr, cp_cons, agg_bufs);
                if (agg_bufs > MAX_SKB_FRAGS)
                        netdev_warn(bp->dev, "TPA frags %d exceeded MAX_SKB_FRAGS %d\n",
                                    agg_bufs, (int)MAX_SKB_FRAGS);
                return NULL;
        }

        if (len <= bp->rx_copy_thresh) {
                skb = bnxt_copy_skb(bnapi, data_ptr, len, mapping);
                if (!skb) {
                        bnxt_abort_tpa(cpr, cp_cons, agg_bufs);
                        return NULL;
                }
        } else {
                u8 *new_data;
                dma_addr_t new_mapping;

                new_data = __bnxt_alloc_rx_data(bp, &new_mapping, GFP_ATOMIC);
                if (!new_data) {
                        bnxt_abort_tpa(cpr, cp_cons, agg_bufs);
                        return NULL;
                }

                tpa_info->data = new_data;
                tpa_info->data_ptr = new_data + bp->rx_offset;
                tpa_info->mapping = new_mapping;

                skb = build_skb(data, 0);
                dma_unmap_single_attrs(&bp->pdev->dev, mapping,
                                       bp->rx_buf_use_size, bp->rx_dir,
                                       DMA_ATTR_WEAK_ORDERING);

                if (!skb) {
                        kfree(data);
                        bnxt_abort_tpa(cpr, cp_cons, agg_bufs);
                        return NULL;
                }
                skb_reserve(skb, bp->rx_offset);
                skb_put(skb, len);
        }

        if (agg_bufs) {
                skb = bnxt_rx_pages(bp, cpr, skb, cp_cons, agg_bufs);
                if (!skb) {
                        /* Page reuse already handled by bnxt_rx_pages(). */
                        return NULL;
                }
        }

        skb->protocol =
                eth_type_trans(skb, bnxt_get_pkt_dev(bp, tpa_info->cfa_code));

        if (tpa_info->hash_type != PKT_HASH_TYPE_NONE)
                skb_set_hash(skb, tpa_info->rss_hash, tpa_info->hash_type);

        if ((tpa_info->flags2 & RX_CMP_FLAGS2_META_FORMAT_VLAN) &&
            (skb->dev->features & NETIF_F_HW_VLAN_CTAG_RX)) {
                u16 vlan_proto = tpa_info->metadata >>
                        RX_CMP_FLAGS2_METADATA_TPID_SFT;
                u16 vtag = tpa_info->metadata & RX_CMP_FLAGS2_METADATA_TCI_MASK;

                __vlan_hwaccel_put_tag(skb, htons(vlan_proto), vtag);
        }

        skb_checksum_none_assert(skb);
        if (likely(tpa_info->flags2 & RX_TPA_START_CMP_FLAGS2_L4_CS_CALC)) {
                skb->ip_summed = CHECKSUM_UNNECESSARY;
                skb->csum_level =
                        (tpa_info->flags2 & RX_CMP_FLAGS2_T_L4_CS_CALC) >> 3;
        }

        if (TPA_END_GRO(tpa_end))
                skb = bnxt_gro_skb(bp, tpa_info, tpa_end, tpa_end1, skb);

        return skb;
}

static void bnxt_deliver_skb(struct bnxt *bp, struct bnxt_napi *bnapi,
                             struct sk_buff *skb)
{
        if (skb->dev != bp->dev) {
                /* this packet belongs to a vf-rep */
                bnxt_vf_rep_rx(bp, skb);
                return;
        }
        skb_record_rx_queue(skb, bnapi->index);
        napi_gro_receive(&bnapi->napi, skb);
}

/* returns the following:
 * 1       - 1 packet successfully received
 * 0       - successful TPA_START, packet not completed yet
 * -EBUSY  - completion ring does not have all the agg buffers yet
 * -ENOMEM - packet aborted due to out of memory
 * -EIO    - packet aborted due to hw error indicated in BD
 */
static int bnxt_rx_pkt(struct bnxt *bp, struct bnxt_cp_ring_info *cpr,
                       u32 *raw_cons, u8 *event)
{
        struct bnxt_napi *bnapi = cpr->bnapi;
        struct bnxt_rx_ring_info *rxr = bnapi->rx_ring;
        struct net_device *dev = bp->dev;
        struct rx_cmp *rxcmp;
        struct rx_cmp_ext *rxcmp1;
        u32 tmp_raw_cons = *raw_cons;
        u16 cfa_code, cons, prod, cp_cons = RING_CMP(tmp_raw_cons);
        struct bnxt_sw_rx_bd *rx_buf;
        unsigned int len;
        u8 *data_ptr, agg_bufs, cmp_type;
        dma_addr_t dma_addr;
        struct sk_buff *skb;
        void *data;
        int rc = 0;
        u32 misc;

        rxcmp = (struct rx_cmp *)
                        &cpr->cp_desc_ring[CP_RING(cp_cons)][CP_IDX(cp_cons)];

        tmp_raw_cons = NEXT_RAW_CMP(tmp_raw_cons);
        cp_cons = RING_CMP(tmp_raw_cons);
        rxcmp1 = (struct rx_cmp_ext *)
                        &cpr->cp_desc_ring[CP_RING(cp_cons)][CP_IDX(cp_cons)];

        if (!RX_CMP_VALID(rxcmp1, tmp_raw_cons))
                return -EBUSY;

        cmp_type = RX_CMP_TYPE(rxcmp);

        prod = rxr->rx_prod;

        if (cmp_type == CMP_TYPE_RX_L2_TPA_START_CMP) {
                bnxt_tpa_start(bp, rxr, (struct rx_tpa_start_cmp *)rxcmp,
                               (struct rx_tpa_start_cmp_ext *)rxcmp1);

                *event |= BNXT_RX_EVENT;
                goto next_rx_no_prod_no_len;

        } else if (cmp_type == CMP_TYPE_RX_L2_TPA_END_CMP) {
                skb = bnxt_tpa_end(bp, cpr, &tmp_raw_cons,
                                   (struct rx_tpa_end_cmp *)rxcmp,
                                   (struct rx_tpa_end_cmp_ext *)rxcmp1, event);

                if (IS_ERR(skb))
                        return -EBUSY;

                rc = -ENOMEM;
                if (likely(skb)) {
                        bnxt_deliver_skb(bp, bnapi, skb);
                        rc = 1;
                }
                *event |= BNXT_RX_EVENT;
                goto next_rx_no_prod_no_len;
        }

        cons = rxcmp->rx_cmp_opaque;
        rx_buf = &rxr->rx_buf_ring[cons];
        data = rx_buf->data;
        data_ptr = rx_buf->data_ptr;
        if (unlikely(cons != rxr->rx_next_cons)) {
                int rc1 = bnxt_discard_rx(bp, cpr, raw_cons, rxcmp);

                bnxt_sched_reset(bp, rxr);
                return rc1;
        }
        prefetch(data_ptr);

        misc = le32_to_cpu(rxcmp->rx_cmp_misc_v1);
        agg_bufs = (misc & RX_CMP_AGG_BUFS) >> RX_CMP_AGG_BUFS_SHIFT;

        if (agg_bufs) {
                if (!bnxt_agg_bufs_valid(bp, cpr, agg_bufs, &tmp_raw_cons))
                        return -EBUSY;

                cp_cons = NEXT_CMP(cp_cons);
                *event |= BNXT_AGG_EVENT;
        }
        *event |= BNXT_RX_EVENT;

        rx_buf->data = NULL;
        if (rxcmp1->rx_cmp_cfa_code_errors_v2 & RX_CMP_L2_ERRORS) {
                bnxt_reuse_rx_data(rxr, cons, data);
                if (agg_bufs)
                        bnxt_reuse_rx_agg_bufs(cpr, cp_cons, agg_bufs);

                rc = -EIO;
                goto next_rx;
        }

        len = le32_to_cpu(rxcmp->rx_cmp_len_flags_type) >> RX_CMP_LEN_SHIFT;
        dma_addr = rx_buf->mapping;

        if (bnxt_rx_xdp(bp, rxr, cons, data, &data_ptr, &len, event)) {
                rc = 1;
                goto next_rx;
        }

        if (len <= bp->rx_copy_thresh) {
                skb = bnxt_copy_skb(bnapi, data_ptr, len, dma_addr);
                bnxt_reuse_rx_data(rxr, cons, data);
                if (!skb) {
                        rc = -ENOMEM;
                        goto next_rx;
                }
        } else {
                u32 payload;

                if (rx_buf->data_ptr == data_ptr)
                        payload = misc & RX_CMP_PAYLOAD_OFFSET;
                else
                        payload = 0;
                skb = bp->rx_skb_func(bp, rxr, cons, data, data_ptr, dma_addr,
                                      payload | len);
                if (!skb) {
                        rc = -ENOMEM;
                        goto next_rx;
                }
        }

        if (agg_bufs) {
                skb = bnxt_rx_pages(bp, cpr, skb, cp_cons, agg_bufs);
                if (!skb) {
                        rc = -ENOMEM;
                        goto next_rx;
                }
        }

        if (RX_CMP_HASH_VALID(rxcmp)) {
                u32 hash_type = RX_CMP_HASH_TYPE(rxcmp);
                enum pkt_hash_types type = PKT_HASH_TYPE_L4;

                /* RSS profiles 1 and 3 with extract code 0 for inner 4-tuple */
                if (hash_type != 1 && hash_type != 3)
                        type = PKT_HASH_TYPE_L3;
                skb_set_hash(skb, le32_to_cpu(rxcmp->rx_cmp_rss_hash), type);
        }

        cfa_code = RX_CMP_CFA_CODE(rxcmp1);
        skb->protocol = eth_type_trans(skb, bnxt_get_pkt_dev(bp, cfa_code));

        if ((rxcmp1->rx_cmp_flags2 &
             cpu_to_le32(RX_CMP_FLAGS2_META_FORMAT_VLAN)) &&
            (skb->dev->features & NETIF_F_HW_VLAN_CTAG_RX)) {
                u32 meta_data = le32_to_cpu(rxcmp1->rx_cmp_meta_data);
                u16 vtag = meta_data & RX_CMP_FLAGS2_METADATA_TCI_MASK;
                u16 vlan_proto = meta_data >> RX_CMP_FLAGS2_METADATA_TPID_SFT;

                __vlan_hwaccel_put_tag(skb, htons(vlan_proto), vtag);
        }

        skb_checksum_none_assert(skb);
        if (RX_CMP_L4_CS_OK(rxcmp1)) {
                if (dev->features & NETIF_F_RXCSUM) {
                        skb->ip_summed = CHECKSUM_UNNECESSARY;
                        skb->csum_level = RX_CMP_ENCAP(rxcmp1);
                }
        } else {
                if (rxcmp1->rx_cmp_cfa_code_errors_v2 & RX_CMP_L4_CS_ERR_BITS) {
                        if (dev->features & NETIF_F_RXCSUM)
                                bnapi->cp_ring.rx_l4_csum_errors++;
                }
        }

        bnxt_deliver_skb(bp, bnapi, skb);
        rc = 1;

next_rx:
        rxr->rx_prod = NEXT_RX(prod);
        rxr->rx_next_cons = NEXT_RX(cons);

        cpr->rx_packets += 1;
        cpr->rx_bytes += len;

next_rx_no_prod_no_len:
        *raw_cons = tmp_raw_cons;

        return rc;
}

/* In netpoll mode, if we are using a combined completion ring, we need to
 * discard the rx packets and recycle the buffers.
 */
static int bnxt_force_rx_discard(struct bnxt *bp,
                                 struct bnxt_cp_ring_info *cpr,
                                 u32 *raw_cons, u8 *event)
{
        u32 tmp_raw_cons = *raw_cons;
        struct rx_cmp_ext *rxcmp1;
        struct rx_cmp *rxcmp;
        u16 cp_cons;
        u8 cmp_type;

        cp_cons = RING_CMP(tmp_raw_cons);
        rxcmp = (struct rx_cmp *)
                        &cpr->cp_desc_ring[CP_RING(cp_cons)][CP_IDX(cp_cons)];

        tmp_raw_cons = NEXT_RAW_CMP(tmp_raw_cons);
        cp_cons = RING_CMP(tmp_raw_cons);
        rxcmp1 = (struct rx_cmp_ext *)
                        &cpr->cp_desc_ring[CP_RING(cp_cons)][CP_IDX(cp_cons)];

        if (!RX_CMP_VALID(rxcmp1, tmp_raw_cons))
                return -EBUSY;

        cmp_type = RX_CMP_TYPE(rxcmp);
        if (cmp_type == CMP_TYPE_RX_L2_CMP) {
                rxcmp1->rx_cmp_cfa_code_errors_v2 |=
                        cpu_to_le32(RX_CMPL_ERRORS_CRC_ERROR);
        } else if (cmp_type == CMP_TYPE_RX_L2_TPA_END_CMP) {
                struct rx_tpa_end_cmp_ext *tpa_end1;

                tpa_end1 = (struct rx_tpa_end_cmp_ext *)rxcmp1;
                tpa_end1->rx_tpa_end_cmp_errors_v2 |=
                        cpu_to_le32(RX_TPA_END_CMP_ERRORS);
        }
        return bnxt_rx_pkt(bp, cpr, raw_cons, event);
}

#define BNXT_GET_EVENT_PORT(data)       \
        ((data) &                       \
         ASYNC_EVENT_CMPL_PORT_CONN_NOT_ALLOWED_EVENT_DATA1_PORT_ID_MASK)

static int bnxt_async_event_process(struct bnxt *bp,
                                    struct hwrm_async_event_cmpl *cmpl)
{
        u16 event_id = le16_to_cpu(cmpl->event_id);

        /* TODO CHIMP_FW: Define event id's for link change, error etc */
        switch (event_id) {
        case ASYNC_EVENT_CMPL_EVENT_ID_LINK_SPEED_CFG_CHANGE: {
                u32 data1 = le32_to_cpu(cmpl->event_data1);
                struct bnxt_link_info *link_info = &bp->link_info;

                if (BNXT_VF(bp))
                        goto async_event_process_exit;

                /* print unsupported speed warning in forced speed mode only */
                if (!(link_info->autoneg & BNXT_AUTONEG_SPEED) &&
                    (data1 & 0x20000)) {
                        u16 fw_speed = link_info->force_link_speed;
                        u32 speed = bnxt_fw_to_ethtool_speed(fw_speed);

                        if (speed != SPEED_UNKNOWN)
                                netdev_warn(bp->dev, "Link speed %d no longer supported\n",
                                            speed);
                }
                set_bit(BNXT_LINK_SPEED_CHNG_SP_EVENT, &bp->sp_event);
        }
        /* fall through */
        case ASYNC_EVENT_CMPL_EVENT_ID_LINK_STATUS_CHANGE:
                set_bit(BNXT_LINK_CHNG_SP_EVENT, &bp->sp_event);
                break;
        case ASYNC_EVENT_CMPL_EVENT_ID_PF_DRVR_UNLOAD:
                set_bit(BNXT_HWRM_PF_UNLOAD_SP_EVENT, &bp->sp_event);
                break;
        case ASYNC_EVENT_CMPL_EVENT_ID_PORT_CONN_NOT_ALLOWED: {
                u32 data1 = le32_to_cpu(cmpl->event_data1);
                u16 port_id = BNXT_GET_EVENT_PORT(data1);

                if (BNXT_VF(bp))
                        break;

                if (bp->pf.port_id != port_id)
                        break;

                set_bit(BNXT_HWRM_PORT_MODULE_SP_EVENT, &bp->sp_event);
                break;
        }
        case ASYNC_EVENT_CMPL_EVENT_ID_VF_CFG_CHANGE:
                if (BNXT_PF(bp))
                        goto async_event_process_exit;
                set_bit(BNXT_RESET_TASK_SILENT_SP_EVENT, &bp->sp_event);
                break;
        default:
                goto async_event_process_exit;
        }
        bnxt_queue_sp_work(bp);
async_event_process_exit:
        bnxt_ulp_async_events(bp, cmpl);
        return 0;
}

static int bnxt_hwrm_handler(struct bnxt *bp, struct tx_cmp *txcmp)
{
        u16 cmpl_type = TX_CMP_TYPE(txcmp), vf_id, seq_id;
        struct hwrm_cmpl *h_cmpl = (struct hwrm_cmpl *)txcmp;
        struct hwrm_fwd_req_cmpl *fwd_req_cmpl =
                                (struct hwrm_fwd_req_cmpl *)txcmp;

        switch (cmpl_type) {
        case CMPL_BASE_TYPE_HWRM_DONE:
                seq_id = le16_to_cpu(h_cmpl->sequence_id);
                if (seq_id == bp->hwrm_intr_seq_id)
                        bp->hwrm_intr_seq_id = (u16)~bp->hwrm_intr_seq_id;
                else
                        netdev_err(bp->dev, "Invalid hwrm seq id %d\n", seq_id);
                break;

        case CMPL_BASE_TYPE_HWRM_FWD_REQ:
                vf_id = le16_to_cpu(fwd_req_cmpl->source_id);

                if ((vf_id < bp->pf.first_vf_id) ||
                    (vf_id >= bp->pf.first_vf_id + bp->pf.active_vfs)) {
                        netdev_err(bp->dev, "Msg contains invalid VF id %x\n",
                                   vf_id);
                        return -EINVAL;
                }

                set_bit(vf_id - bp->pf.first_vf_id, bp->pf.vf_event_bmap);
                set_bit(BNXT_HWRM_EXEC_FWD_REQ_SP_EVENT, &bp->sp_event);
                bnxt_queue_sp_work(bp);
                break;

        case CMPL_BASE_TYPE_HWRM_ASYNC_EVENT:
                bnxt_async_event_process(bp,
                                         (struct hwrm_async_event_cmpl *)txcmp);

        default:
                break;
        }

        return 0;
}

static irqreturn_t bnxt_msix(int irq, void *dev_instance)
{
        struct bnxt_napi *bnapi = dev_instance;
        struct bnxt *bp = bnapi->bp;
        struct bnxt_cp_ring_info *cpr = &bnapi->cp_ring;
        u32 cons = RING_CMP(cpr->cp_raw_cons);

        cpr->event_ctr++;
        prefetch(&cpr->cp_desc_ring[CP_RING(cons)][CP_IDX(cons)]);
        napi_schedule(&bnapi->napi);
        return IRQ_HANDLED;
}

static inline int bnxt_has_work(struct bnxt *bp, struct bnxt_cp_ring_info *cpr)
{
        u32 raw_cons = cpr->cp_raw_cons;
        u16 cons = RING_CMP(raw_cons);
        struct tx_cmp *txcmp;

        txcmp = &cpr->cp_desc_ring[CP_RING(cons)][CP_IDX(cons)];

        return TX_CMP_VALID(txcmp, raw_cons);
}

static irqreturn_t bnxt_inta(int irq, void *dev_instance)
{
        struct bnxt_napi *bnapi = dev_instance;
        struct bnxt *bp = bnapi->bp;
        struct bnxt_cp_ring_info *cpr = &bnapi->cp_ring;
        u32 cons = RING_CMP(cpr->cp_raw_cons);
        u32 int_status;

        prefetch(&cpr->cp_desc_ring[CP_RING(cons)][CP_IDX(cons)]);

        if (!bnxt_has_work(bp, cpr)) {
                int_status = readl(bp->bar0 + BNXT_CAG_REG_LEGACY_INT_STATUS);
                /* return if erroneous interrupt */
                if (!(int_status & (0x10000 << cpr->cp_ring_struct.fw_ring_id)))
                        return IRQ_NONE;
        }

        /* disable ring IRQ */
        BNXT_CP_DB_IRQ_DIS(cpr->cp_db.doorbell);

        /* Return here if interrupt is shared and is disabled. */
        if (unlikely(atomic_read(&bp->intr_sem) != 0))
                return IRQ_HANDLED;

        napi_schedule(&bnapi->napi);
        return IRQ_HANDLED;
}

static int __bnxt_poll_work(struct bnxt *bp, struct bnxt_cp_ring_info *cpr,
                            int budget)
{
        struct bnxt_napi *bnapi = cpr->bnapi;
        u32 raw_cons = cpr->cp_raw_cons;
        u32 cons;
        int tx_pkts = 0;
        int rx_pkts = 0;
        u8 event = 0;
        struct tx_cmp *txcmp;

        cpr->has_more_work = 0;
        while (1) {
                int rc;

                cons = RING_CMP(raw_cons);
                txcmp = &cpr->cp_desc_ring[CP_RING(cons)][CP_IDX(cons)];

                if (!TX_CMP_VALID(txcmp, raw_cons))
                        break;

                /* The valid test of the entry must be done first before
                 * reading any further.
                 */
                dma_rmb();
                cpr->had_work_done = 1;
                if (TX_CMP_TYPE(txcmp) == CMP_TYPE_TX_L2_CMP) {
                        tx_pkts++;
                        /* return full budget so NAPI will complete. */
                        if (unlikely(tx_pkts > bp->tx_wake_thresh)) {
                                rx_pkts = budget;
                                raw_cons = NEXT_RAW_CMP(raw_cons);
                                if (budget)
                                        cpr->has_more_work = 1;
                                break;
                        }
                } else if ((TX_CMP_TYPE(txcmp) & 0x30) == 0x10) {
                        if (likely(budget))
                                rc = bnxt_rx_pkt(bp, cpr, &raw_cons, &event);
                        else
                                rc = bnxt_force_rx_discard(bp, cpr, &raw_cons,
                                                           &event);
                        if (likely(rc >= 0))
                                rx_pkts += rc;
                        /* Increment rx_pkts when rc is -ENOMEM to count towards
                         * the NAPI budget.  Otherwise, we may potentially loop
                         * here forever if we consistently cannot allocate
                         * buffers.
                         */
                        else if (rc == -ENOMEM && budget)
                                rx_pkts++;
                        else if (rc == -EBUSY)  /* partial completion */
                                break;
                } else if (unlikely((TX_CMP_TYPE(txcmp) ==
                                     CMPL_BASE_TYPE_HWRM_DONE) ||
                                    (TX_CMP_TYPE(txcmp) ==
                                     CMPL_BASE_TYPE_HWRM_FWD_REQ) ||
                                    (TX_CMP_TYPE(txcmp) ==
                                     CMPL_BASE_TYPE_HWRM_ASYNC_EVENT))) {
                        bnxt_hwrm_handler(bp, txcmp);
                }
                raw_cons = NEXT_RAW_CMP(raw_cons);

                if (rx_pkts && rx_pkts == budget) {
                        cpr->has_more_work = 1;
                        break;
                }
        }

        if (event & BNXT_TX_EVENT) {
                struct bnxt_tx_ring_info *txr = bnapi->tx_ring;
                u16 prod = txr->tx_prod;

                /* Sync BD data before updating doorbell */
                wmb();

                bnxt_db_write_relaxed(bp, &txr->tx_db, prod);
        }

        cpr->cp_raw_cons = raw_cons;
        bnapi->tx_pkts += tx_pkts;
        bnapi->events |= event;
        return rx_pkts;
}

static void __bnxt_poll_work_done(struct bnxt *bp, struct bnxt_napi *bnapi)
{
        if (bnapi->tx_pkts) {
                bnapi->tx_int(bp, bnapi, bnapi->tx_pkts);
                bnapi->tx_pkts = 0;
        }

        if (bnapi->events & BNXT_RX_EVENT) {
                struct bnxt_rx_ring_info *rxr = bnapi->rx_ring;

                bnxt_db_write(bp, &rxr->rx_db, rxr->rx_prod);
                if (bnapi->events & BNXT_AGG_EVENT)

                        bnxt_db_write(bp, &rxr->rx_agg_db, rxr->rx_agg_prod);
        }
        bnapi->events = 0;
}

static int bnxt_poll_work(struct bnxt *bp, struct bnxt_cp_ring_info *cpr,
                          int budget)
{
        struct bnxt_napi *bnapi = cpr->bnapi;
        int rx_pkts;

        rx_pkts = __bnxt_poll_work(bp, cpr, budget);

        /* ACK completion ring before freeing tx ring and producing new
         * buffers in rx/agg rings to prevent overflowing the completion
         * ring.
         */
        bnxt_db_cq(bp, &cpr->cp_db, cpr->cp_raw_cons);

        __bnxt_poll_work_done(bp, bnapi);
        return rx_pkts;
}

static int bnxt_poll_nitroa0(struct napi_struct *napi, int budget)
{
        struct bnxt_napi *bnapi = container_of(napi, struct bnxt_napi, napi);
        struct bnxt *bp = bnapi->bp;
        struct bnxt_cp_ring_info *cpr = &bnapi->cp_ring;
        struct bnxt_rx_ring_info *rxr = bnapi->rx_ring;
        struct tx_cmp *txcmp;
        struct rx_cmp_ext *rxcmp1;
        u32 cp_cons, tmp_raw_cons;
        u32 raw_cons = cpr->cp_raw_cons;
        u32 rx_pkts = 0;
        u8 event = 0;

        while (1) {
                int rc;

                cp_cons = RING_CMP(raw_cons);
                txcmp = &cpr->cp_desc_ring[CP_RING(cp_cons)][CP_IDX(cp_cons)];

                if (!TX_CMP_VALID(txcmp, raw_cons))
                        break;

                if ((TX_CMP_TYPE(txcmp) & 0x30) == 0x10) {
                        tmp_raw_cons = NEXT_RAW_CMP(raw_cons);
                        cp_cons = RING_CMP(tmp_raw_cons);
                        rxcmp1 = (struct rx_cmp_ext *)
                          &cpr->cp_desc_ring[CP_RING(cp_cons)][CP_IDX(cp_cons)];

                        if (!RX_CMP_VALID(rxcmp1, tmp_raw_cons))
                                break;

                        /* force an error to recycle the buffer */
                        rxcmp1->rx_cmp_cfa_code_errors_v2 |=
                                cpu_to_le32(RX_CMPL_ERRORS_CRC_ERROR);

                        rc = bnxt_rx_pkt(bp, cpr, &raw_cons, &event);
                        if (likely(rc == -EIO) && budget)
                                rx_pkts++;
                        else if (rc == -EBUSY)  /* partial completion */
                                break;
                } else if (unlikely(TX_CMP_TYPE(txcmp) ==
                                    CMPL_BASE_TYPE_HWRM_DONE)) {
                        bnxt_hwrm_handler(bp, txcmp);
                } else {
                        netdev_err(bp->dev,
                                   "Invalid completion received on special ring\n");
                }
                raw_cons = NEXT_RAW_CMP(raw_cons);

                if (rx_pkts == budget)
                        break;
        }

        cpr->cp_raw_cons = raw_cons;
        BNXT_DB_CQ(&cpr->cp_db, cpr->cp_raw_cons);
        bnxt_db_write(bp, &rxr->rx_db, rxr->rx_prod);

        if (event & BNXT_AGG_EVENT)
                bnxt_db_write(bp, &rxr->rx_agg_db, rxr->rx_agg_prod);

        if (!bnxt_has_work(bp, cpr) && rx_pkts < budget) {
                napi_complete_done(napi, rx_pkts);
                BNXT_DB_CQ_ARM(&cpr->cp_db, cpr->cp_raw_cons);
        }
        return rx_pkts;
}

static int bnxt_poll(struct napi_struct *napi, int budget)
{
        struct bnxt_napi *bnapi = container_of(napi, struct bnxt_napi, napi);
        struct bnxt *bp = bnapi->bp;
        struct bnxt_cp_ring_info *cpr = &bnapi->cp_ring;
        int work_done = 0;

        while (1) {
                work_done += bnxt_poll_work(bp, cpr, budget - work_done);

                if (work_done >= budget) {
                        if (!budget)
                                BNXT_DB_CQ_ARM(&cpr->cp_db, cpr->cp_raw_cons);
                        break;
                }

                if (!bnxt_has_work(bp, cpr)) {
                        if (napi_complete_done(napi, work_done))
                                BNXT_DB_CQ_ARM(&cpr->cp_db, cpr->cp_raw_cons);
                        break;
                }
        }
        if (bp->flags & BNXT_FLAG_DIM) {
                struct net_dim_sample dim_sample;

                net_dim_sample(cpr->event_ctr,
                               cpr->rx_packets,
                               cpr->rx_bytes,
                               &dim_sample);
                net_dim(&cpr->dim, dim_sample);
        }
        mmiowb();
        return work_done;
}

static int __bnxt_poll_cqs(struct bnxt *bp, struct bnxt_napi *bnapi, int budget)
{
        struct bnxt_cp_ring_info *cpr = &bnapi->cp_ring;
        int i, work_done = 0;

        for (i = 0; i < 2; i++) {
                struct bnxt_cp_ring_info *cpr2 = cpr->cp_ring_arr[i];

                if (cpr2) {
                        work_done += __bnxt_poll_work(bp, cpr2,
                                                      budget - work_done);
                        cpr->has_more_work |= cpr2->has_more_work;
                }
        }
        return work_done;
}

static void __bnxt_poll_cqs_done(struct bnxt *bp, struct bnxt_napi *bnapi,
                                 u64 dbr_type, bool all)
{
        struct bnxt_cp_ring_info *cpr = &bnapi->cp_ring;
        int i;

        for (i = 0; i < 2; i++) {
                struct bnxt_cp_ring_info *cpr2 = cpr->cp_ring_arr[i];
                struct bnxt_db_info *db;

                if (cpr2 && (all || cpr2->had_work_done)) {
                        db = &cpr2->cp_db;
                        writeq(db->db_key64 | dbr_type |
                               RING_CMP(cpr2->cp_raw_cons), db->doorbell);
                        cpr2->had_work_done = 0;
                }
        }
        __bnxt_poll_work_done(bp, bnapi);
}

static int bnxt_poll_p5(struct napi_struct *napi, int budget)
{
        struct bnxt_napi *bnapi = container_of(napi, struct bnxt_napi, napi);
        struct bnxt_cp_ring_info *cpr = &bnapi->cp_ring;
        u32 raw_cons = cpr->cp_raw_cons;
        struct bnxt *bp = bnapi->bp;
        struct nqe_cn *nqcmp;
        int work_done = 0;
        u32 cons;

        if (cpr->has_more_work) {
                cpr->has_more_work = 0;
                work_done = __bnxt_poll_cqs(bp, bnapi, budget);
                if (cpr->has_more_work) {
                        __bnxt_poll_cqs_done(bp, bnapi, DBR_TYPE_CQ, false);
                        return work_done;
                }
                __bnxt_poll_cqs_done(bp, bnapi, DBR_TYPE_CQ_ARMALL, true);
                if (napi_complete_done(napi, work_done))
                        BNXT_DB_NQ_ARM_P5(&cpr->cp_db, cpr->cp_raw_cons);
                return work_done;
        }
        while (1) {
                cons = RING_CMP(raw_cons);
                nqcmp = &cpr->nq_desc_ring[CP_RING(cons)][CP_IDX(cons)];

                if (!NQ_CMP_VALID(nqcmp, raw_cons)) {
                        __bnxt_poll_cqs_done(bp, bnapi, DBR_TYPE_CQ_ARMALL,
                                             false);
                        cpr->cp_raw_cons = raw_cons;
                        if (napi_complete_done(napi, work_done))
                                BNXT_DB_NQ_ARM_P5(&cpr->cp_db,
                                                  cpr->cp_raw_cons);
                        return work_done;
                }

                /* The valid test of the entry must be done first before
                 * reading any further.
                 */
                dma_rmb();

                if (nqcmp->type == cpu_to_le16(NQ_CN_TYPE_CQ_NOTIFICATION)) {
                        u32 idx = le32_to_cpu(nqcmp->cq_handle_low);
                        struct bnxt_cp_ring_info *cpr2;

                        cpr2 = cpr->cp_ring_arr[idx];
                        work_done += __bnxt_poll_work(bp, cpr2,
                                                      budget - work_done);
                        cpr->has_more_work = cpr2->has_more_work;
                } else {
                        bnxt_hwrm_handler(bp, (struct tx_cmp *)nqcmp);
                }
                raw_cons = NEXT_RAW_CMP(raw_cons);
                if (cpr->has_more_work)
                        break;
        }
        __bnxt_poll_cqs_done(bp, bnapi, DBR_TYPE_CQ, true);
        cpr->cp_raw_cons = raw_cons;
        return work_done;
}

static void bnxt_free_tx_skbs(struct bnxt *bp)
{
        int i, max_idx;
        struct pci_dev *pdev = bp->pdev;

        if (!bp->tx_ring)
                return;

        max_idx = bp->tx_nr_pages * TX_DESC_CNT;
        for (i = 0; i < bp->tx_nr_rings; i++) {
                struct bnxt_tx_ring_info *txr = &bp->tx_ring[i];
                int j;

                for (j = 0; j < max_idx;) {
                        struct bnxt_sw_tx_bd *tx_buf = &txr->tx_buf_ring[j];
                        struct sk_buff *skb = tx_buf->skb;
                        int k, last;

                        if (!skb) {
                                j++;
                                continue;
                        }

                        tx_buf->skb = NULL;

                        if (tx_buf->is_push) {
                                dev_kfree_skb(skb);
                                j += 2;
                                continue;
                        }

                        dma_unmap_single(&pdev->dev,
                                         dma_unmap_addr(tx_buf, mapping),
                                         skb_headlen(skb),
                                         PCI_DMA_TODEVICE);

                        last = tx_buf->nr_frags;
                        j += 2;
                        for (k = 0; k < last; k++, j++) {
                                int ring_idx = j & bp->tx_ring_mask;
                                skb_frag_t *frag = &skb_shinfo(skb)->frags[k];

                                tx_buf = &txr->tx_buf_ring[ring_idx];
                                dma_unmap_page(
                                        &pdev->dev,
                                        dma_unmap_addr(tx_buf, mapping),
                                        skb_frag_size(frag), PCI_DMA_TODEVICE);
                        }
                        dev_kfree_skb(skb);
                }
                netdev_tx_reset_queue(netdev_get_tx_queue(bp->dev, i));
        }
}

static void bnxt_free_rx_skbs(struct bnxt *bp)
{
        int i, max_idx, max_agg_idx;
        struct pci_dev *pdev = bp->pdev;

        if (!bp->rx_ring)
                return;

        max_idx = bp->rx_nr_pages * RX_DESC_CNT;
        max_agg_idx = bp->rx_agg_nr_pages * RX_DESC_CNT;
        for (i = 0; i < bp->rx_nr_rings; i++) {
                struct bnxt_rx_ring_info *rxr = &bp->rx_ring[i];
                int j;

                if (rxr->rx_tpa) {
                        for (j = 0; j < MAX_TPA; j++) {
                                struct bnxt_tpa_info *tpa_info =
                                                        &rxr->rx_tpa[j];
                                u8 *data = tpa_info->data;

                                if (!data)
                                        continue;

                                dma_unmap_single_attrs(&pdev->dev,
                                                       tpa_info->mapping,
                                                       bp->rx_buf_use_size,
                                                       bp->rx_dir,
                                                       DMA_ATTR_WEAK_ORDERING);

                                tpa_info->data = NULL;

                                kfree(data);
                        }
                }

                for (j = 0; j < max_idx; j++) {
                        struct bnxt_sw_rx_bd *rx_buf = &rxr->rx_buf_ring[j];
                        dma_addr_t mapping = rx_buf->mapping;
                        void *data = rx_buf->data;

                        if (!data)
                                continue;

                        rx_buf->data = NULL;

                        if (BNXT_RX_PAGE_MODE(bp)) {
                                mapping -= bp->rx_dma_offset;
                                dma_unmap_page_attrs(&pdev->dev, mapping,
                                                     PAGE_SIZE, bp->rx_dir,
                                                     DMA_ATTR_WEAK_ORDERING);
                                __free_page(data);
                        } else {
                                dma_unmap_single_attrs(&pdev->dev, mapping,
                                                       bp->rx_buf_use_size,
                                                       bp->rx_dir,
                                                       DMA_ATTR_WEAK_ORDERING);
                                kfree(data);
                        }
                }

                for (j = 0; j < max_agg_idx; j++) {
                        struct bnxt_sw_rx_agg_bd *rx_agg_buf =
                                &rxr->rx_agg_ring[j];
                        struct page *page = rx_agg_buf->page;

                        if (!page)
                                continue;

                        dma_unmap_page_attrs(&pdev->dev, rx_agg_buf->mapping,
                                             BNXT_RX_PAGE_SIZE,
                                             PCI_DMA_FROMDEVICE,
                                             DMA_ATTR_WEAK_ORDERING);

                        rx_agg_buf->page = NULL;
                        __clear_bit(j, rxr->rx_agg_bmap);

                        __free_page(page);
                }
                if (rxr->rx_page) {
                        __free_page(rxr->rx_page);
                        rxr->rx_page = NULL;
                }
        }
}

static void bnxt_free_skbs(struct bnxt *bp)
{
        bnxt_free_tx_skbs(bp);
        bnxt_free_rx_skbs(bp);
}

static void bnxt_free_ring(struct bnxt *bp, struct bnxt_ring_mem_info *rmem)
{
        struct pci_dev *pdev = bp->pdev;
        int i;

        for (i = 0; i < rmem->nr_pages; i++) {
                if (!rmem->pg_arr[i])
                        continue;

                dma_free_coherent(&pdev->dev, rmem->page_size,
                                  rmem->pg_arr[i], rmem->dma_arr[i]);

                rmem->pg_arr[i] = NULL;
        }
        if (rmem->pg_tbl) {
                size_t pg_tbl_size = rmem->nr_pages * 8;

                if (rmem->flags & BNXT_RMEM_USE_FULL_PAGE_FLAG)
                        pg_tbl_size = rmem->page_size;
                dma_free_coherent(&pdev->dev, pg_tbl_size,
                                  rmem->pg_tbl, rmem->pg_tbl_map);
                rmem->pg_tbl = NULL;
        }
        if (rmem->vmem_size && *rmem->vmem) {
                vfree(*rmem->vmem);
                *rmem->vmem = NULL;
        }
}

static int bnxt_alloc_ring(struct bnxt *bp, struct bnxt_ring_mem_info *rmem)
{
        struct pci_dev *pdev = bp->pdev;
        u64 valid_bit = 0;
        int i;

        if (rmem->flags & (BNXT_RMEM_VALID_PTE_FLAG | BNXT_RMEM_RING_PTE_FLAG))
                valid_bit = PTU_PTE_VALID;
        if ((rmem->nr_pages > 1 || rmem->depth > 0) && !rmem->pg_tbl) {
                size_t pg_tbl_size = rmem->nr_pages * 8;

                if (rmem->flags & BNXT_RMEM_USE_FULL_PAGE_FLAG)
                        pg_tbl_size = rmem->page_size;
                rmem->pg_tbl = dma_alloc_coherent(&pdev->dev, pg_tbl_size,
                                                  &rmem->pg_tbl_map,
                                                  GFP_KERNEL);
                if (!rmem->pg_tbl)
                        return -ENOMEM;
        }

        for (i = 0; i < rmem->nr_pages; i++) {
                u64 extra_bits = valid_bit;

                rmem->pg_arr[i] = dma_alloc_coherent(&pdev->dev,
                                                     rmem->page_size,
                                                     &rmem->dma_arr[i],
                                                     GFP_KERNEL);
                if (!rmem->pg_arr[i])
                        return -ENOMEM;

                if (rmem->nr_pages > 1 || rmem->depth > 0) {
                        if (i == rmem->nr_pages - 2 &&
                            (rmem->flags & BNXT_RMEM_RING_PTE_FLAG))
                                extra_bits |= PTU_PTE_NEXT_TO_LAST;
                        else if (i == rmem->nr_pages - 1 &&
                                 (rmem->flags & BNXT_RMEM_RING_PTE_FLAG))
                                extra_bits |= PTU_PTE_LAST;
                        rmem->pg_tbl[i] =
                                cpu_to_le64(rmem->dma_arr[i] | extra_bits);
                }
        }

        if (rmem->vmem_size) {
                *rmem->vmem = vzalloc(rmem->vmem_size);
                if (!(*rmem->vmem))
                        return -ENOMEM;
        }
        return 0;
}

static void bnxt_free_rx_rings(struct bnxt *bp)
{
        int i;

        if (!bp->rx_ring)
                return;

        for (i = 0; i < bp->rx_nr_rings; i++) {
                struct bnxt_rx_ring_info *rxr = &bp->rx_ring[i];
                struct bnxt_ring_struct *ring;

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

                if (xdp_rxq_info_is_reg(&rxr->xdp_rxq))
                        xdp_rxq_info_unreg(&rxr->xdp_rxq);

                kfree(rxr->rx_tpa);
                rxr->rx_tpa = NULL;

                kfree(rxr->rx_agg_bmap);
                rxr->rx_agg_bmap = NULL;

                ring = &rxr->rx_ring_struct;
                bnxt_free_ring(bp, &ring->ring_mem);

                ring = &rxr->rx_agg_ring_struct;
                bnxt_free_ring(bp, &ring->ring_mem);
        }
}

static int bnxt_alloc_rx_rings(struct bnxt *bp)
{
        int i, rc, agg_rings = 0, tpa_rings = 0;

        if (!bp->rx_ring)
                return -ENOMEM;

        if (bp->flags & BNXT_FLAG_AGG_RINGS)
                agg_rings = 1;

        if (bp->flags & BNXT_FLAG_TPA)
                tpa_rings = 1;

        for (i = 0; i < bp->rx_nr_rings; i++) {
                struct bnxt_rx_ring_info *rxr = &bp->rx_ring[i];
                struct bnxt_ring_struct *ring;

                ring = &rxr->rx_ring_struct;

                rc = xdp_rxq_info_reg(&rxr->xdp_rxq, bp->dev, i);
                if (rc < 0)
                        return rc;

                rc = bnxt_alloc_ring(bp, &ring->ring_mem);
                if (rc)
                        return rc;

                ring->grp_idx = i;
                if (agg_rings) {
                        u16 mem_size;

                        ring = &rxr->rx_agg_ring_struct;
                        rc = bnxt_alloc_ring(bp, &ring->ring_mem);
                        if (rc)
                                return rc;

                        ring->grp_idx = i;
                        rxr->rx_agg_bmap_size = bp->rx_agg_ring_mask + 1;
                        mem_size = rxr->rx_agg_bmap_size / 8;
                        rxr->rx_agg_bmap = kzalloc(mem_size, GFP_KERNEL);
                        if (!rxr->rx_agg_bmap)
                                return -ENOMEM;

                        if (tpa_rings) {
                                rxr->rx_tpa = kcalloc(MAX_TPA,
                                                sizeof(struct bnxt_tpa_info),
                                                GFP_KERNEL);
                                if (!rxr->rx_tpa)
                                        return -ENOMEM;
                        }
                }
        }
        return 0;
}

static void bnxt_free_tx_rings(struct bnxt *bp)
{
        int i;
        struct pci_dev *pdev = bp->pdev;

        if (!bp->tx_ring)
                return;

        for (i = 0; i < bp->tx_nr_rings; i++) {
                struct bnxt_tx_ring_info *txr = &bp->tx_ring[i];
                struct bnxt_ring_struct *ring;

                if (txr->tx_push) {
                        dma_free_coherent(&pdev->dev, bp->tx_push_size,
                                          txr->tx_push, txr->tx_push_mapping);
                        txr->tx_push = NULL;
                }

                ring = &txr->tx_ring_struct;

                bnxt_free_ring(bp, &ring->ring_mem);
        }
}

static int bnxt_alloc_tx_rings(struct bnxt *bp)
{
        int i, j, rc;
        struct pci_dev *pdev = bp->pdev;

        bp->tx_push_size = 0;
        if (bp->tx_push_thresh) {
                int push_size;

                push_size  = L1_CACHE_ALIGN(sizeof(struct tx_push_bd) +
                                        bp->tx_push_thresh);

                if (push_size > 256) {
                        push_size = 0;
                        bp->tx_push_thresh = 0;
                }

                bp->tx_push_size = push_size;
        }

        for (i = 0, j = 0; i < bp->tx_nr_rings; i++) {
                struct bnxt_tx_ring_info *txr = &bp->tx_ring[i];
                struct bnxt_ring_struct *ring;
                u8 qidx;

                ring = &txr->tx_ring_struct;

                rc = bnxt_alloc_ring(bp, &ring->ring_mem);
                if (rc)
                        return rc;

                ring->grp_idx = txr->bnapi->index;
                if (bp->tx_push_size) {
                        dma_addr_t mapping;

                        /* One pre-allocated DMA buffer to backup
                         * TX push operation
                         */
                        txr->tx_push = dma_alloc_coherent(&pdev->dev,
                                                bp->tx_push_size,
                                                &txr->tx_push_mapping,
                                                GFP_KERNEL);

                        if (!txr->tx_push)
                                return -ENOMEM;

                        mapping = txr->tx_push_mapping +
                                sizeof(struct tx_push_bd);
                        txr->data_mapping = cpu_to_le64(mapping);

                        memset(txr->tx_push, 0, sizeof(struct tx_push_bd));
                }
                qidx = bp->tc_to_qidx[j];
                ring->queue_id = bp->q_info[qidx].queue_id;
                if (i < bp->tx_nr_rings_xdp)
                        continue;
                if (i % bp->tx_nr_rings_per_tc == (bp->tx_nr_rings_per_tc - 1))
                        j++;
        }
        return 0;
}

static void bnxt_free_cp_rings(struct bnxt *bp)
{
        int i;

        if (!bp->bnapi)
                return;

        for (i = 0; i < bp->cp_nr_rings; i++) {
                struct bnxt_napi *bnapi = bp->bnapi[i];
                struct bnxt_cp_ring_info *cpr;
                struct bnxt_ring_struct *ring;
                int j;

                if (!bnapi)
                        continue;

                cpr = &bnapi->cp_ring;
                ring = &cpr->cp_ring_struct;

                bnxt_free_ring(bp, &ring->ring_mem);

                for (j = 0; j < 2; j++) {
                        struct bnxt_cp_ring_info *cpr2 = cpr->cp_ring_arr[j];

                        if (cpr2) {
                                ring = &cpr2->cp_ring_struct;
                                bnxt_free_ring(bp, &ring->ring_mem);
                                kfree(cpr2);
                                cpr->cp_ring_arr[j] = NULL;
                        }
                }
        }
}

static struct bnxt_cp_ring_info *bnxt_alloc_cp_sub_ring(struct bnxt *bp)
{
        struct bnxt_ring_mem_info *rmem;
        struct bnxt_ring_struct *ring;
        struct bnxt_cp_ring_info *cpr;
        int rc;

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

        ring = &cpr->cp_ring_struct;
        rmem = &ring->ring_mem;
        rmem->nr_pages = bp->cp_nr_pages;
        rmem->page_size = HW_CMPD_RING_SIZE;
        rmem->pg_arr = (void **)cpr->cp_desc_ring;
        rmem->dma_arr = cpr->cp_desc_mapping;
        rmem->flags = BNXT_RMEM_RING_PTE_FLAG;
        rc = bnxt_alloc_ring(bp, rmem);
        if (rc) {
                bnxt_free_ring(bp, rmem);
                kfree(cpr);
                cpr = NULL;
        }
        return cpr;
}

static int bnxt_alloc_cp_rings(struct bnxt *bp)
{
        bool sh = !!(bp->flags & BNXT_FLAG_SHARED_RINGS);
        int i, rc, ulp_base_vec, ulp_msix;

        ulp_msix = bnxt_get_ulp_msix_num(bp);
        ulp_base_vec = bnxt_get_ulp_msix_base(bp);
        for (i = 0; i < bp->cp_nr_rings; i++) {
                struct bnxt_napi *bnapi = bp->bnapi[i];
                struct bnxt_cp_ring_info *cpr;
                struct bnxt_ring_struct *ring;

                if (!bnapi)
                        continue;

                cpr = &bnapi->cp_ring;
                cpr->bnapi = bnapi;
                ring = &cpr->cp_ring_struct;

                rc = bnxt_alloc_ring(bp, &ring->ring_mem);
                if (rc)
                        return rc;

                if (ulp_msix && i >= ulp_base_vec)
                        ring->map_idx = i + ulp_msix;
                else
                        ring->map_idx = i;

                if (!(bp->flags & BNXT_FLAG_CHIP_P5))
                        continue;

                if (i < bp->rx_nr_rings) {
                        struct bnxt_cp_ring_info *cpr2 =
                                bnxt_alloc_cp_sub_ring(bp);

                        cpr->cp_ring_arr[BNXT_RX_HDL] = cpr2;
                        if (!cpr2)
                                return -ENOMEM;
                        cpr2->bnapi = bnapi;
                }
                if ((sh && i < bp->tx_nr_rings) ||
                    (!sh && i >= bp->rx_nr_rings)) {
                        struct bnxt_cp_ring_info *cpr2 =
                                bnxt_alloc_cp_sub_ring(bp);

                        cpr->cp_ring_arr[BNXT_TX_HDL] = cpr2;
                        if (!cpr2)
                                return -ENOMEM;
                        cpr2->bnapi = bnapi;
                }
        }
        return 0;
}

static void bnxt_init_ring_struct(struct bnxt *bp)
{
        int i;

        for (i = 0; i < bp->cp_nr_rings; i++) {
                struct bnxt_napi *bnapi = bp->bnapi[i];
                struct bnxt_ring_mem_info *rmem;
                struct bnxt_cp_ring_info *cpr;
                struct bnxt_rx_ring_info *rxr;
                struct bnxt_tx_ring_info *txr;
                struct bnxt_ring_struct *ring;

                if (!bnapi)
                        continue;

                cpr = &bnapi->cp_ring;
                ring = &cpr->cp_ring_struct;
                rmem = &ring->ring_mem;
                rmem->nr_pages = bp->cp_nr_pages;
                rmem->page_size = HW_CMPD_RING_SIZE;
                rmem->pg_arr = (void **)cpr->cp_desc_ring;
                rmem->dma_arr = cpr->cp_desc_mapping;
                rmem->vmem_size = 0;

                rxr = bnapi->rx_ring;
                if (!rxr)
                        goto skip_rx;

                ring = &rxr->rx_ring_struct;
                rmem = &ring->ring_mem;
                rmem->nr_pages = bp->rx_nr_pages;
                rmem->page_size = HW_RXBD_RING_SIZE;
                rmem->pg_arr = (void **)rxr->rx_desc_ring;
                rmem->dma_arr = rxr->rx_desc_mapping;
                rmem->vmem_size = SW_RXBD_RING_SIZE * bp->rx_nr_pages;
                rmem->vmem = (void **)&rxr->rx_buf_ring;

                ring = &rxr->rx_agg_ring_struct;
                rmem = &ring->ring_mem;
                rmem->nr_pages = bp->rx_agg_nr_pages;
                rmem->page_size = HW_RXBD_RING_SIZE;
                rmem->pg_arr = (void **)rxr->rx_agg_desc_ring;
                rmem->dma_arr = rxr->rx_agg_desc_mapping;
                rmem->vmem_size = SW_RXBD_AGG_RING_SIZE * bp->rx_agg_nr_pages;
                rmem->vmem = (void **)&rxr->rx_agg_ring;

skip_rx:
                txr = bnapi->tx_ring;
                if (!txr)
                        continue;

                ring = &txr->tx_ring_struct;
                rmem = &ring->ring_mem;
                rmem->nr_pages = bp->tx_nr_pages;
                rmem->page_size = HW_RXBD_RING_SIZE;
                rmem->pg_arr = (void **)txr->tx_desc_ring;
                rmem->dma_arr = txr->tx_desc_mapping;
                rmem->vmem_size = SW_TXBD_RING_SIZE * bp->tx_nr_pages;
                rmem->vmem = (void **)&txr->tx_buf_ring;
        }
}

static void bnxt_init_rxbd_pages(struct bnxt_ring_struct *ring, u32 type)
{
        int i;
        u32 prod;
        struct rx_bd **rx_buf_ring;

        rx_buf_ring = (struct rx_bd **)ring->ring_mem.pg_arr;
        for (i = 0, prod = 0; i < ring->ring_mem.nr_pages; i++) {
                int j;
                struct rx_bd *rxbd;

                rxbd = rx_buf_ring[i];
                if (!rxbd)
                        continue;

                for (j = 0; j < RX_DESC_CNT; j++, rxbd++, prod++) {
                        rxbd->rx_bd_len_flags_type = cpu_to_le32(type);
                        rxbd->rx_bd_opaque = prod;
                }
        }
}

static int bnxt_init_one_rx_ring(struct bnxt *bp, int ring_nr)
{
        struct net_device *dev = bp->dev;
        struct bnxt_rx_ring_info *rxr;
        struct bnxt_ring_struct *ring;
        u32 prod, type;
        int i;

        type = (bp->rx_buf_use_size << RX_BD_LEN_SHIFT) |
                RX_BD_TYPE_RX_PACKET_BD | RX_BD_FLAGS_EOP;

        if (NET_IP_ALIGN == 2)
                type |= RX_BD_FLAGS_SOP;

        rxr = &bp->rx_ring[ring_nr];
        ring = &rxr->rx_ring_struct;
        bnxt_init_rxbd_pages(ring, type);

        if (BNXT_RX_PAGE_MODE(bp) && bp->xdp_prog) {
                rxr->xdp_prog = bpf_prog_add(bp->xdp_prog, 1);
                if (IS_ERR(rxr->xdp_prog)) {
                        int rc = PTR_ERR(rxr->xdp_prog);

                        rxr->xdp_prog = NULL;
                        return rc;
                }
        }
        prod = rxr->rx_prod;
        for (i = 0; i < bp->rx_ring_size; i++) {
                if (bnxt_alloc_rx_data(bp, rxr, prod, GFP_KERNEL) != 0) {
                        netdev_warn(dev, "init'ed rx ring %d with %d/%d skbs only\n",
                                    ring_nr, i, bp->rx_ring_size);
                        break;
                }
                prod = NEXT_RX(prod);
        }
        rxr->rx_prod = prod;
        ring->fw_ring_id = INVALID_HW_RING_ID;

        ring = &rxr->rx_agg_ring_struct;
        ring->fw_ring_id = INVALID_HW_RING_ID;

        if (!(bp->flags & BNXT_FLAG_AGG_RINGS))
                return 0;

        type = ((u32)BNXT_RX_PAGE_SIZE << RX_BD_LEN_SHIFT) |
                RX_BD_TYPE_RX_AGG_BD | RX_BD_FLAGS_SOP;

        bnxt_init_rxbd_pages(ring, type);

        prod = rxr->rx_agg_prod;
        for (i = 0; i < bp->rx_agg_ring_size; i++) {
                if (bnxt_alloc_rx_page(bp, rxr, prod, GFP_KERNEL) != 0) {
                        netdev_warn(dev, "init'ed rx ring %d with %d/%d pages only\n",
                                    ring_nr, i, bp->rx_ring_size);
                        break;
                }
                prod = NEXT_RX_AGG(prod);
        }
        rxr->rx_agg_prod = prod;

        if (bp->flags & BNXT_FLAG_TPA) {
                if (rxr->rx_tpa) {
                        u8 *data;
                        dma_addr_t mapping;

                        for (i = 0; i < MAX_TPA; i++) {
                                data = __bnxt_alloc_rx_data(bp, &mapping,
                                                            GFP_KERNEL);
                                if (!data)
                                        return -ENOMEM;

                                rxr->rx_tpa[i].data = data;
                                rxr->rx_tpa[i].data_ptr = data + bp->rx_offset;
                                rxr->rx_tpa[i].mapping = mapping;
                        }
                } else {
                        netdev_err(bp->dev, "No resource allocated for LRO/GRO\n");
                        return -ENOMEM;
                }
        }

        return 0;
}

static void bnxt_init_cp_rings(struct bnxt *bp)
{
        int i, j;

        for (i = 0; i < bp->cp_nr_rings; i++) {
                struct bnxt_cp_ring_info *cpr = &bp->bnapi[i]->cp_ring;
                struct bnxt_ring_struct *ring = &cpr->cp_ring_struct;

                ring->fw_ring_id = INVALID_HW_RING_ID;
                cpr->rx_ring_coal.coal_ticks = bp->rx_coal.coal_ticks;
                cpr->rx_ring_coal.coal_bufs = bp->rx_coal.coal_bufs;
                for (j = 0; j < 2; j++) {
                        struct bnxt_cp_ring_info *cpr2 = cpr->cp_ring_arr[j];

                        if (!cpr2)
                                continue;

                        ring = &cpr2->cp_ring_struct;
                        ring->fw_ring_id = INVALID_HW_RING_ID;
                        cpr2->rx_ring_coal.coal_ticks = bp->rx_coal.coal_ticks;
                        cpr2->rx_ring_coal.coal_bufs = bp->rx_coal.coal_bufs;
                }
        }
}

static int bnxt_init_rx_rings(struct bnxt *bp)
{
        int i, rc = 0;

        if (BNXT_RX_PAGE_MODE(bp)) {
                bp->rx_offset = NET_IP_ALIGN + XDP_PACKET_HEADROOM;
                bp->rx_dma_offset = XDP_PACKET_HEADROOM;
        } else {
                bp->rx_offset = BNXT_RX_OFFSET;
                bp->rx_dma_offset = BNXT_RX_DMA_OFFSET;
        }

        for (i = 0; i < bp->rx_nr_rings; i++) {
                rc = bnxt_init_one_rx_ring(bp, i);
                if (rc)
                        break;
        }

        return rc;
}

static int bnxt_init_tx_rings(struct bnxt *bp)
{
        u16 i;

        bp->tx_wake_thresh = max_t(int, bp->tx_ring_size / 2,
                                   MAX_SKB_FRAGS + 1);

        for (i = 0; i < bp->tx_nr_rings; i++) {
                struct bnxt_tx_ring_info *txr = &bp->tx_ring[i];
                struct bnxt_ring_struct *ring = &txr->tx_ring_struct;

                ring->fw_ring_id = INVALID_HW_RING_ID;
        }

        return 0;
}

static void bnxt_free_ring_grps(struct bnxt *bp)
{
        kfree(bp->grp_info);
        bp->grp_info = NULL;
}

static int bnxt_init_ring_grps(struct bnxt *bp, bool irq_re_init)
{
        int i;

        if (irq_re_init) {
                bp->grp_info = kcalloc(bp->cp_nr_rings,
                                       sizeof(struct bnxt_ring_grp_info),
                                       GFP_KERNEL);
                if (!bp->grp_info)
                        return -ENOMEM;
        }
        for (i = 0; i < bp->cp_nr_rings; i++) {
                if (irq_re_init)
                        bp->grp_info[i].fw_stats_ctx = INVALID_HW_RING_ID;
                bp->grp_info[i].fw_grp_id = INVALID_HW_RING_ID;
                bp->grp_info[i].rx_fw_ring_id = INVALID_HW_RING_ID;
                bp->grp_info[i].agg_fw_ring_id = INVALID_HW_RING_ID;
                bp->grp_info[i].cp_fw_ring_id = INVALID_HW_RING_ID;
        }
        return 0;
}

static void bnxt_free_vnics(struct bnxt *bp)
{
        kfree(bp->vnic_info);
        bp->vnic_info = NULL;
        bp->nr_vnics = 0;
}