From: Long Li <lon...@microsoft.com>

Change mana_get_primary_netdev_rcu() to mana_get_primary_netdev(), and
return the ndev with refcount held. The caller is responsible for dropping
the refcount.

Also drop the check for IFF_SLAVE as it is not necessary if the upper
device is present.

Signed-off-by: Long Li <lon...@microsoft.com>
---
Changes
v4: use netdev_hold()/netdev_put() and remove the check for IFF_SLAVE
v5: use netdevice_tracker in mana_ib_dev for netdev_hold()/netdev_put()
v6: rebase to latest rdma-next

 drivers/infiniband/hw/mana/device.c           |  7 +++---
 drivers/infiniband/hw/mana/mana_ib.h          |  1 +
 drivers/net/ethernet/microsoft/mana/mana_en.c | 22 ++++++++++++-------
 include/net/mana/mana.h                       |  4 +++-
 4 files changed, 21 insertions(+), 13 deletions(-)

diff --git a/drivers/infiniband/hw/mana/device.c 
b/drivers/infiniband/hw/mana/device.c
index d1a02c54a236..9357a9845c2c 100644
--- a/drivers/infiniband/hw/mana/device.c
+++ b/drivers/infiniband/hw/mana/device.c
@@ -98,10 +98,8 @@ static int mana_ib_probe(struct auxiliary_device *adev,
        dev->ib_dev.num_comp_vectors = mdev->gdma_context->max_num_queues;
        dev->ib_dev.dev.parent = mdev->gdma_context->dev;
 
-       rcu_read_lock(); /* required to get primary netdev */
-       ndev = mana_get_primary_netdev_rcu(mc, 0);
+       ndev = mana_get_primary_netdev(mc, 0, &dev->dev_tracker);
        if (!ndev) {
-               rcu_read_unlock();
                ret = -ENODEV;
                ibdev_err(&dev->ib_dev, "Failed to get netdev for IB port 1");
                goto free_ib_device;
@@ -109,7 +107,8 @@ static int mana_ib_probe(struct auxiliary_device *adev,
        ether_addr_copy(mac_addr, ndev->dev_addr);
        addrconf_addr_eui48((u8 *)&dev->ib_dev.node_guid, ndev->dev_addr);
        ret = ib_device_set_netdev(&dev->ib_dev, ndev, 1);
-       rcu_read_unlock();
+       /* mana_get_primary_netdev() returns ndev with refcount held */
+       netdev_put(ndev, &dev->dev_tracker);
        if (ret) {
                ibdev_err(&dev->ib_dev, "Failed to set ib netdev, ret %d", ret);
                goto free_ib_device;
diff --git a/drivers/infiniband/hw/mana/mana_ib.h 
b/drivers/infiniband/hw/mana/mana_ib.h
index 77fc1032eda8..81a7e7474462 100644
--- a/drivers/infiniband/hw/mana/mana_ib.h
+++ b/drivers/infiniband/hw/mana/mana_ib.h
@@ -78,6 +78,7 @@ struct mana_ib_dev {
        struct xarray qp_table_wq;
        struct mana_ib_adapter_caps adapter_caps;
        struct dma_pool *av_pool;
+       netdevice_tracker dev_tracker;
 };
 
 struct mana_ib_wq {
diff --git a/drivers/net/ethernet/microsoft/mana/mana_en.c 
b/drivers/net/ethernet/microsoft/mana/mana_en.c
index aa1e47233fe5..4e870b11f946 100644
--- a/drivers/net/ethernet/microsoft/mana/mana_en.c
+++ b/drivers/net/ethernet/microsoft/mana/mana_en.c
@@ -3131,21 +3131,27 @@ void mana_remove(struct gdma_dev *gd, bool suspending)
        kfree(ac);
 }
 
-struct net_device *mana_get_primary_netdev_rcu(struct mana_context *ac, u32 
port_index)
+struct net_device *mana_get_primary_netdev(struct mana_context *ac,
+                                          u32 port_index,
+                                          netdevice_tracker *tracker)
 {
        struct net_device *ndev;
 
-       RCU_LOCKDEP_WARN(!rcu_read_lock_held(),
-                        "Taking primary netdev without holding the RCU read 
lock");
        if (port_index >= ac->num_ports)
                return NULL;
 
-       /* When mana is used in netvsc, the upper netdevice should be returned. 
*/
-       if (ac->ports[port_index]->flags & IFF_SLAVE)
-               ndev = netdev_master_upper_dev_get_rcu(ac->ports[port_index]);
-       else
+       rcu_read_lock();
+
+       /* If mana is used in netvsc, the upper netdevice should be returned. */
+       ndev = netdev_master_upper_dev_get_rcu(ac->ports[port_index]);
+
+       /* If there is no upper device, use the parent Ethernet device */
+       if (!ndev)
                ndev = ac->ports[port_index];
 
+       netdev_hold(ndev, tracker, GFP_ATOMIC);
+       rcu_read_unlock();
+
        return ndev;
 }
-EXPORT_SYMBOL_NS(mana_get_primary_netdev_rcu, "NET_MANA");
+EXPORT_SYMBOL_NS(mana_get_primary_netdev, "NET_MANA");
diff --git a/include/net/mana/mana.h b/include/net/mana/mana.h
index 0d00b24eacaf..0f78065de8fe 100644
--- a/include/net/mana/mana.h
+++ b/include/net/mana/mana.h
@@ -827,5 +827,7 @@ int mana_cfg_vport(struct mana_port_context *apc, u32 
protection_dom_id,
                   u32 doorbell_pg_id);
 void mana_uncfg_vport(struct mana_port_context *apc);
 
-struct net_device *mana_get_primary_netdev_rcu(struct mana_context *ac, u32 
port_index);
+struct net_device *mana_get_primary_netdev(struct mana_context *ac,
+                                          u32 port_index,
+                                          netdevice_tracker *tracker);
 #endif /* _MANA_H */
-- 
2.34.1


Reply via email to