test/Instrumentation/MemorySanitizer/wrap_indirect_calls.ll

   1 ; RUN: opt < %s -msan -msan-check-access-address=0 -msan-wrap-indirect-calls=zzz -msan-wrap-indirect-calls-fast=0 -S | FileCheck %s
   2 ; RUN: opt < %s -msan -msan-check-access-address=0 -msan-wrap-indirect-calls=zzz -msan-wrap-indirect-calls-fast=1 -S | FileCheck -check-prefix=CHECK-FAST %s
   3 target 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-S128"
   4 target triple = "x86_64-unknown-linux-gnu"
   5
   6 ; Test for -msan-wrap-indirect-calls functionality.
   7 ; Replaces indirect call to %f with a call to whatever is returned from the
   8 ; wrapper function.
   9
  10 ; This does not depend on the sanitize_memory attribute.
  11 define i32 @func1(i32 (i32, i32)* nocapture %f, i32 %x, i32 %y) {
  12 entry:
  13   %call = tail call i32 %f(i32 %x, i32 %y)
  14   ret i32 %call
  15 }
  16
  17 ; CHECK: @func1
  18 ; CHECK: bitcast i32 (i32, i32)* %f to void ()*
  19 ; CHECK: call void ()* (void ()*)* @zzz(void ()*
  20 ; CHECK: [[A:%[01-9a-z_.]+]] = bitcast void ()* {{.*}} to i32 (i32, i32)*
  21 ; CHECK: call i32 {{.*}}[[A]](i32 {{.*}}, i32 {{.*}})
  22 ; CHECK: ret i32
  23
  24 ; CHECK-FAST: @func1
  25 ; CHECK-FAST: bitcast i32 (i32, i32)* %f to void ()*
  26 ; CHECK-FAST-DAG: icmp ult void ()* {{.*}}, bitcast (i32* @__executable_start to void ()*)
  27 ; CHECK-FAST-DAG: icmp uge void ()* {{.*}}, bitcast (i32* @_end to void ()*)
  28 ; CHECK-FAST: or i1
  29 ; CHECK-FAST: br i1
  30 ; CHECK-FAST: call void ()* (void ()*)* @zzz(void ()*
  31 ; CHECK-FAST: br label
  32 ; CHECK-FAST: [[A:%[01-9a-z_.]+]] = phi i32 (i32, i32)* [ %f, %entry ], [ {{.*}} ]
  33 ; CHECK-FAST: call i32 {{.*}}[[A]](i32 {{.*}}, i32 {{.*}})
  34 ; CHECK-FAST: ret i32
  35
  36
  37 ; The same test, but with a complex expression as the call target.
  38
  39 declare i8* @callee(i32)
  40
  41 define i8* @func2(i64 %x) #1 {
  42 entry:
  43   %call = tail call i8* bitcast (i8* (i32)* @callee to i8* (i64)*)(i64 %x)
  44   ret i8* %call
  45 }
  46
  47 ; CHECK: @func2
  48 ; CHECK: call {{.*}} @zzz
  49 ; CHECK: [[A:%[01-9a-z_.]+]] = bitcast void ()* {{.*}} to i8* (i64)*
  50 ; CHECK: call i8* {{.*}}[[A]](i64 {{.*}})
  51 ; CHECK: ret i8*
  52
  53 ; CHECK-FAST: @func2
  54 ; CHECK-FAST: {{br i1 or .* icmp ult .* bitcast .* @callee .* @__executable_start.* icmp uge .* bitcast .* @callee .* @_end}}
  55 ; CHECK-FAST: {{call .* @zzz.* bitcast .*@callee}}
  56 ; CHECK-FAST: bitcast void ()* {{.*}} to i8* (i64)*
  57 ; CHECK-FAST: br label
  58 ; CHECK-FAST: [[A:%[01-9a-z_.]+]] = phi i8* (i64)* [{{.*bitcast .* @callee.*, %entry.*}}], [ {{.*}} ]
  59 ; CHECK-FAST: call i8* {{.*}}[[A]](i64 {{.*}})
  60 ; CHECK-FAST: ret i8*