IO—进程线程day5

 使用两个线程完成两个文件的拷贝，并且主线程要阻塞回收子线程资源

#include <myhead.h>

typedef struct file
{
	const char *fd1;
	const char *fd2;
	int start;
	int len;
} Data;


//定义求文件长度函数
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("dstfile open error");
        return -1;
    }

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

    //关闭文件
    close(fd1);
    close(fd2);

    return len;

}
/**********************定义文件拷贝函数***********************************/
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("dstfile 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");
}

void *task1(void *arg){

	printf("子线程开始拷贝\n");
	Data a=*(Data *)arg;
	copy_file(a.fd1,a.fd2,a.start,a.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]);
	Data a={argv[1],argv[2],0,len/2};
	Data b={argv[1],argv[2],len/2,len-len/2};

	pthread_t tid;
	if(pthread_create(&tid,NULL,task1,&a)){
		perror("copy error");
		return -1;
	}
	
	copy_file(b.fd1,b.fd2,b.start,b.len);
	pthread_join(tid,NULL);

	
	return 0;
}

 

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

#include <myhead.h>

//定义一个文件相关参数结构体变量
typedef struct file
{
	const char *fd1;
	const char *fd2;
	int start;
	int len;
} Data;


//定义求文件长度函数
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("dstfile open error");
        return -1;
    }

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

    //关闭文件
    close(fd1);
    close(fd2);

    return len;

}
/**********************定义文件拷贝函数***********************************/
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("dstfile 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");

}

//子线程1
void *task1(void *arg){

	printf("子线程1开始拷贝\n");
	Data b=*(Data *)arg;
	copy_file(b.fd1,b.fd2,b.start,b.len);
	pthread_exit(NULL);
	
}

//子线程2
void *task2(void *arg){

	printf("子线程2开始拷贝\n");
	Data c=*(Data *)arg;
	copy_file(c.fd1,c.fd2,c.start,c.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]);

	//将文件参数封装到结构体变量中，分别定义三个，存放不同段
	Data a={argv[1],argv[2],0,len/3};
	Data b={argv[1],argv[2],len/3,(len-len/3)-(len-len/3*2)};
	Data c={argv[1],argv[2],len/3*2,len-len/3*2};

	//定义主线程1
	pthread_t tid1;
	if(pthread_create(&tid1,NULL,task1,&b)){
		perror("copy error");
		return -1;
	}
	
	//定义主线程2
	pthread_t tid2;
	if(pthread_create(&tid2,NULL,task2,&c)){
		perror("copy error");
		return -1;
	}
	//在主线程内拷贝一部分文件
	copy_file(a.fd1,a.fd2,a.start,a.len);

	//回收两个线程资源
	pthread_join(tid1,NULL);
	pthread_join(tid2,NULL);

	
	return 0;
}