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

#include <linux/pci.h>
#include <linux/phy.h>
#include <linux/netdevice.h>

#include "../libwx/wx_ethtool.h"
#include "../libwx/wx_type.h"
#include "../libwx/wx_lib.h"
#include "../libwx/wx_hw.h"
#include "ngbe_ethtool.h"
#include "ngbe_type.h"

static void ngbe_get_wol(struct net_device *netdev,
                         struct ethtool_wolinfo *wol)
{
        struct wx *wx = netdev_priv(netdev);

        if (!wx->wol_hw_supported)
                return;
        wol->supported = WAKE_MAGIC;
        wol->wolopts = 0;
        if (wx->wol & WX_PSR_WKUP_CTL_MAG)
                wol->wolopts |= WAKE_MAGIC;
}

static int ngbe_set_wol(struct net_device *netdev,
                        struct ethtool_wolinfo *wol)
{
        struct wx *wx = netdev_priv(netdev);
        struct pci_dev *pdev = wx->pdev;

        if (!wx->wol_hw_supported)
                return -EOPNOTSUPP;

        wx->wol = 0;
        if (wol->wolopts & WAKE_MAGIC)
                wx->wol = WX_PSR_WKUP_CTL_MAG;
        netdev->ethtool->wol_enabled = !!(wx->wol);
        wr32(wx, WX_PSR_WKUP_CTL, wx->wol);
        device_set_wakeup_enable(&pdev->dev, netdev->ethtool->wol_enabled);

        return 0;
}

static int ngbe_set_ringparam(struct net_device *netdev,
                              struct ethtool_ringparam *ring,
                              struct kernel_ethtool_ringparam *kernel_ring,
                              struct netlink_ext_ack *extack)
{
        struct wx *wx = netdev_priv(netdev);
        u32 new_rx_count, new_tx_count;
        struct wx_ring *temp_ring;
        int i, err = 0;

        new_tx_count = clamp_t(u32, ring->tx_pending, WX_MIN_TXD, WX_MAX_TXD);
        new_tx_count = ALIGN(new_tx_count, WX_REQ_TX_DESCRIPTOR_MULTIPLE);

        new_rx_count = clamp_t(u32, ring->rx_pending, WX_MIN_RXD, WX_MAX_RXD);
        new_rx_count = ALIGN(new_rx_count, WX_REQ_RX_DESCRIPTOR_MULTIPLE);

        if (new_tx_count == wx->tx_ring_count &&
            new_rx_count == wx->rx_ring_count)
                return 0;

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

        if (!netif_running(wx->netdev)) {
                for (i = 0; i < wx->num_tx_queues; i++)
                        wx->tx_ring[i]->count = new_tx_count;
                for (i = 0; i < wx->num_rx_queues; i++)
                        wx->rx_ring[i]->count = new_rx_count;
                wx->tx_ring_count = new_tx_count;
                wx->rx_ring_count = new_rx_count;

                goto clear_reset;
        }

        /* allocate temporary buffer to store rings in */
        i = max_t(int, wx->num_tx_queues, wx->num_rx_queues);
        temp_ring = kvmalloc_objs(struct wx_ring, i);
        if (!temp_ring) {
                err = -ENOMEM;
                goto clear_reset;
        }

        ngbe_down(wx);

        wx_set_ring(wx, new_tx_count, new_rx_count, temp_ring);
        kvfree(temp_ring);

        wx_configure(wx);
        ngbe_up(wx);

clear_reset:
        clear_bit(WX_STATE_RESETTING, wx->state);
        return err;
}

static int ngbe_set_channels(struct net_device *dev,
                             struct ethtool_channels *ch)
{
        int err;

        err = wx_set_channels(dev, ch);
        if (err < 0)
                return err;

        /* use setup TC to update any traffic class queue mapping */
        return ngbe_setup_tc(dev, netdev_get_num_tc(dev));
}

static const struct ethtool_ops ngbe_ethtool_ops = {
        .supported_coalesce_params = ETHTOOL_COALESCE_USECS |
                                     ETHTOOL_COALESCE_TX_MAX_FRAMES_IRQ |
                                     ETHTOOL_COALESCE_USE_ADAPTIVE,
        .get_drvinfo            = wx_get_drvinfo,
        .get_link               = ethtool_op_get_link,
        .get_link_ksettings     = wx_get_link_ksettings,
        .set_link_ksettings     = wx_set_link_ksettings,
        .nway_reset             = wx_nway_reset,
        .get_wol                = ngbe_get_wol,
        .set_wol                = ngbe_set_wol,
        .get_sset_count         = wx_get_sset_count,
        .get_strings            = wx_get_strings,
        .get_ethtool_stats      = wx_get_ethtool_stats,
        .get_eth_mac_stats      = wx_get_mac_stats,
        .get_pause_stats        = wx_get_pause_stats,
        .get_pauseparam         = wx_get_pauseparam,
        .set_pauseparam         = wx_set_pauseparam,
        .get_ringparam          = wx_get_ringparam,
        .set_ringparam          = ngbe_set_ringparam,
        .get_coalesce           = wx_get_coalesce,
        .set_coalesce           = wx_set_coalesce,
        .get_channels           = wx_get_channels,
        .set_channels           = ngbe_set_channels,
        .get_rxfh_fields        = wx_get_rxfh_fields,
        .set_rxfh_fields        = wx_set_rxfh_fields,
        .get_rxfh_indir_size    = wx_rss_indir_size,
        .get_rxfh_key_size      = wx_get_rxfh_key_size,
        .get_rxfh               = wx_get_rxfh,
        .set_rxfh               = wx_set_rxfh,
        .get_msglevel           = wx_get_msglevel,
        .set_msglevel           = wx_set_msglevel,
        .get_ts_info            = wx_get_ts_info,
        .get_ts_stats           = wx_get_ptp_stats,
};

void ngbe_set_ethtool_ops(struct net_device *netdev)
{
        netdev->ethtool_ops = &ngbe_ethtool_ops;
}