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

#[cfg(test)]
#[cfg(feature = "guest")]
mod tests {
    use std::sync::Mutex;

    use asan::{
        GuestAddr,
        mmap::libc::LibcMmap,
        shadow::{
            Shadow,
            guest::{DefaultShadowLayout, GuestShadow, GuestShadowError},
        },
        symbols::dlsym::{DlSymSymbols, LookupTypeNext},
    };
    use spin::Lazy;

    type GS = GuestShadow<LibcMmap<DlSymSymbols<LookupTypeNext>>, DefaultShadowLayout>;

    const ALIGN: usize = GS::ALLOC_ALIGN_SIZE;

    static INIT_ONCE: Lazy<Mutex<()>> = Lazy::new(|| {
        {
            env_logger::init();
        };
        Mutex::new(())
    });

    fn get_shadow() -> GS {
        drop(INIT_ONCE.lock().unwrap());
        GS::new().unwrap()
    }

    #[test]
    fn test_init() {
        get_shadow();
    }

    // [0x10007fff8000, 0x7fffffffffff] 	HighMem
    // [0x02008fff7000, 0x10007fff7fff] 	HighShadow
    // [0x00008fff7000, 0x02008fff6fff] 	ShadowGap
    // [0x00007fff8000, 0x00008fff6fff] 	LowShadow
    // [0x000000000000, 0x00007fff7fff] 	LowMem

    // [0x40000000, 0xffffffff] 	HighMem
    // [0x28000000, 0x3fffffff] 	HighShadow
    // [0x24000000, 0x27ffffff] 	ShadowGap
    // [0x20000000, 0x23ffffff] 	LowShadow
    // [0x00000000, 0x1fffffff] 	LowMem
    #[test]
    fn test_unposion_bottom_of_low_mem() {
        let mut shadow = get_shadow();
        shadow.unpoison(GS::LOW_MEM_OFFSET, 0x8).unwrap();
    }

    #[test]
    fn test_unposion_top_of_low_mem() {
        let mut shadow = get_shadow();
        shadow.unpoison(GS::LOW_MEM_LIMIT - 0x7, 0x8).unwrap();
    }

    #[test]
    fn test_unposion_bottom_of_low_shadow() {
        let mut shadow = get_shadow();
        let result = shadow.unpoison(GS::LOW_SHADOW_OFFSET, 0x8);
        assert_eq!(
            result,
            Err(GuestShadowError::InvalidMemoryAddress(
                GS::LOW_SHADOW_OFFSET
            ))
        );
    }

    #[test]
    fn test_unposion_top_of_low_shadow() {
        let mut shadow = get_shadow();
        const ADDR: GuestAddr = GS::LOW_SHADOW_OFFSET + GS::LOW_SHADOW_SIZE - 8;
        let result = shadow.unpoison(ADDR, 0x8);
        assert_eq!(result, Err(GuestShadowError::InvalidMemoryAddress(ADDR)));
    }

    #[test]
    fn test_unposion_bottom_of_high_shadow() {
        let mut shadow = get_shadow();
        let result = shadow.unpoison(GS::HIGH_SHADOW_OFFSET, 0x8);
        assert_eq!(
            result,
            Err(GuestShadowError::InvalidMemoryAddress(
                GS::HIGH_SHADOW_OFFSET
            ))
        );
    }

    #[test]
    fn test_unposion_top_of_high_shadow() {
        let mut shadow = get_shadow();
        const ADDR: GuestAddr = GS::HIGH_SHADOW_OFFSET + GS::HIGH_SHADOW_SIZE - 8;
        let result = shadow.unpoison(ADDR, 0x8);
        assert_eq!(result, Err(GuestShadowError::InvalidMemoryAddress(ADDR)));
    }

    #[test]
    fn test_unposion_bottom_of_high_mem() {
        let mut shadow = get_shadow();
        shadow.unpoison(GS::HIGH_MEM_OFFSET, 0x8).unwrap();
    }

    #[test]
    fn test_unposion_top_of_high_mem() {
        let mut shadow = get_shadow();
        shadow.unpoison(GS::HIGH_MEM_LIMIT - 7, 0x8).unwrap();
    }

    #[test]
    fn test_unaligned_start() {
        let mut shadow = get_shadow();
        let result = shadow.unpoison(7, 1);
        assert_eq!(result, Err(GuestShadowError::UnalignedStartAddress(7, 1)));
    }

    #[test]
    fn test_aligned_one() {
        let mut shadow = get_shadow();
        shadow.unpoison(0, 1).unwrap();
    }

    #[test]
    fn test_aligned_two() {
        let mut shadow = get_shadow();
        shadow.unpoison(0, 2).unwrap();
    }

    #[test]
    fn test_aligned_three() {
        let mut shadow = get_shadow();
        shadow.unpoison(0, 3).unwrap();
    }

    #[test]
    fn test_aligned_four() {
        let mut shadow = get_shadow();
        shadow.unpoison(0, 4).unwrap();
    }

    #[test]
    fn test_aligned_five() {
        let mut shadow = get_shadow();
        shadow.unpoison(0, 5).unwrap();
    }

    #[test]
    fn test_aligned_six() {
        let mut shadow = get_shadow();
        shadow.unpoison(0, 6).unwrap();
    }

    #[test]
    fn test_aligned_seven() {
        let mut shadow = get_shadow();
        shadow.unpoison(0, 7).unwrap();
    }

    #[test]
    fn test_overflow_address_range() {
        let mut shadow = get_shadow();
        let start = usize::MAX - (ALIGN - 1);
        let end = ALIGN * 2;
        let result = shadow.unpoison(usize::MAX - (ALIGN - 1), ALIGN * 2);
        assert_eq!(
            result,
            Err(GuestShadowError::AddressRangeOverflow(start, end))
        );
    }
}