// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2015 - 2022 Beijing WangXun Technology Co., Ltd. */

#include <linux/types.h>
#include <linux/module.h>
#include <linux/pci.h>
#include <linux/netdevice.h>
#include <linux/string.h>
#include <linux/etherdevice.h>
#include <linux/phylink.h>
#include <net/udp_tunnel.h>
#include <net/ip.h>
#include <linux/if_vlan.h>

#include "../libwx/wx_type.h"
#include "../libwx/wx_lib.h"
#include "../libwx/wx_ptp.h"
#include "../libwx/wx_hw.h"
#include "../libwx/wx_mbx.h"
#include "../libwx/wx_sriov.h"
#include "txgbe_type.h"
#include "txgbe_hw.h"
#include "txgbe_phy.h"
#include "txgbe_aml.h"
#include "txgbe_irq.h"
#include "txgbe_fdir.h"
#include "txgbe_ethtool.h"

char txgbe_driver_name[] = "txgbe";

/* txgbe_pci_tbl - PCI Device ID Table
 *
 * Wildcard entries (PCI_ANY_ID) should come last
 * Last entry must be all 0s
 *
 * { Vendor ID, Device ID, SubVendor ID, SubDevice ID,
 *   Class, Class Mask, private data (not used) }
 */
static const struct pci_device_id txgbe_pci_tbl[] = {
        { PCI_VDEVICE(WANGXUN, TXGBE_DEV_ID_SP1000), 0},
        { PCI_VDEVICE(WANGXUN, TXGBE_DEV_ID_WX1820), 0},
        { PCI_VDEVICE(WANGXUN, TXGBE_DEV_ID_AML5010), 0},
        { PCI_VDEVICE(WANGXUN, TXGBE_DEV_ID_AML5110), 0},
        { PCI_VDEVICE(WANGXUN, TXGBE_DEV_ID_AML5025), 0},
        { PCI_VDEVICE(WANGXUN, TXGBE_DEV_ID_AML5125), 0},
        { PCI_VDEVICE(WANGXUN, TXGBE_DEV_ID_AML5040), 0},
        { PCI_VDEVICE(WANGXUN, TXGBE_DEV_ID_AML5140), 0},
        /* required last entry */
        { .device = 0 }
};

#define DEFAULT_DEBUG_LEVEL_SHIFT 3

static void txgbe_check_minimum_link(struct wx *wx)
{
        struct pci_dev *pdev;

        pdev = wx->pdev;
        pcie_print_link_status(pdev);
}

/**
 * txgbe_enumerate_functions - Get the number of ports this device has
 * @wx: wx structure
 *
 * This function enumerates the phsyical functions co-located on a single slot,
 * in order to determine how many ports a device has. This is most useful in
 * determining the required GT/s of PCIe bandwidth necessary for optimal
 * performance.
 **/
static int txgbe_enumerate_functions(struct wx *wx)
{
        struct pci_dev *entry, *pdev = wx->pdev;
        int physfns = 0;

        list_for_each_entry(entry, &pdev->bus->devices, bus_list) {
                /* When the devices on the bus don't all match our device ID,
                 * we can't reliably determine the correct number of
                 * functions. This can occur if a function has been direct
                 * attached to a virtual machine using VT-d.
                 */
                if (entry->vendor != pdev->vendor ||
                    entry->device != pdev->device)
                        return -EINVAL;

                physfns++;
        }

        return physfns;
}

static void txgbe_module_detection_subtask(struct wx *wx)
{
        int err;

        if (!test_bit(WX_FLAG_NEED_MODULE_RESET, wx->flags))
                return;

        /* wait for SFF module ready */
        msleep(200);

        err = txgbe_identify_module(wx);
        if (err)
                return;

        clear_bit(WX_FLAG_NEED_MODULE_RESET, wx->flags);
}

static void txgbe_link_config_subtask(struct wx *wx)
{
        int err;

        if (!test_bit(WX_FLAG_NEED_LINK_CONFIG, wx->flags))
                return;

        err = txgbe_set_phy_link(wx);
        if (err)
                return;

        clear_bit(WX_FLAG_NEED_LINK_CONFIG, wx->flags);
}

/**
 * txgbe_service_task - manages and runs subtasks
 * @work: pointer to work_struct containing our data
 **/
