Linux学习（17）：内存映射实现通信

1. 内存映射的概念

两个不同的进程“标记”同一块区域，通过修改区域的内容实现通信

内存映射是一种访问文件的方式，它将文件映射到进程的地址空间中。这样做可以使应用程序像访问内存一样访问文件内容，而不必使用标准的读写函数进行I/O操作。

内存映射通常比传统I/O更快速、更高效。因为当我们使用传统I/O来读取大量数据时，需要多次调用read()或write()等系统调用，并且每次都要从用户态切换到内核态执行相关操作；而使用内存映射，则可以避免这些额外开销并且能够利用现代计算机硬件特性如CPU缓存、预取等加速数据处理过程。

值得注意的是，在使用内存映射时需要考虑安全问题：如果多个进程同时访问同一个共享区域，则可能会出现竞争条件和死锁等问题。

#include <sys/mman.h>
void *mmap(void *addr, size_t length, int prot, int flags,int fd, off_t offset);
    - 功能：将一个文件或者设备的数据映射到内存中
    - 参数：
        - void *addr: NULL, 由内核指定分配一块地址
        - length : 要映射的数据的长度，这个值不能为0。建议使用文件的长度。（并不是设置多少就是多少，会给你一个分页大小的整数倍）
                获取文件的长度：stat lseek
        - prot : 对申请的内存映射区的操作权限
            -PROT_EXEC ：可执行的权限
            -PROT_READ ：读权限
            -PROT_WRITE ：写权限
            -PROT_NONE ：没有权限
            要操作映射内存，必须要有读的权限。
            PROT_READ、PROT_READ|PROT_WRITE
        - flags :
            - MAP_SHARED : 映射区的数据会自动和磁盘文件进行同步，进程间通信，必须要设置这个选项
            - MAP_PRIVATE ：不同步，内存映射区的数据改变了，对原来的文件不会修改，会重新创建一个新的文件。（copy on write）
        - fd: 需要映射的那个文件的文件描述符
            - 通过open得到，open的是一个磁盘文件
            - 注意：文件的大小不能为0，open指定的权限不能和prot参数有冲突。prot权限小于open权限
                prot: PROT_READ                open:只读/读写 
                prot: PROT_READ | PROT_WRITE   open:读写
        - offset：偏移量，一般不用。必须指定的是4k的整数倍，0表示不偏移。
    - 返回值：返回创建的内存的首地址
        失败返回 MAP_FAILED，它的值为(void *) -1

int munmap(void *addr, size_t length);
    - 功能：释放内存映射
    - 参数：
        - addr : 要释放的内存的首地址,mmap函数的返回值
        - length : 要释放的内存的大小，要和mmap函数中的length参数的值一样。

2. 父子进程间通过内存映射通信

- 还没有子进程的时候

- 通过唯一的父进程，先创建内存映射区

- 有了内存映射区以后，创建子进程

- 父子进程共享创建的内存映射区

测试代码：

