进程间通信（3） 共享内存

之前我们学习了进程间通信的两种方式------管道&消息队列，而今天我们再来学习一种方式-----共享内存。

这时我们就要明白一个概念，什么是共享内存呢？

       共享内存就是两个（或多个）进程占有同一段内存空间，这些进程可以是有亲缘关系的进程，也可以是完全不相关的进程。同一块物理内存空间被映射到两个进程，两个进程都可以访问这段共享空间从而实现了进程间通信。但是值得注意的是：Linux的共享内存通信只提供了数据交换功能，并没有提供同步机制，需要使用其它机制来同步不同进程对共享内存的读写操作（如信号量，我们之后再看）。

     共享内存是进程间通信中最简单的方式之一。共享内存区是最快的IPC形式。一旦这样的内存映射到共享它的进程的空间，这些进程就会涉及到内核，换句话说就是进程不再通过执行内核的系统调用来传递彼此的数据。

首先我们来看一下共享内存的示意图

共享内存的示意图

共享内存的数据结构

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 */
};

共享内存函数

shmget函数

功能：⽤用来创建共享内存

原型

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

参数

    key:这个共享内存段名字

    size:共享内存⼤⼩

    shmflg:由九个权限标志构成，它们的⽤法和创建⽂件时使⽤的mode模式标志是一样的

          IPC_CREAT：如果不存在就创建

           IPC_EXCL：  如果已经存在则返回失败

           IPC_EXCL：  如果已经存在则返回失败

返回值：成功返回⼀个⾮负整数，即该共享

shmat函数

功能：将共享内存段连接到进程地址空间

原型

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

参数：

    shmid: 共享内存标识

    shmaddr:指定连接的地址

    shmflg:它的两个可能取值是SHM_RND和SHM_RDONLY

返回值：成功返回⼀一个指针，指向共享内存第⼀一个节；失败返回-1

说明：

    shmaddr为NULL，核心自动选择一个地址

    shmaddr不为NULL且shmflg无SHM_RND标记，则以shmaddr为连接地址。

    shmaddr不为NULL且shmflg设置了SHM_RND标记，则连接的地址会自动向下调整为SHMLBA的整数倍。

            公式：shmaddr - (shmaddr % SHMLBA)

    shmflg=SHM_RDONLY，表⽰连接操作用来只读共享内存

shmdt函数

功能：将共享内存段与当前进程脱离

原型：

int shmdt(const void *shmaddr);

参数

    shmaddr: 由shmat所返回的指针

返回值：成功返回0；失败返回-1

注意：将共享内存段与当前进程脱离不等于删除共享内存段

shmctl函数

功能：⽤用于控制共享内存

原型

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

参数

    shmid:由shmget返回的共享内存标识码

    cmd:将要采取的动作（有三个可取值）

    buf:指向⼀一个保存着共享内存的模式状态和访问权限的数据结构

返回值：成功返回0；失败返回-1

IPC_RMID：删除。(常用 )
IPC_SET：设置 shmid_ds 参数，相当于把共享内存原来的属性值替换为 buf 里的属性值。

IPC_STAT：保存 shmid_ds 参数，把共享内存原来的属性值备份到 buf 里。

现在我们来实现以下进程间通信（server/client之间进行通信）

直接看代码：

#ifndef _COMM_H_
#define _COMM_H_

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

#define PATHNAME "." // ftok函数 生成key使用
#define PROJ_ID 66 // ftok 函数生成key使用

int create_shm(int size);//  分配指定大小的共享内存块
int destroy_shm(int shmid); // 释放指定id的共享内存块
int get_shmid(); // 获取已经存在的共享内存块


#endif /*_COMM_H_*/

comm.c

#include "comm.h"

 
static int comm_shm(int size, int shmflag)
{
	key_t key = ftok(PATHNAME, PROJ_ID); // 获取key 
	if (key < 0){
		perror("ftok");
		return -1;
	}
	int shmid = shmget(key, size, shmflag);
	if (shmid < 0){
		perror("shmget");
		return -2;
	}
	return shmid;
}

int create_shm(int size)
{
	return comm_shm(size, IPC_CREAT | IPC_EXCL | 0666);
}

int get_shmid()
{
	return comm_shm(0, IPC_CREAT);
}

int destroy_shm(int shmid)
{
	if (shmctl(shmid, IPC_RMID, NULL) < 0)
	{
		perror("shmctl");
		return -1;
	}
	return 0;
}

server.c

#include "comm.h"

int main()
{
    int shmid = get_shmid(4096);
    char *buf;
    int index = 0;
        buf = shmat(shmid,NULL, 0 );

    while( index < 4096)
    {
        printf("%s\n", buf);
        sleep(1);
        index++;
        if( index == 27)
            break; // 让程序结束
    }
    return 0;
}

client.c

#include "comm.h"

int main()
{
	int shmid = create_shm(4096);// 创建共享内存块
	char *buf;
	int i = 0;
	buf = shmat(shmid, NULL, 0);

	while (i < 4096)
	{
		buf[i] = 'a' + i;
		i++;
		sleep(1);
		if (i == 26)
			break; // 让程序结束，去释放该共享内存
	}
	destroy_shm(shmid);
	return 0;
}

我们再写一个Makefile

.PHONY:all
all:server client

client:client.c comm.c
	gcc -o $@ $^
server:server.c comm.c
	gcc -o $@ $^

.PHONY:clean
clean:
	rm -f client server

在通信之前我们来学习两个命令：

我们可以执行命令ipcs -m 查看分配的该内存块： 

上图表示该进程被attach到该内存块，等启动client进程后该值变为2；（运行后再放图）

这时我们进行两进程间通信：

此时我们输入命令 ipcs -m ，查看nattch值变为2.

那么两进程间通信就结束了。。下次我们再来学习信号量。。