add resize()

src/lib.rs

@@ -638,6 +638,37 @@ impl MmapMut {
         self.inner.make_exec()?;
         Ok(Mmap { inner: self.inner })
     }
+
+    #[cfg(
+    all(
+        any(
+            all(target_os = "linux", not(target_arch = "mips")),
+            target_os = "freebsd"
+        ),
+        not(target_os = "ios")
+    ))]
+    /// Equivalent to resize_with_flag(len, libc::MREMAP_MAYMOVE)
+    pub fn resize(&mut self, len: usize) -> Result<()> {
+        self.inner.resize(len)
+    }
+
+    #[cfg(
+    all(
+        any(
+            all(target_os = "linux", not(target_arch = "mips")),
+            target_os = "freebsd"
+        ),
+        not(target_os = "ios")
+    ))]
+    /// Corresponding to mremap() in libc, expands (or shrinks) an existing memory mapping, potentially
+    /// moving it at the same time (controlled by the flags argument and the available virtual address space).
+    /// Available flags(See mremap(2) — Linux manual page):
+    ///        MREMAP_MAYMOVE(By default)
+    ///        MREMAP_FIXED (since Linux 2.3.31)
+    ///        MREMAP_DONTUNMAP (since Linux 5.7)
+    pub fn resize_with_flag(&mut self, len: usize, flag: libc::c_int) -> Result<()> {
+        self.inner.resize_with_flag(len, flag)
+    }
 }
 
 impl Deref for MmapMut {
@@ -1057,4 +1088,14 @@ mod test {
         let mmap = mmap.make_exec().expect("make_exec");
         drop(mmap);
     }
-}
+    #[cfg(any(
+    all(target_os = "linux", not(target_arch = "mips")),
+    target_os = "freebsd"
+    ))]
+    #[test]
+    fn resize() {
+        let mut mmap = MmapMut::map_anon(256).expect("map_mut");
+        mmap.resize(512).expect("resize");
+    }
+
+}

src/unix.rs

@@ -20,7 +20,7 @@ const MAP_STACK: libc::c_int = 0;
 
 pub struct MmapInner {
     ptr: *mut libc::c_void,
-    len: usize,
+    len: usize
 }
 
 impl MmapInner {
@@ -60,7 +60,7 @@ impl MmapInner {
             } else {
                 Ok(MmapInner {
                     ptr: ptr.offset(alignment as isize),
-                    len: len,
+                    len
                 })
             }
         }
@@ -174,6 +174,40 @@ impl MmapInner {
         self.mprotect(libc::PROT_READ | libc::PROT_WRITE)
     }
 
+    #[cfg(
+    all(
+        any(
+            all(target_os = "linux", not(target_arch = "mips")),
+            target_os = "freebsd"
+        ),
+        not(target_os = "ios")
+    ))]
+    pub fn resize_with_flag(&mut self, len: usize, flag: libc::c_int) -> io::Result<()> {
+        unsafe {
+            let new_addr = libc::mremap(self.ptr, self.len, len, flag);
+            if new_addr == libc::MAP_FAILED {
+                Err(io::Error::last_os_error())
+            } else {
+                self.ptr = new_addr;
+                self.len = len;
+                Ok(())
+            }
+
+        }
+    }
+
+    #[cfg(
+    all(
+        any(
+            all(target_os = "linux", not(target_arch = "mips")),
+            target_os = "freebsd"
+        ),
+        not(target_os = "ios")
+    ))]
+    pub fn resize(&mut self, len: usize) -> io::Result<()> {
+        self.resize_with_flag(len, libc::MREMAP_MAYMOVE)
+    }
+
     #[inline]
     pub fn ptr(&self) -> *const u8 {
         self.ptr as *const u8
@@ -194,14 +228,10 @@ impl Drop for MmapInner {
     fn drop(&mut self) {
         let alignment = self.ptr as usize % page_size();
         unsafe {
-            assert!(
-                libc::munmap(
-                    self.ptr.offset(-(alignment as isize)),
-                    (self.len + alignment) as libc::size_t
-                ) == 0,
-                "unable to unmap mmap: {}",
-                io::Error::last_os_error()
-            );
+            assert_eq!(libc::munmap(
+                self.ptr.offset(-(alignment as isize)),
+                (self.len + alignment) as libc::size_t
+            ), 0, "unable to unmap mmap: {}", io::Error::last_os_error());
         }
     }
 }