From 7759eef8cdac79028dd5b468bd9ce0cf1c12b162 Mon Sep 17 00:00:00 2001
From: Richard Qin <richard.qin.x@gmail.com>
Date: Tue, 30 Dec 2025 14:21:40 +0800
Subject: dev(spinlock): refactoring the implementation of locks

---
 include/kernel/spinlock.h | 54 ++++++++++++++++++++++++++++++++++++-----------
 1 file changed, 42 insertions(+), 12 deletions(-)

(limited to 'include/kernel/spinlock.h')

diff --git a/include/kernel/spinlock.h b/include/kernel/spinlock.h
index aec4b6a..b3cc93e 100644
--- a/include/kernel/spinlock.h
+++ b/include/kernel/spinlock.h
@@ -1,20 +1,50 @@
 #pragma once
+
 #include "common/types.h"
 
-struct spinlock
+struct cpu;
+struct Proc;
+
+// Spinlock
+// Features: Non-interruptible, busy-wait loop, extremely short hold time.
+// Scenario: Protecting core data such as scheduler, interrupt handling, PMM, etc.
+class Spinlock
 {
-    uint locked;
+public:
+    volatile uint32 locked;     // Whether it is locked (0: unlocked, 1: locked)
+    const char *name;      // Lock Name (for debugging)
+    struct cpu *cpu; // The CPU holding the lock
+
+    // Constructor
+    Spinlock() : locked(0), name("uninit"), cpu(nullptr) {}
+
+    // Initialize lock name
+    void init(const char *name);
 
-    char *name;
-    int cpu_id;
+    // Acquire lock
+    void acquire();
+
+    // Release lock
+    void release();
+
+    // Check if the current CPU holds the lock
+    bool holding();
 };
 
-namespace Spinlock
+// Mutex
+// Features: Allows interruption, allows sleeping (yield CPU), can hold for a long time.
+// Scenarios: File system IO, device driver waiting, long-term buffering operations.
+class Mutex
 {
-    void init(struct spinlock *lk, char *name);
-    void acquire(struct spinlock *lk);
-    void release(struct spinlock *lk);
-    int holding(struct spinlock *lk);
-    void push_off();
-    void pop_off();
-}
\ No newline at end of file
+public:
+    uint32 locked;    // Whether it is locked
+    const char *name; // Lock Name
+    int pid;          // PID of the process holding the lock (for debugging)
+
+    Spinlock lk; // Internal spin lock, protects the state of the Mutex itself
+
+    void init(const char *name);
+    void acquire();
+    void release();
+    bool holding();
+};
\ No newline at end of file
-- 
cgit v1.2.3