进程线程--线程（day2）

思维导图

小练习：

1> 使用两个线程完成两个文件的拷贝，主线程拷贝前一半内容，子线程拷贝后一半内容，并且主线程要阻塞回收子线程资源

#include <myhead.h>

//定义一个结构体
typedef struct{
    char *srcfile;
    char *dstfile;
    int start;
    int len;
}Date;

//定义求文件长度函数
int getlen(const char* srcfile,const char* dstfile){
    //定义两个文件指针
    FILE *fd1,*fd2;
    //以只读的形式打开源文件
    if((fd1=fopen(srcfile,"r"))==NULL){
        perror("srcfile open error");
        return -1;
    }

    //以只写的形式打开目标文件
    if((fd2=fopen(dstfile,"w"))==NULL){
        perror("dstfile open error");
        return -1;
    }

    //求源文件长度
    fseek(fd1,0,SEEK_END);
    int len=ftell(fd1);
    
    //关闭文件
    fclose(fd1);
    fclose(fd2);

    return len;
}

//定义文件拷贝函数
void cope_file(const char* srcfile,const char* dstfile,int start,int len){
    //定义两个文件指针
    FILE *fd1,*fd2;
    //以只读的形式打开源文件
    if((fd1=fopen(srcfile,"r"))==NULL){
        perror("srcfile open error");
        return;
    }

    //以只写的形式打开目标文件
    if((fd2=fopen(dstfile,"w"))==NULL){
        perror("dstfile open error");
        return;
    }

    //将光标定位在要开始拷贝的位置
    fseek(fd1,start,SEEK_SET);
    fseek(fd2,start,SEEK_SET);

    //开始进行拷贝工作
    char buf[128]="";     //数据搬运工
    int sum=0;         //记录搬运的总个数

    while(1){
        //从源文件中读取数据
        int ret = fread(buf,1,sizeof(buf),fd1);
        sum+=ret;     //记录每次搬运的个数

        if(ret==0||sum>len){
            fwrite(buf,1,ret-(sum-len),fd2);     //将最后一次不满128的数据拷贝到新文件中
            break;
        }

        //将数据写入目标文件
        fwrite(buf,1,ret,fd2);    
    }
    printf("拷贝成功\n");
}

//task1函数
void *task1(void *arg){
    Date date=*(Date*)arg;
    cope_file(date.srcfile,date.dstfile,date.start,date.len);
    pthread_exit(NULL);
}

int main(int argc, const char *argv[])
{
    //判断传入文件个数
    if(argc!=3){
        printf("input file error\n");
        printf("usage:./a.out srcfile dstfile\n");
        return -1;
    }

    //求文件长度
    int len=getlen(argv[1],argv[2]);

    //创建1个线程，分别进行拷贝工作
    pthread_t tid;

    Date date={argv[1],argv[2],len-len/2,len/2};

    if(pthread_create(&tid,NULL,task1,&date)){
        printf("线程创建失败\n");
        return -1;
    }
    cope_file(argv[1],argv[2],0,len);
    pthread_join(tid,NULL);

    return 0;
}
 

2> 使用三个线程完成两个文件的拷贝，主线程拷贝前三分之一，子线程1拷贝中间三分之一，子线程2拷贝后三分之一，主线程要设置两个子线程为分离态

#include <myhead.h>

//定义一个数据结构体
typedef struct{
    char *srcfile;
    char *dstfile;
    int start;
    int len;
}Date;

//求源文件长度
int getlen(const char *srcfile,const char *dstfile){
    int fd1,fd2;
    
    //以只读的形式打开源文件
    if((fd1=open(srcfile,O_RDONLY))==-1){
        perror("srcfile open error");
        return -1;
    }

    //以只写的形式打开目标文件
    if((fd2=open(dstfile,O_WRONLY|O_CREAT|O_TRUNC,0664))==-1){
        perror("srcfile open error");
        return -1;
    }

    //求源文件长度
    int len=lseek(fd1,0,SEEK_END);

    //关闭文件
    close(fd1);
    close(fd2);
    return len;
}

//copy函数
void copy_file(const char *srcfile,const char *dstfile,int start,int len){
    //定义两个文件描述符
    int fd1,fd2;

    //以只读的形式打开源文件
    if((fd1=open(srcfile,O_RDONLY))==-1){
        perror("srcfile open error");
        return;
    }

    //以只写的形式打开目标文件
    if((fd2=open(dstfile,O_WRONLY))==-1){
        perror("srcfile open error");
        return;
    }
    
    //将光标定位在要开始拷贝的位置
    lseek(fd1,start,SEEK_SET);
    lseek(fd2,start,SEEK_SET);

    char buf[128]="";     //数据搬运工
    int sum=0;             //记录搬运的总个数

    while(1){
        //从源文件中读取数据
        int ret=read(fd1,buf,sizeof(buf));
        sum+=ret;         //记录每次搬运的个数

        if(ret==0||sum>len){
            write(fd2,buf,ret-(sum-len));         //将最后一次不满128的数据拷贝到新文件中
            break;
        }

        //将数据写入目标文件
        write(fd2,buf,ret);
    }

    printf("拷贝成功\n");
        
}

//task1函数
void* task1(void* arg){
    Date date=*(Date*)arg;
    copy_file(date.srcfile,date.dstfile,date.start,date.len);
    pthread_exit(NULL);
}

int main(int argc, const char *argv[])
{
    //判断传入文件个数
    if(argc!=3){
        printf("input file error\n");
        printf("usage:./a.out srcfile dstfile\n");
        return -1;
    }

    //求文件长度
    int len=getlen(argv[1],argv[2]);

    //创建task传入的值
    Date date1={argv[1],argv[2],len/3,len/3*2};
    Date date2={argv[1],argv[2],len/3*2,len-len/3*2};

    //创建2个线程，分别进行拷贝工作
    pthread_t tid1,tid2;
    if(pthread_create(&tid1,NULL,task1,&date1)){
        printf("pthread create error\n");
        return -1;
    }

    if(pthread_create(&tid2,NULL,task1,&date2)){
        printf("pthread create error\n");
        return -1;
    }

    //主线程的内容
    copy_file(argv[1],argv[2],0,len/3);
    
    //将tid1，tid2设为分离态
    pthread_join(tid1,NULL);
    pthread_join(tid2,NULL);

    return 0;
}