/*
 * Linux driver for VMware's vmxnet3 ethernet NIC.
 *
 * Copyright (C) 2008-2021, VMware, Inc. All Rights Reserved.
 *
 * 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; version 2 of the License and no later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE, GOOD TITLE or
 * NON INFRINGEMENT. See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * The full GNU General Public License is included in this distribution in
 * the file called "COPYING".
 *
 * Maintained by: pv-drivers@vmware.com
 *
 */

#include <linux/module.h>
#include <net/ip6_checksum.h>

#include "vmxnet3_int.h"

char vmxnet3_driver_name[] = "vmxnet3";
#define VMXNET3_DRIVER_DESC "VMware vmxnet3 virtual NIC driver"

/*
 * PCI Device ID Table
 * Last entry must be all 0s
 */
static const struct pci_device_id vmxnet3_pciid_table[] = {
        {PCI_VDEVICE(VMWARE, PCI_DEVICE_ID_VMWARE_VMXNET3)},
        {0}
};

MODULE_DEVICE_TABLE(pci, vmxnet3_pciid_table);

static int enable_mq = 1;

static void
vmxnet3_write_mac_addr(struct vmxnet3_adapter *adapter, u8 *mac);

/*
 *    Enable/Disable the given intr
 */
static void
vmxnet3_enable_intr(struct vmxnet3_adapter *adapter, unsigned intr_idx)
{
        VMXNET3_WRITE_BAR0_REG(adapter, VMXNET3_REG_IMR + intr_idx * 8, 0);
}


static void
vmxnet3_disable_intr(struct vmxnet3_adapter *adapter, unsigned intr_idx)
{
        VMXNET3_WRITE_BAR0_REG(adapter, VMXNET3_REG_IMR + intr_idx * 8, 1);
}


/*
 *    Enable/Disable all intrs used by the device
 */
static void
vmxnet3_enable_all_intrs(struct vmxnet3_adapter *adapter)
{
        int i;

        for (i = 0; i < adapter->intr.num_intrs; i++)
                vmxnet3_enable_intr(adapter, i);
        adapter->shared->devRead.intrConf.intrCtrl &=
                                        cpu_to_le32(~VMXNET3_IC_DISABLE_ALL);
}


static void
vmxnet3_disable_all_intrs(struct vmxnet3_adapter *adapter)
{
        int i;

        adapter->shared->devRead.intrConf.intrCtrl |=
                                        cpu_to_le32(VMXNET3_IC_DISABLE_ALL);
        for (i = 0; i < adapter->intr.num_intrs; i++)
                vmxnet3_disable_intr(adapter, i);
}


static void
vmxnet3_ack_events(struct vmxnet3_adapter *adapter, u32 events)
{
        VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_ECR, events);
}


static bool
vmxnet3_tq_stopped(struct vmxnet3_tx_queue *tq, struct vmxnet3_adapter *adapter)
{
        return tq->stopped;
}


static void
vmxnet3_tq_start(struct vmxnet3_tx_queue *tq, struct vmxnet3_adapter *adapter)
{
        tq->stopped = false;
        netif_start_subqueue(adapter->netdev, tq - adapter->tx_queue);
}


static void
vmxnet3_tq_wake(struct vmxnet3_tx_queue *tq, struct vmxnet3_adapter *adapter)
{
        tq->stopped = false;
        netif_wake_subqueue(adapter->netdev, (tq - adapter->tx_queue));
}


static void
vmxnet3_tq_stop(struct vmxnet3_tx_queue *tq, struct vmxnet3_adapter *adapter)
{
        tq->stopped = true;
        tq->num_stop++;
        netif_stop_subqueue(adapter->netdev, (tq - adapter->tx_queue));
}


/*
 * Check the link state. This may start or stop the tx queue.
 */
static void
vmxnet3_check_link(struct vmxnet3_adapter *adapter, bool affectTxQueue)
{
        u32 ret;
        int i;
        unsigned long flags;

        spin_lock_irqsave(&adapter->cmd_lock, flags);
        VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_CMD, VMXNET3_CMD_GET_LINK);
        ret = VMXNET3_READ_BAR1_REG(adapter, VMXNET3_REG_CMD);
        spin_unlock_irqrestore(&adapter->cmd_lock, flags);

        adapter->link_speed = ret >> 16;
        if (ret & 1) { /* Link is up. */
                netdev_info(adapter->netdev, "NIC Link is Up %d Mbps\n",
                            adapter->link_speed);
                netif_carrier_on(adapter->netdev);

                if (affectTxQueue) {
                        for (i = 0; i < adapter->num_tx_queues; i++)
                                vmxnet3_tq_start(&adapter->tx_queue[i],
                                                 adapter);
                }
        } else {
                netdev_info(adapter->netdev, "NIC Link is Down\n");
                netif_carrier_off(adapter->netdev);

                if (affectTxQueue) {
                        for (i = 0; i < adapter->num_tx_queues; i++)
                                vmxnet3_tq_stop(&adapter->tx_queue[i], adapter);
                }
        }
}

static void
vmxnet3_process_events(struct vmxnet3_adapter *adapter)
{
        int i;
        unsigned long flags;
        u32 events = le32_to_cpu(adapter->shared->ecr);
        if (!events)
                return;

        vmxnet3_ack_events(adapter, events);

        /* Check if link state has changed */
        if (events & VMXNET3_ECR_LINK)
                vmxnet3_check_link(adapter, true);

        /* Check if there is an error on xmit/recv queues */
        if (events & (VMXNET3_ECR_TQERR | VMXNET3_ECR_RQERR)) {
                spin_lock_irqsave(&adapter->cmd_lock, flags);
                VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_CMD,
                                       VMXNET3_CMD_GET_QUEUE_STATUS);
                spin_unlock_irqrestore(&adapter->cmd_lock, flags);

                for (i = 0; i < adapter->num_tx_queues; i++)
                        if (adapter->tqd_start[i].status.stopped)
                                dev_err(&adapter->netdev->dev,
                                        "%s: tq[%d] error 0x%x\n",
                                        adapter->netdev->name, i, le32_to_cpu(
                                        adapter->tqd_start[i].status.error));
                for (i = 0; i < adapter->num_rx_queues; i++)
                        if (adapter->rqd_start[i].status.stopped)
                                dev_err(&adapter->netdev->dev,
                                        "%s: rq[%d] error 0x%x\n",
                                        adapter->netdev->name, i,
                                        adapter->rqd_start[i].status.error);

                schedule_work(&adapter->work);
        }
}

#ifdef __BIG_ENDIAN_BITFIELD
/*
 * The device expects the bitfields in shared structures to be written in
 * little endian. When CPU is big endian, the following routines are used to
 * correctly read and write into ABI.
 * The general technique used here is : double word bitfields are defined in
 * opposite order for big endian architecture. Then before reading them in
 * driver the complete double word is translated using le32_to_cpu. Similarly
 * After the driver writes into bitfields, cpu_to_le32 is used to translate the
 * double words into required format.
 * In order to avoid touching bits in shared structure more than once, temporary
 * descriptors are used. These are passed as srcDesc to following functions.
 */
static void vmxnet3_RxDescToCPU(const struct Vmxnet3_RxDesc *srcDesc,
                                struct Vmxnet3_RxDesc *dstDesc)
{
        u32 *src = (u32 *)srcDesc + 2;
        u32 *dst = (u32 *)dstDesc + 2;
        dstDesc->addr = le64_to_cpu(srcDesc->addr);
        *dst = le32_to_cpu(*src);
        dstDesc->ext1 = le32_to_cpu(srcDesc->ext1);
}

static void vmxnet3_TxDescToLe(const struct Vmxnet3_TxDesc *srcDesc,
                               struct Vmxnet3_TxDesc *dstDesc)
{
        int i;
        u32 *src = (u32 *)(srcDesc + 1);
        u32 *dst = (u32 *)(dstDesc + 1);

        /* Working backwards so that the gen bit is set at the end. */
        for (i = 2; i > 0; i--) {
                src--;
                dst--;
                *dst = cpu_to_le32(*src);
        }
}


static void vmxnet3_RxCompToCPU(const struct Vmxnet3_RxCompDesc *srcDesc,
                                struct Vmxnet3_RxCompDesc *dstDesc)
{
        int i = 0;
        u32 *src = (u32 *)srcDesc;
        u32 *dst = (u32 *)dstDesc;
        for (i = 0; i < sizeof(struct Vmxnet3_RxCompDesc) / sizeof(u32); i++) {
                *dst = le32_to_cpu(*src);
                src++;
                dst++;
        }
}


/* Used to read bitfield values from double words. */
static u32 get_bitfield32(const __le32 *bitfield, u32 pos, u32 size)
{
        u32 temp = le32_to_cpu(*bitfield);
        u32 mask = ((1 << size) - 1) << pos;
        temp &= mask;
        temp >>= pos;
        return temp;
}



#endif  /* __BIG_ENDIAN_BITFIELD */

#ifdef __BIG_ENDIAN_BITFIELD

#   define VMXNET3_TXDESC_GET_GEN(txdesc) get_bitfield32(((const __le32 *) \
                        txdesc) + VMXNET3_TXD_GEN_DWORD_SHIFT, \
                        VMXNET3_TXD_GEN_SHIFT, VMXNET3_TXD_GEN_SIZE)
#   define VMXNET3_TXDESC_GET_EOP(txdesc) get_bitfield32(((const __le32 *) \
                        txdesc) + VMXNET3_TXD_EOP_DWORD_SHIFT, \
                        VMXNET3_TXD_EOP_SHIFT, VMXNET3_TXD_EOP_SIZE)
#   define VMXNET3_TCD_GET_GEN(tcd) get_bitfield32(((const __le32 *)tcd) + \
                        VMXNET3_TCD_GEN_DWORD_SHIFT, VMXNET3_TCD_GEN_SHIFT, \
                        VMXNET3_TCD_GEN_SIZE)
#   define VMXNET3_TCD_GET_TXIDX(tcd) get_bitfield32((const __le32 *)tcd, \
                        VMXNET3_TCD_TXIDX_SHIFT, VMXNET3_TCD_TXIDX_SIZE)
#   define vmxnet3_getRxComp(dstrcd, rcd, tmp) do { \
                        (dstrcd) = (tmp); \
                        vmxnet3_RxCompToCPU((rcd), (tmp)); \
                } while (0)
#   define vmxnet3_getRxDesc(dstrxd, rxd, tmp) do { \
                        (dstrxd) = (tmp); \
                        vmxnet3_RxDescToCPU((rxd), (tmp)); \
                } while (0)

#else

#   define VMXNET3_TXDESC_GET_GEN(txdesc) ((txdesc)->gen)
#   define VMXNET3_TXDESC_GET_EOP(txdesc) ((txdesc)->eop)
#   define VMXNET3_TCD_GET_GEN(tcd) ((tcd)->gen)
#   define VMXNET3_TCD_GET_TXIDX(tcd) ((tcd)->txdIdx)
#   define vmxnet3_getRxComp(dstrcd, rcd, tmp) (dstrcd) = (rcd)
#   define vmxnet3_getRxDesc(dstrxd, rxd, tmp) (dstrxd) = (rxd)

#endif /* __BIG_ENDIAN_BITFIELD  */


static void
vmxnet3_unmap_tx_buf(struct vmxnet3_tx_buf_info *tbi,
                     struct pci_dev *pdev)
{
        if (tbi->map_type == VMXNET3_MAP_SINGLE)
                dma_unmap_single(&pdev->dev, tbi->dma_addr, tbi->len,
                                 DMA_TO_DEVICE);
        else if (tbi->map_type == VMXNET3_MAP_PAGE)
                dma_unmap_page(&pdev->dev, tbi->dma_addr, tbi->len,
                               DMA_TO_DEVICE);
        else
                BUG_ON(tbi->map_type != VMXNET3_MAP_NONE);

        tbi->map_type = VMXNET3_MAP_NONE; /* to help debugging */
}


static int
vmxnet3_unmap_pkt(u32 eop_idx, struct vmxnet3_tx_queue *tq,
                  struct pci_dev *pdev, struct vmxnet3_adapter *adapter)
{
        struct sk_buff *skb;
        int entries = 0;

        /* no out of order completion */
        BUG_ON(tq->buf_info[eop_idx].sop_idx != tq->tx_ring.next2comp);
        BUG_ON(VMXNET3_TXDESC_GET_EOP(&(tq->tx_ring.base[eop_idx].txd)) != 1);

        skb = tq->buf_info[eop_idx].skb;
        BUG_ON(skb == NULL);
        tq->buf_info[eop_idx].skb = NULL;

        VMXNET3_INC_RING_IDX_ONLY(eop_idx, tq->tx_ring.size);

        while (tq->tx_ring.next2comp != eop_idx) {
                vmxnet3_unmap_tx_buf(tq->buf_info + tq->tx_ring.next2comp,
                                     pdev);

                /* update next2comp w/o tx_lock. Since we are marking more,
                 * instead of less, tx ring entries avail, the worst case is
                 * that the tx routine incorrectly re-queues a pkt due to
                 * insufficient tx ring entries.
                 */
                vmxnet3_cmd_ring_adv_next2comp(&tq->tx_ring);
                entries++;
        }

        dev_kfree_skb_any(skb);
        return entries;
}


static int
vmxnet3_tq_tx_complete(struct vmxnet3_tx_queue *tq,
                        struct vmxnet3_adapter *adapter)
{
        int completed = 0;
        union Vmxnet3_GenericDesc *gdesc;

        gdesc = tq->comp_ring.base + tq->comp_ring.next2proc;
        while (VMXNET3_TCD_GET_GEN(&gdesc->tcd) == tq->comp_ring.gen) {
                /* Prevent any &gdesc->tcd field from being (speculatively)
                 * read before (&gdesc->tcd)->gen is read.
                 */
                dma_rmb();

                completed += vmxnet3_unmap_pkt(VMXNET3_TCD_GET_TXIDX(
                                               &gdesc->tcd), tq, adapter->pdev,
                                               adapter);

                vmxnet3_comp_ring_adv_next2proc(&tq->comp_ring);
                gdesc = tq->comp_ring.base + tq->comp_ring.next2proc;
        }

