From e105fa42d5ddb6ff7bfbbb964c4b93415b22b05a Mon Sep 17 00:00:00 2001
From: Ian Moffett <ian@osmora.org>
Date: Fri, 21 Nov 2025 14:20:15 -0500
Subject: kern: Improve IRQ mask modification on spinlocks

Signed-off-by: Ian Moffett <ian@osmora.org>
---
 sys/arch/amd64/cpu/spinlock.c | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

(limited to 'sys/arch/amd64/cpu/spinlock.c')

diff --git a/sys/arch/amd64/cpu/spinlock.c b/sys/arch/amd64/cpu/spinlock.c
index 1cfa088..6857aa6 100644
--- a/sys/arch/amd64/cpu/spinlock.c
+++ b/sys/arch/amd64/cpu/spinlock.c
@@ -29,12 +29,15 @@
 
 #include <sys/cdefs.h>
 #include <mu/spinlock.h>
+#include <mu/irq.h>
 
 void
 mu_spinlock_acq(volatile size_t *lock, int flags)
 {
-    if (ISSET(flags, SPINLOCK_INTTOG)) {
-        asm volatile("cli");
+    bool irq_en = mu_irq_state();
+
+    if (ISSET(flags, SPINLOCK_INTTOG) && irq_en) {
+        __asmv("cli");
     }
 
     __asmv(
@@ -48,6 +51,11 @@ mu_spinlock_acq(volatile size_t *lock, int flags)
         : "r" (lock)
         : "memory", "rax"
     );
+
+    if (ISSET(flags, SPINLOCK_INTTOG)) {
+        if (irq_en && !mu_irq_state())
+            __asmv("sti");
+    }
 }
 
 void
@@ -60,8 +68,4 @@ mu_spinlock_rel(volatile size_t *lock, int flags)
         :
         : "memory", "rax"
     );
-
-    if (ISSET(flags, SPINLOCK_INTTOG)) {
-        asm volatile("sti");
-    }
 }
-- 
cgit v1.2.3