Assert if tracking provider is not empty

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

Author: ldorau

src/provider/provider_tracking.c

@@ -322,7 +322,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) {
@@ -404,3 +434,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 */
+
+    umfMemoryTrackerDestroy((umf_memory_tracker_handle_t)tracker);
+}

src/provider/provider_tracking.h

@@ -78,6 +78,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

@@ -17,7 +17,7 @@ void __attribute__((constructor)) createLibTracker(void) {
     TRACKER = umfMemoryTrackerCreate();
 }
 void __attribute__((destructor)) deleteLibTracker(void) {
-    umfMemoryTrackerDestroy(TRACKER);
+    umfTrackingMemoryProviderFini(TRACKER);
 }
 
 void umfTrackingMemoryProviderInit(void) {

src/provider/provider_tracking_windows.c

@@ -14,10 +14,12 @@
 
 static umf_memory_tracker_handle_t TRACKER = NULL;
 
+static void providerFini(void) { umfTrackingMemoryProviderFini(TRACKER); }
+
 #if defined(UMF_SHARED_LIBRARY)
 BOOL APIENTRY DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved) {
     if (fdwReason == DLL_PROCESS_DETACH) {
-        umfMemoryTrackerDestroy(TRACKER);
+        providerFini();
     } else if (fdwReason == DLL_PROCESS_ATTACH) {
         TRACKER = umfMemoryTrackerCreate();
     }
@@ -30,8 +32,6 @@ void umfTrackingMemoryProviderInit(void) {
 #else
 INIT_ONCE init_once_flag = INIT_ONCE_STATIC_INIT;
 
-static void providerFini(void) { umfMemoryTrackerDestroy(TRACKER); }
-
 BOOL CALLBACK providerInit(PINIT_ONCE InitOnce, PVOID Parameter,
                            PVOID *lpContext) {
     TRACKER = umfMemoryTrackerCreate();