        if (completed) {
                spin_lock(&tq->tx_lock);
                if (unlikely(vmxnet3_tq_stopped(tq, adapter) &&
                             vmxnet3_cmd_ring_desc_avail(&tq->tx_ring) >
                             VMXNET3_WAKE_QUEUE_THRESHOLD(tq) &&
                             netif_carrier_ok(adapter->netdev))) {
                        vmxnet3_tq_wake(tq, adapter);
                }
                spin_unlock(&tq->tx_lock);
        }
        return completed;
}


static void
vmxnet3_tq_cleanup(struct vmxnet3_tx_queue *tq,
                   struct vmxnet3_adapter *adapter)
{
        int i;

        while (tq->tx_ring.next2comp != tq->tx_ring.next2fill) {
                struct vmxnet3_tx_buf_info *tbi;

                tbi = tq->buf_info + tq->tx_ring.next2comp;

                vmxnet3_unmap_tx_buf(tbi, adapter->pdev);
                if (tbi->skb) {
                        dev_kfree_skb_any(tbi->skb);
                        tbi->skb = NULL;
                }
                vmxnet3_cmd_ring_adv_next2comp(&tq->tx_ring);
        }

        /* sanity check, verify all buffers are indeed unmapped and freed */
        for (i = 0; i < tq->tx_ring.size; i++) {
                BUG_ON(tq->buf_info[i].skb != NULL ||
                       tq->buf_info[i].map_type != VMXNET3_MAP_NONE);
        }

        tq->tx_ring.gen = VMXNET3_INIT_GEN;
        tq->tx_ring.next2fill = tq->tx_ring.next2comp = 0;

        tq->comp_ring.gen = VMXNET3_INIT_GEN;
        tq->comp_ring.next2proc = 0;
}


static void
vmxnet3_tq_destroy(struct vmxnet3_tx_queue *tq,
                   struct vmxnet3_adapter *adapter)
{
        if (tq->tx_ring.base) {
                dma_free_coherent(&adapter->pdev->dev, tq->tx_ring.size *
                                  sizeof(struct Vmxnet3_TxDesc),
                                  tq->tx_ring.base, tq->tx_ring.basePA);
                tq->tx_ring.base = NULL;
        }
        if (tq->data_ring.base) {
                dma_free_coherent(&adapter->pdev->dev,
                                  tq->data_ring.size * tq->txdata_desc_size,
                                  tq->data_ring.base, tq->data_ring.basePA);
                tq->data_ring.base = NULL;
        }
        if (tq->comp_ring.base) {
                dma_free_coherent(&adapter->pdev->dev, tq->comp_ring.size *
                                  sizeof(struct Vmxnet3_TxCompDesc),
                                  tq->comp_ring.base, tq->comp_ring.basePA);
                tq->comp_ring.base = NULL;
        }
        kfree(tq->buf_info);
        tq->buf_info = NULL;
}


/* Destroy all tx queues */
void
vmxnet3_tq_destroy_all(struct vmxnet3_adapter *adapter)
{
        int i;

        for (i = 0; i < adapter->num_tx_queues; i++)
                vmxnet3_tq_destroy(&adapter->tx_queue[i], adapter);
}


static void
vmxnet3_tq_init(struct vmxnet3_tx_queue *tq,
                struct vmxnet3_adapter *adapter)
{
        int i;

        /* reset the tx ring contents to 0 and reset the tx ring states */
        memset(tq->tx_ring.base, 0, tq->tx_ring.size *
               sizeof(struct Vmxnet3_TxDesc));
        tq->tx_ring.next2fill = tq->tx_ring.next2comp = 0;
        tq->tx_ring.gen = VMXNET3_INIT_GEN;

        memset(tq->data_ring.base, 0,
               tq->data_ring.size * tq->txdata_desc_size);

        /* reset the tx comp ring contents to 0 and reset comp ring states */
        memset(tq->comp_ring.base, 0, tq->comp_ring.size *
               sizeof(struct Vmxnet3_TxCompDesc));
        tq->comp_ring.next2proc = 0;
        tq->comp_ring.gen = VMXNET3_INIT_GEN;

        /* reset the bookkeeping data */
        memset(tq->buf_info, 0, sizeof(tq->buf_info[0]) * tq->tx_ring.size);
        for (i = 0; i < tq->tx_ring.size; i++)
                tq->buf_info[i].map_type = VMXNET3_MAP_NONE;

        /* stats are not reset */
}


static int
vmxnet3_tq_create(struct vmxnet3_tx_queue *tq,
                  struct vmxnet3_adapter *adapter)
{
        BUG_ON(tq->tx_ring.base || tq->data_ring.base ||
               tq->comp_ring.base || tq->buf_info);

        tq->tx_ring.base = dma_alloc_coherent(&adapter->pdev->dev,
                        tq->tx_ring.size * sizeof(struct Vmxnet3_TxDesc),
                        &tq->tx_ring.basePA, GFP_KERNEL);
        if (!tq->tx_ring.base) {
                netdev_err(adapter->netdev, "failed to allocate tx ring\n");
                goto err;
        }

        tq->data_ring.base = dma_alloc_coherent(&adapter->pdev->dev,
                        tq->data_ring.size * tq->txdata_desc_size,
                        &tq->data_ring.basePA, GFP_KERNEL);
        if (!tq->data_ring.base) {
                netdev_err(adapter->netdev, "failed to allocate tx data ring\n");
                goto err;
        }

        tq->comp_ring.base = dma_alloc_coherent(&adapter->pdev->dev,
                        tq->comp_ring.size * sizeof(struct Vmxnet3_TxCompDesc),
                        &tq->comp_ring.basePA, GFP_KERNEL);
        if (!tq->comp_ring.base) {
                netdev_err(adapter->netdev, "failed to allocate tx comp ring\n");
                goto err;
        }

        tq->buf_info = kcalloc_node(tq->tx_ring.size, sizeof(tq->buf_info[0]),
                                    GFP_KERNEL,
                                    dev_to_node(&adapter->pdev->dev));
        if (!tq->buf_info)
                goto err;

        return 0;

err:
        vmxnet3_tq_destroy(tq, adapter);
        return -ENOMEM;
}

static void
vmxnet3_tq_cleanup_all(struct vmxnet3_adapter *adapter)
{
        int i;

        for (i = 0; i < adapter->num_tx_queues; i++)
                vmxnet3_tq_cleanup(&adapter->tx_queue[i], adapter);
}

/*
 *    starting from ring->next2fill, allocate rx buffers for the given ring
 *    of the rx queue and update the rx desc. stop after @num_to_alloc buffers
 *    are allocated or allocation fails
 */

static int
vmxnet3_rq_alloc_rx_buf(struct vmxnet3_rx_queue *rq, u32 ring_idx,
                        int num_to_alloc, struct vmxnet3_adapter *adapter)
{
        int num_allocated = 0;
        struct vmxnet3_rx_buf_info *rbi_base = rq->buf_info[ring_idx];
        struct vmxnet3_cmd_ring *ring = &rq->rx_ring[ring_idx];
        u32 val;

        while (num_allocated <= num_to_alloc) {
                struct vmxnet3_rx_buf_info *rbi;
                union Vmxnet3_GenericDesc *gd;

                rbi = rbi_base + ring->next2fill;
                gd = ring->base + ring->next2fill;

                if (rbi->buf_type == VMXNET3_RX_BUF_SKB) {
                        if (rbi->skb == NULL) {
                                rbi->skb = __netdev_alloc_skb_ip_align(adapter->netdev,
                                                                       rbi->len,
                                                                       GFP_KERNEL);
                                if (unlikely(rbi->skb == NULL)) {
                                        rq->stats.rx_buf_alloc_failure++;
                                        break;
                                }

                                rbi->dma_addr = dma_map_single(
                                                &adapter->pdev->dev,
                                                rbi->skb->data, rbi->len,
                                                DMA_FROM_DEVICE);
                                if (dma_mapping_error(&adapter->pdev->dev,
                                                      rbi->dma_addr)) {
                                        dev_kfree_skb_any(rbi->skb);
                                        rq->stats.rx_buf_alloc_failure++;
                                        break;
                                }
                        } else {
                                /* rx buffer skipped by the device */
                        }
                        val = VMXNET3_RXD_BTYPE_HEAD << VMXNET3_RXD_BTYPE_SHIFT;
                } else {
                        BUG_ON(rbi->buf_type != VMXNET3_RX_BUF_PAGE ||
                               rbi->len  != PAGE_SIZE);

                        if (rbi->page == NULL) {
                                rbi->page = alloc_page(GFP_ATOMIC);
                                if (unlikely(rbi->page == NULL)) {
                                        rq->stats.rx_buf_alloc_failure++;
                                        break;
                                }
                                rbi->dma_addr = dma_map_page(
                                                &adapter->pdev->dev,
                                                rbi->page, 0, PAGE_SIZE,
                                                DMA_FROM_DEVICE);
                                if (dma_mapping_error(&adapter->pdev->dev,
                                                      rbi->dma_addr)) {
                                        put_page(rbi->page);
                                        rq->stats.rx_buf_alloc_failure++;
                                        break;
                                }
                        } else {
                                /* rx buffers skipped by the device */
                        }
                        val = VMXNET3_RXD_BTYPE_BODY << VMXNET3_RXD_BTYPE_SHIFT;
                }

                gd->rxd.addr = cpu_to_le64(rbi->dma_addr);
                gd->dword[2] = cpu_to_le32((!ring->gen << VMXNET3_RXD_GEN_SHIFT)
                                           | val | rbi->len);

                /* Fill the last buffer but dont mark it ready, or else the
                 * device will think that the queue is full */
                if (num_allocated == num_to_alloc)
                        break;

                gd->dword[2] |= cpu_to_le32(ring->gen << VMXNET3_RXD_GEN_SHIFT);
                num_allocated++;
                vmxnet3_cmd_ring_adv_next2fill(ring);
        }

        netdev_dbg(adapter->netdev,
                "alloc_rx_buf: %d allocated, next2fill %u, next2comp %u\n",
                num_allocated, ring->next2fill, ring->next2comp);

        /* so that the device can distinguish a full ring and an empty ring */
        BUG_ON(num_allocated != 0 && ring->next2fill == ring->next2comp);

        return num_allocated;
}


static void
vmxnet3_append_frag(struct sk_buff *skb, struct Vmxnet3_RxCompDesc *rcd,
                    struct vmxnet3_rx_buf_info *rbi)
{
        skb_frag_t *frag = skb_shinfo(skb)->frags + skb_shinfo(skb)->nr_frags;

        BUG_ON(skb_shinfo(skb)->nr_frags >= MAX_SKB_FRAGS);

        __skb_frag_set_page(frag, rbi->page);
        skb_frag_off_set(frag, 0);
        skb_frag_size_set(frag, rcd->len);
        skb->data_len += rcd->len;
        skb->truesize += PAGE_SIZE;
        skb_shinfo(skb)->nr_frags++;
}


static int
vmxnet3_map_pkt(struct sk_buff *skb, struct vmxnet3_tx_ctx *ctx,
                struct vmxnet3_tx_queue *tq, struct pci_dev *pdev,
                struct vmxnet3_adapter *adapter)
{
        u32 dw2, len;
        unsigned long buf_offset;
        int i;
        union Vmxnet3_GenericDesc *gdesc;
        struct vmxnet3_tx_buf_info *tbi = NULL;

        BUG_ON(ctx->copy_size > skb_headlen(skb));

        /* use the previous gen bit for the SOP desc */
        dw2 = (tq->tx_ring.gen ^ 0x1) << VMXNET3_TXD_GEN_SHIFT;

        ctx->sop_txd = tq->tx_ring.base + tq->tx_ring.next2fill;
        gdesc = ctx->sop_txd; /* both loops below can be skipped */

        /* no need to map the buffer if headers are copied */
        if (ctx->copy_size) {
                ctx->sop_txd->txd.addr = cpu_to_le64(tq->data_ring.basePA +
                                        tq->tx_ring.next2fill *
                                        tq->txdata_desc_size);
                ctx->sop_txd->dword[2] = cpu_to_le32(dw2 | ctx->copy_size);
                ctx->sop_txd->dword[3] = 0;

                tbi = tq->buf_info + tq->tx_ring.next2fill;
                tbi->map_type = VMXNET3_MAP_NONE;

                netdev_dbg(adapter->netdev,
                        "txd[%u]: 0x%Lx 0x%x 0x%x\n",
                        tq->tx_ring.next2fill,
                        le64_to_cpu(ctx->sop_txd->txd.addr),
                        ctx->sop_txd->dword[2], ctx->sop_txd->dword[3]);
                vmxnet3_cmd_ring_adv_next2fill(&tq->tx_ring);

                /* use the right gen for non-SOP desc */
                dw2 = tq->tx_ring.gen << VMXNET3_TXD_GEN_SHIFT;
        }

        /* linear part can use multiple tx desc if it's big */
        len = skb_headlen(skb) - ctx->copy_size;
        buf_offset = ctx->copy_size;
        while (len) {
                u32 buf_size;

                if (len < VMXNET3_MAX_TX_BUF_SIZE) {
                        buf_size = len;
                        dw2 |= len;
                } else {
                        buf_size = VMXNET3_MAX_TX_BUF_SIZE;
                        /* spec says that for TxDesc.len, 0 == 2^14 */
                }

                tbi = tq->buf_info + tq->tx_ring.next2fill;
                tbi->map_type = VMXNET3_MAP_SINGLE;
                tbi->dma_addr = dma_map_single(&adapter->pdev->dev,
                                skb->data + buf_offset, buf_size,
                                DMA_TO_DEVICE);
                if (dma_mapping_error(&adapter->pdev->dev, tbi->dma_addr))
                        return -EFAULT;

                tbi->len = buf_size;

                gdesc = tq->tx_ring.base + tq->tx_ring.next2fill;
                BUG_ON(gdesc->txd.gen == tq->tx_ring.gen);

                gdesc->txd.addr = cpu_to_le64(tbi->dma_addr);
                gdesc->dword[2] = cpu_to_le32(dw2);
                gdesc->dword[3] = 0;

                netdev_dbg(adapter->netdev,
                        "txd[%u]: 0x%Lx 0x%x 0x%x\n",
                        tq->tx_ring.next2fill, le64_to_cpu(gdesc->txd.addr),
                        le32_to_cpu(gdesc->dword[2]), gdesc->dword[3]);
                vmxnet3_cmd_ring_adv_next2fill(&tq->tx_ring);
                dw2 = tq->tx_ring.gen << VMXNET3_TXD_GEN_SHIFT;

                len -= buf_size;
                buf_offset += buf_size;
        }

