From 95d4d316485ad509fd7e440a8d45daf8c3ce1f04 Mon Sep 17 00:00:00 2001 From: David Yat Sin Date: Tue, 23 Aug 2022 11:34:19 -0400 Subject: [PATCH] libhsakmt: Add check for invalid aperture Avoiding segfault when an invalid SharedMemoryHandle is passed in when calling fmm_register_shared_memory. Change-Id: I0e0bbed01487fc10afcbb170eb9330e70b209d14 Signed-off-by: David Yat Sin [ROCm/ROCR-Runtime commit: 1c385fb257c8318384d5065603e6391b1607be59] --- projects/rocr-runtime/src/fmm.c | 2 ++ 1 file changed, 2 insertions(+) diff --git a/projects/rocr-runtime/src/fmm.c b/projects/rocr-runtime/src/fmm.c index 0dea2a8c4c..cf6c2e36c7 100644 --- a/projects/rocr-runtime/src/fmm.c +++ b/projects/rocr-runtime/src/fmm.c @@ -3411,6 +3411,8 @@ HSAKMT_STATUS fmm_register_shared_memory(const HsaSharedMemoryHandle *SharedMemo importArgs.gpu_id = SharedMemoryStruct->ExportGpuId; aperture = fmm_get_aperture(SharedMemoryStruct->ApeInfo); + if (!aperture) + return HSAKMT_STATUS_INVALID_PARAMETER; pthread_mutex_lock(&aperture->fmm_mutex); reservedMem = aperture_allocate_area(aperture, NULL,