static void txgbe_service_task(struct work_struct *work)
{
        struct wx *wx = container_of(work, struct wx, service_task);

        txgbe_module_detection_subtask(wx);
        txgbe_link_config_subtask(wx);

        wx_service_event_complete(wx);
}

static void txgbe_init_service(struct wx *wx)
{
        timer_setup(&wx->service_timer, wx_service_timer, 0);
        INIT_WORK(&wx->service_task, txgbe_service_task);
        clear_bit(WX_STATE_SERVICE_SCHED, wx->state);
}

static void txgbe_up_complete(struct wx *wx)
{
        struct net_device *netdev = wx->netdev;

        wx_control_hw(wx, true);
        wx_configure_vectors(wx);

        /* make sure to complete pre-operations */
        smp_mb__before_atomic();
        wx_napi_enable_all(wx);

        switch (wx->mac.type) {
        case wx_mac_aml40:
                txgbe_setup_link(wx);
                phylink_start(wx->phylink);
                break;
        case wx_mac_aml:
                /* Enable TX laser */
                wr32m(wx, WX_GPIO_DR, TXGBE_GPIOBIT_1, 0);
                txgbe_setup_link(wx);
                phylink_start(wx->phylink);
                break;
        case wx_mac_sp:
                phylink_start(wx->phylink);
                break;
        default:
                break;
        }

        /* clear any pending interrupts, may auto mask */
        rd32(wx, WX_PX_IC(0));
        rd32(wx, WX_PX_IC(1));
        rd32(wx, WX_PX_MISC_IC);
        txgbe_irq_enable(wx, true);

        /* enable transmits */
        netif_tx_start_all_queues(netdev);
        mod_timer(&wx->service_timer, jiffies);

        /* Set PF Reset Done bit so PF/VF Mail Ops can work */
        wr32m(wx, WX_CFG_PORT_CTL, WX_CFG_PORT_CTL_PFRSTD,
              WX_CFG_PORT_CTL_PFRSTD);
        /* update setting rx tx for all active vfs */
        wx_set_all_vfs(wx);
}

static void txgbe_reset(struct wx *wx)
{
        struct net_device *netdev = wx->netdev;
        u8 old_addr[ETH_ALEN];
        int err;

        err = txgbe_reset_hw(wx);
        if (err != 0)
                wx_err(wx, "Hardware Error: %d\n", err);

        wx_start_hw(wx);
        /* do not flush user set addresses */
        memcpy(old_addr, &wx->mac_table[0].addr, netdev->addr_len);
        wx_flush_sw_mac_table(wx);
        wx_mac_set_default_filter(wx, old_addr);

        if (test_bit(WX_STATE_PTP_RUNNING, wx->state))
                wx_ptp_reset(wx);
}

static void txgbe_disable_device(struct wx *wx)
{
        struct net_device *netdev = wx->netdev;
        u32 i;

        wx_disable_pcie_master(wx);
        /* disable receives */
        wx_disable_rx(wx);

        /* disable all enabled rx queues */
        for (i = 0; i < wx->num_rx_queues; i++)
                /* this call also flushes the previous write */
                wx_disable_rx_queue(wx, wx->rx_ring[i]);

        netif_tx_stop_all_queues(netdev);
        netif_tx_disable(netdev);

        wx_irq_disable(wx);
        wx_napi_disable_all(wx);

        timer_delete_sync(&wx->service_timer);

        if (wx->bus.func < 2)
                wr32m(wx, TXGBE_MIS_PRB_CTL, TXGBE_MIS_PRB_CTL_LAN_UP(wx->bus.func), 0);
        else
                wx_err(wx, "%s: invalid bus lan id %d\n",
                       __func__, wx->bus.func);

        if (wx->num_vfs) {
                /* Clear EITR Select mapping */
                wr32(wx, WX_PX_ITRSEL, 0);
                /* Mark all the VFs as inactive */
                for (i = 0; i < wx->num_vfs; i++)
                        wx->vfinfo[i].clear_to_send = 0;
                /* update setting rx tx for all active vfs */
                wx_set_all_vfs(wx);
        }

        if (!(((wx->subsystem_device_id & WX_NCSI_MASK) == WX_NCSI_SUP) ||
              ((wx->subsystem_device_id & WX_WOL_MASK) == WX_WOL_SUP))) {
                /* disable mac transmiter */
                wr32m(wx, WX_MAC_TX_CFG, WX_MAC_TX_CFG_TE, 0);
        }

        /* disable transmits in the hardware now that interrupts are off */
        for (i = 0; i < wx->num_tx_queues; i++) {
                u8 reg_idx = wx->tx_ring[i]->reg_idx;

                wr32(wx, WX_PX_TR_CFG(reg_idx), WX_PX_TR_CFG_SWFLSH);
        }

        /* Disable the Tx DMA engine */
        wr32m(wx, WX_TDM_CTL, WX_TDM_CTL_TE, 0);

        wx_update_stats(wx);
}

