To enhance functionality, we now support reporting statistics through
the netdev-generic netlink (netdev-genl) queue stats interface. However,
this does not extend to all statistics, so a new field, qstat_offset,
has been introduced. This field determines which statistics should be
reported via netdev-genl queue stats.

Given that queue stats are retrieved individually per queue, it's
necessary for the virtnet_get_hw_stats() function to be capable of
fetching statistics for a specific queue.

As the document 
https://docs.kernel.org/next/networking/statistics.html#notes-for-driver-authors

We should not duplicate the stats which get reported via the netlink API in
ethtool. If the stats are for queue stat, that will not be reported by
ethtool -S.

python3 ./tools/net/ynl/cli.py --spec Documentation/netlink/specs/netdev.yaml
    --dump qstats-get --json '{"scope": "queue"}'
[{'ifindex': 2,
  'queue-id': 0,
  'queue-type': 'rx',
  'rx-bytes': 157844011,
  'rx-csum-bad': 0,
  'rx-csum-none': 0,
  'rx-csum-unnecessary': 2195386,
  'rx-hw-drop-overruns': 0,
  'rx-hw-drop-ratelimits': 0,
  'rx-hw-drops': 12964,
  'rx-packets': 598929},
 {'ifindex': 2,
  'queue-id': 0,
  'queue-type': 'tx',
  'tx-bytes': 1938511,
  'tx-csum-none': 0,
  'tx-hw-drop-errors': 0,
  'tx-hw-drop-ratelimits': 0,
  'tx-hw-drops': 0,
  'tx-needs-csum': 61263,
  'tx-packets': 15515}]

Signed-off-by: Xuan Zhuo <xuanz...@linux.alibaba.com>
---
 drivers/net/virtio_net.c | 254 +++++++++++++++++++++++++++++++--------
 1 file changed, 203 insertions(+), 51 deletions(-)

diff --git a/drivers/net/virtio_net.c b/drivers/net/virtio_net.c
index a24cfde30d08..42c38c121a7a 100644
--- a/drivers/net/virtio_net.c
+++ b/drivers/net/virtio_net.c
@@ -24,6 +24,7 @@
 #include <net/xdp.h>
 #include <net/net_failover.h>
 #include <net/netdev_rx_queue.h>
+#include <net/netdev_queues.h>
 
 static int napi_weight = NAPI_POLL_WEIGHT;
 module_param(napi_weight, int, 0444);
