rust/tests/run-pass/union-overwrite.rs
Oliver Schneider a6734cd890
Fix unions
2017-06-20 14:28:18 +02:00

82 lines
2.0 KiB
Rust

// Copyright 2016 The Rust Project Developers. See the COPYRIGHT
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.
#![feature(untagged_unions)]
#![allow(unions_with_drop_fields)]
#[repr(C)]
struct Pair<T, U>(T, U);
#[repr(C)]
struct Triple<T>(T, T, T);
#[repr(C)]
union U<A, B> {
a: Pair<A, A>,
b: B,
}
#[repr(C)]
union W<A, B> {
a: A,
b: B,
}
#[cfg(target_endian = "little")]
unsafe fn check() {
let mut u = U::<u8, u16> { b: 0xDE_DE };
u.a.0 = 0xBE;
assert_eq!(u.b, 0xDE_BE);
let mut u = U::<u16, u32> { b: 0xDEAD_DEAD };
u.a.0 = 0xBEEF;
assert_eq!(u.b, 0xDEAD_BEEF);
let mut u = U::<u32, u64> { b: 0xDEADBEEF_DEADBEEF };
u.a.0 = 0xBAADF00D;
assert_eq!(u.b, 0xDEADBEEF_BAADF00D);
let mut w = W::<Pair<Triple<u8>, u8>, u32> { b: 0xDEAD_DEAD };
w.a.0 = Triple(0, 0, 0);
assert_eq!(w.b, 0xDE00_0000);
let mut w = W::<Pair<u8, Triple<u8>>, u32> { b: 0xDEAD_DEAD };
w.a.1 = Triple(0, 0, 0);
assert_eq!(w.b, 0x0000_00AD);
}
#[cfg(target_endian = "big")]
unsafe fn check() {
let mut u = U::<u8, u16> { b: 0xDE_DE };
u.a.0 = 0xBE;
assert_eq!(u.b, 0xBE_DE);
let mut u = U::<u16, u32> { b: 0xDEAD_DEAD };
u.a.0 = 0xBEEF;
assert_eq!(u.b, 0xBEEF_DEAD);
let mut u = U::<u32, u64> { b: 0xDEADBEEF_DEADBEEF };
u.a.0 = 0xBAADF00D;
assert_eq!(u.b, 0xBAADF00D_DEADBEEF);
let mut w = W::<Pair<Triple<u8>, u8>, u32> { b: 0xDEAD_DEAD };
w.a.0 = Triple(0, 0, 0);
assert_eq!(w.b, 0x0000_00AD);
let mut w = W::<Pair<u8, Triple<u8>>, u32> { b: 0xDEAD_DEAD };
w.a.1 = Triple(0, 0, 0);
assert_eq!(w.b, 0xDE00_0000);
}
fn main() {
unsafe {
check();
}
}