]>
Commit | Line | Data |
---|---|---|
1 | /* GCC instrumentation plugin for ThreadSanitizer. | |
2 | Copyright (C) 2011-2020 Free Software Foundation, Inc. | |
3 | Contributed by Dmitry Vyukov <dvyukov@google.com> | |
4 | ||
5 | This file is part of GCC. | |
6 | ||
7 | GCC is free software; you can redistribute it and/or modify it under | |
8 | the terms of the GNU General Public License as published by the Free | |
9 | Software Foundation; either version 3, or (at your option) any later | |
10 | version. | |
11 | ||
12 | GCC is distributed in the hope that it will be useful, but WITHOUT ANY | |
13 | WARRANTY; without even the implied warranty of MERCHANTABILITY or | |
14 | FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License | |
15 | for more details. | |
16 | ||
17 | You should have received a copy of the GNU General Public License | |
18 | along with GCC; see the file COPYING3. If not see | |
19 | <http://www.gnu.org/licenses/>. */ | |
20 | ||
21 | ||
22 | #include "config.h" | |
23 | #include "system.h" | |
24 | #include "coretypes.h" | |
25 | #include "backend.h" | |
26 | #include "rtl.h" | |
27 | #include "tree.h" | |
28 | #include "memmodel.h" | |
29 | #include "gimple.h" | |
30 | #include "tree-pass.h" | |
31 | #include "ssa.h" | |
32 | #include "cgraph.h" | |
33 | #include "fold-const.h" | |
34 | #include "gimplify.h" | |
35 | #include "gimple-iterator.h" | |
36 | #include "gimplify-me.h" | |
37 | #include "tree-cfg.h" | |
38 | #include "tree-iterator.h" | |
39 | #include "tree-ssa-propagate.h" | |
40 | #include "tree-ssa-loop-ivopts.h" | |
41 | #include "tree-eh.h" | |
42 | #include "tsan.h" | |
43 | #include "stringpool.h" | |
44 | #include "attribs.h" | |
45 | #include "asan.h" | |
46 | #include "builtins.h" | |
47 | #include "target.h" | |
48 | ||
49 | /* Number of instrumented memory accesses in the current function. */ | |
50 | ||
51 | /* Builds the following decl | |
52 | void __tsan_read/writeX (void *addr); */ | |
53 | ||
54 | static tree | |
55 | get_memory_access_decl (bool is_write, unsigned size) | |
56 | { | |
57 | enum built_in_function fcode; | |
58 | ||
59 | if (size <= 1) | |
60 | fcode = is_write ? BUILT_IN_TSAN_WRITE1 | |
61 | : BUILT_IN_TSAN_READ1; | |
62 | else if (size <= 3) | |
63 | fcode = is_write ? BUILT_IN_TSAN_WRITE2 | |
64 | : BUILT_IN_TSAN_READ2; | |
65 | else if (size <= 7) | |
66 | fcode = is_write ? BUILT_IN_TSAN_WRITE4 | |
67 | : BUILT_IN_TSAN_READ4; | |
68 | else if (size <= 15) | |
69 | fcode = is_write ? BUILT_IN_TSAN_WRITE8 | |
70 | : BUILT_IN_TSAN_READ8; | |
71 | else | |
72 | fcode = is_write ? BUILT_IN_TSAN_WRITE16 | |
73 | : BUILT_IN_TSAN_READ16; | |
74 | ||
75 | return builtin_decl_implicit (fcode); | |
76 | } | |
77 | ||
78 | /* Check as to whether EXPR refers to a store to vptr. */ | |
79 | ||
80 | static tree | |
81 | is_vptr_store (gimple *stmt, tree expr, bool is_write) | |
82 | { | |
83 | if (is_write == true | |
84 | && gimple_assign_single_p (stmt) | |
85 | && TREE_CODE (expr) == COMPONENT_REF) | |
86 | { | |
87 | tree field = TREE_OPERAND (expr, 1); | |
88 | if (TREE_CODE (field) == FIELD_DECL | |
89 | && DECL_VIRTUAL_P (field)) | |
90 | return gimple_assign_rhs1 (stmt); | |
91 | } | |
92 | return NULL; | |
93 | } | |
94 | ||
95 | /* Instruments EXPR if needed. If any instrumentation is inserted, | |
96 | return true. */ | |
97 | ||
98 | static bool | |
99 | instrument_expr (gimple_stmt_iterator gsi, tree expr, bool is_write) | |
100 | { | |
101 | tree base, rhs, expr_ptr, builtin_decl; | |
102 | basic_block bb; | |
103 | HOST_WIDE_INT size; | |
104 | gimple *stmt, *g; | |
105 | gimple_seq seq; | |
106 | location_t loc; | |
107 | unsigned int align; | |
108 | ||
109 | size = int_size_in_bytes (TREE_TYPE (expr)); | |
110 | if (size <= 0) | |
111 | return false; | |
112 | ||
113 | poly_int64 unused_bitsize, unused_bitpos; | |
114 | tree offset; | |
115 | machine_mode mode; | |
116 | int unsignedp, reversep, volatilep = 0; | |
117 | base = get_inner_reference (expr, &unused_bitsize, &unused_bitpos, &offset, | |
118 | &mode, &unsignedp, &reversep, &volatilep); | |
119 | ||
120 | /* No need to instrument accesses to decls that don't escape, | |
121 | they can't escape to other threads then. */ | |
122 | if (DECL_P (base) && !is_global_var (base)) | |
123 | { | |
124 | struct pt_solution pt; | |
125 | memset (&pt, 0, sizeof (pt)); | |
126 | pt.escaped = 1; | |
127 | pt.ipa_escaped = flag_ipa_pta != 0; | |
128 | if (!pt_solution_includes (&pt, base)) | |
129 | return false; | |
130 | if (!may_be_aliased (base)) | |
131 | return false; | |
132 | } | |
133 | ||
134 | if (TREE_READONLY (base) || (VAR_P (base) && DECL_HARD_REGISTER (base))) | |
135 | return false; | |
136 | ||
137 | stmt = gsi_stmt (gsi); | |
138 | loc = gimple_location (stmt); | |
139 | rhs = is_vptr_store (stmt, expr, is_write); | |
140 | ||
141 | if ((TREE_CODE (expr) == COMPONENT_REF | |
142 | && DECL_BIT_FIELD_TYPE (TREE_OPERAND (expr, 1))) | |
143 | || TREE_CODE (expr) == BIT_FIELD_REF) | |
144 | { | |
145 | HOST_WIDE_INT bitpos, bitsize; | |
146 | base = TREE_OPERAND (expr, 0); | |
147 | if (TREE_CODE (expr) == COMPONENT_REF) | |
148 | { | |
149 | expr = TREE_OPERAND (expr, 1); | |
150 | if (is_write && DECL_BIT_FIELD_REPRESENTATIVE (expr)) | |
151 | expr = DECL_BIT_FIELD_REPRESENTATIVE (expr); | |
152 | if (!tree_fits_uhwi_p (DECL_FIELD_OFFSET (expr)) | |
153 | || !tree_fits_uhwi_p (DECL_FIELD_BIT_OFFSET (expr)) | |
154 | || !tree_fits_uhwi_p (DECL_SIZE (expr))) | |
155 | return false; | |
156 | bitpos = tree_to_uhwi (DECL_FIELD_OFFSET (expr)) * BITS_PER_UNIT | |
157 | + tree_to_uhwi (DECL_FIELD_BIT_OFFSET (expr)); | |
158 | bitsize = tree_to_uhwi (DECL_SIZE (expr)); | |
159 | } | |
160 | else | |
161 | { | |
162 | if (!tree_fits_uhwi_p (TREE_OPERAND (expr, 2)) | |
163 | || !tree_fits_uhwi_p (TREE_OPERAND (expr, 1))) | |
164 | return false; | |
165 | bitpos = tree_to_uhwi (TREE_OPERAND (expr, 2)); | |
166 | bitsize = tree_to_uhwi (TREE_OPERAND (expr, 1)); | |
167 | } | |
168 | if (bitpos < 0 || bitsize <= 0) | |
169 | return false; | |
170 | size = (bitpos % BITS_PER_UNIT + bitsize + BITS_PER_UNIT - 1) | |
171 | / BITS_PER_UNIT; | |
172 | if (may_be_nonaddressable_p (base)) | |
173 | return false; | |
174 | align = get_object_alignment (base); | |
175 | if (align < BITS_PER_UNIT) | |
176 | return false; | |
177 | bitpos = bitpos & ~(BITS_PER_UNIT - 1); | |
178 | if ((align - 1) & bitpos) | |
179 | { | |
180 | align = (align - 1) & bitpos; | |
181 | align = least_bit_hwi (align); | |
182 | } | |
183 | expr = build_fold_addr_expr (unshare_expr (base)); | |
184 | expr = build2 (MEM_REF, char_type_node, expr, | |
185 | build_int_cst (TREE_TYPE (expr), bitpos / BITS_PER_UNIT)); | |
186 | expr_ptr = build_fold_addr_expr (expr); | |
187 | } | |
188 | else | |
189 | { | |
190 | if (may_be_nonaddressable_p (expr)) | |
191 | return false; | |
192 | align = get_object_alignment (expr); | |
193 | if (align < BITS_PER_UNIT) | |
194 | return false; | |
195 | expr_ptr = build_fold_addr_expr (unshare_expr (expr)); | |
196 | } | |
197 | expr_ptr = force_gimple_operand (expr_ptr, &seq, true, NULL_TREE); | |
198 | if ((size & (size - 1)) != 0 || size > 16 | |
199 | || align < MIN (size, 8) * BITS_PER_UNIT) | |
200 | { | |
201 | builtin_decl = builtin_decl_implicit (is_write | |
202 | ? BUILT_IN_TSAN_WRITE_RANGE | |
203 | : BUILT_IN_TSAN_READ_RANGE); | |
204 | g = gimple_build_call (builtin_decl, 2, expr_ptr, size_int (size)); | |
205 | } | |
206 | else if (rhs == NULL) | |
207 | g = gimple_build_call (get_memory_access_decl (is_write, size), | |
208 | 1, expr_ptr); | |
209 | else | |
210 | { | |
211 | builtin_decl = builtin_decl_implicit (BUILT_IN_TSAN_VPTR_UPDATE); | |
212 | g = gimple_build_call (builtin_decl, 2, expr_ptr, unshare_expr (rhs)); | |
213 | } | |
214 | gimple_set_location (g, loc); | |
215 | gimple_seq_add_stmt_without_update (&seq, g); | |
216 | /* Instrumentation for assignment of a function result | |
217 | must be inserted after the call. Instrumentation for | |
218 | reads of function arguments must be inserted before the call. | |
219 | That's because the call can contain synchronization. */ | |
220 | if (is_gimple_call (stmt) && is_write) | |
221 | { | |
222 | /* If the call can throw, it must be the last stmt in | |
223 | a basic block, so the instrumented stmts need to be | |
224 | inserted in successor bbs. */ | |
225 | if (is_ctrl_altering_stmt (stmt)) | |
226 | { | |
227 | edge e; | |
228 | ||
229 | bb = gsi_bb (gsi); | |
230 | e = find_fallthru_edge (bb->succs); | |
231 | if (e) | |
232 | gsi_insert_seq_on_edge_immediate (e, seq); | |
233 | } | |
234 | else | |
235 | gsi_insert_seq_after (&gsi, seq, GSI_NEW_STMT); | |
236 | } | |
237 | else | |
238 | gsi_insert_seq_before (&gsi, seq, GSI_SAME_STMT); | |
239 | ||
240 | return true; | |
241 | } | |
242 | ||
243 | /* Actions for sync/atomic builtin transformations. */ | |
244 | enum tsan_atomic_action | |
245 | { | |
246 | check_last, add_seq_cst, add_acquire, weak_cas, strong_cas, | |
247 | bool_cas, val_cas, lock_release, fetch_op, fetch_op_seq_cst, | |
248 | bool_clear, bool_test_and_set | |
249 | }; | |
250 | ||
251 | /* Table how to map sync/atomic builtins to their corresponding | |
252 | tsan equivalents. */ | |
253 | static const struct tsan_map_atomic | |
254 | { | |
255 | enum built_in_function fcode, tsan_fcode; | |
256 | enum tsan_atomic_action action; | |
257 | enum tree_code code; | |
258 | } tsan_atomic_table[] = | |
259 | { | |
260 | #define TRANSFORM(fcode, tsan_fcode, action, code) \ | |
261 | { BUILT_IN_##fcode, BUILT_IN_##tsan_fcode, action, code } | |
262 | #define CHECK_LAST(fcode, tsan_fcode) \ | |
263 | TRANSFORM (fcode, tsan_fcode, check_last, ERROR_MARK) | |
264 | #define ADD_SEQ_CST(fcode, tsan_fcode) \ | |
265 | TRANSFORM (fcode, tsan_fcode, add_seq_cst, ERROR_MARK) | |
266 | #define ADD_ACQUIRE(fcode, tsan_fcode) \ | |
267 | TRANSFORM (fcode, tsan_fcode, add_acquire, ERROR_MARK) | |
268 | #define WEAK_CAS(fcode, tsan_fcode) \ | |
269 | TRANSFORM (fcode, tsan_fcode, weak_cas, ERROR_MARK) | |
270 | #define STRONG_CAS(fcode, tsan_fcode) \ | |
271 | TRANSFORM (fcode, tsan_fcode, strong_cas, ERROR_MARK) | |
272 | #define BOOL_CAS(fcode, tsan_fcode) \ | |
273 | TRANSFORM (fcode, tsan_fcode, bool_cas, ERROR_MARK) | |
274 | #define VAL_CAS(fcode, tsan_fcode) \ | |
275 | TRANSFORM (fcode, tsan_fcode, val_cas, ERROR_MARK) | |
276 | #define LOCK_RELEASE(fcode, tsan_fcode) \ | |
277 | TRANSFORM (fcode, tsan_fcode, lock_release, ERROR_MARK) | |
278 | #define FETCH_OP(fcode, tsan_fcode, code) \ | |
279 | TRANSFORM (fcode, tsan_fcode, fetch_op, code) | |
280 | #define FETCH_OPS(fcode, tsan_fcode, code) \ | |
281 | TRANSFORM (fcode, tsan_fcode, fetch_op_seq_cst, code) | |
282 | #define BOOL_CLEAR(fcode, tsan_fcode) \ | |
283 | TRANSFORM (fcode, tsan_fcode, bool_clear, ERROR_MARK) | |
284 | #define BOOL_TEST_AND_SET(fcode, tsan_fcode) \ | |
285 | TRANSFORM (fcode, tsan_fcode, bool_test_and_set, ERROR_MARK) | |
286 | ||
287 | CHECK_LAST (ATOMIC_LOAD_1, TSAN_ATOMIC8_LOAD), | |
288 | CHECK_LAST (ATOMIC_LOAD_2, TSAN_ATOMIC16_LOAD), | |
289 | CHECK_LAST (ATOMIC_LOAD_4, TSAN_ATOMIC32_LOAD), | |
290 | CHECK_LAST (ATOMIC_LOAD_8, TSAN_ATOMIC64_LOAD), | |
291 | CHECK_LAST (ATOMIC_LOAD_16, TSAN_ATOMIC128_LOAD), | |
292 | CHECK_LAST (ATOMIC_STORE_1, TSAN_ATOMIC8_STORE), | |
293 | CHECK_LAST (ATOMIC_STORE_2, TSAN_ATOMIC16_STORE), | |
294 | CHECK_LAST (ATOMIC_STORE_4, TSAN_ATOMIC32_STORE), | |
295 | CHECK_LAST (ATOMIC_STORE_8, TSAN_ATOMIC64_STORE), | |
296 | CHECK_LAST (ATOMIC_STORE_16, TSAN_ATOMIC128_STORE), | |
297 | CHECK_LAST (ATOMIC_EXCHANGE_1, TSAN_ATOMIC8_EXCHANGE), | |
298 | CHECK_LAST (ATOMIC_EXCHANGE_2, TSAN_ATOMIC16_EXCHANGE), | |
299 | CHECK_LAST (ATOMIC_EXCHANGE_4, TSAN_ATOMIC32_EXCHANGE), | |
300 | CHECK_LAST (ATOMIC_EXCHANGE_8, TSAN_ATOMIC64_EXCHANGE), | |
301 | CHECK_LAST (ATOMIC_EXCHANGE_16, TSAN_ATOMIC128_EXCHANGE), | |
302 | CHECK_LAST (ATOMIC_FETCH_ADD_1, TSAN_ATOMIC8_FETCH_ADD), | |
303 | CHECK_LAST (ATOMIC_FETCH_ADD_2, TSAN_ATOMIC16_FETCH_ADD), | |
304 | CHECK_LAST (ATOMIC_FETCH_ADD_4, TSAN_ATOMIC32_FETCH_ADD), | |
305 | CHECK_LAST (ATOMIC_FETCH_ADD_8, TSAN_ATOMIC64_FETCH_ADD), | |
306 | CHECK_LAST (ATOMIC_FETCH_ADD_16, TSAN_ATOMIC128_FETCH_ADD), | |
307 | CHECK_LAST (ATOMIC_FETCH_SUB_1, TSAN_ATOMIC8_FETCH_SUB), | |
308 | CHECK_LAST (ATOMIC_FETCH_SUB_2, TSAN_ATOMIC16_FETCH_SUB), | |
309 | CHECK_LAST (ATOMIC_FETCH_SUB_4, TSAN_ATOMIC32_FETCH_SUB), | |
310 | CHECK_LAST (ATOMIC_FETCH_SUB_8, TSAN_ATOMIC64_FETCH_SUB), | |
311 | CHECK_LAST (ATOMIC_FETCH_SUB_16, TSAN_ATOMIC128_FETCH_SUB), | |
312 | CHECK_LAST (ATOMIC_FETCH_AND_1, TSAN_ATOMIC8_FETCH_AND), | |
313 | CHECK_LAST (ATOMIC_FETCH_AND_2, TSAN_ATOMIC16_FETCH_AND), | |
314 | CHECK_LAST (ATOMIC_FETCH_AND_4, TSAN_ATOMIC32_FETCH_AND), | |
315 | CHECK_LAST (ATOMIC_FETCH_AND_8, TSAN_ATOMIC64_FETCH_AND), | |
316 | CHECK_LAST (ATOMIC_FETCH_AND_16, TSAN_ATOMIC128_FETCH_AND), | |
317 | CHECK_LAST (ATOMIC_FETCH_OR_1, TSAN_ATOMIC8_FETCH_OR), | |
318 | CHECK_LAST (ATOMIC_FETCH_OR_2, TSAN_ATOMIC16_FETCH_OR), | |
319 | CHECK_LAST (ATOMIC_FETCH_OR_4, TSAN_ATOMIC32_FETCH_OR), | |
320 | CHECK_LAST (ATOMIC_FETCH_OR_8, TSAN_ATOMIC64_FETCH_OR), | |
321 | CHECK_LAST (ATOMIC_FETCH_OR_16, TSAN_ATOMIC128_FETCH_OR), | |
322 | CHECK_LAST (ATOMIC_FETCH_XOR_1, TSAN_ATOMIC8_FETCH_XOR), | |
323 | CHECK_LAST (ATOMIC_FETCH_XOR_2, TSAN_ATOMIC16_FETCH_XOR), | |
324 | CHECK_LAST (ATOMIC_FETCH_XOR_4, TSAN_ATOMIC32_FETCH_XOR), | |
325 | CHECK_LAST (ATOMIC_FETCH_XOR_8, TSAN_ATOMIC64_FETCH_XOR), | |
326 | CHECK_LAST (ATOMIC_FETCH_XOR_16, TSAN_ATOMIC128_FETCH_XOR), | |
327 | CHECK_LAST (ATOMIC_FETCH_NAND_1, TSAN_ATOMIC8_FETCH_NAND), | |
328 | CHECK_LAST (ATOMIC_FETCH_NAND_2, TSAN_ATOMIC16_FETCH_NAND), | |
329 | CHECK_LAST (ATOMIC_FETCH_NAND_4, TSAN_ATOMIC32_FETCH_NAND), | |
330 | CHECK_LAST (ATOMIC_FETCH_NAND_8, TSAN_ATOMIC64_FETCH_NAND), | |
331 | CHECK_LAST (ATOMIC_FETCH_NAND_16, TSAN_ATOMIC128_FETCH_NAND), | |
332 | ||
333 | CHECK_LAST (ATOMIC_THREAD_FENCE, TSAN_ATOMIC_THREAD_FENCE), | |
334 | CHECK_LAST (ATOMIC_SIGNAL_FENCE, TSAN_ATOMIC_SIGNAL_FENCE), | |
335 | ||
336 | FETCH_OP (ATOMIC_ADD_FETCH_1, TSAN_ATOMIC8_FETCH_ADD, PLUS_EXPR), | |
337 | FETCH_OP (ATOMIC_ADD_FETCH_2, TSAN_ATOMIC16_FETCH_ADD, PLUS_EXPR), | |
338 | FETCH_OP (ATOMIC_ADD_FETCH_4, TSAN_ATOMIC32_FETCH_ADD, PLUS_EXPR), | |
339 | FETCH_OP (ATOMIC_ADD_FETCH_8, TSAN_ATOMIC64_FETCH_ADD, PLUS_EXPR), | |
340 | FETCH_OP (ATOMIC_ADD_FETCH_16, TSAN_ATOMIC128_FETCH_ADD, PLUS_EXPR), | |
341 | FETCH_OP (ATOMIC_SUB_FETCH_1, TSAN_ATOMIC8_FETCH_SUB, MINUS_EXPR), | |
342 | FETCH_OP (ATOMIC_SUB_FETCH_2, TSAN_ATOMIC16_FETCH_SUB, MINUS_EXPR), | |
343 | FETCH_OP (ATOMIC_SUB_FETCH_4, TSAN_ATOMIC32_FETCH_SUB, MINUS_EXPR), | |
344 | FETCH_OP (ATOMIC_SUB_FETCH_8, TSAN_ATOMIC64_FETCH_SUB, MINUS_EXPR), | |
345 | FETCH_OP (ATOMIC_SUB_FETCH_16, TSAN_ATOMIC128_FETCH_SUB, MINUS_EXPR), | |
346 | FETCH_OP (ATOMIC_AND_FETCH_1, TSAN_ATOMIC8_FETCH_AND, BIT_AND_EXPR), | |
347 | FETCH_OP (ATOMIC_AND_FETCH_2, TSAN_ATOMIC16_FETCH_AND, BIT_AND_EXPR), | |
348 | FETCH_OP (ATOMIC_AND_FETCH_4, TSAN_ATOMIC32_FETCH_AND, BIT_AND_EXPR), | |
349 | FETCH_OP (ATOMIC_AND_FETCH_8, TSAN_ATOMIC64_FETCH_AND, BIT_AND_EXPR), | |
350 | FETCH_OP (ATOMIC_AND_FETCH_16, TSAN_ATOMIC128_FETCH_AND, BIT_AND_EXPR), | |
351 | FETCH_OP (ATOMIC_OR_FETCH_1, TSAN_ATOMIC8_FETCH_OR, BIT_IOR_EXPR), | |
352 | FETCH_OP (ATOMIC_OR_FETCH_2, TSAN_ATOMIC16_FETCH_OR, BIT_IOR_EXPR), | |
353 | FETCH_OP (ATOMIC_OR_FETCH_4, TSAN_ATOMIC32_FETCH_OR, BIT_IOR_EXPR), | |
354 | FETCH_OP (ATOMIC_OR_FETCH_8, TSAN_ATOMIC64_FETCH_OR, BIT_IOR_EXPR), | |
355 | FETCH_OP (ATOMIC_OR_FETCH_16, TSAN_ATOMIC128_FETCH_OR, BIT_IOR_EXPR), | |
356 | FETCH_OP (ATOMIC_XOR_FETCH_1, TSAN_ATOMIC8_FETCH_XOR, BIT_XOR_EXPR), | |
357 | FETCH_OP (ATOMIC_XOR_FETCH_2, TSAN_ATOMIC16_FETCH_XOR, BIT_XOR_EXPR), | |
358 | FETCH_OP (ATOMIC_XOR_FETCH_4, TSAN_ATOMIC32_FETCH_XOR, BIT_XOR_EXPR), | |
359 | FETCH_OP (ATOMIC_XOR_FETCH_8, TSAN_ATOMIC64_FETCH_XOR, BIT_XOR_EXPR), | |
360 | FETCH_OP (ATOMIC_XOR_FETCH_16, TSAN_ATOMIC128_FETCH_XOR, BIT_XOR_EXPR), | |
361 | FETCH_OP (ATOMIC_NAND_FETCH_1, TSAN_ATOMIC8_FETCH_NAND, BIT_NOT_EXPR), | |
362 | FETCH_OP (ATOMIC_NAND_FETCH_2, TSAN_ATOMIC16_FETCH_NAND, BIT_NOT_EXPR), | |
363 | FETCH_OP (ATOMIC_NAND_FETCH_4, TSAN_ATOMIC32_FETCH_NAND, BIT_NOT_EXPR), | |
364 | FETCH_OP (ATOMIC_NAND_FETCH_8, TSAN_ATOMIC64_FETCH_NAND, BIT_NOT_EXPR), | |
365 | FETCH_OP (ATOMIC_NAND_FETCH_16, TSAN_ATOMIC128_FETCH_NAND, BIT_NOT_EXPR), | |
366 | ||
367 | ADD_ACQUIRE (SYNC_LOCK_TEST_AND_SET_1, TSAN_ATOMIC8_EXCHANGE), | |
368 | ADD_ACQUIRE (SYNC_LOCK_TEST_AND_SET_2, TSAN_ATOMIC16_EXCHANGE), | |
369 | ADD_ACQUIRE (SYNC_LOCK_TEST_AND_SET_4, TSAN_ATOMIC32_EXCHANGE), | |
370 | ADD_ACQUIRE (SYNC_LOCK_TEST_AND_SET_8, TSAN_ATOMIC64_EXCHANGE), | |
371 | ADD_ACQUIRE (SYNC_LOCK_TEST_AND_SET_16, TSAN_ATOMIC128_EXCHANGE), | |
372 | ||
373 | ADD_SEQ_CST (SYNC_FETCH_AND_ADD_1, TSAN_ATOMIC8_FETCH_ADD), | |
374 | ADD_SEQ_CST (SYNC_FETCH_AND_ADD_2, TSAN_ATOMIC16_FETCH_ADD), | |
375 | ADD_SEQ_CST (SYNC_FETCH_AND_ADD_4, TSAN_ATOMIC32_FETCH_ADD), | |
376 | ADD_SEQ_CST (SYNC_FETCH_AND_ADD_8, TSAN_ATOMIC64_FETCH_ADD), | |
377 | ADD_SEQ_CST (SYNC_FETCH_AND_ADD_16, TSAN_ATOMIC128_FETCH_ADD), | |
378 | ADD_SEQ_CST (SYNC_FETCH_AND_SUB_1, TSAN_ATOMIC8_FETCH_SUB), | |
379 | ADD_SEQ_CST (SYNC_FETCH_AND_SUB_2, TSAN_ATOMIC16_FETCH_SUB), | |
380 | ADD_SEQ_CST (SYNC_FETCH_AND_SUB_4, TSAN_ATOMIC32_FETCH_SUB), | |
381 | ADD_SEQ_CST (SYNC_FETCH_AND_SUB_8, TSAN_ATOMIC64_FETCH_SUB), | |
382 | ADD_SEQ_CST (SYNC_FETCH_AND_SUB_16, TSAN_ATOMIC128_FETCH_SUB), | |
383 | ADD_SEQ_CST (SYNC_FETCH_AND_AND_1, TSAN_ATOMIC8_FETCH_AND), | |
384 | ADD_SEQ_CST (SYNC_FETCH_AND_AND_2, TSAN_ATOMIC16_FETCH_AND), | |
385 | ADD_SEQ_CST (SYNC_FETCH_AND_AND_4, TSAN_ATOMIC32_FETCH_AND), | |
386 | ADD_SEQ_CST (SYNC_FETCH_AND_AND_8, TSAN_ATOMIC64_FETCH_AND), | |
387 | ADD_SEQ_CST (SYNC_FETCH_AND_AND_16, TSAN_ATOMIC128_FETCH_AND), | |
388 | ADD_SEQ_CST (SYNC_FETCH_AND_OR_1, TSAN_ATOMIC8_FETCH_OR), | |
389 | ADD_SEQ_CST (SYNC_FETCH_AND_OR_2, TSAN_ATOMIC16_FETCH_OR), | |
390 | ADD_SEQ_CST (SYNC_FETCH_AND_OR_4, TSAN_ATOMIC32_FETCH_OR), | |
391 | ADD_SEQ_CST (SYNC_FETCH_AND_OR_8, TSAN_ATOMIC64_FETCH_OR), | |
392 | ADD_SEQ_CST (SYNC_FETCH_AND_OR_16, TSAN_ATOMIC128_FETCH_OR), | |
393 | ADD_SEQ_CST (SYNC_FETCH_AND_XOR_1, TSAN_ATOMIC8_FETCH_XOR), | |
394 | ADD_SEQ_CST (SYNC_FETCH_AND_XOR_2, TSAN_ATOMIC16_FETCH_XOR), | |
395 | ADD_SEQ_CST (SYNC_FETCH_AND_XOR_4, TSAN_ATOMIC32_FETCH_XOR), | |
396 | ADD_SEQ_CST (SYNC_FETCH_AND_XOR_8, TSAN_ATOMIC64_FETCH_XOR), | |
397 | ADD_SEQ_CST (SYNC_FETCH_AND_XOR_16, TSAN_ATOMIC128_FETCH_XOR), | |
398 | ADD_SEQ_CST (SYNC_FETCH_AND_NAND_1, TSAN_ATOMIC8_FETCH_NAND), | |
399 | ADD_SEQ_CST (SYNC_FETCH_AND_NAND_2, TSAN_ATOMIC16_FETCH_NAND), | |
400 | ADD_SEQ_CST (SYNC_FETCH_AND_NAND_4, TSAN_ATOMIC32_FETCH_NAND), | |
401 | ADD_SEQ_CST (SYNC_FETCH_AND_NAND_8, TSAN_ATOMIC64_FETCH_NAND), | |
402 | ADD_SEQ_CST (SYNC_FETCH_AND_NAND_16, TSAN_ATOMIC128_FETCH_NAND), | |
403 | ||
404 | ADD_SEQ_CST (SYNC_SYNCHRONIZE, TSAN_ATOMIC_THREAD_FENCE), | |
405 | ||
406 | FETCH_OPS (SYNC_ADD_AND_FETCH_1, TSAN_ATOMIC8_FETCH_ADD, PLUS_EXPR), | |
407 | FETCH_OPS (SYNC_ADD_AND_FETCH_2, TSAN_ATOMIC16_FETCH_ADD, PLUS_EXPR), | |
408 | FETCH_OPS (SYNC_ADD_AND_FETCH_4, TSAN_ATOMIC32_FETCH_ADD, PLUS_EXPR), | |
409 | FETCH_OPS (SYNC_ADD_AND_FETCH_8, TSAN_ATOMIC64_FETCH_ADD, PLUS_EXPR), | |
410 | FETCH_OPS (SYNC_ADD_AND_FETCH_16, TSAN_ATOMIC128_FETCH_ADD, PLUS_EXPR), | |
411 | FETCH_OPS (SYNC_SUB_AND_FETCH_1, TSAN_ATOMIC8_FETCH_SUB, MINUS_EXPR), | |
412 | FETCH_OPS (SYNC_SUB_AND_FETCH_2, TSAN_ATOMIC16_FETCH_SUB, MINUS_EXPR), | |
413 | FETCH_OPS (SYNC_SUB_AND_FETCH_4, TSAN_ATOMIC32_FETCH_SUB, MINUS_EXPR), | |
414 | FETCH_OPS (SYNC_SUB_AND_FETCH_8, TSAN_ATOMIC64_FETCH_SUB, MINUS_EXPR), | |
415 | FETCH_OPS (SYNC_SUB_AND_FETCH_16, TSAN_ATOMIC128_FETCH_SUB, MINUS_EXPR), | |
416 | FETCH_OPS (SYNC_AND_AND_FETCH_1, TSAN_ATOMIC8_FETCH_AND, BIT_AND_EXPR), | |
417 | FETCH_OPS (SYNC_AND_AND_FETCH_2, TSAN_ATOMIC16_FETCH_AND, BIT_AND_EXPR), | |
418 | FETCH_OPS (SYNC_AND_AND_FETCH_4, TSAN_ATOMIC32_FETCH_AND, BIT_AND_EXPR), | |
419 | FETCH_OPS (SYNC_AND_AND_FETCH_8, TSAN_ATOMIC64_FETCH_AND, BIT_AND_EXPR), | |
420 | FETCH_OPS (SYNC_AND_AND_FETCH_16, TSAN_ATOMIC128_FETCH_AND, BIT_AND_EXPR), | |
421 | FETCH_OPS (SYNC_OR_AND_FETCH_1, TSAN_ATOMIC8_FETCH_OR, BIT_IOR_EXPR), | |
422 | FETCH_OPS (SYNC_OR_AND_FETCH_2, TSAN_ATOMIC16_FETCH_OR, BIT_IOR_EXPR), | |
423 | FETCH_OPS (SYNC_OR_AND_FETCH_4, TSAN_ATOMIC32_FETCH_OR, BIT_IOR_EXPR), | |
424 | FETCH_OPS (SYNC_OR_AND_FETCH_8, TSAN_ATOMIC64_FETCH_OR, BIT_IOR_EXPR), | |
425 | FETCH_OPS (SYNC_OR_AND_FETCH_16, TSAN_ATOMIC128_FETCH_OR, BIT_IOR_EXPR), | |
426 | FETCH_OPS (SYNC_XOR_AND_FETCH_1, TSAN_ATOMIC8_FETCH_XOR, BIT_XOR_EXPR), | |
427 | FETCH_OPS (SYNC_XOR_AND_FETCH_2, TSAN_ATOMIC16_FETCH_XOR, BIT_XOR_EXPR), | |
428 | FETCH_OPS (SYNC_XOR_AND_FETCH_4, TSAN_ATOMIC32_FETCH_XOR, BIT_XOR_EXPR), | |
429 | FETCH_OPS (SYNC_XOR_AND_FETCH_8, TSAN_ATOMIC64_FETCH_XOR, BIT_XOR_EXPR), | |
430 | FETCH_OPS (SYNC_XOR_AND_FETCH_16, TSAN_ATOMIC128_FETCH_XOR, BIT_XOR_EXPR), | |
431 | FETCH_OPS (SYNC_NAND_AND_FETCH_1, TSAN_ATOMIC8_FETCH_NAND, BIT_NOT_EXPR), | |
432 | FETCH_OPS (SYNC_NAND_AND_FETCH_2, TSAN_ATOMIC16_FETCH_NAND, BIT_NOT_EXPR), | |
433 | FETCH_OPS (SYNC_NAND_AND_FETCH_4, TSAN_ATOMIC32_FETCH_NAND, BIT_NOT_EXPR), | |
434 | FETCH_OPS (SYNC_NAND_AND_FETCH_8, TSAN_ATOMIC64_FETCH_NAND, BIT_NOT_EXPR), | |
435 | FETCH_OPS (SYNC_NAND_AND_FETCH_16, TSAN_ATOMIC128_FETCH_NAND, BIT_NOT_EXPR), | |
436 | ||
437 | WEAK_CAS (ATOMIC_COMPARE_EXCHANGE_1, TSAN_ATOMIC8_COMPARE_EXCHANGE_WEAK), | |
438 | WEAK_CAS (ATOMIC_COMPARE_EXCHANGE_2, TSAN_ATOMIC16_COMPARE_EXCHANGE_WEAK), | |
439 | WEAK_CAS (ATOMIC_COMPARE_EXCHANGE_4, TSAN_ATOMIC32_COMPARE_EXCHANGE_WEAK), | |
440 | WEAK_CAS (ATOMIC_COMPARE_EXCHANGE_8, TSAN_ATOMIC64_COMPARE_EXCHANGE_WEAK), | |
441 | WEAK_CAS (ATOMIC_COMPARE_EXCHANGE_16, TSAN_ATOMIC128_COMPARE_EXCHANGE_WEAK), | |
442 | ||
443 | STRONG_CAS (ATOMIC_COMPARE_EXCHANGE_1, TSAN_ATOMIC8_COMPARE_EXCHANGE_STRONG), | |
444 | STRONG_CAS (ATOMIC_COMPARE_EXCHANGE_2, | |
445 | TSAN_ATOMIC16_COMPARE_EXCHANGE_STRONG), | |
446 | STRONG_CAS (ATOMIC_COMPARE_EXCHANGE_4, | |
447 | TSAN_ATOMIC32_COMPARE_EXCHANGE_STRONG), | |
448 | STRONG_CAS (ATOMIC_COMPARE_EXCHANGE_8, | |
449 | TSAN_ATOMIC64_COMPARE_EXCHANGE_STRONG), | |
450 | STRONG_CAS (ATOMIC_COMPARE_EXCHANGE_16, | |
451 | TSAN_ATOMIC128_COMPARE_EXCHANGE_STRONG), | |
452 | ||
453 | BOOL_CAS (SYNC_BOOL_COMPARE_AND_SWAP_1, | |
454 | TSAN_ATOMIC8_COMPARE_EXCHANGE_STRONG), | |
455 | BOOL_CAS (SYNC_BOOL_COMPARE_AND_SWAP_2, | |
456 | TSAN_ATOMIC16_COMPARE_EXCHANGE_STRONG), | |
457 | BOOL_CAS (SYNC_BOOL_COMPARE_AND_SWAP_4, | |
458 | TSAN_ATOMIC32_COMPARE_EXCHANGE_STRONG), | |
459 | BOOL_CAS (SYNC_BOOL_COMPARE_AND_SWAP_8, | |
460 | TSAN_ATOMIC64_COMPARE_EXCHANGE_STRONG), | |
461 | BOOL_CAS (SYNC_BOOL_COMPARE_AND_SWAP_16, | |
462 | TSAN_ATOMIC128_COMPARE_EXCHANGE_STRONG), | |
463 | ||
464 | VAL_CAS (SYNC_VAL_COMPARE_AND_SWAP_1, TSAN_ATOMIC8_COMPARE_EXCHANGE_STRONG), | |
465 | VAL_CAS (SYNC_VAL_COMPARE_AND_SWAP_2, TSAN_ATOMIC16_COMPARE_EXCHANGE_STRONG), | |
466 | VAL_CAS (SYNC_VAL_COMPARE_AND_SWAP_4, TSAN_ATOMIC32_COMPARE_EXCHANGE_STRONG), | |
467 | VAL_CAS (SYNC_VAL_COMPARE_AND_SWAP_8, TSAN_ATOMIC64_COMPARE_EXCHANGE_STRONG), | |
468 | VAL_CAS (SYNC_VAL_COMPARE_AND_SWAP_16, | |
469 | TSAN_ATOMIC128_COMPARE_EXCHANGE_STRONG), | |
470 | ||
471 | LOCK_RELEASE (SYNC_LOCK_RELEASE_1, TSAN_ATOMIC8_STORE), | |
472 | LOCK_RELEASE (SYNC_LOCK_RELEASE_2, TSAN_ATOMIC16_STORE), | |
473 | LOCK_RELEASE (SYNC_LOCK_RELEASE_4, TSAN_ATOMIC32_STORE), | |
474 | LOCK_RELEASE (SYNC_LOCK_RELEASE_8, TSAN_ATOMIC64_STORE), | |
475 | LOCK_RELEASE (SYNC_LOCK_RELEASE_16, TSAN_ATOMIC128_STORE), | |
476 | ||
477 | BOOL_CLEAR (ATOMIC_CLEAR, TSAN_ATOMIC8_STORE), | |
478 | ||
479 | BOOL_TEST_AND_SET (ATOMIC_TEST_AND_SET, TSAN_ATOMIC8_EXCHANGE) | |
480 | }; | |
481 | ||
482 | /* Instrument an atomic builtin. */ | |
483 | ||
484 | static void | |
485 | instrument_builtin_call (gimple_stmt_iterator *gsi) | |
486 | { | |
487 | gimple *stmt = gsi_stmt (*gsi), *g; | |
488 | tree callee = gimple_call_fndecl (stmt), last_arg, args[6], t, lhs; | |
489 | enum built_in_function fcode = DECL_FUNCTION_CODE (callee); | |
490 | unsigned int i, num = gimple_call_num_args (stmt), j; | |
491 | for (j = 0; j < 6 && j < num; j++) | |
492 | args[j] = gimple_call_arg (stmt, j); | |
493 | for (i = 0; i < ARRAY_SIZE (tsan_atomic_table); i++) | |
494 | if (fcode != tsan_atomic_table[i].fcode) | |
495 | continue; | |
496 | else | |
497 | { | |
498 | tree decl = builtin_decl_implicit (tsan_atomic_table[i].tsan_fcode); | |
499 | if (decl == NULL_TREE) | |
500 | return; | |
501 | switch (tsan_atomic_table[i].action) | |
502 | { | |
503 | case check_last: | |
504 | case fetch_op: | |
505 | last_arg = gimple_call_arg (stmt, num - 1); | |
506 | if (tree_fits_uhwi_p (last_arg) | |
507 | && memmodel_base (tree_to_uhwi (last_arg)) >= MEMMODEL_LAST) | |
508 | return; | |
509 | gimple_call_set_fndecl (stmt, decl); | |
510 | update_stmt (stmt); | |
511 | maybe_clean_eh_stmt (stmt); | |
512 | if (tsan_atomic_table[i].action == fetch_op) | |
513 | { | |
514 | args[1] = gimple_call_arg (stmt, 1); | |
515 | goto adjust_result; | |
516 | } | |
517 | return; | |
518 | case add_seq_cst: | |
519 | case add_acquire: | |
520 | case fetch_op_seq_cst: | |
521 | gcc_assert (num <= 2); | |
522 | for (j = 0; j < num; j++) | |
523 | args[j] = gimple_call_arg (stmt, j); | |
524 | for (; j < 2; j++) | |
525 | args[j] = NULL_TREE; | |
526 | args[num] = build_int_cst (NULL_TREE, | |
527 | tsan_atomic_table[i].action | |
528 | != add_acquire | |
529 | ? MEMMODEL_SEQ_CST | |
530 | : MEMMODEL_ACQUIRE); | |
531 | update_gimple_call (gsi, decl, num + 1, args[0], args[1], args[2]); | |
532 | maybe_clean_or_replace_eh_stmt (stmt, gsi_stmt (*gsi)); | |
533 | stmt = gsi_stmt (*gsi); | |
534 | if (tsan_atomic_table[i].action == fetch_op_seq_cst) | |
535 | { | |
536 | adjust_result: | |
537 | lhs = gimple_call_lhs (stmt); | |
538 | if (lhs == NULL_TREE) | |
539 | return; | |
540 | if (!useless_type_conversion_p (TREE_TYPE (lhs), | |
541 | TREE_TYPE (args[1]))) | |
542 | { | |
543 | tree var = make_ssa_name (TREE_TYPE (lhs)); | |
544 | g = gimple_build_assign (var, NOP_EXPR, args[1]); | |
545 | gsi_insert_after (gsi, g, GSI_NEW_STMT); | |
546 | args[1] = var; | |
547 | } | |
548 | gimple_call_set_lhs (stmt, make_ssa_name (TREE_TYPE (lhs))); | |
549 | /* BIT_NOT_EXPR stands for NAND. */ | |
550 | if (tsan_atomic_table[i].code == BIT_NOT_EXPR) | |
551 | { | |
552 | tree var = make_ssa_name (TREE_TYPE (lhs)); | |
553 | g = gimple_build_assign (var, BIT_AND_EXPR, | |
554 | gimple_call_lhs (stmt), args[1]); | |
555 | gsi_insert_after (gsi, g, GSI_NEW_STMT); | |
556 | g = gimple_build_assign (lhs, BIT_NOT_EXPR, var); | |
557 | } | |
558 | else | |
559 | g = gimple_build_assign (lhs, tsan_atomic_table[i].code, | |
560 | gimple_call_lhs (stmt), args[1]); | |
561 | update_stmt (stmt); | |
562 | gsi_insert_after (gsi, g, GSI_NEW_STMT); | |
563 | } | |
564 | return; | |
565 | case weak_cas: | |
566 | if (!integer_nonzerop (gimple_call_arg (stmt, 3))) | |
567 | continue; | |
568 | /* FALLTHRU */ | |
569 | case strong_cas: | |
570 | gcc_assert (num == 6); | |
571 | for (j = 0; j < 6; j++) | |
572 | args[j] = gimple_call_arg (stmt, j); | |
573 | if (tree_fits_uhwi_p (args[4]) | |
574 | && memmodel_base (tree_to_uhwi (args[4])) >= MEMMODEL_LAST) | |
575 | return; | |
576 | if (tree_fits_uhwi_p (args[5]) | |
577 | && memmodel_base (tree_to_uhwi (args[5])) >= MEMMODEL_LAST) | |
578 | return; | |
579 | update_gimple_call (gsi, decl, 5, args[0], args[1], args[2], | |
580 | args[4], args[5]); | |
581 | maybe_clean_or_replace_eh_stmt (stmt, gsi_stmt (*gsi)); | |
582 | return; | |
583 | case bool_cas: | |
584 | case val_cas: | |
585 | gcc_assert (num == 3); | |
586 | for (j = 0; j < 3; j++) | |
587 | args[j] = gimple_call_arg (stmt, j); | |
588 | t = TYPE_ARG_TYPES (TREE_TYPE (decl)); | |
589 | t = TREE_VALUE (TREE_CHAIN (TREE_CHAIN (t))); | |
590 | t = create_tmp_var (t); | |
591 | mark_addressable (t); | |
592 | if (!useless_type_conversion_p (TREE_TYPE (t), | |
593 | TREE_TYPE (args[1]))) | |
594 | { | |
595 | g = gimple_build_assign (make_ssa_name (TREE_TYPE (t)), | |
596 | NOP_EXPR, args[1]); | |
597 | gsi_insert_before (gsi, g, GSI_SAME_STMT); | |
598 | args[1] = gimple_assign_lhs (g); | |
599 | } | |
600 | g = gimple_build_assign (t, args[1]); | |
601 | gsi_insert_before (gsi, g, GSI_SAME_STMT); | |
602 | lhs = gimple_call_lhs (stmt); | |
603 | update_gimple_call (gsi, decl, 5, args[0], | |
604 | build_fold_addr_expr (t), args[2], | |
605 | build_int_cst (NULL_TREE, | |
606 | MEMMODEL_SEQ_CST), | |
607 | build_int_cst (NULL_TREE, | |
608 | MEMMODEL_SEQ_CST)); | |
609 | maybe_clean_or_replace_eh_stmt (stmt, gsi_stmt (*gsi)); | |
610 | if (tsan_atomic_table[i].action == val_cas && lhs) | |
611 | { | |
612 | tree cond; | |
613 | stmt = gsi_stmt (*gsi); | |
614 | g = gimple_build_assign (make_ssa_name (TREE_TYPE (t)), t); | |
615 | gsi_insert_after (gsi, g, GSI_NEW_STMT); | |
616 | t = make_ssa_name (TREE_TYPE (TREE_TYPE (decl)), stmt); | |
617 | cond = build2 (NE_EXPR, boolean_type_node, t, | |
618 | build_int_cst (TREE_TYPE (t), 0)); | |
619 | g = gimple_build_assign (lhs, COND_EXPR, cond, args[1], | |
620 | gimple_assign_lhs (g)); | |
621 | gimple_call_set_lhs (stmt, t); | |
622 | update_stmt (stmt); | |
623 | gsi_insert_after (gsi, g, GSI_NEW_STMT); | |
624 | } | |
625 | return; | |
626 | case lock_release: | |
627 | gcc_assert (num == 1); | |
628 | t = TYPE_ARG_TYPES (TREE_TYPE (decl)); | |
629 | t = TREE_VALUE (TREE_CHAIN (t)); | |
630 | update_gimple_call (gsi, decl, 3, gimple_call_arg (stmt, 0), | |
631 | build_int_cst (t, 0), | |
632 | build_int_cst (NULL_TREE, | |
633 | MEMMODEL_RELEASE)); | |
634 | maybe_clean_or_replace_eh_stmt (stmt, gsi_stmt (*gsi)); | |
635 | return; | |
636 | case bool_clear: | |
637 | case bool_test_and_set: | |
638 | if (BOOL_TYPE_SIZE != 8) | |
639 | { | |
640 | decl = NULL_TREE; | |
641 | for (j = 1; j < 5; j++) | |
642 | if (BOOL_TYPE_SIZE == (8 << j)) | |
643 | { | |
644 | enum built_in_function tsan_fcode | |
645 | = (enum built_in_function) | |
646 | (tsan_atomic_table[i].tsan_fcode + j); | |
647 | decl = builtin_decl_implicit (tsan_fcode); | |
648 | break; | |
649 | } | |
650 | if (decl == NULL_TREE) | |
651 | return; | |
652 | } | |
653 | last_arg = gimple_call_arg (stmt, num - 1); | |
654 | if (tree_fits_uhwi_p (last_arg) | |
655 | && memmodel_base (tree_to_uhwi (last_arg)) >= MEMMODEL_LAST) | |
656 | return; | |
657 | t = TYPE_ARG_TYPES (TREE_TYPE (decl)); | |
658 | t = TREE_VALUE (TREE_CHAIN (t)); | |
659 | if (tsan_atomic_table[i].action == bool_clear) | |
660 | { | |
661 | update_gimple_call (gsi, decl, 3, gimple_call_arg (stmt, 0), | |
662 | build_int_cst (t, 0), last_arg); | |
663 | maybe_clean_or_replace_eh_stmt (stmt, gsi_stmt (*gsi)); | |
664 | return; | |
665 | } | |
666 | t = build_int_cst (t, targetm.atomic_test_and_set_trueval); | |
667 | update_gimple_call (gsi, decl, 3, gimple_call_arg (stmt, 0), | |
668 | t, last_arg); | |
669 | maybe_clean_or_replace_eh_stmt (stmt, gsi_stmt (*gsi)); | |
670 | stmt = gsi_stmt (*gsi); | |
671 | lhs = gimple_call_lhs (stmt); | |
672 | if (lhs == NULL_TREE) | |
673 | return; | |
674 | if (targetm.atomic_test_and_set_trueval != 1 | |
675 | || !useless_type_conversion_p (TREE_TYPE (lhs), | |
676 | TREE_TYPE (t))) | |
677 | { | |
678 | tree new_lhs = make_ssa_name (TREE_TYPE (t)); | |
679 | gimple_call_set_lhs (stmt, new_lhs); | |
680 | if (targetm.atomic_test_and_set_trueval != 1) | |
681 | g = gimple_build_assign (lhs, NE_EXPR, new_lhs, | |
682 | build_int_cst (TREE_TYPE (t), 0)); | |
683 | else | |
684 | g = gimple_build_assign (lhs, NOP_EXPR, new_lhs); | |
685 | gsi_insert_after (gsi, g, GSI_NEW_STMT); | |
686 | update_stmt (stmt); | |
687 | } | |
688 | return; | |
689 | default: | |
690 | continue; | |
691 | } | |
692 | } | |
693 | } | |
694 | ||
695 | /* Instruments the gimple pointed to by GSI. Return | |
696 | true if func entry/exit should be instrumented. */ | |
697 | ||
698 | static bool | |
699 | instrument_gimple (gimple_stmt_iterator *gsi) | |
700 | { | |
701 | gimple *stmt; | |
702 | tree rhs, lhs; | |
703 | bool instrumented = false; | |
704 | ||
705 | stmt = gsi_stmt (*gsi); | |
706 | if (is_gimple_call (stmt) | |
707 | && (gimple_call_fndecl (stmt) | |
708 | != builtin_decl_implicit (BUILT_IN_TSAN_INIT))) | |
709 | { | |
710 | /* All functions with function call will have exit instrumented, | |
711 | therefore no function calls other than __tsan_func_exit | |
712 | shall appear in the functions. */ | |
713 | gimple_call_set_tail (as_a <gcall *> (stmt), false); | |
714 | if (gimple_call_builtin_p (stmt, BUILT_IN_NORMAL)) | |
715 | instrument_builtin_call (gsi); | |
716 | return true; | |
717 | } | |
718 | else if (is_gimple_assign (stmt) | |
719 | && !gimple_clobber_p (stmt)) | |
720 | { | |
721 | if (gimple_store_p (stmt)) | |
722 | { | |
723 | lhs = gimple_assign_lhs (stmt); | |
724 | instrumented = instrument_expr (*gsi, lhs, true); | |
725 | } | |
726 | if (gimple_assign_load_p (stmt)) | |
727 | { | |
728 | rhs = gimple_assign_rhs1 (stmt); | |
729 | instrumented = instrument_expr (*gsi, rhs, false); | |
730 | } | |
731 | } | |
732 | return instrumented; | |
733 | } | |
734 | ||
735 | /* Replace TSAN_FUNC_EXIT internal call with function exit tsan builtin. */ | |
736 | ||
737 | static void | |
738 | replace_func_exit (gimple *stmt) | |
739 | { | |
740 | tree builtin_decl = builtin_decl_implicit (BUILT_IN_TSAN_FUNC_EXIT); | |
741 | gimple *g = gimple_build_call (builtin_decl, 0); | |
742 | gimple_set_location (g, cfun->function_end_locus); | |
743 | gimple_stmt_iterator gsi = gsi_for_stmt (stmt); | |
744 | gsi_replace (&gsi, g, true); | |
745 | } | |
746 | ||
747 | /* Instrument function exit. Used when TSAN_FUNC_EXIT does not exist. */ | |
748 | ||
749 | static void | |
750 | instrument_func_exit (void) | |
751 | { | |
752 | location_t loc; | |
753 | basic_block exit_bb; | |
754 | gimple_stmt_iterator gsi; | |
755 | gimple *stmt, *g; | |
756 | tree builtin_decl; | |
757 | edge e; | |
758 | edge_iterator ei; | |
759 | ||
760 | /* Find all function exits. */ | |
761 | exit_bb = EXIT_BLOCK_PTR_FOR_FN (cfun); | |
762 | FOR_EACH_EDGE (e, ei, exit_bb->preds) | |
763 | { | |
764 | gsi = gsi_last_bb (e->src); | |
765 | stmt = gsi_stmt (gsi); | |
766 | gcc_assert (gimple_code (stmt) == GIMPLE_RETURN | |
767 | || gimple_call_builtin_p (stmt, BUILT_IN_RETURN)); | |
768 | loc = gimple_location (stmt); | |
769 | builtin_decl = builtin_decl_implicit (BUILT_IN_TSAN_FUNC_EXIT); | |
770 | g = gimple_build_call (builtin_decl, 0); | |
771 | gimple_set_location (g, loc); | |
772 | gsi_insert_before (&gsi, g, GSI_SAME_STMT); | |
773 | } | |
774 | } | |
775 | ||
776 | /* Instruments all interesting memory accesses in the current function. | |
777 | Return true if func entry/exit should be instrumented. */ | |
778 | ||
779 | static bool | |
780 | instrument_memory_accesses (bool *cfg_changed) | |
781 | { | |
782 | basic_block bb; | |
783 | gimple_stmt_iterator gsi; | |
784 | bool fentry_exit_instrument = false; | |
785 | bool func_exit_seen = false; | |
786 | auto_vec<gimple *> tsan_func_exits; | |
787 | ||
788 | FOR_EACH_BB_FN (bb, cfun) | |
789 | { | |
790 | for (gsi = gsi_start_bb (bb); !gsi_end_p (gsi); gsi_next (&gsi)) | |
791 | { | |
792 | gimple *stmt = gsi_stmt (gsi); | |
793 | if (gimple_call_internal_p (stmt, IFN_TSAN_FUNC_EXIT)) | |
794 | { | |
795 | if (fentry_exit_instrument) | |
796 | replace_func_exit (stmt); | |
797 | else | |
798 | tsan_func_exits.safe_push (stmt); | |
799 | func_exit_seen = true; | |
800 | } | |
801 | else | |
802 | fentry_exit_instrument |= instrument_gimple (&gsi); | |
803 | } | |
804 | if (gimple_purge_dead_eh_edges (bb)) | |
805 | *cfg_changed = true; | |
806 | } | |
807 | unsigned int i; | |
808 | gimple *stmt; | |
809 | FOR_EACH_VEC_ELT (tsan_func_exits, i, stmt) | |
810 | if (fentry_exit_instrument) | |
811 | replace_func_exit (stmt); | |
812 | else | |
813 | { | |
814 | gsi = gsi_for_stmt (stmt); | |
815 | gsi_remove (&gsi, true); | |
816 | } | |
817 | if (fentry_exit_instrument && !func_exit_seen) | |
818 | instrument_func_exit (); | |
819 | return fentry_exit_instrument; | |
820 | } | |
821 | ||
822 | /* Instruments function entry. */ | |
823 | ||
824 | static void | |
825 | instrument_func_entry (void) | |
826 | { | |
827 | tree ret_addr, builtin_decl; | |
828 | gimple *g; | |
829 | gimple_seq seq = NULL; | |
830 | ||
831 | builtin_decl = builtin_decl_implicit (BUILT_IN_RETURN_ADDRESS); | |
832 | g = gimple_build_call (builtin_decl, 1, integer_zero_node); | |
833 | ret_addr = make_ssa_name (ptr_type_node); | |
834 | gimple_call_set_lhs (g, ret_addr); | |
835 | gimple_set_location (g, cfun->function_start_locus); | |
836 | gimple_seq_add_stmt_without_update (&seq, g); | |
837 | ||
838 | builtin_decl = builtin_decl_implicit (BUILT_IN_TSAN_FUNC_ENTRY); | |
839 | g = gimple_build_call (builtin_decl, 1, ret_addr); | |
840 | gimple_set_location (g, cfun->function_start_locus); | |
841 | gimple_seq_add_stmt_without_update (&seq, g); | |
842 | ||
843 | edge e = single_succ_edge (ENTRY_BLOCK_PTR_FOR_FN (cfun)); | |
844 | gsi_insert_seq_on_edge_immediate (e, seq); | |
845 | } | |
846 | ||
847 | /* ThreadSanitizer instrumentation pass. */ | |
848 | ||
849 | static unsigned | |
850 | tsan_pass (void) | |
851 | { | |
852 | initialize_sanitizer_builtins (); | |
853 | bool cfg_changed = false; | |
854 | if (instrument_memory_accesses (&cfg_changed)) | |
855 | instrument_func_entry (); | |
856 | return cfg_changed ? TODO_cleanup_cfg : 0; | |
857 | } | |
858 | ||
859 | /* Inserts __tsan_init () into the list of CTORs. */ | |
860 | ||
861 | void | |
862 | tsan_finish_file (void) | |
863 | { | |
864 | tree ctor_statements = NULL_TREE; | |
865 | ||
866 | initialize_sanitizer_builtins (); | |
867 | tree init_decl = builtin_decl_implicit (BUILT_IN_TSAN_INIT); | |
868 | append_to_statement_list (build_call_expr (init_decl, 0), | |
869 | &ctor_statements); | |
870 | cgraph_build_static_cdtor ('I', ctor_statements, | |
871 | MAX_RESERVED_INIT_PRIORITY - 1); | |
872 | } | |
873 | ||
874 | /* The pass descriptor. */ | |
875 | ||
876 | namespace { | |
877 | ||
878 | const pass_data pass_data_tsan = | |
879 | { | |
880 | GIMPLE_PASS, /* type */ | |
881 | "tsan", /* name */ | |
882 | OPTGROUP_NONE, /* optinfo_flags */ | |
883 | TV_NONE, /* tv_id */ | |
884 | ( PROP_ssa | PROP_cfg ), /* properties_required */ | |
885 | 0, /* properties_provided */ | |
886 | 0, /* properties_destroyed */ | |
887 | 0, /* todo_flags_start */ | |
888 | TODO_update_ssa, /* todo_flags_finish */ | |
889 | }; | |
890 | ||
891 | class pass_tsan : public gimple_opt_pass | |
892 | { | |
893 | public: | |
894 | pass_tsan (gcc::context *ctxt) | |
895 | : gimple_opt_pass (pass_data_tsan, ctxt) | |
896 | {} | |
897 | ||
898 | /* opt_pass methods: */ | |
899 | opt_pass * clone () { return new pass_tsan (m_ctxt); } | |
900 | virtual bool gate (function *) | |
901 | { | |
902 | return sanitize_flags_p (SANITIZE_THREAD); | |
903 | } | |
904 | ||
905 | virtual unsigned int execute (function *) { return tsan_pass (); } | |
906 | ||
907 | }; // class pass_tsan | |
908 | ||
909 | } // anon namespace | |
910 | ||
911 | gimple_opt_pass * | |
912 | make_pass_tsan (gcc::context *ctxt) | |
913 | { | |
914 | return new pass_tsan (ctxt); | |
915 | } | |
916 | ||
917 | namespace { | |
918 | ||
919 | const pass_data pass_data_tsan_O0 = | |
920 | { | |
921 | GIMPLE_PASS, /* type */ | |
922 | "tsan0", /* name */ | |
923 | OPTGROUP_NONE, /* optinfo_flags */ | |
924 | TV_NONE, /* tv_id */ | |
925 | ( PROP_ssa | PROP_cfg ), /* properties_required */ | |
926 | 0, /* properties_provided */ | |
927 | 0, /* properties_destroyed */ | |
928 | 0, /* todo_flags_start */ | |
929 | TODO_update_ssa, /* todo_flags_finish */ | |
930 | }; | |
931 | ||
932 | class pass_tsan_O0 : public gimple_opt_pass | |
933 | { | |
934 | public: | |
935 | pass_tsan_O0 (gcc::context *ctxt) | |
936 | : gimple_opt_pass (pass_data_tsan_O0, ctxt) | |
937 | {} | |
938 | ||
939 | /* opt_pass methods: */ | |
940 | virtual bool gate (function *) | |
941 | { | |
942 | return (sanitize_flags_p (SANITIZE_THREAD) && !optimize); | |
943 | } | |
944 | ||
945 | virtual unsigned int execute (function *) { return tsan_pass (); } | |
946 | ||
947 | }; // class pass_tsan_O0 | |
948 | ||
949 | } // anon namespace | |
950 | ||
951 | gimple_opt_pass * | |
952 | make_pass_tsan_O0 (gcc::context *ctxt) | |
953 | { | |
954 | return new pass_tsan_O0 (ctxt); | |
955 | } |