fix crash at use-after-free; fix build fail

Author: AllanZyne

source/loader/CMakeLists.txt

@@ -110,14 +110,14 @@ if(UR_ENABLE_SANITIZER)
         ${CMAKE_CURRENT_SOURCE_DIR}/layers/sanitizer/asan_allocator.hpp
         ${CMAKE_CURRENT_SOURCE_DIR}/layers/sanitizer/asan_interceptor.cpp
         ${CMAKE_CURRENT_SOURCE_DIR}/layers/sanitizer/asan_interceptor.hpp
+        ${CMAKE_CURRENT_SOURCE_DIR}/layers/sanitizer/asan_libdevice.hpp
         ${CMAKE_CURRENT_SOURCE_DIR}/layers/sanitizer/asan_quarantine.cpp
         ${CMAKE_CURRENT_SOURCE_DIR}/layers/sanitizer/asan_quarantine.hpp
         ${CMAKE_CURRENT_SOURCE_DIR}/layers/sanitizer/asan_report.cpp
         ${CMAKE_CURRENT_SOURCE_DIR}/layers/sanitizer/asan_report.hpp
         ${CMAKE_CURRENT_SOURCE_DIR}/layers/sanitizer/asan_shadow_setup.cpp
         ${CMAKE_CURRENT_SOURCE_DIR}/layers/sanitizer/asan_shadow_setup.hpp
         ${CMAKE_CURRENT_SOURCE_DIR}/layers/sanitizer/common.hpp
-        ${CMAKE_CURRENT_SOURCE_DIR}/layers/sanitizer/device_sanitizer_report.hpp
         ${CMAKE_CURRENT_SOURCE_DIR}/layers/sanitizer/stacktrace.cpp
         ${CMAKE_CURRENT_SOURCE_DIR}/layers/sanitizer/stacktrace.hpp
         ${CMAKE_CURRENT_SOURCE_DIR}/layers/sanitizer/ur_sanddi.cpp

source/loader/layers/sanitizer/asan_report.cpp

@@ -136,25 +136,24 @@ void ReportUseAfterFree(const DeviceSanitizerReport &Report,
     if (!AllocInfoItOp) {
         context.logger.always("Failed to find which chunck {} is allocated",
                               (void *)Report.Address);
-        return;
+    } else {
+        auto &AllocInfo = (*AllocInfoItOp)->second;
+        if (AllocInfo->Context != Context) {
+            context.logger.always("Failed to find which chunck {} is allocated",
+                                  (void *)Report.Address);
+        }
+        assert(AllocInfo->IsReleased);
+
+        context.logger.always("{} is located inside of {} region [{}, {})",
+                              (void *)Report.Address, ToString(AllocInfo->Type),
+                              (void *)AllocInfo->UserBegin,
+                              (void *)AllocInfo->UserEnd);
+        context.logger.always("allocated here:");
+        AllocInfo->AllocStack.print();
+        context.logger.always("released here:");
+        AllocInfo->ReleaseStack.print();
     }
 
-    auto &AllocInfo = (*AllocInfoItOp)->second;
-    if (AllocInfo->Context != Context) {
-        context.logger.always("Failed to find which chunck {} is allocated",
-                              (void *)Report.Address);
-    }
-    assert(AllocInfo->IsReleased);
-
-    context.logger.always("{} is located inside of {} region [{}, {})",
-                          (void *)Report.Address, ToString(AllocInfo->Type),
-                          (void *)AllocInfo->UserBegin,
-                          (void *)AllocInfo->UserEnd);
-    context.logger.always("allocated here:");
-    AllocInfo->AllocStack.print();
-    context.logger.always("released here:");
-    AllocInfo->ReleaseStack.print();
-
     exit(1);
 }