Преглед изворни кода

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>
Christian König пре 10 година
родитељ
комит
0be52de91c
1 измењених фајлова са 7 додато и 0 уклоњено
  1. 7 0
      drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c

+ 7 - 0
drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c

@@ -936,6 +936,13 @@ int amdgpu_vm_bo_map(struct amdgpu_device *adev,
 	uint64_t eaddr;
 	uint64_t eaddr;
 	int r;
 	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 */
 	/* make sure object fit at this offset */
 	eaddr = saddr + size;
 	eaddr = saddr + size;
 	if ((saddr >= eaddr) || (offset + size > amdgpu_bo_size(bo_va->bo))) {
 	if ((saddr >= eaddr) || (offset + size > amdgpu_bo_size(bo_va->bo))) {