+#include <QObject>
+#include <QIODevice>
+#include <QAbstractSocket>
+#include <QHostAddress>
+#include <QHash>
+#include <QMultiHash>
+#include <QList>
+#include <QSet>
+#include <QDebug>
+#include <QMetaMethod>
+#include <QMetaProperty>
+#include <QRegExp>
+#ifdef HAVE_SSL
+#include <QSslSocket>
+#endif
+#include <QThread>
+#include <QTime>
+#include <QEvent>
+#include <QCoreApplication>
+
+#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<int, Signal> _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<int, Signal>::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 {
+ 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<int> &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<QSslSocket *>(_device);
+ if (sslSocket)
+ return sslSocket->isEncrypted() || sslSocket->localAddress() == QHostAddress::LocalHost || sslSocket->localAddress() == QHostAddress::LocalHostIPv6;
+#endif
+
+ QAbstractSocket *socket = qobject_cast<QAbstractSocket *>(_device);
+ if (socket)
+ return socket->localAddress() == QHostAddress::LocalHost || socket->localAddress() == QHostAddress::LocalHostIPv6;
+
+ return false;
+}
+
+
+QString SignalProxy::IODevicePeer::address() const
+{
+ QAbstractSocket *socket = qobject_cast<QAbstractSocket *>(_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()
+{
+ QHash<QByteArray, ObjectId>::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<IODevicePeer *>(*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()
+{
+ _heartBeatInterval = 0;
+ _maxHeartBeatCount = 0;
+ _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<QSslSocket *>(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<QObject *> 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<QIODevice *>(dev));
+
+ if (peer->type() == AbstractPeer::SignalProxyPeer) {
+ SignalProxy *proxy = static_cast<SignalProxy *>(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<const SyncableObject *>(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<QByteArray, ObjectId>::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<IODevicePeer *>(*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<int>();
+ 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<IODevicePeer *>(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;