pkt_sched: Check .walk and .leaf class handlers

Require qdisc class ops .walk and .leaf for classful qdisc in
register_qdisc(). The checks could be done later insted, but these
ops are really needed and used by most of classful qdiscs.

Signed-off-by: Jarek Poplawski <jarkao2@gmail.com>
Signed-off-by: David S. Miller <davem@davemloft.net>

1 files changed, 1 insertions, 1 deletions

diff --git a/net/sched/sch_api.c b/net/sched/sch_api.c
index 8ed2f5649029..408eea7086aa 100644
--- a/net/sched/sch_api.c
+++ b/net/sched/sch_api.c
@@ -161,7 +161,7 @@ int register_qdisc(struct Qdisc_ops *qops)
         if (qops->cl_ops) {
                 const struct Qdisc_class_ops *cops = qops->cl_ops;
 
-                if (!(cops->get && cops->put))
+                if (!(cops->get && cops->put && cops->walk && cops->leaf))
                         goto out_einval;
 
                 if (cops->tcf_chain && !(cops->bind_tcf && cops->unbind_tcf))