@@ -78,6 +79,7 @@ static const unsigned long guest_offloads[] = {
 struct virtnet_stat_desc {
        char desc[ETH_GSTRING_LEN];
        size_t offset;
+       int qstat_offset;
 };
 
 struct virtnet_sq_stats {
@@ -102,12 +104,27 @@ struct virtnet_rq_stats {
        u64_stats_t kicks;
 };
 
-#define VIRTNET_SQ_STAT(name, m) {name, offsetof(struct virtnet_sq_stats, m)}
-#define VIRTNET_RQ_STAT(name, m) {name, offsetof(struct virtnet_rq_stats, m)}
+#define VIRTNET_SQ_STAT(name, m) {name, offsetof(struct virtnet_sq_stats, m), 
-1}
+#define VIRTNET_RQ_STAT(name, m) {name, offsetof(struct virtnet_rq_stats, m), 
-1}
+
+#define VIRTNET_SQ_STAT_QSTAT(name, m)                         \
+       {                                                       \
+               name,                                           \
+               offsetof(struct virtnet_sq_stats, m),           \
+               offsetof(struct netdev_queue_stats_tx, m),      \
+       }
+
+#define VIRTNET_RQ_STAT_QSTAT(name, m)                         \
+       {                                                       \
+               name,                                           \
+               offsetof(struct virtnet_rq_stats, m),           \
+               offsetof(struct netdev_queue_stats_rx, m),      \
+       }
 
 static const struct virtnet_stat_desc virtnet_sq_stats_desc[] = {
-       VIRTNET_SQ_STAT("packets",      packets),
-       VIRTNET_SQ_STAT("bytes",        bytes),
+       VIRTNET_SQ_STAT_QSTAT("packets", packets),
+       VIRTNET_SQ_STAT_QSTAT("bytes",   bytes),
+
        VIRTNET_SQ_STAT("xdp_tx",       xdp_tx),
        VIRTNET_SQ_STAT("xdp_tx_drops", xdp_tx_drops),
        VIRTNET_SQ_STAT("kicks",        kicks),
@@ -115,8 +132,9 @@ static const struct virtnet_stat_desc 
virtnet_sq_stats_desc[] = {
 };
 
 static const struct virtnet_stat_desc virtnet_rq_stats_desc[] = {
-       VIRTNET_RQ_STAT("packets",       packets),
-       VIRTNET_RQ_STAT("bytes",         bytes),
+       VIRTNET_RQ_STAT_QSTAT("packets", packets),
+       VIRTNET_RQ_STAT_QSTAT("bytes",   bytes),
+
        VIRTNET_RQ_STAT("drops",         drops),
        VIRTNET_RQ_STAT("xdp_packets",   xdp_packets),
        VIRTNET_RQ_STAT("xdp_tx",        xdp_tx),
@@ -126,13 +144,27 @@ static const struct virtnet_stat_desc 
virtnet_rq_stats_desc[] = {
 };
 
 #define VIRTNET_STATS_DESC_CQ(name) \
-       {#name, offsetof(struct virtio_net_stats_cvq, name)}
+       {#name, offsetof(struct virtio_net_stats_cvq, name), -1}
 
 #define VIRTNET_STATS_DESC_RX(class, name) \
-       {#name, offsetof(struct virtio_net_stats_rx_ ## class, rx_ ## name)}
+       {#name, offsetof(struct virtio_net_stats_rx_ ## class, rx_ ## name), -1}
 
 #define VIRTNET_STATS_DESC_TX(class, name) \
-       {#name, offsetof(struct virtio_net_stats_tx_ ## class, tx_ ## name)}
+       {#name, offsetof(struct virtio_net_stats_tx_ ## class, tx_ ## name), -1}
+
+#define VIRTNET_STATS_DESC_RX_QSTAT(class, name, qstat_field)                  
\
+       {                                                                       
\
+               #name,                                                          
\
+               offsetof(struct virtio_net_stats_rx_ ## class, rx_ ## name),    
\
+               offsetof(struct netdev_queue_stats_rx, qstat_field),            
\
+       }
+
+#define VIRTNET_STATS_DESC_TX_QSTAT(class, name, qstat_field)                  
\
+       {                                                                       
\
+               #name,                                                          
\
+               offsetof(struct virtio_net_stats_tx_ ## class, tx_ ## name),    
\
+               offsetof(struct netdev_queue_stats_tx, qstat_field),            
\
+       }
 
 static const struct virtnet_stat_desc virtnet_stats_cvq_desc[] = {
        VIRTNET_STATS_DESC_CQ(command_num),
@@ -146,8 +178,8 @@ static const struct virtnet_stat_desc 
virtnet_stats_rx_basic_desc[] = {
        VIRTNET_STATS_DESC_RX(basic, notifications),
        VIRTNET_STATS_DESC_RX(basic, interrupts),
 
-       VIRTNET_STATS_DESC_RX(basic, drops),
-       VIRTNET_STATS_DESC_RX(basic, drop_overruns),
+       VIRTNET_STATS_DESC_RX_QSTAT(basic, drops,         hw_drops),
+       VIRTNET_STATS_DESC_RX_QSTAT(basic, drop_overruns, hw_drop_overruns),
 };
 
 static const struct virtnet_stat_desc virtnet_stats_tx_basic_desc[] = {
@@ -157,46 +189,47 @@ static const struct virtnet_stat_desc 
virtnet_stats_tx_basic_desc[] = {
        VIRTNET_STATS_DESC_TX(basic, notifications),
        VIRTNET_STATS_DESC_TX(basic, interrupts),
 
-       VIRTNET_STATS_DESC_TX(basic, drops),
-       VIRTNET_STATS_DESC_TX(basic, drop_malformed),
+       VIRTNET_STATS_DESC_TX_QSTAT(basic, drops,          hw_drops),
+       VIRTNET_STATS_DESC_TX_QSTAT(basic, drop_malformed, hw_drop_errors),
 };
 
 static const struct virtnet_stat_desc virtnet_stats_rx_csum_desc[] = {
-       VIRTNET_STATS_DESC_RX(csum, csum_valid),
-       VIRTNET_STATS_DESC_RX(csum, needs_csum),
+       VIRTNET_STATS_DESC_RX_QSTAT(csum, csum_valid, csum_unnecessary),
+       VIRTNET_STATS_DESC_RX_QSTAT(csum, csum_none,  csum_none),
+       VIRTNET_STATS_DESC_RX_QSTAT(csum, csum_bad,   csum_bad),
 
-       VIRTNET_STATS_DESC_RX(csum, csum_none),
-       VIRTNET_STATS_DESC_RX(csum, csum_bad),
+       VIRTNET_STATS_DESC_RX(csum, needs_csum),
 };
 
 static const struct virtnet_stat_desc virtnet_stats_tx_csum_desc[] = {
-       VIRTNET_STATS_DESC_TX(csum, needs_csum),
-       VIRTNET_STATS_DESC_TX(csum, csum_none),
+       VIRTNET_STATS_DESC_TX_QSTAT(csum, csum_none,  csum_none),
+       VIRTNET_STATS_DESC_TX_QSTAT(csum, needs_csum, needs_csum),
 };
 
 static const struct virtnet_stat_desc virtnet_stats_rx_gso_desc[] = {
-       VIRTNET_STATS_DESC_RX(gso, gso_packets),
-       VIRTNET_STATS_DESC_RX(gso, gso_bytes),
-       VIRTNET_STATS_DESC_RX(gso, gso_packets_coalesced),
-       VIRTNET_STATS_DESC_RX(gso, gso_bytes_coalesced),
+       VIRTNET_STATS_DESC_RX_QSTAT(gso, gso_packets,           hw_gro_packets),
+       VIRTNET_STATS_DESC_RX_QSTAT(gso, gso_bytes,             hw_gro_bytes),
+       VIRTNET_STATS_DESC_RX_QSTAT(gso, gso_packets_coalesced, 
hw_gro_wire_packets),
+       VIRTNET_STATS_DESC_RX_QSTAT(gso, gso_bytes_coalesced,   
hw_gro_wire_bytes),
 };
 
 static const struct virtnet_stat_desc virtnet_stats_tx_gso_desc[] = {
-       VIRTNET_STATS_DESC_TX(gso, gso_packets),
-       VIRTNET_STATS_DESC_TX(gso, gso_bytes),
-       VIRTNET_STATS_DESC_TX(gso, gso_segments),
-       VIRTNET_STATS_DESC_TX(gso, gso_segments_bytes),
+       VIRTNET_STATS_DESC_TX_QSTAT(gso, gso_packets,        hw_gso_packets),
+       VIRTNET_STATS_DESC_TX_QSTAT(gso, gso_bytes,          hw_gso_bytes),
+       VIRTNET_STATS_DESC_TX_QSTAT(gso, gso_segments,       
hw_gso_wire_packets),
+       VIRTNET_STATS_DESC_TX_QSTAT(gso, gso_segments_bytes, hw_gso_wire_bytes),
+
        VIRTNET_STATS_DESC_TX(gso, gso_packets_noseg),
        VIRTNET_STATS_DESC_TX(gso, gso_bytes_noseg),
 };
 
 static const struct virtnet_stat_desc virtnet_stats_rx_speed_desc[] = {
-       VIRTNET_STATS_DESC_RX(speed, ratelimit_packets),
+       VIRTNET_STATS_DESC_RX_QSTAT(speed, ratelimit_packets, 
hw_drop_ratelimits),
        VIRTNET_STATS_DESC_RX(speed, ratelimit_bytes),
 };
 
 static const struct virtnet_stat_desc virtnet_stats_tx_speed_desc[] = {
-       VIRTNET_STATS_DESC_TX(speed, ratelimit_packets),
+       VIRTNET_STATS_DESC_TX_QSTAT(speed, ratelimit_packets, 
hw_drop_ratelimits),
        VIRTNET_STATS_DESC_TX(speed, ratelimit_bytes),
 };
 
@@ -3442,6 +3475,9 @@ static void virtnet_get_stats_string(struct virtnet_info 
*vi, int type, int qid,
                for (j = 0; j < m->num; ++j) {
                        desc = m->desc[j].desc;
 
+                       if (m->desc[j].qstat_offset >= 0)
+                               continue;
+
                        if (type == VIRTNET_Q_TYPE_CQ) {
                                ethtool_sprintf(&p, "cq%s_%s", hw, desc);
                                continue;
@@ -3458,6 +3494,9 @@ static void virtnet_get_stats_string(struct virtnet_info 
*vi, int type, int qid,
 }
 
 struct virtnet_stats_ctx {
+       /* the stats are write to qstats or ethtool -S */
+       bool to_qstat;
+
        u32 desc_num[3];
 
        u32 bitmap[3];
@@ -3469,12 +3508,13 @@ struct virtnet_stats_ctx {
 
 static void virtnet_stats_ctx_init(struct virtnet_info *vi,
                                   struct virtnet_stats_ctx *ctx,
-                                  u64 *data)
+                                  u64 *data, bool to_qstat)
 {
        struct virtnet_stats_map *m;
-       int i;
+       int i, num, j;
 
        ctx->data = data;
+       ctx->to_qstat = to_qstat;
 
        for (i = 0; i < ARRAY_SIZE(virtio_net_stats_map); ++i) {
                m = &virtio_net_stats_map[i];
@@ -3482,8 +3522,17 @@ static void virtnet_stats_ctx_init(struct virtnet_info 
*vi,
                if (!virtnet_stats_supported(vi, m))
                        continue;
 
+               num = 0;
+               for (j = 0; j < m->num; ++j) {
+                       if ((m->desc[j].qstat_offset >= 0) == ctx->to_qstat)
+                               ++num;
+               }
+
+               if (!num)
+                       continue;
+
                ctx->bitmap[m->queue_type]   |= m->stat_type;
-               ctx->desc_num[m->queue_type] += m->num;
+               ctx->desc_num[m->queue_type] += num;
                ctx->size[m->queue_type]     += m->len;
        }
 }
@@ -3531,7 +3580,7 @@ static void virtnet_fill_total_fields(struct virtnet_info 
*vi,
        stats_sum_queue(data, num_tx, first_tx_q, vi->curr_queue_pairs);
 }
 
-/* virtnet_fill_stats - copy the stats to ethtool -S
+/* virtnet_fill_stats - copy the stats to qstats or ethtool -S
  * The stats source is the device or the driver.
  *
  * @vi: virtio net info
@@ -3550,8 +3599,9 @@ static void virtnet_fill_stats(struct virtnet_info *vi, 
u32 qid,
        const struct virtnet_stat_desc *desc;
        struct virtnet_stats_map *m;
        const u64_stats_t *v_stat;
+       u64 offset, value;
        const __le64 *v;
-       u64 offset;
+       bool skip;
        int i, j;
 
        num_cq = ctx->desc_num[VIRTNET_Q_TYPE_CQ];
@@ -3577,24 +3627,46 @@ static void virtnet_fill_stats(struct virtnet_info *vi, 
u32 qid,
                if (!virtnet_stats_supported(vi, m))
                        continue;
 
+               skip = false;
+
                /* Checking whether this "m" matches "base" or not. */
-               if (from_driver != m->from_driver || type != m->reply_type) {
-                       offset += m->num;
-                       continue;
-               }
+               if (from_driver != m->from_driver || type != m->reply_type)
+                       skip = true;
 
                for (j = 0; j < m->num; ++j) {
                        desc = &m->desc[j];
+
+                       if ((desc->qstat_offset >= 0) != ctx->to_qstat)
+                               continue;
+
+                       if (skip) {
+                               /* update write offset of the ctx->data. */
+                               ++offset;
+                               continue;
+                       }
+
                        if (!from_driver) {
                                v = (const __le64 *)(base + desc->offset);
-                               ctx->data[offset + j] = le64_to_cpu(*v);
+                               value = le64_to_cpu(*v);
                        } else {
                                v_stat = (const u64_stats_t *)(base + 
desc->offset);
-                               ctx->data[offset + j] = u64_stats_read(v_stat);
+                               value = u64_stats_read(v_stat);
+                       }
+
+                       if (ctx->to_qstat) {
+                               /* store to the queue stats structure */
+                               if (desc->qstat_offset >= 0) {
+                                       offset = desc->qstat_offset / 
sizeof(*ctx->data);
+                                       ctx->data[offset] = value;
+                               }
+                       } else {
+                               /* store to the ethtool -S data area */
+                               ctx->data[offset++] = value;
                        }
                }
 
-               break;
+               if (!skip)
+                       break;
        }
 }
 
@@ -3648,21 +3720,33 @@ static void virtnet_make_stat_req(struct virtnet_info 
*vi,
        *idx += 1;
 }
 
+/* qid: -1: get stats of all vq.
+ *     > 0: get the stats for the special vq. This must not be cvq.
+ */
 static int virtnet_get_hw_stats(struct virtnet_info *vi,
-                               struct virtnet_stats_ctx *ctx)
+                               struct virtnet_stats_ctx *ctx, int qid)
 {
+       int qnum, i, j, res_size, qtype, last_vq, first_vq;
        struct virtio_net_ctrl_queue_stats *req;
-       int qnum, i, j, res_size, qtype, last_vq;
+       bool enable_cvq;
        void *reply;
 
        if (!virtio_has_feature(vi->vdev, VIRTIO_NET_F_DEVICE_STATS))
                return 0;
 
-       last_vq = vi->curr_queue_pairs * 2 - 1;
+       if (qid == -1) {
+               last_vq = vi->curr_queue_pairs * 2 - 1;
+               first_vq = 0;
+               enable_cvq = true;
+       } else {
+               last_vq = qid;
+               first_vq = qid;
+               enable_cvq = false;
+       }
 
        qnum = 0;
        res_size = 0;
-       for (i = 0; i <= last_vq ; ++i) {
+       for (i = first_vq; i <= last_vq ; ++i) {
                qtype = vq_type(vi, i);
                if (ctx->bitmap[qtype]) {
                        ++qnum;
@@ -3670,7 +3754,7 @@ static int virtnet_get_hw_stats(struct virtnet_info *vi,
                }
        }
 
-       if (ctx->bitmap[VIRTNET_Q_TYPE_CQ]) {
+       if (enable_cvq && ctx->bitmap[VIRTNET_Q_TYPE_CQ]) {
                res_size += ctx->size[VIRTNET_Q_TYPE_CQ];
                qnum += 1;
        }
@@ -3686,10 +3770,11 @@ static int virtnet_get_hw_stats(struct virtnet_info *vi,
        }
 
        j = 0;
-       for (i = 0; i <= last_vq ; ++i)
+       for (i = first_vq; i <= last_vq ; ++i)
                virtnet_make_stat_req(vi, ctx, req, i, &j);
 
-       virtnet_make_stat_req(vi, ctx, req, vi->max_queue_pairs * 2, &j);
+       if (enable_cvq)
+               virtnet_make_stat_req(vi, ctx, req, vi->max_queue_pairs * 2, 
&j);
 
        return __virtnet_get_hw_stats(vi, ctx, req, sizeof(*req) * j, reply, 
res_size);
 }
@@ -3742,7 +3827,7 @@ static int virtnet_get_sset_count(struct net_device *dev, 
int sset)
                        }
                }
 
-               virtnet_stats_ctx_init(vi, &ctx, NULL);
+               virtnet_stats_ctx_init(vi, &ctx, NULL, false);
 
                pair_count = ctx.desc_num[VIRTNET_Q_TYPE_RX] + 
ctx.desc_num[VIRTNET_Q_TYPE_TX];
 
@@ -3761,8 +3846,8 @@ static void virtnet_get_ethtool_stats(struct net_device 
*dev,
        unsigned int start, i;
        const u8 *stats_base;
 
-       virtnet_stats_ctx_init(vi, &ctx, data);
-       if (virtnet_get_hw_stats(vi, &ctx))
+       virtnet_stats_ctx_init(vi, &ctx, data, false);
+       if (virtnet_get_hw_stats(vi, &ctx, -1))
                dev_warn(&vi->dev->dev, "Failed to get hw stats.\n");
 
        for (i = 0; i < vi->curr_queue_pairs; i++) {
@@ -4301,6 +4386,72 @@ static const struct ethtool_ops virtnet_ethtool_ops = {
        .set_rxnfc = virtnet_set_rxnfc,
 };
 
+static void virtnet_get_queue_stats_rx(struct net_device *dev, int i,
+                                      struct netdev_queue_stats_rx *stats)
+{
+       struct virtnet_info *vi = netdev_priv(dev);
+       struct receive_queue *rq = &vi->rq[i];
+       struct virtnet_stats_ctx ctx = {0};
+
+       virtnet_stats_ctx_init(vi, &ctx, (void *)stats, true);
+
+       virtnet_get_hw_stats(vi, &ctx, i * 2);
+       virtnet_fill_stats(vi, i * 2, &ctx, (void *)&rq->stats, true, 0);
+}
+
+static void virtnet_get_queue_stats_tx(struct net_device *dev, int i,
+                                      struct netdev_queue_stats_tx *stats)
+{
+       struct virtnet_info *vi = netdev_priv(dev);
+       struct send_queue *sq = &vi->sq[i];
+       struct virtnet_stats_ctx ctx = {0};
+
+       virtnet_stats_ctx_init(vi, &ctx, (void *)stats, true);
+
+       virtnet_get_hw_stats(vi, &ctx, i * 2 + 1);
+       virtnet_fill_stats(vi, i * 2 + 1, &ctx, (void *)&sq->stats, true, 0);
+}
+
+static void virtnet_get_base_stats(struct net_device *dev,
+                                  struct netdev_queue_stats_rx *rx,
+                                  struct netdev_queue_stats_tx *tx)
+{
+       /* The queue stats of the virtio-net will not be reset. So here we
+        * return 0.
+        */
+       rx->bytes = 0;
+       rx->packets = 0;
+       rx->alloc_fail = 0;
+       rx->hw_drops = 0;
+       rx->hw_drop_overruns = 0;
+       rx->csum_unnecessary = 0;
+       rx->csum_none = 0;
+       rx->csum_bad = 0;
+       rx->hw_gro_packets = 0;
+       rx->hw_gro_bytes = 0;
+       rx->hw_gro_wire_packets = 0;
+       rx->hw_gro_wire_bytes = 0;
+       rx->hw_drop_ratelimits = 0;
+
+       tx->bytes = 0;
+       tx->packets = 0;
+       tx->hw_drops = 0;
+       tx->hw_drop_errors = 0;
+       tx->csum_none = 0;
+       tx->needs_csum = 0;
+       tx->hw_gso_packets = 0;
+       tx->hw_gso_bytes = 0;
+       tx->hw_gso_wire_packets = 0;
+       tx->hw_gso_wire_bytes = 0;
+       tx->hw_drop_ratelimits = 0;
+}
+
+static const struct netdev_stat_ops virtnet_stat_ops = {
+       .get_queue_stats_rx     = virtnet_get_queue_stats_rx,
+       .get_queue_stats_tx     = virtnet_get_queue_stats_tx,
+       .get_base_stats         = virtnet_get_base_stats,
+};
+
 static void virtnet_freeze_down(struct virtio_device *vdev)
 {
        struct virtnet_info *vi = vdev->priv;
@@ -5060,6 +5211,7 @@ static int virtnet_probe(struct virtio_device *vdev)
        dev->priv_flags |= IFF_UNICAST_FLT | IFF_LIVE_ADDR_CHANGE |
                           IFF_TX_SKB_NO_LINEAR;
        dev->netdev_ops = &virtnet_netdev;
+       dev->stat_ops = &virtnet_stat_ops;
        dev->features = NETIF_F_HIGHDMA;
 
        dev->ethtool_ops = &virtnet_ethtool_ops;
-- 
2.32.0.3.g01195cf9f


Reply via email to