进程间通信-共享内存

共享内存可以是最有用的进程间通信方式，也是最快的IPC形式。两个不同进程A B共享内存的基本原理是，同一块物理内存被映射到进程 A B各自的进程地址空间。进程A可以即时看到进程B对共享内存中数据的更新，反之亦然。由于多个 进程共享同一块内存区域必然需要某种同步机制，互斥锁和信号量都可以。

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

1、shmget函数

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

#include <sys/ipc.h>
       #include <sys/shm.h>

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

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

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

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

2、shmat函数

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

        #include <sys/types.h>
       #include <sys/shm.h>

       void *shmat(int shmid, const void *shmaddr, int shmflg);

       int shmdt(const void *shmaddr);

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

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

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

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

shmdt用于将共享内存从当前进程中分离

4、shmctl函数用来控制共享内存，它的原型如下：

       #include <sys/ipc.h>
       #include <sys/shm.h>

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

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

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

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

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

    IPC_RMID：删除共享内存段

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

例如：

#include <stdarg.h>
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include <errno.h>
#include <sys/ipc.h>
#include <sys/shm.h>

#define SEGSIZE 100

void writeshm(int shmid, char *segptr, char *text)
{
    strcpy(segptr, text);
    printf("write done\n");
}

void readshm(int shmid, char *segptr)
{
    printf("segptr:%s\n", segptr);
}

void removeshm(int shmid)
{
    shmctl(shmid, IPC_RMID, 0);
    printf("shm deteled!\n");
}

void changemode(int shmid, char *mode)
{
    struct shmid_ds myshmds;
    shmctl(shmid, IPC_STAT, &myshmds);
    printf("old permissions were:[%d]\n", myshmds.shm_perm.mode);
    sscanf(mode, "%d", &myshmds.shm_perm.mode);
    shmctl(shmid, IPC_SET, &myshmds);
    printf("old permissions are:[%d]\n", myshmds.shm_perm.mode);
}

void usage()
{
    fprintf(stderr, "s hmtool -a utility for tinkering with shared memory\n");
    fprintf(stderr, "w rite text\n");
    fprintf(stderr, "r ead\n");
    fprintf(stderr, "d elete");
    fprintf(stderr, "m ode chang");
    exit(1);
}

int main(int argc, char *argv[])
{
    key_t key;
    int shmid, cntr;
    char *segptr;
    if(argc ==1)
        usage();
    key = ftok(".", 's');
    printf("key=[%d]\n");
    if((shmid=shmget(key, SEGSIZE, IPC_CREAT|IPC_EXCL|0666)) == -1)
    {
        printf("shm exist\n");
        if((shmid=shmget(key, SEGSIZE, 0)) == -1)
        {
            perror("shmget");
            exit(1);
        }
    }
    else
    {
        printf("creat shmid=[%d]\n", shmid);
    }
    
    if((segptr = (char *)shmat(shmid, NULL, 0)) == (void *)-1)
    {
        perror("shmat");
        exit(1);
    }
    
    switch(tolower(argv[1][0]))
    {
        case 'w':
            writeshm(shmid, segptr, argv[2]);
            break;
        case 'r':
            readshm(shmid, segptr);
            break;
        case 'd':
            removeshm(shmid);
            break;
        case 'm':
            changemode(shmid, argv[2]);
            break;
        default:
            usage();
    }
    return 0;
}