llvm/test/Instrumentation/DataFlowSanitizer/select.ll

   1 ; RUN: opt < %s -passes=dfsan -dfsan-track-select-control-flow=true -S | FileCheck %s --check-prefixes=CHECK,TRACK_CF
   2 ; RUN: opt < %s -passes=dfsan -dfsan-track-select-control-flow=false -S | FileCheck %s --check-prefixes=CHECK,NO_TRACK_CF
   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 ; CHECK: @__dfsan_arg_tls = external thread_local(initialexec) global [[TLS_ARR:\[100 x i64\]]]
   7 ; CHECK: @__dfsan_retval_tls = external thread_local(initialexec) global [[TLS_ARR]]
   8 define i8 @select8(i1 %c, i8 %t, i8 %f) {
   9   ; TRACK_CF: @select8.dfsan
  10   ; TRACK_CF: %[[#R:]] = load i8, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__dfsan_arg_tls to i64), i64 4) to ptr), align [[ALIGN:2]]
  11   ; TRACK_CF: %[[#R+1]] = load i8, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__dfsan_arg_tls to i64), i64 2) to ptr), align [[ALIGN]]
  12   ; TRACK_CF: %[[#R+2]] = load i8, ptr @__dfsan_arg_tls, align [[ALIGN]]
  13   ; TRACK_CF: %[[#R+3]] = select i1 %c, i8 %[[#R+1]], i8 %[[#R]]
  14   ; TRACK_CF: %[[#RO:]] = or i8 %[[#R+2]], %[[#R+3]]
  15   ; TRACK_CF: %a = select i1 %c, i8 %t, i8 %f
  16   ; TRACK_CF: store i8 %[[#RO]], ptr @__dfsan_retval_tls, align [[ALIGN]]
  17   ; TRACK_CF: ret i8 %a
  18
  19   ; NO_TRACK_CF: @select8.dfsan
  20   ; NO_TRACK_CF: %[[#R:]] = load i8, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__dfsan_arg_tls to i64), i64 4) to ptr), align [[ALIGN:2]]
  21   ; NO_TRACK_CF: %[[#R+1]] = load i8, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__dfsan_arg_tls to i64), i64 2) to ptr), align [[ALIGN]]
  22   ; NO_TRACK_CF: %[[#R+2]] = load i8, ptr @__dfsan_arg_tls, align [[ALIGN]]
  23   ; NO_TRACK_CF: %[[#R+3]] = select i1 %c, i8 %[[#R+1]], i8 %[[#R]]
  24   ; NO_TRACK_CF: %a = select i1 %c, i8 %t, i8 %f
  25   ; NO_TRACK_CF: store i8 %[[#R+3]], ptr @__dfsan_retval_tls, align [[ALIGN]]
  26   ; NO_TRACK_CF: ret i8 %a
  27
  28   %a = select i1 %c, i8 %t, i8 %f
  29   ret i8 %a
  30 }
  31
  32 define i8 @select8e(i1 %c, i8 %tf) {
  33   ; TRACK_CF: @select8e.dfsan
  34   ; TRACK_CF: %[[#R:]] = load i8, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__dfsan_arg_tls to i64), i64 2) to ptr), align [[ALIGN]]
  35   ; TRACK_CF: %[[#R+1]] = load i8, ptr @__dfsan_arg_tls, align [[ALIGN]]
  36   ; TRACK_CF: %[[#RO:]] = or i8 %[[#R+1]], %[[#R]]
  37   ; TRACK_CF: %a = select i1 %c, i8 %tf, i8 %tf
  38   ; TRACK_CF: store i8 %[[#RO]], ptr @__dfsan_retval_tls, align [[ALIGN]]
  39   ; TRACK_CF: ret i8 %a
  40
  41   ; NO_TRACK_CF: @select8e.dfsan
  42   ; NO_TRACK_CF: %[[#R:]] = load i8, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__dfsan_arg_tls to i64), i64 2) to ptr), align [[ALIGN]]
  43   ; NO_TRACK_CF: %[[#R+1]] = load i8, ptr @__dfsan_arg_tls, align [[ALIGN]]
  44   ; NO_TRACK_CF: %a = select i1 %c, i8 %tf, i8 %tf
  45   ; NO_TRACK_CF: store i8 %[[#R]], ptr @__dfsan_retval_tls, align [[ALIGN]]
  46   ; NO_TRACK_CF: ret i8 %a
  47
  48   %a = select i1 %c, i8 %tf, i8 %tf
  49   ret i8 %a
  50 }
  51
  52 define <4 x i8> @select8v(<4 x i1> %c, <4 x i8> %t, <4 x i8> %f) {
  53   ; TRACK_CF: @select8v.dfsan
  54   ; TRACK_CF: %[[#R:]] = load i8, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__dfsan_arg_tls to i64), i64 4) to ptr), align [[ALIGN:2]]
  55   ; TRACK_CF: %[[#R+1]] = load i8, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__dfsan_arg_tls to i64), i64 2) to ptr), align [[ALIGN]]
  56   ; TRACK_CF: %[[#R+2]] = load i8, ptr @__dfsan_arg_tls, align [[ALIGN]]
  57   ; TRACK_CF: %[[#R+3]] = or i8 %[[#R+1]], %[[#R]]
  58   ; TRACK_CF: %[[#RO:]] = or i8 %[[#R+2]], %[[#R+3]]
  59   ; TRACK_CF: %a = select <4 x i1> %c, <4 x i8> %t, <4 x i8> %f
  60   ; TRACK_CF: store i8 %[[#RO]], ptr @__dfsan_retval_tls, align [[ALIGN]]
  61   ; TRACK_CF: ret <4 x i8> %a
  62
  63   ; NO_TRACK_CF: @select8v.dfsan
  64   ; NO_TRACK_CF: %[[#R:]] = load i8, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__dfsan_arg_tls to i64), i64 4) to ptr), align [[ALIGN:2]]
  65   ; NO_TRACK_CF: %[[#R+1]] = load i8, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__dfsan_arg_tls to i64), i64 2) to ptr), align [[ALIGN]]
  66   ; NO_TRACK_CF: %[[#R+2]] = load i8, ptr @__dfsan_arg_tls, align [[ALIGN]]
  67   ; NO_TRACK_CF: %[[#RO:]] = or i8 %[[#R+1]], %[[#R]]
  68   ; NO_TRACK_CF: %a = select <4 x i1> %c, <4 x i8> %t, <4 x i8> %f
  69   ; NO_TRACK_CF: store i8 %[[#RO]], ptr @__dfsan_retval_tls, align [[ALIGN]]
  70   ; NO_TRACK_CF: ret <4 x i8> %a
  71
  72   %a = select <4 x i1> %c, <4 x i8> %t, <4 x i8> %f
  73   ret <4 x i8> %a
  74 }