1*88cb3e2cSTeresa Johnson; Test optional annotation of stack addresses.
2*88cb3e2cSTeresa Johnson;
3*88cb3e2cSTeresa Johnson; RUN: opt < %s -passes='function(memprof),module(memprof-module)' -S | FileCheck --check-prefixes=CHECK %s
4*88cb3e2cSTeresa Johnson; RUN: opt < %s -passes='function(memprof),module(memprof-module)' -memprof-instrument-stack -S | FileCheck --check-prefixes=CHECK,STACK %s
5*88cb3e2cSTeresa Johnson
6*88cb3e2cSTeresa Johnsontarget datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32:64"
7*88cb3e2cSTeresa Johnsontarget triple = "x86_64-unknown-linux-gnu"
8*88cb3e2cSTeresa Johnson
9*88cb3e2cSTeresa Johnsondefine i32 @test_stack_load() {
10*88cb3e2cSTeresa Johnsonentry:
11*88cb3e2cSTeresa Johnson  %x = alloca i32, align 4
12*88cb3e2cSTeresa Johnson  %tmp1 = load i32, i32* %x, align 4
13*88cb3e2cSTeresa Johnson  ret i32 %tmp1
14*88cb3e2cSTeresa Johnson}
15*88cb3e2cSTeresa Johnson; CHECK-LABEL: @test_stack_load
16*88cb3e2cSTeresa Johnson; CHECK:         %[[SHADOW_OFFSET:[^ ]*]] = load i64, i64* @__memprof_shadow_memory_dynamic_address
17*88cb3e2cSTeresa Johnson; CHECK-NEXT:	 %x = alloca i32
18*88cb3e2cSTeresa Johnson; STACK-NEXT:    %[[LOAD_ADDR:[^ ]*]] = ptrtoint i32* %x to i64
19*88cb3e2cSTeresa Johnson; STACK-NEXT:    %[[MASKED_ADDR:[^ ]*]] = and i64 %[[LOAD_ADDR]], -64
20*88cb3e2cSTeresa Johnson; STACK-NEXT:    %[[SHIFTED_ADDR:[^ ]*]] = lshr i64 %[[MASKED_ADDR]], 3
21*88cb3e2cSTeresa Johnson; STACK-NEXT:    add i64 %[[SHIFTED_ADDR]], %[[SHADOW_OFFSET]]
22*88cb3e2cSTeresa Johnson; STACK-NEXT:    %[[LOAD_SHADOW_PTR:[^ ]*]] = inttoptr
23*88cb3e2cSTeresa Johnson; STACK-NEXT:    %[[LOAD_SHADOW:[^ ]*]] = load i64, i64* %[[LOAD_SHADOW_PTR]]
24*88cb3e2cSTeresa Johnson; STACK-NEXT:    %[[NEW_SHADOW:[^ ]*]] = add i64 %[[LOAD_SHADOW]], 1
25*88cb3e2cSTeresa Johnson; STACK-NEXT:    store i64 %[[NEW_SHADOW]], i64* %[[LOAD_SHADOW_PTR]]
26*88cb3e2cSTeresa Johnson; The actual load.
27*88cb3e2cSTeresa Johnson; CHECK-NEXT:    %tmp1 = load i32, i32* %x
28*88cb3e2cSTeresa Johnson; CHECK-NEXT:    ret i32 %tmp1
29*88cb3e2cSTeresa Johnson
30*88cb3e2cSTeresa Johnsondefine void @test_stack_store() {
31*88cb3e2cSTeresa Johnsonentry:
32*88cb3e2cSTeresa Johnson  %x = alloca i32, align 4
33*88cb3e2cSTeresa Johnson  store i32 1, i32* %x, align 4
34*88cb3e2cSTeresa Johnson  ret void
35*88cb3e2cSTeresa Johnson}
36*88cb3e2cSTeresa Johnson; CHECK-LABEL: @test_stack_store
37*88cb3e2cSTeresa Johnson; CHECK:         %[[SHADOW_OFFSET:[^ ]*]] = load i64, i64* @__memprof_shadow_memory_dynamic_address
38*88cb3e2cSTeresa Johnson; CHECK-NEXT:	 %x = alloca i32
39*88cb3e2cSTeresa Johnson; STACK-NEXT:    %[[STORE_ADDR:[^ ]*]] = ptrtoint i32* %x to i64
40*88cb3e2cSTeresa Johnson; STACK-NEXT:    %[[MASKED_ADDR:[^ ]*]] = and i64 %[[STORE_ADDR]], -64
41*88cb3e2cSTeresa Johnson; STACK-NEXT:    %[[SHIFTED_ADDR:[^ ]*]] = lshr i64 %[[MASKED_ADDR]], 3
42*88cb3e2cSTeresa Johnson; STACK-NEXT:    add i64 %[[SHIFTED_ADDR]], %[[SHADOW_OFFSET]]
43*88cb3e2cSTeresa Johnson; STACK-NEXT:    %[[STORE_SHADOW_PTR:[^ ]*]] = inttoptr
44*88cb3e2cSTeresa Johnson; STACK-NEXT:    %[[STORE_SHADOW:[^ ]*]] = load i64, i64* %[[STORE_SHADOW_PTR]]
45*88cb3e2cSTeresa Johnson; STACK-NEXT:    %[[NEW_SHADOW:[^ ]*]] = add i64 %[[STORE_SHADOW]], 1
46*88cb3e2cSTeresa Johnson; STACK-NEXT:    store i64 %[[NEW_SHADOW]], i64* %[[STORE_SHADOW_PTR]]
47*88cb3e2cSTeresa Johnson; The actual store.
48*88cb3e2cSTeresa Johnson; CHECK-NEXT:    store i32 1, i32* %x
49*88cb3e2cSTeresa Johnson; CHECK-NEXT:    ret void
50