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