Linux进程间通信（3）共享内存

共享内存

共享内存原理

共享内存为多个进程之间共享和传递数据提供了一种有效的方式。共享内存是先在物理内存上申请一块空间，多个进程可以将其映射到自己的虚拟地址空间中。所有进程都可以访问共享内存中的地址，就好像它们是由 malloc 分配的一样。如果某个进程向共享内存写入了数据，所做的改动将立刻被可以访问同一段共享内存的任何其他进程看到。由于它并未提供同步机制，所以我们通常需要用其他的机制来同步对共享内存的访问。

共享内存接口介绍

1、头文件

#include <sys/ipc.h>
#include <sys/shm.h>
#include <sys/types.h>

2、接口

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

shmget()用于创建或者获取共享内存
shmget()成功返回共享内存的 ID， 失败返回-1
key： 不同的进程使用相同的 key 值可以获取到同一个共享内存
size： 创建共享内存时，指定要申请的共享内存空间大小
shmflg: IPC_CREAT IPC_EXCL

• void shmat( int shmid, const void shmaddr, int shmflg);

shmat()将申请的共享内存的物理内存映射到当前进程的虚拟地址空间上
shmat()成功返回返回共享内存的首地址，失败返回 NULL
shmaddr：一般给 NULL，由系统自动选择映射的虚拟地址空间
shmflg： 一般给 0， 可以给 SHM_RDONLY 为只读模式，其他的为读写

• *int shmdt( const void shmaddr);

shmdt()断开当前进程的 shmaddr 指向的共享内存映射
shmdt()成功返回 0， 失败返回-1

• *int shmctl( int shmid, int cmd, struct shmid_ds buf);

shmctl()控制共享内存
shmctl()成功返回 0，失败返回-1
cmd： IPC_RMID

示例代码

进程 a 从键盘循环获取数据并拷贝到共享内存中，进程 b 从共享内存中获
取并打印数据。要求进程 a 输入一次，进程 b 输出一次，进程 a 不输入，进程 b 也不输出。
sem.h

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

#define SEM_MAX 2
#define SEM1   0
#define SEM2   1
union semun
{
    int val;
};

void sem_init();
void sem_p(int index);
void sem_v(int index);
void sem_destroy();

sem.c

#include"sem.h"
static int semid=-1;
void sem_init()
{
    int arr[SEM_MAX]={1,0};
    semid=semget((key_t)1234,SEM_MAX,IPC_CREAT|IPC_EXCL|0600);
    if(semid==-1)
    {
        semid=semget((key_t)1234,SEM_MAX,IPC_CREAT|0600);
        if(semid==-1)
        {
            printf("creat sem failed\n");

             return ;
       }
    }
    else
    {
        union semun a;
        int i=0;
        for( ; i<SEM_MAX;i++)
        {
            a.val=arr[i];
            if(semctl(semid,i,SETVAL,a)==-1)
            {
                printf("semctl err\n");
            }
        }
       // a.val=1;
       // if(semctl(semid,0,SETVAL,a)==-1)//semid返回的信号量标识符 0信号量编号 ，setval将信号量初始化初始化为一个已知的值，也就是union semun中的val 其作用是在信号量第一次使用之前进行设置
       // {
       //     printf("semctl setval err\n");
       //     return;
       //}
       //}
    }
}
void sem_p(int index)
{
    if(index<0||index>=SEM_MAX)
    {
        printf("index err");
        return ;
    }
          struct sembuf buf;
          buf.sem_num=index;//信号量编号
          buf.sem_op=-1;//p
          buf.sem_flg=SEM_UNDO;//使操作系统跟踪当前进程对这个信号的修改情况

          if(semop(semid,&buf,1)==-1)
          {
              printf("semop p err\n");
          }
          
}

void  sem_v(int index)
{
    if(index<0||index>=SEM_MAX)
    {
        printf("index err");
        return ;
    }
    struct sembuf buf;
      buf.sem_num=index;
      buf.sem_op=1;
      buf.sem_flg=SEM_UNDO;

      if(semop(semid,&buf,1)==-1)
      {
          printf("semop v err\n");
        
      }

}


void sem_destroy()
{
    if(semctl(semid,0,IPC_RMID)==-1)//IPC_RMID删除一个无关继续使用的信号量标识符
    {
        perror("semctl del error");
    }
}

main.c发送数据

#include<sys/shm.h>
#include"sem.h"

int main()
{
    int shmid = shmget((key_t)2121,128,IPC_CREAT|0600);//key 有效共享内存段命名   shmget返回一个共享内存标识符   IPC..创建一个只属于创建进程的共享内存   size 128 共享内存容量  
    if(shmid == -1)
    {
        printf("shmget err \n");
        exit(0);
    }
    char *s = shmat(shmid, NULL, 0);//const void *shm_addr NULL连接到当前进程中的地址位置通常为空 系统来选择 int shmflg 0 一组标志位            成功返回指针，失败返沪-1
    if(s ==(char*)-1)
    {
        exit(0);
        printf("shmat err\n");
    }
    sem_init();
    while(1)
    {
        sem_p(SEM2);
        if( strncmp(s,"end",3)==0)
        {
            break;
        }
        printf("read:%s\n",s);
        sem_v(SEM1);
    }
    shmdt(s);//将共享内存从当前进程中分离  返回地址指针   并未删除它，只是使得该共享内存对当前进程不再可用
    sem_destroy();
    shmctl(shmid,IPC_RMID,NULL);//int shm_id (shmid) 是shmget 返回的标识符  command 采取的动作（IPC_RMID）删除共享内存段（ipc_stat）把shmid_da结构中的数据设置为共享内存的当前关联值（IPC_SET）如果进程没有足够的权限，就把共享内存当前关联之设置为shmid-ds结构中的值
}

test.c接受数据

#include<sys/shm.h>
#include"sem.h"

int main()
{
    int shmid = shmget((key_t)2121,128,IPC_CREAT|0600);
    if(shmid == -1)
    {
        printf("shmget err \n");
        exit(0);
    }
    char *s = shmat(shmid, NULL, 0);
    if(s ==(char*)-1)
    {
        exit(0);
        printf("shmat err\n");
    }
    sem_init();
    while(1)
    {
        printf("please input:");
        char buff[128]={0};
        fgets(buff,128,stdin);

        sem_p(SEM1);
        strcpy(s,buff);
        sem_v(SEM2);
        if(strncmp(buff,"end",3)==0)
        {
            break;
        }
    }
    shmdt(s);
    sem_destroy();
    shmctl(shmid,IPC_RMID,NULL);
}