You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
2.7 KiB
81 lines
2.7 KiB
use nix::Error;
|
|
use nix::libc::{c_void, size_t};
|
|
use nix::sys::mman::{mmap, MapFlags, ProtFlags};
|
|
|
|
#[cfg(target_os = "linux")]
|
|
use nix::sys::mman::{mremap, MRemapFlags};
|
|
|
|
#[test]
|
|
fn test_mmap_anonymous() {
|
|
let ref mut byte = unsafe {
|
|
let ptr = mmap(std::ptr::null_mut(), 1,
|
|
ProtFlags::PROT_READ | ProtFlags::PROT_WRITE,
|
|
MapFlags::MAP_PRIVATE | MapFlags::MAP_ANONYMOUS, -1, 0)
|
|
.unwrap();
|
|
*(ptr as * mut u8)
|
|
};
|
|
assert_eq !(*byte, 0x00u8);
|
|
*byte = 0xffu8;
|
|
assert_eq !(*byte, 0xffu8);
|
|
}
|
|
|
|
#[test]
|
|
#[cfg(target_os = "linux")]
|
|
fn test_mremap_grow() {
|
|
const ONE_K : size_t = 1024;
|
|
let slice : &mut[u8] = unsafe {
|
|
let mem = mmap(std::ptr::null_mut(), ONE_K,
|
|
ProtFlags::PROT_READ | ProtFlags::PROT_WRITE,
|
|
MapFlags::MAP_ANONYMOUS | MapFlags::MAP_PRIVATE, -1, 0)
|
|
.unwrap();
|
|
std::slice::from_raw_parts_mut(mem as * mut u8, ONE_K)
|
|
};
|
|
assert_eq !(slice[ONE_K - 1], 0x00);
|
|
slice[ONE_K - 1] = 0xFF;
|
|
assert_eq !(slice[ONE_K - 1], 0xFF);
|
|
|
|
let slice : &mut[u8] = unsafe {
|
|
let mem = mremap(slice.as_mut_ptr() as * mut c_void, ONE_K, 10 * ONE_K,
|
|
MRemapFlags::MREMAP_MAYMOVE, None)
|
|
.unwrap();
|
|
std::slice::from_raw_parts_mut(mem as * mut u8, 10 * ONE_K)
|
|
};
|
|
|
|
// The first KB should still have the old data in it.
|
|
assert_eq !(slice[ONE_K - 1], 0xFF);
|
|
|
|
// The additional range should be zero-init'd and accessible.
|
|
assert_eq !(slice[10 * ONE_K - 1], 0x00);
|
|
slice[10 * ONE_K - 1] = 0xFF;
|
|
assert_eq !(slice[10 * ONE_K - 1], 0xFF);
|
|
}
|
|
|
|
#[test]
|
|
#[cfg(target_os = "linux")]
|
|
fn test_mremap_shrink() {
|
|
const ONE_K : size_t = 1024;
|
|
let slice : &mut[u8] = unsafe {
|
|
let mem = mmap(std::ptr::null_mut(), 10 * ONE_K,
|
|
ProtFlags::PROT_READ | ProtFlags::PROT_WRITE,
|
|
MapFlags::MAP_ANONYMOUS | MapFlags::MAP_PRIVATE, -1, 0)
|
|
.unwrap();
|
|
std::slice::from_raw_parts_mut(mem as * mut u8, ONE_K)
|
|
};
|
|
assert_eq !(slice[ONE_K - 1], 0x00);
|
|
slice[ONE_K - 1] = 0xFF;
|
|
assert_eq !(slice[ONE_K - 1], 0xFF);
|
|
|
|
let slice : &mut[u8] = unsafe {
|
|
let mem = mremap(slice.as_mut_ptr() as * mut c_void, 10 * ONE_K, ONE_K,
|
|
MRemapFlags::empty(), None)
|
|
.unwrap();
|
|
// Since we didn't supply MREMAP_MAYMOVE, the address should be the
|
|
// same.
|
|
assert_eq !(mem, slice.as_mut_ptr() as * mut c_void);
|
|
std::slice::from_raw_parts_mut(mem as * mut u8, ONE_K)
|
|
};
|
|
|
|
// The first KB should still be accessible and have the old data in it.
|
|
assert_eq !(slice[ONE_K - 1], 0xFF);
|
|
}
|