1#ifndef __NET_SCHED_GENERIC_H
2#define __NET_SCHED_GENERIC_H
3
4#include <linux/netdevice.h>
5#include <linux/types.h>
6#include <linux/rcupdate.h>
7#include <linux/pkt_sched.h>
8#include <linux/pkt_cls.h>
9#include <net/gen_stats.h>
10#include <net/rtnetlink.h>
11
12struct Qdisc_ops;
13struct qdisc_walker;
14struct tcf_walker;
15struct module;
16
17struct qdisc_rate_table {
18 struct tc_ratespec rate;
19 u32 data[256];
20 struct qdisc_rate_table *next;
21 int refcnt;
22};
23
24enum qdisc_state_t {
25 __QDISC_STATE_SCHED,
26 __QDISC_STATE_DEACTIVATED,
27 __QDISC_STATE_THROTTLED,
28};
29
30
31
32
33enum qdisc___state_t {
34 __QDISC___STATE_RUNNING = 1,
35};
36
37struct qdisc_size_table {
38 struct rcu_head rcu;
39 struct list_head list;
40 struct tc_sizespec szopts;
41 int refcnt;
42 u16 data[];
43};
44
45struct Qdisc {
46 int (*enqueue)(struct sk_buff *skb, struct Qdisc *dev);
47 struct sk_buff * (*dequeue)(struct Qdisc *dev);
48 unsigned int flags;
49#define TCQ_F_BUILTIN 1
50#define TCQ_F_INGRESS 2
51#define TCQ_F_CAN_BYPASS 4
52#define TCQ_F_MQROOT 8
53#define TCQ_F_ONETXQUEUE 0x10
54
55
56
57
58
59
60#define TCQ_F_WARN_NONWC (1 << 16)
61 u32 limit;
62 const struct Qdisc_ops *ops;
63 struct qdisc_size_table __rcu *stab;
64 struct list_head list;
65 u32 handle;
66 u32 parent;
67 int (*reshape_fail)(struct sk_buff *skb,
68 struct Qdisc *q);
69
70 void *u32_node;
71
72
73
74
75 struct Qdisc *__parent;
76 struct netdev_queue *dev_queue;
77
78 struct gnet_stats_rate_est64 rate_est;
79 struct Qdisc *next_sched;
80 struct sk_buff *gso_skb;
81
82
83
84 unsigned long state;
85 struct sk_buff_head q;
86 struct gnet_stats_basic_packed bstats;
87 unsigned int __state;
88 struct gnet_stats_queue qstats;
89 struct rcu_head rcu_head;
90 int padded;
91 atomic_t refcnt;
92
93 spinlock_t busylock ____cacheline_aligned_in_smp;
94};
95
96static inline bool qdisc_is_running(const struct Qdisc *qdisc)
97{
98 return (qdisc->__state & __QDISC___STATE_RUNNING) ? true : false;
99}
100
101static inline bool qdisc_run_begin(struct Qdisc *qdisc)
102{
103 if (qdisc_is_running(qdisc))
104 return false;
105 qdisc->__state |= __QDISC___STATE_RUNNING;
106 return true;
107}
108
109static inline void qdisc_run_end(struct Qdisc *qdisc)
110{
111 qdisc->__state &= ~__QDISC___STATE_RUNNING;
112}
113
114static inline bool qdisc_is_throttled(const struct Qdisc *qdisc)
115{
116 return test_bit(__QDISC_STATE_THROTTLED, &qdisc->state) ? true : false;
117}
118
119static inline void qdisc_throttled(struct Qdisc *qdisc)
120{
121 set_bit(__QDISC_STATE_THROTTLED, &qdisc->state);
122}
123
124static inline void qdisc_unthrottled(struct Qdisc *qdisc)
125{
126 clear_bit(__QDISC_STATE_THROTTLED, &qdisc->state);
127}
128
129struct Qdisc_class_ops {
130
131 struct netdev_queue * (*select_queue)(struct Qdisc *, struct tcmsg *);
132 int (*graft)(struct Qdisc *, unsigned long cl,
133 struct Qdisc *, struct Qdisc **);
134 struct Qdisc * (*leaf)(struct Qdisc *, unsigned long cl);
135 void (*qlen_notify)(struct Qdisc *, unsigned long);
136
137
138 unsigned long (*get)(struct Qdisc *, u32 classid);
139 void (*put)(struct Qdisc *, unsigned long);
140 int (*change)(struct Qdisc *, u32, u32,
141 struct nlattr **, unsigned long *);
142 int (*delete)(struct Qdisc *, unsigned long);
143 void (*walk)(struct Qdisc *, struct qdisc_walker * arg);
144
145
146 struct tcf_proto ** (*tcf_chain)(struct Qdisc *, unsigned long);
147 unsigned long (*bind_tcf)(struct Qdisc *, unsigned long,
148 u32 classid);
149 void (*unbind_tcf)(struct Qdisc *, unsigned long);
150
151
152 int (*dump)(struct Qdisc *, unsigned long,
153 struct sk_buff *skb, struct tcmsg*);
154 int (*dump_stats)(struct Qdisc *, unsigned long,
155 struct gnet_dump *);
156};
157
158struct Qdisc_ops {
159 struct Qdisc_ops *next;
160 const struct Qdisc_class_ops *cl_ops;
161 char id[IFNAMSIZ];
162 int priv_size;
163
164 int (*enqueue)(struct sk_buff *, struct Qdisc *);
165 struct sk_buff * (*dequeue)(struct Qdisc *);
166 struct sk_buff * (*peek)(struct Qdisc *);
167 unsigned int (*drop)(struct Qdisc *);
168
169 int (*init)(struct Qdisc *, struct nlattr *arg);
170 void (*reset)(struct Qdisc *);
171 void (*destroy)(struct Qdisc *);
172 int (*change)(struct Qdisc *, struct nlattr *arg);
173 void (*attach)(struct Qdisc *);
174
175 int (*dump)(struct Qdisc *, struct sk_buff *);
176 int (*dump_stats)(struct Qdisc *, struct gnet_dump *);
177
178 struct module *owner;
179};
180
181
182struct tcf_result {
183 unsigned long class;
184 u32 classid;
185};
186
187struct tcf_proto_ops {
188 struct tcf_proto_ops *next;
189 char kind[IFNAMSIZ];
190
191 int (*classify)(struct sk_buff *,
192 const struct tcf_proto *,
193 struct tcf_result *);
194 int (*init)(struct tcf_proto*);
195 void (*destroy)(struct tcf_proto*);
196
197 unsigned long (*get)(struct tcf_proto*, u32 handle);
198 void (*put)(struct tcf_proto*, unsigned long);
199 int (*change)(struct net *net, struct sk_buff *,
200 struct tcf_proto*, unsigned long,
201 u32 handle, struct nlattr **,
202 unsigned long *);
203 int (*delete)(struct tcf_proto*, unsigned long);
204 void (*walk)(struct tcf_proto*, struct tcf_walker *arg);
205
206
207 int (*dump)(struct tcf_proto*, unsigned long,
208 struct sk_buff *skb, struct tcmsg*);
209
210 struct module *owner;
211};
212
213struct tcf_proto {
214
215 struct tcf_proto *next;
216 void *root;
217 int (*classify)(struct sk_buff *,
218 const struct tcf_proto *,
219 struct tcf_result *);
220 __be16 protocol;
221
222
223 u32 prio;
224 u32 classid;
225 struct Qdisc *q;
226 void *data;
227 const struct tcf_proto_ops *ops;
228};
229
230struct qdisc_skb_cb {
231 unsigned int pkt_len;
232 u16 slave_dev_queue_mapping;
233 u16 _pad;
234 unsigned char data[20];
235};
236
237static inline void qdisc_cb_private_validate(const struct sk_buff *skb, int sz)
238{
239 struct qdisc_skb_cb *qcb;
240
241 BUILD_BUG_ON(sizeof(skb->cb) < offsetof(struct qdisc_skb_cb, data) + sz);
242 BUILD_BUG_ON(sizeof(qcb->data) < sz);
243}
244
245static inline int qdisc_qlen(const struct Qdisc *q)
246{
247 return q->q.qlen;
248}
249
250static inline struct qdisc_skb_cb *qdisc_skb_cb(const struct sk_buff *skb)
251{
252 return (struct qdisc_skb_cb *)skb->cb;
253}
254
255static inline spinlock_t *qdisc_lock(struct Qdisc *qdisc)
256{
257 return &qdisc->q.lock;
258}
259
260static inline struct Qdisc *qdisc_root(const struct Qdisc *qdisc)
261{
262 return qdisc->dev_queue->qdisc;
263}
264
265static inline struct Qdisc *qdisc_root_sleeping(const struct Qdisc *qdisc)
266{
267 return qdisc->dev_queue->qdisc_sleeping;
268}
269
270
271
272
273
274
275
276
277
278
279
280
281static inline spinlock_t *qdisc_root_lock(const struct Qdisc *qdisc)
282{
283 struct Qdisc *root = qdisc_root(qdisc);
284
285 ASSERT_RTNL();
286 return qdisc_lock(root);
287}
288
289static inline spinlock_t *qdisc_root_sleeping_lock(const struct Qdisc *qdisc)
290{
291 struct Qdisc *root = qdisc_root_sleeping(qdisc);
292
293 ASSERT_RTNL();
294 return qdisc_lock(root);
295}
296
297static inline struct net_device *qdisc_dev(const struct Qdisc *qdisc)
298{
299 return qdisc->dev_queue->dev;
300}
301
302static inline void sch_tree_lock(const struct Qdisc *q)
303{
304 spin_lock_bh(qdisc_root_sleeping_lock(q));
305}
306
307static inline void sch_tree_unlock(const struct Qdisc *q)
308{
309 spin_unlock_bh(qdisc_root_sleeping_lock(q));
310}
311
312#define tcf_tree_lock(tp) sch_tree_lock((tp)->q)
313#define tcf_tree_unlock(tp) sch_tree_unlock((tp)->q)
314
315extern struct Qdisc noop_qdisc;
316extern struct Qdisc_ops noop_qdisc_ops;
317extern struct Qdisc_ops pfifo_fast_ops;
318extern struct Qdisc_ops mq_qdisc_ops;
319
320struct Qdisc_class_common {
321 u32 classid;
322 struct hlist_node hnode;
323};
324
325struct Qdisc_class_hash {
326 struct hlist_head *hash;
327 unsigned int hashsize;
328 unsigned int hashmask;
329 unsigned int hashelems;
330};
331
332static inline unsigned int qdisc_class_hash(u32 id, u32 mask)
333{
334 id ^= id >> 8;
335 id ^= id >> 4;
336 return id & mask;
337}
338
339static inline struct Qdisc_class_common *
340qdisc_class_find(const struct Qdisc_class_hash *hash, u32 id)
341{
342 struct Qdisc_class_common *cl;
343 unsigned int h;
344
345 h = qdisc_class_hash(id, hash->hashmask);
346 hlist_for_each_entry(cl, &hash->hash[h], hnode) {
347 if (cl->classid == id)
348 return cl;
349 }
350 return NULL;
351}
352
353extern int qdisc_class_hash_init(struct Qdisc_class_hash *);
354extern void qdisc_class_hash_insert(struct Qdisc_class_hash *, struct Qdisc_class_common *);
355extern void qdisc_class_hash_remove(struct Qdisc_class_hash *, struct Qdisc_class_common *);
356extern void qdisc_class_hash_grow(struct Qdisc *, struct Qdisc_class_hash *);
357extern void qdisc_class_hash_destroy(struct Qdisc_class_hash *);
358
359extern void dev_init_scheduler(struct net_device *dev);
360extern void dev_shutdown(struct net_device *dev);
361extern void dev_activate(struct net_device *dev);
362extern void dev_deactivate(struct net_device *dev);
363extern void dev_deactivate_many(struct list_head *head);
364extern struct Qdisc *dev_graft_qdisc(struct netdev_queue *dev_queue,
365 struct Qdisc *qdisc);
366extern void qdisc_reset(struct Qdisc *qdisc);
367extern void qdisc_destroy(struct Qdisc *qdisc);
368extern void qdisc_tree_decrease_qlen(struct Qdisc *qdisc, unsigned int n);
369extern struct Qdisc *qdisc_alloc(struct netdev_queue *dev_queue,
370 struct Qdisc_ops *ops);
371extern struct Qdisc *qdisc_create_dflt(struct netdev_queue *dev_queue,
372 struct Qdisc_ops *ops, u32 parentid);
373extern void __qdisc_calculate_pkt_len(struct sk_buff *skb,
374 const struct qdisc_size_table *stab);
375extern void tcf_destroy(struct tcf_proto *tp);
376extern void tcf_destroy_chain(struct tcf_proto **fl);
377
378
379static inline void qdisc_reset_all_tx_gt(struct net_device *dev, unsigned int i)
380{
381 struct Qdisc *qdisc;
382
383 for (; i < dev->num_tx_queues; i++) {
384 qdisc = netdev_get_tx_queue(dev, i)->qdisc;
385 if (qdisc) {
386 spin_lock_bh(qdisc_lock(qdisc));
387 qdisc_reset(qdisc);
388 spin_unlock_bh(qdisc_lock(qdisc));
389 }
390 }
391}
392
393static inline void qdisc_reset_all_tx(struct net_device *dev)
394{
395 qdisc_reset_all_tx_gt(dev, 0);
396}
397
398
399static inline bool qdisc_all_tx_empty(const struct net_device *dev)
400{
401 unsigned int i;
402 for (i = 0; i < dev->num_tx_queues; i++) {
403 struct netdev_queue *txq = netdev_get_tx_queue(dev, i);
404 const struct Qdisc *q = txq->qdisc;
405
406 if (q->q.qlen)
407 return false;
408 }
409 return true;
410}
411
412
413static inline bool qdisc_tx_changing(const struct net_device *dev)
414{
415 unsigned int i;
416 for (i = 0; i < dev->num_tx_queues; i++) {
417 struct netdev_queue *txq = netdev_get_tx_queue(dev, i);
418 if (txq->qdisc != txq->qdisc_sleeping)
419 return true;
420 }
421 return false;
422}
423
424
425static inline bool qdisc_tx_is_noop(const struct net_device *dev)
426{
427 unsigned int i;
428 for (i = 0; i < dev->num_tx_queues; i++) {
429 struct netdev_queue *txq = netdev_get_tx_queue(dev, i);
430 if (txq->qdisc != &noop_qdisc)
431 return false;
432 }
433 return true;
434}
435
436static inline unsigned int qdisc_pkt_len(const struct sk_buff *skb)
437{
438 return qdisc_skb_cb(skb)->pkt_len;
439}
440
441
442enum net_xmit_qdisc_t {
443 __NET_XMIT_STOLEN = 0x00010000,
444 __NET_XMIT_BYPASS = 0x00020000,
445};
446
447#ifdef CONFIG_NET_CLS_ACT
448#define net_xmit_drop_count(e) ((e) & __NET_XMIT_STOLEN ? 0 : 1)
449#else
450#define net_xmit_drop_count(e) (1)
451#endif
452
453static inline void qdisc_calculate_pkt_len(struct sk_buff *skb,
454 const struct Qdisc *sch)
455{
456#ifdef CONFIG_NET_SCHED
457 struct qdisc_size_table *stab = rcu_dereference_bh(sch->stab);
458
459 if (stab)
460 __qdisc_calculate_pkt_len(skb, stab);
461#endif
462}
463
464static inline int qdisc_enqueue(struct sk_buff *skb, struct Qdisc *sch)
465{
466 qdisc_calculate_pkt_len(skb, sch);
467 return sch->enqueue(skb, sch);
468}
469
470static inline int qdisc_enqueue_root(struct sk_buff *skb, struct Qdisc *sch)
471{
472 qdisc_skb_cb(skb)->pkt_len = skb->len;
473 return qdisc_enqueue(skb, sch) & NET_XMIT_MASK;
474}
475
476
477static inline void bstats_update(struct gnet_stats_basic_packed *bstats,
478 const struct sk_buff *skb)
479{
480 bstats->bytes += qdisc_pkt_len(skb);
481 bstats->packets += skb_is_gso(skb) ? skb_shinfo(skb)->gso_segs : 1;
482}
483
484static inline void qdisc_bstats_update(struct Qdisc *sch,
485 const struct sk_buff *skb)
486{
487 bstats_update(&sch->bstats, skb);
488}
489
490static inline int __qdisc_enqueue_tail(struct sk_buff *skb, struct Qdisc *sch,
491 struct sk_buff_head *list)
492{
493 __skb_queue_tail(list, skb);
494 sch->qstats.backlog += qdisc_pkt_len(skb);
495
496 return NET_XMIT_SUCCESS;
497}
498
499static inline int qdisc_enqueue_tail(struct sk_buff *skb, struct Qdisc *sch)
500{
501 return __qdisc_enqueue_tail(skb, sch, &sch->q);
502}
503
504static inline struct sk_buff *__qdisc_dequeue_head(struct Qdisc *sch,
505 struct sk_buff_head *list)
506{
507 struct sk_buff *skb = __skb_dequeue(list);
508
509 if (likely(skb != NULL)) {
510 sch->qstats.backlog -= qdisc_pkt_len(skb);
511 qdisc_bstats_update(sch, skb);
512 }
513
514 return skb;
515}
516
517static inline struct sk_buff *qdisc_dequeue_head(struct Qdisc *sch)
518{
519 return __qdisc_dequeue_head(sch, &sch->q);
520}
521
522static inline unsigned int __qdisc_queue_drop_head(struct Qdisc *sch,
523 struct sk_buff_head *list)
524{
525 struct sk_buff *skb = __skb_dequeue(list);
526
527 if (likely(skb != NULL)) {
528 unsigned int len = qdisc_pkt_len(skb);
529 sch->qstats.backlog -= len;
530 kfree_skb(skb);
531 return len;
532 }
533
534 return 0;
535}
536
537static inline unsigned int qdisc_queue_drop_head(struct Qdisc *sch)
538{
539 return __qdisc_queue_drop_head(sch, &sch->q);
540}
541
542static inline struct sk_buff *__qdisc_dequeue_tail(struct Qdisc *sch,
543 struct sk_buff_head *list)
544{
545 struct sk_buff *skb = __skb_dequeue_tail(list);
546
547 if (likely(skb != NULL))
548 sch->qstats.backlog -= qdisc_pkt_len(skb);
549
550 return skb;
551}
552
553static inline struct sk_buff *qdisc_dequeue_tail(struct Qdisc *sch)
554{
555 return __qdisc_dequeue_tail(sch, &sch->q);
556}
557
558static inline struct sk_buff *qdisc_peek_head(struct Qdisc *sch)
559{
560 return skb_peek(&sch->q);
561}
562
563
564static inline struct sk_buff *qdisc_peek_dequeued(struct Qdisc *sch)
565{
566
567 if (!sch->gso_skb) {
568 sch->gso_skb = sch->dequeue(sch);
569 if (sch->gso_skb)
570
571 sch->q.qlen++;
572 }
573
574 return sch->gso_skb;
575}
576
577
578static inline struct sk_buff *qdisc_dequeue_peeked(struct Qdisc *sch)
579{
580 struct sk_buff *skb = sch->gso_skb;
581
582 if (skb) {
583 sch->gso_skb = NULL;
584 sch->q.qlen--;
585 } else {
586 skb = sch->dequeue(sch);
587 }
588
589 return skb;
590}
591
592static inline void __qdisc_reset_queue(struct Qdisc *sch,
593 struct sk_buff_head *list)
594{
595
596
597
598
599 __skb_queue_purge(list);
600}
601
602static inline void qdisc_reset_queue(struct Qdisc *sch)
603{
604 __qdisc_reset_queue(sch, &sch->q);
605 sch->qstats.backlog = 0;
606}
607
608static inline unsigned int __qdisc_queue_drop(struct Qdisc *sch,
609 struct sk_buff_head *list)
610{
611 struct sk_buff *skb = __qdisc_dequeue_tail(sch, list);
612
613 if (likely(skb != NULL)) {
614 unsigned int len = qdisc_pkt_len(skb);
615 kfree_skb(skb);
616 return len;
617 }
618
619 return 0;
620}
621
622static inline unsigned int qdisc_queue_drop(struct Qdisc *sch)
623{
624 return __qdisc_queue_drop(sch, &sch->q);
625}
626
627static inline int qdisc_drop(struct sk_buff *skb, struct Qdisc *sch)
628{
629 kfree_skb(skb);
630 sch->qstats.drops++;
631
632 return NET_XMIT_DROP;
633}
634
635static inline int qdisc_reshape_fail(struct sk_buff *skb, struct Qdisc *sch)
636{
637 sch->qstats.drops++;
638
639#ifdef CONFIG_NET_CLS_ACT
640 if (sch->reshape_fail == NULL || sch->reshape_fail(skb, sch))
641 goto drop;
642
643 return NET_XMIT_SUCCESS;
644
645drop:
646#endif
647 kfree_skb(skb);
648 return NET_XMIT_DROP;
649}
650
651
652
653
654static inline u32 qdisc_l2t(struct qdisc_rate_table* rtab, unsigned int pktlen)
655{
656 int slot = pktlen + rtab->rate.cell_align + rtab->rate.overhead;
657 if (slot < 0)
658 slot = 0;
659 slot >>= rtab->rate.cell_log;
660 if (slot > 255)
661 return rtab->data[255]*(slot >> 8) + rtab->data[slot & 0xFF];
662 return rtab->data[slot];
663}
664
665#ifdef CONFIG_NET_CLS_ACT
666static inline struct sk_buff *skb_act_clone(struct sk_buff *skb, gfp_t gfp_mask,
667 int action)
668{
669 struct sk_buff *n;
670
671 n = skb_clone(skb, gfp_mask);
672
673 if (n) {
674 n->tc_verd = SET_TC_VERD(n->tc_verd, 0);
675 n->tc_verd = CLR_TC_OK2MUNGE(n->tc_verd);
676 n->tc_verd = CLR_TC_MUNGED(n->tc_verd);
677 }
678 return n;
679}
680#endif
681
682struct psched_ratecfg {
683 u64 rate_bytes_ps;
684 u32 mult;
685 u16 overhead;
686 u8 linklayer;
687 u8 shift;
688};
689
690static inline u64 psched_l2t_ns(const struct psched_ratecfg *r,
691 unsigned int len)
692{
693 len += r->overhead;
694
695 if (unlikely(r->linklayer == TC_LINKLAYER_ATM))
696 return ((u64)(DIV_ROUND_UP(len,48)*53) * r->mult) >> r->shift;
697
698 return ((u64)len * r->mult) >> r->shift;
699}
700
701extern void psched_ratecfg_precompute(struct psched_ratecfg *r, const struct tc_ratespec *conf);
702
703static inline void psched_ratecfg_getrate(struct tc_ratespec *res,
704 const struct psched_ratecfg *r)
705{
706 memset(res, 0, sizeof(*res));
707 res->rate = r->rate_bytes_ps;
708 res->overhead = r->overhead;
709 res->linklayer = (r->linklayer & TC_LINKLAYER_MASK);
710}
711
712#endif
713