多线程实现文件的拷贝

#include <stdio.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <sys/mman.h>
#include <memory.h>
#include <sys/wait.h>
#include <pthread.h>
#include <stdlib.h>

int file_size(char* filename)
{
    struct stat statbuf;
    stat(filename, &statbuf);
    int size = statbuf.st_size;
    return size;

}

int size = 0;
int each_size = 0;
int last_size = 0;
char* s_ptr;
char* o_ptr;
void* pthread_copy(void* arg)
{
    int i = (int)arg;

    if (i == 4)
    {
        sleep(i);
        memcpy(o_ptr + i * each_size, s_ptr + i * each_size, each_size);
    }
    else
    {
        sleep(i);
        memcpy(o_ptr + i * each_size, s_ptr + i * each_size, each_size);
    }
    pthread_exit((void*)1);
}


int main(int argc, char* argv[])
{
    //参数一:要拷贝的文件路径名
    if (argv[1] == "")
    {
        printf("参数不能为空");
        exit(1);
    }
    //参数二:目标文件路径名
    if (argv[2] == "")
    {
        printf("参数不能为空");
        exit(1);
    }
    

    int n = 5;//创建子进程的个数
    int s_ret = 0, o_ret = 0;
    int i = 0;
    //打开源文件,只读模式
    s_ret = open(argv[1], O_RDONLY);
    if (s_ret == -1)
    {
        perror("file error:");
        exit(1);
    }
    //目标模式,没有则创建,只写模式
    o_ret = open(argv[2], O_CREAT | O_RDWR, 0644);
    if (o_ret == -1)
    {
        perror("file error:");
        exit(1);
    }
    //获取文件大小
    size = file_size(argv[1]);
    //printf("file size: %d   \n",size);
    //扩展目标文件
    ftruncate(o_ret, size);

    //内存映射
    s_ptr = (char*)mmap(NULL, size, PROT_READ, MAP_PRIVATE, s_ret, 0);
    o_ptr = (char*)mmap(NULL, size, PROT_READ | PROT_WRITE, MAP_SHARED, o_ret, 0);
    if (o_ptr == MAP_FAILED)
    {
        perror("s_ptr mmap error:");
        exit(1);
    }
    //关闭文件描述符
    close(s_ret);
    close(o_ret);
    //实现拷贝
    each_size = size / n;
    last_size = size - (n * each_size);
    pid_t pid;
    pthread_t tid;
    //循环创建子线程
    for (i = 0; i < n; i++)
    {
        int ret=pthread_create(&tid,NULL, pthread_copy,(void*)i);
        if (ret != 0)
        {
            fprintf(stderr, "pthread_create error :%s\n", strerror(ret));
            exit(1);
        }

        ret = pthread_join(tid, NULL);
        if (ret != 0)
        {
            fprintf(stderr, "pthread_join error :%s\n", strerror(ret));
            exit(1);
        }
    }


    //释放映射空间
    munmap(s_ptr, size);
    munmap(o_ptr, size);
    return 0;
}

结果:
在这里插入图片描述

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

chenshida_

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值