linux can do MLOCK_ONFAULT, other unixes have to fall back to bare mlock

bytecodealliance · Feb 17, 2022 · 561f325 · 561f325
1 parent c110a1a
commit 561f325
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/crates/runtime/src/mmap.rs b/crates/runtime/src/mmap.rs
@@ -434,14 +434,21 @@ impl Mmap {
             range.start % region::page::size() == 0,
             "changing of protections isn't page-aligned",
         );
-        #[cfg(unix)]
+        #[cfg(target_os = "linux")]
         unsafe {
             rustix::io::mlock_with(
                 self.as_ptr().add(range.start) as *mut _,
                 range.end - range.start,
                 rustix::io::MlockFlags::ONFAULT,
             )?;
         }
+        #[cfg(all(unix, not(target_os = "linux")))]
+        unsafe {
+            rustix::io::mlock(
+                self.as_ptr().add(range.start) as *mut _,
+                range.end - range.start,
+            )?;
+        }
         Ok(())
     }
 }