lkml.org 
[lkml]   [2021]   [Feb]   [4]   [last100]   RSS Feed
Views: [wrap][no wrap]   [headers]  [forward] 
 
Messages in this thread
    Patch in this message
    /
    From
    Subject[PATCH v3 09/13] vhost/vdpa: remove vhost_vdpa_config_validate()
    Date
    get_config() and set_config() callbacks in the 'struct vdpa_config_ops'
    usually already validated the inputs. Also now they can return an error,
    so we don't need to validate them here anymore.

    Let's use the return value of these callbacks and return it in case of
    error in vhost_vdpa_get_config() and vhost_vdpa_set_config().

    Originally-by: Xie Yongji <xieyongji@bytedance.com>
    Signed-off-by: Stefano Garzarella <sgarzare@redhat.com>
    ---
    drivers/vhost/vdpa.c | 41 +++++++++++++----------------------------
    1 file changed, 13 insertions(+), 28 deletions(-)

    diff --git a/drivers/vhost/vdpa.c b/drivers/vhost/vdpa.c
    index ef688c8c0e0e..d61e779000a8 100644
    --- a/drivers/vhost/vdpa.c
    +++ b/drivers/vhost/vdpa.c
    @@ -185,51 +185,35 @@ static long vhost_vdpa_set_status(struct vhost_vdpa *v, u8 __user *statusp)
    return 0;
    }

    -static int vhost_vdpa_config_validate(struct vhost_vdpa *v,
    - struct vhost_vdpa_config *c)
    -{
    - long size = 0;
    -
    - switch (v->virtio_id) {
    - case VIRTIO_ID_NET:
    - size = sizeof(struct virtio_net_config);
    - break;
    - }
    -
    - if (c->len == 0)
    - return -EINVAL;
    -
    - if (c->len > size - c->off)
    - return -E2BIG;
    -
    - return 0;
    -}
    -
    static long vhost_vdpa_get_config(struct vhost_vdpa *v,
    struct vhost_vdpa_config __user *c)
    {
    struct vdpa_device *vdpa = v->vdpa;
    struct vhost_vdpa_config config;
    unsigned long size = offsetof(struct vhost_vdpa_config, buf);
    + long ret;
    u8 *buf;

    if (copy_from_user(&config, c, size))
    return -EFAULT;
    - if (vhost_vdpa_config_validate(v, &config))
    + if (config.len == 0)
    return -EINVAL;
    buf = kvzalloc(config.len, GFP_KERNEL);
    if (!buf)
    return -ENOMEM;

    - vdpa_get_config(vdpa, config.off, buf, config.len);
    + ret = vdpa_get_config(vdpa, config.off, buf, config.len);
    + if (ret)
    + goto out;

    if (copy_to_user(c->buf, buf, config.len)) {
    - kvfree(buf);
    - return -EFAULT;
    + ret = -EFAULT;
    + goto out;
    }

    +out:
    kvfree(buf);
    - return 0;
    + return ret;
    }

    static long vhost_vdpa_set_config(struct vhost_vdpa *v,
    @@ -239,21 +223,22 @@ static long vhost_vdpa_set_config(struct vhost_vdpa *v,
    const struct vdpa_config_ops *ops = vdpa->config;
    struct vhost_vdpa_config config;
    unsigned long size = offsetof(struct vhost_vdpa_config, buf);
    + long ret;
    u8 *buf;

    if (copy_from_user(&config, c, size))
    return -EFAULT;
    - if (vhost_vdpa_config_validate(v, &config))
    + if (config.len == 0)
    return -EINVAL;

    buf = vmemdup_user(c->buf, config.len);
    if (IS_ERR(buf))
    return PTR_ERR(buf);

    - ops->set_config(vdpa, config.off, buf, config.len);
    + ret = ops->set_config(vdpa, config.off, buf, config.len);

    kvfree(buf);
    - return 0;
    + return ret;
    }

    static long vhost_vdpa_get_features(struct vhost_vdpa *v, u64 __user *featurep)
    --
    2.29.2
    \
     
     \ /
      Last update: 2021-02-04 18:48    [W:5.395 / U:0.008 seconds]
    ©2003-2020 Jasper Spaans|hosted at Digital Ocean and TransIP|Read the blog|Advertise on this site