        for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
                const skb_frag_t *frag = &skb_shinfo(skb)->frags[i];
                u32 buf_size;

                buf_offset = 0;
                len = skb_frag_size(frag);
                while (len) {
                        tbi = tq->buf_info + tq->tx_ring.next2fill;
                        if (len < VMXNET3_MAX_TX_BUF_SIZE) {
                                buf_size = len;
                                dw2 |= len;
                        } else {
                                buf_size = VMXNET3_MAX_TX_BUF_SIZE;
                                /* spec says that for TxDesc.len, 0 == 2^14 */
                        }
                        tbi->map_type = VMXNET3_MAP_PAGE;
                        tbi->dma_addr = skb_frag_dma_map(&adapter->pdev->dev, frag,
                                                         buf_offset, buf_size,
                                                         DMA_TO_DEVICE);
                        if (dma_mapping_error(&adapter->pdev->dev, tbi->dma_addr))
                                return -EFAULT;

                        tbi->len = buf_size;

                        gdesc = tq->tx_ring.base + tq->tx_ring.next2fill;
                        BUG_ON(gdesc->txd.gen == tq->tx_ring.gen);

                        gdesc->txd.addr = cpu_to_le64(tbi->dma_addr);
                        gdesc->dword[2] = cpu_to_le32(dw2);
                        gdesc->dword[3] = 0;

                        netdev_dbg(adapter->netdev,
                                "txd[%u]: 0x%llx %u %u\n",
                                tq->tx_ring.next2fill, le64_to_cpu(gdesc->txd.addr),
                                le32_to_cpu(gdesc->dword[2]), gdesc->dword[3]);
                        vmxnet3_cmd_ring_adv_next2fill(&tq->tx_ring);
                        dw2 = tq->tx_ring.gen << VMXNET3_TXD_GEN_SHIFT;

                        len -= buf_size;
                        buf_offset += buf_size;
                }
        }

        ctx->eop_txd = gdesc;

        /* set the last buf_info for the pkt */
        tbi->skb = skb;
        tbi->sop_idx = ctx->sop_txd - tq->tx_ring.base;

        return 0;
}


/* Init all tx queues */
static void
vmxnet3_tq_init_all(struct vmxnet3_adapter *adapter)
{
        int i;

        for (i = 0; i < adapter->num_tx_queues; i++)
                vmxnet3_tq_init(&adapter->tx_queue[i], adapter);
}


/*
 *    parse relevant protocol headers:
 *      For a tso pkt, relevant headers are L2/3/4 including options
 *      For a pkt requesting csum offloading, they are L2/3 and may include L4
 *      if it's a TCP/UDP pkt
 *
 * Returns:
 *    -1:  error happens during parsing
 *     0:  protocol headers parsed, but too big to be copied
 *     1:  protocol headers parsed and copied
 *
 * Other effects:
 *    1. related *ctx fields are updated.
 *    2. ctx->copy_size is # of bytes copied
 *    3. the portion to be copied is guaranteed to be in the linear part
 *
 */
static int
vmxnet3_parse_hdr(struct sk_buff *skb, struct vmxnet3_tx_queue *tq,
                  struct vmxnet3_tx_ctx *ctx,
                  struct vmxnet3_adapter *adapter)
{
        u8 protocol = 0;

        if (ctx->mss) { /* TSO */
                if (VMXNET3_VERSION_GE_4(adapter) && skb->encapsulation) {
                        ctx->l4_offset = skb_inner_transport_offset(skb);
                        ctx->l4_hdr_size = inner_tcp_hdrlen(skb);
                        ctx->copy_size = ctx->l4_offset + ctx->l4_hdr_size;
                } else {
                        ctx->l4_offset = skb_transport_offset(skb);
                        ctx->l4_hdr_size = tcp_hdrlen(skb);
                        ctx->copy_size = ctx->l4_offset + ctx->l4_hdr_size;
                }
        } else {
                if (skb->ip_summed == CHECKSUM_PARTIAL) {
                        /* For encap packets, skb_checksum_start_offset refers
                         * to inner L4 offset. Thus, below works for encap as
                         * well as non-encap case
                         */
                        ctx->l4_offset = skb_checksum_start_offset(skb);

                        if (VMXNET3_VERSION_GE_4(adapter) &&
                            skb->encapsulation) {
                                struct iphdr *iph = inner_ip_hdr(skb);

                                if (iph->version == 4) {
                                        protocol = iph->protocol;
                                } else {
                                        const struct ipv6hdr *ipv6h;

                                        ipv6h = inner_ipv6_hdr(skb);
                                        protocol = ipv6h->nexthdr;
                                }
                        } else {
                                if (ctx->ipv4) {
                                        const struct iphdr *iph = ip_hdr(skb);

                                        protocol = iph->protocol;
                                } else if (ctx->ipv6) {
                                        const struct ipv6hdr *ipv6h;

                                        ipv6h = ipv6_hdr(skb);
                                        protocol = ipv6h->nexthdr;
                                }
                        }

                        switch (protocol) {
                        case IPPROTO_TCP:
                                ctx->l4_hdr_size = skb->encapsulation ? inner_tcp_hdrlen(skb) :
                                                   tcp_hdrlen(skb);
                                break;
                        case IPPROTO_UDP:
                                ctx->l4_hdr_size = sizeof(struct udphdr);
                                break;
                        default:
                                ctx->l4_hdr_size = 0;
                                break;
                        }

                        ctx->copy_size = min(ctx->l4_offset +
                                         ctx->l4_hdr_size, skb->len);
                } else {
                        ctx->l4_offset = 0;
                        ctx->l4_hdr_size = 0;
                        /* copy as much as allowed */
                        ctx->copy_size = min_t(unsigned int,
                                               tq->txdata_desc_size,
                                               skb_headlen(skb));
                }

                if (skb->len <= VMXNET3_HDR_COPY_SIZE)
                        ctx->copy_size = skb->len;

                /* make sure headers are accessible directly */
                if (unlikely(!pskb_may_pull(skb, ctx->copy_size)))
                        goto err;
        }

        if (unlikely(ctx->copy_size > tq->txdata_desc_size)) {
                tq->stats.oversized_hdr++;
                ctx->copy_size = 0;
                return 0;
        }

        return 1;
err:
        return -1;
}

/*
 *    copy relevant protocol headers to the transmit ring:
 *      For a tso pkt, relevant headers are L2/3/4 including options
 *      For a pkt requesting csum offloading, they are L2/3 and may include L4
 *      if it's a TCP/UDP pkt
 *
 *
 *    Note that this requires that vmxnet3_parse_hdr be called first to set the
 *      appropriate bits in ctx first
 */
static void
vmxnet3_copy_hdr(struct sk_buff *skb, struct vmxnet3_tx_queue *tq,
                 struct vmxnet3_tx_ctx *ctx,
                 struct vmxnet3_adapter *adapter)
{
        struct Vmxnet3_TxDataDesc *tdd;

        tdd = (struct Vmxnet3_TxDataDesc *)((u8 *)tq->data_ring.base +
                                            tq->tx_ring.next2fill *
                                            tq->txdata_desc_size);

        memcpy(tdd->data, skb->data, ctx->copy_size);
        netdev_dbg(adapter->netdev,
                "copy %u bytes to dataRing[%u]\n",
                ctx->copy_size, tq->tx_ring.next2fill);
}


static void
vmxnet3_prepare_inner_tso(struct sk_buff *skb,
                          struct vmxnet3_tx_ctx *ctx)
{
        struct tcphdr *tcph = inner_tcp_hdr(skb);
        struct iphdr *iph = inner_ip_hdr(skb);

        if (iph->version == 4) {
                iph->check = 0;
                tcph->check = ~csum_tcpudp_magic(iph->saddr, iph->daddr, 0,
                                                 IPPROTO_TCP, 0);
        } else {
                struct ipv6hdr *iph = inner_ipv6_hdr(skb);

                tcph->check = ~csum_ipv6_magic(&iph->saddr, &iph->daddr, 0,
                                               IPPROTO_TCP, 0);
        }
}

static void
vmxnet3_prepare_tso(struct sk_buff *skb,
                    struct vmxnet3_tx_ctx *ctx)
{
        struct tcphdr *tcph = tcp_hdr(skb);

        if (ctx->ipv4) {
                struct iphdr *iph = ip_hdr(skb);

                iph->check = 0;
                tcph->check = ~csum_tcpudp_magic(iph->saddr, iph->daddr, 0,
                                                 IPPROTO_TCP, 0);
        } else if (ctx->ipv6) {
                tcp_v6_gso_csum_prep(skb);
        }
}

static int txd_estimate(const struct sk_buff *skb)
{
        int count = VMXNET3_TXD_NEEDED(skb_headlen(skb)) + 1;
        int i;

        for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
                const skb_frag_t *frag = &skb_shinfo(skb)->frags[i];

                count += VMXNET3_TXD_NEEDED(skb_frag_size(frag));
        }
        return count;
}

/*
 * Transmits a pkt thru a given tq
 * Returns:
 *    NETDEV_TX_OK:      descriptors are setup successfully
 *    NETDEV_TX_OK:      error occurred, the pkt is dropped
 *    NETDEV_TX_BUSY:    tx ring is full, queue is stopped
 *
 * Side-effects:
 *    1. tx ring may be changed
 *    2. tq stats may be updated accordingly
 *    3. shared->txNumDeferred may be updated
 */

static int
vmxnet3_tq_xmit(struct sk_buff *skb, struct vmxnet3_tx_queue *tq,
                struct vmxnet3_adapter *adapter, struct net_device *netdev)
{
        int ret;
        u32 count;
        int num_pkts;
        int tx_num_deferred;
        unsigned long flags;
        struct vmxnet3_tx_ctx ctx;
        union Vmxnet3_GenericDesc *gdesc;
#ifdef __BIG_ENDIAN_BITFIELD
        /* Use temporary descriptor to avoid touching bits multiple times */
        union Vmxnet3_GenericDesc tempTxDesc;
#endif

        count = txd_estimate(skb);

        ctx.ipv4 = (vlan_get_protocol(skb) == cpu_to_be16(ETH_P_IP));
        ctx.ipv6 = (vlan_get_protocol(skb) == cpu_to_be16(ETH_P_IPV6));

        ctx.mss = skb_shinfo(skb)->gso_size;
        if (ctx.mss) {
                if (skb_header_cloned(skb)) {
                        if (unlikely(pskb_expand_head(skb, 0, 0,
                                                      GFP_ATOMIC) != 0)) {
                                tq->stats.drop_tso++;
                                goto drop_pkt;
                        }
                        tq->stats.copy_skb_header++;
                }
                if (skb->encapsulation) {
                        vmxnet3_prepare_inner_tso(skb, &ctx);
                } else {
                        vmxnet3_prepare_tso(skb, &ctx);
                }
        } else {
                if (unlikely(count > VMXNET3_MAX_TXD_PER_PKT)) {

                        /* non-tso pkts must not use more than
                         * VMXNET3_MAX_TXD_PER_PKT entries
                         */
                        if (skb_linearize(skb) != 0) {
                                tq->stats.drop_too_many_frags++;
                                goto drop_pkt;
                        }
                        tq->stats.linearized++;

                        /* recalculate the # of descriptors to use */
                        count = VMXNET3_TXD_NEEDED(skb_headlen(skb)) + 1;
                }
        }

        ret = vmxnet3_parse_hdr(skb, tq, &ctx, adapter);
        if (ret >= 0) {
                BUG_ON(ret <= 0 && ctx.copy_size != 0);
                /* hdrs parsed, check against other limits */
                if (ctx.mss) {
                        if (unlikely(ctx.l4_offset + ctx.l4_hdr_size >
                                     VMXNET3_MAX_TX_BUF_SIZE)) {
                                tq->stats.drop_oversized_hdr++;
                                goto drop_pkt;
                        }
                } else {
                        if (skb->ip_summed == CHECKSUM_PARTIAL) {
                                if (unlikely(ctx.l4_offset +
                                             skb->csum_offset >
                                             VMXNET3_MAX_CSUM_OFFSET)) {
                                        tq->stats.drop_oversized_hdr++;
                                        goto drop_pkt;
                                }
                        }
                }
        } else {
                tq->stats.drop_hdr_inspect_err++;
                goto drop_pkt;
        }

        spin_lock_irqsave(&tq->tx_lock, flags);

        if (count > vmxnet3_cmd_ring_desc_avail(&tq->tx_ring)) {
                tq->stats.tx_ring_full++;
                netdev_dbg(adapter->netdev,
                        "tx queue stopped on %s, next2comp %u"
                        " next2fill %u\n", adapter->netdev->name,
                        tq->tx_ring.next2comp, tq->tx_ring.next2fill);

                vmxnet3_tq_stop(tq, adapter);
                spin_unlock_irqrestore(&tq->tx_lock, flags);
                return NETDEV_TX_BUSY;
        }


        vmxnet3_copy_hdr(skb, tq, &ctx, adapter);

        /* fill tx descs related to addr & len */
        if (vmxnet3_map_pkt(skb, &ctx, tq, adapter->pdev, adapter))
                goto unlock_drop_pkt;

        /* setup the EOP desc */
        ctx.eop_txd->dword[3] = cpu_to_le32(VMXNET3_TXD_CQ | VMXNET3_TXD_EOP);

