exec.cc

Go to the documentation of this file.

#include "config.h"

#include <iostream>

#ifdef HAVE_STDLIB_H
#include <stdlib.h>
#endif
#ifdef HAVE_SYS_TYPES_H
#include <sys/types.h>
#endif
#ifdef HAVE_SYS_WAIT_H
#include <sys/wait.h>
#endif
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#ifdef HAVE_SIGNAL_H
#include <signal.h>
#endif
#ifdef HAVE_SYS_SELECT_H
#include <sys/select.h>
#endif
#include <sys/time.h>

#include <cerrno>

#include "asserts.h"
#include "error.h"
#include "fs.h"
#include "exec.h"

/** C'tor */
execute::execute()
{
        clear();
}

/** D'tor */
execute::~execute()
{
        clear();
}

/** Reset the execute class to default values, kill the child processif one is
 * running */
void execute::clear(void)
{
        if (child_running() && is_parent())
                signal_child(SIGKILL);

        m_fd1[0] = 0;
        m_fd1[1] = 0;
        m_fd2[0] = 0;
        m_fd2[1] = 0;
        m_fd3[0] = 0;
        m_fd3[1] = 0;
        m_pid = 0;
        m_status = 0;
        m_in_eof = false;
        m_out_eof = false;
        m_err_eof = false;
        m_child_started = false;
}

/** Generic signal handler */
static void _signal_handler(int signo)
{
        throw(ERROR(0,"Tried to write to a read-only pipe"));
}

/** Fork a child process */
void execute::fork(void)
{
        clear();
        if (signal(SIGPIPE, _signal_handler) == SIG_ERR)
                throw(ERROR(errno,"Could not catch SIGPIPIE signal"));
        if (pipe(m_fd1) < 0)
                throw(ERROR(errno,"Could not create pipe for in"));
        if (pipe(m_fd2) < 0)
                throw(ERROR(errno,"Could not create pipe for out"));
        if (pipe(m_fd3) < 0)
                throw(ERROR(errno,"Could not create pipe for err"));
        if ((m_pid = ::fork()) < 0)
                throw(ERROR(errno,"Could not fork"));
        if (m_pid > 0) {
                // Parent process
                close(m_fd1[0]);
                close(m_fd2[1]);
                close(m_fd3[1]);
                m_child_started = true;
        }
        else {
                // Child process
                close(m_fd1[1]);
                close(m_fd2[0]);
                close(m_fd3[0]);
        }
}

/** Returns true if called by the child */
bool execute::is_child(void)
{
        if (m_pid == 0) {
                return(true);
        }
        return(false);
}

/** Returns true if called by the parent */
bool execute::is_parent(void)
{
        bool value;

        value = !is_child();

        return(value);
}

/** Returns the PID */
pid_t execute::my_pid(void)
{
        pid_t value;

        value = pid();

        return(value);
}

/** Called by the child to exit with a particular code */
void execute::exit(int code)
{
        if (is_child())
		::exit(code);
}

/** Called by the child to reroute the child's stdin, stdout, and stderr to
 * the parent */
void execute::reroute_stdio(void)
{
        if (!is_child())
                return;

        if (m_fd1[0] != STDIN_FILENO) {
                if (dup2(m_fd1[0], STDIN_FILENO) != STDIN_FILENO) {
                        error e(errno,ERROR_INSTANCE("dup2() failed for stdin"));
                        std::cerr << e << std::endl;
                        exit(127);
                }
                close(m_fd1[0]);
        }
        if (m_fd2[1] != STDOUT_FILENO) {
                if (dup2(m_fd2[1], STDOUT_FILENO) != STDOUT_FILENO) {
                        error e(errno,ERROR_INSTANCE("dup2() failed for stdout"));
                        std::cerr << e << std::endl;
                        exit(127);
                }
                close(m_fd2[1]);
        }
        if (m_fd3[1] != STDERR_FILENO) {
                if (dup2(m_fd3[1], STDERR_FILENO) != STDERR_FILENO) {
                        error e(errno,ERROR_INSTANCE("dup2() failed for stderr"));
                        std::cerr << e << std::endl;
                        exit(127);
                }
                close(m_fd3[1]);
        }
}

