pkt_sched: sch_htb: Replace HTB_ACCNT() macro with inlines
[linux-2.6.git] / net / sched / sch_htb.c
1 /*
2  * net/sched/sch_htb.c  Hierarchical token bucket, feed tree version
3  *
4  *              This program is free software; you can redistribute it and/or
5  *              modify it under the terms of the GNU General Public License
6  *              as published by the Free Software Foundation; either version
7  *              2 of the License, or (at your option) any later version.
8  *
9  * Authors:     Martin Devera, <devik@cdi.cz>
10  *
11  * Credits (in time order) for older HTB versions:
12  *              Stef Coene <stef.coene@docum.org>
13  *                      HTB support at LARTC mailing list
14  *              Ondrej Kraus, <krauso@barr.cz>
15  *                      found missing INIT_QDISC(htb)
16  *              Vladimir Smelhaus, Aamer Akhter, Bert Hubert
17  *                      helped a lot to locate nasty class stall bug
18  *              Andi Kleen, Jamal Hadi, Bert Hubert
19  *                      code review and helpful comments on shaping
20  *              Tomasz Wrona, <tw@eter.tym.pl>
21  *                      created test case so that I was able to fix nasty bug
22  *              Wilfried Weissmann
23  *                      spotted bug in dequeue code and helped with fix
24  *              Jiri Fojtasek
25  *                      fixed requeue routine
26  *              and many others. thanks.
27  */
28 #include <linux/module.h>
29 #include <linux/moduleparam.h>
30 #include <linux/types.h>
31 #include <linux/kernel.h>
32 #include <linux/string.h>
33 #include <linux/errno.h>
34 #include <linux/skbuff.h>
35 #include <linux/list.h>
36 #include <linux/compiler.h>
37 #include <linux/rbtree.h>
38 #include <net/netlink.h>
39 #include <net/pkt_sched.h>
40
41 /* HTB algorithm.
42     Author: devik@cdi.cz
43     ========================================================================
44     HTB is like TBF with multiple classes. It is also similar to CBQ because
45     it allows to assign priority to each class in hierarchy.
46     In fact it is another implementation of Floyd's formal sharing.
47
48     Levels:
49     Each class is assigned level. Leaf has ALWAYS level 0 and root
50     classes have level TC_HTB_MAXDEPTH-1. Interior nodes has level
51     one less than their parent.
52 */
53
54 static int htb_hysteresis __read_mostly = 0; /* whether to use mode hysteresis for speedup */
55 #define HTB_VER 0x30011         /* major must be matched with number suplied by TC as version */
56
57 #if HTB_VER >> 16 != TC_HTB_PROTOVER
58 #error "Mismatched sch_htb.c and pkt_sch.h"
59 #endif
60
61 /* Module parameter and sysfs export */
62 module_param    (htb_hysteresis, int, 0640);
63 MODULE_PARM_DESC(htb_hysteresis, "Hysteresis mode, less CPU load, less accurate");
64
65 /* used internaly to keep status of single class */
66 enum htb_cmode {
67         HTB_CANT_SEND,          /* class can't send and can't borrow */
68         HTB_MAY_BORROW,         /* class can't send but may borrow */
69         HTB_CAN_SEND            /* class can send */
70 };
71
72 /* interior & leaf nodes; props specific to leaves are marked L: */
73 struct htb_class {
74         struct Qdisc_class_common common;
75         /* general class parameters */
76         struct gnet_stats_basic bstats;
77         struct gnet_stats_queue qstats;
78         struct gnet_stats_rate_est rate_est;
79         struct tc_htb_xstats xstats;    /* our special stats */
80         int refcnt;             /* usage count of this class */
81
82         /* topology */
83         int level;              /* our level (see above) */
84         unsigned int children;
85         struct htb_class *parent;       /* parent class */
86
87         int prio;               /* these two are used only by leaves... */
88         int quantum;            /* but stored for parent-to-leaf return */
89
90         union {
91                 struct htb_class_leaf {
92                         struct Qdisc *q;
93                         int deficit[TC_HTB_MAXDEPTH];
94                         struct list_head drop_list;
95                 } leaf;
96                 struct htb_class_inner {
97                         struct rb_root feed[TC_HTB_NUMPRIO];    /* feed trees */
98                         struct rb_node *ptr[TC_HTB_NUMPRIO];    /* current class ptr */
99                         /* When class changes from state 1->2 and disconnects from
100                            parent's feed then we lost ptr value and start from the
101                            first child again. Here we store classid of the
102                            last valid ptr (used when ptr is NULL). */
103                         u32 last_ptr_id[TC_HTB_NUMPRIO];
104                 } inner;
105         } un;
106         struct rb_node node[TC_HTB_NUMPRIO];    /* node for self or feed tree */
107         struct rb_node pq_node; /* node for event queue */
108         psched_time_t pq_key;
109
110         int prio_activity;      /* for which prios are we active */
111         enum htb_cmode cmode;   /* current mode of the class */
112
113         /* class attached filters */
114         struct tcf_proto *filter_list;
115         int filter_cnt;
116
117         int warned;             /* only one warning about non work conserving .. */
118
119         /* token bucket parameters */
120         struct qdisc_rate_table *rate;  /* rate table of the class itself */
121         struct qdisc_rate_table *ceil;  /* ceiling rate (limits borrows too) */
122         long buffer, cbuffer;   /* token bucket depth/rate */
123         psched_tdiff_t mbuffer; /* max wait time */
124         long tokens, ctokens;   /* current number of tokens */
125         psched_time_t t_c;      /* checkpoint time */
126 };
127
128 struct htb_sched {
129         struct Qdisc_class_hash clhash;
130         struct list_head drops[TC_HTB_NUMPRIO];/* active leaves (for drops) */
131
132         /* self list - roots of self generating tree */
133         struct rb_root row[TC_HTB_MAXDEPTH][TC_HTB_NUMPRIO];
134         int row_mask[TC_HTB_MAXDEPTH];
135         struct rb_node *ptr[TC_HTB_MAXDEPTH][TC_HTB_NUMPRIO];
136         u32 last_ptr_id[TC_HTB_MAXDEPTH][TC_HTB_NUMPRIO];
137
138         /* self wait list - roots of wait PQs per row */
139         struct rb_root wait_pq[TC_HTB_MAXDEPTH];
140
141         /* time of nearest event per level (row) */
142         psched_time_t near_ev_cache[TC_HTB_MAXDEPTH];
143
144         int defcls;             /* class where unclassified flows go to */
145
146         /* filters for qdisc itself */
147         struct tcf_proto *filter_list;
148
149         int rate2quantum;       /* quant = rate / rate2quantum */
150         psched_time_t now;      /* cached dequeue time */
151         struct qdisc_watchdog watchdog;
152
153         /* non shaped skbs; let them go directly thru */
154         struct sk_buff_head direct_queue;
155         int direct_qlen;        /* max qlen of above */
156
157         long direct_pkts;
158 };
159
160 /* find class in global hash table using given handle */
161 static inline struct htb_class *htb_find(u32 handle, struct Qdisc *sch)
162 {
163         struct htb_sched *q = qdisc_priv(sch);
164         struct Qdisc_class_common *clc;
165
166         clc = qdisc_class_find(&q->clhash, handle);
167         if (clc == NULL)
168                 return NULL;
169         return container_of(clc, struct htb_class, common);
170 }
171
172 /**
173  * htb_classify - classify a packet into class
174  *
175  * It returns NULL if the packet should be dropped or -1 if the packet
176  * should be passed directly thru. In all other cases leaf class is returned.
177  * We allow direct class selection by classid in priority. The we examine
178  * filters in qdisc and in inner nodes (if higher filter points to the inner
179  * node). If we end up with classid MAJOR:0 we enqueue the skb into special
180  * internal fifo (direct). These packets then go directly thru. If we still
181  * have no valid leaf we try to use MAJOR:default leaf. It still unsuccessfull
182  * then finish and return direct queue.
183  */
184 #define HTB_DIRECT (struct htb_class*)-1
185
186 static struct htb_class *htb_classify(struct sk_buff *skb, struct Qdisc *sch,
187                                       int *qerr)
188 {
189         struct htb_sched *q = qdisc_priv(sch);
190         struct htb_class *cl;
191         struct tcf_result res;
192         struct tcf_proto *tcf;
193         int result;
194
195         /* allow to select class by setting skb->priority to valid classid;
196            note that nfmark can be used too by attaching filter fw with no
197            rules in it */
198         if (skb->priority == sch->handle)
199                 return HTB_DIRECT;      /* X:0 (direct flow) selected */
200         if ((cl = htb_find(skb->priority, sch)) != NULL && cl->level == 0)
201                 return cl;
202
203         *qerr = NET_XMIT_SUCCESS | __NET_XMIT_BYPASS;
204         tcf = q->filter_list;
205         while (tcf && (result = tc_classify(skb, tcf, &res)) >= 0) {
206 #ifdef CONFIG_NET_CLS_ACT
207                 switch (result) {
208                 case TC_ACT_QUEUED:
209                 case TC_ACT_STOLEN:
210                         *qerr = NET_XMIT_SUCCESS | __NET_XMIT_STOLEN;
211                 case TC_ACT_SHOT:
212                         return NULL;
213                 }
214 #endif
215                 if ((cl = (void *)res.class) == NULL) {
216                         if (res.classid == sch->handle)
217                                 return HTB_DIRECT;      /* X:0 (direct flow) */
218                         if ((cl = htb_find(res.classid, sch)) == NULL)
219                                 break;  /* filter selected invalid classid */
220                 }
221                 if (!cl->level)
222                         return cl;      /* we hit leaf; return it */
223
224                 /* we have got inner class; apply inner filter chain */
225                 tcf = cl->filter_list;
226         }
227         /* classification failed; try to use default class */
228         cl = htb_find(TC_H_MAKE(TC_H_MAJ(sch->handle), q->defcls), sch);
229         if (!cl || cl->level)
230                 return HTB_DIRECT;      /* bad default .. this is safe bet */
231         return cl;
232 }
233
234 /**
235  * htb_add_to_id_tree - adds class to the round robin list
236  *
237  * Routine adds class to the list (actually tree) sorted by classid.
238  * Make sure that class is not already on such list for given prio.
239  */
240 static void htb_add_to_id_tree(struct rb_root *root,
241                                struct htb_class *cl, int prio)
242 {
243         struct rb_node **p = &root->rb_node, *parent = NULL;
244
245         while (*p) {
246                 struct htb_class *c;
247                 parent = *p;
248                 c = rb_entry(parent, struct htb_class, node[prio]);
249
250                 if (cl->common.classid > c->common.classid)
251                         p = &parent->rb_right;
252                 else
253                         p = &parent->rb_left;
254         }
255         rb_link_node(&cl->node[prio], parent, p);
256         rb_insert_color(&cl->node[prio], root);
257 }
258
259 /**
260  * htb_add_to_wait_tree - adds class to the event queue with delay
261  *
262  * The class is added to priority event queue to indicate that class will
263  * change its mode in cl->pq_key microseconds. Make sure that class is not
264  * already in the queue.
265  */
266 static void htb_add_to_wait_tree(struct htb_sched *q,
267                                  struct htb_class *cl, long delay)
268 {
269         struct rb_node **p = &q->wait_pq[cl->level].rb_node, *parent = NULL;
270
271         cl->pq_key = q->now + delay;
272         if (cl->pq_key == q->now)
273                 cl->pq_key++;
274
275         /* update the nearest event cache */
276         if (q->near_ev_cache[cl->level] > cl->pq_key)
277                 q->near_ev_cache[cl->level] = cl->pq_key;
278
279         while (*p) {
280                 struct htb_class *c;
281                 parent = *p;
282                 c = rb_entry(parent, struct htb_class, pq_node);
283                 if (cl->pq_key >= c->pq_key)
284                         p = &parent->rb_right;
285                 else
286                         p = &parent->rb_left;
287         }
288         rb_link_node(&cl->pq_node, parent, p);
289         rb_insert_color(&cl->pq_node, &q->wait_pq[cl->level]);
290 }
291
292 /**
293  * htb_next_rb_node - finds next node in binary tree
294  *
295  * When we are past last key we return NULL.
296  * Average complexity is 2 steps per call.
297  */
298 static inline void htb_next_rb_node(struct rb_node **n)
299 {
300         *n = rb_next(*n);
301 }
302
303 /**
304  * htb_add_class_to_row - add class to its row
305  *
306  * The class is added to row at priorities marked in mask.
307  * It does nothing if mask == 0.
308  */
309 static inline void htb_add_class_to_row(struct htb_sched *q,
310                                         struct htb_class *cl, int mask)
311 {
312         q->row_mask[cl->level] |= mask;
313         while (mask) {
314                 int prio = ffz(~mask);
315                 mask &= ~(1 << prio);
316                 htb_add_to_id_tree(q->row[cl->level] + prio, cl, prio);
317         }
318 }
319
320 /* If this triggers, it is a bug in this code, but it need not be fatal */
321 static void htb_safe_rb_erase(struct rb_node *rb, struct rb_root *root)
322 {
323         if (RB_EMPTY_NODE(rb)) {
324                 WARN_ON(1);
325         } else {
326                 rb_erase(rb, root);
327                 RB_CLEAR_NODE(rb);
328         }
329 }
330
331
332 /**
333  * htb_remove_class_from_row - removes class from its row
334  *
335  * The class is removed from row at priorities marked in mask.
336  * It does nothing if mask == 0.
337  */
338 static inline void htb_remove_class_from_row(struct htb_sched *q,
339                                                  struct htb_class *cl, int mask)
340 {
341         int m = 0;
342
343         while (mask) {
344                 int prio = ffz(~mask);
345
346                 mask &= ~(1 << prio);
347                 if (q->ptr[cl->level][prio] == cl->node + prio)
348                         htb_next_rb_node(q->ptr[cl->level] + prio);
349
350                 htb_safe_rb_erase(cl->node + prio, q->row[cl->level] + prio);
351                 if (!q->row[cl->level][prio].rb_node)
352                         m |= 1 << prio;
353         }
354         q->row_mask[cl->level] &= ~m;
355 }
356
357 /**
358  * htb_activate_prios - creates active classe's feed chain
359  *
360  * The class is connected to ancestors and/or appropriate rows
361  * for priorities it is participating on. cl->cmode must be new
362  * (activated) mode. It does nothing if cl->prio_activity == 0.
363  */
364 static void htb_activate_prios(struct htb_sched *q, struct htb_class *cl)
365 {
366         struct htb_class *p = cl->parent;
367         long m, mask = cl->prio_activity;
368
369         while (cl->cmode == HTB_MAY_BORROW && p && mask) {
370                 m = mask;
371                 while (m) {
372                         int prio = ffz(~m);
373                         m &= ~(1 << prio);
374
375                         if (p->un.inner.feed[prio].rb_node)
376                                 /* parent already has its feed in use so that
377                                    reset bit in mask as parent is already ok */
378                                 mask &= ~(1 << prio);
379
380                         htb_add_to_id_tree(p->un.inner.feed + prio, cl, prio);
381                 }
382                 p->prio_activity |= mask;
383                 cl = p;
384                 p = cl->parent;
385
386         }
387         if (cl->cmode == HTB_CAN_SEND && mask)
388                 htb_add_class_to_row(q, cl, mask);
389 }
390
391 /**
392  * htb_deactivate_prios - remove class from feed chain
393  *
394  * cl->cmode must represent old mode (before deactivation). It does
395  * nothing if cl->prio_activity == 0. Class is removed from all feed
396  * chains and rows.
397  */
398 static void htb_deactivate_prios(struct htb_sched *q, struct htb_class *cl)
399 {
400         struct htb_class *p = cl->parent;
401         long m, mask = cl->prio_activity;
402
403         while (cl->cmode == HTB_MAY_BORROW && p && mask) {
404                 m = mask;
405                 mask = 0;
406                 while (m) {
407                         int prio = ffz(~m);
408                         m &= ~(1 << prio);
409
410                         if (p->un.inner.ptr[prio] == cl->node + prio) {
411                                 /* we are removing child which is pointed to from
412                                    parent feed - forget the pointer but remember
413                                    classid */
414                                 p->un.inner.last_ptr_id[prio] = cl->common.classid;
415                                 p->un.inner.ptr[prio] = NULL;
416                         }
417
418                         htb_safe_rb_erase(cl->node + prio, p->un.inner.feed + prio);
419
420                         if (!p->un.inner.feed[prio].rb_node)
421                                 mask |= 1 << prio;
422                 }
423
424                 p->prio_activity &= ~mask;
425                 cl = p;
426                 p = cl->parent;
427
428         }
429         if (cl->cmode == HTB_CAN_SEND && mask)
430                 htb_remove_class_from_row(q, cl, mask);
431 }
432
433 static inline long htb_lowater(const struct htb_class *cl)
434 {
435         if (htb_hysteresis)
436                 return cl->cmode != HTB_CANT_SEND ? -cl->cbuffer : 0;
437         else
438                 return 0;
439 }
440 static inline long htb_hiwater(const struct htb_class *cl)
441 {
442         if (htb_hysteresis)
443                 return cl->cmode == HTB_CAN_SEND ? -cl->buffer : 0;
444         else
445                 return 0;
446 }
447
448
449 /**
450  * htb_class_mode - computes and returns current class mode
451  *
452  * It computes cl's mode at time cl->t_c+diff and returns it. If mode
453  * is not HTB_CAN_SEND then cl->pq_key is updated to time difference
454  * from now to time when cl will change its state.
455  * Also it is worth to note that class mode doesn't change simply
456  * at cl->{c,}tokens == 0 but there can rather be hysteresis of
457  * 0 .. -cl->{c,}buffer range. It is meant to limit number of
458  * mode transitions per time unit. The speed gain is about 1/6.
459  */
460 static inline enum htb_cmode
461 htb_class_mode(struct htb_class *cl, long *diff)
462 {
463         long toks;
464
465         if ((toks = (cl->ctokens + *diff)) < htb_lowater(cl)) {
466                 *diff = -toks;
467                 return HTB_CANT_SEND;
468         }
469
470         if ((toks = (cl->tokens + *diff)) >= htb_hiwater(cl))
471                 return HTB_CAN_SEND;
472
473         *diff = -toks;
474         return HTB_MAY_BORROW;
475 }
476
477 /**
478  * htb_change_class_mode - changes classe's mode
479  *
480  * This should be the only way how to change classe's mode under normal
481  * cirsumstances. Routine will update feed lists linkage, change mode
482  * and add class to the wait event queue if appropriate. New mode should
483  * be different from old one and cl->pq_key has to be valid if changing
484  * to mode other than HTB_CAN_SEND (see htb_add_to_wait_tree).
485  */
486 static void
487 htb_change_class_mode(struct htb_sched *q, struct htb_class *cl, long *diff)
488 {
489         enum htb_cmode new_mode = htb_class_mode(cl, diff);
490
491         if (new_mode == cl->cmode)
492                 return;
493
494         if (cl->prio_activity) {        /* not necessary: speed optimization */
495                 if (cl->cmode != HTB_CANT_SEND)
496                         htb_deactivate_prios(q, cl);
497                 cl->cmode = new_mode;
498                 if (new_mode != HTB_CANT_SEND)
499                         htb_activate_prios(q, cl);
500         } else
501                 cl->cmode = new_mode;
502 }
503
504 /**
505  * htb_activate - inserts leaf cl into appropriate active feeds
506  *
507  * Routine learns (new) priority of leaf and activates feed chain
508  * for the prio. It can be called on already active leaf safely.
509  * It also adds leaf into droplist.
510  */
511 static inline void htb_activate(struct htb_sched *q, struct htb_class *cl)
512 {
513         WARN_ON(cl->level || !cl->un.leaf.q || !cl->un.leaf.q->q.qlen);
514
515         if (!cl->prio_activity) {
516                 cl->prio_activity = 1 << cl->prio;
517                 htb_activate_prios(q, cl);
518                 list_add_tail(&cl->un.leaf.drop_list,
519                               q->drops + cl->prio);
520         }
521 }
522
523 /**
524  * htb_deactivate - remove leaf cl from active feeds
525  *
526  * Make sure that leaf is active. In the other words it can't be called
527  * with non-active leaf. It also removes class from the drop list.
528  */
529 static inline void htb_deactivate(struct htb_sched *q, struct htb_class *cl)
530 {
531         WARN_ON(!cl->prio_activity);
532
533         htb_deactivate_prios(q, cl);
534         cl->prio_activity = 0;
535         list_del_init(&cl->un.leaf.drop_list);
536 }
537
538 static int htb_enqueue(struct sk_buff *skb, struct Qdisc *sch)
539 {
540         int uninitialized_var(ret);
541         struct htb_sched *q = qdisc_priv(sch);
542         struct htb_class *cl = htb_classify(skb, sch, &ret);
543
544         if (cl == HTB_DIRECT) {
545                 /* enqueue to helper queue */
546                 if (q->direct_queue.qlen < q->direct_qlen) {
547                         __skb_queue_tail(&q->direct_queue, skb);
548                         q->direct_pkts++;
549                 } else {
550                         kfree_skb(skb);
551                         sch->qstats.drops++;
552                         return NET_XMIT_DROP;
553                 }
554 #ifdef CONFIG_NET_CLS_ACT
555         } else if (!cl) {
556                 if (ret & __NET_XMIT_BYPASS)
557                         sch->qstats.drops++;
558                 kfree_skb(skb);
559                 return ret;
560 #endif
561         } else if ((ret = qdisc_enqueue(skb, cl->un.leaf.q)) != NET_XMIT_SUCCESS) {
562                 if (net_xmit_drop_count(ret)) {
563                         sch->qstats.drops++;
564                         cl->qstats.drops++;
565                 }
566                 return ret;
567         } else {
568                 cl->bstats.packets +=
569                         skb_is_gso(skb)?skb_shinfo(skb)->gso_segs:1;
570                 cl->bstats.bytes += qdisc_pkt_len(skb);
571                 htb_activate(q, cl);
572         }
573
574         sch->q.qlen++;
575         sch->bstats.packets += skb_is_gso(skb)?skb_shinfo(skb)->gso_segs:1;
576         sch->bstats.bytes += qdisc_pkt_len(skb);
577         return NET_XMIT_SUCCESS;
578 }
579
580 static inline void htb_accnt_tokens(struct htb_class *cl, int bytes, long diff)
581 {
582         long toks = diff + cl->tokens;
583
584         if (toks > cl->buffer)
585                 toks = cl->buffer;
586         toks -= (long) qdisc_l2t(cl->rate, bytes);
587         if (toks <= -cl->mbuffer)
588                 toks = 1 - cl->mbuffer;
589
590         cl->tokens = toks;
591 }
592
593 static inline void htb_accnt_ctokens(struct htb_class *cl, int bytes, long diff)
594 {
595         long toks = diff + cl->ctokens;
596
597         if (toks > cl->cbuffer)
598                 toks = cl->cbuffer;
599         toks -= (long) qdisc_l2t(cl->ceil, bytes);
600         if (toks <= -cl->mbuffer)
601                 toks = 1 - cl->mbuffer;
602
603         cl->ctokens = toks;
604 }
605
606 /**
607  * htb_charge_class - charges amount "bytes" to leaf and ancestors
608  *
609  * Routine assumes that packet "bytes" long was dequeued from leaf cl
610  * borrowing from "level". It accounts bytes to ceil leaky bucket for
611  * leaf and all ancestors and to rate bucket for ancestors at levels
612  * "level" and higher. It also handles possible change of mode resulting
613  * from the update. Note that mode can also increase here (MAY_BORROW to
614  * CAN_SEND) because we can use more precise clock that event queue here.
615  * In such case we remove class from event queue first.
616  */
617 static void htb_charge_class(struct htb_sched *q, struct htb_class *cl,
618                              int level, struct sk_buff *skb)
619 {
620         int bytes = qdisc_pkt_len(skb);
621         enum htb_cmode old_mode;
622         long diff;
623
624         while (cl) {
625                 diff = psched_tdiff_bounded(q->now, cl->t_c, cl->mbuffer);
626                 if (cl->level >= level) {
627                         if (cl->level == level)
628                                 cl->xstats.lends++;
629                         htb_accnt_tokens(cl, bytes, diff);
630                 } else {
631                         cl->xstats.borrows++;
632                         cl->tokens += diff;     /* we moved t_c; update tokens */
633                 }
634                 htb_accnt_ctokens(cl, bytes, diff);
635                 cl->t_c = q->now;
636
637                 old_mode = cl->cmode;
638                 diff = 0;
639                 htb_change_class_mode(q, cl, &diff);
640                 if (old_mode != cl->cmode) {
641                         if (old_mode != HTB_CAN_SEND)
642                                 htb_safe_rb_erase(&cl->pq_node, q->wait_pq + cl->level);
643                         if (cl->cmode != HTB_CAN_SEND)
644                                 htb_add_to_wait_tree(q, cl, diff);
645                 }
646
647                 /* update byte stats except for leaves which are already updated */
648                 if (cl->level) {
649                         cl->bstats.bytes += bytes;
650                         cl->bstats.packets += skb_is_gso(skb)?
651                                         skb_shinfo(skb)->gso_segs:1;
652                 }
653                 cl = cl->parent;
654         }
655 }
656
657 /**
658  * htb_do_events - make mode changes to classes at the level
659  *
660  * Scans event queue for pending events and applies them. Returns time of
661  * next pending event (0 for no event in pq).
662  * Note: Applied are events whose have cl->pq_key <= q->now.
663  */
664 static psched_time_t htb_do_events(struct htb_sched *q, int level)
665 {
666         /* don't run for longer than 2 jiffies; 2 is used instead of
667            1 to simplify things when jiffy is going to be incremented
668            too soon */
669         unsigned long stop_at = jiffies + 2;
670         while (time_before(jiffies, stop_at)) {
671                 struct htb_class *cl;
672                 long diff;
673                 struct rb_node *p = rb_first(&q->wait_pq[level]);
674
675                 if (!p)
676                         return 0;
677
678                 cl = rb_entry(p, struct htb_class, pq_node);
679                 if (cl->pq_key > q->now)
680                         return cl->pq_key;
681
682                 htb_safe_rb_erase(p, q->wait_pq + level);
683                 diff = psched_tdiff_bounded(q->now, cl->t_c, cl->mbuffer);
684                 htb_change_class_mode(q, cl, &diff);
685                 if (cl->cmode != HTB_CAN_SEND)
686                         htb_add_to_wait_tree(q, cl, diff);
687         }
688         /* too much load - let's continue on next jiffie */
689         return q->now + PSCHED_TICKS_PER_SEC / HZ;
690 }
691
692 /* Returns class->node+prio from id-tree where classe's id is >= id. NULL
693    is no such one exists. */
694 static struct rb_node *htb_id_find_next_upper(int prio, struct rb_node *n,
695                                               u32 id)
696 {
697         struct rb_node *r = NULL;
698         while (n) {
699                 struct htb_class *cl =
700                     rb_entry(n, struct htb_class, node[prio]);
701                 if (id == cl->common.classid)
702                         return n;
703
704                 if (id > cl->common.classid) {
705                         n = n->rb_right;
706                 } else {
707                         r = n;
708                         n = n->rb_left;
709                 }
710         }
711         return r;
712 }
713
714 /**
715  * htb_lookup_leaf - returns next leaf class in DRR order
716  *
717  * Find leaf where current feed pointers points to.
718  */
719 static struct htb_class *htb_lookup_leaf(struct rb_root *tree, int prio,
720                                          struct rb_node **pptr, u32 * pid)
721 {
722         int i;
723         struct {
724                 struct rb_node *root;
725                 struct rb_node **pptr;
726                 u32 *pid;
727         } stk[TC_HTB_MAXDEPTH], *sp = stk;
728
729         WARN_ON(!tree->rb_node);
730         sp->root = tree->rb_node;
731         sp->pptr = pptr;
732         sp->pid = pid;
733
734         for (i = 0; i < 65535; i++) {
735                 if (!*sp->pptr && *sp->pid) {
736                         /* ptr was invalidated but id is valid - try to recover
737                            the original or next ptr */
738                         *sp->pptr =
739                             htb_id_find_next_upper(prio, sp->root, *sp->pid);
740                 }
741                 *sp->pid = 0;   /* ptr is valid now so that remove this hint as it
742                                    can become out of date quickly */
743                 if (!*sp->pptr) {       /* we are at right end; rewind & go up */
744                         *sp->pptr = sp->root;
745                         while ((*sp->pptr)->rb_left)
746                                 *sp->pptr = (*sp->pptr)->rb_left;
747                         if (sp > stk) {
748                                 sp--;
749                                 WARN_ON(!*sp->pptr);
750                                 if (!*sp->pptr)
751                                         return NULL;
752                                 htb_next_rb_node(sp->pptr);
753                         }
754                 } else {
755                         struct htb_class *cl;
756                         cl = rb_entry(*sp->pptr, struct htb_class, node[prio]);
757                         if (!cl->level)
758                                 return cl;
759                         (++sp)->root = cl->un.inner.feed[prio].rb_node;
760                         sp->pptr = cl->un.inner.ptr + prio;
761                         sp->pid = cl->un.inner.last_ptr_id + prio;
762                 }
763         }
764         WARN_ON(1);
765         return NULL;
766 }
767
768 /* dequeues packet at given priority and level; call only if
769    you are sure that there is active class at prio/level */
770 static struct sk_buff *htb_dequeue_tree(struct htb_sched *q, int prio,
771                                         int level)
772 {
773         struct sk_buff *skb = NULL;
774         struct htb_class *cl, *start;
775         /* look initial class up in the row */
776         start = cl = htb_lookup_leaf(q->row[level] + prio, prio,
777                                      q->ptr[level] + prio,
778                                      q->last_ptr_id[level] + prio);
779
780         do {
781 next:
782                 WARN_ON(!cl);
783                 if (!cl)
784                         return NULL;
785
786                 /* class can be empty - it is unlikely but can be true if leaf
787                    qdisc drops packets in enqueue routine or if someone used
788                    graft operation on the leaf since last dequeue;
789                    simply deactivate and skip such class */
790                 if (unlikely(cl->un.leaf.q->q.qlen == 0)) {
791                         struct htb_class *next;
792                         htb_deactivate(q, cl);
793
794                         /* row/level might become empty */
795                         if ((q->row_mask[level] & (1 << prio)) == 0)
796                                 return NULL;
797
798                         next = htb_lookup_leaf(q->row[level] + prio,
799                                                prio, q->ptr[level] + prio,
800                                                q->last_ptr_id[level] + prio);
801
802                         if (cl == start)        /* fix start if we just deleted it */
803                                 start = next;
804                         cl = next;
805                         goto next;
806                 }
807
808                 skb = cl->un.leaf.q->dequeue(cl->un.leaf.q);
809                 if (likely(skb != NULL))
810                         break;
811                 if (!cl->warned) {
812                         printk(KERN_WARNING
813                                "htb: class %X isn't work conserving ?!\n",
814                                cl->common.classid);
815                         cl->warned = 1;
816                 }
817
818                 htb_next_rb_node((level ? cl->parent->un.inner.ptr : q->
819                                   ptr[0]) + prio);
820                 cl = htb_lookup_leaf(q->row[level] + prio, prio,
821                                      q->ptr[level] + prio,
822                                      q->last_ptr_id[level] + prio);
823
824         } while (cl != start);
825
826         if (likely(skb != NULL)) {
827                 cl->un.leaf.deficit[level] -= qdisc_pkt_len(skb);
828                 if (cl->un.leaf.deficit[level] < 0) {
829                         cl->un.leaf.deficit[level] += cl->quantum;
830                         htb_next_rb_node((level ? cl->parent->un.inner.ptr : q->
831                                           ptr[0]) + prio);
832                 }
833                 /* this used to be after charge_class but this constelation
834                    gives us slightly better performance */
835                 if (!cl->un.leaf.q->q.qlen)
836                         htb_deactivate(q, cl);
837                 htb_charge_class(q, cl, level, skb);
838         }
839         return skb;
840 }
841
842 static struct sk_buff *htb_dequeue(struct Qdisc *sch)
843 {
844         struct sk_buff *skb = NULL;
845         struct htb_sched *q = qdisc_priv(sch);
846         int level;
847         psched_time_t next_event;
848
849         /* try to dequeue direct packets as high prio (!) to minimize cpu work */
850         skb = __skb_dequeue(&q->direct_queue);
851         if (skb != NULL) {
852                 sch->flags &= ~TCQ_F_THROTTLED;
853                 sch->q.qlen--;
854                 return skb;
855         }
856
857         if (!sch->q.qlen)
858                 goto fin;
859         q->now = psched_get_time();
860
861         next_event = q->now + 5 * PSCHED_TICKS_PER_SEC;
862
863         for (level = 0; level < TC_HTB_MAXDEPTH; level++) {
864                 /* common case optimization - skip event handler quickly */
865                 int m;
866                 psched_time_t event;
867
868                 if (q->now >= q->near_ev_cache[level]) {
869                         event = htb_do_events(q, level);
870                         if (!event)
871                                 event = q->now + PSCHED_TICKS_PER_SEC;
872                         q->near_ev_cache[level] = event;
873                 } else
874                         event = q->near_ev_cache[level];
875
876                 if (event && next_event > event)
877                         next_event = event;
878
879                 m = ~q->row_mask[level];
880                 while (m != (int)(-1)) {
881                         int prio = ffz(m);
882                         m |= 1 << prio;
883                         skb = htb_dequeue_tree(q, prio, level);
884                         if (likely(skb != NULL)) {
885                                 sch->q.qlen--;
886                                 sch->flags &= ~TCQ_F_THROTTLED;
887                                 goto fin;
888                         }
889                 }
890         }
891         sch->qstats.overlimits++;
892         qdisc_watchdog_schedule(&q->watchdog, next_event);
893 fin:
894         return skb;
895 }
896
897 /* try to drop from each class (by prio) until one succeed */
898 static unsigned int htb_drop(struct Qdisc *sch)
899 {
900         struct htb_sched *q = qdisc_priv(sch);
901         int prio;
902
903         for (prio = TC_HTB_NUMPRIO - 1; prio >= 0; prio--) {
904                 struct list_head *p;
905                 list_for_each(p, q->drops + prio) {
906                         struct htb_class *cl = list_entry(p, struct htb_class,
907                                                           un.leaf.drop_list);
908                         unsigned int len;
909                         if (cl->un.leaf.q->ops->drop &&
910                             (len = cl->un.leaf.q->ops->drop(cl->un.leaf.q))) {
911                                 sch->q.qlen--;
912                                 if (!cl->un.leaf.q->q.qlen)
913                                         htb_deactivate(q, cl);
914                                 return len;
915                         }
916                 }
917         }
918         return 0;
919 }
920
921 /* reset all classes */
922 /* always caled under BH & queue lock */
923 static void htb_reset(struct Qdisc *sch)
924 {
925         struct htb_sched *q = qdisc_priv(sch);
926         struct htb_class *cl;
927         struct hlist_node *n;
928         unsigned int i;
929
930         for (i = 0; i < q->clhash.hashsize; i++) {
931                 hlist_for_each_entry(cl, n, &q->clhash.hash[i], common.hnode) {
932                         if (cl->level)
933                                 memset(&cl->un.inner, 0, sizeof(cl->un.inner));
934                         else {
935                                 if (cl->un.leaf.q)
936                                         qdisc_reset(cl->un.leaf.q);
937                                 INIT_LIST_HEAD(&cl->un.leaf.drop_list);
938                         }
939                         cl->prio_activity = 0;
940                         cl->cmode = HTB_CAN_SEND;
941
942                 }
943         }
944         qdisc_watchdog_cancel(&q->watchdog);
945         __skb_queue_purge(&q->direct_queue);
946         sch->q.qlen = 0;
947         memset(q->row, 0, sizeof(q->row));
948         memset(q->row_mask, 0, sizeof(q->row_mask));
949         memset(q->wait_pq, 0, sizeof(q->wait_pq));
950         memset(q->ptr, 0, sizeof(q->ptr));
951         for (i = 0; i < TC_HTB_NUMPRIO; i++)
952                 INIT_LIST_HEAD(q->drops + i);
953 }
954
955 static const struct nla_policy htb_policy[TCA_HTB_MAX + 1] = {
956         [TCA_HTB_PARMS] = { .len = sizeof(struct tc_htb_opt) },
957         [TCA_HTB_INIT]  = { .len = sizeof(struct tc_htb_glob) },
958         [TCA_HTB_CTAB]  = { .type = NLA_BINARY, .len = TC_RTAB_SIZE },
959         [TCA_HTB_RTAB]  = { .type = NLA_BINARY, .len = TC_RTAB_SIZE },
960 };
961
962 static int htb_init(struct Qdisc *sch, struct nlattr *opt)
963 {
964         struct htb_sched *q = qdisc_priv(sch);
965         struct nlattr *tb[TCA_HTB_INIT + 1];
966         struct tc_htb_glob *gopt;
967         int err;
968         int i;
969
970         if (!opt)
971                 return -EINVAL;
972
973         err = nla_parse_nested(tb, TCA_HTB_INIT, opt, htb_policy);
974         if (err < 0)
975                 return err;
976
977         if (tb[TCA_HTB_INIT] == NULL) {
978                 printk(KERN_ERR "HTB: hey probably you have bad tc tool ?\n");
979                 return -EINVAL;
980         }
981         gopt = nla_data(tb[TCA_HTB_INIT]);
982         if (gopt->version != HTB_VER >> 16) {
983                 printk(KERN_ERR
984                        "HTB: need tc/htb version %d (minor is %d), you have %d\n",
985                        HTB_VER >> 16, HTB_VER & 0xffff, gopt->version);
986                 return -EINVAL;
987         }
988
989         err = qdisc_class_hash_init(&q->clhash);
990         if (err < 0)
991                 return err;
992         for (i = 0; i < TC_HTB_NUMPRIO; i++)
993                 INIT_LIST_HEAD(q->drops + i);
994
995         qdisc_watchdog_init(&q->watchdog, sch);
996         skb_queue_head_init(&q->direct_queue);
997
998         q->direct_qlen = qdisc_dev(sch)->tx_queue_len;
999         if (q->direct_qlen < 2) /* some devices have zero tx_queue_len */
1000                 q->direct_qlen = 2;
1001
1002         if ((q->rate2quantum = gopt->rate2quantum) < 1)
1003                 q->rate2quantum = 1;
1004         q->defcls = gopt->defcls;
1005
1006         return 0;
1007 }
1008
1009 static int htb_dump(struct Qdisc *sch, struct sk_buff *skb)
1010 {
1011         spinlock_t *root_lock = qdisc_root_sleeping_lock(sch);
1012         struct htb_sched *q = qdisc_priv(sch);
1013         struct nlattr *nest;
1014         struct tc_htb_glob gopt;
1015
1016         spin_lock_bh(root_lock);
1017
1018         gopt.direct_pkts = q->direct_pkts;
1019         gopt.version = HTB_VER;
1020         gopt.rate2quantum = q->rate2quantum;
1021         gopt.defcls = q->defcls;
1022         gopt.debug = 0;
1023
1024         nest = nla_nest_start(skb, TCA_OPTIONS);
1025         if (nest == NULL)
1026                 goto nla_put_failure;
1027         NLA_PUT(skb, TCA_HTB_INIT, sizeof(gopt), &gopt);
1028         nla_nest_end(skb, nest);
1029
1030         spin_unlock_bh(root_lock);
1031         return skb->len;
1032
1033 nla_put_failure:
1034         spin_unlock_bh(root_lock);
1035         nla_nest_cancel(skb, nest);
1036         return -1;
1037 }
1038
1039 static int htb_dump_class(struct Qdisc *sch, unsigned long arg,
1040                           struct sk_buff *skb, struct tcmsg *tcm)
1041 {
1042         struct htb_class *cl = (struct htb_class *)arg;
1043         spinlock_t *root_lock = qdisc_root_sleeping_lock(sch);
1044         struct nlattr *nest;
1045         struct tc_htb_opt opt;
1046
1047         spin_lock_bh(root_lock);
1048         tcm->tcm_parent = cl->parent ? cl->parent->common.classid : TC_H_ROOT;
1049         tcm->tcm_handle = cl->common.classid;
1050         if (!cl->level && cl->un.leaf.q)
1051                 tcm->tcm_info = cl->un.leaf.q->handle;
1052
1053         nest = nla_nest_start(skb, TCA_OPTIONS);
1054         if (nest == NULL)
1055                 goto nla_put_failure;
1056
1057         memset(&opt, 0, sizeof(opt));
1058
1059         opt.rate = cl->rate->rate;
1060         opt.buffer = cl->buffer;
1061         opt.ceil = cl->ceil->rate;
1062         opt.cbuffer = cl->cbuffer;
1063         opt.quantum = cl->quantum;
1064         opt.prio = cl->prio;
1065         opt.level = cl->level;
1066         NLA_PUT(skb, TCA_HTB_PARMS, sizeof(opt), &opt);
1067
1068         nla_nest_end(skb, nest);
1069         spin_unlock_bh(root_lock);
1070         return skb->len;
1071
1072 nla_put_failure:
1073         spin_unlock_bh(root_lock);
1074         nla_nest_cancel(skb, nest);
1075         return -1;
1076 }
1077
1078 static int
1079 htb_dump_class_stats(struct Qdisc *sch, unsigned long arg, struct gnet_dump *d)
1080 {
1081         struct htb_class *cl = (struct htb_class *)arg;
1082
1083         if (!cl->level && cl->un.leaf.q)
1084                 cl->qstats.qlen = cl->un.leaf.q->q.qlen;
1085         cl->xstats.tokens = cl->tokens;
1086         cl->xstats.ctokens = cl->ctokens;
1087
1088         if (gnet_stats_copy_basic(d, &cl->bstats) < 0 ||
1089             gnet_stats_copy_rate_est(d, &cl->rate_est) < 0 ||
1090             gnet_stats_copy_queue(d, &cl->qstats) < 0)
1091                 return -1;
1092
1093         return gnet_stats_copy_app(d, &cl->xstats, sizeof(cl->xstats));
1094 }
1095
1096 static int htb_graft(struct Qdisc *sch, unsigned long arg, struct Qdisc *new,
1097                      struct Qdisc **old)
1098 {
1099         struct htb_class *cl = (struct htb_class *)arg;
1100
1101         if (cl && !cl->level) {
1102                 if (new == NULL &&
1103                     (new = qdisc_create_dflt(qdisc_dev(sch), sch->dev_queue,
1104                                              &pfifo_qdisc_ops,
1105                                              cl->common.classid))
1106                     == NULL)
1107                         return -ENOBUFS;
1108                 sch_tree_lock(sch);
1109                 *old = cl->un.leaf.q;
1110                 cl->un.leaf.q = new;
1111                 if (*old != NULL) {
1112                         qdisc_tree_decrease_qlen(*old, (*old)->q.qlen);
1113                         qdisc_reset(*old);
1114                 }
1115                 sch_tree_unlock(sch);
1116                 return 0;
1117         }
1118         return -ENOENT;
1119 }
1120
1121 static struct Qdisc *htb_leaf(struct Qdisc *sch, unsigned long arg)
1122 {
1123         struct htb_class *cl = (struct htb_class *)arg;
1124         return (cl && !cl->level) ? cl->un.leaf.q : NULL;
1125 }
1126
1127 static void htb_qlen_notify(struct Qdisc *sch, unsigned long arg)
1128 {
1129         struct htb_class *cl = (struct htb_class *)arg;
1130
1131         if (cl->un.leaf.q->q.qlen == 0)
1132                 htb_deactivate(qdisc_priv(sch), cl);
1133 }
1134
1135 static unsigned long htb_get(struct Qdisc *sch, u32 classid)
1136 {
1137         struct htb_class *cl = htb_find(classid, sch);
1138         if (cl)
1139                 cl->refcnt++;
1140         return (unsigned long)cl;
1141 }
1142
1143 static inline int htb_parent_last_child(struct htb_class *cl)
1144 {
1145         if (!cl->parent)
1146                 /* the root class */
1147                 return 0;
1148         if (cl->parent->children > 1)
1149                 /* not the last child */
1150                 return 0;
1151         return 1;
1152 }
1153
1154 static void htb_parent_to_leaf(struct htb_sched *q, struct htb_class *cl,
1155                                struct Qdisc *new_q)
1156 {
1157         struct htb_class *parent = cl->parent;
1158
1159         WARN_ON(cl->level || !cl->un.leaf.q || cl->prio_activity);
1160
1161         if (parent->cmode != HTB_CAN_SEND)
1162                 htb_safe_rb_erase(&parent->pq_node, q->wait_pq + parent->level);
1163
1164         parent->level = 0;
1165         memset(&parent->un.inner, 0, sizeof(parent->un.inner));
1166         INIT_LIST_HEAD(&parent->un.leaf.drop_list);
1167         parent->un.leaf.q = new_q ? new_q : &noop_qdisc;
1168         parent->tokens = parent->buffer;
1169         parent->ctokens = parent->cbuffer;
1170         parent->t_c = psched_get_time();
1171         parent->cmode = HTB_CAN_SEND;
1172 }
1173
1174 static void htb_destroy_class(struct Qdisc *sch, struct htb_class *cl)
1175 {
1176         if (!cl->level) {
1177                 WARN_ON(!cl->un.leaf.q);
1178                 qdisc_destroy(cl->un.leaf.q);
1179         }
1180         gen_kill_estimator(&cl->bstats, &cl->rate_est);
1181         qdisc_put_rtab(cl->rate);
1182         qdisc_put_rtab(cl->ceil);
1183
1184         tcf_destroy_chain(&cl->filter_list);
1185         kfree(cl);
1186 }
1187
1188 /* always caled under BH & queue lock */
1189 static void htb_destroy(struct Qdisc *sch)
1190 {
1191         struct htb_sched *q = qdisc_priv(sch);
1192         struct hlist_node *n, *next;
1193         struct htb_class *cl;
1194         unsigned int i;
1195
1196         qdisc_watchdog_cancel(&q->watchdog);
1197         /* This line used to be after htb_destroy_class call below
1198            and surprisingly it worked in 2.4. But it must precede it
1199            because filter need its target class alive to be able to call
1200            unbind_filter on it (without Oops). */
1201         tcf_destroy_chain(&q->filter_list);
1202
1203         for (i = 0; i < q->clhash.hashsize; i++) {
1204                 hlist_for_each_entry(cl, n, &q->clhash.hash[i], common.hnode)
1205                         tcf_destroy_chain(&cl->filter_list);
1206         }
1207         for (i = 0; i < q->clhash.hashsize; i++) {
1208                 hlist_for_each_entry_safe(cl, n, next, &q->clhash.hash[i],
1209                                           common.hnode)
1210                         htb_destroy_class(sch, cl);
1211         }
1212         qdisc_class_hash_destroy(&q->clhash);
1213         __skb_queue_purge(&q->direct_queue);
1214 }
1215
1216 static int htb_delete(struct Qdisc *sch, unsigned long arg)
1217 {
1218         struct htb_sched *q = qdisc_priv(sch);
1219         struct htb_class *cl = (struct htb_class *)arg;
1220         unsigned int qlen;
1221         struct Qdisc *new_q = NULL;
1222         int last_child = 0;
1223
1224         // TODO: why don't allow to delete subtree ? references ? does
1225         // tc subsys quarantee us that in htb_destroy it holds no class
1226         // refs so that we can remove children safely there ?
1227         if (cl->children || cl->filter_cnt)
1228                 return -EBUSY;
1229
1230         if (!cl->level && htb_parent_last_child(cl)) {
1231                 new_q = qdisc_create_dflt(qdisc_dev(sch), sch->dev_queue,
1232                                           &pfifo_qdisc_ops,
1233                                           cl->parent->common.classid);
1234                 last_child = 1;
1235         }
1236
1237         sch_tree_lock(sch);
1238
1239         if (!cl->level) {
1240                 qlen = cl->un.leaf.q->q.qlen;
1241                 qdisc_reset(cl->un.leaf.q);
1242                 qdisc_tree_decrease_qlen(cl->un.leaf.q, qlen);
1243         }
1244
1245         /* delete from hash and active; remainder in destroy_class */
1246         qdisc_class_hash_remove(&q->clhash, &cl->common);
1247         if (cl->parent)
1248                 cl->parent->children--;
1249
1250         if (cl->prio_activity)
1251                 htb_deactivate(q, cl);
1252
1253         if (cl->cmode != HTB_CAN_SEND)
1254                 htb_safe_rb_erase(&cl->pq_node, q->wait_pq + cl->level);
1255
1256         if (last_child)
1257                 htb_parent_to_leaf(q, cl, new_q);
1258
1259         if (--cl->refcnt == 0)
1260                 htb_destroy_class(sch, cl);
1261
1262         sch_tree_unlock(sch);
1263         return 0;
1264 }
1265
1266 static void htb_put(struct Qdisc *sch, unsigned long arg)
1267 {
1268         struct htb_class *cl = (struct htb_class *)arg;
1269
1270         if (--cl->refcnt == 0)
1271                 htb_destroy_class(sch, cl);
1272 }
1273
1274 static int htb_change_class(struct Qdisc *sch, u32 classid,
1275                             u32 parentid, struct nlattr **tca,
1276                             unsigned long *arg)
1277 {
1278         int err = -EINVAL;
1279         struct htb_sched *q = qdisc_priv(sch);
1280         struct htb_class *cl = (struct htb_class *)*arg, *parent;
1281         struct nlattr *opt = tca[TCA_OPTIONS];
1282         struct qdisc_rate_table *rtab = NULL, *ctab = NULL;
1283         struct nlattr *tb[TCA_HTB_RTAB + 1];
1284         struct tc_htb_opt *hopt;
1285
1286         /* extract all subattrs from opt attr */
1287         if (!opt)
1288                 goto failure;
1289
1290         err = nla_parse_nested(tb, TCA_HTB_RTAB, opt, htb_policy);
1291         if (err < 0)
1292                 goto failure;
1293
1294         err = -EINVAL;
1295         if (tb[TCA_HTB_PARMS] == NULL)
1296                 goto failure;
1297
1298         parent = parentid == TC_H_ROOT ? NULL : htb_find(parentid, sch);
1299
1300         hopt = nla_data(tb[TCA_HTB_PARMS]);
1301
1302         rtab = qdisc_get_rtab(&hopt->rate, tb[TCA_HTB_RTAB]);
1303         ctab = qdisc_get_rtab(&hopt->ceil, tb[TCA_HTB_CTAB]);
1304         if (!rtab || !ctab)
1305                 goto failure;
1306
1307         if (!cl) {              /* new class */
1308                 struct Qdisc *new_q;
1309                 int prio;
1310                 struct {
1311                         struct nlattr           nla;
1312                         struct gnet_estimator   opt;
1313                 } est = {
1314                         .nla = {
1315                                 .nla_len        = nla_attr_size(sizeof(est.opt)),
1316                                 .nla_type       = TCA_RATE,
1317                         },
1318                         .opt = {
1319                                 /* 4s interval, 16s averaging constant */
1320                                 .interval       = 2,
1321                                 .ewma_log       = 2,
1322                         },
1323                 };
1324
1325                 /* check for valid classid */
1326                 if (!classid || TC_H_MAJ(classid ^ sch->handle)
1327                     || htb_find(classid, sch))
1328                         goto failure;
1329
1330                 /* check maximal depth */
1331                 if (parent && parent->parent && parent->parent->level < 2) {
1332                         printk(KERN_ERR "htb: tree is too deep\n");
1333                         goto failure;
1334                 }
1335                 err = -ENOBUFS;
1336                 if ((cl = kzalloc(sizeof(*cl), GFP_KERNEL)) == NULL)
1337                         goto failure;
1338
1339                 err = gen_new_estimator(&cl->bstats, &cl->rate_est,
1340                                         qdisc_root_sleeping_lock(sch),
1341                                         tca[TCA_RATE] ? : &est.nla);
1342                 if (err) {
1343                         kfree(cl);
1344                         goto failure;
1345                 }
1346
1347                 cl->refcnt = 1;
1348                 cl->children = 0;
1349                 INIT_LIST_HEAD(&cl->un.leaf.drop_list);
1350                 RB_CLEAR_NODE(&cl->pq_node);
1351
1352                 for (prio = 0; prio < TC_HTB_NUMPRIO; prio++)
1353                         RB_CLEAR_NODE(&cl->node[prio]);
1354
1355                 /* create leaf qdisc early because it uses kmalloc(GFP_KERNEL)
1356                    so that can't be used inside of sch_tree_lock
1357                    -- thanks to Karlis Peisenieks */
1358                 new_q = qdisc_create_dflt(qdisc_dev(sch), sch->dev_queue,
1359                                           &pfifo_qdisc_ops, classid);
1360                 sch_tree_lock(sch);
1361                 if (parent && !parent->level) {
1362                         unsigned int qlen = parent->un.leaf.q->q.qlen;
1363
1364                         /* turn parent into inner node */
1365                         qdisc_reset(parent->un.leaf.q);
1366                         qdisc_tree_decrease_qlen(parent->un.leaf.q, qlen);
1367                         qdisc_destroy(parent->un.leaf.q);
1368                         if (parent->prio_activity)
1369                                 htb_deactivate(q, parent);
1370
1371                         /* remove from evt list because of level change */
1372                         if (parent->cmode != HTB_CAN_SEND) {
1373                                 htb_safe_rb_erase(&parent->pq_node, q->wait_pq);
1374                                 parent->cmode = HTB_CAN_SEND;
1375                         }
1376                         parent->level = (parent->parent ? parent->parent->level
1377                                          : TC_HTB_MAXDEPTH) - 1;
1378                         memset(&parent->un.inner, 0, sizeof(parent->un.inner));
1379                 }
1380                 /* leaf (we) needs elementary qdisc */
1381                 cl->un.leaf.q = new_q ? new_q : &noop_qdisc;
1382
1383                 cl->common.classid = classid;
1384                 cl->parent = parent;
1385
1386                 /* set class to be in HTB_CAN_SEND state */
1387                 cl->tokens = hopt->buffer;
1388                 cl->ctokens = hopt->cbuffer;
1389                 cl->mbuffer = 60 * PSCHED_TICKS_PER_SEC;        /* 1min */
1390                 cl->t_c = psched_get_time();
1391                 cl->cmode = HTB_CAN_SEND;
1392
1393                 /* attach to the hash list and parent's family */
1394                 qdisc_class_hash_insert(&q->clhash, &cl->common);
1395                 if (parent)
1396                         parent->children++;
1397         } else {
1398                 if (tca[TCA_RATE]) {
1399                         err = gen_replace_estimator(&cl->bstats, &cl->rate_est,
1400                                                     qdisc_root_sleeping_lock(sch),
1401                                                     tca[TCA_RATE]);
1402                         if (err)
1403                                 return err;
1404                 }
1405                 sch_tree_lock(sch);
1406         }
1407
1408         /* it used to be a nasty bug here, we have to check that node
1409            is really leaf before changing cl->un.leaf ! */
1410         if (!cl->level) {
1411                 cl->quantum = rtab->rate.rate / q->rate2quantum;
1412                 if (!hopt->quantum && cl->quantum < 1000) {
1413                         printk(KERN_WARNING
1414                                "HTB: quantum of class %X is small. Consider r2q change.\n",
1415                                cl->common.classid);
1416                         cl->quantum = 1000;
1417                 }
1418                 if (!hopt->quantum && cl->quantum > 200000) {
1419                         printk(KERN_WARNING
1420                                "HTB: quantum of class %X is big. Consider r2q change.\n",
1421                                cl->common.classid);
1422                         cl->quantum = 200000;
1423                 }
1424                 if (hopt->quantum)
1425                         cl->quantum = hopt->quantum;
1426                 if ((cl->prio = hopt->prio) >= TC_HTB_NUMPRIO)
1427                         cl->prio = TC_HTB_NUMPRIO - 1;
1428         }
1429
1430         cl->buffer = hopt->buffer;
1431         cl->cbuffer = hopt->cbuffer;
1432         if (cl->rate)
1433                 qdisc_put_rtab(cl->rate);
1434         cl->rate = rtab;
1435         if (cl->ceil)
1436                 qdisc_put_rtab(cl->ceil);
1437         cl->ceil = ctab;
1438         sch_tree_unlock(sch);
1439
1440         qdisc_class_hash_grow(sch, &q->clhash);
1441
1442         *arg = (unsigned long)cl;
1443         return 0;
1444
1445 failure:
1446         if (rtab)
1447                 qdisc_put_rtab(rtab);
1448         if (ctab)
1449                 qdisc_put_rtab(ctab);
1450         return err;
1451 }
1452
1453 static struct tcf_proto **htb_find_tcf(struct Qdisc *sch, unsigned long arg)
1454 {
1455         struct htb_sched *q = qdisc_priv(sch);
1456         struct htb_class *cl = (struct htb_class *)arg;
1457         struct tcf_proto **fl = cl ? &cl->filter_list : &q->filter_list;
1458
1459         return fl;
1460 }
1461
1462 static unsigned long htb_bind_filter(struct Qdisc *sch, unsigned long parent,
1463                                      u32 classid)
1464 {
1465         struct htb_class *cl = htb_find(classid, sch);
1466
1467         /*if (cl && !cl->level) return 0;
1468            The line above used to be there to prevent attaching filters to
1469            leaves. But at least tc_index filter uses this just to get class
1470            for other reasons so that we have to allow for it.
1471            ----
1472            19.6.2002 As Werner explained it is ok - bind filter is just
1473            another way to "lock" the class - unlike "get" this lock can
1474            be broken by class during destroy IIUC.
1475          */
1476         if (cl)
1477                 cl->filter_cnt++;
1478         return (unsigned long)cl;
1479 }
1480
1481 static void htb_unbind_filter(struct Qdisc *sch, unsigned long arg)
1482 {
1483         struct htb_class *cl = (struct htb_class *)arg;
1484
1485         if (cl)
1486                 cl->filter_cnt--;
1487 }
1488
1489 static void htb_walk(struct Qdisc *sch, struct qdisc_walker *arg)
1490 {
1491         struct htb_sched *q = qdisc_priv(sch);
1492         struct htb_class *cl;
1493         struct hlist_node *n;
1494         unsigned int i;
1495
1496         if (arg->stop)
1497                 return;
1498
1499         for (i = 0; i < q->clhash.hashsize; i++) {
1500                 hlist_for_each_entry(cl, n, &q->clhash.hash[i], common.hnode) {
1501                         if (arg->count < arg->skip) {
1502                                 arg->count++;
1503                                 continue;
1504                         }
1505                         if (arg->fn(sch, (unsigned long)cl, arg) < 0) {
1506                                 arg->stop = 1;
1507                                 return;
1508                         }
1509                         arg->count++;
1510                 }
1511         }
1512 }
1513
1514 static const struct Qdisc_class_ops htb_class_ops = {
1515         .graft          =       htb_graft,
1516         .leaf           =       htb_leaf,
1517         .qlen_notify    =       htb_qlen_notify,
1518         .get            =       htb_get,
1519         .put            =       htb_put,
1520         .change         =       htb_change_class,
1521         .delete         =       htb_delete,
1522         .walk           =       htb_walk,
1523         .tcf_chain      =       htb_find_tcf,
1524         .bind_tcf       =       htb_bind_filter,
1525         .unbind_tcf     =       htb_unbind_filter,
1526         .dump           =       htb_dump_class,
1527         .dump_stats     =       htb_dump_class_stats,
1528 };
1529
1530 static struct Qdisc_ops htb_qdisc_ops __read_mostly = {
1531         .next           =       NULL,
1532         .cl_ops         =       &htb_class_ops,
1533         .id             =       "htb",
1534         .priv_size      =       sizeof(struct htb_sched),
1535         .enqueue        =       htb_enqueue,
1536         .dequeue        =       htb_dequeue,
1537         .peek           =       qdisc_peek_dequeued,
1538         .drop           =       htb_drop,
1539         .init           =       htb_init,
1540         .reset          =       htb_reset,
1541         .destroy        =       htb_destroy,
1542         .change         =       NULL /* htb_change */,
1543         .dump           =       htb_dump,
1544         .owner          =       THIS_MODULE,
1545 };
1546
1547 static int __init htb_module_init(void)
1548 {
1549         return register_qdisc(&htb_qdisc_ops);
1550 }
1551 static void __exit htb_module_exit(void)
1552 {
1553         unregister_qdisc(&htb_qdisc_ops);
1554 }
1555
1556 module_init(htb_module_init)
1557 module_exit(htb_module_exit)
1558 MODULE_LICENSE("GPL");