/*
 * Copyright (C) 2005 - 2011 Emulex
 * 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 version 2
 * as published by the Free Software Foundation.  The full GNU General
 * Public License is included in this distribution in the file called COPYING.
 *
 * Contact Information:
 * linux-drivers@emulex.com
 *
 * Emulex
 * 3333 Susan Street
 * Costa Mesa, CA 92626
 */

#include <linux/prefetch.h>
#include <linux/module.h>
#include "be.h"
#include "be_cmds.h"
#include <asm/div64.h>

MODULE_VERSION(DRV_VER);
MODULE_DEVICE_TABLE(pci, be_dev_ids);
MODULE_DESCRIPTION(DRV_DESC " " DRV_VER);
MODULE_AUTHOR("ServerEngines Corporation");
MODULE_LICENSE("GPL");

static unsigned int num_vfs;
module_param(num_vfs, uint, S_IRUGO);
MODULE_PARM_DESC(num_vfs, "Number of PCI VFs to initialize");

static ushort rx_frag_size = 2048;
module_param(rx_frag_size, ushort, S_IRUGO);
MODULE_PARM_DESC(rx_frag_size, "Size of a fragment that holds rcvd data.");

static DEFINE_PCI_DEVICE_TABLE(be_dev_ids) = {
        { PCI_DEVICE(BE_VENDOR_ID, BE_DEVICE_ID1) },
        { PCI_DEVICE(BE_VENDOR_ID, BE_DEVICE_ID2) },
        { PCI_DEVICE(BE_VENDOR_ID, OC_DEVICE_ID1) },
        { PCI_DEVICE(BE_VENDOR_ID, OC_DEVICE_ID2) },
        { PCI_DEVICE(EMULEX_VENDOR_ID, OC_DEVICE_ID3)},
        { PCI_DEVICE(EMULEX_VENDOR_ID, OC_DEVICE_ID4)},
        { PCI_DEVICE(EMULEX_VENDOR_ID, OC_DEVICE_ID5)},
        { 0 }
};
MODULE_DEVICE_TABLE(pci, be_dev_ids);
/* UE Status Low CSR */
static const char * const ue_status_low_desc[] = {
        "CEV",
        "CTX",
        "DBUF",
        "ERX",
        "Host",
        "MPU",
        "NDMA",
        "PTC ",
        "RDMA ",
        "RXF ",
        "RXIPS ",
        "RXULP0 ",
        "RXULP1 ",
        "RXULP2 ",
        "TIM ",
        "TPOST ",
        "TPRE ",
        "TXIPS ",
        "TXULP0 ",
        "TXULP1 ",
        "UC ",
        "WDMA ",
        "TXULP2 ",
        "HOST1 ",
        "P0_OB_LINK ",
        "P1_OB_LINK ",
        "HOST_GPIO ",
        "MBOX ",
        "AXGMAC0",
        "AXGMAC1",
        "JTAG",
        "MPU_INTPEND"
};
/* UE Status High CSR */
static const char * const ue_status_hi_desc[] = {
        "LPCMEMHOST",
        "MGMT_MAC",
        "PCS0ONLINE",
        "MPU_IRAM",
        "PCS1ONLINE",
        "PCTL0",
        "PCTL1",
        "PMEM",
        "RR",
        "TXPB",
        "RXPP",
        "XAUI",
        "TXP",
        "ARM",
        "IPC",
        "HOST2",
        "HOST3",
        "HOST4",
        "HOST5",
        "HOST6",
        "HOST7",
        "HOST8",
        "HOST9",
        "NETC",
        "Unknown",
        "Unknown",
        "Unknown",
        "Unknown",
        "Unknown",
        "Unknown",
        "Unknown",
        "Unknown"
};

/* Is BE in a multi-channel mode */
static inline bool be_is_mc(struct be_adapter *adapter) {
        return (adapter->function_mode & FLEX10_MODE ||
                adapter->function_mode & VNIC_MODE ||
                adapter->function_mode & UMC_ENABLED);
}

static void be_queue_free(struct be_adapter *adapter, struct be_queue_info *q)
{
        struct be_dma_mem *mem = &q->dma_mem;
        if (mem->va) {
                dma_free_coherent(&adapter->pdev->dev, mem->size, mem->va,
                                  mem->dma);
                mem->va = NULL;
        }
}

static int be_queue_alloc(struct be_adapter *adapter, struct be_queue_info *q,
                u16 len, u16 entry_size)
{
        struct be_dma_mem *mem = &q->dma_mem;

        memset(q, 0, sizeof(*q));
        q->len = len;
        q->entry_size = entry_size;
        mem->size = len * entry_size;
        mem->va = dma_alloc_coherent(&adapter->pdev->dev, mem->size, &mem->dma,
                                     GFP_KERNEL);
        if (!mem->va)
                return -ENOMEM;
        memset(mem->va, 0, mem->size);
        return 0;
}

static void be_intr_set(struct be_adapter *adapter, bool enable)
{
        u32 reg, enabled;

        if (adapter->eeh_err)
                return;

        pci_read_config_dword(adapter->pdev, PCICFG_MEMBAR_CTRL_INT_CTRL_OFFSET,
                                &reg);
        enabled = reg & MEMBAR_CTRL_INT_CTRL_HOSTINTR_MASK;

        if (!enabled && enable)
                reg |= MEMBAR_CTRL_INT_CTRL_HOSTINTR_MASK;
        else if (enabled && !enable)
                reg &= ~MEMBAR_CTRL_INT_CTRL_HOSTINTR_MASK;
        else
                return;

        pci_write_config_dword(adapter->pdev,
                        PCICFG_MEMBAR_CTRL_INT_CTRL_OFFSET, reg);
}

static void be_rxq_notify(struct be_adapter *adapter, u16 qid, u16 posted)
{
        u32 val = 0;
        val |= qid & DB_RQ_RING_ID_MASK;
        val |= posted << DB_RQ_NUM_POSTED_SHIFT;

        wmb();
        iowrite32(val, adapter->db + DB_RQ_OFFSET);
}

static void be_txq_notify(struct be_adapter *adapter, u16 qid, u16 posted)
{
        u32 val = 0;
        val |= qid & DB_TXULP_RING_ID_MASK;
        val |= (posted & DB_TXULP_NUM_POSTED_MASK) << DB_TXULP_NUM_POSTED_SHIFT;

        wmb();
        iowrite32(val, adapter->db + DB_TXULP1_OFFSET);
}

static void be_eq_notify(struct be_adapter *adapter, u16 qid,
                bool arm, bool clear_int, u16 num_popped)
{
        u32 val = 0;
        val |= qid & DB_EQ_RING_ID_MASK;
        val |= ((qid & DB_EQ_RING_ID_EXT_MASK) <<
                        DB_EQ_RING_ID_EXT_MASK_SHIFT);

        if (adapter->eeh_err)
                return;

        if (arm)
                val |= 1 << DB_EQ_REARM_SHIFT;
        if (clear_int)
                val |= 1 << DB_EQ_CLR_SHIFT;
        val |= 1 << DB_EQ_EVNT_SHIFT;
        val |= num_popped << DB_EQ_NUM_POPPED_SHIFT;
        iowrite32(val, adapter->db + DB_EQ_OFFSET);
}

void be_cq_notify(struct be_adapter *adapter, u16 qid, bool arm, u16 num_popped)
{
        u32 val = 0;
        val |= qid & DB_CQ_RING_ID_MASK;
        val |= ((qid & DB_CQ_RING_ID_EXT_MASK) <<
                        DB_CQ_RING_ID_EXT_MASK_SHIFT);

        if (adapter->eeh_err)
                return;

        if (arm)
                val |= 1 << DB_CQ_REARM_SHIFT;
        val |= num_popped << DB_CQ_NUM_POPPED_SHIFT;
        iowrite32(val, adapter->db + DB_CQ_OFFSET);
}

static int be_mac_addr_set(struct net_device *netdev, void *p)
{
        struct be_adapter *adapter = netdev_priv(netdev);
        struct sockaddr *addr = p;
        int status = 0;
        u8 current_mac[ETH_ALEN];
        u32 pmac_id = adapter->pmac_id[0];

        if (!is_valid_ether_addr(addr->sa_data))
                return -EADDRNOTAVAIL;

        status = be_cmd_mac_addr_query(adapter, current_mac,
                                MAC_ADDRESS_TYPE_NETWORK, false,
                                adapter->if_handle, 0);
        if (status)
                goto err;

        if (memcmp(addr->sa_data, current_mac, ETH_ALEN)) {
                status = be_cmd_pmac_add(adapter, (u8 *)addr->sa_data,
                                adapter->if_handle, &adapter->pmac_id[0], 0);
                if (status)
                        goto err;

                be_cmd_pmac_del(adapter, adapter->if_handle, pmac_id, 0);
        }
        memcpy(netdev->dev_addr, addr->sa_data, netdev->addr_len);
        return 0;
err:
        dev_err(&adapter->pdev->dev, "MAC %pM set Failed\n", addr->sa_data);
        return status;
}

static void populate_be2_stats(struct be_adapter *adapter)
{
        struct be_hw_stats_v0 *hw_stats = hw_stats_from_cmd(adapter);
        struct be_pmem_stats *pmem_sts = &hw_stats->pmem;
        struct be_rxf_stats_v0 *rxf_stats = &hw_stats->rxf;
        struct be_port_rxf_stats_v0 *port_stats =
                                        &rxf_stats->port[adapter->port_num];
        struct be_drv_stats *drvs = &adapter->drv_stats;

        be_dws_le_to_cpu(hw_stats, sizeof(*hw_stats));
        drvs->rx_pause_frames = port_stats->rx_pause_frames;
        drvs->rx_crc_errors = port_stats->rx_crc_errors;
        drvs->rx_control_frames = port_stats->rx_control_frames;
        drvs->rx_in_range_errors = port_stats->rx_in_range_errors;
        drvs->rx_frame_too_long = port_stats->rx_frame_too_long;
        drvs->rx_dropped_runt = port_stats->rx_dropped_runt;
        drvs->rx_ip_checksum_errs = port_stats->rx_ip_checksum_errs;
        drvs->rx_tcp_checksum_errs = port_stats->rx_tcp_checksum_errs;
        drvs->rx_udp_checksum_errs = port_stats->rx_udp_checksum_errs;
        drvs->rxpp_fifo_overflow_drop = port_stats->rx_fifo_overflow;
        drvs->rx_dropped_tcp_length = port_stats->rx_dropped_tcp_length;
        drvs->rx_dropped_too_small = port_stats->rx_dropped_too_small;
        drvs->rx_dropped_too_short = port_stats->rx_dropped_too_short;
        drvs->rx_out_range_errors = port_stats->rx_out_range_errors;
        drvs->rx_input_fifo_overflow_drop = port_stats->rx_input_fifo_overflow;
        drvs->rx_dropped_header_too_small =
                port_stats->rx_dropped_header_too_small;
        drvs->rx_address_mismatch_drops =
                                        port_stats->rx_address_mismatch_drops +
                                        port_stats->rx_vlan_mismatch_drops;
        drvs->rx_alignment_symbol_errors =
                port_stats->rx_alignment_symbol_errors;

        drvs->tx_pauseframes = port_stats->tx_pauseframes;
        drvs->tx_controlframes = port_stats->tx_controlframes;

        if (adapter->port_num)
                drvs->jabber_events = rxf_stats->port1_jabber_events;
        else
                drvs->jabber_events = rxf_stats->port0_jabber_events;
        drvs->rx_drops_no_pbuf = rxf_stats->rx_drops_no_pbuf;
        drvs->rx_drops_no_erx_descr = rxf_stats->rx_drops_no_erx_descr;
        drvs->forwarded_packets = rxf_stats->forwarded_packets;
        drvs->rx_drops_mtu = rxf_stats->rx_drops_mtu;
        drvs->rx_drops_no_tpre_descr = rxf_stats->rx_drops_no_tpre_descr;
        drvs->rx_drops_too_many_frags = rxf_stats->rx_drops_too_many_frags;
        adapter->drv_stats.eth_red_drops = pmem_sts->eth_red_drops;
}

static void populate_be3_stats(struct be_adapter *adapter)
{
        struct be_hw_stats_v1 *hw_stats = hw_stats_from_cmd(adapter);
        struct be_pmem_stats *pmem_sts = &hw_stats->pmem;
        struct be_rxf_stats_v1 *rxf_stats = &hw_stats->rxf;
        struct be_port_rxf_stats_v1 *port_stats =
                                        &rxf_stats->port[adapter->port_num];
        struct be_drv_stats *drvs = &adapter->drv_stats;

        be_dws_le_to_cpu(hw_stats, sizeof(*hw_stats));
        drvs->pmem_fifo_overflow_drop = port_stats->pmem_fifo_overflow_drop;
        drvs->rx_priority_pause_frames = port_stats->rx_priority_pause_frames;
        drvs->rx_pause_frames = port_stats->rx_pause_frames;
        drvs->rx_crc_errors = port_stats->rx_crc_errors;
        drvs->rx_control_frames = port_stats->rx_control_frames;
        drvs->rx_in_range_errors = port_stats->rx_in_range_errors;
        drvs->rx_frame_too_long = port_stats->rx_frame_too_long;
        drvs->rx_dropped_runt = port_stats->rx_dropped_runt;
        drvs->rx_ip_checksum_errs = port_stats->rx_ip_checksum_errs;
        drvs->rx_tcp_checksum_errs = port_stats->rx_tcp_checksum_errs;
        drvs->rx_udp_checksum_errs = port_stats->rx_udp_checksum_errs;
        drvs->rx_dropped_tcp_length = port_stats->rx_dropped_tcp_length;
        drvs->rx_dropped_too_small = port_stats->rx_dropped_too_small;
        drvs->rx_dropped_too_short = port_stats->rx_dropped_too_short;
        drvs->rx_out_range_errors = port_stats->rx_out_range_errors;
        drvs->rx_dropped_header_too_small =
                port_stats->rx_dropped_header_too_small;
        drvs->rx_input_fifo_overflow_drop =
                port_stats->rx_input_fifo_overflow_drop;
        drvs->rx_address_mismatch_drops = port_stats->rx_address_mismatch_drops;
        drvs->rx_alignment_symbol_errors =
                port_stats->rx_alignment_symbol_errors;
        drvs->rxpp_fifo_overflow_drop = port_stats->rxpp_fifo_overflow_drop;
        drvs->tx_pauseframes = port_stats->tx_pauseframes;
        drvs->tx_controlframes = port_stats->tx_controlframes;
        drvs->jabber_events = port_stats->jabber_events;
        drvs->rx_drops_no_pbuf = rxf_stats->rx_drops_no_pbuf;
        drvs->rx_drops_no_erx_descr = rxf_stats->rx_drops_no_erx_descr;
        drvs->forwarded_packets = rxf_stats->forwarded_packets;
        drvs->rx_drops_mtu = rxf_stats->rx_drops_mtu;
        drvs->rx_drops_no_tpre_descr = rxf_stats->rx_drops_no_tpre_descr;
        drvs->rx_drops_too_many_frags = rxf_stats->rx_drops_too_many_frags;
        adapter->drv_stats.eth_red_drops = pmem_sts->eth_red_drops;
}

