From: wenxu <wenxu@xxxxxxxxx> Because the new flow-indr-block can't get the tcf_block directly. It provide a callback to find the tcf block immediately when the device register and contain a ingress block. Signed-off-by: wenxu <wenxu@xxxxxxxxx> --- v3: no change include/net/flow_offload.h | 4 ++++ net/core/flow_offload.c | 12 ++++++++++++ net/sched/cls_api.c | 34 ++++++++++++++++++++++++++++++++++ 3 files changed, 50 insertions(+) diff --git a/include/net/flow_offload.h b/include/net/flow_offload.h index 66f89bc..3b2e848 100644 --- a/include/net/flow_offload.h +++ b/include/net/flow_offload.h @@ -391,6 +391,10 @@ struct flow_indr_block_dev { struct flow_block *flow_block; }; +typedef void flow_indr_get_default_block_t(struct flow_indr_block_dev *indr_dev); + +void flow_indr_set_default_block_cb(flow_indr_get_default_block_t *cb); + struct flow_indr_block_dev *flow_indr_block_dev_lookup(struct net_device *dev); int __flow_indr_block_cb_register(struct net_device *dev, void *cb_priv, diff --git a/net/core/flow_offload.c b/net/core/flow_offload.c index 9f1ae67..db8469d 100644 --- a/net/core/flow_offload.c +++ b/net/core/flow_offload.c @@ -298,6 +298,14 @@ struct flow_indr_block_dev * } EXPORT_SYMBOL(flow_indr_block_dev_lookup); +static flow_indr_get_default_block_t *flow_indr_get_default_block; + +void flow_indr_set_default_block_cb(flow_indr_get_default_block_t *cb) +{ + flow_indr_get_default_block = cb; +} +EXPORT_SYMBOL(flow_indr_set_default_block_cb); + static struct flow_indr_block_dev *flow_indr_block_dev_get(struct net_device *dev) { struct flow_indr_block_dev *indr_dev; @@ -312,6 +320,10 @@ static struct flow_indr_block_dev *flow_indr_block_dev_get(struct net_device *de INIT_LIST_HEAD(&indr_dev->cb_list); indr_dev->dev = dev; + + if (flow_indr_get_default_block) + flow_indr_get_default_block(indr_dev); + if (rhashtable_insert_fast(&indr_setup_block_ht, &indr_dev->ht_node, flow_indr_setup_block_ht_params)) { kfree(indr_dev); diff --git a/net/sched/cls_api.c b/net/sched/cls_api.c index d551c56..7c715a8 100644 --- a/net/sched/cls_api.c +++ b/net/sched/cls_api.c @@ -576,6 +576,38 @@ static void tc_indr_block_ing_cmd(struct net_device *dev, tcf_block_setup(block, &bo); } +static struct tcf_block *tc_dev_ingress_block(struct net_device *dev) +{ + const struct Qdisc_class_ops *cops; + struct Qdisc *qdisc; + + if (!dev_ingress_queue(dev)) + return NULL; + + qdisc = dev_ingress_queue(dev)->qdisc_sleeping; + if (!qdisc) + return NULL; + + cops = qdisc->ops->cl_ops; + if (!cops) + return NULL; + + if (!cops->tcf_block) + return NULL; + + return cops->tcf_block(qdisc, TC_H_MIN_INGRESS, NULL); +} + +static void tc_indr_get_default_block(struct flow_indr_block_dev *indr_dev) +{ + struct tcf_block *block = tc_dev_ingress_block(indr_dev->dev); + + if (block) { + indr_dev->flow_block = &block->flow_block; + indr_dev->ing_cmd_cb = tc_indr_block_ing_cmd; + } +} + static void tc_indr_block_call(struct tcf_block *block, struct net_device *dev, struct tcf_block_ext_info *ei, enum flow_block_command command, @@ -3168,6 +3200,8 @@ static int __init tc_filter_init(void) if (err) goto err_register_pernet_subsys; + flow_indr_set_default_block_cb(tc_indr_get_default_block); + rtnl_register(PF_UNSPEC, RTM_NEWTFILTER, tc_new_tfilter, NULL, RTNL_FLAG_DOIT_UNLOCKED); rtnl_register(PF_UNSPEC, RTM_DELTFILTER, tc_del_tfilter, NULL, -- 1.8.3.1