shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
llvm-project/clang/test/CodeGen/ptrauth-qualifier-function.c
Akira Hatanaka a3283a92ae
[PAC] Add support for __ptrauth type qualifier (#100830) 
The qualifier allows programmer to directly control how pointers are
signed when they are stored in a particular variable.

The qualifier takes three arguments: the signing key, a flag specifying
whether address discrimination should be used, and a non-negative
integer that is used for additional discrimination.

```
typedef void (*my_callback)(const void*);
my_callback __ptrauth(ptrauth_key_process_dependent_code, 1, 0xe27a) callback;
```

Co-Authored-By: John McCall rjmccall@apple.com
2025-04-15 12:54:25 -07:00

146 lines
7.3 KiB
C
Raw Blame History

// RUN: %clang_cc1 %s -fptrauth-function-pointer-type-discrimination -triple arm64-apple-ios13 -fptrauth-calls -fptrauth-intrinsics -disable-llvm-passes -emit-llvm -o- | FileCheck --check-prefixes=CHECK,TYPE %s
// RUN: %clang_cc1 %s -fptrauth-function-pointer-type-discrimination -triple aarch64-linux-gnu -fptrauth-calls -fptrauth-intrinsics -disable-llvm-passes -emit-llvm -o- | FileCheck --check-prefixes=CHECK,TYPE %s
// RUN: %clang_cc1 %s -triple arm64-apple-ios13 -fptrauth-calls -fptrauth-intrinsics -disable-llvm-passes -emit-llvm -o- | FileCheck --check-prefixes=CHECK,ZERO %s
// RUN: %clang_cc1 %s -triple aarch64-linux-gnu -fptrauth-calls -fptrauth-intrinsics -disable-llvm-passes -emit-llvm -o- | FileCheck --check-prefixes=CHECK,ZERO %s
// RUN: %clang_cc1 -xc++ %s -fptrauth-function-pointer-type-discrimination -triple arm64-apple-ios13 -fptrauth-calls -fptrauth-intrinsics -disable-llvm-passes -emit-llvm -o- | FileCheck --check-prefixes=CHECK,TYPE,CHECK-CXX %s
// RUN: %clang_cc1 -xc++ %s -fptrauth-function-pointer-type-discrimination -triple aarch64-linux-gnu -fptrauth-calls -fptrauth-intrinsics -disable-llvm-passes -emit-llvm -o- | FileCheck --check-prefixes=CHECK,TYPE,CHECK-CXX %s
#ifdef __cplusplus
extern "C" {
#endif
void (*fptr)(void);
void (* __ptrauth(0, 0, 42) f2ptr_42_discm)(int);
void f(int);
void (* const __ptrauth(0, 0, 42) f_const_ptr)(int) = &f;
// CHECK-LABEL: define {{.*}}void @test_assign_to_qualified
void test_assign_to_qualified() {
f2ptr_42_discm = (void (*)(int))fptr;
// CHECK: [[ENTRY:.*]]:{{$}}
// CHECK: [[FPTR:%.*]] = load ptr, ptr @fptr
// CHECK-NEXT: [[CMP:%.*]] = icmp ne ptr [[FPTR]], null
// TYPE-NEXT: br i1 [[CMP]], label %[[RESIGN1:.*]], label %[[JOIN1:.*]]
// ZERO-NEXT: br i1 [[CMP]], label %[[RESIGN2:.*]], label %[[JOIN2:.*]]
// TYPE: [[RESIGN1]]:
// TYPE-NEXT: [[FPTR2:%.*]] = ptrtoint ptr [[FPTR]] to i64
// TYPE-NEXT: [[FPTR4:%.*]] = call i64 @llvm.ptrauth.resign(i64 [[FPTR2]], i32 0, i64 18983, i32 0, i64 2712)
// TYPE-NEXT: [[FPTR5:%.*]] = inttoptr i64 [[FPTR4]] to ptr
// TYPE-NEXT: br label %[[JOIN1]]
// TYPE: [[JOIN1]]:
// TYPE-NEXT: [[FPTR6:%.*]] = phi ptr [ null, %[[ENTRY]] ], [ [[FPTR5]], %[[RESIGN1]] ]
// TYPE-NEXT: [[CMP:%.*]] = icmp ne ptr [[FPTR6]], null
// TYPE-NEXT: br i1 [[CMP]], label %[[RESIGN2:.*]], label %[[JOIN2:.*]]
// CHECK: [[RESIGN2]]:
// TYPE-NEXT: [[FPTR7:%.*]] = ptrtoint ptr [[FPTR6]] to i64
// TYPE-NEXT: [[FPTR8:%.*]] = call i64 @llvm.ptrauth.resign(i64 [[FPTR7]], i32 0, i64 2712, i32 0, i64 42)
// ZERO-NEXT: [[FPTR7:%.*]] = ptrtoint ptr [[FPTR]] to i64
// ZERO-NEXT: [[FPTR8:%.*]] = call i64 @llvm.ptrauth.resign(i64 [[FPTR7]], i32 0, i64 0, i32 0, i64 42)
// CHECK-NEXT: [[FPTR9:%.*]] = inttoptr i64 [[FPTR8]] to ptr
// CHECK-NEXT: br label %[[JOIN2]]
// CHECK: [[JOIN2]]
// TYPE-NEXT: [[FPTR10:%.*]] = phi ptr [ null, %[[JOIN1]] ], [ [[FPTR9]], %[[RESIGN2]] ]
// ZERO-NEXT: [[FPTR10:%.*]] = phi ptr [ null, %[[ENTRY]] ], [ [[FPTR9]], %[[RESIGN2]] ]
// CHECK-NEXT store void (i32)* [[FPTR10]], void (i32)** @f2ptr_42_discm
}
// CHECK-LABEL: define {{.*}}void @test_assign_from_qualified
void test_assign_from_qualified() {
fptr = (void (*)(void))f2ptr_42_discm;
// CHECK: [[ENTRY:.*]]:{{$}}
// CHECK: [[FPTR:%.*]] = load ptr, ptr @f2ptr_42_discm
// CHECK-NEXT: [[CMP:%.*]] = icmp ne ptr [[FPTR]], null
// TYPE-NEXT: br i1 [[CMP]], label %[[RESIGN1:.*]], label %[[JOIN1:.*]]
// ZERO-NEXT: br i1 [[CMP]], label %[[RESIGN2:.*]], label %[[JOIN2:.*]]
// TYPE: [[RESIGN1]]:
// TYPE-NEXT: [[FPTR1:%.*]] = ptrtoint ptr [[FPTR]] to i64
// TYPE-NEXT: [[FPTR2:%.*]] = call i64 @llvm.ptrauth.resign(i64 [[FPTR1]], i32 0, i64 42, i32 0, i64 2712)
// TYPE-NEXT: [[FPTR3:%.*]] = inttoptr i64 [[FPTR2]] to ptr
// TYPE-NEXT: br label %[[JOIN1]]
// TYPE: [[JOIN1]]:
// TYPE-NEXT: [[FPTR4:%.*]] = phi ptr [ null, %[[ENTRY]] ], [ [[FPTR3]], %[[RESIGN1]] ]
// TYPE-NEXT: [[CMP:%.*]] = icmp ne ptr [[FPTR4]], null
// TYPE-NEXT: br i1 [[CMP]], label %[[RESIGN2:.*]], label %[[JOIN2:.*]]
// CHECK: [[RESIGN2]]:
// TYPE-NEXT: [[FPTR6:%.*]] = ptrtoint ptr [[FPTR4]] to i64
// TYPE-NEXT: [[FPTR7:%.*]] = call i64 @llvm.ptrauth.resign(i64 [[FPTR6]], i32 0, i64 2712, i32 0, i64 18983)
// ZERO-NEXT: [[FPTR6:%.*]] = ptrtoint ptr [[FPTR]] to i64
// ZERO-NEXT: [[FPTR7:%.*]] = call i64 @llvm.ptrauth.resign(i64 [[FPTR6]], i32 0, i64 42, i32 0, i64 0)
// CHECK-NEXT: [[FPTR8:%.*]] = inttoptr i64 [[FPTR7]] to ptr
// CHECK-NEXT: br label %[[JOIN2]]
// CHECK: [[JOIN2]]
// TYPE-NEXT: [[FPTR9:%.*]] = phi ptr [ null, %[[JOIN1]] ], [ [[FPTR8]], %[[RESIGN2]] ]
// ZERO-NEXT: [[FPTR9:%.*]] = phi ptr [ null, %[[ENTRY]] ], [ [[FPTR8]], %[[RESIGN2]] ]
// CHECK-NEXT store void ()* [[FPTR10]], void ()** @f2ptr_42_discm
}
// CHECK-LABEL: define {{.*}}void @test_const_ptr_function_call()
void test_const_ptr_function_call(void) {
f_const_ptr(1);
// TYPE: call void ptrauth (ptr @f, i32 0, i64 2712)(i32 noundef 1) [ "ptrauth"(i32 0, i64 2712) ]
// ZERO: call void ptrauth (ptr @f, i32 0)(i32 noundef 1) [ "ptrauth"(i32 0, i64 0) ]
}
#ifdef __cplusplus
void (* get_fptr(void))(int);
void (* __ptrauth(0, 0, 42) f_const_ptr2)(int) = get_fptr();
void (* const __ptrauth(0, 1, 43) &f_ref)(int) = f_const_ptr2;
// CHECK-CXX-LABEL: define {{.*}}internal void @__cxx_global_var_init()
// CHECK-CXX: [[ENTRY:.*]]:
// CHECK-CXX: %[[CALL:.*]] = call ptr @get_fptr()
// CHECK-CXX: %[[V0:.*]] = icmp ne ptr %[[CALL]], null
// CHECK-CXX: br i1 %[[V0]], label %[[RESIGN_NONNULL:.*]], label %[[RESIGN_CONT:.*]]
// CHECK-CXX: [[RESIGN_NONNULL]]:
// CHECK-CXX: %[[V1:.*]] = ptrtoint ptr %[[CALL]] to i64
// CHECK-CXX: %[[V2:.*]] = call i64 @llvm.ptrauth.resign(i64 %[[V1]], i32 0, i64 2712, i32 0, i64 42)
// CHECK-CXX: %[[V3:.*]] = inttoptr i64 %[[V2]] to ptr
// CHECK-CXX: br label %[[RESIGN_CONT]]
// CHECK-CXX: [[RESIGN_CONT]]:
// CHECK-CXX: %[[V4:.*]] = phi ptr [ null, %[[ENTRY]] ], [ %[[V3]], %[[RESIGN_NONNULL]] ]
// CHECK-CXX: store ptr %[[V4]], ptr @f_const_ptr2, align 8
// CHECK-CXX-LABEL: define {{.*}}internal void @__cxx_global_var_init.1()
// CHECK-CXX: [[ENTRY:.*]]:
// CHECK-CXX: %[[V0:.*]] = load ptr, ptr @f_const_ptr2, align 8
// CHECK-CXX: %[[V1:.*]] = call i64 @llvm.ptrauth.blend(i64 ptrtoint (ptr @_ZGR5f_ref_ to i64), i64 43)
// CHECK-CXX: %[[V2:.*]] = icmp ne ptr %[[V0]], null
// CHECK-CXX: br i1 %[[V2]], label %[[RESIGN_NONNULL:.*]], label %[[RESIGN_CONT:.*]]
// CHECK-CXX: [[RESIGN_NONNULL]]:
// CHECK-CXX: %[[V3:.*]] = ptrtoint ptr %[[V0]] to i64
// CHECK-CXX: %[[V4:.*]] = call i64 @llvm.ptrauth.resign(i64 %[[V3]], i32 0, i64 42, i32 0, i64 %[[V1]])
// CHECK-CXX: %[[V5:.*]] = inttoptr i64 %[[V4]] to ptr
// CHECK-CXX: br label %[[RESIGN_CONT]]
// CHECK-CXX: [[RESIGN_CONT]]:
// CHECK-CXX: %[[V6:.*]] = phi ptr [ null, %[[ENTRY]] ], [ %[[V5]], %[[RESIGN_NONNULL]] ]
// CHECK-CXX: store ptr %[[V6]], ptr @_ZGR5f_ref_, align 8
// CHECK-CXX: store ptr @_ZGR5f_ref_, ptr @f_ref, align 8
// CHECK-CXX-LABEL: define {{.*}}void @test_const_ptr_ref_function_call()
void test_const_ptr_ref_function_call(void) {
f_ref(1);
// CHECK-CXX: %[[V0:.*]] = load ptr, ptr @f_ref, align 8
// CHECK-CXX: %[[V1:.*]] = load ptr, ptr %[[V0]], align 8
// CHECK-CXX: %[[V2:.*]] = ptrtoint ptr %[[V0]] to i64
// CHECK-CXX: %[[V3:.*]] = call i64 @llvm.ptrauth.blend(i64 %[[V2]], i64 43)
// CHECK-CXX: call void %[[V1]](i32 noundef 1) [ "ptrauth"(i32 0, i64 %[[V3]]) ]
}
}
#endif
Reference in New Issue View Git Blame Copy Permalink