X-Git-Url: https://git.quassel-irc.org/?p=quassel.git;a=blobdiff_plain;f=src%2Fcommon%2Fsignalproxy.cpp;h=e7c641b4952460f1f7641e8cb1c75d099197befe;hp=d6c105b313049b880d54f5af2fd59b0f5ddc1d4e;hb=d9f4d3f0eabd3767b4d25438319116f77c158c52;hpb=06a03c2c69ee934aaeec83512bae2fffee83a340 diff --git a/src/common/signalproxy.cpp b/src/common/signalproxy.cpp index d6c105b3..e7c641b4 100644 --- a/src/common/signalproxy.cpp +++ b/src/common/signalproxy.cpp @@ -1,11 +1,11 @@ /*************************************************************************** - * Copyright (C) 2005-07 by The Quassel IRC Development Team * + * Copyright (C) 2005-09 by the Quassel Project * * devel@quassel-irc.org * * * * This program is free software; you can redistribute it and/or modify * * it under the terms of the GNU General Public License as published by * * the Free Software Foundation; either version 2 of the License, or * - * (at your option) any later version. * + * (at your option) version 3. * * * * This program is distributed in the hope that it will be useful, * * but WITHOUT ANY WARRANTY; without even the implied warranty of * @@ -20,83 +20,1227 @@ #include "signalproxy.h" -SignalProxy::SignalProxy(ProxyType _type, QIODevice *dev, QObject *parent) : QObject(parent), type(_type) { - if(dev) { - if(type != Client) { - qWarning() << tr("Device given for ProxyType == Server, ignoring...").toAscii(); +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#ifdef HAVE_SSL +#include +#endif +#include +#include +#include +#include + +#include "syncableobject.h" +#include "util.h" + +// ================================================== +// PeerSignalEvent +// ================================================== +class PeerSignalEvent : public QEvent { +public: + PeerSignalEvent(SignalProxy *sender, SignalProxy::RequestType requestType, const QVariantList ¶ms) : QEvent(QEvent::Type(SignalProxy::PeerSignal)), sender(sender), requestType(requestType), params(params) {} + SignalProxy *sender; + SignalProxy::RequestType requestType; + QVariantList params; +}; + +class RemovePeerEvent : public QEvent { +public: + RemovePeerEvent(QObject *peer) : QEvent(QEvent::Type(SignalProxy::RemovePeer)), peer(peer) {} + QObject *peer; +}; + +// ================================================== +// SignalRelay +// ================================================== +class SignalProxy::SignalRelay : public QObject { +/* Q_OBJECT is not necessary or even allowed, because we implement + qt_metacall ourselves (and don't use any other features of the meta + object system) +*/ +public: + SignalRelay(SignalProxy *parent) : QObject(parent), _proxy(parent) {} + inline SignalProxy *proxy() const { return _proxy; } + + int qt_metacall(QMetaObject::Call _c, int _id, void **_a); + + void attachSignal(QObject *sender, int signalId, const QByteArray &funcName); + void detachSignal(QObject *sender, int signalId = -1); + +private: + struct Signal { + QObject *sender; + int signalId; + QByteArray signature; + Signal(QObject *sender, int sigId, const QByteArray &signature) : sender(sender), signalId(sigId), signature(signature) {} + Signal() : sender(0), signalId(-1) {} + }; + + SignalProxy *_proxy; + QHash _slots; +}; + +void SignalProxy::SignalRelay::attachSignal(QObject *sender, int signalId, const QByteArray &funcName) { + // we ride without safetybelts here... all checking for valid method etc pp has to be done by the caller + // all connected methodIds are offset by the standard methodCount of QObject + int slotId; + for(int i = 0;; i++) { + if(!_slots.contains(i)) { + slotId = i; + break; + } + } + + QByteArray fn; + if(!funcName.isEmpty()) { + fn = QMetaObject::normalizedSignature(funcName); + } else { + fn = SIGNAL(fakeMethodSignature()); + fn = fn.replace("fakeMethodSignature()", sender->metaObject()->method(signalId).signature()); + } + + _slots[slotId] = Signal(sender, signalId, fn); + + QMetaObject::connect(sender, signalId, this, QObject::staticMetaObject.methodCount() + slotId); +} + +void SignalProxy::SignalRelay::detachSignal(QObject *sender, int signalId) { + QHash::iterator slotIter = _slots.begin(); + while(slotIter != _slots.end()) { + if(slotIter->sender == sender && (signalId == -1 || slotIter->signalId == signalId)) { + slotIter = _slots.erase(slotIter); + if(signalId != -1) + break; } else { - addPeer(dev); + slotIter++; + } + } +} + +int SignalProxy::SignalRelay::qt_metacall(QMetaObject::Call _c, int _id, void **_a) { + _id = QObject::qt_metacall(_c, _id, _a); + if(_id < 0) + return _id; + + if(_c == QMetaObject::InvokeMetaMethod) { + if(_slots.contains(_id)) { + QObject *caller = sender(); + + SignalProxy::ExtendedMetaObject *eMeta = proxy()->extendedMetaObject(caller->metaObject()); + Q_ASSERT(eMeta); + + const Signal &signal = _slots[_id]; + + QVariantList params; + params << signal.signature; + + const QList &argTypes = eMeta->argTypes(signal.signalId); + for(int i = 0; i < argTypes.size(); i++) { + if(argTypes[i] == 0) { + qWarning() << "SignalRelay::qt_metacall(): received invalid data for argument number" << i << "of signal" << QString("%1::%2").arg(caller->metaObject()->className()).arg(caller->metaObject()->method(_id).signature()); + qWarning() << " - make sure all your data types are known by the Qt MetaSystem"; + return _id; + } + params << QVariant(argTypes[i], _a[i+1]); + } + + proxy()->dispatchSignal(SignalProxy::RpcCall, params); } + _id -= _slots.count(); + } + return _id; +} + +// ================================================== +// Peers +// ================================================== +void SignalProxy::IODevicePeer::dispatchSignal(const RequestType &requestType, const QVariantList ¶ms) { + QVariantList packedFunc; + packedFunc << (qint16)requestType + << params; + dispatchPackedFunc(QVariant(packedFunc)); +} + +bool SignalProxy::IODevicePeer::isSecure() const { +#ifdef HAVE_SSL + QSslSocket *sslSocket = qobject_cast(_device); + if(sslSocket) + return sslSocket->isEncrypted() || sslSocket->localAddress() == QHostAddress::LocalHost || sslSocket->localAddress() == QHostAddress::LocalHostIPv6; +#endif + + QAbstractSocket *socket = qobject_cast(_device); + if(socket) + return socket->localAddress() == QHostAddress::LocalHost || socket->localAddress() == QHostAddress::LocalHostIPv6; + + return false; +} + +QString SignalProxy::IODevicePeer::address() const { + QAbstractSocket *socket = qobject_cast(_device); + if(socket) + return socket->peerAddress().toString(); + else + return QString(); +} + +void SignalProxy::SignalProxyPeer::dispatchSignal(const RequestType &requestType, const QVariantList ¶ms) { + Qt::ConnectionType type = QThread::currentThread() == receiver->thread() + ? Qt::DirectConnection + : Qt::QueuedConnection; + + if(type == Qt::DirectConnection) { + receiver->receivePeerSignal(sender, requestType, params); + } else { + QCoreApplication::postEvent(receiver, new PeerSignalEvent(sender, requestType, params)); } } +// ================================================== +// SignalProxy +// ================================================== +SignalProxy::SignalProxy(QObject* parent) + : QObject(parent) +{ + setProxyMode(Client); + init(); +} + +SignalProxy::SignalProxy(ProxyMode mode, QObject* parent) + : QObject(parent) +{ + setProxyMode(mode); + init(); +} + +SignalProxy::SignalProxy(ProxyMode mode, QIODevice* device, QObject* parent) + : QObject(parent) +{ + setProxyMode(mode); + addPeer(device); + init(); +} + SignalProxy::~SignalProxy() { - foreach(Connection conn, peers) { - conn.peer->deleteLater(); conn.device->deleteLater(); + QHash::iterator classIter = _syncSlave.begin(); + while(classIter != _syncSlave.end()) { + ObjectId::iterator objIter = classIter->begin(); + while(objIter != classIter->end()) { + SyncableObject *obj = objIter.value(); + objIter = classIter->erase(objIter); + obj->stopSynchronize(this); + } + classIter++; + } + _syncSlave.clear(); + + removeAllPeers(); +} + +void SignalProxy::setProxyMode(ProxyMode mode) { + PeerHash::iterator peer = _peers.begin(); + while(peer != _peers.end()) { + if((*peer)->type() != AbstractPeer::IODevicePeer) { + IODevicePeer *ioPeer = static_cast(*peer); + if(ioPeer->isOpen()) { + qWarning() << "SignalProxy: Cannot change proxy mode while connected"; + return; + } + } + if((*peer)->type() != AbstractPeer::SignalProxyPeer) { + qWarning() << "SignalProxy: Cannot change proxy mode while connected to another internal SignalProxy"; + return; + } + peer++; + } + + _proxyMode = mode; + if(mode == Server) + initServer(); + else + initClient(); +} + +void SignalProxy::init() { + _signalRelay = new SignalRelay(this); + connect(&_heartBeatTimer, SIGNAL(timeout()), this, SLOT(sendHeartBeat())); + setHeartBeatInterval(30); + setMaxHeartBeatCount(2); + _heartBeatTimer.start(); + _secure = false; + updateSecureState(); +} + +void SignalProxy::initServer() { +} + +void SignalProxy::initClient() { + attachSlot("__objectRenamed__", this, SLOT(objectRenamed(QByteArray, QString, QString))); +} + +bool SignalProxy::addPeer(QIODevice* iodev) { + if(!iodev) + return false; + + if(_peers.contains(iodev)) + return true; + + if(proxyMode() == Client && !_peers.isEmpty()) { + qWarning("SignalProxy: only one peer allowed in client mode!"); + return false; + } + + if(!iodev->isOpen()) { + qWarning("SignalProxy::addPeer(QIODevice *iodev): iodev needs to be open!"); + return false; + } + + connect(iodev, SIGNAL(disconnected()), this, SLOT(removePeerBySender())); + connect(iodev, SIGNAL(readyRead()), this, SLOT(dataAvailable())); + +#ifdef HAVE_SSL + QSslSocket *sslSocket = qobject_cast(iodev); + if(sslSocket) { + connect(iodev, SIGNAL(encrypted()), this, SLOT(updateSecureState())); + } +#endif + + if(!iodev->parent()) + iodev->setParent(this); + + _peers[iodev] = new IODevicePeer(iodev, iodev->property("UseCompression").toBool()); + + if(_peers.count() == 1) + emit connected(); + + updateSecureState(); + return true; +} + +void SignalProxy::setHeartBeatInterval(int secs) { + if(secs != _heartBeatInterval) { + _heartBeatInterval = secs; + _heartBeatTimer.setInterval(secs * 1000); + } +} + +void SignalProxy::setMaxHeartBeatCount(int max) { + _maxHeartBeatCount = max; +} + +bool SignalProxy::addPeer(SignalProxy* proxy) { + if(!proxy) + return false; + + if(proxyMode() == proxy->proxyMode()) { + qWarning() << "SignalProxy::addPeer(): adding a SignalProxy as peer requires one proxy to be server and one client!"; + return false; + } + + if(_peers.contains(proxy)) { + return true; + } + + if(proxyMode() == Client && !_peers.isEmpty()) { + qWarning("SignalProxy: only one peer allowed in client mode!"); + return false; + } + + _peers[proxy] = new SignalProxyPeer(this, proxy); + proxy->addPeer(this); + + if(_peers.count() == 1) + emit connected(); + + updateSecureState(); + return true; +} + +void SignalProxy::removeAllPeers() { + Q_ASSERT(proxyMode() == Server || _peers.count() <= 1); + // wee need to copy that list since we modify it in the loop + QList peers = _peers.keys(); + foreach(QObject *peer, peers) { + removePeer(peer); + } +} + +void SignalProxy::removePeer(QObject* dev) { + if(_peers.isEmpty()) { + qWarning() << "SignalProxy::removePeer(): No peers in use!"; + return; + } + + Q_ASSERT(dev); + if(!_peers.contains(dev)) { + qWarning() << "SignalProxy: unknown Peer" << dev; + return; + } + + AbstractPeer *peer = _peers[dev]; + _peers.remove(dev); + + disconnect(dev, 0, this, 0); + if(peer->type() == AbstractPeer::IODevicePeer) + emit peerRemoved(static_cast(dev)); + + if(peer->type() == AbstractPeer::SignalProxyPeer) { + SignalProxy *proxy = static_cast(dev); + if(proxy->_peers.contains(this)) + proxy->removePeer(this); + } + + if(dev->parent() == this) + dev->deleteLater(); + + delete peer; + + updateSecureState(); + + if(_peers.isEmpty()) + emit disconnected(); +} + +void SignalProxy::removePeerBySender() { + removePeer(sender()); +} + +void SignalProxy::renameObject(const SyncableObject *obj, const QString &newname, const QString &oldname) { + if(proxyMode() == Client) + return; + + const QMetaObject *meta = obj->syncMetaObject(); + const QByteArray className(meta->className()); + objectRenamed(className, newname, oldname); + + QVariantList params; + params << "__objectRenamed__" << className << newname << oldname; + dispatchSignal(RpcCall, params); +} + +void SignalProxy::objectRenamed(const QByteArray &classname, const QString &newname, const QString &oldname) { + if(_syncSlave.contains(classname) && _syncSlave[classname].contains(oldname) && oldname != newname) { + SyncableObject *obj = _syncSlave[classname][newname] = _syncSlave[classname].take(oldname); + requestInit(obj); + } +} + +const QMetaObject *SignalProxy::metaObject(const QObject *obj) { + if(const SyncableObject *syncObject = qobject_cast(obj)) + return syncObject->syncMetaObject(); + else + return obj->metaObject(); +} + +SignalProxy::ExtendedMetaObject *SignalProxy::extendedMetaObject(const QMetaObject *meta) const { + if(_extendedMetaObjects.contains(meta)) + return _extendedMetaObjects[meta]; + else + return 0; +} + +SignalProxy::ExtendedMetaObject *SignalProxy::createExtendedMetaObject(const QMetaObject *meta, bool checkConflicts) { + if(!_extendedMetaObjects.contains(meta)) { + _extendedMetaObjects[meta] = new ExtendedMetaObject(meta, checkConflicts); + } + return _extendedMetaObjects[meta]; +} + +bool SignalProxy::attachSignal(QObject *sender, const char* signal, const QByteArray& sigName) { + const QMetaObject* meta = sender->metaObject(); + QByteArray sig(meta->normalizedSignature(signal).mid(1)); + int methodId = meta->indexOfMethod(sig.constData()); + if(methodId == -1 || meta->method(methodId).methodType() != QMetaMethod::Signal) { + qWarning() << "SignalProxy::attachSignal(): No such signal" << signal; + return false; + } + + createExtendedMetaObject(meta); + _signalRelay->attachSignal(sender, methodId, sigName); + + disconnect(sender, SIGNAL(destroyed(QObject *)), this, SLOT(detachObject(QObject *))); + connect(sender, SIGNAL(destroyed(QObject *)), this, SLOT(detachObject(QObject *))); + return true; +} + +bool SignalProxy::attachSlot(const QByteArray& sigName, QObject* recv, const char* slot) { + const QMetaObject* meta = recv->metaObject(); + int methodId = meta->indexOfMethod(meta->normalizedSignature(slot).mid(1)); + if(methodId == -1 || meta->method(methodId).methodType() == QMetaMethod::Method) { + qWarning() << "SignalProxy::attachSlot(): No such slot" << slot; + return false; + } + + createExtendedMetaObject(meta); + + QByteArray funcName = QMetaObject::normalizedSignature(sigName.constData()); + _attachedSlots.insert(funcName, qMakePair(recv, methodId)); + + disconnect(recv, SIGNAL(destroyed(QObject *)), this, SLOT(detachObject(QObject *))); + connect(recv, SIGNAL(destroyed(QObject *)), this, SLOT(detachObject(QObject *))); + return true; +} + +void SignalProxy::synchronize(SyncableObject *obj) { + createExtendedMetaObject(obj, true); + + // attaching as slave to receive sync Calls + QByteArray className(obj->syncMetaObject()->className()); + _syncSlave[className][obj->objectName()] = obj; + + if(proxyMode() == Server) { + obj->setInitialized(); + emit objectInitialized(obj); + } else { + if(obj->isInitialized()) + emit objectInitialized(obj); + else + requestInit(obj); + } + + obj->synchronize(this); +} + +void SignalProxy::detachObject(QObject *obj) { + detachSignals(obj); + detachSlots(obj); +} + +void SignalProxy::detachSignals(QObject *sender) { + _signalRelay->detachSignal(sender); +} + +void SignalProxy::detachSlots(QObject* receiver) { + SlotHash::iterator slotIter = _attachedSlots.begin(); + while(slotIter != _attachedSlots.end()) { + if(slotIter.value().first == receiver) { + slotIter = _attachedSlots.erase(slotIter); + } else + slotIter++; + } +} + +void SignalProxy::stopSynchronize(SyncableObject *obj) { + // we can't use a className here, since it might be effed up, if we receive the call as a result of a decon + // gladly the objectName() is still valid. So we have only to iterate over the classes not each instance! *sigh* + QHash::iterator classIter = _syncSlave.begin(); + while(classIter != _syncSlave.end()) { + if(classIter->contains(obj->objectName()) && classIter.value()[obj->objectName()] == obj) { + classIter->remove(obj->objectName()); + break; + } + classIter++; + } + obj->stopSynchronize(this); +} + +void SignalProxy::dispatchSignal(const RequestType &requestType, const QVariantList ¶ms) { + QVariant packedFunc(QVariantList() << (qint16)requestType << params); + PeerHash::iterator peer = _peers.begin(); + while(peer != _peers.end()) { + switch((*peer)->type()) { + case AbstractPeer::IODevicePeer: + { + IODevicePeer *ioPeer = static_cast(*peer); + if(ioPeer->isOpen()) + ioPeer->dispatchPackedFunc(packedFunc); + else + QCoreApplication::postEvent(this, new RemovePeerEvent(peer.key())); + } + break; + case AbstractPeer::SignalProxyPeer: + (*peer)->dispatchSignal(requestType, params); + break; + default: + Q_ASSERT(false); // there shouldn't be any peers with wrong / unknown type + } + peer++; + } +} + +void SignalProxy::receivePackedFunc(AbstractPeer *sender, const QVariant &packedFunc) { + QVariantList params(packedFunc.toList()); + + if(params.isEmpty()) { + qWarning() << "SignalProxy::receivePeerSignal(): received incompatible Data:" << packedFunc; + return; + } + + RequestType requestType = (RequestType)params.takeFirst().value(); + receivePeerSignal(sender, requestType, params); +} + +void SignalProxy::receivePeerSignal(AbstractPeer *sender, const RequestType &requestType, const QVariantList ¶ms) { + switch(requestType) { + // list all RequestTypes that shouldnot trigger a heartbeat counter reset here + case HeartBeatReply: + break; + default: + if(sender->type() == AbstractPeer::IODevicePeer) { + IODevicePeer *ioPeer = static_cast(sender); + ioPeer->sentHeartBeats = 0; + } + } + + // qDebug() << "SignalProxy::receivePeerSignal)" << requestType << params; + switch(requestType) { + case RpcCall: + if(params.empty()) + qWarning() << "SignalProxy::receivePeerSignal(): received empty RPC-Call"; + else + handleSignal(params); + //handleSignal(params.takeFirst().toByteArray(), params); + break; + + case Sync: + handleSync(sender, params); + break; + + case InitRequest: + handleInitRequest(sender, params); + break; + + case InitData: + handleInitData(sender, params); + break; + + case HeartBeat: + receiveHeartBeat(sender, params); + break; + + case HeartBeatReply: + receiveHeartBeatReply(sender, params); + break; + + default: + qWarning() << "SignalProxy::receivePeerSignal(): received undefined CallType" << requestType << params; + } +} + +void SignalProxy::receivePeerSignal(SignalProxy *sender, const RequestType &requestType, const QVariantList ¶ms) { + if(!_peers.contains(sender)) { + // we output only the pointer value. otherwise Qt would try to pretty print. As the object might already been destroyed, this is not a good idea. + qWarning() << "SignalProxy::receivePeerSignal(): received Signal from unknown Proxy" << reinterpret_cast(sender); + return; + } + receivePeerSignal(_peers[sender], requestType, params); +} + +void SignalProxy::handleSync(AbstractPeer *sender, QVariantList params) { + if(params.count() < 3) { + qWarning() << "received invalid Sync call" << params; + return; + } + + QByteArray className = params.takeFirst().toByteArray(); + QString objectName = params.takeFirst().toString(); + QByteArray slot = params.takeFirst().toByteArray(); + + if(!_syncSlave.contains(className) || !_syncSlave[className].contains(objectName)) { + qWarning() << QString("no registered receiver for sync call: %1::%2 (objectName=\"%3\"). Params are:").arg(QString(className)).arg(QString(slot)).arg(objectName) + << params; + return; + } + + SyncableObject *receiver = _syncSlave[className][objectName]; + ExtendedMetaObject *eMeta = extendedMetaObject(receiver); + if(!eMeta->slotMap().contains(slot)) { + qWarning() << QString("no matching slot for sync call: %1::%2 (objectName=\"%3\"). Params are:").arg(QString(className)).arg(QString(slot)).arg(objectName) + << params; + return; + } + + int slotId = eMeta->slotMap()[slot]; + if(proxyMode() != eMeta->receiverMode(slotId)) { + qWarning("SignalProxy::handleSync(): invokeMethod for \"%s\" failed. Wrong ProxyMode!", eMeta->methodName(slotId).constData()); + return; + } + + QVariant returnValue((QVariant::Type)eMeta->returnType(slotId)); + if(!invokeSlot(receiver, slotId, params, returnValue)) { + qWarning("SignalProxy::handleSync(): invokeMethod for \"%s\" failed ", eMeta->methodName(slotId).constData()); + return; + } + + + if(returnValue.type() != QVariant::Invalid && eMeta->receiveMap().contains(slotId)) { + int receiverId = eMeta->receiveMap()[slotId]; + QVariantList returnParams; + returnParams << className + << objectName + << eMeta->methodName(receiverId); + //QByteArray(receiver->metaObject()->method(receiverId).signature()); + if(eMeta->argTypes(receiverId).count() > 1) + returnParams << params; + returnParams << returnValue; + sender->dispatchSignal(Sync, returnParams); + } + + // send emit update signal + invokeSlot(receiver, eMeta->updatedRemotelyId()); +} + +void SignalProxy::handleInitRequest(AbstractPeer *sender, const QVariantList ¶ms) { + if(params.count() != 2) { + qWarning() << "SignalProxy::handleInitRequest() received initRequest with invalid param Count:" + << params; + return; + } + + QByteArray className(params[0].toByteArray()); + QString objectName(params[1].toString()); + + if(!_syncSlave.contains(className)) { + qWarning() << "SignalProxy::handleInitRequest() received initRequest for unregistered Class:" + << className; + return; + } + + if(!_syncSlave[className].contains(objectName)) { + qWarning() << "SignalProxy::handleInitRequest() received initRequest for unregistered Object:" + << className << objectName; + return; } + + SyncableObject *obj = _syncSlave[className][objectName]; + + QVariantList params_; + params_ << className + << objectName + << initData(obj); + + sender->dispatchSignal(InitData, params_); } -void SignalProxy::addPeer(QIODevice *dev) { - if(type == Client && peers.count()) { - qWarning() << tr("Cannot add more than one peer to a SignalProxy in client mode!").toAscii(); +void SignalProxy::handleInitData(AbstractPeer *sender, const QVariantList ¶ms) { + Q_UNUSED(sender) + if(params.count() != 3) { + qWarning() << "SignalProxy::handleInitData() received initData with invalid param Count:" + << params; return; } - Connection conn; - conn.device = dev; - conn.peer = new QxtRPCPeer(dev, QxtRPCPeer::Peer, this); - connect(conn.peer, SIGNAL(peerDisconnected()), this, SLOT(socketDisconnected())); - foreach(SlotDesc slot, attachedSlots) { - conn.peer->attachSlot(slot.rpcFunction, slot.recv, slot.slot); + QByteArray className(params[0].toByteArray()); + QString objectName(params[1].toString()); + QVariantMap propertyMap(params[2].toMap()); + + if(!_syncSlave.contains(className)) { + qWarning() << "SignalProxy::handleInitData() received initData for unregistered Class:" + << className; + return; } - foreach(SignalDesc sig, attachedSignals) { - conn.peer->attachSignal(sig.sender, sig.signal, sig.rpcFunction); + + if(!_syncSlave[className].contains(objectName)) { + qWarning() << "SignalProxy::handleInitData() received initData for unregistered Object:" + << className << objectName; + return; } - peers.append(conn); + SyncableObject *obj = _syncSlave[className][objectName]; + setInitData(obj, propertyMap); } -void SignalProxy::socketDisconnected() { - for(int i = 0; i < peers.count(); i++) { - Connection conn = peers[i]; - QAbstractSocket *sock = qobject_cast(conn.device); - if(!sock) continue; - if(sock->state() == QAbstractSocket::UnconnectedState) { - peers[i].peer->deleteLater(); peers[i].device->deleteLater(); - peers.removeAt(i); - emit peerDisconnected(); - i--; +//void SignalProxy::handleSignal(const QByteArray &funcName, const QVariantList ¶ms) { +void SignalProxy::handleSignal(const QVariantList &data) { + QVariantList params = data; + QByteArray funcName = params.takeFirst().toByteArray(); + + QObject* receiver; + int methodId; + SlotHash::const_iterator slot = _attachedSlots.constFind(funcName); + while(slot != _attachedSlots.constEnd() && slot.key() == funcName) { + receiver = (*slot).first; + methodId = (*slot).second; + if(!invokeSlot(receiver, methodId, params)) { + ExtendedMetaObject *eMeta = extendedMetaObject(receiver); + qWarning("SignalProxy::handleSignal(): invokeMethod for \"%s\" failed ", eMeta->methodName(methodId).constData()); } + slot++; } } -void SignalProxy::attachSignal(QObject* sender, const char* signal, const QByteArray& rpcFunction) { - foreach(Connection conn, peers) { - conn.peer->attachSignal(sender, signal, rpcFunction); +bool SignalProxy::invokeSlot(QObject *receiver, int methodId, const QVariantList ¶ms, QVariant &returnValue) { + ExtendedMetaObject *eMeta = extendedMetaObject(receiver); + const QList args = eMeta->argTypes(methodId); + const int numArgs = params.count() < args.count() + ? params.count() + : args.count(); + + if(eMeta->minArgCount(methodId) > params.count()) { + qWarning() << "SignalProxy::invokeSlot(): not enough params to invoke" << eMeta->methodName(methodId); + return false; } - attachedSignals.append(SignalDesc(sender, signal, rpcFunction)); + void *_a[] = {0, // return type... + 0, 0, 0, 0 , 0, // and 10 args - that's the max size qt can handle with signals and slots + 0, 0, 0, 0 , 0}; + + // check for argument compatibility and build params array + for(int i = 0; i < numArgs; i++) { + if(!params[i].isValid()) { + qWarning() << "SignalProxy::invokeSlot(): received invalid data for argument number" << i << "of method" << QString("%1::%2()").arg(receiver->metaObject()->className()).arg(receiver->metaObject()->method(methodId).signature()); + qWarning() << " - make sure all your data types are known by the Qt MetaSystem"; + return false; + } + if(args[i] != QMetaType::type(params[i].typeName())) { + qWarning() << "SignalProxy::invokeSlot(): incompatible param types to invoke" << eMeta->methodName(methodId); + return false; + } + _a[i+1] = const_cast(params[i].constData()); + } + + if(returnValue.type() != QVariant::Invalid) + _a[0] = const_cast(returnValue.constData()); + + Qt::ConnectionType type = QThread::currentThread() == receiver->thread() + ? Qt::DirectConnection + : Qt::QueuedConnection; + + if(type == Qt::DirectConnection) { + return receiver->qt_metacall(QMetaObject::InvokeMetaMethod, methodId, _a) < 0; + } else { + qWarning() << "Queued Connections are not implemented yet"; + // note to self: qmetaobject.cpp:990 ff + return false; + } + +} + +bool SignalProxy::invokeSlot(QObject *receiver, int methodId, const QVariantList ¶ms) { + QVariant ret; + return invokeSlot(receiver, methodId, params, ret); } -void SignalProxy::attachSlot(const QByteArray& rpcFunction, QObject* recv, const char* slot) { - foreach(Connection conn, peers) { - conn.peer->attachSlot(rpcFunction, recv, slot); +void SignalProxy::dataAvailable() { + // yet again. it's a private slot. no need for checks. + QIODevice* ioDev = qobject_cast(sender()); + Q_ASSERT(_peers.contains(ioDev) && _peers[ioDev]->type() == AbstractPeer::IODevicePeer); + IODevicePeer *peer = static_cast(_peers[ioDev]); + QVariant var; + while(peer->readData(var)) + receivePackedFunc(peer, var); +} + +void SignalProxy::writeDataToDevice(QIODevice *dev, const QVariant &item, bool compressed) { + QAbstractSocket* sock = qobject_cast(dev); + if(!dev->isOpen() || (sock && sock->state()!=QAbstractSocket::ConnectedState)) { + qWarning("SignalProxy: Can't call write on a closed device"); + return; } - attachedSlots.append(SlotDesc(rpcFunction, recv, slot)); + + QByteArray block; + QDataStream out(&block, QIODevice::WriteOnly); + out.setVersion(QDataStream::Qt_4_2); + out << (quint32)0; + + if(compressed) { + QByteArray rawItem; + QDataStream itemStream(&rawItem, QIODevice::WriteOnly); + + itemStream.setVersion(QDataStream::Qt_4_2); + itemStream << item; + + rawItem = qCompress(rawItem); + + out << rawItem; + } else { + out << item; + } + + out.device()->seek(0); + out << (quint32)(block.size() - sizeof(quint32)); + + dev->write(block); } -void SignalProxy::detachObject(QObject* obj) { - Q_ASSERT(false); // not done yet - foreach(Connection conn, peers) { - conn.peer->detachObject(obj); +bool SignalProxy::readDataFromDevice(QIODevice *dev, quint32 &blockSize, QVariant &item, bool compressed) { + if(!dev) + return false; + + QDataStream in(dev); + in.setVersion(QDataStream::Qt_4_2); + + if(blockSize == 0) { + if(dev->bytesAvailable() < (int)sizeof(quint32)) return false; + in >> blockSize; + } + + if(blockSize > 1 << 22) { + disconnectDevice(dev, tr("Peer tried to send package larger than max package size!")); + return false; + } + + if(blockSize == 0) { + disconnectDevice(dev, tr("Peer tried to send 0 byte package!")); + return false; + } + + if(dev->bytesAvailable() < blockSize) + return false; + + blockSize = 0; + + if(compressed) { + QByteArray rawItem; + in >> rawItem; + + int nbytes = rawItem.size(); + if(nbytes <= 4) { + const char *data = rawItem.constData(); + if(nbytes < 4 || (data[0]!=0 || data[1]!=0 || data[2]!=0 || data[3]!=0)) { + disconnectDevice(dev, tr("Peer sent corrupted compressed data!")); + return false; + } + } + + rawItem = qUncompress(rawItem); + + QDataStream itemStream(&rawItem, QIODevice::ReadOnly); + itemStream.setVersion(QDataStream::Qt_4_2); + itemStream >> item; + } else { + in >> item; } - // FIXME: delete attached signal/slot info + if(!item.isValid()) { + disconnectDevice(dev, tr("Peer sent corrupt data: unable to load QVariant!")); + return false; + } + + return true; +} + +void SignalProxy::requestInit(SyncableObject *obj) { + if(proxyMode() == Server || obj->isInitialized()) + return; + + QVariantList params; + params << obj->syncMetaObject()->className() + << obj->objectName(); + dispatchSignal(InitRequest, params); +} + +QVariantMap SignalProxy::initData(SyncableObject *obj) const { + return obj->toVariantMap(); } -void SignalProxy::sendSignal(const char *signal, QVariant p1, QVariant p2, QVariant p3, QVariant p4, QVariant p5, QVariant p6, QVariant p7, QVariant p8, QVariant p9) { - foreach(Connection conn, peers) { - conn.peer->call(signal, p1, p2, p3, p4, p5, p6, p7, p8, p9); +void SignalProxy::setInitData(SyncableObject *obj, const QVariantMap &properties) { + if(obj->isInitialized()) + return; + obj->fromVariantMap(properties); + obj->setInitialized(); + emit objectInitialized(obj); + invokeSlot(obj, extendedMetaObject(obj)->updatedRemotelyId()); +} + +void SignalProxy::sendHeartBeat() { + QVariantList heartBeatParams; + heartBeatParams << QTime::currentTime(); + QList toClose; + + PeerHash::iterator peer = _peers.begin(); + while(peer != _peers.end()) { + if((*peer)->type() == AbstractPeer::IODevicePeer) { + IODevicePeer *ioPeer = static_cast(*peer); + ioPeer->dispatchSignal(SignalProxy::HeartBeat, heartBeatParams); + if(ioPeer->sentHeartBeats > 0) { + updateLag(ioPeer, ioPeer->sentHeartBeats * _heartBeatTimer.interval()); + } + if(maxHeartBeatCount() >= 0 && ioPeer->sentHeartBeats >= maxHeartBeatCount()) + toClose.append(ioPeer); + else + ioPeer->sentHeartBeats++; + } + ++peer; + } + + foreach(IODevicePeer *ioPeer, toClose) { + qWarning() << "SignalProxy: Disconnecting peer:" << ioPeer->address() + << "(didn't receive a heartbeat for over" << ioPeer->sentHeartBeats * _heartBeatTimer.interval() / 1000 << "seconds)"; + ioPeer->close(); + } +} + +void SignalProxy::receiveHeartBeat(AbstractPeer *peer, const QVariantList ¶ms) { + peer->dispatchSignal(SignalProxy::HeartBeatReply, params); +} + +void SignalProxy::receiveHeartBeatReply(AbstractPeer *peer, const QVariantList ¶ms) { + if(peer->type() != AbstractPeer::IODevicePeer) { + qWarning() << "SignalProxy::receiveHeartBeatReply: received heart beat from a non IODevicePeer!"; + return; + } + + IODevicePeer *ioPeer = static_cast(peer); + ioPeer->sentHeartBeats = 0; + + if(params.isEmpty()) { + qWarning() << "SignalProxy: received heart beat reply with less params then sent from:" << ioPeer->address(); + return; + } + + QTime sendTime = params[0].value(); + updateLag(ioPeer, sendTime.msecsTo(QTime::currentTime()) / 2); +} + +void SignalProxy::customEvent(QEvent *event) { + switch(event->type()) { + case PeerSignal: + { + PeerSignalEvent *e = static_cast(event); + receivePeerSignal(e->sender, e->requestType, e->params); + } + event->accept(); + break; + case RemovePeer: + { + RemovePeerEvent *e = static_cast(event); + removePeer(e->peer); + } + event->accept(); + default: + return; + } +} + +void SignalProxy::sync_call__(const SyncableObject *obj, SignalProxy::ProxyMode modeType, const char *funcname, va_list ap) { + // qDebug() << obj << modeType << "(" << _proxyMode << ")" << funcname; + if(modeType != _proxyMode) + return; + + ExtendedMetaObject *eMeta = extendedMetaObject(obj); + + QVariantList params; + params << eMeta->metaObject()->className() + << obj->objectName() + << QByteArray(funcname); + + const QList &argTypes = eMeta->argTypes(eMeta->methodId(QByteArray(funcname))); + + for(int i = 0; i < argTypes.size(); i++) { + if(argTypes[i] == 0) { + qWarning() << Q_FUNC_INFO << "received invalid data for argument number" << i << "of signal" << QString("%1::%2").arg(eMeta->metaObject()->className()).arg(funcname); + qWarning() << " - make sure all your data types are known by the Qt MetaSystem"; + return; + } + params << QVariant(argTypes[i], va_arg(ap, void *)); } + + dispatchSignal(Sync, params); +} + + + +void SignalProxy::disconnectDevice(QIODevice *dev, const QString &reason) { + if(!reason.isEmpty()) + qWarning() << qPrintable(reason); + QAbstractSocket *sock = qobject_cast(dev); + if(sock) + qWarning() << qPrintable(tr("Disconnecting")) << qPrintable(sock->peerAddress().toString()); + dev->close(); +} + +void SignalProxy::updateLag(IODevicePeer *peer, int lag) { + peer->lag = lag; + if(proxyMode() == Client) { + emit lagUpdated(lag); + } +} + +void SignalProxy::dumpProxyStats() { + QString mode; + if(proxyMode() == Server) + mode = "Server"; + else + mode = "Client"; + + int slaveCount = 0; + foreach(ObjectId oid, _syncSlave.values()) + slaveCount += oid.count(); + + qDebug() << this; + qDebug() << " Proxy Mode:" << mode; + qDebug() << " attached Slots:" << _attachedSlots.count(); + qDebug() << " number of synced Slaves:" << slaveCount; + qDebug() << "number of Classes cached:" << _extendedMetaObjects.count(); +} + +void SignalProxy::updateSecureState() { + bool wasSecure = _secure; + + _secure = !_peers.isEmpty(); + PeerHash::const_iterator peerIter; + for(peerIter = _peers.constBegin(); peerIter != _peers.constEnd(); peerIter++) { + _secure &= (*peerIter)->isSecure(); + } + + if(wasSecure != _secure) + emit secureStateChanged(_secure); +} + + +// ================================================== +// ExtendedMetaObject +// ================================================== +SignalProxy::ExtendedMetaObject::ExtendedMetaObject(const QMetaObject *meta, bool checkConflicts) + : _meta(meta), + _updatedRemotelyId(_meta->indexOfSignal("updatedRemotely()")) +{ + for(int i = 0; i < _meta->methodCount(); i++) { + if(_meta->method(i).methodType() != QMetaMethod::Slot) + continue; + + if(QByteArray(_meta->method(i).signature()).contains('*')) + continue; // skip methods with ptr params + + QByteArray method = methodName(_meta->method(i)); + if(method.startsWith("init")) + continue; // skip initializers + + if(_methodIds.contains(method)) { + /* funny... moc creates for methods containing default parameters multiple metaMethod with separate methodIds. + we don't care... we just need the full fledged version + */ + const QMetaMethod ¤t = _meta->method(_methodIds[method]); + const QMetaMethod &candidate = _meta->method(i); + if(current.parameterTypes().count() > candidate.parameterTypes().count()) { + int minCount = candidate.parameterTypes().count(); + QList commonParams = current.parameterTypes().mid(0, minCount); + if(commonParams == candidate.parameterTypes()) + continue; // we already got the full featured version + } else { + int minCount = current.parameterTypes().count(); + QList commonParams = candidate.parameterTypes().mid(0, minCount); + if(commonParams == current.parameterTypes()) { + _methodIds[method] = i; // use the new one + continue; + } + } + if(checkConflicts) { + qWarning() << "class" << meta->className() << "contains overloaded methods which is currently not supported!"; + qWarning() << " - " << _meta->method(i).signature() << "conflicts with" << _meta->method(_methodIds[method]).signature(); + } + continue; + } + _methodIds[method] = i; + } +} + +const SignalProxy::ExtendedMetaObject::MethodDescriptor &SignalProxy::ExtendedMetaObject::methodDescriptor(int methodId) { + if(!_methods.contains(methodId)) { + _methods[methodId] = MethodDescriptor(_meta->method(methodId)); + } + return _methods[methodId]; +} + +const QHash &SignalProxy::ExtendedMetaObject::receiveMap() { + if(_receiveMap.isEmpty()) { + QHash receiveMap; + + QMetaMethod requestSlot; + QByteArray returnTypeName; + QByteArray signature; + QByteArray methodName; + QByteArray params; + int paramsPos; + int receiverId; + const int methodCount = _meta->methodCount(); + for(int i = 0; i < methodCount; i++) { + requestSlot = _meta->method(i); + if(requestSlot.methodType() != QMetaMethod::Slot) + continue; + + returnTypeName = requestSlot.typeName(); + if(QMetaType::Void == (QMetaType::Type)returnType(i)) + continue; + + signature = QByteArray(requestSlot.signature()); + if(!signature.startsWith("request")) + continue; + + paramsPos = signature.indexOf('('); + if(paramsPos == -1) + continue; + + methodName = signature.left(paramsPos); + params = signature.mid(paramsPos); + + methodName = methodName.replace("request", "receive"); + params = params.left(params.count() - 1) + ", " + returnTypeName + ")"; + + signature = QMetaObject::normalizedSignature(methodName + params); + receiverId = _meta->indexOfSlot(signature); + + if(receiverId == -1) { + signature = QMetaObject::normalizedSignature(methodName + "(" + returnTypeName + ")"); + receiverId = _meta->indexOfSlot(signature); + } + + if(receiverId != -1) { + receiveMap[i] = receiverId; + } + } + _receiveMap = receiveMap; + } + return _receiveMap; +} + +QByteArray SignalProxy::ExtendedMetaObject::methodName(const QMetaMethod &method) { + QByteArray sig(method.signature()); + return sig.left(sig.indexOf("(")); +} + +QString SignalProxy::ExtendedMetaObject::methodBaseName(const QMetaMethod &method) { + QString methodname = QString(method.signature()).section("(", 0, 0); + + // determine where we have to chop: + int upperCharPos; + if(method.methodType() == QMetaMethod::Slot) { + // we take evertyhing from the first uppercase char if it's slot + upperCharPos = methodname.indexOf(QRegExp("[A-Z]")); + if(upperCharPos == -1) + return QString(); + methodname = methodname.mid(upperCharPos); + } else { + // and if it's a signal we discard everything from the last uppercase char + upperCharPos = methodname.lastIndexOf(QRegExp("[A-Z]")); + if(upperCharPos == -1) + return QString(); + methodname = methodname.left(upperCharPos); + } + + methodname[0] = methodname[0].toUpper(); + + return methodname; +} + +SignalProxy::ExtendedMetaObject::MethodDescriptor::MethodDescriptor(const QMetaMethod &method) + : _methodName(SignalProxy::ExtendedMetaObject::methodName(method)), + _returnType(QMetaType::type(method.typeName())) +{ + // determine argTypes + QList paramTypes = method.parameterTypes(); + QList argTypes; + for(int i = 0; i < paramTypes.count(); i++) { + argTypes.append(QMetaType::type(paramTypes[i])); + } + _argTypes = argTypes; + + // determine minArgCount + QString signature(method.signature()); + _minArgCount = method.parameterTypes().count() - signature.count("="); + + _receiverMode = (_methodName.startsWith("request")) + ? SignalProxy::Server + : SignalProxy::Client; }