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