forked from qt-creator/qt-creator
move src/plugins/coreplugin/ssh to src/lib/utils/ssh
Merge-request: 253 Reviewed-by: hjk <qtc-committer@nokia.com>
This commit is contained in:
237
src/libs/utils/ssh/sshremoteprocess.cpp
Normal file
237
src/libs/utils/ssh/sshremoteprocess.cpp
Normal file
@@ -0,0 +1,237 @@
|
||||
/**************************************************************************
|
||||
**
|
||||
** This file is part of Qt Creator
|
||||
**
|
||||
** Copyright (c) 2011 Nokia Corporation and/or its subsidiary(-ies).
|
||||
**
|
||||
** Contact: Nokia Corporation (qt-info@nokia.com)
|
||||
**
|
||||
** No Commercial Usage
|
||||
**
|
||||
** This file contains pre-release code and may not be distributed.
|
||||
** You may use this file in accordance with the terms and conditions
|
||||
** contained in the Technology Preview License Agreement accompanying
|
||||
** this package.
|
||||
**
|
||||
** GNU Lesser General Public License Usage
|
||||
**
|
||||
** Alternatively, this file may be used under the terms of the GNU Lesser
|
||||
** General Public License version 2.1 as published by the Free Software
|
||||
** Foundation and appearing in the file LICENSE.LGPL included in the
|
||||
** packaging of this file. Please review the following information to
|
||||
** ensure the GNU Lesser General Public License version 2.1 requirements
|
||||
** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
|
||||
**
|
||||
** In addition, as a special exception, Nokia gives you certain additional
|
||||
** rights. These rights are described in the Nokia Qt LGPL Exception
|
||||
** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
|
||||
**
|
||||
** If you have questions regarding the use of this file, please contact
|
||||
** Nokia at qt-info@nokia.com.
|
||||
**
|
||||
**************************************************************************/
|
||||
|
||||
#include "sshremoteprocess.h"
|
||||
#include "sshremoteprocess_p.h"
|
||||
|
||||
#include "sshincomingpacket_p.h"
|
||||
#include "sshsendfacility_p.h"
|
||||
|
||||
#include <botan/exceptn.h>
|
||||
|
||||
#include <QtCore/QTimer>
|
||||
|
||||
namespace Utils {
|
||||
|
||||
const QByteArray SshRemoteProcess::AbrtSignal("ABRT");
|
||||
const QByteArray SshRemoteProcess::AlrmSignal("ALRM");
|
||||
const QByteArray SshRemoteProcess::FpeSignal("FPE");
|
||||
const QByteArray SshRemoteProcess::HupSignal("HUP");
|
||||
const QByteArray SshRemoteProcess::IllSignal("ILL");
|
||||
const QByteArray SshRemoteProcess::IntSignal("INT");
|
||||
const QByteArray SshRemoteProcess::KillSignal("KILL");
|
||||
const QByteArray SshRemoteProcess::PipeSignal("PIPE");
|
||||
const QByteArray SshRemoteProcess::QuitSignal("QUIT");
|
||||
const QByteArray SshRemoteProcess::SegvSignal("SEGV");
|
||||
const QByteArray SshRemoteProcess::TermSignal("TERM");
|
||||
const QByteArray SshRemoteProcess::Usr1Signal("USR1");
|
||||
const QByteArray SshRemoteProcess::Usr2Signal("USR2");
|
||||
|
||||
SshRemoteProcess::SshRemoteProcess(const QByteArray &command, quint32 channelId,
|
||||
Internal::SshSendFacility &sendFacility)
|
||||
: d(new Internal::SshRemoteProcessPrivate(command, channelId, sendFacility, this))
|
||||
{
|
||||
connect(d, SIGNAL(started()), this, SIGNAL(started()),
|
||||
Qt::QueuedConnection);
|
||||
connect(d, SIGNAL(outputAvailable(QByteArray)), this,
|
||||
SIGNAL(outputAvailable(QByteArray)), Qt::QueuedConnection);
|
||||
connect(d, SIGNAL(errorOutputAvailable(QByteArray)), this,
|
||||
SIGNAL(errorOutputAvailable(QByteArray)), Qt::QueuedConnection);
|
||||
connect(d, SIGNAL(closed(int)), this, SIGNAL(closed(int)),
|
||||
Qt::QueuedConnection);
|
||||
}
|
||||
|
||||
SshRemoteProcess::~SshRemoteProcess()
|
||||
{
|
||||
Q_ASSERT(d->channelState() == Internal::SshRemoteProcessPrivate::Inactive
|
||||
|| d->channelState() == Internal::SshRemoteProcessPrivate::CloseRequested
|
||||
|| d->channelState() == Internal::SshRemoteProcessPrivate::Closed);
|
||||
delete d;
|
||||
}
|
||||
|
||||
void SshRemoteProcess::addToEnvironment(const QByteArray &var, const QByteArray &value)
|
||||
{
|
||||
if (d->channelState() == Internal::SshRemoteProcessPrivate::Inactive)
|
||||
d->m_env << qMakePair(var, value); // Cached locally and sent on start()
|
||||
}
|
||||
|
||||
void SshRemoteProcess::start()
|
||||
{
|
||||
if (d->channelState() == Internal::SshRemoteProcessPrivate::Inactive) {
|
||||
#ifdef CREATOR_SSH_DEBUG
|
||||
qDebug("process start requested, channel id = %u", d->localChannelId());
|
||||
#endif
|
||||
d->requestSessionStart();
|
||||
}
|
||||
}
|
||||
|
||||
void SshRemoteProcess::sendSignal(const QByteArray &signal)
|
||||
{
|
||||
try {
|
||||
if (isRunning())
|
||||
d->m_sendFacility.sendChannelSignalPacket(d->remoteChannel(),
|
||||
signal);
|
||||
} catch (Botan::Exception &e) {
|
||||
d->setError(QString::fromAscii(e.what()));
|
||||
d->closeChannel();
|
||||
}
|
||||
}
|
||||
|
||||
void SshRemoteProcess::closeChannel()
|
||||
{
|
||||
d->closeChannel();
|
||||
}
|
||||
|
||||
void SshRemoteProcess::sendInput(const QByteArray &data)
|
||||
{
|
||||
if (isRunning())
|
||||
d->sendData(data);
|
||||
}
|
||||
|
||||
bool SshRemoteProcess::isRunning() const
|
||||
{
|
||||
return d->m_procState == Internal::SshRemoteProcessPrivate::Running;
|
||||
}
|
||||
|
||||
QString SshRemoteProcess::errorString() const { return d->errorString(); }
|
||||
|
||||
int SshRemoteProcess::exitCode() const { return d->m_exitCode; }
|
||||
|
||||
QByteArray SshRemoteProcess::exitSignal() const { return d->m_signal; }
|
||||
|
||||
namespace Internal {
|
||||
|
||||
SshRemoteProcessPrivate::SshRemoteProcessPrivate(const QByteArray &command,
|
||||
quint32 channelId, SshSendFacility &sendFacility, SshRemoteProcess *proc)
|
||||
: AbstractSshChannel(channelId, sendFacility), m_procState(NotYetStarted),
|
||||
m_wasRunning(false), m_exitCode(0), m_command(command), m_proc(proc)
|
||||
{
|
||||
}
|
||||
|
||||
void SshRemoteProcessPrivate::setProcState(ProcessState newState)
|
||||
{
|
||||
#ifdef CREATOR_SSH_DEBUG
|
||||
qDebug("channel: old state = %d,new state = %d", m_procState, newState);
|
||||
#endif
|
||||
m_procState = newState;
|
||||
if (newState == StartFailed) {
|
||||
emit closed(SshRemoteProcess::FailedToStart);
|
||||
} else if (newState == Running) {
|
||||
m_wasRunning = true;
|
||||
emit started();
|
||||
}
|
||||
}
|
||||
|
||||
void SshRemoteProcessPrivate::closeHook()
|
||||
{
|
||||
if (m_wasRunning) {
|
||||
if (!m_signal.isEmpty())
|
||||
emit closed(SshRemoteProcess::KilledBySignal);
|
||||
else
|
||||
emit closed(SshRemoteProcess::ExitedNormally);
|
||||
}
|
||||
}
|
||||
|
||||
void SshRemoteProcessPrivate::handleOpenSuccessInternal()
|
||||
{
|
||||
foreach (const EnvVar &envVar, m_env) {
|
||||
m_sendFacility.sendEnvPacket(remoteChannel(), envVar.first,
|
||||
envVar.second);
|
||||
}
|
||||
|
||||
m_sendFacility.sendExecPacket(remoteChannel(), m_command);
|
||||
setProcState(ExecRequested);
|
||||
m_timeoutTimer->start(ReplyTimeout);
|
||||
}
|
||||
|
||||
void SshRemoteProcessPrivate::handleOpenFailureInternal()
|
||||
{
|
||||
setProcState(StartFailed);
|
||||
}
|
||||
|
||||
void SshRemoteProcessPrivate::handleChannelSuccess()
|
||||
{
|
||||
if (m_procState != ExecRequested) {
|
||||
throw SSH_SERVER_EXCEPTION(SSH_DISCONNECT_PROTOCOL_ERROR,
|
||||
"Unexpected SSH_MSG_CHANNEL_SUCCESS message.");
|
||||
}
|
||||
m_timeoutTimer->stop();
|
||||
setProcState(Running);
|
||||
}
|
||||
|
||||
void SshRemoteProcessPrivate::handleChannelFailure()
|
||||
{
|
||||
if (m_procState != ExecRequested) {
|
||||
throw SSH_SERVER_EXCEPTION(SSH_DISCONNECT_PROTOCOL_ERROR,
|
||||
"Unexpected SSH_MSG_CHANNEL_FAILURE message.");
|
||||
}
|
||||
m_timeoutTimer->stop();
|
||||
setProcState(StartFailed);
|
||||
closeChannel();
|
||||
}
|
||||
|
||||
void SshRemoteProcessPrivate::handleChannelDataInternal(const QByteArray &data)
|
||||
{
|
||||
emit outputAvailable(data);
|
||||
}
|
||||
|
||||
void SshRemoteProcessPrivate::handleChannelExtendedDataInternal(quint32 type,
|
||||
const QByteArray &data)
|
||||
{
|
||||
if (type != SSH_EXTENDED_DATA_STDERR)
|
||||
qWarning("Unknown extended data type %u", type);
|
||||
else
|
||||
emit errorOutputAvailable(data);
|
||||
}
|
||||
|
||||
void SshRemoteProcessPrivate::handleExitStatus(const SshChannelExitStatus &exitStatus)
|
||||
{
|
||||
#ifdef CREATOR_SSH_DEBUG
|
||||
qDebug("Process exiting with exit code %d", exitStatus.exitStatus);
|
||||
#endif
|
||||
m_exitCode = exitStatus.exitStatus;
|
||||
m_procState = Exited;
|
||||
}
|
||||
|
||||
void SshRemoteProcessPrivate::handleExitSignal(const SshChannelExitSignal &signal)
|
||||
{
|
||||
#ifdef CREATOR_SSH_DEBUG
|
||||
qDebug("Exit due to signal %s", signal.signal.data());
|
||||
#endif
|
||||
setError(signal.error);
|
||||
m_signal = signal.signal;
|
||||
m_procState = Exited;
|
||||
}
|
||||
|
||||
} // namespace Internal
|
||||
} // namespace Utils
|
||||
Reference in New Issue
Block a user