1 // RUN: %clang_cc1 -triple arm64-apple-ios -fptrauth-intrinsics -emit-llvm %s -o - | FileCheck %s
2 // RUN: %clang_cc1 -triple aarch64-elf -fptrauth-intrinsics -emit-llvm %s -o - | FileCheck %s
3 //
4 // RUN: %clang_cc1 -triple arm64-apple-ios -fptrauth-intrinsics -emit-llvm %s -o - -fexperimental-new-constant-interpreter | FileCheck %s
5 // RUN: %clang_cc1 -triple aarch64-elf -fptrauth-intrinsics -emit-llvm %s -o - -fexperimental-new-constant-interpreter | FileCheck %s
6
7 void (*fnptr)(void);
8 long int_discriminator;
9 void *ptr_discriminator;
10 long signature;
11
12 // CHECK-LABEL: define {{.*}}void @test_auth()
test_auth()13 void test_auth() {
14 // CHECK: [[PTR:%.*]] = load ptr, ptr @fnptr,
15 // CHECK-NEXT: [[DISC0:%.*]] = load ptr, ptr @ptr_discriminator,
16 // CHECK-NEXT: [[T0:%.*]] = ptrtoint ptr [[PTR]] to i64
17 // CHECK-NEXT: [[DISC:%.*]] = ptrtoint ptr [[DISC0]] to i64
18 // CHECK-NEXT: [[T1:%.*]] = call i64 @llvm.ptrauth.auth(i64 [[T0]], i32 0, i64 [[DISC]])
19 // CHECK-NEXT: [[RESULT:%.*]] = inttoptr i64 [[T1]] to ptr
20 // CHECK-NEXT: store ptr [[RESULT]], ptr @fnptr,
21 fnptr = __builtin_ptrauth_auth(fnptr, 0, ptr_discriminator);
22 }
23
24 // CHECK-LABEL: define {{.*}}void @test_strip()
test_strip()25 void test_strip() {
26 // CHECK: [[PTR:%.*]] = load ptr, ptr @fnptr,
27 // CHECK-NEXT: [[T0:%.*]] = ptrtoint ptr [[PTR]] to i64
28 // CHECK-NEXT: [[T1:%.*]] = call i64 @llvm.ptrauth.strip(i64 [[T0]], i32 0)
29 // CHECK-NEXT: [[RESULT:%.*]] = inttoptr i64 [[T1]] to ptr
30 // CHECK-NEXT: store ptr [[RESULT]], ptr @fnptr,
31 fnptr = __builtin_ptrauth_strip(fnptr, 0);
32 }
33
34 // CHECK-LABEL: define {{.*}}void @test_sign_unauthenticated()
test_sign_unauthenticated()35 void test_sign_unauthenticated() {
36 // CHECK: [[PTR:%.*]] = load ptr, ptr @fnptr,
37 // CHECK-NEXT: [[DISC0:%.*]] = load ptr, ptr @ptr_discriminator,
38 // CHECK-NEXT: [[T0:%.*]] = ptrtoint ptr [[PTR]] to i64
39 // CHECK-NEXT: [[DISC:%.*]] = ptrtoint ptr [[DISC0]] to i64
40 // CHECK-NEXT: [[T1:%.*]] = call i64 @llvm.ptrauth.sign(i64 [[T0]], i32 0, i64 [[DISC]])
41 // CHECK-NEXT: [[RESULT:%.*]] = inttoptr i64 [[T1]] to ptr
42 // CHECK-NEXT: store ptr [[RESULT]], ptr @fnptr,
43 fnptr = __builtin_ptrauth_sign_unauthenticated(fnptr, 0, ptr_discriminator);
44 }
45
46 // CHECK-LABEL: define {{.*}}void @test_auth_and_resign()
test_auth_and_resign()47 void test_auth_and_resign() {
48 // CHECK: [[PTR:%.*]] = load ptr, ptr @fnptr,
49 // CHECK-NEXT: [[DISC0:%.*]] = load ptr, ptr @ptr_discriminator,
50 // CHECK-NEXT: [[T0:%.*]] = ptrtoint ptr [[PTR]] to i64
51 // CHECK-NEXT: [[DISC:%.*]] = ptrtoint ptr [[DISC0]] to i64
52 // CHECK-NEXT: [[T1:%.*]] = call i64 @llvm.ptrauth.resign(i64 [[T0]], i32 0, i64 [[DISC]], i32 3, i64 15)
53 // CHECK-NEXT: [[RESULT:%.*]] = inttoptr i64 [[T1]] to ptr
54 // CHECK-NEXT: store ptr [[RESULT]], ptr @fnptr,
55 fnptr = __builtin_ptrauth_auth_and_resign(fnptr, 0, ptr_discriminator, 3, 15);
56 }
57
58 // CHECK-LABEL: define {{.*}}void @test_blend_discriminator()
test_blend_discriminator()59 void test_blend_discriminator() {
60 // CHECK: [[PTR:%.*]] = load ptr, ptr @fnptr,
61 // CHECK-NEXT: [[DISC:%.*]] = load i64, ptr @int_discriminator,
62 // CHECK-NEXT: [[T0:%.*]] = ptrtoint ptr [[PTR]] to i64
63 // CHECK-NEXT: [[RESULT:%.*]] = call i64 @llvm.ptrauth.blend(i64 [[T0]], i64 [[DISC]])
64 // CHECK-NEXT: store i64 [[RESULT]], ptr @int_discriminator,
65 int_discriminator = __builtin_ptrauth_blend_discriminator(fnptr, int_discriminator);
66 }
67
68 // CHECK-LABEL: define {{.*}}void @test_sign_generic_data()
test_sign_generic_data()69 void test_sign_generic_data() {
70 // CHECK: [[PTR:%.*]] = load ptr, ptr @fnptr,
71 // CHECK-NEXT: [[DISC0:%.*]] = load ptr, ptr @ptr_discriminator,
72 // CHECK-NEXT: [[T0:%.*]] = ptrtoint ptr [[PTR]] to i64
73 // CHECK-NEXT: [[DISC:%.*]] = ptrtoint ptr [[DISC0]] to i64
74 // CHECK-NEXT: [[RESULT:%.*]] = call i64 @llvm.ptrauth.sign.generic(i64 [[T0]], i64 [[DISC]])
75 // CHECK-NEXT: store i64 [[RESULT]], ptr @signature,
76 signature = __builtin_ptrauth_sign_generic_data(fnptr, ptr_discriminator);
77 }
78
79 // CHECK-LABEL: define {{.*}}void @test_string_discriminator()
test_string_discriminator()80 void test_string_discriminator() {
81 // CHECK: [[X:%.*]] = alloca i32
82
83 // Check a couple of random discriminators used by Swift.
84
85 // CHECK: store i32 58298, ptr [[X]],
86 int x = __builtin_ptrauth_string_discriminator("InitializeWithCopy");
87
88 // CHECK: store i32 9112, ptr [[X]],
89 x = __builtin_ptrauth_string_discriminator("DestroyArray");
90 }
91