快捷導(dǎo)航

C++使用map實(shí)現(xiàn)多進(jìn)程拷貝文件的程序思路

更新時(shí)間：2021年12月21日 08:31:24 作者：Jxiepc

這篇文章主要介紹了C++使用mmap實(shí)現(xiàn)多進(jìn)程拷貝文件,通過(guò)本文給大家分享程序思路及完整代碼，代碼簡(jiǎn)單易懂，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下

程序思路

1. 指定創(chuàng)建子進(jìn)程的個(gè)數(shù)

2. 打開(kāi)源文件

3. 打開(kāi)目的文件, 不存在則創(chuàng)建

4. 獲取文件大小

5. 根據(jù)文件大小拓展目標(biāo)文件

6. 為源文件創(chuàng)建映射

7. 為目標(biāo)文件創(chuàng)建映射

8. 求出每個(gè)子進(jìn)程該拷貝的字節(jié)數(shù)

9. 創(chuàng)建N個(gè)子進(jìn)程

10. 子進(jìn)程完成分塊拷貝(注意最后一個(gè)子進(jìn)程拷貝起始位置)

11. 釋放映射區(qū)

完整代碼

#include <iostream>
#include <unistd.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <fcntl.h>
#include <cstring>

int main(int argc, char *argv[])
{   
    //1. 指定創(chuàng)建子進(jìn)程的個(gè)數(shù)
    int proNum = 5;

    //2. 打開(kāi)源文件
    int fd_r = open(argv[1], O_RDONLY);
    if(fd_r == -1){
        std::cout << "open error" << std::endl;
        exit(1);
    }
    //3. 打開(kāi)目的文件, 不存在則創(chuàng)建
    int fd_w = open(argv[2], O_RDWR|O_CREAT|O_TRUNC, 0666);
    if(fd_w == -1){
        std::cout << "open error" << std::endl;
        exit(1);
    }
    /*4. 獲取文件大小
     *--------------
     * 相比于seek速度更快
     * */
    struct stat statbuf;
    stat(argv[1], &statbuf);
    int size = statbuf.st_size;
    //5. 根據(jù)文件大小拓展目標(biāo)文件
    int ret = ftruncate(fd_w, size);
    if(ret == -1){
        std::cout << "ftruncate error" << std::endl;
        exit(1);
    }
    std::cout << argv[1] <<"的文件大小為：" << size << std::endl;
    //6. 為源文件創(chuàng)建映射
    char *mmp_r = (char *)mmap(NULL, size, PROT_READ, MAP_SHARED, fd_r, 0);
    if(mmp_r == MAP_FAILED){
        std::cout << "mmap error" << std::endl;
        exit(1);
    }

    close(fd_r);
    //7. 為目標(biāo)文件創(chuàng)建映射
    char *mmp_w = (char *)mmap(NULL, size, PROT_WRITE, MAP_SHARED, fd_w, 0);
    if(mmp_w == MAP_FAILED){
        std::cout << "mmap error" << std::endl;
        exit(1);
    }
    //8. 求出每個(gè)子進(jìn)程該拷貝的字節(jié)數(shù)
    int size_sp = size / proNum;
    int size_sle = size % proNum;
    //9. 創(chuàng)建N個(gè)子進(jìn)程
    int i;
    for(i=0; i<proNum ; ++i){
        pid_t pid = fork();
        if(pid == -1){
            std::cout << "fork error" << std::endl;
            exit(1);
        }else if(pid == 0){ // 子進(jìn)程
            /* 拷貝文件內(nèi)容 */
            if(i < proNum - 1)
            {
                memcpy(mmp_w+i*size_sp, mmp_r+i*size_sp, size_sp);
                std::cout << "我是進(jìn)程" << i << "拷貝文件大小為：" << size_sp<< std::endl;
            }
            else{
                memcpy(mmp_w+i*size_sp, mmp_r+i*size_sp, size_sp+size_sle);
                std::cout << "我是進(jìn)程" << i << "拷貝文件大小為：" << size_sp+size_sle << std::endl;
            }
            break;
        }
    }
    munmap(mmp_r, size);
    munmap(mmp_w, size);

    return 0;
}