1 /* SPDX-License-Identifier: GPL-2.0 */ 2 #ifndef __NET_SCHED_GENERIC_H 3 #define __NET_SCHED_GENERIC_H 4 5 #include <linux/netdevice.h> 6 #include <linux/types.h> 7 #include <linux/rcupdate.h> 8 #include <linux/pkt_sched.h> 9 #include <linux/pkt_cls.h> 10 #include <linux/percpu.h> 11 #include <linux/dynamic_queue_limits.h> 12 #include <linux/list.h> 13 #include <linux/refcount.h> 14 #include <linux/workqueue.h> 15 #include <linux/mutex.h> 16 #include <linux/rwsem.h> 17 #include <linux/atomic.h> 18 #include <linux/hashtable.h> 19 #include <net/gen_stats.h> 20 #include <net/rtnetlink.h> 21 #include <net/flow_offload.h> 22 23 struct Qdisc_ops; 24 struct qdisc_walker; 25 struct tcf_walker; 26 struct module; 27 struct bpf_flow_keys; 28 29 struct qdisc_rate_table { 30 struct tc_ratespec rate; 31 u32 data[256]; 32 struct qdisc_rate_table *next; 33 int refcnt; 34 }; 35 36 enum qdisc_state_t { 37 __QDISC_STATE_SCHED, 38 __QDISC_STATE_DEACTIVATED, 39 __QDISC_STATE_MISSED, 40 __QDISC_STATE_DRAINING, 41 }; 42 43 enum qdisc_state2_t { 44 /* Only for !TCQ_F_NOLOCK qdisc. Never access it directly. 45 * Use qdisc_run_begin/end() or qdisc_is_running() instead. 46 */ 47 __QDISC_STATE2_RUNNING, 48 }; 49 50 #define QDISC_STATE_MISSED BIT(__QDISC_STATE_MISSED) 51 #define QDISC_STATE_DRAINING BIT(__QDISC_STATE_DRAINING) 52 53 #define QDISC_STATE_NON_EMPTY (QDISC_STATE_MISSED | \ 54 QDISC_STATE_DRAINING) 55 56 struct qdisc_size_table { 57 struct rcu_head rcu; 58 struct list_head list; 59 struct tc_sizespec szopts; 60 int refcnt; 61 u16 data[]; 62 }; 63 64 /* similar to sk_buff_head, but skb->prev pointer is undefined. */ 65 struct qdisc_skb_head { 66 struct sk_buff *head; 67 struct sk_buff *tail; 68 __u32 qlen; 69 spinlock_t lock; 70 }; 71 72 struct Qdisc { 73 int (*enqueue)(struct sk_buff *skb, 74 struct Qdisc *sch, 75 struct sk_buff **to_free); 76 struct sk_buff * (*dequeue)(struct Qdisc *sch); 77 unsigned int flags; 78 #define TCQ_F_BUILTIN 1 79 #define TCQ_F_INGRESS 2 80 #define TCQ_F_CAN_BYPASS 4 81 #define TCQ_F_MQROOT 8 82 #define TCQ_F_ONETXQUEUE 0x10 /* dequeue_skb() can assume all skbs are for 83 * q->dev_queue : It can test 84 * netif_xmit_frozen_or_stopped() before 85 * dequeueing next packet. 86 * Its true for MQ/MQPRIO slaves, or non 87 * multiqueue device. 88 */ 89 #define TCQ_F_WARN_NONWC (1 << 16) 90 #define TCQ_F_CPUSTATS 0x20 /* run using percpu statistics */ 91 #define TCQ_F_NOPARENT 0x40 /* root of its hierarchy : 92 * qdisc_tree_decrease_qlen() should stop. 93 */ 94 #define TCQ_F_INVISIBLE 0x80 /* invisible by default in dump */ 95 #define TCQ_F_NOLOCK 0x100 /* qdisc does not require locking */ 96 #define TCQ_F_OFFLOADED 0x200 /* qdisc is offloaded to HW */ 97 u32 limit; 98 const struct Qdisc_ops *ops; 99 struct qdisc_size_table __rcu *stab; 100 struct hlist_node hash; 101 u32 handle; 102 u32 parent; 103 104 struct netdev_queue *dev_queue; 105 106 struct net_rate_estimator __rcu *rate_est; 107 struct gnet_stats_basic_sync __percpu *cpu_bstats; 108 struct gnet_stats_queue __percpu *cpu_qstats; 109 int pad; 110 refcount_t refcnt; 111 112 /* 113 * For performance sake on SMP, we put highly modified fields at the end 114 */ 115 struct sk_buff_head gso_skb ____cacheline_aligned_in_smp; 116 struct qdisc_skb_head q; 117 struct gnet_stats_basic_sync bstats; 118 struct gnet_stats_queue qstats; 119 unsigned long state; 120 unsigned long state2; /* must be written under qdisc spinlock */ 121 struct Qdisc *next_sched; 122 struct sk_buff_head skb_bad_txq; 123 124 spinlock_t busylock ____cacheline_aligned_in_smp; 125 spinlock_t seqlock; 126 127 struct rcu_head rcu; 128 netdevice_tracker dev_tracker; 129 struct lock_class_key root_lock_key; 130 /* private data */ 131 long privdata[] ____cacheline_aligned; 132 }; 133 134 static inline void qdisc_refcount_inc(struct Qdisc *qdisc) 135 { 136 if (qdisc->flags & TCQ_F_BUILTIN) 137 return; 138 refcount_inc(&qdisc->refcnt); 139 } 140 141 static inline bool qdisc_refcount_dec_if_one(struct Qdisc *qdisc) 142 { 143 if (qdisc->flags & TCQ_F_BUILTIN) 144 return true; 145 return refcount_dec_if_one(&qdisc->refcnt); 146 } 147 148 /* Intended to be used by unlocked users, when concurrent qdisc release is 149 * possible. 150 */ 151 152 static inline struct Qdisc *qdisc_refcount_inc_nz(struct Qdisc *qdisc) 153 { 154 if (qdisc->flags & TCQ_F_BUILTIN) 155 return qdisc; 156 if (refcount_inc_not_zero(&qdisc->refcnt)) 157 return qdisc; 158 return NULL; 159 } 160 161 /* For !TCQ_F_NOLOCK qdisc: callers must either call this within a qdisc 162 * root_lock section, or provide their own memory barriers -- ordering 163 * against qdisc_run_begin/end() atomic bit operations. 164 */ 165 static inline bool qdisc_is_running(struct Qdisc *qdisc) 166 { 167 if (qdisc->flags & TCQ_F_NOLOCK) 168 return spin_is_locked(&qdisc->seqlock); 169 return test_bit(__QDISC_STATE2_RUNNING, &qdisc->state2); 170 } 171 172 static inline bool nolock_qdisc_is_empty(const struct Qdisc *qdisc) 173 { 174 return !(READ_ONCE(qdisc->state) & QDISC_STATE_NON_EMPTY); 175 } 176 177 static inline bool qdisc_is_percpu_stats(const struct Qdisc *q) 178 { 179 return q->flags & TCQ_F_CPUSTATS; 180 } 181 182 static inline bool qdisc_is_empty(const struct Qdisc *qdisc) 183 { 184 if (qdisc_is_percpu_stats(qdisc)) 185 return nolock_qdisc_is_empty(qdisc); 186 return !READ_ONCE(qdisc->q.qlen); 187 } 188 189 /* For !TCQ_F_NOLOCK qdisc, qdisc_run_begin/end() must be invoked with 190 * the qdisc root lock acquired. 191 */ 192 static inline bool qdisc_run_begin(struct Qdisc *qdisc) 193 { 194 if (qdisc->flags & TCQ_F_NOLOCK) { 195 if (spin_trylock(&qdisc->seqlock)) 196 return true; 197 198 /* No need to insist if the MISSED flag was already set. 199 * Note that test_and_set_bit() also gives us memory ordering 200 * guarantees wrt potential earlier enqueue() and below 201 * spin_trylock(), both of which are necessary to prevent races 202 */ 203 if (test_and_set_bit(__QDISC_STATE_MISSED, &qdisc->state)) 204 return false; 205 206 /* Try to take the lock again to make sure that we will either 207 * grab it or the CPU that still has it will see MISSED set 208 * when testing it in qdisc_run_end() 209 */ 210 return spin_trylock(&qdisc->seqlock); 211 } 212 return !__test_and_set_bit(__QDISC_STATE2_RUNNING, &qdisc->state2); 213 } 214 215 static inline void qdisc_run_end(struct Qdisc *qdisc) 216 { 217 if (qdisc->flags & TCQ_F_NOLOCK) { 218 spin_unlock(&qdisc->seqlock); 219 220 /* spin_unlock() only has store-release semantic. The unlock 221 * and test_bit() ordering is a store-load ordering, so a full 222 * memory barrier is needed here. 223 */ 224 smp_mb(); 225 226 if (unlikely(test_bit(__QDISC_STATE_MISSED, 227 &qdisc->state))) 228 __netif_schedule(qdisc); 229 } else { 230 __clear_bit(__QDISC_STATE2_RUNNING, &qdisc->state2); 231 } 232 } 233 234 static inline bool qdisc_may_bulk(const struct Qdisc *qdisc) 235 { 236 return qdisc->flags & TCQ_F_ONETXQUEUE; 237 } 238 239 static inline int qdisc_avail_bulklimit(const struct netdev_queue *txq) 240 { 241 #ifdef CONFIG_BQL 242 /* Non-BQL migrated drivers will return 0, too. */ 243 return dql_avail(&txq->dql); 244 #else 245 return 0; 246 #endif 247 } 248 249 struct Qdisc_class_ops { 250 unsigned int flags; 251 /* Child qdisc manipulation */ 252 struct netdev_queue * (*select_queue)(struct Qdisc *, struct tcmsg *); 253 int (*graft)(struct Qdisc *, unsigned long cl, 254 struct Qdisc *, struct Qdisc **, 255 struct netlink_ext_ack *extack); 256 struct Qdisc * (*leaf)(struct Qdisc *, unsigned long cl); 257 void (*qlen_notify)(struct Qdisc *, unsigned long); 258 259 /* Class manipulation routines */ 260 unsigned long (*find)(struct Qdisc *, u32 classid); 261 int (*change)(struct Qdisc *, u32, u32, 262 struct nlattr **, unsigned long *, 263 struct netlink_ext_ack *); 264 int (*delete)(struct Qdisc *, unsigned long, 265 struct netlink_ext_ack *); 266 void (*walk)(struct Qdisc *, struct qdisc_walker * arg); 267 268 /* Filter manipulation */ 269 struct tcf_block * (*tcf_block)(struct Qdisc *sch, 270 unsigned long arg, 271 struct netlink_ext_ack *extack); 272 unsigned long (*bind_tcf)(struct Qdisc *, unsigned long, 273 u32 classid); 274 void (*unbind_tcf)(struct Qdisc *, unsigned long); 275 276 /* rtnetlink specific */ 277 int (*dump)(struct Qdisc *, unsigned long, 278 struct sk_buff *skb, struct tcmsg*); 279 int (*dump_stats)(struct Qdisc *, unsigned long, 280 struct gnet_dump *); 281 }; 282 283 /* Qdisc_class_ops flag values */ 284 285 /* Implements API that doesn't require rtnl lock */ 286 enum qdisc_class_ops_flags { 287 QDISC_CLASS_OPS_DOIT_UNLOCKED = 1, 288 }; 289 290 struct Qdisc_ops { 291 struct Qdisc_ops *next; 292 const struct Qdisc_class_ops *cl_ops; 293 char id[IFNAMSIZ]; 294 int priv_size; 295 unsigned int static_flags; 296 297 int (*enqueue)(struct sk_buff *skb, 298 struct Qdisc *sch, 299 struct sk_buff **to_free); 300 struct sk_buff * (*dequeue)(struct Qdisc *); 301 struct sk_buff * (*peek)(struct Qdisc *); 302 303 int (*init)(struct Qdisc *sch, struct nlattr *arg, 304 struct netlink_ext_ack *extack); 305 void (*reset)(struct Qdisc *); 306 void (*destroy)(struct Qdisc *); 307 int (*change)(struct Qdisc *sch, 308 struct nlattr *arg, 309 struct netlink_ext_ack *extack); 310 void (*attach)(struct Qdisc *sch); 311 int (*change_tx_queue_len)(struct Qdisc *, unsigned int); 312 void (*change_real_num_tx)(struct Qdisc *sch, 313 unsigned int new_real_tx); 314 315 int (*dump)(struct Qdisc *, struct sk_buff *); 316 int (*dump_stats)(struct Qdisc *, struct gnet_dump *); 317 318 void (*ingress_block_set)(struct Qdisc *sch, 319 u32 block_index); 320 void (*egress_block_set)(struct Qdisc *sch, 321 u32 block_index); 322 u32 (*ingress_block_get)(struct Qdisc *sch); 323 u32 (*egress_block_get)(struct Qdisc *sch); 324 325 struct module *owner; 326 }; 327 328 329 struct tcf_result { 330 union { 331 struct { 332 unsigned long class; 333 u32 classid; 334 }; 335 const struct tcf_proto *goto_tp; 336 337 }; 338 }; 339 340 struct tcf_chain; 341 342 struct tcf_proto_ops { 343 struct list_head head; 344 char kind[IFNAMSIZ]; 345 346 int (*classify)(struct sk_buff *, 347 const struct tcf_proto *, 348 struct tcf_result *); 349 int (*init)(struct tcf_proto*); 350 void (*destroy)(struct tcf_proto *tp, bool rtnl_held, 351 struct netlink_ext_ack *extack); 352 353 void* (*get)(struct tcf_proto*, u32 handle); 354 void (*put)(struct tcf_proto *tp, void *f); 355 int (*change)(struct net *net, struct sk_buff *, 356 struct tcf_proto*, unsigned long, 357 u32 handle, struct nlattr **, 358 void **, u32, 359 struct netlink_ext_ack *); 360 int (*delete)(struct tcf_proto *tp, void *arg, 361 bool *last, bool rtnl_held, 362 struct netlink_ext_ack *); 363 bool (*delete_empty)(struct tcf_proto *tp); 364 void (*walk)(struct tcf_proto *tp, 365 struct tcf_walker *arg, bool rtnl_held); 366 int (*reoffload)(struct tcf_proto *tp, bool add, 367 flow_setup_cb_t *cb, void *cb_priv, 368 struct netlink_ext_ack *extack); 369 void (*hw_add)(struct tcf_proto *tp, 370 void *type_data); 371 void (*hw_del)(struct tcf_proto *tp, 372 void *type_data); 373 void (*bind_class)(void *, u32, unsigned long, 374 void *, unsigned long); 375 void * (*tmplt_create)(struct net *net, 376 struct tcf_chain *chain, 377 struct nlattr **tca, 378 struct netlink_ext_ack *extack); 379 void (*tmplt_destroy)(void *tmplt_priv); 380 void (*tmplt_reoffload)(struct tcf_chain *chain, 381 bool add, 382 flow_setup_cb_t *cb, 383 void *cb_priv); 384 struct tcf_exts * (*get_exts)(const struct tcf_proto *tp, 385 u32 handle); 386 387 /* rtnetlink specific */ 388 int (*dump)(struct net*, struct tcf_proto*, void *, 389 struct sk_buff *skb, struct tcmsg*, 390 bool); 391 int (*terse_dump)(struct net *net, 392 struct tcf_proto *tp, void *fh, 393 struct sk_buff *skb, 394 struct tcmsg *t, bool rtnl_held); 395 int (*tmplt_dump)(struct sk_buff *skb, 396 struct net *net, 397 void *tmplt_priv); 398 399 struct module *owner; 400 int flags; 401 }; 402 403 /* Classifiers setting TCF_PROTO_OPS_DOIT_UNLOCKED in tcf_proto_ops->flags 404 * are expected to implement tcf_proto_ops->delete_empty(), otherwise race 405 * conditions can occur when filters are inserted/deleted simultaneously. 406 */ 407 enum tcf_proto_ops_flags { 408 TCF_PROTO_OPS_DOIT_UNLOCKED = 1, 409 }; 410 411 struct tcf_proto { 412 /* Fast access part */ 413 struct tcf_proto __rcu *next; 414 void __rcu *root; 415 416 /* called under RCU BH lock*/ 417 int (*classify)(struct sk_buff *, 418 const struct tcf_proto *, 419 struct tcf_result *); 420 __be16 protocol; 421 422 /* All the rest */ 423 u32 prio; 424 void *data; 425 const struct tcf_proto_ops *ops; 426 struct tcf_chain *chain; 427 /* Lock protects tcf_proto shared state and can be used by unlocked 428 * classifiers to protect their private data. 429 */ 430 spinlock_t lock; 431 bool deleting; 432 refcount_t refcnt; 433 struct rcu_head rcu; 434 struct hlist_node destroy_ht_node; 435 }; 436 437 struct qdisc_skb_cb { 438 struct { 439 unsigned int pkt_len; 440 u16 slave_dev_queue_mapping; 441 u16 tc_classid; 442 }; 443 #define QDISC_CB_PRIV_LEN 20 444 unsigned char data[QDISC_CB_PRIV_LEN]; 445 }; 446 447 typedef void tcf_chain_head_change_t(struct tcf_proto *tp_head, void *priv); 448 449 struct tcf_chain { 450 /* Protects filter_chain. */ 451 struct mutex filter_chain_lock; 452 struct tcf_proto __rcu *filter_chain; 453 struct list_head list; 454 struct tcf_block *block; 455 u32 index; /* chain index */ 456 unsigned int refcnt; 457 unsigned int action_refcnt; 458 bool explicitly_created; 459 bool flushing; 460 const struct tcf_proto_ops *tmplt_ops; 461 void *tmplt_priv; 462 struct rcu_head rcu; 463 }; 464 465 struct tcf_block { 466 /* Lock protects tcf_block and lifetime-management data of chains 467 * attached to the block (refcnt, action_refcnt, explicitly_created). 468 */ 469 struct mutex lock; 470 struct list_head chain_list; 471 u32 index; /* block index for shared blocks */ 472 u32 classid; /* which class this block belongs to */ 473 refcount_t refcnt; 474 struct net *net; 475 struct Qdisc *q; 476 struct rw_semaphore cb_lock; /* protects cb_list and offload counters */ 477 struct flow_block flow_block; 478 struct list_head owner_list; 479 bool keep_dst; 480 atomic_t offloadcnt; /* Number of oddloaded filters */ 481 unsigned int nooffloaddevcnt; /* Number of devs unable to do offload */ 482 unsigned int lockeddevcnt; /* Number of devs that require rtnl lock. */ 483 struct { 484 struct tcf_chain *chain; 485 struct list_head filter_chain_list; 486 } chain0; 487 struct rcu_head rcu; 488 DECLARE_HASHTABLE(proto_destroy_ht, 7); 489 struct mutex proto_destroy_lock; /* Lock for proto_destroy hashtable. */ 490 }; 491 492 static inline bool lockdep_tcf_chain_is_locked(struct tcf_chain *chain) 493 { 494 return lockdep_is_held(&chain->filter_chain_lock); 495 } 496 497 static inline bool lockdep_tcf_proto_is_locked(struct tcf_proto *tp) 498 { 499 return lockdep_is_held(&tp->lock); 500 } 501 502 #define tcf_chain_dereference(p, chain) \ 503 rcu_dereference_protected(p, lockdep_tcf_chain_is_locked(chain)) 504 505 #define tcf_proto_dereference(p, tp) \ 506 rcu_dereference_protected(p, lockdep_tcf_proto_is_locked(tp)) 507 508 static inline void qdisc_cb_private_validate(const struct sk_buff *skb, int sz) 509 { 510 struct qdisc_skb_cb *qcb; 511 512 BUILD_BUG_ON(sizeof(skb->cb) < sizeof(*qcb)); 513 BUILD_BUG_ON(sizeof(qcb->data) < sz); 514 } 515 516 static inline int qdisc_qlen(const struct Qdisc *q) 517 { 518 return q->q.qlen; 519 } 520 521 static inline int qdisc_qlen_sum(const struct Qdisc *q) 522 { 523 __u32 qlen = q->qstats.qlen; 524 int i; 525 526 if (qdisc_is_percpu_stats(q)) { 527 for_each_possible_cpu(i) 528 qlen += per_cpu_ptr(q->cpu_qstats, i)->qlen; 529 } else { 530 qlen += q->q.qlen; 531 } 532 533 return qlen; 534 } 535 536 static inline struct qdisc_skb_cb *qdisc_skb_cb(const struct sk_buff *skb) 537 { 538 return (struct qdisc_skb_cb *)skb->cb; 539 } 540 541 static inline spinlock_t *qdisc_lock(struct Qdisc *qdisc) 542 { 543 return &qdisc->q.lock; 544 } 545 546 static inline struct Qdisc *qdisc_root(const struct Qdisc *qdisc) 547 { 548 struct Qdisc *q = rcu_dereference_rtnl(qdisc->dev_queue->qdisc); 549 550 return q; 551 } 552 553 static inline struct Qdisc *qdisc_root_bh(const struct Qdisc *qdisc) 554 { 555 return rcu_dereference_bh(qdisc->dev_queue->qdisc); 556 } 557 558 static inline struct Qdisc *qdisc_root_sleeping(const struct Qdisc *qdisc) 559 { 560 return rcu_dereference_rtnl(qdisc->dev_queue->qdisc_sleeping); 561 } 562 563 static inline spinlock_t *qdisc_root_sleeping_lock(const struct Qdisc *qdisc) 564 { 565 struct Qdisc *root = qdisc_root_sleeping(qdisc); 566 567 ASSERT_RTNL(); 568 return qdisc_lock(root); 569 } 570 571 static inline struct net_device *qdisc_dev(const struct Qdisc *qdisc) 572 { 573 return qdisc->dev_queue->dev; 574 } 575 576 static inline void sch_tree_lock(struct Qdisc *q) 577 { 578 if (q->flags & TCQ_F_MQROOT) 579 spin_lock_bh(qdisc_lock(q)); 580 else 581 spin_lock_bh(qdisc_root_sleeping_lock(q)); 582 } 583 584 static inline void sch_tree_unlock(struct Qdisc *q) 585 { 586 if (q->flags & TCQ_F_MQROOT) 587 spin_unlock_bh(qdisc_lock(q)); 588 else 589 spin_unlock_bh(qdisc_root_sleeping_lock(q)); 590 } 591 592 extern struct Qdisc noop_qdisc; 593 extern struct Qdisc_ops noop_qdisc_ops; 594 extern struct Qdisc_ops pfifo_fast_ops; 595 extern struct Qdisc_ops mq_qdisc_ops; 596 extern struct Qdisc_ops noqueue_qdisc_ops; 597 extern const struct Qdisc_ops *default_qdisc_ops; 598 static inline const struct Qdisc_ops * 599 get_default_qdisc_ops(const struct net_device *dev, int ntx) 600 { 601 return ntx < dev->real_num_tx_queues ? 602 default_qdisc_ops : &pfifo_fast_ops; 603 } 604 605 struct Qdisc_class_common { 606 u32 classid; 607 unsigned int filter_cnt; 608 struct hlist_node hnode; 609 }; 610 611 struct Qdisc_class_hash { 612 struct hlist_head *hash; 613 unsigned int hashsize; 614 unsigned int hashmask; 615 unsigned int hashelems; 616 }; 617 618 static inline unsigned int qdisc_class_hash(u32 id, u32 mask) 619 { 620 id ^= id >> 8; 621 id ^= id >> 4; 622 return id & mask; 623 } 624 625 static inline struct Qdisc_class_common * 626 qdisc_class_find(const struct Qdisc_class_hash *hash, u32 id) 627 { 628 struct Qdisc_class_common *cl; 629 unsigned int h; 630 631 if (!id) 632 return NULL; 633 634 h = qdisc_class_hash(id, hash->hashmask); 635 hlist_for_each_entry(cl, &hash->hash[h], hnode) { 636 if (cl->classid == id) 637 return cl; 638 } 639 return NULL; 640 } 641 642 static inline bool qdisc_class_in_use(const struct Qdisc_class_common *cl) 643 { 644 return cl->filter_cnt > 0; 645 } 646 647 static inline void qdisc_class_get(struct Qdisc_class_common *cl) 648 { 649 unsigned int res; 650 651 if (check_add_overflow(cl->filter_cnt, 1, &res)) 652 WARN(1, "Qdisc class overflow"); 653 654 cl->filter_cnt = res; 655 } 656 657 static inline void qdisc_class_put(struct Qdisc_class_common *cl) 658 { 659 unsigned int res; 660 661 if (check_sub_overflow(cl->filter_cnt, 1, &res)) 662 WARN(1, "Qdisc class underflow"); 663 664 cl->filter_cnt = res; 665 } 666 667 static inline int tc_classid_to_hwtc(struct net_device *dev, u32 classid) 668 { 669 u32 hwtc = TC_H_MIN(classid) - TC_H_MIN_PRIORITY; 670 671 return (hwtc < netdev_get_num_tc(dev)) ? hwtc : -EINVAL; 672 } 673 674 int qdisc_class_hash_init(struct Qdisc_class_hash *); 675 void qdisc_class_hash_insert(struct Qdisc_class_hash *, 676 struct Qdisc_class_common *); 677 void qdisc_class_hash_remove(struct Qdisc_class_hash *, 678 struct Qdisc_class_common *); 679 void qdisc_class_hash_grow(struct Qdisc *, struct Qdisc_class_hash *); 680 void qdisc_class_hash_destroy(struct Qdisc_class_hash *); 681 682 int dev_qdisc_change_tx_queue_len(struct net_device *dev); 683 void dev_qdisc_change_real_num_tx(struct net_device *dev, 684 unsigned int new_real_tx); 685 void dev_init_scheduler(struct net_device *dev); 686 void dev_shutdown(struct net_device *dev); 687 void dev_activate(struct net_device *dev); 688 void dev_deactivate(struct net_device *dev); 689 void dev_deactivate_many(struct list_head *head); 690 struct Qdisc *dev_graft_qdisc(struct netdev_queue *dev_queue, 691 struct Qdisc *qdisc); 692 void qdisc_reset(struct Qdisc *qdisc); 693 void qdisc_destroy(struct Qdisc *qdisc); 694 void qdisc_put(struct Qdisc *qdisc); 695 void qdisc_put_unlocked(struct Qdisc *qdisc); 696 void qdisc_tree_reduce_backlog(struct Qdisc *qdisc, int n, int len); 697 #ifdef CONFIG_NET_SCHED 698 int qdisc_offload_dump_helper(struct Qdisc *q, enum tc_setup_type type, 699 void *type_data); 700 void qdisc_offload_graft_helper(struct net_device *dev, struct Qdisc *sch, 701 struct Qdisc *new, struct Qdisc *old, 702 enum tc_setup_type type, void *type_data, 703 struct netlink_ext_ack *extack); 704 #else 705 static inline int 706 qdisc_offload_dump_helper(struct Qdisc *q, enum tc_setup_type type, 707 void *type_data) 708 { 709 q->flags &= ~TCQ_F_OFFLOADED; 710 return 0; 711 } 712 713 static inline void 714 qdisc_offload_graft_helper(struct net_device *dev, struct Qdisc *sch, 715 struct Qdisc *new, struct Qdisc *old, 716 enum tc_setup_type type, void *type_data, 717 struct netlink_ext_ack *extack) 718 { 719 } 720 #endif 721 void qdisc_offload_query_caps(struct net_device *dev, 722 enum tc_setup_type type, 723 void *caps, size_t caps_len); 724 struct Qdisc *qdisc_alloc(struct netdev_queue *dev_queue, 725 const struct Qdisc_ops *ops, 726 struct netlink_ext_ack *extack); 727 void qdisc_free(struct Qdisc *qdisc); 728 struct Qdisc *qdisc_create_dflt(struct netdev_queue *dev_queue, 729 const struct Qdisc_ops *ops, u32 parentid, 730 struct netlink_ext_ack *extack); 731 void __qdisc_calculate_pkt_len(struct sk_buff *skb, 732 const struct qdisc_size_table *stab); 733 int skb_do_redirect(struct sk_buff *); 734 735 static inline bool skb_at_tc_ingress(const struct sk_buff *skb) 736 { 737 #ifdef CONFIG_NET_XGRESS 738 return skb->tc_at_ingress; 739 #else 740 return false; 741 #endif 742 } 743 744 static inline bool skb_skip_tc_classify(struct sk_buff *skb) 745 { 746 #ifdef CONFIG_NET_CLS_ACT 747 if (skb->tc_skip_classify) { 748 skb->tc_skip_classify = 0; 749 return true; 750 } 751 #endif 752 return false; 753 } 754 755 /* Reset all TX qdiscs greater than index of a device. */ 756 static inline void qdisc_reset_all_tx_gt(struct net_device *dev, unsigned int i) 757 { 758 struct Qdisc *qdisc; 759 760 for (; i < dev->num_tx_queues; i++) { 761 qdisc = rtnl_dereference(netdev_get_tx_queue(dev, i)->qdisc); 762 if (qdisc) { 763 spin_lock_bh(qdisc_lock(qdisc)); 764 qdisc_reset(qdisc); 765 spin_unlock_bh(qdisc_lock(qdisc)); 766 } 767 } 768 } 769 770 /* Are all TX queues of the device empty? */ 771 static inline bool qdisc_all_tx_empty(const struct net_device *dev) 772 { 773 unsigned int i; 774 775 rcu_read_lock(); 776 for (i = 0; i < dev->num_tx_queues; i++) { 777 struct netdev_queue *txq = netdev_get_tx_queue(dev, i); 778 const struct Qdisc *q = rcu_dereference(txq->qdisc); 779 780 if (!qdisc_is_empty(q)) { 781 rcu_read_unlock(); 782 return false; 783 } 784 } 785 rcu_read_unlock(); 786 return true; 787 } 788 789 /* Are any of the TX qdiscs changing? */ 790 static inline bool qdisc_tx_changing(const struct net_device *dev) 791 { 792 unsigned int i; 793 794 for (i = 0; i < dev->num_tx_queues; i++) { 795 struct netdev_queue *txq = netdev_get_tx_queue(dev, i); 796 797 if (rcu_access_pointer(txq->qdisc) != 798 rcu_access_pointer(txq->qdisc_sleeping)) 799 return true; 800 } 801 return false; 802 } 803 804 /* Is the device using the noop qdisc on all queues? */ 805 static inline bool qdisc_tx_is_noop(const struct net_device *dev) 806 { 807 unsigned int i; 808 809 for (i = 0; i < dev->num_tx_queues; i++) { 810 struct netdev_queue *txq = netdev_get_tx_queue(dev, i); 811 if (rcu_access_pointer(txq->qdisc) != &noop_qdisc) 812 return false; 813 } 814 return true; 815 } 816 817 static inline unsigned int qdisc_pkt_len(const struct sk_buff *skb) 818 { 819 return qdisc_skb_cb(skb)->pkt_len; 820 } 821 822 /* additional qdisc xmit flags (NET_XMIT_MASK in linux/netdevice.h) */ 823 enum net_xmit_qdisc_t { 824 __NET_XMIT_STOLEN = 0x00010000, 825 __NET_XMIT_BYPASS = 0x00020000, 826 }; 827 828 #ifdef CONFIG_NET_CLS_ACT 829 #define net_xmit_drop_count(e) ((e) & __NET_XMIT_STOLEN ? 0 : 1) 830 #else 831 #define net_xmit_drop_count(e) (1) 832 #endif 833 834 static inline void qdisc_calculate_pkt_len(struct sk_buff *skb, 835 const struct Qdisc *sch) 836 { 837 #ifdef CONFIG_NET_SCHED 838 struct qdisc_size_table *stab = rcu_dereference_bh(sch->stab); 839 840 if (stab) 841 __qdisc_calculate_pkt_len(skb, stab); 842 #endif 843 } 844 845 static inline int qdisc_enqueue(struct sk_buff *skb, struct Qdisc *sch, 846 struct sk_buff **to_free) 847 { 848 qdisc_calculate_pkt_len(skb, sch); 849 return sch->enqueue(skb, sch, to_free); 850 } 851 852 static inline void _bstats_update(struct gnet_stats_basic_sync *bstats, 853 __u64 bytes, __u32 packets) 854 { 855 u64_stats_update_begin(&bstats->syncp); 856 u64_stats_add(&bstats->bytes, bytes); 857 u64_stats_add(&bstats->packets, packets); 858 u64_stats_update_end(&bstats->syncp); 859 } 860 861 static inline void bstats_update(struct gnet_stats_basic_sync *bstats, 862 const struct sk_buff *skb) 863 { 864 _bstats_update(bstats, 865 qdisc_pkt_len(skb), 866 skb_is_gso(skb) ? skb_shinfo(skb)->gso_segs : 1); 867 } 868 869 static inline void qdisc_bstats_cpu_update(struct Qdisc *sch, 870 const struct sk_buff *skb) 871 { 872 bstats_update(this_cpu_ptr(sch->cpu_bstats), skb); 873 } 874 875 static inline void qdisc_bstats_update(struct Qdisc *sch, 876 const struct sk_buff *skb) 877 { 878 bstats_update(&sch->bstats, skb); 879 } 880 881 static inline void qdisc_qstats_backlog_dec(struct Qdisc *sch, 882 const struct sk_buff *skb) 883 { 884 sch->qstats.backlog -= qdisc_pkt_len(skb); 885 } 886 887 static inline void qdisc_qstats_cpu_backlog_dec(struct Qdisc *sch, 888 const struct sk_buff *skb) 889 { 890 this_cpu_sub(sch->cpu_qstats->backlog, qdisc_pkt_len(skb)); 891 } 892 893 static inline void qdisc_qstats_backlog_inc(struct Qdisc *sch, 894 const struct sk_buff *skb) 895 { 896 sch->qstats.backlog += qdisc_pkt_len(skb); 897 } 898 899 static inline void qdisc_qstats_cpu_backlog_inc(struct Qdisc *sch, 900 const struct sk_buff *skb) 901 { 902 this_cpu_add(sch->cpu_qstats->backlog, qdisc_pkt_len(skb)); 903 } 904 905 static inline void qdisc_qstats_cpu_qlen_inc(struct Qdisc *sch) 906 { 907 this_cpu_inc(sch->cpu_qstats->qlen); 908 } 909 910 static inline void qdisc_qstats_cpu_qlen_dec(struct Qdisc *sch) 911 { 912 this_cpu_dec(sch->cpu_qstats->qlen); 913 } 914 915 static inline void qdisc_qstats_cpu_requeues_inc(struct Qdisc *sch) 916 { 917 this_cpu_inc(sch->cpu_qstats->requeues); 918 } 919 920 static inline void __qdisc_qstats_drop(struct Qdisc *sch, int count) 921 { 922 sch->qstats.drops += count; 923 } 924 925 static inline void qstats_drop_inc(struct gnet_stats_queue *qstats) 926 { 927 qstats->drops++; 928 } 929 930 static inline void qstats_overlimit_inc(struct gnet_stats_queue *qstats) 931 { 932 qstats->overlimits++; 933 } 934 935 static inline void qdisc_qstats_drop(struct Qdisc *sch) 936 { 937 qstats_drop_inc(&sch->qstats); 938 } 939 940 static inline void qdisc_qstats_cpu_drop(struct Qdisc *sch) 941 { 942 this_cpu_inc(sch->cpu_qstats->drops); 943 } 944 945 static inline void qdisc_qstats_overlimit(struct Qdisc *sch) 946 { 947 sch->qstats.overlimits++; 948 } 949 950 static inline int qdisc_qstats_copy(struct gnet_dump *d, struct Qdisc *sch) 951 { 952 __u32 qlen = qdisc_qlen_sum(sch); 953 954 return gnet_stats_copy_queue(d, sch->cpu_qstats, &sch->qstats, qlen); 955 } 956 957 static inline void qdisc_qstats_qlen_backlog(struct Qdisc *sch, __u32 *qlen, 958 __u32 *backlog) 959 { 960 struct gnet_stats_queue qstats = { 0 }; 961 962 gnet_stats_add_queue(&qstats, sch->cpu_qstats, &sch->qstats); 963 *qlen = qstats.qlen + qdisc_qlen(sch); 964 *backlog = qstats.backlog; 965 } 966 967 static inline void qdisc_tree_flush_backlog(struct Qdisc *sch) 968 { 969 __u32 qlen, backlog; 970 971 qdisc_qstats_qlen_backlog(sch, &qlen, &backlog); 972 qdisc_tree_reduce_backlog(sch, qlen, backlog); 973 } 974 975 static inline void qdisc_purge_queue(struct Qdisc *sch) 976 { 977 __u32 qlen, backlog; 978 979 qdisc_qstats_qlen_backlog(sch, &qlen, &backlog); 980 qdisc_reset(sch); 981 qdisc_tree_reduce_backlog(sch, qlen, backlog); 982 } 983 984 static inline void __qdisc_enqueue_tail(struct sk_buff *skb, 985 struct qdisc_skb_head *qh) 986 { 987 struct sk_buff *last = qh->tail; 988 989 if (last) { 990 skb->next = NULL; 991 last->next = skb; 992 qh->tail = skb; 993 } else { 994 qh->tail = skb; 995 qh->head = skb; 996 } 997 qh->qlen++; 998 } 999 1000 static inline int qdisc_enqueue_tail(struct sk_buff *skb, struct Qdisc *sch) 1001 { 1002 __qdisc_enqueue_tail(skb, &sch->q); 1003 qdisc_qstats_backlog_inc(sch, skb); 1004 return NET_XMIT_SUCCESS; 1005 } 1006 1007 static inline void __qdisc_enqueue_head(struct sk_buff *skb, 1008 struct qdisc_skb_head *qh) 1009 { 1010 skb->next = qh->head; 1011 1012 if (!qh->head) 1013 qh->tail = skb; 1014 qh->head = skb; 1015 qh->qlen++; 1016 } 1017 1018 static inline struct sk_buff *__qdisc_dequeue_head(struct qdisc_skb_head *qh) 1019 { 1020 struct sk_buff *skb = qh->head; 1021 1022 if (likely(skb != NULL)) { 1023 qh->head = skb->next; 1024 qh->qlen--; 1025 if (qh->head == NULL) 1026 qh->tail = NULL; 1027 skb->next = NULL; 1028 } 1029 1030 return skb; 1031 } 1032 1033 static inline struct sk_buff *qdisc_dequeue_head(struct Qdisc *sch) 1034 { 1035 struct sk_buff *skb = __qdisc_dequeue_head(&sch->q); 1036 1037 if (likely(skb != NULL)) { 1038 qdisc_qstats_backlog_dec(sch, skb); 1039 qdisc_bstats_update(sch, skb); 1040 } 1041 1042 return skb; 1043 } 1044 1045 /* Instead of calling kfree_skb() while root qdisc lock is held, 1046 * queue the skb for future freeing at end of __dev_xmit_skb() 1047 */ 1048 static inline void __qdisc_drop(struct sk_buff *skb, struct sk_buff **to_free) 1049 { 1050 skb->next = *to_free; 1051 *to_free = skb; 1052 } 1053 1054 static inline void __qdisc_drop_all(struct sk_buff *skb, 1055 struct sk_buff **to_free) 1056 { 1057 if (skb->prev) 1058 skb->prev->next = *to_free; 1059 else 1060 skb->next = *to_free; 1061 *to_free = skb; 1062 } 1063 1064 static inline unsigned int __qdisc_queue_drop_head(struct Qdisc *sch, 1065 struct qdisc_skb_head *qh, 1066 struct sk_buff **to_free) 1067 { 1068 struct sk_buff *skb = __qdisc_dequeue_head(qh); 1069 1070 if (likely(skb != NULL)) { 1071 unsigned int len = qdisc_pkt_len(skb); 1072 1073 qdisc_qstats_backlog_dec(sch, skb); 1074 __qdisc_drop(skb, to_free); 1075 return len; 1076 } 1077 1078 return 0; 1079 } 1080 1081 static inline struct sk_buff *qdisc_peek_head(struct Qdisc *sch) 1082 { 1083 const struct qdisc_skb_head *qh = &sch->q; 1084 1085 return qh->head; 1086 } 1087 1088 /* generic pseudo peek method for non-work-conserving qdisc */ 1089 static inline struct sk_buff *qdisc_peek_dequeued(struct Qdisc *sch) 1090 { 1091 struct sk_buff *skb = skb_peek(&sch->gso_skb); 1092 1093 /* we can reuse ->gso_skb because peek isn't called for root qdiscs */ 1094 if (!skb) { 1095 skb = sch->dequeue(sch); 1096 1097 if (skb) { 1098 __skb_queue_head(&sch->gso_skb, skb); 1099 /* it's still part of the queue */ 1100 qdisc_qstats_backlog_inc(sch, skb); 1101 sch->q.qlen++; 1102 } 1103 } 1104 1105 return skb; 1106 } 1107 1108 static inline void qdisc_update_stats_at_dequeue(struct Qdisc *sch, 1109 struct sk_buff *skb) 1110 { 1111 if (qdisc_is_percpu_stats(sch)) { 1112 qdisc_qstats_cpu_backlog_dec(sch, skb); 1113 qdisc_bstats_cpu_update(sch, skb); 1114 qdisc_qstats_cpu_qlen_dec(sch); 1115 } else { 1116 qdisc_qstats_backlog_dec(sch, skb); 1117 qdisc_bstats_update(sch, skb); 1118 sch->q.qlen--; 1119 } 1120 } 1121 1122 static inline void qdisc_update_stats_at_enqueue(struct Qdisc *sch, 1123 unsigned int pkt_len) 1124 { 1125 if (qdisc_is_percpu_stats(sch)) { 1126 qdisc_qstats_cpu_qlen_inc(sch); 1127 this_cpu_add(sch->cpu_qstats->backlog, pkt_len); 1128 } else { 1129 sch->qstats.backlog += pkt_len; 1130 sch->q.qlen++; 1131 } 1132 } 1133 1134 /* use instead of qdisc->dequeue() for all qdiscs queried with ->peek() */ 1135 static inline struct sk_buff *qdisc_dequeue_peeked(struct Qdisc *sch) 1136 { 1137 struct sk_buff *skb = skb_peek(&sch->gso_skb); 1138 1139 if (skb) { 1140 skb = __skb_dequeue(&sch->gso_skb); 1141 if (qdisc_is_percpu_stats(sch)) { 1142 qdisc_qstats_cpu_backlog_dec(sch, skb); 1143 qdisc_qstats_cpu_qlen_dec(sch); 1144 } else { 1145 qdisc_qstats_backlog_dec(sch, skb); 1146 sch->q.qlen--; 1147 } 1148 } else { 1149 skb = sch->dequeue(sch); 1150 } 1151 1152 return skb; 1153 } 1154 1155 static inline void __qdisc_reset_queue(struct qdisc_skb_head *qh) 1156 { 1157 /* 1158 * We do not know the backlog in bytes of this list, it 1159 * is up to the caller to correct it 1160 */ 1161 ASSERT_RTNL(); 1162 if (qh->qlen) { 1163 rtnl_kfree_skbs(qh->head, qh->tail); 1164 1165 qh->head = NULL; 1166 qh->tail = NULL; 1167 qh->qlen = 0; 1168 } 1169 } 1170 1171 static inline void qdisc_reset_queue(struct Qdisc *sch) 1172 { 1173 __qdisc_reset_queue(&sch->q); 1174 } 1175 1176 static inline struct Qdisc *qdisc_replace(struct Qdisc *sch, struct Qdisc *new, 1177 struct Qdisc **pold) 1178 { 1179 struct Qdisc *old; 1180 1181 sch_tree_lock(sch); 1182 old = *pold; 1183 *pold = new; 1184 if (old != NULL) 1185 qdisc_purge_queue(old); 1186 sch_tree_unlock(sch); 1187 1188 return old; 1189 } 1190 1191 static inline void rtnl_qdisc_drop(struct sk_buff *skb, struct Qdisc *sch) 1192 { 1193 rtnl_kfree_skbs(skb, skb); 1194 qdisc_qstats_drop(sch); 1195 } 1196 1197 static inline int qdisc_drop_cpu(struct sk_buff *skb, struct Qdisc *sch, 1198 struct sk_buff **to_free) 1199 { 1200 __qdisc_drop(skb, to_free); 1201 qdisc_qstats_cpu_drop(sch); 1202 1203 return NET_XMIT_DROP; 1204 } 1205 1206 static inline int qdisc_drop(struct sk_buff *skb, struct Qdisc *sch, 1207 struct sk_buff **to_free) 1208 { 1209 __qdisc_drop(skb, to_free); 1210 qdisc_qstats_drop(sch); 1211 1212 return NET_XMIT_DROP; 1213 } 1214 1215 static inline int qdisc_drop_all(struct sk_buff *skb, struct Qdisc *sch, 1216 struct sk_buff **to_free) 1217 { 1218 __qdisc_drop_all(skb, to_free); 1219 qdisc_qstats_drop(sch); 1220 1221 return NET_XMIT_DROP; 1222 } 1223 1224 struct psched_ratecfg { 1225 u64 rate_bytes_ps; /* bytes per second */ 1226 u32 mult; 1227 u16 overhead; 1228 u16 mpu; 1229 u8 linklayer; 1230 u8 shift; 1231 }; 1232 1233 static inline u64 psched_l2t_ns(const struct psched_ratecfg *r, 1234 unsigned int len) 1235 { 1236 len += r->overhead; 1237 1238 if (len < r->mpu) 1239 len = r->mpu; 1240 1241 if (unlikely(r->linklayer == TC_LINKLAYER_ATM)) 1242 return ((u64)(DIV_ROUND_UP(len,48)*53) * r->mult) >> r->shift; 1243 1244 return ((u64)len * r->mult) >> r->shift; 1245 } 1246 1247 void psched_ratecfg_precompute(struct psched_ratecfg *r, 1248 const struct tc_ratespec *conf, 1249 u64 rate64); 1250 1251 static inline void psched_ratecfg_getrate(struct tc_ratespec *res, 1252 const struct psched_ratecfg *r) 1253 { 1254 memset(res, 0, sizeof(*res)); 1255 1256 /* legacy struct tc_ratespec has a 32bit @rate field 1257 * Qdisc using 64bit rate should add new attributes 1258 * in order to maintain compatibility. 1259 */ 1260 res->rate = min_t(u64, r->rate_bytes_ps, ~0U); 1261 1262 res->overhead = r->overhead; 1263 res->mpu = r->mpu; 1264 res->linklayer = (r->linklayer & TC_LINKLAYER_MASK); 1265 } 1266 1267 struct psched_pktrate { 1268 u64 rate_pkts_ps; /* packets per second */ 1269 u32 mult; 1270 u8 shift; 1271 }; 1272 1273 static inline u64 psched_pkt2t_ns(const struct psched_pktrate *r, 1274 unsigned int pkt_num) 1275 { 1276 return ((u64)pkt_num * r->mult) >> r->shift; 1277 } 1278 1279 void psched_ppscfg_precompute(struct psched_pktrate *r, u64 pktrate64); 1280 1281 /* Mini Qdisc serves for specific needs of ingress/clsact Qdisc. 1282 * The fast path only needs to access filter list and to update stats 1283 */ 1284 struct mini_Qdisc { 1285 struct tcf_proto *filter_list; 1286 struct tcf_block *block; 1287 struct gnet_stats_basic_sync __percpu *cpu_bstats; 1288 struct gnet_stats_queue __percpu *cpu_qstats; 1289 unsigned long rcu_state; 1290 }; 1291 1292 static inline void mini_qdisc_bstats_cpu_update(struct mini_Qdisc *miniq, 1293 const struct sk_buff *skb) 1294 { 1295 bstats_update(this_cpu_ptr(miniq->cpu_bstats), skb); 1296 } 1297 1298 static inline void mini_qdisc_qstats_cpu_drop(struct mini_Qdisc *miniq) 1299 { 1300 this_cpu_inc(miniq->cpu_qstats->drops); 1301 } 1302 1303 struct mini_Qdisc_pair { 1304 struct mini_Qdisc miniq1; 1305 struct mini_Qdisc miniq2; 1306 struct mini_Qdisc __rcu **p_miniq; 1307 }; 1308 1309 void mini_qdisc_pair_swap(struct mini_Qdisc_pair *miniqp, 1310 struct tcf_proto *tp_head); 1311 void mini_qdisc_pair_init(struct mini_Qdisc_pair *miniqp, struct Qdisc *qdisc, 1312 struct mini_Qdisc __rcu **p_miniq); 1313 void mini_qdisc_pair_block_init(struct mini_Qdisc_pair *miniqp, 1314 struct tcf_block *block); 1315 1316 void mq_change_real_num_tx(struct Qdisc *sch, unsigned int new_real_tx); 1317 1318 int sch_frag_xmit_hook(struct sk_buff *skb, int (*xmit)(struct sk_buff *skb)); 1319 1320 /* Make sure qdisc is no longer in SCHED state. */ 1321 static inline void qdisc_synchronize(const struct Qdisc *q) 1322 { 1323 while (test_bit(__QDISC_STATE_SCHED, &q->state)) 1324 msleep(1); 1325 } 1326 1327 #endif 1328