/** Returns the child's PID */
pid_t execute::child_pid(void)
{
        return(m_pid);
}

/** Send a signal to the child */
void execute::signal_child(int signal_no)
{
        kill(m_pid, signal_no);
}

/** Send a HUP signal to the child */
void execute::hup_child(void)
{
        signal_child(SIGHUP);
}

/** Send a KILL signal to the child */
void execute::kill_child(void)
{
        signal_child(SIGKILL);
}

/** Wait for the child to exit */
void execute::wait(void)
{
        if (is_parent()) {
                pid_t value;
                value = waitpid(m_pid, &m_status, 0);
        }
}

/** Check the child's status */
pid_t execute::check_child_(void)
{
        pid_t value;
        
        errno = 0;
        value = waitpid(m_pid, &m_status, WNOHANG|WUNTRACED);

        return(value);
}

/** Returns true if the child has been started */
bool execute::child_started(void) const
{
        return(m_child_started);
}

/** Returns true if the child is running */
bool execute::child_running(void)
{
        pid_t pid;
        
        pid = check_child_();
        if (pid > 0) return(false); // The child has exited
        if ((pid < 0) && (errno == ECHILD)) return(false); // No such child exists
        return(true);
}

/** Returns true of the child has existed */
bool execute::child_exited(void)
{
        pid_t pid;

        pid = check_child_();
        if (pid > 0) return(true);
        if ((pid < 0) && (errno == ECHILD)) return(true);
        return(false);
}

/** Returns true if the child has exited normally */
bool execute::child_exited_normally(void)
{
        bool value;

        value = (child_exited() && (child_exit_code() == 0));

        return(value);
}

/** Returns true if the child was signaled */
bool execute::child_signaled(void)
{
        bool value;

        check_child_();
        value = WIFSIGNALED(m_status);

        return(value);
}

/** Returns true if the child returned exit code 0 and no caught signals */
bool execute::child_exited_success(void)
{
        bool value;

        value = (child_exited() && (child_exit_code() == 0) && !child_signaled());

        return(value);
}

/** Return the child's exit code */
int execute::child_exit_code(void)
{
        int value;

        check_child_();
        value = WEXITSTATUS(m_status);

        return(value);
}

/** If the child was signaled, return the signal number */
int execute::child_signal_no(void)
{
        int value;

        check_child_();
        if (child_exited()) {
                value = WTERMSIG(m_status);
                return(value);
        }
        else {
                value = WSTOPSIG(m_status);
                return(value);
        }
}

/** Return a file descriptor for I/O between parent and child

        If called by the parent, a writeable file descriptor is returned.
        If called by the child, a readable file descriptor is returned.

        If retroute_stdio() was called by the child, then the returned file
        descriptor is the same as that used by the child for stdin.
 */
int execute::in_fd(void)
{
        if (is_parent()) {
                // Return write end of input pipe
                return(m_fd1[1]);
        }
        else {
                // Return read end of input pipe
                return(m_fd1[0]); 
        }
}

/** Return a file descriptor for I/O between parent a child.

        If called by the parent, a readable file descriptor is returned.
        If called by the child, a writable file descriptor is returned.

        If reroute_stdio() was called by the child, then the returned file
        descriptor is the same as that used by the child for stdout.
 */
int execute::out_fd(void)
{
        if (is_parent()) {
                // Return read end of output pipe
                return(m_fd2[0]);
        }
        else {
                // Return write end of output pipe
                return(m_fd2[1]);
        }
}

/** Return a file descriptor for I/O between parent and child.

        If called by the parent, a readable file descriptor is returned.
        If called by the child, a writeable file descriptor is returned.

        If reroute_stdio() was called by the child, then the returned file
        descriptior is the same as that used by the child for stderr.
 */
