X-Git-Url: https://git.quassel-irc.org/?p=quassel.git;a=blobdiff_plain;f=src%2Fcommon%2Fsignalproxy.cpp;h=c551c24697f7a63a23a9ae63b1076485f7441f5b;hp=1127005cbc2bbbb59d2f6d770a4b61f30fe50e9d;hb=a5dfcc8ecf8b81025d24b3c5c816169e3e030ea4;hpb=7df7429c009d2bea4c516e87f1ec4c956c722e09 diff --git a/src/common/signalproxy.cpp b/src/common/signalproxy.cpp index 1127005c..c551c246 100644 --- a/src/common/signalproxy.cpp +++ b/src/common/signalproxy.cpp @@ -23,6 +23,7 @@ #include #include #include +#include #include #include #include @@ -32,12 +33,34 @@ #include #include #include +#include +#include +#include #include "syncableobject.h" #include "util.h" -class SignalRelay: public QObject { +// ================================================== +// 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 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) @@ -50,12 +73,12 @@ public: void setSynchronize(bool); bool synchronize() const; - + int sigCount() const; - + private: bool isSyncMethod(int i); - + SignalProxy* proxy; QObject* caller; QMultiHash sigNames; @@ -80,17 +103,24 @@ int SignalRelay::qt_metacall(QMetaObject::Call _c, int _id, void **_a) { const QList &argTypes = proxy->argTypes(caller, _id); QVariantList params; int n = argTypes.size(); - for(int i=0; imetaObject()->className()).arg(caller->metaObject()->method(_id).signature()); + qWarning() << " - make sure all your data types are known by the Qt MetaSystem"; + return _id; + } + params.append(QVariant(argTypes[i], _a[i+1])); + } QMultiHash::const_iterator funcIter = sigNames.constFind(_id); while(funcIter != sigNames.constEnd() && funcIter.key() == _id) { - proxy->dispatchSignal(funcIter.value(), params); + proxy->dispatchSignal(SignalProxy::RpcCall, QVariantList() << funcIter.value() << params); funcIter++; } - + // dispatch Sync Signal if necessary QByteArray signature(caller->metaObject()->method(_id).signature()); - if(synchronize() && proxy->syncMap(qobject_cast(caller)).contains(signature)) { + SyncableObject *syncObject = qobject_cast(caller); + if(synchronize() && proxy->syncMap(syncObject).contains(signature)) { //qDebug() << "__SYNC__ >>>" // << caller->metaObject()->className() // << caller->objectName() @@ -98,18 +128,22 @@ int SignalRelay::qt_metacall(QMetaObject::Call _c, int _id, void **_a) { // << params; // params.prepend(QVariant(_id)); params.prepend(signature); - params.prepend(caller->objectName()); - params.prepend(caller->metaObject()->className()); - proxy->dispatchSignal((int)SignalProxy::Sync, params); + params.prepend(syncObject->objectName()); + params.prepend(syncObject->syncMetaObject()->className()); + proxy->dispatchSignal(SignalProxy::Sync, params); } } - _id -= 1; + _id -= QObject::staticMetaObject.methodCount(); } return _id; } void SignalRelay::setSynchronize(bool sync) { - const QMetaObject *meta = caller->metaObject(); + SyncableObject *syncObject = qobject_cast(caller); + if(!syncObject) + return; + + const QMetaObject *meta = syncObject->syncMetaObject(); if(!_sync && sync) { // enable Sync for(int i = 0; i < meta->methodCount(); i++ ) { @@ -127,10 +161,14 @@ void SignalRelay::setSynchronize(bool sync) { } bool SignalRelay::isSyncMethod(int i) { - QByteArray signature = caller->metaObject()->method(i).signature(); - if(!proxy->syncMap(qobject_cast(caller)).contains(signature)) + SyncableObject *syncObject = qobject_cast(caller); + if(!syncObject) return false; - + + QByteArray signature = syncObject->syncMetaObject()->method(i).signature(); + if(!proxy->syncMap(syncObject).contains(signature)) + return false; + if(proxy->proxyMode() == SignalProxy::Server && !signature.contains("Requested")) return true; @@ -163,24 +201,52 @@ void SignalRelay::attachSignal(int methodId, const QByteArray &func) { } sigNames.insert(methodId, fn); } -// ==================== -// END SIGNALRELAY -// ==================== +// ================================================== +// Peers +// ================================================== +void SignalProxy::IODevicePeer::dispatchSignal(const RequestType &requestType, const QVariantList ¶ms) { + QVariantList packedFunc; + packedFunc << (qint16)requestType + << params; + dispatchPackedFunc(QVariant(packedFunc)); +} + +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) @@ -188,21 +254,33 @@ SignalProxy::SignalProxy(ProxyMode mode, QIODevice* device, QObject* parent) { setProxyMode(mode); addPeer(device); -} + init(); +} SignalProxy::~SignalProxy() { QList senders = _relayHash.keys(); foreach(QObject* sender, senders) detachObject(sender); + removeAllPeers(); } void SignalProxy::setProxyMode(ProxyMode mode) { - foreach(QIODevice* peer, _peerByteCount.keys()) { - if(peer->isOpen()) { - qWarning() << "SignalProxy: Cannot change proxy mode while connected"; + 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(); @@ -210,8 +288,9 @@ void SignalProxy::setProxyMode(ProxyMode mode) { initClient(); } -SignalProxy::ProxyMode SignalProxy::proxyMode() const { - return _proxyMode; +void SignalProxy::init() { + connect(&_heartBeatTimer, SIGNAL(timeout()), this, SLOT(sendHeartBeat())); + _heartBeatTimer.start(30 * 1000); } void SignalProxy::initServer() { @@ -224,102 +303,131 @@ void SignalProxy::initClient() { bool SignalProxy::addPeer(QIODevice* iodev) { if(!iodev) return false; - - if(_peerByteCount.contains(iodev)) + + if(_peers.contains(iodev)) return true; - if(proxyMode() == Client && !_peerByteCount.isEmpty()) { + if(proxyMode() == Client && !_peers.isEmpty()) { qWarning("SignalProxy: only one peer allowed in client mode!"); return false; } - if(!iodev->isOpen()) - qWarning("SignalProxy::the device you passed is not open!"); + 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())); - QAbstractSocket* sock = qobject_cast(iodev); - if(sock) { - connect(sock, SIGNAL(disconnected()), this, SLOT(removePeerBySender())); - } + if(!iodev->parent()) + iodev->setParent(this); - _peerByteCount[iodev] = 0; + _peers[iodev] = new IODevicePeer(iodev, iodev->property("UseCompression").toBool()); - if(_peerByteCount.count() == 1) + if(_peers.count() == 1) emit connected(); return true; } -void SignalProxy::removePeerBySender() { - // OK we're brutal here... but since it's a private slot we know what we've got connected to it... - QIODevice *ioDev = (QIODevice *)(sender()); - removePeer(ioDev); - qDebug() << "Client disconnected."; -} +bool SignalProxy::addPeer(SignalProxy* proxy) { + if(!proxy) + return false; -void SignalProxy::objectRenamed(QString oldname, QString newname) { - const QMetaObject *meta = sender()->metaObject(); - const QByteArray className(meta->className()); - objectRenamed(className, oldname, newname); + if(proxyMode() == proxy->proxyMode()) { + qWarning() << "SignalProxy::addPeer(): adding a SignalProxy as peer requires one proxy to be server and one client!"; + return false; + } - if(proxyMode() == Client) - return; - - QVariantList params; - params << className << oldname << newname; - dispatchSignal("__objectRenamed__", params); -} + 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); -void SignalProxy::objectRenamed(QByteArray classname, QString oldname, QString newname) { - if(_syncSlave.contains(classname) && _syncSlave[classname].contains(oldname) && oldname != newname) - _syncSlave[classname][newname] = _syncSlave[classname].take(oldname); + if(_peers.count() == 1) + emit connected(); + + 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(QIODevice* iodev) { - if(_peerByteCount.isEmpty()) { - qWarning() << "SignalProxy: No peers in use!"; +void SignalProxy::removePeer(QObject* dev) { + if(_peers.isEmpty()) { + qWarning() << "SignalProxy::removePeer(): No peers in use!"; return; } - if(proxyMode() == Server && !iodev) { - // disconnect all - QList peers = _peerByteCount.keys(); - foreach(QIODevice *peer, peers) - removePeer(peer); + Q_ASSERT(dev); + if(!_peers.contains(dev)) { + qWarning() << "SignalProxy: unknown Peer" << dev; + return; } - if(proxyMode() != Server && !iodev) - iodev = _peerByteCount.keys().first(); + AbstractPeer *peer = _peers[dev]; + _peers.remove(dev); - Q_ASSERT(iodev); + disconnect(dev, 0, this, 0); + if(peer->type() == AbstractPeer::IODevicePeer) + emit peerRemoved(static_cast(dev)); - if(!_peerByteCount.contains(iodev)) { - qWarning() << "SignalProxy: unknown QIODevice" << iodev; - return; + if(peer->type() == AbstractPeer::SignalProxyPeer) { + SignalProxy *proxy = static_cast(dev); + if(proxy->_peers.contains(this)) + proxy->removePeer(this); } - - // take a last gasp - while(true) { - QVariant var; - if(readDataFromDevice(iodev, _peerByteCount[iodev], var)) - receivePeerSignal(iodev, var); - else - break; - } - _peerByteCount.remove(iodev); - disconnect(iodev, 0, this, 0); - emit peerRemoved(iodev); + if(dev->parent() == this) + dev->deleteLater(); + + delete peer; - if(_peerByteCount.isEmpty()) + if(_peers.isEmpty()) emit disconnected(); } +void SignalProxy::removePeerBySender() { + removePeer(sender()); +} + +void SignalProxy::objectRenamed(const QString &newname, const QString &oldname) { + SyncableObject *syncObject = qobject_cast(sender()); + const QMetaObject *meta = syncObject->metaObject(); + const QByteArray className(meta->className()); + objectRenamed(className, newname, oldname); + + if(proxyMode() == Client) + return; + + 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); + } +} + void SignalProxy::setArgTypes(QObject* obj, int methodId) { - const QMetaObject *meta = obj->metaObject(); + const QMetaObject *meta = metaObject(obj); QList p = meta->method(methodId).parameterTypes(); QList argTypes; int ct = p.count(); @@ -331,37 +439,80 @@ void SignalProxy::setArgTypes(QObject* obj, int methodId) { } const QList &SignalProxy::argTypes(QObject *obj, int methodId) { - Q_ASSERT(_classInfo.contains(obj->metaObject())); - if(!_classInfo[obj->metaObject()]->argTypes.contains(methodId)) + const QMetaObject *meta = metaObject(obj); + Q_ASSERT(_classInfo.contains(meta)); + if(!_classInfo[meta]->argTypes.contains(methodId)) setArgTypes(obj, methodId); - return _classInfo[obj->metaObject()]->argTypes[methodId]; + return _classInfo[meta]->argTypes[methodId]; +} + +void SignalProxy::setReturnType(QObject *obj, int methodId) { + const QMetaObject *meta = metaObject(obj); + int returnType = QMetaType::type(meta->method(methodId).typeName()); + + Q_ASSERT(!_classInfo[meta]->returnType.contains(methodId)); + _classInfo[meta]->returnType[methodId] = returnType; +} + +const int &SignalProxy::returnType(QObject *obj, int methodId) { + const QMetaObject *meta = metaObject(obj); + Q_ASSERT(_classInfo.contains(meta)); + if(!_classInfo[meta]->returnType.contains(methodId)) + setReturnType(obj, methodId); + return _classInfo[meta]->returnType[methodId]; +} + +void SignalProxy::setMinArgCount(QObject *obj, int methodId) { + const QMetaObject *meta = metaObject(obj); + QString signature(meta->method(methodId).signature()); + int minCount = meta->method(methodId).parameterTypes().count() - signature.count("="); + Q_ASSERT(!_classInfo[meta]->minArgCount.contains(methodId)); + _classInfo[meta]->minArgCount[methodId] = minCount; +} + +const int &SignalProxy::minArgCount(QObject *obj, int methodId) { + const QMetaObject *meta = metaObject(obj); + Q_ASSERT(_classInfo.contains(meta)); + if(!_classInfo[meta]->minArgCount.contains(methodId)) + setMinArgCount(obj, methodId); + return _classInfo[meta]->minArgCount[methodId]; } void SignalProxy::setMethodName(QObject *obj, int methodId) { - const QMetaObject *meta = obj->metaObject(); + const QMetaObject *meta = metaObject(obj); QByteArray method(::methodName(meta->method(methodId))); Q_ASSERT(!_classInfo[meta]->methodNames.contains(methodId)); _classInfo[meta]->methodNames[methodId] = method; } const QByteArray &SignalProxy::methodName(QObject *obj, int methodId) { - Q_ASSERT(_classInfo.contains(obj->metaObject())); - if(!_classInfo[obj->metaObject()]->methodNames.contains(methodId)) + const QMetaObject *meta = metaObject(obj); + Q_ASSERT(_classInfo.contains(meta)); + if(!_classInfo[meta]->methodNames.contains(methodId)) setMethodName(obj, methodId); - return _classInfo[obj->metaObject()]->methodNames[methodId]; + return _classInfo[meta]->methodNames[methodId]; } void SignalProxy::setSyncMap(SyncableObject *obj) { - const QMetaObject *meta = obj->metaObject(); + const QMetaObject *meta = obj->syncMetaObject(); QHash syncMap; - + QList slotIndexes; for(int i = 0; i < meta->methodCount(); i++) { if(meta->method(i).methodType() == QMetaMethod::Slot) slotIndexes << i; } + // we're faking sync pairs for sync replies + QByteArray slotSignature; + foreach(int slotIdx, slotIndexes) { + slotSignature = QByteArray(meta->method(slotIdx).signature()); + if(!slotSignature.startsWith("receive")) + continue; + syncMap[slotSignature] = slotIdx; + } + QMetaMethod signal, slot; int matchIdx; for(int signalIdx = 0; signalIdx < meta->methodCount(); signalIdx++) { @@ -388,34 +539,101 @@ void SignalProxy::setSyncMap(SyncableObject *obj) { } const QHash &SignalProxy::syncMap(SyncableObject *obj) { - Q_ASSERT(_classInfo.contains(obj->metaObject())); - if(_classInfo[obj->metaObject()]->syncMap.isEmpty()) + const QMetaObject *meta = obj->syncMetaObject(); + Q_ASSERT(_classInfo.contains(meta)); + if(_classInfo[meta]->syncMap.isEmpty()) setSyncMap(obj); - return _classInfo[obj->metaObject()]->syncMap; + return _classInfo[meta]->syncMap; +} + +void SignalProxy::setReceiveMap(SyncableObject *obj) { + const QMetaObject *meta = obj->syncMetaObject(); + Q_ASSERT(_classInfo.contains(meta)); + + 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(obj, 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; + } + _classInfo[meta]->receiveMap = receiveMap; +} + +const QHash &SignalProxy::receiveMap(SyncableObject *obj) { + const QMetaObject *meta = obj->syncMetaObject(); + Q_ASSERT(_classInfo.contains(meta)); + if(_classInfo[meta]->receiveMap.isEmpty()) + setReceiveMap(obj); + return _classInfo[meta]->receiveMap; } -void SignalProxy::setUpdatedRemotelyId(QObject *obj) { - const QMetaObject *meta = obj->metaObject(); +void SignalProxy::setUpdatedRemotelyId(SyncableObject *obj) { + const QMetaObject *meta = obj->syncMetaObject(); Q_ASSERT(_classInfo.contains(meta)); _classInfo[meta]->updatedRemotelyId = meta->indexOfSignal("updatedRemotely()"); } int SignalProxy::updatedRemotelyId(SyncableObject *obj) { - Q_ASSERT(_classInfo.contains(obj->metaObject())); - return _classInfo[obj->metaObject()]->updatedRemotelyId; + Q_ASSERT(_classInfo.contains(obj->syncMetaObject())); + return _classInfo[obj->syncMetaObject()]->updatedRemotelyId; +} + +const QMetaObject *SignalProxy::metaObject(QObject *obj) { + if(SyncableObject *syncObject = qobject_cast(obj)) + return syncObject->syncMetaObject(); + else + return obj->metaObject(); } void SignalProxy::createClassInfo(QObject *obj) { - if(_classInfo.contains(obj->metaObject())) + const QMetaObject *meta = metaObject(obj); + if(_classInfo.contains(meta)) return; ClassInfo *classInfo = new ClassInfo(); - _classInfo[obj->metaObject()] = classInfo; - setUpdatedRemotelyId(obj); + _classInfo[meta] = classInfo; } bool SignalProxy::attachSignal(QObject* sender, const char* signal, const QByteArray& sigName) { - const QMetaObject* meta = sender->metaObject(); + const QMetaObject* meta = metaObject(sender); QByteArray sig(meta->normalizedSignature(signal).mid(1)); int methodId = meta->indexOfMethod(sig.constData()); if(methodId == -1 || meta->method(methodId).methodType() != QMetaMethod::Signal) { @@ -457,6 +675,7 @@ bool SignalProxy::attachSlot(const QByteArray& sigName, QObject* recv, const cha void SignalProxy::synchronize(SyncableObject *obj) { createClassInfo(obj); + setUpdatedRemotelyId(obj); // attaching all the Signals SignalRelay* relay; @@ -468,36 +687,29 @@ void SignalProxy::synchronize(SyncableObject *obj) { relay->setSynchronize(true); // attaching as slave to receive sync Calls - QByteArray className(obj->metaObject()->className()); + QByteArray className(obj->syncMetaObject()->className()); _syncSlave[className][obj->objectName()] = obj; if(proxyMode() == Server) { - if(obj->metaObject()->indexOfSignal(QMetaObject::normalizedSignature("renameObject(QString, QString)")) != -1) - connect(obj, SIGNAL(renameObject(QString, QString)), this, SLOT(objectRenamed(QString, QString))); - - setInitialized(obj); + connect(obj, SIGNAL(objectRenamed(QString, QString)), this, SLOT(objectRenamed(QString, QString))); + obj->setInitialized(); + emit objectInitialized(obj); } else { - requestInit(obj); + if(obj->isInitialized()) + emit objectInitialized(obj); + else + requestInit(obj); } } -void SignalProxy::setInitialized(SyncableObject *obj) { - obj->setInitialized(); - emit objectInitialized(obj); -} - -bool SignalProxy::isInitialized(SyncableObject *obj) const { - return obj->isInitialized(); -} - void SignalProxy::requestInit(SyncableObject *obj) { - if(proxyMode() == Server || isInitialized(obj)) + if(proxyMode() == Server || obj->isInitialized()) return; QVariantList params; - params << obj->metaObject()->className() + params << obj->syncMetaObject()->className() << obj->objectName(); - dispatchSignal((int)InitRequest, params); + dispatchSignal(InitRequest, params); } void SignalProxy::detachSender() { @@ -505,9 +717,9 @@ void SignalProxy::detachSender() { } void SignalProxy::detachObject(QObject* obj) { + stopSync(static_cast(obj)); detachSignals(obj); detachSlots(obj); - stopSync(static_cast(obj)); } void SignalProxy::detachSignals(QObject* sender) { @@ -534,7 +746,7 @@ void SignalProxy::stopSync(SyncableObject* obj) { // 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())) { + if(classIter->contains(obj->objectName()) && classIter.value()[obj->objectName()] == obj) { classIter->remove(obj->objectName()); break; } @@ -542,83 +754,143 @@ void SignalProxy::stopSync(SyncableObject* obj) { } } -void SignalProxy::dispatchSignal(QIODevice *receiver, const QVariant &identifier, const QVariantList ¶ms) { - QVariantList packedFunc; - packedFunc << identifier; - packedFunc << params; - writeDataToDevice(receiver, QVariant(packedFunc)); -} - -void SignalProxy::dispatchSignal(const QVariant &identifier, const QVariantList ¶ms) { - // yes I know we have a little code duplication here... it's for the sake of performance - QVariantList packedFunc; - packedFunc << identifier; - packedFunc << params; - foreach(QIODevice* dev, _peerByteCount.keys()) - writeDataToDevice(dev, QVariant(packedFunc)); +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::receivePeerSignal(QIODevice *sender, const QVariant &packedFunc) { +void SignalProxy::receivePackedFunc(AbstractPeer *sender, const QVariant &packedFunc) { QVariantList params(packedFunc.toList()); - QVariant call = params.takeFirst(); - if(call.type() != QVariant::Int) - return handleSignal(call.toByteArray(), params); + 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) { + case RpcCall: + if(params.empty()) + qWarning() << "SignalProxy::receivePeerSignal(): received empty RPC-Call"; + else + handleSignal(params); + //handleSignal(params.takeFirst().toByteArray(), params); + break; - switch(call.toInt()) { case Sync: - return handleSync(params); + handleSync(sender, params); + break; + case InitRequest: - return handleInitRequest(sender, params); + handleInitRequest(sender, params); + break; + case InitData: - return handleInitData(sender, params); + handleInitData(sender, params); + break; + + case HeartBeat: + receiveHeartBeat(sender, params); + break; + + case HeartBeatReply: + receiveHeartBeatReply(sender, params); + break; + default: - qWarning() << "received undefined CallType" << call.toInt(); + 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(QVariantList 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 signal = params.takeFirst().toByteArray(); if(!_syncSlave.contains(className) || !_syncSlave[className].contains(objectName)) { - qWarning() << QString("no registered receiver for sync call: %s::%s (objectName=\"%s\"). Params are:").arg(QString(className)).arg(QString(signal)).arg(objectName) + qWarning() << QString("no registered receiver for sync call: %1::%2 (objectName=\"%3\"). Params are:").arg(QString(className)).arg(QString(signal)).arg(objectName) << params; return; } SyncableObject *receiver = _syncSlave[className][objectName]; if(!syncMap(receiver).contains(signal)) { - qWarning() << QString("no matching slot for sync call: %s::%s (objectName=\"%s\"). Params are:").arg(QString(className)).arg(QString(signal)).arg(objectName) + qWarning() << QString("no matching slot for sync call: %1::%2 (objectName=\"%3\"). Params are:").arg(QString(className)).arg(QString(signal)).arg(objectName) << params; return; } int slotId = syncMap(receiver)[signal]; - if(!invokeSlot(receiver, slotId, params)) { + + QVariant returnValue((QVariant::Type)returnType(receiver, slotId)); + if(!invokeSlot(receiver, slotId, params, returnValue)) { qWarning("SignalProxy::handleSync(): invokeMethod for \"%s\" failed ", methodName(receiver, slotId).constData()); return; } + + if(returnValue.type() != QVariant::Invalid && receiveMap(receiver).contains(slotId)) { + int receiverId = receiveMap(receiver)[slotId]; + QVariantList returnParams; + returnParams << className + << objectName + << QByteArray(receiver->metaObject()->method(receiverId).signature()); + if(argTypes(receiver, receiverId).count() > 1) + returnParams << params; + returnParams << returnValue; + sender->dispatchSignal(Sync, returnParams); + } + + // send emit update signal invokeSlot(receiver, updatedRemotelyId(receiver)); } -void SignalProxy::handleInitRequest(QIODevice *sender, const QVariantList ¶ms) { +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; @@ -630,25 +902,25 @@ void SignalProxy::handleInitRequest(QIODevice *sender, const QVariantList ¶m << className << objectName; return; } - + SyncableObject *obj = _syncSlave[className][objectName]; QVariantList params_; - params_ << obj->metaObject()->className() - << obj->objectName() + params_ << className + << objectName << initData(obj); - dispatchSignal(sender, (int)InitData, params_); + sender->dispatchSignal(InitData, params_); } -void SignalProxy::handleInitData(QIODevice *sender, const QVariantList ¶ms) { +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; } - + QByteArray className(params[0].toByteArray()); QString objectName(params[1].toString()); QVariantMap propertyMap(params[2].toMap()); @@ -669,7 +941,11 @@ void SignalProxy::handleInitData(QIODevice *sender, const QVariantList ¶ms) setInitData(obj, propertyMap); } -void SignalProxy::handleSignal(const QByteArray &funcName, const QVariantList ¶ms) { +//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); @@ -682,19 +958,28 @@ void SignalProxy::handleSignal(const QByteArray &funcName, const QVariantList &p } } -bool SignalProxy::invokeSlot(QObject *receiver, int methodId, const QVariantList ¶ms) { +bool SignalProxy::invokeSlot(QObject *receiver, int methodId, const QVariantList ¶ms, QVariant &returnValue) { const QList args = argTypes(receiver, methodId); - const int numArgs = args.count(); + const int numArgs = params.count() < args.count() + ? params.count() + : args.count(); - if(numArgs < params.count()) { + if(minArgCount(receiver, methodId) > params.count()) { qWarning() << "SignalProxy::invokeSlot(): not enough params to invoke" << methodName(receiver, methodId); return false; } - - void *_a[numArgs+1]; - _a[0] = 0; + + 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" << methodName(receiver, methodId); return false; @@ -702,7 +987,9 @@ bool SignalProxy::invokeSlot(QObject *receiver, int methodId, const QVariantList _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; @@ -711,36 +998,60 @@ bool SignalProxy::invokeSlot(QObject *receiver, int methodId, const QVariantList return receiver->qt_metacall(QMetaObject::InvokeMetaMethod, methodId, _a) < 0; } else { qWarning() << "Queued Connections are not implemented yet"; - // not to self: qmetaobject.cpp:990 ff + // 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::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(readDataFromDevice(ioDev, _peerByteCount[ioDev], var)) - receivePeerSignal(ioDev, var); + while(peer->readData(var)) + receivePackedFunc(peer, var); } -void SignalProxy::writeDataToDevice(QIODevice *dev, const QVariant &item) { +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 on a closed device"); + qWarning("SignalProxy: Can't call write on a closed device"); return; } + QByteArray block; QDataStream out(&block, QIODevice::WriteOnly); out.setVersion(QDataStream::Qt_4_2); - out << (quint32)0 << item; + 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); } -bool SignalProxy::readDataFromDevice(QIODevice *dev, quint32 &blockSize, QVariant &item) { +bool SignalProxy::readDataFromDevice(QIODevice *dev, quint32 &blockSize, QVariant &item, bool compressed) { QDataStream in(dev); in.setVersion(QDataStream::Qt_4_2); @@ -749,16 +1060,55 @@ bool SignalProxy::readDataFromDevice(QIODevice *dev, quint32 &blockSize, QVarian in >> blockSize; } + if(blockSize > 1 << 22) { + disconnectDevice(dev, tr("Client tried to send package larger than max package size!")); + return false; + } + + if(blockSize == 0) { + disconnectDevice(dev, tr("Client tried to send 0 byte package!")); + return false; + } + if(dev->bytesAvailable() < blockSize) return false; - in >> item; + 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("Client 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; + } + + if(!item.isValid()) { + disconnectDevice(dev, tr("Client sent corrupt data: unable to load QVariant!")); + return false; + } + return true; } bool SignalProxy::methodsMatch(const QMetaMethod &signal, const QMetaMethod &slot) const { // if we don't even have the same basename it's a sure NO - if(methodBaseName(signal) != methodBaseName(slot)) + QString baseName = methodBaseName(signal); + if(baseName != methodBaseName(slot)) return false; // are the signatures compatible? @@ -766,11 +1116,20 @@ bool SignalProxy::methodsMatch(const QMetaMethod &signal, const QMetaMethod &slo return false; // we take an educated guess if the signals and slots match - QString signalsuffix = ::methodName(signal).mid(QString(::methodName(signal)).lastIndexOf(QRegExp("[A-Z]"))).toLower(); - QString slotprefix = ::methodName(slot).left(QString(::methodName(slot)).indexOf(QRegExp("[A-Z]"))).toLower(); - + QString signalsuffix = ::methodName(signal); + QString slotprefix = ::methodName(slot); + if(!baseName.isEmpty()) { + signalsuffix = signalsuffix.mid(baseName.count()).toLower(); + slotprefix = slotprefix.left(slotprefix.count() - baseName.count()).toLower(); + } + uint sizediff = qAbs(slotprefix.size() - signalsuffix.size()); int ratio = editingDistance(slotprefix, signalsuffix) - sizediff; +// if(ratio < 2) { +// qDebug() << Q_FUNC_INFO; +// qDebug() << methodBaseName(signal) << methodBaseName(slot); +// qDebug() << signalsuffix << slotprefix << sizediff << ratio; +// } return (ratio < 2); } @@ -778,12 +1137,19 @@ QString SignalProxy::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 - methodname = methodname.mid(methodname.indexOf(QRegExp("[A-Z]"))); + 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 - methodname = methodname.left(methodname.lastIndexOf(QRegExp("[A-Z]"))); + upperCharPos = methodname.lastIndexOf(QRegExp("[A-Z]")); + if(upperCharPos == -1) + return QString(); + methodname = methodname.left(upperCharPos); } methodname[0] = methodname[0].toUpper(); @@ -796,12 +1162,96 @@ QVariantMap SignalProxy::initData(SyncableObject *obj) const { } void SignalProxy::setInitData(SyncableObject *obj, const QVariantMap &properties) { - if(isInitialized(obj)) + if(obj->isInitialized()) return; obj->fromVariantMap(properties); - setInitialized(obj); + obj->setInitialized(); + emit objectInitialized(obj); + invokeSlot(obj, updatedRemotelyId(obj)); } +void SignalProxy::sendHeartBeat() { + QVariantList heartBeatParams; + heartBeatParams << QTime::currentTime(); + 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(ioPeer->sentHeartBeats > 1) { + qWarning() << "SignalProxy: Disconnecting peer:" << ioPeer->address() + << "(didn't receive a heartbeat for over" << ioPeer->sentHeartBeats * _heartBeatTimer.interval() / 1000 << "seconds)"; + ioPeer->close(); + } else { + ioPeer->sentHeartBeats++; + } + } + peer++; + } +} + +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::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) @@ -816,7 +1266,7 @@ void SignalProxy::dumpProxyStats() { int slaveCount = 0; foreach(ObjectId oid, _syncSlave.values()) slaveCount += oid.count(); - + qDebug() << this; qDebug() << " Proxy Mode:" << mode; qDebug() << "attached sending Objects:" << _relayHash.count(); @@ -833,13 +1283,7 @@ void SignalProxy::dumpSyncMap(SyncableObject *object) { QHash syncMap_ = syncMap(object); QHash::const_iterator iter = syncMap_.constBegin(); while(iter != syncMap_.constEnd()) { - qDebug() << iter.key() << "-->" << iter.value() << meta->method(iter.value()).signature(); + qDebug() << qPrintable(QString("%1 --> %2 %3").arg(QString(iter.key()), 40).arg(iter.value()).arg(QString(meta->method(iter.value()).signature()))); iter++; } -// QHash syncMap_ = syncMap(object); -// QHash::const_iterator iter = syncMap_.constBegin(); -// while(iter != syncMap_.constEnd()) { -// qDebug() << iter.key() << meta->method(iter.key()).signature() << "-->" << iter.value() << meta->method(iter.value()).signature(); -// iter++; -// } }