Fix asan's user-poisoning flags bug

It add utils_annotate_memory_defined() (which does unpoison on memory region)
after all mmap().
This should be unnecessary change but pairs of mmap/munmap do not reset
asan's user-poisoning flags, leading to invalid error reports.
This bug is describe here - 81619: https://gcc.gnu.org/bugzilla/show_bug.cgi?id=81619

Author: Adam

.github/workflows/sanitizers.yml

@@ -72,9 +72,7 @@ jobs:
       working-directory: ${{env.BUILD_DIR}}
       run: > 
         ${{ matrix.compiler.cxx == 'icpx' && '. /opt/intel/oneapi/setvars.sh &&' || ''}}
-        GTEST_FILTER="-*umfProviderTest.alloc_page64_align_0*" ctest --output-on-failure
-      # TO DO: fix umf-provider_os_memory test for sanitizers
-      # issue 581: https://github.com/oneapi-src/unified-memory-framework/issues/581
+        ctest --output-on-failure
 
   windows-build:
     name: cl and clang-cl on Windows

src/base_alloc/base_alloc_global.c

@@ -195,14 +195,12 @@ void umf_ba_global_free(void *ptr) {
 
     int ac_index = size_to_idx(total_size);
     if (ac_index >= NUM_ALLOCATION_CLASSES) {
-        utils_annotate_memory_inaccessible(ptr, total_size);
         ba_os_free(ptr, total_size);
         return;
     }
 
     if (!BASE_ALLOC.ac[ac_index]) {
         // if creating ac failed, memory must have been allocated by os
-        utils_annotate_memory_inaccessible(ptr, total_size);
         ba_os_free(ptr, total_size);
         return;
     }

src/base_alloc/base_alloc_linux.c

@@ -19,11 +19,17 @@ static UTIL_ONCE_FLAG Page_size_is_initialized = UTIL_ONCE_FLAG_INIT;
 static size_t Page_size;
 
 void *ba_os_alloc(size_t size) {
-    return mmap(NULL, size, PROT_READ | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS,
-                -1, 0);
+    void *ptr = mmap(NULL, size, PROT_READ | PROT_WRITE,
+                     MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
+    // this should be unnecessary but pairs of mmap/munmap do not reset
+    // asan's user-poisoning flags, leading to invalid error reports
+    // Bug 81619: https://gcc.gnu.org/bugzilla/show_bug.cgi?id=81619
+    utils_annotate_memory_defined(ptr, size);
+    return ptr;
 }
 
 void ba_os_free(void *ptr, size_t size) {
+    utils_annotate_memory_inaccessible(ptr, size);
     int ret = munmap(ptr, size);
     assert(ret == 0);
     (void)ret; // unused

src/provider/provider_os_memory_posix.c

@@ -16,6 +16,7 @@
 
 #include "provider_os_memory_internal.h"
 #include "utils_log.h"
+#include "utils_sanitizers.h"
 
 // maximum value of the off_t type
 #define OFF_T_MAX                                                              \
@@ -74,11 +75,17 @@ void *os_mmap(void *hint_addr, size_t length, int prot, int flag, int fd,
     if (ptr == MAP_FAILED) {
         return NULL;
     }
-
+    // this should be unnecessary but pairs of mmap/munmap do not reset
+    // asan's user-poisoning flags, leading to invalid error reports
+    // Bug 81619: https://gcc.gnu.org/bugzilla/show_bug.cgi?id=81619
+    utils_annotate_memory_defined(ptr, length);
     return ptr;
 }
 
-int os_munmap(void *addr, size_t length) { return munmap(addr, length); }
+int os_munmap(void *addr, size_t length) {
+    utils_annotate_memory_inaccessible(addr, length);
+    return munmap(addr, length);
+}
 
 size_t os_get_page_size(void) { return sysconf(_SC_PAGE_SIZE); }