diff options
| -rw-r--r-- | src/codegen.c | 5 | ||||
| -rw-r--r-- | tests/switch.sh | 22 |
2 files changed, 24 insertions, 3 deletions
diff --git a/src/codegen.c b/src/codegen.c index 4eb2042..5ac61ed 100644 --- a/src/codegen.c +++ b/src/codegen.c @@ -704,11 +704,10 @@ static bool codegen_switch_body(CodeGen* g, AstNode* stmt, int* case_values, int break; } } - codegen_stmt(g, stmt->as.case_label->body); - return false; + return codegen_switch_body(g, stmt->as.case_label->body, case_values, case_labels, n_cases); } else if (stmt->kind == AstNodeKind_default_label) { fprintf(g->out, ".Ldefault%d:\n", g->switch_label); - codegen_stmt(g, stmt->as.default_label->body); + codegen_switch_body(g, stmt->as.default_label->body, case_values, case_labels, n_cases); return true; } else if (stmt->kind == AstNodeKind_list) { bool default_label_emitted = false; diff --git a/tests/switch.sh b/tests/switch.sh index e77e250..7d9f894 100644 --- a/tests/switch.sh +++ b/tests/switch.sh @@ -210,3 +210,25 @@ int main() { return 0; } EOF + +test_exit_code 0 <<'EOF' +#include <helpers.h> + +int f(int x) { + switch (x) { + case 1: + case 2: + return 12; + case 3: + case 4: + return 34; + } +} + +int main() { + ASSERT_EQ(12, f(1)); + ASSERT_EQ(12, f(2)); + ASSERT_EQ(34, f(3)); + ASSERT_EQ(34, f(4)); +} +EOF |