42 delete_ptr_list_last((struct ptr_list **)stack_stack);
43 return stack;
44 }
45
46 static struct stree_stack *return_stree_stack;
47 static struct stree_stack_stack *saved_stack_stack;
48 static struct stree *all_return_states;
49 static struct stree_stack *saved_stack;
50
51 void all_return_states_hook(void (*callback)(void))
52 {
53 struct return_states_callback *rs_cb = __alloc_return_states_callback(0);
54
55 rs_cb->callback = callback;
56 add_ptr_list(&callback_list, rs_cb);
57 }
58
59 static void call_hooks(void)
60 {
61 struct return_states_callback *rs_cb;
62
63 __set_fake_cur_stree_fast(all_return_states);
64 FOR_EACH_PTR(callback_list, rs_cb) {
65 rs_cb->callback();
66 } END_FOR_EACH_PTR(rs_cb);
67 __pop_fake_cur_stree_fast();
68 }
69
70 static void match_return(int return_id, char *return_ranges, struct expression *expr)
71 {
72 struct stree *stree;
73
74 stree = clone_stree(__get_cur_stree());
75 merge_stree_no_pools(&all_return_states, stree);
76 push_stree(&return_stree_stack, stree);
77 }
78
79 static void match_end_func(struct symbol *sym)
80 {
81 /*
82 * FIXME: either this isn't needed or we need to copy a stree into the
83 * return_stree_stack as well.
84 */
85 merge_stree(&all_return_states, __get_cur_stree());
86 call_hooks();
87 }
99 {
100 /* This free_stree() isn't needed is it?? */
101 free_stree(&all_return_states);
102
103 all_return_states = pop_stree(&saved_stack);
104 return_stree_stack = pop_stree_stack(&saved_stack_stack);
105 }
106
107 struct stree *get_all_return_states(void)
108 {
109 return all_return_states;
110 }
111
112 struct stree_stack *get_all_return_strees(void)
113 {
114 return return_stree_stack;
115 }
116
117 static void free_resources(struct symbol *sym)
118 {
119 struct stree *tmp;
120
121 free_stree(&all_return_states);
122
123 FOR_EACH_PTR(return_stree_stack, tmp) {
124 free_stree(&tmp);
125 } END_FOR_EACH_PTR(tmp);
126 free_stree_stack(&return_stree_stack);
127 }
128
129 void register_returns_early(int id)
130 {
131 RETURN_ID = id;
132
133 set_dynamic_states(RETURN_ID);
134 add_split_return_callback(match_return);
135 }
136
137 void register_returns(int id)
138 {
139 add_hook(&match_end_func, END_FUNC_HOOK);
140 add_hook(&match_save_states, INLINE_FN_START);
141 add_hook(&match_restore_states, INLINE_FN_END);
142 add_hook(&free_resources, AFTER_FUNC_HOOK);
143 }
|
42 delete_ptr_list_last((struct ptr_list **)stack_stack);
43 return stack;
44 }
45
46 static struct stree_stack *return_stree_stack;
47 static struct stree_stack_stack *saved_stack_stack;
48 static struct stree *all_return_states;
49 static struct stree_stack *saved_stack;
50
51 void all_return_states_hook(void (*callback)(void))
52 {
53 struct return_states_callback *rs_cb = __alloc_return_states_callback(0);
54
55 rs_cb->callback = callback;
56 add_ptr_list(&callback_list, rs_cb);
57 }
58
59 static void call_hooks(void)
60 {
61 struct return_states_callback *rs_cb;
62 struct stree *orig;
63
64 orig = __swap_cur_stree(all_return_states);
65 FOR_EACH_PTR(callback_list, rs_cb) {
66 rs_cb->callback();
67 } END_FOR_EACH_PTR(rs_cb);
68 __swap_cur_stree(orig);
69 }
70
71 static void match_return(int return_id, char *return_ranges, struct expression *expr)
72 {
73 struct stree *stree;
74
75 stree = clone_stree(__get_cur_stree());
76 merge_stree_no_pools(&all_return_states, stree);
77 push_stree(&return_stree_stack, stree);
78 }
79
80 static void match_end_func(struct symbol *sym)
81 {
82 /*
83 * FIXME: either this isn't needed or we need to copy a stree into the
84 * return_stree_stack as well.
85 */
86 merge_stree(&all_return_states, __get_cur_stree());
87 call_hooks();
88 }
100 {
101 /* This free_stree() isn't needed is it?? */
102 free_stree(&all_return_states);
103
104 all_return_states = pop_stree(&saved_stack);
105 return_stree_stack = pop_stree_stack(&saved_stack_stack);
106 }
107
108 struct stree *get_all_return_states(void)
109 {
110 return all_return_states;
111 }
112
113 struct stree_stack *get_all_return_strees(void)
114 {
115 return return_stree_stack;
116 }
117
118 static void free_resources(struct symbol *sym)
119 {
120 free_stree(&all_return_states);
121 free_stack_and_strees(&return_stree_stack);
122 }
123
124 void register_returns_early(int id)
125 {
126 RETURN_ID = id;
127
128 set_dynamic_states(RETURN_ID);
129 add_split_return_callback(match_return);
130 }
131
132 void register_returns(int id)
133 {
134 add_hook(&match_end_func, END_FUNC_HOOK);
135 add_hook(&match_save_states, INLINE_FN_START);
136 add_hook(&match_restore_states, INLINE_FN_END);
137 add_hook(&free_resources, AFTER_FUNC_HOOK);
138 }
|