IOday4

一、思维导图

二、练习

1、使用父子进程完成两个文件的拷贝，父进程拷贝前一半内容，子进程拷贝后一半内容，子进程结束后退出，父进程回收子进程的资源

#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;
}