Value: implement writeToMemory for packed unions

1 files changed, 19 insertions, 0 deletions

diff --git a/src/value.zig b/src/value.zig
index 306e31c0a7..98842a4ca7 100644
--- a/src/value.zig
+++ b/src/value.zig
@@ -1340,6 +1340,14 @@ pub const Value = extern union {
                 const int = mod.global_error_set.get(val.castTag(.@"error").?.data.name).?;
                 std.mem.writeInt(Int, buffer[0..@sizeOf(Int)], @intCast(Int, int), endian);
             },
+            .Union => switch (ty.containerLayout()) {
+                .Auto => unreachable,
+                .Extern => @panic("TODO implement writeToMemory for extern unions"),
+                .Packed => {
+                    const byte_count = (@intCast(usize, ty.bitSize(target)) + 7) / 8;
+                    writeToPackedMemory(val, ty, mod, buffer[0..byte_count], 0);
+                },
+            },
             else => @panic("TODO implement writeToMemory for more types"),
         }
     }
@@ -1430,6 +1438,17 @@ pub const Value = extern union {
                     }
                 },
             },
+            .Union => switch (ty.containerLayout()) {
+                .Auto => unreachable, // Sema is supposed to have emitted a compile error already
+                .Extern => unreachable, // Handled in non-packed writeToMemory
+                .Packed => {
+                    const field_index = ty.unionTagFieldIndex(val.unionTag(), mod);
+                    const field_type = ty.unionFields().values()[field_index.?].ty;
+                    const field_val = val.fieldValue(field_type, field_index.?);
+
+                    field_val.writeToPackedMemory(field_type, mod, buffer, bit_offset);
+                },
+            },
             else => @panic("TODO implement writeToPackedMemory for more types"),
         }
     }