        /* setup the SOP desc */
#ifdef __BIG_ENDIAN_BITFIELD
        gdesc = &tempTxDesc;
        gdesc->dword[2] = ctx.sop_txd->dword[2];
        gdesc->dword[3] = ctx.sop_txd->dword[3];
#else
        gdesc = ctx.sop_txd;
#endif
        tx_num_deferred = le32_to_cpu(tq->shared->txNumDeferred);
        if (ctx.mss) {
                if (VMXNET3_VERSION_GE_4(adapter) && skb->encapsulation) {
                        gdesc->txd.hlen = ctx.l4_offset + ctx.l4_hdr_size;
                        gdesc->txd.om = VMXNET3_OM_ENCAP;
                        gdesc->txd.msscof = ctx.mss;

                        if (skb_shinfo(skb)->gso_type & SKB_GSO_UDP_TUNNEL_CSUM)
                                gdesc->txd.oco = 1;
                } else {
                        gdesc->txd.hlen = ctx.l4_offset + ctx.l4_hdr_size;
                        gdesc->txd.om = VMXNET3_OM_TSO;
                        gdesc->txd.msscof = ctx.mss;
                }
                num_pkts = (skb->len - gdesc->txd.hlen + ctx.mss - 1) / ctx.mss;
        } else {
                if (skb->ip_summed == CHECKSUM_PARTIAL) {
                        if (VMXNET3_VERSION_GE_4(adapter) &&
                            skb->encapsulation) {
                                gdesc->txd.hlen = ctx.l4_offset +
                                                  ctx.l4_hdr_size;
                                gdesc->txd.om = VMXNET3_OM_ENCAP;
                                gdesc->txd.msscof = 0;          /* Reserved */
                        } else {
                                gdesc->txd.hlen = ctx.l4_offset;
                                gdesc->txd.om = VMXNET3_OM_CSUM;
                                gdesc->txd.msscof = ctx.l4_offset +
                                                    skb->csum_offset;
                        }
                } else {
                        gdesc->txd.om = 0;
                        gdesc->txd.msscof = 0;
                }
                num_pkts = 1;
        }
        le32_add_cpu(&tq->shared->txNumDeferred, num_pkts);
        tx_num_deferred += num_pkts;

        if (skb_vlan_tag_present(skb)) {
                gdesc->txd.ti = 1;
                gdesc->txd.tci = skb_vlan_tag_get(skb);
        }

        /* Ensure that the write to (&gdesc->txd)->gen will be observed after
         * all other writes to &gdesc->txd.
         */
        dma_wmb();

        /* finally flips the GEN bit of the SOP desc. */
        gdesc->dword[2] = cpu_to_le32(le32_to_cpu(gdesc->dword[2]) ^
                                                  VMXNET3_TXD_GEN);
#ifdef __BIG_ENDIAN_BITFIELD
        /* Finished updating in bitfields of Tx Desc, so write them in original
         * place.
         */
        vmxnet3_TxDescToLe((struct Vmxnet3_TxDesc *)gdesc,
                           (struct Vmxnet3_TxDesc *)ctx.sop_txd);
        gdesc = ctx.sop_txd;
#endif
        netdev_dbg(adapter->netdev,
                "txd[%u]: SOP 0x%Lx 0x%x 0x%x\n",
                (u32)(ctx.sop_txd -
                tq->tx_ring.base), le64_to_cpu(gdesc->txd.addr),
                le32_to_cpu(gdesc->dword[2]), le32_to_cpu(gdesc->dword[3]));

        spin_unlock_irqrestore(&tq->tx_lock, flags);

        if (tx_num_deferred >= le32_to_cpu(tq->shared->txThreshold)) {
                tq->shared->txNumDeferred = 0;
                VMXNET3_WRITE_BAR0_REG(adapter,
                                       VMXNET3_REG_TXPROD + tq->qid * 8,
                                       tq->tx_ring.next2fill);
        }

        return NETDEV_TX_OK;

unlock_drop_pkt:
        spin_unlock_irqrestore(&tq->tx_lock, flags);
drop_pkt:
        tq->stats.drop_total++;
        dev_kfree_skb_any(skb);
        return NETDEV_TX_OK;
}


static netdev_tx_t
vmxnet3_xmit_frame(struct sk_buff *skb, struct net_device *netdev)
{
        struct vmxnet3_adapter *adapter = netdev_priv(netdev);

        BUG_ON(skb->queue_mapping > adapter->num_tx_queues);
        return vmxnet3_tq_xmit(skb,
                               &adapter->tx_queue[skb->queue_mapping],
                               adapter, netdev);
}


static void
vmxnet3_rx_csum(struct vmxnet3_adapter *adapter,
                struct sk_buff *skb,
                union Vmxnet3_GenericDesc *gdesc)
{
        if (!gdesc->rcd.cnc && adapter->netdev->features & NETIF_F_RXCSUM) {
                if (gdesc->rcd.v4 &&
                    (le32_to_cpu(gdesc->dword[3]) &
                     VMXNET3_RCD_CSUM_OK) == VMXNET3_RCD_CSUM_OK) {
                        skb->ip_summed = CHECKSUM_UNNECESSARY;
                        WARN_ON_ONCE(!(gdesc->rcd.tcp || gdesc->rcd.udp) &&
                                     !(le32_to_cpu(gdesc->dword[0]) &
                                     (1UL << VMXNET3_RCD_HDR_INNER_SHIFT)));
                        WARN_ON_ONCE(gdesc->rcd.frg &&
                                     !(le32_to_cpu(gdesc->dword[0]) &
                                     (1UL << VMXNET3_RCD_HDR_INNER_SHIFT)));
                } else if (gdesc->rcd.v6 && (le32_to_cpu(gdesc->dword[3]) &
                                             (1 << VMXNET3_RCD_TUC_SHIFT))) {
                        skb->ip_summed = CHECKSUM_UNNECESSARY;
                        WARN_ON_ONCE(!(gdesc->rcd.tcp || gdesc->rcd.udp) &&
                                     !(le32_to_cpu(gdesc->dword[0]) &
                                     (1UL << VMXNET3_RCD_HDR_INNER_SHIFT)));
                        WARN_ON_ONCE(gdesc->rcd.frg &&
                                     !(le32_to_cpu(gdesc->dword[0]) &
                                     (1UL << VMXNET3_RCD_HDR_INNER_SHIFT)));
                } else {
                        if (gdesc->rcd.csum) {
                                skb->csum = htons(gdesc->rcd.csum);
                                skb->ip_summed = CHECKSUM_PARTIAL;
                        } else {
                                skb_checksum_none_assert(skb);
                        }
                }
        } else {
                skb_checksum_none_assert(skb);
        }
}


static void
vmxnet3_rx_error(struct vmxnet3_rx_queue *rq, struct Vmxnet3_RxCompDesc *rcd,
                 struct vmxnet3_rx_ctx *ctx,  struct vmxnet3_adapter *adapter)
{
        rq->stats.drop_err++;
        if (!rcd->fcs)
                rq->stats.drop_fcs++;

        rq->stats.drop_total++;

        /*
         * We do not unmap and chain the rx buffer to the skb.
         * We basically pretend this buffer is not used and will be recycled
         * by vmxnet3_rq_alloc_rx_buf()
         */

        /*
         * ctx->skb may be NULL if this is the first and the only one
         * desc for the pkt
         */
        if (ctx->skb)
                dev_kfree_skb_irq(ctx->skb);

        ctx->skb = NULL;
}


static u32
vmxnet3_get_hdr_len(struct vmxnet3_adapter *adapter, struct sk_buff *skb,
                    union Vmxnet3_GenericDesc *gdesc)
{
        u32 hlen, maplen;
        union {
                void *ptr;
                struct ethhdr *eth;
                struct vlan_ethhdr *veth;
                struct iphdr *ipv4;
                struct ipv6hdr *ipv6;
                struct tcphdr *tcp;
        } hdr;
        BUG_ON(gdesc->rcd.tcp == 0);

        maplen = skb_headlen(skb);
        if (unlikely(sizeof(struct iphdr) + sizeof(struct tcphdr) > maplen))
                return 0;

        if (skb->protocol == cpu_to_be16(ETH_P_8021Q) ||
            skb->protocol == cpu_to_be16(ETH_P_8021AD))
                hlen = sizeof(struct vlan_ethhdr);
        else
                hlen = sizeof(struct ethhdr);

        hdr.eth = eth_hdr(skb);
        if (gdesc->rcd.v4) {
                BUG_ON(hdr.eth->h_proto != htons(ETH_P_IP) &&
                       hdr.veth->h_vlan_encapsulated_proto != htons(ETH_P_IP));
                hdr.ptr += hlen;
                BUG_ON(hdr.ipv4->protocol != IPPROTO_TCP);
                hlen = hdr.ipv4->ihl << 2;
                hdr.ptr += hdr.ipv4->ihl << 2;
        } else if (gdesc->rcd.v6) {
                BUG_ON(hdr.eth->h_proto != htons(ETH_P_IPV6) &&
                       hdr.veth->h_vlan_encapsulated_proto != htons(ETH_P_IPV6));
                hdr.ptr += hlen;
                /* Use an estimated value, since we also need to handle
                 * TSO case.
                 */
                if (hdr.ipv6->nexthdr != IPPROTO_TCP)
                        return sizeof(struct ipv6hdr) + sizeof(struct tcphdr);
                hlen = sizeof(struct ipv6hdr);
                hdr.ptr += sizeof(struct ipv6hdr);
        } else {
                /* Non-IP pkt, dont estimate header length */
                return 0;
        }

        if (hlen + sizeof(struct tcphdr) > maplen)
                return 0;

        return (hlen + (hdr.tcp->doff << 2));
}

