父子进程间的共享内存通信

We can't give up trying,the fight was worth it.  --我们不能放弃尝试，奋斗是值得的。

======================================================================================

以前也学过共享内存，但是没有实战过，今天一起来学习下吧。

这里面的概念来自于我关注的一个博主的，写得非常棒，他是分为2个程序写的，我这个是一个父子进程间通信的

http://blog.csdn.net/ljianhui/article/details/10253345

一、什么是共享内存

顾名思义，共享内存就是允许两个不相关的进程访问同一个逻辑内存。共享内存是在两个正在运行的进程之间共享和传递数据的一种非常有效的方式。不同进程之间共享的内存通常安排为同一段物理内存。进程可以将同一段共享内存连接到它们自己的地址空间中，所有进程都可以访问共享内存中的地址，就好像它们是由用C语言函数malloc分配的内存一样。而如果某个进程向共享内存写入数据，所做的改动将立即影响到可以访问同一段共享内存的任何其他进程。

 二、共享内存的使得

与信号量一样，在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  

2.  {  

3.       uid_t shm_perm.uid;  

4.      uid_t shm_perm.gid;  

5.       mode_t shm_perm.mode;  

6.  };  

父子进程的通信代码：

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

struct shareMemory{

    int is_write;
    char data[20];
};

int main(int argc, const char *argv[])
{
    key_t  key_n ;
    int    shm_id;
    pid_t  pid;
    int running = 1;   //决定是否继续运行的变量
    void *shm = NULL;  //共享内存的首地址
    struct shareMemory *shared;
    char buf[1024+1];

    if((key_n = ftok("/",'s')<0))  //生成共享内存的key值
    {
        perror("Fail to ftok");
        exit(EXIT_SUCCESS);
    }
    printf("key num:%d\n",key_n);

    if((shm_id = shmget(key_n,1024,0666|IPC_CREAT))==-1)  //创建共享内存
    {
        perror("Fail to shmget");
        exit(EXIT_SUCCESS);
    }
    printf("share memory id:%d\n",shm_id);

    if((pid = fork())==-1)
    {
        perror("Fail to fork");
        exit(EXIT_SUCCESS);
    }

    if(pid==0)                                         //子进程，用来读
    {
        printf("I is son process!\n");
        shm = shmat(shm_id,0,0);                     //将共享内存连接到当前进程的地址空间
        if(shm==(void *)-1)
        {
            perror("Fail to shmat");
            exit(EXIT_SUCCESS);
        }
        printf(" son Memory attached at %x\n",(int)shm);

        shared = (struct shareMemory *)shm;        //设置共享内存
        shared->is_write = 0;

        while(running)  //读取共享内存中的数据
        {
            if(shared->is_write!=0)
            {
                printf("YOu worte:%s\n",shared->data);
                sleep(rand()%3);
                shared->is_write = 0;

                if(strncmp(shared->data,"end",3)==0)  //输入end退出循环
                {
                    running = 0;
                }
                else      //有其他进程在写数据，不能读
                    sleep(1);
            }
        }
        if(shmdt(shm)==-1)   //把共享内存从当前进程分离
        {
            perror("Fail to shmdt");
            exit(EXIT_SUCCESS);
        }

        if(shmctl(shm_id,IPC_RMID,0)==-1)      //删除这块共享内存
        {
            perror("Fail to shmctl(IPC_RMID)");
            exit(EXIT_SUCCESS);
        }
        exit(EXIT_SUCCESS);
    }

    if(pid!=0)                                        //父进程，用来写
    {

        printf("I is father process:%d\n",pid);
        shm = shmat(shm_id,0,0);                     //将共享内存连接到当前进程的地址空间
        if(shm==(void *)-1)
        {
            perror("Fail to shmat");
            exit(EXIT_SUCCESS);
        }
        printf("father Memory attached at %x\n",(int)shm);

        shared = (struct shareMemory *)shm;        //设置共享内存
        shared->is_write = 0;

        while(running)  //向共享内存中写数据
        {
            while(shared->is_write==1)
            {
                printf("wait...\n");
                sleep(1);
            }
            //向共享内存中写入数据
            printf("input some data:\n");
            fgets(buf,1024,stdin);
            strncpy(shared->data,buf,1024);
            shared->is_write = 1;        //写完数据后，置1使可读

            if(strncmp(shared->data,"end",3)==0)
            {
                running = 0;
            }
        } 


        if(shmdt(shm)==-1)   //把共享内存从这块进程分离
        {
            perror("Fail to shmdt");
            exit(EXIT_SUCCESS);
        }
        sleep(2);
        exit(EXIT_SUCCESS);
    }

    return 0;
}

运行结果：