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