src/client/execwrapper.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-2018 by the Quassel Project                        *
   3  *   devel@quassel-irc.org                                                 *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) version 3.                                           *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
  19  ***************************************************************************/
  20
  21 #include <QFile>
  22 #include <QTextCodec>
  23
  24 #include "execwrapper.h"
  25
  26 #include "client.h"
  27 #include "messagemodel.h"
  28 #include "quassel.h"
  29 #include "util.h"
  30
  31 ExecWrapper::ExecWrapper(QObject *parent) : QObject(parent)
  32 {
  33     connect(&_process, &QProcess::readyReadStandardOutput, this, &ExecWrapper::processReadStdout);
  34     connect(&_process, &QProcess::readyReadStandardError, this, &ExecWrapper::processReadStderr);
  35     connect(&_process, selectOverload<int, QProcess::ExitStatus>(&QProcess::finished), this, &ExecWrapper::processFinished);
  36     connect(&_process, selectOverload<QProcess::ProcessError>(&QProcess::error), this, &ExecWrapper::processError);
  37
  38     connect(this, &ExecWrapper::output, this, &ExecWrapper::postStdout);
  39     connect(this, &ExecWrapper::error, this, &ExecWrapper::postStderr);
  40 }
  41
  42
  43 void ExecWrapper::start(const BufferInfo &info, const QString &command)
  44 {
  45     _bufferInfo = info;
  46     QString params;
  47
  48     QRegExp rx(R"(^\s*(\S+)(\s+(.*))?$)");
  49     if (!rx.exactMatch(command)) {
  50         emit error(tr("Invalid command string for /exec: %1").arg(command));
  51     }
  52     else {
  53         _scriptName = rx.cap(1);
  54         params = rx.cap(3);
  55     }
  56
  57     // Make sure we don't execute something outside a script dir
  58     if (_scriptName.contains("../") || _scriptName.contains("..\\"))
  59         emit error(tr(R"(Name "%1" is invalid: ../ or ..\ are not allowed!)").arg(_scriptName));
  60
  61     else {
  62         foreach(QString scriptDir, Quassel::scriptDirPaths()) {
  63             QString fileName = scriptDir + _scriptName;
  64             if (!QFile::exists(fileName))
  65                 continue;
  66             _process.setWorkingDirectory(scriptDir);
  67             _process.start('"' + fileName + "\" " + params);
  68             return;
  69         }
  70         emit error(tr("Could not find script \"%1\"").arg(_scriptName));
  71     }
  72
  73     deleteLater(); // self-destruct
  74 }
  75
  76
  77 void ExecWrapper::postStdout(const QString &msg)
  78 {
  79     if (_bufferInfo.isValid())
  80         Client::userInput(_bufferInfo, msg);
  81 }
  82
  83
  84 void ExecWrapper::postStderr(const QString &msg)
  85 {
  86     if (_bufferInfo.isValid())
  87         Client::messageModel()->insertErrorMessage(_bufferInfo, msg);
  88 }
  89
  90
  91 void ExecWrapper::processFinished(int exitCode, QProcess::ExitStatus status)
  92 {
  93     if (status == QProcess::CrashExit) {
  94         emit error(tr("Script \"%1\" crashed with exit code %2.").arg(_scriptName).arg(exitCode));
  95     }
  96
  97     // empty buffers
  98     if (!_stdoutBuffer.isEmpty())
  99         foreach(QString msg, _stdoutBuffer.split('\n'))
 100         emit output(msg);
 101     if (!_stderrBuffer.isEmpty())
 102         foreach(QString msg, _stderrBuffer.split('\n'))
 103         emit error(msg);
 104
 105     deleteLater();
 106 }
 107
 108
 109 void ExecWrapper::processError(QProcess::ProcessError err)
 110 {
 111     if (err == QProcess::FailedToStart)
 112         emit error(tr("Script \"%1\" could not start.").arg(_scriptName));
 113     else
 114         emit error(tr("Script \"%1\" caused error %2.").arg(_scriptName).arg(err));
 115
 116     if (_process.state() != QProcess::Running)
 117         deleteLater();
 118 }
 119
 120
 121 void ExecWrapper::processReadStdout()
 122 {
 123     QString str = QTextCodec::codecForLocale()->toUnicode(_process.readAllStandardOutput());
 124     str.replace(QRegExp("\r\n?"), "\n");
 125     _stdoutBuffer.append(str);
 126     int idx;
 127     while ((idx = _stdoutBuffer.indexOf('\n')) >= 0) {
 128         emit output(_stdoutBuffer.left(idx));
 129         _stdoutBuffer = _stdoutBuffer.mid(idx + 1);
 130     }
 131 }
 132
 133
 134 void ExecWrapper::processReadStderr()
 135 {
 136     QString str = QTextCodec::codecForLocale()->toUnicode(_process.readAllStandardError());
 137     str.replace(QRegExp("\r\n?"), "\n");
 138     _stderrBuffer.append(str);
 139     int idx;
 140     while ((idx = _stderrBuffer.indexOf('\n')) >= 0) {
 141         emit error(_stderrBuffer.left(idx));
 142         _stderrBuffer = _stderrBuffer.mid(idx + 1);
 143     }
 144 }