void txgbe_down(struct wx *wx)
{
        txgbe_disable_device(wx);
        txgbe_reset(wx);

        switch (wx->mac.type) {
        case wx_mac_aml40:
                phylink_stop(wx->phylink);
                break;
        case wx_mac_aml:
                phylink_stop(wx->phylink);
                /* Disable TX laser */
                wr32m(wx, WX_GPIO_DR, TXGBE_GPIOBIT_1, TXGBE_GPIOBIT_1);
                break;
        case wx_mac_sp:
                phylink_stop(wx->phylink);
                break;
        default:
                break;
        }

        wx_clean_all_tx_rings(wx);
        wx_clean_all_rx_rings(wx);
}

void txgbe_up(struct wx *wx)
{
        wx_configure(wx);
        wx_ptp_init(wx);
        txgbe_up_complete(wx);
}

/**
 *  txgbe_init_type_code - Initialize the shared code
 *  @wx: pointer to hardware structure
 **/
static void txgbe_init_type_code(struct wx *wx)
{
        u8 device_type = wx->subsystem_device_id & 0xF0;

        switch (wx->device_id) {
        case TXGBE_DEV_ID_SP1000:
        case TXGBE_DEV_ID_WX1820:
                wx->mac.type = wx_mac_sp;
                break;
        case TXGBE_DEV_ID_AML5010:
        case TXGBE_DEV_ID_AML5110:
        case TXGBE_DEV_ID_AML5025:
        case TXGBE_DEV_ID_AML5125:
                wx->mac.type = wx_mac_aml;
                break;
        case TXGBE_DEV_ID_AML5040:
        case TXGBE_DEV_ID_AML5140:
                wx->mac.type = wx_mac_aml40;
                break;
        default:
                wx->mac.type = wx_mac_unknown;
                break;
        }

        switch (device_type) {
        case TXGBE_ID_SFP:
                wx->media_type = wx_media_fiber;
                break;
        case TXGBE_ID_XAUI:
        case TXGBE_ID_SGMII:
                wx->media_type = wx_media_copper;
                break;
        case TXGBE_ID_KR_KX_KX4:
        case TXGBE_ID_MAC_XAUI:
        case TXGBE_ID_MAC_SGMII:
                wx->media_type = wx_media_backplane;
                break;
        case TXGBE_ID_SFI_XAUI:
                if (wx->bus.func == 0)
                        wx->media_type = wx_media_fiber;
                else
                        wx->media_type = wx_media_copper;
                break;
        default:
                wx->media_type = wx_media_unknown;
                break;
        }
}

/**
 * txgbe_sw_init - Initialize general software structures (struct wx)
 * @wx: board private structure to initialize
 **/
