Merge git://git.kernel.org/pub/scm/linux/kernel/git/davem/net
[cascardo/linux.git] / drivers / net / ethernet / mellanox / mlx4 / resource_tracker.c
1 /*
2  * Copyright (c) 2004, 2005 Topspin Communications.  All rights reserved.
3  * Copyright (c) 2005, 2006, 2007, 2008 Mellanox Technologies.
4  * All rights reserved.
5  * Copyright (c) 2005, 2006, 2007 Cisco Systems, Inc.  All rights reserved.
6  *
7  * This software is available to you under a choice of one of two
8  * licenses.  You may choose to be licensed under the terms of the GNU
9  * General Public License (GPL) Version 2, available from the file
10  * COPYING in the main directory of this source tree, or the
11  * OpenIB.org BSD license below:
12  *
13  *     Redistribution and use in source and binary forms, with or
14  *     without modification, are permitted provided that the following
15  *     conditions are met:
16  *
17  *      - Redistributions of source code must retain the above
18  *        copyright notice, this list of conditions and the following
19  *        disclaimer.
20  *
21  *      - Redistributions in binary form must reproduce the above
22  *        copyright notice, this list of conditions and the following
23  *        disclaimer in the documentation and/or other materials
24  *        provided with the distribution.
25  *
26  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
27  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
28  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
29  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
30  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
31  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
32  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
33  * SOFTWARE.
34  */
35
36 #include <linux/sched.h>
37 #include <linux/pci.h>
38 #include <linux/errno.h>
39 #include <linux/kernel.h>
40 #include <linux/io.h>
41 #include <linux/slab.h>
42 #include <linux/mlx4/cmd.h>
43 #include <linux/mlx4/qp.h>
44 #include <linux/if_ether.h>
45 #include <linux/etherdevice.h>
46
47 #include "mlx4.h"
48 #include "fw.h"
49
50 #define MLX4_MAC_VALID          (1ull << 63)
51
52 struct mac_res {
53         struct list_head list;
54         u64 mac;
55         int ref_count;
56         u8 smac_index;
57         u8 port;
58 };
59
60 struct vlan_res {
61         struct list_head list;
62         u16 vlan;
63         int ref_count;
64         int vlan_index;
65         u8 port;
66 };
67
68 struct res_common {
69         struct list_head        list;
70         struct rb_node          node;
71         u64                     res_id;
72         int                     owner;
73         int                     state;
74         int                     from_state;
75         int                     to_state;
76         int                     removing;
77 };
78
79 enum {
80         RES_ANY_BUSY = 1
81 };
82
83 struct res_gid {
84         struct list_head        list;
85         u8                      gid[16];
86         enum mlx4_protocol      prot;
87         enum mlx4_steer_type    steer;
88         u64                     reg_id;
89 };
90
91 enum res_qp_states {
92         RES_QP_BUSY = RES_ANY_BUSY,
93
94         /* QP number was allocated */
95         RES_QP_RESERVED,
96
97         /* ICM memory for QP context was mapped */
98         RES_QP_MAPPED,
99
100         /* QP is in hw ownership */
101         RES_QP_HW
102 };
103
104 struct res_qp {
105         struct res_common       com;
106         struct res_mtt         *mtt;
107         struct res_cq          *rcq;
108         struct res_cq          *scq;
109         struct res_srq         *srq;
110         struct list_head        mcg_list;
111         spinlock_t              mcg_spl;
112         int                     local_qpn;
113         atomic_t                ref_count;
114         u32                     qpc_flags;
115         /* saved qp params before VST enforcement in order to restore on VGT */
116         u8                      sched_queue;
117         __be32                  param3;
118         u8                      vlan_control;
119         u8                      fvl_rx;
120         u8                      pri_path_fl;
121         u8                      vlan_index;
122         u8                      feup;
123 };
124
125 enum res_mtt_states {
126         RES_MTT_BUSY = RES_ANY_BUSY,
127         RES_MTT_ALLOCATED,
128 };
129
130 static inline const char *mtt_states_str(enum res_mtt_states state)
131 {
132         switch (state) {
133         case RES_MTT_BUSY: return "RES_MTT_BUSY";
134         case RES_MTT_ALLOCATED: return "RES_MTT_ALLOCATED";
135         default: return "Unknown";
136         }
137 }
138
139 struct res_mtt {
140         struct res_common       com;
141         int                     order;
142         atomic_t                ref_count;
143 };
144
145 enum res_mpt_states {
146         RES_MPT_BUSY = RES_ANY_BUSY,
147         RES_MPT_RESERVED,
148         RES_MPT_MAPPED,
149         RES_MPT_HW,
150 };
151
152 struct res_mpt {
153         struct res_common       com;
154         struct res_mtt         *mtt;
155         int                     key;
156 };
157
158 enum res_eq_states {
159         RES_EQ_BUSY = RES_ANY_BUSY,
160         RES_EQ_RESERVED,
161         RES_EQ_HW,
162 };
163
164 struct res_eq {
165         struct res_common       com;
166         struct res_mtt         *mtt;
167 };
168
169 enum res_cq_states {
170         RES_CQ_BUSY = RES_ANY_BUSY,
171         RES_CQ_ALLOCATED,
172         RES_CQ_HW,
173 };
174
175 struct res_cq {
176         struct res_common       com;
177         struct res_mtt         *mtt;
178         atomic_t                ref_count;
179 };
180
181 enum res_srq_states {
182         RES_SRQ_BUSY = RES_ANY_BUSY,
183         RES_SRQ_ALLOCATED,
184         RES_SRQ_HW,
185 };
186
187 struct res_srq {
188         struct res_common       com;
189         struct res_mtt         *mtt;
190         struct res_cq          *cq;
191         atomic_t                ref_count;
192 };
193
194 enum res_counter_states {
195         RES_COUNTER_BUSY = RES_ANY_BUSY,
196         RES_COUNTER_ALLOCATED,
197 };
198
199 struct res_counter {
200         struct res_common       com;
201         int                     port;
202 };
203
204 enum res_xrcdn_states {
205         RES_XRCD_BUSY = RES_ANY_BUSY,
206         RES_XRCD_ALLOCATED,
207 };
208
209 struct res_xrcdn {
210         struct res_common       com;
211         int                     port;
212 };
213
214 enum res_fs_rule_states {
215         RES_FS_RULE_BUSY = RES_ANY_BUSY,
216         RES_FS_RULE_ALLOCATED,
217 };
218
219 struct res_fs_rule {
220         struct res_common       com;
221         int                     qpn;
222 };
223
224 static int mlx4_is_eth(struct mlx4_dev *dev, int port)
225 {
226         return dev->caps.port_mask[port] == MLX4_PORT_TYPE_IB ? 0 : 1;
227 }
228
229 static void *res_tracker_lookup(struct rb_root *root, u64 res_id)
230 {
231         struct rb_node *node = root->rb_node;
232
233         while (node) {
234                 struct res_common *res = container_of(node, struct res_common,
235                                                       node);
236
237                 if (res_id < res->res_id)
238                         node = node->rb_left;
239                 else if (res_id > res->res_id)
240                         node = node->rb_right;
241                 else
242                         return res;
243         }
244         return NULL;
245 }
246
247 static int res_tracker_insert(struct rb_root *root, struct res_common *res)
248 {
249         struct rb_node **new = &(root->rb_node), *parent = NULL;
250
251         /* Figure out where to put new node */
252         while (*new) {
253                 struct res_common *this = container_of(*new, struct res_common,
254                                                        node);
255
256                 parent = *new;
257                 if (res->res_id < this->res_id)
258                         new = &((*new)->rb_left);
259                 else if (res->res_id > this->res_id)
260                         new = &((*new)->rb_right);
261                 else
262                         return -EEXIST;
263         }
264
265         /* Add new node and rebalance tree. */
266         rb_link_node(&res->node, parent, new);
267         rb_insert_color(&res->node, root);
268
269         return 0;
270 }
271
272 enum qp_transition {
273         QP_TRANS_INIT2RTR,
274         QP_TRANS_RTR2RTS,
275         QP_TRANS_RTS2RTS,
276         QP_TRANS_SQERR2RTS,
277         QP_TRANS_SQD2SQD,
278         QP_TRANS_SQD2RTS
279 };
280
281 /* For Debug uses */
282 static const char *resource_str(enum mlx4_resource rt)
283 {
284         switch (rt) {
285         case RES_QP: return "RES_QP";
286         case RES_CQ: return "RES_CQ";
287         case RES_SRQ: return "RES_SRQ";
288         case RES_MPT: return "RES_MPT";
289         case RES_MTT: return "RES_MTT";
290         case RES_MAC: return  "RES_MAC";
291         case RES_VLAN: return  "RES_VLAN";
292         case RES_EQ: return "RES_EQ";
293         case RES_COUNTER: return "RES_COUNTER";
294         case RES_FS_RULE: return "RES_FS_RULE";
295         case RES_XRCD: return "RES_XRCD";
296         default: return "Unknown resource type !!!";
297         };
298 }
299
300 static void rem_slave_vlans(struct mlx4_dev *dev, int slave);
301 static inline int mlx4_grant_resource(struct mlx4_dev *dev, int slave,
302                                       enum mlx4_resource res_type, int count,
303                                       int port)
304 {
305         struct mlx4_priv *priv = mlx4_priv(dev);
306         struct resource_allocator *res_alloc =
307                 &priv->mfunc.master.res_tracker.res_alloc[res_type];
308         int err = -EINVAL;
309         int allocated, free, reserved, guaranteed, from_free;
310         int from_rsvd;
311
312         if (slave > dev->num_vfs)
313                 return -EINVAL;
314
315         spin_lock(&res_alloc->alloc_lock);
316         allocated = (port > 0) ?
317                 res_alloc->allocated[(port - 1) * (dev->num_vfs + 1) + slave] :
318                 res_alloc->allocated[slave];
319         free = (port > 0) ? res_alloc->res_port_free[port - 1] :
320                 res_alloc->res_free;
321         reserved = (port > 0) ? res_alloc->res_port_rsvd[port - 1] :
322                 res_alloc->res_reserved;
323         guaranteed = res_alloc->guaranteed[slave];
324
325         if (allocated + count > res_alloc->quota[slave]) {
326                 mlx4_warn(dev, "VF %d port %d res %s: quota exceeded, count %d alloc %d quota %d\n",
327                           slave, port, resource_str(res_type), count,
328                           allocated, res_alloc->quota[slave]);
329                 goto out;
330         }
331
332         if (allocated + count <= guaranteed) {
333                 err = 0;
334                 from_rsvd = count;
335         } else {
336                 /* portion may need to be obtained from free area */
337                 if (guaranteed - allocated > 0)
338                         from_free = count - (guaranteed - allocated);
339                 else
340                         from_free = count;
341
342                 from_rsvd = count - from_free;
343
344                 if (free - from_free >= reserved)
345                         err = 0;
346                 else
347                         mlx4_warn(dev, "VF %d port %d res %s: free pool empty, free %d from_free %d rsvd %d\n",
348                                   slave, port, resource_str(res_type), free,
349                                   from_free, reserved);
350         }
351
352         if (!err) {
353                 /* grant the request */
354                 if (port > 0) {
355                         res_alloc->allocated[(port - 1) * (dev->num_vfs + 1) + slave] += count;
356                         res_alloc->res_port_free[port - 1] -= count;
357                         res_alloc->res_port_rsvd[port - 1] -= from_rsvd;
358                 } else {
359                         res_alloc->allocated[slave] += count;
360                         res_alloc->res_free -= count;
361                         res_alloc->res_reserved -= from_rsvd;
362                 }
363         }
364
365 out:
366         spin_unlock(&res_alloc->alloc_lock);
367         return err;
368 }
369
370 static inline void mlx4_release_resource(struct mlx4_dev *dev, int slave,
371                                     enum mlx4_resource res_type, int count,
372                                     int port)
373 {
374         struct mlx4_priv *priv = mlx4_priv(dev);
375         struct resource_allocator *res_alloc =
376                 &priv->mfunc.master.res_tracker.res_alloc[res_type];
377         int allocated, guaranteed, from_rsvd;
378
379         if (slave > dev->num_vfs)
380                 return;
381
382         spin_lock(&res_alloc->alloc_lock);
383
384         allocated = (port > 0) ?
385                 res_alloc->allocated[(port - 1) * (dev->num_vfs + 1) + slave] :
386                 res_alloc->allocated[slave];
387         guaranteed = res_alloc->guaranteed[slave];
388
389         if (allocated - count >= guaranteed) {
390                 from_rsvd = 0;
391         } else {
392                 /* portion may need to be returned to reserved area */
393                 if (allocated - guaranteed > 0)
394                         from_rsvd = count - (allocated - guaranteed);
395                 else
396                         from_rsvd = count;
397         }
398
399         if (port > 0) {
400                 res_alloc->allocated[(port - 1) * (dev->num_vfs + 1) + slave] -= count;
401                 res_alloc->res_port_free[port - 1] += count;
402                 res_alloc->res_port_rsvd[port - 1] += from_rsvd;
403         } else {
404                 res_alloc->allocated[slave] -= count;
405                 res_alloc->res_free += count;
406                 res_alloc->res_reserved += from_rsvd;
407         }
408
409         spin_unlock(&res_alloc->alloc_lock);
410         return;
411 }
412
413 static inline void initialize_res_quotas(struct mlx4_dev *dev,
414                                          struct resource_allocator *res_alloc,
415                                          enum mlx4_resource res_type,
416                                          int vf, int num_instances)
417 {
418         res_alloc->guaranteed[vf] = num_instances / (2 * (dev->num_vfs + 1));
419         res_alloc->quota[vf] = (num_instances / 2) + res_alloc->guaranteed[vf];
420         if (vf == mlx4_master_func_num(dev)) {
421                 res_alloc->res_free = num_instances;
422                 if (res_type == RES_MTT) {
423                         /* reserved mtts will be taken out of the PF allocation */
424                         res_alloc->res_free += dev->caps.reserved_mtts;
425                         res_alloc->guaranteed[vf] += dev->caps.reserved_mtts;
426                         res_alloc->quota[vf] += dev->caps.reserved_mtts;
427                 }
428         }
429 }
430
431 void mlx4_init_quotas(struct mlx4_dev *dev)
432 {
433         struct mlx4_priv *priv = mlx4_priv(dev);
434         int pf;
435
436         /* quotas for VFs are initialized in mlx4_slave_cap */
437         if (mlx4_is_slave(dev))
438                 return;
439
440         if (!mlx4_is_mfunc(dev)) {
441                 dev->quotas.qp = dev->caps.num_qps - dev->caps.reserved_qps -
442                         mlx4_num_reserved_sqps(dev);
443                 dev->quotas.cq = dev->caps.num_cqs - dev->caps.reserved_cqs;
444                 dev->quotas.srq = dev->caps.num_srqs - dev->caps.reserved_srqs;
445                 dev->quotas.mtt = dev->caps.num_mtts - dev->caps.reserved_mtts;
446                 dev->quotas.mpt = dev->caps.num_mpts - dev->caps.reserved_mrws;
447                 return;
448         }
449
450         pf = mlx4_master_func_num(dev);
451         dev->quotas.qp =
452                 priv->mfunc.master.res_tracker.res_alloc[RES_QP].quota[pf];
453         dev->quotas.cq =
454                 priv->mfunc.master.res_tracker.res_alloc[RES_CQ].quota[pf];
455         dev->quotas.srq =
456                 priv->mfunc.master.res_tracker.res_alloc[RES_SRQ].quota[pf];
457         dev->quotas.mtt =
458                 priv->mfunc.master.res_tracker.res_alloc[RES_MTT].quota[pf];
459         dev->quotas.mpt =
460                 priv->mfunc.master.res_tracker.res_alloc[RES_MPT].quota[pf];
461 }
462 int mlx4_init_resource_tracker(struct mlx4_dev *dev)
463 {
464         struct mlx4_priv *priv = mlx4_priv(dev);
465         int i, j;
466         int t;
467
468         priv->mfunc.master.res_tracker.slave_list =
469                 kzalloc(dev->num_slaves * sizeof(struct slave_list),
470                         GFP_KERNEL);
471         if (!priv->mfunc.master.res_tracker.slave_list)
472                 return -ENOMEM;
473
474         for (i = 0 ; i < dev->num_slaves; i++) {
475                 for (t = 0; t < MLX4_NUM_OF_RESOURCE_TYPE; ++t)
476                         INIT_LIST_HEAD(&priv->mfunc.master.res_tracker.
477                                        slave_list[i].res_list[t]);
478                 mutex_init(&priv->mfunc.master.res_tracker.slave_list[i].mutex);
479         }
480
481         mlx4_dbg(dev, "Started init_resource_tracker: %ld slaves\n",
482                  dev->num_slaves);
483         for (i = 0 ; i < MLX4_NUM_OF_RESOURCE_TYPE; i++)
484                 priv->mfunc.master.res_tracker.res_tree[i] = RB_ROOT;
485
486         for (i = 0; i < MLX4_NUM_OF_RESOURCE_TYPE; i++) {
487                 struct resource_allocator *res_alloc =
488                         &priv->mfunc.master.res_tracker.res_alloc[i];
489                 res_alloc->quota = kmalloc((dev->num_vfs + 1) * sizeof(int), GFP_KERNEL);
490                 res_alloc->guaranteed = kmalloc((dev->num_vfs + 1) * sizeof(int), GFP_KERNEL);
491                 if (i == RES_MAC || i == RES_VLAN)
492                         res_alloc->allocated = kzalloc(MLX4_MAX_PORTS *
493                                                        (dev->num_vfs + 1) * sizeof(int),
494                                                         GFP_KERNEL);
495                 else
496                         res_alloc->allocated = kzalloc((dev->num_vfs + 1) * sizeof(int), GFP_KERNEL);
497
498                 if (!res_alloc->quota || !res_alloc->guaranteed ||
499                     !res_alloc->allocated)
500                         goto no_mem_err;
501
502                 spin_lock_init(&res_alloc->alloc_lock);
503                 for (t = 0; t < dev->num_vfs + 1; t++) {
504                         struct mlx4_active_ports actv_ports =
505                                 mlx4_get_active_ports(dev, t);
506                         switch (i) {
507                         case RES_QP:
508                                 initialize_res_quotas(dev, res_alloc, RES_QP,
509                                                       t, dev->caps.num_qps -
510                                                       dev->caps.reserved_qps -
511                                                       mlx4_num_reserved_sqps(dev));
512                                 break;
513                         case RES_CQ:
514                                 initialize_res_quotas(dev, res_alloc, RES_CQ,
515                                                       t, dev->caps.num_cqs -
516                                                       dev->caps.reserved_cqs);
517                                 break;
518                         case RES_SRQ:
519                                 initialize_res_quotas(dev, res_alloc, RES_SRQ,
520                                                       t, dev->caps.num_srqs -
521                                                       dev->caps.reserved_srqs);
522                                 break;
523                         case RES_MPT:
524                                 initialize_res_quotas(dev, res_alloc, RES_MPT,
525                                                       t, dev->caps.num_mpts -
526                                                       dev->caps.reserved_mrws);
527                                 break;
528                         case RES_MTT:
529                                 initialize_res_quotas(dev, res_alloc, RES_MTT,
530                                                       t, dev->caps.num_mtts -
531                                                       dev->caps.reserved_mtts);
532                                 break;
533                         case RES_MAC:
534                                 if (t == mlx4_master_func_num(dev)) {
535                                         int max_vfs_pport = 0;
536                                         /* Calculate the max vfs per port for */
537                                         /* both ports.                        */
538                                         for (j = 0; j < dev->caps.num_ports;
539                                              j++) {
540                                                 struct mlx4_slaves_pport slaves_pport =
541                                                         mlx4_phys_to_slaves_pport(dev, j + 1);
542                                                 unsigned current_slaves =
543                                                         bitmap_weight(slaves_pport.slaves,
544                                                                       dev->caps.num_ports) - 1;
545                                                 if (max_vfs_pport < current_slaves)
546                                                         max_vfs_pport =
547                                                                 current_slaves;
548                                         }
549                                         res_alloc->quota[t] =
550                                                 MLX4_MAX_MAC_NUM -
551                                                 2 * max_vfs_pport;
552                                         res_alloc->guaranteed[t] = 2;
553                                         for (j = 0; j < MLX4_MAX_PORTS; j++)
554                                                 res_alloc->res_port_free[j] =
555                                                         MLX4_MAX_MAC_NUM;
556                                 } else {
557                                         res_alloc->quota[t] = MLX4_MAX_MAC_NUM;
558                                         res_alloc->guaranteed[t] = 2;
559                                 }
560                                 break;
561                         case RES_VLAN:
562                                 if (t == mlx4_master_func_num(dev)) {
563                                         res_alloc->quota[t] = MLX4_MAX_VLAN_NUM;
564                                         res_alloc->guaranteed[t] = MLX4_MAX_VLAN_NUM / 2;
565                                         for (j = 0; j < MLX4_MAX_PORTS; j++)
566                                                 res_alloc->res_port_free[j] =
567                                                         res_alloc->quota[t];
568                                 } else {
569                                         res_alloc->quota[t] = MLX4_MAX_VLAN_NUM / 2;
570                                         res_alloc->guaranteed[t] = 0;
571                                 }
572                                 break;
573                         case RES_COUNTER:
574                                 res_alloc->quota[t] = dev->caps.max_counters;
575                                 res_alloc->guaranteed[t] = 0;
576                                 if (t == mlx4_master_func_num(dev))
577                                         res_alloc->res_free = res_alloc->quota[t];
578                                 break;
579                         default:
580                                 break;
581                         }
582                         if (i == RES_MAC || i == RES_VLAN) {
583                                 for (j = 0; j < dev->caps.num_ports; j++)
584                                         if (test_bit(j, actv_ports.ports))
585                                                 res_alloc->res_port_rsvd[j] +=
586                                                         res_alloc->guaranteed[t];
587                         } else {
588                                 res_alloc->res_reserved += res_alloc->guaranteed[t];
589                         }
590                 }
591         }
592         spin_lock_init(&priv->mfunc.master.res_tracker.lock);
593         return 0;
594
595 no_mem_err:
596         for (i = 0; i < MLX4_NUM_OF_RESOURCE_TYPE; i++) {
597                 kfree(priv->mfunc.master.res_tracker.res_alloc[i].allocated);
598                 priv->mfunc.master.res_tracker.res_alloc[i].allocated = NULL;
599                 kfree(priv->mfunc.master.res_tracker.res_alloc[i].guaranteed);
600                 priv->mfunc.master.res_tracker.res_alloc[i].guaranteed = NULL;
601                 kfree(priv->mfunc.master.res_tracker.res_alloc[i].quota);
602                 priv->mfunc.master.res_tracker.res_alloc[i].quota = NULL;
603         }
604         return -ENOMEM;
605 }
606
607 void mlx4_free_resource_tracker(struct mlx4_dev *dev,
608                                 enum mlx4_res_tracker_free_type type)
609 {
610         struct mlx4_priv *priv = mlx4_priv(dev);
611         int i;
612
613         if (priv->mfunc.master.res_tracker.slave_list) {
614                 if (type != RES_TR_FREE_STRUCTS_ONLY) {
615                         for (i = 0; i < dev->num_slaves; i++) {
616                                 if (type == RES_TR_FREE_ALL ||
617                                     dev->caps.function != i)
618                                         mlx4_delete_all_resources_for_slave(dev, i);
619                         }
620                         /* free master's vlans */
621                         i = dev->caps.function;
622                         mlx4_reset_roce_gids(dev, i);
623                         mutex_lock(&priv->mfunc.master.res_tracker.slave_list[i].mutex);
624                         rem_slave_vlans(dev, i);
625                         mutex_unlock(&priv->mfunc.master.res_tracker.slave_list[i].mutex);
626                 }
627
628                 if (type != RES_TR_FREE_SLAVES_ONLY) {
629                         for (i = 0; i < MLX4_NUM_OF_RESOURCE_TYPE; i++) {
630                                 kfree(priv->mfunc.master.res_tracker.res_alloc[i].allocated);
631                                 priv->mfunc.master.res_tracker.res_alloc[i].allocated = NULL;
632                                 kfree(priv->mfunc.master.res_tracker.res_alloc[i].guaranteed);
633                                 priv->mfunc.master.res_tracker.res_alloc[i].guaranteed = NULL;
634                                 kfree(priv->mfunc.master.res_tracker.res_alloc[i].quota);
635                                 priv->mfunc.master.res_tracker.res_alloc[i].quota = NULL;
636                         }
637                         kfree(priv->mfunc.master.res_tracker.slave_list);
638                         priv->mfunc.master.res_tracker.slave_list = NULL;
639                 }
640         }
641 }
642
643 static void update_pkey_index(struct mlx4_dev *dev, int slave,
644                               struct mlx4_cmd_mailbox *inbox)
645 {
646         u8 sched = *(u8 *)(inbox->buf + 64);
647         u8 orig_index = *(u8 *)(inbox->buf + 35);
648         u8 new_index;
649         struct mlx4_priv *priv = mlx4_priv(dev);
650         int port;
651
652         port = (sched >> 6 & 1) + 1;
653
654         new_index = priv->virt2phys_pkey[slave][port - 1][orig_index];
655         *(u8 *)(inbox->buf + 35) = new_index;
656 }
657
658 static void update_gid(struct mlx4_dev *dev, struct mlx4_cmd_mailbox *inbox,
659                        u8 slave)
660 {
661         struct mlx4_qp_context  *qp_ctx = inbox->buf + 8;
662         enum mlx4_qp_optpar     optpar = be32_to_cpu(*(__be32 *) inbox->buf);
663         u32                     ts = (be32_to_cpu(qp_ctx->flags) >> 16) & 0xff;
664         int port;
665
666         if (MLX4_QP_ST_UD == ts) {
667                 port = (qp_ctx->pri_path.sched_queue >> 6 & 1) + 1;
668                 if (mlx4_is_eth(dev, port))
669                         qp_ctx->pri_path.mgid_index =
670                                 mlx4_get_base_gid_ix(dev, slave, port) | 0x80;
671                 else
672                         qp_ctx->pri_path.mgid_index = slave | 0x80;
673
674         } else if (MLX4_QP_ST_RC == ts || MLX4_QP_ST_XRC == ts || MLX4_QP_ST_UC == ts) {
675                 if (optpar & MLX4_QP_OPTPAR_PRIMARY_ADDR_PATH) {
676                         port = (qp_ctx->pri_path.sched_queue >> 6 & 1) + 1;
677                         if (mlx4_is_eth(dev, port)) {
678                                 qp_ctx->pri_path.mgid_index +=
679                                         mlx4_get_base_gid_ix(dev, slave, port);
680                                 qp_ctx->pri_path.mgid_index &= 0x7f;
681                         } else {
682                                 qp_ctx->pri_path.mgid_index = slave & 0x7F;
683                         }
684                 }
685                 if (optpar & MLX4_QP_OPTPAR_ALT_ADDR_PATH) {
686                         port = (qp_ctx->alt_path.sched_queue >> 6 & 1) + 1;
687                         if (mlx4_is_eth(dev, port)) {
688                                 qp_ctx->alt_path.mgid_index +=
689                                         mlx4_get_base_gid_ix(dev, slave, port);
690                                 qp_ctx->alt_path.mgid_index &= 0x7f;
691                         } else {
692                                 qp_ctx->alt_path.mgid_index = slave & 0x7F;
693                         }
694                 }
695         }
696 }
697
698 static int update_vport_qp_param(struct mlx4_dev *dev,
699                                  struct mlx4_cmd_mailbox *inbox,
700                                  u8 slave, u32 qpn)
701 {
702         struct mlx4_qp_context  *qpc = inbox->buf + 8;
703         struct mlx4_vport_oper_state *vp_oper;
704         struct mlx4_priv *priv;
705         u32 qp_type;
706         int port;
707
708         port = (qpc->pri_path.sched_queue & 0x40) ? 2 : 1;
709         priv = mlx4_priv(dev);
710         vp_oper = &priv->mfunc.master.vf_oper[slave].vport[port];
711         qp_type = (be32_to_cpu(qpc->flags) >> 16) & 0xff;
712
713         if (MLX4_VGT != vp_oper->state.default_vlan) {
714                 /* the reserved QPs (special, proxy, tunnel)
715                  * do not operate over vlans
716                  */
717                 if (mlx4_is_qp_reserved(dev, qpn))
718                         return 0;
719
720                 /* force strip vlan by clear vsd, MLX QP refers to Raw Ethernet */
721                 if (qp_type == MLX4_QP_ST_UD ||
722                     (qp_type == MLX4_QP_ST_MLX && mlx4_is_eth(dev, port))) {
723                         if (dev->caps.bmme_flags & MLX4_BMME_FLAG_VSD_INIT2RTR) {
724                                 *(__be32 *)inbox->buf =
725                                         cpu_to_be32(be32_to_cpu(*(__be32 *)inbox->buf) |
726                                         MLX4_QP_OPTPAR_VLAN_STRIPPING);
727                                 qpc->param3 &= ~cpu_to_be32(MLX4_STRIP_VLAN);
728                         } else {
729                                 struct mlx4_update_qp_params params = {.flags = 0};
730
731                                 mlx4_update_qp(dev, qpn, MLX4_UPDATE_QP_VSD, &params);
732                         }
733                 }
734
735                 if (vp_oper->state.link_state == IFLA_VF_LINK_STATE_DISABLE &&
736                     dev->caps.flags2 & MLX4_DEV_CAP_FLAG2_UPDATE_QP) {
737                         qpc->pri_path.vlan_control =
738                                 MLX4_VLAN_CTRL_ETH_TX_BLOCK_TAGGED |
739                                 MLX4_VLAN_CTRL_ETH_TX_BLOCK_PRIO_TAGGED |
740                                 MLX4_VLAN_CTRL_ETH_TX_BLOCK_UNTAGGED |
741                                 MLX4_VLAN_CTRL_ETH_RX_BLOCK_PRIO_TAGGED |
742                                 MLX4_VLAN_CTRL_ETH_RX_BLOCK_UNTAGGED |
743                                 MLX4_VLAN_CTRL_ETH_RX_BLOCK_TAGGED;
744                 } else if (0 != vp_oper->state.default_vlan) {
745                         qpc->pri_path.vlan_control =
746                                 MLX4_VLAN_CTRL_ETH_TX_BLOCK_TAGGED |
747                                 MLX4_VLAN_CTRL_ETH_RX_BLOCK_PRIO_TAGGED |
748                                 MLX4_VLAN_CTRL_ETH_RX_BLOCK_UNTAGGED;
749                 } else { /* priority tagged */
750                         qpc->pri_path.vlan_control =
751                                 MLX4_VLAN_CTRL_ETH_TX_BLOCK_TAGGED |
752                                 MLX4_VLAN_CTRL_ETH_RX_BLOCK_TAGGED;
753                 }
754
755                 qpc->pri_path.fvl_rx |= MLX4_FVL_RX_FORCE_ETH_VLAN;
756                 qpc->pri_path.vlan_index = vp_oper->vlan_idx;
757                 qpc->pri_path.fl |= MLX4_FL_CV | MLX4_FL_ETH_HIDE_CQE_VLAN;
758                 qpc->pri_path.feup |= MLX4_FEUP_FORCE_ETH_UP | MLX4_FVL_FORCE_ETH_VLAN;
759                 qpc->pri_path.sched_queue &= 0xC7;
760                 qpc->pri_path.sched_queue |= (vp_oper->state.default_qos) << 3;
761         }
762         if (vp_oper->state.spoofchk) {
763                 qpc->pri_path.feup |= MLX4_FSM_FORCE_ETH_SRC_MAC;
764                 qpc->pri_path.grh_mylmc = (0x80 & qpc->pri_path.grh_mylmc) + vp_oper->mac_idx;
765         }
766         return 0;
767 }
768
769 static int mpt_mask(struct mlx4_dev *dev)
770 {
771         return dev->caps.num_mpts - 1;
772 }
773
774 static void *find_res(struct mlx4_dev *dev, u64 res_id,
775                       enum mlx4_resource type)
776 {
777         struct mlx4_priv *priv = mlx4_priv(dev);
778
779         return res_tracker_lookup(&priv->mfunc.master.res_tracker.res_tree[type],
780                                   res_id);
781 }
782
783 static int get_res(struct mlx4_dev *dev, int slave, u64 res_id,
784                    enum mlx4_resource type,
785                    void *res)
786 {
787         struct res_common *r;
788         int err = 0;
789
790         spin_lock_irq(mlx4_tlock(dev));
791         r = find_res(dev, res_id, type);
792         if (!r) {
793                 err = -ENONET;
794                 goto exit;
795         }
796
797         if (r->state == RES_ANY_BUSY) {
798                 err = -EBUSY;
799                 goto exit;
800         }
801
802         if (r->owner != slave) {
803                 err = -EPERM;
804                 goto exit;
805         }
806
807         r->from_state = r->state;
808         r->state = RES_ANY_BUSY;
809
810         if (res)
811                 *((struct res_common **)res) = r;
812
813 exit:
814         spin_unlock_irq(mlx4_tlock(dev));
815         return err;
816 }
817
818 int mlx4_get_slave_from_resource_id(struct mlx4_dev *dev,
819                                     enum mlx4_resource type,
820                                     u64 res_id, int *slave)
821 {
822
823         struct res_common *r;
824         int err = -ENOENT;
825         int id = res_id;
826
827         if (type == RES_QP)
828                 id &= 0x7fffff;
829         spin_lock(mlx4_tlock(dev));
830
831         r = find_res(dev, id, type);
832         if (r) {
833                 *slave = r->owner;
834                 err = 0;
835         }
836         spin_unlock(mlx4_tlock(dev));
837
838         return err;
839 }
840
841 static void put_res(struct mlx4_dev *dev, int slave, u64 res_id,
842                     enum mlx4_resource type)
843 {
844         struct res_common *r;
845
846         spin_lock_irq(mlx4_tlock(dev));
847         r = find_res(dev, res_id, type);
848         if (r)
849                 r->state = r->from_state;
850         spin_unlock_irq(mlx4_tlock(dev));
851 }
852
853 static struct res_common *alloc_qp_tr(int id)
854 {
855         struct res_qp *ret;
856
857         ret = kzalloc(sizeof *ret, GFP_KERNEL);
858         if (!ret)
859                 return NULL;
860
861         ret->com.res_id = id;
862         ret->com.state = RES_QP_RESERVED;
863         ret->local_qpn = id;
864         INIT_LIST_HEAD(&ret->mcg_list);
865         spin_lock_init(&ret->mcg_spl);
866         atomic_set(&ret->ref_count, 0);
867
868         return &ret->com;
869 }
870
871 static struct res_common *alloc_mtt_tr(int id, int order)
872 {
873         struct res_mtt *ret;
874
875         ret = kzalloc(sizeof *ret, GFP_KERNEL);
876         if (!ret)
877                 return NULL;
878
879         ret->com.res_id = id;
880         ret->order = order;
881         ret->com.state = RES_MTT_ALLOCATED;
882         atomic_set(&ret->ref_count, 0);
883
884         return &ret->com;
885 }
886
887 static struct res_common *alloc_mpt_tr(int id, int key)
888 {
889         struct res_mpt *ret;
890
891         ret = kzalloc(sizeof *ret, GFP_KERNEL);
892         if (!ret)
893                 return NULL;
894
895         ret->com.res_id = id;
896         ret->com.state = RES_MPT_RESERVED;
897         ret->key = key;
898
899         return &ret->com;
900 }
901
902 static struct res_common *alloc_eq_tr(int id)
903 {
904         struct res_eq *ret;
905
906         ret = kzalloc(sizeof *ret, GFP_KERNEL);
907         if (!ret)
908                 return NULL;
909
910         ret->com.res_id = id;
911         ret->com.state = RES_EQ_RESERVED;
912
913         return &ret->com;
914 }
915
916 static struct res_common *alloc_cq_tr(int id)
917 {
918         struct res_cq *ret;
919
920         ret = kzalloc(sizeof *ret, GFP_KERNEL);
921         if (!ret)
922                 return NULL;
923
924         ret->com.res_id = id;
925         ret->com.state = RES_CQ_ALLOCATED;
926         atomic_set(&ret->ref_count, 0);
927
928         return &ret->com;
929 }
930
931 static struct res_common *alloc_srq_tr(int id)
932 {
933         struct res_srq *ret;
934
935         ret = kzalloc(sizeof *ret, GFP_KERNEL);
936         if (!ret)
937                 return NULL;
938
939         ret->com.res_id = id;
940         ret->com.state = RES_SRQ_ALLOCATED;
941         atomic_set(&ret->ref_count, 0);
942
943         return &ret->com;
944 }
945
946 static struct res_common *alloc_counter_tr(int id)
947 {
948         struct res_counter *ret;
949
950         ret = kzalloc(sizeof *ret, GFP_KERNEL);
951         if (!ret)
952                 return NULL;
953
954         ret->com.res_id = id;
955         ret->com.state = RES_COUNTER_ALLOCATED;
956
957         return &ret->com;
958 }
959
960 static struct res_common *alloc_xrcdn_tr(int id)
961 {
962         struct res_xrcdn *ret;
963
964         ret = kzalloc(sizeof *ret, GFP_KERNEL);
965         if (!ret)
966                 return NULL;
967
968         ret->com.res_id = id;
969         ret->com.state = RES_XRCD_ALLOCATED;
970
971         return &ret->com;
972 }
973
974 static struct res_common *alloc_fs_rule_tr(u64 id, int qpn)
975 {
976         struct res_fs_rule *ret;
977
978         ret = kzalloc(sizeof *ret, GFP_KERNEL);
979         if (!ret)
980                 return NULL;
981
982         ret->com.res_id = id;
983         ret->com.state = RES_FS_RULE_ALLOCATED;
984         ret->qpn = qpn;
985         return &ret->com;
986 }
987
988 static struct res_common *alloc_tr(u64 id, enum mlx4_resource type, int slave,
989                                    int extra)
990 {
991         struct res_common *ret;
992
993         switch (type) {
994         case RES_QP:
995                 ret = alloc_qp_tr(id);
996                 break;
997         case RES_MPT:
998                 ret = alloc_mpt_tr(id, extra);
999                 break;
1000         case RES_MTT:
1001                 ret = alloc_mtt_tr(id, extra);
1002                 break;
1003         case RES_EQ:
1004                 ret = alloc_eq_tr(id);
1005                 break;
1006         case RES_CQ:
1007                 ret = alloc_cq_tr(id);
1008                 break;
1009         case RES_SRQ:
1010                 ret = alloc_srq_tr(id);
1011                 break;
1012         case RES_MAC:
1013                 pr_err("implementation missing\n");
1014                 return NULL;
1015         case RES_COUNTER:
1016                 ret = alloc_counter_tr(id);
1017                 break;
1018         case RES_XRCD:
1019                 ret = alloc_xrcdn_tr(id);
1020                 break;
1021         case RES_FS_RULE:
1022                 ret = alloc_fs_rule_tr(id, extra);
1023                 break;
1024         default:
1025                 return NULL;
1026         }
1027         if (ret)
1028                 ret->owner = slave;
1029
1030         return ret;
1031 }
1032
1033 static int add_res_range(struct mlx4_dev *dev, int slave, u64 base, int count,
1034                          enum mlx4_resource type, int extra)
1035 {
1036         int i;
1037         int err;
1038         struct mlx4_priv *priv = mlx4_priv(dev);
1039         struct res_common **res_arr;
1040         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
1041         struct rb_root *root = &tracker->res_tree[type];
1042
1043         res_arr = kzalloc(count * sizeof *res_arr, GFP_KERNEL);
1044         if (!res_arr)
1045                 return -ENOMEM;
1046
1047         for (i = 0; i < count; ++i) {
1048                 res_arr[i] = alloc_tr(base + i, type, slave, extra);
1049                 if (!res_arr[i]) {
1050                         for (--i; i >= 0; --i)
1051                                 kfree(res_arr[i]);
1052
1053                         kfree(res_arr);
1054                         return -ENOMEM;
1055                 }
1056         }
1057
1058         spin_lock_irq(mlx4_tlock(dev));
1059         for (i = 0; i < count; ++i) {
1060                 if (find_res(dev, base + i, type)) {
1061                         err = -EEXIST;
1062                         goto undo;
1063                 }
1064                 err = res_tracker_insert(root, res_arr[i]);
1065                 if (err)
1066                         goto undo;
1067                 list_add_tail(&res_arr[i]->list,
1068                               &tracker->slave_list[slave].res_list[type]);
1069         }
1070         spin_unlock_irq(mlx4_tlock(dev));
1071         kfree(res_arr);
1072
1073         return 0;
1074
1075 undo:
1076         for (--i; i >= base; --i)
1077                 rb_erase(&res_arr[i]->node, root);
1078
1079         spin_unlock_irq(mlx4_tlock(dev));
1080
1081         for (i = 0; i < count; ++i)
1082                 kfree(res_arr[i]);
1083
1084         kfree(res_arr);
1085
1086         return err;
1087 }
1088
1089 static int remove_qp_ok(struct res_qp *res)
1090 {
1091         if (res->com.state == RES_QP_BUSY || atomic_read(&res->ref_count) ||
1092             !list_empty(&res->mcg_list)) {
1093                 pr_err("resource tracker: fail to remove qp, state %d, ref_count %d\n",
1094                        res->com.state, atomic_read(&res->ref_count));
1095                 return -EBUSY;
1096         } else if (res->com.state != RES_QP_RESERVED) {
1097                 return -EPERM;
1098         }
1099
1100         return 0;
1101 }
1102
1103 static int remove_mtt_ok(struct res_mtt *res, int order)
1104 {
1105         if (res->com.state == RES_MTT_BUSY ||
1106             atomic_read(&res->ref_count)) {
1107                 pr_devel("%s-%d: state %s, ref_count %d\n",
1108                          __func__, __LINE__,
1109                          mtt_states_str(res->com.state),
1110                          atomic_read(&res->ref_count));
1111                 return -EBUSY;
1112         } else if (res->com.state != RES_MTT_ALLOCATED)
1113                 return -EPERM;
1114         else if (res->order != order)
1115                 return -EINVAL;
1116
1117         return 0;
1118 }
1119
1120 static int remove_mpt_ok(struct res_mpt *res)
1121 {
1122         if (res->com.state == RES_MPT_BUSY)
1123                 return -EBUSY;
1124         else if (res->com.state != RES_MPT_RESERVED)
1125                 return -EPERM;
1126
1127         return 0;
1128 }
1129
1130 static int remove_eq_ok(struct res_eq *res)
1131 {
1132         if (res->com.state == RES_MPT_BUSY)
1133                 return -EBUSY;
1134         else if (res->com.state != RES_MPT_RESERVED)
1135                 return -EPERM;
1136
1137         return 0;
1138 }
1139
1140 static int remove_counter_ok(struct res_counter *res)
1141 {
1142         if (res->com.state == RES_COUNTER_BUSY)
1143                 return -EBUSY;
1144         else if (res->com.state != RES_COUNTER_ALLOCATED)
1145                 return -EPERM;
1146
1147         return 0;
1148 }
1149
1150 static int remove_xrcdn_ok(struct res_xrcdn *res)
1151 {
1152         if (res->com.state == RES_XRCD_BUSY)
1153                 return -EBUSY;
1154         else if (res->com.state != RES_XRCD_ALLOCATED)
1155                 return -EPERM;
1156
1157         return 0;
1158 }
1159
1160 static int remove_fs_rule_ok(struct res_fs_rule *res)
1161 {
1162         if (res->com.state == RES_FS_RULE_BUSY)
1163                 return -EBUSY;
1164         else if (res->com.state != RES_FS_RULE_ALLOCATED)
1165                 return -EPERM;
1166
1167         return 0;
1168 }
1169
1170 static int remove_cq_ok(struct res_cq *res)
1171 {
1172         if (res->com.state == RES_CQ_BUSY)
1173                 return -EBUSY;
1174         else if (res->com.state != RES_CQ_ALLOCATED)
1175                 return -EPERM;
1176
1177         return 0;
1178 }
1179
1180 static int remove_srq_ok(struct res_srq *res)
1181 {
1182         if (res->com.state == RES_SRQ_BUSY)
1183                 return -EBUSY;
1184         else if (res->com.state != RES_SRQ_ALLOCATED)
1185                 return -EPERM;
1186
1187         return 0;
1188 }
1189
1190 static int remove_ok(struct res_common *res, enum mlx4_resource type, int extra)
1191 {
1192         switch (type) {
1193         case RES_QP:
1194                 return remove_qp_ok((struct res_qp *)res);
1195         case RES_CQ:
1196                 return remove_cq_ok((struct res_cq *)res);
1197         case RES_SRQ:
1198                 return remove_srq_ok((struct res_srq *)res);
1199         case RES_MPT:
1200                 return remove_mpt_ok((struct res_mpt *)res);
1201         case RES_MTT:
1202                 return remove_mtt_ok((struct res_mtt *)res, extra);
1203         case RES_MAC:
1204                 return -ENOSYS;
1205         case RES_EQ:
1206                 return remove_eq_ok((struct res_eq *)res);
1207         case RES_COUNTER:
1208                 return remove_counter_ok((struct res_counter *)res);
1209         case RES_XRCD:
1210                 return remove_xrcdn_ok((struct res_xrcdn *)res);
1211         case RES_FS_RULE:
1212                 return remove_fs_rule_ok((struct res_fs_rule *)res);
1213         default:
1214                 return -EINVAL;
1215         }
1216 }
1217
1218 static int rem_res_range(struct mlx4_dev *dev, int slave, u64 base, int count,
1219                          enum mlx4_resource type, int extra)
1220 {
1221         u64 i;
1222         int err;
1223         struct mlx4_priv *priv = mlx4_priv(dev);
1224         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
1225         struct res_common *r;
1226
1227         spin_lock_irq(mlx4_tlock(dev));
1228         for (i = base; i < base + count; ++i) {
1229                 r = res_tracker_lookup(&tracker->res_tree[type], i);
1230                 if (!r) {
1231                         err = -ENOENT;
1232                         goto out;
1233                 }
1234                 if (r->owner != slave) {
1235                         err = -EPERM;
1236                         goto out;
1237                 }
1238                 err = remove_ok(r, type, extra);
1239                 if (err)
1240                         goto out;
1241         }
1242
1243         for (i = base; i < base + count; ++i) {
1244                 r = res_tracker_lookup(&tracker->res_tree[type], i);
1245                 rb_erase(&r->node, &tracker->res_tree[type]);
1246                 list_del(&r->list);
1247                 kfree(r);
1248         }
1249         err = 0;
1250
1251 out:
1252         spin_unlock_irq(mlx4_tlock(dev));
1253
1254         return err;
1255 }
1256
1257 static int qp_res_start_move_to(struct mlx4_dev *dev, int slave, int qpn,
1258                                 enum res_qp_states state, struct res_qp **qp,
1259                                 int alloc)
1260 {
1261         struct mlx4_priv *priv = mlx4_priv(dev);
1262         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
1263         struct res_qp *r;
1264         int err = 0;
1265
1266         spin_lock_irq(mlx4_tlock(dev));
1267         r = res_tracker_lookup(&tracker->res_tree[RES_QP], qpn);
1268         if (!r)
1269                 err = -ENOENT;
1270         else if (r->com.owner != slave)
1271                 err = -EPERM;
1272         else {
1273                 switch (state) {
1274                 case RES_QP_BUSY:
1275                         mlx4_dbg(dev, "%s: failed RES_QP, 0x%llx\n",
1276                                  __func__, r->com.res_id);
1277                         err = -EBUSY;
1278                         break;
1279
1280                 case RES_QP_RESERVED:
1281                         if (r->com.state == RES_QP_MAPPED && !alloc)
1282                                 break;
1283
1284                         mlx4_dbg(dev, "failed RES_QP, 0x%llx\n", r->com.res_id);
1285                         err = -EINVAL;
1286                         break;
1287
1288                 case RES_QP_MAPPED:
1289                         if ((r->com.state == RES_QP_RESERVED && alloc) ||
1290                             r->com.state == RES_QP_HW)
1291                                 break;
1292                         else {
1293                                 mlx4_dbg(dev, "failed RES_QP, 0x%llx\n",
1294                                           r->com.res_id);
1295                                 err = -EINVAL;
1296                         }
1297
1298                         break;
1299
1300                 case RES_QP_HW:
1301                         if (r->com.state != RES_QP_MAPPED)
1302                                 err = -EINVAL;
1303                         break;
1304                 default:
1305                         err = -EINVAL;
1306                 }
1307
1308                 if (!err) {
1309                         r->com.from_state = r->com.state;
1310                         r->com.to_state = state;
1311                         r->com.state = RES_QP_BUSY;
1312                         if (qp)
1313                                 *qp = r;
1314                 }
1315         }
1316
1317         spin_unlock_irq(mlx4_tlock(dev));
1318
1319         return err;
1320 }
1321
1322 static int mr_res_start_move_to(struct mlx4_dev *dev, int slave, int index,
1323                                 enum res_mpt_states state, struct res_mpt **mpt)
1324 {
1325         struct mlx4_priv *priv = mlx4_priv(dev);
1326         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
1327         struct res_mpt *r;
1328         int err = 0;
1329
1330         spin_lock_irq(mlx4_tlock(dev));
1331         r = res_tracker_lookup(&tracker->res_tree[RES_MPT], index);
1332         if (!r)
1333                 err = -ENOENT;
1334         else if (r->com.owner != slave)
1335                 err = -EPERM;
1336         else {
1337                 switch (state) {
1338                 case RES_MPT_BUSY:
1339                         err = -EINVAL;
1340                         break;
1341
1342                 case RES_MPT_RESERVED:
1343                         if (r->com.state != RES_MPT_MAPPED)
1344                                 err = -EINVAL;
1345                         break;
1346
1347                 case RES_MPT_MAPPED:
1348                         if (r->com.state != RES_MPT_RESERVED &&
1349                             r->com.state != RES_MPT_HW)
1350                                 err = -EINVAL;
1351                         break;
1352
1353                 case RES_MPT_HW:
1354                         if (r->com.state != RES_MPT_MAPPED)
1355                                 err = -EINVAL;
1356                         break;
1357                 default:
1358                         err = -EINVAL;
1359                 }
1360
1361                 if (!err) {
1362                         r->com.from_state = r->com.state;
1363                         r->com.to_state = state;
1364                         r->com.state = RES_MPT_BUSY;
1365                         if (mpt)
1366                                 *mpt = r;
1367                 }
1368         }
1369
1370         spin_unlock_irq(mlx4_tlock(dev));
1371
1372         return err;
1373 }
1374
1375 static int eq_res_start_move_to(struct mlx4_dev *dev, int slave, int index,
1376                                 enum res_eq_states state, struct res_eq **eq)
1377 {
1378         struct mlx4_priv *priv = mlx4_priv(dev);
1379         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
1380         struct res_eq *r;
1381         int err = 0;
1382
1383         spin_lock_irq(mlx4_tlock(dev));
1384         r = res_tracker_lookup(&tracker->res_tree[RES_EQ], index);
1385         if (!r)
1386                 err = -ENOENT;
1387         else if (r->com.owner != slave)
1388                 err = -EPERM;
1389         else {
1390                 switch (state) {
1391                 case RES_EQ_BUSY:
1392                         err = -EINVAL;
1393                         break;
1394
1395                 case RES_EQ_RESERVED:
1396                         if (r->com.state != RES_EQ_HW)
1397                                 err = -EINVAL;
1398                         break;
1399
1400                 case RES_EQ_HW:
1401                         if (r->com.state != RES_EQ_RESERVED)
1402                                 err = -EINVAL;
1403                         break;
1404
1405                 default:
1406                         err = -EINVAL;
1407                 }
1408
1409                 if (!err) {
1410                         r->com.from_state = r->com.state;
1411                         r->com.to_state = state;
1412                         r->com.state = RES_EQ_BUSY;
1413                         if (eq)
1414                                 *eq = r;
1415                 }
1416         }
1417
1418         spin_unlock_irq(mlx4_tlock(dev));
1419
1420         return err;
1421 }
1422
1423 static int cq_res_start_move_to(struct mlx4_dev *dev, int slave, int cqn,
1424                                 enum res_cq_states state, struct res_cq **cq)
1425 {
1426         struct mlx4_priv *priv = mlx4_priv(dev);
1427         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
1428         struct res_cq *r;
1429         int err;
1430
1431         spin_lock_irq(mlx4_tlock(dev));
1432         r = res_tracker_lookup(&tracker->res_tree[RES_CQ], cqn);
1433         if (!r) {
1434                 err = -ENOENT;
1435         } else if (r->com.owner != slave) {
1436                 err = -EPERM;
1437         } else if (state == RES_CQ_ALLOCATED) {
1438                 if (r->com.state != RES_CQ_HW)
1439                         err = -EINVAL;
1440                 else if (atomic_read(&r->ref_count))
1441                         err = -EBUSY;
1442                 else
1443                         err = 0;
1444         } else if (state != RES_CQ_HW || r->com.state != RES_CQ_ALLOCATED) {
1445                 err = -EINVAL;
1446         } else {
1447                 err = 0;
1448         }
1449
1450         if (!err) {
1451                 r->com.from_state = r->com.state;
1452                 r->com.to_state = state;
1453                 r->com.state = RES_CQ_BUSY;
1454                 if (cq)
1455                         *cq = r;
1456         }
1457
1458         spin_unlock_irq(mlx4_tlock(dev));
1459
1460         return err;
1461 }
1462
1463 static int srq_res_start_move_to(struct mlx4_dev *dev, int slave, int index,
1464                                  enum res_srq_states state, struct res_srq **srq)
1465 {
1466         struct mlx4_priv *priv = mlx4_priv(dev);
1467         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
1468         struct res_srq *r;
1469         int err = 0;
1470
1471         spin_lock_irq(mlx4_tlock(dev));
1472         r = res_tracker_lookup(&tracker->res_tree[RES_SRQ], index);
1473         if (!r) {
1474                 err = -ENOENT;
1475         } else if (r->com.owner != slave) {
1476                 err = -EPERM;
1477         } else if (state == RES_SRQ_ALLOCATED) {
1478                 if (r->com.state != RES_SRQ_HW)
1479                         err = -EINVAL;
1480                 else if (atomic_read(&r->ref_count))
1481                         err = -EBUSY;
1482         } else if (state != RES_SRQ_HW || r->com.state != RES_SRQ_ALLOCATED) {
1483                 err = -EINVAL;
1484         }
1485
1486         if (!err) {
1487                 r->com.from_state = r->com.state;
1488                 r->com.to_state = state;
1489                 r->com.state = RES_SRQ_BUSY;
1490                 if (srq)
1491                         *srq = r;
1492         }
1493
1494         spin_unlock_irq(mlx4_tlock(dev));
1495
1496         return err;
1497 }
1498
1499 static void res_abort_move(struct mlx4_dev *dev, int slave,
1500                            enum mlx4_resource type, int id)
1501 {
1502         struct mlx4_priv *priv = mlx4_priv(dev);
1503         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
1504         struct res_common *r;
1505
1506         spin_lock_irq(mlx4_tlock(dev));
1507         r = res_tracker_lookup(&tracker->res_tree[type], id);
1508         if (r && (r->owner == slave))
1509                 r->state = r->from_state;
1510         spin_unlock_irq(mlx4_tlock(dev));
1511 }
1512
1513 static void res_end_move(struct mlx4_dev *dev, int slave,
1514                          enum mlx4_resource type, int id)
1515 {
1516         struct mlx4_priv *priv = mlx4_priv(dev);
1517         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
1518         struct res_common *r;
1519
1520         spin_lock_irq(mlx4_tlock(dev));
1521         r = res_tracker_lookup(&tracker->res_tree[type], id);
1522         if (r && (r->owner == slave))
1523                 r->state = r->to_state;
1524         spin_unlock_irq(mlx4_tlock(dev));
1525 }
1526
1527 static int valid_reserved(struct mlx4_dev *dev, int slave, int qpn)
1528 {
1529         return mlx4_is_qp_reserved(dev, qpn) &&
1530                 (mlx4_is_master(dev) || mlx4_is_guest_proxy(dev, slave, qpn));
1531 }
1532
1533 static int fw_reserved(struct mlx4_dev *dev, int qpn)
1534 {
1535         return qpn < dev->caps.reserved_qps_cnt[MLX4_QP_REGION_FW];
1536 }
1537
1538 static int qp_alloc_res(struct mlx4_dev *dev, int slave, int op, int cmd,
1539                         u64 in_param, u64 *out_param)
1540 {
1541         int err;
1542         int count;
1543         int align;
1544         int base;
1545         int qpn;
1546
1547         switch (op) {
1548         case RES_OP_RESERVE:
1549                 count = get_param_l(&in_param) & 0xffffff;
1550                 align = get_param_h(&in_param);
1551                 err = mlx4_grant_resource(dev, slave, RES_QP, count, 0);
1552                 if (err)
1553                         return err;
1554
1555                 err = __mlx4_qp_reserve_range(dev, count, align, &base);
1556                 if (err) {
1557                         mlx4_release_resource(dev, slave, RES_QP, count, 0);
1558                         return err;
1559                 }
1560
1561                 err = add_res_range(dev, slave, base, count, RES_QP, 0);
1562                 if (err) {
1563                         mlx4_release_resource(dev, slave, RES_QP, count, 0);
1564                         __mlx4_qp_release_range(dev, base, count);
1565                         return err;
1566                 }
1567                 set_param_l(out_param, base);
1568                 break;
1569         case RES_OP_MAP_ICM:
1570                 qpn = get_param_l(&in_param) & 0x7fffff;
1571                 if (valid_reserved(dev, slave, qpn)) {
1572                         err = add_res_range(dev, slave, qpn, 1, RES_QP, 0);
1573                         if (err)
1574                                 return err;
1575                 }
1576
1577                 err = qp_res_start_move_to(dev, slave, qpn, RES_QP_MAPPED,
1578                                            NULL, 1);
1579                 if (err)
1580                         return err;
1581
1582                 if (!fw_reserved(dev, qpn)) {
1583                         err = __mlx4_qp_alloc_icm(dev, qpn, GFP_KERNEL);
1584                         if (err) {
1585                                 res_abort_move(dev, slave, RES_QP, qpn);
1586                                 return err;
1587                         }
1588                 }
1589
1590                 res_end_move(dev, slave, RES_QP, qpn);
1591                 break;
1592
1593         default:
1594                 err = -EINVAL;
1595                 break;
1596         }
1597         return err;
1598 }
1599
1600 static int mtt_alloc_res(struct mlx4_dev *dev, int slave, int op, int cmd,
1601                          u64 in_param, u64 *out_param)
1602 {
1603         int err = -EINVAL;
1604         int base;
1605         int order;
1606
1607         if (op != RES_OP_RESERVE_AND_MAP)
1608                 return err;
1609
1610         order = get_param_l(&in_param);
1611
1612         err = mlx4_grant_resource(dev, slave, RES_MTT, 1 << order, 0);
1613         if (err)
1614                 return err;
1615
1616         base = __mlx4_alloc_mtt_range(dev, order);
1617         if (base == -1) {
1618                 mlx4_release_resource(dev, slave, RES_MTT, 1 << order, 0);
1619                 return -ENOMEM;
1620         }
1621
1622         err = add_res_range(dev, slave, base, 1, RES_MTT, order);
1623         if (err) {
1624                 mlx4_release_resource(dev, slave, RES_MTT, 1 << order, 0);
1625                 __mlx4_free_mtt_range(dev, base, order);
1626         } else {
1627                 set_param_l(out_param, base);
1628         }
1629
1630         return err;
1631 }
1632
1633 static int mpt_alloc_res(struct mlx4_dev *dev, int slave, int op, int cmd,
1634                          u64 in_param, u64 *out_param)
1635 {
1636         int err = -EINVAL;
1637         int index;
1638         int id;
1639         struct res_mpt *mpt;
1640
1641         switch (op) {
1642         case RES_OP_RESERVE:
1643                 err = mlx4_grant_resource(dev, slave, RES_MPT, 1, 0);
1644                 if (err)
1645                         break;
1646
1647                 index = __mlx4_mpt_reserve(dev);
1648                 if (index == -1) {
1649                         mlx4_release_resource(dev, slave, RES_MPT, 1, 0);
1650                         break;
1651                 }
1652                 id = index & mpt_mask(dev);
1653
1654                 err = add_res_range(dev, slave, id, 1, RES_MPT, index);
1655                 if (err) {
1656                         mlx4_release_resource(dev, slave, RES_MPT, 1, 0);
1657                         __mlx4_mpt_release(dev, index);
1658                         break;
1659                 }
1660                 set_param_l(out_param, index);
1661                 break;
1662         case RES_OP_MAP_ICM:
1663                 index = get_param_l(&in_param);
1664                 id = index & mpt_mask(dev);
1665                 err = mr_res_start_move_to(dev, slave, id,
1666                                            RES_MPT_MAPPED, &mpt);
1667                 if (err)
1668                         return err;
1669
1670                 err = __mlx4_mpt_alloc_icm(dev, mpt->key, GFP_KERNEL);
1671                 if (err) {
1672                         res_abort_move(dev, slave, RES_MPT, id);
1673                         return err;
1674                 }
1675
1676                 res_end_move(dev, slave, RES_MPT, id);
1677                 break;
1678         }
1679         return err;
1680 }
1681
1682 static int cq_alloc_res(struct mlx4_dev *dev, int slave, int op, int cmd,
1683                         u64 in_param, u64 *out_param)
1684 {
1685         int cqn;
1686         int err;
1687
1688         switch (op) {
1689         case RES_OP_RESERVE_AND_MAP:
1690                 err = mlx4_grant_resource(dev, slave, RES_CQ, 1, 0);
1691                 if (err)
1692                         break;
1693
1694                 err = __mlx4_cq_alloc_icm(dev, &cqn);
1695                 if (err) {
1696                         mlx4_release_resource(dev, slave, RES_CQ, 1, 0);
1697                         break;
1698                 }
1699
1700                 err = add_res_range(dev, slave, cqn, 1, RES_CQ, 0);
1701                 if (err) {
1702                         mlx4_release_resource(dev, slave, RES_CQ, 1, 0);
1703                         __mlx4_cq_free_icm(dev, cqn);
1704                         break;
1705                 }
1706
1707                 set_param_l(out_param, cqn);
1708                 break;
1709
1710         default:
1711                 err = -EINVAL;
1712         }
1713
1714         return err;
1715 }
1716
1717 static int srq_alloc_res(struct mlx4_dev *dev, int slave, int op, int cmd,
1718                          u64 in_param, u64 *out_param)
1719 {
1720         int srqn;
1721         int err;
1722
1723         switch (op) {
1724         case RES_OP_RESERVE_AND_MAP:
1725                 err = mlx4_grant_resource(dev, slave, RES_SRQ, 1, 0);
1726                 if (err)
1727                         break;
1728
1729                 err = __mlx4_srq_alloc_icm(dev, &srqn);
1730                 if (err) {
1731                         mlx4_release_resource(dev, slave, RES_SRQ, 1, 0);
1732                         break;
1733                 }
1734
1735                 err = add_res_range(dev, slave, srqn, 1, RES_SRQ, 0);
1736                 if (err) {
1737                         mlx4_release_resource(dev, slave, RES_SRQ, 1, 0);
1738                         __mlx4_srq_free_icm(dev, srqn);
1739                         break;
1740                 }
1741
1742                 set_param_l(out_param, srqn);
1743                 break;
1744
1745         default:
1746                 err = -EINVAL;
1747         }
1748
1749         return err;
1750 }
1751
1752 static int mac_find_smac_ix_in_slave(struct mlx4_dev *dev, int slave, int port,
1753                                      u8 smac_index, u64 *mac)
1754 {
1755         struct mlx4_priv *priv = mlx4_priv(dev);
1756         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
1757         struct list_head *mac_list =
1758                 &tracker->slave_list[slave].res_list[RES_MAC];
1759         struct mac_res *res, *tmp;
1760
1761         list_for_each_entry_safe(res, tmp, mac_list, list) {
1762                 if (res->smac_index == smac_index && res->port == (u8) port) {
1763                         *mac = res->mac;
1764                         return 0;
1765                 }
1766         }
1767         return -ENOENT;
1768 }
1769
1770 static int mac_add_to_slave(struct mlx4_dev *dev, int slave, u64 mac, int port, u8 smac_index)
1771 {
1772         struct mlx4_priv *priv = mlx4_priv(dev);
1773         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
1774         struct list_head *mac_list =
1775                 &tracker->slave_list[slave].res_list[RES_MAC];
1776         struct mac_res *res, *tmp;
1777
1778         list_for_each_entry_safe(res, tmp, mac_list, list) {
1779                 if (res->mac == mac && res->port == (u8) port) {
1780                         /* mac found. update ref count */
1781                         ++res->ref_count;
1782                         return 0;
1783                 }
1784         }
1785
1786         if (mlx4_grant_resource(dev, slave, RES_MAC, 1, port))
1787                 return -EINVAL;
1788         res = kzalloc(sizeof *res, GFP_KERNEL);
1789         if (!res) {
1790                 mlx4_release_resource(dev, slave, RES_MAC, 1, port);
1791                 return -ENOMEM;
1792         }
1793         res->mac = mac;
1794         res->port = (u8) port;
1795         res->smac_index = smac_index;
1796         res->ref_count = 1;
1797         list_add_tail(&res->list,
1798                       &tracker->slave_list[slave].res_list[RES_MAC]);
1799         return 0;
1800 }
1801
1802 static void mac_del_from_slave(struct mlx4_dev *dev, int slave, u64 mac,
1803                                int port)
1804 {
1805         struct mlx4_priv *priv = mlx4_priv(dev);
1806         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
1807         struct list_head *mac_list =
1808                 &tracker->slave_list[slave].res_list[RES_MAC];
1809         struct mac_res *res, *tmp;
1810
1811         list_for_each_entry_safe(res, tmp, mac_list, list) {
1812                 if (res->mac == mac && res->port == (u8) port) {
1813                         if (!--res->ref_count) {
1814                                 list_del(&res->list);
1815                                 mlx4_release_resource(dev, slave, RES_MAC, 1, port);
1816                                 kfree(res);
1817                         }
1818                         break;
1819                 }
1820         }
1821 }
1822
1823 static void rem_slave_macs(struct mlx4_dev *dev, int slave)
1824 {
1825         struct mlx4_priv *priv = mlx4_priv(dev);
1826         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
1827         struct list_head *mac_list =
1828                 &tracker->slave_list[slave].res_list[RES_MAC];
1829         struct mac_res *res, *tmp;
1830         int i;
1831
1832         list_for_each_entry_safe(res, tmp, mac_list, list) {
1833                 list_del(&res->list);
1834                 /* dereference the mac the num times the slave referenced it */
1835                 for (i = 0; i < res->ref_count; i++)
1836                         __mlx4_unregister_mac(dev, res->port, res->mac);
1837                 mlx4_release_resource(dev, slave, RES_MAC, 1, res->port);
1838                 kfree(res);
1839         }
1840 }
1841
1842 static int mac_alloc_res(struct mlx4_dev *dev, int slave, int op, int cmd,
1843                          u64 in_param, u64 *out_param, int in_port)
1844 {
1845         int err = -EINVAL;
1846         int port;
1847         u64 mac;
1848         u8 smac_index;
1849
1850         if (op != RES_OP_RESERVE_AND_MAP)
1851                 return err;
1852
1853         port = !in_port ? get_param_l(out_param) : in_port;
1854         port = mlx4_slave_convert_port(
1855                         dev, slave, port);
1856
1857         if (port < 0)
1858                 return -EINVAL;
1859         mac = in_param;
1860
1861         err = __mlx4_register_mac(dev, port, mac);
1862         if (err >= 0) {
1863                 smac_index = err;
1864                 set_param_l(out_param, err);
1865                 err = 0;
1866         }
1867
1868         if (!err) {
1869                 err = mac_add_to_slave(dev, slave, mac, port, smac_index);
1870                 if (err)
1871                         __mlx4_unregister_mac(dev, port, mac);
1872         }
1873         return err;
1874 }
1875
1876 static int vlan_add_to_slave(struct mlx4_dev *dev, int slave, u16 vlan,
1877                              int port, int vlan_index)
1878 {
1879         struct mlx4_priv *priv = mlx4_priv(dev);
1880         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
1881         struct list_head *vlan_list =
1882                 &tracker->slave_list[slave].res_list[RES_VLAN];
1883         struct vlan_res *res, *tmp;
1884
1885         list_for_each_entry_safe(res, tmp, vlan_list, list) {
1886                 if (res->vlan == vlan && res->port == (u8) port) {
1887                         /* vlan found. update ref count */
1888                         ++res->ref_count;
1889                         return 0;
1890                 }
1891         }
1892
1893         if (mlx4_grant_resource(dev, slave, RES_VLAN, 1, port))
1894                 return -EINVAL;
1895         res = kzalloc(sizeof(*res), GFP_KERNEL);
1896         if (!res) {
1897                 mlx4_release_resource(dev, slave, RES_VLAN, 1, port);
1898                 return -ENOMEM;
1899         }
1900         res->vlan = vlan;
1901         res->port = (u8) port;
1902         res->vlan_index = vlan_index;
1903         res->ref_count = 1;
1904         list_add_tail(&res->list,
1905                       &tracker->slave_list[slave].res_list[RES_VLAN]);
1906         return 0;
1907 }
1908
1909
1910 static void vlan_del_from_slave(struct mlx4_dev *dev, int slave, u16 vlan,
1911                                 int port)
1912 {
1913         struct mlx4_priv *priv = mlx4_priv(dev);
1914         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
1915         struct list_head *vlan_list =
1916                 &tracker->slave_list[slave].res_list[RES_VLAN];
1917         struct vlan_res *res, *tmp;
1918
1919         list_for_each_entry_safe(res, tmp, vlan_list, list) {
1920                 if (res->vlan == vlan && res->port == (u8) port) {
1921                         if (!--res->ref_count) {
1922                                 list_del(&res->list);
1923                                 mlx4_release_resource(dev, slave, RES_VLAN,
1924                                                       1, port);
1925                                 kfree(res);
1926                         }
1927                         break;
1928                 }
1929         }
1930 }
1931
1932 static void rem_slave_vlans(struct mlx4_dev *dev, int slave)
1933 {
1934         struct mlx4_priv *priv = mlx4_priv(dev);
1935         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
1936         struct list_head *vlan_list =
1937                 &tracker->slave_list[slave].res_list[RES_VLAN];
1938         struct vlan_res *res, *tmp;
1939         int i;
1940
1941         list_for_each_entry_safe(res, tmp, vlan_list, list) {
1942                 list_del(&res->list);
1943                 /* dereference the vlan the num times the slave referenced it */
1944                 for (i = 0; i < res->ref_count; i++)
1945                         __mlx4_unregister_vlan(dev, res->port, res->vlan);
1946                 mlx4_release_resource(dev, slave, RES_VLAN, 1, res->port);
1947                 kfree(res);
1948         }
1949 }
1950
1951 static int vlan_alloc_res(struct mlx4_dev *dev, int slave, int op, int cmd,
1952                           u64 in_param, u64 *out_param, int in_port)
1953 {
1954         struct mlx4_priv *priv = mlx4_priv(dev);
1955         struct mlx4_slave_state *slave_state = priv->mfunc.master.slave_state;
1956         int err;
1957         u16 vlan;
1958         int vlan_index;
1959         int port;
1960
1961         port = !in_port ? get_param_l(out_param) : in_port;
1962
1963         if (!port || op != RES_OP_RESERVE_AND_MAP)
1964                 return -EINVAL;
1965
1966         port = mlx4_slave_convert_port(
1967                         dev, slave, port);
1968
1969         if (port < 0)
1970                 return -EINVAL;
1971         /* upstream kernels had NOP for reg/unreg vlan. Continue this. */
1972         if (!in_port && port > 0 && port <= dev->caps.num_ports) {
1973                 slave_state[slave].old_vlan_api = true;
1974                 return 0;
1975         }
1976
1977         vlan = (u16) in_param;
1978
1979         err = __mlx4_register_vlan(dev, port, vlan, &vlan_index);
1980         if (!err) {
1981                 set_param_l(out_param, (u32) vlan_index);
1982                 err = vlan_add_to_slave(dev, slave, vlan, port, vlan_index);
1983                 if (err)
1984                         __mlx4_unregister_vlan(dev, port, vlan);
1985         }
1986         return err;
1987 }
1988
1989 static int counter_alloc_res(struct mlx4_dev *dev, int slave, int op, int cmd,
1990                              u64 in_param, u64 *out_param)
1991 {
1992         u32 index;
1993         int err;
1994
1995         if (op != RES_OP_RESERVE)
1996                 return -EINVAL;
1997
1998         err = mlx4_grant_resource(dev, slave, RES_COUNTER, 1, 0);
1999         if (err)
2000                 return err;
2001
2002         err = __mlx4_counter_alloc(dev, &index);
2003         if (err) {
2004                 mlx4_release_resource(dev, slave, RES_COUNTER, 1, 0);
2005                 return err;
2006         }
2007
2008         err = add_res_range(dev, slave, index, 1, RES_COUNTER, 0);
2009         if (err) {
2010                 __mlx4_counter_free(dev, index);
2011                 mlx4_release_resource(dev, slave, RES_COUNTER, 1, 0);
2012         } else {
2013                 set_param_l(out_param, index);
2014         }
2015
2016         return err;
2017 }
2018
2019 static int xrcdn_alloc_res(struct mlx4_dev *dev, int slave, int op, int cmd,
2020                            u64 in_param, u64 *out_param)
2021 {
2022         u32 xrcdn;
2023         int err;
2024
2025         if (op != RES_OP_RESERVE)
2026                 return -EINVAL;
2027
2028         err = __mlx4_xrcd_alloc(dev, &xrcdn);
2029         if (err)
2030                 return err;
2031
2032         err = add_res_range(dev, slave, xrcdn, 1, RES_XRCD, 0);
2033         if (err)
2034                 __mlx4_xrcd_free(dev, xrcdn);
2035         else
2036                 set_param_l(out_param, xrcdn);
2037
2038         return err;
2039 }
2040
2041 int mlx4_ALLOC_RES_wrapper(struct mlx4_dev *dev, int slave,
2042                            struct mlx4_vhcr *vhcr,
2043                            struct mlx4_cmd_mailbox *inbox,
2044                            struct mlx4_cmd_mailbox *outbox,
2045                            struct mlx4_cmd_info *cmd)
2046 {
2047         int err;
2048         int alop = vhcr->op_modifier;
2049
2050         switch (vhcr->in_modifier & 0xFF) {
2051         case RES_QP:
2052                 err = qp_alloc_res(dev, slave, vhcr->op_modifier, alop,
2053                                    vhcr->in_param, &vhcr->out_param);
2054                 break;
2055
2056         case RES_MTT:
2057                 err = mtt_alloc_res(dev, slave, vhcr->op_modifier, alop,
2058                                     vhcr->in_param, &vhcr->out_param);
2059                 break;
2060
2061         case RES_MPT:
2062                 err = mpt_alloc_res(dev, slave, vhcr->op_modifier, alop,
2063                                     vhcr->in_param, &vhcr->out_param);
2064                 break;
2065
2066         case RES_CQ:
2067                 err = cq_alloc_res(dev, slave, vhcr->op_modifier, alop,
2068                                    vhcr->in_param, &vhcr->out_param);
2069                 break;
2070
2071         case RES_SRQ:
2072                 err = srq_alloc_res(dev, slave, vhcr->op_modifier, alop,
2073                                     vhcr->in_param, &vhcr->out_param);
2074                 break;
2075
2076         case RES_MAC:
2077                 err = mac_alloc_res(dev, slave, vhcr->op_modifier, alop,
2078                                     vhcr->in_param, &vhcr->out_param,
2079                                     (vhcr->in_modifier >> 8) & 0xFF);
2080                 break;
2081
2082         case RES_VLAN:
2083                 err = vlan_alloc_res(dev, slave, vhcr->op_modifier, alop,
2084                                      vhcr->in_param, &vhcr->out_param,
2085                                      (vhcr->in_modifier >> 8) & 0xFF);
2086                 break;
2087
2088         case RES_COUNTER:
2089                 err = counter_alloc_res(dev, slave, vhcr->op_modifier, alop,
2090                                         vhcr->in_param, &vhcr->out_param);
2091                 break;
2092
2093         case RES_XRCD:
2094                 err = xrcdn_alloc_res(dev, slave, vhcr->op_modifier, alop,
2095                                       vhcr->in_param, &vhcr->out_param);
2096                 break;
2097
2098         default:
2099                 err = -EINVAL;
2100                 break;
2101         }
2102
2103         return err;
2104 }
2105
2106 static int qp_free_res(struct mlx4_dev *dev, int slave, int op, int cmd,
2107                        u64 in_param)
2108 {
2109         int err;
2110         int count;
2111         int base;
2112         int qpn;
2113
2114         switch (op) {
2115         case RES_OP_RESERVE:
2116                 base = get_param_l(&in_param) & 0x7fffff;
2117                 count = get_param_h(&in_param);
2118                 err = rem_res_range(dev, slave, base, count, RES_QP, 0);
2119                 if (err)
2120                         break;
2121                 mlx4_release_resource(dev, slave, RES_QP, count, 0);
2122                 __mlx4_qp_release_range(dev, base, count);
2123                 break;
2124         case RES_OP_MAP_ICM:
2125                 qpn = get_param_l(&in_param) & 0x7fffff;
2126                 err = qp_res_start_move_to(dev, slave, qpn, RES_QP_RESERVED,
2127                                            NULL, 0);
2128                 if (err)
2129                         return err;
2130
2131                 if (!fw_reserved(dev, qpn))
2132                         __mlx4_qp_free_icm(dev, qpn);
2133
2134                 res_end_move(dev, slave, RES_QP, qpn);
2135
2136                 if (valid_reserved(dev, slave, qpn))
2137                         err = rem_res_range(dev, slave, qpn, 1, RES_QP, 0);
2138                 break;
2139         default:
2140                 err = -EINVAL;
2141                 break;
2142         }
2143         return err;
2144 }
2145
2146 static int mtt_free_res(struct mlx4_dev *dev, int slave, int op, int cmd,
2147                         u64 in_param, u64 *out_param)
2148 {
2149         int err = -EINVAL;
2150         int base;
2151         int order;
2152
2153         if (op != RES_OP_RESERVE_AND_MAP)
2154                 return err;
2155
2156         base = get_param_l(&in_param);
2157         order = get_param_h(&in_param);
2158         err = rem_res_range(dev, slave, base, 1, RES_MTT, order);
2159         if (!err) {
2160                 mlx4_release_resource(dev, slave, RES_MTT, 1 << order, 0);
2161                 __mlx4_free_mtt_range(dev, base, order);
2162         }
2163         return err;
2164 }
2165
2166 static int mpt_free_res(struct mlx4_dev *dev, int slave, int op, int cmd,
2167                         u64 in_param)
2168 {
2169         int err = -EINVAL;
2170         int index;
2171         int id;
2172         struct res_mpt *mpt;
2173
2174         switch (op) {
2175         case RES_OP_RESERVE:
2176                 index = get_param_l(&in_param);
2177                 id = index & mpt_mask(dev);
2178                 err = get_res(dev, slave, id, RES_MPT, &mpt);
2179                 if (err)
2180                         break;
2181                 index = mpt->key;
2182                 put_res(dev, slave, id, RES_MPT);
2183
2184                 err = rem_res_range(dev, slave, id, 1, RES_MPT, 0);
2185                 if (err)
2186                         break;
2187                 mlx4_release_resource(dev, slave, RES_MPT, 1, 0);
2188                 __mlx4_mpt_release(dev, index);
2189                 break;
2190         case RES_OP_MAP_ICM:
2191                         index = get_param_l(&in_param);
2192                         id = index & mpt_mask(dev);
2193                         err = mr_res_start_move_to(dev, slave, id,
2194                                                    RES_MPT_RESERVED, &mpt);
2195                         if (err)
2196                                 return err;
2197
2198                         __mlx4_mpt_free_icm(dev, mpt->key);
2199                         res_end_move(dev, slave, RES_MPT, id);
2200                         return err;
2201                 break;
2202         default:
2203                 err = -EINVAL;
2204                 break;
2205         }
2206         return err;
2207 }
2208
2209 static int cq_free_res(struct mlx4_dev *dev, int slave, int op, int cmd,
2210                        u64 in_param, u64 *out_param)
2211 {
2212         int cqn;
2213         int err;
2214
2215         switch (op) {
2216         case RES_OP_RESERVE_AND_MAP:
2217                 cqn = get_param_l(&in_param);
2218                 err = rem_res_range(dev, slave, cqn, 1, RES_CQ, 0);
2219                 if (err)
2220                         break;
2221
2222                 mlx4_release_resource(dev, slave, RES_CQ, 1, 0);
2223                 __mlx4_cq_free_icm(dev, cqn);
2224                 break;
2225
2226         default:
2227                 err = -EINVAL;
2228                 break;
2229         }
2230
2231         return err;
2232 }
2233
2234 static int srq_free_res(struct mlx4_dev *dev, int slave, int op, int cmd,
2235                         u64 in_param, u64 *out_param)
2236 {
2237         int srqn;
2238         int err;
2239
2240         switch (op) {
2241         case RES_OP_RESERVE_AND_MAP:
2242                 srqn = get_param_l(&in_param);
2243                 err = rem_res_range(dev, slave, srqn, 1, RES_SRQ, 0);
2244                 if (err)
2245                         break;
2246
2247                 mlx4_release_resource(dev, slave, RES_SRQ, 1, 0);
2248                 __mlx4_srq_free_icm(dev, srqn);
2249                 break;
2250
2251         default:
2252                 err = -EINVAL;
2253                 break;
2254         }
2255
2256         return err;
2257 }
2258
2259 static int mac_free_res(struct mlx4_dev *dev, int slave, int op, int cmd,
2260                             u64 in_param, u64 *out_param, int in_port)
2261 {
2262         int port;
2263         int err = 0;
2264
2265         switch (op) {
2266         case RES_OP_RESERVE_AND_MAP:
2267                 port = !in_port ? get_param_l(out_param) : in_port;
2268                 port = mlx4_slave_convert_port(
2269                                 dev, slave, port);
2270
2271                 if (port < 0)
2272                         return -EINVAL;
2273                 mac_del_from_slave(dev, slave, in_param, port);
2274                 __mlx4_unregister_mac(dev, port, in_param);
2275                 break;
2276         default:
2277                 err = -EINVAL;
2278                 break;
2279         }
2280
2281         return err;
2282
2283 }
2284
2285 static int vlan_free_res(struct mlx4_dev *dev, int slave, int op, int cmd,
2286                             u64 in_param, u64 *out_param, int port)
2287 {
2288         struct mlx4_priv *priv = mlx4_priv(dev);
2289         struct mlx4_slave_state *slave_state = priv->mfunc.master.slave_state;
2290         int err = 0;
2291
2292         port = mlx4_slave_convert_port(
2293                         dev, slave, port);
2294
2295         if (port < 0)
2296                 return -EINVAL;
2297         switch (op) {
2298         case RES_OP_RESERVE_AND_MAP:
2299                 if (slave_state[slave].old_vlan_api)
2300                         return 0;
2301                 if (!port)
2302                         return -EINVAL;
2303                 vlan_del_from_slave(dev, slave, in_param, port);
2304                 __mlx4_unregister_vlan(dev, port, in_param);
2305                 break;
2306         default:
2307                 err = -EINVAL;
2308                 break;
2309         }
2310
2311         return err;
2312 }
2313
2314 static int counter_free_res(struct mlx4_dev *dev, int slave, int op, int cmd,
2315                             u64 in_param, u64 *out_param)
2316 {
2317         int index;
2318         int err;
2319
2320         if (op != RES_OP_RESERVE)
2321                 return -EINVAL;
2322
2323         index = get_param_l(&in_param);
2324         err = rem_res_range(dev, slave, index, 1, RES_COUNTER, 0);
2325         if (err)
2326                 return err;
2327
2328         __mlx4_counter_free(dev, index);
2329         mlx4_release_resource(dev, slave, RES_COUNTER, 1, 0);
2330
2331         return err;
2332 }
2333
2334 static int xrcdn_free_res(struct mlx4_dev *dev, int slave, int op, int cmd,
2335                           u64 in_param, u64 *out_param)
2336 {
2337         int xrcdn;
2338         int err;
2339
2340         if (op != RES_OP_RESERVE)
2341                 return -EINVAL;
2342
2343         xrcdn = get_param_l(&in_param);
2344         err = rem_res_range(dev, slave, xrcdn, 1, RES_XRCD, 0);
2345         if (err)
2346                 return err;
2347
2348         __mlx4_xrcd_free(dev, xrcdn);
2349
2350         return err;
2351 }
2352
2353 int mlx4_FREE_RES_wrapper(struct mlx4_dev *dev, int slave,
2354                           struct mlx4_vhcr *vhcr,
2355                           struct mlx4_cmd_mailbox *inbox,
2356                           struct mlx4_cmd_mailbox *outbox,
2357                           struct mlx4_cmd_info *cmd)
2358 {
2359         int err = -EINVAL;
2360         int alop = vhcr->op_modifier;
2361
2362         switch (vhcr->in_modifier & 0xFF) {
2363         case RES_QP:
2364                 err = qp_free_res(dev, slave, vhcr->op_modifier, alop,
2365                                   vhcr->in_param);
2366                 break;
2367
2368         case RES_MTT:
2369                 err = mtt_free_res(dev, slave, vhcr->op_modifier, alop,
2370                                    vhcr->in_param, &vhcr->out_param);
2371                 break;
2372
2373         case RES_MPT:
2374                 err = mpt_free_res(dev, slave, vhcr->op_modifier, alop,
2375                                    vhcr->in_param);
2376                 break;
2377
2378         case RES_CQ:
2379                 err = cq_free_res(dev, slave, vhcr->op_modifier, alop,
2380                                   vhcr->in_param, &vhcr->out_param);
2381                 break;
2382
2383         case RES_SRQ:
2384                 err = srq_free_res(dev, slave, vhcr->op_modifier, alop,
2385                                    vhcr->in_param, &vhcr->out_param);
2386                 break;
2387
2388         case RES_MAC:
2389                 err = mac_free_res(dev, slave, vhcr->op_modifier, alop,
2390                                    vhcr->in_param, &vhcr->out_param,
2391                                    (vhcr->in_modifier >> 8) & 0xFF);
2392                 break;
2393
2394         case RES_VLAN:
2395                 err = vlan_free_res(dev, slave, vhcr->op_modifier, alop,
2396                                     vhcr->in_param, &vhcr->out_param,
2397                                     (vhcr->in_modifier >> 8) & 0xFF);
2398                 break;
2399
2400         case RES_COUNTER:
2401                 err = counter_free_res(dev, slave, vhcr->op_modifier, alop,
2402                                        vhcr->in_param, &vhcr->out_param);
2403                 break;
2404
2405         case RES_XRCD:
2406                 err = xrcdn_free_res(dev, slave, vhcr->op_modifier, alop,
2407                                      vhcr->in_param, &vhcr->out_param);
2408
2409         default:
2410                 break;
2411         }
2412         return err;
2413 }
2414
2415 /* ugly but other choices are uglier */
2416 static int mr_phys_mpt(struct mlx4_mpt_entry *mpt)
2417 {
2418         return (be32_to_cpu(mpt->flags) >> 9) & 1;
2419 }
2420
2421 static int mr_get_mtt_addr(struct mlx4_mpt_entry *mpt)
2422 {
2423         return (int)be64_to_cpu(mpt->mtt_addr) & 0xfffffff8;
2424 }
2425
2426 static int mr_get_mtt_size(struct mlx4_mpt_entry *mpt)
2427 {
2428         return be32_to_cpu(mpt->mtt_sz);
2429 }
2430
2431 static u32 mr_get_pd(struct mlx4_mpt_entry *mpt)
2432 {
2433         return be32_to_cpu(mpt->pd_flags) & 0x00ffffff;
2434 }
2435
2436 static int mr_is_fmr(struct mlx4_mpt_entry *mpt)
2437 {
2438         return be32_to_cpu(mpt->pd_flags) & MLX4_MPT_PD_FLAG_FAST_REG;
2439 }
2440
2441 static int mr_is_bind_enabled(struct mlx4_mpt_entry *mpt)
2442 {
2443         return be32_to_cpu(mpt->flags) & MLX4_MPT_FLAG_BIND_ENABLE;
2444 }
2445
2446 static int mr_is_region(struct mlx4_mpt_entry *mpt)
2447 {
2448         return be32_to_cpu(mpt->flags) & MLX4_MPT_FLAG_REGION;
2449 }
2450
2451 static int qp_get_mtt_addr(struct mlx4_qp_context *qpc)
2452 {
2453         return be32_to_cpu(qpc->mtt_base_addr_l) & 0xfffffff8;
2454 }
2455
2456 static int srq_get_mtt_addr(struct mlx4_srq_context *srqc)
2457 {
2458         return be32_to_cpu(srqc->mtt_base_addr_l) & 0xfffffff8;
2459 }
2460
2461 static int qp_get_mtt_size(struct mlx4_qp_context *qpc)
2462 {
2463         int page_shift = (qpc->log_page_size & 0x3f) + 12;
2464         int log_sq_size = (qpc->sq_size_stride >> 3) & 0xf;
2465         int log_sq_sride = qpc->sq_size_stride & 7;
2466         int log_rq_size = (qpc->rq_size_stride >> 3) & 0xf;
2467         int log_rq_stride = qpc->rq_size_stride & 7;
2468         int srq = (be32_to_cpu(qpc->srqn) >> 24) & 1;
2469         int rss = (be32_to_cpu(qpc->flags) >> 13) & 1;
2470         u32 ts = (be32_to_cpu(qpc->flags) >> 16) & 0xff;
2471         int xrc = (ts == MLX4_QP_ST_XRC) ? 1 : 0;
2472         int sq_size;
2473         int rq_size;
2474         int total_pages;
2475         int total_mem;
2476         int page_offset = (be32_to_cpu(qpc->params2) >> 6) & 0x3f;
2477
2478         sq_size = 1 << (log_sq_size + log_sq_sride + 4);
2479         rq_size = (srq|rss|xrc) ? 0 : (1 << (log_rq_size + log_rq_stride + 4));
2480         total_mem = sq_size + rq_size;
2481         total_pages =
2482                 roundup_pow_of_two((total_mem + (page_offset << 6)) >>
2483                                    page_shift);
2484
2485         return total_pages;
2486 }
2487
2488 static int check_mtt_range(struct mlx4_dev *dev, int slave, int start,
2489                            int size, struct res_mtt *mtt)
2490 {
2491         int res_start = mtt->com.res_id;
2492         int res_size = (1 << mtt->order);
2493
2494         if (start < res_start || start + size > res_start + res_size)
2495                 return -EPERM;
2496         return 0;
2497 }
2498
2499 int mlx4_SW2HW_MPT_wrapper(struct mlx4_dev *dev, int slave,
2500                            struct mlx4_vhcr *vhcr,
2501                            struct mlx4_cmd_mailbox *inbox,
2502                            struct mlx4_cmd_mailbox *outbox,
2503                            struct mlx4_cmd_info *cmd)
2504 {
2505         int err;
2506         int index = vhcr->in_modifier;
2507         struct res_mtt *mtt;
2508         struct res_mpt *mpt;
2509         int mtt_base = mr_get_mtt_addr(inbox->buf) / dev->caps.mtt_entry_sz;
2510         int phys;
2511         int id;
2512         u32 pd;
2513         int pd_slave;
2514
2515         id = index & mpt_mask(dev);
2516         err = mr_res_start_move_to(dev, slave, id, RES_MPT_HW, &mpt);
2517         if (err)
2518                 return err;
2519
2520         /* Disable memory windows for VFs. */
2521         if (!mr_is_region(inbox->buf)) {
2522                 err = -EPERM;
2523                 goto ex_abort;
2524         }
2525
2526         /* Make sure that the PD bits related to the slave id are zeros. */
2527         pd = mr_get_pd(inbox->buf);
2528         pd_slave = (pd >> 17) & 0x7f;
2529         if (pd_slave != 0 && pd_slave != slave) {
2530                 err = -EPERM;
2531                 goto ex_abort;
2532         }
2533
2534         if (mr_is_fmr(inbox->buf)) {
2535                 /* FMR and Bind Enable are forbidden in slave devices. */
2536                 if (mr_is_bind_enabled(inbox->buf)) {
2537                         err = -EPERM;
2538                         goto ex_abort;
2539                 }
2540                 /* FMR and Memory Windows are also forbidden. */
2541                 if (!mr_is_region(inbox->buf)) {
2542                         err = -EPERM;
2543                         goto ex_abort;
2544                 }
2545         }
2546
2547         phys = mr_phys_mpt(inbox->buf);
2548         if (!phys) {
2549                 err = get_res(dev, slave, mtt_base, RES_MTT, &mtt);
2550                 if (err)
2551                         goto ex_abort;
2552
2553                 err = check_mtt_range(dev, slave, mtt_base,
2554                                       mr_get_mtt_size(inbox->buf), mtt);
2555                 if (err)
2556                         goto ex_put;
2557
2558                 mpt->mtt = mtt;
2559         }
2560
2561         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
2562         if (err)
2563                 goto ex_put;
2564
2565         if (!phys) {
2566                 atomic_inc(&mtt->ref_count);
2567                 put_res(dev, slave, mtt->com.res_id, RES_MTT);
2568         }
2569
2570         res_end_move(dev, slave, RES_MPT, id);
2571         return 0;
2572
2573 ex_put:
2574         if (!phys)
2575                 put_res(dev, slave, mtt->com.res_id, RES_MTT);
2576 ex_abort:
2577         res_abort_move(dev, slave, RES_MPT, id);
2578
2579         return err;
2580 }
2581
2582 int mlx4_HW2SW_MPT_wrapper(struct mlx4_dev *dev, int slave,
2583                            struct mlx4_vhcr *vhcr,
2584                            struct mlx4_cmd_mailbox *inbox,
2585                            struct mlx4_cmd_mailbox *outbox,
2586                            struct mlx4_cmd_info *cmd)
2587 {
2588         int err;
2589         int index = vhcr->in_modifier;
2590         struct res_mpt *mpt;
2591         int id;
2592
2593         id = index & mpt_mask(dev);
2594         err = mr_res_start_move_to(dev, slave, id, RES_MPT_MAPPED, &mpt);
2595         if (err)
2596                 return err;
2597
2598         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
2599         if (err)
2600                 goto ex_abort;
2601
2602         if (mpt->mtt)
2603                 atomic_dec(&mpt->mtt->ref_count);
2604
2605         res_end_move(dev, slave, RES_MPT, id);
2606         return 0;
2607
2608 ex_abort:
2609         res_abort_move(dev, slave, RES_MPT, id);
2610
2611         return err;
2612 }
2613
2614 int mlx4_QUERY_MPT_wrapper(struct mlx4_dev *dev, int slave,
2615                            struct mlx4_vhcr *vhcr,
2616                            struct mlx4_cmd_mailbox *inbox,
2617                            struct mlx4_cmd_mailbox *outbox,
2618                            struct mlx4_cmd_info *cmd)
2619 {
2620         int err;
2621         int index = vhcr->in_modifier;
2622         struct res_mpt *mpt;
2623         int id;
2624
2625         id = index & mpt_mask(dev);
2626         err = get_res(dev, slave, id, RES_MPT, &mpt);
2627         if (err)
2628                 return err;
2629
2630         if (mpt->com.from_state == RES_MPT_MAPPED) {
2631                 /* In order to allow rereg in SRIOV, we need to alter the MPT entry. To do
2632                  * that, the VF must read the MPT. But since the MPT entry memory is not
2633                  * in the VF's virtual memory space, it must use QUERY_MPT to obtain the
2634                  * entry contents. To guarantee that the MPT cannot be changed, the driver
2635                  * must perform HW2SW_MPT before this query and return the MPT entry to HW
2636                  * ownership fofollowing the change. The change here allows the VF to
2637                  * perform QUERY_MPT also when the entry is in SW ownership.
2638                  */
2639                 struct mlx4_mpt_entry *mpt_entry = mlx4_table_find(
2640                                         &mlx4_priv(dev)->mr_table.dmpt_table,
2641                                         mpt->key, NULL);
2642
2643                 if (NULL == mpt_entry || NULL == outbox->buf) {
2644                         err = -EINVAL;
2645                         goto out;
2646                 }
2647
2648                 memcpy(outbox->buf, mpt_entry, sizeof(*mpt_entry));
2649
2650                 err = 0;
2651         } else if (mpt->com.from_state == RES_MPT_HW) {
2652                 err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
2653         } else {
2654                 err = -EBUSY;
2655                 goto out;
2656         }
2657
2658
2659 out:
2660         put_res(dev, slave, id, RES_MPT);
2661         return err;
2662 }
2663
2664 static int qp_get_rcqn(struct mlx4_qp_context *qpc)
2665 {
2666         return be32_to_cpu(qpc->cqn_recv) & 0xffffff;
2667 }
2668
2669 static int qp_get_scqn(struct mlx4_qp_context *qpc)
2670 {
2671         return be32_to_cpu(qpc->cqn_send) & 0xffffff;
2672 }
2673
2674 static u32 qp_get_srqn(struct mlx4_qp_context *qpc)
2675 {
2676         return be32_to_cpu(qpc->srqn) & 0x1ffffff;
2677 }
2678
2679 static void adjust_proxy_tun_qkey(struct mlx4_dev *dev, struct mlx4_vhcr *vhcr,
2680                                   struct mlx4_qp_context *context)
2681 {
2682         u32 qpn = vhcr->in_modifier & 0xffffff;
2683         u32 qkey = 0;
2684
2685         if (mlx4_get_parav_qkey(dev, qpn, &qkey))
2686                 return;
2687
2688         /* adjust qkey in qp context */
2689         context->qkey = cpu_to_be32(qkey);
2690 }
2691
2692 int mlx4_RST2INIT_QP_wrapper(struct mlx4_dev *dev, int slave,
2693                              struct mlx4_vhcr *vhcr,
2694                              struct mlx4_cmd_mailbox *inbox,
2695                              struct mlx4_cmd_mailbox *outbox,
2696                              struct mlx4_cmd_info *cmd)
2697 {
2698         int err;
2699         int qpn = vhcr->in_modifier & 0x7fffff;
2700         struct res_mtt *mtt;
2701         struct res_qp *qp;
2702         struct mlx4_qp_context *qpc = inbox->buf + 8;
2703         int mtt_base = qp_get_mtt_addr(qpc) / dev->caps.mtt_entry_sz;
2704         int mtt_size = qp_get_mtt_size(qpc);
2705         struct res_cq *rcq;
2706         struct res_cq *scq;
2707         int rcqn = qp_get_rcqn(qpc);
2708         int scqn = qp_get_scqn(qpc);
2709         u32 srqn = qp_get_srqn(qpc) & 0xffffff;
2710         int use_srq = (qp_get_srqn(qpc) >> 24) & 1;
2711         struct res_srq *srq;
2712         int local_qpn = be32_to_cpu(qpc->local_qpn) & 0xffffff;
2713
2714         err = qp_res_start_move_to(dev, slave, qpn, RES_QP_HW, &qp, 0);
2715         if (err)
2716                 return err;
2717         qp->local_qpn = local_qpn;
2718         qp->sched_queue = 0;
2719         qp->param3 = 0;
2720         qp->vlan_control = 0;
2721         qp->fvl_rx = 0;
2722         qp->pri_path_fl = 0;
2723         qp->vlan_index = 0;
2724         qp->feup = 0;
2725         qp->qpc_flags = be32_to_cpu(qpc->flags);
2726
2727         err = get_res(dev, slave, mtt_base, RES_MTT, &mtt);
2728         if (err)
2729                 goto ex_abort;
2730
2731         err = check_mtt_range(dev, slave, mtt_base, mtt_size, mtt);
2732         if (err)
2733                 goto ex_put_mtt;
2734
2735         err = get_res(dev, slave, rcqn, RES_CQ, &rcq);
2736         if (err)
2737                 goto ex_put_mtt;
2738
2739         if (scqn != rcqn) {
2740                 err = get_res(dev, slave, scqn, RES_CQ, &scq);
2741                 if (err)
2742                         goto ex_put_rcq;
2743         } else
2744                 scq = rcq;
2745
2746         if (use_srq) {
2747                 err = get_res(dev, slave, srqn, RES_SRQ, &srq);
2748                 if (err)
2749                         goto ex_put_scq;
2750         }
2751
2752         adjust_proxy_tun_qkey(dev, vhcr, qpc);
2753         update_pkey_index(dev, slave, inbox);
2754         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
2755         if (err)
2756                 goto ex_put_srq;
2757         atomic_inc(&mtt->ref_count);
2758         qp->mtt = mtt;
2759         atomic_inc(&rcq->ref_count);
2760         qp->rcq = rcq;
2761         atomic_inc(&scq->ref_count);
2762         qp->scq = scq;
2763
2764         if (scqn != rcqn)
2765                 put_res(dev, slave, scqn, RES_CQ);
2766
2767         if (use_srq) {
2768                 atomic_inc(&srq->ref_count);
2769                 put_res(dev, slave, srqn, RES_SRQ);
2770                 qp->srq = srq;
2771         }
2772         put_res(dev, slave, rcqn, RES_CQ);
2773         put_res(dev, slave, mtt_base, RES_MTT);
2774         res_end_move(dev, slave, RES_QP, qpn);
2775
2776         return 0;
2777
2778 ex_put_srq:
2779         if (use_srq)
2780                 put_res(dev, slave, srqn, RES_SRQ);
2781 ex_put_scq:
2782         if (scqn != rcqn)
2783                 put_res(dev, slave, scqn, RES_CQ);
2784 ex_put_rcq:
2785         put_res(dev, slave, rcqn, RES_CQ);
2786 ex_put_mtt:
2787         put_res(dev, slave, mtt_base, RES_MTT);
2788 ex_abort:
2789         res_abort_move(dev, slave, RES_QP, qpn);
2790
2791         return err;
2792 }
2793
2794 static int eq_get_mtt_addr(struct mlx4_eq_context *eqc)
2795 {
2796         return be32_to_cpu(eqc->mtt_base_addr_l) & 0xfffffff8;
2797 }
2798
2799 static int eq_get_mtt_size(struct mlx4_eq_context *eqc)
2800 {
2801         int log_eq_size = eqc->log_eq_size & 0x1f;
2802         int page_shift = (eqc->log_page_size & 0x3f) + 12;
2803
2804         if (log_eq_size + 5 < page_shift)
2805                 return 1;
2806
2807         return 1 << (log_eq_size + 5 - page_shift);
2808 }
2809
2810 static int cq_get_mtt_addr(struct mlx4_cq_context *cqc)
2811 {
2812         return be32_to_cpu(cqc->mtt_base_addr_l) & 0xfffffff8;
2813 }
2814
2815 static int cq_get_mtt_size(struct mlx4_cq_context *cqc)
2816 {
2817         int log_cq_size = (be32_to_cpu(cqc->logsize_usrpage) >> 24) & 0x1f;
2818         int page_shift = (cqc->log_page_size & 0x3f) + 12;
2819
2820         if (log_cq_size + 5 < page_shift)
2821                 return 1;
2822
2823         return 1 << (log_cq_size + 5 - page_shift);
2824 }
2825
2826 int mlx4_SW2HW_EQ_wrapper(struct mlx4_dev *dev, int slave,
2827                           struct mlx4_vhcr *vhcr,
2828                           struct mlx4_cmd_mailbox *inbox,
2829                           struct mlx4_cmd_mailbox *outbox,
2830                           struct mlx4_cmd_info *cmd)
2831 {
2832         int err;
2833         int eqn = vhcr->in_modifier;
2834         int res_id = (slave << 8) | eqn;
2835         struct mlx4_eq_context *eqc = inbox->buf;
2836         int mtt_base = eq_get_mtt_addr(eqc) / dev->caps.mtt_entry_sz;
2837         int mtt_size = eq_get_mtt_size(eqc);
2838         struct res_eq *eq;
2839         struct res_mtt *mtt;
2840
2841         err = add_res_range(dev, slave, res_id, 1, RES_EQ, 0);
2842         if (err)
2843                 return err;
2844         err = eq_res_start_move_to(dev, slave, res_id, RES_EQ_HW, &eq);
2845         if (err)
2846                 goto out_add;
2847
2848         err = get_res(dev, slave, mtt_base, RES_MTT, &mtt);
2849         if (err)
2850                 goto out_move;
2851
2852         err = check_mtt_range(dev, slave, mtt_base, mtt_size, mtt);
2853         if (err)
2854                 goto out_put;
2855
2856         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
2857         if (err)
2858                 goto out_put;
2859
2860         atomic_inc(&mtt->ref_count);
2861         eq->mtt = mtt;
2862         put_res(dev, slave, mtt->com.res_id, RES_MTT);
2863         res_end_move(dev, slave, RES_EQ, res_id);
2864         return 0;
2865
2866 out_put:
2867         put_res(dev, slave, mtt->com.res_id, RES_MTT);
2868 out_move:
2869         res_abort_move(dev, slave, RES_EQ, res_id);
2870 out_add:
2871         rem_res_range(dev, slave, res_id, 1, RES_EQ, 0);
2872         return err;
2873 }
2874
2875 int mlx4_CONFIG_DEV_wrapper(struct mlx4_dev *dev, int slave,
2876                             struct mlx4_vhcr *vhcr,
2877                             struct mlx4_cmd_mailbox *inbox,
2878                             struct mlx4_cmd_mailbox *outbox,
2879                             struct mlx4_cmd_info *cmd)
2880 {
2881         int err;
2882         u8 get = vhcr->op_modifier;
2883
2884         if (get != 1)
2885                 return -EPERM;
2886
2887         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
2888
2889         return err;
2890 }
2891
2892 static int get_containing_mtt(struct mlx4_dev *dev, int slave, int start,
2893                               int len, struct res_mtt **res)
2894 {
2895         struct mlx4_priv *priv = mlx4_priv(dev);
2896         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
2897         struct res_mtt *mtt;
2898         int err = -EINVAL;
2899
2900         spin_lock_irq(mlx4_tlock(dev));
2901         list_for_each_entry(mtt, &tracker->slave_list[slave].res_list[RES_MTT],
2902                             com.list) {
2903                 if (!check_mtt_range(dev, slave, start, len, mtt)) {
2904                         *res = mtt;
2905                         mtt->com.from_state = mtt->com.state;
2906                         mtt->com.state = RES_MTT_BUSY;
2907                         err = 0;
2908                         break;
2909                 }
2910         }
2911         spin_unlock_irq(mlx4_tlock(dev));
2912
2913         return err;
2914 }
2915
2916 static int verify_qp_parameters(struct mlx4_dev *dev,
2917                                 struct mlx4_vhcr *vhcr,
2918                                 struct mlx4_cmd_mailbox *inbox,
2919                                 enum qp_transition transition, u8 slave)
2920 {
2921         u32                     qp_type;
2922         u32                     qpn;
2923         struct mlx4_qp_context  *qp_ctx;
2924         enum mlx4_qp_optpar     optpar;
2925         int port;
2926         int num_gids;
2927
2928         qp_ctx  = inbox->buf + 8;
2929         qp_type = (be32_to_cpu(qp_ctx->flags) >> 16) & 0xff;
2930         optpar  = be32_to_cpu(*(__be32 *) inbox->buf);
2931
2932         switch (qp_type) {
2933         case MLX4_QP_ST_RC:
2934         case MLX4_QP_ST_XRC:
2935         case MLX4_QP_ST_UC:
2936                 switch (transition) {
2937                 case QP_TRANS_INIT2RTR:
2938                 case QP_TRANS_RTR2RTS:
2939                 case QP_TRANS_RTS2RTS:
2940                 case QP_TRANS_SQD2SQD:
2941                 case QP_TRANS_SQD2RTS:
2942                         if (slave != mlx4_master_func_num(dev))
2943                                 if (optpar & MLX4_QP_OPTPAR_PRIMARY_ADDR_PATH) {
2944                                         port = (qp_ctx->pri_path.sched_queue >> 6 & 1) + 1;
2945                                         if (dev->caps.port_mask[port] != MLX4_PORT_TYPE_IB)
2946                                                 num_gids = mlx4_get_slave_num_gids(dev, slave, port);
2947                                         else
2948                                                 num_gids = 1;
2949                                         if (qp_ctx->pri_path.mgid_index >= num_gids)
2950                                                 return -EINVAL;
2951                                 }
2952                                 if (optpar & MLX4_QP_OPTPAR_ALT_ADDR_PATH) {
2953                                         port = (qp_ctx->alt_path.sched_queue >> 6 & 1) + 1;
2954                                         if (dev->caps.port_mask[port] != MLX4_PORT_TYPE_IB)
2955                                                 num_gids = mlx4_get_slave_num_gids(dev, slave, port);
2956                                         else
2957                                                 num_gids = 1;
2958                                         if (qp_ctx->alt_path.mgid_index >= num_gids)
2959                                                 return -EINVAL;
2960                                 }
2961                         break;
2962                 default:
2963                         break;
2964                 }
2965                 break;
2966
2967         case MLX4_QP_ST_MLX:
2968                 qpn = vhcr->in_modifier & 0x7fffff;
2969                 port = (qp_ctx->pri_path.sched_queue >> 6 & 1) + 1;
2970                 if (transition == QP_TRANS_INIT2RTR &&
2971                     slave != mlx4_master_func_num(dev) &&
2972                     mlx4_is_qp_reserved(dev, qpn) &&
2973                     !mlx4_vf_smi_enabled(dev, slave, port)) {
2974                         /* only enabled VFs may create MLX proxy QPs */
2975                         mlx4_err(dev, "%s: unprivileged slave %d attempting to create an MLX proxy special QP on port %d\n",
2976                                  __func__, slave, port);
2977                         return -EPERM;
2978                 }
2979                 break;
2980
2981         default:
2982                 break;
2983         }
2984
2985         return 0;
2986 }
2987
2988 int mlx4_WRITE_MTT_wrapper(struct mlx4_dev *dev, int slave,
2989                            struct mlx4_vhcr *vhcr,
2990                            struct mlx4_cmd_mailbox *inbox,
2991                            struct mlx4_cmd_mailbox *outbox,
2992                            struct mlx4_cmd_info *cmd)
2993 {
2994         struct mlx4_mtt mtt;
2995         __be64 *page_list = inbox->buf;
2996         u64 *pg_list = (u64 *)page_list;
2997         int i;
2998         struct res_mtt *rmtt = NULL;
2999         int start = be64_to_cpu(page_list[0]);
3000         int npages = vhcr->in_modifier;
3001         int err;
3002
3003         err = get_containing_mtt(dev, slave, start, npages, &rmtt);
3004         if (err)
3005                 return err;
3006
3007         /* Call the SW implementation of write_mtt:
3008          * - Prepare a dummy mtt struct
3009          * - Translate inbox contents to simple addresses in host endianess */
3010         mtt.offset = 0;  /* TBD this is broken but I don't handle it since
3011                             we don't really use it */
3012         mtt.order = 0;
3013         mtt.page_shift = 0;
3014         for (i = 0; i < npages; ++i)
3015                 pg_list[i + 2] = (be64_to_cpu(page_list[i + 2]) & ~1ULL);
3016
3017         err = __mlx4_write_mtt(dev, &mtt, be64_to_cpu(page_list[0]), npages,
3018                                ((u64 *)page_list + 2));
3019
3020         if (rmtt)
3021                 put_res(dev, slave, rmtt->com.res_id, RES_MTT);
3022
3023         return err;
3024 }
3025
3026 int mlx4_HW2SW_EQ_wrapper(struct mlx4_dev *dev, int slave,
3027                           struct mlx4_vhcr *vhcr,
3028                           struct mlx4_cmd_mailbox *inbox,
3029                           struct mlx4_cmd_mailbox *outbox,
3030                           struct mlx4_cmd_info *cmd)
3031 {
3032         int eqn = vhcr->in_modifier;
3033         int res_id = eqn | (slave << 8);
3034         struct res_eq *eq;
3035         int err;
3036
3037         err = eq_res_start_move_to(dev, slave, res_id, RES_EQ_RESERVED, &eq);
3038         if (err)
3039                 return err;
3040
3041         err = get_res(dev, slave, eq->mtt->com.res_id, RES_MTT, NULL);
3042         if (err)
3043                 goto ex_abort;
3044
3045         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3046         if (err)
3047                 goto ex_put;
3048
3049         atomic_dec(&eq->mtt->ref_count);
3050         put_res(dev, slave, eq->mtt->com.res_id, RES_MTT);
3051         res_end_move(dev, slave, RES_EQ, res_id);
3052         rem_res_range(dev, slave, res_id, 1, RES_EQ, 0);
3053
3054         return 0;
3055
3056 ex_put:
3057         put_res(dev, slave, eq->mtt->com.res_id, RES_MTT);
3058 ex_abort:
3059         res_abort_move(dev, slave, RES_EQ, res_id);
3060
3061         return err;
3062 }
3063
3064 int mlx4_GEN_EQE(struct mlx4_dev *dev, int slave, struct mlx4_eqe *eqe)
3065 {
3066         struct mlx4_priv *priv = mlx4_priv(dev);
3067         struct mlx4_slave_event_eq_info *event_eq;
3068         struct mlx4_cmd_mailbox *mailbox;
3069         u32 in_modifier = 0;
3070         int err;
3071         int res_id;
3072         struct res_eq *req;
3073
3074         if (!priv->mfunc.master.slave_state)
3075                 return -EINVAL;
3076
3077         event_eq = &priv->mfunc.master.slave_state[slave].event_eq[eqe->type];
3078
3079         /* Create the event only if the slave is registered */
3080         if (event_eq->eqn < 0)
3081                 return 0;
3082
3083         mutex_lock(&priv->mfunc.master.gen_eqe_mutex[slave]);
3084         res_id = (slave << 8) | event_eq->eqn;
3085         err = get_res(dev, slave, res_id, RES_EQ, &req);
3086         if (err)
3087                 goto unlock;
3088
3089         if (req->com.from_state != RES_EQ_HW) {
3090                 err = -EINVAL;
3091                 goto put;
3092         }
3093
3094         mailbox = mlx4_alloc_cmd_mailbox(dev);
3095         if (IS_ERR(mailbox)) {
3096                 err = PTR_ERR(mailbox);
3097                 goto put;
3098         }
3099
3100         if (eqe->type == MLX4_EVENT_TYPE_CMD) {
3101                 ++event_eq->token;
3102                 eqe->event.cmd.token = cpu_to_be16(event_eq->token);
3103         }
3104
3105         memcpy(mailbox->buf, (u8 *) eqe, 28);
3106
3107         in_modifier = (slave & 0xff) | ((event_eq->eqn & 0xff) << 16);
3108
3109         err = mlx4_cmd(dev, mailbox->dma, in_modifier, 0,
3110                        MLX4_CMD_GEN_EQE, MLX4_CMD_TIME_CLASS_B,
3111                        MLX4_CMD_NATIVE);
3112
3113         put_res(dev, slave, res_id, RES_EQ);
3114         mutex_unlock(&priv->mfunc.master.gen_eqe_mutex[slave]);
3115         mlx4_free_cmd_mailbox(dev, mailbox);
3116         return err;
3117
3118 put:
3119         put_res(dev, slave, res_id, RES_EQ);
3120
3121 unlock:
3122         mutex_unlock(&priv->mfunc.master.gen_eqe_mutex[slave]);
3123         return err;
3124 }
3125
3126 int mlx4_QUERY_EQ_wrapper(struct mlx4_dev *dev, int slave,
3127                           struct mlx4_vhcr *vhcr,
3128                           struct mlx4_cmd_mailbox *inbox,
3129                           struct mlx4_cmd_mailbox *outbox,
3130                           struct mlx4_cmd_info *cmd)
3131 {
3132         int eqn = vhcr->in_modifier;
3133         int res_id = eqn | (slave << 8);
3134         struct res_eq *eq;
3135         int err;
3136
3137         err = get_res(dev, slave, res_id, RES_EQ, &eq);
3138         if (err)
3139                 return err;
3140
3141         if (eq->com.from_state != RES_EQ_HW) {
3142                 err = -EINVAL;
3143                 goto ex_put;
3144         }
3145
3146         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3147
3148 ex_put:
3149         put_res(dev, slave, res_id, RES_EQ);
3150         return err;
3151 }
3152
3153 int mlx4_SW2HW_CQ_wrapper(struct mlx4_dev *dev, int slave,
3154                           struct mlx4_vhcr *vhcr,
3155                           struct mlx4_cmd_mailbox *inbox,
3156                           struct mlx4_cmd_mailbox *outbox,
3157                           struct mlx4_cmd_info *cmd)
3158 {
3159         int err;
3160         int cqn = vhcr->in_modifier;
3161         struct mlx4_cq_context *cqc = inbox->buf;
3162         int mtt_base = cq_get_mtt_addr(cqc) / dev->caps.mtt_entry_sz;
3163         struct res_cq *cq;
3164         struct res_mtt *mtt;
3165
3166         err = cq_res_start_move_to(dev, slave, cqn, RES_CQ_HW, &cq);
3167         if (err)
3168                 return err;
3169         err = get_res(dev, slave, mtt_base, RES_MTT, &mtt);
3170         if (err)
3171                 goto out_move;
3172         err = check_mtt_range(dev, slave, mtt_base, cq_get_mtt_size(cqc), mtt);
3173         if (err)
3174                 goto out_put;
3175         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3176         if (err)
3177                 goto out_put;
3178         atomic_inc(&mtt->ref_count);
3179         cq->mtt = mtt;
3180         put_res(dev, slave, mtt->com.res_id, RES_MTT);
3181         res_end_move(dev, slave, RES_CQ, cqn);
3182         return 0;
3183
3184 out_put:
3185         put_res(dev, slave, mtt->com.res_id, RES_MTT);
3186 out_move:
3187         res_abort_move(dev, slave, RES_CQ, cqn);
3188         return err;
3189 }
3190
3191 int mlx4_HW2SW_CQ_wrapper(struct mlx4_dev *dev, int slave,
3192                           struct mlx4_vhcr *vhcr,
3193                           struct mlx4_cmd_mailbox *inbox,
3194                           struct mlx4_cmd_mailbox *outbox,
3195                           struct mlx4_cmd_info *cmd)
3196 {
3197         int err;
3198         int cqn = vhcr->in_modifier;
3199         struct res_cq *cq;
3200
3201         err = cq_res_start_move_to(dev, slave, cqn, RES_CQ_ALLOCATED, &cq);
3202         if (err)
3203                 return err;
3204         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3205         if (err)
3206                 goto out_move;
3207         atomic_dec(&cq->mtt->ref_count);
3208         res_end_move(dev, slave, RES_CQ, cqn);
3209         return 0;
3210
3211 out_move:
3212         res_abort_move(dev, slave, RES_CQ, cqn);
3213         return err;
3214 }
3215
3216 int mlx4_QUERY_CQ_wrapper(struct mlx4_dev *dev, int slave,
3217                           struct mlx4_vhcr *vhcr,
3218                           struct mlx4_cmd_mailbox *inbox,
3219                           struct mlx4_cmd_mailbox *outbox,
3220                           struct mlx4_cmd_info *cmd)
3221 {
3222         int cqn = vhcr->in_modifier;
3223         struct res_cq *cq;
3224         int err;
3225
3226         err = get_res(dev, slave, cqn, RES_CQ, &cq);
3227         if (err)
3228                 return err;
3229
3230         if (cq->com.from_state != RES_CQ_HW)
3231                 goto ex_put;
3232
3233         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3234 ex_put:
3235         put_res(dev, slave, cqn, RES_CQ);
3236
3237         return err;
3238 }
3239
3240 static int handle_resize(struct mlx4_dev *dev, int slave,
3241                          struct mlx4_vhcr *vhcr,
3242                          struct mlx4_cmd_mailbox *inbox,
3243                          struct mlx4_cmd_mailbox *outbox,
3244                          struct mlx4_cmd_info *cmd,
3245                          struct res_cq *cq)
3246 {
3247         int err;
3248         struct res_mtt *orig_mtt;
3249         struct res_mtt *mtt;
3250         struct mlx4_cq_context *cqc = inbox->buf;
3251         int mtt_base = cq_get_mtt_addr(cqc) / dev->caps.mtt_entry_sz;
3252
3253         err = get_res(dev, slave, cq->mtt->com.res_id, RES_MTT, &orig_mtt);
3254         if (err)
3255                 return err;
3256
3257         if (orig_mtt != cq->mtt) {
3258                 err = -EINVAL;
3259                 goto ex_put;
3260         }
3261
3262         err = get_res(dev, slave, mtt_base, RES_MTT, &mtt);
3263         if (err)
3264                 goto ex_put;
3265
3266         err = check_mtt_range(dev, slave, mtt_base, cq_get_mtt_size(cqc), mtt);
3267         if (err)
3268                 goto ex_put1;
3269         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3270         if (err)
3271                 goto ex_put1;
3272         atomic_dec(&orig_mtt->ref_count);
3273         put_res(dev, slave, orig_mtt->com.res_id, RES_MTT);
3274         atomic_inc(&mtt->ref_count);
3275         cq->mtt = mtt;
3276         put_res(dev, slave, mtt->com.res_id, RES_MTT);
3277         return 0;
3278
3279 ex_put1:
3280         put_res(dev, slave, mtt->com.res_id, RES_MTT);
3281 ex_put:
3282         put_res(dev, slave, orig_mtt->com.res_id, RES_MTT);
3283
3284         return err;
3285
3286 }
3287
3288 int mlx4_MODIFY_CQ_wrapper(struct mlx4_dev *dev, int slave,
3289                            struct mlx4_vhcr *vhcr,
3290                            struct mlx4_cmd_mailbox *inbox,
3291                            struct mlx4_cmd_mailbox *outbox,
3292                            struct mlx4_cmd_info *cmd)
3293 {
3294         int cqn = vhcr->in_modifier;
3295         struct res_cq *cq;
3296         int err;
3297
3298         err = get_res(dev, slave, cqn, RES_CQ, &cq);
3299         if (err)
3300                 return err;
3301
3302         if (cq->com.from_state != RES_CQ_HW)
3303                 goto ex_put;
3304
3305         if (vhcr->op_modifier == 0) {
3306                 err = handle_resize(dev, slave, vhcr, inbox, outbox, cmd, cq);
3307                 goto ex_put;
3308         }
3309
3310         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3311 ex_put:
3312         put_res(dev, slave, cqn, RES_CQ);
3313
3314         return err;
3315 }
3316
3317 static int srq_get_mtt_size(struct mlx4_srq_context *srqc)
3318 {
3319         int log_srq_size = (be32_to_cpu(srqc->state_logsize_srqn) >> 24) & 0xf;
3320         int log_rq_stride = srqc->logstride & 7;
3321         int page_shift = (srqc->log_page_size & 0x3f) + 12;
3322
3323         if (log_srq_size + log_rq_stride + 4 < page_shift)
3324                 return 1;
3325
3326         return 1 << (log_srq_size + log_rq_stride + 4 - page_shift);
3327 }
3328
3329 int mlx4_SW2HW_SRQ_wrapper(struct mlx4_dev *dev, int slave,
3330                            struct mlx4_vhcr *vhcr,
3331                            struct mlx4_cmd_mailbox *inbox,
3332                            struct mlx4_cmd_mailbox *outbox,
3333                            struct mlx4_cmd_info *cmd)
3334 {
3335         int err;
3336         int srqn = vhcr->in_modifier;
3337         struct res_mtt *mtt;
3338         struct res_srq *srq;
3339         struct mlx4_srq_context *srqc = inbox->buf;
3340         int mtt_base = srq_get_mtt_addr(srqc) / dev->caps.mtt_entry_sz;
3341
3342         if (srqn != (be32_to_cpu(srqc->state_logsize_srqn) & 0xffffff))
3343                 return -EINVAL;
3344
3345         err = srq_res_start_move_to(dev, slave, srqn, RES_SRQ_HW, &srq);
3346         if (err)
3347                 return err;
3348         err = get_res(dev, slave, mtt_base, RES_MTT, &mtt);
3349         if (err)
3350                 goto ex_abort;
3351         err = check_mtt_range(dev, slave, mtt_base, srq_get_mtt_size(srqc),
3352                               mtt);
3353         if (err)
3354                 goto ex_put_mtt;
3355
3356         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3357         if (err)
3358                 goto ex_put_mtt;
3359
3360         atomic_inc(&mtt->ref_count);
3361         srq->mtt = mtt;
3362         put_res(dev, slave, mtt->com.res_id, RES_MTT);
3363         res_end_move(dev, slave, RES_SRQ, srqn);
3364         return 0;
3365
3366 ex_put_mtt:
3367         put_res(dev, slave, mtt->com.res_id, RES_MTT);
3368 ex_abort:
3369         res_abort_move(dev, slave, RES_SRQ, srqn);
3370
3371         return err;
3372 }
3373
3374 int mlx4_HW2SW_SRQ_wrapper(struct mlx4_dev *dev, int slave,
3375                            struct mlx4_vhcr *vhcr,
3376                            struct mlx4_cmd_mailbox *inbox,
3377                            struct mlx4_cmd_mailbox *outbox,
3378                            struct mlx4_cmd_info *cmd)
3379 {
3380         int err;
3381         int srqn = vhcr->in_modifier;
3382         struct res_srq *srq;
3383
3384         err = srq_res_start_move_to(dev, slave, srqn, RES_SRQ_ALLOCATED, &srq);
3385         if (err)
3386                 return err;
3387         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3388         if (err)
3389                 goto ex_abort;
3390         atomic_dec(&srq->mtt->ref_count);
3391         if (srq->cq)
3392                 atomic_dec(&srq->cq->ref_count);
3393         res_end_move(dev, slave, RES_SRQ, srqn);
3394
3395         return 0;
3396
3397 ex_abort:
3398         res_abort_move(dev, slave, RES_SRQ, srqn);
3399
3400         return err;
3401 }
3402
3403 int mlx4_QUERY_SRQ_wrapper(struct mlx4_dev *dev, int slave,
3404                            struct mlx4_vhcr *vhcr,
3405                            struct mlx4_cmd_mailbox *inbox,
3406                            struct mlx4_cmd_mailbox *outbox,
3407                            struct mlx4_cmd_info *cmd)
3408 {
3409         int err;
3410         int srqn = vhcr->in_modifier;
3411         struct res_srq *srq;
3412
3413         err = get_res(dev, slave, srqn, RES_SRQ, &srq);
3414         if (err)
3415                 return err;
3416         if (srq->com.from_state != RES_SRQ_HW) {
3417                 err = -EBUSY;
3418                 goto out;
3419         }
3420         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3421 out:
3422         put_res(dev, slave, srqn, RES_SRQ);
3423         return err;
3424 }
3425
3426 int mlx4_ARM_SRQ_wrapper(struct mlx4_dev *dev, int slave,
3427                          struct mlx4_vhcr *vhcr,
3428                          struct mlx4_cmd_mailbox *inbox,
3429                          struct mlx4_cmd_mailbox *outbox,
3430                          struct mlx4_cmd_info *cmd)
3431 {
3432         int err;
3433         int srqn = vhcr->in_modifier;
3434         struct res_srq *srq;
3435
3436         err = get_res(dev, slave, srqn, RES_SRQ, &srq);
3437         if (err)
3438                 return err;
3439
3440         if (srq->com.from_state != RES_SRQ_HW) {
3441                 err = -EBUSY;
3442                 goto out;
3443         }
3444
3445         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3446 out:
3447         put_res(dev, slave, srqn, RES_SRQ);
3448         return err;
3449 }
3450
3451 int mlx4_GEN_QP_wrapper(struct mlx4_dev *dev, int slave,
3452                         struct mlx4_vhcr *vhcr,
3453                         struct mlx4_cmd_mailbox *inbox,
3454                         struct mlx4_cmd_mailbox *outbox,
3455                         struct mlx4_cmd_info *cmd)
3456 {
3457         int err;
3458         int qpn = vhcr->in_modifier & 0x7fffff;
3459         struct res_qp *qp;
3460
3461         err = get_res(dev, slave, qpn, RES_QP, &qp);
3462         if (err)
3463                 return err;
3464         if (qp->com.from_state != RES_QP_HW) {
3465                 err = -EBUSY;
3466                 goto out;
3467         }
3468
3469         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3470 out:
3471         put_res(dev, slave, qpn, RES_QP);
3472         return err;
3473 }
3474
3475 int mlx4_INIT2INIT_QP_wrapper(struct mlx4_dev *dev, int slave,
3476                               struct mlx4_vhcr *vhcr,
3477                               struct mlx4_cmd_mailbox *inbox,
3478                               struct mlx4_cmd_mailbox *outbox,
3479                               struct mlx4_cmd_info *cmd)
3480 {
3481         struct mlx4_qp_context *context = inbox->buf + 8;
3482         adjust_proxy_tun_qkey(dev, vhcr, context);
3483         update_pkey_index(dev, slave, inbox);
3484         return mlx4_GEN_QP_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3485 }
3486
3487 static int adjust_qp_sched_queue(struct mlx4_dev *dev, int slave,
3488                                   struct mlx4_qp_context *qpc,
3489                                   struct mlx4_cmd_mailbox *inbox)
3490 {
3491         enum mlx4_qp_optpar optpar = be32_to_cpu(*(__be32 *)inbox->buf);
3492         u8 pri_sched_queue;
3493         int port = mlx4_slave_convert_port(
3494                    dev, slave, (qpc->pri_path.sched_queue >> 6 & 1) + 1) - 1;
3495
3496         if (port < 0)
3497                 return -EINVAL;
3498
3499         pri_sched_queue = (qpc->pri_path.sched_queue & ~(1 << 6)) |
3500                           ((port & 1) << 6);
3501
3502         if (optpar & MLX4_QP_OPTPAR_PRIMARY_ADDR_PATH ||
3503             mlx4_is_eth(dev, port + 1)) {
3504                 qpc->pri_path.sched_queue = pri_sched_queue;
3505         }
3506
3507         if (optpar & MLX4_QP_OPTPAR_ALT_ADDR_PATH) {
3508                 port = mlx4_slave_convert_port(
3509                                 dev, slave, (qpc->alt_path.sched_queue >> 6 & 1)
3510                                 + 1) - 1;
3511                 if (port < 0)
3512                         return -EINVAL;
3513                 qpc->alt_path.sched_queue =
3514                         (qpc->alt_path.sched_queue & ~(1 << 6)) |
3515                         (port & 1) << 6;
3516         }
3517         return 0;
3518 }
3519
3520 static int roce_verify_mac(struct mlx4_dev *dev, int slave,
3521                                 struct mlx4_qp_context *qpc,
3522                                 struct mlx4_cmd_mailbox *inbox)
3523 {
3524         u64 mac;
3525         int port;
3526         u32 ts = (be32_to_cpu(qpc->flags) >> 16) & 0xff;
3527         u8 sched = *(u8 *)(inbox->buf + 64);
3528         u8 smac_ix;
3529
3530         port = (sched >> 6 & 1) + 1;
3531         if (mlx4_is_eth(dev, port) && (ts != MLX4_QP_ST_MLX)) {
3532                 smac_ix = qpc->pri_path.grh_mylmc & 0x7f;
3533                 if (mac_find_smac_ix_in_slave(dev, slave, port, smac_ix, &mac))
3534                         return -ENOENT;
3535         }
3536         return 0;
3537 }
3538
3539 int mlx4_INIT2RTR_QP_wrapper(struct mlx4_dev *dev, int slave,
3540                              struct mlx4_vhcr *vhcr,
3541                              struct mlx4_cmd_mailbox *inbox,
3542                              struct mlx4_cmd_mailbox *outbox,
3543                              struct mlx4_cmd_info *cmd)
3544 {
3545         int err;
3546         struct mlx4_qp_context *qpc = inbox->buf + 8;
3547         int qpn = vhcr->in_modifier & 0x7fffff;
3548         struct res_qp *qp;
3549         u8 orig_sched_queue;
3550         __be32  orig_param3 = qpc->param3;
3551         u8 orig_vlan_control = qpc->pri_path.vlan_control;
3552         u8 orig_fvl_rx = qpc->pri_path.fvl_rx;
3553         u8 orig_pri_path_fl = qpc->pri_path.fl;
3554         u8 orig_vlan_index = qpc->pri_path.vlan_index;
3555         u8 orig_feup = qpc->pri_path.feup;
3556
3557         err = adjust_qp_sched_queue(dev, slave, qpc, inbox);
3558         if (err)
3559                 return err;
3560         err = verify_qp_parameters(dev, vhcr, inbox, QP_TRANS_INIT2RTR, slave);
3561         if (err)
3562                 return err;
3563
3564         if (roce_verify_mac(dev, slave, qpc, inbox))
3565                 return -EINVAL;
3566
3567         update_pkey_index(dev, slave, inbox);
3568         update_gid(dev, inbox, (u8)slave);
3569         adjust_proxy_tun_qkey(dev, vhcr, qpc);
3570         orig_sched_queue = qpc->pri_path.sched_queue;
3571         err = update_vport_qp_param(dev, inbox, slave, qpn);
3572         if (err)
3573                 return err;
3574
3575         err = get_res(dev, slave, qpn, RES_QP, &qp);
3576         if (err)
3577                 return err;
3578         if (qp->com.from_state != RES_QP_HW) {
3579                 err = -EBUSY;
3580                 goto out;
3581         }
3582
3583         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3584 out:
3585         /* if no error, save sched queue value passed in by VF. This is
3586          * essentially the QOS value provided by the VF. This will be useful
3587          * if we allow dynamic changes from VST back to VGT
3588          */
3589         if (!err) {
3590                 qp->sched_queue = orig_sched_queue;
3591                 qp->param3      = orig_param3;
3592                 qp->vlan_control = orig_vlan_control;
3593                 qp->fvl_rx      =  orig_fvl_rx;
3594                 qp->pri_path_fl = orig_pri_path_fl;
3595                 qp->vlan_index  = orig_vlan_index;
3596                 qp->feup        = orig_feup;
3597         }
3598         put_res(dev, slave, qpn, RES_QP);
3599         return err;
3600 }
3601
3602 int mlx4_RTR2RTS_QP_wrapper(struct mlx4_dev *dev, int slave,
3603                             struct mlx4_vhcr *vhcr,
3604                             struct mlx4_cmd_mailbox *inbox,
3605                             struct mlx4_cmd_mailbox *outbox,
3606                             struct mlx4_cmd_info *cmd)
3607 {
3608         int err;
3609         struct mlx4_qp_context *context = inbox->buf + 8;
3610
3611         err = adjust_qp_sched_queue(dev, slave, context, inbox);
3612         if (err)
3613                 return err;
3614         err = verify_qp_parameters(dev, vhcr, inbox, QP_TRANS_RTR2RTS, slave);
3615         if (err)
3616                 return err;
3617
3618         update_pkey_index(dev, slave, inbox);
3619         update_gid(dev, inbox, (u8)slave);
3620         adjust_proxy_tun_qkey(dev, vhcr, context);
3621         return mlx4_GEN_QP_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3622 }
3623
3624 int mlx4_RTS2RTS_QP_wrapper(struct mlx4_dev *dev, int slave,
3625                             struct mlx4_vhcr *vhcr,
3626                             struct mlx4_cmd_mailbox *inbox,
3627                             struct mlx4_cmd_mailbox *outbox,
3628                             struct mlx4_cmd_info *cmd)
3629 {
3630         int err;
3631         struct mlx4_qp_context *context = inbox->buf + 8;
3632
3633         err = adjust_qp_sched_queue(dev, slave, context, inbox);
3634         if (err)
3635                 return err;
3636         err = verify_qp_parameters(dev, vhcr, inbox, QP_TRANS_RTS2RTS, slave);
3637         if (err)
3638                 return err;
3639
3640         update_pkey_index(dev, slave, inbox);
3641         update_gid(dev, inbox, (u8)slave);
3642         adjust_proxy_tun_qkey(dev, vhcr, context);
3643         return mlx4_GEN_QP_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3644 }
3645
3646
3647 int mlx4_SQERR2RTS_QP_wrapper(struct mlx4_dev *dev, int slave,
3648                               struct mlx4_vhcr *vhcr,
3649                               struct mlx4_cmd_mailbox *inbox,
3650                               struct mlx4_cmd_mailbox *outbox,
3651                               struct mlx4_cmd_info *cmd)
3652 {
3653         struct mlx4_qp_context *context = inbox->buf + 8;
3654         int err = adjust_qp_sched_queue(dev, slave, context, inbox);
3655         if (err)
3656                 return err;
3657         adjust_proxy_tun_qkey(dev, vhcr, context);
3658         return mlx4_GEN_QP_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3659 }
3660
3661 int mlx4_SQD2SQD_QP_wrapper(struct mlx4_dev *dev, int slave,
3662                             struct mlx4_vhcr *vhcr,
3663                             struct mlx4_cmd_mailbox *inbox,
3664                             struct mlx4_cmd_mailbox *outbox,
3665                             struct mlx4_cmd_info *cmd)
3666 {
3667         int err;
3668         struct mlx4_qp_context *context = inbox->buf + 8;
3669
3670         err = adjust_qp_sched_queue(dev, slave, context, inbox);
3671         if (err)
3672                 return err;
3673         err = verify_qp_parameters(dev, vhcr, inbox, QP_TRANS_SQD2SQD, slave);
3674         if (err)
3675                 return err;
3676
3677         adjust_proxy_tun_qkey(dev, vhcr, context);
3678         update_gid(dev, inbox, (u8)slave);
3679         update_pkey_index(dev, slave, inbox);
3680         return mlx4_GEN_QP_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3681 }
3682
3683 int mlx4_SQD2RTS_QP_wrapper(struct mlx4_dev *dev, int slave,
3684                             struct mlx4_vhcr *vhcr,
3685                             struct mlx4_cmd_mailbox *inbox,
3686                             struct mlx4_cmd_mailbox *outbox,
3687                             struct mlx4_cmd_info *cmd)
3688 {
3689         int err;
3690         struct mlx4_qp_context *context = inbox->buf + 8;
3691
3692         err = adjust_qp_sched_queue(dev, slave, context, inbox);
3693         if (err)
3694                 return err;
3695         err = verify_qp_parameters(dev, vhcr, inbox, QP_TRANS_SQD2RTS, slave);
3696         if (err)
3697                 return err;
3698
3699         adjust_proxy_tun_qkey(dev, vhcr, context);
3700         update_gid(dev, inbox, (u8)slave);
3701         update_pkey_index(dev, slave, inbox);
3702         return mlx4_GEN_QP_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3703 }
3704
3705 int mlx4_2RST_QP_wrapper(struct mlx4_dev *dev, int slave,
3706                          struct mlx4_vhcr *vhcr,
3707                          struct mlx4_cmd_mailbox *inbox,
3708                          struct mlx4_cmd_mailbox *outbox,
3709                          struct mlx4_cmd_info *cmd)
3710 {
3711         int err;
3712         int qpn = vhcr->in_modifier & 0x7fffff;
3713         struct res_qp *qp;
3714
3715         err = qp_res_start_move_to(dev, slave, qpn, RES_QP_MAPPED, &qp, 0);
3716         if (err)
3717                 return err;
3718         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
3719         if (err)
3720                 goto ex_abort;
3721
3722         atomic_dec(&qp->mtt->ref_count);
3723         atomic_dec(&qp->rcq->ref_count);
3724         atomic_dec(&qp->scq->ref_count);
3725         if (qp->srq)
3726                 atomic_dec(&qp->srq->ref_count);
3727         res_end_move(dev, slave, RES_QP, qpn);
3728         return 0;
3729
3730 ex_abort:
3731         res_abort_move(dev, slave, RES_QP, qpn);
3732
3733         return err;
3734 }
3735
3736 static struct res_gid *find_gid(struct mlx4_dev *dev, int slave,
3737                                 struct res_qp *rqp, u8 *gid)
3738 {
3739         struct res_gid *res;
3740
3741         list_for_each_entry(res, &rqp->mcg_list, list) {
3742                 if (!memcmp(res->gid, gid, 16))
3743                         return res;
3744         }
3745         return NULL;
3746 }
3747
3748 static int add_mcg_res(struct mlx4_dev *dev, int slave, struct res_qp *rqp,
3749                        u8 *gid, enum mlx4_protocol prot,
3750                        enum mlx4_steer_type steer, u64 reg_id)
3751 {
3752         struct res_gid *res;
3753         int err;
3754
3755         res = kzalloc(sizeof *res, GFP_KERNEL);
3756         if (!res)
3757                 return -ENOMEM;
3758
3759         spin_lock_irq(&rqp->mcg_spl);
3760         if (find_gid(dev, slave, rqp, gid)) {
3761                 kfree(res);
3762                 err = -EEXIST;
3763         } else {
3764                 memcpy(res->gid, gid, 16);
3765                 res->prot = prot;
3766                 res->steer = steer;
3767                 res->reg_id = reg_id;
3768                 list_add_tail(&res->list, &rqp->mcg_list);
3769                 err = 0;
3770         }
3771         spin_unlock_irq(&rqp->mcg_spl);
3772
3773         return err;
3774 }
3775
3776 static int rem_mcg_res(struct mlx4_dev *dev, int slave, struct res_qp *rqp,
3777                        u8 *gid, enum mlx4_protocol prot,
3778                        enum mlx4_steer_type steer, u64 *reg_id)
3779 {
3780         struct res_gid *res;
3781         int err;
3782
3783         spin_lock_irq(&rqp->mcg_spl);
3784         res = find_gid(dev, slave, rqp, gid);
3785         if (!res || res->prot != prot || res->steer != steer)
3786                 err = -EINVAL;
3787         else {
3788                 *reg_id = res->reg_id;
3789                 list_del(&res->list);
3790                 kfree(res);
3791                 err = 0;
3792         }
3793         spin_unlock_irq(&rqp->mcg_spl);
3794
3795         return err;
3796 }
3797
3798 static int qp_attach(struct mlx4_dev *dev, int slave, struct mlx4_qp *qp,
3799                      u8 gid[16], int block_loopback, enum mlx4_protocol prot,
3800                      enum mlx4_steer_type type, u64 *reg_id)
3801 {
3802         switch (dev->caps.steering_mode) {
3803         case MLX4_STEERING_MODE_DEVICE_MANAGED: {
3804                 int port = mlx4_slave_convert_port(dev, slave, gid[5]);
3805                 if (port < 0)
3806                         return port;
3807                 return mlx4_trans_to_dmfs_attach(dev, qp, gid, port,
3808                                                 block_loopback, prot,
3809                                                 reg_id);
3810         }
3811         case MLX4_STEERING_MODE_B0:
3812                 if (prot == MLX4_PROT_ETH) {
3813                         int port = mlx4_slave_convert_port(dev, slave, gid[5]);
3814                         if (port < 0)
3815                                 return port;
3816                         gid[5] = port;
3817                 }
3818                 return mlx4_qp_attach_common(dev, qp, gid,
3819                                             block_loopback, prot, type);
3820         default:
3821                 return -EINVAL;
3822         }
3823 }
3824
3825 static int qp_detach(struct mlx4_dev *dev, struct mlx4_qp *qp,
3826                      u8 gid[16], enum mlx4_protocol prot,
3827                      enum mlx4_steer_type type, u64 reg_id)
3828 {
3829         switch (dev->caps.steering_mode) {
3830         case MLX4_STEERING_MODE_DEVICE_MANAGED:
3831                 return mlx4_flow_detach(dev, reg_id);
3832         case MLX4_STEERING_MODE_B0:
3833                 return mlx4_qp_detach_common(dev, qp, gid, prot, type);
3834         default:
3835                 return -EINVAL;
3836         }
3837 }
3838
3839 static int mlx4_adjust_port(struct mlx4_dev *dev, int slave,
3840                             u8 *gid, enum mlx4_protocol prot)
3841 {
3842         int real_port;
3843
3844         if (prot != MLX4_PROT_ETH)
3845                 return 0;
3846
3847         if (dev->caps.steering_mode == MLX4_STEERING_MODE_B0 ||
3848             dev->caps.steering_mode == MLX4_STEERING_MODE_DEVICE_MANAGED) {
3849                 real_port = mlx4_slave_convert_port(dev, slave, gid[5]);
3850                 if (real_port < 0)
3851                         return -EINVAL;
3852                 gid[5] = real_port;
3853         }
3854
3855         return 0;
3856 }
3857
3858 int mlx4_QP_ATTACH_wrapper(struct mlx4_dev *dev, int slave,
3859                                struct mlx4_vhcr *vhcr,
3860                                struct mlx4_cmd_mailbox *inbox,
3861                                struct mlx4_cmd_mailbox *outbox,
3862                                struct mlx4_cmd_info *cmd)
3863 {
3864         struct mlx4_qp qp; /* dummy for calling attach/detach */
3865         u8 *gid = inbox->buf;
3866         enum mlx4_protocol prot = (vhcr->in_modifier >> 28) & 0x7;
3867         int err;
3868         int qpn;
3869         struct res_qp *rqp;
3870         u64 reg_id = 0;
3871         int attach = vhcr->op_modifier;
3872         int block_loopback = vhcr->in_modifier >> 31;
3873         u8 steer_type_mask = 2;
3874         enum mlx4_steer_type type = (gid[7] & steer_type_mask) >> 1;
3875
3876         qpn = vhcr->in_modifier & 0xffffff;
3877         err = get_res(dev, slave, qpn, RES_QP, &rqp);
3878         if (err)
3879                 return err;
3880
3881         qp.qpn = qpn;
3882         if (attach) {
3883                 err = qp_attach(dev, slave, &qp, gid, block_loopback, prot,
3884                                 type, &reg_id);
3885                 if (err) {
3886                         pr_err("Fail to attach rule to qp 0x%x\n", qpn);
3887                         goto ex_put;
3888                 }
3889                 err = add_mcg_res(dev, slave, rqp, gid, prot, type, reg_id);
3890                 if (err)
3891                         goto ex_detach;
3892         } else {
3893                 err = mlx4_adjust_port(dev, slave, gid, prot);
3894                 if (err)
3895                         goto ex_put;
3896
3897                 err = rem_mcg_res(dev, slave, rqp, gid, prot, type, &reg_id);
3898                 if (err)
3899                         goto ex_put;
3900
3901                 err = qp_detach(dev, &qp, gid, prot, type, reg_id);
3902                 if (err)
3903                         pr_err("Fail to detach rule from qp 0x%x reg_id = 0x%llx\n",
3904                                qpn, reg_id);
3905         }
3906         put_res(dev, slave, qpn, RES_QP);
3907         return err;
3908
3909 ex_detach:
3910         qp_detach(dev, &qp, gid, prot, type, reg_id);
3911 ex_put:
3912         put_res(dev, slave, qpn, RES_QP);
3913         return err;
3914 }
3915
3916 /*
3917  * MAC validation for Flow Steering rules.
3918  * VF can attach rules only with a mac address which is assigned to it.
3919  */
3920 static int validate_eth_header_mac(int slave, struct _rule_hw *eth_header,
3921                                    struct list_head *rlist)
3922 {
3923         struct mac_res *res, *tmp;
3924         __be64 be_mac;
3925
3926         /* make sure it isn't multicast or broadcast mac*/
3927         if (!is_multicast_ether_addr(eth_header->eth.dst_mac) &&
3928             !is_broadcast_ether_addr(eth_header->eth.dst_mac)) {
3929                 list_for_each_entry_safe(res, tmp, rlist, list) {
3930                         be_mac = cpu_to_be64(res->mac << 16);
3931                         if (ether_addr_equal((u8 *)&be_mac, eth_header->eth.dst_mac))
3932                                 return 0;
3933                 }
3934                 pr_err("MAC %pM doesn't belong to VF %d, Steering rule rejected\n",
3935                        eth_header->eth.dst_mac, slave);
3936                 return -EINVAL;
3937         }
3938         return 0;
3939 }
3940
3941 /*
3942  * In case of missing eth header, append eth header with a MAC address
3943  * assigned to the VF.
3944  */
3945 static int add_eth_header(struct mlx4_dev *dev, int slave,
3946                           struct mlx4_cmd_mailbox *inbox,
3947                           struct list_head *rlist, int header_id)
3948 {
3949         struct mac_res *res, *tmp;
3950         u8 port;
3951         struct mlx4_net_trans_rule_hw_ctrl *ctrl;
3952         struct mlx4_net_trans_rule_hw_eth *eth_header;
3953         struct mlx4_net_trans_rule_hw_ipv4 *ip_header;
3954         struct mlx4_net_trans_rule_hw_tcp_udp *l4_header;
3955         __be64 be_mac = 0;
3956         __be64 mac_msk = cpu_to_be64(MLX4_MAC_MASK << 16);
3957
3958         ctrl = (struct mlx4_net_trans_rule_hw_ctrl *)inbox->buf;
3959         port = ctrl->port;
3960         eth_header = (struct mlx4_net_trans_rule_hw_eth *)(ctrl + 1);
3961
3962         /* Clear a space in the inbox for eth header */
3963         switch (header_id) {
3964         case MLX4_NET_TRANS_RULE_ID_IPV4:
3965                 ip_header =
3966                         (struct mlx4_net_trans_rule_hw_ipv4 *)(eth_header + 1);
3967                 memmove(ip_header, eth_header,
3968                         sizeof(*ip_header) + sizeof(*l4_header));
3969                 break;
3970         case MLX4_NET_TRANS_RULE_ID_TCP:
3971         case MLX4_NET_TRANS_RULE_ID_UDP:
3972                 l4_header = (struct mlx4_net_trans_rule_hw_tcp_udp *)
3973                             (eth_header + 1);
3974                 memmove(l4_header, eth_header, sizeof(*l4_header));
3975                 break;
3976         default:
3977                 return -EINVAL;
3978         }
3979         list_for_each_entry_safe(res, tmp, rlist, list) {
3980                 if (port == res->port) {
3981                         be_mac = cpu_to_be64(res->mac << 16);
3982                         break;
3983                 }
3984         }
3985         if (!be_mac) {
3986                 pr_err("Failed adding eth header to FS rule, Can't find matching MAC for port %d\n",
3987                        port);
3988                 return -EINVAL;
3989         }
3990
3991         memset(eth_header, 0, sizeof(*eth_header));
3992         eth_header->size = sizeof(*eth_header) >> 2;
3993         eth_header->id = cpu_to_be16(__sw_id_hw[MLX4_NET_TRANS_RULE_ID_ETH]);
3994         memcpy(eth_header->dst_mac, &be_mac, ETH_ALEN);
3995         memcpy(eth_header->dst_mac_msk, &mac_msk, ETH_ALEN);
3996
3997         return 0;
3998
3999 }
4000
4001 #define MLX4_UPD_QP_PATH_MASK_SUPPORTED (1ULL << MLX4_UPD_QP_PATH_MASK_MAC_INDEX)
4002 int mlx4_UPDATE_QP_wrapper(struct mlx4_dev *dev, int slave,
4003                            struct mlx4_vhcr *vhcr,
4004                            struct mlx4_cmd_mailbox *inbox,
4005                            struct mlx4_cmd_mailbox *outbox,
4006                            struct mlx4_cmd_info *cmd_info)
4007 {
4008         int err;
4009         u32 qpn = vhcr->in_modifier & 0xffffff;
4010         struct res_qp *rqp;
4011         u64 mac;
4012         unsigned port;
4013         u64 pri_addr_path_mask;
4014         struct mlx4_update_qp_context *cmd;
4015         int smac_index;
4016
4017         cmd = (struct mlx4_update_qp_context *)inbox->buf;
4018
4019         pri_addr_path_mask = be64_to_cpu(cmd->primary_addr_path_mask);
4020         if (cmd->qp_mask || cmd->secondary_addr_path_mask ||
4021             (pri_addr_path_mask & ~MLX4_UPD_QP_PATH_MASK_SUPPORTED))
4022                 return -EPERM;
4023
4024         /* Just change the smac for the QP */
4025         err = get_res(dev, slave, qpn, RES_QP, &rqp);
4026         if (err) {
4027                 mlx4_err(dev, "Updating qpn 0x%x for slave %d rejected\n", qpn, slave);
4028                 return err;
4029         }
4030
4031         port = (rqp->sched_queue >> 6 & 1) + 1;
4032
4033         if (pri_addr_path_mask & (1ULL << MLX4_UPD_QP_PATH_MASK_MAC_INDEX)) {
4034                 smac_index = cmd->qp_context.pri_path.grh_mylmc;
4035                 err = mac_find_smac_ix_in_slave(dev, slave, port,
4036                                                 smac_index, &mac);
4037
4038                 if (err) {
4039                         mlx4_err(dev, "Failed to update qpn 0x%x, MAC is invalid. smac_ix: %d\n",
4040                                  qpn, smac_index);
4041                         goto err_mac;
4042                 }
4043         }
4044
4045         err = mlx4_cmd(dev, inbox->dma,
4046                        vhcr->in_modifier, 0,
4047                        MLX4_CMD_UPDATE_QP, MLX4_CMD_TIME_CLASS_A,
4048                        MLX4_CMD_NATIVE);
4049         if (err) {
4050                 mlx4_err(dev, "Failed to update qpn on qpn 0x%x, command failed\n", qpn);
4051                 goto err_mac;
4052         }
4053
4054 err_mac:
4055         put_res(dev, slave, qpn, RES_QP);
4056         return err;
4057 }
4058
4059 int mlx4_QP_FLOW_STEERING_ATTACH_wrapper(struct mlx4_dev *dev, int slave,
4060                                          struct mlx4_vhcr *vhcr,
4061                                          struct mlx4_cmd_mailbox *inbox,
4062                                          struct mlx4_cmd_mailbox *outbox,
4063                                          struct mlx4_cmd_info *cmd)
4064 {
4065
4066         struct mlx4_priv *priv = mlx4_priv(dev);
4067         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
4068         struct list_head *rlist = &tracker->slave_list[slave].res_list[RES_MAC];
4069         int err;
4070         int qpn;
4071         struct res_qp *rqp;
4072         struct mlx4_net_trans_rule_hw_ctrl *ctrl;
4073         struct _rule_hw  *rule_header;
4074         int header_id;
4075
4076         if (dev->caps.steering_mode !=
4077             MLX4_STEERING_MODE_DEVICE_MANAGED)
4078                 return -EOPNOTSUPP;
4079
4080         ctrl = (struct mlx4_net_trans_rule_hw_ctrl *)inbox->buf;
4081         ctrl->port = mlx4_slave_convert_port(dev, slave, ctrl->port);
4082         if (ctrl->port <= 0)
4083                 return -EINVAL;
4084         qpn = be32_to_cpu(ctrl->qpn) & 0xffffff;
4085         err = get_res(dev, slave, qpn, RES_QP, &rqp);
4086         if (err) {
4087                 pr_err("Steering rule with qpn 0x%x rejected\n", qpn);
4088                 return err;
4089         }
4090         rule_header = (struct _rule_hw *)(ctrl + 1);
4091         header_id = map_hw_to_sw_id(be16_to_cpu(rule_header->id));
4092
4093         switch (header_id) {
4094         case MLX4_NET_TRANS_RULE_ID_ETH:
4095                 if (validate_eth_header_mac(slave, rule_header, rlist)) {
4096                         err = -EINVAL;
4097                         goto err_put;
4098                 }
4099                 break;
4100         case MLX4_NET_TRANS_RULE_ID_IB:
4101                 break;
4102         case MLX4_NET_TRANS_RULE_ID_IPV4:
4103         case MLX4_NET_TRANS_RULE_ID_TCP:
4104         case MLX4_NET_TRANS_RULE_ID_UDP:
4105                 pr_warn("Can't attach FS rule without L2 headers, adding L2 header\n");
4106                 if (add_eth_header(dev, slave, inbox, rlist, header_id)) {
4107                         err = -EINVAL;
4108                         goto err_put;
4109                 }
4110                 vhcr->in_modifier +=
4111                         sizeof(struct mlx4_net_trans_rule_hw_eth) >> 2;
4112                 break;
4113         default:
4114                 pr_err("Corrupted mailbox\n");
4115                 err = -EINVAL;
4116                 goto err_put;
4117         }
4118
4119         err = mlx4_cmd_imm(dev, inbox->dma, &vhcr->out_param,
4120                            vhcr->in_modifier, 0,
4121                            MLX4_QP_FLOW_STEERING_ATTACH, MLX4_CMD_TIME_CLASS_A,
4122                            MLX4_CMD_NATIVE);
4123         if (err)
4124                 goto err_put;
4125
4126         err = add_res_range(dev, slave, vhcr->out_param, 1, RES_FS_RULE, qpn);
4127         if (err) {
4128                 mlx4_err(dev, "Fail to add flow steering resources\n");
4129                 /* detach rule*/
4130                 mlx4_cmd(dev, vhcr->out_param, 0, 0,
4131                          MLX4_QP_FLOW_STEERING_DETACH, MLX4_CMD_TIME_CLASS_A,
4132                          MLX4_CMD_NATIVE);
4133                 goto err_put;
4134         }
4135         atomic_inc(&rqp->ref_count);
4136 err_put:
4137         put_res(dev, slave, qpn, RES_QP);
4138         return err;
4139 }
4140
4141 int mlx4_QP_FLOW_STEERING_DETACH_wrapper(struct mlx4_dev *dev, int slave,
4142                                          struct mlx4_vhcr *vhcr,
4143                                          struct mlx4_cmd_mailbox *inbox,
4144                                          struct mlx4_cmd_mailbox *outbox,
4145                                          struct mlx4_cmd_info *cmd)
4146 {
4147         int err;
4148         struct res_qp *rqp;
4149         struct res_fs_rule *rrule;
4150
4151         if (dev->caps.steering_mode !=
4152             MLX4_STEERING_MODE_DEVICE_MANAGED)
4153                 return -EOPNOTSUPP;
4154
4155         err = get_res(dev, slave, vhcr->in_param, RES_FS_RULE, &rrule);
4156         if (err)
4157                 return err;
4158         /* Release the rule form busy state before removal */
4159         put_res(dev, slave, vhcr->in_param, RES_FS_RULE);
4160         err = get_res(dev, slave, rrule->qpn, RES_QP, &rqp);
4161         if (err)
4162                 return err;
4163
4164         err = rem_res_range(dev, slave, vhcr->in_param, 1, RES_FS_RULE, 0);
4165         if (err) {
4166                 mlx4_err(dev, "Fail to remove flow steering resources\n");
4167                 goto out;
4168         }
4169
4170         err = mlx4_cmd(dev, vhcr->in_param, 0, 0,
4171                        MLX4_QP_FLOW_STEERING_DETACH, MLX4_CMD_TIME_CLASS_A,
4172                        MLX4_CMD_NATIVE);
4173         if (!err)
4174                 atomic_dec(&rqp->ref_count);
4175 out:
4176         put_res(dev, slave, rrule->qpn, RES_QP);
4177         return err;
4178 }
4179
4180 enum {
4181         BUSY_MAX_RETRIES = 10
4182 };
4183
4184 int mlx4_QUERY_IF_STAT_wrapper(struct mlx4_dev *dev, int slave,
4185                                struct mlx4_vhcr *vhcr,
4186                                struct mlx4_cmd_mailbox *inbox,
4187                                struct mlx4_cmd_mailbox *outbox,
4188                                struct mlx4_cmd_info *cmd)
4189 {
4190         int err;
4191         int index = vhcr->in_modifier & 0xffff;
4192
4193         err = get_res(dev, slave, index, RES_COUNTER, NULL);
4194         if (err)
4195                 return err;
4196
4197         err = mlx4_DMA_wrapper(dev, slave, vhcr, inbox, outbox, cmd);
4198         put_res(dev, slave, index, RES_COUNTER);
4199         return err;
4200 }
4201
4202 static void detach_qp(struct mlx4_dev *dev, int slave, struct res_qp *rqp)
4203 {
4204         struct res_gid *rgid;
4205         struct res_gid *tmp;
4206         struct mlx4_qp qp; /* dummy for calling attach/detach */
4207
4208         list_for_each_entry_safe(rgid, tmp, &rqp->mcg_list, list) {
4209                 switch (dev->caps.steering_mode) {
4210                 case MLX4_STEERING_MODE_DEVICE_MANAGED:
4211                         mlx4_flow_detach(dev, rgid->reg_id);
4212                         break;
4213                 case MLX4_STEERING_MODE_B0:
4214                         qp.qpn = rqp->local_qpn;
4215                         (void) mlx4_qp_detach_common(dev, &qp, rgid->gid,
4216                                                      rgid->prot, rgid->steer);
4217                         break;
4218                 }
4219                 list_del(&rgid->list);
4220                 kfree(rgid);
4221         }
4222 }
4223
4224 static int _move_all_busy(struct mlx4_dev *dev, int slave,
4225                           enum mlx4_resource type, int print)
4226 {
4227         struct mlx4_priv *priv = mlx4_priv(dev);
4228         struct mlx4_resource_tracker *tracker =
4229                 &priv->mfunc.master.res_tracker;
4230         struct list_head *rlist = &tracker->slave_list[slave].res_list[type];
4231         struct res_common *r;
4232         struct res_common *tmp;
4233         int busy;
4234
4235         busy = 0;
4236         spin_lock_irq(mlx4_tlock(dev));
4237         list_for_each_entry_safe(r, tmp, rlist, list) {
4238                 if (r->owner == slave) {
4239                         if (!r->removing) {
4240                                 if (r->state == RES_ANY_BUSY) {
4241                                         if (print)
4242                                                 mlx4_dbg(dev,
4243                                                          "%s id 0x%llx is busy\n",
4244                                                           resource_str(type),
4245                                                           r->res_id);
4246                                         ++busy;
4247                                 } else {
4248                                         r->from_state = r->state;
4249                                         r->state = RES_ANY_BUSY;
4250                                         r->removing = 1;
4251                                 }
4252                         }
4253                 }
4254         }
4255         spin_unlock_irq(mlx4_tlock(dev));
4256
4257         return busy;
4258 }
4259
4260 static int move_all_busy(struct mlx4_dev *dev, int slave,
4261                          enum mlx4_resource type)
4262 {
4263         unsigned long begin;
4264         int busy;
4265
4266         begin = jiffies;
4267         do {
4268                 busy = _move_all_busy(dev, slave, type, 0);
4269                 if (time_after(jiffies, begin + 5 * HZ))
4270                         break;
4271                 if (busy)
4272                         cond_resched();
4273         } while (busy);
4274
4275         if (busy)
4276                 busy = _move_all_busy(dev, slave, type, 1);
4277
4278         return busy;
4279 }
4280 static void rem_slave_qps(struct mlx4_dev *dev, int slave)
4281 {
4282         struct mlx4_priv *priv = mlx4_priv(dev);
4283         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
4284         struct list_head *qp_list =
4285                 &tracker->slave_list[slave].res_list[RES_QP];
4286         struct res_qp *qp;
4287         struct res_qp *tmp;
4288         int state;
4289         u64 in_param;
4290         int qpn;
4291         int err;
4292
4293         err = move_all_busy(dev, slave, RES_QP);
4294         if (err)
4295                 mlx4_warn(dev, "rem_slave_qps: Could not move all qps to busy for slave %d\n",
4296                           slave);
4297
4298         spin_lock_irq(mlx4_tlock(dev));
4299         list_for_each_entry_safe(qp, tmp, qp_list, com.list) {
4300                 spin_unlock_irq(mlx4_tlock(dev));
4301                 if (qp->com.owner == slave) {
4302                         qpn = qp->com.res_id;
4303                         detach_qp(dev, slave, qp);
4304                         state = qp->com.from_state;
4305                         while (state != 0) {
4306                                 switch (state) {
4307                                 case RES_QP_RESERVED:
4308                                         spin_lock_irq(mlx4_tlock(dev));
4309                                         rb_erase(&qp->com.node,
4310                                                  &tracker->res_tree[RES_QP]);
4311                                         list_del(&qp->com.list);
4312                                         spin_unlock_irq(mlx4_tlock(dev));
4313                                         if (!valid_reserved(dev, slave, qpn)) {
4314                                                 __mlx4_qp_release_range(dev, qpn, 1);
4315                                                 mlx4_release_resource(dev, slave,
4316                                                                       RES_QP, 1, 0);
4317                                         }
4318                                         kfree(qp);
4319                                         state = 0;
4320                                         break;
4321                                 case RES_QP_MAPPED:
4322                                         if (!valid_reserved(dev, slave, qpn))
4323                                                 __mlx4_qp_free_icm(dev, qpn);
4324                                         state = RES_QP_RESERVED;
4325                                         break;
4326                                 case RES_QP_HW:
4327                                         in_param = slave;
4328                                         err = mlx4_cmd(dev, in_param,
4329                                                        qp->local_qpn, 2,
4330                                                        MLX4_CMD_2RST_QP,
4331                                                        MLX4_CMD_TIME_CLASS_A,
4332                                                        MLX4_CMD_NATIVE);
4333                                         if (err)
4334                                                 mlx4_dbg(dev, "rem_slave_qps: failed to move slave %d qpn %d to reset\n",
4335                                                          slave, qp->local_qpn);
4336                                         atomic_dec(&qp->rcq->ref_count);
4337                                         atomic_dec(&qp->scq->ref_count);
4338                                         atomic_dec(&qp->mtt->ref_count);
4339                                         if (qp->srq)
4340                                                 atomic_dec(&qp->srq->ref_count);
4341                                         state = RES_QP_MAPPED;
4342                                         break;
4343                                 default:
4344                                         state = 0;
4345                                 }
4346                         }
4347                 }
4348                 spin_lock_irq(mlx4_tlock(dev));
4349         }
4350         spin_unlock_irq(mlx4_tlock(dev));
4351 }
4352
4353 static void rem_slave_srqs(struct mlx4_dev *dev, int slave)
4354 {
4355         struct mlx4_priv *priv = mlx4_priv(dev);
4356         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
4357         struct list_head *srq_list =
4358                 &tracker->slave_list[slave].res_list[RES_SRQ];
4359         struct res_srq *srq;
4360         struct res_srq *tmp;
4361         int state;
4362         u64 in_param;
4363         LIST_HEAD(tlist);
4364         int srqn;
4365         int err;
4366
4367         err = move_all_busy(dev, slave, RES_SRQ);
4368         if (err)
4369                 mlx4_warn(dev, "rem_slave_srqs: Could not move all srqs - too busy for slave %d\n",
4370                           slave);
4371
4372         spin_lock_irq(mlx4_tlock(dev));
4373         list_for_each_entry_safe(srq, tmp, srq_list, com.list) {
4374                 spin_unlock_irq(mlx4_tlock(dev));
4375                 if (srq->com.owner == slave) {
4376                         srqn = srq->com.res_id;
4377                         state = srq->com.from_state;
4378                         while (state != 0) {
4379                                 switch (state) {
4380                                 case RES_SRQ_ALLOCATED:
4381                                         __mlx4_srq_free_icm(dev, srqn);
4382                                         spin_lock_irq(mlx4_tlock(dev));
4383                                         rb_erase(&srq->com.node,
4384                                                  &tracker->res_tree[RES_SRQ]);
4385                                         list_del(&srq->com.list);
4386                                         spin_unlock_irq(mlx4_tlock(dev));
4387                                         mlx4_release_resource(dev, slave,
4388                                                               RES_SRQ, 1, 0);
4389                                         kfree(srq);
4390                                         state = 0;
4391                                         break;
4392
4393                                 case RES_SRQ_HW:
4394                                         in_param = slave;
4395                                         err = mlx4_cmd(dev, in_param, srqn, 1,
4396                                                        MLX4_CMD_HW2SW_SRQ,
4397                                                        MLX4_CMD_TIME_CLASS_A,
4398                                                        MLX4_CMD_NATIVE);
4399                                         if (err)
4400                                                 mlx4_dbg(dev, "rem_slave_srqs: failed to move slave %d srq %d to SW ownership\n",
4401                                                          slave, srqn);
4402
4403                                         atomic_dec(&srq->mtt->ref_count);
4404                                         if (srq->cq)
4405                                                 atomic_dec(&srq->cq->ref_count);
4406                                         state = RES_SRQ_ALLOCATED;
4407                                         break;
4408
4409                                 default:
4410                                         state = 0;
4411                                 }
4412                         }
4413                 }
4414                 spin_lock_irq(mlx4_tlock(dev));
4415         }
4416         spin_unlock_irq(mlx4_tlock(dev));
4417 }
4418
4419 static void rem_slave_cqs(struct mlx4_dev *dev, int slave)
4420 {
4421         struct mlx4_priv *priv = mlx4_priv(dev);
4422         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
4423         struct list_head *cq_list =
4424                 &tracker->slave_list[slave].res_list[RES_CQ];
4425         struct res_cq *cq;
4426         struct res_cq *tmp;
4427         int state;
4428         u64 in_param;
4429         LIST_HEAD(tlist);
4430         int cqn;
4431         int err;
4432
4433         err = move_all_busy(dev, slave, RES_CQ);
4434         if (err)
4435                 mlx4_warn(dev, "rem_slave_cqs: Could not move all cqs - too busy for slave %d\n",
4436                           slave);
4437
4438         spin_lock_irq(mlx4_tlock(dev));
4439         list_for_each_entry_safe(cq, tmp, cq_list, com.list) {
4440                 spin_unlock_irq(mlx4_tlock(dev));
4441                 if (cq->com.owner == slave && !atomic_read(&cq->ref_count)) {
4442                         cqn = cq->com.res_id;
4443                         state = cq->com.from_state;
4444                         while (state != 0) {
4445                                 switch (state) {
4446                                 case RES_CQ_ALLOCATED:
4447                                         __mlx4_cq_free_icm(dev, cqn);
4448                                         spin_lock_irq(mlx4_tlock(dev));
4449                                         rb_erase(&cq->com.node,
4450                                                  &tracker->res_tree[RES_CQ]);
4451                                         list_del(&cq->com.list);
4452                                         spin_unlock_irq(mlx4_tlock(dev));
4453                                         mlx4_release_resource(dev, slave,
4454                                                               RES_CQ, 1, 0);
4455                                         kfree(cq);
4456                                         state = 0;
4457                                         break;
4458
4459                                 case RES_CQ_HW:
4460                                         in_param = slave;
4461                                         err = mlx4_cmd(dev, in_param, cqn, 1,
4462                                                        MLX4_CMD_HW2SW_CQ,
4463                                                        MLX4_CMD_TIME_CLASS_A,
4464                                                        MLX4_CMD_NATIVE);
4465                                         if (err)
4466                                                 mlx4_dbg(dev, "rem_slave_cqs: failed to move slave %d cq %d to SW ownership\n",
4467                                                          slave, cqn);
4468                                         atomic_dec(&cq->mtt->ref_count);
4469                                         state = RES_CQ_ALLOCATED;
4470                                         break;
4471
4472                                 default:
4473                                         state = 0;
4474                                 }
4475                         }
4476                 }
4477                 spin_lock_irq(mlx4_tlock(dev));
4478         }
4479         spin_unlock_irq(mlx4_tlock(dev));
4480 }
4481
4482 static void rem_slave_mrs(struct mlx4_dev *dev, int slave)
4483 {
4484         struct mlx4_priv *priv = mlx4_priv(dev);
4485         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
4486         struct list_head *mpt_list =
4487                 &tracker->slave_list[slave].res_list[RES_MPT];
4488         struct res_mpt *mpt;
4489         struct res_mpt *tmp;
4490         int state;
4491         u64 in_param;
4492         LIST_HEAD(tlist);
4493         int mptn;
4494         int err;
4495
4496         err = move_all_busy(dev, slave, RES_MPT);
4497         if (err)
4498                 mlx4_warn(dev, "rem_slave_mrs: Could not move all mpts - too busy for slave %d\n",
4499                           slave);
4500
4501         spin_lock_irq(mlx4_tlock(dev));
4502         list_for_each_entry_safe(mpt, tmp, mpt_list, com.list) {
4503                 spin_unlock_irq(mlx4_tlock(dev));
4504                 if (mpt->com.owner == slave) {
4505                         mptn = mpt->com.res_id;
4506                         state = mpt->com.from_state;
4507                         while (state != 0) {
4508                                 switch (state) {
4509                                 case RES_MPT_RESERVED:
4510                                         __mlx4_mpt_release(dev, mpt->key);
4511                                         spin_lock_irq(mlx4_tlock(dev));
4512                                         rb_erase(&mpt->com.node,
4513                                                  &tracker->res_tree[RES_MPT]);
4514                                         list_del(&mpt->com.list);
4515                                         spin_unlock_irq(mlx4_tlock(dev));
4516                                         mlx4_release_resource(dev, slave,
4517                                                               RES_MPT, 1, 0);
4518                                         kfree(mpt);
4519                                         state = 0;
4520                                         break;
4521
4522                                 case RES_MPT_MAPPED:
4523                                         __mlx4_mpt_free_icm(dev, mpt->key);
4524                                         state = RES_MPT_RESERVED;
4525                                         break;
4526
4527                                 case RES_MPT_HW:
4528                                         in_param = slave;
4529                                         err = mlx4_cmd(dev, in_param, mptn, 0,
4530                                                      MLX4_CMD_HW2SW_MPT,
4531                                                      MLX4_CMD_TIME_CLASS_A,
4532                                                      MLX4_CMD_NATIVE);
4533                                         if (err)
4534                                                 mlx4_dbg(dev, "rem_slave_mrs: failed to move slave %d mpt %d to SW ownership\n",
4535                                                          slave, mptn);
4536                                         if (mpt->mtt)
4537                                                 atomic_dec(&mpt->mtt->ref_count);
4538                                         state = RES_MPT_MAPPED;
4539                                         break;
4540                                 default:
4541                                         state = 0;
4542                                 }
4543                         }
4544                 }
4545                 spin_lock_irq(mlx4_tlock(dev));
4546         }
4547         spin_unlock_irq(mlx4_tlock(dev));
4548 }
4549
4550 static void rem_slave_mtts(struct mlx4_dev *dev, int slave)
4551 {
4552         struct mlx4_priv *priv = mlx4_priv(dev);
4553         struct mlx4_resource_tracker *tracker =
4554                 &priv->mfunc.master.res_tracker;
4555         struct list_head *mtt_list =
4556                 &tracker->slave_list[slave].res_list[RES_MTT];
4557         struct res_mtt *mtt;
4558         struct res_mtt *tmp;
4559         int state;
4560         LIST_HEAD(tlist);
4561         int base;
4562         int err;
4563
4564         err = move_all_busy(dev, slave, RES_MTT);
4565         if (err)
4566                 mlx4_warn(dev, "rem_slave_mtts: Could not move all mtts  - too busy for slave %d\n",
4567                           slave);
4568
4569         spin_lock_irq(mlx4_tlock(dev));
4570         list_for_each_entry_safe(mtt, tmp, mtt_list, com.list) {
4571                 spin_unlock_irq(mlx4_tlock(dev));
4572                 if (mtt->com.owner == slave) {
4573                         base = mtt->com.res_id;
4574                         state = mtt->com.from_state;
4575                         while (state != 0) {
4576                                 switch (state) {
4577                                 case RES_MTT_ALLOCATED:
4578                                         __mlx4_free_mtt_range(dev, base,
4579                                                               mtt->order);
4580                                         spin_lock_irq(mlx4_tlock(dev));
4581                                         rb_erase(&mtt->com.node,
4582                                                  &tracker->res_tree[RES_MTT]);
4583                                         list_del(&mtt->com.list);
4584                                         spin_unlock_irq(mlx4_tlock(dev));
4585                                         mlx4_release_resource(dev, slave, RES_MTT,
4586                                                               1 << mtt->order, 0);
4587                                         kfree(mtt);
4588                                         state = 0;
4589                                         break;
4590
4591                                 default:
4592                                         state = 0;
4593                                 }
4594                         }
4595                 }
4596                 spin_lock_irq(mlx4_tlock(dev));
4597         }
4598         spin_unlock_irq(mlx4_tlock(dev));
4599 }
4600
4601 static void rem_slave_fs_rule(struct mlx4_dev *dev, int slave)
4602 {
4603         struct mlx4_priv *priv = mlx4_priv(dev);
4604         struct mlx4_resource_tracker *tracker =
4605                 &priv->mfunc.master.res_tracker;
4606         struct list_head *fs_rule_list =
4607                 &tracker->slave_list[slave].res_list[RES_FS_RULE];
4608         struct res_fs_rule *fs_rule;
4609         struct res_fs_rule *tmp;
4610         int state;
4611         u64 base;
4612         int err;
4613
4614         err = move_all_busy(dev, slave, RES_FS_RULE);
4615         if (err)
4616                 mlx4_warn(dev, "rem_slave_fs_rule: Could not move all mtts to busy for slave %d\n",
4617                           slave);
4618
4619         spin_lock_irq(mlx4_tlock(dev));
4620         list_for_each_entry_safe(fs_rule, tmp, fs_rule_list, com.list) {
4621                 spin_unlock_irq(mlx4_tlock(dev));
4622                 if (fs_rule->com.owner == slave) {
4623                         base = fs_rule->com.res_id;
4624                         state = fs_rule->com.from_state;
4625                         while (state != 0) {
4626                                 switch (state) {
4627                                 case RES_FS_RULE_ALLOCATED:
4628                                         /* detach rule */
4629                                         err = mlx4_cmd(dev, base, 0, 0,
4630                                                        MLX4_QP_FLOW_STEERING_DETACH,
4631                                                        MLX4_CMD_TIME_CLASS_A,
4632                                                        MLX4_CMD_NATIVE);
4633
4634                                         spin_lock_irq(mlx4_tlock(dev));
4635                                         rb_erase(&fs_rule->com.node,
4636                                                  &tracker->res_tree[RES_FS_RULE]);
4637                                         list_del(&fs_rule->com.list);
4638                                         spin_unlock_irq(mlx4_tlock(dev));
4639                                         kfree(fs_rule);
4640                                         state = 0;
4641                                         break;
4642
4643                                 default:
4644                                         state = 0;
4645                                 }
4646                         }
4647                 }
4648                 spin_lock_irq(mlx4_tlock(dev));
4649         }
4650         spin_unlock_irq(mlx4_tlock(dev));
4651 }
4652
4653 static void rem_slave_eqs(struct mlx4_dev *dev, int slave)
4654 {
4655         struct mlx4_priv *priv = mlx4_priv(dev);
4656         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
4657         struct list_head *eq_list =
4658                 &tracker->slave_list[slave].res_list[RES_EQ];
4659         struct res_eq *eq;
4660         struct res_eq *tmp;
4661         int err;
4662         int state;
4663         LIST_HEAD(tlist);
4664         int eqn;
4665         struct mlx4_cmd_mailbox *mailbox;
4666
4667         err = move_all_busy(dev, slave, RES_EQ);
4668         if (err)
4669                 mlx4_warn(dev, "rem_slave_eqs: Could not move all eqs - too busy for slave %d\n",
4670                           slave);
4671
4672         spin_lock_irq(mlx4_tlock(dev));
4673         list_for_each_entry_safe(eq, tmp, eq_list, com.list) {
4674                 spin_unlock_irq(mlx4_tlock(dev));
4675                 if (eq->com.owner == slave) {
4676                         eqn = eq->com.res_id;
4677                         state = eq->com.from_state;
4678                         while (state != 0) {
4679                                 switch (state) {
4680                                 case RES_EQ_RESERVED:
4681                                         spin_lock_irq(mlx4_tlock(dev));
4682                                         rb_erase(&eq->com.node,
4683                                                  &tracker->res_tree[RES_EQ]);
4684                                         list_del(&eq->com.list);
4685                                         spin_unlock_irq(mlx4_tlock(dev));
4686                                         kfree(eq);
4687                                         state = 0;
4688                                         break;
4689
4690                                 case RES_EQ_HW:
4691                                         mailbox = mlx4_alloc_cmd_mailbox(dev);
4692                                         if (IS_ERR(mailbox)) {
4693                                                 cond_resched();
4694                                                 continue;
4695                                         }
4696                                         err = mlx4_cmd_box(dev, slave, 0,
4697                                                            eqn & 0xff, 0,
4698                                                            MLX4_CMD_HW2SW_EQ,
4699                                                            MLX4_CMD_TIME_CLASS_A,
4700                                                            MLX4_CMD_NATIVE);
4701                                         if (err)
4702                                                 mlx4_dbg(dev, "rem_slave_eqs: failed to move slave %d eqs %d to SW ownership\n",
4703                                                          slave, eqn);
4704                                         mlx4_free_cmd_mailbox(dev, mailbox);
4705                                         atomic_dec(&eq->mtt->ref_count);
4706                                         state = RES_EQ_RESERVED;
4707                                         break;
4708
4709                                 default:
4710                                         state = 0;
4711                                 }
4712                         }
4713                 }
4714                 spin_lock_irq(mlx4_tlock(dev));
4715         }
4716         spin_unlock_irq(mlx4_tlock(dev));
4717 }
4718
4719 static void rem_slave_counters(struct mlx4_dev *dev, int slave)
4720 {
4721         struct mlx4_priv *priv = mlx4_priv(dev);
4722         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
4723         struct list_head *counter_list =
4724                 &tracker->slave_list[slave].res_list[RES_COUNTER];
4725         struct res_counter *counter;
4726         struct res_counter *tmp;
4727         int err;
4728         int index;
4729
4730         err = move_all_busy(dev, slave, RES_COUNTER);
4731         if (err)
4732                 mlx4_warn(dev, "rem_slave_counters: Could not move all counters - too busy for slave %d\n",
4733                           slave);
4734
4735         spin_lock_irq(mlx4_tlock(dev));
4736         list_for_each_entry_safe(counter, tmp, counter_list, com.list) {
4737                 if (counter->com.owner == slave) {
4738                         index = counter->com.res_id;
4739                         rb_erase(&counter->com.node,
4740                                  &tracker->res_tree[RES_COUNTER]);
4741                         list_del(&counter->com.list);
4742                         kfree(counter);
4743                         __mlx4_counter_free(dev, index);
4744                         mlx4_release_resource(dev, slave, RES_COUNTER, 1, 0);
4745                 }
4746         }
4747         spin_unlock_irq(mlx4_tlock(dev));
4748 }
4749
4750 static void rem_slave_xrcdns(struct mlx4_dev *dev, int slave)
4751 {
4752         struct mlx4_priv *priv = mlx4_priv(dev);
4753         struct mlx4_resource_tracker *tracker = &priv->mfunc.master.res_tracker;
4754         struct list_head *xrcdn_list =
4755                 &tracker->slave_list[slave].res_list[RES_XRCD];
4756         struct res_xrcdn *xrcd;
4757         struct res_xrcdn *tmp;
4758         int err;
4759         int xrcdn;
4760
4761         err = move_all_busy(dev, slave, RES_XRCD);
4762         if (err)
4763                 mlx4_warn(dev, "rem_slave_xrcdns: Could not move all xrcdns - too busy for slave %d\n",
4764                           slave);
4765
4766         spin_lock_irq(mlx4_tlock(dev));
4767         list_for_each_entry_safe(xrcd, tmp, xrcdn_list, com.list) {
4768                 if (xrcd->com.owner == slave) {
4769                         xrcdn = xrcd->com.res_id;
4770                         rb_erase(&xrcd->com.node, &tracker->res_tree[RES_XRCD]);
4771                         list_del(&xrcd->com.list);
4772                         kfree(xrcd);
4773                         __mlx4_xrcd_free(dev, xrcdn);
4774                 }
4775         }
4776         spin_unlock_irq(mlx4_tlock(dev));
4777 }
4778
4779 void mlx4_delete_all_resources_for_slave(struct mlx4_dev *dev, int slave)
4780 {
4781         struct mlx4_priv *priv = mlx4_priv(dev);
4782         mlx4_reset_roce_gids(dev, slave);
4783         mutex_lock(&priv->mfunc.master.res_tracker.slave_list[slave].mutex);
4784         rem_slave_vlans(dev, slave);
4785         rem_slave_macs(dev, slave);
4786         rem_slave_fs_rule(dev, slave);
4787         rem_slave_qps(dev, slave);
4788         rem_slave_srqs(dev, slave);
4789         rem_slave_cqs(dev, slave);
4790         rem_slave_mrs(dev, slave);
4791         rem_slave_eqs(dev, slave);
4792         rem_slave_mtts(dev, slave);
4793         rem_slave_counters(dev, slave);
4794         rem_slave_xrcdns(dev, slave);
4795         mutex_unlock(&priv->mfunc.master.res_tracker.slave_list[slave].mutex);
4796 }
4797
4798 void mlx4_vf_immed_vlan_work_handler(struct work_struct *_work)
4799 {
4800         struct mlx4_vf_immed_vlan_work *work =
4801                 container_of(_work, struct mlx4_vf_immed_vlan_work, work);
4802         struct mlx4_cmd_mailbox *mailbox;
4803         struct mlx4_update_qp_context *upd_context;
4804         struct mlx4_dev *dev = &work->priv->dev;
4805         struct mlx4_resource_tracker *tracker =
4806                 &work->priv->mfunc.master.res_tracker;
4807         struct list_head *qp_list =
4808                 &tracker->slave_list[work->slave].res_list[RES_QP];
4809         struct res_qp *qp;
4810         struct res_qp *tmp;
4811         u64 qp_path_mask_vlan_ctrl =
4812                        ((1ULL << MLX4_UPD_QP_PATH_MASK_ETH_TX_BLOCK_UNTAGGED) |
4813                        (1ULL << MLX4_UPD_QP_PATH_MASK_ETH_TX_BLOCK_1P) |
4814                        (1ULL << MLX4_UPD_QP_PATH_MASK_ETH_TX_BLOCK_TAGGED) |
4815                        (1ULL << MLX4_UPD_QP_PATH_MASK_ETH_RX_BLOCK_UNTAGGED) |
4816                        (1ULL << MLX4_UPD_QP_PATH_MASK_ETH_RX_BLOCK_1P) |
4817                        (1ULL << MLX4_UPD_QP_PATH_MASK_ETH_RX_BLOCK_TAGGED));
4818
4819         u64 qp_path_mask = ((1ULL << MLX4_UPD_QP_PATH_MASK_VLAN_INDEX) |
4820                        (1ULL << MLX4_UPD_QP_PATH_MASK_FVL) |
4821                        (1ULL << MLX4_UPD_QP_PATH_MASK_CV) |
4822                        (1ULL << MLX4_UPD_QP_PATH_MASK_ETH_HIDE_CQE_VLAN) |
4823                        (1ULL << MLX4_UPD_QP_PATH_MASK_FEUP) |
4824                        (1ULL << MLX4_UPD_QP_PATH_MASK_FVL_RX) |
4825                        (1ULL << MLX4_UPD_QP_PATH_MASK_SCHED_QUEUE));
4826
4827         int err;
4828         int port, errors = 0;
4829         u8 vlan_control;
4830
4831         if (mlx4_is_slave(dev)) {
4832                 mlx4_warn(dev, "Trying to update-qp in slave %d\n",
4833                           work->slave);
4834                 goto out;
4835         }
4836
4837         mailbox = mlx4_alloc_cmd_mailbox(dev);
4838         if (IS_ERR(mailbox))
4839                 goto out;
4840         if (work->flags & MLX4_VF_IMMED_VLAN_FLAG_LINK_DISABLE) /* block all */
4841                 vlan_control = MLX4_VLAN_CTRL_ETH_TX_BLOCK_TAGGED |
4842                         MLX4_VLAN_CTRL_ETH_TX_BLOCK_PRIO_TAGGED |
4843                         MLX4_VLAN_CTRL_ETH_TX_BLOCK_UNTAGGED |
4844                         MLX4_VLAN_CTRL_ETH_RX_BLOCK_PRIO_TAGGED |
4845                         MLX4_VLAN_CTRL_ETH_RX_BLOCK_UNTAGGED |
4846                         MLX4_VLAN_CTRL_ETH_RX_BLOCK_TAGGED;
4847         else if (!work->vlan_id)
4848                 vlan_control = MLX4_VLAN_CTRL_ETH_TX_BLOCK_TAGGED |
4849                         MLX4_VLAN_CTRL_ETH_RX_BLOCK_TAGGED;
4850         else
4851                 vlan_control = MLX4_VLAN_CTRL_ETH_TX_BLOCK_TAGGED |
4852                         MLX4_VLAN_CTRL_ETH_RX_BLOCK_PRIO_TAGGED |
4853                         MLX4_VLAN_CTRL_ETH_RX_BLOCK_UNTAGGED;
4854
4855         upd_context = mailbox->buf;
4856         upd_context->qp_mask = cpu_to_be64(1ULL << MLX4_UPD_QP_MASK_VSD);
4857
4858         spin_lock_irq(mlx4_tlock(dev));
4859         list_for_each_entry_safe(qp, tmp, qp_list, com.list) {
4860                 spin_unlock_irq(mlx4_tlock(dev));
4861                 if (qp->com.owner == work->slave) {
4862                         if (qp->com.from_state != RES_QP_HW ||
4863                             !qp->sched_queue ||  /* no INIT2RTR trans yet */
4864                             mlx4_is_qp_reserved(dev, qp->local_qpn) ||
4865                             qp->qpc_flags & (1 << MLX4_RSS_QPC_FLAG_OFFSET)) {
4866                                 spin_lock_irq(mlx4_tlock(dev));
4867                                 continue;
4868                         }
4869                         port = (qp->sched_queue >> 6 & 1) + 1;
4870                         if (port != work->port) {
4871                                 spin_lock_irq(mlx4_tlock(dev));
4872                                 continue;
4873                         }
4874                         if (MLX4_QP_ST_RC == ((qp->qpc_flags >> 16) & 0xff))
4875                                 upd_context->primary_addr_path_mask = cpu_to_be64(qp_path_mask);
4876                         else
4877                                 upd_context->primary_addr_path_mask =
4878                                         cpu_to_be64(qp_path_mask | qp_path_mask_vlan_ctrl);
4879                         if (work->vlan_id == MLX4_VGT) {
4880                                 upd_context->qp_context.param3 = qp->param3;
4881                                 upd_context->qp_context.pri_path.vlan_control = qp->vlan_control;
4882                                 upd_context->qp_context.pri_path.fvl_rx = qp->fvl_rx;
4883                                 upd_context->qp_context.pri_path.vlan_index = qp->vlan_index;
4884                                 upd_context->qp_context.pri_path.fl = qp->pri_path_fl;
4885                                 upd_context->qp_context.pri_path.feup = qp->feup;
4886                                 upd_context->qp_context.pri_path.sched_queue =
4887                                         qp->sched_queue;
4888                         } else {
4889                                 upd_context->qp_context.param3 = qp->param3 & ~cpu_to_be32(MLX4_STRIP_VLAN);
4890                                 upd_context->qp_context.pri_path.vlan_control = vlan_control;
4891                                 upd_context->qp_context.pri_path.vlan_index = work->vlan_ix;
4892                                 upd_context->qp_context.pri_path.fvl_rx =
4893                                         qp->fvl_rx | MLX4_FVL_RX_FORCE_ETH_VLAN;
4894                                 upd_context->qp_context.pri_path.fl =
4895                                         qp->pri_path_fl | MLX4_FL_CV | MLX4_FL_ETH_HIDE_CQE_VLAN;
4896                                 upd_context->qp_context.pri_path.feup =
4897                                         qp->feup | MLX4_FEUP_FORCE_ETH_UP | MLX4_FVL_FORCE_ETH_VLAN;
4898                                 upd_context->qp_context.pri_path.sched_queue =
4899                                         qp->sched_queue & 0xC7;
4900                                 upd_context->qp_context.pri_path.sched_queue |=
4901                                         ((work->qos & 0x7) << 3);
4902                         }
4903
4904                         err = mlx4_cmd(dev, mailbox->dma,
4905                                        qp->local_qpn & 0xffffff,
4906                                        0, MLX4_CMD_UPDATE_QP,
4907                                        MLX4_CMD_TIME_CLASS_C, MLX4_CMD_NATIVE);
4908                         if (err) {
4909                                 mlx4_info(dev, "UPDATE_QP failed for slave %d, port %d, qpn %d (%d)\n",
4910                                           work->slave, port, qp->local_qpn, err);
4911                                 errors++;
4912                         }
4913                 }
4914                 spin_lock_irq(mlx4_tlock(dev));
4915         }
4916         spin_unlock_irq(mlx4_tlock(dev));
4917         mlx4_free_cmd_mailbox(dev, mailbox);
4918
4919         if (errors)
4920                 mlx4_err(dev, "%d UPDATE_QP failures for slave %d, port %d\n",
4921                          errors, work->slave, work->port);
4922
4923         /* unregister previous vlan_id if needed and we had no errors
4924          * while updating the QPs
4925          */
4926         if (work->flags & MLX4_VF_IMMED_VLAN_FLAG_VLAN && !errors &&
4927             NO_INDX != work->orig_vlan_ix)
4928                 __mlx4_unregister_vlan(&work->priv->dev, work->port,
4929                                        work->orig_vlan_id);
4930 out:
4931         kfree(work);
4932         return;
4933 }