static void populate_lancer_stats(struct be_adapter *adapter)
{

        struct be_drv_stats *drvs = &adapter->drv_stats;
        struct lancer_pport_stats *pport_stats =
                                        pport_stats_from_cmd(adapter);

        be_dws_le_to_cpu(pport_stats, sizeof(*pport_stats));
        drvs->rx_pause_frames = pport_stats->rx_pause_frames_lo;
        drvs->rx_crc_errors = pport_stats->rx_crc_errors_lo;
        drvs->rx_control_frames = pport_stats->rx_control_frames_lo;
        drvs->rx_in_range_errors = pport_stats->rx_in_range_errors;
        drvs->rx_frame_too_long = pport_stats->rx_frames_too_long_lo;
        drvs->rx_dropped_runt = pport_stats->rx_dropped_runt;
        drvs->rx_ip_checksum_errs = pport_stats->rx_ip_checksum_errors;
        drvs->rx_tcp_checksum_errs = pport_stats->rx_tcp_checksum_errors;
        drvs->rx_udp_checksum_errs = pport_stats->rx_udp_checksum_errors;
        drvs->rx_dropped_tcp_length =
                                pport_stats->rx_dropped_invalid_tcp_length;
        drvs->rx_dropped_too_small = pport_stats->rx_dropped_too_small;
        drvs->rx_dropped_too_short = pport_stats->rx_dropped_too_short;
        drvs->rx_out_range_errors = pport_stats->rx_out_of_range_errors;
        drvs->rx_dropped_header_too_small =
                                pport_stats->rx_dropped_header_too_small;
        drvs->rx_input_fifo_overflow_drop = pport_stats->rx_fifo_overflow;
        drvs->rx_address_mismatch_drops =
                                        pport_stats->rx_address_mismatch_drops +
                                        pport_stats->rx_vlan_mismatch_drops;
        drvs->rx_alignment_symbol_errors = pport_stats->rx_symbol_errors_lo;
        drvs->rxpp_fifo_overflow_drop = pport_stats->rx_fifo_overflow;
        drvs->tx_pauseframes = pport_stats->tx_pause_frames_lo;
        drvs->tx_controlframes = pport_stats->tx_control_frames_lo;
        drvs->jabber_events = pport_stats->rx_jabbers;
        drvs->forwarded_packets = pport_stats->num_forwards_lo;
        drvs->rx_drops_mtu = pport_stats->rx_drops_mtu_lo;
        drvs->rx_drops_too_many_frags =
                                pport_stats->rx_drops_too_many_frags_lo;
}

static void accumulate_16bit_val(u32 *acc, u16 val)
{
#define lo(x)                   (x & 0xFFFF)
#define hi(x)                   (x & 0xFFFF0000)
        bool wrapped = val < lo(*acc);
        u32 newacc = hi(*acc) + val;

        if (wrapped)
                newacc += 65536;
        ACCESS_ONCE(*acc) = newacc;
}

void be_parse_stats(struct be_adapter *adapter)
{
        struct be_erx_stats_v1 *erx = be_erx_stats_from_cmd(adapter);
        struct be_rx_obj *rxo;
        int i;

        if (adapter->generation == BE_GEN3) {
                if (lancer_chip(adapter))
                        populate_lancer_stats(adapter);
                 else
                        populate_be3_stats(adapter);
        } else {
                populate_be2_stats(adapter);
        }

        /* as erx_v1 is longer than v0, ok to use v1 defn for v0 access */
        for_all_rx_queues(adapter, rxo, i) {
                /* below erx HW counter can actually wrap around after
                 * 65535. Driver accumulates a 32-bit value
                 */
                accumulate_16bit_val(&rx_stats(rxo)->rx_drops_no_frags,
                                (u16)erx->rx_drops_no_fragments[rxo->q.id]);
        }
}

static struct rtnl_link_stats64 *be_get_stats64(struct net_device *netdev,
                                        struct rtnl_link_stats64 *stats)
{
        struct be_adapter *adapter = netdev_priv(netdev);
        struct be_drv_stats *drvs = &adapter->drv_stats;
        struct be_rx_obj *rxo;
        struct be_tx_obj *txo;
        u64 pkts, bytes;
        unsigned int start;
        int i;

        for_all_rx_queues(adapter, rxo, i) {
                const struct be_rx_stats *rx_stats = rx_stats(rxo);
                do {
                        start = u64_stats_fetch_begin_bh(&rx_stats->sync);
                        pkts = rx_stats(rxo)->rx_pkts;
                        bytes = rx_stats(rxo)->rx_bytes;
                } while (u64_stats_fetch_retry_bh(&rx_stats->sync, start));
                stats->rx_packets += pkts;
                stats->rx_bytes += bytes;
                stats->multicast += rx_stats(rxo)->rx_mcast_pkts;
                stats->rx_dropped += rx_stats(rxo)->rx_drops_no_skbs +
                                        rx_stats(rxo)->rx_drops_no_frags;
        }

        for_all_tx_queues(adapter, txo, i) {
                const struct be_tx_stats *tx_stats = tx_stats(txo);
                do {
                        start = u64_stats_fetch_begin_bh(&tx_stats->sync);
                        pkts = tx_stats(txo)->tx_pkts;
                        bytes = tx_stats(txo)->tx_bytes;
                } while (u64_stats_fetch_retry_bh(&tx_stats->sync, start));
                stats->tx_packets += pkts;
                stats->tx_bytes += bytes;
        }

        /* bad pkts received */
        stats->rx_errors = drvs->rx_crc_errors +
                drvs->rx_alignment_symbol_errors +
                drvs->rx_in_range_errors +
                drvs->rx_out_range_errors +
                drvs->rx_frame_too_long +
                drvs->rx_dropped_too_small +
                drvs->rx_dropped_too_short +
                drvs->rx_dropped_header_too_small +
                drvs->rx_dropped_tcp_length +
                drvs->rx_dropped_runt;

        /* detailed rx errors */
        stats->rx_length_errors = drvs->rx_in_range_errors +
                drvs->rx_out_range_errors +
                drvs->rx_frame_too_long;

        stats->rx_crc_errors = drvs->rx_crc_errors;

        /* frame alignment errors */
        stats->rx_frame_errors = drvs->rx_alignment_symbol_errors;

        /* receiver fifo overrun */
        /* drops_no_pbuf is no per i/f, it's per BE card */
        stats->rx_fifo_errors = drvs->rxpp_fifo_overflow_drop +
                                drvs->rx_input_fifo_overflow_drop +
                                drvs->rx_drops_no_pbuf;
        return stats;
}

void be_link_status_update(struct be_adapter *adapter, u8 link_status)
{
        struct net_device *netdev = adapter->netdev;

        if (!(adapter->flags & BE_FLAGS_LINK_STATUS_INIT)) {
                netif_carrier_off(netdev);
                adapter->flags |= BE_FLAGS_LINK_STATUS_INIT;
        }

        if ((link_status & LINK_STATUS_MASK) == LINK_UP)
                netif_carrier_on(netdev);
        else
                netif_carrier_off(netdev);
}

static void be_tx_stats_update(struct be_tx_obj *txo,
                        u32 wrb_cnt, u32 copied, u32 gso_segs, bool stopped)
{
        struct be_tx_stats *stats = tx_stats(txo);

        u64_stats_update_begin(&stats->sync);
        stats->tx_reqs++;
        stats->tx_wrbs += wrb_cnt;
        stats->tx_bytes += copied;
        stats->tx_pkts += (gso_segs ? gso_segs : 1);
        if (stopped)
                stats->tx_stops++;
        u64_stats_update_end(&stats->sync);
}

/* Determine number of WRB entries needed to xmit data in an skb */
static u32 wrb_cnt_for_skb(struct be_adapter *adapter, struct sk_buff *skb,
                                                                bool *dummy)
{
        int cnt = (skb->len > skb->data_len);

        cnt += skb_shinfo(skb)->nr_frags;

        /* to account for hdr wrb */
        cnt++;
        if (lancer_chip(adapter) || !(cnt & 1)) {
                *dummy = false;
        } else {
                /* add a dummy to make it an even num */
                cnt++;
                *dummy = true;
        }
        BUG_ON(cnt > BE_MAX_TX_FRAG_COUNT);
        return cnt;
}

static inline void wrb_fill(struct be_eth_wrb *wrb, u64 addr, int len)
{
        wrb->frag_pa_hi = upper_32_bits(addr);
        wrb->frag_pa_lo = addr & 0xFFFFFFFF;
        wrb->frag_len = len & ETH_WRB_FRAG_LEN_MASK;
}

static inline u16 be_get_tx_vlan_tag(struct be_adapter *adapter,
                                        struct sk_buff *skb)
{
        u8 vlan_prio;
        u16 vlan_tag;

        vlan_tag = vlan_tx_tag_get(skb);
        vlan_prio = (vlan_tag & VLAN_PRIO_MASK) >> VLAN_PRIO_SHIFT;
        /* If vlan priority provided by OS is NOT in available bmap */
        if (!(adapter->vlan_prio_bmap & (1 << vlan_prio)))
                vlan_tag = (vlan_tag & ~VLAN_PRIO_MASK) |
                                adapter->recommended_prio;

        return vlan_tag;
}

static void wrb_fill_hdr(struct be_adapter *adapter, struct be_eth_hdr_wrb *hdr,
                struct sk_buff *skb, u32 wrb_cnt, u32 len)
{
        u16 vlan_tag;

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

        AMAP_SET_BITS(struct amap_eth_hdr_wrb, crc, hdr, 1);

        if (skb_is_gso(skb)) {
                AMAP_SET_BITS(struct amap_eth_hdr_wrb, lso, hdr, 1);
                AMAP_SET_BITS(struct amap_eth_hdr_wrb, lso_mss,
                        hdr, skb_shinfo(skb)->gso_size);
                if (skb_is_gso_v6(skb) && !lancer_chip(adapter))
                        AMAP_SET_BITS(struct amap_eth_hdr_wrb, lso6, hdr, 1);
                if (lancer_chip(adapter) && adapter->sli_family  ==
                                                        LANCER_A0_SLI_FAMILY) {
                        AMAP_SET_BITS(struct amap_eth_hdr_wrb, ipcs, hdr, 1);
                        if (is_tcp_pkt(skb))
                                AMAP_SET_BITS(struct amap_eth_hdr_wrb,
                                                                tcpcs, hdr, 1);
                        else if (is_udp_pkt(skb))
                                AMAP_SET_BITS(struct amap_eth_hdr_wrb,
                                                                udpcs, hdr, 1);
                }
        } else if (skb->ip_summed == CHECKSUM_PARTIAL) {
                if (is_tcp_pkt(skb))
                        AMAP_SET_BITS(struct amap_eth_hdr_wrb, tcpcs, hdr, 1);
                else if (is_udp_pkt(skb))
                        AMAP_SET_BITS(struct amap_eth_hdr_wrb, udpcs, hdr, 1);
        }

        if (vlan_tx_tag_present(skb)) {
                AMAP_SET_BITS(struct amap_eth_hdr_wrb, vlan, hdr, 1);
                vlan_tag = be_get_tx_vlan_tag(adapter, skb);
                AMAP_SET_BITS(struct amap_eth_hdr_wrb, vlan_tag, hdr, vlan_tag);
        }

        AMAP_SET_BITS(struct amap_eth_hdr_wrb, event, hdr, 1);
        AMAP_SET_BITS(struct amap_eth_hdr_wrb, complete, hdr, 1);
        AMAP_SET_BITS(struct amap_eth_hdr_wrb, num_wrb, hdr, wrb_cnt);
        AMAP_SET_BITS(struct amap_eth_hdr_wrb, len, hdr, len);
}

static void unmap_tx_frag(struct device *dev, struct be_eth_wrb *wrb,
                bool unmap_single)
{
        dma_addr_t dma;

        be_dws_le_to_cpu(wrb, sizeof(*wrb));

        dma = (u64)wrb->frag_pa_hi << 32 | (u64)wrb->frag_pa_lo;
        if (wrb->frag_len) {
                if (unmap_single)
                        dma_unmap_single(dev, dma, wrb->frag_len,
                                         DMA_TO_DEVICE);
                else
                        dma_unmap_page(dev, dma, wrb->frag_len, DMA_TO_DEVICE);
        }
}

static int make_tx_wrbs(struct be_adapter *adapter, struct be_queue_info *txq,
                struct sk_buff *skb, u32 wrb_cnt, bool dummy_wrb)
{
        dma_addr_t busaddr;
        int i, copied = 0;
        struct device *dev = &adapter->pdev->dev;
        struct sk_buff *first_skb = skb;
        struct be_eth_wrb *wrb;
        struct be_eth_hdr_wrb *hdr;
        bool map_single = false;
        u16 map_head;

