从零实现高效文件复制工具：命令行参数解析与缓冲区读写优化

1次阅读

共计 1770 个字符，预计需要花费 5 分钟才能阅读完成。

在实际开发中，我们经常需要处理文件复制操作。虽然 Linux 系统自带的 cp 命令能满足基本需求，但在某些场景下存在明显不足：

权限控制不灵活 ：cp 命令会默认保留源文件权限，而我们需要精确控制目标文件权限（如 0644）
特殊处理需求：当目标文件存在时需要清空内容（O_TRUNC），而非追加写入
集成复杂度：在应用程序中直接调用系统命令存在安全风险和性能开销

使用 getopt 规范处理参数，这是 POSIX 标准推荐的做法：

#include <unistd.h>

int main(int argc, char *argv[]) {
    int opt;
    char *src = NULL, *dst = NULL;

    while ((opt = getopt(argc, argv, "s:d:")) != -1) {switch (opt) {
        case 's': src = optarg; break;
        case 'd': dst = optarg; break;
        default: 
            fprintf(stderr, "Usage: %s -s src -d dst\n", argv[0]);
            return 1;
        }
    }

    if (!src || !dst) {fprintf(stderr, "Missing source or destination\n");
        return 1;
    }
    // ...
}

关键标志位组合：

O_WRONLY：只写模式
O_CREAT：文件不存在时创建
O_TRUNC：文件存在时清空

配合权限位 0644（所有者可读写，组和其他人只读）：

int fd_dst = open(dst, O_WRONLY|O_CREAT|O_TRUNC, 0644);
if (fd_dst == -1) {perror("open dst");
    return 1;
}

最佳实践是使用文件系统的块大小（通过 stat 获取）：

struct stat st;
if (stat(src, &st) == -1) {perror("stat");
    return 1;
}

size_t buf_size = st.st_blksize;
char *buf = malloc(buf_size);

#include <fcntl.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>

int main(int argc, char *argv[]) {
    // 参数解析（同上）// 打开源文件
    int fd_src = open(src, O_RDONLY);
    if (fd_src == -1) {perror("open src");
        return 1;
    }

    // 获取最佳缓冲区大小
    struct stat st;
    if (fstat(fd_src, &st) == -1) {perror("fstat");
        close(fd_src);
        return 1;
    }

    // 内存分配（推荐堆分配应对大文件）char *buf = malloc(st.st_blksize);
    if (!buf) {perror("malloc");
        close(fd_src);
        return 1;
    }

    // 复制循环
    ssize_t n;
    while ((n = read(fd_src, buf, st.st_blksize)) > 0) {if (write(fd_dst, buf, n) != n) {perror("write");
            break;
        }
    }

    // 资源清理
    free(buf);
    close(fd_src);
    close(fd_dst);
    return 0;
}