Close loophole in `when` statements that allowed mutating inner field

members

2 files changed, 9 insertions, 6 deletions

diff --git a/src/compile/expressions.c b/src/compile/expressions.c
index a69a7d56..19e0672e 100644
--- a/src/compile/expressions.c
+++ b/src/compile/expressions.c
@@ -30,8 +30,7 @@ Text_t compile_maybe_incref(env_t *env, ast_t *ast, type_t *t) {
         } else {
             Text_t code = Texts("({ ", compile_declaration(t, Text("_tmp")), " = ", compile_to_type(env, ast, t), "; ",
                                 "((", compile_type(t), "){");
-            ast_t *tmp = WrapAST(ast, InlineCCode,
-                                 .chunks = new (ast_list_t, .ast = WrapAST(ast, TextLiteral, Text("_tmp"))), .type = t);
+            ast_t *tmp = WrapLiteralCode(ast, Text("_tmp"), .type = t);
             for (arg_t *field = Match(t, StructType)->fields; field; field = field->next) {
                 Text_t val = compile_maybe_incref(env, WrapAST(ast, FieldAccess, .fielded = tmp, .field = field->name),
                                                   get_arg_type(env, field));
diff --git a/src/compile/whens.c b/src/compile/whens.c
index 4f6a2a40..122c581c 100644
--- a/src/compile/whens.c
+++ b/src/compile/whens.c
@@ -83,8 +83,10 @@ Text_t compile_when_statement(env_t *env, ast_t *ast) {
             const char *var_name = Match(args->value, Var)->name;
             if (!streq(var_name, "_")) {
                 Text_t var = Texts("_$", var_name);
-                code = Texts(code, compile_declaration(tag_type, var), " = _when_subject.",
-                             valid_c_name(clause_tag_name), ";\n");
+                ast_t *member =
+                    WrapLiteralCode(ast, Texts("_when_subject.", valid_c_name(clause_tag_name)), .type = tag_type);
+                code = Texts(code, compile_declaration(tag_type, var), " = ",
+                             compile_maybe_incref(env, member, tag_type), ";\n");
                 scope = fresh_scope(scope);
                 set_binding(scope, Match(args->value, Var)->name, tag_type, EMPTY_TEXT);
             }
@@ -101,8 +103,10 @@ Text_t compile_when_statement(env_t *env, ast_t *ast) {
                 const char *var_name = Match(arg->value, Var)->name;
                 if (!streq(var_name, "_")) {
                     Text_t var = Texts("_$", var_name);
-                    code = Texts(code, compile_declaration(field->type, var), " = _when_subject.",
-                                 valid_c_name(clause_tag_name), ".", valid_c_name(field->name), ";\n");
+                    ast_t *member =
+                        WrapLiteralCode(ast, Texts("_when_subject.", valid_c_name(clause_tag_name)), .type = tag_type);
+                    code = Texts(code, compile_declaration(field->type, var), " = ",
+                                 compile_maybe_incref(env, member, tag_type), ".", valid_c_name(field->name), ";\n");
                     set_binding(scope, Match(arg->value, Var)->name, field->type, var);
                 }
                 field = field->next;