Subzero: Replace pointers to allocation functions in stores

tests_lit/asan_tests/localreplacement.ll

Index: tests_lit/asan_tests/localreplacement.ll
diff --git a/tests_lit/asan_tests/localreplacement.ll b/tests_lit/asan_tests/localreplacement.ll
index 06b810c0c773a02747c316f96b1a26a63afdd48a..b2c121617ea2a8481ac3ae852597f51ac8ad6422 100644
--- a/tests_lit/asan_tests/localreplacement.ll
+++ b/tests_lit/asan_tests/localreplacement.ll
@@ -1,4 +1,5 @@
-; Test that loads of local pointers to allocation functions are instrumented
+; Test that loads of local pointers to allocation functions and stores
+; of pointers to allocation functions are instrumented.
 
 ; REQUIRES: allow_dump
 
@@ -10,16 +11,28 @@ declare external i32 @realloc(i32, i32)
 declare external i32 @calloc(i32, i32)
 declare external void @free(i32)
 
-define internal void @func() {
-  %malloc_addr = bitcast i32 (i32)* @malloc to i32*
-  %realloc_addr = bitcast i32 (i32, i32)* @realloc to i32*
-  %calloc_addr = bitcast i32 (i32, i32)* @calloc to i32*
-  %free_addr = bitcast void (i32)* @free to i32*
+define internal void @func(i32 %store_loc) {
+  %store_dest = inttoptr i32 %store_loc to i32*
 
-  %local_malloc = load i32, i32* %malloc_addr, align 1
-  %local_realloc = load i32, i32* %realloc_addr, align 1
-  %local_calloc = load i32, i32* %calloc_addr, align 1
-  %local_free = load i32, i32* %free_addr, align 1
+  %malloc_ptr = bitcast i32 (i32)* @malloc to i32*
+  %realloc_ptr = bitcast i32 (i32, i32)* @realloc to i32*
+  %calloc_ptr = bitcast i32 (i32, i32)* @calloc to i32*
+  %free_ptr = bitcast void (i32)* @free to i32*
+
+  %malloc_addr = ptrtoint i32 (i32)* @malloc to i32
+  %realloc_addr = ptrtoint i32 (i32, i32)* @realloc to i32
+  %calloc_addr = ptrtoint i32 (i32, i32)* @calloc to i32
+  %free_addr = ptrtoint void (i32)* @free to i32
+
+  store i32 %malloc_addr, i32* %store_dest, align 1
+  store i32 %realloc_addr, i32* %store_dest, align 1
+  store i32 %calloc_addr, i32* %store_dest, align 1
+  store i32 %free_addr, i32* %store_dest, align 1
+
+  %local_malloc = load i32, i32* %malloc_ptr, align 1
+  %local_realloc = load i32, i32* %realloc_ptr, align 1
+  %local_calloc = load i32, i32* %calloc_ptr, align 1
+  %local_free = load i32, i32* %free_ptr, align 1
 
   %local_mallocfunc = inttoptr i32 %local_malloc to i32 (i32)*
   %local_reallocfunc = inttoptr i32 %local_realloc to i32 (i32, i32)*
@@ -32,8 +45,13 @@ define internal void @func() {
 }
 
 ; DUMP-LABEL: ================ Instrumented CFG ================
-; DUMP-NEXT: @func() {
+; DUMP-NEXT: @func(i32 %store_loc) {
 ; DUMP-NEXT: __0:
+; DUMP-NEXT:   call void @__asan_check_store(i32 %store_loc, i32 4)
+; DUMP-NEXT:   store i32 @__asan_malloc, i32* %store_loc, align 1
+; DUMP-NEXT:   store i32 @__asan_realloc, i32* %store_loc, align 1
+; DUMP-NEXT:   store i32 @__asan_calloc, i32* %store_loc, align 1
+; DUMP-NEXT:   store i32 @__asan_free, i32* %store_loc, align 1
 ; DUMP-NEXT:   call void @__asan_check_load(i32 @__asan_malloc, i32 4)
 ; DUMP-NEXT:   %local_malloc = load i32, i32* @__asan_malloc, align 1
 ; DUMP-NEXT:   call void @__asan_check_load(i32 @__asan_realloc, i32 4)