linux進程間通訊-共享內存

轉載：http://www.cnblogs.com/fangshenghui/p/4039720.html

一 共享內存介紹

共享內存能夠從字面上去理解，就把一片邏輯內存共享出來，讓不一樣的進程去訪問它，修改它。共享內存是在兩個正在運行的進程之間共享和傳遞數據的一種很是有效的方式。不一樣進程之間共享的內存一般安排爲同一段物理內存。進程能夠將同一段共享內存鏈接到它們本身的地址空間中，全部進程均可以訪問共享內存中的地址，就好像它們是由用C語言函數malloc分配的內存同樣。而若是某個進程向共享內存寫入數據，所作的改動將當即影響到能夠訪問同一段共享內存的任何其餘進程。

但有一點特別要注意：共享內存並未提供同步機制。也就是說，在第一個進程結束對共享內存的寫操做以前，並沒有自動機制能夠阻止第二個進程開始對它進行讀取。因此咱們一般須要用其餘的機制來同步對共享內存的訪問，例如信號量。

二 共享內存的使用

1.建立共享內存

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

◇第一個參數，共享內存段的命名，shmget函數成功時返回一個與key相關的共享內存標識符（非負整數），用於後續的共享內存函數。調用失敗返回-1.

 

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

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

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

　　　　0:(取共享內存標識符，若不存在則函數會報錯)。

　　　　IPC_CREAT:當shmflg&IPC_CREAT爲真時，若是內核中不存在鍵值與key相等的共享內存，則新建一個共享內存；若是存在這樣的共享內存，返回此共享內存的標識符

　　　　IPC_CREAT|IPC_EXCL：若是內核中不存在鍵值與key相等的共享內存，則新建一個消息隊列；若是存在這樣的共享內存則報錯

 

2.啓動對該共享內存的訪問

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

◇第一次建立完共享內存時，它還不能被任何進程訪問，shmat函數的做用就是用來啓動對該共享內存的訪問，並把共享內存鏈接到當前進程的地址空間。

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

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

◇第三個參數，shm_flg是一組標誌位，一般爲0。(SHM_RDONLY：爲只讀模式，其餘爲讀寫模式)　　　　　

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

 

3.將共享內存從當前進程中分離

int shmdt(const void *shmaddr);

◇該函數用於將共享內存從當前進程中分離。注意，將共享內存分離並非刪除它，只是使該共享內存對當前進程再也不可用。

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

 

4.控制共享內存

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是一個結構指針，它指向共享內存模式和訪問權限的結構。

◇成功：0　　出錯：-1，錯誤緣由存於error中

1 struct shmid_ds
2 {
3 　　uid_t shm_perm.uid;
4 　　uid_t shm_perm.gid;
5 　　mode_t shm_perm.mode;
6 ｝

三 例子

 1 //shmdata.h
 2 #ifndef _SHMDATA_H_HEADER
 3 #define _SHMDATA_H_HEADER
 4 #define TEXT_SZ 2048
 5 
 6 struct shared_use_st
 7 {
 8     int written;/* 做爲一個標誌，非0：表示可讀，0表示可寫 */
 9     char text[TEXT_SZ];/* 記錄寫入和讀取的文本 */
10 };
11 
12 #endi 1 //shmread.c

//shmread.c
int main()
{
　　int running =1; //程序是否繼續運行的標誌
　　void*shm = NULL; //分配的共享內存的原始首地址
　　struct shared_use_st *shared;//指向shm
　　int shmid; //共享內存標識符

　　//建立共享內存
    shmid = shmget((key_t)MEM_KEY,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"
#define MEM_KEY (1234)
 
int main()
{
　　int running =1;
　　void*shm = NULL;
　　struct shared_use_st *shared = NULL;
　　char buffer[BUFSIZ +1];//用於保存輸入的文本
　　int shmid;
　　//建立共享內存
　　shmid = shmget((key_t)MEM_KEY,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);
}

代碼分析：

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

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

關於前面程序安全性的討論：

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

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

使用共享內存的優缺點

◇優勢：咱們能夠看到使用共享內存進行進程間的通訊真的是很是方便，並且函數的接口也簡單，數據的共享還使進程間的數據不用傳送，而是直接訪問內存，也加快了程序的效率。同時，它也不像匿名管道那樣要求通訊的進程有必定的父子關係。

◇缺點：共享內存沒有提供同步的機制，這使得咱們在使用共享內存進行進程間通訊時，每每要藉助其餘的手段來進行進程間的同步工做。