Simplify macro mumbo jumbo.
[cparser] / walk_statements.c
1 #include <config.h>
2
3 #include "adt/error.h"
4 #include "ast_t.h"
5 #include "entity_t.h"
6 #include "walk_statements.h"
7
8
9 static void walk_expression(expression_t const *const expr,
10                             statement_callback const callback, void *const env)
11 {
12         switch (expr->base.kind) {
13         case EXPR_STATEMENT:
14                 walk_statements(expr->statement.statement, callback, env);
15                 return;
16
17         EXPR_BINARY_CASES
18                 walk_expression(expr->binary.left,  callback, env);
19                 walk_expression(expr->binary.right, callback, env);
20                 return;
21
22         EXPR_UNARY_CASES_OPTIONAL
23                 if (expr->unary.value == NULL)
24                         return;
25                 /* FALLTHROUGH */
26         EXPR_UNARY_CASES_MANDATORY
27                 walk_expression(expr->unary.value, callback, env);
28                 return;
29
30         case EXPR_CALL:
31                 for (call_argument_t *arg = expr->call.arguments; arg != NULL;
32                      arg = arg->next) {
33                         walk_expression(arg->expression, callback, env);
34                 }
35                 return;
36
37         case EXPR_UNKNOWN:
38                 panic("unexpected expr kind");
39
40         case EXPR_COMPOUND_LITERAL:
41                 /* TODO... */
42                 break;
43
44         case EXPR_CONDITIONAL:
45                 walk_expression(expr->conditional.condition,        callback, env);
46                 /* may be NULL because of gnu extension */
47                 if (expr->conditional.true_expression != NULL)
48                         walk_expression(expr->conditional.true_expression,  callback, env);
49                 walk_expression(expr->conditional.false_expression, callback, env);
50                 return;
51
52         case EXPR_BUILTIN_CONSTANT_P:
53                 walk_expression(expr->builtin_constant.value, callback, env);
54                 return;
55
56         case EXPR_SELECT:
57                 walk_expression(expr->select.compound, callback, env);
58                 return;
59
60         case EXPR_ARRAY_ACCESS:
61                 walk_expression(expr->array_access.array_ref, callback, env);
62                 walk_expression(expr->array_access.index,     callback, env);
63                 return;
64
65         case EXPR_CLASSIFY_TYPE:
66                 walk_expression(expr->classify_type.type_expression, callback, env);
67                 return;
68
69         case EXPR_SIZEOF:
70         case EXPR_ALIGNOF: {
71                 expression_t *tp_expression = expr->typeprop.tp_expression;
72                 if (tp_expression != NULL) {
73                         walk_expression(tp_expression, callback, env);
74                 }
75                 return;
76         }
77
78         case EXPR_INVALID:
79         case EXPR_OFFSETOF:
80         case EXPR_REFERENCE:
81         case EXPR_REFERENCE_ENUM_VALUE:
82         case EXPR_CONST:
83         case EXPR_CHARACTER_CONSTANT:
84         case EXPR_WIDE_CHARACTER_CONSTANT:
85         case EXPR_STRING_LITERAL:
86         case EXPR_WIDE_STRING_LITERAL:
87         case EXPR_FUNCNAME:
88         case EXPR_VA_START:
89         case EXPR_VA_ARG:
90         case EXPR_LABEL_ADDRESS:
91         case EXPR_BUILTIN_TYPES_COMPATIBLE_P:
92                 break;
93         }
94
95         /* TODO FIXME: implement all the missing expressions here */
96 }
97
98 static void walk_initializer(const initializer_t  *initializer,
99                              statement_callback    callback,
100                                                          void                 *env)
101 {
102         switch(initializer->kind) {
103         case INITIALIZER_VALUE:
104                 walk_expression(initializer->value.value, callback, env);
105                 return;
106         default:
107                 /* FIXME: should walk initializer hierarchies... */
108                 break;
109         }
110 }
111
112 static void walk_declarations(const entity_t*            entity,
113                               const entity_t*      const last,
114                               statement_callback   const callback,
115                               void                *const env)
116 {
117         entity_t const *const end = last != NULL ? last->base.next : NULL;
118         for (; entity != end; entity = entity->base.next) {
119                 /* we only look at variables */
120                 if (entity->kind != ENTITY_VARIABLE)
121                         continue;
122
123                 const variable_t    *variable    = &entity->variable;
124                 const initializer_t *initializer = variable->initializer;
125                 if (initializer != NULL) {
126                         walk_initializer(initializer, callback, env);
127                 }
128         }
129 }
130
131
132 void walk_statements(statement_t *const stmt, statement_callback const callback, void *const env)
133 {
134         callback(stmt, env);
135
136         switch (stmt->kind) {
137                 case STATEMENT_COMPOUND:
138                         for (statement_t *s = stmt->compound.statements; s != NULL; s = s->base.next) {
139                                 walk_statements(s, callback, env);
140                         }
141                         return;
142
143                 case STATEMENT_FOR:
144                         walk_declarations(stmt->fors.scope.entities, NULL, callback, env);
145                         if (stmt->fors.initialisation != NULL)
146                                 walk_expression(stmt->fors.initialisation, callback, env);
147                         if (stmt->fors.condition != NULL)
148                                 walk_expression(stmt->fors.condition, callback, env);
149                         if (stmt->fors.step != NULL)
150                                 walk_expression(stmt->fors.step, callback, env);
151                         walk_statements(stmt->fors.body,           callback, env);
152                         return;
153
154                 case STATEMENT_IF:
155                         walk_expression(stmt->ifs.condition,      callback, env);
156                         walk_statements(stmt->ifs.true_statement, callback, env);
157                         if (stmt->ifs.false_statement != NULL)
158                                 walk_statements(stmt->ifs.false_statement, callback, env);
159                         return;
160
161                 case STATEMENT_SWITCH:
162                         walk_expression(stmt->switchs.expression, callback, env);
163                         walk_statements(stmt->switchs.body,       callback, env);
164                         return;
165
166                 case STATEMENT_LABEL:
167                         walk_statements(stmt->label.statement, callback, env);
168                         return;
169
170                 case STATEMENT_CASE_LABEL:
171                         walk_statements(stmt->case_label.statement, callback, env);
172                         return;
173
174                 case STATEMENT_WHILE:
175                         walk_expression(stmt->whiles.condition, callback, env);
176                         walk_statements(stmt->whiles.body,      callback, env);
177                         return;
178
179                 case STATEMENT_DO_WHILE:
180                         walk_statements(stmt->do_while.body,      callback, env);
181                         walk_expression(stmt->do_while.condition, callback, env);
182                         return;
183
184                 case STATEMENT_EXPRESSION:
185                         walk_expression(stmt->expression.expression, callback, env);
186                         return;
187
188                 case STATEMENT_RETURN:
189                         if (stmt->returns.value != NULL)
190                                 walk_expression(stmt->returns.value, callback, env);
191                         return;
192
193                 case STATEMENT_DECLARATION:
194                         walk_declarations(stmt->declaration.declarations_begin,
195                                         stmt->declaration.declarations_end, callback, env);
196                         return;
197
198                 case STATEMENT_MS_TRY:
199                         walk_statements(stmt->ms_try.try_statement,   callback, env);
200                         walk_statements(stmt->ms_try.final_statement, callback, env);
201                         return;
202
203                 case STATEMENT_INVALID:
204                 case STATEMENT_EMPTY:
205                 case STATEMENT_CONTINUE:
206                 case STATEMENT_BREAK:
207                 case STATEMENT_GOTO:
208                 case STATEMENT_ASM:
209                 case STATEMENT_LEAVE:
210                         return;
211         }
212
213         panic("unhandled statement");
214 }