fix: codegen of switch fallthrough

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