1. 程式人生 > >linux程序控制之訊號量 semget,semctl,semop

linux程序控制之訊號量 semget,semctl,semop

轉載自 https://www.cnblogs.com/52php/p/5851570.html 這篇文章將講述別一種程序間通訊的機制——訊號量。注意請不要把它與之前所說的訊號混淆起來,訊號與訊號量是不同的兩種事物。有關訊號的更多內容,可以閱讀我的另一篇文章:Linux程序間通訊 – 訊號。下面就進入訊號量的講解。 一、什麼是訊號量

為了防止出現因多個程式同時訪問一個共享資源而引發的一系列問題,我們需要一種方法,它可以通過生成並使用令牌來授權,在任一時刻只能有一個執行執行緒訪問程式碼的臨界區域。臨界區域是指執行資料更新的程式碼需要獨佔式地執行。而訊號量就可以提供這樣的一種訪問機制,讓一個臨界區同一時間只有一個執行緒在訪問它,也就是說訊號量是用來調協程序對共享資源的訪問的。

訊號量是一個特殊的變數,程式對其訪問都是原子操作,且只允許對它進行等待(即P(訊號變數))和傳送(即V(訊號變數))資訊操作。最簡單的訊號量是隻能取0和1的變數,這也是訊號量最常見的一種形式,叫做二進位制訊號量。而可以取多個正整數的訊號量被稱為通用訊號量。這裡主要討論二進位制訊號量。 二、訊號量的工作原理

由於訊號量只能進行兩種操作等待和傳送訊號,即P(sv)和V(sv),他們的行為是這樣的:

P(sv):如果sv的值大於零,就給它減1;如果它的值為零,就掛起該程序的執行

V(sv):如果有其他程序因等待sv而被掛起,就讓它恢復執行,如果沒有程序因等待sv而掛起,就給它加1.

舉個例子,就是兩個程序共享訊號量sv,一旦其中一個程序執行了P(sv)操作,它將得到訊號量,並可以進入臨界區,使sv減1。而第二個程序將被阻止進入臨界區,因為當它試圖執行P(sv)時,sv為0,它會被掛起以等待第一個程序離開臨界區域並執行V(sv)釋放訊號量,這時第二個程序就可以恢復執行。 三、Linux的訊號量機制

Linux提供了一組精心設計的訊號量介面來對訊號進行操作,它們不只是針對二進位制訊號量,下面將會對這些函式進行介紹,但請注意,這些函式都是用來對成組的訊號量值進行操作的。它們宣告在標頭檔案sys/sem.h中。

1、semget()函式

它的作用是建立一個新訊號量或取得一個已有訊號量,原型為:

int semget(key_t key, int num_sems, int sem_flags);

第一個引數key是整數值(唯一非零),不相關的程序可以通過它訪問一個訊號量,它代表程式可能要使用的某個資源,程式對所有訊號量的訪問都是間接的,程式先通過呼叫semget()函式並提供一個鍵,再由系統生成一個相應的訊號識別符號(semget()函式的返回值),只有semget()函式才直接使用訊號量鍵,所有其他的訊號量函式使用由semget()函式返回的訊號量識別符號。如果多個程式使用相同的key值,key將負責協調工作。

第二個引數num_sems指定需要的訊號量數目,它的值幾乎總是1。

第三個引數sem_flags是一組標誌,當想要當訊號量不存在時建立一個新的訊號量,可以和值IPC_CREAT做按位或操作。設定了IPC_CREAT標誌後,即使給出的鍵是一個已有訊號量的鍵,也不會產生錯誤。而IPC_CREAT | IPC_EXCL則可以建立一個新的,唯一的訊號量,如果訊號量已存在,返回一個錯誤。

semget()函式成功返回一個相應訊號識別符號(非零),失敗返回-1.

2、semop()函式

它的作用是改變訊號量的值,原型為:

int semop(int sem_id, struct sembuf *sem_opa, size_t num_sem_ops);

sem_id是由semget()返回的訊號量識別符號,sembuf結構的定義如下:

#include <unistd.h> #include <sys/types.h> #include <sys/stat.h> #include <fcntl.h> #include <stdlib.h> #include <stdio.h> #include <string.h> #include <sys/sem.h>

//union semun //{ // int val; // struct semid_ds *buf; // unsigned short *arry; //};

static int sem_id = 0; static int set_semvalue(); static void del_semvalue(); static int semaphore_p(); static int semaphore_v();

int main(int argc, char *argv[]) { char message = ‘X’; int i = 0;

// 建立訊號量
sem_id = semget((key_t) 1234, 1, 0666 | IPC_CREAT);

if (argc > 1)
{
    // 程式第一次被呼叫,初始化訊號量
    if (!set_semvalue())
    {
        fprintf(stderr, "Failed to initialize semaphore\n");
        exit(EXIT_FAILURE);
    }

    // 設定要輸出到螢幕中的資訊,即其引數的第一個字元
    message = argv[1][0];
    sleep(2);
}

for (i = 0; i < 10; ++i)
{
    // 進入臨界區
    if (!semaphore_p())
    {
        exit(EXIT_FAILURE);
    }

    // 向螢幕中輸出資料
    printf("%c", message);

    // 清理緩衝區,然後休眠隨機時間
    fflush(stdout);
    sleep(rand() % 3);

    // 離開臨界區前再一次向螢幕輸出資料
    printf("%c", message);
    fflush(stdout);

    // 離開臨界區,休眠隨機時間後繼續迴圈
    if (!semaphore_v())
    {
        exit(EXIT_FAILURE);
    }
    sleep(rand() % 2);
}

sleep(10);
printf("\n%d - finished\n", getpid());

if (argc > 1)
{
    // 如果程式是第一次被呼叫,則在退出前刪除訊號量
    sleep(3);
    del_semvalue();
}
exit(EXIT_SUCCESS);

}

