annotate gcc/c-family/c-ubsan.c @ 111:04ced10e8804

gcc 7
author kono
date Fri, 27 Oct 2017 22:46:09 +0900
parents
children 84e7813d76e9
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
111
kono
parents:
diff changeset
1 /* UndefinedBehaviorSanitizer, undefined behavior detector.
kono
parents:
diff changeset
2 Copyright (C) 2013-2017 Free Software Foundation, Inc.
kono
parents:
diff changeset
3 Contributed by Marek Polacek <polacek@redhat.com>
kono
parents:
diff changeset
4
kono
parents:
diff changeset
5 This file is part of GCC.
kono
parents:
diff changeset
6
kono
parents:
diff changeset
7 GCC is free software; you can redistribute it and/or modify it under
kono
parents:
diff changeset
8 the terms of the GNU General Public License as published by the Free
kono
parents:
diff changeset
9 Software Foundation; either version 3, or (at your option) any later
kono
parents:
diff changeset
10 version.
kono
parents:
diff changeset
11
kono
parents:
diff changeset
12 GCC is distributed in the hope that it will be useful, but WITHOUT ANY
kono
parents:
diff changeset
13 WARRANTY; without even the implied warranty of MERCHANTABILITY or
kono
parents:
diff changeset
14 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
kono
parents:
diff changeset
15 for more details.
kono
parents:
diff changeset
16
kono
parents:
diff changeset
17 You should have received a copy of the GNU General Public License
kono
parents:
diff changeset
18 along with GCC; see the file COPYING3. If not see
kono
parents:
diff changeset
19 <http://www.gnu.org/licenses/>. */
kono
parents:
diff changeset
20
kono
parents:
diff changeset
21 #include "config.h"
kono
parents:
diff changeset
22 #include "system.h"
kono
parents:
diff changeset
23 #include "coretypes.h"
kono
parents:
diff changeset
24 #include "tm.h"
kono
parents:
diff changeset
25 #include "c-family/c-common.h"
kono
parents:
diff changeset
26 #include "ubsan.h"
kono
parents:
diff changeset
27 #include "c-family/c-ubsan.h"
kono
parents:
diff changeset
28 #include "stor-layout.h"
kono
parents:
diff changeset
29 #include "builtins.h"
kono
parents:
diff changeset
30 #include "gimplify.h"
kono
parents:
diff changeset
31 #include "stringpool.h"
kono
parents:
diff changeset
32 #include "attribs.h"
kono
parents:
diff changeset
33 #include "asan.h"
kono
parents:
diff changeset
34
kono
parents:
diff changeset
35 /* Instrument division by zero and INT_MIN / -1. If not instrumenting,
kono
parents:
diff changeset
36 return NULL_TREE. */
kono
parents:
diff changeset
37
kono
parents:
diff changeset
38 tree
kono
parents:
diff changeset
39 ubsan_instrument_division (location_t loc, tree op0, tree op1)
kono
parents:
diff changeset
40 {
kono
parents:
diff changeset
41 tree t, tt;
kono
parents:
diff changeset
42 tree type = TREE_TYPE (op0);
kono
parents:
diff changeset
43
kono
parents:
diff changeset
44 /* At this point both operands should have the same type,
kono
parents:
diff changeset
45 because they are already converted to RESULT_TYPE.
kono
parents:
diff changeset
46 Use TYPE_MAIN_VARIANT since typedefs can confuse us. */
kono
parents:
diff changeset
47 gcc_assert (TYPE_MAIN_VARIANT (TREE_TYPE (op0))
kono
parents:
diff changeset
48 == TYPE_MAIN_VARIANT (TREE_TYPE (op1)));
kono
parents:
diff changeset
49
kono
parents:
diff changeset
50 op0 = unshare_expr (op0);
kono
parents:
diff changeset
51 op1 = unshare_expr (op1);
kono
parents:
diff changeset
52
kono
parents:
diff changeset
53 if (TREE_CODE (type) == INTEGER_TYPE
kono
parents:
diff changeset
54 && sanitize_flags_p (SANITIZE_DIVIDE))
kono
parents:
diff changeset
55 t = fold_build2 (EQ_EXPR, boolean_type_node,
kono
parents:
diff changeset
56 op1, build_int_cst (type, 0));
kono
parents:
diff changeset
57 else if (TREE_CODE (type) == REAL_TYPE
kono
parents:
diff changeset
58 && sanitize_flags_p (SANITIZE_FLOAT_DIVIDE))
kono
parents:
diff changeset
59 t = fold_build2 (EQ_EXPR, boolean_type_node,
kono
parents:
diff changeset
60 op1, build_real (type, dconst0));
kono
parents:
diff changeset
61 else
kono
parents:
diff changeset
62 return NULL_TREE;
kono
parents:
diff changeset
63
kono
parents:
diff changeset
64 /* We check INT_MIN / -1 only for signed types. */
kono
parents:
diff changeset
65 if (TREE_CODE (type) == INTEGER_TYPE
kono
parents:
diff changeset
66 && sanitize_flags_p (SANITIZE_DIVIDE)
kono
parents:
diff changeset
67 && !TYPE_UNSIGNED (type))
kono
parents:
diff changeset
68 {
kono
parents:
diff changeset
69 tree x;
kono
parents:
diff changeset
70 tt = fold_build2 (EQ_EXPR, boolean_type_node, unshare_expr (op1),
kono
parents:
diff changeset
71 build_int_cst (type, -1));
kono
parents:
diff changeset
72 x = fold_build2 (EQ_EXPR, boolean_type_node, op0,
kono
parents:
diff changeset
73 TYPE_MIN_VALUE (type));
kono
parents:
diff changeset
74 x = fold_build2 (TRUTH_AND_EXPR, boolean_type_node, x, tt);
kono
parents:
diff changeset
75 t = fold_build2 (TRUTH_OR_EXPR, boolean_type_node, t, x);
kono
parents:
diff changeset
76 }
kono
parents:
diff changeset
77
kono
parents:
diff changeset
78 /* If the condition was folded to 0, no need to instrument
kono
parents:
diff changeset
79 this expression. */
kono
parents:
diff changeset
80 if (integer_zerop (t))
kono
parents:
diff changeset
81 return NULL_TREE;
kono
parents:
diff changeset
82
kono
parents:
diff changeset
83 /* In case we have a SAVE_EXPR in a conditional context, we need to
kono
parents:
diff changeset
84 make sure it gets evaluated before the condition. */
kono
parents:
diff changeset
85 t = fold_build2 (COMPOUND_EXPR, TREE_TYPE (t), unshare_expr (op0), t);
kono
parents:
diff changeset
86 t = fold_build2 (COMPOUND_EXPR, TREE_TYPE (t), unshare_expr (op1), t);
kono
parents:
diff changeset
87 if (flag_sanitize_undefined_trap_on_error)
kono
parents:
diff changeset
88 tt = build_call_expr_loc (loc, builtin_decl_explicit (BUILT_IN_TRAP), 0);
kono
parents:
diff changeset
89 else
kono
parents:
diff changeset
90 {
kono
parents:
diff changeset
91 tree data = ubsan_create_data ("__ubsan_overflow_data", 1, &loc,
kono
parents:
diff changeset
92 ubsan_type_descriptor (type), NULL_TREE,
kono
parents:
diff changeset
93 NULL_TREE);
kono
parents:
diff changeset
94 data = build_fold_addr_expr_loc (loc, data);
kono
parents:
diff changeset
95 enum built_in_function bcode
kono
parents:
diff changeset
96 = (flag_sanitize_recover & SANITIZE_DIVIDE)
kono
parents:
diff changeset
97 ? BUILT_IN_UBSAN_HANDLE_DIVREM_OVERFLOW
kono
parents:
diff changeset
98 : BUILT_IN_UBSAN_HANDLE_DIVREM_OVERFLOW_ABORT;
kono
parents:
diff changeset
99 tt = builtin_decl_explicit (bcode);
kono
parents:
diff changeset
100 op0 = unshare_expr (op0);
kono
parents:
diff changeset
101 op1 = unshare_expr (op1);
kono
parents:
diff changeset
102 tt = build_call_expr_loc (loc, tt, 3, data, ubsan_encode_value (op0),
kono
parents:
diff changeset
103 ubsan_encode_value (op1));
kono
parents:
diff changeset
104 }
kono
parents:
diff changeset
105 t = fold_build3 (COND_EXPR, void_type_node, t, tt, void_node);
kono
parents:
diff changeset
106
kono
parents:
diff changeset
107 return t;
kono
parents:
diff changeset
108 }
kono
parents:
diff changeset
109
kono
parents:
diff changeset
110 /* Instrument left and right shifts. */
kono
parents:
diff changeset
111
kono
parents:
diff changeset
112 tree
kono
parents:
diff changeset
113 ubsan_instrument_shift (location_t loc, enum tree_code code,
kono
parents:
diff changeset
114 tree op0, tree op1)
kono
parents:
diff changeset
115 {
kono
parents:
diff changeset
116 tree t, tt = NULL_TREE;
kono
parents:
diff changeset
117 tree type0 = TREE_TYPE (op0);
kono
parents:
diff changeset
118 tree type1 = TREE_TYPE (op1);
kono
parents:
diff changeset
119 if (!INTEGRAL_TYPE_P (type0))
kono
parents:
diff changeset
120 return NULL_TREE;
kono
parents:
diff changeset
121
kono
parents:
diff changeset
122 tree op1_utype = unsigned_type_for (type1);
kono
parents:
diff changeset
123 HOST_WIDE_INT op0_prec = TYPE_PRECISION (type0);
kono
parents:
diff changeset
124 tree uprecm1 = build_int_cst (op1_utype, op0_prec - 1);
kono
parents:
diff changeset
125
kono
parents:
diff changeset
126 op0 = unshare_expr (op0);
kono
parents:
diff changeset
127 op1 = unshare_expr (op1);
kono
parents:
diff changeset
128
kono
parents:
diff changeset
129 t = fold_convert_loc (loc, op1_utype, op1);
kono
parents:
diff changeset
130 t = fold_build2 (GT_EXPR, boolean_type_node, t, uprecm1);
kono
parents:
diff changeset
131
kono
parents:
diff changeset
132 /* If this is not a signed operation, don't perform overflow checks.
kono
parents:
diff changeset
133 Also punt on bit-fields. */
kono
parents:
diff changeset
134 if (TYPE_OVERFLOW_WRAPS (type0)
kono
parents:
diff changeset
135 || GET_MODE_BITSIZE (TYPE_MODE (type0)) != TYPE_PRECISION (type0)
kono
parents:
diff changeset
136 || !sanitize_flags_p (SANITIZE_SHIFT_BASE))
kono
parents:
diff changeset
137 ;
kono
parents:
diff changeset
138
kono
parents:
diff changeset
139 /* For signed x << y, in C99/C11, the following:
kono
parents:
diff changeset
140 (unsigned) x >> (uprecm1 - y)
kono
parents:
diff changeset
141 if non-zero, is undefined. */
kono
parents:
diff changeset
142 else if (code == LSHIFT_EXPR && flag_isoc99 && cxx_dialect < cxx11)
kono
parents:
diff changeset
143 {
kono
parents:
diff changeset
144 tree x = fold_build2 (MINUS_EXPR, op1_utype, uprecm1,
kono
parents:
diff changeset
145 fold_convert (op1_utype, unshare_expr (op1)));
kono
parents:
diff changeset
146 tt = fold_convert_loc (loc, unsigned_type_for (type0), op0);
kono
parents:
diff changeset
147 tt = fold_build2 (RSHIFT_EXPR, TREE_TYPE (tt), tt, x);
kono
parents:
diff changeset
148 tt = fold_build2 (NE_EXPR, boolean_type_node, tt,
kono
parents:
diff changeset
149 build_int_cst (TREE_TYPE (tt), 0));
kono
parents:
diff changeset
150 }
kono
parents:
diff changeset
151
kono
parents:
diff changeset
152 /* For signed x << y, in C++11 and later, the following:
kono
parents:
diff changeset
153 x < 0 || ((unsigned) x >> (uprecm1 - y))
kono
parents:
diff changeset
154 if > 1, is undefined. */
kono
parents:
diff changeset
155 else if (code == LSHIFT_EXPR && cxx_dialect >= cxx11)
kono
parents:
diff changeset
156 {
kono
parents:
diff changeset
157 tree x = fold_build2 (MINUS_EXPR, op1_utype, uprecm1,
kono
parents:
diff changeset
158 fold_convert (op1_utype, unshare_expr (op1)));
kono
parents:
diff changeset
159 tt = fold_convert_loc (loc, unsigned_type_for (type0),
kono
parents:
diff changeset
160 unshare_expr (op0));
kono
parents:
diff changeset
161 tt = fold_build2 (RSHIFT_EXPR, TREE_TYPE (tt), tt, x);
kono
parents:
diff changeset
162 tt = fold_build2 (GT_EXPR, boolean_type_node, tt,
kono
parents:
diff changeset
163 build_int_cst (TREE_TYPE (tt), 1));
kono
parents:
diff changeset
164 x = fold_build2 (LT_EXPR, boolean_type_node, unshare_expr (op0),
kono
parents:
diff changeset
165 build_int_cst (type0, 0));
kono
parents:
diff changeset
166 tt = fold_build2 (TRUTH_OR_EXPR, boolean_type_node, x, tt);
kono
parents:
diff changeset
167 }
kono
parents:
diff changeset
168
kono
parents:
diff changeset
169 /* If the condition was folded to 0, no need to instrument
kono
parents:
diff changeset
170 this expression. */
kono
parents:
diff changeset
171 if (integer_zerop (t) && (tt == NULL_TREE || integer_zerop (tt)))
kono
parents:
diff changeset
172 return NULL_TREE;
kono
parents:
diff changeset
173
kono
parents:
diff changeset
174 /* In case we have a SAVE_EXPR in a conditional context, we need to
kono
parents:
diff changeset
175 make sure it gets evaluated before the condition. */
kono
parents:
diff changeset
176 t = fold_build2 (COMPOUND_EXPR, TREE_TYPE (t), unshare_expr (op0), t);
kono
parents:
diff changeset
177 t = fold_build2 (COMPOUND_EXPR, TREE_TYPE (t), unshare_expr (op1), t);
kono
parents:
diff changeset
178
kono
parents:
diff changeset
179 enum sanitize_code recover_kind = SANITIZE_SHIFT_EXPONENT;
kono
parents:
diff changeset
180 tree else_t = void_node;
kono
parents:
diff changeset
181 if (tt)
kono
parents:
diff changeset
182 {
kono
parents:
diff changeset
183 if (!sanitize_flags_p (SANITIZE_SHIFT_EXPONENT))
kono
parents:
diff changeset
184 {
kono
parents:
diff changeset
185 t = fold_build1 (TRUTH_NOT_EXPR, boolean_type_node, t);
kono
parents:
diff changeset
186 t = fold_build2 (TRUTH_AND_EXPR, boolean_type_node, t, tt);
kono
parents:
diff changeset
187 recover_kind = SANITIZE_SHIFT_BASE;
kono
parents:
diff changeset
188 }
kono
parents:
diff changeset
189 else
kono
parents:
diff changeset
190 {
kono
parents:
diff changeset
191 if (flag_sanitize_undefined_trap_on_error
kono
parents:
diff changeset
192 || ((!(flag_sanitize_recover & SANITIZE_SHIFT_EXPONENT))
kono
parents:
diff changeset
193 == (!(flag_sanitize_recover & SANITIZE_SHIFT_BASE))))
kono
parents:
diff changeset
194 t = fold_build2 (TRUTH_OR_EXPR, boolean_type_node, t, tt);
kono
parents:
diff changeset
195 else
kono
parents:
diff changeset
196 else_t = tt;
kono
parents:
diff changeset
197 }
kono
parents:
diff changeset
198 }
kono
parents:
diff changeset
199
kono
parents:
diff changeset
200 if (flag_sanitize_undefined_trap_on_error)
kono
parents:
diff changeset
201 tt = build_call_expr_loc (loc, builtin_decl_explicit (BUILT_IN_TRAP), 0);
kono
parents:
diff changeset
202 else
kono
parents:
diff changeset
203 {
kono
parents:
diff changeset
204 tree data = ubsan_create_data ("__ubsan_shift_data", 1, &loc,
kono
parents:
diff changeset
205 ubsan_type_descriptor (type0),
kono
parents:
diff changeset
206 ubsan_type_descriptor (type1), NULL_TREE,
kono
parents:
diff changeset
207 NULL_TREE);
kono
parents:
diff changeset
208 data = build_fold_addr_expr_loc (loc, data);
kono
parents:
diff changeset
209
kono
parents:
diff changeset
210 enum built_in_function bcode
kono
parents:
diff changeset
211 = (flag_sanitize_recover & recover_kind)
kono
parents:
diff changeset
212 ? BUILT_IN_UBSAN_HANDLE_SHIFT_OUT_OF_BOUNDS
kono
parents:
diff changeset
213 : BUILT_IN_UBSAN_HANDLE_SHIFT_OUT_OF_BOUNDS_ABORT;
kono
parents:
diff changeset
214 tt = builtin_decl_explicit (bcode);
kono
parents:
diff changeset
215 op0 = unshare_expr (op0);
kono
parents:
diff changeset
216 op1 = unshare_expr (op1);
kono
parents:
diff changeset
217 tt = build_call_expr_loc (loc, tt, 3, data, ubsan_encode_value (op0),
kono
parents:
diff changeset
218 ubsan_encode_value (op1));
kono
parents:
diff changeset
219 if (else_t != void_node)
kono
parents:
diff changeset
220 {
kono
parents:
diff changeset
221 bcode = (flag_sanitize_recover & SANITIZE_SHIFT_BASE)
kono
parents:
diff changeset
222 ? BUILT_IN_UBSAN_HANDLE_SHIFT_OUT_OF_BOUNDS
kono
parents:
diff changeset
223 : BUILT_IN_UBSAN_HANDLE_SHIFT_OUT_OF_BOUNDS_ABORT;
kono
parents:
diff changeset
224 tree else_tt = builtin_decl_explicit (bcode);
kono
parents:
diff changeset
225 op0 = unshare_expr (op0);
kono
parents:
diff changeset
226 op1 = unshare_expr (op1);
kono
parents:
diff changeset
227 else_tt = build_call_expr_loc (loc, else_tt, 3, data,
kono
parents:
diff changeset
228 ubsan_encode_value (op0),
kono
parents:
diff changeset
229 ubsan_encode_value (op1));
kono
parents:
diff changeset
230 else_t = fold_build3 (COND_EXPR, void_type_node, else_t,
kono
parents:
diff changeset
231 else_tt, void_node);
kono
parents:
diff changeset
232 }
kono
parents:
diff changeset
233 }
kono
parents:
diff changeset
234 t = fold_build3 (COND_EXPR, void_type_node, t, tt, else_t);
kono
parents:
diff changeset
235
kono
parents:
diff changeset
236 return t;
kono
parents:
diff changeset
237 }
kono
parents:
diff changeset
238
kono
parents:
diff changeset
239 /* Instrument variable length array bound. */
kono
parents:
diff changeset
240
kono
parents:
diff changeset
241 tree
kono
parents:
diff changeset
242 ubsan_instrument_vla (location_t loc, tree size)
kono
parents:
diff changeset
243 {
kono
parents:
diff changeset
244 tree type = TREE_TYPE (size);
kono
parents:
diff changeset
245 tree t, tt;
kono
parents:
diff changeset
246
kono
parents:
diff changeset
247 t = fold_build2 (LE_EXPR, boolean_type_node, size, build_int_cst (type, 0));
kono
parents:
diff changeset
248 if (flag_sanitize_undefined_trap_on_error)
kono
parents:
diff changeset
249 tt = build_call_expr_loc (loc, builtin_decl_explicit (BUILT_IN_TRAP), 0);
kono
parents:
diff changeset
250 else
kono
parents:
diff changeset
251 {
kono
parents:
diff changeset
252 tree data = ubsan_create_data ("__ubsan_vla_data", 1, &loc,
kono
parents:
diff changeset
253 ubsan_type_descriptor (type), NULL_TREE,
kono
parents:
diff changeset
254 NULL_TREE);
kono
parents:
diff changeset
255 data = build_fold_addr_expr_loc (loc, data);
kono
parents:
diff changeset
256 enum built_in_function bcode
kono
parents:
diff changeset
257 = (flag_sanitize_recover & SANITIZE_VLA)
kono
parents:
diff changeset
258 ? BUILT_IN_UBSAN_HANDLE_VLA_BOUND_NOT_POSITIVE
kono
parents:
diff changeset
259 : BUILT_IN_UBSAN_HANDLE_VLA_BOUND_NOT_POSITIVE_ABORT;
kono
parents:
diff changeset
260 tt = builtin_decl_explicit (bcode);
kono
parents:
diff changeset
261 tt = build_call_expr_loc (loc, tt, 2, data, ubsan_encode_value (size));
kono
parents:
diff changeset
262 }
kono
parents:
diff changeset
263 t = fold_build3 (COND_EXPR, void_type_node, t, tt, void_node);
kono
parents:
diff changeset
264
kono
parents:
diff changeset
265 return t;
kono
parents:
diff changeset
266 }
kono
parents:
diff changeset
267
kono
parents:
diff changeset
268 /* Instrument missing return in C++ functions returning non-void. */
kono
parents:
diff changeset
269
kono
parents:
diff changeset
270 tree
kono
parents:
diff changeset
271 ubsan_instrument_return (location_t loc)
kono
parents:
diff changeset
272 {
kono
parents:
diff changeset
273 if (flag_sanitize_undefined_trap_on_error)
kono
parents:
diff changeset
274 return build_call_expr_loc (loc, builtin_decl_explicit (BUILT_IN_TRAP), 0);
kono
parents:
diff changeset
275
kono
parents:
diff changeset
276 tree data = ubsan_create_data ("__ubsan_missing_return_data", 1, &loc,
kono
parents:
diff changeset
277 NULL_TREE, NULL_TREE);
kono
parents:
diff changeset
278 tree t = builtin_decl_explicit (BUILT_IN_UBSAN_HANDLE_MISSING_RETURN);
kono
parents:
diff changeset
279 return build_call_expr_loc (loc, t, 1, build_fold_addr_expr_loc (loc, data));
kono
parents:
diff changeset
280 }
kono
parents:
diff changeset
281
kono
parents:
diff changeset
282 /* Instrument array bounds for ARRAY_REFs. We create special builtin,
kono
parents:
diff changeset
283 that gets expanded in the sanopt pass, and make an array dimension
kono
parents:
diff changeset
284 of it. ARRAY is the array, *INDEX is an index to the array.
kono
parents:
diff changeset
285 Return NULL_TREE if no instrumentation is emitted.
kono
parents:
diff changeset
286 IGNORE_OFF_BY_ONE is true if the ARRAY_REF is inside a ADDR_EXPR. */
kono
parents:
diff changeset
287
kono
parents:
diff changeset
288 tree
kono
parents:
diff changeset
289 ubsan_instrument_bounds (location_t loc, tree array, tree *index,
kono
parents:
diff changeset
290 bool ignore_off_by_one)
kono
parents:
diff changeset
291 {
kono
parents:
diff changeset
292 tree type = TREE_TYPE (array);
kono
parents:
diff changeset
293 tree domain = TYPE_DOMAIN (type);
kono
parents:
diff changeset
294
kono
parents:
diff changeset
295 if (domain == NULL_TREE || TYPE_MAX_VALUE (domain) == NULL_TREE)
kono
parents:
diff changeset
296 return NULL_TREE;
kono
parents:
diff changeset
297
kono
parents:
diff changeset
298 tree bound = TYPE_MAX_VALUE (domain);
kono
parents:
diff changeset
299 if (ignore_off_by_one)
kono
parents:
diff changeset
300 bound = fold_build2 (PLUS_EXPR, TREE_TYPE (bound), bound,
kono
parents:
diff changeset
301 build_int_cst (TREE_TYPE (bound), 1));
kono
parents:
diff changeset
302
kono
parents:
diff changeset
303 /* Detect flexible array members and suchlike, unless
kono
parents:
diff changeset
304 -fsanitize=bounds-strict. */
kono
parents:
diff changeset
305 tree base = get_base_address (array);
kono
parents:
diff changeset
306 if (!sanitize_flags_p (SANITIZE_BOUNDS_STRICT)
kono
parents:
diff changeset
307 && TREE_CODE (array) == COMPONENT_REF
kono
parents:
diff changeset
308 && base && (INDIRECT_REF_P (base) || TREE_CODE (base) == MEM_REF))
kono
parents:
diff changeset
309 {
kono
parents:
diff changeset
310 tree next = NULL_TREE;
kono
parents:
diff changeset
311 tree cref = array;
kono
parents:
diff changeset
312
kono
parents:
diff changeset
313 /* Walk all structs/unions. */
kono
parents:
diff changeset
314 while (TREE_CODE (cref) == COMPONENT_REF)
kono
parents:
diff changeset
315 {
kono
parents:
diff changeset
316 if (TREE_CODE (TREE_TYPE (TREE_OPERAND (cref, 0))) == RECORD_TYPE)
kono
parents:
diff changeset
317 for (next = DECL_CHAIN (TREE_OPERAND (cref, 1));
kono
parents:
diff changeset
318 next && TREE_CODE (next) != FIELD_DECL;
kono
parents:
diff changeset
319 next = DECL_CHAIN (next))
kono
parents:
diff changeset
320 ;
kono
parents:
diff changeset
321 if (next)
kono
parents:
diff changeset
322 /* Not a last element. Instrument it. */
kono
parents:
diff changeset
323 break;
kono
parents:
diff changeset
324 /* Ok, this is the last field of the structure/union. But the
kono
parents:
diff changeset
325 aggregate containing the field must be the last field too,
kono
parents:
diff changeset
326 recursively. */
kono
parents:
diff changeset
327 cref = TREE_OPERAND (cref, 0);
kono
parents:
diff changeset
328 }
kono
parents:
diff changeset
329 if (!next)
kono
parents:
diff changeset
330 /* Don't instrument this flexible array member-like array in non-strict
kono
parents:
diff changeset
331 -fsanitize=bounds mode. */
kono
parents:
diff changeset
332 return NULL_TREE;
kono
parents:
diff changeset
333 }
kono
parents:
diff changeset
334
kono
parents:
diff changeset
335 /* Don't emit instrumentation in the most common cases. */
kono
parents:
diff changeset
336 tree idx = NULL_TREE;
kono
parents:
diff changeset
337 if (TREE_CODE (*index) == INTEGER_CST)
kono
parents:
diff changeset
338 idx = *index;
kono
parents:
diff changeset
339 else if (TREE_CODE (*index) == BIT_AND_EXPR
kono
parents:
diff changeset
340 && TREE_CODE (TREE_OPERAND (*index, 1)) == INTEGER_CST)
kono
parents:
diff changeset
341 idx = TREE_OPERAND (*index, 1);
kono
parents:
diff changeset
342 if (idx
kono
parents:
diff changeset
343 && TREE_CODE (bound) == INTEGER_CST
kono
parents:
diff changeset
344 && tree_int_cst_sgn (idx) >= 0
kono
parents:
diff changeset
345 && tree_int_cst_le (idx, bound))
kono
parents:
diff changeset
346 return NULL_TREE;
kono
parents:
diff changeset
347
kono
parents:
diff changeset
348 *index = save_expr (*index);
kono
parents:
diff changeset
349 /* Create a "(T *) 0" tree node to describe the array type. */
kono
parents:
diff changeset
350 tree zero_with_type = build_int_cst (build_pointer_type (type), 0);
kono
parents:
diff changeset
351 return build_call_expr_internal_loc (loc, IFN_UBSAN_BOUNDS,
kono
parents:
diff changeset
352 void_type_node, 3, zero_with_type,
kono
parents:
diff changeset
353 *index, bound);
kono
parents:
diff changeset
354 }
kono
parents:
diff changeset
355
kono
parents:
diff changeset
356 /* Return true iff T is an array that was instrumented by SANITIZE_BOUNDS. */
kono
parents:
diff changeset
357
kono
parents:
diff changeset
358 bool
kono
parents:
diff changeset
359 ubsan_array_ref_instrumented_p (const_tree t)
kono
parents:
diff changeset
360 {
kono
parents:
diff changeset
361 if (TREE_CODE (t) != ARRAY_REF)
kono
parents:
diff changeset
362 return false;
kono
parents:
diff changeset
363
kono
parents:
diff changeset
364 tree op1 = TREE_OPERAND (t, 1);
kono
parents:
diff changeset
365 return TREE_CODE (op1) == COMPOUND_EXPR
kono
parents:
diff changeset
366 && TREE_CODE (TREE_OPERAND (op1, 0)) == CALL_EXPR
kono
parents:
diff changeset
367 && CALL_EXPR_FN (TREE_OPERAND (op1, 0)) == NULL_TREE
kono
parents:
diff changeset
368 && CALL_EXPR_IFN (TREE_OPERAND (op1, 0)) == IFN_UBSAN_BOUNDS;
kono
parents:
diff changeset
369 }
kono
parents:
diff changeset
370
kono
parents:
diff changeset
371 /* Instrument an ARRAY_REF, if it hasn't already been instrumented.
kono
parents:
diff changeset
372 IGNORE_OFF_BY_ONE is true if the ARRAY_REF is inside a ADDR_EXPR. */
kono
parents:
diff changeset
373
kono
parents:
diff changeset
374 void
kono
parents:
diff changeset
375 ubsan_maybe_instrument_array_ref (tree *expr_p, bool ignore_off_by_one)
kono
parents:
diff changeset
376 {
kono
parents:
diff changeset
377 if (!ubsan_array_ref_instrumented_p (*expr_p)
kono
parents:
diff changeset
378 && sanitize_flags_p (SANITIZE_BOUNDS | SANITIZE_BOUNDS_STRICT)
kono
parents:
diff changeset
379 && current_function_decl != NULL_TREE)
kono
parents:
diff changeset
380 {
kono
parents:
diff changeset
381 tree op0 = TREE_OPERAND (*expr_p, 0);
kono
parents:
diff changeset
382 tree op1 = TREE_OPERAND (*expr_p, 1);
kono
parents:
diff changeset
383 tree e = ubsan_instrument_bounds (EXPR_LOCATION (*expr_p), op0, &op1,
kono
parents:
diff changeset
384 ignore_off_by_one);
kono
parents:
diff changeset
385 if (e != NULL_TREE)
kono
parents:
diff changeset
386 {
kono
parents:
diff changeset
387 tree t = copy_node (*expr_p);
kono
parents:
diff changeset
388 TREE_OPERAND (t, 1) = build2 (COMPOUND_EXPR, TREE_TYPE (op1),
kono
parents:
diff changeset
389 e, op1);
kono
parents:
diff changeset
390 *expr_p = t;
kono
parents:
diff changeset
391 }
kono
parents:
diff changeset
392 }
kono
parents:
diff changeset
393 }
kono
parents:
diff changeset
394
kono
parents:
diff changeset
395 static tree
kono
parents:
diff changeset
396 ubsan_maybe_instrument_reference_or_call (location_t loc, tree op, tree ptype,
kono
parents:
diff changeset
397 enum ubsan_null_ckind ckind)
kono
parents:
diff changeset
398 {
kono
parents:
diff changeset
399 if (!sanitize_flags_p (SANITIZE_ALIGNMENT | SANITIZE_NULL)
kono
parents:
diff changeset
400 || current_function_decl == NULL_TREE)
kono
parents:
diff changeset
401 return NULL_TREE;
kono
parents:
diff changeset
402
kono
parents:
diff changeset
403 tree type = TREE_TYPE (ptype);
kono
parents:
diff changeset
404 tree orig_op = op;
kono
parents:
diff changeset
405 bool instrument = false;
kono
parents:
diff changeset
406 unsigned int mina = 0;
kono
parents:
diff changeset
407
kono
parents:
diff changeset
408 if (sanitize_flags_p (SANITIZE_ALIGNMENT))
kono
parents:
diff changeset
409 {
kono
parents:
diff changeset
410 mina = min_align_of_type (type);
kono
parents:
diff changeset
411 if (mina <= 1)
kono
parents:
diff changeset
412 mina = 0;
kono
parents:
diff changeset
413 }
kono
parents:
diff changeset
414 while ((TREE_CODE (op) == NOP_EXPR
kono
parents:
diff changeset
415 || TREE_CODE (op) == NON_LVALUE_EXPR)
kono
parents:
diff changeset
416 && TREE_CODE (TREE_TYPE (op)) == POINTER_TYPE)
kono
parents:
diff changeset
417 op = TREE_OPERAND (op, 0);
kono
parents:
diff changeset
418 if (TREE_CODE (op) == NOP_EXPR
kono
parents:
diff changeset
419 && TREE_CODE (TREE_TYPE (op)) == REFERENCE_TYPE)
kono
parents:
diff changeset
420 {
kono
parents:
diff changeset
421 if (mina && mina > min_align_of_type (TREE_TYPE (TREE_TYPE (op))))
kono
parents:
diff changeset
422 instrument = true;
kono
parents:
diff changeset
423 }
kono
parents:
diff changeset
424 else
kono
parents:
diff changeset
425 {
kono
parents:
diff changeset
426 if (sanitize_flags_p (SANITIZE_NULL) && TREE_CODE (op) == ADDR_EXPR)
kono
parents:
diff changeset
427 {
kono
parents:
diff changeset
428 bool strict_overflow_p = false;
kono
parents:
diff changeset
429 /* tree_single_nonzero_warnv_p will not return true for non-weak
kono
parents:
diff changeset
430 non-automatic decls with -fno-delete-null-pointer-checks,
kono
parents:
diff changeset
431 which is disabled during -fsanitize=null. We don't want to
kono
parents:
diff changeset
432 instrument those, just weak vars though. */
kono
parents:
diff changeset
433 int save_flag_delete_null_pointer_checks
kono
parents:
diff changeset
434 = flag_delete_null_pointer_checks;
kono
parents:
diff changeset
435 flag_delete_null_pointer_checks = 1;
kono
parents:
diff changeset
436 if (!tree_single_nonzero_warnv_p (op, &strict_overflow_p)
kono
parents:
diff changeset
437 || strict_overflow_p)
kono
parents:
diff changeset
438 instrument = true;
kono
parents:
diff changeset
439 flag_delete_null_pointer_checks
kono
parents:
diff changeset
440 = save_flag_delete_null_pointer_checks;
kono
parents:
diff changeset
441 }
kono
parents:
diff changeset
442 else if (sanitize_flags_p (SANITIZE_NULL))
kono
parents:
diff changeset
443 instrument = true;
kono
parents:
diff changeset
444 if (mina && mina > 1)
kono
parents:
diff changeset
445 {
kono
parents:
diff changeset
446 if (!POINTER_TYPE_P (TREE_TYPE (op))
kono
parents:
diff changeset
447 || mina > get_pointer_alignment (op) / BITS_PER_UNIT)
kono
parents:
diff changeset
448 instrument = true;
kono
parents:
diff changeset
449 }
kono
parents:
diff changeset
450 }
kono
parents:
diff changeset
451 if (!instrument)
kono
parents:
diff changeset
452 return NULL_TREE;
kono
parents:
diff changeset
453 op = save_expr (orig_op);
kono
parents:
diff changeset
454 gcc_assert (POINTER_TYPE_P (ptype));
kono
parents:
diff changeset
455 if (TREE_CODE (ptype) == REFERENCE_TYPE)
kono
parents:
diff changeset
456 ptype = build_pointer_type (TREE_TYPE (ptype));
kono
parents:
diff changeset
457 tree kind = build_int_cst (ptype, ckind);
kono
parents:
diff changeset
458 tree align = build_int_cst (pointer_sized_int_node, mina);
kono
parents:
diff changeset
459 tree call
kono
parents:
diff changeset
460 = build_call_expr_internal_loc (loc, IFN_UBSAN_NULL, void_type_node,
kono
parents:
diff changeset
461 3, op, kind, align);
kono
parents:
diff changeset
462 TREE_SIDE_EFFECTS (call) = 1;
kono
parents:
diff changeset
463 return fold_build2 (COMPOUND_EXPR, TREE_TYPE (op), call, op);
kono
parents:
diff changeset
464 }
kono
parents:
diff changeset
465
kono
parents:
diff changeset
466 /* Instrument a NOP_EXPR to REFERENCE_TYPE or INTEGER_CST with REFERENCE_TYPE
kono
parents:
diff changeset
467 type if needed. */
kono
parents:
diff changeset
468
kono
parents:
diff changeset
469 void
kono
parents:
diff changeset
470 ubsan_maybe_instrument_reference (tree *stmt_p)
kono
parents:
diff changeset
471 {
kono
parents:
diff changeset
472 tree stmt = *stmt_p;
kono
parents:
diff changeset
473 tree op = stmt;
kono
parents:
diff changeset
474 if (TREE_CODE (stmt) == NOP_EXPR)
kono
parents:
diff changeset
475 op = TREE_OPERAND (stmt, 0);
kono
parents:
diff changeset
476 op = ubsan_maybe_instrument_reference_or_call (EXPR_LOCATION (stmt), op,
kono
parents:
diff changeset
477 TREE_TYPE (stmt),
kono
parents:
diff changeset
478 UBSAN_REF_BINDING);
kono
parents:
diff changeset
479 if (op)
kono
parents:
diff changeset
480 {
kono
parents:
diff changeset
481 if (TREE_CODE (stmt) == NOP_EXPR)
kono
parents:
diff changeset
482 TREE_OPERAND (stmt, 0) = op;
kono
parents:
diff changeset
483 else
kono
parents:
diff changeset
484 *stmt_p = op;
kono
parents:
diff changeset
485 }
kono
parents:
diff changeset
486 }
kono
parents:
diff changeset
487
kono
parents:
diff changeset
488 /* Instrument a CALL_EXPR to a method if needed. */
kono
parents:
diff changeset
489
kono
parents:
diff changeset
490 void
kono
parents:
diff changeset
491 ubsan_maybe_instrument_member_call (tree stmt, bool is_ctor)
kono
parents:
diff changeset
492 {
kono
parents:
diff changeset
493 if (call_expr_nargs (stmt) == 0)
kono
parents:
diff changeset
494 return;
kono
parents:
diff changeset
495 tree op = CALL_EXPR_ARG (stmt, 0);
kono
parents:
diff changeset
496 if (op == error_mark_node
kono
parents:
diff changeset
497 || !POINTER_TYPE_P (TREE_TYPE (op)))
kono
parents:
diff changeset
498 return;
kono
parents:
diff changeset
499 op = ubsan_maybe_instrument_reference_or_call (EXPR_LOCATION (stmt), op,
kono
parents:
diff changeset
500 TREE_TYPE (op),
kono
parents:
diff changeset
501 is_ctor ? UBSAN_CTOR_CALL
kono
parents:
diff changeset
502 : UBSAN_MEMBER_CALL);
kono
parents:
diff changeset
503 if (op)
kono
parents:
diff changeset
504 CALL_EXPR_ARG (stmt, 0) = op;
kono
parents:
diff changeset
505 }