【Linux】共享内存

共享内存允许两个或更多进程访问同一块内存，当一个进程改变了这块地址中的内容的时候，其它进程都会察觉到这个更改。因为所有进程共享同一块内存，共享内存在各种进程间通信方式中具有最高的效率。访问共享内存区域和访问进程独有的内存区域一样快，并不需要通过系统调用或者其它需要切入内核的过程来完成。同时它也避免了对数据的各种不必要的复制。

因为系统内核没有对访问共享内存进行同步，必须提供自己的同步措施。例如，在数据被写入之前不允许进程从共享内存中读取信息、不允许两个进程同时向同一个共享内存地址写入数据等。解决这些问题的常用方法是通过使用信号量进行同步。

要使用一块共享内存，进程必须首先分配它。随后需要访问这个共享内存块的每一个进程都必须将这个共享内存绑定到自己的地址空间中。当完成通信之后，所有进程都将脱离共享内存，并且由一个进程释放该共享内存块。

创建共享内存：

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

key：用来标识共享内存块的键值，通常在代码中可以利用宏定义来实现。彼此无关的进程可以通过指定同一个键以获取对同一个共享内存块的访问。但是，其它程序也可能挑选了同样的特定值作为自己分配共享内存的键值，从而产生冲突。用特殊常量 IPC_PRIVATE 作为键值可以保证系统建立一个全新的共享内存块。 
size：代表创建的共享内存的大小。系统分配的共享内存的大小是以页面为最小单位的，在 Linux 系统中，内存页面大小是4KB，在申请共享内存时，大小是页面大小的整数倍，不足一页也会分配给一页的大小（实际分配的内存块大小将被扩大到页面大小的整数倍）。
shmflg：通常是使用一组常量的按位或操作来表示。IPC_CREAT：这个标志表示应创建一个新的共享内存块。通过指定这个标志，我们可以创建一个具有指定键值的新共享内存块。IPC_EXCL：这个标志只能与 IPC_CREAT 同时使用。当指定这个标志的时候，如果已有一个具有这个键值的共享内存块存在，则shmget会调用失败。也就是说，这个标志将使线程获得一个“独有”的共享内存块。如果没有指定这个标志而系统中存在一个具有相同键值的共享内存块，shmget会返回这个已经建立的共享内存块，而不是重新创建一个。
返回值：成功返回该共享内存的标识码，一个非负整数，失败返回-1。

将共享内存映射到进程地址空间：

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

shmid：共享内存标识。
shmaddr：是一个指针，指向您希望用于映射该共享内存块的进程内存地址。如果为NULL则内核会选择一个合适的地址用于映射。
shmflg：标志位
1. SHM_RND表示第二个参数指定的地址应被向下靠拢到内存页面大小的整数倍。
2. SHM_RDONLY表示这个内存块将仅允许读取操作而禁止写入。
返回值：成功则会返回绑定的共享内存块对应的地址，失败返回-1。
当进程调用 fork() 后，映射的地址会被继承到子进程。当调用 exec() 后，映射的地址连接会自动脱离。程序运行结束之后，映射的地址连接也会自动脱离。
对于一个共享内存，实现采用的是引用计数的原理，当进程脱离共享内存后，计数器减一，映射成功时，计数器加一，当计数器变为零时，共享内存将被删除。

将进程地址空间与共享内存脱离：

int shmdt(const void *shmaddr);

shmaddr：由shmat所返回的指针。
返回值：成功返回0，失败返回-1。 

控制共享内存：

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

shmid：由shmget返回的共享内存标识码。 
cmd：
1. 要获取一个共享内存块的相关信息，则为该函数传递 IPC_STAT 作为第二个参数，同时传递一个指向一个 struct shmid_ds 对象的指针作为第三个参数。 即把共享内存的 shmid_ds 结构数据复制到 buf。
2. 要删除一个共享内存块，则应将 IPC_RMID 作为第二个参数，而将 NULL 作为第三个参数。当最后一个绑定该共享内存块的进程与其脱离时，该共享内存块将被删除。
3. IPC_SET 将 buf 所指的结构复制到共享内存结构中。

struct shmid_ds 结构：

struct shmid_ds {
 struct ipc_perm shm_perm; /* operation perms */
 int shm_segsz; /* size of segment (bytes) */
 __kernel_time_t shm_atime; /* last attach time */
 __kernel_time_t shm_dtime; /* last detach time */
 __kernel_time_t shm_ctime; /* last change time */
 __kernel_ipc_pid_t shm_cpid; /* pid of creator */
 __kernel_ipc_pid_t shm_lpid; /* pid of last operator */
 unsigned short shm_nattch; /* no. of current attaches */
 unsigned short shm_unused; /* compatibility */
 void *shm_unused2; /* ditto - used by DIPC */
 void *shm_unused3; /* unused */
};

通过两个简单程序演示共享内存：

test1.c

#include<stdio.h>
#include<unistd.h>
#include<sys/ipc.h>
#include<sys/shm.h>
#include<errno.h>
#include<stdlib.h>
#include<string.h>
#define IPC_KEY 0x12345678
int main()
{
    int shm_id;
    shm_id=shmget(IPC_KEY,4096,IPC_CREAT|0644);
    if(shm_id<0)
    {
        perror("shmget error");
        return -1;
    }
    void* memaddr=shmat(shm_id,NULL,0);
    if(*(int*)memaddr==-1)
    {
        perror("shmat error");
        return -1;
    }
    while(1)
    {
        char input;
        printf("please input y or n to continue\n");
        scanf("%c",&input);
        getchar();
        if(input=='n')
        {
            printf("quit process and shmdt\n");
            break;
        }
        printf("please input some words to write in shared memory\n");
        read(0,memaddr,4096);
        sleep(1);
    }
    shmdt(memaddr);
    shmctl(shm_id,IPC_RMID,NULL);
    return 0;
}

test2.c

#include<stdio.h>
#include<unistd.h>
#include<sys/ipc.h>
#include<sys/shm.h>
#include<errno.h>

#define IPC_KEY 0x12345678
int main()
{
    int shm_id;
    shm_id=shmget(IPC_KEY,4096,IPC_CREAT|0644);
    if(shm_id<0)
    {
        perror("shmget error");
        return -1;
    }
    void *memaddr=shmat(shm_id,NULL,0);
    if(*(int*)memaddr==-1)
    {
        perror("shmat error");
        return -1;
    }
    while(1)
    {
        char input;
        printf("please input y or n to continue\n");
        scanf("%c",&input);
        getchar();
        if(input=='n')
        {
            printf("quit process and shmdt\n");
            break;
        }	
        write(1,memaddr,4096);
    }
    shmdt(memaddr);    
    shmctl(shm_id,IPC_RMID,NULL);
    return 0;
}

程序运行结果：

我们可以看到，当进程 test1 将字符串“sharedmemory!\n”写入到共享内存之后，进程 test2 同样输出了字符串“sharedmemory!\n”。
然后进程test1再将字符串“hello\n”写入共享内存，我们发现进程test2输出的是“hello\nmemory\n”。可以看到已经完全符合我们预期的要求了。

通过命令查看和删除共享内存：

我们再将上面两个进程执行一遍。然后使用 ipcs -m 命令查看共享内存资源。

我们可以看到之前利用程序创建的共享内存，shmid为 753679，key值为 0x12345678，大小为 4096 字节，nattch 选项为 2，意思是有两个进程的地址空间映射到共享区内存。
我们也可以使用命令 ipcrm -m shmid 来删除共享内存。