drm/amdgpu: validate amdgpu_vm_bo_map parameters

Signed-off-by: Christian König <christian.koenig@amd.com>
Reviewed-by: Jammy Zhou <Jammy.Zhou@amd.com>
Reviewed-by: Alex Deucher <alexander.deucher@amd.com>

1 files changed, 7 insertions, 0 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
index 1cc01fb409dc..b25e533fd6e8 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
@@ -936,6 +936,13 @@ int amdgpu_vm_bo_map(struct amdgpu_device *adev,
         uint64_t eaddr;
         int r;
 
+        /* validate the parameters */
+        if (saddr & AMDGPU_GPU_PAGE_MASK || offset & AMDGPU_GPU_PAGE_MASK ||
+            size == 0 || size & AMDGPU_GPU_PAGE_MASK) {
+                amdgpu_bo_unreserve(bo_va->bo);
+                return -EINVAL;
+        }
+
         /* make sure object fit at this offset */
         eaddr = saddr + size;
         if ((saddr >= eaddr) || (offset + size > amdgpu_bo_size(bo_va->bo))) {