大神教你設定Linux進程的睡眠和喚醒

導讀	在Linux中，只等待CPU時間的進程稱為就緒進程，它們被放置在一個運行佇列中，一個就緒進程的狀 態標誌位元為TASK_RUNNING。一旦一個運行中的進程時間片用完， Linux 核心的調度器會剝奪這個進程對CPU的控制權，並且從運行隊列中選擇一個合適的進程投入運行。

當然，一個行程也可以主動釋放CPU的控制權。函數 schedule()是一個調度函數，它可以被一個進程主動調用，從而調度其它進程佔用CPU。一旦這個主動放棄CPU的程序被重新調度佔用 CPU，那麼它將從上次停止執行的位置開始執行，也就是說它將從呼叫schedule()的下一行程式碼開始執行。
有時候，進程需要等待直到某個特定的事件發生，例如裝置初始化完成、I/O 操作完成或計時器到時等。在這種情況下，進程則必須從運行佇列移出，加入到一個等待佇列中，這個時候進程就進入了睡眠狀態。

Linux 中的進程睡眠狀態分類

一種是可中斷的睡眠狀態，其狀態標誌位元TASK_INTERRUPTIBLE;
另一種是不可中斷 的睡眠狀態，其狀態標誌位元為TASK_UNINTERRUPTIBLE。可中斷的睡眠狀態的進程會睡眠直到某個條件變成真，比如說產生一個硬體中斷、釋放 進程正在等待的系統資源或是傳遞一個訊號都可以是喚醒進程的條件。不可中斷睡眠狀態與可中斷睡眠狀態類似，但是它有一個例外，那就是把訊號傳遞到這種睡眠 狀態的進程不能改變它的狀態，也就是說它不回應訊號的喚醒。不可中斷睡眠狀態一般較少用到，但在某些特定情況下這種狀態還是很有用的，比如說：進程必須等 待，不能被中斷，直到某個特定的事件發生。
在現代的Linux作業系統中，進程一般都是用呼叫schedule()的方法進入睡眠狀態的，下面的程式碼演
示如何讓正在運行的進程進入睡眠狀態。
sleeping_task = current;
set_current_state(TASK_INTERRUPTIBLE);
schedule();
func1();
/* Rest of the code ... */
在第一個語句中，程式儲存了一份進程結構指標sleeping_task，current 是一個宏，它指向正在執行
的進程結構。 set_current_state()將該行程的狀態從執行狀態TASK_RUNNING 變成睡眠狀態
TASK_INTERRUPTIBLE。如果schedule()是被一個狀態為TASK_RUNNING 的進程調度，那麼schedule()將調度另一個進程佔用CPU；如果schedule()是被一個狀態為TASK_INTERRUPTIBLE 或TASK_UNINTERRUPTIBLE 的進程調度，那麼還有一個附加的步驟將被執行：目前執行的進程在另一個進程被調度之前會被從運行隊列中移出，這將導致正在運行的進程進入睡眠，因為它已經不在運行隊列中了。
我們可以使用下面的這個函數將剛才那個進入睡眠的進程喚醒。
wake_up_process(sleeping_task);
在呼叫了wake_up_process()以後，這個睡眠行程的狀態會被設定為TASK_RUNNING，而且調度器
會把它加入到運行隊列中去。當然，這個進程只有在下次被調度器調度到的時候才能真正投入運作。

無效喚醒

幾乎在所有的情況下，進程都會在檢查了某些條件之後，發現條件不滿足才進入睡眠。可是有的時候
進程卻會在 判定條件為真後開始睡眠，如果這樣的話進程就會無限期地休眠下去，這就是所謂的無效喚醒問題。在作業系統中，當多個進程都企圖對共享資料進行某種處理，而最後的結果又取決於進程運行的順序時，就會發生競爭條件，這是作業系統中一個典型的問題，無效喚醒恰恰就是由於競爭條件導致的。
設想有兩個進程A 和B，A 進程正在處理一個鍊錶，它需要檢查這個鍊錶是否為空，如果不空就對鏈
表裡面的資料進行一些操作，同時B進程也正在往這個鍊錶加入節點。當這個鍊錶是空的時候，由於無資料可操作，這時A進程就進入睡眠，當B進程向鍊錶裡面添加了節點之後它就喚醒A 進程，其程式碼如下：
A進程:
1 spin_lock(&list_lock);
2 if(list_empty(&list_head)) {
3 spin_unlock(&list_lock);
4 set_current_state(TASK_INTERRUPTIBLE);
5 schedule();
6 spin_lock(&list_lock);
7 }
8
9 /* Rest of the code ... */
10 spin_unlock(&list_lock);
B進程:
100 spin_lock(&list_lock);
101 list_add_tail(&list_head, new_node);
102 spin_unlock(&list_lock);
103 wake_up_process(processa_task);
這裡會出現一個問題，假如當A行程執行到第3行後第4行前的時候，B行程被另外一個處理器調度
投 入運行。在這個時間片內，B進程執行完了它所有的指令，因此它試圖喚醒A進程，而此時的A進程還沒有進入睡眠，所以喚醒操作無效。在這之後，A 進程繼續執行，它會錯誤地認為這個時候鍊錶仍然是空的，於是將自己的狀態設為TASK_INTERRUPTIBLE然後調用schedule()進入睡 眠。由於錯過了B進程喚醒，它將會無限期的睡眠下去，這就是無效喚醒問題，因為即使鍊錶中有資料需要處理，A 進程也還是睡眠了。

