type_val: record fewer "unknown" assignments
[smatch.git] / smatch_states.c
blob21d42020fe1a29207c461dd071b0b9c69ade1deb
1 /*
2 * Copyright (C) 2006 Dan Carpenter.
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License
6 * as published by the Free Software Foundation; either version 2
7 * of the License, or (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, see http://www.gnu.org/copyleft/gpl.txt
19 * You have a lists of states. kernel = locked, foo = NULL, ...
20 * When you hit an if {} else {} statement then you swap the list
21 * of states for a different list of states. The lists are stored
22 * on stacks.
24 * At the beginning of this file there are list of the stacks that
25 * we use. Each function in this file does something to one of
26 * of the stacks.
28 * So the smatch_flow.c understands code but it doesn't understand states.
29 * smatch_flow calls functions in this file. This file calls functions
30 * in smatch_slist.c which just has boring generic plumbing for handling
31 * state lists. But really it's this file where all the magic happens.
34 #include <stdlib.h>
35 #include <stdio.h>
36 #include "smatch.h"
37 #include "smatch_slist.h"
38 #include "smatch_extra.h"
40 struct smatch_state undefined = { .name = "undefined" };
41 struct smatch_state ghost = { .name = "ghost" };
42 struct smatch_state merged = { .name = "merged" };
43 struct smatch_state true_state = { .name = "true" };
44 struct smatch_state false_state = { .name = "false" };
46 static struct stree *cur_stree; /* current states */
48 static struct stree_stack *true_stack; /* states after a t/f branch */
49 static struct stree_stack *false_stack;
50 static struct stree_stack *pre_cond_stack; /* states before a t/f branch */
52 static struct stree_stack *cond_true_stack; /* states affected by a branch */
53 static struct stree_stack *cond_false_stack;
55 static struct stree_stack *fake_cur_stree_stack;
56 static int read_only;
58 static struct stree_stack *break_stack;
59 static struct stree_stack *fake_break_stack;
60 static struct stree_stack *switch_stack;
61 static struct range_list_stack *remaining_cases;
62 static struct stree_stack *default_stack;
63 static struct stree_stack *continue_stack;
65 static struct named_stree_stack *goto_stack;
67 static struct ptr_list *backup;
69 int option_debug;
71 void __print_cur_stree(void)
73 __print_stree(cur_stree);
76 int unreachable(void)
78 if (!cur_stree)
79 return 1;
80 return 0;
83 struct sm_state *set_state(int owner, const char *name, struct symbol *sym, struct smatch_state *state)
85 struct sm_state *ret;
87 if (!name)
88 return NULL;
90 if (read_only)
91 sm_msg("Smatch Internal Error: cur_stree is read only.");
93 if (option_debug || strcmp(check_name(owner), option_debug_check) == 0) {
94 struct smatch_state *s;
96 s = get_state(owner, name, sym);
97 if (!s)
98 sm_msg("%s new [%s] '%s' %s", __func__,
99 check_name(owner), name, show_state(state));
100 else
101 sm_msg("%s change [%s] '%s' %s => %s",
102 __func__, check_name(owner), name, show_state(s),
103 show_state(state));
106 if (owner != -1 && unreachable())
107 return NULL;
109 if (fake_cur_stree_stack)
110 set_state_stree_stack(&fake_cur_stree_stack, owner, name, sym, state);
112 ret = set_state_stree(&cur_stree, owner, name, sym, state);
114 return ret;
117 struct sm_state *set_state_expr(int owner, struct expression *expr, struct smatch_state *state)
119 char *name;
120 struct symbol *sym;
121 struct sm_state *ret = NULL;
123 expr = strip_expr(expr);
124 name = expr_to_var_sym(expr, &sym);
125 if (!name || !sym)
126 goto free;
127 ret = set_state(owner, name, sym, state);
128 free:
129 free_string(name);
130 return ret;
133 void __swap_cur_stree(struct stree *stree)
135 free_stree(&cur_stree);
136 cur_stree = stree;
139 void __push_fake_cur_stree(void)
141 push_stree(&fake_cur_stree_stack, NULL);
142 __save_pre_cond_states();
145 struct stree *__pop_fake_cur_stree(void)
147 if (!fake_cur_stree_stack)
148 sm_msg("INTERNAL error: popping too many fake cur strees.");
149 __use_pre_cond_states();
150 return pop_stree(&fake_cur_stree_stack);
153 void __free_fake_cur_stree(void)
155 struct stree *stree;
157 stree = __pop_fake_cur_stree();
158 free_stree(&stree);
161 void __set_fake_cur_stree_fast(struct stree *stree)
163 push_stree(&pre_cond_stack, cur_stree);
164 cur_stree = stree;
165 read_only = 1;
168 void __pop_fake_cur_stree_fast(void)
170 cur_stree = pop_stree(&pre_cond_stack);
171 read_only = 0;
174 void __merge_stree_into_cur(struct stree *stree)
176 struct sm_state *sm;
177 struct sm_state *orig;
178 struct sm_state *merged;
180 FOR_EACH_SM(stree, sm) {
181 orig = get_sm_state(sm->owner, sm->name, sm->sym);
182 if (orig)
183 merged = merge_sm_states(orig, sm);
184 else
185 merged = sm;
186 __set_sm(merged);
187 } END_FOR_EACH_SM(sm);
190 void __set_sm(struct sm_state *sm)
192 if (read_only)
193 sm_msg("Smatch Internal Error: cur_stree is read only.");
195 if (option_debug ||
196 strcmp(check_name(sm->owner), option_debug_check) == 0) {
197 struct smatch_state *s;
199 s = get_state(sm->owner, sm->name, sm->sym);
200 if (!s)
201 sm_msg("%s new %s", __func__, show_sm(sm));
202 else
203 sm_msg("%s change %s (was %s)", __func__, show_sm(sm),
204 show_state(s));
207 if (unreachable())
208 return;
210 if (fake_cur_stree_stack)
211 overwrite_sm_state_stree_stack(&fake_cur_stree_stack, sm);
213 overwrite_sm_state_stree(&cur_stree, sm);
216 void __set_sm_cur_stree(struct sm_state *sm)
218 if (read_only)
219 sm_msg("Smatch Internal Error: cur_stree is read only.");
221 if (option_debug ||
222 strcmp(check_name(sm->owner), option_debug_check) == 0) {
223 struct smatch_state *s;
225 s = get_state(sm->owner, sm->name, sm->sym);
226 if (!s)
227 sm_msg("%s new %s", __func__, show_sm(sm));
228 else
229 sm_msg("%s change %s (was %s)",
230 __func__, show_sm(sm), show_state(s));
233 if (unreachable())
234 return;
236 overwrite_sm_state_stree(&cur_stree, sm);
239 void __set_sm_fake_stree(struct sm_state *sm)
241 if (read_only)
242 sm_msg("Smatch Internal Error: cur_stree is read only.");
244 if (option_debug ||
245 strcmp(check_name(sm->owner), option_debug_check) == 0) {
246 struct smatch_state *s;
248 s = get_state(sm->owner, sm->name, sm->sym);
249 if (!s)
250 sm_msg("%s new %s", __func__, show_sm(sm));
251 else
252 sm_msg("%s change %s (was %s)",
253 __func__, show_sm(sm), show_state(s));
256 if (unreachable())
257 return;
259 overwrite_sm_state_stree_stack(&fake_cur_stree_stack, sm);
263 typedef void (get_state_hook)(int owner, const char *name, struct symbol *sym);
264 DECLARE_PTR_LIST(fn_list, get_state_hook *);
265 static struct fn_list *get_state_hooks;
267 void add_get_state_hook(get_state_hook *fn)
269 get_state_hook **p = malloc(sizeof(get_state_hook *));
270 *p = fn;
271 add_ptr_list(&get_state_hooks, p);
274 static void call_get_state_hooks(int owner, const char *name, struct symbol *sym)
276 static int recursion;
277 get_state_hook **fn;
279 if (recursion)
280 return;
281 recursion = 1;
283 FOR_EACH_PTR(get_state_hooks, fn) {
284 (*fn)(owner, name, sym);
285 } END_FOR_EACH_PTR(fn);
287 recursion = 0;
290 struct smatch_state *__get_state(int owner, const char *name, struct symbol *sym)
292 return get_state_stree(cur_stree, owner, name, sym);
295 struct smatch_state *get_state(int owner, const char *name, struct symbol *sym)
297 call_get_state_hooks(owner, name, sym);
299 return __get_state(owner, name, sym);
302 struct smatch_state *get_state_expr(int owner, struct expression *expr)
304 char *name;
305 struct symbol *sym;
306 struct smatch_state *ret = NULL;
308 expr = strip_expr(expr);
309 name = expr_to_var_sym(expr, &sym);
310 if (!name || !sym)
311 goto free;
312 ret = get_state(owner, name, sym);
313 free:
314 free_string(name);
315 return ret;
318 struct state_list *get_possible_states(int owner, const char *name, struct symbol *sym)
320 struct sm_state *sms;
322 sms = get_sm_state_stree(cur_stree, owner, name, sym);
323 if (sms)
324 return sms->possible;
325 return NULL;
328 struct state_list *get_possible_states_expr(int owner, struct expression *expr)
330 char *name;
331 struct symbol *sym;
332 struct state_list *ret = NULL;
334 expr = strip_expr(expr);
335 name = expr_to_var_sym(expr, &sym);
336 if (!name || !sym)
337 goto free;
338 ret = get_possible_states(owner, name, sym);
339 free:
340 free_string(name);
341 return ret;
344 struct sm_state *get_sm_state(int owner, const char *name, struct symbol *sym)
346 return get_sm_state_stree(cur_stree, owner, name, sym);
349 struct sm_state *get_sm_state_expr(int owner, struct expression *expr)
351 char *name;
352 struct symbol *sym;
353 struct sm_state *ret = NULL;
355 expr = strip_expr(expr);
356 name = expr_to_var_sym(expr, &sym);
357 if (!name || !sym)
358 goto free;
359 ret = get_sm_state(owner, name, sym);
360 free:
361 free_string(name);
362 return ret;
365 void delete_state(int owner, const char *name, struct symbol *sym)
367 delete_state_stree(&cur_stree, owner, name, sym);
368 if (cond_true_stack) {
369 delete_state_stree_stack(&pre_cond_stack, owner, name, sym);
370 delete_state_stree_stack(&cond_true_stack, owner, name, sym);
371 delete_state_stree_stack(&cond_false_stack, owner, name, sym);
375 void delete_state_expr(int owner, struct expression *expr)
377 char *name;
378 struct symbol *sym;
380 expr = strip_expr(expr);
381 name = expr_to_var_sym(expr, &sym);
382 if (!name || !sym)
383 goto free;
384 delete_state(owner, name, sym);
385 free:
386 free_string(name);
389 static void delete_all_states_stree_sym(struct stree **stree, struct symbol *sym)
391 struct state_list *slist = NULL;
392 struct sm_state *sm;
394 FOR_EACH_SM(*stree, sm) {
395 if (sm->sym == sym)
396 add_ptr_list(&slist, sm);
397 } END_FOR_EACH_SM(sm);
399 FOR_EACH_PTR(slist, sm) {
400 delete_state_stree(stree, sm->owner, sm->name, sm->sym);
401 } END_FOR_EACH_PTR(sm);
403 free_slist(&slist);
406 static void delete_all_states_stree_stack_sym(struct stree_stack **stack, struct symbol *sym)
408 struct stree *stree;
410 if (!*stack)
411 return;
413 stree = pop_stree(stack);
414 delete_all_states_stree_sym(&stree, sym);
415 push_stree(stack, stree);
418 void __delete_all_states_sym(struct symbol *sym)
420 delete_all_states_stree_sym(&cur_stree, sym);
422 delete_all_states_stree_stack_sym(&true_stack, sym);
423 delete_all_states_stree_stack_sym(&true_stack, sym);
424 delete_all_states_stree_stack_sym(&false_stack, sym);
425 delete_all_states_stree_stack_sym(&pre_cond_stack, sym);
426 delete_all_states_stree_stack_sym(&cond_true_stack, sym);
427 delete_all_states_stree_stack_sym(&cond_false_stack, sym);
428 delete_all_states_stree_stack_sym(&fake_cur_stree_stack, sym);
429 delete_all_states_stree_stack_sym(&break_stack, sym);
430 delete_all_states_stree_stack_sym(&fake_break_stack, sym);
431 delete_all_states_stree_stack_sym(&switch_stack, sym);
432 delete_all_states_stree_stack_sym(&continue_stack, sym);
435 * deleting from the goto stack is problematic because we don't know
436 * if the label is in scope and also we need the value for --two-passes.
440 struct stree *get_all_states_from_stree(int owner, struct stree *source)
442 struct stree *ret = NULL;
443 struct sm_state *tmp;
445 FOR_EACH_SM(source, tmp) {
446 if (tmp->owner == owner)
447 avl_insert(&ret, tmp);
448 } END_FOR_EACH_SM(tmp);
450 return ret;
453 struct stree *get_all_states_stree(int owner)
455 return get_all_states_from_stree(owner, cur_stree);
458 struct stree *__get_cur_stree(void)
460 return cur_stree;
463 int is_reachable(void)
465 if (cur_stree)
466 return 1;
467 return 0;
470 void set_true_false_states(int owner, const char *name, struct symbol *sym,
471 struct smatch_state *true_state,
472 struct smatch_state *false_state)
474 if (read_only)
475 sm_msg("Smatch Internal Error: cur_stree is read only.");
477 if (option_debug || strcmp(check_name(owner), option_debug_check) == 0) {
478 struct smatch_state *tmp;
480 tmp = get_state(owner, name, sym);
481 sm_msg("%s [%s] '%s'. Was %s. Now T:%s F:%s", __func__,
482 check_name(owner), name, show_state(tmp),
483 show_state(true_state), show_state(false_state));
486 if (unreachable())
487 return;
489 if (!cond_false_stack || !cond_true_stack) {
490 printf("Error: missing true/false stacks\n");
491 return;
494 if (true_state)
495 set_state_stree_stack(&cond_true_stack, owner, name, sym, true_state);
496 if (false_state)
497 set_state_stree_stack(&cond_false_stack, owner, name, sym, false_state);
500 void set_true_false_states_expr(int owner, struct expression *expr,
501 struct smatch_state *true_state,
502 struct smatch_state *false_state)
504 char *name;
505 struct symbol *sym;
507 expr = strip_expr(expr);
508 name = expr_to_var_sym(expr, &sym);
509 if (!name || !sym)
510 goto free;
511 set_true_false_states(owner, name, sym, true_state, false_state);
512 free:
513 free_string(name);
516 void __set_true_false_sm(struct sm_state *true_sm, struct sm_state *false_sm)
518 int owner;
519 const char *name;
520 struct symbol *sym;
522 if (!true_sm && !false_sm)
523 return;
525 if (unreachable())
526 return;
528 owner = true_sm ? true_sm->owner : false_sm->owner;
529 name = true_sm ? true_sm->name : false_sm->name;
530 sym = true_sm ? true_sm->sym : false_sm->sym;
531 if (option_debug || strcmp(check_name(owner), option_debug_check) == 0) {
532 struct smatch_state *tmp;
534 tmp = get_state(owner, name, sym);
535 sm_msg("%s [%s] '%s'. Was %s. Now T:%s F:%s", __func__,
536 check_name(owner), name, show_state(tmp),
537 show_state(true_sm ? true_sm->state : NULL),
538 show_state(false_sm ? false_sm->state : NULL));
541 if (!cond_false_stack || !cond_true_stack) {
542 printf("Error: missing true/false stacks\n");
543 return;
546 if (true_sm)
547 overwrite_sm_state_stree_stack(&cond_true_stack, true_sm);
548 if (false_sm)
549 overwrite_sm_state_stree_stack(&cond_false_stack, false_sm);
552 void nullify_path(void)
554 if (fake_cur_stree_stack) {
555 __free_fake_cur_stree();
556 __push_fake_cur_stree();
558 free_stree(&cur_stree);
561 void __match_nullify_path_hook(const char *fn, struct expression *expr,
562 void *unused)
564 nullify_path();
568 * At the start of every function we mark the path
569 * as unnull. That way there is always at least one state
570 * in the cur_stree until nullify_path is called. This
571 * is used in merge_slist() for the first null check.
573 void __unnullify_path(void)
575 if (!cur_stree)
576 set_state(-1, "unnull_path", NULL, &true_state);
579 int __path_is_null(void)
581 if (cur_stree)
582 return 0;
583 return 1;
586 static void check_stree_stack_free(struct stree_stack **stack)
588 if (*stack) {
589 sm_msg("smatch internal error: stack not empty");
590 free_stack_and_strees(stack);
594 void save_all_states(void)
596 __add_ptr_list(&backup, cur_stree, 0);
597 cur_stree = NULL;
599 __add_ptr_list(&backup, true_stack, 0);
600 true_stack = NULL;
601 __add_ptr_list(&backup, false_stack, 0);
602 false_stack = NULL;
603 __add_ptr_list(&backup, pre_cond_stack, 0);
604 pre_cond_stack = NULL;
606 __add_ptr_list(&backup, cond_true_stack, 0);
607 cond_true_stack = NULL;
608 __add_ptr_list(&backup, cond_false_stack, 0);
609 cond_false_stack = NULL;
611 __add_ptr_list(&backup, fake_cur_stree_stack, 0);
612 fake_cur_stree_stack = NULL;
614 __add_ptr_list(&backup, break_stack, 0);
615 break_stack = NULL;
616 __add_ptr_list(&backup, fake_break_stack, 0);
617 fake_break_stack = NULL;
619 __add_ptr_list(&backup, switch_stack, 0);
620 switch_stack = NULL;
621 __add_ptr_list(&backup, remaining_cases, 0);
622 remaining_cases = NULL;
623 __add_ptr_list(&backup, default_stack, 0);
624 default_stack = NULL;
625 __add_ptr_list(&backup, continue_stack, 0);
626 continue_stack = NULL;
628 __add_ptr_list(&backup, goto_stack, 0);
629 goto_stack = NULL;
632 static void *pop_backup(void)
634 void *ret;
636 ret = last_ptr_list(backup);
637 delete_ptr_list_last(&backup);
638 return ret;
641 void restore_all_states(void)
643 goto_stack = pop_backup();
645 continue_stack = pop_backup();
646 default_stack = pop_backup();
647 remaining_cases = pop_backup();
648 switch_stack = pop_backup();
649 fake_break_stack = pop_backup();
650 break_stack = pop_backup();
652 fake_cur_stree_stack = pop_backup();
654 cond_false_stack = pop_backup();
655 cond_true_stack = pop_backup();
657 pre_cond_stack = pop_backup();
658 false_stack = pop_backup();
659 true_stack = pop_backup();
661 cur_stree = pop_backup();
664 void free_goto_stack(void)
666 struct named_stree *named_stree;
668 FOR_EACH_PTR(goto_stack, named_stree) {
669 free_stree(&named_stree->stree);
670 } END_FOR_EACH_PTR(named_stree);
671 __free_ptr_list((struct ptr_list **)&goto_stack);
674 void clear_all_states(void)
676 nullify_path();
677 check_stree_stack_free(&true_stack);
678 check_stree_stack_free(&false_stack);
679 check_stree_stack_free(&pre_cond_stack);
680 check_stree_stack_free(&cond_true_stack);
681 check_stree_stack_free(&cond_false_stack);
682 check_stree_stack_free(&break_stack);
683 check_stree_stack_free(&fake_break_stack);
684 check_stree_stack_free(&switch_stack);
685 check_stree_stack_free(&continue_stack);
686 check_stree_stack_free(&fake_cur_stree_stack);
688 free_goto_stack();
690 free_every_single_sm_state();
693 void __push_cond_stacks(void)
695 push_stree(&cond_true_stack, NULL);
696 push_stree(&cond_false_stack, NULL);
697 __push_fake_cur_stree();
700 void __fold_in_set_states(void)
702 struct stree *new_states;
703 struct sm_state *sm;
705 new_states = __pop_fake_cur_stree();
706 FOR_EACH_SM(new_states, sm) {
707 __set_sm(sm);
708 __set_true_false_sm(sm, sm);
709 } END_FOR_EACH_SM(sm);
710 free_stree(&new_states);
713 void __free_set_states(void)
715 struct stree *new_states;
717 new_states = __pop_fake_cur_stree();
718 free_stree(&new_states);
721 struct stree *__copy_cond_true_states(void)
723 struct stree *ret;
725 ret = pop_stree(&cond_true_stack);
726 push_stree(&cond_true_stack, clone_stree(ret));
727 return ret;
730 struct stree *__copy_cond_false_states(void)
732 struct stree *ret;
734 ret = pop_stree(&cond_false_stack);
735 push_stree(&cond_false_stack, clone_stree(ret));
736 return ret;
739 struct stree *__pop_cond_true_stack(void)
741 return pop_stree(&cond_true_stack);
744 struct stree *__pop_cond_false_stack(void)
746 return pop_stree(&cond_false_stack);
750 * This combines the pre cond states with either the true or false states.
751 * For example:
752 * a = kmalloc() ; if (a !! foo(a)
753 * In the pre state a is possibly null. In the true state it is non null.
754 * In the false state it is null. Combine the pre and the false to get
755 * that when we call 'foo', 'a' is null.
757 static void __use_cond_stack(struct stree_stack **stack)
759 struct stree *stree;
761 free_stree(&cur_stree);
763 cur_stree = pop_stree(&pre_cond_stack);
764 push_stree(&pre_cond_stack, clone_stree(cur_stree));
766 stree = pop_stree(stack);
767 overwrite_stree(stree, &cur_stree);
768 push_stree(stack, stree);
771 void __use_pre_cond_states(void)
773 free_stree(&cur_stree);
774 cur_stree = pop_stree(&pre_cond_stack);
777 void __use_cond_true_states(void)
779 __use_cond_stack(&cond_true_stack);
782 void __use_cond_false_states(void)
784 __use_cond_stack(&cond_false_stack);
787 void __negate_cond_stacks(void)
789 struct stree *old_false, *old_true;
791 __use_cond_stack(&cond_false_stack);
792 old_false = pop_stree(&cond_false_stack);
793 old_true = pop_stree(&cond_true_stack);
794 push_stree(&cond_false_stack, old_true);
795 push_stree(&cond_true_stack, old_false);
798 void __and_cond_states(void)
800 and_stree_stack(&cond_true_stack);
801 or_stree_stack(&pre_cond_stack, cur_stree, &cond_false_stack);
804 void __or_cond_states(void)
806 or_stree_stack(&pre_cond_stack, cur_stree, &cond_true_stack);
807 and_stree_stack(&cond_false_stack);
810 void __save_pre_cond_states(void)
812 push_stree(&pre_cond_stack, clone_stree(cur_stree));
815 void __discard_pre_cond_states(void)
817 struct stree *tmp;
819 tmp = pop_stree(&pre_cond_stack);
820 free_stree(&tmp);
823 struct stree *__get_true_states(void)
825 return clone_stree(top_stree(cond_true_stack));
828 struct stree *__get_false_states(void)
830 return clone_stree(top_stree(cond_false_stack));
833 void __use_cond_states(void)
835 struct stree *pre, *pre_clone, *true_states, *false_states;
837 pre = pop_stree(&pre_cond_stack);
838 pre_clone = clone_stree(pre);
840 true_states = pop_stree(&cond_true_stack);
841 overwrite_stree(true_states, &pre);
842 free_stree(&true_states);
843 /* we use the true states right away */
844 free_stree(&cur_stree);
845 cur_stree = pre;
847 false_states = pop_stree(&cond_false_stack);
848 overwrite_stree(false_states, &pre_clone);
849 free_stree(&false_states);
850 push_stree(&false_stack, pre_clone);
853 void __push_true_states(void)
855 push_stree(&true_stack, clone_stree(cur_stree));
858 void __use_false_states(void)
860 free_stree(&cur_stree);
861 cur_stree = pop_stree(&false_stack);
864 void __discard_false_states(void)
866 struct stree *stree;
868 stree = pop_stree(&false_stack);
869 free_stree(&stree);
872 void __merge_false_states(void)
874 struct stree *stree;
876 stree = pop_stree(&false_stack);
877 merge_stree(&cur_stree, stree);
878 free_stree(&stree);
882 * This function probably seemed common sensical when I wrote it but, oh wow,
883 * does it look subtle in retrospect. Say we set a state on one side of the if
884 * else path but not on the other, then what we should record in the fake stree
885 * is the merged state.
887 * This function relies on the fact that the we always set the cur_stree as well
888 * and we already have the infrastructure to merge things correctly into the
889 * cur_stree.
891 * So instead of merging fake strees together which is probably a lot of work,
892 * we just use it as a list of set states and look up the actual current values
893 * in the cur_stree.
896 static void update_stree_with_merged(struct stree **stree)
898 struct state_list *slist = NULL;
899 struct sm_state *sm, *new;
901 FOR_EACH_SM(*stree, sm) {
902 new = get_sm_state(sm->owner, sm->name, sm->sym);
903 if (!new) /* This can happen if we go out of scope */
904 continue;
905 add_ptr_list(&slist, new);
906 } END_FOR_EACH_SM(sm);
908 FOR_EACH_PTR(slist, sm) {
909 overwrite_sm_state_stree(stree, sm);
910 } END_FOR_EACH_PTR(sm);
912 free_slist(&slist);
915 static void update_fake_stree_with_merged(void)
917 struct stree *stree;
919 if (!fake_cur_stree_stack)
920 return;
921 stree = pop_stree(&fake_cur_stree_stack);
922 update_stree_with_merged(&stree);
923 push_stree(&fake_cur_stree_stack, stree);
926 void __merge_true_states(void)
928 struct stree *stree;
930 stree = pop_stree(&true_stack);
931 merge_stree(&cur_stree, stree);
932 update_fake_stree_with_merged();
933 free_stree(&stree);
936 void __push_continues(void)
938 push_stree(&continue_stack, NULL);
941 void __discard_continues(void)
943 struct stree *stree;
945 stree = pop_stree(&continue_stack);
946 free_stree(&stree);
949 void __process_continues(void)
951 struct stree *stree;
953 stree = pop_stree(&continue_stack);
954 if (!stree)
955 stree = clone_stree(cur_stree);
956 else
957 merge_stree(&stree, cur_stree);
959 push_stree(&continue_stack, stree);
962 void __merge_continues(void)
964 struct stree *stree;
966 stree = pop_stree(&continue_stack);
967 merge_stree(&cur_stree, stree);
968 free_stree(&stree);
971 void __push_breaks(void)
973 push_stree(&break_stack, NULL);
974 if (fake_cur_stree_stack)
975 push_stree(&fake_break_stack, NULL);
978 void __process_breaks(void)
980 struct stree *stree;
982 stree = pop_stree(&break_stack);
983 if (!stree)
984 stree = clone_stree(cur_stree);
985 else
986 merge_stree(&stree, cur_stree);
987 push_stree(&break_stack, stree);
989 if (!fake_cur_stree_stack)
990 return;
992 stree = pop_stree(&fake_break_stack);
993 if (!stree)
994 stree = clone_stree(top_stree(fake_cur_stree_stack));
995 else
996 merge_stree(&stree, top_stree(fake_cur_stree_stack));
997 push_stree(&fake_break_stack, stree);
1000 int __has_breaks(void)
1002 struct stree *stree;
1003 int ret;
1005 stree = pop_stree(&break_stack);
1006 ret = !!stree;
1007 push_stree(&break_stack, stree);
1008 return ret;
1011 void __merge_breaks(void)
1013 struct stree *stree;
1014 struct sm_state *sm;
1016 stree = pop_stree(&break_stack);
1017 merge_stree(&cur_stree, stree);
1018 free_stree(&stree);
1020 if (!fake_cur_stree_stack)
1021 return;
1023 stree = pop_stree(&fake_break_stack);
1024 update_stree_with_merged(&stree);
1025 FOR_EACH_SM(stree, sm) {
1026 overwrite_sm_state_stree_stack(&fake_cur_stree_stack, sm);
1027 } END_FOR_EACH_SM(sm);
1028 free_stree(&stree);
1031 void __use_breaks(void)
1033 struct stree *stree;
1034 struct sm_state *sm;
1036 free_stree(&cur_stree);
1037 cur_stree = pop_stree(&break_stack);
1039 if (!fake_cur_stree_stack)
1040 return;
1041 stree = pop_stree(&fake_break_stack);
1042 FOR_EACH_SM(stree, sm) {
1043 overwrite_sm_state_stree_stack(&fake_cur_stree_stack, sm);
1044 } END_FOR_EACH_SM(sm);
1045 free_stree(&stree);
1050 void __save_switch_states(struct expression *switch_expr)
1052 struct range_list *rl;
1054 get_absolute_rl(switch_expr, &rl);
1056 push_rl(&remaining_cases, rl);
1057 push_stree(&switch_stack, clone_stree(cur_stree));
1060 void __merge_switches(struct expression *switch_expr, struct range_list *case_rl)
1062 struct stree *stree;
1063 struct stree *implied_stree;
1065 stree = pop_stree(&switch_stack);
1066 implied_stree = __implied_case_stree(switch_expr, case_rl, &remaining_cases, &stree);
1067 merge_stree(&cur_stree, implied_stree);
1068 free_stree(&implied_stree);
1069 push_stree(&switch_stack, stree);
1072 void __discard_switches(void)
1074 struct stree *stree;
1076 pop_rl(&remaining_cases);
1077 stree = pop_stree(&switch_stack);
1078 free_stree(&stree);
1081 void __push_default(void)
1083 push_stree(&default_stack, NULL);
1086 void __set_default(void)
1088 set_state_stree_stack(&default_stack, 0, "has_default", NULL, &true_state);
1091 int __pop_default(void)
1093 struct stree *stree;
1095 stree = pop_stree(&default_stack);
1096 if (stree) {
1097 free_stree(&stree);
1098 return 1;
1100 return 0;
1103 static struct named_stree *alloc_named_stree(const char *name, struct symbol *sym, struct stree *stree)
1105 struct named_stree *named_stree = __alloc_named_stree(0);
1107 named_stree->name = (char *)name;
1108 named_stree->stree = stree;
1109 named_stree->sym = sym;
1110 return named_stree;
1113 void __save_gotos(const char *name, struct symbol *sym)
1115 struct stree **stree;
1116 struct stree *clone;
1118 stree = get_named_stree(goto_stack, name, sym);
1119 if (stree) {
1120 merge_stree(stree, cur_stree);
1121 return;
1122 } else {
1123 struct named_stree *named_stree;
1125 clone = clone_stree(cur_stree);
1126 named_stree = alloc_named_stree(name, sym, clone);
1127 add_ptr_list(&goto_stack, named_stree);
1131 void __merge_gotos(const char *name, struct symbol *sym)
1133 struct stree **stree;
1135 stree = get_named_stree(goto_stack, name, sym);
1136 if (stree)
1137 merge_stree(&cur_stree, *stree);