linux 实现cp命令递归复制

#include<func.h>

#define LENGTH 4096

int DFS_copy(char* path1, char* path2);  //递归复制

int main(int argc, char* argv[]) {
    ARGS_CHECK(argc, 3); //检查参数是否错误

    struct stat statbuf;
    stat(argv[1], &statbuf);
    if (S_ISDIR(statbuf.st_mode)) { //若参数1是目录，直接进行递归
        DFS_copy(argv[1], argv[2]);

    }
    else { //否则参数1就是普通文件
        stat(argv[2], &statbuf);
        if (S_ISDIR(statbuf.st_mode)) { //若参数2是目录
            char newpath1[1024] = { 0 };
            char newpath2[1024] = { 0 };
            sprintf(newpath1, "%s", argv[1]);//重建待复制的源文件的地址
            sprintf(newpath2, "%s%s%s", argv[2], "/", argv[1]);//重建目的文件的地址

            int fdr = open(newpath1, O_RDONLY); //只读方式打开源文件
            ERROR_CHECK(fdr, -1, "open fdr");
            int fdw = open(newpath2, O_WRONLY | O_CREAT | O_TRUNC, 0666);//只写方式打开目的文件,若不存在就创建,给予权限0666
            ERROR_CHECK(fdw, -1, "open fdw");
            char buf[4096] = { 0 }; //数据缓存
            while (1) {
                memset(buf, 0, sizeof(buf)); //每次清空内容
                int ret = read(fdr, buf, sizeof(buf)); //读内容，返回成功读到的字节数
                if (ret == 0) {
                    break;
                }
                write(fdw, buf, ret); //写入到目的文件
            }
            close(fdr); //关闭文件
            close(fdw);
        }
        else { //若参数2是普通文件,则就是两个普通文件的复制
            int fdr = open(argv[1], O_RDONLY); //只读方式打开源文件
            ERROR_CHECK(fdr, -1, "open fdr");
            int fdw = open(argv[2], O_WRONLY | O_CREAT | O_TRUNC, 0666);//只写方式打开目的文件,若不存在就创建,给予权限0666
            ERROR_CHECK(fdw, -1, "open fdw");
            char buf[4096] = { 0 }; //数据缓存
            while (1) {
                memset(buf, 0, sizeof(buf)); //每次清空内容
                int ret = read(fdr, buf, sizeof(buf)); //读内容，返回成功读到的字节数
                if (ret == 0) {
                    break;
                }
                write(fdw, buf, ret); //写入到目的文件
            }
            close(fdr); //关闭文件
            close(fdw);
        }

    }

    return 0;
}

int DFS_copy(char* path1, char* path2) {
    DIR* pdir1 = opendir(path1); //打开源目录流
    ERROR_CHECK(pdir1, NULL, "opendir1"); //检查返回值是否出错
    if (NULL == opendir(path2)) { //打开目的文件的目录流,若不是目录就创建文件夹
        int ret = mkdir(path2, 0777); //创建目的文件夹
        ERROR_CHECK(ret, -1, "mkdir");
    }
    DIR* pdir2 = opendir(path2); //打开目的文件夹的目录流
    ERROR_CHECK(pdir2, NULL, "opendir2"); //检查返回值是否出错
    struct dirent* pdirent; //存储目录项信息
    while ((pdirent = readdir(pdir1)) != NULL) {  //遍历所有目录项
        //过滤.和..目录，不然复制的时候会死循环
        if (strcmp(pdirent->d_name, ".") == 0 || strcmp(pdirent->d_name, "..") == 0) {
            continue;
        }

        char newpath1[1024] = { 0 };
        char newpath2[1024] = { 0 };
        sprintf(newpath1, "%s%s%s", path1, "/", pdirent->d_name);//重建待复制的源文件的地址
        sprintf(newpath2, "%s%s%s", path2, "/", pdirent->d_name);//重建目的文件的地址
        if (pdirent->d_type == DT_DIR) { //判断当前目录项是否是个目录
            DFS_copy(newpath1, newpath2); //是目录就进行递归复制
        }
        else { //不是目录就直接复制
        //------------------read/write方式复制文件------------------
            // int fdr = open(newpath1, O_RDONLY); //只读方式打开源文件
            // ERROR_CHECK(fdr, -1, "open fdr");
            // int fdw = open(newpath2, O_WRONLY | O_CREAT | O_TRUNC, 0666);//只写方式打开目的文件,若不存在就创建,给予权限0666
            // ERROR_CHECK(fdw, -1, "open fdw");
            // char buf[4096] = { 0 }; //数据缓存
            // while (1) {
            //     memset(buf, 0, sizeof(buf)); //每次清空内容
            //     int ret = read(fdr, buf, sizeof(buf)); //读内容，返回成功读到的字节数
            //     if (ret == 0) {
            //         break;
            //     }
            //     write(fdw, buf, ret); //写入到目的文件
            // }
            // close(fdr); //关闭文件
            // close(fdw);
        //--------------------------------------------------------
        //------------------mmap方式复制文件-----------------------
            int fdr = open(newpath1, O_RDONLY);
            ERROR_CHECK(fdr, -1, "open fdr");
            int fdw = open(newpath2, O_RDWR | O_CREAT | O_TRUNC, 0666);
            ERROR_CHECK(fdw, -1, "open fdw");

            struct stat statbuf;
            fstat(fdr, &statbuf); //和stat函数一样,只是参数变为文件描述符
            ftruncate(fdw, statbuf.st_size);
            off_t fsize = 0;
            int cp_size;
            while (fsize < statbuf.st_size) {
                if (statbuf.st_size - fsize >= LENGTH) {
                    cp_size = LENGTH;
                }
                else {
                    cp_size = statbuf.st_size - fsize;
                }

                char* src = (char*)mmap(NULL, cp_size, PROT_READ, MAP_SHARED, fdr, fsize);
                ERROR_CHECK(src, MAP_FAILED, "mmap");

                char* dest = (char*)mmap(NULL, cp_size, PROT_WRITE | PROT_READ, MAP_SHARED, fdw, fsize);
                ERROR_CHECK(dest, MAP_FAILED, "mmap");

                memcpy(dest, src, cp_size);
                // for(int i = 0; i < cp_size; i++){  //另一种转移数据的方式
                //     dest[i] = src[i];
                // }
                munmap(src, cp_size);
                munmap(dest, cp_size);
                fsize += cp_size;
            }

            //--------------------------------------------------------
        }
    }
    closedir(pdir1); //关闭目录流
    closedir(pdir2);
    return 0;
}