Add IPC test for devdax memory provider

Signed-off-by: Lukasz Dorau <[email protected]>

Author: ldorau

test/CMakeLists.txt

@@ -356,6 +356,36 @@ if(LINUX)
         add_umf_ipc_test(TEST ipc_os_prov_anon_fd)
         add_umf_ipc_test(TEST ipc_os_prov_shm)
     endif()
+    if(UMF_TESTS_DEVDAX_PATH AND UMF_TESTS_DEVDAX_SIZE)
+        build_umf_test(
+            NAME
+            ipc_devdax_prov_consumer
+            SRCS
+            ipc_devdax_prov_consumer.c
+            common/ipc_common.c
+            common/ipc_os_prov_common.c)
+        build_umf_test(
+            NAME
+            ipc_devdax_prov_producer
+            SRCS
+            ipc_devdax_prov_producer.c
+            common/ipc_common.c
+            common/ipc_os_prov_common.c)
+        target_compile_definitions(
+            umf_test-ipc_devdax_prov_consumer
+            PRIVATE UMF_TESTS_DEVDAX_PATH="${UMF_TESTS_DEVDAX_PATH}"
+                    UMF_TESTS_DEVDAX_SIZE=${UMF_TESTS_DEVDAX_SIZE})
+        target_compile_definitions(
+            umf_test-ipc_devdax_prov_producer
+            PRIVATE UMF_TESTS_DEVDAX_PATH="${UMF_TESTS_DEVDAX_PATH}"
+                    UMF_TESTS_DEVDAX_SIZE=${UMF_TESTS_DEVDAX_SIZE})
+        add_umf_ipc_test(TEST ipc_devdax_prov)
+    else()
+        message(
+            STATUS
+                "UMF_TESTS_DEVDAX_PATH or UMF_TESTS_DEVDAX_SIZE is not set - skipping the IPC devdax memory provider test"
+        )
+    endif()
     if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_LEVEL_ZERO_PROVIDER)
         build_umf_test(
             NAME

test/ipc_devdax_prov.sh

@@ -0,0 +1,24 @@
+#
+# Copyright (C) 2024 Intel Corporation
+#
+# Under the Apache License v2.0 with LLVM Exceptions. See LICENSE.TXT.
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+#
+
+#!/bin/bash
+
+set -e
+
+# port should be a number from the range <1024, 65535>
+PORT=$(( 1024 + ( $$ % ( 65535 - 1024 ))))
+
+UMF_LOG_VAL="level:debug;flush:debug;output:stderr;pid:yes"
+
+echo "Starting ipc_devdax_prov CONSUMER on port $PORT ..."
+UMF_LOG=$UMF_LOG_VAL ./umf_test-ipc_devdax_prov_consumer $PORT &
+
+echo "Waiting 1 sec ..."
+sleep 1
+
+echo "Starting ipc_devdax_prov PRODUCER on port $PORT ..."
+UMF_LOG=$UMF_LOG_VAL ./umf_test-ipc_devdax_prov_producer $PORT

test/ipc_devdax_prov_consumer.c

@@ -0,0 +1,30 @@
+/*
+ * Copyright (C) 2024 Intel Corporation
+ *
+ * Under the Apache License v2.0 with LLVM Exceptions. See LICENSE.TXT.
+ * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+
+#include <umf/providers/provider_devdax_memory.h>
+
+#include "ipc_common.h"
+#include "ipc_os_prov_common.h"
+
+int main(int argc, char *argv[]) {
+    if (argc < 2) {
+        fprintf(stderr, "usage: %s <port>\n", argv[0]);
+        return -1;
+    }
+
+    int port = atoi(argv[1]);
+
+    umf_devdax_memory_provider_params_t devdax_params =
+        umfDevDaxMemoryProviderParamsDefault(UMF_TESTS_DEVDAX_PATH,
+                                             UMF_TESTS_DEVDAX_SIZE);
+
+    return run_consumer(port, umfDevDaxMemoryProviderOps(), &devdax_params,
+                        memcopy, NULL);
+}

test/ipc_devdax_prov_producer.c

@@ -0,0 +1,30 @@
+/*
+ * Copyright (C) 2024 Intel Corporation
+ *
+ * Under the Apache License v2.0 with LLVM Exceptions. See LICENSE.TXT.
+ * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+
+#include <umf/providers/provider_devdax_memory.h>
+
+#include "ipc_common.h"
+#include "ipc_os_prov_common.h"
+
+int main(int argc, char *argv[]) {
+    if (argc < 2) {
+        fprintf(stderr, "usage: %s <port>\n", argv[0]);
+        return -1;
+    }
+
+    int port = atoi(argv[1]);
+
+    umf_devdax_memory_provider_params_t devdax_params =
+        umfDevDaxMemoryProviderParamsDefault(UMF_TESTS_DEVDAX_PATH,
+                                             UMF_TESTS_DEVDAX_SIZE);
+
+    return run_producer(port, umfDevDaxMemoryProviderOps(), &devdax_params,
+                        memcopy, NULL);
+}