diff --git a/drivers/gpu/nvgpu/common/mm/vm.c b/drivers/gpu/nvgpu/common/mm/vm.c index 49c2f64bd..64cca5edc 100644 --- a/drivers/gpu/nvgpu/common/mm/vm.c +++ b/drivers/gpu/nvgpu/common/mm/vm.c @@ -1014,6 +1014,10 @@ int nvgpu_vm_map(struct vm_gk20a *vm, binfo.flags = flags; binfo.size = nvgpu_os_buf_get_size(os_buf); + if (binfo.size == 0UL) { + nvgpu_err(g, "Invalid buffer size"); + return -EINVAL; + } binfo.incompr_kind = incompr_kind; #ifdef CONFIG_NVGPU_COMPRESSION