src/test/ui/union/union-overwrite.rs

   1 // run-pass
   2 // revisions: mirunsafeck thirunsafeck
   3 // [thirunsafeck]compile-flags: -Z thir-unsafeck
   4
   5 #[repr(C)]
   6 #[derive(Copy, Clone)]
   7 struct Pair<T, U>(T, U);
   8 #[repr(C)]
   9 #[derive(Copy, Clone)]
  10 struct Triple<T>(T, T, T);
  11
  12 #[repr(C)]
  13 union U<A, B>
  14 where
  15     A: Copy, B: Copy
  16 {
  17     a: Pair<A, A>,
  18     b: B,
  19 }
  20
  21 #[repr(C)]
  22 union W<A, B>
  23 where
  24     A: Copy, B: Copy
  25 {
  26     a: A,
  27     b: B,
  28 }
  29
  30 #[cfg(target_endian = "little")]
  31 unsafe fn check() {
  32     let mut u = U::<u8, u16> { b: 0xDE_DE };
  33     u.a.0 = 0xBE;
  34     assert_eq!(u.b, 0xDE_BE);
  35
  36     let mut u = U::<u16, u32> { b: 0xDEAD_DEAD };
  37     u.a.0 = 0xBEEF;
  38     assert_eq!(u.b, 0xDEAD_BEEF);
  39
  40     let mut u = U::<u32, u64> { b: 0xDEADBEEF_DEADBEEF };
  41     u.a.0 = 0xBAADF00D;
  42     assert_eq!(u.b, 0xDEADBEEF_BAADF00D);
  43
  44     let mut w = W::<Pair<Triple<u8>, u8>, u32> { b: 0xDEAD_DEAD };
  45     w.a.0 = Triple(0, 0, 0);
  46     assert_eq!(w.b, 0xDE00_0000);
  47
  48     let mut w = W::<Pair<u8, Triple<u8>>, u32> { b: 0xDEAD_DEAD };
  49     w.a.1 = Triple(0, 0, 0);
  50     assert_eq!(w.b, 0x0000_00AD);
  51 }
  52
  53 #[cfg(target_endian = "big")]
  54 unsafe fn check() {
  55     let mut u = U::<u8, u16> { b: 0xDE_DE };
  56     u.a.0 = 0xBE;
  57     assert_eq!(u.b, 0xBE_DE);
  58
  59     let mut u = U::<u16, u32> { b: 0xDEAD_DEAD };
  60     u.a.0 = 0xBEEF;
  61     assert_eq!(u.b, 0xBEEF_DEAD);
  62
  63     let mut u = U::<u32, u64> { b: 0xDEADBEEF_DEADBEEF };
  64     u.a.0 = 0xBAADF00D;
  65     assert_eq!(u.b, 0xBAADF00D_DEADBEEF);
  66
  67     let mut w = W::<Pair<Triple<u8>, u8>, u32> { b: 0xDEAD_DEAD };
  68     w.a.0 = Triple(0, 0, 0);
  69     assert_eq!(w.b, 0x0000_00AD);
  70
  71     let mut w = W::<Pair<u8, Triple<u8>>, u32> { b: 0xDEAD_DEAD };
  72     w.a.1 = Triple(0, 0, 0);
  73     assert_eq!(w.b, 0xDE00_0000);
  74 }
  75
  76 fn main() {
  77     unsafe {
  78         check();
  79     }
  80 }