首页 > 代码库 > linux进程间通信-共享内存

linux进程间通信-共享内存

一 共享内存介绍

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

但有一点特别要注意:共享内存并未提供同步机制。也就是说,在第一个进程结束对共享内存的写操作之前,并无自动机制可以阻止第二个进程开始对它进行读取。所以我们通常需要用其他的机制来同步对共享内存的访问,例如信号量。

二 共享内存的使用

 创建共享内存

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

第一个参数,共享内存段命名,shmget函数成功时返回一个与key相关的共享内存标识符(非负整数),用于后续的共享内存函数。调用失败返回-1.

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

第二个参数,size以字节为单位指定需要共享的内存容量

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

 ◆启动对该共享内存的访问

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

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

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

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

第三个参数,shm_flg是一组标志位,通常为0

调用成功时返回一个指向共享内存第一个字节的指针,如果调用失败返回-1.

 ◆将共享内存从当前进程中分离

int shmdt(const void *shmaddr);

该函数用于将共享内存从当前进程中分离。注意,将共享内存分离并不是删除它,只是使该共享内存对当前进程不再可用

参数shmaddrshmat函数返回的地址指针,调用成功时返回0,失败时返回-1

 ◆控制共享内存

int shmctl(int shm_id, int command, struct shmid_ds *buf);

第一个参数,shm_idshmget函数返回的共享内存标识符。

