#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. 指定創建子進程的個數
    int proNum = 5;

    //2. 打開源文件
    int fd_r = open(argv[1], O_RDONLY);
    if(fd_r == -1){
        std::cout << "open error" << std::endl;
        exit(1);
    }
    //3. 打開目的文件, 不存在則創建
    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. 根據文件大小拓展目標文件
    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. 為源文件創建映射
    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. 為目標文件創建映射
    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. 求出每個子進程該拷貝的字節數
    int size_sp = size / proNum;
    int size_sle = size % proNum;
    //9. 創建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){ // 子進程
            /* 拷貝文件內容 */
            if(i < proNum - 1)
            {
                memcpy(mmp_w+i*size_sp, mmp_r+i*size_sp, size_sp);
                std::cout << "我是進程" << i << "拷貝文件大小為：" << size_sp<< std::endl;
            }
            else{
                memcpy(mmp_w+i*size_sp, mmp_r+i*size_sp, size_sp+size_sle);
                std::cout << "我是進程" << i << "拷貝文件大小為：" << size_sp+size_sle << std::endl;
            }
            break;
        }
    }
    munmap(mmp_r, size);
    munmap(mmp_w, size);

    return 0;
}