root/drivers/net/ethernet/google/gve/gve_ethtool.c
// SPDX-License-Identifier: (GPL-2.0 OR MIT)
/* Google virtual Ethernet (gve) driver
 *
 * Copyright (C) 2015-2024 Google LLC
 */

#include <net/netdev_lock.h>
#include "gve.h"
#include "gve_adminq.h"
#include "gve_dqo.h"
#include "gve_utils.h"

static void gve_get_drvinfo(struct net_device *netdev,
                            struct ethtool_drvinfo *info)
{
        struct gve_priv *priv = netdev_priv(netdev);

        strscpy(info->driver, gve_driver_name, sizeof(info->driver));
        strscpy(info->version, gve_version_str, sizeof(info->version));
        strscpy(info->bus_info, pci_name(priv->pdev), sizeof(info->bus_info));
}

static void gve_set_msglevel(struct net_device *netdev, u32 value)
{
        struct gve_priv *priv = netdev_priv(netdev);

        priv->msg_enable = value;
}

static u32 gve_get_msglevel(struct net_device *netdev)
{
        struct gve_priv *priv = netdev_priv(netdev);

        return priv->msg_enable;
}

/* For the following stats column string names, make sure the order
 * matches how it is filled in the code. For xdp_aborted, xdp_drop,
 * xdp_pass, xdp_tx, xdp_redirect, make sure it also matches the order
 * as declared in enum xdp_action inside file uapi/linux/bpf.h .
 */
static const char gve_gstrings_main_stats[][ETH_GSTRING_LEN] = {
        "rx_packets", "rx_hsplit_pkt", "tx_packets", "rx_bytes",
        "tx_bytes", "rx_dropped", "tx_dropped", "tx_timeouts",
        "rx_skb_alloc_fail", "rx_buf_alloc_fail", "rx_desc_err_dropped_pkt",
        "rx_hsplit_unsplit_pkt",
        "interface_up_cnt", "interface_down_cnt", "reset_cnt",
        "page_alloc_fail", "dma_mapping_error", "stats_report_trigger_cnt",
};

static const char gve_gstrings_rx_stats[][ETH_GSTRING_LEN] = {
        "rx_posted_desc[%u]", "rx_completed_desc[%u]", "rx_consumed_desc[%u]",
        "rx_bytes[%u]", "rx_hsplit_bytes[%u]", "rx_cont_packet_cnt[%u]",
        "rx_frag_flip_cnt[%u]", "rx_frag_copy_cnt[%u]", "rx_frag_alloc_cnt[%u]",
        "rx_dropped_pkt[%u]", "rx_copybreak_pkt[%u]", "rx_copied_pkt[%u]",
        "rx_queue_drop_cnt[%u]", "rx_no_buffers_posted[%u]",
        "rx_drops_packet_over_mru[%u]", "rx_drops_invalid_checksum[%u]",
        "rx_xdp_aborted[%u]", "rx_xdp_drop[%u]", "rx_xdp_pass[%u]",
        "rx_xdp_tx[%u]", "rx_xdp_redirect[%u]",
        "rx_xdp_tx_errors[%u]", "rx_xdp_redirect_errors[%u]", "rx_xdp_alloc_fails[%u]",
};

static const char gve_gstrings_tx_stats[][ETH_GSTRING_LEN] = {
        "tx_posted_desc[%u]", "tx_completed_desc[%u]", "tx_consumed_desc[%u]", "tx_bytes[%u]",
        "tx_wake[%u]", "tx_stop[%u]", "tx_event_counter[%u]",
        "tx_dma_mapping_error[%u]",
        "tx_xsk_sent[%u]", "tx_xdp_xmit[%u]", "tx_xdp_xmit_errors[%u]"
};

static const char gve_gstrings_adminq_stats[][ETH_GSTRING_LEN] __nonstring_array = {
        "adminq_prod_cnt", "adminq_cmd_fail", "adminq_timeouts",
        "adminq_describe_device_cnt", "adminq_cfg_device_resources_cnt",
        "adminq_register_page_list_cnt", "adminq_unregister_page_list_cnt",
        "adminq_create_tx_queue_cnt", "adminq_create_rx_queue_cnt",
        "adminq_destroy_tx_queue_cnt", "adminq_destroy_rx_queue_cnt",
        "adminq_dcfg_device_resources_cnt", "adminq_set_driver_parameter_cnt",
        "adminq_report_stats_cnt", "adminq_report_link_speed_cnt", "adminq_get_ptype_map_cnt",
        "adminq_query_flow_rules", "adminq_cfg_flow_rule", "adminq_cfg_rss_cnt",
        "adminq_query_rss_cnt", "adminq_report_nic_timestamp_cnt",
};

static const char gve_gstrings_priv_flags[][ETH_GSTRING_LEN] = {
        "report-stats",
};

#define GVE_MAIN_STATS_LEN  ARRAY_SIZE(gve_gstrings_main_stats)
#define GVE_ADMINQ_STATS_LEN  ARRAY_SIZE(gve_gstrings_adminq_stats)
#define NUM_GVE_TX_CNTS ARRAY_SIZE(gve_gstrings_tx_stats)
#define NUM_GVE_RX_CNTS ARRAY_SIZE(gve_gstrings_rx_stats)
#define GVE_PRIV_FLAGS_STR_LEN ARRAY_SIZE(gve_gstrings_priv_flags)

