【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;
}
相关推荐
管理大亨2 小时前
ELK的操作应用
开发语言·python·elk
步达硬件2 小时前
【matlab】代码库-双线性插值
开发语言·matlab
就叫飞六吧2 小时前
api vs jsp 绑定风格
java·开发语言
花北城2 小时前
【C#】DbToLinq多表连接
开发语言·c#
Lueeee.2 小时前
Linux内核Kbuild编译系统
linux·服务器
曹牧2 小时前
C#:string.IndexOf
服务器·开发语言·c#
福大大架构师每日一题2 小时前
2025年12月TIOBE编程语言排行榜,Go语言排名第15,Rust语言排名17。编程语言 R 重返前十。
开发语言·后端·rust
Percep_gan2 小时前
Linux中安装rabbitmq,很详细
linux·运维·rabbitmq
vortex52 小时前
Linux .forward 文件详解
linux·运维·前端