static int
vmxnet3_rq_rx_complete(struct vmxnet3_rx_queue *rq,
                       struct vmxnet3_adapter *adapter, int quota)
{
        static const u32 rxprod_reg[2] = {
                VMXNET3_REG_RXPROD, VMXNET3_REG_RXPROD2
        };
        u32 num_pkts = 0;
        bool skip_page_frags = false;
        struct Vmxnet3_RxCompDesc *rcd;
        struct vmxnet3_rx_ctx *ctx = &rq->rx_ctx;
        u16 segCnt = 0, mss = 0;
#ifdef __BIG_ENDIAN_BITFIELD
        struct Vmxnet3_RxDesc rxCmdDesc;
        struct Vmxnet3_RxCompDesc rxComp;
#endif
        vmxnet3_getRxComp(rcd, &rq->comp_ring.base[rq->comp_ring.next2proc].rcd,
                          &rxComp);
        while (rcd->gen == rq->comp_ring.gen) {
                struct vmxnet3_rx_buf_info *rbi;
                struct sk_buff *skb, *new_skb = NULL;
                struct page *new_page = NULL;
                dma_addr_t new_dma_addr;
                int num_to_alloc;
                struct Vmxnet3_RxDesc *rxd;
                u32 idx, ring_idx;
                struct vmxnet3_cmd_ring *ring = NULL;
                if (num_pkts >= quota) {
                        /* we may stop even before we see the EOP desc of
                         * the current pkt
                         */
                        break;
                }

                /* Prevent any rcd field from being (speculatively) read before
                 * rcd->gen is read.
                 */
                dma_rmb();

                BUG_ON(rcd->rqID != rq->qid && rcd->rqID != rq->qid2 &&
                       rcd->rqID != rq->dataRingQid);
                idx = rcd->rxdIdx;
                ring_idx = VMXNET3_GET_RING_IDX(adapter, rcd->rqID);
                ring = rq->rx_ring + ring_idx;
                vmxnet3_getRxDesc(rxd, &rq->rx_ring[ring_idx].base[idx].rxd,
                                  &rxCmdDesc);
                rbi = rq->buf_info[ring_idx] + idx;

                BUG_ON(rxd->addr != rbi->dma_addr ||
                       rxd->len != rbi->len);

                if (unlikely(rcd->eop && rcd->err)) {
                        vmxnet3_rx_error(rq, rcd, ctx, adapter);
                        goto rcd_done;
                }

                if (rcd->sop) { /* first buf of the pkt */
                        bool rxDataRingUsed;
                        u16 len;

                        BUG_ON(rxd->btype != VMXNET3_RXD_BTYPE_HEAD ||
                               (rcd->rqID != rq->qid &&
                                rcd->rqID != rq->dataRingQid));

                        BUG_ON(rbi->buf_type != VMXNET3_RX_BUF_SKB);
                        BUG_ON(ctx->skb != NULL || rbi->skb == NULL);

                        if (unlikely(rcd->len == 0)) {
                                /* Pretend the rx buffer is skipped. */
                                BUG_ON(!(rcd->sop && rcd->eop));
                                netdev_dbg(adapter->netdev,
                                        "rxRing[%u][%u] 0 length\n",
                                        ring_idx, idx);
                                goto rcd_done;
                        }

                        skip_page_frags = false;
                        ctx->skb = rbi->skb;

                        rxDataRingUsed =
                                VMXNET3_RX_DATA_RING(adapter, rcd->rqID);
                        len = rxDataRingUsed ? rcd->len : rbi->len;
                        new_skb = netdev_alloc_skb_ip_align(adapter->netdev,
                                                            len);
                        if (new_skb == NULL) {
                                /* Skb allocation failed, do not handover this
                                 * skb to stack. Reuse it. Drop the existing pkt
                                 */
                                rq->stats.rx_buf_alloc_failure++;
                                ctx->skb = NULL;
                                rq->stats.drop_total++;
                                skip_page_frags = true;
                                goto rcd_done;
                        }

                        if (rxDataRingUsed) {
                                size_t sz;

                                BUG_ON(rcd->len > rq->data_ring.desc_size);

                                ctx->skb = new_skb;
                                sz = rcd->rxdIdx * rq->data_ring.desc_size;
                                memcpy(new_skb->data,
                                       &rq->data_ring.base[sz], rcd->len);
                        } else {
                                ctx->skb = rbi->skb;

                                new_dma_addr =
                                        dma_map_single(&adapter->pdev->dev,
                                                       new_skb->data, rbi->len,
                                                       DMA_FROM_DEVICE);
                                if (dma_mapping_error(&adapter->pdev->dev,
                                                      new_dma_addr)) {
                                        dev_kfree_skb(new_skb);
                                        /* Skb allocation failed, do not
                                         * handover this skb to stack. Reuse
                                         * it. Drop the existing pkt.
                                         */
                                        rq->stats.rx_buf_alloc_failure++;
                                        ctx->skb = NULL;
                                        rq->stats.drop_total++;
                                        skip_page_frags = true;
                                        goto rcd_done;
                                }

                                dma_unmap_single(&adapter->pdev->dev,
                                                 rbi->dma_addr,
                                                 rbi->len,
                                                 DMA_FROM_DEVICE);

                                /* Immediate refill */
                                rbi->skb = new_skb;
                                rbi->dma_addr = new_dma_addr;
                                rxd->addr = cpu_to_le64(rbi->dma_addr);
                                rxd->len = rbi->len;
                        }

#ifdef VMXNET3_RSS
                        if (rcd->rssType != VMXNET3_RCD_RSS_TYPE_NONE &&
                            (adapter->netdev->features & NETIF_F_RXHASH)) {
                                enum pkt_hash_types hash_type;

                                switch (rcd->rssType) {
                                case VMXNET3_RCD_RSS_TYPE_IPV4:
                                case VMXNET3_RCD_RSS_TYPE_IPV6:
                                        hash_type = PKT_HASH_TYPE_L3;
                                        break;
                                case VMXNET3_RCD_RSS_TYPE_TCPIPV4:
                                case VMXNET3_RCD_RSS_TYPE_TCPIPV6:
                                case VMXNET3_RCD_RSS_TYPE_UDPIPV4:
                                case VMXNET3_RCD_RSS_TYPE_UDPIPV6:
                                        hash_type = PKT_HASH_TYPE_L4;
                                        break;
                                default:
                                        hash_type = PKT_HASH_TYPE_L3;
                                        break;
                                }
                                skb_set_hash(ctx->skb,
                                             le32_to_cpu(rcd->rssHash),
                                             hash_type);
                        }
#endif
                        skb_put(ctx->skb, rcd->len);

                        if (VMXNET3_VERSION_GE_2(adapter) &&
                            rcd->type == VMXNET3_CDTYPE_RXCOMP_LRO) {
                                struct Vmxnet3_RxCompDescExt *rcdlro;
                                rcdlro = (struct Vmxnet3_RxCompDescExt *)rcd;

                                segCnt = rcdlro->segCnt;
                                WARN_ON_ONCE(segCnt == 0);
                                mss = rcdlro->mss;
                                if (unlikely(segCnt <= 1))
                                        segCnt = 0;
                        } else {
                                segCnt = 0;
                        }
                } else {
                        BUG_ON(ctx->skb == NULL && !skip_page_frags);

                        /* non SOP buffer must be type 1 in most cases */
                        BUG_ON(rbi->buf_type != VMXNET3_RX_BUF_PAGE);
                        BUG_ON(rxd->btype != VMXNET3_RXD_BTYPE_BODY);

                        /* If an sop buffer was dropped, skip all
                         * following non-sop fragments. They will be reused.
                         */
                        if (skip_page_frags)
                                goto rcd_done;

                        if (rcd->len) {
                                new_page = alloc_page(GFP_ATOMIC);
                                /* Replacement page frag could not be allocated.
                                 * Reuse this page. Drop the pkt and free the
                                 * skb which contained this page as a frag. Skip
                                 * processing all the following non-sop frags.
                                 */
                                if (unlikely(!new_page)) {
                                        rq->stats.rx_buf_alloc_failure++;
                                        dev_kfree_skb(ctx->skb);
                                        ctx->skb = NULL;
                                        skip_page_frags = true;
                                        goto rcd_done;
                                }
                                new_dma_addr = dma_map_page(&adapter->pdev->dev,
                                                            new_page,
                                                            0, PAGE_SIZE,
                                                            DMA_FROM_DEVICE);
                                if (dma_mapping_error(&adapter->pdev->dev,
                                                      new_dma_addr)) {
                                        put_page(new_page);
                                        rq->stats.rx_buf_alloc_failure++;
                                        dev_kfree_skb(ctx->skb);
                                        ctx->skb = NULL;
                                        skip_page_frags = true;
                                        goto rcd_done;
                                }

                                dma_unmap_page(&adapter->pdev->dev,
                                               rbi->dma_addr, rbi->len,
                                               DMA_FROM_DEVICE);

                                vmxnet3_append_frag(ctx->skb, rcd, rbi);

                                /* Immediate refill */
                                rbi->page = new_page;
                                rbi->dma_addr = new_dma_addr;
                                rxd->addr = cpu_to_le64(rbi->dma_addr);
                                rxd->len = rbi->len;
                        }
                }


                skb = ctx->skb;
                if (rcd->eop) {
                        u32 mtu = adapter->netdev->mtu;
                        skb->len += skb->data_len;

                        vmxnet3_rx_csum(adapter, skb,
                                        (union Vmxnet3_GenericDesc *)rcd);
                        skb->protocol = eth_type_trans(skb, adapter->netdev);
                        if (!rcd->tcp ||
                            !(adapter->netdev->features & NETIF_F_LRO))
                                goto not_lro;

                        if (segCnt != 0 && mss != 0) {
                                skb_shinfo(skb)->gso_type = rcd->v4 ?
                                        SKB_GSO_TCPV4 : SKB_GSO_TCPV6;
                                skb_shinfo(skb)->gso_size = mss;
                                skb_shinfo(skb)->gso_segs = segCnt;
                        } else if (segCnt != 0 || skb->len > mtu) {
                                u32 hlen;

                                hlen = vmxnet3_get_hdr_len(adapter, skb,
                                        (union Vmxnet3_GenericDesc *)rcd);
                                if (hlen == 0)
                                        goto not_lro;

                                skb_shinfo(skb)->gso_type =
                                        rcd->v4 ? SKB_GSO_TCPV4 : SKB_GSO_TCPV6;
                                if (segCnt != 0) {
                                        skb_shinfo(skb)->gso_segs = segCnt;
                                        skb_shinfo(skb)->gso_size =
                                                DIV_ROUND_UP(skb->len -
                                                        hlen, segCnt);
                                } else {
                                        skb_shinfo(skb)->gso_size = mtu - hlen;
                                }
                        }
not_lro:
                        if (unlikely(rcd->ts))
                                __vlan_hwaccel_put_tag(skb, htons(ETH_P_8021Q), rcd->tci);

                        if (adapter->netdev->features & NETIF_F_LRO)
                                netif_receive_skb(skb);
                        else
                                napi_gro_receive(&rq->napi, skb);

                        ctx->skb = NULL;
                        num_pkts++;
                }

rcd_done:
                /* device may have skipped some rx descs */
                ring->next2comp = idx;
                num_to_alloc = vmxnet3_cmd_ring_desc_avail(ring);
                ring = rq->rx_ring + ring_idx;

                /* Ensure that the writes to rxd->gen bits will be observed
                 * after all other writes to rxd objects.
                 */
                dma_wmb();

                while (num_to_alloc) {
                        vmxnet3_getRxDesc(rxd, &ring->base[ring->next2fill].rxd,
                                          &rxCmdDesc);
                        BUG_ON(!rxd->addr);

                        /* Recv desc is ready to be used by the device */
                        rxd->gen = ring->gen;
                        vmxnet3_cmd_ring_adv_next2fill(ring);
                        num_to_alloc--;
                }

                /* if needed, update the register */
                if (unlikely(rq->shared->updateRxProd)) {
                        VMXNET3_WRITE_BAR0_REG(adapter,
                                               rxprod_reg[ring_idx] + rq->qid * 8,
                                               ring->next2fill);
                }

                vmxnet3_comp_ring_adv_next2proc(&rq->comp_ring);
                vmxnet3_getRxComp(rcd,
                                  &rq->comp_ring.base[rq->comp_ring.next2proc].rcd, &rxComp);
        }

        return num_pkts;
}


static void
vmxnet3_rq_cleanup(struct vmxnet3_rx_queue *rq,
                   struct vmxnet3_adapter *adapter)
{
        u32 i, ring_idx;
        struct Vmxnet3_RxDesc *rxd;

        for (ring_idx = 0; ring_idx < 2; ring_idx++) {
                for (i = 0; i < rq->rx_ring[ring_idx].size; i++) {
#ifdef __BIG_ENDIAN_BITFIELD
                        struct Vmxnet3_RxDesc rxDesc;
#endif
                        vmxnet3_getRxDesc(rxd,
                                &rq->rx_ring[ring_idx].base[i].rxd, &rxDesc);

                        if (rxd->btype == VMXNET3_RXD_BTYPE_HEAD &&
                                        rq->buf_info[ring_idx][i].skb) {
                                dma_unmap_single(&adapter->pdev->dev, rxd->addr,
                                                 rxd->len, DMA_FROM_DEVICE);
                                dev_kfree_skb(rq->buf_info[ring_idx][i].skb);
                                rq->buf_info[ring_idx][i].skb = NULL;
                        } else if (rxd->btype == VMXNET3_RXD_BTYPE_BODY &&
                                        rq->buf_info[ring_idx][i].page) {
                                dma_unmap_page(&adapter->pdev->dev, rxd->addr,
                                               rxd->len, DMA_FROM_DEVICE);
                                put_page(rq->buf_info[ring_idx][i].page);
                                rq->buf_info[ring_idx][i].page = NULL;
                        }
                }

                rq->rx_ring[ring_idx].gen = VMXNET3_INIT_GEN;
                rq->rx_ring[ring_idx].next2fill =
                                        rq->rx_ring[ring_idx].next2comp = 0;
        }

        rq->comp_ring.gen = VMXNET3_INIT_GEN;
        rq->comp_ring.next2proc = 0;
}


static void
vmxnet3_rq_cleanup_all(struct vmxnet3_adapter *adapter)
{
        int i;

        for (i = 0; i < adapter->num_rx_queues; i++)
                vmxnet3_rq_cleanup(&adapter->rx_queue[i], adapter);
}


static void vmxnet3_rq_destroy(struct vmxnet3_rx_queue *rq,
                               struct vmxnet3_adapter *adapter)
{
        int i;
        int j;

        /* all rx buffers must have already been freed */
        for (i = 0; i < 2; i++) {
                if (rq->buf_info[i]) {
                        for (j = 0; j < rq->rx_ring[i].size; j++)
                                BUG_ON(rq->buf_info[i][j].page != NULL);
                }
        }


        for (i = 0; i < 2; i++) {
                if (rq->rx_ring[i].base) {
                        dma_free_coherent(&adapter->pdev->dev,
                                          rq->rx_ring[i].size
                                          * sizeof(struct Vmxnet3_RxDesc),
                                          rq->rx_ring[i].base,
                                          rq->rx_ring[i].basePA);
                        rq->rx_ring[i].base = NULL;
                }
        }

        if (rq->data_ring.base) {
                dma_free_coherent(&adapter->pdev->dev,
                                  rq->rx_ring[0].size * rq->data_ring.desc_size,
                                  rq->data_ring.base, rq->data_ring.basePA);
                rq->data_ring.base = NULL;
        }

        if (rq->comp_ring.base) {
                dma_free_coherent(&adapter->pdev->dev, rq->comp_ring.size
                                  * sizeof(struct Vmxnet3_RxCompDesc),
                                  rq->comp_ring.base, rq->comp_ring.basePA);
                rq->comp_ring.base = NULL;
        }

        kfree(rq->buf_info[0]);
        rq->buf_info[0] = NULL;
        rq->buf_info[1] = NULL;
}

static void
vmxnet3_rq_destroy_all_rxdataring(struct vmxnet3_adapter *adapter)
{
        int i;

        for (i = 0; i < adapter->num_rx_queues; i++) {
                struct vmxnet3_rx_queue *rq = &adapter->rx_queue[i];

                if (rq->data_ring.base) {
                        dma_free_coherent(&adapter->pdev->dev,
                                          (rq->rx_ring[0].size *
                                          rq->data_ring.desc_size),
                                          rq->data_ring.base,
                                          rq->data_ring.basePA);
                        rq->data_ring.base = NULL;
                        rq->data_ring.desc_size = 0;
                }
        }
}

static int
vmxnet3_rq_init(struct vmxnet3_rx_queue *rq,
                struct vmxnet3_adapter  *adapter)
{
        int i;

        /* initialize buf_info */
        for (i = 0; i < rq->rx_ring[0].size; i++) {

                /* 1st buf for a pkt is skbuff */
                if (i % adapter->rx_buf_per_pkt == 0) {
                        rq->buf_info[0][i].buf_type = VMXNET3_RX_BUF_SKB;
                        rq->buf_info[0][i].len = adapter->skb_buf_size;
                } else { /* subsequent bufs for a pkt is frag */
                        rq->buf_info[0][i].buf_type = VMXNET3_RX_BUF_PAGE;
                        rq->buf_info[0][i].len = PAGE_SIZE;
                }
        }
        for (i = 0; i < rq->rx_ring[1].size; i++) {
                rq->buf_info[1][i].buf_type = VMXNET3_RX_BUF_PAGE;
                rq->buf_info[1][i].len = PAGE_SIZE;
        }

        /* reset internal state and allocate buffers for both rings */
        for (i = 0; i < 2; i++) {
                rq->rx_ring[i].next2fill = rq->rx_ring[i].next2comp = 0;

                memset(rq->rx_ring[i].base, 0, rq->rx_ring[i].size *
                       sizeof(struct Vmxnet3_RxDesc));
                rq->rx_ring[i].gen = VMXNET3_INIT_GEN;
        }
        if (vmxnet3_rq_alloc_rx_buf(rq, 0, rq->rx_ring[0].size - 1,
                                    adapter) == 0) {
                /* at least has 1 rx buffer for the 1st ring */
                return -ENOMEM;
        }
        vmxnet3_rq_alloc_rx_buf(rq, 1, rq->rx_ring[1].size - 1, adapter);

        /* reset the comp ring */
        rq->comp_ring.next2proc = 0;
        memset(rq->comp_ring.base, 0, rq->comp_ring.size *
               sizeof(struct Vmxnet3_RxCompDesc));
        rq->comp_ring.gen = VMXNET3_INIT_GEN;

        /* reset rxctx */
        rq->rx_ctx.skb = NULL;

        /* stats are not reset */
        return 0;
}


static int
vmxnet3_rq_init_all(struct vmxnet3_adapter *adapter)
{
        int i, err = 0;

        for (i = 0; i < adapter->num_rx_queues; i++) {
                err = vmxnet3_rq_init(&adapter->rx_queue[i], adapter);
                if (unlikely(err)) {
                        dev_err(&adapter->netdev->dev, "%s: failed to "
                                "initialize rx queue%i\n",
                                adapter->netdev->name, i);
                        break;
                }
        }
        return err;

}


