[PATCH 1/1] RDMA/mlx5: Make mlx5 device work with ib_device_get_by_netdev

[Zhu Yanjun <yanjun.zhu@intel.com>]

From: Zhu Yanjun <yanjun.zhu@linux.dev>

Before mlx5 ib device is registered, the function ib_device_set_netdev
is not called to map the mlx5 ib device with the related net device.

As such, when the function ib_device_get_by_netdev is called to get ib
device, NULL is returned.

Other ib devices, such as irdma, rxe and so on, the function
ib_device_get_by_netdev can get ib device from the related net device.

Signed-off-by: Zhu Yanjun <yanjun.zhu@linux.dev>
---
 drivers/infiniband/hw/mlx5/main.c | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/drivers/infiniband/hw/mlx5/main.c b/drivers/infiniband/hw/mlx5/main.c
index 883d7c60143e..6899c3f73509 100644
--- a/drivers/infiniband/hw/mlx5/main.c
+++ b/drivers/infiniband/hw/mlx5/main.c
@@ -168,6 +168,7 @@ static int mlx5_netdev_event(struct notifier_block *this,
 	u32 port_num = roce->native_port_num;
 	struct mlx5_core_dev *mdev;
 	struct mlx5_ib_dev *ibdev;
+	int ret = 0;
 
 	ibdev = roce->dev;
 	mdev = mlx5_ib_get_native_port_mdev(ibdev, port_num, NULL);
@@ -183,6 +184,14 @@ static int mlx5_netdev_event(struct notifier_block *this,
 		if (ndev->dev.parent == mdev->device)
 			roce->netdev = ndev;
 		write_unlock(&roce->netdev_lock);
+		if (ndev->dev.parent == mdev->device) {
+			ret = ib_device_set_netdev(&ibdev->ib_dev, ndev, port_num);
+			if (ret) {
+				pr_warn("func: %s, error: %d\n", __func__, ret);
+				goto done;
+			}
+		}
+
 		break;
 
 	case NETDEV_UNREGISTER:
@@ -191,6 +200,15 @@ static int mlx5_netdev_event(struct notifier_block *this,
 		if (roce->netdev == ndev)
 			roce->netdev = NULL;
 		write_unlock(&roce->netdev_lock);
+
+		if (roce->netdev == ndev) {
+			ret = ib_device_set_netdev(&ibdev->ib_dev, NULL, port_num);
+			if (ret) {
+				pr_warn("func: %s, error: %d\n", __func__, ret);
+				goto done;
+			}
+		}
+
 		break;
 
 	case NETDEV_CHANGE:
-- 
2.27.0