        hdr = queue_head_node(txq);
        queue_head_inc(txq);
        map_head = txq->head;

        if (skb->len > skb->data_len) {
                int len = skb_headlen(skb);
                busaddr = dma_map_single(dev, skb->data, len, DMA_TO_DEVICE);
                if (dma_mapping_error(dev, busaddr))
                        goto dma_err;
                map_single = true;
                wrb = queue_head_node(txq);
                wrb_fill(wrb, busaddr, len);
                be_dws_cpu_to_le(wrb, sizeof(*wrb));
                queue_head_inc(txq);
                copied += len;
        }

        for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
                const struct skb_frag_struct *frag =
                        &skb_shinfo(skb)->frags[i];
                busaddr = skb_frag_dma_map(dev, frag, 0,
                                           skb_frag_size(frag), DMA_TO_DEVICE);
                if (dma_mapping_error(dev, busaddr))
                        goto dma_err;
                wrb = queue_head_node(txq);
                wrb_fill(wrb, busaddr, skb_frag_size(frag));
                be_dws_cpu_to_le(wrb, sizeof(*wrb));
                queue_head_inc(txq);
                copied += skb_frag_size(frag);
        }

        if (dummy_wrb) {
                wrb = queue_head_node(txq);
                wrb_fill(wrb, 0, 0);
                be_dws_cpu_to_le(wrb, sizeof(*wrb));
                queue_head_inc(txq);
        }

        wrb_fill_hdr(adapter, hdr, first_skb, wrb_cnt, copied);
        be_dws_cpu_to_le(hdr, sizeof(*hdr));

        return copied;
dma_err:
        txq->head = map_head;
        while (copied) {
                wrb = queue_head_node(txq);
                unmap_tx_frag(dev, wrb, map_single);
                map_single = false;
                copied -= wrb->frag_len;
                queue_head_inc(txq);
        }
        return 0;
}

static netdev_tx_t be_xmit(struct sk_buff *skb,
                        struct net_device *netdev)
{
        struct be_adapter *adapter = netdev_priv(netdev);
        struct be_tx_obj *txo = &adapter->tx_obj[skb_get_queue_mapping(skb)];
        struct be_queue_info *txq = &txo->q;
        u32 wrb_cnt = 0, copied = 0;
        u32 start = txq->head;
        bool dummy_wrb, stopped = false;

        /* For vlan tagged pkts, BE
         * 1) calculates checksum even when CSO is not requested
         * 2) calculates checksum wrongly for padded pkt less than
         * 60 bytes long.
         * As a workaround disable TX vlan offloading in such cases.
         */
        if (unlikely(vlan_tx_tag_present(skb) &&
                     (skb->ip_summed != CHECKSUM_PARTIAL || skb->len <= 60))) {
                skb = skb_share_check(skb, GFP_ATOMIC);
                if (unlikely(!skb))
                        goto tx_drop;

                skb = __vlan_put_tag(skb, be_get_tx_vlan_tag(adapter, skb));
                if (unlikely(!skb))
                        goto tx_drop;

                skb->vlan_tci = 0;
        }

        wrb_cnt = wrb_cnt_for_skb(adapter, skb, &dummy_wrb);

        copied = make_tx_wrbs(adapter, txq, skb, wrb_cnt, dummy_wrb);
        if (copied) {
                /* record the sent skb in the sent_skb table */
                BUG_ON(txo->sent_skb_list[start]);
                txo->sent_skb_list[start] = skb;

                /* Ensure txq has space for the next skb; Else stop the queue
                 * *BEFORE* ringing the tx doorbell, so that we serialze the
                 * tx compls of the current transmit which'll wake up the queue
                 */
                atomic_add(wrb_cnt, &txq->used);
                if ((BE_MAX_TX_FRAG_COUNT + atomic_read(&txq->used)) >=
                                                                txq->len) {
                        netif_stop_subqueue(netdev, skb_get_queue_mapping(skb));
                        stopped = true;
                }

                be_txq_notify(adapter, txq->id, wrb_cnt);

                be_tx_stats_update(txo, wrb_cnt, copied,
                                skb_shinfo(skb)->gso_segs, stopped);
        } else {
                txq->head = start;
                dev_kfree_skb_any(skb);
        }
tx_drop:
        return NETDEV_TX_OK;
}

static int be_change_mtu(struct net_device *netdev, int new_mtu)
{
        struct be_adapter *adapter = netdev_priv(netdev);
        if (new_mtu < BE_MIN_MTU ||
                        new_mtu > (BE_MAX_JUMBO_FRAME_SIZE -
                                        (ETH_HLEN + ETH_FCS_LEN))) {
                dev_info(&adapter->pdev->dev,
                        "MTU must be between %d and %d bytes\n",
                        BE_MIN_MTU,
                        (BE_MAX_JUMBO_FRAME_SIZE - (ETH_HLEN + ETH_FCS_LEN)));
                return -EINVAL;
        }
        dev_info(&adapter->pdev->dev, "MTU changed from %d to %d bytes\n",
                        netdev->mtu, new_mtu);
        netdev->mtu = new_mtu;
        return 0;
}

/*
 * A max of 64 (BE_NUM_VLANS_SUPPORTED) vlans can be configured in BE.
 * If the user configures more, place BE in vlan promiscuous mode.
 */
static int be_vid_config(struct be_adapter *adapter, bool vf, u32 vf_num)
{
        struct be_vf_cfg *vf_cfg = &adapter->vf_cfg[vf_num];
        u16 vtag[BE_NUM_VLANS_SUPPORTED];
        u16 ntags = 0, i;
        int status = 0;

        if (vf) {
                vtag[0] = cpu_to_le16(vf_cfg->vlan_tag);
                status = be_cmd_vlan_config(adapter, vf_cfg->if_handle, vtag,
                                            1, 1, 0);
        }

        /* No need to further configure vids if in promiscuous mode */
        if (adapter->promiscuous)
                return 0;

        if (adapter->vlans_added <= adapter->max_vlans)  {
                /* Construct VLAN Table to give to HW */
                for (i = 0; i < VLAN_N_VID; i++) {
                        if (adapter->vlan_tag[i]) {
                                vtag[ntags] = cpu_to_le16(i);
                                ntags++;
                        }
                }
                status = be_cmd_vlan_config(adapter, adapter->if_handle,
                                        vtag, ntags, 1, 0);
        } else {
                status = be_cmd_vlan_config(adapter, adapter->if_handle,
                                        NULL, 0, 1, 1);
        }

        return status;
}

static int be_vlan_add_vid(struct net_device *netdev, u16 vid)
{
        struct be_adapter *adapter = netdev_priv(netdev);
        int status = 0;

        if (!be_physfn(adapter)) {
                status = -EINVAL;
                goto ret;
        }

        adapter->vlan_tag[vid] = 1;
        if (adapter->vlans_added <= (adapter->max_vlans + 1))
                status = be_vid_config(adapter, false, 0);

        if (!status)
                adapter->vlans_added++;
        else
                adapter->vlan_tag[vid] = 0;
ret:
        return status;
}

static int be_vlan_rem_vid(struct net_device *netdev, u16 vid)
{
        struct be_adapter *adapter = netdev_priv(netdev);
        int status = 0;

        if (!be_physfn(adapter)) {
                status = -EINVAL;
                goto ret;
        }

        adapter->vlan_tag[vid] = 0;
        if (adapter->vlans_added <= adapter->max_vlans)
                status = be_vid_config(adapter, false, 0);

        if (!status)
                adapter->vlans_added--;
        else
                adapter->vlan_tag[vid] = 1;
ret:
        return status;
}

static void be_set_rx_mode(struct net_device *netdev)
{
        struct be_adapter *adapter = netdev_priv(netdev);

        if (netdev->flags & IFF_PROMISC) {
                be_cmd_rx_filter(adapter, IFF_PROMISC, ON);
                adapter->promiscuous = true;
                goto done;
        }

        /* BE was previously in promiscuous mode; disable it */
        if (adapter->promiscuous) {
                adapter->promiscuous = false;
                be_cmd_rx_filter(adapter, IFF_PROMISC, OFF);

                if (adapter->vlans_added)
                        be_vid_config(adapter, false, 0);
        }

        /* Enable multicast promisc if num configured exceeds what we support */
        if (netdev->flags & IFF_ALLMULTI ||
                        netdev_mc_count(netdev) > BE_MAX_MC) {
                be_cmd_rx_filter(adapter, IFF_ALLMULTI, ON);
                goto done;
        }

        if (netdev_uc_count(netdev) != adapter->uc_macs) {
                struct netdev_hw_addr *ha;
                int i = 1; /* First slot is claimed by the Primary MAC */

                for (; adapter->uc_macs > 0; adapter->uc_macs--, i++) {
                        be_cmd_pmac_del(adapter, adapter->if_handle,
                                        adapter->pmac_id[i], 0);
                }

                if (netdev_uc_count(netdev) > adapter->max_pmac_cnt) {
                        be_cmd_rx_filter(adapter, IFF_PROMISC, ON);
                        adapter->promiscuous = true;
                        goto done;
                }

                netdev_for_each_uc_addr(ha, adapter->netdev) {
                        adapter->uc_macs++; /* First slot is for Primary MAC */
                        be_cmd_pmac_add(adapter, (u8 *)ha->addr,
                                        adapter->if_handle,
                                        &adapter->pmac_id[adapter->uc_macs], 0);
                }
        }

        be_cmd_rx_filter(adapter, IFF_MULTICAST, ON);
done:
        return;
}

static int be_set_vf_mac(struct net_device *netdev, int vf, u8 *mac)
{
        struct be_adapter *adapter = netdev_priv(netdev);
        struct be_vf_cfg *vf_cfg = &adapter->vf_cfg[vf];
        int status;

        if (!sriov_enabled(adapter))
                return -EPERM;

        if (!is_valid_ether_addr(mac) || vf >= adapter->num_vfs)
                return -EINVAL;

        if (lancer_chip(adapter)) {
                status = be_cmd_set_mac_list(adapter,  mac, 1, vf + 1);
        } else {
                status = be_cmd_pmac_del(adapter, vf_cfg->if_handle,
                                         vf_cfg->pmac_id, vf + 1);

                status = be_cmd_pmac_add(adapter, mac, vf_cfg->if_handle,
                                         &vf_cfg->pmac_id, vf + 1);
        }

        if (status)
                dev_err(&adapter->pdev->dev, "MAC %pM set on VF %d Failed\n",
                                mac, vf);
        else
                memcpy(vf_cfg->mac_addr, mac, ETH_ALEN);

        return status;
}

static int be_get_vf_config(struct net_device *netdev, int vf,
                        struct ifla_vf_info *vi)
{
        struct be_adapter *adapter = netdev_priv(netdev);
        struct be_vf_cfg *vf_cfg = &adapter->vf_cfg[vf];

        if (!sriov_enabled(adapter))
                return -EPERM;

        if (vf >= adapter->num_vfs)
                return -EINVAL;

        vi->vf = vf;
        vi->tx_rate = vf_cfg->tx_rate;
        vi->vlan = vf_cfg->vlan_tag;
        vi->qos = 0;
        memcpy(&vi->mac, vf_cfg->mac_addr, ETH_ALEN);

        return 0;
}

static int be_set_vf_vlan(struct net_device *netdev,
                        int vf, u16 vlan, u8 qos)
{
        struct be_adapter *adapter = netdev_priv(netdev);
        int status = 0;

        if (!sriov_enabled(adapter))
                return -EPERM;

        if (vf >= adapter->num_vfs || vlan > 4095)
                return -EINVAL;

        if (vlan) {
                if (adapter->vf_cfg[vf].vlan_tag != vlan) {
                        /* If this is new value, program it. Else skip. */
                        adapter->vf_cfg[vf].vlan_tag = vlan;

                        status = be_cmd_set_hsw_config(adapter, vlan,
                                vf + 1, adapter->vf_cfg[vf].if_handle);
                }
        } else {
                /* Reset Transparent Vlan Tagging. */
                adapter->vf_cfg[vf].vlan_tag = 0;
                vlan = adapter->vf_cfg[vf].def_vid;
                status = be_cmd_set_hsw_config(adapter, vlan, vf + 1,
                        adapter->vf_cfg[vf].if_handle);
        }


        if (status)
                dev_info(&adapter->pdev->dev,
                                "VLAN %d config on VF %d failed\n", vlan, vf);
        return status;

}

static int be_set_vf_tx_rate(struct net_device *netdev,
                        int vf, int rate)
{
        struct be_adapter *adapter = netdev_priv(netdev);
        int status = 0;

        if (!sriov_enabled(adapter))
                return -EPERM;

        if (vf >= adapter->num_vfs)
                return -EINVAL;

        if (rate < 100 || rate > 10000) {
                dev_err(&adapter->pdev->dev,
                        "tx rate must be between 100 and 10000 Mbps\n");
                return -EINVAL;
        }

        status = be_cmd_set_qos(adapter, rate / 10, vf + 1);

        if (status)
                dev_err(&adapter->pdev->dev,
                                "tx rate %d on VF %d failed\n", rate, vf);
        else
                adapter->vf_cfg[vf].tx_rate = rate;
        return status;
}

static void be_eqd_update(struct be_adapter *adapter, struct be_eq_obj *eqo)
{
        struct be_rx_stats *stats = rx_stats(&adapter->rx_obj[eqo->idx]);
        ulong now = jiffies;
        ulong delta = now - stats->rx_jiffies;
        u64 pkts;
        unsigned int start, eqd;

        if (!eqo->enable_aic) {
                eqd = eqo->eqd;
                goto modify_eqd;
        }

        if (eqo->idx >= adapter->num_rx_qs)
                return;

        stats = rx_stats(&adapter->rx_obj[eqo->idx]);

        /* Wrapped around */
        if (time_before(now, stats->rx_jiffies)) {
                stats->rx_jiffies = now;
                return;
        }

        /* Update once a second */
        if (delta < HZ)
                return;

        do {
                start = u64_stats_fetch_begin_bh(&stats->sync);
                pkts = stats->rx_pkts;
        } while (u64_stats_fetch_retry_bh(&stats->sync, start));

        stats->rx_pps = (unsigned long)(pkts - stats->rx_pkts_prev) / (delta / HZ);
        stats->rx_pkts_prev = pkts;
        stats->rx_jiffies = now;
        eqd = (stats->rx_pps / 110000) << 3;
        eqd = min(eqd, eqo->max_eqd);
        eqd = max(eqd, eqo->min_eqd);
        if (eqd < 10)
                eqd = 0;

modify_eqd:
        if (eqd != eqo->cur_eqd) {
                be_cmd_modify_eqd(adapter, eqo->q.id, eqd);
                eqo->cur_eqd = eqd;
        }
}

