使用mmap實現(xiàn)多進(jìn)程對大文件拷貝

#include<stdio.h>
#include<stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/mman.h>
#include<sys/wait.h>
#include<string.h>
#include <fcntl.h>
 
 
int main(int argc,char*argv[])
{
 int n;
 if(argc < 3 || argc > 4)
 {
 printf("Enter like this : ./a.out file_src file_dst [proc_number]\n");
 exit(1);
 }
 else if(argc == 3) //用戶未指定，默認(rèn)創(chuàng)建5個進(jìn)程。
 n = 5;
 else
 n = atoi(argv[3]);
 
 //2.打開源文件
 int fd_src = open(argv[1],O_RDONLY);
 if(fd_src < 0)
 {
 perror("open");
 exit(2);
 }
 //3.打開目標(biāo)文件，不存在就創(chuàng)建，存在則截斷為0的大小。
 int fd_dst = open(argv[2],O_RDWR |O_CREAT |O_TRUNC,0664);
 if(fd_dst < 0)
 {
 perror("open");
 exit(3);
 }
 //4.獲取源文件大小。
 struct stat sbuf;
 int ret = fstat(fd_src,&sbuf); //fd_src所指向的文件信息保存到結(jié)構(gòu)體sbuf中。
 if(ret < 0)
 {
 perror("fstat");
 exit(4);
 }
 int flen = sbuf.st_size; //源文件大小。
 if(flen < n) //文件長度小于進(jìn)程個數(shù)。
 {
 n = flen;
 }
 //5.根據(jù)文件大小拓展目標(biāo)文件。
 ret = ftruncate(fd_dst,flen);//將參數(shù)fd指定的文件大小改為參數(shù)length指定的大小
 if(ret < 0)
 {
 perror("ftruncate");
 exit(5);
 }
 //6.為源文件創(chuàng)建映射。
// void *mmap(void *addr, size_t length, int prot, int flags, int fd, off_t offset);
//addr == NULL,表示內(nèi)核選擇一個合適的地址創(chuàng)建一個length大小的共享內(nèi)存，
 char *mp_src = (char*)mmap(NULL,flen,PROT_READ,MAP_SHARED,fd_src,0); //0，表示將fd_src所指向的文件從起始映射到共享內(nèi)存中，共享內(nèi)存的權(quán)限為只讀，進(jìn)程間共享。
 if(mp_src == MAP_FAILED) //mmap一定要檢查返回值。
 {
 perror("mmap");
 exit(6);
 }
 close(fd_src);
 
 //7.為目標(biāo)文件創(chuàng)建映射。
 char *mp_dst = (char*)mmap(NULL,flen,PROT_READ|PROT_WRITE,MAP_SHARED,fd_dst,0);
 if(mp_dst == MAP_FAILED) //mmap一定要檢查返回值。
 {
 perror("mmap");
 exit(7);
 }
 close(fd_dst);
 
 //8.求出每個進(jìn)程拷貝的字節(jié)數(shù)。
 int bs = flen / n;
 int mod = flen % bs; //求出均分后余下的字節(jié)數(shù)，讓最后一個子進(jìn)程處理。
 
 char *temp_src = mp_src;
 char *temp_dst = mp_dst;
 
 //9.創(chuàng)建n個子進(jìn)程。
 int i ;
 pid_t pid;
 for(i = 0; i < n; ++i)
 {
 printf("create %dth proc\n",i);
 if( (pid =fork()) == 0 )
  break;
 }
 if(n == i) //父進(jìn)程。
 {
 int j = 0;
 for(j = 0; j < n; ++j)
  wait(NULL);
 }
 else if(i == (n-1)) //10.子進(jìn)程拷貝，最后一個子進(jìn)程，它多處理均分后剩下的字節(jié)數(shù)。
 {
 printf("i = %d\n",i);
 memcpy(temp_dst+i*bs,temp_src+i*bs,bs+mod);
 }
 else if(i == 0)
 {
 printf("i = %d\n",i);
 memcpy(temp_dst,temp_src,bs);
 }
 else 
 {
 printf("i = %d\n",i);
 memcpy(temp_dst+i*bs,temp_src+i*bs,bs);
 }
 
 //11.釋放映射區(qū)。
 
 munmap(mp_src,flen);
 munmap(mp_dst,flen);
 
 return 0;
}