struct netlink_ext_ack *extack)
 {
        struct bnxt *bp = bnxt_get_bp_from_dl(devlink);
-       int rc = 0;
 
-       devl_lock(devlink);
        if (bp->eswitch_mode == mode) {
                netdev_info(bp->dev, "already in %s eswitch mode\n",
                            mode == DEVLINK_ESWITCH_MODE_LEGACY ?
                            "legacy" : "switchdev");
-               rc = -EINVAL;
-               goto done;
+               return -EINVAL;
        }
 
        switch (mode) {
        case DEVLINK_ESWITCH_MODE_LEGACY:
                bnxt_vf_reps_destroy(bp);
-               break;
+               return 0;
 
        case DEVLINK_ESWITCH_MODE_SWITCHDEV:
                if (bp->hwrm_spec_code < 0x10803) {
                        netdev_warn(bp->dev, "FW does not support SRIOV E-Switch SWITCHDEV mode\n");
-                       rc = -ENOTSUPP;
-                       goto done;
+                       return -ENOTSUPP;
                }
 
                if (pci_num_vf(bp->pdev) == 0) {
                        netdev_info(bp->dev, "Enable VFs before setting switchdev mode\n");
-                       rc = -EPERM;
-                       goto done;
+                       return -EPERM;
                }
-               rc = bnxt_vf_reps_create(bp);
-               break;
+               return bnxt_vf_reps_create(bp);
 
        default:
-               rc = -EINVAL;
-               goto done;
+               return -EINVAL;
        }
-done:
-       devl_unlock(devlink);
-       return rc;
 }
 
 #endif
 
                !mlx5_core_is_ecpf_esw_manager(esw->dev)) ? -EOPNOTSUPP : 0;
 }
 
+/* FIXME: devl_unlock() followed by devl_lock() inside driver callback
+ * is never correct and prone to races. It's a transitional workaround,
+ * never repeat this pattern.
+ *
+ * This code MUST be fixed before removing devlink_mutex as it is safe
+ * to do only because of that mutex.
+ */
+static void mlx5_eswtich_mode_callback_enter(struct devlink *devlink,
+                                            struct mlx5_eswitch *esw)
+{
+       devl_unlock(devlink);
+       down_write(&esw->mode_lock);
+}
+
+static void mlx5_eswtich_mode_callback_exit(struct devlink *devlink,
+                                           struct mlx5_eswitch *esw)
+{
+       up_write(&esw->mode_lock);
+       devl_lock(devlink);
+}
+
 int mlx5_devlink_eswitch_mode_set(struct devlink *devlink, u16 mode,
                                  struct netlink_ext_ack *extack)
 {
        if (esw_mode_from_devlink(mode, &mlx5_mode))
                return -EINVAL;
 
+       /* FIXME: devl_unlock() followed by devl_lock() inside driver callback
+        * is never correct and prone to races. It's a transitional workaround,
+        * never repeat this pattern.
+        *
+        * This code MUST be fixed before removing devlink_mutex as it is safe
+        * to do only because of that mutex.
+        */
+       devl_unlock(devlink);
+
        mlx5_lag_disable_change(esw->dev);
        err = mlx5_esw_try_lock(esw);
        if (err < 0) {
        mlx5_esw_unlock(esw);
 enable_lag:
        mlx5_lag_enable_change(esw->dev);
+       devl_lock(devlink);
        return err;
 }
 
        if (IS_ERR(esw))
                return PTR_ERR(esw);
 
-       down_write(&esw->mode_lock);
+       mlx5_eswtich_mode_callback_enter(devlink, esw);
        err = eswitch_devlink_esw_mode_check(esw);
        if (err)
                goto unlock;
 
        err = esw_mode_to_devlink(esw->mode, mode);
 unlock:
-       up_write(&esw->mode_lock);
+       mlx5_eswtich_mode_callback_exit(devlink, esw);
        return err;
 }
 
        if (IS_ERR(esw))
                return PTR_ERR(esw);
 
-       down_write(&esw->mode_lock);
+       mlx5_eswtich_mode_callback_enter(devlink, esw);
        err = eswitch_devlink_esw_mode_check(esw);
        if (err)
                goto out;
                goto out;
 
        esw->offloads.inline_mode = mlx5_mode;
