xref: /openbmc/linux/drivers/infiniband/hw/mlx5/main.c (revision 8795a739)
1 /*
2  * Copyright (c) 2013-2015, Mellanox Technologies. All rights reserved.
3  *
4  * This software is available to you under a choice of one of two
5  * licenses.  You may choose to be licensed under the terms of the GNU
6  * General Public License (GPL) Version 2, available from the file
7  * COPYING in the main directory of this source tree, or the
8  * OpenIB.org BSD license below:
9  *
10  *     Redistribution and use in source and binary forms, with or
11  *     without modification, are permitted provided that the following
12  *     conditions are met:
13  *
14  *      - Redistributions of source code must retain the above
15  *        copyright notice, this list of conditions and the following
16  *        disclaimer.
17  *
18  *      - Redistributions in binary form must reproduce the above
19  *        copyright notice, this list of conditions and the following
20  *        disclaimer in the documentation and/or other materials
21  *        provided with the distribution.
22  *
23  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
24  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
25  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
26  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
27  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
28  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
29  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
30  * SOFTWARE.
31  */
32 
33 #include <linux/debugfs.h>
34 #include <linux/highmem.h>
35 #include <linux/module.h>
36 #include <linux/init.h>
37 #include <linux/errno.h>
38 #include <linux/pci.h>
39 #include <linux/dma-mapping.h>
40 #include <linux/slab.h>
41 #include <linux/bitmap.h>
42 #if defined(CONFIG_X86)
43 #include <asm/pat.h>
44 #endif
45 #include <linux/sched.h>
46 #include <linux/sched/mm.h>
47 #include <linux/sched/task.h>
48 #include <linux/delay.h>
49 #include <rdma/ib_user_verbs.h>
50 #include <rdma/ib_addr.h>
51 #include <rdma/ib_cache.h>
52 #include <linux/mlx5/port.h>
53 #include <linux/mlx5/vport.h>
54 #include <linux/mlx5/fs.h>
55 #include <linux/mlx5/eswitch.h>
56 #include <linux/list.h>
57 #include <rdma/ib_smi.h>
58 #include <rdma/ib_umem.h>
59 #include <linux/in.h>
60 #include <linux/etherdevice.h>
61 #include "mlx5_ib.h"
62 #include "ib_rep.h"
63 #include "cmd.h"
64 #include "srq.h"
65 #include <linux/mlx5/fs_helpers.h>
66 #include <linux/mlx5/accel.h>
67 #include <rdma/uverbs_std_types.h>
68 #include <rdma/mlx5_user_ioctl_verbs.h>
69 #include <rdma/mlx5_user_ioctl_cmds.h>
70 
71 #define UVERBS_MODULE_NAME mlx5_ib
72 #include <rdma/uverbs_named_ioctl.h>
73 
74 #define DRIVER_NAME "mlx5_ib"
75 #define DRIVER_VERSION "5.0-0"
76 
77 MODULE_AUTHOR("Eli Cohen <eli@mellanox.com>");
78 MODULE_DESCRIPTION("Mellanox Connect-IB HCA IB driver");
79 MODULE_LICENSE("Dual BSD/GPL");
80 
81 static char mlx5_version[] =
82 	DRIVER_NAME ": Mellanox Connect-IB Infiniband driver v"
83 	DRIVER_VERSION "\n";
84 
85 struct mlx5_ib_event_work {
86 	struct work_struct	work;
87 	union {
88 		struct mlx5_ib_dev	      *dev;
89 		struct mlx5_ib_multiport_info *mpi;
90 	};
91 	bool			is_slave;
92 	unsigned int		event;
93 	void			*param;
94 };
95 
96 enum {
97 	MLX5_ATOMIC_SIZE_QP_8BYTES = 1 << 3,
98 };
99 
100 static struct workqueue_struct *mlx5_ib_event_wq;
101 static LIST_HEAD(mlx5_ib_unaffiliated_port_list);
102 static LIST_HEAD(mlx5_ib_dev_list);
103 /*
104  * This mutex should be held when accessing either of the above lists
105  */
106 static DEFINE_MUTEX(mlx5_ib_multiport_mutex);
107 
108 /* We can't use an array for xlt_emergency_page because dma_map_single
109  * doesn't work on kernel modules memory
110  */
111 static unsigned long xlt_emergency_page;
112 static struct mutex xlt_emergency_page_mutex;
113 
114 struct mlx5_ib_dev *mlx5_ib_get_ibdev_from_mpi(struct mlx5_ib_multiport_info *mpi)
115 {
116 	struct mlx5_ib_dev *dev;
117 
118 	mutex_lock(&mlx5_ib_multiport_mutex);
119 	dev = mpi->ibdev;
120 	mutex_unlock(&mlx5_ib_multiport_mutex);
121 	return dev;
122 }
123 
124 static enum rdma_link_layer
125 mlx5_port_type_cap_to_rdma_ll(int port_type_cap)
126 {
127 	switch (port_type_cap) {
128 	case MLX5_CAP_PORT_TYPE_IB:
129 		return IB_LINK_LAYER_INFINIBAND;
130 	case MLX5_CAP_PORT_TYPE_ETH:
131 		return IB_LINK_LAYER_ETHERNET;
132 	default:
133 		return IB_LINK_LAYER_UNSPECIFIED;
134 	}
135 }
136 
137 static enum rdma_link_layer
138 mlx5_ib_port_link_layer(struct ib_device *device, u8 port_num)
139 {
140 	struct mlx5_ib_dev *dev = to_mdev(device);
141 	int port_type_cap = MLX5_CAP_GEN(dev->mdev, port_type);
142 
143 	return mlx5_port_type_cap_to_rdma_ll(port_type_cap);
144 }
145 
146 static int get_port_state(struct ib_device *ibdev,
147 			  u8 port_num,
148 			  enum ib_port_state *state)
149 {
150 	struct ib_port_attr attr;
151 	int ret;
152 
153 	memset(&attr, 0, sizeof(attr));
154 	ret = ibdev->ops.query_port(ibdev, port_num, &attr);
155 	if (!ret)
156 		*state = attr.state;
157 	return ret;
158 }
159 
160 static struct mlx5_roce *mlx5_get_rep_roce(struct mlx5_ib_dev *dev,
161 					   struct net_device *ndev,
162 					   u8 *port_num)
163 {
164 	struct mlx5_eswitch *esw = dev->mdev->priv.eswitch;
165 	struct net_device *rep_ndev;
166 	struct mlx5_ib_port *port;
167 	int i;
168 
169 	for (i = 0; i < dev->num_ports; i++) {
170 		port  = &dev->port[i];
171 		if (!port->rep)
172 			continue;
173 
174 		read_lock(&port->roce.netdev_lock);
175 		rep_ndev = mlx5_ib_get_rep_netdev(esw,
176 						  port->rep->vport);
177 		if (rep_ndev == ndev) {
178 			read_unlock(&port->roce.netdev_lock);
179 			*port_num = i + 1;
180 			return &port->roce;
181 		}
182 		read_unlock(&port->roce.netdev_lock);
183 	}
184 
185 	return NULL;
186 }
187 
188 static int mlx5_netdev_event(struct notifier_block *this,
189 			     unsigned long event, void *ptr)
190 {
191 	struct mlx5_roce *roce = container_of(this, struct mlx5_roce, nb);
192 	struct net_device *ndev = netdev_notifier_info_to_dev(ptr);
193 	u8 port_num = roce->native_port_num;
194 	struct mlx5_core_dev *mdev;
195 	struct mlx5_ib_dev *ibdev;
196 
197 	ibdev = roce->dev;
198 	mdev = mlx5_ib_get_native_port_mdev(ibdev, port_num, NULL);
199 	if (!mdev)
200 		return NOTIFY_DONE;
201 
202 	switch (event) {
203 	case NETDEV_REGISTER:
204 		/* Should already be registered during the load */
205 		if (ibdev->is_rep)
206 			break;
207 		write_lock(&roce->netdev_lock);
208 		if (ndev->dev.parent == mdev->device)
209 			roce->netdev = ndev;
210 		write_unlock(&roce->netdev_lock);
211 		break;
212 
213 	case NETDEV_UNREGISTER:
214 		/* In case of reps, ib device goes away before the netdevs */
215 		write_lock(&roce->netdev_lock);
216 		if (roce->netdev == ndev)
217 			roce->netdev = NULL;
218 		write_unlock(&roce->netdev_lock);
219 		break;
220 
221 	case NETDEV_CHANGE:
222 	case NETDEV_UP:
223 	case NETDEV_DOWN: {
224 		struct net_device *lag_ndev = mlx5_lag_get_roce_netdev(mdev);
225 		struct net_device *upper = NULL;
226 
227 		if (lag_ndev) {
228 			upper = netdev_master_upper_dev_get(lag_ndev);
229 			dev_put(lag_ndev);
230 		}
231 
232 		if (ibdev->is_rep)
233 			roce = mlx5_get_rep_roce(ibdev, ndev, &port_num);
234 		if (!roce)
235 			return NOTIFY_DONE;
236 		if ((upper == ndev || (!upper && ndev == roce->netdev))
237 		    && ibdev->ib_active) {
238 			struct ib_event ibev = { };
239 			enum ib_port_state port_state;
240 
241 			if (get_port_state(&ibdev->ib_dev, port_num,
242 					   &port_state))
243 				goto done;
244 
245 			if (roce->last_port_state == port_state)
246 				goto done;
247 
248 			roce->last_port_state = port_state;
249 			ibev.device = &ibdev->ib_dev;
250 			if (port_state == IB_PORT_DOWN)
251 				ibev.event = IB_EVENT_PORT_ERR;
252 			else if (port_state == IB_PORT_ACTIVE)
253 				ibev.event = IB_EVENT_PORT_ACTIVE;
254 			else
255 				goto done;
256 
257 			ibev.element.port_num = port_num;
258 			ib_dispatch_event(&ibev);
259 		}
260 		break;
261 	}
262 
263 	default:
264 		break;
265 	}
266 done:
267 	mlx5_ib_put_native_port_mdev(ibdev, port_num);
268 	return NOTIFY_DONE;
269 }
270 
271 static struct net_device *mlx5_ib_get_netdev(struct ib_device *device,
272 					     u8 port_num)
273 {
274 	struct mlx5_ib_dev *ibdev = to_mdev(device);
275 	struct net_device *ndev;
276 	struct mlx5_core_dev *mdev;
277 
278 	mdev = mlx5_ib_get_native_port_mdev(ibdev, port_num, NULL);
279 	if (!mdev)
280 		return NULL;
281 
282 	ndev = mlx5_lag_get_roce_netdev(mdev);
283 	if (ndev)
284 		goto out;
285 
286 	/* Ensure ndev does not disappear before we invoke dev_hold()
287 	 */
288 	read_lock(&ibdev->port[port_num - 1].roce.netdev_lock);
289 	ndev = ibdev->port[port_num - 1].roce.netdev;
290 	if (ndev)
291 		dev_hold(ndev);
292 	read_unlock(&ibdev->port[port_num - 1].roce.netdev_lock);
293 
294 out:
295 	mlx5_ib_put_native_port_mdev(ibdev, port_num);
296 	return ndev;
297 }
298 
299 struct mlx5_core_dev *mlx5_ib_get_native_port_mdev(struct mlx5_ib_dev *ibdev,
300 						   u8 ib_port_num,
301 						   u8 *native_port_num)
302 {
303 	enum rdma_link_layer ll = mlx5_ib_port_link_layer(&ibdev->ib_dev,
304 							  ib_port_num);
305 	struct mlx5_core_dev *mdev = NULL;
306 	struct mlx5_ib_multiport_info *mpi;
307 	struct mlx5_ib_port *port;
308 
309 	if (!mlx5_core_mp_enabled(ibdev->mdev) ||
310 	    ll != IB_LINK_LAYER_ETHERNET) {
311 		if (native_port_num)
312 			*native_port_num = ib_port_num;
313 		return ibdev->mdev;
314 	}
315 
316 	if (native_port_num)
317 		*native_port_num = 1;
318 
319 	port = &ibdev->port[ib_port_num - 1];
320 	if (!port)
321 		return NULL;
322 
323 	spin_lock(&port->mp.mpi_lock);
324 	mpi = ibdev->port[ib_port_num - 1].mp.mpi;
325 	if (mpi && !mpi->unaffiliate) {
326 		mdev = mpi->mdev;
327 		/* If it's the master no need to refcount, it'll exist
328 		 * as long as the ib_dev exists.
329 		 */
330 		if (!mpi->is_master)
331 			mpi->mdev_refcnt++;
332 	}
333 	spin_unlock(&port->mp.mpi_lock);
334 
335 	return mdev;
336 }
337 
338 void mlx5_ib_put_native_port_mdev(struct mlx5_ib_dev *ibdev, u8 port_num)
339 {
340 	enum rdma_link_layer ll = mlx5_ib_port_link_layer(&ibdev->ib_dev,
341 							  port_num);
342 	struct mlx5_ib_multiport_info *mpi;
343 	struct mlx5_ib_port *port;
344 
345 	if (!mlx5_core_mp_enabled(ibdev->mdev) || ll != IB_LINK_LAYER_ETHERNET)
346 		return;
347 
348 	port = &ibdev->port[port_num - 1];
349 
350 	spin_lock(&port->mp.mpi_lock);
351 	mpi = ibdev->port[port_num - 1].mp.mpi;
352 	if (mpi->is_master)
353 		goto out;
354 
355 	mpi->mdev_refcnt--;
356 	if (mpi->unaffiliate)
357 		complete(&mpi->unref_comp);
358 out:
359 	spin_unlock(&port->mp.mpi_lock);
360 }
361 
362 static int translate_eth_legacy_proto_oper(u32 eth_proto_oper, u8 *active_speed,
363 					   u8 *active_width)
364 {
365 	switch (eth_proto_oper) {
366 	case MLX5E_PROT_MASK(MLX5E_1000BASE_CX_SGMII):
367 	case MLX5E_PROT_MASK(MLX5E_1000BASE_KX):
368 	case MLX5E_PROT_MASK(MLX5E_100BASE_TX):
369 	case MLX5E_PROT_MASK(MLX5E_1000BASE_T):
370 		*active_width = IB_WIDTH_1X;
371 		*active_speed = IB_SPEED_SDR;
372 		break;
373 	case MLX5E_PROT_MASK(MLX5E_10GBASE_T):
374 	case MLX5E_PROT_MASK(MLX5E_10GBASE_CX4):
375 	case MLX5E_PROT_MASK(MLX5E_10GBASE_KX4):
376 	case MLX5E_PROT_MASK(MLX5E_10GBASE_KR):
377 	case MLX5E_PROT_MASK(MLX5E_10GBASE_CR):
378 	case MLX5E_PROT_MASK(MLX5E_10GBASE_SR):
379 	case MLX5E_PROT_MASK(MLX5E_10GBASE_ER):
380 		*active_width = IB_WIDTH_1X;
381 		*active_speed = IB_SPEED_QDR;
382 		break;
383 	case MLX5E_PROT_MASK(MLX5E_25GBASE_CR):
384 	case MLX5E_PROT_MASK(MLX5E_25GBASE_KR):
385 	case MLX5E_PROT_MASK(MLX5E_25GBASE_SR):
386 		*active_width = IB_WIDTH_1X;
387 		*active_speed = IB_SPEED_EDR;
388 		break;
389 	case MLX5E_PROT_MASK(MLX5E_40GBASE_CR4):
390 	case MLX5E_PROT_MASK(MLX5E_40GBASE_KR4):
391 	case MLX5E_PROT_MASK(MLX5E_40GBASE_SR4):
392 	case MLX5E_PROT_MASK(MLX5E_40GBASE_LR4):
393 		*active_width = IB_WIDTH_4X;
394 		*active_speed = IB_SPEED_QDR;
395 		break;
396 	case MLX5E_PROT_MASK(MLX5E_50GBASE_CR2):
397 	case MLX5E_PROT_MASK(MLX5E_50GBASE_KR2):
398 	case MLX5E_PROT_MASK(MLX5E_50GBASE_SR2):
399 		*active_width = IB_WIDTH_1X;
400 		*active_speed = IB_SPEED_HDR;
401 		break;
402 	case MLX5E_PROT_MASK(MLX5E_56GBASE_R4):
403 		*active_width = IB_WIDTH_4X;
404 		*active_speed = IB_SPEED_FDR;
405 		break;
406 	case MLX5E_PROT_MASK(MLX5E_100GBASE_CR4):
407 	case MLX5E_PROT_MASK(MLX5E_100GBASE_SR4):
408 	case MLX5E_PROT_MASK(MLX5E_100GBASE_KR4):
409 	case MLX5E_PROT_MASK(MLX5E_100GBASE_LR4):
410 		*active_width = IB_WIDTH_4X;
411 		*active_speed = IB_SPEED_EDR;
412 		break;
413 	default:
414 		return -EINVAL;
415 	}
416 
417 	return 0;
418 }
419 
420 static int translate_eth_ext_proto_oper(u32 eth_proto_oper, u8 *active_speed,
421 					u8 *active_width)
422 {
423 	switch (eth_proto_oper) {
424 	case MLX5E_PROT_MASK(MLX5E_SGMII_100M):
425 	case MLX5E_PROT_MASK(MLX5E_1000BASE_X_SGMII):
426 		*active_width = IB_WIDTH_1X;
427 		*active_speed = IB_SPEED_SDR;
428 		break;
429 	case MLX5E_PROT_MASK(MLX5E_5GBASE_R):
430 		*active_width = IB_WIDTH_1X;
431 		*active_speed = IB_SPEED_DDR;
432 		break;
433 	case MLX5E_PROT_MASK(MLX5E_10GBASE_XFI_XAUI_1):
434 		*active_width = IB_WIDTH_1X;
435 		*active_speed = IB_SPEED_QDR;
436 		break;
437 	case MLX5E_PROT_MASK(MLX5E_40GBASE_XLAUI_4_XLPPI_4):
438 		*active_width = IB_WIDTH_4X;
439 		*active_speed = IB_SPEED_QDR;
440 		break;
441 	case MLX5E_PROT_MASK(MLX5E_25GAUI_1_25GBASE_CR_KR):
442 		*active_width = IB_WIDTH_1X;
443 		*active_speed = IB_SPEED_EDR;
444 		break;
445 	case MLX5E_PROT_MASK(MLX5E_50GAUI_2_LAUI_2_50GBASE_CR2_KR2):
446 		*active_width = IB_WIDTH_2X;
447 		*active_speed = IB_SPEED_EDR;
448 		break;
449 	case MLX5E_PROT_MASK(MLX5E_50GAUI_1_LAUI_1_50GBASE_CR_KR):
450 		*active_width = IB_WIDTH_1X;
451 		*active_speed = IB_SPEED_HDR;
452 		break;
453 	case MLX5E_PROT_MASK(MLX5E_CAUI_4_100GBASE_CR4_KR4):
454 		*active_width = IB_WIDTH_4X;
455 		*active_speed = IB_SPEED_EDR;
456 		break;
457 	case MLX5E_PROT_MASK(MLX5E_100GAUI_2_100GBASE_CR2_KR2):
458 		*active_width = IB_WIDTH_2X;
459 		*active_speed = IB_SPEED_HDR;
460 		break;
461 	case MLX5E_PROT_MASK(MLX5E_200GAUI_4_200GBASE_CR4_KR4):
462 		*active_width = IB_WIDTH_4X;
463 		*active_speed = IB_SPEED_HDR;
464 		break;
465 	default:
466 		return -EINVAL;
467 	}
468 
469 	return 0;
470 }
471 
472 static int translate_eth_proto_oper(u32 eth_proto_oper, u8 *active_speed,
473 				    u8 *active_width, bool ext)
474 {
475 	return ext ?
476 		translate_eth_ext_proto_oper(eth_proto_oper, active_speed,
477 					     active_width) :
478 		translate_eth_legacy_proto_oper(eth_proto_oper, active_speed,
479 						active_width);
480 }
481 
482 static int mlx5_query_port_roce(struct ib_device *device, u8 port_num,
483 				struct ib_port_attr *props)
484 {
485 	struct mlx5_ib_dev *dev = to_mdev(device);
486 	u32 out[MLX5_ST_SZ_DW(ptys_reg)] = {0};
487 	struct mlx5_core_dev *mdev;
488 	struct net_device *ndev, *upper;
489 	enum ib_mtu ndev_ib_mtu;
490 	bool put_mdev = true;
491 	u16 qkey_viol_cntr;
492 	u32 eth_prot_oper;
493 	u8 mdev_port_num;
494 	bool ext;
495 	int err;
496 
497 	mdev = mlx5_ib_get_native_port_mdev(dev, port_num, &mdev_port_num);
498 	if (!mdev) {
499 		/* This means the port isn't affiliated yet. Get the
500 		 * info for the master port instead.
501 		 */
502 		put_mdev = false;
503 		mdev = dev->mdev;
504 		mdev_port_num = 1;
505 		port_num = 1;
506 	}
507 
508 	/* Possible bad flows are checked before filling out props so in case
509 	 * of an error it will still be zeroed out.
510 	 * Use native port in case of reps
511 	 */
512 	if (dev->is_rep)
513 		err = mlx5_query_port_ptys(mdev, out, sizeof(out), MLX5_PTYS_EN,
514 					   1);
515 	else
516 		err = mlx5_query_port_ptys(mdev, out, sizeof(out), MLX5_PTYS_EN,
517 					   mdev_port_num);
518 	if (err)
519 		goto out;
520 	ext = MLX5_CAP_PCAM_FEATURE(dev->mdev, ptys_extended_ethernet);
521 	eth_prot_oper = MLX5_GET_ETH_PROTO(ptys_reg, out, ext, eth_proto_oper);
522 
523 	props->active_width     = IB_WIDTH_4X;
524 	props->active_speed     = IB_SPEED_QDR;
525 
526 	translate_eth_proto_oper(eth_prot_oper, &props->active_speed,
527 				 &props->active_width, ext);
528 
529 	props->port_cap_flags |= IB_PORT_CM_SUP;
530 	props->ip_gids = true;
531 
532 	props->gid_tbl_len      = MLX5_CAP_ROCE(dev->mdev,
533 						roce_address_table_size);
534 	props->max_mtu          = IB_MTU_4096;
535 	props->max_msg_sz       = 1 << MLX5_CAP_GEN(dev->mdev, log_max_msg);
536 	props->pkey_tbl_len     = 1;
537 	props->state            = IB_PORT_DOWN;
538 	props->phys_state       = IB_PORT_PHYS_STATE_DISABLED;
539 
540 	mlx5_query_nic_vport_qkey_viol_cntr(mdev, &qkey_viol_cntr);
541 	props->qkey_viol_cntr = qkey_viol_cntr;
542 
543 	/* If this is a stub query for an unaffiliated port stop here */
544 	if (!put_mdev)
545 		goto out;
546 
547 	ndev = mlx5_ib_get_netdev(device, port_num);
548 	if (!ndev)
549 		goto out;
550 
551 	if (dev->lag_active) {
552 		rcu_read_lock();
553 		upper = netdev_master_upper_dev_get_rcu(ndev);
554 		if (upper) {
555 			dev_put(ndev);
556 			ndev = upper;
557 			dev_hold(ndev);
558 		}
559 		rcu_read_unlock();
560 	}
561 
562 	if (netif_running(ndev) && netif_carrier_ok(ndev)) {
563 		props->state      = IB_PORT_ACTIVE;
564 		props->phys_state = IB_PORT_PHYS_STATE_LINK_UP;
565 	}
566 
567 	ndev_ib_mtu = iboe_get_mtu(ndev->mtu);
568 
569 	dev_put(ndev);
570 
571 	props->active_mtu	= min(props->max_mtu, ndev_ib_mtu);
572 out:
573 	if (put_mdev)
574 		mlx5_ib_put_native_port_mdev(dev, port_num);
575 	return err;
576 }
577 
578 static int set_roce_addr(struct mlx5_ib_dev *dev, u8 port_num,
579 			 unsigned int index, const union ib_gid *gid,
580 			 const struct ib_gid_attr *attr)
581 {
582 	enum ib_gid_type gid_type = IB_GID_TYPE_IB;
583 	u16 vlan_id = 0xffff;
584 	u8 roce_version = 0;
585 	u8 roce_l3_type = 0;
586 	u8 mac[ETH_ALEN];
587 	int ret;
588 
589 	if (gid) {
590 		gid_type = attr->gid_type;
591 		ret = rdma_read_gid_l2_fields(attr, &vlan_id, &mac[0]);
592 		if (ret)
593 			return ret;
594 	}
595 
596 	switch (gid_type) {
597 	case IB_GID_TYPE_IB:
598 		roce_version = MLX5_ROCE_VERSION_1;
599 		break;
600 	case IB_GID_TYPE_ROCE_UDP_ENCAP:
601 		roce_version = MLX5_ROCE_VERSION_2;
602 		if (ipv6_addr_v4mapped((void *)gid))
603 			roce_l3_type = MLX5_ROCE_L3_TYPE_IPV4;
604 		else
605 			roce_l3_type = MLX5_ROCE_L3_TYPE_IPV6;
606 		break;
607 
608 	default:
609 		mlx5_ib_warn(dev, "Unexpected GID type %u\n", gid_type);
610 	}
611 
612 	return mlx5_core_roce_gid_set(dev->mdev, index, roce_version,
613 				      roce_l3_type, gid->raw, mac,
614 				      vlan_id < VLAN_CFI_MASK, vlan_id,
615 				      port_num);
616 }
617 
618 static int mlx5_ib_add_gid(const struct ib_gid_attr *attr,
619 			   __always_unused void **context)
620 {
621 	return set_roce_addr(to_mdev(attr->device), attr->port_num,
622 			     attr->index, &attr->gid, attr);
623 }
624 
625 static int mlx5_ib_del_gid(const struct ib_gid_attr *attr,
626 			   __always_unused void **context)
627 {
628 	return set_roce_addr(to_mdev(attr->device), attr->port_num,
629 			     attr->index, NULL, NULL);
630 }
631 
632 __be16 mlx5_get_roce_udp_sport(struct mlx5_ib_dev *dev,
633 			       const struct ib_gid_attr *attr)
634 {
635 	if (attr->gid_type != IB_GID_TYPE_ROCE_UDP_ENCAP)
636 		return 0;
637 
638 	return cpu_to_be16(MLX5_CAP_ROCE(dev->mdev, r_roce_min_src_udp_port));
639 }
640 
641 static int mlx5_use_mad_ifc(struct mlx5_ib_dev *dev)
642 {
643 	if (MLX5_CAP_GEN(dev->mdev, port_type) == MLX5_CAP_PORT_TYPE_IB)
644 		return !MLX5_CAP_GEN(dev->mdev, ib_virt);
645 	return 0;
646 }
647 
648 enum {
649 	MLX5_VPORT_ACCESS_METHOD_MAD,
650 	MLX5_VPORT_ACCESS_METHOD_HCA,
651 	MLX5_VPORT_ACCESS_METHOD_NIC,
652 };
653 
654 static int mlx5_get_vport_access_method(struct ib_device *ibdev)
655 {
656 	if (mlx5_use_mad_ifc(to_mdev(ibdev)))
657 		return MLX5_VPORT_ACCESS_METHOD_MAD;
658 
659 	if (mlx5_ib_port_link_layer(ibdev, 1) ==
660 	    IB_LINK_LAYER_ETHERNET)
661 		return MLX5_VPORT_ACCESS_METHOD_NIC;
662 
663 	return MLX5_VPORT_ACCESS_METHOD_HCA;
664 }
665 
666 static void get_atomic_caps(struct mlx5_ib_dev *dev,
667 			    u8 atomic_size_qp,
668 			    struct ib_device_attr *props)
669 {
670 	u8 tmp;
671 	u8 atomic_operations = MLX5_CAP_ATOMIC(dev->mdev, atomic_operations);
672 	u8 atomic_req_8B_endianness_mode =
673 		MLX5_CAP_ATOMIC(dev->mdev, atomic_req_8B_endianness_mode);
674 
675 	/* Check if HW supports 8 bytes standard atomic operations and capable
676 	 * of host endianness respond
677 	 */
678 	tmp = MLX5_ATOMIC_OPS_CMP_SWAP | MLX5_ATOMIC_OPS_FETCH_ADD;
679 	if (((atomic_operations & tmp) == tmp) &&
680 	    (atomic_size_qp & MLX5_ATOMIC_SIZE_QP_8BYTES) &&
681 	    (atomic_req_8B_endianness_mode)) {
682 		props->atomic_cap = IB_ATOMIC_HCA;
683 	} else {
684 		props->atomic_cap = IB_ATOMIC_NONE;
685 	}
686 }
687 
688 static void get_atomic_caps_qp(struct mlx5_ib_dev *dev,
689 			       struct ib_device_attr *props)
690 {
691 	u8 atomic_size_qp = MLX5_CAP_ATOMIC(dev->mdev, atomic_size_qp);
692 
693 	get_atomic_caps(dev, atomic_size_qp, props);
694 }
695 
696 static void get_atomic_caps_dc(struct mlx5_ib_dev *dev,
697 			       struct ib_device_attr *props)
698 {
699 	u8 atomic_size_qp = MLX5_CAP_ATOMIC(dev->mdev, atomic_size_dc);
700 
701 	get_atomic_caps(dev, atomic_size_qp, props);
702 }
703 
704 bool mlx5_ib_dc_atomic_is_supported(struct mlx5_ib_dev *dev)
705 {
706 	struct ib_device_attr props = {};
707 
708 	get_atomic_caps_dc(dev, &props);
709 	return (props.atomic_cap == IB_ATOMIC_HCA) ? true : false;
710 }
711 static int mlx5_query_system_image_guid(struct ib_device *ibdev,
712 					__be64 *sys_image_guid)
713 {
714 	struct mlx5_ib_dev *dev = to_mdev(ibdev);
715 	struct mlx5_core_dev *mdev = dev->mdev;
716 	u64 tmp;
717 	int err;
718 
719 	switch (mlx5_get_vport_access_method(ibdev)) {
720 	case MLX5_VPORT_ACCESS_METHOD_MAD:
721 		return mlx5_query_mad_ifc_system_image_guid(ibdev,
722 							    sys_image_guid);
723 
724 	case MLX5_VPORT_ACCESS_METHOD_HCA:
725 		err = mlx5_query_hca_vport_system_image_guid(mdev, &tmp);
726 		break;
727 
728 	case MLX5_VPORT_ACCESS_METHOD_NIC:
729 		err = mlx5_query_nic_vport_system_image_guid(mdev, &tmp);
730 		break;
731 
732 	default:
733 		return -EINVAL;
734 	}
735 
736 	if (!err)
737 		*sys_image_guid = cpu_to_be64(tmp);
738 
739 	return err;
740 
741 }
742 
743 static int mlx5_query_max_pkeys(struct ib_device *ibdev,
744 				u16 *max_pkeys)
745 {
746 	struct mlx5_ib_dev *dev = to_mdev(ibdev);
747 	struct mlx5_core_dev *mdev = dev->mdev;
748 
749 	switch (mlx5_get_vport_access_method(ibdev)) {
750 	case MLX5_VPORT_ACCESS_METHOD_MAD:
751 		return mlx5_query_mad_ifc_max_pkeys(ibdev, max_pkeys);
752 
753 	case MLX5_VPORT_ACCESS_METHOD_HCA:
754 	case MLX5_VPORT_ACCESS_METHOD_NIC:
755 		*max_pkeys = mlx5_to_sw_pkey_sz(MLX5_CAP_GEN(mdev,
756 						pkey_table_size));
757 		return 0;
758 
759 	default:
760 		return -EINVAL;
761 	}
762 }
763 
764 static int mlx5_query_vendor_id(struct ib_device *ibdev,
765 				u32 *vendor_id)
766 {
767 	struct mlx5_ib_dev *dev = to_mdev(ibdev);
768 
769 	switch (mlx5_get_vport_access_method(ibdev)) {
770 	case MLX5_VPORT_ACCESS_METHOD_MAD:
771 		return mlx5_query_mad_ifc_vendor_id(ibdev, vendor_id);
772 
773 	case MLX5_VPORT_ACCESS_METHOD_HCA:
774 	case MLX5_VPORT_ACCESS_METHOD_NIC:
775 		return mlx5_core_query_vendor_id(dev->mdev, vendor_id);
776 
777 	default:
778 		return -EINVAL;
779 	}
780 }
781 
782 static int mlx5_query_node_guid(struct mlx5_ib_dev *dev,
783 				__be64 *node_guid)
784 {
785 	u64 tmp;
786 	int err;
787 
788 	switch (mlx5_get_vport_access_method(&dev->ib_dev)) {
789 	case MLX5_VPORT_ACCESS_METHOD_MAD:
790 		return mlx5_query_mad_ifc_node_guid(dev, node_guid);
791 
792 	case MLX5_VPORT_ACCESS_METHOD_HCA:
793 		err = mlx5_query_hca_vport_node_guid(dev->mdev, &tmp);
794 		break;
795 
796 	case MLX5_VPORT_ACCESS_METHOD_NIC:
797 		err = mlx5_query_nic_vport_node_guid(dev->mdev, &tmp);
798 		break;
799 
800 	default:
801 		return -EINVAL;
802 	}
803 
804 	if (!err)
805 		*node_guid = cpu_to_be64(tmp);
806 
807 	return err;
808 }
809 
810 struct mlx5_reg_node_desc {
811 	u8	desc[IB_DEVICE_NODE_DESC_MAX];
812 };
813 
814 static int mlx5_query_node_desc(struct mlx5_ib_dev *dev, char *node_desc)
815 {
816 	struct mlx5_reg_node_desc in;
817 
818 	if (mlx5_use_mad_ifc(dev))
819 		return mlx5_query_mad_ifc_node_desc(dev, node_desc);
820 
821 	memset(&in, 0, sizeof(in));
822 
823 	return mlx5_core_access_reg(dev->mdev, &in, sizeof(in), node_desc,
824 				    sizeof(struct mlx5_reg_node_desc),
825 				    MLX5_REG_NODE_DESC, 0, 0);
826 }
827 
828 static int mlx5_ib_query_device(struct ib_device *ibdev,
829 				struct ib_device_attr *props,
830 				struct ib_udata *uhw)
831 {
832 	struct mlx5_ib_dev *dev = to_mdev(ibdev);
833 	struct mlx5_core_dev *mdev = dev->mdev;
834 	int err = -ENOMEM;
835 	int max_sq_desc;
836 	int max_rq_sg;
837 	int max_sq_sg;
838 	u64 min_page_size = 1ull << MLX5_CAP_GEN(mdev, log_pg_sz);
839 	bool raw_support = !mlx5_core_mp_enabled(mdev);
840 	struct mlx5_ib_query_device_resp resp = {};
841 	size_t resp_len;
842 	u64 max_tso;
843 
844 	resp_len = sizeof(resp.comp_mask) + sizeof(resp.response_length);
845 	if (uhw->outlen && uhw->outlen < resp_len)
846 		return -EINVAL;
847 	else
848 		resp.response_length = resp_len;
849 
850 	if (uhw->inlen && !ib_is_udata_cleared(uhw, 0, uhw->inlen))
851 		return -EINVAL;
852 
853 	memset(props, 0, sizeof(*props));
854 	err = mlx5_query_system_image_guid(ibdev,
855 					   &props->sys_image_guid);
856 	if (err)
857 		return err;
858 
859 	err = mlx5_query_max_pkeys(ibdev, &props->max_pkeys);
860 	if (err)
861 		return err;
862 
863 	err = mlx5_query_vendor_id(ibdev, &props->vendor_id);
864 	if (err)
865 		return err;
866 
867 	props->fw_ver = ((u64)fw_rev_maj(dev->mdev) << 32) |
868 		(fw_rev_min(dev->mdev) << 16) |
869 		fw_rev_sub(dev->mdev);
870 	props->device_cap_flags    = IB_DEVICE_CHANGE_PHY_PORT |
871 		IB_DEVICE_PORT_ACTIVE_EVENT		|
872 		IB_DEVICE_SYS_IMAGE_GUID		|
873 		IB_DEVICE_RC_RNR_NAK_GEN;
874 
875 	if (MLX5_CAP_GEN(mdev, pkv))
876 		props->device_cap_flags |= IB_DEVICE_BAD_PKEY_CNTR;
877 	if (MLX5_CAP_GEN(mdev, qkv))
878 		props->device_cap_flags |= IB_DEVICE_BAD_QKEY_CNTR;
879 	if (MLX5_CAP_GEN(mdev, apm))
880 		props->device_cap_flags |= IB_DEVICE_AUTO_PATH_MIG;
881 	if (MLX5_CAP_GEN(mdev, xrc))
882 		props->device_cap_flags |= IB_DEVICE_XRC;
883 	if (MLX5_CAP_GEN(mdev, imaicl)) {
884 		props->device_cap_flags |= IB_DEVICE_MEM_WINDOW |
885 					   IB_DEVICE_MEM_WINDOW_TYPE_2B;
886 		props->max_mw = 1 << MLX5_CAP_GEN(mdev, log_max_mkey);
887 		/* We support 'Gappy' memory registration too */
888 		props->device_cap_flags |= IB_DEVICE_SG_GAPS_REG;
889 	}
890 	props->device_cap_flags |= IB_DEVICE_MEM_MGT_EXTENSIONS;
891 	if (MLX5_CAP_GEN(mdev, sho)) {
892 		props->device_cap_flags |= IB_DEVICE_INTEGRITY_HANDOVER;
893 		/* At this stage no support for signature handover */
894 		props->sig_prot_cap = IB_PROT_T10DIF_TYPE_1 |
895 				      IB_PROT_T10DIF_TYPE_2 |
896 				      IB_PROT_T10DIF_TYPE_3;
897 		props->sig_guard_cap = IB_GUARD_T10DIF_CRC |
898 				       IB_GUARD_T10DIF_CSUM;
899 	}
900 	if (MLX5_CAP_GEN(mdev, block_lb_mc))
901 		props->device_cap_flags |= IB_DEVICE_BLOCK_MULTICAST_LOOPBACK;
902 
903 	if (MLX5_CAP_GEN(dev->mdev, eth_net_offloads) && raw_support) {
904 		if (MLX5_CAP_ETH(mdev, csum_cap)) {
905 			/* Legacy bit to support old userspace libraries */
906 			props->device_cap_flags |= IB_DEVICE_RAW_IP_CSUM;
907 			props->raw_packet_caps |= IB_RAW_PACKET_CAP_IP_CSUM;
908 		}
909 
910 		if (MLX5_CAP_ETH(dev->mdev, vlan_cap))
911 			props->raw_packet_caps |=
912 				IB_RAW_PACKET_CAP_CVLAN_STRIPPING;
913 
914 		if (field_avail(typeof(resp), tso_caps, uhw->outlen)) {
915 			max_tso = MLX5_CAP_ETH(mdev, max_lso_cap);
916 			if (max_tso) {
917 				resp.tso_caps.max_tso = 1 << max_tso;
918 				resp.tso_caps.supported_qpts |=
919 					1 << IB_QPT_RAW_PACKET;
920 				resp.response_length += sizeof(resp.tso_caps);
921 			}
922 		}
923 
924 		if (field_avail(typeof(resp), rss_caps, uhw->outlen)) {
925 			resp.rss_caps.rx_hash_function =
926 						MLX5_RX_HASH_FUNC_TOEPLITZ;
927 			resp.rss_caps.rx_hash_fields_mask =
928 						MLX5_RX_HASH_SRC_IPV4 |
929 						MLX5_RX_HASH_DST_IPV4 |
930 						MLX5_RX_HASH_SRC_IPV6 |
931 						MLX5_RX_HASH_DST_IPV6 |
932 						MLX5_RX_HASH_SRC_PORT_TCP |
933 						MLX5_RX_HASH_DST_PORT_TCP |
934 						MLX5_RX_HASH_SRC_PORT_UDP |
935 						MLX5_RX_HASH_DST_PORT_UDP |
936 						MLX5_RX_HASH_INNER;
937 			if (mlx5_accel_ipsec_device_caps(dev->mdev) &
938 			    MLX5_ACCEL_IPSEC_CAP_DEVICE)
939 				resp.rss_caps.rx_hash_fields_mask |=
940 					MLX5_RX_HASH_IPSEC_SPI;
941 			resp.response_length += sizeof(resp.rss_caps);
942 		}
943 	} else {
944 		if (field_avail(typeof(resp), tso_caps, uhw->outlen))
945 			resp.response_length += sizeof(resp.tso_caps);
946 		if (field_avail(typeof(resp), rss_caps, uhw->outlen))
947 			resp.response_length += sizeof(resp.rss_caps);
948 	}
949 
950 	if (MLX5_CAP_GEN(mdev, ipoib_basic_offloads)) {
951 		props->device_cap_flags |= IB_DEVICE_UD_IP_CSUM;
952 		props->device_cap_flags |= IB_DEVICE_UD_TSO;
953 	}
954 
955 	if (MLX5_CAP_GEN(dev->mdev, rq_delay_drop) &&
956 	    MLX5_CAP_GEN(dev->mdev, general_notification_event) &&
957 	    raw_support)
958 		props->raw_packet_caps |= IB_RAW_PACKET_CAP_DELAY_DROP;
959 
960 	if (MLX5_CAP_GEN(mdev, ipoib_enhanced_offloads) &&
961 	    MLX5_CAP_IPOIB_ENHANCED(mdev, csum_cap))
962 		props->device_cap_flags |= IB_DEVICE_UD_IP_CSUM;
963 
964 	if (MLX5_CAP_GEN(dev->mdev, eth_net_offloads) &&
965 	    MLX5_CAP_ETH(dev->mdev, scatter_fcs) &&
966 	    raw_support) {
967 		/* Legacy bit to support old userspace libraries */
968 		props->device_cap_flags |= IB_DEVICE_RAW_SCATTER_FCS;
969 		props->raw_packet_caps |= IB_RAW_PACKET_CAP_SCATTER_FCS;
970 	}
971 
972 	if (MLX5_CAP_DEV_MEM(mdev, memic)) {
973 		props->max_dm_size =
974 			MLX5_CAP_DEV_MEM(mdev, max_memic_size);
975 	}
976 
977 	if (mlx5_get_flow_namespace(dev->mdev, MLX5_FLOW_NAMESPACE_BYPASS))
978 		props->device_cap_flags |= IB_DEVICE_MANAGED_FLOW_STEERING;
979 
980 	if (MLX5_CAP_GEN(mdev, end_pad))
981 		props->device_cap_flags |= IB_DEVICE_PCI_WRITE_END_PADDING;
982 
983 	props->vendor_part_id	   = mdev->pdev->device;
984 	props->hw_ver		   = mdev->pdev->revision;
985 
986 	props->max_mr_size	   = ~0ull;
987 	props->page_size_cap	   = ~(min_page_size - 1);
988 	props->max_qp		   = 1 << MLX5_CAP_GEN(mdev, log_max_qp);
989 	props->max_qp_wr	   = 1 << MLX5_CAP_GEN(mdev, log_max_qp_sz);
990 	max_rq_sg =  MLX5_CAP_GEN(mdev, max_wqe_sz_rq) /
991 		     sizeof(struct mlx5_wqe_data_seg);
992 	max_sq_desc = min_t(int, MLX5_CAP_GEN(mdev, max_wqe_sz_sq), 512);
993 	max_sq_sg = (max_sq_desc - sizeof(struct mlx5_wqe_ctrl_seg) -
994 		     sizeof(struct mlx5_wqe_raddr_seg)) /
995 		sizeof(struct mlx5_wqe_data_seg);
996 	props->max_send_sge = max_sq_sg;
997 	props->max_recv_sge = max_rq_sg;
998 	props->max_sge_rd	   = MLX5_MAX_SGE_RD;
999 	props->max_cq		   = 1 << MLX5_CAP_GEN(mdev, log_max_cq);
1000 	props->max_cqe = (1 << MLX5_CAP_GEN(mdev, log_max_cq_sz)) - 1;
1001 	props->max_mr		   = 1 << MLX5_CAP_GEN(mdev, log_max_mkey);
1002 	props->max_pd		   = 1 << MLX5_CAP_GEN(mdev, log_max_pd);
1003 	props->max_qp_rd_atom	   = 1 << MLX5_CAP_GEN(mdev, log_max_ra_req_qp);
1004 	props->max_qp_init_rd_atom = 1 << MLX5_CAP_GEN(mdev, log_max_ra_res_qp);
1005 	props->max_srq		   = 1 << MLX5_CAP_GEN(mdev, log_max_srq);
1006 	props->max_srq_wr = (1 << MLX5_CAP_GEN(mdev, log_max_srq_sz)) - 1;
1007 	props->local_ca_ack_delay  = MLX5_CAP_GEN(mdev, local_ca_ack_delay);
1008 	props->max_res_rd_atom	   = props->max_qp_rd_atom * props->max_qp;
1009 	props->max_srq_sge	   = max_rq_sg - 1;
1010 	props->max_fast_reg_page_list_len =
1011 		1 << MLX5_CAP_GEN(mdev, log_max_klm_list_size);
1012 	props->max_pi_fast_reg_page_list_len =
1013 		props->max_fast_reg_page_list_len / 2;
1014 	get_atomic_caps_qp(dev, props);
1015 	props->masked_atomic_cap   = IB_ATOMIC_NONE;
1016 	props->max_mcast_grp	   = 1 << MLX5_CAP_GEN(mdev, log_max_mcg);
1017 	props->max_mcast_qp_attach = MLX5_CAP_GEN(mdev, max_qp_mcg);
1018 	props->max_total_mcast_qp_attach = props->max_mcast_qp_attach *
1019 					   props->max_mcast_grp;
1020 	props->max_map_per_fmr = INT_MAX; /* no limit in ConnectIB */
1021 	props->max_ah = INT_MAX;
1022 	props->hca_core_clock = MLX5_CAP_GEN(mdev, device_frequency_khz);
1023 	props->timestamp_mask = 0x7FFFFFFFFFFFFFFFULL;
1024 
1025 	if (IS_ENABLED(CONFIG_INFINIBAND_ON_DEMAND_PAGING)) {
1026 		if (dev->odp_caps.general_caps & IB_ODP_SUPPORT)
1027 			props->device_cap_flags |= IB_DEVICE_ON_DEMAND_PAGING;
1028 		props->odp_caps = dev->odp_caps;
1029 	}
1030 
1031 	if (MLX5_CAP_GEN(mdev, cd))
1032 		props->device_cap_flags |= IB_DEVICE_CROSS_CHANNEL;
1033 
1034 	if (!mlx5_core_is_pf(mdev))
1035 		props->device_cap_flags |= IB_DEVICE_VIRTUAL_FUNCTION;
1036 
1037 	if (mlx5_ib_port_link_layer(ibdev, 1) ==
1038 	    IB_LINK_LAYER_ETHERNET && raw_support) {
1039 		props->rss_caps.max_rwq_indirection_tables =
1040 			1 << MLX5_CAP_GEN(dev->mdev, log_max_rqt);
1041 		props->rss_caps.max_rwq_indirection_table_size =
1042 			1 << MLX5_CAP_GEN(dev->mdev, log_max_rqt_size);
1043 		props->rss_caps.supported_qpts = 1 << IB_QPT_RAW_PACKET;
1044 		props->max_wq_type_rq =
1045 			1 << MLX5_CAP_GEN(dev->mdev, log_max_rq);
1046 	}
1047 
1048 	if (MLX5_CAP_GEN(mdev, tag_matching)) {
1049 		props->tm_caps.max_num_tags =
1050 			(1 << MLX5_CAP_GEN(mdev, log_tag_matching_list_sz)) - 1;
1051 		props->tm_caps.max_ops =
1052 			1 << MLX5_CAP_GEN(mdev, log_max_qp_sz);
1053 		props->tm_caps.max_sge = MLX5_TM_MAX_SGE;
1054 	}
1055 
1056 	if (MLX5_CAP_GEN(mdev, tag_matching) &&
1057 	    MLX5_CAP_GEN(mdev, rndv_offload_rc)) {
1058 		props->tm_caps.flags = IB_TM_CAP_RNDV_RC;
1059 		props->tm_caps.max_rndv_hdr_size = MLX5_TM_MAX_RNDV_MSG_SIZE;
1060 	}
1061 
1062 	if (MLX5_CAP_GEN(dev->mdev, cq_moderation)) {
1063 		props->cq_caps.max_cq_moderation_count =
1064 						MLX5_MAX_CQ_COUNT;
1065 		props->cq_caps.max_cq_moderation_period =
1066 						MLX5_MAX_CQ_PERIOD;
1067 	}
1068 
1069 	if (field_avail(typeof(resp), cqe_comp_caps, uhw->outlen)) {
1070 		resp.response_length += sizeof(resp.cqe_comp_caps);
1071 
1072 		if (MLX5_CAP_GEN(dev->mdev, cqe_compression)) {
1073 			resp.cqe_comp_caps.max_num =
1074 				MLX5_CAP_GEN(dev->mdev,
1075 					     cqe_compression_max_num);
1076 
1077 			resp.cqe_comp_caps.supported_format =
1078 				MLX5_IB_CQE_RES_FORMAT_HASH |
1079 				MLX5_IB_CQE_RES_FORMAT_CSUM;
1080 
1081 			if (MLX5_CAP_GEN(dev->mdev, mini_cqe_resp_stride_index))
1082 				resp.cqe_comp_caps.supported_format |=
1083 					MLX5_IB_CQE_RES_FORMAT_CSUM_STRIDX;
1084 		}
1085 	}
1086 
1087 	if (field_avail(typeof(resp), packet_pacing_caps, uhw->outlen) &&
1088 	    raw_support) {
1089 		if (MLX5_CAP_QOS(mdev, packet_pacing) &&
1090 		    MLX5_CAP_GEN(mdev, qos)) {
1091 			resp.packet_pacing_caps.qp_rate_limit_max =
1092 				MLX5_CAP_QOS(mdev, packet_pacing_max_rate);
1093 			resp.packet_pacing_caps.qp_rate_limit_min =
1094 				MLX5_CAP_QOS(mdev, packet_pacing_min_rate);
1095 			resp.packet_pacing_caps.supported_qpts |=
1096 				1 << IB_QPT_RAW_PACKET;
1097 			if (MLX5_CAP_QOS(mdev, packet_pacing_burst_bound) &&
1098 			    MLX5_CAP_QOS(mdev, packet_pacing_typical_size))
1099 				resp.packet_pacing_caps.cap_flags |=
1100 					MLX5_IB_PP_SUPPORT_BURST;
1101 		}
1102 		resp.response_length += sizeof(resp.packet_pacing_caps);
1103 	}
1104 
1105 	if (field_avail(typeof(resp), mlx5_ib_support_multi_pkt_send_wqes,
1106 			uhw->outlen)) {
1107 		if (MLX5_CAP_ETH(mdev, multi_pkt_send_wqe))
1108 			resp.mlx5_ib_support_multi_pkt_send_wqes =
1109 				MLX5_IB_ALLOW_MPW;
1110 
1111 		if (MLX5_CAP_ETH(mdev, enhanced_multi_pkt_send_wqe))
1112 			resp.mlx5_ib_support_multi_pkt_send_wqes |=
1113 				MLX5_IB_SUPPORT_EMPW;
1114 
1115 		resp.response_length +=
1116 			sizeof(resp.mlx5_ib_support_multi_pkt_send_wqes);
1117 	}
1118 
1119 	if (field_avail(typeof(resp), flags, uhw->outlen)) {
1120 		resp.response_length += sizeof(resp.flags);
1121 
1122 		if (MLX5_CAP_GEN(mdev, cqe_compression_128))
1123 			resp.flags |=
1124 				MLX5_IB_QUERY_DEV_RESP_FLAGS_CQE_128B_COMP;
1125 
1126 		if (MLX5_CAP_GEN(mdev, cqe_128_always))
1127 			resp.flags |= MLX5_IB_QUERY_DEV_RESP_FLAGS_CQE_128B_PAD;
1128 		if (MLX5_CAP_GEN(mdev, qp_packet_based))
1129 			resp.flags |=
1130 				MLX5_IB_QUERY_DEV_RESP_PACKET_BASED_CREDIT_MODE;
1131 
1132 		resp.flags |= MLX5_IB_QUERY_DEV_RESP_FLAGS_SCAT2CQE_DCT;
1133 	}
1134 
1135 	if (field_avail(typeof(resp), sw_parsing_caps,
1136 			uhw->outlen)) {
1137 		resp.response_length += sizeof(resp.sw_parsing_caps);
1138 		if (MLX5_CAP_ETH(mdev, swp)) {
1139 			resp.sw_parsing_caps.sw_parsing_offloads |=
1140 				MLX5_IB_SW_PARSING;
1141 
1142 			if (MLX5_CAP_ETH(mdev, swp_csum))
1143 				resp.sw_parsing_caps.sw_parsing_offloads |=
1144 					MLX5_IB_SW_PARSING_CSUM;
1145 
1146 			if (MLX5_CAP_ETH(mdev, swp_lso))
1147 				resp.sw_parsing_caps.sw_parsing_offloads |=
1148 					MLX5_IB_SW_PARSING_LSO;
1149 
1150 			if (resp.sw_parsing_caps.sw_parsing_offloads)
1151 				resp.sw_parsing_caps.supported_qpts =
1152 					BIT(IB_QPT_RAW_PACKET);
1153 		}
1154 	}
1155 
1156 	if (field_avail(typeof(resp), striding_rq_caps, uhw->outlen) &&
1157 	    raw_support) {
1158 		resp.response_length += sizeof(resp.striding_rq_caps);
1159 		if (MLX5_CAP_GEN(mdev, striding_rq)) {
1160 			resp.striding_rq_caps.min_single_stride_log_num_of_bytes =
1161 				MLX5_MIN_SINGLE_STRIDE_LOG_NUM_BYTES;
1162 			resp.striding_rq_caps.max_single_stride_log_num_of_bytes =
1163 				MLX5_MAX_SINGLE_STRIDE_LOG_NUM_BYTES;
1164 			resp.striding_rq_caps.min_single_wqe_log_num_of_strides =
1165 				MLX5_MIN_SINGLE_WQE_LOG_NUM_STRIDES;
1166 			resp.striding_rq_caps.max_single_wqe_log_num_of_strides =
1167 				MLX5_MAX_SINGLE_WQE_LOG_NUM_STRIDES;
1168 			resp.striding_rq_caps.supported_qpts =
1169 				BIT(IB_QPT_RAW_PACKET);
1170 		}
1171 	}
1172 
1173 	if (field_avail(typeof(resp), tunnel_offloads_caps,
1174 			uhw->outlen)) {
1175 		resp.response_length += sizeof(resp.tunnel_offloads_caps);
1176 		if (MLX5_CAP_ETH(mdev, tunnel_stateless_vxlan))
1177 			resp.tunnel_offloads_caps |=
1178 				MLX5_IB_TUNNELED_OFFLOADS_VXLAN;
1179 		if (MLX5_CAP_ETH(mdev, tunnel_stateless_geneve_rx))
1180 			resp.tunnel_offloads_caps |=
1181 				MLX5_IB_TUNNELED_OFFLOADS_GENEVE;
1182 		if (MLX5_CAP_ETH(mdev, tunnel_stateless_gre))
1183 			resp.tunnel_offloads_caps |=
1184 				MLX5_IB_TUNNELED_OFFLOADS_GRE;
1185 		if (MLX5_CAP_GEN(mdev, flex_parser_protocols) &
1186 		    MLX5_FLEX_PROTO_CW_MPLS_GRE)
1187 			resp.tunnel_offloads_caps |=
1188 				MLX5_IB_TUNNELED_OFFLOADS_MPLS_GRE;
1189 		if (MLX5_CAP_GEN(mdev, flex_parser_protocols) &
1190 		    MLX5_FLEX_PROTO_CW_MPLS_UDP)
1191 			resp.tunnel_offloads_caps |=
1192 				MLX5_IB_TUNNELED_OFFLOADS_MPLS_UDP;
1193 	}
1194 
1195 	if (uhw->outlen) {
1196 		err = ib_copy_to_udata(uhw, &resp, resp.response_length);
1197 
1198 		if (err)
1199 			return err;
1200 	}
1201 
1202 	return 0;
1203 }
1204 
1205 enum mlx5_ib_width {
1206 	MLX5_IB_WIDTH_1X	= 1 << 0,
1207 	MLX5_IB_WIDTH_2X	= 1 << 1,
1208 	MLX5_IB_WIDTH_4X	= 1 << 2,
1209 	MLX5_IB_WIDTH_8X	= 1 << 3,
1210 	MLX5_IB_WIDTH_12X	= 1 << 4
1211 };
1212 
1213 static void translate_active_width(struct ib_device *ibdev, u8 active_width,
1214 				  u8 *ib_width)
1215 {
1216 	struct mlx5_ib_dev *dev = to_mdev(ibdev);
1217 
1218 	if (active_width & MLX5_IB_WIDTH_1X)
1219 		*ib_width = IB_WIDTH_1X;
1220 	else if (active_width & MLX5_IB_WIDTH_2X)
1221 		*ib_width = IB_WIDTH_2X;
1222 	else if (active_width & MLX5_IB_WIDTH_4X)
1223 		*ib_width = IB_WIDTH_4X;
1224 	else if (active_width & MLX5_IB_WIDTH_8X)
1225 		*ib_width = IB_WIDTH_8X;
1226 	else if (active_width & MLX5_IB_WIDTH_12X)
1227 		*ib_width = IB_WIDTH_12X;
1228 	else {
1229 		mlx5_ib_dbg(dev, "Invalid active_width %d, setting width to default value: 4x\n",
1230 			    (int)active_width);
1231 		*ib_width = IB_WIDTH_4X;
1232 	}
1233 
1234 	return;
1235 }
1236 
1237 static int mlx5_mtu_to_ib_mtu(int mtu)
1238 {
1239 	switch (mtu) {
1240 	case 256: return 1;
1241 	case 512: return 2;
1242 	case 1024: return 3;
1243 	case 2048: return 4;
1244 	case 4096: return 5;
1245 	default:
1246 		pr_warn("invalid mtu\n");
1247 		return -1;
1248 	}
1249 }
1250 
1251 enum ib_max_vl_num {
1252 	__IB_MAX_VL_0		= 1,
1253 	__IB_MAX_VL_0_1		= 2,
1254 	__IB_MAX_VL_0_3		= 3,
1255 	__IB_MAX_VL_0_7		= 4,
1256 	__IB_MAX_VL_0_14	= 5,
1257 };
1258 
1259 enum mlx5_vl_hw_cap {
1260 	MLX5_VL_HW_0	= 1,
1261 	MLX5_VL_HW_0_1	= 2,
1262 	MLX5_VL_HW_0_2	= 3,
1263 	MLX5_VL_HW_0_3	= 4,
1264 	MLX5_VL_HW_0_4	= 5,
1265 	MLX5_VL_HW_0_5	= 6,
1266 	MLX5_VL_HW_0_6	= 7,
1267 	MLX5_VL_HW_0_7	= 8,
1268 	MLX5_VL_HW_0_14	= 15
1269 };
1270 
1271 static int translate_max_vl_num(struct ib_device *ibdev, u8 vl_hw_cap,
1272 				u8 *max_vl_num)
1273 {
1274 	switch (vl_hw_cap) {
1275 	case MLX5_VL_HW_0:
1276 		*max_vl_num = __IB_MAX_VL_0;
1277 		break;
1278 	case MLX5_VL_HW_0_1:
1279 		*max_vl_num = __IB_MAX_VL_0_1;
1280 		break;
1281 	case MLX5_VL_HW_0_3:
1282 		*max_vl_num = __IB_MAX_VL_0_3;
1283 		break;
1284 	case MLX5_VL_HW_0_7:
1285 		*max_vl_num = __IB_MAX_VL_0_7;
1286 		break;
1287 	case MLX5_VL_HW_0_14:
1288 		*max_vl_num = __IB_MAX_VL_0_14;
1289 		break;
1290 
1291 	default:
1292 		return -EINVAL;
1293 	}
1294 
1295 	return 0;
1296 }
1297 
1298 static int mlx5_query_hca_port(struct ib_device *ibdev, u8 port,
1299 			       struct ib_port_attr *props)
1300 {
1301 	struct mlx5_ib_dev *dev = to_mdev(ibdev);
1302 	struct mlx5_core_dev *mdev = dev->mdev;
1303 	struct mlx5_hca_vport_context *rep;
1304 	u16 max_mtu;
1305 	u16 oper_mtu;
1306 	int err;
1307 	u8 ib_link_width_oper;
1308 	u8 vl_hw_cap;
1309 
1310 	rep = kzalloc(sizeof(*rep), GFP_KERNEL);
1311 	if (!rep) {
1312 		err = -ENOMEM;
1313 		goto out;
1314 	}
1315 
1316 	/* props being zeroed by the caller, avoid zeroing it here */
1317 
1318 	err = mlx5_query_hca_vport_context(mdev, 0, port, 0, rep);
1319 	if (err)
1320 		goto out;
1321 
1322 	props->lid		= rep->lid;
1323 	props->lmc		= rep->lmc;
1324 	props->sm_lid		= rep->sm_lid;
1325 	props->sm_sl		= rep->sm_sl;
1326 	props->state		= rep->vport_state;
1327 	props->phys_state	= rep->port_physical_state;
1328 	props->port_cap_flags	= rep->cap_mask1;
1329 	props->gid_tbl_len	= mlx5_get_gid_table_len(MLX5_CAP_GEN(mdev, gid_table_size));
1330 	props->max_msg_sz	= 1 << MLX5_CAP_GEN(mdev, log_max_msg);
1331 	props->pkey_tbl_len	= mlx5_to_sw_pkey_sz(MLX5_CAP_GEN(mdev, pkey_table_size));
1332 	props->bad_pkey_cntr	= rep->pkey_violation_counter;
1333 	props->qkey_viol_cntr	= rep->qkey_violation_counter;
1334 	props->subnet_timeout	= rep->subnet_timeout;
1335 	props->init_type_reply	= rep->init_type_reply;
1336 
1337 	if (props->port_cap_flags & IB_PORT_CAP_MASK2_SUP)
1338 		props->port_cap_flags2 = rep->cap_mask2;
1339 
1340 	err = mlx5_query_port_link_width_oper(mdev, &ib_link_width_oper, port);
1341 	if (err)
1342 		goto out;
1343 
1344 	translate_active_width(ibdev, ib_link_width_oper, &props->active_width);
1345 
1346 	err = mlx5_query_port_ib_proto_oper(mdev, &props->active_speed, port);
1347 	if (err)
1348 		goto out;
1349 
1350 	mlx5_query_port_max_mtu(mdev, &max_mtu, port);
1351 
1352 	props->max_mtu = mlx5_mtu_to_ib_mtu(max_mtu);
1353 
1354 	mlx5_query_port_oper_mtu(mdev, &oper_mtu, port);
1355 
1356 	props->active_mtu = mlx5_mtu_to_ib_mtu(oper_mtu);
1357 
1358 	err = mlx5_query_port_vl_hw_cap(mdev, &vl_hw_cap, port);
1359 	if (err)
1360 		goto out;
1361 
1362 	err = translate_max_vl_num(ibdev, vl_hw_cap,
1363 				   &props->max_vl_num);
1364 out:
1365 	kfree(rep);
1366 	return err;
1367 }
1368 
1369 int mlx5_ib_query_port(struct ib_device *ibdev, u8 port,
1370 		       struct ib_port_attr *props)
1371 {
1372 	unsigned int count;
1373 	int ret;
1374 
1375 	switch (mlx5_get_vport_access_method(ibdev)) {
1376 	case MLX5_VPORT_ACCESS_METHOD_MAD:
1377 		ret = mlx5_query_mad_ifc_port(ibdev, port, props);
1378 		break;
1379 
1380 	case MLX5_VPORT_ACCESS_METHOD_HCA:
1381 		ret = mlx5_query_hca_port(ibdev, port, props);
1382 		break;
1383 
1384 	case MLX5_VPORT_ACCESS_METHOD_NIC:
1385 		ret = mlx5_query_port_roce(ibdev, port, props);
1386 		break;
1387 
1388 	default:
1389 		ret = -EINVAL;
1390 	}
1391 
1392 	if (!ret && props) {
1393 		struct mlx5_ib_dev *dev = to_mdev(ibdev);
1394 		struct mlx5_core_dev *mdev;
1395 		bool put_mdev = true;
1396 
1397 		mdev = mlx5_ib_get_native_port_mdev(dev, port, NULL);
1398 		if (!mdev) {
1399 			/* If the port isn't affiliated yet query the master.
1400 			 * The master and slave will have the same values.
1401 			 */
1402 			mdev = dev->mdev;
1403 			port = 1;
1404 			put_mdev = false;
1405 		}
1406 		count = mlx5_core_reserved_gids_count(mdev);
1407 		if (put_mdev)
1408 			mlx5_ib_put_native_port_mdev(dev, port);
1409 		props->gid_tbl_len -= count;
1410 	}
1411 	return ret;
1412 }
1413 
1414 static int mlx5_ib_rep_query_port(struct ib_device *ibdev, u8 port,
1415 				  struct ib_port_attr *props)
1416 {
1417 	int ret;
1418 
1419 	/* Only link layer == ethernet is valid for representors
1420 	 * and we always use port 1
1421 	 */
1422 	ret = mlx5_query_port_roce(ibdev, port, props);
1423 	if (ret || !props)
1424 		return ret;
1425 
1426 	/* We don't support GIDS */
1427 	props->gid_tbl_len = 0;
1428 
1429 	return ret;
1430 }
1431 
1432 static int mlx5_ib_query_gid(struct ib_device *ibdev, u8 port, int index,
1433 			     union ib_gid *gid)
1434 {
1435 	struct mlx5_ib_dev *dev = to_mdev(ibdev);
1436 	struct mlx5_core_dev *mdev = dev->mdev;
1437 
1438 	switch (mlx5_get_vport_access_method(ibdev)) {
1439 	case MLX5_VPORT_ACCESS_METHOD_MAD:
1440 		return mlx5_query_mad_ifc_gids(ibdev, port, index, gid);
1441 
1442 	case MLX5_VPORT_ACCESS_METHOD_HCA:
1443 		return mlx5_query_hca_vport_gid(mdev, 0, port, 0, index, gid);
1444 
1445 	default:
1446 		return -EINVAL;
1447 	}
1448 
1449 }
1450 
1451 static int mlx5_query_hca_nic_pkey(struct ib_device *ibdev, u8 port,
1452 				   u16 index, u16 *pkey)
1453 {
1454 	struct mlx5_ib_dev *dev = to_mdev(ibdev);
1455 	struct mlx5_core_dev *mdev;
1456 	bool put_mdev = true;
1457 	u8 mdev_port_num;
1458 	int err;
1459 
1460 	mdev = mlx5_ib_get_native_port_mdev(dev, port, &mdev_port_num);
1461 	if (!mdev) {
1462 		/* The port isn't affiliated yet, get the PKey from the master
1463 		 * port. For RoCE the PKey tables will be the same.
1464 		 */
1465 		put_mdev = false;
1466 		mdev = dev->mdev;
1467 		mdev_port_num = 1;
1468 	}
1469 
1470 	err = mlx5_query_hca_vport_pkey(mdev, 0, mdev_port_num, 0,
1471 					index, pkey);
1472 	if (put_mdev)
1473 		mlx5_ib_put_native_port_mdev(dev, port);
1474 
1475 	return err;
1476 }
1477 
1478 static int mlx5_ib_query_pkey(struct ib_device *ibdev, u8 port, u16 index,
1479 			      u16 *pkey)
1480 {
1481 	switch (mlx5_get_vport_access_method(ibdev)) {
1482 	case MLX5_VPORT_ACCESS_METHOD_MAD:
1483 		return mlx5_query_mad_ifc_pkey(ibdev, port, index, pkey);
1484 
1485 	case MLX5_VPORT_ACCESS_METHOD_HCA:
1486 	case MLX5_VPORT_ACCESS_METHOD_NIC:
1487 		return mlx5_query_hca_nic_pkey(ibdev, port, index, pkey);
1488 	default:
1489 		return -EINVAL;
1490 	}
1491 }
1492 
1493 static int mlx5_ib_modify_device(struct ib_device *ibdev, int mask,
1494 				 struct ib_device_modify *props)
1495 {
1496 	struct mlx5_ib_dev *dev = to_mdev(ibdev);
1497 	struct mlx5_reg_node_desc in;
1498 	struct mlx5_reg_node_desc out;
1499 	int err;
1500 
1501 	if (mask & ~IB_DEVICE_MODIFY_NODE_DESC)
1502 		return -EOPNOTSUPP;
1503 
1504 	if (!(mask & IB_DEVICE_MODIFY_NODE_DESC))
1505 		return 0;
1506 
1507 	/*
1508 	 * If possible, pass node desc to FW, so it can generate
1509 	 * a 144 trap.  If cmd fails, just ignore.
1510 	 */
1511 	memcpy(&in, props->node_desc, IB_DEVICE_NODE_DESC_MAX);
1512 	err = mlx5_core_access_reg(dev->mdev, &in, sizeof(in), &out,
1513 				   sizeof(out), MLX5_REG_NODE_DESC, 0, 1);
1514 	if (err)
1515 		return err;
1516 
1517 	memcpy(ibdev->node_desc, props->node_desc, IB_DEVICE_NODE_DESC_MAX);
1518 
1519 	return err;
1520 }
1521 
1522 static int set_port_caps_atomic(struct mlx5_ib_dev *dev, u8 port_num, u32 mask,
1523 				u32 value)
1524 {
1525 	struct mlx5_hca_vport_context ctx = {};
1526 	struct mlx5_core_dev *mdev;
1527 	u8 mdev_port_num;
1528 	int err;
1529 
1530 	mdev = mlx5_ib_get_native_port_mdev(dev, port_num, &mdev_port_num);
1531 	if (!mdev)
1532 		return -ENODEV;
1533 
1534 	err = mlx5_query_hca_vport_context(mdev, 0, mdev_port_num, 0, &ctx);
1535 	if (err)
1536 		goto out;
1537 
1538 	if (~ctx.cap_mask1_perm & mask) {
1539 		mlx5_ib_warn(dev, "trying to change bitmask 0x%X but change supported 0x%X\n",
1540 			     mask, ctx.cap_mask1_perm);
1541 		err = -EINVAL;
1542 		goto out;
1543 	}
1544 
1545 	ctx.cap_mask1 = value;
1546 	ctx.cap_mask1_perm = mask;
1547 	err = mlx5_core_modify_hca_vport_context(mdev, 0, mdev_port_num,
1548 						 0, &ctx);
1549 
1550 out:
1551 	mlx5_ib_put_native_port_mdev(dev, port_num);
1552 
1553 	return err;
1554 }
1555 
1556 static int mlx5_ib_modify_port(struct ib_device *ibdev, u8 port, int mask,
1557 			       struct ib_port_modify *props)
1558 {
1559 	struct mlx5_ib_dev *dev = to_mdev(ibdev);
1560 	struct ib_port_attr attr;
1561 	u32 tmp;
1562 	int err;
1563 	u32 change_mask;
1564 	u32 value;
1565 	bool is_ib = (mlx5_ib_port_link_layer(ibdev, port) ==
1566 		      IB_LINK_LAYER_INFINIBAND);
1567 
1568 	/* CM layer calls ib_modify_port() regardless of the link layer. For
1569 	 * Ethernet ports, qkey violation and Port capabilities are meaningless.
1570 	 */
1571 	if (!is_ib)
1572 		return 0;
1573 
1574 	if (MLX5_CAP_GEN(dev->mdev, ib_virt) && is_ib) {
1575 		change_mask = props->clr_port_cap_mask | props->set_port_cap_mask;
1576 		value = ~props->clr_port_cap_mask | props->set_port_cap_mask;
1577 		return set_port_caps_atomic(dev, port, change_mask, value);
1578 	}
1579 
1580 	mutex_lock(&dev->cap_mask_mutex);
1581 
1582 	err = ib_query_port(ibdev, port, &attr);
1583 	if (err)
1584 		goto out;
1585 
1586 	tmp = (attr.port_cap_flags | props->set_port_cap_mask) &
1587 		~props->clr_port_cap_mask;
1588 
1589 	err = mlx5_set_port_caps(dev->mdev, port, tmp);
1590 
1591 out:
1592 	mutex_unlock(&dev->cap_mask_mutex);
1593 	return err;
1594 }
1595 
1596 static void print_lib_caps(struct mlx5_ib_dev *dev, u64 caps)
1597 {
1598 	mlx5_ib_dbg(dev, "MLX5_LIB_CAP_4K_UAR = %s\n",
1599 		    caps & MLX5_LIB_CAP_4K_UAR ? "y" : "n");
1600 }
1601 
1602 static u16 calc_dynamic_bfregs(int uars_per_sys_page)
1603 {
1604 	/* Large page with non 4k uar support might limit the dynamic size */
1605 	if (uars_per_sys_page == 1  && PAGE_SIZE > 4096)
1606 		return MLX5_MIN_DYN_BFREGS;
1607 
1608 	return MLX5_MAX_DYN_BFREGS;
1609 }
1610 
1611 static int calc_total_bfregs(struct mlx5_ib_dev *dev, bool lib_uar_4k,
1612 			     struct mlx5_ib_alloc_ucontext_req_v2 *req,
1613 			     struct mlx5_bfreg_info *bfregi)
1614 {
1615 	int uars_per_sys_page;
1616 	int bfregs_per_sys_page;
1617 	int ref_bfregs = req->total_num_bfregs;
1618 
1619 	if (req->total_num_bfregs == 0)
1620 		return -EINVAL;
1621 
1622 	BUILD_BUG_ON(MLX5_MAX_BFREGS % MLX5_NON_FP_BFREGS_IN_PAGE);
1623 	BUILD_BUG_ON(MLX5_MAX_BFREGS < MLX5_NON_FP_BFREGS_IN_PAGE);
1624 
1625 	if (req->total_num_bfregs > MLX5_MAX_BFREGS)
1626 		return -ENOMEM;
1627 
1628 	uars_per_sys_page = get_uars_per_sys_page(dev, lib_uar_4k);
1629 	bfregs_per_sys_page = uars_per_sys_page * MLX5_NON_FP_BFREGS_PER_UAR;
1630 	/* This holds the required static allocation asked by the user */
1631 	req->total_num_bfregs = ALIGN(req->total_num_bfregs, bfregs_per_sys_page);
1632 	if (req->num_low_latency_bfregs > req->total_num_bfregs - 1)
1633 		return -EINVAL;
1634 
1635 	bfregi->num_static_sys_pages = req->total_num_bfregs / bfregs_per_sys_page;
1636 	bfregi->num_dyn_bfregs = ALIGN(calc_dynamic_bfregs(uars_per_sys_page), bfregs_per_sys_page);
1637 	bfregi->total_num_bfregs = req->total_num_bfregs + bfregi->num_dyn_bfregs;
1638 	bfregi->num_sys_pages = bfregi->total_num_bfregs / bfregs_per_sys_page;
1639 
1640 	mlx5_ib_dbg(dev, "uar_4k: fw support %s, lib support %s, user requested %d bfregs, allocated %d, total bfregs %d, using %d sys pages\n",
1641 		    MLX5_CAP_GEN(dev->mdev, uar_4k) ? "yes" : "no",
1642 		    lib_uar_4k ? "yes" : "no", ref_bfregs,
1643 		    req->total_num_bfregs, bfregi->total_num_bfregs,
1644 		    bfregi->num_sys_pages);
1645 
1646 	return 0;
1647 }
1648 
1649 static int allocate_uars(struct mlx5_ib_dev *dev, struct mlx5_ib_ucontext *context)
1650 {
1651 	struct mlx5_bfreg_info *bfregi;
1652 	int err;
1653 	int i;
1654 
1655 	bfregi = &context->bfregi;
1656 	for (i = 0; i < bfregi->num_static_sys_pages; i++) {
1657 		err = mlx5_cmd_alloc_uar(dev->mdev, &bfregi->sys_pages[i]);
1658 		if (err)
1659 			goto error;
1660 
1661 		mlx5_ib_dbg(dev, "allocated uar %d\n", bfregi->sys_pages[i]);
1662 	}
1663 
1664 	for (i = bfregi->num_static_sys_pages; i < bfregi->num_sys_pages; i++)
1665 		bfregi->sys_pages[i] = MLX5_IB_INVALID_UAR_INDEX;
1666 
1667 	return 0;
1668 
1669 error:
1670 	for (--i; i >= 0; i--)
1671 		if (mlx5_cmd_free_uar(dev->mdev, bfregi->sys_pages[i]))
1672 			mlx5_ib_warn(dev, "failed to free uar %d\n", i);
1673 
1674 	return err;
1675 }
1676 
1677 static void deallocate_uars(struct mlx5_ib_dev *dev,
1678 			    struct mlx5_ib_ucontext *context)
1679 {
1680 	struct mlx5_bfreg_info *bfregi;
1681 	int i;
1682 
1683 	bfregi = &context->bfregi;
1684 	for (i = 0; i < bfregi->num_sys_pages; i++)
1685 		if (i < bfregi->num_static_sys_pages ||
1686 		    bfregi->sys_pages[i] != MLX5_IB_INVALID_UAR_INDEX)
1687 			mlx5_cmd_free_uar(dev->mdev, bfregi->sys_pages[i]);
1688 }
1689 
1690 int mlx5_ib_enable_lb(struct mlx5_ib_dev *dev, bool td, bool qp)
1691 {
1692 	int err = 0;
1693 
1694 	mutex_lock(&dev->lb.mutex);
1695 	if (td)
1696 		dev->lb.user_td++;
1697 	if (qp)
1698 		dev->lb.qps++;
1699 
1700 	if (dev->lb.user_td == 2 ||
1701 	    dev->lb.qps == 1) {
1702 		if (!dev->lb.enabled) {
1703 			err = mlx5_nic_vport_update_local_lb(dev->mdev, true);
1704 			dev->lb.enabled = true;
1705 		}
1706 	}
1707 
1708 	mutex_unlock(&dev->lb.mutex);
1709 
1710 	return err;
1711 }
1712 
1713 void mlx5_ib_disable_lb(struct mlx5_ib_dev *dev, bool td, bool qp)
1714 {
1715 	mutex_lock(&dev->lb.mutex);
1716 	if (td)
1717 		dev->lb.user_td--;
1718 	if (qp)
1719 		dev->lb.qps--;
1720 
1721 	if (dev->lb.user_td == 1 &&
1722 	    dev->lb.qps == 0) {
1723 		if (dev->lb.enabled) {
1724 			mlx5_nic_vport_update_local_lb(dev->mdev, false);
1725 			dev->lb.enabled = false;
1726 		}
1727 	}
1728 
1729 	mutex_unlock(&dev->lb.mutex);
1730 }
1731 
1732 static int mlx5_ib_alloc_transport_domain(struct mlx5_ib_dev *dev, u32 *tdn,
1733 					  u16 uid)
1734 {
1735 	int err;
1736 
1737 	if (!MLX5_CAP_GEN(dev->mdev, log_max_transport_domain))
1738 		return 0;
1739 
1740 	err = mlx5_cmd_alloc_transport_domain(dev->mdev, tdn, uid);
1741 	if (err)
1742 		return err;
1743 
1744 	if ((MLX5_CAP_GEN(dev->mdev, port_type) != MLX5_CAP_PORT_TYPE_ETH) ||
1745 	    (!MLX5_CAP_GEN(dev->mdev, disable_local_lb_uc) &&
1746 	     !MLX5_CAP_GEN(dev->mdev, disable_local_lb_mc)))
1747 		return err;
1748 
1749 	return mlx5_ib_enable_lb(dev, true, false);
1750 }
1751 
1752 static void mlx5_ib_dealloc_transport_domain(struct mlx5_ib_dev *dev, u32 tdn,
1753 					     u16 uid)
1754 {
1755 	if (!MLX5_CAP_GEN(dev->mdev, log_max_transport_domain))
1756 		return;
1757 
1758 	mlx5_cmd_dealloc_transport_domain(dev->mdev, tdn, uid);
1759 
1760 	if ((MLX5_CAP_GEN(dev->mdev, port_type) != MLX5_CAP_PORT_TYPE_ETH) ||
1761 	    (!MLX5_CAP_GEN(dev->mdev, disable_local_lb_uc) &&
1762 	     !MLX5_CAP_GEN(dev->mdev, disable_local_lb_mc)))
1763 		return;
1764 
1765 	mlx5_ib_disable_lb(dev, true, false);
1766 }
1767 
1768 static int mlx5_ib_alloc_ucontext(struct ib_ucontext *uctx,
1769 				  struct ib_udata *udata)
1770 {
1771 	struct ib_device *ibdev = uctx->device;
1772 	struct mlx5_ib_dev *dev = to_mdev(ibdev);
1773 	struct mlx5_ib_alloc_ucontext_req_v2 req = {};
1774 	struct mlx5_ib_alloc_ucontext_resp resp = {};
1775 	struct mlx5_core_dev *mdev = dev->mdev;
1776 	struct mlx5_ib_ucontext *context = to_mucontext(uctx);
1777 	struct mlx5_bfreg_info *bfregi;
1778 	int ver;
1779 	int err;
1780 	size_t min_req_v2 = offsetof(struct mlx5_ib_alloc_ucontext_req_v2,
1781 				     max_cqe_version);
1782 	u32 dump_fill_mkey;
1783 	bool lib_uar_4k;
1784 
1785 	if (!dev->ib_active)
1786 		return -EAGAIN;
1787 
1788 	if (udata->inlen == sizeof(struct mlx5_ib_alloc_ucontext_req))
1789 		ver = 0;
1790 	else if (udata->inlen >= min_req_v2)
1791 		ver = 2;
1792 	else
1793 		return -EINVAL;
1794 
1795 	err = ib_copy_from_udata(&req, udata, min(udata->inlen, sizeof(req)));
1796 	if (err)
1797 		return err;
1798 
1799 	if (req.flags & ~MLX5_IB_ALLOC_UCTX_DEVX)
1800 		return -EOPNOTSUPP;
1801 
1802 	if (req.comp_mask || req.reserved0 || req.reserved1 || req.reserved2)
1803 		return -EOPNOTSUPP;
1804 
1805 	req.total_num_bfregs = ALIGN(req.total_num_bfregs,
1806 				    MLX5_NON_FP_BFREGS_PER_UAR);
1807 	if (req.num_low_latency_bfregs > req.total_num_bfregs - 1)
1808 		return -EINVAL;
1809 
1810 	resp.qp_tab_size = 1 << MLX5_CAP_GEN(dev->mdev, log_max_qp);
1811 	if (mlx5_core_is_pf(dev->mdev) && MLX5_CAP_GEN(dev->mdev, bf))
1812 		resp.bf_reg_size = 1 << MLX5_CAP_GEN(dev->mdev, log_bf_reg_size);
1813 	resp.cache_line_size = cache_line_size();
1814 	resp.max_sq_desc_sz = MLX5_CAP_GEN(dev->mdev, max_wqe_sz_sq);
1815 	resp.max_rq_desc_sz = MLX5_CAP_GEN(dev->mdev, max_wqe_sz_rq);
1816 	resp.max_send_wqebb = 1 << MLX5_CAP_GEN(dev->mdev, log_max_qp_sz);
1817 	resp.max_recv_wr = 1 << MLX5_CAP_GEN(dev->mdev, log_max_qp_sz);
1818 	resp.max_srq_recv_wr = 1 << MLX5_CAP_GEN(dev->mdev, log_max_srq_sz);
1819 	resp.cqe_version = min_t(__u8,
1820 				 (__u8)MLX5_CAP_GEN(dev->mdev, cqe_version),
1821 				 req.max_cqe_version);
1822 	resp.log_uar_size = MLX5_CAP_GEN(dev->mdev, uar_4k) ?
1823 				MLX5_ADAPTER_PAGE_SHIFT : PAGE_SHIFT;
1824 	resp.num_uars_per_page = MLX5_CAP_GEN(dev->mdev, uar_4k) ?
1825 					MLX5_CAP_GEN(dev->mdev, num_of_uars_per_page) : 1;
1826 	resp.response_length = min(offsetof(typeof(resp), response_length) +
1827 				   sizeof(resp.response_length), udata->outlen);
1828 
1829 	if (mlx5_accel_ipsec_device_caps(dev->mdev) & MLX5_ACCEL_IPSEC_CAP_DEVICE) {
1830 		if (mlx5_get_flow_namespace(dev->mdev, MLX5_FLOW_NAMESPACE_EGRESS))
1831 			resp.flow_action_flags |= MLX5_USER_ALLOC_UCONTEXT_FLOW_ACTION_FLAGS_ESP_AES_GCM;
1832 		if (mlx5_accel_ipsec_device_caps(dev->mdev) & MLX5_ACCEL_IPSEC_CAP_REQUIRED_METADATA)
1833 			resp.flow_action_flags |= MLX5_USER_ALLOC_UCONTEXT_FLOW_ACTION_FLAGS_ESP_AES_GCM_REQ_METADATA;
1834 		if (MLX5_CAP_FLOWTABLE(dev->mdev, flow_table_properties_nic_receive.ft_field_support.outer_esp_spi))
1835 			resp.flow_action_flags |= MLX5_USER_ALLOC_UCONTEXT_FLOW_ACTION_FLAGS_ESP_AES_GCM_SPI_STEERING;
1836 		if (mlx5_accel_ipsec_device_caps(dev->mdev) & MLX5_ACCEL_IPSEC_CAP_TX_IV_IS_ESN)
1837 			resp.flow_action_flags |= MLX5_USER_ALLOC_UCONTEXT_FLOW_ACTION_FLAGS_ESP_AES_GCM_TX_IV_IS_ESN;
1838 		/* MLX5_USER_ALLOC_UCONTEXT_FLOW_ACTION_FLAGS_ESP_AES_GCM_FULL_OFFLOAD is currently always 0 */
1839 	}
1840 
1841 	lib_uar_4k = req.lib_caps & MLX5_LIB_CAP_4K_UAR;
1842 	bfregi = &context->bfregi;
1843 
1844 	/* updates req->total_num_bfregs */
1845 	err = calc_total_bfregs(dev, lib_uar_4k, &req, bfregi);
1846 	if (err)
1847 		goto out_ctx;
1848 
1849 	mutex_init(&bfregi->lock);
1850 	bfregi->lib_uar_4k = lib_uar_4k;
1851 	bfregi->count = kcalloc(bfregi->total_num_bfregs, sizeof(*bfregi->count),
1852 				GFP_KERNEL);
1853 	if (!bfregi->count) {
1854 		err = -ENOMEM;
1855 		goto out_ctx;
1856 	}
1857 
1858 	bfregi->sys_pages = kcalloc(bfregi->num_sys_pages,
1859 				    sizeof(*bfregi->sys_pages),
1860 				    GFP_KERNEL);
1861 	if (!bfregi->sys_pages) {
1862 		err = -ENOMEM;
1863 		goto out_count;
1864 	}
1865 
1866 	err = allocate_uars(dev, context);
1867 	if (err)
1868 		goto out_sys_pages;
1869 
1870 	if (req.flags & MLX5_IB_ALLOC_UCTX_DEVX) {
1871 		err = mlx5_ib_devx_create(dev, true);
1872 		if (err < 0)
1873 			goto out_uars;
1874 		context->devx_uid = err;
1875 	}
1876 
1877 	err = mlx5_ib_alloc_transport_domain(dev, &context->tdn,
1878 					     context->devx_uid);
1879 	if (err)
1880 		goto out_devx;
1881 
1882 	if (MLX5_CAP_GEN(dev->mdev, dump_fill_mkey)) {
1883 		err = mlx5_cmd_dump_fill_mkey(dev->mdev, &dump_fill_mkey);
1884 		if (err)
1885 			goto out_mdev;
1886 	}
1887 
1888 	INIT_LIST_HEAD(&context->db_page_list);
1889 	mutex_init(&context->db_page_mutex);
1890 
1891 	resp.tot_bfregs = req.total_num_bfregs;
1892 	resp.num_ports = dev->num_ports;
1893 
1894 	if (field_avail(typeof(resp), cqe_version, udata->outlen))
1895 		resp.response_length += sizeof(resp.cqe_version);
1896 
1897 	if (field_avail(typeof(resp), cmds_supp_uhw, udata->outlen)) {
1898 		resp.cmds_supp_uhw |= MLX5_USER_CMDS_SUPP_UHW_QUERY_DEVICE |
1899 				      MLX5_USER_CMDS_SUPP_UHW_CREATE_AH;
1900 		resp.response_length += sizeof(resp.cmds_supp_uhw);
1901 	}
1902 
1903 	if (field_avail(typeof(resp), eth_min_inline, udata->outlen)) {
1904 		if (mlx5_ib_port_link_layer(ibdev, 1) == IB_LINK_LAYER_ETHERNET) {
1905 			mlx5_query_min_inline(dev->mdev, &resp.eth_min_inline);
1906 			resp.eth_min_inline++;
1907 		}
1908 		resp.response_length += sizeof(resp.eth_min_inline);
1909 	}
1910 
1911 	if (field_avail(typeof(resp), clock_info_versions, udata->outlen)) {
1912 		if (mdev->clock_info)
1913 			resp.clock_info_versions = BIT(MLX5_IB_CLOCK_INFO_V1);
1914 		resp.response_length += sizeof(resp.clock_info_versions);
1915 	}
1916 
1917 	/*
1918 	 * We don't want to expose information from the PCI bar that is located
1919 	 * after 4096 bytes, so if the arch only supports larger pages, let's
1920 	 * pretend we don't support reading the HCA's core clock. This is also
1921 	 * forced by mmap function.
1922 	 */
1923 	if (field_avail(typeof(resp), hca_core_clock_offset, udata->outlen)) {
1924 		if (PAGE_SIZE <= 4096) {
1925 			resp.comp_mask |=
1926 				MLX5_IB_ALLOC_UCONTEXT_RESP_MASK_CORE_CLOCK_OFFSET;
1927 			resp.hca_core_clock_offset =
1928 				offsetof(struct mlx5_init_seg, internal_timer_h) % PAGE_SIZE;
1929 		}
1930 		resp.response_length += sizeof(resp.hca_core_clock_offset);
1931 	}
1932 
1933 	if (field_avail(typeof(resp), log_uar_size, udata->outlen))
1934 		resp.response_length += sizeof(resp.log_uar_size);
1935 
1936 	if (field_avail(typeof(resp), num_uars_per_page, udata->outlen))
1937 		resp.response_length += sizeof(resp.num_uars_per_page);
1938 
1939 	if (field_avail(typeof(resp), num_dyn_bfregs, udata->outlen)) {
1940 		resp.num_dyn_bfregs = bfregi->num_dyn_bfregs;
1941 		resp.response_length += sizeof(resp.num_dyn_bfregs);
1942 	}
1943 
1944 	if (field_avail(typeof(resp), dump_fill_mkey, udata->outlen)) {
1945 		if (MLX5_CAP_GEN(dev->mdev, dump_fill_mkey)) {
1946 			resp.dump_fill_mkey = dump_fill_mkey;
1947 			resp.comp_mask |=
1948 				MLX5_IB_ALLOC_UCONTEXT_RESP_MASK_DUMP_FILL_MKEY;
1949 		}
1950 		resp.response_length += sizeof(resp.dump_fill_mkey);
1951 	}
1952 
1953 	err = ib_copy_to_udata(udata, &resp, resp.response_length);
1954 	if (err)
1955 		goto out_mdev;
1956 
1957 	bfregi->ver = ver;
1958 	bfregi->num_low_latency_bfregs = req.num_low_latency_bfregs;
1959 	context->cqe_version = resp.cqe_version;
1960 	context->lib_caps = req.lib_caps;
1961 	print_lib_caps(dev, context->lib_caps);
1962 
1963 	if (dev->lag_active) {
1964 		u8 port = mlx5_core_native_port_num(dev->mdev) - 1;
1965 
1966 		atomic_set(&context->tx_port_affinity,
1967 			   atomic_add_return(
1968 				   1, &dev->port[port].roce.tx_port_affinity));
1969 	}
1970 
1971 	return 0;
1972 
1973 out_mdev:
1974 	mlx5_ib_dealloc_transport_domain(dev, context->tdn, context->devx_uid);
1975 out_devx:
1976 	if (req.flags & MLX5_IB_ALLOC_UCTX_DEVX)
1977 		mlx5_ib_devx_destroy(dev, context->devx_uid);
1978 
1979 out_uars:
1980 	deallocate_uars(dev, context);
1981 
1982 out_sys_pages:
1983 	kfree(bfregi->sys_pages);
1984 
1985 out_count:
1986 	kfree(bfregi->count);
1987 
1988 out_ctx:
1989 	return err;
1990 }
1991 
1992 static void mlx5_ib_dealloc_ucontext(struct ib_ucontext *ibcontext)
1993 {
1994 	struct mlx5_ib_ucontext *context = to_mucontext(ibcontext);
1995 	struct mlx5_ib_dev *dev = to_mdev(ibcontext->device);
1996 	struct mlx5_bfreg_info *bfregi;
1997 
1998 	bfregi = &context->bfregi;
1999 	mlx5_ib_dealloc_transport_domain(dev, context->tdn, context->devx_uid);
2000 
2001 	if (context->devx_uid)
2002 		mlx5_ib_devx_destroy(dev, context->devx_uid);
2003 
2004 	deallocate_uars(dev, context);
2005 	kfree(bfregi->sys_pages);
2006 	kfree(bfregi->count);
2007 }
2008 
2009 static phys_addr_t uar_index2pfn(struct mlx5_ib_dev *dev,
2010 				 int uar_idx)
2011 {
2012 	int fw_uars_per_page;
2013 
2014 	fw_uars_per_page = MLX5_CAP_GEN(dev->mdev, uar_4k) ? MLX5_UARS_IN_PAGE : 1;
2015 
2016 	return (dev->mdev->bar_addr >> PAGE_SHIFT) + uar_idx / fw_uars_per_page;
2017 }
2018 
2019 static int get_command(unsigned long offset)
2020 {
2021 	return (offset >> MLX5_IB_MMAP_CMD_SHIFT) & MLX5_IB_MMAP_CMD_MASK;
2022 }
2023 
2024 static int get_arg(unsigned long offset)
2025 {
2026 	return offset & ((1 << MLX5_IB_MMAP_CMD_SHIFT) - 1);
2027 }
2028 
2029 static int get_index(unsigned long offset)
2030 {
2031 	return get_arg(offset);
2032 }
2033 
2034 /* Index resides in an extra byte to enable larger values than 255 */
2035 static int get_extended_index(unsigned long offset)
2036 {
2037 	return get_arg(offset) | ((offset >> 16) & 0xff) << 8;
2038 }
2039 
2040 
2041 static void mlx5_ib_disassociate_ucontext(struct ib_ucontext *ibcontext)
2042 {
2043 }
2044 
2045 static inline char *mmap_cmd2str(enum mlx5_ib_mmap_cmd cmd)
2046 {
2047 	switch (cmd) {
2048 	case MLX5_IB_MMAP_WC_PAGE:
2049 		return "WC";
2050 	case MLX5_IB_MMAP_REGULAR_PAGE:
2051 		return "best effort WC";
2052 	case MLX5_IB_MMAP_NC_PAGE:
2053 		return "NC";
2054 	case MLX5_IB_MMAP_DEVICE_MEM:
2055 		return "Device Memory";
2056 	default:
2057 		return NULL;
2058 	}
2059 }
2060 
2061 static int mlx5_ib_mmap_clock_info_page(struct mlx5_ib_dev *dev,
2062 					struct vm_area_struct *vma,
2063 					struct mlx5_ib_ucontext *context)
2064 {
2065 	if ((vma->vm_end - vma->vm_start != PAGE_SIZE) ||
2066 	    !(vma->vm_flags & VM_SHARED))
2067 		return -EINVAL;
2068 
2069 	if (get_index(vma->vm_pgoff) != MLX5_IB_CLOCK_INFO_V1)
2070 		return -EOPNOTSUPP;
2071 
2072 	if (vma->vm_flags & (VM_WRITE | VM_EXEC))
2073 		return -EPERM;
2074 	vma->vm_flags &= ~VM_MAYWRITE;
2075 
2076 	if (!dev->mdev->clock_info)
2077 		return -EOPNOTSUPP;
2078 
2079 	return vm_insert_page(vma, vma->vm_start,
2080 			      virt_to_page(dev->mdev->clock_info));
2081 }
2082 
2083 static int uar_mmap(struct mlx5_ib_dev *dev, enum mlx5_ib_mmap_cmd cmd,
2084 		    struct vm_area_struct *vma,
2085 		    struct mlx5_ib_ucontext *context)
2086 {
2087 	struct mlx5_bfreg_info *bfregi = &context->bfregi;
2088 	int err;
2089 	unsigned long idx;
2090 	phys_addr_t pfn;
2091 	pgprot_t prot;
2092 	u32 bfreg_dyn_idx = 0;
2093 	u32 uar_index;
2094 	int dyn_uar = (cmd == MLX5_IB_MMAP_ALLOC_WC);
2095 	int max_valid_idx = dyn_uar ? bfregi->num_sys_pages :
2096 				bfregi->num_static_sys_pages;
2097 
2098 	if (vma->vm_end - vma->vm_start != PAGE_SIZE)
2099 		return -EINVAL;
2100 
2101 	if (dyn_uar)
2102 		idx = get_extended_index(vma->vm_pgoff) + bfregi->num_static_sys_pages;
2103 	else
2104 		idx = get_index(vma->vm_pgoff);
2105 
2106 	if (idx >= max_valid_idx) {
2107 		mlx5_ib_warn(dev, "invalid uar index %lu, max=%d\n",
2108 			     idx, max_valid_idx);
2109 		return -EINVAL;
2110 	}
2111 
2112 	switch (cmd) {
2113 	case MLX5_IB_MMAP_WC_PAGE:
2114 	case MLX5_IB_MMAP_ALLOC_WC:
2115 /* Some architectures don't support WC memory */
2116 #if defined(CONFIG_X86)
2117 		if (!pat_enabled())
2118 			return -EPERM;
2119 #elif !(defined(CONFIG_PPC) || (defined(CONFIG_ARM) && defined(CONFIG_MMU)))
2120 			return -EPERM;
2121 #endif
2122 	/* fall through */
2123 	case MLX5_IB_MMAP_REGULAR_PAGE:
2124 		/* For MLX5_IB_MMAP_REGULAR_PAGE do the best effort to get WC */
2125 		prot = pgprot_writecombine(vma->vm_page_prot);
2126 		break;
2127 	case MLX5_IB_MMAP_NC_PAGE:
2128 		prot = pgprot_noncached(vma->vm_page_prot);
2129 		break;
2130 	default:
2131 		return -EINVAL;
2132 	}
2133 
2134 	if (dyn_uar) {
2135 		int uars_per_page;
2136 
2137 		uars_per_page = get_uars_per_sys_page(dev, bfregi->lib_uar_4k);
2138 		bfreg_dyn_idx = idx * (uars_per_page * MLX5_NON_FP_BFREGS_PER_UAR);
2139 		if (bfreg_dyn_idx >= bfregi->total_num_bfregs) {
2140 			mlx5_ib_warn(dev, "invalid bfreg_dyn_idx %u, max=%u\n",
2141 				     bfreg_dyn_idx, bfregi->total_num_bfregs);
2142 			return -EINVAL;
2143 		}
2144 
2145 		mutex_lock(&bfregi->lock);
2146 		/* Fail if uar already allocated, first bfreg index of each
2147 		 * page holds its count.
2148 		 */
2149 		if (bfregi->count[bfreg_dyn_idx]) {
2150 			mlx5_ib_warn(dev, "wrong offset, idx %lu is busy, bfregn=%u\n", idx, bfreg_dyn_idx);
2151 			mutex_unlock(&bfregi->lock);
2152 			return -EINVAL;
2153 		}
2154 
2155 		bfregi->count[bfreg_dyn_idx]++;
2156 		mutex_unlock(&bfregi->lock);
2157 
2158 		err = mlx5_cmd_alloc_uar(dev->mdev, &uar_index);
2159 		if (err) {
2160 			mlx5_ib_warn(dev, "UAR alloc failed\n");
2161 			goto free_bfreg;
2162 		}
2163 	} else {
2164 		uar_index = bfregi->sys_pages[idx];
2165 	}
2166 
2167 	pfn = uar_index2pfn(dev, uar_index);
2168 	mlx5_ib_dbg(dev, "uar idx 0x%lx, pfn %pa\n", idx, &pfn);
2169 
2170 	err = rdma_user_mmap_io(&context->ibucontext, vma, pfn, PAGE_SIZE,
2171 				prot);
2172 	if (err) {
2173 		mlx5_ib_err(dev,
2174 			    "rdma_user_mmap_io failed with error=%d, mmap_cmd=%s\n",
2175 			    err, mmap_cmd2str(cmd));
2176 		goto err;
2177 	}
2178 
2179 	if (dyn_uar)
2180 		bfregi->sys_pages[idx] = uar_index;
2181 	return 0;
2182 
2183 err:
2184 	if (!dyn_uar)
2185 		return err;
2186 
2187 	mlx5_cmd_free_uar(dev->mdev, idx);
2188 
2189 free_bfreg:
2190 	mlx5_ib_free_bfreg(dev, bfregi, bfreg_dyn_idx);
2191 
2192 	return err;
2193 }
2194 
2195 static int dm_mmap(struct ib_ucontext *context, struct vm_area_struct *vma)
2196 {
2197 	struct mlx5_ib_ucontext *mctx = to_mucontext(context);
2198 	struct mlx5_ib_dev *dev = to_mdev(context->device);
2199 	u16 page_idx = get_extended_index(vma->vm_pgoff);
2200 	size_t map_size = vma->vm_end - vma->vm_start;
2201 	u32 npages = map_size >> PAGE_SHIFT;
2202 	phys_addr_t pfn;
2203 
2204 	if (find_next_zero_bit(mctx->dm_pages, page_idx + npages, page_idx) !=
2205 	    page_idx + npages)
2206 		return -EINVAL;
2207 
2208 	pfn = ((dev->mdev->bar_addr +
2209 	      MLX5_CAP64_DEV_MEM(dev->mdev, memic_bar_start_addr)) >>
2210 	      PAGE_SHIFT) +
2211 	      page_idx;
2212 	return rdma_user_mmap_io(context, vma, pfn, map_size,
2213 				 pgprot_writecombine(vma->vm_page_prot));
2214 }
2215 
2216 static int mlx5_ib_mmap(struct ib_ucontext *ibcontext, struct vm_area_struct *vma)
2217 {
2218 	struct mlx5_ib_ucontext *context = to_mucontext(ibcontext);
2219 	struct mlx5_ib_dev *dev = to_mdev(ibcontext->device);
2220 	unsigned long command;
2221 	phys_addr_t pfn;
2222 
2223 	command = get_command(vma->vm_pgoff);
2224 	switch (command) {
2225 	case MLX5_IB_MMAP_WC_PAGE:
2226 	case MLX5_IB_MMAP_NC_PAGE:
2227 	case MLX5_IB_MMAP_REGULAR_PAGE:
2228 	case MLX5_IB_MMAP_ALLOC_WC:
2229 		return uar_mmap(dev, command, vma, context);
2230 
2231 	case MLX5_IB_MMAP_GET_CONTIGUOUS_PAGES:
2232 		return -ENOSYS;
2233 
2234 	case MLX5_IB_MMAP_CORE_CLOCK:
2235 		if (vma->vm_end - vma->vm_start != PAGE_SIZE)
2236 			return -EINVAL;
2237 
2238 		if (vma->vm_flags & VM_WRITE)
2239 			return -EPERM;
2240 		vma->vm_flags &= ~VM_MAYWRITE;
2241 
2242 		/* Don't expose to user-space information it shouldn't have */
2243 		if (PAGE_SIZE > 4096)
2244 			return -EOPNOTSUPP;
2245 
2246 		pfn = (dev->mdev->iseg_base +
2247 		       offsetof(struct mlx5_init_seg, internal_timer_h)) >>
2248 			PAGE_SHIFT;
2249 		return rdma_user_mmap_io(&context->ibucontext, vma, pfn,
2250 					 PAGE_SIZE,
2251 					 pgprot_noncached(vma->vm_page_prot));
2252 	case MLX5_IB_MMAP_CLOCK_INFO:
2253 		return mlx5_ib_mmap_clock_info_page(dev, vma, context);
2254 
2255 	case MLX5_IB_MMAP_DEVICE_MEM:
2256 		return dm_mmap(ibcontext, vma);
2257 
2258 	default:
2259 		return -EINVAL;
2260 	}
2261 
2262 	return 0;
2263 }
2264 
2265 static inline int check_dm_type_support(struct mlx5_ib_dev *dev,
2266 					u32 type)
2267 {
2268 	switch (type) {
2269 	case MLX5_IB_UAPI_DM_TYPE_MEMIC:
2270 		if (!MLX5_CAP_DEV_MEM(dev->mdev, memic))
2271 			return -EOPNOTSUPP;
2272 		break;
2273 	case MLX5_IB_UAPI_DM_TYPE_STEERING_SW_ICM:
2274 	case MLX5_IB_UAPI_DM_TYPE_HEADER_MODIFY_SW_ICM:
2275 		if (!capable(CAP_SYS_RAWIO) ||
2276 		    !capable(CAP_NET_RAW))
2277 			return -EPERM;
2278 
2279 		if (!(MLX5_CAP_FLOWTABLE_NIC_RX(dev->mdev, sw_owner) ||
2280 		      MLX5_CAP_FLOWTABLE_NIC_TX(dev->mdev, sw_owner)))
2281 			return -EOPNOTSUPP;
2282 		break;
2283 	}
2284 
2285 	return 0;
2286 }
2287 
2288 static int handle_alloc_dm_memic(struct ib_ucontext *ctx,
2289 				 struct mlx5_ib_dm *dm,
2290 				 struct ib_dm_alloc_attr *attr,
2291 				 struct uverbs_attr_bundle *attrs)
2292 {
2293 	struct mlx5_dm *dm_db = &to_mdev(ctx->device)->dm;
2294 	u64 start_offset;
2295 	u32 page_idx;
2296 	int err;
2297 
2298 	dm->size = roundup(attr->length, MLX5_MEMIC_BASE_SIZE);
2299 
2300 	err = mlx5_cmd_alloc_memic(dm_db, &dm->dev_addr,
2301 				   dm->size, attr->alignment);
2302 	if (err)
2303 		return err;
2304 
2305 	page_idx = (dm->dev_addr - pci_resource_start(dm_db->dev->pdev, 0) -
2306 		    MLX5_CAP64_DEV_MEM(dm_db->dev, memic_bar_start_addr)) >>
2307 		    PAGE_SHIFT;
2308 
2309 	err = uverbs_copy_to(attrs,
2310 			     MLX5_IB_ATTR_ALLOC_DM_RESP_PAGE_INDEX,
2311 			     &page_idx, sizeof(page_idx));
2312 	if (err)
2313 		goto err_dealloc;
2314 
2315 	start_offset = dm->dev_addr & ~PAGE_MASK;
2316 	err = uverbs_copy_to(attrs,
2317 			     MLX5_IB_ATTR_ALLOC_DM_RESP_START_OFFSET,
2318 			     &start_offset, sizeof(start_offset));
2319 	if (err)
2320 		goto err_dealloc;
2321 
2322 	bitmap_set(to_mucontext(ctx)->dm_pages, page_idx,
2323 		   DIV_ROUND_UP(dm->size, PAGE_SIZE));
2324 
2325 	return 0;
2326 
2327 err_dealloc:
2328 	mlx5_cmd_dealloc_memic(dm_db, dm->dev_addr, dm->size);
2329 
2330 	return err;
2331 }
2332 
2333 static int handle_alloc_dm_sw_icm(struct ib_ucontext *ctx,
2334 				  struct mlx5_ib_dm *dm,
2335 				  struct ib_dm_alloc_attr *attr,
2336 				  struct uverbs_attr_bundle *attrs,
2337 				  int type)
2338 {
2339 	struct mlx5_core_dev *dev = to_mdev(ctx->device)->mdev;
2340 	u64 act_size;
2341 	int err;
2342 
2343 	/* Allocation size must a multiple of the basic block size
2344 	 * and a power of 2.
2345 	 */
2346 	act_size = round_up(attr->length, MLX5_SW_ICM_BLOCK_SIZE(dev));
2347 	act_size = roundup_pow_of_two(act_size);
2348 
2349 	dm->size = act_size;
2350 	err = mlx5_dm_sw_icm_alloc(dev, type, act_size,
2351 				   to_mucontext(ctx)->devx_uid, &dm->dev_addr,
2352 				   &dm->icm_dm.obj_id);
2353 	if (err)
2354 		return err;
2355 
2356 	err = uverbs_copy_to(attrs,
2357 			     MLX5_IB_ATTR_ALLOC_DM_RESP_START_OFFSET,
2358 			     &dm->dev_addr, sizeof(dm->dev_addr));
2359 	if (err)
2360 		mlx5_dm_sw_icm_dealloc(dev, type, dm->size,
2361 				       to_mucontext(ctx)->devx_uid, dm->dev_addr,
2362 				       dm->icm_dm.obj_id);
2363 
2364 	return err;
2365 }
2366 
2367 struct ib_dm *mlx5_ib_alloc_dm(struct ib_device *ibdev,
2368 			       struct ib_ucontext *context,
2369 			       struct ib_dm_alloc_attr *attr,
2370 			       struct uverbs_attr_bundle *attrs)
2371 {
2372 	struct mlx5_ib_dm *dm;
2373 	enum mlx5_ib_uapi_dm_type type;
2374 	int err;
2375 
2376 	err = uverbs_get_const_default(&type, attrs,
2377 				       MLX5_IB_ATTR_ALLOC_DM_REQ_TYPE,
2378 				       MLX5_IB_UAPI_DM_TYPE_MEMIC);
2379 	if (err)
2380 		return ERR_PTR(err);
2381 
2382 	mlx5_ib_dbg(to_mdev(ibdev), "alloc_dm req: dm_type=%d user_length=0x%llx log_alignment=%d\n",
2383 		    type, attr->length, attr->alignment);
2384 
2385 	err = check_dm_type_support(to_mdev(ibdev), type);
2386 	if (err)
2387 		return ERR_PTR(err);
2388 
2389 	dm = kzalloc(sizeof(*dm), GFP_KERNEL);
2390 	if (!dm)
2391 		return ERR_PTR(-ENOMEM);
2392 
2393 	dm->type = type;
2394 
2395 	switch (type) {
2396 	case MLX5_IB_UAPI_DM_TYPE_MEMIC:
2397 		err = handle_alloc_dm_memic(context, dm,
2398 					    attr,
2399 					    attrs);
2400 		break;
2401 	case MLX5_IB_UAPI_DM_TYPE_STEERING_SW_ICM:
2402 		err = handle_alloc_dm_sw_icm(context, dm,
2403 					     attr, attrs,
2404 					     MLX5_SW_ICM_TYPE_STEERING);
2405 		break;
2406 	case MLX5_IB_UAPI_DM_TYPE_HEADER_MODIFY_SW_ICM:
2407 		err = handle_alloc_dm_sw_icm(context, dm,
2408 					     attr, attrs,
2409 					     MLX5_SW_ICM_TYPE_HEADER_MODIFY);
2410 		break;
2411 	default:
2412 		err = -EOPNOTSUPP;
2413 	}
2414 
2415 	if (err)
2416 		goto err_free;
2417 
2418 	return &dm->ibdm;
2419 
2420 err_free:
2421 	kfree(dm);
2422 	return ERR_PTR(err);
2423 }
2424 
2425 int mlx5_ib_dealloc_dm(struct ib_dm *ibdm, struct uverbs_attr_bundle *attrs)
2426 {
2427 	struct mlx5_ib_ucontext *ctx = rdma_udata_to_drv_context(
2428 		&attrs->driver_udata, struct mlx5_ib_ucontext, ibucontext);
2429 	struct mlx5_core_dev *dev = to_mdev(ibdm->device)->mdev;
2430 	struct mlx5_dm *dm_db = &to_mdev(ibdm->device)->dm;
2431 	struct mlx5_ib_dm *dm = to_mdm(ibdm);
2432 	u32 page_idx;
2433 	int ret;
2434 
2435 	switch (dm->type) {
2436 	case MLX5_IB_UAPI_DM_TYPE_MEMIC:
2437 		ret = mlx5_cmd_dealloc_memic(dm_db, dm->dev_addr, dm->size);
2438 		if (ret)
2439 			return ret;
2440 
2441 		page_idx = (dm->dev_addr - pci_resource_start(dev->pdev, 0) -
2442 			    MLX5_CAP64_DEV_MEM(dev, memic_bar_start_addr)) >>
2443 			    PAGE_SHIFT;
2444 		bitmap_clear(ctx->dm_pages, page_idx,
2445 			     DIV_ROUND_UP(dm->size, PAGE_SIZE));
2446 		break;
2447 	case MLX5_IB_UAPI_DM_TYPE_STEERING_SW_ICM:
2448 		ret = mlx5_dm_sw_icm_dealloc(dev, MLX5_SW_ICM_TYPE_STEERING,
2449 					     dm->size, ctx->devx_uid, dm->dev_addr,
2450 					     dm->icm_dm.obj_id);
2451 		if (ret)
2452 			return ret;
2453 		break;
2454 	case MLX5_IB_UAPI_DM_TYPE_HEADER_MODIFY_SW_ICM:
2455 		ret = mlx5_dm_sw_icm_dealloc(dev, MLX5_SW_ICM_TYPE_HEADER_MODIFY,
2456 					     dm->size, ctx->devx_uid, dm->dev_addr,
2457 					     dm->icm_dm.obj_id);
2458 		if (ret)
2459 			return ret;
2460 		break;
2461 	default:
2462 		return -EOPNOTSUPP;
2463 	}
2464 
2465 	kfree(dm);
2466 
2467 	return 0;
2468 }
2469 
2470 static int mlx5_ib_alloc_pd(struct ib_pd *ibpd, struct ib_udata *udata)
2471 {
2472 	struct mlx5_ib_pd *pd = to_mpd(ibpd);
2473 	struct ib_device *ibdev = ibpd->device;
2474 	struct mlx5_ib_alloc_pd_resp resp;
2475 	int err;
2476 	u32 out[MLX5_ST_SZ_DW(alloc_pd_out)] = {};
2477 	u32 in[MLX5_ST_SZ_DW(alloc_pd_in)]   = {};
2478 	u16 uid = 0;
2479 	struct mlx5_ib_ucontext *context = rdma_udata_to_drv_context(
2480 		udata, struct mlx5_ib_ucontext, ibucontext);
2481 
2482 	uid = context ? context->devx_uid : 0;
2483 	MLX5_SET(alloc_pd_in, in, opcode, MLX5_CMD_OP_ALLOC_PD);
2484 	MLX5_SET(alloc_pd_in, in, uid, uid);
2485 	err = mlx5_cmd_exec(to_mdev(ibdev)->mdev, in, sizeof(in),
2486 			    out, sizeof(out));
2487 	if (err)
2488 		return err;
2489 
2490 	pd->pdn = MLX5_GET(alloc_pd_out, out, pd);
2491 	pd->uid = uid;
2492 	if (udata) {
2493 		resp.pdn = pd->pdn;
2494 		if (ib_copy_to_udata(udata, &resp, sizeof(resp))) {
2495 			mlx5_cmd_dealloc_pd(to_mdev(ibdev)->mdev, pd->pdn, uid);
2496 			return -EFAULT;
2497 		}
2498 	}
2499 
2500 	return 0;
2501 }
2502 
2503 static void mlx5_ib_dealloc_pd(struct ib_pd *pd, struct ib_udata *udata)
2504 {
2505 	struct mlx5_ib_dev *mdev = to_mdev(pd->device);
2506 	struct mlx5_ib_pd *mpd = to_mpd(pd);
2507 
2508 	mlx5_cmd_dealloc_pd(mdev->mdev, mpd->pdn, mpd->uid);
2509 }
2510 
2511 enum {
2512 	MATCH_CRITERIA_ENABLE_OUTER_BIT,
2513 	MATCH_CRITERIA_ENABLE_MISC_BIT,
2514 	MATCH_CRITERIA_ENABLE_INNER_BIT,
2515 	MATCH_CRITERIA_ENABLE_MISC2_BIT
2516 };
2517 
2518 #define HEADER_IS_ZERO(match_criteria, headers)			           \
2519 	!(memchr_inv(MLX5_ADDR_OF(fte_match_param, match_criteria, headers), \
2520 		    0, MLX5_FLD_SZ_BYTES(fte_match_param, headers)))       \
2521 
2522 static u8 get_match_criteria_enable(u32 *match_criteria)
2523 {
2524 	u8 match_criteria_enable;
2525 
2526 	match_criteria_enable =
2527 		(!HEADER_IS_ZERO(match_criteria, outer_headers)) <<
2528 		MATCH_CRITERIA_ENABLE_OUTER_BIT;
2529 	match_criteria_enable |=
2530 		(!HEADER_IS_ZERO(match_criteria, misc_parameters)) <<
2531 		MATCH_CRITERIA_ENABLE_MISC_BIT;
2532 	match_criteria_enable |=
2533 		(!HEADER_IS_ZERO(match_criteria, inner_headers)) <<
2534 		MATCH_CRITERIA_ENABLE_INNER_BIT;
2535 	match_criteria_enable |=
2536 		(!HEADER_IS_ZERO(match_criteria, misc_parameters_2)) <<
2537 		MATCH_CRITERIA_ENABLE_MISC2_BIT;
2538 
2539 	return match_criteria_enable;
2540 }
2541 
2542 static int set_proto(void *outer_c, void *outer_v, u8 mask, u8 val)
2543 {
2544 	u8 entry_mask;
2545 	u8 entry_val;
2546 	int err = 0;
2547 
2548 	if (!mask)
2549 		goto out;
2550 
2551 	entry_mask = MLX5_GET(fte_match_set_lyr_2_4, outer_c,
2552 			      ip_protocol);
2553 	entry_val = MLX5_GET(fte_match_set_lyr_2_4, outer_v,
2554 			     ip_protocol);
2555 	if (!entry_mask) {
2556 		MLX5_SET(fte_match_set_lyr_2_4, outer_c, ip_protocol, mask);
2557 		MLX5_SET(fte_match_set_lyr_2_4, outer_v, ip_protocol, val);
2558 		goto out;
2559 	}
2560 	/* Don't override existing ip protocol */
2561 	if (mask != entry_mask || val != entry_val)
2562 		err = -EINVAL;
2563 out:
2564 	return err;
2565 }
2566 
2567 static void set_flow_label(void *misc_c, void *misc_v, u32 mask, u32 val,
2568 			   bool inner)
2569 {
2570 	if (inner) {
2571 		MLX5_SET(fte_match_set_misc,
2572 			 misc_c, inner_ipv6_flow_label, mask);
2573 		MLX5_SET(fte_match_set_misc,
2574 			 misc_v, inner_ipv6_flow_label, val);
2575 	} else {
2576 		MLX5_SET(fte_match_set_misc,
2577 			 misc_c, outer_ipv6_flow_label, mask);
2578 		MLX5_SET(fte_match_set_misc,
2579 			 misc_v, outer_ipv6_flow_label, val);
2580 	}
2581 }
2582 
2583 static void set_tos(void *outer_c, void *outer_v, u8 mask, u8 val)
2584 {
2585 	MLX5_SET(fte_match_set_lyr_2_4, outer_c, ip_ecn, mask);
2586 	MLX5_SET(fte_match_set_lyr_2_4, outer_v, ip_ecn, val);
2587 	MLX5_SET(fte_match_set_lyr_2_4, outer_c, ip_dscp, mask >> 2);
2588 	MLX5_SET(fte_match_set_lyr_2_4, outer_v, ip_dscp, val >> 2);
2589 }
2590 
2591 static int check_mpls_supp_fields(u32 field_support, const __be32 *set_mask)
2592 {
2593 	if (MLX5_GET(fte_match_mpls, set_mask, mpls_label) &&
2594 	    !(field_support & MLX5_FIELD_SUPPORT_MPLS_LABEL))
2595 		return -EOPNOTSUPP;
2596 
2597 	if (MLX5_GET(fte_match_mpls, set_mask, mpls_exp) &&
2598 	    !(field_support & MLX5_FIELD_SUPPORT_MPLS_EXP))
2599 		return -EOPNOTSUPP;
2600 
2601 	if (MLX5_GET(fte_match_mpls, set_mask, mpls_s_bos) &&
2602 	    !(field_support & MLX5_FIELD_SUPPORT_MPLS_S_BOS))
2603 		return -EOPNOTSUPP;
2604 
2605 	if (MLX5_GET(fte_match_mpls, set_mask, mpls_ttl) &&
2606 	    !(field_support & MLX5_FIELD_SUPPORT_MPLS_TTL))
2607 		return -EOPNOTSUPP;
2608 
2609 	return 0;
2610 }
2611 
2612 #define LAST_ETH_FIELD vlan_tag
2613 #define LAST_IB_FIELD sl
2614 #define LAST_IPV4_FIELD tos
2615 #define LAST_IPV6_FIELD traffic_class
2616 #define LAST_TCP_UDP_FIELD src_port
2617 #define LAST_TUNNEL_FIELD tunnel_id
2618 #define LAST_FLOW_TAG_FIELD tag_id
2619 #define LAST_DROP_FIELD size
2620 #define LAST_COUNTERS_FIELD counters
2621 
2622 /* Field is the last supported field */
2623 #define FIELDS_NOT_SUPPORTED(filter, field)\
2624 	memchr_inv((void *)&filter.field  +\
2625 		   sizeof(filter.field), 0,\
2626 		   sizeof(filter) -\
2627 		   offsetof(typeof(filter), field) -\
2628 		   sizeof(filter.field))
2629 
2630 int parse_flow_flow_action(struct mlx5_ib_flow_action *maction,
2631 			   bool is_egress,
2632 			   struct mlx5_flow_act *action)
2633 {
2634 
2635 	switch (maction->ib_action.type) {
2636 	case IB_FLOW_ACTION_ESP:
2637 		if (action->action & (MLX5_FLOW_CONTEXT_ACTION_ENCRYPT |
2638 				      MLX5_FLOW_CONTEXT_ACTION_DECRYPT))
2639 			return -EINVAL;
2640 		/* Currently only AES_GCM keymat is supported by the driver */
2641 		action->esp_id = (uintptr_t)maction->esp_aes_gcm.ctx;
2642 		action->action |= is_egress ?
2643 			MLX5_FLOW_CONTEXT_ACTION_ENCRYPT :
2644 			MLX5_FLOW_CONTEXT_ACTION_DECRYPT;
2645 		return 0;
2646 	case IB_FLOW_ACTION_UNSPECIFIED:
2647 		if (maction->flow_action_raw.sub_type ==
2648 		    MLX5_IB_FLOW_ACTION_MODIFY_HEADER) {
2649 			if (action->action & MLX5_FLOW_CONTEXT_ACTION_MOD_HDR)
2650 				return -EINVAL;
2651 			action->action |= MLX5_FLOW_CONTEXT_ACTION_MOD_HDR;
2652 			action->modify_hdr =
2653 				maction->flow_action_raw.modify_hdr;
2654 			return 0;
2655 		}
2656 		if (maction->flow_action_raw.sub_type ==
2657 		    MLX5_IB_FLOW_ACTION_DECAP) {
2658 			if (action->action & MLX5_FLOW_CONTEXT_ACTION_DECAP)
2659 				return -EINVAL;
2660 			action->action |= MLX5_FLOW_CONTEXT_ACTION_DECAP;
2661 			return 0;
2662 		}
2663 		if (maction->flow_action_raw.sub_type ==
2664 		    MLX5_IB_FLOW_ACTION_PACKET_REFORMAT) {
2665 			if (action->action &
2666 			    MLX5_FLOW_CONTEXT_ACTION_PACKET_REFORMAT)
2667 				return -EINVAL;
2668 			action->action |=
2669 				MLX5_FLOW_CONTEXT_ACTION_PACKET_REFORMAT;
2670 			action->pkt_reformat =
2671 				maction->flow_action_raw.pkt_reformat;
2672 			return 0;
2673 		}
2674 		/* fall through */
2675 	default:
2676 		return -EOPNOTSUPP;
2677 	}
2678 }
2679 
2680 static int parse_flow_attr(struct mlx5_core_dev *mdev,
2681 			   struct mlx5_flow_spec *spec,
2682 			   const union ib_flow_spec *ib_spec,
2683 			   const struct ib_flow_attr *flow_attr,
2684 			   struct mlx5_flow_act *action, u32 prev_type)
2685 {
2686 	struct mlx5_flow_context *flow_context = &spec->flow_context;
2687 	u32 *match_c = spec->match_criteria;
2688 	u32 *match_v = spec->match_value;
2689 	void *misc_params_c = MLX5_ADDR_OF(fte_match_param, match_c,
2690 					   misc_parameters);
2691 	void *misc_params_v = MLX5_ADDR_OF(fte_match_param, match_v,
2692 					   misc_parameters);
2693 	void *misc_params2_c = MLX5_ADDR_OF(fte_match_param, match_c,
2694 					    misc_parameters_2);
2695 	void *misc_params2_v = MLX5_ADDR_OF(fte_match_param, match_v,
2696 					    misc_parameters_2);
2697 	void *headers_c;
2698 	void *headers_v;
2699 	int match_ipv;
2700 	int ret;
2701 
2702 	if (ib_spec->type & IB_FLOW_SPEC_INNER) {
2703 		headers_c = MLX5_ADDR_OF(fte_match_param, match_c,
2704 					 inner_headers);
2705 		headers_v = MLX5_ADDR_OF(fte_match_param, match_v,
2706 					 inner_headers);
2707 		match_ipv = MLX5_CAP_FLOWTABLE_NIC_RX(mdev,
2708 					ft_field_support.inner_ip_version);
2709 	} else {
2710 		headers_c = MLX5_ADDR_OF(fte_match_param, match_c,
2711 					 outer_headers);
2712 		headers_v = MLX5_ADDR_OF(fte_match_param, match_v,
2713 					 outer_headers);
2714 		match_ipv = MLX5_CAP_FLOWTABLE_NIC_RX(mdev,
2715 					ft_field_support.outer_ip_version);
2716 	}
2717 
2718 	switch (ib_spec->type & ~IB_FLOW_SPEC_INNER) {
2719 	case IB_FLOW_SPEC_ETH:
2720 		if (FIELDS_NOT_SUPPORTED(ib_spec->eth.mask, LAST_ETH_FIELD))
2721 			return -EOPNOTSUPP;
2722 
2723 		ether_addr_copy(MLX5_ADDR_OF(fte_match_set_lyr_2_4, headers_c,
2724 					     dmac_47_16),
2725 				ib_spec->eth.mask.dst_mac);
2726 		ether_addr_copy(MLX5_ADDR_OF(fte_match_set_lyr_2_4, headers_v,
2727 					     dmac_47_16),
2728 				ib_spec->eth.val.dst_mac);
2729 
2730 		ether_addr_copy(MLX5_ADDR_OF(fte_match_set_lyr_2_4, headers_c,
2731 					     smac_47_16),
2732 				ib_spec->eth.mask.src_mac);
2733 		ether_addr_copy(MLX5_ADDR_OF(fte_match_set_lyr_2_4, headers_v,
2734 					     smac_47_16),
2735 				ib_spec->eth.val.src_mac);
2736 
2737 		if (ib_spec->eth.mask.vlan_tag) {
2738 			MLX5_SET(fte_match_set_lyr_2_4, headers_c,
2739 				 cvlan_tag, 1);
2740 			MLX5_SET(fte_match_set_lyr_2_4, headers_v,
2741 				 cvlan_tag, 1);
2742 
2743 			MLX5_SET(fte_match_set_lyr_2_4, headers_c,
2744 				 first_vid, ntohs(ib_spec->eth.mask.vlan_tag));
2745 			MLX5_SET(fte_match_set_lyr_2_4, headers_v,
2746 				 first_vid, ntohs(ib_spec->eth.val.vlan_tag));
2747 
2748 			MLX5_SET(fte_match_set_lyr_2_4, headers_c,
2749 				 first_cfi,
2750 				 ntohs(ib_spec->eth.mask.vlan_tag) >> 12);
2751 			MLX5_SET(fte_match_set_lyr_2_4, headers_v,
2752 				 first_cfi,
2753 				 ntohs(ib_spec->eth.val.vlan_tag) >> 12);
2754 
2755 			MLX5_SET(fte_match_set_lyr_2_4, headers_c,
2756 				 first_prio,
2757 				 ntohs(ib_spec->eth.mask.vlan_tag) >> 13);
2758 			MLX5_SET(fte_match_set_lyr_2_4, headers_v,
2759 				 first_prio,
2760 				 ntohs(ib_spec->eth.val.vlan_tag) >> 13);
2761 		}
2762 		MLX5_SET(fte_match_set_lyr_2_4, headers_c,
2763 			 ethertype, ntohs(ib_spec->eth.mask.ether_type));
2764 		MLX5_SET(fte_match_set_lyr_2_4, headers_v,
2765 			 ethertype, ntohs(ib_spec->eth.val.ether_type));
2766 		break;
2767 	case IB_FLOW_SPEC_IPV4:
2768 		if (FIELDS_NOT_SUPPORTED(ib_spec->ipv4.mask, LAST_IPV4_FIELD))
2769 			return -EOPNOTSUPP;
2770 
2771 		if (match_ipv) {
2772 			MLX5_SET(fte_match_set_lyr_2_4, headers_c,
2773 				 ip_version, 0xf);
2774 			MLX5_SET(fte_match_set_lyr_2_4, headers_v,
2775 				 ip_version, MLX5_FS_IPV4_VERSION);
2776 		} else {
2777 			MLX5_SET(fte_match_set_lyr_2_4, headers_c,
2778 				 ethertype, 0xffff);
2779 			MLX5_SET(fte_match_set_lyr_2_4, headers_v,
2780 				 ethertype, ETH_P_IP);
2781 		}
2782 
2783 		memcpy(MLX5_ADDR_OF(fte_match_set_lyr_2_4, headers_c,
2784 				    src_ipv4_src_ipv6.ipv4_layout.ipv4),
2785 		       &ib_spec->ipv4.mask.src_ip,
2786 		       sizeof(ib_spec->ipv4.mask.src_ip));
2787 		memcpy(MLX5_ADDR_OF(fte_match_set_lyr_2_4, headers_v,
2788 				    src_ipv4_src_ipv6.ipv4_layout.ipv4),
2789 		       &ib_spec->ipv4.val.src_ip,
2790 		       sizeof(ib_spec->ipv4.val.src_ip));
2791 		memcpy(MLX5_ADDR_OF(fte_match_set_lyr_2_4, headers_c,
2792 				    dst_ipv4_dst_ipv6.ipv4_layout.ipv4),
2793 		       &ib_spec->ipv4.mask.dst_ip,
2794 		       sizeof(ib_spec->ipv4.mask.dst_ip));
2795 		memcpy(MLX5_ADDR_OF(fte_match_set_lyr_2_4, headers_v,
2796 				    dst_ipv4_dst_ipv6.ipv4_layout.ipv4),
2797 		       &ib_spec->ipv4.val.dst_ip,
2798 		       sizeof(ib_spec->ipv4.val.dst_ip));
2799 
2800 		set_tos(headers_c, headers_v,
2801 			ib_spec->ipv4.mask.tos, ib_spec->ipv4.val.tos);
2802 
2803 		if (set_proto(headers_c, headers_v,
2804 			      ib_spec->ipv4.mask.proto,
2805 			      ib_spec->ipv4.val.proto))
2806 			return -EINVAL;
2807 		break;
2808 	case IB_FLOW_SPEC_IPV6:
2809 		if (FIELDS_NOT_SUPPORTED(ib_spec->ipv6.mask, LAST_IPV6_FIELD))
2810 			return -EOPNOTSUPP;
2811 
2812 		if (match_ipv) {
2813 			MLX5_SET(fte_match_set_lyr_2_4, headers_c,
2814 				 ip_version, 0xf);
2815 			MLX5_SET(fte_match_set_lyr_2_4, headers_v,
2816 				 ip_version, MLX5_FS_IPV6_VERSION);
2817 		} else {
2818 			MLX5_SET(fte_match_set_lyr_2_4, headers_c,
2819 				 ethertype, 0xffff);
2820 			MLX5_SET(fte_match_set_lyr_2_4, headers_v,
2821 				 ethertype, ETH_P_IPV6);
2822 		}
2823 
2824 		memcpy(MLX5_ADDR_OF(fte_match_set_lyr_2_4, headers_c,
2825 				    src_ipv4_src_ipv6.ipv6_layout.ipv6),
2826 		       &ib_spec->ipv6.mask.src_ip,
2827 		       sizeof(ib_spec->ipv6.mask.src_ip));
2828 		memcpy(MLX5_ADDR_OF(fte_match_set_lyr_2_4, headers_v,
2829 				    src_ipv4_src_ipv6.ipv6_layout.ipv6),
2830 		       &ib_spec->ipv6.val.src_ip,
2831 		       sizeof(ib_spec->ipv6.val.src_ip));
2832 		memcpy(MLX5_ADDR_OF(fte_match_set_lyr_2_4, headers_c,
2833 				    dst_ipv4_dst_ipv6.ipv6_layout.ipv6),
2834 		       &ib_spec->ipv6.mask.dst_ip,
2835 		       sizeof(ib_spec->ipv6.mask.dst_ip));
2836 		memcpy(MLX5_ADDR_OF(fte_match_set_lyr_2_4, headers_v,
2837 				    dst_ipv4_dst_ipv6.ipv6_layout.ipv6),
2838 		       &ib_spec->ipv6.val.dst_ip,
2839 		       sizeof(ib_spec->ipv6.val.dst_ip));
2840 
2841 		set_tos(headers_c, headers_v,
2842 			ib_spec->ipv6.mask.traffic_class,
2843 			ib_spec->ipv6.val.traffic_class);
2844 
2845 		if (set_proto(headers_c, headers_v,
2846 			      ib_spec->ipv6.mask.next_hdr,
2847 			      ib_spec->ipv6.val.next_hdr))
2848 			return -EINVAL;
2849 
2850 		set_flow_label(misc_params_c, misc_params_v,
2851 			       ntohl(ib_spec->ipv6.mask.flow_label),
2852 			       ntohl(ib_spec->ipv6.val.flow_label),
2853 			       ib_spec->type & IB_FLOW_SPEC_INNER);
2854 		break;
2855 	case IB_FLOW_SPEC_ESP:
2856 		if (ib_spec->esp.mask.seq)
2857 			return -EOPNOTSUPP;
2858 
2859 		MLX5_SET(fte_match_set_misc, misc_params_c, outer_esp_spi,
2860 			 ntohl(ib_spec->esp.mask.spi));
2861 		MLX5_SET(fte_match_set_misc, misc_params_v, outer_esp_spi,
2862 			 ntohl(ib_spec->esp.val.spi));
2863 		break;
2864 	case IB_FLOW_SPEC_TCP:
2865 		if (FIELDS_NOT_SUPPORTED(ib_spec->tcp_udp.mask,
2866 					 LAST_TCP_UDP_FIELD))
2867 			return -EOPNOTSUPP;
2868 
2869 		if (set_proto(headers_c, headers_v, 0xff, IPPROTO_TCP))
2870 			return -EINVAL;
2871 
2872 		MLX5_SET(fte_match_set_lyr_2_4, headers_c, tcp_sport,
2873 			 ntohs(ib_spec->tcp_udp.mask.src_port));
2874 		MLX5_SET(fte_match_set_lyr_2_4, headers_v, tcp_sport,
2875 			 ntohs(ib_spec->tcp_udp.val.src_port));
2876 
2877 		MLX5_SET(fte_match_set_lyr_2_4, headers_c, tcp_dport,
2878 			 ntohs(ib_spec->tcp_udp.mask.dst_port));
2879 		MLX5_SET(fte_match_set_lyr_2_4, headers_v, tcp_dport,
2880 			 ntohs(ib_spec->tcp_udp.val.dst_port));
2881 		break;
2882 	case IB_FLOW_SPEC_UDP:
2883 		if (FIELDS_NOT_SUPPORTED(ib_spec->tcp_udp.mask,
2884 					 LAST_TCP_UDP_FIELD))
2885 			return -EOPNOTSUPP;
2886 
2887 		if (set_proto(headers_c, headers_v, 0xff, IPPROTO_UDP))
2888 			return -EINVAL;
2889 
2890 		MLX5_SET(fte_match_set_lyr_2_4, headers_c, udp_sport,
2891 			 ntohs(ib_spec->tcp_udp.mask.src_port));
2892 		MLX5_SET(fte_match_set_lyr_2_4, headers_v, udp_sport,
2893 			 ntohs(ib_spec->tcp_udp.val.src_port));
2894 
2895 		MLX5_SET(fte_match_set_lyr_2_4, headers_c, udp_dport,
2896 			 ntohs(ib_spec->tcp_udp.mask.dst_port));
2897 		MLX5_SET(fte_match_set_lyr_2_4, headers_v, udp_dport,
2898 			 ntohs(ib_spec->tcp_udp.val.dst_port));
2899 		break;
2900 	case IB_FLOW_SPEC_GRE:
2901 		if (ib_spec->gre.mask.c_ks_res0_ver)
2902 			return -EOPNOTSUPP;
2903 
2904 		if (set_proto(headers_c, headers_v, 0xff, IPPROTO_GRE))
2905 			return -EINVAL;
2906 
2907 		MLX5_SET(fte_match_set_lyr_2_4, headers_c, ip_protocol,
2908 			 0xff);
2909 		MLX5_SET(fte_match_set_lyr_2_4, headers_v, ip_protocol,
2910 			 IPPROTO_GRE);
2911 
2912 		MLX5_SET(fte_match_set_misc, misc_params_c, gre_protocol,
2913 			 ntohs(ib_spec->gre.mask.protocol));
2914 		MLX5_SET(fte_match_set_misc, misc_params_v, gre_protocol,
2915 			 ntohs(ib_spec->gre.val.protocol));
2916 
2917 		memcpy(MLX5_ADDR_OF(fte_match_set_misc, misc_params_c,
2918 				    gre_key.nvgre.hi),
2919 		       &ib_spec->gre.mask.key,
2920 		       sizeof(ib_spec->gre.mask.key));
2921 		memcpy(MLX5_ADDR_OF(fte_match_set_misc, misc_params_v,
2922 				    gre_key.nvgre.hi),
2923 		       &ib_spec->gre.val.key,
2924 		       sizeof(ib_spec->gre.val.key));
2925 		break;
2926 	case IB_FLOW_SPEC_MPLS:
2927 		switch (prev_type) {
2928 		case IB_FLOW_SPEC_UDP:
2929 			if (check_mpls_supp_fields(MLX5_CAP_FLOWTABLE_NIC_RX(mdev,
2930 						   ft_field_support.outer_first_mpls_over_udp),
2931 						   &ib_spec->mpls.mask.tag))
2932 				return -EOPNOTSUPP;
2933 
2934 			memcpy(MLX5_ADDR_OF(fte_match_set_misc2, misc_params2_v,
2935 					    outer_first_mpls_over_udp),
2936 			       &ib_spec->mpls.val.tag,
2937 			       sizeof(ib_spec->mpls.val.tag));
2938 			memcpy(MLX5_ADDR_OF(fte_match_set_misc2, misc_params2_c,
2939 					    outer_first_mpls_over_udp),
2940 			       &ib_spec->mpls.mask.tag,
2941 			       sizeof(ib_spec->mpls.mask.tag));
2942 			break;
2943 		case IB_FLOW_SPEC_GRE:
2944 			if (check_mpls_supp_fields(MLX5_CAP_FLOWTABLE_NIC_RX(mdev,
2945 						   ft_field_support.outer_first_mpls_over_gre),
2946 						   &ib_spec->mpls.mask.tag))
2947 				return -EOPNOTSUPP;
2948 
2949 			memcpy(MLX5_ADDR_OF(fte_match_set_misc2, misc_params2_v,
2950 					    outer_first_mpls_over_gre),
2951 			       &ib_spec->mpls.val.tag,
2952 			       sizeof(ib_spec->mpls.val.tag));
2953 			memcpy(MLX5_ADDR_OF(fte_match_set_misc2, misc_params2_c,
2954 					    outer_first_mpls_over_gre),
2955 			       &ib_spec->mpls.mask.tag,
2956 			       sizeof(ib_spec->mpls.mask.tag));
2957 			break;
2958 		default:
2959 			if (ib_spec->type & IB_FLOW_SPEC_INNER) {
2960 				if (check_mpls_supp_fields(MLX5_CAP_FLOWTABLE_NIC_RX(mdev,
2961 							   ft_field_support.inner_first_mpls),
2962 							   &ib_spec->mpls.mask.tag))
2963 					return -EOPNOTSUPP;
2964 
2965 				memcpy(MLX5_ADDR_OF(fte_match_set_misc2, misc_params2_v,
2966 						    inner_first_mpls),
2967 				       &ib_spec->mpls.val.tag,
2968 				       sizeof(ib_spec->mpls.val.tag));
2969 				memcpy(MLX5_ADDR_OF(fte_match_set_misc2, misc_params2_c,
2970 						    inner_first_mpls),
2971 				       &ib_spec->mpls.mask.tag,
2972 				       sizeof(ib_spec->mpls.mask.tag));
2973 			} else {
2974 				if (check_mpls_supp_fields(MLX5_CAP_FLOWTABLE_NIC_RX(mdev,
2975 							   ft_field_support.outer_first_mpls),
2976 							   &ib_spec->mpls.mask.tag))
2977 					return -EOPNOTSUPP;
2978 
2979 				memcpy(MLX5_ADDR_OF(fte_match_set_misc2, misc_params2_v,
2980 						    outer_first_mpls),
2981 				       &ib_spec->mpls.val.tag,
2982 				       sizeof(ib_spec->mpls.val.tag));
2983 				memcpy(MLX5_ADDR_OF(fte_match_set_misc2, misc_params2_c,
2984 						    outer_first_mpls),
2985 				       &ib_spec->mpls.mask.tag,
2986 				       sizeof(ib_spec->mpls.mask.tag));
2987 			}
2988 		}
2989 		break;
2990 	case IB_FLOW_SPEC_VXLAN_TUNNEL:
2991 		if (FIELDS_NOT_SUPPORTED(ib_spec->tunnel.mask,
2992 					 LAST_TUNNEL_FIELD))
2993 			return -EOPNOTSUPP;
2994 
2995 		MLX5_SET(fte_match_set_misc, misc_params_c, vxlan_vni,
2996 			 ntohl(ib_spec->tunnel.mask.tunnel_id));
2997 		MLX5_SET(fte_match_set_misc, misc_params_v, vxlan_vni,
2998 			 ntohl(ib_spec->tunnel.val.tunnel_id));
2999 		break;
3000 	case IB_FLOW_SPEC_ACTION_TAG:
3001 		if (FIELDS_NOT_SUPPORTED(ib_spec->flow_tag,
3002 					 LAST_FLOW_TAG_FIELD))
3003 			return -EOPNOTSUPP;
3004 		if (ib_spec->flow_tag.tag_id >= BIT(24))
3005 			return -EINVAL;
3006 
3007 		flow_context->flow_tag = ib_spec->flow_tag.tag_id;
3008 		flow_context->flags |= FLOW_CONTEXT_HAS_TAG;
3009 		break;
3010 	case IB_FLOW_SPEC_ACTION_DROP:
3011 		if (FIELDS_NOT_SUPPORTED(ib_spec->drop,
3012 					 LAST_DROP_FIELD))
3013 			return -EOPNOTSUPP;
3014 		action->action |= MLX5_FLOW_CONTEXT_ACTION_DROP;
3015 		break;
3016 	case IB_FLOW_SPEC_ACTION_HANDLE:
3017 		ret = parse_flow_flow_action(to_mflow_act(ib_spec->action.act),
3018 			flow_attr->flags & IB_FLOW_ATTR_FLAGS_EGRESS, action);
3019 		if (ret)
3020 			return ret;
3021 		break;
3022 	case IB_FLOW_SPEC_ACTION_COUNT:
3023 		if (FIELDS_NOT_SUPPORTED(ib_spec->flow_count,
3024 					 LAST_COUNTERS_FIELD))
3025 			return -EOPNOTSUPP;
3026 
3027 		/* for now support only one counters spec per flow */
3028 		if (action->action & MLX5_FLOW_CONTEXT_ACTION_COUNT)
3029 			return -EINVAL;
3030 
3031 		action->counters = ib_spec->flow_count.counters;
3032 		action->action |= MLX5_FLOW_CONTEXT_ACTION_COUNT;
3033 		break;
3034 	default:
3035 		return -EINVAL;
3036 	}
3037 
3038 	return 0;
3039 }
3040 
3041 /* If a flow could catch both multicast and unicast packets,
3042  * it won't fall into the multicast flow steering table and this rule
3043  * could steal other multicast packets.
3044  */
3045 static bool flow_is_multicast_only(const struct ib_flow_attr *ib_attr)
3046 {
3047 	union ib_flow_spec *flow_spec;
3048 
3049 	if (ib_attr->type != IB_FLOW_ATTR_NORMAL ||
3050 	    ib_attr->num_of_specs < 1)
3051 		return false;
3052 
3053 	flow_spec = (union ib_flow_spec *)(ib_attr + 1);
3054 	if (flow_spec->type == IB_FLOW_SPEC_IPV4) {
3055 		struct ib_flow_spec_ipv4 *ipv4_spec;
3056 
3057 		ipv4_spec = (struct ib_flow_spec_ipv4 *)flow_spec;
3058 		if (ipv4_is_multicast(ipv4_spec->val.dst_ip))
3059 			return true;
3060 
3061 		return false;
3062 	}
3063 
3064 	if (flow_spec->type == IB_FLOW_SPEC_ETH) {
3065 		struct ib_flow_spec_eth *eth_spec;
3066 
3067 		eth_spec = (struct ib_flow_spec_eth *)flow_spec;
3068 		return is_multicast_ether_addr(eth_spec->mask.dst_mac) &&
3069 		       is_multicast_ether_addr(eth_spec->val.dst_mac);
3070 	}
3071 
3072 	return false;
3073 }
3074 
3075 enum valid_spec {
3076 	VALID_SPEC_INVALID,
3077 	VALID_SPEC_VALID,
3078 	VALID_SPEC_NA,
3079 };
3080 
3081 static enum valid_spec
3082 is_valid_esp_aes_gcm(struct mlx5_core_dev *mdev,
3083 		     const struct mlx5_flow_spec *spec,
3084 		     const struct mlx5_flow_act *flow_act,
3085 		     bool egress)
3086 {
3087 	const u32 *match_c = spec->match_criteria;
3088 	bool is_crypto =
3089 		(flow_act->action & (MLX5_FLOW_CONTEXT_ACTION_ENCRYPT |
3090 				     MLX5_FLOW_CONTEXT_ACTION_DECRYPT));
3091 	bool is_ipsec = mlx5_fs_is_ipsec_flow(match_c);
3092 	bool is_drop = flow_act->action & MLX5_FLOW_CONTEXT_ACTION_DROP;
3093 
3094 	/*
3095 	 * Currently only crypto is supported in egress, when regular egress
3096 	 * rules would be supported, always return VALID_SPEC_NA.
3097 	 */
3098 	if (!is_crypto)
3099 		return VALID_SPEC_NA;
3100 
3101 	return is_crypto && is_ipsec &&
3102 		(!egress || (!is_drop &&
3103 			     !(spec->flow_context.flags & FLOW_CONTEXT_HAS_TAG))) ?
3104 		VALID_SPEC_VALID : VALID_SPEC_INVALID;
3105 }
3106 
3107 static bool is_valid_spec(struct mlx5_core_dev *mdev,
3108 			  const struct mlx5_flow_spec *spec,
3109 			  const struct mlx5_flow_act *flow_act,
3110 			  bool egress)
3111 {
3112 	/* We curretly only support ipsec egress flow */
3113 	return is_valid_esp_aes_gcm(mdev, spec, flow_act, egress) != VALID_SPEC_INVALID;
3114 }
3115 
3116 static bool is_valid_ethertype(struct mlx5_core_dev *mdev,
3117 			       const struct ib_flow_attr *flow_attr,
3118 			       bool check_inner)
3119 {
3120 	union ib_flow_spec *ib_spec = (union ib_flow_spec *)(flow_attr + 1);
3121 	int match_ipv = check_inner ?
3122 			MLX5_CAP_FLOWTABLE_NIC_RX(mdev,
3123 					ft_field_support.inner_ip_version) :
3124 			MLX5_CAP_FLOWTABLE_NIC_RX(mdev,
3125 					ft_field_support.outer_ip_version);
3126 	int inner_bit = check_inner ? IB_FLOW_SPEC_INNER : 0;
3127 	bool ipv4_spec_valid, ipv6_spec_valid;
3128 	unsigned int ip_spec_type = 0;
3129 	bool has_ethertype = false;
3130 	unsigned int spec_index;
3131 	bool mask_valid = true;
3132 	u16 eth_type = 0;
3133 	bool type_valid;
3134 
3135 	/* Validate that ethertype is correct */
3136 	for (spec_index = 0; spec_index < flow_attr->num_of_specs; spec_index++) {
3137 		if ((ib_spec->type == (IB_FLOW_SPEC_ETH | inner_bit)) &&
3138 		    ib_spec->eth.mask.ether_type) {
3139 			mask_valid = (ib_spec->eth.mask.ether_type ==
3140 				      htons(0xffff));
3141 			has_ethertype = true;
3142 			eth_type = ntohs(ib_spec->eth.val.ether_type);
3143 		} else if ((ib_spec->type == (IB_FLOW_SPEC_IPV4 | inner_bit)) ||
3144 			   (ib_spec->type == (IB_FLOW_SPEC_IPV6 | inner_bit))) {
3145 			ip_spec_type = ib_spec->type;
3146 		}
3147 		ib_spec = (void *)ib_spec + ib_spec->size;
3148 	}
3149 
3150 	type_valid = (!has_ethertype) || (!ip_spec_type);
3151 	if (!type_valid && mask_valid) {
3152 		ipv4_spec_valid = (eth_type == ETH_P_IP) &&
3153 			(ip_spec_type == (IB_FLOW_SPEC_IPV4 | inner_bit));
3154 		ipv6_spec_valid = (eth_type == ETH_P_IPV6) &&
3155 			(ip_spec_type == (IB_FLOW_SPEC_IPV6 | inner_bit));
3156 
3157 		type_valid = (ipv4_spec_valid) || (ipv6_spec_valid) ||
3158 			     (((eth_type == ETH_P_MPLS_UC) ||
3159 			       (eth_type == ETH_P_MPLS_MC)) && match_ipv);
3160 	}
3161 
3162 	return type_valid;
3163 }
3164 
3165 static bool is_valid_attr(struct mlx5_core_dev *mdev,
3166 			  const struct ib_flow_attr *flow_attr)
3167 {
3168 	return is_valid_ethertype(mdev, flow_attr, false) &&
3169 	       is_valid_ethertype(mdev, flow_attr, true);
3170 }
3171 
3172 static void put_flow_table(struct mlx5_ib_dev *dev,
3173 			   struct mlx5_ib_flow_prio *prio, bool ft_added)
3174 {
3175 	prio->refcount -= !!ft_added;
3176 	if (!prio->refcount) {
3177 		mlx5_destroy_flow_table(prio->flow_table);
3178 		prio->flow_table = NULL;
3179 	}
3180 }
3181 
3182 static void counters_clear_description(struct ib_counters *counters)
3183 {
3184 	struct mlx5_ib_mcounters *mcounters = to_mcounters(counters);
3185 
3186 	mutex_lock(&mcounters->mcntrs_mutex);
3187 	kfree(mcounters->counters_data);
3188 	mcounters->counters_data = NULL;
3189 	mcounters->cntrs_max_index = 0;
3190 	mutex_unlock(&mcounters->mcntrs_mutex);
3191 }
3192 
3193 static int mlx5_ib_destroy_flow(struct ib_flow *flow_id)
3194 {
3195 	struct mlx5_ib_flow_handler *handler = container_of(flow_id,
3196 							  struct mlx5_ib_flow_handler,
3197 							  ibflow);
3198 	struct mlx5_ib_flow_handler *iter, *tmp;
3199 	struct mlx5_ib_dev *dev = handler->dev;
3200 
3201 	mutex_lock(&dev->flow_db->lock);
3202 
3203 	list_for_each_entry_safe(iter, tmp, &handler->list, list) {
3204 		mlx5_del_flow_rules(iter->rule);
3205 		put_flow_table(dev, iter->prio, true);
3206 		list_del(&iter->list);
3207 		kfree(iter);
3208 	}
3209 
3210 	mlx5_del_flow_rules(handler->rule);
3211 	put_flow_table(dev, handler->prio, true);
3212 	if (handler->ibcounters &&
3213 	    atomic_read(&handler->ibcounters->usecnt) == 1)
3214 		counters_clear_description(handler->ibcounters);
3215 
3216 	mutex_unlock(&dev->flow_db->lock);
3217 	if (handler->flow_matcher)
3218 		atomic_dec(&handler->flow_matcher->usecnt);
3219 	kfree(handler);
3220 
3221 	return 0;
3222 }
3223 
3224 static int ib_prio_to_core_prio(unsigned int priority, bool dont_trap)
3225 {
3226 	priority *= 2;
3227 	if (!dont_trap)
3228 		priority++;
3229 	return priority;
3230 }
3231 
3232 enum flow_table_type {
3233 	MLX5_IB_FT_RX,
3234 	MLX5_IB_FT_TX
3235 };
3236 
3237 #define MLX5_FS_MAX_TYPES	 6
3238 #define MLX5_FS_MAX_ENTRIES	 BIT(16)
3239 
3240 static struct mlx5_ib_flow_prio *_get_prio(struct mlx5_flow_namespace *ns,
3241 					   struct mlx5_ib_flow_prio *prio,
3242 					   int priority,
3243 					   int num_entries, int num_groups,
3244 					   u32 flags)
3245 {
3246 	struct mlx5_flow_table *ft;
3247 
3248 	ft = mlx5_create_auto_grouped_flow_table(ns, priority,
3249 						 num_entries,
3250 						 num_groups,
3251 						 0, flags);
3252 	if (IS_ERR(ft))
3253 		return ERR_CAST(ft);
3254 
3255 	prio->flow_table = ft;
3256 	prio->refcount = 0;
3257 	return prio;
3258 }
3259 
3260 static struct mlx5_ib_flow_prio *get_flow_table(struct mlx5_ib_dev *dev,
3261 						struct ib_flow_attr *flow_attr,
3262 						enum flow_table_type ft_type)
3263 {
3264 	bool dont_trap = flow_attr->flags & IB_FLOW_ATTR_FLAGS_DONT_TRAP;
3265 	struct mlx5_flow_namespace *ns = NULL;
3266 	struct mlx5_ib_flow_prio *prio;
3267 	struct mlx5_flow_table *ft;
3268 	int max_table_size;
3269 	int num_entries;
3270 	int num_groups;
3271 	bool esw_encap;
3272 	u32 flags = 0;
3273 	int priority;
3274 
3275 	max_table_size = BIT(MLX5_CAP_FLOWTABLE_NIC_RX(dev->mdev,
3276 						       log_max_ft_size));
3277 	esw_encap = mlx5_eswitch_get_encap_mode(dev->mdev) !=
3278 		DEVLINK_ESWITCH_ENCAP_MODE_NONE;
3279 	if (flow_attr->type == IB_FLOW_ATTR_NORMAL) {
3280 		enum mlx5_flow_namespace_type fn_type;
3281 
3282 		if (flow_is_multicast_only(flow_attr) &&
3283 		    !dont_trap)
3284 			priority = MLX5_IB_FLOW_MCAST_PRIO;
3285 		else
3286 			priority = ib_prio_to_core_prio(flow_attr->priority,
3287 							dont_trap);
3288 		if (ft_type == MLX5_IB_FT_RX) {
3289 			fn_type = MLX5_FLOW_NAMESPACE_BYPASS;
3290 			prio = &dev->flow_db->prios[priority];
3291 			if (!dev->is_rep && !esw_encap &&
3292 			    MLX5_CAP_FLOWTABLE_NIC_RX(dev->mdev, decap))
3293 				flags |= MLX5_FLOW_TABLE_TUNNEL_EN_DECAP;
3294 			if (!dev->is_rep && !esw_encap &&
3295 			    MLX5_CAP_FLOWTABLE_NIC_RX(dev->mdev,
3296 					reformat_l3_tunnel_to_l2))
3297 				flags |= MLX5_FLOW_TABLE_TUNNEL_EN_REFORMAT;
3298 		} else {
3299 			max_table_size =
3300 				BIT(MLX5_CAP_FLOWTABLE_NIC_TX(dev->mdev,
3301 							      log_max_ft_size));
3302 			fn_type = MLX5_FLOW_NAMESPACE_EGRESS;
3303 			prio = &dev->flow_db->egress_prios[priority];
3304 			if (!dev->is_rep && !esw_encap &&
3305 			    MLX5_CAP_FLOWTABLE_NIC_TX(dev->mdev, reformat))
3306 				flags |= MLX5_FLOW_TABLE_TUNNEL_EN_REFORMAT;
3307 		}
3308 		ns = mlx5_get_flow_namespace(dev->mdev, fn_type);
3309 		num_entries = MLX5_FS_MAX_ENTRIES;
3310 		num_groups = MLX5_FS_MAX_TYPES;
3311 	} else if (flow_attr->type == IB_FLOW_ATTR_ALL_DEFAULT ||
3312 		   flow_attr->type == IB_FLOW_ATTR_MC_DEFAULT) {
3313 		ns = mlx5_get_flow_namespace(dev->mdev,
3314 					     MLX5_FLOW_NAMESPACE_LEFTOVERS);
3315 		build_leftovers_ft_param(&priority,
3316 					 &num_entries,
3317 					 &num_groups);
3318 		prio = &dev->flow_db->prios[MLX5_IB_FLOW_LEFTOVERS_PRIO];
3319 	} else if (flow_attr->type == IB_FLOW_ATTR_SNIFFER) {
3320 		if (!MLX5_CAP_FLOWTABLE(dev->mdev,
3321 					allow_sniffer_and_nic_rx_shared_tir))
3322 			return ERR_PTR(-ENOTSUPP);
3323 
3324 		ns = mlx5_get_flow_namespace(dev->mdev, ft_type == MLX5_IB_FT_RX ?
3325 					     MLX5_FLOW_NAMESPACE_SNIFFER_RX :
3326 					     MLX5_FLOW_NAMESPACE_SNIFFER_TX);
3327 
3328 		prio = &dev->flow_db->sniffer[ft_type];
3329 		priority = 0;
3330 		num_entries = 1;
3331 		num_groups = 1;
3332 	}
3333 
3334 	if (!ns)
3335 		return ERR_PTR(-ENOTSUPP);
3336 
3337 	max_table_size = min_t(int, num_entries, max_table_size);
3338 
3339 	ft = prio->flow_table;
3340 	if (!ft)
3341 		return _get_prio(ns, prio, priority, max_table_size, num_groups,
3342 				 flags);
3343 
3344 	return prio;
3345 }
3346 
3347 static void set_underlay_qp(struct mlx5_ib_dev *dev,
3348 			    struct mlx5_flow_spec *spec,
3349 			    u32 underlay_qpn)
3350 {
3351 	void *misc_params_c = MLX5_ADDR_OF(fte_match_param,
3352 					   spec->match_criteria,
3353 					   misc_parameters);
3354 	void *misc_params_v = MLX5_ADDR_OF(fte_match_param, spec->match_value,
3355 					   misc_parameters);
3356 
3357 	if (underlay_qpn &&
3358 	    MLX5_CAP_FLOWTABLE_NIC_RX(dev->mdev,
3359 				      ft_field_support.bth_dst_qp)) {
3360 		MLX5_SET(fte_match_set_misc,
3361 			 misc_params_v, bth_dst_qp, underlay_qpn);
3362 		MLX5_SET(fte_match_set_misc,
3363 			 misc_params_c, bth_dst_qp, 0xffffff);
3364 	}
3365 }
3366 
3367 static int read_flow_counters(struct ib_device *ibdev,
3368 			      struct mlx5_read_counters_attr *read_attr)
3369 {
3370 	struct mlx5_fc *fc = read_attr->hw_cntrs_hndl;
3371 	struct mlx5_ib_dev *dev = to_mdev(ibdev);
3372 
3373 	return mlx5_fc_query(dev->mdev, fc,
3374 			     &read_attr->out[IB_COUNTER_PACKETS],
3375 			     &read_attr->out[IB_COUNTER_BYTES]);
3376 }
3377 
3378 /* flow counters currently expose two counters packets and bytes */
3379 #define FLOW_COUNTERS_NUM 2
3380 static int counters_set_description(struct ib_counters *counters,
3381 				    enum mlx5_ib_counters_type counters_type,
3382 				    struct mlx5_ib_flow_counters_desc *desc_data,
3383 				    u32 ncounters)
3384 {
3385 	struct mlx5_ib_mcounters *mcounters = to_mcounters(counters);
3386 	u32 cntrs_max_index = 0;
3387 	int i;
3388 
3389 	if (counters_type != MLX5_IB_COUNTERS_FLOW)
3390 		return -EINVAL;
3391 
3392 	/* init the fields for the object */
3393 	mcounters->type = counters_type;
3394 	mcounters->read_counters = read_flow_counters;
3395 	mcounters->counters_num = FLOW_COUNTERS_NUM;
3396 	mcounters->ncounters = ncounters;
3397 	/* each counter entry have both description and index pair */
3398 	for (i = 0; i < ncounters; i++) {
3399 		if (desc_data[i].description > IB_COUNTER_BYTES)
3400 			return -EINVAL;
3401 
3402 		if (cntrs_max_index <= desc_data[i].index)
3403 			cntrs_max_index = desc_data[i].index + 1;
3404 	}
3405 
3406 	mutex_lock(&mcounters->mcntrs_mutex);
3407 	mcounters->counters_data = desc_data;
3408 	mcounters->cntrs_max_index = cntrs_max_index;
3409 	mutex_unlock(&mcounters->mcntrs_mutex);
3410 
3411 	return 0;
3412 }
3413 
3414 #define MAX_COUNTERS_NUM (USHRT_MAX / (sizeof(u32) * 2))
3415 static int flow_counters_set_data(struct ib_counters *ibcounters,
3416 				  struct mlx5_ib_create_flow *ucmd)
3417 {
3418 	struct mlx5_ib_mcounters *mcounters = to_mcounters(ibcounters);
3419 	struct mlx5_ib_flow_counters_data *cntrs_data = NULL;
3420 	struct mlx5_ib_flow_counters_desc *desc_data = NULL;
3421 	bool hw_hndl = false;
3422 	int ret = 0;
3423 
3424 	if (ucmd && ucmd->ncounters_data != 0) {
3425 		cntrs_data = ucmd->data;
3426 		if (cntrs_data->ncounters > MAX_COUNTERS_NUM)
3427 			return -EINVAL;
3428 
3429 		desc_data = kcalloc(cntrs_data->ncounters,
3430 				    sizeof(*desc_data),
3431 				    GFP_KERNEL);
3432 		if (!desc_data)
3433 			return  -ENOMEM;
3434 
3435 		if (copy_from_user(desc_data,
3436 				   u64_to_user_ptr(cntrs_data->counters_data),
3437 				   sizeof(*desc_data) * cntrs_data->ncounters)) {
3438 			ret = -EFAULT;
3439 			goto free;
3440 		}
3441 	}
3442 
3443 	if (!mcounters->hw_cntrs_hndl) {
3444 		mcounters->hw_cntrs_hndl = mlx5_fc_create(
3445 			to_mdev(ibcounters->device)->mdev, false);
3446 		if (IS_ERR(mcounters->hw_cntrs_hndl)) {
3447 			ret = PTR_ERR(mcounters->hw_cntrs_hndl);
3448 			goto free;
3449 		}
3450 		hw_hndl = true;
3451 	}
3452 
3453 	if (desc_data) {
3454 		/* counters already bound to at least one flow */
3455 		if (mcounters->cntrs_max_index) {
3456 			ret = -EINVAL;
3457 			goto free_hndl;
3458 		}
3459 
3460 		ret = counters_set_description(ibcounters,
3461 					       MLX5_IB_COUNTERS_FLOW,
3462 					       desc_data,
3463 					       cntrs_data->ncounters);
3464 		if (ret)
3465 			goto free_hndl;
3466 
3467 	} else if (!mcounters->cntrs_max_index) {
3468 		/* counters not bound yet, must have udata passed */
3469 		ret = -EINVAL;
3470 		goto free_hndl;
3471 	}
3472 
3473 	return 0;
3474 
3475 free_hndl:
3476 	if (hw_hndl) {
3477 		mlx5_fc_destroy(to_mdev(ibcounters->device)->mdev,
3478 				mcounters->hw_cntrs_hndl);
3479 		mcounters->hw_cntrs_hndl = NULL;
3480 	}
3481 free:
3482 	kfree(desc_data);
3483 	return ret;
3484 }
3485 
3486 static void mlx5_ib_set_rule_source_port(struct mlx5_ib_dev *dev,
3487 					 struct mlx5_flow_spec *spec,
3488 					 struct mlx5_eswitch_rep *rep)
3489 {
3490 	struct mlx5_eswitch *esw = dev->mdev->priv.eswitch;
3491 	void *misc;
3492 
3493 	if (mlx5_eswitch_vport_match_metadata_enabled(esw)) {
3494 		misc = MLX5_ADDR_OF(fte_match_param, spec->match_value,
3495 				    misc_parameters_2);
3496 
3497 		MLX5_SET(fte_match_set_misc2, misc, metadata_reg_c_0,
3498 			 mlx5_eswitch_get_vport_metadata_for_match(esw,
3499 								   rep->vport));
3500 		misc = MLX5_ADDR_OF(fte_match_param, spec->match_criteria,
3501 				    misc_parameters_2);
3502 
3503 		MLX5_SET_TO_ONES(fte_match_set_misc2, misc, metadata_reg_c_0);
3504 	} else {
3505 		misc = MLX5_ADDR_OF(fte_match_param, spec->match_value,
3506 				    misc_parameters);
3507 
3508 		MLX5_SET(fte_match_set_misc, misc, source_port, rep->vport);
3509 
3510 		misc = MLX5_ADDR_OF(fte_match_param, spec->match_criteria,
3511 				    misc_parameters);
3512 
3513 		MLX5_SET_TO_ONES(fte_match_set_misc, misc, source_port);
3514 	}
3515 }
3516 
3517 static struct mlx5_ib_flow_handler *_create_flow_rule(struct mlx5_ib_dev *dev,
3518 						      struct mlx5_ib_flow_prio *ft_prio,
3519 						      const struct ib_flow_attr *flow_attr,
3520 						      struct mlx5_flow_destination *dst,
3521 						      u32 underlay_qpn,
3522 						      struct mlx5_ib_create_flow *ucmd)
3523 {
3524 	struct mlx5_flow_table	*ft = ft_prio->flow_table;
3525 	struct mlx5_ib_flow_handler *handler;
3526 	struct mlx5_flow_act flow_act = {};
3527 	struct mlx5_flow_spec *spec;
3528 	struct mlx5_flow_destination dest_arr[2] = {};
3529 	struct mlx5_flow_destination *rule_dst = dest_arr;
3530 	const void *ib_flow = (const void *)flow_attr + sizeof(*flow_attr);
3531 	unsigned int spec_index;
3532 	u32 prev_type = 0;
3533 	int err = 0;
3534 	int dest_num = 0;
3535 	bool is_egress = flow_attr->flags & IB_FLOW_ATTR_FLAGS_EGRESS;
3536 
3537 	if (!is_valid_attr(dev->mdev, flow_attr))
3538 		return ERR_PTR(-EINVAL);
3539 
3540 	if (dev->is_rep && is_egress)
3541 		return ERR_PTR(-EINVAL);
3542 
3543 	spec = kvzalloc(sizeof(*spec), GFP_KERNEL);
3544 	handler = kzalloc(sizeof(*handler), GFP_KERNEL);
3545 	if (!handler || !spec) {
3546 		err = -ENOMEM;
3547 		goto free;
3548 	}
3549 
3550 	INIT_LIST_HEAD(&handler->list);
3551 	if (dst) {
3552 		memcpy(&dest_arr[0], dst, sizeof(*dst));
3553 		dest_num++;
3554 	}
3555 
3556 	for (spec_index = 0; spec_index < flow_attr->num_of_specs; spec_index++) {
3557 		err = parse_flow_attr(dev->mdev, spec,
3558 				      ib_flow, flow_attr, &flow_act,
3559 				      prev_type);
3560 		if (err < 0)
3561 			goto free;
3562 
3563 		prev_type = ((union ib_flow_spec *)ib_flow)->type;
3564 		ib_flow += ((union ib_flow_spec *)ib_flow)->size;
3565 	}
3566 
3567 	if (!flow_is_multicast_only(flow_attr))
3568 		set_underlay_qp(dev, spec, underlay_qpn);
3569 
3570 	if (dev->is_rep) {
3571 		struct mlx5_eswitch_rep *rep;
3572 
3573 		rep = dev->port[flow_attr->port - 1].rep;
3574 		if (!rep) {
3575 			err = -EINVAL;
3576 			goto free;
3577 		}
3578 
3579 		mlx5_ib_set_rule_source_port(dev, spec, rep);
3580 	}
3581 
3582 	spec->match_criteria_enable = get_match_criteria_enable(spec->match_criteria);
3583 
3584 	if (is_egress &&
3585 	    !is_valid_spec(dev->mdev, spec, &flow_act, is_egress)) {
3586 		err = -EINVAL;
3587 		goto free;
3588 	}
3589 
3590 	if (flow_act.action & MLX5_FLOW_CONTEXT_ACTION_COUNT) {
3591 		struct mlx5_ib_mcounters *mcounters;
3592 
3593 		err = flow_counters_set_data(flow_act.counters, ucmd);
3594 		if (err)
3595 			goto free;
3596 
3597 		mcounters = to_mcounters(flow_act.counters);
3598 		handler->ibcounters = flow_act.counters;
3599 		dest_arr[dest_num].type =
3600 			MLX5_FLOW_DESTINATION_TYPE_COUNTER;
3601 		dest_arr[dest_num].counter_id =
3602 			mlx5_fc_id(mcounters->hw_cntrs_hndl);
3603 		dest_num++;
3604 	}
3605 
3606 	if (flow_act.action & MLX5_FLOW_CONTEXT_ACTION_DROP) {
3607 		if (!(flow_act.action & MLX5_FLOW_CONTEXT_ACTION_COUNT)) {
3608 			rule_dst = NULL;
3609 			dest_num = 0;
3610 		}
3611 	} else {
3612 		if (is_egress)
3613 			flow_act.action |= MLX5_FLOW_CONTEXT_ACTION_ALLOW;
3614 		else
3615 			flow_act.action |=
3616 				dest_num ?  MLX5_FLOW_CONTEXT_ACTION_FWD_DEST :
3617 					MLX5_FLOW_CONTEXT_ACTION_FWD_NEXT_PRIO;
3618 	}
3619 
3620 	if ((spec->flow_context.flags & FLOW_CONTEXT_HAS_TAG)  &&
3621 	    (flow_attr->type == IB_FLOW_ATTR_ALL_DEFAULT ||
3622 	     flow_attr->type == IB_FLOW_ATTR_MC_DEFAULT)) {
3623 		mlx5_ib_warn(dev, "Flow tag %u and attribute type %x isn't allowed in leftovers\n",
3624 			     spec->flow_context.flow_tag, flow_attr->type);
3625 		err = -EINVAL;
3626 		goto free;
3627 	}
3628 	handler->rule = mlx5_add_flow_rules(ft, spec,
3629 					    &flow_act,
3630 					    rule_dst, dest_num);
3631 
3632 	if (IS_ERR(handler->rule)) {
3633 		err = PTR_ERR(handler->rule);
3634 		goto free;
3635 	}
3636 
3637 	ft_prio->refcount++;
3638 	handler->prio = ft_prio;
3639 	handler->dev = dev;
3640 
3641 	ft_prio->flow_table = ft;
3642 free:
3643 	if (err && handler) {
3644 		if (handler->ibcounters &&
3645 		    atomic_read(&handler->ibcounters->usecnt) == 1)
3646 			counters_clear_description(handler->ibcounters);
3647 		kfree(handler);
3648 	}
3649 	kvfree(spec);
3650 	return err ? ERR_PTR(err) : handler;
3651 }
3652 
3653 static struct mlx5_ib_flow_handler *create_flow_rule(struct mlx5_ib_dev *dev,
3654 						     struct mlx5_ib_flow_prio *ft_prio,
3655 						     const struct ib_flow_attr *flow_attr,
3656 						     struct mlx5_flow_destination *dst)
3657 {
3658 	return _create_flow_rule(dev, ft_prio, flow_attr, dst, 0, NULL);
3659 }
3660 
3661 static struct mlx5_ib_flow_handler *create_dont_trap_rule(struct mlx5_ib_dev *dev,
3662 							  struct mlx5_ib_flow_prio *ft_prio,
3663 							  struct ib_flow_attr *flow_attr,
3664 							  struct mlx5_flow_destination *dst)
3665 {
3666 	struct mlx5_ib_flow_handler *handler_dst = NULL;
3667 	struct mlx5_ib_flow_handler *handler = NULL;
3668 
3669 	handler = create_flow_rule(dev, ft_prio, flow_attr, NULL);
3670 	if (!IS_ERR(handler)) {
3671 		handler_dst = create_flow_rule(dev, ft_prio,
3672 					       flow_attr, dst);
3673 		if (IS_ERR(handler_dst)) {
3674 			mlx5_del_flow_rules(handler->rule);
3675 			ft_prio->refcount--;
3676 			kfree(handler);
3677 			handler = handler_dst;
3678 		} else {
3679 			list_add(&handler_dst->list, &handler->list);
3680 		}
3681 	}
3682 
3683 	return handler;
3684 }
3685 enum {
3686 	LEFTOVERS_MC,
3687 	LEFTOVERS_UC,
3688 };
3689 
3690 static struct mlx5_ib_flow_handler *create_leftovers_rule(struct mlx5_ib_dev *dev,
3691 							  struct mlx5_ib_flow_prio *ft_prio,
3692 							  struct ib_flow_attr *flow_attr,
3693 							  struct mlx5_flow_destination *dst)
3694 {
3695 	struct mlx5_ib_flow_handler *handler_ucast = NULL;
3696 	struct mlx5_ib_flow_handler *handler = NULL;
3697 
3698 	static struct {
3699 		struct ib_flow_attr	flow_attr;
3700 		struct ib_flow_spec_eth eth_flow;
3701 	} leftovers_specs[] = {
3702 		[LEFTOVERS_MC] = {
3703 			.flow_attr = {
3704 				.num_of_specs = 1,
3705 				.size = sizeof(leftovers_specs[0])
3706 			},
3707 			.eth_flow = {
3708 				.type = IB_FLOW_SPEC_ETH,
3709 				.size = sizeof(struct ib_flow_spec_eth),
3710 				.mask = {.dst_mac = {0x1} },
3711 				.val =  {.dst_mac = {0x1} }
3712 			}
3713 		},
3714 		[LEFTOVERS_UC] = {
3715 			.flow_attr = {
3716 				.num_of_specs = 1,
3717 				.size = sizeof(leftovers_specs[0])
3718 			},
3719 			.eth_flow = {
3720 				.type = IB_FLOW_SPEC_ETH,
3721 				.size = sizeof(struct ib_flow_spec_eth),
3722 				.mask = {.dst_mac = {0x1} },
3723 				.val = {.dst_mac = {} }
3724 			}
3725 		}
3726 	};
3727 
3728 	handler = create_flow_rule(dev, ft_prio,
3729 				   &leftovers_specs[LEFTOVERS_MC].flow_attr,
3730 				   dst);
3731 	if (!IS_ERR(handler) &&
3732 	    flow_attr->type == IB_FLOW_ATTR_ALL_DEFAULT) {
3733 		handler_ucast = create_flow_rule(dev, ft_prio,
3734 						 &leftovers_specs[LEFTOVERS_UC].flow_attr,
3735 						 dst);
3736 		if (IS_ERR(handler_ucast)) {
3737 			mlx5_del_flow_rules(handler->rule);
3738 			ft_prio->refcount--;
3739 			kfree(handler);
3740 			handler = handler_ucast;
3741 		} else {
3742 			list_add(&handler_ucast->list, &handler->list);
3743 		}
3744 	}
3745 
3746 	return handler;
3747 }
3748 
3749 static struct mlx5_ib_flow_handler *create_sniffer_rule(struct mlx5_ib_dev *dev,
3750 							struct mlx5_ib_flow_prio *ft_rx,
3751 							struct mlx5_ib_flow_prio *ft_tx,
3752 							struct mlx5_flow_destination *dst)
3753 {
3754 	struct mlx5_ib_flow_handler *handler_rx;
3755 	struct mlx5_ib_flow_handler *handler_tx;
3756 	int err;
3757 	static const struct ib_flow_attr flow_attr  = {
3758 		.num_of_specs = 0,
3759 		.size = sizeof(flow_attr)
3760 	};
3761 
3762 	handler_rx = create_flow_rule(dev, ft_rx, &flow_attr, dst);
3763 	if (IS_ERR(handler_rx)) {
3764 		err = PTR_ERR(handler_rx);
3765 		goto err;
3766 	}
3767 
3768 	handler_tx = create_flow_rule(dev, ft_tx, &flow_attr, dst);
3769 	if (IS_ERR(handler_tx)) {
3770 		err = PTR_ERR(handler_tx);
3771 		goto err_tx;
3772 	}
3773 
3774 	list_add(&handler_tx->list, &handler_rx->list);
3775 
3776 	return handler_rx;
3777 
3778 err_tx:
3779 	mlx5_del_flow_rules(handler_rx->rule);
3780 	ft_rx->refcount--;
3781 	kfree(handler_rx);
3782 err:
3783 	return ERR_PTR(err);
3784 }
3785 
3786 static struct ib_flow *mlx5_ib_create_flow(struct ib_qp *qp,
3787 					   struct ib_flow_attr *flow_attr,
3788 					   int domain,
3789 					   struct ib_udata *udata)
3790 {
3791 	struct mlx5_ib_dev *dev = to_mdev(qp->device);
3792 	struct mlx5_ib_qp *mqp = to_mqp(qp);
3793 	struct mlx5_ib_flow_handler *handler = NULL;
3794 	struct mlx5_flow_destination *dst = NULL;
3795 	struct mlx5_ib_flow_prio *ft_prio_tx = NULL;
3796 	struct mlx5_ib_flow_prio *ft_prio;
3797 	bool is_egress = flow_attr->flags & IB_FLOW_ATTR_FLAGS_EGRESS;
3798 	struct mlx5_ib_create_flow *ucmd = NULL, ucmd_hdr;
3799 	size_t min_ucmd_sz, required_ucmd_sz;
3800 	int err;
3801 	int underlay_qpn;
3802 
3803 	if (udata && udata->inlen) {
3804 		min_ucmd_sz = offsetof(typeof(ucmd_hdr), reserved) +
3805 				sizeof(ucmd_hdr.reserved);
3806 		if (udata->inlen < min_ucmd_sz)
3807 			return ERR_PTR(-EOPNOTSUPP);
3808 
3809 		err = ib_copy_from_udata(&ucmd_hdr, udata, min_ucmd_sz);
3810 		if (err)
3811 			return ERR_PTR(err);
3812 
3813 		/* currently supports only one counters data */
3814 		if (ucmd_hdr.ncounters_data > 1)
3815 			return ERR_PTR(-EINVAL);
3816 
3817 		required_ucmd_sz = min_ucmd_sz +
3818 			sizeof(struct mlx5_ib_flow_counters_data) *
3819 			ucmd_hdr.ncounters_data;
3820 		if (udata->inlen > required_ucmd_sz &&
3821 		    !ib_is_udata_cleared(udata, required_ucmd_sz,
3822 					 udata->inlen - required_ucmd_sz))
3823 			return ERR_PTR(-EOPNOTSUPP);
3824 
3825 		ucmd = kzalloc(required_ucmd_sz, GFP_KERNEL);
3826 		if (!ucmd)
3827 			return ERR_PTR(-ENOMEM);
3828 
3829 		err = ib_copy_from_udata(ucmd, udata, required_ucmd_sz);
3830 		if (err)
3831 			goto free_ucmd;
3832 	}
3833 
3834 	if (flow_attr->priority > MLX5_IB_FLOW_LAST_PRIO) {
3835 		err = -ENOMEM;
3836 		goto free_ucmd;
3837 	}
3838 
3839 	if (domain != IB_FLOW_DOMAIN_USER ||
3840 	    flow_attr->port > dev->num_ports ||
3841 	    (flow_attr->flags & ~(IB_FLOW_ATTR_FLAGS_DONT_TRAP |
3842 				  IB_FLOW_ATTR_FLAGS_EGRESS))) {
3843 		err = -EINVAL;
3844 		goto free_ucmd;
3845 	}
3846 
3847 	if (is_egress &&
3848 	    (flow_attr->type == IB_FLOW_ATTR_ALL_DEFAULT ||
3849 	     flow_attr->type == IB_FLOW_ATTR_MC_DEFAULT)) {
3850 		err = -EINVAL;
3851 		goto free_ucmd;
3852 	}
3853 
3854 	dst = kzalloc(sizeof(*dst), GFP_KERNEL);
3855 	if (!dst) {
3856 		err = -ENOMEM;
3857 		goto free_ucmd;
3858 	}
3859 
3860 	mutex_lock(&dev->flow_db->lock);
3861 
3862 	ft_prio = get_flow_table(dev, flow_attr,
3863 				 is_egress ? MLX5_IB_FT_TX : MLX5_IB_FT_RX);
3864 	if (IS_ERR(ft_prio)) {
3865 		err = PTR_ERR(ft_prio);
3866 		goto unlock;
3867 	}
3868 	if (flow_attr->type == IB_FLOW_ATTR_SNIFFER) {
3869 		ft_prio_tx = get_flow_table(dev, flow_attr, MLX5_IB_FT_TX);
3870 		if (IS_ERR(ft_prio_tx)) {
3871 			err = PTR_ERR(ft_prio_tx);
3872 			ft_prio_tx = NULL;
3873 			goto destroy_ft;
3874 		}
3875 	}
3876 
3877 	if (is_egress) {
3878 		dst->type = MLX5_FLOW_DESTINATION_TYPE_PORT;
3879 	} else {
3880 		dst->type = MLX5_FLOW_DESTINATION_TYPE_TIR;
3881 		if (mqp->flags & MLX5_IB_QP_RSS)
3882 			dst->tir_num = mqp->rss_qp.tirn;
3883 		else
3884 			dst->tir_num = mqp->raw_packet_qp.rq.tirn;
3885 	}
3886 
3887 	if (flow_attr->type == IB_FLOW_ATTR_NORMAL) {
3888 		if (flow_attr->flags & IB_FLOW_ATTR_FLAGS_DONT_TRAP)  {
3889 			handler = create_dont_trap_rule(dev, ft_prio,
3890 							flow_attr, dst);
3891 		} else {
3892 			underlay_qpn = (mqp->flags & MLX5_IB_QP_UNDERLAY) ?
3893 					mqp->underlay_qpn : 0;
3894 			handler = _create_flow_rule(dev, ft_prio, flow_attr,
3895 						    dst, underlay_qpn, ucmd);
3896 		}
3897 	} else if (flow_attr->type == IB_FLOW_ATTR_ALL_DEFAULT ||
3898 		   flow_attr->type == IB_FLOW_ATTR_MC_DEFAULT) {
3899 		handler = create_leftovers_rule(dev, ft_prio, flow_attr,
3900 						dst);
3901 	} else if (flow_attr->type == IB_FLOW_ATTR_SNIFFER) {
3902 		handler = create_sniffer_rule(dev, ft_prio, ft_prio_tx, dst);
3903 	} else {
3904 		err = -EINVAL;
3905 		goto destroy_ft;
3906 	}
3907 
3908 	if (IS_ERR(handler)) {
3909 		err = PTR_ERR(handler);
3910 		handler = NULL;
3911 		goto destroy_ft;
3912 	}
3913 
3914 	mutex_unlock(&dev->flow_db->lock);
3915 	kfree(dst);
3916 	kfree(ucmd);
3917 
3918 	return &handler->ibflow;
3919 
3920 destroy_ft:
3921 	put_flow_table(dev, ft_prio, false);
3922 	if (ft_prio_tx)
3923 		put_flow_table(dev, ft_prio_tx, false);
3924 unlock:
3925 	mutex_unlock(&dev->flow_db->lock);
3926 	kfree(dst);
3927 free_ucmd:
3928 	kfree(ucmd);
3929 	return ERR_PTR(err);
3930 }
3931 
3932 static struct mlx5_ib_flow_prio *
3933 _get_flow_table(struct mlx5_ib_dev *dev,
3934 		struct mlx5_ib_flow_matcher *fs_matcher,
3935 		bool mcast)
3936 {
3937 	struct mlx5_flow_namespace *ns = NULL;
3938 	struct mlx5_ib_flow_prio *prio = NULL;
3939 	int max_table_size = 0;
3940 	bool esw_encap;
3941 	u32 flags = 0;
3942 	int priority;
3943 
3944 	if (mcast)
3945 		priority = MLX5_IB_FLOW_MCAST_PRIO;
3946 	else
3947 		priority = ib_prio_to_core_prio(fs_matcher->priority, false);
3948 
3949 	esw_encap = mlx5_eswitch_get_encap_mode(dev->mdev) !=
3950 		DEVLINK_ESWITCH_ENCAP_MODE_NONE;
3951 	if (fs_matcher->ns_type == MLX5_FLOW_NAMESPACE_BYPASS) {
3952 		max_table_size = BIT(MLX5_CAP_FLOWTABLE_NIC_RX(dev->mdev,
3953 					log_max_ft_size));
3954 		if (MLX5_CAP_FLOWTABLE_NIC_RX(dev->mdev, decap) && !esw_encap)
3955 			flags |= MLX5_FLOW_TABLE_TUNNEL_EN_DECAP;
3956 		if (MLX5_CAP_FLOWTABLE_NIC_RX(dev->mdev,
3957 					      reformat_l3_tunnel_to_l2) &&
3958 		    !esw_encap)
3959 			flags |= MLX5_FLOW_TABLE_TUNNEL_EN_REFORMAT;
3960 	} else if (fs_matcher->ns_type == MLX5_FLOW_NAMESPACE_EGRESS) {
3961 		max_table_size = BIT(
3962 			MLX5_CAP_FLOWTABLE_NIC_TX(dev->mdev, log_max_ft_size));
3963 		if (MLX5_CAP_FLOWTABLE_NIC_TX(dev->mdev, reformat) && !esw_encap)
3964 			flags |= MLX5_FLOW_TABLE_TUNNEL_EN_REFORMAT;
3965 	} else if (fs_matcher->ns_type == MLX5_FLOW_NAMESPACE_FDB) {
3966 		max_table_size = BIT(
3967 			MLX5_CAP_ESW_FLOWTABLE_FDB(dev->mdev, log_max_ft_size));
3968 		if (MLX5_CAP_ESW_FLOWTABLE_FDB(dev->mdev, decap) && esw_encap)
3969 			flags |= MLX5_FLOW_TABLE_TUNNEL_EN_DECAP;
3970 		if (MLX5_CAP_ESW_FLOWTABLE_FDB(dev->mdev, reformat_l3_tunnel_to_l2) &&
3971 		    esw_encap)
3972 			flags |= MLX5_FLOW_TABLE_TUNNEL_EN_REFORMAT;
3973 		priority = FDB_BYPASS_PATH;
3974 	} else if (fs_matcher->ns_type == MLX5_FLOW_NAMESPACE_RDMA_RX) {
3975 		max_table_size =
3976 			BIT(MLX5_CAP_FLOWTABLE_RDMA_RX(dev->mdev,
3977 						       log_max_ft_size));
3978 		priority = fs_matcher->priority;
3979 	}
3980 
3981 	max_table_size = min_t(int, max_table_size, MLX5_FS_MAX_ENTRIES);
3982 
3983 	ns = mlx5_get_flow_namespace(dev->mdev, fs_matcher->ns_type);
3984 	if (!ns)
3985 		return ERR_PTR(-ENOTSUPP);
3986 
3987 	if (fs_matcher->ns_type == MLX5_FLOW_NAMESPACE_BYPASS)
3988 		prio = &dev->flow_db->prios[priority];
3989 	else if (fs_matcher->ns_type == MLX5_FLOW_NAMESPACE_EGRESS)
3990 		prio = &dev->flow_db->egress_prios[priority];
3991 	else if (fs_matcher->ns_type == MLX5_FLOW_NAMESPACE_FDB)
3992 		prio = &dev->flow_db->fdb;
3993 	else if (fs_matcher->ns_type == MLX5_FLOW_NAMESPACE_RDMA_RX)
3994 		prio = &dev->flow_db->rdma_rx[priority];
3995 
3996 	if (!prio)
3997 		return ERR_PTR(-EINVAL);
3998 
3999 	if (prio->flow_table)
4000 		return prio;
4001 
4002 	return _get_prio(ns, prio, priority, max_table_size,
4003 			 MLX5_FS_MAX_TYPES, flags);
4004 }
4005 
4006 static struct mlx5_ib_flow_handler *
4007 _create_raw_flow_rule(struct mlx5_ib_dev *dev,
4008 		      struct mlx5_ib_flow_prio *ft_prio,
4009 		      struct mlx5_flow_destination *dst,
4010 		      struct mlx5_ib_flow_matcher  *fs_matcher,
4011 		      struct mlx5_flow_context *flow_context,
4012 		      struct mlx5_flow_act *flow_act,
4013 		      void *cmd_in, int inlen,
4014 		      int dst_num)
4015 {
4016 	struct mlx5_ib_flow_handler *handler;
4017 	struct mlx5_flow_spec *spec;
4018 	struct mlx5_flow_table *ft = ft_prio->flow_table;
4019 	int err = 0;
4020 
4021 	spec = kvzalloc(sizeof(*spec), GFP_KERNEL);
4022 	handler = kzalloc(sizeof(*handler), GFP_KERNEL);
4023 	if (!handler || !spec) {
4024 		err = -ENOMEM;
4025 		goto free;
4026 	}
4027 
4028 	INIT_LIST_HEAD(&handler->list);
4029 
4030 	memcpy(spec->match_value, cmd_in, inlen);
4031 	memcpy(spec->match_criteria, fs_matcher->matcher_mask.match_params,
4032 	       fs_matcher->mask_len);
4033 	spec->match_criteria_enable = fs_matcher->match_criteria_enable;
4034 	spec->flow_context = *flow_context;
4035 
4036 	handler->rule = mlx5_add_flow_rules(ft, spec,
4037 					    flow_act, dst, dst_num);
4038 
4039 	if (IS_ERR(handler->rule)) {
4040 		err = PTR_ERR(handler->rule);
4041 		goto free;
4042 	}
4043 
4044 	ft_prio->refcount++;
4045 	handler->prio = ft_prio;
4046 	handler->dev = dev;
4047 	ft_prio->flow_table = ft;
4048 
4049 free:
4050 	if (err)
4051 		kfree(handler);
4052 	kvfree(spec);
4053 	return err ? ERR_PTR(err) : handler;
4054 }
4055 
4056 static bool raw_fs_is_multicast(struct mlx5_ib_flow_matcher *fs_matcher,
4057 				void *match_v)
4058 {
4059 	void *match_c;
4060 	void *match_v_set_lyr_2_4, *match_c_set_lyr_2_4;
4061 	void *dmac, *dmac_mask;
4062 	void *ipv4, *ipv4_mask;
4063 
4064 	if (!(fs_matcher->match_criteria_enable &
4065 	      (1 << MATCH_CRITERIA_ENABLE_OUTER_BIT)))
4066 		return false;
4067 
4068 	match_c = fs_matcher->matcher_mask.match_params;
4069 	match_v_set_lyr_2_4 = MLX5_ADDR_OF(fte_match_param, match_v,
4070 					   outer_headers);
4071 	match_c_set_lyr_2_4 = MLX5_ADDR_OF(fte_match_param, match_c,
4072 					   outer_headers);
4073 
4074 	dmac = MLX5_ADDR_OF(fte_match_set_lyr_2_4, match_v_set_lyr_2_4,
4075 			    dmac_47_16);
4076 	dmac_mask = MLX5_ADDR_OF(fte_match_set_lyr_2_4, match_c_set_lyr_2_4,
4077 				 dmac_47_16);
4078 
4079 	if (is_multicast_ether_addr(dmac) &&
4080 	    is_multicast_ether_addr(dmac_mask))
4081 		return true;
4082 
4083 	ipv4 = MLX5_ADDR_OF(fte_match_set_lyr_2_4, match_v_set_lyr_2_4,
4084 			    dst_ipv4_dst_ipv6.ipv4_layout.ipv4);
4085 
4086 	ipv4_mask = MLX5_ADDR_OF(fte_match_set_lyr_2_4, match_c_set_lyr_2_4,
4087 				 dst_ipv4_dst_ipv6.ipv4_layout.ipv4);
4088 
4089 	if (ipv4_is_multicast(*(__be32 *)(ipv4)) &&
4090 	    ipv4_is_multicast(*(__be32 *)(ipv4_mask)))
4091 		return true;
4092 
4093 	return false;
4094 }
4095 
4096 struct mlx5_ib_flow_handler *
4097 mlx5_ib_raw_fs_rule_add(struct mlx5_ib_dev *dev,
4098 			struct mlx5_ib_flow_matcher *fs_matcher,
4099 			struct mlx5_flow_context *flow_context,
4100 			struct mlx5_flow_act *flow_act,
4101 			u32 counter_id,
4102 			void *cmd_in, int inlen, int dest_id,
4103 			int dest_type)
4104 {
4105 	struct mlx5_flow_destination *dst;
4106 	struct mlx5_ib_flow_prio *ft_prio;
4107 	struct mlx5_ib_flow_handler *handler;
4108 	int dst_num = 0;
4109 	bool mcast;
4110 	int err;
4111 
4112 	if (fs_matcher->flow_type != MLX5_IB_FLOW_TYPE_NORMAL)
4113 		return ERR_PTR(-EOPNOTSUPP);
4114 
4115 	if (fs_matcher->priority > MLX5_IB_FLOW_LAST_PRIO)
4116 		return ERR_PTR(-ENOMEM);
4117 
4118 	dst = kcalloc(2, sizeof(*dst), GFP_KERNEL);
4119 	if (!dst)
4120 		return ERR_PTR(-ENOMEM);
4121 
4122 	mcast = raw_fs_is_multicast(fs_matcher, cmd_in);
4123 	mutex_lock(&dev->flow_db->lock);
4124 
4125 	ft_prio = _get_flow_table(dev, fs_matcher, mcast);
4126 	if (IS_ERR(ft_prio)) {
4127 		err = PTR_ERR(ft_prio);
4128 		goto unlock;
4129 	}
4130 
4131 	if (dest_type == MLX5_FLOW_DESTINATION_TYPE_TIR) {
4132 		dst[dst_num].type = dest_type;
4133 		dst[dst_num].tir_num = dest_id;
4134 		flow_act->action |= MLX5_FLOW_CONTEXT_ACTION_FWD_DEST;
4135 	} else if (dest_type == MLX5_FLOW_DESTINATION_TYPE_FLOW_TABLE) {
4136 		dst[dst_num].type = MLX5_FLOW_DESTINATION_TYPE_FLOW_TABLE_NUM;
4137 		dst[dst_num].ft_num = dest_id;
4138 		flow_act->action |= MLX5_FLOW_CONTEXT_ACTION_FWD_DEST;
4139 	} else {
4140 		dst[dst_num].type = MLX5_FLOW_DESTINATION_TYPE_PORT;
4141 		flow_act->action |= MLX5_FLOW_CONTEXT_ACTION_ALLOW;
4142 	}
4143 
4144 	dst_num++;
4145 
4146 	if (flow_act->action & MLX5_FLOW_CONTEXT_ACTION_COUNT) {
4147 		dst[dst_num].type = MLX5_FLOW_DESTINATION_TYPE_COUNTER;
4148 		dst[dst_num].counter_id = counter_id;
4149 		dst_num++;
4150 	}
4151 
4152 	handler = _create_raw_flow_rule(dev, ft_prio, dst, fs_matcher,
4153 					flow_context, flow_act,
4154 					cmd_in, inlen, dst_num);
4155 
4156 	if (IS_ERR(handler)) {
4157 		err = PTR_ERR(handler);
4158 		goto destroy_ft;
4159 	}
4160 
4161 	mutex_unlock(&dev->flow_db->lock);
4162 	atomic_inc(&fs_matcher->usecnt);
4163 	handler->flow_matcher = fs_matcher;
4164 
4165 	kfree(dst);
4166 
4167 	return handler;
4168 
4169 destroy_ft:
4170 	put_flow_table(dev, ft_prio, false);
4171 unlock:
4172 	mutex_unlock(&dev->flow_db->lock);
4173 	kfree(dst);
4174 
4175 	return ERR_PTR(err);
4176 }
4177 
4178 static u32 mlx5_ib_flow_action_flags_to_accel_xfrm_flags(u32 mlx5_flags)
4179 {
4180 	u32 flags = 0;
4181 
4182 	if (mlx5_flags & MLX5_IB_UAPI_FLOW_ACTION_FLAGS_REQUIRE_METADATA)
4183 		flags |= MLX5_ACCEL_XFRM_FLAG_REQUIRE_METADATA;
4184 
4185 	return flags;
4186 }
4187 
4188 #define MLX5_FLOW_ACTION_ESP_CREATE_LAST_SUPPORTED	MLX5_IB_UAPI_FLOW_ACTION_FLAGS_REQUIRE_METADATA
4189 static struct ib_flow_action *
4190 mlx5_ib_create_flow_action_esp(struct ib_device *device,
4191 			       const struct ib_flow_action_attrs_esp *attr,
4192 			       struct uverbs_attr_bundle *attrs)
4193 {
4194 	struct mlx5_ib_dev *mdev = to_mdev(device);
4195 	struct ib_uverbs_flow_action_esp_keymat_aes_gcm *aes_gcm;
4196 	struct mlx5_accel_esp_xfrm_attrs accel_attrs = {};
4197 	struct mlx5_ib_flow_action *action;
4198 	u64 action_flags;
4199 	u64 flags;
4200 	int err = 0;
4201 
4202 	err = uverbs_get_flags64(
4203 		&action_flags, attrs, MLX5_IB_ATTR_CREATE_FLOW_ACTION_FLAGS,
4204 		((MLX5_FLOW_ACTION_ESP_CREATE_LAST_SUPPORTED << 1) - 1));
4205 	if (err)
4206 		return ERR_PTR(err);
4207 
4208 	flags = mlx5_ib_flow_action_flags_to_accel_xfrm_flags(action_flags);
4209 
4210 	/* We current only support a subset of the standard features. Only a
4211 	 * keymat of type AES_GCM, with icv_len == 16, iv_algo == SEQ and esn
4212 	 * (with overlap). Full offload mode isn't supported.
4213 	 */
4214 	if (!attr->keymat || attr->replay || attr->encap ||
4215 	    attr->spi || attr->seq || attr->tfc_pad ||
4216 	    attr->hard_limit_pkts ||
4217 	    (attr->flags & ~(IB_FLOW_ACTION_ESP_FLAGS_ESN_TRIGGERED |
4218 			     IB_UVERBS_FLOW_ACTION_ESP_FLAGS_ENCRYPT)))
4219 		return ERR_PTR(-EOPNOTSUPP);
4220 
4221 	if (attr->keymat->protocol !=
4222 	    IB_UVERBS_FLOW_ACTION_ESP_KEYMAT_AES_GCM)
4223 		return ERR_PTR(-EOPNOTSUPP);
4224 
4225 	aes_gcm = &attr->keymat->keymat.aes_gcm;
4226 
4227 	if (aes_gcm->icv_len != 16 ||
4228 	    aes_gcm->iv_algo != IB_UVERBS_FLOW_ACTION_IV_ALGO_SEQ)
4229 		return ERR_PTR(-EOPNOTSUPP);
4230 
4231 	action = kmalloc(sizeof(*action), GFP_KERNEL);
4232 	if (!action)
4233 		return ERR_PTR(-ENOMEM);
4234 
4235 	action->esp_aes_gcm.ib_flags = attr->flags;
4236 	memcpy(&accel_attrs.keymat.aes_gcm.aes_key, &aes_gcm->aes_key,
4237 	       sizeof(accel_attrs.keymat.aes_gcm.aes_key));
4238 	accel_attrs.keymat.aes_gcm.key_len = aes_gcm->key_len * 8;
4239 	memcpy(&accel_attrs.keymat.aes_gcm.salt, &aes_gcm->salt,
4240 	       sizeof(accel_attrs.keymat.aes_gcm.salt));
4241 	memcpy(&accel_attrs.keymat.aes_gcm.seq_iv, &aes_gcm->iv,
4242 	       sizeof(accel_attrs.keymat.aes_gcm.seq_iv));
4243 	accel_attrs.keymat.aes_gcm.icv_len = aes_gcm->icv_len * 8;
4244 	accel_attrs.keymat.aes_gcm.iv_algo = MLX5_ACCEL_ESP_AES_GCM_IV_ALGO_SEQ;
4245 	accel_attrs.keymat_type = MLX5_ACCEL_ESP_KEYMAT_AES_GCM;
4246 
4247 	accel_attrs.esn = attr->esn;
4248 	if (attr->flags & IB_FLOW_ACTION_ESP_FLAGS_ESN_TRIGGERED)
4249 		accel_attrs.flags |= MLX5_ACCEL_ESP_FLAGS_ESN_TRIGGERED;
4250 	if (attr->flags & IB_UVERBS_FLOW_ACTION_ESP_FLAGS_ESN_NEW_WINDOW)
4251 		accel_attrs.flags |= MLX5_ACCEL_ESP_FLAGS_ESN_STATE_OVERLAP;
4252 
4253 	if (attr->flags & IB_UVERBS_FLOW_ACTION_ESP_FLAGS_ENCRYPT)
4254 		accel_attrs.action |= MLX5_ACCEL_ESP_ACTION_ENCRYPT;
4255 
4256 	action->esp_aes_gcm.ctx =
4257 		mlx5_accel_esp_create_xfrm(mdev->mdev, &accel_attrs, flags);
4258 	if (IS_ERR(action->esp_aes_gcm.ctx)) {
4259 		err = PTR_ERR(action->esp_aes_gcm.ctx);
4260 		goto err_parse;
4261 	}
4262 
4263 	action->esp_aes_gcm.ib_flags = attr->flags;
4264 
4265 	return &action->ib_action;
4266 
4267 err_parse:
4268 	kfree(action);
4269 	return ERR_PTR(err);
4270 }
4271 
4272 static int
4273 mlx5_ib_modify_flow_action_esp(struct ib_flow_action *action,
4274 			       const struct ib_flow_action_attrs_esp *attr,
4275 			       struct uverbs_attr_bundle *attrs)
4276 {
4277 	struct mlx5_ib_flow_action *maction = to_mflow_act(action);
4278 	struct mlx5_accel_esp_xfrm_attrs accel_attrs;
4279 	int err = 0;
4280 
4281 	if (attr->keymat || attr->replay || attr->encap ||
4282 	    attr->spi || attr->seq || attr->tfc_pad ||
4283 	    attr->hard_limit_pkts ||
4284 	    (attr->flags & ~(IB_FLOW_ACTION_ESP_FLAGS_ESN_TRIGGERED |
4285 			     IB_FLOW_ACTION_ESP_FLAGS_MOD_ESP_ATTRS |
4286 			     IB_UVERBS_FLOW_ACTION_ESP_FLAGS_ESN_NEW_WINDOW)))
4287 		return -EOPNOTSUPP;
4288 
4289 	/* Only the ESN value or the MLX5_ACCEL_ESP_FLAGS_ESN_STATE_OVERLAP can
4290 	 * be modified.
4291 	 */
4292 	if (!(maction->esp_aes_gcm.ib_flags &
4293 	      IB_FLOW_ACTION_ESP_FLAGS_ESN_TRIGGERED) &&
4294 	    attr->flags & (IB_FLOW_ACTION_ESP_FLAGS_ESN_TRIGGERED |
4295 			   IB_UVERBS_FLOW_ACTION_ESP_FLAGS_ESN_NEW_WINDOW))
4296 		return -EINVAL;
4297 
4298 	memcpy(&accel_attrs, &maction->esp_aes_gcm.ctx->attrs,
4299 	       sizeof(accel_attrs));
4300 
4301 	accel_attrs.esn = attr->esn;
4302 	if (attr->flags & IB_UVERBS_FLOW_ACTION_ESP_FLAGS_ESN_NEW_WINDOW)
4303 		accel_attrs.flags |= MLX5_ACCEL_ESP_FLAGS_ESN_STATE_OVERLAP;
4304 	else
4305 		accel_attrs.flags &= ~MLX5_ACCEL_ESP_FLAGS_ESN_STATE_OVERLAP;
4306 
4307 	err = mlx5_accel_esp_modify_xfrm(maction->esp_aes_gcm.ctx,
4308 					 &accel_attrs);
4309 	if (err)
4310 		return err;
4311 
4312 	maction->esp_aes_gcm.ib_flags &=
4313 		~IB_UVERBS_FLOW_ACTION_ESP_FLAGS_ESN_NEW_WINDOW;
4314 	maction->esp_aes_gcm.ib_flags |=
4315 		attr->flags & IB_UVERBS_FLOW_ACTION_ESP_FLAGS_ESN_NEW_WINDOW;
4316 
4317 	return 0;
4318 }
4319 
4320 static int mlx5_ib_destroy_flow_action(struct ib_flow_action *action)
4321 {
4322 	struct mlx5_ib_flow_action *maction = to_mflow_act(action);
4323 
4324 	switch (action->type) {
4325 	case IB_FLOW_ACTION_ESP:
4326 		/*
4327 		 * We only support aes_gcm by now, so we implicitly know this is
4328 		 * the underline crypto.
4329 		 */
4330 		mlx5_accel_esp_destroy_xfrm(maction->esp_aes_gcm.ctx);
4331 		break;
4332 	case IB_FLOW_ACTION_UNSPECIFIED:
4333 		mlx5_ib_destroy_flow_action_raw(maction);
4334 		break;
4335 	default:
4336 		WARN_ON(true);
4337 		break;
4338 	}
4339 
4340 	kfree(maction);
4341 	return 0;
4342 }
4343 
4344 static int mlx5_ib_mcg_attach(struct ib_qp *ibqp, union ib_gid *gid, u16 lid)
4345 {
4346 	struct mlx5_ib_dev *dev = to_mdev(ibqp->device);
4347 	struct mlx5_ib_qp *mqp = to_mqp(ibqp);
4348 	int err;
4349 	u16 uid;
4350 
4351 	uid = ibqp->pd ?
4352 		to_mpd(ibqp->pd)->uid : 0;
4353 
4354 	if (mqp->flags & MLX5_IB_QP_UNDERLAY) {
4355 		mlx5_ib_dbg(dev, "Attaching a multi cast group to underlay QP is not supported\n");
4356 		return -EOPNOTSUPP;
4357 	}
4358 
4359 	err = mlx5_cmd_attach_mcg(dev->mdev, gid, ibqp->qp_num, uid);
4360 	if (err)
4361 		mlx5_ib_warn(dev, "failed attaching QPN 0x%x, MGID %pI6\n",
4362 			     ibqp->qp_num, gid->raw);
4363 
4364 	return err;
4365 }
4366 
4367 static int mlx5_ib_mcg_detach(struct ib_qp *ibqp, union ib_gid *gid, u16 lid)
4368 {
4369 	struct mlx5_ib_dev *dev = to_mdev(ibqp->device);
4370 	int err;
4371 	u16 uid;
4372 
4373 	uid = ibqp->pd ?
4374 		to_mpd(ibqp->pd)->uid : 0;
4375 	err = mlx5_cmd_detach_mcg(dev->mdev, gid, ibqp->qp_num, uid);
4376 	if (err)
4377 		mlx5_ib_warn(dev, "failed detaching QPN 0x%x, MGID %pI6\n",
4378 			     ibqp->qp_num, gid->raw);
4379 
4380 	return err;
4381 }
4382 
4383 static int init_node_data(struct mlx5_ib_dev *dev)
4384 {
4385 	int err;
4386 
4387 	err = mlx5_query_node_desc(dev, dev->ib_dev.node_desc);
4388 	if (err)
4389 		return err;
4390 
4391 	dev->mdev->rev_id = dev->mdev->pdev->revision;
4392 
4393 	return mlx5_query_node_guid(dev, &dev->ib_dev.node_guid);
4394 }
4395 
4396 static ssize_t fw_pages_show(struct device *device,
4397 			     struct device_attribute *attr, char *buf)
4398 {
4399 	struct mlx5_ib_dev *dev =
4400 		rdma_device_to_drv_device(device, struct mlx5_ib_dev, ib_dev);
4401 
4402 	return sprintf(buf, "%d\n", dev->mdev->priv.fw_pages);
4403 }
4404 static DEVICE_ATTR_RO(fw_pages);
4405 
4406 static ssize_t reg_pages_show(struct device *device,
4407 			      struct device_attribute *attr, char *buf)
4408 {
4409 	struct mlx5_ib_dev *dev =
4410 		rdma_device_to_drv_device(device, struct mlx5_ib_dev, ib_dev);
4411 
4412 	return sprintf(buf, "%d\n", atomic_read(&dev->mdev->priv.reg_pages));
4413 }
4414 static DEVICE_ATTR_RO(reg_pages);
4415 
4416 static ssize_t hca_type_show(struct device *device,
4417 			     struct device_attribute *attr, char *buf)
4418 {
4419 	struct mlx5_ib_dev *dev =
4420 		rdma_device_to_drv_device(device, struct mlx5_ib_dev, ib_dev);
4421 
4422 	return sprintf(buf, "MT%d\n", dev->mdev->pdev->device);
4423 }
4424 static DEVICE_ATTR_RO(hca_type);
4425 
4426 static ssize_t hw_rev_show(struct device *device,
4427 			   struct device_attribute *attr, char *buf)
4428 {
4429 	struct mlx5_ib_dev *dev =
4430 		rdma_device_to_drv_device(device, struct mlx5_ib_dev, ib_dev);
4431 
4432 	return sprintf(buf, "%x\n", dev->mdev->rev_id);
4433 }
4434 static DEVICE_ATTR_RO(hw_rev);
4435 
4436 static ssize_t board_id_show(struct device *device,
4437 			     struct device_attribute *attr, char *buf)
4438 {
4439 	struct mlx5_ib_dev *dev =
4440 		rdma_device_to_drv_device(device, struct mlx5_ib_dev, ib_dev);
4441 
4442 	return sprintf(buf, "%.*s\n", MLX5_BOARD_ID_LEN,
4443 		       dev->mdev->board_id);
4444 }
4445 static DEVICE_ATTR_RO(board_id);
4446 
4447 static struct attribute *mlx5_class_attributes[] = {
4448 	&dev_attr_hw_rev.attr,
4449 	&dev_attr_hca_type.attr,
4450 	&dev_attr_board_id.attr,
4451 	&dev_attr_fw_pages.attr,
4452 	&dev_attr_reg_pages.attr,
4453 	NULL,
4454 };
4455 
4456 static const struct attribute_group mlx5_attr_group = {
4457 	.attrs = mlx5_class_attributes,
4458 };
4459 
4460 static void pkey_change_handler(struct work_struct *work)
4461 {
4462 	struct mlx5_ib_port_resources *ports =
4463 		container_of(work, struct mlx5_ib_port_resources,
4464 			     pkey_change_work);
4465 
4466 	mutex_lock(&ports->devr->mutex);
4467 	mlx5_ib_gsi_pkey_change(ports->gsi);
4468 	mutex_unlock(&ports->devr->mutex);
4469 }
4470 
4471 static void mlx5_ib_handle_internal_error(struct mlx5_ib_dev *ibdev)
4472 {
4473 	struct mlx5_ib_qp *mqp;
4474 	struct mlx5_ib_cq *send_mcq, *recv_mcq;
4475 	struct mlx5_core_cq *mcq;
4476 	struct list_head cq_armed_list;
4477 	unsigned long flags_qp;
4478 	unsigned long flags_cq;
4479 	unsigned long flags;
4480 
4481 	INIT_LIST_HEAD(&cq_armed_list);
4482 
4483 	/* Go over qp list reside on that ibdev, sync with create/destroy qp.*/
4484 	spin_lock_irqsave(&ibdev->reset_flow_resource_lock, flags);
4485 	list_for_each_entry(mqp, &ibdev->qp_list, qps_list) {
4486 		spin_lock_irqsave(&mqp->sq.lock, flags_qp);
4487 		if (mqp->sq.tail != mqp->sq.head) {
4488 			send_mcq = to_mcq(mqp->ibqp.send_cq);
4489 			spin_lock_irqsave(&send_mcq->lock, flags_cq);
4490 			if (send_mcq->mcq.comp &&
4491 			    mqp->ibqp.send_cq->comp_handler) {
4492 				if (!send_mcq->mcq.reset_notify_added) {
4493 					send_mcq->mcq.reset_notify_added = 1;
4494 					list_add_tail(&send_mcq->mcq.reset_notify,
4495 						      &cq_armed_list);
4496 				}
4497 			}
4498 			spin_unlock_irqrestore(&send_mcq->lock, flags_cq);
4499 		}
4500 		spin_unlock_irqrestore(&mqp->sq.lock, flags_qp);
4501 		spin_lock_irqsave(&mqp->rq.lock, flags_qp);
4502 		/* no handling is needed for SRQ */
4503 		if (!mqp->ibqp.srq) {
4504 			if (mqp->rq.tail != mqp->rq.head) {
4505 				recv_mcq = to_mcq(mqp->ibqp.recv_cq);
4506 				spin_lock_irqsave(&recv_mcq->lock, flags_cq);
4507 				if (recv_mcq->mcq.comp &&
4508 				    mqp->ibqp.recv_cq->comp_handler) {
4509 					if (!recv_mcq->mcq.reset_notify_added) {
4510 						recv_mcq->mcq.reset_notify_added = 1;
4511 						list_add_tail(&recv_mcq->mcq.reset_notify,
4512 							      &cq_armed_list);
4513 					}
4514 				}
4515 				spin_unlock_irqrestore(&recv_mcq->lock,
4516 						       flags_cq);
4517 			}
4518 		}
4519 		spin_unlock_irqrestore(&mqp->rq.lock, flags_qp);
4520 	}
4521 	/*At that point all inflight post send were put to be executed as of we
4522 	 * lock/unlock above locks Now need to arm all involved CQs.
4523 	 */
4524 	list_for_each_entry(mcq, &cq_armed_list, reset_notify) {
4525 		mcq->comp(mcq, NULL);
4526 	}
4527 	spin_unlock_irqrestore(&ibdev->reset_flow_resource_lock, flags);
4528 }
4529 
4530 static void delay_drop_handler(struct work_struct *work)
4531 {
4532 	int err;
4533 	struct mlx5_ib_delay_drop *delay_drop =
4534 		container_of(work, struct mlx5_ib_delay_drop,
4535 			     delay_drop_work);
4536 
4537 	atomic_inc(&delay_drop->events_cnt);
4538 
4539 	mutex_lock(&delay_drop->lock);
4540 	err = mlx5_core_set_delay_drop(delay_drop->dev->mdev,
4541 				       delay_drop->timeout);
4542 	if (err) {
4543 		mlx5_ib_warn(delay_drop->dev, "Failed to set delay drop, timeout=%u\n",
4544 			     delay_drop->timeout);
4545 		delay_drop->activate = false;
4546 	}
4547 	mutex_unlock(&delay_drop->lock);
4548 }
4549 
4550 static void handle_general_event(struct mlx5_ib_dev *ibdev, struct mlx5_eqe *eqe,
4551 				 struct ib_event *ibev)
4552 {
4553 	u8 port = (eqe->data.port.port >> 4) & 0xf;
4554 
4555 	switch (eqe->sub_type) {
4556 	case MLX5_GENERAL_SUBTYPE_DELAY_DROP_TIMEOUT:
4557 		if (mlx5_ib_port_link_layer(&ibdev->ib_dev, port) ==
4558 					    IB_LINK_LAYER_ETHERNET)
4559 			schedule_work(&ibdev->delay_drop.delay_drop_work);
4560 		break;
4561 	default: /* do nothing */
4562 		return;
4563 	}
4564 }
4565 
4566 static int handle_port_change(struct mlx5_ib_dev *ibdev, struct mlx5_eqe *eqe,
4567 			      struct ib_event *ibev)
4568 {
4569 	u8 port = (eqe->data.port.port >> 4) & 0xf;
4570 
4571 	ibev->element.port_num = port;
4572 
4573 	switch (eqe->sub_type) {
4574 	case MLX5_PORT_CHANGE_SUBTYPE_ACTIVE:
4575 	case MLX5_PORT_CHANGE_SUBTYPE_DOWN:
4576 	case MLX5_PORT_CHANGE_SUBTYPE_INITIALIZED:
4577 		/* In RoCE, port up/down events are handled in
4578 		 * mlx5_netdev_event().
4579 		 */
4580 		if (mlx5_ib_port_link_layer(&ibdev->ib_dev, port) ==
4581 					    IB_LINK_LAYER_ETHERNET)
4582 			return -EINVAL;
4583 
4584 		ibev->event = (eqe->sub_type == MLX5_PORT_CHANGE_SUBTYPE_ACTIVE) ?
4585 				IB_EVENT_PORT_ACTIVE : IB_EVENT_PORT_ERR;
4586 		break;
4587 
4588 	case MLX5_PORT_CHANGE_SUBTYPE_LID:
4589 		ibev->event = IB_EVENT_LID_CHANGE;
4590 		break;
4591 
4592 	case MLX5_PORT_CHANGE_SUBTYPE_PKEY:
4593 		ibev->event = IB_EVENT_PKEY_CHANGE;
4594 		schedule_work(&ibdev->devr.ports[port - 1].pkey_change_work);
4595 		break;
4596 
4597 	case MLX5_PORT_CHANGE_SUBTYPE_GUID:
4598 		ibev->event = IB_EVENT_GID_CHANGE;
4599 		break;
4600 
4601 	case MLX5_PORT_CHANGE_SUBTYPE_CLIENT_REREG:
4602 		ibev->event = IB_EVENT_CLIENT_REREGISTER;
4603 		break;
4604 	default:
4605 		return -EINVAL;
4606 	}
4607 
4608 	return 0;
4609 }
4610 
4611 static void mlx5_ib_handle_event(struct work_struct *_work)
4612 {
4613 	struct mlx5_ib_event_work *work =
4614 		container_of(_work, struct mlx5_ib_event_work, work);
4615 	struct mlx5_ib_dev *ibdev;
4616 	struct ib_event ibev;
4617 	bool fatal = false;
4618 
4619 	if (work->is_slave) {
4620 		ibdev = mlx5_ib_get_ibdev_from_mpi(work->mpi);
4621 		if (!ibdev)
4622 			goto out;
4623 	} else {
4624 		ibdev = work->dev;
4625 	}
4626 
4627 	switch (work->event) {
4628 	case MLX5_DEV_EVENT_SYS_ERROR:
4629 		ibev.event = IB_EVENT_DEVICE_FATAL;
4630 		mlx5_ib_handle_internal_error(ibdev);
4631 		ibev.element.port_num  = (u8)(unsigned long)work->param;
4632 		fatal = true;
4633 		break;
4634 	case MLX5_EVENT_TYPE_PORT_CHANGE:
4635 		if (handle_port_change(ibdev, work->param, &ibev))
4636 			goto out;
4637 		break;
4638 	case MLX5_EVENT_TYPE_GENERAL_EVENT:
4639 		handle_general_event(ibdev, work->param, &ibev);
4640 		/* fall through */
4641 	default:
4642 		goto out;
4643 	}
4644 
4645 	ibev.device = &ibdev->ib_dev;
4646 
4647 	if (!rdma_is_port_valid(&ibdev->ib_dev, ibev.element.port_num)) {
4648 		mlx5_ib_warn(ibdev, "warning: event on port %d\n",  ibev.element.port_num);
4649 		goto out;
4650 	}
4651 
4652 	if (ibdev->ib_active)
4653 		ib_dispatch_event(&ibev);
4654 
4655 	if (fatal)
4656 		ibdev->ib_active = false;
4657 out:
4658 	kfree(work);
4659 }
4660 
4661 static int mlx5_ib_event(struct notifier_block *nb,
4662 			 unsigned long event, void *param)
4663 {
4664 	struct mlx5_ib_event_work *work;
4665 
4666 	work = kmalloc(sizeof(*work), GFP_ATOMIC);
4667 	if (!work)
4668 		return NOTIFY_DONE;
4669 
4670 	INIT_WORK(&work->work, mlx5_ib_handle_event);
4671 	work->dev = container_of(nb, struct mlx5_ib_dev, mdev_events);
4672 	work->is_slave = false;
4673 	work->param = param;
4674 	work->event = event;
4675 
4676 	queue_work(mlx5_ib_event_wq, &work->work);
4677 
4678 	return NOTIFY_OK;
4679 }
4680 
4681 static int mlx5_ib_event_slave_port(struct notifier_block *nb,
4682 				    unsigned long event, void *param)
4683 {
4684 	struct mlx5_ib_event_work *work;
4685 
4686 	work = kmalloc(sizeof(*work), GFP_ATOMIC);
4687 	if (!work)
4688 		return NOTIFY_DONE;
4689 
4690 	INIT_WORK(&work->work, mlx5_ib_handle_event);
4691 	work->mpi = container_of(nb, struct mlx5_ib_multiport_info, mdev_events);
4692 	work->is_slave = true;
4693 	work->param = param;
4694 	work->event = event;
4695 	queue_work(mlx5_ib_event_wq, &work->work);
4696 
4697 	return NOTIFY_OK;
4698 }
4699 
4700 static int set_has_smi_cap(struct mlx5_ib_dev *dev)
4701 {
4702 	struct mlx5_hca_vport_context vport_ctx;
4703 	int err;
4704 	int port;
4705 
4706 	for (port = 1; port <= ARRAY_SIZE(dev->mdev->port_caps); port++) {
4707 		dev->mdev->port_caps[port - 1].has_smi = false;
4708 		if (MLX5_CAP_GEN(dev->mdev, port_type) ==
4709 		    MLX5_CAP_PORT_TYPE_IB) {
4710 			if (MLX5_CAP_GEN(dev->mdev, ib_virt)) {
4711 				err = mlx5_query_hca_vport_context(dev->mdev, 0,
4712 								   port, 0,
4713 								   &vport_ctx);
4714 				if (err) {
4715 					mlx5_ib_err(dev, "query_hca_vport_context for port=%d failed %d\n",
4716 						    port, err);
4717 					return err;
4718 				}
4719 				dev->mdev->port_caps[port - 1].has_smi =
4720 					vport_ctx.has_smi;
4721 			} else {
4722 				dev->mdev->port_caps[port - 1].has_smi = true;
4723 			}
4724 		}
4725 	}
4726 	return 0;
4727 }
4728 
4729 static void get_ext_port_caps(struct mlx5_ib_dev *dev)
4730 {
4731 	int port;
4732 
4733 	for (port = 1; port <= dev->num_ports; port++)
4734 		mlx5_query_ext_port_caps(dev, port);
4735 }
4736 
4737 static int __get_port_caps(struct mlx5_ib_dev *dev, u8 port)
4738 {
4739 	struct ib_device_attr *dprops = NULL;
4740 	struct ib_port_attr *pprops = NULL;
4741 	int err = -ENOMEM;
4742 	struct ib_udata uhw = {.inlen = 0, .outlen = 0};
4743 
4744 	pprops = kzalloc(sizeof(*pprops), GFP_KERNEL);
4745 	if (!pprops)
4746 		goto out;
4747 
4748 	dprops = kmalloc(sizeof(*dprops), GFP_KERNEL);
4749 	if (!dprops)
4750 		goto out;
4751 
4752 	err = mlx5_ib_query_device(&dev->ib_dev, dprops, &uhw);
4753 	if (err) {
4754 		mlx5_ib_warn(dev, "query_device failed %d\n", err);
4755 		goto out;
4756 	}
4757 
4758 	err = mlx5_ib_query_port(&dev->ib_dev, port, pprops);
4759 	if (err) {
4760 		mlx5_ib_warn(dev, "query_port %d failed %d\n",
4761 			     port, err);
4762 		goto out;
4763 	}
4764 
4765 	dev->mdev->port_caps[port - 1].pkey_table_len =
4766 					dprops->max_pkeys;
4767 	dev->mdev->port_caps[port - 1].gid_table_len =
4768 					pprops->gid_tbl_len;
4769 	mlx5_ib_dbg(dev, "port %d: pkey_table_len %d, gid_table_len %d\n",
4770 		    port, dprops->max_pkeys, pprops->gid_tbl_len);
4771 
4772 out:
4773 	kfree(pprops);
4774 	kfree(dprops);
4775 
4776 	return err;
4777 }
4778 
4779 static int get_port_caps(struct mlx5_ib_dev *dev, u8 port)
4780 {
4781 	/* For representors use port 1, is this is the only native
4782 	 * port
4783 	 */
4784 	if (dev->is_rep)
4785 		return __get_port_caps(dev, 1);
4786 	return __get_port_caps(dev, port);
4787 }
4788 
4789 static void destroy_umrc_res(struct mlx5_ib_dev *dev)
4790 {
4791 	int err;
4792 
4793 	err = mlx5_mr_cache_cleanup(dev);
4794 	if (err)
4795 		mlx5_ib_warn(dev, "mr cache cleanup failed\n");
4796 
4797 	if (dev->umrc.qp)
4798 		mlx5_ib_destroy_qp(dev->umrc.qp, NULL);
4799 	if (dev->umrc.cq)
4800 		ib_free_cq(dev->umrc.cq);
4801 	if (dev->umrc.pd)
4802 		ib_dealloc_pd(dev->umrc.pd);
4803 }
4804 
4805 enum {
4806 	MAX_UMR_WR = 128,
4807 };
4808 
4809 static int create_umr_res(struct mlx5_ib_dev *dev)
4810 {
4811 	struct ib_qp_init_attr *init_attr = NULL;
4812 	struct ib_qp_attr *attr = NULL;
4813 	struct ib_pd *pd;
4814 	struct ib_cq *cq;
4815 	struct ib_qp *qp;
4816 	int ret;
4817 
4818 	attr = kzalloc(sizeof(*attr), GFP_KERNEL);
4819 	init_attr = kzalloc(sizeof(*init_attr), GFP_KERNEL);
4820 	if (!attr || !init_attr) {
4821 		ret = -ENOMEM;
4822 		goto error_0;
4823 	}
4824 
4825 	pd = ib_alloc_pd(&dev->ib_dev, 0);
4826 	if (IS_ERR(pd)) {
4827 		mlx5_ib_dbg(dev, "Couldn't create PD for sync UMR QP\n");
4828 		ret = PTR_ERR(pd);
4829 		goto error_0;
4830 	}
4831 
4832 	cq = ib_alloc_cq(&dev->ib_dev, NULL, 128, 0, IB_POLL_SOFTIRQ);
4833 	if (IS_ERR(cq)) {
4834 		mlx5_ib_dbg(dev, "Couldn't create CQ for sync UMR QP\n");
4835 		ret = PTR_ERR(cq);
4836 		goto error_2;
4837 	}
4838 
4839 	init_attr->send_cq = cq;
4840 	init_attr->recv_cq = cq;
4841 	init_attr->sq_sig_type = IB_SIGNAL_ALL_WR;
4842 	init_attr->cap.max_send_wr = MAX_UMR_WR;
4843 	init_attr->cap.max_send_sge = 1;
4844 	init_attr->qp_type = MLX5_IB_QPT_REG_UMR;
4845 	init_attr->port_num = 1;
4846 	qp = mlx5_ib_create_qp(pd, init_attr, NULL);
4847 	if (IS_ERR(qp)) {
4848 		mlx5_ib_dbg(dev, "Couldn't create sync UMR QP\n");
4849 		ret = PTR_ERR(qp);
4850 		goto error_3;
4851 	}
4852 	qp->device     = &dev->ib_dev;
4853 	qp->real_qp    = qp;
4854 	qp->uobject    = NULL;
4855 	qp->qp_type    = MLX5_IB_QPT_REG_UMR;
4856 	qp->send_cq    = init_attr->send_cq;
4857 	qp->recv_cq    = init_attr->recv_cq;
4858 
4859 	attr->qp_state = IB_QPS_INIT;
4860 	attr->port_num = 1;
4861 	ret = mlx5_ib_modify_qp(qp, attr, IB_QP_STATE | IB_QP_PKEY_INDEX |
4862 				IB_QP_PORT, NULL);
4863 	if (ret) {
4864 		mlx5_ib_dbg(dev, "Couldn't modify UMR QP\n");
4865 		goto error_4;
4866 	}
4867 
4868 	memset(attr, 0, sizeof(*attr));
4869 	attr->qp_state = IB_QPS_RTR;
4870 	attr->path_mtu = IB_MTU_256;
4871 
4872 	ret = mlx5_ib_modify_qp(qp, attr, IB_QP_STATE, NULL);
4873 	if (ret) {
4874 		mlx5_ib_dbg(dev, "Couldn't modify umr QP to rtr\n");
4875 		goto error_4;
4876 	}
4877 
4878 	memset(attr, 0, sizeof(*attr));
4879 	attr->qp_state = IB_QPS_RTS;
4880 	ret = mlx5_ib_modify_qp(qp, attr, IB_QP_STATE, NULL);
4881 	if (ret) {
4882 		mlx5_ib_dbg(dev, "Couldn't modify umr QP to rts\n");
4883 		goto error_4;
4884 	}
4885 
4886 	dev->umrc.qp = qp;
4887 	dev->umrc.cq = cq;
4888 	dev->umrc.pd = pd;
4889 
4890 	sema_init(&dev->umrc.sem, MAX_UMR_WR);
4891 	ret = mlx5_mr_cache_init(dev);
4892 	if (ret) {
4893 		mlx5_ib_warn(dev, "mr cache init failed %d\n", ret);
4894 		goto error_4;
4895 	}
4896 
4897 	kfree(attr);
4898 	kfree(init_attr);
4899 
4900 	return 0;
4901 
4902 error_4:
4903 	mlx5_ib_destroy_qp(qp, NULL);
4904 	dev->umrc.qp = NULL;
4905 
4906 error_3:
4907 	ib_free_cq(cq);
4908 	dev->umrc.cq = NULL;
4909 
4910 error_2:
4911 	ib_dealloc_pd(pd);
4912 	dev->umrc.pd = NULL;
4913 
4914 error_0:
4915 	kfree(attr);
4916 	kfree(init_attr);
4917 	return ret;
4918 }
4919 
4920 static u8 mlx5_get_umr_fence(u8 umr_fence_cap)
4921 {
4922 	switch (umr_fence_cap) {
4923 	case MLX5_CAP_UMR_FENCE_NONE:
4924 		return MLX5_FENCE_MODE_NONE;
4925 	case MLX5_CAP_UMR_FENCE_SMALL:
4926 		return MLX5_FENCE_MODE_INITIATOR_SMALL;
4927 	default:
4928 		return MLX5_FENCE_MODE_STRONG_ORDERING;
4929 	}
4930 }
4931 
4932 static int create_dev_resources(struct mlx5_ib_resources *devr)
4933 {
4934 	struct ib_srq_init_attr attr;
4935 	struct mlx5_ib_dev *dev;
4936 	struct ib_device *ibdev;
4937 	struct ib_cq_init_attr cq_attr = {.cqe = 1};
4938 	int port;
4939 	int ret = 0;
4940 
4941 	dev = container_of(devr, struct mlx5_ib_dev, devr);
4942 	ibdev = &dev->ib_dev;
4943 
4944 	mutex_init(&devr->mutex);
4945 
4946 	devr->p0 = rdma_zalloc_drv_obj(ibdev, ib_pd);
4947 	if (!devr->p0)
4948 		return -ENOMEM;
4949 
4950 	devr->p0->device  = ibdev;
4951 	devr->p0->uobject = NULL;
4952 	atomic_set(&devr->p0->usecnt, 0);
4953 
4954 	ret = mlx5_ib_alloc_pd(devr->p0, NULL);
4955 	if (ret)
4956 		goto error0;
4957 
4958 	devr->c0 = rdma_zalloc_drv_obj(ibdev, ib_cq);
4959 	if (!devr->c0) {
4960 		ret = -ENOMEM;
4961 		goto error1;
4962 	}
4963 
4964 	devr->c0->device = &dev->ib_dev;
4965 	atomic_set(&devr->c0->usecnt, 0);
4966 
4967 	ret = mlx5_ib_create_cq(devr->c0, &cq_attr, NULL);
4968 	if (ret)
4969 		goto err_create_cq;
4970 
4971 	devr->x0 = mlx5_ib_alloc_xrcd(&dev->ib_dev, NULL);
4972 	if (IS_ERR(devr->x0)) {
4973 		ret = PTR_ERR(devr->x0);
4974 		goto error2;
4975 	}
4976 	devr->x0->device = &dev->ib_dev;
4977 	devr->x0->inode = NULL;
4978 	atomic_set(&devr->x0->usecnt, 0);
4979 	mutex_init(&devr->x0->tgt_qp_mutex);
4980 	INIT_LIST_HEAD(&devr->x0->tgt_qp_list);
4981 
4982 	devr->x1 = mlx5_ib_alloc_xrcd(&dev->ib_dev, NULL);
4983 	if (IS_ERR(devr->x1)) {
4984 		ret = PTR_ERR(devr->x1);
4985 		goto error3;
4986 	}
4987 	devr->x1->device = &dev->ib_dev;
4988 	devr->x1->inode = NULL;
4989 	atomic_set(&devr->x1->usecnt, 0);
4990 	mutex_init(&devr->x1->tgt_qp_mutex);
4991 	INIT_LIST_HEAD(&devr->x1->tgt_qp_list);
4992 
4993 	memset(&attr, 0, sizeof(attr));
4994 	attr.attr.max_sge = 1;
4995 	attr.attr.max_wr = 1;
4996 	attr.srq_type = IB_SRQT_XRC;
4997 	attr.ext.cq = devr->c0;
4998 	attr.ext.xrc.xrcd = devr->x0;
4999 
5000 	devr->s0 = rdma_zalloc_drv_obj(ibdev, ib_srq);
5001 	if (!devr->s0) {
5002 		ret = -ENOMEM;
5003 		goto error4;
5004 	}
5005 
5006 	devr->s0->device	= &dev->ib_dev;
5007 	devr->s0->pd		= devr->p0;
5008 	devr->s0->srq_type      = IB_SRQT_XRC;
5009 	devr->s0->ext.xrc.xrcd	= devr->x0;
5010 	devr->s0->ext.cq	= devr->c0;
5011 	ret = mlx5_ib_create_srq(devr->s0, &attr, NULL);
5012 	if (ret)
5013 		goto err_create;
5014 
5015 	atomic_inc(&devr->s0->ext.xrc.xrcd->usecnt);
5016 	atomic_inc(&devr->s0->ext.cq->usecnt);
5017 	atomic_inc(&devr->p0->usecnt);
5018 	atomic_set(&devr->s0->usecnt, 0);
5019 
5020 	memset(&attr, 0, sizeof(attr));
5021 	attr.attr.max_sge = 1;
5022 	attr.attr.max_wr = 1;
5023 	attr.srq_type = IB_SRQT_BASIC;
5024 	devr->s1 = rdma_zalloc_drv_obj(ibdev, ib_srq);
5025 	if (!devr->s1) {
5026 		ret = -ENOMEM;
5027 		goto error5;
5028 	}
5029 
5030 	devr->s1->device	= &dev->ib_dev;
5031 	devr->s1->pd		= devr->p0;
5032 	devr->s1->srq_type      = IB_SRQT_BASIC;
5033 	devr->s1->ext.cq	= devr->c0;
5034 
5035 	ret = mlx5_ib_create_srq(devr->s1, &attr, NULL);
5036 	if (ret)
5037 		goto error6;
5038 
5039 	atomic_inc(&devr->p0->usecnt);
5040 	atomic_set(&devr->s1->usecnt, 0);
5041 
5042 	for (port = 0; port < ARRAY_SIZE(devr->ports); ++port) {
5043 		INIT_WORK(&devr->ports[port].pkey_change_work,
5044 			  pkey_change_handler);
5045 		devr->ports[port].devr = devr;
5046 	}
5047 
5048 	return 0;
5049 
5050 error6:
5051 	kfree(devr->s1);
5052 error5:
5053 	mlx5_ib_destroy_srq(devr->s0, NULL);
5054 err_create:
5055 	kfree(devr->s0);
5056 error4:
5057 	mlx5_ib_dealloc_xrcd(devr->x1, NULL);
5058 error3:
5059 	mlx5_ib_dealloc_xrcd(devr->x0, NULL);
5060 error2:
5061 	mlx5_ib_destroy_cq(devr->c0, NULL);
5062 err_create_cq:
5063 	kfree(devr->c0);
5064 error1:
5065 	mlx5_ib_dealloc_pd(devr->p0, NULL);
5066 error0:
5067 	kfree(devr->p0);
5068 	return ret;
5069 }
5070 
5071 static void destroy_dev_resources(struct mlx5_ib_resources *devr)
5072 {
5073 	int port;
5074 
5075 	mlx5_ib_destroy_srq(devr->s1, NULL);
5076 	kfree(devr->s1);
5077 	mlx5_ib_destroy_srq(devr->s0, NULL);
5078 	kfree(devr->s0);
5079 	mlx5_ib_dealloc_xrcd(devr->x0, NULL);
5080 	mlx5_ib_dealloc_xrcd(devr->x1, NULL);
5081 	mlx5_ib_destroy_cq(devr->c0, NULL);
5082 	kfree(devr->c0);
5083 	mlx5_ib_dealloc_pd(devr->p0, NULL);
5084 	kfree(devr->p0);
5085 
5086 	/* Make sure no change P_Key work items are still executing */
5087 	for (port = 0; port < ARRAY_SIZE(devr->ports); ++port)
5088 		cancel_work_sync(&devr->ports[port].pkey_change_work);
5089 }
5090 
5091 static u32 get_core_cap_flags(struct ib_device *ibdev,
5092 			      struct mlx5_hca_vport_context *rep)
5093 {
5094 	struct mlx5_ib_dev *dev = to_mdev(ibdev);
5095 	enum rdma_link_layer ll = mlx5_ib_port_link_layer(ibdev, 1);
5096 	u8 l3_type_cap = MLX5_CAP_ROCE(dev->mdev, l3_type);
5097 	u8 roce_version_cap = MLX5_CAP_ROCE(dev->mdev, roce_version);
5098 	bool raw_support = !mlx5_core_mp_enabled(dev->mdev);
5099 	u32 ret = 0;
5100 
5101 	if (rep->grh_required)
5102 		ret |= RDMA_CORE_CAP_IB_GRH_REQUIRED;
5103 
5104 	if (ll == IB_LINK_LAYER_INFINIBAND)
5105 		return ret | RDMA_CORE_PORT_IBA_IB;
5106 
5107 	if (raw_support)
5108 		ret |= RDMA_CORE_PORT_RAW_PACKET;
5109 
5110 	if (!(l3_type_cap & MLX5_ROCE_L3_TYPE_IPV4_CAP))
5111 		return ret;
5112 
5113 	if (!(l3_type_cap & MLX5_ROCE_L3_TYPE_IPV6_CAP))
5114 		return ret;
5115 
5116 	if (roce_version_cap & MLX5_ROCE_VERSION_1_CAP)
5117 		ret |= RDMA_CORE_PORT_IBA_ROCE;
5118 
5119 	if (roce_version_cap & MLX5_ROCE_VERSION_2_CAP)
5120 		ret |= RDMA_CORE_PORT_IBA_ROCE_UDP_ENCAP;
5121 
5122 	return ret;
5123 }
5124 
5125 static int mlx5_port_immutable(struct ib_device *ibdev, u8 port_num,
5126 			       struct ib_port_immutable *immutable)
5127 {
5128 	struct ib_port_attr attr;
5129 	struct mlx5_ib_dev *dev = to_mdev(ibdev);
5130 	enum rdma_link_layer ll = mlx5_ib_port_link_layer(ibdev, port_num);
5131 	struct mlx5_hca_vport_context rep = {0};
5132 	int err;
5133 
5134 	err = ib_query_port(ibdev, port_num, &attr);
5135 	if (err)
5136 		return err;
5137 
5138 	if (ll == IB_LINK_LAYER_INFINIBAND) {
5139 		err = mlx5_query_hca_vport_context(dev->mdev, 0, port_num, 0,
5140 						   &rep);
5141 		if (err)
5142 			return err;
5143 	}
5144 
5145 	immutable->pkey_tbl_len = attr.pkey_tbl_len;
5146 	immutable->gid_tbl_len = attr.gid_tbl_len;
5147 	immutable->core_cap_flags = get_core_cap_flags(ibdev, &rep);
5148 	if ((ll == IB_LINK_LAYER_INFINIBAND) || MLX5_CAP_GEN(dev->mdev, roce))
5149 		immutable->max_mad_size = IB_MGMT_MAD_SIZE;
5150 
5151 	return 0;
5152 }
5153 
5154 static int mlx5_port_rep_immutable(struct ib_device *ibdev, u8 port_num,
5155 				   struct ib_port_immutable *immutable)
5156 {
5157 	struct ib_port_attr attr;
5158 	int err;
5159 
5160 	immutable->core_cap_flags = RDMA_CORE_PORT_RAW_PACKET;
5161 
5162 	err = ib_query_port(ibdev, port_num, &attr);
5163 	if (err)
5164 		return err;
5165 
5166 	immutable->pkey_tbl_len = attr.pkey_tbl_len;
5167 	immutable->gid_tbl_len = attr.gid_tbl_len;
5168 	immutable->core_cap_flags = RDMA_CORE_PORT_RAW_PACKET;
5169 
5170 	return 0;
5171 }
5172 
5173 static void get_dev_fw_str(struct ib_device *ibdev, char *str)
5174 {
5175 	struct mlx5_ib_dev *dev =
5176 		container_of(ibdev, struct mlx5_ib_dev, ib_dev);
5177 	snprintf(str, IB_FW_VERSION_NAME_MAX, "%d.%d.%04d",
5178 		 fw_rev_maj(dev->mdev), fw_rev_min(dev->mdev),
5179 		 fw_rev_sub(dev->mdev));
5180 }
5181 
5182 static int mlx5_eth_lag_init(struct mlx5_ib_dev *dev)
5183 {
5184 	struct mlx5_core_dev *mdev = dev->mdev;
5185 	struct mlx5_flow_namespace *ns = mlx5_get_flow_namespace(mdev,
5186 								 MLX5_FLOW_NAMESPACE_LAG);
5187 	struct mlx5_flow_table *ft;
5188 	int err;
5189 
5190 	if (!ns || !mlx5_lag_is_roce(mdev))
5191 		return 0;
5192 
5193 	err = mlx5_cmd_create_vport_lag(mdev);
5194 	if (err)
5195 		return err;
5196 
5197 	ft = mlx5_create_lag_demux_flow_table(ns, 0, 0);
5198 	if (IS_ERR(ft)) {
5199 		err = PTR_ERR(ft);
5200 		goto err_destroy_vport_lag;
5201 	}
5202 
5203 	dev->flow_db->lag_demux_ft = ft;
5204 	dev->lag_active = true;
5205 	return 0;
5206 
5207 err_destroy_vport_lag:
5208 	mlx5_cmd_destroy_vport_lag(mdev);
5209 	return err;
5210 }
5211 
5212 static void mlx5_eth_lag_cleanup(struct mlx5_ib_dev *dev)
5213 {
5214 	struct mlx5_core_dev *mdev = dev->mdev;
5215 
5216 	if (dev->lag_active) {
5217 		dev->lag_active = false;
5218 
5219 		mlx5_destroy_flow_table(dev->flow_db->lag_demux_ft);
5220 		dev->flow_db->lag_demux_ft = NULL;
5221 
5222 		mlx5_cmd_destroy_vport_lag(mdev);
5223 	}
5224 }
5225 
5226 static int mlx5_add_netdev_notifier(struct mlx5_ib_dev *dev, u8 port_num)
5227 {
5228 	int err;
5229 
5230 	dev->port[port_num].roce.nb.notifier_call = mlx5_netdev_event;
5231 	err = register_netdevice_notifier(&dev->port[port_num].roce.nb);
5232 	if (err) {
5233 		dev->port[port_num].roce.nb.notifier_call = NULL;
5234 		return err;
5235 	}
5236 
5237 	return 0;
5238 }
5239 
5240 static void mlx5_remove_netdev_notifier(struct mlx5_ib_dev *dev, u8 port_num)
5241 {
5242 	if (dev->port[port_num].roce.nb.notifier_call) {
5243 		unregister_netdevice_notifier(&dev->port[port_num].roce.nb);
5244 		dev->port[port_num].roce.nb.notifier_call = NULL;
5245 	}
5246 }
5247 
5248 static int mlx5_enable_eth(struct mlx5_ib_dev *dev)
5249 {
5250 	int err;
5251 
5252 	if (MLX5_CAP_GEN(dev->mdev, roce)) {
5253 		err = mlx5_nic_vport_enable_roce(dev->mdev);
5254 		if (err)
5255 			return err;
5256 	}
5257 
5258 	err = mlx5_eth_lag_init(dev);
5259 	if (err)
5260 		goto err_disable_roce;
5261 
5262 	return 0;
5263 
5264 err_disable_roce:
5265 	if (MLX5_CAP_GEN(dev->mdev, roce))
5266 		mlx5_nic_vport_disable_roce(dev->mdev);
5267 
5268 	return err;
5269 }
5270 
5271 static void mlx5_disable_eth(struct mlx5_ib_dev *dev)
5272 {
5273 	mlx5_eth_lag_cleanup(dev);
5274 	if (MLX5_CAP_GEN(dev->mdev, roce))
5275 		mlx5_nic_vport_disable_roce(dev->mdev);
5276 }
5277 
5278 struct mlx5_ib_counter {
5279 	const char *name;
5280 	size_t offset;
5281 };
5282 
5283 #define INIT_Q_COUNTER(_name)		\
5284 	{ .name = #_name, .offset = MLX5_BYTE_OFF(query_q_counter_out, _name)}
5285 
5286 static const struct mlx5_ib_counter basic_q_cnts[] = {
5287 	INIT_Q_COUNTER(rx_write_requests),
5288 	INIT_Q_COUNTER(rx_read_requests),
5289 	INIT_Q_COUNTER(rx_atomic_requests),
5290 	INIT_Q_COUNTER(out_of_buffer),
5291 };
5292 
5293 static const struct mlx5_ib_counter out_of_seq_q_cnts[] = {
5294 	INIT_Q_COUNTER(out_of_sequence),
5295 };
5296 
5297 static const struct mlx5_ib_counter retrans_q_cnts[] = {
5298 	INIT_Q_COUNTER(duplicate_request),
5299 	INIT_Q_COUNTER(rnr_nak_retry_err),
5300 	INIT_Q_COUNTER(packet_seq_err),
5301 	INIT_Q_COUNTER(implied_nak_seq_err),
5302 	INIT_Q_COUNTER(local_ack_timeout_err),
5303 };
5304 
5305 #define INIT_CONG_COUNTER(_name)		\
5306 	{ .name = #_name, .offset =	\
5307 		MLX5_BYTE_OFF(query_cong_statistics_out, _name ## _high)}
5308 
5309 static const struct mlx5_ib_counter cong_cnts[] = {
5310 	INIT_CONG_COUNTER(rp_cnp_ignored),
5311 	INIT_CONG_COUNTER(rp_cnp_handled),
5312 	INIT_CONG_COUNTER(np_ecn_marked_roce_packets),
5313 	INIT_CONG_COUNTER(np_cnp_sent),
5314 };
5315 
5316 static const struct mlx5_ib_counter extended_err_cnts[] = {
5317 	INIT_Q_COUNTER(resp_local_length_error),
5318 	INIT_Q_COUNTER(resp_cqe_error),
5319 	INIT_Q_COUNTER(req_cqe_error),
5320 	INIT_Q_COUNTER(req_remote_invalid_request),
5321 	INIT_Q_COUNTER(req_remote_access_errors),
5322 	INIT_Q_COUNTER(resp_remote_access_errors),
5323 	INIT_Q_COUNTER(resp_cqe_flush_error),
5324 	INIT_Q_COUNTER(req_cqe_flush_error),
5325 };
5326 
5327 #define INIT_EXT_PPCNT_COUNTER(_name)		\
5328 	{ .name = #_name, .offset =	\
5329 	MLX5_BYTE_OFF(ppcnt_reg, \
5330 		      counter_set.eth_extended_cntrs_grp_data_layout._name##_high)}
5331 
5332 static const struct mlx5_ib_counter ext_ppcnt_cnts[] = {
5333 	INIT_EXT_PPCNT_COUNTER(rx_icrc_encapsulated),
5334 };
5335 
5336 static bool is_mdev_switchdev_mode(const struct mlx5_core_dev *mdev)
5337 {
5338 	return MLX5_ESWITCH_MANAGER(mdev) &&
5339 	       mlx5_ib_eswitch_mode(mdev->priv.eswitch) ==
5340 		       MLX5_ESWITCH_OFFLOADS;
5341 }
5342 
5343 static void mlx5_ib_dealloc_counters(struct mlx5_ib_dev *dev)
5344 {
5345 	int num_cnt_ports;
5346 	int i;
5347 
5348 	num_cnt_ports = is_mdev_switchdev_mode(dev->mdev) ? 1 : dev->num_ports;
5349 
5350 	for (i = 0; i < num_cnt_ports; i++) {
5351 		if (dev->port[i].cnts.set_id_valid)
5352 			mlx5_core_dealloc_q_counter(dev->mdev,
5353 						    dev->port[i].cnts.set_id);
5354 		kfree(dev->port[i].cnts.names);
5355 		kfree(dev->port[i].cnts.offsets);
5356 	}
5357 }
5358 
5359 static int __mlx5_ib_alloc_counters(struct mlx5_ib_dev *dev,
5360 				    struct mlx5_ib_counters *cnts)
5361 {
5362 	u32 num_counters;
5363 
5364 	num_counters = ARRAY_SIZE(basic_q_cnts);
5365 
5366 	if (MLX5_CAP_GEN(dev->mdev, out_of_seq_cnt))
5367 		num_counters += ARRAY_SIZE(out_of_seq_q_cnts);
5368 
5369 	if (MLX5_CAP_GEN(dev->mdev, retransmission_q_counters))
5370 		num_counters += ARRAY_SIZE(retrans_q_cnts);
5371 
5372 	if (MLX5_CAP_GEN(dev->mdev, enhanced_error_q_counters))
5373 		num_counters += ARRAY_SIZE(extended_err_cnts);
5374 
5375 	cnts->num_q_counters = num_counters;
5376 
5377 	if (MLX5_CAP_GEN(dev->mdev, cc_query_allowed)) {
5378 		cnts->num_cong_counters = ARRAY_SIZE(cong_cnts);
5379 		num_counters += ARRAY_SIZE(cong_cnts);
5380 	}
5381 	if (MLX5_CAP_PCAM_FEATURE(dev->mdev, rx_icrc_encapsulated_counter)) {
5382 		cnts->num_ext_ppcnt_counters = ARRAY_SIZE(ext_ppcnt_cnts);
5383 		num_counters += ARRAY_SIZE(ext_ppcnt_cnts);
5384 	}
5385 	cnts->names = kcalloc(num_counters, sizeof(cnts->names), GFP_KERNEL);
5386 	if (!cnts->names)
5387 		return -ENOMEM;
5388 
5389 	cnts->offsets = kcalloc(num_counters,
5390 				sizeof(cnts->offsets), GFP_KERNEL);
5391 	if (!cnts->offsets)
5392 		goto err_names;
5393 
5394 	return 0;
5395 
5396 err_names:
5397 	kfree(cnts->names);
5398 	cnts->names = NULL;
5399 	return -ENOMEM;
5400 }
5401 
5402 static void mlx5_ib_fill_counters(struct mlx5_ib_dev *dev,
5403 				  const char **names,
5404 				  size_t *offsets)
5405 {
5406 	int i;
5407 	int j = 0;
5408 
5409 	for (i = 0; i < ARRAY_SIZE(basic_q_cnts); i++, j++) {
5410 		names[j] = basic_q_cnts[i].name;
5411 		offsets[j] = basic_q_cnts[i].offset;
5412 	}
5413 
5414 	if (MLX5_CAP_GEN(dev->mdev, out_of_seq_cnt)) {
5415 		for (i = 0; i < ARRAY_SIZE(out_of_seq_q_cnts); i++, j++) {
5416 			names[j] = out_of_seq_q_cnts[i].name;
5417 			offsets[j] = out_of_seq_q_cnts[i].offset;
5418 		}
5419 	}
5420 
5421 	if (MLX5_CAP_GEN(dev->mdev, retransmission_q_counters)) {
5422 		for (i = 0; i < ARRAY_SIZE(retrans_q_cnts); i++, j++) {
5423 			names[j] = retrans_q_cnts[i].name;
5424 			offsets[j] = retrans_q_cnts[i].offset;
5425 		}
5426 	}
5427 
5428 	if (MLX5_CAP_GEN(dev->mdev, enhanced_error_q_counters)) {
5429 		for (i = 0; i < ARRAY_SIZE(extended_err_cnts); i++, j++) {
5430 			names[j] = extended_err_cnts[i].name;
5431 			offsets[j] = extended_err_cnts[i].offset;
5432 		}
5433 	}
5434 
5435 	if (MLX5_CAP_GEN(dev->mdev, cc_query_allowed)) {
5436 		for (i = 0; i < ARRAY_SIZE(cong_cnts); i++, j++) {
5437 			names[j] = cong_cnts[i].name;
5438 			offsets[j] = cong_cnts[i].offset;
5439 		}
5440 	}
5441 
5442 	if (MLX5_CAP_PCAM_FEATURE(dev->mdev, rx_icrc_encapsulated_counter)) {
5443 		for (i = 0; i < ARRAY_SIZE(ext_ppcnt_cnts); i++, j++) {
5444 			names[j] = ext_ppcnt_cnts[i].name;
5445 			offsets[j] = ext_ppcnt_cnts[i].offset;
5446 		}
5447 	}
5448 }
5449 
5450 static int mlx5_ib_alloc_counters(struct mlx5_ib_dev *dev)
5451 {
5452 	int num_cnt_ports;
5453 	int err = 0;
5454 	int i;
5455 	bool is_shared;
5456 
5457 	is_shared = MLX5_CAP_GEN(dev->mdev, log_max_uctx) != 0;
5458 	num_cnt_ports = is_mdev_switchdev_mode(dev->mdev) ? 1 : dev->num_ports;
5459 
5460 	for (i = 0; i < num_cnt_ports; i++) {
5461 		err = __mlx5_ib_alloc_counters(dev, &dev->port[i].cnts);
5462 		if (err)
5463 			goto err_alloc;
5464 
5465 		mlx5_ib_fill_counters(dev, dev->port[i].cnts.names,
5466 				      dev->port[i].cnts.offsets);
5467 
5468 		err = mlx5_cmd_alloc_q_counter(dev->mdev,
5469 					       &dev->port[i].cnts.set_id,
5470 					       is_shared ?
5471 					       MLX5_SHARED_RESOURCE_UID : 0);
5472 		if (err) {
5473 			mlx5_ib_warn(dev,
5474 				     "couldn't allocate queue counter for port %d, err %d\n",
5475 				     i + 1, err);
5476 			goto err_alloc;
5477 		}
5478 		dev->port[i].cnts.set_id_valid = true;
5479 	}
5480 	return 0;
5481 
5482 err_alloc:
5483 	mlx5_ib_dealloc_counters(dev);
5484 	return err;
5485 }
5486 
5487 static const struct mlx5_ib_counters *get_counters(struct mlx5_ib_dev *dev,
5488 						   u8 port_num)
5489 {
5490 	return is_mdev_switchdev_mode(dev->mdev) ? &dev->port[0].cnts :
5491 						   &dev->port[port_num].cnts;
5492 }
5493 
5494 /**
5495  * mlx5_ib_get_counters_id - Returns counters id to use for device+port
5496  * @dev:	Pointer to mlx5 IB device
5497  * @port_num:	Zero based port number
5498  *
5499  * mlx5_ib_get_counters_id() Returns counters set id to use for given
5500  * device port combination in switchdev and non switchdev mode of the
5501  * parent device.
5502  */
5503 u16 mlx5_ib_get_counters_id(struct mlx5_ib_dev *dev, u8 port_num)
5504 {
5505 	const struct mlx5_ib_counters *cnts = get_counters(dev, port_num);
5506 
5507 	return cnts->set_id;
5508 }
5509 
5510 static struct rdma_hw_stats *mlx5_ib_alloc_hw_stats(struct ib_device *ibdev,
5511 						    u8 port_num)
5512 {
5513 	struct mlx5_ib_dev *dev = to_mdev(ibdev);
5514 	const struct mlx5_ib_counters *cnts;
5515 	bool is_switchdev = is_mdev_switchdev_mode(dev->mdev);
5516 
5517 	if ((is_switchdev && port_num) || (!is_switchdev && !port_num))
5518 		return NULL;
5519 
5520 	cnts = get_counters(dev, port_num - 1);
5521 
5522 	return rdma_alloc_hw_stats_struct(cnts->names,
5523 					  cnts->num_q_counters +
5524 					  cnts->num_cong_counters +
5525 					  cnts->num_ext_ppcnt_counters,
5526 					  RDMA_HW_STATS_DEFAULT_LIFESPAN);
5527 }
5528 
5529 static int mlx5_ib_query_q_counters(struct mlx5_core_dev *mdev,
5530 				    const struct mlx5_ib_counters *cnts,
5531 				    struct rdma_hw_stats *stats,
5532 				    u16 set_id)
5533 {
5534 	int outlen = MLX5_ST_SZ_BYTES(query_q_counter_out);
5535 	void *out;
5536 	__be32 val;
5537 	int ret, i;
5538 
5539 	out = kvzalloc(outlen, GFP_KERNEL);
5540 	if (!out)
5541 		return -ENOMEM;
5542 
5543 	ret = mlx5_core_query_q_counter(mdev, set_id, 0, out, outlen);
5544 	if (ret)
5545 		goto free;
5546 
5547 	for (i = 0; i < cnts->num_q_counters; i++) {
5548 		val = *(__be32 *)(out + cnts->offsets[i]);
5549 		stats->value[i] = (u64)be32_to_cpu(val);
5550 	}
5551 
5552 free:
5553 	kvfree(out);
5554 	return ret;
5555 }
5556 
5557 static int mlx5_ib_query_ext_ppcnt_counters(struct mlx5_ib_dev *dev,
5558 					    const struct mlx5_ib_counters *cnts,
5559 					    struct rdma_hw_stats *stats)
5560 {
5561 	int offset = cnts->num_q_counters + cnts->num_cong_counters;
5562 	int sz = MLX5_ST_SZ_BYTES(ppcnt_reg);
5563 	int ret, i;
5564 	void *out;
5565 
5566 	out = kvzalloc(sz, GFP_KERNEL);
5567 	if (!out)
5568 		return -ENOMEM;
5569 
5570 	ret = mlx5_cmd_query_ext_ppcnt_counters(dev->mdev, out);
5571 	if (ret)
5572 		goto free;
5573 
5574 	for (i = 0; i < cnts->num_ext_ppcnt_counters; i++)
5575 		stats->value[i + offset] =
5576 			be64_to_cpup((__be64 *)(out +
5577 				    cnts->offsets[i + offset]));
5578 free:
5579 	kvfree(out);
5580 	return ret;
5581 }
5582 
5583 static int mlx5_ib_get_hw_stats(struct ib_device *ibdev,
5584 				struct rdma_hw_stats *stats,
5585 				u8 port_num, int index)
5586 {
5587 	struct mlx5_ib_dev *dev = to_mdev(ibdev);
5588 	const struct mlx5_ib_counters *cnts = get_counters(dev, port_num - 1);
5589 	struct mlx5_core_dev *mdev;
5590 	int ret, num_counters;
5591 	u8 mdev_port_num;
5592 
5593 	if (!stats)
5594 		return -EINVAL;
5595 
5596 	num_counters = cnts->num_q_counters +
5597 		       cnts->num_cong_counters +
5598 		       cnts->num_ext_ppcnt_counters;
5599 
5600 	/* q_counters are per IB device, query the master mdev */
5601 	ret = mlx5_ib_query_q_counters(dev->mdev, cnts, stats, cnts->set_id);
5602 	if (ret)
5603 		return ret;
5604 
5605 	if (MLX5_CAP_PCAM_FEATURE(dev->mdev, rx_icrc_encapsulated_counter)) {
5606 		ret =  mlx5_ib_query_ext_ppcnt_counters(dev, cnts, stats);
5607 		if (ret)
5608 			return ret;
5609 	}
5610 
5611 	if (MLX5_CAP_GEN(dev->mdev, cc_query_allowed)) {
5612 		mdev = mlx5_ib_get_native_port_mdev(dev, port_num,
5613 						    &mdev_port_num);
5614 		if (!mdev) {
5615 			/* If port is not affiliated yet, its in down state
5616 			 * which doesn't have any counters yet, so it would be
5617 			 * zero. So no need to read from the HCA.
5618 			 */
5619 			goto done;
5620 		}
5621 		ret = mlx5_lag_query_cong_counters(dev->mdev,
5622 						   stats->value +
5623 						   cnts->num_q_counters,
5624 						   cnts->num_cong_counters,
5625 						   cnts->offsets +
5626 						   cnts->num_q_counters);
5627 
5628 		mlx5_ib_put_native_port_mdev(dev, port_num);
5629 		if (ret)
5630 			return ret;
5631 	}
5632 
5633 done:
5634 	return num_counters;
5635 }
5636 
5637 static struct rdma_hw_stats *
5638 mlx5_ib_counter_alloc_stats(struct rdma_counter *counter)
5639 {
5640 	struct mlx5_ib_dev *dev = to_mdev(counter->device);
5641 	const struct mlx5_ib_counters *cnts =
5642 		get_counters(dev, counter->port - 1);
5643 
5644 	/* Q counters are in the beginning of all counters */
5645 	return rdma_alloc_hw_stats_struct(cnts->names,
5646 					  cnts->num_q_counters,
5647 					  RDMA_HW_STATS_DEFAULT_LIFESPAN);
5648 }
5649 
5650 static int mlx5_ib_counter_update_stats(struct rdma_counter *counter)
5651 {
5652 	struct mlx5_ib_dev *dev = to_mdev(counter->device);
5653 	const struct mlx5_ib_counters *cnts =
5654 		get_counters(dev, counter->port - 1);
5655 
5656 	return mlx5_ib_query_q_counters(dev->mdev, cnts,
5657 					counter->stats, counter->id);
5658 }
5659 
5660 static int mlx5_ib_counter_bind_qp(struct rdma_counter *counter,
5661 				   struct ib_qp *qp)
5662 {
5663 	struct mlx5_ib_dev *dev = to_mdev(qp->device);
5664 	u16 cnt_set_id = 0;
5665 	int err;
5666 
5667 	if (!counter->id) {
5668 		err = mlx5_cmd_alloc_q_counter(dev->mdev,
5669 					       &cnt_set_id,
5670 					       MLX5_SHARED_RESOURCE_UID);
5671 		if (err)
5672 			return err;
5673 		counter->id = cnt_set_id;
5674 	}
5675 
5676 	err = mlx5_ib_qp_set_counter(qp, counter);
5677 	if (err)
5678 		goto fail_set_counter;
5679 
5680 	return 0;
5681 
5682 fail_set_counter:
5683 	mlx5_core_dealloc_q_counter(dev->mdev, cnt_set_id);
5684 	counter->id = 0;
5685 
5686 	return err;
5687 }
5688 
5689 static int mlx5_ib_counter_unbind_qp(struct ib_qp *qp)
5690 {
5691 	return mlx5_ib_qp_set_counter(qp, NULL);
5692 }
5693 
5694 static int mlx5_ib_counter_dealloc(struct rdma_counter *counter)
5695 {
5696 	struct mlx5_ib_dev *dev = to_mdev(counter->device);
5697 
5698 	return mlx5_core_dealloc_q_counter(dev->mdev, counter->id);
5699 }
5700 
5701 static int mlx5_ib_rn_get_params(struct ib_device *device, u8 port_num,
5702 				 enum rdma_netdev_t type,
5703 				 struct rdma_netdev_alloc_params *params)
5704 {
5705 	if (type != RDMA_NETDEV_IPOIB)
5706 		return -EOPNOTSUPP;
5707 
5708 	return mlx5_rdma_rn_get_params(to_mdev(device)->mdev, device, params);
5709 }
5710 
5711 static void delay_drop_debugfs_cleanup(struct mlx5_ib_dev *dev)
5712 {
5713 	if (!dev->delay_drop.dbg)
5714 		return;
5715 	debugfs_remove_recursive(dev->delay_drop.dbg->dir_debugfs);
5716 	kfree(dev->delay_drop.dbg);
5717 	dev->delay_drop.dbg = NULL;
5718 }
5719 
5720 static void cancel_delay_drop(struct mlx5_ib_dev *dev)
5721 {
5722 	if (!(dev->ib_dev.attrs.raw_packet_caps & IB_RAW_PACKET_CAP_DELAY_DROP))
5723 		return;
5724 
5725 	cancel_work_sync(&dev->delay_drop.delay_drop_work);
5726 	delay_drop_debugfs_cleanup(dev);
5727 }
5728 
5729 static ssize_t delay_drop_timeout_read(struct file *filp, char __user *buf,
5730 				       size_t count, loff_t *pos)
5731 {
5732 	struct mlx5_ib_delay_drop *delay_drop = filp->private_data;
5733 	char lbuf[20];
5734 	int len;
5735 
5736 	len = snprintf(lbuf, sizeof(lbuf), "%u\n", delay_drop->timeout);
5737 	return simple_read_from_buffer(buf, count, pos, lbuf, len);
5738 }
5739 
5740 static ssize_t delay_drop_timeout_write(struct file *filp, const char __user *buf,
5741 					size_t count, loff_t *pos)
5742 {
5743 	struct mlx5_ib_delay_drop *delay_drop = filp->private_data;
5744 	u32 timeout;
5745 	u32 var;
5746 
5747 	if (kstrtouint_from_user(buf, count, 0, &var))
5748 		return -EFAULT;
5749 
5750 	timeout = min_t(u32, roundup(var, 100), MLX5_MAX_DELAY_DROP_TIMEOUT_MS *
5751 			1000);
5752 	if (timeout != var)
5753 		mlx5_ib_dbg(delay_drop->dev, "Round delay drop timeout to %u usec\n",
5754 			    timeout);
5755 
5756 	delay_drop->timeout = timeout;
5757 
5758 	return count;
5759 }
5760 
5761 static const struct file_operations fops_delay_drop_timeout = {
5762 	.owner	= THIS_MODULE,
5763 	.open	= simple_open,
5764 	.write	= delay_drop_timeout_write,
5765 	.read	= delay_drop_timeout_read,
5766 };
5767 
5768 static int delay_drop_debugfs_init(struct mlx5_ib_dev *dev)
5769 {
5770 	struct mlx5_ib_dbg_delay_drop *dbg;
5771 
5772 	if (!mlx5_debugfs_root)
5773 		return 0;
5774 
5775 	dbg = kzalloc(sizeof(*dbg), GFP_KERNEL);
5776 	if (!dbg)
5777 		return -ENOMEM;
5778 
5779 	dev->delay_drop.dbg = dbg;
5780 
5781 	dbg->dir_debugfs =
5782 		debugfs_create_dir("delay_drop",
5783 				   dev->mdev->priv.dbg_root);
5784 	if (!dbg->dir_debugfs)
5785 		goto out_debugfs;
5786 
5787 	dbg->events_cnt_debugfs =
5788 		debugfs_create_atomic_t("num_timeout_events", 0400,
5789 					dbg->dir_debugfs,
5790 					&dev->delay_drop.events_cnt);
5791 	if (!dbg->events_cnt_debugfs)
5792 		goto out_debugfs;
5793 
5794 	dbg->rqs_cnt_debugfs =
5795 		debugfs_create_atomic_t("num_rqs", 0400,
5796 					dbg->dir_debugfs,
5797 					&dev->delay_drop.rqs_cnt);
5798 	if (!dbg->rqs_cnt_debugfs)
5799 		goto out_debugfs;
5800 
5801 	dbg->timeout_debugfs =
5802 		debugfs_create_file("timeout", 0600,
5803 				    dbg->dir_debugfs,
5804 				    &dev->delay_drop,
5805 				    &fops_delay_drop_timeout);
5806 	if (!dbg->timeout_debugfs)
5807 		goto out_debugfs;
5808 
5809 	return 0;
5810 
5811 out_debugfs:
5812 	delay_drop_debugfs_cleanup(dev);
5813 	return -ENOMEM;
5814 }
5815 
5816 static void init_delay_drop(struct mlx5_ib_dev *dev)
5817 {
5818 	if (!(dev->ib_dev.attrs.raw_packet_caps & IB_RAW_PACKET_CAP_DELAY_DROP))
5819 		return;
5820 
5821 	mutex_init(&dev->delay_drop.lock);
5822 	dev->delay_drop.dev = dev;
5823 	dev->delay_drop.activate = false;
5824 	dev->delay_drop.timeout = MLX5_MAX_DELAY_DROP_TIMEOUT_MS * 1000;
5825 	INIT_WORK(&dev->delay_drop.delay_drop_work, delay_drop_handler);
5826 	atomic_set(&dev->delay_drop.rqs_cnt, 0);
5827 	atomic_set(&dev->delay_drop.events_cnt, 0);
5828 
5829 	if (delay_drop_debugfs_init(dev))
5830 		mlx5_ib_warn(dev, "Failed to init delay drop debugfs\n");
5831 }
5832 
5833 static void mlx5_ib_unbind_slave_port(struct mlx5_ib_dev *ibdev,
5834 				      struct mlx5_ib_multiport_info *mpi)
5835 {
5836 	u8 port_num = mlx5_core_native_port_num(mpi->mdev) - 1;
5837 	struct mlx5_ib_port *port = &ibdev->port[port_num];
5838 	int comps;
5839 	int err;
5840 	int i;
5841 
5842 	lockdep_assert_held(&mlx5_ib_multiport_mutex);
5843 
5844 	mlx5_ib_cleanup_cong_debugfs(ibdev, port_num);
5845 
5846 	spin_lock(&port->mp.mpi_lock);
5847 	if (!mpi->ibdev) {
5848 		spin_unlock(&port->mp.mpi_lock);
5849 		return;
5850 	}
5851 
5852 	mpi->ibdev = NULL;
5853 
5854 	spin_unlock(&port->mp.mpi_lock);
5855 	if (mpi->mdev_events.notifier_call)
5856 		mlx5_notifier_unregister(mpi->mdev, &mpi->mdev_events);
5857 	mpi->mdev_events.notifier_call = NULL;
5858 	mlx5_remove_netdev_notifier(ibdev, port_num);
5859 	spin_lock(&port->mp.mpi_lock);
5860 
5861 	comps = mpi->mdev_refcnt;
5862 	if (comps) {
5863 		mpi->unaffiliate = true;
5864 		init_completion(&mpi->unref_comp);
5865 		spin_unlock(&port->mp.mpi_lock);
5866 
5867 		for (i = 0; i < comps; i++)
5868 			wait_for_completion(&mpi->unref_comp);
5869 
5870 		spin_lock(&port->mp.mpi_lock);
5871 		mpi->unaffiliate = false;
5872 	}
5873 
5874 	port->mp.mpi = NULL;
5875 
5876 	list_add_tail(&mpi->list, &mlx5_ib_unaffiliated_port_list);
5877 
5878 	spin_unlock(&port->mp.mpi_lock);
5879 
5880 	err = mlx5_nic_vport_unaffiliate_multiport(mpi->mdev);
5881 
5882 	mlx5_ib_dbg(ibdev, "unaffiliated port %d\n", port_num + 1);
5883 	/* Log an error, still needed to cleanup the pointers and add
5884 	 * it back to the list.
5885 	 */
5886 	if (err)
5887 		mlx5_ib_err(ibdev, "Failed to unaffiliate port %u\n",
5888 			    port_num + 1);
5889 
5890 	ibdev->port[port_num].roce.last_port_state = IB_PORT_DOWN;
5891 }
5892 
5893 static bool mlx5_ib_bind_slave_port(struct mlx5_ib_dev *ibdev,
5894 				    struct mlx5_ib_multiport_info *mpi)
5895 {
5896 	u8 port_num = mlx5_core_native_port_num(mpi->mdev) - 1;
5897 	int err;
5898 
5899 	lockdep_assert_held(&mlx5_ib_multiport_mutex);
5900 
5901 	spin_lock(&ibdev->port[port_num].mp.mpi_lock);
5902 	if (ibdev->port[port_num].mp.mpi) {
5903 		mlx5_ib_dbg(ibdev, "port %d already affiliated.\n",
5904 			    port_num + 1);
5905 		spin_unlock(&ibdev->port[port_num].mp.mpi_lock);
5906 		return false;
5907 	}
5908 
5909 	ibdev->port[port_num].mp.mpi = mpi;
5910 	mpi->ibdev = ibdev;
5911 	mpi->mdev_events.notifier_call = NULL;
5912 	spin_unlock(&ibdev->port[port_num].mp.mpi_lock);
5913 
5914 	err = mlx5_nic_vport_affiliate_multiport(ibdev->mdev, mpi->mdev);
5915 	if (err)
5916 		goto unbind;
5917 
5918 	err = get_port_caps(ibdev, mlx5_core_native_port_num(mpi->mdev));
5919 	if (err)
5920 		goto unbind;
5921 
5922 	err = mlx5_add_netdev_notifier(ibdev, port_num);
5923 	if (err) {
5924 		mlx5_ib_err(ibdev, "failed adding netdev notifier for port %u\n",
5925 			    port_num + 1);
5926 		goto unbind;
5927 	}
5928 
5929 	mpi->mdev_events.notifier_call = mlx5_ib_event_slave_port;
5930 	mlx5_notifier_register(mpi->mdev, &mpi->mdev_events);
5931 
5932 	mlx5_ib_init_cong_debugfs(ibdev, port_num);
5933 
5934 	return true;
5935 
5936 unbind:
5937 	mlx5_ib_unbind_slave_port(ibdev, mpi);
5938 	return false;
5939 }
5940 
5941 static int mlx5_ib_init_multiport_master(struct mlx5_ib_dev *dev)
5942 {
5943 	int port_num = mlx5_core_native_port_num(dev->mdev) - 1;
5944 	enum rdma_link_layer ll = mlx5_ib_port_link_layer(&dev->ib_dev,
5945 							  port_num + 1);
5946 	struct mlx5_ib_multiport_info *mpi;
5947 	int err;
5948 	int i;
5949 
5950 	if (!mlx5_core_is_mp_master(dev->mdev) || ll != IB_LINK_LAYER_ETHERNET)
5951 		return 0;
5952 
5953 	err = mlx5_query_nic_vport_system_image_guid(dev->mdev,
5954 						     &dev->sys_image_guid);
5955 	if (err)
5956 		return err;
5957 
5958 	err = mlx5_nic_vport_enable_roce(dev->mdev);
5959 	if (err)
5960 		return err;
5961 
5962 	mutex_lock(&mlx5_ib_multiport_mutex);
5963 	for (i = 0; i < dev->num_ports; i++) {
5964 		bool bound = false;
5965 
5966 		/* build a stub multiport info struct for the native port. */
5967 		if (i == port_num) {
5968 			mpi = kzalloc(sizeof(*mpi), GFP_KERNEL);
5969 			if (!mpi) {
5970 				mutex_unlock(&mlx5_ib_multiport_mutex);
5971 				mlx5_nic_vport_disable_roce(dev->mdev);
5972 				return -ENOMEM;
5973 			}
5974 
5975 			mpi->is_master = true;
5976 			mpi->mdev = dev->mdev;
5977 			mpi->sys_image_guid = dev->sys_image_guid;
5978 			dev->port[i].mp.mpi = mpi;
5979 			mpi->ibdev = dev;
5980 			mpi = NULL;
5981 			continue;
5982 		}
5983 
5984 		list_for_each_entry(mpi, &mlx5_ib_unaffiliated_port_list,
5985 				    list) {
5986 			if (dev->sys_image_guid == mpi->sys_image_guid &&
5987 			    (mlx5_core_native_port_num(mpi->mdev) - 1) == i) {
5988 				bound = mlx5_ib_bind_slave_port(dev, mpi);
5989 			}
5990 
5991 			if (bound) {
5992 				dev_dbg(mpi->mdev->device,
5993 					"removing port from unaffiliated list.\n");
5994 				mlx5_ib_dbg(dev, "port %d bound\n", i + 1);
5995 				list_del(&mpi->list);
5996 				break;
5997 			}
5998 		}
5999 		if (!bound) {
6000 			get_port_caps(dev, i + 1);
6001 			mlx5_ib_dbg(dev, "no free port found for port %d\n",
6002 				    i + 1);
6003 		}
6004 	}
6005 
6006 	list_add_tail(&dev->ib_dev_list, &mlx5_ib_dev_list);
6007 	mutex_unlock(&mlx5_ib_multiport_mutex);
6008 	return err;
6009 }
6010 
6011 static void mlx5_ib_cleanup_multiport_master(struct mlx5_ib_dev *dev)
6012 {
6013 	int port_num = mlx5_core_native_port_num(dev->mdev) - 1;
6014 	enum rdma_link_layer ll = mlx5_ib_port_link_layer(&dev->ib_dev,
6015 							  port_num + 1);
6016 	int i;
6017 
6018 	if (!mlx5_core_is_mp_master(dev->mdev) || ll != IB_LINK_LAYER_ETHERNET)
6019 		return;
6020 
6021 	mutex_lock(&mlx5_ib_multiport_mutex);
6022 	for (i = 0; i < dev->num_ports; i++) {
6023 		if (dev->port[i].mp.mpi) {
6024 			/* Destroy the native port stub */
6025 			if (i == port_num) {
6026 				kfree(dev->port[i].mp.mpi);
6027 				dev->port[i].mp.mpi = NULL;
6028 			} else {
6029 				mlx5_ib_dbg(dev, "unbinding port_num: %d\n", i + 1);
6030 				mlx5_ib_unbind_slave_port(dev, dev->port[i].mp.mpi);
6031 			}
6032 		}
6033 	}
6034 
6035 	mlx5_ib_dbg(dev, "removing from devlist\n");
6036 	list_del(&dev->ib_dev_list);
6037 	mutex_unlock(&mlx5_ib_multiport_mutex);
6038 
6039 	mlx5_nic_vport_disable_roce(dev->mdev);
6040 }
6041 
6042 ADD_UVERBS_ATTRIBUTES_SIMPLE(
6043 	mlx5_ib_dm,
6044 	UVERBS_OBJECT_DM,
6045 	UVERBS_METHOD_DM_ALLOC,
6046 	UVERBS_ATTR_PTR_OUT(MLX5_IB_ATTR_ALLOC_DM_RESP_START_OFFSET,
6047 			    UVERBS_ATTR_TYPE(u64),
6048 			    UA_MANDATORY),
6049 	UVERBS_ATTR_PTR_OUT(MLX5_IB_ATTR_ALLOC_DM_RESP_PAGE_INDEX,
6050 			    UVERBS_ATTR_TYPE(u16),
6051 			    UA_OPTIONAL),
6052 	UVERBS_ATTR_CONST_IN(MLX5_IB_ATTR_ALLOC_DM_REQ_TYPE,
6053 			     enum mlx5_ib_uapi_dm_type,
6054 			     UA_OPTIONAL));
6055 
6056 ADD_UVERBS_ATTRIBUTES_SIMPLE(
6057 	mlx5_ib_flow_action,
6058 	UVERBS_OBJECT_FLOW_ACTION,
6059 	UVERBS_METHOD_FLOW_ACTION_ESP_CREATE,
6060 	UVERBS_ATTR_FLAGS_IN(MLX5_IB_ATTR_CREATE_FLOW_ACTION_FLAGS,
6061 			     enum mlx5_ib_uapi_flow_action_flags));
6062 
6063 static const struct uapi_definition mlx5_ib_defs[] = {
6064 #if IS_ENABLED(CONFIG_INFINIBAND_USER_ACCESS)
6065 	UAPI_DEF_CHAIN(mlx5_ib_devx_defs),
6066 	UAPI_DEF_CHAIN(mlx5_ib_flow_defs),
6067 #endif
6068 
6069 	UAPI_DEF_CHAIN_OBJ_TREE(UVERBS_OBJECT_FLOW_ACTION,
6070 				&mlx5_ib_flow_action),
6071 	UAPI_DEF_CHAIN_OBJ_TREE(UVERBS_OBJECT_DM, &mlx5_ib_dm),
6072 	{}
6073 };
6074 
6075 static int mlx5_ib_read_counters(struct ib_counters *counters,
6076 				 struct ib_counters_read_attr *read_attr,
6077 				 struct uverbs_attr_bundle *attrs)
6078 {
6079 	struct mlx5_ib_mcounters *mcounters = to_mcounters(counters);
6080 	struct mlx5_read_counters_attr mread_attr = {};
6081 	struct mlx5_ib_flow_counters_desc *desc;
6082 	int ret, i;
6083 
6084 	mutex_lock(&mcounters->mcntrs_mutex);
6085 	if (mcounters->cntrs_max_index > read_attr->ncounters) {
6086 		ret = -EINVAL;
6087 		goto err_bound;
6088 	}
6089 
6090 	mread_attr.out = kcalloc(mcounters->counters_num, sizeof(u64),
6091 				 GFP_KERNEL);
6092 	if (!mread_attr.out) {
6093 		ret = -ENOMEM;
6094 		goto err_bound;
6095 	}
6096 
6097 	mread_attr.hw_cntrs_hndl = mcounters->hw_cntrs_hndl;
6098 	mread_attr.flags = read_attr->flags;
6099 	ret = mcounters->read_counters(counters->device, &mread_attr);
6100 	if (ret)
6101 		goto err_read;
6102 
6103 	/* do the pass over the counters data array to assign according to the
6104 	 * descriptions and indexing pairs
6105 	 */
6106 	desc = mcounters->counters_data;
6107 	for (i = 0; i < mcounters->ncounters; i++)
6108 		read_attr->counters_buff[desc[i].index] += mread_attr.out[desc[i].description];
6109 
6110 err_read:
6111 	kfree(mread_attr.out);
6112 err_bound:
6113 	mutex_unlock(&mcounters->mcntrs_mutex);
6114 	return ret;
6115 }
6116 
6117 static int mlx5_ib_destroy_counters(struct ib_counters *counters)
6118 {
6119 	struct mlx5_ib_mcounters *mcounters = to_mcounters(counters);
6120 
6121 	counters_clear_description(counters);
6122 	if (mcounters->hw_cntrs_hndl)
6123 		mlx5_fc_destroy(to_mdev(counters->device)->mdev,
6124 				mcounters->hw_cntrs_hndl);
6125 
6126 	kfree(mcounters);
6127 
6128 	return 0;
6129 }
6130 
6131 static struct ib_counters *mlx5_ib_create_counters(struct ib_device *device,
6132 						   struct uverbs_attr_bundle *attrs)
6133 {
6134 	struct mlx5_ib_mcounters *mcounters;
6135 
6136 	mcounters = kzalloc(sizeof(*mcounters), GFP_KERNEL);
6137 	if (!mcounters)
6138 		return ERR_PTR(-ENOMEM);
6139 
6140 	mutex_init(&mcounters->mcntrs_mutex);
6141 
6142 	return &mcounters->ibcntrs;
6143 }
6144 
6145 static void mlx5_ib_stage_init_cleanup(struct mlx5_ib_dev *dev)
6146 {
6147 	mlx5_ib_cleanup_multiport_master(dev);
6148 	if (IS_ENABLED(CONFIG_INFINIBAND_ON_DEMAND_PAGING)) {
6149 		srcu_barrier(&dev->mr_srcu);
6150 		cleanup_srcu_struct(&dev->mr_srcu);
6151 	}
6152 
6153 	WARN_ON(!bitmap_empty(dev->dm.memic_alloc_pages, MLX5_MAX_MEMIC_PAGES));
6154 }
6155 
6156 static int mlx5_ib_stage_init_init(struct mlx5_ib_dev *dev)
6157 {
6158 	struct mlx5_core_dev *mdev = dev->mdev;
6159 	int err;
6160 	int i;
6161 
6162 	for (i = 0; i < dev->num_ports; i++) {
6163 		spin_lock_init(&dev->port[i].mp.mpi_lock);
6164 		rwlock_init(&dev->port[i].roce.netdev_lock);
6165 		dev->port[i].roce.dev = dev;
6166 		dev->port[i].roce.native_port_num = i + 1;
6167 		dev->port[i].roce.last_port_state = IB_PORT_DOWN;
6168 	}
6169 
6170 	mlx5_ib_internal_fill_odp_caps(dev);
6171 
6172 	err = mlx5_ib_init_multiport_master(dev);
6173 	if (err)
6174 		return err;
6175 
6176 	err = set_has_smi_cap(dev);
6177 	if (err)
6178 		return err;
6179 
6180 	if (!mlx5_core_mp_enabled(mdev)) {
6181 		for (i = 1; i <= dev->num_ports; i++) {
6182 			err = get_port_caps(dev, i);
6183 			if (err)
6184 				break;
6185 		}
6186 	} else {
6187 		err = get_port_caps(dev, mlx5_core_native_port_num(mdev));
6188 	}
6189 	if (err)
6190 		goto err_mp;
6191 
6192 	if (mlx5_use_mad_ifc(dev))
6193 		get_ext_port_caps(dev);
6194 
6195 	dev->ib_dev.node_type		= RDMA_NODE_IB_CA;
6196 	dev->ib_dev.local_dma_lkey	= 0 /* not supported for now */;
6197 	dev->ib_dev.phys_port_cnt	= dev->num_ports;
6198 	dev->ib_dev.num_comp_vectors    = mlx5_comp_vectors_count(mdev);
6199 	dev->ib_dev.dev.parent		= mdev->device;
6200 
6201 	mutex_init(&dev->cap_mask_mutex);
6202 	INIT_LIST_HEAD(&dev->qp_list);
6203 	spin_lock_init(&dev->reset_flow_resource_lock);
6204 
6205 	spin_lock_init(&dev->dm.lock);
6206 	dev->dm.dev = mdev;
6207 
6208 	if (IS_ENABLED(CONFIG_INFINIBAND_ON_DEMAND_PAGING)) {
6209 		err = init_srcu_struct(&dev->mr_srcu);
6210 		if (err)
6211 			goto err_mp;
6212 	}
6213 
6214 	return 0;
6215 
6216 err_mp:
6217 	mlx5_ib_cleanup_multiport_master(dev);
6218 
6219 	return -ENOMEM;
6220 }
6221 
6222 static int mlx5_ib_stage_flow_db_init(struct mlx5_ib_dev *dev)
6223 {
6224 	dev->flow_db = kzalloc(sizeof(*dev->flow_db), GFP_KERNEL);
6225 
6226 	if (!dev->flow_db)
6227 		return -ENOMEM;
6228 
6229 	mutex_init(&dev->flow_db->lock);
6230 
6231 	return 0;
6232 }
6233 
6234 static void mlx5_ib_stage_flow_db_cleanup(struct mlx5_ib_dev *dev)
6235 {
6236 	kfree(dev->flow_db);
6237 }
6238 
6239 static const struct ib_device_ops mlx5_ib_dev_ops = {
6240 	.owner = THIS_MODULE,
6241 	.driver_id = RDMA_DRIVER_MLX5,
6242 	.uverbs_abi_ver	= MLX5_IB_UVERBS_ABI_VERSION,
6243 
6244 	.add_gid = mlx5_ib_add_gid,
6245 	.alloc_mr = mlx5_ib_alloc_mr,
6246 	.alloc_mr_integrity = mlx5_ib_alloc_mr_integrity,
6247 	.alloc_pd = mlx5_ib_alloc_pd,
6248 	.alloc_ucontext = mlx5_ib_alloc_ucontext,
6249 	.attach_mcast = mlx5_ib_mcg_attach,
6250 	.check_mr_status = mlx5_ib_check_mr_status,
6251 	.create_ah = mlx5_ib_create_ah,
6252 	.create_counters = mlx5_ib_create_counters,
6253 	.create_cq = mlx5_ib_create_cq,
6254 	.create_flow = mlx5_ib_create_flow,
6255 	.create_qp = mlx5_ib_create_qp,
6256 	.create_srq = mlx5_ib_create_srq,
6257 	.dealloc_pd = mlx5_ib_dealloc_pd,
6258 	.dealloc_ucontext = mlx5_ib_dealloc_ucontext,
6259 	.del_gid = mlx5_ib_del_gid,
6260 	.dereg_mr = mlx5_ib_dereg_mr,
6261 	.destroy_ah = mlx5_ib_destroy_ah,
6262 	.destroy_counters = mlx5_ib_destroy_counters,
6263 	.destroy_cq = mlx5_ib_destroy_cq,
6264 	.destroy_flow = mlx5_ib_destroy_flow,
6265 	.destroy_flow_action = mlx5_ib_destroy_flow_action,
6266 	.destroy_qp = mlx5_ib_destroy_qp,
6267 	.destroy_srq = mlx5_ib_destroy_srq,
6268 	.detach_mcast = mlx5_ib_mcg_detach,
6269 	.disassociate_ucontext = mlx5_ib_disassociate_ucontext,
6270 	.drain_rq = mlx5_ib_drain_rq,
6271 	.drain_sq = mlx5_ib_drain_sq,
6272 	.get_dev_fw_str = get_dev_fw_str,
6273 	.get_dma_mr = mlx5_ib_get_dma_mr,
6274 	.get_link_layer = mlx5_ib_port_link_layer,
6275 	.map_mr_sg = mlx5_ib_map_mr_sg,
6276 	.map_mr_sg_pi = mlx5_ib_map_mr_sg_pi,
6277 	.mmap = mlx5_ib_mmap,
6278 	.modify_cq = mlx5_ib_modify_cq,
6279 	.modify_device = mlx5_ib_modify_device,
6280 	.modify_port = mlx5_ib_modify_port,
6281 	.modify_qp = mlx5_ib_modify_qp,
6282 	.modify_srq = mlx5_ib_modify_srq,
6283 	.poll_cq = mlx5_ib_poll_cq,
6284 	.post_recv = mlx5_ib_post_recv,
6285 	.post_send = mlx5_ib_post_send,
6286 	.post_srq_recv = mlx5_ib_post_srq_recv,
6287 	.process_mad = mlx5_ib_process_mad,
6288 	.query_ah = mlx5_ib_query_ah,
6289 	.query_device = mlx5_ib_query_device,
6290 	.query_gid = mlx5_ib_query_gid,
6291 	.query_pkey = mlx5_ib_query_pkey,
6292 	.query_qp = mlx5_ib_query_qp,
6293 	.query_srq = mlx5_ib_query_srq,
6294 	.read_counters = mlx5_ib_read_counters,
6295 	.reg_user_mr = mlx5_ib_reg_user_mr,
6296 	.req_notify_cq = mlx5_ib_arm_cq,
6297 	.rereg_user_mr = mlx5_ib_rereg_user_mr,
6298 	.resize_cq = mlx5_ib_resize_cq,
6299 
6300 	INIT_RDMA_OBJ_SIZE(ib_ah, mlx5_ib_ah, ibah),
6301 	INIT_RDMA_OBJ_SIZE(ib_cq, mlx5_ib_cq, ibcq),
6302 	INIT_RDMA_OBJ_SIZE(ib_pd, mlx5_ib_pd, ibpd),
6303 	INIT_RDMA_OBJ_SIZE(ib_srq, mlx5_ib_srq, ibsrq),
6304 	INIT_RDMA_OBJ_SIZE(ib_ucontext, mlx5_ib_ucontext, ibucontext),
6305 };
6306 
6307 static const struct ib_device_ops mlx5_ib_dev_flow_ipsec_ops = {
6308 	.create_flow_action_esp = mlx5_ib_create_flow_action_esp,
6309 	.modify_flow_action_esp = mlx5_ib_modify_flow_action_esp,
6310 };
6311 
6312 static const struct ib_device_ops mlx5_ib_dev_ipoib_enhanced_ops = {
6313 	.rdma_netdev_get_params = mlx5_ib_rn_get_params,
6314 };
6315 
6316 static const struct ib_device_ops mlx5_ib_dev_sriov_ops = {
6317 	.get_vf_config = mlx5_ib_get_vf_config,
6318 	.get_vf_stats = mlx5_ib_get_vf_stats,
6319 	.set_vf_guid = mlx5_ib_set_vf_guid,
6320 	.set_vf_link_state = mlx5_ib_set_vf_link_state,
6321 };
6322 
6323 static const struct ib_device_ops mlx5_ib_dev_mw_ops = {
6324 	.alloc_mw = mlx5_ib_alloc_mw,
6325 	.dealloc_mw = mlx5_ib_dealloc_mw,
6326 };
6327 
6328 static const struct ib_device_ops mlx5_ib_dev_xrc_ops = {
6329 	.alloc_xrcd = mlx5_ib_alloc_xrcd,
6330 	.dealloc_xrcd = mlx5_ib_dealloc_xrcd,
6331 };
6332 
6333 static const struct ib_device_ops mlx5_ib_dev_dm_ops = {
6334 	.alloc_dm = mlx5_ib_alloc_dm,
6335 	.dealloc_dm = mlx5_ib_dealloc_dm,
6336 	.reg_dm_mr = mlx5_ib_reg_dm_mr,
6337 };
6338 
6339 static int mlx5_ib_stage_caps_init(struct mlx5_ib_dev *dev)
6340 {
6341 	struct mlx5_core_dev *mdev = dev->mdev;
6342 	int err;
6343 
6344 	dev->ib_dev.uverbs_cmd_mask	=
6345 		(1ull << IB_USER_VERBS_CMD_GET_CONTEXT)		|
6346 		(1ull << IB_USER_VERBS_CMD_QUERY_DEVICE)	|
6347 		(1ull << IB_USER_VERBS_CMD_QUERY_PORT)		|
6348 		(1ull << IB_USER_VERBS_CMD_ALLOC_PD)		|
6349 		(1ull << IB_USER_VERBS_CMD_DEALLOC_PD)		|
6350 		(1ull << IB_USER_VERBS_CMD_CREATE_AH)		|
6351 		(1ull << IB_USER_VERBS_CMD_DESTROY_AH)		|
6352 		(1ull << IB_USER_VERBS_CMD_REG_MR)		|
6353 		(1ull << IB_USER_VERBS_CMD_REREG_MR)		|
6354 		(1ull << IB_USER_VERBS_CMD_DEREG_MR)		|
6355 		(1ull << IB_USER_VERBS_CMD_CREATE_COMP_CHANNEL)	|
6356 		(1ull << IB_USER_VERBS_CMD_CREATE_CQ)		|
6357 		(1ull << IB_USER_VERBS_CMD_RESIZE_CQ)		|
6358 		(1ull << IB_USER_VERBS_CMD_DESTROY_CQ)		|
6359 		(1ull << IB_USER_VERBS_CMD_CREATE_QP)		|
6360 		(1ull << IB_USER_VERBS_CMD_MODIFY_QP)		|
6361 		(1ull << IB_USER_VERBS_CMD_QUERY_QP)		|
6362 		(1ull << IB_USER_VERBS_CMD_DESTROY_QP)		|
6363 		(1ull << IB_USER_VERBS_CMD_ATTACH_MCAST)	|
6364 		(1ull << IB_USER_VERBS_CMD_DETACH_MCAST)	|
6365 		(1ull << IB_USER_VERBS_CMD_CREATE_SRQ)		|
6366 		(1ull << IB_USER_VERBS_CMD_MODIFY_SRQ)		|
6367 		(1ull << IB_USER_VERBS_CMD_QUERY_SRQ)		|
6368 		(1ull << IB_USER_VERBS_CMD_DESTROY_SRQ)		|
6369 		(1ull << IB_USER_VERBS_CMD_CREATE_XSRQ)		|
6370 		(1ull << IB_USER_VERBS_CMD_OPEN_QP);
6371 	dev->ib_dev.uverbs_ex_cmd_mask =
6372 		(1ull << IB_USER_VERBS_EX_CMD_QUERY_DEVICE)	|
6373 		(1ull << IB_USER_VERBS_EX_CMD_CREATE_CQ)	|
6374 		(1ull << IB_USER_VERBS_EX_CMD_CREATE_QP)	|
6375 		(1ull << IB_USER_VERBS_EX_CMD_MODIFY_QP)	|
6376 		(1ull << IB_USER_VERBS_EX_CMD_MODIFY_CQ)	|
6377 		(1ull << IB_USER_VERBS_EX_CMD_CREATE_FLOW)	|
6378 		(1ull << IB_USER_VERBS_EX_CMD_DESTROY_FLOW);
6379 
6380 	if (MLX5_CAP_GEN(mdev, ipoib_enhanced_offloads) &&
6381 	    IS_ENABLED(CONFIG_MLX5_CORE_IPOIB))
6382 		ib_set_device_ops(&dev->ib_dev,
6383 				  &mlx5_ib_dev_ipoib_enhanced_ops);
6384 
6385 	if (mlx5_core_is_pf(mdev))
6386 		ib_set_device_ops(&dev->ib_dev, &mlx5_ib_dev_sriov_ops);
6387 
6388 	dev->umr_fence = mlx5_get_umr_fence(MLX5_CAP_GEN(mdev, umr_fence));
6389 
6390 	if (MLX5_CAP_GEN(mdev, imaicl)) {
6391 		dev->ib_dev.uverbs_cmd_mask |=
6392 			(1ull << IB_USER_VERBS_CMD_ALLOC_MW)	|
6393 			(1ull << IB_USER_VERBS_CMD_DEALLOC_MW);
6394 		ib_set_device_ops(&dev->ib_dev, &mlx5_ib_dev_mw_ops);
6395 	}
6396 
6397 	if (MLX5_CAP_GEN(mdev, xrc)) {
6398 		dev->ib_dev.uverbs_cmd_mask |=
6399 			(1ull << IB_USER_VERBS_CMD_OPEN_XRCD) |
6400 			(1ull << IB_USER_VERBS_CMD_CLOSE_XRCD);
6401 		ib_set_device_ops(&dev->ib_dev, &mlx5_ib_dev_xrc_ops);
6402 	}
6403 
6404 	if (MLX5_CAP_DEV_MEM(mdev, memic) ||
6405 	    MLX5_CAP_GEN_64(dev->mdev, general_obj_types) &
6406 	    MLX5_GENERAL_OBJ_TYPES_CAP_SW_ICM)
6407 		ib_set_device_ops(&dev->ib_dev, &mlx5_ib_dev_dm_ops);
6408 
6409 	if (mlx5_accel_ipsec_device_caps(dev->mdev) &
6410 	    MLX5_ACCEL_IPSEC_CAP_DEVICE)
6411 		ib_set_device_ops(&dev->ib_dev, &mlx5_ib_dev_flow_ipsec_ops);
6412 	ib_set_device_ops(&dev->ib_dev, &mlx5_ib_dev_ops);
6413 
6414 	if (IS_ENABLED(CONFIG_INFINIBAND_USER_ACCESS))
6415 		dev->ib_dev.driver_def = mlx5_ib_defs;
6416 
6417 	err = init_node_data(dev);
6418 	if (err)
6419 		return err;
6420 
6421 	if ((MLX5_CAP_GEN(dev->mdev, port_type) == MLX5_CAP_PORT_TYPE_ETH) &&
6422 	    (MLX5_CAP_GEN(dev->mdev, disable_local_lb_uc) ||
6423 	     MLX5_CAP_GEN(dev->mdev, disable_local_lb_mc)))
6424 		mutex_init(&dev->lb.mutex);
6425 
6426 	dev->ib_dev.use_cq_dim = true;
6427 
6428 	return 0;
6429 }
6430 
6431 static const struct ib_device_ops mlx5_ib_dev_port_ops = {
6432 	.get_port_immutable = mlx5_port_immutable,
6433 	.query_port = mlx5_ib_query_port,
6434 };
6435 
6436 static int mlx5_ib_stage_non_default_cb(struct mlx5_ib_dev *dev)
6437 {
6438 	ib_set_device_ops(&dev->ib_dev, &mlx5_ib_dev_port_ops);
6439 	return 0;
6440 }
6441 
6442 static const struct ib_device_ops mlx5_ib_dev_port_rep_ops = {
6443 	.get_port_immutable = mlx5_port_rep_immutable,
6444 	.query_port = mlx5_ib_rep_query_port,
6445 };
6446 
6447 static int mlx5_ib_stage_rep_non_default_cb(struct mlx5_ib_dev *dev)
6448 {
6449 	ib_set_device_ops(&dev->ib_dev, &mlx5_ib_dev_port_rep_ops);
6450 	return 0;
6451 }
6452 
6453 static const struct ib_device_ops mlx5_ib_dev_common_roce_ops = {
6454 	.create_rwq_ind_table = mlx5_ib_create_rwq_ind_table,
6455 	.create_wq = mlx5_ib_create_wq,
6456 	.destroy_rwq_ind_table = mlx5_ib_destroy_rwq_ind_table,
6457 	.destroy_wq = mlx5_ib_destroy_wq,
6458 	.get_netdev = mlx5_ib_get_netdev,
6459 	.modify_wq = mlx5_ib_modify_wq,
6460 };
6461 
6462 static int mlx5_ib_stage_common_roce_init(struct mlx5_ib_dev *dev)
6463 {
6464 	u8 port_num;
6465 
6466 	dev->ib_dev.uverbs_ex_cmd_mask |=
6467 			(1ull << IB_USER_VERBS_EX_CMD_CREATE_WQ) |
6468 			(1ull << IB_USER_VERBS_EX_CMD_MODIFY_WQ) |
6469 			(1ull << IB_USER_VERBS_EX_CMD_DESTROY_WQ) |
6470 			(1ull << IB_USER_VERBS_EX_CMD_CREATE_RWQ_IND_TBL) |
6471 			(1ull << IB_USER_VERBS_EX_CMD_DESTROY_RWQ_IND_TBL);
6472 	ib_set_device_ops(&dev->ib_dev, &mlx5_ib_dev_common_roce_ops);
6473 
6474 	port_num = mlx5_core_native_port_num(dev->mdev) - 1;
6475 
6476 	/* Register only for native ports */
6477 	return mlx5_add_netdev_notifier(dev, port_num);
6478 }
6479 
6480 static void mlx5_ib_stage_common_roce_cleanup(struct mlx5_ib_dev *dev)
6481 {
6482 	u8 port_num = mlx5_core_native_port_num(dev->mdev) - 1;
6483 
6484 	mlx5_remove_netdev_notifier(dev, port_num);
6485 }
6486 
6487 static int mlx5_ib_stage_rep_roce_init(struct mlx5_ib_dev *dev)
6488 {
6489 	struct mlx5_core_dev *mdev = dev->mdev;
6490 	enum rdma_link_layer ll;
6491 	int port_type_cap;
6492 	int err = 0;
6493 
6494 	port_type_cap = MLX5_CAP_GEN(mdev, port_type);
6495 	ll = mlx5_port_type_cap_to_rdma_ll(port_type_cap);
6496 
6497 	if (ll == IB_LINK_LAYER_ETHERNET)
6498 		err = mlx5_ib_stage_common_roce_init(dev);
6499 
6500 	return err;
6501 }
6502 
6503 static void mlx5_ib_stage_rep_roce_cleanup(struct mlx5_ib_dev *dev)
6504 {
6505 	mlx5_ib_stage_common_roce_cleanup(dev);
6506 }
6507 
6508 static int mlx5_ib_stage_roce_init(struct mlx5_ib_dev *dev)
6509 {
6510 	struct mlx5_core_dev *mdev = dev->mdev;
6511 	enum rdma_link_layer ll;
6512 	int port_type_cap;
6513 	int err;
6514 
6515 	port_type_cap = MLX5_CAP_GEN(mdev, port_type);
6516 	ll = mlx5_port_type_cap_to_rdma_ll(port_type_cap);
6517 
6518 	if (ll == IB_LINK_LAYER_ETHERNET) {
6519 		err = mlx5_ib_stage_common_roce_init(dev);
6520 		if (err)
6521 			return err;
6522 
6523 		err = mlx5_enable_eth(dev);
6524 		if (err)
6525 			goto cleanup;
6526 	}
6527 
6528 	return 0;
6529 cleanup:
6530 	mlx5_ib_stage_common_roce_cleanup(dev);
6531 
6532 	return err;
6533 }
6534 
6535 static void mlx5_ib_stage_roce_cleanup(struct mlx5_ib_dev *dev)
6536 {
6537 	struct mlx5_core_dev *mdev = dev->mdev;
6538 	enum rdma_link_layer ll;
6539 	int port_type_cap;
6540 
6541 	port_type_cap = MLX5_CAP_GEN(mdev, port_type);
6542 	ll = mlx5_port_type_cap_to_rdma_ll(port_type_cap);
6543 
6544 	if (ll == IB_LINK_LAYER_ETHERNET) {
6545 		mlx5_disable_eth(dev);
6546 		mlx5_ib_stage_common_roce_cleanup(dev);
6547 	}
6548 }
6549 
6550 static int mlx5_ib_stage_dev_res_init(struct mlx5_ib_dev *dev)
6551 {
6552 	return create_dev_resources(&dev->devr);
6553 }
6554 
6555 static void mlx5_ib_stage_dev_res_cleanup(struct mlx5_ib_dev *dev)
6556 {
6557 	destroy_dev_resources(&dev->devr);
6558 }
6559 
6560 static int mlx5_ib_stage_odp_init(struct mlx5_ib_dev *dev)
6561 {
6562 	return mlx5_ib_odp_init_one(dev);
6563 }
6564 
6565 static void mlx5_ib_stage_odp_cleanup(struct mlx5_ib_dev *dev)
6566 {
6567 	mlx5_ib_odp_cleanup_one(dev);
6568 }
6569 
6570 static const struct ib_device_ops mlx5_ib_dev_hw_stats_ops = {
6571 	.alloc_hw_stats = mlx5_ib_alloc_hw_stats,
6572 	.get_hw_stats = mlx5_ib_get_hw_stats,
6573 	.counter_bind_qp = mlx5_ib_counter_bind_qp,
6574 	.counter_unbind_qp = mlx5_ib_counter_unbind_qp,
6575 	.counter_dealloc = mlx5_ib_counter_dealloc,
6576 	.counter_alloc_stats = mlx5_ib_counter_alloc_stats,
6577 	.counter_update_stats = mlx5_ib_counter_update_stats,
6578 };
6579 
6580 static int mlx5_ib_stage_counters_init(struct mlx5_ib_dev *dev)
6581 {
6582 	if (MLX5_CAP_GEN(dev->mdev, max_qp_cnt)) {
6583 		ib_set_device_ops(&dev->ib_dev, &mlx5_ib_dev_hw_stats_ops);
6584 
6585 		return mlx5_ib_alloc_counters(dev);
6586 	}
6587 
6588 	return 0;
6589 }
6590 
6591 static void mlx5_ib_stage_counters_cleanup(struct mlx5_ib_dev *dev)
6592 {
6593 	if (MLX5_CAP_GEN(dev->mdev, max_qp_cnt))
6594 		mlx5_ib_dealloc_counters(dev);
6595 }
6596 
6597 static int mlx5_ib_stage_cong_debugfs_init(struct mlx5_ib_dev *dev)
6598 {
6599 	mlx5_ib_init_cong_debugfs(dev,
6600 				  mlx5_core_native_port_num(dev->mdev) - 1);
6601 	return 0;
6602 }
6603 
6604 static void mlx5_ib_stage_cong_debugfs_cleanup(struct mlx5_ib_dev *dev)
6605 {
6606 	mlx5_ib_cleanup_cong_debugfs(dev,
6607 				     mlx5_core_native_port_num(dev->mdev) - 1);
6608 }
6609 
6610 static int mlx5_ib_stage_uar_init(struct mlx5_ib_dev *dev)
6611 {
6612 	dev->mdev->priv.uar = mlx5_get_uars_page(dev->mdev);
6613 	return PTR_ERR_OR_ZERO(dev->mdev->priv.uar);
6614 }
6615 
6616 static void mlx5_ib_stage_uar_cleanup(struct mlx5_ib_dev *dev)
6617 {
6618 	mlx5_put_uars_page(dev->mdev, dev->mdev->priv.uar);
6619 }
6620 
6621 static int mlx5_ib_stage_bfrag_init(struct mlx5_ib_dev *dev)
6622 {
6623 	int err;
6624 
6625 	err = mlx5_alloc_bfreg(dev->mdev, &dev->bfreg, false, false);
6626 	if (err)
6627 		return err;
6628 
6629 	err = mlx5_alloc_bfreg(dev->mdev, &dev->fp_bfreg, false, true);
6630 	if (err)
6631 		mlx5_free_bfreg(dev->mdev, &dev->fp_bfreg);
6632 
6633 	return err;
6634 }
6635 
6636 static void mlx5_ib_stage_bfrag_cleanup(struct mlx5_ib_dev *dev)
6637 {
6638 	mlx5_free_bfreg(dev->mdev, &dev->fp_bfreg);
6639 	mlx5_free_bfreg(dev->mdev, &dev->bfreg);
6640 }
6641 
6642 static int mlx5_ib_stage_ib_reg_init(struct mlx5_ib_dev *dev)
6643 {
6644 	const char *name;
6645 
6646 	rdma_set_device_sysfs_group(&dev->ib_dev, &mlx5_attr_group);
6647 	if (!mlx5_lag_is_roce(dev->mdev))
6648 		name = "mlx5_%d";
6649 	else
6650 		name = "mlx5_bond_%d";
6651 	return ib_register_device(&dev->ib_dev, name);
6652 }
6653 
6654 static void mlx5_ib_stage_pre_ib_reg_umr_cleanup(struct mlx5_ib_dev *dev)
6655 {
6656 	destroy_umrc_res(dev);
6657 }
6658 
6659 static void mlx5_ib_stage_ib_reg_cleanup(struct mlx5_ib_dev *dev)
6660 {
6661 	ib_unregister_device(&dev->ib_dev);
6662 }
6663 
6664 static int mlx5_ib_stage_post_ib_reg_umr_init(struct mlx5_ib_dev *dev)
6665 {
6666 	return create_umr_res(dev);
6667 }
6668 
6669 static int mlx5_ib_stage_delay_drop_init(struct mlx5_ib_dev *dev)
6670 {
6671 	init_delay_drop(dev);
6672 
6673 	return 0;
6674 }
6675 
6676 static void mlx5_ib_stage_delay_drop_cleanup(struct mlx5_ib_dev *dev)
6677 {
6678 	cancel_delay_drop(dev);
6679 }
6680 
6681 static int mlx5_ib_stage_dev_notifier_init(struct mlx5_ib_dev *dev)
6682 {
6683 	dev->mdev_events.notifier_call = mlx5_ib_event;
6684 	mlx5_notifier_register(dev->mdev, &dev->mdev_events);
6685 	return 0;
6686 }
6687 
6688 static void mlx5_ib_stage_dev_notifier_cleanup(struct mlx5_ib_dev *dev)
6689 {
6690 	mlx5_notifier_unregister(dev->mdev, &dev->mdev_events);
6691 }
6692 
6693 static int mlx5_ib_stage_devx_init(struct mlx5_ib_dev *dev)
6694 {
6695 	int uid;
6696 
6697 	uid = mlx5_ib_devx_create(dev, false);
6698 	if (uid > 0) {
6699 		dev->devx_whitelist_uid = uid;
6700 		mlx5_ib_devx_init_event_table(dev);
6701 	}
6702 
6703 	return 0;
6704 }
6705 static void mlx5_ib_stage_devx_cleanup(struct mlx5_ib_dev *dev)
6706 {
6707 	if (dev->devx_whitelist_uid) {
6708 		mlx5_ib_devx_cleanup_event_table(dev);
6709 		mlx5_ib_devx_destroy(dev, dev->devx_whitelist_uid);
6710 	}
6711 }
6712 
6713 void __mlx5_ib_remove(struct mlx5_ib_dev *dev,
6714 		      const struct mlx5_ib_profile *profile,
6715 		      int stage)
6716 {
6717 	/* Number of stages to cleanup */
6718 	while (stage) {
6719 		stage--;
6720 		if (profile->stage[stage].cleanup)
6721 			profile->stage[stage].cleanup(dev);
6722 	}
6723 
6724 	kfree(dev->port);
6725 	ib_dealloc_device(&dev->ib_dev);
6726 }
6727 
6728 void *__mlx5_ib_add(struct mlx5_ib_dev *dev,
6729 		    const struct mlx5_ib_profile *profile)
6730 {
6731 	int err;
6732 	int i;
6733 
6734 	for (i = 0; i < MLX5_IB_STAGE_MAX; i++) {
6735 		if (profile->stage[i].init) {
6736 			err = profile->stage[i].init(dev);
6737 			if (err)
6738 				goto err_out;
6739 		}
6740 	}
6741 
6742 	dev->profile = profile;
6743 	dev->ib_active = true;
6744 
6745 	return dev;
6746 
6747 err_out:
6748 	__mlx5_ib_remove(dev, profile, i);
6749 
6750 	return NULL;
6751 }
6752 
6753 static const struct mlx5_ib_profile pf_profile = {
6754 	STAGE_CREATE(MLX5_IB_STAGE_INIT,
6755 		     mlx5_ib_stage_init_init,
6756 		     mlx5_ib_stage_init_cleanup),
6757 	STAGE_CREATE(MLX5_IB_STAGE_FLOW_DB,
6758 		     mlx5_ib_stage_flow_db_init,
6759 		     mlx5_ib_stage_flow_db_cleanup),
6760 	STAGE_CREATE(MLX5_IB_STAGE_CAPS,
6761 		     mlx5_ib_stage_caps_init,
6762 		     NULL),
6763 	STAGE_CREATE(MLX5_IB_STAGE_NON_DEFAULT_CB,
6764 		     mlx5_ib_stage_non_default_cb,
6765 		     NULL),
6766 	STAGE_CREATE(MLX5_IB_STAGE_ROCE,
6767 		     mlx5_ib_stage_roce_init,
6768 		     mlx5_ib_stage_roce_cleanup),
6769 	STAGE_CREATE(MLX5_IB_STAGE_SRQ,
6770 		     mlx5_init_srq_table,
6771 		     mlx5_cleanup_srq_table),
6772 	STAGE_CREATE(MLX5_IB_STAGE_DEVICE_RESOURCES,
6773 		     mlx5_ib_stage_dev_res_init,
6774 		     mlx5_ib_stage_dev_res_cleanup),
6775 	STAGE_CREATE(MLX5_IB_STAGE_DEVICE_NOTIFIER,
6776 		     mlx5_ib_stage_dev_notifier_init,
6777 		     mlx5_ib_stage_dev_notifier_cleanup),
6778 	STAGE_CREATE(MLX5_IB_STAGE_ODP,
6779 		     mlx5_ib_stage_odp_init,
6780 		     mlx5_ib_stage_odp_cleanup),
6781 	STAGE_CREATE(MLX5_IB_STAGE_COUNTERS,
6782 		     mlx5_ib_stage_counters_init,
6783 		     mlx5_ib_stage_counters_cleanup),
6784 	STAGE_CREATE(MLX5_IB_STAGE_CONG_DEBUGFS,
6785 		     mlx5_ib_stage_cong_debugfs_init,
6786 		     mlx5_ib_stage_cong_debugfs_cleanup),
6787 	STAGE_CREATE(MLX5_IB_STAGE_UAR,
6788 		     mlx5_ib_stage_uar_init,
6789 		     mlx5_ib_stage_uar_cleanup),
6790 	STAGE_CREATE(MLX5_IB_STAGE_BFREG,
6791 		     mlx5_ib_stage_bfrag_init,
6792 		     mlx5_ib_stage_bfrag_cleanup),
6793 	STAGE_CREATE(MLX5_IB_STAGE_PRE_IB_REG_UMR,
6794 		     NULL,
6795 		     mlx5_ib_stage_pre_ib_reg_umr_cleanup),
6796 	STAGE_CREATE(MLX5_IB_STAGE_WHITELIST_UID,
6797 		     mlx5_ib_stage_devx_init,
6798 		     mlx5_ib_stage_devx_cleanup),
6799 	STAGE_CREATE(MLX5_IB_STAGE_IB_REG,
6800 		     mlx5_ib_stage_ib_reg_init,
6801 		     mlx5_ib_stage_ib_reg_cleanup),
6802 	STAGE_CREATE(MLX5_IB_STAGE_POST_IB_REG_UMR,
6803 		     mlx5_ib_stage_post_ib_reg_umr_init,
6804 		     NULL),
6805 	STAGE_CREATE(MLX5_IB_STAGE_DELAY_DROP,
6806 		     mlx5_ib_stage_delay_drop_init,
6807 		     mlx5_ib_stage_delay_drop_cleanup),
6808 };
6809 
6810 const struct mlx5_ib_profile uplink_rep_profile = {
6811 	STAGE_CREATE(MLX5_IB_STAGE_INIT,
6812 		     mlx5_ib_stage_init_init,
6813 		     mlx5_ib_stage_init_cleanup),
6814 	STAGE_CREATE(MLX5_IB_STAGE_FLOW_DB,
6815 		     mlx5_ib_stage_flow_db_init,
6816 		     mlx5_ib_stage_flow_db_cleanup),
6817 	STAGE_CREATE(MLX5_IB_STAGE_CAPS,
6818 		     mlx5_ib_stage_caps_init,
6819 		     NULL),
6820 	STAGE_CREATE(MLX5_IB_STAGE_NON_DEFAULT_CB,
6821 		     mlx5_ib_stage_rep_non_default_cb,
6822 		     NULL),
6823 	STAGE_CREATE(MLX5_IB_STAGE_ROCE,
6824 		     mlx5_ib_stage_rep_roce_init,
6825 		     mlx5_ib_stage_rep_roce_cleanup),
6826 	STAGE_CREATE(MLX5_IB_STAGE_SRQ,
6827 		     mlx5_init_srq_table,
6828 		     mlx5_cleanup_srq_table),
6829 	STAGE_CREATE(MLX5_IB_STAGE_DEVICE_RESOURCES,
6830 		     mlx5_ib_stage_dev_res_init,
6831 		     mlx5_ib_stage_dev_res_cleanup),
6832 	STAGE_CREATE(MLX5_IB_STAGE_DEVICE_NOTIFIER,
6833 		     mlx5_ib_stage_dev_notifier_init,
6834 		     mlx5_ib_stage_dev_notifier_cleanup),
6835 	STAGE_CREATE(MLX5_IB_STAGE_COUNTERS,
6836 		     mlx5_ib_stage_counters_init,
6837 		     mlx5_ib_stage_counters_cleanup),
6838 	STAGE_CREATE(MLX5_IB_STAGE_UAR,
6839 		     mlx5_ib_stage_uar_init,
6840 		     mlx5_ib_stage_uar_cleanup),
6841 	STAGE_CREATE(MLX5_IB_STAGE_BFREG,
6842 		     mlx5_ib_stage_bfrag_init,
6843 		     mlx5_ib_stage_bfrag_cleanup),
6844 	STAGE_CREATE(MLX5_IB_STAGE_PRE_IB_REG_UMR,
6845 		     NULL,
6846 		     mlx5_ib_stage_pre_ib_reg_umr_cleanup),
6847 	STAGE_CREATE(MLX5_IB_STAGE_WHITELIST_UID,
6848 		     mlx5_ib_stage_devx_init,
6849 		     mlx5_ib_stage_devx_cleanup),
6850 	STAGE_CREATE(MLX5_IB_STAGE_IB_REG,
6851 		     mlx5_ib_stage_ib_reg_init,
6852 		     mlx5_ib_stage_ib_reg_cleanup),
6853 	STAGE_CREATE(MLX5_IB_STAGE_POST_IB_REG_UMR,
6854 		     mlx5_ib_stage_post_ib_reg_umr_init,
6855 		     NULL),
6856 };
6857 
6858 static void *mlx5_ib_add_slave_port(struct mlx5_core_dev *mdev)
6859 {
6860 	struct mlx5_ib_multiport_info *mpi;
6861 	struct mlx5_ib_dev *dev;
6862 	bool bound = false;
6863 	int err;
6864 
6865 	mpi = kzalloc(sizeof(*mpi), GFP_KERNEL);
6866 	if (!mpi)
6867 		return NULL;
6868 
6869 	mpi->mdev = mdev;
6870 
6871 	err = mlx5_query_nic_vport_system_image_guid(mdev,
6872 						     &mpi->sys_image_guid);
6873 	if (err) {
6874 		kfree(mpi);
6875 		return NULL;
6876 	}
6877 
6878 	mutex_lock(&mlx5_ib_multiport_mutex);
6879 	list_for_each_entry(dev, &mlx5_ib_dev_list, ib_dev_list) {
6880 		if (dev->sys_image_guid == mpi->sys_image_guid)
6881 			bound = mlx5_ib_bind_slave_port(dev, mpi);
6882 
6883 		if (bound) {
6884 			rdma_roce_rescan_device(&dev->ib_dev);
6885 			break;
6886 		}
6887 	}
6888 
6889 	if (!bound) {
6890 		list_add_tail(&mpi->list, &mlx5_ib_unaffiliated_port_list);
6891 		dev_dbg(mdev->device,
6892 			"no suitable IB device found to bind to, added to unaffiliated list.\n");
6893 	}
6894 	mutex_unlock(&mlx5_ib_multiport_mutex);
6895 
6896 	return mpi;
6897 }
6898 
6899 static void *mlx5_ib_add(struct mlx5_core_dev *mdev)
6900 {
6901 	enum rdma_link_layer ll;
6902 	struct mlx5_ib_dev *dev;
6903 	int port_type_cap;
6904 	int num_ports;
6905 
6906 	printk_once(KERN_INFO "%s", mlx5_version);
6907 
6908 	if (MLX5_ESWITCH_MANAGER(mdev) &&
6909 	    mlx5_ib_eswitch_mode(mdev->priv.eswitch) == MLX5_ESWITCH_OFFLOADS) {
6910 		if (!mlx5_core_mp_enabled(mdev))
6911 			mlx5_ib_register_vport_reps(mdev);
6912 		return mdev;
6913 	}
6914 
6915 	port_type_cap = MLX5_CAP_GEN(mdev, port_type);
6916 	ll = mlx5_port_type_cap_to_rdma_ll(port_type_cap);
6917 
6918 	if (mlx5_core_is_mp_slave(mdev) && ll == IB_LINK_LAYER_ETHERNET)
6919 		return mlx5_ib_add_slave_port(mdev);
6920 
6921 	num_ports = max(MLX5_CAP_GEN(mdev, num_ports),
6922 			MLX5_CAP_GEN(mdev, num_vhca_ports));
6923 	dev = ib_alloc_device(mlx5_ib_dev, ib_dev);
6924 	if (!dev)
6925 		return NULL;
6926 	dev->port = kcalloc(num_ports, sizeof(*dev->port),
6927 			     GFP_KERNEL);
6928 	if (!dev->port) {
6929 		ib_dealloc_device(&dev->ib_dev);
6930 		return NULL;
6931 	}
6932 
6933 	dev->mdev = mdev;
6934 	dev->num_ports = num_ports;
6935 
6936 	return __mlx5_ib_add(dev, &pf_profile);
6937 }
6938 
6939 static void mlx5_ib_remove(struct mlx5_core_dev *mdev, void *context)
6940 {
6941 	struct mlx5_ib_multiport_info *mpi;
6942 	struct mlx5_ib_dev *dev;
6943 
6944 	if (MLX5_ESWITCH_MANAGER(mdev) && context == mdev) {
6945 		mlx5_ib_unregister_vport_reps(mdev);
6946 		return;
6947 	}
6948 
6949 	if (mlx5_core_is_mp_slave(mdev)) {
6950 		mpi = context;
6951 		mutex_lock(&mlx5_ib_multiport_mutex);
6952 		if (mpi->ibdev)
6953 			mlx5_ib_unbind_slave_port(mpi->ibdev, mpi);
6954 		list_del(&mpi->list);
6955 		mutex_unlock(&mlx5_ib_multiport_mutex);
6956 		kfree(mpi);
6957 		return;
6958 	}
6959 
6960 	dev = context;
6961 	__mlx5_ib_remove(dev, dev->profile, MLX5_IB_STAGE_MAX);
6962 }
6963 
6964 static struct mlx5_interface mlx5_ib_interface = {
6965 	.add            = mlx5_ib_add,
6966 	.remove         = mlx5_ib_remove,
6967 	.protocol	= MLX5_INTERFACE_PROTOCOL_IB,
6968 };
6969 
6970 unsigned long mlx5_ib_get_xlt_emergency_page(void)
6971 {
6972 	mutex_lock(&xlt_emergency_page_mutex);
6973 	return xlt_emergency_page;
6974 }
6975 
6976 void mlx5_ib_put_xlt_emergency_page(void)
6977 {
6978 	mutex_unlock(&xlt_emergency_page_mutex);
6979 }
6980 
6981 static int __init mlx5_ib_init(void)
6982 {
6983 	int err;
6984 
6985 	xlt_emergency_page = __get_free_page(GFP_KERNEL);
6986 	if (!xlt_emergency_page)
6987 		return -ENOMEM;
6988 
6989 	mutex_init(&xlt_emergency_page_mutex);
6990 
6991 	mlx5_ib_event_wq = alloc_ordered_workqueue("mlx5_ib_event_wq", 0);
6992 	if (!mlx5_ib_event_wq) {
6993 		free_page(xlt_emergency_page);
6994 		return -ENOMEM;
6995 	}
6996 
6997 	mlx5_ib_odp_init();
6998 
6999 	err = mlx5_register_interface(&mlx5_ib_interface);
7000 
7001 	return err;
7002 }
7003 
7004 static void __exit mlx5_ib_cleanup(void)
7005 {
7006 	mlx5_unregister_interface(&mlx5_ib_interface);
7007 	destroy_workqueue(mlx5_ib_event_wq);
7008 	mutex_destroy(&xlt_emergency_page_mutex);
7009 	free_page(xlt_emergency_page);
7010 }
7011 
7012 module_init(mlx5_ib_init);
7013 module_exit(mlx5_ib_cleanup);
7014