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