static int
vmxnet3_rq_create(struct vmxnet3_rx_queue *rq, struct vmxnet3_adapter *adapter)
{
        int i;
        size_t sz;
        struct vmxnet3_rx_buf_info *bi;

        for (i = 0; i < 2; i++) {

                sz = rq->rx_ring[i].size * sizeof(struct Vmxnet3_RxDesc);
                rq->rx_ring[i].base = dma_alloc_coherent(
                                                &adapter->pdev->dev, sz,
                                                &rq->rx_ring[i].basePA,
                                                GFP_KERNEL);
                if (!rq->rx_ring[i].base) {
                        netdev_err(adapter->netdev,
                                   "failed to allocate rx ring %d\n", i);
                        goto err;
                }
        }

        if ((adapter->rxdataring_enabled) && (rq->data_ring.desc_size != 0)) {
                sz = rq->rx_ring[0].size * rq->data_ring.desc_size;
                rq->data_ring.base =
                        dma_alloc_coherent(&adapter->pdev->dev, sz,
                                           &rq->data_ring.basePA,
                                           GFP_KERNEL);
                if (!rq->data_ring.base) {
                        netdev_err(adapter->netdev,
                                   "rx data ring will be disabled\n");
                        adapter->rxdataring_enabled = false;
                }
        } else {
                rq->data_ring.base = NULL;
                rq->data_ring.desc_size = 0;
        }

        sz = rq->comp_ring.size * sizeof(struct Vmxnet3_RxCompDesc);
        rq->comp_ring.base = dma_alloc_coherent(&adapter->pdev->dev, sz,
                                                &rq->comp_ring.basePA,
                                                GFP_KERNEL);
        if (!rq->comp_ring.base) {
                netdev_err(adapter->netdev, "failed to allocate rx comp ring\n");
                goto err;
        }

        bi = kcalloc_node(rq->rx_ring[0].size + rq->rx_ring[1].size,
                          sizeof(rq->buf_info[0][0]), GFP_KERNEL,
                          dev_to_node(&adapter->pdev->dev));
        if (!bi)
                goto err;

        rq->buf_info[0] = bi;
        rq->buf_info[1] = bi + rq->rx_ring[0].size;

        return 0;

err:
        vmxnet3_rq_destroy(rq, adapter);
        return -ENOMEM;
}


static int
vmxnet3_rq_create_all(struct vmxnet3_adapter *adapter)
{
        int i, err = 0;

        adapter->rxdataring_enabled = VMXNET3_VERSION_GE_3(adapter);

        for (i = 0; i < adapter->num_rx_queues; i++) {
                err = vmxnet3_rq_create(&adapter->rx_queue[i], adapter);
                if (unlikely(err)) {
                        dev_err(&adapter->netdev->dev,
                                "%s: failed to create rx queue%i\n",
                                adapter->netdev->name, i);
                        goto err_out;
                }
        }

        if (!adapter->rxdataring_enabled)
                vmxnet3_rq_destroy_all_rxdataring(adapter);

        return err;
err_out:
        vmxnet3_rq_destroy_all(adapter);
        return err;

}

/* Multiple queue aware polling function for tx and rx */

static int
vmxnet3_do_poll(struct vmxnet3_adapter *adapter, int budget)
{
        int rcd_done = 0, i;
        if (unlikely(adapter->shared->ecr))
                vmxnet3_process_events(adapter);
        for (i = 0; i < adapter->num_tx_queues; i++)
                vmxnet3_tq_tx_complete(&adapter->tx_queue[i], adapter);

        for (i = 0; i < adapter->num_rx_queues; i++)
                rcd_done += vmxnet3_rq_rx_complete(&adapter->rx_queue[i],
                                                   adapter, budget);
        return rcd_done;
}


static int
vmxnet3_poll(struct napi_struct *napi, int budget)
{
        struct vmxnet3_rx_queue *rx_queue = container_of(napi,
                                          struct vmxnet3_rx_queue, napi);
        int rxd_done;

        rxd_done = vmxnet3_do_poll(rx_queue->adapter, budget);

        if (rxd_done < budget) {
                napi_complete_done(napi, rxd_done);
                vmxnet3_enable_all_intrs(rx_queue->adapter);
        }
        return rxd_done;
}

/*
 * NAPI polling function for MSI-X mode with multiple Rx queues
 * Returns the # of the NAPI credit consumed (# of rx descriptors processed)
 */

static int
vmxnet3_poll_rx_only(struct napi_struct *napi, int budget)
{
        struct vmxnet3_rx_queue *rq = container_of(napi,
                                                struct vmxnet3_rx_queue, napi);
        struct vmxnet3_adapter *adapter = rq->adapter;
        int rxd_done;

        /* When sharing interrupt with corresponding tx queue, process
         * tx completions in that queue as well
         */
        if (adapter->share_intr == VMXNET3_INTR_BUDDYSHARE) {
                struct vmxnet3_tx_queue *tq =
                                &adapter->tx_queue[rq - adapter->rx_queue];
                vmxnet3_tq_tx_complete(tq, adapter);
        }

        rxd_done = vmxnet3_rq_rx_complete(rq, adapter, budget);

        if (rxd_done < budget) {
                napi_complete_done(napi, rxd_done);
                vmxnet3_enable_intr(adapter, rq->comp_ring.intr_idx);
        }
        return rxd_done;

}


#ifdef CONFIG_PCI_MSI

/*
 * Handle completion interrupts on tx queues
 * Returns whether or not the intr is handled
 */

static irqreturn_t
vmxnet3_msix_tx(int irq, void *data)
{
        struct vmxnet3_tx_queue *tq = data;
        struct vmxnet3_adapter *adapter = tq->adapter;

        if (adapter->intr.mask_mode == VMXNET3_IMM_ACTIVE)
                vmxnet3_disable_intr(adapter, tq->comp_ring.intr_idx);

        /* Handle the case where only one irq is allocate for all tx queues */
        if (adapter->share_intr == VMXNET3_INTR_TXSHARE) {
                int i;
                for (i = 0; i < adapter->num_tx_queues; i++) {
                        struct vmxnet3_tx_queue *txq = &adapter->tx_queue[i];
                        vmxnet3_tq_tx_complete(txq, adapter);
                }
        } else {
                vmxnet3_tq_tx_complete(tq, adapter);
        }
        vmxnet3_enable_intr(adapter, tq->comp_ring.intr_idx);

        return IRQ_HANDLED;
}


/*
 * Handle completion interrupts on rx queues. Returns whether or not the
 * intr is handled
 */

static irqreturn_t
vmxnet3_msix_rx(int irq, void *data)
{
        struct vmxnet3_rx_queue *rq = data;
        struct vmxnet3_adapter *adapter = rq->adapter;

        /* disable intr if needed */
        if (adapter->intr.mask_mode == VMXNET3_IMM_ACTIVE)
                vmxnet3_disable_intr(adapter, rq->comp_ring.intr_idx);
        napi_schedule(&rq->napi);

        return IRQ_HANDLED;
}

/*
 *----------------------------------------------------------------------------
 *
 * vmxnet3_msix_event --
 *
 *    vmxnet3 msix event intr handler
 *
 * Result:
 *    whether or not the intr is handled
 *
 *----------------------------------------------------------------------------
 */

static irqreturn_t
vmxnet3_msix_event(int irq, void *data)
{
        struct net_device *dev = data;
        struct vmxnet3_adapter *adapter = netdev_priv(dev);

        /* disable intr if needed */
        if (adapter->intr.mask_mode == VMXNET3_IMM_ACTIVE)
                vmxnet3_disable_intr(adapter, adapter->intr.event_intr_idx);

        if (adapter->shared->ecr)
                vmxnet3_process_events(adapter);

        vmxnet3_enable_intr(adapter, adapter->intr.event_intr_idx);

        return IRQ_HANDLED;
}

#endif /* CONFIG_PCI_MSI  */


/* Interrupt handler for vmxnet3  */
static irqreturn_t
vmxnet3_intr(int irq, void *dev_id)
{
        struct net_device *dev = dev_id;
        struct vmxnet3_adapter *adapter = netdev_priv(dev);

        if (adapter->intr.type == VMXNET3_IT_INTX) {
                u32 icr = VMXNET3_READ_BAR1_REG(adapter, VMXNET3_REG_ICR);
                if (unlikely(icr == 0))
                        /* not ours */
                        return IRQ_NONE;
        }


        /* disable intr if needed */
        if (adapter->intr.mask_mode == VMXNET3_IMM_ACTIVE)
                vmxnet3_disable_all_intrs(adapter);

        napi_schedule(&adapter->rx_queue[0].napi);

        return IRQ_HANDLED;
}

#ifdef CONFIG_NET_POLL_CONTROLLER

/* netpoll callback. */
static void
vmxnet3_netpoll(struct net_device *netdev)
{
        struct vmxnet3_adapter *adapter = netdev_priv(netdev);

        switch (adapter->intr.type) {
#ifdef CONFIG_PCI_MSI
        case VMXNET3_IT_MSIX: {
                int i;
                for (i = 0; i < adapter->num_rx_queues; i++)
                        vmxnet3_msix_rx(0, &adapter->rx_queue[i]);
                break;
        }
#endif
        case VMXNET3_IT_MSI:
        default:
                vmxnet3_intr(0, adapter->netdev);
                break;
        }

}
#endif  /* CONFIG_NET_POLL_CONTROLLER */

static int
vmxnet3_request_irqs(struct vmxnet3_adapter *adapter)
{
        struct vmxnet3_intr *intr = &adapter->intr;
        int err = 0, i;
        int vector = 0;

#ifdef CONFIG_PCI_MSI
        if (adapter->intr.type == VMXNET3_IT_MSIX) {
                for (i = 0; i < adapter->num_tx_queues; i++) {
                        if (adapter->share_intr != VMXNET3_INTR_BUDDYSHARE) {
                                sprintf(adapter->tx_queue[i].name, "%s-tx-%d",
                                        adapter->netdev->name, vector);
                                err = request_irq(
                                              intr->msix_entries[vector].vector,
                                              vmxnet3_msix_tx, 0,
                                              adapter->tx_queue[i].name,
                                              &adapter->tx_queue[i]);
                        } else {
                                sprintf(adapter->tx_queue[i].name, "%s-rxtx-%d",
                                        adapter->netdev->name, vector);
                        }
                        if (err) {
                                dev_err(&adapter->netdev->dev,
                                        "Failed to request irq for MSIX, %s, "
                                        "error %d\n",
                                        adapter->tx_queue[i].name, err);
                                return err;
                        }

                        /* Handle the case where only 1 MSIx was allocated for
                         * all tx queues */
                        if (adapter->share_intr == VMXNET3_INTR_TXSHARE) {
                                for (; i < adapter->num_tx_queues; i++)
                                        adapter->tx_queue[i].comp_ring.intr_idx
                                                                = vector;
                                vector++;
                                break;
                        } else {
                                adapter->tx_queue[i].comp_ring.intr_idx
                                                                = vector++;
                        }
                }
                if (adapter->share_intr == VMXNET3_INTR_BUDDYSHARE)
                        vector = 0;

                for (i = 0; i < adapter->num_rx_queues; i++) {
                        if (adapter->share_intr != VMXNET3_INTR_BUDDYSHARE)
                                sprintf(adapter->rx_queue[i].name, "%s-rx-%d",
                                        adapter->netdev->name, vector);
                        else
                                sprintf(adapter->rx_queue[i].name, "%s-rxtx-%d",
                                        adapter->netdev->name, vector);
                        err = request_irq(intr->msix_entries[vector].vector,
                                          vmxnet3_msix_rx, 0,
                                          adapter->rx_queue[i].name,
                                          &(adapter->rx_queue[i]));
                        if (err) {
                                netdev_err(adapter->netdev,
                                           "Failed to request irq for MSIX, "
                                           "%s, error %d\n",
                                           adapter->rx_queue[i].name, err);
                                return err;
                        }

                        adapter->rx_queue[i].comp_ring.intr_idx = vector++;
                }

                sprintf(intr->event_msi_vector_name, "%s-event-%d",
                        adapter->netdev->name, vector);
                err = request_irq(intr->msix_entries[vector].vector,
                                  vmxnet3_msix_event, 0,
                                  intr->event_msi_vector_name, adapter->netdev);
                intr->event_intr_idx = vector;

        } else if (intr->type == VMXNET3_IT_MSI) {
                adapter->num_rx_queues = 1;
                err = request_irq(adapter->pdev->irq, vmxnet3_intr, 0,
                                  adapter->netdev->name, adapter->netdev);
        } else {
#endif
                adapter->num_rx_queues = 1;
                err = request_irq(adapter->pdev->irq, vmxnet3_intr,
                                  IRQF_SHARED, adapter->netdev->name,
                                  adapter->netdev);
#ifdef CONFIG_PCI_MSI
        }
#endif
        intr->num_intrs = vector + 1;
        if (err) {
                netdev_err(adapter->netdev,
                           "Failed to request irq (intr type:%d), error %d\n",
                           intr->type, err);
        } else {
                /* Number of rx queues will not change after this */
                for (i = 0; i < adapter->num_rx_queues; i++) {
                        struct vmxnet3_rx_queue *rq = &adapter->rx_queue[i];
                        rq->qid = i;
                        rq->qid2 = i + adapter->num_rx_queues;
                        rq->dataRingQid = i + 2 * adapter->num_rx_queues;
                }

                /* init our intr settings */
                for (i = 0; i < intr->num_intrs; i++)
                        intr->mod_levels[i] = UPT1_IML_ADAPTIVE;
                if (adapter->intr.type != VMXNET3_IT_MSIX) {
                        adapter->intr.event_intr_idx = 0;
                        for (i = 0; i < adapter->num_tx_queues; i++)
                                adapter->tx_queue[i].comp_ring.intr_idx = 0;
                        adapter->rx_queue[0].comp_ring.intr_idx = 0;
                }

                netdev_info(adapter->netdev,
                            "intr type %u, mode %u, %u vectors allocated\n",
                            intr->type, intr->mask_mode, intr->num_intrs);
        }

        return err;
}


static void
vmxnet3_free_irqs(struct vmxnet3_adapter *adapter)
{
        struct vmxnet3_intr *intr = &adapter->intr;
        BUG_ON(intr->type == VMXNET3_IT_AUTO || intr->num_intrs <= 0);

        switch (intr->type) {
#ifdef CONFIG_PCI_MSI
        case VMXNET3_IT_MSIX:
        {
                int i, vector = 0;

                if (adapter->share_intr != VMXNET3_INTR_BUDDYSHARE) {
                        for (i = 0; i < adapter->num_tx_queues; i++) {
                                free_irq(intr->msix_entries[vector++].vector,
                                         &(adapter->tx_queue[i]));
                                if (adapter->share_intr == VMXNET3_INTR_TXSHARE)
                                        break;
                        }
                }

                for (i = 0; i < adapter->num_rx_queues; i++) {
                        free_irq(intr->msix_entries[vector++].vector,
                                 &(adapter->rx_queue[i]));
                }

                free_irq(intr->msix_entries[vector].vector,
                         adapter->netdev);
                BUG_ON(vector >= intr->num_intrs);
                break;
        }
#endif
        case VMXNET3_IT_MSI:
                free_irq(adapter->pdev->irq, adapter->netdev);
                break;
        case VMXNET3_IT_INTX:
                free_irq(adapter->pdev->irq, adapter->netdev);
                break;
        default:
                BUG();
        }
}


