Skip to content

Commit

Permalink
rustc: ensure optimized enums have a properly aligned size.
Browse files Browse the repository at this point in the history
  • Loading branch information
eddyb committed Dec 18, 2017
1 parent a3a7203 commit 087f1c2
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 4 deletions.
3 changes: 2 additions & 1 deletion src/librustc/ty/layout.rs
Original file line number Diff line number Diff line change
Expand Up @@ -1492,7 +1492,7 @@ impl<'a, 'tcx> LayoutDetails {
}).collect::<Result<Vec<_>, _>>()?;

let offset = st[i].fields.offset(field_index) + offset;
let LayoutDetails { size, mut align, .. } = st[i];
let LayoutDetails { mut size, mut align, .. } = st[i];

let mut niche_align = niche.value.align(dl);
let abi = if offset.bytes() == 0 && niche.value.size(dl) == size {
Expand All @@ -1504,6 +1504,7 @@ impl<'a, 'tcx> LayoutDetails {
Abi::Aggregate { sized: true }
};
align = align.max(niche_align);
size = size.abi_align(align);

return Ok(tcx.intern_layout(LayoutDetails {
variants: Variants::NicheFilling {
Expand Down
13 changes: 10 additions & 3 deletions src/test/run-pass/packed-struct-optimized-enum.rs
Original file line number Diff line number Diff line change
Expand Up @@ -16,14 +16,21 @@ impl<T: Copy> Clone for Packed<T> {
fn clone(&self) -> Self { *self }
}

fn main() {
let one = (Some(Packed((&(), 0))), true);
fn sanity_check_size<T: Copy>(one: T) {
let two = [one, one];
let stride = (&two[1] as *const _ as usize) - (&two[0] as *const _ as usize);
assert_eq!(stride, std::mem::size_of_val(&one));
}

fn main() {
// This can fail if rustc and LLVM disagree on the size of a type.
// In this case, `Option<Packed<(&(), u32)>>` was erronously not
// marked as packed despite needing alignment `1` and containing
// its `&()` discriminant, which has alignment larger than `1`.
assert_eq!(stride, std::mem::size_of_val(&one));
sanity_check_size((Some(Packed((&(), 0))), true));

// In #46769, `Option<(Packed<&()>, bool)>` was found to have
// pointer alignment, without actually being aligned in size.
// E.g. on 64-bit platforms, it had alignment `8` but size `9`.
sanity_check_size(Some((Packed(&()), true)));
}

0 comments on commit 087f1c2

Please sign in to comment.