（轉）Linux進程間通信——使用共享內存

一、什麼是共享內存

顧名思義，共享內存就是允許兩個不相關的進程訪問同一個邏輯內存。共享內存是在兩個正在運行的進程之間共享和傳遞數據的一種非常有效的方式。不同進程之間共享的內存通常安排爲同一段物理內存。進程可以將同一段共享內存連接到它們自己的地址空間中，所有進程都可以訪問共享內存中的地址，就好像它們是由用C語言函數malloc分配的內存一樣。而如果某個進程向共享內存寫入數據，所做的改動將立即影響到可以訪問同一段共享內存的任何其他進程。

特別提醒：共享內存並未提供同步機制，也就是說，在第一個進程結束對共享內存的寫操作之前，並無自動機制可以阻止第二個進程開始對它進行讀取。所以我們通常需要用其他的機制來同步對共享內存的訪問，例如前面說到的信號量。有關信號量的更多內容，可以查閱我的另一篇文章：Linux進程間通信——使用信號量

二、共享內存的使得

與信號量一樣，在Linux中也提供了一組函數接口用於使用共享內存，而且使用共享共存的接口還與信號量的非常相似，而且比使用信號量的接口來得簡單。它們聲明在頭文件 sys/shm.h中。

1、shmget函數

該函數用來創建共享內存，它的原型爲：

int shmget(key_t key, size_t size, int shmflg);

第一個參數，與信號量的semget函數一樣，程序需要提供一個參數key（非0整數），它有效地爲共享內存段命名，shmget函數成功時返回一個與key相關的共享內存標識符（非負整數），用於後續的共享內存函數。調用失敗返回-1.

不相關的進程可以通過該函數的返回值訪問同一共享內存，它代表程序可能要使用的某個資源，程序對所有共享內存的訪問都是間接的，程序先通過調用shmget函數並提供一個鍵，再由系統生成一個相應的共享內存標識符（shmget函數的返回值），只有shmget函數才直接使用信號量鍵，所有其他的信號量函數使用由semget函數返回的信號量標識符。

第二個參數，size以字節爲單位指定需要共享的內存容量

第三個參數，shmflg是權限標誌，它的作用與open函數的mode參數一樣，如果要想在key標識的共享內存不存在時，創建它的話，可以與IPC_CREAT做或操作。共享內存的權限標誌與文件的讀寫權限一樣，舉例來說，0644,它表示允許一個進程創建的共享內存被內存創建者所擁有的進程向共享內存讀取和寫入數據，同時其他用戶創建的進程只能讀取共享內存。

2、shmat函數

第一次創建完共享內存時，它還不能被任何進程訪問，shmat函數的作用就是用來啓動對該共享內存的訪問，並把共享內存連接到當前進程的地址空間。它的原型如下：

void *shmat(int shm_id, const void *shm_addr, int shmflg);

第一個參數，shm_id是由shmget函數返回的共享內存標識。

第二個參數，shm_addr指定共享內存連接到當前進程中的地址位置，通常爲空，表示讓系統來選擇共享內存的地址。

第三個參數，shm_flg是一組標誌位，通常爲0。

調用成功時返回一個指向共享內存第一個字節的指針，如果調用失敗返回-1.

3、shmdt函數

該函數用於將共享內存從當前進程中分離。注意，將共享內存分離並不是刪除它，只是使該共享內存對當前進程不再可用。它的原型如下：

int shmdt(const void *shmaddr);

參數shmaddr是shmat函數返回的地址指針，調用成功時返回0，失敗時返回-1.

4、shmctl函數

與信號量的semctl函數一樣，用來控制共享內存，它的原型如下：

int shmctl(int shm_id, int command, struct shmid_ds *buf);

第一個參數，shm_id是shmget函數返回的共享內存標識符。

