Always check shootdown reqs when locking addrsp.

src/percpu.rs

@@ -50,9 +50,7 @@ pub fn shootdown_tlb_ipi(target: Option<LogicalCpuId>) {
         while percpublock.wants_tlb_shootdown.swap(true, Ordering::Release) == true {
             // Load is faster than CAS or on x86, LOCK BTS
             while percpublock.wants_tlb_shootdown.load(Ordering::Relaxed) == true {
-                if my_percpublock.wants_tlb_shootdown.swap(false, Ordering::Release) == true {
-                    log::warn!("oops");
-                }
+                my_percpublock.maybe_handle_tlb_shootdown();
                 core::hint::spin_loop();
             }
         }
@@ -66,3 +64,22 @@ pub fn shootdown_tlb_ipi(target: Option<LogicalCpuId>) {
         }
     }
 }
+impl PercpuBlock {
+    pub fn maybe_handle_tlb_shootdown(&self) {
+        if self.wants_tlb_shootdown.swap(false, Ordering::Relaxed) == false {
+            return;
+        }
+
+        // TODO: Finer-grained flush
+        unsafe {
+            x86::tlb::flush_all();
+        }
+
+        {
+            let addrsp = self.current_addrsp.borrow();
+            if let Some(ref addrsp) = &*addrsp {
+                addrsp.tlb_ack.fetch_add(1, Ordering::Release);
+            }
+        }
+    }
+}