llvm/test/Transforms/SafeStack/X86/musttail.ll

   1 ; To test that safestack does not break the musttail call contract.
   2 ;
   3 ; RUN: opt < %s --safe-stack -S | FileCheck %s
   4
   5 target triple = "x86_64-unknown-linux-gnu"
   6
   7 declare i32 @foo(ptr %p)
   8 declare void @alloca_test_use(ptr)
   9
  10 define i32 @call_foo(ptr %a) safestack {
  11 ; CHECK-LABEL: @call_foo(
  12 ; CHECK-NEXT:    [[UNSAFE_STACK_PTR:%.*]] = load ptr, ptr @__safestack_unsafe_stack_ptr, align 8
  13 ; CHECK-NEXT:    [[UNSAFE_STACK_STATIC_TOP:%.*]] = getelementptr i8, ptr [[UNSAFE_STACK_PTR]], i32 -16
  14 ; CHECK-NEXT:    store ptr [[UNSAFE_STACK_STATIC_TOP]], ptr @__safestack_unsafe_stack_ptr, align 8
  15 ; CHECK-NEXT:    [[TMP1:%.*]] = getelementptr i8, ptr [[UNSAFE_STACK_PTR]], i32 -10
  16 ; CHECK-NEXT:    call void @alloca_test_use(ptr [[TMP1]])
  17 ; CHECK-NEXT:    store ptr [[UNSAFE_STACK_PTR]], ptr @__safestack_unsafe_stack_ptr, align 8
  18 ; CHECK-NEXT:    [[R:%.*]] = musttail call i32 @foo(ptr [[A:%.*]])
  19 ; CHECK-NEXT:    ret i32 [[R]]
  20 ;
  21   %x = alloca [10 x i8], align 1
  22   call void @alloca_test_use(ptr %x)
  23   %r = musttail call i32 @foo(ptr %a)
  24   ret i32 %r
  25 }
  26
  27 define i32 @call_foo_cast(ptr %a) safestack {
  28 ; CHECK-LABEL: @call_foo_cast(
  29 ; CHECK-NEXT:    [[UNSAFE_STACK_PTR:%.*]] = load ptr, ptr @__safestack_unsafe_stack_ptr, align 8
  30 ; CHECK-NEXT:    [[UNSAFE_STACK_STATIC_TOP:%.*]] = getelementptr i8, ptr [[UNSAFE_STACK_PTR]], i32 -16
  31 ; CHECK-NEXT:    store ptr [[UNSAFE_STACK_STATIC_TOP]], ptr @__safestack_unsafe_stack_ptr, align 8
  32 ; CHECK-NEXT:    [[TMP1:%.*]] = getelementptr i8, ptr [[UNSAFE_STACK_PTR]], i32 -10
  33 ; CHECK-NEXT:    call void @alloca_test_use(ptr [[TMP1]])
  34 ; CHECK-NEXT:    store ptr [[UNSAFE_STACK_PTR]], ptr @__safestack_unsafe_stack_ptr, align 8
  35 ; CHECK-NEXT:    [[R:%.*]] = musttail call i32 @foo(ptr [[A:%.*]])
  36 ; CHECK-NEXT:    [[T:%.*]] = bitcast i32 [[R]] to i32
  37 ; CHECK-NEXT:    ret i32 [[T]]
  38 ;
  39   %x = alloca [10 x i8], align 1
  40   call void @alloca_test_use(ptr %x)
  41   %r = musttail call i32 @foo(ptr %a)
  42   %t = bitcast i32 %r to i32
  43   ret i32 %t
  44 }