int execute::err_fd(void)
{
        if (is_parent()) {
                // return read end of error pipe
                return(m_fd3[0]); 
        }
        else {
                // return write end of error pipe
                return(m_fd3[1]); 
        }
}

/** Return true if the file descriptor is ready to be written to */
bool execute::check_write_ready_(int fd)
{
        struct timeval timeout = { 0, 0 };
        fd_set wset;

        FD_ZERO(&wset);
        FD_SET(fd, &wset);
        select(fd+1, 0, &wset, 0, &timeout);
        if (FD_ISSET(fd, &wset)) {
                return(true);
        }
        return(false);
}

/** Return true if the file descriptor has input ready to be read */
bool execute::check_read_ready_(int fd)
{
        struct timeval timeout = { 0, 0 };
        fd_set rset;

        FD_ZERO(&rset);
        FD_SET(fd, &rset);
        select(fd+1, &rset, 0, 0, &timeout);
        if (FD_ISSET(fd, &rset)) {
                return(true);
        }
        return(false);
}

/** Execute a command, rerouting stdin, stdout, and stderr to parent */
void execute::exec(const std::string command)
{
        fork();
        if (is_parent())
                return;

        /*
        {
                bool wait_for_debugger = true;

                std::cerr << "Waiting for debugger to attach..." << std::endl;
                while (wait_for_debugger);
                std::cerr << "Debugger attached." << std::endl;
        }
        */
        reroute_stdio();
        
        execl("/bin/sh", "sh", "-c", command.c_str(), (char *)0);
        exit(127);
}

/** Execute a command, rerouting stdin, stdout, and stderr to parent */
void execute::exec(const std::string binary, const std::vector<std::string> argv)
{
        char *bin = 0;
        char *base = 0;
        char **args = 0;
        uint16 c;
        size_t size;

        fork();
        if (is_parent())
                return;
        
        /*
        {
                bool wait_for_debugger = true;

                std::cerr << "Waiting for debugger to attach..." << std::endl;
                while (wait_for_debugger);
                std::cerr << "Debugger attached." << std::endl;
        }
        */

        reroute_stdio();

        bin = (char *)binary.c_str();
        base = bin+strlen(bin);
        while ((base > bin) && (*base != '/')) {
                base--;
        }
        if (*base == '/') {
                base++;
        }
        size = (argv.size() + 2) * sizeof(char *);
        args = (char **)malloc(size);
        memset((void *)args, 0, size);
        if (args == 0) {
                exit(127);
        }
        args[0] = base;
        for (c = 0; c < argv.size(); c++) {
                args[c+1] = (char *)argv[c].c_str();
        }
        args[c+1] = 0;

        execv(bin, args);
        exit(127);
}

/** Check I/O for input

        If called by the parent, check if ready to write to child's input.
        If called by the child, check if input is ready to be read.

        If reroute_stdio() was called by the child, then this pipe is the same as
        used by the child for stdin.
 */
bool execute::in_ready(void)
{
        bool value;

        if (is_parent())
                value = check_write_ready_(in_fd());
        else
                value = check_read_ready_(in_fd());
        
        return(value);
}

/** Check I/O for output

        If called by the parent, check if output from child is ready to be read.
        If called by the child, check if output to parent is ready to be written to.

        If reroute_stdio() was called by the child, then this pipe is the same as
        used by the child for stdout.
 */
bool execute::out_ready(void)
{
        bool value;

        if (is_parent())
                value = check_read_ready_(out_fd());
        else
                value = check_write_ready_(out_fd());
        
        return(value);
}

/** Check I/O for output

        If called by the parent, check if output from child is ready to be read.
        If called by the child, check if output to parent is ready to be written to.

        If reroute_stdio() was called by the child, then this pipe is the same as
        used by the child for stderr.
 */