static void gve_get_strings(struct net_device *netdev, u32 stringset, u8 *data)
{
        struct gve_priv *priv = netdev_priv(netdev);
        u8 *s = (char *)data;
        int num_tx_queues;
        int i, j;

        num_tx_queues = gve_num_tx_queues(priv);
        switch (stringset) {
        case ETH_SS_STATS:
                for (i = 0; i < ARRAY_SIZE(gve_gstrings_main_stats); i++)
                        ethtool_puts(&s, gve_gstrings_main_stats[i]);

                for (i = 0; i < priv->rx_cfg.num_queues; i++)
                        for (j = 0; j < NUM_GVE_RX_CNTS; j++)
                                ethtool_sprintf(&s, gve_gstrings_rx_stats[j],
                                                i);

                for (i = 0; i < num_tx_queues; i++)
                        for (j = 0; j < NUM_GVE_TX_CNTS; j++)
                                ethtool_sprintf(&s, gve_gstrings_tx_stats[j],
                                                i);

                for (i = 0; i < ARRAY_SIZE(gve_gstrings_adminq_stats); i++)
                        ethtool_cpy(&s, gve_gstrings_adminq_stats[i]);

                break;

        case ETH_SS_PRIV_FLAGS:
                for (i = 0; i < ARRAY_SIZE(gve_gstrings_priv_flags); i++)
                        ethtool_puts(&s, gve_gstrings_priv_flags[i]);
                break;

        default:
                break;
        }
}

static int gve_get_sset_count(struct net_device *netdev, int sset)
{
        struct gve_priv *priv = netdev_priv(netdev);
        int num_tx_queues;

        num_tx_queues = gve_num_tx_queues(priv);
        switch (sset) {
        case ETH_SS_STATS:
                return GVE_MAIN_STATS_LEN + GVE_ADMINQ_STATS_LEN +
                       (priv->rx_cfg.num_queues * NUM_GVE_RX_CNTS) +
                       (num_tx_queues * NUM_GVE_TX_CNTS);
        case ETH_SS_PRIV_FLAGS:
                return GVE_PRIV_FLAGS_STR_LEN;
        default:
                return -EOPNOTSUPP;
        }
}

static void
gve_get_ethtool_stats(struct net_device *netdev,
                      struct ethtool_stats *stats, u64 *data)
{
        u64 tmp_rx_pkts, tmp_rx_hsplit_pkt, tmp_rx_bytes, tmp_rx_hsplit_bytes,
                tmp_rx_skb_alloc_fail, tmp_rx_buf_alloc_fail,
                tmp_rx_desc_err_dropped_pkt, tmp_rx_hsplit_unsplit_pkt,
                tmp_tx_pkts, tmp_tx_bytes,
                tmp_xdp_tx_errors, tmp_xdp_redirect_errors;
        u64 rx_buf_alloc_fail, rx_desc_err_dropped_pkt, rx_hsplit_unsplit_pkt,
                rx_pkts, rx_hsplit_pkt, rx_skb_alloc_fail, rx_bytes, tx_pkts, tx_bytes,
                tx_dropped, xdp_tx_errors, xdp_redirect_errors;
        int rx_base_stats_idx, max_rx_stats_idx, max_tx_stats_idx;
        int stats_idx, stats_region_len, nic_stats_len;
        struct stats *report_stats;
        int *rx_qid_to_stats_idx;
        int *tx_qid_to_stats_idx;
        int num_stopped_rxqs = 0;
        int num_stopped_txqs = 0;
        struct gve_priv *priv;
        bool skip_nic_stats;
        unsigned int start;
        int num_tx_queues;
        int ring;
        int i, j;

        netdev_assert_locked(netdev);

        priv = netdev_priv(netdev);
        num_tx_queues = gve_num_tx_queues(priv);
        report_stats = priv->stats_report->stats;
        rx_qid_to_stats_idx = kmalloc_objs(int, priv->rx_cfg.num_queues);
        if (!rx_qid_to_stats_idx)
                return;
        for (ring = 0; ring < priv->rx_cfg.num_queues; ring++) {
                rx_qid_to_stats_idx[ring] = -1;
                if (!gve_rx_was_added_to_block(priv, ring))
                        num_stopped_rxqs++;
        }
        tx_qid_to_stats_idx = kmalloc_objs(int, num_tx_queues);
        if (!tx_qid_to_stats_idx) {
                kfree(rx_qid_to_stats_idx);
                return;
        }
        for (ring = 0; ring < num_tx_queues; ring++) {
                tx_qid_to_stats_idx[ring] = -1;
                if (!gve_tx_was_added_to_block(priv, ring))
                        num_stopped_txqs++;
        }