static int txgbe_sw_init(struct wx *wx)
{
        u16 msix_count = 0;
        int err;

        wx->mac.num_rar_entries = TXGBE_RAR_ENTRIES;
        wx->mac.max_tx_queues = TXGBE_MAX_TXQ;
        wx->mac.max_rx_queues = TXGBE_MAX_RXQ;
        wx->mac.mcft_size = TXGBE_MC_TBL_SIZE;
        wx->mac.vft_size = TXGBE_VFT_TBL_SIZE;
        wx->mac.rx_pb_size = TXGBE_RX_PB_SIZE;
        wx->mac.tx_pb_size = TXGBE_TDB_PB_SZ;

        /* PCI config space info */
        err = wx_sw_init(wx);
        if (err < 0)
                return err;

        txgbe_init_type_code(wx);

        /* Set common capability flags and settings */
        wx->max_q_vectors = TXGBE_MAX_MSIX_VECTORS;
        err = wx_get_pcie_msix_counts(wx, &msix_count, TXGBE_MAX_MSIX_VECTORS);
        if (err)
                wx_err(wx, "Do not support MSI-X\n");
        wx->mac.max_msix_vectors = msix_count;

        wx->ring_feature[RING_F_RSS].limit = min_t(int, TXGBE_MAX_RSS_INDICES,
                                                   num_online_cpus());
        wx->rss_enabled = true;

        wx->ring_feature[RING_F_FDIR].limit = min_t(int, TXGBE_MAX_FDIR_INDICES,
                                                    num_online_cpus());
        set_bit(WX_FLAG_FDIR_CAPABLE, wx->flags);
        set_bit(WX_FLAG_FDIR_HASH, wx->flags);
        wx->atr_sample_rate = TXGBE_DEFAULT_ATR_SAMPLE_RATE;
        wx->atr = txgbe_atr;
        wx->configure_fdir = txgbe_configure_fdir;

        set_bit(WX_FLAG_RSC_CAPABLE, wx->flags);
        set_bit(WX_FLAG_RSC_ENABLED, wx->flags);
        set_bit(WX_FLAG_MULTI_64_FUNC, wx->flags);

        /* enable itr by default in dynamic mode */
        wx->adaptive_itr = true;
        wx->rx_itr_setting = 1;
        wx->tx_itr_setting = 1;

        /* set default ring sizes */
        wx->tx_ring_count = TXGBE_DEFAULT_TXD;
        wx->rx_ring_count = TXGBE_DEFAULT_RXD;
        wx->mbx.size = WX_VXMAILBOX_SIZE;

        /* set default work limits */
        wx->tx_work_limit = TXGBE_DEFAULT_TX_WORK;
        wx->rx_work_limit = TXGBE_DEFAULT_RX_WORK;

        wx->setup_tc = txgbe_setup_tc;
        wx->do_reset = txgbe_do_reset;
        set_bit(0, &wx->fwd_bitmask);

        switch (wx->mac.type) {
        case wx_mac_sp:
                break;
        case wx_mac_aml:
        case wx_mac_aml40:
                set_bit(WX_FLAG_RX_MERGE_ENABLED, wx->flags);
                set_bit(WX_FLAG_TXHEAD_WB_ENABLED, wx->flags);
                set_bit(WX_FLAG_SWFW_RING, wx->flags);
                wx->swfw_index = 0;
                break;
        default:
                break;
        }

        return 0;
}

static void txgbe_init_fdir(struct txgbe *txgbe)
{
        txgbe->fdir_filter_count = 0;
        spin_lock_init(&txgbe->fdir_perfect_lock);
}

/**
 * txgbe_open - Called when a network interface is made active
 * @netdev: network interface device structure
 *
 * Returns 0 on success, negative value on failure
 *
 * The open entry point is called when a network interface is made
 * active by the system (IFF_UP).
 **/
static int txgbe_open(struct net_device *netdev)
{
        struct wx *wx = netdev_priv(netdev);
        int err;

        err = wx_setup_resources(wx);
        if (err)
                goto err_reset;

        wx_configure(wx);

        err = txgbe_setup_misc_irq(wx->priv);
        if (err)
                goto err_free_resources;

        err = txgbe_request_queue_irqs(wx);
        if (err)
                goto err_free_misc_irq;

        /* Notify the stack of the actual queue counts. */
        err = netif_set_real_num_tx_queues(netdev, wx->num_tx_queues);
        if (err)
                goto err_free_irq;

        err = netif_set_real_num_rx_queues(netdev, wx->num_rx_queues);
        if (err)
                goto err_free_irq;

        wx_ptp_init(wx);

        txgbe_up_complete(wx);

        return 0;

err_free_irq:
        wx_free_irq(wx);
err_free_misc_irq:
        txgbe_free_misc_irq(wx->priv);
        wx_reset_interrupt_capability(wx);
err_free_resources:
        wx_free_resources(wx);
err_reset:
        txgbe_reset(wx);

        return err;
}

/**
 * txgbe_close_suspend - actions necessary to both suspend and close flows
 * @wx: the private wx struct
 *
 * This function should contain the necessary work common to both suspending
 * and closing of the device.
 */
static void txgbe_close_suspend(struct wx *wx)
{
        wx_ptp_suspend(wx);
        txgbe_disable_device(wx);
        wx_free_resources(wx);
}

