Linux內核中的互斥鎖、讀寫鎖、自旋鎖、信號量該如何選擇？

一、前言

Linux內核中有許多不同類型的鎖，它們都可以用來保護關鍵資源，以避免多個線程或進程之間發生競爭條件，從而保護系統的穩定性和可靠性。這些鎖的類型包括：互斥鎖（mutex）、讀寫鎖（rwlock）、自旋鎖（spinlock）和信號量（semaphore）。今天就給大家介紹一下Linux內核中的各種鎖，以及我們在實際項目中該如何選擇使用哪個鎖。

二、幾種鎖的介紹

互斥鎖（mutex） 是最常用的鎖，它可以保護共享資源，使得在某個時刻只有一個線程或進程可以訪問它。讀寫鎖（rwlock）則可以同時允許多個線程或進程讀取共享資源，但只允許一個線程或進程寫入它。自旋鎖（spinlock）可以用來保護共享資源，使得在某個時刻只有一個線程或進程可以訪問它，但它會使線程或進程“自旋”，直到獲得鎖為止。最后，信號量（semaphore）可以用來控制對共享資源的訪問，以保證其他線程或進程可以安全地訪問它們。

讀寫鎖（rwlock） 是一種用于控制多線程訪問共享資源的同步機制。當一個線程需要讀取共享資源時，可以獲取讀取鎖，這樣其他線程就可以同時讀取該資源，而不會引發沖突。當一個線程需要寫入共享資源時，可以獲取寫入鎖，這樣其他線程就不能訪問該資源，從而保證數據的完整性和一致性。

自旋鎖（spinlock） 是一種簡單而有效的用于解決多線程同步問題的鎖。它是一種排他鎖，可以在多線程環境下保護共享資源，以防止多個線程同時對該資源進行訪問。自旋鎖的基本原理是，當一個線程試圖獲取鎖時，它會不斷嘗試獲取鎖，直到成功為止。在這期間，線程不會進入休眠狀態，而是一直處于忙等待（busy-waiting）狀態，這也就是自旋鎖的由來。

信號量（semaphore） 是一種常用的同步機制，它可以用來控制多個線程對共享資源的訪問。它有助于確保同一時間只有一個線程能夠訪問共享資源，從而避免資源沖突和競爭。信號量是一種整數計數器，用于跟蹤可用資源的數量。當一個線程需要訪問共享資源時，它首先必須獲取信號量，這會將信號量的計數器減少1，而當它完成訪問共享資源后，它必須釋放信號量，以便其他線程也可以訪問共享資源。

四、互斥鎖（Mutex）

互斥鎖是最基本的鎖類型，在內核中使用較為廣泛。它是一種二元鎖，只能同時有一個線程持有該鎖。當一個線程請求該鎖時，如果鎖已被占用，則線程會被阻塞直到鎖被釋放。互斥鎖的實現使用了原子操作，因此它的性能比較高，但也容易出現死鎖情況。

在內核中，互斥鎖的定義如下：

struct?mutex?{
????raw_spinlock_t??????wait_lock;
????struct?list_head????wait_list;
????struct?task_struct??*owner;
????int?????????????????recursion;
#ifdef?CONFIG_DEBUG_LOCK_ALLOC
????struct?lockdep_map??dep_map;
#endif
};

互斥鎖的使用非常簡單，通常只需要調用兩個函數即可完成：

void?mutex_init(struct?mutex?*lock)：函數用于初始化互斥鎖
void?mutex_lock(struct?mutex?*lock)：函數用于獲取互斥鎖
void?mutex_unlock(struct?mutex?*lock)：函數用于釋放互斥鎖

五、讀寫鎖（Reader-Writer Lock）

讀寫鎖是一種特殊的鎖類型，它允許多個線程同時讀取共享資源，但只允許一個線程寫入共享資源。讀寫鎖的實現使用了兩個計數器，分別記錄當前持有鎖的讀線程數和寫線程數。

在內核中，讀寫鎖的定義如下：

struct?rw_semaphore?{
????long????????????count;
????struct?list_head????wait_list;
#ifdef?CONFIG_DEBUG_LOCK_ALLOC
????struct?lockdep_map??dep_map;
#endif
};

讀寫鎖的使用也比較簡單，通常只需要調用三個函數即可完成：

init_rwsem(struct rw_semaphore *sem)：函數用于初始化讀寫鎖
down_read(struct rw_semaphore *sem)：函數用于獲取讀鎖
up_read(struct rw_semaphore *sem)：函數用于釋放讀鎖
down_write(struct rw_semaphore *sem)：函數用于獲取寫鎖
up_write(struct rw_semaphore *sem)：函數用于釋放寫鎖

六、自旋鎖（spinlock）

自旋鎖是一種保護共享資源的鎖，它會在等待期間一直占用CPU。自旋鎖適用于代碼臨界區比較小的情況，且共享資源的獨占時間比較短，這樣就可以避免上下文切換的開銷。自旋鎖不能用于需要睡眠的代碼臨界區，因為在睡眠期間自旋鎖會一直占用CPU。

