1 // SPDX-License-Identifier: GPL-2.0-or-later 1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /* 2 /*
3 * net/sched/sch_htb.c Hierarchical token buc 3 * net/sched/sch_htb.c Hierarchical token bucket, feed tree version
4 * 4 *
5 * Authors: Martin Devera, <devik@cdi.cz> 5 * Authors: Martin Devera, <devik@cdi.cz>
6 * 6 *
7 * Credits (in time order) for older HTB versi 7 * Credits (in time order) for older HTB versions:
8 * Stef Coene <stef.coene@docum.o 8 * Stef Coene <stef.coene@docum.org>
9 * HTB support at LARTC m 9 * HTB support at LARTC mailing list
10 * Ondrej Kraus, <krauso@barr.cz> 10 * Ondrej Kraus, <krauso@barr.cz>
11 * found missing INIT_QDI 11 * found missing INIT_QDISC(htb)
12 * Vladimir Smelhaus, Aamer Akhte 12 * Vladimir Smelhaus, Aamer Akhter, Bert Hubert
13 * helped a lot to locate 13 * helped a lot to locate nasty class stall bug
14 * Andi Kleen, Jamal Hadi, Bert H 14 * Andi Kleen, Jamal Hadi, Bert Hubert
15 * code review and helpfu 15 * code review and helpful comments on shaping
16 * Tomasz Wrona, <tw@eter.tym.pl> 16 * Tomasz Wrona, <tw@eter.tym.pl>
17 * created test case so t 17 * created test case so that I was able to fix nasty bug
18 * Wilfried Weissmann 18 * Wilfried Weissmann
19 * spotted bug in dequeue 19 * spotted bug in dequeue code and helped with fix
20 * Jiri Fojtasek 20 * Jiri Fojtasek
21 * fixed requeue routine 21 * fixed requeue routine
22 * and many others. thanks. 22 * and many others. thanks.
23 */ 23 */
24 #include <linux/module.h> 24 #include <linux/module.h>
25 #include <linux/moduleparam.h> 25 #include <linux/moduleparam.h>
26 #include <linux/types.h> 26 #include <linux/types.h>
27 #include <linux/kernel.h> 27 #include <linux/kernel.h>
28 #include <linux/string.h> 28 #include <linux/string.h>
29 #include <linux/errno.h> 29 #include <linux/errno.h>
30 #include <linux/skbuff.h> 30 #include <linux/skbuff.h>
31 #include <linux/list.h> 31 #include <linux/list.h>
32 #include <linux/compiler.h> 32 #include <linux/compiler.h>
33 #include <linux/rbtree.h> 33 #include <linux/rbtree.h>
34 #include <linux/workqueue.h> 34 #include <linux/workqueue.h>
35 #include <linux/slab.h> 35 #include <linux/slab.h>
36 #include <net/netlink.h> 36 #include <net/netlink.h>
37 #include <net/sch_generic.h> 37 #include <net/sch_generic.h>
38 #include <net/pkt_sched.h> 38 #include <net/pkt_sched.h>
39 #include <net/pkt_cls.h> 39 #include <net/pkt_cls.h>
40 40
41 /* HTB algorithm. 41 /* HTB algorithm.
42 Author: devik@cdi.cz 42 Author: devik@cdi.cz
43 ========================================== 43 ========================================================================
44 HTB is like TBF with multiple classes. It 44 HTB is like TBF with multiple classes. It is also similar to CBQ because
45 it allows to assign priority to each class 45 it allows to assign priority to each class in hierarchy.
46 In fact it is another implementation of Fl 46 In fact it is another implementation of Floyd's formal sharing.
47 47
48 Levels: 48 Levels:
49 Each class is assigned level. Leaf has ALW 49 Each class is assigned level. Leaf has ALWAYS level 0 and root
50 classes have level TC_HTB_MAXDEPTH-1. Inte 50 classes have level TC_HTB_MAXDEPTH-1. Interior nodes has level
51 one less than their parent. 51 one less than their parent.
52 */ 52 */
53 53
54 static int htb_hysteresis __read_mostly = 0; / 54 static int htb_hysteresis __read_mostly = 0; /* whether to use mode hysteresis for speedup */
55 #define HTB_VER 0x30011 /* major must !! 55 #define HTB_VER 0x30011 /* major must be matched with number suplied by TC as version */
56 56
57 #if HTB_VER >> 16 != TC_HTB_PROTOVER 57 #if HTB_VER >> 16 != TC_HTB_PROTOVER
58 #error "Mismatched sch_htb.c and pkt_sch.h" 58 #error "Mismatched sch_htb.c and pkt_sch.h"
59 #endif 59 #endif
60 60
61 /* Module parameter and sysfs export */ 61 /* Module parameter and sysfs export */
62 module_param (htb_hysteresis, int, 0640); 62 module_param (htb_hysteresis, int, 0640);
63 MODULE_PARM_DESC(htb_hysteresis, "Hysteresis m 63 MODULE_PARM_DESC(htb_hysteresis, "Hysteresis mode, less CPU load, less accurate");
64 64
65 static int htb_rate_est = 0; /* htb classes ha 65 static int htb_rate_est = 0; /* htb classes have a default rate estimator */
66 module_param(htb_rate_est, int, 0640); 66 module_param(htb_rate_est, int, 0640);
67 MODULE_PARM_DESC(htb_rate_est, "setup a defaul 67 MODULE_PARM_DESC(htb_rate_est, "setup a default rate estimator (4sec 16sec) for htb classes");
68 68
69 /* used internaly to keep status of single cla 69 /* used internaly to keep status of single class */
70 enum htb_cmode { 70 enum htb_cmode {
71 HTB_CANT_SEND, /* class can't 71 HTB_CANT_SEND, /* class can't send and can't borrow */
72 HTB_MAY_BORROW, /* class can't 72 HTB_MAY_BORROW, /* class can't send but may borrow */
73 HTB_CAN_SEND /* class can s 73 HTB_CAN_SEND /* class can send */
74 }; 74 };
75 75
76 struct htb_prio { 76 struct htb_prio {
77 union { 77 union {
78 struct rb_root row; 78 struct rb_root row;
79 struct rb_root feed; 79 struct rb_root feed;
80 }; 80 };
81 struct rb_node *ptr; 81 struct rb_node *ptr;
82 /* When class changes from state 1->2 82 /* When class changes from state 1->2 and disconnects from
83 * parent's feed then we lost ptr valu 83 * parent's feed then we lost ptr value and start from the
84 * first child again. Here we store cl 84 * first child again. Here we store classid of the
85 * last valid ptr (used when ptr is NU 85 * last valid ptr (used when ptr is NULL).
86 */ 86 */
87 u32 last_ptr_id; 87 u32 last_ptr_id;
88 }; 88 };
89 89
90 /* interior & leaf nodes; props specific to le 90 /* interior & leaf nodes; props specific to leaves are marked L:
91 * To reduce false sharing, place mostly read 91 * To reduce false sharing, place mostly read fields at beginning,
92 * and mostly written ones at the end. 92 * and mostly written ones at the end.
93 */ 93 */
94 struct htb_class { 94 struct htb_class {
95 struct Qdisc_class_common common; 95 struct Qdisc_class_common common;
96 struct psched_ratecfg rate; 96 struct psched_ratecfg rate;
97 struct psched_ratecfg ceil; 97 struct psched_ratecfg ceil;
98 s64 buffer, cbuffe 98 s64 buffer, cbuffer;/* token bucket depth/rate */
99 s64 mbuffer; 99 s64 mbuffer; /* max wait time */
100 u32 prio; 100 u32 prio; /* these two are used only by leaves... */
101 int quantum; 101 int quantum; /* but stored for parent-to-leaf return */
102 102
103 struct tcf_proto __rcu *filter_list; 103 struct tcf_proto __rcu *filter_list; /* class attached filters */
104 struct tcf_block *block; 104 struct tcf_block *block;
>> 105 int filter_cnt;
105 106
106 int level; 107 int level; /* our level (see above) */
107 unsigned int children; 108 unsigned int children;
108 struct htb_class *parent; 109 struct htb_class *parent; /* parent class */
109 110
110 struct net_rate_estimator __rcu *rate_ 111 struct net_rate_estimator __rcu *rate_est;
111 112
112 /* 113 /*
113 * Written often fields 114 * Written often fields
114 */ 115 */
115 struct gnet_stats_basic_sync bstats; !! 116 struct gnet_stats_basic_packed bstats;
116 struct gnet_stats_basic_sync bstats_bi <<
117 struct tc_htb_xstats xstats; /* our 117 struct tc_htb_xstats xstats; /* our special stats */
118 118
119 /* token bucket parameters */ 119 /* token bucket parameters */
120 s64 tokens, ctoken 120 s64 tokens, ctokens;/* current number of tokens */
121 s64 t_c; 121 s64 t_c; /* checkpoint time */
122 122
123 union { 123 union {
124 struct htb_class_leaf { 124 struct htb_class_leaf {
125 int defici 125 int deficit[TC_HTB_MAXDEPTH];
126 struct Qdisc *q; 126 struct Qdisc *q;
127 struct netdev_queue *o <<
128 } leaf; 127 } leaf;
129 struct htb_class_inner { 128 struct htb_class_inner {
130 struct htb_prio clprio 129 struct htb_prio clprio[TC_HTB_NUMPRIO];
131 } inner; 130 } inner;
132 }; 131 };
133 s64 pq_key; 132 s64 pq_key;
134 133
135 int prio_activity; 134 int prio_activity; /* for which prios are we active */
136 enum htb_cmode cmode; 135 enum htb_cmode cmode; /* current mode of the class */
137 struct rb_node pq_node; 136 struct rb_node pq_node; /* node for event queue */
138 struct rb_node node[TC_HTB_NU 137 struct rb_node node[TC_HTB_NUMPRIO]; /* node for self or feed tree */
139 138
140 unsigned int drops ____cacheline_align 139 unsigned int drops ____cacheline_aligned_in_smp;
141 unsigned int overlimits; 140 unsigned int overlimits;
142 }; 141 };
143 142
144 struct htb_level { 143 struct htb_level {
145 struct rb_root wait_pq; 144 struct rb_root wait_pq;
146 struct htb_prio hprio[TC_HTB_NUMPRIO]; 145 struct htb_prio hprio[TC_HTB_NUMPRIO];
147 }; 146 };
148 147
149 struct htb_sched { 148 struct htb_sched {
150 struct Qdisc_class_hash clhash; 149 struct Qdisc_class_hash clhash;
151 int defcls; 150 int defcls; /* class where unclassified flows go to */
152 int rate2quantum; 151 int rate2quantum; /* quant = rate / rate2quantum */
153 152
154 /* filters for qdisc itself */ 153 /* filters for qdisc itself */
155 struct tcf_proto __rcu *filter_list; 154 struct tcf_proto __rcu *filter_list;
156 struct tcf_block *block; 155 struct tcf_block *block;
157 156
158 #define HTB_WARN_TOOMANYEVENTS 0x1 157 #define HTB_WARN_TOOMANYEVENTS 0x1
159 unsigned int warned; /* onl 158 unsigned int warned; /* only one warning */
160 int direct_qlen; 159 int direct_qlen;
161 struct work_struct work; 160 struct work_struct work;
162 161
163 /* non shaped skbs; let them go direct 162 /* non shaped skbs; let them go directly thru */
164 struct qdisc_skb_head direct_queue; 163 struct qdisc_skb_head direct_queue;
165 u32 direct_pkts; 164 u32 direct_pkts;
166 u32 overlimits; 165 u32 overlimits;
167 166
168 struct qdisc_watchdog watchdog; 167 struct qdisc_watchdog watchdog;
169 168
170 s64 now; /* cac 169 s64 now; /* cached dequeue time */
171 170
172 /* time of nearest event per level (ro 171 /* time of nearest event per level (row) */
173 s64 near_ev_cache[ 172 s64 near_ev_cache[TC_HTB_MAXDEPTH];
174 173
175 int row_mask[TC_HT 174 int row_mask[TC_HTB_MAXDEPTH];
176 175
177 struct htb_level hlevel[TC_HTB_ 176 struct htb_level hlevel[TC_HTB_MAXDEPTH];
178 <<
179 struct Qdisc **direct_qdisc <<
180 unsigned int num_direct_qdi <<
181 <<
182 bool offload; <<
183 }; 177 };
184 178
185 /* find class in global hash table using given 179 /* find class in global hash table using given handle */
186 static inline struct htb_class *htb_find(u32 h 180 static inline struct htb_class *htb_find(u32 handle, struct Qdisc *sch)
187 { 181 {
188 struct htb_sched *q = qdisc_priv(sch); 182 struct htb_sched *q = qdisc_priv(sch);
189 struct Qdisc_class_common *clc; 183 struct Qdisc_class_common *clc;
190 184
191 clc = qdisc_class_find(&q->clhash, han 185 clc = qdisc_class_find(&q->clhash, handle);
192 if (clc == NULL) 186 if (clc == NULL)
193 return NULL; 187 return NULL;
194 return container_of(clc, struct htb_cl 188 return container_of(clc, struct htb_class, common);
195 } 189 }
196 190
197 static unsigned long htb_search(struct Qdisc * 191 static unsigned long htb_search(struct Qdisc *sch, u32 handle)
198 { 192 {
199 return (unsigned long)htb_find(handle, 193 return (unsigned long)htb_find(handle, sch);
200 } 194 }
201 <<
202 #define HTB_DIRECT ((struct htb_class *)-1L) <<
203 <<
204 /** 195 /**
205 * htb_classify - classify a packet into class 196 * htb_classify - classify a packet into class
206 * @skb: the socket buffer <<
207 * @sch: the active queue discipline <<
208 * @qerr: pointer for returned status code <<
209 * 197 *
210 * It returns NULL if the packet should be dro 198 * It returns NULL if the packet should be dropped or -1 if the packet
211 * should be passed directly thru. In all othe 199 * should be passed directly thru. In all other cases leaf class is returned.
212 * We allow direct class selection by classid 200 * We allow direct class selection by classid in priority. The we examine
213 * filters in qdisc and in inner nodes (if hig 201 * filters in qdisc and in inner nodes (if higher filter points to the inner
214 * node). If we end up with classid MAJOR:0 we 202 * node). If we end up with classid MAJOR:0 we enqueue the skb into special
215 * internal fifo (direct). These packets then 203 * internal fifo (direct). These packets then go directly thru. If we still
216 * have no valid leaf we try to use MAJOR:defa 204 * have no valid leaf we try to use MAJOR:default leaf. It still unsuccessful
217 * then finish and return direct queue. 205 * then finish and return direct queue.
218 */ 206 */
>> 207 #define HTB_DIRECT ((struct htb_class *)-1L)
>> 208
219 static struct htb_class *htb_classify(struct s 209 static struct htb_class *htb_classify(struct sk_buff *skb, struct Qdisc *sch,
220 int *qer 210 int *qerr)
221 { 211 {
222 struct htb_sched *q = qdisc_priv(sch); 212 struct htb_sched *q = qdisc_priv(sch);
223 struct htb_class *cl; 213 struct htb_class *cl;
224 struct tcf_result res; 214 struct tcf_result res;
225 struct tcf_proto *tcf; 215 struct tcf_proto *tcf;
226 int result; 216 int result;
227 217
228 /* allow to select class by setting sk 218 /* allow to select class by setting skb->priority to valid classid;
229 * note that nfmark can be used too by 219 * note that nfmark can be used too by attaching filter fw with no
230 * rules in it 220 * rules in it
231 */ 221 */
232 if (skb->priority == sch->handle) 222 if (skb->priority == sch->handle)
233 return HTB_DIRECT; /* X:0 223 return HTB_DIRECT; /* X:0 (direct flow) selected */
234 cl = htb_find(skb->priority, sch); 224 cl = htb_find(skb->priority, sch);
235 if (cl) { 225 if (cl) {
236 if (cl->level == 0) 226 if (cl->level == 0)
237 return cl; 227 return cl;
238 /* Start with inner filter cha 228 /* Start with inner filter chain if a non-leaf class is selected */
239 tcf = rcu_dereference_bh(cl->f 229 tcf = rcu_dereference_bh(cl->filter_list);
240 } else { 230 } else {
241 tcf = rcu_dereference_bh(q->fi 231 tcf = rcu_dereference_bh(q->filter_list);
242 } 232 }
243 233
244 *qerr = NET_XMIT_SUCCESS | __NET_XMIT_ 234 *qerr = NET_XMIT_SUCCESS | __NET_XMIT_BYPASS;
245 while (tcf && (result = tcf_classify(s !! 235 while (tcf && (result = tcf_classify(skb, tcf, &res, false)) >= 0) {
246 #ifdef CONFIG_NET_CLS_ACT 236 #ifdef CONFIG_NET_CLS_ACT
247 switch (result) { 237 switch (result) {
248 case TC_ACT_QUEUED: 238 case TC_ACT_QUEUED:
249 case TC_ACT_STOLEN: 239 case TC_ACT_STOLEN:
250 case TC_ACT_TRAP: 240 case TC_ACT_TRAP:
251 *qerr = NET_XMIT_SUCCE 241 *qerr = NET_XMIT_SUCCESS | __NET_XMIT_STOLEN;
252 fallthrough; !! 242 /* fall through */
253 case TC_ACT_SHOT: 243 case TC_ACT_SHOT:
254 return NULL; 244 return NULL;
255 } 245 }
256 #endif 246 #endif
257 cl = (void *)res.class; 247 cl = (void *)res.class;
258 if (!cl) { 248 if (!cl) {
259 if (res.classid == sch 249 if (res.classid == sch->handle)
260 return HTB_DIR 250 return HTB_DIRECT; /* X:0 (direct flow) */
261 cl = htb_find(res.clas 251 cl = htb_find(res.classid, sch);
262 if (!cl) 252 if (!cl)
263 break; /* fil 253 break; /* filter selected invalid classid */
264 } 254 }
265 if (!cl->level) 255 if (!cl->level)
266 return cl; /* we 256 return cl; /* we hit leaf; return it */
267 257
268 /* we have got inner class; ap 258 /* we have got inner class; apply inner filter chain */
269 tcf = rcu_dereference_bh(cl->f 259 tcf = rcu_dereference_bh(cl->filter_list);
270 } 260 }
271 /* classification failed; try to use d 261 /* classification failed; try to use default class */
272 cl = htb_find(TC_H_MAKE(TC_H_MAJ(sch-> 262 cl = htb_find(TC_H_MAKE(TC_H_MAJ(sch->handle), q->defcls), sch);
273 if (!cl || cl->level) 263 if (!cl || cl->level)
274 return HTB_DIRECT; /* bad 264 return HTB_DIRECT; /* bad default .. this is safe bet */
275 return cl; 265 return cl;
276 } 266 }
277 267
278 /** 268 /**
279 * htb_add_to_id_tree - adds class to the roun 269 * htb_add_to_id_tree - adds class to the round robin list
280 * @root: the root of the tree <<
281 * @cl: the class to add <<
282 * @prio: the give prio in class <<
283 * 270 *
284 * Routine adds class to the list (actually tr 271 * Routine adds class to the list (actually tree) sorted by classid.
285 * Make sure that class is not already on such 272 * Make sure that class is not already on such list for given prio.
286 */ 273 */
287 static void htb_add_to_id_tree(struct rb_root 274 static void htb_add_to_id_tree(struct rb_root *root,
288 struct htb_clas 275 struct htb_class *cl, int prio)
289 { 276 {
290 struct rb_node **p = &root->rb_node, * 277 struct rb_node **p = &root->rb_node, *parent = NULL;
291 278
292 while (*p) { 279 while (*p) {
293 struct htb_class *c; 280 struct htb_class *c;
294 parent = *p; 281 parent = *p;
295 c = rb_entry(parent, struct ht 282 c = rb_entry(parent, struct htb_class, node[prio]);
296 283
297 if (cl->common.classid > c->co 284 if (cl->common.classid > c->common.classid)
298 p = &parent->rb_right; 285 p = &parent->rb_right;
299 else 286 else
300 p = &parent->rb_left; 287 p = &parent->rb_left;
301 } 288 }
302 rb_link_node(&cl->node[prio], parent, 289 rb_link_node(&cl->node[prio], parent, p);
303 rb_insert_color(&cl->node[prio], root) 290 rb_insert_color(&cl->node[prio], root);
304 } 291 }
305 292
306 /** 293 /**
307 * htb_add_to_wait_tree - adds class to the ev 294 * htb_add_to_wait_tree - adds class to the event queue with delay
308 * @q: the priority event queue <<
309 * @cl: the class to add <<
310 * @delay: delay in microseconds <<
311 * 295 *
312 * The class is added to priority event queue 296 * The class is added to priority event queue to indicate that class will
313 * change its mode in cl->pq_key microseconds. 297 * change its mode in cl->pq_key microseconds. Make sure that class is not
314 * already in the queue. 298 * already in the queue.
315 */ 299 */
316 static void htb_add_to_wait_tree(struct htb_sc 300 static void htb_add_to_wait_tree(struct htb_sched *q,
317 struct htb_cl 301 struct htb_class *cl, s64 delay)
318 { 302 {
319 struct rb_node **p = &q->hlevel[cl->le 303 struct rb_node **p = &q->hlevel[cl->level].wait_pq.rb_node, *parent = NULL;
320 304
321 cl->pq_key = q->now + delay; 305 cl->pq_key = q->now + delay;
322 if (cl->pq_key == q->now) 306 if (cl->pq_key == q->now)
323 cl->pq_key++; 307 cl->pq_key++;
324 308
325 /* update the nearest event cache */ 309 /* update the nearest event cache */
326 if (q->near_ev_cache[cl->level] > cl-> 310 if (q->near_ev_cache[cl->level] > cl->pq_key)
327 q->near_ev_cache[cl->level] = 311 q->near_ev_cache[cl->level] = cl->pq_key;
328 312
329 while (*p) { 313 while (*p) {
330 struct htb_class *c; 314 struct htb_class *c;
331 parent = *p; 315 parent = *p;
332 c = rb_entry(parent, struct ht 316 c = rb_entry(parent, struct htb_class, pq_node);
333 if (cl->pq_key >= c->pq_key) 317 if (cl->pq_key >= c->pq_key)
334 p = &parent->rb_right; 318 p = &parent->rb_right;
335 else 319 else
336 p = &parent->rb_left; 320 p = &parent->rb_left;
337 } 321 }
338 rb_link_node(&cl->pq_node, parent, p); 322 rb_link_node(&cl->pq_node, parent, p);
339 rb_insert_color(&cl->pq_node, &q->hlev 323 rb_insert_color(&cl->pq_node, &q->hlevel[cl->level].wait_pq);
340 } 324 }
341 325
342 /** 326 /**
343 * htb_next_rb_node - finds next node in binar 327 * htb_next_rb_node - finds next node in binary tree
344 * @n: the current node in binary tree <<
345 * 328 *
346 * When we are past last key we return NULL. 329 * When we are past last key we return NULL.
347 * Average complexity is 2 steps per call. 330 * Average complexity is 2 steps per call.
348 */ 331 */
349 static inline void htb_next_rb_node(struct rb_ 332 static inline void htb_next_rb_node(struct rb_node **n)
350 { 333 {
351 *n = rb_next(*n); 334 *n = rb_next(*n);
352 } 335 }
353 336
354 /** 337 /**
355 * htb_add_class_to_row - add class to its row 338 * htb_add_class_to_row - add class to its row
356 * @q: the priority event queue <<
357 * @cl: the class to add <<
358 * @mask: the given priorities in class in bit <<
359 * 339 *
360 * The class is added to row at priorities mar 340 * The class is added to row at priorities marked in mask.
361 * It does nothing if mask == 0. 341 * It does nothing if mask == 0.
362 */ 342 */
363 static inline void htb_add_class_to_row(struct 343 static inline void htb_add_class_to_row(struct htb_sched *q,
364 struct 344 struct htb_class *cl, int mask)
365 { 345 {
366 q->row_mask[cl->level] |= mask; 346 q->row_mask[cl->level] |= mask;
367 while (mask) { 347 while (mask) {
368 int prio = ffz(~mask); 348 int prio = ffz(~mask);
369 mask &= ~(1 << prio); 349 mask &= ~(1 << prio);
370 htb_add_to_id_tree(&q->hlevel[ 350 htb_add_to_id_tree(&q->hlevel[cl->level].hprio[prio].row, cl, prio);
371 } 351 }
372 } 352 }
373 353
374 /* If this triggers, it is a bug in this code, 354 /* If this triggers, it is a bug in this code, but it need not be fatal */
375 static void htb_safe_rb_erase(struct rb_node * 355 static void htb_safe_rb_erase(struct rb_node *rb, struct rb_root *root)
376 { 356 {
377 if (RB_EMPTY_NODE(rb)) { 357 if (RB_EMPTY_NODE(rb)) {
378 WARN_ON(1); 358 WARN_ON(1);
379 } else { 359 } else {
380 rb_erase(rb, root); 360 rb_erase(rb, root);
381 RB_CLEAR_NODE(rb); 361 RB_CLEAR_NODE(rb);
382 } 362 }
383 } 363 }
384 364
385 365
386 /** 366 /**
387 * htb_remove_class_from_row - removes class f 367 * htb_remove_class_from_row - removes class from its row
388 * @q: the priority event queue <<
389 * @cl: the class to add <<
390 * @mask: the given priorities in class in bit <<
391 * 368 *
392 * The class is removed from row at priorities 369 * The class is removed from row at priorities marked in mask.
393 * It does nothing if mask == 0. 370 * It does nothing if mask == 0.
394 */ 371 */
395 static inline void htb_remove_class_from_row(s 372 static inline void htb_remove_class_from_row(struct htb_sched *q,
396 373 struct htb_class *cl, int mask)
397 { 374 {
398 int m = 0; 375 int m = 0;
399 struct htb_level *hlevel = &q->hlevel[ 376 struct htb_level *hlevel = &q->hlevel[cl->level];
400 377
401 while (mask) { 378 while (mask) {
402 int prio = ffz(~mask); 379 int prio = ffz(~mask);
403 struct htb_prio *hprio = &hlev 380 struct htb_prio *hprio = &hlevel->hprio[prio];
404 381
405 mask &= ~(1 << prio); 382 mask &= ~(1 << prio);
406 if (hprio->ptr == cl->node + p 383 if (hprio->ptr == cl->node + prio)
407 htb_next_rb_node(&hpri 384 htb_next_rb_node(&hprio->ptr);
408 385
409 htb_safe_rb_erase(cl->node + p 386 htb_safe_rb_erase(cl->node + prio, &hprio->row);
410 if (!hprio->row.rb_node) 387 if (!hprio->row.rb_node)
411 m |= 1 << prio; 388 m |= 1 << prio;
412 } 389 }
413 q->row_mask[cl->level] &= ~m; 390 q->row_mask[cl->level] &= ~m;
414 } 391 }
415 392
416 /** 393 /**
417 * htb_activate_prios - creates active classe' 394 * htb_activate_prios - creates active classe's feed chain
418 * @q: the priority event queue <<
419 * @cl: the class to activate <<
420 * 395 *
421 * The class is connected to ancestors and/or 396 * The class is connected to ancestors and/or appropriate rows
422 * for priorities it is participating on. cl-> 397 * for priorities it is participating on. cl->cmode must be new
423 * (activated) mode. It does nothing if cl->pr 398 * (activated) mode. It does nothing if cl->prio_activity == 0.
424 */ 399 */
425 static void htb_activate_prios(struct htb_sche 400 static void htb_activate_prios(struct htb_sched *q, struct htb_class *cl)
426 { 401 {
427 struct htb_class *p = cl->parent; 402 struct htb_class *p = cl->parent;
428 long m, mask = cl->prio_activity; 403 long m, mask = cl->prio_activity;
429 404
430 while (cl->cmode == HTB_MAY_BORROW && 405 while (cl->cmode == HTB_MAY_BORROW && p && mask) {
431 m = mask; 406 m = mask;
432 while (m) { 407 while (m) {
433 unsigned int prio = ff 408 unsigned int prio = ffz(~m);
434 409
435 if (WARN_ON_ONCE(prio 410 if (WARN_ON_ONCE(prio >= ARRAY_SIZE(p->inner.clprio)))
436 break; 411 break;
437 m &= ~(1 << prio); 412 m &= ~(1 << prio);
438 413
439 if (p->inner.clprio[pr 414 if (p->inner.clprio[prio].feed.rb_node)
440 /* parent alre 415 /* parent already has its feed in use so that
441 * reset bit i 416 * reset bit in mask as parent is already ok
442 */ 417 */
443 mask &= ~(1 << 418 mask &= ~(1 << prio);
444 419
445 htb_add_to_id_tree(&p- 420 htb_add_to_id_tree(&p->inner.clprio[prio].feed, cl, prio);
446 } 421 }
447 p->prio_activity |= mask; 422 p->prio_activity |= mask;
448 cl = p; 423 cl = p;
449 p = cl->parent; 424 p = cl->parent;
450 425
451 } 426 }
452 if (cl->cmode == HTB_CAN_SEND && mask) 427 if (cl->cmode == HTB_CAN_SEND && mask)
453 htb_add_class_to_row(q, cl, ma 428 htb_add_class_to_row(q, cl, mask);
454 } 429 }
455 430
456 /** 431 /**
457 * htb_deactivate_prios - remove class from fe 432 * htb_deactivate_prios - remove class from feed chain
458 * @q: the priority event queue <<
459 * @cl: the class to deactivate <<
460 * 433 *
461 * cl->cmode must represent old mode (before d 434 * cl->cmode must represent old mode (before deactivation). It does
462 * nothing if cl->prio_activity == 0. Class is 435 * nothing if cl->prio_activity == 0. Class is removed from all feed
463 * chains and rows. 436 * chains and rows.
464 */ 437 */
465 static void htb_deactivate_prios(struct htb_sc 438 static void htb_deactivate_prios(struct htb_sched *q, struct htb_class *cl)
466 { 439 {
467 struct htb_class *p = cl->parent; 440 struct htb_class *p = cl->parent;
468 long m, mask = cl->prio_activity; 441 long m, mask = cl->prio_activity;
469 442
470 while (cl->cmode == HTB_MAY_BORROW && 443 while (cl->cmode == HTB_MAY_BORROW && p && mask) {
471 m = mask; 444 m = mask;
472 mask = 0; 445 mask = 0;
473 while (m) { 446 while (m) {
474 int prio = ffz(~m); 447 int prio = ffz(~m);
475 m &= ~(1 << prio); 448 m &= ~(1 << prio);
476 449
477 if (p->inner.clprio[pr 450 if (p->inner.clprio[prio].ptr == cl->node + prio) {
478 /* we are remo 451 /* we are removing child which is pointed to from
479 * parent feed 452 * parent feed - forget the pointer but remember
480 * classid 453 * classid
481 */ 454 */
482 p->inner.clpri 455 p->inner.clprio[prio].last_ptr_id = cl->common.classid;
483 p->inner.clpri 456 p->inner.clprio[prio].ptr = NULL;
484 } 457 }
485 458
486 htb_safe_rb_erase(cl-> 459 htb_safe_rb_erase(cl->node + prio,
487 &p-> 460 &p->inner.clprio[prio].feed);
488 461
489 if (!p->inner.clprio[p 462 if (!p->inner.clprio[prio].feed.rb_node)
490 mask |= 1 << p 463 mask |= 1 << prio;
491 } 464 }
492 465
493 p->prio_activity &= ~mask; 466 p->prio_activity &= ~mask;
494 cl = p; 467 cl = p;
495 p = cl->parent; 468 p = cl->parent;
496 469
497 } 470 }
498 if (cl->cmode == HTB_CAN_SEND && mask) 471 if (cl->cmode == HTB_CAN_SEND && mask)
499 htb_remove_class_from_row(q, c 472 htb_remove_class_from_row(q, cl, mask);
500 } 473 }
501 474
502 static inline s64 htb_lowater(const struct htb 475 static inline s64 htb_lowater(const struct htb_class *cl)
503 { 476 {
504 if (htb_hysteresis) 477 if (htb_hysteresis)
505 return cl->cmode != HTB_CANT_S 478 return cl->cmode != HTB_CANT_SEND ? -cl->cbuffer : 0;
506 else 479 else
507 return 0; 480 return 0;
508 } 481 }
509 static inline s64 htb_hiwater(const struct htb 482 static inline s64 htb_hiwater(const struct htb_class *cl)
510 { 483 {
511 if (htb_hysteresis) 484 if (htb_hysteresis)
512 return cl->cmode == HTB_CAN_SE 485 return cl->cmode == HTB_CAN_SEND ? -cl->buffer : 0;
513 else 486 else
514 return 0; 487 return 0;
515 } 488 }
516 489
517 490
518 /** 491 /**
519 * htb_class_mode - computes and returns curre 492 * htb_class_mode - computes and returns current class mode
520 * @cl: the target class <<
521 * @diff: diff time in microseconds <<
522 * 493 *
523 * It computes cl's mode at time cl->t_c+diff 494 * It computes cl's mode at time cl->t_c+diff and returns it. If mode
524 * is not HTB_CAN_SEND then cl->pq_key is upda 495 * is not HTB_CAN_SEND then cl->pq_key is updated to time difference
525 * from now to time when cl will change its st 496 * from now to time when cl will change its state.
526 * Also it is worth to note that class mode do 497 * Also it is worth to note that class mode doesn't change simply
527 * at cl->{c,}tokens == 0 but there can rather 498 * at cl->{c,}tokens == 0 but there can rather be hysteresis of
528 * 0 .. -cl->{c,}buffer range. It is meant to 499 * 0 .. -cl->{c,}buffer range. It is meant to limit number of
529 * mode transitions per time unit. The speed g 500 * mode transitions per time unit. The speed gain is about 1/6.
530 */ 501 */
531 static inline enum htb_cmode 502 static inline enum htb_cmode
532 htb_class_mode(struct htb_class *cl, s64 *diff 503 htb_class_mode(struct htb_class *cl, s64 *diff)
533 { 504 {
534 s64 toks; 505 s64 toks;
535 506
536 if ((toks = (cl->ctokens + *diff)) < h 507 if ((toks = (cl->ctokens + *diff)) < htb_lowater(cl)) {
537 *diff = -toks; 508 *diff = -toks;
538 return HTB_CANT_SEND; 509 return HTB_CANT_SEND;
539 } 510 }
540 511
541 if ((toks = (cl->tokens + *diff)) >= h 512 if ((toks = (cl->tokens + *diff)) >= htb_hiwater(cl))
542 return HTB_CAN_SEND; 513 return HTB_CAN_SEND;
543 514
544 *diff = -toks; 515 *diff = -toks;
545 return HTB_MAY_BORROW; 516 return HTB_MAY_BORROW;
546 } 517 }
547 518
548 /** 519 /**
549 * htb_change_class_mode - changes classe's mo 520 * htb_change_class_mode - changes classe's mode
550 * @q: the priority event queue <<
551 * @cl: the target class <<
552 * @diff: diff time in microseconds <<
553 * 521 *
554 * This should be the only way how to change c 522 * This should be the only way how to change classe's mode under normal
555 * circumstances. Routine will update feed lis !! 523 * cirsumstances. Routine will update feed lists linkage, change mode
556 * and add class to the wait event queue if ap 524 * and add class to the wait event queue if appropriate. New mode should
557 * be different from old one and cl->pq_key ha 525 * be different from old one and cl->pq_key has to be valid if changing
558 * to mode other than HTB_CAN_SEND (see htb_ad 526 * to mode other than HTB_CAN_SEND (see htb_add_to_wait_tree).
559 */ 527 */
560 static void 528 static void
561 htb_change_class_mode(struct htb_sched *q, str 529 htb_change_class_mode(struct htb_sched *q, struct htb_class *cl, s64 *diff)
562 { 530 {
563 enum htb_cmode new_mode = htb_class_mo 531 enum htb_cmode new_mode = htb_class_mode(cl, diff);
564 532
565 if (new_mode == cl->cmode) 533 if (new_mode == cl->cmode)
566 return; 534 return;
567 535
568 if (new_mode == HTB_CANT_SEND) { 536 if (new_mode == HTB_CANT_SEND) {
569 cl->overlimits++; 537 cl->overlimits++;
570 q->overlimits++; 538 q->overlimits++;
571 } 539 }
572 540
573 if (cl->prio_activity) { /* not 541 if (cl->prio_activity) { /* not necessary: speed optimization */
574 if (cl->cmode != HTB_CANT_SEND 542 if (cl->cmode != HTB_CANT_SEND)
575 htb_deactivate_prios(q 543 htb_deactivate_prios(q, cl);
576 cl->cmode = new_mode; 544 cl->cmode = new_mode;
577 if (new_mode != HTB_CANT_SEND) 545 if (new_mode != HTB_CANT_SEND)
578 htb_activate_prios(q, 546 htb_activate_prios(q, cl);
579 } else 547 } else
580 cl->cmode = new_mode; 548 cl->cmode = new_mode;
581 } 549 }
582 550
583 /** 551 /**
584 * htb_activate - inserts leaf cl into appropr 552 * htb_activate - inserts leaf cl into appropriate active feeds
585 * @q: the priority event queue <<
586 * @cl: the target class <<
587 * 553 *
588 * Routine learns (new) priority of leaf and a 554 * Routine learns (new) priority of leaf and activates feed chain
589 * for the prio. It can be called on already a 555 * for the prio. It can be called on already active leaf safely.
590 * It also adds leaf into droplist. 556 * It also adds leaf into droplist.
591 */ 557 */
592 static inline void htb_activate(struct htb_sch 558 static inline void htb_activate(struct htb_sched *q, struct htb_class *cl)
593 { 559 {
594 WARN_ON(cl->level || !cl->leaf.q || !c 560 WARN_ON(cl->level || !cl->leaf.q || !cl->leaf.q->q.qlen);
595 561
596 if (!cl->prio_activity) { 562 if (!cl->prio_activity) {
597 cl->prio_activity = 1 << cl->p 563 cl->prio_activity = 1 << cl->prio;
598 htb_activate_prios(q, cl); 564 htb_activate_prios(q, cl);
599 } 565 }
600 } 566 }
601 567
602 /** 568 /**
603 * htb_deactivate - remove leaf cl from active 569 * htb_deactivate - remove leaf cl from active feeds
604 * @q: the priority event queue <<
605 * @cl: the target class <<
606 * 570 *
607 * Make sure that leaf is active. In the other 571 * Make sure that leaf is active. In the other words it can't be called
608 * with non-active leaf. It also removes class 572 * with non-active leaf. It also removes class from the drop list.
609 */ 573 */
610 static inline void htb_deactivate(struct htb_s 574 static inline void htb_deactivate(struct htb_sched *q, struct htb_class *cl)
611 { 575 {
612 WARN_ON(!cl->prio_activity); 576 WARN_ON(!cl->prio_activity);
613 577
614 htb_deactivate_prios(q, cl); 578 htb_deactivate_prios(q, cl);
615 cl->prio_activity = 0; 579 cl->prio_activity = 0;
616 } 580 }
617 581
618 static int htb_enqueue(struct sk_buff *skb, st 582 static int htb_enqueue(struct sk_buff *skb, struct Qdisc *sch,
619 struct sk_buff **to_fre 583 struct sk_buff **to_free)
620 { 584 {
621 int ret; 585 int ret;
622 unsigned int len = qdisc_pkt_len(skb); 586 unsigned int len = qdisc_pkt_len(skb);
623 struct htb_sched *q = qdisc_priv(sch); 587 struct htb_sched *q = qdisc_priv(sch);
624 struct htb_class *cl = htb_classify(sk 588 struct htb_class *cl = htb_classify(skb, sch, &ret);
625 589
626 if (cl == HTB_DIRECT) { 590 if (cl == HTB_DIRECT) {
627 /* enqueue to helper queue */ 591 /* enqueue to helper queue */
628 if (q->direct_queue.qlen < q-> 592 if (q->direct_queue.qlen < q->direct_qlen) {
629 __qdisc_enqueue_tail(s 593 __qdisc_enqueue_tail(skb, &q->direct_queue);
630 q->direct_pkts++; 594 q->direct_pkts++;
631 } else { 595 } else {
632 return qdisc_drop(skb, 596 return qdisc_drop(skb, sch, to_free);
633 } 597 }
634 #ifdef CONFIG_NET_CLS_ACT 598 #ifdef CONFIG_NET_CLS_ACT
635 } else if (!cl) { 599 } else if (!cl) {
636 if (ret & __NET_XMIT_BYPASS) 600 if (ret & __NET_XMIT_BYPASS)
637 qdisc_qstats_drop(sch) 601 qdisc_qstats_drop(sch);
638 __qdisc_drop(skb, to_free); 602 __qdisc_drop(skb, to_free);
639 return ret; 603 return ret;
640 #endif 604 #endif
641 } else if ((ret = qdisc_enqueue(skb, c 605 } else if ((ret = qdisc_enqueue(skb, cl->leaf.q,
642 to_fre 606 to_free)) != NET_XMIT_SUCCESS) {
643 if (net_xmit_drop_count(ret)) 607 if (net_xmit_drop_count(ret)) {
644 qdisc_qstats_drop(sch) 608 qdisc_qstats_drop(sch);
645 cl->drops++; 609 cl->drops++;
646 } 610 }
647 return ret; 611 return ret;
648 } else { 612 } else {
649 htb_activate(q, cl); 613 htb_activate(q, cl);
650 } 614 }
651 615
652 sch->qstats.backlog += len; 616 sch->qstats.backlog += len;
653 sch->q.qlen++; 617 sch->q.qlen++;
654 return NET_XMIT_SUCCESS; 618 return NET_XMIT_SUCCESS;
655 } 619 }
656 620
657 static inline void htb_accnt_tokens(struct htb 621 static inline void htb_accnt_tokens(struct htb_class *cl, int bytes, s64 diff)
658 { 622 {
659 s64 toks = diff + cl->tokens; 623 s64 toks = diff + cl->tokens;
660 624
661 if (toks > cl->buffer) 625 if (toks > cl->buffer)
662 toks = cl->buffer; 626 toks = cl->buffer;
663 toks -= (s64) psched_l2t_ns(&cl->rate, 627 toks -= (s64) psched_l2t_ns(&cl->rate, bytes);
664 if (toks <= -cl->mbuffer) 628 if (toks <= -cl->mbuffer)
665 toks = 1 - cl->mbuffer; 629 toks = 1 - cl->mbuffer;
666 630
667 cl->tokens = toks; 631 cl->tokens = toks;
668 } 632 }
669 633
670 static inline void htb_accnt_ctokens(struct ht 634 static inline void htb_accnt_ctokens(struct htb_class *cl, int bytes, s64 diff)
671 { 635 {
672 s64 toks = diff + cl->ctokens; 636 s64 toks = diff + cl->ctokens;
673 637
674 if (toks > cl->cbuffer) 638 if (toks > cl->cbuffer)
675 toks = cl->cbuffer; 639 toks = cl->cbuffer;
676 toks -= (s64) psched_l2t_ns(&cl->ceil, 640 toks -= (s64) psched_l2t_ns(&cl->ceil, bytes);
677 if (toks <= -cl->mbuffer) 641 if (toks <= -cl->mbuffer)
678 toks = 1 - cl->mbuffer; 642 toks = 1 - cl->mbuffer;
679 643
680 cl->ctokens = toks; 644 cl->ctokens = toks;
681 } 645 }
682 646
683 /** 647 /**
684 * htb_charge_class - charges amount "bytes" t 648 * htb_charge_class - charges amount "bytes" to leaf and ancestors
685 * @q: the priority event queue <<
686 * @cl: the class to start iterate <<
687 * @level: the minimum level to account <<
688 * @skb: the socket buffer <<
689 * 649 *
690 * Routine assumes that packet "bytes" long wa 650 * Routine assumes that packet "bytes" long was dequeued from leaf cl
691 * borrowing from "level". It accounts bytes t 651 * borrowing from "level". It accounts bytes to ceil leaky bucket for
692 * leaf and all ancestors and to rate bucket f 652 * leaf and all ancestors and to rate bucket for ancestors at levels
693 * "level" and higher. It also handles possibl 653 * "level" and higher. It also handles possible change of mode resulting
694 * from the update. Note that mode can also in 654 * from the update. Note that mode can also increase here (MAY_BORROW to
695 * CAN_SEND) because we can use more precise c 655 * CAN_SEND) because we can use more precise clock that event queue here.
696 * In such case we remove class from event que 656 * In such case we remove class from event queue first.
697 */ 657 */
698 static void htb_charge_class(struct htb_sched 658 static void htb_charge_class(struct htb_sched *q, struct htb_class *cl,
699 int level, struct 659 int level, struct sk_buff *skb)
700 { 660 {
701 int bytes = qdisc_pkt_len(skb); 661 int bytes = qdisc_pkt_len(skb);
702 enum htb_cmode old_mode; 662 enum htb_cmode old_mode;
703 s64 diff; 663 s64 diff;
704 664
705 while (cl) { 665 while (cl) {
706 diff = min_t(s64, q->now - cl- 666 diff = min_t(s64, q->now - cl->t_c, cl->mbuffer);
707 if (cl->level >= level) { 667 if (cl->level >= level) {
708 if (cl->level == level 668 if (cl->level == level)
709 cl->xstats.len 669 cl->xstats.lends++;
710 htb_accnt_tokens(cl, b 670 htb_accnt_tokens(cl, bytes, diff);
711 } else { 671 } else {
712 cl->xstats.borrows++; 672 cl->xstats.borrows++;
713 cl->tokens += diff; 673 cl->tokens += diff; /* we moved t_c; update tokens */
714 } 674 }
715 htb_accnt_ctokens(cl, bytes, d 675 htb_accnt_ctokens(cl, bytes, diff);
716 cl->t_c = q->now; 676 cl->t_c = q->now;
717 677
718 old_mode = cl->cmode; 678 old_mode = cl->cmode;
719 diff = 0; 679 diff = 0;
720 htb_change_class_mode(q, cl, & 680 htb_change_class_mode(q, cl, &diff);
721 if (old_mode != cl->cmode) { 681 if (old_mode != cl->cmode) {
722 if (old_mode != HTB_CA 682 if (old_mode != HTB_CAN_SEND)
723 htb_safe_rb_er 683 htb_safe_rb_erase(&cl->pq_node, &q->hlevel[cl->level].wait_pq);
724 if (cl->cmode != HTB_C 684 if (cl->cmode != HTB_CAN_SEND)
725 htb_add_to_wai 685 htb_add_to_wait_tree(q, cl, diff);
726 } 686 }
727 687
728 /* update basic stats except f 688 /* update basic stats except for leaves which are already updated */
729 if (cl->level) 689 if (cl->level)
730 bstats_update(&cl->bst 690 bstats_update(&cl->bstats, skb);
731 691
732 cl = cl->parent; 692 cl = cl->parent;
733 } 693 }
734 } 694 }
735 695
736 /** 696 /**
737 * htb_do_events - make mode changes to classe 697 * htb_do_events - make mode changes to classes at the level
738 * @q: the priority event queue <<
739 * @level: which wait_pq in 'q->hlevel' <<
740 * @start: start jiffies <<
741 * 698 *
742 * Scans event queue for pending events and ap 699 * Scans event queue for pending events and applies them. Returns time of
743 * next pending event (0 for no event in pq, q 700 * next pending event (0 for no event in pq, q->now for too many events).
744 * Note: Applied are events whose have cl->pq_ 701 * Note: Applied are events whose have cl->pq_key <= q->now.
745 */ 702 */
746 static s64 htb_do_events(struct htb_sched *q, 703 static s64 htb_do_events(struct htb_sched *q, const int level,
747 unsigned long start) 704 unsigned long start)
748 { 705 {
749 /* don't run for longer than 2 jiffies 706 /* don't run for longer than 2 jiffies; 2 is used instead of
750 * 1 to simplify things when jiffy is 707 * 1 to simplify things when jiffy is going to be incremented
751 * too soon 708 * too soon
752 */ 709 */
753 unsigned long stop_at = start + 2; 710 unsigned long stop_at = start + 2;
754 struct rb_root *wait_pq = &q->hlevel[l 711 struct rb_root *wait_pq = &q->hlevel[level].wait_pq;
755 712
756 while (time_before(jiffies, stop_at)) 713 while (time_before(jiffies, stop_at)) {
757 struct htb_class *cl; 714 struct htb_class *cl;
758 s64 diff; 715 s64 diff;
759 struct rb_node *p = rb_first(w 716 struct rb_node *p = rb_first(wait_pq);
760 717
761 if (!p) 718 if (!p)
762 return 0; 719 return 0;
763 720
764 cl = rb_entry(p, struct htb_cl 721 cl = rb_entry(p, struct htb_class, pq_node);
765 if (cl->pq_key > q->now) 722 if (cl->pq_key > q->now)
766 return cl->pq_key; 723 return cl->pq_key;
767 724
768 htb_safe_rb_erase(p, wait_pq); 725 htb_safe_rb_erase(p, wait_pq);
769 diff = min_t(s64, q->now - cl- 726 diff = min_t(s64, q->now - cl->t_c, cl->mbuffer);
770 htb_change_class_mode(q, cl, & 727 htb_change_class_mode(q, cl, &diff);
771 if (cl->cmode != HTB_CAN_SEND) 728 if (cl->cmode != HTB_CAN_SEND)
772 htb_add_to_wait_tree(q 729 htb_add_to_wait_tree(q, cl, diff);
773 } 730 }
774 731
775 /* too much load - let's continue afte 732 /* too much load - let's continue after a break for scheduling */
776 if (!(q->warned & HTB_WARN_TOOMANYEVEN 733 if (!(q->warned & HTB_WARN_TOOMANYEVENTS)) {
777 pr_warn("htb: too many events! 734 pr_warn("htb: too many events!\n");
778 q->warned |= HTB_WARN_TOOMANYE 735 q->warned |= HTB_WARN_TOOMANYEVENTS;
779 } 736 }
780 737
781 return q->now; 738 return q->now;
782 } 739 }
783 740
784 /* Returns class->node+prio from id-tree where 741 /* Returns class->node+prio from id-tree where classe's id is >= id. NULL
785 * is no such one exists. 742 * is no such one exists.
786 */ 743 */
787 static struct rb_node *htb_id_find_next_upper( 744 static struct rb_node *htb_id_find_next_upper(int prio, struct rb_node *n,
788 745 u32 id)
789 { 746 {
790 struct rb_node *r = NULL; 747 struct rb_node *r = NULL;
791 while (n) { 748 while (n) {
792 struct htb_class *cl = 749 struct htb_class *cl =
793 rb_entry(n, struct htb_cla 750 rb_entry(n, struct htb_class, node[prio]);
794 751
795 if (id > cl->common.classid) { 752 if (id > cl->common.classid) {
796 n = n->rb_right; 753 n = n->rb_right;
797 } else if (id < cl->common.cla 754 } else if (id < cl->common.classid) {
798 r = n; 755 r = n;
799 n = n->rb_left; 756 n = n->rb_left;
800 } else { 757 } else {
801 return n; 758 return n;
802 } 759 }
803 } 760 }
804 return r; 761 return r;
805 } 762 }
806 763
807 /** 764 /**
808 * htb_lookup_leaf - returns next leaf class i 765 * htb_lookup_leaf - returns next leaf class in DRR order
809 * @hprio: the current one <<
810 * @prio: which prio in class <<
811 * 766 *
812 * Find leaf where current feed pointers point 767 * Find leaf where current feed pointers points to.
813 */ 768 */
814 static struct htb_class *htb_lookup_leaf(struc 769 static struct htb_class *htb_lookup_leaf(struct htb_prio *hprio, const int prio)
815 { 770 {
816 int i; 771 int i;
817 struct { 772 struct {
818 struct rb_node *root; 773 struct rb_node *root;
819 struct rb_node **pptr; 774 struct rb_node **pptr;
820 u32 *pid; 775 u32 *pid;
821 } stk[TC_HTB_MAXDEPTH], *sp = stk; 776 } stk[TC_HTB_MAXDEPTH], *sp = stk;
822 777
823 BUG_ON(!hprio->row.rb_node); 778 BUG_ON(!hprio->row.rb_node);
824 sp->root = hprio->row.rb_node; 779 sp->root = hprio->row.rb_node;
825 sp->pptr = &hprio->ptr; 780 sp->pptr = &hprio->ptr;
826 sp->pid = &hprio->last_ptr_id; 781 sp->pid = &hprio->last_ptr_id;
827 782
828 for (i = 0; i < 65535; i++) { 783 for (i = 0; i < 65535; i++) {
829 if (!*sp->pptr && *sp->pid) { 784 if (!*sp->pptr && *sp->pid) {
830 /* ptr was invalidated 785 /* ptr was invalidated but id is valid - try to recover
831 * the original or nex 786 * the original or next ptr
832 */ 787 */
833 *sp->pptr = 788 *sp->pptr =
834 htb_id_find_next_u 789 htb_id_find_next_upper(prio, sp->root, *sp->pid);
835 } 790 }
836 *sp->pid = 0; /* ptr is vali 791 *sp->pid = 0; /* ptr is valid now so that remove this hint as it
837 * can become 792 * can become out of date quickly
838 */ 793 */
839 if (!*sp->pptr) { /* we 794 if (!*sp->pptr) { /* we are at right end; rewind & go up */
840 *sp->pptr = sp->root; 795 *sp->pptr = sp->root;
841 while ((*sp->pptr)->rb 796 while ((*sp->pptr)->rb_left)
842 *sp->pptr = (* 797 *sp->pptr = (*sp->pptr)->rb_left;
843 if (sp > stk) { 798 if (sp > stk) {
844 sp--; 799 sp--;
845 if (!*sp->pptr 800 if (!*sp->pptr) {
846 WARN_O 801 WARN_ON(1);
847 return 802 return NULL;
848 } 803 }
849 htb_next_rb_no 804 htb_next_rb_node(sp->pptr);
850 } 805 }
851 } else { 806 } else {
852 struct htb_class *cl; 807 struct htb_class *cl;
853 struct htb_prio *clp; 808 struct htb_prio *clp;
854 809
855 cl = rb_entry(*sp->ppt 810 cl = rb_entry(*sp->pptr, struct htb_class, node[prio]);
856 if (!cl->level) 811 if (!cl->level)
857 return cl; 812 return cl;
858 clp = &cl->inner.clpri 813 clp = &cl->inner.clprio[prio];
859 (++sp)->root = clp->fe 814 (++sp)->root = clp->feed.rb_node;
860 sp->pptr = &clp->ptr; 815 sp->pptr = &clp->ptr;
861 sp->pid = &clp->last_p 816 sp->pid = &clp->last_ptr_id;
862 } 817 }
863 } 818 }
864 WARN_ON(1); 819 WARN_ON(1);
865 return NULL; 820 return NULL;
866 } 821 }
867 822
868 /* dequeues packet at given priority and level 823 /* dequeues packet at given priority and level; call only if
869 * you are sure that there is active class at 824 * you are sure that there is active class at prio/level
870 */ 825 */
871 static struct sk_buff *htb_dequeue_tree(struct 826 static struct sk_buff *htb_dequeue_tree(struct htb_sched *q, const int prio,
872 const 827 const int level)
873 { 828 {
874 struct sk_buff *skb = NULL; 829 struct sk_buff *skb = NULL;
875 struct htb_class *cl, *start; 830 struct htb_class *cl, *start;
876 struct htb_level *hlevel = &q->hlevel[ 831 struct htb_level *hlevel = &q->hlevel[level];
877 struct htb_prio *hprio = &hlevel->hpri 832 struct htb_prio *hprio = &hlevel->hprio[prio];
878 833
879 /* look initial class up in the row */ 834 /* look initial class up in the row */
880 start = cl = htb_lookup_leaf(hprio, pr 835 start = cl = htb_lookup_leaf(hprio, prio);
881 836
882 do { 837 do {
883 next: 838 next:
884 if (unlikely(!cl)) 839 if (unlikely(!cl))
885 return NULL; 840 return NULL;
886 841
887 /* class can be empty - it is 842 /* class can be empty - it is unlikely but can be true if leaf
888 * qdisc drops packets in enqu 843 * qdisc drops packets in enqueue routine or if someone used
889 * graft operation on the leaf 844 * graft operation on the leaf since last dequeue;
890 * simply deactivate and skip 845 * simply deactivate and skip such class
891 */ 846 */
892 if (unlikely(cl->leaf.q->q.qle 847 if (unlikely(cl->leaf.q->q.qlen == 0)) {
893 struct htb_class *next 848 struct htb_class *next;
894 htb_deactivate(q, cl); 849 htb_deactivate(q, cl);
895 850
896 /* row/level might bec 851 /* row/level might become empty */
897 if ((q->row_mask[level 852 if ((q->row_mask[level] & (1 << prio)) == 0)
898 return NULL; 853 return NULL;
899 854
900 next = htb_lookup_leaf 855 next = htb_lookup_leaf(hprio, prio);
901 856
902 if (cl == start) 857 if (cl == start) /* fix start if we just deleted it */
903 start = next; 858 start = next;
904 cl = next; 859 cl = next;
905 goto next; 860 goto next;
906 } 861 }
907 862
908 skb = cl->leaf.q->dequeue(cl-> 863 skb = cl->leaf.q->dequeue(cl->leaf.q);
909 if (likely(skb != NULL)) 864 if (likely(skb != NULL))
910 break; 865 break;
911 866
912 qdisc_warn_nonwc("htb", cl->le 867 qdisc_warn_nonwc("htb", cl->leaf.q);
913 htb_next_rb_node(level ? &cl-> 868 htb_next_rb_node(level ? &cl->parent->inner.clprio[prio].ptr:
914 &q->h 869 &q->hlevel[0].hprio[prio].ptr);
915 cl = htb_lookup_leaf(hprio, pr 870 cl = htb_lookup_leaf(hprio, prio);
916 871
917 } while (cl != start); 872 } while (cl != start);
918 873
919 if (likely(skb != NULL)) { 874 if (likely(skb != NULL)) {
920 bstats_update(&cl->bstats, skb 875 bstats_update(&cl->bstats, skb);
921 cl->leaf.deficit[level] -= qdi 876 cl->leaf.deficit[level] -= qdisc_pkt_len(skb);
922 if (cl->leaf.deficit[level] < 877 if (cl->leaf.deficit[level] < 0) {
923 cl->leaf.deficit[level 878 cl->leaf.deficit[level] += cl->quantum;
924 htb_next_rb_node(level 879 htb_next_rb_node(level ? &cl->parent->inner.clprio[prio].ptr :
925 880 &q->hlevel[0].hprio[prio].ptr);
926 } 881 }
927 /* this used to be after charg 882 /* this used to be after charge_class but this constelation
928 * gives us slightly better pe 883 * gives us slightly better performance
929 */ 884 */
930 if (!cl->leaf.q->q.qlen) 885 if (!cl->leaf.q->q.qlen)
931 htb_deactivate(q, cl); 886 htb_deactivate(q, cl);
932 htb_charge_class(q, cl, level, 887 htb_charge_class(q, cl, level, skb);
933 } 888 }
934 return skb; 889 return skb;
935 } 890 }
936 891
937 static struct sk_buff *htb_dequeue(struct Qdis 892 static struct sk_buff *htb_dequeue(struct Qdisc *sch)
938 { 893 {
939 struct sk_buff *skb; 894 struct sk_buff *skb;
940 struct htb_sched *q = qdisc_priv(sch); 895 struct htb_sched *q = qdisc_priv(sch);
941 int level; 896 int level;
942 s64 next_event; 897 s64 next_event;
943 unsigned long start_at; 898 unsigned long start_at;
944 899
945 /* try to dequeue direct packets as hi 900 /* try to dequeue direct packets as high prio (!) to minimize cpu work */
946 skb = __qdisc_dequeue_head(&q->direct_ 901 skb = __qdisc_dequeue_head(&q->direct_queue);
947 if (skb != NULL) { 902 if (skb != NULL) {
948 ok: 903 ok:
949 qdisc_bstats_update(sch, skb); 904 qdisc_bstats_update(sch, skb);
950 qdisc_qstats_backlog_dec(sch, 905 qdisc_qstats_backlog_dec(sch, skb);
951 sch->q.qlen--; 906 sch->q.qlen--;
952 return skb; 907 return skb;
953 } 908 }
954 909
955 if (!sch->q.qlen) 910 if (!sch->q.qlen)
956 goto fin; 911 goto fin;
957 q->now = ktime_get_ns(); 912 q->now = ktime_get_ns();
958 start_at = jiffies; 913 start_at = jiffies;
959 914
960 next_event = q->now + 5LLU * NSEC_PER_ 915 next_event = q->now + 5LLU * NSEC_PER_SEC;
961 916
962 for (level = 0; level < TC_HTB_MAXDEPT 917 for (level = 0; level < TC_HTB_MAXDEPTH; level++) {
963 /* common case optimization - 918 /* common case optimization - skip event handler quickly */
964 int m; 919 int m;
965 s64 event = q->near_ev_cache[l 920 s64 event = q->near_ev_cache[level];
966 921
967 if (q->now >= event) { 922 if (q->now >= event) {
968 event = htb_do_events( 923 event = htb_do_events(q, level, start_at);
969 if (!event) 924 if (!event)
970 event = q->now 925 event = q->now + NSEC_PER_SEC;
971 q->near_ev_cache[level 926 q->near_ev_cache[level] = event;
972 } 927 }
973 928
974 if (next_event > event) 929 if (next_event > event)
975 next_event = event; 930 next_event = event;
976 931
977 m = ~q->row_mask[level]; 932 m = ~q->row_mask[level];
978 while (m != (int)(-1)) { 933 while (m != (int)(-1)) {
979 int prio = ffz(m); 934 int prio = ffz(m);
980 935
981 m |= 1 << prio; 936 m |= 1 << prio;
982 skb = htb_dequeue_tree 937 skb = htb_dequeue_tree(q, prio, level);
983 if (likely(skb != NULL 938 if (likely(skb != NULL))
984 goto ok; 939 goto ok;
985 } 940 }
986 } 941 }
987 if (likely(next_event > q->now)) 942 if (likely(next_event > q->now))
988 qdisc_watchdog_schedule_ns(&q- 943 qdisc_watchdog_schedule_ns(&q->watchdog, next_event);
989 else 944 else
990 schedule_work(&q->work); 945 schedule_work(&q->work);
991 fin: 946 fin:
992 return skb; 947 return skb;
993 } 948 }
994 949
995 /* reset all classes */ 950 /* reset all classes */
996 /* always caled under BH & queue lock */ 951 /* always caled under BH & queue lock */
997 static void htb_reset(struct Qdisc *sch) 952 static void htb_reset(struct Qdisc *sch)
998 { 953 {
999 struct htb_sched *q = qdisc_priv(sch); 954 struct htb_sched *q = qdisc_priv(sch);
1000 struct htb_class *cl; 955 struct htb_class *cl;
1001 unsigned int i; 956 unsigned int i;
1002 957
1003 for (i = 0; i < q->clhash.hashsize; i 958 for (i = 0; i < q->clhash.hashsize; i++) {
1004 hlist_for_each_entry(cl, &q-> 959 hlist_for_each_entry(cl, &q->clhash.hash[i], common.hnode) {
1005 if (cl->level) 960 if (cl->level)
1006 memset(&cl->i 961 memset(&cl->inner, 0, sizeof(cl->inner));
1007 else { 962 else {
1008 if (cl->leaf. !! 963 if (cl->leaf.q)
1009 qdisc 964 qdisc_reset(cl->leaf.q);
1010 } 965 }
1011 cl->prio_activity = 0 966 cl->prio_activity = 0;
1012 cl->cmode = HTB_CAN_S 967 cl->cmode = HTB_CAN_SEND;
1013 } 968 }
1014 } 969 }
1015 qdisc_watchdog_cancel(&q->watchdog); 970 qdisc_watchdog_cancel(&q->watchdog);
1016 __qdisc_reset_queue(&q->direct_queue) 971 __qdisc_reset_queue(&q->direct_queue);
>> 972 sch->q.qlen = 0;
>> 973 sch->qstats.backlog = 0;
1017 memset(q->hlevel, 0, sizeof(q->hlevel 974 memset(q->hlevel, 0, sizeof(q->hlevel));
1018 memset(q->row_mask, 0, sizeof(q->row_ 975 memset(q->row_mask, 0, sizeof(q->row_mask));
1019 } 976 }
1020 977
1021 static const struct nla_policy htb_policy[TCA 978 static const struct nla_policy htb_policy[TCA_HTB_MAX + 1] = {
1022 [TCA_HTB_PARMS] = { .len = sizeof(str 979 [TCA_HTB_PARMS] = { .len = sizeof(struct tc_htb_opt) },
1023 [TCA_HTB_INIT] = { .len = sizeof(str 980 [TCA_HTB_INIT] = { .len = sizeof(struct tc_htb_glob) },
1024 [TCA_HTB_CTAB] = { .type = NLA_BINAR 981 [TCA_HTB_CTAB] = { .type = NLA_BINARY, .len = TC_RTAB_SIZE },
1025 [TCA_HTB_RTAB] = { .type = NLA_BINAR 982 [TCA_HTB_RTAB] = { .type = NLA_BINARY, .len = TC_RTAB_SIZE },
1026 [TCA_HTB_DIRECT_QLEN] = { .type = NLA 983 [TCA_HTB_DIRECT_QLEN] = { .type = NLA_U32 },
1027 [TCA_HTB_RATE64] = { .type = NLA_U64 984 [TCA_HTB_RATE64] = { .type = NLA_U64 },
1028 [TCA_HTB_CEIL64] = { .type = NLA_U64 985 [TCA_HTB_CEIL64] = { .type = NLA_U64 },
1029 [TCA_HTB_OFFLOAD] = { .type = NLA_FLA <<
1030 }; 986 };
1031 987
1032 static void htb_work_func(struct work_struct 988 static void htb_work_func(struct work_struct *work)
1033 { 989 {
1034 struct htb_sched *q = container_of(wo 990 struct htb_sched *q = container_of(work, struct htb_sched, work);
1035 struct Qdisc *sch = q->watchdog.qdisc 991 struct Qdisc *sch = q->watchdog.qdisc;
1036 992
1037 rcu_read_lock(); 993 rcu_read_lock();
1038 __netif_schedule(qdisc_root(sch)); 994 __netif_schedule(qdisc_root(sch));
1039 rcu_read_unlock(); 995 rcu_read_unlock();
1040 } 996 }
1041 997
1042 static int htb_offload(struct net_device *dev <<
1043 { <<
1044 return dev->netdev_ops->ndo_setup_tc( <<
1045 } <<
1046 <<
1047 static int htb_init(struct Qdisc *sch, struct 998 static int htb_init(struct Qdisc *sch, struct nlattr *opt,
1048 struct netlink_ext_ack *e 999 struct netlink_ext_ack *extack)
1049 { 1000 {
1050 struct net_device *dev = qdisc_dev(sc <<
1051 struct tc_htb_qopt_offload offload_op <<
1052 struct htb_sched *q = qdisc_priv(sch) 1001 struct htb_sched *q = qdisc_priv(sch);
1053 struct nlattr *tb[TCA_HTB_MAX + 1]; 1002 struct nlattr *tb[TCA_HTB_MAX + 1];
1054 struct tc_htb_glob *gopt; 1003 struct tc_htb_glob *gopt;
1055 unsigned int ntx; <<
1056 bool offload; <<
1057 int err; 1004 int err;
1058 1005
1059 qdisc_watchdog_init(&q->watchdog, sch 1006 qdisc_watchdog_init(&q->watchdog, sch);
1060 INIT_WORK(&q->work, htb_work_func); 1007 INIT_WORK(&q->work, htb_work_func);
1061 1008
1062 if (!opt) 1009 if (!opt)
1063 return -EINVAL; 1010 return -EINVAL;
1064 1011
1065 err = tcf_block_get(&q->block, &q->fi 1012 err = tcf_block_get(&q->block, &q->filter_list, sch, extack);
1066 if (err) 1013 if (err)
1067 return err; 1014 return err;
1068 1015
1069 err = nla_parse_nested_deprecated(tb, 1016 err = nla_parse_nested_deprecated(tb, TCA_HTB_MAX, opt, htb_policy,
1070 NUL 1017 NULL);
1071 if (err < 0) 1018 if (err < 0)
1072 return err; 1019 return err;
1073 1020
1074 if (!tb[TCA_HTB_INIT]) 1021 if (!tb[TCA_HTB_INIT])
1075 return -EINVAL; 1022 return -EINVAL;
1076 1023
1077 gopt = nla_data(tb[TCA_HTB_INIT]); 1024 gopt = nla_data(tb[TCA_HTB_INIT]);
1078 if (gopt->version != HTB_VER >> 16) 1025 if (gopt->version != HTB_VER >> 16)
1079 return -EINVAL; 1026 return -EINVAL;
1080 1027
1081 offload = nla_get_flag(tb[TCA_HTB_OFF <<
1082 <<
1083 if (offload) { <<
1084 if (sch->parent != TC_H_ROOT) <<
1085 NL_SET_ERR_MSG(extack <<
1086 return -EOPNOTSUPP; <<
1087 } <<
1088 <<
1089 if (!tc_can_offload(dev) || ! <<
1090 NL_SET_ERR_MSG(extack <<
1091 return -EOPNOTSUPP; <<
1092 } <<
1093 <<
1094 q->num_direct_qdiscs = dev->r <<
1095 q->direct_qdiscs = kcalloc(q- <<
1096 si <<
1097 GF <<
1098 if (!q->direct_qdiscs) <<
1099 return -ENOMEM; <<
1100 } <<
1101 <<
1102 err = qdisc_class_hash_init(&q->clhas 1028 err = qdisc_class_hash_init(&q->clhash);
1103 if (err < 0) 1029 if (err < 0)
1104 return err; 1030 return err;
1105 1031
>> 1032 qdisc_skb_head_init(&q->direct_queue);
>> 1033
1106 if (tb[TCA_HTB_DIRECT_QLEN]) 1034 if (tb[TCA_HTB_DIRECT_QLEN])
1107 q->direct_qlen = nla_get_u32( 1035 q->direct_qlen = nla_get_u32(tb[TCA_HTB_DIRECT_QLEN]);
1108 else 1036 else
1109 q->direct_qlen = qdisc_dev(sc 1037 q->direct_qlen = qdisc_dev(sch)->tx_queue_len;
1110 1038
1111 if ((q->rate2quantum = gopt->rate2qua 1039 if ((q->rate2quantum = gopt->rate2quantum) < 1)
1112 q->rate2quantum = 1; 1040 q->rate2quantum = 1;
1113 q->defcls = gopt->defcls; 1041 q->defcls = gopt->defcls;
1114 1042
1115 if (!offload) <<
1116 return 0; <<
1117 <<
1118 for (ntx = 0; ntx < q->num_direct_qdi <<
1119 struct netdev_queue *dev_queu <<
1120 struct Qdisc *qdisc; <<
1121 <<
1122 qdisc = qdisc_create_dflt(dev <<
1123 TC_ <<
1124 if (!qdisc) { <<
1125 return -ENOMEM; <<
1126 } <<
1127 <<
1128 q->direct_qdiscs[ntx] = qdisc <<
1129 qdisc->flags |= TCQ_F_ONETXQU <<
1130 } <<
1131 <<
1132 sch->flags |= TCQ_F_MQROOT; <<
1133 <<
1134 offload_opt = (struct tc_htb_qopt_off <<
1135 .command = TC_HTB_CREATE, <<
1136 .parent_classid = TC_H_MAJ(sc <<
1137 .classid = TC_H_MIN(q->defcls <<
1138 .extack = extack, <<
1139 }; <<
1140 err = htb_offload(dev, &offload_opt); <<
1141 if (err) <<
1142 return err; <<
1143 <<
1144 /* Defer this assignment, so that htb <<
1145 * parts (especially calling ndo_setu <<
1146 */ <<
1147 q->offload = true; <<
1148 <<
1149 return 0; 1043 return 0;
1150 } 1044 }
1151 1045
1152 static void htb_attach_offload(struct Qdisc * <<
1153 { <<
1154 struct net_device *dev = qdisc_dev(sc <<
1155 struct htb_sched *q = qdisc_priv(sch) <<
1156 unsigned int ntx; <<
1157 <<
1158 for (ntx = 0; ntx < q->num_direct_qdi <<
1159 struct Qdisc *old, *qdisc = q <<
1160 <<
1161 old = dev_graft_qdisc(qdisc-> <<
1162 qdisc_put(old); <<
1163 qdisc_hash_add(qdisc, false); <<
1164 } <<
1165 for (ntx = q->num_direct_qdiscs; ntx <<
1166 struct netdev_queue *dev_queu <<
1167 struct Qdisc *old = dev_graft <<
1168 <<
1169 qdisc_put(old); <<
1170 } <<
1171 <<
1172 kfree(q->direct_qdiscs); <<
1173 q->direct_qdiscs = NULL; <<
1174 } <<
1175 <<
1176 static void htb_attach_software(struct Qdisc <<
1177 { <<
1178 struct net_device *dev = qdisc_dev(sc <<
1179 unsigned int ntx; <<
1180 <<
1181 /* Resemble qdisc_graft behavior. */ <<
1182 for (ntx = 0; ntx < dev->num_tx_queue <<
1183 struct netdev_queue *dev_queu <<
1184 struct Qdisc *old = dev_graft <<
1185 <<
1186 qdisc_refcount_inc(sch); <<
1187 <<
1188 qdisc_put(old); <<
1189 } <<
1190 } <<
1191 <<
1192 static void htb_attach(struct Qdisc *sch) <<
1193 { <<
1194 struct htb_sched *q = qdisc_priv(sch) <<
1195 <<
1196 if (q->offload) <<
1197 htb_attach_offload(sch); <<
1198 else <<
1199 htb_attach_software(sch); <<
1200 } <<
1201 <<
1202 static int htb_dump(struct Qdisc *sch, struct 1046 static int htb_dump(struct Qdisc *sch, struct sk_buff *skb)
1203 { 1047 {
1204 struct htb_sched *q = qdisc_priv(sch) 1048 struct htb_sched *q = qdisc_priv(sch);
1205 struct nlattr *nest; 1049 struct nlattr *nest;
1206 struct tc_htb_glob gopt; 1050 struct tc_htb_glob gopt;
1207 1051
1208 if (q->offload) <<
1209 sch->flags |= TCQ_F_OFFLOADED <<
1210 else <<
1211 sch->flags &= ~TCQ_F_OFFLOADE <<
1212 <<
1213 sch->qstats.overlimits = q->overlimit 1052 sch->qstats.overlimits = q->overlimits;
1214 /* Its safe to not acquire qdisc lock 1053 /* Its safe to not acquire qdisc lock. As we hold RTNL,
1215 * no change can happen on the qdisc 1054 * no change can happen on the qdisc parameters.
1216 */ 1055 */
1217 1056
1218 gopt.direct_pkts = q->direct_pkts; 1057 gopt.direct_pkts = q->direct_pkts;
1219 gopt.version = HTB_VER; 1058 gopt.version = HTB_VER;
1220 gopt.rate2quantum = q->rate2quantum; 1059 gopt.rate2quantum = q->rate2quantum;
1221 gopt.defcls = q->defcls; 1060 gopt.defcls = q->defcls;
1222 gopt.debug = 0; 1061 gopt.debug = 0;
1223 1062
1224 nest = nla_nest_start_noflag(skb, TCA 1063 nest = nla_nest_start_noflag(skb, TCA_OPTIONS);
1225 if (nest == NULL) 1064 if (nest == NULL)
1226 goto nla_put_failure; 1065 goto nla_put_failure;
1227 if (nla_put(skb, TCA_HTB_INIT, sizeof 1066 if (nla_put(skb, TCA_HTB_INIT, sizeof(gopt), &gopt) ||
1228 nla_put_u32(skb, TCA_HTB_DIRECT_Q 1067 nla_put_u32(skb, TCA_HTB_DIRECT_QLEN, q->direct_qlen))
1229 goto nla_put_failure; 1068 goto nla_put_failure;
1230 if (q->offload && nla_put_flag(skb, T <<
1231 goto nla_put_failure; <<
1232 1069
1233 return nla_nest_end(skb, nest); 1070 return nla_nest_end(skb, nest);
1234 1071
1235 nla_put_failure: 1072 nla_put_failure:
1236 nla_nest_cancel(skb, nest); 1073 nla_nest_cancel(skb, nest);
1237 return -1; 1074 return -1;
1238 } 1075 }
1239 1076
1240 static int htb_dump_class(struct Qdisc *sch, 1077 static int htb_dump_class(struct Qdisc *sch, unsigned long arg,
1241 struct sk_buff *skb 1078 struct sk_buff *skb, struct tcmsg *tcm)
1242 { 1079 {
1243 struct htb_class *cl = (struct htb_cl 1080 struct htb_class *cl = (struct htb_class *)arg;
1244 struct htb_sched *q = qdisc_priv(sch) <<
1245 struct nlattr *nest; 1081 struct nlattr *nest;
1246 struct tc_htb_opt opt; 1082 struct tc_htb_opt opt;
1247 1083
1248 /* Its safe to not acquire qdisc lock 1084 /* Its safe to not acquire qdisc lock. As we hold RTNL,
1249 * no change can happen on the class 1085 * no change can happen on the class parameters.
1250 */ 1086 */
1251 tcm->tcm_parent = cl->parent ? cl->pa 1087 tcm->tcm_parent = cl->parent ? cl->parent->common.classid : TC_H_ROOT;
1252 tcm->tcm_handle = cl->common.classid; 1088 tcm->tcm_handle = cl->common.classid;
1253 if (!cl->level && cl->leaf.q) 1089 if (!cl->level && cl->leaf.q)
1254 tcm->tcm_info = cl->leaf.q->h 1090 tcm->tcm_info = cl->leaf.q->handle;
1255 1091
1256 nest = nla_nest_start_noflag(skb, TCA 1092 nest = nla_nest_start_noflag(skb, TCA_OPTIONS);
1257 if (nest == NULL) 1093 if (nest == NULL)
1258 goto nla_put_failure; 1094 goto nla_put_failure;
1259 1095
1260 memset(&opt, 0, sizeof(opt)); 1096 memset(&opt, 0, sizeof(opt));
1261 1097
1262 psched_ratecfg_getrate(&opt.rate, &cl 1098 psched_ratecfg_getrate(&opt.rate, &cl->rate);
1263 opt.buffer = PSCHED_NS2TICKS(cl->buff 1099 opt.buffer = PSCHED_NS2TICKS(cl->buffer);
1264 psched_ratecfg_getrate(&opt.ceil, &cl 1100 psched_ratecfg_getrate(&opt.ceil, &cl->ceil);
1265 opt.cbuffer = PSCHED_NS2TICKS(cl->cbu 1101 opt.cbuffer = PSCHED_NS2TICKS(cl->cbuffer);
1266 opt.quantum = cl->quantum; 1102 opt.quantum = cl->quantum;
1267 opt.prio = cl->prio; 1103 opt.prio = cl->prio;
1268 opt.level = cl->level; 1104 opt.level = cl->level;
1269 if (nla_put(skb, TCA_HTB_PARMS, sizeo 1105 if (nla_put(skb, TCA_HTB_PARMS, sizeof(opt), &opt))
1270 goto nla_put_failure; 1106 goto nla_put_failure;
1271 if (q->offload && nla_put_flag(skb, T <<
1272 goto nla_put_failure; <<
1273 if ((cl->rate.rate_bytes_ps >= (1ULL 1107 if ((cl->rate.rate_bytes_ps >= (1ULL << 32)) &&
1274 nla_put_u64_64bit(skb, TCA_HTB_RA 1108 nla_put_u64_64bit(skb, TCA_HTB_RATE64, cl->rate.rate_bytes_ps,
1275 TCA_HTB_PAD)) 1109 TCA_HTB_PAD))
1276 goto nla_put_failure; 1110 goto nla_put_failure;
1277 if ((cl->ceil.rate_bytes_ps >= (1ULL 1111 if ((cl->ceil.rate_bytes_ps >= (1ULL << 32)) &&
1278 nla_put_u64_64bit(skb, TCA_HTB_CE 1112 nla_put_u64_64bit(skb, TCA_HTB_CEIL64, cl->ceil.rate_bytes_ps,
1279 TCA_HTB_PAD)) 1113 TCA_HTB_PAD))
1280 goto nla_put_failure; 1114 goto nla_put_failure;
1281 1115
1282 return nla_nest_end(skb, nest); 1116 return nla_nest_end(skb, nest);
1283 1117
1284 nla_put_failure: 1118 nla_put_failure:
1285 nla_nest_cancel(skb, nest); 1119 nla_nest_cancel(skb, nest);
1286 return -1; 1120 return -1;
1287 } 1121 }
1288 1122
1289 static void htb_offload_aggregate_stats(struc <<
1290 struc <<
1291 { <<
1292 u64 bytes = 0, packets = 0; <<
1293 struct htb_class *c; <<
1294 unsigned int i; <<
1295 <<
1296 gnet_stats_basic_sync_init(&cl->bstat <<
1297 <<
1298 for (i = 0; i < q->clhash.hashsize; i <<
1299 hlist_for_each_entry(c, &q->c <<
1300 struct htb_class *p = <<
1301 <<
1302 while (p && p->level <<
1303 p = p->parent <<
1304 <<
1305 if (p != cl) <<
1306 continue; <<
1307 <<
1308 bytes += u64_stats_re <<
1309 packets += u64_stats_ <<
1310 if (c->level == 0) { <<
1311 bytes += u64_ <<
1312 packets += u6 <<
1313 } <<
1314 } <<
1315 } <<
1316 _bstats_update(&cl->bstats, bytes, pa <<
1317 } <<
1318 <<
1319 static int 1123 static int
1320 htb_dump_class_stats(struct Qdisc *sch, unsig 1124 htb_dump_class_stats(struct Qdisc *sch, unsigned long arg, struct gnet_dump *d)
1321 { 1125 {
1322 struct htb_class *cl = (struct htb_cl 1126 struct htb_class *cl = (struct htb_class *)arg;
1323 struct htb_sched *q = qdisc_priv(sch) <<
1324 struct gnet_stats_queue qs = { 1127 struct gnet_stats_queue qs = {
1325 .drops = cl->drops, 1128 .drops = cl->drops,
1326 .overlimits = cl->overlimits, 1129 .overlimits = cl->overlimits,
1327 }; 1130 };
1328 __u32 qlen = 0; 1131 __u32 qlen = 0;
1329 1132
1330 if (!cl->level && cl->leaf.q) 1133 if (!cl->level && cl->leaf.q)
1331 qdisc_qstats_qlen_backlog(cl- 1134 qdisc_qstats_qlen_backlog(cl->leaf.q, &qlen, &qs.backlog);
1332 1135
1333 cl->xstats.tokens = clamp_t(s64, PSCH 1136 cl->xstats.tokens = clamp_t(s64, PSCHED_NS2TICKS(cl->tokens),
1334 INT_MIN, 1137 INT_MIN, INT_MAX);
1335 cl->xstats.ctokens = clamp_t(s64, PSC 1138 cl->xstats.ctokens = clamp_t(s64, PSCHED_NS2TICKS(cl->ctokens),
1336 INT_MIN, 1139 INT_MIN, INT_MAX);
1337 1140
1338 if (q->offload) { !! 1141 if (gnet_stats_copy_basic(qdisc_root_sleeping_running(sch),
1339 if (!cl->level) { !! 1142 d, NULL, &cl->bstats) < 0 ||
1340 if (cl->leaf.q) <<
1341 cl->bstats = <<
1342 else <<
1343 gnet_stats_ba <<
1344 _bstats_update(&cl->b <<
1345 u64_st <<
1346 u64_st <<
1347 } else { <<
1348 htb_offload_aggregate <<
1349 } <<
1350 } <<
1351 <<
1352 if (gnet_stats_copy_basic(d, NULL, &c <<
1353 gnet_stats_copy_rate_est(d, &cl-> 1143 gnet_stats_copy_rate_est(d, &cl->rate_est) < 0 ||
1354 gnet_stats_copy_queue(d, NULL, &q 1144 gnet_stats_copy_queue(d, NULL, &qs, qlen) < 0)
1355 return -1; 1145 return -1;
1356 1146
1357 return gnet_stats_copy_app(d, &cl->xs 1147 return gnet_stats_copy_app(d, &cl->xstats, sizeof(cl->xstats));
1358 } 1148 }
1359 1149
1360 static struct netdev_queue * <<
1361 htb_select_queue(struct Qdisc *sch, struct tc <<
1362 { <<
1363 struct net_device *dev = qdisc_dev(sc <<
1364 struct tc_htb_qopt_offload offload_op <<
1365 struct htb_sched *q = qdisc_priv(sch) <<
1366 int err; <<
1367 <<
1368 if (!q->offload) <<
1369 return sch->dev_queue; <<
1370 <<
1371 offload_opt = (struct tc_htb_qopt_off <<
1372 .command = TC_HTB_LEAF_QUERY_ <<
1373 .classid = TC_H_MIN(tcm->tcm_ <<
1374 }; <<
1375 err = htb_offload(dev, &offload_opt); <<
1376 if (err || offload_opt.qid >= dev->nu <<
1377 return NULL; <<
1378 return netdev_get_tx_queue(dev, offlo <<
1379 } <<
1380 <<
1381 static struct Qdisc * <<
1382 htb_graft_helper(struct netdev_queue *dev_que <<
1383 { <<
1384 struct net_device *dev = dev_queue->d <<
1385 struct Qdisc *old_q; <<
1386 <<
1387 if (dev->flags & IFF_UP) <<
1388 dev_deactivate(dev); <<
1389 old_q = dev_graft_qdisc(dev_queue, ne <<
1390 if (new_q) <<
1391 new_q->flags |= TCQ_F_ONETXQU <<
1392 if (dev->flags & IFF_UP) <<
1393 dev_activate(dev); <<
1394 <<
1395 return old_q; <<
1396 } <<
1397 <<
1398 static struct netdev_queue *htb_offload_get_q <<
1399 { <<
1400 struct netdev_queue *queue; <<
1401 <<
1402 queue = cl->leaf.offload_queue; <<
1403 if (!(cl->leaf.q->flags & TCQ_F_BUILT <<
1404 WARN_ON(cl->leaf.q->dev_queue <<
1405 <<
1406 return queue; <<
1407 } <<
1408 <<
1409 static void htb_offload_move_qdisc(struct Qdi <<
1410 struct htb <<
1411 { <<
1412 struct netdev_queue *queue_old, *queu <<
1413 struct net_device *dev = qdisc_dev(sc <<
1414 <<
1415 queue_old = htb_offload_get_queue(cl_ <<
1416 queue_new = htb_offload_get_queue(cl_ <<
1417 <<
1418 if (!destroying) { <<
1419 struct Qdisc *qdisc; <<
1420 <<
1421 if (dev->flags & IFF_UP) <<
1422 dev_deactivate(dev); <<
1423 qdisc = dev_graft_qdisc(queue <<
1424 WARN_ON(qdisc != cl_old->leaf <<
1425 } <<
1426 <<
1427 if (!(cl_old->leaf.q->flags & TCQ_F_B <<
1428 cl_old->leaf.q->dev_queue = q <<
1429 cl_old->leaf.offload_queue = queue_ne <<
1430 <<
1431 if (!destroying) { <<
1432 struct Qdisc *qdisc; <<
1433 <<
1434 qdisc = dev_graft_qdisc(queue <<
1435 if (dev->flags & IFF_UP) <<
1436 dev_activate(dev); <<
1437 WARN_ON(!(qdisc->flags & TCQ_ <<
1438 } <<
1439 } <<
1440 <<
1441 static int htb_graft(struct Qdisc *sch, unsig 1150 static int htb_graft(struct Qdisc *sch, unsigned long arg, struct Qdisc *new,
1442 struct Qdisc **old, stru 1151 struct Qdisc **old, struct netlink_ext_ack *extack)
1443 { 1152 {
1444 struct netdev_queue *dev_queue = sch- <<
1445 struct htb_class *cl = (struct htb_cl 1153 struct htb_class *cl = (struct htb_class *)arg;
1446 struct htb_sched *q = qdisc_priv(sch) <<
1447 struct Qdisc *old_q; <<
1448 1154
1449 if (cl->level) 1155 if (cl->level)
1450 return -EINVAL; 1156 return -EINVAL;
1451 !! 1157 if (new == NULL &&
1452 if (q->offload) !! 1158 (new = qdisc_create_dflt(sch->dev_queue, &pfifo_qdisc_ops,
1453 dev_queue = htb_offload_get_q !! 1159 cl->common.classid, extack)) == NULL)
1454 !! 1160 return -ENOBUFS;
1455 if (!new) { <<
1456 new = qdisc_create_dflt(dev_q <<
1457 cl->c <<
1458 if (!new) <<
1459 return -ENOBUFS; <<
1460 } <<
1461 <<
1462 if (q->offload) { <<
1463 /* One ref for cl->leaf.q, th <<
1464 qdisc_refcount_inc(new); <<
1465 old_q = htb_graft_helper(dev_ <<
1466 } <<
1467 1161
1468 *old = qdisc_replace(sch, new, &cl->l 1162 *old = qdisc_replace(sch, new, &cl->leaf.q);
1469 <<
1470 if (q->offload) { <<
1471 WARN_ON(old_q != *old); <<
1472 qdisc_put(old_q); <<
1473 } <<
1474 <<
1475 return 0; 1163 return 0;
1476 } 1164 }
1477 1165
1478 static struct Qdisc *htb_leaf(struct Qdisc *s 1166 static struct Qdisc *htb_leaf(struct Qdisc *sch, unsigned long arg)
1479 { 1167 {
1480 struct htb_class *cl = (struct htb_cl 1168 struct htb_class *cl = (struct htb_class *)arg;
1481 return !cl->level ? cl->leaf.q : NULL 1169 return !cl->level ? cl->leaf.q : NULL;
1482 } 1170 }
1483 1171
1484 static void htb_qlen_notify(struct Qdisc *sch 1172 static void htb_qlen_notify(struct Qdisc *sch, unsigned long arg)
1485 { 1173 {
1486 struct htb_class *cl = (struct htb_cl 1174 struct htb_class *cl = (struct htb_class *)arg;
1487 1175
1488 htb_deactivate(qdisc_priv(sch), cl); 1176 htb_deactivate(qdisc_priv(sch), cl);
1489 } 1177 }
1490 1178
1491 static inline int htb_parent_last_child(struc 1179 static inline int htb_parent_last_child(struct htb_class *cl)
1492 { 1180 {
1493 if (!cl->parent) 1181 if (!cl->parent)
1494 /* the root class */ 1182 /* the root class */
1495 return 0; 1183 return 0;
1496 if (cl->parent->children > 1) 1184 if (cl->parent->children > 1)
1497 /* not the last child */ 1185 /* not the last child */
1498 return 0; 1186 return 0;
1499 return 1; 1187 return 1;
1500 } 1188 }
1501 1189
1502 static void htb_parent_to_leaf(struct Qdisc * !! 1190 static void htb_parent_to_leaf(struct htb_sched *q, struct htb_class *cl,
1503 struct Qdisc * 1191 struct Qdisc *new_q)
1504 { 1192 {
1505 struct htb_sched *q = qdisc_priv(sch) <<
1506 struct htb_class *parent = cl->parent 1193 struct htb_class *parent = cl->parent;
1507 1194
1508 WARN_ON(cl->level || !cl->leaf.q || c 1195 WARN_ON(cl->level || !cl->leaf.q || cl->prio_activity);
1509 1196
1510 if (parent->cmode != HTB_CAN_SEND) 1197 if (parent->cmode != HTB_CAN_SEND)
1511 htb_safe_rb_erase(&parent->pq 1198 htb_safe_rb_erase(&parent->pq_node,
1512 &q->hlevel[ 1199 &q->hlevel[parent->level].wait_pq);
1513 1200
1514 parent->level = 0; 1201 parent->level = 0;
1515 memset(&parent->inner, 0, sizeof(pare 1202 memset(&parent->inner, 0, sizeof(parent->inner));
1516 parent->leaf.q = new_q ? new_q : &noo 1203 parent->leaf.q = new_q ? new_q : &noop_qdisc;
1517 parent->tokens = parent->buffer; 1204 parent->tokens = parent->buffer;
1518 parent->ctokens = parent->cbuffer; 1205 parent->ctokens = parent->cbuffer;
1519 parent->t_c = ktime_get_ns(); 1206 parent->t_c = ktime_get_ns();
1520 parent->cmode = HTB_CAN_SEND; 1207 parent->cmode = HTB_CAN_SEND;
1521 if (q->offload) <<
1522 parent->leaf.offload_queue = <<
1523 } <<
1524 <<
1525 static void htb_parent_to_leaf_offload(struct <<
1526 struct <<
1527 struct <<
1528 { <<
1529 struct Qdisc *old_q; <<
1530 <<
1531 /* One ref for cl->leaf.q, the other <<
1532 if (new_q) <<
1533 qdisc_refcount_inc(new_q); <<
1534 old_q = htb_graft_helper(dev_queue, n <<
1535 WARN_ON(!(old_q->flags & TCQ_F_BUILTI <<
1536 } <<
1537 <<
1538 static int htb_destroy_class_offload(struct Q <<
1539 bool las <<
1540 struct n <<
1541 { <<
1542 struct tc_htb_qopt_offload offload_op <<
1543 struct netdev_queue *dev_queue; <<
1544 struct Qdisc *q = cl->leaf.q; <<
1545 struct Qdisc *old; <<
1546 int err; <<
1547 <<
1548 if (cl->level) <<
1549 return -EINVAL; <<
1550 <<
1551 WARN_ON(!q); <<
1552 dev_queue = htb_offload_get_queue(cl) <<
1553 /* When destroying, caller qdisc_graf <<
1554 * qdisc_put for the qdisc being dest <<
1555 * does not need to graft or qdisc_pu <<
1556 */ <<
1557 if (!destroying) { <<
1558 old = htb_graft_helper(dev_qu <<
1559 /* Last qdisc grafted should <<
1560 * calling htb_delete. <<
1561 */ <<
1562 WARN_ON(old != q); <<
1563 } <<
1564 <<
1565 if (cl->parent) { <<
1566 _bstats_update(&cl->parent->b <<
1567 u64_stats_read <<
1568 u64_stats_read <<
1569 } <<
1570 <<
1571 offload_opt = (struct tc_htb_qopt_off <<
1572 .command = !last_child ? TC_H <<
1573 destroying ? TC_HT <<
1574 TC_HTB_LEAF_DEL_LA <<
1575 .classid = cl->common.classid <<
1576 .extack = extack, <<
1577 }; <<
1578 err = htb_offload(qdisc_dev(sch), &of <<
1579 <<
1580 if (!destroying) { <<
1581 if (!err) <<
1582 qdisc_put(old); <<
1583 else <<
1584 htb_graft_helper(dev_ <<
1585 } <<
1586 <<
1587 if (last_child) <<
1588 return err; <<
1589 <<
1590 if (!err && offload_opt.classid != TC <<
1591 u32 classid = TC_H_MAJ(sch->h <<
1592 TC_H_MIN(offloa <<
1593 struct htb_class *moved_cl = <<
1594 <<
1595 htb_offload_move_qdisc(sch, m <<
1596 } <<
1597 <<
1598 return err; <<
1599 } 1208 }
1600 1209
1601 static void htb_destroy_class(struct Qdisc *s 1210 static void htb_destroy_class(struct Qdisc *sch, struct htb_class *cl)
1602 { 1211 {
1603 if (!cl->level) { 1212 if (!cl->level) {
1604 WARN_ON(!cl->leaf.q); 1213 WARN_ON(!cl->leaf.q);
1605 qdisc_put(cl->leaf.q); 1214 qdisc_put(cl->leaf.q);
1606 } 1215 }
1607 gen_kill_estimator(&cl->rate_est); 1216 gen_kill_estimator(&cl->rate_est);
1608 tcf_block_put(cl->block); 1217 tcf_block_put(cl->block);
1609 kfree(cl); 1218 kfree(cl);
1610 } 1219 }
1611 1220
1612 static void htb_destroy(struct Qdisc *sch) 1221 static void htb_destroy(struct Qdisc *sch)
1613 { 1222 {
1614 struct net_device *dev = qdisc_dev(sc <<
1615 struct tc_htb_qopt_offload offload_op <<
1616 struct htb_sched *q = qdisc_priv(sch) 1223 struct htb_sched *q = qdisc_priv(sch);
1617 struct hlist_node *next; 1224 struct hlist_node *next;
1618 bool nonempty, changed; <<
1619 struct htb_class *cl; 1225 struct htb_class *cl;
1620 unsigned int i; 1226 unsigned int i;
1621 1227
1622 cancel_work_sync(&q->work); 1228 cancel_work_sync(&q->work);
1623 qdisc_watchdog_cancel(&q->watchdog); 1229 qdisc_watchdog_cancel(&q->watchdog);
1624 /* This line used to be after htb_des 1230 /* This line used to be after htb_destroy_class call below
1625 * and surprisingly it worked in 2.4. 1231 * and surprisingly it worked in 2.4. But it must precede it
1626 * because filter need its target cla 1232 * because filter need its target class alive to be able to call
1627 * unbind_filter on it (without Oops) 1233 * unbind_filter on it (without Oops).
1628 */ 1234 */
1629 tcf_block_put(q->block); 1235 tcf_block_put(q->block);
1630 1236
1631 for (i = 0; i < q->clhash.hashsize; i 1237 for (i = 0; i < q->clhash.hashsize; i++) {
1632 hlist_for_each_entry(cl, &q-> 1238 hlist_for_each_entry(cl, &q->clhash.hash[i], common.hnode) {
1633 tcf_block_put(cl->blo 1239 tcf_block_put(cl->block);
1634 cl->block = NULL; 1240 cl->block = NULL;
1635 } 1241 }
1636 } 1242 }
1637 !! 1243 for (i = 0; i < q->clhash.hashsize; i++) {
1638 do { !! 1244 hlist_for_each_entry_safe(cl, next, &q->clhash.hash[i],
1639 nonempty = false; !! 1245 common.hnode)
1640 changed = false; !! 1246 htb_destroy_class(sch, cl);
1641 for (i = 0; i < q->clhash.has !! 1247 }
1642 hlist_for_each_entry_ <<
1643 <<
1644 bool last_chi <<
1645 <<
1646 if (!q->offlo <<
1647 htb_d <<
1648 conti <<
1649 } <<
1650 <<
1651 nonempty = tr <<
1652 <<
1653 if (cl->level <<
1654 conti <<
1655 <<
1656 changed = tru <<
1657 <<
1658 last_child = <<
1659 htb_destroy_c <<
1660 <<
1661 qdisc_class_h <<
1662 <<
1663 if (cl->paren <<
1664 cl->p <<
1665 if (last_chil <<
1666 htb_p <<
1667 htb_destroy_c <<
1668 } <<
1669 } <<
1670 } while (changed); <<
1671 WARN_ON(nonempty); <<
1672 <<
1673 qdisc_class_hash_destroy(&q->clhash); 1248 qdisc_class_hash_destroy(&q->clhash);
1674 __qdisc_reset_queue(&q->direct_queue) 1249 __qdisc_reset_queue(&q->direct_queue);
1675 <<
1676 if (q->offload) { <<
1677 offload_opt = (struct tc_htb_ <<
1678 .command = TC_HTB_DES <<
1679 }; <<
1680 htb_offload(dev, &offload_opt <<
1681 } <<
1682 <<
1683 if (!q->direct_qdiscs) <<
1684 return; <<
1685 for (i = 0; i < q->num_direct_qdiscs <<
1686 qdisc_put(q->direct_qdiscs[i] <<
1687 kfree(q->direct_qdiscs); <<
1688 } 1250 }
1689 1251
1690 static int htb_delete(struct Qdisc *sch, unsi !! 1252 static int htb_delete(struct Qdisc *sch, unsigned long arg)
1691 struct netlink_ext_ack <<
1692 { 1253 {
1693 struct htb_sched *q = qdisc_priv(sch) 1254 struct htb_sched *q = qdisc_priv(sch);
1694 struct htb_class *cl = (struct htb_cl 1255 struct htb_class *cl = (struct htb_class *)arg;
1695 struct Qdisc *new_q = NULL; 1256 struct Qdisc *new_q = NULL;
1696 int last_child = 0; 1257 int last_child = 0;
1697 int err; <<
1698 1258
1699 /* TODO: why don't allow to delete su 1259 /* TODO: why don't allow to delete subtree ? references ? does
1700 * tc subsys guarantee us that in htb 1260 * tc subsys guarantee us that in htb_destroy it holds no class
1701 * refs so that we can remove childre 1261 * refs so that we can remove children safely there ?
1702 */ 1262 */
1703 if (cl->children || qdisc_class_in_us !! 1263 if (cl->children || cl->filter_cnt)
1704 NL_SET_ERR_MSG(extack, "HTB c <<
1705 return -EBUSY; 1264 return -EBUSY;
1706 } <<
1707 <<
1708 if (!cl->level && htb_parent_last_chi <<
1709 last_child = 1; <<
1710 <<
1711 if (q->offload) { <<
1712 err = htb_destroy_class_offlo <<
1713 <<
1714 if (err) <<
1715 return err; <<
1716 } <<
1717 1265
1718 if (last_child) { !! 1266 if (!cl->level && htb_parent_last_child(cl)) {
1719 struct netdev_queue *dev_queu !! 1267 new_q = qdisc_create_dflt(sch->dev_queue, &pfifo_qdisc_ops,
1720 <<
1721 if (q->offload) <<
1722 dev_queue = htb_offlo <<
1723 <<
1724 new_q = qdisc_create_dflt(dev <<
1725 cl- 1268 cl->parent->common.classid,
1726 NUL 1269 NULL);
1727 if (q->offload) !! 1270 last_child = 1;
1728 htb_parent_to_leaf_of <<
1729 } 1271 }
1730 1272
1731 sch_tree_lock(sch); 1273 sch_tree_lock(sch);
1732 1274
1733 if (!cl->level) 1275 if (!cl->level)
1734 qdisc_purge_queue(cl->leaf.q) 1276 qdisc_purge_queue(cl->leaf.q);
1735 1277
1736 /* delete from hash and active; remai 1278 /* delete from hash and active; remainder in destroy_class */
1737 qdisc_class_hash_remove(&q->clhash, & 1279 qdisc_class_hash_remove(&q->clhash, &cl->common);
1738 if (cl->parent) 1280 if (cl->parent)
1739 cl->parent->children--; 1281 cl->parent->children--;
1740 1282
1741 if (cl->prio_activity) 1283 if (cl->prio_activity)
1742 htb_deactivate(q, cl); 1284 htb_deactivate(q, cl);
1743 1285
1744 if (cl->cmode != HTB_CAN_SEND) 1286 if (cl->cmode != HTB_CAN_SEND)
1745 htb_safe_rb_erase(&cl->pq_nod 1287 htb_safe_rb_erase(&cl->pq_node,
1746 &q->hlevel[ 1288 &q->hlevel[cl->level].wait_pq);
1747 1289
1748 if (last_child) 1290 if (last_child)
1749 htb_parent_to_leaf(sch, cl, n !! 1291 htb_parent_to_leaf(q, cl, new_q);
1750 1292
1751 sch_tree_unlock(sch); 1293 sch_tree_unlock(sch);
1752 1294
1753 htb_destroy_class(sch, cl); 1295 htb_destroy_class(sch, cl);
1754 return 0; 1296 return 0;
1755 } 1297 }
1756 1298
1757 static int htb_change_class(struct Qdisc *sch 1299 static int htb_change_class(struct Qdisc *sch, u32 classid,
1758 u32 parentid, str 1300 u32 parentid, struct nlattr **tca,
1759 unsigned long *ar 1301 unsigned long *arg, struct netlink_ext_ack *extack)
1760 { 1302 {
1761 int err = -EINVAL; 1303 int err = -EINVAL;
1762 struct htb_sched *q = qdisc_priv(sch) 1304 struct htb_sched *q = qdisc_priv(sch);
1763 struct htb_class *cl = (struct htb_cl 1305 struct htb_class *cl = (struct htb_class *)*arg, *parent;
1764 struct tc_htb_qopt_offload offload_op <<
1765 struct nlattr *opt = tca[TCA_OPTIONS] 1306 struct nlattr *opt = tca[TCA_OPTIONS];
1766 struct nlattr *tb[TCA_HTB_MAX + 1]; 1307 struct nlattr *tb[TCA_HTB_MAX + 1];
1767 struct Qdisc *parent_qdisc = NULL; 1308 struct Qdisc *parent_qdisc = NULL;
1768 struct netdev_queue *dev_queue; <<
1769 struct tc_htb_opt *hopt; 1309 struct tc_htb_opt *hopt;
1770 u64 rate64, ceil64; 1310 u64 rate64, ceil64;
1771 int warn = 0; 1311 int warn = 0;
1772 1312
1773 /* extract all subattrs from opt attr 1313 /* extract all subattrs from opt attr */
1774 if (!opt) 1314 if (!opt)
1775 goto failure; 1315 goto failure;
1776 1316
1777 err = nla_parse_nested_deprecated(tb, 1317 err = nla_parse_nested_deprecated(tb, TCA_HTB_MAX, opt, htb_policy,
1778 ext !! 1318 NULL);
1779 if (err < 0) 1319 if (err < 0)
1780 goto failure; 1320 goto failure;
1781 1321
1782 err = -EINVAL; 1322 err = -EINVAL;
1783 if (tb[TCA_HTB_PARMS] == NULL) 1323 if (tb[TCA_HTB_PARMS] == NULL)
1784 goto failure; 1324 goto failure;
1785 1325
1786 parent = parentid == TC_H_ROOT ? NULL 1326 parent = parentid == TC_H_ROOT ? NULL : htb_find(parentid, sch);
1787 1327
1788 hopt = nla_data(tb[TCA_HTB_PARMS]); 1328 hopt = nla_data(tb[TCA_HTB_PARMS]);
1789 if (!hopt->rate.rate || !hopt->ceil.r 1329 if (!hopt->rate.rate || !hopt->ceil.rate)
1790 goto failure; 1330 goto failure;
1791 1331
1792 if (q->offload) { <<
1793 /* Options not supported by t <<
1794 if (hopt->rate.overhead || ho <<
1795 NL_SET_ERR_MSG(extack <<
1796 goto failure; <<
1797 } <<
1798 if (hopt->rate.mpu || hopt->c <<
1799 NL_SET_ERR_MSG(extack <<
1800 goto failure; <<
1801 } <<
1802 } <<
1803 <<
1804 /* Keeping backward compatible with r 1332 /* Keeping backward compatible with rate_table based iproute2 tc */
1805 if (hopt->rate.linklayer == TC_LINKLA 1333 if (hopt->rate.linklayer == TC_LINKLAYER_UNAWARE)
1806 qdisc_put_rtab(qdisc_get_rtab 1334 qdisc_put_rtab(qdisc_get_rtab(&hopt->rate, tb[TCA_HTB_RTAB],
1807 1335 NULL));
1808 1336
1809 if (hopt->ceil.linklayer == TC_LINKLA 1337 if (hopt->ceil.linklayer == TC_LINKLAYER_UNAWARE)
1810 qdisc_put_rtab(qdisc_get_rtab 1338 qdisc_put_rtab(qdisc_get_rtab(&hopt->ceil, tb[TCA_HTB_CTAB],
1811 1339 NULL));
1812 1340
1813 rate64 = tb[TCA_HTB_RATE64] ? nla_get <<
1814 ceil64 = tb[TCA_HTB_CEIL64] ? nla_get <<
1815 <<
1816 if (!cl) { /* new class 1341 if (!cl) { /* new class */
1817 struct net_device *dev = qdis !! 1342 struct Qdisc *new_q;
1818 struct Qdisc *new_q, *old_q; <<
1819 int prio; 1343 int prio;
1820 struct { 1344 struct {
1821 struct nlattr 1345 struct nlattr nla;
1822 struct gnet_estimator 1346 struct gnet_estimator opt;
1823 } est = { 1347 } est = {
1824 .nla = { 1348 .nla = {
1825 .nla_len 1349 .nla_len = nla_attr_size(sizeof(est.opt)),
1826 .nla_type 1350 .nla_type = TCA_RATE,
1827 }, 1351 },
1828 .opt = { 1352 .opt = {
1829 /* 4s interva 1353 /* 4s interval, 16s averaging constant */
1830 .interval 1354 .interval = 2,
1831 .ewma_log 1355 .ewma_log = 2,
1832 }, 1356 },
1833 }; 1357 };
1834 1358
1835 /* check for valid classid */ 1359 /* check for valid classid */
1836 if (!classid || TC_H_MAJ(clas 1360 if (!classid || TC_H_MAJ(classid ^ sch->handle) ||
1837 htb_find(classid, sch)) 1361 htb_find(classid, sch))
1838 goto failure; 1362 goto failure;
1839 1363
1840 /* check maximal depth */ 1364 /* check maximal depth */
1841 if (parent && parent->parent 1365 if (parent && parent->parent && parent->parent->level < 2) {
1842 NL_SET_ERR_MSG_MOD(ex !! 1366 pr_err("htb: tree is too deep\n");
1843 goto failure; 1367 goto failure;
1844 } 1368 }
1845 err = -ENOBUFS; 1369 err = -ENOBUFS;
1846 cl = kzalloc(sizeof(*cl), GFP 1370 cl = kzalloc(sizeof(*cl), GFP_KERNEL);
1847 if (!cl) 1371 if (!cl)
1848 goto failure; 1372 goto failure;
1849 1373
1850 gnet_stats_basic_sync_init(&c <<
1851 gnet_stats_basic_sync_init(&c <<
1852 <<
1853 err = tcf_block_get(&cl->bloc 1374 err = tcf_block_get(&cl->block, &cl->filter_list, sch, extack);
1854 if (err) { 1375 if (err) {
1855 kfree(cl); 1376 kfree(cl);
1856 goto failure; 1377 goto failure;
1857 } 1378 }
1858 if (htb_rate_est || tca[TCA_R 1379 if (htb_rate_est || tca[TCA_RATE]) {
1859 err = gen_new_estimat 1380 err = gen_new_estimator(&cl->bstats, NULL,
1860 1381 &cl->rate_est,
1861 1382 NULL,
1862 !! 1383 qdisc_root_sleeping_running(sch),
1863 1384 tca[TCA_RATE] ? : &est.nla);
1864 if (err) !! 1385 if (err) {
1865 goto err_bloc !! 1386 tcf_block_put(cl->block);
>> 1387 kfree(cl);
>> 1388 goto failure;
>> 1389 }
1866 } 1390 }
1867 1391
1868 cl->children = 0; 1392 cl->children = 0;
1869 RB_CLEAR_NODE(&cl->pq_node); 1393 RB_CLEAR_NODE(&cl->pq_node);
1870 1394
1871 for (prio = 0; prio < TC_HTB_ 1395 for (prio = 0; prio < TC_HTB_NUMPRIO; prio++)
1872 RB_CLEAR_NODE(&cl->no 1396 RB_CLEAR_NODE(&cl->node[prio]);
1873 1397
1874 cl->common.classid = classid; <<
1875 <<
1876 /* Make sure nothing interrup <<
1877 * ndo_setup_tc calls. <<
1878 */ <<
1879 ASSERT_RTNL(); <<
1880 <<
1881 /* create leaf qdisc early be 1398 /* create leaf qdisc early because it uses kmalloc(GFP_KERNEL)
1882 * so that can't be used insi 1399 * so that can't be used inside of sch_tree_lock
1883 * -- thanks to Karlis Peisen 1400 * -- thanks to Karlis Peisenieks
1884 */ 1401 */
1885 if (!q->offload) { !! 1402 new_q = qdisc_create_dflt(sch->dev_queue, &pfifo_qdisc_ops,
1886 dev_queue = sch->dev_ <<
1887 } else if (!(parent && !paren <<
1888 /* Assign a dev_queue <<
1889 offload_opt = (struct <<
1890 .command = TC <<
1891 .classid = cl <<
1892 .parent_class <<
1893 TC_H_ <<
1894 TC_HT <<
1895 .rate = max_t <<
1896 .ceil = max_t <<
1897 .prio = hopt- <<
1898 .quantum = ho <<
1899 .extack = ext <<
1900 }; <<
1901 err = htb_offload(dev <<
1902 if (err) { <<
1903 NL_SET_ERR_MS <<
1904 <<
1905 goto err_kill <<
1906 } <<
1907 dev_queue = netdev_ge <<
1908 } else { /* First child. */ <<
1909 dev_queue = htb_offlo <<
1910 old_q = htb_graft_hel <<
1911 WARN_ON(old_q != pare <<
1912 offload_opt = (struct <<
1913 .command = TC <<
1914 .classid = cl <<
1915 .parent_class <<
1916 TC_H_ <<
1917 .rate = max_t <<
1918 .ceil = max_t <<
1919 .prio = hopt- <<
1920 .quantum = ho <<
1921 .extack = ext <<
1922 }; <<
1923 err = htb_offload(dev <<
1924 if (err) { <<
1925 NL_SET_ERR_MS <<
1926 <<
1927 htb_graft_hel <<
1928 goto err_kill <<
1929 } <<
1930 _bstats_update(&paren <<
1931 u64_st <<
1932 u64_st <<
1933 qdisc_put(old_q); <<
1934 } <<
1935 new_q = qdisc_create_dflt(dev <<
1936 cla 1403 classid, NULL);
1937 if (q->offload) { <<
1938 /* One ref for cl->le <<
1939 if (new_q) <<
1940 qdisc_refcoun <<
1941 old_q = htb_graft_hel <<
1942 /* No qdisc_put neede <<
1943 WARN_ON(!(old_q->flag <<
1944 } <<
1945 sch_tree_lock(sch); 1404 sch_tree_lock(sch);
1946 if (parent && !parent->level) 1405 if (parent && !parent->level) {
1947 /* turn parent into i 1406 /* turn parent into inner node */
1948 qdisc_purge_queue(par 1407 qdisc_purge_queue(parent->leaf.q);
1949 parent_qdisc = parent 1408 parent_qdisc = parent->leaf.q;
1950 if (parent->prio_acti 1409 if (parent->prio_activity)
1951 htb_deactivat 1410 htb_deactivate(q, parent);
1952 1411
1953 /* remove from evt li 1412 /* remove from evt list because of level change */
1954 if (parent->cmode != 1413 if (parent->cmode != HTB_CAN_SEND) {
1955 htb_safe_rb_e 1414 htb_safe_rb_erase(&parent->pq_node, &q->hlevel[0].wait_pq);
1956 parent->cmode 1415 parent->cmode = HTB_CAN_SEND;
1957 } 1416 }
1958 parent->level = (pare 1417 parent->level = (parent->parent ? parent->parent->level
1959 : TC 1418 : TC_HTB_MAXDEPTH) - 1;
1960 memset(&parent->inner 1419 memset(&parent->inner, 0, sizeof(parent->inner));
1961 } 1420 }
1962 <<
1963 /* leaf (we) needs elementary 1421 /* leaf (we) needs elementary qdisc */
1964 cl->leaf.q = new_q ? new_q : 1422 cl->leaf.q = new_q ? new_q : &noop_qdisc;
1965 if (q->offload) <<
1966 cl->leaf.offload_queu <<
1967 1423
>> 1424 cl->common.classid = classid;
1968 cl->parent = parent; 1425 cl->parent = parent;
1969 1426
1970 /* set class to be in HTB_CAN 1427 /* set class to be in HTB_CAN_SEND state */
1971 cl->tokens = PSCHED_TICKS2NS( 1428 cl->tokens = PSCHED_TICKS2NS(hopt->buffer);
1972 cl->ctokens = PSCHED_TICKS2NS 1429 cl->ctokens = PSCHED_TICKS2NS(hopt->cbuffer);
1973 cl->mbuffer = 60ULL * NSEC_PE 1430 cl->mbuffer = 60ULL * NSEC_PER_SEC; /* 1min */
1974 cl->t_c = ktime_get_ns(); 1431 cl->t_c = ktime_get_ns();
1975 cl->cmode = HTB_CAN_SEND; 1432 cl->cmode = HTB_CAN_SEND;
1976 1433
1977 /* attach to the hash list an 1434 /* attach to the hash list and parent's family */
1978 qdisc_class_hash_insert(&q->c 1435 qdisc_class_hash_insert(&q->clhash, &cl->common);
1979 if (parent) 1436 if (parent)
1980 parent->children++; 1437 parent->children++;
1981 if (cl->leaf.q != &noop_qdisc 1438 if (cl->leaf.q != &noop_qdisc)
1982 qdisc_hash_add(cl->le 1439 qdisc_hash_add(cl->leaf.q, true);
1983 } else { 1440 } else {
1984 if (tca[TCA_RATE]) { 1441 if (tca[TCA_RATE]) {
1985 err = gen_replace_est 1442 err = gen_replace_estimator(&cl->bstats, NULL,
1986 1443 &cl->rate_est,
1987 1444 NULL,
1988 !! 1445 qdisc_root_sleeping_running(sch),
1989 1446 tca[TCA_RATE]);
1990 if (err) 1447 if (err)
1991 return err; 1448 return err;
1992 } 1449 }
1993 <<
1994 if (q->offload) { <<
1995 struct net_device *de <<
1996 <<
1997 offload_opt = (struct <<
1998 .command = TC <<
1999 .classid = cl <<
2000 .rate = max_t <<
2001 .ceil = max_t <<
2002 .prio = hopt- <<
2003 .quantum = ho <<
2004 .extack = ext <<
2005 }; <<
2006 err = htb_offload(dev <<
2007 if (err) <<
2008 /* Estimator <<
2009 * as well, s <<
2010 * the estima <<
2011 * offload an <<
2012 * only when <<
2013 */ <<
2014 return err; <<
2015 } <<
2016 <<
2017 sch_tree_lock(sch); 1450 sch_tree_lock(sch);
2018 } 1451 }
2019 1452
>> 1453 rate64 = tb[TCA_HTB_RATE64] ? nla_get_u64(tb[TCA_HTB_RATE64]) : 0;
>> 1454
>> 1455 ceil64 = tb[TCA_HTB_CEIL64] ? nla_get_u64(tb[TCA_HTB_CEIL64]) : 0;
>> 1456
2020 psched_ratecfg_precompute(&cl->rate, 1457 psched_ratecfg_precompute(&cl->rate, &hopt->rate, rate64);
2021 psched_ratecfg_precompute(&cl->ceil, 1458 psched_ratecfg_precompute(&cl->ceil, &hopt->ceil, ceil64);
2022 1459
2023 /* it used to be a nasty bug here, we 1460 /* it used to be a nasty bug here, we have to check that node
2024 * is really leaf before changing cl- 1461 * is really leaf before changing cl->leaf !
2025 */ 1462 */
2026 if (!cl->level) { 1463 if (!cl->level) {
2027 u64 quantum = cl->rate.rate_b 1464 u64 quantum = cl->rate.rate_bytes_ps;
2028 1465
2029 do_div(quantum, q->rate2quant 1466 do_div(quantum, q->rate2quantum);
2030 cl->quantum = min_t(u64, quan 1467 cl->quantum = min_t(u64, quantum, INT_MAX);
2031 1468
2032 if (!hopt->quantum && cl->qua 1469 if (!hopt->quantum && cl->quantum < 1000) {
2033 warn = -1; 1470 warn = -1;
2034 cl->quantum = 1000; 1471 cl->quantum = 1000;
2035 } 1472 }
2036 if (!hopt->quantum && cl->qua 1473 if (!hopt->quantum && cl->quantum > 200000) {
2037 warn = 1; 1474 warn = 1;
2038 cl->quantum = 200000; 1475 cl->quantum = 200000;
2039 } 1476 }
2040 if (hopt->quantum) 1477 if (hopt->quantum)
2041 cl->quantum = hopt->q 1478 cl->quantum = hopt->quantum;
2042 if ((cl->prio = hopt->prio) > 1479 if ((cl->prio = hopt->prio) >= TC_HTB_NUMPRIO)
2043 cl->prio = TC_HTB_NUM 1480 cl->prio = TC_HTB_NUMPRIO - 1;
2044 } 1481 }
2045 1482
2046 cl->buffer = PSCHED_TICKS2NS(hopt->bu 1483 cl->buffer = PSCHED_TICKS2NS(hopt->buffer);
2047 cl->cbuffer = PSCHED_TICKS2NS(hopt->c 1484 cl->cbuffer = PSCHED_TICKS2NS(hopt->cbuffer);
2048 1485
2049 sch_tree_unlock(sch); 1486 sch_tree_unlock(sch);
2050 qdisc_put(parent_qdisc); 1487 qdisc_put(parent_qdisc);
2051 1488
2052 if (warn) 1489 if (warn)
2053 NL_SET_ERR_MSG_FMT_MOD(extack !! 1490 pr_warn("HTB: quantum of class %X is %s. Consider r2q change.\n",
2054 "quant !! 1491 cl->common.classid, (warn == -1 ? "small" : "big"));
2055 cl->co <<
2056 1492
2057 qdisc_class_hash_grow(sch, &q->clhash 1493 qdisc_class_hash_grow(sch, &q->clhash);
2058 1494
2059 *arg = (unsigned long)cl; 1495 *arg = (unsigned long)cl;
2060 return 0; 1496 return 0;
2061 1497
2062 err_kill_estimator: <<
2063 gen_kill_estimator(&cl->rate_est); <<
2064 err_block_put: <<
2065 tcf_block_put(cl->block); <<
2066 kfree(cl); <<
2067 failure: 1498 failure:
2068 return err; 1499 return err;
2069 } 1500 }
2070 1501
2071 static struct tcf_block *htb_tcf_block(struct 1502 static struct tcf_block *htb_tcf_block(struct Qdisc *sch, unsigned long arg,
2072 struct 1503 struct netlink_ext_ack *extack)
2073 { 1504 {
2074 struct htb_sched *q = qdisc_priv(sch) 1505 struct htb_sched *q = qdisc_priv(sch);
2075 struct htb_class *cl = (struct htb_cl 1506 struct htb_class *cl = (struct htb_class *)arg;
2076 1507
2077 return cl ? cl->block : q->block; 1508 return cl ? cl->block : q->block;
2078 } 1509 }
2079 1510
2080 static unsigned long htb_bind_filter(struct Q 1511 static unsigned long htb_bind_filter(struct Qdisc *sch, unsigned long parent,
2081 u32 clas 1512 u32 classid)
2082 { 1513 {
2083 struct htb_class *cl = htb_find(class 1514 struct htb_class *cl = htb_find(classid, sch);
2084 1515
2085 /*if (cl && !cl->level) return 0; 1516 /*if (cl && !cl->level) return 0;
2086 * The line above used to be there to 1517 * The line above used to be there to prevent attaching filters to
2087 * leaves. But at least tc_index filt 1518 * leaves. But at least tc_index filter uses this just to get class
2088 * for other reasons so that we have 1519 * for other reasons so that we have to allow for it.
2089 * ---- 1520 * ----
2090 * 19.6.2002 As Werner explained it i 1521 * 19.6.2002 As Werner explained it is ok - bind filter is just
2091 * another way to "lock" the class - 1522 * another way to "lock" the class - unlike "get" this lock can
2092 * be broken by class during destroy 1523 * be broken by class during destroy IIUC.
2093 */ 1524 */
2094 if (cl) 1525 if (cl)
2095 qdisc_class_get(&cl->common); !! 1526 cl->filter_cnt++;
2096 return (unsigned long)cl; 1527 return (unsigned long)cl;
2097 } 1528 }
2098 1529
2099 static void htb_unbind_filter(struct Qdisc *s 1530 static void htb_unbind_filter(struct Qdisc *sch, unsigned long arg)
2100 { 1531 {
2101 struct htb_class *cl = (struct htb_cl 1532 struct htb_class *cl = (struct htb_class *)arg;
2102 1533
2103 qdisc_class_put(&cl->common); !! 1534 if (cl)
>> 1535 cl->filter_cnt--;
2104 } 1536 }
2105 1537
2106 static void htb_walk(struct Qdisc *sch, struc 1538 static void htb_walk(struct Qdisc *sch, struct qdisc_walker *arg)
2107 { 1539 {
2108 struct htb_sched *q = qdisc_priv(sch) 1540 struct htb_sched *q = qdisc_priv(sch);
2109 struct htb_class *cl; 1541 struct htb_class *cl;
2110 unsigned int i; 1542 unsigned int i;
2111 1543
2112 if (arg->stop) 1544 if (arg->stop)
2113 return; 1545 return;
2114 1546
2115 for (i = 0; i < q->clhash.hashsize; i 1547 for (i = 0; i < q->clhash.hashsize; i++) {
2116 hlist_for_each_entry(cl, &q-> 1548 hlist_for_each_entry(cl, &q->clhash.hash[i], common.hnode) {
2117 if (!tc_qdisc_stats_d !! 1549 if (arg->count < arg->skip) {
>> 1550 arg->count++;
>> 1551 continue;
>> 1552 }
>> 1553 if (arg->fn(sch, (unsigned long)cl, arg) < 0) {
>> 1554 arg->stop = 1;
2118 return; 1555 return;
>> 1556 }
>> 1557 arg->count++;
2119 } 1558 }
2120 } 1559 }
2121 } 1560 }
2122 1561
2123 static const struct Qdisc_class_ops htb_class 1562 static const struct Qdisc_class_ops htb_class_ops = {
2124 .select_queue = htb_select_qu <<
2125 .graft = htb_graft, 1563 .graft = htb_graft,
2126 .leaf = htb_leaf, 1564 .leaf = htb_leaf,
2127 .qlen_notify = htb_qlen_noti 1565 .qlen_notify = htb_qlen_notify,
2128 .find = htb_search, 1566 .find = htb_search,
2129 .change = htb_change_cl 1567 .change = htb_change_class,
2130 .delete = htb_delete, 1568 .delete = htb_delete,
2131 .walk = htb_walk, 1569 .walk = htb_walk,
2132 .tcf_block = htb_tcf_block 1570 .tcf_block = htb_tcf_block,
2133 .bind_tcf = htb_bind_filt 1571 .bind_tcf = htb_bind_filter,
2134 .unbind_tcf = htb_unbind_fi 1572 .unbind_tcf = htb_unbind_filter,
2135 .dump = htb_dump_clas 1573 .dump = htb_dump_class,
2136 .dump_stats = htb_dump_clas 1574 .dump_stats = htb_dump_class_stats,
2137 }; 1575 };
2138 1576
2139 static struct Qdisc_ops htb_qdisc_ops __read_ 1577 static struct Qdisc_ops htb_qdisc_ops __read_mostly = {
2140 .cl_ops = &htb_class_op 1578 .cl_ops = &htb_class_ops,
2141 .id = "htb", 1579 .id = "htb",
2142 .priv_size = sizeof(struct 1580 .priv_size = sizeof(struct htb_sched),
2143 .enqueue = htb_enqueue, 1581 .enqueue = htb_enqueue,
2144 .dequeue = htb_dequeue, 1582 .dequeue = htb_dequeue,
2145 .peek = qdisc_peek_de 1583 .peek = qdisc_peek_dequeued,
2146 .init = htb_init, 1584 .init = htb_init,
2147 .attach = htb_attach, <<
2148 .reset = htb_reset, 1585 .reset = htb_reset,
2149 .destroy = htb_destroy, 1586 .destroy = htb_destroy,
2150 .dump = htb_dump, 1587 .dump = htb_dump,
2151 .owner = THIS_MODULE, 1588 .owner = THIS_MODULE,
2152 }; 1589 };
2153 MODULE_ALIAS_NET_SCH("htb"); <<
2154 1590
2155 static int __init htb_module_init(void) 1591 static int __init htb_module_init(void)
2156 { 1592 {
2157 return register_qdisc(&htb_qdisc_ops) 1593 return register_qdisc(&htb_qdisc_ops);
2158 } 1594 }
2159 static void __exit htb_module_exit(void) 1595 static void __exit htb_module_exit(void)
2160 { 1596 {
2161 unregister_qdisc(&htb_qdisc_ops); 1597 unregister_qdisc(&htb_qdisc_ops);
2162 } 1598 }
2163 1599
2164 module_init(htb_module_init) 1600 module_init(htb_module_init)
2165 module_exit(htb_module_exit) 1601 module_exit(htb_module_exit)
2166 MODULE_LICENSE("GPL"); 1602 MODULE_LICENSE("GPL");
2167 MODULE_DESCRIPTION("Hierarchical Token Bucket <<
2168 1603
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.