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

猜你喜欢

转载自blog.csdn.net/weixin_44962938/article/details/119701752