static void be_rx_stats_update(struct be_rx_obj *rxo,
                struct be_rx_compl_info *rxcp)
{
        struct be_rx_stats *stats = rx_stats(rxo);

        u64_stats_update_begin(&stats->sync);
        stats->rx_compl++;
        stats->rx_bytes += rxcp->pkt_size;
        stats->rx_pkts++;
        if (rxcp->pkt_type == BE_MULTICAST_PACKET)
                stats->rx_mcast_pkts++;
        if (rxcp->err)
                stats->rx_compl_err++;
        u64_stats_update_end(&stats->sync);
}

static inline bool csum_passed(struct be_rx_compl_info *rxcp)
{
        /* L4 checksum is not reliable for non TCP/UDP packets.
         * Also ignore ipcksm for ipv6 pkts */
        return (rxcp->tcpf || rxcp->udpf) && rxcp->l4_csum &&
                                (rxcp->ip_csum || rxcp->ipv6);
}

static struct be_rx_page_info *get_rx_page_info(struct be_rx_obj *rxo,
                                                u16 frag_idx)
{
        struct be_adapter *adapter = rxo->adapter;
        struct be_rx_page_info *rx_page_info;
        struct be_queue_info *rxq = &rxo->q;

        rx_page_info = &rxo->page_info_tbl[frag_idx];
        BUG_ON(!rx_page_info->page);

        if (rx_page_info->last_page_user) {
                dma_unmap_page(&adapter->pdev->dev,
                               dma_unmap_addr(rx_page_info, bus),
                               adapter->big_page_size, DMA_FROM_DEVICE);
                rx_page_info->last_page_user = false;
        }

        atomic_dec(&rxq->used);
        return rx_page_info;
}

/* Throwaway the data in the Rx completion */
static void be_rx_compl_discard(struct be_rx_obj *rxo,
                                struct be_rx_compl_info *rxcp)
{
        struct be_queue_info *rxq = &rxo->q;
        struct be_rx_page_info *page_info;
        u16 i, num_rcvd = rxcp->num_rcvd;

        for (i = 0; i < num_rcvd; i++) {
                page_info = get_rx_page_info(rxo, rxcp->rxq_idx);
                put_page(page_info->page);
                memset(page_info, 0, sizeof(*page_info));
                index_inc(&rxcp->rxq_idx, rxq->len);
        }
}

/*
 * skb_fill_rx_data forms a complete skb for an ether frame
 * indicated by rxcp.
 */
static void skb_fill_rx_data(struct be_rx_obj *rxo, struct sk_buff *skb,
                             struct be_rx_compl_info *rxcp)
{
        struct be_queue_info *rxq = &rxo->q;
        struct be_rx_page_info *page_info;
        u16 i, j;
        u16 hdr_len, curr_frag_len, remaining;
        u8 *start;

        page_info = get_rx_page_info(rxo, rxcp->rxq_idx);
        start = page_address(page_info->page) + page_info->page_offset;
        prefetch(start);

        /* Copy data in the first descriptor of this completion */
        curr_frag_len = min(rxcp->pkt_size, rx_frag_size);

        /* Copy the header portion into skb_data */
        hdr_len = min(BE_HDR_LEN, curr_frag_len);
        memcpy(skb->data, start, hdr_len);
        skb->len = curr_frag_len;
        if (curr_frag_len <= BE_HDR_LEN) { /* tiny packet */
                /* Complete packet has now been moved to data */
                put_page(page_info->page);
                skb->data_len = 0;
                skb->tail += curr_frag_len;
        } else {
                skb_shinfo(skb)->nr_frags = 1;
                skb_frag_set_page(skb, 0, page_info->page);
                skb_shinfo(skb)->frags[0].page_offset =
                                        page_info->page_offset + hdr_len;
                skb_frag_size_set(&skb_shinfo(skb)->frags[0], curr_frag_len - hdr_len);
                skb->data_len = curr_frag_len - hdr_len;
                skb->truesize += rx_frag_size;
                skb->tail += hdr_len;
        }
        page_info->page = NULL;

        if (rxcp->pkt_size <= rx_frag_size) {
                BUG_ON(rxcp->num_rcvd != 1);
                return;
        }

        /* More frags present for this completion */
        index_inc(&rxcp->rxq_idx, rxq->len);
        remaining = rxcp->pkt_size - curr_frag_len;
        for (i = 1, j = 0; i < rxcp->num_rcvd; i++) {
                page_info = get_rx_page_info(rxo, rxcp->rxq_idx);
                curr_frag_len = min(remaining, rx_frag_size);

                /* Coalesce all frags from the same physical page in one slot */
                if (page_info->page_offset == 0) {
                        /* Fresh page */
                        j++;
                        skb_frag_set_page(skb, j, page_info->page);
                        skb_shinfo(skb)->frags[j].page_offset =
                                                        page_info->page_offset;
                        skb_frag_size_set(&skb_shinfo(skb)->frags[j], 0);
                        skb_shinfo(skb)->nr_frags++;
                } else {
                        put_page(page_info->page);
                }

                skb_frag_size_add(&skb_shinfo(skb)->frags[j], curr_frag_len);
                skb->len += curr_frag_len;
                skb->data_len += curr_frag_len;
                skb->truesize += rx_frag_size;
                remaining -= curr_frag_len;
                index_inc(&rxcp->rxq_idx, rxq->len);
                page_info->page = NULL;
        }
        BUG_ON(j > MAX_SKB_FRAGS);
}

/* Process the RX completion indicated by rxcp when GRO is disabled */
static void be_rx_compl_process(struct be_rx_obj *rxo,
                                struct be_rx_compl_info *rxcp)
{
        struct be_adapter *adapter = rxo->adapter;
        struct net_device *netdev = adapter->netdev;
        struct sk_buff *skb;

        skb = netdev_alloc_skb_ip_align(netdev, BE_RX_SKB_ALLOC_SIZE);
        if (unlikely(!skb)) {
                rx_stats(rxo)->rx_drops_no_skbs++;
                be_rx_compl_discard(rxo, rxcp);
                return;
        }

        skb_fill_rx_data(rxo, skb, rxcp);

        if (likely((netdev->features & NETIF_F_RXCSUM) && csum_passed(rxcp)))
                skb->ip_summed = CHECKSUM_UNNECESSARY;
        else
                skb_checksum_none_assert(skb);

        skb->protocol = eth_type_trans(skb, netdev);
        if (netdev->features & NETIF_F_RXHASH)
                skb->rxhash = rxcp->rss_hash;


        if (rxcp->vlanf)
                __vlan_hwaccel_put_tag(skb, rxcp->vlan_tag);

        netif_receive_skb(skb);
}

/* Process the RX completion indicated by rxcp when GRO is enabled */
void be_rx_compl_process_gro(struct be_rx_obj *rxo, struct napi_struct *napi,
                             struct be_rx_compl_info *rxcp)
{
        struct be_adapter *adapter = rxo->adapter;
        struct be_rx_page_info *page_info;
        struct sk_buff *skb = NULL;
        struct be_queue_info *rxq = &rxo->q;
        u16 remaining, curr_frag_len;
        u16 i, j;

        skb = napi_get_frags(napi);
        if (!skb) {
                be_rx_compl_discard(rxo, rxcp);
                return;
        }

        remaining = rxcp->pkt_size;
        for (i = 0, j = -1; i < rxcp->num_rcvd; i++) {
                page_info = get_rx_page_info(rxo, rxcp->rxq_idx);

                curr_frag_len = min(remaining, rx_frag_size);

                /* Coalesce all frags from the same physical page in one slot */
                if (i == 0 || page_info->page_offset == 0) {
                        /* First frag or Fresh page */
                        j++;
                        skb_frag_set_page(skb, j, page_info->page);
                        skb_shinfo(skb)->frags[j].page_offset =
                                                        page_info->page_offset;
                        skb_frag_size_set(&skb_shinfo(skb)->frags[j], 0);
                } else {
                        put_page(page_info->page);
                }
                skb_frag_size_add(&skb_shinfo(skb)->frags[j], curr_frag_len);
                skb->truesize += rx_frag_size;
                remaining -= curr_frag_len;
                index_inc(&rxcp->rxq_idx, rxq->len);
                memset(page_info, 0, sizeof(*page_info));
        }
        BUG_ON(j > MAX_SKB_FRAGS);

        skb_shinfo(skb)->nr_frags = j + 1;
        skb->len = rxcp->pkt_size;
        skb->data_len = rxcp->pkt_size;
        skb->ip_summed = CHECKSUM_UNNECESSARY;
        if (adapter->netdev->features & NETIF_F_RXHASH)
                skb->rxhash = rxcp->rss_hash;

        if (rxcp->vlanf)
                __vlan_hwaccel_put_tag(skb, rxcp->vlan_tag);

        napi_gro_frags(napi);
}

static void be_parse_rx_compl_v1(struct be_eth_rx_compl *compl,
                                 struct be_rx_compl_info *rxcp)
{
        rxcp->pkt_size =
                AMAP_GET_BITS(struct amap_eth_rx_compl_v1, pktsize, compl);
        rxcp->vlanf = AMAP_GET_BITS(struct amap_eth_rx_compl_v1, vtp, compl);
        rxcp->err = AMAP_GET_BITS(struct amap_eth_rx_compl_v1, err, compl);
        rxcp->tcpf = AMAP_GET_BITS(struct amap_eth_rx_compl_v1, tcpf, compl);
        rxcp->udpf = AMAP_GET_BITS(struct amap_eth_rx_compl_v1, udpf, compl);
        rxcp->ip_csum =
                AMAP_GET_BITS(struct amap_eth_rx_compl_v1, ipcksm, compl);
        rxcp->l4_csum =
                AMAP_GET_BITS(struct amap_eth_rx_compl_v1, l4_cksm, compl);
        rxcp->ipv6 =
                AMAP_GET_BITS(struct amap_eth_rx_compl_v1, ip_version, compl);
        rxcp->rxq_idx =
                AMAP_GET_BITS(struct amap_eth_rx_compl_v1, fragndx, compl);
        rxcp->num_rcvd =
                AMAP_GET_BITS(struct amap_eth_rx_compl_v1, numfrags, compl);
        rxcp->pkt_type =
                AMAP_GET_BITS(struct amap_eth_rx_compl_v1, cast_enc, compl);
        rxcp->rss_hash =
                AMAP_GET_BITS(struct amap_eth_rx_compl_v1, rsshash, rxcp);
        if (rxcp->vlanf) {
                rxcp->vtm = AMAP_GET_BITS(struct amap_eth_rx_compl_v1, vtm,
                                          compl);
                rxcp->vlan_tag = AMAP_GET_BITS(struct amap_eth_rx_compl_v1, vlan_tag,
                                               compl);
        }
        rxcp->port = AMAP_GET_BITS(struct amap_eth_rx_compl_v1, port, compl);
}

static void be_parse_rx_compl_v0(struct be_eth_rx_compl *compl,
                                 struct be_rx_compl_info *rxcp)
{
        rxcp->pkt_size =
                AMAP_GET_BITS(struct amap_eth_rx_compl_v0, pktsize, compl);
        rxcp->vlanf = AMAP_GET_BITS(struct amap_eth_rx_compl_v0, vtp, compl);
        rxcp->err = AMAP_GET_BITS(struct amap_eth_rx_compl_v0, err, compl);
        rxcp->tcpf = AMAP_GET_BITS(struct amap_eth_rx_compl_v0, tcpf, compl);
        rxcp->udpf = AMAP_GET_BITS(struct amap_eth_rx_compl_v0, udpf, compl);
        rxcp->ip_csum =
                AMAP_GET_BITS(struct amap_eth_rx_compl_v0, ipcksm, compl);
        rxcp->l4_csum =
                AMAP_GET_BITS(struct amap_eth_rx_compl_v0, l4_cksm, compl);
        rxcp->ipv6 =
                AMAP_GET_BITS(struct amap_eth_rx_compl_v0, ip_version, compl);
        rxcp->rxq_idx =
                AMAP_GET_BITS(struct amap_eth_rx_compl_v0, fragndx, compl);
        rxcp->num_rcvd =
                AMAP_GET_BITS(struct amap_eth_rx_compl_v0, numfrags, compl);
        rxcp->pkt_type =
                AMAP_GET_BITS(struct amap_eth_rx_compl_v0, cast_enc, compl);
        rxcp->rss_hash =
                AMAP_GET_BITS(struct amap_eth_rx_compl_v0, rsshash, rxcp);
        if (rxcp->vlanf) {
                rxcp->vtm = AMAP_GET_BITS(struct amap_eth_rx_compl_v0, vtm,
                                          compl);
                rxcp->vlan_tag = AMAP_GET_BITS(struct amap_eth_rx_compl_v0, vlan_tag,
                                               compl);
        }
        rxcp->port = AMAP_GET_BITS(struct amap_eth_rx_compl_v0, port, compl);
}

static struct be_rx_compl_info *be_rx_compl_get(struct be_rx_obj *rxo)
{
        struct be_eth_rx_compl *compl = queue_tail_node(&rxo->cq);
        struct be_rx_compl_info *rxcp = &rxo->rxcp;
        struct be_adapter *adapter = rxo->adapter;

