linux c之使用共享内存实现进程间通信

2/22/2017来源:ASP.NET技巧人气:2046

这篇博客有别人的也有自己改的,作为读书笔记,勿喷。

1、共享内存的介绍

共享内存就是允许两个不相关的进程访问同一个逻辑内存。共享内存是在两个正在运行的进程之间共享和传递数据的一种非常有效的方式。不同进程之间共享的内存通常安排为同一段物理内存。进程可以将同一段共享内存连接到它们自己的地址空间中,所有进程都可以访问共享内存中的地址,就好像它们是由用C语言函数malloc分配的内存一样。而如果某个进程向共享内存写入数据,所做的改动将立即影响到可以访问同一段共享内存的任何其他进程。 特别提醒:共享内存并未提供同步机制,也就是说,在第一个进程结束对共享内存的写操作之前,并无自动机制可以阻止第二个进程开始对它进行读取。所以我们通常需要用其他的机制来同步对共享内存的访问

2、共享内存的常用api介绍

头文件 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函数的作用就是用来启动对该共享内存的访问,并把共享内存连接到当前进程的地址空间。它的原型如下: 在CODE上查看代码片派生到我的代码片

    void *shmat(int shm_id, const void *shm_addr, int shmflg);  
第一个参数,shm_id是由shmget函数返回的共享内存标识。
第二个参数,shm_addr指定共享内存连接到当前进程中的地址位置,通常为空,表示让系统来选择共享内存的地址。
第三个参数,shm_flg是一组标志位,通常为0。

调用成功时返回一个指向共享内存第一个字节的指针,如果调用失败返回-1.

3、shmdt函数

该函数用于将共享内存从当前进程中分离。注意,将共享内存分离并不是删除它,只是使该共享内存对当前进程不再可用。它的原型如下: 在CODE上查看代码片派生到我的代码片

    int shmdt(const void *shmaddr);  
参数shmaddr是shmat函数返回的地址指针,调用成功时返回0,失败时返回-1.

4、shmctl函数

与信号量的semctl函数一样,用来控制共享内存,它的原型如下: 在CODE上查看代码片派生到我的代码片

    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结构至少包括以下成员:
[cpp] view plain copy
PRint?在CODE上查看代码片派生到我的代码片

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

3、通过共享内存实现进程间通信

         其中一个文件shm_read.c创建共享内存,并读取其中的信息,另一个文件shm_write.c向共享内存中写入数据。为了方便操作和数据结构的统一,为这两个文件定义了相同的数据结构,定义在文件shmdata.c中。结构shared_use_st中的written作为一个可读或可写的标志,非0:表示可读,0表示可写,text则是内存中的文件。

shared_use_st.h 文件如下

#ifndef UNTITLED_SHMDATA_H
#define UNTITLED_SHMDATA_H



#define TEXT_SIZE 2048

struct share_use_st
{
    int written; //作为一个标志,非0表示可读,0表示可写
    char text[TEXT_SIZE];
};

#endif //UNTITLED_SHMDATA_H

shm_write.c 文件如下

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

#define TEXT_SIZE 2048

//#include "shmdata.h"
struct shared_use_st
{
    int written; //作为一个标志,非0表示可读,0表示可写
    char text[TEXT_SIZE];
};

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 fail\n");
        exit(EXIT_FAILURE);
    }
    //将共享内存连接到当前进程的地址空间
    shm = shmat(shmid,  (void*)0, 0);
    if (shm == (void*) -1)
    {
        fprintf(stderr, "shmat fail\n");
    }
    printf("Memory attached at %x\n", (int)shm);
    //设置共享内存
    shared = (struct shared_use_st *)shm;
    while (running)  //向内存共享中写入数据
    {
        //数据还没有被读取,则等待数据被读取,不能向共享内存中写入文本
        while (shared->written == 1)
        {
            sleep(1);
            puts("waitting ....");
        }
        //向共享内存中写入数据
        printf("Enter some text:\n");
        fgets(buffer , BUFSIZ, stdin);
        strncpy(shared -> text, buffer, TEXT_SIZE);
        //写完数据,设置written使共享内存可读
        shared -> written = 1;
        //输入end,退出循环
        if (strncmp(buffer, "end", 3) == 0)
        {
            running = 0;
        }
    }
    //把内存共享从当前进程中分离
    if (shmdt(shm) == -1)
    {
        fprintf(stderr, "shmdt fail\n");
        exit(EXIT_FAILURE);
    }
    sleep(1);
    puts("exit_success\n");
    exit(EXIT_SUCCESS);
}


shm_read.c文件如下

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

#include "shared_use_st.h"

int main() {
    int running = 1;//程序是否运行标志
    void *shm = NULL;//分配的共享内存的原始首地址
    struct shared_use_st *shared = NULL; //指向内存
    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("Memory 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);
    }
    puts("exit success\n");
    exit(EXIT_SUCCESS);
}


4、运行结果和总结

这个程序是不安全的,当有多个程序同时向共享内存中读写数据时,问题就会出现。可能你会认为,可以改变一下written的使用方式,例如,只有当written为0时进程才可以向共享内存写入数据,而当一个进程只有在written不为0时才能对其进行读取,同时把written进行加1操作,读取完后进行减1操作。这就有点像文件锁中的读写锁的功能。咋看之下,它似乎能行得通。但是这都不是原子操作,所以这种做法是行不能的。试想当written为0时,如果有两个进程同时访问共享内存,它们就会发现written为0,于是两个进程都对其进行写操作,显然不行。当written为1时,有两个进程同时对共享内存进行读操作时也是如些,当这两个进程都读取完是,written就变成了-1. 要想让程序安全地执行,就要有一种进程同步的进制,保证在进入临界区的操作是原子操作。例如,可以使用前面所讲的信号量来进行进程的同步。因为信号量的操作都是原子性的。

1、优点:我们可以看到使用共享内存进行进程间的通信真的是非常方便,而且函数的接口也简单,数据的共享还使进程间的数据不用传送,而是直接访问内存,也加快了程序的效率。同时,它也不像匿名管道那样要求通信的进程有一定的父子关系。 2、缺点:共享内存没有提供同步的机制,这使得我们在使用共享内存进行进程间通信时,往往要借助其他的手段来进行进程间的同步工作。