        for (rx_pkts = 0, rx_bytes = 0, rx_hsplit_pkt = 0,
             rx_skb_alloc_fail = 0, rx_buf_alloc_fail = 0,
             rx_desc_err_dropped_pkt = 0, rx_hsplit_unsplit_pkt = 0,
             xdp_tx_errors = 0, xdp_redirect_errors = 0,
             ring = 0;
             ring < priv->rx_cfg.num_queues; ring++) {
                if (priv->rx) {
                        do {
                                struct gve_rx_ring *rx = &priv->rx[ring];

                                start =
                                  u64_stats_fetch_begin(&priv->rx[ring].statss);
                                tmp_rx_pkts = rx->rpackets;
                                tmp_rx_hsplit_pkt = rx->rx_hsplit_pkt;
                                tmp_rx_bytes = rx->rbytes;
                                tmp_rx_skb_alloc_fail = rx->rx_skb_alloc_fail;
                                tmp_rx_buf_alloc_fail = rx->rx_buf_alloc_fail;
                                tmp_rx_desc_err_dropped_pkt =
                                        rx->rx_desc_err_dropped_pkt;
                                tmp_rx_hsplit_unsplit_pkt =
                                        rx->rx_hsplit_unsplit_pkt;
                                tmp_xdp_tx_errors = rx->xdp_tx_errors;
                                tmp_xdp_redirect_errors =
                                        rx->xdp_redirect_errors;
                        } while (u64_stats_fetch_retry(&priv->rx[ring].statss,
                                                       start));
                        rx_pkts += tmp_rx_pkts;
                        rx_hsplit_pkt += tmp_rx_hsplit_pkt;
                        rx_bytes += tmp_rx_bytes;
                        rx_skb_alloc_fail += tmp_rx_skb_alloc_fail;
                        rx_buf_alloc_fail += tmp_rx_buf_alloc_fail;
                        rx_desc_err_dropped_pkt += tmp_rx_desc_err_dropped_pkt;
                        rx_hsplit_unsplit_pkt += tmp_rx_hsplit_unsplit_pkt;
                        xdp_tx_errors += tmp_xdp_tx_errors;
                        xdp_redirect_errors += tmp_xdp_redirect_errors;
                }
        }
        for (tx_pkts = 0, tx_bytes = 0, tx_dropped = 0, ring = 0;
             ring < num_tx_queues; ring++) {
                if (priv->tx) {
                        do {
                                start =
                                  u64_stats_fetch_begin(&priv->tx[ring].statss);
                                tmp_tx_pkts = priv->tx[ring].pkt_done;
                                tmp_tx_bytes = priv->tx[ring].bytes_done;
                        } while (u64_stats_fetch_retry(&priv->tx[ring].statss,
                                                       start));
                        tx_pkts += tmp_tx_pkts;
                        tx_bytes += tmp_tx_bytes;
                        tx_dropped += priv->tx[ring].dropped_pkt;
                }
        }

        i = 0;
        data[i++] = rx_pkts;
        data[i++] = rx_hsplit_pkt;
        data[i++] = tx_pkts;
        data[i++] = rx_bytes;
        data[i++] = tx_bytes;
        /* total rx dropped packets */
        data[i++] = rx_skb_alloc_fail + rx_desc_err_dropped_pkt +
                    xdp_tx_errors + xdp_redirect_errors;
        data[i++] = tx_dropped;
        data[i++] = priv->tx_timeo_cnt;
        data[i++] = rx_skb_alloc_fail;
        data[i++] = rx_buf_alloc_fail;
        data[i++] = rx_desc_err_dropped_pkt;
        data[i++] = rx_hsplit_unsplit_pkt;
        data[i++] = priv->interface_up_cnt;
        data[i++] = priv->interface_down_cnt;
        data[i++] = priv->reset_cnt;
        data[i++] = priv->page_alloc_fail;
        data[i++] = priv->dma_mapping_error;
        data[i++] = priv->stats_report_trigger_cnt;
        i = GVE_MAIN_STATS_LEN;

