++API, ++EventFlags
[quassel.git] / src / common / signalproxy.cpp
1 /***************************************************************************
2  *   Copyright (C) 2005-09 by the Quassel Project                          *
3  *   devel@quassel-irc.org                                                 *
4  *                                                                         *
5  *   This program is free software; you can redistribute it and/or modify  *
6  *   it under the terms of the GNU General Public License as published by  *
7  *   the Free Software Foundation; either version 2 of the License, or     *
8  *   (at your option) version 3.                                           *
9  *                                                                         *
10  *   This program is distributed in the hope that it will be useful,       *
11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
13  *   GNU General Public License for more details.                          *
14  *                                                                         *
15  *   You should have received a copy of the GNU General Public License     *
16  *   along with this program; if not, write to the                         *
17  *   Free Software Foundation, Inc.,                                       *
18  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
19  ***************************************************************************/
20
21 #include "signalproxy.h"
22
23 #include <QObject>
24 #include <QIODevice>
25 #include <QAbstractSocket>
26 #include <QHostAddress>
27 #include <QHash>
28 #include <QMultiHash>
29 #include <QList>
30 #include <QSet>
31 #include <QDebug>
32 #include <QMetaMethod>
33 #include <QMetaProperty>
34 #include <QRegExp>
35 #ifdef HAVE_SSL
36 #include <QSslSocket>
37 #endif
38 #include <QThread>
39 #include <QTime>
40 #include <QEvent>
41 #include <QCoreApplication>
42
43 #include "syncableobject.h"
44 #include "util.h"
45
46 // ==================================================
47 //  PeerSignalEvent
48 // ==================================================
49 class PeerSignalEvent : public QEvent {
50 public:
51   PeerSignalEvent(SignalProxy *sender, SignalProxy::RequestType requestType, const QVariantList &params) : QEvent(QEvent::Type(SignalProxy::PeerSignal)), sender(sender), requestType(requestType), params(params) {}
52   SignalProxy *sender;
53   SignalProxy::RequestType requestType;
54   QVariantList params;
55 };
56
57 class RemovePeerEvent : public QEvent {
58 public:
59   RemovePeerEvent(QObject *peer) : QEvent(QEvent::Type(SignalProxy::RemovePeer)), peer(peer) {}
60   QObject *peer;
61 };
62
63 // ==================================================
64 //  SignalRelay
65 // ==================================================
66 class SignalProxy::SignalRelay : public QObject {
67 /* Q_OBJECT is not necessary or even allowed, because we implement
68    qt_metacall ourselves (and don't use any other features of the meta
69    object system)
70 */
71 public:
72   SignalRelay(SignalProxy *parent) : QObject(parent), _proxy(parent) {}
73   inline SignalProxy *proxy() const { return _proxy; }
74
75   int qt_metacall(QMetaObject::Call _c, int _id, void **_a);
76
77   void attachSignal(QObject *sender, int signalId, const QByteArray &funcName);
78   void detachSignal(QObject *sender, int signalId = -1);
79
80 private:
81   struct Signal {
82     QObject *sender;
83     int signalId;
84     QByteArray signature;
85     Signal(QObject *sender, int sigId, const QByteArray &signature) : sender(sender), signalId(sigId), signature(signature) {}
86     Signal() : sender(0), signalId(-1) {}
87   };
88
89   SignalProxy *_proxy;
90   QHash<int, Signal> _slots;
91 };
92
93 void SignalProxy::SignalRelay::attachSignal(QObject *sender, int signalId, const QByteArray &funcName) {
94   // we ride without safetybelts here... all checking for valid method etc pp has to be done by the caller
95   // all connected methodIds are offset by the standard methodCount of QObject
96   int slotId;
97   for(int i = 0;; i++) {
98     if(!_slots.contains(i)) {
99       slotId = i;
100       break;
101     }
102   }
103
104   QByteArray fn;
105   if(!funcName.isEmpty()) {
106     fn = QMetaObject::normalizedSignature(funcName);
107   } else {
108     fn = SIGNAL(fakeMethodSignature());
109     fn = fn.replace("fakeMethodSignature()", sender->metaObject()->method(signalId).signature());
110   }
111
112   _slots[slotId] = Signal(sender, signalId, fn);
113
114   QMetaObject::connect(sender, signalId, this, QObject::staticMetaObject.methodCount() + slotId);
115 }
116
117 void SignalProxy::SignalRelay::detachSignal(QObject *sender, int signalId) {
118   QHash<int, Signal>::iterator slotIter = _slots.begin();
119   while(slotIter != _slots.end()) {
120     if(slotIter->sender == sender && (signalId == -1 || slotIter->signalId == signalId)) {
121       slotIter = _slots.erase(slotIter);
122       if(signalId != -1)
123         break;
124     } else {
125       slotIter++;
126     }
127   }
128 }
129
130 int SignalProxy::SignalRelay::qt_metacall(QMetaObject::Call _c, int _id, void **_a) {
131   _id = QObject::qt_metacall(_c, _id, _a);
132   if(_id < 0)
133     return _id;
134
135   if(_c == QMetaObject::InvokeMetaMethod) {
136     if(_slots.contains(_id)) {
137       QObject *caller = sender();
138
139       SignalProxy::ExtendedMetaObject *eMeta = proxy()->extendedMetaObject(caller->metaObject());
140       Q_ASSERT(eMeta);
141
142       const Signal &signal = _slots[_id];
143
144       QVariantList params;
145       params << signal.signature;
146
147       const QList<int> &argTypes = eMeta->argTypes(signal.signalId);
148       for(int i = 0; i < argTypes.size(); i++) {
149         if(argTypes[i] == 0) {
150           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());
151           qWarning() << "                            - make sure all your data types are known by the Qt MetaSystem";
152           return _id;
153         }
154         params << QVariant(argTypes[i], _a[i+1]);
155       }
156
157       proxy()->dispatchSignal(SignalProxy::RpcCall, params);
158     }
159     _id -= _slots.count();
160   }
161   return _id;
162 }
163
164 // ==================================================
165 //  Peers
166 // ==================================================
167 void SignalProxy::IODevicePeer::dispatchSignal(const RequestType &requestType, const QVariantList &params) {
168   QVariantList packedFunc;
169   packedFunc << (qint16)requestType
170              << params;
171   dispatchPackedFunc(QVariant(packedFunc));
172 }
173
174 bool SignalProxy::IODevicePeer::isSecure() const {
175 #ifdef HAVE_SSL
176   QSslSocket *sslSocket = qobject_cast<QSslSocket *>(_device);
177   if(sslSocket)
178     return sslSocket->isEncrypted() || sslSocket->localAddress() == QHostAddress::LocalHost || sslSocket->localAddress() == QHostAddress::LocalHostIPv6;
179 #endif
180
181   QAbstractSocket *socket = qobject_cast<QAbstractSocket *>(_device);
182   if(socket)
183     return socket->localAddress() == QHostAddress::LocalHost || socket->localAddress() == QHostAddress::LocalHostIPv6;
184
185   return false;
186 }
187
188 QString SignalProxy::IODevicePeer::address() const {
189   QAbstractSocket *socket = qobject_cast<QAbstractSocket *>(_device);
190   if(socket)
191     return socket->peerAddress().toString();
192   else
193     return QString();
194 }
195
196 void SignalProxy::SignalProxyPeer::dispatchSignal(const RequestType &requestType, const QVariantList &params) {
197   Qt::ConnectionType type = QThread::currentThread() == receiver->thread()
198     ? Qt::DirectConnection
199     : Qt::QueuedConnection;
200
201   if(type == Qt::DirectConnection) {
202     receiver->receivePeerSignal(sender, requestType, params);
203   } else {
204     QCoreApplication::postEvent(receiver, new PeerSignalEvent(sender, requestType, params));
205   }
206 }
207
208 // ==================================================
209 //  SignalProxy
210 // ==================================================
211 SignalProxy::SignalProxy(QObject* parent)
212   : QObject(parent)
213 {
214   setProxyMode(Client);
215   init();
216 }
217
218 SignalProxy::SignalProxy(ProxyMode mode, QObject* parent)
219   : QObject(parent)
220 {
221   setProxyMode(mode);
222   init();
223 }
224
225 SignalProxy::SignalProxy(ProxyMode mode, QIODevice* device, QObject* parent)
226   : QObject(parent)
227 {
228   setProxyMode(mode);
229   addPeer(device);
230   init();
231 }
232
233 SignalProxy::~SignalProxy() {
234   QHash<QByteArray, ObjectId>::iterator classIter = _syncSlave.begin();
235   while(classIter != _syncSlave.end()) {
236     ObjectId::iterator objIter = classIter->begin();
237     while(objIter != classIter->end()) {
238       SyncableObject *obj = objIter.value();
239       objIter = classIter->erase(objIter);
240       obj->stopSynchronize(this);
241     }
242     classIter++;
243   }
244   _syncSlave.clear();
245
246   removeAllPeers();
247 }
248
249 void SignalProxy::setProxyMode(ProxyMode mode) {
250   PeerHash::iterator peer = _peers.begin();
251   while(peer != _peers.end()) {
252     if((*peer)->type() != AbstractPeer::IODevicePeer) {
253       IODevicePeer *ioPeer = static_cast<IODevicePeer *>(*peer);
254       if(ioPeer->isOpen()) {
255         qWarning() << "SignalProxy: Cannot change proxy mode while connected";
256         return;
257       }
258     }
259     if((*peer)->type() != AbstractPeer::SignalProxyPeer) {
260       qWarning() << "SignalProxy: Cannot change proxy mode while connected to another internal SignalProxy";
261       return;
262     }
263     peer++;
264   }
265
266   _proxyMode = mode;
267   if(mode == Server)
268     initServer();
269   else
270     initClient();
271 }
272
273 void SignalProxy::init() {
274   _heartBeatInterval = 0;
275   _maxHeartBeatCount = 0;
276   _signalRelay = new SignalRelay(this);
277   connect(&_heartBeatTimer, SIGNAL(timeout()), this, SLOT(sendHeartBeat()));
278   setHeartBeatInterval(30);
279   setMaxHeartBeatCount(2);
280   _heartBeatTimer.start();
281   _secure = false;
282   updateSecureState();
283 }
284
285 void SignalProxy::initServer() {
286 }
287
288 void SignalProxy::initClient() {
289   attachSlot("__objectRenamed__", this, SLOT(objectRenamed(QByteArray, QString, QString)));
290 }
291
292 bool SignalProxy::addPeer(QIODevice* iodev) {
293   if(!iodev)
294     return false;
295
296   if(_peers.contains(iodev))
297     return true;
298
299   if(proxyMode() == Client && !_peers.isEmpty()) {
300     qWarning("SignalProxy: only one peer allowed in client mode!");
301     return false;
302   }
303
304   if(!iodev->isOpen()) {
305     qWarning("SignalProxy::addPeer(QIODevice *iodev): iodev needs to be open!");
306     return false;
307   }
308
309   connect(iodev, SIGNAL(disconnected()), this, SLOT(removePeerBySender()));
310   connect(iodev, SIGNAL(readyRead()), this, SLOT(dataAvailable()));
311
312 #ifdef HAVE_SSL
313   QSslSocket *sslSocket = qobject_cast<QSslSocket *>(iodev);
314   if(sslSocket) {
315     connect(iodev, SIGNAL(encrypted()), this, SLOT(updateSecureState()));
316   }
317 #endif
318
319   if(!iodev->parent())
320     iodev->setParent(this);
321
322   _peers[iodev] = new IODevicePeer(iodev, iodev->property("UseCompression").toBool());
323
324   if(_peers.count() == 1)
325     emit connected();
326
327   updateSecureState();
328   return true;
329 }
330
331 void SignalProxy::setHeartBeatInterval(int secs) {
332   if(secs != _heartBeatInterval) {
333     _heartBeatInterval = secs;
334     _heartBeatTimer.setInterval(secs * 1000);
335   }
336 }
337
338 void SignalProxy::setMaxHeartBeatCount(int max) {
339   _maxHeartBeatCount = max;
340 }
341
342 bool SignalProxy::addPeer(SignalProxy* proxy) {
343   if(!proxy)
344     return false;
345
346   if(proxyMode() == proxy->proxyMode()) {
347     qWarning() << "SignalProxy::addPeer(): adding a SignalProxy as peer requires one proxy to be server and one client!";
348     return false;
349   }
350
351   if(_peers.contains(proxy)) {
352     return true;
353   }
354
355   if(proxyMode() == Client && !_peers.isEmpty()) {
356     qWarning("SignalProxy: only one peer allowed in client mode!");
357     return false;
358   }
359
360   _peers[proxy] = new SignalProxyPeer(this, proxy);
361   proxy->addPeer(this);
362
363   if(_peers.count() == 1)
364     emit connected();
365
366   updateSecureState();
367   return true;
368 }
369
370 void SignalProxy::removeAllPeers() {
371   Q_ASSERT(proxyMode() == Server || _peers.count() <= 1);
372   // wee need to copy that list since we modify it in the loop
373   QList<QObject *> peers = _peers.keys();
374   foreach(QObject *peer, peers) {
375     removePeer(peer);
376   }
377 }
378
379 void SignalProxy::removePeer(QObject* dev) {
380   if(_peers.isEmpty()) {
381     qWarning() << "SignalProxy::removePeer(): No peers in use!";
382     return;
383   }
384
385   Q_ASSERT(dev);
386   if(!_peers.contains(dev)) {
387     qWarning() << "SignalProxy: unknown Peer" << dev;
388     return;
389   }
390
391   AbstractPeer *peer = _peers[dev];
392   _peers.remove(dev);
393
394   disconnect(dev, 0, this, 0);
395   if(peer->type() == AbstractPeer::IODevicePeer)
396     emit peerRemoved(static_cast<QIODevice *>(dev));
397
398   if(peer->type() == AbstractPeer::SignalProxyPeer) {
399     SignalProxy *proxy = static_cast<SignalProxy *>(dev);
400     if(proxy->_peers.contains(this))
401       proxy->removePeer(this);
402   }
403
404   if(dev->parent() == this)
405     dev->deleteLater();
406
407   delete peer;
408
409   updateSecureState();
410
411   if(_peers.isEmpty())
412     emit disconnected();
413 }
414
415 void SignalProxy::removePeerBySender() {
416   removePeer(sender());
417 }
418
419 void SignalProxy::renameObject(const SyncableObject *obj, const QString &newname, const QString &oldname) {
420   if(proxyMode() == Client)
421     return;
422
423   const QMetaObject *meta = obj->syncMetaObject();
424   const QByteArray className(meta->className());
425   objectRenamed(className, newname, oldname);
426
427   QVariantList params;
428   params << "__objectRenamed__" << className << newname << oldname;
429   dispatchSignal(RpcCall, params);
430 }
431
432 void SignalProxy::objectRenamed(const QByteArray &classname, const QString &newname, const QString &oldname) {
433   if(_syncSlave.contains(classname) && _syncSlave[classname].contains(oldname) && oldname != newname) {
434     SyncableObject *obj = _syncSlave[classname][newname] = _syncSlave[classname].take(oldname);
435     requestInit(obj);
436   }
437 }
438
439 const QMetaObject *SignalProxy::metaObject(const QObject *obj) {
440   if(const SyncableObject *syncObject = qobject_cast<const SyncableObject *>(obj))
441     return syncObject->syncMetaObject();
442   else
443     return obj->metaObject();
444 }
445
446 SignalProxy::ExtendedMetaObject *SignalProxy::extendedMetaObject(const QMetaObject *meta) const {
447   if(_extendedMetaObjects.contains(meta))
448     return _extendedMetaObjects[meta];
449   else
450     return 0;
451 }
452
453 SignalProxy::ExtendedMetaObject *SignalProxy::createExtendedMetaObject(const QMetaObject *meta, bool checkConflicts) {
454   if(!_extendedMetaObjects.contains(meta)) {
455     _extendedMetaObjects[meta] = new ExtendedMetaObject(meta, checkConflicts);
456   }
457   return _extendedMetaObjects[meta];
458 }
459
460 bool SignalProxy::attachSignal(QObject *sender, const char* signal, const QByteArray& sigName) {
461   const QMetaObject* meta = sender->metaObject();
462   QByteArray sig(meta->normalizedSignature(signal).mid(1));
463   int methodId = meta->indexOfMethod(sig.constData());
464   if(methodId == -1 || meta->method(methodId).methodType() != QMetaMethod::Signal) {
465     qWarning() << "SignalProxy::attachSignal(): No such signal" << signal;
466     return false;
467   }
468
469   createExtendedMetaObject(meta);
470   _signalRelay->attachSignal(sender, methodId, sigName);
471
472   disconnect(sender, SIGNAL(destroyed(QObject *)), this, SLOT(detachObject(QObject *)));
473   connect(sender, SIGNAL(destroyed(QObject *)), this, SLOT(detachObject(QObject *)));
474   return true;
475 }
476
477 bool SignalProxy::attachSlot(const QByteArray& sigName, QObject* recv, const char* slot) {
478   const QMetaObject* meta = recv->metaObject();
479   int methodId = meta->indexOfMethod(meta->normalizedSignature(slot).mid(1));
480   if(methodId == -1 || meta->method(methodId).methodType() == QMetaMethod::Method) {
481     qWarning() << "SignalProxy::attachSlot(): No such slot" << slot;
482     return false;
483   }
484
485   createExtendedMetaObject(meta);
486
487   QByteArray funcName = QMetaObject::normalizedSignature(sigName.constData());
488   _attachedSlots.insert(funcName, qMakePair(recv, methodId));
489
490   disconnect(recv, SIGNAL(destroyed(QObject *)), this, SLOT(detachObject(QObject *)));
491   connect(recv, SIGNAL(destroyed(QObject *)), this, SLOT(detachObject(QObject *)));
492   return true;
493 }
494
495 void SignalProxy::synchronize(SyncableObject *obj) {
496   createExtendedMetaObject(obj, true);
497
498   // attaching as slave to receive sync Calls
499   QByteArray className(obj->syncMetaObject()->className());
500   _syncSlave[className][obj->objectName()] = obj;
501
502   if(proxyMode() == Server) {
503     obj->setInitialized();
504     emit objectInitialized(obj);
505   } else {
506     if(obj->isInitialized())
507       emit objectInitialized(obj);
508     else
509       requestInit(obj);
510   }
511
512   obj->synchronize(this);
513 }
514
515 void SignalProxy::detachObject(QObject *obj) {
516   detachSignals(obj);
517   detachSlots(obj);
518 }
519
520 void SignalProxy::detachSignals(QObject *sender) {
521   _signalRelay->detachSignal(sender);
522 }
523
524 void SignalProxy::detachSlots(QObject* receiver) {
525   SlotHash::iterator slotIter = _attachedSlots.begin();
526   while(slotIter != _attachedSlots.end()) {
527     if(slotIter.value().first == receiver) {
528       slotIter = _attachedSlots.erase(slotIter);
529     } else
530       slotIter++;
531   }
532 }
533
534 void SignalProxy::stopSynchronize(SyncableObject *obj) {
535   // we can't use a className here, since it might be effed up, if we receive the call as a result of a decon
536   // gladly the objectName() is still valid. So we have only to iterate over the classes not each instance! *sigh*
537   QHash<QByteArray, ObjectId>::iterator classIter = _syncSlave.begin();
538   while(classIter != _syncSlave.end()) {
539     if(classIter->contains(obj->objectName()) && classIter.value()[obj->objectName()] == obj) {
540       classIter->remove(obj->objectName());
541       break;
542     }
543     classIter++;
544   }
545   obj->stopSynchronize(this);
546 }
547
548 void SignalProxy::dispatchSignal(const RequestType &requestType, const QVariantList &params) {
549   QVariant packedFunc(QVariantList() << (qint16)requestType << params);
550   PeerHash::iterator peer = _peers.begin();
551   while(peer != _peers.end()) {
552     switch((*peer)->type()) {
553     case AbstractPeer::IODevicePeer:
554       {
555         IODevicePeer *ioPeer = static_cast<IODevicePeer *>(*peer);
556         if(ioPeer->isOpen())
557           ioPeer->dispatchPackedFunc(packedFunc);
558         else
559           QCoreApplication::postEvent(this, new RemovePeerEvent(peer.key()));
560       }
561       break;
562     case AbstractPeer::SignalProxyPeer:
563       (*peer)->dispatchSignal(requestType, params);
564       break;
565     default:
566       Q_ASSERT(false); // there shouldn't be any peers with wrong / unknown type
567     }
568     peer++;
569   }
570 }
571
572 void SignalProxy::receivePackedFunc(AbstractPeer *sender, const QVariant &packedFunc) {
573   QVariantList params(packedFunc.toList());
574
575   if(params.isEmpty()) {
576     qWarning() << "SignalProxy::receivePeerSignal(): received incompatible Data:" << packedFunc;
577     return;
578   }
579
580   RequestType requestType = (RequestType)params.takeFirst().value<int>();
581   receivePeerSignal(sender, requestType, params);
582 }
583
584 void SignalProxy::receivePeerSignal(AbstractPeer *sender, const RequestType &requestType, const QVariantList &params) {
585   switch(requestType) {
586     // list all RequestTypes that shouldnot trigger a heartbeat counter reset here
587   case HeartBeatReply:
588     break;
589   default:
590     if(sender->type() == AbstractPeer::IODevicePeer) {
591       IODevicePeer *ioPeer = static_cast<IODevicePeer *>(sender);
592       ioPeer->sentHeartBeats = 0;
593     }
594   }
595
596   // qDebug() << "SignalProxy::receivePeerSignal)" << requestType << params;
597   switch(requestType) {
598   case RpcCall:
599     if(params.empty())
600       qWarning() << "SignalProxy::receivePeerSignal(): received empty RPC-Call";
601     else
602       handleSignal(params);
603       //handleSignal(params.takeFirst().toByteArray(), params);
604     break;
605
606   case Sync:
607     handleSync(sender, params);
608     break;
609
610   case InitRequest:
611     handleInitRequest(sender, params);
612     break;
613
614   case InitData:
615     handleInitData(sender, params);
616     break;
617
618   case HeartBeat:
619     receiveHeartBeat(sender, params);
620     break;
621
622   case HeartBeatReply:
623     receiveHeartBeatReply(sender, params);
624     break;
625
626   default:
627     qWarning() << "SignalProxy::receivePeerSignal(): received undefined CallType" << requestType << params;
628   }
629 }
630
631 void SignalProxy::receivePeerSignal(SignalProxy *sender, const RequestType &requestType, const QVariantList &params) {
632   if(!_peers.contains(sender)) {
633     // 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.
634     qWarning() << "SignalProxy::receivePeerSignal(): received Signal from unknown Proxy" << reinterpret_cast<void *>(sender);
635     return;
636   }
637   receivePeerSignal(_peers[sender], requestType, params);
638 }
639
640 void SignalProxy::handleSync(AbstractPeer *sender, QVariantList params) {
641   if(params.count() < 3) {
642     qWarning() << "received invalid Sync call" << params;
643     return;
644   }
645
646   QByteArray className = params.takeFirst().toByteArray();
647   QString objectName = params.takeFirst().toString();
648   QByteArray slot = params.takeFirst().toByteArray();
649
650   if(!_syncSlave.contains(className) || !_syncSlave[className].contains(objectName)) {
651     qWarning() << QString("no registered receiver for sync call: %1::%2 (objectName=\"%3\"). Params are:").arg(QString(className)).arg(QString(slot)).arg(objectName)
652                << params;
653     return;
654   }
655
656   SyncableObject *receiver = _syncSlave[className][objectName];
657   ExtendedMetaObject *eMeta = extendedMetaObject(receiver);
658   if(!eMeta->slotMap().contains(slot)) {
659     qWarning() << QString("no matching slot for sync call: %1::%2 (objectName=\"%3\"). Params are:").arg(QString(className)).arg(QString(slot)).arg(objectName)
660                << params;
661     return;
662   }
663
664   int slotId = eMeta->slotMap()[slot];
665   if(proxyMode() != eMeta->receiverMode(slotId)) {
666     qWarning("SignalProxy::handleSync(): invokeMethod for \"%s\" failed. Wrong ProxyMode!", eMeta->methodName(slotId).constData());
667     return;
668   }
669
670   QVariant returnValue((QVariant::Type)eMeta->returnType(slotId));
671   if(!invokeSlot(receiver, slotId, params, returnValue)) {
672     qWarning("SignalProxy::handleSync(): invokeMethod for \"%s\" failed ", eMeta->methodName(slotId).constData());
673     return;
674   }
675
676
677   if(returnValue.type() != QVariant::Invalid && eMeta->receiveMap().contains(slotId)) {
678     int receiverId = eMeta->receiveMap()[slotId];
679     QVariantList returnParams;
680     returnParams << className
681                  << objectName
682                  << eMeta->methodName(receiverId);
683     //QByteArray(receiver->metaObject()->method(receiverId).signature());
684     if(eMeta->argTypes(receiverId).count() > 1)
685       returnParams << params;
686     returnParams << returnValue;
687     sender->dispatchSignal(Sync, returnParams);
688   }
689
690   // send emit update signal
691   invokeSlot(receiver, eMeta->updatedRemotelyId());
692 }
693
694 void SignalProxy::handleInitRequest(AbstractPeer *sender, const QVariantList &params) {
695   if(params.count() != 2) {
696     qWarning() << "SignalProxy::handleInitRequest() received initRequest with invalid param Count:"
697                << params;
698     return;
699   }
700
701   QByteArray className(params[0].toByteArray());
702   QString objectName(params[1].toString());
703
704   if(!_syncSlave.contains(className)) {
705     qWarning() << "SignalProxy::handleInitRequest() received initRequest for unregistered Class:"
706                << className;
707     return;
708   }
709
710   if(!_syncSlave[className].contains(objectName)) {
711     qWarning() << "SignalProxy::handleInitRequest() received initRequest for unregistered Object:"
712                << className << objectName;
713     return;
714   }
715
716   SyncableObject *obj = _syncSlave[className][objectName];
717
718   QVariantList params_;
719   params_ << className
720           << objectName
721           << initData(obj);
722
723   sender->dispatchSignal(InitData, params_);
724 }
725
726 void SignalProxy::handleInitData(AbstractPeer *sender, const QVariantList &params) {
727   Q_UNUSED(sender)
728   if(params.count() != 3) {
729     qWarning() << "SignalProxy::handleInitData() received initData with invalid param Count:"
730                << params;
731     return;
732   }
733
734   QByteArray className(params[0].toByteArray());
735   QString objectName(params[1].toString());
736   QVariantMap propertyMap(params[2].toMap());
737
738   if(!_syncSlave.contains(className)) {
739     qWarning() << "SignalProxy::handleInitData() received initData for unregistered Class:"
740                << className;
741     return;
742   }
743
744   if(!_syncSlave[className].contains(objectName)) {
745     qWarning() << "SignalProxy::handleInitData() received initData for unregistered Object:"
746                << className << objectName;
747     return;
748   }
749
750   SyncableObject *obj = _syncSlave[className][objectName];
751   setInitData(obj, propertyMap);
752 }
753
754 //void SignalProxy::handleSignal(const QByteArray &funcName, const QVariantList &params) {
755 void SignalProxy::handleSignal(const QVariantList &data) {
756   QVariantList params = data;
757   QByteArray funcName = params.takeFirst().toByteArray();
758
759   QObject* receiver;
760   int methodId;
761   SlotHash::const_iterator slot = _attachedSlots.constFind(funcName);
762   while(slot != _attachedSlots.constEnd() && slot.key() == funcName) {
763     receiver = (*slot).first;
764     methodId = (*slot).second;
765     if(!invokeSlot(receiver, methodId, params)) {
766       ExtendedMetaObject *eMeta = extendedMetaObject(receiver);
767       qWarning("SignalProxy::handleSignal(): invokeMethod for \"%s\" failed ", eMeta->methodName(methodId).constData());
768     }
769     slot++;
770   }
771 }
772
773 bool SignalProxy::invokeSlot(QObject *receiver, int methodId, const QVariantList &params, QVariant &returnValue) {
774   ExtendedMetaObject *eMeta = extendedMetaObject(receiver);
775   const QList<int> args = eMeta->argTypes(methodId);
776   const int numArgs = params.count() < args.count()
777     ? params.count()
778     : args.count();
779
780   if(eMeta->minArgCount(methodId) > params.count()) {
781       qWarning() << "SignalProxy::invokeSlot(): not enough params to invoke" << eMeta->methodName(methodId);
782       return false;
783   }
784
785   void *_a[] = {0,              // return type...
786                 0, 0, 0, 0 , 0, // and 10 args - that's the max size qt can handle with signals and slots
787                 0, 0, 0, 0 , 0};
788
789   // check for argument compatibility and build params array
790   for(int i = 0; i < numArgs; i++) {
791     if(!params[i].isValid()) {
792       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());
793       qWarning() << "                            - make sure all your data types are known by the Qt MetaSystem";
794       return false;
795     }
796     if(args[i] != QMetaType::type(params[i].typeName())) {
797       qWarning() << "SignalProxy::invokeSlot(): incompatible param types to invoke" << eMeta->methodName(methodId);
798       return false;
799     }
800     _a[i+1] = const_cast<void *>(params[i].constData());
801   }
802
803   if(returnValue.type() != QVariant::Invalid)
804     _a[0] = const_cast<void *>(returnValue.constData());
805
806   Qt::ConnectionType type = QThread::currentThread() == receiver->thread()
807     ? Qt::DirectConnection
808     : Qt::QueuedConnection;
809
810   if(type == Qt::DirectConnection) {
811     return receiver->qt_metacall(QMetaObject::InvokeMetaMethod, methodId, _a) < 0;
812   } else {
813     qWarning() << "Queued Connections are not implemented yet";
814     // note to self: qmetaobject.cpp:990 ff
815     return false;
816   }
817
818 }
819
820 bool SignalProxy::invokeSlot(QObject *receiver, int methodId, const QVariantList &params) {
821   QVariant ret;
822   return invokeSlot(receiver, methodId, params, ret);
823 }
824
825 void SignalProxy::dataAvailable() {
826   // yet again. it's a private slot. no need for checks.
827   QIODevice* ioDev = qobject_cast<QIODevice* >(sender());
828   Q_ASSERT(_peers.contains(ioDev) && _peers[ioDev]->type() == AbstractPeer::IODevicePeer);
829   IODevicePeer *peer = static_cast<IODevicePeer *>(_peers[ioDev]);
830   QVariant var;
831   while(peer->readData(var))
832     receivePackedFunc(peer, var);
833 }
834
835 void SignalProxy::writeDataToDevice(QIODevice *dev, const QVariant &item, bool compressed) {
836   QAbstractSocket* sock  = qobject_cast<QAbstractSocket*>(dev);
837   if(!dev->isOpen() || (sock && sock->state()!=QAbstractSocket::ConnectedState)) {
838     qWarning("SignalProxy: Can't call write on a closed device");
839     return;
840   }
841
842   QByteArray block;
843   QDataStream out(&block, QIODevice::WriteOnly);
844   out.setVersion(QDataStream::Qt_4_2);
845   out << (quint32)0;
846
847   if(compressed) {
848     QByteArray rawItem;
849     QDataStream itemStream(&rawItem, QIODevice::WriteOnly);
850
851     itemStream.setVersion(QDataStream::Qt_4_2);
852     itemStream << item;
853
854     rawItem = qCompress(rawItem);
855
856     out << rawItem;
857   } else {
858     out << item;
859   }
860
861   out.device()->seek(0);
862   out << (quint32)(block.size() - sizeof(quint32));
863
864   dev->write(block);
865 }
866
867 bool SignalProxy::readDataFromDevice(QIODevice *dev, quint32 &blockSize, QVariant &item, bool compressed) {
868   if(!dev)
869     return false;
870
871   QDataStream in(dev);
872   in.setVersion(QDataStream::Qt_4_2);
873
874   if(blockSize == 0) {
875     if(dev->bytesAvailable() < (int)sizeof(quint32)) return false;
876     in >> blockSize;
877   }
878
879   if(blockSize > 1 << 22) {
880     disconnectDevice(dev, tr("Peer tried to send package larger than max package size!"));
881     return false;
882   }
883
884   if(blockSize == 0) {
885     disconnectDevice(dev, tr("Peer tried to send 0 byte package!"));
886     return false;
887   }
888
889   if(dev->bytesAvailable() < blockSize)
890     return false;
891
892   blockSize = 0;
893
894   if(compressed) {
895     QByteArray rawItem;
896     in >> rawItem;
897
898     int nbytes = rawItem.size();
899     if(nbytes <= 4) {
900       const char *data = rawItem.constData();
901       if(nbytes < 4 || (data[0]!=0 || data[1]!=0 || data[2]!=0 || data[3]!=0)) {
902         disconnectDevice(dev, tr("Peer sent corrupted compressed data!"));
903         return false;
904       }
905     }
906
907     rawItem = qUncompress(rawItem);
908
909     QDataStream itemStream(&rawItem, QIODevice::ReadOnly);
910     itemStream.setVersion(QDataStream::Qt_4_2);
911     itemStream >> item;
912   } else {
913     in >> item;
914   }
915
916   if(!item.isValid()) {
917     disconnectDevice(dev, tr("Peer sent corrupt data: unable to load QVariant!"));
918     return false;
919   }
920
921   return true;
922 }
923
924 void SignalProxy::requestInit(SyncableObject *obj) {
925   if(proxyMode() == Server || obj->isInitialized())
926     return;
927
928   QVariantList params;
929   params << obj->syncMetaObject()->className()
930          << obj->objectName();
931   dispatchSignal(InitRequest, params);
932 }
933
934 QVariantMap SignalProxy::initData(SyncableObject *obj) const {
935   return obj->toVariantMap();
936 }
937
938 void SignalProxy::setInitData(SyncableObject *obj, const QVariantMap &properties) {
939   if(obj->isInitialized())
940     return;
941   obj->fromVariantMap(properties);
942   obj->setInitialized();
943   emit objectInitialized(obj);
944   invokeSlot(obj, extendedMetaObject(obj)->updatedRemotelyId());
945 }
946
947 void SignalProxy::sendHeartBeat() {
948   QVariantList heartBeatParams;
949   heartBeatParams << QTime::currentTime();
950   QList<IODevicePeer *> toClose;
951
952   PeerHash::iterator peer = _peers.begin();
953   while(peer != _peers.end()) {
954     if((*peer)->type() == AbstractPeer::IODevicePeer) {
955       IODevicePeer *ioPeer = static_cast<IODevicePeer *>(*peer);
956       ioPeer->dispatchSignal(SignalProxy::HeartBeat, heartBeatParams);
957       if(ioPeer->sentHeartBeats > 0) {
958         updateLag(ioPeer, ioPeer->sentHeartBeats * _heartBeatTimer.interval());
959       }
960       if(maxHeartBeatCount() >= 0 && ioPeer->sentHeartBeats >= maxHeartBeatCount())
961         toClose.append(ioPeer);
962       else
963         ioPeer->sentHeartBeats++;
964     }
965     ++peer;
966   }
967
968   foreach(IODevicePeer *ioPeer, toClose) {
969     qWarning() << "SignalProxy: Disconnecting peer:" << ioPeer->address()
970                << "(didn't receive a heartbeat for over" << ioPeer->sentHeartBeats * _heartBeatTimer.interval() / 1000 << "seconds)";
971     ioPeer->close();
972   }
973 }
974
975 void SignalProxy::receiveHeartBeat(AbstractPeer *peer, const QVariantList &params) {
976   peer->dispatchSignal(SignalProxy::HeartBeatReply, params);
977 }
978
979 void SignalProxy::receiveHeartBeatReply(AbstractPeer *peer, const QVariantList &params) {
980   if(peer->type() != AbstractPeer::IODevicePeer) {
981     qWarning() << "SignalProxy::receiveHeartBeatReply: received heart beat from a non IODevicePeer!";
982     return;
983   }
984
985   IODevicePeer *ioPeer = static_cast<IODevicePeer *>(peer);
986   ioPeer->sentHeartBeats = 0;
987
988   if(params.isEmpty()) {
989     qWarning() << "SignalProxy: received heart beat reply with less params then sent from:" << ioPeer->address();
990     return;
991   }
992
993   QTime sendTime = params[0].value<QTime>();
994   updateLag(ioPeer, sendTime.msecsTo(QTime::currentTime()) / 2);
995 }
996
997 void SignalProxy::customEvent(QEvent *event) {
998   switch(event->type()) {
999   case PeerSignal:
1000     {
1001       PeerSignalEvent *e = static_cast<PeerSignalEvent *>(event);
1002       receivePeerSignal(e->sender, e->requestType, e->params);
1003     }
1004     event->accept();
1005     break;
1006   case RemovePeer:
1007     {
1008       RemovePeerEvent *e = static_cast<RemovePeerEvent *>(event);
1009       removePeer(e->peer);
1010     }
1011     event->accept();
1012   default:
1013     return;
1014   }
1015 }
1016
1017 void SignalProxy::sync_call__(const SyncableObject *obj, SignalProxy::ProxyMode modeType, const char *funcname, va_list ap) {
1018   // qDebug() << obj << modeType << "(" << _proxyMode << ")" << funcname;
1019   if(modeType != _proxyMode)
1020     return;
1021
1022   ExtendedMetaObject *eMeta = extendedMetaObject(obj);
1023
1024   QVariantList params;
1025   params << eMeta->metaObject()->className()
1026          << obj->objectName()
1027          << QByteArray(funcname);
1028
1029   const QList<int> &argTypes = eMeta->argTypes(eMeta->methodId(QByteArray(funcname)));
1030
1031   for(int i = 0; i < argTypes.size(); i++) {
1032     if(argTypes[i] == 0) {
1033       qWarning() << Q_FUNC_INFO << "received invalid data for argument number" << i << "of signal" << QString("%1::%2").arg(eMeta->metaObject()->className()).arg(funcname);
1034       qWarning() << "        - make sure all your data types are known by the Qt MetaSystem";
1035       return;
1036     }
1037     params << QVariant(argTypes[i], va_arg(ap, void *));
1038   }
1039
1040   dispatchSignal(Sync, params);
1041 }
1042
1043
1044
1045 void SignalProxy::disconnectDevice(QIODevice *dev, const QString &reason) {
1046   if(!reason.isEmpty())
1047     qWarning() << qPrintable(reason);
1048   QAbstractSocket *sock  = qobject_cast<QAbstractSocket *>(dev);
1049   if(sock)
1050     qWarning() << qPrintable(tr("Disconnecting")) << qPrintable(sock->peerAddress().toString());
1051   dev->close();
1052 }
1053
1054 void SignalProxy::updateLag(IODevicePeer *peer, int lag) {
1055   peer->lag = lag;
1056   if(proxyMode() == Client) {
1057     emit lagUpdated(lag);
1058   }
1059 }
1060
1061 void SignalProxy::dumpProxyStats() {
1062   QString mode;
1063   if(proxyMode() == Server)
1064     mode = "Server";
1065   else
1066     mode = "Client";
1067
1068   int slaveCount = 0;
1069   foreach(ObjectId oid, _syncSlave.values())
1070     slaveCount += oid.count();
1071
1072   qDebug() << this;
1073   qDebug() << "              Proxy Mode:" << mode;
1074   qDebug() << "          attached Slots:" << _attachedSlots.count();
1075   qDebug() << " number of synced Slaves:" << slaveCount;
1076   qDebug() << "number of Classes cached:" << _extendedMetaObjects.count();
1077 }
1078
1079 void SignalProxy::updateSecureState() {
1080   bool wasSecure = _secure;
1081
1082   _secure = !_peers.isEmpty();
1083   PeerHash::const_iterator peerIter;
1084   for(peerIter = _peers.constBegin(); peerIter != _peers.constEnd(); peerIter++) {
1085     _secure &= (*peerIter)->isSecure();
1086   }
1087
1088   if(wasSecure != _secure)
1089     emit secureStateChanged(_secure);
1090 }
1091
1092
1093 // ==================================================
1094 //  ExtendedMetaObject
1095 // ==================================================
1096 SignalProxy::ExtendedMetaObject::ExtendedMetaObject(const QMetaObject *meta, bool checkConflicts)
1097   : _meta(meta),
1098     _updatedRemotelyId(_meta->indexOfSignal("updatedRemotely()"))
1099 {
1100   for(int i = 0; i < _meta->methodCount(); i++) {
1101     if(_meta->method(i).methodType() != QMetaMethod::Slot)
1102       continue;
1103
1104     if(QByteArray(_meta->method(i).signature()).contains('*'))
1105       continue; // skip methods with ptr params
1106
1107     QByteArray method = methodName(_meta->method(i));
1108     if(method.startsWith("init"))
1109       continue; // skip initializers
1110
1111     if(_methodIds.contains(method)) {
1112       /* funny... moc creates for methods containing default parameters multiple metaMethod with separate methodIds.
1113          we don't care... we just need the full fledged version
1114        */
1115       const QMetaMethod &current = _meta->method(_methodIds[method]);
1116       const QMetaMethod &candidate = _meta->method(i);
1117       if(current.parameterTypes().count() > candidate.parameterTypes().count()) {
1118         int minCount = candidate.parameterTypes().count();
1119         QList<QByteArray> commonParams = current.parameterTypes().mid(0, minCount);
1120         if(commonParams == candidate.parameterTypes())
1121           continue; // we already got the full featured version
1122       } else {
1123         int minCount = current.parameterTypes().count();
1124         QList<QByteArray> commonParams = candidate.parameterTypes().mid(0, minCount);
1125         if(commonParams == current.parameterTypes()) {
1126           _methodIds[method] = i; // use the new one
1127           continue;
1128         }
1129       }
1130       if(checkConflicts) {
1131         qWarning() << "class" << meta->className() << "contains overloaded methods which is currently not supported!";
1132         qWarning() << " - " << _meta->method(i).signature() << "conflicts with" << _meta->method(_methodIds[method]).signature();
1133       }
1134       continue;
1135     }
1136     _methodIds[method] = i;
1137   }
1138 }
1139
1140 const SignalProxy::ExtendedMetaObject::MethodDescriptor &SignalProxy::ExtendedMetaObject::methodDescriptor(int methodId) {
1141   if(!_methods.contains(methodId)) {
1142     _methods[methodId] = MethodDescriptor(_meta->method(methodId));
1143   }
1144   return _methods[methodId];
1145 }
1146
1147 const QHash<int, int> &SignalProxy::ExtendedMetaObject::receiveMap() {
1148   if(_receiveMap.isEmpty()) {
1149     QHash<int, int> receiveMap;
1150
1151     QMetaMethod requestSlot;
1152     QByteArray returnTypeName;
1153     QByteArray signature;
1154     QByteArray methodName;
1155     QByteArray params;
1156     int paramsPos;
1157     int receiverId;
1158     const int methodCount = _meta->methodCount();
1159     for(int i = 0; i < methodCount; i++) {
1160       requestSlot = _meta->method(i);
1161       if(requestSlot.methodType() != QMetaMethod::Slot)
1162         continue;
1163
1164       returnTypeName = requestSlot.typeName();
1165       if(QMetaType::Void == (QMetaType::Type)returnType(i))
1166         continue;
1167
1168       signature = QByteArray(requestSlot.signature());
1169       if(!signature.startsWith("request"))
1170         continue;
1171
1172       paramsPos = signature.indexOf('(');
1173       if(paramsPos == -1)
1174         continue;
1175
1176       methodName = signature.left(paramsPos);
1177       params = signature.mid(paramsPos);
1178
1179       methodName = methodName.replace("request", "receive");
1180       params = params.left(params.count() - 1) + ", " + returnTypeName + ")";
1181
1182       signature = QMetaObject::normalizedSignature(methodName + params);
1183       receiverId = _meta->indexOfSlot(signature);
1184
1185       if(receiverId == -1) {
1186         signature = QMetaObject::normalizedSignature(methodName + "(" + returnTypeName + ")");
1187         receiverId = _meta->indexOfSlot(signature);
1188       }
1189
1190       if(receiverId != -1) {
1191         receiveMap[i] = receiverId;
1192       }
1193     }
1194     _receiveMap = receiveMap;
1195   }
1196   return _receiveMap;
1197 }
1198
1199 QByteArray SignalProxy::ExtendedMetaObject::methodName(const QMetaMethod &method) {
1200   QByteArray sig(method.signature());
1201   return sig.left(sig.indexOf("("));
1202 }
1203
1204 QString SignalProxy::ExtendedMetaObject::methodBaseName(const QMetaMethod &method) {
1205   QString methodname = QString(method.signature()).section("(", 0, 0);
1206
1207   // determine where we have to chop:
1208   int upperCharPos;
1209   if(method.methodType() == QMetaMethod::Slot) {
1210     // we take evertyhing from the first uppercase char if it's slot
1211     upperCharPos = methodname.indexOf(QRegExp("[A-Z]"));
1212     if(upperCharPos == -1)
1213       return QString();
1214     methodname = methodname.mid(upperCharPos);
1215   } else {
1216     // and if it's a signal we discard everything from the last uppercase char
1217     upperCharPos = methodname.lastIndexOf(QRegExp("[A-Z]"));
1218     if(upperCharPos == -1)
1219       return QString();
1220     methodname = methodname.left(upperCharPos);
1221   }
1222
1223   methodname[0] = methodname[0].toUpper();
1224
1225   return methodname;
1226 }
1227
1228 SignalProxy::ExtendedMetaObject::MethodDescriptor::MethodDescriptor(const QMetaMethod &method)
1229   : _methodName(SignalProxy::ExtendedMetaObject::methodName(method)),
1230     _returnType(QMetaType::type(method.typeName()))
1231 {
1232   // determine argTypes
1233   QList<QByteArray> paramTypes = method.parameterTypes();
1234   QList<int> argTypes;
1235   for(int i = 0; i < paramTypes.count(); i++) {
1236     argTypes.append(QMetaType::type(paramTypes[i]));
1237   }
1238   _argTypes = argTypes;
1239
1240   // determine minArgCount
1241   QString signature(method.signature());
1242   _minArgCount = method.parameterTypes().count() - signature.count("=");
1243
1244   _receiverMode = (_methodName.startsWith("request"))
1245     ? SignalProxy::Server
1246     : SignalProxy::Client;
1247 }
1248