Revert "Add lock to umfMemoryTrackerAdd() and umfMemoryTrackerRemove()"

This reverts commit 33d8b73.

Author: ldorau

src/provider/provider_tracking.c

@@ -37,7 +37,7 @@ struct umf_memory_tracker_t {
     // when one memory pool acts as a memory provider
     // for another memory pool (nested memory pooling).
     critnib *alloc_segments_map[MAX_LEVELS_OF_ALLOC_SEGMENT_MAP];
-    utils_mutex_t mutex;
+    utils_mutex_t splitMergeMutex;
 };
 
 typedef struct tracker_alloc_info_t {
@@ -179,11 +179,6 @@ static umf_result_t umfMemoryTrackerAdd(umf_memory_tracker_handle_t hTracker,
     int level = 0;
     int found = 0;
 
-    int ret = utils_mutex_lock(&hTracker->mutex);
-    if (ret) {
-        return UMF_RESULT_ERROR_UNKNOWN;
-    }
-
     // Find the most nested (in the highest level) entry
     // in the critnib maps that contains the given 'ptr' pointer.
     do {
@@ -202,8 +197,7 @@ static umf_result_t umfMemoryTrackerAdd(umf_memory_tracker_handle_t hTracker,
                 // TODO: we need to support an arbitrary amount of layers in the future
                 LOG_ERR("tracker level is too high, ptr=%p, size=%zu", ptr,
                         size);
-                umf_result = UMF_RESULT_ERROR_OUT_OF_RESOURCES;
-                goto err_unlock;
+                return UMF_RESULT_ERROR_OUT_OF_RESOURCES;
             }
             if (((uintptr_t)ptr + size) > (rkey + rsize)) {
                 LOG_ERR(
@@ -212,8 +206,7 @@ static umf_result_t umfMemoryTrackerAdd(umf_memory_tracker_handle_t hTracker,
                     "that exceeds the parent value (pool=%p, ptr=%p, size=%zu)",
                     (void *)pool, ptr, size, (void *)rvalue->pool, (void *)rkey,
                     (size_t)rsize);
-                umf_result = UMF_RESULT_ERROR_INVALID_ARGUMENT;
-                goto err_unlock;
+                return UMF_RESULT_ERROR_INVALID_ARGUMENT;
             }
             parent_key = rkey;
             parent_value = rvalue;
@@ -224,15 +217,10 @@ static umf_result_t umfMemoryTrackerAdd(umf_memory_tracker_handle_t hTracker,
     umf_result = umfMemoryTrackerAddAtLevel(hTracker, level, pool, ptr, size,
                                             parent_key, parent_value);
     if (umf_result != UMF_RESULT_SUCCESS) {
-        goto err_unlock;
+        return umf_result;
     }
 
-    umf_result = UMF_RESULT_SUCCESS;
-
-err_unlock:
-    utils_mutex_unlock(&hTracker->mutex);
-
-    return umf_result;
+    return UMF_RESULT_SUCCESS;
 }
 
 static umf_result_t umfMemoryTrackerRemove(umf_memory_tracker_handle_t hTracker,
@@ -244,23 +232,17 @@ static umf_result_t umfMemoryTrackerRemove(umf_memory_tracker_handle_t hTracker,
     // Every umfMemoryTrackerAdd(..., ptr, ...) should have a corresponding
     // umfMemoryTrackerRemove call with the same ptr value.
 
-    umf_result_t umf_result = UMF_RESULT_ERROR_UNKNOWN;
     tracker_alloc_info_t *parent_value = NULL;
     uintptr_t parent_key = 0;
     int level = 0;
 
-    int ret = utils_mutex_lock(&hTracker->mutex);
-    if (ret) {
-        return UMF_RESULT_ERROR_UNKNOWN;
-    }
-
     // Find the most nested (on the highest level) entry in the map
     // with the `ptr` key and with no children - only such entry can be removed.
     tracker_alloc_info_t *value = get_most_nested_alloc_segment(
         hTracker, ptr, &level, &parent_key, &parent_value, 1 /* no_children */);
     if (!value) {
         LOG_ERR("pointer %p not found in the alloc_segments_map", ptr);
-        goto err_unlock;
+        return UMF_RESULT_ERROR_UNKNOWN;
     }
 
     assert(level < MAX_LEVELS_OF_ALLOC_SEGMENT_MAP);
@@ -282,12 +264,7 @@ static umf_result_t umfMemoryTrackerRemove(umf_memory_tracker_handle_t hTracker,
 
     umf_ba_free(hTracker->alloc_info_allocator, value);
 
-    umf_result = UMF_RESULT_SUCCESS;
-
-err_unlock:
-    utils_mutex_unlock(&hTracker->mutex);
-
-    return umf_result;
+    return UMF_RESULT_SUCCESS;
 }
 
 umf_memory_pool_handle_t umfMemoryTrackerGetPool(const void *ptr) {
@@ -427,7 +404,7 @@ static umf_result_t trackingAllocationSplit(void *hProvider, void *ptr,
     splitValue->size = firstSize;
     splitValue->n_children = 0;
 
-    int r = utils_mutex_lock(&provider->hTracker->mutex);
+    int r = utils_mutex_lock(&provider->hTracker->splitMergeMutex);
     if (r) {
         goto err_lock;
     }
@@ -493,7 +470,7 @@ static umf_result_t trackingAllocationSplit(void *hProvider, void *ptr,
 
     // free the original value
     umf_ba_free(provider->hTracker->alloc_info_allocator, value);
-    utils_mutex_unlock(&provider->hTracker->mutex);
+    utils_mutex_unlock(&provider->hTracker->splitMergeMutex);
 
     LOG_DEBUG(
         "split memory region (level=%i): ptr=%p, totalSize=%zu, firstSize=%zu",
@@ -502,7 +479,7 @@ static umf_result_t trackingAllocationSplit(void *hProvider, void *ptr,
     return UMF_RESULT_SUCCESS;
 
 err:
-    utils_mutex_unlock(&provider->hTracker->mutex);
+    utils_mutex_unlock(&provider->hTracker->splitMergeMutex);
 err_lock:
     umf_ba_free(provider->hTracker->alloc_info_allocator, splitValue);
 
@@ -534,7 +511,7 @@ static umf_result_t trackingAllocationMerge(void *hProvider, void *lowPtr,
     int lowLevel = -2;
     int highLevel = -1;
 
-    int r = utils_mutex_lock(&provider->hTracker->mutex);
+    int r = utils_mutex_lock(&provider->hTracker->splitMergeMutex);
     if (r) {
         goto err_lock;
     }
@@ -601,7 +578,7 @@ static umf_result_t trackingAllocationMerge(void *hProvider, void *lowPtr,
 
     umf_ba_free(provider->hTracker->alloc_info_allocator, erasedhighValue);
 
-    utils_mutex_unlock(&provider->hTracker->mutex);
+    utils_mutex_unlock(&provider->hTracker->splitMergeMutex);
 
     LOG_DEBUG("merged memory regions (level=%i): lowPtr=%p (child=%zu), "
               "highPtr=%p (child=%zu), totalSize=%zu",
@@ -616,7 +593,7 @@ static umf_result_t trackingAllocationMerge(void *hProvider, void *lowPtr,
     assert(0);
 
 not_merged:
-    utils_mutex_unlock(&provider->hTracker->mutex);
+    utils_mutex_unlock(&provider->hTracker->splitMergeMutex);
 
 err_lock:
     umf_ba_free(provider->hTracker->alloc_info_allocator, mergedValue);
@@ -1100,7 +1077,7 @@ umf_memory_tracker_handle_t umfMemoryTrackerCreate(void) {
 
     handle->alloc_info_allocator = alloc_info_allocator;
 
-    void *mutex_ptr = utils_mutex_init(&handle->mutex);
+    void *mutex_ptr = utils_mutex_init(&handle->splitMergeMutex);
     if (!mutex_ptr) {
         goto err_destroy_alloc_info_allocator;
     }
@@ -1124,7 +1101,7 @@ umf_memory_tracker_handle_t umfMemoryTrackerCreate(void) {
             critnib_delete(handle->alloc_segments_map[j]);
         }
     }
-    utils_mutex_destroy_not_free(&handle->mutex);
+    utils_mutex_destroy_not_free(&handle->splitMergeMutex);
 err_destroy_alloc_info_allocator:
     umf_ba_destroy(alloc_info_allocator);
 err_free_handle:
@@ -1157,7 +1134,7 @@ void umfMemoryTrackerDestroy(umf_memory_tracker_handle_t handle) {
             handle->alloc_segments_map[i] = NULL;
         }
     }
-    utils_mutex_destroy_not_free(&handle->mutex);
+    utils_mutex_destroy_not_free(&handle->splitMergeMutex);
     umf_ba_destroy(handle->alloc_info_allocator);
     handle->alloc_info_allocator = NULL;
     umf_ba_global_free(handle);