Linux中进程间通信——共享内存

Linux中进程间通信——共享内存

 

一、共享内存的定义

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

 

特别提醒：共享内存并未提供同步机制，也就是说，在第一个进程结束对共享内存的写操作之前，并无自动机制可以阻止第二个进程开始对它进行读取。所以我们通常需要用其他的机制来同步对共享内存的访问，例如前面说到的信号量。有关信号量的更多内容，可以查阅我的另一篇博客：Linux中进程间通信——信号量机制

 

二、共享内存的函数

与信号量一样，在Linux中也提供了一组函数接口用于使用共享内存，而且使用共享共存的接口还与信号量的非常相似，而且比使用信号量的接口来得简单，它们声明在头文件 sys/shm.h中。

 

1、shmget函数

该函数用来创建共享内存，它的原型为：

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

 

（1）第一个参数，与信号量的semget函数一样，程序需要提供一个参数key（非 0整数），它有效地为共享内存段命名，shmget函数成功时返回一个与key相关的共享内存标识符（非负整数），用于后续的共享内存函数。调用失败返回-1.

 

不相关的进程可以通过该函数的返回值访问同一共享内存，它代表程序可能要使用的某个资源，程序对所有共享内存的访问都是间接的，程序先通过调用shmget函数并提供一个键，再由系统生成一个相应的共享内存标识符（shmget函数的返回值），只有shmget函数才直接使用信号量键，所有其他的信号量函数使用由semget函数返回的信号量标识符。

（2）第二个参数，size以字节为单位指定需要共享的内存容量。

（3）第三个参数，shmflg是权限标志，它的作用与open函数的mode参数一样，如果要想在key标识的共享内存不存在时，创建它的话，可以与IPC_CREAT做或操作。共享内存的权限标志与文件的读写权限一样，举例来说，0644,它表示允许一个进程创建的共享内存被内存创建者所拥有的进程向共享内存读取和写入数据，同时其他用户创建的进程只能读取共享内存。

 

2、shmat函数

第一次创建完共享内存时，它还不能被任何进程访问，shmat函数的作用就是用来启动对该共享内存的访问，并把共享内存连接到当前进程的地址空间。它的原型如下：

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

（1）第一个参数，shm_id是由shmget函数返回的共享内存标识。

（2）第二个参数，shm_addr指定共享内存连接到当前进程中的地址位置，通常为空，表示让系统来选择共享内存的地址。

（3）第三个参数，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);  

 

（1）第一个参数，shm_id是shmget函数返回的共享内存标识符。

（2）第二个参数，command是要采取的操作，它可以取下面的三个值 ：

    IPC_STAT：把shmid_ds结构中的数据设置为共享内存的当前关联值，即用共享内存的当前关联值覆盖shmid_ds的值。

    IPC_SET：如果进程有足够的权限，就把共享内存的当前关联值设置为shmid_ds结构中给出的值

    IPC_RMID：删除共享内存段

（3）第三个参数，buf是一个结构指针，它指向共享内存模式和访问权限的结构。

shmid_ds结构至少包括以下成员：

struct shmid_ds  

{  

    uid_t shm_perm.uid;  

    uid_t shm_perm.gid;  

    mode_t shm_perm.mode;  

};  

三、使用共享内存完成两个进程间的通信(读、写)

sem.h

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

union semun
{
	int val;
};

void sem_init(int,int);//因为要用到两个信号量，所以初始化时直接初始两个
void sem_p(int);
void sem_v(int);
void sem_destroy();

sem.c

#include "sem.h"

static int semid = 0;

void init(int in1,int in2)
{
	if((semid  =  semget((key_t)1234,2,IPC_CREAT|IPC_EXCL|0600)) == -1)
	{
		if((semid = semget((key_t)1234,2,IPC_EXCL)) == -1)
		{
			perror("semget error;\n");
		}
	}
	else
	{
		union semun buf;
		buf.val = in1;
		if(semctl(semid,0,SETVAL,buf) == -1)
		{
			perror("semctl error");
		}

		union semun buf1;
		buf1.val = in2;
		if(semctl(semid,1,SETVAL,buf1) == -1)
		{
			perror("semctl error");
		}
	}
}
void sem_p(int num)
{
	struct sembuf buff;
	buff.sem_num = num;
	buff.sem_op = -1;
	buff.sem_flg = SEM_UNDO;
	if(semop(semid,&buff,1) == -1)
	{
		perror("semop error");
	}
}
void sem_v(int num)
{

	struct sembuf buff;
	buff.sem_num = num;
	buff.sem_op = 1;
	buff.sem_flg = SEM_UNDO;
	if(semop(semid,&buff,1) == -1)
	{
		perror("semop error");
	}
}

void destroy()
{
	if(semctl(semid,0,IPC_RMID) == -1)
	{
		perror("destory error");
	}

}

write.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <assert.h>
#include <fcntl.h>
#include <sys/sem.h>
#include <sys/shm.h>
#include "sem.h"

int main()
{
	int shmid = shmget((key_t)1234,128,IPC_CREAT|0600);
	if(shmid == -1)
	{
		perror("shmget error:");
	}
	void * id = (void *)0;
	id = shmat(shmid,NULL,0);
	if(id  == (void  *)-1)
	{
		perror("shmat error:");
	}

	char * s = (char * )id;

	init(1,0);
	int i = 0;
	while(1)
	{
		sem_p(0);
		fgets(s,128,stdin);

		sem_v(1);
		if(strncmp(s,"end",3) == 0)
		{
			break;
		}
	}

	if(shmdt(s) == -1)
	{
		perror("shmdt error:");
	}
	sleep(10);
	shmctl(shmid,IPC_RMID,0);
	destroy();
}

read.c

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

int main()
{
	int shmid = shmget((key_t)1234,128,IPC_CREAT|0600);
	if(shmid == -1)
	{
		perror("shmget error:");
	}
	char *s = (char *)shmat(shmid,NULL,0);
	if(s == (char *)-1)
	{
		perror("shmat error:");
	}
	init(1,0);
       //init(0);
        int i = 0;
	while(1)
	{
		sem_p(1);
		if(strncmp(s,"end",3) == 0)
		{
			break;
		}
		printf("%s",s);
		
		sem_v(0);
	}
	if(shmdt(s) == -1)
	{
		perror("shmdt error:");
	}
}

运行结果：

解析：打开两个终端(两个进程)，一个进行写操作，一个进行读操作，

      当写进程写入"end"之后，读进程结束。