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

[Eric Huang]

to prevent svm range to be overflow or underflow.

Signed-off-by: Eric Huang <jinhuieric.hu...@amd.com>
---
 drivers/gpu/drm/amd/amdkfd/kfd_svm.c | 8 +++++++-
 1 file changed, 7 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..37b79ccfc2ad 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_svm.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_svm.c
@@ -3473,7 +3473,13 @@ 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 (size == 0)
+               retrune -EINVAL;
+
+       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