c - 编写我自己的 linux shell I/O 重定向 '>' 函数

Question

我正在编写将命令输出写入给定文件名的重定向函数。

例如:

echo Hello World > hello.txt 会将“Hello World”写入 hello.txt。

ls -al > file_list.txt 会将当前目录中所有文件/目录名称的列表写入 file_list.txt。

到目前为止，我的函数定义为:

int my_redirect(char **args, int count) {
    if (count == 0 || args[count + 1] == NULL) {
        printf("The redirect function must follow a command and be followed by a target filename.\n");
        return 1;
    }
    char *filename = args[count + 1];

    //Concatenates each argument into a string separated by spaces to form the command
    char *command = (char *) malloc(256);
    for (int i = 0; i < (count); i++) {
        if (i == 0) {
            strcpy(command, args[i]);
            strcat(command, " ");
        }
        else if (i == count - 1) {
            strcat(command, args[i]);
        }
        else {
            strcat(command, args[i]);
            strcat(command, " ");
        }
    }

    //command execution to file goes here

    free(command);
    return 1;
}

其中 args[count] 是 ">"。

如何将字符串从 args[0] 到 args[count - 1] 执行到 args[count 处给出的文件中+ 1]?

编辑

这些是我们得到的指示:

“通过将 stdout 的重定向添加到文件来改进您的 shell。仅在完成功能 1 后尝试。解析 > 的行，将前面的所有内容作为命令，并将后面的第一个单词作为文件名(忽略 <, >> , | 等)。

标准输出被写到文件描述符 1(stdin 为 0，stderr 为 2)。所以这个任务可以通过打开文件，并使用 dup2 系统调用将它的文件描述符复制到 1 来完成。

int f = open( filename , O_WRONLY|O_CREAT|O_TRUNC, 0666) ;
dup2( f , 1 ) ;

注意:这里使用系统调用 open 而不是库包装器 fopen。"

Answer 1

如果允许您以特殊方式解决此问题，那么它只适用于范围狭窄的问题，例如将命令的标准输出捕获到文件中，您可以避免使用 popen() 重新发明轮子来自 <stdio.h> 的函数.

程序示意图:

1. 确定输出文件名
2. 打开输出文件进行写入
3. 确定命令和参数
4. 构造从参数到 > 的命令字符串.
5. 调用FILE *cmd = popen(command, "r");
6. 从 cmd 读取行流，写入输出文件
7. 转到 6 而不是 cmd 上的 EOF流。
8. pclose(cmd) , fclose输出流

仅当您的导师不希望您使用 fork、dup 和 friends 时才这样做。