使用pipe和dup2重定向标准输入输出时出现Broken pipe问题

你在尝试用C语言复现ps eaux | grep "^USERNAME" | wc -l这个管道命令的功能时遇到了Broken pipe错误，先看看你给出的不完整代码片段：

#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>
int main(int argc, char * argv[]) {
    int i;
    int pipefd1[2];
    int pipefd2[2];
    pipe(pipefd1);
    pipe(pipefd2);
    for (i=0; i<2; i++) {
        switch (fork()) {
            case -1: perror("F...");

问题根源分析 #

Broken pipe错误通常发生在往一个已经没有读端的管道写入数据时。你的代码大概率存在这些问题：

• 没有正确关闭每个进程不需要的管道描述符，导致管道的引用计数不为0，读写端状态异常
• 只fork了2个子进程，但原命令需要3个进程（ps、grep、wc）
• 没有处理exec失败的情况，子进程会继续执行父进程的代码，干扰管道状态
• 没有将标准输入/输出正确重定向到对应的管道两端

完整的正确实现代码 #

下面是可以正常工作的代码，我会逐部分解释细节：

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

int main(int argc, char *argv[]) {
    // 确定要过滤的用户名：优先用命令行参数，否则读取环境变量USER
    const char *username = argc > 1 ? argv[1] : getenv("USER");
    if (!username) {
        fprintf(stderr, "无法获取用户名\n");
        return 1;
    }
    // 构造grep的匹配模式，比如"^username"
    char grep_pattern[256];
    snprintf(grep_pattern, sizeof(grep_pattern), "^%s", username);

    int pipefd1[2]; // ps -> grep 的管道
    int pipefd2[2]; // grep -> wc 的管道

    // 创建管道，检查是否成功
    if (pipe(pipefd1) == -1 || pipe(pipefd2) == -1) {
        perror("创建管道失败");
        return 1;
    }

    // 忽略SIGPIPE信号，避免进程因管道写入失败而意外终止（可选但更健壮）
    signal(SIGPIPE, SIG_IGN);

    // 第一个子进程：执行ps eaux
    pid_t pid1 = fork();
    if (pid1 == -1) {
        perror("fork ps进程失败");
        return 1;
    } else if (pid1 == 0) {
        // 关闭不需要的管道描述符：pipefd1的读端、pipefd2的所有端
        close(pipefd1[0]);
        close(pipefd2[0]);
        close(pipefd2[1]);

        // 将stdout重定向到pipefd1的写端
        dup2(pipefd1[1], STDOUT_FILENO);
        close(pipefd1[1]); // 重定向后关闭原描述符，避免资源泄漏

        // 执行ps命令
        execlp("ps", "ps", "eaux", NULL);
        // 如果execlp返回，说明执行失败
        perror("执行ps失败");
        exit(EXIT_FAILURE);
    }

    // 第二个子进程：执行grep
    pid_t pid2 = fork();
    if (pid2 == -1) {
        perror("fork grep进程失败");
        return 1;
    } else if (pid2 == 0) {
        // 关闭不需要的管道描述符：pipefd1的写端、pipefd2的读端
        close(pipefd1[1]);
        close(pipefd2[0]);

        // 将stdin重定向到pipefd1的读端，stdout重定向到pipefd2的写端
        dup2(pipefd1[0], STDIN_FILENO);
        close(pipefd1[0]);
        dup2(pipefd2[1], STDOUT_FILENO);
        close(pipefd2[1]);

        // 执行grep命令
        execlp("grep", "grep", grep_pattern, NULL);
        perror("执行grep失败");
        exit(EXIT_FAILURE);
    }

    // 第三个子进程：执行wc -l
    pid_t pid3 = fork();
    if (pid3 == -1) {
        perror("fork wc进程失败");
        return 1;
    } else if (pid3 == 0) {
        // 关闭不需要的管道描述符：pipefd1的所有端、pipefd2的写端
        close(pipefd1[0]);
        close(pipefd1[1]);
        close(pipefd2[1]);

        // 将stdin重定向到pipefd2的读端
        dup2(pipefd2[0], STDIN_FILENO);
        close(pipefd2[0]);

        // 执行wc命令
        execlp("wc", "wc", "-l", NULL);
        perror("执行wc失败");
        exit(EXIT_FAILURE);
    }

    // 父进程：关闭所有管道描述符，等待所有子进程结束
    close(pipefd1[0]);
    close(pipefd1[1]);
    close(pipefd2[0]);
    close(pipefd2[1]);

    // 等待三个子进程退出，回收资源
    waitpid(pid1, NULL, 0);
    waitpid(pid2, NULL, 0);
    waitpid(pid3, NULL, 0);

    return 0;
}

关键注意事项 #

1. 关闭无用的管道描述符：每个进程只保留自己需要的管道端，比如ps进程只需要往pipefd1写，所以要关闭pipefd1的读端和pipefd2的所有端。如果不关闭，管道的读/写端不会被正确标记为关闭，可能导致其他进程误判管道状态。
2. 重定向后关闭原描述符：用dup2重定向标准输入/输出后，必须关闭原来的管道描述符，避免资源泄漏。
3. 处理exec失败：如果execlp返回，说明命令执行失败，必须调用exit终止子进程，否则子进程会继续执行父进程的代码，破坏整个流程。
4. 忽略SIGPIPE：如果某个子进程提前退出（比如grep没找到匹配内容就退出），后续进程往管道写数据会触发SIGPIPE信号，默认会终止进程。忽略这个信号可以让进程继续执行，或者你也可以自定义信号处理逻辑。

内容的提问来源于stack exchange，提问作者Pop Flamingo