Toy shell is not properly piping

I'm not going to lie. This is a homework question. However, as far as I know, the glasses are gone. Right now, I'm just looking for an answer because I ... I think I would be crazy.

The purpose of this program is to execute a command ps -A | grep (inputstring) | wc -l

in a manner similar to how the shell does. So, I start processes and wait for each other. The newest process, great-grandson execlp("ps","ps","-A",NULL)

, which replaces the process ps -A

. Before doing this execlp

, I make sure its standard output goes to the pipe's output. The next process is on line wait()

ing and is already set up so that the input pipe goes to standard and the standard output goes to the output pipe and it will grep, etc.

I'm pretty sure I configured it correctly. And yet ... the program does. Not. Job.

#include <stdlib.h>
#include <iostream>
#include <string>

#define MAXLINE 1500
#define READ 0
#define WRITE 1

using namespace std;

int main( int argc, char** argv ) {
//* start of input block
if ( argc != 2 ) {
    cout << "Usage: ./a.out arg1" << endl;
    return 0;
}
string in = argv[1];
// end of input block */
int pipeA[2], pipeB[2], pid, stat;

// get our first set of pipes
if ( pipe(pipeA) < 0 ) {
    cerr << "Pipe error.\n";
    exit(-1);
}
if ( pipe(pipeB) < 0 ) {
    cerr << "Pipe error.\n";
    exit(-1);
}

// make the first fork
if ( (pid = fork() ) < 0 ) { cerr << "Fork error.\n"; exit(-1); }

if ( pid > 0 ) {    // parent case
    wait(&stat);
} else {            // child case
    if ( (pid = fork()) < 0 ) { cerr << "Fork Error\n"; exit(-1); }
    if ( pid > 0 ) {    // child
        wait(&stat);
        dup2(pipeA[READ],READ);
        execlp("wc","wc","-l",NULL);
    } else {    // grand-child
        if ( (pid = fork()) < 0 ) { cerr << "Fork Error\n"; exit(-1); }
        if ( pid > 0 ) {    // still grand-child
            wait(&stat);
            dup2(pipeB[READ],READ);  
            dup2(pipeA[WRITE],WRITE); 
            close(pipeB[READ]);
            execlp("grep","grep",in.c_str(),NULL);
        } else {    // great grand-child
            dup2(pipeB[WRITE],WRITE); // t now goes to pipeB[1]
            close(READ);
            close(pipeB[READ]);
            execlp("ps", "ps", "-A", NULL);
        }
    }
}
return 0;
}

      

EDIT: Changed for the two-pipe version of my code.

+1


source to share


2 answers


I'm pretty sure this is what you are trying to do. Sorry in advance for the sloppy coding. its somewhat late here and i really should be sleeping right now:

#include <iostream>
#include <string>
#include <cstdlib>
#include <cstdio>
#include <unistd.h>

#define READ 0
#define WRITE 1

// ps -A | grep argv[1] | wc -l

int main( int argc, char** argv )
{
    //* start of input block
    if ( argc != 2 )
    {
        std::cout << "Usage: ./a.out arg1" << std::endl;
        return 0;
    }

    // make local copy of argument
    std::string in = argv[1];
    int fd1[2], fd2[2], pid;

    // allocate two pipe sets
    if (pipe(fd1) < 0 || pipe(fd2) < 0)
    {
        perror("Failed to create pipe.");
        return EXIT_FAILURE;
    }

    // launch first child process.
    if ((pid = fork()) < 0)
    {
        perror("Failed to fork child(1)");
        return EXIT_FAILURE;
    }

    if (pid == 0)
    {
        // wc -l process. 
        //  stdin  = fd2(read)
        close(fd1[READ]);
        close(fd1[WRITE]);
        close(fd2[WRITE]);
        dup2(fd2[READ],STDIN_FILENO);
        execlp("wc","wc","-l",NULL);
    }

    // fork again. this time for grep
    if ((pid = fork()) < 0)
    {
        perror("Failed to fork child(2)");
        return EXIT_FAILURE;
    }

    if (pid == 0)
    {
        // grep argv[1] process.
        //  stdin  = fd1(read)
        //  stdout = fd2(write)            
        close(fd1[WRITE]);
        close(fd2[READ]);
        dup2(fd2[WRITE], STDOUT_FILENO);
        dup2(fd1[READ], STDIN_FILENO);
        execlp("grep", "grep", in.c_str(), NULL);
    }

    //  fork once more. this time for ps -A
    if ((pid = fork()) < 0)
    {
        perror("Failed to fork child(3)");
        return EXIT_FAILURE;
    }

    if (pid == 0)
    {
        // ps -A process.
        //  stdout = fd1(write)
        close(fd2[WRITE]);
        close(fd2[READ]);
        close(fd1[READ]);
        dup2(fd1[WRITE], STDOUT_FILENO);
        execlp("ps", "ps", "-A", NULL);
    }

    int stat=0;
    wait(&stat);

    return EXIT_SUCCESS;
}

      



On my system, ps -A

it reports 141 lines, of which 41 have a word System

somewhere inside, checked by a simple run ps -A | grep System | wc -l

. The above code generates exactly the same result.

Hope this helps you.

+1


source


I'm not sure, but maybe calling dup2 before waiting for the child to solve the pipe issue.

The reason I'm not sure is that it's usually stdin and stdout buffered, so I suppose that even if you hook a pipe with them after the child is done, you should get the same results, but maybe be (if anyone knows the answer to this please correct me) the buffers for stdin and stdout will be erased with the termination of the child process.



Also could you update the code in your question to contain the modified code with two sets of pipes?

0


source







All Articles