Function test_mlock

tests/mm/mmap.rs:94–133

()

#[test]
fn test_mlock() {
    use rustix::mm::{mlock, mmap_anonymous, munlock, munmap, MapFlags, ProtFlags};
    #[cfg(linux_kernel)]
    use rustix::mm::{mlock_with, MlockFlags};
    use std::ptr::null_mut;

    unsafe {
        let addr = mmap_anonymous(null_mut(), 8192, ProtFlags::READ, MapFlags::PRIVATE).unwrap();

        match mlock(addr, 8192) {
            Ok(()) => munlock(addr, 8192).unwrap(),
            // Tests won't always have enough memory or permissions, and that's ok.
            Err(rustix::io::Errno::PERM | rustix::io::Errno::NOMEM) => (),
            // But they shouldn't fail otherwise.
            Err(other) => panic!("{:?}", other),
        }

        #[cfg(linux_kernel)]
        {
            match mlock_with(addr, 8192, MlockFlags::empty()) {
                Err(rustix::io::Errno::NOSYS) => (),
                Err(err) => panic!("{:?}", err),
                Ok(()) => munlock(addr, 8192).unwrap(),
            }

            match mlock_with(addr, 8192, MlockFlags::ONFAULT) {
                // Linux versions that lack `mlock` return this.
                Err(rustix::io::Errno::NOSYS) => (),
                // Linux versions that don't recognize `ONFAULT` return this.
                Err(rustix::io::Errno::INVAL) => (),
                Err(err) => panic!("{:?}", err),
                Ok(()) => munlock(addr, 8192).unwrap(),
            }

            munlock(addr, 8192).unwrap();
        }

        munmap(addr, 8192).unwrap();
    }
}

#[cfg(not(target_os = "redox"))]
#[test]