/****************************************************************************
**
** Copyright (C) 1992-2004 Trolltech AS. All rights reserved.
**
** This file is part of the core module of the Qt Toolkit.
**
** This file may be distributed under the terms of the Q Public License
** as defined by Trolltech AS of Norway and appearing in the file
** LICENSE.QPL included in the packaging of this file.
**
** This file may be distributed and/or modified under the terms of the
** GNU General Public License version 2 as published by the Free Software
** Foundation and appearing in the file LICENSE.GPL included in the
** packaging of this file.
**
** See http://www.trolltech.com/pricing.html or email sales@trolltech.com for
**   information about Qt Commercial License Agreements.
** See http://www.trolltech.com/qpl/ for QPL licensing information.
** See http://www.trolltech.com/gpl/ for GPL licensing information.
**
** Contact info@trolltech.com if any conditions of this licensing are
** not clear to you.
**
** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
**
****************************************************************************/

#include "qplatformdefs.h"

#include "qprocess.h"
#include "qprocess_p.h"

#include <qabstracteventdispatcher.h>
#include <qdatetime.h>
#include <qfile.h>
#include <qfileinfo.h>
#include <qlist.h>
#include <qmap.h>
#include <qmutex.h>
#include <qsignal.h>
#include <qsocketnotifier.h>
#include <qdebug.h>

#include <errno.h>
#include <stdlib.h>

static int qt_qprocess_deadChild_pipe[2];
static void (*qt_sa_old_sigchld_handler)(int) = 0;
static void qt_sa_sigchld_handler(int signum)
{
    ::write(qt_qprocess_deadChild_pipe[1], "", 1);
    if (qt_sa_old_sigchld_handler && qt_sa_old_sigchld_handler != SIG_IGN)
        qt_sa_old_sigchld_handler(signum);
}

class QProcessManager : public QObject
{
    Q_OBJECT
public:
    ~QProcessManager();

    static inline QProcessManager &instance()
    {
        // ### add reentrancy
        static QProcessManager singleton;
        return singleton;
    }

    void initialize();

    inline void add(pid_t pid, QProcess *process)
        {
            QMutexLocker lock(&mutex);
            children[pid] = process;
        }

    inline bool has(pid_t pid)
        {
            return children.contains(pid);
        }

public slots:
    void deadChildNotification(int);

protected:
    inline QProcessManager() : old_sigchld_handler(0) { }

private:

    QMutex mutex;
    QMap<pid_t, QProcess *> children;

    void (*old_sigchld_handler)(int);

    QSocketNotifier *shutdownNotifier;
};

QProcessManager::~QProcessManager()
{
}

void QProcessManager::initialize()
{
    QMutexLocker lock(&mutex);
    if (qt_sa_old_sigchld_handler == 0) {
        ::pipe(qt_qprocess_deadChild_pipe);
        ::fcntl(qt_qprocess_deadChild_pipe[0], F_SETFL,
                ::fcntl(qt_qprocess_deadChild_pipe[0], F_GETFL) | O_NONBLOCK);
        struct sigaction oldAction;
        struct sigaction action;
        memset(&action, 0, sizeof(action));
        action.sa_handler = qt_sa_sigchld_handler;
        action.sa_flags = SA_NOCLDSTOP;
        ::sigaction(SIGCHLD, &action, &oldAction);
        if (oldAction.sa_handler != qt_sa_sigchld_handler) {
            old_sigchld_handler = qt_sa_old_sigchld_handler = oldAction.sa_handler;
        }

        if (QAbstractEventDispatcher::instance(thread())) {
            shutdownNotifier = new QSocketNotifier(qt_qprocess_deadChild_pipe[0],
                                                   QSocketNotifier::Read, this);
            connect(shutdownNotifier, SIGNAL(activated(int)),
                    this, SLOT(deadChildNotification(int)));
        }
    }
}

void QProcessManager::deadChildNotification(int)
{
    char c;
    ::read(qt_qprocess_deadChild_pipe[0], &c, 1);

    for (;;) {
        int result;
        pid_t childpid = waitpid(0, &result, WNOHANG);
        if (childpid <= 0)
            break;

        QMutexLocker lock(&mutex);
        QProcess *child = children.value(childpid, 0);
        if (child) {
            ((QProcessPrivate *)child->d_ptr)->exitCode = WEXITSTATUS(result);
            ((QProcessPrivate *)child->d_ptr)->crashed = !WIFEXITED(result);
            qInvokeMetaMember(child, "processDied");
            children.remove(childpid);
        }
    }
}

