summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorQintaoShen <unSimple1993@163.com>2022-03-24 11:26:23 +0300
committerGreg Kroah-Hartman <gregkh@linuxfoundation.org>2022-04-20 10:36:20 +0300
commit0a692c625e373fef692ffbc7fc41f8a025f01cb7 (patch)
treefb529cc73b6f7ea9fbef8b27a9f39adb12e879df
parentc6f7f5f67fc5c907ce244b0016ff8d7028557575 (diff)
downloadlinux-0a692c625e373fef692ffbc7fc41f8a025f01cb7.tar.xz
drm/amdkfd: Check for potential null return of kmalloc_array()
[ Upstream commit ebbb7bb9e80305820dc2328a371c1b35679f2667 ] As the kmalloc_array() may return null, the 'event_waiters[i].wait' would lead to null-pointer dereference. Therefore, it is better to check the return value of kmalloc_array() to avoid this confusion. Signed-off-by: QintaoShen <unSimple1993@163.com> Signed-off-by: Alex Deucher <alexander.deucher@amd.com> Signed-off-by: Sasha Levin <sashal@kernel.org>
-rw-r--r--drivers/gpu/drm/amd/amdkfd/kfd_events.c2
1 files changed, 2 insertions, 0 deletions
diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_events.c b/drivers/gpu/drm/amd/amdkfd/kfd_events.c
index afe72dd11325..6ca7e12bdab8 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_events.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_events.c
@@ -531,6 +531,8 @@ static struct kfd_event_waiter *alloc_event_waiters(uint32_t num_events)
event_waiters = kmalloc_array(num_events,
sizeof(struct kfd_event_waiter),
GFP_KERNEL);
+ if (!event_waiters)
+ return NULL;
for (i = 0; (event_waiters) && (i < num_events) ; i++) {
init_wait(&event_waiters[i].wait);