Commit 23e2368ac339842b907ccc36580c3a471bb26d43 | zig

author	Luuk de Gram <luuk@degram.dev> 2022-03-06 15:57:28+01:00
committer	Luuk de Gram <luuk@degram.dev> 2022-03-06 16:04:15+01:00
log	23e2368ac339842b907ccc36580c3a471bb26d43
tree	1522a9e220c037f9a019335b212e1901e6801102
parent	bf972e44d5d0ce704cae99957d565c55ea16335d
signature	Commit is signed but in an unrecognized format.

stage2: Fix codegen for unions and error unions

When an union had a zero-sized payload type, we would lower the tag twice. This is fixed
by exiting early when `payload_size` is 0.

With regards to error unions, we were only accounting for padding for the payload field.
However, the errorset value can have a smaller alignment than the payload as well, i.e. error!usize.
We fix this by also accounting for padding/alignment of the error set tag of an error union.

1 files changed, 29 insertions(+), 26 deletions(-)

src
- codegen.zig

src/codegen.zig+29-26

 @@ -573,16 +573,10 @@ pub fn generateSymbol(
             const layout = typed_value.ty.unionGetLayout(target);
             if (layout.payload_size == 0) {
-                switch (try generateSymbol(bin_file, src_loc, .{
+                return generateSymbol(bin_file, src_loc, .{
                     .ty = typed_value.ty.unionTagType().?,
                     .val = union_obj.tag,
-                }, code, debug_output, reloc_info)) {
+                }, code, debug_output, reloc_info);
-                    .appended => {},
-                    .externally_managed => |external_slice| {
-                        code.appendSliceAssumeCapacity(external_slice);
-                    },
-                    .fail => |em| return Result{ .fail = em },
-                }
             }
             // Check if we should store the tag first.
 @@ -703,20 +697,30 @@ pub fn generateSymbol(
             const abi_align = typed_value.ty.abiAlignment(target);
-            const error_val = if (!is_payload) typed_value.val else Value.initTag(.zero);
+            {
-            const begin = code.items.len;
+                const error_val = if (!is_payload) typed_value.val else Value.initTag(.zero);
-            switch (try generateSymbol(bin_file, src_loc, .{
+                const begin = code.items.len;
-                .ty = error_ty,
+                switch (try generateSymbol(bin_file, src_loc, .{
-                .val = error_val,
+                    .ty = error_ty,
-            }, code, debug_output, reloc_info)) {
+                    .val = error_val,
-                .appended => {},
+                }, code, debug_output, reloc_info)) {
-                .externally_managed => |external_slice| {
+                    .appended => {},
-                    code.appendSliceAssumeCapacity(external_slice);
+                    .externally_managed => |external_slice| {
-                },
+                        code.appendSliceAssumeCapacity(external_slice);
-                .fail => |em| return Result{ .fail = em },
+                    },
+                    .fail => |em| return Result{ .fail = em },
+                }
+                const unpadded_end = code.items.len - begin;
+                const padded_end = mem.alignForwardGeneric(u64, unpadded_end, abi_align);
+                const padding = try math.cast(usize, padded_end - unpadded_end);
+                if (padding > 0) {
+                    try code.writer().writeByteNTimes(0, padding);
+                }
             }
             if (payload_ty.hasRuntimeBits()) {
+                const begin = code.items.len;
                 const payload_val = if (typed_value.val.castTag(.eu_payload)) |val| val.data else Value.initTag(.undef);
                 switch (try generateSymbol(bin_file, src_loc, .{
                     .ty = payload_ty,
 @@ -728,14 +732,13 @@ pub fn generateSymbol(
                     },
                     .fail => |em| return Result{ .fail = em },
                 }
-            }
+                const unpadded_end = code.items.len - begin;
+                const padded_end = mem.alignForwardGeneric(u64, unpadded_end, abi_align);
+                const padding = try math.cast(usize, padded_end - unpadded_end);
-            const unpadded_end = code.items.len - begin;
+                if (padding > 0) {
-            const padded_end = mem.alignForwardGeneric(u64, unpadded_end, abi_align);
+                    try code.writer().writeByteNTimes(0, padding);
-            const padding = try math.cast(usize, padded_end - unpadded_end);
+                }
-            if (padding > 0) {
-                try code.writer().writeByteNTimes(0, padding);
             }
             return Result{ .appended = {} };