/**
 * txgbe_close - Disables a network interface
 * @netdev: network interface device structure
 *
 * Returns 0, this is not allowed to fail
 *
 * The close entry point is called when an interface is de-activated
 * by the OS.  The hardware is still under the drivers control, but
 * needs to be disabled.  A global MAC reset is issued to stop the
 * hardware, and all transmit and receive resources are freed.
 **/
static int txgbe_close(struct net_device *netdev)
{
        struct wx *wx = netdev_priv(netdev);

        wx_ptp_stop(wx);
        txgbe_down(wx);
        wx_free_irq(wx);
        txgbe_free_misc_irq(wx->priv);
        wx_free_resources(wx);
        txgbe_fdir_filter_exit(wx);
        wx_control_hw(wx, false);

        return 0;
}

static void txgbe_dev_shutdown(struct pci_dev *pdev)
{
        struct wx *wx = pci_get_drvdata(pdev);
        struct net_device *netdev;

        netdev = wx->netdev;
        netif_device_detach(netdev);

        rtnl_lock();
        if (netif_running(netdev))
                txgbe_close_suspend(wx);
        rtnl_unlock();

        wx_control_hw(wx, false);

        pci_disable_device(pdev);
}

static void txgbe_shutdown(struct pci_dev *pdev)
{
        txgbe_dev_shutdown(pdev);

        if (system_state == SYSTEM_POWER_OFF) {
                pci_wake_from_d3(pdev, false);
                pci_set_power_state(pdev, PCI_D3hot);
        }
}

/**
 * txgbe_setup_tc - routine to configure net_device for multiple traffic
 * classes.
 *
 * @dev: net device to configure
 * @tc: number of traffic classes to enable
 */
int txgbe_setup_tc(struct net_device *dev, u8 tc)
{
        struct wx *wx = netdev_priv(dev);

        /* Hardware has to reinitialize queues and interrupts to
         * match packet buffer alignment. Unfortunately, the
         * hardware is not flexible enough to do this dynamically.
         */
        if (netif_running(dev))
                txgbe_close(dev);
        else
                txgbe_reset(wx);

        wx_clear_interrupt_scheme(wx);

        if (tc)
                netdev_set_num_tc(dev, tc);
        else
                netdev_reset_tc(dev);

        wx_init_interrupt_scheme(wx);

        if (netif_running(dev))
                txgbe_open(dev);

        return 0;
}

static void txgbe_reinit_locked(struct wx *wx)
{
        int err = 0;

        netif_trans_update(wx->netdev);

        err = wx_set_state_reset(wx);
        if (err) {
                wx_err(wx, "wait device reset timeout\n");
                return;
        }

        txgbe_down(wx);
        txgbe_up(wx);

        clear_bit(WX_STATE_RESETTING, wx->state);
}

void txgbe_do_reset(struct net_device *netdev)
{
        struct wx *wx = netdev_priv(netdev);

        if (netif_running(netdev))
                txgbe_reinit_locked(wx);
        else
                txgbe_reset(wx);
}

static int txgbe_udp_tunnel_sync(struct net_device *dev, unsigned int table)
{
        struct wx *wx = netdev_priv(dev);
        struct udp_tunnel_info ti;

        udp_tunnel_nic_get_port(dev, table, 0, &ti);
        switch (ti.type) {
        case UDP_TUNNEL_TYPE_VXLAN:
                wr32(wx, TXGBE_CFG_VXLAN, ntohs(ti.port));
                break;
        case UDP_TUNNEL_TYPE_VXLAN_GPE:
                wr32(wx, TXGBE_CFG_VXLAN_GPE, ntohs(ti.port));
                break;
        case UDP_TUNNEL_TYPE_GENEVE:
                wr32(wx, TXGBE_CFG_GENEVE, ntohs(ti.port));
                break;
        default:
                break;
        }

        return 0;
}

static const struct udp_tunnel_nic_info txgbe_udp_tunnels = {
        .sync_table     = txgbe_udp_tunnel_sync,
        .flags          = UDP_TUNNEL_NIC_INFO_OPEN_ONLY,
        .tables         = {
                { .n_entries = 1, .tunnel_types = UDP_TUNNEL_TYPE_VXLAN, },
                { .n_entries = 1, .tunnel_types = UDP_TUNNEL_TYPE_VXLAN_GPE, },
                { .n_entries = 1, .tunnel_types = UDP_TUNNEL_TYPE_GENEVE, },
        },
};

