Source code
Revision control
Copy as Markdown
Other Tools
#![allow(clippy::unnecessary_cast)]↩
#![allow(clippy::manual_slice_size_calculation)]↩
↩
use core::{↩
mem::size_of,↩
num::{NonZeroU32, NonZeroU8},↩
};↩
↩
use bytemuck::{checked::CheckedCastError, *};↩
↩
#[test]↩
fn test_try_cast_slice() {↩
// some align4 data↩
let nonzero_u32_slice: &[NonZeroU32] = &[↩
NonZeroU32::new(4).unwrap(),↩
NonZeroU32::new(5).unwrap(),↩
NonZeroU32::new(6).unwrap(),↩
];↩
↩
// contains bytes with invalid bitpattern for NonZeroU8↩
assert_eq!(↩
checked::try_cast_slice::<NonZeroU32, NonZeroU8>(nonzero_u32_slice),↩
Err(CheckedCastError::InvalidBitPattern)↩
);↩
↩
// the same data as align1↩
let the_bytes: &[u8] = checked::try_cast_slice(nonzero_u32_slice).unwrap();↩
↩
assert_eq!(↩
nonzero_u32_slice.as_ptr() as *const NonZeroU32 as usize,↩
the_bytes.as_ptr() as *const u8 as usize↩
);↩
assert_eq!(↩
nonzero_u32_slice.len() * size_of::<NonZeroU32>(),↩
the_bytes.len() * size_of::<u8>()↩
);↩
↩
// by taking one byte off the front, we're definitely mis-aligned for↩
// NonZeroU32.↩
let mis_aligned_bytes = &the_bytes[1..];↩
assert_eq!(↩
checked::try_cast_slice::<u8, NonZeroU32>(mis_aligned_bytes),↩
Err(CheckedCastError::PodCastError(↩
PodCastError::TargetAlignmentGreaterAndInputNotAligned↩
))↩
);↩
↩
// by taking one byte off the end, we're aligned but would have slop bytes for↩
// NonZeroU32↩
let the_bytes_len_minus1 = the_bytes.len() - 1;↩
let slop_bytes = &the_bytes[..the_bytes_len_minus1];↩
assert_eq!(↩
checked::try_cast_slice::<u8, NonZeroU32>(slop_bytes),↩
Err(CheckedCastError::PodCastError(PodCastError::OutputSliceWouldHaveSlop))↩
);↩
↩
// if we don't mess with it we can up-alignment cast↩
checked::try_cast_slice::<u8, NonZeroU32>(the_bytes).unwrap();↩
}↩
↩
#[test]↩
fn test_try_cast_slice_mut() {↩
// some align4 data↩
let u32_slice: &mut [u32] = &mut [4, 5, 6];↩
↩
// contains bytes with invalid bitpattern for NonZeroU8↩
assert_eq!(↩
checked::try_cast_slice_mut::<u32, NonZeroU8>(u32_slice),↩
Err(CheckedCastError::InvalidBitPattern)↩
);↩
↩
// some align4 data↩
let u32_slice: &mut [u32] = &mut [0x4444_4444, 0x5555_5555, 0x6666_6666];↩
let u32_len = u32_slice.len();↩
let u32_ptr = u32_slice.as_ptr();↩
↩
// the same data as align1, nonzero bytes↩
let the_nonzero_bytes: &mut [NonZeroU8] =↩
checked::try_cast_slice_mut(u32_slice).unwrap();↩
let the_nonzero_bytes_len = the_nonzero_bytes.len();↩
let the_nonzero_bytes_ptr = the_nonzero_bytes.as_ptr();↩
↩
assert_eq!(↩
u32_ptr as *const u32 as usize,↩
the_nonzero_bytes_ptr as *const NonZeroU8 as usize↩
);↩
assert_eq!(↩
u32_len * size_of::<u32>(),↩
the_nonzero_bytes_len * size_of::<NonZeroU8>()↩
);↩
↩
// the same data as align1↩
let the_bytes: &mut [u8] = checked::try_cast_slice_mut(u32_slice).unwrap();↩
let the_bytes_len = the_bytes.len();↩
let the_bytes_ptr = the_bytes.as_ptr();↩
↩
assert_eq!(↩
u32_ptr as *const u32 as usize,↩
the_bytes_ptr as *const u8 as usize↩
);↩
assert_eq!(↩
u32_len * size_of::<u32>(),↩
the_bytes_len * size_of::<NonZeroU8>()↩
);↩
↩
// by taking one byte off the front, we're definitely mis-aligned for u32.↩
let mis_aligned_bytes = &mut the_bytes[1..];↩
assert_eq!(↩
checked::try_cast_slice_mut::<u8, NonZeroU32>(mis_aligned_bytes),↩
Err(CheckedCastError::PodCastError(↩
PodCastError::TargetAlignmentGreaterAndInputNotAligned↩
))↩
);↩
↩
// by taking one byte off the end, we're aligned but would have slop bytes for↩
// NonZeroU32↩
let the_bytes_len_minus1 = the_bytes.len() - 1;↩
let slop_bytes = &mut the_bytes[..the_bytes_len_minus1];↩
assert_eq!(↩
checked::try_cast_slice_mut::<u8, NonZeroU32>(slop_bytes),↩
Err(CheckedCastError::PodCastError(PodCastError::OutputSliceWouldHaveSlop))↩
);↩
↩
// if we don't mess with it we can up-alignment cast, since there are no↩
// zeroes in the original slice↩
checked::try_cast_slice_mut::<u8, NonZeroU32>(the_bytes).unwrap();↩
}↩
↩
#[test]↩
fn test_types() {↩
let _: NonZeroU32 = checked::cast(1.0_f32);↩
let _: &mut NonZeroU32 = checked::cast_mut(&mut 1.0_f32);↩
let _: &NonZeroU32 = checked::cast_ref(&1.0_f32);↩
let _: &[NonZeroU32] = checked::cast_slice(&[1.0_f32]);↩
let _: &mut [NonZeroU32] = checked::cast_slice_mut(&mut [1.0_f32]);↩
//↩
let _: Result<NonZeroU32, CheckedCastError> = checked::try_cast(1.0_f32);↩
let _: Result<&mut NonZeroU32, CheckedCastError> =↩
checked::try_cast_mut(&mut 1.0_f32);↩
let _: Result<&NonZeroU32, CheckedCastError> =↩
checked::try_cast_ref(&1.0_f32);↩
let _: Result<&[NonZeroU32], CheckedCastError> =↩
checked::try_cast_slice(&[1.0_f32]);↩
let _: Result<&mut [NonZeroU32], CheckedCastError> =↩
checked::try_cast_slice_mut(&mut [1.0_f32]);↩
}↩
↩
#[test]↩
fn test_try_pod_read_unaligned() {↩
let u32s = [0xaabbccdd, 0x11223344_u32];↩
let bytes = bytemuck::checked::cast_slice::<u32, u8>(&u32s);↩
↩
#[cfg(target_endian = "big")]↩
assert_eq!(↩
checked::try_pod_read_unaligned::<NonZeroU32>(&bytes[1..5]),↩
Ok(NonZeroU32::new(0xbbccdd11).unwrap())↩
);↩
#[cfg(target_endian = "little")]↩
assert_eq!(↩
checked::try_pod_read_unaligned::<NonZeroU32>(&bytes[1..5]),↩
Ok(NonZeroU32::new(0x44aabbcc).unwrap())↩
);↩
↩
let u32s = [0; 2];↩
let bytes = bytemuck::checked::cast_slice::<u32, u8>(&u32s);↩
↩
assert_eq!(↩
checked::try_pod_read_unaligned::<NonZeroU32>(&bytes[1..5]),↩
Err(CheckedCastError::InvalidBitPattern)↩
);↩
}↩
↩
#[test]↩
fn test_try_from_bytes() {↩
let nonzero_u32s = [↩
NonZeroU32::new(0xaabbccdd).unwrap(),↩
NonZeroU32::new(0x11223344).unwrap(),↩
];↩
let bytes = bytemuck::checked::cast_slice::<NonZeroU32, u8>(&nonzero_u32s);↩
assert_eq!(↩
checked::try_from_bytes::<NonZeroU32>(&bytes[..4]),↩
Ok(&nonzero_u32s[0])↩
);↩
assert_eq!(↩
checked::try_from_bytes::<NonZeroU32>(&bytes[..5]),↩
Err(CheckedCastError::PodCastError(PodCastError::SizeMismatch))↩
);↩
assert_eq!(↩
checked::try_from_bytes::<NonZeroU32>(&bytes[..3]),↩
Err(CheckedCastError::PodCastError(PodCastError::SizeMismatch))↩
);↩
assert_eq!(↩
checked::try_from_bytes::<NonZeroU32>(&bytes[1..5]),↩
Err(CheckedCastError::PodCastError(↩
PodCastError::TargetAlignmentGreaterAndInputNotAligned↩
))↩
);↩
↩
let zero_u32s = [0, 0x11223344_u32];↩
let bytes = bytemuck::checked::cast_slice::<u32, u8>(&zero_u32s);↩
assert_eq!(↩
checked::try_from_bytes::<NonZeroU32>(&bytes[..4]),↩
Err(CheckedCastError::InvalidBitPattern)↩
);↩
assert_eq!(↩
checked::try_from_bytes::<NonZeroU32>(&bytes[4..]),↩
Ok(&NonZeroU32::new(zero_u32s[1]).unwrap())↩
);↩
assert_eq!(↩
checked::try_from_bytes::<NonZeroU32>(&bytes[..5]),↩
Err(CheckedCastError::PodCastError(PodCastError::SizeMismatch))↩
);↩
assert_eq!(↩
checked::try_from_bytes::<NonZeroU32>(&bytes[..3]),↩
Err(CheckedCastError::PodCastError(PodCastError::SizeMismatch))↩
);↩
assert_eq!(↩
checked::try_from_bytes::<NonZeroU32>(&bytes[1..5]),↩
Err(CheckedCastError::PodCastError(↩
PodCastError::TargetAlignmentGreaterAndInputNotAligned↩
))↩
);↩
}↩
↩
#[test]↩
fn test_try_from_bytes_mut() {↩
let a = 0xaabbccdd_u32;↩
let b = 0x11223344_u32;↩
let mut u32s = [a, b];↩
let bytes = bytemuck::checked::cast_slice_mut::<u32, u8>(&mut u32s);↩
assert_eq!(↩
checked::try_from_bytes_mut::<NonZeroU32>(&mut bytes[..4]),↩
Ok(&mut NonZeroU32::new(a).unwrap())↩
);↩
assert_eq!(↩
checked::try_from_bytes_mut::<NonZeroU32>(&mut bytes[4..]),↩
Ok(&mut NonZeroU32::new(b).unwrap())↩
);↩
assert_eq!(↩
checked::try_from_bytes_mut::<NonZeroU32>(&mut bytes[..5]),↩
Err(CheckedCastError::PodCastError(PodCastError::SizeMismatch))↩
);↩
assert_eq!(↩
checked::try_from_bytes_mut::<NonZeroU32>(&mut bytes[..3]),↩
Err(CheckedCastError::PodCastError(PodCastError::SizeMismatch))↩
);↩
assert_eq!(↩
checked::try_from_bytes::<NonZeroU32>(&bytes[1..5]),↩
Err(CheckedCastError::PodCastError(↩
PodCastError::TargetAlignmentGreaterAndInputNotAligned↩
))↩
);↩
↩
let mut u32s = [0, b];↩
let bytes = bytemuck::checked::cast_slice_mut::<u32, u8>(&mut u32s);↩
assert_eq!(↩
checked::try_from_bytes_mut::<NonZeroU32>(&mut bytes[..4]),↩
Err(CheckedCastError::InvalidBitPattern)↩
);↩
assert_eq!(↩
checked::try_from_bytes_mut::<NonZeroU32>(&mut bytes[4..]),↩
Ok(&mut NonZeroU32::new(b).unwrap())↩
);↩
assert_eq!(↩
checked::try_from_bytes_mut::<NonZeroU32>(&mut bytes[..5]),↩
Err(CheckedCastError::PodCastError(PodCastError::SizeMismatch))↩
);↩
assert_eq!(↩
checked::try_from_bytes_mut::<NonZeroU32>(&mut bytes[..3]),↩
Err(CheckedCastError::PodCastError(PodCastError::SizeMismatch))↩
);↩
assert_eq!(↩
checked::try_from_bytes::<NonZeroU32>(&bytes[1..5]),↩
Err(CheckedCastError::PodCastError(↩
PodCastError::TargetAlignmentGreaterAndInputNotAligned↩
))↩
);↩
}↩
↩
#[test]↩
fn test_from_bytes() {↩
let abcd = 0xaabbccdd_u32;↩
let aligned_bytes = bytemuck::bytes_of(&abcd);↩
assert_eq!(↩
checked::from_bytes::<NonZeroU32>(aligned_bytes),↩
&NonZeroU32::new(abcd).unwrap()↩
);↩
assert!(core::ptr::eq(↩
checked::from_bytes(aligned_bytes) as *const NonZeroU32 as *const u32,↩
&abcd↩
));↩
}↩
↩
#[test]↩
fn test_from_bytes_mut() {↩
let mut a = 0xaabbccdd_u32;↩
let a_addr = &a as *const _ as usize;↩
let aligned_bytes = bytemuck::bytes_of_mut(&mut a);↩
assert_eq!(↩
*checked::from_bytes_mut::<NonZeroU32>(aligned_bytes),↩
NonZeroU32::new(0xaabbccdd).unwrap()↩
);↩
assert_eq!(↩
checked::from_bytes_mut::<NonZeroU32>(aligned_bytes) as *const NonZeroU32↩
as usize,↩
a_addr↩
);↩
}↩
↩
// like #[should_panic], but can be a part of another test, instead of requiring↩
// it to be it's own test.↩
macro_rules! should_panic {↩
($ex:expr) => {↩
assert!(↩
std::panic::catch_unwind(|| {↩
let _ = $ex;↩
})↩
.is_err(),↩
concat!("should have panicked: `", stringify!($ex), "`")↩
);↩
};↩
}↩
↩
#[test]↩
fn test_panics() {↩
should_panic!(checked::cast::<u32, NonZeroU32>(0));↩
should_panic!(checked::cast_ref::<u32, NonZeroU32>(&0));↩
should_panic!(checked::cast_mut::<u32, NonZeroU32>(&mut 0));↩
should_panic!(checked::cast_slice::<u8, NonZeroU32>(&[1u8, 2u8]));↩
should_panic!(checked::cast_slice_mut::<u8, NonZeroU32>(&mut [1u8, 2u8]));↩
should_panic!(checked::from_bytes::<NonZeroU32>(&[1u8, 2]));↩
should_panic!(checked::from_bytes::<NonZeroU32>(&[1u8, 2, 3, 4, 5]));↩
should_panic!(checked::from_bytes_mut::<NonZeroU32>(&mut [1u8, 2]));↩
should_panic!(checked::from_bytes_mut::<NonZeroU32>(&mut [1u8, 2, 3, 4, 5]));↩
// use cast_slice on some u32s to get some align>=4 bytes, so we can know↩
// we'll give from_bytes unaligned ones.↩
let aligned_bytes = bytemuck::cast_slice::<u32, u8>(&[0, 0]);↩
should_panic!(checked::from_bytes::<NonZeroU32>(aligned_bytes));↩
should_panic!(checked::from_bytes::<NonZeroU32>(&aligned_bytes[1..5]));↩
should_panic!(checked::pod_read_unaligned::<NonZeroU32>(↩
&aligned_bytes[1..5]↩
));↩
}↩
↩
#[test]↩
fn test_char() {↩
assert_eq!(checked::try_cast::<u32, char>(0), Ok('\0'));↩
assert_eq!(checked::try_cast::<u32, char>(0xd7ff), Ok('\u{d7ff}'));↩
assert_eq!(↩
checked::try_cast::<u32, char>(0xd800),↩
Err(CheckedCastError::InvalidBitPattern)↩
);↩
assert_eq!(↩
checked::try_cast::<u32, char>(0xdfff),↩
Err(CheckedCastError::InvalidBitPattern)↩
);↩
assert_eq!(checked::try_cast::<u32, char>(0xe000), Ok('\u{e000}'));↩
assert_eq!(checked::try_cast::<u32, char>(0x10ffff), Ok('\u{10ffff}'));↩
assert_eq!(↩
checked::try_cast::<u32, char>(0x110000),↩
Err(CheckedCastError::InvalidBitPattern)↩
);↩
assert_eq!(↩
checked::try_cast::<u32, char>(-1i32 as u32),↩
Err(CheckedCastError::InvalidBitPattern)↩
);↩
}↩
↩
#[test]↩
fn test_bool() {↩
assert_eq!(checked::try_cast::<u8, bool>(0), Ok(false));↩
assert_eq!(checked::try_cast::<u8, bool>(1), Ok(true));↩
for i in 2..=255 {↩
assert_eq!(↩
checked::try_cast::<u8, bool>(i),↩
Err(CheckedCastError::InvalidBitPattern)↩
);↩
}↩
↩
assert_eq!(checked::try_from_bytes::<bool>(&[1]), Ok(&true));↩
assert_eq!(↩
checked::try_from_bytes::<bool>(&[3]),↩
Err(CheckedCastError::InvalidBitPattern)↩
);↩
assert_eq!(↩
checked::try_from_bytes::<bool>(&[0, 1]),↩
Err(CheckedCastError::PodCastError(PodCastError::SizeMismatch))↩
);↩
}↩
↩
#[test]↩
fn test_all_nonzero() {↩
use core::num::*;↩
macro_rules! test_nonzero {↩
($nonzero:ty: $primitive:ty) => {↩
assert_eq!(↩
checked::try_cast::<$primitive, $nonzero>(0),↩
Err(CheckedCastError::InvalidBitPattern)↩
);↩
assert_eq!(↩
checked::try_cast::<$primitive, $nonzero>(1),↩
Ok(<$nonzero>::new(1).unwrap())↩
);↩
};↩
}↩
↩
test_nonzero!(NonZeroU8: u8);↩
test_nonzero!(NonZeroI8: i8);↩
test_nonzero!(NonZeroU16: u16);↩
test_nonzero!(NonZeroI16: i16);↩
test_nonzero!(NonZeroU32: u32);↩
test_nonzero!(NonZeroI32: i32);↩
test_nonzero!(NonZeroU64: u64);↩
test_nonzero!(NonZeroI64: i64);↩
test_nonzero!(NonZeroU128: u128);↩
test_nonzero!(NonZeroI128: i128);↩
test_nonzero!(NonZeroUsize: usize);↩
test_nonzero!(NonZeroIsize: isize);↩
}↩