do not destroy proxy lib pool before umf lib

Author: bratpiorka

src/utils/utils_common.h

@@ -14,10 +14,14 @@
 #include <stdbool.h>
 #include <stddef.h>
 #include <stdint.h>
+#include <stdio.h>
+#include <string.h>
 
 #include <umf/base.h>
 #include <umf/memory_provider.h>
 
+#include "utils_load_library.h"
+
 #ifdef __cplusplus
 extern "C" {
 #endif
@@ -79,6 +83,23 @@ static inline int utils_env_var_has_str(const char *envvar, const char *str) {
 
 // check if we are running in the proxy library
 static inline int utils_is_running_in_proxy_lib(void) {
+    // check if the proxy library is loaded using /proc/self/maps
+    FILE *fp = fopen("/proc/self/maps", "r");
+    if (!fp) {
+        perror("Failed to open /proc/self/maps");
+        return -1;
+    }
+
+    char line[256];
+    while (fgets(line, sizeof(line), fp)) {
+        if (strstr(line, "libumf_proxy.so") != NULL) {
+            fclose(fp);
+            return 1; // Shared object is loaded
+        }
+    }
+    fclose(fp);
+
+    // check if the proxy library is loaded using LD_PRELOAD
     return utils_env_var_get_str("LD_PRELOAD", "libumf_proxy.so") ? 1 : 0;
 }
 

src/utils/utils_load_library.c

@@ -71,18 +71,31 @@ void *utils_open_library(const char *filename, int userFlags) {
         dlopenFlags |= RTLD_GLOBAL;
     }
     if (userFlags & UMF_UTIL_OPEN_LIBRARY_NO_LOAD) {
-        dlopenFlags |= RTLD_NOLOAD;
+        dlopenFlags = RTLD_NOLOAD | RTLD_NOW;
     }
 
     void *handle = dlopen(filename, dlopenFlags);
     if (handle == NULL) {
+        if (userFlags & UMF_UTIL_OPEN_LIBRARY_NO_LOAD) {
+            // if we are not loading the library, just return NULL if it is not
+            // found
+            LOG_DEBUG("Library %s not found, returning NULL", filename);
+            return NULL;
+        }
+
         LOG_FATAL("dlopen(%s) failed with error: %s", filename, dlerror());
+        return NULL;
     }
 
+    LOG_DEBUG("Opened library %s with handle %p", filename, handle);
     return handle;
 }
 
-int utils_close_library(void *handle) { return dlclose(handle); }
+int utils_close_library(void *handle) {
+    LOG_DEBUG("Closing library handle %p", handle);
+
+    return dlclose(handle);
+}
 
 void *utils_get_symbol_addr(void *handle, const char *symbol,
                             const char *libname) {