From 7280f28b40faf880cdd7c3b93807b13fd4f0d645 Mon Sep 17 00:00:00 2001
From: Chris Freehill <Chris.Freehill@amd.com>
Date: Tue, 10 Nov 2020 15:45:56 -0600
Subject: [PATCH] rsmitst address sanitizer support

Also, add libasan flag variants for librocm_smi

Change-Id: Ibd012e40d26907addf8c0550aaf9f78c11b8d51f


[ROCm/rocm_smi_lib commit: bf6af909088f6e545844c4098759361a18cb5218]
---
 projects/rocm-smi-lib/CMakeLists.txt                |  6 ++++++
 .../rocm-smi-lib/tests/rocm_smi_test/CMakeLists.txt | 13 +++++++++++++
 2 files changed, 19 insertions(+)

diff --git a/projects/rocm-smi-lib/CMakeLists.txt b/projects/rocm-smi-lib/CMakeLists.txt
index 1c6d7c9c74..3464baf18a 100755
--- a/projects/rocm-smi-lib/CMakeLists.txt
+++ b/projects/rocm-smi-lib/CMakeLists.txt
@@ -103,6 +103,12 @@ set(CMAKE_CXX_FLAGS
 if (${ADDRESS_SANITIZER})
     set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fsanitize=address")
     set(CMAKE_EXE_LINKER_FLAGS -fsanitize=address)
+
+    if (BUILD_SHARED_LIBS})
+        set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -shared-libasan" )
+    else ()
+        set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -static-libasan" )
+    endif ()
 else ()
     ## Security breach mitigation flags
     set(CMAKE_CXX_FLAGS
diff --git a/projects/rocm-smi-lib/tests/rocm_smi_test/CMakeLists.txt b/projects/rocm-smi-lib/tests/rocm_smi_test/CMakeLists.txt
index 78953c0e03..db6a5af136 100755
--- a/projects/rocm-smi-lib/tests/rocm_smi_test/CMakeLists.txt
+++ b/projects/rocm-smi-lib/tests/rocm_smi_test/CMakeLists.txt
@@ -140,6 +140,19 @@ set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wall")
 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wextra")
 # set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -pedantic")
 
+## Address Sanitize Flag
+if (${ADDRESS_SANITIZER})
+    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fsanitize=address")
+    set(CMAKE_EXE_LINKER_FLAGS -fsanitize=address)
+
+    if (BUILD_SHARED_LIBS})
+        set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -shared-libasan" )
+    else ()
+        set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -static-libasan" )
+    endif ()
+endif()
+
+
 #
 # Extend the compiler flags for 64-bit builds
 #