58 lines
1.6 KiB
Rust
58 lines
1.6 KiB
Rust
#[cfg(test)]
|
|
#[cfg(feature = "hooks")]
|
|
mod tests {
|
|
use core::{ffi::c_char, ptr::null_mut};
|
|
|
|
use asan::{expect_panic, hooks::stpcpy::stpcpy};
|
|
|
|
#[test]
|
|
fn test_stpcpy_dst_null() {
|
|
let src = [0u8; 10];
|
|
expect_panic();
|
|
unsafe { stpcpy(null_mut(), src.as_ptr() as *const c_char) };
|
|
unreachable!();
|
|
}
|
|
|
|
#[test]
|
|
fn test_stpcpy_src_null() {
|
|
let dst = [0u8; 10];
|
|
expect_panic();
|
|
unsafe { stpcpy(dst.as_ptr() as *mut c_char, null_mut()) };
|
|
unreachable!();
|
|
}
|
|
|
|
#[test]
|
|
fn test_stpcpy_full() {
|
|
let src = [0xffu8; 10];
|
|
let dst = [0u8; 10];
|
|
let ret = unsafe { stpcpy(dst.as_ptr() as *mut c_char, src.as_ptr() as *const c_char) };
|
|
assert_eq!(ret, unsafe { dst.as_ptr().add(src.len()) as *mut c_char });
|
|
}
|
|
|
|
#[test]
|
|
fn test_stpcpy_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 {
|
|
stpcpy(
|
|
dst.as_ptr().add(2) as *mut c_char,
|
|
src.as_ptr() as *const c_char,
|
|
)
|
|
};
|
|
assert_eq!(ret, unsafe {
|
|
dst.as_ptr().add(2).add(src.len()) as *mut c_char
|
|
});
|
|
let expected = c"abuvwxyz".to_bytes();
|
|
expected
|
|
.iter()
|
|
.zip(dst.iter())
|
|
.for_each(|(x, y)| assert_eq!(*x, *y));
|
|
let remaining = c"jklmnopqrstuvwxyz".to_bytes();
|
|
remaining
|
|
.iter()
|
|
.zip(dst.iter().skip(2).skip(src.len()).skip(1))
|
|
.for_each(|(x, y)| assert_eq!(*x, *y));
|
|
}
|
|
}
|