FRET-LibAFL/libafl_qemu/librasan/asan/tests/hooks_strcat.rs

#[cfg(test)]
#[cfg(feature = "hooks")]
mod tests {
    use core::{
        ffi::c_char,
        ptr::{null, null_mut},
    };

    use asan::{expect_panic, hooks::strcat::strcat};

    #[test]
    fn test_strcat_null_s() {
        expect_panic();
        let data = [0u8; 10];
        unsafe { strcat(null_mut(), data.as_ptr() as *const c_char) };
        unreachable!();
    }

    #[test]
    fn test_strcat_null_s2() {
        expect_panic();
        let data = [0u8; 10];
        unsafe { strcat(data.as_ptr() as *mut c_char, null()) };
        unreachable!();
    }

    #[test]
    fn test_strcat_zero_length_both() {
        let data = [0u8; 10];
        let ret = unsafe { strcat(data.as_ptr() as *mut c_char, data.as_ptr() as *const c_char) };
        assert_eq!(ret, data.as_ptr() as *mut c_char);
    }

    #[test]
    fn test_strcat_appends() {
        let mut vec = "abcde\0zzzzzzzzzzzzzzz".as_bytes().to_vec();
        let s = vec.as_mut_slice();
        let ct = c"fghij";
        let ret = unsafe { strcat(s.as_ptr() as *mut c_char, ct.as_ptr() as *const c_char) };
        assert_eq!(ret, s.as_ptr() as *mut c_char);
        let expected = c"abcdefghij";
        expected
            .to_bytes()
            .iter()
            .zip(s.iter())
            .for_each(|(x, y)| assert_eq!(*x, *y));
    }
}