bool execute::err_ready(void)
{
        bool value;

        if (is_parent())
                value = check_read_ready_(err_fd());
        else
                value = check_write_ready_(err_fd());
        
        return(value);
}

/** Check for input EOF */
bool execute::in_eof(void)
{
        return(m_in_eof);
}

/** Check for output EOF */
bool execute::out_eof(void)
{
        return(m_out_eof);
}

/** Check for err EOF */
bool execute::err_eof(void)
{
        return(m_err_eof);
}

/** Allow child to read input from in_fd() */
int execute::in_read(char* buf, const int len)
{
        int n;
        int err;

        n = read(in_fd(), buf, len);
        err = errno;
        errno = 0;
        if (n > 0) {
                return(n);
        }
        if (child_started() && !child_running() && (err != EAGAIN))
                m_in_eof = true;
        return(0);
}

/** Allow parent to write output to in_fd() */
int execute::in_write(const char* buf, const int len)
{
        int n;
        int err;

        n = write(in_fd(), buf, len);
        err = errno;
        errno = 0;
        if (n > 0) {
                return(n);
        }
        if (child_started() && !child_running() && (err != EAGAIN))
                m_in_eof = true;
        return(0);
}

/** Allow parent to read out_fd() */
int execute::out_read(char* buf, const int len)
{
        int n;
        int err;

        n = read(out_fd(), buf, len);
        err = errno;
        errno = 0;
        if (n > 0) {
                return(n);
        }
        if (child_started() && !child_running() && (err != EAGAIN))
                m_out_eof = true;
        return(0);
}

/** Allow child to write to out_fd() */
int execute::out_write(const char* buf, const int len)
{
        int n;
        int err;

        n = write(out_fd(), buf, len);
        err = errno;
        errno = 0;
        if (n > 0) {
                return(n);
        }
        if (child_started() && !child_running() && (err != EAGAIN))
                m_out_eof = true;
        return(0);
}

/** Allow parent to read from err_fd() */
int execute::err_read(char* buf, const int len)
{
        int n;
        int err;

        n = read(err_fd(), buf, len);
        err = errno;
        errno = 0;
        if (n > 0) {
                return(n);
        }
        if (child_started() && !child_running() && (err != EAGAIN))
                m_err_eof = true;
        return(0);
}

/** Allow child to write to err_fd() */
int execute::err_write(const char* buf, const int len)
{
        int n;
        int err;

        n = write(err_fd(), buf, len);
        err = errno;
        errno = 0;
        if (n > 0) {
                return(n);
        }
        if (child_started() && !child_running() && (err != EAGAIN))
                m_err_eof = true;
        return(0);
}

/** Dump execute object information -- used for debugging */
void execute::print(std::ostream& out)
{
        out << "execute::is_child() = " << is_child() << std::endl;
        out << "execute::is_parent() = " << is_parent() << std::endl;
        out << "execute::my_pid() = " << my_pid() << std::endl;
        out << "execute::child_running() = " << child_running() << std::endl;
        out << "execute::child_exited() = " << child_exited() << std::endl;
        out << "execute::child_exited_normally() = " << child_exited_normally() 
                << std::endl;
        out << "execute::child_exited_success() = " << child_exited_success() 
                << std::endl;
        out << "execute::child_signaled() = " << child_signaled() << std::endl;
        out << "execute::child_exit_code() = " << child_exit_code() << std::endl;
        out << "execute::child_signal_no() = " << child_signal_no() << std::endl;
        out << "execute::in_fd() = " << in_fd() << std::endl;
        out << "execute::out_fd() = " << out_fd() << std::endl;
        out << "execute::err_fd() = " << err_fd() << std::endl;
        out << "execute::in_ready() = " << in_ready() << std::endl;
        out << "execute::out_ready() = " << out_ready() << std::endl;
        out << "execute::err_ready() = " << err_ready() << std::endl;
}

/** Convenience function to call execute::print() */
std::ostream& operator << (std::ostream& out, execute& exe)
{
        exe.print(out);

        return(out);
}