Add UMF_BUILD_OS_MEMORY_PROVIDER option to disable OS memory provider

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

Author: ldorau

.github/workflows/basic.yml

@@ -16,6 +16,7 @@ jobs:
         compiler: [{c: gcc, cxx: g++}]
         pool_tracking: ['ON', 'OFF']
         shared_library: ['OFF']
+        os_provider: ['ON']
         include:
           - os: 'ubuntu-20.04'
             build_type: Release
@@ -29,6 +30,11 @@ jobs:
             build_type: Release
             compiler: {c: gcc, cxx: g++}
             shared_library: 'ON'
+          - os: 'ubuntu-22.04'
+            build_type: Release
+            compiler: {c: gcc, cxx: g++}
+            shared_library: ['ON', 'OFF']
+            os_provider: 'OFF'
     runs-on: ${{matrix.os}}
 
     steps:
@@ -54,6 +60,7 @@ jobs:
           -DUMF_BUILD_SHARED_LIBRARY=${{matrix.shared_library}}
           -DCMAKE_C_COMPILER=${{matrix.compiler.c}}
           -DCMAKE_CXX_COMPILER=${{matrix.compiler.cxx}}
+          -DUMF_BUILD_OS_MEMORY_PROVIDER=${{matrix.os_provider}}
           -DUMF_ENABLE_POOL_TRACKING=${{matrix.pool_tracking}}
           -DUMF_FORMAT_CODE_STYLE=OFF
           -DUMF_DEVELOPER_MODE=ON

CMakeLists.txt

@@ -17,6 +17,7 @@ include(helpers)
 
 # Build Options
 option(UMF_BUILD_SHARED_LIBRARY "Build UMF as shared library" OFF)
+option(UMF_BUILD_OS_MEMORY_PROVIDER "Build OS memory provider" ON)
 option(UMF_BUILD_LIBUMF_POOL_DISJOINT "Build the libumf_pool_disjoint static library" OFF)
 option(UMF_BUILD_LIBUMF_POOL_JEMALLOC "Build the libumf_pool_jemalloc static library" OFF)
 option(UMF_BUILD_LIBUMF_POOL_SCALABLE "Build the libumf_pool_scalable static library" OFF)

README.md

