在Linux上实现自定义的 cp命令

二、编写cp命令

1. 阅读联机帮助

将 SOURCE 复制到 DEST，或将多个 SOURCE(s) 复制到 DIRECTORY。 长选项的强制性参数对于短选项也是强制性的。

也就是说，cp命令是用来复制文件/目录的。

基本用法：cp 源文件 目标文件
如果目标文件不存在，cp就创建这个文件；如果存在就覆盖。

2. cp的工作原理

很容易猜到，cp命令无非就是从一个文件中读数据然后写入另外的文件中，与 who命令 从文件中读数据然后以一定的个数输出类似。

1. 打开 源文件
2. 创建 目标文件
3. 从 源文件中读取数据 并写入 目标文件 中
4. 关闭文件

3. 如何编写cp

首先编写一个最基本的，只能复制1个文件的cp命令

#include <stdio.h>
#include <stdlib.h>     // exit()
#include <unistd.h>     // write() read() close()
#include <fcntl.h>      // create()

#define BUFFERSIZE 4096
#define COPYMODE 0644

void show_error(char*, char*);

// argc记录程序启动时就传入的参数个数，argv[]记录每个实参的地址
int main(int argc, char* argv[]) {
    int source_fd, dest_fd, n_chars;
    char buf[BUFFERSIZE];   // 缓冲区

    if (argc < 3) {
        // printf("xxxx")其实就是向stdout中输出，等同于fprintf(stdout，"xxxx");
        printf("传入参数有误，应输入: %s source destination\n", argv[0]);
        exit(1);
    }

    // 打开源文件
    if ((source_fd = open(argv[1], O_RDONLY)) == -1) {
        show_error("Cannot open", argv[1]);
    }

    // 创建目标文件，在早期的Unix实现中，open()只有两个参数，无法创建新文件，而是使用creat()创建
    if ((dest_fd = creat(argv[2], COPYMODE)) == -1) {
        show_error("Cannot create", argv[2]);
    }

    // 写入数据
    while ((n_chars = read(source_fd, buf, BUFFERSIZE)) > 0) {
        if (write(dest_fd, buf, n_chars) != n_chars) {
            show_error("Write error to", argv[2]);
        }
    }
    if (n_chars == -1) {
        show_error("Read error from", argv[1]);
    }

    if (close(source_fd) == -1 || close(dest_fd) == -1) {
        show_error("Error close files", "");
    }

    return 0;
}

void show_error(char* s1, char* s2) {
    printf("Error: %s ", s1);
    perror(s2);
    exit(1);
}

运行结果：

当输入参数不够时：

输入错误参数时：

以上仅仅完成了最基本的 cp功能，很多部分还需继续改进优化。