Linux 多进程使用mmap拷贝文件的简单实现

多进程利用内存映射区拷贝文件的简单实现代码着重实现功能
百度搜索相关代码结果靠前的几份都有相同的错误:将每个子进程要拷贝的内容以子进程执行顺序进行指定(子进程执行顺序是不定的),这将导致拷贝出的新文件很容易出现内容错乱。


程序会自动将文件text.txt的内容拷贝到新文件text_t.txt中,新文件由程序创建,但你得预先准备一个带内容的text.txt文件(与代码文件同路径)。

#include <stdio.h>
#include <unistd.h>
#include <sys/mman.h>
#include <sys/wait.h>
#include <string.h>
#include <fcntl.h>
#include <assert.h>  //简洁起见,引入assert
#include <sys/stat.h>
const int PROCESSNUM = 5;  //进程数

int main(){
    int ret;
    int src_fd = open("text.txt", O_RDONLY);  //待拷贝文件text.txt
    assert(src_fd >= 0);

    struct stat statBuf;
    ret = fstat(src_fd, &statBuf);  //获取文件属性到statBuf
    assert(ret >= 0);
    close(src_fd);
	int fileSize = statBuf.st_size;  //获取文件大小
    int length = (fileSize+PROCESSNUM-1)/PROCESSNUM;  //平均每个进程需要拷贝的字节数

    int dct_fd = open("text_t.txt", O_CREAT|O_TRUNC|O_RDWR, 0664);  //新建文件text_t.txt
    ret = ftruncate(dct_fd, fileSize);  //填充新文件到指定大小
    assert(ret >= 0);
    close(dct_fd);
    
    pid_t pid;
    for(int i=0; i<PROCESSNUM; i++){
        pid = fork();
        if(pid != 0) continue;  //父进程止步...

        int src_fd = open("text.txt", O_RDWR);  //子进程打开待读文件text.txt
        assert(src_fd >= 0);
        int dct_fd = open("text_t.txt", O_CREAT|O_RDWR, 0664);  //子进程打开待写文件text_t.txt
        assert(dct_fd >=0);

        int curLength = i+1==PROCESSNUM?fileSize-(length*i):length;  //特判,最后一个进程实际需要拷贝的字节数
        void *src_p = mmap(NULL, curLength, PROT_READ, MAP_SHARED, src_fd, 0);  //普普通通的内存映射
        assert(src_p != MAP_FAILED);
        void *dct_p = mmap(NULL, curLength, PROT_READ|PROT_WRITE, MAP_SHARED, dct_fd, 0);
        assert(dct_p != MAP_FAILED);
        
        memcpy(dct_p+(length*i), src_p+(length*i), curLength);  //偏移指针,相应进程拷贝相应数据
        close(src_fd);
        close(dct_fd);
        munmap(src_p, curLength);
        munmap(dct_p, curLength);
        return 0;  //子进程去死
    }

    while(wait(NULL) >= 0);  //给所有子进程收尸
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

刺子

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值