进程间通信系列（3）匿名管道范例代码

r囧r小猫 2022-07-03 12:17 222阅读 0赞

范例程序通过父进程创建两个子进程，两个子进程通过管道进行通信，实现了cat /passwd | grep root的功能。

#include <unistd.h>
    #include <stdlib.h>
    #include <stdio.h>
    #include <sys/types.h>
    #include <wait.h>
    
    
    char *cmd1[3] = {"/bin/cat", "/etc/passwd", NULL};
    char *cmd2[3] = {"/bin/grep", "root", NULL};
    
    int main()
    {
        int fd[2] = {0};
        if(pipe(fd) < 0)
        {
            perror("pipe error");
            exit(1);
        }
    
        int i = 0;
        pid_t pid;
        for(; i < 2; i++)
        {
           if(pid < 0)
           {
               perror("fork error");
               exit(0);
           } 
           else if(pid == 0)//child process
           {
                if(0 == i)//第一个子进程，负责往管道写入数据
                {
                    //关闭读端
                    close(fd[0]);
    
                    /*
                     *将标准输出重定向到管道的写端
                     *(下面命令执行的结果会写入到管道中
                     *而不是输出到屏幕)
                     */
                    if(dup2(fd[1], STDOUT_FILENO) != STDOUT_FILENO)
                    {
                        perror("dup2 error");
                        exit(1);
                    }
                    close(fd[1]);//因为被复制，可以直接关闭了
    
                    //调用exec函数执行cat命令
                    if(execvp(cmd1[0], cmd1) < 0)
                    {
                        perror("execvp error");
                        exit(1);
                    }
    
                    break;
                }
                else if(1 == i)//第二个子进程，负责从管道读取数据
                {
                    //关闭写端，负责向管道写入数据
                    close(fd[1]);
                    
                    /*
                     *将标准输入重定向到管道的读端
                     *下面的命令grep的执行是从管道的读端
                     *读取内容，而不是从标准输入读取
                     */                
                    if(dup2(fd[0], STDIN_FILENO) != STDIN_FILENO)
                    {
                        perror("dup2 error");
                        exit(1);
                    }
                    close(fd[0]);
    
    
                    //调用exec函数执行grep命令
                    if(execvp(cmd2[0], cmd2) < 0)
                    {
                        perror("execvp error");
                        exit(1);
                    }
    
                    break;
                }
           }
           else//parent process
           {
                if(1 == i)
                {
                    close(fd[0]);
                    close(fd[1]);
    
                    //父进程要等到子进程全部创建完毕才去回收
                    wait(0);
                    wait(0);
                }
           }
        }
    
        exit(0);
    }