一、相关练习
1. 使用父子进程完成两个文件的拷贝
要求:父进程拷贝前一半内容,子进程拷贝后一半内容
子进程结束后退出,父进程回收子进程的资源
1.1> copy_fork.c
#include <myhead.h>
int main(int argc, const char *argv[])
{
//判断传入文件个数
if(argc != 3)
{
perror("input file error");
return -1;
}
//定义文件描述符
int sfd,dfd;
//以只读的形式打开源文件
if((sfd = open(argv[1], O_RDONLY)) == -1)
{
perror("src file open error");
return -1;
}
//定义文件搬运工
char buf[128] = "";
//计算源文件文件大小
int res = 0;
int src_size = 0;
while((res= read(sfd, buf, sizeof(buf))))
{
src_size += res;
}
//定义进程pid号,创建子进程
pid_t pid = fork();
//判断父子进程
if(pid > 0)
{
//将源文件光标置位文件开头
lseek(sfd, 0, SEEK_SET);
//以只写的形式打开目标文件
if((dfd = open(argv[2], O_WRONLY|O_CREAT|O_TRUNC, 0664)) == -1)
{
perror("dest file open error");
return -1;
}
int dest_size = 0;
//将源文件前一半内容写入目标文件
while(1)
{
res = read(sfd, buf, sizeof(buf));
dest_size += res; //写入后目标文件当前大小
if(dest_size > src_size/2)
{
write(dfd, buf, (src_size/2)%sizeof(buf));
break;
}
write(dfd, buf, res);
}
}else if(pid == 0)
{
sleep(0.01);
//以追加的形式打开目标文件
if((dfd = open(argv[2], O_WRONLY|O_APPEND|O_CREAT, 0664)) == -1)
{
perror("dest file open error");
return -1;
}
//将源文件光标置位文件中间
lseek(sfd, src_size/2, SEEK_SET);
//将源文件剩余内容写入目标文件
while(write(dfd, buf, read(sfd, buf, sizeof(buf))));
//退出子程序,不刷新缓IO冲区
_exit(EXIT_SUCCESS);
}
//回收子进程
wait(NULL);
return 0;
}
1.2> 程序实现效果
1.3> 代码修正
#include<myhead.h>
//定义求源文件大小的函数
int get_file_len(const char *srcfile, const char *destfile)
{
//以只读的形式打开源文件
int sfd = open(srcfile, O_RDONLY);
if(sfd == -1)
{
perror("open srcfile error");
return -1;
}
//以创建的形式打开目标文件
int dfd = open(destfile, O_WRONLY|O_CREAT|O_TRUNC, 0664);
if(dfd == -1)
{
perror("open destfile error");
return -1;
}
int size = lseek(sfd, 0, SEEK_END);
//关闭文件
close(sfd);
close(dfd);
return size; //将文件大小返回
}
//定义文件拷贝函数
void copy_file(const char *srcfile, const char *destfile, int start, int len)
{
//以只读的形式打开源文件
int sfd = open(srcfile, O_RDONLY);
if(sfd == -1)
{
perror("open srcfile error");
return ;
}
//以创建的形式打开目标文件
int dfd = open(destfile, O_WRONLY);
if(dfd == -1)
{
perror("open destfile error");
return ;
}
//移动光标位置
lseek(sfd, start, SEEK_SET);
lseek(dfd, start, SEEK_SET);
//定义搬运工
char buf[128] = "";
int sum = 0; //记录拷贝的总个数
while(1)
{
//从源文件中读取数据
int res = read(sfd, buf, sizeof(buf));
sum += res; //将读取的个数累加
if(res == 0 || sum>len) //表示文件读取结束
{
write(dfd, buf, res-(sum-len)); //父进程将最后一次拷贝结束
break;
}
//写入到目标文件中
write(dfd, buf, res);
}
//关闭文件
close(sfd);
close(dfd);
printf("拷贝成功\n");
}
/******************************主程序*************************/
int main(int argc, const char *argv[])
{
//判断是否传入两个文件
if(argc != 3)
{
printf("input file error\n");
return -1;
}
//求出源文件长度
int len = get_file_len(argv[1], argv[2]);
//创建两个进程
pid_t pid = fork();
if(pid > 0)
{
//父进程,拷贝前一半内容
copy_file(argv[1], argv[2], 0, len/2);
}else if(pid == 0)
{
//子进程拷贝后一半内容
copy_file(argv[1], argv[2], len/2, len-len/2);
//退出进程
exit(EXIT_SUCCESS);
}else
{
perror("fork error");
return -1;
}
//回收子进程资源
wait(NULL);
return 0;
}