45cdde694d20049af85391ffcc376438fadb921e
[sfrench/cifs-2.6.git] / drivers / net / ethernet / mellanox / mlx5 / core / en_arfs.c
1 /*
2  * Copyright (c) 2016, Mellanox Technologies. All rights reserved.
3  *
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:
9  *
10  *     Redistribution and use in source and binary forms, with or
11  *     without modification, are permitted provided that the following
12  *     conditions are met:
13  *
14  *      - Redistributions of source code must retain the above
15  *        copyright notice, this list of conditions and the following
16  *        disclaimer.
17  *
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.
22  *
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
30  * SOFTWARE.
31  */
32
33 #include <linux/hash.h>
34 #include <linux/mlx5/fs.h>
35 #include <linux/ip.h>
36 #include <linux/ipv6.h>
37 #include "en.h"
38
39 struct arfs_tuple {
40         __be16 etype;
41         u8     ip_proto;
42         union {
43                 __be32 src_ipv4;
44                 struct in6_addr src_ipv6;
45         };
46         union {
47                 __be32 dst_ipv4;
48                 struct in6_addr dst_ipv6;
49         };
50         __be16 src_port;
51         __be16 dst_port;
52 };
53
54 struct arfs_rule {
55         struct mlx5e_priv       *priv;
56         struct work_struct      arfs_work;
57         struct mlx5_flow_handle *rule;
58         struct hlist_node       hlist;
59         int                     rxq;
60         /* Flow ID passed to ndo_rx_flow_steer */
61         int                     flow_id;
62         /* Filter ID returned by ndo_rx_flow_steer */
63         int                     filter_id;
64         struct arfs_tuple       tuple;
65 };
66
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)
70
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)
74
75 static enum mlx5e_traffic_types arfs_get_tt(enum arfs_type type)
76 {
77         switch (type) {
78         case ARFS_IPV4_TCP:
79                 return MLX5E_TT_IPV4_TCP;
80         case ARFS_IPV4_UDP:
81                 return MLX5E_TT_IPV4_UDP;
82         case ARFS_IPV6_TCP:
83                 return MLX5E_TT_IPV6_TCP;
84         case ARFS_IPV6_UDP:
85                 return MLX5E_TT_IPV6_UDP;
86         default:
87                 return -EINVAL;
88         }
89 }
90
91 static int arfs_disable(struct mlx5e_priv *priv)
92 {
93         struct mlx5_flow_destination dest = {};
94         struct mlx5e_tir *tir = priv->indir_tir;
95         int err = 0;
96         int tt;
97         int i;
98
99         dest.type = MLX5_FLOW_DESTINATION_TYPE_TIR;
100         for (i = 0; i < ARFS_NUM_TYPES; i++) {
101                 dest.tir_num = tir[i].tirn;
102                 tt = arfs_get_tt(i);
103                 /* Modify ttc rules destination to bypass the aRFS tables*/
104                 err = mlx5_modify_rule_destination(priv->fs.ttc.rules[tt],
105                                                    &dest, NULL);
106                 if (err) {
107                         netdev_err(priv->netdev,
108                                    "%s: modify ttc destination failed\n",
109                                    __func__);
110                         return err;
111                 }
112         }
113         return 0;
114 }
115
116 static void arfs_del_rules(struct mlx5e_priv *priv);
117
118 int mlx5e_arfs_disable(struct mlx5e_priv *priv)
119 {
120         arfs_del_rules(priv);
121
122         return arfs_disable(priv);
123 }
124
125 int mlx5e_arfs_enable(struct mlx5e_priv *priv)
126 {
127         struct mlx5_flow_destination dest = {};
128         int err = 0;
129         int tt;
130         int i;
131
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;
135                 tt = arfs_get_tt(i);
136                 /* Modify ttc rules destination to point on the aRFS FTs */
137                 err = mlx5_modify_rule_destination(priv->fs.ttc.rules[tt],
138                                                    &dest, NULL);
139                 if (err) {
140                         netdev_err(priv->netdev,
141                                    "%s: modify ttc destination failed err=%d\n",
142                                    __func__, err);
143                         arfs_disable(priv);
144                         return err;
145                 }
146         }
147         return 0;
148 }
149
150 static void arfs_destroy_table(struct arfs_table *arfs_t)
151 {
152         mlx5_del_flow_rules(arfs_t->default_rule);
153         mlx5e_destroy_flow_table(&arfs_t->ft);
154 }
155
156 void mlx5e_arfs_destroy_tables(struct mlx5e_priv *priv)
157 {
158         int i;
159
160         if (!(priv->netdev->hw_features & NETIF_F_NTUPLE))
161                 return;
162
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]);
168         }
169 }
170
171 static int arfs_add_default_rule(struct mlx5e_priv *priv,
172                                  enum arfs_type type)
173 {
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;
180         int err = 0;
181
182         spec = kvzalloc(sizeof(*spec), GFP_KERNEL);
183         if (!spec) {
184                 err = -ENOMEM;
185                 goto out;
186         }
187
188         dest.type = MLX5_FLOW_DESTINATION_TYPE_TIR;
189         tt = arfs_get_tt(type);
190         if (tt == -EINVAL) {
191                 netdev_err(priv->netdev, "%s: bad arfs_type: %d\n",
192                            __func__, type);
193                 err = -EINVAL;
194                 goto out;
195         }
196
197         dest.tir_num = tir[tt].tirn;
198
199         arfs_t->default_rule = mlx5_add_flow_rules(arfs_t->ft.t, spec,
200                                                    &flow_act,
201                                                    &dest, 1);
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",
206                            __func__, type);
207         }
208 out:
209         kvfree(spec);
210         return err;
211 }
212
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,
219                               enum  arfs_type type)
220 {
221         int inlen = MLX5_ST_SZ_BYTES(create_flow_group_in);
222         void *outer_headers_c;
223         int ix = 0;
224         u32 *in;
225         int err;
226         u8 *mc;
227
228         ft->g = kcalloc(MLX5E_ARFS_NUM_GROUPS,
229                         sizeof(*ft->g), GFP_KERNEL);
230         in = kvzalloc(inlen, GFP_KERNEL);
231         if  (!in || !ft->g) {
232                 kvfree(ft->g);
233                 kvfree(in);
234                 return -ENOMEM;
235         }
236
237         mc = MLX5_ADDR_OF(create_flow_group_in, in, match_criteria);
238         outer_headers_c = MLX5_ADDR_OF(fte_match_param, mc,
239                                        outer_headers);
240         MLX5_SET_TO_ONES(fte_match_set_lyr_2_4, outer_headers_c, ethertype);
241         switch (type) {
242         case ARFS_IPV4_TCP:
243         case ARFS_IPV6_TCP:
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);
246                 break;
247         case ARFS_IPV4_UDP:
248         case ARFS_IPV6_UDP:
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);
251                 break;
252         default:
253                 err = -EINVAL;
254                 goto out;
255         }
256
257         switch (type) {
258         case ARFS_IPV4_TCP:
259         case ARFS_IPV4_UDP:
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);
264                 break;
265         case ARFS_IPV6_TCP:
266         case ARFS_IPV6_UDP:
267                 memset(MLX5_ADDR_OF(fte_match_set_lyr_2_4, outer_headers_c,
268                                     src_ipv4_src_ipv6.ipv6_layout.ipv6),
269                        0xff, 16);
270                 memset(MLX5_ADDR_OF(fte_match_set_lyr_2_4, outer_headers_c,
271                                     dst_ipv4_dst_ipv6.ipv6_layout.ipv6),
272                        0xff, 16);
273                 break;
274         default:
275                 err = -EINVAL;
276                 goto out;
277         }
278
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]))
285                 goto err;
286         ft->num_groups++;
287
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]))
294                 goto err;
295         ft->num_groups++;
296
297         kvfree(in);
298         return 0;
299
300 err:
301         err = PTR_ERR(ft->g[ft->num_groups]);
302         ft->g[ft->num_groups] = NULL;
303 out:
304         kvfree(in);
305
306         return err;
307 }
308
309 static int arfs_create_table(struct mlx5e_priv *priv,
310                              enum arfs_type type)
311 {
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 = {};
315         int err;
316
317         ft->num_groups = 0;
318
319         ft_attr.max_fte = MLX5E_ARFS_TABLE_SIZE;
320         ft_attr.level = MLX5E_ARFS_FT_LEVEL;
321         ft_attr.prio = MLX5E_NIC_PRIO;
322
323         ft->t = mlx5_create_flow_table(priv->fs.ns, &ft_attr);
324         if (IS_ERR(ft->t)) {
325                 err = PTR_ERR(ft->t);
326                 ft->t = NULL;
327                 return err;
328         }
329
330         err = arfs_create_groups(ft, type);
331         if (err)
332                 goto err;
333
334         err = arfs_add_default_rule(priv, type);
335         if (err)
336                 goto err;
337
338         return 0;
339 err:
340         mlx5e_destroy_flow_table(ft);
341         return err;
342 }
343
344 int mlx5e_arfs_create_tables(struct mlx5e_priv *priv)
345 {
346         int err = 0;
347         int i;
348
349         if (!(priv->netdev->hw_features & NETIF_F_NTUPLE))
350                 return 0;
351
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)
356                 return -ENOMEM;
357
358         for (i = 0; i < ARFS_NUM_TYPES; i++) {
359                 err = arfs_create_table(priv, i);
360                 if (err)
361                         goto err;
362         }
363         return 0;
364 err:
365         mlx5e_arfs_destroy_tables(priv);
366         return err;
367 }
368
369 #define MLX5E_ARFS_EXPIRY_QUOTA 60
370
371 static void arfs_may_expire_flow(struct mlx5e_priv *priv)
372 {
373         struct arfs_rule *arfs_rule;
374         struct hlist_node *htmp;
375         int quota = 0;
376         int i;
377         int j;
378
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)
389                                 break;
390                 }
391         }
392         spin_unlock_bh(&priv->fs.arfs.arfs_lock);
393         hlist_for_each_entry_safe(arfs_rule, htmp, &del_list, hlist) {
394                 if (arfs_rule->rule)
395                         mlx5_del_flow_rules(arfs_rule->rule);
396                 hlist_del(&arfs_rule->hlist);
397                 kfree(arfs_rule);
398         }
399 }
400
401 static void arfs_del_rules(struct mlx5e_priv *priv)
402 {
403         struct hlist_node *htmp;
404         struct arfs_rule *rule;
405         int i;
406         int j;
407
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);
413         }
414         spin_unlock_bh(&priv->fs.arfs.arfs_lock);
415
416         hlist_for_each_entry_safe(rule, htmp, &del_list, hlist) {
417                 cancel_work_sync(&rule->arfs_work);
418                 if (rule->rule)
419                         mlx5_del_flow_rules(rule->rule);
420                 hlist_del(&rule->hlist);
421                 kfree(rule);
422         }
423 }
424
425 static struct hlist_head *
426 arfs_hash_bucket(struct arfs_table *arfs_t, __be16 src_port,
427                  __be16 dst_port)
428 {
429         unsigned long l;
430         int bucket_idx;
431
432         l = (__force unsigned long)src_port |
433             ((__force unsigned long)dst_port << 2);
434
435         bucket_idx = hash_long(l, ARFS_HASH_SHIFT);
436
437         return &arfs_t->rules_hash[bucket_idx];
438 }
439
440 static u8 arfs_get_ip_proto(const struct sk_buff *skb)
441 {
442         return (skb->protocol == htons(ETH_P_IP)) ?
443                 ip_hdr(skb)->protocol : ipv6_hdr(skb)->nexthdr;
444 }
445
446 static struct arfs_table *arfs_get_table(struct mlx5e_arfs_tables *arfs,
447                                          u8 ip_proto, __be16 etype)
448 {
449         if (etype == htons(ETH_P_IP) && ip_proto == IPPROTO_TCP)
450                 return &arfs->arfs_tables[ARFS_IPV4_TCP];
451         if (etype == htons(ETH_P_IP) && ip_proto == IPPROTO_UDP)
452                 return &arfs->arfs_tables[ARFS_IPV4_UDP];
453         if (etype == htons(ETH_P_IPV6) && ip_proto == IPPROTO_TCP)
454                 return &arfs->arfs_tables[ARFS_IPV6_TCP];
455         if (etype == htons(ETH_P_IPV6) && ip_proto == IPPROTO_UDP)
456                 return &arfs->arfs_tables[ARFS_IPV6_UDP];
457
458         return NULL;
459 }
460
461 static struct mlx5_flow_handle *arfs_add_rule(struct mlx5e_priv *priv,
462                                               struct arfs_rule *arfs_rule)
463 {
464         struct mlx5e_arfs_tables *arfs = &priv->fs.arfs;
465         struct arfs_tuple *tuple = &arfs_rule->tuple;
466         struct mlx5_flow_handle *rule = NULL;
467         struct mlx5_flow_destination dest = {};
468         MLX5_DECLARE_FLOW_ACT(flow_act);
469         struct arfs_table *arfs_table;
470         struct mlx5_flow_spec *spec;
471         struct mlx5_flow_table *ft;
472         int err = 0;
473
474         spec = kvzalloc(sizeof(*spec), GFP_KERNEL);
475         if (!spec) {
476                 err = -ENOMEM;
477                 goto out;
478         }
479         spec->match_criteria_enable = MLX5_MATCH_OUTER_HEADERS;
480         MLX5_SET_TO_ONES(fte_match_param, spec->match_criteria,
481                          outer_headers.ethertype);
482         MLX5_SET(fte_match_param, spec->match_value, outer_headers.ethertype,
483                  ntohs(tuple->etype));
484         arfs_table = arfs_get_table(arfs, tuple->ip_proto, tuple->etype);
485         if (!arfs_table) {
486                 err = -EINVAL;
487                 goto out;
488         }
489
490         ft = arfs_table->ft.t;
491         if (tuple->ip_proto == IPPROTO_TCP) {
492                 MLX5_SET_TO_ONES(fte_match_param, spec->match_criteria,
493                                  outer_headers.tcp_dport);
494                 MLX5_SET_TO_ONES(fte_match_param, spec->match_criteria,
495                                  outer_headers.tcp_sport);
496                 MLX5_SET(fte_match_param, spec->match_value, outer_headers.tcp_dport,
497                          ntohs(tuple->dst_port));
498                 MLX5_SET(fte_match_param, spec->match_value, outer_headers.tcp_sport,
499                          ntohs(tuple->src_port));
500         } else {
501                 MLX5_SET_TO_ONES(fte_match_param, spec->match_criteria,
502                                  outer_headers.udp_dport);
503                 MLX5_SET_TO_ONES(fte_match_param, spec->match_criteria,
504                                  outer_headers.udp_sport);
505                 MLX5_SET(fte_match_param, spec->match_value, outer_headers.udp_dport,
506                          ntohs(tuple->dst_port));
507                 MLX5_SET(fte_match_param, spec->match_value, outer_headers.udp_sport,
508                          ntohs(tuple->src_port));
509         }
510         if (tuple->etype == htons(ETH_P_IP)) {
511                 memcpy(MLX5_ADDR_OF(fte_match_param, spec->match_value,
512                                     outer_headers.src_ipv4_src_ipv6.ipv4_layout.ipv4),
513                        &tuple->src_ipv4,
514                        4);
515                 memcpy(MLX5_ADDR_OF(fte_match_param, spec->match_value,
516                                     outer_headers.dst_ipv4_dst_ipv6.ipv4_layout.ipv4),
517                        &tuple->dst_ipv4,
518                        4);
519                 MLX5_SET_TO_ONES(fte_match_param, spec->match_criteria,
520                                  outer_headers.src_ipv4_src_ipv6.ipv4_layout.ipv4);
521                 MLX5_SET_TO_ONES(fte_match_param, spec->match_criteria,
522                                  outer_headers.dst_ipv4_dst_ipv6.ipv4_layout.ipv4);
523         } else {
524                 memcpy(MLX5_ADDR_OF(fte_match_param, spec->match_value,
525                                     outer_headers.src_ipv4_src_ipv6.ipv6_layout.ipv6),
526                        &tuple->src_ipv6,
527                        16);
528                 memcpy(MLX5_ADDR_OF(fte_match_param, spec->match_value,
529                                     outer_headers.dst_ipv4_dst_ipv6.ipv6_layout.ipv6),
530                        &tuple->dst_ipv6,
531                        16);
532                 memset(MLX5_ADDR_OF(fte_match_param, spec->match_criteria,
533                                     outer_headers.src_ipv4_src_ipv6.ipv6_layout.ipv6),
534                        0xff,
535                        16);
536                 memset(MLX5_ADDR_OF(fte_match_param, spec->match_criteria,
537                                     outer_headers.dst_ipv4_dst_ipv6.ipv6_layout.ipv6),
538                        0xff,
539                        16);
540         }
541         dest.type = MLX5_FLOW_DESTINATION_TYPE_TIR;
542         dest.tir_num = priv->direct_tir[arfs_rule->rxq].tirn;
543         rule = mlx5_add_flow_rules(ft, spec, &flow_act, &dest, 1);
544         if (IS_ERR(rule)) {
545                 err = PTR_ERR(rule);
546                 netdev_err(priv->netdev, "%s: add rule(filter id=%d, rq idx=%d) failed, err=%d\n",
547                            __func__, arfs_rule->filter_id, arfs_rule->rxq, err);
548         }
549
550 out:
551         kvfree(spec);
552         return err ? ERR_PTR(err) : rule;
553 }
554
555 static void arfs_modify_rule_rq(struct mlx5e_priv *priv,
556                                 struct mlx5_flow_handle *rule, u16 rxq)
557 {
558         struct mlx5_flow_destination dst = {};
559         int err = 0;
560
561         dst.type = MLX5_FLOW_DESTINATION_TYPE_TIR;
562         dst.tir_num = priv->direct_tir[rxq].tirn;
563         err =  mlx5_modify_rule_destination(rule, &dst, NULL);
564         if (err)
565                 netdev_warn(priv->netdev,
566                             "Failed to modify aRFS rule destination to rq=%d\n", rxq);
567 }
568
569 static void arfs_handle_work(struct work_struct *work)
570 {
571         struct arfs_rule *arfs_rule = container_of(work,
572                                                    struct arfs_rule,
573                                                    arfs_work);
574         struct mlx5e_priv *priv = arfs_rule->priv;
575         struct mlx5_flow_handle *rule;
576
577         mutex_lock(&priv->state_lock);
578         if (!test_bit(MLX5E_STATE_OPENED, &priv->state)) {
579                 spin_lock_bh(&priv->fs.arfs.arfs_lock);
580                 hlist_del(&arfs_rule->hlist);
581                 spin_unlock_bh(&priv->fs.arfs.arfs_lock);
582
583                 mutex_unlock(&priv->state_lock);
584                 kfree(arfs_rule);
585                 goto out;
586         }
587         mutex_unlock(&priv->state_lock);
588
589         if (!arfs_rule->rule) {
590                 rule = arfs_add_rule(priv, arfs_rule);
591                 if (IS_ERR(rule))
592                         goto out;
593                 arfs_rule->rule = rule;
594         } else {
595                 arfs_modify_rule_rq(priv, arfs_rule->rule,
596                                     arfs_rule->rxq);
597         }
598 out:
599         arfs_may_expire_flow(priv);
600 }
601
602 /* return L4 destination port from ip4/6 packets */
603 static __be16 arfs_get_dst_port(const struct sk_buff *skb)
604 {
605         char *transport_header;
606
607         transport_header = skb_transport_header(skb);
608         if (arfs_get_ip_proto(skb) == IPPROTO_TCP)
609                 return ((struct tcphdr *)transport_header)->dest;
610         return ((struct udphdr *)transport_header)->dest;
611 }
612
613 /* return L4 source port from ip4/6 packets */
614 static __be16 arfs_get_src_port(const struct sk_buff *skb)
615 {
616         char *transport_header;
617
618         transport_header = skb_transport_header(skb);
619         if (arfs_get_ip_proto(skb) == IPPROTO_TCP)
620                 return ((struct tcphdr *)transport_header)->source;
621         return ((struct udphdr *)transport_header)->source;
622 }
623
624 static struct arfs_rule *arfs_alloc_rule(struct mlx5e_priv *priv,
625                                          struct arfs_table *arfs_t,
626                                          const struct sk_buff *skb,
627                                          u16 rxq, u32 flow_id)
628 {
629         struct arfs_rule *rule;
630         struct arfs_tuple *tuple;
631
632         rule = kzalloc(sizeof(*rule), GFP_ATOMIC);
633         if (!rule)
634                 return NULL;
635
636         rule->priv = priv;
637         rule->rxq = rxq;
638         INIT_WORK(&rule->arfs_work, arfs_handle_work);
639
640         tuple = &rule->tuple;
641         tuple->etype = skb->protocol;
642         if (tuple->etype == htons(ETH_P_IP)) {
643                 tuple->src_ipv4 = ip_hdr(skb)->saddr;
644                 tuple->dst_ipv4 = ip_hdr(skb)->daddr;
645         } else {
646                 memcpy(&tuple->src_ipv6, &ipv6_hdr(skb)->saddr,
647                        sizeof(struct in6_addr));
648                 memcpy(&tuple->dst_ipv6, &ipv6_hdr(skb)->daddr,
649                        sizeof(struct in6_addr));
650         }
651         tuple->ip_proto = arfs_get_ip_proto(skb);
652         tuple->src_port = arfs_get_src_port(skb);
653         tuple->dst_port = arfs_get_dst_port(skb);
654
655         rule->flow_id = flow_id;
656         rule->filter_id = priv->fs.arfs.last_filter_id++ % RPS_NO_FILTER;
657
658         hlist_add_head(&rule->hlist,
659                        arfs_hash_bucket(arfs_t, tuple->src_port,
660                                         tuple->dst_port));
661         return rule;
662 }
663
664 static bool arfs_cmp_ips(struct arfs_tuple *tuple,
665                          const struct sk_buff *skb)
666 {
667         if (tuple->etype == htons(ETH_P_IP) &&
668             tuple->src_ipv4 == ip_hdr(skb)->saddr &&
669             tuple->dst_ipv4 == ip_hdr(skb)->daddr)
670                 return true;
671         if (tuple->etype == htons(ETH_P_IPV6) &&
672             (!memcmp(&tuple->src_ipv6, &ipv6_hdr(skb)->saddr,
673                      sizeof(struct in6_addr))) &&
674             (!memcmp(&tuple->dst_ipv6, &ipv6_hdr(skb)->daddr,
675                      sizeof(struct in6_addr))))
676                 return true;
677         return false;
678 }
679
680 static struct arfs_rule *arfs_find_rule(struct arfs_table *arfs_t,
681                                         const struct sk_buff *skb)
682 {
683         struct arfs_rule *arfs_rule;
684         struct hlist_head *head;
685         __be16 src_port = arfs_get_src_port(skb);
686         __be16 dst_port = arfs_get_dst_port(skb);
687
688         head = arfs_hash_bucket(arfs_t, src_port, dst_port);
689         hlist_for_each_entry(arfs_rule, head, hlist) {
690                 if (arfs_rule->tuple.src_port == src_port &&
691                     arfs_rule->tuple.dst_port == dst_port &&
692                     arfs_cmp_ips(&arfs_rule->tuple, skb)) {
693                         return arfs_rule;
694                 }
695         }
696
697         return NULL;
698 }
699
700 int mlx5e_rx_flow_steer(struct net_device *dev, const struct sk_buff *skb,
701                         u16 rxq_index, u32 flow_id)
702 {
703         struct mlx5e_priv *priv = netdev_priv(dev);
704         struct mlx5e_arfs_tables *arfs = &priv->fs.arfs;
705         struct arfs_table *arfs_t;
706         struct arfs_rule *arfs_rule;
707
708         if (skb->protocol != htons(ETH_P_IP) &&
709             skb->protocol != htons(ETH_P_IPV6))
710                 return -EPROTONOSUPPORT;
711
712         if (skb->encapsulation)
713                 return -EPROTONOSUPPORT;
714
715         arfs_t = arfs_get_table(arfs, arfs_get_ip_proto(skb), skb->protocol);
716         if (!arfs_t)
717                 return -EPROTONOSUPPORT;
718
719         spin_lock_bh(&arfs->arfs_lock);
720         arfs_rule = arfs_find_rule(arfs_t, skb);
721         if (arfs_rule) {
722                 if (arfs_rule->rxq == rxq_index) {
723                         spin_unlock_bh(&arfs->arfs_lock);
724                         return arfs_rule->filter_id;
725                 }
726                 arfs_rule->rxq = rxq_index;
727         } else {
728                 arfs_rule = arfs_alloc_rule(priv, arfs_t, skb,
729                                             rxq_index, flow_id);
730                 if (!arfs_rule) {
731                         spin_unlock_bh(&arfs->arfs_lock);
732                         return -ENOMEM;
733                 }
734         }
735         queue_work(priv->fs.arfs.wq, &arfs_rule->arfs_work);
736         spin_unlock_bh(&arfs->arfs_lock);
737         return arfs_rule->filter_id;
738 }
739