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 29f65632ebc5987d22ca7c8134ccf557e58d102e..2ff80cd04c5ca87d4d4624cb8b5f6479fd469a65 100644 (file)
--- 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