|
@@ -648,11 +648,11 @@ int virtio_gpu_cmd_get_capset(struct virtio_gpu_device *vgdev,
|
|
{
|
|
{
|
|
struct virtio_gpu_get_capset *cmd_p;
|
|
struct virtio_gpu_get_capset *cmd_p;
|
|
struct virtio_gpu_vbuffer *vbuf;
|
|
struct virtio_gpu_vbuffer *vbuf;
|
|
- int max_size = vgdev->capsets[idx].max_size;
|
|
|
|
|
|
+ int max_size;
|
|
struct virtio_gpu_drv_cap_cache *cache_ent;
|
|
struct virtio_gpu_drv_cap_cache *cache_ent;
|
|
void *resp_buf;
|
|
void *resp_buf;
|
|
|
|
|
|
- if (idx > vgdev->num_capsets)
|
|
|
|
|
|
+ if (idx >= vgdev->num_capsets)
|
|
return -EINVAL;
|
|
return -EINVAL;
|
|
|
|
|
|
if (version > vgdev->capsets[idx].max_version)
|
|
if (version > vgdev->capsets[idx].max_version)
|
|
@@ -662,6 +662,7 @@ int virtio_gpu_cmd_get_capset(struct virtio_gpu_device *vgdev,
|
|
if (!cache_ent)
|
|
if (!cache_ent)
|
|
return -ENOMEM;
|
|
return -ENOMEM;
|
|
|
|
|
|
|
|
+ max_size = vgdev->capsets[idx].max_size;
|
|
cache_ent->caps_cache = kmalloc(max_size, GFP_KERNEL);
|
|
cache_ent->caps_cache = kmalloc(max_size, GFP_KERNEL);
|
|
if (!cache_ent->caps_cache) {
|
|
if (!cache_ent->caps_cache) {
|
|
kfree(cache_ent);
|
|
kfree(cache_ent);
|