clang/test/CodeGen/ptrauth-function-lvalue-cast-disc.c

   1 // RUN: %clang_cc1 %s -triple arm64e-apple-ios13 -fptrauth-calls -fptrauth-intrinsics -emit-llvm -o- -fptrauth-function-pointer-type-discrimination | FileCheck -check-prefixes CHECK,TYPE %s
   2 // RUN: %clang_cc1 %s -triple aarch64-linux-gnu  -fptrauth-calls -fptrauth-intrinsics -emit-llvm -o- -fptrauth-function-pointer-type-discrimination | FileCheck -check-prefixes CHECK,TYPE %s
   3 // RUN: %clang_cc1 %s -triple arm64e-apple-ios13 -fptrauth-calls -fptrauth-intrinsics -emit-llvm -o- | FileCheck -check-prefixes CHECK,ZERO %s
   4 // RUN: %clang_cc1 %s -triple aarch64-linux-gnu  -fptrauth-calls -fptrauth-intrinsics -emit-llvm -o- | FileCheck -check-prefixes CHECK,ZERO %s
   5
   6 typedef void (*fptr_t)(void);
   7
   8 char *cptr;
   9 void (*fptr)(void);
  10
  11 // CHECK-LABEL: define{{.*}} void @test1
  12 void test1() {
  13   // TYPE: [[LOAD:%.*]] = load ptr, ptr @cptr
  14   // TYPE: [[TOINT:%.*]] = ptrtoint ptr [[LOAD]] to i64
  15   // TYPE: call i64 @llvm.ptrauth.resign(i64 [[TOINT]], i32 0, i64 0, i32 0, i64 18983)
  16   // TYPE: call void {{.*}}() [ "ptrauth"(i32 0, i64 18983) ]
  17   // ZERO-NOT: @llvm.ptrauth.resign
  18
  19   (*(fptr_t)cptr)();
  20 }
  21
  22 // CHECK-LABEL: define{{.*}} i8 @test2
  23 char test2() {
  24   return *(char *)fptr;
  25
  26   // TYPE: [[LOAD:%.*]] = load ptr, ptr @fptr
  27   // TYPE: [[CMP:%.*]] = icmp ne ptr [[LOAD]], null
  28   // TYPE-NEXT: br i1 [[CMP]], label %[[NONNULL:.*]], label %[[CONT:.*]]
  29
  30   // TYPE: [[NONNULL]]:
  31   // TYPE: [[TOINT:%.*]] = ptrtoint ptr [[LOAD]] to i64
  32   // TYPE: [[CALL:%.*]] = call i64 @llvm.ptrauth.resign(i64 [[TOINT]], i32 0, i64 18983, i32 0, i64 0)
  33   // TYPE: [[TOPTR:%.*]] = inttoptr i64 [[CALL]] to ptr
  34
  35   // TYPE: [[CONT]]:
  36   // TYPE: phi ptr [ null, {{.*}} ], [ [[TOPTR]], %[[NONNULL]] ]
  37   // ZERO-NOT: @llvm.ptrauth.resign
  38 }
  39
  40 // CHECK-LABEL: define{{.*}} void @test4
  41 void test4() {
  42   (*((fptr_t)(&*((char *)(&*(fptr_t)cptr)))))();
  43
  44   // CHECK: [[LOAD:%.*]] = load ptr, ptr @cptr
  45   // TYPE-NEXT: [[CAST4:%.*]] = ptrtoint ptr [[LOAD]] to i64
  46   // TYPE-NEXT: [[RESIGN:%.*]] = call i64 @llvm.ptrauth.resign(i64 [[CAST4]], i32 0, i64 0, i32 0, i64 18983)
  47   // TYPE-NEXT: [[CAST5:%.*]] = inttoptr i64 [[RESIGN]] to ptr
  48   // TYPE-NEXT: call void [[CAST5]]() [ "ptrauth"(i32 0, i64 18983) ]
  49   // ZERO-NOT: @llvm.ptrauth.resign
  50   // ZERO: call void [[LOAD]]() [ "ptrauth"(i32 0, i64 0) ]
  51 }
  52
  53 void *vptr;
  54 // CHECK-LABEL: define{{.*}} void @test5
  55 void test5() {
  56   vptr = &*(char *)fptr;
  57
  58   // TYPE: [[LOAD:%.*]] = load ptr, ptr @fptr
  59   // TYPE-NEXT: [[CMP]] = icmp ne ptr [[LOAD]], null
  60   // TYPE-NEXT: br i1 [[CMP]], label %[[NONNULL:.*]], label %[[CONT:.*]]
  61
  62   // TYPE: [[NONNULL]]:
  63   // TYPE: [[RESIGN:%.*]] = call i64 @llvm.ptrauth.resign(i64 {{.*}}, i32 0, i64 18983, i32 0, i64 0)
  64   // TYPE: [[CAST:%.*]] = inttoptr i64 [[RESIGN]] to ptr
  65
  66   // TYPE: [[CONT]]:
  67   // TYPE: [[PHI:%.*]] = phi ptr [ null, {{.*}} ], [ [[CAST]], %[[NONNULL]] ]
  68   // TYPE: store ptr [[PHI]], ptr @vptr
  69   // ZERO-NOT: @llvm.ptrauth.resign
  70 }