第二個參數，command是要採取的操作，它可以取下面的三個值 ：

    IPC_STAT：把shmid_ds結構中的數據設置爲共享內存的當前關聯值，即用共享內存的當前關聯值覆蓋shmid_ds的值。

    IPC_SET：如果進程有足夠的權限，就把共享內存的當前關聯值設置爲shmid_ds結構中給出的值

    IPC_RMID：刪除共享內存段

第三個參數，buf是一個結構指針，它指向共享內存模式和訪問權限的結構。

shmid_ds結構至少包括以下成員：

struct shmid_ds
{
    uid_t shm_perm.uid;
    uid_t shm_perm.gid;
    mode_t shm_perm.mode;
};

三、使用共享內存進行進程間通信

說了這麼多，又到了實戰的時候了。下面就以兩個不相關的進程來說明進程間如何通過共享內存來進行通信。其中一個文件shmread.c創建共享內存，並讀取其中的信息，另一個文件shmwrite.c向共享內存中寫入數據。爲了方便操作和數據結構的統一，爲這兩個文件定義了相同的數據結構，定義在文件shmdata.c中。結構shared_use_st中的written作爲一個可讀或可寫的標誌，非0：表示可讀，0表示可寫，text則是內存中的文件。

shmdata.c的源代碼如下：

#ifndef _SHMDATA_H_HEADER
#define _SHMDATA_H_HEADER

#define TEXT_SZ 2048

struct shared_use_st
{
    int written;//作爲一個標誌，非0：表示可讀，0表示可寫
    char text[TEXT_SZ];//記錄寫入和讀取的文本
};

#endif

源文件shmread.c的源代碼如下：

#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <sys/shm.h>
#include "shmdata.h"

int main()
{
    int running = 1;//程序是否繼續運行的標誌
    void *shm = NULL;//分配的共享內存的原始首地址
    struct shared_use_st *shared;//指向shm
    int shmid;//共享內存標識符
    //創建共享內存
    shmid = shmget((key_t)1234, sizeof(struct shared_use_st), 0666|IPC_CREAT);
    if(shmid == -1)
    {
        fprintf(stderr, "shmget failed\n");
        exit(EXIT_FAILURE);
    }
    //將共享內存連接到當前進程的地址空間
    shm = shmat(shmid, 0, 0);
    if(shm == (void*)-1)
    {
        fprintf(stderr, "shmat failed\n");
        exit(EXIT_FAILURE);
    }
    printf("\nMemory attached at %X\n", (int)shm);
    //設置共享內存
    shared = (struct shared_use_st*)shm;
    shared->written = 0;
    while(running)//讀取共享內存中的數據
    {
        //沒有進程向共享內存定數據有數據可讀取
        if(shared->written != 0)
        {
            printf("You wrote: %s", shared->text);
            sleep(rand() % 3);
            //讀取完數據，設置written使共享內存段可寫
            shared->written = 0;
            //輸入了end，退出循環（程序）
            if(strncmp(shared->text, "end", 3) == 0)
                running = 0;
        }
        else//有其他進程在寫數據，不能讀取數據
            sleep(1);
    }
    //把共享內存從當前進程中分離
    if(shmdt(shm) == -1)
    {
        fprintf(stderr, "shmdt failed\n");
        exit(EXIT_FAILURE);
    }
    //刪除共享內存
    if(shmctl(shmid, IPC_RMID, 0) == -1)
    {
        fprintf(stderr, "shmctl(IPC_RMID) failed\n");
        exit(EXIT_FAILURE);
    }
    exit(EXIT_SUCCESS);
}

源文件shmwrite.c的源代碼如下：

#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <sys/shm.h>
#include "shmdata.h"

