（在 C 中模擬 UNIX SHELL）如何在 for 循環中實現多個管道？

Question

我正在嘗試在 C 程序中模擬 unix shell，但它仍處於開始階段，並且最多只能用於兩個管道。 我有一個命令向量 (char *com[3][3])，考慮到字符“|”將它們分開，但我的問題是如何在 for 循環中處理更多管道？ 在下面的當前實現中，我嘗試執行由管道分隔的 3 個命令：

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

int main(int argc, char **argv){

    //Vector with positions of pipes found, position 0 reserved for the total amount of commands.

    char* com[3][3] = { { "/bin/ls", "-la", 0 },
        { "/bin/grep", ".", 0}, { "/usr/bin/wc", "-l", 0 }};

    //EXECUTE COMMANDS
    pid_t fork1, fork2, fork3;
    int fd1[2], fd2[2];

    if(pipe(fd1) < 0){
        perror("pipe1");
    }
    if(pipe(fd2) < 0){
        perror("pipe2");
    }



    //COMMAND 1
    fork1 = fork();
    if(fork1 == 0){
        dup2(fd1[1], STDOUT_FILENO);
        close(fd1[0]);
        close(fd2[0]);
        close(fd2[1]);
        execvp(com[0][0], com[0]);
        perror("execvp 1");
        exit(EXIT_FAILURE);
    }

    //COMMAND 2
    fork2 = fork();
    if(fork2 == 0){
        dup2(fd1[0], STDIN_FILENO);
        dup2(fd2[1], STDOUT_FILENO);
        close(fd1[1]);
        close(fd2[0]);
        execvp(com[1][0], com[1]);
        perror("execvp 2");
        exit(EXIT_FAILURE);
    }

    //COMMAND 3
    fork3 = fork();
    if(fork3 == 0){
        dup2(fd2[0], STDIN_FILENO);
        close(fd2[1]);
        close(fd1[0]);
        close(fd1[1]);
        execvp(com[2][0], com[2]);
        perror("execvp 3");
        exit(EXIT_FAILURE);
    }

    close(fd1[0]);
    close(fd1[1]);
    close(fd2[0]);
    close(fd2[1]);

    waitpid(-1, NULL, 0);
    waitpid(-1, NULL, 0);
    waitpid(-1, NULL, 0);

    return 0;
}

如何在 for 循環中生成 com[n][3]？

Answer 1

“迭代是人類，遞歸是神聖的”——Anon。

我會用遞歸方法來解決這個問題。 這是成為三星級程序員幾乎是合理的非常罕見的情況之一。 ;)

這是完全未經測試的，但應該讓您指向正確的方向。

// You'll need to rearrange your command strings into this three dimensional array
// of pointers, but by doing so you allow an arbitrary number of commands, each with
// an arbitrary number of arguments.
int executePipe(char ***commands, int inputfd)
{
    // commands is NULL terminated
    if (commands[1] == NULL)
    {
        // If we get here there's no further commands to execute, so run the 
        // current one, and send its result back.
        pid_t pid;
        int status;
        if ((pid = fork()) == 0)
        {
            // Set up stdin for this process.  Leave stdout alone so output goes to the
            // terminal.  If you want '>' / '>>' redirection to work, you'd do that here
            if (inputfd != -1)
            {
                dup2(inputfd, STDIN_FILENO);
                close(inputfd);
            }
            execvp(commands[0][0], commands[0]);
            perror("execvp");
            exit(EXIT_FAILURE);
        }
        else if (pid < 0)
        {
            perror("fork");
            exit(EXIT_FAILURE);
        }
        waitpid(pid, &status, 0);
        return status;
    }
    else
    {
        // Somewhat similar to the above, except we also redirect stdout for the
        // next process in the chain
        int fds[2];
        if (pipe(fds) != 0)
        {
            perror("pipe");
            exit(EXIT_FAILURE);
        }
        pid_t pid;
        int status;
        if ((pid = fork()) == 0)
        {
            // Redirect stdin if needed
            if (inputfd != -1)
            {
                dup2(inputfd, STDIN_FILENO);
                close(inputfd);
            }
            dup2(fds[1], STDOUT_FILENO);
            close(fds[1]);
            execvp(commands[0][0], commands[0]);
            perror("execvp");
            exit(EXIT_FAILURE);
        }
        else if (pid < 0)
        {
            perror("fork");
            exit(EXIT_FAILURE);
        }
        // This is where we handle piped commands.  We've just executed
        // commands[0], and we know there's another command in the chain.
        // We have everything needed to execute that next command, so call
        // ourselves recursively to do the heavy lifting.
        status = executePipe(++commands, fds[0]);
        // As written, this returns the exit status of the very last command
        // in the chain.  If you pass &status as the second parameter here
        // to waitpid, you'll get the exit status of the first command.
        // It is left as an exercise to the reader to figure how to get the
        // the complete list of exit statuses
        waitpid(pid, NULL, 0);
        return status;
    }
}

要使用它，首先使用如所述設置的commands數組調用它，並inputfd最初-1 。

如果您想處理<類型重定向，您可能需要在最頂部檢查inputfd == -1 ，如果需要進行重定向，並在進入正文的其余部分之前將inputfd替換為適當的值。