static int set_semvalue() { // 用於初始化訊號量,在使用訊號量前必須這樣做 union semun sem_union;

sem_union.val = 1;
if (semctl(sem_id, 0, SETVAL, sem_union) == -1)
{
    return 0;
}
return 1;

}

static void del_semvalue() { // 刪除訊號量 union semun sem_union;

if (semctl(sem_id, 0, IPC_RMID, sem_union) == -1)
{
    fprintf(stderr, "Failed to delete semaphore\n");
}

}

static int semaphore_p() { // 對訊號量做減1操作,即等待P(sv) struct sembuf sem_b; sem_b.sem_num = 0; sem_b.sem_op = -1;//P() sem_b.sem_flg = SEM_UNDO; if (semop(sem_id, &sem_b, 1) == -1) { fprintf(stderr, “semaphore_p failed\n”); return 0; }

return 1;

}

static int semaphore_v() { // 這是一個釋放操作,它使訊號量變為可用,即傳送訊號V(sv) struct sembuf sem_b; sem_b.sem_num = 0; sem_b.sem_op = 1; // V() sem_b.sem_flg = SEM_UNDO; if (semop(sem_id, &sem_b, 1) == -1) { fprintf(stderr, “semaphore_v failed\n”); return 0; }

return 1;

}

執行結果如下:

注:這個程式的臨界區為main函式for迴圈不的semaphore_p()和semaphore_v()函式中間的程式碼。

例子分析:

同時執行一個程式的兩個例項,注意第一次執行時,要加上一個字元作為引數,例如本例中的字元‘O’,它用於區分是否為第一次呼叫,同時這個字元輸出到螢幕中。因為每個程式都在其進入臨界區後和離開臨界區前列印一個字元,所以每個字元都應該成對出現,正如你看到的上圖的輸出那樣。在main函式中迴圈中我們可以看到,每次程序要訪問stdout(標準輸出),即要輸出字元時,每次都要檢查訊號量是否可用(即stdout有沒有正在被其他程序使用)。所以,當一個程序A在呼叫函式semaphore_p()進入了臨界區,輸出字元後,呼叫sleep()時,另一個程序B可能想訪問stdout,但是訊號量的P請求操作失敗,只能掛起自己的執行,當程序A呼叫函式semaphore_v()離開了臨界區,程序B馬上被恢復執行。然後程序A和程序B就這樣一直迴圈了10次。 五、對比例子——程序間的資源競爭

看了上面的例子,你可能還不是很明白,不過沒關係,下面我就以另一個例子來說明一下,它實現的功能與前面的例子一樣,執行方式也一樣,都是兩個相同的程序,同時向stdout中輸出字元,只是沒有使用訊號量,兩個程序在互相競爭stdout。它的程式碼非常簡單,檔名為normalprint.c,程式碼如下:

#include <stdio.h> #include <stdlib.h>

int main(int argc, char *argv[]) { char message = ‘X’; int i = 0; if (argc > 1) { message = argv[1][0]; }

for (i = 0; i < 10; ++i)
{
    printf("%c", message);
    fflush(stdout);
    sleep(rand() % 3);
    printf("%c", message);
    fflush(stdout);
    sleep(rand() % 2);
}
sleep(10);
printf("\n%d - finished\n", getpid());
 
exit(EXIT_SUCCESS);

}

執行結果如下:

例子分析:

從上面的輸出結果,我們可以看到字元‘X’和‘O’並不像前面的例子那樣,總是成對出現,因為當第一個程序A輸出了字元後,呼叫sleep休眠時,另一個程序B立即輸出並休眠,而程序A醒來時,再繼續執行輸出,同樣的程序B也是如此。所以輸出的字元就是不成對的出現。這兩個程序在競爭stdout這一共同的資源。通過兩個例子的對比,我想訊號量的意義和使用應該比較清楚了。 六、訊號量的總結

訊號量是一個特殊的變數,程式對其訪問都是原子操作,且只允許對它進行等待(即P(訊號變數))和傳送(即V(訊號變數))資訊操作。我們通常通過訊號來解決多個程序對同一資源的訪問競爭的問題,使在任一時刻只能有一個執行執行緒訪問程式碼的臨界區域,也可以說它是協調程序間的對同一資源的訪問權,也就是用於同步程序的。

注意:

例子中的程序分為兩種,一種是執行時帶有命令列引數的程序,一種是執行時不帶有引數的程序。前者負責訊號量的建立、設定初值並銷燬,而後者直接獲得並使用前者建立並設定好的訊號量。

參考:

《Linux 高效能伺服器程式設計》