2 * Copyright (c) 2016, Mellanox Technologies. All rights reserved.
4 * This software is available to you under a choice of one of two
5 * licenses. You may choose to be licensed under the terms of the GNU
6 * General Public License (GPL) Version 2, available from the file
7 * COPYING in the main directory of this source tree, or the
8 * OpenIB.org BSD license below:
10 * Redistribution and use in source and binary forms, with or
11 * without modification, are permitted provided that the following
14 * - Redistributions of source code must retain the above
15 * copyright notice, this list of conditions and the following
18 * - Redistributions in binary form must reproduce the above
19 * copyright notice, this list of conditions and the following
20 * disclaimer in the documentation and/or other materials
21 * provided with the distribution.
23 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
24 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
25 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
26 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
27 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
28 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
29 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
33 #include <linux/hash.h>
34 #include <linux/mlx5/fs.h>
36 #include <linux/ipv6.h>
44 struct in6_addr src_ipv6;
48 struct in6_addr dst_ipv6;
55 struct mlx5e_priv *priv;
56 struct work_struct arfs_work;
57 struct mlx5_flow_handle *rule;
58 struct hlist_node hlist;
60 /* Flow ID passed to ndo_rx_flow_steer */
62 /* Filter ID returned by ndo_rx_flow_steer */
64 struct arfs_tuple tuple;
67 #define mlx5e_for_each_arfs_rule(hn, tmp, arfs_tables, i, j) \
68 for (i = 0; i < ARFS_NUM_TYPES; i++) \
69 mlx5e_for_each_hash_arfs_rule(hn, tmp, arfs_tables[i].rules_hash, j)
71 #define mlx5e_for_each_hash_arfs_rule(hn, tmp, hash, j) \
72 for (j = 0; j < ARFS_HASH_SIZE; j++) \
73 hlist_for_each_entry_safe(hn, tmp, &hash[j], hlist)
75 static enum mlx5e_traffic_types arfs_get_tt(enum arfs_type type)
79 return MLX5E_TT_IPV4_TCP;
81 return MLX5E_TT_IPV4_UDP;
83 return MLX5E_TT_IPV6_TCP;
85 return MLX5E_TT_IPV6_UDP;
91 static int arfs_disable(struct mlx5e_priv *priv)
93 struct mlx5_flow_destination dest = {};
94 struct mlx5e_tir *tir = priv->indir_tir;
99 dest.type = MLX5_FLOW_DESTINATION_TYPE_TIR;
100 for (i = 0; i < ARFS_NUM_TYPES; i++) {
101 dest.tir_num = tir[i].tirn;
103 /* Modify ttc rules destination to bypass the aRFS tables*/
104 err = mlx5_modify_rule_destination(priv->fs.ttc.rules[tt],
107 netdev_err(priv->netdev,
108 "%s: modify ttc destination failed\n",
116 static void arfs_del_rules(struct mlx5e_priv *priv);
118 int mlx5e_arfs_disable(struct mlx5e_priv *priv)
120 arfs_del_rules(priv);
122 return arfs_disable(priv);
125 int mlx5e_arfs_enable(struct mlx5e_priv *priv)
127 struct mlx5_flow_destination dest = {};
132 dest.type = MLX5_FLOW_DESTINATION_TYPE_FLOW_TABLE;
133 for (i = 0; i < ARFS_NUM_TYPES; i++) {
134 dest.ft = priv->fs.arfs.arfs_tables[i].ft.t;
136 /* Modify ttc rules destination to point on the aRFS FTs */
137 err = mlx5_modify_rule_destination(priv->fs.ttc.rules[tt],
140 netdev_err(priv->netdev,
141 "%s: modify ttc destination failed err=%d\n",
150 static void arfs_destroy_table(struct arfs_table *arfs_t)
152 mlx5_del_flow_rules(arfs_t->default_rule);
153 mlx5e_destroy_flow_table(&arfs_t->ft);
156 void mlx5e_arfs_destroy_tables(struct mlx5e_priv *priv)
160 if (!(priv->netdev->hw_features & NETIF_F_NTUPLE))
163 arfs_del_rules(priv);
164 destroy_workqueue(priv->fs.arfs.wq);
165 for (i = 0; i < ARFS_NUM_TYPES; i++) {
166 if (!IS_ERR_OR_NULL(priv->fs.arfs.arfs_tables[i].ft.t))
167 arfs_destroy_table(&priv->fs.arfs.arfs_tables[i]);
171 static int arfs_add_default_rule(struct mlx5e_priv *priv,
174 struct arfs_table *arfs_t = &priv->fs.arfs.arfs_tables[type];
175 struct mlx5e_tir *tir = priv->indir_tir;
176 struct mlx5_flow_destination dest = {};
177 MLX5_DECLARE_FLOW_ACT(flow_act);
178 struct mlx5_flow_spec *spec;
179 enum mlx5e_traffic_types tt;
182 spec = kvzalloc(sizeof(*spec), GFP_KERNEL);
188 dest.type = MLX5_FLOW_DESTINATION_TYPE_TIR;
189 tt = arfs_get_tt(type);
191 netdev_err(priv->netdev, "%s: bad arfs_type: %d\n",
197 dest.tir_num = tir[tt].tirn;
199 arfs_t->default_rule = mlx5_add_flow_rules(arfs_t->ft.t, spec,
202 if (IS_ERR(arfs_t->default_rule)) {
203 err = PTR_ERR(arfs_t->default_rule);
204 arfs_t->default_rule = NULL;
205 netdev_err(priv->netdev, "%s: add rule failed, arfs type=%d\n",
213 #define MLX5E_ARFS_NUM_GROUPS 2
214 #define MLX5E_ARFS_GROUP1_SIZE (BIT(16) - 1)
215 #define MLX5E_ARFS_GROUP2_SIZE BIT(0)
216 #define MLX5E_ARFS_TABLE_SIZE (MLX5E_ARFS_GROUP1_SIZE +\
217 MLX5E_ARFS_GROUP2_SIZE)
218 static int arfs_create_groups(struct mlx5e_flow_table *ft,
221 int inlen = MLX5_ST_SZ_BYTES(create_flow_group_in);
222 void *outer_headers_c;
228 ft->g = kcalloc(MLX5E_ARFS_NUM_GROUPS,
229 sizeof(*ft->g), GFP_KERNEL);
230 in = kvzalloc(inlen, GFP_KERNEL);
237 mc = MLX5_ADDR_OF(create_flow_group_in, in, match_criteria);
238 outer_headers_c = MLX5_ADDR_OF(fte_match_param, mc,
240 MLX5_SET_TO_ONES(fte_match_set_lyr_2_4, outer_headers_c, ethertype);
244 MLX5_SET_TO_ONES(fte_match_set_lyr_2_4, outer_headers_c, tcp_dport);
245 MLX5_SET_TO_ONES(fte_match_set_lyr_2_4, outer_headers_c, tcp_sport);
249 MLX5_SET_TO_ONES(fte_match_set_lyr_2_4, outer_headers_c, udp_dport);
250 MLX5_SET_TO_ONES(fte_match_set_lyr_2_4, outer_headers_c, udp_sport);
260 MLX5_SET_TO_ONES(fte_match_set_lyr_2_4, outer_headers_c,
261 src_ipv4_src_ipv6.ipv4_layout.ipv4);
262 MLX5_SET_TO_ONES(fte_match_set_lyr_2_4, outer_headers_c,
263 dst_ipv4_dst_ipv6.ipv4_layout.ipv4);
267 memset(MLX5_ADDR_OF(fte_match_set_lyr_2_4, outer_headers_c,
268 src_ipv4_src_ipv6.ipv6_layout.ipv6),
270 memset(MLX5_ADDR_OF(fte_match_set_lyr_2_4, outer_headers_c,
271 dst_ipv4_dst_ipv6.ipv6_layout.ipv6),
279 MLX5_SET_CFG(in, match_criteria_enable, MLX5_MATCH_OUTER_HEADERS);
280 MLX5_SET_CFG(in, start_flow_index, ix);
281 ix += MLX5E_ARFS_GROUP1_SIZE;
282 MLX5_SET_CFG(in, end_flow_index, ix - 1);
283 ft->g[ft->num_groups] = mlx5_create_flow_group(ft->t, in);
284 if (IS_ERR(ft->g[ft->num_groups]))
288 memset(in, 0, inlen);
289 MLX5_SET_CFG(in, start_flow_index, ix);
290 ix += MLX5E_ARFS_GROUP2_SIZE;
291 MLX5_SET_CFG(in, end_flow_index, ix - 1);
292 ft->g[ft->num_groups] = mlx5_create_flow_group(ft->t, in);
293 if (IS_ERR(ft->g[ft->num_groups]))
301 err = PTR_ERR(ft->g[ft->num_groups]);
302 ft->g[ft->num_groups] = NULL;
309 static int arfs_create_table(struct mlx5e_priv *priv,
312 struct mlx5e_arfs_tables *arfs = &priv->fs.arfs;
313 struct mlx5e_flow_table *ft = &arfs->arfs_tables[type].ft;
314 struct mlx5_flow_table_attr ft_attr = {};
319 ft_attr.max_fte = MLX5E_ARFS_TABLE_SIZE;
320 ft_attr.level = MLX5E_ARFS_FT_LEVEL;
321 ft_attr.prio = MLX5E_NIC_PRIO;
323 ft->t = mlx5_create_flow_table(priv->fs.ns, &ft_attr);
325 err = PTR_ERR(ft->t);
330 err = arfs_create_groups(ft, type);
334 err = arfs_add_default_rule(priv, type);
340 mlx5e_destroy_flow_table(ft);
344 int mlx5e_arfs_create_tables(struct mlx5e_priv *priv)
349 if (!(priv->netdev->hw_features & NETIF_F_NTUPLE))
352 spin_lock_init(&priv->fs.arfs.arfs_lock);
353 INIT_LIST_HEAD(&priv->fs.arfs.rules);
354 priv->fs.arfs.wq = create_singlethread_workqueue("mlx5e_arfs");
355 if (!priv->fs.arfs.wq)
358 for (i = 0; i < ARFS_NUM_TYPES; i++) {
359 err = arfs_create_table(priv, i);
365 mlx5e_arfs_destroy_tables(priv);
369 #define MLX5E_ARFS_EXPIRY_QUOTA 60
371 static void arfs_may_expire_flow(struct mlx5e_priv *priv)
373 struct arfs_rule *arfs_rule;
374 struct hlist_node *htmp;
379 HLIST_HEAD(del_list);
380 spin_lock_bh(&priv->fs.arfs.arfs_lock);
381 mlx5e_for_each_arfs_rule(arfs_rule, htmp, priv->fs.arfs.arfs_tables, i, j) {
382 if (!work_pending(&arfs_rule->arfs_work) &&
383 rps_may_expire_flow(priv->netdev,
384 arfs_rule->rxq, arfs_rule->flow_id,
385 arfs_rule->filter_id)) {
386 hlist_del_init(&arfs_rule->hlist);
387 hlist_add_head(&arfs_rule->hlist, &del_list);
388 if (quota++ > MLX5E_ARFS_EXPIRY_QUOTA)
392 spin_unlock_bh(&priv->fs.arfs.arfs_lock);
393 hlist_for_each_entry_safe(arfs_rule, htmp, &del_list, hlist) {
395 mlx5_del_flow_rules(arfs_rule->rule);
396 hlist_del(&arfs_rule->hlist);
401 static void arfs_del_rules(struct mlx5e_priv *priv)
403 struct hlist_node *htmp;
404 struct arfs_rule *rule;
408 HLIST_HEAD(del_list);
409 spin_lock_bh(&priv->fs.arfs.arfs_lock);
410 mlx5e_for_each_arfs_rule(rule, htmp, priv->fs.arfs.arfs_tables, i, j) {
411 hlist_del_init(&rule->hlist);
412 hlist_add_head(&rule->hlist, &del_list);
414 spin_unlock_bh(&priv->fs.arfs.arfs_lock);
416 hlist_for_each_entry_safe(rule, htmp, &del_list, hlist) {
417 cancel_work_sync(&rule->arfs_work);
419 mlx5_del_flow_rules(rule->rule);
420 hlist_del(&rule->hlist);
425 static struct hlist_head *
426 arfs_hash_bucket(struct arfs_table *arfs_t, __be16 src_port,
432 l = (__force unsigned long)src_port |
433 ((__force unsigned long)dst_port << 2);
435 bucket_idx = hash_long(l, ARFS_HASH_SHIFT);
437 return &arfs_t->rules_hash[bucket_idx];
440 static struct arfs_table *arfs_get_table(struct mlx5e_arfs_tables *arfs,
441 u8 ip_proto, __be16 etype)
443 if (etype == htons(ETH_P_IP) && ip_proto == IPPROTO_TCP)
444 return &arfs->arfs_tables[ARFS_IPV4_TCP];
445 if (etype == htons(ETH_P_IP) && ip_proto == IPPROTO_UDP)
446 return &arfs->arfs_tables[ARFS_IPV4_UDP];
447 if (etype == htons(ETH_P_IPV6) && ip_proto == IPPROTO_TCP)
448 return &arfs->arfs_tables[ARFS_IPV6_TCP];
449 if (etype == htons(ETH_P_IPV6) && ip_proto == IPPROTO_UDP)
450 return &arfs->arfs_tables[ARFS_IPV6_UDP];
455 static struct mlx5_flow_handle *arfs_add_rule(struct mlx5e_priv *priv,
456 struct arfs_rule *arfs_rule)
458 struct mlx5e_arfs_tables *arfs = &priv->fs.arfs;
459 struct arfs_tuple *tuple = &arfs_rule->tuple;
460 struct mlx5_flow_handle *rule = NULL;
461 struct mlx5_flow_destination dest = {};
462 MLX5_DECLARE_FLOW_ACT(flow_act);
463 struct arfs_table *arfs_table;
464 struct mlx5_flow_spec *spec;
465 struct mlx5_flow_table *ft;
468 spec = kvzalloc(sizeof(*spec), GFP_KERNEL);
473 spec->match_criteria_enable = MLX5_MATCH_OUTER_HEADERS;
474 MLX5_SET_TO_ONES(fte_match_param, spec->match_criteria,
475 outer_headers.ethertype);
476 MLX5_SET(fte_match_param, spec->match_value, outer_headers.ethertype,
477 ntohs(tuple->etype));
478 arfs_table = arfs_get_table(arfs, tuple->ip_proto, tuple->etype);
484 ft = arfs_table->ft.t;
485 if (tuple->ip_proto == IPPROTO_TCP) {
486 MLX5_SET_TO_ONES(fte_match_param, spec->match_criteria,
487 outer_headers.tcp_dport);
488 MLX5_SET_TO_ONES(fte_match_param, spec->match_criteria,
489 outer_headers.tcp_sport);
490 MLX5_SET(fte_match_param, spec->match_value, outer_headers.tcp_dport,
491 ntohs(tuple->dst_port));
492 MLX5_SET(fte_match_param, spec->match_value, outer_headers.tcp_sport,
493 ntohs(tuple->src_port));
495 MLX5_SET_TO_ONES(fte_match_param, spec->match_criteria,
496 outer_headers.udp_dport);
497 MLX5_SET_TO_ONES(fte_match_param, spec->match_criteria,
498 outer_headers.udp_sport);
499 MLX5_SET(fte_match_param, spec->match_value, outer_headers.udp_dport,
500 ntohs(tuple->dst_port));
501 MLX5_SET(fte_match_param, spec->match_value, outer_headers.udp_sport,
502 ntohs(tuple->src_port));
504 if (tuple->etype == htons(ETH_P_IP)) {
505 memcpy(MLX5_ADDR_OF(fte_match_param, spec->match_value,
506 outer_headers.src_ipv4_src_ipv6.ipv4_layout.ipv4),
509 memcpy(MLX5_ADDR_OF(fte_match_param, spec->match_value,
510 outer_headers.dst_ipv4_dst_ipv6.ipv4_layout.ipv4),
513 MLX5_SET_TO_ONES(fte_match_param, spec->match_criteria,
514 outer_headers.src_ipv4_src_ipv6.ipv4_layout.ipv4);
515 MLX5_SET_TO_ONES(fte_match_param, spec->match_criteria,
516 outer_headers.dst_ipv4_dst_ipv6.ipv4_layout.ipv4);
518 memcpy(MLX5_ADDR_OF(fte_match_param, spec->match_value,
519 outer_headers.src_ipv4_src_ipv6.ipv6_layout.ipv6),
522 memcpy(MLX5_ADDR_OF(fte_match_param, spec->match_value,
523 outer_headers.dst_ipv4_dst_ipv6.ipv6_layout.ipv6),
526 memset(MLX5_ADDR_OF(fte_match_param, spec->match_criteria,
527 outer_headers.src_ipv4_src_ipv6.ipv6_layout.ipv6),
530 memset(MLX5_ADDR_OF(fte_match_param, spec->match_criteria,
531 outer_headers.dst_ipv4_dst_ipv6.ipv6_layout.ipv6),
535 dest.type = MLX5_FLOW_DESTINATION_TYPE_TIR;
536 dest.tir_num = priv->direct_tir[arfs_rule->rxq].tirn;
537 rule = mlx5_add_flow_rules(ft, spec, &flow_act, &dest, 1);
540 netdev_err(priv->netdev, "%s: add rule(filter id=%d, rq idx=%d) failed, err=%d\n",
541 __func__, arfs_rule->filter_id, arfs_rule->rxq, err);
546 return err ? ERR_PTR(err) : rule;
549 static void arfs_modify_rule_rq(struct mlx5e_priv *priv,
550 struct mlx5_flow_handle *rule, u16 rxq)
552 struct mlx5_flow_destination dst = {};
555 dst.type = MLX5_FLOW_DESTINATION_TYPE_TIR;
556 dst.tir_num = priv->direct_tir[rxq].tirn;
557 err = mlx5_modify_rule_destination(rule, &dst, NULL);
559 netdev_warn(priv->netdev,
560 "Failed to modify aRFS rule destination to rq=%d\n", rxq);
563 static void arfs_handle_work(struct work_struct *work)
565 struct arfs_rule *arfs_rule = container_of(work,
568 struct mlx5e_priv *priv = arfs_rule->priv;
569 struct mlx5_flow_handle *rule;
571 mutex_lock(&priv->state_lock);
572 if (!test_bit(MLX5E_STATE_OPENED, &priv->state)) {
573 spin_lock_bh(&priv->fs.arfs.arfs_lock);
574 hlist_del(&arfs_rule->hlist);
575 spin_unlock_bh(&priv->fs.arfs.arfs_lock);
577 mutex_unlock(&priv->state_lock);
581 mutex_unlock(&priv->state_lock);
583 if (!arfs_rule->rule) {
584 rule = arfs_add_rule(priv, arfs_rule);
587 arfs_rule->rule = rule;
589 arfs_modify_rule_rq(priv, arfs_rule->rule,
593 arfs_may_expire_flow(priv);
596 static struct arfs_rule *arfs_alloc_rule(struct mlx5e_priv *priv,
597 struct arfs_table *arfs_t,
598 const struct flow_keys *fk,
599 u16 rxq, u32 flow_id)
601 struct arfs_rule *rule;
602 struct arfs_tuple *tuple;
604 rule = kzalloc(sizeof(*rule), GFP_ATOMIC);
610 INIT_WORK(&rule->arfs_work, arfs_handle_work);
612 tuple = &rule->tuple;
613 tuple->etype = fk->basic.n_proto;
614 tuple->ip_proto = fk->basic.ip_proto;
615 if (tuple->etype == htons(ETH_P_IP)) {
616 tuple->src_ipv4 = fk->addrs.v4addrs.src;
617 tuple->dst_ipv4 = fk->addrs.v4addrs.dst;
619 memcpy(&tuple->src_ipv6, &fk->addrs.v6addrs.src,
620 sizeof(struct in6_addr));
621 memcpy(&tuple->dst_ipv6, &fk->addrs.v6addrs.dst,
622 sizeof(struct in6_addr));
624 tuple->src_port = fk->ports.src;
625 tuple->dst_port = fk->ports.dst;
627 rule->flow_id = flow_id;
628 rule->filter_id = priv->fs.arfs.last_filter_id++ % RPS_NO_FILTER;
630 hlist_add_head(&rule->hlist,
631 arfs_hash_bucket(arfs_t, tuple->src_port,
636 static bool arfs_cmp(const struct arfs_tuple *tuple, const struct flow_keys *fk)
638 if (tuple->src_port != fk->ports.src || tuple->dst_port != fk->ports.dst)
640 if (tuple->etype != fk->basic.n_proto)
642 if (tuple->etype == htons(ETH_P_IP))
643 return tuple->src_ipv4 == fk->addrs.v4addrs.src &&
644 tuple->dst_ipv4 == fk->addrs.v4addrs.dst;
645 if (tuple->etype == htons(ETH_P_IPV6))
646 return !memcmp(&tuple->src_ipv6, &fk->addrs.v6addrs.src,
647 sizeof(struct in6_addr)) &&
648 !memcmp(&tuple->dst_ipv6, &fk->addrs.v6addrs.dst,
649 sizeof(struct in6_addr));
653 static struct arfs_rule *arfs_find_rule(struct arfs_table *arfs_t,
654 const struct flow_keys *fk)
656 struct arfs_rule *arfs_rule;
657 struct hlist_head *head;
659 head = arfs_hash_bucket(arfs_t, fk->ports.src, fk->ports.dst);
660 hlist_for_each_entry(arfs_rule, head, hlist) {
661 if (arfs_cmp(&arfs_rule->tuple, fk))
668 int mlx5e_rx_flow_steer(struct net_device *dev, const struct sk_buff *skb,
669 u16 rxq_index, u32 flow_id)
671 struct mlx5e_priv *priv = netdev_priv(dev);
672 struct mlx5e_arfs_tables *arfs = &priv->fs.arfs;
673 struct arfs_table *arfs_t;
674 struct arfs_rule *arfs_rule;
677 if (!skb_flow_dissect_flow_keys(skb, &fk, 0))
678 return -EPROTONOSUPPORT;
680 if (fk.basic.n_proto != htons(ETH_P_IP) &&
681 fk.basic.n_proto != htons(ETH_P_IPV6))
682 return -EPROTONOSUPPORT;
684 if (skb->encapsulation)
685 return -EPROTONOSUPPORT;
687 arfs_t = arfs_get_table(arfs, fk.basic.ip_proto, fk.basic.n_proto);
689 return -EPROTONOSUPPORT;
691 spin_lock_bh(&arfs->arfs_lock);
692 arfs_rule = arfs_find_rule(arfs_t, &fk);
694 if (arfs_rule->rxq == rxq_index) {
695 spin_unlock_bh(&arfs->arfs_lock);
696 return arfs_rule->filter_id;
698 arfs_rule->rxq = rxq_index;
700 arfs_rule = arfs_alloc_rule(priv, arfs_t, &fk, rxq_index, flow_id);
702 spin_unlock_bh(&arfs->arfs_lock);
706 queue_work(priv->fs.arfs.wq, &arfs_rule->arfs_work);
707 spin_unlock_bh(&arfs->arfs_lock);
708 return arfs_rule->filter_id;