        rx_base_stats_idx = 0;
        max_rx_stats_idx = 0;
        max_tx_stats_idx = 0;
        stats_region_len = priv->stats_report_len -
                                sizeof(struct gve_stats_report);
        nic_stats_len = (NIC_RX_STATS_REPORT_NUM * priv->rx_cfg.num_queues +
                NIC_TX_STATS_REPORT_NUM * num_tx_queues) * sizeof(struct stats);
        if (unlikely((stats_region_len -
                                nic_stats_len) % sizeof(struct stats))) {
                net_err_ratelimited("Starting index of NIC stats should be multiple of stats size");
        } else {
                /* For rx cross-reporting stats,
                 * start from nic rx stats in report
                 */
                rx_base_stats_idx = (stats_region_len - nic_stats_len) /
                                                        sizeof(struct stats);
                /* The boundary between driver stats and NIC stats
                 * shifts if there are stopped queues
                 */
                rx_base_stats_idx += NIC_RX_STATS_REPORT_NUM *
                        num_stopped_rxqs + NIC_TX_STATS_REPORT_NUM *
                        num_stopped_txqs;
                max_rx_stats_idx = NIC_RX_STATS_REPORT_NUM *
                        (priv->rx_cfg.num_queues - num_stopped_rxqs) +
                        rx_base_stats_idx;
                max_tx_stats_idx = NIC_TX_STATS_REPORT_NUM *
                        (num_tx_queues - num_stopped_txqs) +
                        max_rx_stats_idx;
        }
        /* Preprocess the stats report for rx, map queue id to start index */
        skip_nic_stats = false;
        for (stats_idx = rx_base_stats_idx; stats_idx < max_rx_stats_idx;
                stats_idx += NIC_RX_STATS_REPORT_NUM) {
                u32 stat_name = be32_to_cpu(report_stats[stats_idx].stat_name);
                u32 queue_id = be32_to_cpu(report_stats[stats_idx].queue_id);

                if (stat_name == 0) {
                        /* no stats written by NIC yet */
                        skip_nic_stats = true;
                        break;
                }
                if (queue_id < 0 || queue_id >= priv->rx_cfg.num_queues) {
                        net_err_ratelimited("Invalid rxq id in NIC stats\n");
                        continue;
                }
                rx_qid_to_stats_idx[queue_id] = stats_idx;
        }
        /* walk RX rings */
        if (priv->rx) {
                for (ring = 0; ring < priv->rx_cfg.num_queues; ring++) {
                        struct gve_rx_ring *rx = &priv->rx[ring];

                        data[i++] = rx->fill_cnt;
                        data[i++] = rx->cnt;
                        data[i++] = rx->fill_cnt - rx->cnt;
                        do {
                                start =
                                  u64_stats_fetch_begin(&priv->rx[ring].statss);
                                tmp_rx_bytes = rx->rbytes;
                                tmp_rx_hsplit_bytes = rx->rx_hsplit_bytes;
                                tmp_rx_skb_alloc_fail = rx->rx_skb_alloc_fail;
                                tmp_rx_buf_alloc_fail = rx->rx_buf_alloc_fail;
                                tmp_rx_desc_err_dropped_pkt =
                                        rx->rx_desc_err_dropped_pkt;
                                tmp_xdp_tx_errors = rx->xdp_tx_errors;
                                tmp_xdp_redirect_errors =
                                        rx->xdp_redirect_errors;
                        } while (u64_stats_fetch_retry(&priv->rx[ring].statss,
                                                       start));
                        data[i++] = tmp_rx_bytes;
                        data[i++] = tmp_rx_hsplit_bytes;
                        data[i++] = rx->rx_cont_packet_cnt;
                        data[i++] = rx->rx_frag_flip_cnt;
                        data[i++] = rx->rx_frag_copy_cnt;
                        data[i++] = rx->rx_frag_alloc_cnt;
                        /* rx dropped packets */
                        data[i++] = tmp_rx_skb_alloc_fail +
                                    tmp_rx_desc_err_dropped_pkt +
                                    tmp_xdp_tx_errors +
                                    tmp_xdp_redirect_errors;
                        data[i++] = rx->rx_copybreak_pkt;
                        data[i++] = rx->rx_copied_pkt;
                        /* stats from NIC */
                        stats_idx = rx_qid_to_stats_idx[ring];
                        if (skip_nic_stats || stats_idx < 0) {
                                /* skip NIC rx stats */
                                i += NIC_RX_STATS_REPORT_NUM;
                        } else {
                                for (j = 0; j < NIC_RX_STATS_REPORT_NUM; j++) {
                                        u64 value =
                                                be64_to_cpu(report_stats[stats_idx + j].value);

                                        data[i++] = value;
                                }
                        }
                        /* XDP rx counters */
                        do {
                                start = u64_stats_fetch_begin(&priv->rx[ring].statss);
                                for (j = 0; j < GVE_XDP_ACTIONS; j++)
                                        data[i + j] = rx->xdp_actions[j];
                                data[i + j++] = rx->xdp_tx_errors;
                                data[i + j++] = rx->xdp_redirect_errors;
                                data[i + j++] = rx->xdp_alloc_fails;
                        } while (u64_stats_fetch_retry(&priv->rx[ring].statss,
                                                       start));
                        i += GVE_XDP_ACTIONS + 3; /* XDP rx counters */
                }
        } else {
                i += priv->rx_cfg.num_queues * NUM_GVE_RX_CNTS;
        }

        skip_nic_stats = false;
        /* NIC TX stats start right after NIC RX stats */
        for (stats_idx = max_rx_stats_idx; stats_idx < max_tx_stats_idx;
                stats_idx += NIC_TX_STATS_REPORT_NUM) {
                u32 stat_name = be32_to_cpu(report_stats[stats_idx].stat_name);
                u32 queue_id = be32_to_cpu(report_stats[stats_idx].queue_id);

                if (stat_name == 0) {
                        /* no stats written by NIC yet */
                        skip_nic_stats = true;
                        break;
                }
                if (queue_id < 0 || queue_id >= num_tx_queues) {
                        net_err_ratelimited("Invalid txq id in NIC stats\n");
                        continue;
                }
                tx_qid_to_stats_idx[queue_id] = stats_idx;
        }
        /* walk TX rings */
        if (priv->tx) {
                for (ring = 0; ring < num_tx_queues; ring++) {
                        struct gve_tx_ring *tx = &priv->tx[ring];

                        if (gve_is_gqi(priv)) {
                                data[i++] = tx->req;
                                data[i++] = tx->done;
                                data[i++] = tx->req - tx->done;
                        } else {
                                /* DQO doesn't currently support
                                 * posted/completed descriptor counts;
                                 */
                                data[i++] = 0;
                                data[i++] = 0;
                                data[i++] =
                                        (tx->dqo_tx.tail - tx->dqo_tx.head) &
                                        tx->mask;
                        }
                        do {
                                start =
                                  u64_stats_fetch_begin(&priv->tx[ring].statss);
                                tmp_tx_bytes = tx->bytes_done;
                        } while (u64_stats_fetch_retry(&priv->tx[ring].statss,
                                                       start));
                        data[i++] = tmp_tx_bytes;
                        data[i++] = tx->wake_queue;
                        data[i++] = tx->stop_queue;
                        data[i++] = gve_tx_load_event_counter(priv, tx);
                        data[i++] = tx->dma_mapping_error;
                        /* stats from NIC */
                        stats_idx = tx_qid_to_stats_idx[ring];
                        if (skip_nic_stats || stats_idx < 0) {
                                /* skip NIC tx stats */
                                i += NIC_TX_STATS_REPORT_NUM;
                        } else {
                                for (j = 0; j < NIC_TX_STATS_REPORT_NUM; j++) {
                                        u64 value =
                                                be64_to_cpu(report_stats[stats_idx + j].value);
                                        data[i++] = value;
                                }
                        }
                        /* XDP counters */
                        do {
                                start = u64_stats_fetch_begin(&priv->tx[ring].statss);
                                data[i] = tx->xdp_xsk_sent;
                                data[i + 1] = tx->xdp_xmit;
                                data[i + 2] = tx->xdp_xmit_errors;
                        } while (u64_stats_fetch_retry(&priv->tx[ring].statss,
                                                       start));
                        i += 3; /* XDP tx counters */
                }
        } else {
                i += num_tx_queues * NUM_GVE_TX_CNTS;
        }

