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

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

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

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

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

    #[test]
    fn test_strstr_ct_too_long() {
        let data1 = c"abcdefghij";
        let data2 = c"abcdefghijk";
        let ret = unsafe {
            strstr(
                data1.as_ptr() as *const c_char,
                data2.as_ptr() as *const c_char,
            )
        };
        assert_eq!(ret, null_mut());
    }

    #[test]
    fn test_strstr_found_at_start() {
        let data1 = c"abcdefghijk";
        let data2 = c"abc";
        let ret = unsafe {
            strstr(
                data1.as_ptr() as *const c_char,
                data2.as_ptr() as *const c_char,
            )
        };
        assert_eq!(ret, data1.as_ptr() as *mut c_char);
    }

    #[test]
    fn test_strstr_found_at_end() {
        let data1 = c"abcdefghijk";
        let data2 = c"ijk";
        let ret = unsafe {
            strstr(
                data1.as_ptr() as *const c_char,
                data2.as_ptr() as *const c_char,
            )
        };
        assert_eq!(ret, unsafe {
            data1
                .as_ptr()
                .add(data1.count_bytes() - data2.count_bytes()) as *mut c_char
        });
    }

    #[test]
    fn test_strstr_found_in_middle() {
        let data1 = c"abcdefghijk";
        let data2 = c"def";
        let ret = unsafe {
            strstr(
                data1.as_ptr() as *const c_char,
                data2.as_ptr() as *const c_char,
            )
        };
        assert_eq!(ret, unsafe { data1.as_ptr().add(3) as *mut c_char });
    }

    #[test]
    fn test_strstr_case_not_ignored() {
        let data1 = c"abcdefghijklmnopqrstuvwxyz";
        let data2 = c"ABCDEFGHIJKLMNOPQRSTUVWXYZ";
        let ret = unsafe {
            strstr(
                data1.as_ptr() as *const c_char,
                data2.as_ptr() as *const c_char,
            )
        };
        assert_eq!(ret, null_mut());
    }
}