1

我必须用 C (Unix) 编写程序,该程序捕获命令行参数中传递的命令的标准输入、标准输出和标准错误。

例如,

./capture echo Hi
== stdout ===
Hi
== end of stdout ==
./capture bc -q
== stdin ==
1+2
== end of stdin ==
== stdout ==
3
== end of stdout ==

我已经为 stdout 和 stderr 实现了这样的行为:

int pipe_out[2];
int pipe_in[2];
int finished = 0;

void sig_handler(int signo)
{
    if (signo == SIGCHLD)
    {
        finished = 1;
        wait(NULL);
    }
}

void ChildProcess (char * argv[], char * env[])
{

    dup2(pipe_out[1], 1);
    dup2(pipe_out[1], 2);
    dup2(pipe_in[0], 0);

    close(pipe_out[0]);
    close(pipe_out[1]);

    close(pipe_in[0]);
    close(pipe_in[1]);

    int return_code = execvpe(argv[0], argv, env);
    printf("This should not appear: %d\n", return_code);
    printf("Error: %s\n\n", strerror(errno));
}   

void ParentProcess(pid_t pid)
{
    int status;
    char buf[10000];

    fd_set out_fds;
    fd_set in_fds;

    signal(SIGCHLD, sig_handler);

    do
    {
        FD_ZERO(&out_fds);
        FD_ZERO(&in_fds);
        FD_SET(pipe_in[1], &in_fds); 
        FD_SET(pipe_out[0], &out_fds);
        int cnt = select(max(pipe_out[0], pipe_out[0]) + 1, &out_fds, NULL, NULL, NULL);
        if (cnt > 0 && FD_ISSET(pipe_out[0], &out_fds))
        {
            puts("== stdout ==");
            read(pipe_out[0], buf, sizeof(buf));
            printf("%s", buf);
            puts("== end of stdout == ");
            continue;
        }
        if (0 && FD_ISSET(pipe_in[1], &in_fds))
        {
            puts("== stdin ==");
            write(pipe_in[1], "1+2\n", sizeof("1+2\n"));
            puts("== end of stdin ==");
            continue;
        }
    } while (!finished);
}   

据我了解,它以这种方式工作:父进程调用 select() 并等待子进程的输出。之后,父进程打印附加信息(== stdout ==)和子进程的输出。

当我尝试为标准输入实现相同的行为时,我发现它in_fds总是独立于子进程需要什么。所以capture总是从标准输入读取。即使子进程需要输出一些东西。

所以我的问题是如何实现这种行为?

我认为必须有一个特殊信号指示子进程何时需要一些输入,但我找不到任何东西。

4

0 回答 0