        kfree(rx_qid_to_stats_idx);
        kfree(tx_qid_to_stats_idx);
        /* AQ Stats */
        data[i++] = priv->adminq_prod_cnt;
        data[i++] = priv->adminq_cmd_fail;
        data[i++] = priv->adminq_timeouts;
        data[i++] = priv->adminq_describe_device_cnt;
        data[i++] = priv->adminq_cfg_device_resources_cnt;
        data[i++] = priv->adminq_register_page_list_cnt;
        data[i++] = priv->adminq_unregister_page_list_cnt;
        data[i++] = priv->adminq_create_tx_queue_cnt;
        data[i++] = priv->adminq_create_rx_queue_cnt;
        data[i++] = priv->adminq_destroy_tx_queue_cnt;
        data[i++] = priv->adminq_destroy_rx_queue_cnt;
        data[i++] = priv->adminq_dcfg_device_resources_cnt;
        data[i++] = priv->adminq_set_driver_parameter_cnt;
        data[i++] = priv->adminq_report_stats_cnt;
        data[i++] = priv->adminq_report_link_speed_cnt;
        data[i++] = priv->adminq_get_ptype_map_cnt;
        data[i++] = priv->adminq_query_flow_rules_cnt;
        data[i++] = priv->adminq_cfg_flow_rule_cnt;
        data[i++] = priv->adminq_cfg_rss_cnt;
        data[i++] = priv->adminq_query_rss_cnt;
        data[i++] = priv->adminq_report_nic_timestamp_cnt;
}

static void gve_get_channels(struct net_device *netdev,
                             struct ethtool_channels *cmd)
{
        struct gve_priv *priv = netdev_priv(netdev);

        cmd->max_rx = priv->rx_cfg.max_queues;
        cmd->max_tx = priv->tx_cfg.max_queues;
        cmd->max_other = 0;
        cmd->max_combined = 0;
        cmd->rx_count = priv->rx_cfg.num_queues;
        cmd->tx_count = priv->tx_cfg.num_queues;
        cmd->other_count = 0;
        cmd->combined_count = 0;
}

static int gve_set_channels(struct net_device *netdev,
                            struct ethtool_channels *cmd)
{
        struct gve_priv *priv = netdev_priv(netdev);
        struct gve_tx_queue_config new_tx_cfg = priv->tx_cfg;
        struct gve_rx_queue_config new_rx_cfg = priv->rx_cfg;
        struct ethtool_channels old_settings;
        int new_tx = cmd->tx_count;
        int new_rx = cmd->rx_count;
        bool reset_rss = false;

        gve_get_channels(netdev, &old_settings);

        /* Changing combined is not allowed */
        if (cmd->combined_count != old_settings.combined_count)
                return -EINVAL;

        if (!new_rx || !new_tx)
                return -EINVAL;

        if (priv->xdp_prog) {
                if (new_tx != new_rx ||
                    (2 * new_tx > priv->tx_cfg.max_queues)) {
                        dev_err(&priv->pdev->dev, "The number of configured RX queues should be equal to the number of configured TX queues and the number of configured RX/TX queues should be less than or equal to half the maximum number of RX/TX queues when XDP program is installed");
                        return -EINVAL;
                }

                /* One XDP TX queue per RX queue. */
                new_tx_cfg.num_xdp_queues = new_rx;
        } else {
                new_tx_cfg.num_xdp_queues = 0;
        }

        if (new_rx != priv->rx_cfg.num_queues &&
            priv->cache_rss_config && !netif_is_rxfh_configured(netdev))
                reset_rss = true;

        new_tx_cfg.num_queues = new_tx;
        new_rx_cfg.num_queues = new_rx;

        return gve_adjust_queues(priv, new_rx_cfg, new_tx_cfg, reset_rss);
}

static void gve_get_ringparam(struct net_device *netdev,
                              struct ethtool_ringparam *cmd,
                              struct kernel_ethtool_ringparam *kernel_cmd,
                              struct netlink_ext_ack *extack)
{
        struct gve_priv *priv = netdev_priv(netdev);

        cmd->rx_max_pending = priv->max_rx_desc_cnt;
        cmd->tx_max_pending = priv->max_tx_desc_cnt;
        cmd->rx_pending = priv->rx_desc_cnt;
        cmd->tx_pending = priv->tx_desc_cnt;

        kernel_cmd->rx_buf_len = priv->rx_cfg.packet_buffer_size;

        if (!gve_header_split_supported(priv))
                kernel_cmd->tcp_data_split = ETHTOOL_TCP_DATA_SPLIT_UNKNOWN;
        else if (priv->header_split_enabled)
                kernel_cmd->tcp_data_split = ETHTOOL_TCP_DATA_SPLIT_ENABLED;
        else
                kernel_cmd->tcp_data_split = ETHTOOL_TCP_DATA_SPLIT_DISABLED;
}