static void qt_create_pipe(int *pipe)
{
    if (pipe[0] != -1)
        ::close(pipe[0]);
    if (pipe[1] != -1)
        ::close(pipe[1]);
#ifdef Q_OS_IRIX
    if (::socketpair(AF_UNIX, SOCK_STREAM, 0, pipe) == -1) {
        qWarning("QProcessPrivate::createPipe(%p) failed: %s",
                 pipe, strerror(errno));
    }
#else
    if (::pipe(pipe) != 0) {
        qWarning("QProcessPrivate::createPipe(%p) failed: %s",
                 pipe, strerror(errno));
    }
#endif
}

void QProcessPrivate::destroyPipe(int *pipe)
{
    if (pipe[0] != -1) {
        ::close(pipe[0]);
        pipe[0] = -1;
    }
    if (pipe[1] != -1) {
        ::close(pipe[1]);
        pipe[1] = -1;
    }
}

void QProcessPrivate::startProcess()
{
    Q_Q(QProcess);

    // Initialize pipes
    qt_create_pipe(childStartedPipe);
    if (QAbstractEventDispatcher::instance(q->thread())) {
        startupSocketNotifier = new QSocketNotifier(childStartedPipe[0],
                                                    QSocketNotifier::Read, q);
        QObject::connect(startupSocketNotifier, SIGNAL(activated(int)),
                         q, SLOT(startupNotification()));
    }
    qt_create_pipe(writePipe);

    if (QAbstractEventDispatcher::instance(q->thread())) {
        writeSocketNotifier = new QSocketNotifier(writePipe[1],
                                                  QSocketNotifier::Write, q);
        QObject::connect(writeSocketNotifier, SIGNAL(activated(int)),
                         q, SLOT(canWrite()));
        writeSocketNotifier->setEnabled(false);
    }

    qt_create_pipe(standardReadPipe);
    qt_create_pipe(errorReadPipe);

    if (QAbstractEventDispatcher::instance(q->thread())) {
        standardReadSocketNotifier = new QSocketNotifier(standardReadPipe[0],
                                                         QSocketNotifier::Read,
                                                         q);
        QObject::connect(standardReadSocketNotifier, SIGNAL(activated(int)),
                         q, SLOT(canReadStandardOutput()));

        errorReadSocketNotifier = new QSocketNotifier(errorReadPipe[0],
                                                      QSocketNotifier::Read,
                                                      q);
        QObject::connect(errorReadSocketNotifier, SIGNAL(activated(int)),
                         q, SLOT(canReadStandardError()));
    }

    // Start the process (platform dependent)
    processState = QProcess::Starting;
    emit q->stateChanged(processState);

    QProcessManager::instance().initialize();

    pid = (Q_PID) fork();
    if (pid == 0) {
        execChild();
        ::_exit(-1);
    }

    QProcessManager::instance().add(pid, q);

    // parent
    ::close(childStartedPipe[1]);
    ::close(standardReadPipe[1]);
    ::close(errorReadPipe[1]);
    ::close(writePipe[0]);

    childStartedPipe[1] = -1;
    standardReadPipe[1] = -1;
    errorReadPipe[1] = -1;
    writePipe[0] = -1;

    // make all pipes non-blocking
    ::fcntl(standardReadPipe[0], F_SETFL, ::fcntl(standardReadPipe[0], F_GETFL) | O_NONBLOCK);
    ::fcntl(errorReadPipe[0], F_SETFL, ::fcntl(errorReadPipe[0], F_GETFL) | O_NONBLOCK);
    ::fcntl(writePipe[1], F_SETFL, ::fcntl(writePipe[1], F_GETFL) | O_NONBLOCK);
}

