add a simple check to reject any size not aligned to the
min_page_size.

Signed-off-by: Arunpravin Paneer Selvam <arunpravin.paneersel...@amd.com>
---
 drivers/gpu/drm/drm_buddy.c | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/drivers/gpu/drm/drm_buddy.c b/drivers/gpu/drm/drm_buddy.c
index 72f52f293249..b503c88786b0 100644
--- a/drivers/gpu/drm/drm_buddy.c
+++ b/drivers/gpu/drm/drm_buddy.c
@@ -661,6 +661,9 @@ int drm_buddy_alloc_blocks(struct drm_buddy *mm,
        if (range_overflows(start, size, mm->size))
                return -EINVAL;
 
+       if (WARN_ON(!IS_ALIGNED(size, min_page_size)))
+               return -EINVAL;
+
        /* Actual range allocation */
        if (start + size == end)
                return __drm_buddy_alloc_range(mm, start, size, blocks);
-- 
2.25.1

Reply via email to