core/coreproxy.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005/06 by The Quassel Team                             *
   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) any later version.                                   *
   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 <QDebug>
  22
  23 #include "coreproxy.h"
  24 #include "global.h"
  25 #include "util.h"
  26
  27 CoreProxy::CoreProxy() {
  28   if(coreProxy) qFatal("Trying to instantiate more than one CoreProxy object!");
  29
  30   connect(global, SIGNAL(dataPutLocally(QString)), this, SLOT(updateGlobalData(QString)));
  31   connect(&server, SIGNAL(newConnection()), this, SLOT(incomingConnection()));
  32   if(!server.listen(QHostAddress::Any, 4242)) {
  33     qFatal(QString(QString("Could not open GUI client port %1: %2").arg(4242).arg(server.errorString())).toAscii());
  34   }
  35   qDebug() << "Listening for GUI clients on port" << server.serverPort() << ".";
  36 }
  37
  38 void CoreProxy::incomingConnection() {
  39   QTcpSocket *socket = server.nextPendingConnection();
  40   connect(socket, SIGNAL(disconnected()), this, SLOT(clientDisconnected()));
  41   connect(socket, SIGNAL(readyRead()), this, SLOT(clientHasData()));
  42   clients.append(socket);
  43   blockSizes.insert(socket, (quint32)0);
  44   qDebug() << "Client connected from " << socket->peerAddress().toString();
  45 }
  46
  47 void CoreProxy::clientHasData() {
  48   QTcpSocket *socket = dynamic_cast<QTcpSocket*>(sender());
  49   Q_ASSERT(socket && blockSizes.contains(socket));
  50   quint32 bsize = blockSizes.value(socket);
  51   QVariant item;
  52   while(readDataFromDevice(socket, bsize, item)) {
  53     QList<QVariant> sigdata = item.toList();
  54     Q_ASSERT(sigdata.size() == 4);
  55     switch((GUISignal)sigdata[0].toInt()) {
  56       case GS_CLIENT_INIT:  processClientInit(socket, sigdata[1]); break;
  57       case GS_UPDATE_GLOBAL_DATA: processClientUpdate(socket, sigdata[1].toString(), sigdata[2]); break;
  58       //case GS_CLIENT_READY: processClientReady(sigdata[1], sigdata[2], sigdata[3]); break;
  59       default: recv((GUISignal)sigdata[0].toInt(), sigdata[1], sigdata[2], sigdata[3]); break;
  60     }
  61     blockSizes[socket] = bsize = 0;
  62   }
  63   blockSizes[socket] = bsize;
  64 }
  65
  66 void CoreProxy::clientDisconnected() {
  67   QTcpSocket *socket = dynamic_cast<QTcpSocket*>(sender());
  68   blockSizes.remove(socket);
  69   clients.removeAll(socket);
  70   qDebug() << "Client disconnected.";
  71 }
  72
  73 void CoreProxy::processClientInit(QTcpSocket *socket, const QVariant &v) {
  74   VarMap msg = v.toMap();
  75   if(msg["GUIProtocol"].toUInt() != GUI_PROTOCOL) {
  76     qDebug() << "Client version mismatch. Disconnecting.";
  77     socket->close();
  78     return;
  79   }
  80   VarMap reply;
  81   VarMap coreData;
  82   QStringList dataKeys = global->getKeys();
  83   QString key;
  84   foreach(key, dataKeys) {
  85     coreData[key] = global->getData(key);
  86   }
  87   reply["CoreData"] = coreData;
  88   //QVariant payload = QByteArray(1000000, 'a');
  89   //reply["payload"] = payload;
  90   QList<QVariant> sigdata;
  91   sigdata.append(CS_CORE_STATE); sigdata.append(QVariant(reply)); sigdata.append(QVariant()); sigdata.append(QVariant());
  92   writeDataToDevice(socket, QVariant(sigdata));
  93 }
  94
  95 void CoreProxy::processClientUpdate(QTcpSocket *socket, QString key, QVariant data) {
  96   global->updateData(key, data);
  97   QList<QVariant> sigdata;
  98   sigdata.append(CS_UPDATE_GLOBAL_DATA); sigdata.append(key); sigdata.append(data); sigdata.append(QVariant());
  99   QTcpSocket *s;
 100   foreach(s, clients) {
 101     if(s != socket) writeDataToDevice(s, QVariant(sigdata));
 102   }
 103 }
 104
 105 void CoreProxy::updateGlobalData(QString key) {
 106   QVariant data = global->getData(key);
 107   emit csUpdateGlobalData(key, data);
 108 }
 109
 110 void CoreProxy::send(CoreSignal sig, QVariant arg1, QVariant arg2, QVariant arg3) {
 111   sendToGUI(sig, arg1, arg2, arg3);
 112   QList<QVariant> sigdata;
 113   sigdata.append(sig); sigdata.append(arg1); sigdata.append(arg2); sigdata.append(arg3);
 114   //qDebug() << "Sending signal: " << sigdata;
 115   QTcpSocket *socket;
 116   foreach(socket, clients) {
 117     writeDataToDevice(socket, QVariant(sigdata));
 118   }
 119 }
 120
 121 void CoreProxy::recv(GUISignal sig, QVariant arg1, QVariant arg2, QVariant arg3) {
 122   //qDebug() << "[CORE] Received signal" << sig << ":" << arg1<<arg2<<arg3;
 123   switch(sig) {
 124     case GS_UPDATE_GLOBAL_DATA: emit gsPutGlobalData(arg1.toString(), arg2); break;
 125     case GS_USER_INPUT: emit gsUserInput(arg1.toString()); break;
 126     case GS_REQUEST_CONNECT: emit gsRequestConnect(arg1.toString(), arg2.toUInt()); break;
 127     default: qWarning() << "Unknown signal in CoreProxy::recv: " << sig;
 128   }
 129 }
 130
 131 CoreProxy *coreProxy;