1、使用两个线程完成两个文件的拷贝,分支线程1拷贝前一半,分支线程2拷贝后一半,主线程回收两个分支线程的资源
#include <myhead.h>
//定义文件相关数据结构体
typedef struct
{
int fd1;
int fd2;
int mid;
int num;
}DATA, *DATAPTR;
//定义拷贝函数
void *task(void *arg)
{
int fd1 = ((DATAPTR)arg)->fd1;
int fd2 = ((DATAPTR)arg)->fd2;
int mid = ((DATAPTR)arg)->mid;
int num = ((DATAPTR)arg)->num;
char buf[128] = "";
if(num == 1) //拷贝前一半
{
lseek(fd1, 0, SEEK_SET);//偏移读取文件光标
lseek(fd2, 0, SEEK_SET);//偏移写入文件光标
while(1)
{
if(lseek(fd1, 0, SEEK_CUR) == mid)
{
break;
}
int res = read(fd1, buf, sizeof(buf));
write(fd2, buf, res);
}
}
else if(num == 2)//拷贝后一半
{
lseek(fd1, mid, SEEK_SET);//偏移读取文件光标
lseek(fd2, mid, SEEK_SET);//偏移写入文件光标
while(1)
{
int res = read(fd1, buf, sizeof(buf));
if(res == 0)
{
break;
}
write(fd2, buf, res);
}
}
pthread_exit(NULL);
}
int main(int argc, const char *argv[])
{
if(argc != 3)
{
printf("input error\n");
return -1;
}
int fd1 = open(argv[1], O_RDONLY); //定义文件描述符读取文件
int fd2 = open(argv[2], O_WRONLY|O_CREAT|O_TRUNC, 0664);//定义文件描述符写入文件
if(fd1 == -1 || fd2 == -1)
{
perror("fd open error");
}
off_t file_size = lseek(fd1, 0, SEEK_END);//返回文件大小
int mid = file_size/2-file_size/2%128; //找到文件中间位置
DATA file_data1, file_data2;
file_data1.fd1 = fd1;
file_data1.fd2 = fd2;
file_data1.mid = mid;
file_data1.num = 1;
file_data2.fd1 = fd1;
file_data2.fd2 = fd2;
file_data2.mid = mid;
file_data2.num = 2;
pthread_t tid1 = -1, tid2 = -1;//定义变量存储线程号
//创建分支线程1,拷贝文件前一半内容
if(pthread_create(&tid1, NULL, task, &file_data1) != 0)
{
printf("tid1 error\n");
return -1;
}
//创建分支线程2,拷贝文件后一半内容
if(pthread_create(&tid2, NULL, task, &file_data2) != 0)
{
printf("tid2 error\n");
return -1;
}
pthread_join(tid1, NULL);
pthread_join(tid2, NULL);
close(fd1);
close(fd2);
printf("拷贝成功\n");
return 0;
}
思维导图: