clang/test/CodeGenCXX/ptrauth-apple-kext-indirect-call-2.cpp

   1 // RUN: %clang_cc1 -triple arm64-apple-ios -fptrauth-calls -fapple-kext -fno-rtti -emit-llvm -o - %s | FileCheck %s
   2
   3 // CHECK: @_ZTV1A = unnamed_addr constant { [4 x ptr] } { [4 x ptr] [ptr null, ptr null, ptr ptrauth (ptr @_ZNK1A3abcEv, i32 0, i64 12401, ptr getelementptr inbounds ({ [4 x ptr] }, ptr @_ZTV1A, i32 0, i32 0, i32 2)), ptr null] }, align 8
   4 // CHECK: @_ZTV4Base = unnamed_addr constant { [4 x ptr] } { [4 x ptr] [ptr null, ptr null, ptr ptrauth (ptr @_ZNK4Base3abcEv, i32 0, i64 64320, ptr getelementptr inbounds ({ [4 x ptr] }, ptr @_ZTV4Base, i32 0, i32 0, i32 2)), ptr null] }, align 8
   5 // CHECK: @_ZTV8Derived2 = unnamed_addr constant { [5 x ptr] } { [5 x ptr] [ptr null, ptr null, ptr null, ptr ptrauth (ptr @_ZNK8Derived23efgEv, i32 0, i64 36603, ptr getelementptr inbounds ({ [5 x ptr] }, ptr @_ZTV8Derived2, i32 0, i32 0, i32 3)), ptr null] }, align 8
   6 // CHECK: @_ZTV2D2 = unnamed_addr constant { [5 x ptr] } { [5 x ptr] [ptr null, ptr null, ptr null, ptr ptrauth (ptr @_ZNK2D23abcEv, i32 0, i64 20222, ptr getelementptr inbounds ({ [5 x ptr] }, ptr @_ZTV2D2, i32 0, i32 0, i32 3)), ptr null] }, align 8
   7
   8 struct A {
   9   virtual const char* abc(void) const;
  10 };
  11
  12 const char* A::abc(void) const {return "A"; };
  13
  14 struct B : virtual A {
  15   virtual void VF();
  16 };
  17
  18 void B::VF() {}
  19
  20 void FUNC(B* p) {
  21 // CHECK: [[T1:%.*]] = load ptr, ptr getelementptr inbounds (ptr, ptr @_ZTV1A, i64 2)
  22 // CHECK-NEXT:  [[BT1:%.*]] = call i64 @llvm.ptrauth.blend(i64 ptrtoint (ptr getelementptr inbounds (ptr, ptr @_ZTV1A, i64 2) to i64), i64 12401)
  23 // CHECK-NEXT:  [[T2:%.*]] = call noundef ptr [[T1]](ptr noundef {{.*}}) [ "ptrauth"(i32 0, i64 [[BT1]]) ]
  24   const char* c = p->A::abc();
  25 }
  26
  27
  28 // Test2
  29 struct Base { virtual char* abc(void) const; };
  30
  31 char* Base::abc() const { return 0; }
  32
  33 struct Derived : public Base {
  34 };
  35
  36 void FUNC1(Derived* p) {
  37 // CHECK: [[U1:%.*]] = load ptr, ptr getelementptr inbounds (ptr, ptr @_ZTV4Base, i64 2)
  38 // CHECK-NEXT: [[BU1:%.*]] = call i64 @llvm.ptrauth.blend(i64 ptrtoint (ptr getelementptr inbounds (ptr, ptr @_ZTV4Base, i64 2) to i64), i64 64320)
  39 // CHECK-NEXT:  [[U2:%.*]] = call noundef ptr [[U1]](ptr noundef {{.*}}) [ "ptrauth"(i32 0, i64 [[BU1]]) ]
  40   char* c = p->Base::abc();
  41 }
  42
  43
  44 // Test3
  45 struct Base2 { };
  46
  47 struct Derived2 : virtual Base2 {
  48   virtual char* efg(void) const;
  49 };
  50
  51 char* Derived2::efg(void) const { return 0; }
  52
  53 void FUNC2(Derived2* p) {
  54 // CHECK: [[V1:%.*]] = load ptr, ptr getelementptr inbounds (ptr, ptr @_ZTV8Derived2, i64 3)
  55 // CHECK-NEXT:  [[BV1:%.*]] = call i64 @llvm.ptrauth.blend(i64 ptrtoint (ptr getelementptr inbounds (ptr, ptr @_ZTV8Derived2, i64 3) to i64), i64 36603)
  56 // CHECK-NEXT:  [[V2:%.*]] = call noundef ptr [[V1]](ptr noundef {{.*}}) [ "ptrauth"(i32 0, i64 [[BV1]]) ]
  57   char* c = p->Derived2::efg();
  58 }
  59
  60 // Test4
  61 struct Base3 { };
  62
  63 struct D1 : virtual Base3 {
  64 };
  65
  66 struct D2 : virtual Base3 {
  67  virtual char *abc(void) const;
  68 };
  69
  70 struct Sub : D1, D2 {
  71 };
  72
  73 char* D2::abc(void) const { return 0; }
  74
  75 void FUNC3(Sub* p) {
  76 // CHECK: [[W1:%.*]] = load ptr, ptr getelementptr inbounds (ptr, ptr @_ZTV2D2, i64 3)
  77 // CHECK-NEXT:  [[BW1:%.*]] = call i64 @llvm.ptrauth.blend(i64 ptrtoint (ptr getelementptr inbounds (ptr, ptr @_ZTV2D2, i64 3) to i64), i64 20222)
  78 // CHECK-NEXT:  [[W2:%.*]] = call noundef ptr [[W1]](ptr noundef {{.*}}) [ "ptrauth"(i32 0, i64 [[BW1]]) ]
  79   char* c = p->D2::abc();
  80 }
  81
  82
  83 // Test4
  84 struct Base4 { virtual void abc(); };
  85
  86 void Base4::abc() {}
  87
  88 struct Derived4 : public Base4 {
  89   void abc() override;
  90 };
  91
  92 void Derived4::abc() {}
  93
  94 void FUNC4(Derived4* p) {
  95 // CHECK: %[[VTABLE:[a-z]+]] = load ptr, ptr %{{.*}}
  96 // CHECK: %[[T0:[0-9]+]] = ptrtoint ptr %[[VTABLE]] to i64
  97 // CHECK: %[[T3:[0-9]+]] = call i64 @llvm.ptrauth.auth(i64 %[[T0]], i32 2, i64 0)
  98 // CHECK: %[[T4:[0-9]+]] = inttoptr i64 %[[T3]] to ptr
  99 // CHECK: %[[VFN:[a-z]+]] = getelementptr inbounds ptr, ptr %[[T4]], i64 0
 100 // CHECK: %[[T5:[0-9]+]] = load ptr, ptr %[[VFN]]
 101 // CHECK: %[[T6:[0-9]+]] = ptrtoint ptr %[[VFN]] to i64
 102 // CHECK: %[[T7:[0-9]+]] = call i64 @llvm.ptrauth.blend(i64 %[[T6]], i64 426)
 103 // CHECK: call void %[[T5]](ptr noundef nonnull align {{[0-9]+}} dereferenceable(8) %{{.*}}) [ "ptrauth"(i32 0, i64 %[[T7]]) ]
 104   p->abc();
 105 }