static const struct net_device_ops txgbe_netdev_ops = {
        .ndo_open               = txgbe_open,
        .ndo_stop               = txgbe_close,
        .ndo_change_mtu         = wx_change_mtu,
        .ndo_start_xmit         = wx_xmit_frame,
        .ndo_set_rx_mode        = wx_set_rx_mode,
        .ndo_set_features       = wx_set_features,
        .ndo_fix_features       = wx_fix_features,
        .ndo_features_check     = wx_features_check,
        .ndo_validate_addr      = eth_validate_addr,
        .ndo_set_mac_address    = wx_set_mac,
        .ndo_get_stats64        = wx_get_stats64,
        .ndo_vlan_rx_add_vid    = wx_vlan_rx_add_vid,
        .ndo_vlan_rx_kill_vid   = wx_vlan_rx_kill_vid,
        .ndo_hwtstamp_set       = wx_hwtstamp_set,
        .ndo_hwtstamp_get       = wx_hwtstamp_get,
};

/**
 * txgbe_probe - Device Initialization Routine
 * @pdev: PCI device information struct
 * @ent: entry in txgbe_pci_tbl
 *
 * Returns 0 on success, negative on failure
 *
 * txgbe_probe initializes an adapter identified by a pci_dev structure.
 * The OS initialization, configuring of the wx private structure,
 * and a hardware reset occur.
 **/
