如何为已实现基础执行的Shell添加输入输出重定向功能？

看起来你已经搞定了Shell的基础命令解析和执行，接下来要处理输入输出重定向其实没那么复杂，核心就是在子进程执行命令前，修改它的标准输入/输出文件描述符，让它们指向指定的文件。我给你梳理下具体步骤和注意事项：

一、先处理输出重定向（> 或 >>） #

当解析到outfile不为空时，你需要把子进程的标准输出（对应文件描述符STDOUT_FILENO，值为1）替换成目标文件：

• 首先用open()打开目标文件：
  • 对于覆盖写入（>），用参数O_WRONLY | O_CREAT | O_TRUNC，权限设为0644（普通文件的默认权限）。
  • 对于追加写入（>>），把O_TRUNC换成O_APPEND即可。
• 然后用dup2()把打开的文件描述符复制到标准输出的位置，这样后续命令的输出就会直接写入目标文件。
• 记得复制完成后关闭原来的文件描述符，避免资源泄漏。

二、再处理输入重定向（<） #

逻辑和输出重定向类似，只是操作的是标准输入（STDIN_FILENO，值为0）：

• 用open()打开输入文件，参数用O_RDONLY即可。
• 同样用dup2()把文件描述符复制到标准输入的位置，这样命令就会从指定文件读取输入，而不是默认的终端。

三、关键注意事项 #

• 操作时机：一定要在vfork()之后、exec*()之前处理重定向。因为exec*()会替换子进程的整个地址空间，之后修改文件描述符就没用了；而vfork()后子进程共享父进程地址空间，但修改文件描述符是进程级的操作，不会影响父进程。
• 错误处理：打开文件或调用dup2()都可能失败（比如文件不存在、没有权限），一定要在子进程中检查返回值，出错时用perror()打印错误信息，然后用_exit()退出（不要用exit()，因为exit()会刷新缓冲区，可能干扰父进程）。
• 多重重定向：如果同时存在输入和输出重定向，处理顺序不影响结果，因为修改的是不同的文件描述符。
• vfork的坑：子进程里不要做任何会修改父进程数据的操作，比如修改全局变量、调用malloc()，尽量只做重定向和exec*()操作。

示例代码片段 #

这里给你一段参考代码，把重定向逻辑整合到现有的执行流程里：

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

// 假设已解析得到以下变量：
// char *cmd = "ls";
// char *args[] = {"ls", "-l", NULL}; // 注意args必须以NULL结尾
// char *outfile = "directory.txt"; // 可为NULL，代表无输出重定向
// char *infile = NULL; // 可为NULL，代表无输入重定向

int main() {
    pid_t pid = vfork();
    if (pid == -1) {
        perror("vfork failed");
        exit(EXIT_FAILURE);
    } else if (pid == 0) {
        // 子进程：处理输出重定向
        if (outfile != NULL) {
            int fd = open(outfile, O_WRONLY | O_CREAT | O_TRUNC, 0644);
            if (fd == -1) {
                perror("Failed to open output file");
                _exit(EXIT_FAILURE);
            }
            if (dup2(fd, STDOUT_FILENO) == -1) {
                perror("Failed to redirect stdout");
                _exit(EXIT_FAILURE);
            }
            close(fd); // 复制完成后关闭原文件描述符
        }

        // 子进程：处理输入重定向
        if (infile != NULL) {
            int fd = open(infile, O_RDONLY);
            if (fd == -1) {
                perror("Failed to open input file");
                _exit(EXIT_FAILURE);
            }
            if (dup2(fd, STDIN_FILENO) == -1) {
                perror("Failed to redirect stdin");
                _exit(EXIT_FAILURE);
            }
            close(fd);
        }

        // 执行命令
        execvp(cmd, args);
        // 如果execvp返回，说明执行失败
        perror("execvp failed");
        _exit(EXIT_FAILURE);
    } else {
        // 父进程：等待子进程结束
        int status;
        waitpid(pid, &status, 0);
        // 可以根据status判断命令执行结果，比如WIFEXITED(status)和WEXITSTATUS(status)
    }
    return 0;
}

进阶方向 #

如果之后想扩展功能，还可以考虑：

• 支持错误输出重定向（2> 或 2>>），逻辑和标准输出重定向一致，只是操作的是STDERR_FILENO（值为2）。
• 支持文件描述符复制（比如>&、<&），需要解析文件描述符编号，然后用dup2()对应复制。
• 支持管道（|），这需要创建管道，然后把前一个命令的输出重定向到管道写入端，后一个命令的输入重定向到管道读取端，还要处理多进程的同步。

内容的提问来源于stack exchange，提问作者anush95