@@ -22,6 +22,10 @@ UMF comes with predefined pool allocators (see include/pool) and providers (see
 
 A memory provider that provides memory from an operating system.
 
+#### Requirements
+
+To enable this feature, the `UMF_BUILD_OS_MEMORY_PROVIDER` option needs to be turned `ON` (default: `ON`).
+
 ## Memory pool managers
 
 ### libumf_pool_disjoint
@@ -115,6 +119,7 @@ List of options provided by CMake:
 | Name | Description | Values | Default |
 | - | - | - | - |
 | UMF_BUILD_SHARED_LIBRARY | Build UMF as shared library | ON/OFF | OFF |
+| UMF_BUILD_OS_MEMORY_PROVIDER | Build OS memory provider | ON/OFF | ON |
 | UMF_BUILD_LIBUMF_POOL_DISJOINT | Build the libumf_pool_disjoint static library | ON/OFF | OFF |
 | UMF_BUILD_LIBUMF_POOL_JEMALLOC | Build the libumf_pool_jemalloc static library | ON/OFF | OFF |
 | UMF_BUILD_LIBUMF_POOL_SCALABLE | Build the libumf_pool_scalable static library | ON/OFF | OFF |

benchmark/CMakeLists.txt

@@ -32,6 +32,10 @@ target_link_libraries(ubench
 	pthread
 	m)
 
+if (UMF_BUILD_OS_MEMORY_PROVIDER)
+	target_compile_definitions(ubench PRIVATE UMF_BUILD_OS_MEMORY_PROVIDER=1)
+endif()
+
 if (UMF_BUILD_LIBUMF_POOL_DISJOINT)
 	target_compile_definitions(ubench PRIVATE UMF_BUILD_LIBUMF_POOL_DISJOINT=1)
 endif()

benchmark/ubench.c

@@ -93,6 +93,7 @@ UBENCH_EX(simple, glibc_malloc) {
     free(array);
 }
 
+#ifdef UMF_BUILD_OS_MEMORY_PROVIDER
 ////////////////// OS MEMORY PROVIDER
 
 static umf_os_memory_provider_params_t UMF_OS_MEMORY_PROVIDER_PARAMS = {
@@ -156,10 +157,12 @@ UBENCH_EX(simple, os_memory_provider) {
     umfMemoryProviderDestroy(os_memory_provider);
     free(array);
 }
+#endif /* UMF_BUILD_OS_MEMORY_PROVIDER */
 
-#if (defined UMF_BUILD_LIBUMF_POOL_DISJOINT) ||                                \
-    (defined UMF_BUILD_LIBUMF_POOL_JEMALLOC) ||                                \
-    (defined UMF_BUILD_LIBUMF_POOL_SCALABLE)
+#if (defined UMF_BUILD_OS_MEMORY_PROVIDER) &&                                  \
+    ((defined UMF_BUILD_LIBUMF_POOL_DISJOINT) ||                               \
+     (defined UMF_BUILD_LIBUMF_POOL_JEMALLOC) ||                               \
+     (defined UMF_BUILD_LIBUMF_POOL_SCALABLE))
 
 static void *w_umfPoolMalloc(void *provider, size_t size, size_t alignment) {
     umf_memory_pool_handle_t hPool = (umf_memory_pool_handle_t)provider;
@@ -174,9 +177,10 @@ static void w_umfPoolFree(void *provider, void *ptr, size_t size) {
         exit(-1);
     }
 }
-#endif /* (defined UMF_BUILD_LIBUMF_POOL_DISJOINT) || (defined UMF_BUILD_LIBUMF_POOL_JEMALLOC) || (defined UMF_BUILD_LIBUMF_POOL_SCALABLE) */
+#endif /* (defined UMF_BUILD_OS_MEMORY_PROVIDER) && ((defined UMF_BUILD_LIBUMF_POOL_DISJOINT) || (defined UMF_BUILD_LIBUMF_POOL_JEMALLOC) || (defined UMF_BUILD_LIBUMF_POOL_SCALABLE)) */
 
-#ifdef UMF_BUILD_LIBUMF_POOL_DISJOINT
+#if (defined UMF_BUILD_LIBUMF_POOL_DISJOINT) &&                                \
+    (defined UMF_BUILD_OS_MEMORY_PROVIDER)
 ////////////////// DISJOINT POOL WITH OS MEMORY PROVIDER
 
 UBENCH_EX(simple, disjoint_pool_with_os_memory_provider) {
@@ -218,9 +222,10 @@ UBENCH_EX(simple, disjoint_pool_with_os_memory_provider) {
     umfMemoryProviderDestroy(os_memory_provider);
     free(array);
 }
-#endif /* UMF_BUILD_LIBUMF_POOL_DISJOINT */
+#endif /* (defined UMF_BUILD_LIBUMF_POOL_DISJOINT) && (defined UMF_BUILD_OS_MEMORY_PROVIDER) */
 
-#ifdef UMF_BUILD_LIBUMF_POOL_JEMALLOC
+#if (defined UMF_BUILD_LIBUMF_POOL_JEMALLOC) &&                                \
+    (defined UMF_BUILD_OS_MEMORY_PROVIDER)
 ////////////////// JEMALLOC POOL WITH OS MEMORY PROVIDER
 
 UBENCH_EX(simple, jemalloc_pool_with_os_memory_provider) {
@@ -254,9 +259,10 @@ UBENCH_EX(simple, jemalloc_pool_with_os_memory_provider) {
     umfMemoryProviderDestroy(os_memory_provider);
     free(array);
 }
-#endif /* UMF_BUILD_LIBUMF_POOL_JEMALLOC */
+#endif /* (defined UMF_BUILD_LIBUMF_POOL_JEMALLOC) && (defined UMF_BUILD_OS_MEMORY_PROVIDER) */
 
-#ifdef UMF_BUILD_LIBUMF_POOL_SCALABLE
+#if (defined UMF_BUILD_LIBUMF_POOL_SCALABLE) &&                                \
+    (defined UMF_BUILD_OS_MEMORY_PROVIDER)
 ////////////////// SCALABLE (TBB) POOL WITH OS MEMORY PROVIDER
 
 UBENCH_EX(simple, scalable_pool_with_os_memory_provider) {
@@ -290,6 +296,6 @@ UBENCH_EX(simple, scalable_pool_with_os_memory_provider) {
     umfMemoryProviderDestroy(os_memory_provider);
     free(array);
 }
-#endif /* UMF_BUILD_LIBUMF_POOL_SCALABLE */
+#endif /* (defined UMF_BUILD_LIBUMF_POOL_SCALABLE) && (defined UMF_BUILD_OS_MEMORY_PROVIDER) */
 
 UBENCH_MAIN();

src/CMakeLists.txt

@@ -10,20 +10,20 @@ set(UMF_SOURCES
     memory_pool.c
     memory_provider.c
     memory_provider_get_last_failed.c
-    provider/provider_os_memory.c
     provider/provider_tracking.c
     critnib/critnib.c
 )
 
-set(UMF_SOURCES_LINUX
-    provider/provider_os_memory_linux.c
-)
-
 set(UMF_SOURCES_WINDOWS
-    provider/provider_os_memory_windows.c
     provider/provider_tracking_windows.cpp
 )
 
+if(UMF_BUILD_OS_MEMORY_PROVIDER)
+    set(UMF_SOURCES ${UMF_SOURCES} provider/provider_os_memory.c)
+    set(UMF_SOURCES_LINUX ${UMF_SOURCES_LINUX} provider/provider_os_memory_linux.c)
+    set(UMF_SOURCES_WINDOWS ${UMF_SOURCES_WINDOWS} provider/provider_os_memory_windows.c)
+endif()
+
 if(LINUX)
     set(UMF_SOURCES ${UMF_SOURCES} ${UMF_SOURCES_LINUX})
 elseif(WINDOWS)

test/CMakeLists.txt

@@ -78,19 +78,27 @@ if(UMF_BUILD_LIBUMF_POOL_DISJOINT)
                  LIBS disjoint_pool)
 endif()
 
-if(UMF_BUILD_LIBUMF_POOL_JEMALLOC)
-    add_umf_test(NAME jemalloc_pool
-                 SRCS pools/jemalloc_pool.cpp malloc_compliance_tests.cpp
-                 LIBS jemalloc_pool)
+if(UMF_BUILD_OS_MEMORY_PROVIDER)
+    if(UMF_BUILD_LIBUMF_POOL_JEMALLOC)
+        add_umf_test(NAME jemalloc_pool
+                    SRCS pools/jemalloc_pool.cpp malloc_compliance_tests.cpp
+                    LIBS jemalloc_pool)
+    endif()
+else()
+    message(STATUS "The jemalloc_pool test is DISABLED, because UMF_BUILD_OS_MEMORY_PROVIDER is set to OFF")
 endif()
 
-if(UMF_BUILD_LIBUMF_POOL_SCALABLE)
-    add_umf_test(NAME scalable_pool
-                 SRCS pools/scalable_pool.cpp malloc_compliance_tests.cpp
-                 LIBS scalable_pool)
+if(UMF_BUILD_OS_MEMORY_PROVIDER)
+    if(UMF_BUILD_LIBUMF_POOL_SCALABLE)
+        add_umf_test(NAME scalable_pool
+                    SRCS pools/scalable_pool.cpp malloc_compliance_tests.cpp
+                    LIBS scalable_pool)
+    endif()
+else()
+    message(STATUS "The scalable_pool test is DISABLED, because UMF_BUILD_OS_MEMORY_PROVIDER is set to OFF")
 endif()
 
-if(LINUX) # OS-specific functions are implemented only for Linux now
+if(UMF_BUILD_OS_MEMORY_PROVIDER AND LINUX) # OS-specific functions are implemented only for Linux now
     add_umf_test(NAME provider_os_memory
                  SRCS provider_os_memory.cpp
                  LIBS umf_utils)