C:将stdin,stdout重定向到管道

时间:2016-01-20 22:27:45

标签: c pipe fork io-redirection

我正在正确理解管道和FD,我正在尝试编写以下内容: 该程序基本上压缩像gzip一样的东西与选项-cf。 基本思路是:

我在父进程中创建了两个管道,然后我将它分叉两次,以便我有两个孩子。在第一个子节点中,我将第一个管道的读取结束重定向到stdin,第二个管道的写入结束到stdout。然后我用-cf选项执行gzip,以便它写入stdout(现在是pipe2的写入结束)

在第二个孩子中,我从pipe2读取并直接输出或将其保存到文件中。 然而,问题是没有数据到达第二个孩子,我不确定为什么。这是代码:

int main(int argc, char **argv) {
    char *file;
    int out = 0;
    if(argc == 2) {
        file = argv[1];
        out = 1;
    } else if (argc > 2) {
        exit(EXIT_FAILURE);         
    }

    int c1pipe[2];
    int c2pipe[2];
    pipe(c1pipe);
    pipe(c2pipe);

    int f;    
    for(int i = 0; i < 2; i++) {
        switch(f = fork()) {
            case 0: //child
                if(i == 0) { //first loop iteration, child 1
                    close(c1pipe[1]); 
                    dup2(c1pipe[0], fileno(stdin));
                    close(c1pipe[0]); 
                    close(c2pipe[0]); 
                    dup2(c2pipe[1], fileno(stdout));
                    close(c2pipe[1]); 
                    execlp("gzip", "gzip", "-cf", (char *) NULL);
                } else if (i == 1) { //second loop iteration, child2
                    close(c1pipe[0]);
                    close(c1pipe[1]);
                    close(c2pipe[1]);
                    FILE *read = fdopen(c2pipe[0], "r");    
                    char buffer[1024];

                    if(out == 0) { //output to stdout
                        while(fgets(buffer, 1024, read) != NULL) {
                            fprintf(stdout, "%s", buffer);
                            fflush(stdout);
                        }
                    } else { //write to specified file
                        FILE *writeto = fopen(file, "w");
                        while(fread(buffer, sizeof(char), strlen(buffer)+1, read) > 0) {
                            fwrite(buffer, sizeof(char), strlen(buffer)+1, writeto);
                            fflush(writeto);
                        }
                        fclose(writeto);
                    }

                    close(c2pipe[0]);
                    fclose(read);
                }           
            break;
            case -1: //err
                //not implemented
            break;
            default: //parent
                if(i == 0) {
                    close(c2pipe[0]);
                    close(c2pipe[1]);
                    close(c1pipe[0]);
                    FILE *writer;
                    writer = fdopen(c1pipe[1], "w");
                    char buffer[1024];
                    while(fgets(buffer, sizeof buffer, stdin) != NULL) {
                        fwrite(buffer, sizeof (char), strlen(buffer)+1, writer);
                    }
                    close(c1pipe[1]);
                    fclose(writer);
                }
            break;
        }
    }

    return 0;
}

请原谅我丢失的错误处理,因为我想创建一个快速而肮脏的版本。

感谢任何帮助。

1 个答案:

答案 0 :(得分:1)

在父进程中,在分叉第二个子进程之前,你正在关闭c2pipe的两端。

如果你对任何读/写调用都进行了任何错误处理,你可能已经想到了这一点。实际上,如果您检查了dup2调用上的错误,然后查看了errno,您可能会发现它是EBADF(错误的文件描述符)。

另一个问题是您的父进程在知道两个子进程都已完成之前退出。这意味着子进程将收到一个信号并自行终止。父母需要调用wait()的一个变体来确保两个孩子都离开了。