aboutsummaryrefslogtreecommitdiff
path: root/clang/test/CodeGen/cfi-unchecked-callee-attribute-member-function.cpp
blob: 637b5201ad6b4839469c41df6c4f869b6d235168 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
// RUN: %clang_cc1 -triple x86_64-unknown-linux-gnu -emit-llvm -fsanitize=cfi-mfcall -o - %s -fvisibility=hidden | FileCheck %s

#define CFI_UNCHECKED_CALLEE __attribute__((cfi_unchecked_callee))

class A {};

// CHECK-LABEL: _Z14MemberFuncCallP1AMS_FvvE
void MemberFuncCall(A *s, void (A::*p)()) {
  // CHECK:      memptr.virtual:
  // CHECK-NEXT:   [[VTABLE:%.*]] = load ptr, ptr {{.*}}, align 8
  // CHECK-NEXT:   [[OFFSET:%.*]] = sub i64 %memptr.ptr, 1
  // CHECK-NEXT:   [[FUNC:%.*]] = getelementptr i8, ptr [[VTABLE]], i64 [[OFFSET]]
  // CHECK-NEXT:   [[VALID:%.*]] = call i1 @llvm.type.test(ptr [[FUNC]], metadata !"_ZTSM1AFvvE.virtual")
  // CHECK-NEXT:   [[FUNC:%.*]] = getelementptr i8, ptr [[VTABLE]], i64 [[OFFSET]]
  // CHECK-NEXT:   %memptr.virtualfn = load ptr, ptr [[FUNC]], align 8
  // CHECK-NEXT:   {{.*}}= call i1 @llvm.type.test(ptr [[VTABLE]], metadata !"all-vtables")
  // CHECK-NEXT:   br i1 [[VALID]], label %[[CONT:.*]], label %handler.cfi_check_fail{{.*}}

  // CHECK:      [[CONT]]:
  // CHECK-NEXT:   br label %memptr.end

  // CHECK:      memptr.nonvirtual:
  // CHECK-NEXT:   %memptr.nonvirtualfn = inttoptr i64 %memptr.ptr to ptr
  // CHECK-NEXT:   [[VALID:%.*]] = call i1 @llvm.type.test(ptr %memptr.nonvirtualfn, metadata !"_ZTSM1AFvvE")
  // CHECK-NEXT:   [[VALID2:%.*]] = or i1 false, [[VALID]]
  // CHECK-NEXT:   br i1 [[VALID2]], label %[[CONT2:.*]], label %handler.cfi_check_fail{{.*}}

  // CHECK:      [[CONT2]]:
  // CHECK-NEXT:   br label %memptr.end

  // CHECK:      memptr.end:
  // CHECK-NEXT:   {{.*}} = phi ptr [ %memptr.virtualfn, %[[CONT]] ], [ %memptr.nonvirtualfn, %[[CONT2]] ]
  (s->*p)();
}

// CHECK-LABEL: _Z19MemberFuncCallNoCFIP1AMS_FvvE
// CHECK-NOT: llvm.type.test
void MemberFuncCallNoCFI(A *s, void (CFI_UNCHECKED_CALLEE A::*p)()) {
  // CHECK:      memptr.virtual:
  // CHECK-NEXT:   [[VTABLE:%.*]] = load ptr, ptr {{.*}}, align 8
  // CHECK-NEXT:   [[OFFSET:%.*]] = sub i64 %memptr.ptr, 1
  // CHECK-NEXT:   [[FUNC:%.*]] = getelementptr i8, ptr [[VTABLE]], i64 [[OFFSET]]
  // CHECK-NEXT:   %memptr.virtualfn = load ptr, ptr [[FUNC]], align 8
  // CHECK-NEXT:   br label %memptr.end

  // CHECK:      memptr.nonvirtual:
  // CHECK-NEXT:   %memptr.nonvirtualfn = inttoptr i64 %memptr.ptr to ptr
  // CHECK-NEXT:   br label %memptr.end

  // CHECK:      memptr.end:
  // CHECK-NEXT:   {{.*}} = phi ptr [ %memptr.virtualfn, %memptr.virtual ], [ %memptr.nonvirtualfn, %memptr.nonvirtual ]
  (s->*p)();
}