第二个参数,command是要采取的操作,它可以取下面的三个值 :

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

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

    IPC_RMID:删除共享内存段

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

 

  1. struct shmid_ds
  2. {
  3. uid_t shm_perm.uid;
  4. uid_t shm_perm.gid;
  5. mode_t shm_perm.mode;

 

三 例子

shmdata.h的源码:

 

  1. #ifndef _SHMDATA_H_HEADER
  2. #define _SHMDATA_H_HEADER
  3. #define TEXT_SZ 2048
  4. struct shared_use_st
  5. {
  6. int written;/* 作为一个标志,非0:表示可读,0表示可写 */
  7. char text[TEXT_SZ];/* 记录写入和读取的文本 */
  8. };
  9. #endif

 

shmread.c的源代码

 

  1. #include<unistd.h>
  2. #include<stdlib.h>
  3. #include<stdio.h>
  4. #include<sys/shm.h>
  5. #include"shmdata.h"
  1.  
  2. #define MEM_KEY (1234)
  3.  
  4. int main()
  5. {
  6. int running =1; //程序是否继续运行的标志
  7. void*shm = NULL; //分配的共享内存的原始首地址
  8. struct shared_use_st *shared;//指向shm
  9. int shmid; //共享内存标识符
  10. //创建共享内存
  11.  
  12. shmid = shmget((key_t)MEM_KEY,sizeof(struct shared_use_st),0666|IPC_CREAT);
  13. if(shmid ==-1)
  14. {
  15. fprintf(stderr,"shmget failed\n");
  16. exit(EXIT_FAILURE);
  17. }
  18. //将共享内存连接到当前进程的地址空间
  19. shm = shmat(shmid,0,0);
  20. if(shm ==(void*)-1)
  21. {
  22. fprintf(stderr,"shmat failed\n");
  23. exit(EXIT_FAILURE);
  24. }
  25. printf("\nMemory attached at %X\n",(int)shm);
  26. //设置共享内存
  27. shared =(struct shared_use_st*)shm;
  28. shared->written =0;
  29. while(running)//读取共享内存中的数据
  30. {
  31. //没有进程向共享内存定数据有数据可读取
  32. if(shared->written !=0)
  33. {
  34. printf("You wrote: %s", shared->text);
  35. sleep(rand()%3);
  36. //读取完数据,设置written使共享内存段可写
  37. shared->written =0;
  38. //输入了end,退出循环(程序)
  39. if(strncmp(shared->text,"end",3)==0)
  40. running =0;
  41. }
  42. else//有其他进程在写数据,不能读取数据
  43. sleep(1);
  44. }
  45. //把共享内存从当前进程中分离
  46. if(shmdt(shm)==-1)
  47. {
  48. fprintf(stderr,"shmdt failed\n");
  49. exit(EXIT_FAILURE);
  50. }
  51. //删除共享内存
  52. if(shmctl(shmid, IPC_RMID,0)==-1)
  53. {
  54. fprintf(stderr,"shmctl(IPC_RMID) failed\n");
  55. exit(EXIT_FAILURE);
  56. }
  57. exit(EXIT_SUCCESS);
  58. }

 

shmwrite.c的源代码

 

  1. #include<unistd.h>
  2. #include<stdlib.h>
  3. #include<stdio.h>
  4. #include<string.h>
  5. #include<sys/shm.h>
  6. #include"shmdata.h"
  7. #define MEM_KEY (1234)
  8.  
  9. int main()
  10. {
  11. int running =1;
  12. void*shm = NULL;
  13. struct shared_use_st *shared = NULL;
  14. char buffer[BUFSIZ +1];//用于保存输入的文本
  15. int shmid;
  16. //创建共享内存
  17. shmid = shmget((key_t)MEM_KEY,sizeof(struct shared_use_st),0666|IPC_CREAT);
  18. if(shmid ==-1)
  19. {
  20. fprintf(stderr,"shmget failed\n");
  21. exit(EXIT_FAILURE);
  22. }
  23. //将共享内存连接到当前进程的地址空间
  24. shm = shmat(shmid,(void*)0,0);
  25. if(shm ==(void*)-1)
  26. {
  27. fprintf(stderr,"shmat failed\n");
  28. exit(EXIT_FAILURE);
  29. }
  30. printf("Memory attached at %X\n",(int)shm);
  31. //设置共享内存
  32. shared =(struct shared_use_st*)shm;
  33. while(running)//向共享内存中写数据
  34. {
  35. //数据还没有被读取,则等待数据被读取,不能向共享内存中写入文本
  36. while(shared->written ==1)
  37. {
  38. sleep(1);
  39. printf("Waiting...\n");
  40. }
  41. //向共享内存中写入数据
  42. printf("Enter some text: ");
  43. fgets(buffer, BUFSIZ, stdin);
  44. strncpy(shared->text, buffer, TEXT_SZ);
  45. //写完数据,设置written使共享内存段可读
  46. shared->written =1;
  47. //输入了end,退出循环(程序)
  48. if(strncmp(buffer,"end",3)==0)
  49. running =0;
  50. }
  51. //把共享内存从当前进程中分离
  52. if(shmdt(shm)==-1)
  53. {
  54. fprintf(stderr,"shmdt failed\n");
  55. exit(EXIT_FAILURE);
  56. }
  57. sleep(2);
  58. exit(EXIT_SUCCESS);
  59. }

 

代码分析:

程序shmread创建共享内存,然后将它连接到自己的地址空间。在共享内存的开始处使用了一个结构struct_use_st。该结构中有个标志written,当共享内存中有其他进程向它写入数据时,共享内存中的written被设置为0,程序等待。当它不为0时,表示没有进程对共享内存写入数据,程序就从共享内存中读取数据并输出,然后重置设置共享内存中的written0,即让其可被shmwrite进程写入数据。

程序shmwrite取得共享内存并连接到自己的地址空间中。检查共享内存中的written,是否为0,若不是,表示共享内存中的数据还没有被完,则等待其他进程读取完成,并提示用户等待。若共享内存的written0,表示没有其他进程对共享内存进行读取,则提示用户输入文本,并再次设置共享内存中的written1,表示写完成,其他进程可对共享内存进行读操作。

 关于前面的例子的安全性讨论

这个程序是不安全的,当有多个程序同时向共享内存中读写数据时,问题就会出现。可能你会认为,可以改变一下written的使用方式,例如,只有当written0时进程才可以向共享内存写入数据,而当一个进程只有在written不为0时才能对其进行读取,同时把written进行加1操作,读取完后进行减1操作。这就有点像文件锁中的读写锁的功能。咋看之下,它似乎能行得通。但是这都不是原子操作,所以这种做法是行不能的。试想当written0时,如果有两个进程同时访问共享内存,它们就会发现written0,于是两个进程都对其进行写操作,显然不行。当written1时,有两个进程同时对共享内存进行读操作时也是如些,当这两个进程都读取完是,written就变成了-1.

要想让程序安全地执行,就要有一种进程同步的进制,保证在进入临界区的操作是原子操作。例如,可以使用前面所讲的信号量来进行进程的同步。因为信号量的操作都是原子性的。

使用共享内存的优缺点

优点:我们可以看到使用共享内存进行进程间的通信真的是非常方便,而且函数的接口也简单,数据的共享还使进程间的数据不用传送,而是直接访问内存,也加快了程序的效率。同时,它也不像匿名管道那样要求通信的进程有一定的父子关系。

缺点:共享内存没有提供同步的机制,这使得我们在使用共享内存进行进程间通信时,往往要借助其他的手段来进行进程间的同步工作。

四 更好的例子

1、server.c

 

  1. /*server.c:向共享内存中写入People*/
  2. #include<stdio.h>
  3. #include<sys/types.h>
  4. #include<sys/ipc.h>
  5. #include<sys/sem.h>
  6. #include<string.h>
  7. #include"credis.h"
  8. int semid;
  9. int shmid;
  10. /*信号量的P操作*/
  11. void p()
  12. {
  13. struct sembuf sem_p;
  14. sem_p.sem_num=0;/*设置哪个信号量*/
  15. sem_p.sem_op=-1;/*定义操作*/
  16. if(semop(semid,&sem_p,1)==-1)
  17. printf("p operation is fail\n");
  18. /*semop函数自动执行信号量集合上的操作数组。
  19.    int semop(int semid, struct sembuf semoparray[], size_t nops);
  20.    semoparray是一个指针,它指向一个信号量操作数组。nops规定该数组中操作的数量。*/
  21. }
  22. /*信号量的V操作*/
  23. void v()
  24. {
  25. struct sembuf sem_v;
  26. sem_v.sem_num=0;
  27. sem_v.sem_op=1;
  28. if(semop(semid,&sem_v,1)==-1)
  29. printf("v operation is fail\n");
  30. }
  31. int main()
  32. {
  33. structPeople{
  34. char name[10];
  35. int age;
  36. };
  37. key_t semkey;
  38. key_t shmkey;
  39. semkey=ftok("../test/VenusDB.cbp",0);//用来产生唯一的标志符,便于区分信号量及共享内存
  40. shmkey=ftok("../test/main.c",0);
  41. /*创建信号量的XSI IPC*/
  42. semid=semget(semkey,1,0666|IPC_CREAT);//参数nsems,此时为中间值1,指定信号灯集包含信号灯的数目
  43. //0666|IPC_CREAT用来表示对信号灯的读写权限
  44. /*
  45. 从左向右:
  46. 第一位:0表示这是一个8进制数
  47. 第二位:当前用户的经权限:6=110(二进制),每一位分别对就 可读,可写,可执行,6说明当前用户可读可写不可执行
  48. 第三位:group组用户,6的意义同上
  49. 第四位:其它用户,每一位的意义同上,0表示不可读不可写也不可执行
  50. */
  51. if(semid==-1)
  52. printf("creat sem is fail\n");
  53. //创建共享内存
  54. shmid=shmget(shmkey,1024,0666|IPC_CREAT);//对共享内存
  55. if(shmid==-1)
  56. printf("creat shm is fail\n");
  57. /*设置信号量的初始值,就是资源个数*/
  58. union semun{
  59. int val;
  60. struct semid_ds *buf;
  61. unsignedshort*array;
  62. }sem_u;
  63. sem_u.val=1;/*设置变量值*/
  64. semctl(semid,0,SETVAL,sem_u);//初始化信号量,设置第0个信号量,p()操作为非阻塞的
  65. /*将共享内存映射到当前进程的地址中,之后直接对进程中的地址addr操作就是对共享内存操作*/
  66. structPeople*addr;
  67. addr=(structPeople*)shmat(shmid,0,0);//将共享内存映射到调用此函数的内存段
  68. if(addr==(structPeople*)-1)
  69. printf("shm shmat is fail\n");
  70. /*向共享内存写入数据*/
  71. p();
  72. strcpy((*addr).name,"xiaoming");
  73. /*注意:①此处只能给指针指向的地址直接赋值,不能在定义一个 struct People people_1;addr=&people_1;因为addr在addr=(struct People*)shmat(shmid,0,0);时,已经由系统自动分配了一个地址,这个地址与共享内存相关联,所以不能改变这个指针的指向,否则他将不指向共享内存,无法完成通信了。
  74. 注意:②给字符数组赋值的方法。刚才太虎了。。*/
  75. (*addr).age=10;
  76. v();
  77. /*将共享内存与当前进程断开*/
  78. if(shmdt(addr)==-1)
  79. printf("shmdt is fail\n");
  80. }
2、clinet.c
  1. /*client.c:从共享内存中读出People*/
  2. #include<stdio.h>
  3. #include<sys/types.h>
  4. #include<sys/ipc.h>
  5. #include<sys/sem.h>
  6. int semid;
  7. int shmid;
  8. /*信号量的P操作*/
  9. void p()
  10. {
  11. struct sembuf sem_p;
  12. sem_p.sem_num=0;
  13. sem_p.sem_op=-1;
  14. if(semop(semid,&sem_p,1)==-1)
  15. printf("p operation is fail\n");
  16. }
  17. /*信号量的V操作*/
  18. void v()
  19. {
  20. struct sembuf sem_v;
  21. sem_v.sem_num=0;
  22. sem_v.sem_op=1;
  23. if(semop(semid,&sem_v,1)==-1)
  24. printf("v operation is fail\n");
  25. }
  26. int main()
  27. {
  28. key_t semkey;
  29. key_t shmkey;
  30. semkey=ftok("../test/client/VenusDB.cbp",0);
  31. shmkey=ftok("../test/client/main.c",0);
  32. structPeople{
  33. char name[10];
  34. int age;
  35. };
  36. /*读取共享内存和信号量的IPC*/
  37. semid=semget(semkey,0,0666);
  38. if(semid==-1)
  39. printf("creat sem is fail\n");
  40. shmid=shmget(shmkey,0,0666);
  41. if(shmid==-1)
  42. printf("creat shm is fail\n");
  43. /*将共享内存映射到当前进程的地址中,之后直接对进程中的地址addr操作就是对共享内存操作*/
  44. structPeople*addr;
  45. addr=(structPeople*)shmat(shmid,0,0);
  46. if(addr==(structPeople*)-1)
  47. printf("shm shmat is fail\n");
  48. /*从共享内存读出数据*/
  49. p();
  50. printf("name:%s\n",addr->name);
  51. printf("age:%d\n",addr->age);
  52. v();
  53. /*将共享内存与当前进程断开*/
  54. if(shmdt(addr)==-1)
  55. printf("shmdt is fail\n");
  56. /*IPC必须显示删除。否则会一直留存在系统中*/
  57. if(semctl(semid,0,IPC_RMID,0)==-1)
  58. printf("semctl delete error\n");
  59. if(shmctl(shmid,IPC_RMID,NULL)==-1)
  60. printf("shmctl delete error\n");
  61. }

五 父子进程共享内存的个子

  1. #include<stdio.h>
  2. #include<sys/types.h>
  3. #include<sys/ipc.h>
  4. #include<sys/sem.h>
  5. #define SHM_KEY 0x33
  6. #define SEM_KEY 0x44
  7. union semun {
  8. int val;
  9. struct semid_ds *buf;
  10. unsignedshort*array;
  11. };
  12. int P(int semid)
  13. {
  14. struct sembuf sb;
  15. sb.sem_num =0;
  16. sb.sem_op =-1;
  17. sb.sem_flg = SEM_UNDO;
  18. if(semop(semid,&sb,1)==-1){
  19. perror("semop");
  20. return-1;
  21. }
  22. return0;
  23. }
  24. int V(int semid)
  25. {
  26. struct sembuf sb;
  27. sb.sem_num =0;
  28. sb.sem_op =1;
  29. sb.sem_flg = SEM_UNDO;
  30. if(semop(semid,&sb,1)==-1){
  31. perror("semop");
  32. return-1;
  33. }
  34. return0;
  35. }
  36. int main(int argc,char**argv)
  37. {
  38. pid_t pid;
  39. int i, shmid, semid;
  40. int*ptr;
  41. union semun semopts;
  42. /* 创建一块共享内存, 存一个int变量 */
  43. if((shmid = shmget(SHM_KEY,sizeof(int), IPC_CREAT |0600))==-1){
  44. perror("msgget");
  45. }
  46. /* 将共享内存映射到进程, fork后子进程可以继承映射 */
  47. if((ptr =(int*)shmat(shmid, NULL,0))==(void*)-1){
  48. perror("shmat");
  49. }
  50. *ptr =0;
  51. /* 创建一个信号量用来同步共享内存的操作 */
  52. if((semid = semget(SEM_KEY,1, IPC_CREAT |0600))==-1){
  53. perror("semget");
  54. }
  55. /* 初始化信号量 */
  56. semopts.val =1;
  57. if(semctl(semid,0, SETVAL, semopts)<0){
  58. perror("semctl");
  59. }
  60. if((pid = fork())<0){
  61. perror("fork");
  62. }elseif(pid ==0){/* Child */
  63. /* 子进程对共享内存加1 */
  64. for(i =0; i <100; i++){
  65. P(semid);
  66. (*ptr)++;
  67. V(semid);
  68. printf("child: %d\n",*ptr);
  69. }
  70. }else{/* Parent */
  71. /* 父进程对共享内存减1 */
  72. for(i =0; i <100; i++){
  73. P(semid);
  74. (*ptr)--;
  75. V(semid);
  76. printf("parent: %d\n",*ptr);
  77. }
  78. waitpid(pid);
  79. sleep(2);
  80. /* 如果同步成功, 共享内存的值为0 */
  81. printf("finally: %d\n",*ptr);
  82. }
  83. return0;
  84. }
 
 

 

 



来自为知笔记(Wiz)



linux进程间通信-共享内存