1 // RUN: %clang_cc1 -fsanitize=alignment,null,object-size,shift-base,shift-exponent,return,signed-integer-overflow,vla-bound,float-cast-overflow,integer-divide-by-zero,bool,returns-nonnull-attribute,nonnull-attribute -fsanitize-recover=alignment,null,object-size,shift-base,shift-exponent,signed-integer-overflow,vla-bound,float-cast-overflow,integer-divide-by-zero,bool,returns-nonnull-attribute,nonnull-attribute -emit-llvm %s -o - -triple x86_64-linux-gnu | opt -instnamer -S | FileCheck %s --check-prefix=CHECK-COMMON --check-prefix=CHECK-UBSAN
2 // RUN: %clang_cc1 -fsanitize-trap=alignment,null,object-size,shift-base,shift-exponent,return,signed-integer-overflow,vla-bound,float-cast-overflow,integer-divide-by-zero,bool,returns-nonnull-attribute,nonnull-attribute -fsanitize-recover=alignment,null,object-size,shift-base,shift-exponent,signed-integer-overflow,vla-bound,float-cast-overflow,integer-divide-by-zero,bool,returns-nonnull-attribute,nonnull-attribute -fsanitize=alignment,null,object-size,shift-base,shift-exponent,return,signed-integer-overflow,vla-bound,float-cast-overflow,integer-divide-by-zero,bool,returns-nonnull-attribute,nonnull-attribute -fsanitize-recover=alignment,null,object-size,shift-base,shift-exponent,signed-integer-overflow,vla-bound,float-cast-overflow,integer-divide-by-zero,bool,returns-nonnull-attribute,nonnull-attribute -emit-llvm %s -o - -triple x86_64-linux-gnu | opt -instnamer -S | FileCheck %s --check-prefix=CHECK-COMMON --check-prefix=CHECK-TRAP
3 // RUN: %clang_cc1 -fsanitize=signed-integer-overflow -emit-llvm %s -o - -triple x86_64-linux-gnu | FileCheck %s --check-prefix=CHECK-OVERFLOW
4 
5 // CHECK-UBSAN: @[[INT:.*]] = private unnamed_addr constant { i16, i16, [6 x i8] } { i16 0, i16 11, [6 x i8] c"'int'\00" }
6 
7 // FIXME: When we only emit each type once, use [[INT]] more below.
8 // CHECK-UBSAN: @[[LINE_100:.*]] = private unnamed_addr global {{.*}}, i32 100, i32 5 {{.*}} @[[INT]], i8 2, i8 1
9 // CHECK-UBSAN: @[[LINE_200:.*]] = {{.*}}, i32 200, i32 10 {{.*}}, i8 2, i8 0
10 // CHECK-UBSAN: @[[LINE_300:.*]] = {{.*}}, i32 300, i32 12 {{.*}} @{{.*}}, {{.*}} @{{.*}}
11 // CHECK-UBSAN: @[[LINE_400:.*]] = {{.*}}, i32 400, i32 12 {{.*}} @{{.*}}, {{.*}} @{{.*}}
12 // CHECK-UBSAN: @[[LINE_500:.*]] = {{.*}}, i32 500, i32 10 {{.*}} @{{.*}}, i8 2, i8 0 }
13 // CHECK-UBSAN: @[[LINE_600:.*]] = {{.*}}, i32 600, i32 3 {{.*}} @{{.*}}, i8 2, i8 1 }
14 
15 // CHECK-UBSAN: @[[STRUCT_S:.*]] = private unnamed_addr constant { i16, i16, [11 x i8] } { i16 -1, i16 0, [11 x i8] c"'struct S'\00" }
16 
17 // CHECK-UBSAN: @[[LINE_700:.*]] = {{.*}}, i32 700, i32 14 {{.*}} @[[STRUCT_S]], i8 2, i8 3 }
18 // CHECK-UBSAN: @[[LINE_800:.*]] = {{.*}}, i32 800, i32 12 {{.*}} @{{.*}} }
19 // CHECK-UBSAN: @[[LINE_900:.*]] = {{.*}}, i32 900, i32 11 {{.*}} @{{.*}} }
20 // CHECK-UBSAN: @[[LINE_1000:.*]] = {{.*}}, i32 1000, i32 10 {{.*}} @{{.*}} }
21 // CHECK-UBSAN: @[[FP16:.*]] = private unnamed_addr constant { i16, i16, [9 x i8] } { i16 1, i16 16, [9 x i8] c"'__fp16'\00" }
22 // CHECK-UBSAN: @[[LINE_1100:.*]] = {{.*}}, i32 1100, i32 8 {{.*}} @{{.*}} }
23 // CHECK-UBSAN: @[[LINE_1200:.*]] = {{.*}}, i32 1200, i32 10 {{.*}} @{{.*}} }
24 // CHECK-UBSAN: @[[LINE_1300:.*]] = {{.*}}, i32 1300, i32 10 {{.*}} @{{.*}} }
25 // CHECK-UBSAN: @[[LINE_1400:.*]] = {{.*}}, i32 1400, i32 10 {{.*}} @{{.*}} }
26 // Make sure we check the fp16 type_mismatch data so we can easily match the signed char float_cast_overflow
27 // CHECK-UBSAN: @[[LINE_1500:.*]] = {{.*}}, i32 1500, i32 10 {{.*}} @[[FP16]], {{.*}} }
28 // CHECK-UBSAN: @[[SCHAR:.*]] = private unnamed_addr constant { i16, i16, [14 x i8] } { i16 0, i16 7, [14 x i8] c"'signed char'\00" }
29 // CHECK-UBSAN: @[[LINE_1500:.*]] = {{.*}}, i32 1500, i32 10 {{.*}} @[[FP16]], {{.*}} }
30 // CHECK-UBSAN: @[[LINE_1600:.*]] = {{.*}}, i32 1600, i32 10 {{.*}} @{{.*}} }
31 
32 // PR6805
33 // CHECK-COMMON-LABEL: @foo
34 void foo() {
35   union { int i; } u;
36 
37   // CHECK-COMMON:      %[[I8PTR:.*]] = bitcast i32* %[[PTR:.*]] to i8*
38   // CHECK-COMMON-NEXT: %[[SIZE:.*]] = call i64 @llvm.objectsize.i64.p0i8(i8* %[[I8PTR]], i1 false, i1 false)
39   // CHECK-COMMON-NEXT: %[[OK:.*]] = icmp uge i64 %[[SIZE]], 4
40 
41   // CHECK-UBSAN: br i1 %[[OK]], {{.*}} !prof ![[WEIGHT_MD:.*]], !nosanitize
42   // CHECK-TRAP:  br i1 %[[OK]], {{.*}}
43 
44   // CHECK-UBSAN:      %[[ARG:.*]] = ptrtoint {{.*}} %[[PTR]] to i64
45   // CHECK-UBSAN-NEXT: call void @__ubsan_handle_type_mismatch_v1(i8* bitcast ({{.*}} @[[LINE_100]] to i8*), i64 %[[ARG]])
46 
47   // CHECK-TRAP:      call void @llvm.trap() [[NR_NUW:#[0-9]+]]
48   // CHECK-TRAP-NEXT: unreachable
49 #line 100
50   u.i=1;
51 }
52 
53 // CHECK-COMMON-LABEL: @bar
54 int bar(int *a) {
55   // CHECK-COMMON:      %[[SIZE:.*]] = call i64 @llvm.objectsize.i64
56   // CHECK-COMMON-NEXT: icmp uge i64 %[[SIZE]], 4
57 
58   // CHECK-COMMON:      %[[PTRINT:.*]] = ptrtoint
59   // CHECK-COMMON-NEXT: %[[MISALIGN:.*]] = and i64 %[[PTRINT]], 3
60   // CHECK-COMMON-NEXT: icmp eq i64 %[[MISALIGN]], 0
61 
62   // CHECK-UBSAN: call void @__ubsan_handle_type_mismatch_v1(i8* bitcast ({{.*}} @[[LINE_200]] to i8*), i64 %[[PTRINT]])
63 
64   // CHECK-TRAP:      call void @llvm.trap() [[NR_NUW]]
65   // CHECK-TRAP-NEXT: unreachable
66 
67 #line 200
68   return *a;
69 }
70 
71 // CHECK-UBSAN-LABEL: @addr_space
72 int addr_space(int __attribute__((address_space(256))) *a) {
73   // CHECK-UBSAN-NOT: __ubsan
74   return *a;
75 }
76 
77 // CHECK-COMMON-LABEL: @lsh_overflow
78 int lsh_overflow(int a, int b) {
79   // CHECK-COMMON:      %[[RHS_INBOUNDS:.*]] = icmp ule i32 %[[RHS:.*]], 31
80   // CHECK-COMMON-NEXT: br i1 %[[RHS_INBOUNDS]], label %[[CHECK_BB:.*]], label %[[CONT_BB:.*]],
81 
82   // CHECK-COMMON:      [[CHECK_BB]]:
83   // CHECK-COMMON-NEXT: %[[SHIFTED_OUT_WIDTH:.*]] = sub nuw nsw i32 31, %[[RHS]]
84   // CHECK-COMMON-NEXT: %[[SHIFTED_OUT:.*]] = lshr i32 %[[LHS:.*]], %[[SHIFTED_OUT_WIDTH]]
85   // CHECK-COMMON-NEXT: %[[NO_OVERFLOW:.*]] = icmp eq i32 %[[SHIFTED_OUT]], 0
86   // CHECK-COMMON-NEXT: br label %[[CONT_BB]]
87 
88   // CHECK-COMMON:      [[CONT_BB]]:
89   // CHECK-COMMON-NEXT: %[[VALID_BASE:.*]] = phi i1 [ true, {{.*}} ], [ %[[NO_OVERFLOW]], %[[CHECK_BB]] ]
90   // CHECK-COMMON-NEXT: %[[VALID:.*]] = and i1 %[[RHS_INBOUNDS]], %[[VALID_BASE]]
91 
92   // CHECK-UBSAN: br i1 %[[VALID]], {{.*}} !prof ![[WEIGHT_MD]]
93   // CHECK-TRAP:  br i1 %[[VALID]]
94 
95   // CHECK-UBSAN:      %[[ARG1:.*]] = zext
96   // CHECK-UBSAN-NEXT: %[[ARG2:.*]] = zext
97   // CHECK-UBSAN-NEXT: call void @__ubsan_handle_shift_out_of_bounds(i8* bitcast ({{.*}} @[[LINE_300]] to i8*), i64 %[[ARG1]], i64 %[[ARG2]])
98   // CHECK-UBSAN-NOT:  call void @__ubsan_handle_shift_out_of_bounds
99 
100   // CHECK-TRAP:      call void @llvm.trap() [[NR_NUW]]
101   // CHECK-TRAP:      unreachable
102   // CHECK-TRAP-NOT:  call void @llvm.trap()
103 
104   // CHECK-COMMON:      %[[RET:.*]] = shl i32 %[[LHS]], %[[RHS]]
105   // CHECK-COMMON-NEXT: ret i32 %[[RET]]
106 #line 300
107   return a << b;
108 }
109 
110 // CHECK-COMMON-LABEL: @rsh_inbounds
111 int rsh_inbounds(int a, int b) {
112   // CHECK-COMMON:      %[[INBOUNDS:.*]] = icmp ule i32 %[[RHS:.*]], 31
113   // CHECK-COMMON:      br i1 %[[INBOUNDS]]
114 
115   // CHECK-UBSAN:      %[[ARG1:.*]] = zext
116   // CHECK-UBSAN-NEXT: %[[ARG2:.*]] = zext
117   // CHECK-UBSAN-NEXT: call void @__ubsan_handle_shift_out_of_bounds(i8* bitcast ({{.*}} @[[LINE_400]] to i8*), i64 %[[ARG1]], i64 %[[ARG2]])
118 
119   // CHECK-TRAP:      call void @llvm.trap() [[NR_NUW]]
120   // CHECK-TRAP-NEXT: unreachable
121 
122   // CHECK-COMMON:      %[[RET:.*]] = ashr i32 {{.*}}, %[[RHS]]
123   // CHECK-COMMON-NEXT: ret i32 %[[RET]]
124 #line 400
125   return a >> b;
126 }
127 
128 // CHECK-COMMON-LABEL: @load
129 int load(int *p) {
130   // CHECK-UBSAN: call void @__ubsan_handle_type_mismatch_v1(i8* bitcast ({{.*}} @[[LINE_500]] to i8*), i64 %{{.*}})
131 
132   // CHECK-TRAP:      call void @llvm.trap() [[NR_NUW]]
133   // CHECK-TRAP-NEXT: unreachable
134 #line 500
135   return *p;
136 }
137 
138 // CHECK-COMMON-LABEL: @store
139 void store(int *p, int q) {
140   // CHECK-UBSAN: call void @__ubsan_handle_type_mismatch_v1(i8* bitcast ({{.*}} @[[LINE_600]] to i8*), i64 %{{.*}})
141 
142   // CHECK-TRAP:      call void @llvm.trap() [[NR_NUW]]
143   // CHECK-TRAP-NEXT: unreachable
144 #line 600
145   *p = q;
146 }
147 
148 struct S { int k; };
149 
150 // CHECK-COMMON-LABEL: @member_access
151 int *member_access(struct S *p) {
152   // CHECK-UBSAN: call void @__ubsan_handle_type_mismatch_v1(i8* bitcast ({{.*}} @[[LINE_700]] to i8*), i64 %{{.*}})
153 
154   // CHECK-TRAP:      call void @llvm.trap() [[NR_NUW]]
155   // CHECK-TRAP-NEXT: unreachable
156 #line 700
157   return &p->k;
158 }
159 
160 // CHECK-COMMON-LABEL: @signed_overflow
161 int signed_overflow(int a, int b) {
162   // CHECK-UBSAN:      %[[ARG1:.*]] = zext
163   // CHECK-UBSAN-NEXT: %[[ARG2:.*]] = zext
164   // CHECK-UBSAN-NEXT: call void @__ubsan_handle_add_overflow(i8* bitcast ({{.*}} @[[LINE_800]] to i8*), i64 %[[ARG1]], i64 %[[ARG2]])
165 
166   // CHECK-TRAP:      call void @llvm.trap() [[NR_NUW]]
167   // CHECK-TRAP-NEXT: unreachable
168 #line 800
169   return a + b;
170 }
171 
172 // CHECK-COMMON-LABEL: @no_return
173 int no_return() {
174   // Reaching the end of a noreturn function is fine in C.
175   // FIXME: If the user explicitly requests -fsanitize=return, we should catch
176   //        that here even though it's not undefined behavior.
177   // CHECK-COMMON-NOT: call
178   // CHECK-COMMON-NOT: unreachable
179   // CHECK-COMMON: ret i32
180 }
181 
182 // CHECK-UBSAN-LABEL: @vla_bound
183 void vla_bound(int n) {
184   // CHECK-UBSAN:      icmp sgt i32 %[[PARAM:.*]], 0
185   //
186   // CHECK-UBSAN:      %[[ARG:.*]] = zext i32 %[[PARAM]] to i64
187   // CHECK-UBSAN-NEXT: call void @__ubsan_handle_vla_bound_not_positive(i8* bitcast ({{.*}} @[[LINE_900]] to i8*), i64 %[[ARG]])
188 #line 900
189   int arr[n * 3];
190 }
191 
192 // CHECK-UBSAN-LABEL: @int_float_no_overflow
193 float int_float_no_overflow(__int128 n) {
194   // CHECK-UBSAN-NOT: call void @__ubsan_handle
195   return n;
196 }
197 
198 // CHECK-COMMON-LABEL: @int_float_overflow
199 float int_float_overflow(unsigned __int128 n) {
200   // This is 2**104. FLT_MAX is 2**128 - 2**104.
201   // CHECK-COMMON: %[[INBOUNDS:.*]] = icmp ule i128 %{{.*}}, -20282409603651670423947251286016
202   // CHECK-COMMON-NEXT: br i1 %[[INBOUNDS]]
203 
204   // CHECK-UBSAN: call void @__ubsan_handle_float_cast_overflow(i8* bitcast ({{.*}} @[[LINE_1000]] to i8*),
205 
206   // CHECK-TRAP:      call void @llvm.trap() [[NR_NUW]]
207   // CHECK-TRAP-NEXT: unreachable
208 #line 1000
209   return n;
210 }
211 
212 // CHECK-COMMON-LABEL: @int_fp16_overflow
213 void int_fp16_overflow(int n, __fp16 *p) {
214   // CHECK-COMMON: %[[GE:.*]] = icmp sge i32 %{{.*}}, -65504
215   // CHECK-COMMON: %[[LE:.*]] = icmp sle i32 %{{.*}}, 65504
216   // CHECK-COMMON: %[[INBOUNDS:.*]] = and i1 %[[GE]], %[[LE]]
217   // CHECK-COMMON-NEXT: br i1 %[[INBOUNDS]]
218 
219   // CHECK-UBSAN: call void @__ubsan_handle_float_cast_overflow(i8* bitcast ({{.*}} @[[LINE_1100]] to i8*),
220 
221   // CHECK-TRAP:      call void @llvm.trap() [[NR_NUW]]
222   // CHECK-TRAP-NEXT: unreachable
223 #line 1100
224   *p = n;
225 }
226 
227 // CHECK-COMMON-LABEL: @float_int_overflow
228 int float_int_overflow(float f) {
229   // CHECK-COMMON: %[[GE:.*]] = fcmp ogt float %[[F:.*]], 0xC1E0000020000000
230   // CHECK-COMMON: %[[LE:.*]] = fcmp olt float %[[F]], 0x41E0000000000000
231   // CHECK-COMMON: %[[INBOUNDS:.*]] = and i1 %[[GE]], %[[LE]]
232   // CHECK-COMMON-NEXT: br i1 %[[INBOUNDS]]
233 
234   // CHECK-UBSAN: %[[CAST:.*]] = bitcast float %[[F]] to i32
235   // CHECK-UBSAN: %[[ARG:.*]] = zext i32 %[[CAST]] to i64
236   // CHECK-UBSAN: call void @__ubsan_handle_float_cast_overflow(i8* bitcast ({{.*}} @[[LINE_1200]] to i8*), i64 %[[ARG]]
237 
238   // CHECK-TRAP:      call void @llvm.trap() [[NR_NUW]]
239   // CHECK-TRAP-NEXT: unreachable
240 #line 1200
241   return f;
242 }
243 
244 // CHECK-COMMON-LABEL: @long_double_int_overflow
245 int long_double_int_overflow(long double ld) {
246   // CHECK-UBSAN: alloca x86_fp80
247 
248   // CHECK-COMMON: %[[GE:.*]] = fcmp ogt x86_fp80 %[[F:.*]], 0xKC01E800000010000000
249   // CHECK-COMMON: %[[LE:.*]] = fcmp olt x86_fp80 %[[F]], 0xK401E800000000000000
250   // CHECK-COMMON: %[[INBOUNDS:.*]] = and i1 %[[GE]], %[[LE]]
251   // CHECK-COMMON-NEXT: br i1 %[[INBOUNDS]]
252 
253   // CHECK-UBSAN: store x86_fp80 %[[F]], x86_fp80* %[[ALLOCA:.*]], align 16, !nosanitize
254   // CHECK-UBSAN: %[[ARG:.*]] = ptrtoint x86_fp80* %[[ALLOCA]] to i64
255   // CHECK-UBSAN: call void @__ubsan_handle_float_cast_overflow(i8* bitcast ({{.*}} @[[LINE_1300]] to i8*), i64 %[[ARG]]
256 
257   // CHECK-TRAP:      call void @llvm.trap() [[NR_NUW]]
258   // CHECK-TRAP-NEXT: unreachable
259 #line 1300
260   return ld;
261 }
262 
263 // CHECK-COMMON-LABEL: @float_uint_overflow
264 unsigned float_uint_overflow(float f) {
265   // CHECK-COMMON: %[[GE:.*]] = fcmp ogt float %[[F:.*]], -1.{{0*}}e+00
266   // CHECK-COMMON: %[[LE:.*]] = fcmp olt float %[[F]], 0x41F0000000000000
267   // CHECK-COMMON: %[[INBOUNDS:.*]] = and i1 %[[GE]], %[[LE]]
268   // CHECK-COMMON-NEXT: br i1 %[[INBOUNDS]]
269 
270   // CHECK-UBSAN: call void @__ubsan_handle_float_cast_overflow(i8* bitcast ({{.*}} @[[LINE_1400]] to i8*),
271 
272   // CHECK-TRAP:      call void @llvm.trap() [[NR_NUW]]
273   // CHECK-TRAP-NEXT: unreachable
274 #line 1400
275   return f;
276 }
277 
278 // CHECK-COMMON-LABEL: @fp16_char_overflow
279 signed char fp16_char_overflow(__fp16 *p) {
280   // CHECK-COMMON: %[[GE:.*]] = fcmp ogt float %[[F:.*]], -1.29{{0*}}e+02
281   // CHECK-COMMON: %[[LE:.*]] = fcmp olt float %[[F]], 1.28{{0*}}e+02
282   // CHECK-COMMON: %[[INBOUNDS:.*]] = and i1 %[[GE]], %[[LE]]
283   // CHECK-COMMON-NEXT: br i1 %[[INBOUNDS]]
284 
285   // CHECK-UBSAN: call void @__ubsan_handle_float_cast_overflow(i8* bitcast ({{.*}} @[[LINE_1500]] to i8*),
286 
287   // CHECK-TRAP:      call void @llvm.trap() [[NR_NUW]]
288   // CHECK-TRAP-NEXT: unreachable
289 #line 1500
290   return *p;
291 }
292 
293 // CHECK-COMMON-LABEL: @float_float_overflow
294 float float_float_overflow(double f) {
295   // CHECK-COMMON: %[[F:.*]] = call double @llvm.fabs.f64(
296   // CHECK-COMMON: %[[GE:.*]] = fcmp ogt double %[[F]], 0x47EFFFFFE0000000
297   // CHECK-COMMON: %[[LE:.*]] = fcmp olt double %[[F]], 0x7FF0000000000000
298   // CHECK-COMMON: %[[OUTOFBOUNDS:.*]] = and i1 %[[GE]], %[[LE]]
299   // CHECK-COMMON: %[[INBOUNDS:.*]] = xor i1 %[[OUTOFBOUNDS]], true
300   // CHECK-COMMON-NEXT: br i1 %[[INBOUNDS]]
301 
302   // CHECK-UBSAN: call void @__ubsan_handle_float_cast_overflow(i8* bitcast ({{.*}} @[[LINE_1600]] to i8*),
303 
304   // CHECK-TRAP:      call void @llvm.trap() [[NR_NUW]]
305   // CHECK-TRAP-NEXT: unreachable
306 #line 1600
307   return f;
308 }
309 
310 // CHECK-COMMON-LABEL:   @int_divide_overflow
311 // CHECK-OVERFLOW-LABEL: @int_divide_overflow
312 int int_divide_overflow(int a, int b) {
313   // CHECK-COMMON:         %[[ZERO:.*]] = icmp ne i32 %[[B:.*]], 0
314   // CHECK-OVERFLOW-NOT:  icmp ne i32 %{{.*}}, 0
315 
316   // CHECK-COMMON:               %[[AOK:.*]] = icmp ne i32 %[[A:.*]], -2147483648
317   // CHECK-COMMON-NEXT:          %[[BOK:.*]] = icmp ne i32 %[[B]], -1
318   // CHECK-COMMON-NEXT:          %[[OVER:.*]] = or i1 %[[AOK]], %[[BOK]]
319   // CHECK-COMMON:         %[[OK:.*]] = and i1 %[[ZERO]], %[[OVER]]
320   // CHECK-COMMON:         br i1 %[[OK]]
321 
322   // CHECK-OVERFLOW:      %[[AOK:.*]] = icmp ne i32 %[[A:.*]], -2147483648
323   // CHECK-OVERFLOW-NEXT: %[[BOK:.*]] = icmp ne i32 %[[B:.*]], -1
324   // CHECK-OVERFLOW-NEXT: %[[OK:.*]] = or i1 %[[AOK]], %[[BOK]]
325   // CHECK-OVERFLOW:      br i1 %[[OK]]
326 
327   // CHECK-TRAP: call void @llvm.trap() [[NR_NUW]]
328   // CHECK-TRAP: unreachable
329   return a / b;
330 
331   // CHECK-COMMON:          }
332   // CHECK-OVERFLOW: }
333 }
334 
335 // CHECK-COMMON-LABEL: @sour_bool
336 _Bool sour_bool(_Bool *p) {
337   // CHECK-COMMON: %[[OK:.*]] = icmp ule i8 {{.*}}, 1
338   // CHECK-COMMON: br i1 %[[OK]]
339 
340   // CHECK-UBSAN: call void @__ubsan_handle_load_invalid_value(i8* bitcast ({{.*}}), i64 {{.*}})
341 
342   // CHECK-TRAP: call void @llvm.trap() [[NR_NUW]]
343   // CHECK-TRAP: unreachable
344   return *p;
345 }
346 
347 // CHECK-COMMON-LABEL: @ret_nonnull
348 __attribute__((returns_nonnull))
349 int *ret_nonnull(int *a) {
350   // CHECK-COMMON: [[OK:%.*]] = icmp ne i32* {{.*}}, null
351   // CHECK-COMMON: br i1 [[OK]]
352 
353   // CHECK-UBSAN: call void @__ubsan_handle_nonnull_return
354 
355   // CHECK-TRAP: call void @llvm.trap() [[NR_NUW]]
356   // CHECK-TRAP: unreachable
357   return a;
358 }
359 
360 // CHECK-COMMON-LABEL: @call_decl_nonnull
361 __attribute__((nonnull)) void decl_nonnull(int *a);
362 void call_decl_nonnull(int *a) {
363   // CHECK-COMMON: [[OK:%.*]] = icmp ne i32* {{.*}}, null
364   // CHECK-COMMON: br i1 [[OK]]
365 
366   // CHECK-UBSAN: call void @__ubsan_handle_nonnull_arg
367 
368   // CHECK-TRAP: call void @llvm.trap() [[NR_NUW]]
369   // CHECK-TRAP: unreachable
370   decl_nonnull(a);
371 }
372 
373 extern void *memcpy (void *, const void *, unsigned) __attribute__((nonnull(1, 2)));
374 
375 // CHECK-COMMON-LABEL: @call_memcpy_nonnull
376 void call_memcpy_nonnull(void *p, void *q, int sz) {
377   // CHECK-COMMON: icmp ne i8* {{.*}}, null
378   // CHECK-UBSAN: call void @__ubsan_handle_nonnull_arg
379   // CHECK-TRAP: call void @llvm.trap()
380 
381   // CHECK-COMMON: icmp ne i8* {{.*}}, null
382   // CHECK-UBSAN: call void @__ubsan_handle_nonnull_arg
383   // CHECK-TRAP: call void @llvm.trap()
384   memcpy(p, q, sz);
385 }
386 
387 extern void *memmove (void *, const void *, unsigned) __attribute__((nonnull(1, 2)));
388 
389 // CHECK-COMMON-LABEL: @call_memmove_nonnull
390 void call_memmove_nonnull(void *p, void *q, int sz) {
391   // CHECK-COMMON: icmp ne i8* {{.*}}, null
392   // CHECK-UBSAN: call void @__ubsan_handle_nonnull_arg
393   // CHECK-TRAP: call void @llvm.trap()
394 
395   // CHECK-COMMON: icmp ne i8* {{.*}}, null
396   // CHECK-UBSAN: call void @__ubsan_handle_nonnull_arg
397   // CHECK-TRAP: call void @llvm.trap()
398   memmove(p, q, sz);
399 }
400 
401 // CHECK-COMMON-LABEL: @call_nonnull_variadic
402 __attribute__((nonnull)) void nonnull_variadic(int a, ...);
403 void call_nonnull_variadic(int a, int *b) {
404   // CHECK-COMMON: [[OK:%.*]] = icmp ne i32* {{.*}}, null
405   // CHECK-COMMON: br i1 [[OK]]
406 
407   // CHECK-UBSAN: call void @__ubsan_handle_nonnull_arg
408   // CHECK-UBSAN-NOT: __ubsan_handle_nonnull_arg
409 
410   // CHECK-COMMON: call void (i32, ...) @nonnull_variadic
411   nonnull_variadic(a, b);
412 }
413 
414 // CHECK-UBSAN: ![[WEIGHT_MD]] = !{!"branch_weights", i32 1048575, i32 1}
415 
416 // CHECK-TRAP: attributes [[NR_NUW]] = { noreturn nounwind }
417