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 @func(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: @func
  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: @func
  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