From cd761ee679ea2a72aa6cd273b75811e5f579c009 Mon Sep 17 00:00:00 2001
From: bjorn3 <17426603+bjorn3@users.noreply.github.com>
Date: Fri, 26 Sep 2025 23:27:01 +0200
Subject: [PATCH] Handle all cases in foreach_addrsp

---
 src/scheme/proc.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/scheme/proc.rs b/src/scheme/proc.rs
index 7a0cc8e97b..6840564982 100644
--- a/src/scheme/proc.rs
+++ b/src/scheme/proc.rs
@@ -160,7 +160,10 @@ static HANDLES: RwLock<L1, HashMap<usize, Handle>> =
 pub fn foreach_addrsp(token: &mut CleanLockToken, mut f: impl FnMut(&Arc<AddrSpaceWrapper>)) {
     for (_, handle) in HANDLES.read(token.token()).iter() {
         let Handle {
-            kind: ContextHandle::AddrSpace { addrspace, .. },
+            kind:
+                ContextHandle::AddrSpace { addrspace, .. }
+                | ContextHandle::AwaitingAddrSpaceChange { new: addrspace, .. }
+                | ContextHandle::MmapMinAddr(addrspace),
             ..
         } = handle
         else {