        /* For checking the valid bit it is Ok to use either definition as the
         * valid bit is at the same position in both v0 and v1 Rx compl */
        if (compl->dw[offsetof(struct amap_eth_rx_compl_v1, valid) / 32] == 0)
                return NULL;

        rmb();
        be_dws_le_to_cpu(compl, sizeof(*compl));

        if (adapter->be3_native)
                be_parse_rx_compl_v1(compl, rxcp);
        else
                be_parse_rx_compl_v0(compl, rxcp);

        if (rxcp->vlanf) {
                /* vlanf could be wrongly set in some cards.
                 * ignore if vtm is not set */
                if ((adapter->function_mode & FLEX10_MODE) && !rxcp->vtm)
                        rxcp->vlanf = 0;

                if (!lancer_chip(adapter))
                        rxcp->vlan_tag = swab16(rxcp->vlan_tag);

                if (adapter->pvid == (rxcp->vlan_tag & VLAN_VID_MASK) &&
                    !adapter->vlan_tag[rxcp->vlan_tag])
                        rxcp->vlanf = 0;
        }

        /* As the compl has been parsed, reset it; we wont touch it again */
        compl->dw[offsetof(struct amap_eth_rx_compl_v1, valid) / 32] = 0;

        queue_tail_inc(&rxo->cq);
        return rxcp;
}

static inline struct page *be_alloc_pages(u32 size, gfp_t gfp)
{
        u32 order = get_order(size);

        if (order > 0)
                gfp |= __GFP_COMP;
        return  alloc_pages(gfp, order);
}

/*
 * Allocate a page, split it to fragments of size rx_frag_size and post as
 * receive buffers to BE
 */
static void be_post_rx_frags(struct be_rx_obj *rxo, gfp_t gfp)
{
        struct be_adapter *adapter = rxo->adapter;
        struct be_rx_page_info *page_info = NULL, *prev_page_info = NULL;
        struct be_queue_info *rxq = &rxo->q;
        struct page *pagep = NULL;
        struct be_eth_rx_d *rxd;
        u64 page_dmaaddr = 0, frag_dmaaddr;
        u32 posted, page_offset = 0;

        page_info = &rxo->page_info_tbl[rxq->head];
        for (posted = 0; posted < MAX_RX_POST && !page_info->page; posted++) {
                if (!pagep) {
                        pagep = be_alloc_pages(adapter->big_page_size, gfp);
                        if (unlikely(!pagep)) {
                                rx_stats(rxo)->rx_post_fail++;
                                break;
                        }
                        page_dmaaddr = dma_map_page(&adapter->pdev->dev, pagep,
                                                    0, adapter->big_page_size,
                                                    DMA_FROM_DEVICE);
                        page_info->page_offset = 0;
                } else {
                        get_page(pagep);
                        page_info->page_offset = page_offset + rx_frag_size;
                }
                page_offset = page_info->page_offset;
                page_info->page = pagep;
                dma_unmap_addr_set(page_info, bus, page_dmaaddr);
                frag_dmaaddr = page_dmaaddr + page_info->page_offset;

                rxd = queue_head_node(rxq);
                rxd->fragpa_lo = cpu_to_le32(frag_dmaaddr & 0xFFFFFFFF);
                rxd->fragpa_hi = cpu_to_le32(upper_32_bits(frag_dmaaddr));

                /* Any space left in the current big page for another frag? */
                if ((page_offset + rx_frag_size + rx_frag_size) >
                                        adapter->big_page_size) {
                        pagep = NULL;
                        page_info->last_page_user = true;
                }

                prev_page_info = page_info;
                queue_head_inc(rxq);
                page_info = &rxo->page_info_tbl[rxq->head];
        }
        if (pagep)
                prev_page_info->last_page_user = true;

        if (posted) {
                atomic_add(posted, &rxq->used);
                be_rxq_notify(adapter, rxq->id, posted);
        } else if (atomic_read(&rxq->used) == 0) {
                /* Let be_worker replenish when memory is available */
                rxo->rx_post_starved = true;
        }
}

static struct be_eth_tx_compl *be_tx_compl_get(struct be_queue_info *tx_cq)
{
        struct be_eth_tx_compl *txcp = queue_tail_node(tx_cq);

        if (txcp->dw[offsetof(struct amap_eth_tx_compl, valid) / 32] == 0)
                return NULL;

        rmb();
        be_dws_le_to_cpu(txcp, sizeof(*txcp));

        txcp->dw[offsetof(struct amap_eth_tx_compl, valid) / 32] = 0;

        queue_tail_inc(tx_cq);
        return txcp;
}

static u16 be_tx_compl_process(struct be_adapter *adapter,
                struct be_tx_obj *txo, u16 last_index)
{
        struct be_queue_info *txq = &txo->q;
        struct be_eth_wrb *wrb;
        struct sk_buff **sent_skbs = txo->sent_skb_list;
        struct sk_buff *sent_skb;
        u16 cur_index, num_wrbs = 1; /* account for hdr wrb */
        bool unmap_skb_hdr = true;

        sent_skb = sent_skbs[txq->tail];
        BUG_ON(!sent_skb);
        sent_skbs[txq->tail] = NULL;

        /* skip header wrb */
        queue_tail_inc(txq);

        do {
                cur_index = txq->tail;
                wrb = queue_tail_node(txq);
                unmap_tx_frag(&adapter->pdev->dev, wrb,
                              (unmap_skb_hdr && skb_headlen(sent_skb)));
                unmap_skb_hdr = false;

                num_wrbs++;
                queue_tail_inc(txq);
        } while (cur_index != last_index);

        kfree_skb(sent_skb);
        return num_wrbs;
}

/* Return the number of events in the event queue */
static inline int events_get(struct be_eq_obj *eqo)
{
        struct be_eq_entry *eqe;
        int num = 0;

        do {
                eqe = queue_tail_node(&eqo->q);
                if (eqe->evt == 0)
                        break;

                rmb();
                eqe->evt = 0;
                num++;
                queue_tail_inc(&eqo->q);
        } while (true);

        return num;
}

static int event_handle(struct be_eq_obj *eqo)
{
        bool rearm = false;
        int num = events_get(eqo);

        /* Deal with any spurious interrupts that come without events */
        if (!num)
                rearm = true;

        be_eq_notify(eqo->adapter, eqo->q.id, rearm, true, num);
        if (num)
                napi_schedule(&eqo->napi);

        return num;
}

/* Leaves the EQ is disarmed state */
static void be_eq_clean(struct be_eq_obj *eqo)
{
        int num = events_get(eqo);

        be_eq_notify(eqo->adapter, eqo->q.id, false, true, num);
}

static void be_rx_cq_clean(struct be_rx_obj *rxo)
{
        struct be_rx_page_info *page_info;
        struct be_queue_info *rxq = &rxo->q;
        struct be_queue_info *rx_cq = &rxo->cq;
        struct be_rx_compl_info *rxcp;
        u16 tail;

        /* First cleanup pending rx completions */
        while ((rxcp = be_rx_compl_get(rxo)) != NULL) {
                be_rx_compl_discard(rxo, rxcp);
                be_cq_notify(rxo->adapter, rx_cq->id, false, 1);
        }

        /* Then free posted rx buffer that were not used */
        tail = (rxq->head + rxq->len - atomic_read(&rxq->used)) % rxq->len;
        for (; atomic_read(&rxq->used) > 0; index_inc(&tail, rxq->len)) {
                page_info = get_rx_page_info(rxo, tail);
                put_page(page_info->page);
                memset(page_info, 0, sizeof(*page_info));
        }
        BUG_ON(atomic_read(&rxq->used));
        rxq->tail = rxq->head = 0;
}

static void be_tx_compl_clean(struct be_adapter *adapter)
{
        struct be_tx_obj *txo;
        struct be_queue_info *txq;
        struct be_eth_tx_compl *txcp;
        u16 end_idx, cmpl = 0, timeo = 0, num_wrbs = 0;
        struct sk_buff *sent_skb;
        bool dummy_wrb;
        int i, pending_txqs;

        /* Wait for a max of 200ms for all the tx-completions to arrive. */
        do {
                pending_txqs = adapter->num_tx_qs;

                for_all_tx_queues(adapter, txo, i) {
                        txq = &txo->q;
                        while ((txcp = be_tx_compl_get(&txo->cq))) {
                                end_idx =
                                        AMAP_GET_BITS(struct amap_eth_tx_compl,
                                                      wrb_index, txcp);
                                num_wrbs += be_tx_compl_process(adapter, txo,
                                                                end_idx);
                                cmpl++;
                        }
                        if (cmpl) {
                                be_cq_notify(adapter, txo->cq.id, false, cmpl);
                                atomic_sub(num_wrbs, &txq->used);
                                cmpl = 0;
                                num_wrbs = 0;
                        }
                        if (atomic_read(&txq->used) == 0)
                                pending_txqs--;
                }

                if (pending_txqs == 0 || ++timeo > 200)
                        break;

                mdelay(1);
        } while (true);

        for_all_tx_queues(adapter, txo, i) {
                txq = &txo->q;
                if (atomic_read(&txq->used))
                        dev_err(&adapter->pdev->dev, "%d pending tx-compls\n",
                                atomic_read(&txq->used));

                /* free posted tx for which compls will never arrive */
                while (atomic_read(&txq->used)) {
                        sent_skb = txo->sent_skb_list[txq->tail];
                        end_idx = txq->tail;
                        num_wrbs = wrb_cnt_for_skb(adapter, sent_skb,
                                                   &dummy_wrb);
                        index_adv(&end_idx, num_wrbs - 1, txq->len);
                        num_wrbs = be_tx_compl_process(adapter, txo, end_idx);
                        atomic_sub(num_wrbs, &txq->used);
                }
        }
}

static void be_evt_queues_destroy(struct be_adapter *adapter)
{
        struct be_eq_obj *eqo;
        int i;

        for_all_evt_queues(adapter, eqo, i) {
                be_eq_clean(eqo);
                if (eqo->q.created)
                        be_cmd_q_destroy(adapter, &eqo->q, QTYPE_EQ);
                be_queue_free(adapter, &eqo->q);
        }
}

static int be_evt_queues_create(struct be_adapter *adapter)
{
        struct be_queue_info *eq;
        struct be_eq_obj *eqo;
        int i, rc;

        adapter->num_evt_qs = num_irqs(adapter);

        for_all_evt_queues(adapter, eqo, i) {
                eqo->adapter = adapter;
                eqo->tx_budget = BE_TX_BUDGET;
                eqo->idx = i;
                eqo->max_eqd = BE_MAX_EQD;
                eqo->enable_aic = true;

                eq = &eqo->q;
                rc = be_queue_alloc(adapter, eq, EVNT_Q_LEN,
                                        sizeof(struct be_eq_entry));
                if (rc)
                        return rc;

                rc = be_cmd_eq_create(adapter, eq, eqo->cur_eqd);
                if (rc)
                        return rc;
        }
        return 0;
}

static void be_mcc_queues_destroy(struct be_adapter *adapter)
{
        struct be_queue_info *q;

        q = &adapter->mcc_obj.q;
        if (q->created)
                be_cmd_q_destroy(adapter, q, QTYPE_MCCQ);
        be_queue_free(adapter, q);

        q = &adapter->mcc_obj.cq;
        if (q->created)
                be_cmd_q_destroy(adapter, q, QTYPE_CQ);
        be_queue_free(adapter, q);
}

/* Must be called only after TX qs are created as MCC shares TX EQ */
static int be_mcc_queues_create(struct be_adapter *adapter)
{
        struct be_queue_info *q, *cq;

        cq = &adapter->mcc_obj.cq;
        if (be_queue_alloc(adapter, cq, MCC_CQ_LEN,
                        sizeof(struct be_mcc_compl)))
                goto err;

        /* Use the default EQ for MCC completions */
        if (be_cmd_cq_create(adapter, cq, &mcc_eqo(adapter)->q, true, 0))
                goto mcc_cq_free;

        q = &adapter->mcc_obj.q;
        if (be_queue_alloc(adapter, q, MCC_Q_LEN, sizeof(struct be_mcc_wrb)))
                goto mcc_cq_destroy;

        if (be_cmd_mccq_create(adapter, q, cq))
                goto mcc_q_free;

        return 0;

mcc_q_free:
        be_queue_free(adapter, q);
mcc_cq_destroy:
        be_cmd_q_destroy(adapter, cq, QTYPE_CQ);
mcc_cq_free:
        be_queue_free(adapter, cq);
err:
        return -1;
}

static void be_tx_queues_destroy(struct be_adapter *adapter)
{
        struct be_queue_info *q;
        struct be_tx_obj *txo;
        u8 i;

        for_all_tx_queues(adapter, txo, i) {
                q = &txo->q;
                if (q->created)
                        be_cmd_q_destroy(adapter, q, QTYPE_TXQ);
                be_queue_free(adapter, q);

                q = &txo->cq;
                if (q->created)
                        be_cmd_q_destroy(adapter, q, QTYPE_CQ);
                be_queue_free(adapter, q);
        }
}

static int be_num_txqs_want(struct be_adapter *adapter)
{
        if (sriov_enabled(adapter) || be_is_mc(adapter) ||
                lancer_chip(adapter) || !be_physfn(adapter) ||
                adapter->generation == BE_GEN2)
                return 1;
        else
                return MAX_TX_QS;
}

static int be_tx_cqs_create(struct be_adapter *adapter)
{
        struct be_queue_info *cq, *eq;
        int status;
        struct be_tx_obj *txo;
        u8 i;

        adapter->num_tx_qs = be_num_txqs_want(adapter);
        if (adapter->num_tx_qs != MAX_TX_QS) {
                rtnl_lock();
                netif_set_real_num_tx_queues(adapter->netdev,
                        adapter->num_tx_qs);
                rtnl_unlock();
        }

        for_all_tx_queues(adapter, txo, i) {
                cq = &txo->cq;
                status = be_queue_alloc(adapter, cq, TX_CQ_LEN,
                                        sizeof(struct be_eth_tx_compl));
                if (status)
                        return status;

                /* If num_evt_qs is less than num_tx_qs, then more than
                 * one txq share an eq
                 */
                eq = &adapter->eq_obj[i % adapter->num_evt_qs].q;
                status = be_cmd_cq_create(adapter, cq, eq, false, 3);
                if (status)
                        return status;
        }
        return 0;
}

