65 lines
1.8 KiB
Rust
65 lines
1.8 KiB
Rust
#[cfg(test)]
|
|
#[cfg(feature = "hooks")]
|
|
mod tests {
|
|
use core::{ffi::c_char, ptr::null_mut};
|
|
|
|
use asan::{expect_panic, hooks::stpncpy::stpncpy};
|
|
|
|
#[test]
|
|
fn test_stpncpy_zero_length() {
|
|
let ret = unsafe { stpncpy(null_mut(), null_mut(), 0) };
|
|
assert_eq!(ret, null_mut());
|
|
}
|
|
|
|
#[test]
|
|
fn test_stpncpy_dst_null() {
|
|
let src = [0u8; 10];
|
|
expect_panic();
|
|
unsafe { stpncpy(null_mut(), src.as_ptr() as *const c_char, 10) };
|
|
unreachable!();
|
|
}
|
|
|
|
#[test]
|
|
fn test_stpncpy_src_null() {
|
|
let dst = [0u8; 10];
|
|
expect_panic();
|
|
unsafe { stpncpy(dst.as_ptr() as *mut c_char, null_mut(), dst.len()) };
|
|
unreachable!();
|
|
}
|
|
|
|
#[test]
|
|
fn test_stpncpy_full() {
|
|
let src = [0xffu8; 10];
|
|
let dst = [0u8; 10];
|
|
let ret = unsafe {
|
|
stpncpy(
|
|
dst.as_ptr() as *mut c_char,
|
|
src.as_ptr() as *const c_char,
|
|
dst.len(),
|
|
)
|
|
};
|
|
assert_eq!(ret, unsafe { dst.as_ptr().add(dst.len()) as *mut c_char });
|
|
}
|
|
|
|
#[test]
|
|
fn test_stpncpy_partial() {
|
|
let mut vec = c"abcdefghijklmnopqrstuvwxyz".to_bytes().to_vec();
|
|
let dst = vec.as_mut_slice();
|
|
let src = c"uvwxyz".to_bytes();
|
|
let ret = unsafe {
|
|
stpncpy(
|
|
dst.as_ptr() as *mut c_char,
|
|
src.as_ptr() as *const c_char,
|
|
dst.len(),
|
|
)
|
|
};
|
|
assert_eq!(ret, unsafe { dst.as_ptr().add(dst.len()) as *mut c_char });
|
|
let expected = c"uvwxyz".to_bytes();
|
|
expected
|
|
.iter()
|
|
.zip(dst.iter())
|
|
.for_each(|(x, y)| assert_eq!(*x, *y));
|
|
dst.iter().skip(src.len()).for_each(|x| assert_eq!(*x, 0));
|
|
}
|
|
}
|