Skip to content

Commit

Permalink
test that coercions still work under randomization
Browse files Browse the repository at this point in the history
  • Loading branch information
the8472 committed Jan 10, 2025
1 parent 8b1de16 commit d89b6d5
Showing 1 changed file with 40 additions and 8 deletions.
48 changes: 40 additions & 8 deletions tests/ui/layout/randomize.rs
Original file line number Diff line number Diff line change
@@ -1,18 +1,23 @@
//@ build-pass
//@ run-pass
//@ revisions: normal randomize-layout
//@ [randomize-layout]compile-flags: -Zrandomize-layout
//@ [randomize-layout]compile-flags: -Zrandomize-layout -Zlayout-seed=2

#![crate_type = "lib"]
#![feature(offset_of_enum)]

struct Foo<T>(u32, T, u8);
use std::ptr;

struct Wrapper<T>(T);

// these types only have their field offsets taken, they're never constructed
#[allow(dead_code)]
pub struct Foo<T>(u32, T, u8);
#[allow(dead_code)]
pub struct Wrapper<T>(T);
#[repr(transparent)]
struct TransparentWrapper(u16);
#[allow(dead_code)]
pub struct TransparentWrapper(u16);

const _: () = {
// behavior of the current implementation, not guaranteed
// Behavior of the current non-randomized implementation, not guaranteed
#[cfg(not(randomize_layout))]
assert!(std::mem::offset_of!(Foo::<u16>, 1) == std::mem::offset_of!(Foo::<Wrapper<u16>>, 1));

Expand All @@ -21,10 +26,37 @@ const _: () = {
assert!(std::mem::offset_of!(Foo::<u16>, 1) != std::mem::offset_of!(Foo::<Wrapper<u16>>, 1));

// Even transparent wrapper inner types get a different layout since associated type
// pecialization could result in the outer type behaving differently depending on the exact
// specialization could result in the outer type behaving differently depending on the exact
// inner type.
#[cfg(randomize_layout)]
assert!(
std::mem::offset_of!(Foo::<u16>, 1) != std::mem::offset_of!(Foo::<TransparentWrapper>, 1)
);

// Currently all fn pointers are treated interchangably even with randomization. Not guaranteed.
// Associated type specialization could also break this.
assert!(
std::mem::offset_of!(Foo::<fn(u32)>, 1) == std::mem::offset_of!(Foo::<fn() -> usize>, 1)
);

// But subtype coercions must always result in the same layout.
assert!(
std::mem::offset_of!(Foo::<fn(&u32)>, 1) == std::mem::offset_of!(Foo::<fn(&'static u32)>, 1)
);

// Randomization must uphold NPO guarantees
assert!(std::mem::offset_of!(Option::<&usize>, Some.0) == 0);
assert!(std::mem::offset_of!(Result::<&usize, ()>, Ok.0) == 0);
};

#[allow(dead_code)]
struct Unsizable<T: ?Sized>(usize, T);

fn main() {
// offset_of doesn't let us probe the unsized field, check at runtime.
let x = &Unsizable::<[u32; 4]>(0, [0; 4]);
let y: &Unsizable::<[u32]> = x;

// type coercion must not change the layout.
assert_eq!(ptr::from_ref(&x.1).addr(), ptr::from_ref(&y.1).addr());
}

0 comments on commit d89b6d5

Please sign in to comment.