Notify schemes when mmaps are unmapped.

src/scheme/proc.rs

@@ -592,7 +592,7 @@ impl Scheme for ProcScheme {
 
         match handle.info.operation {
             Operation::AwaitingAddrSpaceChange { new, new_sp, new_ip } => {
-                stop_context(handle.info.pid, |context: &mut Context| unsafe {
+                let prev_addr_space = stop_context(handle.info.pid, |context: &mut Context| unsafe {
                     if let Some(saved_regs) = ptrace::regs_for_mut(context) {
                         #[cfg(target_arch = "aarch64")]
                         {
@@ -615,9 +615,7 @@ impl Scheme for ProcScheme {
                         context.clone_entry = Some([new_ip, new_sp]);
                     }
 
-                    let _prev_addr_space = context.set_addr_space(new);
-
-                    Ok(())
+                    Ok(context.set_addr_space(new))
                 })?;
                 let _ = ptrace::send_event(crate::syscall::ptrace_event!(PTRACE_EVENT_ADDRSPACE_SWITCH, 0));
             }

src/scheme/user.rs

@@ -543,10 +543,15 @@ impl UserInner {
         };
 
         let page_count_nz = NonZeroUsize::new(page_count).expect("already validated map.size != 0");
-        let dst_base = dst_addr_space.write().mmap(dst_base, page_count_nz, map.flags | MAP_FIXED_NOREPLACE, &mut Vec::new(), |dst_base, flags, mapper, flusher| {
+        let mut notify_files = Vec::new();
+        let dst_base = dst_addr_space.write().mmap(dst_base, page_count_nz, map.flags | MAP_FIXED_NOREPLACE, &mut notify_files, |dst_base, flags, mapper, flusher| {
             Grant::borrow_fmap(PageSpan::new(dst_base, page_count), page_flags(map.flags), file_ref, src, mapper, flusher)
         })?;
 
+        for map in notify_files {
+            let _ = map.unmap();
+        }
+
         Ok(dst_base.start_address().data())
     }
 }
@@ -829,6 +834,20 @@ impl KernelScheme for UserScheme {
 
         inner.fmap_inner(Arc::clone(addr_space), file, map)
     }
+    fn kfunmap(&self, number: usize, offset: usize, size: usize) -> Result<()> {
+        let inner = self.inner.upgrade().ok_or(Error::new(ENODEV))?;
+
+        let res = inner.call_extended(CallerCtx {
+            pid: context::context_id().into(),
+            uid: offset as u32,
+            gid: (offset >> 32) as u32,
+        }, [KSMSG_MUNMAP, number, size, 0])?;
+
+        match res {
+            Response::Regular(_) => Ok(()),
+            Response::Fd(_) => Err(Error::new(EIO)),
+        }
+    }
 
     fn as_user_inner(&self) -> Option<Result<Arc<UserInner>>> {
         Some(self.inner.upgrade().ok_or(Error::new(ENODEV)))