1 /***************************************************************************
2 * Copyright (C) 2005-09 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 * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
19 ***************************************************************************/
23 #include "execwrapper.h"
26 #include "messagemodel.h"
29 ExecWrapper::ExecWrapper(QObject* parent) : QObject(parent) {
30 connect(&_process, SIGNAL(readyReadStandardOutput()), SLOT(processReadStdout()));
31 connect(&_process, SIGNAL(readyReadStandardError()), SLOT(processReadStderr()));
32 connect(&_process, SIGNAL(finished(int, QProcess::ExitStatus)), SLOT(processFinished(int, QProcess::ExitStatus)));
33 connect(&_process, SIGNAL(error(QProcess::ProcessError)), SLOT(processError(QProcess::ProcessError)));
35 connect(this, SIGNAL(output(QString)), SLOT(postStdout(QString)));
36 connect(this, SIGNAL(error(QString)), SLOT(postStderr(QString)));
39 void ExecWrapper::start(const BufferInfo &info, const QString &command) {
43 QRegExp rx("^\\s*(\\S+)(\\s+(.*))?$");
44 if(!rx.exactMatch(command)) {
45 emit error(tr("Invalid command string for /exec: %1").arg(command));
47 _scriptName = rx.cap(1);
51 // Make sure we don't execute something outside a script dir
52 if(_scriptName.contains("../") || _scriptName.contains("..\\"))
53 emit error(tr("Name \"%1\" is invalid: ../ or ..\\ are not allowed!").arg(_scriptName));
56 foreach(QString scriptDir, Quassel::scriptDirPaths()) {
57 QString fileName = scriptDir + _scriptName;
58 if(!QFile::exists(fileName))
60 _process.setWorkingDirectory(scriptDir);
61 _process.start('"' + fileName + "\" " + params);
64 emit error(tr("Could not find script \"%1\"").arg(_scriptName));
67 deleteLater(); // self-destruct
70 void ExecWrapper::postStdout(const QString &msg) {
71 if(_bufferInfo.isValid())
72 Client::userInput(_bufferInfo, msg);
75 void ExecWrapper::postStderr(const QString &msg) {
76 if(_bufferInfo.isValid())
77 Client::messageModel()->insertErrorMessage(_bufferInfo, msg);
80 void ExecWrapper::processFinished(int exitCode, QProcess::ExitStatus status) {
81 if(status == QProcess::CrashExit) {
82 emit error(tr("Script \"%1\" crashed with exit code %2.").arg(_scriptName).arg(exitCode));
86 if(!_stdoutBuffer.isEmpty())
87 foreach(QString msg, _stdoutBuffer.split('\n'))
89 if(!_stderrBuffer.isEmpty())
90 foreach(QString msg, _stderrBuffer.split('\n'))
96 void ExecWrapper::processError(QProcess::ProcessError err) {
97 if(err == QProcess::FailedToStart)
98 emit error(tr("Script \"%1\" could not start.").arg(_scriptName));
100 emit error(tr("Script \"%1\" caused error %2.").arg(_scriptName).arg(err));
102 if(_process.state() != QProcess::Running)
106 void ExecWrapper::processReadStdout() {
107 QString str = QTextCodec::codecForLocale()->toUnicode(_process.readAllStandardOutput());
108 str.replace(QRegExp("\r\n?"), "\n");
109 _stdoutBuffer.append(str);
111 while((idx = _stdoutBuffer.indexOf('\n')) >= 0) {
112 emit output(_stdoutBuffer.left(idx));
113 _stdoutBuffer = _stdoutBuffer.mid(idx + 1);
117 void ExecWrapper::processReadStderr() {
118 QString str = QTextCodec::codecForLocale()->toUnicode(_process.readAllStandardError());
119 str.replace(QRegExp("\r\n?"), "\n");
120 _stderrBuffer.append(str);
122 while((idx = _stderrBuffer.indexOf('\n')) >= 0) {
123 emit error(_stderrBuffer.left(idx));
124 _stderrBuffer = _stderrBuffer.mid(idx + 1);