Put linux specific code into linux/san_utils.cpp

Author: zhaomaosu

source/loader/layers/sanitizer/asan_interceptor.cpp

@@ -14,8 +14,6 @@
 #include "asan_interceptor.hpp"
 #include "ur_sanitizer_layer.hpp"
 
-#include <dlfcn.h>
-
 namespace ur_sanitizer_layer {
 
 namespace {
@@ -365,20 +363,7 @@ ur_result_t SanitizerInterceptor::enqueueMemSetShadow(
 
         // Poison shadow memory outside of asan runtime is not allowed, so we
         // need to avoid memset's call from being intercepted.
-        static void *memset_ptr = []() {
-            void *handle = dlopen("libc.so.6", RTLD_LAZY);
-            if (!handle) {
-                context.logger.error("dlopen failed: {}", dlerror());
-                return (void *)nullptr;
-            }
-            void *ptr = dlsym(handle, "memset");
-            if (!ptr) {
-                context.logger.error("dlsym failed: {}", dlerror());
-                return (void *)nullptr;
-            }
-            return ptr;
-        }();
-
+        static void *memset_ptr = GetMemFunctionPointer("memset");
         assert(nullptr != memset_ptr);
         ((void *(*)(void *, int, size_t))memset_ptr)(
             (void *)ShadowBegin, Value, ShadowEnd - ShadowBegin + 1);

source/loader/layers/sanitizer/common.hpp

@@ -105,4 +105,6 @@ bool SetupShadowMem();
 
 bool DestroyShadowMem();
 
+void *GetMemFunctionPointer(const char *);
+
 } // namespace ur_sanitizer_layer

source/loader/layers/sanitizer/linux/san_utils.cpp

@@ -13,6 +13,7 @@
 
 #include "common.hpp"
 #include <asm/param.h>
+#include <dlfcn.h>
 #include <sys/mman.h>
 
 extern "C" __attribute__((weak)) void __asan_init(void);
@@ -67,4 +68,16 @@ bool DestroyShadowMem() {
     return true;
 }
 
+void *GetMemFunctionPointer(const char *FuncName) {
+    void *handle = dlopen("libc.so.6", RTLD_LAZY);
+    if (!handle) {
+        return (void *)nullptr;
+    }
+    void *ptr = dlsym(handle, FuncName);
+    if (!ptr) {
+        return (void *)nullptr;
+    }
+    return ptr;
+}
+
 } // namespace ur_sanitizer_layer