X-Git-Url: https://git.quassel-irc.org/?p=quassel.git;a=blobdiff_plain;f=src%2Fcommon%2Fsignalproxy.cpp;h=0350ac6542e9160ec1398858bcaf470b2bb96b62;hp=794e878a1121e271a3ad71ea3ba7a7246d54ef5e;hb=f12ff6555fc60d165d5057902a91cdff355816b9;hpb=1ed8c48b0e7d309e1a92c905a19fcb7128b73d3d diff --git a/src/common/signalproxy.cpp b/src/common/signalproxy.cpp index 794e878a..0350ac65 100644 --- a/src/common/signalproxy.cpp +++ b/src/common/signalproxy.cpp @@ -23,6 +23,7 @@ #include #include #include +#include #include #include #include @@ -32,6 +33,7 @@ #include #include #include +#include #include "syncableobject.h" #include "util.h" @@ -96,7 +98,8 @@ int SignalRelay::qt_metacall(QMetaObject::Call _c, int _id, void **_a) { // 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() @@ -104,8 +107,8 @@ 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()); + params.prepend(syncObject->objectName()); + params.prepend(syncObject->syncMetaObject()->className()); proxy->dispatchSignal(SignalProxy::Sync, params); } } @@ -115,7 +118,11 @@ int SignalRelay::qt_metacall(QMetaObject::Call _c, int _id, void **_a) { } 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++ ) { @@ -133,8 +140,12 @@ 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")) @@ -170,7 +181,7 @@ void SignalRelay::attachSignal(int methodId, const QByteArray &func) { sigNames.insert(methodId, fn); } // ==================== -// END SIGNALRELAY +// /SIGNALRELAY // ==================== @@ -181,12 +192,14 @@ 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) @@ -194,6 +207,7 @@ SignalProxy::SignalProxy(ProxyMode mode, QIODevice* device, QObject* parent) { setProxyMode(mode); addPeer(device); + init(); } SignalProxy::~SignalProxy() { @@ -202,14 +216,14 @@ SignalProxy::~SignalProxy() { detachObject(sender); // close peer connections - foreach(QIODevice *device, _peerByteCount.keys()) { + foreach(QIODevice *device, _peers.keys()) { device->close(); delete device; } } void SignalProxy::setProxyMode(ProxyMode mode) { - foreach(QIODevice* peer, _peerByteCount.keys()) { + foreach(QIODevice* peer, _peers.keys()) { if(peer->isOpen()) { qWarning() << "SignalProxy: Cannot change proxy mode while connected"; return; @@ -222,30 +236,26 @@ 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() { - disconnect(&_heartBeatTimer, 0, this, 0); - _heartBeatTimer.stop(); } void SignalProxy::initClient() { attachSlot("__objectRenamed__", this, SLOT(objectRenamed(QByteArray, QString, QString))); - connect(&_heartBeatTimer, SIGNAL(timeout()), - this, SLOT(sendHeartBeat())); - _heartBeatTimer.start(60 * 1000); // msecs: one beep per minute } 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; } @@ -261,57 +271,60 @@ bool SignalProxy::addPeer(QIODevice* iodev) { connect(sock, SIGNAL(disconnected()), this, SLOT(removePeerBySender())); } - _peerByteCount[iodev] = 0; + _peers[iodev] = peerInfo(); + if(iodev->property("UseCompression").toBool()) + _peers[iodev].usesCompression = true; - if(_peerByteCount.count() == 1) + if(_peers.count() == 1) emit connected(); return true; } void SignalProxy::removePeer(QIODevice* iodev) { - if(_peerByteCount.isEmpty()) { + if(_peers.isEmpty()) { qWarning() << "SignalProxy::removePeer(): No peers in use!"; return; } if(proxyMode() == Server && !iodev) { // disconnect all - QList peers = _peerByteCount.keys(); + QList peers = _peers.keys(); foreach(QIODevice *peer, peers) removePeer(peer); } if(proxyMode() != Server && !iodev) - iodev = _peerByteCount.keys().first(); + iodev = _peers.keys().first(); Q_ASSERT(iodev); - if(!_peerByteCount.contains(iodev)) { + if(!_peers.contains(iodev)) { qWarning() << "SignalProxy: unknown QIODevice" << iodev; return; } - _peerByteCount.remove(iodev); + _peers.remove(iodev); disconnect(iodev, 0, this, 0); emit peerRemoved(iodev); - if(_peerByteCount.isEmpty()) + if(_peers.isEmpty()) emit disconnected(); } void SignalProxy::removePeerBySender() { // OK we're brutal here... but since it's a private slot we know what we've got connected to it... + // this Slot is not triggered by destroyed, so the object is still alive and can be used! QIODevice *ioDev = (QIODevice *)(sender()); removePeer(ioDev); - qDebug() << "Client disconnected."; } void SignalProxy::objectRenamed(const QString &newname, const QString &oldname) { - const QMetaObject *meta = sender()->metaObject(); + SyncableObject *syncObject = qobject_cast(sender()); + const QMetaObject *meta = syncObject->metaObject(); const QByteArray className(meta->className()); - objectRenamed(className, oldname, newname); + objectRenamed(className, newname, oldname); if(proxyMode() == Client) return; @@ -329,7 +342,7 @@ void SignalProxy::objectRenamed(const QByteArray &classname, const QString &newn } 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(); @@ -341,14 +354,31 @@ 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 = obj->metaObject(); + 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)); @@ -356,29 +386,31 @@ void SignalProxy::setMinArgCount(QObject *obj, int methodId) { } const int &SignalProxy::minArgCount(QObject *obj, int methodId) { - Q_ASSERT(_classInfo.contains(obj->metaObject())); - if(!_classInfo[obj->metaObject()]->minArgCount.contains(methodId)) + const QMetaObject *meta = metaObject(obj); + Q_ASSERT(_classInfo.contains(meta)); + if(!_classInfo[meta]->minArgCount.contains(methodId)) setMinArgCount(obj, methodId); - return _classInfo[obj->metaObject()]->minArgCount[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; @@ -387,6 +419,15 @@ void SignalProxy::setSyncMap(SyncableObject *obj) { 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++) { @@ -413,34 +454,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::setUpdatedRemotelyId(QObject *obj) { - const QMetaObject *meta = obj->metaObject(); +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(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) { @@ -482,7 +590,8 @@ 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; if(_relayHash.contains(obj)) @@ -493,32 +602,27 @@ 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) { connect(obj, SIGNAL(objectRenamed(QString, QString)), this, SLOT(objectRenamed(QString, QString))); - setInitialized(obj); + 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(InitRequest, params); } @@ -566,10 +670,11 @@ void SignalProxy::stopSync(SyncableObject* obj) { } void SignalProxy::dispatchSignal(QIODevice *receiver, const RequestType &requestType, const QVariantList ¶ms) { + Q_ASSERT(_peers.contains(receiver)); QVariantList packedFunc; packedFunc << (qint16)requestType; packedFunc << params; - writeDataToDevice(receiver, QVariant(packedFunc)); + writeDataToDevice(receiver, QVariant(packedFunc), _peers[receiver].usesCompression); } void SignalProxy::dispatchSignal(const RequestType &requestType, const QVariantList ¶ms) { @@ -577,8 +682,10 @@ void SignalProxy::dispatchSignal(const RequestType &requestType, const QVariantL QVariantList packedFunc; packedFunc << (qint16)requestType; packedFunc << params; - foreach(QIODevice* dev, _peerByteCount.keys()) - writeDataToDevice(dev, QVariant(packedFunc)); + foreach(QIODevice* dev, _peers.keys()) { + Q_ASSERT(_peers.contains(dev)); + writeDataToDevice(dev, QVariant(packedFunc), _peers[dev].usesCompression); + } } void SignalProxy::receivePeerSignal(QIODevice *sender, const QVariant &packedFunc) { @@ -593,27 +700,38 @@ void SignalProxy::receivePeerSignal(QIODevice *sender, const QVariant &packedFun switch(callType) { case RpcCall: - if(params.empty()) { + if(params.empty()) qWarning() << "SignalProxy::receivePeerSignal(): received empty RPC-Call"; - return; - } else { - return handleSignal(params.takeFirst().toByteArray(), params); - } + else + handleSignal(params.takeFirst().toByteArray(), params); + break; + 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: - return; + receiveHeartBeat(sender, params); + break; + + case HeartBeatReply: + receiveHeartBeatReply(sender, params); + break; + default: qWarning() << "SignalProxy::receivePeerSignal(): received undefined CallType" << callType << params; - return; } } -void SignalProxy::handleSync(QVariantList params) { +void SignalProxy::handleSync(QIODevice *sender, QVariantList params) { if(params.count() < 3) { qWarning() << "received invalid Sync call" << params; return; @@ -637,10 +755,26 @@ void SignalProxy::handleSync(QVariantList params) { } 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; + dispatchSignal(sender, Sync, returnParams); + } + + // send emit update signal invokeSlot(receiver, updatedRemotelyId(receiver)); } @@ -669,8 +803,8 @@ void SignalProxy::handleInitRequest(QIODevice *sender, const QVariantList ¶m SyncableObject *obj = _syncSlave[className][objectName]; QVariantList params_; - params_ << obj->metaObject()->className() - << obj->objectName() + params_ << className + << objectName << initData(obj); dispatchSignal(sender, InitData, params_); @@ -717,7 +851,7 @@ 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 = params.count() < args.count() ? params.count() @@ -728,9 +862,10 @@ bool SignalProxy::invokeSlot(QObject *receiver, int methodId, const QVariantList return false; } - void *_a[] = {0, // return type + 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()) { @@ -745,6 +880,8 @@ 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 @@ -754,36 +891,59 @@ 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)); QVariant var; - while(readDataFromDevice(ioDev, _peerByteCount[ioDev], var)) + while(readDataFromDevice(ioDev, _peers[ioDev].byteCount, var, _peers[ioDev].usesCompression)) receivePeerSignal(ioDev, 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); @@ -792,10 +952,31 @@ bool SignalProxy::readDataFromDevice(QIODevice *dev, quint32 &blockSize, QVarian in >> blockSize; } + if(blockSize > 1 << 22) { + qWarning() << qPrintable(tr("Client tried to send package larger than max package size!")); + QAbstractSocket *sock = qobject_cast(dev); + qWarning() << qPrintable(tr("Disconnecting")) << (sock ? qPrintable(sock->peerAddress().toString()) : qPrintable(tr("local client"))); + dev->close(); + return false; + } + if(dev->bytesAvailable() < blockSize) return false; - in >> item; + + if(compressed) { + QByteArray rawItem; + in >> rawItem; + rawItem = qUncompress(rawItem); + + QDataStream itemStream(&rawItem, QIODevice::ReadOnly); + itemStream.setVersion(QDataStream::Qt_4_2); + itemStream >> item; + } else { + in >> item; + } + blockSize = 0; + return true; } @@ -821,12 +1002,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(); @@ -839,17 +1027,68 @@ 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() { - dispatchSignal(SignalProxy::HeartBeat, QVariantList()); + dispatchSignal(SignalProxy::HeartBeat, QVariantList() << QTime::currentTime()); + QHash::iterator peerIter = _peers.begin(); + QHash::iterator peerIterEnd = _peers.end(); + while(peerIter != peerIterEnd) { + if(peerIter->sentHeartBeats > 0) { + updateLag(peerIter.key(), _heartBeatTimer.interval()); + } + if(peerIter->sentHeartBeats > 1) { + QAbstractSocket *socket = qobject_cast(peerIter.key()); + qWarning() << "SignalProxy: Disconnecting peer:" + << (socket ? qPrintable(socket->peerAddress().toString()) : "local client") + << "(didn't receive a heartbeat for over" << peerIter->sentHeartBeats * _heartBeatTimer.interval() / 1000 << "seconds)"; + peerIter.key()->close(); + } else { + peerIter->sentHeartBeats++; + } + peerIter++; + } +} + +void SignalProxy::receiveHeartBeat(QIODevice *dev, const QVariantList ¶ms) { + if(!_peers.contains(dev)) { + qWarning() << "SignalProxy: received heart beat from unknown Device:" << dev; + } + dispatchSignal(dev, SignalProxy::HeartBeatReply, params); } +void SignalProxy::receiveHeartBeatReply(QIODevice *dev, const QVariantList ¶ms) { + if(!_peers.contains(dev)) { + qWarning() << "SignalProxy: received heart beat reply from unknown Device:" << dev; + return; + } + + _peers[dev].sentHeartBeats = 0; + + if(params.isEmpty()) { + qWarning() << "SignalProxy: received heart beat reply with less params then sent from:" << dev; + return; + } + + QTime sendTime = params[0].value(); + updateLag(dev, sendTime.msecsTo(QTime::currentTime()) / 2); +} + +void SignalProxy::updateLag(QIODevice *dev, int lag) { + Q_ASSERT(_peers.contains(dev)); + _peers[dev].lag = lag; + if(proxyMode() == Client) { + emit lagUpdated(lag); + } +} + + void SignalProxy::dumpProxyStats() { QString mode; if(proxyMode() == Server)