static void
vmxnet3_restore_vlan(struct vmxnet3_adapter *adapter)
{
        u32 *vfTable = adapter->shared->devRead.rxFilterConf.vfTable;
        u16 vid;

        /* allow untagged pkts */
        VMXNET3_SET_VFTABLE_ENTRY(vfTable, 0);

        for_each_set_bit(vid, adapter->active_vlans, VLAN_N_VID)
                VMXNET3_SET_VFTABLE_ENTRY(vfTable, vid);
}


static int
vmxnet3_vlan_rx_add_vid(struct net_device *netdev, __be16 proto, u16 vid)
{
        struct vmxnet3_adapter *adapter = netdev_priv(netdev);

        if (!(netdev->flags & IFF_PROMISC)) {
                u32 *vfTable = adapter->shared->devRead.rxFilterConf.vfTable;
                unsigned long flags;

                VMXNET3_SET_VFTABLE_ENTRY(vfTable, vid);
                spin_lock_irqsave(&adapter->cmd_lock, flags);
                VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_CMD,
                                       VMXNET3_CMD_UPDATE_VLAN_FILTERS);
                spin_unlock_irqrestore(&adapter->cmd_lock, flags);
        }

        set_bit(vid, adapter->active_vlans);

        return 0;
}


static int
vmxnet3_vlan_rx_kill_vid(struct net_device *netdev, __be16 proto, u16 vid)
{
        struct vmxnet3_adapter *adapter = netdev_priv(netdev);

        if (!(netdev->flags & IFF_PROMISC)) {
                u32 *vfTable = adapter->shared->devRead.rxFilterConf.vfTable;
                unsigned long flags;

                VMXNET3_CLEAR_VFTABLE_ENTRY(vfTable, vid);
                spin_lock_irqsave(&adapter->cmd_lock, flags);
                VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_CMD,
                                       VMXNET3_CMD_UPDATE_VLAN_FILTERS);
                spin_unlock_irqrestore(&adapter->cmd_lock, flags);
        }

        clear_bit(vid, adapter->active_vlans);

        return 0;
}


static u8 *
vmxnet3_copy_mc(struct net_device *netdev)
{
        u8 *buf = NULL;
        u32 sz = netdev_mc_count(netdev) * ETH_ALEN;

        /* struct Vmxnet3_RxFilterConf.mfTableLen is u16. */
        if (sz <= 0xffff) {
                /* We may be called with BH disabled */
                buf = kmalloc(sz, GFP_ATOMIC);
                if (buf) {
                        struct netdev_hw_addr *ha;
                        int i = 0;

                        netdev_for_each_mc_addr(ha, netdev)
                                memcpy(buf + i++ * ETH_ALEN, ha->addr,
                                       ETH_ALEN);
                }
        }
        return buf;
}


static void
vmxnet3_set_mc(struct net_device *netdev)
{
        struct vmxnet3_adapter *adapter = netdev_priv(netdev);
        unsigned long flags;
        struct Vmxnet3_RxFilterConf *rxConf =
                                        &adapter->shared->devRead.rxFilterConf;
        u8 *new_table = NULL;
        dma_addr_t new_table_pa = 0;
        bool new_table_pa_valid = false;
        u32 new_mode = VMXNET3_RXM_UCAST;

        if (netdev->flags & IFF_PROMISC) {
                u32 *vfTable = adapter->shared->devRead.rxFilterConf.vfTable;
                memset(vfTable, 0, VMXNET3_VFT_SIZE * sizeof(*vfTable));

                new_mode |= VMXNET3_RXM_PROMISC;
        } else {
                vmxnet3_restore_vlan(adapter);
        }

        if (netdev->flags & IFF_BROADCAST)
                new_mode |= VMXNET3_RXM_BCAST;

        if (netdev->flags & IFF_ALLMULTI)
                new_mode |= VMXNET3_RXM_ALL_MULTI;
        else
                if (!netdev_mc_empty(netdev)) {
                        new_table = vmxnet3_copy_mc(netdev);
                        if (new_table) {
                                size_t sz = netdev_mc_count(netdev) * ETH_ALEN;

                                rxConf->mfTableLen = cpu_to_le16(sz);
                                new_table_pa = dma_map_single(
                                                        &adapter->pdev->dev,
                                                        new_table,
                                                        sz,
                                                        DMA_TO_DEVICE);
                                if (!dma_mapping_error(&adapter->pdev->dev,
                                                       new_table_pa)) {
                                        new_mode |= VMXNET3_RXM_MCAST;
                                        new_table_pa_valid = true;
                                        rxConf->mfTablePA = cpu_to_le64(
                                                                new_table_pa);
                                }
                        }
                        if (!new_table_pa_valid) {
                                netdev_info(netdev,
                                            "failed to copy mcast list, setting ALL_MULTI\n");
                                new_mode |= VMXNET3_RXM_ALL_MULTI;
                        }
                }

        if (!(new_mode & VMXNET3_RXM_MCAST)) {
                rxConf->mfTableLen = 0;
                rxConf->mfTablePA = 0;
        }

        spin_lock_irqsave(&adapter->cmd_lock, flags);
        if (new_mode != rxConf->rxMode) {
                rxConf->rxMode = cpu_to_le32(new_mode);
                VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_CMD,
                                       VMXNET3_CMD_UPDATE_RX_MODE);
                VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_CMD,
                                       VMXNET3_CMD_UPDATE_VLAN_FILTERS);
        }

        VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_CMD,
                               VMXNET3_CMD_UPDATE_MAC_FILTERS);
        spin_unlock_irqrestore(&adapter->cmd_lock, flags);

        if (new_table_pa_valid)
                dma_unmap_single(&adapter->pdev->dev, new_table_pa,
                                 rxConf->mfTableLen, DMA_TO_DEVICE);
        kfree(new_table);
}

void
vmxnet3_rq_destroy_all(struct vmxnet3_adapter *adapter)
{
        int i;

        for (i = 0; i < adapter->num_rx_queues; i++)
                vmxnet3_rq_destroy(&adapter->rx_queue[i], adapter);
}


/*
 *   Set up driver_shared based on settings in adapter.
 */

static void
vmxnet3_setup_driver_shared(struct vmxnet3_adapter *adapter)
{
        struct Vmxnet3_DriverShared *shared = adapter->shared;
        struct Vmxnet3_DSDevRead *devRead = &shared->devRead;
        struct Vmxnet3_DSDevReadExt *devReadExt = &shared->devReadExt;
        struct Vmxnet3_TxQueueConf *tqc;
        struct Vmxnet3_RxQueueConf *rqc;
        int i;

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

        /* driver settings */
        shared->magic = cpu_to_le32(VMXNET3_REV1_MAGIC);
        devRead->misc.driverInfo.version = cpu_to_le32(
                                                VMXNET3_DRIVER_VERSION_NUM);
        devRead->misc.driverInfo.gos.gosBits = (sizeof(void *) == 4 ?
                                VMXNET3_GOS_BITS_32 : VMXNET3_GOS_BITS_64);
        devRead->misc.driverInfo.gos.gosType = VMXNET3_GOS_TYPE_LINUX;
        *((u32 *)&devRead->misc.driverInfo.gos) = cpu_to_le32(
                                *((u32 *)&devRead->misc.driverInfo.gos));
        devRead->misc.driverInfo.vmxnet3RevSpt = cpu_to_le32(1);
        devRead->misc.driverInfo.uptVerSpt = cpu_to_le32(1);

        devRead->misc.ddPA = cpu_to_le64(adapter->adapter_pa);
        devRead->misc.ddLen = cpu_to_le32(sizeof(struct vmxnet3_adapter));

        /* set up feature flags */
        if (adapter->netdev->features & NETIF_F_RXCSUM)
                devRead->misc.uptFeatures |= UPT1_F_RXCSUM;

        if (adapter->netdev->features & NETIF_F_LRO) {
                devRead->misc.uptFeatures |= UPT1_F_LRO;
                devRead->misc.maxNumRxSG = cpu_to_le16(1 + MAX_SKB_FRAGS);
        }
        if (adapter->netdev->features & NETIF_F_HW_VLAN_CTAG_RX)
                devRead->misc.uptFeatures |= UPT1_F_RXVLAN;

        if (adapter->netdev->features & (NETIF_F_GSO_UDP_TUNNEL |
                                         NETIF_F_GSO_UDP_TUNNEL_CSUM))
                devRead->misc.uptFeatures |= UPT1_F_RXINNEROFLD;

        devRead->misc.mtu = cpu_to_le32(adapter->netdev->mtu);
        devRead->misc.queueDescPA = cpu_to_le64(adapter->queue_desc_pa);
        devRead->misc.queueDescLen = cpu_to_le32(
                adapter->num_tx_queues * sizeof(struct Vmxnet3_TxQueueDesc) +
                adapter->num_rx_queues * sizeof(struct Vmxnet3_RxQueueDesc));

        /* tx queue settings */
        devRead->misc.numTxQueues =  adapter->num_tx_queues;
        for (i = 0; i < adapter->num_tx_queues; i++) {
                struct vmxnet3_tx_queue *tq = &adapter->tx_queue[i];
                BUG_ON(adapter->tx_queue[i].tx_ring.base == NULL);
                tqc = &adapter->tqd_start[i].conf;
                tqc->txRingBasePA   = cpu_to_le64(tq->tx_ring.basePA);
                tqc->dataRingBasePA = cpu_to_le64(tq->data_ring.basePA);
                tqc->compRingBasePA = cpu_to_le64(tq->comp_ring.basePA);
                tqc->ddPA           = cpu_to_le64(~0ULL);
                tqc->txRingSize     = cpu_to_le32(tq->tx_ring.size);
                tqc->dataRingSize   = cpu_to_le32(tq->data_ring.size);
                tqc->txDataRingDescSize = cpu_to_le32(tq->txdata_desc_size);
                tqc->compRingSize   = cpu_to_le32(tq->comp_ring.size);
                tqc->ddLen          = cpu_to_le32(0);
                tqc->intrIdx        = tq->comp_ring.intr_idx;
        }

        /* rx queue settings */
        devRead->misc.numRxQueues = adapter->num_rx_queues;
        for (i = 0; i < adapter->num_rx_queues; i++) {
                struct vmxnet3_rx_queue *rq = &adapter->rx_queue[i];
                rqc = &adapter->rqd_start[i].conf;
                rqc->rxRingBasePA[0] = cpu_to_le64(rq->rx_ring[0].basePA);
                rqc->rxRingBasePA[1] = cpu_to_le64(rq->rx_ring[1].basePA);
                rqc->compRingBasePA  = cpu_to_le64(rq->comp_ring.basePA);
                rqc->ddPA            = cpu_to_le64(~0ULL);
                rqc->rxRingSize[0]   = cpu_to_le32(rq->rx_ring[0].size);
                rqc->rxRingSize[1]   = cpu_to_le32(rq->rx_ring[1].size);
                rqc->compRingSize    = cpu_to_le32(rq->comp_ring.size);
                rqc->ddLen           = cpu_to_le32(0);
                rqc->intrIdx         = rq->comp_ring.intr_idx;
                if (VMXNET3_VERSION_GE_3(adapter)) {
                        rqc->rxDataRingBasePA =
                                cpu_to_le64(rq->data_ring.basePA);
                        rqc->rxDataRingDescSize =
                                cpu_to_le16(rq->data_ring.desc_size);
                }
        }

#ifdef VMXNET3_RSS
        memset(adapter->rss_conf, 0, sizeof(*adapter->rss_conf));

        if (adapter->rss) {
                struct UPT1_RSSConf *rssConf = adapter->rss_conf;

                devRead->misc.uptFeatures |= UPT1_F_RSS;
                devRead->misc.numRxQueues = adapter->num_rx_queues;
                rssConf->hashType = UPT1_RSS_HASH_TYPE_TCP_IPV4 |
                                    UPT1_RSS_HASH_TYPE_IPV4 |
                                    UPT1_RSS_HASH_TYPE_TCP_IPV6 |
                                    UPT1_RSS_HASH_TYPE_IPV6;
                rssConf->hashFunc = UPT1_RSS_HASH_FUNC_TOEPLITZ;
                rssConf->hashKeySize = UPT1_RSS_MAX_KEY_SIZE;
                rssConf->indTableSize = VMXNET3_RSS_IND_TABLE_SIZE;
                netdev_rss_key_fill(rssConf->hashKey, sizeof(rssConf->hashKey));

                for (i = 0; i < rssConf->indTableSize; i++)
                        rssConf->indTable[i] = ethtool_rxfh_indir_default(
                                i, adapter->num_rx_queues);

                devRead->rssConfDesc.confVer = 1;
                devRead->rssConfDesc.confLen = cpu_to_le32(sizeof(*rssConf));
                devRead->rssConfDesc.confPA =
                        cpu_to_le64(adapter->rss_conf_pa);
        }

#endif /* VMXNET3_RSS */

        /* intr settings */
        if (!VMXNET3_VERSION_GE_6(adapter) ||
            !adapter->queuesExtEnabled) {
                devRead->intrConf.autoMask = adapter->intr.mask_mode ==
                                             VMXNET3_IMM_AUTO;
                devRead->intrConf.numIntrs = adapter->intr.num_intrs;
                for (i = 0; i < adapter->intr.num_intrs; i++)
                        devRead->intrConf.modLevels[i] = adapter->intr.mod_levels[i];

                devRead->intrConf.eventIntrIdx = adapter->intr.event_intr_idx;
                devRead->intrConf.intrCtrl |= cpu_to_le32(VMXNET3_IC_DISABLE_ALL);
        } else {
                devReadExt->intrConfExt.autoMask = adapter->intr.mask_mode ==
                                                   VMXNET3_IMM_AUTO;
                devReadExt->intrConfExt.numIntrs = adapter->intr.num_intrs;
                for (i = 0; i < adapter->intr.num_intrs; i++)
                        devReadExt->intrConfExt.modLevels[i] = adapter->intr.mod_levels[i];

                devReadExt->intrConfExt.eventIntrIdx = adapter->intr.event_intr_idx;
                devReadExt->intrConfExt.intrCtrl |= cpu_to_le32(VMXNET3_IC_DISABLE_ALL);
        }

        /* rx filter settings */
        devRead->rxFilterConf.rxMode = 0;
        vmxnet3_restore_vlan(adapter);
        vmxnet3_write_mac_addr(adapter, adapter->netdev->dev_addr);

        /* the rest are already zeroed */
}

