作业一:
使用两个线程完成两个文件的拷贝,分支线程1拷贝前一半,分支线程2拷贝后一半,主线程回收两个分支线程的资源
#include <myhead.h>
#define BUF_SIZE 10 //定义缓冲区大小
typedef struct{
char* sfile;
char* tfile;
int index;
int file_size;
}Data,*DataPtr;
void * task(void *arg){
DataPtr data=(DataPtr)arg;
//打开文件
int sf=open(data->sfile,O_RDONLY);
int tf=open(data->tfile,O_WRONLY);
//移动光标
lseek(sf,data->index,SEEK_SET);
lseek(tf,data->index,SEEK_SET);
//定义缓冲区
char buf[BUF_SIZE]="";
//设置实际读取到的个数,要去读的个数
int count=-1,read_size=-1;
while(1){
//如果缓存区的容量大于剩余读取文件大小,则读取的数量为file_size
read_size=BUF_SIZE>data->file_size?data->file_size:BUF_SIZE;
count=read(sf,buf,read_size);
//如果返回值小于等于0,代表读取失败或者遇到文件结束标志EOF
if(count<=0){
break;
}
write(tf,buf,count);
//减去读取到的大小
data->file_size-=count;
}
printf("线程%ld结束\n",pthread_self());
pthread_exit(NULL);
}
int main(int argc, const char *argv[])
{
if(argc!=3){
return -1;
}
int sfd=open(argv[1],O_RDONLY);
//清空目标文件
int tfd=open(argv[2],O_WRONLY|O_CREAT|O_TRUNC,0777);
//求出文件总的大小
int file_size=lseek(sfd,0,SEEK_END);
pthread_t tid1,tid2;
//封装传入的参数
Data data1,data2;
data1.sfile=argv[1];
data1.tfile=argv[2];
data1.index=0;
data1.file_size=file_size/2;
//创建复制前半段的线程
if(pthread_create(&tid1,NULL,task,&data1)!=0){
return -1;
}
//更改传入后半段的参数
data2.sfile=argv[1];
data2.tfile=argv[2];
data2.index=file_size/2;
data2.file_size=file_size-(file_size/2);
//创建辅助后半段的线程
if(pthread_create(&tid2,NULL,task,&data2)!=0){
return -1;
}
//阻塞回收线程
pthread_join(tid1,NULL);
pthread_join(tid2,NULL);
//关闭文件流
close(sfd);
close(tfd);
return 0;
}