static int be_tx_qs_create(struct be_adapter *adapter)
{
        struct be_tx_obj *txo;
        int i, status;

        for_all_tx_queues(adapter, txo, i) {
                status = be_queue_alloc(adapter, &txo->q, TX_Q_LEN,
                                        sizeof(struct be_eth_wrb));
                if (status)
                        return status;

                status = be_cmd_txq_create(adapter, &txo->q, &txo->cq);
                if (status)
                        return status;
        }

        return 0;
}

static void be_rx_cqs_destroy(struct be_adapter *adapter)
{
        struct be_queue_info *q;
        struct be_rx_obj *rxo;
        int i;

        for_all_rx_queues(adapter, rxo, i) {
                q = &rxo->cq;
                if (q->created)
                        be_cmd_q_destroy(adapter, q, QTYPE_CQ);
                be_queue_free(adapter, q);
        }
}

static int be_rx_cqs_create(struct be_adapter *adapter)
{
        struct be_queue_info *eq, *cq;
        struct be_rx_obj *rxo;
        int rc, i;

        /* We'll create as many RSS rings as there are irqs.
         * But when there's only one irq there's no use creating RSS rings
         */
        adapter->num_rx_qs = (num_irqs(adapter) > 1) ?
                                num_irqs(adapter) + 1 : 1;

        adapter->big_page_size = (1 << get_order(rx_frag_size)) * PAGE_SIZE;
        for_all_rx_queues(adapter, rxo, i) {
                rxo->adapter = adapter;
                cq = &rxo->cq;
                rc = be_queue_alloc(adapter, cq, RX_CQ_LEN,
                                sizeof(struct be_eth_rx_compl));
                if (rc)
                        return rc;

                eq = &adapter->eq_obj[i % adapter->num_evt_qs].q;
                rc = be_cmd_cq_create(adapter, cq, eq, false, 3);
                if (rc)
                        return rc;
        }

        if (adapter->num_rx_qs != MAX_RX_QS)
                dev_info(&adapter->pdev->dev,
                        "Created only %d receive queues", adapter->num_rx_qs);

        return 0;
}

static irqreturn_t be_intx(int irq, void *dev)
{
        struct be_adapter *adapter = dev;
        int num_evts;

        /* With INTx only one EQ is used */
        num_evts = event_handle(&adapter->eq_obj[0]);
        if (num_evts)
                return IRQ_HANDLED;
        else
                return IRQ_NONE;
}

static irqreturn_t be_msix(int irq, void *dev)
{
        struct be_eq_obj *eqo = dev;

        event_handle(eqo);
        return IRQ_HANDLED;
}

static inline bool do_gro(struct be_rx_compl_info *rxcp)
{
        return (rxcp->tcpf && !rxcp->err) ? true : false;
}

static int be_process_rx(struct be_rx_obj *rxo, struct napi_struct *napi,
                        int budget)
{
        struct be_adapter *adapter = rxo->adapter;
        struct be_queue_info *rx_cq = &rxo->cq;
        struct be_rx_compl_info *rxcp;
        u32 work_done;

        for (work_done = 0; work_done < budget; work_done++) {
                rxcp = be_rx_compl_get(rxo);
                if (!rxcp)
                        break;

                /* Is it a flush compl that has no data */
                if (unlikely(rxcp->num_rcvd == 0))
                        goto loop_continue;

                /* Discard compl with partial DMA Lancer B0 */
                if (unlikely(!rxcp->pkt_size)) {
                        be_rx_compl_discard(rxo, rxcp);
                        goto loop_continue;
                }

                /* On BE drop pkts that arrive due to imperfect filtering in
                 * promiscuous mode on some skews
                 */
                if (unlikely(rxcp->port != adapter->port_num &&
                                !lancer_chip(adapter))) {
                        be_rx_compl_discard(rxo, rxcp);
                        goto loop_continue;
                }

                if (do_gro(rxcp))
                        be_rx_compl_process_gro(rxo, napi, rxcp);
                else
                        be_rx_compl_process(rxo, rxcp);
loop_continue:
                be_rx_stats_update(rxo, rxcp);
        }

        if (work_done) {
                be_cq_notify(adapter, rx_cq->id, true, work_done);

                if (atomic_read(&rxo->q.used) < RX_FRAGS_REFILL_WM)
                        be_post_rx_frags(rxo, GFP_ATOMIC);
        }

        return work_done;
}

static bool be_process_tx(struct be_adapter *adapter, struct be_tx_obj *txo,
                          int budget, int idx)
{
        struct be_eth_tx_compl *txcp;
        int num_wrbs = 0, work_done;

        for (work_done = 0; work_done < budget; work_done++) {
                txcp = be_tx_compl_get(&txo->cq);
                if (!txcp)
                        break;
                num_wrbs += be_tx_compl_process(adapter, txo,
                                AMAP_GET_BITS(struct amap_eth_tx_compl,
                                        wrb_index, txcp));
        }

        if (work_done) {
                be_cq_notify(adapter, txo->cq.id, true, work_done);
                atomic_sub(num_wrbs, &txo->q.used);

                /* As Tx wrbs have been freed up, wake up netdev queue
                 * if it was stopped due to lack of tx wrbs.  */
                if (__netif_subqueue_stopped(adapter->netdev, idx) &&
                        atomic_read(&txo->q.used) < txo->q.len / 2) {
                        netif_wake_subqueue(adapter->netdev, idx);
                }

                u64_stats_update_begin(&tx_stats(txo)->sync_compl);
                tx_stats(txo)->tx_compl += work_done;
                u64_stats_update_end(&tx_stats(txo)->sync_compl);
        }
        return (work_done < budget); /* Done */
}

int be_poll(struct napi_struct *napi, int budget)
{
        struct be_eq_obj *eqo = container_of(napi, struct be_eq_obj, napi);
        struct be_adapter *adapter = eqo->adapter;
        int max_work = 0, work, i;
        bool tx_done;

        /* Process all TXQs serviced by this EQ */
        for (i = eqo->idx; i < adapter->num_tx_qs; i += adapter->num_evt_qs) {
                tx_done = be_process_tx(adapter, &adapter->tx_obj[i],
                                        eqo->tx_budget, i);
                if (!tx_done)
                        max_work = budget;
        }

        /* This loop will iterate twice for EQ0 in which
         * completions of the last RXQ (default one) are also processed

         * For other EQs the loop iterates only once
         */
        for (i = eqo->idx; i < adapter->num_rx_qs; i += adapter->num_evt_qs) {
                work = be_process_rx(&adapter->rx_obj[i], napi, budget);
                max_work = max(work, max_work);
        }

        if (is_mcc_eqo(eqo))
                be_process_mcc(adapter);

        if (max_work < budget) {
                napi_complete(napi);
                be_eq_notify(adapter, eqo->q.id, true, false, 0);
        } else {
                /* As we'll continue in polling mode, count and clear events */
                be_eq_notify(adapter, eqo->q.id, false, false, events_get(eqo));
        }
        return max_work;
}

void be_detect_dump_ue(struct be_adapter *adapter)
{
        u32 ue_lo = 0, ue_hi = 0, ue_lo_mask = 0, ue_hi_mask = 0;
        u32 sliport_status = 0, sliport_err1 = 0, sliport_err2 = 0;
        u32 i;

        if (adapter->eeh_err || adapter->ue_detected)
                return;

        if (lancer_chip(adapter)) {
                sliport_status = ioread32(adapter->db + SLIPORT_STATUS_OFFSET);
                if (sliport_status & SLIPORT_STATUS_ERR_MASK) {
                        sliport_err1 = ioread32(adapter->db +
                                        SLIPORT_ERROR1_OFFSET);
                        sliport_err2 = ioread32(adapter->db +
                                        SLIPORT_ERROR2_OFFSET);
                }
        } else {
                pci_read_config_dword(adapter->pdev,
                                PCICFG_UE_STATUS_LOW, &ue_lo);
                pci_read_config_dword(adapter->pdev,
                                PCICFG_UE_STATUS_HIGH, &ue_hi);
                pci_read_config_dword(adapter->pdev,
                                PCICFG_UE_STATUS_LOW_MASK, &ue_lo_mask);
                pci_read_config_dword(adapter->pdev,
                                PCICFG_UE_STATUS_HI_MASK, &ue_hi_mask);

                ue_lo = (ue_lo & (~ue_lo_mask));
                ue_hi = (ue_hi & (~ue_hi_mask));
        }

        if (ue_lo || ue_hi ||
                sliport_status & SLIPORT_STATUS_ERR_MASK) {
                adapter->ue_detected = true;
                adapter->eeh_err = true;
                dev_err(&adapter->pdev->dev,
                        "Unrecoverable error in the card\n");
        }

        if (ue_lo) {
                for (i = 0; ue_lo; ue_lo >>= 1, i++) {
                        if (ue_lo & 1)
                                dev_err(&adapter->pdev->dev,
                                "UE: %s bit set\n", ue_status_low_desc[i]);
                }
        }
        if (ue_hi) {
                for (i = 0; ue_hi; ue_hi >>= 1, i++) {
                        if (ue_hi & 1)
                                dev_err(&adapter->pdev->dev,
                                "UE: %s bit set\n", ue_status_hi_desc[i]);
                }
        }

        if (sliport_status & SLIPORT_STATUS_ERR_MASK) {
                dev_err(&adapter->pdev->dev,
                        "sliport status 0x%x\n", sliport_status);
                dev_err(&adapter->pdev->dev,
                        "sliport error1 0x%x\n", sliport_err1);
                dev_err(&adapter->pdev->dev,
                        "sliport error2 0x%x\n", sliport_err2);
        }
}

static void be_msix_disable(struct be_adapter *adapter)
{
        if (msix_enabled(adapter)) {
                pci_disable_msix(adapter->pdev);
                adapter->num_msix_vec = 0;
        }
}

static uint be_num_rss_want(struct be_adapter *adapter)
{
        if ((adapter->function_caps & BE_FUNCTION_CAPS_RSS) &&
             adapter->num_vfs == 0 && be_physfn(adapter) &&
             !be_is_mc(adapter))
                return (adapter->be3_native) ? BE3_MAX_RSS_QS : BE2_MAX_RSS_QS;
        else
                return 0;
}

static void be_msix_enable(struct be_adapter *adapter)
{
#define BE_MIN_MSIX_VECTORS             1
        int i, status, num_vec;

        /* If RSS queues are not used, need a vec for default RX Q */
        num_vec = min(be_num_rss_want(adapter), num_online_cpus());
        num_vec = max(num_vec, BE_MIN_MSIX_VECTORS);

        for (i = 0; i < num_vec; i++)
                adapter->msix_entries[i].entry = i;

        status = pci_enable_msix(adapter->pdev, adapter->msix_entries, num_vec);
        if (status == 0) {
                goto done;
        } else if (status >= BE_MIN_MSIX_VECTORS) {
                num_vec = status;
                if (pci_enable_msix(adapter->pdev, adapter->msix_entries,
                                num_vec) == 0)
                        goto done;
        }
        return;
done:
        adapter->num_msix_vec = num_vec;
        return;
}

static int be_sriov_enable(struct be_adapter *adapter)
{
        be_check_sriov_fn_type(adapter);

#ifdef CONFIG_PCI_IOV
        if (be_physfn(adapter) && num_vfs) {
                int status, pos;
                u16 dev_vfs;

                pos = pci_find_ext_capability(adapter->pdev,
                                                PCI_EXT_CAP_ID_SRIOV);
                pci_read_config_word(adapter->pdev,
                                     pos + PCI_SRIOV_TOTAL_VF, &dev_vfs);

                adapter->num_vfs = min_t(u16, num_vfs, dev_vfs);
                if (adapter->num_vfs != num_vfs)
                        dev_info(&adapter->pdev->dev,
                                 "Device supports %d VFs and not %d\n",
                                 adapter->num_vfs, num_vfs);

                status = pci_enable_sriov(adapter->pdev, adapter->num_vfs);
                if (status)
                        adapter->num_vfs = 0;

                if (adapter->num_vfs) {
                        adapter->vf_cfg = kcalloc(num_vfs,
                                                sizeof(struct be_vf_cfg),
                                                GFP_KERNEL);
                        if (!adapter->vf_cfg)
                                return -ENOMEM;
                }
        }
#endif
        return 0;
}

static void be_sriov_disable(struct be_adapter *adapter)
{
#ifdef CONFIG_PCI_IOV
        if (sriov_enabled(adapter)) {
                pci_disable_sriov(adapter->pdev);
                kfree(adapter->vf_cfg);
                adapter->num_vfs = 0;
        }
#endif
}

static inline int be_msix_vec_get(struct be_adapter *adapter,
                                struct be_eq_obj *eqo)
{
        return adapter->msix_entries[eqo->idx].vector;
}

static int be_msix_register(struct be_adapter *adapter)
{
        struct net_device *netdev = adapter->netdev;
        struct be_eq_obj *eqo;
        int status, i, vec;

        for_all_evt_queues(adapter, eqo, i) {
                sprintf(eqo->desc, "%s-q%d", netdev->name, i);
                vec = be_msix_vec_get(adapter, eqo);
                status = request_irq(vec, be_msix, 0, eqo->desc, eqo);
                if (status)
                        goto err_msix;
        }

        return 0;
err_msix:
        for (i--, eqo = &adapter->eq_obj[i]; i >= 0; i--, eqo--)
                free_irq(be_msix_vec_get(adapter, eqo), eqo);
        dev_warn(&adapter->pdev->dev, "MSIX Request IRQ failed - err %d\n",
                status);
        be_msix_disable(adapter);
        return status;
}