在Linux內核中，自旋鎖使用spinlock_t類型表示，可以通過spin_lock()和spin_unlock()函數對其進行操作。

spin_lock_init(spinlock_t?*lock)：用于初始化自旋鎖，將自旋鎖的初始狀態設置為未加鎖狀態。
spin_lock(spinlock_t?*lock)：用于獲得自旋鎖，如果自旋鎖已經被占用，則當前進程會自旋等待，直到自旋鎖可用。
spin_trylock(spinlock_t?*lock)：用于嘗試獲取自旋鎖，如果自旋鎖當前被占用，則返回0，否則返回1。
spin_unlock(spinlock_t?*lock)：用于釋放自旋鎖。

在使用自旋鎖時，需要注意以下幾點：

• 自旋鎖只適用于臨界區代碼比較短的情況，因為自旋等待的過程會占用CPU資源。

• 自旋鎖不可重入，也就是說，如果一個進程已經持有了自旋鎖，那么它不能再次獲取該自旋鎖。

• 在持有自旋鎖的情況下，應該盡量避免調用可能會導致調度的內核函數，比如睡眠函數，因為這可能會導致死鎖的發生。

• 在使用自旋鎖的時候，應該盡量避免嵌套使用不同類型的鎖，比如自旋鎖和讀寫鎖，因為這可能會導致死鎖的發生。

• 當臨界區代碼較長或者需要睡眠時，應該使用信號量或者讀寫鎖來代替自旋鎖。

七、信號量（semaphore）

信號量是一種更高級的鎖機制，它可以控制對共享資源的訪問次數。信號量可分為二元信號量和計數信號量。二元信號量只有0和1兩種狀態，常用于互斥鎖的實現；計數信號量則可以允許多個進程同時訪問同一共享資源，只要它們申請信號量的數量不超過該資源所允許的最大數量。

在Linux內核中，信號量使用struct semaphore結構表示，可以通過down()和up()函數對其進行操作。

void?sema_init(struct?semaphore?*sem,?int?val)：初始化一個信號量，val參數表示初始值。
void?down(struct?semaphore?*sem)：嘗試獲取信號量，如果信號量值為?0，調用進程將被阻塞。
int?down_interruptible(struct?semaphore?*sem)：嘗試獲取信號量，如果信號量值為?0，調用進程將被阻塞，并可以被中斷。
int?down_trylock(struct?semaphore?*sem)：嘗試獲取信號量，如果信號量值為?0，則立即返回，否則返回錯誤。
void?up(struct?semaphore?*sem)：釋放信號量，將信號量的值加 1，并喚醒可能正在等待信號量的進程。

八、該如何選擇正確的鎖

當需要對共享資源進行訪問和修改時，我們通常需要采用同步機制來保證數據的一致性和正確性，其中鎖是最基本的同步機制之一。不同類型的鎖適用于不同的場景。

互斥鎖適用于需要保護共享資源，只允許一個線程或進程訪問共享資源的場景。例如，當一個線程正在修改一個數據結構時，其他線程必須等待該線程釋放鎖后才能修改該數據結構。

讀寫鎖適用于共享資源的讀寫操作頻繁且讀操作遠大于寫操作的場景。讀寫鎖允許多個線程同時讀取共享資源，但只允許一個線程寫入共享資源。例如，在一個數據庫管理系統中，讀取操作比寫入操作頻繁，使用讀寫鎖可以提高系統的并發性能。

自旋鎖適用于保護共享資源的訪問時間很短的場景，當線程需要等待的時間很短時，自旋鎖比互斥鎖的性能更好。例如，在訪問共享資源時需要進行一些簡單的操作，如對共享資源進行遞增或遞減等操作。

信號量適用于需要協調多個線程或進程對共享資源的訪問的場景，允許多個線程或進程同時訪問共享資源，但同時訪問的線程或進程數量有限。例如，在一個并發下載系統中，可以使用信號量來限制同時下載的文件數量。

舉個生活中的例子：當我們在買咖啡的時候，柜臺前可能會有一個小桶，上面寫著“請取走您需要的糖果，每人一顆”這樣的字樣。這個小桶就是一個信號量，它限制了每個人能夠取走的糖果的數量，從而保證了公平性。

如果我們把這個小桶換成互斥鎖，那么就可以只允許一個人在柜臺前取走糖果。如果使用讀寫鎖，那么在非高峰期的時候，多個人可以同時取走糖果，但在高峰期時只允許一個人取走。

而如果我們把這個小桶換成自旋鎖，那么當有人在取走糖果時，其他人就需要一直在那里等待，直到糖果被取走為止。這樣可能會造成浪費時間的情況，因為其他人可能有更緊急的事情需要處理。

九、總結

在Linux內核中，有四種常見的鎖：互斥鎖、讀寫鎖、自旋鎖和信號量。這些鎖適用于不同的場景，開發者需要根據實際情況選擇適當的鎖來確保并發訪問的正確性和性能。