#include <stdio.h>
#include <sys/mman.h>
#include <fcntl.h>
#include <sys/types.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <wait.h>
int main() {

    // 1.打开一个文件
    int fd = open("test.txt", O_RDWR);
    int size = lseek(fd, 0, SEEK_END);  // 获取文件的大小

    // 2.创建内存映射区
    void *ptr = mmap(NULL, size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
    if(ptr == MAP_FAILED) {
        perror("mmap");
        exit(0);
    }

    // 3.创建子进程
    pid_t pid = fork();
    if(pid > 0) {
        // 父进程
        wait(NULL);//回收
        char buf[64];
        strcpy(buf, (char *)ptr);
        printf("read data : %s\n", buf);
       
    }else if(pid == 0){
        // 子进程
        strcpy((char *)ptr, "nihao a, son!!!");
    }
    // 关闭内存映射区
    munmap(ptr, size);
    return 0;
}

strcpy函数：

char *strcpy(char *dest, const char *src);

strcpy函数不会检查目标和源区域是否重叠或者越界，在使用时应该保证这些问题不会发生

由于第二个参数const修饰了src指针，因此调用者不能通过该指针修改其所指向内存中保存的值；

函数返回值为char*类型，并且总是返回dest（也就是目标地址）

参数是char*类型，而ptr是一个空指针类型（void*），需要进行强制转换

3. 无关进程间通过内存映射通信

- 准备一个大小不是0的磁盘文件

- 进程1 通过磁盘文件创建内存映射区

- 得到一个操作这块内存的指针

- 进程2 通过磁盘文件创建内存映射区

- 得到一个操作这块内存的指针

- 使用内存映射区通信

注意：内存映射区通信，是非阻塞。

因此，如果令一个进程读另一个进程写，读的进程需要一直while循环读

//写进程
int main(){

    //打开文件
    int fd = open("test.txt",O_RDWR);
    int len = lseek(fd,0,SEEK_END);
    //创建映射
    void *ptr = mmap(NULL,len,PROT_READ|PROT_WRITE,MAP_SHARED,fd,0);
    //往里面写东西
    strcpy((char*)ptr,"12345678");
    munmap(ptr,len);
    return 0;
}

//读进程
int main(){

    //打开文件
    int fd = open("test.txt",O_RDWR);
    int len = lseek(fd,0,SEEK_END);
    //创建映射
    void *ptr = mmap(NULL,len,PROT_READ|PROT_WRITE,MAP_SHARED,fd,0);
    //从里面读取
    char buf[64];
    strcpy(buf,(char*)ptr);
    printf("read:%s",buf);
    munmap(ptr,len);
    return 0;
}

4. 一些思考问题

1. 如果对mmap的返回值(ptr)做++操作(ptr++), munmap是否能够成功?

ptr指针允许++操作，但是munmap必须和mmap创建的内存映射区的首地址相同才能够对mmap映射进行关闭

2.open时O_RDONLY, mmap时prot参数指定PROT_READ | PROT_WRITE会怎样?

错误，prot的权限必须小于open的fd权限，会报错MAP_FAILED

因此建议将两者的权限写为一致

3.如果文件偏移量为1000会怎样?

会返回MAP_FAILED，因为偏移量必须是4k的整数倍

通常情况下，操作系统会将物理内存和磁盘上的数据都按页进行划分和管理（一页通常为4KB或8KB），所以offset参数应该是页大小（4096字节）的整数倍才能保证正确地进行内存映射操作

4.mmap什么情况下会调用失败?

length=0；或者prot权限大于open的fd权限

5.可以open的时候O_CREAT一个新文件来创建映射区吗?

可以，但是必须保证创建的大小不为0

使用lseek函数和truncate函数对文件进行拓展

要注意的是 lseek(fd,0,SEEK_END)并不能实际扩展文件，我们要写一个字符占位

write(fd," ",1);

6.mmap后关闭文件描述符，对mmap映射有没有影响？

没有影响；

使用mmap创建映射区后，关闭创建映射区的文件描述符fd并不会影响已经建立的映射关系。这是因为在调用mmap函数时，内核会将打开的文件和进程地址空间中的一段虚拟地址进行绑定，并返回一个指向该虚拟地址起始位置的指针。

这个指针可以用来访问这段映射区域，而与之相关联的文件描述符只是表示了进程打开了一个文件，并没有直接作用于该内存区域。所以即使关闭了该文件描述符，也不会对已经建立好的映射关系产生任何影响。（用户态的文件指针不会对内核态的内存有作用）

7.对ptr越界操作会怎样？

和所有的指针访问一样，越界操作操作的是一个非法的内存，会产生段错误

除此之外，还有其他一些原因也可能导致段错误，比如调用野指针、栈溢出等等。无论何种原因造成的段错误都表明程序存在严重问题需要修复。

5.匿名映射

定义：指的是不需要文件实体的一个内存映射

因为没有文件实体，所以只能做父子关系进程的映射

做法：

flags中多一个MAP_ANONYMOUS

fd文件描述符填-1

int main() {

    // 1.创建匿名内存映射区
    int len = 4096;
    void * ptr = mmap(NULL, len, PROT_READ | PROT_WRITE, MAP_SHARED | MAP_ANONYMOUS, -1, 0);
    if(ptr == MAP_FAILED) {
        perror("mmap");
        exit(0);
    }

    // 父子进程间通信
    pid_t pid = fork();

    if(pid > 0) {
        // 父进程
        strcpy((char *) ptr, "hello, world");
        wait(NULL);
    }else if(pid == 0) {
        // 子进程
        sleep(1);
        printf("%s\n", (char *)ptr);
    }

    // 释放内存映射区
    int ret = munmap(ptr, len);

    if(ret == -1) {
        perror("munmap");
        exit(0);
    }
    return 0;
}