1 // RUN: %clang_cc1 -std=c++11 -triple x86_64-apple-darwin10 -emit-llvm -o - %s -fsanitize=alignment | FileCheck %s --check-prefixes=CHECK,ALIGN
2 // RUN: %clang_cc1 -std=c++11 -triple x86_64-apple-darwin10 -emit-llvm -o - %s -fsanitize=null | FileCheck %s --check-prefixes=CHECK,NULL
3 // RUN: %clang_cc1 -std=c++11 -triple x86_64-apple-darwin10 -emit-llvm -o - %s -fsanitize=alignment,null -DCHECK_LAMBDA | FileCheck %s --check-prefixes=LAMBDA
4
5 // CHECK-LABEL: define void @_Z22load_non_null_pointersv
load_non_null_pointers()6 void load_non_null_pointers() {
7 int var;
8 var = *&var;
9
10 int arr[1];
11 arr[0] = arr[0];
12
13 char c = "foo"[0];
14
15 // CHECK-NOT: and i64 {{.*}}, !nosanitize
16 // CHECK-NOT: icmp ne {{.*}}, null, !nosanitize
17 // CHECK: ret void
18 }
19
20 // CHECK-LABEL: define void @_Z31use_us16_aligned_array_elementsv
use_us16_aligned_array_elements()21 void use_us16_aligned_array_elements() {
22 static const unsigned short Arr[] = {0, 1, 2};
23 auto use_array = [](const unsigned short(&X)[3]) -> void {};
24 use_array(Arr);
25
26 // CHECK-NOT: br i1 true
27 // ALIGN-NOT: call void @__ubsan_handle_type_mismatch
28 // CHECK: ret void
29 }
30
31 struct A {
32 int foo;
33
34 // CHECK-LABEL: define linkonce_odr void @_ZN1A10do_nothingEv
do_nothingA35 void do_nothing() {
36 // ALIGN: %[[THISINT1:[0-9]+]] = ptrtoint %struct.A* %{{.*}} to i64, !nosanitize
37 // ALIGN: and i64 %[[THISINT1]], 3, !nosanitize
38 // NULL: icmp ne %struct.A* %[[THIS1:[a-z0-9]+]], null, !nosanitize
39 // NULL: ptrtoint %struct.A* %[[THIS1]] to i64, !nosanitize
40 // CHECK: call void @__ubsan_handle_type_mismatch
41 // CHECK-NOT: call void @__ubsan_handle_type_mismatch
42 // CHECK: ret void
43 }
44
45 #ifdef CHECK_LAMBDA
46 // LAMBDA-LABEL: define linkonce_odr void @_ZN1A22do_nothing_with_lambdaEv
do_nothing_with_lambdaA47 void do_nothing_with_lambda() {
48 // LAMBDA: icmp ne %struct.A* %[[THIS2:[a-z0-9]+]], null, !nosanitize
49 // LAMBDA: %[[THISINT2:[0-9]+]] = ptrtoint %struct.A* %[[THIS2]] to i64, !nosanitize
50 // LAMBDA: and i64 %[[THISINT2]], 3, !nosanitize
51 // LAMBDA: call void @__ubsan_handle_type_mismatch
52
53 auto f = [&] {
54 foo = 0;
55 };
56 f();
57
58 // LAMBDA-NOT: call void @__ubsan_handle_type_mismatch
59 // LAMBDA: ret void
60 }
61
62 // Check the IR for the lambda:
63 //
64 // LAMBDA-LABEL: define linkonce_odr void @_ZZN1A22do_nothing_with_lambdaEvENKUlvE_clEv
65 // LAMBDA: call void @__ubsan_handle_type_mismatch
66 // LAMBDA-NOT: call void @__ubsan_handle_type_mismatch
67 // LAMBDA: ret void
68 #endif
69
70 // CHECK-LABEL: define linkonce_odr i32 @_ZN1A11load_memberEv
load_memberA71 int load_member() {
72 // ALIGN: %[[THISINT3:[0-9]+]] = ptrtoint %struct.A* %{{.*}} to i64, !nosanitize
73 // ALIGN: and i64 %[[THISINT3]], 3, !nosanitize
74 // NULL: icmp ne %struct.A* %[[THIS3:[a-z0-9]+]], null, !nosanitize
75 // NULL: ptrtoint %struct.A* %[[THIS3]] to i64, !nosanitize
76 // CHECK: call void @__ubsan_handle_type_mismatch
77 // CHECK-NOT: call void @__ubsan_handle_type_mismatch
78 return foo;
79 // CHECK: ret i32
80 }
81
82 // CHECK-LABEL: define linkonce_odr i32 @_ZN1A11call_methodEv
call_methodA83 int call_method() {
84 // ALIGN: %[[THISINT4:[0-9]+]] = ptrtoint %struct.A* %{{.*}} to i64, !nosanitize
85 // ALIGN: and i64 %[[THISINT4]], 3, !nosanitize
86 // NULL: icmp ne %struct.A* %[[THIS4:[a-z0-9]+]], null, !nosanitize
87 // NULL: ptrtoint %struct.A* %[[THIS4]] to i64, !nosanitize
88 // CHECK: call void @__ubsan_handle_type_mismatch
89 // CHECK-NOT: call void @__ubsan_handle_type_mismatch
90 return load_member();
91 // CHECK: ret i32
92 }
93
94 // CHECK-LABEL: define linkonce_odr void @_ZN1A15assign_member_1Ev
assign_member_1A95 void assign_member_1() {
96 // ALIGN: %[[THISINT5:[0-9]+]] = ptrtoint %struct.A* %{{.*}} to i64, !nosanitize
97 // ALIGN: and i64 %[[THISINT5]], 3, !nosanitize
98 // NULL: icmp ne %struct.A* %[[THIS5:[a-z0-9]+]], null, !nosanitize
99 // NULL: ptrtoint %struct.A* %[[THIS5]] to i64, !nosanitize
100 // CHECK: call void @__ubsan_handle_type_mismatch
101 // CHECK-NOT: call void @__ubsan_handle_type_mismatch
102 foo = 0;
103 // CHECK: ret void
104 }
105
106 // CHECK-LABEL: define linkonce_odr void @_ZN1A15assign_member_2Ev
assign_member_2A107 void assign_member_2() {
108 // ALIGN: %[[THISINT6:[0-9]+]] = ptrtoint %struct.A* %{{.*}} to i64, !nosanitize
109 // ALIGN: and i64 %[[THISINT6]], 3, !nosanitize
110 // NULL: icmp ne %struct.A* %[[THIS6:[a-z0-9]+]], null, !nosanitize
111 // NULL: ptrtoint %struct.A* %[[THIS6]] to i64, !nosanitize
112 // CHECK: call void @__ubsan_handle_type_mismatch
113 // CHECK-NOT: call void @__ubsan_handle_type_mismatch
114 (__extension__ (this))->foo = 0;
115 // CHECK: ret void
116 }
117
118 // CHECK-LABEL: define linkonce_odr void @_ZNK1A15assign_member_3Ev
assign_member_3A119 void assign_member_3() const {
120 // ALIGN: %[[THISINT7:[0-9]+]] = ptrtoint %struct.A* %{{.*}} to i64, !nosanitize
121 // ALIGN: and i64 %[[THISINT7]], 3, !nosanitize
122 // NULL: icmp ne %struct.A* %[[THIS7:[a-z0-9]+]], null, !nosanitize
123 // NULL: ptrtoint %struct.A* %[[THIS7]] to i64, !nosanitize
124 // CHECK: call void @__ubsan_handle_type_mismatch
125 // CHECK-NOT: call void @__ubsan_handle_type_mismatch
126 const_cast<A *>(this)->foo = 0;
127 // CHECK: ret void
128 }
129
130 // CHECK-LABEL: define linkonce_odr i32 @_ZN1A22call_through_referenceERS_
call_through_referenceA131 static int call_through_reference(A &a) {
132 // ALIGN: %[[OBJINT:[0-9]+]] = ptrtoint %struct.A* %{{.*}} to i64, !nosanitize
133 // ALIGN: and i64 %[[OBJINT]], 3, !nosanitize
134 // ALIGN: call void @__ubsan_handle_type_mismatch
135 // NULL-NOT: call void @__ubsan_handle_type_mismatch
136 return a.load_member();
137 // CHECK: ret i32
138 }
139
140 // CHECK-LABEL: define linkonce_odr i32 @_ZN1A20call_through_pointerEPS_
call_through_pointerA141 static int call_through_pointer(A *a) {
142 // CHECK: call void @__ubsan_handle_type_mismatch
143 return a->load_member();
144 // CHECK: ret i32
145 }
146 };
147
148 struct B {
operator A*B149 operator A*() const { return nullptr; }
150
151 // CHECK-LABEL: define linkonce_odr i32 @_ZN1B11load_memberEPS_
load_memberB152 static int load_member(B *bp) {
153 // Check &b before converting it to an A*.
154 // CHECK: call void @__ubsan_handle_type_mismatch
155 //
156 // Check the result of the conversion before using it.
157 // NULL: call void @__ubsan_handle_type_mismatch
158 //
159 // CHECK-NOT: call void @__ubsan_handle_type_mismatch
160 return static_cast<A *>(*bp)->load_member();
161 // CHECK: ret i32
162 }
163 };
164
165 struct Base {
166 int foo;
167
168 virtual int load_member_1() = 0;
169 };
170
171 struct Derived : public Base {
172 int bar;
173
174 // CHECK-LABEL: define linkonce_odr i32 @_ZN7Derived13load_member_2Ev
load_member_2Derived175 int load_member_2() {
176 // ALIGN: %[[THISINT8:[0-9]+]] = ptrtoint %struct.Derived* %{{.*}} to i64, !nosanitize
177 // ALIGN: and i64 %[[THISINT8]], 7, !nosanitize
178 // ALIGN: call void @__ubsan_handle_type_mismatch
179 // NULL: icmp ne %struct.Derived* %[[THIS8:[a-z0-9]+]], null, !nosanitize
180 // NULL: ptrtoint %struct.Derived* %[[THIS8]] to i64, !nosanitize
181 // CHECK: call void @__ubsan_handle_type_mismatch
182 //
183 // Check the result of the cast before using it.
184 // CHECK: call void @__ubsan_handle_type_mismatch
185 //
186 // CHECK-NOT: call void @__ubsan_handle_type_mismatch
187 return dynamic_cast<Base *>(this)->load_member_1();
188 // CHECK: ret i32
189 }
190
191 // CHECK-LABEL: define linkonce_odr i32 @_ZN7Derived13load_member_3Ev
load_member_3Derived192 int load_member_3() {
193 // ALIGN: %[[THISINT9:[0-9]+]] = ptrtoint %struct.Derived* %{{.*}} to i64, !nosanitize
194 // ALIGN: and i64 %[[THISINT9]], 7, !nosanitize
195 // ALIGN: call void @__ubsan_handle_type_mismatch
196 // ALIGN: call void @__ubsan_handle_type_mismatch
197 // NULL: icmp ne %struct.Derived* %[[THIS9:[a-z0-9]+]], null, !nosanitize
198 // NULL: ptrtoint %struct.Derived* %[[THIS9]] to i64, !nosanitize
199 // CHECK: call void @__ubsan_handle_type_mismatch
200 // CHECK-NOT: call void @__ubsan_handle_type_mismatch
201 return reinterpret_cast<Derived *>(static_cast<Base *>(this))->foo;
202 // CHECK: ret i32
203 }
204
205 // CHECK-LABEL: define linkonce_odr i32 @_ZN7Derived13load_member_1Ev
load_member_1Derived206 int load_member_1() override {
207 // ALIGN: %[[THISINT10:[0-9]+]] = ptrtoint %struct.Derived* %{{.*}} to i64, !nosanitize
208 // ALIGN: and i64 %[[THISINT10]], 7, !nosanitize
209 // ALIGN: call void @__ubsan_handle_type_mismatch
210 // NULL: icmp ne %struct.Derived* %[[THIS10:[a-z0-9]+]], null, !nosanitize
211 // NULL: ptrtoint %struct.Derived* %[[THIS10]] to i64, !nosanitize
212 // CHECK: call void @__ubsan_handle_type_mismatch
213 // CHECK-NOT: call void @__ubsan_handle_type_mismatch
214 return foo + bar;
215 // CHECK: ret i32
216 }
217 };
218
force_irgen()219 void force_irgen() {
220 A *a;
221 a->do_nothing();
222 #ifdef CHECK_LAMBDA
223 a->do_nothing_with_lambda();
224 #endif
225 a->load_member();
226 a->call_method();
227 a->assign_member_1();
228 a->assign_member_2();
229 a->assign_member_3();
230 A::call_through_reference(*a);
231 A::call_through_pointer(a);
232
233 B::load_member(nullptr);
234
235 Base *b = new Derived;
236 b->load_member_1();
237
238 Derived *d;
239 d->load_member_2();
240 d->load_member_3();
241
242 load_non_null_pointers();
243 use_us16_aligned_array_elements();
244 }
245