Assert if tracking provider is not empty

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

Author: ldorau

src/provider/provider_tracking.c

@@ -16,6 +16,7 @@
 
 #include <assert.h>
 #include <errno.h>
+#include <stdio.h>
 #include <stdlib.h>
 
 typedef struct tracker_value_t {
@@ -160,7 +161,37 @@ static umf_result_t trackingInitialize(void *params, void **ret) {
     return UMF_RESULT_SUCCESS;
 }
 
-static void trackingFinalize(void *provider) { free(provider); }
+static void trackingFinalize(void *provider) {
+#ifndef NDEBUG
+    umf_tracking_memory_provider_t *p =
+        (umf_tracking_memory_provider_t *)provider;
+
+    uintptr_t rkey;
+    void *rvalue;
+    size_t n_items = 0;
+    uintptr_t last_key = 0;
+
+    while (1 == critnib_find((critnib *)p->hTracker, last_key, FIND_G, &rkey,
+                             &rvalue)) {
+        tracker_value_t *value = (tracker_value_t *)rvalue;
+        if (value->pool == p->pool) {
+            n_items++;
+        }
+
+        last_key = rkey;
+    }
+
+    if (n_items) {
+        fprintf(stderr,
+                "ASSERT: tracking provider of pool %p is not empty! (%zu items "
+                "left)\n",
+                (void *)p->pool, n_items);
+        assert(n_items == 0);
+    }
+#endif /* NDEBUG */
+
+    free(provider);
+}
 
 static void trackingGetLastError(void *provider, const char **msg,
                                  int32_t *pError) {
@@ -241,3 +272,27 @@ void umfTrackingMemoryProviderGetUpstreamProvider(
         (umf_tracking_memory_provider_t *)hTrackingProvider;
     *hUpstream = p->hUpstream;
 }
+
+void umfTrackingMemoryProviderFini(void *tracker) {
+#ifndef NDEBUG
+    uintptr_t rkey;
+    void *rvalue;
+    size_t n_items = 0;
+    uintptr_t last_key = 0;
+
+    while (1 ==
+           critnib_find((critnib *)tracker, last_key, FIND_G, &rkey, &rvalue)) {
+        n_items++;
+        last_key = rkey;
+    }
+
+    if (n_items) {
+        fprintf(stderr,
+                "ASSERT: tracking provider is not empty! (%zu items left)\n",
+                n_items);
+        assert(n_items == 0);
+    }
+#endif /* NDEBUG */
+
+    critnib_delete((critnib *)tracker);
+}

src/provider/provider_tracking.h

@@ -35,6 +35,8 @@ umf_result_t umfTrackingMemoryProviderCreate(
 // Initialize critnib for a UMF static library build on Windows
 void umfTrackingMemoryProviderInit(void);
 
+void umfTrackingMemoryProviderFini(void *tracker);
+
 void umfTrackingMemoryProviderGetUpstreamProvider(
     umf_memory_provider_handle_t hTrackingProvider,
     umf_memory_provider_handle_t *hUpstream);

src/provider/provider_tracking_linux.c

@@ -18,7 +18,7 @@ void __attribute__((constructor)) createLibTracker(void) {
     TRACKER = critnib_new();
 }
 void __attribute__((destructor)) deleteLibTracker(void) {
-    critnib_delete(TRACKER);
+    umfTrackingMemoryProviderFini(TRACKER);
 }
 
 void umfTrackingMemoryProviderInit(void) {

src/provider/provider_tracking_windows.c

@@ -17,7 +17,7 @@ static critnib *TRACKER = NULL;
 #if defined(UMF_SHARED_LIBRARY)
 BOOL APIENTRY DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved) {
     if (fdwReason == DLL_PROCESS_DETACH) {
-        critnib_delete(TRACKER);
+        providerFini();
     } else if (fdwReason == DLL_PROCESS_ATTACH) {
         TRACKER = critnib_new();
     }
@@ -30,7 +30,7 @@ void umfTrackingMemoryProviderInit(void) {
 #else
 INIT_ONCE init_once_flag = INIT_ONCE_STATIC_INIT;
 
-static void providerFini(void) { critnib_delete(TRACKER); }
+static void providerFini(void) { umfTrackingMemoryProviderFini(TRACKER); }
 
 BOOL CALLBACK providerInit(PINIT_ONCE InitOnce, PVOID Parameter,
                            PVOID *lpContext) {