From b0a00a5387697a0450f9d67061f0976a803363de Mon Sep 17 00:00:00 2001
From: liuwenfeng <bluecodeman@outlook.com>
Date: Thu, 29 May 2025 04:08:20 +0000
Subject: [PATCH] support asan

Signed-off-by: liuwenfeng <bluecodeman@outlook.com>
---
 BUILD.gn                      |  9 ++++++
 ohos/build_ohos64.py          |  6 ++--
 ohos/meson_cross_process64.py | 59 ++++++++++++++++++++++++++++++-----
 3 files changed, 63 insertions(+), 11 deletions(-)

diff --git a/BUILD.gn b/BUILD.gn
index 2a03af468f8..0c45e230ef9 100644
--- a/BUILD.gn
+++ b/BUILD.gn
@@ -28,10 +28,19 @@ action("mesa3d_action_build") {
   ohos_root_path = rebase_path("//")
   product_name = rebase_path("${root_build_dir}", "//out")
   mesa3d_source_path = rebase_path("//third_party/mesa3d", root_build_dir)
+  asan_option = "noasan"
+  if (is_asan) {
+    if (use_hwasan) {
+      asan_option = "hwasan"
+    } else {
+      asan_option = "swasan"
+    }
+  }
   args = [
     "$ohos_root_path",
     "$product_name",
     "$mesa3d_source_path",
+    "$asan_option",
   ]
   external_deps = [
     "hilog:libhilog",
diff --git a/ohos/build_ohos64.py b/ohos/build_ohos64.py
index a103065eeb0..b7c8340b719 100755
--- a/ohos/build_ohos64.py
+++ b/ohos/build_ohos64.py
@@ -24,11 +24,11 @@ import sys
 import ntpath
 import os
 if __name__ == '__main__':
-    if len(sys.argv) < 4:
-        print("must input the OpenHarmony directory and the product name and the source dir")
+    if len(sys.argv) < 5:
+        print("must input the OpenHarmony directory and the product name and the source dir and the asan option")
         exit(-1)
     script_dir = os.path.split(os.path.abspath( __file__))[0]
-    run_cross_pross_cmd = 'python3 ' + script_dir + '/meson_cross_process64.py ' + sys.argv[1] + ' ' + sys.argv[2]
+    run_cross_pross_cmd = 'python3 ' + script_dir + '/meson_cross_process64.py ' + sys.argv[1] + ' ' + sys.argv[2] + ' ' + sys.argv[4]
     os.system(run_cross_pross_cmd)
 
     run_build_cmd = 'PKG_CONFIG_PATH=./thirdparty/mesa3d/pkgconfig '
diff --git a/ohos/meson_cross_process64.py b/ohos/meson_cross_process64.py
index 6c8d0c85fab..f42c4cf5423 100755
--- a/ohos/meson_cross_process64.py
+++ b/ohos/meson_cross_process64.py
@@ -36,13 +36,17 @@ c_args = [
     '--target=aarch64-linux-ohosmusl',
     '--sysroot=sysroot_stub',
     '-fno-emulated-tls',
-    '-fPIC']
+    '-fPIC',
+    asan_compile_args_stub
+    ]
 
 cpp_args = [
     '--target=aarch64-linux-ohosmusl',
     '--sysroot=sysroot_stub',
     '-fno-emulated-tls',
-    '-fPIC']
+    '-fPIC',
+    asan_compile_args_stub
+    ]
     
 c_link_args = [
     '--target=aarch64-linux-ohosmusl',
@@ -52,6 +56,7 @@ c_link_args = [
     '-Lproject_stub/prebuilts/clang/ohos/linux-x86_64/llvm/lib/clang/current/lib/aarch64-linux-ohos',
     '-Lproject_stub/prebuilts/clang/ohos/linux-x86_64/llvm/lib/aarch64-linux-ohos/c++',
     '--rtlib=compiler-rt',
+    asan_link_args_stub
     ]
 
 cpp_link_args = [
@@ -66,6 +71,7 @@ cpp_link_args = [
     '-Wl,--exclude-libs=libvpx_assembly_arm.a',
     '-Wl,--warn-shared-textrel',
     '--rtlib=compiler-rt',
+    asan_link_args_stub
     ]
 	
 [binaries]
@@ -84,10 +90,47 @@ cpu = 'armv8'
 endian = 'little'
 '''
 
-def generate_cross_file(project_stub_in, sysroot_stub_in):
+hwasan_compile_args = '''
+    '-shared-libasan',
+    '-fsanitize=hwaddress',
+    '-mllvm',
+    '-hwasan-globals=0',
+    '-fno-lto',
+    '-fno-whole-program-vtables',
+'''
+hwasan_link_args = '''
+    '-shared-libasan',
+    '-fsanitize=hwaddress',
+'''
+
+swasan_compile_args = '''
+    '-fsanitize=address',
+    '-fno-inline-functions',
+    '-fno-inline',
+    '-fsanitize-address-use-after-scope',
+    '-fno-omit-frame-pointer',
+'''
+swasan_link_args = '''
+    '-lclang_rt.asan',
+'''
+
+noasan_compile_args = '''
+'''
+noasan_link_args = '''
+'''
+def generate_cross_file(project_stub_in, sysroot_stub_in, asan_option):
     with open("thirdparty/mesa3d/cross_file", 'w+') as file:
         result = corss_file_content.replace("project_stub", project_stub_in)
         result = result.replace("sysroot_stub", sysroot_stub_in)
+        if asan_option == "hwasan":
+            result = result.replace("asan_compile_args_stub", hwasan_compile_args)
+            result = result.replace("asan_link_args_stub", hwasan_link_args)
+        elif asan_option == "swasan":
+            result = result.replace("asan_compile_args_stub", swasan_compile_args)
+            result = result.replace("asan_link_args_stub", swasan_link_args)
+        else:
+            result = result.replace("asan_compile_args_stub", noasan_compile_args)
+            result = result.replace("asan_link_args_stub", noasan_link_args)
         file.write(result)
     print("generate_cross_file")
 
@@ -113,17 +156,17 @@ def process_pkgconfig(project_dir, product_name):
             generate_pc_file(template_dir + '/' + template, project_dir, product_name)
     print("process_pkgconfig")
 
-def prepare_environment(project_path, product):
+def prepare_environment(project_path, product, asan_option):
     global project_stub
     global sysroot_stub
     product = product.lower()
     project_stub = project_path
     sysroot_stub = os.path.join(project_stub, "out", product, "obj", "third_party", "musl")
-    generate_cross_file(project_path, sysroot_stub)
+    generate_cross_file(project_path, sysroot_stub, asan_option)
     process_pkgconfig(project_path, product)
 
 if __name__ == '__main__':
-    if len(sys.argv) < 3:
-        print("must input the OpenHarmony directory and the product name")
+    if len(sys.argv) < 4:
+        print("must input the OpenHarmony directory and the product name and the asan option")
         exit(-1)
-    prepare_environment(sys.argv[1], sys.argv[2])
+    prepare_environment(sys.argv[1], sys.argv[2], sys.argv[3])
-- 
Gitee