static int be_irq_register(struct be_adapter *adapter)
{
        struct net_device *netdev = adapter->netdev;
        int status;

        if (msix_enabled(adapter)) {
                status = be_msix_register(adapter);
                if (status == 0)
                        goto done;
                /* INTx is not supported for VF */
                if (!be_physfn(adapter))
                        return status;
        }

        /* INTx */
        netdev->irq = adapter->pdev->irq;
        status = request_irq(netdev->irq, be_intx, IRQF_SHARED, netdev->name,
                        adapter);
        if (status) {
                dev_err(&adapter->pdev->dev,
                        "INTx request IRQ failed - err %d\n", status);
                return status;
        }
done:
        adapter->isr_registered = true;
        return 0;
}

static void be_irq_unregister(struct be_adapter *adapter)
{
        struct net_device *netdev = adapter->netdev;
        struct be_eq_obj *eqo;
        int i;

        if (!adapter->isr_registered)
                return;

        /* INTx */
        if (!msix_enabled(adapter)) {
                free_irq(netdev->irq, adapter);
                goto done;
        }

        /* MSIx */
        for_all_evt_queues(adapter, eqo, i)
                free_irq(be_msix_vec_get(adapter, eqo), eqo);

done:
        adapter->isr_registered = false;
}

static void be_rx_qs_destroy(struct be_adapter *adapter)
{
        struct be_queue_info *q;
        struct be_rx_obj *rxo;
        int i;

        for_all_rx_queues(adapter, rxo, i) {
                q = &rxo->q;
                if (q->created) {
                        be_cmd_rxq_destroy(adapter, q);
                        /* After the rxq is invalidated, wait for a grace time
                         * of 1ms for all dma to end and the flush compl to
                         * arrive
                         */
                        mdelay(1);
                        be_rx_cq_clean(rxo);
                }
                be_queue_free(adapter, q);
        }
}

static int be_close(struct net_device *netdev)
{
        struct be_adapter *adapter = netdev_priv(netdev);
        struct be_eq_obj *eqo;
        int i;

        be_async_mcc_disable(adapter);

        if (!lancer_chip(adapter))
                be_intr_set(adapter, false);

        for_all_evt_queues(adapter, eqo, i) {
                napi_disable(&eqo->napi);
                if (msix_enabled(adapter))
                        synchronize_irq(be_msix_vec_get(adapter, eqo));
                else
                        synchronize_irq(netdev->irq);
                be_eq_clean(eqo);
        }

        be_irq_unregister(adapter);

        /* Wait for all pending tx completions to arrive so that
         * all tx skbs are freed.
         */
        be_tx_compl_clean(adapter);

        be_rx_qs_destroy(adapter);
        return 0;
}

static int be_rx_qs_create(struct be_adapter *adapter)
{
        struct be_rx_obj *rxo;
        int rc, i, j;
        u8 rsstable[128];

        for_all_rx_queues(adapter, rxo, i) {
                rc = be_queue_alloc(adapter, &rxo->q, RX_Q_LEN,
                                    sizeof(struct be_eth_rx_d));
                if (rc)
                        return rc;
        }

        /* The FW would like the default RXQ to be created first */
        rxo = default_rxo(adapter);
        rc = be_cmd_rxq_create(adapter, &rxo->q, rxo->cq.id, rx_frag_size,
                               adapter->if_handle, false, &rxo->rss_id);
        if (rc)
                return rc;

        for_all_rss_queues(adapter, rxo, i) {
                rc = be_cmd_rxq_create(adapter, &rxo->q, rxo->cq.id,
                                       rx_frag_size, adapter->if_handle,
                                       true, &rxo->rss_id);
                if (rc)
                        return rc;
        }

        if (be_multi_rxq(adapter)) {
                for (j = 0; j < 128; j += adapter->num_rx_qs - 1) {
                        for_all_rss_queues(adapter, rxo, i) {
                                if ((j + i) >= 128)
                                        break;
                                rsstable[j + i] = rxo->rss_id;
                        }
                }
                rc = be_cmd_rss_config(adapter, rsstable, 128);
                if (rc)
                        return rc;
        }

        /* First time posting */
        for_all_rx_queues(adapter, rxo, i)
                be_post_rx_frags(rxo, GFP_KERNEL);
        return 0;
}

static int be_open(struct net_device *netdev)
{
        struct be_adapter *adapter = netdev_priv(netdev);
        struct be_eq_obj *eqo;
        struct be_rx_obj *rxo;
        struct be_tx_obj *txo;
        u8 link_status;
        int status, i;

        status = be_rx_qs_create(adapter);
        if (status)
                goto err;

        be_irq_register(adapter);

        if (!lancer_chip(adapter))
                be_intr_set(adapter, true);

        for_all_rx_queues(adapter, rxo, i)
                be_cq_notify(adapter, rxo->cq.id, true, 0);

        for_all_tx_queues(adapter, txo, i)
                be_cq_notify(adapter, txo->cq.id, true, 0);

        be_async_mcc_enable(adapter);

        for_all_evt_queues(adapter, eqo, i) {
                napi_enable(&eqo->napi);
                be_eq_notify(adapter, eqo->q.id, true, false, 0);
        }

        status = be_cmd_link_status_query(adapter, NULL, NULL,
                                          &link_status, 0);
        if (!status)
                be_link_status_update(adapter, link_status);

        return 0;
err:
        be_close(adapter->netdev);
        return -EIO;
}

static int be_setup_wol(struct be_adapter *adapter, bool enable)
{
        struct be_dma_mem cmd;
        int status = 0;
        u8 mac[ETH_ALEN];

        memset(mac, 0, ETH_ALEN);

        cmd.size = sizeof(struct be_cmd_req_acpi_wol_magic_config);
        cmd.va = dma_alloc_coherent(&adapter->pdev->dev, cmd.size, &cmd.dma,
                                    GFP_KERNEL);
        if (cmd.va == NULL)
                return -1;
        memset(cmd.va, 0, cmd.size);

        if (enable) {
                status = pci_write_config_dword(adapter->pdev,
                        PCICFG_PM_CONTROL_OFFSET, PCICFG_PM_CONTROL_MASK);
                if (status) {
                        dev_err(&adapter->pdev->dev,
                                "Could not enable Wake-on-lan\n");
                        dma_free_coherent(&adapter->pdev->dev, cmd.size, cmd.va,
                                          cmd.dma);
                        return status;
                }
                status = be_cmd_enable_magic_wol(adapter,
                                adapter->netdev->dev_addr, &cmd);
                pci_enable_wake(adapter->pdev, PCI_D3hot, 1);
                pci_enable_wake(adapter->pdev, PCI_D3cold, 1);
        } else {
                status = be_cmd_enable_magic_wol(adapter, mac, &cmd);
                pci_enable_wake(adapter->pdev, PCI_D3hot, 0);
                pci_enable_wake(adapter->pdev, PCI_D3cold, 0);
        }

        dma_free_coherent(&adapter->pdev->dev, cmd.size, cmd.va, cmd.dma);
        return status;
}

/*
 * Generate a seed MAC address from the PF MAC Address using jhash.
 * MAC Address for VFs are assigned incrementally starting from the seed.
 * These addresses are programmed in the ASIC by the PF and the VF driver
 * queries for the MAC address during its probe.
 */
static inline int be_vf_eth_addr_config(struct be_adapter *adapter)
{
        u32 vf;
        int status = 0;
        u8 mac[ETH_ALEN];
        struct be_vf_cfg *vf_cfg;

        be_vf_eth_addr_generate(adapter, mac);

        for_all_vfs(adapter, vf_cfg, vf) {
                if (lancer_chip(adapter)) {
                        status = be_cmd_set_mac_list(adapter,  mac, 1, vf + 1);
                } else {
                        status = be_cmd_pmac_add(adapter, mac,
                                                 vf_cfg->if_handle,
                                                 &vf_cfg->pmac_id, vf + 1);
                }

                if (status)
                        dev_err(&adapter->pdev->dev,
                        "Mac address assignment failed for VF %d\n", vf);
                else
                        memcpy(vf_cfg->mac_addr, mac, ETH_ALEN);

                mac[5] += 1;
        }
        return status;
}

static void be_vf_clear(struct be_adapter *adapter)
{
        struct be_vf_cfg *vf_cfg;
        u32 vf;

        for_all_vfs(adapter, vf_cfg, vf) {
                if (lancer_chip(adapter))
                        be_cmd_set_mac_list(adapter, NULL, 0, vf + 1);
                else
                        be_cmd_pmac_del(adapter, vf_cfg->if_handle,
                                        vf_cfg->pmac_id, vf + 1);

                be_cmd_if_destroy(adapter, vf_cfg->if_handle, vf + 1);
        }
}

static int be_clear(struct be_adapter *adapter)
{
        int i = 1;

        if (adapter->flags & BE_FLAGS_WORKER_SCHEDULED) {
                cancel_delayed_work_sync(&adapter->work);
                adapter->flags &= ~BE_FLAGS_WORKER_SCHEDULED;
        }

        if (sriov_enabled(adapter))
                be_vf_clear(adapter);

        for (; adapter->uc_macs > 0; adapter->uc_macs--, i++)
                be_cmd_pmac_del(adapter, adapter->if_handle,
                        adapter->pmac_id[i], 0);

        be_cmd_if_destroy(adapter, adapter->if_handle,  0);

        be_mcc_queues_destroy(adapter);
        be_rx_cqs_destroy(adapter);
        be_tx_queues_destroy(adapter);
        be_evt_queues_destroy(adapter);

        /* tell fw we're done with firing cmds */
        be_cmd_fw_clean(adapter);

        be_msix_disable(adapter);
        kfree(adapter->pmac_id);
        return 0;
}

static void be_vf_setup_init(struct be_adapter *adapter)
{
        struct be_vf_cfg *vf_cfg;
        int vf;

        for_all_vfs(adapter, vf_cfg, vf) {
                vf_cfg->if_handle = -1;
                vf_cfg->pmac_id = -1;
        }
}

static int be_vf_setup(struct be_adapter *adapter)
{
        struct be_vf_cfg *vf_cfg;
        u32 cap_flags, en_flags, vf;
        u16 def_vlan, lnk_speed;
        int status;

        be_vf_setup_init(adapter);

        cap_flags = en_flags = BE_IF_FLAGS_UNTAGGED | BE_IF_FLAGS_BROADCAST |
                                BE_IF_FLAGS_MULTICAST;
        for_all_vfs(adapter, vf_cfg, vf) {
                status = be_cmd_if_create(adapter, cap_flags, en_flags, NULL,
                                          &vf_cfg->if_handle, NULL, vf + 1);
                if (status)
                        goto err;
        }

        status = be_vf_eth_addr_config(adapter);
        if (status)
                goto err;

        for_all_vfs(adapter, vf_cfg, vf) {
                status = be_cmd_link_status_query(adapter, NULL, &lnk_speed,
                                                  NULL, vf + 1);
                if (status)
                        goto err;
                vf_cfg->tx_rate = lnk_speed * 10;

                status = be_cmd_get_hsw_config(adapter, &def_vlan,
                                vf + 1, vf_cfg->if_handle);
                if (status)
                        goto err;
                vf_cfg->def_vid = def_vlan;
        }
        return 0;
err:
        return status;
}

static void be_setup_init(struct be_adapter *adapter)
{
        adapter->vlan_prio_bmap = 0xff;
        adapter->link_speed = -1;
        adapter->if_handle = -1;
        adapter->be3_native = false;
        adapter->promiscuous = false;
        adapter->eq_next_idx = 0;
}

static int be_add_mac_from_list(struct be_adapter *adapter, u8 *mac)
{
        u32 pmac_id;
        int status;
        bool pmac_id_active;

        status = be_cmd_get_mac_from_list(adapter, 0, &pmac_id_active,
                                                        &pmac_id, mac);
        if (status != 0)
                goto do_none;

        if (pmac_id_active) {
                status = be_cmd_mac_addr_query(adapter, mac,
                                MAC_ADDRESS_TYPE_NETWORK,
                                false, adapter->if_handle, pmac_id);

                if (!status)
                        adapter->pmac_id[0] = pmac_id;
        } else {
                status = be_cmd_pmac_add(adapter, mac,
                                adapter->if_handle, &adapter->pmac_id[0], 0);
        }
do_none:
        return status;
}

