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 "server.h"
  23 #include "global.h"
  24 #include "util.h"
  25 #include "coreproxy.h"
  26 #include "sqlitestorage.h"
  27
  28 #include <QtSql>
  29 #include <QSettings>
  30
  31 Core *Core::instanceptr = 0;
  32
  33 Core * Core::instance() {
  34   if(instanceptr) return instanceptr;
  35   instanceptr = new Core();
  36   instanceptr->init();
  37   return instanceptr;
  38 }
  39
  40 void Core::destroy() {
  41   delete instanceptr;
  42   instanceptr = 0;
  43 }
  44
  45 Core::Core() {
  46   qDebug() << "core";
  47 }
  48
  49 void Core::init() {
  50   if(!SqliteStorage::isAvailable()) {
  51     qFatal("Sqlite is currently required! Please make sure your Qt library has sqlite support enabled.");
  52   }
  53   //SqliteStorage::init();
  54   storage = new SqliteStorage();
  55   connect(Global::instance(), SIGNAL(dataPutLocally(UserId, QString)), this, SLOT(updateGlobalData(UserId, QString)));
  56   connect(&server, SIGNAL(newConnection()), this, SLOT(incomingConnection()));
  57   //startListening(); // FIXME
  58   if(Global::runMode == Global::Monolithic) {  // TODO Make GUI user configurable
  59     try {
  60       guiUser = storage->validateUser("Default", "password");
  61     } catch(Storage::AuthError) {
  62       guiUser = storage->addUser("Default", "password");
  63     }
  64     Q_ASSERT(guiUser);
  65     Global::setGuiUser(guiUser);
  66     createSession(guiUser);
  67   } else guiUser = 0;
  68
  69   // Read global settings from config file
  70   QSettings s;
  71   s.beginGroup("Global");
  72   foreach(QString unum, s.childGroups()) {
  73     UserId uid = unum.toUInt();
  74     s.beginGroup(unum);
  75     foreach(QString key, s.childKeys()) {
  76       Global::updateData(uid, key, s.value(key));
  77     }
  78     s.endGroup();
  79   }
  80   s.endGroup();
  81 }
  82
  83 Core::~Core() {
  84   foreach(QTcpSocket *sock, validClients.keys()) {
  85     delete sock;
  86   }
  87   qDeleteAll(sessions);
  88   delete storage;
  89 }
  90
  91 CoreSession *Core::session(UserId uid) {
  92   Core *core = instance();
  93   if(core->sessions.contains(uid)) return core->sessions[uid];
  94   else return 0;
  95 }
  96
  97 CoreSession *Core::localSession() {
  98   Core *core = instance();
  99   if(core->guiUser && core->sessions.contains(core->guiUser)) return core->sessions[core->guiUser];
 100   else return 0;
 101 }
 102
 103 CoreSession *Core::createSession(UserId uid) {
 104   Core *core = instance();
 105   Q_ASSERT(!core->sessions.contains(uid));
 106   CoreSession *sess = new CoreSession(uid, core->storage);
 107   core->sessions[uid] = sess;
 108   connect(sess, SIGNAL(proxySignal(CoreSignal, QVariant, QVariant, QVariant)), core, SLOT(recvProxySignal(CoreSignal, QVariant, QVariant, QVariant)));
 109   return sess;
 110 }
 111
 112
 113 bool Core::startListening(uint port) {
 114   if(!server.listen(QHostAddress::Any, port)) {
 115     qWarning(QString(QString("Could not open GUI client port %1: %2").arg(port).arg(server.errorString())).toAscii());
 116     return false;
 117   }
 118   qDebug() << "Listening for GUI clients on port" << server.serverPort();
 119   return true;
 120 }
 121
 122 void Core::stopListening() {
 123   server.close();
 124   qDebug() << "No longer listening for GUI clients.";
 125 }
 126
 127 void Core::incomingConnection() {
 128   // TODO implement SSL
 129   QTcpSocket *socket = server.nextPendingConnection();
 130   connect(socket, SIGNAL(disconnected()), this, SLOT(clientDisconnected()));
 131   connect(socket, SIGNAL(readyRead()), this, SLOT(clientHasData()));
 132   blockSizes.insert(socket, (quint32)0);
 133   qDebug() << "Client connected from " << socket->peerAddress().toString();
 134 }
 135
 136 void Core::clientHasData() {
 137   QTcpSocket *socket = dynamic_cast<QTcpSocket*>(sender());
 138   Q_ASSERT(socket && blockSizes.contains(socket));
 139   quint32 bsize = blockSizes.value(socket);
 140   QVariant item;
 141   while(readDataFromDevice(socket, bsize, item)) {
 142     if(validClients.contains(socket)) {
 143       QList<QVariant> sigdata = item.toList();
 144       if((ClientSignal)sigdata[0].toInt() == GS_UPDATE_GLOBAL_DATA) {
 145         processClientUpdate(socket, sigdata[1].toString(), sigdata[2]);
 146       } else {
 147         sessions[validClients[socket]]->processSignal((ClientSignal)sigdata[0].toInt(), sigdata[1], sigdata[2], sigdata[3]);
 148       }
 149     } else {
 150       // we need to auth the client
 151       try {
 152         processClientInit(socket, item);
 153       } catch(Storage::AuthError) {
 154         qWarning() << "Authentification error!";  // FIXME
 155         socket->close();
 156         return;
 157       } catch(Exception e) {
 158         qWarning() << "Client init error:" << e.msg();
 159         socket->close();
 160         return;
 161       }
 162     }
 163     blockSizes[socket] = bsize = 0;
 164   }
 165   blockSizes[socket] = bsize;
 166 }
 167
 168 void Core::clientDisconnected() {
 169   QTcpSocket *socket = dynamic_cast<QTcpSocket*>(sender());
 170   blockSizes.remove(socket);
 171   validClients.remove(socket);
 172   qDebug() << "Client disconnected.";
 173   // TODO remove unneeded sessions - if necessary/possible...
 174 }
 175
 176 void Core::processClientInit(QTcpSocket *socket, const QVariant &v) {
 177   VarMap msg = v.toMap();
 178   if(msg["GUIProtocol"].toUInt() != GUI_PROTOCOL) {
 179     //qWarning() << "Client version mismatch.";
 180     throw Exception("GUI client version mismatch");
 181   }
 182   // Auth
 183   UserId uid = storage->validateUser(msg["User"].toString(), msg["Password"].toString());  // throws exception if this failed
 184
 185   // Find or create session for validated user
 186   CoreSession *sess;
 187   if(sessions.contains(uid)) sess = sessions[uid];
 188   else {
 189     sess = createSession(uid);
 190     validClients[socket] = uid;
 191   }
 192   VarMap reply;
 193   VarMap coreData;
 194   // FIXME
 195   QStringList dataKeys = Global::keys(uid);
 196   QString key;
 197   foreach(key, dataKeys) {
 198     coreData[key] = Global::data(key);
 199   }
 200   reply["CoreData"] = coreData;
 201   reply["SessionState"] = sess->sessionState();
 202   QList<QVariant> sigdata;
 203   sigdata.append(CS_CORE_STATE); sigdata.append(QVariant(reply)); sigdata.append(QVariant()); sigdata.append(QVariant());
 204   writeDataToDevice(socket, QVariant(sigdata));
 205   sess->sendServerStates();
 206 }
 207
 208 void Core::processClientUpdate(QTcpSocket *socket, QString key, const QVariant &data) {
 209   UserId uid = validClients[socket];
 210   Global::updateData(uid, key, data);
 211   QList<QVariant> sigdata;
 212   sigdata.append(CS_UPDATE_GLOBAL_DATA); sigdata.append(key); sigdata.append(data); sigdata.append(QVariant());
 213   foreach(QTcpSocket *s, validClients.keys()) {
 214     if(validClients[s] == uid && s != socket) writeDataToDevice(s, QVariant(sigdata));
 215   }
 216 }
 217
 218 void Core::updateGlobalData(UserId uid, QString key) {
 219   QVariant data = Global::data(uid, key);
 220   QList<QVariant> sigdata;
 221   sigdata.append(CS_UPDATE_GLOBAL_DATA); sigdata.append(key); sigdata.append(data); sigdata.append(QVariant());
 222   foreach(QTcpSocket *socket, validClients.keys()) {
 223     if(validClients[socket] == uid) writeDataToDevice(socket, QVariant(sigdata));
 224   }
 225 }
 226
 227 void Core::recvProxySignal(CoreSignal sig, QVariant arg1, QVariant arg2, QVariant arg3) {
 228   CoreSession *sess = qobject_cast<CoreSession*>(sender());
 229   Q_ASSERT(sess);
 230   UserId uid = sess->userId();
 231   QList<QVariant> sigdata;
 232   sigdata.append(sig); sigdata.append(arg1); sigdata.append(arg2); sigdata.append(arg3);
 233   //qDebug() << "Sending signal: " << sigdata;
 234   foreach(QTcpSocket *socket, validClients.keys()) {
 235     if(validClients[socket] == uid) writeDataToDevice(socket, QVariant(sigdata));
 236   }
 237 }
 238
 239 /*
 240   // Read global settings from config file
 241   QSettings s;
 242   s.beginGroup("Global");
 243   QString key;
 244   foreach(key, s.childKeys()) {
 245     global->updateData(key, s.value(key));
 246   }
 247
 248   global->updateData("CoreReady", true);
 249   // Now that we are in sync, we can connect signals to automatically store further updates.
 250   // I don't think we care if global data changed locally or if it was updated by a client.
 251   connect(global, SIGNAL(dataUpdatedRemotely(QString)), SLOT(globalDataUpdated(QString)));
 252   connect(global, SIGNAL(dataPutLocally(QString)), SLOT(globalDataUpdated(QString)));
 253
 254 }
 255   */
 256
 257 CoreSession::CoreSession(UserId uid, Storage *_storage) : user(uid), storage(_storage) {
 258   coreProxy = new CoreProxy();
 259
 260   connect(coreProxy, SIGNAL(send(CoreSignal, QVariant, QVariant, QVariant)), this, SIGNAL(proxySignal(CoreSignal, QVariant, QVariant, QVariant)));
 261   connect(coreProxy, SIGNAL(requestServerStates()), this, SIGNAL(serverStateRequested()));
 262   connect(coreProxy, SIGNAL(gsRequestConnect(QStringList)), this, SLOT(connectToIrc(QStringList)));
 263   connect(coreProxy, SIGNAL(gsUserInput(BufferId, QString)), this, SLOT(msgFromGui(BufferId, QString)));
 264   connect(coreProxy, SIGNAL(gsImportBacklog()), storage, SLOT(importOldBacklog()));
 265   connect(coreProxy, SIGNAL(gsRequestBacklog(BufferId, QVariant, QVariant)), this, SLOT(sendBacklog(BufferId, QVariant, QVariant)));
 266   connect(this, SIGNAL(displayMsg(Message)), coreProxy, SLOT(csDisplayMsg(Message)));
 267   connect(this, SIGNAL(displayStatusMsg(QString, QString)), coreProxy, SLOT(csDisplayStatusMsg(QString, QString)));
 268   connect(this, SIGNAL(backlogData(BufferId, QList<QVariant>, bool)), coreProxy, SLOT(csBacklogData(BufferId, QList<QVariant>, bool)));
 269   connect(this, SIGNAL(bufferIdUpdated(BufferId)), coreProxy, SLOT(csUpdateBufferId(BufferId)));
 270   connect(storage, SIGNAL(bufferIdUpdated(BufferId)), coreProxy, SLOT(csUpdateBufferId(BufferId)));
 271   connect(Global::instance(), SIGNAL(dataUpdatedRemotely(UserId, QString)), this, SLOT(globalDataUpdated(UserId, QString)));
 272   connect(Global::instance(), SIGNAL(dataPutLocally(UserId, QString)), this, SLOT(globalDataUpdated(UserId, QString)));
 273 }
 274
 275 CoreSession::~CoreSession() {
 276
 277 }
 278
 279 UserId CoreSession::userId() {
 280   return user;
 281 }
 282
 283 void CoreSession::processSignal(ClientSignal sig, QVariant arg1, QVariant arg2, QVariant arg3) {
 284   coreProxy->recv(sig, arg1, arg2, arg3);
 285 }
 286
 287 void CoreSession::globalDataUpdated(UserId uid, QString key) {
 288   Q_ASSERT(uid == userId());
 289   QVariant data = Global::data(userId(), key);
 290   QSettings s;
 291   s.setValue(QString("Global/%1/").arg(userId())+key, data);
 292 }
 293
 294 void CoreSession::connectToIrc(QStringList networks) {
 295   foreach(QString net, networks) {
 296     if(servers.contains(net)) {
 297
 298     } else {
 299       Server *server = new Server(userId(), net);
 300       connect(this, SIGNAL(serverStateRequested()), server, SLOT(sendState()));
 301       connect(this, SIGNAL(connectToIrc(QString)), server, SLOT(connectToIrc(QString)));
 302       connect(this, SIGNAL(disconnectFromIrc(QString)), server, SLOT(disconnectFromIrc(QString)));
 303       connect(this, SIGNAL(msgFromGui(QString, QString, QString)), server, SLOT(userInput(QString, QString, QString)));
 304
 305       connect(server, SIGNAL(connected(QString)), this, SLOT(serverConnected(QString)));
 306       connect(server, SIGNAL(disconnected(QString)), this, SLOT(serverDisconnected(QString)));
 307
 308       connect(server, SIGNAL(serverState(QString, VarMap)), coreProxy, SLOT(csServerState(QString, VarMap)));
 309       //connect(server, SIGNAL(displayMsg(Message)), this, SLOT(recvMessageFromServer(Message)));
 310       connect(server, SIGNAL(displayMsg(Message::Type, QString, QString, QString, quint8)), this, SLOT(recvMessageFromServer(Message::Type, QString, QString, QString, quint8)));
 311       connect(server, SIGNAL(displayStatusMsg(QString)), this, SLOT(recvStatusMsgFromServer(QString)));
 312       connect(server, SIGNAL(modeSet(QString, QString, QString)), coreProxy, SLOT(csModeSet(QString, QString, QString)));
 313       connect(server, SIGNAL(topicSet(QString, QString, QString)), coreProxy, SLOT(csTopicSet(QString, QString, QString)));
 314       connect(server, SIGNAL(nickAdded(QString, QString, VarMap)), coreProxy, SLOT(csNickAdded(QString, QString, VarMap)));
 315       connect(server, SIGNAL(nickRenamed(QString, QString, QString)), coreProxy, SLOT(csNickRenamed(QString, QString, QString)));
 316       connect(server, SIGNAL(nickRemoved(QString, QString)), coreProxy, SLOT(csNickRemoved(QString, QString)));
 317       connect(server, SIGNAL(nickUpdated(QString, QString, VarMap)), coreProxy, SLOT(csNickUpdated(QString, QString, VarMap)));
 318       connect(server, SIGNAL(ownNickSet(QString, QString)), coreProxy, SLOT(csOwnNickSet(QString, QString)));
 319       connect(server, SIGNAL(queryRequested(QString, QString)), coreProxy, SLOT(csQueryRequested(QString, QString)));
 320       // TODO add error handling
 321       connect(server, SIGNAL(connected(QString)), coreProxy, SLOT(csServerConnected(QString)));
 322       connect(server, SIGNAL(disconnected(QString)), coreProxy, SLOT(csServerDisconnected(QString)));
 323
 324       server->start();
 325       servers[net] = server;
 326     }
 327     emit connectToIrc(net);
 328   }
 329 }
 330
 331 void CoreSession::serverConnected(QString net) {
 332   storage->getBufferId(userId(), net); // create status buffer
 333 }
 334
 335 void CoreSession::serverDisconnected(QString net) {
 336   delete servers[net];
 337   servers.remove(net);
 338   coreProxy->csServerDisconnected(net);
 339 }
 340
 341 void CoreSession::msgFromGui(BufferId bufid, QString msg) {
 342   emit msgFromGui(bufid.network(), bufid.buffer(), msg);
 343 }
 344
 345 // ALL messages coming pass through these functions before going to the GUI.
 346 // So this is the perfect place for storing the backlog and log stuff.
 347
 348 void CoreSession::recvMessageFromServer(Message::Type type, QString target, QString text, QString sender, quint8 flags) {
 349   Server *s = qobject_cast<Server*>(this->sender());
 350   Q_ASSERT(s);
 351   BufferId buf;
 352   if((flags & Message::PrivMsg) && !(flags & Message::Self)) {
 353     buf = storage->getBufferId(user, s->getNetwork(), nickFromMask(sender));
 354   } else {
 355     buf = storage->getBufferId(user, s->getNetwork(), target);
 356   }
 357   Message msg(buf, type, text, sender, flags);
 358   msg.msgId = storage->logMessage(msg); //qDebug() << msg.msgId;
 359   Q_ASSERT(msg.msgId);
 360   emit displayMsg(msg);
 361 }
 362
 363 void CoreSession::recvStatusMsgFromServer(QString msg) {
 364   Server *s = qobject_cast<Server*>(sender());
 365   Q_ASSERT(s);
 366   emit displayStatusMsg(s->getNetwork(), msg);
 367 }
 368
 369
 370 QList<BufferId> CoreSession::buffers() const {
 371   return storage->requestBuffers(user);
 372 }
 373
 374
 375 QVariant CoreSession::sessionState() {
 376   VarMap v;
 377   QList<QVariant> bufs;
 378   foreach(BufferId id, storage->requestBuffers(user)) { bufs.append(QVariant::fromValue(id)); }
 379   v["Buffers"] = bufs;
 380
 381   return v;
 382 }
 383
 384 void CoreSession::sendServerStates() {
 385   emit serverStateRequested();
 386 }
 387
 388 void CoreSession::sendBacklog(BufferId id, QVariant v1, QVariant v2) {
 389   QList<QVariant> log;
 390   QList<Message> msglist;
 391   if(v1.type() == QVariant::DateTime) {
 392
 393
 394   } else {
 395     msglist = storage->requestMsgs(id, v1.toInt(), v2.toInt());
 396   }
 397
 398   // Send messages out in smaller packages - we don't want to make the signal data too large!
 399   for(int i = 0; i < msglist.count(); i++) {
 400     log.append(QVariant::fromValue(msglist[i]));
 401     if(log.count() >= 5) {
 402       emit backlogData(id, log, i >= msglist.count() - 1);
 403       log.clear();
 404     }
 405   }
 406   if(log.count() > 0) emit backlogData(id, log, true);
 407 }
 408
 409
 410 //Core *core = 0;