共享内存概述
共享内存是所有IPC手段中最快的一种。它之所以快是因为共享内存一旦映射到进程的地址空间,进程之间数据的传递就不须要涉及内核了。
进程间通信主要包括管道, 系统IPC(Inter-Process Communication,进程间通信)(包括 消息队列, 信号,共享存储), 套接字(SOCKET).
回顾一下前面已经讨论过的管道、FIFO,任意两个进程之间想要交换信息,都必须通过内核,内核在其中发挥了中转站的作用:
·发送信息的一方,通过系统调用(write或msgsnd)将信息从用户层拷贝到内核层,由内核暂存这部分信息。
·提取信息的一方,通过系统调用(read或msgrcv)将信息从内核层提取到应用层。
共享内存的思想
一个通信周期内,上述过程至少牵扯到两次内存拷贝(从用户拷贝到内核空间和从内核空间拷贝到用户空间)和两次系统调用,这其中的开销不容小觑。用户层的体验固然不佳,内核层想必也是不堪其扰,双方的内心都是崩溃的。
建立共享内存之后,内核完全不参与进程间的通信,这种说法严格来讲并不是正确的。因为当进程使用共享内存时,可能会发生缺页,引发缺页中断,这种情况下,内核还是会参与进来的。
进程从此就像操作普通进程的地址空间一样操作这块共享内存,一个进程可以将信息写入这片内存区域,而另一个进程也可以看到共享内存里面的信息,从而达到通信的目的。
允许多个进程同时操作共享内存,就不得不防范竞争条件的出现,比如有两个进程同时执行更新操作,或者一个进程在执行读取操作时,另外一个进程正在执行更新操作。因此,共享内存这种进程间通信的手段通常不会单独出现,总是和信号量、文件锁等同步的手段配合使用。
shmget函数
与信号量一样,程序需要提供一个参数key,它有效地为共享内存段命名,shmget函数返回一个共享内存标识符,该标识符将用于后续的共享内存函数。有一个特殊的键值IPC_PRIVATE,它用于创建一个只属于创建进程的共享内存。通常你不会用到这个值,而且你可能会发现在一些Linux系统中,私有的共享内存其实并不是真正的私有。
其中第二个参数size必须是正整数,表示要创建的共享内存的大小。内核以页面大小的整数倍来分配共享内存,因此,实际size会被向上取整为页面大小的整数倍。
第三个参数shmflg包含9个比特的权限标志,它们的作用与创建文件时使用的mode标志一样。由IPC_CREAT定义的一个特殊比特必须和权限标志按位或才能创建一个新的共享内存段。设置IPC_CREAT标志的同时,给shmget函数传递一个已有共享内存段的键并不是一个错误,如果无需用到IPC_CREAT标志,该标志就会被悄悄地忽略掉。
权限标志对共享内存非常有用,因为它们允许一个进程创建的共享内存可以被共享内存的创建者所拥有的进程写入,同时其他用户创建的进程只能读取该共享内存。我们可以利用这个功能来提供一种有效的对数据进行只读访问的方法,通过将数据放入共享内存并设置它的权限,就可以避免数据被其他用户修改。
如果共享内存创建成功,shmget返回一个非负整数,即共享内存标识符;如果失败,就返回-1。
shmat函数
第一次创建共享内存段时,它不能被任何进程访问。要想启用对该共享内存的访问,必须将其连接到一个进程的地址空间中。这项工作由shmat函数来完成,它的定义如下所示:
第一个参数shm_id是由shmget返回的共享内存标识符。
第二个参数shm_addr指定的是共享内存连接到当前进程中的地址位置。它通常是一个空指针,表示让系统来选择共享内存出现的地址。大部分的普通青年都会将第二个参数设置为NULL,表示用户并不在意,一切交由内核做主。
第三个参数shmflg是一组位标志。它的两个可能取值是SHM_RND(这个标志与shm_addr联合使用,用来控制共享内存连接的地址)和SHM_RDONLY(它使得连接的内存只读)。我们很少需要控制共享内存连接的地址,通常都是让系统来选择一个地址,否则就会使应用程序对硬件的依赖性过高。
如果shmat调用成功,它返回一个指向共享内存第一个字节的指针;如果失败,它就返回-1。
共享内存的读写权限由它的属主(共享内存的创建者)、它的访问权限和当前进程的属主决定。共享内存的访问权限类似于文件的访问权限。这个规则的一个例外是,当shmflg & SHM_RDONLY为true时的情况。此时即使该共享内存的访问权限允许写操作,它都不能被写入。
shmdt函数
shmdt函数的作用是将共享内存从当前进程中分离。它的参数是shmat返回的地址指针。成功时它返回0,失败时返回-1。注意,将共享内存分离并未删除它,只是使得该共享内存对当前进程不再可用。
shmctl函数
共享内存的控制函数
第一个参数shm_id是shmget返回的共享内存标识符。
第二个参数command是要采取的动作,它可以取3个值
第三个参数buf是一个指针,它指向包含共享内存模式和访问权限的结构。
成功时返回0,失败时返回-1。X/Open规范没有定义当你试图删除一个正处于连接状态的共享内存段时将会发生的情况。通常这个已经被删除的处于连接状态的共享内存段还能继续使用,直到它从最后一个进程中分离为止。但因为这个行为并未在规范中定义,所以最好不要依赖它。
共享内存实验
在这个实验中,我们将编写一对程序shm1.c和shm2.c。第一个程序(消费者)将创建一个共享内存段,然后把写到它里面的数据都显示出来。第二个程序(生产者)将连接一个已有的共享内存段,并允许我们向其中输入数据。
(1)我们首先创建一个公共的头文件,来定义我们希望分发的共享内存。我们将其命名为shm_com.h:
#define TEXT_SZ 2048
struct shared_use_st{
int written_by_you; //值为1,则读出数据;值为0,则写数据
char some_text[TEXT_SZ]; //用于读或写的字符数据
};
这里定义的结构在消费者和生产者程序中都会用到。当有数据写入这个结构时,我们用该结构中的一个整型标志written_by_you来通知消费者。需要传输的文本长度2K是由我们随意决定的。
(2)第一个程序shm1.c是消费者程序。在头文件之后,通过设置了IPC_CREAT标志位的shmget调用来创建共享内存段(其长度就是我们的共享内存结构的长度)
#include<stdlib.h>
#include<stdio.h>
#include<string.h>
#include<sys/shm.h>
#include "shm_com.h"
int main()
{
int running = 1;//程序是否继续运行的标志
void *shared_memory = (void *)0;//分配的共享内存的原始首地址
struct shared_use_st *shared_stuff;
int shmid;
srand((unsigned int)getpid());
shmid = shmget((key_t)1234, sizeof(struct shared_use_st), 0666|IPC_CREAT);
if(shmid == -1)
{
fprintf(stderr, "shmget failed\n");
exit(EXIT_FAILURE);
}
}
(3)现在,让程序可以访问这个共享内存
//将共享内存连接到当前进程的地址空间
shared_memory = shmat(shmid, (void *)0, 0);
if(shared_memory == (void *)-1)
{
fprintf(stderr, "shmat failed\n");
exit(EXIT_FAILURE);
}
printf("Memory attached at %X\n", (int)shared_memory);
(4)程序的下一部分将shared_memory分配给shared_stuff,然后它输出written_by_you中的文本。循环将一直执行到在written_by_you中找到end字符串为止。sleep调用强迫消费者程序在临界区域多待一会儿,让生产者程序等待:
shared_stuff = (struct shared_use_st *)shared_memory;
shared_stuff->written_by_you = 0;
while(running)
{
if(shared_stuff->written_by_you)
{
printf("You wrote:%s", shared_stuff->some_text);
sleep(rand() % 4);
shared_stuff->written_by_you = 0;
if(strncmp(shared_stuff->some_text, "end", 3) == 0)
{
running = 0;
}
}
}
(5)最后,共享内存被分离,然后被删除:
if(shmdt(shared_memory) == -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);
}
exit(EXIT_SUCCESS);
}
总的shm1.c
#include<unistd.h>
#include<stdlib.h>
#include<stdio.h>
#include<string.h>
#include<sys/shm.h>
#include "shm_com.h"
int main()
{
int running = 1;//程序是否继续运行的标志
void *shared_memory = (void *)0;//分配的共享内存的原始首地址
struct shared_use_st *shared_stuff;
int shmid;//共享内存标识符
srand((unsigned int)getpid());//获取种子
//即每次运行程序rand()的数都不一样
//创建共享内存
shmid = shmget((key_t)1234, sizeof(struct shared_use_st), 0666|IPC_CREAT);
if(shmid == -1)
{
fprintf(stderr, "shmget failed\n");
exit(EXIT_FAILURE);
}
//将共享内存连接到当前进程的地址空间
shared_memory = shmat(shmid, (void *)0, 0);
if(shared_memory == (void *)-1)
{
fprintf(stderr, "shmat failed\n");
exit(EXIT_FAILURE);
}
printf("Memory attached at %X\n", (int)shared_memory);
//设置共享内存
shared_stuff = (struct shared_use_st *)shared_memory;
shared_stuff->written_by_you = 0;
while(running)
{ //读取共享内存中的数据
if(shared_stuff->written_by_you)
{
printf("You wrote:%s", shared_stuff->some_text);
sleep(rand() % 4);
//读取完数据,设置written使共享内存段可写
shared_stuff->written_by_you = 0;
//输入了end,退出循环(程序)
if(strncmp(shared_stuff->some_text, "end", 3) == 0)
{
running = 0;
}
}
}
//把共享内存从当前进程中分离
if(shmdt(shared_memory) == -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);
}
exit(EXIT_SUCCESS);
}
(6)第二个程序shm2.c是生产者程序,我们通过它向消费者程序输入数据。它与shm1.c很相似,程序代码如下所示:
#include<unistd.h>
#include<stdlib.h>
#include<stdio.h>
#include<string.h>
#include<sys/shm.h>
#include"shm_com.h"
int main()
{
int running = 1;
void *shared_memory = (void*)0;
struct shared_use_st * shared_stuff;
//用于保存输入的文本
char buffer[BUFSIZ];
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);
}
//将共享内存连接到当前进程的地址空间
shared_memory = shmat(shmid, (void *)0, 0);
if(shared_memory == (void*)-1)
{
fprintf(stderr, "shmat failed\n");
exit(EXIT_FAILURE);
}
printf("Memory attached at %X\n", (int)shared_memory);
//设置共享内存
shared_stuff = (struct shared_use_st *)shared_memory;
while(running)
{
//向共享内存中写数据
//数据还没有被读取,则等待数据被读取,不能向共享内存中写入文本
while(shared_stuff->written_by_you == 1)
{
sleep(1);
printf("waiting for client...\n");
}
printf("Enter some text:");
//向共享内存中写入数据
fgets(buffer, BUFSIZ, stdin);
strncpy(shared_stuff->some_text, buffer, TEXT_SZ);
//写完数据,设置written使共享内存段可读
shared_stuff->written_by_you = 1;
//输入了end,退出循环(程序)
if(strncmp(buffer, "end", 3) == 0)
{
running = 0;
}
}
//把共享内存从当前进程中分离
if(shmdt(shared_memory) == -1)
{
fprintf(stderr, "shmdt failed\n");
exit(EXIT_FAILURE);
}
exit(EXIT_SUCCESS);
}
开两个终端分别运行shm1和shm2
实验分析
第一个程序shm1创建共享内存段,然后将它连接到自己的地址空间中。我们在共享内存的开始处使用了一个结构shared_use_st。该结构中有个标志written_by_you,当共享内存中有数据写入时,就设置这个标志。这个标志被设置时,程序就从共享内存中读取文本,将它打印出来,然后清除这个标志表示已经读完数据。我们用一个特殊字符串end来退出循环。接下来,程序分离共享内存段并删除它。
第二个程序shm2使用相同的键1234来取得并连接同一个共享内存段。然后它提示用户输入一些文本。如果标志written_by_you被设置,shm2就知道客户进程还未读完上一次的数据,因此就继续等待。当其他进程清除了这个标志后,shm2写入新数据并设置该标志。它还使用字符串end来终止并分离共享内存段。