278 lines
10 KiB
C++
278 lines
10 KiB
C++
/*
|
|
* Copyright © 2009 Rob Scheepmaker <r.scheepmaker@student.utwente.nl>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU Library General Public License version 2 as
|
|
* published by the Free Software Foundation
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details
|
|
*
|
|
* You should have received a copy of the GNU Library General Public
|
|
* License along with this program; if not, write to the
|
|
* Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#include "remoteservice_p.h"
|
|
#include "joliemessagehelper_p.h"
|
|
|
|
#include <plasma/accessmanager.h>
|
|
#include <plasma/authorizationinterface.h>
|
|
#include <plasma/authorizationmanager.h>
|
|
#include <plasma/clientpinrequest.h>
|
|
#include <plasma/service.h>
|
|
|
|
#include "authorizationmanager_p.h"
|
|
#include "remoteservicejob_p.h"
|
|
#include "service_p.h"
|
|
|
|
#include <QtCore/QBuffer>
|
|
#include <QtCore/QObject>
|
|
#include <QtCore/QTimer>
|
|
#include <QtCore/QUuid>
|
|
|
|
#include <QtJolie/Client>
|
|
#include <QtJolie/Message>
|
|
#include <QtJolie/PendingReply>
|
|
#include <QtJolie/PendingCall>
|
|
#include <QtJolie/PendingCallWatcher>
|
|
#include <QtJolie/Value>
|
|
|
|
#include <kdebug.h>
|
|
#include <kurl.h>
|
|
|
|
namespace Plasma
|
|
{
|
|
|
|
RemoteService::RemoteService(QObject* parent)
|
|
: Service(parent),
|
|
m_client(0),
|
|
m_ready(false),
|
|
m_busy(false)
|
|
{
|
|
}
|
|
|
|
RemoteService::RemoteService(QObject* parent, KUrl location)
|
|
: Service(parent),
|
|
m_location(location),
|
|
m_client(0),
|
|
m_ready(false),
|
|
m_busy(false)
|
|
{
|
|
if (AuthorizationManager::self()->d->myCredentials.isValid()) {
|
|
setLocation(location);
|
|
} else {
|
|
connect(AuthorizationManager::self(), SIGNAL(readyForRemoteAccess()),
|
|
this, SLOT(slotReadyForRemoteAccess()));
|
|
}
|
|
}
|
|
|
|
RemoteService::~RemoteService()
|
|
{
|
|
delete m_client;
|
|
}
|
|
|
|
void RemoteService::slotReadyForRemoteAccess()
|
|
{
|
|
kDebug() << "AuthorizationManager is now ready for remote access!";
|
|
setLocation(m_location);
|
|
}
|
|
|
|
void RemoteService::setLocation(const KUrl &location)
|
|
{
|
|
kDebug() << "Setting RemoteService location to " << location.prettyUrl();
|
|
|
|
m_uuid = QUuid::createUuid().toString();
|
|
Credentials identity = AuthorizationManager::self()->d->myCredentials;
|
|
if (!identity.canSign()) {
|
|
kDebug() << "we can't sign? how did we get here?";
|
|
return;
|
|
}
|
|
|
|
if (m_client && (m_location != location)) {
|
|
delete m_client;
|
|
m_client = new Jolie::Client(location.host(), location.port());
|
|
}
|
|
|
|
if (!m_client) {
|
|
m_client = new Jolie::Client(location.host(), location.port());
|
|
}
|
|
|
|
m_location = location;
|
|
|
|
QByteArray identityByteArray;
|
|
QDataStream stream(&identityByteArray, QIODevice::WriteOnly);
|
|
stream << identity.toPublicCredentials();
|
|
|
|
Jolie::Message getOpDesc(location.path(KUrl::RemoveTrailingSlash).remove(0, 1).toUtf8(),
|
|
"startConnection");
|
|
Jolie::Value data;
|
|
data.children(Message::Field::IDENTITY) << Jolie::Value(identityByteArray);
|
|
data.children(Message::Field::UUID) << Jolie::Value(m_uuid.toAscii());
|
|
getOpDesc.setData(data);
|
|
|
|
Jolie::PendingCall pendingReply = m_client->asyncCall(getOpDesc);
|
|
Jolie::PendingCallWatcher *watcher = new Jolie::PendingCallWatcher(pendingReply, this);
|
|
connect(watcher, SIGNAL(finished(Jolie::PendingCallWatcher*)),
|
|
this, SLOT(callCompleted(Jolie::PendingCallWatcher*)));
|
|
}
|
|
|
|
QString RemoteService::location() const
|
|
{
|
|
return m_location.prettyUrl();
|
|
}
|
|
|
|
void RemoteService::callCompleted(Jolie::PendingCallWatcher *watcher)
|
|
{
|
|
Jolie::PendingReply reply = *watcher;
|
|
Jolie::Message response = reply.reply();
|
|
|
|
if (response.operationName() == "startConnection") {
|
|
kDebug() << "Started connection: fetching .operations";
|
|
m_token = Message::field(Message::Field::TOKEN, response);
|
|
Jolie::Message getOpDesc(m_location.path(KUrl::RemoveTrailingSlash).remove(0, 1).toUtf8(),
|
|
"getOperations");
|
|
//TODO: async
|
|
Jolie::PendingCall pendingReply = m_client->asyncCall(signMessage(getOpDesc));
|
|
Jolie::PendingCallWatcher *watcher = new Jolie::PendingCallWatcher(pendingReply, this);
|
|
connect(watcher, SIGNAL(finished(Jolie::PendingCallWatcher*)),
|
|
this, SLOT(callCompleted(Jolie::PendingCallWatcher*)));
|
|
} else if (response.operationName() == "getOperations") {
|
|
if (response.fault().name() == Message::Error::REQUIREPIN) {
|
|
kDebug() << "pin required, request auth interface";
|
|
ClientPinRequest *request = new ClientPinRequest(this);
|
|
connect(request, SIGNAL(changed(Plasma::ClientPinRequest*)),
|
|
this, SLOT(slotGotPin(Plasma::ClientPinRequest*)));
|
|
AuthorizationManager::self()->d->authorizationInterface->clientPinRequest(*request);
|
|
} else {
|
|
kDebug() << "RemoteService is now ready for use!";
|
|
m_operationsScheme = Message::field(Message::Field::OPERATIONSDESCRIPTION, response);
|
|
m_token = Message::field(Message::Field::TOKEN, response);
|
|
m_ready = true;
|
|
setName(m_location.prettyUrl());
|
|
}
|
|
} else if (response.operationName() == "getEnabledOperations") {
|
|
//TODO: optimize.
|
|
m_token = Message::field(Message::Field::TOKEN, response);
|
|
QByteArray enabledOperations = Message::field(Message::Field::ENABLEDOPERATIONS, response);
|
|
QDataStream in(&enabledOperations, QIODevice::ReadOnly);
|
|
QStringList enabledOperationsList;
|
|
in >> enabledOperationsList;
|
|
|
|
foreach (const QString &operation, operationNames()) {
|
|
if (enabledOperationsList.contains(operation) && !isOperationEnabled(operation)) {
|
|
kDebug() << "yeah, we're enabling the operation with the name " << operation;
|
|
setOperationEnabled(operation, true);
|
|
} else if (!enabledOperationsList.contains(operation) && isOperationEnabled(operation)) {
|
|
kDebug() << "we're disabling the operation with the name " << operation;
|
|
setOperationEnabled(operation, false);
|
|
}
|
|
}
|
|
|
|
//if there's stuff in the queue, let it continue.
|
|
m_busy = false;
|
|
slotFinished();
|
|
} else {
|
|
kDebug() << "How did we end up here?";
|
|
}
|
|
}
|
|
|
|
void RemoteService::slotGotPin(Plasma::ClientPinRequest *request)
|
|
{
|
|
Jolie::Message getOpDesc(m_location.path(KUrl::RemoveTrailingSlash).remove(0, 1).toUtf8(),
|
|
"getOperations");
|
|
Jolie::Value value;
|
|
value.children(Message::Field::PARAMETERS) << Jolie::Value(QByteArray());
|
|
if (!request->pin().isEmpty()) {
|
|
value.children(Message::Field::PIN) << Jolie::Value(request->pin().toAscii());
|
|
}
|
|
getOpDesc.setData(value);
|
|
//TODO: async
|
|
Jolie::PendingCall pendingReply = m_client->asyncCall(signMessage(getOpDesc));
|
|
Jolie::PendingCallWatcher *watcher = new Jolie::PendingCallWatcher(pendingReply, this);
|
|
connect(watcher, SIGNAL(finished(Jolie::PendingCallWatcher*)),
|
|
this, SLOT(callCompleted(Jolie::PendingCallWatcher*)));
|
|
}
|
|
|
|
void RemoteService::registerOperationsScheme()
|
|
{
|
|
QBuffer buffer(&m_operationsScheme);
|
|
buffer.open(QBuffer::ReadWrite);
|
|
setOperationsScheme(&buffer);
|
|
QTimer *timer = new QTimer(this);
|
|
connect(timer, SIGNAL(timeout()), this, SLOT(slotUpdateEnabledOperations()));
|
|
//FIXME: get some sensible interval depending on the connection speed. This is kind of stupid.
|
|
timer->start(2000);
|
|
}
|
|
|
|
void RemoteService::slotUpdateEnabledOperations()
|
|
{
|
|
//TODO: maybe push the get enabled operations also on the queue?
|
|
if (!m_busy) {
|
|
m_busy = true;
|
|
Jolie::Message getOpDesc(m_location.path(KUrl::RemoveTrailingSlash).remove(0, 1).toUtf8(),
|
|
"getEnabledOperations");
|
|
|
|
Jolie::PendingCall pendingReply = m_client->asyncCall(signMessage(getOpDesc));
|
|
Jolie::PendingCallWatcher *watcher = new Jolie::PendingCallWatcher(pendingReply, this);
|
|
connect(watcher, SIGNAL(finished(Jolie::PendingCallWatcher*)),
|
|
this, SLOT(callCompleted(Jolie::PendingCallWatcher*)));
|
|
} else {
|
|
kDebug() << "We would like to update enabled operations, but are still busy so let's wait for now.";
|
|
}
|
|
}
|
|
|
|
ServiceJob* RemoteService::createJob(const QString& operation,
|
|
QMap<QString,QVariant>& parameters)
|
|
{
|
|
if (m_ready) {
|
|
ServiceJob *job = new RemoteServiceJob(m_location, destination(),
|
|
operation, parameters, m_token, this);
|
|
connect(job, SIGNAL(finished(KJob *)), this, SLOT(slotFinished()));
|
|
return job;
|
|
} else {
|
|
kWarning() << "We're not yet ready, so we're starting a NullServiceJob.";
|
|
kWarning() << "This means some plasmoid doesn't check for the serviceReady signal, which it should.";
|
|
return new NullServiceJob(destination(), operation, this);
|
|
}
|
|
}
|
|
|
|
void RemoteService::slotFinished()
|
|
{
|
|
if (m_queue.isEmpty()) {
|
|
return;
|
|
}
|
|
|
|
kDebug() << "Job finished, there are still service jobs in queue, starting next in queue.";
|
|
ServiceJob *job = m_queue.dequeue();
|
|
QTimer::singleShot(0, job, SLOT(slotStart()));
|
|
}
|
|
|
|
Jolie::Message RemoteService::signMessage(const Jolie::Message &message) const
|
|
{
|
|
Jolie::Message response(message);
|
|
|
|
Credentials identity = AuthorizationManager::self()->d->myCredentials;
|
|
if (!identity.isValid()) {
|
|
kDebug() << "We don't have our identity yet, just drop this message";
|
|
return response;
|
|
}
|
|
|
|
Jolie::Value data = response.data();
|
|
data.children(Message::Field::IDENTITYID) << Jolie::Value(identity.id().toAscii());
|
|
data.children(Message::Field::TOKEN) << Jolie::Value(m_token);
|
|
data.children(Message::Field::UUID) << Jolie::Value(m_uuid.toAscii());
|
|
response.setData(data);
|
|
data.children(Message::Field::SIGNATURE) <<
|
|
Jolie::Value(identity.signMessage(Message::payload(response)));
|
|
response.setData(data);
|
|
return response;
|
|
}
|
|
|
|
} //namespace Plasma
|
|
|
|
#include "remoteservice_p.moc"
|