1 /* GCC instrumentation plugin for ThreadSanitizer.
2 Copyright (C) 2011-2013 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 "tree.h"
26 #include "intl.h"
27 #include "tm.h"
28 #include "basic-block.h"
29 #include "gimple.h"
30 #include "function.h"
31 #include "tree-flow.h"
32 #include "tree-pass.h"
33 #include "tree-iterator.h"
34 #include "langhooks.h"
35 #include "output.h"
36 #include "options.h"
37 #include "target.h"
38 #include "cgraph.h"
39 #include "diagnostic.h"
40 #include "tree-ssa-propagate.h"
41 #include "tsan.h"
42 #include "asan.h"
43
44 /* Number of instrumented memory accesses in the current function. */
45
46 /* Builds the following decl
47 void __tsan_read/writeX (void *addr); */
48
49 static tree
get_memory_access_decl(bool is_write,unsigned size)50 get_memory_access_decl (bool is_write, unsigned size)
51 {
52 enum built_in_function fcode;
53
54 if (size <= 1)
55 fcode = is_write ? BUILT_IN_TSAN_WRITE1
56 : BUILT_IN_TSAN_READ1;
57 else if (size <= 3)
58 fcode = is_write ? BUILT_IN_TSAN_WRITE2
59 : BUILT_IN_TSAN_READ2;
60 else if (size <= 7)
61 fcode = is_write ? BUILT_IN_TSAN_WRITE4
62 : BUILT_IN_TSAN_READ4;
63 else if (size <= 15)
64 fcode = is_write ? BUILT_IN_TSAN_WRITE8
65 : BUILT_IN_TSAN_READ8;
66 else
67 fcode = is_write ? BUILT_IN_TSAN_WRITE16
68 : BUILT_IN_TSAN_READ16;
69
70 return builtin_decl_implicit (fcode);
71 }
72
73 /* Check as to whether EXPR refers to a store to vptr. */
74
75 static tree
is_vptr_store(gimple stmt,tree expr,bool is_write)76 is_vptr_store (gimple stmt, tree expr, bool is_write)
77 {
78 if (is_write == true
79 && gimple_assign_single_p (stmt)
80 && TREE_CODE (expr) == COMPONENT_REF)
81 {
82 tree field = TREE_OPERAND (expr, 1);
83 if (TREE_CODE (field) == FIELD_DECL
84 && DECL_VIRTUAL_P (field))
85 return gimple_assign_rhs1 (stmt);
86 }
87 return NULL;
88 }
89
90 /* Instruments EXPR if needed. If any instrumentation is inserted,
91 return true. */
92
93 static bool
instrument_expr(gimple_stmt_iterator gsi,tree expr,bool is_write)94 instrument_expr (gimple_stmt_iterator gsi, tree expr, bool is_write)
95 {
96 tree base, rhs, expr_ptr, builtin_decl;
97 basic_block bb;
98 HOST_WIDE_INT size;
99 gimple stmt, g;
100 gimple_seq seq;
101 location_t loc;
102
103 size = int_size_in_bytes (TREE_TYPE (expr));
104 if (size == -1)
105 return false;
106
107 /* For now just avoid instrumenting bit field acceses.
108 TODO: handle bit-fields as if touching the whole field. */
109 HOST_WIDE_INT bitsize, bitpos;
110 tree offset;
111 enum machine_mode mode;
112 int volatilep = 0, unsignedp = 0;
113 base = get_inner_reference (expr, &bitsize, &bitpos, &offset,
114 &mode, &unsignedp, &volatilep, false);
115
116 /* No need to instrument accesses to decls that don't escape,
117 they can't escape to other threads then. */
118 if (DECL_P (base))
119 {
120 struct pt_solution pt;
121 memset (&pt, 0, sizeof (pt));
122 pt.escaped = 1;
123 pt.ipa_escaped = flag_ipa_pta != 0;
124 pt.nonlocal = 1;
125 if (!pt_solution_includes (&pt, base))
126 return false;
127 if (!is_global_var (base) && !may_be_aliased (base))
128 return false;
129 }
130
131 if (TREE_READONLY (base)
132 || (TREE_CODE (base) == VAR_DECL
133 && DECL_HARD_REGISTER (base)))
134 return false;
135
136 if (size == 0
137 || bitpos % (size * BITS_PER_UNIT)
138 || bitsize != size * BITS_PER_UNIT)
139 return false;
140
141 stmt = gsi_stmt (gsi);
142 loc = gimple_location (stmt);
143 rhs = is_vptr_store (stmt, expr, is_write);
144 gcc_checking_assert (rhs != NULL || is_gimple_addressable (expr));
145 expr_ptr = build_fold_addr_expr (unshare_expr (expr));
146 seq = NULL;
147 if (!is_gimple_val (expr_ptr))
148 {
149 g = gimple_build_assign (make_ssa_name (TREE_TYPE (expr_ptr), NULL),
150 expr_ptr);
151 expr_ptr = gimple_assign_lhs (g);
152 gimple_set_location (g, loc);
153 gimple_seq_add_stmt_without_update (&seq, g);
154 }
155 if (rhs == NULL)
156 g = gimple_build_call (get_memory_access_decl (is_write, size),
157 1, expr_ptr);
158 else
159 {
160 builtin_decl = builtin_decl_implicit (BUILT_IN_TSAN_VPTR_UPDATE);
161 g = gimple_build_call (builtin_decl, 1, expr_ptr);
162 }
163 gimple_set_location (g, loc);
164 gimple_seq_add_stmt_without_update (&seq, g);
165 /* Instrumentation for assignment of a function result
166 must be inserted after the call. Instrumentation for
167 reads of function arguments must be inserted before the call.
168 That's because the call can contain synchronization. */
169 if (is_gimple_call (stmt) && is_write)
170 {
171 /* If the call can throw, it must be the last stmt in
172 a basic block, so the instrumented stmts need to be
173 inserted in successor bbs. */
174 if (is_ctrl_altering_stmt (stmt))
175 {
176 edge e;
177
178 bb = gsi_bb (gsi);
179 e = find_fallthru_edge (bb->succs);
180 if (e)
181 gsi_insert_seq_on_edge_immediate (e, seq);
182 }
183 else
184 gsi_insert_seq_after (&gsi, seq, GSI_NEW_STMT);
185 }
186 else
187 gsi_insert_seq_before (&gsi, seq, GSI_SAME_STMT);
188
189 return true;
190 }
191
192 /* Actions for sync/atomic builtin transformations. */
193 enum tsan_atomic_action
194 {
195 check_last, add_seq_cst, add_acquire, weak_cas, strong_cas,
196 bool_cas, val_cas, lock_release, fetch_op, fetch_op_seq_cst
197 };
198
199 /* Table how to map sync/atomic builtins to their corresponding
200 tsan equivalents. */
201 static struct tsan_map_atomic
202 {
203 enum built_in_function fcode, tsan_fcode;
204 enum tsan_atomic_action action;
205 enum tree_code code;
206 } tsan_atomic_table[] =
207 {
208 #define TRANSFORM(fcode, tsan_fcode, action, code) \
209 { BUILT_IN_##fcode, BUILT_IN_##tsan_fcode, action, code }
210 #define CHECK_LAST(fcode, tsan_fcode) \
211 TRANSFORM (fcode, tsan_fcode, check_last, ERROR_MARK)
212 #define ADD_SEQ_CST(fcode, tsan_fcode) \
213 TRANSFORM (fcode, tsan_fcode, add_seq_cst, ERROR_MARK)
214 #define ADD_ACQUIRE(fcode, tsan_fcode) \
215 TRANSFORM (fcode, tsan_fcode, add_acquire, ERROR_MARK)
216 #define WEAK_CAS(fcode, tsan_fcode) \
217 TRANSFORM (fcode, tsan_fcode, weak_cas, ERROR_MARK)
218 #define STRONG_CAS(fcode, tsan_fcode) \
219 TRANSFORM (fcode, tsan_fcode, strong_cas, ERROR_MARK)
220 #define BOOL_CAS(fcode, tsan_fcode) \
221 TRANSFORM (fcode, tsan_fcode, bool_cas, ERROR_MARK)
222 #define VAL_CAS(fcode, tsan_fcode) \
223 TRANSFORM (fcode, tsan_fcode, val_cas, ERROR_MARK)
224 #define LOCK_RELEASE(fcode, tsan_fcode) \
225 TRANSFORM (fcode, tsan_fcode, lock_release, ERROR_MARK)
226 #define FETCH_OP(fcode, tsan_fcode, code) \
227 TRANSFORM (fcode, tsan_fcode, fetch_op, code)
228 #define FETCH_OPS(fcode, tsan_fcode, code) \
229 TRANSFORM (fcode, tsan_fcode, fetch_op_seq_cst, code)
230
231 CHECK_LAST (ATOMIC_LOAD_1, TSAN_ATOMIC8_LOAD),
232 CHECK_LAST (ATOMIC_LOAD_2, TSAN_ATOMIC16_LOAD),
233 CHECK_LAST (ATOMIC_LOAD_4, TSAN_ATOMIC32_LOAD),
234 CHECK_LAST (ATOMIC_LOAD_8, TSAN_ATOMIC64_LOAD),
235 CHECK_LAST (ATOMIC_LOAD_16, TSAN_ATOMIC128_LOAD),
236 CHECK_LAST (ATOMIC_STORE_1, TSAN_ATOMIC8_STORE),
237 CHECK_LAST (ATOMIC_STORE_2, TSAN_ATOMIC16_STORE),
238 CHECK_LAST (ATOMIC_STORE_4, TSAN_ATOMIC32_STORE),
239 CHECK_LAST (ATOMIC_STORE_8, TSAN_ATOMIC64_STORE),
240 CHECK_LAST (ATOMIC_STORE_16, TSAN_ATOMIC128_STORE),
241 CHECK_LAST (ATOMIC_EXCHANGE_1, TSAN_ATOMIC8_EXCHANGE),
242 CHECK_LAST (ATOMIC_EXCHANGE_2, TSAN_ATOMIC16_EXCHANGE),
243 CHECK_LAST (ATOMIC_EXCHANGE_4, TSAN_ATOMIC32_EXCHANGE),
244 CHECK_LAST (ATOMIC_EXCHANGE_8, TSAN_ATOMIC64_EXCHANGE),
245 CHECK_LAST (ATOMIC_EXCHANGE_16, TSAN_ATOMIC128_EXCHANGE),
246 CHECK_LAST (ATOMIC_FETCH_ADD_1, TSAN_ATOMIC8_FETCH_ADD),
247 CHECK_LAST (ATOMIC_FETCH_ADD_2, TSAN_ATOMIC16_FETCH_ADD),
248 CHECK_LAST (ATOMIC_FETCH_ADD_4, TSAN_ATOMIC32_FETCH_ADD),
249 CHECK_LAST (ATOMIC_FETCH_ADD_8, TSAN_ATOMIC64_FETCH_ADD),
250 CHECK_LAST (ATOMIC_FETCH_ADD_16, TSAN_ATOMIC128_FETCH_ADD),
251 CHECK_LAST (ATOMIC_FETCH_SUB_1, TSAN_ATOMIC8_FETCH_SUB),
252 CHECK_LAST (ATOMIC_FETCH_SUB_2, TSAN_ATOMIC16_FETCH_SUB),
253 CHECK_LAST (ATOMIC_FETCH_SUB_4, TSAN_ATOMIC32_FETCH_SUB),
254 CHECK_LAST (ATOMIC_FETCH_SUB_8, TSAN_ATOMIC64_FETCH_SUB),
255 CHECK_LAST (ATOMIC_FETCH_SUB_16, TSAN_ATOMIC128_FETCH_SUB),
256 CHECK_LAST (ATOMIC_FETCH_AND_1, TSAN_ATOMIC8_FETCH_AND),
257 CHECK_LAST (ATOMIC_FETCH_AND_2, TSAN_ATOMIC16_FETCH_AND),
258 CHECK_LAST (ATOMIC_FETCH_AND_4, TSAN_ATOMIC32_FETCH_AND),
259 CHECK_LAST (ATOMIC_FETCH_AND_8, TSAN_ATOMIC64_FETCH_AND),
260 CHECK_LAST (ATOMIC_FETCH_AND_16, TSAN_ATOMIC128_FETCH_AND),
261 CHECK_LAST (ATOMIC_FETCH_OR_1, TSAN_ATOMIC8_FETCH_OR),
262 CHECK_LAST (ATOMIC_FETCH_OR_2, TSAN_ATOMIC16_FETCH_OR),
263 CHECK_LAST (ATOMIC_FETCH_OR_4, TSAN_ATOMIC32_FETCH_OR),
264 CHECK_LAST (ATOMIC_FETCH_OR_8, TSAN_ATOMIC64_FETCH_OR),
265 CHECK_LAST (ATOMIC_FETCH_OR_16, TSAN_ATOMIC128_FETCH_OR),
266 CHECK_LAST (ATOMIC_FETCH_XOR_1, TSAN_ATOMIC8_FETCH_XOR),
267 CHECK_LAST (ATOMIC_FETCH_XOR_2, TSAN_ATOMIC16_FETCH_XOR),
268 CHECK_LAST (ATOMIC_FETCH_XOR_4, TSAN_ATOMIC32_FETCH_XOR),
269 CHECK_LAST (ATOMIC_FETCH_XOR_8, TSAN_ATOMIC64_FETCH_XOR),
270 CHECK_LAST (ATOMIC_FETCH_XOR_16, TSAN_ATOMIC128_FETCH_XOR),
271 CHECK_LAST (ATOMIC_FETCH_NAND_1, TSAN_ATOMIC8_FETCH_NAND),
272 CHECK_LAST (ATOMIC_FETCH_NAND_2, TSAN_ATOMIC16_FETCH_NAND),
273 CHECK_LAST (ATOMIC_FETCH_NAND_4, TSAN_ATOMIC32_FETCH_NAND),
274 CHECK_LAST (ATOMIC_FETCH_NAND_8, TSAN_ATOMIC64_FETCH_NAND),
275 CHECK_LAST (ATOMIC_FETCH_NAND_16, TSAN_ATOMIC128_FETCH_NAND),
276
277 CHECK_LAST (ATOMIC_THREAD_FENCE, TSAN_ATOMIC_THREAD_FENCE),
278 CHECK_LAST (ATOMIC_SIGNAL_FENCE, TSAN_ATOMIC_SIGNAL_FENCE),
279
280 FETCH_OP (ATOMIC_ADD_FETCH_1, TSAN_ATOMIC8_FETCH_ADD, PLUS_EXPR),
281 FETCH_OP (ATOMIC_ADD_FETCH_2, TSAN_ATOMIC16_FETCH_ADD, PLUS_EXPR),
282 FETCH_OP (ATOMIC_ADD_FETCH_4, TSAN_ATOMIC32_FETCH_ADD, PLUS_EXPR),
283 FETCH_OP (ATOMIC_ADD_FETCH_8, TSAN_ATOMIC64_FETCH_ADD, PLUS_EXPR),
284 FETCH_OP (ATOMIC_ADD_FETCH_16, TSAN_ATOMIC128_FETCH_ADD, PLUS_EXPR),
285 FETCH_OP (ATOMIC_SUB_FETCH_1, TSAN_ATOMIC8_FETCH_SUB, MINUS_EXPR),
286 FETCH_OP (ATOMIC_SUB_FETCH_2, TSAN_ATOMIC16_FETCH_SUB, MINUS_EXPR),
287 FETCH_OP (ATOMIC_SUB_FETCH_4, TSAN_ATOMIC32_FETCH_SUB, MINUS_EXPR),
288 FETCH_OP (ATOMIC_SUB_FETCH_8, TSAN_ATOMIC64_FETCH_SUB, MINUS_EXPR),
289 FETCH_OP (ATOMIC_SUB_FETCH_16, TSAN_ATOMIC128_FETCH_SUB, MINUS_EXPR),
290 FETCH_OP (ATOMIC_AND_FETCH_1, TSAN_ATOMIC8_FETCH_AND, BIT_AND_EXPR),
291 FETCH_OP (ATOMIC_AND_FETCH_2, TSAN_ATOMIC16_FETCH_AND, BIT_AND_EXPR),
292 FETCH_OP (ATOMIC_AND_FETCH_4, TSAN_ATOMIC32_FETCH_AND, BIT_AND_EXPR),
293 FETCH_OP (ATOMIC_AND_FETCH_8, TSAN_ATOMIC64_FETCH_AND, BIT_AND_EXPR),
294 FETCH_OP (ATOMIC_AND_FETCH_16, TSAN_ATOMIC128_FETCH_AND, BIT_AND_EXPR),
295 FETCH_OP (ATOMIC_OR_FETCH_1, TSAN_ATOMIC8_FETCH_OR, BIT_IOR_EXPR),
296 FETCH_OP (ATOMIC_OR_FETCH_2, TSAN_ATOMIC16_FETCH_OR, BIT_IOR_EXPR),
297 FETCH_OP (ATOMIC_OR_FETCH_4, TSAN_ATOMIC32_FETCH_OR, BIT_IOR_EXPR),
298 FETCH_OP (ATOMIC_OR_FETCH_8, TSAN_ATOMIC64_FETCH_OR, BIT_IOR_EXPR),
299 FETCH_OP (ATOMIC_OR_FETCH_16, TSAN_ATOMIC128_FETCH_OR, BIT_IOR_EXPR),
300 FETCH_OP (ATOMIC_XOR_FETCH_1, TSAN_ATOMIC8_FETCH_XOR, BIT_XOR_EXPR),
301 FETCH_OP (ATOMIC_XOR_FETCH_2, TSAN_ATOMIC16_FETCH_XOR, BIT_XOR_EXPR),
302 FETCH_OP (ATOMIC_XOR_FETCH_4, TSAN_ATOMIC32_FETCH_XOR, BIT_XOR_EXPR),
303 FETCH_OP (ATOMIC_XOR_FETCH_8, TSAN_ATOMIC64_FETCH_XOR, BIT_XOR_EXPR),
304 FETCH_OP (ATOMIC_XOR_FETCH_16, TSAN_ATOMIC128_FETCH_XOR, BIT_XOR_EXPR),
305 FETCH_OP (ATOMIC_NAND_FETCH_1, TSAN_ATOMIC8_FETCH_NAND, BIT_NOT_EXPR),
306 FETCH_OP (ATOMIC_NAND_FETCH_2, TSAN_ATOMIC16_FETCH_NAND, BIT_NOT_EXPR),
307 FETCH_OP (ATOMIC_NAND_FETCH_4, TSAN_ATOMIC32_FETCH_NAND, BIT_NOT_EXPR),
308 FETCH_OP (ATOMIC_NAND_FETCH_8, TSAN_ATOMIC64_FETCH_NAND, BIT_NOT_EXPR),
309 FETCH_OP (ATOMIC_NAND_FETCH_16, TSAN_ATOMIC128_FETCH_NAND, BIT_NOT_EXPR),
310
311 ADD_ACQUIRE (SYNC_LOCK_TEST_AND_SET_1, TSAN_ATOMIC8_EXCHANGE),
312 ADD_ACQUIRE (SYNC_LOCK_TEST_AND_SET_2, TSAN_ATOMIC16_EXCHANGE),
313 ADD_ACQUIRE (SYNC_LOCK_TEST_AND_SET_4, TSAN_ATOMIC32_EXCHANGE),
314 ADD_ACQUIRE (SYNC_LOCK_TEST_AND_SET_8, TSAN_ATOMIC64_EXCHANGE),
315 ADD_ACQUIRE (SYNC_LOCK_TEST_AND_SET_16, TSAN_ATOMIC128_EXCHANGE),
316
317 ADD_SEQ_CST (SYNC_FETCH_AND_ADD_1, TSAN_ATOMIC8_FETCH_ADD),
318 ADD_SEQ_CST (SYNC_FETCH_AND_ADD_2, TSAN_ATOMIC16_FETCH_ADD),
319 ADD_SEQ_CST (SYNC_FETCH_AND_ADD_4, TSAN_ATOMIC32_FETCH_ADD),
320 ADD_SEQ_CST (SYNC_FETCH_AND_ADD_8, TSAN_ATOMIC64_FETCH_ADD),
321 ADD_SEQ_CST (SYNC_FETCH_AND_ADD_16, TSAN_ATOMIC128_FETCH_ADD),
322 ADD_SEQ_CST (SYNC_FETCH_AND_SUB_1, TSAN_ATOMIC8_FETCH_SUB),
323 ADD_SEQ_CST (SYNC_FETCH_AND_SUB_2, TSAN_ATOMIC16_FETCH_SUB),
324 ADD_SEQ_CST (SYNC_FETCH_AND_SUB_4, TSAN_ATOMIC32_FETCH_SUB),
325 ADD_SEQ_CST (SYNC_FETCH_AND_SUB_8, TSAN_ATOMIC64_FETCH_SUB),
326 ADD_SEQ_CST (SYNC_FETCH_AND_SUB_16, TSAN_ATOMIC128_FETCH_SUB),
327 ADD_SEQ_CST (SYNC_FETCH_AND_AND_1, TSAN_ATOMIC8_FETCH_AND),
328 ADD_SEQ_CST (SYNC_FETCH_AND_AND_2, TSAN_ATOMIC16_FETCH_AND),
329 ADD_SEQ_CST (SYNC_FETCH_AND_AND_4, TSAN_ATOMIC32_FETCH_AND),
330 ADD_SEQ_CST (SYNC_FETCH_AND_AND_8, TSAN_ATOMIC64_FETCH_AND),
331 ADD_SEQ_CST (SYNC_FETCH_AND_AND_16, TSAN_ATOMIC128_FETCH_AND),
332 ADD_SEQ_CST (SYNC_FETCH_AND_OR_1, TSAN_ATOMIC8_FETCH_OR),
333 ADD_SEQ_CST (SYNC_FETCH_AND_OR_2, TSAN_ATOMIC16_FETCH_OR),
334 ADD_SEQ_CST (SYNC_FETCH_AND_OR_4, TSAN_ATOMIC32_FETCH_OR),
335 ADD_SEQ_CST (SYNC_FETCH_AND_OR_8, TSAN_ATOMIC64_FETCH_OR),
336 ADD_SEQ_CST (SYNC_FETCH_AND_OR_16, TSAN_ATOMIC128_FETCH_OR),
337 ADD_SEQ_CST (SYNC_FETCH_AND_XOR_1, TSAN_ATOMIC8_FETCH_XOR),
338 ADD_SEQ_CST (SYNC_FETCH_AND_XOR_2, TSAN_ATOMIC16_FETCH_XOR),
339 ADD_SEQ_CST (SYNC_FETCH_AND_XOR_4, TSAN_ATOMIC32_FETCH_XOR),
340 ADD_SEQ_CST (SYNC_FETCH_AND_XOR_8, TSAN_ATOMIC64_FETCH_XOR),
341 ADD_SEQ_CST (SYNC_FETCH_AND_XOR_16, TSAN_ATOMIC128_FETCH_XOR),
342 ADD_SEQ_CST (SYNC_FETCH_AND_NAND_1, TSAN_ATOMIC8_FETCH_NAND),
343 ADD_SEQ_CST (SYNC_FETCH_AND_NAND_2, TSAN_ATOMIC16_FETCH_NAND),
344 ADD_SEQ_CST (SYNC_FETCH_AND_NAND_4, TSAN_ATOMIC32_FETCH_NAND),
345 ADD_SEQ_CST (SYNC_FETCH_AND_NAND_8, TSAN_ATOMIC64_FETCH_NAND),
346 ADD_SEQ_CST (SYNC_FETCH_AND_NAND_16, TSAN_ATOMIC128_FETCH_NAND),
347
348 ADD_SEQ_CST (SYNC_SYNCHRONIZE, TSAN_ATOMIC_THREAD_FENCE),
349
350 FETCH_OPS (SYNC_ADD_AND_FETCH_1, TSAN_ATOMIC8_FETCH_ADD, PLUS_EXPR),
351 FETCH_OPS (SYNC_ADD_AND_FETCH_2, TSAN_ATOMIC16_FETCH_ADD, PLUS_EXPR),
352 FETCH_OPS (SYNC_ADD_AND_FETCH_4, TSAN_ATOMIC32_FETCH_ADD, PLUS_EXPR),
353 FETCH_OPS (SYNC_ADD_AND_FETCH_8, TSAN_ATOMIC64_FETCH_ADD, PLUS_EXPR),
354 FETCH_OPS (SYNC_ADD_AND_FETCH_16, TSAN_ATOMIC128_FETCH_ADD, PLUS_EXPR),
355 FETCH_OPS (SYNC_SUB_AND_FETCH_1, TSAN_ATOMIC8_FETCH_SUB, MINUS_EXPR),
356 FETCH_OPS (SYNC_SUB_AND_FETCH_2, TSAN_ATOMIC16_FETCH_SUB, MINUS_EXPR),
357 FETCH_OPS (SYNC_SUB_AND_FETCH_4, TSAN_ATOMIC32_FETCH_SUB, MINUS_EXPR),
358 FETCH_OPS (SYNC_SUB_AND_FETCH_8, TSAN_ATOMIC64_FETCH_SUB, MINUS_EXPR),
359 FETCH_OPS (SYNC_SUB_AND_FETCH_16, TSAN_ATOMIC128_FETCH_SUB, MINUS_EXPR),
360 FETCH_OPS (SYNC_AND_AND_FETCH_1, TSAN_ATOMIC8_FETCH_AND, BIT_AND_EXPR),
361 FETCH_OPS (SYNC_AND_AND_FETCH_2, TSAN_ATOMIC16_FETCH_AND, BIT_AND_EXPR),
362 FETCH_OPS (SYNC_AND_AND_FETCH_4, TSAN_ATOMIC32_FETCH_AND, BIT_AND_EXPR),
363 FETCH_OPS (SYNC_AND_AND_FETCH_8, TSAN_ATOMIC64_FETCH_AND, BIT_AND_EXPR),
364 FETCH_OPS (SYNC_AND_AND_FETCH_16, TSAN_ATOMIC128_FETCH_AND, BIT_AND_EXPR),
365 FETCH_OPS (SYNC_OR_AND_FETCH_1, TSAN_ATOMIC8_FETCH_OR, BIT_IOR_EXPR),
366 FETCH_OPS (SYNC_OR_AND_FETCH_2, TSAN_ATOMIC16_FETCH_OR, BIT_IOR_EXPR),
367 FETCH_OPS (SYNC_OR_AND_FETCH_4, TSAN_ATOMIC32_FETCH_OR, BIT_IOR_EXPR),
368 FETCH_OPS (SYNC_OR_AND_FETCH_8, TSAN_ATOMIC64_FETCH_OR, BIT_IOR_EXPR),
369 FETCH_OPS (SYNC_OR_AND_FETCH_16, TSAN_ATOMIC128_FETCH_OR, BIT_IOR_EXPR),
370 FETCH_OPS (SYNC_XOR_AND_FETCH_1, TSAN_ATOMIC8_FETCH_XOR, BIT_XOR_EXPR),
371 FETCH_OPS (SYNC_XOR_AND_FETCH_2, TSAN_ATOMIC16_FETCH_XOR, BIT_XOR_EXPR),
372 FETCH_OPS (SYNC_XOR_AND_FETCH_4, TSAN_ATOMIC32_FETCH_XOR, BIT_XOR_EXPR),
373 FETCH_OPS (SYNC_XOR_AND_FETCH_8, TSAN_ATOMIC64_FETCH_XOR, BIT_XOR_EXPR),
374 FETCH_OPS (SYNC_XOR_AND_FETCH_16, TSAN_ATOMIC128_FETCH_XOR, BIT_XOR_EXPR),
375 FETCH_OPS (SYNC_NAND_AND_FETCH_1, TSAN_ATOMIC8_FETCH_NAND, BIT_NOT_EXPR),
376 FETCH_OPS (SYNC_NAND_AND_FETCH_2, TSAN_ATOMIC16_FETCH_NAND, BIT_NOT_EXPR),
377 FETCH_OPS (SYNC_NAND_AND_FETCH_4, TSAN_ATOMIC32_FETCH_NAND, BIT_NOT_EXPR),
378 FETCH_OPS (SYNC_NAND_AND_FETCH_8, TSAN_ATOMIC64_FETCH_NAND, BIT_NOT_EXPR),
379 FETCH_OPS (SYNC_NAND_AND_FETCH_16, TSAN_ATOMIC128_FETCH_NAND, BIT_NOT_EXPR),
380
381 WEAK_CAS (ATOMIC_COMPARE_EXCHANGE_1, TSAN_ATOMIC8_COMPARE_EXCHANGE_WEAK),
382 WEAK_CAS (ATOMIC_COMPARE_EXCHANGE_2, TSAN_ATOMIC16_COMPARE_EXCHANGE_WEAK),
383 WEAK_CAS (ATOMIC_COMPARE_EXCHANGE_4, TSAN_ATOMIC32_COMPARE_EXCHANGE_WEAK),
384 WEAK_CAS (ATOMIC_COMPARE_EXCHANGE_8, TSAN_ATOMIC64_COMPARE_EXCHANGE_WEAK),
385 WEAK_CAS (ATOMIC_COMPARE_EXCHANGE_16, TSAN_ATOMIC128_COMPARE_EXCHANGE_WEAK),
386
387 STRONG_CAS (ATOMIC_COMPARE_EXCHANGE_1, TSAN_ATOMIC8_COMPARE_EXCHANGE_STRONG),
388 STRONG_CAS (ATOMIC_COMPARE_EXCHANGE_2,
389 TSAN_ATOMIC16_COMPARE_EXCHANGE_STRONG),
390 STRONG_CAS (ATOMIC_COMPARE_EXCHANGE_4,
391 TSAN_ATOMIC32_COMPARE_EXCHANGE_STRONG),
392 STRONG_CAS (ATOMIC_COMPARE_EXCHANGE_8,
393 TSAN_ATOMIC64_COMPARE_EXCHANGE_STRONG),
394 STRONG_CAS (ATOMIC_COMPARE_EXCHANGE_16,
395 TSAN_ATOMIC128_COMPARE_EXCHANGE_STRONG),
396
397 BOOL_CAS (SYNC_BOOL_COMPARE_AND_SWAP_1,
398 TSAN_ATOMIC8_COMPARE_EXCHANGE_STRONG),
399 BOOL_CAS (SYNC_BOOL_COMPARE_AND_SWAP_2,
400 TSAN_ATOMIC16_COMPARE_EXCHANGE_STRONG),
401 BOOL_CAS (SYNC_BOOL_COMPARE_AND_SWAP_4,
402 TSAN_ATOMIC32_COMPARE_EXCHANGE_STRONG),
403 BOOL_CAS (SYNC_BOOL_COMPARE_AND_SWAP_8,
404 TSAN_ATOMIC64_COMPARE_EXCHANGE_STRONG),
405 BOOL_CAS (SYNC_BOOL_COMPARE_AND_SWAP_16,
406 TSAN_ATOMIC128_COMPARE_EXCHANGE_STRONG),
407
408 VAL_CAS (SYNC_VAL_COMPARE_AND_SWAP_1, TSAN_ATOMIC8_COMPARE_EXCHANGE_STRONG),
409 VAL_CAS (SYNC_VAL_COMPARE_AND_SWAP_2, TSAN_ATOMIC16_COMPARE_EXCHANGE_STRONG),
410 VAL_CAS (SYNC_VAL_COMPARE_AND_SWAP_4, TSAN_ATOMIC32_COMPARE_EXCHANGE_STRONG),
411 VAL_CAS (SYNC_VAL_COMPARE_AND_SWAP_8, TSAN_ATOMIC64_COMPARE_EXCHANGE_STRONG),
412 VAL_CAS (SYNC_VAL_COMPARE_AND_SWAP_16,
413 TSAN_ATOMIC128_COMPARE_EXCHANGE_STRONG),
414
415 LOCK_RELEASE (SYNC_LOCK_RELEASE_1, TSAN_ATOMIC8_STORE),
416 LOCK_RELEASE (SYNC_LOCK_RELEASE_2, TSAN_ATOMIC16_STORE),
417 LOCK_RELEASE (SYNC_LOCK_RELEASE_4, TSAN_ATOMIC32_STORE),
418 LOCK_RELEASE (SYNC_LOCK_RELEASE_8, TSAN_ATOMIC64_STORE),
419 LOCK_RELEASE (SYNC_LOCK_RELEASE_16, TSAN_ATOMIC128_STORE)
420 };
421
422 /* Instrument an atomic builtin. */
423
424 static void
instrument_builtin_call(gimple_stmt_iterator * gsi)425 instrument_builtin_call (gimple_stmt_iterator *gsi)
426 {
427 gimple stmt = gsi_stmt (*gsi), g;
428 tree callee = gimple_call_fndecl (stmt), last_arg, args[6], t, lhs;
429 enum built_in_function fcode = DECL_FUNCTION_CODE (callee);
430 unsigned int i, num = gimple_call_num_args (stmt), j;
431 for (j = 0; j < 6 && j < num; j++)
432 args[j] = gimple_call_arg (stmt, j);
433 for (i = 0; i < ARRAY_SIZE (tsan_atomic_table); i++)
434 if (fcode != tsan_atomic_table[i].fcode)
435 continue;
436 else
437 {
438 tree decl = builtin_decl_implicit (tsan_atomic_table[i].tsan_fcode);
439 if (decl == NULL_TREE)
440 return;
441 switch (tsan_atomic_table[i].action)
442 {
443 case check_last:
444 case fetch_op:
445 last_arg = gimple_call_arg (stmt, num - 1);
446 if (!host_integerp (last_arg, 1)
447 || (unsigned HOST_WIDE_INT) tree_low_cst (last_arg, 1)
448 > MEMMODEL_SEQ_CST)
449 return;
450 gimple_call_set_fndecl (stmt, decl);
451 update_stmt (stmt);
452 if (tsan_atomic_table[i].action == fetch_op)
453 {
454 args[1] = gimple_call_arg (stmt, 1);
455 goto adjust_result;
456 }
457 return;
458 case add_seq_cst:
459 case add_acquire:
460 case fetch_op_seq_cst:
461 gcc_assert (num <= 2);
462 for (j = 0; j < num; j++)
463 args[j] = gimple_call_arg (stmt, j);
464 for (; j < 2; j++)
465 args[j] = NULL_TREE;
466 args[num] = build_int_cst (NULL_TREE,
467 tsan_atomic_table[i].action
468 != add_acquire
469 ? MEMMODEL_SEQ_CST
470 : MEMMODEL_ACQUIRE);
471 update_gimple_call (gsi, decl, num + 1, args[0], args[1], args[2]);
472 stmt = gsi_stmt (*gsi);
473 if (tsan_atomic_table[i].action == fetch_op_seq_cst)
474 {
475 adjust_result:
476 lhs = gimple_call_lhs (stmt);
477 if (lhs == NULL_TREE)
478 return;
479 if (!useless_type_conversion_p (TREE_TYPE (lhs),
480 TREE_TYPE (args[1])))
481 {
482 tree var = make_ssa_name (TREE_TYPE (lhs), NULL);
483 g = gimple_build_assign_with_ops (NOP_EXPR, var,
484 args[1], NULL_TREE);
485 gsi_insert_after (gsi, g, GSI_NEW_STMT);
486 args[1] = var;
487 }
488 gimple_call_set_lhs (stmt,
489 make_ssa_name (TREE_TYPE (lhs), NULL));
490 /* BIT_NOT_EXPR stands for NAND. */
491 if (tsan_atomic_table[i].code == BIT_NOT_EXPR)
492 {
493 tree var = make_ssa_name (TREE_TYPE (lhs), NULL);
494 g = gimple_build_assign_with_ops (BIT_AND_EXPR, var,
495 gimple_call_lhs (stmt),
496 args[1]);
497 gsi_insert_after (gsi, g, GSI_NEW_STMT);
498 g = gimple_build_assign_with_ops (BIT_NOT_EXPR, lhs, var,
499 NULL_TREE);
500 }
501 else
502 g = gimple_build_assign_with_ops (tsan_atomic_table[i].code,
503 lhs,
504 gimple_call_lhs (stmt),
505 args[1]);
506 update_stmt (stmt);
507 gsi_insert_after (gsi, g, GSI_NEW_STMT);
508 }
509 return;
510 case weak_cas:
511 if (!integer_nonzerop (gimple_call_arg (stmt, 3)))
512 continue;
513 /* FALLTHRU */
514 case strong_cas:
515 gcc_assert (num == 6);
516 for (j = 0; j < 6; j++)
517 args[j] = gimple_call_arg (stmt, j);
518 if (!host_integerp (args[4], 1)
519 || (unsigned HOST_WIDE_INT) tree_low_cst (args[4], 1)
520 > MEMMODEL_SEQ_CST)
521 return;
522 if (!host_integerp (args[5], 1)
523 || (unsigned HOST_WIDE_INT) tree_low_cst (args[5], 1)
524 > MEMMODEL_SEQ_CST)
525 return;
526 update_gimple_call (gsi, decl, 5, args[0], args[1], args[2],
527 args[4], args[5]);
528 return;
529 case bool_cas:
530 case val_cas:
531 gcc_assert (num == 3);
532 for (j = 0; j < 3; j++)
533 args[j] = gimple_call_arg (stmt, j);
534 t = TYPE_ARG_TYPES (TREE_TYPE (decl));
535 t = TREE_VALUE (TREE_CHAIN (TREE_CHAIN (t)));
536 t = create_tmp_var (t, NULL);
537 mark_addressable (t);
538 if (!useless_type_conversion_p (TREE_TYPE (t),
539 TREE_TYPE (args[1])))
540 {
541 g = gimple_build_assign_with_ops (NOP_EXPR,
542 make_ssa_name (TREE_TYPE (t),
543 NULL),
544 args[1], NULL_TREE);
545 gsi_insert_before (gsi, g, GSI_SAME_STMT);
546 args[1] = gimple_assign_lhs (g);
547 }
548 g = gimple_build_assign (t, args[1]);
549 gsi_insert_before (gsi, g, GSI_SAME_STMT);
550 lhs = gimple_call_lhs (stmt);
551 update_gimple_call (gsi, decl, 5, args[0],
552 build_fold_addr_expr (t), args[2],
553 build_int_cst (NULL_TREE,
554 MEMMODEL_SEQ_CST),
555 build_int_cst (NULL_TREE,
556 MEMMODEL_SEQ_CST));
557 if (tsan_atomic_table[i].action == val_cas && lhs)
558 {
559 tree cond;
560 stmt = gsi_stmt (*gsi);
561 g = gimple_build_assign (make_ssa_name (TREE_TYPE (t), NULL),
562 t);
563 gsi_insert_after (gsi, g, GSI_NEW_STMT);
564 t = make_ssa_name (TREE_TYPE (TREE_TYPE (decl)), stmt);
565 cond = build2 (NE_EXPR, boolean_type_node, t,
566 build_int_cst (TREE_TYPE (t), 0));
567 g = gimple_build_assign_with_ops (COND_EXPR, lhs, cond,
568 args[1],
569 gimple_assign_lhs (g));
570 gimple_call_set_lhs (stmt, t);
571 update_stmt (stmt);
572 gsi_insert_after (gsi, g, GSI_NEW_STMT);
573 }
574 return;
575 case lock_release:
576 gcc_assert (num == 1);
577 t = TYPE_ARG_TYPES (TREE_TYPE (decl));
578 t = TREE_VALUE (TREE_CHAIN (t));
579 update_gimple_call (gsi, decl, 3, gimple_call_arg (stmt, 0),
580 build_int_cst (t, 0),
581 build_int_cst (NULL_TREE,
582 MEMMODEL_RELEASE));
583 return;
584 default:
585 continue;
586 }
587 }
588 }
589
590 /* Instruments the gimple pointed to by GSI. Return
591 true if func entry/exit should be instrumented. */
592
593 static bool
instrument_gimple(gimple_stmt_iterator * gsi)594 instrument_gimple (gimple_stmt_iterator *gsi)
595 {
596 gimple stmt;
597 tree rhs, lhs;
598 bool instrumented = false;
599
600 stmt = gsi_stmt (*gsi);
601 if (is_gimple_call (stmt)
602 && (gimple_call_fndecl (stmt)
603 != builtin_decl_implicit (BUILT_IN_TSAN_INIT)))
604 {
605 if (is_gimple_builtin_call (stmt))
606 instrument_builtin_call (gsi);
607 return true;
608 }
609 else if (is_gimple_assign (stmt)
610 && !gimple_clobber_p (stmt))
611 {
612 if (gimple_store_p (stmt))
613 {
614 lhs = gimple_assign_lhs (stmt);
615 instrumented = instrument_expr (*gsi, lhs, true);
616 }
617 if (gimple_assign_load_p (stmt))
618 {
619 rhs = gimple_assign_rhs1 (stmt);
620 instrumented = instrument_expr (*gsi, rhs, false);
621 }
622 }
623 return instrumented;
624 }
625
626 /* Instruments all interesting memory accesses in the current function.
627 Return true if func entry/exit should be instrumented. */
628
629 static bool
instrument_memory_accesses(void)630 instrument_memory_accesses (void)
631 {
632 basic_block bb;
633 gimple_stmt_iterator gsi;
634 bool fentry_exit_instrument = false;
635
636 FOR_EACH_BB (bb)
637 for (gsi = gsi_start_bb (bb); !gsi_end_p (gsi); gsi_next (&gsi))
638 fentry_exit_instrument |= instrument_gimple (&gsi);
639 return fentry_exit_instrument;
640 }
641
642 /* Instruments function entry. */
643
644 static void
instrument_func_entry(void)645 instrument_func_entry (void)
646 {
647 tree ret_addr, builtin_decl;
648 gimple g;
649 gimple_seq seq = NULL;
650
651 builtin_decl = builtin_decl_implicit (BUILT_IN_RETURN_ADDRESS);
652 g = gimple_build_call (builtin_decl, 1, integer_zero_node);
653 ret_addr = make_ssa_name (ptr_type_node, NULL);
654 gimple_call_set_lhs (g, ret_addr);
655 gimple_set_location (g, cfun->function_start_locus);
656 gimple_seq_add_stmt_without_update (&seq, g);
657
658 builtin_decl = builtin_decl_implicit (BUILT_IN_TSAN_FUNC_ENTRY);
659 g = gimple_build_call (builtin_decl, 1, ret_addr);
660 gimple_set_location (g, cfun->function_start_locus);
661 gimple_seq_add_stmt_without_update (&seq, g);
662
663 edge e = single_succ_edge (ENTRY_BLOCK_PTR);
664 gsi_insert_seq_on_edge_immediate (e, seq);
665 }
666
667 /* Instruments function exits. */
668
669 static void
instrument_func_exit(void)670 instrument_func_exit (void)
671 {
672 location_t loc;
673 basic_block exit_bb;
674 gimple_stmt_iterator gsi;
675 gimple stmt, g;
676 tree builtin_decl;
677 edge e;
678 edge_iterator ei;
679
680 /* Find all function exits. */
681 exit_bb = EXIT_BLOCK_PTR;
682 FOR_EACH_EDGE (e, ei, exit_bb->preds)
683 {
684 gsi = gsi_last_bb (e->src);
685 stmt = gsi_stmt (gsi);
686 gcc_assert (gimple_code (stmt) == GIMPLE_RETURN
687 || gimple_call_builtin_p (stmt, BUILT_IN_RETURN));
688 loc = gimple_location (stmt);
689 builtin_decl = builtin_decl_implicit (BUILT_IN_TSAN_FUNC_EXIT);
690 g = gimple_build_call (builtin_decl, 0);
691 gimple_set_location (g, loc);
692 gsi_insert_before (&gsi, g, GSI_SAME_STMT);
693 }
694 }
695
696 /* ThreadSanitizer instrumentation pass. */
697
698 static unsigned
tsan_pass(void)699 tsan_pass (void)
700 {
701 initialize_sanitizer_builtins ();
702 if (instrument_memory_accesses ())
703 {
704 instrument_func_entry ();
705 instrument_func_exit ();
706 }
707 return 0;
708 }
709
710 /* The pass's gate. */
711
712 static bool
tsan_gate(void)713 tsan_gate (void)
714 {
715 return flag_tsan != 0;
716 }
717
718 /* Inserts __tsan_init () into the list of CTORs. */
719
720 void
tsan_finish_file(void)721 tsan_finish_file (void)
722 {
723 tree ctor_statements = NULL_TREE;
724
725 initialize_sanitizer_builtins ();
726 tree init_decl = builtin_decl_implicit (BUILT_IN_TSAN_INIT);
727 append_to_statement_list (build_call_expr (init_decl, 0),
728 &ctor_statements);
729 cgraph_build_static_cdtor ('I', ctor_statements,
730 MAX_RESERVED_INIT_PRIORITY - 1);
731 }
732
733 /* The pass descriptor. */
734
735 struct gimple_opt_pass pass_tsan =
736 {
737 {
738 GIMPLE_PASS,
739 "tsan", /* name */
740 OPTGROUP_NONE, /* optinfo_flags */
741 tsan_gate, /* gate */
742 tsan_pass, /* execute */
743 NULL, /* sub */
744 NULL, /* next */
745 0, /* static_pass_number */
746 TV_NONE, /* tv_id */
747 PROP_ssa | PROP_cfg, /* properties_required */
748 0, /* properties_provided */
749 0, /* properties_destroyed */
750 0, /* todo_flags_start */
751 TODO_verify_all | TODO_update_ssa /* todo_flags_finish */
752 }
753 };
754
755 static bool
tsan_gate_O0(void)756 tsan_gate_O0 (void)
757 {
758 return flag_tsan != 0 && !optimize;
759 }
760
761 struct gimple_opt_pass pass_tsan_O0 =
762 {
763 {
764 GIMPLE_PASS,
765 "tsan0", /* name */
766 OPTGROUP_NONE, /* optinfo_flags */
767 tsan_gate_O0, /* gate */
768 tsan_pass, /* execute */
769 NULL, /* sub */
770 NULL, /* next */
771 0, /* static_pass_number */
772 TV_NONE, /* tv_id */
773 PROP_ssa | PROP_cfg, /* properties_required */
774 0, /* properties_provided */
775 0, /* properties_destroyed */
776 0, /* todo_flags_start */
777 TODO_verify_all | TODO_update_ssa /* todo_flags_finish */
778 }
779 };
780