int main()
{
    int running = 1;
    void *shm = NULL;
    struct shared_use_st *shared = NULL;
    char buffer[BUFSIZ + 1];//用於保存輸入的文本
    int shmid;
    //創建共享內存
    shmid = shmget((key_t)1234, sizeof(struct shared_use_st), 0666|IPC_CREAT);
    if(shmid == -1)
    {
        fprintf(stderr, "shmget failed\n");
        exit(EXIT_FAILURE);
    }
    //將共享內存連接到當前進程的地址空間
    shm = shmat(shmid, (void*)0, 0);
    if(shm == (void*)-1)
    {
        fprintf(stderr, "shmat failed\n");
        exit(EXIT_FAILURE);
    }
    printf("Memory attached at %X\n", (int)shm);
    //設置共享內存
    shared = (struct shared_use_st*)shm;
    while(running)//向共享內存中寫數據
    {
        //數據還沒有被讀取，則等待數據被讀取,不能向共享內存中寫入文本
        while(shared->written == 1)
        {
            sleep(1);
            printf("Waiting...\n");
        }
        //向共享內存中寫入數據
        printf("Enter some text: ");
        fgets(buffer, BUFSIZ, stdin);
        strncpy(shared->text, buffer, TEXT_SZ);
        //寫完數據，設置written使共享內存段可讀
        shared->written = 1;
        //輸入了end，退出循環（程序）
        if(strncmp(buffer, "end", 3) == 0)
            running = 0;
    }
    //把共享內存從當前進程中分離
    if(shmdt(shm) == -1)
    {
        fprintf(stderr, "shmdt failed\n");
        exit(EXIT_FAILURE);
    }
    sleep(2);
    exit(EXIT_SUCCESS);
}

再來看看運行的結果：

分析：

1、程序shmread創建共享內存，然後將它連接到自己的地址空間。在共享內存的開始處使用了一個結構struct_use_st。該結構中有個標誌written，當共享內存中有其他進程向它寫入數據時，共享內存中的written被設置爲0，程序等待。當它不爲0時，表示沒有進程對共享內存寫入數據，程序就從共享內存中讀取數據並輸出，然後重置設置共享內存中的written爲0，即讓其可被shmwrite進程寫入數據。

2、程序shmwrite取得共享內存並連接到自己的地址空間中。檢查共享內存中的written，是否爲0，若不是，表示共享內存中的數據還沒有被完，則等待其他進程讀取完成，並提示用戶等待。若共享內存的written爲0，表示沒有其他進程對共享內存進行讀取，則提示用戶輸入文本，並再次設置共享內存中的written爲1，表示寫完成，其他進程可對共享內存進行讀操作。

四、關於前面的例子的安全性討論

這個程序是不安全的，當有多個程序同時向共享內存中讀寫數據時，問題就會出現。可能你會認爲，可以改變一下written的使用方式，例如，只有當written爲0時進程纔可以向共享內存寫入數據，而當一個進程只有在written不爲0時才能對其進行讀取，同時把written進行加1操作，讀取完後進行減1操作。這就有點像文件鎖中的讀寫鎖的功能。咋看之下，它似乎能行得通。但是這都不是原子操作，所以這種做法是行不能的。試想當written爲0時，如果有兩個進程同時訪問共享內存，它們就會發現written爲0，於是兩個進程都對其進行寫操作，顯然不行。當written爲1時，有兩個進程同時對共享內存進行讀操作時也是如些，當這兩個進程都讀取完是，written就變成了-1.

要想讓程序安全地執行，就要有一種進程同步的進制，保證在進入臨界區的操作是原子操作。例如，可以使用前面所講的信號量來進行進程的同步。因爲信號量的操作都是原子性的。

五、使用共享內存的優缺點

1、優點：我們可以看到使用共享內存進行進程間的通信真的是非常方便，而且函數的接口也簡單，數據的共享還使進程間的數據不用傳送，而是直接訪問內存，也加快了程序的效率。同時，它也不像匿名管道那樣要求通信的進程有一定的父子關係。

2、缺點：共享內存沒有提供同步的機制，這使得我們在使用共享內存進行進程間通信時，往往要藉助其他的手段來進行進程間的同步工作。