From: Zhengchao Shao <shaozhengchao@huawei.com>
Date: Wed, 21 Sep 2022 02:41:04 +0000 (+0800)
Subject: net/sched: sch_api: add helper for tc qdisc walker stats dump
X-Git-Url: http://git.maquefel.me/?a=commitdiff_plain;h=d7a68e564e29bcd1c70d76d9e2f65591e6183f46;p=linux.git

net/sched: sch_api: add helper for tc qdisc walker stats dump

The walk implementation of most qdisc class modules is basically the
same. That is, the values of count and skip are checked first. If
count is greater than or equal to skip, the registered fn function is
executed. Otherwise, increase the value of count. So we can reconstruct
them.

Signed-off-by: Zhengchao Shao <shaozhengchao@huawei.com>
Acked-by: Toke Høiland-Jørgensen <toke@redhat.com>
Signed-off-by: Jakub Kicinski <kuba@kernel.org>
---

diff --git a/include/net/pkt_sched.h b/include/net/pkt_sched.h
index 29f65632ebc59..2ff80cd04c5ca 100644
--- a/include/net/pkt_sched.h
+++ b/include/net/pkt_sched.h
@@ -222,4 +222,17 @@ static inline struct tc_skb_cb *tc_skb_cb(const struct sk_buff *skb)
 	return cb;
 }
 
+static inline bool tc_qdisc_stats_dump(struct Qdisc *sch,
+				       unsigned long cl,
+				       struct qdisc_walker *arg)
+{
+	if (arg->count >= arg->skip && arg->fn(sch, cl, arg) < 0) {
+		arg->stop = 1;
+		return false;
+	}
+
+	arg->count++;
+	return true;
+}
+
 #endif