static int txgbe_probe(struct pci_dev *pdev,
                       const struct pci_device_id __always_unused *ent)
{
        struct net_device *netdev;
        int err, expected_gts;
        struct wx *wx = NULL;
        struct txgbe *txgbe;

        u16 eeprom_verh = 0, eeprom_verl = 0, offset = 0;
        u16 eeprom_cfg_blkh = 0, eeprom_cfg_blkl = 0;
        u16 build = 0, major = 0, patch = 0;
        u32 etrack_id = 0;

        err = pci_enable_device_mem(pdev);
        if (err)
                return err;

        err = dma_set_mask_and_coherent(&pdev->dev, DMA_BIT_MASK(64));
        if (err) {
                dev_err(&pdev->dev,
                        "No usable DMA configuration, aborting\n");
                goto err_pci_disable_dev;
        }

        err = pci_request_selected_regions(pdev,
                                           pci_select_bars(pdev, IORESOURCE_MEM),
                                           txgbe_driver_name);
        if (err) {
                dev_err(&pdev->dev,
                        "pci_request_selected_regions failed 0x%x\n", err);
                goto err_pci_disable_dev;
        }

        pci_set_master(pdev);

        netdev = devm_alloc_etherdev_mqs(&pdev->dev,
                                         sizeof(struct wx),
                                         TXGBE_MAX_TX_QUEUES,
                                         TXGBE_MAX_RX_QUEUES);
        if (!netdev) {
                err = -ENOMEM;
                goto err_pci_release_regions;
        }

        SET_NETDEV_DEV(netdev, &pdev->dev);

        wx = netdev_priv(netdev);
        wx->netdev = netdev;
        wx->pdev = pdev;

        wx->msg_enable = (1 << DEFAULT_DEBUG_LEVEL_SHIFT) - 1;

        wx->hw_addr = devm_ioremap(&pdev->dev,
                                   pci_resource_start(pdev, 0),
                                   pci_resource_len(pdev, 0));
        if (!wx->hw_addr) {
                err = -EIO;
                goto err_pci_release_regions;
        }

        /* The sapphire supports up to 63 VFs per pf, but physical
         * function also need one pool for basic networking.
         */
        pci_sriov_set_totalvfs(pdev, TXGBE_MAX_VFS_DRV_LIMIT);
        wx->driver_name = txgbe_driver_name;
        txgbe_set_ethtool_ops(netdev);
        netdev->netdev_ops = &txgbe_netdev_ops;
        netdev->udp_tunnel_nic_info = &txgbe_udp_tunnels;

        /* setup the private structure */
        err = txgbe_sw_init(wx);
        if (err)
                goto err_pci_release_regions;

        /* check if flash load is done after hw power up */
        err = wx_check_flash_load(wx, TXGBE_SPI_ILDR_STATUS_PERST);
        if (err)
                goto err_free_mac_table;
        err = wx_check_flash_load(wx, TXGBE_SPI_ILDR_STATUS_PWRRST);
        if (err)
                goto err_free_mac_table;

        err = wx_mng_present(wx);
        if (err) {
                dev_err(&pdev->dev, "Management capability is not present\n");
                goto err_free_mac_table;
        }

        err = txgbe_reset_hw(wx);
        if (err) {
                dev_err(&pdev->dev, "HW Init failed: %d\n", err);
                goto err_free_mac_table;
        }

        netdev->features = NETIF_F_SG |
                           NETIF_F_TSO |
                           NETIF_F_TSO6 |
                           NETIF_F_RXHASH |
                           NETIF_F_RXCSUM |
                           NETIF_F_HW_CSUM;

        netdev->gso_partial_features =  NETIF_F_GSO_ENCAP_ALL;
        netdev->features |= netdev->gso_partial_features;
        netdev->features |= NETIF_F_SCTP_CRC;
        netdev->vlan_features |= netdev->features | NETIF_F_TSO_MANGLEID;
        netdev->hw_enc_features |= netdev->vlan_features;
        netdev->features |= NETIF_F_VLAN_FEATURES;
        /* copy netdev features into list of user selectable features */
        netdev->hw_features |= netdev->features | NETIF_F_RXALL;
        netdev->hw_features |= NETIF_F_NTUPLE | NETIF_F_HW_TC;
        netdev->features |= NETIF_F_HIGHDMA;
        netdev->hw_features |= NETIF_F_GRO;
        netdev->features |= NETIF_F_GRO;
        netdev->hw_features |= NETIF_F_LRO;
        netdev->features |= NETIF_F_LRO;
        netdev->features |= NETIF_F_RX_UDP_TUNNEL_PORT;

        netdev->priv_flags |= IFF_UNICAST_FLT;
        netdev->priv_flags |= IFF_SUPP_NOFCS;
        netdev->priv_flags |= IFF_LIVE_ADDR_CHANGE;

        netdev->min_mtu = ETH_MIN_MTU;
        netdev->max_mtu = WX_MAX_JUMBO_FRAME_SIZE -
                          (ETH_HLEN + ETH_FCS_LEN + VLAN_HLEN);

        /* make sure the EEPROM is good */
        err = txgbe_validate_eeprom_checksum(wx, NULL);
        if (err != 0) {
                dev_err(&pdev->dev, "The EEPROM Checksum Is Not Valid\n");
                wr32(wx, WX_MIS_RST, WX_MIS_RST_SW_RST);
                err = -EIO;
                goto err_free_mac_table;
        }

        eth_hw_addr_set(netdev, wx->mac.perm_addr);
        wx_mac_set_default_filter(wx, wx->mac.perm_addr);

        txgbe_init_service(wx);

        err = wx_init_interrupt_scheme(wx);
        if (err)
                goto err_cancel_service;

        /* Save off EEPROM version number and Option Rom version which
         * together make a unique identify for the eeprom
         */
        wx_read_ee_hostif(wx,
                          wx->eeprom.sw_region_offset + TXGBE_EEPROM_VERSION_H,
                          &eeprom_verh);
        wx_read_ee_hostif(wx,
                          wx->eeprom.sw_region_offset + TXGBE_EEPROM_VERSION_L,
                          &eeprom_verl);
        etrack_id = (eeprom_verh << 16) | eeprom_verl;

        wx_read_ee_hostif(wx,
                          wx->eeprom.sw_region_offset + TXGBE_ISCSI_BOOT_CONFIG,
                          &offset);

        /* Make sure offset to SCSI block is valid */
        if (!(offset == 0x0) && !(offset == 0xffff)) {
                wx_read_ee_hostif(wx, offset + 0x84, &eeprom_cfg_blkh);
                wx_read_ee_hostif(wx, offset + 0x83, &eeprom_cfg_blkl);

                /* Only display Option Rom if exist */
                if (eeprom_cfg_blkl && eeprom_cfg_blkh) {
                        major = eeprom_cfg_blkl >> 8;
                        build = (eeprom_cfg_blkl << 8) | (eeprom_cfg_blkh >> 8);
                        patch = eeprom_cfg_blkh & 0x00ff;

                        snprintf(wx->eeprom_id, sizeof(wx->eeprom_id),
                                 "0x%08x, %d.%d.%d", etrack_id, major, build,
                                 patch);
                } else {
                        snprintf(wx->eeprom_id, sizeof(wx->eeprom_id),
                                 "0x%08x", etrack_id);
                }
        } else {
                snprintf(wx->eeprom_id, sizeof(wx->eeprom_id),
                         "0x%08x", etrack_id);
        }

        if (etrack_id < 0x20010)
                dev_warn(&pdev->dev, "Please upgrade the firmware to 0x20010 or above.\n");

        err = txgbe_test_hostif(wx);
        if (err != 0) {
                dev_err(&pdev->dev, "Mismatched Firmware version\n");
                err = -EIO;
                goto err_release_hw;
        }

        txgbe = devm_kzalloc(&pdev->dev, sizeof(*txgbe), GFP_KERNEL);
        if (!txgbe) {
                err = -ENOMEM;
                goto err_release_hw;
        }

        txgbe->wx = wx;
        wx->priv = txgbe;

        txgbe_init_fdir(txgbe);

        err = txgbe_init_phy(txgbe);
        if (err)
                goto err_release_hw;

        err = register_netdev(netdev);
        if (err)
                goto err_remove_phy;

        pci_set_drvdata(pdev, wx);

        netif_tx_stop_all_queues(netdev);

        /* calculate the expected PCIe bandwidth required for optimal
         * performance. Note that some older parts will never have enough
         * bandwidth due to being older generation PCIe parts. We clamp these
         * parts to ensure that no warning is displayed, as this could confuse
         * users otherwise.
         */
        expected_gts = txgbe_enumerate_functions(wx) * 10;

        /* don't check link if we failed to enumerate functions */
        if (expected_gts > 0)
                txgbe_check_minimum_link(wx);
        else
                dev_warn(&pdev->dev, "Failed to enumerate PF devices.\n");

        return 0;

err_remove_phy:
        txgbe_remove_phy(txgbe);
err_release_hw:
        wx_clear_interrupt_scheme(wx);
        wx_control_hw(wx, false);
err_cancel_service:
        timer_delete_sync(&wx->service_timer);
        cancel_work_sync(&wx->service_task);
err_free_mac_table:
        kfree(wx->rss_key);
        kfree(wx->mac_table);
err_pci_release_regions:
        pci_release_selected_regions(pdev,
                                     pci_select_bars(pdev, IORESOURCE_MEM));
err_pci_disable_dev:
        pci_disable_device(pdev);
        return err;
}

