【Linux】mmap文件映射的使用

目录

mmap文件映射介绍

参数介绍:

返回值介绍:

对文件进行操作的常用系统调用

写入映射

读取映射

使用mmap极简模拟实现malloc


mmap文件映射介绍

允许用户空间程序将文件或设备的内容直接映射到进程的虚拟地址空间中,通过mmap程序可以高效的方法问文件数据,而无需通过调用传统的read或write进行数据的复制。

参数介绍:

  • addr:表示希望映射区域在进程地址空间中的开始虚拟地址,如果是NULL,则操作系统会自动选择一个合适的地址。
  • length:要映射到进程地址空间中的字节数,这个大小必须是操作系统页面大小的整数倍(通常是4KB也就是4096B,不同的操作系统页面大小可能不同)。如果指定的len大小不是系统页面大小的整数倍,系统可能会自动向上舍入到最近的页面大小边界。
  • prot:指定映射区域的内存保护属性,可以是以下组合(使用按位或|运算符)。PROT_READ:映射区域可读。PROT_WRITE:映射区域可写。PROT_EXEC:映射区域可执行。
  • flags:指定映射的类型和其他选项。MAP_PRIVATE:私有映射,对映射区域的修改不会反映到底层文件中。MAP_SHARED:共有映射,对映射区域的修改会反映到底层文件中(前提是文件是以写的方式打开的)。其他选项(MAP_ANONYMOUS)用于创建不与文件关联的匿名映射。
  • fd:一个文件描述符,指向要映射的文件。对于匿名映射,这个参数为-1 。
  • offset:文件中的起始偏移量,从文件那个位置的内容开始映射。

返回值介绍:

成功返回进程地址空间中的映射的区域对应的开始的虚拟地址。

失败返回MAP_FAILED,其实就是(void*)-1 。

对文件进行操作的常用系统调用

调整文件大小的系统调用truncate和ftruncate:

获取文件状态信息的系统调用stat和fsate:

写入映射

cpp 复制代码
#include <iostream>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/man.h>
#include <unistd.h>
#include <fcntl.h>

#define PAGE_SIZE 4096
enum Status
{
    USAGE_ERROR = 1,
    OPEN_ERROR = 2,
    FTRUNCATE_ERROR = 3,
    MMAP_ERROR = 4,
    MUNMAP_ERROR = 5
};

int main(int argc, char* argv[])
{
    if(argc != 2){
        std::cerr << "Usage: " << argv[0] << " filename" << std::endl;
        return USAGE_ERROR;
    }
    
    //映射不存在的文件先创建该文件
    int fd = open(argv[1], O_RDWR | O_CREAT | O_TRUNC, 0666);
    if(fd < 0){
        std::cerr << "Failed to open file: " << argv[1] << std::endl;
        return OPEN_ERROR;
    }

    if(ftruncate(fd, PAGE_SIZE) == -1){
        std::cerr << "Failed to ftruncate file: " << argv[1] << std::endl;
        return FTRUNCATE_ERROR; 
    }

    char* addr = (char*)::mmap(nullptr, PAGE_SIZE, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
    if(addr == MAP_FAILED){
        std::cerr << "Failed to mmap file: " << argv[1] << std::endl;
        return MMAP_ERROR;
    }

    //向文件中写入数据
    for(int i = 'A';i < 'Z';++i)
    {
        addr[i-'A'] = 'A' + i;
        sleep(1);
    }

    if(::munmap(addr, PAGE_SIZE) == -1){
        std::cerr << "Failed to munmap file: " << argv[1] << std::endl;
        return MUNMAP_ERROR;
    }

    close(fd);
    return 0;
}

读取映射

cpp 复制代码
#include <iostream>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/man.h>
#include <unistd.h>
#include <fcntl.h>

#define PAGE_SIZE 4096
enum Status
{
    USAGE_ERROR = 1,
    OPEN_ERROR = 2,
    FTRUNCATE_ERROR = 3,
    MMAP_ERROR = 4,
    MUNMAP_ERROR = 5
};

int main(int argc, char* argv[])
{
    if(argc != 2){
        std::cerr << "Usage: " << argv[0] << " filename" << std::endl;
        return USAGE_ERROR;
    }
    
    int fd = open(argv[1], O_RDONLY);
    if(fd < 0){
        std::cerr << "Failed to open file: " << argv[1] << std::endl;
        return OPEN_ERROR;
    }

    //获取文件属性
    struct stat st;
    ::fstat(fd, &st);

    char* addr = (char*)::mmap(nullptr, st.st_size, PROT_READ, MAP_SHARED, fd, 0);
    if(addr == MAP_FAILED){
        std::cerr << "Failed to mmap file: " << argv[1] << std::endl;
        return MMAP_ERROR;
    }

    //读取文件中的数据
    std::cout << "File content: " << addr << std::endl;

    if(::munmap(addr, st.st_size) == -1){
        std::cerr << "Failed to munmap file: " << argv[1] << std::endl;
        return MUNMAP_ERROR;
    }

    close(fd);
    return 0;
}

使用mmap极简模拟实现malloc

cpp 复制代码
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <sys/mman.h>
#include <unistd.h>

void* my_malloc(size_t size)
{
    if(size <= 0){
        return nullptr;
    }
    void* addr = ::mmap(nullptr, size, PROT_READ | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
    if(addr == (void*)-1){
        perror("mmap");
        return nullptr;
    }
    return addr;
}

void myf_free(void* ptr, size_t size)
{
    if(!ptr && size > 0){
        if(::munmap(ptr, size) == -1){
            perror("munmap");
        }
    }
}

int main()
{
    size_t size = 1024;
    char* ptr = (char*)my_malloc(size);
    if(ptr){
        memset(ptr,'A',size);
        for(int i = 0;i < size;++i){
        printf("%c ", ptr[i]);
        fflush(stdout);
        sleep(1);
    }
   }
    my_free(ptr, size);

    return 0;
}
相关推荐
JAVA面经实录9175 小时前
Java企业级工程化·终极完整版背诵手册(无遗漏、全覆盖、面试+落地通用)
java·开发语言·面试
HHFQ5 小时前
在 systemd 场景下的 CPU 限制方式
linux
道清茗6 小时前
【RH294知识点汇总】第 9 章 《 自动执行 Linux 管理任务 》常见问题
linux·运维·服务器
山羊硬件Time6 小时前
自动化管理Linux的好工具:shell script
linux·嵌入式硬件·硬件工程师·基带工程·硬件开发
王老师青少年编程6 小时前
csp信奥赛C++高频考点专项训练之贪心算法 --【哈夫曼贪心】:合并果子
c++·算法·贪心·csp·信奥赛·哈夫曼贪心·合并果子
周杰伦fans6 小时前
AutoCAD .NET 二次开发:深入理解 EntityJig 的工作原理与正确实现
开发语言·.net
叼烟扛炮6 小时前
C++第二讲:类和对象(上)
数据结构·c++·算法·类和对象·struct·实例化
wj3055853786 小时前
Codex + Git 开发环境配置指南(WSL版)
linux·运维·git
星马梦缘7 小时前
如何切换window-ubuntu双系统【方案二】
linux·运维·ubuntu
样例过了就是过了8 小时前
LeetCode热题100 最长公共子序列
c++·算法·leetcode·动态规划