[PATCH] drm/amdkfd: add overflow check in svm_range_is_valid

[Eric Huang]

to prevent svm range to be overflow.

Signed-off-by: Eric Huang <jinhuieric.hu...@amd.com>
---
 drivers/gpu/drm/amd/amdkfd/kfd_svm.c | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_svm.c 
b/drivers/gpu/drm/amd/amdkfd/kfd_svm.c
index ffecd818e611..ad4325d25e7c 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_svm.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_svm.c
@@ -3473,7 +3473,9 @@ svm_range_is_valid(struct kfd_process *p, uint64_t start, 
uint64_t size)
        unsigned long start_unchg = start;
 
        start <<= PAGE_SHIFT;
-       end = start + (size << PAGE_SHIFT);
+       if (check_add_overflow(start, size << PAGE_SHIFT, &end))
+               return -EOVERFLOW;
+
        do {
                vma = vma_lookup(p->mm, start);
                if (!vma || (vma->vm_flags & device_vma))
-- 
2.34.1