handling the ast-destruction in computed goto

This commit is contained in:
Wolfgang Bumiller 2013-01-03 13:20:57 +01:00
parent 7d14fdf530
commit cf293bc669
2 changed files with 25 additions and 15 deletions

9
ast.c
View file

@ -724,9 +724,12 @@ ast_ternary* ast_ternary_new(lex_ctx ctx, ast_expression *cond, ast_expression *
void ast_ternary_delete(ast_ternary *self) void ast_ternary_delete(ast_ternary *self)
{ {
ast_unref(self->cond); /* the if()s are only there because computed-gotos can set them
ast_unref(self->on_true); * to NULL
ast_unref(self->on_false); */
if (self->cond) ast_unref(self->cond);
if (self->on_true) ast_unref(self->on_true);
if (self->on_false) ast_unref(self->on_false);
ast_expression_delete((ast_expression*)self); ast_expression_delete((ast_expression*)self);
mem_d(self); mem_d(self);
} }

View file

@ -2999,28 +2999,35 @@ static bool parse_switch_go(parser_t *parser, ast_block *block, ast_expression *
} }
/* parse computed goto sides */ /* parse computed goto sides */
static ast_expression *parse_goto_computed(parser_t *parser, ast_expression *side) { static ast_expression *parse_goto_computed(parser_t *parser, ast_expression **side) {
ast_expression *on_true; ast_expression *on_true;
ast_expression *on_false; ast_expression *on_false;
ast_expression *cond;
if (!side) if (!*side)
return NULL; return NULL;
if (ast_istype(side, ast_ternary)) { if (ast_istype(*side, ast_ternary)) {
on_true = parse_goto_computed(parser, ((ast_ternary*)side)->on_true); ast_ternary *tern = (ast_ternary*)*side;
on_false = parse_goto_computed(parser, ((ast_ternary*)side)->on_false); on_true = parse_goto_computed(parser, &tern->on_true);
on_false = parse_goto_computed(parser, &tern->on_false);
if (!on_true || !on_false) { if (!on_true || !on_false) {
parseerror(parser, "expected label or expression in ternary"); parseerror(parser, "expected label or expression in ternary");
if (((ast_ternary*)side)->on_false) ast_unref(((ast_ternary*)side)->on_false); if (on_true) ast_unref(on_true);
if (((ast_ternary*)side)->on_true) ast_unref(((ast_ternary*)side)->on_true); if (on_false) ast_unref(on_false);
return NULL; return NULL;
} }
return (ast_expression*)ast_ifthen_new(parser_ctx(parser), ((ast_ternary*)side)->cond, on_true, on_false); cond = tern->cond;
} else if (ast_istype(side, ast_label)) { tern->cond = NULL;
ast_goto *gt = ast_goto_new(parser_ctx(parser), ((ast_label*)side)->name); ast_delete(tern);
ast_goto_set_label(gt, ((ast_label*)side)); *side = NULL;
return (ast_expression*)ast_ifthen_new(parser_ctx(parser), cond, on_true, on_false);
} else if (ast_istype(*side, ast_label)) {
ast_goto *gt = ast_goto_new(parser_ctx(parser), ((ast_label*)*side)->name);
ast_goto_set_label(gt, ((ast_label*)*side));
*side = NULL;
return (ast_expression*)gt; return (ast_expression*)gt;
} }
return NULL; return NULL;
@ -3045,7 +3052,7 @@ static bool parse_goto(parser_t *parser, ast_expression **out)
/* failed to parse expression for goto */ /* failed to parse expression for goto */
if (!(expression = parse_expression(parser, false, true)) || if (!(expression = parse_expression(parser, false, true)) ||
!(*out = parse_goto_computed(parser, expression))) { !(*out = parse_goto_computed(parser, &expression))) {
parseerror(parser, "invalid goto expression"); parseerror(parser, "invalid goto expression");
ast_unref(expression); ast_unref(expression);
return false; return false;