src/core/core.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-07 by The Quassel IRC Development 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 "core.h"
  22 #include "coresession.h"
  23 #include "sqlitestorage.h"
  24 #include "util.h"
  25
  26 Core *Core::instanceptr = 0;
  27
  28 Core * Core::instance() {
  29   if(instanceptr) return instanceptr;
  30   instanceptr = new Core();
  31   instanceptr->init();
  32   return instanceptr;
  33 }
  34
  35 void Core::destroy() {
  36   delete instanceptr;
  37   instanceptr = 0;
  38 }
  39
  40 Core::Core() {
  41
  42 }
  43
  44 void Core::init() {
  45   if(!SqliteStorage::isAvailable()) {
  46     qFatal("Sqlite is currently required! Please make sure your Qt library has sqlite support enabled.");
  47   }
  48   //SqliteStorage::init();
  49   storage = new SqliteStorage();
  50   connect(&server, SIGNAL(newConnection()), this, SLOT(incomingConnection()));
  51   startListening(); // FIXME make configurable
  52   guiUser = 0;
  53 }
  54
  55 Core::~Core() {
  56   foreach(QTcpSocket *sock, validClients.keys()) {
  57     delete sock;
  58   }
  59   qDeleteAll(sessions);
  60   delete storage;
  61 }
  62
  63 CoreSession *Core::session(UserId uid) {
  64   Core *core = instance();
  65   if(core->sessions.contains(uid)) return core->sessions[uid];
  66   else return 0;
  67 }
  68
  69 CoreSession *Core::localSession() {
  70   Core *core = instance();
  71   if(core->guiUser && core->sessions.contains(core->guiUser)) return core->sessions[core->guiUser];
  72   else return 0;
  73 }
  74
  75 CoreSession *Core::createSession(UserId uid) {
  76   Core *core = instance();
  77   Q_ASSERT(!core->sessions.contains(uid));
  78   CoreSession *sess = new CoreSession(uid, core->storage);
  79   core->sessions[uid] = sess;
  80   connect(sess, SIGNAL(proxySignal(CoreSignal, QVariant, QVariant, QVariant)), core, SLOT(recvProxySignal(CoreSignal, QVariant, QVariant, QVariant)));
  81   return sess;
  82 }
  83
  84
  85 bool Core::startListening(uint port) {
  86   if(!server.listen(QHostAddress::Any, port)) {
  87     qWarning(QString(QString("Could not open GUI client port %1: %2").arg(port).arg(server.errorString())).toAscii());
  88     return false;
  89   }
  90   qDebug() << "Listening for GUI clients on port" << server.serverPort();
  91   return true;
  92 }
  93
  94 void Core::stopListening() {
  95   server.close();
  96   qDebug() << "No longer listening for GUI clients.";
  97 }
  98
  99 void Core::incomingConnection() {
 100   // TODO implement SSL
 101   QTcpSocket *socket = server.nextPendingConnection();
 102   connect(socket, SIGNAL(disconnected()), this, SLOT(clientDisconnected()));
 103   connect(socket, SIGNAL(readyRead()), this, SLOT(clientHasData()));
 104   blockSizes.insert(socket, (quint32)0);
 105   qDebug() << "Client connected from " << socket->peerAddress().toString();
 106 }
 107
 108 void Core::clientHasData() {
 109   QTcpSocket *socket = dynamic_cast<QTcpSocket*>(sender());
 110   Q_ASSERT(socket && blockSizes.contains(socket));
 111   quint32 bsize = blockSizes.value(socket);
 112   QVariant item;
 113   while(readDataFromDevice(socket, bsize, item)) {
 114     if(validClients.contains(socket)) {
 115       QList<QVariant> sigdata = item.toList();
 116       sessions[validClients[socket]]->processSignal((ClientSignal)sigdata[0].toInt(), sigdata[1], sigdata[2], sigdata[3]);
 117     } else {
 118       // we need to auth the client
 119       try {
 120         processClientInit(socket, item);
 121       } catch(Storage::AuthError) {
 122         qWarning() << "Authentification error!";  // FIXME
 123         socket->close();
 124         return;
 125       } catch(Exception e) {
 126         qWarning() << "Client init error:" << e.msg();
 127         socket->close();
 128         return;
 129       }
 130     }
 131     blockSizes[socket] = bsize = 0;
 132   }
 133   blockSizes[socket] = bsize;
 134 }
 135
 136 void Core::clientDisconnected() {
 137   QTcpSocket *socket = dynamic_cast<QTcpSocket*>(sender());
 138   blockSizes.remove(socket);
 139   validClients.remove(socket);
 140   qDebug() << "Client disconnected.";
 141   // TODO remove unneeded sessions - if necessary/possible...
 142 }
 143
 144 QVariant Core::connectLocalClient(QString user, QString passwd) {
 145   UserId uid = instance()->storage->validateUser(user, passwd);
 146   QVariant reply = instance()->initSession(uid);
 147   instance()->guiUser = uid;
 148   qDebug() << "Local client connected.";
 149   return reply;
 150 }
 151
 152 void Core::disconnectLocalClient() {
 153   qDebug() << "Local client disconnected.";
 154   instance()->guiUser = 0;
 155 }
 156
 157 void Core::processClientInit(QTcpSocket *socket, const QVariant &v) {
 158   VarMap msg = v.toMap();
 159   if(msg["GuiProtocol"].toUInt() != GUI_PROTOCOL) {
 160     //qWarning() << "Client version mismatch.";
 161     throw Exception("GUI client version mismatch");
 162   }
 163     // Auth
 164   UserId uid = storage->validateUser(msg["User"].toString(), msg["Password"].toString());  // throws exception if this failed
 165   QVariant reply = initSession(uid);
 166   validClients[socket] = uid;
 167   QList<QVariant> sigdata;
 168   sigdata.append(CS_CORE_STATE); sigdata.append(reply); sigdata.append(QVariant()); sigdata.append(QVariant());
 169   writeDataToDevice(socket, QVariant(sigdata));
 170 }
 171
 172 QVariant Core::initSession(UserId uid) {
 173   // Find or create session for validated user
 174   CoreSession *sess;
 175   if(sessions.contains(uid)) sess = sessions[uid];
 176   else {
 177     sess = createSession(uid);
 178     //validClients[socket] = uid;
 179   }
 180   VarMap reply;
 181   reply["SessionState"] = sess->sessionState();
 182   return reply;
 183 }
 184
 185 void Core::recvProxySignal(CoreSignal sig, QVariant arg1, QVariant arg2, QVariant arg3) {
 186   CoreSession *sess = qobject_cast<CoreSession*>(sender());
 187   Q_ASSERT(sess);
 188   UserId uid = sess->userId();
 189   QList<QVariant> sigdata;
 190   sigdata.append(sig); sigdata.append(arg1); sigdata.append(arg2); sigdata.append(arg3);
 191   //qDebug() << "Sending signal: " << sigdata;
 192   foreach(QTcpSocket *socket, validClients.keys()) {
 193     if(validClients[socket] == uid) writeDataToDevice(socket, QVariant(sigdata));
 194   }
 195 }