void QProcessPrivate::execChild()
{
    QByteArray prog = QFile::encodeName(program);

    // create argument list with right number of elements, and set the
    // final one to 0.
    char **argv = new char *[arguments.count() + 2];
    argv[arguments.count() + 1] = 0;

    // allow invoking of .app bundles on the Mac.
#ifdef Q_OS_MAC
    QFileInfo fileInfo(prog);
    if (fileInfo.isDir() && prog.endsWith(".app")) {
        QByteArray tmp = prog;
        int lastSlashPos = tmp.lastIndexOf('/');
        if(lastSlashPos != -1)
            tmp.remove(0, lastSlashPos + 1);
        tmp = prog + "/Contents/MacOS/" + tmp;
        tmp.resize(tmp.size() - 4); // chop off the .app
        if(QFile::exists(tmp))
            prog = tmp;
    }
#endif

    // add the program name
    argv[0] = new char[prog.size() + 1];
    memcpy(argv[0], prog.data(), prog.size());
    argv[0][prog.size()] = '\0';

    // add every argument to the list
    for (int i = 0; i < arguments.count(); ++i) {
        QString arg = arguments.at(i);
        argv[i + 1] = new char[arg.size() + 1];
        memcpy(argv[i + 1], arg.local8Bit(), arg.size());
        argv[i + 1][arg.size()] = '\0';
    }

    // on all pipes, close the end that we don't use
    ::close(standardReadPipe[0]);
    ::close(errorReadPipe[0]);
    ::close(writePipe[1]);

    // copy the stdout, stderr and stdin sockets
    ::dup2(standardReadPipe[1], fileno(stdout));
    ::dup2(errorReadPipe[1], fileno(stderr));
    ::dup2(writePipe[0], fileno(stdin));

    // make sure this fd is closed if execvp() succeeds
    ::close(childStartedPipe[0]);
    ::fcntl(childStartedPipe[1], F_SETFD, FD_CLOEXEC);

    // enter the working directory
    if (!workingDirectory.isEmpty())
        ::chdir(QFile::encodeName(workingDirectory).data());

    // execute the process
    if (environment.isEmpty()) {
        ::execvp(prog.data(), argv);
    } else {
        // if LD_LIBRARY_PATH exists in the current environment, but
        // not in the environment list passed by the programmer, then
        // copy it over.
#if defined(Q_OS_MACX)
        static const char *libraryPath = "DYLD_LIBRARY_PATH";
#else
        static const char *libraryPath = "LD_LIBRARY_PATH";
#endif
        QStringList matches = environment.filter(QRegExp("^" + QByteArray(libraryPath) + "="));
        char *envLibraryPath = ::getenv(libraryPath);
        if (matches.isEmpty() && envLibraryPath != 0) {
            QString entry = libraryPath;
            entry += "=";
            entry += envLibraryPath;
            environment << QString(libraryPath) +  "=" + QString(envLibraryPath);
        }

        char **envp = new char *[environment.count() + 1];
        envp[environment.count()] = 0;

        for (int j = 0; j < environment.count(); ++j) {
            QString item = environment.at(j);
            envp[j] = new char[item.size() + 1];
            memcpy(envp[j], item.local8Bit(), item.size());
            envp[j][item.size()] = '\0';
        }

        if (!prog.contains("/")) {
            char *path = ::getenv("PATH");
            if (path) {
                QStringList pathEntries = QString(path).split(":");
                for (int k = 0; k < pathEntries.size(); ++k) {
                    QByteArray tmp = QFile::encodeName(pathEntries.at(k));
                    if (!tmp.endsWith('/')) tmp += '/';
                    tmp += prog;
                    ::execve(tmp.data(), argv, envp);
                }
            }
        } else {
            ::execve(prog.data(), argv, envp);
        }
    }

    // notify failure
    ::write(childStartedPipe[1], "", 1);
    ::close(childStartedPipe[1]);
    childStartedPipe[1] = -1;
}

bool QProcessPrivate::processStarted()
{
    char c;
    int i = ::read(childStartedPipe[0], &c, 1);
    if (startupSocketNotifier) {
        startupSocketNotifier->setEnabled(false);
        delete startupSocketNotifier;
        startupSocketNotifier = 0;
    }
    ::close(childStartedPipe[0]);
    childStartedPipe[0] = -1;
    return i <= 0;
}

Q_LONGLONG QProcessPrivate::bytesAvailableFromStdout() const
{
    size_t nbytes = 0;
    Q_LONGLONG available = 0;
    if (::ioctl(standardReadPipe[0], FIONREAD, (char *) &nbytes) >= 0)
        available = (Q_LONGLONG) *((int *) &nbytes);
    return available;
}

Q_LONGLONG QProcessPrivate::bytesAvailableFromStderr() const
{
    size_t nbytes = 0;
    Q_LONGLONG available = 0;
    if (::ioctl(errorReadPipe[0], FIONREAD, (char *) &nbytes) >= 0)
        available = (Q_LONGLONG) *((int *) &nbytes);
    return available;
}

Q_LONGLONG QProcessPrivate::readFromStdout(char *data, Q_LONGLONG maxlen)
{
    return Q_LONGLONG(::read(standardReadPipe[0], data, maxlen));
}

