Linux进程间通信-共享内存深入理解
继前三篇分析了进程间通信机制管道，命名管道（FIFO），消息队列后，本文将介绍最后一种进程间通信机制，也是进程间通信机制效率最高的一种-共享内存
1、共享内存
考虑前三种进程间通信机制，一个客户-服务器文件复制程序将设计到一下步骤：
（1）服务器从输入文件读取。该文件的数据由内核读入自己的内存空间，然后从内核复制到服务器进程。
（2）服务器往管道、FIFO和消息队列以一条消息的形式写入这些数据。这些IPC形式需要把进程中的数据复制到内核。
（3）客户端从这些IPC通道中读取数据，需要把内核数据复制到进程的地址空间。
（4）客户端将进程空间内的数据再写入内核，由内核写入文件。
这些IPC形式的问题在于，共享数据需要多次经过内核。
共享内存的方式是让同一块物理内存被映射到进程A、B各自的进程地址空间。进程A可以即时看到进程B对共享内存中数据的更新，反之亦然。由于多个进程共享同一块内存区域，必然需要某种同步机制（Linux同步的机制系列文章）。同样是上面的例子，共享内存只需经过如下 流程：
这种形式的通信数据只需复制两次：一次从输入文件到共享内存，另一次是从共享内存到输出文件。
2、共享内存基本操作
2.1 打开/创建一个共享内存对象
#include <sys/mman.h>
int shm_open(const char* name, int oflag, mode_t mode);
成功返回非负描述符，若失败返回-1
#include <unistd.h>
#include <stdio.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/mman.h>
int main(int argc, char* argv[])
{
int flag = O_RDWR | O_CREAT | O_EXCL;
int mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH;
int fd = 0;
fd = shm_open("/shm_test", flag, mode);
if (-1 == fd)
{
printf("open shm failed!\n");
return 1;
}
ftruncate(fd, 1024);
mmap(NULL,1024,PROT_READ|PROT_WRITE,MAP_SHARED,fd,0);
return 0;
}
2.2 删除一个共享对象的名字
#include <sys/mman.h>
int shm_unlink(const char* name);
成功返回0，失败返回-1
#include <unistd.h>
#include <stdio.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/mman.h>
int main(int argc, char* argv[])
{
shm_unlink("/shm_test");
return 0;
}
2.3 ftruncate 和 fstat
处理mmap的时候，普通文件或共享内存对象的大小可以通过ftruncate函数修改：
#include <unistd.h>
int ftruncate(int fd, off_t length);
成功返回0，出错返回-1
当打开一个已存在的共享内存区对象时，我们可调用fstat来获取有关该对象的信息：
#include <sys/types.h>
#include <sys/stat.h>
int fstat(int fd, struct stat* stat);
成功返回0，失败返回-1
对于普通文件stat结构可以获得12个以上的成员信息，然而当fd指代一个共享内存区对象时，只有四个成员含有信息。
struct stat{
...
mode_t st_mode;
uid_t st_uid;
gid_t st_gid;
off_t st_size;
};
2.4 共享内存的读写
read端代码如下：
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/mman.h>
int main(int argc, char* argv[])
{
int flag = O_RDONLY;
int mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH;
int fd = 0;
unsigned char* ptr, c;
fd = shm_open("/shm_test", flag, mode);
if (-1 == fd)
{
printf("open shm failed!\n");
return 1;
}
struct stat stat;
fstat(fd,&stat);
ptr = mmap(NULL, stat.st_size, PROT_READ, MAP_SHARED, fd, 0);
close(fd);
for (int i = 0; i < stat.st_size; i++)
{
if((c = *ptr++) != (i%256))
{
printf("read error!\n");
exit(-1);
}
else{
printf("%c\n",c);
}
}
return 0;
}
write端代码如下：
#include <unistd.h>
#include <stdio.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/mman.h>
int main(int argc, char* argv[])
{
int flag = O_RDWR;
int mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH;
int fd = 0;
unsigned char* ptr, c;
fd = shm_open("/shm_test", flag, mode);
if (-1 == fd)
{
printf("open shm failed!\n");
return 1;
}
struct stat stat;
fstat(fd,&stat);
ptr = mmap(NULL, stat.st_size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
close(fd);
for (int i = 0; i < stat.st_size; i++)
{
*ptr++ = i % 256;
printf("write %d\n",i);
}
return 0;
}
这里与mqueue不同的是，read端读取的时候并不会阻塞等待，如果没有提前write内容到共享内存里，read读取到的内容为空。
3、共享内存的起始地址
同一共享内存区对象映射到不同进程的地址空间，起始地址可以不一样。
#include <unistd.h>
#include <sys/types.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <stdio.h>
#include <sys/wait.h>
#include <errno.h>
#include <string.h>
int main(int argc, char* argv[])
{
int fd1,fd2;
unsigned char *ptr1, *ptr2;
pid_t pid;
int mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH;
fd1 = shm_open("/shm_test",O_RDWR,mode);
if (fd1 == -1)
{
printf("Message : %s\n", strerror(errno));
return 1;
}
struct stat stat;
fstat(fd1, &stat);
fd2 = open("/dev/shm/shm_test1",O_RDWR | O_CREAT,mode);
if (fd2 == -1)
{
printf("Message : %s\n", strerror(errno));
return 1;
}
pid = fork();
if (pid == 0)
{
ptr2 = mmap(NULL, 4, PROT_READ|PROT_WRITE, MAP_SHARED, fd2,0);
ptr1 = mmap(NULL, 1024, PROT_READ|PROT_WRITE, MAP_SHARED, fd1, 0);
printf("child: shm ptr=%p, mmf ptr=%p\n",ptr1, ptr2);
sleep(5);
printf("share memory integer:%d", *ptr1);
}
else if (pid > 0)
{
ptr1 = mmap(NULL, 1024, PROT_READ|PROT_WRITE, MAP_SHARED, fd1, 0);
ptr2 = mmap(NULL, 4, PROT_READ|PROT_WRITE, MAP_SHARED, fd2,0);
printf("parent: shm ptr=%p, mmf ptr=%p\n",ptr1, ptr2);
*ptr1 = 128;
waitpid(0, NULL, 0);
}
else{
}
return 0;
}
代码运行结果如下：
可见，在不同的进程中，相同的共享内存区对象起始地址可以不一样。