clang/test/CodeGen/ubsan-builtin-checks.c

   1 // RUN: %clang_cc1 -triple x86_64-apple-darwin10 -w -emit-llvm -o - %s -fsanitize=builtin | FileCheck %s --check-prefixes=CHECK,POISON
   2 // RUN: %clang_cc1 -triple arm64-none-linux-gnu -w -emit-llvm -o - %s -fsanitize=builtin | FileCheck %s --check-prefixes=CHECK,NOPOISON
   3
   4 // A zero input to __bultin_ctz/clz is considered UB even if the target does not
   5 // want to optimize based on zero input being undefined.
   6
   7 // CHECK: define{{.*}} void @check_ctz
   8 void check_ctz(int n) {
   9   // CHECK: [[NOT_ZERO:%.*]] = icmp ne i32 [[N:%.*]], 0, !nosanitize
  10   // CHECK-NEXT: br i1 [[NOT_ZERO]]
  11   //
  12   // Handler block:
  13   // CHECK: call void @__ubsan_handle_invalid_builtin
  14   // CHECK-NEXT: unreachable
  15   //
  16   // Continuation block:
  17   // POISON: call i32 @llvm.cttz.i32(i32 [[N]], i1 true)
  18   // NOPOISON: call i32 @llvm.cttz.i32(i32 [[N]], i1 false)
  19   __builtin_ctz(n);
  20
  21   // CHECK: call void @__ubsan_handle_invalid_builtin
  22   __builtin_ctzl(n);
  23
  24   // CHECK: call void @__ubsan_handle_invalid_builtin
  25   __builtin_ctzll(n);
  26
  27   // CHECK: call void @__ubsan_handle_invalid_builtin
  28   __builtin_ctzg((unsigned int)n);
  29 }
  30
  31 // CHECK: define{{.*}} void @check_clz
  32 void check_clz(int n) {
  33   // CHECK: [[NOT_ZERO:%.*]] = icmp ne i32 [[N:%.*]], 0, !nosanitize
  34   // CHECK-NEXT: br i1 [[NOT_ZERO]]
  35   //
  36   // Handler block:
  37   // CHECK: call void @__ubsan_handle_invalid_builtin
  38   // CHECK-NEXT: unreachable
  39   //
  40   // Continuation block:
  41   // POISON: call i32 @llvm.ctlz.i32(i32 [[N]], i1 true)
  42   // NOPOISON: call i32 @llvm.ctlz.i32(i32 [[N]], i1 false)
  43   __builtin_clz(n);
  44
  45   // CHECK: call void @__ubsan_handle_invalid_builtin
  46   __builtin_clzl(n);
  47
  48   // CHECK: call void @__ubsan_handle_invalid_builtin
  49   __builtin_clzll(n);
  50
  51   // CHECK: call void @__ubsan_handle_invalid_builtin
  52   __builtin_clzg((unsigned int)n);
  53 }
  54
  55 // CHECK: define{{.*}} void @check_assume
  56 void check_assume(int n) {
  57   // CHECK: [[TOBOOL:%.*]] = icmp ne i32 [[N:%.*]], 0
  58   // CHECK-NEXT: br i1 [[TOBOOL]]
  59   //
  60   // Handler block:
  61   // CHECK: call void @__ubsan_handle_invalid_builtin
  62   // CHECK-NEXT: unreachable
  63   //
  64   // Continuation block:
  65   // CHECK: call void @llvm.assume(i1 [[TOBOOL]])
  66   __builtin_assume(n);
  67
  68   // CHECK: call void @__ubsan_handle_invalid_builtin
  69   __attribute__((assume(n)));
  70 }