SWDEV-543340 - Added Unit_hipEventIpc_shm_cleanup test

projects/hip-tests/catch/unit/event/CMakeLists.txt

@@ -17,6 +17,11 @@ if(HIP_PLATFORM MATCHES "amd")
   set(AMD_SRC
       Unit_hipEventQuery.cc
   )
+  if(UNIX)
+    set(AMD_SRC
+      ${AMD_SRC}
+      Unit_hipEventIpc_shm_cleanup.cc)
+  endif()
   set(TEST_SRC ${TEST_SRC} ${AMD_SRC})
 endif()
 

projects/hip-tests/catch/unit/event/Unit_hipEventIpc_shm_cleanup.cc

@@ -0,0 +1,143 @@
+/*
+Copyright (c) 2025 Advanced Micro Devices, Inc. All rights reserved.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.
+*/
+
+// Test hipEventIpc behavior.
+
+#include <hip_test_checkers.hh>
+#include <hip_test_kernels.hh>
+
+#include <hip_test_common.hh>
+#include <dirent.h>
+#include <string>
+#include <set>
+#include <regex>
+#include <sys/wait.h>
+#include <unistd.h>
+
+/**
+ * Test Description
+ * ------------------------
+ *  - Verify that all shared memory objects for IPC events used internally by HIP are properly
+ *    cleaned up after use and do not leave persistent files in /dev/shm.
+ * Test source
+ * ------------------------
+ *  - unit/event/Unit_hipEventIpc_shm_cleanup.cc
+ * Test requirements
+ * ------------------------
+ *  - HIP_VERSION >= 7.0
+ */
+
+// Helper to get current /dev/shm files that look like HIP IPC
+std::set<std::string> get_hip_ipc_shm_files() {
+  std::set<std::string> result;
+  DIR* dir = opendir("/dev/shm");
+  if (!dir) return result;
+
+  struct dirent* entry;
+  std::regex hip_pattern("^hip_.*");  // Matches typical HIP shm names
+
+  while ((entry = readdir(dir)) != nullptr) {
+    if (std::regex_match(entry->d_name, hip_pattern)) {
+      result.insert(entry->d_name);
+    }
+  }
+  closedir(dir);
+  return result;
+}
+
+TEST_CASE("Unit_hipEventIpc_shm_cleanup") {
+  auto before = get_hip_ipc_shm_files();
+
+  int pipefd[2];
+  REQUIRE(pipe(pipefd) == 0);  // Create pipe: pipefd[0] for reading, pipefd[1] for writing
+
+  pid_t pid = fork();
+  REQUIRE(pid >= 0);  // Ensure fork succeeded
+
+  if (pid == 0) {
+    // === Child Process ===
+    close(pipefd[1]);  // Close write end
+
+    hipIpcEventHandle_t handle;
+    ssize_t total = 0;
+    char* ptr = reinterpret_cast<char*>(&handle);
+    while (total < sizeof(handle)) {
+      ssize_t r = read(pipefd[0], ptr + total, sizeof(handle) - total);
+      if (r <= 0) {
+        perror("Child failed to read from pipe");
+        exit(1);
+      }
+      total += r;
+    }
+    close(pipefd[0]);
+
+    hipEvent_t event;
+    if (hipIpcOpenEventHandle(&event, handle) != hipSuccess) {
+      perror("Child failed to open event handle");
+      exit(1);
+    }
+
+    if (hipEventDestroy(event) != hipSuccess) {
+      perror("Child failed to destroy event");
+      exit(1);
+    }
+
+    exit(0);
+  } else {
+    // === Parent Process ===
+    close(pipefd[0]);  // Close read end
+
+    hipEvent_t event;
+    REQUIRE(hipEventCreateWithFlags(&event, hipEventInterprocess | hipEventDisableTiming) ==
+            hipSuccess);
+
+    hipIpcEventHandle_t ipcHandle;
+    REQUIRE(hipIpcGetEventHandle(&ipcHandle, event) == hipSuccess);
+
+    // Write handle to child
+    ssize_t total = 0;
+    const char* ptr = reinterpret_cast<const char*>(&ipcHandle);
+    while (total < sizeof(ipcHandle)) {
+      ssize_t w = write(pipefd[1], ptr + total, sizeof(ipcHandle) - total);
+      if (w <= 0) {
+        perror("Parent failed to write to pipe");
+        exit(1);
+      }
+      total += w;
+    }
+    close(pipefd[1]);
+
+    // Wait for child
+    int status;
+    waitpid(pid, &status, 0);
+    REQUIRE(WIFEXITED(status));
+    REQUIRE(WEXITSTATUS(status) == 0);
+
+    REQUIRE(hipEventDestroy(event) == hipSuccess);
+
+    auto after = get_hip_ipc_shm_files();
+    std::cout << before.size() << after.size() << std::endl;
+    for (const auto& file : after) {
+      REQUIRE(before.count(file) == 1);  // No new HIP shm files left behind
+    }
+  }
+}