管道第二次卡住

问题描述

我使用 pipe 和 dup2 来重定向文件描述符,但是在第一个命令之后,第二个 execvp 卡住了,我使用 for 循环来循环多个命令。我很确定标记化和缓冲区的东西有效,如果我运行第一个命令,它有效,但对于第二个命令,它只是在第一个命令之后卡住了。

#include <fstream>
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/wait.h>
#include <fcntl.h>

using namespace std;

char buffer[10][500];
char *args[50];
int fd[20];

void tokenFnc(int counter);

//int counter = 0;
int main(int argc,char *argv[])
{

    string input;
    cout << "myshell$";
    getline(cin,input);
    int k = 0;
    int counter = 0;
    int status_code;

    for (unsigned i = 0; i <= input.size(); i++)
    {
        if (i > 0 && input[i] == ' ' && input[i - 1] == '|')
        {
            continue;
        }
        if (input[i] == '|')
        {
            counter++;
            k = 0;
        }
        else
        {
            buffer[counter][k] = input[i];
            k++;
        }
    }

    //  int fd[(counter-1)*2];
    for (int i = 0; i < counter; i++)
    {
        pipe(fd + i * 2);
    }

    counter += 1; //how many commands

    for (int i = 0; i < counter; i++)
    {
        if (pipe(fd) == -1)
        {
            perror("pipe Failed");
            return 1;
        }
        tokenFnc(i);

        int rc = fork();
        if (rc < 0)
        { // fork creation fail
            perror("fork Failed\n");
            return 1;
        }
        if (rc == 0)
        { //child process
            if (i == 0 && counter > 1)
                dup2(fd[1],1);

            if (i > 0 && i != counter - 1)
            {
                dup2(fd[(i - 1) * 2],0);
                dup2(fd[(2 * i) + 1],1);
            }
            if (i == counter - 1 && counter > 1)
                dup2(fd[(i - 1) * 2],0);

            for (int k = 0; k < (counter - 1) * 2; k++)
            {
                close(fd[k]);
            }

            status_code = execvp(args[0],(char *const *)args);
            if (status_code == -1)
            { //error handling
                perror("Child process terminated fail");
                return 1;
            }
        }
    }
    int status;
    for (int i = 0; i < counter; i++)
    {
        pid_t pid = wait(&status);
        if (pid == -1)
        {
            perror("wait exective fail \n");
            return 1;
        }
        printf("process %i exits with %i\n",pid,status);
    }
    return 0;
}

void tokenFnc(int counter)
{
    int tokenCount = 0;
    char delim[] = " ";
    // cout << counter << endl;

    // cout << "The token are: " << endl;
    char *token = strtok(buffer[counter],delim);
    while (token)
    {
        //   cout << token << " ";
        token = strtok(NULL,delim);
        tokenCount++;
    }
    // cout << endl;

    int pos = 0;
    for (int i = 0; i < tokenCount; i++)
    {
        args[i] = (char *)&buffer[counter][pos];
        pos += strlen(args[i]) + 1;
        // cout << args[i] << ",";
    }
    args[tokenCount] = (char *)nullptr;
    // cout << endl;
}

解决方法

        for (int k = 0; k < (counter - 1) * 2; k++)
        {
            close(fd[k]);
        }

此处显示的代码close是每个子进程中所有预pipe的文件描述符。

但是在启动所有子进程后,它在父进程中无法执行相同的操作。父进程中打开的文件描述符阻止子进程正确检测连接到其标准输入的管道上的文件结束条件。所有管道描述符仍然在父进程中打开。

总的来说,管道文件描述符的显示处理似乎过于复杂。在启动每个子进程时,不必预先pipe 一切,而是一次创建一个管道。将新进程的标准输入连接到前一次迭代管道中的文件描述符(或第一个子进程的标准输入);为子进程标准输出(或父进程的标准输出,为管道中的最后一个进程)设置管道;然后只需关闭父级中的一对文件描述符。任务完成。