Skip to content
Closed
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
fix(ast_tools): fix miscalculation of enum layouts.
  • Loading branch information
rzvxa committed Sep 14, 2024
commit ea1c733b844a16b4b5755d66420a6fc73d17d428
15 changes: 10 additions & 5 deletions tasks/ast_tools/src/passes/calc_layout.rs
Original file line number Diff line number Diff line change
Expand Up @@ -123,9 +123,14 @@ fn calc_enum_layout(ty: &mut Enum, ctx: &EarlyCtx) -> Result<PlatformLayout> {
acc
}

let with_tag = |mut acc: KnownLayout| -> KnownLayout {
acc.consume_niches(ty.item.variants.len() as u128, true);
acc
let with_tag = |layout: KnownLayout| -> Result<KnownLayout> {
let niches = layout.niches();
let layout = std::alloc::Layout::from_size_align(layout.size(), layout.align())
.normalize()?
.pad_to_align();
let mut layout = KnownLayout::new(layout.size(), layout.align(), niches);
layout.consume_niches(ty.item.variants.len() as u128, true);
Ok(layout)
};

let layouts = collect_variant_layouts(ty, ctx)?;
Expand All @@ -137,8 +142,8 @@ fn calc_enum_layout(ty: &mut Enum, ctx: &EarlyCtx) -> Result<PlatformLayout> {
.collect::<Option<_>>()
.expect("already checked.");

let x32 = with_tag(layouts_x32.into_iter().fold(KnownLayout::default(), fold_layout));
let x64 = with_tag(layouts_x64.into_iter().fold(KnownLayout::default(), fold_layout));
let x32 = with_tag(layouts_x32.into_iter().fold(KnownLayout::default(), fold_layout))?;
let x64 = with_tag(layouts_x64.into_iter().fold(KnownLayout::default(), fold_layout))?;
Ok(PlatformLayout(Layout::from(x64), Layout::from(x32)))
}

Expand Down