static void esw_destroy_legacy_fdb_table(struct mlx5_eswitch *esw);
 static void esw_cleanup_vepa_rules(struct mlx5_eswitch *esw);
 
+static int mlx5_eswitch_check(const struct mlx5_core_dev *dev)
+{
+       if (MLX5_CAP_GEN(dev, port_type) != MLX5_CAP_PORT_TYPE_ETH)
+               return -EOPNOTSUPP;
+
+       if (!MLX5_ESWITCH_MANAGER(dev))
+               return -EPERM;
+
+       return 0;
+}
+
+struct mlx5_eswitch *mlx5_devlink_eswitch_get(struct devlink *devlink)
+{
+       struct mlx5_core_dev *dev = devlink_priv(devlink);
+       int err;
+
+       err = mlx5_eswitch_check(dev);
+       if (err)
+               return ERR_PTR(err);
+
+       return dev->priv.eswitch;
+}
+
 struct mlx5_vport *__must_check
 mlx5_eswitch_get_vport(struct mlx5_eswitch *esw, u16 vport_num)
 {
 
        return 0;
 }
 
-static int mlx5_eswitch_check(const struct mlx5_core_dev *dev)
-{
-       if (MLX5_CAP_GEN(dev, port_type) != MLX5_CAP_PORT_TYPE_ETH)
-               return -EOPNOTSUPP;
-
-       if(!MLX5_ESWITCH_MANAGER(dev))
-               return -EPERM;
-
-       return 0;
-}
-
 static int eswitch_devlink_esw_mode_check(const struct mlx5_eswitch *esw)
 {
        /* devlink commands in NONE eswitch mode are currently supported only
 int mlx5_devlink_eswitch_mode_set(struct devlink *devlink, u16 mode,
                                  struct netlink_ext_ack *extack)
 {
-       struct mlx5_core_dev *dev = devlink_priv(devlink);
-       struct mlx5_eswitch *esw = dev->priv.eswitch;
        u16 cur_mlx5_mode, mlx5_mode = 0;
+       struct mlx5_eswitch *esw;
        int err;
 
-       err = mlx5_eswitch_check(dev);
-       if (err)
-               return err;
+       esw = mlx5_devlink_eswitch_get(devlink);
+       if (IS_ERR(esw))
+               return PTR_ERR(esw);
 
        if (esw_mode_from_devlink(mode, &mlx5_mode))
                return -EINVAL;
 
 int mlx5_devlink_eswitch_mode_get(struct devlink *devlink, u16 *mode)
 {
-       struct mlx5_core_dev *dev = devlink_priv(devlink);
-       struct mlx5_eswitch *esw = dev->priv.eswitch;
+       struct mlx5_eswitch *esw;
        int err;
 
-       err = mlx5_eswitch_check(dev);
-       if (err)
-               return err;
+       esw = mlx5_devlink_eswitch_get(devlink);
+       if (IS_ERR(esw))
+               return PTR_ERR(esw);
 
        mutex_lock(&esw->mode_lock);
-       err = eswitch_devlink_esw_mode_check(dev->priv.eswitch);
+       err = eswitch_devlink_esw_mode_check(esw);
        if (err)
                goto unlock;
 
                                         struct netlink_ext_ack *extack)
 {
        struct mlx5_core_dev *dev = devlink_priv(devlink);
-       struct mlx5_eswitch *esw = dev->priv.eswitch;
        int err, vport, num_vport;
+       struct mlx5_eswitch *esw;
        u8 mlx5_mode;
 
-       err = mlx5_eswitch_check(dev);
-       if (err)
-               return err;
+       esw = mlx5_devlink_eswitch_get(devlink);
+       if (IS_ERR(esw))
+               return PTR_ERR(esw);
 
        mutex_lock(&esw->mode_lock);
        err = eswitch_devlink_esw_mode_check(esw);
 
 int mlx5_devlink_eswitch_inline_mode_get(struct devlink *devlink, u8 *mode)
 {
-       struct mlx5_core_dev *dev = devlink_priv(devlink);
-       struct mlx5_eswitch *esw = dev->priv.eswitch;
+       struct mlx5_eswitch *esw;
        int err;
 
-       err = mlx5_eswitch_check(dev);
-       if (err)
-               return err;
+       esw = mlx5_devlink_eswitch_get(devlink);
+       if (IS_ERR(esw))
+               return PTR_ERR(esw);
 
        mutex_lock(&esw->mode_lock);
        err = eswitch_devlink_esw_mode_check(esw);
                                        struct netlink_ext_ack *extack)
 {
        struct mlx5_core_dev *dev = devlink_priv(devlink);
-       struct mlx5_eswitch *esw = dev->priv.eswitch;
+       struct mlx5_eswitch *esw;
        int err;
 
-       err = mlx5_eswitch_check(dev);
-       if (err)
-               return err;
+       esw = mlx5_devlink_eswitch_get(devlink);
+       if (IS_ERR(esw))
+               return PTR_ERR(esw);
 
        mutex_lock(&esw->mode_lock);
        err = eswitch_devlink_esw_mode_check(esw);
 int mlx5_devlink_eswitch_encap_mode_get(struct devlink *devlink,
                                        enum devlink_eswitch_encap_mode *encap)
 {
-       struct mlx5_core_dev *dev = devlink_priv(devlink);
-       struct mlx5_eswitch *esw = dev->priv.eswitch;
+       struct mlx5_eswitch *esw;
        int err;
 
-       err = mlx5_eswitch_check(dev);
-       if (err)
-               return err;
+       esw = mlx5_devlink_eswitch_get(devlink);
+       if (IS_ERR(esw))
+               return PTR_ERR(esw);
+
 
        mutex_lock(&esw->mode_lock);
        err = eswitch_devlink_esw_mode_check(esw);