1 /***************************************************************************
2 * Copyright (C) 2005-2018 by the Quassel Project *
3 * devel@quassel-irc.org *
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. *
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. *
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 ***************************************************************************/
24 #include "execwrapper.h"
27 #include "messagemodel.h"
31 ExecWrapper::ExecWrapper(QObject *parent) : QObject(parent)
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);
38 connect(this, &ExecWrapper::output, this, &ExecWrapper::postStdout);
39 connect(this, &ExecWrapper::error, this, &ExecWrapper::postStderr);
43 void ExecWrapper::start(const BufferInfo &info, const QString &command)
48 QRegExp rx(R"(^\s*(\S+)(\s+(.*))?$)");
49 if (!rx.exactMatch(command)) {
50 emit error(tr("Invalid command string for /exec: %1").arg(command));
53 _scriptName = rx.cap(1);
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));
62 foreach(QString scriptDir, Quassel::scriptDirPaths()) {
63 QString fileName = scriptDir + _scriptName;
64 if (!QFile::exists(fileName))
66 _process.setWorkingDirectory(scriptDir);
67 _process.start('"' + fileName + "\" " + params);
70 emit error(tr("Could not find script \"%1\"").arg(_scriptName));
73 deleteLater(); // self-destruct
77 void ExecWrapper::postStdout(const QString &msg)
79 if (_bufferInfo.isValid())
80 Client::userInput(_bufferInfo, msg);
84 void ExecWrapper::postStderr(const QString &msg)
86 if (_bufferInfo.isValid())
87 Client::messageModel()->insertErrorMessage(_bufferInfo, msg);
91 void ExecWrapper::processFinished(int exitCode, QProcess::ExitStatus status)
93 if (status == QProcess::CrashExit) {
94 emit error(tr("Script \"%1\" crashed with exit code %2.").arg(_scriptName).arg(exitCode));
98 if (!_stdoutBuffer.isEmpty())
99 foreach(QString msg, _stdoutBuffer.split('\n'))
101 if (!_stderrBuffer.isEmpty())
102 foreach(QString msg, _stderrBuffer.split('\n'))
109 void ExecWrapper::processError(QProcess::ProcessError err)
111 if (err == QProcess::FailedToStart)
112 emit error(tr("Script \"%1\" could not start.").arg(_scriptName));
114 emit error(tr("Script \"%1\" caused error %2.").arg(_scriptName).arg(err));
116 if (_process.state() != QProcess::Running)
121 void ExecWrapper::processReadStdout()
123 QString str = QTextCodec::codecForLocale()->toUnicode(_process.readAllStandardOutput());
124 str.replace(QRegExp("\r\n?"), "\n");
125 _stdoutBuffer.append(str);
127 while ((idx = _stdoutBuffer.indexOf('\n')) >= 0) {
128 emit output(_stdoutBuffer.left(idx));
129 _stdoutBuffer = _stdoutBuffer.mid(idx + 1);
134 void ExecWrapper::processReadStderr()
136 QString str = QTextCodec::codecForLocale()->toUnicode(_process.readAllStandardError());
137 str.replace(QRegExp("\r\n?"), "\n");
138 _stderrBuffer.append(str);
140 while ((idx = _stderrBuffer.indexOf('\n')) >= 0) {
141 emit error(_stderrBuffer.left(idx));
142 _stderrBuffer = _stderrBuffer.mid(idx + 1);