From a9cbff0a4900607622319b7db1ed80e5659f3a98 Mon Sep 17 00:00:00 2001
From: pghafari <pghafari@amd.com>
Date: Thu, 5 Jan 2023 07:56:33 -0500
Subject: [PATCH] SWDEV-374361, SWDEV-374362, SWDEV-374364, SWDEV-374365 -
 error checking

Change-Id: I0b729976ed794239fa54e1ad2965133b319d2f5d
---
 hipamd/src/hip_memory.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/hipamd/src/hip_memory.cpp b/hipamd/src/hip_memory.cpp
index 5a93909579..681b07860a 100644
--- a/hipamd/src/hip_memory.cpp
+++ b/hipamd/src/hip_memory.cpp
@@ -103,6 +103,11 @@ hipError_t hipImportExternalMemory(
     hipExternalMemory_t* extMem_out,
     const hipExternalMemoryHandleDesc* memHandleDesc) {
   HIP_INIT_API(hipImportExternalMemory, extMem_out, memHandleDesc);
+  if (extMem_out == nullptr || memHandleDesc == nullptr ||
+      (memHandleDesc->flags != 0 && memHandleDesc->flags != hipExternalMemoryDedicated) ||
+      memHandleDesc->size == 0) {
+    HIP_RETURN(hipErrorInvalidValue);
+  }
 
   size_t sizeBytes = memHandleDesc->size;
   amd::Context& amdContext = *hip::getCurrentDevice()->asContext();