static int be_setup(struct be_adapter *adapter)
{
        struct net_device *netdev = adapter->netdev;
        u32 cap_flags, en_flags;
        u32 tx_fc, rx_fc;
        int status;
        u8 mac[ETH_ALEN];

        be_setup_init(adapter);

        be_cmd_req_native_mode(adapter);

        be_msix_enable(adapter);

        status = be_evt_queues_create(adapter);
        if (status)
                goto err;

        status = be_tx_cqs_create(adapter);
        if (status)
                goto err;

        status = be_rx_cqs_create(adapter);
        if (status)
                goto err;

        status = be_mcc_queues_create(adapter);
        if (status)
                goto err;

        memset(mac, 0, ETH_ALEN);
        status = be_cmd_mac_addr_query(adapter, mac, MAC_ADDRESS_TYPE_NETWORK,
                        true /*permanent */, 0, 0);
        if (status)
                return status;
        memcpy(adapter->netdev->dev_addr, mac, ETH_ALEN);
        memcpy(adapter->netdev->perm_addr, mac, ETH_ALEN);

        en_flags = BE_IF_FLAGS_UNTAGGED | BE_IF_FLAGS_BROADCAST |
                        BE_IF_FLAGS_MULTICAST | BE_IF_FLAGS_PASS_L3L4_ERRORS;
        cap_flags = en_flags | BE_IF_FLAGS_MCAST_PROMISCUOUS |
                        BE_IF_FLAGS_VLAN_PROMISCUOUS | BE_IF_FLAGS_PROMISCUOUS;

        if (adapter->function_caps & BE_FUNCTION_CAPS_RSS) {
                cap_flags |= BE_IF_FLAGS_RSS;
                en_flags |= BE_IF_FLAGS_RSS;
        }
        status = be_cmd_if_create(adapter, cap_flags, en_flags,
                        netdev->dev_addr, &adapter->if_handle,
                        &adapter->pmac_id[0], 0);
        if (status != 0)
                goto err;

         /* The VF's permanent mac queried from card is incorrect.
          * For BEx: Query the mac configued by the PF using if_handle
          * For Lancer: Get and use mac_list to obtain mac address.
          */
        if (!be_physfn(adapter)) {
                if (lancer_chip(adapter))
                        status = be_add_mac_from_list(adapter, mac);
                else
                        status = be_cmd_mac_addr_query(adapter, mac,
                                        MAC_ADDRESS_TYPE_NETWORK, false,
                                        adapter->if_handle, 0);
                if (!status) {
                        memcpy(adapter->netdev->dev_addr, mac, ETH_ALEN);
                        memcpy(adapter->netdev->perm_addr, mac, ETH_ALEN);
                }
        }

        status = be_tx_qs_create(adapter);
        if (status)
                goto err;

        be_cmd_get_fw_ver(adapter, adapter->fw_ver, NULL);

        status = be_vid_config(adapter, false, 0);
        if (status)
                goto err;

        be_set_rx_mode(adapter->netdev);

        status = be_cmd_get_flow_control(adapter, &tx_fc, &rx_fc);
        /* For Lancer: It is legal for this cmd to fail on VF */
        if (status && (be_physfn(adapter) || !lancer_chip(adapter)))
                goto err;

        if (rx_fc != adapter->rx_fc || tx_fc != adapter->tx_fc) {
                status = be_cmd_set_flow_control(adapter, adapter->tx_fc,
                                        adapter->rx_fc);
                /* For Lancer: It is legal for this cmd to fail on VF */
                if (status && (be_physfn(adapter) || !lancer_chip(adapter)))
                        goto err;
        }

        pcie_set_readrq(adapter->pdev, 4096);

        if (sriov_enabled(adapter)) {
                status = be_vf_setup(adapter);
                if (status)
                        goto err;
        }

        schedule_delayed_work(&adapter->work, msecs_to_jiffies(1000));
        adapter->flags |= BE_FLAGS_WORKER_SCHEDULED;

        return 0;
err:
        be_clear(adapter);
        return status;
}

#ifdef CONFIG_NET_POLL_CONTROLLER
static void be_netpoll(struct net_device *netdev)
{
        struct be_adapter *adapter = netdev_priv(netdev);
        struct be_eq_obj *eqo;
        int i;

        for_all_evt_queues(adapter, eqo, i)
                event_handle(eqo);

        return;
}
#endif

#define FW_FILE_HDR_SIGN        "ServerEngines Corp. "
static bool be_flash_redboot(struct be_adapter *adapter,
                        const u8 *p, u32 img_start, int image_size,
                        int hdr_size)
{
        u32 crc_offset;
        u8 flashed_crc[4];
        int status;

        crc_offset = hdr_size + img_start + image_size - 4;

        p += crc_offset;

        status = be_cmd_get_flash_crc(adapter, flashed_crc,
                        (image_size - 4));
        if (status) {
                dev_err(&adapter->pdev->dev,
                "could not get crc from flash, not flashing redboot\n");
                return false;
        }

        /*update redboot only if crc does not match*/
        if (!memcmp(flashed_crc, p, 4))
                return false;
        else
                return true;
}

static bool phy_flashing_required(struct be_adapter *adapter)
{
        int status = 0;
        struct be_phy_info phy_info;

        status = be_cmd_get_phy_info(adapter, &phy_info);
        if (status)
                return false;
        if ((phy_info.phy_type == TN_8022) &&
                (phy_info.interface_type == PHY_TYPE_BASET_10GB)) {
                return true;
        }
        return false;
}

static int be_flash_data(struct be_adapter *adapter,
                        const struct firmware *fw,
                        struct be_dma_mem *flash_cmd, int num_of_images)

{
        int status = 0, i, filehdr_size = 0;
        u32 total_bytes = 0, flash_op;
        int num_bytes;
        const u8 *p = fw->data;
        struct be_cmd_write_flashrom *req = flash_cmd->va;
        const struct flash_comp *pflashcomp;
        int num_comp;

        static const struct flash_comp gen3_flash_types[10] = {
                { FLASH_iSCSI_PRIMARY_IMAGE_START_g3, IMG_TYPE_ISCSI_ACTIVE,
                        FLASH_IMAGE_MAX_SIZE_g3},
                { FLASH_REDBOOT_START_g3, IMG_TYPE_REDBOOT,
                        FLASH_REDBOOT_IMAGE_MAX_SIZE_g3},
                { FLASH_iSCSI_BIOS_START_g3, IMG_TYPE_BIOS,
                        FLASH_BIOS_IMAGE_MAX_SIZE_g3},
                { FLASH_PXE_BIOS_START_g3, IMG_TYPE_PXE_BIOS,
                        FLASH_BIOS_IMAGE_MAX_SIZE_g3},
                { FLASH_FCoE_BIOS_START_g3, IMG_TYPE_FCOE_BIOS,
                        FLASH_BIOS_IMAGE_MAX_SIZE_g3},
                { FLASH_iSCSI_BACKUP_IMAGE_START_g3, IMG_TYPE_ISCSI_BACKUP,
                        FLASH_IMAGE_MAX_SIZE_g3},
                { FLASH_FCoE_PRIMARY_IMAGE_START_g3, IMG_TYPE_FCOE_FW_ACTIVE,
                        FLASH_IMAGE_MAX_SIZE_g3},
                { FLASH_FCoE_BACKUP_IMAGE_START_g3, IMG_TYPE_FCOE_FW_BACKUP,
                        FLASH_IMAGE_MAX_SIZE_g3},
                { FLASH_NCSI_START_g3, IMG_TYPE_NCSI_FW,
                        FLASH_NCSI_IMAGE_MAX_SIZE_g3},
                { FLASH_PHY_FW_START_g3, IMG_TYPE_PHY_FW,
                        FLASH_PHY_FW_IMAGE_MAX_SIZE_g3}
        };
        static const struct flash_comp gen2_flash_types[8] = {
                { FLASH_iSCSI_PRIMARY_IMAGE_START_g2, IMG_TYPE_ISCSI_ACTIVE,
                        FLASH_IMAGE_MAX_SIZE_g2},
                { FLASH_REDBOOT_START_g2, IMG_TYPE_REDBOOT,
                        FLASH_REDBOOT_IMAGE_MAX_SIZE_g2},
                { FLASH_iSCSI_BIOS_START_g2, IMG_TYPE_BIOS,
                        FLASH_BIOS_IMAGE_MAX_SIZE_g2},
                { FLASH_PXE_BIOS_START_g2, IMG_TYPE_PXE_BIOS,
                        FLASH_BIOS_IMAGE_MAX_SIZE_g2},
                { FLASH_FCoE_BIOS_START_g2, IMG_TYPE_FCOE_BIOS,
                        FLASH_BIOS_IMAGE_MAX_SIZE_g2},
                { FLASH_iSCSI_BACKUP_IMAGE_START_g2, IMG_TYPE_ISCSI_BACKUP,
                        FLASH_IMAGE_MAX_SIZE_g2},
                { FLASH_FCoE_PRIMARY_IMAGE_START_g2, IMG_TYPE_FCOE_FW_ACTIVE,
                        FLASH_IMAGE_MAX_SIZE_g2},
                { FLASH_FCoE_BACKUP_IMAGE_START_g2, IMG_TYPE_FCOE_FW_BACKUP,
                         FLASH_IMAGE_MAX_SIZE_g2}
        };

        if (adapter->generation == BE_GEN3) {
                pflashcomp = gen3_flash_types;
                filehdr_size = sizeof(struct flash_file_hdr_g3);
                num_comp = ARRAY_SIZE(gen3_flash_types);
        } else {
                pflashcomp = gen2_flash_types;
                filehdr_size = sizeof(struct flash_file_hdr_g2);
                num_comp = ARRAY_SIZE(gen2_flash_types);
        }
        for (i = 0; i < num_comp; i++) {
                if ((pflashcomp[i].optype == IMG_TYPE_NCSI_FW) &&
                                memcmp(adapter->fw_ver, "3.102.148.0", 11) < 0)
                        continue;
                if (pflashcomp[i].optype == IMG_TYPE_PHY_FW) {
                        if (!phy_flashing_required(adapter))
                                continue;
                }
                if ((pflashcomp[i].optype == IMG_TYPE_REDBOOT) &&
                        (!be_flash_redboot(adapter, fw->data,
                        pflashcomp[i].offset, pflashcomp[i].size, filehdr_size +
                        (num_of_images * sizeof(struct image_hdr)))))
                        continue;
                p = fw->data;
                p += filehdr_size + pflashcomp[i].offset
                        + (num_of_images * sizeof(struct image_hdr));
                if (p + pflashcomp[i].size > fw->data + fw->size)
                        return -1;
                total_bytes = pflashcomp[i].size;
                while (total_bytes) {
                        if (total_bytes > 32*1024)
                                num_bytes = 32*1024;
                        else
                                num_bytes = total_bytes;
                        total_bytes -= num_bytes;
                        if (!total_bytes) {
                                if (pflashcomp[i].optype == IMG_TYPE_PHY_FW)
                                        flash_op = FLASHROM_OPER_PHY_FLASH;
                                else
                                        flash_op = FLASHROM_OPER_FLASH;
                        } else {
                                if (pflashcomp[i].optype == IMG_TYPE_PHY_FW)
                                        flash_op = FLASHROM_OPER_PHY_SAVE;
                                else
                                        flash_op = FLASHROM_OPER_SAVE;
                        }
                        memcpy(req->params.data_buf, p, num_bytes);
                        p += num_bytes;
                        status = be_cmd_write_flashrom(adapter, flash_cmd,
                                pflashcomp[i].optype, flash_op, num_bytes);
                        if (status) {
                                if ((status == ILLEGAL_IOCTL_REQ) &&
                                        (pflashcomp[i].optype ==
                                                IMG_TYPE_PHY_FW))
                                        break;
                                dev_err(&adapter->pdev->dev,
                                        "cmd to write to flash rom failed.\n");
                                return -1;
                        }
                }
        }
        return 0;
}

static int get_ufigen_type(struct flash_file_hdr_g2 *fhdr)
{
        if (fhdr == NULL)
                return 0;
        if (fhdr->build[0] == '3')
                return BE_GEN3;
        else if (fhdr->build[0] == '2')
                return BE_GEN2;
        else
                return 0;
}

static int lancer_fw_download(struct be_adapter *adapter,
                                const struct firmware *fw)
{
#define LANCER_FW_DOWNLOAD_CHUNK      (32 * 1024)
#define LANCER_FW_DOWNLOAD_LOCATION   "/prg"
        struct be_dma_mem flash_cmd;
        const u8 *data_ptr = NULL;
        u8 *dest_image_ptr = NULL;
        size_t image_size = 0;
        u32 chunk_size = 0;
        u32 data_written = 0;
        u32 offset = 0;
        int status = 0;
        u8 add_status = 0;

        if (!IS_ALIGNED(fw->size, sizeof(u32))) {
                dev_err(&adapter->pdev->dev,
                        "FW Image not properly aligned. "
                        "Length must be 4 byte aligned.\n");
                status = -EINVAL;
                goto lancer_fw_exit;
        }

        flash_cmd.size = sizeof(struct lancer_cmd_req_write_object)
                                + LANCER_FW_DOWNLOAD_CHUNK;
        flash_cmd.va = dma_alloc_coherent(&adapter->pdev->dev, flash_cmd.size,
                                                &flash_cmd.dma, GFP_KERNEL);
        if (!flash_cmd.va) {
                status = -ENOMEM;
                dev_err(&adapter->pdev->dev,
                        "Memory allocation failure while flashing\n");
                goto lancer_fw_exit;
        }

        dest_image_ptr = flash_cmd.va +
                                sizeof(struct lancer_cmd_req_write_object);
        image_size = fw->size;
        data_ptr = fw->data;

        while (image_size) {
                chunk_size = min_t(u32, image_size, LANCER_FW_DOWNLOAD_CHUNK);

                /* Copy the image chunk content. */
                memcpy(dest_image_ptr, data_ptr, chunk_size);

                status = lancer_cmd_write_object(adapter, &flash_cmd,
                                chunk_size, offset, LANCER_FW_DOWNLOAD_LOCATION,
                                &data_written, &add_status);

                if (status)
                        break;

                offset += data_written;
                data_ptr += data_written;
                image_size -= data_written;
        }

        if (!status) {
                /* Commit the FW written */
                status = lancer_cmd_write_object(adapter, &flash_cmd,
                                        0, offset, LANCER_FW_DOWNLOAD_LOCATION,
                                        &data_written, &add_status);
        }

        dma_free_coherent(&adapter->pdev->dev, flash_cmd.size, flash_cmd.va,
                                flash_cmd.dma);
        if (status) {
                dev_err(&adapter->pdev->dev,
                        "Firmware load error. "
                        "Status code: 0x%x Additional Status: 0x%x\n",
                        status, add_status);
                goto lancer_fw_exit;
        }

        dev_info(&adapter->pdev->dev, "Firmware flashed successfully\n");
lancer_fw_exit:
        return status;
}

static int be_fw_download(struct be_adapter *adapter, const struct firmware* fw)
{
        struct flash_file_hdr_g2 *fhdr;
        struct flash_file_hdr_g3 *fhdr3;
        struct image_hdr *img_hdr_ptr = NULL;
        struct be_dma_mem flash_cmd;
        const u8 *p;
        int status = 0, i = 0, num_imgs = 0;

        p = fw->data;
        fhdr = (struct flash_file_hdr_g2 *) p;

        flash_cmd.size = sizeof(struct be_cmd_write_flashrom) + 32*1024;
        flash_cmd.va = dma_alloc_coherent(&adapter->pdev->dev, flash_cmd.size,
                                          &flash_cmd.dma, GFP_KERNEL);
        if (!flash_cmd.va) {
                status = -ENOMEM;