作业:用多线程完成文件拷贝
#include<myhead.h>
typedef struct
{
int size;//文件大小
char srcname[20];//源文件文件名
char destname[20];//目标文件文件名
}data;
data file_data;
//task1,拷贝前半段
void *task1(void *arg)
{
int fd1,fd2;
data *file_data=(data *)arg;
if((fd1=open(file_data->srcname,O_RDONLY))==-1)
{
perror("open file");
}
if((fd2=open(file_data->destname,O_RDWR))==-1)
{
perror("open file");
}
char buf[20]="";
lseek(fd1,0,SEEK_SET);
lseek(fd2,0,SEEK_SET);
int ret=read(fd1,buf,(file_data->size)/2);
write(fd2,buf,ret);
close(fd1);
close(fd2);
pthread_exit(NULL);
}
//task2,拷贝后半段
void *task2(void *arg)
{
int fd1,fd2;
data *file_data=(data *)arg;
if((fd1=open(file_data->srcname,O_RDONLY))==-1)
{
perror("open file");
}
if((fd2=open(file_data->destname,O_RDWR))==-1)
{
perror("open file");
}
char buf[20]="";
lseek(fd1,(file_data->size)/2,SEEK_SET);
lseek(fd2,(file_data->size)/2,SEEK_SET);
int ret=read(fd1,buf,(file_data->size)/2);
write(fd2,buf,ret);
close(fd1);
close(fd2);
pthread_exit(NULL);
}
int main(int argc, char const *argv[])
{
if(argc!=3)
{
printf("input file error\n");
return -1;
}
//创建两个文件描述符
int fd1,fd2;
//打开文件,并且创建文件
if((fd1=open(argv[1],O_RDONLY))==-1)
{
perror("open file");
return -1;
}
if((fd2=open(argv[2],O_RDWR|O_CREAT|O_TRUNC,0664))==-1)
{
perror("open file");
return -1;
}
close(fd1);
close(fd2);
//求文件大小
struct stat sb;
stat(argv[1],&sb);
int len=sb.st_size;
//给结构体初始化
file_data.size=len;
strcpy(file_data.srcname,argv[1]);//名字是字符串,用复制,不可以用=
strcpy(file_data.destname,argv[2]);
//创建线程,tid1负责前半段,tid2负责后半段
pthread_t tid1,tid2;
if(pthread_create(&tid1,NULL,task1,&file_data))
{
perror("create1 error");
return -1;
}
if(pthread_create(&tid2,NULL,task2,&file_data))
{
perror("create2 error");
return -1;
}
pthread_join(tid1,NULL);//阻塞态,等待线程结束收尸
pthread_join(tid2,NULL);
return 0;
}