/**
 * txgbe_remove - Device Removal Routine
 * @pdev: PCI device information struct
 *
 * txgbe_remove is called by the PCI subsystem to alert the driver
 * that it should release a PCI device.  The could be caused by a
 * Hot-Plug event, or because the driver is going to be removed from
 * memory.
 **/
static void txgbe_remove(struct pci_dev *pdev)
{
        struct wx *wx = pci_get_drvdata(pdev);
        struct txgbe *txgbe = wx->priv;
        struct net_device *netdev;

        cancel_work_sync(&wx->service_task);

        netdev = wx->netdev;
        wx_disable_sriov(wx);
        unregister_netdev(netdev);

        txgbe_remove_phy(txgbe);
        wx_free_isb_resources(wx);

        pci_release_selected_regions(pdev,
                                     pci_select_bars(pdev, IORESOURCE_MEM));

        kfree(wx->rss_key);
        kfree(wx->mac_table);
        wx_clear_interrupt_scheme(wx);

        pci_disable_device(pdev);
}

static struct pci_driver txgbe_driver = {
        .name     = txgbe_driver_name,
        .id_table = txgbe_pci_tbl,
        .probe    = txgbe_probe,
        .remove   = txgbe_remove,
        .shutdown = txgbe_shutdown,
        .sriov_configure = wx_pci_sriov_configure,
};

module_pci_driver(txgbe_driver);

MODULE_DEVICE_TABLE(pci, txgbe_pci_tbl);
MODULE_AUTHOR("Beijing WangXun Technology Co., Ltd, <software@trustnetic.com>");
MODULE_DESCRIPTION("WangXun(R) 10/25/40 Gigabit PCI Express Network Driver");
MODULE_LICENSE("GPL");