static int gve_validate_req_ring_size(struct gve_priv *priv, u16 new_tx_desc_cnt,
                                      u16 new_rx_desc_cnt)
{
        /* check for valid range */
        if (new_tx_desc_cnt < priv->min_tx_desc_cnt ||
            new_tx_desc_cnt > priv->max_tx_desc_cnt ||
            new_rx_desc_cnt < priv->min_rx_desc_cnt ||
            new_rx_desc_cnt > priv->max_rx_desc_cnt) {
                dev_err(&priv->pdev->dev, "Requested descriptor count out of range\n");
                return -EINVAL;
        }

        if (!is_power_of_2(new_tx_desc_cnt) || !is_power_of_2(new_rx_desc_cnt)) {
                dev_err(&priv->pdev->dev, "Requested descriptor count has to be a power of 2\n");
                return -EINVAL;
        }
        return 0;
}

static int gve_set_ring_sizes_config(struct gve_priv *priv, u16 new_tx_desc_cnt,
                                     u16 new_rx_desc_cnt,
                                     struct gve_tx_alloc_rings_cfg *tx_alloc_cfg,
                                     struct gve_rx_alloc_rings_cfg *rx_alloc_cfg)
{
        if (new_tx_desc_cnt == priv->tx_desc_cnt &&
            new_rx_desc_cnt == priv->rx_desc_cnt)
                return 0;

        if (!priv->modify_ring_size_enabled) {
                dev_err(&priv->pdev->dev, "Modify ring size is not supported.\n");
                return -EOPNOTSUPP;
        }

        if (gve_validate_req_ring_size(priv, new_tx_desc_cnt, new_rx_desc_cnt))
                return -EINVAL;

        tx_alloc_cfg->ring_size = new_tx_desc_cnt;
        rx_alloc_cfg->ring_size = new_rx_desc_cnt;
        return 0;
}

static int gve_set_ringparam(struct net_device *netdev,
                             struct ethtool_ringparam *cmd,
                             struct kernel_ethtool_ringparam *kernel_cmd,
                             struct netlink_ext_ack *extack)
{
        struct gve_tx_alloc_rings_cfg tx_alloc_cfg = {0};
        struct gve_rx_alloc_rings_cfg rx_alloc_cfg = {0};
        struct gve_priv *priv = netdev_priv(netdev);
        int err;

        gve_get_curr_alloc_cfgs(priv, &tx_alloc_cfg, &rx_alloc_cfg);

        err = gve_set_rx_buf_len_config(priv, kernel_cmd->rx_buf_len, extack,
                                        &rx_alloc_cfg);
        if (err)
                return err;

        err = gve_set_hsplit_config(priv, kernel_cmd->tcp_data_split,
                                    &rx_alloc_cfg);
        if (err)
                return err;

        err = gve_set_ring_sizes_config(priv, cmd->tx_pending, cmd->rx_pending,
                                        &tx_alloc_cfg, &rx_alloc_cfg);
        if (err)
                return err;

        if (netif_running(priv->dev)) {
                err = gve_adjust_config(priv, &tx_alloc_cfg, &rx_alloc_cfg);
                if (err)
                        return err;
        } else {
                /* Set ring params for the next up */
                priv->rx_cfg.packet_buffer_size =
                        rx_alloc_cfg.packet_buffer_size;
                priv->header_split_enabled = rx_alloc_cfg.enable_header_split;
                priv->tx_desc_cnt = tx_alloc_cfg.ring_size;
                priv->rx_desc_cnt = rx_alloc_cfg.ring_size;
        }
        return 0;
}

static int gve_user_reset(struct net_device *netdev, u32 *flags)
{
        struct gve_priv *priv = netdev_priv(netdev);

        if (*flags == ETH_RESET_ALL) {
                *flags = 0;
                return gve_reset(priv, true);
        }

        return -EOPNOTSUPP;
}

static int gve_get_tunable(struct net_device *netdev,
                           const struct ethtool_tunable *etuna, void *value)
{
        struct gve_priv *priv = netdev_priv(netdev);

        switch (etuna->id) {
        case ETHTOOL_RX_COPYBREAK:
                *(u32 *)value = priv->rx_copybreak;
                return 0;
        default:
                return -EOPNOTSUPP;
        }
}

static int gve_set_tunable(struct net_device *netdev,
                           const struct ethtool_tunable *etuna,
                           const void *value)
{
        struct gve_priv *priv = netdev_priv(netdev);
        u32 len;

        switch (etuna->id) {
        case ETHTOOL_RX_COPYBREAK:
        {
                u32 max_copybreak = priv->rx_cfg.packet_buffer_size;

                len = *(u32 *)value;
                if (len > max_copybreak)
                        return -EINVAL;
                priv->rx_copybreak = len;
                return 0;
        }
        default:
                return -EOPNOTSUPP;
        }
}

static u32 gve_get_priv_flags(struct net_device *netdev)
{
        struct gve_priv *priv = netdev_priv(netdev);
        u32 ret_flags = 0;

        /* Only 1 flag exists currently: report-stats (BIT(0)), so set that flag. */
        if (priv->ethtool_flags & BIT(0))
                ret_flags |= BIT(0);
        return ret_flags;
}

