feisky

云计算、虚拟化与Linux技术笔记
  博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理

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

Posted on 2010-03-24 10:10  feisky  阅读(6616)  评论(1编辑  收藏  举报

共享内存是运行在同一台机器上的进程间通信最快的方式,因为数据不需要在不同的进程间复制。通常由一个进程创建一块共享内存区,其余进程对这块内存区进行读写。共享内存往往与其它通信机制,如信号量结合使用,来达到进程间的同步及互斥。
首先要用的函数是shmget,它获得一个共享存储标识符。

#i nclude <sys/types.h>
        #i nclude <sys/ipc.h>
        #i nclude <sys/shm.h>
        int shmget(key_t key, int size, int flag);

这个函数有点类似大家熟悉的malloc函数,系统按照请求分配size大小的内存用作共享内存。Linux系统内核中每个IPC结构都有的一个非负整数的标识符,这样对一个消息队列发送消息时只要引用标识符就可以了。这个标识符是内核由IPC结构的关键字得到的,这个关键字,就是上面第一个函数的key。数据类型key_t是在头文件sys/types.h中定义的,它是一个长整形的数据。在我们后面的章节中,还会碰到这个关键字。

当共享内存创建后,其余进程可以调用shmat()将其连接到自身的地址空间中。

void *shmat(int shmid, void *addr, int flag);

shmid为shmget函数返回的共享存储标识符,addr和flag参数决定了以什么方式来确定连接的地址,函数的返回值即是该进程数据段所连接的实际地址,进程可以对此进程进行读写操作。

使用共享存储来实现进程间通信的注意点是对数据存取的同步,必须确保当一个进程去读取数据时,它所想要的数据已经写好了。通常,信号量被要来实现对共享存储数据存取的同步,另外,可以通过使用shmctl函数设置共享存储内存的某些标志位如SHM_LOCK、SHM_UNLOCK等来实现。

共享内存机制使得两个不相关进程可以读取和修改同一段内存从而实现数据共享。主要函数定义:

#include <sys/shm.h>

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

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

int shmdt(const void *shm_addr);

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

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. int main()  
  7. {  
  8. struct People{  
  9. char name[10];  
  10. int age;  
  11.     };  
  12. int semid;  
  13. int shmid;  
  14.     key_t semkey;  
  15.     key_t shmkey;  
  16.     semkey=ftok("server.c",0);  
  17.     shmkey=ftok("client.c",0);  
  18. /*创建共享内存和信号量的IPC*/
  19.     semid=semget(semkey,1,0666|IPC_CREAT);  
  20. if(semid==-1)  
  21.     printf("creat sem is fail\n");  
  22.     shmid=shmget(shmkey,1024,0666|IPC_CREAT);  
  23. if(shmid==-1)  
  24.     printf("creat shm is fail\n");  
  25. /*设置信号量的初始值,就是资源个数*/
  26. union semun{  
  27. int val;  
  28. struct semid_ds *buf;  
  29.         ushort *array;  
  30.     }sem_u;  
  31.     sem_u.val=1;  
  32.     semctl(semid,0,SETVAL,sem_u);  
  33. /*将共享内存映射到当前进程的地址中,之后直接对进程中的地址addr操作就是对共享内存操作*/
  34. struct People * addr;  
  35.     addr=(struct People*)shmat(shmid,0,0);  
  36. if(addr==(struct People*)-1)  
  37.     printf("shm shmat is fail\n");  
  38. /*信号量的P操作*/
  39. void p()  
  40.     {  
  41. struct sembuf sem_p;  
  42.         sem_p.sem_num=0;  
  43.         sem_p.sem_op=-1;  
  44. if(semop(semid,&sem_p,1)==-1)  
  45.         printf("p operation is fail\n");          
  46.     }  
  47. /*信号量的V操作*/
  48. void v()  
  49.     {  
  50. struct sembuf sem_v;  
  51.         sem_v.sem_num=0;  
  52.         sem_v.sem_op=1;  
  53. if(semop(semid,&sem_v,1)==-1)  
  54.         printf("v operation is fail\n");  
  55.     }  
  56. /*向共享内存写入数据*/
  57.     p();  
  58.     strcpy((*addr).name,"xiaoming");  
  59. /*注意:①此处只能给指针指向的地址直接赋值,不能在定义一个  struct People people_1;addr=&people_1;因为addr在addr=(struct People*)shmat(shmid,0,0);时,已经由系统自动分配了一个地址,这个地址与共享内存相关联,所以不能改变这个指针的指向,否则他将不指向共享内存,无法完成通信了。
  60. 注意:②给字符数组赋值的方法。刚才太虎了。。*/
  61.     (*addr).age=10;  
  62.     v();  
  63. /*将共享内存与当前进程断开*/
  64. if(shmdt(addr)==-1)  
  65.     printf("shmdt is fail\n");    

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 main()  
  7. {  
  8. int semid;  
  9. int shmid;  
  10.     key_t semkey;  
  11.     key_t shmkey;  
  12.     semkey=ftok("server.c",0);  
  13.     shmkey=ftok("client.c",0);  
  14. struct People{  
  15. char name[10];  
  16. int age;  
  17.     };  
  18. /*读取共享内存和信号量的IPC*/
  19.     semid=semget(semkey,0,0666);  
  20. if(semid==-1)  
  21.     printf("creat sem is fail\n");  
  22.     shmid=shmget(shmkey,0,0666);  
  23. if(shmid==-1)  
  24.     printf("creat shm is fail\n");  
  25. /*将共享内存映射到当前进程的地址中,之后直接对进程中的地址addr操作就是对共享内存操作*/
  26. struct People * addr;  
  27.     addr=(struct People*)shmat(shmid,0,0);  
  28. if(addr==(struct People*)-1)  
  29.     printf("shm shmat is fail\n");  
  30. /*信号量的P操作*/
  31. void p()  
  32.     {  
  33. struct sembuf sem_p;  
  34.         sem_p.sem_num=0;  
  35.         sem_p.sem_op=-1;  
  36. if(semop(semid,&sem_p,1)==-1)  
  37.         printf("p operation is fail\n");          
  38.     }  
  39. /*信号量的V操作*/
  40. void v()  
  41.     {  
  42. struct sembuf sem_v;  
  43.         sem_v.sem_num=0;  
  44.         sem_v.sem_op=1;  
  45. if(semop(semid,&sem_v,1)==-1)  
  46.         printf("v operation is fail\n");  
  47.     }  
  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");  
无觅相关文章插件,快速提升流量