避免無效喚醒

如何避免無效喚醒問題呢？我們發現無效喚醒主要發生在檢查條件之後和進程狀態被設定為睡眠狀
態之前， 本來B進程的wake_up_process()提供了一次將A進程狀態置為TASK_RUNNING 的機會，可惜這個時候A進程的狀態仍然是TASK_RUNNING，所以wake_up_process()將A進程狀態從睡眠狀態轉變為運行狀態的努力沒有起到預期的作用。要解決這個問題，必須使用一種保障機制使得判斷鍊錶為空和設定進程狀態為睡眠狀態成為一個不可分割的步驟才行，也就是必須消除競爭條件產生的根源，這樣在這之後出現的wake_up_process ( )就可以起到喚醒狀態是睡眠狀態的進程的作用了。
找到了原因後，重新設計一下A程序的程式碼結構，就可以避免上面例子的無效喚醒問題了。
A進程:
1 set_current_state(TASK_INTERRUPTIBLE);
2 spin_lock(&list_lock);
3 if(list_empty(&list_head)) {
4 spin_unlock(&list_lock);
5 schedule();
6 spin_lock(&list_lock);
7 }
8 set_current_state(TASK_RUNNING);
9
10 /* Rest of the code ... */
11 spin_unlock(&list_lock);
可以看到，這段程式碼在測試條件之前就將目前執行進程狀態轉設定成TASK_INTERRUPTIBLE了，並且在鍊錶不為空的情況下又將自己置為TASK_RUNNING狀態。這樣一來如果B進程在A進程進程檢查
了鍊錶為空以後呼叫wake_up_process()，那麼A進程的狀態就會自動由原來TASK_INTERRUPTIBLE
變成TASK_RUNNING，此後即使進程又調用了schedule()，由於它現在的狀態是TASK_RUNNING，所以仍然不會被從運行隊列中移出，因而不會錯誤的進入睡眠，當然也就避免了無效喚醒問題。

Linux核心的範例

在Linux作業系統中，核心的穩定性至關重要，為了避免Linux作業系統核心中出現無效喚醒問題，
Linux核心在需要進程睡眠的時候應該使用類似如下的操作：
/* ‘q’是我們希望睡眠的等待隊列 */
DECLARE_WAITQUEUE(wait,current);
add_wait_queue(q, &wait);
set_current_state(TASK_INTERRUPTIBLE);
/* 或TASK_INTERRUPTIBLE */
while(!condition) /* ‘condition’ 是等待的條件*/
schedule();
set_current_state(TASK_RUNNING);
remove_wait_queue(q, &wait);
上面的操作，使得進程透過下面的一系列步驟安全地將自己加入到一個等待隊列中進行睡眠：首先調
用DECLARE_WAITQUEUE ()建立一個等待佇列的項，然後呼叫add_wait_queue()把自己加入到等待佇列中，並且將行程的狀態設為 TASK_INTERRUPTIBLE 或TASK_INTERRUPTIBLE。然後循環檢查條件是否為真：如果是的話就沒有必要睡眠，如果條件不為真，就呼叫schedule()。當進程 檢查的條件滿足後，進程又將自己設定為TASK_RUNNING 並呼叫remove_wait_queue()將自己移出等待佇列。
從上面可以看到，Linux的核心程式碼維護者也是在進程檢查條件之前就設定進程的狀態為睡眠狀態，
然後才循環檢查條件。如果在進程開始睡眠之前條件就已經達成了，那麼循環會退出並用set_current_state()將自己的狀態設為就緒，這同樣保證了進程不會存在錯誤的進入睡眠的傾向，當然也就不會導致出現無效喚醒問題。
以下讓我們用linux 核心中的實例來看看Linux 核心是如何避免無效睡眠的，這段程式碼出自Linux2.6的核心(linux-2.6.11/kernel/sched.c: 4254):
4253 /* Wait for kthread_stop */
4254 set_current_state(TASK_INTERRUPTIBLE);
4255 while (!kthread_should_stop()) {
4256 schedule();
4257 set_current_state(TASK_INTERRUPTIBLE);
4258 }
4259 __set_current_state(TASK_RUNNING);
4260 return 0;
上面的這些程式碼屬於遷移服務執行緒migration_thread，這個執行緒不斷地檢查kthread_should_stop()，
直 到kthread_should_stop()返回1它才可以退出循環，也就是說只要kthread_should_stop()返回0該進程就會一直睡 眠。從程式碼中我們可以看出，檢查kthread_should_stop()確實是在進程的狀態被置為TASK_INTERRUPTIBLE後才開始執行 的。因此，如果在條件檢查之後但是在schedule()之前有其他進程試圖喚醒它，那麼該進程的喚醒操作不會失效。

總結

透過上面的討論，可以發現在Linux 中避免進程的無效喚醒的關鍵是在進程檢查條件之前就將進程的
狀態置為TASK_INTERRUPTIBLE或TASK_UNINTERRUPTIBLE，如果檢查的條件滿足的話就應該
將其狀態重新設定為TASK_RUNNING。這樣無論進程等待的條件是否滿足， 進程都不會因為被移出就緒佇列而錯誤地進入睡眠狀態，從而避免了無效喚醒問題。

以上是大神教你設定Linux進程的睡眠和喚醒的詳細內容。更多資訊請關注PHP中文網其他相關文章！