C:将子进程输出重定向到其他子进程输入和stdout

时间:2015-11-26 14:08:16

标签: c linux process pipe exec

我是一个用C语言编程Linux的新手(我搜索了类似的线程,但没有帮助),所以我遇到了以下问题:

我想在C语言中为Linux创建一个shell(使用fork(),exec(),pipe(),它获取一个带有参数和管道的命令作为来自终端stdin的输入(例如" sort foo) | uniq -c | wc -l"),执行它然后请求下一个命令等。

我分离了不同的命令,它们的参数等,我为每个命令创建了1个子进程,但我不能将每个子进程的输出链接到下一个的输出(以及stdout的最后一个输出)在终端)。

任何人都可以帮助做正确的管道以使其运行起来吗?

如需更多信息,请询问...... 提前致谢

完整代码如下:

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

#define P_READ 0
#define P_WRITE 1

pid_t pID, chID;
char input[100];
char * params[100][100];
char * funcs[100];
char * par;
char * fun;
int i, j, k, stat, infd[2], outfd[2];


//Pipe read
void read_en(int * infd)
{
    dup2(infd[P_READ], STDIN_FILENO);
    close(infd[P_READ]);
    close(infd[P_WRITE]);
}

//Pipe write
void write_en(int * outfd)
{
    dup2(outfd[P_WRITE], STDOUT_FILENO);
    close(outfd[P_READ]);
    close(outfd[P_WRITE]);
}

//Fork, read from pipe, write to pipe, exec   
void fork_chain(int * infd, int * outfd, int i)
{
        pID = fork();

        if (pID == 0)
        {
            if (infd != NULL)
            {
                read_en(infd);
            }

            if (outfd != NULL)
            {
                write_en(outfd);
            }

            execvp(params[i][0], params[i]);
            fprintf(stderr, "Command not found!\n");
            exit(1);
        }
        else if (pID < 0)
        {
            fprintf(stderr, "Fork error!\n");
            exit(1);
        }
        else
        {
            chID = waitpid(-1, &stat, 0);
        }
}


int main()
{
    printf("\n$");
    fgets(input, sizeof(input), stdin);
    strtok(input, "\n");


    while (strcmp(input, "exit") != 0)
    {
        //Separate each command
        k = 0;
        fun = strtok(input, "|");

        while (fun != NULL)
        {
            funcs[k] = fun;
            fun = strtok(NULL, "|");
            k++;
        }


        //Separate each command's parameters
        for (i = 0; i < k; i++)
        {
            j = 0;
            par = strtok(funcs[i], " ");

            while (par != NULL)
            {
                params[i][j] = par;
                par = strtok(NULL, " ");
                j++;
            }
            params[i][j] = NULL;
        }

        //Fork, pipe and exec for each command
        for (i = 0; i < k; i++)
        {
            if (i == 0)
            {
                pipe(outfd);
                fork_chain(NULL, outfd, 0);
                infd[P_READ] = outfd[P_READ];
                infd[P_WRITE] = outfd[P_WRITE];
            }
            else if (i == k-1)
            {
                fork_chain(infd, NULL, 1);
                close(infd[P_READ]);
                close(infd[P_WRITE]);
            }
            else
            {
                pipe(outfd);
                fork_chain(infd, outfd, i);
                close(infd[P_READ]);
                close(infd[P_WRITE]);
                infd[P_READ] = outfd[P_READ];
                infd[P_WRITE] = outfd[P_WRITE];
            }
        }

        //Ask for next input
        printf("\n$");
        fgets(input, sizeof(input), stdin);
        strtok(input, "\n");
    }

    return (0);
}

1 个答案:

答案 0 :(得分:0)

shell进程必须关闭管道的写入端,否则子进程在其输入上不会获得EOF并在调用read()时永远阻塞。您的代码将关闭写入结束,但是在最后一个子进程的情况下为时已晚,因为fork_chain()在返回之前等待它退出(如果管道缓冲区已满,则会出现另一个问题)。如果我们在合适的时间关闭了写入结尾,则close(infd[P_WRITE])必须从read_en()中删除,因为它已经被关闭了。
除此之外,由于

处的代码乘法而出现错误
            else if (i == k-1)
            {
                fork_chain(infd, NULL, 1);

参数1仅在k等于2时才正确,一般情况下必须为i

因此,随着read_en()的更改,主fork-pipe-and-exec循环可以重写为

        for (i = 0; i < k; i++)
        {
            fork_chain(i ? infd : NULL, i < k-1 ? pipe(outfd), outfd : NULL, i);
            if (i)
                close(infd[P_READ]);
            if (i < k-1)
                infd[P_READ] = outfd[P_READ],
                close(outfd[P_WRITE]);  // shell is done with it
        }

(请注意infd[P_WRITE]不再在任何地方使用。)

相关问题