nsz Git - cparser/blob - walk_statements.c

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