static int gve_set_priv_flags(struct net_device *netdev, u32 flags)
{
        struct gve_priv *priv = netdev_priv(netdev);
        u64 ori_flags, new_flags;
        int num_tx_queues;

        num_tx_queues = gve_num_tx_queues(priv);
        ori_flags = READ_ONCE(priv->ethtool_flags);
        new_flags = ori_flags;

        /* Only one priv flag exists: report-stats (BIT(0))*/
        if (flags & BIT(0))
                new_flags |= BIT(0);
        else
                new_flags &= ~(BIT(0));
        priv->ethtool_flags = new_flags;
        /* start report-stats timer when user turns report stats on. */
        if (flags & BIT(0)) {
                mod_timer(&priv->stats_report_timer,
                          round_jiffies(jiffies +
                                        msecs_to_jiffies(priv->stats_report_timer_period)));
        }
        /* Zero off gve stats when report-stats turned off and */
        /* delete report stats timer. */
        if (!(flags & BIT(0)) && (ori_flags & BIT(0))) {
                int tx_stats_num = GVE_TX_STATS_REPORT_NUM *
                        num_tx_queues;
                int rx_stats_num = GVE_RX_STATS_REPORT_NUM *
                        priv->rx_cfg.num_queues;

                memset(priv->stats_report->stats, 0, (tx_stats_num + rx_stats_num) *
                                   sizeof(struct stats));
                timer_delete_sync(&priv->stats_report_timer);
        }
        return 0;
}

static int gve_get_link_ksettings(struct net_device *netdev,
                                  struct ethtool_link_ksettings *cmd)
{
        struct gve_priv *priv = netdev_priv(netdev);
        int err = 0;

        if (priv->link_speed == 0)
                err = gve_adminq_report_link_speed(priv);

        cmd->base.speed = priv->link_speed;

        cmd->base.duplex = DUPLEX_FULL;

        return err;
}

static int gve_get_coalesce(struct net_device *netdev,
                            struct ethtool_coalesce *ec,
                            struct kernel_ethtool_coalesce *kernel_ec,
                            struct netlink_ext_ack *extack)
{
        struct gve_priv *priv = netdev_priv(netdev);

        if (gve_is_gqi(priv))
                return -EOPNOTSUPP;
        ec->tx_coalesce_usecs = priv->tx_coalesce_usecs;
        ec->rx_coalesce_usecs = priv->rx_coalesce_usecs;

        return 0;
}

static int gve_set_coalesce(struct net_device *netdev,
                            struct ethtool_coalesce *ec,
                            struct kernel_ethtool_coalesce *kernel_ec,
                            struct netlink_ext_ack *extack)
{
        struct gve_priv *priv = netdev_priv(netdev);
        u32 tx_usecs_orig = priv->tx_coalesce_usecs;
        u32 rx_usecs_orig = priv->rx_coalesce_usecs;
        int idx;

        if (gve_is_gqi(priv))
                return -EOPNOTSUPP;

        if (ec->tx_coalesce_usecs > GVE_MAX_ITR_INTERVAL_DQO ||
            ec->rx_coalesce_usecs > GVE_MAX_ITR_INTERVAL_DQO)
                return -EINVAL;
        priv->tx_coalesce_usecs = ec->tx_coalesce_usecs;
        priv->rx_coalesce_usecs = ec->rx_coalesce_usecs;

        if (tx_usecs_orig != priv->tx_coalesce_usecs) {
                for (idx = 0; idx < priv->tx_cfg.num_queues; idx++) {
                        int ntfy_idx = gve_tx_idx_to_ntfy(priv, idx);
                        struct gve_notify_block *block = &priv->ntfy_blocks[ntfy_idx];

                        gve_set_itr_coalesce_usecs_dqo(priv, block,
                                                       priv->tx_coalesce_usecs);
                }
        }

        if (rx_usecs_orig != priv->rx_coalesce_usecs) {
                for (idx = 0; idx < priv->rx_cfg.num_queues; idx++) {
                        int ntfy_idx = gve_rx_idx_to_ntfy(priv, idx);
                        struct gve_notify_block *block = &priv->ntfy_blocks[ntfy_idx];

                        gve_set_itr_coalesce_usecs_dqo(priv, block,
                                                       priv->rx_coalesce_usecs);
                }
        }

        return 0;
}

static int gve_set_rxnfc(struct net_device *netdev, struct ethtool_rxnfc *cmd)
{
        struct gve_priv *priv = netdev_priv(netdev);
        int err = 0;

        if (!(netdev->features & NETIF_F_NTUPLE))
                return -EOPNOTSUPP;

        switch (cmd->cmd) {
        case ETHTOOL_SRXCLSRLINS:
                err = gve_add_flow_rule(priv, cmd);
                break;
        case ETHTOOL_SRXCLSRLDEL:
                err = gve_del_flow_rule(priv, cmd);
                break;
        default:
                err = -EOPNOTSUPP;
                break;
        }

        return err;
}

static u32 gve_get_rx_ring_count(struct net_device *netdev)
{
        struct gve_priv *priv = netdev_priv(netdev);

        return priv->rx_cfg.num_queues;
}

static int gve_get_rxnfc(struct net_device *netdev, struct ethtool_rxnfc *cmd, u32 *rule_locs)
{
        struct gve_priv *priv = netdev_priv(netdev);
        int err = 0;

        switch (cmd->cmd) {
        case ETHTOOL_GRXCLSRLCNT:
                if (!priv->max_flow_rules)
                        return -EOPNOTSUPP;

                err = gve_adminq_query_flow_rules(priv, GVE_FLOW_RULE_QUERY_STATS, 0);
                if (err)
                        return err;

                cmd->rule_cnt = priv->num_flow_rules;
                cmd->data = priv->max_flow_rules;
                break;
        case ETHTOOL_GRXCLSRULE:
                err = gve_get_flow_rule_entry(priv, cmd);
                break;
        case ETHTOOL_GRXCLSRLALL:
                err = gve_get_flow_rule_ids(priv, cmd, (u32 *)rule_locs);
                break;
        default:
                err = -EOPNOTSUPP;
                break;
        }

        return err;
}