-       up_write(&esw->mode_lock);
+       mlx5_eswtich_mode_callback_exit(devlink, esw);
        return 0;
 
 out:
-       up_write(&esw->mode_lock);
+       mlx5_eswtich_mode_callback_exit(devlink, esw);
        return err;
 }
 
        if (IS_ERR(esw))
                return PTR_ERR(esw);
 
-       down_write(&esw->mode_lock);
+       mlx5_eswtich_mode_callback_enter(devlink, esw);
        err = eswitch_devlink_esw_mode_check(esw);
        if (err)
                goto unlock;
 
        err = esw_inline_mode_to_devlink(esw->offloads.inline_mode, mode);
 unlock:
-       up_write(&esw->mode_lock);
+       mlx5_eswtich_mode_callback_exit(devlink, esw);
        return err;
 }
 
        if (IS_ERR(esw))
                return PTR_ERR(esw);
 
-       down_write(&esw->mode_lock);
+       mlx5_eswtich_mode_callback_enter(devlink, esw);
        err = eswitch_devlink_esw_mode_check(esw);
        if (err)
                goto unlock;
        }
 
 unlock:
-       up_write(&esw->mode_lock);
+       mlx5_eswtich_mode_callback_exit(devlink, esw);
        return err;
 }
 
        if (IS_ERR(esw))
                return PTR_ERR(esw);
 
-
-       down_write(&esw->mode_lock);
+       mlx5_eswtich_mode_callback_enter(devlink, esw);
        err = eswitch_devlink_esw_mode_check(esw);
        if (err)
                goto unlock;
 
        *encap = esw->offloads.encap;
 unlock:
-       up_write(&esw->mode_lock);
+       mlx5_eswtich_mode_callback_exit(devlink, esw);
        return err;
 }
 
 
                                         struct netlink_ext_ack *extack)
 {
        struct nsim_dev *nsim_dev = devlink_priv(devlink);
-       int err = 0;
 
-       devl_lock(devlink);
        if (mode == nsim_dev->esw_mode)
-               goto unlock;
+               return 0;
 
        if (mode == DEVLINK_ESWITCH_MODE_LEGACY)
-               err = nsim_esw_legacy_enable(nsim_dev, extack);
-       else if (mode == DEVLINK_ESWITCH_MODE_SWITCHDEV)
-               err = nsim_esw_switchdev_enable(nsim_dev, extack);
-       else
-               err = -EINVAL;
+               return nsim_esw_legacy_enable(nsim_dev, extack);
+       if (mode == DEVLINK_ESWITCH_MODE_SWITCHDEV)
+               return nsim_esw_switchdev_enable(nsim_dev, extack);
 
-unlock:
-       devl_unlock(devlink);
-       return err;
+       return -EINVAL;
 }
 
 static int nsim_devlink_eswitch_mode_get(struct devlink *devlink, u16 *mode)
 
 {
        struct devlink_rate *devlink_rate;
 
-       /* Take the lock to sync with destroy */
-       mutex_lock(&devlink->lock);
        list_for_each_entry(devlink_rate, &devlink->rate_list, list)
                if (devlink_rate_is_node(devlink_rate)) {
-                       mutex_unlock(&devlink->lock);
                        NL_SET_ERR_MSG_MOD(extack, "Rate node(s) exists.");
                        return -EBUSY;
                }
-       mutex_unlock(&devlink->lock);
        return 0;
 }
 
                .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
                .doit = devlink_nl_cmd_eswitch_get_doit,
                .flags = GENL_ADMIN_PERM,
-               .internal_flags = DEVLINK_NL_FLAG_NO_LOCK,
        },
        {
                .cmd = DEVLINK_CMD_ESWITCH_SET,
                .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
                .doit = devlink_nl_cmd_eswitch_set_doit,
                .flags = GENL_ADMIN_PERM,
-               .internal_flags = DEVLINK_NL_FLAG_NO_LOCK,
        },
        {
                .cmd = DEVLINK_CMD_DPIPE_TABLE_GET,