gcc/ada/
[official-gcc.git] / gcc / tree-chkp.c
blob4c3317af3e0e71866c96158b6a3b7bb10db17c0d
1 /* Pointer Bounds Checker insrumentation pass.
2 Copyright (C) 2014-2015 Free Software Foundation, Inc.
3 Contributed by Ilya Enkovich (ilya.enkovich@intel.com)
5 This file is part of GCC.
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.
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.
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/>. */
21 #include "config.h"
22 #include "system.h"
23 #include "coretypes.h"
24 #include "alias.h"
25 #include "symtab.h"
26 #include "options.h"
27 #include "tree.h"
28 #include "fold-const.h"
29 #include "stor-layout.h"
30 #include "varasm.h"
31 #include "target.h"
32 #include "tree-iterator.h"
33 #include "tree-cfg.h"
34 #include "langhooks.h"
35 #include "tree-pass.h"
36 #include "diagnostic.h"
37 #include "cfgloop.h"
38 #include "stringpool.h"
39 #include "tree-ssa-alias.h"
40 #include "tree-ssanames.h"
41 #include "tree-ssa-operands.h"
42 #include "tree-ssa-address.h"
43 #include "tree-ssa.h"
44 #include "predict.h"
45 #include "dominance.h"
46 #include "cfg.h"
47 #include "basic-block.h"
48 #include "tree-ssa-loop-niter.h"
49 #include "gimple-expr.h"
50 #include "gimple.h"
51 #include "tree-phinodes.h"
52 #include "gimple-ssa.h"
53 #include "ssa-iterators.h"
54 #include "gimple-pretty-print.h"
55 #include "gimple-iterator.h"
56 #include "gimplify.h"
57 #include "gimplify-me.h"
58 #include "print-tree.h"
59 #include "tm.h"
60 #include "hard-reg-set.h"
61 #include "function.h"
62 #include "rtl.h"
63 #include "flags.h"
64 #include "insn-config.h"
65 #include "expmed.h"
66 #include "dojump.h"
67 #include "explow.h"
68 #include "calls.h"
69 #include "emit-rtl.h"
70 #include "stmt.h"
71 #include "expr.h"
72 #include "tree-ssa-propagate.h"
73 #include "gimple-fold.h"
74 #include "tree-chkp.h"
75 #include "gimple-walk.h"
76 #include "rtl.h" /* For MEM_P, assign_temp. */
77 #include "tree-dfa.h"
78 #include "ipa-ref.h"
79 #include "lto-streamer.h"
80 #include "cgraph.h"
81 #include "ipa-chkp.h"
82 #include "params.h"
84 /* Pointer Bounds Checker instruments code with memory checks to find
85 out-of-bounds memory accesses. Checks are performed by computing
86 bounds for each pointer and then comparing address of accessed
87 memory before pointer dereferencing.
89 1. Function clones.
91 See ipa-chkp.c.
93 2. Instrumentation.
95 There are few things to instrument:
97 a) Memory accesses - add checker calls to check address of accessed memory
98 against bounds of dereferenced pointer. Obviously safe memory
99 accesses like static variable access does not have to be instrumented
100 with checks.
102 Example:
104 val_2 = *p_1;
106 with 4 bytes access is transformed into:
108 __builtin___chkp_bndcl (__bound_tmp.1_3, p_1);
109 D.1_4 = p_1 + 3;
110 __builtin___chkp_bndcu (__bound_tmp.1_3, D.1_4);
111 val_2 = *p_1;
113 where __bound_tmp.1_3 are bounds computed for pointer p_1,
114 __builtin___chkp_bndcl is a lower bound check and
115 __builtin___chkp_bndcu is an upper bound check.
117 b) Pointer stores.
119 When pointer is stored in memory we need to store its bounds. To
120 achieve compatibility of instrumented code with regular codes
121 we have to keep data layout and store bounds in special bound tables
122 via special checker call. Implementation of bounds table may vary for
123 different platforms. It has to associate pointer value and its
124 location (it is required because we may have two equal pointers
125 with different bounds stored in different places) with bounds.
126 Another checker builtin allows to get bounds for specified pointer
127 loaded from specified location.
129 Example:
131 buf1[i_1] = &buf2;
133 is transformed into:
135 buf1[i_1] = &buf2;
136 D.1_2 = &buf1[i_1];
137 __builtin___chkp_bndstx (D.1_2, &buf2, __bound_tmp.1_2);
139 where __bound_tmp.1_2 are bounds of &buf2.
141 c) Static initialization.
143 The special case of pointer store is static pointer initialization.
144 Bounds initialization is performed in a few steps:
145 - register all static initializations in front-end using
146 chkp_register_var_initializer
147 - when file compilation finishes we create functions with special
148 attribute 'chkp ctor' and put explicit initialization code
149 (assignments) for all statically initialized pointers.
150 - when checker constructor is compiled checker pass adds required
151 bounds initialization for all statically initialized pointers
152 - since we do not actually need excess pointers initialization
153 in checker constructor we remove such assignments from them
155 d) Calls.
157 For each call in the code we add additional arguments to pass
158 bounds for pointer arguments. We determine type of call arguments
159 using arguments list from function declaration; if function
160 declaration is not available we use function type; otherwise
161 (e.g. for unnamed arguments) we use type of passed value. Function
162 declaration/type is replaced with the instrumented one.
164 Example:
166 val_1 = foo (&buf1, &buf2, &buf1, 0);
168 is translated into:
170 val_1 = foo.chkp (&buf1, __bound_tmp.1_2, &buf2, __bound_tmp.1_3,
171 &buf1, __bound_tmp.1_2, 0);
173 e) Returns.
175 If function returns a pointer value we have to return bounds also.
176 A new operand was added for return statement to hold returned bounds.
178 Example:
180 return &_buf1;
182 is transformed into
184 return &_buf1, __bound_tmp.1_1;
186 3. Bounds computation.
188 Compiler is fully responsible for computing bounds to be used for each
189 memory access. The first step for bounds computation is to find the
190 origin of pointer dereferenced for memory access. Basing on pointer
191 origin we define a way to compute its bounds. There are just few
192 possible cases:
194 a) Pointer is returned by call.
196 In this case we use corresponding checker builtin method to obtain returned
197 bounds.
199 Example:
201 buf_1 = malloc (size_2);
202 foo (buf_1);
204 is translated into:
206 buf_1 = malloc (size_2);
207 __bound_tmp.1_3 = __builtin___chkp_bndret (buf_1);
208 foo (buf_1, __bound_tmp.1_3);
210 b) Pointer is an address of an object.
212 In this case compiler tries to compute objects size and create corresponding
213 bounds. If object has incomplete type then special checker builtin is used to
214 obtain its size at runtime.
216 Example:
218 foo ()
220 <unnamed type> __bound_tmp.3;
221 static int buf[100];
223 <bb 3>:
224 __bound_tmp.3_2 = __builtin___chkp_bndmk (&buf, 400);
226 <bb 2>:
227 return &buf, __bound_tmp.3_2;
230 Example:
232 Address of an object 'extern int buf[]' with incomplete type is
233 returned.
235 foo ()
237 <unnamed type> __bound_tmp.4;
238 long unsigned int __size_tmp.3;
240 <bb 3>:
241 __size_tmp.3_4 = __builtin_ia32_sizeof (buf);
242 __bound_tmp.4_3 = __builtin_ia32_bndmk (&buf, __size_tmp.3_4);
244 <bb 2>:
245 return &buf, __bound_tmp.4_3;
248 c) Pointer is the result of object narrowing.
250 It happens when we use pointer to an object to compute pointer to a part
251 of an object. E.g. we take pointer to a field of a structure. In this
252 case we perform bounds intersection using bounds of original object and
253 bounds of object's part (which are computed basing on its type).
255 There may be some debatable questions about when narrowing should occur
256 and when it should not. To avoid false bound violations in correct
257 programs we do not perform narrowing when address of an array element is
258 obtained (it has address of the whole array) and when address of the first
259 structure field is obtained (because it is guaranteed to be equal to
260 address of the whole structure and it is legal to cast it back to structure).
262 Default narrowing behavior may be changed using compiler flags.
264 Example:
266 In this example address of the second structure field is returned.
268 foo (struct A * p, __bounds_type __bounds_of_p)
270 <unnamed type> __bound_tmp.3;
271 int * _2;
272 int * _5;
274 <bb 2>:
275 _5 = &p_1(D)->second_field;
276 __bound_tmp.3_6 = __builtin___chkp_bndmk (_5, 4);
277 __bound_tmp.3_8 = __builtin___chkp_intersect (__bound_tmp.3_6,
278 __bounds_of_p_3(D));
279 _2 = &p_1(D)->second_field;
280 return _2, __bound_tmp.3_8;
283 Example:
285 In this example address of the first field of array element is returned.
287 foo (struct A * p, __bounds_type __bounds_of_p, int i)
289 long unsigned int _3;
290 long unsigned int _4;
291 struct A * _6;
292 int * _7;
294 <bb 2>:
295 _3 = (long unsigned int) i_1(D);
296 _4 = _3 * 8;
297 _6 = p_5(D) + _4;
298 _7 = &_6->first_field;
299 return _7, __bounds_of_p_2(D);
303 d) Pointer is the result of pointer arithmetic or type cast.
305 In this case bounds of the base pointer are used. In case of binary
306 operation producing a pointer we are analyzing data flow further
307 looking for operand's bounds. One operand is considered as a base
308 if it has some valid bounds. If we fall into a case when none of
309 operands (or both of them) has valid bounds, a default bounds value
310 is used.
312 Trying to find out bounds for binary operations we may fall into
313 cyclic dependencies for pointers. To avoid infinite recursion all
314 walked phi nodes instantly obtain corresponding bounds but created
315 bounds are marked as incomplete. It helps us to stop DF walk during
316 bounds search.
318 When we reach pointer source, some args of incomplete bounds phi obtain
319 valid bounds and those values are propagated further through phi nodes.
320 If no valid bounds were found for phi node then we mark its result as
321 invalid bounds. Process stops when all incomplete bounds become either
322 valid or invalid and we are able to choose a pointer base.
324 e) Pointer is loaded from the memory.
326 In this case we just need to load bounds from the bounds table.
328 Example:
330 foo ()
332 <unnamed type> __bound_tmp.3;
333 static int * buf;
334 int * _2;
336 <bb 2>:
337 _2 = buf;
338 __bound_tmp.3_4 = __builtin___chkp_bndldx (&buf, _2);
339 return _2, __bound_tmp.3_4;
344 typedef void (*assign_handler)(tree, tree, void *);
346 static tree chkp_get_zero_bounds ();
347 static tree chkp_find_bounds (tree ptr, gimple_stmt_iterator *iter);
348 static tree chkp_find_bounds_loaded (tree ptr, tree ptr_src,
349 gimple_stmt_iterator *iter);
350 static void chkp_parse_array_and_component_ref (tree node, tree *ptr,
351 tree *elt, bool *safe,
352 bool *bitfield,
353 tree *bounds,
354 gimple_stmt_iterator *iter,
355 bool innermost_bounds);
357 #define chkp_bndldx_fndecl \
358 (targetm.builtin_chkp_function (BUILT_IN_CHKP_BNDLDX))
359 #define chkp_bndstx_fndecl \
360 (targetm.builtin_chkp_function (BUILT_IN_CHKP_BNDSTX))
361 #define chkp_checkl_fndecl \
362 (targetm.builtin_chkp_function (BUILT_IN_CHKP_BNDCL))
363 #define chkp_checku_fndecl \
364 (targetm.builtin_chkp_function (BUILT_IN_CHKP_BNDCU))
365 #define chkp_bndmk_fndecl \
366 (targetm.builtin_chkp_function (BUILT_IN_CHKP_BNDMK))
367 #define chkp_ret_bnd_fndecl \
368 (targetm.builtin_chkp_function (BUILT_IN_CHKP_BNDRET))
369 #define chkp_intersect_fndecl \
370 (targetm.builtin_chkp_function (BUILT_IN_CHKP_INTERSECT))
371 #define chkp_narrow_bounds_fndecl \
372 (targetm.builtin_chkp_function (BUILT_IN_CHKP_NARROW))
373 #define chkp_sizeof_fndecl \
374 (targetm.builtin_chkp_function (BUILT_IN_CHKP_SIZEOF))
375 #define chkp_extract_lower_fndecl \
376 (targetm.builtin_chkp_function (BUILT_IN_CHKP_EXTRACT_LOWER))
377 #define chkp_extract_upper_fndecl \
378 (targetm.builtin_chkp_function (BUILT_IN_CHKP_EXTRACT_UPPER))
380 static GTY (()) tree chkp_uintptr_type;
382 static GTY (()) tree chkp_zero_bounds_var;
383 static GTY (()) tree chkp_none_bounds_var;
385 static GTY (()) basic_block entry_block;
386 static GTY (()) tree zero_bounds;
387 static GTY (()) tree none_bounds;
388 static GTY (()) tree incomplete_bounds;
389 static GTY (()) tree tmp_var;
390 static GTY (()) tree size_tmp_var;
391 static GTY (()) bitmap chkp_abnormal_copies;
393 struct hash_set<tree> *chkp_invalid_bounds;
394 struct hash_set<tree> *chkp_completed_bounds_set;
395 struct hash_map<tree, tree> *chkp_reg_bounds;
396 struct hash_map<tree, tree> *chkp_bound_vars;
397 struct hash_map<tree, tree> *chkp_reg_addr_bounds;
398 struct hash_map<tree, tree> *chkp_incomplete_bounds_map;
399 struct hash_map<tree, tree> *chkp_bounds_map;
400 struct hash_map<tree, tree> *chkp_static_var_bounds;
402 static bool in_chkp_pass;
404 #define CHKP_BOUND_TMP_NAME "__bound_tmp"
405 #define CHKP_SIZE_TMP_NAME "__size_tmp"
406 #define CHKP_BOUNDS_OF_SYMBOL_PREFIX "__chkp_bounds_of_"
407 #define CHKP_STRING_BOUNDS_PREFIX "__chkp_string_bounds_"
408 #define CHKP_VAR_BOUNDS_PREFIX "__chkp_var_bounds_"
409 #define CHKP_ZERO_BOUNDS_VAR_NAME "__chkp_zero_bounds"
410 #define CHKP_NONE_BOUNDS_VAR_NAME "__chkp_none_bounds"
412 /* Static checker constructors may become very large and their
413 compilation with optimization may take too much time.
414 Therefore we put a limit to number of statements in one
415 constructor. Tests with 100 000 statically initialized
416 pointers showed following compilation times on Sandy Bridge
417 server (used -O2):
418 limit 100 => ~18 sec.
419 limit 300 => ~22 sec.
420 limit 1000 => ~30 sec.
421 limit 3000 => ~49 sec.
422 limit 5000 => ~55 sec.
423 limit 10000 => ~76 sec.
424 limit 100000 => ~532 sec. */
425 #define MAX_STMTS_IN_STATIC_CHKP_CTOR (PARAM_VALUE (PARAM_CHKP_MAX_CTOR_SIZE))
427 struct chkp_ctor_stmt_list
429 tree stmts;
430 int avail;
433 /* Return 1 if function FNDECL is instrumented by Pointer
434 Bounds Checker. */
435 bool
436 chkp_function_instrumented_p (tree fndecl)
438 return fndecl
439 && lookup_attribute ("chkp instrumented", DECL_ATTRIBUTES (fndecl));
442 /* Mark function FNDECL as instrumented. */
443 void
444 chkp_function_mark_instrumented (tree fndecl)
446 if (chkp_function_instrumented_p (fndecl))
447 return;
449 DECL_ATTRIBUTES (fndecl)
450 = tree_cons (get_identifier ("chkp instrumented"), NULL,
451 DECL_ATTRIBUTES (fndecl));
454 /* Return true when STMT is builtin call to instrumentation function
455 corresponding to CODE. */
457 bool
458 chkp_gimple_call_builtin_p (gimple call,
459 enum built_in_function code)
461 tree fndecl;
462 if (is_gimple_call (call)
463 && (fndecl = targetm.builtin_chkp_function (code))
464 && gimple_call_fndecl (call) == fndecl)
465 return true;
466 return false;
469 /* Emit code to build zero bounds and return RTL holding
470 the result. */
472 chkp_expand_zero_bounds ()
474 tree zero_bnd;
476 if (flag_chkp_use_static_const_bounds)
477 zero_bnd = chkp_get_zero_bounds_var ();
478 else
479 zero_bnd = chkp_build_make_bounds_call (integer_zero_node,
480 integer_zero_node);
481 return expand_normal (zero_bnd);
484 /* Emit code to store zero bounds for PTR located at MEM. */
485 void
486 chkp_expand_bounds_reset_for_mem (tree mem, tree ptr)
488 tree zero_bnd, bnd, addr, bndstx;
490 if (flag_chkp_use_static_const_bounds)
491 zero_bnd = chkp_get_zero_bounds_var ();
492 else
493 zero_bnd = chkp_build_make_bounds_call (integer_zero_node,
494 integer_zero_node);
495 bnd = make_tree (pointer_bounds_type_node,
496 assign_temp (pointer_bounds_type_node, 0, 1));
497 addr = build1 (ADDR_EXPR,
498 build_pointer_type (TREE_TYPE (mem)), mem);
499 bndstx = chkp_build_bndstx_call (addr, ptr, bnd);
501 expand_assignment (bnd, zero_bnd, false);
502 expand_normal (bndstx);
505 /* Build retbnd call for returned value RETVAL.
507 If BNDVAL is not NULL then result is stored
508 in it. Otherwise a temporary is created to
509 hold returned value.
511 GSI points to a position for a retbnd call
512 and is set to created stmt.
514 Cgraph edge is created for a new call if
515 UPDATE_EDGE is 1.
517 Obtained bounds are returned. */
518 tree
519 chkp_insert_retbnd_call (tree bndval, tree retval,
520 gimple_stmt_iterator *gsi)
522 gimple call;
524 if (!bndval)
525 bndval = create_tmp_reg (pointer_bounds_type_node, "retbnd");
527 call = gimple_build_call (chkp_ret_bnd_fndecl, 1, retval);
528 gimple_call_set_lhs (call, bndval);
529 gsi_insert_after (gsi, call, GSI_CONTINUE_LINKING);
531 return bndval;
534 /* Build a GIMPLE_CALL identical to CALL but skipping bounds
535 arguments. */
537 gcall *
538 chkp_copy_call_skip_bounds (gcall *call)
540 bitmap bounds;
541 unsigned i;
543 bitmap_obstack_initialize (NULL);
544 bounds = BITMAP_ALLOC (NULL);
546 for (i = 0; i < gimple_call_num_args (call); i++)
547 if (POINTER_BOUNDS_P (gimple_call_arg (call, i)))
548 bitmap_set_bit (bounds, i);
550 if (!bitmap_empty_p (bounds))
551 call = gimple_call_copy_skip_args (call, bounds);
552 gimple_call_set_with_bounds (call, false);
554 BITMAP_FREE (bounds);
555 bitmap_obstack_release (NULL);
557 return call;
560 /* Redirect edge E to the correct node according to call_stmt.
561 Return 1 if bounds removal from call_stmt should be done
562 instead of redirection. */
564 bool
565 chkp_redirect_edge (cgraph_edge *e)
567 bool instrumented = false;
568 tree decl = e->callee->decl;
570 if (e->callee->instrumentation_clone
571 || chkp_function_instrumented_p (decl))
572 instrumented = true;
574 if (instrumented
575 && !gimple_call_with_bounds_p (e->call_stmt))
576 e->redirect_callee (cgraph_node::get_create (e->callee->orig_decl));
577 else if (!instrumented
578 && gimple_call_with_bounds_p (e->call_stmt)
579 && !chkp_gimple_call_builtin_p (e->call_stmt, BUILT_IN_CHKP_BNDCL)
580 && !chkp_gimple_call_builtin_p (e->call_stmt, BUILT_IN_CHKP_BNDCU)
581 && !chkp_gimple_call_builtin_p (e->call_stmt, BUILT_IN_CHKP_BNDSTX))
583 if (e->callee->instrumented_version)
584 e->redirect_callee (e->callee->instrumented_version);
585 else
587 tree args = TYPE_ARG_TYPES (TREE_TYPE (decl));
588 /* Avoid bounds removal if all args will be removed. */
589 if (!args || TREE_VALUE (args) != void_type_node)
590 return true;
591 else
592 gimple_call_set_with_bounds (e->call_stmt, false);
596 return false;
599 /* Mark statement S to not be instrumented. */
600 static void
601 chkp_mark_stmt (gimple s)
603 gimple_set_plf (s, GF_PLF_1, true);
606 /* Mark statement S to be instrumented. */
607 static void
608 chkp_unmark_stmt (gimple s)
610 gimple_set_plf (s, GF_PLF_1, false);
613 /* Return 1 if statement S should not be instrumented. */
614 static bool
615 chkp_marked_stmt_p (gimple s)
617 return gimple_plf (s, GF_PLF_1);
620 /* Get var to be used for bound temps. */
621 static tree
622 chkp_get_tmp_var (void)
624 if (!tmp_var)
625 tmp_var = create_tmp_reg (pointer_bounds_type_node, CHKP_BOUND_TMP_NAME);
627 return tmp_var;
630 /* Get SSA_NAME to be used as temp. */
631 static tree
632 chkp_get_tmp_reg (gimple stmt)
634 if (in_chkp_pass)
635 return make_ssa_name (chkp_get_tmp_var (), stmt);
637 return make_temp_ssa_name (pointer_bounds_type_node, stmt,
638 CHKP_BOUND_TMP_NAME);
641 /* Get var to be used for size temps. */
642 static tree
643 chkp_get_size_tmp_var (void)
645 if (!size_tmp_var)
646 size_tmp_var = create_tmp_reg (chkp_uintptr_type, CHKP_SIZE_TMP_NAME);
648 return size_tmp_var;
651 /* Register bounds BND for address of OBJ. */
652 static void
653 chkp_register_addr_bounds (tree obj, tree bnd)
655 if (bnd == incomplete_bounds)
656 return;
658 chkp_reg_addr_bounds->put (obj, bnd);
660 if (dump_file && (dump_flags & TDF_DETAILS))
662 fprintf (dump_file, "Regsitered bound ");
663 print_generic_expr (dump_file, bnd, 0);
664 fprintf (dump_file, " for address of ");
665 print_generic_expr (dump_file, obj, 0);
666 fprintf (dump_file, "\n");
670 /* Return bounds registered for address of OBJ. */
671 static tree
672 chkp_get_registered_addr_bounds (tree obj)
674 tree *slot = chkp_reg_addr_bounds->get (obj);
675 return slot ? *slot : NULL_TREE;
678 /* Mark BOUNDS as completed. */
679 static void
680 chkp_mark_completed_bounds (tree bounds)
682 chkp_completed_bounds_set->add (bounds);
684 if (dump_file && (dump_flags & TDF_DETAILS))
686 fprintf (dump_file, "Marked bounds ");
687 print_generic_expr (dump_file, bounds, 0);
688 fprintf (dump_file, " as completed\n");
692 /* Return 1 if BOUNDS were marked as completed and 0 otherwise. */
693 static bool
694 chkp_completed_bounds (tree bounds)
696 return chkp_completed_bounds_set->contains (bounds);
699 /* Clear comleted bound marks. */
700 static void
701 chkp_erase_completed_bounds (void)
703 delete chkp_completed_bounds_set;
704 chkp_completed_bounds_set = new hash_set<tree>;
707 /* Mark BOUNDS associated with PTR as incomplete. */
708 static void
709 chkp_register_incomplete_bounds (tree bounds, tree ptr)
711 chkp_incomplete_bounds_map->put (bounds, ptr);
713 if (dump_file && (dump_flags & TDF_DETAILS))
715 fprintf (dump_file, "Regsitered incomplete bounds ");
716 print_generic_expr (dump_file, bounds, 0);
717 fprintf (dump_file, " for ");
718 print_generic_expr (dump_file, ptr, 0);
719 fprintf (dump_file, "\n");
723 /* Return 1 if BOUNDS are incomplete and 0 otherwise. */
724 static bool
725 chkp_incomplete_bounds (tree bounds)
727 if (bounds == incomplete_bounds)
728 return true;
730 if (chkp_completed_bounds (bounds))
731 return false;
733 return chkp_incomplete_bounds_map->get (bounds) != NULL;
736 /* Clear incomleted bound marks. */
737 static void
738 chkp_erase_incomplete_bounds (void)
740 delete chkp_incomplete_bounds_map;
741 chkp_incomplete_bounds_map = new hash_map<tree, tree>;
744 /* Build and return bndmk call which creates bounds for structure
745 pointed by PTR. Structure should have complete type. */
746 tree
747 chkp_make_bounds_for_struct_addr (tree ptr)
749 tree type = TREE_TYPE (ptr);
750 tree size;
752 gcc_assert (POINTER_TYPE_P (type));
754 size = TYPE_SIZE (TREE_TYPE (type));
756 gcc_assert (size);
758 return build_call_nary (pointer_bounds_type_node,
759 build_fold_addr_expr (chkp_bndmk_fndecl),
760 2, ptr, size);
763 /* Traversal function for chkp_may_finish_incomplete_bounds.
764 Set RES to 0 if at least one argument of phi statement
765 defining bounds (passed in KEY arg) is unknown.
766 Traversal stops when first unknown phi argument is found. */
767 bool
768 chkp_may_complete_phi_bounds (tree const &bounds, tree *slot ATTRIBUTE_UNUSED,
769 bool *res)
771 gimple phi;
772 unsigned i;
774 gcc_assert (TREE_CODE (bounds) == SSA_NAME);
776 phi = SSA_NAME_DEF_STMT (bounds);
778 gcc_assert (phi && gimple_code (phi) == GIMPLE_PHI);
780 for (i = 0; i < gimple_phi_num_args (phi); i++)
782 tree phi_arg = gimple_phi_arg_def (phi, i);
783 if (!phi_arg)
785 *res = false;
786 /* Do not need to traverse further. */
787 return false;
791 return true;
794 /* Return 1 if all phi nodes created for bounds have their
795 arguments computed. */
796 static bool
797 chkp_may_finish_incomplete_bounds (void)
799 bool res = true;
801 chkp_incomplete_bounds_map
802 ->traverse<bool *, chkp_may_complete_phi_bounds> (&res);
804 return res;
807 /* Helper function for chkp_finish_incomplete_bounds.
808 Recompute args for bounds phi node. */
809 bool
810 chkp_recompute_phi_bounds (tree const &bounds, tree *slot,
811 void *res ATTRIBUTE_UNUSED)
813 tree ptr = *slot;
814 gphi *bounds_phi;
815 gphi *ptr_phi;
816 unsigned i;
818 gcc_assert (TREE_CODE (bounds) == SSA_NAME);
819 gcc_assert (TREE_CODE (ptr) == SSA_NAME);
821 bounds_phi = as_a <gphi *> (SSA_NAME_DEF_STMT (bounds));
822 ptr_phi = as_a <gphi *> (SSA_NAME_DEF_STMT (ptr));
824 for (i = 0; i < gimple_phi_num_args (bounds_phi); i++)
826 tree ptr_arg = gimple_phi_arg_def (ptr_phi, i);
827 tree bound_arg = chkp_find_bounds (ptr_arg, NULL);
829 add_phi_arg (bounds_phi, bound_arg,
830 gimple_phi_arg_edge (ptr_phi, i),
831 UNKNOWN_LOCATION);
834 return true;
837 /* Mark BOUNDS as invalid. */
838 static void
839 chkp_mark_invalid_bounds (tree bounds)
841 chkp_invalid_bounds->add (bounds);
843 if (dump_file && (dump_flags & TDF_DETAILS))
845 fprintf (dump_file, "Marked bounds ");
846 print_generic_expr (dump_file, bounds, 0);
847 fprintf (dump_file, " as invalid\n");
851 /* Return 1 if BOUNDS were marked as invalid and 0 otherwise. */
852 static bool
853 chkp_valid_bounds (tree bounds)
855 if (bounds == zero_bounds || bounds == none_bounds)
856 return false;
858 return !chkp_invalid_bounds->contains (bounds);
861 /* Helper function for chkp_finish_incomplete_bounds.
862 Check all arguments of phi nodes trying to find
863 valid completed bounds. If there is at least one
864 such arg then bounds produced by phi node are marked
865 as valid completed bounds and all phi args are
866 recomputed. */
867 bool
868 chkp_find_valid_phi_bounds (tree const &bounds, tree *slot, bool *res)
870 gimple phi;
871 unsigned i;
873 gcc_assert (TREE_CODE (bounds) == SSA_NAME);
875 if (chkp_completed_bounds (bounds))
876 return true;
878 phi = SSA_NAME_DEF_STMT (bounds);
880 gcc_assert (phi && gimple_code (phi) == GIMPLE_PHI);
882 for (i = 0; i < gimple_phi_num_args (phi); i++)
884 tree phi_arg = gimple_phi_arg_def (phi, i);
886 gcc_assert (phi_arg);
888 if (chkp_valid_bounds (phi_arg) && !chkp_incomplete_bounds (phi_arg))
890 *res = true;
891 chkp_mark_completed_bounds (bounds);
892 chkp_recompute_phi_bounds (bounds, slot, NULL);
893 return true;
897 return true;
900 /* Helper function for chkp_finish_incomplete_bounds.
901 Marks all incompleted bounds as invalid. */
902 bool
903 chkp_mark_invalid_bounds_walker (tree const &bounds,
904 tree *slot ATTRIBUTE_UNUSED,
905 void *res ATTRIBUTE_UNUSED)
907 if (!chkp_completed_bounds (bounds))
909 chkp_mark_invalid_bounds (bounds);
910 chkp_mark_completed_bounds (bounds);
912 return true;
915 /* When all bound phi nodes have all their args computed
916 we have enough info to find valid bounds. We iterate
917 through all incompleted bounds searching for valid
918 bounds. Found valid bounds are marked as completed
919 and all remaining incompleted bounds are recomputed.
920 Process continues until no new valid bounds may be
921 found. All remained incompleted bounds are marked as
922 invalid (i.e. have no valid source of bounds). */
923 static void
924 chkp_finish_incomplete_bounds (void)
926 bool found_valid;
928 while (found_valid)
930 found_valid = false;
932 chkp_incomplete_bounds_map->
933 traverse<bool *, chkp_find_valid_phi_bounds> (&found_valid);
935 if (found_valid)
936 chkp_incomplete_bounds_map->
937 traverse<void *, chkp_recompute_phi_bounds> (NULL);
940 chkp_incomplete_bounds_map->
941 traverse<void *, chkp_mark_invalid_bounds_walker> (NULL);
942 chkp_incomplete_bounds_map->
943 traverse<void *, chkp_recompute_phi_bounds> (NULL);
945 chkp_erase_completed_bounds ();
946 chkp_erase_incomplete_bounds ();
949 /* Return 1 if type TYPE is a pointer type or a
950 structure having a pointer type as one of its fields.
951 Otherwise return 0. */
952 bool
953 chkp_type_has_pointer (const_tree type)
955 bool res = false;
957 if (BOUNDED_TYPE_P (type))
958 res = true;
959 else if (RECORD_OR_UNION_TYPE_P (type))
961 tree field;
963 for (field = TYPE_FIELDS (type); field; field = DECL_CHAIN (field))
964 if (TREE_CODE (field) == FIELD_DECL)
965 res = res || chkp_type_has_pointer (TREE_TYPE (field));
967 else if (TREE_CODE (type) == ARRAY_TYPE)
968 res = chkp_type_has_pointer (TREE_TYPE (type));
970 return res;
973 unsigned
974 chkp_type_bounds_count (const_tree type)
976 unsigned res = 0;
978 if (!type)
979 res = 0;
980 else if (BOUNDED_TYPE_P (type))
981 res = 1;
982 else if (RECORD_OR_UNION_TYPE_P (type))
984 bitmap have_bound;
986 bitmap_obstack_initialize (NULL);
987 have_bound = BITMAP_ALLOC (NULL);
988 chkp_find_bound_slots (type, have_bound);
989 res = bitmap_count_bits (have_bound);
990 BITMAP_FREE (have_bound);
991 bitmap_obstack_release (NULL);
994 return res;
997 /* Get bounds associated with NODE via
998 chkp_set_bounds call. */
999 tree
1000 chkp_get_bounds (tree node)
1002 tree *slot;
1004 if (!chkp_bounds_map)
1005 return NULL_TREE;
1007 slot = chkp_bounds_map->get (node);
1008 return slot ? *slot : NULL_TREE;
1011 /* Associate bounds VAL with NODE. */
1012 void
1013 chkp_set_bounds (tree node, tree val)
1015 if (!chkp_bounds_map)
1016 chkp_bounds_map = new hash_map<tree, tree>;
1018 chkp_bounds_map->put (node, val);
1021 /* Check if statically initialized variable VAR require
1022 static bounds initialization. If VAR is added into
1023 bounds initlization list then 1 is returned. Otherwise
1024 return 0. */
1025 extern bool
1026 chkp_register_var_initializer (tree var)
1028 if (!flag_check_pointer_bounds
1029 || DECL_INITIAL (var) == error_mark_node)
1030 return false;
1032 gcc_assert (TREE_CODE (var) == VAR_DECL);
1033 gcc_assert (DECL_INITIAL (var));
1035 if (TREE_STATIC (var)
1036 && chkp_type_has_pointer (TREE_TYPE (var)))
1038 varpool_node::get_create (var)->need_bounds_init = 1;
1039 return true;
1042 return false;
1045 /* Helper function for chkp_finish_file.
1047 Add new modification statement (RHS is assigned to LHS)
1048 into list of static initializer statementes (passed in ARG).
1049 If statements list becomes too big, emit checker constructor
1050 and start the new one. */
1051 static void
1052 chkp_add_modification_to_stmt_list (tree lhs,
1053 tree rhs,
1054 void *arg)
1056 struct chkp_ctor_stmt_list *stmts = (struct chkp_ctor_stmt_list *)arg;
1057 tree modify;
1059 if (!useless_type_conversion_p (TREE_TYPE (lhs), TREE_TYPE (rhs)))
1060 rhs = build1 (CONVERT_EXPR, TREE_TYPE (lhs), rhs);
1062 modify = build2 (MODIFY_EXPR, TREE_TYPE (lhs), lhs, rhs);
1063 append_to_statement_list (modify, &stmts->stmts);
1065 stmts->avail--;
1068 /* Build and return ADDR_EXPR for specified object OBJ. */
1069 static tree
1070 chkp_build_addr_expr (tree obj)
1072 return TREE_CODE (obj) == TARGET_MEM_REF
1073 ? tree_mem_ref_addr (ptr_type_node, obj)
1074 : build_fold_addr_expr (obj);
1077 /* Helper function for chkp_finish_file.
1078 Initialize bound variable BND_VAR with bounds of variable
1079 VAR to statements list STMTS. If statements list becomes
1080 too big, emit checker constructor and start the new one. */
1081 static void
1082 chkp_output_static_bounds (tree bnd_var, tree var,
1083 struct chkp_ctor_stmt_list *stmts)
1085 tree lb, ub, size;
1087 if (TREE_CODE (var) == STRING_CST)
1089 lb = build1 (CONVERT_EXPR, size_type_node, chkp_build_addr_expr (var));
1090 size = build_int_cst (size_type_node, TREE_STRING_LENGTH (var) - 1);
1092 else if (DECL_SIZE (var)
1093 && !chkp_variable_size_type (TREE_TYPE (var)))
1095 /* Compute bounds using statically known size. */
1096 lb = build1 (CONVERT_EXPR, size_type_node, chkp_build_addr_expr (var));
1097 size = size_binop (MINUS_EXPR, DECL_SIZE_UNIT (var), size_one_node);
1099 else
1101 /* Compute bounds using dynamic size. */
1102 tree call;
1104 lb = build1 (CONVERT_EXPR, size_type_node, chkp_build_addr_expr (var));
1105 call = build1 (ADDR_EXPR,
1106 build_pointer_type (TREE_TYPE (chkp_sizeof_fndecl)),
1107 chkp_sizeof_fndecl);
1108 size = build_call_nary (TREE_TYPE (TREE_TYPE (chkp_sizeof_fndecl)),
1109 call, 1, var);
1111 if (flag_chkp_zero_dynamic_size_as_infinite)
1113 tree max_size, cond;
1115 max_size = build2 (MINUS_EXPR, size_type_node, size_zero_node, lb);
1116 cond = build2 (NE_EXPR, boolean_type_node, size, size_zero_node);
1117 size = build3 (COND_EXPR, size_type_node, cond, size, max_size);
1120 size = size_binop (MINUS_EXPR, size, size_one_node);
1123 ub = size_binop (PLUS_EXPR, lb, size);
1124 stmts->avail -= targetm.chkp_initialize_bounds (bnd_var, lb, ub,
1125 &stmts->stmts);
1126 if (stmts->avail <= 0)
1128 cgraph_build_static_cdtor ('B', stmts->stmts,
1129 MAX_RESERVED_INIT_PRIORITY + 2);
1130 stmts->avail = MAX_STMTS_IN_STATIC_CHKP_CTOR;
1131 stmts->stmts = NULL;
1135 /* Return entry block to be used for checker initilization code.
1136 Create new block if required. */
1137 static basic_block
1138 chkp_get_entry_block (void)
1140 if (!entry_block)
1141 entry_block
1142 = split_block_after_labels (ENTRY_BLOCK_PTR_FOR_FN (cfun))->dest;
1144 return entry_block;
1147 /* Return a bounds var to be used for pointer var PTR_VAR. */
1148 static tree
1149 chkp_get_bounds_var (tree ptr_var)
1151 tree bnd_var;
1152 tree *slot;
1154 slot = chkp_bound_vars->get (ptr_var);
1155 if (slot)
1156 bnd_var = *slot;
1157 else
1159 bnd_var = create_tmp_reg (pointer_bounds_type_node,
1160 CHKP_BOUND_TMP_NAME);
1161 chkp_bound_vars->put (ptr_var, bnd_var);
1164 return bnd_var;
1167 /* If BND is an abnormal bounds copy, return a copied value.
1168 Otherwise return BND. */
1169 static tree
1170 chkp_get_orginal_bounds_for_abnormal_copy (tree bnd)
1172 if (bitmap_bit_p (chkp_abnormal_copies, SSA_NAME_VERSION (bnd)))
1174 gimple bnd_def = SSA_NAME_DEF_STMT (bnd);
1175 gcc_checking_assert (gimple_code (bnd_def) == GIMPLE_ASSIGN);
1176 bnd = gimple_assign_rhs1 (bnd_def);
1179 return bnd;
1182 /* Register bounds BND for object PTR in global bounds table.
1183 A copy of bounds may be created for abnormal ssa names.
1184 Returns bounds to use for PTR. */
1185 static tree
1186 chkp_maybe_copy_and_register_bounds (tree ptr, tree bnd)
1188 bool abnormal_ptr;
1190 if (!chkp_reg_bounds)
1191 return bnd;
1193 /* Do nothing if bounds are incomplete_bounds
1194 because it means bounds will be recomputed. */
1195 if (bnd == incomplete_bounds)
1196 return bnd;
1198 abnormal_ptr = (TREE_CODE (ptr) == SSA_NAME
1199 && SSA_NAME_OCCURS_IN_ABNORMAL_PHI (ptr)
1200 && gimple_code (SSA_NAME_DEF_STMT (ptr)) != GIMPLE_PHI);
1202 /* A single bounds value may be reused multiple times for
1203 different pointer values. It may cause coalescing issues
1204 for abnormal SSA names. To avoid it we create a bounds
1205 copy in case it is computed for abnormal SSA name.
1207 We also cannot reuse such created copies for other pointers */
1208 if (abnormal_ptr
1209 || bitmap_bit_p (chkp_abnormal_copies, SSA_NAME_VERSION (bnd)))
1211 tree bnd_var = NULL_TREE;
1213 if (abnormal_ptr)
1215 if (SSA_NAME_VAR (ptr))
1216 bnd_var = chkp_get_bounds_var (SSA_NAME_VAR (ptr));
1218 else
1219 bnd_var = chkp_get_tmp_var ();
1221 /* For abnormal copies we may just find original
1222 bounds and use them. */
1223 if (!abnormal_ptr && !SSA_NAME_IS_DEFAULT_DEF (bnd))
1224 bnd = chkp_get_orginal_bounds_for_abnormal_copy (bnd);
1225 /* For undefined values we usually use none bounds
1226 value but in case of abnormal edge it may cause
1227 coalescing failures. Use default definition of
1228 bounds variable instead to avoid it. */
1229 else if (SSA_NAME_IS_DEFAULT_DEF (ptr)
1230 && TREE_CODE (SSA_NAME_VAR (ptr)) != PARM_DECL)
1232 bnd = get_or_create_ssa_default_def (cfun, bnd_var);
1234 if (dump_file && (dump_flags & TDF_DETAILS))
1236 fprintf (dump_file, "Using default def bounds ");
1237 print_generic_expr (dump_file, bnd, 0);
1238 fprintf (dump_file, " for abnormal default def SSA name ");
1239 print_generic_expr (dump_file, ptr, 0);
1240 fprintf (dump_file, "\n");
1243 else
1245 tree copy;
1246 gimple def = SSA_NAME_DEF_STMT (ptr);
1247 gimple assign;
1248 gimple_stmt_iterator gsi;
1250 if (bnd_var)
1251 copy = make_ssa_name (bnd_var);
1252 else
1253 copy = make_temp_ssa_name (pointer_bounds_type_node,
1254 NULL,
1255 CHKP_BOUND_TMP_NAME);
1256 bnd = chkp_get_orginal_bounds_for_abnormal_copy (bnd);
1257 assign = gimple_build_assign (copy, bnd);
1259 if (dump_file && (dump_flags & TDF_DETAILS))
1261 fprintf (dump_file, "Creating a copy of bounds ");
1262 print_generic_expr (dump_file, bnd, 0);
1263 fprintf (dump_file, " for abnormal SSA name ");
1264 print_generic_expr (dump_file, ptr, 0);
1265 fprintf (dump_file, "\n");
1268 if (gimple_code (def) == GIMPLE_NOP)
1270 gsi = gsi_last_bb (chkp_get_entry_block ());
1271 if (!gsi_end_p (gsi) && is_ctrl_stmt (gsi_stmt (gsi)))
1272 gsi_insert_before (&gsi, assign, GSI_CONTINUE_LINKING);
1273 else
1274 gsi_insert_after (&gsi, assign, GSI_CONTINUE_LINKING);
1276 else
1278 gimple bnd_def = SSA_NAME_DEF_STMT (bnd);
1279 /* Sometimes (e.g. when we load a pointer from a
1280 memory) bounds are produced later than a pointer.
1281 We need to insert bounds copy appropriately. */
1282 if (gimple_code (bnd_def) != GIMPLE_NOP
1283 && stmt_dominates_stmt_p (def, bnd_def))
1284 gsi = gsi_for_stmt (bnd_def);
1285 else
1286 gsi = gsi_for_stmt (def);
1287 gsi_insert_after (&gsi, assign, GSI_CONTINUE_LINKING);
1290 bnd = copy;
1293 if (abnormal_ptr)
1294 bitmap_set_bit (chkp_abnormal_copies, SSA_NAME_VERSION (bnd));
1297 chkp_reg_bounds->put (ptr, bnd);
1299 if (dump_file && (dump_flags & TDF_DETAILS))
1301 fprintf (dump_file, "Regsitered bound ");
1302 print_generic_expr (dump_file, bnd, 0);
1303 fprintf (dump_file, " for pointer ");
1304 print_generic_expr (dump_file, ptr, 0);
1305 fprintf (dump_file, "\n");
1308 return bnd;
1311 /* Get bounds registered for object PTR in global bounds table. */
1312 static tree
1313 chkp_get_registered_bounds (tree ptr)
1315 tree *slot;
1317 if (!chkp_reg_bounds)
1318 return NULL_TREE;
1320 slot = chkp_reg_bounds->get (ptr);
1321 return slot ? *slot : NULL_TREE;
1324 /* Add bound retvals to return statement pointed by GSI. */
1326 static void
1327 chkp_add_bounds_to_ret_stmt (gimple_stmt_iterator *gsi)
1329 greturn *ret = as_a <greturn *> (gsi_stmt (*gsi));
1330 tree retval = gimple_return_retval (ret);
1331 tree ret_decl = DECL_RESULT (cfun->decl);
1332 tree bounds;
1334 if (!retval)
1335 return;
1337 if (BOUNDED_P (ret_decl))
1339 bounds = chkp_find_bounds (retval, gsi);
1340 bounds = chkp_maybe_copy_and_register_bounds (ret_decl, bounds);
1341 gimple_return_set_retbnd (ret, bounds);
1344 update_stmt (ret);
1347 /* Force OP to be suitable for using as an argument for call.
1348 New statements (if any) go to SEQ. */
1349 static tree
1350 chkp_force_gimple_call_op (tree op, gimple_seq *seq)
1352 gimple_seq stmts;
1353 gimple_stmt_iterator si;
1355 op = force_gimple_operand (unshare_expr (op), &stmts, true, NULL_TREE);
1357 for (si = gsi_start (stmts); !gsi_end_p (si); gsi_next (&si))
1358 chkp_mark_stmt (gsi_stmt (si));
1360 gimple_seq_add_seq (seq, stmts);
1362 return op;
1365 /* Generate lower bound check for memory access by ADDR.
1366 Check is inserted before the position pointed by ITER.
1367 DIRFLAG indicates whether memory access is load or store. */
1368 static void
1369 chkp_check_lower (tree addr, tree bounds,
1370 gimple_stmt_iterator iter,
1371 location_t location,
1372 tree dirflag)
1374 gimple_seq seq;
1375 gimple check;
1376 tree node;
1378 if (!chkp_function_instrumented_p (current_function_decl)
1379 && bounds == chkp_get_zero_bounds ())
1380 return;
1382 if (dirflag == integer_zero_node
1383 && !flag_chkp_check_read)
1384 return;
1386 if (dirflag == integer_one_node
1387 && !flag_chkp_check_write)
1388 return;
1390 seq = NULL;
1392 node = chkp_force_gimple_call_op (addr, &seq);
1394 check = gimple_build_call (chkp_checkl_fndecl, 2, node, bounds);
1395 chkp_mark_stmt (check);
1396 gimple_call_set_with_bounds (check, true);
1397 gimple_set_location (check, location);
1398 gimple_seq_add_stmt (&seq, check);
1400 gsi_insert_seq_before (&iter, seq, GSI_SAME_STMT);
1402 if (dump_file && (dump_flags & TDF_DETAILS))
1404 gimple before = gsi_stmt (iter);
1405 fprintf (dump_file, "Generated lower bound check for statement ");
1406 print_gimple_stmt (dump_file, before, 0, TDF_VOPS|TDF_MEMSYMS);
1407 fprintf (dump_file, " ");
1408 print_gimple_stmt (dump_file, check, 0, TDF_VOPS|TDF_MEMSYMS);
1412 /* Generate upper bound check for memory access by ADDR.
1413 Check is inserted before the position pointed by ITER.
1414 DIRFLAG indicates whether memory access is load or store. */
1415 static void
1416 chkp_check_upper (tree addr, tree bounds,
1417 gimple_stmt_iterator iter,
1418 location_t location,
1419 tree dirflag)
1421 gimple_seq seq;
1422 gimple check;
1423 tree node;
1425 if (!chkp_function_instrumented_p (current_function_decl)
1426 && bounds == chkp_get_zero_bounds ())
1427 return;
1429 if (dirflag == integer_zero_node
1430 && !flag_chkp_check_read)
1431 return;
1433 if (dirflag == integer_one_node
1434 && !flag_chkp_check_write)
1435 return;
1437 seq = NULL;
1439 node = chkp_force_gimple_call_op (addr, &seq);
1441 check = gimple_build_call (chkp_checku_fndecl, 2, node, bounds);
1442 chkp_mark_stmt (check);
1443 gimple_call_set_with_bounds (check, true);
1444 gimple_set_location (check, location);
1445 gimple_seq_add_stmt (&seq, check);
1447 gsi_insert_seq_before (&iter, seq, GSI_SAME_STMT);
1449 if (dump_file && (dump_flags & TDF_DETAILS))
1451 gimple before = gsi_stmt (iter);
1452 fprintf (dump_file, "Generated upper bound check for statement ");
1453 print_gimple_stmt (dump_file, before, 0, TDF_VOPS|TDF_MEMSYMS);
1454 fprintf (dump_file, " ");
1455 print_gimple_stmt (dump_file, check, 0, TDF_VOPS|TDF_MEMSYMS);
1459 /* Generate lower and upper bound checks for memory access
1460 to memory slot [FIRST, LAST] againsr BOUNDS. Checks
1461 are inserted before the position pointed by ITER.
1462 DIRFLAG indicates whether memory access is load or store. */
1463 void
1464 chkp_check_mem_access (tree first, tree last, tree bounds,
1465 gimple_stmt_iterator iter,
1466 location_t location,
1467 tree dirflag)
1469 chkp_check_lower (first, bounds, iter, location, dirflag);
1470 chkp_check_upper (last, bounds, iter, location, dirflag);
1473 /* Replace call to _bnd_chk_* pointed by GSI with
1474 bndcu and bndcl calls. DIRFLAG determines whether
1475 check is for read or write. */
1477 void
1478 chkp_replace_address_check_builtin (gimple_stmt_iterator *gsi,
1479 tree dirflag)
1481 gimple_stmt_iterator call_iter = *gsi;
1482 gimple call = gsi_stmt (*gsi);
1483 tree fndecl = gimple_call_fndecl (call);
1484 tree addr = gimple_call_arg (call, 0);
1485 tree bounds = chkp_find_bounds (addr, gsi);
1487 if (DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_CHECK_PTR_LBOUNDS
1488 || DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_CHECK_PTR_BOUNDS)
1489 chkp_check_lower (addr, bounds, *gsi, gimple_location (call), dirflag);
1491 if (DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_CHECK_PTR_UBOUNDS)
1492 chkp_check_upper (addr, bounds, *gsi, gimple_location (call), dirflag);
1494 if (DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_CHECK_PTR_BOUNDS)
1496 tree size = gimple_call_arg (call, 1);
1497 addr = fold_build_pointer_plus (addr, size);
1498 addr = fold_build_pointer_plus_hwi (addr, -1);
1499 chkp_check_upper (addr, bounds, *gsi, gimple_location (call), dirflag);
1502 gsi_remove (&call_iter, true);
1505 /* Replace call to _bnd_get_ptr_* pointed by GSI with
1506 corresponding bounds extract call. */
1508 void
1509 chkp_replace_extract_builtin (gimple_stmt_iterator *gsi)
1511 gimple call = gsi_stmt (*gsi);
1512 tree fndecl = gimple_call_fndecl (call);
1513 tree addr = gimple_call_arg (call, 0);
1514 tree bounds = chkp_find_bounds (addr, gsi);
1515 gimple extract;
1517 if (DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_GET_PTR_LBOUND)
1518 fndecl = chkp_extract_lower_fndecl;
1519 else if (DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_GET_PTR_UBOUND)
1520 fndecl = chkp_extract_upper_fndecl;
1521 else
1522 gcc_unreachable ();
1524 extract = gimple_build_call (fndecl, 1, bounds);
1525 gimple_call_set_lhs (extract, gimple_call_lhs (call));
1526 chkp_mark_stmt (extract);
1528 gsi_replace (gsi, extract, false);
1531 /* Return COMPONENT_REF accessing FIELD in OBJ. */
1532 static tree
1533 chkp_build_component_ref (tree obj, tree field)
1535 tree res;
1537 /* If object is TMR then we do not use component_ref but
1538 add offset instead. We need it to be able to get addr
1539 of the reasult later. */
1540 if (TREE_CODE (obj) == TARGET_MEM_REF)
1542 tree offs = TMR_OFFSET (obj);
1543 offs = fold_binary_to_constant (PLUS_EXPR, TREE_TYPE (offs),
1544 offs, DECL_FIELD_OFFSET (field));
1546 gcc_assert (offs);
1548 res = copy_node (obj);
1549 TREE_TYPE (res) = TREE_TYPE (field);
1550 TMR_OFFSET (res) = offs;
1552 else
1553 res = build3 (COMPONENT_REF, TREE_TYPE (field), obj, field, NULL_TREE);
1555 return res;
1558 /* Return ARRAY_REF for array ARR and index IDX with
1559 specified element type ETYPE and element size ESIZE. */
1560 static tree
1561 chkp_build_array_ref (tree arr, tree etype, tree esize,
1562 unsigned HOST_WIDE_INT idx)
1564 tree index = build_int_cst (size_type_node, idx);
1565 tree res;
1567 /* If object is TMR then we do not use array_ref but
1568 add offset instead. We need it to be able to get addr
1569 of the reasult later. */
1570 if (TREE_CODE (arr) == TARGET_MEM_REF)
1572 tree offs = TMR_OFFSET (arr);
1574 esize = fold_binary_to_constant (MULT_EXPR, TREE_TYPE (esize),
1575 esize, index);
1576 gcc_assert(esize);
1578 offs = fold_binary_to_constant (PLUS_EXPR, TREE_TYPE (offs),
1579 offs, esize);
1580 gcc_assert (offs);
1582 res = copy_node (arr);
1583 TREE_TYPE (res) = etype;
1584 TMR_OFFSET (res) = offs;
1586 else
1587 res = build4 (ARRAY_REF, etype, arr, index, NULL_TREE, NULL_TREE);
1589 return res;
1592 /* Helper function for chkp_add_bounds_to_call_stmt.
1593 Fill ALL_BOUNDS output array with created bounds.
1595 OFFS is used for recursive calls and holds basic
1596 offset of TYPE in outer structure in bits.
1598 ITER points a position where bounds are searched.
1600 ALL_BOUNDS[i] is filled with elem bounds if there
1601 is a field in TYPE which has pointer type and offset
1602 equal to i * POINTER_SIZE in bits. */
1603 static void
1604 chkp_find_bounds_for_elem (tree elem, tree *all_bounds,
1605 HOST_WIDE_INT offs,
1606 gimple_stmt_iterator *iter)
1608 tree type = TREE_TYPE (elem);
1610 if (BOUNDED_TYPE_P (type))
1612 if (!all_bounds[offs / POINTER_SIZE])
1614 tree temp = make_temp_ssa_name (type, NULL, "");
1615 gimple assign = gimple_build_assign (temp, elem);
1616 gimple_stmt_iterator gsi;
1618 gsi_insert_before (iter, assign, GSI_SAME_STMT);
1619 gsi = gsi_for_stmt (assign);
1621 all_bounds[offs / POINTER_SIZE] = chkp_find_bounds (temp, &gsi);
1624 else if (RECORD_OR_UNION_TYPE_P (type))
1626 tree field;
1628 for (field = TYPE_FIELDS (type); field; field = DECL_CHAIN (field))
1629 if (TREE_CODE (field) == FIELD_DECL)
1631 tree base = unshare_expr (elem);
1632 tree field_ref = chkp_build_component_ref (base, field);
1633 HOST_WIDE_INT field_offs
1634 = TREE_INT_CST_LOW (DECL_FIELD_BIT_OFFSET (field));
1635 if (DECL_FIELD_OFFSET (field))
1636 field_offs += TREE_INT_CST_LOW (DECL_FIELD_OFFSET (field)) * 8;
1638 chkp_find_bounds_for_elem (field_ref, all_bounds,
1639 offs + field_offs, iter);
1642 else if (TREE_CODE (type) == ARRAY_TYPE)
1644 tree maxval = TYPE_MAX_VALUE (TYPE_DOMAIN (type));
1645 tree etype = TREE_TYPE (type);
1646 HOST_WIDE_INT esize = TREE_INT_CST_LOW (TYPE_SIZE (etype));
1647 unsigned HOST_WIDE_INT cur;
1649 if (!maxval || integer_minus_onep (maxval))
1650 return;
1652 for (cur = 0; cur <= TREE_INT_CST_LOW (maxval); cur++)
1654 tree base = unshare_expr (elem);
1655 tree arr_elem = chkp_build_array_ref (base, etype,
1656 TYPE_SIZE (etype),
1657 cur);
1658 chkp_find_bounds_for_elem (arr_elem, all_bounds, offs + cur * esize,
1659 iter);
1664 /* Fill HAVE_BOUND output bitmap with information about
1665 bounds requred for object of type TYPE.
1667 OFFS is used for recursive calls and holds basic
1668 offset of TYPE in outer structure in bits.
1670 HAVE_BOUND[i] is set to 1 if there is a field
1671 in TYPE which has pointer type and offset
1672 equal to i * POINTER_SIZE - OFFS in bits. */
1673 void
1674 chkp_find_bound_slots_1 (const_tree type, bitmap have_bound,
1675 HOST_WIDE_INT offs)
1677 if (BOUNDED_TYPE_P (type))
1678 bitmap_set_bit (have_bound, offs / POINTER_SIZE);
1679 else if (RECORD_OR_UNION_TYPE_P (type))
1681 tree field;
1683 for (field = TYPE_FIELDS (type); field; field = DECL_CHAIN (field))
1684 if (TREE_CODE (field) == FIELD_DECL)
1686 HOST_WIDE_INT field_offs
1687 = TREE_INT_CST_LOW (DECL_FIELD_BIT_OFFSET (field));
1688 if (DECL_FIELD_OFFSET (field))
1689 field_offs += TREE_INT_CST_LOW (DECL_FIELD_OFFSET (field)) * 8;
1690 chkp_find_bound_slots_1 (TREE_TYPE (field), have_bound,
1691 offs + field_offs);
1694 else if (TREE_CODE (type) == ARRAY_TYPE)
1696 tree maxval = TYPE_MAX_VALUE (TYPE_DOMAIN (type));
1697 tree etype = TREE_TYPE (type);
1698 HOST_WIDE_INT esize = TREE_INT_CST_LOW (TYPE_SIZE (etype));
1699 unsigned HOST_WIDE_INT cur;
1701 if (!maxval
1702 || TREE_CODE (maxval) != INTEGER_CST
1703 || integer_minus_onep (maxval))
1704 return;
1706 for (cur = 0; cur <= TREE_INT_CST_LOW (maxval); cur++)
1707 chkp_find_bound_slots_1 (etype, have_bound, offs + cur * esize);
1711 /* Fill bitmap RES with information about bounds for
1712 type TYPE. See chkp_find_bound_slots_1 for more
1713 details. */
1714 void
1715 chkp_find_bound_slots (const_tree type, bitmap res)
1717 bitmap_clear (res);
1718 chkp_find_bound_slots_1 (type, res, 0);
1721 /* Return 1 if call to FNDECL should be instrumented
1722 and 0 otherwise. */
1724 static bool
1725 chkp_instrument_normal_builtin (tree fndecl)
1727 switch (DECL_FUNCTION_CODE (fndecl))
1729 case BUILT_IN_STRLEN:
1730 case BUILT_IN_STRCPY:
1731 case BUILT_IN_STRNCPY:
1732 case BUILT_IN_STPCPY:
1733 case BUILT_IN_STPNCPY:
1734 case BUILT_IN_STRCAT:
1735 case BUILT_IN_STRNCAT:
1736 case BUILT_IN_MEMCPY:
1737 case BUILT_IN_MEMPCPY:
1738 case BUILT_IN_MEMSET:
1739 case BUILT_IN_MEMMOVE:
1740 case BUILT_IN_BZERO:
1741 case BUILT_IN_STRCMP:
1742 case BUILT_IN_STRNCMP:
1743 case BUILT_IN_BCMP:
1744 case BUILT_IN_MEMCMP:
1745 case BUILT_IN_MEMCPY_CHK:
1746 case BUILT_IN_MEMPCPY_CHK:
1747 case BUILT_IN_MEMMOVE_CHK:
1748 case BUILT_IN_MEMSET_CHK:
1749 case BUILT_IN_STRCPY_CHK:
1750 case BUILT_IN_STRNCPY_CHK:
1751 case BUILT_IN_STPCPY_CHK:
1752 case BUILT_IN_STPNCPY_CHK:
1753 case BUILT_IN_STRCAT_CHK:
1754 case BUILT_IN_STRNCAT_CHK:
1755 case BUILT_IN_MALLOC:
1756 case BUILT_IN_CALLOC:
1757 case BUILT_IN_REALLOC:
1758 return 1;
1760 default:
1761 return 0;
1765 /* Add bound arguments to call statement pointed by GSI.
1766 Also performs a replacement of user checker builtins calls
1767 with internal ones. */
1769 static void
1770 chkp_add_bounds_to_call_stmt (gimple_stmt_iterator *gsi)
1772 gcall *call = as_a <gcall *> (gsi_stmt (*gsi));
1773 unsigned arg_no = 0;
1774 tree fndecl = gimple_call_fndecl (call);
1775 tree fntype;
1776 tree first_formal_arg;
1777 tree arg;
1778 bool use_fntype = false;
1779 tree op;
1780 ssa_op_iter iter;
1781 gcall *new_call;
1783 /* Do nothing for internal functions. */
1784 if (gimple_call_internal_p (call))
1785 return;
1787 fntype = TREE_TYPE (TREE_TYPE (gimple_call_fn (call)));
1789 /* Do nothing if back-end builtin is called. */
1790 if (fndecl && DECL_BUILT_IN_CLASS (fndecl) == BUILT_IN_MD)
1791 return;
1793 /* Do nothing for some middle-end builtins. */
1794 if (fndecl && DECL_BUILT_IN_CLASS (fndecl) == BUILT_IN_NORMAL
1795 && DECL_FUNCTION_CODE (fndecl) == BUILT_IN_OBJECT_SIZE)
1796 return;
1798 /* Do nothing for calls to not instrumentable functions. */
1799 if (fndecl && !chkp_instrumentable_p (fndecl))
1800 return;
1802 /* Ignore CHKP_INIT_PTR_BOUNDS, CHKP_NULL_PTR_BOUNDS
1803 and CHKP_COPY_PTR_BOUNDS. */
1804 if (fndecl && DECL_BUILT_IN_CLASS (fndecl) == BUILT_IN_NORMAL
1805 && (DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_INIT_PTR_BOUNDS
1806 || DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_NULL_PTR_BOUNDS
1807 || DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_COPY_PTR_BOUNDS
1808 || DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_SET_PTR_BOUNDS))
1809 return;
1811 /* Check user builtins are replaced with checks. */
1812 if (fndecl && DECL_BUILT_IN_CLASS (fndecl) == BUILT_IN_NORMAL
1813 && (DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_CHECK_PTR_LBOUNDS
1814 || DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_CHECK_PTR_UBOUNDS
1815 || DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_CHECK_PTR_BOUNDS))
1817 chkp_replace_address_check_builtin (gsi, integer_minus_one_node);
1818 return;
1821 /* Check user builtins are replaced with bound extract. */
1822 if (fndecl && DECL_BUILT_IN_CLASS (fndecl) == BUILT_IN_NORMAL
1823 && (DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_GET_PTR_LBOUND
1824 || DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_GET_PTR_UBOUND))
1826 chkp_replace_extract_builtin (gsi);
1827 return;
1830 /* BUILT_IN_CHKP_NARROW_PTR_BOUNDS call is replaced with
1831 target narrow bounds call. */
1832 if (fndecl && DECL_BUILT_IN_CLASS (fndecl) == BUILT_IN_NORMAL
1833 && DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_NARROW_PTR_BOUNDS)
1835 tree arg = gimple_call_arg (call, 1);
1836 tree bounds = chkp_find_bounds (arg, gsi);
1838 gimple_call_set_fndecl (call, chkp_narrow_bounds_fndecl);
1839 gimple_call_set_arg (call, 1, bounds);
1840 update_stmt (call);
1842 return;
1845 /* BUILT_IN_CHKP_STORE_PTR_BOUNDS call is replaced with
1846 bndstx call. */
1847 if (fndecl && DECL_BUILT_IN_CLASS (fndecl) == BUILT_IN_NORMAL
1848 && DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_STORE_PTR_BOUNDS)
1850 tree addr = gimple_call_arg (call, 0);
1851 tree ptr = gimple_call_arg (call, 1);
1852 tree bounds = chkp_find_bounds (ptr, gsi);
1853 gimple_stmt_iterator iter = gsi_for_stmt (call);
1855 chkp_build_bndstx (addr, ptr, bounds, gsi);
1856 gsi_remove (&iter, true);
1858 return;
1861 if (!flag_chkp_instrument_calls)
1862 return;
1864 /* We instrument only some subset of builtins. We also instrument
1865 builtin calls to be inlined. */
1866 if (fndecl
1867 && DECL_BUILT_IN_CLASS (fndecl) == BUILT_IN_NORMAL
1868 && !chkp_instrument_normal_builtin (fndecl))
1870 if (!lookup_attribute ("always_inline", DECL_ATTRIBUTES (fndecl)))
1871 return;
1873 struct cgraph_node *clone = chkp_maybe_create_clone (fndecl);
1874 if (!clone
1875 || !gimple_has_body_p (clone->decl))
1876 return;
1879 /* If function decl is available then use it for
1880 formal arguments list. Otherwise use function type. */
1881 if (fndecl && DECL_ARGUMENTS (fndecl))
1882 first_formal_arg = DECL_ARGUMENTS (fndecl);
1883 else
1885 first_formal_arg = TYPE_ARG_TYPES (fntype);
1886 use_fntype = true;
1889 /* Fill vector of new call args. */
1890 vec<tree> new_args = vNULL;
1891 new_args.create (gimple_call_num_args (call));
1892 arg = first_formal_arg;
1893 for (arg_no = 0; arg_no < gimple_call_num_args (call); arg_no++)
1895 tree call_arg = gimple_call_arg (call, arg_no);
1896 tree type;
1898 /* Get arg type using formal argument description
1899 or actual argument type. */
1900 if (arg)
1901 if (use_fntype)
1902 if (TREE_VALUE (arg) != void_type_node)
1904 type = TREE_VALUE (arg);
1905 arg = TREE_CHAIN (arg);
1907 else
1908 type = TREE_TYPE (call_arg);
1909 else
1911 type = TREE_TYPE (arg);
1912 arg = TREE_CHAIN (arg);
1914 else
1915 type = TREE_TYPE (call_arg);
1917 new_args.safe_push (call_arg);
1919 if (BOUNDED_TYPE_P (type)
1920 || pass_by_reference (NULL, TYPE_MODE (type), type, true))
1921 new_args.safe_push (chkp_find_bounds (call_arg, gsi));
1922 else if (chkp_type_has_pointer (type))
1924 HOST_WIDE_INT max_bounds
1925 = TREE_INT_CST_LOW (TYPE_SIZE (type)) / POINTER_SIZE;
1926 tree *all_bounds = (tree *)xmalloc (sizeof (tree) * max_bounds);
1927 HOST_WIDE_INT bnd_no;
1929 memset (all_bounds, 0, sizeof (tree) * max_bounds);
1931 chkp_find_bounds_for_elem (call_arg, all_bounds, 0, gsi);
1933 for (bnd_no = 0; bnd_no < max_bounds; bnd_no++)
1934 if (all_bounds[bnd_no])
1935 new_args.safe_push (all_bounds[bnd_no]);
1937 free (all_bounds);
1941 if (new_args.length () == gimple_call_num_args (call))
1942 new_call = call;
1943 else
1945 new_call = gimple_build_call_vec (gimple_op (call, 1), new_args);
1946 gimple_call_set_lhs (new_call, gimple_call_lhs (call));
1947 gimple_call_copy_flags (new_call, call);
1948 gimple_call_set_chain (new_call, gimple_call_chain (call));
1950 new_args.release ();
1952 /* For direct calls fndecl is replaced with instrumented version. */
1953 if (fndecl)
1955 tree new_decl = chkp_maybe_create_clone (fndecl)->decl;
1956 gimple_call_set_fndecl (new_call, new_decl);
1957 gimple_call_set_fntype (new_call, TREE_TYPE (new_decl));
1959 /* For indirect call we should fix function pointer type if
1960 pass some bounds. */
1961 else if (new_call != call)
1963 tree type = gimple_call_fntype (call);
1964 type = chkp_copy_function_type_adding_bounds (type);
1965 gimple_call_set_fntype (new_call, type);
1968 /* replace old call statement with the new one. */
1969 if (call != new_call)
1971 FOR_EACH_SSA_TREE_OPERAND (op, call, iter, SSA_OP_ALL_DEFS)
1973 SSA_NAME_DEF_STMT (op) = new_call;
1975 gsi_replace (gsi, new_call, true);
1977 else
1978 update_stmt (new_call);
1980 gimple_call_set_with_bounds (new_call, true);
1983 /* Return constant static bounds var with specified bounds LB and UB.
1984 If such var does not exists then new var is created with specified NAME. */
1985 static tree
1986 chkp_make_static_const_bounds (HOST_WIDE_INT lb,
1987 HOST_WIDE_INT ub,
1988 const char *name)
1990 tree id = get_identifier (name);
1991 tree var;
1992 varpool_node *node;
1993 symtab_node *snode;
1995 var = build_decl (UNKNOWN_LOCATION, VAR_DECL, id,
1996 pointer_bounds_type_node);
1997 TREE_STATIC (var) = 1;
1998 TREE_PUBLIC (var) = 1;
2000 /* With LTO we may have constant bounds already in varpool.
2001 Try to find it. */
2002 if ((snode = symtab_node::get_for_asmname (DECL_ASSEMBLER_NAME (var))))
2004 /* We don't allow this symbol usage for non bounds. */
2005 if (snode->type != SYMTAB_VARIABLE
2006 || !POINTER_BOUNDS_P (snode->decl))
2007 sorry ("-fcheck-pointer-bounds requires '%s' "
2008 "name for internal usage",
2009 IDENTIFIER_POINTER (DECL_ASSEMBLER_NAME (var)));
2011 return snode->decl;
2014 TREE_USED (var) = 1;
2015 TREE_READONLY (var) = 1;
2016 TREE_ADDRESSABLE (var) = 0;
2017 DECL_ARTIFICIAL (var) = 1;
2018 DECL_READ_P (var) = 1;
2019 DECL_INITIAL (var) = targetm.chkp_make_bounds_constant (lb, ub);
2020 make_decl_one_only (var, DECL_ASSEMBLER_NAME (var));
2021 /* We may use this symbol during ctors generation in chkp_finish_file
2022 when all symbols are emitted. Force output to avoid undefined
2023 symbols in ctors. */
2024 node = varpool_node::get_create (var);
2025 node->force_output = 1;
2027 varpool_node::finalize_decl (var);
2029 return var;
2032 /* Generate code to make bounds with specified lower bound LB and SIZE.
2033 if AFTER is 1 then code is inserted after position pointed by ITER
2034 otherwise code is inserted before position pointed by ITER.
2035 If ITER is NULL then code is added to entry block. */
2036 static tree
2037 chkp_make_bounds (tree lb, tree size, gimple_stmt_iterator *iter, bool after)
2039 gimple_seq seq;
2040 gimple_stmt_iterator gsi;
2041 gimple stmt;
2042 tree bounds;
2044 if (iter)
2045 gsi = *iter;
2046 else
2047 gsi = gsi_start_bb (chkp_get_entry_block ());
2049 seq = NULL;
2051 lb = chkp_force_gimple_call_op (lb, &seq);
2052 size = chkp_force_gimple_call_op (size, &seq);
2054 stmt = gimple_build_call (chkp_bndmk_fndecl, 2, lb, size);
2055 chkp_mark_stmt (stmt);
2057 bounds = chkp_get_tmp_reg (stmt);
2058 gimple_call_set_lhs (stmt, bounds);
2060 gimple_seq_add_stmt (&seq, stmt);
2062 if (iter && after)
2063 gsi_insert_seq_after (&gsi, seq, GSI_SAME_STMT);
2064 else
2065 gsi_insert_seq_before (&gsi, seq, GSI_SAME_STMT);
2067 if (dump_file && (dump_flags & TDF_DETAILS))
2069 fprintf (dump_file, "Made bounds: ");
2070 print_gimple_stmt (dump_file, stmt, 0, TDF_VOPS|TDF_MEMSYMS);
2071 if (iter)
2073 fprintf (dump_file, " inserted before statement: ");
2074 print_gimple_stmt (dump_file, gsi_stmt (*iter), 0, TDF_VOPS|TDF_MEMSYMS);
2076 else
2077 fprintf (dump_file, " at function entry\n");
2080 /* update_stmt (stmt); */
2082 return bounds;
2085 /* Return var holding zero bounds. */
2086 tree
2087 chkp_get_zero_bounds_var (void)
2089 if (!chkp_zero_bounds_var)
2090 chkp_zero_bounds_var
2091 = chkp_make_static_const_bounds (0, -1,
2092 CHKP_ZERO_BOUNDS_VAR_NAME);
2093 return chkp_zero_bounds_var;
2096 /* Return var holding none bounds. */
2097 tree
2098 chkp_get_none_bounds_var (void)
2100 if (!chkp_none_bounds_var)
2101 chkp_none_bounds_var
2102 = chkp_make_static_const_bounds (-1, 0,
2103 CHKP_NONE_BOUNDS_VAR_NAME);
2104 return chkp_none_bounds_var;
2107 /* Return SSA_NAME used to represent zero bounds. */
2108 static tree
2109 chkp_get_zero_bounds (void)
2111 if (zero_bounds)
2112 return zero_bounds;
2114 if (dump_file && (dump_flags & TDF_DETAILS))
2115 fprintf (dump_file, "Creating zero bounds...");
2117 if ((flag_chkp_use_static_bounds && flag_chkp_use_static_const_bounds)
2118 || flag_chkp_use_static_const_bounds > 0)
2120 gimple_stmt_iterator gsi = gsi_start_bb (chkp_get_entry_block ());
2121 gimple stmt;
2123 zero_bounds = chkp_get_tmp_reg (NULL);
2124 stmt = gimple_build_assign (zero_bounds, chkp_get_zero_bounds_var ());
2125 gsi_insert_before (&gsi, stmt, GSI_SAME_STMT);
2127 else
2128 zero_bounds = chkp_make_bounds (integer_zero_node,
2129 integer_zero_node,
2130 NULL,
2131 false);
2133 return zero_bounds;
2136 /* Return SSA_NAME used to represent none bounds. */
2137 static tree
2138 chkp_get_none_bounds (void)
2140 if (none_bounds)
2141 return none_bounds;
2143 if (dump_file && (dump_flags & TDF_DETAILS))
2144 fprintf (dump_file, "Creating none bounds...");
2147 if ((flag_chkp_use_static_bounds && flag_chkp_use_static_const_bounds)
2148 || flag_chkp_use_static_const_bounds > 0)
2150 gimple_stmt_iterator gsi = gsi_start_bb (chkp_get_entry_block ());
2151 gimple stmt;
2153 none_bounds = chkp_get_tmp_reg (NULL);
2154 stmt = gimple_build_assign (none_bounds, chkp_get_none_bounds_var ());
2155 gsi_insert_before (&gsi, stmt, GSI_SAME_STMT);
2157 else
2158 none_bounds = chkp_make_bounds (integer_minus_one_node,
2159 build_int_cst (size_type_node, 2),
2160 NULL,
2161 false);
2163 return none_bounds;
2166 /* Return bounds to be used as a result of operation which
2167 should not create poiunter (e.g. MULT_EXPR). */
2168 static tree
2169 chkp_get_invalid_op_bounds (void)
2171 return chkp_get_zero_bounds ();
2174 /* Return bounds to be used for loads of non-pointer values. */
2175 static tree
2176 chkp_get_nonpointer_load_bounds (void)
2178 return chkp_get_zero_bounds ();
2181 /* Return 1 if may use bndret call to get bounds for pointer
2182 returned by CALL. */
2183 static bool
2184 chkp_call_returns_bounds_p (gcall *call)
2186 if (gimple_call_internal_p (call))
2187 return false;
2189 if (gimple_call_builtin_p (call, BUILT_IN_CHKP_NARROW_PTR_BOUNDS)
2190 || chkp_gimple_call_builtin_p (call, BUILT_IN_CHKP_NARROW))
2191 return true;
2193 if (gimple_call_with_bounds_p (call))
2194 return true;
2196 tree fndecl = gimple_call_fndecl (call);
2198 if (fndecl && DECL_BUILT_IN_CLASS (fndecl) == BUILT_IN_MD)
2199 return false;
2201 if (fndecl && !chkp_instrumentable_p (fndecl))
2202 return false;
2204 if (fndecl && DECL_BUILT_IN_CLASS (fndecl) == BUILT_IN_NORMAL)
2206 if (chkp_instrument_normal_builtin (fndecl))
2207 return true;
2209 if (!lookup_attribute ("always_inline", DECL_ATTRIBUTES (fndecl)))
2210 return false;
2212 struct cgraph_node *clone = chkp_maybe_create_clone (fndecl);
2213 return (clone && gimple_has_body_p (clone->decl));
2216 return true;
2219 /* Build bounds returned by CALL. */
2220 static tree
2221 chkp_build_returned_bound (gcall *call)
2223 gimple_stmt_iterator gsi;
2224 tree bounds;
2225 gimple stmt;
2226 tree fndecl = gimple_call_fndecl (call);
2227 unsigned int retflags;
2229 /* To avoid fixing alloca expands in targets we handle
2230 it separately. */
2231 if (fndecl
2232 && DECL_BUILT_IN_CLASS (fndecl) == BUILT_IN_NORMAL
2233 && (DECL_FUNCTION_CODE (fndecl) == BUILT_IN_ALLOCA
2234 || DECL_FUNCTION_CODE (fndecl) == BUILT_IN_ALLOCA_WITH_ALIGN))
2236 tree size = gimple_call_arg (call, 0);
2237 tree lb = gimple_call_lhs (call);
2238 gimple_stmt_iterator iter = gsi_for_stmt (call);
2239 bounds = chkp_make_bounds (lb, size, &iter, true);
2241 /* We know bounds returned by set_bounds builtin call. */
2242 else if (fndecl
2243 && DECL_BUILT_IN_CLASS (fndecl) == BUILT_IN_NORMAL
2244 && DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_SET_PTR_BOUNDS)
2246 tree lb = gimple_call_arg (call, 0);
2247 tree size = gimple_call_arg (call, 1);
2248 gimple_stmt_iterator iter = gsi_for_stmt (call);
2249 bounds = chkp_make_bounds (lb, size, &iter, true);
2251 /* Detect bounds initialization calls. */
2252 else if (fndecl
2253 && DECL_BUILT_IN_CLASS (fndecl) == BUILT_IN_NORMAL
2254 && DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_INIT_PTR_BOUNDS)
2255 bounds = chkp_get_zero_bounds ();
2256 /* Detect bounds nullification calls. */
2257 else if (fndecl
2258 && DECL_BUILT_IN_CLASS (fndecl) == BUILT_IN_NORMAL
2259 && DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_NULL_PTR_BOUNDS)
2260 bounds = chkp_get_none_bounds ();
2261 /* Detect bounds copy calls. */
2262 else if (fndecl
2263 && DECL_BUILT_IN_CLASS (fndecl) == BUILT_IN_NORMAL
2264 && DECL_FUNCTION_CODE (fndecl) == BUILT_IN_CHKP_COPY_PTR_BOUNDS)
2266 gimple_stmt_iterator iter = gsi_for_stmt (call);
2267 bounds = chkp_find_bounds (gimple_call_arg (call, 1), &iter);
2269 /* Do not use retbnd when returned bounds are equal to some
2270 of passed bounds. */
2271 else if (((retflags = gimple_call_return_flags (call)) & ERF_RETURNS_ARG)
2272 && (retflags & ERF_RETURN_ARG_MASK) < gimple_call_num_args (call))
2274 gimple_stmt_iterator iter = gsi_for_stmt (call);
2275 unsigned int retarg = retflags & ERF_RETURN_ARG_MASK, argno;
2276 if (gimple_call_with_bounds_p (call))
2278 for (argno = 0; argno < gimple_call_num_args (call); argno++)
2279 if (!POINTER_BOUNDS_P (gimple_call_arg (call, argno)))
2281 if (retarg)
2282 retarg--;
2283 else
2284 break;
2287 else
2288 argno = retarg;
2290 bounds = chkp_find_bounds (gimple_call_arg (call, argno), &iter);
2292 else if (chkp_call_returns_bounds_p (call))
2294 gcc_assert (TREE_CODE (gimple_call_lhs (call)) == SSA_NAME);
2296 /* In general case build checker builtin call to
2297 obtain returned bounds. */
2298 stmt = gimple_build_call (chkp_ret_bnd_fndecl, 1,
2299 gimple_call_lhs (call));
2300 chkp_mark_stmt (stmt);
2302 gsi = gsi_for_stmt (call);
2303 gsi_insert_after (&gsi, stmt, GSI_SAME_STMT);
2305 bounds = chkp_get_tmp_reg (stmt);
2306 gimple_call_set_lhs (stmt, bounds);
2308 update_stmt (stmt);
2310 else
2311 bounds = chkp_get_zero_bounds ();
2313 if (dump_file && (dump_flags & TDF_DETAILS))
2315 fprintf (dump_file, "Built returned bounds (");
2316 print_generic_expr (dump_file, bounds, 0);
2317 fprintf (dump_file, ") for call: ");
2318 print_gimple_stmt (dump_file, call, 0, TDF_VOPS|TDF_MEMSYMS);
2321 bounds = chkp_maybe_copy_and_register_bounds (gimple_call_lhs (call), bounds);
2323 return bounds;
2326 /* Return bounds used as returned by call
2327 which produced SSA name VAL. */
2328 gcall *
2329 chkp_retbnd_call_by_val (tree val)
2331 if (TREE_CODE (val) != SSA_NAME)
2332 return NULL;
2334 gcc_assert (gimple_code (SSA_NAME_DEF_STMT (val)) == GIMPLE_CALL);
2336 imm_use_iterator use_iter;
2337 use_operand_p use_p;
2338 FOR_EACH_IMM_USE_FAST (use_p, use_iter, val)
2339 if (gimple_code (USE_STMT (use_p)) == GIMPLE_CALL
2340 && gimple_call_fndecl (USE_STMT (use_p)) == chkp_ret_bnd_fndecl)
2341 return as_a <gcall *> (USE_STMT (use_p));
2343 return NULL;
2346 /* Check the next parameter for the given PARM is bounds
2347 and return it's default SSA_NAME (create if required). */
2348 static tree
2349 chkp_get_next_bounds_parm (tree parm)
2351 tree bounds = TREE_CHAIN (parm);
2352 gcc_assert (POINTER_BOUNDS_P (bounds));
2353 bounds = ssa_default_def (cfun, bounds);
2354 if (!bounds)
2356 bounds = make_ssa_name (TREE_CHAIN (parm), gimple_build_nop ());
2357 set_ssa_default_def (cfun, TREE_CHAIN (parm), bounds);
2359 return bounds;
2362 /* Return bounds to be used for input argument PARM. */
2363 static tree
2364 chkp_get_bound_for_parm (tree parm)
2366 tree decl = SSA_NAME_VAR (parm);
2367 tree bounds;
2369 gcc_assert (TREE_CODE (decl) == PARM_DECL);
2371 bounds = chkp_get_registered_bounds (parm);
2373 if (!bounds)
2374 bounds = chkp_get_registered_bounds (decl);
2376 if (!bounds)
2378 tree orig_decl = cgraph_node::get (cfun->decl)->orig_decl;
2380 /* For static chain param we return zero bounds
2381 because currently we do not check dereferences
2382 of this pointer. */
2383 if (cfun->static_chain_decl == decl)
2384 bounds = chkp_get_zero_bounds ();
2385 /* If non instrumented runtime is used then it may be useful
2386 to use zero bounds for input arguments of main
2387 function. */
2388 else if (flag_chkp_zero_input_bounds_for_main
2389 && strcmp (IDENTIFIER_POINTER (DECL_ASSEMBLER_NAME (orig_decl)),
2390 "main") == 0)
2391 bounds = chkp_get_zero_bounds ();
2392 else if (BOUNDED_P (parm))
2394 bounds = chkp_get_next_bounds_parm (decl);
2395 bounds = chkp_maybe_copy_and_register_bounds (decl, bounds);
2397 if (dump_file && (dump_flags & TDF_DETAILS))
2399 fprintf (dump_file, "Built arg bounds (");
2400 print_generic_expr (dump_file, bounds, 0);
2401 fprintf (dump_file, ") for arg: ");
2402 print_node (dump_file, "", decl, 0);
2405 else
2406 bounds = chkp_get_zero_bounds ();
2409 if (!chkp_get_registered_bounds (parm))
2410 bounds = chkp_maybe_copy_and_register_bounds (parm, bounds);
2412 if (dump_file && (dump_flags & TDF_DETAILS))
2414 fprintf (dump_file, "Using bounds ");
2415 print_generic_expr (dump_file, bounds, 0);
2416 fprintf (dump_file, " for parm ");
2417 print_generic_expr (dump_file, parm, 0);
2418 fprintf (dump_file, " of type ");
2419 print_generic_expr (dump_file, TREE_TYPE (parm), 0);
2420 fprintf (dump_file, ".\n");
2423 return bounds;
2426 /* Build and return CALL_EXPR for bndstx builtin with specified
2427 arguments. */
2428 tree
2429 chkp_build_bndldx_call (tree addr, tree ptr)
2431 tree fn = build1 (ADDR_EXPR,
2432 build_pointer_type (TREE_TYPE (chkp_bndldx_fndecl)),
2433 chkp_bndldx_fndecl);
2434 tree call = build_call_nary (TREE_TYPE (TREE_TYPE (chkp_bndldx_fndecl)),
2435 fn, 2, addr, ptr);
2436 CALL_WITH_BOUNDS_P (call) = true;
2437 return call;
2440 /* Insert code to load bounds for PTR located by ADDR.
2441 Code is inserted after position pointed by GSI.
2442 Loaded bounds are returned. */
2443 static tree
2444 chkp_build_bndldx (tree addr, tree ptr, gimple_stmt_iterator *gsi)
2446 gimple_seq seq;
2447 gimple stmt;
2448 tree bounds;
2450 seq = NULL;
2452 addr = chkp_force_gimple_call_op (addr, &seq);
2453 ptr = chkp_force_gimple_call_op (ptr, &seq);
2455 stmt = gimple_build_call (chkp_bndldx_fndecl, 2, addr, ptr);
2456 chkp_mark_stmt (stmt);
2457 bounds = chkp_get_tmp_reg (stmt);
2458 gimple_call_set_lhs (stmt, bounds);
2460 gimple_seq_add_stmt (&seq, stmt);
2462 gsi_insert_seq_after (gsi, seq, GSI_CONTINUE_LINKING);
2464 if (dump_file && (dump_flags & TDF_DETAILS))
2466 fprintf (dump_file, "Generated bndldx for pointer ");
2467 print_generic_expr (dump_file, ptr, 0);
2468 fprintf (dump_file, ": ");
2469 print_gimple_stmt (dump_file, stmt, 0, TDF_VOPS|TDF_MEMSYMS);
2472 return bounds;
2475 /* Build and return CALL_EXPR for bndstx builtin with specified
2476 arguments. */
2477 tree
2478 chkp_build_bndstx_call (tree addr, tree ptr, tree bounds)
2480 tree fn = build1 (ADDR_EXPR,
2481 build_pointer_type (TREE_TYPE (chkp_bndstx_fndecl)),
2482 chkp_bndstx_fndecl);
2483 tree call = build_call_nary (TREE_TYPE (TREE_TYPE (chkp_bndstx_fndecl)),
2484 fn, 3, ptr, bounds, addr);
2485 CALL_WITH_BOUNDS_P (call) = true;
2486 return call;
2489 /* Insert code to store BOUNDS for PTR stored by ADDR.
2490 New statements are inserted after position pointed
2491 by GSI. */
2492 void
2493 chkp_build_bndstx (tree addr, tree ptr, tree bounds,
2494 gimple_stmt_iterator *gsi)
2496 gimple_seq seq;
2497 gimple stmt;
2499 seq = NULL;
2501 addr = chkp_force_gimple_call_op (addr, &seq);
2502 ptr = chkp_force_gimple_call_op (ptr, &seq);
2504 stmt = gimple_build_call (chkp_bndstx_fndecl, 3, ptr, bounds, addr);
2505 chkp_mark_stmt (stmt);
2506 gimple_call_set_with_bounds (stmt, true);
2508 gimple_seq_add_stmt (&seq, stmt);
2510 gsi_insert_seq_after (gsi, seq, GSI_CONTINUE_LINKING);
2512 if (dump_file && (dump_flags & TDF_DETAILS))
2514 fprintf (dump_file, "Generated bndstx for pointer store ");
2515 print_gimple_stmt (dump_file, gsi_stmt (*gsi), 0, TDF_VOPS|TDF_MEMSYMS);
2516 print_gimple_stmt (dump_file, stmt, 2, TDF_VOPS|TDF_MEMSYMS);
2520 /* Compute bounds for pointer NODE which was assigned in
2521 assignment statement ASSIGN. Return computed bounds. */
2522 static tree
2523 chkp_compute_bounds_for_assignment (tree node, gimple assign)
2525 enum tree_code rhs_code = gimple_assign_rhs_code (assign);
2526 tree rhs1 = gimple_assign_rhs1 (assign);
2527 tree bounds = NULL_TREE;
2528 gimple_stmt_iterator iter = gsi_for_stmt (assign);
2529 tree base = NULL;
2531 if (dump_file && (dump_flags & TDF_DETAILS))
2533 fprintf (dump_file, "Computing bounds for assignment: ");
2534 print_gimple_stmt (dump_file, assign, 0, TDF_VOPS|TDF_MEMSYMS);
2537 switch (rhs_code)
2539 case MEM_REF:
2540 case TARGET_MEM_REF:
2541 case COMPONENT_REF:
2542 case ARRAY_REF:
2543 /* We need to load bounds from the bounds table. */
2544 bounds = chkp_find_bounds_loaded (node, rhs1, &iter);
2545 break;
2547 case VAR_DECL:
2548 case SSA_NAME:
2549 case ADDR_EXPR:
2550 case POINTER_PLUS_EXPR:
2551 case NOP_EXPR:
2552 case CONVERT_EXPR:
2553 case INTEGER_CST:
2554 /* Bounds are just propagated from RHS. */
2555 bounds = chkp_find_bounds (rhs1, &iter);
2556 base = rhs1;
2557 break;
2559 case VIEW_CONVERT_EXPR:
2560 /* Bounds are just propagated from RHS. */
2561 bounds = chkp_find_bounds (TREE_OPERAND (rhs1, 0), &iter);
2562 break;
2564 case PARM_DECL:
2565 if (BOUNDED_P (rhs1))
2567 /* We need to load bounds from the bounds table. */
2568 bounds = chkp_build_bndldx (chkp_build_addr_expr (rhs1),
2569 node, &iter);
2570 TREE_ADDRESSABLE (rhs1) = 1;
2572 else
2573 bounds = chkp_get_nonpointer_load_bounds ();
2574 break;
2576 case MINUS_EXPR:
2577 case PLUS_EXPR:
2578 case BIT_AND_EXPR:
2579 case BIT_IOR_EXPR:
2580 case BIT_XOR_EXPR:
2582 tree rhs2 = gimple_assign_rhs2 (assign);
2583 tree bnd1 = chkp_find_bounds (rhs1, &iter);
2584 tree bnd2 = chkp_find_bounds (rhs2, &iter);
2586 /* First we try to check types of operands. If it
2587 does not help then look at bound values.
2589 If some bounds are incomplete and other are
2590 not proven to be valid (i.e. also incomplete
2591 or invalid because value is not pointer) then
2592 resulting value is incomplete and will be
2593 recomputed later in chkp_finish_incomplete_bounds. */
2594 if (BOUNDED_P (rhs1)
2595 && !BOUNDED_P (rhs2))
2596 bounds = bnd1;
2597 else if (BOUNDED_P (rhs2)
2598 && !BOUNDED_P (rhs1)
2599 && rhs_code != MINUS_EXPR)
2600 bounds = bnd2;
2601 else if (chkp_incomplete_bounds (bnd1))
2602 if (chkp_valid_bounds (bnd2) && rhs_code != MINUS_EXPR
2603 && !chkp_incomplete_bounds (bnd2))
2604 bounds = bnd2;
2605 else
2606 bounds = incomplete_bounds;
2607 else if (chkp_incomplete_bounds (bnd2))
2608 if (chkp_valid_bounds (bnd1)
2609 && !chkp_incomplete_bounds (bnd1))
2610 bounds = bnd1;
2611 else
2612 bounds = incomplete_bounds;
2613 else if (!chkp_valid_bounds (bnd1))
2614 if (chkp_valid_bounds (bnd2) && rhs_code != MINUS_EXPR)
2615 bounds = bnd2;
2616 else if (bnd2 == chkp_get_zero_bounds ())
2617 bounds = bnd2;
2618 else
2619 bounds = bnd1;
2620 else if (!chkp_valid_bounds (bnd2))
2621 bounds = bnd1;
2622 else
2623 /* Seems both operands may have valid bounds
2624 (e.g. pointer minus pointer). In such case
2625 use default invalid op bounds. */
2626 bounds = chkp_get_invalid_op_bounds ();
2628 base = (bounds == bnd1) ? rhs1 : (bounds == bnd2) ? rhs2 : NULL;
2630 break;
2632 case BIT_NOT_EXPR:
2633 case NEGATE_EXPR:
2634 case LSHIFT_EXPR:
2635 case RSHIFT_EXPR:
2636 case LROTATE_EXPR:
2637 case RROTATE_EXPR:
2638 case EQ_EXPR:
2639 case NE_EXPR:
2640 case LT_EXPR:
2641 case LE_EXPR:
2642 case GT_EXPR:
2643 case GE_EXPR:
2644 case MULT_EXPR:
2645 case RDIV_EXPR:
2646 case TRUNC_DIV_EXPR:
2647 case FLOOR_DIV_EXPR:
2648 case CEIL_DIV_EXPR:
2649 case ROUND_DIV_EXPR:
2650 case TRUNC_MOD_EXPR:
2651 case FLOOR_MOD_EXPR:
2652 case CEIL_MOD_EXPR:
2653 case ROUND_MOD_EXPR:
2654 case EXACT_DIV_EXPR:
2655 case FIX_TRUNC_EXPR:
2656 case FLOAT_EXPR:
2657 case REALPART_EXPR:
2658 case IMAGPART_EXPR:
2659 /* No valid bounds may be produced by these exprs. */
2660 bounds = chkp_get_invalid_op_bounds ();
2661 break;
2663 case COND_EXPR:
2665 tree val1 = gimple_assign_rhs2 (assign);
2666 tree val2 = gimple_assign_rhs3 (assign);
2667 tree bnd1 = chkp_find_bounds (val1, &iter);
2668 tree bnd2 = chkp_find_bounds (val2, &iter);
2669 gimple stmt;
2671 if (chkp_incomplete_bounds (bnd1) || chkp_incomplete_bounds (bnd2))
2672 bounds = incomplete_bounds;
2673 else if (bnd1 == bnd2)
2674 bounds = bnd1;
2675 else
2677 rhs1 = unshare_expr (rhs1);
2679 bounds = chkp_get_tmp_reg (assign);
2680 stmt = gimple_build_assign (bounds, COND_EXPR, rhs1, bnd1, bnd2);
2681 gsi_insert_after (&iter, stmt, GSI_SAME_STMT);
2683 if (!chkp_valid_bounds (bnd1) && !chkp_valid_bounds (bnd2))
2684 chkp_mark_invalid_bounds (bounds);
2687 break;
2689 case MAX_EXPR:
2690 case MIN_EXPR:
2692 tree rhs2 = gimple_assign_rhs2 (assign);
2693 tree bnd1 = chkp_find_bounds (rhs1, &iter);
2694 tree bnd2 = chkp_find_bounds (rhs2, &iter);
2696 if (chkp_incomplete_bounds (bnd1) || chkp_incomplete_bounds (bnd2))
2697 bounds = incomplete_bounds;
2698 else if (bnd1 == bnd2)
2699 bounds = bnd1;
2700 else
2702 gimple stmt;
2703 tree cond = build2 (rhs_code == MAX_EXPR ? GT_EXPR : LT_EXPR,
2704 boolean_type_node, rhs1, rhs2);
2705 bounds = chkp_get_tmp_reg (assign);
2706 stmt = gimple_build_assign (bounds, COND_EXPR, cond, bnd1, bnd2);
2708 gsi_insert_after (&iter, stmt, GSI_SAME_STMT);
2710 if (!chkp_valid_bounds (bnd1) && !chkp_valid_bounds (bnd2))
2711 chkp_mark_invalid_bounds (bounds);
2714 break;
2716 default:
2717 bounds = chkp_get_zero_bounds ();
2718 warning (0, "pointer bounds were lost due to unexpected expression %s",
2719 get_tree_code_name (rhs_code));
2722 gcc_assert (bounds);
2724 /* We may reuse bounds of other pointer we copy/modify. But it is not
2725 allowed for abnormal ssa names. If we produced a pointer using
2726 abnormal ssa name, we better make a bounds copy to avoid coalescing
2727 issues. */
2728 if (base
2729 && TREE_CODE (base) == SSA_NAME
2730 && SSA_NAME_OCCURS_IN_ABNORMAL_PHI (base))
2732 gimple stmt = gimple_build_assign (chkp_get_tmp_reg (NULL), bounds);
2733 gsi_insert_after (&iter, stmt, GSI_SAME_STMT);
2734 bounds = gimple_assign_lhs (stmt);
2737 if (node)
2738 bounds = chkp_maybe_copy_and_register_bounds (node, bounds);
2740 return bounds;
2743 /* Compute bounds for ssa name NODE defined by DEF_STMT pointed by ITER.
2745 There are just few statement codes allowed: NOP (for default ssa names),
2746 ASSIGN, CALL, PHI, ASM.
2748 Return computed bounds. */
2749 static tree
2750 chkp_get_bounds_by_definition (tree node, gimple def_stmt,
2751 gphi_iterator *iter)
2753 tree var, bounds;
2754 enum gimple_code code = gimple_code (def_stmt);
2755 gphi *stmt;
2757 if (dump_file && (dump_flags & TDF_DETAILS))
2759 fprintf (dump_file, "Searching for bounds for node: ");
2760 print_generic_expr (dump_file, node, 0);
2762 fprintf (dump_file, " using its definition: ");
2763 print_gimple_stmt (dump_file, def_stmt, 0, TDF_VOPS|TDF_MEMSYMS);
2766 switch (code)
2768 case GIMPLE_NOP:
2769 var = SSA_NAME_VAR (node);
2770 switch (TREE_CODE (var))
2772 case PARM_DECL:
2773 bounds = chkp_get_bound_for_parm (node);
2774 break;
2776 case VAR_DECL:
2777 /* For uninitialized pointers use none bounds. */
2778 bounds = chkp_get_none_bounds ();
2779 bounds = chkp_maybe_copy_and_register_bounds (node, bounds);
2780 break;
2782 case RESULT_DECL:
2784 tree base_type;
2786 gcc_assert (TREE_CODE (TREE_TYPE (node)) == REFERENCE_TYPE);
2788 base_type = TREE_TYPE (TREE_TYPE (node));
2790 gcc_assert (TYPE_SIZE (base_type)
2791 && TREE_CODE (TYPE_SIZE (base_type)) == INTEGER_CST
2792 && tree_to_uhwi (TYPE_SIZE (base_type)) != 0);
2794 bounds = chkp_make_bounds (node, TYPE_SIZE_UNIT (base_type),
2795 NULL, false);
2796 bounds = chkp_maybe_copy_and_register_bounds (node, bounds);
2798 break;
2800 default:
2801 if (dump_file && (dump_flags & TDF_DETAILS))
2803 fprintf (dump_file, "Unexpected var with no definition\n");
2804 print_generic_expr (dump_file, var, 0);
2806 internal_error ("chkp_get_bounds_by_definition: Unexpected var of type %s",
2807 get_tree_code_name (TREE_CODE (var)));
2809 break;
2811 case GIMPLE_ASSIGN:
2812 bounds = chkp_compute_bounds_for_assignment (node, def_stmt);
2813 break;
2815 case GIMPLE_CALL:
2816 bounds = chkp_build_returned_bound (as_a <gcall *> (def_stmt));
2817 break;
2819 case GIMPLE_PHI:
2820 if (SSA_NAME_OCCURS_IN_ABNORMAL_PHI (node))
2821 if (SSA_NAME_VAR (node))
2822 var = chkp_get_bounds_var (SSA_NAME_VAR (node));
2823 else
2824 var = make_temp_ssa_name (pointer_bounds_type_node,
2825 NULL,
2826 CHKP_BOUND_TMP_NAME);
2827 else
2828 var = chkp_get_tmp_var ();
2829 stmt = create_phi_node (var, gimple_bb (def_stmt));
2830 bounds = gimple_phi_result (stmt);
2831 *iter = gsi_for_phi (stmt);
2833 bounds = chkp_maybe_copy_and_register_bounds (node, bounds);
2835 /* Created bounds do not have all phi args computed and
2836 therefore we do not know if there is a valid source
2837 of bounds for that node. Therefore we mark bounds
2838 as incomplete and then recompute them when all phi
2839 args are computed. */
2840 chkp_register_incomplete_bounds (bounds, node);
2841 break;
2843 case GIMPLE_ASM:
2844 bounds = chkp_get_zero_bounds ();
2845 bounds = chkp_maybe_copy_and_register_bounds (node, bounds);
2846 break;
2848 default:
2849 internal_error ("chkp_get_bounds_by_definition: Unexpected GIMPLE code %s",
2850 gimple_code_name[code]);
2853 return bounds;
2856 /* Return CALL_EXPR for bndmk with specified LOWER_BOUND and SIZE. */
2857 tree
2858 chkp_build_make_bounds_call (tree lower_bound, tree size)
2860 tree call = build1 (ADDR_EXPR,
2861 build_pointer_type (TREE_TYPE (chkp_bndmk_fndecl)),
2862 chkp_bndmk_fndecl);
2863 return build_call_nary (TREE_TYPE (TREE_TYPE (chkp_bndmk_fndecl)),
2864 call, 2, lower_bound, size);
2867 /* Create static bounds var of specfified OBJ which is
2868 is either VAR_DECL or string constant. */
2869 static tree
2870 chkp_make_static_bounds (tree obj)
2872 static int string_id = 1;
2873 static int var_id = 1;
2874 tree *slot;
2875 const char *var_name;
2876 char *bnd_var_name;
2877 tree bnd_var;
2879 /* First check if we already have required var. */
2880 if (chkp_static_var_bounds)
2882 /* For vars we use assembler name as a key in
2883 chkp_static_var_bounds map. It allows to
2884 avoid duplicating bound vars for decls
2885 sharing assembler name. */
2886 if (TREE_CODE (obj) == VAR_DECL)
2888 tree name = DECL_ASSEMBLER_NAME (obj);
2889 slot = chkp_static_var_bounds->get (name);
2890 if (slot)
2891 return *slot;
2893 else
2895 slot = chkp_static_var_bounds->get (obj);
2896 if (slot)
2897 return *slot;
2901 /* Build decl for bounds var. */
2902 if (TREE_CODE (obj) == VAR_DECL)
2904 if (DECL_IGNORED_P (obj))
2906 bnd_var_name = (char *) xmalloc (strlen (CHKP_VAR_BOUNDS_PREFIX) + 10);
2907 sprintf (bnd_var_name, "%s%d", CHKP_VAR_BOUNDS_PREFIX, var_id++);
2909 else
2911 var_name = IDENTIFIER_POINTER (DECL_ASSEMBLER_NAME (obj));
2913 /* For hidden symbols we want to skip first '*' char. */
2914 if (*var_name == '*')
2915 var_name++;
2917 bnd_var_name = (char *) xmalloc (strlen (var_name)
2918 + strlen (CHKP_BOUNDS_OF_SYMBOL_PREFIX) + 1);
2919 strcpy (bnd_var_name, CHKP_BOUNDS_OF_SYMBOL_PREFIX);
2920 strcat (bnd_var_name, var_name);
2923 bnd_var = build_decl (UNKNOWN_LOCATION, VAR_DECL,
2924 get_identifier (bnd_var_name),
2925 pointer_bounds_type_node);
2927 /* Address of the obj will be used as lower bound. */
2928 TREE_ADDRESSABLE (obj) = 1;
2930 else
2932 bnd_var_name = (char *) xmalloc (strlen (CHKP_STRING_BOUNDS_PREFIX) + 10);
2933 sprintf (bnd_var_name, "%s%d", CHKP_STRING_BOUNDS_PREFIX, string_id++);
2935 bnd_var = build_decl (UNKNOWN_LOCATION, VAR_DECL,
2936 get_identifier (bnd_var_name),
2937 pointer_bounds_type_node);
2940 TREE_PUBLIC (bnd_var) = 0;
2941 TREE_USED (bnd_var) = 1;
2942 TREE_READONLY (bnd_var) = 0;
2943 TREE_STATIC (bnd_var) = 1;
2944 TREE_ADDRESSABLE (bnd_var) = 0;
2945 DECL_ARTIFICIAL (bnd_var) = 1;
2946 DECL_COMMON (bnd_var) = 1;
2947 DECL_COMDAT (bnd_var) = 1;
2948 DECL_READ_P (bnd_var) = 1;
2949 DECL_INITIAL (bnd_var) = chkp_build_addr_expr (obj);
2950 /* Force output similar to constant bounds.
2951 See chkp_make_static_const_bounds. */
2952 varpool_node::get_create (bnd_var)->force_output = 1;
2953 /* Mark symbol as requiring bounds initialization. */
2954 varpool_node::get_create (bnd_var)->need_bounds_init = 1;
2955 varpool_node::finalize_decl (bnd_var);
2957 /* Add created var to the map to use it for other references
2958 to obj. */
2959 if (!chkp_static_var_bounds)
2960 chkp_static_var_bounds = new hash_map<tree, tree>;
2962 if (TREE_CODE (obj) == VAR_DECL)
2964 tree name = DECL_ASSEMBLER_NAME (obj);
2965 chkp_static_var_bounds->put (name, bnd_var);
2967 else
2968 chkp_static_var_bounds->put (obj, bnd_var);
2970 return bnd_var;
2973 /* When var has incomplete type we cannot get size to
2974 compute its bounds. In such cases we use checker
2975 builtin call which determines object size at runtime. */
2976 static tree
2977 chkp_generate_extern_var_bounds (tree var)
2979 tree bounds, size_reloc, lb, size, max_size, cond;
2980 gimple_stmt_iterator gsi;
2981 gimple_seq seq = NULL;
2982 gimple stmt;
2984 /* If instrumentation is not enabled for vars having
2985 incomplete type then just return zero bounds to avoid
2986 checks for this var. */
2987 if (!flag_chkp_incomplete_type)
2988 return chkp_get_zero_bounds ();
2990 if (dump_file && (dump_flags & TDF_DETAILS))
2992 fprintf (dump_file, "Generating bounds for extern symbol '");
2993 print_generic_expr (dump_file, var, 0);
2994 fprintf (dump_file, "'\n");
2997 stmt = gimple_build_call (chkp_sizeof_fndecl, 1, var);
2999 size_reloc = create_tmp_reg (chkp_uintptr_type, CHKP_SIZE_TMP_NAME);
3000 gimple_call_set_lhs (stmt, size_reloc);
3002 gimple_seq_add_stmt (&seq, stmt);
3004 lb = chkp_build_addr_expr (var);
3005 size = make_ssa_name (chkp_get_size_tmp_var ());
3007 if (flag_chkp_zero_dynamic_size_as_infinite)
3009 /* We should check that size relocation was resolved.
3010 If it was not then use maximum possible size for the var. */
3011 max_size = build2 (MINUS_EXPR, chkp_uintptr_type, integer_zero_node,
3012 fold_convert (chkp_uintptr_type, lb));
3013 max_size = chkp_force_gimple_call_op (max_size, &seq);
3015 cond = build2 (NE_EXPR, boolean_type_node,
3016 size_reloc, integer_zero_node);
3017 stmt = gimple_build_assign (size, COND_EXPR, cond, size_reloc, max_size);
3018 gimple_seq_add_stmt (&seq, stmt);
3020 else
3022 stmt = gimple_build_assign (size, size_reloc);
3023 gimple_seq_add_stmt (&seq, stmt);
3026 gsi = gsi_start_bb (chkp_get_entry_block ());
3027 gsi_insert_seq_after (&gsi, seq, GSI_CONTINUE_LINKING);
3029 bounds = chkp_make_bounds (lb, size, &gsi, true);
3031 return bounds;
3034 /* Return 1 if TYPE has fields with zero size or fields
3035 marked with chkp_variable_size attribute. */
3036 bool
3037 chkp_variable_size_type (tree type)
3039 bool res = false;
3040 tree field;
3042 if (RECORD_OR_UNION_TYPE_P (type))
3043 for (field = TYPE_FIELDS (type); field; field = DECL_CHAIN (field))
3045 if (TREE_CODE (field) == FIELD_DECL)
3046 res = res
3047 || lookup_attribute ("bnd_variable_size", DECL_ATTRIBUTES (field))
3048 || chkp_variable_size_type (TREE_TYPE (field));
3050 else
3051 res = !TYPE_SIZE (type)
3052 || TREE_CODE (TYPE_SIZE (type)) != INTEGER_CST
3053 || tree_to_uhwi (TYPE_SIZE (type)) == 0;
3055 return res;
3058 /* Compute and return bounds for address of DECL which is
3059 one of VAR_DECL, PARM_DECL, RESULT_DECL. */
3060 static tree
3061 chkp_get_bounds_for_decl_addr (tree decl)
3063 tree bounds;
3065 gcc_assert (TREE_CODE (decl) == VAR_DECL
3066 || TREE_CODE (decl) == PARM_DECL
3067 || TREE_CODE (decl) == RESULT_DECL);
3069 bounds = chkp_get_registered_addr_bounds (decl);
3071 if (bounds)
3072 return bounds;
3074 if (dump_file && (dump_flags & TDF_DETAILS))
3076 fprintf (dump_file, "Building bounds for address of decl ");
3077 print_generic_expr (dump_file, decl, 0);
3078 fprintf (dump_file, "\n");
3081 /* Use zero bounds if size is unknown and checks for
3082 unknown sizes are restricted. */
3083 if ((!DECL_SIZE (decl)
3084 || (chkp_variable_size_type (TREE_TYPE (decl))
3085 && (TREE_STATIC (decl)
3086 || DECL_EXTERNAL (decl)
3087 || TREE_PUBLIC (decl))))
3088 && !flag_chkp_incomplete_type)
3089 return chkp_get_zero_bounds ();
3091 if (flag_chkp_use_static_bounds
3092 && TREE_CODE (decl) == VAR_DECL
3093 && (TREE_STATIC (decl)
3094 || DECL_EXTERNAL (decl)
3095 || TREE_PUBLIC (decl))
3096 && !DECL_THREAD_LOCAL_P (decl))
3098 tree bnd_var = chkp_make_static_bounds (decl);
3099 gimple_stmt_iterator gsi = gsi_start_bb (chkp_get_entry_block ());
3100 gimple stmt;
3102 bounds = chkp_get_tmp_reg (NULL);
3103 stmt = gimple_build_assign (bounds, bnd_var);
3104 gsi_insert_before (&gsi, stmt, GSI_SAME_STMT);
3106 else if (!DECL_SIZE (decl)
3107 || (chkp_variable_size_type (TREE_TYPE (decl))
3108 && (TREE_STATIC (decl)
3109 || DECL_EXTERNAL (decl)
3110 || TREE_PUBLIC (decl))))
3112 gcc_assert (TREE_CODE (decl) == VAR_DECL);
3113 bounds = chkp_generate_extern_var_bounds (decl);
3115 else
3117 tree lb = chkp_build_addr_expr (decl);
3118 bounds = chkp_make_bounds (lb, DECL_SIZE_UNIT (decl), NULL, false);
3121 return bounds;
3124 /* Compute and return bounds for constant string. */
3125 static tree
3126 chkp_get_bounds_for_string_cst (tree cst)
3128 tree bounds;
3129 tree lb;
3130 tree size;
3132 gcc_assert (TREE_CODE (cst) == STRING_CST);
3134 bounds = chkp_get_registered_bounds (cst);
3136 if (bounds)
3137 return bounds;
3139 if ((flag_chkp_use_static_bounds && flag_chkp_use_static_const_bounds)
3140 || flag_chkp_use_static_const_bounds > 0)
3142 tree bnd_var = chkp_make_static_bounds (cst);
3143 gimple_stmt_iterator gsi = gsi_start_bb (chkp_get_entry_block ());
3144 gimple stmt;
3146 bounds = chkp_get_tmp_reg (NULL);
3147 stmt = gimple_build_assign (bounds, bnd_var);
3148 gsi_insert_before (&gsi, stmt, GSI_SAME_STMT);
3150 else
3152 lb = chkp_build_addr_expr (cst);
3153 size = build_int_cst (chkp_uintptr_type, TREE_STRING_LENGTH (cst));
3154 bounds = chkp_make_bounds (lb, size, NULL, false);
3157 bounds = chkp_maybe_copy_and_register_bounds (cst, bounds);
3159 return bounds;
3162 /* Generate code to instersect bounds BOUNDS1 and BOUNDS2 and
3163 return the result. if ITER is not NULL then Code is inserted
3164 before position pointed by ITER. Otherwise code is added to
3165 entry block. */
3166 static tree
3167 chkp_intersect_bounds (tree bounds1, tree bounds2, gimple_stmt_iterator *iter)
3169 if (!bounds1 || bounds1 == chkp_get_zero_bounds ())
3170 return bounds2 ? bounds2 : bounds1;
3171 else if (!bounds2 || bounds2 == chkp_get_zero_bounds ())
3172 return bounds1;
3173 else
3175 gimple_seq seq;
3176 gimple stmt;
3177 tree bounds;
3179 seq = NULL;
3181 stmt = gimple_build_call (chkp_intersect_fndecl, 2, bounds1, bounds2);
3182 chkp_mark_stmt (stmt);
3184 bounds = chkp_get_tmp_reg (stmt);
3185 gimple_call_set_lhs (stmt, bounds);
3187 gimple_seq_add_stmt (&seq, stmt);
3189 /* We are probably doing narrowing for constant expression.
3190 In such case iter may be undefined. */
3191 if (!iter)
3193 gimple_stmt_iterator gsi = gsi_last_bb (chkp_get_entry_block ());
3194 iter = &gsi;
3195 gsi_insert_seq_after (iter, seq, GSI_SAME_STMT);
3197 else
3198 gsi_insert_seq_before (iter, seq, GSI_SAME_STMT);
3200 if (dump_file && (dump_flags & TDF_DETAILS))
3202 fprintf (dump_file, "Bounds intersection: ");
3203 print_gimple_stmt (dump_file, stmt, 0, TDF_VOPS|TDF_MEMSYMS);
3204 fprintf (dump_file, " inserted before statement: ");
3205 print_gimple_stmt (dump_file, gsi_stmt (*iter), 0,
3206 TDF_VOPS|TDF_MEMSYMS);
3209 return bounds;
3213 /* Return 1 if we are allowed to narrow bounds for addressed FIELD
3214 and 0 othersize. */
3215 static bool
3216 chkp_may_narrow_to_field (tree field)
3218 return DECL_SIZE (field) && TREE_CODE (DECL_SIZE (field)) == INTEGER_CST
3219 && tree_to_uhwi (DECL_SIZE (field)) != 0
3220 && (!DECL_FIELD_OFFSET (field)
3221 || TREE_CODE (DECL_FIELD_OFFSET (field)) == INTEGER_CST)
3222 && (!DECL_FIELD_BIT_OFFSET (field)
3223 || TREE_CODE (DECL_FIELD_BIT_OFFSET (field)) == INTEGER_CST)
3224 && !lookup_attribute ("bnd_variable_size", DECL_ATTRIBUTES (field))
3225 && !chkp_variable_size_type (TREE_TYPE (field));
3228 /* Return 1 if bounds for FIELD should be narrowed to
3229 field's own size. */
3230 static bool
3231 chkp_narrow_bounds_for_field (tree field)
3233 HOST_WIDE_INT offs;
3234 HOST_WIDE_INT bit_offs;
3236 if (!chkp_may_narrow_to_field (field))
3237 return false;
3239 /* Accesse to compiler generated fields should not cause
3240 bounds narrowing. */
3241 if (DECL_ARTIFICIAL (field))
3242 return false;
3244 offs = tree_to_uhwi (DECL_FIELD_OFFSET (field));
3245 bit_offs = tree_to_uhwi (DECL_FIELD_BIT_OFFSET (field));
3247 return (flag_chkp_narrow_bounds
3248 && (flag_chkp_first_field_has_own_bounds
3249 || offs
3250 || bit_offs));
3253 /* Perform narrowing for BOUNDS using bounds computed for field
3254 access COMPONENT. ITER meaning is the same as for
3255 chkp_intersect_bounds. */
3256 static tree
3257 chkp_narrow_bounds_to_field (tree bounds, tree component,
3258 gimple_stmt_iterator *iter)
3260 tree field = TREE_OPERAND (component, 1);
3261 tree size = DECL_SIZE_UNIT (field);
3262 tree field_ptr = chkp_build_addr_expr (component);
3263 tree field_bounds;
3265 field_bounds = chkp_make_bounds (field_ptr, size, iter, false);
3267 return chkp_intersect_bounds (field_bounds, bounds, iter);
3270 /* Parse field or array access NODE.
3272 PTR ouput parameter holds a pointer to the outermost
3273 object.
3275 BITFIELD output parameter is set to 1 if bitfield is
3276 accessed and to 0 otherwise. If it is 1 then ELT holds
3277 outer component for accessed bit field.
3279 SAFE outer parameter is set to 1 if access is safe and
3280 checks are not required.
3282 BOUNDS outer parameter holds bounds to be used to check
3283 access (may be NULL).
3285 If INNERMOST_BOUNDS is 1 then try to narrow bounds to the
3286 innermost accessed component. */
3287 static void
3288 chkp_parse_array_and_component_ref (tree node, tree *ptr,
3289 tree *elt, bool *safe,
3290 bool *bitfield,
3291 tree *bounds,
3292 gimple_stmt_iterator *iter,
3293 bool innermost_bounds)
3295 tree comp_to_narrow = NULL_TREE;
3296 tree last_comp = NULL_TREE;
3297 bool array_ref_found = false;
3298 tree *nodes;
3299 tree var;
3300 int len;
3301 int i;
3303 /* Compute tree height for expression. */
3304 var = node;
3305 len = 1;
3306 while (TREE_CODE (var) == COMPONENT_REF
3307 || TREE_CODE (var) == ARRAY_REF
3308 || TREE_CODE (var) == VIEW_CONVERT_EXPR)
3310 var = TREE_OPERAND (var, 0);
3311 len++;
3314 gcc_assert (len > 1);
3316 /* It is more convenient for us to scan left-to-right,
3317 so walk tree again and put all node to nodes vector
3318 in reversed order. */
3319 nodes = XALLOCAVEC (tree, len);
3320 nodes[len - 1] = node;
3321 for (i = len - 2; i >= 0; i--)
3322 nodes[i] = TREE_OPERAND (nodes[i + 1], 0);
3324 if (bounds)
3325 *bounds = NULL;
3326 *safe = true;
3327 *bitfield = (TREE_CODE (node) == COMPONENT_REF
3328 && DECL_BIT_FIELD_TYPE (TREE_OPERAND (node, 1)));
3329 /* To get bitfield address we will need outer elemnt. */
3330 if (*bitfield)
3331 *elt = nodes[len - 2];
3332 else
3333 *elt = NULL_TREE;
3335 /* If we have indirection in expression then compute
3336 outermost structure bounds. Computed bounds may be
3337 narrowed later. */
3338 if (TREE_CODE (nodes[0]) == MEM_REF || INDIRECT_REF_P (nodes[0]))
3340 *safe = false;
3341 *ptr = TREE_OPERAND (nodes[0], 0);
3342 if (bounds)
3343 *bounds = chkp_find_bounds (*ptr, iter);
3345 else
3347 gcc_assert (TREE_CODE (var) == VAR_DECL
3348 || TREE_CODE (var) == PARM_DECL
3349 || TREE_CODE (var) == RESULT_DECL
3350 || TREE_CODE (var) == STRING_CST
3351 || TREE_CODE (var) == SSA_NAME);
3353 *ptr = chkp_build_addr_expr (var);
3356 /* In this loop we are trying to find a field access
3357 requiring narrowing. There are two simple rules
3358 for search:
3359 1. Leftmost array_ref is chosen if any.
3360 2. Rightmost suitable component_ref is chosen if innermost
3361 bounds are required and no array_ref exists. */
3362 for (i = 1; i < len; i++)
3364 var = nodes[i];
3366 if (TREE_CODE (var) == ARRAY_REF)
3368 *safe = false;
3369 array_ref_found = true;
3370 if (flag_chkp_narrow_bounds
3371 && !flag_chkp_narrow_to_innermost_arrray
3372 && (!last_comp
3373 || chkp_may_narrow_to_field (TREE_OPERAND (last_comp, 1))))
3375 comp_to_narrow = last_comp;
3376 break;
3379 else if (TREE_CODE (var) == COMPONENT_REF)
3381 tree field = TREE_OPERAND (var, 1);
3383 if (innermost_bounds
3384 && !array_ref_found
3385 && chkp_narrow_bounds_for_field (field))
3386 comp_to_narrow = var;
3387 last_comp = var;
3389 if (flag_chkp_narrow_bounds
3390 && flag_chkp_narrow_to_innermost_arrray
3391 && TREE_CODE (TREE_TYPE (field)) == ARRAY_TYPE)
3393 if (bounds)
3394 *bounds = chkp_narrow_bounds_to_field (*bounds, var, iter);
3395 comp_to_narrow = NULL;
3398 else if (TREE_CODE (var) == VIEW_CONVERT_EXPR)
3399 /* Nothing to do for it. */
3401 else
3402 gcc_unreachable ();
3405 if (comp_to_narrow && DECL_SIZE (TREE_OPERAND (comp_to_narrow, 1)) && bounds)
3406 *bounds = chkp_narrow_bounds_to_field (*bounds, comp_to_narrow, iter);
3408 if (innermost_bounds && bounds && !*bounds)
3409 *bounds = chkp_find_bounds (*ptr, iter);
3412 /* Compute and return bounds for address of OBJ. */
3413 static tree
3414 chkp_make_addressed_object_bounds (tree obj, gimple_stmt_iterator *iter)
3416 tree bounds = chkp_get_registered_addr_bounds (obj);
3418 if (bounds)
3419 return bounds;
3421 switch (TREE_CODE (obj))
3423 case VAR_DECL:
3424 case PARM_DECL:
3425 case RESULT_DECL:
3426 bounds = chkp_get_bounds_for_decl_addr (obj);
3427 break;
3429 case STRING_CST:
3430 bounds = chkp_get_bounds_for_string_cst (obj);
3431 break;
3433 case ARRAY_REF:
3434 case COMPONENT_REF:
3436 tree elt;
3437 tree ptr;
3438 bool safe;
3439 bool bitfield;
3441 chkp_parse_array_and_component_ref (obj, &ptr, &elt, &safe,
3442 &bitfield, &bounds, iter, true);
3444 gcc_assert (bounds);
3446 break;
3448 case FUNCTION_DECL:
3449 case LABEL_DECL:
3450 bounds = chkp_get_zero_bounds ();
3451 break;
3453 case MEM_REF:
3454 bounds = chkp_find_bounds (TREE_OPERAND (obj, 0), iter);
3455 break;
3457 case REALPART_EXPR:
3458 case IMAGPART_EXPR:
3459 bounds = chkp_make_addressed_object_bounds (TREE_OPERAND (obj, 0), iter);
3460 break;
3462 default:
3463 if (dump_file && (dump_flags & TDF_DETAILS))
3465 fprintf (dump_file, "chkp_make_addressed_object_bounds: "
3466 "unexpected object of type %s\n",
3467 get_tree_code_name (TREE_CODE (obj)));
3468 print_node (dump_file, "", obj, 0);
3470 internal_error ("chkp_make_addressed_object_bounds: "
3471 "Unexpected tree code %s",
3472 get_tree_code_name (TREE_CODE (obj)));
3475 chkp_register_addr_bounds (obj, bounds);
3477 return bounds;
3480 /* Compute bounds for pointer PTR loaded from PTR_SRC. Generate statements
3481 to compute bounds if required. Computed bounds should be available at
3482 position pointed by ITER.
3484 If PTR_SRC is NULL_TREE then pointer definition is identified.
3486 If PTR_SRC is not NULL_TREE then ITER points to statements which loads
3487 PTR. If PTR is a any memory reference then ITER points to a statement
3488 after which bndldx will be inserterd. In both cases ITER will be updated
3489 to point to the inserted bndldx statement. */
3491 static tree
3492 chkp_find_bounds_1 (tree ptr, tree ptr_src, gimple_stmt_iterator *iter)
3494 tree addr = NULL_TREE;
3495 tree bounds = NULL_TREE;
3497 if (!ptr_src)
3498 ptr_src = ptr;
3500 bounds = chkp_get_registered_bounds (ptr_src);
3502 if (bounds)
3503 return bounds;
3505 switch (TREE_CODE (ptr_src))
3507 case MEM_REF:
3508 case VAR_DECL:
3509 if (BOUNDED_P (ptr_src))
3510 if (TREE_CODE (ptr) == VAR_DECL && DECL_REGISTER (ptr))
3511 bounds = chkp_get_zero_bounds ();
3512 else
3514 addr = chkp_build_addr_expr (ptr_src);
3515 bounds = chkp_build_bndldx (addr, ptr, iter);
3517 else
3518 bounds = chkp_get_nonpointer_load_bounds ();
3519 break;
3521 case ARRAY_REF:
3522 case COMPONENT_REF:
3523 addr = get_base_address (ptr_src);
3524 if (DECL_P (addr)
3525 || TREE_CODE (addr) == MEM_REF
3526 || TREE_CODE (addr) == TARGET_MEM_REF)
3528 if (BOUNDED_P (ptr_src))
3529 if (TREE_CODE (ptr) == VAR_DECL && DECL_REGISTER (ptr))
3530 bounds = chkp_get_zero_bounds ();
3531 else
3533 addr = chkp_build_addr_expr (ptr_src);
3534 bounds = chkp_build_bndldx (addr, ptr, iter);
3536 else
3537 bounds = chkp_get_nonpointer_load_bounds ();
3539 else
3541 gcc_assert (TREE_CODE (addr) == SSA_NAME);
3542 bounds = chkp_find_bounds (addr, iter);
3544 break;
3546 case PARM_DECL:
3547 gcc_unreachable ();
3548 bounds = chkp_get_bound_for_parm (ptr_src);
3549 break;
3551 case TARGET_MEM_REF:
3552 addr = chkp_build_addr_expr (ptr_src);
3553 bounds = chkp_build_bndldx (addr, ptr, iter);
3554 break;
3556 case SSA_NAME:
3557 bounds = chkp_get_registered_bounds (ptr_src);
3558 if (!bounds)
3560 gimple def_stmt = SSA_NAME_DEF_STMT (ptr_src);
3561 gphi_iterator phi_iter;
3563 bounds = chkp_get_bounds_by_definition (ptr_src, def_stmt, &phi_iter);
3565 gcc_assert (bounds);
3567 if (gphi *def_phi = dyn_cast <gphi *> (def_stmt))
3569 unsigned i;
3571 for (i = 0; i < gimple_phi_num_args (def_phi); i++)
3573 tree arg = gimple_phi_arg_def (def_phi, i);
3574 tree arg_bnd;
3575 gphi *phi_bnd;
3577 arg_bnd = chkp_find_bounds (arg, NULL);
3579 /* chkp_get_bounds_by_definition created new phi
3580 statement and phi_iter points to it.
3582 Previous call to chkp_find_bounds could create
3583 new basic block and therefore change phi statement
3584 phi_iter points to. */
3585 phi_bnd = phi_iter.phi ();
3587 add_phi_arg (phi_bnd, arg_bnd,
3588 gimple_phi_arg_edge (def_phi, i),
3589 UNKNOWN_LOCATION);
3592 /* If all bound phi nodes have their arg computed
3593 then we may finish its computation. See
3594 chkp_finish_incomplete_bounds for more details. */
3595 if (chkp_may_finish_incomplete_bounds ())
3596 chkp_finish_incomplete_bounds ();
3599 gcc_assert (bounds == chkp_get_registered_bounds (ptr_src)
3600 || chkp_incomplete_bounds (bounds));
3602 break;
3604 case ADDR_EXPR:
3605 bounds = chkp_make_addressed_object_bounds (TREE_OPERAND (ptr_src, 0), iter);
3606 break;
3608 case INTEGER_CST:
3609 if (integer_zerop (ptr_src))
3610 bounds = chkp_get_none_bounds ();
3611 else
3612 bounds = chkp_get_invalid_op_bounds ();
3613 break;
3615 default:
3616 if (dump_file && (dump_flags & TDF_DETAILS))
3618 fprintf (dump_file, "chkp_find_bounds: unexpected ptr of type %s\n",
3619 get_tree_code_name (TREE_CODE (ptr_src)));
3620 print_node (dump_file, "", ptr_src, 0);
3622 internal_error ("chkp_find_bounds: Unexpected tree code %s",
3623 get_tree_code_name (TREE_CODE (ptr_src)));
3626 if (!bounds)
3628 if (dump_file && (dump_flags & TDF_DETAILS))
3630 fprintf (stderr, "chkp_find_bounds: cannot find bounds for pointer\n");
3631 print_node (dump_file, "", ptr_src, 0);
3633 internal_error ("chkp_find_bounds: Cannot find bounds for pointer");
3636 return bounds;
3639 /* Normal case for bounds search without forced narrowing. */
3640 static tree
3641 chkp_find_bounds (tree ptr, gimple_stmt_iterator *iter)
3643 return chkp_find_bounds_1 (ptr, NULL_TREE, iter);
3646 /* Search bounds for pointer PTR loaded from PTR_SRC
3647 by statement *ITER points to. */
3648 static tree
3649 chkp_find_bounds_loaded (tree ptr, tree ptr_src, gimple_stmt_iterator *iter)
3651 return chkp_find_bounds_1 (ptr, ptr_src, iter);
3654 /* Helper function which checks type of RHS and finds all pointers in
3655 it. For each found pointer we build it's accesses in LHS and RHS
3656 objects and then call HANDLER for them. Function is used to copy
3657 or initilize bounds for copied object. */
3658 static void
3659 chkp_walk_pointer_assignments (tree lhs, tree rhs, void *arg,
3660 assign_handler handler)
3662 tree type = TREE_TYPE (lhs);
3664 /* We have nothing to do with clobbers. */
3665 if (TREE_CLOBBER_P (rhs))
3666 return;
3668 if (BOUNDED_TYPE_P (type))
3669 handler (lhs, rhs, arg);
3670 else if (RECORD_OR_UNION_TYPE_P (type))
3672 tree field;
3674 if (TREE_CODE (rhs) == CONSTRUCTOR)
3676 unsigned HOST_WIDE_INT cnt;
3677 tree val;
3679 FOR_EACH_CONSTRUCTOR_ELT (CONSTRUCTOR_ELTS (rhs), cnt, field, val)
3681 if (chkp_type_has_pointer (TREE_TYPE (field)))
3683 tree lhs_field = chkp_build_component_ref (lhs, field);
3684 chkp_walk_pointer_assignments (lhs_field, val, arg, handler);
3688 else
3689 for (field = TYPE_FIELDS (type); field; field = DECL_CHAIN (field))
3690 if (TREE_CODE (field) == FIELD_DECL
3691 && chkp_type_has_pointer (TREE_TYPE (field)))
3693 tree rhs_field = chkp_build_component_ref (rhs, field);
3694 tree lhs_field = chkp_build_component_ref (lhs, field);
3695 chkp_walk_pointer_assignments (lhs_field, rhs_field, arg, handler);
3698 else if (TREE_CODE (type) == ARRAY_TYPE)
3700 unsigned HOST_WIDE_INT cur = 0;
3701 tree maxval = TYPE_MAX_VALUE (TYPE_DOMAIN (type));
3702 tree etype = TREE_TYPE (type);
3703 tree esize = TYPE_SIZE (etype);
3705 if (TREE_CODE (rhs) == CONSTRUCTOR)
3707 unsigned HOST_WIDE_INT cnt;
3708 tree purp, val, lhs_elem;
3710 FOR_EACH_CONSTRUCTOR_ELT (CONSTRUCTOR_ELTS (rhs), cnt, purp, val)
3712 if (purp && TREE_CODE (purp) == RANGE_EXPR)
3714 tree lo_index = TREE_OPERAND (purp, 0);
3715 tree hi_index = TREE_OPERAND (purp, 1);
3717 for (cur = (unsigned)tree_to_uhwi (lo_index);
3718 cur <= (unsigned)tree_to_uhwi (hi_index);
3719 cur++)
3721 lhs_elem = chkp_build_array_ref (lhs, etype, esize, cur);
3722 chkp_walk_pointer_assignments (lhs_elem, val, arg, handler);
3725 else
3727 if (purp)
3729 gcc_assert (TREE_CODE (purp) == INTEGER_CST);
3730 cur = tree_to_uhwi (purp);
3733 lhs_elem = chkp_build_array_ref (lhs, etype, esize, cur++);
3735 chkp_walk_pointer_assignments (lhs_elem, val, arg, handler);
3739 /* Copy array only when size is known. */
3740 else if (maxval && !integer_minus_onep (maxval))
3741 for (cur = 0; cur <= TREE_INT_CST_LOW (maxval); cur++)
3743 tree lhs_elem = chkp_build_array_ref (lhs, etype, esize, cur);
3744 tree rhs_elem = chkp_build_array_ref (rhs, etype, esize, cur);
3745 chkp_walk_pointer_assignments (lhs_elem, rhs_elem, arg, handler);
3748 else
3749 internal_error("chkp_walk_pointer_assignments: unexpected RHS type: %s",
3750 get_tree_code_name (TREE_CODE (type)));
3753 /* Add code to copy bounds for assignment of RHS to LHS.
3754 ARG is an iterator pointing ne code position. */
3755 static void
3756 chkp_copy_bounds_for_elem (tree lhs, tree rhs, void *arg)
3758 gimple_stmt_iterator *iter = (gimple_stmt_iterator *)arg;
3759 tree bounds = chkp_find_bounds (rhs, iter);
3760 tree addr = chkp_build_addr_expr(lhs);
3762 chkp_build_bndstx (addr, rhs, bounds, iter);
3765 /* Emit static bound initilizers and size vars. */
3766 void
3767 chkp_finish_file (void)
3769 struct varpool_node *node;
3770 struct chkp_ctor_stmt_list stmts;
3772 if (seen_error ())
3773 return;
3775 /* Iterate through varpool and generate bounds initialization
3776 constructors for all statically initialized pointers. */
3777 stmts.avail = MAX_STMTS_IN_STATIC_CHKP_CTOR;
3778 stmts.stmts = NULL;
3779 FOR_EACH_VARIABLE (node)
3780 /* Check that var is actually emitted and we need and may initialize
3781 its bounds. */
3782 if (node->need_bounds_init
3783 && !POINTER_BOUNDS_P (node->decl)
3784 && DECL_RTL (node->decl)
3785 && MEM_P (DECL_RTL (node->decl))
3786 && TREE_ASM_WRITTEN (node->decl))
3788 chkp_walk_pointer_assignments (node->decl,
3789 DECL_INITIAL (node->decl),
3790 &stmts,
3791 chkp_add_modification_to_stmt_list);
3793 if (stmts.avail <= 0)
3795 cgraph_build_static_cdtor ('P', stmts.stmts,
3796 MAX_RESERVED_INIT_PRIORITY + 3);
3797 stmts.avail = MAX_STMTS_IN_STATIC_CHKP_CTOR;
3798 stmts.stmts = NULL;
3802 if (stmts.stmts)
3803 cgraph_build_static_cdtor ('P', stmts.stmts,
3804 MAX_RESERVED_INIT_PRIORITY + 3);
3806 /* Iterate through varpool and generate bounds initialization
3807 constructors for all static bounds vars. */
3808 stmts.avail = MAX_STMTS_IN_STATIC_CHKP_CTOR;
3809 stmts.stmts = NULL;
3810 FOR_EACH_VARIABLE (node)
3811 if (node->need_bounds_init
3812 && POINTER_BOUNDS_P (node->decl)
3813 && TREE_ASM_WRITTEN (node->decl))
3815 tree bnd = node->decl;
3816 tree var;
3818 gcc_assert (DECL_INITIAL (bnd)
3819 && TREE_CODE (DECL_INITIAL (bnd)) == ADDR_EXPR);
3821 var = TREE_OPERAND (DECL_INITIAL (bnd), 0);
3822 chkp_output_static_bounds (bnd, var, &stmts);
3825 if (stmts.stmts)
3826 cgraph_build_static_cdtor ('B', stmts.stmts,
3827 MAX_RESERVED_INIT_PRIORITY + 2);
3829 delete chkp_static_var_bounds;
3830 delete chkp_bounds_map;
3833 /* An instrumentation function which is called for each statement
3834 having memory access we want to instrument. It inserts check
3835 code and bounds copy code.
3837 ITER points to statement to instrument.
3839 NODE holds memory access in statement to check.
3841 LOC holds the location information for statement.
3843 DIRFLAGS determines whether access is read or write.
3845 ACCESS_OFFS should be added to address used in NODE
3846 before check.
3848 ACCESS_SIZE holds size of checked access.
3850 SAFE indicates if NODE access is safe and should not be
3851 checked. */
3852 static void
3853 chkp_process_stmt (gimple_stmt_iterator *iter, tree node,
3854 location_t loc, tree dirflag,
3855 tree access_offs, tree access_size,
3856 bool safe)
3858 tree node_type = TREE_TYPE (node);
3859 tree size = access_size ? access_size : TYPE_SIZE_UNIT (node_type);
3860 tree addr_first = NULL_TREE; /* address of the first accessed byte */
3861 tree addr_last = NULL_TREE; /* address of the last accessed byte */
3862 tree ptr = NULL_TREE; /* a pointer used for dereference */
3863 tree bounds = NULL_TREE;
3865 /* We do not need instrumentation for clobbers. */
3866 if (dirflag == integer_one_node
3867 && gimple_code (gsi_stmt (*iter)) == GIMPLE_ASSIGN
3868 && TREE_CLOBBER_P (gimple_assign_rhs1 (gsi_stmt (*iter))))
3869 return;
3871 switch (TREE_CODE (node))
3873 case ARRAY_REF:
3874 case COMPONENT_REF:
3876 bool bitfield;
3877 tree elt;
3879 if (safe)
3881 /* We are not going to generate any checks, so do not
3882 generate bounds as well. */
3883 addr_first = chkp_build_addr_expr (node);
3884 break;
3887 chkp_parse_array_and_component_ref (node, &ptr, &elt, &safe,
3888 &bitfield, &bounds, iter, false);
3890 /* Break if there is no dereference and operation is safe. */
3892 if (bitfield)
3894 tree field = TREE_OPERAND (node, 1);
3896 if (TREE_CODE (DECL_SIZE_UNIT (field)) == INTEGER_CST)
3897 size = DECL_SIZE_UNIT (field);
3899 if (elt)
3900 elt = chkp_build_addr_expr (elt);
3901 addr_first = fold_convert_loc (loc, ptr_type_node, elt ? elt : ptr);
3902 addr_first = fold_build_pointer_plus_loc (loc,
3903 addr_first,
3904 byte_position (field));
3906 else
3907 addr_first = chkp_build_addr_expr (node);
3909 break;
3911 case INDIRECT_REF:
3912 ptr = TREE_OPERAND (node, 0);
3913 addr_first = ptr;
3914 break;
3916 case MEM_REF:
3917 ptr = TREE_OPERAND (node, 0);
3918 addr_first = chkp_build_addr_expr (node);
3919 break;
3921 case TARGET_MEM_REF:
3922 ptr = TMR_BASE (node);
3923 addr_first = chkp_build_addr_expr (node);
3924 break;
3926 case ARRAY_RANGE_REF:
3927 printf("ARRAY_RANGE_REF\n");
3928 debug_gimple_stmt(gsi_stmt(*iter));
3929 debug_tree(node);
3930 gcc_unreachable ();
3931 break;
3933 case BIT_FIELD_REF:
3935 tree offs, rem, bpu;
3937 gcc_assert (!access_offs);
3938 gcc_assert (!access_size);
3940 bpu = fold_convert (size_type_node, bitsize_int (BITS_PER_UNIT));
3941 offs = fold_convert (size_type_node, TREE_OPERAND (node, 2));
3942 rem = size_binop_loc (loc, TRUNC_MOD_EXPR, offs, bpu);
3943 offs = size_binop_loc (loc, TRUNC_DIV_EXPR, offs, bpu);
3945 size = fold_convert (size_type_node, TREE_OPERAND (node, 1));
3946 size = size_binop_loc (loc, PLUS_EXPR, size, rem);
3947 size = size_binop_loc (loc, CEIL_DIV_EXPR, size, bpu);
3948 size = fold_convert (size_type_node, size);
3950 chkp_process_stmt (iter, TREE_OPERAND (node, 0), loc,
3951 dirflag, offs, size, safe);
3952 return;
3954 break;
3956 case VAR_DECL:
3957 case RESULT_DECL:
3958 case PARM_DECL:
3959 if (dirflag != integer_one_node
3960 || DECL_REGISTER (node))
3961 return;
3963 safe = true;
3964 addr_first = chkp_build_addr_expr (node);
3965 break;
3967 default:
3968 return;
3971 /* If addr_last was not computed then use (addr_first + size - 1)
3972 expression to compute it. */
3973 if (!addr_last)
3975 addr_last = fold_build_pointer_plus_loc (loc, addr_first, size);
3976 addr_last = fold_build_pointer_plus_hwi_loc (loc, addr_last, -1);
3979 /* Shift both first_addr and last_addr by access_offs if specified. */
3980 if (access_offs)
3982 addr_first = fold_build_pointer_plus_loc (loc, addr_first, access_offs);
3983 addr_last = fold_build_pointer_plus_loc (loc, addr_last, access_offs);
3986 /* Generate bndcl/bndcu checks if memory access is not safe. */
3987 if (!safe)
3989 gimple_stmt_iterator stmt_iter = *iter;
3991 if (!bounds)
3992 bounds = chkp_find_bounds (ptr, iter);
3994 chkp_check_mem_access (addr_first, addr_last, bounds,
3995 stmt_iter, loc, dirflag);
3998 /* We need to store bounds in case pointer is stored. */
3999 if (dirflag == integer_one_node
4000 && chkp_type_has_pointer (node_type)
4001 && flag_chkp_store_bounds)
4003 gimple stmt = gsi_stmt (*iter);
4004 tree rhs1 = gimple_assign_rhs1 (stmt);
4005 enum tree_code rhs_code = gimple_assign_rhs_code (stmt);
4007 if (get_gimple_rhs_class (rhs_code) == GIMPLE_SINGLE_RHS)
4008 chkp_walk_pointer_assignments (node, rhs1, iter,
4009 chkp_copy_bounds_for_elem);
4010 else
4012 bounds = chkp_compute_bounds_for_assignment (NULL_TREE, stmt);
4013 chkp_build_bndstx (addr_first, rhs1, bounds, iter);
4018 /* Add code to copy bounds for all pointers copied
4019 in ASSIGN created during inline of EDGE. */
4020 void
4021 chkp_copy_bounds_for_assign (gimple assign, struct cgraph_edge *edge)
4023 tree lhs = gimple_assign_lhs (assign);
4024 tree rhs = gimple_assign_rhs1 (assign);
4025 gimple_stmt_iterator iter = gsi_for_stmt (assign);
4027 if (!flag_chkp_store_bounds)
4028 return;
4030 chkp_walk_pointer_assignments (lhs, rhs, &iter, chkp_copy_bounds_for_elem);
4032 /* We should create edges for all created calls to bndldx and bndstx. */
4033 while (gsi_stmt (iter) != assign)
4035 gimple stmt = gsi_stmt (iter);
4036 if (gimple_code (stmt) == GIMPLE_CALL)
4038 tree fndecl = gimple_call_fndecl (stmt);
4039 struct cgraph_node *callee = cgraph_node::get_create (fndecl);
4040 struct cgraph_edge *new_edge;
4042 gcc_assert (fndecl == chkp_bndstx_fndecl
4043 || fndecl == chkp_bndldx_fndecl
4044 || fndecl == chkp_ret_bnd_fndecl);
4046 new_edge = edge->caller->create_edge (callee,
4047 as_a <gcall *> (stmt),
4048 edge->count,
4049 edge->frequency);
4050 new_edge->frequency = compute_call_stmt_bb_frequency
4051 (edge->caller->decl, gimple_bb (stmt));
4053 gsi_prev (&iter);
4057 /* Some code transformation made during instrumentation pass
4058 may put code into inconsistent state. Here we find and fix
4059 such flaws. */
4060 void
4061 chkp_fix_cfg ()
4063 basic_block bb;
4064 gimple_stmt_iterator i;
4066 /* We could insert some code right after stmt which ends bb.
4067 We wanted to put this code on fallthru edge but did not
4068 add new edges from the beginning because it may cause new
4069 phi node creation which may be incorrect due to incomplete
4070 bound phi nodes. */
4071 FOR_ALL_BB_FN (bb, cfun)
4072 for (i = gsi_start_bb (bb); !gsi_end_p (i); gsi_next (&i))
4074 gimple stmt = gsi_stmt (i);
4075 gimple_stmt_iterator next = i;
4077 gsi_next (&next);
4079 if (stmt_ends_bb_p (stmt)
4080 && !gsi_end_p (next))
4082 edge fall = find_fallthru_edge (bb->succs);
4083 basic_block dest = NULL;
4084 int flags = 0;
4086 gcc_assert (fall);
4088 /* We cannot split abnormal edge. Therefore we
4089 store its params, make it regular and then
4090 rebuild abnormal edge after split. */
4091 if (fall->flags & EDGE_ABNORMAL)
4093 flags = fall->flags & ~EDGE_FALLTHRU;
4094 dest = fall->dest;
4096 fall->flags &= ~EDGE_COMPLEX;
4099 while (!gsi_end_p (next))
4101 gimple next_stmt = gsi_stmt (next);
4102 gsi_remove (&next, false);
4103 gsi_insert_on_edge (fall, next_stmt);
4106 gsi_commit_edge_inserts ();
4108 /* Re-create abnormal edge. */
4109 if (dest)
4110 make_edge (bb, dest, flags);
4115 /* Walker callback for chkp_replace_function_pointers. Replaces
4116 function pointer in the specified operand with pointer to the
4117 instrumented function version. */
4118 static tree
4119 chkp_replace_function_pointer (tree *op, int *walk_subtrees,
4120 void *data ATTRIBUTE_UNUSED)
4122 if (TREE_CODE (*op) == FUNCTION_DECL
4123 && chkp_instrumentable_p (*op)
4124 && (DECL_BUILT_IN_CLASS (*op) == NOT_BUILT_IN
4125 /* For builtins we replace pointers only for selected
4126 function and functions having definitions. */
4127 || (DECL_BUILT_IN_CLASS (*op) == BUILT_IN_NORMAL
4128 && (chkp_instrument_normal_builtin (*op)
4129 || gimple_has_body_p (*op)))))
4131 struct cgraph_node *node = cgraph_node::get_create (*op);
4132 struct cgraph_node *clone = NULL;
4134 if (!node->instrumentation_clone)
4135 clone = chkp_maybe_create_clone (*op);
4137 if (clone)
4138 *op = clone->decl;
4139 *walk_subtrees = 0;
4142 return NULL;
4145 /* This function searches for function pointers in statement
4146 pointed by GSI and replaces them with pointers to instrumented
4147 function versions. */
4148 static void
4149 chkp_replace_function_pointers (gimple_stmt_iterator *gsi)
4151 gimple stmt = gsi_stmt (*gsi);
4152 /* For calls we want to walk call args only. */
4153 if (gimple_code (stmt) == GIMPLE_CALL)
4155 unsigned i;
4156 for (i = 0; i < gimple_call_num_args (stmt); i++)
4157 walk_tree (gimple_call_arg_ptr (stmt, i),
4158 chkp_replace_function_pointer, NULL, NULL);
4160 else
4161 walk_gimple_stmt (gsi, NULL, chkp_replace_function_pointer, NULL);
4164 /* This function instruments all statements working with memory,
4165 calls and rets.
4167 It also removes excess statements from static initializers. */
4168 static void
4169 chkp_instrument_function (void)
4171 basic_block bb, next;
4172 gimple_stmt_iterator i;
4173 enum gimple_rhs_class grhs_class;
4174 bool safe = lookup_attribute ("chkp ctor", DECL_ATTRIBUTES (cfun->decl));
4176 bb = ENTRY_BLOCK_PTR_FOR_FN (cfun)->next_bb;
4179 next = bb->next_bb;
4180 for (i = gsi_start_bb (bb); !gsi_end_p (i); )
4182 gimple s = gsi_stmt (i);
4184 /* Skip statement marked to not be instrumented. */
4185 if (chkp_marked_stmt_p (s))
4187 gsi_next (&i);
4188 continue;
4191 chkp_replace_function_pointers (&i);
4193 switch (gimple_code (s))
4195 case GIMPLE_ASSIGN:
4196 chkp_process_stmt (&i, gimple_assign_lhs (s),
4197 gimple_location (s), integer_one_node,
4198 NULL_TREE, NULL_TREE, safe);
4199 chkp_process_stmt (&i, gimple_assign_rhs1 (s),
4200 gimple_location (s), integer_zero_node,
4201 NULL_TREE, NULL_TREE, safe);
4202 grhs_class = get_gimple_rhs_class (gimple_assign_rhs_code (s));
4203 if (grhs_class == GIMPLE_BINARY_RHS)
4204 chkp_process_stmt (&i, gimple_assign_rhs2 (s),
4205 gimple_location (s), integer_zero_node,
4206 NULL_TREE, NULL_TREE, safe);
4207 break;
4209 case GIMPLE_RETURN:
4211 greturn *r = as_a <greturn *> (s);
4212 if (gimple_return_retval (r) != NULL_TREE)
4214 chkp_process_stmt (&i, gimple_return_retval (r),
4215 gimple_location (r),
4216 integer_zero_node,
4217 NULL_TREE, NULL_TREE, safe);
4219 /* Additionally we need to add bounds
4220 to return statement. */
4221 chkp_add_bounds_to_ret_stmt (&i);
4224 break;
4226 case GIMPLE_CALL:
4227 chkp_add_bounds_to_call_stmt (&i);
4228 break;
4230 default:
4234 gsi_next (&i);
4236 /* We do not need any actual pointer stores in checker
4237 static initializer. */
4238 if (lookup_attribute ("chkp ctor", DECL_ATTRIBUTES (cfun->decl))
4239 && gimple_code (s) == GIMPLE_ASSIGN
4240 && gimple_store_p (s))
4242 gimple_stmt_iterator del_iter = gsi_for_stmt (s);
4243 gsi_remove (&del_iter, true);
4244 unlink_stmt_vdef (s);
4245 release_defs(s);
4248 bb = next;
4250 while (bb);
4252 /* Some input params may have bounds and be address taken. In this case
4253 we should store incoming bounds into bounds table. */
4254 tree arg;
4255 if (flag_chkp_store_bounds)
4256 for (arg = DECL_ARGUMENTS (cfun->decl); arg; arg = DECL_CHAIN (arg))
4257 if (TREE_ADDRESSABLE (arg))
4259 if (BOUNDED_P (arg))
4261 tree bounds = chkp_get_next_bounds_parm (arg);
4262 tree def_ptr = ssa_default_def (cfun, arg);
4263 gimple_stmt_iterator iter
4264 = gsi_start_bb (chkp_get_entry_block ());
4265 chkp_build_bndstx (chkp_build_addr_expr (arg),
4266 def_ptr ? def_ptr : arg,
4267 bounds, &iter);
4269 /* Skip bounds arg. */
4270 arg = TREE_CHAIN (arg);
4272 else if (chkp_type_has_pointer (TREE_TYPE (arg)))
4274 tree orig_arg = arg;
4275 bitmap slots = BITMAP_ALLOC (NULL);
4276 gimple_stmt_iterator iter
4277 = gsi_start_bb (chkp_get_entry_block ());
4278 bitmap_iterator bi;
4279 unsigned bnd_no;
4281 chkp_find_bound_slots (TREE_TYPE (arg), slots);
4283 EXECUTE_IF_SET_IN_BITMAP (slots, 0, bnd_no, bi)
4285 tree bounds = chkp_get_next_bounds_parm (arg);
4286 HOST_WIDE_INT offs = bnd_no * POINTER_SIZE / BITS_PER_UNIT;
4287 tree addr = chkp_build_addr_expr (orig_arg);
4288 tree ptr = build2 (MEM_REF, ptr_type_node, addr,
4289 build_int_cst (ptr_type_node, offs));
4290 chkp_build_bndstx (chkp_build_addr_expr (ptr), ptr,
4291 bounds, &iter);
4293 arg = DECL_CHAIN (arg);
4295 BITMAP_FREE (slots);
4300 /* Find init/null/copy_ptr_bounds calls and replace them
4301 with assignments. It should allow better code
4302 optimization. */
4304 static void
4305 chkp_remove_useless_builtins ()
4307 basic_block bb;
4308 gimple_stmt_iterator gsi;
4310 FOR_EACH_BB_FN (bb, cfun)
4312 for (gsi = gsi_start_bb (bb); !gsi_end_p (gsi); gsi_next (&gsi))
4314 gimple stmt = gsi_stmt (gsi);
4315 tree fndecl;
4316 enum built_in_function fcode;
4318 /* Find builtins returning first arg and replace
4319 them with assignments. */
4320 if (gimple_code (stmt) == GIMPLE_CALL
4321 && (fndecl = gimple_call_fndecl (stmt))
4322 && DECL_BUILT_IN_CLASS (fndecl) == BUILT_IN_NORMAL
4323 && (fcode = DECL_FUNCTION_CODE (fndecl))
4324 && (fcode == BUILT_IN_CHKP_INIT_PTR_BOUNDS
4325 || fcode == BUILT_IN_CHKP_NULL_PTR_BOUNDS
4326 || fcode == BUILT_IN_CHKP_COPY_PTR_BOUNDS
4327 || fcode == BUILT_IN_CHKP_SET_PTR_BOUNDS))
4329 tree res = gimple_call_arg (stmt, 0);
4330 update_call_from_tree (&gsi, res);
4331 stmt = gsi_stmt (gsi);
4332 update_stmt (stmt);
4338 /* Initialize pass. */
4339 static void
4340 chkp_init (void)
4342 basic_block bb;
4343 gimple_stmt_iterator i;
4345 in_chkp_pass = true;
4347 for (bb = ENTRY_BLOCK_PTR_FOR_FN (cfun)->next_bb; bb; bb = bb->next_bb)
4348 for (i = gsi_start_bb (bb); !gsi_end_p (i); gsi_next (&i))
4349 chkp_unmark_stmt (gsi_stmt (i));
4351 chkp_invalid_bounds = new hash_set<tree>;
4352 chkp_completed_bounds_set = new hash_set<tree>;
4353 delete chkp_reg_bounds;
4354 chkp_reg_bounds = new hash_map<tree, tree>;
4355 delete chkp_bound_vars;
4356 chkp_bound_vars = new hash_map<tree, tree>;
4357 chkp_reg_addr_bounds = new hash_map<tree, tree>;
4358 chkp_incomplete_bounds_map = new hash_map<tree, tree>;
4359 delete chkp_bounds_map;
4360 chkp_bounds_map = new hash_map<tree, tree>;
4361 chkp_abnormal_copies = BITMAP_GGC_ALLOC ();
4363 entry_block = NULL;
4364 zero_bounds = NULL_TREE;
4365 none_bounds = NULL_TREE;
4366 incomplete_bounds = integer_zero_node;
4367 tmp_var = NULL_TREE;
4368 size_tmp_var = NULL_TREE;
4370 chkp_uintptr_type = lang_hooks.types.type_for_mode (ptr_mode, true);
4372 /* We create these constant bounds once for each object file.
4373 These symbols go to comdat section and result in single copy
4374 of each one in the final binary. */
4375 chkp_get_zero_bounds_var ();
4376 chkp_get_none_bounds_var ();
4378 calculate_dominance_info (CDI_DOMINATORS);
4379 calculate_dominance_info (CDI_POST_DOMINATORS);
4381 bitmap_obstack_initialize (NULL);
4384 /* Finalize instrumentation pass. */
4385 static void
4386 chkp_fini (void)
4388 in_chkp_pass = false;
4390 delete chkp_invalid_bounds;
4391 delete chkp_completed_bounds_set;
4392 delete chkp_reg_addr_bounds;
4393 delete chkp_incomplete_bounds_map;
4395 free_dominance_info (CDI_DOMINATORS);
4396 free_dominance_info (CDI_POST_DOMINATORS);
4398 bitmap_obstack_release (NULL);
4400 entry_block = NULL;
4401 zero_bounds = NULL_TREE;
4402 none_bounds = NULL_TREE;
4405 /* Main instrumentation pass function. */
4406 static unsigned int
4407 chkp_execute (void)
4409 chkp_init ();
4411 chkp_instrument_function ();
4413 chkp_remove_useless_builtins ();
4415 chkp_function_mark_instrumented (cfun->decl);
4417 chkp_fix_cfg ();
4419 chkp_fini ();
4421 return 0;
4424 /* Instrumentation pass gate. */
4425 static bool
4426 chkp_gate (void)
4428 cgraph_node *node = cgraph_node::get (cfun->decl);
4429 return ((node != NULL
4430 && node->instrumentation_clone)
4431 || lookup_attribute ("chkp ctor", DECL_ATTRIBUTES (cfun->decl)));
4434 namespace {
4436 const pass_data pass_data_chkp =
4438 GIMPLE_PASS, /* type */
4439 "chkp", /* name */
4440 OPTGROUP_NONE, /* optinfo_flags */
4441 TV_NONE, /* tv_id */
4442 PROP_ssa | PROP_cfg, /* properties_required */
4443 0, /* properties_provided */
4444 0, /* properties_destroyed */
4445 0, /* todo_flags_start */
4446 TODO_verify_il
4447 | TODO_update_ssa /* todo_flags_finish */
4450 class pass_chkp : public gimple_opt_pass
4452 public:
4453 pass_chkp (gcc::context *ctxt)
4454 : gimple_opt_pass (pass_data_chkp, ctxt)
4457 /* opt_pass methods: */
4458 virtual opt_pass * clone ()
4460 return new pass_chkp (m_ctxt);
4463 virtual bool gate (function *)
4465 return chkp_gate ();
4468 virtual unsigned int execute (function *)
4470 return chkp_execute ();
4473 }; // class pass_chkp
4475 } // anon namespace
4477 gimple_opt_pass *
4478 make_pass_chkp (gcc::context *ctxt)
4480 return new pass_chkp (ctxt);
4483 #include "gt-tree-chkp.h"