Q_LONGLONG QProcessPrivate::readFromStderr(char *data, Q_LONGLONG maxlen)
{
    return Q_LONGLONG(::read(errorReadPipe[0], data, maxlen));
}

Q_LONGLONG QProcessPrivate::writeToStdin(const char *data, Q_LONGLONG maxlen)
{
    return Q_LONGLONG(::write(writePipe[1], data, maxlen));
}

void QProcessPrivate::killProcess()
{
    ::kill(pid, SIGTERM);
}

static int qt_native_select(const QList<int> &fd, int timeout, bool selectForRead)
{
    fd_set fds;
    FD_ZERO(&fds);

    int fdmax = 0;
    for (int i = 0; i < fd.count(); ++i) {
        int f = fd.at(i);
        if (f > fdmax)
            fdmax = f;
        FD_SET(f, &fds);
    }

    struct timeval tv;
    tv.tv_sec = timeout / 1000;
    tv.tv_usec = (timeout % 1000) * 1000;

    if (selectForRead)
        select(fdmax + 1, &fds, 0, 0, timeout < 0 ? 0 : &tv);
    else
        select(fdmax + 1, 0, &fds, 0, timeout < 0 ? 0 : &tv);

    int ret = 0;
    for (int i = 0; i < fd.count(); ++i) {
        if (FD_ISSET(fd.at(i), &fds))
            ret += 1<<i;
    }
    return ret;
}

bool QProcessPrivate::waitForStarted(int msecs)
{
    Q_Q(QProcess);
    int ret = qt_native_select(QList<int>() << childStartedPipe[0], msecs, true);
    if (ret == 0) {
        processError = QProcess::Timedout;
        q->setErrorString(QT_TRANSLATE_NOOP(QProcess, "Process opeation timed out"));
        return false;
    }

    return processStarted();
}

bool QProcessPrivate::waitForReadyRead(int msecs)
{
    Q_Q(QProcess);
    if (QProcessManager::instance().has(pid)) {
        QTime stopWatch;
        stopWatch.start();
        int timeout = msecs;
        forever {
            int ret = qt_native_select(QList<int>() << qt_qprocess_deadChild_pipe[0]
                                       << standardReadPipe[0] << errorReadPipe[0],
                                       timeout, true);
            int channel = (processChannel == QProcess::StandardOutput? 2 : 4);

            if (ret == 0) {
                processError = QProcess::Timedout;
                q->setErrorString(QT_TRANSLATE_NOOP(QProcess, "Process operation timed out"));
                return false;
            }
            if (ret & 2)
                canReadStandardOutput();
            if (ret & 4)
                canReadStandardError();
            if (ret & 1) {
                QProcessManager::instance().deadChildNotification(0);
                return (ret & channel);
            }
            if (ret & channel)
                return true;
            timeout = msecs - stopWatch.elapsed();
        }
    }
    return false;
}

bool QProcessPrivate::waitForBytesWritten(int msecs)
{
    Q_Q(QProcess);
    if (writeBuffer.isEmpty())
        return false;

    int ret = qt_native_select(QList<int>() << writePipe[1], msecs < 0 ? 0 : msecs, false);
    if (ret == 0) {
        processError = QProcess::Timedout;
        q->setErrorString(QT_TRANSLATE_NOOP(QProcess, "Process operation timed out"));
        return false;
    }

    canWrite();
    return true;
}

bool QProcessPrivate::waitForFinished(int msecs)
{
    Q_Q(QProcess);
    if (QProcessManager::instance().has(pid)) {
        forever {
            int ret = qt_native_select(QList<int>() << qt_qprocess_deadChild_pipe[0]
                                       << standardReadPipe[0] << errorReadPipe[0],
                                       msecs, true);
            if (ret == 0) {
                processError = QProcess::Timedout;
                q->setErrorString(QT_TRANSLATE_NOOP(QProcess, "Process opeation timed out"));
                return false;
            }
            if (ret & 2)
                canReadStandardOutput();
            if (ret & 4)
                canReadStandardError();
            if (ret & 1) {
                QProcessManager::instance().deadChildNotification(0);
                return true;
            }
        }
    }
    return false;
}

bool QProcessPrivate::waitForWrite(int msecs)
{
    return qt_native_select(QList<int>() << writePipe[1], msecs < 0 ? 0 : msecs, false) == 1;
}

void QProcessPrivate::findExitCode()
{
}

void QProcessPrivate::notified()
{
}

#include "qprocess_unix.moc"

