aboutsummaryrefslogtreecommitdiff
path: root/llvm/test/Instrumentation/AllocToken/invoke.ll
blob: 347c99a2e8f8d8d07369c3c3fdf793d1f6e75399 (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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
; RUN: opt < %s -passes=inferattrs,alloc-token -alloc-token-mode=increment -S | FileCheck %s

target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"

define ptr @test_invoke_malloc() sanitize_alloc_token personality ptr @__gxx_personality_v0 {
; CHECK-LABEL: define ptr @test_invoke_malloc(
; CHECK-SAME: ) #[[ATTR0:[0-9]+]] personality ptr @__gxx_personality_v0 {
; CHECK-NEXT:  [[ENTRY:.*:]]
; CHECK-NEXT:    [[TMP0:%.*]] = invoke ptr @__alloc_token_malloc(i64 64, i64 0)
; CHECK-NEXT:            to label %[[NORMAL:.*]] unwind label %[[CLEANUP:.*]]
; CHECK:       [[NORMAL]]:
; CHECK-NEXT:    ret ptr [[TMP0]]
; CHECK:       [[CLEANUP]]:
; CHECK-NEXT:    [[LP:%.*]] = landingpad { ptr, i32 }
; CHECK-NEXT:            cleanup
; CHECK-NEXT:    ret ptr null
;
entry:
  %ptr = invoke ptr @malloc(i64 64) to label %normal unwind label %cleanup

normal:
  ret ptr %ptr

cleanup:
  %lp = landingpad { ptr, i32 } cleanup
  ret ptr null
}

define ptr @test_invoke_operator_new() sanitize_alloc_token personality ptr @__gxx_personality_v0 {
; CHECK-LABEL: define ptr @test_invoke_operator_new(
; CHECK-SAME: ) #[[ATTR0]] personality ptr @__gxx_personality_v0 {
; CHECK-NEXT:  [[ENTRY:.*:]]
; CHECK-NEXT:    [[TMP0:%.*]] = invoke ptr @__alloc_token__Znwm(i64 32, i64 1)
; CHECK-NEXT:            to label %[[NORMAL:.*]] unwind label %[[CLEANUP:.*]]
; CHECK:       [[NORMAL]]:
; CHECK-NEXT:    ret ptr [[TMP0]]
; CHECK:       [[CLEANUP]]:
; CHECK-NEXT:    [[LP:%.*]] = landingpad { ptr, i32 }
; CHECK-NEXT:            cleanup
; CHECK-NEXT:    ret ptr null
;
entry:
  %ptr = invoke ptr @_Znwm(i64 32) to label %normal unwind label %cleanup

normal:
  ret ptr %ptr

cleanup:
  %lp = landingpad { ptr, i32 } cleanup
  ret ptr null
}

; Test complex exception flow with multiple invoke allocations
define ptr @test_complex_invoke_flow() sanitize_alloc_token personality ptr @__gxx_personality_v0 {
; CHECK-LABEL: define ptr @test_complex_invoke_flow(
; CHECK-SAME: ) #[[ATTR0]] personality ptr @__gxx_personality_v0 {
; CHECK-NEXT:  [[ENTRY:.*:]]
; CHECK-NEXT:    [[TMP0:%.*]] = invoke ptr @__alloc_token_malloc(i64 16, i64 2)
; CHECK-NEXT:            to label %[[FIRST_OK:.*]] unwind label %[[CLEANUP1:.*]]
; CHECK:       [[FIRST_OK]]:
; CHECK-NEXT:    [[TMP1:%.*]] = invoke ptr @__alloc_token__Znwm(i64 32, i64 3)
; CHECK-NEXT:            to label %[[SECOND_OK:.*]] unwind label %[[CLEANUP2:.*]]
; CHECK:       [[SECOND_OK]]:
; CHECK-NEXT:    ret ptr [[TMP0]]
; CHECK:       [[CLEANUP1]]:
; CHECK-NEXT:    [[LP1:%.*]] = landingpad { ptr, i32 }
; CHECK-NEXT:            cleanup
; CHECK-NEXT:    ret ptr null
; CHECK:       [[CLEANUP2]]:
; CHECK-NEXT:    [[LP2:%.*]] = landingpad { ptr, i32 }
; CHECK-NEXT:            cleanup
; CHECK-NEXT:    ret ptr null
;
entry:
  %ptr1 = invoke ptr @malloc(i64 16) to label %first_ok unwind label %cleanup1

first_ok:
  %ptr2 = invoke ptr @_Znwm(i64 32) to label %second_ok unwind label %cleanup2

second_ok:
  ret ptr %ptr1

cleanup1:
  %lp1 = landingpad { ptr, i32 } cleanup
  ret ptr null

cleanup2:
  %lp2 = landingpad { ptr, i32 } cleanup
  ret ptr null
}

; Test mixed call/invoke
define ptr @test_mixed_call_invoke() sanitize_alloc_token personality ptr @__gxx_personality_v0 {
; CHECK-LABEL: define ptr @test_mixed_call_invoke(
; CHECK-SAME: ) #[[ATTR0]] personality ptr @__gxx_personality_v0 {
; CHECK-NEXT:  [[ENTRY:.*:]]
; CHECK-NEXT:    [[TMP0:%.*]] = call ptr @__alloc_token_malloc(i64 8, i64 4)
; CHECK-NEXT:    [[TMP1:%.*]] = invoke ptr @__alloc_token_malloc(i64 16, i64 5)
; CHECK-NEXT:            to label %[[NORMAL:.*]] unwind label %[[CLEANUP:.*]]
; CHECK:       [[NORMAL]]:
; CHECK-NEXT:    ret ptr [[TMP0]]
; CHECK:       [[CLEANUP]]:
; CHECK-NEXT:    [[LP:%.*]] = landingpad { ptr, i32 }
; CHECK-NEXT:            cleanup
; CHECK-NEXT:    ret ptr null
;
entry:
  %ptr1 = call ptr @malloc(i64 8)

  %ptr2 = invoke ptr @malloc(i64 16) to label %normal unwind label %cleanup

normal:
  ret ptr %ptr1

cleanup:
  %lp = landingpad { ptr, i32 } cleanup
  ret ptr null
}

declare ptr @malloc(i64)
declare ptr @_Znwm(i64)
declare i32 @__gxx_personality_v0(...)