static u32 gve_get_rxfh_key_size(struct net_device *netdev)
{
        struct gve_priv *priv = netdev_priv(netdev);

        return priv->rss_key_size;
}

static u32 gve_get_rxfh_indir_size(struct net_device *netdev)
{
        struct gve_priv *priv = netdev_priv(netdev);

        return priv->rss_lut_size;
}

static void gve_get_rss_config_cache(struct gve_priv *priv,
                                     struct ethtool_rxfh_param *rxfh)
{
        struct gve_rss_config *rss_config = &priv->rss_config;

        rxfh->hfunc = ETH_RSS_HASH_TOP;

        if (rxfh->key) {
                rxfh->key_size = priv->rss_key_size;
                memcpy(rxfh->key, rss_config->hash_key, priv->rss_key_size);
        }

        if (rxfh->indir) {
                rxfh->indir_size = priv->rss_lut_size;
                memcpy(rxfh->indir, rss_config->hash_lut,
                       priv->rss_lut_size * sizeof(*rxfh->indir));
        }
}

static int gve_get_rxfh(struct net_device *netdev, struct ethtool_rxfh_param *rxfh)
{
        struct gve_priv *priv = netdev_priv(netdev);

        if (!priv->rss_key_size || !priv->rss_lut_size)
                return -EOPNOTSUPP;

        if (priv->cache_rss_config) {
                gve_get_rss_config_cache(priv, rxfh);
                return 0;
        }

        return gve_adminq_query_rss_config(priv, rxfh);
}

static void gve_set_rss_config_cache(struct gve_priv *priv,
                                     struct ethtool_rxfh_param *rxfh)
{
        struct gve_rss_config *rss_config = &priv->rss_config;

        if (rxfh->key)
                memcpy(rss_config->hash_key, rxfh->key, priv->rss_key_size);

        if (rxfh->indir)
                memcpy(rss_config->hash_lut, rxfh->indir,
                       priv->rss_lut_size * sizeof(*rxfh->indir));
}

static int gve_set_rxfh(struct net_device *netdev, struct ethtool_rxfh_param *rxfh,
                        struct netlink_ext_ack *extack)
{
        struct gve_priv *priv = netdev_priv(netdev);
        int err;

        if (!priv->rss_key_size || !priv->rss_lut_size)
                return -EOPNOTSUPP;

        err = gve_adminq_configure_rss(priv, rxfh);
        if (err) {
                NL_SET_ERR_MSG_MOD(extack, "Fail to configure RSS config");
                return err;
        }

        if (priv->cache_rss_config)
                gve_set_rss_config_cache(priv, rxfh);

        return 0;
}

static int gve_get_ts_info(struct net_device *netdev,
                           struct kernel_ethtool_ts_info *info)
{
        struct gve_priv *priv = netdev_priv(netdev);

        ethtool_op_get_ts_info(netdev, info);

        if (gve_is_clock_enabled(priv)) {
                info->so_timestamping |= SOF_TIMESTAMPING_RX_HARDWARE |
                                         SOF_TIMESTAMPING_RAW_HARDWARE;

                info->rx_filters |= BIT(HWTSTAMP_FILTER_NONE) |
                                    BIT(HWTSTAMP_FILTER_ALL);

                if (priv->ptp)
                        info->phc_index = ptp_clock_index(priv->ptp->clock);
        }

        return 0;
}

const struct ethtool_ops gve_ethtool_ops = {
        .supported_coalesce_params = ETHTOOL_COALESCE_USECS,
        .supported_ring_params = ETHTOOL_RING_USE_TCP_DATA_SPLIT |
                                 ETHTOOL_RING_USE_RX_BUF_LEN,
        .op_needs_rtnl = ETHTOOL_OP_NEEDS_RTNL_SCHANNELS |
                         ETHTOOL_OP_NEEDS_RTNL_SRINGPARAM |
                         ETHTOOL_OP_NEEDS_RTNL_GLINK,
        .get_drvinfo = gve_get_drvinfo,
        .get_strings = gve_get_strings,
        .get_sset_count = gve_get_sset_count,
        .get_ethtool_stats = gve_get_ethtool_stats,
        .set_msglevel = gve_set_msglevel,
        .get_msglevel = gve_get_msglevel,
        .set_channels = gve_set_channels,
        .get_channels = gve_get_channels,
        .set_rxnfc = gve_set_rxnfc,
        .get_rxnfc = gve_get_rxnfc,
        .get_rx_ring_count = gve_get_rx_ring_count,
        .get_rxfh_indir_size = gve_get_rxfh_indir_size,
        .get_rxfh_key_size = gve_get_rxfh_key_size,
        .get_rxfh = gve_get_rxfh,
        .set_rxfh = gve_set_rxfh,
        .get_link = ethtool_op_get_link,
        .get_coalesce = gve_get_coalesce,
        .set_coalesce = gve_set_coalesce,
        .get_ringparam = gve_get_ringparam,
        .set_ringparam = gve_set_ringparam,
        .reset = gve_user_reset,
        .get_tunable = gve_get_tunable,
        .set_tunable = gve_set_tunable,
        .get_priv_flags = gve_get_priv_flags,
        .set_priv_flags = gve_set_priv_flags,
        .get_link_ksettings = gve_get_link_ksettings,
        .get_ts_info = gve_get_ts_info,
};