static void
vmxnet3_init_coalesce(struct vmxnet3_adapter *adapter)
{
        struct Vmxnet3_DriverShared *shared = adapter->shared;
        union Vmxnet3_CmdInfo *cmdInfo = &shared->cu.cmdInfo;
        unsigned long flags;

        if (!VMXNET3_VERSION_GE_3(adapter))
                return;

        spin_lock_irqsave(&adapter->cmd_lock, flags);
        cmdInfo->varConf.confVer = 1;
        cmdInfo->varConf.confLen =
                cpu_to_le32(sizeof(*adapter->coal_conf));
        cmdInfo->varConf.confPA  = cpu_to_le64(adapter->coal_conf_pa);

        if (adapter->default_coal_mode) {
                VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_CMD,
                                       VMXNET3_CMD_GET_COALESCE);
        } else {
                VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_CMD,
                                       VMXNET3_CMD_SET_COALESCE);
        }

        spin_unlock_irqrestore(&adapter->cmd_lock, flags);
}

static void
vmxnet3_init_rssfields(struct vmxnet3_adapter *adapter)
{
        struct Vmxnet3_DriverShared *shared = adapter->shared;
        union Vmxnet3_CmdInfo *cmdInfo = &shared->cu.cmdInfo;
        unsigned long flags;

        if (!VMXNET3_VERSION_GE_4(adapter))
                return;

        spin_lock_irqsave(&adapter->cmd_lock, flags);

        if (adapter->default_rss_fields) {
                VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_CMD,
                                       VMXNET3_CMD_GET_RSS_FIELDS);
                adapter->rss_fields =
                        VMXNET3_READ_BAR1_REG(adapter, VMXNET3_REG_CMD);
        } else {
                cmdInfo->setRssFields = adapter->rss_fields;
                VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_CMD,
                                       VMXNET3_CMD_SET_RSS_FIELDS);
                /* Not all requested RSS may get applied, so get and
                 * cache what was actually applied.
                 */
                VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_CMD,
                                       VMXNET3_CMD_GET_RSS_FIELDS);
                adapter->rss_fields =
                        VMXNET3_READ_BAR1_REG(adapter, VMXNET3_REG_CMD);
        }

        spin_unlock_irqrestore(&adapter->cmd_lock, flags);
}

int
vmxnet3_activate_dev(struct vmxnet3_adapter *adapter)
{
        int err, i;
        u32 ret;
        unsigned long flags;

        netdev_dbg(adapter->netdev, "%s: skb_buf_size %d, rx_buf_per_pkt %d,"
                " ring sizes %u %u %u\n", adapter->netdev->name,
                adapter->skb_buf_size, adapter->rx_buf_per_pkt,
                adapter->tx_queue[0].tx_ring.size,
                adapter->rx_queue[0].rx_ring[0].size,
                adapter->rx_queue[0].rx_ring[1].size);

        vmxnet3_tq_init_all(adapter);
        err = vmxnet3_rq_init_all(adapter);
        if (err) {
                netdev_err(adapter->netdev,
                           "Failed to init rx queue error %d\n", err);
                goto rq_err;
        }

        err = vmxnet3_request_irqs(adapter);
        if (err) {
                netdev_err(adapter->netdev,
                           "Failed to setup irq for error %d\n", err);
                goto irq_err;
        }

        vmxnet3_setup_driver_shared(adapter);

        VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_DSAL, VMXNET3_GET_ADDR_LO(
                               adapter->shared_pa));
        VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_DSAH, VMXNET3_GET_ADDR_HI(
                               adapter->shared_pa));
        spin_lock_irqsave(&adapter->cmd_lock, flags);
        VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_CMD,
                               VMXNET3_CMD_ACTIVATE_DEV);
        ret = VMXNET3_READ_BAR1_REG(adapter, VMXNET3_REG_CMD);
        spin_unlock_irqrestore(&adapter->cmd_lock, flags);

        if (ret != 0) {
                netdev_err(adapter->netdev,
                           "Failed to activate dev: error %u\n", ret);
                err = -EINVAL;
                goto activate_err;
        }

        vmxnet3_init_coalesce(adapter);
        vmxnet3_init_rssfields(adapter);

        for (i = 0; i < adapter->num_rx_queues; i++) {
                VMXNET3_WRITE_BAR0_REG(adapter,
                                VMXNET3_REG_RXPROD + i * VMXNET3_REG_ALIGN,
                                adapter->rx_queue[i].rx_ring[0].next2fill);
                VMXNET3_WRITE_BAR0_REG(adapter, (VMXNET3_REG_RXPROD2 +
                                (i * VMXNET3_REG_ALIGN)),
                                adapter->rx_queue[i].rx_ring[1].next2fill);
        }

        /* Apply the rx filter settins last. */
        vmxnet3_set_mc(adapter->netdev);

        /*
         * Check link state when first activating device. It will start the
         * tx queue if the link is up.
         */
        vmxnet3_check_link(adapter, true);
        netif_tx_wake_all_queues(adapter->netdev);
        for (i = 0; i < adapter->num_rx_queues; i++)
                napi_enable(&adapter->rx_queue[i].napi);
        vmxnet3_enable_all_intrs(adapter);
        clear_bit(VMXNET3_STATE_BIT_QUIESCED, &adapter->state);
        return 0;

activate_err:
        VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_DSAL, 0);
        VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_DSAH, 0);
        vmxnet3_free_irqs(adapter);
irq_err:
rq_err:
        /* free up buffers we allocated */
        vmxnet3_rq_cleanup_all(adapter);
        return err;
}


void
vmxnet3_reset_dev(struct vmxnet3_adapter *adapter)
{
        unsigned long flags;
        spin_lock_irqsave(&adapter->cmd_lock, flags);
        VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_CMD, VMXNET3_CMD_RESET_DEV);
        spin_unlock_irqrestore(&adapter->cmd_lock, flags);
}


int
vmxnet3_quiesce_dev(struct vmxnet3_adapter *adapter)
{
        int i;
        unsigned long flags;
        if (test_and_set_bit(VMXNET3_STATE_BIT_QUIESCED, &adapter->state))
                return 0;


        spin_lock_irqsave(&adapter->cmd_lock, flags);
        VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_CMD,
                               VMXNET3_CMD_QUIESCE_DEV);
        spin_unlock_irqrestore(&adapter->cmd_lock, flags);
        vmxnet3_disable_all_intrs(adapter);

        for (i = 0; i < adapter->num_rx_queues; i++)
                napi_disable(&adapter->rx_queue[i].napi);
        netif_tx_disable(adapter->netdev);
        adapter->link_speed = 0;
        netif_carrier_off(adapter->netdev);

        vmxnet3_tq_cleanup_all(adapter);
        vmxnet3_rq_cleanup_all(adapter);
        vmxnet3_free_irqs(adapter);
        return 0;
}


static void
vmxnet3_write_mac_addr(struct vmxnet3_adapter *adapter, u8 *mac)
{
        u32 tmp;

        tmp = *(u32 *)mac;
        VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_MACL, tmp);

        tmp = (mac[5] << 8) | mac[4];
        VMXNET3_WRITE_BAR1_REG(adapter, VMXNET3_REG_MACH, tmp);
}


static int
vmxnet3_set_mac_addr(struct net_device *netdev, void *p)
{
        struct sockaddr *addr = p;
        struct vmxnet3_adapter *adapter = netdev_priv(netdev);

        memcpy(netdev->dev_addr, addr->sa_data, netdev->addr_len);
        vmxnet3_write_mac_addr(adapter, addr->sa_data);

        return 0;
}


/* ==================== initialization and cleanup routines ============ */

static int
vmxnet3_alloc_pci_resources(struct vmxnet3_adapter *adapter)
{
        int err;
        unsigned long mmio_start, mmio_len;
        struct pci_dev *pdev = adapter->pdev;

        err = pci_enable_device(pdev);
        if (err) {
                dev_err(&pdev->dev, "Failed to enable adapter: error %d\n", err);
                return err;
        }

        err = pci_request_selected_regions(pdev, (1 << 2) - 1,
                                           vmxnet3_driver_name);
        if (err) {
                dev_err(&pdev->dev,
                        "Failed to request region for adapter: error %d\n", err);
                goto err_enable_device;
        }

        pci_set_master(pdev);

        mmio_start = pci_resource_start(pdev, 0);
        mmio_len = pci_resource_len(pdev, 0);
        adapter->hw_addr0 = ioremap(mmio_start, mmio_len);
        if (!adapter->hw_addr0) {
                dev_err(&pdev->dev, "Failed to map bar0\n");
                err = -EIO;
                goto err_ioremap;
        }

        mmio_start = pci_resource_start(pdev, 1);
        mmio_len = pci_resource_len(pdev, 1);
        adapter->hw_addr1 = ioremap(mmio_start, mmio_len);
        if (!adapter->hw_addr1) {
                dev_err(&pdev->dev, "Failed to map bar1\n");
                err = -EIO;
                goto err_bar1;
        }
        return 0;

err_bar1:
        iounmap(adapter->hw_addr0);
err_ioremap:
        pci_release_selected_regions(pdev, (1 << 2) - 1);
err_enable_device:
        pci_disable_device(pdev);
        return err;
}


static void
vmxnet3_free_pci_resources(struct vmxnet3_adapter *adapter)
{
        BUG_ON(!adapter->pdev);

        iounmap(adapter->hw_addr0);
        iounmap(adapter->hw_addr1);
        pci_release_selected_regions(adapter->pdev, (1 << 2) - 1);
        pci_disable_device(adapter->pdev);
}


static void
vmxnet3_adjust_rx_ring_size(struct vmxnet3_adapter *adapter)
{
        size_t sz, i, ring0_size, ring1_size, comp_size;
        if (adapter->netdev->mtu <= VMXNET3_MAX_SKB_BUF_SIZE -
                                    VMXNET3_MAX_ETH_HDR_SIZE) {
                adapter->skb_buf_size = adapter->netdev->mtu +
                                        VMXNET3_MAX_ETH_HDR_SIZE;
                if (adapter->skb_buf_size < VMXNET3_MIN_T0_BUF_SIZE)
                        adapter->skb_buf_size = VMXNET3_MIN_T0_BUF_SIZE;

                adapter->rx_buf_per_pkt = 1;
        } else {
                adapter->skb_buf_size = VMXNET3_MAX_SKB_BUF_SIZE;
                sz = adapter->netdev->mtu - VMXNET3_MAX_SKB_BUF_SIZE +
                                            VMXNET3_MAX_ETH_HDR_SIZE;
                adapter->rx_buf_per_pkt = 1 + (sz + PAGE_SIZE - 1) / PAGE_SIZE;
        }

        /*
         * for simplicity, force the ring0 size to be a multiple of
         * rx_buf_per_pkt * VMXNET3_RING_SIZE_ALIGN
         */
        sz = adapter->rx_buf_per_pkt * VMXNET3_RING_SIZE_ALIGN;
        ring0_size = adapter->rx_queue[0].rx_ring[0].size;
        ring0_size = (ring0_size + sz - 1) / sz * sz;
        ring0_size = min_t(u32, ring0_size, VMXNET3_RX_RING_MAX_SIZE /
                           sz * sz);
        ring1_size = adapter->rx_queue[0].rx_ring[1].size;
        ring1_size = (ring1_size + sz - 1) / sz * sz;
        ring1_size = min_t(u32, ring1_size, VMXNET3_RX_RING2_MAX_SIZE /
                           sz * sz);
        comp_size = ring0_size + ring1_size;

        for (i = 0; i < adapter->num_rx_queues; i++) {
                struct vmxnet3_rx_queue *rq = &adapter->rx_queue[i];

                rq->rx_ring[0].size = ring0_size;
                rq->rx_ring[1].size = ring1_size;
                rq->comp_ring.size = comp_size;
        }
}


int
vmxnet3_create_queues(struct vmxnet3_adapter *adapter, u32 tx_ring_size,
                      u32 rx_ring_size, u32 rx_ring2_size,
                      u16 txdata_desc_size, u16 rxdata_desc_size)
{
        int err = 0, i;

        for (i = 0; i < adapter->num_tx_queues; i++) {
                struct vmxnet3_tx_queue *tq = &adapter->tx_queue[i];
                tq->tx_ring.size   = tx_ring_size;
                tq->data_ring.size = tx_ring_size;
                tq->comp_ring.size = tx_ring_size;
                tq->txdata_desc_size = txdata_desc_size;
                tq->shared = &adapter->tqd_start[i].ctrl;
                tq->stopped = true;
                tq->adapter = adapter;
                tq->qid = i;
                err = vmxnet3_tq_create(tq, adapter);
                /*
                 * Too late to change num_tx_queues. We cannot do away with
                 * lesser number of queues than what we asked for
                 */
                if (err)
                        goto queue_err;
        }

        adapter->rx_queue[0].rx_ring[0].size = rx_ring_size;
        adapter->rx_queue[0].rx_ring[1].size = rx_ring2_size;
        vmxnet3_adjust_rx_ring_size(adapter);

        adapter->rxdataring_enabled = VMXNET3_VERSION_GE_3(adapter);
        for (i = 0; i < adapter->num_rx_queues; i++) {
                struct vmxnet3_rx_queue *rq = &adapter->rx_queue[i];
                /* qid and qid2 for rx queues will be assigned later when num
                 * of rx queues is finalized after allocating intrs */
                rq->shared = &adapter->rqd_start[i].ctrl;
                rq->adapter = adapter;
                rq->data_ring.desc_size = rxdata_desc_size;
                err = vmxnet3_rq_create(rq, adapter);
                if (err) {
                        if (i == 0) {
                                netdev_err(adapter->netdev,
                                           "Could not allocate any rx queues. "
                                           "Aborting.\n");
                                goto queue_err;
                        } else {
                                netdev_info(adapter->netdev,
                                            "Number of rx queues changed "
                                            "to : %d.\n", i);
                                adapter->num_rx_queues = i;
                                err = 0;
                                break;
                        }
                }
        }

        if (!adapter->rxdataring_enabled)