bool mlx5_eswitch_block_encap(struct mlx5_core_dev *dev)
 {
-       struct devlink *devlink = priv_to_devlink(dev);
-       struct mlx5_eswitch *esw;
+       struct mlx5_eswitch *esw = dev->priv.eswitch;
 
-       devl_lock(devlink);
-       esw = mlx5_devlink_eswitch_get(devlink);
-       if (IS_ERR(esw)) {
-               devl_unlock(devlink);
-               /* Failure means no eswitch => not possible to change encap */
+       if (!mlx5_esw_allowed(esw))
                return true;
-       }
 
        down_write(&esw->mode_lock);
        if (esw->mode != MLX5_ESWITCH_LEGACY &&
            esw->offloads.encap != DEVLINK_ESWITCH_ENCAP_MODE_NONE) {
                up_write(&esw->mode_lock);
-               devl_unlock(devlink);
                return false;
        }
 
        esw->offloads.num_block_encap++;
        up_write(&esw->mode_lock);
-       devl_unlock(devlink);
        return true;
 }
 
 void mlx5_eswitch_unblock_encap(struct mlx5_core_dev *dev)
 {
-       struct devlink *devlink = priv_to_devlink(dev);
-       struct mlx5_eswitch *esw;
+       struct mlx5_eswitch *esw = dev->priv.eswitch;
 
-       esw = mlx5_devlink_eswitch_get(devlink);
-       if (IS_ERR(esw))
+       if (!mlx5_esw_allowed(esw))
                return;
 
        down_write(&esw->mode_lock);